Svelte/src/edu/berkeley/cs/bodik/svelte/experimental/autodebug/InstrumentationSet.java

   1 package edu.berkeley.cs.bodik.svelte.experimental.autodebug;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collection;
   5 import java.util.HashSet;
   6 import java.util.Hashtable;
   7
   8 import com.ibm.wala.ipa.slicer.NormalStatement;
   9 import com.ibm.wala.ipa.slicer.ParamCaller;
  10 import com.ibm.wala.ipa.slicer.Statement;
  11 import com.ibm.wala.ssa.SSAInstruction;
  12 import com.ibm.wala.ssa.SSAPutInstruction;
  13 import com.ibm.wala.ssa.SSAReturnInstruction;
  14
  15 public class InstrumentationSet {
  16         HashSet<String> fullyQualifiedClassNames;
  17         Hashtable<String,ArrayList<InstrumentationPoint>> methodsToIPs;
  18         public InstrumentationSet() {
  19                 fullyQualifiedClassNames = new HashSet<String>();
  20                 methodsToIPs = new Hashtable<String,ArrayList<InstrumentationPoint>>();
  21         }
  22
  23         private void addInstrumentationPoint(String methodSignature,InstrumentationPoint ip) {
  24                 methodSignature = methodSignature.replace("/",".");
  25                 ArrayList<InstrumentationPoint> methodsIPs = methodsToIPs.get(methodSignature);
  26                 if ( methodsIPs == null ) {
  27                         methodsIPs = new ArrayList<InstrumentationPoint>();
  28                         methodsToIPs.put(methodSignature, methodsIPs);
  29                 }
  30                 methodsIPs.add(ip);
  31         }
  32
  33         public boolean toBeInstrumented(String methodSignature) {
  34                 return methodsToIPs.containsKey(methodSignature.replace("/","."));
  35         }
  36
  37         public ArrayList<InstrumentationPoint> instrumentationPointsForMethod(String methodSignature) {
  38                 return methodsToIPs.get(methodSignature.replace("/","."));
  39         }
  40
  41         public void addInstrumentationPoint(Statement statement) {
  42                 InstrumentationPoint newIP = null;
  43                 if ( statement instanceof NormalStatement ) {
  44                         SSAInstruction instr = ((NormalStatement)statement).getInstruction();
  45                         if ( instr instanceof SSAPutInstruction || instr instanceof SSAReturnInstruction )
  46                                 newIP = new InstrumentationPoint.PrintTopOfStackIP(statement);
  47
  48                 } else if ( statement instanceof ParamCaller ) {
  49                         newIP = new InstrumentationPoint.ParamCallerIP(statement);
  50                 }
  51                 if ( newIP != null ) {
  52                         addInstrumentationPoint(statement.getNode().getMethod().getSignature(), newIP);
  53                         String classname = statement.getNode().getMethod().getDeclaringClass().getName().toString();
  54                         fullyQualifiedClassNames.add(classname);
  55                 }
  56         }
  57
  58         public Collection<String> getFullyQualifiedClassNames() {
  59                 return fullyQualifiedClassNames;
  60         }
  61 }