/*

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

 */

import jdk.test.lib.Platform;

import jdk.test.lib.Utils;

import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.Arrays;

/**

 * Starts a new process to execute a command.

 * <p>Usage: --java|--cmd|--launcher <arg>+

 * <p>If {@code --cmd} flag is specified, the arguments are treated as

 * a program to run and its arguments. Non-zero exit code of the created process

 * will be reported as an {@link AssertionError}.

 * <p>If {@code --java} flag is specified, the arguments are passed to {@code java}

 * from JDK under test. If exit code doesn't equal to 0 or 95, {@link AssertionError}

 * will be thrown.

 * <p>If {@code --launcher} flag is specified, the arguments treated similar as

 * for {@code --cmd}, but the started process will have the directory which

 * contains {@code jvm.so} in dynamic library path, and {@code test.class.path}

 * as CLASSPATH environment variable. Exit codes are checked as in

 * {@code --java}, i.e. 0 or 95 means pass.

 */

public class ExecDriver {

    public static void main(String[] args) throws IOException, InterruptedException {

        boolean java = false;

        boolean launcher = false;

        String type = args[0];

        switch (type) {

            case "--java":

                String[] oldArgs = args;

                int count;

                String libraryPath = System.getProperty("test.nativepath");

                if (libraryPath != null && !libraryPath.isEmpty()) {

                    count = 4;

                    args = new String[args.length + 3];

                    args[3] = "-Djava.library.path=" + libraryPath;

                } else {

                    count = 3;

                    args = new String[args.length + 2];

                }

                args[0] = javaBin();

                args[1] = "-cp";

                args[2] = Utils.TEST_CLASS_PATH;

                System.arraycopy(oldArgs, 1, args, count, oldArgs.length - 1);

                java = true;

                break;

            case "--launcher":

                java = true;

                launcher = true;

            case "--cmd":

                args = Arrays.copyOfRange(args, 1, args.length);

                break;

            default:

                throw new Error("unknown type: " + type);

        }

        // adding 'test.vm.opts' and 'test.java.opts'

        if (java) {

            String[] oldArgs = args;

            String[] testJavaOpts = Utils.getTestJavaOpts();

            if (testJavaOpts.length > 0) {

                args = new String[args.length + testJavaOpts.length];

                // bin/java goes before options

                args[0] = oldArgs[0];

                // then external java options

                System.arraycopy(testJavaOpts, 0, args, 1, testJavaOpts.length);

                // and then options and args from a test

                System.arraycopy(oldArgs, 1, args, 1 + testJavaOpts.length, oldArgs.length - 1);

            }

        }

        String command = Arrays.toString(args);

        System.out.println("exec " + command);

        ProcessBuilder pb = new ProcessBuilder(args);

        // adding jvm.so to library path

        if (launcher) {

            Path dir = Paths.get(Utils.TEST_JDK);

            String value;

            String name;

            if (Platform.isWindows()) {

                value = dir.resolve("bin")

                           .resolve(variant())

                           .toAbsolutePath()

                           .toString();

                value += File.pathSeparator;

                value += dir.resolve("bin")

                            .toAbsolutePath()

                            .toString();

                name = "PATH";

            } else {

                value = dir.resolve("lib")

                           .resolve(variant())

                           .toAbsolutePath()

                           .toString();

                name = Platform.isOSX() ? "DYLD_LIBRARY_PATH" : "LD_LIBRARY_PATH";

            }

            System.out.println("  with " + name + " = " +

                    pb.environment()

                      .merge(name, value, (x, y) -> y + File.pathSeparator + x));

            System.out.println("  with CLASSPATH = " +

                    pb.environment()

                      .put("CLASSPATH", Utils.TEST_CLASS_PATH));

        }

        Process p = pb.start();

        // inheritIO does not work as expected for @run driver

        new Thread(() -> copy(p.getInputStream(), System.out)).start();

        new Thread(() -> copy(p.getErrorStream(), System.out)).start();

        int exitCode = p.waitFor();

        if (exitCode != 0 && (!java || exitCode != 95)) {

            throw new AssertionError(command + " exit code is " + exitCode);

        }

    }

    private static String variant() {

        if (Platform.isServer()) {

            return "server";

        } else if (Platform.isClient()) {

            return "client";

        } else if (Platform.isMinimal()) {

            return "minimal";

        } else {

            throw new Error("TESTBUG: unsuppported vm variant");

        }

    }

    private static void copy(InputStream is, OutputStream os) {

        byte[] buffer = new byte[1024];

        int n;

        try (InputStream close = is) {

            while ((n = is.read(buffer)) != -1) {

                os.write(buffer, 0, n);

            }

            os.flush();

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    private static String javaBin() {

        return Paths.get(Utils.TEST_JDK)

                    .resolve("bin")

                    .resolve("java")

                    .toAbsolutePath()

                    .toString();

    }

}