small fix

SVF-tools · Nov 5, 2024 · 1bcb6b6 · 1bcb6b6
1 parent 4deec17
commit 1bcb6b6
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/svf/include/Util/CallGraphBuilder.h b/svf/include/Util/CallGraphBuilder.h
@@ -49,6 +49,9 @@ class CallGraphBuilder
     /// Buidl SVFIR callgraoh
     CallGraph* buildSVFIRCallGraph(SVFModule* svfModule);
 
+    /// Buidl PTA callgraoh
+    PTACallGraph* buildPTACallGraph();
+
     /// Build thread-aware callgraph
     ThreadCallGraph* buildThreadCallGraph();
 };

diff --git a/svf/lib/MemoryModel/PointerAnalysis.cpp b/svf/lib/MemoryModel/PointerAnalysis.cpp
@@ -116,8 +116,8 @@ void PointerAnalysis::initialize()
     }
     else
     {
-        CallGraph* cg = pag->getCallGraph();
-        callgraph = new PTACallGraph(*cg);
+        CallGraphBuilder bd;
+        callgraph = bd.buildPTACallGraph();
     }
     callGraphSCCDetection();
 

diff --git a/svf/lib/Util/CallGraphBuilder.cpp b/svf/lib/Util/CallGraphBuilder.cpp
@@ -66,6 +66,12 @@ CallGraph* CallGraphBuilder::buildSVFIRCallGraph(SVFModule* svfModule)
     return callgraph;
 }
 
+PTACallGraph* CallGraphBuilder::buildPTACallGraph()
+{
+    CallGraph* svfirCallGraph = PAG::getPAG()->getCallGraph();
+    return new PTACallGraph(*svfirCallGraph);
+}
+
 ThreadCallGraph* CallGraphBuilder::buildThreadCallGraph()
 {
     CallGraph* svfirCallGraph = PAG::getPAG()->getCallGraph();