/*
 * Copyright (c) 2013, 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 7147084
 * @run main/othervm InheritIOEHandle
 * @summary inherit IOE handles and MS CreateProcess limitations (kb315939)
 */

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;

public class InheritIOEHandle {
    private static enum APP {
        B, C;
    }
    private static File stopC = new File(".\\StopC.txt");
    private static String SIGNAL = "After call child process";
    private static String JAVA_EXE = System.getProperty("java.home")
        + File.separator + "bin"
        + File.separator + "java";

    private static String[] getCommandArray(String processName) {
        String[] cmdArray = {
            JAVA_EXE,
            "-cp",
            System.getProperty("java.class.path"),
            InheritIOEHandle.class.getName(),
            processName
        };
        return cmdArray;
    }

    public static void main(String[] args) throws Exception {
        if (!System.getProperty("os.name").startsWith("Windows")) {
            return;
        }

        if (args.length > 0) {
            APP app = APP.valueOf(args[0]);
            switch (app) {
            case B:
                performB();
                break;
            case C:
                performC();
                break;
            }
            return;
        }
        performA();
    }

    private static void performA() {
        try {
            stopC.delete();

            ProcessBuilder builder = new ProcessBuilder(
                    getCommandArray(APP.B.name()));
            builder.redirectErrorStream(true);

            Process process = builder.start();

            process.getOutputStream().close();
            process.getErrorStream().close();

            try (BufferedReader in = new BufferedReader( new InputStreamReader(
                         process.getInputStream(), "utf-8")))
            {
                String result;
                while ((result = in.readLine()) != null) {
                    if (!SIGNAL.equals(result)) {
                        throw new Error("Catastrophe in process B! Bad output.");
                    }
                }
            }

            // If JDK-7147084 is not fixed that point is unreachable.

            // write signal file
            stopC.createNewFile();

            System.err.println("Read stream finished.");
        } catch (IOException ex) {
            throw new Error("Catastrophe in process A!", ex);
        }
    }

    private static void performB() {
        try {
            ProcessBuilder builder = new ProcessBuilder(
                    getCommandArray(APP.C.name()));

            Process process = builder.start();

            process.getInputStream().close();
            process.getOutputStream().close();
            process.getErrorStream().close();

            System.out.println(SIGNAL);

            // JDK-7147084 subject:
            // Process C inherits the [System.out] handle and
            // handle close in B does not finalize the streaming for A.
            // (handle reference count > 1).
        } catch (IOException ex) {
            throw new Error("Catastrophe in process B!", ex);
        }
    }

    private static void performC() {
        // If JDK-7147084 is not fixed the loop is 5min long.
        for (int i = 0; i < 5*60; ++i) {
            try {
                Thread.sleep(1000);
                // check for sucess
                if (stopC.exists())
                    break;
            } catch (InterruptedException ex) {
                // that is ok. Longer sleep - better effect.
            }
        }
    }
}