Chris Lattner | fe11a97 | 2002-12-23 23:59:41 +0000 | [diff] [blame] | 1 | //===- lli.cpp - LLVM Interpreter / Dynamic compiler ----------------------===// |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 2 | // |
John Criswell | 7c0e022 | 2003-10-20 17:47:21 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 21c62da | 2007-12-29 20:44:31 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 7 | // |
John Criswell | 7c0e022 | 2003-10-20 17:47:21 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
Chris Lattner | 92101ac | 2001-08-23 17:05:04 +0000 | [diff] [blame] | 9 | // |
Chris Lattner | 7efea1d | 2003-12-26 05:07:35 +0000 | [diff] [blame] | 10 | // This utility provides a simple wrapper around the LLVM Execution Engines, |
| 11 | // which allow the direct execution of LLVM programs through a Just-In-Time |
Torok Edwin | db9c028 | 2009-07-03 12:11:32 +0000 | [diff] [blame] | 12 | // compiler, or through an interpreter if no JIT is available for this platform. |
Chris Lattner | 92101ac | 2001-08-23 17:05:04 +0000 | [diff] [blame] | 13 | // |
| 14 | //===----------------------------------------------------------------------===// |
| 15 | |
Owen Anderson | 8b477ed | 2009-07-01 16:58:40 +0000 | [diff] [blame] | 16 | #include "llvm/LLVMContext.h" |
Chris Lattner | fd13129 | 2003-09-05 20:08:15 +0000 | [diff] [blame] | 17 | #include "llvm/Module.h" |
Chris Lattner | 269a428 | 2003-12-26 06:49:53 +0000 | [diff] [blame] | 18 | #include "llvm/Type.h" |
Duncan Sands | 75ebbce | 2010-08-28 01:30:02 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/Triple.h" |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 20 | #include "llvm/Bitcode/ReaderWriter.h" |
Chris Lattner | d3a680a | 2006-08-01 22:34:35 +0000 | [diff] [blame] | 21 | #include "llvm/CodeGen/LinkAllCodegenComponents.h" |
Brian Gaeke | d1cab3e | 2003-09-05 19:42:34 +0000 | [diff] [blame] | 22 | #include "llvm/ExecutionEngine/GenericValue.h" |
Jeffrey Yasskin | df5a7da | 2009-06-25 02:04:04 +0000 | [diff] [blame] | 23 | #include "llvm/ExecutionEngine/Interpreter.h" |
| 24 | #include "llvm/ExecutionEngine/JIT.h" |
| 25 | #include "llvm/ExecutionEngine/JITEventListener.h" |
Daniel Dunbar | 6aec298 | 2010-11-17 16:06:43 +0000 | [diff] [blame] | 26 | #include "llvm/ExecutionEngine/MCJIT.h" |
Reid Spencer | 551ccae | 2004-09-01 22:55:40 +0000 | [diff] [blame] | 27 | #include "llvm/Support/CommandLine.h" |
Daniel Dunbar | 46a2716 | 2010-11-13 00:28:01 +0000 | [diff] [blame] | 28 | #include "llvm/Support/IRReader.h" |
Chris Lattner | c30598b | 2006-12-06 01:18:01 +0000 | [diff] [blame] | 29 | #include "llvm/Support/ManagedStatic.h" |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 30 | #include "llvm/Support/MemoryBuffer.h" |
Reid Spencer | 551ccae | 2004-09-01 22:55:40 +0000 | [diff] [blame] | 31 | #include "llvm/Support/PluginLoader.h" |
Chris Lattner | cc14d25 | 2009-03-06 05:34:10 +0000 | [diff] [blame] | 32 | #include "llvm/Support/PrettyStackTrace.h" |
Chris Lattner | 74382b7 | 2009-08-23 22:45:37 +0000 | [diff] [blame] | 33 | #include "llvm/Support/raw_ostream.h" |
Michael J. Spencer | 1f6efa3 | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 34 | #include "llvm/Support/Process.h" |
| 35 | #include "llvm/Support/Signals.h" |
Evan Cheng | 3e74d6f | 2011-08-24 18:08:43 +0000 | [diff] [blame] | 36 | #include "llvm/Support/TargetSelect.h" |
Chris Lattner | 0204032 | 2007-04-27 17:02:33 +0000 | [diff] [blame] | 37 | #include <cerrno> |
NAKAMURA Takumi | a13d14a | 2010-10-22 14:53:59 +0000 | [diff] [blame] | 38 | |
| 39 | #ifdef __CYGWIN__ |
| 40 | #include <cygwin/version.h> |
| 41 | #if defined(CYGWIN_VERSION_DLL_MAJOR) && CYGWIN_VERSION_DLL_MAJOR<1007 |
| 42 | #define DO_NOTHING_ATEXIT 1 |
| 43 | #endif |
| 44 | #endif |
| 45 | |
Brian Gaeke | d0fde30 | 2003-11-11 22:41:34 +0000 | [diff] [blame] | 46 | using namespace llvm; |
| 47 | |
Chris Lattner | fe11a97 | 2002-12-23 23:59:41 +0000 | [diff] [blame] | 48 | namespace { |
| 49 | cl::opt<std::string> |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 50 | InputFile(cl::desc("<input bitcode>"), cl::Positional, cl::init("-")); |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 51 | |
Chris Lattner | fe11a97 | 2002-12-23 23:59:41 +0000 | [diff] [blame] | 52 | cl::list<std::string> |
| 53 | InputArgv(cl::ConsumeAfter, cl::desc("<program arguments>...")); |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 54 | |
Chris Lattner | fe11a97 | 2002-12-23 23:59:41 +0000 | [diff] [blame] | 55 | cl::opt<bool> ForceInterpreter("force-interpreter", |
Misha Brukman | 3d8a54d | 2003-09-25 18:10:34 +0000 | [diff] [blame] | 56 | cl::desc("Force interpretation: disable JIT"), |
| 57 | cl::init(false)); |
Evan Cheng | e1a4eda | 2008-08-08 08:12:06 +0000 | [diff] [blame] | 58 | |
Daniel Dunbar | 6d13597 | 2010-11-17 16:06:37 +0000 | [diff] [blame] | 59 | cl::opt<bool> UseMCJIT( |
| 60 | "use-mcjit", cl::desc("Enable use of the MC-based JIT (if available)"), |
| 61 | cl::init(false)); |
| 62 | |
Evan Cheng | 712e80e | 2009-05-04 23:05:19 +0000 | [diff] [blame] | 63 | // Determine optimization level. |
| 64 | cl::opt<char> |
| 65 | OptLevel("O", |
| 66 | cl::desc("Optimization level. [-O0, -O1, -O2, or -O3] " |
| 67 | "(default = '-O2')"), |
| 68 | cl::Prefix, |
| 69 | cl::ZeroOrMore, |
| 70 | cl::init(' ')); |
Evan Cheng | e1a4eda | 2008-08-08 08:12:06 +0000 | [diff] [blame] | 71 | |
Chris Lattner | 3015e60 | 2005-12-16 05:00:21 +0000 | [diff] [blame] | 72 | cl::opt<std::string> |
Chris Lattner | 60844d4 | 2005-12-16 05:19:18 +0000 | [diff] [blame] | 73 | TargetTriple("mtriple", cl::desc("Override target triple for module")); |
Evan Cheng | ec740e3 | 2008-11-05 23:21:52 +0000 | [diff] [blame] | 74 | |
| 75 | cl::opt<std::string> |
Jeffrey Yasskin | 4688261 | 2010-02-05 16:19:36 +0000 | [diff] [blame] | 76 | MArch("march", |
| 77 | cl::desc("Architecture to generate assembly for (see --version)")); |
| 78 | |
| 79 | cl::opt<std::string> |
| 80 | MCPU("mcpu", |
| 81 | cl::desc("Target a specific cpu type (-mcpu=help for details)"), |
| 82 | cl::value_desc("cpu-name"), |
| 83 | cl::init("")); |
| 84 | |
| 85 | cl::list<std::string> |
| 86 | MAttrs("mattr", |
| 87 | cl::CommaSeparated, |
| 88 | cl::desc("Target specific attributes (-mattr=help for details)"), |
| 89 | cl::value_desc("a1,+a2,-a3,...")); |
| 90 | |
| 91 | cl::opt<std::string> |
Evan Cheng | ec740e3 | 2008-11-05 23:21:52 +0000 | [diff] [blame] | 92 | EntryFunc("entry-function", |
| 93 | cl::desc("Specify the entry function (default = 'main') " |
| 94 | "of the executable"), |
| 95 | cl::value_desc("function"), |
| 96 | cl::init("main")); |
Chris Lattner | 3015e60 | 2005-12-16 05:00:21 +0000 | [diff] [blame] | 97 | |
Chris Lattner | e69671d | 2003-10-28 22:51:44 +0000 | [diff] [blame] | 98 | cl::opt<std::string> |
| 99 | FakeArgv0("fake-argv0", |
| 100 | cl::desc("Override the 'argv[0]' value passed into the executing" |
| 101 | " program"), cl::value_desc("executable")); |
Chris Lattner | 43f249a | 2006-09-14 06:17:09 +0000 | [diff] [blame] | 102 | |
| 103 | cl::opt<bool> |
| 104 | DisableCoreFiles("disable-core-files", cl::Hidden, |
| 105 | cl::desc("Disable emission of core files if possible")); |
Evan Cheng | c290a5b | 2008-04-22 06:51:41 +0000 | [diff] [blame] | 106 | |
| 107 | cl::opt<bool> |
Evan Cheng | 03dace8 | 2008-05-21 18:20:21 +0000 | [diff] [blame] | 108 | NoLazyCompilation("disable-lazy-compilation", |
Evan Cheng | c290a5b | 2008-04-22 06:51:41 +0000 | [diff] [blame] | 109 | cl::desc("Disable JIT lazy compilation"), |
| 110 | cl::init(false)); |
Evan Cheng | 4396613 | 2011-07-19 06:37:02 +0000 | [diff] [blame] | 111 | |
| 112 | cl::opt<Reloc::Model> |
| 113 | RelocModel("relocation-model", |
| 114 | cl::desc("Choose relocation model"), |
| 115 | cl::init(Reloc::Default), |
| 116 | cl::values( |
| 117 | clEnumValN(Reloc::Default, "default", |
| 118 | "Target default relocation model"), |
| 119 | clEnumValN(Reloc::Static, "static", |
| 120 | "Non-relocatable code"), |
| 121 | clEnumValN(Reloc::PIC_, "pic", |
| 122 | "Fully relocatable, position independent code"), |
| 123 | clEnumValN(Reloc::DynamicNoPIC, "dynamic-no-pic", |
| 124 | "Relocatable external references, non-relocatable code"), |
| 125 | clEnumValEnd)); |
Evan Cheng | 34ad6db | 2011-07-20 07:51:56 +0000 | [diff] [blame] | 126 | |
| 127 | cl::opt<llvm::CodeModel::Model> |
| 128 | CMModel("code-model", |
| 129 | cl::desc("Choose code model"), |
| 130 | cl::init(CodeModel::JITDefault), |
| 131 | cl::values(clEnumValN(CodeModel::JITDefault, "default", |
| 132 | "Target default JIT code model"), |
| 133 | clEnumValN(CodeModel::Small, "small", |
| 134 | "Small code model"), |
| 135 | clEnumValN(CodeModel::Kernel, "kernel", |
| 136 | "Kernel code model"), |
| 137 | clEnumValN(CodeModel::Medium, "medium", |
| 138 | "Medium code model"), |
| 139 | clEnumValN(CodeModel::Large, "large", |
| 140 | "Large code model"), |
| 141 | clEnumValEnd)); |
| 142 | |
Chris Lattner | fe11a97 | 2002-12-23 23:59:41 +0000 | [diff] [blame] | 143 | } |
Chris Lattner | 43e3f7c | 2001-10-27 08:43:52 +0000 | [diff] [blame] | 144 | |
Reid Spencer | f70d677 | 2007-03-03 18:21:44 +0000 | [diff] [blame] | 145 | static ExecutionEngine *EE = 0; |
| 146 | |
| 147 | static void do_shutdown() { |
NAKAMURA Takumi | a13d14a | 2010-10-22 14:53:59 +0000 | [diff] [blame] | 148 | // Cygwin-1.5 invokes DLL's dtors before atexit handler. |
| 149 | #ifndef DO_NOTHING_ATEXIT |
Reid Spencer | f70d677 | 2007-03-03 18:21:44 +0000 | [diff] [blame] | 150 | delete EE; |
| 151 | llvm_shutdown(); |
NAKAMURA Takumi | a13d14a | 2010-10-22 14:53:59 +0000 | [diff] [blame] | 152 | #endif |
Reid Spencer | f70d677 | 2007-03-03 18:21:44 +0000 | [diff] [blame] | 153 | } |
| 154 | |
Chris Lattner | 92101ac | 2001-08-23 17:05:04 +0000 | [diff] [blame] | 155 | //===----------------------------------------------------------------------===// |
Chris Lattner | 92101ac | 2001-08-23 17:05:04 +0000 | [diff] [blame] | 156 | // main Driver function |
| 157 | // |
Misha Brukman | c4fb6fd | 2003-10-14 21:39:53 +0000 | [diff] [blame] | 158 | int main(int argc, char **argv, char * const *envp) { |
Chris Lattner | cc14d25 | 2009-03-06 05:34:10 +0000 | [diff] [blame] | 159 | sys::PrintStackTraceOnErrorSignal(); |
| 160 | PrettyStackTraceProgram X(argc, argv); |
| 161 | |
Owen Anderson | 0d7c695 | 2009-07-15 22:16:10 +0000 | [diff] [blame] | 162 | LLVMContext &Context = getGlobalContext(); |
Reid Spencer | f70d677 | 2007-03-03 18:21:44 +0000 | [diff] [blame] | 163 | atexit(do_shutdown); // Call llvm_shutdown() on exit. |
Daniel Dunbar | 494d663 | 2009-07-16 02:04:54 +0000 | [diff] [blame] | 164 | |
| 165 | // If we have a native target, initialize it to ensure it is linked in and |
| 166 | // usable by the JIT. |
| 167 | InitializeNativeTarget(); |
Jim Grosbach | 31649e6 | 2011-03-18 22:48:41 +0000 | [diff] [blame] | 168 | InitializeNativeTargetAsmPrinter(); |
Daniel Dunbar | 494d663 | 2009-07-16 02:04:54 +0000 | [diff] [blame] | 169 | |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 170 | cl::ParseCommandLineOptions(argc, argv, |
Dan Gohman | 82a13c9 | 2007-10-08 15:45:12 +0000 | [diff] [blame] | 171 | "llvm interpreter & dynamic compiler\n"); |
Reid Spencer | 1ef8bda | 2004-12-30 05:36:08 +0000 | [diff] [blame] | 172 | |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 173 | // If the user doesn't want core files, disable them. |
| 174 | if (DisableCoreFiles) |
| 175 | sys::Process::PreventCoreFiles(); |
| 176 | |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 177 | // Load the bitcode... |
Daniel Dunbar | 46a2716 | 2010-11-13 00:28:01 +0000 | [diff] [blame] | 178 | SMDiagnostic Err; |
| 179 | Module *Mod = ParseIRFile(InputFile, Err, Context); |
Jeffrey Yasskin | f0356fe | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 180 | if (!Mod) { |
Chris Lattner | d8b7aa2 | 2011-10-16 04:47:35 +0000 | [diff] [blame^] | 181 | Err.print(argv[0], errs()); |
Daniel Dunbar | 46a2716 | 2010-11-13 00:28:01 +0000 | [diff] [blame] | 182 | return 1; |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 183 | } |
| 184 | |
Jeffrey Yasskin | f0356fe | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 185 | // If not jitting lazily, load the whole bitcode file eagerly too. |
Daniel Dunbar | 46a2716 | 2010-11-13 00:28:01 +0000 | [diff] [blame] | 186 | std::string ErrorMsg; |
Jeffrey Yasskin | f0356fe | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 187 | if (NoLazyCompilation) { |
| 188 | if (Mod->MaterializeAllPermanently(&ErrorMsg)) { |
| 189 | errs() << argv[0] << ": bitcode didn't read correctly.\n"; |
| 190 | errs() << "Reason: " << ErrorMsg << "\n"; |
| 191 | exit(1); |
| 192 | } |
Evan Cheng | c290a5b | 2008-04-22 06:51:41 +0000 | [diff] [blame] | 193 | } |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 194 | |
Jeffrey Yasskin | f0356fe | 2010-01-27 20:34:15 +0000 | [diff] [blame] | 195 | EngineBuilder builder(Mod); |
Jeffrey Yasskin | 4688261 | 2010-02-05 16:19:36 +0000 | [diff] [blame] | 196 | builder.setMArch(MArch); |
| 197 | builder.setMCPU(MCPU); |
| 198 | builder.setMAttrs(MAttrs); |
Evan Cheng | 4396613 | 2011-07-19 06:37:02 +0000 | [diff] [blame] | 199 | builder.setRelocationModel(RelocModel); |
Evan Cheng | 34ad6db | 2011-07-20 07:51:56 +0000 | [diff] [blame] | 200 | builder.setCodeModel(CMModel); |
Daniel Dunbar | 4dc3136 | 2009-07-18 08:07:13 +0000 | [diff] [blame] | 201 | builder.setErrorStr(&ErrorMsg); |
| 202 | builder.setEngineKind(ForceInterpreter |
Reid Kleckner | 4b1511b | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 203 | ? EngineKind::Interpreter |
| 204 | : EngineKind::JIT); |
| 205 | |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 206 | // If we are supposed to override the target triple, do so now. |
| 207 | if (!TargetTriple.empty()) |
Duncan Sands | 75ebbce | 2010-08-28 01:30:02 +0000 | [diff] [blame] | 208 | Mod->setTargetTriple(Triple::normalize(TargetTriple)); |
Evan Cheng | c290a5b | 2008-04-22 06:51:41 +0000 | [diff] [blame] | 209 | |
Daniel Dunbar | 6d13597 | 2010-11-17 16:06:37 +0000 | [diff] [blame] | 210 | // Enable MCJIT, if desired. |
| 211 | if (UseMCJIT) |
| 212 | builder.setUseMCJIT(true); |
| 213 | |
Evan Cheng | 712e80e | 2009-05-04 23:05:19 +0000 | [diff] [blame] | 214 | CodeGenOpt::Level OLvl = CodeGenOpt::Default; |
| 215 | switch (OptLevel) { |
| 216 | default: |
Dan Gohman | 65f57c2 | 2009-07-15 16:35:29 +0000 | [diff] [blame] | 217 | errs() << argv[0] << ": invalid optimization level.\n"; |
Evan Cheng | 712e80e | 2009-05-04 23:05:19 +0000 | [diff] [blame] | 218 | return 1; |
| 219 | case ' ': break; |
| 220 | case '0': OLvl = CodeGenOpt::None; break; |
Evan Cheng | bf57b52 | 2009-10-16 21:02:20 +0000 | [diff] [blame] | 221 | case '1': OLvl = CodeGenOpt::Less; break; |
Evan Cheng | 712e80e | 2009-05-04 23:05:19 +0000 | [diff] [blame] | 222 | case '2': OLvl = CodeGenOpt::Default; break; |
| 223 | case '3': OLvl = CodeGenOpt::Aggressive; break; |
| 224 | } |
Reid Kleckner | 4b1511b | 2009-07-18 00:42:18 +0000 | [diff] [blame] | 225 | builder.setOptLevel(OLvl); |
| 226 | |
| 227 | EE = builder.create(); |
Chris Lattner | fd15bee | 2009-07-07 18:31:09 +0000 | [diff] [blame] | 228 | if (!EE) { |
| 229 | if (!ErrorMsg.empty()) |
Dan Gohman | 65f57c2 | 2009-07-15 16:35:29 +0000 | [diff] [blame] | 230 | errs() << argv[0] << ": error creating EE: " << ErrorMsg << "\n"; |
Chris Lattner | fd15bee | 2009-07-07 18:31:09 +0000 | [diff] [blame] | 231 | else |
Dan Gohman | 65f57c2 | 2009-07-15 16:35:29 +0000 | [diff] [blame] | 232 | errs() << argv[0] << ": unknown error creating EE!\n"; |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 233 | exit(1); |
| 234 | } |
| 235 | |
Jeffrey Yasskin | 8f6d9a7 | 2009-07-10 21:08:20 +0000 | [diff] [blame] | 236 | EE->RegisterJITEventListener(createOProfileJITEventListener()); |
Jeffrey Yasskin | df5a7da | 2009-06-25 02:04:04 +0000 | [diff] [blame] | 237 | |
Jeffrey Yasskin | 18fec73 | 2009-10-27 22:39:42 +0000 | [diff] [blame] | 238 | EE->DisableLazyCompilation(NoLazyCompilation); |
Evan Cheng | c290a5b | 2008-04-22 06:51:41 +0000 | [diff] [blame] | 239 | |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 240 | // If the user specifically requested an argv[0] to pass into the program, |
| 241 | // do it now. |
| 242 | if (!FakeArgv0.empty()) { |
| 243 | InputFile = FakeArgv0; |
| 244 | } else { |
| 245 | // Otherwise, if there is a .bc suffix on the executable strip it off, it |
| 246 | // might confuse the program. |
Benjamin Kramer | 3bb37e9 | 2010-04-15 11:33:14 +0000 | [diff] [blame] | 247 | if (StringRef(InputFile).endswith(".bc")) |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 248 | InputFile.erase(InputFile.length() - 3); |
| 249 | } |
| 250 | |
| 251 | // Add the module's name to the start of the vector of arguments to main(). |
| 252 | InputArgv.insert(InputArgv.begin(), InputFile); |
| 253 | |
| 254 | // Call the main function from M as if its signature were: |
| 255 | // int main (int argc, char **argv, const char **envp) |
| 256 | // using the contents of Args to determine argc & argv, and the contents of |
| 257 | // EnvVars to determine envp. |
| 258 | // |
Evan Cheng | ec740e3 | 2008-11-05 23:21:52 +0000 | [diff] [blame] | 259 | Function *EntryFn = Mod->getFunction(EntryFunc); |
| 260 | if (!EntryFn) { |
Dan Gohman | 65f57c2 | 2009-07-15 16:35:29 +0000 | [diff] [blame] | 261 | errs() << '\'' << EntryFunc << "\' function not found in module.\n"; |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 262 | return -1; |
| 263 | } |
| 264 | |
| 265 | // If the program doesn't explicitly call exit, we will need the Exit |
| 266 | // function later on to make an explicit call, so get the function now. |
Owen Anderson | 1d0be15 | 2009-08-13 21:58:54 +0000 | [diff] [blame] | 267 | Constant *Exit = Mod->getOrInsertFunction("exit", Type::getVoidTy(Context), |
| 268 | Type::getInt32Ty(Context), |
| 269 | NULL); |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 270 | |
| 271 | // Reset errno to zero on entry to main. |
| 272 | errno = 0; |
| 273 | |
| 274 | // Run static constructors. |
| 275 | EE->runStaticConstructorsDestructors(false); |
Evan Cheng | c290a5b | 2008-04-22 06:51:41 +0000 | [diff] [blame] | 276 | |
| 277 | if (NoLazyCompilation) { |
| 278 | for (Module::iterator I = Mod->begin(), E = Mod->end(); I != E; ++I) { |
| 279 | Function *Fn = &*I; |
Evan Cheng | ec740e3 | 2008-11-05 23:21:52 +0000 | [diff] [blame] | 280 | if (Fn != EntryFn && !Fn->isDeclaration()) |
Evan Cheng | c290a5b | 2008-04-22 06:51:41 +0000 | [diff] [blame] | 281 | EE->getPointerToFunction(Fn); |
| 282 | } |
| 283 | } |
| 284 | |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 285 | // Run main. |
Evan Cheng | ec740e3 | 2008-11-05 23:21:52 +0000 | [diff] [blame] | 286 | int Result = EE->runFunctionAsMain(EntryFn, InputArgv, envp); |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 287 | |
| 288 | // Run static destructors. |
| 289 | EE->runStaticConstructorsDestructors(true); |
| 290 | |
| 291 | // If the program didn't call exit explicitly, we should call it now. |
| 292 | // This ensures that any atexit handlers get called correctly. |
| 293 | if (Function *ExitF = dyn_cast<Function>(Exit)) { |
| 294 | std::vector<GenericValue> Args; |
| 295 | GenericValue ResultGV; |
| 296 | ResultGV.IntVal = APInt(32, Result); |
| 297 | Args.push_back(ResultGV); |
| 298 | EE->runFunction(ExitF, Args); |
Dan Gohman | 65f57c2 | 2009-07-15 16:35:29 +0000 | [diff] [blame] | 299 | errs() << "ERROR: exit(" << Result << ") returned!\n"; |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 300 | abort(); |
| 301 | } else { |
Dan Gohman | 65f57c2 | 2009-07-15 16:35:29 +0000 | [diff] [blame] | 302 | errs() << "ERROR: exit defined with wrong prototype!\n"; |
Chris Lattner | c1e6d68 | 2007-05-06 04:58:26 +0000 | [diff] [blame] | 303 | abort(); |
| 304 | } |
Chris Lattner | 92101ac | 2001-08-23 17:05:04 +0000 | [diff] [blame] | 305 | } |