/*

 * Copyright 2003-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 4199068 4738465 4937983 4930681 4926230 4931433 4932663 4986689

 *      5026830 5023243 5070673 4052517 4811767 6192449 6397034 6413313

 *      6464154 6523983 6206031

 * @summary Basic tests for Process and Environment Variable code

 * @run main/othervm Basic

 * @author Martin Buchholz

 */

import java.io.*;

import java.util.*;

import java.security.*;

import java.util.regex.Pattern;

import static java.lang.System.getenv;

import static java.lang.System.out;

import static java.lang.Boolean.TRUE;

import static java.util.AbstractMap.SimpleImmutableEntry;

public class Basic {

    private static String commandOutput(Reader r) throws Throwable {

        StringBuilder sb = new StringBuilder();

        int c;

        while ((c = r.read()) > 0)

            if (c != '\r')

                sb.append((char) c);

        return sb.toString();

    }

    private static String commandOutput(Process p) throws Throwable {

        check(p.getInputStream()  == p.getInputStream());

        check(p.getOutputStream() == p.getOutputStream());

        check(p.getErrorStream()  == p.getErrorStream());

        Reader r = new InputStreamReader(p.getInputStream(),"UTF-8");

        String output = commandOutput(r);

        equal(p.waitFor(), 0);

        equal(p.exitValue(), 0);

        return output;

    }

    private static String commandOutput(ProcessBuilder pb) {

        try {

            return commandOutput(pb.start());

        } catch (Throwable t) {

            String commandline = "";

            for (String arg : pb.command())

                commandline += " " + arg;

            System.out.println("Exception trying to run process: " + commandline);

            unexpected(t);

            return "";

        }

    }

    private static String commandOutput(String...command) {

        try {

            return commandOutput(Runtime.getRuntime().exec(command));

        } catch (Throwable t) {

            String commandline = "";

            for (String arg : command)

                commandline += " " + arg;

            System.out.println("Exception trying to run process: " + commandline);

            unexpected(t);

            return "";

        }

    }

    private static void checkCommandOutput(ProcessBuilder pb,

                                           String expected,

                                           String failureMsg) {

        String got = commandOutput(pb);

        check(got.equals(expected),

              failureMsg + "\n" +

              "Expected: \"" + expected + "\"\n" +

              "Got: \"" + got + "\"");

    }

    private static String absolutifyPath(String path) {

        StringBuilder sb = new StringBuilder();

        for (String file : path.split(File.pathSeparator)) {

            if (sb.length() != 0)

                sb.append(File.pathSeparator);

            sb.append(new File(file).getAbsolutePath());

        }

        return sb.toString();

    }

    // compare windows-style, by canonicalizing to upper case,

    // not lower case as String.compareToIgnoreCase does

    private static class WindowsComparator

        implements Comparator<String> {

        public int compare(String x, String y) {

            return x.toUpperCase(Locale.US)

                .compareTo(y.toUpperCase(Locale.US));

        }

    }

    private static String sortedLines(String lines) {

        String[] arr = lines.split("\n");

        List<String> ls = new ArrayList<String>();

        for (String s : arr)

            ls.add(s);

        Collections.sort(ls, new WindowsComparator());

        StringBuilder sb = new StringBuilder();

        for (String s : ls)

            sb.append(s + "\n");

        return sb.toString();

    }

    private static void compareLinesIgnoreCase(String lines1, String lines2) {

        if (! (sortedLines(lines1).equalsIgnoreCase(sortedLines(lines2)))) {

            String dashes =

                "-----------------------------------------------------";

            out.println(dashes);

            out.print(sortedLines(lines1));

            out.println(dashes);

            out.print(sortedLines(lines2));

            out.println(dashes);

            out.println("sizes: " + sortedLines(lines1).length() +

                        " " + sortedLines(lines2).length());

            fail("Sorted string contents differ");

        }

    }

    private static final Runtime runtime = Runtime.getRuntime();

    private static final String[] winEnvCommand = {"cmd.exe", "/c", "set"};

    private static String winEnvFilter(String env) {

        return env.replaceAll("\r", "")

            .replaceAll("(?m)^(?:COMSPEC|PROMPT|PATHEXT)=.*\n","");

    }

