| // Copyright 2015 PDFium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "testing/utils/path_service.h" |
| |
| #ifdef _WIN32 |
| #include <Windows.h> |
| #elif defined(__APPLE__) |
| #include <mach-o/dyld.h> |
| #else // Linux |
| #include <linux/limits.h> |
| #include <unistd.h> |
| #endif // _WIN32 |
| |
| #include <string> |
| |
| #include "core/fxcrt/include/fx_system.h" |
| |
| // static |
| bool PathService::EndsWithSeparator(const std::string& path) { |
| return path.size() > 1 && path[path.size() - 1] == PATH_SEPARATOR; |
| } |
| |
| // static |
| bool PathService::GetExecutableDir(std::string* path) { |
| // Get the current executable file path. |
| #ifdef _WIN32 |
| char path_buffer[MAX_PATH]; |
| path_buffer[0] = 0; |
| |
| if (GetModuleFileNameA(NULL, path_buffer, MAX_PATH) == 0) |
| return false; |
| *path = std::string(path_buffer); |
| #elif defined(__APPLE__) |
| ASSERT(path); |
| unsigned int path_length = 0; |
| _NSGetExecutablePath(NULL, &path_length); |
| if (path_length == 0) |
| return false; |
| |
| path->reserve(path_length); |
| path->resize(path_length - 1); |
| if (_NSGetExecutablePath(&((*path)[0]), &path_length)) |
| return false; |
| #else // Linux |
| static const char kProcSelfExe[] = "/proc/self/exe"; |
| char buf[PATH_MAX]; |
| ssize_t count = ::readlink(kProcSelfExe, buf, PATH_MAX); |
| if (count <= 0) |
| return false; |
| |
| *path = std::string(buf, count); |
| #endif // _WIN32 |
| |
| // Get the directory path. |
| std::size_t pos = path->size() - 1; |
| if (EndsWithSeparator(*path)) |
| pos--; |
| std::size_t found = path->find_last_of(PATH_SEPARATOR, pos); |
| if (found == std::string::npos) |
| return false; |
| path->resize(found); |
| return true; |
| } |
| |
| // static |
| bool PathService::GetSourceDir(std::string* path) { |
| if (!GetExecutableDir(path)) |
| return false; |
| |
| if (!EndsWithSeparator(*path)) |
| path->push_back(PATH_SEPARATOR); |
| path->append(".."); |
| path->push_back(PATH_SEPARATOR); |
| path->append(".."); |
| return true; |
| } |
| |
| // static |
| bool PathService::GetTestDataDir(std::string* path) { |
| if (!GetSourceDir(path)) |
| return false; |
| |
| if (!EndsWithSeparator(*path)) |
| path->push_back(PATH_SEPARATOR); |
| path->append("testing"); |
| path->push_back(PATH_SEPARATOR); |
| path->append("resources"); |
| return true; |
| } |
| |
| // static |
| bool PathService::GetTestFilePath(const std::string& file_name, |
| std::string* path) { |
| if (!GetTestDataDir(path)) |
| return false; |
| |
| if (!EndsWithSeparator(*path)) |
| path->push_back(PATH_SEPARATOR); |
| path->append(file_name); |
| return true; |
| } |