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