fix(native-image): register array types and key interfaces for Conscrypt JNI in GraalVM

Author: lqiu96

sdk-platform-java/gax-java/gax/src/main/java/com/google/api/gax/nativeimage/GoogleJsonClientFeature.java

@@ -93,7 +93,12 @@ private void loadConscrypt(BeforeAnalysisAccess access) {
       registerClassHierarchyForJni(access, "org.conscrypt.NativeRef");
       registerClassHierarchyForJni(access, "org.conscrypt.OpenSSLBIOInputStream");
 
-      // Register standard JDK classes accessed via JNI by Conscrypt native C code
+      // Register array types and JDK classes accessed via JNI by Conscrypt native C code
+      registerClassForJni(Object[].class);
+      registerClassForJni(byte[].class);
+      registerClassForJni(byte[][].class);
+      registerClassForJni(int[].class);
+
       registerClassHierarchyForJni(access, "java.util.Calendar");
       registerClassHierarchyForJni(access, "java.io.InputStream");
       registerClassHierarchyForJni(access, "java.io.OutputStream");
@@ -103,6 +108,8 @@ private void loadConscrypt(BeforeAnalysisAccess access) {
       registerClassHierarchyForJni(access, "java.lang.Object");
       registerClassHierarchyForJni(access, "java.nio.Buffer");
       registerClassHierarchyForJni(access, "java.nio.ByteBuffer");
+      registerClassHierarchyForJni(access, "java.security.PrivateKey");
+      registerClassHierarchyForJni(access, "java.security.Key");
     }
   }
 

sdk-platform-java/gax-java/gax/src/main/java/com/google/api/gax/nativeimage/NativeImageUtils.java

@@ -86,27 +86,37 @@ public static void registerConstructorsForReflection(FeatureAccess access, Strin
     }
   }
 
+  /** Registers an entire class for reflection use. */
+  public static void registerClassForReflection(Class<?> clazz) {
+    RuntimeReflection.register(clazz);
+    RuntimeReflection.register(clazz.getDeclaredConstructors());
+    RuntimeReflection.register(clazz.getDeclaredFields());
+    RuntimeReflection.register(clazz.getDeclaredMethods());
+  }
+
   /** Registers an entire class for reflection use. */
   public static void registerClassForReflection(FeatureAccess access, String name) {
     Class<?> clazz = access.findClassByName(name);
     if (clazz != null) {
-      RuntimeReflection.register(clazz);
-      RuntimeReflection.register(clazz.getDeclaredConstructors());
-      RuntimeReflection.register(clazz.getDeclaredFields());
-      RuntimeReflection.register(clazz.getDeclaredMethods());
+      registerClassForReflection(clazz);
     } else {
       LOGGER.log(Level.WARNING, CLASS_REFLECTION_ERROR_MESSAGE, name);
     }
   }
 
+  /** Registers an entire class for JNI use. */
+  public static void registerClassForJni(Class<?> clazz) {
+    RuntimeJNIAccess.register(clazz);
+    RuntimeJNIAccess.register(clazz.getDeclaredConstructors());
+    RuntimeJNIAccess.register(clazz.getDeclaredFields());
+    RuntimeJNIAccess.register(clazz.getDeclaredMethods());
+  }
+
   /** Registers an entire class for JNI use. */
   public static void registerClassForJni(FeatureAccess access, String name) {
     Class<?> clazz = access.findClassByName(name);
     if (clazz != null) {
-      RuntimeJNIAccess.register(clazz);
-      RuntimeJNIAccess.register(clazz.getDeclaredConstructors());
-      RuntimeJNIAccess.register(clazz.getDeclaredFields());
-      RuntimeJNIAccess.register(clazz.getDeclaredMethods());
+      registerClassForJni(clazz);
     } else {
       LOGGER.log(Level.WARNING, CLASS_REFLECTION_ERROR_MESSAGE, name);
     }