/*
 * Copyright (c) 2015, 2019, Oracle and/or its affiliates. All rights reserved.
 * 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.
 *
 */

import java.io.*;
import java.net.*;
import sun.hotspot.WhiteBox;

public class LoaderSegregation {
    // Use these definitions instead of literal strings, to avoid typos.
    static final String ONLY_BUILTIN      = "OnlyBuiltin";
    static final String ONLY_UNREGISTERED = "OnlyUnregistered";

    public static void main(String args[]) throws Exception {
        WhiteBox wb = WhiteBox.getWhiteBox();

        // [A] An archived BUILTIN class cannot be a subclass of a non-BUILTIN class.
        // [B] An archived BUILTIN class cannot implement a non-BUILTIN interface.
        if (wb.isSharedClass(LoaderSegregation.class)) {
            // [1] check that CustomLoadee is loadable from archive
            if (!wb.isSharedClass(CustomLoadee.class)) {
                throw new RuntimeException("wb.isSharedClass(CustomLoadee.class) should be true");
            }

            // [2] CustomInterface2_ia should be archived, even though it was not specified in the classlist.
            //     It was successfully dumped as a side effect of attempting to load CustomLoadee2
            //     during dump time. Note that CustomLoadee2 failed to dump because one of its interfaces,
            //     CustomInterface2_ib, was not loadable from the BOOT/EXT/APP classpath. during dump time.
            if (!wb.isSharedClass(CustomInterface2_ia.class)) {
                throw new RuntimeException("wb.isSharedClass(CustomInterface2_ia.class) should be true");
            }

            // [3] Check that the BUILTIN versions of CustomLoadee2 and CustomLoadee3Child are loadable
            //     at run time (since we have append LoaderSegregation_app2.jar the classpath),
            //     but these classes must be loaded from the JAR file.
            if (wb.isSharedClass(CustomLoadee2.class)) {
                throw new RuntimeException("wb.isSharedClass(CustomLoadee2.class) should be false");
            }
            if (wb.isSharedClass(CustomLoadee3.class)) {
                throw new RuntimeException("wb.isSharedClass(CustomLoadee3.class) should be false");
            }
            if (wb.isSharedClass(CustomLoadee3Child.class)) {
                throw new RuntimeException("wb.isSharedClass(CustomLoadee3Child.class) should be false");
            }
        }

        // [C] BUILTIN and UNREGISTERED classes can be loaded only by their corresponding
        //     type of loaders.

        String path = args[0];
        File jarFile = new File(path);
        URL url = new File(path).toURI().toURL();
        URL[] urls = new URL[] {url};
        ClassLoader appLoader = LoaderSegregation.class.getClassLoader();

        { // BUILTIN LOADER
            try {
                appLoader.loadClass(ONLY_UNREGISTERED);
                throw new RuntimeException("BUILTIN loader cannot load archived UNREGISTERED class");
            } catch (ClassNotFoundException expected) {}
        }

        { // UNREGISTERED LOADER
            URLClassLoader urlClassLoader = new URLClassLoader(urls) {
                protected Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException {
                    synchronized (getClassLoadingLock(name)) {
                        Class<?> c = findLoadedClass(name);
                        if (c == null) {
                            try {
                                c = findClass(name);
                            } catch (ClassNotFoundException e) {
                                c = getParent().loadClass(name);
                            }
                        }
                        if (resolve) {
                            resolveClass(c);
                        }
                        return c;
                    }
                }
            };
            Class<?> c2 = urlClassLoader.loadClass(ONLY_BUILTIN);

            if (c2.getClassLoader() != urlClassLoader) {
                throw new RuntimeException("Error in test");
            }

            if (wb.isSharedClass(LoaderSegregation.class)) {
                if (wb.isSharedClass(c2)) {
                    throw new RuntimeException("wb.isSharedClass(c2) should be false - " +
                                               "unregistered loader cannot load an archived BUILTIN class");
                }
            }
        }
    }
}