1 /*
   2  * Copyright (c) 2016, 2017, 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 // This is a utlitity test class for loading classes-under-test
  26 // by means of custom class loader.
  27 // See AppCDS/jvmti/transformRelatedClasses/TransformRelatedClasses.java
  28 // for an example.
  29 // Use this test app in conjunction with other tests
  30 // to load and exercise classes using custom class loader(s).
  31 // This class is intended to be called by the "main test driver"
  32 // inside a child process, normally with sharing enabled.
  33 //
  34 // Arguments: customJarPath, loaderType, testClass
  35 //     customJarPath - a path to jar file containing classes for
  36 //         loading via this custom class loader, including the
  37 //         testClass
  38 //     loaderType - Currently only "unregistered"
  39 //         (Fingerprint verification method) is allowed
  40 //     testClass - the class to be loader; the test method with
  41 //         signature 'public static void test()' will be called
  42 //         on this class, so class must contain such method
  43 
  44 
  45 import java.io.File;
  46 import java.lang.reflect.Method;
  47 import java.net.URL;
  48 import java.net.URLClassLoader;
  49 import java.util.logging.Logger;
  50 
  51 public class CustomLoaderApp {
  52     public static void ping() {};
  53 
  54     private static void log(String msg) {
  55         System.out.println("CustomLoaderApp: " + msg);
  56     }
  57 
  58     public static void main(String[] args) throws Exception {
  59         String path = args[0];
  60         URL url = new File(path).toURI().toURL();
  61         URL[] urls = new URL[] {url};
  62 
  63         String loaderType = args[1];
  64         log("loaderType = " + loaderType);
  65 
  66         String testClass = args[2];
  67         log("testClass = " + testClass);
  68 
  69         switch(loaderType) {
  70         case "unregistered":
  71             loadAndUseWithUnregisteredLoader(urls, testClass);
  72             break;
  73         default:
  74             throw new IllegalArgumentException("loader type is wrong: " + loaderType);
  75         }
  76     }
  77 
  78 
  79     // Load the test classes using unregistered loader
  80     // (i.e. loader that is not using AppCDS API)
  81     private static void loadAndUseWithUnregisteredLoader(URL[] urls, String testClass)
  82         throws Exception {
  83         URLClassLoader urlClassLoader = new URLClassLoader(urls);
  84         callTestMethod(loadAndCheck(urlClassLoader, testClass));
  85     }
  86 
  87     private static Class loadAndCheck(ClassLoader loader, String className)
  88         throws ClassNotFoundException {
  89         Class c = loader.loadClass(className);
  90         log("class =" + c);
  91         log("loader = " + c.getClassLoader());
  92 
  93         // Check that c is defined by the correct loader
  94         if (c.getClassLoader() != loader) {
  95             String msg = String.format("c.getClassLoader() equals to <%s>, expected <%s>",
  96                                        c.getClassLoader(), loader);
  97             throw new RuntimeException(msg);
  98         }
  99         return c;
 100     }
 101 
 102     private static void callTestMethod(Class c) throws Exception {
 103         Method[] methods = c.getDeclaredMethods();
 104         for (Method m : methods) {
 105             log("method = " + m.getName());
 106             if (m.getName().equals("test"))
 107                 m.invoke(null);
 108         }
 109     }
 110 }