mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 03:02:36 +01:00
a30052855f
If exiting using _Exit or ExitProcess, DLLs are still unloaded cleanly before exiting, running destructors and other cleanup in those DLLs. When the caller expects to exit without cleanup, running destructors in some loaded DLLs (which can be either libLLVM.dll or e.g. libc++.dll) can cause deadlocks occasionally. This is an alternative to D102684. Differential Revision: https://reviews.llvm.org/D102944
113 lines
3.4 KiB
C++
113 lines
3.4 KiB
C++
//===-- 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"
|
|
|
|
#include <stdlib.h> // for _Exit
|
|
|
|
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, char Separator) {
|
|
return FindInEnvPath(EnvName, FileName, {}, Separator);
|
|
}
|
|
|
|
Optional<std::string> Process::FindInEnvPath(StringRef EnvName,
|
|
StringRef FileName,
|
|
ArrayRef<std::string> IgnoreList,
|
|
char Separator) {
|
|
assert(!path::is_absolute(FileName));
|
|
Optional<std::string> FoundPath;
|
|
Optional<std::string> OptPath = Process::GetEnv(EnvName);
|
|
if (!OptPath.hasValue())
|
|
return FoundPath;
|
|
|
|
const char EnvPathSeparatorStr[] = {Separator, '\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, bool NoCleanup) {
|
|
if (CrashRecoveryContext *CRC = CrashRecoveryContext::GetCurrent())
|
|
CRC->HandleExit(RetCode);
|
|
|
|
if (NoCleanup)
|
|
ExitNoCleanup(RetCode);
|
|
else
|
|
::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
|