llvm/lib/Support/Process.cpp

   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
  23 #include <stdlib.h> // for _Exit
  24
  25 using namespace llvm;
  26 using namespace sys;
  27
  28 //===----------------------------------------------------------------------===//
  29 //=== WARNING: Implementation here must contain only TRULY operating system
  30 //===          independent code.
  31 //===----------------------------------------------------------------------===//
  32
  33 Optional<std::string>
  34 Process::FindInEnvPath(StringRef EnvName, StringRef FileName, char Separator) {
  35   return FindInEnvPath(EnvName, FileName, {}, Separator);
  36 }
  37
  38 Optional<std::string> Process::FindInEnvPath(StringRef EnvName,
  39                                              StringRef FileName,
  40                                              ArrayRef<std::string> IgnoreList,
  41                                              char Separator) {
  42   assert(!path::is_absolute(FileName));
  43   Optional<std::string> FoundPath;
  44   Optional<std::string> OptPath = Process::GetEnv(EnvName);
  45   if (!OptPath.hasValue())
  46     return FoundPath;
  47
  48   const char EnvPathSeparatorStr[] = {Separator, '\0'};
  49   SmallVector<StringRef, 8> Dirs;
  50   SplitString(OptPath.getValue(), Dirs, EnvPathSeparatorStr);
  51
  52   for (StringRef Dir : Dirs) {
  53     if (Dir.empty())
  54       continue;
  55
  56     if (any_of(IgnoreList, [&](StringRef S) { return fs::equivalent(S, Dir); }))
  57       continue;
  58
  59     SmallString<128> FilePath(Dir);
  60     path::append(FilePath, FileName);
  61     if (fs::exists(Twine(FilePath))) {
  62       FoundPath = std::string(FilePath.str());
  63       break;
  64     }
  65   }
  66
  67   return FoundPath;
  68 }
  69
  70
  71 #define COLOR(FGBG, CODE, BOLD) "\033[0;" BOLD FGBG CODE "m"
  72
  73 #define ALLCOLORS(FGBG,BOLD) {\
  74     COLOR(FGBG, "0", BOLD),\
  75     COLOR(FGBG, "1", BOLD),\
  76     COLOR(FGBG, "2", BOLD),\
  77     COLOR(FGBG, "3", BOLD),\
  78     COLOR(FGBG, "4", BOLD),\
  79     COLOR(FGBG, "5", BOLD),\
  80     COLOR(FGBG, "6", BOLD),\
  81     COLOR(FGBG, "7", BOLD)\
  82   }
  83
  84 static const char colorcodes[2][2][8][10] = {
  85  { ALLCOLORS("3",""), ALLCOLORS("3","1;") },
  86  { ALLCOLORS("4",""), ALLCOLORS("4","1;") }
  87 };
  88
  89 // A CMake option controls wheter we emit core dumps by default. An application
  90 // may disable core dumps by calling Process::PreventCoreFiles().
  91 static bool coreFilesPrevented = !LLVM_ENABLE_CRASH_DUMPS;
  92
  93 bool Process::AreCoreFilesPrevented() { return coreFilesPrevented; }
  94
  95 [[noreturn]] void Process::Exit(int RetCode, bool NoCleanup) {
  96   if (CrashRecoveryContext *CRC = CrashRecoveryContext::GetCurrent())
  97     CRC->HandleExit(RetCode);
  98
  99   if (NoCleanup)
 100     ExitNoCleanup(RetCode);
 101   else
 102     ::exit(RetCode);
 103 }
 104
 105 // Include the platform-specific parts of this class.
 106 #ifdef LLVM_ON_UNIX
 107 #include "Unix/Process.inc"
 108 #endif
 109 #ifdef _WIN32
 110 #include "Windows/Process.inc"
 111 #endif