Line data    Source code

       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/Support/Process.h"
      15             : #include "llvm/ADT/STLExtras.h"
      16             : #include "llvm/ADT/StringExtras.h"
      17             : #include "llvm/Config/llvm-config.h"
      18             : #include "llvm/Config/config.h"
      19             : #include "llvm/Support/FileSystem.h"
      20             : #include "llvm/Support/Path.h"
      21             : #include "llvm/Support/Program.h"
      22             : 
      23             : using namespace llvm;
      24             : using namespace sys;
      25             : 
      26             : //===----------------------------------------------------------------------===//
      27             : //=== WARNING: Implementation here must contain only TRULY operating system
      28             : //===          independent code.
      29             : //===----------------------------------------------------------------------===//
      30             : 
      31           7 : Optional<std::string> Process::FindInEnvPath(StringRef EnvName,
      32             :                                              StringRef FileName) {
      33           7 :   return FindInEnvPath(EnvName, FileName, {});
      34             : }
      35             : 
      36          18 : Optional<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           9 :   Optional<std::string> OptPath = Process::GetEnv(EnvName);
      42           9 :   if (!OptPath.hasValue())
      43             :     return FoundPath;
      44             : 
      45           4 :   const char EnvPathSeparatorStr[] = {EnvPathSeparator, '\0'};
      46             :   SmallVector<StringRef, 8> Dirs;
      47           4 :   SplitString(OptPath.getValue(), Dirs, EnvPathSeparatorStr);
      48             : 
      49           5 :   for (StringRef Dir : Dirs) {
      50           4 :     if (Dir.empty())
      51           0 :       continue;
      52             : 
      53           4 :     if (any_of(IgnoreList, [&](StringRef S) { return fs::equivalent(S, Dir); }))
      54             :       continue;
      55             : 
      56             :     SmallString<128> FilePath(Dir);
      57           4 :     path::append(FilePath, FileName);
      58           4 :     if (fs::exists(Twine(FilePath))) {
      59           3 :       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             : 
      81             : static 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().
      88             : static bool coreFilesPrevented = !LLVM_ENABLE_CRASH_DUMPS;
      89             : 
      90           0 : bool Process::AreCoreFilesPrevented() { return coreFilesPrevented; }
      91             : 
      92             : // Include the platform-specific parts of this class.
      93             : #ifdef LLVM_ON_UNIX
      94             : #include "Unix/Process.inc"
      95             : #endif
      96             : #ifdef _WIN32
      97             : #include "Windows/Process.inc"
      98             : #endif