Chris Lattner | dbc2318 | 2003-10-15 21:49:57 +0000 | [diff] [blame] | 1 | //===-- gccas.cpp - The "optimizing assembler" used by the GCC frontend ---===// |
John Criswell | 7c0e022 | 2003-10-20 17:47:21 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file was developed by the LLVM research group and is distributed under |
| 6 | // the University of Illinois Open Source License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 9 | // |
Misha Brukman | 57d708b | 2003-08-07 21:23:52 +0000 | [diff] [blame] | 10 | // This utility is designed to be used by the GCC frontend for creating bytecode |
| 11 | // files from its intermediate LLVM assembly. The requirements for this utility |
| 12 | // are thus slightly different than that of the standard `as' util. |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 13 | // |
Chris Lattner | 3dc67dd | 2002-01-22 03:30:46 +0000 | [diff] [blame] | 14 | //===----------------------------------------------------------------------===// |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 15 | |
| 16 | #include "llvm/Module.h" |
Chris Lattner | 0f3bfff | 2002-01-31 00:46:22 +0000 | [diff] [blame] | 17 | #include "llvm/PassManager.h" |
Misha Brukman | 57d708b | 2003-08-07 21:23:52 +0000 | [diff] [blame] | 18 | #include "llvm/Analysis/LoadValueNumbering.h" |
| 19 | #include "llvm/Analysis/Verifier.h" |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 20 | #include "llvm/Assembly/Parser.h" |
Misha Brukman | 57d708b | 2003-08-07 21:23:52 +0000 | [diff] [blame] | 21 | #include "llvm/Bytecode/WriteBytecodePass.h" |
| 22 | #include "llvm/Target/TargetData.h" |
Chris Lattner | d9d8c07 | 2002-07-23 22:04:43 +0000 | [diff] [blame] | 23 | #include "llvm/Transforms/IPO.h" |
Chris Lattner | 65f1b89 | 2002-05-07 20:03:27 +0000 | [diff] [blame] | 24 | #include "llvm/Transforms/Scalar.h" |
Reid Spencer | 551ccae | 2004-09-01 22:55:40 +0000 | [diff] [blame] | 25 | #include "llvm/Support/CommandLine.h" |
Chris Lattner | bed85ff | 2004-05-27 05:41:36 +0000 | [diff] [blame] | 26 | #include "llvm/System/Signals.h" |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 27 | #include <memory> |
| 28 | #include <fstream> |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 29 | |
Brian Gaeke | d0fde30 | 2003-11-11 22:41:34 +0000 | [diff] [blame] | 30 | using namespace llvm; |
| 31 | |
Chris Lattner | f2956fc | 2003-04-16 17:34:29 +0000 | [diff] [blame] | 32 | namespace { |
Chris Lattner | f2956fc | 2003-04-16 17:34:29 +0000 | [diff] [blame] | 33 | cl::opt<std::string> |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 34 | InputFilename(cl::Positional,cl::desc("<input llvm assembly>"),cl::init("-")); |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 35 | |
Chris Lattner | f2956fc | 2003-04-16 17:34:29 +0000 | [diff] [blame] | 36 | cl::opt<std::string> |
| 37 | OutputFilename("o", cl::desc("Override output filename"), |
| 38 | cl::value_desc("filename")); |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 39 | |
Chris Lattner | f2956fc | 2003-04-16 17:34:29 +0000 | [diff] [blame] | 40 | cl::opt<bool> |
| 41 | Verify("verify", cl::desc("Verify each pass result")); |
Chris Lattner | 0cea3ec | 2003-10-10 18:18:53 +0000 | [diff] [blame] | 42 | |
| 43 | cl::opt<bool> |
| 44 | DisableInline("disable-inlining", cl::desc("Do not run the inliner pass")); |
Chris Lattner | 74295c0 | 2003-12-30 03:24:27 +0000 | [diff] [blame] | 45 | |
| 46 | cl::opt<bool> |
| 47 | DisableOptimizations("disable-opt", |
| 48 | cl::desc("Do not run any optimization passes")); |
Chris Lattner | b11a463 | 2004-07-22 08:34:33 +0000 | [diff] [blame] | 49 | |
| 50 | cl::opt<bool> |
Chris Lattner | 40add59 | 2004-12-03 05:45:58 +0000 | [diff] [blame^] | 51 | StripDebug("strip-debug", |
| 52 | cl::desc("Strip debugger symbol info from translation unit")); |
| 53 | |
Reid Spencer | cc0bd56 | 2004-11-08 17:37:04 +0000 | [diff] [blame] | 54 | cl::opt<bool> |
| 55 | NoCompress("disable-compression", cl::init(false), |
| 56 | cl::desc("Don't ompress the generated bytecode")); |
Chris Lattner | f2956fc | 2003-04-16 17:34:29 +0000 | [diff] [blame] | 57 | } |
Chris Lattner | 5ff62e9 | 2002-07-22 02:10:13 +0000 | [diff] [blame] | 58 | |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 59 | |
| 60 | static inline void addPass(PassManager &PM, Pass *P) { |
Chris Lattner | 36f18ae | 2003-08-31 21:47:24 +0000 | [diff] [blame] | 61 | // Add the pass to the pass manager... |
| 62 | PM.add(P); |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 63 | |
Chris Lattner | 36f18ae | 2003-08-31 21:47:24 +0000 | [diff] [blame] | 64 | // If we are verifying all of the intermediate steps, add the verifier... |
| 65 | if (Verify) PM.add(createVerifierPass()); |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 66 | } |
| 67 | |
| 68 | |
| 69 | void AddConfiguredTransformationPasses(PassManager &PM) { |
Chris Lattner | e643a6c | 2003-06-22 20:11:45 +0000 | [diff] [blame] | 70 | PM.add(createVerifierPass()); // Verify that input is correct |
Chris Lattner | 40add59 | 2004-12-03 05:45:58 +0000 | [diff] [blame^] | 71 | |
Chris Lattner | f7c7f5a | 2003-09-15 04:56:44 +0000 | [diff] [blame] | 72 | addPass(PM, createLowerSetJmpPass()); // Lower llvm.setjmp/.longjmp |
Chris Lattner | c539483 | 2002-08-30 22:55:32 +0000 | [diff] [blame] | 73 | addPass(PM, createFunctionResolvingPass()); // Resolve (...) functions |
Chris Lattner | 74295c0 | 2003-12-30 03:24:27 +0000 | [diff] [blame] | 74 | |
Chris Lattner | 40add59 | 2004-12-03 05:45:58 +0000 | [diff] [blame^] | 75 | // If the -strip-debug command line option was specified, do it. |
| 76 | if (StripDebug) |
| 77 | addPass(PM, createStripSymbolsPass(true)); |
| 78 | |
Chris Lattner | 74295c0 | 2003-12-30 03:24:27 +0000 | [diff] [blame] | 79 | if (DisableOptimizations) return; |
| 80 | |
Chris Lattner | 14b170f | 2003-11-21 21:44:35 +0000 | [diff] [blame] | 81 | addPass(PM, createRaiseAllocationsPass()); // call %malloc -> malloc inst |
Chris Lattner | 05e4e8a | 2004-02-01 07:24:53 +0000 | [diff] [blame] | 82 | addPass(PM, createCFGSimplificationPass()); // Clean up disgusting code |
| 83 | addPass(PM, createPromoteMemoryToRegister()); // Kill useless allocas |
Chris Lattner | 93a00e4 | 2004-10-07 04:12:02 +0000 | [diff] [blame] | 84 | addPass(PM, createGlobalOptimizerPass()); // Optimize out global vars |
| 85 | addPass(PM, createGlobalDCEPass()); // Remove unused fns and globs |
Chris Lattner | eaa35bb | 2003-10-23 18:25:57 +0000 | [diff] [blame] | 86 | addPass(PM, createIPConstantPropagationPass());// IP Constant Propagation |
| 87 | addPass(PM, createDeadArgEliminationPass()); // Dead argument elimination |
Chris Lattner | 05e4e8a | 2004-02-01 07:24:53 +0000 | [diff] [blame] | 88 | addPass(PM, createInstructionCombiningPass()); // Clean up after IPCP & DAE |
| 89 | addPass(PM, createCFGSimplificationPass()); // Clean up after IPCP & DAE |
Chris Lattner | eaa35bb | 2003-10-23 18:25:57 +0000 | [diff] [blame] | 90 | |
Chris Lattner | cf37c23 | 2003-08-31 21:45:55 +0000 | [diff] [blame] | 91 | addPass(PM, createPruneEHPass()); // Remove dead EH info |
Chris Lattner | 0cea3ec | 2003-10-10 18:18:53 +0000 | [diff] [blame] | 92 | |
| 93 | if (!DisableInline) |
| 94 | addPass(PM, createFunctionInliningPass()); // Inline small functions |
Chris Lattner | 2c7b430 | 2004-03-13 21:38:35 +0000 | [diff] [blame] | 95 | addPass(PM, createArgumentPromotionPass()); // Scalarize uninlined fn args |
Chris Lattner | cf37c23 | 2003-08-31 21:45:55 +0000 | [diff] [blame] | 96 | |
Chris Lattner | f07c833 | 2003-10-16 16:50:34 +0000 | [diff] [blame] | 97 | addPass(PM, createRaisePointerReferencesPass());// Recover type information |
Chris Lattner | e643a6c | 2003-06-22 20:11:45 +0000 | [diff] [blame] | 98 | addPass(PM, createTailDuplicationPass()); // Simplify cfg by copying code |
| 99 | addPass(PM, createCFGSimplificationPass()); // Merge & remove BBs |
Chris Lattner | 42ed21b | 2003-05-30 19:24:06 +0000 | [diff] [blame] | 100 | addPass(PM, createScalarReplAggregatesPass()); // Break up aggregate allocas |
Chris Lattner | cf37c23 | 2003-08-31 21:45:55 +0000 | [diff] [blame] | 101 | addPass(PM, createInstructionCombiningPass()); // Combine silly seq's |
| 102 | |
Chris Lattner | 52af630 | 2002-10-31 17:13:11 +0000 | [diff] [blame] | 103 | addPass(PM, createReassociatePass()); // Reassociate expressions |
Chris Lattner | 4dd7d3e | 2002-09-06 18:41:33 +0000 | [diff] [blame] | 104 | addPass(PM, createInstructionCombiningPass()); // Combine silly seq's |
Chris Lattner | ca6cc6f | 2003-12-11 17:50:32 +0000 | [diff] [blame] | 105 | addPass(PM, createTailCallEliminationPass()); // Eliminate tail calls |
Chris Lattner | 4dd7d3e | 2002-09-06 18:41:33 +0000 | [diff] [blame] | 106 | addPass(PM, createCFGSimplificationPass()); // Merge & remove BBs |
Chris Lattner | c539483 | 2002-08-30 22:55:32 +0000 | [diff] [blame] | 107 | addPass(PM, createLICMPass()); // Hoist loop invariants |
Chris Lattner | 93d8202 | 2004-04-18 05:21:01 +0000 | [diff] [blame] | 108 | addPass(PM, createInstructionCombiningPass()); // Clean up after the unroller |
| 109 | addPass(PM, createIndVarSimplifyPass()); // Canonicalize indvars |
| 110 | addPass(PM, createLoopUnrollPass()); // Unroll small loops |
| 111 | addPass(PM, createInstructionCombiningPass()); // Clean up after the unroller |
Chris Lattner | c539483 | 2002-08-30 22:55:32 +0000 | [diff] [blame] | 112 | addPass(PM, createLoadValueNumberingPass()); // GVN for load instructions |
| 113 | addPass(PM, createGCSEPass()); // Remove common subexprs |
| 114 | addPass(PM, createSCCPPass()); // Constant prop with SCCP |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 115 | |
| 116 | // Run instcombine after redundancy elimination to exploit opportunities |
| 117 | // opened up by them. |
| 118 | addPass(PM, createInstructionCombiningPass()); |
Chris Lattner | 40add59 | 2004-12-03 05:45:58 +0000 | [diff] [blame^] | 119 | addPass(PM, createDeadStoreEliminationPass()); // Delete dead stores |
Chris Lattner | e643a6c | 2003-06-22 20:11:45 +0000 | [diff] [blame] | 120 | addPass(PM, createAggressiveDCEPass()); // SSA based 'Aggressive DCE' |
Chris Lattner | 4dd7d3e | 2002-09-06 18:41:33 +0000 | [diff] [blame] | 121 | addPass(PM, createCFGSimplificationPass()); // Merge & remove BBs |
Chris Lattner | cf37c23 | 2003-08-31 21:45:55 +0000 | [diff] [blame] | 122 | addPass(PM, createDeadTypeEliminationPass()); // Eliminate dead types |
| 123 | addPass(PM, createConstantMergePass()); // Merge dup global constants |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 124 | } |
| 125 | |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 126 | |
| 127 | int main(int argc, char **argv) { |
Chris Lattner | 11c862c | 2001-10-31 04:33:33 +0000 | [diff] [blame] | 128 | cl::ParseCommandLineOptions(argc, argv, " llvm .s -> .o assembler for GCC\n"); |
Reid Spencer | 9de7b33 | 2004-08-29 19:28:55 +0000 | [diff] [blame] | 129 | sys::PrintStackTraceOnErrorSignal(); |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 130 | |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 131 | std::auto_ptr<Module> M; |
| 132 | try { |
| 133 | // Parse the file now... |
| 134 | M.reset(ParseAssemblyFile(InputFilename)); |
| 135 | } catch (const ParseException &E) { |
Chris Lattner | 2c1d2f2 | 2003-04-16 17:41:08 +0000 | [diff] [blame] | 136 | std::cerr << argv[0] << ": " << E.getMessage() << "\n"; |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 137 | return 1; |
| 138 | } |
| 139 | |
| 140 | if (M.get() == 0) { |
Chris Lattner | 2c1d2f2 | 2003-04-16 17:41:08 +0000 | [diff] [blame] | 141 | std::cerr << argv[0] << ": assembly didn't read correctly.\n"; |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 142 | return 1; |
| 143 | } |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 144 | |
| 145 | std::ostream *Out = 0; |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 146 | if (OutputFilename == "") { // Didn't specify an output filename? |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 147 | if (InputFilename == "-") { |
| 148 | OutputFilename = "-"; |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 149 | } else { |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 150 | std::string IFN = InputFilename; |
| 151 | int Len = IFN.length(); |
| 152 | if (IFN[Len-2] == '.' && IFN[Len-1] == 's') { // Source ends in .s? |
| 153 | OutputFilename = std::string(IFN.begin(), IFN.end()-2); |
| 154 | } else { |
| 155 | OutputFilename = IFN; // Append a .o to it |
| 156 | } |
| 157 | OutputFilename += ".o"; |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 158 | } |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 159 | } |
| 160 | |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 161 | if (OutputFilename == "-") |
| 162 | Out = &std::cout; |
| 163 | else { |
| 164 | Out = new std::ofstream(OutputFilename.c_str(), std::ios::out); |
| 165 | |
Misha Brukman | 452fea9 | 2003-10-10 17:56:49 +0000 | [diff] [blame] | 166 | // Make sure that the Out file gets unlinked from the disk if we get a |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 167 | // signal |
Reid Spencer | 227b6d0 | 2004-11-14 22:30:54 +0000 | [diff] [blame] | 168 | sys::RemoveFileOnSignal(sys::Path(OutputFilename)); |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 169 | } |
| 170 | |
| 171 | |
| 172 | if (!Out->good()) { |
Chris Lattner | 2c1d2f2 | 2003-04-16 17:41:08 +0000 | [diff] [blame] | 173 | std::cerr << argv[0] << ": error opening " << OutputFilename << "!\n"; |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 174 | return 1; |
| 175 | } |
| 176 | |
| 177 | // In addition to just parsing the input from GCC, we also want to spiff it up |
| 178 | // a little bit. Do this now. |
| 179 | // |
Chris Lattner | f4de63f | 2002-01-21 07:31:50 +0000 | [diff] [blame] | 180 | PassManager Passes; |
Chris Lattner | 4ad5322 | 2002-05-14 16:23:14 +0000 | [diff] [blame] | 181 | |
Chris Lattner | 9c3b55e | 2003-04-24 19:13:02 +0000 | [diff] [blame] | 182 | // Add an appropriate TargetData instance for this module... |
| 183 | Passes.add(new TargetData("gccas", M.get())); |
| 184 | |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 185 | // Add all of the transformation passes to the pass manager to do the cleanup |
| 186 | // and optimization of the GCC output. |
| 187 | // |
| 188 | AddConfiguredTransformationPasses(Passes); |
| 189 | |
Chris Lattner | 0cccb18 | 2004-01-14 03:39:46 +0000 | [diff] [blame] | 190 | // Make sure everything is still good. |
| 191 | Passes.add(createVerifierPass()); |
| 192 | |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 193 | // Write bytecode to file... |
Reid Spencer | cc0bd56 | 2004-11-08 17:37:04 +0000 | [diff] [blame] | 194 | Passes.add(new WriteBytecodePass(Out,false,!NoCompress)); |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 195 | |
Chris Lattner | 3dc67dd | 2002-01-22 03:30:46 +0000 | [diff] [blame] | 196 | // Run our queue of passes all at once now, efficiently. |
Chris Lattner | 624c3e0 | 2002-06-25 15:57:43 +0000 | [diff] [blame] | 197 | Passes.run(*M.get()); |
Chris Lattner | 8c7b055 | 2003-04-16 17:49:18 +0000 | [diff] [blame] | 198 | |
| 199 | if (Out != &std::cout) delete Out; |
Chris Lattner | ecbde33 | 2001-10-31 04:28:11 +0000 | [diff] [blame] | 200 | return 0; |
| 201 | } |