/*
 * Copyright 2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

/**
 * @test
 * @bug 6545058 6611182
 * @summary validate and test -version, -fullversion, and internal
 * @compile VersionCheck.java
 * @run main VersionCheck
 */

import java.lang.*;
import java.io.File;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.Map;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class VersionCheck {

    private static String javaBin;

    // A known set of programs we know for sure will behave correctly.
    private static String[] programs = new String[]{
        "appletviewer",
        "extcheck",
        "idlj",
        "jar",
        "jarsigner",
        "javac",
        "javadoc",
        "javah",
        "javap",
        "jconsole",
        "jdb",
        "jhat",
        "jinfo",
        "jmap",
        "jps",
        "jstack",
        "jstat",
        "jstatd",
        "keytool",
        "native2ascii",
        "orbd",
        "pack200",
        "policytool",
        "rmic",
        "rmid",
        "rmiregistry",
        "schemagen",
        "serialver",
        "servertool",
        "tnameserv",
        "wsgen",
        "wsimport",
        "xjc"
    };

    // expected reference strings
    static String refVersion;
    static String refFullVersion;

    private static List<String> getProcessStreamAsList(boolean javaDebug,
                                                       String... argv) {
        List<String> out = new ArrayList<String>();
        List<String> javaCmds = new ArrayList<String>();

        String prog = javaBin + File.separator + argv[0];
        if (System.getProperty("os.name").startsWith("Windows")) {
            prog = prog.concat(".exe");
        }
        javaCmds.add(prog);
        for (int i = 1; i < argv.length; i++) {
            javaCmds.add(argv[i]);
        }

        ProcessBuilder pb = new ProcessBuilder(javaCmds);
        Map<String, String> env = pb.environment();
        if (javaDebug) {
            env.put("_JAVA_LAUNCHER_DEBUG", "true");
        }
        try {
            Process p = pb.start();
            BufferedReader r = (javaDebug) ?
                new BufferedReader(new InputStreamReader(p.getInputStream())) :
                new BufferedReader(new InputStreamReader(p.getErrorStream())) ;

            String s = r.readLine();
            while (s != null) {
                out.add(s.trim());
                s = r.readLine();
            }
            p.waitFor();
            p.destroy();
        } catch (Exception ex) {
            ex.printStackTrace();
            throw new RuntimeException(ex.getMessage());
        }
        return out;
    }

    static String getVersion(String... argv) {
        List<String> alist = getProcessStreamAsList(false, argv);
        if (alist.size() == 0) {
            throw new AssertionError("unexpected process returned null");
        }
        StringBuilder out = new StringBuilder();
        // remove the HotSpot line
        for (String x : alist) {
            if (!x.contains("HotSpot")) {
                out = out.append(x + "\n");
            }
        }
        return out.toString();
    }

    static boolean compareVersionStrings() {
        int failcount = 0;
        for (String x : programs) {
            String testStr;

            testStr = getVersion(x, "-J-version");
            if (refVersion.compareTo(testStr) != 0) {
                failcount++;
                System.out.println("Error: " + x +
                                   " fails -J-version comparison");
                System.out.println("Expected:");
                System.out.print(refVersion);
                System.out.println("Actual:");
                System.out.print(testStr);
            }

            testStr = getVersion(x, "-J-fullversion");
            if (refFullVersion.compareTo(testStr) != 0) {
                failcount++;
                System.out.println("Error: " + x +
                                   " fails -J-fullversion comparison");
                System.out.println("Expected:");
                System.out.print(refFullVersion);
                System.out.println("Actual:");
                System.out.print(testStr);
            }
        }
        System.out.println("Version Test: " + failcount);
        return failcount == 0;
    }

    static boolean compareInternalStrings() {
        int failcount = 0;
        String bStr = refVersion.substring(refVersion.lastIndexOf("build") +
                                           "build".length() + 1,
                                           refVersion.lastIndexOf(")"));

        String[] vStr = bStr.split("\\.|-|_");
        String jdkMajor = vStr[0];
        String jdkMinor = vStr[1];
        String jdkMicro = vStr[2];
        String jdkBuild = vStr[vStr.length - 1];

        String expectedDotVersion = "dotversion:" + jdkMajor + "." + jdkMinor;
        String expectedFullVersion = "fullversion:" + bStr;

        List<String> alist = getProcessStreamAsList(true, "java", "-version");

        if (!alist.contains(expectedDotVersion)) {
            System.out.println("Error: could not find " + expectedDotVersion);
            failcount++;
        }

        if (!alist.contains(expectedFullVersion)) {
            System.out.println("Error: could not find " + expectedFullVersion);
            failcount++;
        }
        System.out.println("Internal Strings Test: " + failcount);
        return failcount == 0;
    }

    // Initialize
    static void init() {
        String javaHome = System.getProperty("java.home");
        if (javaHome.endsWith("jre")) {
            javaHome = new File(javaHome).getParent();
        }
        javaBin = javaHome + File.separator + "bin";
        refVersion = getVersion("java", "-version");
        refFullVersion = getVersion("java", "-fullversion");
    }

    public static void main(String[] args) {
        init();
        if (compareVersionStrings() && compareInternalStrings()) {
            System.out.println("All Version string comparisons: PASS");
        } else {
            throw new AssertionError("Some tests failed");
        }
    }
}