From eee959a4046f8e652802ad76dfb6e0e8990acd43 Mon Sep 17 00:00:00 2001
From: jumormt <jumormt@163.com>
Date: Wed, 28 Aug 2024 20:41:01 +1000
Subject: [PATCH] sync with SVF

---
 Assignment-2/Assignment-2.h   | 4 ++--
 Assignment-4/Assignment-4.cpp | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Assignment-2/Assignment-2.h b/Assignment-2/Assignment-2.h
index a6af1dc..38e9787 100644
--- 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
index 2910faa..23341b7 100644
--- 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);