diff --git a/lib/Transforms/Instrumentation/EmitFunctions.cpp b/lib/Transforms/Instrumentation/EmitFunctions.cpp index 7b9810748fb..a676edcc01b 100644 --- a/lib/Transforms/Instrumentation/EmitFunctions.cpp +++ b/lib/Transforms/Instrumentation/EmitFunctions.cpp @@ -8,6 +8,13 @@ #include "llvm/DerivedTypes.h" #include "llvm/Module.h" #include "llvm/Pass.h" +#include "llvm/Support/CFG.h" + +enum Color{ + WHITE, + GREY, + BLACK +}; namespace { struct EmitFunctionTable : public Pass { @@ -17,15 +24,52 @@ namespace { RegisterOpt X("emitfuncs", "Emit a Function Table"); } +char doDFS(BasicBlock * node,std::map &color){ + color[node] = GREY; + + for(BasicBlock::succ_iterator vl = succ_begin(node), + ve = succ_end(node); vl != ve; ++vl){ + + BasicBlock *BB = *vl; + + if(color[BB]!=GREY && color[BB]!=BLACK){ + if(!doDFS(BB, color)){ + return 0; + } + } + + //if has backedge + else if(color[BB]==GREY) + return 0; + + } + + color[node] = BLACK; + return 1; +} + +char hasBackEdge(Function *F){ + std::map color; + return doDFS(F->begin(), color); +} + // Per Module pass for inserting function table bool EmitFunctionTable::run(Module &M){ std::vector vType; + std::vector vConsts; - unsigned char counter = 0; + std::vector sBCons; + + unsigned int counter = 0; for(Module::iterator MI = M.begin(), ME = M.end(); MI != ME; ++MI) if (!MI->isExternal()) { vType.push_back(MI->getType()); + + //std::cerr<getType(), true, + GlobalValue::ExternalLinkage, + constArray, "llvmSimpleFunction"); + + M.getGlobalList().push_back(funcArray); + ConstantInt *cnst = ConstantSInt::get(Type::IntTy, counter); GlobalVariable *fnCount = new GlobalVariable(Type::IntTy, true, GlobalValue::ExternalLinkage,