jdk8-mips64-public/langtools: test/tools/javac/lambda/TestInvokeDynamic.java@26d93df3905a (annotated)

test/tools/javac/lambda/TestInvokeDynamic.java@26d93df3905a (annotated)

test/tools/javac/lambda/TestInvokeDynamic.java

Tue, 25 Sep 2012 11:53:18 +0100

author: mcimadamore
date: Tue, 25 Sep 2012 11:53:18 +0100
changeset 1336: 26d93df3905a
child 1343: f1e6b361a329
permissions: -rw-r--r--

7194586: Add back-end support for invokedynamic
Summary: Add support for invokedynamic bytecode instruction; includes suppot for generation of all related classfile attributes
Reviewed-by: jjg

 /*
  * Copyright (c) 2012, 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
  *
  * @summary Add back-end support for invokedynamic
  *
  */
 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.Method;
 import com.sun.tools.javac.api.JavacTaskImpl;
 import com.sun.tools.javac.api.JavacTool;
 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.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 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.JavaCompiler;
 import javax.tools.JavaFileManager;
 import javax.tools.JavaFileObject;
 import javax.tools.SimpleJavaFileObject;
 import javax.tools.StandardJavaFileManager;
 import javax.tools.ToolProvider;
 import static com.sun.tools.javac.jvm.ClassFile.*;
 public class TestInvokeDynamic {
     static int checkCount = 0;
     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) {
             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, names);
                 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 {
         // Create a single file manager and compiler and reuse it for each compile to save time.
         StandardJavaFileManager fm = JavacTool.create().getStandardFileManager(null, null, null);
         final JavaCompiler tool = ToolProvider.getSystemJavaCompiler();
         for (StaticArgumentsArity arity : StaticArgumentsArity.values()) {
             if (arity.arity == 0) {
                 new TestInvokeDynamic(arity).compileAndCheck(fm, tool);
             } else {
                 for (StaticArgumentKind sak1 : StaticArgumentKind.values()) {
                     if (arity.arity == 1) {
                         new TestInvokeDynamic(arity, sak1).compileAndCheck(fm, tool);
                     } else {
                         for (StaticArgumentKind sak2 : StaticArgumentKind.values()) {
                             if (arity.arity == 2) {
                                 new TestInvokeDynamic(arity, sak1, sak2).compileAndCheck(fm, tool);
                             } else {
                                 for (StaticArgumentKind sak3 : StaticArgumentKind.values()) {
                                     new TestInvokeDynamic(arity, sak1, sak2, sak3).compileAndCheck(fm, tool);
                                 }
                             }
                         }
                     }
                 }
             }
         }
         System.out.println("Total checks made: " + checkCount);
     }
     StaticArgumentsArity arity;
     StaticArgumentKind[] saks;
     JavaSource source;
     DiagChecker dc;
     TestInvokeDynamic(StaticArgumentsArity arity, StaticArgumentKind... saks) {
         this.arity = arity;
         this.saks = saks;
         source = new JavaSource();
         dc = new DiagChecker();
     }
     void compileAndCheck(JavaFileManager fm, JavaCompiler tool) throws Exception {
         JavacTaskImpl ct = (JavacTaskImpl)tool.getTask(null, fm, dc,
                 null, null, Arrays.asList(source));
         Context context = ct.getContext();
         Symtab syms = Symtab.instance(context);
         Names names = Names.instance(context);
         ct.addTaskListener(new Indifier(syms, names));
         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();
     }
     void verifyBytecode() {
         File compiledTest = new File("Test.class");
         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());
             }
         } 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 {\n" +
                 "   void m() { }\n" +
                 "   void test() { m(); }\n" +
                 "}";
         String source;
         JavaSource() {
             super(URI.create("myfo:/Test.java"), JavaFileObject.Kind.SOURCE);
             source = source_template.replace("#SARGS", asSignatureString());
         }
         @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<Void, Void> implements TaskListener {
         MethodSymbol bsm;
         Symtab syms;
         Names names;
         Indifier(Symtab syms, Names names) {
             this.syms = syms;
             this.names = names;
         }
         @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);
                 }
                 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<JavaFileObject> {
         boolean diagFound;
         ArrayList<String> diags = new ArrayList<>();
         public void report(Diagnostic<? extends JavaFileObject> 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();
         }
     }
 }

Mercurial > jdk8-mips64-public > langtools / annotate

test/tools/javac/lambda/TestInvokeDynamic.java@26d93df3905a (annotated)

test/tools/javac/lambda/TestInvokeDynamic.java