Fix

pascal-lab · May 29, 2024 · a99ba1a · a99ba1a
1 parent 92c9f00
commit a99ba1a
Showing 1 changed file with 3 additions and 3 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
@@ -34,7 +34,6 @@
 import pascal.taie.analysis.pta.core.solver.Solver;
 import pascal.taie.analysis.pta.plugin.CompositePlugin;
 import pascal.taie.analysis.pta.pts.PointsToSet;
-import pascal.taie.config.AnalysisOptions;
 import pascal.taie.ir.IR;
 import pascal.taie.language.classes.JMethod;
 import pascal.taie.util.Timer;
@@ -91,11 +90,12 @@ private void initialize() {
                 JMethod method = csMethod.getMethod();
                 Context ctxt = csMethod.getContext();
                 IR ir = csMethod.getMethod().getIR();
-                if (context.config().callSiteMode()) {
+                if (context.config().callSiteMode() ||
+                        context.config().sources().stream().anyMatch(FieldSource.class::isInstance)) {
                     ir.forEach(stmt -> onNewStmt(stmt, method));
                 }
-                csMethod.getEdges().forEach(this::onNewCallEdge);
                 this.onNewCSMethod(csMethod);
+                csMethod.getEdges().forEach(this::onNewCallEdge);
                 ir.getParams().forEach(param -> {
                     CSVar csParam = csManager.getCSVar(ctxt, param);
                     onNewPointsToSet(csParam, csParam.getPointsToSet());