set last bb as exit bb for no return func

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()));
 }
 
 

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);
 

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();
     }
 }
 

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!");

svf/lib/Util/CFBasicBlockGBuilder.cpp

@@ -194,8 +194,8 @@ void CFBasicBlockGBuilder::addInterProceduralEdge(ICFG *icfg,
                 {
                     if (const RetCFGEdge *retEdge = SVFUtil::dyn_cast<RetCFGEdge>(icfgEdge))
                     {
-                        // no return function does not have an exit BB
-                        if(!retEdge->getSrcNode()->getBB()) continue;
+                        // no return function
+                        if(!retEdge->getSrcNode()->getFun()->hasReturn()) continue;
                         CFBasicBlockEdge *pEdge = new CFBasicBlockEdge(bbToNodes[retEdge->getSrcNode()->getBB()].back(),
                                 bbNodes.second[i],
                                 retEdge);