23
24 import java.lang.classfile.ClassFile;
25 import jdk.test.lib.util.ForceGC;
26 import org.junit.jupiter.api.Test;
27 import org.junit.jupiter.params.ParameterizedTest;
28 import org.junit.jupiter.params.provider.ValueSource;
29
30 import java.lang.constant.ClassDesc;
31 import java.lang.constant.MethodTypeDesc;
32 import java.lang.invoke.MethodHandle;
33 import java.lang.invoke.MethodHandleProxies;
34 import java.lang.invoke.MethodHandles;
35 import java.lang.invoke.MethodType;
36 import java.lang.ref.WeakReference;
37 import java.util.Comparator;
38
39 import static java.lang.constant.ConstantDescs.*;
40 import static java.lang.invoke.MethodHandleProxies.*;
41 import static java.lang.invoke.MethodType.methodType;
42 import static java.lang.classfile.ClassFile.*;
43 import static org.junit.jupiter.api.Assertions.*;
44
45 /*
46 * @test
47 * @bug 6983726
48 * @library /test/lib
49 * @summary Tests on implementation hidden classes spinned by MethodHandleProxies
50 * @build WrapperHiddenClassTest Client jdk.test.lib.util.ForceGC
51 * @run junit WrapperHiddenClassTest
52 */
53 public class WrapperHiddenClassTest {
54
55 /**
56 * Tests an adversary "implementation" class will not be
57 * "recovered" by the wrapperInstance* APIs
58 */
59 @Test
60 public void testWrapperInstance() throws Throwable {
61 Comparator<Integer> hostile = createHostileInstance();
62 var mh = MethodHandles.publicLookup()
63 .findVirtual(Integer.class, "compareTo", methodType(int.class, Integer.class));
64 @SuppressWarnings("unchecked")
65 Comparator<Integer> proxy = (Comparator<Integer>) asInterfaceInstance(Comparator.class, mh);
66
67 assertTrue(isWrapperInstance(proxy));
68 assertFalse(isWrapperInstance(hostile));
69 assertSame(mh, wrapperInstanceTarget(proxy));
70 assertThrows(IllegalArgumentException.class, () -> wrapperInstanceTarget(hostile));
71 assertSame(Comparator.class, wrapperInstanceType(proxy));
72 assertThrows(IllegalArgumentException.class, () -> wrapperInstanceType(hostile));
73 }
74
75 private static final String TYPE = "interfaceType";
76 private static final String TARGET = "target";
77 private static final ClassDesc CD_HostileWrapper = ClassDesc.of("HostileWrapper");
78 private static final ClassDesc CD_Comparator = ClassDesc.of("java.util.Comparator");
79 private static final MethodTypeDesc MTD_int_Object_Object = MethodTypeDesc.of(CD_int, CD_Object, CD_Object);
80 private static final MethodTypeDesc MTD_int_Integer = MethodTypeDesc.of(CD_int, CD_Integer);
81
82 // Update this template when the MHP template is updated
83 @SuppressWarnings("unchecked")
84 private Comparator<Integer> createHostileInstance() throws Throwable {
85 var cf = ClassFile.of();
86 var bytes = cf.build(CD_HostileWrapper, clb -> {
87 clb.withSuperclass(CD_Object);
88 clb.withFlags(ACC_FINAL | ACC_SYNTHETIC);
89 clb.withInterfaceSymbols(CD_Comparator);
90
91 // static and instance fields
92 clb.withField(TYPE, CD_Class, ACC_PRIVATE | ACC_STATIC | ACC_FINAL);
93 clb.withField(TARGET, CD_MethodHandle, ACC_PRIVATE | ACC_FINAL);
94
95 // <clinit>
96 clb.withMethodBody(CLASS_INIT_NAME, MTD_void, ACC_STATIC, cob -> {
97 cob.loadConstant(CD_Comparator);
98 cob.putstatic(CD_HostileWrapper, TYPE, CD_Class);
99 cob.return_();
100 });
101
102 // <init>
103 clb.withMethodBody(INIT_NAME, MTD_void, ACC_PUBLIC, cob -> {
104 cob.aload(0);
105 cob.invokespecial(CD_Object, INIT_NAME, MTD_void);
106 cob.return_();
107 });
108
|
23
24 import java.lang.classfile.ClassFile;
25 import jdk.test.lib.util.ForceGC;
26 import org.junit.jupiter.api.Test;
27 import org.junit.jupiter.params.ParameterizedTest;
28 import org.junit.jupiter.params.provider.ValueSource;
29
30 import java.lang.constant.ClassDesc;
31 import java.lang.constant.MethodTypeDesc;
32 import java.lang.invoke.MethodHandle;
33 import java.lang.invoke.MethodHandleProxies;
34 import java.lang.invoke.MethodHandles;
35 import java.lang.invoke.MethodType;
36 import java.lang.ref.WeakReference;
37 import java.util.Comparator;
38
39 import static java.lang.constant.ConstantDescs.*;
40 import static java.lang.invoke.MethodHandleProxies.*;
41 import static java.lang.invoke.MethodType.methodType;
42 import static java.lang.classfile.ClassFile.*;
43
44 import jdk.internal.misc.PreviewFeatures;
45
46 import static org.junit.jupiter.api.Assertions.*;
47
48 /*
49 * @test
50 * @bug 6983726
51 * @library /test/lib
52 * @modules java.base/jdk.internal.misc
53 * @summary Tests on implementation hidden classes spinned by MethodHandleProxies
54 * @build WrapperHiddenClassTest Client jdk.test.lib.util.ForceGC
55 * @run junit WrapperHiddenClassTest
56 */
57 public class WrapperHiddenClassTest {
58
59 /**
60 * Tests an adversary "implementation" class will not be
61 * "recovered" by the wrapperInstance* APIs
62 */
63 @Test
64 public void testWrapperInstance() throws Throwable {
65 Comparator<Integer> hostile = createHostileInstance();
66 var mh = MethodHandles.publicLookup()
67 .findVirtual(Integer.class, "compareTo", methodType(int.class, Integer.class));
68 @SuppressWarnings("unchecked")
69 Comparator<Integer> proxy = (Comparator<Integer>) asInterfaceInstance(Comparator.class, mh);
70
71 assertTrue(isWrapperInstance(proxy));
72 assertFalse(isWrapperInstance(hostile));
73 assertSame(mh, wrapperInstanceTarget(proxy));
74 assertThrows(IllegalArgumentException.class, () -> wrapperInstanceTarget(hostile));
75 assertSame(Comparator.class, wrapperInstanceType(proxy));
76 assertThrows(IllegalArgumentException.class, () -> wrapperInstanceType(hostile));
77 }
78
79 private static final String TYPE = "interfaceType";
80 private static final String TARGET = "target";
81 private static final ClassDesc CD_HostileWrapper = ClassDesc.of("HostileWrapper");
82 private static final ClassDesc CD_Comparator = ClassDesc.of("java.util.Comparator");
83 private static final MethodTypeDesc MTD_int_Object_Object = MethodTypeDesc.of(CD_int, CD_Object, CD_Object);
84 private static final MethodTypeDesc MTD_int_Integer = MethodTypeDesc.of(CD_int, CD_Integer);
85
86 // Update this template when the MHP template is updated
87 @SuppressWarnings("unchecked")
88 private Comparator<Integer> createHostileInstance() throws Throwable {
89 var cf = ClassFile.of();
90 var bytes = cf.build(CD_HostileWrapper, clb -> {
91 clb.withSuperclass(CD_Object);
92 clb.withFlags((PreviewFeatures.isEnabled() ? ACC_IDENTITY : 0) | ACC_FINAL | ACC_SYNTHETIC);
93 clb.withInterfaceSymbols(CD_Comparator);
94
95 // static and instance fields
96 clb.withField(TYPE, CD_Class, ACC_PRIVATE | ACC_STATIC | ACC_FINAL);
97 clb.withField(TARGET, CD_MethodHandle, ACC_PRIVATE | ACC_FINAL);
98
99 // <clinit>
100 clb.withMethodBody(CLASS_INIT_NAME, MTD_void, ACC_STATIC, cob -> {
101 cob.loadConstant(CD_Comparator);
102 cob.putstatic(CD_HostileWrapper, TYPE, CD_Class);
103 cob.return_();
104 });
105
106 // <init>
107 clb.withMethodBody(INIT_NAME, MTD_void, ACC_PUBLIC, cob -> {
108 cob.aload(0);
109 cob.invokespecial(CD_Object, INIT_NAME, MTD_void);
110 cob.return_();
111 });
112
|