diff -r 000000000000 -r 959103a6100f test/tools/javac/lambda/TestInvokeDynamic.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/tools/javac/lambda/TestInvokeDynamic.java Wed Apr 27 01:34:52 2016 +0800 @@ -0,0 +1,494 @@ +/* + * Copyright (c) 2012, 2013, 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. + */ + +/* + * @test + * @bug 7194586 8003280 8006694 8010404 + * @summary Add lambda tests + * Add back-end support for invokedynamic + * temporarily workaround combo tests are causing time out in several platforms + * @library ../lib + * @build JavacTestingAbstractThreadedTest + * @run main/othervm TestInvokeDynamic + */ + +// use /othervm to avoid jtreg timeout issues (CODETOOLS-7900047) +// see JDK-8006746 + +import java.io.File; +import java.net.URI; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.Locale; + +import javax.tools.Diagnostic; +import javax.tools.JavaFileObject; +import javax.tools.SimpleJavaFileObject; + +import com.sun.source.tree.MethodInvocationTree; +import com.sun.source.tree.MethodTree; +import com.sun.source.util.TaskEvent; +import com.sun.source.util.TaskListener; +import com.sun.source.util.TreeScanner; + +import com.sun.tools.classfile.Attribute; +import com.sun.tools.classfile.BootstrapMethods_attribute; +import com.sun.tools.classfile.ClassFile; +import com.sun.tools.classfile.Code_attribute; +import com.sun.tools.classfile.ConstantPool.*; +import com.sun.tools.classfile.Instruction; +import com.sun.tools.classfile.LineNumberTable_attribute; +import com.sun.tools.classfile.Method; + +import com.sun.tools.javac.api.JavacTaskImpl; +import com.sun.tools.javac.code.Symbol; +import com.sun.tools.javac.code.Symbol.MethodSymbol; +import com.sun.tools.javac.code.Symtab; +import com.sun.tools.javac.code.Types; +import com.sun.tools.javac.jvm.Pool; +import com.sun.tools.javac.tree.JCTree.JCMethodInvocation; +import com.sun.tools.javac.tree.JCTree.JCMethodDecl; +import com.sun.tools.javac.tree.JCTree.JCIdent; +import com.sun.tools.javac.util.Context; +import com.sun.tools.javac.util.Names; + +import static com.sun.tools.javac.jvm.ClassFile.*; + +public class TestInvokeDynamic + extends JavacTestingAbstractThreadedTest + implements Runnable { + + enum StaticArgumentKind { + STRING("Hello!", "String", "Ljava/lang/String;") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + return (cpInfo instanceof CONSTANT_String_info) && + ((CONSTANT_String_info)cpInfo).getString() + .equals(value); + } + }, + CLASS(null, "Class", "Ljava/lang/Class;") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + return (cpInfo instanceof CONSTANT_Class_info) && + ((CONSTANT_Class_info)cpInfo).getName() + .equals("java/lang/String"); + } + }, + INTEGER(1, "int", "I") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + return (cpInfo instanceof CONSTANT_Integer_info) && + ((CONSTANT_Integer_info)cpInfo).value == + ((Integer)value).intValue(); + } + }, + LONG(1L, "long", "J") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + return (cpInfo instanceof CONSTANT_Long_info) && + ((CONSTANT_Long_info)cpInfo).value == + ((Long)value).longValue(); + } + }, + FLOAT(1.0f, "float", "F") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + return (cpInfo instanceof CONSTANT_Float_info) && + ((CONSTANT_Float_info)cpInfo).value == + ((Float)value).floatValue(); + } + }, + DOUBLE(1.0, "double","D") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + return (cpInfo instanceof CONSTANT_Double_info) && + ((CONSTANT_Double_info)cpInfo).value == + ((Double)value).doubleValue(); + } + }, + METHOD_HANDLE(null, "MethodHandle", "Ljava/lang/invoke/MethodHandle;") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + if (!(cpInfo instanceof CONSTANT_MethodHandle_info)) + return false; + CONSTANT_MethodHandle_info handleInfo = + (CONSTANT_MethodHandle_info)cpInfo; + return handleInfo.getCPRefInfo().getClassName().equals("Array") && + handleInfo.reference_kind == RefKind.REF_invokeVirtual && + handleInfo.getCPRefInfo() + .getNameAndTypeInfo().getName().equals("clone") && + handleInfo.getCPRefInfo() + .getNameAndTypeInfo().getType().equals("()Ljava/lang/Object;"); + } + }, + METHOD_TYPE(null, "MethodType", "Ljava/lang/invoke/MethodType;") { + @Override + boolean check(CPInfo cpInfo) throws Exception { + return (cpInfo instanceof CONSTANT_MethodType_info) && + ((CONSTANT_MethodType_info)cpInfo).getType() + .equals("()Ljava/lang/Object;"); + } + }; + + Object value; + String sourceTypeStr; + String bytecodeTypeStr; + + StaticArgumentKind(Object value, String sourceTypeStr, + String bytecodeTypeStr) { + this.value = value; + this.sourceTypeStr = sourceTypeStr; + this.bytecodeTypeStr = bytecodeTypeStr; + } + + abstract boolean check(CPInfo cpInfo) throws Exception; + + Object getValue(Symtab syms, Names names, Types types) { + switch (this) { + case STRING: + case INTEGER: + case LONG: + case FLOAT: + case DOUBLE: + return value; + case CLASS: + return syms.stringType.tsym; + case METHOD_HANDLE: + return new Pool.MethodHandle(REF_invokeVirtual, + syms.arrayCloneMethod, types); + case METHOD_TYPE: + return syms.arrayCloneMethod.type; + default: + throw new AssertionError(); + } + } + } + + enum StaticArgumentsArity { + ZERO(0), + ONE(1), + TWO(2), + THREE(3); + + int arity; + + StaticArgumentsArity(int arity) { + this.arity = arity; + } + } + + public static void main(String... args) throws Exception { + for (StaticArgumentsArity arity : StaticArgumentsArity.values()) { + if (arity.arity == 0) { + pool.execute(new TestInvokeDynamic(arity)); + } else { + for (StaticArgumentKind sak1 : StaticArgumentKind.values()) { + if (arity.arity == 1) { + pool.execute(new TestInvokeDynamic(arity, sak1)); + } else { + for (StaticArgumentKind sak2 : StaticArgumentKind.values()) { + if (arity.arity == 2) { + pool.execute(new TestInvokeDynamic(arity, sak1, sak2)); + } else { + for (StaticArgumentKind sak3 : StaticArgumentKind.values()) { + pool.execute( + new TestInvokeDynamic(arity, sak1, sak2, sak3)); + } + } + } + } + } + } + } + + checkAfterExec(); + } + + StaticArgumentsArity arity; + StaticArgumentKind[] saks; + DiagChecker dc; + + TestInvokeDynamic(StaticArgumentsArity arity, StaticArgumentKind... saks) { + this.arity = arity; + this.saks = saks; + dc = new DiagChecker(); + } + + public void run() { + int id = checkCount.incrementAndGet(); + JavaSource source = new JavaSource(id); + JavacTaskImpl ct = (JavacTaskImpl)comp.getTask(null, fm.get(), dc, + Arrays.asList("-g"), null, Arrays.asList(source)); + Context context = ct.getContext(); + Symtab syms = Symtab.instance(context); + Names names = Names.instance(context); + Types types = Types.instance(context); + ct.addTaskListener(new Indifier(syms, names, types)); + try { + ct.generate(); + } catch (Throwable t) { + t.printStackTrace(); + throw new AssertionError( + String.format("Error thrown when compiling following code\n%s", + source.source)); + } + if (dc.diagFound) { + throw new AssertionError( + String.format("Diags found when compiling following code\n%s\n\n%s", + source.source, dc.printDiags())); + } + verifyBytecode(id); + } + + void verifyBytecode(int id) { + File compiledTest = new File(String.format("Test%d.class", id)); + try { + ClassFile cf = ClassFile.read(compiledTest); + Method testMethod = null; + for (Method m : cf.methods) { + if (m.getName(cf.constant_pool).equals("test")) { + testMethod = m; + break; + } + } + if (testMethod == null) { + throw new Error("Test method not found"); + } + Code_attribute ea = + (Code_attribute)testMethod.attributes.get(Attribute.Code); + if (testMethod == null) { + throw new Error("Code attribute for test() method not found"); + } + + int bsmIdx = -1; + + for (Instruction i : ea.getInstructions()) { + if (i.getMnemonic().equals("invokedynamic")) { + CONSTANT_InvokeDynamic_info indyInfo = + (CONSTANT_InvokeDynamic_info)cf + .constant_pool.get(i.getShort(1)); + bsmIdx = indyInfo.bootstrap_method_attr_index; + if (!indyInfo.getNameAndTypeInfo().getType().equals("()V")) { + throw new + AssertionError("type mismatch for CONSTANT_InvokeDynamic_info"); + } + } + } + if (bsmIdx == -1) { + throw new Error("Missing invokedynamic in generated code"); + } + + BootstrapMethods_attribute bsm_attr = + (BootstrapMethods_attribute)cf + .getAttribute(Attribute.BootstrapMethods); + if (bsm_attr.bootstrap_method_specifiers.length != 1) { + throw new Error("Bad number of method specifiers " + + "in BootstrapMethods attribute"); + } + BootstrapMethods_attribute.BootstrapMethodSpecifier bsm_spec = + bsm_attr.bootstrap_method_specifiers[0]; + + if (bsm_spec.bootstrap_arguments.length != arity.arity) { + throw new Error("Bad number of static invokedynamic args " + + "in BootstrapMethod attribute"); + } + + int count = 0; + for (StaticArgumentKind sak : saks) { + if (!sak.check(cf.constant_pool + .get(bsm_spec.bootstrap_arguments[count]))) { + throw new Error("Bad static argument value " + sak); + } + count++; + } + + CONSTANT_MethodHandle_info bsm_handle = + (CONSTANT_MethodHandle_info)cf.constant_pool + .get(bsm_spec.bootstrap_method_ref); + + if (bsm_handle.reference_kind != RefKind.REF_invokeStatic) { + throw new Error("Bad kind on boostrap method handle"); + } + + CONSTANT_Methodref_info bsm_ref = + (CONSTANT_Methodref_info)cf.constant_pool + .get(bsm_handle.reference_index); + + if (!bsm_ref.getClassInfo().getName().equals("Bootstrap")) { + throw new Error("Bad owner of boostrap method"); + } + + if (!bsm_ref.getNameAndTypeInfo().getName().equals("bsm")) { + throw new Error("Bad boostrap method name"); + } + + if (!bsm_ref.getNameAndTypeInfo() + .getType().equals(asBSMSignatureString())) { + throw new Error("Bad boostrap method type" + + bsm_ref.getNameAndTypeInfo().getType() + " " + + asBSMSignatureString()); + } + + LineNumberTable_attribute lnt = + (LineNumberTable_attribute)ea.attributes.get(Attribute.LineNumberTable); + + if (lnt == null) { + throw new Error("No LineNumberTable attribute"); + } + if (lnt.line_number_table_length != 3) { + throw new Error("Wrong number of entries in LineNumberTable"); + } + } catch (Exception e) { + e.printStackTrace(); + throw new Error("error reading " + compiledTest +": " + e); + } + } + + String asBSMSignatureString() { + StringBuilder buf = new StringBuilder(); + buf.append("(Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;"); + for (StaticArgumentKind sak : saks) { + buf.append(sak.bytecodeTypeStr); + } + buf.append(")Ljava/lang/invoke/CallSite;"); + return buf.toString(); + } + + class JavaSource extends SimpleJavaFileObject { + + static final String source_template = "import java.lang.invoke.*;\n" + + "class Bootstrap {\n" + + " public static CallSite bsm(MethodHandles.Lookup lookup, " + + "String name, MethodType methodType #SARGS) {\n" + + " return null;\n" + + " }\n" + + "}\n" + + "class Test#ID {\n" + + " void m() { }\n" + + " void test() {\n" + + " Object o = this; // marker statement \n" + + " m();\n" + + " }\n" + + "}"; + + String source; + + JavaSource(int id) { + super(URI.create("myfo:/Test.java"), JavaFileObject.Kind.SOURCE); + source = source_template.replace("#SARGS", asSignatureString()) + .replace("#ID", String.valueOf(id)); + } + + @Override + public CharSequence getCharContent(boolean ignoreEncodingErrors) { + return source; + } + + String asSignatureString() { + int count = 0; + StringBuilder buf = new StringBuilder(); + for (StaticArgumentKind sak : saks) { + buf.append(","); + buf.append(sak.sourceTypeStr); + buf.append(' '); + buf.append(String.format("x%d", count++)); + } + return buf.toString(); + } + } + + class Indifier extends TreeScanner implements TaskListener { + + MethodSymbol bsm; + Symtab syms; + Names names; + Types types; + + Indifier(Symtab syms, Names names, Types types) { + this.syms = syms; + this.names = names; + this.types = types; + } + + @Override + public void started(TaskEvent e) { + //do nothing + } + + @Override + public void finished(TaskEvent e) { + if (e.getKind() == TaskEvent.Kind.ANALYZE) { + scan(e.getCompilationUnit(), null); + } + } + + @Override + public Void visitMethodInvocation(MethodInvocationTree node, Void p) { + super.visitMethodInvocation(node, p); + JCMethodInvocation apply = (JCMethodInvocation)node; + JCIdent ident = (JCIdent)apply.meth; + Symbol oldSym = ident.sym; + if (!oldSym.isConstructor()) { + Object[] staticArgs = new Object[arity.arity]; + for (int i = 0; i < arity.arity ; i++) { + staticArgs[i] = saks[i].getValue(syms, names, types); + } + ident.sym = new Symbol.DynamicMethodSymbol(oldSym.name, + oldSym.owner, REF_invokeStatic, bsm, oldSym.type, staticArgs); + } + return null; + } + + @Override + public Void visitMethod(MethodTree node, Void p) { + super.visitMethod(node, p); + if (node.getName().toString().equals("bsm")) { + bsm = ((JCMethodDecl)node).sym; + } + return null; + } + } + + static class DiagChecker + implements javax.tools.DiagnosticListener { + + boolean diagFound; + ArrayList diags = new ArrayList<>(); + + public void report(Diagnostic diagnostic) { + diags.add(diagnostic.getMessage(Locale.getDefault())); + diagFound = true; + } + + String printDiags() { + StringBuilder buf = new StringBuilder(); + for (String s : diags) { + buf.append(s); + buf.append("\n"); + } + return buf.toString(); + } + } + +}