aoqi@0: /* aoqi@0: * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved. aoqi@0: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. aoqi@0: * aoqi@0: * This code is free software; you can redistribute it and/or modify it aoqi@0: * under the terms of the GNU General Public License version 2 only, as aoqi@0: * published by the Free Software Foundation. aoqi@0: * aoqi@0: * This code is distributed in the hope that it will be useful, but WITHOUT aoqi@0: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or aoqi@0: * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License aoqi@0: * version 2 for more details (a copy is included in the LICENSE file that aoqi@0: * accompanied this code). aoqi@0: * aoqi@0: * You should have received a copy of the GNU General Public License version aoqi@0: * 2 along with this work; if not, write to the Free Software Foundation, aoqi@0: * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. aoqi@0: * aoqi@0: * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA aoqi@0: * or visit www.oracle.com if you need additional information or have any aoqi@0: * questions. aoqi@0: */ aoqi@0: aoqi@0: /* aoqi@0: * @test aoqi@0: * @bug 7194586 8003280 8006694 8010404 aoqi@0: * @summary Add lambda tests aoqi@0: * Add back-end support for invokedynamic aoqi@0: * temporarily workaround combo tests are causing time out in several platforms aoqi@0: * @library ../lib aoqi@0: * @build JavacTestingAbstractThreadedTest aoqi@0: * @run main/othervm TestInvokeDynamic aoqi@0: */ aoqi@0: aoqi@0: // use /othervm to avoid jtreg timeout issues (CODETOOLS-7900047) aoqi@0: // see JDK-8006746 aoqi@0: aoqi@0: import java.io.File; aoqi@0: import java.net.URI; aoqi@0: import java.util.ArrayList; aoqi@0: import java.util.Arrays; aoqi@0: import java.util.Locale; aoqi@0: aoqi@0: import javax.tools.Diagnostic; aoqi@0: import javax.tools.JavaFileObject; aoqi@0: import javax.tools.SimpleJavaFileObject; aoqi@0: aoqi@0: import com.sun.source.tree.MethodInvocationTree; aoqi@0: import com.sun.source.tree.MethodTree; aoqi@0: import com.sun.source.util.TaskEvent; aoqi@0: import com.sun.source.util.TaskListener; aoqi@0: import com.sun.source.util.TreeScanner; aoqi@0: aoqi@0: import com.sun.tools.classfile.Attribute; aoqi@0: import com.sun.tools.classfile.BootstrapMethods_attribute; aoqi@0: import com.sun.tools.classfile.ClassFile; aoqi@0: import com.sun.tools.classfile.Code_attribute; aoqi@0: import com.sun.tools.classfile.ConstantPool.*; aoqi@0: import com.sun.tools.classfile.Instruction; aoqi@0: import com.sun.tools.classfile.LineNumberTable_attribute; aoqi@0: import com.sun.tools.classfile.Method; aoqi@0: aoqi@0: import com.sun.tools.javac.api.JavacTaskImpl; aoqi@0: import com.sun.tools.javac.code.Symbol; aoqi@0: import com.sun.tools.javac.code.Symbol.MethodSymbol; aoqi@0: import com.sun.tools.javac.code.Symtab; aoqi@0: import com.sun.tools.javac.code.Types; aoqi@0: import com.sun.tools.javac.jvm.Pool; aoqi@0: import com.sun.tools.javac.tree.JCTree.JCMethodInvocation; aoqi@0: import com.sun.tools.javac.tree.JCTree.JCMethodDecl; aoqi@0: import com.sun.tools.javac.tree.JCTree.JCIdent; aoqi@0: import com.sun.tools.javac.util.Context; aoqi@0: import com.sun.tools.javac.util.Names; aoqi@0: aoqi@0: import static com.sun.tools.javac.jvm.ClassFile.*; aoqi@0: aoqi@0: public class TestInvokeDynamic aoqi@0: extends JavacTestingAbstractThreadedTest aoqi@0: implements Runnable { aoqi@0: aoqi@0: enum StaticArgumentKind { aoqi@0: STRING("Hello!", "String", "Ljava/lang/String;") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: return (cpInfo instanceof CONSTANT_String_info) && aoqi@0: ((CONSTANT_String_info)cpInfo).getString() aoqi@0: .equals(value); aoqi@0: } aoqi@0: }, aoqi@0: CLASS(null, "Class", "Ljava/lang/Class;") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: return (cpInfo instanceof CONSTANT_Class_info) && aoqi@0: ((CONSTANT_Class_info)cpInfo).getName() aoqi@0: .equals("java/lang/String"); aoqi@0: } aoqi@0: }, aoqi@0: INTEGER(1, "int", "I") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: return (cpInfo instanceof CONSTANT_Integer_info) && aoqi@0: ((CONSTANT_Integer_info)cpInfo).value == aoqi@0: ((Integer)value).intValue(); aoqi@0: } aoqi@0: }, aoqi@0: LONG(1L, "long", "J") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: return (cpInfo instanceof CONSTANT_Long_info) && aoqi@0: ((CONSTANT_Long_info)cpInfo).value == aoqi@0: ((Long)value).longValue(); aoqi@0: } aoqi@0: }, aoqi@0: FLOAT(1.0f, "float", "F") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: return (cpInfo instanceof CONSTANT_Float_info) && aoqi@0: ((CONSTANT_Float_info)cpInfo).value == aoqi@0: ((Float)value).floatValue(); aoqi@0: } aoqi@0: }, aoqi@0: DOUBLE(1.0, "double","D") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: return (cpInfo instanceof CONSTANT_Double_info) && aoqi@0: ((CONSTANT_Double_info)cpInfo).value == aoqi@0: ((Double)value).doubleValue(); aoqi@0: } aoqi@0: }, aoqi@0: METHOD_HANDLE(null, "MethodHandle", "Ljava/lang/invoke/MethodHandle;") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: if (!(cpInfo instanceof CONSTANT_MethodHandle_info)) aoqi@0: return false; aoqi@0: CONSTANT_MethodHandle_info handleInfo = aoqi@0: (CONSTANT_MethodHandle_info)cpInfo; aoqi@0: return handleInfo.getCPRefInfo().getClassName().equals("Array") && aoqi@0: handleInfo.reference_kind == RefKind.REF_invokeVirtual && aoqi@0: handleInfo.getCPRefInfo() aoqi@0: .getNameAndTypeInfo().getName().equals("clone") && aoqi@0: handleInfo.getCPRefInfo() aoqi@0: .getNameAndTypeInfo().getType().equals("()Ljava/lang/Object;"); aoqi@0: } aoqi@0: }, aoqi@0: METHOD_TYPE(null, "MethodType", "Ljava/lang/invoke/MethodType;") { aoqi@0: @Override aoqi@0: boolean check(CPInfo cpInfo) throws Exception { aoqi@0: return (cpInfo instanceof CONSTANT_MethodType_info) && aoqi@0: ((CONSTANT_MethodType_info)cpInfo).getType() aoqi@0: .equals("()Ljava/lang/Object;"); aoqi@0: } aoqi@0: }; aoqi@0: aoqi@0: Object value; aoqi@0: String sourceTypeStr; aoqi@0: String bytecodeTypeStr; aoqi@0: aoqi@0: StaticArgumentKind(Object value, String sourceTypeStr, aoqi@0: String bytecodeTypeStr) { aoqi@0: this.value = value; aoqi@0: this.sourceTypeStr = sourceTypeStr; aoqi@0: this.bytecodeTypeStr = bytecodeTypeStr; aoqi@0: } aoqi@0: aoqi@0: abstract boolean check(CPInfo cpInfo) throws Exception; aoqi@0: aoqi@0: Object getValue(Symtab syms, Names names, Types types) { aoqi@0: switch (this) { aoqi@0: case STRING: aoqi@0: case INTEGER: aoqi@0: case LONG: aoqi@0: case FLOAT: aoqi@0: case DOUBLE: aoqi@0: return value; aoqi@0: case CLASS: aoqi@0: return syms.stringType.tsym; aoqi@0: case METHOD_HANDLE: aoqi@0: return new Pool.MethodHandle(REF_invokeVirtual, aoqi@0: syms.arrayCloneMethod, types); aoqi@0: case METHOD_TYPE: aoqi@0: return syms.arrayCloneMethod.type; aoqi@0: default: aoqi@0: throw new AssertionError(); aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: enum StaticArgumentsArity { aoqi@0: ZERO(0), aoqi@0: ONE(1), aoqi@0: TWO(2), aoqi@0: THREE(3); aoqi@0: aoqi@0: int arity; aoqi@0: aoqi@0: StaticArgumentsArity(int arity) { aoqi@0: this.arity = arity; aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: public static void main(String... args) throws Exception { aoqi@0: for (StaticArgumentsArity arity : StaticArgumentsArity.values()) { aoqi@0: if (arity.arity == 0) { aoqi@0: pool.execute(new TestInvokeDynamic(arity)); aoqi@0: } else { aoqi@0: for (StaticArgumentKind sak1 : StaticArgumentKind.values()) { aoqi@0: if (arity.arity == 1) { aoqi@0: pool.execute(new TestInvokeDynamic(arity, sak1)); aoqi@0: } else { aoqi@0: for (StaticArgumentKind sak2 : StaticArgumentKind.values()) { aoqi@0: if (arity.arity == 2) { aoqi@0: pool.execute(new TestInvokeDynamic(arity, sak1, sak2)); aoqi@0: } else { aoqi@0: for (StaticArgumentKind sak3 : StaticArgumentKind.values()) { aoqi@0: pool.execute( aoqi@0: new TestInvokeDynamic(arity, sak1, sak2, sak3)); aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: checkAfterExec(); aoqi@0: } aoqi@0: aoqi@0: StaticArgumentsArity arity; aoqi@0: StaticArgumentKind[] saks; aoqi@0: DiagChecker dc; aoqi@0: aoqi@0: TestInvokeDynamic(StaticArgumentsArity arity, StaticArgumentKind... saks) { aoqi@0: this.arity = arity; aoqi@0: this.saks = saks; aoqi@0: dc = new DiagChecker(); aoqi@0: } aoqi@0: aoqi@0: public void run() { aoqi@0: int id = checkCount.incrementAndGet(); aoqi@0: JavaSource source = new JavaSource(id); aoqi@0: JavacTaskImpl ct = (JavacTaskImpl)comp.getTask(null, fm.get(), dc, aoqi@0: Arrays.asList("-g"), null, Arrays.asList(source)); aoqi@0: Context context = ct.getContext(); aoqi@0: Symtab syms = Symtab.instance(context); aoqi@0: Names names = Names.instance(context); aoqi@0: Types types = Types.instance(context); aoqi@0: ct.addTaskListener(new Indifier(syms, names, types)); aoqi@0: try { aoqi@0: ct.generate(); aoqi@0: } catch (Throwable t) { aoqi@0: t.printStackTrace(); aoqi@0: throw new AssertionError( aoqi@0: String.format("Error thrown when compiling following code\n%s", aoqi@0: source.source)); aoqi@0: } aoqi@0: if (dc.diagFound) { aoqi@0: throw new AssertionError( aoqi@0: String.format("Diags found when compiling following code\n%s\n\n%s", aoqi@0: source.source, dc.printDiags())); aoqi@0: } aoqi@0: verifyBytecode(id); aoqi@0: } aoqi@0: aoqi@0: void verifyBytecode(int id) { aoqi@0: File compiledTest = new File(String.format("Test%d.class", id)); aoqi@0: try { aoqi@0: ClassFile cf = ClassFile.read(compiledTest); aoqi@0: Method testMethod = null; aoqi@0: for (Method m : cf.methods) { aoqi@0: if (m.getName(cf.constant_pool).equals("test")) { aoqi@0: testMethod = m; aoqi@0: break; aoqi@0: } aoqi@0: } aoqi@0: if (testMethod == null) { aoqi@0: throw new Error("Test method not found"); aoqi@0: } aoqi@0: Code_attribute ea = aoqi@0: (Code_attribute)testMethod.attributes.get(Attribute.Code); aoqi@0: if (testMethod == null) { aoqi@0: throw new Error("Code attribute for test() method not found"); aoqi@0: } aoqi@0: aoqi@0: int bsmIdx = -1; aoqi@0: aoqi@0: for (Instruction i : ea.getInstructions()) { aoqi@0: if (i.getMnemonic().equals("invokedynamic")) { aoqi@0: CONSTANT_InvokeDynamic_info indyInfo = aoqi@0: (CONSTANT_InvokeDynamic_info)cf aoqi@0: .constant_pool.get(i.getShort(1)); aoqi@0: bsmIdx = indyInfo.bootstrap_method_attr_index; aoqi@0: if (!indyInfo.getNameAndTypeInfo().getType().equals("()V")) { aoqi@0: throw new aoqi@0: AssertionError("type mismatch for CONSTANT_InvokeDynamic_info"); aoqi@0: } aoqi@0: } aoqi@0: } aoqi@0: if (bsmIdx == -1) { aoqi@0: throw new Error("Missing invokedynamic in generated code"); aoqi@0: } aoqi@0: aoqi@0: BootstrapMethods_attribute bsm_attr = aoqi@0: (BootstrapMethods_attribute)cf aoqi@0: .getAttribute(Attribute.BootstrapMethods); aoqi@0: if (bsm_attr.bootstrap_method_specifiers.length != 1) { aoqi@0: throw new Error("Bad number of method specifiers " + aoqi@0: "in BootstrapMethods attribute"); aoqi@0: } aoqi@0: BootstrapMethods_attribute.BootstrapMethodSpecifier bsm_spec = aoqi@0: bsm_attr.bootstrap_method_specifiers[0]; aoqi@0: aoqi@0: if (bsm_spec.bootstrap_arguments.length != arity.arity) { aoqi@0: throw new Error("Bad number of static invokedynamic args " + aoqi@0: "in BootstrapMethod attribute"); aoqi@0: } aoqi@0: aoqi@0: int count = 0; aoqi@0: for (StaticArgumentKind sak : saks) { aoqi@0: if (!sak.check(cf.constant_pool aoqi@0: .get(bsm_spec.bootstrap_arguments[count]))) { aoqi@0: throw new Error("Bad static argument value " + sak); aoqi@0: } aoqi@0: count++; aoqi@0: } aoqi@0: aoqi@0: CONSTANT_MethodHandle_info bsm_handle = aoqi@0: (CONSTANT_MethodHandle_info)cf.constant_pool aoqi@0: .get(bsm_spec.bootstrap_method_ref); aoqi@0: aoqi@0: if (bsm_handle.reference_kind != RefKind.REF_invokeStatic) { aoqi@0: throw new Error("Bad kind on boostrap method handle"); aoqi@0: } aoqi@0: aoqi@0: CONSTANT_Methodref_info bsm_ref = aoqi@0: (CONSTANT_Methodref_info)cf.constant_pool aoqi@0: .get(bsm_handle.reference_index); aoqi@0: aoqi@0: if (!bsm_ref.getClassInfo().getName().equals("Bootstrap")) { aoqi@0: throw new Error("Bad owner of boostrap method"); aoqi@0: } aoqi@0: aoqi@0: if (!bsm_ref.getNameAndTypeInfo().getName().equals("bsm")) { aoqi@0: throw new Error("Bad boostrap method name"); aoqi@0: } aoqi@0: aoqi@0: if (!bsm_ref.getNameAndTypeInfo() aoqi@0: .getType().equals(asBSMSignatureString())) { aoqi@0: throw new Error("Bad boostrap method type" + aoqi@0: bsm_ref.getNameAndTypeInfo().getType() + " " + aoqi@0: asBSMSignatureString()); aoqi@0: } aoqi@0: aoqi@0: LineNumberTable_attribute lnt = aoqi@0: (LineNumberTable_attribute)ea.attributes.get(Attribute.LineNumberTable); aoqi@0: aoqi@0: if (lnt == null) { aoqi@0: throw new Error("No LineNumberTable attribute"); aoqi@0: } aoqi@0: if (lnt.line_number_table_length != 3) { aoqi@0: throw new Error("Wrong number of entries in LineNumberTable"); aoqi@0: } aoqi@0: } catch (Exception e) { aoqi@0: e.printStackTrace(); aoqi@0: throw new Error("error reading " + compiledTest +": " + e); aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: String asBSMSignatureString() { aoqi@0: StringBuilder buf = new StringBuilder(); aoqi@0: buf.append("(Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;"); aoqi@0: for (StaticArgumentKind sak : saks) { aoqi@0: buf.append(sak.bytecodeTypeStr); aoqi@0: } aoqi@0: buf.append(")Ljava/lang/invoke/CallSite;"); aoqi@0: return buf.toString(); aoqi@0: } aoqi@0: aoqi@0: class JavaSource extends SimpleJavaFileObject { aoqi@0: aoqi@0: static final String source_template = "import java.lang.invoke.*;\n" + aoqi@0: "class Bootstrap {\n" + aoqi@0: " public static CallSite bsm(MethodHandles.Lookup lookup, " + aoqi@0: "String name, MethodType methodType #SARGS) {\n" + aoqi@0: " return null;\n" + aoqi@0: " }\n" + aoqi@0: "}\n" + aoqi@0: "class Test#ID {\n" + aoqi@0: " void m() { }\n" + aoqi@0: " void test() {\n" + aoqi@0: " Object o = this; // marker statement \n" + aoqi@0: " m();\n" + aoqi@0: " }\n" + aoqi@0: "}"; aoqi@0: aoqi@0: String source; aoqi@0: aoqi@0: JavaSource(int id) { aoqi@0: super(URI.create("myfo:/Test.java"), JavaFileObject.Kind.SOURCE); aoqi@0: source = source_template.replace("#SARGS", asSignatureString()) aoqi@0: .replace("#ID", String.valueOf(id)); aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public CharSequence getCharContent(boolean ignoreEncodingErrors) { aoqi@0: return source; aoqi@0: } aoqi@0: aoqi@0: String asSignatureString() { aoqi@0: int count = 0; aoqi@0: StringBuilder buf = new StringBuilder(); aoqi@0: for (StaticArgumentKind sak : saks) { aoqi@0: buf.append(","); aoqi@0: buf.append(sak.sourceTypeStr); aoqi@0: buf.append(' '); aoqi@0: buf.append(String.format("x%d", count++)); aoqi@0: } aoqi@0: return buf.toString(); aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: class Indifier extends TreeScanner implements TaskListener { aoqi@0: aoqi@0: MethodSymbol bsm; aoqi@0: Symtab syms; aoqi@0: Names names; aoqi@0: Types types; aoqi@0: aoqi@0: Indifier(Symtab syms, Names names, Types types) { aoqi@0: this.syms = syms; aoqi@0: this.names = names; aoqi@0: this.types = types; aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public void started(TaskEvent e) { aoqi@0: //do nothing aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public void finished(TaskEvent e) { aoqi@0: if (e.getKind() == TaskEvent.Kind.ANALYZE) { aoqi@0: scan(e.getCompilationUnit(), null); aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public Void visitMethodInvocation(MethodInvocationTree node, Void p) { aoqi@0: super.visitMethodInvocation(node, p); aoqi@0: JCMethodInvocation apply = (JCMethodInvocation)node; aoqi@0: JCIdent ident = (JCIdent)apply.meth; aoqi@0: Symbol oldSym = ident.sym; aoqi@0: if (!oldSym.isConstructor()) { aoqi@0: Object[] staticArgs = new Object[arity.arity]; aoqi@0: for (int i = 0; i < arity.arity ; i++) { aoqi@0: staticArgs[i] = saks[i].getValue(syms, names, types); aoqi@0: } aoqi@0: ident.sym = new Symbol.DynamicMethodSymbol(oldSym.name, aoqi@0: oldSym.owner, REF_invokeStatic, bsm, oldSym.type, staticArgs); aoqi@0: } aoqi@0: return null; aoqi@0: } aoqi@0: aoqi@0: @Override aoqi@0: public Void visitMethod(MethodTree node, Void p) { aoqi@0: super.visitMethod(node, p); aoqi@0: if (node.getName().toString().equals("bsm")) { aoqi@0: bsm = ((JCMethodDecl)node).sym; aoqi@0: } aoqi@0: return null; aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: static class DiagChecker aoqi@0: implements javax.tools.DiagnosticListener { aoqi@0: aoqi@0: boolean diagFound; aoqi@0: ArrayList diags = new ArrayList<>(); aoqi@0: aoqi@0: public void report(Diagnostic diagnostic) { aoqi@0: diags.add(diagnostic.getMessage(Locale.getDefault())); aoqi@0: diagFound = true; aoqi@0: } aoqi@0: aoqi@0: String printDiags() { aoqi@0: StringBuilder buf = new StringBuilder(); aoqi@0: for (String s : diags) { aoqi@0: buf.append(s); aoqi@0: buf.append("\n"); aoqi@0: } aoqi@0: return buf.toString(); aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: }