/*

 * Copyright (c) 2016, 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.

 */

/*

 * @test

 * @bug 8159393

 * @summary Test signed jars involved in image creation

 * @modules java.base/jdk.internal.jimage

 *          java.base/sun.security.tools.keytool

 *          jdk.compiler/com.sun.tools.javac

 *          jdk.jartool/sun.security.tools.jarsigner

 *          jdk.jartool/sun.tools.jar

 *          jdk.jlink/jdk.tools.jlink.internal

 * @run main/othervm JLinkSigningTest

 */

import java.io.File;

import java.io.IOException;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.Arrays;

import java.util.spi.ToolProvider;

public class JLinkSigningTest {

    private static final ToolProvider JAR_TOOL = ToolProvider.findFirst("jar")

            .orElseThrow(() -> new RuntimeException("jar tool not found"));

    private static final ToolProvider JAVAC_TOOL = ToolProvider.findFirst("javac")

            .orElseThrow(() -> new RuntimeException("javac tool not found"));

    private static final ToolProvider JLINK_TOOL = ToolProvider.findFirst("jlink")

            .orElseThrow(() -> new RuntimeException("jlink tool not found"));

    static final String[] MODULE_INFO = {

        "module test {",

        "}",

    };

    static final String[] TEST_CLASS = {

        "package test;",

        "public class test {",

        "    public static void main(String[] args) {",

        "    }",

        "}",

    };

    static void report(String command, String[] args) {

        System.out.println(command + " " + String.join(" ", Arrays.asList(args)));

    }

    static void jar(String[] args) {

        report("jar", args);

        JAR_TOOL.run(System.out, System.err, args);

    }

    static void jarsigner(String[] args) {

        report("jarsigner", args);

        try {

            sun.security.tools.jarsigner.Main.main(args);

        } catch (Exception ex) {

            throw new RuntimeException("jarsigner not found");

        }

    }

    static void javac(String[] args) {

        report("javac", args);

        JAVAC_TOOL.run(System.out, System.err, args);

    }

    static void jlink(String[] args) {

        report("jlink", args);

        JLINK_TOOL.run(System.out, System.err, args);

    }

    static void keytool(String[] args) {

        report("keytool", args);

        try {

            sun.security.tools.keytool.Main.main(args);

        } catch (Exception ex) {

            throw new RuntimeException("keytool failed");

        }

    }

    public static void main(String[] args) {

        final String JAVA_HOME = System.getProperty("java.home");

        Path moduleInfoJavaPath = Paths.get("module-info.java");

        Path moduleInfoClassPath = Paths.get("module-info.class");

        Path testDirectoryPath = Paths.get("test");

        Path testJavaPath = testDirectoryPath.resolve("test.java");

        Path testClassPath = testDirectoryPath.resolve("test.class");

        Path testModsDirectoryPath = Paths.get("testmods");

        Path jmodsPath = Paths.get(JAVA_HOME, "jmods");

        Path testjarPath = testModsDirectoryPath.resolve("test.jar");

        String modulesPath = testjarPath.toString() +

                             File.pathSeparator +

                             jmodsPath.toString();

        try {

            Files.write(moduleInfoJavaPath, Arrays.asList(MODULE_INFO));

            Files.createDirectories(testDirectoryPath);

            Files.write(testJavaPath, Arrays.asList(TEST_CLASS));

            Files.createDirectories(testModsDirectoryPath);

        } catch (IOException ex) {

            throw new RuntimeException("file construction failed");

        }

        javac(new String[] {

            testJavaPath.toString(),

            moduleInfoJavaPath.toString(),

        });

        jar(new String[] {

            "-c",

            "-f", testjarPath.toString(),

            "--module-path", jmodsPath.toString(),

            testClassPath.toString(),

            moduleInfoClassPath.toString(),

        });

        keytool(new String[] {

            "-genkey",

            "-keyalg", "RSA",

            "-dname", "CN=John Doe, OU=JPG, O=Oracle, L=Santa Clara, ST=California, C=US",

            "-alias", "examplekey",

            "-storepass", "password",

            "-keypass", "password",

            "-keystore", "examplekeystore",

            "-validity", "365",

        });

        jarsigner(new String[] {

            "-keystore", "examplekeystore",

            "-verbose", testjarPath.toString(),

            "-storepass", "password",

            "-keypass", "password",

            "examplekey",

        });

        try {

            jlink(new String[] {

                "--module-path", modulesPath,

                "--add-modules", "test",

                "--output", "foo",

            });

        } catch (Throwable ex) {

            System.out.println("Failed as should");

        }

        try {

            jlink(new String[] {

                "--module-path", modulesPath,

                "--add-modules", "test",

                "--ignore-signing-information",

                "--output", "foo",

            });

            System.out.println("Suceeded as should");

        } catch (Throwable ex) {

            System.err.println("Should not have failed");

            throw new RuntimeException(ex);

        }

        System.out.println("Done");

    }

}