1 //===------------------------------------------------------------------------=== 2 // LLVM 'OPT' UTILITY 3 // 4 // This utility may be invoked in the following manner: 5 // opt --help - Output information about command line switches 6 // opt [options] -dce - Run a dead code elimination pass on input 7 // bytecodes 8 // opt [options] -constprop - Run a constant propogation pass on input 9 // bytecodes 10 // opt [options] -inline - Run a method inlining pass on input bytecodes 11 // opt [options] -strip - Strip symbol tables out of methods 12 // opt [options] -mstrip - Strip module & method symbol tables 13 // 14 // Optimizations may be specified an arbitrary number of times on the command 15 // line, they are run in the order specified. 16 // 17 // TODO: Add a -all option to keep applying all optimizations until the program 18 // stops permuting. 19 // 20 //===------------------------------------------------------------------------=== 21 22 #include <iostream.h> 23 #include <fstream.h> 24 #include "llvm/Module.h" 25 #include "llvm/Bytecode/Reader.h" 26 #include "llvm/Bytecode/Writer.h" 27 #include "llvm/Tools/CommandLine.h" 28 #include "llvm/Opt/AllOpts.h" 29 30 struct { 31 const string ArgName, Name; 32 bool (*OptPtr)(Module *C); 33 } OptTable[] = { 34 { "-dce", "Dead Code Elimination", DoDeadCodeElimination }, 35 { "-constprop","Constant Propogation", DoConstantPropogation }, 36 { "-inline" ,"Method Inlining", DoMethodInlining }, 37 { "-strip" ,"Strip Symbols", DoSymbolStripping }, 38 { "-mstrip" ,"Strip Module Symbols", DoFullSymbolStripping }, 39 { "-indvars" ,"Simplify Induction Vars", DoInductionVariableCannonicalize }, 40 }; 41 42 int main(int argc, char **argv) { 43 ToolCommandLine Opts(argc, argv, false); 44 bool Quiet = false; 45 46 for (int i = 1; i < argc; i++) { 47 if (string(argv[i]) == string("--help")) { 48 cerr << argv[0] << " usage:\n" 49 << " " << argv[0] << " --help - Print this usage information\n"; 50 for (unsigned j = 0; j < sizeof(OptTable)/sizeof(OptTable[0]); ++j) { 51 cerr << "\t" << OptTable[j].ArgName << "\t - Enable " 52 << OptTable[j].Name << endl; 53 } 54 return 1; 55 } else if (string(argv[i]) == string("-q")) { 56 Quiet = true; argv[i] = 0; 57 } 58 } 59 60 ostream *Out = &cout; // Default to printing to stdout... 61 62 Module *C = ParseBytecodeFile(Opts.getInputFilename()); 63 if (C == 0) { 64 cerr << "bytecode didn't read correctly.\n"; 65 return 1; 66 } 67 68 69 for (int i = 1; i < argc; i++) { 70 if (argv[i] == 0) continue; 71 unsigned j; 72 for (j = 0; j < sizeof(OptTable)/sizeof(OptTable[0]); j++) { 73 if (string(argv[i]) == OptTable[j].ArgName) { 74 if (OptTable[j].OptPtr(C) && !Quiet) 75 cerr << OptTable[j].Name << " pass made modifications!\n"; 76 break; 77 } 78 } 79 80 if (j == sizeof(OptTable)/sizeof(OptTable[0])) 81 cerr << "'" << argv[i] << "' argument unrecognized: ignored\n"; 82 } 83 84 if (Opts.getOutputFilename() != "-") { 85 Out = new ofstream(Opts.getOutputFilename().c_str(), 86 (Opts.getForce() ? 0 : ios::noreplace)|ios::out); 87 if (!Out->good()) { 88 cerr << "Error opening " << Opts.getOutputFilename() 89 << "!\n"; 90 delete C; 91 return 1; 92 } 93 } 94 95 // Okay, we're done now... write out result... 96 WriteBytecodeToFile(C, *Out); 97 delete C; 98 99 if (Out != &cout) delete Out; 100 return 0; 101 } 102