1 /*
   2  * Copyright (c) 2014, 2015, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug 8067422
  27  * @summary Check that the lambda names are not unnecessarily unstable
  28  * @library /tools/lib
  29  * @modules jdk.jdeps/com.sun.tools.classfile
  30  *          jdk.compiler/com.sun.tools.javac.api
  31  *          jdk.compiler/com.sun.tools.javac.file
  32  *          jdk.compiler/com.sun.tools.javac.main
  33  * @build ToolBox
  34  * @run main TestNonSerializableLambdaNameStability
  35  */
  36 
  37 import com.sun.tools.classfile.ClassFile;
  38 import com.sun.tools.classfile.Method;
  39 import java.io.ByteArrayInputStream;
  40 import java.io.InputStream;
  41 import java.util.ArrayList;
  42 import java.util.List;
  43 import javax.tools.StandardLocation;
  44 
  45 public class TestNonSerializableLambdaNameStability {
  46 
  47     public static void main(String... args) throws Exception {
  48         new TestNonSerializableLambdaNameStability().run();
  49     }
  50 
  51     String lambdaSource = "public class L%d {\n" +
  52                           "    public static class A {\n" +
  53                           "        private Runnable r = () -> { };\n" +
  54                           "    }\n" +
  55                           "    public static class B {\n" +
  56                           "        private Runnable r = () -> { };\n" +
  57                           "    }\n" +
  58                           "    private Runnable r = () -> { };\n" +
  59                           "}\n";
  60 
  61     String expectedLambdaMethodName = "lambda$new$0";
  62 
  63     void run() throws Exception {
  64         List<String> sources = new ArrayList<>();
  65 
  66         for (int i = 0; i < 3; i++) {
  67             sources.add(String.format(lambdaSource, i));
  68         }
  69 
  70         ToolBox tb = new ToolBox();
  71 
  72         try (ToolBox.MemoryFileManager fm = new ToolBox.MemoryFileManager()) {
  73             tb.new JavacTask()
  74               .sources(sources.toArray(new String[sources.size()]))
  75               .fileManager(fm)
  76               .run();
  77 
  78             for (String file : fm.files.get(StandardLocation.CLASS_OUTPUT).keySet()) {
  79                 byte[] fileBytes = fm.getFileBytes(StandardLocation.CLASS_OUTPUT, file);
  80                 try (InputStream in = new ByteArrayInputStream(fileBytes)) {
  81                     boolean foundLambdaMethod = false;
  82                     ClassFile cf = ClassFile.read(in);
  83                     StringBuilder seenMethods = new StringBuilder();
  84                     String sep = "";
  85                     for (Method m : cf.methods) {
  86                         String methodName = m.getName(cf.constant_pool);
  87                         if (expectedLambdaMethodName.equals(methodName)) {
  88                             foundLambdaMethod = true;
  89                             break;
  90                         }
  91                         seenMethods.append(sep);
  92                         seenMethods.append(methodName);
  93                         sep = ", ";
  94                     }
  95 
  96                     if (!foundLambdaMethod) {
  97                         throw new AbstractMethodError("Did not find the lambda method, " +
  98                                                       "found methods: " + seenMethods.toString());
  99                     }
 100                 }
 101             }
 102         }
 103     }
 104 }