Polish

pascal-lab · May 29, 2024 · 943c4ec · 943c4ec
1 parent a99ba1a
commit 943c4ec
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/pascal/taie/analysis/pta/plugin/taint/TaintAnalysis.java b/src/main/java/pascal/taie/analysis/pta/plugin/taint/TaintAnalysis.java
@@ -86,12 +86,13 @@ private void initialize() {
         // trigger the creation of taint objects
         CallGraph<CSCallSite, CSMethod> cg = solver.getCallGraph();
         if (cg != null) {
+            boolean handleStmt = context.config().callSiteMode()
+                    || context.config().sources().stream().anyMatch(FieldSource.class::isInstance);
             cg.reachableMethods().forEach(csMethod -> {
                 JMethod method = csMethod.getMethod();
                 Context ctxt = csMethod.getContext();
                 IR ir = csMethod.getMethod().getIR();
-                if (context.config().callSiteMode() ||
-                        context.config().sources().stream().anyMatch(FieldSource.class::isInstance)) {
+                if (handleStmt) {
                     ir.forEach(stmt -> onNewStmt(stmt, method));
                 }
                 this.onNewCSMethod(csMethod);