/* Copyright (c) 2013 Microsoft Corporation. All rights reserved. Released under Apache 2.0 license as described in the file LICENSE. Author: Leonardo de Moura */ #include #include #include #include #include "util/exception.h" #include "util/sstream.h" namespace lean { #if defined(LEAN_WINDOWS) // Windows version #include static char g_path_sep = ';'; static char g_sep = '\\'; static std::string get_exe_location() { HMODULE hModule = GetModuleHandleW(NULL); WCHAR path[MAX_PATH]; GetModuleFileNameW(hModule, path, MAX_PATH); return std::string(path); } #elif defined(__APPLE__) // OSX version #include #include static char g_path_sep = ':'; static char g_sep = '/'; static std::string get_exe_location() { char buf[PATH_MAX]; uint32_t bufsize = PATH_MAX; if (_NSGetExecutablePath(buf, &bufsize) != 0) throw exception("failed to locate Lean executable location"); return std::string(buf); } #else // Linux version #include #include #include #include // NOLINT #include static char g_path_sep = ':'; static char g_sep = '/'; static std::string get_exe_location() { char path[PATH_MAX]; char dest[PATH_MAX]; pid_t pid = getpid(); snprintf(path, PATH_MAX, "/proc/%d/exe", pid); if (readlink(path, dest, PATH_MAX) == -1) { throw exception("failed to locate Lean executable location"); } else { return std::string(dest); } } #endif static std::string g_lean_path; static std::vector g_lean_path_vector; struct init_lean_path { init_lean_path() { char * r = getenv("LEAN_PATH"); if (r == nullptr) { g_lean_path = "."; g_lean_path += g_path_sep; g_lean_path += get_exe_location(); } else { g_lean_path = r; } unsigned i = 0; unsigned j = 0; unsigned sz = g_lean_path.size(); for (; j < sz; j++) { if (g_lean_path[j] == g_path_sep) { if (j > i) g_lean_path_vector.push_back(g_lean_path.substr(i, j - i)); i = j + 1; } } } }; static init_lean_path g_init_lean_path; std::string find_file(char const * fname) { for (auto path : g_lean_path_vector) { auto file = path + g_sep + fname; std::ifstream ifile(file); if (ifile) return file; } throw exception(sstream() << "file '" << fname << "' not found in the LEAN_PATH"); } char const * get_lean_path() { return g_lean_path.c_str(); } }