22fd11fe66
This AutoConvert.h header frequently gets mislabeled as an unused include because it is guarded by MVS internally and every usage is also guarded. This refactors the change to remove this guard and instead make these functions a noop on other non-z/OS platforms.
113 lines
3.8 KiB
C++
113 lines
3.8 KiB
C++
//===-- InitLLVM.cpp -----------------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Support/AutoConvert.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/ManagedStatic.h"
|
|
#include "llvm/Support/Signals.h"
|
|
|
|
#ifdef _WIN32
|
|
#include "llvm/Support/Windows/WindowsSupport.h"
|
|
#endif
|
|
|
|
#if defined(HAVE_UNISTD_H)
|
|
#include <unistd.h>
|
|
#else
|
|
#ifndef STDIN_FILENO
|
|
#define STDIN_FILENO 0
|
|
#endif
|
|
#ifndef STDOUT_FILENO
|
|
#define STDOUT_FILENO 1
|
|
#endif
|
|
#ifndef STDERR_FILENO
|
|
#define STDERR_FILENO 2
|
|
#endif
|
|
#endif
|
|
|
|
void CleanupStdHandles(void *Cookie) {
|
|
llvm::raw_ostream *Outs = &llvm::outs(), *Errs = &llvm::errs();
|
|
Outs->flush();
|
|
Errs->flush();
|
|
llvm::restoreStdHandleAutoConversion(STDIN_FILENO);
|
|
llvm::restoreStdHandleAutoConversion(STDOUT_FILENO);
|
|
llvm::restoreStdHandleAutoConversion(STDERR_FILENO);
|
|
}
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::sys;
|
|
|
|
InitLLVM::InitLLVM(int &Argc, const char **&Argv,
|
|
bool InstallPipeSignalExitHandler) {
|
|
#ifndef NDEBUG
|
|
static std::atomic<bool> Initialized{false};
|
|
assert(!Initialized && "InitLLVM was already initialized!");
|
|
Initialized = true;
|
|
#endif
|
|
|
|
// Bring stdin/stdout/stderr into a known state.
|
|
sys::AddSignalHandler(CleanupStdHandles, nullptr);
|
|
|
|
if (InstallPipeSignalExitHandler)
|
|
// The pipe signal handler must be installed before any other handlers are
|
|
// registered. This is because the Unix \ref RegisterHandlers function does
|
|
// not perform a sigaction() for SIGPIPE unless a one-shot handler is
|
|
// present, to allow long-lived processes (like lldb) to fully opt-out of
|
|
// llvm's SIGPIPE handling and ignore the signal safely.
|
|
sys::SetOneShotPipeSignalFunction(sys::DefaultOneShotPipeSignalHandler);
|
|
// Initialize the stack printer after installing the one-shot pipe signal
|
|
// handler, so we can perform a sigaction() for SIGPIPE on Unix if requested.
|
|
StackPrinter.emplace(Argc, Argv);
|
|
sys::PrintStackTraceOnErrorSignal(Argv[0]);
|
|
install_out_of_memory_new_handler();
|
|
|
|
#ifdef __MVS__
|
|
|
|
// We use UTF-8 as the internal character encoding. On z/OS, all external
|
|
// output is encoded in EBCDIC. In order to be able to read all
|
|
// error messages, we turn conversion to EBCDIC on for stderr fd.
|
|
std::string Banner = std::string(Argv[0]) + ": ";
|
|
ExitOnError ExitOnErr(Banner);
|
|
|
|
// If turning on conversion for stderr fails then the error message
|
|
// may be garbled. There is no solution to this problem.
|
|
ExitOnErr(errorCodeToError(llvm::enableAutoConversion(STDERR_FILENO)));
|
|
ExitOnErr(errorCodeToError(llvm::enableAutoConversion(STDOUT_FILENO)));
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
// We use UTF-8 as the internal character encoding. On Windows,
|
|
// arguments passed to main() may not be encoded in UTF-8. In order
|
|
// to reliably detect encoding of command line arguments, we use an
|
|
// Windows API to obtain arguments, convert them to UTF-8, and then
|
|
// write them back to the Argv vector.
|
|
//
|
|
// There's probably other way to do the same thing (e.g. using
|
|
// wmain() instead of main()), but this way seems less intrusive
|
|
// than that.
|
|
std::string Banner = std::string(Argv[0]) + ": ";
|
|
ExitOnError ExitOnErr(Banner);
|
|
|
|
ExitOnErr(errorCodeToError(windows::GetCommandLineArguments(Args, Alloc)));
|
|
|
|
// GetCommandLineArguments doesn't terminate the vector with a
|
|
// nullptr. Do it to make it compatible with the real argv.
|
|
Args.push_back(nullptr);
|
|
|
|
Argc = Args.size() - 1;
|
|
Argv = Args.data();
|
|
#endif
|
|
}
|
|
|
|
InitLLVM::~InitLLVM() {
|
|
CleanupStdHandles(nullptr);
|
|
llvm_shutdown();
|
|
}
|