/*

 * 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.

 */

/*

 * @test

 * @summary Test the System properties for JarFile that support multi-release jar files

 * @library /lib/testlibrary/java/util/jar

 * @build Compiler JarBuilder CreateMultiReleaseTestJars

 * @run testng MultiReleaseJarProperties

 * @run testng/othervm -Djdk.util.jar.version=100 MultiReleaseJarProperties

 * @run testng/othervm -Djdk.util.jar.enableMultiRelease=false MultiReleaseJarProperties

 * @run testng/othervm -Djdk.util.jar.enableMultiRelease=force MultiReleaseJarProperties

 */

import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import java.lang.invoke.MethodHandle;

import java.lang.invoke.MethodHandles;

import java.lang.invoke.MethodType;

import java.net.URL;

import java.net.URLClassLoader;

import java.nio.file.Files;

import java.util.jar.JarEntry;

import java.util.jar.JarFile;

import org.testng.Assert;

import org.testng.annotations.AfterClass;

import org.testng.annotations.BeforeClass;

import org.testng.annotations.Test;

public class MultiReleaseJarProperties {

    final static int BASE_VERSION = JarFile.baseVersion().major();

    final static String userdir = System.getProperty("user.dir", ".");

    final static File multirelease = new File(userdir, "multi-release.jar");

    protected int rtVersion;

    boolean force;

    protected ClassLoader cldr;

    protected Class<?> rootClass;

    @BeforeClass

    public void initialize() throws Exception {

        CreateMultiReleaseTestJars creator =  new CreateMultiReleaseTestJars();

        creator.compileEntries();

        creator.buildMultiReleaseJar();

        int RUNTIME_VERSION = Runtime.version().major();

        rtVersion = Integer.getInteger("jdk.util.jar.version", RUNTIME_VERSION);

        String mrprop = System.getProperty("jdk.util.jar.enableMultiRelease", "");

        if (mrprop.equals("false")) {

            rtVersion = BASE_VERSION;

        } else if (rtVersion < BASE_VERSION) {

            rtVersion = BASE_VERSION;

        } else if (rtVersion > RUNTIME_VERSION) {

            rtVersion = RUNTIME_VERSION;

        }

        force = mrprop.equals("force");

        initializeClassLoader();

    }

    protected void initializeClassLoader() throws Exception {

        URL[] urls = new URL[]{multirelease.toURI().toURL()};

        cldr = new URLClassLoader(urls);

        // load any class, Main is convenient and in the root entries

        rootClass = cldr.loadClass("version.Main");

    }

    @AfterClass

    public void close() throws IOException {

        ((URLClassLoader)cldr).close();

        Files.delete(multirelease.toPath());

    }

    /*

     * jdk.util.jar.enableMultiRelease=force is a no-op for URLClassLoader

     */

    @Test

    public void testURLClassLoader() throws Throwable {

        Class<?> vcls = cldr.loadClass("version.Version");

        invokeMethod(vcls, rtVersion);

    }

    protected void invokeMethod(Class<?> vcls, int expected) throws Throwable {

        MethodType mt = MethodType.methodType(int.class);

        MethodHandle mh = MethodHandles.lookup().findVirtual(vcls, "getVersion", mt);

        Assert.assertEquals(expected, (int) mh.invoke(vcls.newInstance()));

    }

    /*

     * jdk.util.jar.enableMultiRelease=force should affect a custom class loader

     */

    @Test

    public void testClassLoader() throws Throwable {

        try (JarFile jf = new JarFile(multirelease)) {  // do not set runtime versioning

            ClassLoader cldr = new CustomClassLoader(jf);

            Class<?> vcls = cldr.loadClass("version.Version");

            if (rtVersion == 9) {

                try {

                    cldr.loadClass("version.PackagePrivate");

                } catch (ClassNotFoundException x) {

                    if (force) throw x;

                }

            }

            invokeMethod(vcls, force ? rtVersion : BASE_VERSION);

        }

    }

    private static class CustomClassLoader extends ClassLoader {

        private final JarFile jf;

        CustomClassLoader(JarFile jf) throws Exception {

            super(null);

            this.jf = jf;

        }

        protected Class<?> findClass(String name) throws ClassNotFoundException {

            try {

                byte[] b;

                String entryName = name.replace(".", "/") + ".class";

                JarEntry je = jf.getJarEntry(entryName);

                if (je != null) {

                    try (InputStream is = jf.getInputStream(je)) {

                        b = new byte[(int) je.getSize()];

                        is.read(b);

                    }

                    return defineClass(name, b, 0, b.length);

                }

                throw new ClassNotFoundException(name);

            } catch (IOException x) {

                throw new ClassNotFoundException(x.getMessage());

            }

        }

    }

    @Test

    public void testGetResourceAsStream() throws Exception {

        String resource = rtVersion == 9 ? "/version/PackagePrivate.java" : "/version/Version.java";

        // use fileRootClass as a base for getting resources

        getResourceAsStream(rootClass, resource);

    }

    protected void getResourceAsStream(Class<?> rootClass, String resource) throws Exception {

        try (InputStream is = rootClass.getResourceAsStream(resource)) {

            byte[] bytes = is.readAllBytes();

            resource = new String(bytes);

        }

        String match = "return " + rtVersion + ";";

        Assert.assertTrue(resource.contains(match));

    }

    @Test

    public void testGetResource() throws Exception {

        String resource = rtVersion == 9 ? "/version/PackagePrivate.java" : "/version/Version.java";

        // use rootClass as a base for getting resources

        getResource(rootClass, resource);

    }

    protected void getResource(Class<?> rootClass, String resource) throws Exception {

        URL url = rootClass.getResource(resource);

        try (InputStream is = url.openStream()) {

            byte[] bytes = is.readAllBytes();

            resource = new String(bytes);

        }

        String match = "return " + rtVersion + ";";

        Assert.assertTrue(resource.contains(match));

    }

}