/*

 * 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.

            }

        }

    }

}