8263459: Add better support for restricted methods

src/jdk.incubator.foreign/share/classes/jdk/internal/foreign/MemoryAddressImpl.java

@@ -93,6 +93,12 @@ public String toString() {
         return "MemoryAddress{ base: " + base + " offset=0x" + Long.toHexString(offset) + " }";
     }
 
+    @Override
+    @NativeAccess
+    public MemorySegment asSegmentRestricted(long bytesSize) {
+        return MemoryAddress.super.asSegmentRestricted(bytesSize);
+    }
+
     @Override
     @NativeAccess
     public MemorySegment asSegmentRestricted(long bytesSize, Runnable cleanupAction, Object attachment) {

test/jdk/java/foreign/TestRestricted.java

@@ -21,6 +21,7 @@
  * questions.
  */
 
+import jdk.incubator.foreign.MemoryAddress;
 import jdk.incubator.foreign.MemorySegment;
 import org.testng.annotations.Test;
 
@@ -33,7 +34,6 @@
  * @run testng TestRestricted
  */
 public class TestRestricted {
-
     @Test(expectedExceptions = IllegalAccessException.class)
     public void testReflection() throws Throwable {
         Method method = MemorySegment.class.getDeclaredMethod("ofNativeRestricted");
@@ -49,4 +49,21 @@ public void testLookup() throws Throwable {
     public void testDirectAccess() throws Throwable {
         MemorySegment.ofNativeRestricted();
     }
+
+    @Test(expectedExceptions = IllegalAccessException.class)
+    public void testReflection2() throws Throwable {
+        Method method = MemoryAddress.class.getDeclaredMethod("asSegmentRestricted", long.class);
+        method.invoke(MemoryAddress.NULL, 4000L);
+    }
+
+    @Test(expectedExceptions = IllegalAccessException.class)
+    public void testLookup2() throws Throwable {
+        MethodHandles.lookup().findVirtual(MemoryAddress.class, "asSegmentRestricted",
+            MethodType.methodType(MemorySegment.class, long.class));
+    }
+
+    @Test(expectedExceptions = IllegalAccessException.class)
+    public void testDirectAccess2() throws Throwable {
+        MemoryAddress.NULL.asSegmentRestricted(4000L);
+    }
 }