1 //===-- InitLLVM.cpp -----------------------------------------------------===// 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 #include "llvm/Support/InitLLVM.h" 10 #include "llvm/ADT/StringRef.h" 11 #include "llvm/Support/ErrorHandling.h" 12 #include "llvm/Support/ManagedStatic.h" 13 #include "llvm/Support/PrettyStackTrace.h" 14 #include "llvm/Support/Signals.h" 15 #include "llvm/Support/SwapByteOrder.h" 16 17 #ifdef _WIN32 18 #include "llvm/Support/Error.h" 19 #include "llvm/Support/Windows/WindowsSupport.h" 20 #endif 21 22 using namespace llvm; 23 using namespace llvm::sys; 24 25 InitLLVM::InitLLVM(int &Argc, const char **&Argv, 26 bool InstallPipeSignalExitHandler) { 27 if (InstallPipeSignalExitHandler) 28 // The pipe signal handler must be installed before any other handlers are 29 // registered. This is because the Unix \ref RegisterHandlers function does 30 // not perform a sigaction() for SIGPIPE unless a one-shot handler is 31 // present, to allow long-lived processes (like lldb) to fully opt-out of 32 // llvm's SIGPIPE handling and ignore the signal safely. 33 sys::SetOneShotPipeSignalFunction(sys::DefaultOneShotPipeSignalHandler); 34 // Initialize the stack printer after installing the one-shot pipe signal 35 // handler, so we can perform a sigaction() for SIGPIPE on Unix if requested. 36 StackPrinter.emplace(Argc, Argv); 37 sys::PrintStackTraceOnErrorSignal(Argv[0]); 38 install_out_of_memory_new_handler(); 39 40 #ifdef _WIN32 41 // We use UTF-8 as the internal character encoding. On Windows, 42 // arguments passed to main() may not be encoded in UTF-8. In order 43 // to reliably detect encoding of command line arguments, we use an 44 // Windows API to obtain arguments, convert them to UTF-8, and then 45 // write them back to the Argv vector. 46 // 47 // There's probably other way to do the same thing (e.g. using 48 // wmain() instead of main()), but this way seems less intrusive 49 // than that. 50 std::string Banner = std::string(Argv[0]) + ": "; 51 ExitOnError ExitOnErr(Banner); 52 53 ExitOnErr(errorCodeToError(windows::GetCommandLineArguments(Args, Alloc))); 54 55 // GetCommandLineArguments doesn't terminate the vector with a 56 // nullptr. Do it to make it compatible with the real argv. 57 Args.push_back(nullptr); 58 59 Argc = Args.size() - 1; 60 Argv = Args.data(); 61 #endif 62 } 63 64 InitLLVM::~InitLLVM() { llvm_shutdown(); } 65