/*
 * Copyright (c) 2012, 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 8003256
 * @compile -XDignore.symbol.file VersionCheck.java
 * @run main VersionCheck
 * @summary Tests that "java -version" includes the name of the profile and that
 *     it matches the name in the release file
 */

import java.nio.file.*;
import java.io.*;
import java.util.Properties;

public class VersionCheck {

    static final String JAVA_HOME = System.getProperty("java.home");
    static final String OS_NAME = System.getProperty("os.name");
    static final String OS_ARCH = System.getProperty("os.arch");

    static final String JAVA_CMD =
            OS_NAME.startsWith("Windows") ? "java.exe" : "java";

    static final boolean NEED_D64 =
            OS_NAME.equals("SunOS") &&
            (OS_ARCH.equals("sparcv9") || OS_ARCH.equals("amd64"));

    /**
     * Returns {@code true} if the given class is present.
     */
    static boolean isPresent(String cn) {
        try {
            Class.forName(cn);
            return true;
        } catch (ClassNotFoundException ignore) {
            return false;
        }
    }

    /**
     * Determines the profile by checking whether specific classes are present.
     * Returns the empty string if this runtime does not appear to be a profile
     * of Java SE.
     */
    static String probeProfile() {
        if (isPresent("java.awt.Window"))
            return "";
        if (isPresent("java.lang.management.ManagementFactory"))
            return "compact3";
        if (isPresent("java.sql.DriverManager"))
            return "compact2";
        return "compact1";
    }

    /**
     * Execs java with the given parameters. The method blocks until the
     * process terminates. Returns a {@code ByteArrayOutputStream} with any
     * stdout or stderr from the process.
     */
    static ByteArrayOutputStream execJava(String... args)
        throws IOException
    {
        StringBuilder sb = new StringBuilder();
        sb.append(Paths.get(JAVA_HOME, "bin", JAVA_CMD).toString());
        if (NEED_D64)
            sb.append(" -d64");
        for (String arg: args) {
            sb.append(' ');
            sb.append(arg);
        }
        String[] cmd = sb.toString().split(" ");
        ProcessBuilder pb = new ProcessBuilder(cmd);
        pb.redirectErrorStream(true);
        Process p = pb.start();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] buf = new byte[1024];
        int n;
        do {
            n = p.getInputStream().read(buf);
            if (n > 0)
               baos.write(buf, 0, n);
        } while (n > 0);
        try {
            int exitCode = p.waitFor();
            if (exitCode != 0)
                throw new RuntimeException("Exit code: " + exitCode);
        } catch (InterruptedException e) {
            throw new RuntimeException("Should not happen");
        }
        return baos;
    }

    public static void main(String[] args) throws IOException {
        String reported = sun.misc.Version.profileName();
        String probed = probeProfile();
        if (!reported.equals(probed)) {
            throw new RuntimeException("sun.misc.Version reports: " + reported
               + ", but probing reports: " + probed);
        }

        String profile = probed;
        boolean isFullJre = (profile.length() == 0);

        // check that java -version includes "profile compactN"
        String expected = "profile " + profile;
        System.out.println("Checking java -version ...");
        ByteArrayOutputStream baos = execJava("-version");
        ByteArrayInputStream bain = new ByteArrayInputStream(baos.toByteArray());
        BufferedReader reader = new BufferedReader(new InputStreamReader(bain));
        boolean found = false;
        String line;
        while ((line = reader.readLine()) != null) {
            if (line.contains(expected)) {
                found = true;
                break;
            }
        }
        if (found && isFullJre)
           throw new RuntimeException(expected + " found in java -version output");
        if (!found && !isFullJre)
            throw new RuntimeException("java -version did not include " + expected);

        // check that the profile name matches the release file
        System.out.println("Checking release file ...");
        Properties props = new Properties();

        Path home = Paths.get(JAVA_HOME);
        if (home.getFileName().toString().equals("jre"))
            home = home.getParent();
        Path release = home.resolve("release");
        try (InputStream in = Files.newInputStream(release)) {
            props.load(in);
        }
        String value = props.getProperty("JAVA_PROFILE");
        if (isFullJre) {
            if (value != null)
                throw new RuntimeException("JAVA_PROFILE should not be present");
        } else {
            if (value == null)
                throw new RuntimeException("JAVA_PROFILE not present in release file");
            if (!value.equals("\"" + profile + "\""))
                throw new RuntimeException("Unexpected value of JAVA_PROFILE: " + value);
        }

        System.out.println("Test passed.");
    }
}