test/java/lang/ProcessBuilder/SiblingIOEHandle.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 6921885
 * @run main/othervm SiblingIOEHandle
 * @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;
import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CyclicBarrier;

public class SiblingIOEHandle {
    private static enum APP {
        B, C;
    }
    private static File stopC = new File(".\\StopCs.txt");
    private static String SIGNAL = "B child reported.";
    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"),
            SiblingIOEHandle.class.getName(),
            processName
        };
        return cmdArray;
    }

    public static void main(String[] args) {
        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(true);
        performA(false);
    }

    static boolean procClaunched = false;

    private static void waitAbit() {
        try {
            Thread.sleep(0);
        } catch (InterruptedException ex) {
            // that was long enough
        }
    }
    private static boolean waitBarrier(CyclicBarrier barrier) {
        while (true) try {
            barrier.await();
            return true;
        } catch (InterruptedException ex) {
            continue;
        } catch (BrokenBarrierException ex) {
            ex.printStackTrace();
            return false;
        }
    }

    private static void performA(boolean fileOut) {
        try {
            stopC.delete();
            ProcessBuilder builderB = new ProcessBuilder(
                    getCommandArray(APP.B.name()));

            File outB = null;
            if (fileOut) {
                outB = new File("outB.txt");
                builderB.redirectOutput(outB);
            }
            builderB.redirectErrorStream(true);

            final CyclicBarrier barrier = new CyclicBarrier(2);
            Thread procCRunner = new Thread(new Runnable() {
                @Override public void run() {
                    try {
                        if (waitBarrier(barrier)) {
                            waitAbit();
                            // Run process C next to B ASAP to make an attempt
                            // to capture the B-process IOE handles in C process.
                            Runtime.getRuntime().exec(getCommandArray(APP.C.name()));
                            procClaunched = true;
                        }
                    } catch (IOException ex) {
                        ex.printStackTrace();
                    }
                }
            });
            procCRunner.start();


            if (!waitBarrier(barrier)) {
                throw new Error("Catastrophe in process A! Synchronization failed.");
            }
            // Run process B first.
            Process processB = builderB.start();

            while (true) try {
                procCRunner.join();
                break;
            } catch (InterruptedException ex) {
                continue;
            }

            if (!procClaunched) {
                throw new Error("Catastrophe in process A! C was not launched.");
            }

            processB.getOutputStream().close();
            processB.getErrorStream().close();

            if (fileOut) {
                try {
                    processB.waitFor();
                } catch (InterruptedException ex) {
                    throw new Error("Catastrophe in process B! B hung up.");
                }
                System.err.println("Trying to delete [outB.txt].");
                if (!outB.delete()) {
                    throw new Error("Greedy brother C deadlock! File share.");
                }
                System.err.println("Succeeded in delete [outB.txt].");
            } else {
                System.err.println("Read stream start.");
                try (BufferedReader in = new BufferedReader( new InputStreamReader(
                             processB.getInputStream(), "utf-8")))
                {
                    String result;
                    while ((result = in.readLine()) != null) {
                        if (!SIGNAL.equals(result)) {
                            throw new Error("Catastrophe in process B! Bad output.");
                        }
                    }
                }
                System.err.println("Read stream finished.");
            }
            // If JDK-6921885 is not fixed that point is unreachable.
            // Test timeout exception.

            // write signal file to stop C process.
            stopC.createNewFile();
        } catch (IOException ex) {
            throw new Error("Catastrophe in process A!", ex);
        }
    }

    private static void performB() {
        System.out.println(SIGNAL);
    }

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