/*

 * Copyright (c) 2016, 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 java.io.OutputStream;

import java.io.IOException;

/**

 * Class to run and control rmiregistry in a sub-process.

 *

 * We can't kill a registry if we have too-close control

 * over it.  We must make it in a subprocess, and then kill the

 * subprocess when it has served our needs.

 */

public class REGISTRY extends JavaVM {

    private static double startTimeout = 20_000 * TestLibrary.getTimeoutFactor();

    private int port = -1;

    private REGISTRY(OutputStream out, OutputStream err,

                    String options, int port) {

        super("RegistryRunner", options, Integer.toString(port), out, err);

        this.port = port;

    }

    public static REGISTRY createREGISTRY() {

        return createREGISTRY(System.out, System.err, "", 0);

    }

    public static REGISTRY createREGISTRY(OutputStream out, OutputStream err,

                                    String options, int port) {

        options += " --add-exports=java.rmi/sun.rmi.registry=ALL-UNNAMED"

                 + " --add-exports=java.rmi/sun.rmi.server=ALL-UNNAMED"

                 + " --add-exports=java.rmi/sun.rmi.transport=ALL-UNNAMED"

                 + " --add-exports=java.rmi/sun.rmi.transport.tcp=ALL-UNNAMED";

        REGISTRY reg = new REGISTRY(out, err, options, port);

        return reg;

    }

    /**

     * Starts the registry in a sub-process and waits up to

     * the given timeout period to confirm that it's running,

     * and get the port where it's running.

     */

    public void start() throws IOException {

        super.start();

        long startTime = System.currentTimeMillis();

        long deadline = TestLibrary.computeDeadline(startTime, (long)startTimeout);

        while (true) {

            try {

                Thread.sleep(1000);

            } catch (InterruptedException ignore) { }

            String output = outputStream.ba.toString();

            port = RegistryRunner.getRegistryPort(output);

            if (port != -1) {

                break;

            }

            if (System.currentTimeMillis() > deadline) {

                TestLibrary.bomb("Failed to start registry, giving up after " +

                    (System.currentTimeMillis() - startTime) + "ms.", null);

            }

        }

    }

    /**

     * Shuts down the registry.

     */

    public void shutdown() {

        RegistryRunner.requestExit(port);

    }

    /**

     * Gets the port where the registry is serving.

     */

    public int getPort() {

        return port;

    }

}