+throw exception

Rroscha · May 22, 2014 · f4fb967 · f4fb967
1 parent 0e99855
commit f4fb967
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/src/soot/toolkits/scalar/SmartLocalDefs.java b/src/soot/toolkits/scalar/SmartLocalDefs.java
@@ -65,22 +65,24 @@ protected StaticSingleAssignment(List<Unit>[] units, Local[] locals) {
 
 		@Override
 		public List<Unit> getDefsOfAt(Local l, Unit s) {
-			List<Unit> r = localToDefs.get(l);
-			if (r == null) {
-				r = emptyList();
+			for (ValueBox useBox : s.getUseBoxes()) {
+				if (l == useBox.getValue()) {
+					return getDefsOfAt(useBox);
+				}
 			}
-			return r;
+			throw new RuntimeException();
 		}
 
 		public List<Unit> getDefsOfAt(ValueBox valueBox) {
 			Value v = valueBox.getValue();
 			if (v instanceof Local) {
 				List<Unit> r = localToDefs.get(v);
-				if (r != null) {
-					return r;
+				if (r == null) {
+					return emptyList();
 				}
+				return r;
 			}
-			return emptyList();
+			throw new RuntimeException();
 		}
 	}
 
@@ -141,13 +143,17 @@ public List<Unit> getDefsOfAt(Local l, Unit s) {
 					return getDefsOfAt(useBox);
 				}
 			}
-			return emptyList();
+			throw new RuntimeException();
 		}
-
+		
 		public List<Unit> getDefsOfAt(ValueBox valueBox) {
 			List<Unit> r = resultValueBoxes.get(valueBox);
 			if (r == null) {
-				r = emptyList();
+				Value v = valueBox.getValue();
+				if (v instanceof Local) {
+					return emptyList();
+				}
+				throw new RuntimeException();	
 			}
 			return r;
 		}