1 /*
   2  * Copyright (c) 2016, 2019, 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 import sun.hotspot.WhiteBox;
  26 
  27 class LoadMe {
  28     static String getValue() {
  29         return "beforeHook";
  30     }
  31     static String getOtherValue() {
  32         return "abc-beforeHook-xyz";
  33     }
  34 }
  35 
  36 public class ClassFileLoadHook {
  37     public enum TestCaseId {
  38         SHARING_OFF_CFLH_ON,   // test case to establish a baseline
  39         SHARING_ON_CFLH_OFF,
  40         SHARING_AUTO_CFLH_ON,
  41         SHARING_ON_CFLH_ON
  42     }
  43 
  44     public static void main(String args[]) throws Exception {
  45         TestCaseId testCase = TestCaseId.valueOf(args[0]);
  46         test1(testCase);
  47         test2(testCase);
  48     }
  49 
  50     // Test rewriting the classfile data using CFLH
  51     static void test1(TestCaseId testCase) {
  52         WhiteBox wb = WhiteBox.getWhiteBox();
  53 
  54         System.out.println("====== ClassFileLoadHook.main():testCase = " + testCase);
  55         System.out.println("getValue():" + LoadMe.getValue());
  56         System.out.println("getOtherValue():" + LoadMe.getOtherValue());
  57 
  58         switch (testCase) {
  59             case SHARING_OFF_CFLH_ON:
  60                 assertTrue("after_Hook".equals(LoadMe.getValue()) &&
  61                            "abc-after_Hook-xyz".equals(LoadMe.getOtherValue()),
  62                            "Not sharing, this test should replace beforeHook " +
  63                            "with after_Hook");
  64             break;
  65 
  66             case SHARING_ON_CFLH_OFF:
  67                 assertTrue(wb.isSharedClass(LoadMe.class),
  68                     "LoadMe should be shared, but is not");
  69                 assertTrue("beforeHook".equals(LoadMe.getValue()) &&
  70                            "abc-beforeHook-xyz".equals(LoadMe.getOtherValue()),
  71                            "CFLH off, bug values are redefined");
  72             break;
  73 
  74             case SHARING_AUTO_CFLH_ON:
  75             case SHARING_ON_CFLH_ON:
  76                 // LoadMe is rewritten on CFLH
  77                 assertFalse(wb.isSharedClass(LoadMe.class),
  78                     "LoadMe should not be shared if CFLH has modified the class");
  79                 assertFalse("beforeHook".equals(LoadMe.getValue()) &&
  80                            "abc-beforeHook-xyz".equals(LoadMe.getOtherValue()),
  81                            "Class contents should be changed if CFLH is enabled");
  82              break;
  83 
  84              default:
  85                  throw new RuntimeException("Invalid testcase");
  86 
  87         }
  88     }
  89 
  90     // Test the loading of classfile data for non-boot shared classes from jrt:/xxx.
  91     // See JDK-8221351.
  92     static void test2(TestCaseId testCase) throws Exception {
  93         WhiteBox wb = WhiteBox.getWhiteBox();
  94         Class c = Class.forName("java.sql.SQLException"); // defined by platform class loader.
  95 
  96         switch (testCase) {
  97             case SHARING_ON_CFLH_OFF:
  98             case SHARING_ON_CFLH_ON:
  99                 assertTrue(wb.isSharedClass(c), "must be shared");
 100                 break;
 101             case SHARING_AUTO_CFLH_ON:
 102                 // With -Xshare:auto, the test continues to run with mapping failure.
 103                 // In case of mapping failure, java/lang/Object and the app class
 104                 // won't be loaded from the archive.
 105                 Class objClass = Class.forName("java.lang.Object");
 106                 if (wb.isSharedClass(objClass)) {
 107                     assertTrue(wb.isSharedClass(c), "must be shared");
 108                 } else {
 109                     assertFalse(wb.isSharedClass(c), "must not be shared");
 110                 }
 111                 break;
 112             default:
 113                 // this test is not applicable to -Xshare:off
 114                 if (testCase != TestCaseId.SHARING_OFF_CFLH_ON) {
 115                     throw new RuntimeException("Invalid testcase");
 116                 }
 117         }
 118     }
 119 
 120     private static void assertTrue(boolean expr, String msg) {
 121         if (!expr)
 122             throw new RuntimeException(msg);
 123     }
 124 
 125     private static void assertFalse(boolean expr, String msg) {
 126         if (expr)
 127             throw new RuntimeException(msg);
 128     }
 129 }