检测BasicBlock是否在循环中

时间:2018-05-02 15:25:59

标签: llvm static-analysis

我是LLVM的初学者,我正在尝试编写LLVM传递。我的传递很简单:它检测循环和函数调用。我想知道函数调用是否在循环中并检测子循环。

我遇到的问题是,当调用处于循环中时,我的pass会打印两次函数调用。这是我的代码:

void BlocksInLoop(Loop *L,unsigned nlvl){
  errs() <<"Loop level"<< nlvl << " {\n";
  BasicBlock* h = L->getHeader();
  ScalarEvolution *SE = &getAnalysis<ScalarEvolutionWrapperPass().getSE();
  errs() <<"Loop trip count :"<< SE->getSmallConstantTripCount(L) << "\n";
  std::vector<Loop*> subLoops = L-> getSubLoops();
  Loop::iterator j,f; 
  for (j = subLoops.begin(), f = subLoops.end();j!=f ; ++j)
    BlocksInLoop(*j ,nlvl+1);
  unsigned numBlocks = 0;
  Loop::block_iterator bb;
  for(bb = L-> block_begin(); bb != L-> block_end();++bb){
    BasicBlock* BB = *bb;
    for(BasicBlock::iterator i = BB->begin() , e = BB->end(); i!=e; ++i){
      if(isa<CallInst>(&(*i)) || isa<InvokeInst>(&(*i))){
        errs()<<"Call "<< cast<CallInst((*i))->getCalledFunction()->getName() << "\n"; 
        }
     }
  }
  errs()<< "}\n";
  } 

  virtual bool runOnFunction(Function &F){
    LoopInfo &LI = getAnalysis<LoopInfoWrapperPass>().getLoopInfo();
    errs() << "Function " << F.getName () + "{\n";
    for( Function::iterator b = F.begin() , be = F.end() ;b != be; ++b){
      for(BasicBlock::iterator i = b->begin() , ie = b->end();i != ie; i ++){
        if(isa<CallInst>(&(*i)) || isa<InvokeInst>(&(*i))){
          errs()<<"Call "<< cast<CallInst>(&(*i))->getCalledFunction()->getName() << "\n"; 
        }
      }
    }
    for(LoopInfo::iterator i = LI.begin(), e = LI.end(); i!=e; ++i)
      BlocksInLoop (*i,0);
    errs()<< "}\n\n"; 
    return(false);
  }

如果输入是例如:

void foo(){
for(int i=0; i++; i<10)
  foo();
}

这是我得到的输出:

Function foo{
Call foo
Loop level0 {
Loop trip count :0
Call foo
}
}

所以我正在寻找一种方法来检测Call是否处于循环中,如果是,则忽略第一次打印。有可能,如果可能,怎么样? (行程计数也是错误的,总是为0但我稍后会尝试修复它。)

1 个答案:

答案 0 :(得分:1)

这是实现目标的一种方法,跳过属于循环的那些基本块。

mkdir -p /etc/mysql/mysql.conf.d

}

注意:我没有运行此代码,可能会出现语法问题,但这是整体想法。