1 //===-- InitLLVM.cpp -----------------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "llvm/Support/InitLLVM.h"
11 #include "llvm/Support/Error.h"
12 #include "llvm/Support/ManagedStatic.h"
13 #include "llvm/Support/PrettyStackTrace.h"
14 #include "llvm/Support/Process.h"
15 #include "llvm/Support/Signals.h"
16 #include <string>
17 
18 using namespace llvm;
19 
20 InitLLVM::InitLLVM(int &Argc, const char **&Argv) : StackPrinter(Argc, Argv) {
21   sys::PrintStackTraceOnErrorSignal(Argv[0]);
22 
23 #ifdef _WIN32
24   // We use UTF-8 as the internal character encoding. On Windows,
25   // arguments passed to main() may not be encoded in UTF-8. In order
26   // to reliably detect encoding of command line arguments, we use an
27   // Windows API to obtain arguments, convert them to UTF-8, and then
28   // write them back to the Argv vector.
29   //
30   // There's probably other way to do the same thing (e.g. using
31   // wmain() instead of main()), but this way seems less intrusive
32   // than that.
33   std::string Banner = std::string(Argv[0]) + ": ";
34   ExitOnError ExitOnErr(Banner);
35 
36   ExitOnErr(errorCodeToError(
37       sys::Process::GetArgumentVector(Args, makeArrayRef(Argv, Argc), Alloc)));
38 
39   // GetArgumentVector doesn't terminate the vector with a nullptr.
40   // Do it to make it compatible with the real argv.
41   Args.push_back(nullptr);
42 
43   Argc = Args.size() - 1;
44   Argv = Args.data();
45 #endif
46 }
47 
48 InitLLVM::~InitLLVM() { llvm_shutdown(); }
49