1// Copyright 2015 PDFium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "testing/utils/path_service.h" 6 7#ifdef _WIN32 8#include <Windows.h> 9#elif defined(__APPLE__) 10#include <mach-o/dyld.h> 11#else // Linux 12#include <linux/limits.h> 13#include <unistd.h> 14#endif // _WIN32 15 16#include <string> 17 18#include "core/fxcrt/fx_system.h" 19 20// static 21bool PathService::EndsWithSeparator(const std::string& path) { 22 return path.size() > 1 && path[path.size() - 1] == PATH_SEPARATOR; 23} 24 25// static 26bool PathService::GetExecutableDir(std::string* path) { 27// Get the current executable file path. 28#ifdef _WIN32 29 char path_buffer[MAX_PATH]; 30 path_buffer[0] = 0; 31 32 if (GetModuleFileNameA(NULL, path_buffer, MAX_PATH) == 0) 33 return false; 34 *path = std::string(path_buffer); 35#elif defined(__APPLE__) 36 ASSERT(path); 37 unsigned int path_length = 0; 38 _NSGetExecutablePath(NULL, &path_length); 39 if (path_length == 0) 40 return false; 41 42 path->reserve(path_length); 43 path->resize(path_length - 1); 44 if (_NSGetExecutablePath(&((*path)[0]), &path_length)) 45 return false; 46#else // Linux 47 static const char kProcSelfExe[] = "/proc/self/exe"; 48 char buf[PATH_MAX]; 49 ssize_t count = ::readlink(kProcSelfExe, buf, PATH_MAX); 50 if (count <= 0) 51 return false; 52 53 *path = std::string(buf, count); 54#endif // _WIN32 55 56 // Get the directory path. 57 std::size_t pos = path->size() - 1; 58 if (EndsWithSeparator(*path)) 59 pos--; 60 std::size_t found = path->find_last_of(PATH_SEPARATOR, pos); 61 if (found == std::string::npos) 62 return false; 63 path->resize(found); 64 return true; 65} 66 67// static 68bool PathService::GetSourceDir(std::string* path) { 69 if (!GetExecutableDir(path)) 70 return false; 71 72 if (!EndsWithSeparator(*path)) 73 path->push_back(PATH_SEPARATOR); 74 path->append(".."); 75 path->push_back(PATH_SEPARATOR); 76#if defined(ANDROID) 77 path->append("chromium_tests_root"); 78#else // Non-Android 79 path->append(".."); 80#endif // defined(ANDROID) 81 return true; 82} 83 84// static 85bool PathService::GetTestDataDir(std::string* path) { 86 if (!GetSourceDir(path)) 87 return false; 88 89 if (!EndsWithSeparator(*path)) 90 path->push_back(PATH_SEPARATOR); 91 path->append("testing"); 92 path->push_back(PATH_SEPARATOR); 93 path->append("resources"); 94 return true; 95} 96 97// static 98bool PathService::GetTestFilePath(const std::string& file_name, 99 std::string* path) { 100 if (!GetTestDataDir(path)) 101 return false; 102 103 if (!EndsWithSeparator(*path)) 104 path->push_back(PATH_SEPARATOR); 105 path->append(file_name); 106 return true; 107} 108