1 /*

     2  * Copyright (c) 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  * Asm plugin testing.

    26  * @test

    27  * @summary Test basic functionality.

    28  * @author Jean-Francois Denise

    29  * @modules java.base/jdk.internal.org.objectweb.asm

    30  *          jdk.jlink/jdk.tools.jlink.internal

    31  *          jdk.jlink/jdk.tools.jlink.internal.plugins.asm

    32  * @build AsmPluginTestBase

    33  * @run main IdentityPluginTest

    34  */

    35 

    36 import java.io.ByteArrayInputStream;

    37 import java.io.IOException;

    38 import java.io.UncheckedIOException;

    39 

    40 import jdk.internal.org.objectweb.asm.ClassReader;

    41 import jdk.internal.org.objectweb.asm.ClassVisitor;

    42 import jdk.internal.org.objectweb.asm.ClassWriter;

    43 import jdk.internal.org.objectweb.asm.Opcodes;

    44 import jdk.tools.jlink.internal.plugins.asm.AsmPool.WritableClassPool;

    45 import jdk.tools.jlink.plugin.ModuleEntry;

    46 import jdk.tools.jlink.plugin.ModulePool;

    47 

    48 public class IdentityPluginTest extends AsmPluginTestBase {

    49 

    50     public static void main(String[] args) throws Exception {

    51         if (!isImageBuild()) {

    52             System.err.println("Test not run. Not image build.");

    53             return;

    54         }

    55         new IdentityPluginTest().test();

    56     }

    57 

    58     public void test() throws Exception {

    59         IdentityPlugin asm = new IdentityPlugin();

    60         ModulePool resourcePool = asm.visit(getPool());

    61         asm.test(getPool(), resourcePool);

    62     }

    63 

    64     private class IdentityPlugin extends TestPlugin {

    65 

    66         @Override

    67         public void visit() {

    68             for (ModuleEntry res : getPools().getGlobalPool().getClasses()) {

    69                 if (res.getPath().endsWith("module-info.class")) {

    70                     continue;

    71                 }

    72                 ClassReader reader = getPools().getGlobalPool().getClassReader(res);

    73                 ClassWriter writer = new ClassWriter(reader, ClassWriter.COMPUTE_FRAMES);

    74                 IdentityClassVisitor visitor = new IdentityClassVisitor(writer);

    75                 reader.accept(visitor, ClassReader.EXPAND_FRAMES);

    76                 getPools().getGlobalPool().getTransformedClasses().addClass(writer);

    77             }

    78         }

    79 

    80         @Override

    81         public void test(ModulePool inResources, ModulePool outResources) throws IOException {

    82             if (outResources.isEmpty()) {

    83                 throw new AssertionError("Empty result");

    84             }

    85             if (!isVisitCalled()) {

    86                 throw new AssertionError("Resources not visited");

    87             }

    88             WritableClassPool transformedClasses = getPools().getGlobalPool().getTransformedClasses();

    89             if (transformedClasses.getClasses().size() != getClasses().size()) {

    90                 throw new AssertionError("Number of transformed classes not equal to expected");

    91             }

    92             for (String className : getClasses()) {

    93                 if (transformedClasses.getClassReader(className) == null) {

    94                     throw new AssertionError("Class not transformed " + className);

    95                 }

    96             }

    97             outResources.entries().forEach(r -> {

    98                 if (r.getPath().endsWith(".class") && !r.getPath().endsWith("module-info.class")) {

    99                     try {

   100                         ClassReader reader = new ClassReader(new ByteArrayInputStream(r.getBytes()));

   101                         ClassWriter w = new ClassWriter(reader, ClassWriter.COMPUTE_FRAMES);

   102                         reader.accept(w, ClassReader.EXPAND_FRAMES);

   103                     } catch (IOException exp) {

   104                         throw new UncheckedIOException(exp);

   105                     }

   106                 }

   107             });

   108         }

   109 

   110         @Override

   111         public String getName() {

   112             return "identity-plugin";

   113         }

   114     }

   115 

   116     private static class IdentityClassVisitor extends ClassVisitor {

   117         public IdentityClassVisitor(ClassWriter cv) {

   118             super(Opcodes.ASM5, cv);

   119         }

   120     }

   121 }