Performance

soot-oss · Oct 21, 2024 · 987955d · 987955d
1 parent 3299680
commit 987955d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/main/java/soot/dexpler/DexBody.java b/src/main/java/soot/dexpler/DexBody.java
@@ -881,6 +881,15 @@ public Type promote(Type tlow, Type thigh) {
 
       }
 
+      protected Type getDefiniteType(Local v) {
+        Collection<Type> r = definiteConstraints.get(v);
+        if (r != null && r.size() == 1) {
+          return r.iterator().next();
+        } else {
+          return null;
+        }
+      }
+
       protected soot.jimple.toolkits.typing.fast.BytecodeHierarchy createBytecodeHierarchy() {
         return new soot.jimple.toolkits.typing.fast.BytecodeHierarchy() {
           public java.util.Collection<Type> lcas(Type a, Type b, boolean useWeakObjectType) {

diff --git a/src/main/java/soot/jimple/toolkits/typing/fast/TypeResolver.java b/src/main/java/soot/jimple/toolkits/typing/fast/TypeResolver.java
@@ -565,6 +565,14 @@ protected Collection<Typing> applyAssignmentConstraints(Typing tg, IEvalFunction
         Value lhs = stmt.getLeftOp();
         if (lhs instanceof Local) {
           Local v = (Local) lhs;
+          Type t = getDefiniteType(v);
+          if (t != null) {
+            simple.set(i);
+            wl.clear(i);
+            tg.set(v, t);
+            continue;
+          }
+
           if (singleAssignments.contains(v)) {
             Collection<Type> d = ef.eval(tg, stmt.getRightOp(), stmt);
             if (d.size() == 1) {
@@ -699,6 +707,10 @@ protected Collection<Typing> applyAssignmentConstraints(Typing tg, IEvalFunction
     return r;
   }
 
+  protected Type getDefiniteType(Local v) {
+    return null;
+  }
+
   protected Collection<Type> reduceToAllowedTypesForLocal(Collection<Type> lcas, Local v) {
     return lcas;
   }