[FFI/JDK22] Performance improvement via NativeMethodHandle

Signed-off-by: ChengJin01 <[email protected]>
ChengJin01 · Nov 27, 2023 · 6f34398 · 6f34398
1 parent 1d9233c
commit 6f34398
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 15 deletions.
diff --git a/src/java.base/share/classes/java/lang/invoke/MethodHandleImpl.java b/src/java.base/share/classes/java/lang/invoke/MethodHandleImpl.java
@@ -25,15 +25,14 @@
 
 /*
  * ===========================================================================
- * (c) Copyright IBM Corp. 2022, 2022 All Rights Reserved
+ * (c) Copyright IBM Corp. 2022, 2023 All Rights Reserved
  * ===========================================================================
  */
 
 package java.lang.invoke;
 
 import jdk.internal.access.JavaLangInvokeAccess;
 import jdk.internal.access.SharedSecrets;
-import jdk.internal.foreign.abi.NativeEntryPoint;
 import jdk.internal.org.objectweb.asm.ClassWriter;
 import jdk.internal.reflect.CallerSensitive;
 import jdk.internal.reflect.Reflection;
@@ -1580,7 +1579,7 @@ public VarHandle memorySegmentViewHandle(Class<?> carrier, long alignmentMask, B
             }
 
             @Override
-            public MethodHandle nativeMethodHandle(NativeEntryPoint nep) {
+            public MethodHandle nativeMethodHandle(MethodHandle nep) {
                 return NativeMethodHandle.make(nep);
             }
 

diff --git a/src/java.base/share/classes/java/lang/invoke/NativeMethodHandle.java b/src/java.base/share/classes/java/lang/invoke/NativeMethodHandle.java
@@ -23,34 +23,43 @@
  * questions.
  */
 
+/*
+ * ===========================================================================
+ * (c) Copyright IBM Corp. 2023, 2023 All Rights Reserved
+ * ===========================================================================
+ */
+
 package java.lang.invoke;
 
 import jdk.internal.vm.annotation.ForceInline;
-import jdk.internal.foreign.abi.NativeEntryPoint;
+
+import java.lang.foreign.MemorySegment;
+import java.lang.foreign.SegmentAllocator;
 
 import static java.lang.invoke.LambdaForm.*;
 import static java.lang.invoke.MethodHandleNatives.Constants.LM_TRUSTED;
 import static java.lang.invoke.MethodHandleNatives.Constants.REF_invokeStatic;
 import static java.lang.invoke.MethodHandleStatics.newInternalError;
 
 /**
- * This class models a method handle to a native function. A native method handle is made up of a {@link NativeEntryPoint},
+ * This class models a method handle to a native function. A native method handle is made up of a bounded {@link MethodHandle},
  * which is used to capture the characteristics of the native call (such as calling convention to be used,
  * or whether a native transition is required) and a <em>fallback</em> method handle, which can be used
  * when intrinsification of this method handle is not possible.
  */
 /*non-public*/ final class NativeMethodHandle extends MethodHandle {
-    final NativeEntryPoint nep;
+    /* The native entry point is literally bounded MethodHandle implemented in OpenJ9. */
+    final MethodHandle nep;
 
-    private NativeMethodHandle(MethodType type, LambdaForm form, NativeEntryPoint nep) {
+    private NativeMethodHandle(MethodType type, LambdaForm form, MethodHandle nep) {
         super(type, form);
         this.nep = nep;
     }
 
     /**
-     * Creates a new native method handle with given {@link NativeEntryPoint} and <em>fallback</em> method handle.
+     * Creates a new native method handle with given {@link MethodHandle} and <em>fallback</em> method handle.
      */
-    public static MethodHandle make(NativeEntryPoint nep) {
+    public static MethodHandle make(MethodHandle nep) {
         MethodType type = nep.type();
         if (hasIllegalType(type))
             throw new IllegalArgumentException("Illegal type(s) found: " + type);
@@ -74,11 +83,12 @@ private static boolean hasIllegalType(MethodType type) {
 
     private static boolean isIllegalType(Class<?> pType) {
         return !(pType == long.class
-              || pType == int.class
-              || pType == float.class
-              || pType == double.class
-              || pType == void.class
-              || pType == Object.class);
+                || pType == int.class
+                || pType == float.class
+                || pType == double.class
+                || pType == void.class
+                || pType == MemorySegment.class  // Skip checking the argument preset in OpenJ9 for the moment.
+                || pType == SegmentAllocator.class); // Skip checking the argument preset OpenJ9 for the moment.
     }
 
     private static final MemberName.Factory IMPL_NAMES = MemberName.getFactory();

diff --git a/src/java.base/share/classes/jdk/internal/access/JavaLangInvokeAccess.java b/src/java.base/share/classes/jdk/internal/access/JavaLangInvokeAccess.java
@@ -23,6 +23,12 @@
  * questions.
  */
 
+/*
+ * ===========================================================================
+ * (c) Copyright IBM Corp. 2023, 2023 All Rights Reserved
+ * ===========================================================================
+ */
+
 package jdk.internal.access;
 
 import jdk.internal.foreign.abi.NativeEntryPoint;
@@ -121,7 +127,7 @@ public interface JavaLangInvokeAccess {
      * @param nep the native entry point
      * @return the native method handle
      */
-    MethodHandle nativeMethodHandle(NativeEntryPoint nep);
+    MethodHandle nativeMethodHandle(MethodHandle nep);
 
     /**
      * Produces a method handle unreflecting from a {@code Constructor} with