sync with SVF

SVF-tools · Aug 28, 2024 · eee959a · eee959a
1 parent 6c1a9b5
commit eee959a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Assignment-2/Assignment-2.h b/Assignment-2/Assignment-2.h
@@ -51,7 +51,7 @@ class ICFGTraversal
     {
         for (const CallICFGNode *cs : pag->getCallSiteSet())
         {
-            const SVFFunction *fun = SVFUtil::getCallee(cs->getCallSite());
+            const SVFFunction *fun = cs->getCalledFunction();
             if (fun->getName() == "source")
             {
                 sources.insert(cs);
@@ -65,7 +65,7 @@ class ICFGTraversal
     {
         for (const CallICFGNode *cs : pag->getCallSiteSet())
         {
-            const SVFFunction *fun = SVFUtil::getCallee(cs->getCallSite());
+            const SVFFunction *fun = cs->getCalledFunction();
             if (fun->getName() == "sink")
             {
                 sinks.insert(cs);

diff --git a/Assignment-4/Assignment-4.cpp b/Assignment-4/Assignment-4.cpp
@@ -41,7 +41,7 @@ std::set<const CallICFGNode *>& TaintGraphTraversal::identifySources()
 {
     for (const CallICFGNode *cs : pag->getCallSiteSet())
     {
-        const SVFFunction *fun = SVFUtil::getCallee(cs->getCallSite());
+        const SVFFunction *fun = cs->getCalledFunction();
         if (checker_source_api.find(fun->getName()) != checker_source_api.end())
         {
             sources.insert(cs);
@@ -54,7 +54,7 @@ std::set<const CallICFGNode *>& TaintGraphTraversal::identifySinks()
 {
     for (const CallICFGNode *cs : pag->getCallSiteSet())
     {
-        const SVFFunction *fun = SVFUtil::getCallee(cs->getCallSite());
+        const SVFFunction *fun = cs->getCalledFunction();
         if (checker_sink_api.find(fun->getName()) != checker_sink_api.end())
         {
             sinks.insert(cs);