1//===-- Process.cpp - Implement OS Process Concept --------------*- C++ -*-===// 2// 3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4// See https://llvm.org/LICENSE.txt for license information. 5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6// 7//===----------------------------------------------------------------------===// 8// 9// This file implements the operating system Process concept. 10// 11//===----------------------------------------------------------------------===// 12 13#include "llvm/Support/Process.h" 14#include "llvm/ADT/STLExtras.h" 15#include "llvm/ADT/StringExtras.h" 16#include "llvm/Config/config.h" 17#include "llvm/Config/llvm-config.h" 18#include "llvm/Support/CrashRecoveryContext.h" 19#include "llvm/Support/FileSystem.h" 20#include "llvm/Support/Path.h" 21#include "llvm/Support/Program.h" 22 23using namespace llvm; 24using namespace sys; 25 26//===----------------------------------------------------------------------===// 27//=== WARNING: Implementation here must contain only TRULY operating system 28//=== independent code. 29//===----------------------------------------------------------------------===// 30 31Optional<std::string> Process::FindInEnvPath(StringRef EnvName, 32 StringRef FileName) { 33 return FindInEnvPath(EnvName, FileName, {}); 34} 35 36Optional<std::string> Process::FindInEnvPath(StringRef EnvName, 37 StringRef FileName, 38 ArrayRef<std::string> IgnoreList) { 39 assert(!path::is_absolute(FileName)); 40 Optional<std::string> FoundPath; 41 Optional<std::string> OptPath = Process::GetEnv(EnvName); 42 if (!OptPath.hasValue()) 43 return FoundPath; 44 45 const char EnvPathSeparatorStr[] = {EnvPathSeparator, '\0'}; 46 SmallVector<StringRef, 8> Dirs; 47 SplitString(OptPath.getValue(), Dirs, EnvPathSeparatorStr); 48 49 for (StringRef Dir : Dirs) { 50 if (Dir.empty()) 51 continue; 52 53 if (any_of(IgnoreList, [&](StringRef S) { return fs::equivalent(S, Dir); })) 54 continue; 55 56 SmallString<128> FilePath(Dir); 57 path::append(FilePath, FileName); 58 if (fs::exists(Twine(FilePath))) { 59 FoundPath = FilePath.str(); 60 break; 61 } 62 } 63 64 return FoundPath; 65} 66 67 68#define COLOR(FGBG, CODE, BOLD) "\033[0;" BOLD FGBG CODE "m" 69 70#define ALLCOLORS(FGBG,BOLD) {\ 71 COLOR(FGBG, "0", BOLD),\ 72 COLOR(FGBG, "1", BOLD),\ 73 COLOR(FGBG, "2", BOLD),\ 74 COLOR(FGBG, "3", BOLD),\ 75 COLOR(FGBG, "4", BOLD),\ 76 COLOR(FGBG, "5", BOLD),\ 77 COLOR(FGBG, "6", BOLD),\ 78 COLOR(FGBG, "7", BOLD)\ 79 } 80 81static const char colorcodes[2][2][8][10] = { 82 { ALLCOLORS("3",""), ALLCOLORS("3","1;") }, 83 { ALLCOLORS("4",""), ALLCOLORS("4","1;") } 84}; 85 86// A CMake option controls wheter we emit core dumps by default. An application 87// may disable core dumps by calling Process::PreventCoreFiles(). 88static bool coreFilesPrevented = !LLVM_ENABLE_CRASH_DUMPS; 89 90bool Process::AreCoreFilesPrevented() { return coreFilesPrevented; } 91 92LLVM_ATTRIBUTE_NORETURN 93void Process::Exit(int RetCode) { 94 if (CrashRecoveryContext *CRC = CrashRecoveryContext::GetCurrent()) 95 CRC->HandleExit(RetCode); 96 ::exit(RetCode); 97} 98 99// Include the platform-specific parts of this class. 100#ifdef LLVM_ON_UNIX 101#include "Unix/Process.inc" 102#endif 103#ifdef _WIN32 104#include "Windows/Process.inc" 105#endif 106