ExecutionEngine.cpp: Move execution engine creation stuff into a new
[oota-llvm.git] / tools / lli / lli.cpp
1 //===- lli.cpp - LLVM Interpreter / Dynamic compiler ----------------------===//
2 //
3 // This utility provides a way to execute LLVM bytecode without static
4 // compilation.  This consists of a very simple and slow (but portable)
5 // interpreter, along with capability for system specific dynamic compilers.  At
6 // runtime, the fastest (stable) execution engine is selected to run the
7 // program.  This means the JIT compiler for the current platform if it's
8 // available.
9 //
10 //===----------------------------------------------------------------------===//
11
12 #include "ExecutionEngine.h"
13 #include "Support/CommandLine.h"
14 #include "llvm/Bytecode/Reader.h"
15 #include "llvm/Module.h"
16 #include "llvm/Target/TargetMachineImpls.h"
17
18 namespace {
19   cl::opt<std::string>
20   InputFile(cl::desc("<input bytecode>"), cl::Positional, cl::init("-"));
21
22   cl::list<std::string>
23   InputArgv(cl::ConsumeAfter, cl::desc("<program arguments>..."));
24
25   cl::opt<std::string>
26   MainFunction ("f", cl::desc("Function to execute"), cl::init("main"),
27                 cl::value_desc("function name"));
28
29   cl::opt<bool> DebugMode("d", cl::desc("Start program in debugger"));
30
31   cl::opt<bool> TraceMode("trace", cl::desc("Enable Tracing"));
32
33   cl::opt<bool> ForceInterpreter("force-interpreter",
34                                  cl::desc("Force interpretation: disable JIT"),
35                                  cl::init(false));
36 }
37
38 //===----------------------------------------------------------------------===//
39 // ExecutionEngine Class Implementation
40 //
41
42 ExecutionEngine::~ExecutionEngine() {
43   delete &CurMod;
44 }
45
46 //===----------------------------------------------------------------------===//
47 // main Driver function
48 //
49 int main(int argc, char** argv, const char ** envp) {
50   cl::ParseCommandLineOptions(argc, argv,
51                               " llvm interpreter & dynamic compiler\n");
52
53   // Load the bytecode...
54   std::string ErrorMsg;
55   Module *M = ParseBytecodeFile(InputFile, &ErrorMsg);
56   if (M == 0) {
57     std::cout << "Error parsing '" << InputFile << "': "
58               << ErrorMsg << "\n";
59     exit(1);
60   }
61
62   ExecutionEngine *EE =
63     ExecutionEngine::create (M, ForceInterpreter, DebugMode, TraceMode);
64   assert (EE && "Couldn't create an ExecutionEngine, not even an interpreter?");
65
66   // Add the module name to the start of the argv vector...
67   // But delete .bc first, since programs (and users) might not expect to
68   // see it.
69   const std::string ByteCodeFileSuffix (".bc");
70   if (InputFile.rfind (ByteCodeFileSuffix) ==
71       InputFile.length () - ByteCodeFileSuffix.length ()) {
72     InputFile.erase (InputFile.length () - ByteCodeFileSuffix.length ());
73   }
74   InputArgv.insert(InputArgv.begin(), InputFile);
75
76   // Run the main function!
77   int ExitCode = EE->run(MainFunction, InputArgv, envp);
78
79   // Now that we are done executing the program, shut down the execution engine
80   delete EE;
81   return ExitCode;
82 }