//===- TraceBasicBlocks.cpp - Insert basic-block trace instrumentation ----===//
-//
+//
// The LLVM Compiler Infrastructure
//
// This file was developed by the LLVM research group and is distributed under
// the University of Illinois Open Source License. See LICENSE.TXT for details.
-//
+//
//===----------------------------------------------------------------------===//
//
// This pass instruments the specified program with calls into a runtime
#include "llvm/Module.h"
#include "llvm/Pass.h"
#include "llvm/Transforms/Utils/BasicBlockUtils.h"
-#include "llvm/iOther.h"
-#include "llvm/iMemory.h"
-#include "llvm/iPHINode.h"
+#include "llvm/Transforms/Instrumentation.h"
+#include "llvm/Instructions.h"
#include "ProfilingUtils.h"
-#include "Support/Debug.h"
+#include "llvm/Support/Debug.h"
#include <set>
+#include <iostream>
using namespace llvm;
namespace {
- class TraceBasicBlocks : public Pass {
- bool run(Module &M);
+ class TraceBasicBlocks : public ModulePass {
+ bool runOnModule(Module &M);
};
- RegisterOpt<TraceBasicBlocks> X("trace-basic-blocks",
+ RegisterPass<TraceBasicBlocks> X("trace-basic-blocks",
"Insert instrumentation for basic block tracing");
}
+ModulePass *llvm::createTraceBasicBlockPass()
+{
+ return new TraceBasicBlocks();
+}
+
static void InsertInstrumentationCall (BasicBlock *BB,
const std::string FnName,
unsigned BBNumber) {
<< "\", \"" << FnName << "\", " << BBNumber << ")\n");
Module &M = *BB->getParent ()->getParent ();
Function *InstrFn = M.getOrInsertFunction (FnName, Type::VoidTy,
- Type::UIntTy, 0);
+ Type::UIntTy, (Type *)0);
std::vector<Value*> Args (1);
- Args[0] = ConstantUInt::get (Type::UIntTy, BBNumber);
+ Args[0] = ConstantInt::get (Type::UIntTy, BBNumber);
// Insert the call after any alloca or PHI instructions...
BasicBlock::iterator InsertPos = BB->begin();
Instruction *InstrCall = new CallInst (InstrFn, Args, "", InsertPos);
}
-bool TraceBasicBlocks::run(Module &M) {
+bool TraceBasicBlocks::runOnModule(Module &M) {
Function *Main = M.getMainFunction();
if (Main == 0) {
std::cerr << "WARNING: cannot insert basic-block trace instrumentation"