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:+open
  31  * @modules java.base/jdk.internal.org.objectweb.asm
  32  *          java.base/jdk.internal.org.objectweb.asm.tree
  33  * @modules jdk.internal.vm.ci/jdk.vm.ci.hotspot:+open
  34  *          jdk.internal.vm.ci/jdk.vm.ci.code
  35  *          jdk.internal.vm.ci/jdk.vm.ci.meta
  36  * @build jdk.internal.vm.ci/jdk.vm.ci.hotspot.CompilerToVMHelper
  37  * @run main/othervm -XX:+UnlockExperimentalVMOptions -XX:+EnableJVMCI
  38  *                   -Djvmci.Compiler=null
  39  *                  compiler.jvmci.compilerToVM.GetSymbolTest
  40  */
  41 
  42 package compiler.jvmci.compilerToVM;
  43 
  44 import compiler.jvmci.common.CTVMUtilities;
  45 import compiler.jvmci.common.testcases.SingleImplementer;
  46 import jdk.test.lib.Utils;
  47 import jdk.vm.ci.hotspot.CompilerToVMHelper;
  48 import jdk.vm.ci.hotspot.HotSpotResolvedJavaMethod;
  49 import jdk.vm.ci.meta.ConstantPool;
  50 
  51 import java.lang.reflect.Field;
  52 import java.lang.reflect.Member;
  53 import java.lang.reflect.Method;
  54 import java.util.ArrayList;
  55 import java.util.List;
  56 import java.util.function.Function;
  57 import java.util.stream.Collectors;
  58 import java.util.stream.Stream;
  59 
  60 public class GetSymbolTest {
  61     private static final int CONSTANT_POOL_UTF8_TAG = 1; // see jvms, section 4.4
  62 
  63     private static final Function<Member[], List<String>> NAMES = members ->
  64             Stream.of(members)
  65                     .map(Member::getName)
  66                     .collect(Collectors.toList());
  67 
  68     public static void main(String[] args) {
  69         new GetSymbolTest().test(SingleImplementer.class);
  70     }
  71 
  72     private void test(Class<?> aClass) {
  73         Utils.ensureClassIsLoaded(aClass);
  74         Method method;
  75         try {
  76             method = aClass.getDeclaredMethod("nonInterfaceMethod");
  77         } catch (NoSuchMethodException e) {
  78             throw new Error("TEST BUG: can't find test method", e);
  79         }
  80         HotSpotResolvedJavaMethod resolvedMethod
  81                 = CTVMUtilities.getResolvedMethod(aClass, method);
  82         List<String> symbols;
  83         try {
  84             symbols = getSymbols(resolvedMethod);
  85         } catch (ReflectiveOperationException e) {
  86             throw new Error("TEST BUG: can't access private members", e);
  87         }
  88         List<String> classSymbols = new ArrayList<>();
  89         classSymbols.addAll(NAMES.apply(aClass.getDeclaredFields()));
  90         classSymbols.addAll(NAMES.apply(aClass.getDeclaredMethods()));
  91         // Check that all members of test class have symbols from constant pool
  92         for (String s : classSymbols) {
  93             if (!symbols.contains(s)) {
  94                 // failed. print all symbols found by getSymbol
  95                 System.out.println("getSymbol data:");
  96                 for (String ctvmValue : symbols) {
  97                     System.out.println(ctvmValue);
  98                 }
  99                 throw new AssertionError("Unable to find symbol " + s
 100                         + " using CompilerToVM.getSymbol");
 101             }
 102         }
 103     }
 104 
 105     private List<String> getSymbols(HotSpotResolvedJavaMethod
 106             metaspaceMethod) throws ReflectiveOperationException {
 107         List<String> symbols = new ArrayList<>();
 108         ConstantPool pool = metaspaceMethod.getConstantPool();
 109         long length = pool.length();
 110         // jvms-4.1: The constant_pool table is indexed from 1 ...
 111         for (int i = 1; i < length; i++) {
 112             if (getTag(pool, i) == CONSTANT_POOL_UTF8_TAG) {
 113                 long entryPointer;
 114                 Method getEntryAt = pool.getClass()
 115                         .getDeclaredMethod("getEntryAt", int.class);
 116                 getEntryAt.setAccessible(true);
 117                 entryPointer = (Long) getEntryAt.invoke(pool, i);
 118                 String symbol = CompilerToVMHelper.getSymbol(entryPointer);
 119                 symbols.add(symbol);
 120             }
 121         }
 122         return symbols;
 123     }
 124 
 125     private int getTag(ConstantPool pool, int index)
 126             throws ReflectiveOperationException {
 127         Object jvmConstant;
 128         Method getTag = pool.getClass().getDeclaredMethod("getTagAt",
 129                 int.class);
 130         getTag.setAccessible(true);
 131         jvmConstant = getTag.invoke(pool, index);
 132         Field tagCode = jvmConstant.getClass().getDeclaredField("tag");
 133         tagCode.setAccessible(true);
 134         return tagCode.getInt(jvmConstant);
 135     }
 136 }