jdk/test/java/lang/ProcessBuilder/InheritIOEHandle.java

/*
 * 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 {
        A, 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;
        }

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

    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();

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

            }
            if (!isSignalReceived) {
                throw new RuntimeException("Signal from B was not received");
            }

            // If JDK-7147084 is not fixed that point is unreachable.
            System.out.println("Received signal from B, creating file StopC");
            // write signal file
            boolean isFileStopC = stopC.createNewFile();
            if (!isFileStopC) {
                throw new RuntimeException("Signal file StopC.txt was not created. TEST or INFRA bug");
            }

            process.waitFor();

            System.err.println("Read stream finished.");
        } catch (IOException ex) {
            throw new RuntimeException("Catastrophe in process A!", ex);
        } catch (InterruptedException ex) {
            throw new RuntimeException("A was interrupted while waiting for B", 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);
            process.waitFor();

            // 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 RuntimeException("Catastrophe in process B!", ex);
        } catch (InterruptedException ex) {
            throw new RuntimeException("B was interrupted while waiting for C", 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);
            } catch (InterruptedException ex) {
                // that is ok. Longer sleep - better effect.
            }
            // check for success
            if (stopC.exists()) {
                break;
            }
        }
    }
}