1 //===- TraceBasicBlocks.cpp - Insert basic-block trace instrumentation ----===//
3 // The LLVM Compiler Infrastructure
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.
8 //===----------------------------------------------------------------------===//
10 // This pass instruments the specified program with calls into a runtime
11 // library that cause it to output a trace of basic blocks as a side effect
12 // of normal execution.
14 //===----------------------------------------------------------------------===//
16 #include "llvm/Constants.h"
17 #include "llvm/DerivedTypes.h"
18 #include "llvm/Module.h"
19 #include "llvm/Pass.h"
20 #include "llvm/Transforms/Utils/BasicBlockUtils.h"
21 #include "llvm/Transforms/Instrumentation.h"
22 #include "llvm/Instructions.h"
23 #include "ProfilingUtils.h"
24 #include "llvm/Support/Debug.h"
30 class TraceBasicBlocks : public ModulePass {
31 bool runOnModule(Module &M);
34 RegisterPass<TraceBasicBlocks> X("trace-basic-blocks",
35 "Insert instrumentation for basic block tracing");
38 ModulePass *llvm::createTraceBasicBlockPass()
40 return new TraceBasicBlocks();
43 static void InsertInstrumentationCall (BasicBlock *BB,
44 const std::string FnName,
46 DEBUG (std::cerr << "InsertInstrumentationCall (\"" << BB->getName ()
47 << "\", \"" << FnName << "\", " << BBNumber << ")\n");
48 Module &M = *BB->getParent ()->getParent ();
49 Function *InstrFn = M.getOrInsertFunction (FnName, Type::VoidTy,
50 Type::UIntTy, (Type *)0);
51 std::vector<Value*> Args (1);
52 Args[0] = ConstantUInt::get (Type::UIntTy, BBNumber);
54 // Insert the call after any alloca or PHI instructions...
55 BasicBlock::iterator InsertPos = BB->begin();
56 while (isa<AllocaInst>(InsertPos) || isa<PHINode>(InsertPos))
59 Instruction *InstrCall = new CallInst (InstrFn, Args, "", InsertPos);
62 bool TraceBasicBlocks::runOnModule(Module &M) {
63 Function *Main = M.getMainFunction();
65 std::cerr << "WARNING: cannot insert basic-block trace instrumentation"
66 << " into a module with no main function!\n";
67 return false; // No main, no instrumentation!
70 unsigned BBNumber = 0;
71 for (Module::iterator F = M.begin(), E = M.end(); F != E; ++F)
72 for (Function::iterator BB = F->begin(), E = F->end(); BB != E; ++BB) {
73 InsertInstrumentationCall (BB, "llvm_trace_basic_block", BBNumber);
77 // Add the initialization call to main.
78 InsertProfilingInitCall(Main, "llvm_start_basic_block_tracing");