Minor improvements

Author: MarcMil

src/main/java/soot/dotnet/members/ByReferenceWrapperGenerator.java

@@ -50,6 +50,7 @@
  * @author Marc Miltenberger
  */
 public class ByReferenceWrapperGenerator {
+  public static final String WRAPPER_FIELD_NAME = "r";
   public static final String WRAPPER_CLASS_NAME = "ByReferenceWrappers.Wrapper";
 
   public synchronized static SootClass getWrapperClass(Type t) {
@@ -61,7 +62,7 @@ public synchronized static SootClass getWrapperClass(Type t) {
     }
     SootClass sc = scene.makeSootClass(name, Modifier.FINAL | Modifier.STATIC);
     sc.setApplicationClass();
-    SootField r = scene.makeSootField("r", RefType.v("System.Object"));
+    SootField r = scene.makeSootField(WRAPPER_FIELD_NAME, RefType.v("System.Object"));
     r.setModifiers(Modifier.PUBLIC);
     sc.addField(r);
 
@@ -120,7 +121,7 @@ public static Unit getUnwrapCall(SootClass wrapperClass, Value argToUnwrap, Valu
   }
 
   public static SootField getWrapperField(SootClass wrapperClass) {
-    return wrapperClass.getFieldByName("r");
+    return wrapperClass.getFieldByName(WRAPPER_FIELD_NAME);
   }
 
   /**
@@ -134,7 +135,7 @@ public static SootField getWrapperField(SootClass wrapperClass) {
    */
   public static Unit getUpdateWrappedValueCall(Local wrapped, Local unwrapped) {
     RefType rt = (RefType) wrapped.getType();
-    SootField f = rt.getSootClass().getFieldByName("r");
+    SootField f = rt.getSootClass().getFieldByName(WRAPPER_FIELD_NAME);
     Jimple j = Jimple.v();
     return j.newAssignStmt(j.newInstanceFieldRef(wrapped, f.makeRef()), unwrapped);
 

src/main/java/soot/jimple/spark/solver/PropWorklist.java

@@ -56,7 +56,7 @@
 
 public class PropWorklist extends Propagator {
   private static final Logger logger = LoggerFactory.getLogger(PropWorklist.class);
-  protected final Set<VarNode> varNodeWorkList = new TreeSet<VarNode>();
+  protected final TreeSet<VarNode> varNodeWorkList = new TreeSet<VarNode>();
 
   public PropWorklist(PAG pag) {
     this.pag = pag;
@@ -75,10 +75,9 @@ public void propagate() {
       if (verbose) {
         logger.debug("Worklist has " + varNodeWorkList.size() + " nodes.");
       }
-      while (!varNodeWorkList.isEmpty()) {
-        VarNode src = varNodeWorkList.iterator().next();
-        varNodeWorkList.remove(src);
-        handleVarNode(src);
+      VarNode vsrc;
+      while ((vsrc = varNodeWorkList.pollFirst()) != null) {
+        handleVarNode(vsrc);
       }
       if (verbose) {
         logger.debug("Now handling field references");

src/main/java/soot/jimple/toolkits/callgraph/ReachableMethods.java

@@ -25,7 +25,6 @@
 import java.util.Collection;
 import java.util.HashSet;
 import java.util.Iterator;
-import java.util.NoSuchElementException;
 import java.util.Set;
 
 import soot.MethodOrMethodContext;
@@ -91,13 +90,15 @@ public void update() {
     while (unprocessedMethods.hasNext()) {
       MethodOrMethodContext m = unprocessedMethods.next();
       if (m == null) {
-    	continue;
+        continue;
       }
       Iterator<Edge> targets = cg.edgesOutOf(m);
       if (filter != null) {
         targets = filter.wrap(targets);
       }
-      addMethods(new Targets(targets));
+      if (targets.hasNext()) {
+        addMethods(new Targets(targets));
+      }
     }
   }