    private static String unixEnvProg() {

        return new File("/usr/bin/env").canExecute() ? "/usr/bin/env"

            : "/bin/env";

    }

    private static String nativeEnv(String[] env) {

        try {

            if (Windows.is()) {

                return winEnvFilter

                    (commandOutput(runtime.exec(winEnvCommand, env)));

            } else {

                return commandOutput(runtime.exec(unixEnvProg(), env));

            }

        } catch (Throwable t) { throw new Error(t); }

    }

    private static String nativeEnv(ProcessBuilder pb) {

        try {

            if (Windows.is()) {

                pb.command(winEnvCommand);

                return winEnvFilter(commandOutput(pb));

            } else {

                pb.command(new String[]{unixEnvProg()});

                return commandOutput(pb);

            }

        } catch (Throwable t) { throw new Error(t); }

    }

    private static void checkSizes(Map<String,String> environ, int size) {

        try {

            equal(size, environ.size());

            equal(size, environ.entrySet().size());

            equal(size, environ.keySet().size());

            equal(size, environ.values().size());

            boolean isEmpty = (size == 0);

            equal(isEmpty, environ.isEmpty());

            equal(isEmpty, environ.entrySet().isEmpty());

            equal(isEmpty, environ.keySet().isEmpty());

            equal(isEmpty, environ.values().isEmpty());

        } catch (Throwable t) { unexpected(t); }

    }

    private interface EnvironmentFrobber {

        void doIt(Map<String,String> environ);

    }

    private static void testVariableDeleter(EnvironmentFrobber fooDeleter) {

        try {

            Map<String,String> environ = new ProcessBuilder().environment();

            environ.put("Foo", "BAAR");

            fooDeleter.doIt(environ);

            equal(environ.get("Foo"), null);

            equal(environ.remove("Foo"), null);

        } catch (Throwable t) { unexpected(t); }

    }

    private static void testVariableAdder(EnvironmentFrobber fooAdder) {

        try {

            Map<String,String> environ = new ProcessBuilder().environment();

            environ.remove("Foo");

            fooAdder.doIt(environ);

            equal(environ.get("Foo"), "Bahrein");

        } catch (Throwable t) { unexpected(t); }

    }

    private static void testVariableModifier(EnvironmentFrobber fooModifier) {

        try {

            Map<String,String> environ = new ProcessBuilder().environment();

            environ.put("Foo","OldValue");

            fooModifier.doIt(environ);

            equal(environ.get("Foo"), "NewValue");

        } catch (Throwable t) { unexpected(t); }

    }

    private static void printUTF8(String s) throws IOException {

        out.write(s.getBytes("UTF-8"));

    }

    private static String getenvAsString(Map<String,String> environment) {

        StringBuilder sb = new StringBuilder();

        for (Map.Entry<String,String> e : environment.entrySet())

            // Ignore magic environment variables added by the launcher

            if (! e.getKey().equals("NLSPATH") &&

                ! e.getKey().equals("XFILESEARCHPATH") &&

                ! e.getKey().equals("LD_LIBRARY_PATH"))

                sb.append(e.getKey())

                    .append('=')

                    .append(e.getValue())

                    .append(',');

        return sb.toString();

    }

    static void print4095(OutputStream s) throws Throwable {

        byte[] bytes = new byte[4095];

        Arrays.fill(bytes, (byte) '!');

        s.write(bytes);         // Might hang!

    }

    public static class JavaChild {

