removed entrypoints field from jimple cfg

soot-oss · Oct 1, 2024 · dc383ff · dc383ff
1 parent f5ba85d
commit dc383ff
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/...ral/src/main/java/sootup/analysis/interprocedural/icfg/JimpleBasedInterproceduralCFG.java b/...ral/src/main/java/sootup/analysis/interprocedural/icfg/JimpleBasedInterproceduralCFG.java
@@ -55,7 +55,6 @@ public class JimpleBasedInterproceduralCFG extends AbstractJimpleBasedICFG {
 
   protected static final Logger logger =
       LoggerFactory.getLogger(JimpleBasedInterproceduralCFG.class);
-  private List<MethodSignature> entryPoints;
 
   protected boolean includeReflectiveCalls;
 
@@ -123,14 +122,17 @@ public JimpleBasedInterproceduralCFG(
     super(enableExceptions);
     this.includeReflectiveCalls = includeReflectiveCalls;
     this.view = view;
-    this.entryPoints = cg.getEntryMethods();
     this.cg = cg;
     initializeStmtToOwner();
   }
 
+  public CallGraph getCg() {
+    return cg;
+  }
+
   public String buildICFGGraph(CallGraph callGraph) {
     Map<MethodSignature, StmtGraph<?>> signatureToStmtGraph = new LinkedHashMap<>();
-    computeAllCalls(entryPoints, signatureToStmtGraph, callGraph);
+    computeAllCalls(callGraph.getEntryMethods(), signatureToStmtGraph, callGraph);
     return ICFGDotExporter.buildICFGGraph(signatureToStmtGraph, view, callGraph);
   }
 

diff --git a/...l/src/test/java/sootup/analysis/interprocedural/icfg/BackwardsInterproceduralCFGTest.java b/...l/src/test/java/sootup/analysis/interprocedural/icfg/BackwardsInterproceduralCFGTest.java
@@ -42,7 +42,8 @@ public static void setup() {
   @Test
   void methodStartAndEndPointTest() {
     JimpleBasedInterproceduralCFG forwardICFG =
-        new JimpleBasedInterproceduralCFG(view, entryMethodSignature, false, false);
+        new JimpleBasedInterproceduralCFG(
+            view, Collections.singletonList(entryMethodSignature), false, false);
 
     BackwardsInterproceduralCFG backwardsInterproceduralCFG =
         new BackwardsInterproceduralCFG(forwardICFG);
@@ -58,7 +59,8 @@ void methodStartAndEndPointTest() {
   @Test
   void methodToCallerStmtTest() {
     JimpleBasedInterproceduralCFG forwardICFG =
-        new JimpleBasedInterproceduralCFG(view, entryMethodSignature, false, false);
+        new JimpleBasedInterproceduralCFG(
+            view, Collections.singletonList(entryMethodSignature), false, false);
 
     BackwardsInterproceduralCFG backwardsInterproceduralCFG =
         new BackwardsInterproceduralCFG(forwardICFG);