/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 *

 */

// This is a utlitity test class for loading classes-under-test

// by means of custom class loader.

// See AppCDS/jvmti/transformRelatedClasses/TransformRelatedClasses.java

// for an example.

// Use this test app in conjunction with other tests

// to load and exercise classes using custom class loader(s).

// This class is intended to be called by the "main test driver"

// inside a child process, normally with sharing enabled.

//

// Arguments: customJarPath, loaderType, testClass

//     customJarPath - a path to jar file containing classes for

//         loading via this custom class loader, including the

//         testClass

//     loaderType - Currently only "unregistered"

//         (Fingerprint verification method) is allowed

//     testClass - the class to be loader; the test method with

//         signature 'public static void test()' will be called

//         on this class, so class must contain such method

import java.io.File;

import java.lang.reflect.Method;

import java.net.URL;

import java.net.URLClassLoader;

import java.util.logging.Logger;

public class CustomLoaderApp {

    public static void ping() {};

    private static void log(String msg) {

        System.out.println("CustomLoaderApp: " + msg);

    }

    public static void main(String[] args) throws Exception {

        String path = args[0];

        URL url = new File(path).toURI().toURL();

        URL[] urls = new URL[] {url};

        String loaderType = args[1];

        log("loaderType = " + loaderType);

        String testClass = args[2];

        log("testClass = " + testClass);

        switch(loaderType) {

        case "unregistered":

            loadAndUseWithUnregisteredLoader(urls, testClass);

            break;

        default:

            throw new IllegalArgumentException("loader type is wrong: " + loaderType);

        }

    }

    // Load the test classes using unregistered loader

    // (i.e. loader that is not using AppCDS API)

    private static void loadAndUseWithUnregisteredLoader(URL[] urls, String testClass)

        throws Exception {

        URLClassLoader urlClassLoader = new URLClassLoader(urls);

        callTestMethod(loadAndCheck(urlClassLoader, testClass));

    }

    private static Class loadAndCheck(ClassLoader loader, String className)

        throws ClassNotFoundException {

        Class c = loader.loadClass(className);

        log("class =" + c);

        log("loader = " + c.getClassLoader());

        // Check that c is defined by the correct loader

        if (c.getClassLoader() != loader) {

            String msg = String.format("c.getClassLoader() equals to <%s>, expected <%s>",

                                       c.getClassLoader(), loader);

            throw new RuntimeException(msg);

        }

        return c;

    }

    private static void callTestMethod(Class c) throws Exception {

        Method[] methods = c.getDeclaredMethods();

        for (Method m : methods) {

            log("method = " + m.getName());

            if (m.getName().equals("test"))

                m.invoke(null);

        }

    }

}