//===-- Process.cpp - Implement OS Process Concept --------------*- C++ -*-===//

//

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.

// See https://llvm.org/LICENSE.txt for license information.

// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

//

//===----------------------------------------------------------------------===//

//

//  This file implements the operating system Process concept.

//

//===----------------------------------------------------------------------===//

#include "llvm/Support/Process.h"

#include "llvm/ADT/STLExtras.h"

#include "llvm/ADT/StringExtras.h"

#include "llvm/Config/config.h"

#include "llvm/Config/llvm-config.h"

#include "llvm/Support/CrashRecoveryContext.h"

#include "llvm/Support/FileSystem.h"

#include "llvm/Support/Path.h"

#include "llvm/Support/Program.h"

using namespace llvm;

using namespace sys;

//===----------------------------------------------------------------------===//

//=== WARNING: Implementation here must contain only TRULY operating system

//===          independent code.

//===----------------------------------------------------------------------===//

Optional<std::string> Process::FindInEnvPath(StringRef EnvName,

                                             StringRef FileName) {

  return FindInEnvPath(EnvName, FileName, {});

}

Optional<std::string> Process::FindInEnvPath(StringRef EnvName,

                                             StringRef FileName,

                                             ArrayRef<std::string> IgnoreList) {

  assert(!path::is_absolute(FileName));

  Optional<std::string> FoundPath;

  Optional<std::string> OptPath = Process::GetEnv(EnvName);

  if (!OptPath.hasValue())

    return FoundPath;

  const char EnvPathSeparatorStr[] = {EnvPathSeparator, '\0'};

  SmallVector<StringRef, 8> Dirs;

  SplitString(OptPath.getValue(), Dirs, EnvPathSeparatorStr);

  for (StringRef Dir : Dirs) {

    if (Dir.empty())

      continue;

    if (any_of(IgnoreList, [&](StringRef S) { return fs::equivalent(S, Dir); }))

      continue;

    SmallString<128> FilePath(Dir);

    path::append(FilePath, FileName);

    if (fs::exists(Twine(FilePath))) {

      FoundPath = std::string(FilePath.str());

      break;

    }

  }

  return FoundPath;

}

#define COLOR(FGBG, CODE, BOLD) "\033[0;" BOLD FGBG CODE "m"

#define ALLCOLORS(FGBG,BOLD) {\

    COLOR(FGBG, "0", BOLD),\

    COLOR(FGBG, "1", BOLD),\

    COLOR(FGBG, "2", BOLD),\

    COLOR(FGBG, "3", BOLD),\

    COLOR(FGBG, "4", BOLD),\

    COLOR(FGBG, "5", BOLD),\

    COLOR(FGBG, "6", BOLD),\

    COLOR(FGBG, "7", BOLD)\

  }

static const char colorcodes[2][2][8][10] = {

 { ALLCOLORS("3",""), ALLCOLORS("3","1;") },

 { ALLCOLORS("4",""), ALLCOLORS("4","1;") }

};

// A CMake option controls wheter we emit core dumps by default. An application

// may disable core dumps by calling Process::PreventCoreFiles().

static bool coreFilesPrevented = !LLVM_ENABLE_CRASH_DUMPS;

bool Process::AreCoreFilesPrevented() { return coreFilesPrevented; }

LLVM_ATTRIBUTE_NORETURN

void Process::Exit(int RetCode) {

  if (CrashRecoveryContext *CRC = CrashRecoveryContext::GetCurrent())

    CRC->HandleExit(RetCode);

  ::exit(RetCode);

}

// Include the platform-specific parts of this class.

#ifdef LLVM_ON_UNIX

#include "Unix/Process.inc"

#endif

#ifdef _WIN32

#include "Windows/Process.inc"

#endif