        public static void main(String args[]) throws Throwable {

            String action = args[0];

            if (action.equals("System.getenv(String)")) {

                String val = System.getenv(args[1]);

                printUTF8(val == null ? "null" : val);

            } else if (action.equals("System.getenv(\\u1234)")) {

                String val = System.getenv("\u1234");

                printUTF8(val == null ? "null" : val);

            } else if (action.equals("System.getenv()")) {

                printUTF8(getenvAsString(System.getenv()));

            } else if (action.equals("pwd")) {

                printUTF8(new File(System.getProperty("user.dir"))

                          .getCanonicalPath());

            } else if (action.equals("print4095")) {

                print4095(System.out);

                System.exit(5);

            } else if (action.equals("OutErr")) {

                // You might think the system streams would be

                // buffered, and in fact they are implemented using

                // BufferedOutputStream, but each and every print

                // causes immediate operating system I/O.

                System.out.print("out");

                System.err.print("err");

                System.out.print("out");

                System.err.print("err");

            } else if (action.equals("null PATH")) {

                equal(System.getenv("PATH"), null);

                check(new File("/bin/true").exists());

                check(new File("/bin/false").exists());

                ProcessBuilder pb1 = new ProcessBuilder();

                ProcessBuilder pb2 = new ProcessBuilder();

                pb2.environment().put("PATH", "anyOldPathIgnoredAnyways");

                ProcessResults r;

                for (final ProcessBuilder pb :

                         new ProcessBuilder[] {pb1, pb2}) {

                    pb.command("true");

                    r = run(pb.start());

                    equal(r.exitValue(), True.exitValue());

                    pb.command("false");

                    r = run(pb.start());

                    equal(r.exitValue(), False.exitValue());

                }

                if (failed != 0) throw new Error("null PATH");

            } else if (action.equals("PATH search algorithm")) {

                equal(System.getenv("PATH"), "dir1:dir2:");

                check(new File("/bin/true").exists());

                check(new File("/bin/false").exists());

                String[] cmd = {"prog"};

                ProcessBuilder pb1 = new ProcessBuilder(cmd);

                ProcessBuilder pb2 = new ProcessBuilder(cmd);

                ProcessBuilder pb3 = new ProcessBuilder(cmd);

                pb2.environment().put("PATH", "anyOldPathIgnoredAnyways");

                pb3.environment().remove("PATH");

                for (final ProcessBuilder pb :

                         new ProcessBuilder[] {pb1, pb2, pb3}) {

                    try {

                        // Not on PATH at all; directories don't exist

                        try {

                            pb.start();

                            fail("Expected IOException not thrown");

                        } catch (IOException e) {

                            String m = e.getMessage();

                            if (EnglishUnix.is() &&

                                ! matches(m, "No such file"))

                                unexpected(e);

                        } catch (Throwable t) { unexpected(t); }

                        // Not on PATH at all; directories exist

                        new File("dir1").mkdirs();

                        new File("dir2").mkdirs();

                        try {

                            pb.start();

                            fail("Expected IOException not thrown");

                        } catch (IOException e) {

                            String m = e.getMessage();

                            if (EnglishUnix.is() &&

                                ! matches(m, "No such file"))

                                unexpected(e);

                        } catch (Throwable t) { unexpected(t); }

                        // Can't execute a directory -- permission denied

                        // Report EACCES errno

                        new File("dir1/prog").mkdirs();

                        try {

                            pb.start();

                            fail("Expected IOException not thrown");

                        } catch (IOException e) {

                            String m = e.getMessage();

                            if (EnglishUnix.is() &&

                                ! matches(m, "Permission denied"))

                                unexpected(e);

                        } catch (Throwable t) { unexpected(t); }

                        // continue searching if EACCES

                        copy("/bin/true", "dir2/prog");

                        equal(run(pb.start()).exitValue(), True.exitValue());

                        new File("dir1/prog").delete();

                        new File("dir2/prog").delete();

                        new File("dir2/prog").mkdirs();

                        copy("/bin/true", "dir1/prog");

                        equal(run(pb.start()).exitValue(), True.exitValue());

                        // Check empty PATH component means current directory

                        new File("dir1/prog").delete();

                        new File("dir2/prog").delete();

                        copy("/bin/true", "./prog");

                        equal(run(pb.start()).exitValue(), True.exitValue());

                        // If prog found on both parent and child's PATH,

                        // parent's is used.

                        new File("dir1/prog").delete();

                        new File("dir2/prog").delete();

                        new File("prog").delete();

                        new File("dir3").mkdirs();

                        copy("/bin/true", "dir1/prog");

                        copy("/bin/false", "dir3/prog");

                        pb.environment().put("PATH","dir3");

                        equal(run(pb.start()).exitValue(), True.exitValue());

                        copy("/bin/true", "dir3/prog");

                        copy("/bin/false", "dir1/prog");

                        equal(run(pb.start()).exitValue(), False.exitValue());

                    } finally {

                        // cleanup

                        new File("dir1/prog").delete();

                        new File("dir2/prog").delete();

                        new File("dir3/prog").delete();

                        new File("dir1").delete();

                        new File("dir2").delete();

                        new File("dir3").delete();

                        new File("prog").delete();

                    }

                }

                if (failed != 0) throw new Error("PATH search algorithm");

            }

            else throw new Error("JavaChild invocation error");

        }

    }

