1/*
2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * @test DefineAnon
26 * @bug 8058575
27 * @library /testlibrary
28 * @modules java.base/jdk.internal.org.objectweb.asm
29 *          java.management
30 * @compile -XDignore.symbol.file=true DefineAnon.java
31 * @run main/othervm p1.DefineAnon
32 */
33
34package p1;
35
36import jdk.internal.org.objectweb.asm.ClassWriter;
37import jdk.internal.org.objectweb.asm.MethodVisitor;
38import jdk.internal.org.objectweb.asm.Opcodes;
39import sun.misc.Unsafe;
40
41
42class T {
43    static           protected void test0() { System.out.println("test0 (public)"); }
44    static           protected void test1() { System.out.println("test1 (protected)"); }
45    static /*package-private*/ void test2() { System.out.println("test2 (package)"); }
46    static             private void test3() { System.out.println("test3 (private)"); }
47}
48
49public class DefineAnon {
50
51    private static Unsafe getUnsafe() {
52        try {
53            java.lang.reflect.Field singleoneInstanceField = Unsafe.class.getDeclaredField("theUnsafe");
54            singleoneInstanceField.setAccessible(true);
55            return (Unsafe) singleoneInstanceField.get(null);
56        } catch (Throwable ex) {
57            throw new RuntimeException("Was unable to get Unsafe instance.");
58        }
59    }
60
61    static Unsafe UNSAFE = DefineAnon.getUnsafe();
62
63    static Class<?> getAnonClass(Class<?> hostClass, final String className) {
64        final String superName = "java/lang/Object";
65        ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS + ClassWriter.COMPUTE_FRAMES);
66        cw.visit(Opcodes.V1_8, Opcodes.ACC_PUBLIC + Opcodes.ACC_FINAL + Opcodes.ACC_SUPER, className, null, superName, null);
67
68        MethodVisitor mv = cw.visitMethod(Opcodes.ACC_STATIC | Opcodes.ACC_PUBLIC, "test", "()V", null, null);
69        mv.visitMethodInsn(Opcodes.INVOKESTATIC, "p1/T", "test0", "()V", false);
70        mv.visitMethodInsn(Opcodes.INVOKESTATIC, "p1/T", "test1", "()V", false);
71        mv.visitMethodInsn(Opcodes.INVOKESTATIC, "p1/T", "test2", "()V", false);
72        mv.visitMethodInsn(Opcodes.INVOKESTATIC, "p1/T", "test3", "()V", false);
73        mv.visitInsn(Opcodes.RETURN);
74        mv.visitMaxs(0, 0);
75        mv.visitEnd();
76
77        final byte[] classBytes = cw.toByteArray();
78        Class<?> invokerClass = UNSAFE.defineAnonymousClass(hostClass, classBytes, new Object[0]);
79        UNSAFE.ensureClassInitialized(invokerClass);
80        return invokerClass;
81    }
82
83    public static void main(String[] args) throws Throwable {
84        Throwable fail = null;
85
86        // Anonymous class has the privileges of its host class, so test[0123] should all work.
87        System.out.println("Injecting from the same package (p1):");
88        Class<?> p1cls = getAnonClass(T.class, "p1/AnonClass");
89        try {
90            p1cls.getMethod("test").invoke(null);
91        } catch (Throwable ex) {
92            ex.printStackTrace();
93            fail = ex;  // throw this to make test fail, since subtest failed
94        }
95
96        // Anonymous class has different package name from host class.  Should throw
97        // IllegalArgumentException.
98        System.out.println("Injecting from the wrong package (p2):");
99        try {
100            Class<?> p2cls = getAnonClass(DefineAnon.class, "p2/AnonClass");
101            p2cls.getMethod("test").invoke(null);
102            System.out.println("Failed, did not get expected IllegalArgumentException");
103        } catch (java.lang.IllegalArgumentException e) {
104            if (e.getMessage().contains("Host class p1/DefineAnon and anonymous class p2/AnonClass")) {
105                System.out.println("Got expected IllegalArgumentException: " + e.getMessage());
106            } else {
107                throw new RuntimeException("Unexpected message: " + e.getMessage());
108            }
109        } catch (Throwable ex) {
110            ex.printStackTrace();
111            fail = ex;  // throw this to make test fail, since subtest failed
112        }
113
114        // Inject a class in the unnamed package into p1.T.  It should be able
115        // to access all methods in p1.T.
116        System.out.println("Injecting unnamed package into correct host class:");
117        try {
118            Class<?> p3cls = getAnonClass(T.class, "AnonClass");
119            p3cls.getMethod("test").invoke(null);
120        } catch (Throwable ex) {
121            ex.printStackTrace();
122            fail = ex;  // throw this to make test fail, since subtest failed
123        }
124
125        // Try using an array class as the host class.  This should throw IllegalArgumentException.
126        try {
127            Class<?> p3cls = getAnonClass(String[].class, "AnonClass");
128            throw new RuntimeException("Expected IllegalArgumentException not thrown");
129        } catch (IllegalArgumentException ex) {
130        }
131
132        if (fail != null) throw fail;  // make test fail, since subtest failed
133    }
134}
135