--- /dev/null 2016-05-31 09:42:47.975716356 -0700 +++ new/src/jdk.vm.compiler/share/classes/org.graalvm.compiler.core.test/src/org/graalvm/compiler/core/test/InterfaceMethodHandleTest.java 2016-12-07 13:48:25.721643199 -0800 @@ -0,0 +1,213 @@ +/* + * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ +package org.graalvm.compiler.core.test; + +import java.lang.invoke.MethodHandle; +import java.lang.invoke.MethodHandles; +import java.lang.invoke.MethodType; + +import org.junit.Test; + +import org.graalvm.compiler.code.CompilationResult; +import org.graalvm.compiler.test.ExportingClassLoader; + +import jdk.internal.org.objectweb.asm.ClassWriter; +import jdk.internal.org.objectweb.asm.Label; +import jdk.internal.org.objectweb.asm.MethodVisitor; +import jdk.internal.org.objectweb.asm.Opcodes; +import jdk.vm.ci.code.InstalledCode; +import jdk.vm.ci.meta.ResolvedJavaMethod; + +public final class InterfaceMethodHandleTest extends GraalCompilerTest implements Opcodes { + private static final MethodHandle INTERFACE_HANDLE_M; + private static final MethodHandle INTERFACE_HANDLE_M2; + + public interface I { + int m(); + + int m2(int a, int b, int c, int d, int e, int f, int g, int h, int i, int j); + } + + static class A implements I { + @Override + public int m() { + return 0; + } + + @Override + public int m2(int a, int b, int c, int d, int e, int f, int g, int h, int i, int j) { + return 1; + } + + } + + static class M2Thrower implements I { + @Override + public int m() { + return 0; + } + + @Override + public int m2(int a, int b, int c, int d, int e, int f, int g, int h, int i, int j) { + throw new InternalError(); + } + + } + + static { + try { + MethodType type = MethodType.fromMethodDescriptorString("()I", I.class.getClassLoader()); + INTERFACE_HANDLE_M = MethodHandles.lookup().findVirtual(I.class, "m", type); + MethodType type2 = MethodType.fromMethodDescriptorString("(IIIIIIIIII)I", I.class.getClassLoader()); + INTERFACE_HANDLE_M2 = MethodHandles.lookup().findVirtual(I.class, "m2", type2); + } catch (IllegalAccessException | NoSuchMethodException e) { + throw new RuntimeException("unable to initialize method handle", e); + } + } + + public static Object invokeInterfaceHandle(I o) throws Throwable { + return (int) INTERFACE_HANDLE_M.invokeExact(o); + } + + @Test + public void testInvokeInterface01() { + test("invokeInterfaceHandle", new A()); + + } + + @Test + public void testInvokeInterface02() throws InstantiationException, IllegalAccessException, ClassNotFoundException { + test("invokeInterfaceHandle", loader.findClass(NAME).newInstance()); + } + + public static Object invokeInterfaceHandle2(I o, int a, int b, int c, int d, int e, int f, int g, int h, int i, int j) throws Throwable { + return (int) INTERFACE_HANDLE_M2.invokeExact(o, a, b, c, d, e, f, g, h, i, j); + } + + @Override + protected InstalledCode addMethod(ResolvedJavaMethod method, CompilationResult compResult) { + if (method.getDeclaringClass().equals(getMetaAccess().lookupJavaType(M2Thrower.class))) { + // Make sure M2Thrower.m2 is invoked from normal code + return getBackend().createDefaultInstalledCode(method, compResult); + } + return super.addMethod(method, compResult); + } + + /** + * Try to exercise a mixed calling sequence with regular JIT code calling a method handle that + * can't be inlined with an implementation compiled by Graal that throws an exception. + */ + @Test + public void testInvokeInterface03() throws Throwable { + A goodInstance = new A(); + I badInstance = new M2Thrower(); + getCode(getMetaAccess().lookupJavaMethod(getMethod(M2Thrower.class, "m2"))); + for (int x = 0; x < 1000; x++) { + final int limit = 20000; + for (int i = 0; i <= limit; i++) { + try { + invokeInterfaceHandle2(i < limit - 1 ? goodInstance : badInstance, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10); + } catch (InternalError e) { + + } + } + } + } + + private static final String BASENAME = InterfaceMethodHandleTest.class.getName(); + private static final String NAME = BASENAME + "_B"; + private AsmLoader loader = new AsmLoader(UnbalancedMonitorsTest.class.getClassLoader()); + + /** + * Construct a type which claims to implement {@link I} but with incorrect access on {@link I#m} + * so that an exception must be thrown. + */ + public static byte[] bytesForB() { + + ClassWriter cw = new ClassWriter(0); + MethodVisitor mv; + String jvmName = NAME.replace('.', '/'); + cw.visit(52, ACC_SUPER | ACC_PUBLIC, jvmName, null, "java/lang/Object", new String[]{BASENAME.replace('.', '/') + "$I"}); + + mv = cw.visitMethod(ACC_PUBLIC, "", "()V", null, null); + mv.visitCode(); + Label l0 = new Label(); + mv.visitLabel(l0); + mv.visitVarInsn(ALOAD, 0); + mv.visitMethodInsn(INVOKESPECIAL, "java/lang/Object", "", "()V", false); + mv.visitInsn(RETURN); + Label l1 = new Label(); + mv.visitLabel(l1); + mv.visitMaxs(1, 1); + mv.visitEnd(); + + mv = cw.visitMethod(ACC_PRIVATE, "m", "()I", null, null); + mv.visitCode(); + l0 = new Label(); + mv.visitLabel(l0); + mv.visitInsn(ICONST_0); + mv.visitInsn(IRETURN); + l1 = new Label(); + mv.visitLabel(l1); + mv.visitMaxs(1, 1); + mv.visitEnd(); + + cw.visitEnd(); + + mv = cw.visitMethod(ACC_PRIVATE, "m2", "(IIIIIIIIII)I", null, null); + mv.visitCode(); + l0 = new Label(); + mv.visitLabel(l0); + mv.visitInsn(ICONST_0); + mv.visitInsn(IRETURN); + l1 = new Label(); + mv.visitLabel(l1); + mv.visitMaxs(1, 11); + mv.visitEnd(); + + cw.visitEnd(); + + return cw.toByteArray(); + } + + public static class AsmLoader extends ExportingClassLoader { + Class loaded; + + public AsmLoader(ClassLoader parent) { + super(parent); + } + + @Override + protected Class findClass(String name) throws ClassNotFoundException { + if (name.equals(NAME)) { + if (loaded != null) { + return loaded; + } + byte[] bytes = bytesForB(); + return (loaded = defineClass(name, bytes, 0, bytes.length)); + } else { + return super.findClass(name); + } + } + } +}