- Function *F = M.get()->getNamedFunction(ExtractFunc);
- if (F == 0) {
- std::cerr << argv[0] << ": program doesn't contain function named '"
- << ExtractFunc << "'!\n";
+ GlobalVariable *G = ExtractGlobal.size() ?
+ M.get()->getNamedGlobal(ExtractGlobal) : 0;
+
+ // Figure out which function we should extract
+ if (!ExtractFunc.size() && !ExtractGlobal.size()) ExtractFunc = "main";
+ Function *F = M.get()->getFunction(ExtractFunc);
+
+ if (F == 0 && G == 0) {
+ cerr << argv[0] << ": program doesn't contain function named '"
+ << ExtractFunc << "' or a global named '" << ExtractGlobal << "'!\n";