X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FSupport%2FProcess.cpp;h=0d42e0e35b92e7e3691d7b6eb7092375efe03978;hb=9c23685dd25370a9a255052e44fa609b8ce455c0;hp=d5168f03a6de0686c0b0692b7b84b34c8b01fdb3;hpb=91b585bccc23fafe004024c47fa555a638547a2b;p=oota-llvm.git diff --git a/lib/Support/Process.cpp b/lib/Support/Process.cpp index d5168f03a6d..0d42e0e35b9 100644 --- a/lib/Support/Process.cpp +++ b/lib/Support/Process.cpp @@ -7,13 +7,16 @@ // //===----------------------------------------------------------------------===// // -// This header file implements the operating system Process concept. +// This file implements the operating system Process concept. // //===----------------------------------------------------------------------===// +#include "llvm/ADT/StringExtras.h" #include "llvm/Config/config.h" #include "llvm/Support/ErrorHandling.h" +#include "llvm/Support/FileSystem.h" #include "llvm/Support/Process.h" +#include "llvm/Support/Program.h" using namespace llvm; using namespace sys; @@ -34,14 +37,6 @@ self_process *process::get_self() { return SP; } -#if defined(_MSC_VER) -// Visual Studio complains that the self_process destructor never exits. This -// doesn't make much sense, as that's the whole point of calling abort... Just -// silence this warning. -#pragma warning(push) -#pragma warning(disable:4722) -#endif - // The destructor for the self_process subclass must never actually be // executed. There should be at most one instance of this class, and that // instance should live until the process terminates to avoid the potential for @@ -74,10 +69,32 @@ TimeValue self_process::get_wall_time() const { return getElapsedWallTime(); } +Optional Process::FindInEnvPath(const std::string& EnvName, + const std::string& FileName) +{ + Optional FoundPath; + Optional OptPath = Process::GetEnv(EnvName); + if (!OptPath.hasValue()) + return FoundPath; + + const char EnvPathSeparatorStr[] = {EnvPathSeparator, '\0'}; + SmallVector Dirs; + SplitString(OptPath.getValue(), Dirs, EnvPathSeparatorStr); + + for (const auto &Dir : Dirs) { + if (Dir.empty()) + continue; + + SmallString<128> FilePath(Dir); + path::append(FilePath, FileName); + if (fs::exists(Twine(FilePath))) { + FoundPath = FilePath.str(); + break; + } + } -#if defined(_MSC_VER) -#pragma warning(pop) -#endif + return FoundPath; +} #define COLOR(FGBG, CODE, BOLD) "\033[0;" BOLD FGBG CODE "m"