Merge pull request #1273 from jumormt/master

fix CFBBGbuilder bug
SVF-tools · Dec 6, 2023 · 9831850 · 9831850
2 parents e98ad8a + d2c158b
commit 9831850
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 11 deletions.
diff --git a/svf-llvm/lib/LLVMModule.cpp b/svf-llvm/lib/LLVMModule.cpp
@@ -364,6 +364,10 @@ void LLVMModuleSet::initSVFBasicBlock(const Function* func)
             LLVMUtil::getPrevInsts(inst, getSVFInstruction(inst)->getPredInstructions());
         }
     }
+    // For no return functions, we set the last block as exit BB
+    // This ensures that each function that has definition must have an exit BB
+    if(svfFun->exitBlock == nullptr && svfFun->hasBasicBlock()) svfFun->setExitBlock(
+                const_cast<SVFBasicBlock *>(svfFun->back()));
 }
 
 

diff --git a/svf/include/SVFIR/SVFValue.h b/svf/include/SVFIR/SVFValue.h
@@ -415,12 +415,7 @@ class SVFFunction : public SVFValue
 
     /// Carefully! when you call getExitBB, you need ensure the function has return instruction
     /// more refer to: https://github.com/SVF-tools/SVF/pull/1262
-    inline const SVFBasicBlock* getExitBB() const
-    {
-        assert(hasBasicBlock() && "function does not have any Basicblock, external function?");
-        assert((hasReturn() && exitBlock) && "ensure the function has a single basic block containing a return instruction!");
-        return exitBlock;
-    }
+    const SVFBasicBlock* getExitBB() const;
 
     void setExitBlock(SVFBasicBlock *bb);
 

diff --git a/svf/lib/Graphs/ICFG.cpp b/svf/lib/Graphs/ICFG.cpp
@@ -54,11 +54,7 @@ FunExitICFGNode::FunExitICFGNode(NodeID id, const SVFFunction* f)
     // if function is implemented
     if (f->begin() != f->end())
     {
-        // ensure the enclosing function has exit basic block
-        if (f->hasReturn())
-        {
-            bb = f->getExitBB();
-        }
+        bb = f->getExitBB();
     }
 }
 

diff --git a/svf/lib/SVFIR/SVFValue.cpp b/svf/lib/SVFIR/SVFValue.cpp
@@ -190,6 +190,14 @@ bool SVFFunction::isVarArg() const
     return varArg;
 }
 
+const SVFBasicBlock *SVFFunction::getExitBB() const
+{
+    assert(hasBasicBlock() && "function does not have any Basicblock, external function?");
+    assert((!hasReturn() || exitBlock->back()->isRetInst()) && "last inst must be return inst");
+    assert(exitBlock && "must have an exitBlock");
+    return exitBlock;
+}
+
 void SVFFunction::setExitBlock(SVFBasicBlock *bb)
 {
     assert(!exitBlock && "have already set exit Basicblock!");

diff --git a/svf/lib/Util/CFBasicBlockGBuilder.cpp b/svf/lib/Util/CFBasicBlockGBuilder.cpp
@@ -194,6 +194,8 @@ void CFBasicBlockGBuilder::addInterProceduralEdge(ICFG *icfg,
                 {
                     if (const RetCFGEdge *retEdge = SVFUtil::dyn_cast<RetCFGEdge>(icfgEdge))
                     {
+                        // no return function
+                        if(!retEdge->getSrcNode()->getFun()->hasReturn()) continue;
                         CFBasicBlockEdge *pEdge = new CFBasicBlockEdge(bbToNodes[retEdge->getSrcNode()->getBB()].back(),
                                 bbNodes.second[i],
                                 retEdge);