1 /*
   2  * Copyright (c) 2015, 2016, 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 8136421
  27  * @requires vm.jvmci
  28  * @library /test/lib /
  29  * @library ../common/patches
  30  * @modules java.base/jdk.internal.misc
  31  *          java.base/jdk.internal.reflect
  32  *          java.base/jdk.internal.org.objectweb.asm
  33  *          jdk.internal.vm.ci/jdk.vm.ci.hotspot
  34  *          jdk.internal.vm.ci/jdk.vm.ci.runtime
  35  *          jdk.internal.vm.ci/jdk.vm.ci.meta
  36  *
  37  * @build jdk.internal.vm.ci/jdk.vm.ci.hotspot.CompilerToVMHelper sun.hotspot.WhiteBox
  38  * @run driver ClassFileInstaller sun.hotspot.WhiteBox
  39  *                                sun.hotspot.WhiteBox$WhiteBoxPermission
  40  * @run main/othervm -Xbootclasspath/a:.
  41  *                   -XX:+UnlockExperimentalVMOptions -XX:+EnableJVMCI
  42  *                   -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI
  43  *                   compiler.jvmci.compilerToVM.ResolveConstantInPoolTest
  44  */
  45 
  46 package compiler.jvmci.compilerToVM;
  47 
  48 import compiler.jvmci.compilerToVM.ConstantPoolTestCase.ConstantTypes;
  49 import compiler.jvmci.compilerToVM.ConstantPoolTestCase.Validator;
  50 import compiler.jvmci.compilerToVM.ConstantPoolTestsHelper.DummyClasses;
  51 import jdk.test.lib.Asserts;
  52 import jdk.vm.ci.hotspot.CompilerToVMHelper;
  53 import jdk.vm.ci.meta.ConstantPool;
  54 
  55 import java.lang.invoke.MethodHandle;
  56 import java.lang.invoke.MethodType;
  57 import java.util.HashMap;
  58 import java.util.Map;
  59 
  60 import static compiler.jvmci.compilerToVM.ConstantPoolTestCase.ConstantTypes.CONSTANT_METHODHANDLE;
  61 import static compiler.jvmci.compilerToVM.ConstantPoolTestCase.ConstantTypes.CONSTANT_METHODTYPE;
  62 
  63 /**
  64  * Test for {@code jdk.vm.ci.hotspot.CompilerToVM.resolveConstantInPool} method
  65  */
  66 public class ResolveConstantInPoolTest {
  67 
  68     private static final String NOT_NULL_MSG
  69             = "Object returned by resolveConstantInPool method should not be null";
  70 
  71     public static void main(String[] args) throws Exception {
  72         Map<ConstantTypes, Validator> typeTests = new HashMap<>();
  73         typeTests.put(CONSTANT_METHODHANDLE, ResolveConstantInPoolTest::validateMethodHandle);
  74         typeTests.put(CONSTANT_METHODTYPE, ResolveConstantInPoolTest::validateMethodType);
  75         ConstantPoolTestCase testCase = new ConstantPoolTestCase(typeTests);
  76         testCase.test();
  77         // The next "Class.forName" and repeating "testCase.test()"
  78         // are here for the following reason.
  79         // The first test run is without dummy class initialization,
  80         // which means no constant pool cache exists.
  81         // The second run is with initialized class (with constant pool cache available).
  82         // Some CompilerToVM methods require different input
  83         // depending on whether CP cache exists or not.
  84         for (DummyClasses dummy : DummyClasses.values()) {
  85             Class.forName(dummy.klass.getName());
  86         }
  87         testCase.test();
  88     }
  89 
  90     private static void validateMethodHandle(ConstantPool constantPoolCTVM,
  91                                              ConstantTypes cpType,
  92                                              DummyClasses dummyClass,
  93                                              int index) {
  94         Object constantInPool = CompilerToVMHelper.resolveConstantInPool(constantPoolCTVM, index);
  95         String msg = String.format("%s for index %d", NOT_NULL_MSG, index);
  96         Asserts.assertNotNull(constantInPool, msg);
  97         if (!(constantInPool instanceof MethodHandle)) {
  98             msg = String.format("Wrong constant pool entry accessed by index"
  99                                         + " %d: %s, but should be subclass of %s",
 100                                 index,
 101                                 constantInPool.getClass(),
 102                                 MethodHandle.class.getName());
 103             throw new AssertionError(msg);
 104         }
 105     }
 106 
 107     private static void validateMethodType(ConstantPool constantPoolCTVM,
 108                                            ConstantTypes cpType,
 109                                            DummyClasses dummyClass,
 110                                            int index) {
 111         Object constantInPool = CompilerToVMHelper.resolveConstantInPool(constantPoolCTVM, index);
 112         String msg = String.format("%s for index %d", NOT_NULL_MSG, index);
 113         Asserts.assertNotNull(constantInPool, msg);
 114         Class mtToVerify = constantInPool.getClass();
 115         Class mtToRefer = MethodType.class;
 116         msg = String.format("Wrong method type class accessed by"
 117                                     + " constant pool index %d",
 118                             index);
 119         Asserts.assertEQ(mtToRefer, mtToVerify, msg);
 120     }
 121 }