/*
 * Copyright (c) 2012, 2014, 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 7124089 7131021 8042469
 * @summary Checks for MacOSX specific flags are accepted or rejected, and
 *          MacOSX platforms specific environment is consistent.
 * @compile -XDignore.symbol.file TestSpecialArgs.java EnvironmentVariables.java
 * @run main TestSpecialArgs
 */
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

public class TestSpecialArgs extends TestHelper {

    public static void main(String... args) {
        final Map<String, String> envMap = new HashMap<>();
        envMap.put("_JAVA_LAUNCHER_DEBUG", "true");

        TestResult tr = doExec(envMap, javaCmd, "-XstartOnFirstThread", "-version");
        if (isMacOSX) {
            if (!tr.contains("In same thread")) {
                System.out.println(tr);
                throw new RuntimeException("Error: not running in the same thread ?");
            }
            if (!tr.isOK()) {
                System.out.println(tr);
                throw new RuntimeException("Error: arg was rejected ????");
            }
        } else {
            if (tr.isOK()) {
                System.out.println(tr);
                throw new RuntimeException("Error: argument was accepted ????");
            }
        }

        tr = doExec(javaCmd, "-Xdock:/tmp/not-available", "-version");
        if (isMacOSX) {
            if (!tr.isOK()) {
                System.out.println(tr);
                throw new RuntimeException("Error: arg was rejected ????");
            }
        } else {
            if (tr.isOK()) {
                System.out.println(tr);
                throw new RuntimeException("Error: argument was accepted ????");
            }
        }

        /*
         * test argument : -XX:NativeMemoryTracking=value
         * A JVM flag, comsumed by the JVM, but requiring launcher
         * to set an environmental variable if and only if value is supplied.
         * Test and order:
         * 1) execute with valid parameter: -XX:NativeMemoryTracking=MyValue
         *    a) check for correct env variable name: "NMT_LEVEL_" + pid
         *    b) check that "MyValue" was found in local env.
         * 2) execute with invalid parameter: -XX:NativeMemoryTracking=
         *    !) Won't find "NativeMemoryTracking:"
         *       Code to create env variable not executed.
         * 3) execute with invalid parameter: -XX:NativeMemoryTracking
         *    !) Won't find "NativeMemoryTracking:"
         *       Code to create env variable not executed.
         * 4) give and invalid value and check to make sure JVM commented
         */
        { // NativeMemoryTracking
            String launcherPidString = "launcher.pid=";
            String envVarPidString = "TRACER_MARKER: NativeMemoryTracking: env var is NMT_LEVEL_";
            String NMT_Option_Value = "off";
            String myClassName = "helloworld";
            boolean haveLauncherPid = false;

            // === Run the tests ===

            // ---Test 1a
            tr = doExec(envMap,javaCmd, "-XX:NativeMemoryTracking=" + NMT_Option_Value,
                        "-version");

            // get the PID from the env var we set for the JVM
            String envVarPid = null;
            for (String line : tr.testOutput) {
                if (line.contains(envVarPidString)) {
                    int sindex = envVarPidString.length();
                    envVarPid = line.substring(sindex);
                    break;
                }
            }
            // did we find envVarPid?
            if (envVarPid == null) {
                System.out.println(tr);
                throw new RuntimeException("Error: failed to find env Var Pid in tracking info");
            }
            // we think we found the pid string.  min test, not "".
            if (envVarPid.length() < 1) {
                System.out.println(tr);
                throw new RuntimeException("Error: env Var Pid in tracking info is empty string");
            }

            /*
             * On Linux, Launcher Tracking will print the PID.  Use this info
             * to validate what we got as the PID in the Launcher itself.
             * Linux is the only one that prints this, and trying to get it
             * here for win is awful.  So let the linux test make sure we get
             * the valid pid, and for non-linux, just make sure pid string is
             * non-zero.
             */
            if (isLinux) {
                // get what the test says is the launcher pid
                String launcherPid = null;
                for (String line : tr.testOutput) {
                    int index = line.indexOf(launcherPidString);
                    if (index >= 0) {
                        int sindex = index + launcherPidString.length();
                        int tindex = sindex + line.substring(sindex).indexOf("'");
                        System.out.println("DEBUG INFO: sindex = " + sindex);
                        System.out.println("DEBUG INFO: searching substring: " + line.substring(sindex));
                        System.out.println("DEBUG INFO: tindex = " + tindex);
                        // DEBUG INFO
                        System.out.println(tr);
                        launcherPid = line.substring(sindex, tindex);
                        break;
                    }
                }
                if (launcherPid == null) {
                    System.out.println(tr);
                    throw new RuntimeException("Error: failed to find launcher Pid in launcher tracking info");
                }

                // did we create the env var with the correct pid?
                if (!launcherPid.equals(envVarPid)) {
                    System.out.println(tr);
                    System.out.println("Error: wrong pid in creating env var");
                    System.out.println("Error Info: launcherPid = " + launcherPid);
                    System.out.println("Error Info: envVarPid   = " + envVarPid);
                    throw new RuntimeException("Error: wrong pid in creating env var");
                }
            }


            // --- Test 1b
            if (!tr.contains("NativeMemoryTracking: got value " + NMT_Option_Value)) {
                System.out.println(tr);
                throw new RuntimeException("Error: Valid param failed to set env variable");
            }

            // --- Test 2
            tr = doExec(envMap,javaCmd, "-XX:NativeMemoryTracking=",
                        "-version");
            if (tr.contains("NativeMemoryTracking:")) {
                System.out.println(tr);
                throw new RuntimeException("Error: invalid param caused env variable to be erroneously created");
            }
            if (!tr.contains("Syntax error, expecting -XX:NativeMemoryTracking=")) {
                System.out.println(tr);
                throw new RuntimeException("Error: invalid param not checked by JVM");
            }

            // --- Test 3
            tr = doExec(envMap,javaCmd, "-XX:NativeMemoryTracking",
                        "-version");
            if (tr.contains("NativeMemoryTracking:")) {
                System.out.println(tr);
                throw new RuntimeException("Error: invalid param caused env variable to be erroneously created");
            }
            if (!tr.contains("Syntax error, expecting -XX:NativeMemoryTracking=")) {
                System.out.println(tr);
                throw new RuntimeException("Error: invalid param not checked by JVM");
            }
            // --- Test 4
            tr = doExec(envMap,javaCmd, "-XX:NativeMemoryTracking=BADVALUE",
                        "-version");
            if (!tr.contains("expecting -XX:NativeMemoryTracking")) {
                System.out.println(tr);
                throw new RuntimeException("Error: invalid param did not get JVM Syntax error message");
            }

        } // NativeMemoryTracking


        // MacOSX specific tests ensue......
        if (!isMacOSX)
            return;
        Set<String> envToRemove = new HashSet<>();
        Map<String, String> map = System.getenv();
        for (String s : map.keySet()) {
            if (s.startsWith("JAVA_MAIN_CLASS_")
                    || s.startsWith("APP_NAME_")
                    || s.startsWith("APP_ICON_")) {
                envToRemove.add(s);
            }
        }
        runTest(envToRemove, javaCmd, "-cp", TEST_CLASSES_DIR.getAbsolutePath(),
                "EnvironmentVariables", "JAVA_MAIN_CLASS_*",
                "EnvironmentVariables");

        runTest(envToRemove, javaCmd, "-cp", TEST_CLASSES_DIR.getAbsolutePath(),
                "-Xdock:name=TestAppName", "EnvironmentVariables",
                "APP_NAME_*", "TestAppName");

        runTest(envToRemove, javaCmd, "-cp", TEST_CLASSES_DIR.getAbsolutePath(),
                "-Xdock:icon=TestAppIcon", "EnvironmentVariables",
                "APP_ICON_*", "TestAppIcon");
    }

    static void runTest(Set<String> envToRemove, String... args) {
        TestResult tr = doExec(null, envToRemove, args);
        if (!tr.isOK()) {
            System.err.println(tr.toString());
            throw new RuntimeException("Test Fails");
        }
    }
}