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  * @test
  26  * @bug 8072008
  27  * @library /testlibrary /test/lib
  28  * @compile RedefineTest.java Agent.java
  29  * @run main ClassFileInstaller sun.hotspot.WhiteBox
  30  *                              sun.hotspot.WhiteBox$WhiteBoxPermission
  31  *                              java.lang.invoke.RedefineTest
  32  *                              Agent
  33  *                              jdk.test.lib.Asserts
  34  * @run main Agent agent.jar java.lang.invoke.RedefineTest
  35  * @run main/othervm -Xbootclasspath/a:. -javaagent:agent.jar
  36  *                   -XX:+IgnoreUnrecognizedVMOptions
  37  *                   -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI
  38  *                   -Xbatch -XX:-TieredCompilation -XX:CICompilerCount=1
  39  *                      java.lang.invoke.RedefineTest
  40  */
  41 package java.lang.invoke;
  42 
  43 import sun.hotspot.WhiteBox;
  44 import sun.misc.Unsafe;
  45 import jdk.internal.org.objectweb.asm.*;
  46 import jdk.internal.vm.annotation.DontInline;
  47 import java.lang.instrument.ClassDefinition;
  48 import java.lang.instrument.Instrumentation;
  49 
  50 import static jdk.internal.org.objectweb.asm.Opcodes.*;
  51 
  52 public class RedefineTest {
  53     static final MethodHandles.Lookup LOOKUP = MethodHandles.Lookup.IMPL_LOOKUP;
  54     static final Unsafe UNSAFE = Unsafe.getUnsafe();
  55 
  56     static final String NAME = "java/lang/invoke/RedefineTest$T";
  57 
  58     static Class<?> getClass(int r) {
  59         byte[] classFile = getClassFile(r);
  60         return UNSAFE.defineClass(NAME, classFile, 0, classFile.length, null, null);
  61     }
  62 
  63     /**
  64      * Generates a class of the following shape:
  65      *     static class T {
  66      *         @DontInline public static int f() { return $r; }
  67      *     }
  68      */
  69     static byte[] getClassFile(int r) {
  70         ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS);
  71         MethodVisitor mv;
  72         cw.visit(52, ACC_PUBLIC | ACC_SUPER, NAME, null, "java/lang/Object", null);
  73         {
  74             mv = cw.visitMethod(ACC_PUBLIC | ACC_STATIC, "f", "()I", null, null);
  75             mv.visitAnnotation("Ljdk/internal/vm/annotation/DontInline;", true);
  76             mv.visitCode();
  77             mv.visitLdcInsn(r);
  78             mv.visitInsn(IRETURN);
  79             mv.visitMaxs(0, 0);
  80             mv.visitEnd();
  81         }
  82         cw.visitEnd();
  83         return cw.toByteArray();
  84     }
  85 
  86     static final MethodHandle mh;
  87     static final Class<?> CLS = getClass(0);
  88     static {
  89         try {
  90             mh = LOOKUP.findStatic(CLS, "f", MethodType.methodType(int.class));
  91         } catch (Exception e) {
  92             throw new Error(e);
  93         }
  94     }
  95 
  96     static final WhiteBox WB = WhiteBox.getWhiteBox();
  97 
  98     @DontInline
  99     static int invokeBasic() {
 100         try {
 101             return (int)mh.invokeExact();
 102         } catch (Throwable e) {
 103             throw new Error(e);
 104         }
 105     }
 106 
 107     static Instrumentation instr;
 108     public static void premain(String args, Instrumentation instr) {
 109         RedefineTest.instr = instr;
 110     }
 111 
 112 
 113     public static void main(String[] args) throws Exception {
 114         for (int i = 0; i < 20_000; i++) {
 115             int r = invokeBasic();
 116             if (r != 0) {
 117                 throw new Error(r + " != 0");
 118             }
 119         }
 120         // WB.ensureCompiled();
 121 
 122         redefine();
 123 
 124         int exp = (instr != null) ? 1 : 0;
 125 
 126         for (int i = 0; i < 20_000; i++) {
 127             if (invokeBasic() != exp) {
 128                 throw new Error();
 129             }
 130         }
 131 
 132         WB.clearInlineCaches();
 133 
 134         for (int i = 0; i < 20_000; i++) {
 135             if (invokeBasic() != exp) {
 136                 throw new Error();
 137             }
 138         }
 139 
 140         // WB.ensureCompiled();
 141     }
 142 
 143     static void redefine() {
 144         if (instr == null) {
 145             System.out.println("NOT REDEFINED");
 146             return;
 147         }
 148         ClassDefinition cd = new ClassDefinition(CLS, getClassFile(1));
 149         try {
 150             instr.redefineClasses(cd);
 151         } catch (Exception e) {
 152             throw new Error(e);
 153         }
 154         System.out.println("REDEFINED");
 155     }
 156 }