lib/Transforms/Instrumentation/ProfilePaths/EmitFunctions/EmitFunctions.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

//===-- EmitFunctions.cpp - interface to insert instrumentation --*- C++ -*--=//
//
// This inserts a global constant table with function pointers all along
//
//===----------------------------------------------------------------------===//

//#include "llvm/Transforms/Instrumentation/ProfilePaths.h"
//#include "llvm/Transforms/Utils/UnifyFunctionExitNodes.h"
#include "llvm/Support/CFG.h"
#include "llvm/Transforms/Instrumentation/EmitFunctions.h"
#include "llvm/Constants.h"
#include "llvm/DerivedTypes.h"
#include "llvm/iMemory.h"
#include "llvm/Constants.h"
#include "llvm/Module.h"
#include <iostream>
#include <fstream>

using std::vector;

struct EmitFunctionTable : public Pass {
  const char *getPassName() const { return "EmitFunctionTablePass"; }

  bool run(Module &M);
};

//  Create a new pass to add function table
//
Pass *createEmitFunctionTablePass() {
  return new EmitFunctionTable();
}

//Per Module pass for inserting function table
bool EmitFunctionTable::run(Module &M){
  vector<const Type*> vType;
  vector<Constant *> vConsts;
  for(Module::iterator MI = M.begin(), ME = M.end(); MI!=ME; ++MI){
    Function *F = MI;
    if(F->size()<=1) continue;
    ConstantPointerRef *cpref = ConstantPointerRef::get(F);
    vType.push_back(MI->getType());
    vConsts.push_back(cpref);
  }
  
  StructType *sttype = StructType::get(vType);
  ConstantStruct *cstruct = ConstantStruct::get(sttype, vConsts);

  GlobalVariable *gb = new GlobalVariable(cstruct->getType(), true, true, 
                                          cstruct, "llvmFunctionTable");
  M.getGlobalList().push_back(gb);
  return true;  // Always modifies function
}