Intercept Cipher#getCurrentSpi to avoid running error

robolectric · Jun 1, 2023 · c735405 · c735405
1 parent cf9c4b0
commit c735405
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/robolectric/src/test/java/org/robolectric/interceptors/AndroidInterceptorsTest.java b/robolectric/src/test/java/org/robolectric/interceptors/AndroidInterceptorsTest.java
@@ -27,7 +27,8 @@ public void allMethodRefs() throws Exception {
             new MethodRef("java.util.Locale", "adjustLanguageCode"),
             new MethodRef("java.io.FileDescriptor", "release$"),
             new MethodRef("java.io.FileDescriptor", "getInt$"),
-            new MethodRef("java.io.FileDescriptor", "setInt$"));
+            new MethodRef("java.io.FileDescriptor", "setInt$"),
+            new MethodRef("javax.crypto.Cipher", "getCurrentSpi"));
   }
 
   @Test

diff --git a/sandbox/src/main/java/org/robolectric/interceptors/AndroidInterceptors.java b/sandbox/src/main/java/org/robolectric/interceptors/AndroidInterceptors.java
@@ -24,6 +24,8 @@
 import java.util.List;
 import java.util.Locale;
 import javax.annotation.Nullable;
+import javax.crypto.Cipher;
+import javax.crypto.CipherSpi;
 import org.robolectric.fakes.CleanerCompat;
 import org.robolectric.internal.bytecode.Interceptor;
 import org.robolectric.internal.bytecode.MethodRef;
@@ -46,6 +48,7 @@ public static Collection<Interceptor> all() {
                 new FileDescriptorInterceptor(),
                 new NoOpInterceptor(),
                 new SocketInterceptor(),
+                new CipherInterceptor(),
                 new ReferenceRefersToInterceptor()));
 
     if (Util.getJavaVersion() >= 9) {
@@ -465,6 +468,35 @@ public MethodHandle getMethodHandle(String methodName, MethodType type)
     }
   }
 
+  /** Intercepts calls to methods in {@link javax.crypto.Cipher} not present in the OpenJDK. */
+  public static class CipherInterceptor extends Interceptor {
+    public CipherInterceptor() {
+      super(new MethodRef(Cipher.class, "getCurrentSpi"));
+    }
+
+    @Nullable
+    static CipherSpi getCurrentSpi() {
+      return null;
+    }
+
+    @Override
+    public Function<Object, Object> handle(MethodSignature methodSignature) {
+      return new Function<Object, Object>() {
+        @Override
+        public Object call(Class<?> theClass, Object value, Object[] params) {
+          return getCurrentSpi();
+        }
+      };
+    }
+
+    @Override
+    public MethodHandle getMethodHandle(String methodName, MethodType type)
+        throws NoSuchMethodException, IllegalAccessException {
+      return lookup.findStatic(
+          getClass(), "getCurrentSpi", methodType(CipherSpi.class, void.class));
+    }
+  }
+
   /** AndroidInterceptor for Reference.refersTo which is not available until JDK 16. */
   public static class ReferenceRefersToInterceptor extends Interceptor {
     private static final String METHOD = "refersTo";