naiveproxy/tools/gn/util/exe_path.cc
2018-08-14 22:19:20 +00:00

62 lines
1.7 KiB
C++

// Copyright 2018 The Chromium 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 "util/exe_path.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/strings/string_util.h"
#include "util/build_config.h"
#if defined(OS_MACOSX)
#include <mach-o/dyld.h>
#elif defined(OS_WIN)
#include <windows.h>
#endif
#if defined(OS_MACOSX)
base::FilePath GetExePath() {
// Executable path can have relative references ("..") depending on
// how the app was launched.
uint32_t executable_length = 0;
_NSGetExecutablePath(NULL, &executable_length);
DCHECK_GT(executable_length, 1u);
std::string executable_path;
int rv = _NSGetExecutablePath(
base::WriteInto(&executable_path, executable_length), &executable_length);
DCHECK_EQ(rv, 0);
// _NSGetExecutablePath may return paths containing ./ or ../ which makes
// FilePath::DirName() work incorrectly, convert it to absolute path so that
// paths such as DIR_SOURCE_ROOT can work, since we expect absolute paths to
// be returned here.
return base::MakeAbsoluteFilePath(base::FilePath(executable_path));
}
#elif defined(OS_WIN)
base::FilePath GetExePath() {
wchar_t system_buffer[MAX_PATH];
system_buffer[0] = 0;
if (GetModuleFileName(NULL, system_buffer, MAX_PATH) == 0) {
return base::FilePath();
}
return base::FilePath(system_buffer);
}
#else
base::FilePath GetExePath() {
base::FilePath result;
const char kProcSelfExe[] = "/proc/self/exe";
if (!ReadSymbolicLink(base::FilePath(kProcSelfExe), &result)) {
NOTREACHED() << "Unable to resolve " << kProcSelfExe << ".";
return base::FilePath();
}
return result;
}
#endif