diff --git a/tools/lli/lli.cpp b/tools/lli/lli.cpp index 74ea3dd4154..fee48ed2101 100644 --- a/tools/lli/lli.cpp +++ b/tools/lli/lli.cpp @@ -49,6 +49,13 @@ namespace { cl::opt TargetTriple("mtriple", cl::desc("Override target triple for module")); + + cl::opt + EntryFunc("entry-function", + cl::desc("Specify the entry function (default = 'main') " + "of the executable"), + cl::value_desc("function"), + cl::init("main")); cl::opt FakeArgv0("fake-argv0", @@ -140,9 +147,9 @@ int main(int argc, char **argv, char * const *envp) { // using the contents of Args to determine argc & argv, and the contents of // EnvVars to determine envp. // - Function *MainFn = Mod->getFunction("main"); - if (!MainFn) { - std::cerr << "'main' function not found in module.\n"; + Function *EntryFn = Mod->getFunction(EntryFunc); + if (!EntryFn) { + std::cerr << '\'' << EntryFunc << "\' function not found in module.\n"; return -1; } @@ -160,13 +167,13 @@ int main(int argc, char **argv, char * const *envp) { if (NoLazyCompilation) { for (Module::iterator I = Mod->begin(), E = Mod->end(); I != E; ++I) { Function *Fn = &*I; - if (Fn != MainFn && !Fn->isDeclaration()) + if (Fn != EntryFn && !Fn->isDeclaration()) EE->getPointerToFunction(Fn); } } // Run main. - int Result = EE->runFunctionAsMain(MainFn, InputArgv, envp); + int Result = EE->runFunctionAsMain(EntryFn, InputArgv, envp); // Run static destructors. EE->runStaticConstructorsDestructors(true);