test/jdk/java/lang/invoke/callerSensitive/CallerSensitiveAccess.java

*** 20,11 ***
   * or visit www.oracle.com if you need additional information or have any
   * questions.
   */
  
  /* @test
!  * @bug 8196830 8235351 8257874
   * @modules java.base/jdk.internal.reflect
   * @run junit/othervm CallerSensitiveAccess
   * @summary Check Lookup findVirtual, findStatic and unreflect behavior with
   *          caller sensitive methods with focus on AccessibleObject.setAccessible
   */
--- 20,11 ---
   * or visit www.oracle.com if you need additional information or have any
   * questions.
   */
  
  /* @test
!  * @bug 8196830 8235351 8257874 8327639
   * @modules java.base/jdk.internal.reflect
   * @run junit/othervm CallerSensitiveAccess
   * @summary Check Lookup findVirtual, findStatic and unreflect behavior with
   *          caller sensitive methods with focus on AccessibleObject.setAccessible
   */

*** 49,24 ***
  import java.util.stream.Stream;
  
  import jdk.internal.reflect.CallerSensitive;
  
  import static org.junit.jupiter.api.Assertions.*;
  import org.junit.jupiter.api.Test;
  import org.junit.jupiter.params.ParameterizedTest;
  import org.junit.jupiter.params.provider.MethodSource;
  
  public class CallerSensitiveAccess {
  
      /**
       * Caller sensitive methods in APIs exported by java.base.
       */
      static Object[][] callerSensitiveMethods() {
!         try (Stream<Method> stream = callerSensitiveMethods(Object.class.getModule())) {
!             return stream.map(m -> new Object[]{m, shortDescription(m)})
!                     .toArray(Object[][]::new);
-         }
      }
  
      /**
       * Using publicLookup, attempt to use findVirtual or findStatic to obtain a
       * method handle to a caller sensitive method.
--- 49,31 ---
  import java.util.stream.Stream;
  
  import jdk.internal.reflect.CallerSensitive;
  
  import static org.junit.jupiter.api.Assertions.*;
+ import org.junit.jupiter.api.BeforeAll;
  import org.junit.jupiter.api.Test;
  import org.junit.jupiter.params.ParameterizedTest;
  import org.junit.jupiter.params.provider.MethodSource;
  
  public class CallerSensitiveAccess {
+     // Cache the list of Caller Sensitive Methods
+     private static List<Method> CALLER_SENSITIVE_METHODS;
+ 
+     @BeforeAll
+     public static void setupCallerSensitiveMethods() {
+         CALLER_SENSITIVE_METHODS = callerSensitiveMethods(Object.class.getModule());
+     }
  
      /**
       * Caller sensitive methods in APIs exported by java.base.
       */
      static Object[][] callerSensitiveMethods() {
!         return CALLER_SENSITIVE_METHODS.stream()
!                 .map(m -> new Object[]{m, shortDescription(m)})
!                 .toArray(Object[][]::new);
      }
  
      /**
       * Using publicLookup, attempt to use findVirtual or findStatic to obtain a
       * method handle to a caller sensitive method.

*** 99,17 ***
  
      /**
       * public accessible caller sensitive methods in APIs exported by java.base.
       */
      static Object[][] accessibleCallerSensitiveMethods() {
!         try (Stream<Method> stream = callerSensitiveMethods(Object.class.getModule())) {
-             return stream
                  .filter(m -> Modifier.isPublic(m.getModifiers()))
                  .map(m -> { m.setAccessible(true); return m; })
                  .map(m -> new Object[] { m, shortDescription(m) })
                  .toArray(Object[][]::new);
-         }
      }
  
      /**
       * Using publicLookup, attempt to use unreflect to obtain a method handle to a
       * caller sensitive method.
--- 106,15 ---
  
      /**
       * public accessible caller sensitive methods in APIs exported by java.base.
       */
      static Object[][] accessibleCallerSensitiveMethods() {
!         return CALLER_SENSITIVE_METHODS.stream()
                  .filter(m -> Modifier.isPublic(m.getModifiers()))
                  .map(m -> { m.setAccessible(true); return m; })
                  .map(m -> new Object[] { m, shortDescription(m) })
                  .toArray(Object[][]::new);
      }
  
      /**
       * Using publicLookup, attempt to use unreflect to obtain a method handle to a
       * caller sensitive method.

*** 388,14 ***
      }
  
      // -- supporting methods --
  
      /**
!      * Returns a stream of all caller sensitive methods on public classes in packages
       * exported by a named module.
       */
!     static Stream<Method> callerSensitiveMethods(Module module) {
          assertTrue(module.isNamed());
          ModuleReference mref = module.getLayer().configuration()
                  .findModule(module.getName())
                  .orElseThrow(() -> new RuntimeException())
                  .reference();
--- 393,15 ---
      }
  
      // -- supporting methods --
  
      /**
!      * Returns a List of all caller sensitive methods on public classes in packages
       * exported by a named module.
+      * Returns a List instead of a stream so the ModuleReader can be closed before returning.
       */
!     static List<Method> callerSensitiveMethods(Module module) {
          assertTrue(module.isNamed());
          ModuleReference mref = module.getLayer().configuration()
                  .findModule(module.getName())
                  .orElseThrow(() -> new RuntimeException())
                  .reference();

*** 411,11 ***
                      .filter(cn -> module.isExported(packageName(cn)))
                      .map(cn -> Class.forName(module, cn))
                      .filter(refc -> refc != null
                                      && Modifier.isPublic(refc.getModifiers()))
                      .map(refc -> callerSensitiveMethods(refc))
!                     .flatMap(List::stream);
          } catch (IOException ioe) {
              throw new UncheckedIOException(ioe);
          }
      }
  
--- 417,11 ---
                      .filter(cn -> module.isExported(packageName(cn)))
                      .map(cn -> Class.forName(module, cn))
                      .filter(refc -> refc != null
                                      && Modifier.isPublic(refc.getModifiers()))
                      .map(refc -> callerSensitiveMethods(refc))
!                     .flatMap(List::stream).toList();
          } catch (IOException ioe) {
              throw new UncheckedIOException(ioe);
          }
      }