Slightly (sort of) improve proxy objects tracking.

[cycript.git] / Java / Cycript.java

diff --git a/Java/Cycript.java b/Java/Cycript.java
index 79ebe8e29109e79b0b9e6b885da24155a7ee8bea..8fb7817f14e4c2e309ceaf2f07c46fadb61fe12a 100644 (file)
--- a/Java/Cycript.java
+++ b/Java/Cycript.java
@@ -80,16 +80,18 @@ public static class Wrapper
         if (false)
             return null;
         else if (method.equals(Object$equals))
+            // XXX: this assumes there is only one proxy
             return proxy == args[0];
         else if (method == Object$hashCode)
-            return System.identityHashCode(proxy);
+            // XXX: this assumes there is only one wrapper
+            return hashCode();
         else
             return handle(protect_, method.getName(), args);
     }
 }
 
-public static Object proxy(Class proxy, long protect) {
-    return Proxy.newProxyInstance(proxy.getClassLoader(), new Class[] {proxy}, new Wrapper(protect));
+public static Object proxy(Class proxy, Wrapper wrapper) {
+    return Proxy.newProxyInstance(proxy.getClassLoader(), new Class[] {proxy}, wrapper);
 }
 
 }