-static bool
-insertTraceCodeFor(Module &M)
-{
- PassManager Passes;
-
- // Insert trace code in all functions in the module
- if (TraceBasicBlocks)
- Passes.add(createTraceValuesPassForBasicBlocks());
- else if (TraceFunctions)
- Passes.add(createTraceValuesPassForFunction());
- else
- return false;
-
- // Eliminate duplication in constant pool
- Passes.add(createConstantMergePass());
-
- // Run passes to insert and clean up trace code...
- Passes.run(M);
-
- std::string ErrorMessage;
-
- // Load the module that contains the runtime helper routines neccesary for
- // pointer hashing and stuff... link this module into the program if possible
- //
- Module *TraceModule = ParseBytecodeFile(TraceLibPath+"libinstr.bc");
-
- // Check if the TraceLibPath contains a valid module. If not, try to load
- // the module from the current LLVM-GCC install directory. This is kindof
- // a hack, but allows people to not HAVE to have built the library.
- //
- if (TraceModule == 0)
- TraceModule = ParseBytecodeFile("/home/vadve/lattner/cvs/gcc_install/lib/"
- "gcc-lib/llvm/3.1/libinstr.bc");
-
- // If we still didn't get it, cancel trying to link it in...
- if (TraceModule == 0)
- std::cerr <<"WARNING: couldn't load trace routines to link into program!\n";
- else
- {
- // Link in the trace routines... if this fails, don't panic, because the
- // compile should still succeed, but the native linker will probably fail.
- //
- std::auto_ptr<Module> TraceRoutines(TraceModule);
- if (LinkModules(&M, TraceRoutines.get(), &ErrorMessage))
- std::cerr << "WARNING: Error linking in trace routines: "
- << ErrorMessage << "\n";
- }
-
- // Write out the module with tracing code just before code generation
- assert (InputFilename != "-"
- && "Cannot write out traced bytecode when reading input from stdin");
- std::string TraceFilename = GetFileNameRoot(InputFilename) + ".trace.bc";
-
- std::ofstream Out(TraceFilename.c_str());
- if (!Out.good())
- std::cerr << "Error opening '" << TraceFilename
- << "'!: Skipping output of trace code as bytecode\n";
- else
- {
- std::cerr << "Emitting trace code to '" << TraceFilename
- << "' for comparison...\n";
- WriteBytecodeToFile(&M, Out);
- }
-
- return true;
-}
-
-// Making tracing a module pass so the entire module with tracing
-// can be written out before continuing.
-struct InsertTracingCodePass: public Pass {
- virtual bool run(Module &M) {
- return insertTraceCodeFor(M);
- }
-};