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