package Analysis.SSJava;
+import java.io.BufferedWriter;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Comparator;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.Iterator;
+import java.util.List;
import java.util.Set;
import java.util.StringTokenizer;
import java.util.Vector;
import Analysis.CallGraph.CallGraph;
+import Analysis.Loops.GlobalFieldType;
import Analysis.Loops.LoopOptimize;
import Analysis.Loops.LoopTerminate;
import IR.AnnotationDescriptor;
import IR.ClassDescriptor;
+import IR.Descriptor;
+import IR.FieldDescriptor;
import IR.MethodDescriptor;
import IR.State;
+import IR.SymbolTable;
import IR.TypeUtil;
+import IR.Flat.BuildFlat;
import IR.Flat.FlatMethod;
+import IR.Flat.FlatNode;
+import Util.Pair;
public class SSJavaAnalysis {
public static final String LOC = "LOC";
public static final String DELTA = "DELTA";
public static final String TERMINATE = "TERMINATE";
+ public static final String DELEGATE = "DELEGATE";
+ public static final String DELEGATETHIS = "DELEGATETHIS";
+ public static final String TRUST = "TRUST";
State state;
TypeUtil tu;
FlowDownCheck flowDownChecker;
MethodAnnotationCheck methodAnnotationChecker;
+ BuildFlat bf;
- // if a method has annotations, the mapping has true
+ // set containing method requires to be annoated
Set<MethodDescriptor> annotationRequireSet;
// class -> field lattice
// method -> local variable lattice
Hashtable<MethodDescriptor, MethodLattice<String>> md2lattice;
- // method set that does not have loop termination analysis
+ // method set that does not want to have loop termination analysis
Hashtable<MethodDescriptor, Integer> skipLoopTerminate;
+ // map shared location to its descriptors
+ Hashtable<Location, Set<Descriptor>> mapSharedLocation2DescriptorSet;
+
+ // set containing a class that has at least one annoated method
+ Set<ClassDescriptor> annotationRequireClassSet;
+
+ // the set of method descriptor required to check the linear type property
+ Set<MethodDescriptor> linearTypeCheckMethodSet;
+
+ // the set of method descriptors annotated as "TRUST"
+ Set<MethodDescriptor> trustWorthyMDSet;
+
+ // points to method containing SSJAVA Loop
+ private MethodDescriptor methodContainingSSJavaLoop;
+
+ private FlatNode ssjavaLoopEntrance;
+
+ // keep the field ownership from the linear type checking
+ Hashtable<MethodDescriptor, Set<FieldDescriptor>> mapMethodToOwnedFieldSet;
+
+ Set<FlatNode> sameHeightWriteFlatNodeSet;
+
CallGraph callgraph;
- public SSJavaAnalysis(State state, TypeUtil tu, CallGraph callgraph) {
+ LinearTypeCheck checker;
+
+ public SSJavaAnalysis(State state, TypeUtil tu, BuildFlat bf, CallGraph callgraph) {
this.state = state;
this.tu = tu;
this.callgraph = callgraph;
this.cd2methodDefault = new Hashtable<ClassDescriptor, MethodLattice<String>>();
this.md2lattice = new Hashtable<MethodDescriptor, MethodLattice<String>>();
this.annotationRequireSet = new HashSet<MethodDescriptor>();
+ this.annotationRequireClassSet = new HashSet<ClassDescriptor>();
this.skipLoopTerminate = new Hashtable<MethodDescriptor, Integer>();
+ this.mapSharedLocation2DescriptorSet = new Hashtable<Location, Set<Descriptor>>();
+ this.linearTypeCheckMethodSet = new HashSet<MethodDescriptor>();
+ this.bf = bf;
+ this.trustWorthyMDSet = new HashSet<MethodDescriptor>();
+ this.mapMethodToOwnedFieldSet = new Hashtable<MethodDescriptor, Set<FieldDescriptor>>();
+ this.sameHeightWriteFlatNodeSet = new HashSet<FlatNode>();
}
public void doCheck() {
doMethodAnnotationCheck();
- if (state.SSJAVADEBUG) {
- debugPrint();
+ computeLinearTypeCheckMethodSet();
+ doLinearTypeCheck();
+ // if (state.SSJAVADEBUG) {
+ // debugPrint();
+ // }
+ if (state.SSJAVAINFER) {
+ inference();
+ } else {
+ parseLocationAnnotation();
+ doFlowDownCheck();
+ doDefinitelyWrittenCheck();
+ doLoopCheck();
+ }
+ }
+
+ private void inference() {
+ LocationInference inferEngine = new LocationInference(this, state);
+ inferEngine.inference();
+ }
+
+ private void doLoopCheck() {
+ GlobalFieldType gft = new GlobalFieldType(callgraph, state, tu.getMain());
+ LoopOptimize lo = new LoopOptimize(gft, tu);
+
+ SymbolTable classtable = state.getClassSymbolTable();
+
+ List<ClassDescriptor> toanalyzeList = new ArrayList<ClassDescriptor>();
+ List<MethodDescriptor> toanalyzeMethodList = new ArrayList<MethodDescriptor>();
+
+ toanalyzeList.addAll(classtable.getValueSet());
+ Collections.sort(toanalyzeList, new Comparator<ClassDescriptor>() {
+ public int compare(ClassDescriptor o1, ClassDescriptor o2) {
+ return o1.getClassName().compareTo(o2.getClassName());
+ }
+ });
+
+ for (int i = 0; i < toanalyzeList.size(); i++) {
+ ClassDescriptor cd = toanalyzeList.get(i);
+
+ SymbolTable methodtable = cd.getMethodTable();
+ toanalyzeMethodList.clear();
+ toanalyzeMethodList.addAll(methodtable.getValueSet());
+ Collections.sort(toanalyzeMethodList, new Comparator<MethodDescriptor>() {
+ public int compare(MethodDescriptor o1, MethodDescriptor o2) {
+ return o1.getSymbol().compareTo(o2.getSymbol());
+ }
+ });
+
+ for (int mdIdx = 0; mdIdx < toanalyzeMethodList.size(); mdIdx++) {
+ MethodDescriptor md = toanalyzeMethodList.get(mdIdx);
+ if (needTobeAnnotated(md)) {
+ lo.analyze(state.getMethodFlat(md));
+ doLoopTerminationCheck(lo, state.getMethodFlat(md));
+ }
+ }
+
}
- parseLocationAnnotation();
- doFlowDownCheck();
- doDefinitelyWrittenCheck();
- doSingleReferenceCheck();
+
+ }
+
+ public void addTrustMethod(MethodDescriptor md) {
+ trustWorthyMDSet.add(md);
+ }
+
+ public boolean isTrustMethod(MethodDescriptor md) {
+ return trustWorthyMDSet.contains(md);
+ }
+
+ private void computeLinearTypeCheckMethodSet() {
+
+ Set<MethodDescriptor> allCalledSet = callgraph.getMethodCalls(tu.getMain());
+ linearTypeCheckMethodSet.addAll(allCalledSet);
+
+ Set<MethodDescriptor> trustedSet = new HashSet<MethodDescriptor>();
+
+ for (Iterator iterator = trustWorthyMDSet.iterator(); iterator.hasNext();) {
+ MethodDescriptor trustMethod = (MethodDescriptor) iterator.next();
+ Set<MethodDescriptor> calledFromTrustMethodSet = callgraph.getMethodCalls(trustMethod);
+ trustedSet.add(trustMethod);
+ trustedSet.addAll(calledFromTrustMethodSet);
+ }
+
+ linearTypeCheckMethodSet.removeAll(trustedSet);
+
+ // if a method is called only by trusted method, no need to check linear
+ // type & flow down rule
+ for (Iterator iterator = trustedSet.iterator(); iterator.hasNext();) {
+ MethodDescriptor md = (MethodDescriptor) iterator.next();
+ Set<MethodDescriptor> callerSet = callgraph.getCallerSet(md);
+ if (!trustedSet.containsAll(callerSet) && !trustWorthyMDSet.contains(md)) {
+ linearTypeCheckMethodSet.add(md);
+ }
+ }
+
+ }
+
+ private void doLinearTypeCheck() {
+ LinearTypeCheck checker = new LinearTypeCheck(this, state);
+ checker.linearTypeCheck();
}
public void debugPrint() {
System.out.println("SSJAVA: SSJava is checking the following methods:");
for (Iterator<MethodDescriptor> iterator = annotationRequireSet.iterator(); iterator.hasNext();) {
MethodDescriptor md = iterator.next();
- System.out.println("SSJAVA: " + md);
+ System.out.print(" " + md);
}
+ System.out.println();
}
private void doMethodAnnotationCheck() {
checker.definitelyWrittenCheck();
}
- public void doSingleReferenceCheck() {
- SingleReferenceCheck checker = new SingleReferenceCheck(this, state);
- checker.singleReferenceCheck();
- }
-
private void parseLocationAnnotation() {
Iterator it = state.getClassSymbolTable().getDescriptorsIterator();
while (it.hasNext()) {
new SSJavaLattice<String>(SSJavaLattice.TOP, SSJavaLattice.BOTTOM);
cd2lattice.put(cd, locOrder);
parseClassLatticeDefinition(cd, an.getValue(), locOrder);
+
+ if (state.SSJAVADEBUG) {
+ // generate lattice dot file
+ writeLatticeDotFile(cd, locOrder);
+ }
+
} else if (marker.equals(METHODDEFAULT)) {
MethodLattice<String> locOrder =
new MethodLattice<String>(SSJavaLattice.TOP, SSJavaLattice.BOTTOM);
cd2methodDefault.put(cd, locOrder);
- parseMethodLatticeDefinition(cd, an.getValue(), locOrder);
+ parseMethodDefaultLatticeDefinition(cd, an.getValue(), locOrder);
}
}
MethodLattice<String> locOrder =
new MethodLattice<String>(SSJavaLattice.TOP, SSJavaLattice.BOTTOM);
md2lattice.put(md, locOrder);
- parseMethodLatticeDefinition(cd, an.getValue(), locOrder);
+ parseMethodDefaultLatticeDefinition(cd, an.getValue(), locOrder);
} else if (an.getMarker().equals(TERMINATE)) {
// developer explicitly wants to skip loop termination analysis
String value = an.getValue();
}
}
- private void parseMethodLatticeDefinition(ClassDescriptor cd, String value,
+ private void writeLatticeDotFile(ClassDescriptor cd, SSJavaLattice<String> locOrder) {
+
+ String className = cd.getSymbol().replaceAll("[\\W_]", "");
+
+ Set<Pair<String, String>> pairSet = locOrder.getOrderingPairSet();
+
+ try {
+ BufferedWriter bw = new BufferedWriter(new FileWriter(className + ".dot"));
+
+ bw.write("digraph " + className + " {\n");
+
+ for (Iterator iterator = pairSet.iterator(); iterator.hasNext();) {
+ // pair is in the form of <higher, lower>
+ Pair<String, String> pair = (Pair<String, String>) iterator.next();
+
+ String highLocId = pair.getFirst();
+ if (locOrder.isSharedLoc(highLocId)) {
+ highLocId = "\"" + highLocId + "*\"";
+ }
+ String lowLocId = pair.getSecond();
+ if (locOrder.isSharedLoc(lowLocId)) {
+ lowLocId = "\"" + lowLocId + "*\"";
+ }
+ bw.write(highLocId + " -> " + lowLocId + ";\n");
+ }
+ bw.write("}\n");
+ bw.close();
+
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+
+ }
+
+ private void parseMethodDefaultLatticeDefinition(ClassDescriptor cd, String value,
MethodLattice<String> locOrder) {
value = value.replaceAll(" ", ""); // remove all blank spaces
} else if (orderElement.startsWith(GLOBALLOC + "=")) {
String globalLoc = orderElement.substring(10);
locOrder.setGlobalLoc(globalLoc);
- } else if (orderElement.contains("*")) {
+ } else if (orderElement.startsWith(RETURNLOC + "=")) {
+ String returnLoc = orderElement.substring(10);
+ locOrder.setReturnLoc(returnLoc);
+ } else if (orderElement.endsWith("*")) {
// spin loc definition
- locOrder.addSpinLoc(orderElement.substring(0, orderElement.length() - 1));
+ locOrder.addSharedLoc(orderElement.substring(0, orderElement.length() - 1));
} else {
// single element
locOrder.put(orderElement);
// sanity checks
if (locOrder.getThisLoc() != null && !locOrder.containsKey(locOrder.getThisLoc())) {
throw new Error("Variable 'this' location '" + locOrder.getThisLoc()
- + "' is not defined in the default local variable lattice at " + cd.getSourceFileName());
+ + "' is not defined in the local variable lattice at " + cd.getSourceFileName());
}
if (locOrder.getGlobalLoc() != null && !locOrder.containsKey(locOrder.getGlobalLoc())) {
throw new Error("Variable global location '" + locOrder.getGlobalLoc()
- + "' is not defined in the default local variable lattice at " + cd.getSourceFileName());
+ + "' is not defined in the local variable lattice at " + cd.getSourceFileName());
}
}
}
} else if (orderElement.contains("*")) {
// spin loc definition
- locOrder.addSpinLoc(orderElement.substring(0, orderElement.length() - 1));
+ locOrder.addSharedLoc(orderElement.substring(0, orderElement.length() - 1));
} else {
// single element
locOrder.put(orderElement);
}
// sanity check
- Set<String> spinLocSet = locOrder.getSpinLocSet();
+ Set<String> spinLocSet = locOrder.getSharedLocSet();
for (Iterator iterator = spinLocSet.iterator(); iterator.hasNext();) {
String spinLoc = (String) iterator.next();
if (!locOrder.containsKey(spinLoc)) {
return cd2lattice.get(cd);
}
+ public MethodLattice<String> getMethodDefaultLattice(ClassDescriptor cd) {
+ return cd2methodDefault.get(cd);
+ }
+
public MethodLattice<String> getMethodLattice(MethodDescriptor md) {
if (md2lattice.containsKey(md)) {
return md2lattice.get(md);
} else {
- return cd2methodDefault.get(md.getClassDesc());
+
+ if (cd2methodDefault.containsKey(md.getClassDesc())) {
+ return cd2methodDefault.get(md.getClassDesc());
+ } else {
+ throw new Error("Method Lattice of " + md + " is not defined.");
+ }
+
}
}
+ public boolean needToCheckLinearType(MethodDescriptor md) {
+ return linearTypeCheckMethodSet.contains(md);
+ }
+
public boolean needTobeAnnotated(MethodDescriptor md) {
return annotationRequireSet.contains(md);
}
+ public boolean needToBeAnnoated(ClassDescriptor cd) {
+ return annotationRequireClassSet.contains(cd);
+ }
+
+ public void addAnnotationRequire(ClassDescriptor cd) {
+ annotationRequireClassSet.add(cd);
+ }
+
public void addAnnotationRequire(MethodDescriptor md) {
- annotationRequireSet.add(md);
+
+ ClassDescriptor cd = md.getClassDesc();
+ // if a method requires to be annotated, class containg that method also
+ // requires to be annotated
+ if (!isSSJavaUtil(cd)) {
+ annotationRequireClassSet.add(cd);
+ annotationRequireSet.add(md);
+ }
}
public Set<MethodDescriptor> getAnnotationRequireSet() {
}
public void doLoopTerminationCheck(LoopOptimize lo, FlatMethod fm) {
- LoopTerminate lt = new LoopTerminate();
- lt.terminateAnalysis(fm, lo.getLoopInvariant(fm));
+ LoopTerminate lt = new LoopTerminate(this, state);
+ if (needTobeAnnotated(fm.getMethod())) {
+ lt.terminateAnalysis(fm, lo.getLoopInvariant(fm));
+ }
}
- public void doLoopTerminationCheck(LoopOptimize lo) {
- LoopTerminate lt = new LoopTerminate();
- for (Iterator iterator = annotationRequireSet.iterator(); iterator.hasNext();) {
- MethodDescriptor md = (MethodDescriptor) iterator.next();
- if (!skipLoopTerminate.containsKey(md)) {
- FlatMethod fm = state.getMethodFlat(md);
- lt.terminateAnalysis(fm, lo.getLoopInvariant(fm));
- }
+ public CallGraph getCallGraph() {
+ return callgraph;
+ }
+
+ public SSJavaLattice<String> getLattice(Descriptor d) {
+
+ if (d instanceof MethodDescriptor) {
+ return getMethodLattice((MethodDescriptor) d);
+ } else {
+ return getClassLattice((ClassDescriptor) d);
}
}
- public CallGraph getCallGraph() {
- return callgraph;
+ public boolean isSharedLocation(Location loc) {
+ SSJavaLattice<String> lattice = getLattice(loc.getDescriptor());
+ return lattice.getSharedLocSet().contains(loc.getLocIdentifier());
+ }
+
+ public void mapSharedLocation2Descriptor(Location loc, Descriptor d) {
+ Set<Descriptor> set = mapSharedLocation2DescriptorSet.get(loc);
+ if (set == null) {
+ set = new HashSet<Descriptor>();
+ mapSharedLocation2DescriptorSet.put(loc, set);
+ }
+ set.add(d);
+ }
+
+ public BuildFlat getBuildFlat() {
+ return bf;
+ }
+
+ public MethodDescriptor getMethodContainingSSJavaLoop() {
+ return methodContainingSSJavaLoop;
+ }
+
+ public void setMethodContainingSSJavaLoop(MethodDescriptor methodContainingSSJavaLoop) {
+ this.methodContainingSSJavaLoop = methodContainingSSJavaLoop;
+ }
+
+ public boolean isSSJavaUtil(ClassDescriptor cd) {
+ if (cd.getSymbol().equals("SSJAVA")) {
+ return true;
+ }
+ return false;
+ }
+
+ public void setFieldOnwership(MethodDescriptor md, FieldDescriptor field) {
+
+ Set<FieldDescriptor> fieldSet = mapMethodToOwnedFieldSet.get(md);
+ if (fieldSet == null) {
+ fieldSet = new HashSet<FieldDescriptor>();
+ mapMethodToOwnedFieldSet.put(md, fieldSet);
+ }
+ fieldSet.add(field);
+ }
+
+ public boolean isOwnedByMethod(MethodDescriptor md, FieldDescriptor field) {
+ Set<FieldDescriptor> fieldSet = mapMethodToOwnedFieldSet.get(md);
+ if (fieldSet != null) {
+ return fieldSet.contains(field);
+ }
+ return false;
+ }
+
+ public FlatNode getSSJavaLoopEntrance() {
+ return ssjavaLoopEntrance;
+ }
+
+ public void setSSJavaLoopEntrance(FlatNode ssjavaLoopEntrance) {
+ this.ssjavaLoopEntrance = ssjavaLoopEntrance;
+ }
+
+ public void addSameHeightWriteFlatNode(FlatNode fn) {
+ this.sameHeightWriteFlatNodeSet.add(fn);
+ }
+
+ public boolean isSameHeightWrite(FlatNode fn) {
+ return this.sameHeightWriteFlatNodeSet.contains(fn);
}
}