passes the flow down rule
[IRC.git] / Robust / src / Analysis / SSJava / FlowDownCheck.java
1 package Analysis.SSJava;
2
3 import java.util.ArrayList;
4 import java.util.Collections;
5 import java.util.Comparator;
6 import java.util.HashSet;
7 import java.util.Hashtable;
8 import java.util.Iterator;
9 import java.util.List;
10 import java.util.Set;
11 import java.util.StringTokenizer;
12 import java.util.Vector;
13
14 import Analysis.SSJava.FlowDownCheck.ComparisonResult;
15 import Analysis.SSJava.FlowDownCheck.CompositeLattice;
16 import IR.AnnotationDescriptor;
17 import IR.ClassDescriptor;
18 import IR.Descriptor;
19 import IR.FieldDescriptor;
20 import IR.MethodDescriptor;
21 import IR.NameDescriptor;
22 import IR.Operation;
23 import IR.State;
24 import IR.SymbolTable;
25 import IR.TypeDescriptor;
26 import IR.TypeExtension;
27 import IR.VarDescriptor;
28 import IR.Tree.ArrayAccessNode;
29 import IR.Tree.AssignmentNode;
30 import IR.Tree.BlockExpressionNode;
31 import IR.Tree.BlockNode;
32 import IR.Tree.BlockStatementNode;
33 import IR.Tree.CastNode;
34 import IR.Tree.CreateObjectNode;
35 import IR.Tree.DeclarationNode;
36 import IR.Tree.ExpressionNode;
37 import IR.Tree.FieldAccessNode;
38 import IR.Tree.IfStatementNode;
39 import IR.Tree.Kind;
40 import IR.Tree.LiteralNode;
41 import IR.Tree.LoopNode;
42 import IR.Tree.MethodInvokeNode;
43 import IR.Tree.NameNode;
44 import IR.Tree.OpNode;
45 import IR.Tree.ReturnNode;
46 import IR.Tree.SubBlockNode;
47 import IR.Tree.SwitchBlockNode;
48 import IR.Tree.SwitchStatementNode;
49 import IR.Tree.SynchronizedNode;
50 import IR.Tree.TertiaryNode;
51 import IR.Tree.TreeNode;
52 import Util.Pair;
53
54 public class FlowDownCheck {
55
56   State state;
57   static SSJavaAnalysis ssjava;
58
59   Set<ClassDescriptor> toanalyze;
60   List<ClassDescriptor> toanalyzeList;
61
62   Set<MethodDescriptor> toanalyzeMethod;
63   List<MethodDescriptor> toanalyzeMethodList;
64
65   // mapping from 'descriptor' to 'composite location'
66   Hashtable<Descriptor, CompositeLocation> d2loc;
67
68   Hashtable<MethodDescriptor, CompositeLocation> md2ReturnLoc;
69   Hashtable<MethodDescriptor, ReturnLocGenerator> md2ReturnLocGen;
70
71   // mapping from 'locID' to 'class descriptor'
72   Hashtable<String, ClassDescriptor> fieldLocName2cd;
73
74   boolean deterministic = true;
75
76   public FlowDownCheck(SSJavaAnalysis ssjava, State state) {
77     this.ssjava = ssjava;
78     this.state = state;
79     if (deterministic) {
80       this.toanalyzeList = new ArrayList<ClassDescriptor>();
81     } else {
82       this.toanalyze = new HashSet<ClassDescriptor>();
83     }
84     if (deterministic) {
85       this.toanalyzeMethodList = new ArrayList<MethodDescriptor>();
86     } else {
87       this.toanalyzeMethod = new HashSet<MethodDescriptor>();
88     }
89     this.d2loc = new Hashtable<Descriptor, CompositeLocation>();
90     this.fieldLocName2cd = new Hashtable<String, ClassDescriptor>();
91     this.md2ReturnLoc = new Hashtable<MethodDescriptor, CompositeLocation>();
92     this.md2ReturnLocGen = new Hashtable<MethodDescriptor, ReturnLocGenerator>();
93   }
94
95   public void init() {
96
97     // construct mapping from the location name to the class descriptor
98     // assume that the location name is unique through the whole program
99
100     Set<ClassDescriptor> cdSet = ssjava.getCd2lattice().keySet();
101     for (Iterator iterator = cdSet.iterator(); iterator.hasNext();) {
102       ClassDescriptor cd = (ClassDescriptor) iterator.next();
103       SSJavaLattice<String> lattice = ssjava.getCd2lattice().get(cd);
104       Set<String> fieldLocNameSet = lattice.getKeySet();
105
106       for (Iterator iterator2 = fieldLocNameSet.iterator(); iterator2.hasNext();) {
107         String fieldLocName = (String) iterator2.next();
108         fieldLocName2cd.put(fieldLocName, cd);
109       }
110
111     }
112
113   }
114
115   public boolean toAnalyzeIsEmpty() {
116     if (deterministic) {
117       return toanalyzeList.isEmpty();
118     } else {
119       return toanalyze.isEmpty();
120     }
121   }
122
123   public ClassDescriptor toAnalyzeNext() {
124     if (deterministic) {
125       return toanalyzeList.remove(0);
126     } else {
127       ClassDescriptor cd = toanalyze.iterator().next();
128       toanalyze.remove(cd);
129       return cd;
130     }
131   }
132
133   public void setupToAnalyze() {
134     SymbolTable classtable = state.getClassSymbolTable();
135     if (deterministic) {
136       toanalyzeList.clear();
137       toanalyzeList.addAll(classtable.getValueSet());
138       Collections.sort(toanalyzeList, new Comparator<ClassDescriptor>() {
139         public int compare(ClassDescriptor o1, ClassDescriptor o2) {
140           return o1.getClassName().compareToIgnoreCase(o2.getClassName());
141         }
142       });
143     } else {
144       toanalyze.clear();
145       toanalyze.addAll(classtable.getValueSet());
146     }
147   }
148
149   public void setupToAnalazeMethod(ClassDescriptor cd) {
150
151     SymbolTable methodtable = cd.getMethodTable();
152     if (deterministic) {
153       toanalyzeMethodList.clear();
154       toanalyzeMethodList.addAll(methodtable.getValueSet());
155       Collections.sort(toanalyzeMethodList, new Comparator<MethodDescriptor>() {
156         public int compare(MethodDescriptor o1, MethodDescriptor o2) {
157           return o1.getSymbol().compareToIgnoreCase(o2.getSymbol());
158         }
159       });
160     } else {
161       toanalyzeMethod.clear();
162       toanalyzeMethod.addAll(methodtable.getValueSet());
163     }
164   }
165
166   public boolean toAnalyzeMethodIsEmpty() {
167     if (deterministic) {
168       return toanalyzeMethodList.isEmpty();
169     } else {
170       return toanalyzeMethod.isEmpty();
171     }
172   }
173
174   public MethodDescriptor toAnalyzeMethodNext() {
175     if (deterministic) {
176       return toanalyzeMethodList.remove(0);
177     } else {
178       MethodDescriptor md = toanalyzeMethod.iterator().next();
179       toanalyzeMethod.remove(md);
180       return md;
181     }
182   }
183
184   public void flowDownCheck() {
185
186     // phase 1 : checking declaration node and creating mapping of 'type
187     // desciptor' & 'location'
188     setupToAnalyze();
189
190     while (!toAnalyzeIsEmpty()) {
191       ClassDescriptor cd = toAnalyzeNext();
192
193       if (ssjava.needToBeAnnoated(cd)) {
194
195         ClassDescriptor superDesc = cd.getSuperDesc();
196
197         if (superDesc != null && (!superDesc.getSymbol().equals("Object"))) {
198           checkOrderingInheritance(superDesc, cd);
199         }
200
201         checkDeclarationInClass(cd);
202
203         setupToAnalazeMethod(cd);
204         while (!toAnalyzeMethodIsEmpty()) {
205           MethodDescriptor md = toAnalyzeMethodNext();
206           if (ssjava.needTobeAnnotated(md)) {
207             checkDeclarationInMethodBody(cd, md);
208           }
209         }
210
211       }
212
213     }
214
215     // phase2 : checking assignments
216     setupToAnalyze();
217
218     while (!toAnalyzeIsEmpty()) {
219       ClassDescriptor cd = toAnalyzeNext();
220
221       setupToAnalazeMethod(cd);
222       while (!toAnalyzeMethodIsEmpty()) {
223         MethodDescriptor md = toAnalyzeMethodNext();
224         if (ssjava.needTobeAnnotated(md)) {
225           if (state.SSJAVADEBUG) {
226             System.out.println("SSJAVA: Checking Flow-down Rules: " + md);
227           }
228           checkMethodBody(cd, md, null);
229         }
230       }
231     }
232
233   }
234
235   private void checkOrderingInheritance(ClassDescriptor superCd, ClassDescriptor cd) {
236     // here, we're going to check that sub class keeps same relative orderings
237     // in respect to super class
238
239     SSJavaLattice<String> superLattice = ssjava.getClassLattice(superCd);
240     SSJavaLattice<String> subLattice = ssjava.getClassLattice(cd);
241
242     if (superLattice != null) {
243       // if super class doesn't define lattice, then we don't need to check its
244       // subclass
245       if (subLattice == null) {
246         throw new Error("If a parent class '" + superCd
247             + "' has a ordering lattice, its subclass '" + cd + "' should have one.");
248       }
249
250       Set<Pair<String, String>> superPairSet = superLattice.getOrderingPairSet();
251       Set<Pair<String, String>> subPairSet = subLattice.getOrderingPairSet();
252
253       for (Iterator iterator = superPairSet.iterator(); iterator.hasNext();) {
254         Pair<String, String> pair = (Pair<String, String>) iterator.next();
255
256         if (!subPairSet.contains(pair)) {
257           throw new Error("Subclass '" + cd + "' does not have the relative ordering '"
258               + pair.getSecond() + " < " + pair.getFirst()
259               + "' that is defined by its superclass '" + superCd + "'.");
260         }
261       }
262     }
263
264     MethodLattice<String> superMethodDefaultLattice = ssjava.getMethodDefaultLattice(superCd);
265     MethodLattice<String> subMethodDefaultLattice = ssjava.getMethodDefaultLattice(cd);
266
267     if (superMethodDefaultLattice != null) {
268       if (subMethodDefaultLattice == null) {
269         throw new Error("When a parent class '" + superCd
270             + "' defines a default method lattice, its subclass '" + cd + "' should define one.");
271       }
272
273       Set<Pair<String, String>> superPairSet = superMethodDefaultLattice.getOrderingPairSet();
274       Set<Pair<String, String>> subPairSet = subMethodDefaultLattice.getOrderingPairSet();
275
276       for (Iterator iterator = superPairSet.iterator(); iterator.hasNext();) {
277         Pair<String, String> pair = (Pair<String, String>) iterator.next();
278
279         if (!subPairSet.contains(pair)) {
280           throw new Error("Subclass '" + cd + "' does not have the relative ordering '"
281               + pair.getSecond() + " < " + pair.getFirst()
282               + "' that is defined by its superclass '" + superCd
283               + "' in the method default lattice.");
284         }
285       }
286
287     }
288
289   }
290
291   public Hashtable getMap() {
292     return d2loc;
293   }
294
295   private void checkDeclarationInMethodBody(ClassDescriptor cd, MethodDescriptor md) {
296     BlockNode bn = state.getMethodBody(md);
297
298     // first, check annotations on method parameters
299     List<CompositeLocation> paramList = new ArrayList<CompositeLocation>();
300     for (int i = 0; i < md.numParameters(); i++) {
301       // process annotations on method parameters
302       VarDescriptor vd = (VarDescriptor) md.getParameter(i);
303       assignLocationOfVarDescriptor(vd, md, md.getParameterTable(), null);
304       paramList.add(d2loc.get(vd));
305     }
306     Vector<AnnotationDescriptor> methodAnnotations = md.getModifiers().getAnnotations();
307
308     // second, check return location annotation
309     if (!md.getReturnType().isVoid()) {
310       CompositeLocation returnLocComp = null;
311
312       boolean hasReturnLocDeclaration = false;
313       if (methodAnnotations != null) {
314         for (int i = 0; i < methodAnnotations.size(); i++) {
315           AnnotationDescriptor an = methodAnnotations.elementAt(i);
316           if (an.getMarker().equals(ssjava.RETURNLOC)) {
317             // this case, developer explicitly defines method lattice
318             String returnLocDeclaration = an.getValue();
319             returnLocComp = parseLocationDeclaration(md, null, returnLocDeclaration);
320             hasReturnLocDeclaration = true;
321           }
322         }
323       }
324
325       if (!hasReturnLocDeclaration) {
326         // if developer does not define method lattice
327         // search return location in the method default lattice
328         String rtrStr = ssjava.getMethodLattice(md).getReturnLoc();
329         if (rtrStr != null) {
330           returnLocComp = new CompositeLocation(new Location(md, rtrStr));
331         }
332       }
333
334       if (returnLocComp == null) {
335         throw new Error("Return location is not specified for the method " + md + " at "
336             + cd.getSourceFileName());
337       }
338
339       md2ReturnLoc.put(md, returnLocComp);
340
341       // check this location
342       MethodLattice<String> methodLattice = ssjava.getMethodLattice(md);
343       String thisLocId = methodLattice.getThisLoc();
344       if (thisLocId == null) {
345         throw new Error("Method '" + md + "' does not have the definition of 'this' location at "
346             + md.getClassDesc().getSourceFileName());
347       }
348       CompositeLocation thisLoc = new CompositeLocation(new Location(md, thisLocId));
349       paramList.add(0, thisLoc);
350
351       // System.out.println("### ReturnLocGenerator=" + md);
352       // System.out.println("### md2ReturnLoc.get(md)=" + md2ReturnLoc.get(md));
353
354       md2ReturnLocGen.put(md, new ReturnLocGenerator(md2ReturnLoc.get(md), md, paramList, md
355           + " of " + cd.getSourceFileName()));
356     }
357
358     // fourth, check declarations inside of method
359
360     checkDeclarationInBlockNode(md, md.getParameterTable(), bn);
361
362   }
363
364   private void checkDeclarationInBlockNode(MethodDescriptor md, SymbolTable nametable, BlockNode bn) {
365     bn.getVarTable().setParent(nametable);
366     for (int i = 0; i < bn.size(); i++) {
367       BlockStatementNode bsn = bn.get(i);
368       checkDeclarationInBlockStatementNode(md, bn.getVarTable(), bsn);
369     }
370   }
371
372   private void checkDeclarationInBlockStatementNode(MethodDescriptor md, SymbolTable nametable,
373       BlockStatementNode bsn) {
374
375     switch (bsn.kind()) {
376     case Kind.SubBlockNode:
377       checkDeclarationInSubBlockNode(md, nametable, (SubBlockNode) bsn);
378       return;
379
380     case Kind.DeclarationNode:
381       checkDeclarationNode(md, nametable, (DeclarationNode) bsn);
382       break;
383
384     case Kind.LoopNode:
385       checkDeclarationInLoopNode(md, nametable, (LoopNode) bsn);
386       break;
387
388     case Kind.IfStatementNode:
389       checkDeclarationInIfStatementNode(md, nametable, (IfStatementNode) bsn);
390       return;
391
392     case Kind.SwitchStatementNode:
393       checkDeclarationInSwitchStatementNode(md, nametable, (SwitchStatementNode) bsn);
394       return;
395
396     case Kind.SynchronizedNode:
397       checkDeclarationInSynchronizedNode(md, nametable, (SynchronizedNode) bsn);
398       return;
399
400     }
401   }
402
403   private void checkDeclarationInSynchronizedNode(MethodDescriptor md, SymbolTable nametable,
404       SynchronizedNode sbn) {
405     checkDeclarationInBlockNode(md, nametable, sbn.getBlockNode());
406   }
407
408   private void checkDeclarationInSwitchStatementNode(MethodDescriptor md, SymbolTable nametable,
409       SwitchStatementNode ssn) {
410     BlockNode sbn = ssn.getSwitchBody();
411     for (int i = 0; i < sbn.size(); i++) {
412       SwitchBlockNode node = (SwitchBlockNode) sbn.get(i);
413       checkDeclarationInBlockNode(md, nametable, node.getSwitchBlockStatement());
414     }
415   }
416
417   private void checkDeclarationInIfStatementNode(MethodDescriptor md, SymbolTable nametable,
418       IfStatementNode isn) {
419     checkDeclarationInBlockNode(md, nametable, isn.getTrueBlock());
420     if (isn.getFalseBlock() != null)
421       checkDeclarationInBlockNode(md, nametable, isn.getFalseBlock());
422   }
423
424   private void checkDeclarationInLoopNode(MethodDescriptor md, SymbolTable nametable, LoopNode ln) {
425
426     if (ln.getType() == LoopNode.FORLOOP) {
427       // check for loop case
428       ClassDescriptor cd = md.getClassDesc();
429       BlockNode bn = ln.getInitializer();
430       for (int i = 0; i < bn.size(); i++) {
431         BlockStatementNode bsn = bn.get(i);
432         checkDeclarationInBlockStatementNode(md, nametable, bsn);
433       }
434     }
435
436     // check loop body
437     checkDeclarationInBlockNode(md, nametable, ln.getBody());
438   }
439
440   private void checkMethodBody(ClassDescriptor cd, MethodDescriptor md,
441       CompositeLocation constraints) {
442     BlockNode bn = state.getMethodBody(md);
443     checkLocationFromBlockNode(md, md.getParameterTable(), bn, constraints);
444   }
445
446   private String generateErrorMessage(ClassDescriptor cd, TreeNode tn) {
447     if (tn != null) {
448       return cd.getSourceFileName() + "::" + tn.getNumLine();
449     } else {
450       return cd.getSourceFileName();
451     }
452
453   }
454
455   private CompositeLocation checkLocationFromBlockNode(MethodDescriptor md, SymbolTable nametable,
456       BlockNode bn, CompositeLocation constraint) {
457
458     bn.getVarTable().setParent(nametable);
459     for (int i = 0; i < bn.size(); i++) {
460       BlockStatementNode bsn = bn.get(i);
461       checkLocationFromBlockStatementNode(md, bn.getVarTable(), bsn, constraint);
462     }
463     return new CompositeLocation();
464
465   }
466
467   private CompositeLocation checkLocationFromBlockStatementNode(MethodDescriptor md,
468       SymbolTable nametable, BlockStatementNode bsn, CompositeLocation constraint) {
469
470     CompositeLocation compLoc = null;
471     switch (bsn.kind()) {
472     case Kind.BlockExpressionNode:
473       compLoc =
474           checkLocationFromBlockExpressionNode(md, nametable, (BlockExpressionNode) bsn, constraint);
475       break;
476
477     case Kind.DeclarationNode:
478       compLoc = checkLocationFromDeclarationNode(md, nametable, (DeclarationNode) bsn, constraint);
479       break;
480
481     case Kind.IfStatementNode:
482       compLoc = checkLocationFromIfStatementNode(md, nametable, (IfStatementNode) bsn, constraint);
483       break;
484
485     case Kind.LoopNode:
486       compLoc = checkLocationFromLoopNode(md, nametable, (LoopNode) bsn, constraint);
487       break;
488
489     case Kind.ReturnNode:
490       compLoc = checkLocationFromReturnNode(md, nametable, (ReturnNode) bsn, constraint);
491       break;
492
493     case Kind.SubBlockNode:
494       compLoc = checkLocationFromSubBlockNode(md, nametable, (SubBlockNode) bsn, constraint);
495       break;
496
497     case Kind.ContinueBreakNode:
498       compLoc = new CompositeLocation();
499       break;
500
501     case Kind.SwitchStatementNode:
502       compLoc =
503           checkLocationFromSwitchStatementNode(md, nametable, (SwitchStatementNode) bsn, constraint);
504
505     }
506     return compLoc;
507   }
508
509   private CompositeLocation checkLocationFromSwitchStatementNode(MethodDescriptor md,
510       SymbolTable nametable, SwitchStatementNode ssn, CompositeLocation constraint) {
511
512     ClassDescriptor cd = md.getClassDesc();
513     CompositeLocation condLoc =
514         checkLocationFromExpressionNode(md, nametable, ssn.getCondition(), new CompositeLocation(),
515             constraint, false);
516     BlockNode sbn = ssn.getSwitchBody();
517
518     constraint = generateNewConstraint(constraint, condLoc);
519
520     for (int i = 0; i < sbn.size(); i++) {
521       checkLocationFromSwitchBlockNode(md, nametable, (SwitchBlockNode) sbn.get(i), constraint);
522     }
523     return new CompositeLocation();
524   }
525
526   private CompositeLocation checkLocationFromSwitchBlockNode(MethodDescriptor md,
527       SymbolTable nametable, SwitchBlockNode sbn, CompositeLocation constraint) {
528
529     CompositeLocation blockLoc =
530         checkLocationFromBlockNode(md, nametable, sbn.getSwitchBlockStatement(), constraint);
531
532     return blockLoc;
533
534   }
535
536   private CompositeLocation checkLocationFromReturnNode(MethodDescriptor md, SymbolTable nametable,
537       ReturnNode rn, CompositeLocation constraint) {
538
539     ExpressionNode returnExp = rn.getReturnExpression();
540
541     CompositeLocation returnValueLoc;
542     if (returnExp != null) {
543       returnValueLoc =
544           checkLocationFromExpressionNode(md, nametable, returnExp, new CompositeLocation(),
545               constraint, false);
546
547       // System.out.println("# RETURN VALUE LOC=" + returnValueLoc +
548       // " with constraint=" + constraint);
549
550       // TODO: do we need to check here?
551       // if this return statement is inside branch, return value has an implicit
552       // flow from conditional location
553       // if (constraint != null) {
554       // Set<CompositeLocation> inputGLB = new HashSet<CompositeLocation>();
555       // inputGLB.add(returnValueLoc);
556       // inputGLB.add(constraint);
557       // returnValueLoc =
558       // CompositeLattice.calculateGLB(inputGLB,
559       // generateErrorMessage(md.getClassDesc(), rn));
560       // }
561
562       // check if return value is equal or higher than RETRUNLOC of method
563       // declaration annotation
564       CompositeLocation declaredReturnLoc = md2ReturnLoc.get(md);
565
566       int compareResult =
567           CompositeLattice.compare(returnValueLoc, declaredReturnLoc, false,
568               generateErrorMessage(md.getClassDesc(), rn));
569
570       if (compareResult == ComparisonResult.LESS || compareResult == ComparisonResult.INCOMPARABLE) {
571         throw new Error(
572             "Return value location is not equal or higher than the declaraed return location at "
573                 + md.getClassDesc().getSourceFileName() + "::" + rn.getNumLine());
574       }
575     }
576
577     return new CompositeLocation();
578   }
579
580   private boolean hasOnlyLiteralValue(ExpressionNode en) {
581     if (en.kind() == Kind.LiteralNode) {
582       return true;
583     } else {
584       return false;
585     }
586   }
587
588   private CompositeLocation checkLocationFromLoopNode(MethodDescriptor md, SymbolTable nametable,
589       LoopNode ln, CompositeLocation constraint) {
590
591     ClassDescriptor cd = md.getClassDesc();
592     if (ln.getType() == LoopNode.WHILELOOP || ln.getType() == LoopNode.DOWHILELOOP) {
593
594       CompositeLocation condLoc =
595           checkLocationFromExpressionNode(md, nametable, ln.getCondition(),
596               new CompositeLocation(), constraint, false);
597       // addLocationType(ln.getCondition().getType(), (condLoc));
598
599       constraint = generateNewConstraint(constraint, condLoc);
600       checkLocationFromBlockNode(md, nametable, ln.getBody(), constraint);
601
602       return new CompositeLocation();
603
604     } else {
605       // check 'for loop' case
606       BlockNode bn = ln.getInitializer();
607       bn.getVarTable().setParent(nametable);
608
609       // calculate glb location of condition and update statements
610       CompositeLocation condLoc =
611           checkLocationFromExpressionNode(md, bn.getVarTable(), ln.getCondition(),
612               new CompositeLocation(), constraint, false);
613       // addLocationType(ln.getCondition().getType(), condLoc);
614
615       constraint = generateNewConstraint(constraint, condLoc);
616
617       checkLocationFromBlockNode(md, bn.getVarTable(), ln.getUpdate(), constraint);
618       checkLocationFromBlockNode(md, bn.getVarTable(), ln.getBody(), constraint);
619
620       return new CompositeLocation();
621
622     }
623
624   }
625
626   private CompositeLocation checkLocationFromSubBlockNode(MethodDescriptor md,
627       SymbolTable nametable, SubBlockNode sbn, CompositeLocation constraint) {
628     CompositeLocation compLoc =
629         checkLocationFromBlockNode(md, nametable, sbn.getBlockNode(), constraint);
630     return compLoc;
631   }
632
633   private CompositeLocation generateNewConstraint(CompositeLocation currentCon,
634       CompositeLocation newCon) {
635
636     if (currentCon == null) {
637       return newCon;
638     } else {
639       // compute GLB of current constraint and new constraint
640       Set<CompositeLocation> inputSet = new HashSet<CompositeLocation>();
641       inputSet.add(currentCon);
642       inputSet.add(newCon);
643       return CompositeLattice.calculateGLB(inputSet, "");
644     }
645
646   }
647
648   private CompositeLocation checkLocationFromIfStatementNode(MethodDescriptor md,
649       SymbolTable nametable, IfStatementNode isn, CompositeLocation constraint) {
650
651     CompositeLocation condLoc =
652         checkLocationFromExpressionNode(md, nametable, isn.getCondition(), new CompositeLocation(),
653             constraint, false);
654
655     // addLocationType(isn.getCondition().getType(), condLoc);
656
657     constraint = generateNewConstraint(constraint, condLoc);
658     checkLocationFromBlockNode(md, nametable, isn.getTrueBlock(), constraint);
659
660     if (isn.getFalseBlock() != null) {
661       checkLocationFromBlockNode(md, nametable, isn.getFalseBlock(), constraint);
662     }
663
664     return new CompositeLocation();
665   }
666
667   private void checkOwnership(MethodDescriptor md, TreeNode tn, ExpressionNode srcExpNode) {
668
669     if (srcExpNode.kind() == Kind.NameNode || srcExpNode.kind() == Kind.FieldAccessNode) {
670       if (srcExpNode.getType().isPtr() && !srcExpNode.getType().isNull()) {
671         // first, check the linear type
672         // RHS reference should be owned by the current method
673         FieldDescriptor fd = getFieldDescriptorFromExpressionNode(srcExpNode);
674         boolean isOwned;
675         if (fd == null) {
676           // local var case
677           isOwned = ((SSJavaType) srcExpNode.getType().getExtension()).isOwned();
678         } else {
679           // field case
680           isOwned = ssjava.isOwnedByMethod(md, fd);
681         }
682         if (!isOwned) {
683           throw new Error(
684               "It is now allowed to create the reference alias from the reference not owned by the method at "
685                   + generateErrorMessage(md.getClassDesc(), tn));
686         }
687
688       }
689     }
690
691   }
692
693   private CompositeLocation checkLocationFromDeclarationNode(MethodDescriptor md,
694       SymbolTable nametable, DeclarationNode dn, CompositeLocation constraint) {
695
696     VarDescriptor vd = dn.getVarDescriptor();
697
698     CompositeLocation destLoc = d2loc.get(vd);
699
700     if (dn.getExpression() != null) {
701
702       checkOwnership(md, dn, dn.getExpression());
703
704       CompositeLocation expressionLoc =
705           checkLocationFromExpressionNode(md, nametable, dn.getExpression(),
706               new CompositeLocation(), constraint, false);
707       // addTypeLocation(dn.getExpression().getType(), expressionLoc);
708
709       if (expressionLoc != null) {
710
711         // checking location order
712         if (!CompositeLattice.isGreaterThan(expressionLoc, destLoc,
713             generateErrorMessage(md.getClassDesc(), dn))) {
714           throw new Error("The value flow from " + expressionLoc + " to " + destLoc
715               + " does not respect location hierarchy on the assignment " + dn.printNode(0)
716               + " at " + md.getClassDesc().getSourceFileName() + "::" + dn.getNumLine());
717         }
718       }
719       return expressionLoc;
720
721     } else {
722
723       return new CompositeLocation();
724
725     }
726
727   }
728
729   private void checkDeclarationInSubBlockNode(MethodDescriptor md, SymbolTable nametable,
730       SubBlockNode sbn) {
731     checkDeclarationInBlockNode(md, nametable.getParent(), sbn.getBlockNode());
732   }
733
734   private CompositeLocation checkLocationFromBlockExpressionNode(MethodDescriptor md,
735       SymbolTable nametable, BlockExpressionNode ben, CompositeLocation constraint) {
736     CompositeLocation compLoc =
737         checkLocationFromExpressionNode(md, nametable, ben.getExpression(), null, constraint, false);
738     // addTypeLocation(ben.getExpression().getType(), compLoc);
739     return compLoc;
740   }
741
742   private CompositeLocation checkLocationFromExpressionNode(MethodDescriptor md,
743       SymbolTable nametable, ExpressionNode en, CompositeLocation loc,
744       CompositeLocation constraint, boolean isLHS) {
745
746     CompositeLocation compLoc = null;
747     switch (en.kind()) {
748
749     case Kind.AssignmentNode:
750       compLoc =
751           checkLocationFromAssignmentNode(md, nametable, (AssignmentNode) en, loc, constraint);
752       break;
753
754     case Kind.FieldAccessNode:
755       compLoc =
756           checkLocationFromFieldAccessNode(md, nametable, (FieldAccessNode) en, loc, constraint);
757       break;
758
759     case Kind.NameNode:
760       compLoc = checkLocationFromNameNode(md, nametable, (NameNode) en, loc, constraint);
761       break;
762
763     case Kind.OpNode:
764       compLoc = checkLocationFromOpNode(md, nametable, (OpNode) en, constraint);
765       break;
766
767     case Kind.CreateObjectNode:
768       compLoc = checkLocationFromCreateObjectNode(md, nametable, (CreateObjectNode) en);
769       break;
770
771     case Kind.ArrayAccessNode:
772       compLoc =
773           checkLocationFromArrayAccessNode(md, nametable, (ArrayAccessNode) en, constraint, isLHS);
774       break;
775
776     case Kind.LiteralNode:
777       compLoc = checkLocationFromLiteralNode(md, nametable, (LiteralNode) en, loc);
778       break;
779
780     case Kind.MethodInvokeNode:
781       compLoc =
782           checkLocationFromMethodInvokeNode(md, nametable, (MethodInvokeNode) en, loc, constraint);
783       break;
784
785     case Kind.TertiaryNode:
786       compLoc = checkLocationFromTertiaryNode(md, nametable, (TertiaryNode) en, constraint);
787       break;
788
789     case Kind.CastNode:
790       compLoc = checkLocationFromCastNode(md, nametable, (CastNode) en, constraint);
791       break;
792
793     // case Kind.InstanceOfNode:
794     // checkInstanceOfNode(md, nametable, (InstanceOfNode) en, td);
795     // return null;
796
797     // case Kind.ArrayInitializerNode:
798     // checkArrayInitializerNode(md, nametable, (ArrayInitializerNode) en,
799     // td);
800     // return null;
801
802     // case Kind.ClassTypeNode:
803     // checkClassTypeNode(md, nametable, (ClassTypeNode) en, td);
804     // return null;
805
806     // case Kind.OffsetNode:
807     // checkOffsetNode(md, nametable, (OffsetNode)en, td);
808     // return null;
809
810     default:
811       return null;
812
813     }
814     // addTypeLocation(en.getType(), compLoc);
815     return compLoc;
816
817   }
818
819   private CompositeLocation checkLocationFromCastNode(MethodDescriptor md, SymbolTable nametable,
820       CastNode cn, CompositeLocation constraint) {
821
822     ExpressionNode en = cn.getExpression();
823     return checkLocationFromExpressionNode(md, nametable, en, new CompositeLocation(), constraint,
824         false);
825
826   }
827
828   private CompositeLocation checkLocationFromTertiaryNode(MethodDescriptor md,
829       SymbolTable nametable, TertiaryNode tn, CompositeLocation constraint) {
830     ClassDescriptor cd = md.getClassDesc();
831
832     CompositeLocation condLoc =
833         checkLocationFromExpressionNode(md, nametable, tn.getCond(), new CompositeLocation(),
834             constraint, false);
835     // addLocationType(tn.getCond().getType(), condLoc);
836     CompositeLocation trueLoc =
837         checkLocationFromExpressionNode(md, nametable, tn.getTrueExpr(), new CompositeLocation(),
838             constraint, false);
839     // addLocationType(tn.getTrueExpr().getType(), trueLoc);
840     CompositeLocation falseLoc =
841         checkLocationFromExpressionNode(md, nametable, tn.getFalseExpr(), new CompositeLocation(),
842             constraint, false);
843     // addLocationType(tn.getFalseExpr().getType(), falseLoc);
844
845     // locations from true/false branches can be TOP when there are only literal
846     // values
847     // in this case, we don't need to check flow down rule!
848
849     // System.out.println("\n#tertiary cond=" + tn.getCond().printNode(0) +
850     // " Loc=" + condLoc);
851     // System.out.println("# true=" + tn.getTrueExpr().printNode(0) + " Loc=" +
852     // trueLoc);
853     // System.out.println("# false=" + tn.getFalseExpr().printNode(0) + " Loc="
854     // + falseLoc);
855
856     // check if condLoc is higher than trueLoc & falseLoc
857     if (!trueLoc.get(0).isTop()
858         && !CompositeLattice.isGreaterThan(condLoc, trueLoc, generateErrorMessage(cd, tn))) {
859       throw new Error(
860           "The location of the condition expression is lower than the true expression at "
861               + cd.getSourceFileName() + ":" + tn.getCond().getNumLine());
862     }
863
864     if (!falseLoc.get(0).isTop()
865         && !CompositeLattice.isGreaterThan(condLoc, falseLoc,
866             generateErrorMessage(cd, tn.getCond()))) {
867       throw new Error(
868           "The location of the condition expression is lower than the false expression at "
869               + cd.getSourceFileName() + ":" + tn.getCond().getNumLine());
870     }
871
872     // then, return glb of trueLoc & falseLoc
873     Set<CompositeLocation> glbInputSet = new HashSet<CompositeLocation>();
874     glbInputSet.add(trueLoc);
875     glbInputSet.add(falseLoc);
876
877     if (glbInputSet.size() == 1) {
878       return trueLoc;
879     } else {
880       return CompositeLattice.calculateGLB(glbInputSet, generateErrorMessage(cd, tn));
881     }
882
883   }
884
885   private CompositeLocation checkLocationFromMethodInvokeNode(MethodDescriptor md,
886       SymbolTable nametable, MethodInvokeNode min, CompositeLocation loc,
887       CompositeLocation constraint) {
888
889     ClassDescriptor cd = md.getClassDesc();
890     MethodDescriptor calleeMD = min.getMethod();
891
892     NameDescriptor baseName = min.getBaseName();
893     boolean isSystemout = false;
894     if (baseName != null) {
895       isSystemout = baseName.getSymbol().equals("System.out");
896     }
897
898     if (!ssjava.isSSJavaUtil(calleeMD.getClassDesc()) && !ssjava.isTrustMethod(calleeMD)
899         && !calleeMD.getModifiers().isNative() && !isSystemout) {
900
901       CompositeLocation baseLocation = null;
902       if (min.getExpression() != null) {
903         baseLocation =
904             checkLocationFromExpressionNode(md, nametable, min.getExpression(),
905                 new CompositeLocation(), constraint, false);
906       } else {
907         if (min.getMethod().isStatic()) {
908           String globalLocId = ssjava.getMethodLattice(md).getGlobalLoc();
909           if (globalLocId == null) {
910             throw new Error("Method lattice does not define global variable location at "
911                 + generateErrorMessage(md.getClassDesc(), min));
912           }
913           baseLocation = new CompositeLocation(new Location(md, globalLocId));
914         } else {
915           String thisLocId = ssjava.getMethodLattice(md).getThisLoc();
916           baseLocation = new CompositeLocation(new Location(md, thisLocId));
917         }
918       }
919
920       // System.out.println("\n#checkLocationFromMethodInvokeNode=" +
921       // min.printNode(0)
922       // + " baseLocation=" + baseLocation + " constraint=" + constraint);
923
924       if (constraint != null) {
925         int compareResult =
926             CompositeLattice.compare(constraint, baseLocation, true, generateErrorMessage(cd, min));
927         if (compareResult != ComparisonResult.GREATER) {
928           // if the current constraint is higher than method's THIS location
929           // no need to check constraints!
930           CompositeLocation calleeConstraint =
931               translateCallerLocToCalleeLoc(calleeMD, baseLocation, constraint);
932           // System.out.println("check method body for constraint:" + calleeMD +
933           // " calleeConstraint="
934           // + calleeConstraint);
935           checkMethodBody(calleeMD.getClassDesc(), calleeMD, calleeConstraint);
936         }
937       }
938
939       checkCalleeConstraints(md, nametable, min, baseLocation, constraint);
940
941       // checkCallerArgumentLocationConstraints(md, nametable, min,
942       // baseLocation, constraint);
943
944       if (!min.getMethod().getReturnType().isVoid()) {
945         // If method has a return value, compute the highest possible return
946         // location in the caller's perspective
947         CompositeLocation ceilingLoc =
948             computeCeilingLocationForCaller(md, nametable, min, baseLocation, constraint);
949         return ceilingLoc;
950       }
951     }
952
953     return new CompositeLocation(Location.createTopLocation(md));
954
955   }
956
957   private CompositeLocation translateCallerLocToCalleeLoc(MethodDescriptor calleeMD,
958       CompositeLocation calleeBaseLoc, CompositeLocation constraint) {
959
960     CompositeLocation calleeConstraint = new CompositeLocation();
961
962     // if (constraint.startsWith(calleeBaseLoc)) {
963     // if the first part of constraint loc is matched with callee base loc
964     Location thisLoc = new Location(calleeMD, ssjava.getMethodLattice(calleeMD).getThisLoc());
965     calleeConstraint.addLocation(thisLoc);
966     for (int i = calleeBaseLoc.getSize(); i < constraint.getSize(); i++) {
967       calleeConstraint.addLocation(constraint.get(i));
968     }
969
970     // }
971
972     return calleeConstraint;
973   }
974
975   private void checkCallerArgumentLocationConstraints(MethodDescriptor md, SymbolTable nametable,
976       MethodInvokeNode min, CompositeLocation callerBaseLoc, CompositeLocation constraint) {
977     // if parameter location consists of THIS and FIELD location,
978     // caller should pass an argument that is comparable to the declared
979     // parameter location
980     // and is not lower than the declared parameter location in the field
981     // lattice.
982
983     MethodDescriptor calleemd = min.getMethod();
984
985     List<CompositeLocation> callerArgList = new ArrayList<CompositeLocation>();
986     List<CompositeLocation> calleeParamList = new ArrayList<CompositeLocation>();
987
988     MethodLattice<String> calleeLattice = ssjava.getMethodLattice(calleemd);
989     Location calleeThisLoc = new Location(calleemd, calleeLattice.getThisLoc());
990
991     for (int i = 0; i < min.numArgs(); i++) {
992       ExpressionNode en = min.getArg(i);
993       CompositeLocation callerArgLoc =
994           checkLocationFromExpressionNode(md, nametable, en, new CompositeLocation(), constraint,
995               false);
996       callerArgList.add(callerArgLoc);
997     }
998
999     // setup callee params set
1000     for (int i = 0; i < calleemd.numParameters(); i++) {
1001       VarDescriptor calleevd = (VarDescriptor) calleemd.getParameter(i);
1002       CompositeLocation calleeLoc = d2loc.get(calleevd);
1003       calleeParamList.add(calleeLoc);
1004     }
1005
1006     String errorMsg = generateErrorMessage(md.getClassDesc(), min);
1007
1008     // System.out.println("checkCallerArgumentLocationConstraints=" +
1009     // min.printNode(0));
1010     // System.out.println("base location=" + callerBaseLoc + " constraint=" +
1011     // constraint);
1012
1013     for (int i = 0; i < calleeParamList.size(); i++) {
1014       CompositeLocation calleeParamLoc = calleeParamList.get(i);
1015       if (calleeParamLoc.get(0).equals(calleeThisLoc) && calleeParamLoc.getSize() > 1) {
1016
1017         // callee parameter location has field information
1018         CompositeLocation callerArgLoc = callerArgList.get(i);
1019
1020         CompositeLocation paramLocation =
1021             translateCalleeParamLocToCaller(md, calleeParamLoc, callerBaseLoc, errorMsg);
1022
1023         Set<CompositeLocation> inputGLBSet = new HashSet<CompositeLocation>();
1024         if (constraint != null) {
1025           inputGLBSet.add(callerArgLoc);
1026           inputGLBSet.add(constraint);
1027           callerArgLoc =
1028               CompositeLattice.calculateGLB(inputGLBSet,
1029                   generateErrorMessage(md.getClassDesc(), min));
1030         }
1031
1032         if (!CompositeLattice.isGreaterThan(callerArgLoc, paramLocation, errorMsg)) {
1033           throw new Error("Caller argument '" + min.getArg(i).printNode(0) + " : " + callerArgLoc
1034               + "' should be higher than corresponding callee's parameter : " + paramLocation
1035               + " at " + errorMsg);
1036         }
1037
1038       }
1039     }
1040
1041   }
1042
1043   private CompositeLocation translateCalleeParamLocToCaller(MethodDescriptor md,
1044       CompositeLocation calleeParamLoc, CompositeLocation callerBaseLocation, String errorMsg) {
1045
1046     CompositeLocation translate = new CompositeLocation();
1047
1048     for (int i = 0; i < callerBaseLocation.getSize(); i++) {
1049       translate.addLocation(callerBaseLocation.get(i));
1050     }
1051
1052     for (int i = 1; i < calleeParamLoc.getSize(); i++) {
1053       translate.addLocation(calleeParamLoc.get(i));
1054     }
1055
1056     // System.out.println("TRANSLATED=" + translate + " from calleeParamLoc=" +
1057     // calleeParamLoc);
1058
1059     return translate;
1060   }
1061
1062   private CompositeLocation computeCeilingLocationForCaller(MethodDescriptor md,
1063       SymbolTable nametable, MethodInvokeNode min, CompositeLocation baseLocation,
1064       CompositeLocation constraint) {
1065     List<CompositeLocation> argList = new ArrayList<CompositeLocation>();
1066
1067     // by default, method has a THIS parameter
1068     argList.add(baseLocation);
1069
1070     for (int i = 0; i < min.numArgs(); i++) {
1071       ExpressionNode en = min.getArg(i);
1072       CompositeLocation callerArg =
1073           checkLocationFromExpressionNode(md, nametable, en, new CompositeLocation(), constraint,
1074               false);
1075       argList.add(callerArg);
1076     }
1077
1078     // System.out.println("\n## computeReturnLocation=" + min.getMethod() +
1079     // " argList=" + argList);
1080     CompositeLocation ceilLoc = md2ReturnLocGen.get(min.getMethod()).computeReturnLocation(argList);
1081     // System.out.println("## ReturnLocation=" + ceilLoc);
1082
1083     return ceilLoc;
1084
1085   }
1086
1087   private void checkCalleeConstraints(MethodDescriptor md, SymbolTable nametable,
1088       MethodInvokeNode min, CompositeLocation callerBaseLoc, CompositeLocation constraint) {
1089
1090     // System.out.println("checkCalleeConstraints=" + min.printNode(0));
1091
1092     MethodDescriptor calleemd = min.getMethod();
1093
1094     MethodLattice<String> calleeLattice = ssjava.getMethodLattice(calleemd);
1095     CompositeLocation calleeThisLoc =
1096         new CompositeLocation(new Location(calleemd, calleeLattice.getThisLoc()));
1097
1098     List<CompositeLocation> callerArgList = new ArrayList<CompositeLocation>();
1099     List<CompositeLocation> calleeParamList = new ArrayList<CompositeLocation>();
1100
1101     if (min.numArgs() > 0) {
1102       // caller needs to guarantee that it passes arguments in regarding to
1103       // callee's hierarchy
1104
1105       // setup caller args set
1106       // first, add caller's base(this) location
1107       callerArgList.add(callerBaseLoc);
1108       // second, add caller's arguments
1109       for (int i = 0; i < min.numArgs(); i++) {
1110         ExpressionNode en = min.getArg(i);
1111         CompositeLocation callerArgLoc =
1112             checkLocationFromExpressionNode(md, nametable, en, new CompositeLocation(), constraint,
1113                 false);
1114         callerArgList.add(callerArgLoc);
1115       }
1116
1117       // setup callee params set
1118       // first, add callee's this location
1119       calleeParamList.add(calleeThisLoc);
1120       // second, add callee's parameters
1121       for (int i = 0; i < calleemd.numParameters(); i++) {
1122         VarDescriptor calleevd = (VarDescriptor) calleemd.getParameter(i);
1123         CompositeLocation calleeLoc = d2loc.get(calleevd);
1124         // System.out.println("calleevd=" + calleevd + " loc=" + calleeLoc);
1125         calleeParamList.add(calleeLoc);
1126       }
1127
1128       // here, check if ordering relations among caller's args respect
1129       // ordering relations in-between callee's args
1130       CHECK: for (int i = 0; i < calleeParamList.size(); i++) {
1131         CompositeLocation calleeLoc1 = calleeParamList.get(i);
1132         CompositeLocation callerLoc1 = callerArgList.get(i);
1133
1134         for (int j = 0; j < calleeParamList.size(); j++) {
1135           if (i != j) {
1136             CompositeLocation calleeLoc2 = calleeParamList.get(j);
1137             CompositeLocation callerLoc2 = callerArgList.get(j);
1138
1139             if (callerLoc1.get(callerLoc1.getSize() - 1).isTop()
1140                 || callerLoc2.get(callerLoc2.getSize() - 1).isTop()) {
1141               continue CHECK;
1142             }
1143
1144             // System.out.println("calleeLoc1=" + calleeLoc1);
1145             // System.out.println("calleeLoc2=" + calleeLoc2 +
1146             // "calleeParamList=" + calleeParamList);
1147
1148             int callerResult =
1149                 CompositeLattice.compare(callerLoc1, callerLoc2, true,
1150                     generateErrorMessage(md.getClassDesc(), min));
1151             // System.out.println("callerResult=" + callerResult);
1152             int calleeResult =
1153                 CompositeLattice.compare(calleeLoc1, calleeLoc2, true,
1154                     generateErrorMessage(md.getClassDesc(), min));
1155             // System.out.println("calleeResult=" + calleeResult);
1156
1157             if (callerResult == ComparisonResult.EQUAL) {
1158               if (ssjava.isSharedLocation(callerLoc1.get(callerLoc1.getSize() - 1))
1159                   && ssjava.isSharedLocation(callerLoc2.get(callerLoc2.getSize() - 1))) {
1160                 // if both of them are shared locations, promote them to
1161                 // "GREATER relation"
1162                 callerResult = ComparisonResult.GREATER;
1163               }
1164             }
1165
1166             if (calleeResult == ComparisonResult.GREATER
1167                 && callerResult != ComparisonResult.GREATER) {
1168               // If calleeLoc1 is higher than calleeLoc2
1169               // then, caller should have same ordering relation in-bet
1170               // callerLoc1 & callerLoc2
1171
1172               String paramName1, paramName2;
1173
1174               if (i == 0) {
1175                 paramName1 = "'THIS'";
1176               } else {
1177                 paramName1 = "'parameter " + calleemd.getParamName(i - 1) + "'";
1178               }
1179
1180               if (j == 0) {
1181                 paramName2 = "'THIS'";
1182               } else {
1183                 paramName2 = "'parameter " + calleemd.getParamName(j - 1) + "'";
1184               }
1185
1186               throw new Error(
1187                   "Caller doesn't respect an ordering relation among method arguments: callee expects that "
1188                       + paramName1 + " should be higher than " + paramName2 + " in " + calleemd
1189                       + " at " + md.getClassDesc().getSourceFileName() + ":" + min.getNumLine());
1190             }
1191           }
1192
1193         }
1194       }
1195
1196     }
1197
1198   }
1199
1200   private CompositeLocation checkLocationFromArrayAccessNode(MethodDescriptor md,
1201       SymbolTable nametable, ArrayAccessNode aan, CompositeLocation constraint, boolean isLHS) {
1202
1203     ClassDescriptor cd = md.getClassDesc();
1204
1205     CompositeLocation arrayLoc =
1206         checkLocationFromExpressionNode(md, nametable, aan.getExpression(),
1207             new CompositeLocation(), constraint, isLHS);
1208     // addTypeLocation(aan.getExpression().getType(), arrayLoc);
1209     CompositeLocation indexLoc =
1210         checkLocationFromExpressionNode(md, nametable, aan.getIndex(), new CompositeLocation(),
1211             constraint, isLHS);
1212     // addTypeLocation(aan.getIndex().getType(), indexLoc);
1213
1214     if (isLHS) {
1215       if (!CompositeLattice.isGreaterThan(indexLoc, arrayLoc, generateErrorMessage(cd, aan))) {
1216         throw new Error("Array index value is not higher than array location at "
1217             + generateErrorMessage(cd, aan));
1218       }
1219       return arrayLoc;
1220     } else {
1221       Set<CompositeLocation> inputGLB = new HashSet<CompositeLocation>();
1222       inputGLB.add(arrayLoc);
1223       inputGLB.add(indexLoc);
1224       return CompositeLattice.calculateGLB(inputGLB, generateErrorMessage(cd, aan));
1225     }
1226
1227   }
1228
1229   private CompositeLocation checkLocationFromCreateObjectNode(MethodDescriptor md,
1230       SymbolTable nametable, CreateObjectNode con) {
1231
1232     ClassDescriptor cd = md.getClassDesc();
1233
1234     CompositeLocation compLoc = new CompositeLocation();
1235     compLoc.addLocation(Location.createTopLocation(md));
1236     return compLoc;
1237
1238   }
1239
1240   private CompositeLocation checkLocationFromOpNode(MethodDescriptor md, SymbolTable nametable,
1241       OpNode on, CompositeLocation constraint) {
1242
1243     ClassDescriptor cd = md.getClassDesc();
1244     CompositeLocation leftLoc = new CompositeLocation();
1245     leftLoc =
1246         checkLocationFromExpressionNode(md, nametable, on.getLeft(), leftLoc, constraint, false);
1247     // addTypeLocation(on.getLeft().getType(), leftLoc);
1248
1249     CompositeLocation rightLoc = new CompositeLocation();
1250     if (on.getRight() != null) {
1251       rightLoc =
1252           checkLocationFromExpressionNode(md, nametable, on.getRight(), rightLoc, constraint, false);
1253       // addTypeLocation(on.getRight().getType(), rightLoc);
1254     }
1255
1256     // System.out.println("\n# OP NODE=" + on.printNode(0));
1257     // System.out.println("# left loc=" + leftLoc + " from " +
1258     // on.getLeft().getClass());
1259     // if (on.getRight() != null) {
1260     // System.out.println("# right loc=" + rightLoc + " from " +
1261     // on.getRight().getClass());
1262     // }
1263
1264     Operation op = on.getOp();
1265
1266     switch (op.getOp()) {
1267
1268     case Operation.UNARYPLUS:
1269     case Operation.UNARYMINUS:
1270     case Operation.LOGIC_NOT:
1271       // single operand
1272       return leftLoc;
1273
1274     case Operation.LOGIC_OR:
1275     case Operation.LOGIC_AND:
1276     case Operation.COMP:
1277     case Operation.BIT_OR:
1278     case Operation.BIT_XOR:
1279     case Operation.BIT_AND:
1280     case Operation.ISAVAILABLE:
1281     case Operation.EQUAL:
1282     case Operation.NOTEQUAL:
1283     case Operation.LT:
1284     case Operation.GT:
1285     case Operation.LTE:
1286     case Operation.GTE:
1287     case Operation.ADD:
1288     case Operation.SUB:
1289     case Operation.MULT:
1290     case Operation.DIV:
1291     case Operation.MOD:
1292     case Operation.LEFTSHIFT:
1293     case Operation.RIGHTSHIFT:
1294     case Operation.URIGHTSHIFT:
1295
1296       Set<CompositeLocation> inputSet = new HashSet<CompositeLocation>();
1297       inputSet.add(leftLoc);
1298       inputSet.add(rightLoc);
1299       CompositeLocation glbCompLoc =
1300           CompositeLattice.calculateGLB(inputSet, generateErrorMessage(cd, on));
1301       // System.out.println("# glbCompLoc=" + glbCompLoc);
1302       return glbCompLoc;
1303
1304     default:
1305       throw new Error(op.toString());
1306     }
1307
1308   }
1309
1310   private CompositeLocation checkLocationFromLiteralNode(MethodDescriptor md,
1311       SymbolTable nametable, LiteralNode en, CompositeLocation loc) {
1312
1313     // literal value has the top location so that value can be flowed into any
1314     // location
1315     Location literalLoc = Location.createTopLocation(md);
1316     loc.addLocation(literalLoc);
1317     return loc;
1318
1319   }
1320
1321   private CompositeLocation checkLocationFromNameNode(MethodDescriptor md, SymbolTable nametable,
1322       NameNode nn, CompositeLocation loc, CompositeLocation constraint) {
1323
1324     NameDescriptor nd = nn.getName();
1325     if (nd.getBase() != null) {
1326       loc =
1327           checkLocationFromExpressionNode(md, nametable, nn.getExpression(), loc, constraint, false);
1328     } else {
1329       String varname = nd.toString();
1330       if (varname.equals("this")) {
1331         // 'this' itself!
1332         MethodLattice<String> methodLattice = ssjava.getMethodLattice(md);
1333         String thisLocId = methodLattice.getThisLoc();
1334         if (thisLocId == null) {
1335           throw new Error("The location for 'this' is not defined at "
1336               + md.getClassDesc().getSourceFileName() + "::" + nn.getNumLine());
1337         }
1338         Location locElement = new Location(md, thisLocId);
1339         loc.addLocation(locElement);
1340         return loc;
1341
1342       }
1343
1344       Descriptor d = (Descriptor) nametable.get(varname);
1345
1346       // CompositeLocation localLoc = null;
1347       if (d instanceof VarDescriptor) {
1348         VarDescriptor vd = (VarDescriptor) d;
1349         // localLoc = d2loc.get(vd);
1350         // the type of var descriptor has a composite location!
1351         loc = ((SSJavaType) vd.getType().getExtension()).getCompLoc().clone();
1352       } else if (d instanceof FieldDescriptor) {
1353         // the type of field descriptor has a location!
1354         FieldDescriptor fd = (FieldDescriptor) d;
1355         if (fd.isStatic()) {
1356           if (fd.isFinal()) {
1357             // if it is 'static final', the location has TOP since no one can
1358             // change its value
1359             loc.addLocation(Location.createTopLocation(md));
1360             return loc;
1361           } else {
1362             // if 'static', the location has pre-assigned global loc
1363             MethodLattice<String> localLattice = ssjava.getMethodLattice(md);
1364             String globalLocId = localLattice.getGlobalLoc();
1365             if (globalLocId == null) {
1366               throw new Error("Global location element is not defined in the method " + md);
1367             }
1368             Location globalLoc = new Location(md, globalLocId);
1369
1370             loc.addLocation(globalLoc);
1371           }
1372         } else {
1373           // the location of field access starts from this, followed by field
1374           // location
1375           MethodLattice<String> localLattice = ssjava.getMethodLattice(md);
1376           Location thisLoc = new Location(md, localLattice.getThisLoc());
1377           loc.addLocation(thisLoc);
1378         }
1379
1380         Location fieldLoc = (Location) fd.getType().getExtension();
1381         loc.addLocation(fieldLoc);
1382       } else if (d == null) {
1383         // access static field
1384         ClassDescriptor cd = nn.getClassDesc();
1385
1386         MethodLattice<String> localLattice = ssjava.getMethodLattice(md);
1387         String globalLocId = localLattice.getGlobalLoc();
1388         if (globalLocId == null) {
1389           throw new Error("Method lattice does not define global variable location at "
1390               + generateErrorMessage(md.getClassDesc(), nn));
1391         }
1392         loc.addLocation(new Location(md, globalLocId));
1393         return loc;
1394
1395       }
1396     }
1397     return loc;
1398   }
1399
1400   private CompositeLocation checkLocationFromFieldAccessNode(MethodDescriptor md,
1401       SymbolTable nametable, FieldAccessNode fan, CompositeLocation loc,
1402       CompositeLocation constraint) {
1403
1404     ExpressionNode left = fan.getExpression();
1405     TypeDescriptor ltd = left.getType();
1406
1407     FieldDescriptor fd = fan.getField();
1408
1409     String varName = null;
1410     if (left.kind() == Kind.NameNode) {
1411       NameDescriptor nd = ((NameNode) left).getName();
1412       varName = nd.toString();
1413     }
1414
1415     if (ltd.isClassNameRef() || (varName != null && varName.equals("this"))) {
1416       // using a class name directly or access using this
1417       if (fd.isStatic() && fd.isFinal()) {
1418         loc.addLocation(Location.createTopLocation(md));
1419         return loc;
1420       }
1421     }
1422
1423     if (left instanceof ArrayAccessNode) {
1424       ArrayAccessNode aan = (ArrayAccessNode) left;
1425       left = aan.getExpression();
1426     }
1427
1428     loc = checkLocationFromExpressionNode(md, nametable, left, loc, constraint, false);
1429     // System.out.println("### checkLocationFromFieldAccessNode=" +
1430     // fan.printNode(0));
1431     // System.out.println("### left=" + left.printNode(0));
1432
1433     if (left.getType().isArray()) {
1434       // array.length case: return the location of the array
1435       return loc;
1436     } else if (!left.getType().isPrimitive()) {
1437       Location fieldLoc = getFieldLocation(fd);
1438       loc.addLocation(fieldLoc);
1439     }
1440     // System.out.println("### field loc=" + loc);
1441     return loc;
1442   }
1443
1444   private Location getFieldLocation(FieldDescriptor fd) {
1445
1446     // System.out.println("### getFieldLocation=" + fd);
1447     // System.out.println("### fd.getType().getExtension()=" +
1448     // fd.getType().getExtension());
1449
1450     Location fieldLoc = (Location) fd.getType().getExtension();
1451
1452     // handle the case that method annotation checking skips checking field
1453     // declaration
1454     if (fieldLoc == null) {
1455       fieldLoc = checkFieldDeclaration(fd.getClassDescriptor(), fd);
1456     }
1457
1458     return fieldLoc;
1459
1460   }
1461
1462   private FieldDescriptor getFieldDescriptorFromExpressionNode(ExpressionNode en) {
1463
1464     if (en.kind() == Kind.NameNode) {
1465       NameNode nn = (NameNode) en;
1466       if (nn.getField() != null) {
1467         return nn.getField();
1468       }
1469
1470       if (nn.getName() != null && nn.getName().getBase() != null) {
1471         return getFieldDescriptorFromExpressionNode(nn.getExpression());
1472       }
1473
1474     } else if (en.kind() == Kind.FieldAccessNode) {
1475       FieldAccessNode fan = (FieldAccessNode) en;
1476       return fan.getField();
1477     }
1478
1479     return null;
1480   }
1481
1482   private CompositeLocation checkLocationFromAssignmentNode(MethodDescriptor md,
1483       SymbolTable nametable, AssignmentNode an, CompositeLocation loc, CompositeLocation constraint) {
1484
1485     // System.out.println("\n# ASSIGNMENTNODE=" + an.printNode(0));
1486
1487     ClassDescriptor cd = md.getClassDesc();
1488
1489     Set<CompositeLocation> inputGLBSet = new HashSet<CompositeLocation>();
1490
1491     boolean postinc = true;
1492     if (an.getOperation().getBaseOp() == null
1493         || (an.getOperation().getBaseOp().getOp() != Operation.POSTINC && an.getOperation()
1494             .getBaseOp().getOp() != Operation.POSTDEC))
1495       postinc = false;
1496
1497     // if LHS is array access node, need to check if array index is higher
1498     // than array itself
1499     CompositeLocation destLocation =
1500         checkLocationFromExpressionNode(md, nametable, an.getDest(), new CompositeLocation(),
1501             constraint, true);
1502
1503     CompositeLocation rhsLocation;
1504     CompositeLocation srcLocation;
1505
1506     if (!postinc) {
1507
1508       checkOwnership(md, an, an.getSrc());
1509
1510       rhsLocation =
1511           checkLocationFromExpressionNode(md, nametable, an.getSrc(), new CompositeLocation(),
1512               constraint, false);
1513
1514       srcLocation = rhsLocation;
1515
1516       // if (!rhsLocation.get(rhsLocation.getSize() - 1).isTop()) {
1517       if (constraint != null) {
1518         inputGLBSet.add(rhsLocation);
1519         inputGLBSet.add(constraint);
1520         srcLocation = CompositeLattice.calculateGLB(inputGLBSet, generateErrorMessage(cd, an));
1521       }
1522       // }
1523
1524       // System.out.println("dstLocation=" + destLocation);
1525       // System.out.println("rhsLocation=" + rhsLocation);
1526       // System.out.println("srcLocation=" + srcLocation);
1527       // System.out.println("constraint=" + constraint);
1528
1529       if (!CompositeLattice.isGreaterThan(srcLocation, destLocation, generateErrorMessage(cd, an))) {
1530
1531         String context = "";
1532         if (constraint != null) {
1533           context = " and the current context constraint is " + constraint;
1534         }
1535
1536         throw new Error("The value flow from " + srcLocation + " to " + destLocation
1537             + " does not respect location hierarchy on the assignment " + an.printNode(0) + context
1538             + " at " + cd.getSourceFileName() + "::" + an.getNumLine());
1539       }
1540
1541     } else {
1542       destLocation =
1543           rhsLocation =
1544               checkLocationFromExpressionNode(md, nametable, an.getDest(), new CompositeLocation(),
1545                   constraint, false);
1546
1547       if (constraint != null) {
1548         inputGLBSet.add(rhsLocation);
1549         inputGLBSet.add(constraint);
1550         srcLocation = CompositeLattice.calculateGLB(inputGLBSet, generateErrorMessage(cd, an));
1551       } else {
1552         srcLocation = rhsLocation;
1553       }
1554
1555       // System.out.println("srcLocation=" + srcLocation);
1556       // System.out.println("rhsLocation=" + rhsLocation);
1557       // System.out.println("constraint=" + constraint);
1558
1559       if (!CompositeLattice.isGreaterThan(srcLocation, destLocation, generateErrorMessage(cd, an))) {
1560
1561         if (srcLocation.equals(destLocation)) {
1562           throw new Error("Location " + srcLocation
1563               + " is not allowed to have the value flow that moves within the same location at '"
1564               + an.printNode(0) + "' of " + cd.getSourceFileName() + "::" + an.getNumLine());
1565         } else {
1566           throw new Error("The value flow from " + srcLocation + " to " + destLocation
1567               + " does not respect location hierarchy on the assignment " + an.printNode(0)
1568               + " at " + cd.getSourceFileName() + "::" + an.getNumLine());
1569         }
1570
1571       }
1572
1573     }
1574
1575     return destLocation;
1576   }
1577
1578   private void assignLocationOfVarDescriptor(VarDescriptor vd, MethodDescriptor md,
1579       SymbolTable nametable, TreeNode n) {
1580
1581     ClassDescriptor cd = md.getClassDesc();
1582     Vector<AnnotationDescriptor> annotationVec = vd.getType().getAnnotationMarkers();
1583
1584     // currently enforce every variable to have corresponding location
1585     if (annotationVec.size() == 0) {
1586       throw new Error("Location is not assigned to variable '" + vd.getSymbol()
1587           + "' in the method '" + md + "' of the class " + cd.getSymbol() + " at "
1588           + generateErrorMessage(cd, n));
1589     }
1590
1591     int locDecCount = 0;
1592     for (int i = 0; i < annotationVec.size(); i++) {
1593       AnnotationDescriptor ad = annotationVec.elementAt(i);
1594
1595       if (ad.getType() == AnnotationDescriptor.SINGLE_ANNOTATION) {
1596
1597         if (ad.getMarker().equals(SSJavaAnalysis.LOC)) {
1598           locDecCount++;
1599           if (locDecCount > 1) {// variable can have at most one location
1600             throw new Error(vd.getSymbol() + " has more than one location declaration.");
1601           }
1602           String locDec = ad.getValue(); // check if location is defined
1603
1604           if (locDec.startsWith(SSJavaAnalysis.DELTA)) {
1605             DeltaLocation deltaLoc = parseDeltaDeclaration(md, n, locDec);
1606             d2loc.put(vd, deltaLoc);
1607             addLocationType(vd.getType(), deltaLoc);
1608           } else {
1609             CompositeLocation compLoc = parseLocationDeclaration(md, n, locDec);
1610
1611             Location lastElement = compLoc.get(compLoc.getSize() - 1);
1612             if (ssjava.isSharedLocation(lastElement)) {
1613               ssjava.mapSharedLocation2Descriptor(lastElement, vd);
1614             }
1615
1616             d2loc.put(vd, compLoc);
1617             addLocationType(vd.getType(), compLoc);
1618           }
1619
1620         }
1621       }
1622     }
1623
1624   }
1625
1626   private DeltaLocation parseDeltaDeclaration(MethodDescriptor md, TreeNode n, String locDec) {
1627
1628     int deltaCount = 0;
1629     int dIdx = locDec.indexOf(SSJavaAnalysis.DELTA);
1630     while (dIdx >= 0) {
1631       deltaCount++;
1632       int beginIdx = dIdx + 6;
1633       locDec = locDec.substring(beginIdx, locDec.length() - 1);
1634       dIdx = locDec.indexOf(SSJavaAnalysis.DELTA);
1635     }
1636
1637     CompositeLocation compLoc = parseLocationDeclaration(md, n, locDec);
1638     DeltaLocation deltaLoc = new DeltaLocation(compLoc, deltaCount);
1639
1640     return deltaLoc;
1641   }
1642
1643   private Location parseFieldLocDeclaraton(String decl, String msg) throws Exception {
1644
1645     int idx = decl.indexOf(".");
1646
1647     String className = decl.substring(0, idx);
1648     String fieldName = decl.substring(idx + 1);
1649
1650     className.replaceAll(" ", "");
1651     fieldName.replaceAll(" ", "");
1652
1653     Descriptor d = state.getClassSymbolTable().get(className);
1654
1655     if (d == null) {
1656       // System.out.println("state.getClassSymbolTable()=" +
1657       // state.getClassSymbolTable());
1658       throw new Error("The class in the location declaration '" + decl + "' does not exist at "
1659           + msg);
1660     }
1661
1662     assert (d instanceof ClassDescriptor);
1663     SSJavaLattice<String> lattice = ssjava.getClassLattice((ClassDescriptor) d);
1664     if (!lattice.containsKey(fieldName)) {
1665       throw new Error("The location " + fieldName + " is not defined in the field lattice of '"
1666           + className + "' at " + msg);
1667     }
1668
1669     return new Location(d, fieldName);
1670   }
1671
1672   private CompositeLocation parseLocationDeclaration(MethodDescriptor md, TreeNode n, String locDec) {
1673
1674     CompositeLocation compLoc = new CompositeLocation();
1675
1676     StringTokenizer tokenizer = new StringTokenizer(locDec, ",");
1677     List<String> locIdList = new ArrayList<String>();
1678     while (tokenizer.hasMoreTokens()) {
1679       String locId = tokenizer.nextToken();
1680       locIdList.add(locId);
1681     }
1682
1683     // at least,one location element needs to be here!
1684     assert (locIdList.size() > 0);
1685
1686     // assume that loc with idx 0 comes from the local lattice
1687     // loc with idx 1 comes from the field lattice
1688
1689     String localLocId = locIdList.get(0);
1690     SSJavaLattice<String> localLattice = CompositeLattice.getLatticeByDescriptor(md);
1691     Location localLoc = new Location(md, localLocId);
1692     if (localLattice == null || (!localLattice.containsKey(localLocId))) {
1693       throw new Error("Location " + localLocId
1694           + " is not defined in the local variable lattice at "
1695           + md.getClassDesc().getSourceFileName() + "::" + (n != null ? n.getNumLine() : md) + ".");
1696     }
1697     compLoc.addLocation(localLoc);
1698
1699     for (int i = 1; i < locIdList.size(); i++) {
1700       String locName = locIdList.get(i);
1701       try {
1702         Location fieldLoc =
1703             parseFieldLocDeclaraton(locName, generateErrorMessage(md.getClassDesc(), n));
1704         compLoc.addLocation(fieldLoc);
1705       } catch (Exception e) {
1706         throw new Error("The location declaration '" + locName + "' is wrong  at "
1707             + generateErrorMessage(md.getClassDesc(), n));
1708       }
1709     }
1710
1711     return compLoc;
1712
1713   }
1714
1715   private void checkDeclarationNode(MethodDescriptor md, SymbolTable nametable, DeclarationNode dn) {
1716     VarDescriptor vd = dn.getVarDescriptor();
1717     assignLocationOfVarDescriptor(vd, md, nametable, dn);
1718   }
1719
1720   private void checkDeclarationInClass(ClassDescriptor cd) {
1721     // Check to see that fields are okay
1722     for (Iterator field_it = cd.getFields(); field_it.hasNext();) {
1723       FieldDescriptor fd = (FieldDescriptor) field_it.next();
1724
1725       if (!(fd.isFinal() && fd.isStatic())) {
1726         checkFieldDeclaration(cd, fd);
1727       } else {
1728         // for static final, assign top location by default
1729         Location loc = Location.createTopLocation(cd);
1730         addLocationType(fd.getType(), loc);
1731       }
1732     }
1733   }
1734
1735   private Location checkFieldDeclaration(ClassDescriptor cd, FieldDescriptor fd) {
1736
1737     Vector<AnnotationDescriptor> annotationVec = fd.getType().getAnnotationMarkers();
1738
1739     // currently enforce every field to have corresponding location
1740     if (annotationVec.size() == 0) {
1741       throw new Error("Location is not assigned to the field '" + fd.getSymbol()
1742           + "' of the class " + cd.getSymbol() + " at " + cd.getSourceFileName());
1743     }
1744
1745     if (annotationVec.size() > 1) {
1746       // variable can have at most one location
1747       throw new Error("Field " + fd.getSymbol() + " of class " + cd
1748           + " has more than one location.");
1749     }
1750
1751     AnnotationDescriptor ad = annotationVec.elementAt(0);
1752     Location loc = null;
1753
1754     if (ad.getType() == AnnotationDescriptor.SINGLE_ANNOTATION) {
1755       if (ad.getMarker().equals(SSJavaAnalysis.LOC)) {
1756         String locationID = ad.getValue();
1757         // check if location is defined
1758         SSJavaLattice<String> lattice = ssjava.getClassLattice(cd);
1759         if (lattice == null || (!lattice.containsKey(locationID))) {
1760           throw new Error("Location " + locationID
1761               + " is not defined in the field lattice of class " + cd.getSymbol() + " at"
1762               + cd.getSourceFileName() + ".");
1763         }
1764         loc = new Location(cd, locationID);
1765
1766         if (ssjava.isSharedLocation(loc)) {
1767           ssjava.mapSharedLocation2Descriptor(loc, fd);
1768         }
1769
1770         addLocationType(fd.getType(), loc);
1771
1772       }
1773     }
1774
1775     return loc;
1776   }
1777
1778   private void addLocationType(TypeDescriptor type, CompositeLocation loc) {
1779     if (type != null) {
1780       TypeExtension te = type.getExtension();
1781       SSJavaType ssType;
1782       if (te != null) {
1783         ssType = (SSJavaType) te;
1784         ssType.setCompLoc(loc);
1785       } else {
1786         ssType = new SSJavaType(loc);
1787         type.setExtension(ssType);
1788       }
1789     }
1790   }
1791
1792   private void addLocationType(TypeDescriptor type, Location loc) {
1793     if (type != null) {
1794       type.setExtension(loc);
1795     }
1796   }
1797
1798   static class CompositeLattice {
1799
1800     public static boolean isGreaterThan(CompositeLocation loc1, CompositeLocation loc2, String msg) {
1801
1802       // System.out.println("\nisGreaterThan=" + loc1 + " " + loc2 + " msg=" +
1803       // msg);
1804       int baseCompareResult = compareBaseLocationSet(loc1, loc2, true, false, msg);
1805       if (baseCompareResult == ComparisonResult.EQUAL) {
1806         if (compareDelta(loc1, loc2) == ComparisonResult.GREATER) {
1807           return true;
1808         } else {
1809           return false;
1810         }
1811       } else if (baseCompareResult == ComparisonResult.GREATER) {
1812         return true;
1813       } else {
1814         return false;
1815       }
1816
1817     }
1818
1819     public static int compare(CompositeLocation loc1, CompositeLocation loc2, boolean ignore,
1820         String msg) {
1821
1822       // System.out.println("compare=" + loc1 + " " + loc2);
1823       int baseCompareResult = compareBaseLocationSet(loc1, loc2, false, ignore, msg);
1824
1825       if (baseCompareResult == ComparisonResult.EQUAL) {
1826         return compareDelta(loc1, loc2);
1827       } else {
1828         return baseCompareResult;
1829       }
1830
1831     }
1832
1833     private static int compareDelta(CompositeLocation dLoc1, CompositeLocation dLoc2) {
1834
1835       int deltaCount1 = 0;
1836       int deltaCount2 = 0;
1837       if (dLoc1 instanceof DeltaLocation) {
1838         deltaCount1 = ((DeltaLocation) dLoc1).getNumDelta();
1839       }
1840
1841       if (dLoc2 instanceof DeltaLocation) {
1842         deltaCount2 = ((DeltaLocation) dLoc2).getNumDelta();
1843       }
1844       if (deltaCount1 < deltaCount2) {
1845         return ComparisonResult.GREATER;
1846       } else if (deltaCount1 == deltaCount2) {
1847         return ComparisonResult.EQUAL;
1848       } else {
1849         return ComparisonResult.LESS;
1850       }
1851
1852     }
1853
1854     private static int compareBaseLocationSet(CompositeLocation compLoc1,
1855         CompositeLocation compLoc2, boolean awareSharedLoc, boolean ignore, String msg) {
1856
1857       // if compLoc1 is greater than compLoc2, return true
1858       // else return false;
1859
1860       // compare one by one in according to the order of the tuple
1861       int numOfTie = 0;
1862       for (int i = 0; i < compLoc1.getSize(); i++) {
1863         Location loc1 = compLoc1.get(i);
1864         if (i >= compLoc2.getSize()) {
1865           if (ignore) {
1866             return ComparisonResult.INCOMPARABLE;
1867           } else {
1868             throw new Error("Failed to compare two locations of " + compLoc1 + " and " + compLoc2
1869                 + " because they are not comparable at " + msg);
1870           }
1871         }
1872         Location loc2 = compLoc2.get(i);
1873
1874         Descriptor descriptor = getCommonParentDescriptor(loc1, loc2, msg);
1875         SSJavaLattice<String> lattice = getLatticeByDescriptor(descriptor);
1876
1877         // check if the shared location is appeared only at the end of the
1878         // composite location
1879         if (lattice.getSharedLocSet().contains(loc1.getLocIdentifier())) {
1880           if (i != (compLoc1.getSize() - 1)) {
1881             throw new Error("The shared location " + loc1.getLocIdentifier()
1882                 + " cannot be appeared in the middle of composite location at" + msg);
1883           }
1884         }
1885
1886         if (lattice.getSharedLocSet().contains(loc2.getLocIdentifier())) {
1887           if (i != (compLoc2.getSize() - 1)) {
1888             throw new Error("The shared location " + loc2.getLocIdentifier()
1889                 + " cannot be appeared in the middle of composite location at " + msg);
1890           }
1891         }
1892
1893         // if (!lattice1.equals(lattice2)) {
1894         // throw new Error("Failed to compare two locations of " + compLoc1 +
1895         // " and " + compLoc2
1896         // + " because they are not comparable at " + msg);
1897         // }
1898
1899         if (loc1.getLocIdentifier().equals(loc2.getLocIdentifier())) {
1900           numOfTie++;
1901           // check if the current location is the spinning location
1902           // note that the spinning location only can be appeared in the last
1903           // part of the composite location
1904           if (awareSharedLoc && numOfTie == compLoc1.getSize()
1905               && lattice.getSharedLocSet().contains(loc1.getLocIdentifier())) {
1906             return ComparisonResult.GREATER;
1907           }
1908           continue;
1909         } else if (lattice.isGreaterThan(loc1.getLocIdentifier(), loc2.getLocIdentifier())) {
1910           return ComparisonResult.GREATER;
1911         } else {
1912           return ComparisonResult.LESS;
1913         }
1914
1915       }
1916
1917       if (numOfTie == compLoc1.getSize()) {
1918
1919         if (numOfTie != compLoc2.getSize()) {
1920
1921           if (ignore) {
1922             return ComparisonResult.INCOMPARABLE;
1923           } else {
1924             throw new Error("Failed to compare two locations of " + compLoc1 + " and " + compLoc2
1925                 + " because they are not comparable at " + msg);
1926           }
1927
1928         }
1929
1930         return ComparisonResult.EQUAL;
1931       }
1932
1933       return ComparisonResult.LESS;
1934
1935     }
1936
1937     public static CompositeLocation calculateGLB(Set<CompositeLocation> inputSet, String errMsg) {
1938
1939       // System.out.println("Calculating GLB=" + inputSet);
1940       CompositeLocation glbCompLoc = new CompositeLocation();
1941
1942       // calculate GLB of the first(priority) element
1943       Set<String> priorityLocIdentifierSet = new HashSet<String>();
1944       Descriptor priorityDescriptor = null;
1945
1946       Hashtable<String, Set<CompositeLocation>> locId2CompLocSet =
1947           new Hashtable<String, Set<CompositeLocation>>();
1948       // mapping from the priority loc ID to its full representation by the
1949       // composite location
1950
1951       int maxTupleSize = 0;
1952       CompositeLocation maxCompLoc = null;
1953
1954       Location prevPriorityLoc = null;
1955       for (Iterator iterator = inputSet.iterator(); iterator.hasNext();) {
1956         CompositeLocation compLoc = (CompositeLocation) iterator.next();
1957         if (compLoc.getSize() > maxTupleSize) {
1958           maxTupleSize = compLoc.getSize();
1959           maxCompLoc = compLoc;
1960         }
1961         Location priorityLoc = compLoc.get(0);
1962         String priorityLocId = priorityLoc.getLocIdentifier();
1963         priorityLocIdentifierSet.add(priorityLocId);
1964
1965         if (locId2CompLocSet.containsKey(priorityLocId)) {
1966           locId2CompLocSet.get(priorityLocId).add(compLoc);
1967         } else {
1968           Set<CompositeLocation> newSet = new HashSet<CompositeLocation>();
1969           newSet.add(compLoc);
1970           locId2CompLocSet.put(priorityLocId, newSet);
1971         }
1972
1973         // check if priority location are coming from the same lattice
1974         if (priorityDescriptor == null) {
1975           priorityDescriptor = priorityLoc.getDescriptor();
1976         } else {
1977           priorityDescriptor = getCommonParentDescriptor(priorityLoc, prevPriorityLoc, errMsg);
1978         }
1979         prevPriorityLoc = priorityLoc;
1980         // else if (!priorityDescriptor.equals(priorityLoc.getDescriptor())) {
1981         // throw new Error("Failed to calculate GLB of " + inputSet
1982         // + " because they are from different lattices.");
1983         // }
1984       }
1985
1986       SSJavaLattice<String> locOrder = getLatticeByDescriptor(priorityDescriptor);
1987       String glbOfPriorityLoc = locOrder.getGLB(priorityLocIdentifierSet);
1988
1989       glbCompLoc.addLocation(new Location(priorityDescriptor, glbOfPriorityLoc));
1990       Set<CompositeLocation> compSet = locId2CompLocSet.get(glbOfPriorityLoc);
1991
1992       if (compSet == null) {
1993         // when GLB(x1,x2)!=x1 and !=x2 : GLB case 4
1994         // mean that the result is already lower than <x1,y1> and <x2,y2>
1995         // assign TOP to the rest of the location elements
1996
1997         // in this case, do not take care about delta
1998         // CompositeLocation inputComp = inputSet.iterator().next();
1999         for (int i = 1; i < maxTupleSize; i++) {
2000           glbCompLoc.addLocation(Location.createTopLocation(maxCompLoc.get(i).getDescriptor()));
2001         }
2002       } else {
2003
2004         // here find out composite location that has a maximum length tuple
2005         // if we have three input set: [A], [A,B], [A,B,C]
2006         // maximum length tuple will be [A,B,C]
2007         int max = 0;
2008         CompositeLocation maxFromCompSet = null;
2009         for (Iterator iterator = compSet.iterator(); iterator.hasNext();) {
2010           CompositeLocation c = (CompositeLocation) iterator.next();
2011           if (c.getSize() > max) {
2012             max = c.getSize();
2013             maxFromCompSet = c;
2014           }
2015         }
2016
2017         if (compSet.size() == 1) {
2018           // if GLB(x1,x2)==x1 or x2 : GLB case 2,3
2019           CompositeLocation comp = compSet.iterator().next();
2020           for (int i = 1; i < comp.getSize(); i++) {
2021             glbCompLoc.addLocation(comp.get(i));
2022           }
2023
2024           // if input location corresponding to glb is a delta, need to apply
2025           // delta to glb result
2026           if (comp instanceof DeltaLocation) {
2027             glbCompLoc = new DeltaLocation(glbCompLoc, 1);
2028           }
2029
2030         } else {
2031           // when GLB(x1,x2)==x1 and x2 : GLB case 1
2032           // if more than one location shares the same priority GLB
2033           // need to calculate the rest of GLB loc
2034
2035           // setup input set starting from the second tuple item
2036           Set<CompositeLocation> innerGLBInput = new HashSet<CompositeLocation>();
2037           for (Iterator iterator = compSet.iterator(); iterator.hasNext();) {
2038             CompositeLocation compLoc = (CompositeLocation) iterator.next();
2039             CompositeLocation innerCompLoc = new CompositeLocation();
2040             for (int idx = 1; idx < compLoc.getSize(); idx++) {
2041               innerCompLoc.addLocation(compLoc.get(idx));
2042             }
2043             if (innerCompLoc.getSize() > 0) {
2044               innerGLBInput.add(innerCompLoc);
2045             }
2046           }
2047
2048           if (innerGLBInput.size() > 0) {
2049             CompositeLocation innerGLB = CompositeLattice.calculateGLB(innerGLBInput, errMsg);
2050             for (int idx = 0; idx < innerGLB.getSize(); idx++) {
2051               glbCompLoc.addLocation(innerGLB.get(idx));
2052             }
2053           }
2054
2055           // if input location corresponding to glb is a delta, need to apply
2056           // delta to glb result
2057
2058           for (Iterator iterator = compSet.iterator(); iterator.hasNext();) {
2059             CompositeLocation compLoc = (CompositeLocation) iterator.next();
2060             if (compLoc instanceof DeltaLocation) {
2061               if (glbCompLoc.equals(compLoc)) {
2062                 glbCompLoc = new DeltaLocation(glbCompLoc, 1);
2063                 break;
2064               }
2065             }
2066           }
2067
2068         }
2069       }
2070
2071       // System.out.println("GLB=" + glbCompLoc);
2072       return glbCompLoc;
2073
2074     }
2075
2076     static SSJavaLattice<String> getLatticeByDescriptor(Descriptor d) {
2077
2078       SSJavaLattice<String> lattice = null;
2079
2080       if (d instanceof ClassDescriptor) {
2081         lattice = ssjava.getCd2lattice().get(d);
2082       } else if (d instanceof MethodDescriptor) {
2083         if (ssjava.getMd2lattice().containsKey(d)) {
2084           lattice = ssjava.getMd2lattice().get(d);
2085         } else {
2086           // use default lattice for the method
2087           lattice = ssjava.getCd2methodDefault().get(((MethodDescriptor) d).getClassDesc());
2088         }
2089       }
2090
2091       return lattice;
2092     }
2093
2094     static Descriptor getCommonParentDescriptor(Location loc1, Location loc2, String msg) {
2095
2096       Descriptor d1 = loc1.getDescriptor();
2097       Descriptor d2 = loc2.getDescriptor();
2098
2099       Descriptor descriptor;
2100
2101       if (d1 instanceof ClassDescriptor && d2 instanceof ClassDescriptor) {
2102
2103         if (d1.equals(d2)) {
2104           descriptor = d1;
2105         } else {
2106           // identifying which one is parent class
2107           Set<Descriptor> d1SubClassesSet = ssjava.tu.getSubClasses((ClassDescriptor) d1);
2108           Set<Descriptor> d2SubClassesSet = ssjava.tu.getSubClasses((ClassDescriptor) d2);
2109
2110           if (d1 == null && d2 == null) {
2111             throw new Error("Failed to compare two locations of " + loc1 + " and " + loc2
2112                 + " because they are not comparable at " + msg);
2113           } else if (d1SubClassesSet != null && d1SubClassesSet.contains(d2)) {
2114             descriptor = d1;
2115           } else if (d2SubClassesSet != null && d2SubClassesSet.contains(d1)) {
2116             descriptor = d2;
2117           } else {
2118             throw new Error("Failed to compare two locations of " + loc1 + " and " + loc2
2119                 + " because they are not comparable at " + msg);
2120           }
2121         }
2122
2123       } else if (d1 instanceof MethodDescriptor && d2 instanceof MethodDescriptor) {
2124
2125         if (d1.equals(d2)) {
2126           descriptor = d1;
2127         } else {
2128
2129           // identifying which one is parent class
2130           MethodDescriptor md1 = (MethodDescriptor) d1;
2131           MethodDescriptor md2 = (MethodDescriptor) d2;
2132
2133           if (!md1.matches(md2)) {
2134             throw new Error("Failed to compare two locations of " + loc1 + " and " + loc2
2135                 + " because they are not comparable at " + msg);
2136           }
2137
2138           Set<Descriptor> d1SubClassesSet =
2139               ssjava.tu.getSubClasses(((MethodDescriptor) d1).getClassDesc());
2140           Set<Descriptor> d2SubClassesSet =
2141               ssjava.tu.getSubClasses(((MethodDescriptor) d2).getClassDesc());
2142
2143           if (d1 == null && d2 == null) {
2144             throw new Error("Failed to compare two locations of " + loc1 + " and " + loc2
2145                 + " because they are not comparable at " + msg);
2146           } else if (d1 != null && d1SubClassesSet.contains(d2)) {
2147             descriptor = d1;
2148           } else if (d2 != null && d2SubClassesSet.contains(d1)) {
2149             descriptor = d2;
2150           } else {
2151             throw new Error("Failed to compare two locations of " + loc1 + " and " + loc2
2152                 + " because they are not comparable at " + msg);
2153           }
2154         }
2155
2156       } else {
2157         throw new Error("Failed to compare two locations of " + loc1 + " and " + loc2
2158             + " because they are not comparable at " + msg);
2159       }
2160
2161       return descriptor;
2162
2163     }
2164
2165   }
2166
2167   class ComparisonResult {
2168
2169     public static final int GREATER = 0;
2170     public static final int EQUAL = 1;
2171     public static final int LESS = 2;
2172     public static final int INCOMPARABLE = 3;
2173     int result;
2174
2175   }
2176
2177 }
2178
2179 class ReturnLocGenerator {
2180
2181   public static final int PARAMISHIGHER = 0;
2182   public static final int PARAMISSAME = 1;
2183   public static final int IGNORE = 2;
2184
2185   private Hashtable<Integer, Integer> paramIdx2paramType;
2186
2187   private CompositeLocation declaredReturnLoc = null;
2188
2189   public ReturnLocGenerator(CompositeLocation returnLoc, MethodDescriptor md,
2190       List<CompositeLocation> params, String msg) {
2191
2192     CompositeLocation thisLoc = params.get(0);
2193     if (returnLoc.get(0).equals(thisLoc.get(0)) && returnLoc.getSize() > 1) {
2194       // if the declared return location consists of THIS and field location,
2195       // return location for the caller's side has to have same field element
2196       this.declaredReturnLoc = returnLoc;
2197     } else {
2198       // creating mappings
2199       paramIdx2paramType = new Hashtable<Integer, Integer>();
2200       for (int i = 0; i < params.size(); i++) {
2201         CompositeLocation param = params.get(i);
2202         int compareResult = CompositeLattice.compare(param, returnLoc, true, msg);
2203
2204         int type;
2205         if (compareResult == ComparisonResult.GREATER) {
2206           type = 0;
2207         } else if (compareResult == ComparisonResult.EQUAL) {
2208           type = 1;
2209         } else {
2210           type = 2;
2211         }
2212         paramIdx2paramType.put(new Integer(i), new Integer(type));
2213       }
2214     }
2215
2216   }
2217
2218   public CompositeLocation computeReturnLocation(List<CompositeLocation> args) {
2219
2220     if (declaredReturnLoc != null) {
2221       // when developer specify that the return value is [THIS,field]
2222       // needs to translate to the caller's location
2223       CompositeLocation callerLoc = new CompositeLocation();
2224       CompositeLocation callerBaseLocation = args.get(0);
2225
2226       for (int i = 0; i < callerBaseLocation.getSize(); i++) {
2227         callerLoc.addLocation(callerBaseLocation.get(i));
2228       }
2229       for (int i = 1; i < declaredReturnLoc.getSize(); i++) {
2230         callerLoc.addLocation(declaredReturnLoc.get(i));
2231       }
2232       return callerLoc;
2233     } else {
2234       // compute the highest possible location in caller's side
2235       assert paramIdx2paramType.keySet().size() == args.size();
2236
2237       Set<CompositeLocation> inputGLB = new HashSet<CompositeLocation>();
2238       for (int i = 0; i < args.size(); i++) {
2239         int type = (paramIdx2paramType.get(new Integer(i))).intValue();
2240         CompositeLocation argLoc = args.get(i);
2241         if (type == PARAMISHIGHER || type == PARAMISSAME) {
2242           // return loc is equal to or lower than param
2243           inputGLB.add(argLoc);
2244         }
2245       }
2246
2247       // compute GLB of arguments subset that are same or higher than return
2248       // location
2249       if (inputGLB.isEmpty()) {
2250         CompositeLocation rtr =
2251             new CompositeLocation(Location.createTopLocation(args.get(0).get(0).getDescriptor()));
2252         return rtr;
2253       } else {
2254         CompositeLocation glb = CompositeLattice.calculateGLB(inputGLB, "");
2255         return glb;
2256       }
2257     }
2258
2259   }
2260 }