Record context-sensitive variable (args) for processing reflective ar…

…guments

src/main/java/pascal/taie/analysis/pta/plugin/reflection/ReflectiveActionModel.java

@@ -91,7 +91,7 @@ public class ReflectiveActionModel extends AnalysisModelPlugin {
     /**
      * Associates argument variable (Object[]) to reflective call edges.
      */
-    private final MultiMap<Var, ReflectiveCallEdge> reflectiveArgs = Maps.newMultiMap();
+    private final MultiMap<CSVar, ReflectiveCallEdge> reflectiveArgs = Maps.newMultiMap();
 
     /**
      * Set of invocations that are annotated by the reflection log.
@@ -308,7 +308,7 @@ public void onNewCallEdge(Edge<CSCallSite, CSMethod> edge) {
                 CSVar csArgs = csManager.getCSVar(callerCtx, args);
                 passReflectiveArgs(refEdge, solver.getPointsToSetOf(csArgs));
                 // record args for later-arrive array objects
-                reflectiveArgs.put(args, refEdge);
+                reflectiveArgs.put(csArgs, refEdge);
             }
             // pass return value
             Invoke invoke = refEdge.getCallSite().getCallSite();
@@ -349,8 +349,7 @@ private static boolean isConcerned(Type type) {
     @Override
     public void onNewPointsToSet(CSVar csVar, PointsToSet pts) {
         super.onNewPointsToSet(csVar, pts);
-        reflectiveArgs.get(csVar.getVar())
-                .forEach(edge -> passReflectiveArgs(edge, pts));
+        reflectiveArgs.get(csVar).forEach(edge -> passReflectiveArgs(edge, pts));
     }
 
     MultiMap<Invoke, Object> getAllTargets() {