1//===-- Process.cpp - Implement OS Process Concept --------------*- C++ -*-===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file implements the operating system Process concept. 11// 12//===----------------------------------------------------------------------===// 13 14#include "llvm/ADT/StringExtras.h" 15#include "llvm/Config/config.h" 16#include "llvm/Support/FileSystem.h" 17#include "llvm/Support/Path.h" 18#include "llvm/Support/Process.h" 19#include "llvm/Support/Program.h" 20 21using namespace llvm; 22using namespace sys; 23 24//===----------------------------------------------------------------------===// 25//=== WARNING: Implementation here must contain only TRULY operating system 26//=== independent code. 27//===----------------------------------------------------------------------===// 28 29Optional<std::string> Process::FindInEnvPath(const std::string& EnvName, 30 const std::string& FileName) 31{ 32 assert(!path::is_absolute(FileName)); 33 Optional<std::string> FoundPath; 34 Optional<std::string> OptPath = Process::GetEnv(EnvName); 35 if (!OptPath.hasValue()) 36 return FoundPath; 37 38 const char EnvPathSeparatorStr[] = {EnvPathSeparator, '\0'}; 39 SmallVector<StringRef, 8> Dirs; 40 SplitString(OptPath.getValue(), Dirs, EnvPathSeparatorStr); 41 42 for (const auto &Dir : Dirs) { 43 if (Dir.empty()) 44 continue; 45 46 SmallString<128> FilePath(Dir); 47 path::append(FilePath, FileName); 48 if (fs::exists(Twine(FilePath))) { 49 FoundPath = FilePath.str(); 50 break; 51 } 52 } 53 54 return FoundPath; 55} 56 57 58#define COLOR(FGBG, CODE, BOLD) "\033[0;" BOLD FGBG CODE "m" 59 60#define ALLCOLORS(FGBG,BOLD) {\ 61 COLOR(FGBG, "0", BOLD),\ 62 COLOR(FGBG, "1", BOLD),\ 63 COLOR(FGBG, "2", BOLD),\ 64 COLOR(FGBG, "3", BOLD),\ 65 COLOR(FGBG, "4", BOLD),\ 66 COLOR(FGBG, "5", BOLD),\ 67 COLOR(FGBG, "6", BOLD),\ 68 COLOR(FGBG, "7", BOLD)\ 69 } 70 71static const char colorcodes[2][2][8][10] = { 72 { ALLCOLORS("3",""), ALLCOLORS("3","1;") }, 73 { ALLCOLORS("4",""), ALLCOLORS("4","1;") } 74}; 75 76// This is set to true when Process::PreventCoreFiles() is called. 77static bool coreFilesPrevented = false; 78 79bool Process::AreCoreFilesPrevented() { 80 return coreFilesPrevented; 81} 82 83// Include the platform-specific parts of this class. 84#ifdef LLVM_ON_UNIX 85#include "Unix/Process.inc" 86#endif 87#ifdef LLVM_ON_WIN32 88#include "Windows/Process.inc" 89#endif 90