| /* |
| * Copyright (c) 2013, 2015, 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 8011738 |
| * @author sogoel |
| * @summary Code translation test for Lambda expressions, method references |
| * @modules jdk.jdeps/com.sun.tools.classfile |
| * @run main ByteCodeTest |
| */ |
| |
| import com.sun.tools.classfile.Attribute; |
| import com.sun.tools.classfile.BootstrapMethods_attribute; |
| import com.sun.tools.classfile.ClassFile; |
| import com.sun.tools.classfile.ConstantPool; |
| import com.sun.tools.classfile.ConstantPoolException; |
| import com.sun.tools.classfile.ConstantPool.*; |
| |
| import java.io.BufferedWriter; |
| import java.io.File; |
| import java.io.FileWriter; |
| import java.io.IOException; |
| import java.io.PrintWriter; |
| import java.util.ArrayList; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.HashSet; |
| import java.util.List; |
| import java.util.Map; |
| |
| public class ByteCodeTest { |
| |
| static boolean IS_DEBUG = false; |
| public static void main(String[] args) { |
| File classFile = null; |
| int err = 0; |
| boolean verifyResult = false; |
| for(TestCases tc : TestCases.values()) { |
| classFile = getCompiledFile(tc.name(), tc.srcCode); |
| if(classFile == null) { // either testFile or classFile was not created |
| err++; |
| } else { |
| verifyResult = verifyClassFileAttributes(classFile, tc); |
| if(!verifyResult) |
| System.out.println("Bootstrap class file attributes did not match for " + tc.name()); |
| } |
| } |
| if(err > 0) |
| throw new RuntimeException("Found " + err + " found"); |
| else |
| System.out.println("Test passed"); |
| } |
| |
| private static boolean verifyClassFileAttributes(File classFile, TestCases tc) { |
| ClassFile c = null; |
| try { |
| c = ClassFile.read(classFile); |
| } catch (IOException | ConstantPoolException e) { |
| e.printStackTrace(); |
| } |
| ConstantPoolVisitor cpv = new ConstantPoolVisitor(c, c.constant_pool.size()); |
| Map<Integer, String> hm = cpv.getBSMMap(); |
| |
| List<String> expectedValList = tc.getExpectedArgValues(); |
| expectedValList.add(tc.bsmSpecifier.specifier); |
| if(!(hm.values().containsAll(new HashSet<String>(expectedValList)))) { |
| System.out.println("Values do not match"); |
| return false; |
| } |
| return true; |
| } |
| |
| private static File getCompiledFile(String fname, String srcString) { |
| File testFile = null, classFile = null; |
| boolean isTestFileCreated = true; |
| |
| try { |
| testFile = writeTestFile(fname+".java", srcString); |
| } catch(IOException ioe) { |
| isTestFileCreated = false; |
| System.err.println("fail to write" + ioe); |
| } |
| |
| if(isTestFileCreated) { |
| try { |
| classFile = compile(testFile); |
| } catch (Error err) { |
| System.err.println("fail compile. Source:\n" + srcString); |
| throw err; |
| } |
| } |
| return classFile; |
| } |
| |
| static File writeTestFile(String fname, String source) throws IOException { |
| File f = new File(fname); |
| PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(f))); |
| out.println(source); |
| out.close(); |
| return f; |
| } |
| |
| static File compile(File f) { |
| int rc = com.sun.tools.javac.Main.compile(new String[] { |
| "-g", f.getPath() }); |
| if (rc != 0) |
| throw new Error("compilation failed. rc=" + rc); |
| String path = f.getPath(); |
| return new File(path.substring(0, path.length() - 5) + ".class"); |
| } |
| |
| static void debugln(String str) { |
| if(IS_DEBUG) |
| System.out.println(str); |
| } |
| |
| enum BSMSpecifier { |
| SPECIFIER1("REF_invokeStatic java/lang/invoke/LambdaMetafactory metaFactory " + |
| "(Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;" + |
| "Ljava/lang/invoke/MethodHandle;Ljava/lang/invoke/MethodHandle;Ljava/lang/invoke/MethodType;)" + |
| "Ljava/lang/invoke/CallSite;"), |
| SPECIFIER2("REF_invokeStatic java/lang/invoke/LambdaMetafactory altMetaFactory " + |
| "(Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;" + |
| "[Ljava/lang/Object;)Ljava/lang/invoke/CallSite;"); |
| |
| String specifier; |
| private BSMSpecifier(String specifier) { |
| this.specifier = specifier; |
| } |
| } |
| |
| enum TestCases { |
| // Single line lambda expression |
| TC1("class TC1 {\n" + |
| " public static void main(String[] args) {\n" + |
| " Object o = (Runnable) () -> { System.out.println(\"hi\");};\n" + |
| " }\n"+ |
| "}", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface java/lang/Runnable run ()V"); |
| valList.add("REF_invokeStatic TC1 lambda$0 ()V"); |
| valList.add("()V"); |
| return valList; |
| } |
| }, |
| |
| // Lambda expression in a for loop |
| TC2("import java.util.*;\n" + |
| "public class TC2 {\n" + |
| " void TC2_test() {\n" + |
| " List<String> list = new ArrayList<>();\n" + |
| " list.add(\"A\");\n" + |
| " list.add(\"B\");\n" + |
| " list.stream().forEach( s -> { System.out.println(s); } );\n" + |
| " }\n" + |
| " public static void main(String[] args) {\n" + |
| " new TC2().TC2_test();\n" + |
| " }\n" + |
| "}", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface java/util/function/Consumer accept (Ljava/lang/Object;)V"); |
| valList.add("REF_invokeStatic TC2 lambda$0 (Ljava/lang/String;)V"); |
| valList.add("(Ljava/lang/String;)V"); |
| return valList; |
| } |
| }, |
| |
| // Lambda initializer |
| TC3("class TC3 {\n" + |
| " interface SAM {\n" + |
| " void m(int i);\n" + |
| " }\n" + |
| " SAM lambda_03 = (int pos) -> { };\n" + |
| "}", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface TC3$SAM m (I)V"); |
| valList.add("REF_invokeStatic TC3 lambda$0 (I)V"); |
| valList.add("(I)V"); |
| return valList; |
| } |
| }, |
| |
| // Array initializer |
| TC4("class TC4 {\n" + |
| " interface Block<T> {\n" + |
| " void m(T t);\n" + |
| " }\n" + |
| " void test1() {\n" + |
| " Block<?>[] arr1 = { t -> { }, t -> { } };\n" + |
| " }\n" + |
| "}", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface TC4$Block m (Ljava/lang/Object;)V"); |
| valList.add("REF_invokeStatic TC4 lambda$0 (Ljava/lang/Object;)V"); |
| valList.add("(Ljava/lang/Object;)V"); |
| valList.add("REF_invokeStatic TC4 lambda$1 (Ljava/lang/Object;)V"); |
| return valList; |
| } |
| }, |
| |
| //Lambda expression as a method arg |
| TC5("class TC5 {\n"+ |
| " interface MapFun<T,R> { R m( T n); }\n" + |
| " void meth( MapFun<String,Integer> mf ) {\n" + |
| " assert( mf.m(\"four\") == 4);\n" + |
| " }\n"+ |
| " void test(Integer i) {\n" + |
| " meth(s -> { Integer len = s.length(); return len; } );\n" + |
| " }\n"+ |
| "}", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface TC5$MapFun m (Ljava/lang/Object;)Ljava/lang/Object;"); |
| valList.add("REF_invokeStatic TC5 lambda$0 (Ljava/lang/String;)Ljava/lang/Integer;"); |
| valList.add("(Ljava/lang/String;)Ljava/lang/Integer;"); |
| return valList; |
| } |
| }, |
| |
| //Inner class of Lambda expression |
| TC6("class TC6 {\n" + |
| " interface MapFun<T, R> { R m( T n); }\n" + |
| " MapFun<Class<?>,String> cs;\n" + |
| " void test() {\n" + |
| " cs = c -> {\n" + |
| " class innerClass {\n" + |
| " Class<?> icc;\n" + |
| " innerClass(Class<?> _c) { icc = _c; }\n" + |
| " String getString() { return icc.toString(); }\n" + |
| " }\n" + |
| " return new innerClass(c).getString();\n"+ |
| " };\n" + |
| " }\n" + |
| "}\n", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface TC6$MapFun m (Ljava/lang/Object;)Ljava/lang/Object;"); |
| valList.add("REF_invokeSpecial TC6 lambda$0 (Ljava/lang/Class;)Ljava/lang/String;"); |
| valList.add("(Ljava/lang/Class;)Ljava/lang/String;"); |
| return valList; |
| } |
| }, |
| |
| // Method reference |
| TC7("class TC7 {\n" + |
| " static interface SAM {\n" + |
| " void m(Integer i);\n" + |
| " }\n" + |
| " void m(Integer i) {}\n" + |
| " SAM s = this::m;\n" + |
| "}\n", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface TC7$SAM m (Ljava/lang/Integer;)V"); |
| valList.add("REF_invokeVirtual TC7 m (Ljava/lang/Integer;)V"); |
| valList.add("(Ljava/lang/Integer;)V"); |
| return valList; |
| } |
| }, |
| |
| // Constructor reference |
| TC8("public class TC8 {\n" + |
| " static interface A {Fee<String> m();}\n" + |
| " static class Fee<T> {\n" + |
| " private T t;\n" + |
| " public Fee() {}\n" + |
| " }\n" + |
| " public static void main(String[] args) {\n" + |
| " A a = Fee<String>::new; \n" + |
| " }\n" + |
| "}\n", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface TC8$A m ()LTC8$Fee;"); |
| valList.add("REF_newInvokeSpecial TC8$Fee <init> ()V"); |
| valList.add("()LTC8$Fee;"); |
| return valList; |
| } |
| }, |
| |
| // Recursive lambda expression |
| TC9("class TC9 {\n" + |
| " interface Recursive<T, R> { T apply(R n); };\n" + |
| " Recursive<Integer,Integer> factorial;\n" + |
| " void test(Integer j) {\n" + |
| " factorial = i -> { return i == 0 ? 1 : i * factorial.apply( i - 1 ); };\n" + |
| " }\n" + |
| "}\n", BSMSpecifier.SPECIFIER1) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeInterface TC9$Recursive apply (Ljava/lang/Object;)Ljava/lang/Object;"); |
| valList.add("REF_invokeSpecial TC9 lambda$0 (Ljava/lang/Integer;)Ljava/lang/Integer;"); |
| valList.add("(Ljava/lang/Integer;)Ljava/lang/Integer;"); |
| return valList; |
| } |
| }, |
| |
| //Serializable Lambda |
| TC10("import java.io.Serializable;\n" + |
| "class TC10 {\n" + |
| " interface Foo { int m(); }\n" + |
| " public static void main(String[] args) {\n" + |
| " Foo f1 = (Foo & Serializable)() -> 3;\n" + |
| " }\n" + |
| "}\n", BSMSpecifier.SPECIFIER2) { |
| |
| @Override |
| List<String> getExpectedArgValues() { |
| List<String> valList = new ArrayList<>(); |
| valList.add("REF_invokeStatic java/lang/invoke/LambdaMetafactory altMetaFactory (Ljava/lang/invoke/MethodHandles$Lookup;Ljava/lang/String;Ljava/lang/invoke/MethodType;[Ljava/lang/Object;)Ljava/lang/invoke/CallSite;"); |
| valList.add("REF_invokeInterface TC10$Foo m ()I"); |
| valList.add("REF_invokeStatic TC10 lambda$main$3231c38a$0 ()I"); |
| valList.add("()I"); |
| valList.add("1"); |
| return valList; |
| } |
| }; |
| |
| String srcCode; |
| BSMSpecifier bsmSpecifier; |
| |
| TestCases(String src, BSMSpecifier bsmSpecifier) { |
| this.srcCode = src; |
| // By default, all test cases will have bootstrap method specifier as Lambda.MetaFactory |
| // For serializable lambda test cases, bootstrap method specifier changed to altMetaFactory |
| this.bsmSpecifier = bsmSpecifier; |
| } |
| |
| List<String> getExpectedArgValues() { |
| return null; |
| } |
| |
| void setSrcCode(String src) { |
| srcCode = src; |
| } |
| } |
| |
| static class ConstantPoolVisitor implements ConstantPool.Visitor<String, Integer> { |
| final List<String> slist; |
| final ClassFile cf; |
| final ConstantPool cfpool; |
| final Map<Integer, String> bsmMap; |
| |
| |
| public ConstantPoolVisitor(ClassFile cf, int size) { |
| slist = new ArrayList<>(size); |
| for (int i = 0 ; i < size; i++) { |
| slist.add(null); |
| } |
| this.cf = cf; |
| this.cfpool = cf.constant_pool; |
| bsmMap = readBSM(); |
| } |
| |
| public Map<Integer, String> getBSMMap() { |
| return Collections.unmodifiableMap(bsmMap); |
| } |
| |
| public String visit(CPInfo c, int index) { |
| return c.accept(this, index); |
| } |
| |
| private Map<Integer, String> readBSM() { |
| BootstrapMethods_attribute bsmAttr = |
| (BootstrapMethods_attribute) cf.getAttribute(Attribute.BootstrapMethods); |
| if (bsmAttr != null) { |
| Map<Integer, String> out = |
| new HashMap<>(bsmAttr.bootstrap_method_specifiers.length); |
| for (BootstrapMethods_attribute.BootstrapMethodSpecifier bsms : |
| bsmAttr.bootstrap_method_specifiers) { |
| int index = bsms.bootstrap_method_ref; |
| try { |
| String value = slist.get(index); |
| if (value == null) { |
| value = visit(cfpool.get(index), index); |
| debugln("[SG]: index " + index); |
| debugln("[SG]: value " + value); |
| slist.set(index, value); |
| out.put(index, value); |
| } |
| for (int idx : bsms.bootstrap_arguments) { |
| value = slist.get(idx); |
| if (value == null) { |
| value = visit(cfpool.get(idx), idx); |
| debugln("[SG]: idx " + idx); |
| debugln("[SG]: value " + value); |
| slist.set(idx, value); |
| out.put(idx, value); |
| } |
| } |
| } catch (InvalidIndex ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return out; |
| } |
| return new HashMap<>(0); |
| } |
| |
| @Override |
| public String visitClass(CONSTANT_Class_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.name_index), c.name_index); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitDouble(CONSTANT_Double_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| value = Double.toString(c.value); |
| slist.set(p, value); |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitFieldref(CONSTANT_Fieldref_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.class_index), c.class_index); |
| value = value.concat(" " + visit(cfpool.get(c.name_and_type_index), |
| c.name_and_type_index)); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitFloat(CONSTANT_Float_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| value = Float.toString(c.value); |
| slist.set(p, value); |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitInteger(CONSTANT_Integer_info cnstnt, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| value = Integer.toString(cnstnt.value); |
| slist.set(p, value); |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitInterfaceMethodref(CONSTANT_InterfaceMethodref_info c, |
| Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.class_index), c.class_index); |
| value = value.concat(" " + |
| visit(cfpool.get(c.name_and_type_index), |
| c.name_and_type_index)); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitInvokeDynamic(CONSTANT_InvokeDynamic_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = bsmMap.get(c.bootstrap_method_attr_index) + " " |
| + visit(cfpool.get(c.name_and_type_index), c.name_and_type_index); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitLong(CONSTANT_Long_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| value = Long.toString(c.value); |
| slist.set(p, value); |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitNameAndType(CONSTANT_NameAndType_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.name_index), c.name_index); |
| value = value.concat(" " + |
| visit(cfpool.get(c.type_index), c.type_index)); |
| slist.set(p, value); |
| } catch (InvalidIndex ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitMethodref(CONSTANT_Methodref_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.class_index), c.class_index); |
| value = value.concat(" " + |
| visit(cfpool.get(c.name_and_type_index), |
| c.name_and_type_index)); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitMethodHandle(CONSTANT_MethodHandle_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = c.reference_kind.name(); |
| value = value.concat(" " |
| + visit(cfpool.get(c.reference_index), c.reference_index)); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitMethodType(CONSTANT_MethodType_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.descriptor_index), c.descriptor_index); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitModule(CONSTANT_Module_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.name_index), c.name_index); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitPackage(CONSTANT_Package_info c, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| try { |
| value = visit(cfpool.get(c.name_index), c.name_index); |
| slist.set(p, value); |
| } catch (ConstantPoolException ex) { |
| ex.printStackTrace(); |
| } |
| } |
| return value; |
| } |
| |
| @Override |
| public String visitString(CONSTANT_String_info c, Integer p) { |
| |
| try { |
| String value = slist.get(p); |
| if (value == null) { |
| value = c.getString(); |
| slist.set(p, value); |
| } |
| return value; |
| } catch (ConstantPoolException ex) { |
| throw new RuntimeException("Fatal error", ex); |
| } |
| } |
| |
| @Override |
| public String visitUtf8(CONSTANT_Utf8_info cnstnt, Integer p) { |
| |
| String value = slist.get(p); |
| if (value == null) { |
| value = cnstnt.value; |
| slist.set(p, value); |
| } |
| return value; |
| } |
| } |
| } |
| |