1 /*
   2  * Copyright (c) 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 package gc.g1.humongousObjects;
  26 
  27 import gc.testlibrary.Helpers;
  28 import jdk.test.lib.Asserts;
  29 import sun.hotspot.WhiteBox;
  30 
  31 import java.io.IOException;
  32 import java.lang.reflect.InvocationTargetException;
  33 import java.net.URL;
  34 import java.net.URLClassLoader;
  35 import java.nio.file.Path;
  36 import java.nio.file.Paths;
  37 
  38 /**
  39  * @test gc.g1.humongousObjects.TestHumongousClassLoader
  40  * @summary Checks that unreachable classes and unreachable humongous class loader are unloaded after GC
  41  * @requires vm.gc=="G1" | vm.gc=="null"
  42  * @requires vm.opt.G1HeapRegionSize == "null" | vm.opt.G1HeapRegionSize == "1M"
  43  * @library /testlibrary /test/lib /
  44  * @modules java.management
  45  * @build sun.hotspot.WhiteBox
  46  *        gc.testlibrary.Helpers
  47  *        gc.g1.humongousObjects.G1SampleClass
  48  *        gc.g1.humongousObjects.ClassLoaderGenerator
  49  *        gc.g1.humongousObjects.TestHumongousClassLoader
  50  *
  51  * @run driver ClassFileInstaller sun.hotspot.WhiteBox
  52  *                                sun.hotspot.WhiteBox$WhiteBoxPermission
  53  *
  54  * @run main/othervm/timeout=240  -Xms256M -Xmx256M -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -Xbootclasspath/a:.
  55  *                                gc.g1.humongousObjects.ClassLoaderGenerator 1
  56  *
  57  * @run main/othervm -Xms256M -Xmx256M -XX:+UseG1GC -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -Xbootclasspath/a:.
  58  *                   -Xlog:classload,classunload=debug:file=TestHumongousClassLoader_Full_GC.log
  59  *                   -XX:G1HeapRegionSize=1M
  60  *                   gc.g1.humongousObjects.TestHumongousClassLoader FULL_GC
  61  *
  62  * @run main/othervm -Xms256M -Xmx256M -XX:+UseG1GC -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -Xbootclasspath/a:.
  63  *                   -Xlog:classload,classunload=debug:file=TestHumongousClassLoader_Full_GC_Mem_Pressure.log
  64  *                   -XX:G1HeapRegionSize=1M
  65  *                   gc.g1.humongousObjects.TestHumongousClassLoader FULL_GC_MEMORY_PRESSURE
  66  *
  67  *@run main/othervm -Xms256M -Xmx256M -XX:+UseG1GC -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -Xbootclasspath/a:.
  68  *                   -Xlog:classload,classunload=debug:file=TestHumongousClassLoader_CMC.log
  69  *                   -XX:G1HeapRegionSize=1M -XX:MaxTenuringThreshold=1
  70  *                   gc.g1.humongousObjects.TestHumongousClassLoader CMC
  71  *
  72  */
  73 
  74 public class TestHumongousClassLoader {
  75 
  76     private static final WhiteBox WB = WhiteBox.getWhiteBox();
  77     private static final String SAMPLE_CLASS_NAME_PREFIX = "SampleClassFiller";
  78     public static final String SIMPLE_CLASSLOADER_NAME = "SimpleClassLoader";
  79     public static final String HUMONGOUS_CLASSLOADER_NAME = "HumongousClassLoader";
  80 
  81 
  82     public static final String LOAD_CLASS_METHOD_PROTOTYPE =
  83             "    @Override\n"
  84                     + "    public Class loadClass(String fileName) throws ClassNotFoundException {\n"
  85                     + "        if (${ClassLoadFilter}) {\n"
  86                     + "            System.out.println(\"Loading class \" + fileName);\n"
  87                     + "            byte[] b = null;\n"
  88                     + "            try {\n"
  89                     + "                b = Files.readAllBytes(new File(fileName + \".class\").toPath());\n"
  90                     + "            } catch (IOException e) {\n"
  91                     + "                e.printStackTrace();\n"
  92                     + "            }\n"
  93                     + "            Class c = defineClass(fileName, b, 0, b.length);\n"
  94                     + "            resolveClass(c);\n"
  95                     + "            return c;\n"
  96                     + "        } else {\n"
  97                     + "            return super.loadClass(fileName);\n"
  98                     + "        }\n"
  99                     + "\n"
 100                     + "\n"
 101                     + "    }\n";
 102 
 103     public static final String CLASS_HEADER = "import java.io.File;\n"
 104             + "import java.io.IOException;\n"
 105             + "import java.nio.file.Files;\n"
 106             + "import java.nio.file.Paths;\n";
 107 
 108     public static final String GENERIC_PROTOTYPE = "${ClassHeader}\n"
 109             + "public class ${ClassName} extends ${BaseClass}{\n"
 110             + "    ${ConstructorClause}\n"
 111             + "    ${Methods}\n"
 112             + "    ${Fields}\n"
 113             + "}\n";
 114 
 115     public static final String CONSTUCTOR_PROTOTYPE = "public ${ClassName}(ClassLoader parent) { super(parent);}\n";
 116 
 117     private enum GC {
 118         FULL_GC {
 119             @Override
 120             public void provoke() {
 121                 System.gc();
 122             }
 123         },
 124         CMC {
 125             @Override
 126             public void provoke() {
 127                 // We need 2 young gc to promote class loader to old gen
 128                 // Otherwise it will not be unloaded after CMC
 129                 WHITE_BOX.youngGC();
 130                 Helpers.waitTillCMCFinished(WHITE_BOX, 0);
 131                 WHITE_BOX.youngGC();
 132                 Helpers.waitTillCMCFinished(WHITE_BOX, 0);
 133                 WHITE_BOX.g1StartConcMarkCycle();
 134                 Helpers.waitTillCMCFinished(WHITE_BOX, 0);
 135             }
 136         },
 137         FULL_GC_MEMORY_PRESSURE {
 138             @Override
 139             public void provoke() {
 140                 WHITE_BOX.fullGC();
 141             }
 142         };
 143         private static final WhiteBox WHITE_BOX = WhiteBox.getWhiteBox();
 144 
 145         public abstract void provoke();
 146     }
 147 
 148     public static void main(String[] args) throws ClassNotFoundException, InstantiationException,
 149             IllegalAccessException, IOException, NoSuchMethodException, InvocationTargetException {
 150 
 151         if (args.length != 1) {
 152             throw new Error("Test Bug: Expected GC type wasn't provided as command line argument");
 153         }
 154         GC gc = GC.valueOf(args[0]);
 155 
 156         Path wrkDir = Paths.get("");
 157         URL[] url = {wrkDir.toUri().toURL()};
 158         URLClassLoader urlLoader = new URLClassLoader(url);
 159 
 160         Class<?> simpleClassLoaderClass = urlLoader.loadClass(SIMPLE_CLASSLOADER_NAME);
 161 
 162         ClassLoader simpleClassLoader = (ClassLoader) simpleClassLoaderClass
 163                 .getConstructor(java.lang.ClassLoader.class)
 164                 .newInstance(TestHumongousClassLoader.class.getClassLoader());
 165 
 166         // Sanity check
 167         Asserts.assertEquals(WB.g1IsHumongous(simpleClassLoader), false,
 168                 "Test Bug: simpleClassLoader is expected to be non-humongous but it's humongous");
 169 
 170 
 171         Class<?> humongousClassLoaderClass = simpleClassLoader.loadClass(HUMONGOUS_CLASSLOADER_NAME);
 172 
 173         ClassLoader humongousClassLoader = (ClassLoader) humongousClassLoaderClass
 174                 .getConstructor(java.lang.ClassLoader.class)
 175                 .newInstance(simpleClassLoader);
 176 
 177         // Sanity check
 178         Asserts.assertEquals(WB.g1IsHumongous(humongousClassLoader), true,
 179                 "Test Bug: humongousClassLoader is expected to be humongous but it's non-humongous");
 180 
 181         //Asserts.assertEquals(1,0);
 182 
 183         Object[] loadedClasses = new Object[]{
 184                 G1SampleClass.LARGEST_NON_HUMONGOUS.getCls(humongousClassLoader, wrkDir, SAMPLE_CLASS_NAME_PREFIX)
 185                         .newInstance(),
 186                 G1SampleClass.SMALLEST_HUMONGOUS.getCls(humongousClassLoader, wrkDir, SAMPLE_CLASS_NAME_PREFIX)
 187                         .newInstance(),
 188                 G1SampleClass.ONE_REGION_HUMONGOUS.getCls(humongousClassLoader, wrkDir, SAMPLE_CLASS_NAME_PREFIX)
 189                         .newInstance(),
 190                 G1SampleClass.TWO_REGION_HUMONGOUS.getCls(humongousClassLoader, wrkDir, SAMPLE_CLASS_NAME_PREFIX)
 191                         .newInstance(),
 192         };
 193 
 194         // forgetting references to loaded classes
 195         for (int i = 0; i < loadedClasses.length; ++i) {
 196             loadedClasses[i] = null;
 197         }
 198 
 199         // forgetting referencies to classloaders
 200         humongousClassLoader = null;
 201         humongousClassLoaderClass = null;
 202 
 203         simpleClassLoader = null;
 204         simpleClassLoaderClass = null;
 205 
 206         gc.provoke();
 207 
 208         // Test checks
 209         Asserts.assertEquals(WB.isClassAlive(HUMONGOUS_CLASSLOADER_NAME), false,
 210                 String.format("Classloader class %s is loaded after we forget all references to it",
 211                         HUMONGOUS_CLASSLOADER_NAME));
 212 
 213         for (G1SampleClass sampleClass : G1SampleClass.values()) {
 214             String className = Helpers.enumNameToClassName(sampleClass.name()) + "Class";
 215             Asserts.assertEquals(WB.isClassAlive(className), false,
 216                     String.format("Class %s is loaded after we forget all references to it", className));
 217         }
 218     }
 219 
 220 }