    private static void copy(String src, String dst) {

        system("/bin/cp", "-fp", src, dst);

    }

    private static void system(String... command) {

        try {

            ProcessBuilder pb = new ProcessBuilder(command);

            ProcessResults r = run(pb.start());

            equal(r.exitValue(), 0);

            equal(r.out(), "");

            equal(r.err(), "");

        } catch (Throwable t) { unexpected(t); }

    }

    private static String javaChildOutput(ProcessBuilder pb, String...args) {

        List<String> list = new ArrayList<String>(javaChildArgs);

        for (String arg : args)

            list.add(arg);

        pb.command(list);

        return commandOutput(pb);

    }

    private static String getenvInChild(ProcessBuilder pb) {

        return javaChildOutput(pb, "System.getenv()");

    }

    private static String getenvInChild1234(ProcessBuilder pb) {

        return javaChildOutput(pb, "System.getenv(\\u1234)");

    }

    private static String getenvInChild(ProcessBuilder pb, String name) {

        return javaChildOutput(pb, "System.getenv(String)", name);

    }

    private static String pwdInChild(ProcessBuilder pb) {

        return javaChildOutput(pb, "pwd");

    }

    private static final String javaExe =

        System.getProperty("java.home") +

        File.separator + "bin" + File.separator + "java";

    private static final String classpath =

        System.getProperty("java.class.path");

    private static final List<String> javaChildArgs =

        Arrays.asList(new String[]

            { javaExe, "-classpath", absolutifyPath(classpath),

              "Basic$JavaChild"});

    private static void testEncoding(String encoding, String tested) {

        try {

            // If round trip conversion works, should be able to set env vars

            // correctly in child.

            if (new String(tested.getBytes()).equals(tested)) {

                out.println("Testing " + encoding + " environment values");

                ProcessBuilder pb = new ProcessBuilder();

                pb.environment().put("ASCIINAME",tested);

                equal(getenvInChild(pb,"ASCIINAME"), tested);

            }

        } catch (Throwable t) { unexpected(t); }

    }

    static class Windows {

        public static boolean is() { return is; }

        private static final boolean is =

            System.getProperty("os.name").startsWith("Windows");

    }

    static class Unix {

        public static boolean is() { return is; }

        private static final boolean is =

            (! Windows.is() &&

             new File("/bin/sh").exists() &&

             new File("/bin/true").exists() &&

             new File("/bin/false").exists());

    }

    static class UnicodeOS {

        public static boolean is() { return is; }

        private static final String osName = System.getProperty("os.name");

        private static final boolean is =

            // MacOS X would probably also qualify

            osName.startsWith("Windows")   &&

            ! osName.startsWith("Windows 9") &&

            ! osName.equals("Windows Me");

    }

    static class True {

        public static int exitValue() { return 0; }

    }

    private static class False {

        public static int exitValue() { return exitValue; }

        private static final int exitValue = exitValue0();

        private static int exitValue0() {

            // /bin/false returns an *unspecified* non-zero number.

            try {

                if (! Unix.is())

                    return -1;

                else {

                    int rc = new ProcessBuilder("/bin/false")

                        .start().waitFor();

                    check(rc != 0);

                    return rc;

                }

            } catch (Throwable t) { unexpected(t); return -1; }

        }

    }

    static class EnglishUnix {

        private final static Boolean is =

            (! Windows.is() && isEnglish("LANG") && isEnglish("LC_ALL"));

        private static boolean isEnglish(String envvar) {

            String val = getenv(envvar);

            return (val == null) || val.matches("en.*");

        }

        /** Returns true if we can expect English OS error strings */

        static boolean is() { return is; }

    }

    private static boolean matches(String str, String regex) {

        return Pattern.compile(regex).matcher(str).find();

    }