/*

 * Copyright (c) 1999, 2008, 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 4208804

 *

 * @summary Incoming connections should be subject to timeout

 * @author Adrian Colley

 *

 * @build TestIface TestImpl TestImpl_Stub

 * @run main/othervm/policy=security.policy/timeout=60

 *     -Dsun.rmi.transport.tcp.readTimeout=5000 ReadTimeoutTest

 */

/* This test sets a very short read timeout, exports an object, and then

 * connects to the port and does nothing.  The server should close the

 * connection after the timeout.  If that doesn't happen, the test fails.

 *

 * A background thread does the read.  The foreground waits for DELAY*4

 * and then aborts.  This should give sufficient margin for timing slop.

 */

import java.rmi.*;

import java.rmi.server.RMISocketFactory;

import java.io.*;

import java.net.*;

public class ReadTimeoutTest

{

    private static final int DELAY = 5000;      // milliseconds

    public static void main(String[] args)

        throws Exception

    {

        // Make trouble for ourselves

        if (System.getSecurityManager() == null)

            System.setSecurityManager(new RMISecurityManager());

        // Flaky code alert - hope this is executed before TCPTransport.<clinit>

        System.setProperty("sun.rmi.transport.tcp.readTimeout", Integer.toString(DELAY));

        // Set the socket factory.

        System.err.println("(installing socket factory)");

        SomeFactory fac = new SomeFactory();

        RMISocketFactory.setSocketFactory(fac);

        // Create remote object

        TestImpl impl = new TestImpl();

        // Export and get which port.

        System.err.println("(exporting remote object)");

        TestIface stub = impl.export();

        Socket DoS = null;

        try {

            int port = fac.whichPort();

            // Sanity

            if (port == 0)

                throw new Error("TEST FAILED: export didn't reserve a port(?)");

            // Now, connect to that port

            //Thread.sleep(2000);

            System.err.println("(connecting to listening port on 127.0.0.1:" +

                               port + ")");

            DoS = new Socket("127.0.0.1", port);

            InputStream stream = DoS.getInputStream();

            // Read on the socket in the background

            // Wait for completion

                System.err.println("TEST PASSED.");

            } else {

                throw new Error("TEST FAILED.");

            }

        } finally {

            try {

                if (DoS != null)

                    DoS.close();        // aborts the reader if still blocked

                impl.unexport();

            } catch (Throwable unmatter) {

            }

        }

        // Should exit here

    }

    private static class SomeFactory

        extends RMISocketFactory

    {

        private int servport = 0;

        public Socket createSocket(String h, int p)

            throws IOException

        {

            return (new Socket(h, p));

        }

        /** Create a server socket and remember which port it's on.

         * Aborts if createServerSocket(0) is called twice, because then

         * it doesn't know whether to remember the first or second port.

         */

        public ServerSocket createServerSocket(int p)

            throws IOException

        {

            ServerSocket ss;

            ss = new ServerSocket(p);

            if (p == 0) {

                if (servport != 0) {

                    System.err.println("TEST FAILED: " +

                                       "Duplicate createServerSocket(0)");

                    throw new Error("Test aborted (createServerSocket)");

                }

                servport = ss.getLocalPort();

            }

            return (ss);

        }

        /** Return which port was reserved by createServerSocket(0).

         * If the return value was 0, createServerSocket(0) wasn't called.

         */

        public int whichPort() {

            return (servport);

        }

    } // end class SomeFactory

    protected static class SomeReader extends Thread {

            super();

            this.setDaemon(true);

            this.readon = s;

        }

        public void run() {

            try {

                int c = this.readon.read();

                if (c != -1)

                    throw new Error ("Server returned " + c);

            } catch (IOException e) {

                e.printStackTrace();

            }

        }

    } // end class SomeReader

}