1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
|
// 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/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;
}
|