1 package Analysis.SSJava;
3 import java.io.BufferedReader;
4 import java.io.BufferedWriter;
5 import java.io.FileReader;
6 import java.io.FileWriter;
7 import java.io.IOException;
8 import java.util.ArrayList;
9 import java.util.Collections;
10 import java.util.Comparator;
11 import java.util.HashMap;
12 import java.util.HashSet;
13 import java.util.Iterator;
14 import java.util.LinkedList;
15 import java.util.List;
18 import java.util.Stack;
19 import java.util.Vector;
21 import IR.ClassDescriptor;
23 import IR.FieldDescriptor;
24 import IR.MethodDescriptor;
25 import IR.NameDescriptor;
28 import IR.SymbolTable;
29 import IR.TypeDescriptor;
31 import IR.VarDescriptor;
32 import IR.Tree.ArrayAccessNode;
33 import IR.Tree.AssignmentNode;
34 import IR.Tree.BlockExpressionNode;
35 import IR.Tree.BlockNode;
36 import IR.Tree.BlockStatementNode;
37 import IR.Tree.CastNode;
38 import IR.Tree.CreateObjectNode;
39 import IR.Tree.DeclarationNode;
40 import IR.Tree.ExpressionNode;
41 import IR.Tree.FieldAccessNode;
42 import IR.Tree.IfStatementNode;
44 import IR.Tree.LiteralNode;
45 import IR.Tree.LoopNode;
46 import IR.Tree.MethodInvokeNode;
47 import IR.Tree.NameNode;
48 import IR.Tree.OpNode;
49 import IR.Tree.ReturnNode;
50 import IR.Tree.SubBlockNode;
51 import IR.Tree.SwitchBlockNode;
52 import IR.Tree.SwitchStatementNode;
53 import IR.Tree.TertiaryNode;
54 import IR.Tree.TreeNode;
57 public class LocationInference {
62 SSJavaAnalysis ssjava;
65 List<ClassDescriptor> temp_toanalyzeList;
66 List<MethodDescriptor> temp_toanalyzeMethodList;
67 Map<MethodDescriptor, FlowGraph> mapMethodDescriptorToFlowGraph;
69 LinkedList<MethodDescriptor> toanalyze_methodDescList;
70 Set<ClassDescriptor> toanalyze_classDescSet;
72 // InheritanceTree<ClassDescriptor> inheritanceTree;
74 // map a method descriptor to its set of parameter descriptors
75 Map<MethodDescriptor, Set<Descriptor>> mapMethodDescriptorToParamDescSet;
77 // keep current descriptors to visit in fixed-point interprocedural analysis,
78 private Stack<MethodDescriptor> methodDescriptorsToVisitStack;
80 // map a descriptor to a naive lattice
81 private Map<Descriptor, SSJavaLattice<String>> desc2naiveLattice;
83 // map a class descriptor to a field lattice
84 private Map<ClassDescriptor, SSJavaLattice<String>> cd2lattice;
86 // map a method descriptor to a method lattice
87 private Map<MethodDescriptor, SSJavaLattice<String>> md2lattice;
89 // map a method/class descriptor to a hierarchy graph
90 private Map<Descriptor, HierarchyGraph> mapDescriptorToHierarchyGraph;
92 // map a method/class descriptor to a skeleton hierarchy graph
93 private Map<Descriptor, HierarchyGraph> mapDescriptorToSkeletonHierarchyGraph;
95 private Map<Descriptor, HierarchyGraph> mapDescriptorToSimpleHierarchyGraph;
97 // map a method/class descriptor to a skeleton hierarchy graph with combination nodes
98 private Map<Descriptor, HierarchyGraph> mapDescriptorToCombineSkeletonHierarchyGraph;
100 // map a descriptor to a simple lattice
101 private Map<Descriptor, SSJavaLattice<String>> mapDescriptorToSimpleLattice;
103 // map a method descriptor to the set of method invocation nodes which are
104 // invoked by the method descriptor
105 private Map<MethodDescriptor, Set<MethodInvokeNode>> mapMethodDescriptorToMethodInvokeNodeSet;
107 private Map<MethodInvokeNode, Map<Integer, NTuple<Descriptor>>> mapMethodInvokeNodeToArgIdxMap;
109 private Map<MethodInvokeNode, NTuple<Descriptor>> mapMethodInvokeNodeToBaseTuple;
111 private Map<MethodInvokeNode, Set<NTuple<Location>>> mapMethodInvokeNodeToPCLocTupleSet;
113 private Map<MethodDescriptor, MethodLocationInfo> mapMethodDescToMethodLocationInfo;
115 private Map<ClassDescriptor, LocationInfo> mapClassToLocationInfo;
117 private Map<MethodDescriptor, Set<MethodDescriptor>> mapMethodToCalleeSet;
119 private Map<MethodDescriptor, Set<FlowNode>> mapMethodDescToParamNodeFlowsToReturnValue;
121 private Map<String, Vector<String>> mapFileNameToLineVector;
123 private Map<Descriptor, Integer> mapDescToDefinitionLine;
125 private Map<Descriptor, LocationSummary> mapDescToLocationSummary;
127 private Map<MethodDescriptor, Set<MethodInvokeNode>> mapMethodDescToMethodInvokeNodeSet;
129 // maps a method descriptor to a sub global flow graph that captures all value flows caused by the
130 // set of callees reachable from the method
131 private Map<MethodDescriptor, GlobalFlowGraph> mapMethodDescriptorToSubGlobalFlowGraph;
133 private Map<MethodInvokeNode, Map<NTuple<Descriptor>, NTuple<Descriptor>>> mapMethodInvokeNodeToMapCallerArgToCalleeArg;
135 private Map<MethodDescriptor, Boolean> mapMethodDescriptorToCompositeReturnCase;
137 private Map<MethodDescriptor, MethodDescriptor> mapMethodDescToHighestOverriddenMethodDesc;
139 private Map<MethodDescriptor, Set<MethodDescriptor>> mapHighestOverriddenMethodDescToMethodDescSet;
141 private Map<MethodDescriptor, NTuple<Descriptor>> mapHighestOverriddenMethodDescToReturnLocTuple;
143 private Map<MethodDescriptor, NTuple<Descriptor>> mapHighestOverriddenMethodDescToPCLocTuple;
145 private Map<MethodDescriptor, Set<NTuple<Descriptor>>> mapHighestOverriddenMethodDescToSetLowerThanPCLoc;
147 private Map<MethodDescriptor, Set<NTuple<Descriptor>>> mapHighestOverriddenMethodDescToSetHigherThanRETURNLoc;
149 public static final String GLOBALLOC = "GLOBALLOC";
151 public static final String INTERLOC = "INTERLOC";
153 public static final String PCLOC = "_PCLOC_";
155 public static final String RLOC = "_RLOC_";
157 public static final Descriptor GLOBALDESC = new NameDescriptor(GLOBALLOC);
159 public static final Descriptor TOPDESC = new NameDescriptor(SSJavaAnalysis.TOP);
161 public static final Descriptor BOTTOMDESC = new NameDescriptor(SSJavaAnalysis.BOTTOM);
163 public static final Descriptor RETURNLOC = new NameDescriptor(RLOC);
165 public static final Descriptor LITERALDESC = new NameDescriptor("LITERAL");
167 public static final HNode TOPHNODE = new HNode(TOPDESC);
169 public static final HNode BOTTOMHNODE = new HNode(BOTTOMDESC);
171 public static String newline = System.getProperty("line.separator");
173 LocationInfo curMethodInfo;
175 private boolean hasChanges = false;
177 boolean debug = true;
179 public static int locSeed = 0;
181 private Stack<String> arrayAccessNodeStack;
183 private ClassDescriptor rootClassDescriptor;
185 private BuildLattice buildLattice;
187 public static int numLocationsSInfer = 0;
188 public static int numLocationsNaive = 0;
190 public LocationInference(SSJavaAnalysis ssjava, State state, TypeUtil tu) {
191 this.ssjava = ssjava;
194 this.toanalyze_classDescSet = new HashSet<ClassDescriptor>();
195 this.temp_toanalyzeList = new ArrayList<ClassDescriptor>();
196 this.temp_toanalyzeMethodList = new ArrayList<MethodDescriptor>();
197 this.mapMethodDescriptorToFlowGraph = new HashMap<MethodDescriptor, FlowGraph>();
198 this.cd2lattice = new HashMap<ClassDescriptor, SSJavaLattice<String>>();
199 this.md2lattice = new HashMap<MethodDescriptor, SSJavaLattice<String>>();
200 this.desc2naiveLattice = new HashMap<Descriptor, SSJavaLattice<String>>();
202 this.methodDescriptorsToVisitStack = new Stack<MethodDescriptor>();
203 this.mapMethodDescriptorToMethodInvokeNodeSet =
204 new HashMap<MethodDescriptor, Set<MethodInvokeNode>>();
205 this.mapMethodInvokeNodeToArgIdxMap =
206 new HashMap<MethodInvokeNode, Map<Integer, NTuple<Descriptor>>>();
207 this.mapMethodDescToMethodLocationInfo = new HashMap<MethodDescriptor, MethodLocationInfo>();
208 this.mapMethodToCalleeSet = new HashMap<MethodDescriptor, Set<MethodDescriptor>>();
209 this.mapClassToLocationInfo = new HashMap<ClassDescriptor, LocationInfo>();
211 this.mapFileNameToLineVector = new HashMap<String, Vector<String>>();
212 this.mapDescToDefinitionLine = new HashMap<Descriptor, Integer>();
213 this.mapMethodDescToParamNodeFlowsToReturnValue =
214 new HashMap<MethodDescriptor, Set<FlowNode>>();
216 this.mapDescriptorToHierarchyGraph = new HashMap<Descriptor, HierarchyGraph>();
217 this.mapMethodInvokeNodeToBaseTuple = new HashMap<MethodInvokeNode, NTuple<Descriptor>>();
219 this.mapDescriptorToSkeletonHierarchyGraph = new HashMap<Descriptor, HierarchyGraph>();
220 this.mapDescriptorToCombineSkeletonHierarchyGraph = new HashMap<Descriptor, HierarchyGraph>();
221 this.mapDescriptorToSimpleHierarchyGraph = new HashMap<Descriptor, HierarchyGraph>();
223 this.mapDescriptorToSimpleLattice = new HashMap<Descriptor, SSJavaLattice<String>>();
225 this.mapDescToLocationSummary = new HashMap<Descriptor, LocationSummary>();
227 this.mapMethodDescriptorToSubGlobalFlowGraph = new HashMap<MethodDescriptor, GlobalFlowGraph>();
229 this.mapMethodInvokeNodeToMapCallerArgToCalleeArg =
230 new HashMap<MethodInvokeNode, Map<NTuple<Descriptor>, NTuple<Descriptor>>>();
232 this.mapMethodInvokeNodeToPCLocTupleSet =
233 new HashMap<MethodInvokeNode, Set<NTuple<Location>>>();
235 this.arrayAccessNodeStack = new Stack<String>();
237 this.mapMethodDescToMethodInvokeNodeSet =
238 new HashMap<MethodDescriptor, Set<MethodInvokeNode>>();
240 this.mapMethodDescriptorToCompositeReturnCase = new HashMap<MethodDescriptor, Boolean>();
242 mapMethodDescToHighestOverriddenMethodDesc = new HashMap<MethodDescriptor, MethodDescriptor>();
244 mapHighestOverriddenMethodDescToSetLowerThanPCLoc =
245 new HashMap<MethodDescriptor, Set<NTuple<Descriptor>>>();
247 mapHighestOverriddenMethodDescToMethodDescSet =
248 new HashMap<MethodDescriptor, Set<MethodDescriptor>>();
250 mapHighestOverriddenMethodDescToReturnLocTuple =
251 new HashMap<MethodDescriptor, NTuple<Descriptor>>();
253 mapHighestOverriddenMethodDescToPCLocTuple =
254 new HashMap<MethodDescriptor, NTuple<Descriptor>>();
256 mapHighestOverriddenMethodDescToSetHigherThanRETURNLoc =
257 new HashMap<MethodDescriptor, Set<NTuple<Descriptor>>>();
259 this.buildLattice = new BuildLattice(this);
263 public void setupToAnalyze() {
264 SymbolTable classtable = state.getClassSymbolTable();
265 temp_toanalyzeList.clear();
266 temp_toanalyzeList.addAll(classtable.getValueSet());
267 // Collections.sort(toanalyzeList, new Comparator<ClassDescriptor>() {
268 // public int compare(ClassDescriptor o1, ClassDescriptor o2) {
269 // return o1.getClassName().compareToIgnoreCase(o2.getClassName());
274 public void setupToAnalazeMethod(ClassDescriptor cd) {
276 SymbolTable methodtable = cd.getMethodTable();
277 temp_toanalyzeMethodList.clear();
278 temp_toanalyzeMethodList.addAll(methodtable.getValueSet());
279 Collections.sort(temp_toanalyzeMethodList, new Comparator<MethodDescriptor>() {
280 public int compare(MethodDescriptor o1, MethodDescriptor o2) {
281 return o1.getSymbol().compareToIgnoreCase(o2.getSymbol());
286 public boolean toAnalyzeMethodIsEmpty() {
287 return temp_toanalyzeMethodList.isEmpty();
290 public boolean toAnalyzeIsEmpty() {
291 return temp_toanalyzeList.isEmpty();
294 public ClassDescriptor toAnalyzeNext() {
295 return temp_toanalyzeList.remove(0);
298 public MethodDescriptor toAnalyzeMethodNext() {
299 return temp_toanalyzeMethodList.remove(0);
302 public void inference() {
306 // construct value flow graph
307 constructFlowGraph();
309 constructGlobalFlowGraph();
313 assignCompositeLocation();
315 calculateExtraLocations();
316 addAdditionalOrderingConstraints();
318 _debug_writeFlowGraph();
320 buildInheritanceTree();
321 calculateReturnPCLocInheritance();
323 constructHierarchyGraph();
325 addInheritanceConstraintsToHierarchyGraph();
327 debug_writeHierarchyDotFiles();
329 simplifyHierarchyGraph();
331 debug_writeSimpleHierarchyDotFiles();
333 constructSkeletonHierarchyGraph();
335 debug_writeSkeletonHierarchyDotFiles();
337 insertCombinationNodes();
339 debug_writeSkeletonCombinationHierarchyDotFiles();
341 buildLatticeInheritanceTree();
344 debug_writeLattices();
346 updateCompositeLocationAssignments();
348 generateMethodSummary();
350 generateAnnoatedCode();
352 for (Iterator iterator = cd2lattice.keySet().iterator(); iterator.hasNext();) {
353 ClassDescriptor cd = (ClassDescriptor) iterator.next();
354 SSJavaLattice<String> lattice = getLattice(cd);
355 HierarchyGraph hg = mapDescriptorToHierarchyGraph.get(cd);
356 // System.out.println("~~~\t" + cd + "\t" + lattice.getKeySet().size() + "\t"
357 // + hg.getNodeSet().size());
360 for (Iterator iterator = md2lattice.keySet().iterator(); iterator.hasNext();) {
361 MethodDescriptor md = (MethodDescriptor) iterator.next();
362 SSJavaLattice<String> locOrder = getLattice(md);
363 // writeLatticeDotFile(md.getClassDesc(), md, getMethodLattice(md));
364 HierarchyGraph hg = mapDescriptorToHierarchyGraph.get(md);
365 // System.out.println("~~~\t" + md.getClassDesc() + "_" + md + "\t"
366 // + locOrder.getKeySet().size() + "\t" + hg.getNodeSet().size());
369 if (state.SSJAVA_INFER_NAIVE_WRITEDOTS) {
370 System.out.println("The number of elements: Naive=" + numLocationsNaive);
372 System.out.println("The number of elements: SInfer=" + numLocationsSInfer);
378 private void calculateReturnPCLocInheritance() {
379 calculateHighestPCLocInheritance();
380 calculateLowestReturnLocInheritance();
381 updateFlowGraphPCReturnLocInheritance();
384 private void updateFlowGraphPCReturnLocInheritance() {
386 Set<MethodDescriptor> keySet = mapHighestOverriddenMethodDescToMethodDescSet.keySet();
387 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
388 MethodDescriptor highestMethodDesc = (MethodDescriptor) iterator.next();
390 if (mapHighestOverriddenMethodDescToMethodDescSet.get(highestMethodDesc).size() == 1) {
394 Set<MethodDescriptor> methodDescSet =
395 mapHighestOverriddenMethodDescToMethodDescSet.get(highestMethodDesc);
397 NTuple<Descriptor> highestPCLocDescTuple =
398 mapHighestOverriddenMethodDescToPCLocTuple.get(highestMethodDesc);
400 NTuple<Descriptor> highestRETURNLocDescTuple =
401 mapHighestOverriddenMethodDescToReturnLocTuple.get(highestMethodDesc);
403 // System.out.println("\n$$$$$$$$$$$$$$$$updateFlowGraphPCReturnLocInheritance="
404 // + highestMethodDesc);
405 // System.out.println("-----highestPCLoc=" + highestPCLocDescTuple);
406 // System.out.println("-----highestRETURNLoc=" + highestRETURNLocDescTuple);
408 for (Iterator iterator2 = methodDescSet.iterator(); iterator2.hasNext();) {
409 MethodDescriptor md = (MethodDescriptor) iterator2.next();
410 // System.out.println("\n --------MD=" + md);
411 FlowGraph flowGraph = getFlowGraph(md);
413 MethodSummary summary = getMethodSummary(md);
414 CompositeLocation curPCLoc = summary.getPCLoc();
415 NTuple<Descriptor> curPCDescTuple = translateToDescTuple(curPCLoc.getTuple());
416 // System.out.println("md=" + md + " curPCLoc=" + curPCLoc);
417 // System.out.println("highestPCLoc=" + highestPCLocDescTuple);
419 if (highestPCLocDescTuple == null) {
420 // this case: PCLOC is top
421 // System.out.println("###SET PCLOC AS TOP");
422 if (curPCDescTuple != null && !curPCLoc.get(0).isTop()) {
423 FlowNode pcFlowNode = flowGraph.getFlowNode(curPCDescTuple);
424 flowGraph.removeNode(pcFlowNode);
426 summary.setPCLoc(new CompositeLocation(new Location(md, Location.TOP)));
428 NTuple<Descriptor> newPCDescTuple = new NTuple<Descriptor>();
429 if (highestPCLocDescTuple.size() == 1) {
430 newPCDescTuple.add(highestPCLocDescTuple.get(0));
432 newPCDescTuple.add(md.getThis());
433 newPCDescTuple.add(highestPCLocDescTuple.get(1));
435 if (!curPCDescTuple.equals(newPCDescTuple)) {
436 FlowNode pcFlowNode = flowGraph.getFlowNode(curPCDescTuple);
437 flowGraph.updateTuple(pcFlowNode, newPCDescTuple);
438 // flowGraph.removeNode(pcFlowNode);
439 Set<NTuple<Descriptor>> descSetLowerThanPCLoc =
440 mapHighestOverriddenMethodDescToSetLowerThanPCLoc.get(highestMethodDesc);
441 for (Iterator iterator3 = descSetLowerThanPCLoc.iterator(); iterator3.hasNext();) {
442 NTuple<Descriptor> lowerNTuple = (NTuple<Descriptor>) iterator3.next();
443 flowGraph.addValueFlowEdge(newPCDescTuple, lowerNTuple);
445 CompositeLocation newPCCompLoc =
446 new CompositeLocation(translateToLocTuple(md, newPCDescTuple));
447 summary.setPCLoc(newPCCompLoc);
453 if (highestRETURNLocDescTuple != null) {
454 CompositeLocation curRETURNLoc = summary.getRETURNLoc();
455 NTuple<Descriptor> curReturnDescTuple = translateToDescTuple(curRETURNLoc.getTuple());
457 if (!curReturnDescTuple.equals(highestRETURNLocDescTuple)) {
458 // handle the case that RETURNLOC is started with 'this'...
459 NTuple<Descriptor> newRETURNLocDescTuple = new NTuple<Descriptor>();
460 if (highestRETURNLocDescTuple.size() == 1) {
461 newRETURNLocDescTuple.add(highestRETURNLocDescTuple.get(0));
463 newRETURNLocDescTuple.add(md.getThis());
464 newRETURNLocDescTuple.add(highestRETURNLocDescTuple.get(1));
467 FlowNode returnFlowNode = flowGraph.getFlowNode(curReturnDescTuple);
468 flowGraph.updateTuple(returnFlowNode, newRETURNLocDescTuple);
470 Set<NTuple<Descriptor>> descSetHigherThanRETURNLoc =
471 mapHighestOverriddenMethodDescToSetHigherThanRETURNLoc.get(highestMethodDesc);
472 for (Iterator iterator3 = descSetHigherThanRETURNLoc.iterator(); iterator3.hasNext();) {
473 NTuple<Descriptor> higherNTuple = (NTuple<Descriptor>) iterator3.next();
474 flowGraph.addValueFlowEdge(higherNTuple, newRETURNLocDescTuple);
477 CompositeLocation newRETURNLocCompLoc =
478 new CompositeLocation(translateToLocTuple(md, newRETURNLocDescTuple));
479 summary.setRETURNLoc(newRETURNLocCompLoc);
486 private void calculateHighestPCLocInheritance() {
488 Set<MethodDescriptor> keySet = mapHighestOverriddenMethodDescToMethodDescSet.keySet();
490 Map<MethodDescriptor, Integer> mapMethodDescToParamCount =
491 new HashMap<MethodDescriptor, Integer>();
493 next: for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
494 MethodDescriptor highestMethodDesc = (MethodDescriptor) iterator.next();
496 NTuple<Descriptor> tempTuple = null;
498 if (getMethodSummary(highestMethodDesc).getPCLoc() != null) {
500 Set<MethodDescriptor> methodDescSet =
501 mapHighestOverriddenMethodDescToMethodDescSet.get(highestMethodDesc);
503 if (methodDescSet.size() > 1) {
504 // System.out.println("---method desc set=" + methodDescSet + " from=" +
505 // highestMethodDesc);
510 for (Iterator iterator2 = methodDescSet.iterator(); iterator2.hasNext();) {
511 MethodDescriptor md = (MethodDescriptor) iterator2.next();
513 FlowGraph flowGraph = getFlowGraph(md);
514 if (flowGraph == null) {
517 Set<FlowNode> paramNodeSet = flowGraph.getParamFlowNodeSet();
518 // System.out.println("###md=" + md + " paramNodeSet=" + paramNodeSet);
520 CompositeLocation pcLOC = getMethodSummary(md).getPCLoc();
521 // System.out.println("---pcLOC=" + pcLOC);
523 if (md.equals(highestMethodDesc)) {
524 mapHighestOverriddenMethodDescToPCLocTuple.put(highestMethodDesc,
525 translateToDescTuple(pcLOC.getTuple()));
528 if (!pcLOC.get(0).isTop()) {
530 FlowNode pcFlowNode = flowGraph.getFlowNode(translateToDescTuple(pcLOC.getTuple()));
533 for (Iterator iterator3 = paramNodeSet.iterator(); iterator3.hasNext();) {
534 FlowNode paramNode = (FlowNode) iterator3.next();
535 if (flowGraph.getReachableSetFrom(pcFlowNode.getCurrentDescTuple().subList(0, 1))
536 .contains(paramNode)) {
540 mapMethodDescToParamCount.put(md, count);
544 // the PC location is top
545 // if one of pcloc among the method inheritance chain has the TOP,
546 // all methods in the same chain should have the TOP.
547 mapHighestOverriddenMethodDescToPCLocTuple.remove(highestMethodDesc);
548 // System.out.println("highest=" + highestMethodDesc + " HIGHEST PCLOC="
549 // + mapHighestOverriddenMethodDescToPCLocTuple.get(highestMethodDesc));
551 Set<NTuple<Descriptor>> descTupleSetLowerThanPC = new HashSet<NTuple<Descriptor>>();
552 for (Iterator iterator3 = paramNodeSet.iterator(); iterator3.hasNext();) {
553 FlowNode flowNode = (FlowNode) iterator3.next();
554 descTupleSetLowerThanPC.add(flowNode.getCurrentDescTuple());
556 mapHighestOverriddenMethodDescToSetLowerThanPCLoc.put(highestMethodDesc,
557 descTupleSetLowerThanPC);
563 // identify which method in the inheritance chain has the highest PCLOC
564 // basically, finds a method that has the highest count or TOP location
565 int highestCount = -1;
566 MethodDescriptor methodDescHighestCount = null;
567 for (Iterator iterator2 = methodDescSet.iterator(); iterator2.hasNext();) {
568 MethodDescriptor methodDesc = (MethodDescriptor) iterator2.next();
569 if (mapMethodDescToParamCount.containsKey(methodDesc)) {
570 int curCount = mapMethodDescToParamCount.get(methodDesc).intValue();
571 if (highestCount < curCount) {
572 highestCount = curCount;
573 methodDescHighestCount = methodDesc;
578 if (methodDescHighestCount != null) {
579 FlowGraph flowGraph = getFlowGraph(methodDescHighestCount);
580 CompositeLocation pcLOC = getMethodSummary(methodDescHighestCount).getPCLoc();
581 FlowNode pcFlowNode = flowGraph.getFlowNode(translateToDescTuple(pcLOC.getTuple()));
582 Set<FlowNode> reachableSet =
583 flowGraph.getReachableSetFrom(pcFlowNode.getCurrentDescTuple().subList(0, 1));
585 Set<FlowNode> reachableParamNodeSet = new HashSet<FlowNode>();
586 for (Iterator iterator3 = reachableSet.iterator(); iterator3.hasNext();) {
587 FlowNode flowNode = (FlowNode) iterator3.next();
588 if (flowGraph.isParameter(flowNode.getCurrentDescTuple())) {
589 reachableParamNodeSet.add(flowNode);
594 Set<NTuple<Descriptor>> descTupleSetLowerThanPC = new HashSet<NTuple<Descriptor>>();
595 for (Iterator iterator2 = reachableParamNodeSet.iterator(); iterator2.hasNext();) {
596 FlowNode flowNode = (FlowNode) iterator2.next();
597 descTupleSetLowerThanPC.add(flowNode.getCurrentDescTuple());
600 // mapHighestOverriddenMethodDescToPCLocTuple.remove(highestMethodDesc);
601 mapHighestOverriddenMethodDescToSetLowerThanPCLoc.put(highestMethodDesc,
602 descTupleSetLowerThanPC);
607 // System.out.println("####################################");
608 // System.out.println(" highest=" + highestMethodDesc + " HIGHEST PCLOC="
609 // + mapHighestOverriddenMethodDescToPCLocTuple.get(highestMethodDesc));
610 // System.out.println(" setLowerThanPCLoc="
611 // + mapHighestOverriddenMethodDescToSetLowerThanPCLoc.get(highestMethodDesc));
616 private void calculateLowestReturnLocInheritance() {
618 Set<MethodDescriptor> keySet = mapHighestOverriddenMethodDescToMethodDescSet.keySet();
620 Map<MethodDescriptor, Integer> mapMethodDescToParamCount =
621 new HashMap<MethodDescriptor, Integer>();
623 next: for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
624 MethodDescriptor highestMethodDesc = (MethodDescriptor) iterator.next();
626 Set<MethodDescriptor> methodDescSet =
627 mapHighestOverriddenMethodDescToMethodDescSet.get(highestMethodDesc);
629 if (methodDescSet.size() > 1 && getMethodSummary(highestMethodDesc).getRETURNLoc() != null) {
634 for (Iterator iterator2 = methodDescSet.iterator(); iterator2.hasNext();) {
635 MethodDescriptor md = (MethodDescriptor) iterator2.next();
637 FlowGraph flowGraph = getFlowGraph(md);
638 Set<FlowNode> paramNodeSet = flowGraph.getParamFlowNodeSet();
640 CompositeLocation returnLoc = getMethodSummary(md).getRETURNLoc();
642 FlowNode returnFlowNode = flowGraph.getFlowNode(translateToDescTuple(returnLoc.getTuple()));
645 for (Iterator iterator3 = paramNodeSet.iterator(); iterator3.hasNext();) {
646 FlowNode paramNode = (FlowNode) iterator3.next();
647 if (flowGraph.getReachableSetFrom(paramNode.getCurrentDescTuple().subList(0, 1))
648 .contains(returnFlowNode)) {
652 mapMethodDescToParamCount.put(md, count);
653 // System.out.println("###returnLoc=" + returnLoc + " count higher=" + count);
656 // identify which method in the inheritance chain has the highest PCLOC
657 // basically, finds a method that has the highest count or TOP location
658 int highestCount = -1;
659 MethodDescriptor methodDescHighestCount = null;
660 for (Iterator iterator2 = methodDescSet.iterator(); iterator2.hasNext();) {
661 MethodDescriptor methodDesc = (MethodDescriptor) iterator2.next();
662 int curCount = mapMethodDescToParamCount.get(methodDesc).intValue();
663 if (highestCount < curCount) {
664 highestCount = curCount;
665 methodDescHighestCount = methodDesc;
669 if (methodDescHighestCount != null) {
670 FlowGraph flowGraph = getFlowGraph(methodDescHighestCount);
671 CompositeLocation returnLOC = getMethodSummary(methodDescHighestCount).getRETURNLoc();
672 NTuple<Descriptor> returnLocTuple = translateToDescTuple(returnLOC.getTuple());
673 FlowNode returnFlowNode = flowGraph.getFlowNode(returnLocTuple);
675 Set<FlowNode> curMethodParamNodeSet = flowGraph.getParamFlowNodeSet();
676 Set<NTuple<Descriptor>> descTupleSetHigherThanPC = new HashSet<NTuple<Descriptor>>();
677 for (Iterator iterator3 = curMethodParamNodeSet.iterator(); iterator3.hasNext();) {
678 FlowNode paramNode = (FlowNode) iterator3.next();
679 if (flowGraph.getReachableSetFrom(paramNode.getCurrentDescTuple().subList(0, 1))
680 .contains(returnFlowNode)) {
681 descTupleSetHigherThanPC.add(paramNode.getCurrentDescTuple());
685 mapHighestOverriddenMethodDescToReturnLocTuple.put(highestMethodDesc, returnLocTuple);
686 mapHighestOverriddenMethodDescToSetHigherThanRETURNLoc.put(highestMethodDesc,
687 descTupleSetHigherThanPC);
691 // System.out.println("####################################");
692 // System.out.println(" highest=" + highestMethodDesc + " LOWEST RETURNLOC="
693 // + mapHighestOverriddenMethodDescToReturnLocTuple.get(highestMethodDesc));
694 // System.out.println(" setHigherThanReturnLoc="
695 // + mapHighestOverriddenMethodDescToSetHigherThanRETURNLoc.get(highestMethodDesc));
701 private void addMapHighestMethodDescToMethodDesc(MethodDescriptor highest, MethodDescriptor md) {
702 if (!mapHighestOverriddenMethodDescToMethodDescSet.containsKey(highest)) {
703 mapHighestOverriddenMethodDescToMethodDescSet.put(highest, new HashSet<MethodDescriptor>());
705 mapHighestOverriddenMethodDescToMethodDescSet.get(highest).add(md);
708 private void DFSInheritanceTreeCalculatingHighestOverriddenMethod(ClassDescriptor cd) {
710 // ClassDescriptor cd = node.getData();
712 for (Iterator iterator = cd.getMethods(); iterator.hasNext();) {
713 MethodDescriptor md = (MethodDescriptor) iterator.next();
714 MethodDescriptor highestMethodDesc = getHighestOverriddenMethod(md.getClassDesc(), md);
715 mapMethodDescToHighestOverriddenMethodDesc.put(md, highestMethodDesc);
716 addMapHighestMethodDescToMethodDesc(highestMethodDesc, md);
721 Set<ClassDescriptor> children = getDirectSubClasses(cd);
722 for (Iterator iterator = children.iterator(); iterator.hasNext();) {
723 ClassDescriptor child = (ClassDescriptor) iterator.next();
724 DFSInheritanceTreeCalculatingHighestOverriddenMethod(child);
729 private MethodDescriptor getHighestOverriddenMethod(ClassDescriptor curClassDesc,
730 MethodDescriptor curMethodDesc) {
732 // Node<ClassDescriptor> curNode = inheritanceTree.getTreeNode(curClassDesc);
733 // Node<ClassDescriptor> parentNode = curNode.getParent();
734 ClassDescriptor parentClassDesc = curClassDesc.getSuperDesc();
736 if (parentClassDesc != null) {
737 if (parentClassDesc.getMethodTable().contains(curMethodDesc.getSymbol())) {
738 Set<MethodDescriptor> methodDescSet =
739 parentClassDesc.getMethodTable().getSet(curMethodDesc.getSymbol());
740 for (Iterator iterator = methodDescSet.iterator(); iterator.hasNext();) {
741 MethodDescriptor md = (MethodDescriptor) iterator.next();
742 if (md.matches(curMethodDesc)) {
743 return getHighestOverriddenMethod(parentClassDesc, md);
747 // traverse to the parent!
748 return getHighestOverriddenMethod(parentClassDesc, curMethodDesc);
750 return curMethodDesc;
753 private void buildInheritanceTree() {
755 DFSInheritanceTreeCalculatingHighestOverriddenMethod(rootClassDescriptor);
759 private void addInheritanceConstraintsToHierarchyGraph() {
761 // DFS the inheritance tree and propagates nodes/edges of parent to child
763 // Node<ClassDescriptor> rootNode = inheritanceTree.getRootNode();
764 DFSInheritanceTree(rootClassDescriptor);
768 private void DFSInheritanceTree(ClassDescriptor parentClassDescriptor) {
770 // ClassDescriptor parentClassDescriptor = parentNode.getData();
772 Set<ClassDescriptor> children = getDirectSubClasses(parentClassDescriptor);
773 for (Iterator iterator = children.iterator(); iterator.hasNext();) {
774 ClassDescriptor childClassDescriptor = (ClassDescriptor) iterator.next();
776 HierarchyGraph parentGraph = getHierarchyGraph(parentClassDescriptor);
777 HierarchyGraph childGraph = getHierarchyGraph(childClassDescriptor);
779 Set<HNode> parentNodeSet = parentGraph.getNodeSet();
780 for (Iterator iterator2 = parentNodeSet.iterator(); iterator2.hasNext();) {
781 HNode hNode = (HNode) iterator2.next();
782 childGraph.addNode(hNode);
785 // copies extra information from the parent hierarchy graph
786 Map<HNode, Set<HNode>> parentMergeNodeMap = parentGraph.getMapHNodetoMergeSet();
787 Map<HNode, Set<HNode>> childMergeNodeMap = childGraph.getMapHNodetoMergeSet();
789 Set<HNode> keySet = parentMergeNodeMap.keySet();
790 for (Iterator iterator2 = keySet.iterator(); iterator2.hasNext();) {
791 HNode parentKey = (HNode) iterator2.next();
792 if (!childMergeNodeMap.containsKey(parentKey)) {
793 childMergeNodeMap.put(parentKey, new HashSet<HNode>());
795 childMergeNodeMap.get(parentKey).addAll(parentMergeNodeMap.get(parentKey));
798 // copies nodes/edges from the parent class...
799 for (Iterator iterator2 = parentNodeSet.iterator(); iterator2.hasNext();) {
800 HNode parentHNode = (HNode) iterator2.next();
802 Set<HNode> parentIncomingHNode = parentGraph.getIncomingNodeSet(parentHNode);
803 Set<HNode> parentOutgoingHNode = parentGraph.getOutgoingNodeSet(parentHNode);
805 for (Iterator iterator3 = parentIncomingHNode.iterator(); iterator3.hasNext();) {
806 HNode inHNode = (HNode) iterator3.next();
807 childGraph.addEdge(inHNode.getDescriptor(), parentHNode.getDescriptor());
810 for (Iterator iterator3 = parentOutgoingHNode.iterator(); iterator3.hasNext();) {
811 HNode outHNode = (HNode) iterator3.next();
812 childGraph.addEdge(parentHNode.getDescriptor(), outHNode.getDescriptor());
817 // copies nodes/edges from parent methods to overridden methods
819 for (Iterator iterator3 = childClassDescriptor.getMethods(); iterator3.hasNext();) {
820 MethodDescriptor childMethodDescriptor = (MethodDescriptor) iterator3.next();
822 MethodDescriptor parentMethodDesc =
823 getParentMethodDesc(childMethodDescriptor.getClassDesc(), childMethodDescriptor);
825 if (parentMethodDesc != null) {
827 HierarchyGraph parentMethodGraph = getHierarchyGraph(parentMethodDesc);
828 HierarchyGraph childMethodGraph = getHierarchyGraph(childMethodDescriptor);
830 Set<HNode> parentMethodNodeSet = parentMethodGraph.getNodeSet();
831 for (Iterator iterator2 = parentMethodNodeSet.iterator(); iterator2.hasNext();) {
832 HNode hNode = (HNode) iterator2.next();
833 childMethodGraph.addNode(hNode);
836 // copies extra information from the parent hierarchy graph
837 Map<HNode, Set<HNode>> parentMethodMergeNodeMap =
838 parentMethodGraph.getMapHNodetoMergeSet();
839 Map<HNode, Set<HNode>> childMethodMergeNodeMap = childMethodGraph.getMapHNodetoMergeSet();
841 Set<HNode> methodKeySet = parentMethodMergeNodeMap.keySet();
842 for (Iterator iterator2 = methodKeySet.iterator(); iterator2.hasNext();) {
843 HNode parentKey = (HNode) iterator2.next();
844 if (!childMethodMergeNodeMap.containsKey(parentKey)) {
845 childMethodMergeNodeMap.put(parentKey, new HashSet<HNode>());
847 childMethodMergeNodeMap.get(parentKey).addAll(parentMethodMergeNodeMap.get(parentKey));
850 // copies nodes/edges from the parent method...
851 for (Iterator iterator2 = parentMethodGraph.getNodeSet().iterator(); iterator2.hasNext();) {
852 HNode parentHNode = (HNode) iterator2.next();
854 Set<HNode> parentIncomingHNode = parentMethodGraph.getIncomingNodeSet(parentHNode);
855 Set<HNode> parentOutgoingHNode = parentMethodGraph.getOutgoingNodeSet(parentHNode);
857 for (Iterator iterator4 = parentIncomingHNode.iterator(); iterator4.hasNext();) {
858 HNode inHNode = (HNode) iterator4.next();
859 childMethodGraph.addEdge(inHNode, parentHNode);
862 for (Iterator iterator4 = parentOutgoingHNode.iterator(); iterator4.hasNext();) {
863 HNode outHNode = (HNode) iterator4.next();
864 childMethodGraph.addEdge(parentHNode, outHNode);
873 DFSInheritanceTree(childClassDescriptor);
878 public MethodDescriptor getParentMethodDesc(ClassDescriptor classDesc, MethodDescriptor methodDesc) {
880 // Node<ClassDescriptor> childNode = inheritanceTree.getTreeNode(classDesc);
881 ClassDescriptor parentClassDesc = classDesc.getSuperDesc();
882 // Node<ClassDescriptor> parentNode = childNode.getParent();
884 if (parentClassDesc != null) {
885 // ClassDescriptor parentClassDesc = parentNode.getData();
886 if (parentClassDesc.getMethodTable().contains(methodDesc.getSymbol())) {
887 Set<MethodDescriptor> methodDescSet =
888 parentClassDesc.getMethodTable().getSet(methodDesc.getSymbol());
889 for (Iterator iterator = methodDescSet.iterator(); iterator.hasNext();) {
890 MethodDescriptor md = (MethodDescriptor) iterator.next();
891 if (md.matches(methodDesc)) {
897 // traverse to the parent!
898 getParentMethodDesc(parentClassDesc, methodDesc);
905 private void checkReturnNodes() {
906 LinkedList<MethodDescriptor> methodDescList =
907 (LinkedList<MethodDescriptor>) toanalyze_methodDescList.clone();
909 while (!methodDescList.isEmpty()) {
910 MethodDescriptor md = methodDescList.removeLast();
912 if (md.getReturnType() != null && !md.getReturnType().isVoid()) {
913 checkFlowNodeReturnThisField(md);
915 // // in this case, this method will return the composite location that starts with 'this'
916 // FlowGraph flowGraph = getFlowGraph(md);
917 // Set<FlowNode> returnNodeSet = flowGraph.getReturnNodeSet();
924 private void addSuperClasses(ClassDescriptor cd) {
925 ClassDescriptor parentClassDesc = cd.getSuperDesc();
926 if (parentClassDesc != null) {
927 toanalyze_classDescSet.add(parentClassDesc);
928 addSuperClasses(parentClassDesc);
932 private void updateFlowGraph() {
934 LinkedList<MethodDescriptor> methodDescList =
935 (LinkedList<MethodDescriptor>) toanalyze_methodDescList.clone();
937 while (!methodDescList.isEmpty()) {
938 MethodDescriptor md = methodDescList.removeLast();
939 if (state.SSJAVADEBUG) {
940 System.out.println();
941 System.out.println("SSJAVA: Updating a flow graph: " + md);
942 propagateFlowsFromCalleesWithNoCompositeLocation(md);
945 Set<FlowNode> nodeSet = getFlowGraph(md).getNodeSet();
946 for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
947 FlowNode flowNode = (FlowNode) iterator.next();
948 NTuple<Descriptor> descTuple = flowNode.getCurrentDescTuple();
949 NTuple<Location> locTuple = translateToLocTuple(md, descTuple);
950 for (int i = 0; i < locTuple.size(); i++) {
951 Location loc = locTuple.get(i);
952 if (loc.getDescriptor() instanceof ClassDescriptor) {
953 ClassDescriptor classDesc = (ClassDescriptor) loc.getDescriptor();
954 toanalyze_classDescSet.add(classDesc);
955 addSuperClasses(classDesc);
956 } else if (loc.getDescriptor() instanceof MethodDescriptor) {
957 toanalyze_classDescSet.add(((MethodDescriptor) loc.getDescriptor()).getClassDesc());
966 public Map<NTuple<Descriptor>, NTuple<Descriptor>> getMapCallerArgToCalleeParam(
967 MethodInvokeNode min) {
969 if (!mapMethodInvokeNodeToMapCallerArgToCalleeArg.containsKey(min)) {
970 mapMethodInvokeNodeToMapCallerArgToCalleeArg.put(min,
971 new HashMap<NTuple<Descriptor>, NTuple<Descriptor>>());
974 return mapMethodInvokeNodeToMapCallerArgToCalleeArg.get(min);
977 public void addMapCallerArgToCalleeParam(MethodInvokeNode min, NTuple<Descriptor> callerArg,
978 NTuple<Descriptor> calleeParam) {
979 getMapCallerArgToCalleeParam(min).put(callerArg, calleeParam);
982 private void assignCompositeLocation() {
983 calculateGlobalValueFlowCompositeLocation();
984 translateCompositeLocationAssignmentToFlowGraph();
987 private void translateCompositeLocationAssignmentToFlowGraph() {
988 System.out.println("\nSSJAVA: Translate composite location assignments to flow graphs:");
989 MethodDescriptor methodEventLoopDesc = ssjava.getMethodContainingSSJavaLoop();
990 translateCompositeLocationAssignmentToFlowGraph(methodEventLoopDesc);
993 private void translateCompositeLocationAssignmentToFlowGraph2() {
994 System.out.println("\nSSJAVA: Translate composite location assignments to flow graphs:");
995 MethodDescriptor methodEventLoopDesc = ssjava.getMethodContainingSSJavaLoop();
996 translateCompositeLocationAssignmentToFlowGraph(methodEventLoopDesc);
999 private void addAdditionalOrderingConstraints() {
1000 System.out.println("\nSSJAVA: Add addtional ordering constriants:");
1001 MethodDescriptor methodEventLoopDesc = ssjava.getMethodContainingSSJavaLoop();
1002 addAddtionalOrderingConstraints(methodEventLoopDesc);
1003 // calculateReturnHolderLocation();
1006 private void calculateReturnHolderLocation() {
1007 LinkedList<MethodDescriptor> methodDescList =
1008 (LinkedList<MethodDescriptor>) toanalyze_methodDescList.clone();
1010 while (!methodDescList.isEmpty()) {
1011 MethodDescriptor md = methodDescList.removeLast();
1013 FlowGraph fg = getFlowGraph(md);
1014 Set<FlowNode> nodeSet = fg.getNodeSet();
1015 for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
1016 FlowNode flowNode = (FlowNode) iterator.next();
1017 if (flowNode.isFromHolder()) {
1018 calculateCompositeLocationFromFlowGraph(md, flowNode);
1025 private void updateCompositeLocationAssignments() {
1027 LinkedList<MethodDescriptor> methodDescList =
1028 (LinkedList<MethodDescriptor>) toanalyze_methodDescList.clone();
1030 while (!methodDescList.isEmpty()) {
1031 MethodDescriptor md = methodDescList.removeLast();
1033 // System.out.println("\n#updateCompositeLocationAssignments=" + md);
1035 FlowGraph flowGraph = getFlowGraph(md);
1037 MethodSummary methodSummary = getMethodSummary(md);
1039 Set<FlowNode> nodeSet = flowGraph.getNodeSet();
1040 for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
1041 FlowNode node = (FlowNode) iterator.next();
1042 // System.out.println("-node=" + node + " node.getDescTuple=" + node.getDescTuple());
1043 if (node.getCompositeLocation() != null) {
1044 CompositeLocation compLoc = node.getCompositeLocation();
1045 CompositeLocation updatedCompLoc = updateCompositeLocation(compLoc);
1046 node.setCompositeLocation(updatedCompLoc);
1047 // System.out.println("---updatedCompLoc1=" + updatedCompLoc);
1049 NTuple<Descriptor> descTuple = node.getDescTuple();
1050 // System.out.println("update desc=" + descTuple);
1051 CompositeLocation compLoc = convertToCompositeLocation(md, descTuple);
1052 compLoc = updateCompositeLocation(compLoc);
1053 node.setCompositeLocation(compLoc);
1054 // System.out.println("---updatedCompLoc2=" + compLoc);
1057 if (node.isDeclaratonNode()) {
1058 Descriptor localVarDesc = node.getDescTuple().get(0);
1059 CompositeLocation compLoc = updateCompositeLocation(node.getCompositeLocation());
1060 methodSummary.addMapVarNameToInferCompLoc(localVarDesc, compLoc);
1064 // update PCLOC and RETURNLOC if they have a composite location assignment
1065 if (methodSummary.getRETURNLoc() != null) {
1066 methodSummary.setRETURNLoc(updateCompositeLocation(methodSummary.getRETURNLoc()));
1068 if (methodSummary.getPCLoc() != null) {
1069 methodSummary.setPCLoc(updateCompositeLocation(methodSummary.getPCLoc()));
1076 private CompositeLocation updateCompositeLocation(CompositeLocation compLoc) {
1077 CompositeLocation updatedCompLoc = new CompositeLocation();
1078 for (int i = 0; i < compLoc.getSize(); i++) {
1079 Location loc = compLoc.get(i);
1080 String nodeIdentifier = loc.getLocIdentifier();
1081 Descriptor enclosingDesc = loc.getDescriptor();
1083 if (!enclosingDesc.equals(GLOBALDESC)) {
1084 LocationSummary locSummary = getLocationSummary(enclosingDesc);
1085 // HierarchyGraph scGraph = getSkeletonCombinationHierarchyGraph(enclosingDesc);
1086 HierarchyGraph scGraph = getSimpleHierarchyGraph(enclosingDesc);
1087 if (scGraph != null) {
1088 HNode curNode = scGraph.getCurrentHNode(nodeIdentifier);
1089 // System.out.println("nodeID=" + nodeIdentifier + " curNode=" + curNode
1090 // + " enclosingDesc=" + enclosingDesc);
1091 if (curNode != null) {
1092 nodeIdentifier = curNode.getName();
1095 locName = locSummary.getLocationName(nodeIdentifier);
1097 locName = nodeIdentifier;
1099 Location updatedLoc = new Location(enclosingDesc, locName);
1100 updatedCompLoc.addLocation(updatedLoc);
1103 return updatedCompLoc;
1106 private void translateCompositeLocationAssignmentToFlowGraph(MethodDescriptor mdCaller) {
1108 // System.out.println("\n\n###translateCompositeLocationAssignmentToFlowGraph mdCaller="
1111 // First, assign a composite location to a node in the flow graph
1112 GlobalFlowGraph callerGlobalFlowGraph = getSubGlobalFlowGraph(mdCaller);
1114 FlowGraph callerFlowGraph = getFlowGraph(mdCaller);
1115 Map<Location, CompositeLocation> callerMapLocToCompLoc =
1116 callerGlobalFlowGraph.getMapLocationToInferCompositeLocation();
1118 Set<Location> methodLocSet = callerMapLocToCompLoc.keySet();
1119 for (Iterator iterator = methodLocSet.iterator(); iterator.hasNext();) {
1120 Location methodLoc = (Location) iterator.next();
1121 if (methodLoc.getDescriptor().equals(mdCaller)) {
1122 CompositeLocation inferCompLoc = callerMapLocToCompLoc.get(methodLoc);
1123 assignCompositeLocationToFlowGraph(callerFlowGraph, methodLoc, inferCompLoc);
1127 Set<MethodInvokeNode> minSet = mapMethodDescriptorToMethodInvokeNodeSet.get(mdCaller);
1129 Set<MethodDescriptor> calleeSet = new HashSet<MethodDescriptor>();
1130 for (Iterator iterator = minSet.iterator(); iterator.hasNext();) {
1131 MethodInvokeNode min = (MethodInvokeNode) iterator.next();
1132 // need to translate a composite location that is started with the base
1134 translateMapLocationToInferCompositeLocationToCalleeGraph(callerGlobalFlowGraph, min);
1135 MethodDescriptor mdCallee = min.getMethod();
1136 calleeSet.add(mdCallee);
1140 for (Iterator iterator = calleeSet.iterator(); iterator.hasNext();) {
1141 MethodDescriptor callee = (MethodDescriptor) iterator.next();
1142 translateCompositeLocationAssignmentToFlowGraph(callee);
1147 private void addAddtionalOrderingConstraints(MethodDescriptor mdCaller) {
1149 // First, assign a composite location to a node in the flow graph
1150 GlobalFlowGraph callerGlobalFlowGraph = getSubGlobalFlowGraph(mdCaller);
1152 FlowGraph callerFlowGraph = getFlowGraph(mdCaller);
1153 Map<Location, CompositeLocation> callerMapLocToCompLoc =
1154 callerGlobalFlowGraph.getMapLocationToInferCompositeLocation();
1155 Set<Location> methodLocSet = callerMapLocToCompLoc.keySet();
1157 Set<MethodInvokeNode> minSet = mapMethodDescriptorToMethodInvokeNodeSet.get(mdCaller);
1159 Set<MethodDescriptor> calleeSet = new HashSet<MethodDescriptor>();
1160 for (Iterator iterator = minSet.iterator(); iterator.hasNext();) {
1161 MethodInvokeNode min = (MethodInvokeNode) iterator.next();
1162 MethodDescriptor mdCallee = min.getMethod();
1163 calleeSet.add(mdCallee);
1166 // add an additional ordering constraint
1167 // if the first element of a parameter composite location matches 'this' reference,
1168 // the corresponding argument in the caller is required to be higher than the translated
1169 // parameter location in the caller lattice
1171 // addOrderingConstraintFromCompLocParamToArg(mdCaller, min);
1174 // update return flow nodes in the caller
1175 CompositeLocation returnLoc = getMethodSummary(mdCallee).getRETURNLoc();
1176 // System.out.println("### min=" + min.printNode(0) + " returnLoc=" + returnLoc);
1177 if (returnLoc != null && returnLoc.get(0).getLocDescriptor().equals(mdCallee.getThis())
1178 && returnLoc.getSize() > 1) {
1179 // System.out.println("###RETURN COMP LOC=" + returnLoc);
1180 NTuple<Location> returnLocTuple = returnLoc.getTuple();
1181 NTuple<Descriptor> baseTuple = mapMethodInvokeNodeToBaseTuple.get(min);
1182 // System.out.println("###basetuple=" + baseTuple);
1183 NTuple<Descriptor> newReturnTuple = baseTuple.clone();
1184 for (int i = 1; i < returnLocTuple.size(); i++) {
1185 newReturnTuple.add(returnLocTuple.get(i).getLocDescriptor());
1187 // System.out.println("###NEW RETURN TUPLE FOR CALLER=" + newReturnTuple);
1189 FlowReturnNode holderNode = callerFlowGraph.getFlowReturnNode(min);
1190 NodeTupleSet holderTupleSet =
1191 getNodeTupleSetFromReturnNode(getFlowGraph(mdCaller), holderNode);
1193 callerFlowGraph.getFlowReturnNode(min).setNewTuple(newReturnTuple);
1195 // then need to remove old constraints
1197 // System.out.println("###REMOVE OLD CONSTRAINTS=" + holderNode);
1198 for (Iterator<NTuple<Descriptor>> iter = holderTupleSet.iterator(); iter.hasNext();) {
1199 NTuple<Descriptor> tupleFromHolder = iter.next();
1200 Set<FlowEdge> holderOutEdge = callerFlowGraph.getOutEdgeSet(holderNode);
1201 for (Iterator iterator2 = holderOutEdge.iterator(); iterator2.hasNext();) {
1202 FlowEdge outEdge = (FlowEdge) iterator2.next();
1203 NTuple<Descriptor> toberemovedTuple = outEdge.getEndTuple();
1204 // System.out.println("---remove " + tupleFromHolder + " -> " + toberemovedTuple);
1205 callerFlowGraph.removeEdge(tupleFromHolder, toberemovedTuple);
1210 // if the return loc set was empty and later pcloc was connected to the return loc
1211 // need to make sure that return loc reflects to this changes.
1212 FlowReturnNode flowReturnNode = callerFlowGraph.getFlowReturnNode(min);
1213 if (flowReturnNode != null && flowReturnNode.getReturnTupleSet().isEmpty()) {
1215 if (needToUpdateReturnLocHolder(min.getMethod(), flowReturnNode)) {
1216 NTuple<Descriptor> baseTuple = mapMethodInvokeNodeToBaseTuple.get(min);
1217 NTuple<Descriptor> newReturnTuple = baseTuple.clone();
1218 flowReturnNode.addTuple(newReturnTuple);
1227 for (Iterator iterator = calleeSet.iterator(); iterator.hasNext();) {
1228 MethodDescriptor callee = (MethodDescriptor) iterator.next();
1229 addAddtionalOrderingConstraints(callee);
1234 private boolean needToUpdateReturnLocHolder(MethodDescriptor mdCallee,
1235 FlowReturnNode flowReturnNode) {
1236 FlowGraph fg = getFlowGraph(mdCallee);
1237 MethodSummary summary = getMethodSummary(mdCallee);
1238 CompositeLocation returnCompLoc = summary.getRETURNLoc();
1239 NTuple<Descriptor> returnDescTuple = translateToDescTuple(returnCompLoc.getTuple());
1240 Set<FlowNode> incomingNodeToReturnNode =
1241 fg.getIncomingFlowNodeSet(fg.getFlowNode(returnDescTuple));
1242 for (Iterator iterator = incomingNodeToReturnNode.iterator(); iterator.hasNext();) {
1243 FlowNode inNode = (FlowNode) iterator.next();
1244 if (inNode.getDescTuple().get(0).equals(mdCallee.getThis())) {
1251 private void addMapMethodDescToMethodInvokeNodeSet(MethodInvokeNode min) {
1252 MethodDescriptor md = min.getMethod();
1253 if (!mapMethodDescToMethodInvokeNodeSet.containsKey(md)) {
1254 mapMethodDescToMethodInvokeNodeSet.put(md, new HashSet<MethodInvokeNode>());
1256 mapMethodDescToMethodInvokeNodeSet.get(md).add(min);
1259 private Set<MethodInvokeNode> getMethodInvokeNodeSetByMethodDesc(MethodDescriptor md) {
1260 if (!mapMethodDescToMethodInvokeNodeSet.containsKey(md)) {
1261 mapMethodDescToMethodInvokeNodeSet.put(md, new HashSet<MethodInvokeNode>());
1263 return mapMethodDescToMethodInvokeNodeSet.get(md);
1266 public void assignCompositeLocationToFlowGraph(FlowGraph flowGraph, Location loc,
1267 CompositeLocation inferCompLoc) {
1268 Descriptor localDesc = loc.getLocDescriptor();
1270 Set<FlowNode> nodeSet = flowGraph.getNodeSet();
1271 for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
1272 FlowNode node = (FlowNode) iterator.next();
1273 if (node.getDescTuple().startsWith(localDesc)
1274 && !node.getDescTuple().get(0).equals(LITERALDESC)) {
1275 // need to assign the inferred composite location to this node
1276 CompositeLocation newCompLoc = generateCompositeLocation(node.getDescTuple(), inferCompLoc);
1277 node.setCompositeLocation(newCompLoc);
1278 // System.out.println("SET Node=" + node + " inferCompLoc=" + newCompLoc);
1283 private CompositeLocation generateCompositeLocation(NTuple<Descriptor> nodeDescTuple,
1284 CompositeLocation inferCompLoc) {
1286 // System.out.println("generateCompositeLocation=" + nodeDescTuple + " with inferCompLoc="
1289 MethodDescriptor md = (MethodDescriptor) inferCompLoc.get(0).getDescriptor();
1291 CompositeLocation newCompLoc = new CompositeLocation();
1292 for (int i = 0; i < inferCompLoc.getSize(); i++) {
1293 newCompLoc.addLocation(inferCompLoc.get(i));
1296 Descriptor lastDescOfPrefix = nodeDescTuple.get(0);
1297 Descriptor enclosingDescriptor;
1298 if (lastDescOfPrefix instanceof InterDescriptor) {
1299 enclosingDescriptor = getFlowGraph(md).getEnclosingDescriptor(lastDescOfPrefix);
1301 enclosingDescriptor = ((VarDescriptor) lastDescOfPrefix).getType().getClassDesc();
1304 for (int i = 1; i < nodeDescTuple.size(); i++) {
1305 Descriptor desc = nodeDescTuple.get(i);
1306 Location locElement = new Location(enclosingDescriptor, desc);
1307 newCompLoc.addLocation(locElement);
1309 enclosingDescriptor = ((FieldDescriptor) desc).getClassDescriptor();
1315 private void translateMapLocationToInferCompositeLocationToCalleeGraph(
1316 GlobalFlowGraph callerGraph, MethodInvokeNode min) {
1318 MethodDescriptor mdCallee = min.getMethod();
1319 MethodDescriptor mdCaller = callerGraph.getMethodDescriptor();
1320 Map<Location, CompositeLocation> callerMapLocToCompLoc =
1321 callerGraph.getMapLocationToInferCompositeLocation();
1323 Map<Integer, NTuple<Descriptor>> mapIdxToArgTuple = mapMethodInvokeNodeToArgIdxMap.get(min);
1325 FlowGraph calleeFlowGraph = getFlowGraph(mdCallee);
1326 GlobalFlowGraph calleeGlobalGraph = getSubGlobalFlowGraph(mdCallee);
1328 NTuple<Location> baseLocTuple = null;
1329 if (mapMethodInvokeNodeToBaseTuple.containsKey(min)) {
1330 baseLocTuple = translateToLocTuple(mdCaller, mapMethodInvokeNodeToBaseTuple.get(min));
1333 // System.out.println("\n-#translate caller=" + mdCaller + " infer composite loc to callee="
1334 // + mdCallee + " baseLocTuple=" + baseLocTuple);
1336 Set<Location> keySet = callerMapLocToCompLoc.keySet();
1337 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
1338 Location key = (Location) iterator.next();
1339 CompositeLocation callerCompLoc = callerMapLocToCompLoc.get(key);
1341 if (!key.getDescriptor().equals(mdCaller)) {
1343 CompositeLocation newCalleeCompLoc;
1344 if (baseLocTuple != null && callerCompLoc.getTuple().startsWith(baseLocTuple)) {
1345 // System.out.println("-----need to translate callerCompLoc=" + callerCompLoc
1346 // + " with baseTuple=" + baseLocTuple);
1348 translateCompositeLocationToCallee(callerCompLoc, baseLocTuple, mdCallee);
1350 calleeGlobalGraph.addMapLocationToInferCompositeLocation(key, newCalleeCompLoc);
1351 // System.out.println("1---key=" + key + " callerCompLoc=" + callerCompLoc
1352 // + " newCalleeCompLoc=" + newCalleeCompLoc);
1353 // System.out.println("-----caller=" + mdCaller + " callee=" + mdCallee);
1354 if (!newCalleeCompLoc.get(0).getDescriptor().equals(mdCallee)) {
1358 // System.out.println("-----baseLoctuple=" + baseLocTuple);
1360 // check if it is the global access
1361 Location compLocFirstElement = callerCompLoc.getTuple().get(0);
1362 if (compLocFirstElement.getDescriptor().equals(mdCallee)
1363 && compLocFirstElement.getLocDescriptor().equals(GLOBALDESC)) {
1365 newCalleeCompLoc = new CompositeLocation();
1366 Location newMethodLoc = new Location(mdCallee, GLOBALDESC);
1368 newCalleeCompLoc.addLocation(newMethodLoc);
1369 for (int i = 1; i < callerCompLoc.getSize(); i++) {
1370 newCalleeCompLoc.addLocation(callerCompLoc.get(i));
1372 calleeGlobalGraph.addMapLocationToInferCompositeLocation(key, newCalleeCompLoc);
1373 // System.out.println("2---key=" + key + " callerCompLoc=" + callerCompLoc
1374 // + " newCalleeCompLoc=" + newCalleeCompLoc);
1375 // System.out.println("-----caller=" + mdCaller + " callee=" + mdCallee);
1378 int paramIdx = getParamIdx(callerCompLoc, mapIdxToArgTuple);
1379 if (paramIdx == -1) {
1380 // here, the first element of the current composite location comes from the current
1382 // so transfer the same composite location to the callee
1383 if (!calleeGlobalGraph.contrainsInferCompositeLocationMapKey(key)) {
1384 if (callerCompLoc.get(0).getDescriptor().equals(mdCallee)) {
1385 // System.out.println("3---key=" + key + " callerCompLoc=" + callerCompLoc
1386 // + " newCalleeCompLoc=" + callerCompLoc);
1387 // System.out.println("-----caller=" + mdCaller + " callee=" + mdCallee);
1388 calleeGlobalGraph.addMapLocationToInferCompositeLocation(key, callerCompLoc);
1390 // System.out.println("3---SKIP key=" + key + " callerCompLoc=" + callerCompLoc);
1396 // It is the case where two parameters have relative orderings between them by having
1397 // composite locations
1398 // if we found the param idx, it means that the first part of the caller composite
1399 // location corresponds to the one of arguments.
1400 // for example, if the caller argument is <<caller.this>,<Decoder.br>>
1401 // and the current caller composite location mapping
1402 // <<caller.this>,<Decoder.br>,<Br.value>>
1403 // and the parameter which matches with the caller argument is 'Br brParam'
1404 // then, the translated callee composite location will be <<callee.brParam>,<Br.value>>
1405 NTuple<Descriptor> argTuple = mapIdxToArgTuple.get(paramIdx);
1407 FlowNode paramFlowNode = calleeFlowGraph.getParamFlowNode(paramIdx);
1408 NTuple<Location> paramLocTuple =
1409 translateToLocTuple(mdCallee, paramFlowNode.getDescTuple());
1410 newCalleeCompLoc = new CompositeLocation();
1411 for (int i = 0; i < paramLocTuple.size(); i++) {
1412 newCalleeCompLoc.addLocation(paramLocTuple.get(i));
1414 for (int i = argTuple.size(); i < callerCompLoc.getSize(); i++) {
1415 newCalleeCompLoc.addLocation(callerCompLoc.get(i));
1417 calleeGlobalGraph.addMapLocationToInferCompositeLocation(key, newCalleeCompLoc);
1418 // System.out.println("4---key=" + key + " callerCompLoc=" + callerCompLoc
1419 // + " newCalleeCompLoc=" + newCalleeCompLoc);
1420 // System.out.println("-----caller=" + mdCaller + " callee=" + mdCallee);
1429 // System.out.println("#ASSIGN COMP LOC TO CALLEE PARAMS: callee=" + mdCallee + " caller="
1431 // If the location of an argument has a composite location
1432 // need to assign a proper composite location to the corresponding callee parameter
1433 Set<Integer> idxSet = mapIdxToArgTuple.keySet();
1434 for (Iterator iterator = idxSet.iterator(); iterator.hasNext();) {
1435 Integer idx = (Integer) iterator.next();
1437 if (idx == 0 && !min.getMethod().isStatic()) {
1441 NTuple<Descriptor> argTuple = mapIdxToArgTuple.get(idx);
1442 // System.out.println("-argTuple=" + argTuple + " idx=" + idx);
1443 if (argTuple.size() > 0) {
1444 // check if an arg tuple has been already assigned to a composite location
1445 NTuple<Location> argLocTuple = translateToLocTuple(mdCaller, argTuple);
1446 Location argLocalLoc = argLocTuple.get(0);
1448 // if (!isPrimitiveType(argTuple)) {
1449 if (callerMapLocToCompLoc.containsKey(argLocalLoc)) {
1451 CompositeLocation callerCompLoc = callerMapLocToCompLoc.get(argLocalLoc);
1452 for (int i = 1; i < argLocTuple.size(); i++) {
1453 callerCompLoc.addLocation(argLocTuple.get(i));
1456 // System.out.println("---callerCompLoc=" + callerCompLoc);
1458 // if (baseLocTuple != null && callerCompLoc.getTuple().startsWith(baseLocTuple)) {
1460 FlowNode calleeParamFlowNode = calleeFlowGraph.getParamFlowNode(idx);
1462 NTuple<Descriptor> calleeParamDescTuple = calleeParamFlowNode.getDescTuple();
1463 NTuple<Location> calleeParamLocTuple =
1464 translateToLocTuple(mdCallee, calleeParamDescTuple);
1466 int refParamIdx = getParamIdx(callerCompLoc, mapIdxToArgTuple);
1467 // System.out.println("-----paramIdx=" + refParamIdx);
1468 if (refParamIdx == 0 && !mdCallee.isStatic()) {
1470 // System.out.println("-------need to translate callerCompLoc=" + callerCompLoc
1471 // + " with baseTuple=" + baseLocTuple + " calleeParamLocTuple="
1472 // + calleeParamLocTuple);
1474 CompositeLocation newCalleeCompLoc =
1475 translateCompositeLocationToCallee(callerCompLoc, baseLocTuple, mdCallee);
1477 calleeGlobalGraph.addMapLocationToInferCompositeLocation(calleeParamLocTuple.get(0),
1480 // System.out.println("---------key=" + calleeParamLocTuple.get(0) + " callerCompLoc="
1481 // + callerCompLoc + " newCalleeCompLoc=" + newCalleeCompLoc);
1483 } else if (refParamIdx != -1) {
1484 // the first element of an argument composite location matches with one of paramtere
1485 // composite locations
1487 // System.out.println("-------param match case=");
1489 NTuple<Descriptor> argTupleRef = mapIdxToArgTuple.get(refParamIdx);
1490 FlowNode refParamFlowNode = calleeFlowGraph.getParamFlowNode(refParamIdx);
1491 NTuple<Location> refParamLocTuple =
1492 translateToLocTuple(mdCallee, refParamFlowNode.getDescTuple());
1494 // System.out.println("---------refParamLocTuple=" + refParamLocTuple
1495 // + " from argTupleRef=" + argTupleRef);
1497 CompositeLocation newCalleeCompLoc = new CompositeLocation();
1498 for (int i = 0; i < refParamLocTuple.size(); i++) {
1499 newCalleeCompLoc.addLocation(refParamLocTuple.get(i));
1501 for (int i = argTupleRef.size(); i < callerCompLoc.getSize(); i++) {
1502 newCalleeCompLoc.addLocation(callerCompLoc.get(i));
1505 calleeGlobalGraph.addMapLocationToInferCompositeLocation(calleeParamLocTuple.get(0),
1508 calleeParamFlowNode.setCompositeLocation(newCalleeCompLoc);
1509 // System.out.println("-----------key=" + calleeParamLocTuple.get(0) +
1510 // " callerCompLoc="
1511 // + callerCompLoc + " newCalleeCompLoc=" + newCalleeCompLoc);
1514 CompositeLocation newCalleeCompLoc =
1515 calculateCompositeLocationFromSubGlobalGraph(mdCallee, calleeParamFlowNode);
1516 if (newCalleeCompLoc != null) {
1517 calleeGlobalGraph.addMapLocationToInferCompositeLocation(calleeParamLocTuple.get(0),
1519 calleeParamFlowNode.setCompositeLocation(newCalleeCompLoc);
1523 // System.out.println("-----------------calleeParamFlowNode="
1524 // + calleeParamFlowNode.getCompositeLocation());
1535 private CompositeLocation calculateCompositeLocationFromSubGlobalGraph(MethodDescriptor md,
1536 FlowNode paramNode) {
1538 // System.out.println("#############################################################");
1539 // System.out.println("calculateCompositeLocationFromSubGlobalGraph=" + paramNode);
1541 GlobalFlowGraph subGlobalFlowGraph = getSubGlobalFlowGraph(md);
1542 NTuple<Location> paramLocTuple = translateToLocTuple(md, paramNode.getDescTuple());
1543 GlobalFlowNode paramGlobalNode = subGlobalFlowGraph.getFlowNode(paramLocTuple);
1545 List<NTuple<Location>> prefixList = calculatePrefixList(subGlobalFlowGraph, paramGlobalNode);
1547 Location prefixLoc = paramLocTuple.get(0);
1549 Set<GlobalFlowNode> reachableNodeSet =
1550 subGlobalFlowGraph.getReachableNodeSetByPrefix(paramGlobalNode.getLocTuple().get(0));
1551 // Set<GlobalFlowNode> reachNodeSet = globalFlowGraph.getReachableNodeSetFrom(node);
1553 // System.out.println("node=" + node + " prefixList=" + prefixList);
1555 for (int i = 0; i < prefixList.size(); i++) {
1556 NTuple<Location> curPrefix = prefixList.get(i);
1557 Set<NTuple<Location>> reachableCommonPrefixSet = new HashSet<NTuple<Location>>();
1559 for (Iterator iterator2 = reachableNodeSet.iterator(); iterator2.hasNext();) {
1560 GlobalFlowNode reachNode = (GlobalFlowNode) iterator2.next();
1561 if (reachNode.getLocTuple().startsWith(curPrefix)) {
1562 reachableCommonPrefixSet.add(reachNode.getLocTuple());
1565 // System.out.println("reachableCommonPrefixSet=" + reachableCommonPrefixSet);
1567 if (!reachableCommonPrefixSet.isEmpty()) {
1569 MethodDescriptor curPrefixFirstElementMethodDesc =
1570 (MethodDescriptor) curPrefix.get(0).getDescriptor();
1572 MethodDescriptor nodePrefixLocFirstElementMethodDesc =
1573 (MethodDescriptor) prefixLoc.getDescriptor();
1575 // System.out.println("curPrefixFirstElementMethodDesc=" +
1576 // curPrefixFirstElementMethodDesc);
1577 // System.out.println("nodePrefixLocFirstElementMethodDesc="
1578 // + nodePrefixLocFirstElementMethodDesc);
1580 if (curPrefixFirstElementMethodDesc.equals(nodePrefixLocFirstElementMethodDesc)
1581 || isTransitivelyCalledFrom(nodePrefixLocFirstElementMethodDesc,
1582 curPrefixFirstElementMethodDesc)) {
1585 // if (!node.getLocTuple().startsWith(curPrefix.get(0))) {
1587 Location curPrefixLocalLoc = curPrefix.get(0);
1588 if (subGlobalFlowGraph.mapLocationToInferCompositeLocation.containsKey(curPrefixLocalLoc)) {
1589 // in this case, the local variable of the current prefix has already got a composite
1591 // so we just ignore the current composite location.
1593 // System.out.println("HERE WE DO NOT ASSIGN A COMPOSITE LOCATION TO =" + node
1594 // + " DUE TO " + curPrefix);
1598 if (!needToGenerateCompositeLocation(paramGlobalNode, curPrefix)) {
1599 // System.out.println("NO NEED TO GENERATE COMP LOC to " + paramGlobalNode
1600 // + " with prefix=" + curPrefix);
1604 Location targetLocalLoc = paramGlobalNode.getLocTuple().get(0);
1605 CompositeLocation newCompLoc = generateCompositeLocation(curPrefix);
1606 // System.out.println("NEED TO ASSIGN COMP LOC TO " + paramGlobalNode + " with prefix="
1608 // System.out.println("-targetLocalLoc=" + targetLocalLoc + " - newCompLoc=" +
1611 // makes sure that a newly generated location appears in the hierarchy graph
1612 for (int compIdx = 0; compIdx < newCompLoc.getSize(); compIdx++) {
1613 Location curLoc = newCompLoc.get(compIdx);
1614 getHierarchyGraph(curLoc.getDescriptor()).getHNode(curLoc.getLocDescriptor());
1617 subGlobalFlowGraph.addMapLocationToInferCompositeLocation(targetLocalLoc, newCompLoc);
1629 private int getParamIdx(CompositeLocation compLoc,
1630 Map<Integer, NTuple<Descriptor>> mapIdxToArgTuple) {
1632 // if the composite location is started with the argument descriptor
1633 // return the argument's index. o.t. return -1
1635 Set<Integer> keySet = mapIdxToArgTuple.keySet();
1636 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
1637 Integer key = (Integer) iterator.next();
1638 NTuple<Descriptor> argTuple = mapIdxToArgTuple.get(key);
1639 if (argTuple.size() > 0 && translateToDescTuple(compLoc.getTuple()).startsWith(argTuple)) {
1640 // System.out.println("compLoc.getTuple=" + compLoc + " is started with " + argTuple);
1641 return key.intValue();
1648 private boolean isPrimitiveType(NTuple<Descriptor> argTuple) {
1650 Descriptor lastDesc = argTuple.get(argTuple.size() - 1);
1652 if (lastDesc instanceof FieldDescriptor) {
1653 return ((FieldDescriptor) lastDesc).getType().isPrimitive();
1654 } else if (lastDesc instanceof VarDescriptor) {
1655 return ((VarDescriptor) lastDesc).getType().isPrimitive();
1656 } else if (lastDesc instanceof InterDescriptor) {
1663 private CompositeLocation translateCompositeLocationToCallee(CompositeLocation callerCompLoc,
1664 NTuple<Location> baseLocTuple, MethodDescriptor mdCallee) {
1666 CompositeLocation newCalleeCompLoc = new CompositeLocation();
1668 Location calleeThisLoc = new Location(mdCallee, mdCallee.getThis());
1669 newCalleeCompLoc.addLocation(calleeThisLoc);
1671 // remove the base tuple from the caller
1672 // ex; In the method invoation foo.bar.methodA(), the callee will have the composite location
1673 // ,which is relative to the 'this' variable, <THIS,...>
1674 for (int i = baseLocTuple.size(); i < callerCompLoc.getSize(); i++) {
1675 newCalleeCompLoc.addLocation(callerCompLoc.get(i));
1678 return newCalleeCompLoc;
1682 private void calculateGlobalValueFlowCompositeLocation() {
1684 System.out.println("SSJAVA: Calculate composite locations in the global value flow graph");
1685 MethodDescriptor methodDescEventLoop = ssjava.getMethodContainingSSJavaLoop();
1686 GlobalFlowGraph globalFlowGraph = getSubGlobalFlowGraph(methodDescEventLoop);
1688 Set<Location> calculatedPrefixSet = new HashSet<Location>();
1690 Set<GlobalFlowNode> nodeSet = globalFlowGraph.getNodeSet();
1692 next: for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
1693 GlobalFlowNode node = (GlobalFlowNode) iterator.next();
1695 Location prefixLoc = node.getLocTuple().get(0);
1697 if (calculatedPrefixSet.contains(prefixLoc)) {
1698 // the prefix loc has been already assigned to a composite location
1702 calculatedPrefixSet.add(prefixLoc);
1704 // Set<GlobalFlowNode> incomingNodeSet = globalFlowGraph.getIncomingNodeSet(node);
1705 List<NTuple<Location>> prefixList = calculatePrefixList(globalFlowGraph, node);
1707 Set<GlobalFlowNode> reachableNodeSet =
1708 globalFlowGraph.getReachableNodeSetByPrefix(node.getLocTuple().get(0));
1709 // Set<GlobalFlowNode> reachNodeSet = globalFlowGraph.getReachableNodeSetFrom(node);
1711 // System.out.println("node=" + node + " prefixList=" + prefixList);
1712 // System.out.println("---prefixList=" + prefixList);
1714 nextprefix: for (int i = 0; i < prefixList.size(); i++) {
1715 NTuple<Location> curPrefix = prefixList.get(i);
1716 // System.out.println("---curPrefix=" + curPrefix);
1717 Set<NTuple<Location>> reachableCommonPrefixSet = new HashSet<NTuple<Location>>();
1719 for (Iterator iterator2 = reachableNodeSet.iterator(); iterator2.hasNext();) {
1720 GlobalFlowNode reachNode = (GlobalFlowNode) iterator2.next();
1721 if (reachNode.getLocTuple().startsWith(curPrefix)) {
1722 reachableCommonPrefixSet.add(reachNode.getLocTuple());
1725 // System.out.println("reachableCommonPrefixSet=" + reachableCommonPrefixSet);
1727 if (!reachableCommonPrefixSet.isEmpty()) {
1729 MethodDescriptor curPrefixFirstElementMethodDesc =
1730 (MethodDescriptor) curPrefix.get(0).getDescriptor();
1732 MethodDescriptor nodePrefixLocFirstElementMethodDesc =
1733 (MethodDescriptor) prefixLoc.getDescriptor();
1735 // System.out.println("curPrefixFirstElementMethodDesc=" +
1736 // curPrefixFirstElementMethodDesc);
1737 // System.out.println("nodePrefixLocFirstElementMethodDesc="
1738 // + nodePrefixLocFirstElementMethodDesc);
1740 if (curPrefixFirstElementMethodDesc.equals(nodePrefixLocFirstElementMethodDesc)
1741 || isTransitivelyCalledFrom(nodePrefixLocFirstElementMethodDesc,
1742 curPrefixFirstElementMethodDesc)) {
1745 // if (!node.getLocTuple().startsWith(curPrefix.get(0))) {
1747 Location curPrefixLocalLoc = curPrefix.get(0);
1748 if (globalFlowGraph.mapLocationToInferCompositeLocation.containsKey(curPrefixLocalLoc)) {
1749 // in this case, the local variable of the current prefix has already got a composite
1751 // so we just ignore the current composite location.
1753 // System.out.println("HERE WE DO NOT ASSIGN A COMPOSITE LOCATION TO =" + node
1754 // + " DUE TO " + curPrefix);
1759 if (!needToGenerateCompositeLocation(node, curPrefix)) {
1760 // System.out.println("NO NEED TO GENERATE COMP LOC to " + node + " with prefix="
1762 // System.out.println("prefixList=" + prefixList);
1763 // System.out.println("reachableNodeSet=" + reachableNodeSet);
1764 continue nextprefix;
1767 Location targetLocalLoc = node.getLocTuple().get(0);
1768 CompositeLocation newCompLoc = generateCompositeLocation(curPrefix);
1769 // System.out.println("NEED TO ASSIGN COMP LOC TO " + node + " with prefix=" +
1771 // System.out.println("-targetLocalLoc=" + targetLocalLoc + " - newCompLoc="
1773 globalFlowGraph.addMapLocationToInferCompositeLocation(targetLocalLoc, newCompLoc);
1788 private boolean checkFlowNodeReturnThisField(MethodDescriptor md) {
1790 MethodDescriptor methodDescEventLoop = ssjava.getMethodContainingSSJavaLoop();
1791 GlobalFlowGraph globalFlowGraph = getSubGlobalFlowGraph(methodDescEventLoop);
1793 FlowGraph flowGraph = getFlowGraph(md);
1795 ClassDescriptor enclosingDesc = getClassTypeDescriptor(md.getThis());
1796 if (enclosingDesc == null) {
1801 Set<FlowNode> returnNodeSet = flowGraph.getReturnNodeSet();
1802 Set<GlobalFlowNode> globalReturnNodeSet = new HashSet<GlobalFlowNode>();
1803 for (Iterator iterator = returnNodeSet.iterator(); iterator.hasNext();) {
1804 FlowNode flowNode = (FlowNode) iterator.next();
1805 NTuple<Location> locTuple = translateToLocTuple(md, flowNode.getDescTuple());
1806 GlobalFlowNode globalReturnNode = globalFlowGraph.getFlowNode(locTuple);
1807 globalReturnNodeSet.add(globalReturnNode);
1809 List<NTuple<Location>> prefixList = calculatePrefixList(globalFlowGraph, globalReturnNode);
1810 for (int i = 0; i < prefixList.size(); i++) {
1811 NTuple<Location> curPrefix = prefixList.get(i);
1812 ClassDescriptor cd =
1813 getClassTypeDescriptor(curPrefix.get(curPrefix.size() - 1).getLocDescriptor());
1814 if (cd != null && cd.equals(enclosingDesc)) {
1822 if (count == returnNodeSet.size()) {
1823 // in this case, all return nodes in the method returns values coming from a location that
1824 // starts with "this"
1826 // System.out.println("$$$SET RETURN LOC TRUE=" + md);
1827 mapMethodDescriptorToCompositeReturnCase.put(md, Boolean.TRUE);
1829 // NameDescriptor returnLocDesc = new NameDescriptor("RLOC" + (locSeed++));
1830 // NTuple<Descriptor> rDescTuple = new NTuple<Descriptor>();
1831 // rDescTuple.add(md.getThis());
1832 // rDescTuple.add(returnLocDesc);
1834 // for (Iterator iterator = returnNodeSet.iterator(); iterator.hasNext();) {
1835 // FlowNode rnode = (FlowNode) iterator.next();
1836 // flowGraph.addValueFlowEdge(rnode.getDescTuple(), rDescTuple);
1839 // getMethodSummary(md).setRETURNLoc(new CompositeLocation(translateToLocTuple(md,
1843 mapMethodDescriptorToCompositeReturnCase.put(md, Boolean.FALSE);
1846 return mapMethodDescriptorToCompositeReturnCase.get(md).booleanValue();
1850 private boolean needToGenerateCompositeLocation(GlobalFlowNode node, NTuple<Location> curPrefix) {
1851 // return true if there is a path between a node to which we want to give a composite location
1852 // and nodes which start with curPrefix
1854 // System.out.println("---needToGenerateCompositeLocation curPrefix=" + curPrefix);
1856 Location targetLocalLoc = node.getLocTuple().get(0);
1858 MethodDescriptor md = (MethodDescriptor) targetLocalLoc.getDescriptor();
1859 FlowGraph flowGraph = getFlowGraph(md);
1861 FlowNode flowNode = flowGraph.getFlowNode(node.getDescTuple());
1862 Set<FlowNode> reachableSet = flowGraph.getReachFlowNodeSetFrom(flowNode);
1864 Set<FlowNode> paramNodeSet = flowGraph.getParamFlowNodeSet();
1865 for (Iterator iterator = paramNodeSet.iterator(); iterator.hasNext();) {
1866 FlowNode paramFlowNode = (FlowNode) iterator.next();
1867 if (curPrefix.startsWith(translateToLocTuple(md, paramFlowNode.getDescTuple()))) {
1872 if (targetLocalLoc.getLocDescriptor() instanceof InterDescriptor) {
1873 Pair<MethodInvokeNode, Integer> pair =
1874 ((InterDescriptor) targetLocalLoc.getLocDescriptor()).getMethodArgIdxPair();
1877 // System.out.println("$$$TARGETLOCALLOC HOLDER=" + targetLocalLoc);
1879 MethodInvokeNode min = pair.getFirst();
1880 Integer paramIdx = pair.getSecond();
1881 MethodDescriptor mdCallee = min.getMethod();
1883 FlowNode paramNode = getFlowGraph(mdCallee).getParamFlowNode(paramIdx);
1884 if (checkNodeReachToReturnNode(mdCallee, paramNode)) {
1892 GlobalFlowGraph subGlobalFlowGraph = getSubGlobalFlowGraph(md);
1893 Set<GlobalFlowNode> subGlobalReachableSet = subGlobalFlowGraph.getReachableNodeSetFrom(node);
1895 if (!md.isStatic()) {
1896 ClassDescriptor currentMethodThisType = getClassTypeDescriptor(md.getThis());
1897 for (int i = 0; i < curPrefix.size(); i++) {
1898 ClassDescriptor prefixType = getClassTypeDescriptor(curPrefix.get(i).getLocDescriptor());
1899 if (prefixType != null && prefixType.equals(currentMethodThisType)) {
1900 // System.out.println("PREFIX TYPE MATCHES WITH=" + currentMethodThisType);
1902 if (mapMethodDescriptorToCompositeReturnCase.containsKey(md)) {
1903 boolean hasCompReturnLocWithThis =
1904 mapMethodDescriptorToCompositeReturnCase.get(md).booleanValue();
1905 if (hasCompReturnLocWithThis) {
1906 if (checkNodeReachToReturnNode(md, flowNode)) {
1912 for (Iterator iterator3 = subGlobalReachableSet.iterator(); iterator3.hasNext();) {
1913 GlobalFlowNode subGlobalReachalbeNode = (GlobalFlowNode) iterator3.next();
1914 if (subGlobalReachalbeNode.getLocTuple().get(0).getLocDescriptor().equals(md.getThis())) {
1915 // System.out.println("PREFIX FOUND=" + subGlobalReachalbeNode);
1923 Location lastLocationOfPrefix = curPrefix.get(curPrefix.size() - 1);
1924 // check whether prefix appears in the list of parameters
1925 Set<MethodInvokeNode> minSet = mapMethodDescToMethodInvokeNodeSet.get(md);
1926 // System.out.println("$$$md=" + md + " minSet=" + minSet);
1927 if (minSet == null) {
1930 found: for (Iterator iterator = minSet.iterator(); iterator.hasNext();) {
1931 MethodInvokeNode min = (MethodInvokeNode) iterator.next();
1932 Map<Integer, NTuple<Descriptor>> map = mapMethodInvokeNodeToArgIdxMap.get(min);
1933 Set<Integer> keySet = map.keySet();
1934 // System.out.println("min=" + min.printNode(0));
1936 for (Iterator iterator2 = keySet.iterator(); iterator2.hasNext();) {
1937 Integer argIdx = (Integer) iterator2.next();
1938 NTuple<Descriptor> argTuple = map.get(argIdx);
1940 if (!(!md.isStatic() && argIdx == 0)) {
1941 // if the argTuple is empty, we don't need to do with anything(LITERAL CASE).
1942 if (argTuple.size() > 0
1943 && argTuple.get(argTuple.size() - 1).equals(lastLocationOfPrefix.getLocDescriptor())) {
1944 NTuple<Location> locTuple =
1945 translateToLocTuple(md, flowGraph.getParamFlowNode(argIdx).getDescTuple());
1946 lastLocationOfPrefix = locTuple.get(0);
1947 // System.out.println("ARG CASE=" + locTuple);
1948 for (Iterator iterator3 = subGlobalReachableSet.iterator(); iterator3.hasNext();) {
1949 GlobalFlowNode subGlobalReachalbeNode = (GlobalFlowNode) iterator3.next();
1950 // NTuple<Location> locTuple = translateToLocTuple(md, reachalbeNode.getDescTuple());
1951 NTuple<Location> globalReachlocTuple = subGlobalReachalbeNode.getLocTuple();
1952 for (int i = 0; i < globalReachlocTuple.size(); i++) {
1953 if (globalReachlocTuple.get(i).equals(lastLocationOfPrefix)) {
1954 // System.out.println("ARG " + argTuple + " IS MATCHED WITH="
1955 // + lastLocationOfPrefix);
1968 private boolean checkNodeReachToReturnNode(MethodDescriptor md, FlowNode node) {
1970 FlowGraph flowGraph = getFlowGraph(md);
1971 Set<FlowNode> reachableSet = flowGraph.getReachFlowNodeSetFrom(node);
1972 if (mapMethodDescriptorToCompositeReturnCase.containsKey(md)) {
1973 boolean hasCompReturnLocWithThis =
1974 mapMethodDescriptorToCompositeReturnCase.get(md).booleanValue();
1976 if (hasCompReturnLocWithThis) {
1977 for (Iterator iterator = flowGraph.getReturnNodeSet().iterator(); iterator.hasNext();) {
1978 FlowNode returnFlowNode = (FlowNode) iterator.next();
1979 if (reachableSet.contains(returnFlowNode)) {
1988 private void assignCompositeLocation(CompositeLocation compLocPrefix, GlobalFlowNode node) {
1989 CompositeLocation newCompLoc = compLocPrefix.clone();
1990 NTuple<Location> locTuple = node.getLocTuple();
1991 for (int i = 1; i < locTuple.size(); i++) {
1992 newCompLoc.addLocation(locTuple.get(i));
1994 node.setInferCompositeLocation(newCompLoc);
1997 private List<NTuple<Location>> calculatePrefixList(GlobalFlowGraph graph, GlobalFlowNode node) {
1999 // System.out.println("\n##### calculatePrefixList node=" + node);
2001 Set<GlobalFlowNode> incomingNodeSetPrefix =
2002 graph.getIncomingNodeSetByPrefix(node.getLocTuple().get(0));
2003 // System.out.println("---incomingNodeSetPrefix=" + incomingNodeSetPrefix);
2005 Set<GlobalFlowNode> reachableNodeSetPrefix =
2006 graph.getReachableNodeSetByPrefix(node.getLocTuple().get(0));
2007 // System.out.println("---reachableNodeSetPrefix=" + reachableNodeSetPrefix);
2009 List<NTuple<Location>> prefixList = new ArrayList<NTuple<Location>>();
2011 for (Iterator iterator = incomingNodeSetPrefix.iterator(); iterator.hasNext();) {
2012 GlobalFlowNode inNode = (GlobalFlowNode) iterator.next();
2013 NTuple<Location> inNodeTuple = inNode.getLocTuple();
2015 if (inNodeTuple.get(0).getLocDescriptor() instanceof InterDescriptor
2016 || inNodeTuple.get(0).getLocDescriptor().equals(GLOBALDESC)) {
2020 for (int i = 1; i < inNodeTuple.size(); i++) {
2021 NTuple<Location> prefix = inNodeTuple.subList(0, i);
2022 if (!prefixList.contains(prefix)) {
2023 prefixList.add(prefix);
2028 Collections.sort(prefixList, new Comparator<NTuple<Location>>() {
2029 public int compare(NTuple<Location> arg0, NTuple<Location> arg1) {
2030 int s0 = arg0.size();
2031 int s1 = arg1.size();
2034 } else if (s0 == s1) {
2046 private CompositeLocation calculateCompositeLocationFromFlowGraph(MethodDescriptor md,
2049 // System.out.println("#############################################################");
2050 // System.out.println("calculateCompositeLocationFromFlowGraph=" + node);
2052 FlowGraph flowGraph = getFlowGraph(md);
2053 // NTuple<Location> paramLocTuple = translateToLocTuple(md, paramNode.getDescTuple());
2054 // GlobalFlowNode paramGlobalNode = subGlobalFlowGraph.getFlowNode(paramLocTuple);
2056 List<NTuple<Location>> prefixList = calculatePrefixListFlowGraph(flowGraph, node);
2058 // Set<GlobalFlowNode> reachableNodeSet =
2059 // subGlobalFlowGraph.getReachableNodeSetByPrefix(paramGlobalNode.getLocTuple().get(0));
2061 Set<FlowNode> reachableNodeSet =
2062 flowGraph.getReachableSetFrom(node.getDescTuple().subList(0, 1));
2064 // Set<GlobalFlowNode> reachNodeSet = globalFlowGraph.getReachableNodeSetFrom(node);
2066 // System.out.println("node=" + node + " prefixList=" + prefixList);
2068 for (int i = 0; i < prefixList.size(); i++) {
2069 NTuple<Location> curPrefix = prefixList.get(i);
2070 Set<NTuple<Location>> reachableCommonPrefixSet = new HashSet<NTuple<Location>>();
2072 for (Iterator iterator2 = reachableNodeSet.iterator(); iterator2.hasNext();) {
2073 FlowNode reachNode = (FlowNode) iterator2.next();
2074 NTuple<Location> reachLocTuple = translateToLocTuple(md, reachNode.getCurrentDescTuple());
2075 if (reachLocTuple.startsWith(curPrefix)) {
2076 reachableCommonPrefixSet.add(reachLocTuple);
2079 // System.out.println("reachableCommonPrefixSet=" + reachableCommonPrefixSet);
2081 if (!reachableCommonPrefixSet.isEmpty()) {
2083 MethodDescriptor curPrefixFirstElementMethodDesc =
2084 (MethodDescriptor) curPrefix.get(0).getDescriptor();
2086 Location curPrefixLocalLoc = curPrefix.get(0);
2088 Location targetLocalLoc = new Location(md, node.getDescTuple().get(0));
2089 // Location targetLocalLoc = paramGlobalNode.getLocTuple().get(0);
2091 CompositeLocation newCompLoc = generateCompositeLocation(curPrefix);
2092 // System.out.println("NEED2ASSIGN COMP LOC TO " + node + " with prefix=" + curPrefix);
2093 // System.out.println("-targetLocalLoc=" + targetLocalLoc + " - newCompLoc=" +
2096 node.setCompositeLocation(newCompLoc);
2106 private List<NTuple<Location>> calculatePrefixListFlowGraph(FlowGraph graph, FlowNode node) {
2108 // System.out.println("\n##### calculatePrefixList node=" + node);
2110 MethodDescriptor md = graph.getMethodDescriptor();
2111 Set<FlowNode> incomingNodeSetPrefix =
2112 graph.getIncomingNodeSetByPrefix(node.getDescTuple().get(0));
2113 // System.out.println("---incomingNodeSetPrefix=" + incomingNodeSetPrefix);
2115 Set<FlowNode> reachableNodeSetPrefix =
2116 graph.getReachableSetFrom(node.getDescTuple().subList(0, 1));
2117 // System.out.println("---reachableNodeSetPrefix=" + reachableNodeSetPrefix);
2119 List<NTuple<Location>> prefixList = new ArrayList<NTuple<Location>>();
2121 for (Iterator iterator = incomingNodeSetPrefix.iterator(); iterator.hasNext();) {
2122 FlowNode inNode = (FlowNode) iterator.next();
2123 NTuple<Location> inNodeTuple = translateToLocTuple(md, inNode.getCurrentDescTuple());
2125 // if (inNodeTuple.get(0).getLocDescriptor() instanceof InterDescriptor
2126 // || inNodeTuple.get(0).getLocDescriptor().equals(GLOBALDESC)) {
2130 for (int i = 1; i < inNodeTuple.size(); i++) {
2131 NTuple<Location> prefix = inNodeTuple.subList(0, i);
2132 if (!prefixList.contains(prefix)) {
2133 prefixList.add(prefix);
2138 Collections.sort(prefixList, new Comparator<NTuple<Location>>() {
2139 public int compare(NTuple<Location> arg0, NTuple<Location> arg1) {
2140 int s0 = arg0.size();
2141 int s1 = arg1.size();
2144 } else if (s0 == s1) {
2156 private GlobalFlowGraph constructSubGlobalFlowGraph(FlowGraph flowGraph) {
2158 MethodDescriptor md = flowGraph.getMethodDescriptor();
2160 GlobalFlowGraph globalGraph = getSubGlobalFlowGraph(md);
2162 // Set<FlowNode> nodeSet = flowGraph.getNodeSet();
2163 Set<FlowEdge> edgeSet = flowGraph.getEdgeSet();
2165 for (Iterator iterator = edgeSet.iterator(); iterator.hasNext();) {
2167 FlowEdge edge = (FlowEdge) iterator.next();
2168 NTuple<Descriptor> srcDescTuple = edge.getInitTuple();
2169 NTuple<Descriptor> dstDescTuple = edge.getEndTuple();
2171 if (flowGraph.getFlowNode(srcDescTuple) instanceof FlowReturnNode
2172 || flowGraph.getFlowNode(dstDescTuple) instanceof FlowReturnNode) {
2176 // here only keep the first element(method location) of the descriptor
2178 NTuple<Location> srcLocTuple = translateToLocTuple(md, srcDescTuple);
2179 NTuple<Location> dstLocTuple = translateToLocTuple(md, dstDescTuple);
2181 globalGraph.addValueFlowEdge(srcLocTuple, dstLocTuple);
2188 private NTuple<Location> translateToLocTuple(MethodDescriptor md, NTuple<Descriptor> descTuple) {
2190 NTuple<Location> locTuple = new NTuple<Location>();
2192 Descriptor enclosingDesc = md;
2193 for (int i = 0; i < descTuple.size(); i++) {
2194 Descriptor desc = descTuple.get(i);
2196 Location loc = new Location(enclosingDesc, desc);
2199 if (desc instanceof VarDescriptor) {
2200 enclosingDesc = ((VarDescriptor) desc).getType().getClassDesc();
2201 } else if (desc instanceof FieldDescriptor) {
2202 enclosingDesc = ((FieldDescriptor) desc).getType().getClassDesc();
2204 enclosingDesc = desc;
2213 private void addValueFlowsFromCalleeSubGlobalFlowGraph(MethodDescriptor mdCaller) {
2215 // the transformation for a call site propagates flows through parameters
2216 // if the method is virtual, it also grab all relations from any possible
2219 Set<MethodInvokeNode> setMethodInvokeNode = getMethodInvokeNodeSet(mdCaller);
2221 for (Iterator iterator = setMethodInvokeNode.iterator(); iterator.hasNext();) {
2222 MethodInvokeNode min = (MethodInvokeNode) iterator.next();
2223 MethodDescriptor mdCallee = min.getMethod();
2224 Set<MethodDescriptor> setPossibleCallees = new HashSet<MethodDescriptor>();
2225 if (mdCallee.isStatic()) {
2226 setPossibleCallees.add(mdCallee);
2228 Set<MethodDescriptor> calleeSet = ssjava.getCallGraph().getMethods(mdCallee);
2229 // removes method descriptors that are not invoked by the caller
2230 calleeSet.retainAll(mapMethodToCalleeSet.get(mdCaller));
2231 setPossibleCallees.addAll(calleeSet);
2234 for (Iterator iterator2 = setPossibleCallees.iterator(); iterator2.hasNext();) {
2235 MethodDescriptor possibleMdCallee = (MethodDescriptor) iterator2.next();
2236 propagateValueFlowsToCallerFromSubGlobalFlowGraph(min, mdCaller, possibleMdCallee);
2243 private void propagateValueFlowsToCallerFromSubGlobalFlowGraph(MethodInvokeNode min,
2244 MethodDescriptor mdCaller, MethodDescriptor possibleMdCallee) {
2246 // System.out.println("---propagate from " + min.printNode(0) + " to caller=" + mdCaller);
2247 FlowGraph calleeFlowGraph = getFlowGraph(possibleMdCallee);
2248 Map<Integer, NTuple<Descriptor>> mapIdxToArg = mapMethodInvokeNodeToArgIdxMap.get(min);
2250 // System.out.println("-----mapMethodInvokeNodeToArgIdxMap.get(min)="
2251 // + mapMethodInvokeNodeToArgIdxMap.get(min));
2253 Set<Integer> keySet = mapIdxToArg.keySet();
2254 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2255 Integer idx = (Integer) iterator.next();
2256 NTuple<Descriptor> argDescTuple = mapIdxToArg.get(idx);
2257 if (argDescTuple.size() > 0) {
2258 NTuple<Location> argLocTuple = translateToLocTuple(mdCaller, argDescTuple);
2259 NTuple<Descriptor> paramDescTuple = calleeFlowGraph.getParamFlowNode(idx).getDescTuple();
2260 NTuple<Location> paramLocTuple = translateToLocTuple(possibleMdCallee, paramDescTuple);
2261 // System.out.println("-------paramDescTuple=" + paramDescTuple + "->argDescTuple="
2263 addMapCallerArgToCalleeParam(min, argDescTuple, paramDescTuple);
2267 // addValueFlowBetweenParametersToCaller(min, mdCaller, possibleMdCallee);
2269 NTuple<Descriptor> baseTuple = mapMethodInvokeNodeToBaseTuple.get(min);
2270 GlobalFlowGraph calleeSubGlobalGraph = getSubGlobalFlowGraph(possibleMdCallee);
2271 Set<GlobalFlowNode> calleeNodeSet = calleeSubGlobalGraph.getNodeSet();
2272 for (Iterator iterator = calleeNodeSet.iterator(); iterator.hasNext();) {
2273 GlobalFlowNode calleeNode = (GlobalFlowNode) iterator.next();
2274 addValueFlowFromCalleeNode(min, mdCaller, possibleMdCallee, calleeNode);
2277 // System.out.println("$$$GLOBAL PC LOC ADD=" + mdCaller);
2278 Set<NTuple<Location>> pcLocTupleSet = mapMethodInvokeNodeToPCLocTupleSet.get(min);
2279 // System.out.println("---pcLocTupleSet=" + pcLocTupleSet);
2280 GlobalFlowGraph callerSubGlobalGraph = getSubGlobalFlowGraph(mdCaller);
2281 for (Iterator iterator = calleeNodeSet.iterator(); iterator.hasNext();) {
2282 GlobalFlowNode calleeNode = (GlobalFlowNode) iterator.next();
2283 if (calleeNode.isParamNodeWithIncomingFlows()) {
2284 // System.out.println("calleeNode.getLocTuple()" + calleeNode.getLocTuple());
2285 NTuple<Location> callerSrcNodeLocTuple =
2286 translateToCallerLocTuple(min, possibleMdCallee, mdCaller, calleeNode.getLocTuple());
2287 // System.out.println("---callerSrcNodeLocTuple=" + callerSrcNodeLocTuple);
2288 if (callerSrcNodeLocTuple != null && callerSrcNodeLocTuple.size() > 0) {
2289 for (Iterator iterator2 = pcLocTupleSet.iterator(); iterator2.hasNext();) {
2290 NTuple<Location> pcLocTuple = (NTuple<Location>) iterator2.next();
2292 callerSubGlobalGraph.addValueFlowEdge(pcLocTuple, callerSrcNodeLocTuple);
2301 private void addValueFlowFromCalleeNode(MethodInvokeNode min, MethodDescriptor mdCaller,
2302 MethodDescriptor mdCallee, GlobalFlowNode calleeSrcNode) {
2304 GlobalFlowGraph calleeSubGlobalGraph = getSubGlobalFlowGraph(mdCallee);
2305 GlobalFlowGraph callerSubGlobalGraph = getSubGlobalFlowGraph(mdCaller);
2307 // System.out.println("$addValueFlowFromCalleeNode calleeSrcNode=" + calleeSrcNode);
2309 NTuple<Location> callerSrcNodeLocTuple =
2310 translateToCallerLocTuple(min, mdCallee, mdCaller, calleeSrcNode.getLocTuple());
2311 // System.out.println("---callerSrcNodeLocTuple=" + callerSrcNodeLocTuple);
2313 if (callerSrcNodeLocTuple != null && callerSrcNodeLocTuple.size() > 0) {
2315 Set<GlobalFlowNode> outNodeSet = calleeSubGlobalGraph.getOutNodeSet(calleeSrcNode);
2317 for (Iterator iterator = outNodeSet.iterator(); iterator.hasNext();) {
2318 GlobalFlowNode outNode = (GlobalFlowNode) iterator.next();
2319 NTuple<Location> callerDstNodeLocTuple =
2320 translateToCallerLocTuple(min, mdCallee, mdCaller, outNode.getLocTuple());
2321 // System.out.println("outNode=" + outNode + " callerDstNodeLocTuple="
2322 // + callerDstNodeLocTuple);
2323 if (callerSrcNodeLocTuple != null && callerDstNodeLocTuple != null
2324 && callerSrcNodeLocTuple.size() > 0 && callerDstNodeLocTuple.size() > 0) {
2325 callerSubGlobalGraph.addValueFlowEdge(callerSrcNodeLocTuple, callerDstNodeLocTuple);
2332 private NTuple<Location> translateToCallerLocTuple(MethodInvokeNode min,
2333 MethodDescriptor mdCallee, MethodDescriptor mdCaller, NTuple<Location> nodeLocTuple) {
2334 // this method will return the same nodeLocTuple if the corresponding argument is literal
2337 // System.out.println("translateToCallerLocTuple=" + nodeLocTuple);
2339 FlowGraph calleeFlowGraph = getFlowGraph(mdCallee);
2340 NTuple<Descriptor> nodeDescTuple = translateToDescTuple(nodeLocTuple);
2341 if (calleeFlowGraph.isParameter(nodeDescTuple)) {
2342 int paramIdx = calleeFlowGraph.getParamIdx(nodeDescTuple);
2343 NTuple<Descriptor> argDescTuple = mapMethodInvokeNodeToArgIdxMap.get(min).get(paramIdx);
2345 // if (isPrimitive(nodeLocTuple.get(0).getLocDescriptor())) {
2346 // // the type of argument is primitive.
2347 // return nodeLocTuple.clone();
2349 // System.out.println("paramIdx=" + paramIdx + " argDescTuple=" + argDescTuple + " from min="
2350 // + min.printNode(0));
2351 NTuple<Location> argLocTuple = translateToLocTuple(mdCaller, argDescTuple);
2353 NTuple<Location> callerLocTuple = new NTuple<Location>();
2355 callerLocTuple.addAll(argLocTuple);
2356 for (int i = 1; i < nodeLocTuple.size(); i++) {
2357 callerLocTuple.add(nodeLocTuple.get(i));
2359 return callerLocTuple;
2361 return nodeLocTuple.clone();
2366 public static boolean isPrimitive(Descriptor desc) {
2368 if (desc instanceof FieldDescriptor) {
2369 return ((FieldDescriptor) desc).getType().isPrimitive();
2370 } else if (desc instanceof VarDescriptor) {
2371 return ((VarDescriptor) desc).getType().isPrimitive();
2372 } else if (desc instanceof InterDescriptor) {
2379 public static boolean isReference(Descriptor desc) {
2381 if (desc instanceof FieldDescriptor) {
2383 TypeDescriptor type = ((FieldDescriptor) desc).getType();
2384 if (type.isArray()) {
2385 return !type.isPrimitive();
2387 return type.isPtr();
2390 } else if (desc instanceof VarDescriptor) {
2391 TypeDescriptor type = ((VarDescriptor) desc).getType();
2392 if (type.isArray()) {
2393 return !type.isPrimitive();
2395 return type.isPtr();
2402 private NTuple<Descriptor> translateToDescTuple(NTuple<Location> locTuple) {
2404 NTuple<Descriptor> descTuple = new NTuple<Descriptor>();
2405 for (int i = 0; i < locTuple.size(); i++) {
2406 descTuple.add(locTuple.get(i).getLocDescriptor());
2412 public LocationSummary getLocationSummary(Descriptor d) {
2413 if (!mapDescToLocationSummary.containsKey(d)) {
2414 if (d instanceof MethodDescriptor) {
2415 mapDescToLocationSummary.put(d, new MethodSummary((MethodDescriptor) d));
2416 } else if (d instanceof ClassDescriptor) {
2417 mapDescToLocationSummary.put(d, new FieldSummary());
2420 return mapDescToLocationSummary.get(d);
2423 private void generateMethodSummary() {
2425 Set<MethodDescriptor> keySet = md2lattice.keySet();
2426 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2427 MethodDescriptor md = (MethodDescriptor) iterator.next();
2429 System.out.println("\nSSJAVA: generate method summary: " + md);
2431 FlowGraph flowGraph = getFlowGraph(md);
2432 if (flowGraph == null) {
2435 MethodSummary methodSummary = getMethodSummary(md);
2437 HierarchyGraph scGraph = getSkeletonCombinationHierarchyGraph(md);
2439 // set the 'this' reference location
2440 if (!md.isStatic()) {
2441 // System.out.println("setThisLocName=" + scGraph.getHNode(md.getThis()).getName());
2442 methodSummary.setThisLocName(scGraph.getHNode(md.getThis()).getName());
2445 // set the 'global' reference location if needed
2446 if (methodSummary.hasGlobalAccess()) {
2447 methodSummary.setGlobalLocName(scGraph.getHNode(GLOBALDESC).getName());
2450 // construct a parameter mapping that maps a parameter descriptor to an
2451 // inferred composite location
2452 for (int paramIdx = 0; paramIdx < flowGraph.getNumParameters(); paramIdx++) {
2453 FlowNode flowNode = flowGraph.getParamFlowNode(paramIdx);
2454 CompositeLocation inferredCompLoc =
2455 updateCompositeLocation(flowNode.getCompositeLocation());
2456 // System.out.println("-paramIdx=" + paramIdx + " infer=" + inferredCompLoc + " original="
2457 // + flowNode.getCompositeLocation());
2459 Descriptor localVarDesc = flowNode.getDescTuple().get(0);
2460 methodSummary.addMapVarNameToInferCompLoc(localVarDesc, inferredCompLoc);
2461 methodSummary.addMapParamIdxToInferLoc(paramIdx, inferredCompLoc);
2468 private boolean hasOrderingRelation(NTuple<Location> locTuple1, NTuple<Location> locTuple2) {
2470 int size = locTuple1.size() >= locTuple2.size() ? locTuple2.size() : locTuple1.size();
2472 for (int idx = 0; idx < size; idx++) {
2473 Location loc1 = locTuple1.get(idx);
2474 Location loc2 = locTuple2.get(idx);
2476 Descriptor desc1 = loc1.getDescriptor();
2477 Descriptor desc2 = loc2.getDescriptor();
2479 if (!desc1.equals(desc2)) {
2480 throw new Error("Fail to compare " + locTuple1 + " and " + locTuple2);
2483 Descriptor locDesc1 = loc1.getLocDescriptor();
2484 Descriptor locDesc2 = loc2.getLocDescriptor();
2486 HierarchyGraph hierarchyGraph = getHierarchyGraph(desc1);
2488 HNode node1 = hierarchyGraph.getHNode(locDesc1);
2489 HNode node2 = hierarchyGraph.getHNode(locDesc2);
2491 // System.out.println("---node1=" + node1 + " node2=" + node2);
2492 // System.out.println("---hierarchyGraph.getIncomingNodeSet(node2)="
2493 // + hierarchyGraph.getIncomingNodeSet(node2));
2495 if (locDesc1.equals(locDesc2)) {
2497 } else if (!hierarchyGraph.getIncomingNodeSet(node2).contains(node1)
2498 && !hierarchyGraph.getIncomingNodeSet(node1).contains(node2)) {
2510 private boolean isHigherThan(NTuple<Location> locTuple1, NTuple<Location> locTuple2) {
2512 int size = locTuple1.size() >= locTuple2.size() ? locTuple2.size() : locTuple1.size();
2514 for (int idx = 0; idx < size; idx++) {
2515 Location loc1 = locTuple1.get(idx);
2516 Location loc2 = locTuple2.get(idx);
2518 Descriptor desc1 = loc1.getDescriptor();
2519 Descriptor desc2 = loc2.getDescriptor();
2521 if (!desc1.equals(desc2)) {
2522 throw new Error("Fail to compare " + locTuple1 + " and " + locTuple2);
2525 Descriptor locDesc1 = loc1.getLocDescriptor();
2526 Descriptor locDesc2 = loc2.getLocDescriptor();
2528 HierarchyGraph hierarchyGraph = getHierarchyGraph(desc1);
2530 HNode node1 = hierarchyGraph.getHNode(locDesc1);
2531 HNode node2 = hierarchyGraph.getHNode(locDesc2);
2533 // System.out.println("---node1=" + node1 + " node2=" + node2);
2534 // System.out.println("---hierarchyGraph.getIncomingNodeSet(node2)="
2535 // + hierarchyGraph.getIncomingNodeSet(node2));
2537 if (locDesc1.equals(locDesc2)) {
2539 } else if (hierarchyGraph.getIncomingNodeSet(node2).contains(node1)) {
2550 private void debug_writeLattices() {
2552 Set<Descriptor> keySet = mapDescriptorToSimpleLattice.keySet();
2553 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2554 Descriptor key = (Descriptor) iterator.next();
2555 SSJavaLattice<String> simpleLattice = mapDescriptorToSimpleLattice.get(key);
2556 // HierarchyGraph simpleHierarchyGraph = getSimpleHierarchyGraph(key);
2557 HierarchyGraph scHierarchyGraph = getSkeletonCombinationHierarchyGraph(key);
2558 if (key instanceof ClassDescriptor) {
2559 writeInferredLatticeDotFile((ClassDescriptor) key, simpleLattice, "_SIMPLE");
2560 } else if (key instanceof MethodDescriptor) {
2561 MethodDescriptor md = (MethodDescriptor) key;
2562 writeInferredLatticeDotFile(md.getClassDesc(), md, simpleLattice, "_SIMPLE");
2565 LocationSummary ls = getLocationSummary(key);
2566 // System.out.println("####LOC SUMMARY=" + key + "\n" + ls.getMapHNodeNameToLocationName());
2569 Set<ClassDescriptor> cdKeySet = cd2lattice.keySet();
2570 for (Iterator iterator = cdKeySet.iterator(); iterator.hasNext();) {
2571 ClassDescriptor cd = (ClassDescriptor) iterator.next();
2572 // System.out.println("########cd=" + cd);
2573 writeInferredLatticeDotFile((ClassDescriptor) cd, cd2lattice.get(cd), "");
2574 COUNT += cd2lattice.get(cd).getKeySet().size();
2577 Set<MethodDescriptor> mdKeySet = md2lattice.keySet();
2578 for (Iterator iterator = mdKeySet.iterator(); iterator.hasNext();) {
2579 MethodDescriptor md = (MethodDescriptor) iterator.next();
2580 writeInferredLatticeDotFile(md.getClassDesc(), md, md2lattice.get(md), "");
2581 COUNT += md2lattice.get(md).getKeySet().size();
2583 System.out.println("###COUNT=" + COUNT);
2585 Set<Descriptor> descKeySet = desc2naiveLattice.keySet();
2586 for (Iterator iterator = descKeySet.iterator(); iterator.hasNext();) {
2587 Descriptor desc = (Descriptor) iterator.next();
2588 // System.out.println("########cd=" + cd);
2590 ClassDescriptor cd_naive;
2591 MethodDescriptor md_naive;
2592 if (desc instanceof ClassDescriptor) {
2593 cd_naive = (ClassDescriptor) desc;
2596 md_naive = (MethodDescriptor) desc;
2597 cd_naive = md_naive.getClassDesc();
2600 writeInferredLatticeDotFile(cd_naive, md_naive, desc2naiveLattice.get(desc), "_naive");
2604 private void buildLattice(Descriptor desc) {
2605 // System.out.println("buildLattice=" + desc);
2606 SSJavaLattice<String> simpleLattice = buildLattice.buildLattice(desc);
2608 addMapDescToSimpleLattice(desc, simpleLattice);
2610 if (desc instanceof ClassDescriptor) {
2611 writeInferredLatticeDotFile((ClassDescriptor) desc, null, simpleLattice, "_SC");
2613 MethodDescriptor md = (MethodDescriptor) desc;
2614 writeInferredLatticeDotFile(md.getClassDesc(), md, simpleLattice, "_SC");
2617 HierarchyGraph simpleHierarchyGraph = getSimpleHierarchyGraph(desc);
2619 // System.out.println("\n## insertIntermediateNodesToStraightLine:"
2620 // + simpleHierarchyGraph.getName());
2621 SSJavaLattice<String> lattice =
2622 buildLattice.insertIntermediateNodesToStraightLine(desc, simpleLattice);
2624 if (lattice == null) {
2627 lattice.removeRedundantEdges();
2629 LocationInference.numLocationsSInfer += lattice.getKeySet().size();
2631 if (desc instanceof ClassDescriptor) {
2633 cd2lattice.put((ClassDescriptor) desc, lattice);
2634 // ssjava.writeLatticeDotFile((ClassDescriptor) desc, null, lattice);
2635 } else if (desc instanceof MethodDescriptor) {
2637 md2lattice.put((MethodDescriptor) desc, lattice);
2638 MethodDescriptor md = (MethodDescriptor) desc;
2639 ClassDescriptor cd = md.getClassDesc();
2640 // ssjava.writeLatticeDotFile(cd, md, lattice);
2645 // deprecated: it builds method/class lattices without considering class inheritance
2646 private void buildLattice() {
2648 Set<Descriptor> keySet = mapDescriptorToCombineSkeletonHierarchyGraph.keySet();
2649 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2650 Descriptor desc = (Descriptor) iterator.next();
2652 SSJavaLattice<String> simpleLattice = buildLattice.buildLattice(desc);
2654 addMapDescToSimpleLattice(desc, simpleLattice);
2656 HierarchyGraph simpleHierarchyGraph = getSimpleHierarchyGraph(desc);
2657 System.out.println("\n## insertIntermediateNodesToStraightLine:"
2658 + simpleHierarchyGraph.getName());
2659 SSJavaLattice<String> lattice =
2660 buildLattice.insertIntermediateNodesToStraightLine(desc, simpleLattice);
2661 lattice.removeRedundantEdges();
2663 LocationInference.numLocationsSInfer += lattice.getKeySet().size();
2665 if (desc instanceof ClassDescriptor) {
2667 cd2lattice.put((ClassDescriptor) desc, lattice);
2668 // ssjava.writeLatticeDotFile((ClassDescriptor) desc, null, lattice);
2669 } else if (desc instanceof MethodDescriptor) {
2671 md2lattice.put((MethodDescriptor) desc, lattice);
2672 MethodDescriptor md = (MethodDescriptor) desc;
2673 ClassDescriptor cd = md.getClassDesc();
2674 // ssjava.writeLatticeDotFile(cd, md, lattice);
2681 private void buildLatticeInheritanceTree() {
2682 // DFS the inheritance tree and propagates lattice nodes/edges from the parent to children
2683 // Node<ClassDescriptor> rootNode = inheritanceTree.getRootNode();
2684 DFSBuildLatticeInheritanceTree(rootClassDescriptor);
2687 public Set<ClassDescriptor> getDirectSubClasses(ClassDescriptor parent) {
2689 Set<ClassDescriptor> result = new HashSet<ClassDescriptor>();
2691 Set<ClassDescriptor> children = tu.getDirectSubClasses(parent);
2692 if (children == null) {
2693 children = new HashSet<ClassDescriptor>();
2696 for (Iterator iterator = children.iterator(); iterator.hasNext();) {
2697 ClassDescriptor child = (ClassDescriptor) iterator.next();
2698 if (toanalyze_classDescSet.contains(child)) {
2706 private void DFSBuildLatticeInheritanceTree(ClassDescriptor cd) {
2707 // ClassDescriptor cd = node.getData();
2709 ClassDescriptor parentClassDesc = cd.getSuperDesc();
2710 if (parentClassDesc != null) {
2711 Map<TripleItem, String> parentMap = buildLattice.getIntermediateLocMap(parentClassDesc);
2712 buildLattice.setIntermediateLocMap(cd, parentMap);
2717 for (Iterator iterator = cd.getMethods(); iterator.hasNext();) {
2718 MethodDescriptor md = (MethodDescriptor) iterator.next();
2719 if (toanalyze_methodDescList.contains(md)) {
2720 MethodDescriptor parentMethodDesc = getParentMethodDesc(md.getClassDesc(), md);
2721 if (parentMethodDesc != null) {
2722 Map<TripleItem, String> parentMap = buildLattice.getIntermediateLocMap(parentMethodDesc);
2723 Map<TripleItem, String> childMap = new HashMap<TripleItem, String>();
2724 Set<TripleItem> keySet = parentMap.keySet();
2725 for (Iterator iterator2 = keySet.iterator(); iterator2.hasNext();) {
2726 TripleItem key = (TripleItem) iterator2.next();
2727 childMap.put(key, parentMap.get(key));
2729 buildLattice.setIntermediateLocMap(md, childMap);
2735 // traverse children
2736 Set<ClassDescriptor> children = tu.getDirectSubClasses(cd);
2737 if (children != null) {
2738 for (Iterator iterator = children.iterator(); iterator.hasNext();) {
2739 ClassDescriptor classDescriptor = (ClassDescriptor) iterator.next();
2740 if (toanalyze_classDescSet.contains(classDescriptor)) {
2741 DFSBuildLatticeInheritanceTree(classDescriptor);
2743 if (classDescriptor.isAbstract()) {
2744 DFSBuildLatticeInheritanceTree(classDescriptor);
2752 public void addMapDescToSimpleLattice(Descriptor desc, SSJavaLattice<String> lattice) {
2753 mapDescriptorToSimpleLattice.put(desc, lattice);
2756 public SSJavaLattice<String> getSimpleLattice(Descriptor desc) {
2757 return mapDescriptorToSimpleLattice.get(desc);
2760 private void simplifyHierarchyGraph() {
2761 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
2762 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2763 Descriptor desc = (Descriptor) iterator.next();
2764 // System.out.println("SSJAVA: remove redundant edges: " + desc);
2765 HierarchyGraph simpleHierarchyGraph = getHierarchyGraph(desc).clone();
2766 simpleHierarchyGraph.setName(desc + "_SIMPLE");
2767 simpleHierarchyGraph.removeRedundantEdges();
2768 mapDescriptorToSimpleHierarchyGraph.put(desc, simpleHierarchyGraph);
2772 private void insertCombinationNodes() {
2773 Set<Descriptor> keySet = mapDescriptorToSkeletonHierarchyGraph.keySet();
2774 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2775 Descriptor desc = (Descriptor) iterator.next();
2776 System.out.println("\nSSJAVA: Inserting Combination Nodes:" + desc);
2777 HierarchyGraph skeletonGraph = getSkeletonHierarchyGraph(desc);
2778 HierarchyGraph skeletonGraphWithCombinationNode = skeletonGraph.clone();
2779 skeletonGraphWithCombinationNode.setName(desc + "_SC");
2781 HierarchyGraph simpleHierarchyGraph = getSimpleHierarchyGraph(desc);
2782 skeletonGraphWithCombinationNode.insertCombinationNodesToGraph(simpleHierarchyGraph);
2783 // skeletonGraphWithCombinationNode.insertCombinationNodesToGraph(simpleHierarchyGraph,
2785 // skeletonGraphWithCombinationNode.simplifySkeletonCombinationHierarchyGraph();
2786 skeletonGraphWithCombinationNode.removeRedundantEdges();
2787 mapDescriptorToCombineSkeletonHierarchyGraph.put(desc, skeletonGraphWithCombinationNode);
2791 private void constructSkeletonHierarchyGraph() {
2792 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
2793 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2794 Descriptor desc = (Descriptor) iterator.next();
2795 System.out.println("SSJAVA: Constructing Skeleton Hierarchy Graph: " + desc);
2796 HierarchyGraph simpleGraph = getSimpleHierarchyGraph(desc);
2797 HierarchyGraph skeletonGraph = simpleGraph.generateSkeletonGraph();
2798 skeletonGraph.setMapDescToHNode(simpleGraph.getMapDescToHNode());
2799 skeletonGraph.setMapHNodeToDescSet(simpleGraph.getMapHNodeToDescSet());
2800 skeletonGraph.simplifyHierarchyGraph(this);
2801 mapDescriptorToSkeletonHierarchyGraph.put(desc, skeletonGraph);
2805 private void recurUpAccumulateInheritanceDesc(Descriptor curDesc, Set<Descriptor> set) {
2807 if (curDesc instanceof ClassDescriptor) {
2808 ClassDescriptor cd = (ClassDescriptor) curDesc;
2809 ClassDescriptor parentClassDesc = cd.getSuperDesc();
2810 if (parentClassDesc != null && !parentClassDesc.equals(rootClassDescriptor)) {
2811 set.add(parentClassDesc);
2812 recurUpAccumulateInheritanceDesc(parentClassDesc, set);
2815 MethodDescriptor md = (MethodDescriptor) curDesc;
2816 ClassDescriptor cd = md.getClassDesc();
2819 ClassDescriptor parentClassDesc = cd.getSuperDesc();
2820 if (parentClassDesc != null && !parentClassDesc.equals(rootClassDescriptor)) {
2822 Set<MethodDescriptor> methodDescSet =
2823 parentClassDesc.getMethodTable().getSet(md.getSymbol());
2824 for (Iterator iterator = methodDescSet.iterator(); iterator.hasNext();) {
2825 MethodDescriptor parentMethodDesc = (MethodDescriptor) iterator.next();
2826 if (parentMethodDesc.matches(md)) {
2827 set.add(parentMethodDesc);
2828 recurUpAccumulateInheritanceDesc(parentMethodDesc, set);
2837 private void recurDownAccumulateInheritanceDesc(Descriptor curDesc, Set<Descriptor> set) {
2839 if (curDesc instanceof ClassDescriptor) {
2840 ClassDescriptor cd = (ClassDescriptor) curDesc;
2841 ClassDescriptor parentClassDesc = cd.getSuperDesc();
2842 Set<ClassDescriptor> directSubClasses = tu.getDirectSubClasses(cd);
2843 for (Iterator iterator = directSubClasses.iterator(); iterator.hasNext();) {
2844 ClassDescriptor child = (ClassDescriptor) iterator.next();
2845 recurDownAccumulateInheritanceDesc(child, set);
2848 MethodDescriptor md = (MethodDescriptor) curDesc;
2849 ClassDescriptor cd = md.getClassDesc();
2852 Set<ClassDescriptor> directSubClasses = tu.getDirectSubClasses(cd);
2853 for (Iterator iterator = directSubClasses.iterator(); iterator.hasNext();) {
2854 ClassDescriptor child = (ClassDescriptor) iterator.next();
2856 Set<MethodDescriptor> methodDescSet = child.getMethodTable().getSet(md.getSymbol());
2857 for (Iterator iterator2 = methodDescSet.iterator(); iterator2.hasNext();) {
2858 MethodDescriptor childMethodDesc = (MethodDescriptor) iterator2.next();
2859 if (childMethodDesc.matches(md)) {
2860 set.add(childMethodDesc);
2861 recurDownAccumulateInheritanceDesc(childMethodDesc, set);
2870 private void accumulateInheritanceDesc(Descriptor curDesc, Set<Descriptor> set) {
2872 recurUpAccumulateInheritanceDesc(curDesc, set);
2873 recurDownAccumulateInheritanceDesc(curDesc, set);
2877 public boolean isValidMergeInheritanceCheck(Descriptor desc, Set<HNode> mergeSet) {
2879 // set up inheritance chain set...
2880 Set<Descriptor> inheritanceDescSet = new HashSet<Descriptor>();
2881 recurUpAccumulateInheritanceDesc(desc, inheritanceDescSet);
2883 nextgraph: for (Iterator iterator = inheritanceDescSet.iterator(); iterator.hasNext();) {
2884 Descriptor inheritDesc = (Descriptor) iterator.next();
2886 if (!desc.equals(inheritDesc)) {
2887 HierarchyGraph graph = getSkeletonCombinationHierarchyGraph(inheritDesc);
2889 // first check whether this graph includes all elements of the merge set
2890 for (Iterator iterator2 = mergeSet.iterator(); iterator2.hasNext();) {
2891 HNode node = (HNode) iterator2.next();
2892 if (!graph.contains(node)) {
2897 HNode firstNode = mergeSet.iterator().next();
2899 Set<HNode> incomingNode = graph.getIncomingNodeSet(firstNode);
2900 Set<HNode> outgoingNode = graph.getOutgoingNodeSet(firstNode);
2902 for (Iterator iterator2 = mergeSet.iterator(); iterator2.hasNext();) {
2903 HNode node = (HNode) iterator2.next();
2905 if (!graph.getIncomingNodeSet(node).equals(incomingNode)
2906 || !graph.getOutgoingNodeSet(node).equals(outgoingNode)) {
2918 private void debug_writeHierarchyDotFiles() {
2920 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
2921 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2922 Descriptor desc = (Descriptor) iterator.next();
2923 getHierarchyGraph(desc).writeGraph();
2928 private void debug_writeSimpleHierarchyDotFiles() {
2930 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
2931 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2932 Descriptor desc = (Descriptor) iterator.next();
2933 getHierarchyGraph(desc).writeGraph();
2934 getSimpleHierarchyGraph(desc).writeGraph();
2935 getSimpleHierarchyGraph(desc).writeGraph(true);
2940 private void debug_writeSkeletonHierarchyDotFiles() {
2942 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
2943 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2944 Descriptor desc = (Descriptor) iterator.next();
2945 getSkeletonHierarchyGraph(desc).writeGraph();
2950 private void debug_writeSkeletonCombinationHierarchyDotFiles() {
2952 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
2953 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
2954 Descriptor desc = (Descriptor) iterator.next();
2955 getSkeletonCombinationHierarchyGraph(desc).writeGraph();
2960 public HierarchyGraph getSimpleHierarchyGraph(Descriptor d) {
2961 return mapDescriptorToSimpleHierarchyGraph.get(d);
2964 private HierarchyGraph getSkeletonHierarchyGraph(Descriptor d) {
2965 if (!mapDescriptorToSkeletonHierarchyGraph.containsKey(d)) {
2966 mapDescriptorToSkeletonHierarchyGraph.put(d, new HierarchyGraph(d));
2968 return mapDescriptorToSkeletonHierarchyGraph.get(d);
2971 public HierarchyGraph getSkeletonCombinationHierarchyGraph(Descriptor d) {
2972 if (!mapDescriptorToCombineSkeletonHierarchyGraph.containsKey(d)) {
2973 mapDescriptorToCombineSkeletonHierarchyGraph.put(d, new HierarchyGraph(d));
2975 return mapDescriptorToCombineSkeletonHierarchyGraph.get(d);
2978 private void constructHierarchyGraph() {
2980 LinkedList<MethodDescriptor> methodDescList =
2981 (LinkedList<MethodDescriptor>) toanalyze_methodDescList.clone();
2983 while (!methodDescList.isEmpty()) {
2984 MethodDescriptor md = methodDescList.removeLast();
2985 if (state.SSJAVADEBUG) {
2986 HierarchyGraph hierarchyGraph = new HierarchyGraph(md);
2987 System.out.println();
2988 System.out.println("SSJAVA: Construcing the hierarchy graph from " + md);
2989 constructHierarchyGraph(md, hierarchyGraph);
2990 mapDescriptorToHierarchyGraph.put(md, hierarchyGraph);
2996 while (!toAnalyzeIsEmpty()) {
2997 ClassDescriptor cd = toAnalyzeNext();
2998 HierarchyGraph graph = getHierarchyGraph(cd);
2999 for (Iterator iter = cd.getFields(); iter.hasNext();) {
3000 FieldDescriptor fieldDesc = (FieldDescriptor) iter.next();
3001 if (!(fieldDesc.isStatic() && fieldDesc.isFinal())) {
3002 graph.getHNode(fieldDesc);
3007 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
3008 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
3009 Descriptor key = (Descriptor) iterator.next();
3010 HierarchyGraph graph = getHierarchyGraph(key);
3012 Set<HNode> nodeToBeConnected = new HashSet<HNode>();
3013 for (Iterator iterator2 = graph.getNodeSet().iterator(); iterator2.hasNext();) {
3014 HNode node = (HNode) iterator2.next();
3015 if (!node.isSkeleton() && !node.isCombinationNode()) {
3016 if (graph.getIncomingNodeSet(node).size() == 0) {
3017 nodeToBeConnected.add(node);
3022 for (Iterator iterator2 = nodeToBeConnected.iterator(); iterator2.hasNext();) {
3023 HNode node = (HNode) iterator2.next();
3024 // System.out.println("NEED TO BE CONNECTED TO TOP=" + node);
3025 graph.addEdge(graph.getHNode(TOPDESC), node);
3032 private void constructHierarchyGraph2() {
3034 // do fixed-point analysis
3036 LinkedList<MethodDescriptor> descriptorListToAnalyze = ssjava.getSortedDescriptors();
3038 // Collections.sort(descriptorListToAnalyze, new
3039 // Comparator<MethodDescriptor>() {
3040 // public int compare(MethodDescriptor o1, MethodDescriptor o2) {
3041 // return o1.getSymbol().compareToIgnoreCase(o2.getSymbol());
3045 // current descriptors to visit in fixed-point interprocedural analysis,
3046 // prioritized by dependency in the call graph
3047 methodDescriptorsToVisitStack.clear();
3049 Set<MethodDescriptor> methodDescriptorToVistSet = new HashSet<MethodDescriptor>();
3050 methodDescriptorToVistSet.addAll(descriptorListToAnalyze);
3052 while (!descriptorListToAnalyze.isEmpty()) {
3053 MethodDescriptor md = descriptorListToAnalyze.removeFirst();
3054 methodDescriptorsToVisitStack.add(md);
3057 // analyze scheduled methods until there are no more to visit
3058 while (!methodDescriptorsToVisitStack.isEmpty()) {
3059 // start to analyze leaf node
3060 MethodDescriptor md = methodDescriptorsToVisitStack.pop();
3062 HierarchyGraph hierarchyGraph = new HierarchyGraph(md);
3063 // MethodSummary methodSummary = new MethodSummary(md);
3065 // MethodLocationInfo methodInfo = new MethodLocationInfo(md);
3066 // curMethodInfo = methodInfo;
3068 System.out.println();
3069 System.out.println("SSJAVA: Construcing the hierarchy graph from " + md);
3071 constructHierarchyGraph(md, hierarchyGraph);
3073 HierarchyGraph prevHierarchyGraph = getHierarchyGraph(md);
3074 // MethodSummary prevMethodSummary = getMethodSummary(md);
3076 if (!hierarchyGraph.equals(prevHierarchyGraph)) {
3078 mapDescriptorToHierarchyGraph.put(md, hierarchyGraph);
3079 // mapDescToLocationSummary.put(md, methodSummary);
3081 // results for callee changed, so enqueue dependents caller for
3083 Iterator<MethodDescriptor> depsItr = ssjava.getDependents(md).iterator();
3084 while (depsItr.hasNext()) {
3085 MethodDescriptor methodNext = depsItr.next();
3086 if (!methodDescriptorsToVisitStack.contains(methodNext)
3087 && methodDescriptorToVistSet.contains(methodNext)) {
3088 methodDescriptorsToVisitStack.add(methodNext);
3097 while (!toAnalyzeIsEmpty()) {
3098 ClassDescriptor cd = toAnalyzeNext();
3099 HierarchyGraph graph = getHierarchyGraph(cd);
3100 for (Iterator iter = cd.getFields(); iter.hasNext();) {
3101 FieldDescriptor fieldDesc = (FieldDescriptor) iter.next();
3102 if (!(fieldDesc.isStatic() && fieldDesc.isFinal())) {
3103 graph.getHNode(fieldDesc);
3108 Set<Descriptor> keySet = mapDescriptorToHierarchyGraph.keySet();
3109 for (Iterator iterator = keySet.iterator(); iterator.hasNext();) {
3110 Descriptor key = (Descriptor) iterator.next();
3111 HierarchyGraph graph = getHierarchyGraph(key);
3113 Set<HNode> nodeToBeConnected = new HashSet<HNode>();
3114 for (Iterator iterator2 = graph.getNodeSet().iterator(); iterator2.hasNext();) {
3115 HNode node = (HNode) iterator2.next();
3116 if (!node.isSkeleton() && !node.isCombinationNode()) {
3117 if (graph.getIncomingNodeSet(node).size() == 0) {
3118 nodeToBeConnected.add(node);
3123 for (Iterator iterator2 = nodeToBeConnected.iterator(); iterator2.hasNext();) {
3124 HNode node = (HNode) iterator2.next();
3125 // System.out.println("NEED TO BE CONNECTED TO TOP=" + node);
3126 graph.addEdge(graph.getHNode(TOPDESC), node);
3133 private HierarchyGraph getHierarchyGraph(Descriptor d) {
3134 if (!mapDescriptorToHierarchyGraph.containsKey(d)) {
3135 mapDescriptorToHierarchyGraph.put(d, new HierarchyGraph(d));
3137 return mapDescriptorToHierarchyGraph.get(d);
3140 private void constructHierarchyGraph(MethodDescriptor md, HierarchyGraph methodGraph) {
3142 // visit each node of method flow graph
3143 FlowGraph fg = getFlowGraph(md);
3144 // Set<FlowNode> nodeSet = fg.getNodeSet();
3146 Set<FlowEdge> edgeSet = fg.getEdgeSet();
3148 Set<Descriptor> paramDescSet = fg.getMapParamDescToIdx().keySet();
3149 for (Iterator iterator = paramDescSet.iterator(); iterator.hasNext();) {
3150 Descriptor desc = (Descriptor) iterator.next();
3151 methodGraph.getHNode(desc).setSkeleton(true);
3154 // for the method lattice, we need to look at the first element of
3155 // NTuple<Descriptor>
3156 boolean hasGlobalAccess = false;
3157 // for (Iterator iterator = nodeSet.iterator(); iterator.hasNext();) {
3158 // FlowNode originalSrcNode = (FlowNode) iterator.next();
3159 for (Iterator iterator = edgeSet.iterator(); iterator.hasNext();) {
3160 FlowEdge edge = (FlowEdge) iterator.next();
3162 FlowNode originalSrcNode = fg.getFlowNode(edge.getInitTuple());
3163 Set<FlowNode> sourceNodeSet = new HashSet<FlowNode>();
3164 if (originalSrcNode instanceof FlowReturnNode) {
3165 FlowReturnNode rnode = (FlowReturnNode) originalSrcNode;
3166 // System.out.println("rnode=" + rnode);
3167 Set<NTuple<Descriptor>> tupleSet = rnode.getReturnTupleSet();
3168 for (Iterator iterator2 = tupleSet.iterator(); iterator2.hasNext();) {
3169 NTuple<Descriptor> nTuple = (NTuple<Descriptor>) iterator2.next();
3170 sourceNodeSet.add(fg.getFlowNode(nTuple));
3171 // System.out.println("&&&SOURCE fg.getFlowNode(nTuple)=" + fg.getFlowNode(nTuple));
3174 sourceNodeSet.add(originalSrcNode);
3177 // System.out.println("---sourceNodeSet=" + sourceNodeSet + " from originalSrcNode="
3178 // + originalSrcNode);
3180 for (Iterator iterator3 = sourceNodeSet.iterator(); iterator3.hasNext();) {
3181 FlowNode srcNode = (FlowNode) iterator3.next();
3183 NTuple<Descriptor> srcNodeTuple = srcNode.getDescTuple();
3184 Descriptor srcLocalDesc = srcNodeTuple.get(0);
3186 if (srcLocalDesc instanceof InterDescriptor
3187 && ((InterDescriptor) srcLocalDesc).getMethodArgIdxPair() != null) {
3189 if (srcNode.getCompositeLocation() == null) {
3194 // if the srcNode is started with the global descriptor
3195 // need to set as a skeleton node
3196 if (!hasGlobalAccess && srcNode.getDescTuple().startsWith(GLOBALDESC)) {
3197 hasGlobalAccess = true;
3200 // Set<FlowEdge> outEdgeSet = fg.getOutEdgeSet(originalSrcNode);
3201 // for (Iterator iterator2 = outEdgeSet.iterator(); iterator2.hasNext();) {
3202 // FlowEdge outEdge = (FlowEdge) iterator2.next();
3203 // FlowNode originalDstNode = outEdge.getDst();
3204 FlowNode originalDstNode = fg.getFlowNode(edge.getEndTuple());
3206 Set<FlowNode> dstNodeSet = new HashSet<FlowNode>();
3207 if (originalDstNode instanceof FlowReturnNode) {
3208 FlowReturnNode rnode = (FlowReturnNode) originalDstNode;
3209 // System.out.println("\n-returnNode=" + rnode);
3210 Set<NTuple<Descriptor>> tupleSet = rnode.getReturnTupleSet();
3211 for (Iterator iterator4 = tupleSet.iterator(); iterator4.hasNext();) {
3212 NTuple<Descriptor> nTuple = (NTuple<Descriptor>) iterator4.next();
3213 dstNodeSet.add(fg.getFlowNode(nTuple));
3214 // System.out.println("&&&DST fg.getFlowNode(nTuple)=" + fg.getFlowNode(nTuple));
3217 dstNodeSet.add(originalDstNode);
3219 // System.out.println("---dstNodeSet=" + dstNodeSet);
3220 for (Iterator iterator4 = dstNodeSet.iterator(); iterator4.hasNext();) {
3221 FlowNode dstNode = (FlowNode) iterator4.next();
3223 NTuple<Descriptor> dstNodeTuple = dstNode.getDescTuple();
3224 Descriptor dstLocalDesc = dstNodeTuple.get(0);
3226 if (dstLocalDesc instanceof InterDescriptor
3227 && ((InterDescriptor) dstLocalDesc).getMethodArgIdxPair() != null) {
3228 if (dstNode.getCompositeLocation() == null) {
3229 // System.out.println("%%%%%%%%%%%%%SKIP=" + dstNode);
3234 // if (outEdge.getInitTuple().equals(srcNodeTuple)
3235 // && outEdge.getEndTuple().equals(dstNodeTuple)) {
3237 NTuple<Descriptor> srcCurTuple = srcNode.getCurrentDescTuple();
3238 NTuple<Descriptor> dstCurTuple = dstNode.getCurrentDescTuple();
3240 // //////////////////////////
3241 // inheritance check
3242 if (mapMethodDescToHighestOverriddenMethodDesc.containsKey(md)) {
3244 MethodDescriptor highestOverriddenMethodDesc =
3245 mapMethodDescToHighestOverriddenMethodDesc.get(md);
3247 if (srcCurTuple.get(srcCurTuple.size() - 1).getSymbol().startsWith(PCLOC)) {
3251 // //////////////////////////
3253 // System.out.println("-srcCurTuple=" + srcCurTuple + " dstCurTuple=" + dstCurTuple
3254 // + " srcNode=" + srcNode + " dstNode=" + dstNode);
3256 // srcCurTuple = translateBaseTuple(srcNode, srcCurTuple);
3257 // dstCurTuple = translateBaseTuple(dstNode, dstCurTuple);
3259 if ((srcCurTuple.size() > 1 && dstCurTuple.size() > 1)
3260 && srcCurTuple.get(0).equals(dstCurTuple.get(0))) {
3262 // value flows between fields
3263 Descriptor desc = srcCurTuple.get(0);
3264 ClassDescriptor classDesc;
3266 if (desc.equals(GLOBALDESC)) {
3267 classDesc = md.getClassDesc();
3269 VarDescriptor varDesc = (VarDescriptor) srcCurTuple.get(0);
3270 classDesc = varDesc.getType().getClassDesc();
3272 extractFlowsBetweenFields(classDesc, srcNode, dstNode, 1);
3274 } else if ((srcCurTuple.size() == 1 && dstCurTuple.size() == 1)
3275 || ((srcCurTuple.size() > 1 || dstCurTuple.size() > 1) && !srcCurTuple.get(0).equals(
3276 dstCurTuple.get(0)))) {
3278 // value flow between a primitive local var - a primitive local var or local var -
3281 Descriptor srcDesc = srcCurTuple.get(0);
3282 Descriptor dstDesc = dstCurTuple.get(0);
3284 methodGraph.addEdge(srcDesc, dstDesc);
3286 if (fg.isParamDesc(srcDesc)) {
3287 methodGraph.setParamHNode(srcDesc);
3289 if (fg.isParamDesc(dstDesc)) {
3290 methodGraph.setParamHNode(dstDesc);
3304 // If the method accesses static fields
3305 // set hasGloabalAccess true in the method summary.
3306 if (hasGlobalAccess) {
3307 getMethodSummary(md).setHasGlobalAccess();
3309 methodGraph.getHNode(GLOBALDESC).setSkeleton(true);
3311 if (ssjava.getMethodContainingSSJavaLoop().equals(md)) {
3312 // if the current method contains the event loop
3313 // we need to set all nodes of the hierarchy graph as a skeleton node
3314 Set<HNode> hnodeSet = methodGraph.getNodeSet();
3315 for (Iterator iterator = hnodeSet.iterator(); iterator.hasNext();) {
3316 HNode hnode = (HNode) iterator.next();
3317 hnode.setSkeleton(true);
3323 private NTuple<Descriptor> translateBaseTuple(FlowNode flowNode, NTuple<Descriptor> inTuple) {