/*

 * Copyright (c) 2019, 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

 * @library /test/lib

 * @build jdk.test.lib.Utils

 * @run testng/timeout=180 Timeouts

 * @summary Test Socket timeouts

 */

import java.io.Closeable;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.net.ConnectException;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.SocketAddress;

import java.net.SocketException;

import java.net.SocketTimeoutException;

import java.util.concurrent.Executors;

import java.util.concurrent.ExecutionException;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.Future;

import java.util.concurrent.ScheduledExecutorService;

import java.util.concurrent.TimeUnit;

import org.testng.annotations.Test;

import static org.testng.Assert.*;

import jdk.test.lib.Utils;

@Test

public class Timeouts {

    /**

     * Test timed connect where connection is established

     */

    public void testTimedConnect1() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            try (Socket s = new Socket()) {

                s.connect(ss.getLocalSocketAddress(), 2000);

            }

        }

    }

    /**

     * Test timed connect where connection is refused

     */

    public void testTimedConnect2() throws IOException {

        try (Socket s = new Socket()) {

            SocketAddress remote = Utils.refusingEndpoint();

            try {

                s.connect(remote, 2000);

            } catch (ConnectException expected) { }

        }

    }

    /**

     * Test connect with a timeout of Integer.MAX_VALUE

     */

    public void testTimedConnect3() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            try (Socket s = new Socket()) {

                s.connect(ss.getLocalSocketAddress(), Integer.MAX_VALUE);

            }

        }

    }

    /**

     * Test connect with a negative timeout.

     */

    public void testTimedConnect4() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            try (Socket s = new Socket()) {

                try {

                    s.connect(ss.getLocalSocketAddress(), -1);

                    assertTrue(false);

                } catch (IllegalArgumentException expected) { }

            }

        }

    }

    /**

     * Test timed read where the read succeeds immediately

     */

    public void testTimedRead1() throws IOException {

        withConnection((s1, s2) -> {

            s1.getOutputStream().write(99);

            s2.setSoTimeout(30*1000);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

        });

    }

    /**

     * Test timed read where the read succeeds after a delay

     */

    public void testTimedRead2() throws IOException {

        withConnection((s1, s2) -> {

            scheduleWrite(s1.getOutputStream(), 99, 2000);

            s2.setSoTimeout(30*1000);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

        });

    }

    /**

     * Test timed read where the read times out

     */

    public void testTimedRead3() throws IOException {

        withConnection((s1, s2) -> {

            s2.setSoTimeout(2000);

            try {

                s2.getInputStream().read();

                assertTrue(false);

            } catch (SocketTimeoutException expected) {

                int timeout = s2.getSoTimeout();

            }

        });

    }

    /**

     * Test timed read that succeeds after a previous read has timed out

     */

    public void testTimedRead4() throws IOException {

        withConnection((s1, s2) -> {

            s2.setSoTimeout(2000);

            try {

                s2.getInputStream().read();

                assertTrue(false);

            } catch (SocketTimeoutException e) { }

            s1.getOutputStream().write(99);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

        });

    }

    /**

     * Test timed read that succeeds after a previous read has timed out and

     * after a short delay

     */

    public void testTimedRead5() throws IOException {

        withConnection((s1, s2) -> {

            s2.setSoTimeout(2000);

            try {

                s2.getInputStream().read();

                assertTrue(false);

            } catch (SocketTimeoutException e) { }

            s2.setSoTimeout(30*3000);

            scheduleWrite(s1.getOutputStream(), 99, 2000);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

        });

    }

    /**

     * Test untimed read that succeeds after a previous read has timed out

     */

    public void testTimedRead6() throws IOException {

        withConnection((s1, s2) -> {

            s2.setSoTimeout(2000);

            try {

                s2.getInputStream().read();

                assertTrue(false);

            } catch (SocketTimeoutException e) { }

            s1.getOutputStream().write(99);

            s2.setSoTimeout(0);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

        });

    }

    /**

     * Test untimed read that succeeds after a previous read has timed out and

     * after a short delay

     */

    public void testTimedRead7() throws IOException {

        withConnection((s1, s2) -> {

            s2.setSoTimeout(2000);

            try {

                s2.getInputStream().read();

                assertTrue(false);

            } catch (SocketTimeoutException e) { }

            scheduleWrite(s1.getOutputStream(), 99, 2000);

            s2.setSoTimeout(0);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

        });

    }

    /**

     * Test async close of timed read

     */

    public void testTimedRead8() throws IOException {

        withConnection((s1, s2) -> {

            s2.setSoTimeout(30*1000);

            scheduleClose(s2, 2000);

            try {

                s2.getInputStream().read();

                assertTrue(false);

            } catch (SocketException expected) { }

        });

    }

    /**

     * Test read with a timeout of Integer.MAX_VALUE

     */

    public void testTimedRead9() throws IOException {

        withConnection((s1, s2) -> {

            scheduleWrite(s1.getOutputStream(), 99, 2000);

            s2.setSoTimeout(Integer.MAX_VALUE);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

        });

    }

    /**

     * Test writing after a timed read.

     */

    public void testTimedWrite1() throws IOException {

        withConnection((s1, s2) -> {

            s1.getOutputStream().write(99);

            s2.setSoTimeout(3000);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

            // schedule thread to read s1 to EOF

            scheduleReadToEOF(s1.getInputStream(), 3000);

            // write a lot so that write blocks

            byte[] data = new byte[128*1024];

            for (int i = 0; i < 100; i++) {

                s2.getOutputStream().write(data);

            }

        });

    }

    /**

     * Test async close of writer (after a timed read).

     */

    public void testTimedWrite2() throws IOException {

        withConnection((s1, s2) -> {

            s1.getOutputStream().write(99);

            s2.setSoTimeout(3000);

            int b = s2.getInputStream().read();

            assertTrue(b == 99);

            // schedule s2 to be be closed

            scheduleClose(s2, 3000);

            // write a lot so that write blocks

            byte[] data = new byte[128*1024];

            try {

                while (true) {

                    s2.getOutputStream().write(data);

                }

            } catch (SocketException expected) { }

        });

    }

    /**

     * Test timed accept where a connection is established immediately

     */

    public void testTimedAccept1() throws IOException {

        Socket s1 = null;

        Socket s2 = null;

        try (ServerSocket ss = new ServerSocket(0)) {

            s1 = new Socket();

            s1.connect(ss.getLocalSocketAddress());

            ss.setSoTimeout(30*1000);

            s2 = ss.accept();

        } finally {

            if (s1 != null) s1.close();

            if (s2 != null) s2.close();

        }

    }

    /**

     * Test timed accept where a connection is established after a short delay

     */

    public void testTimedAccept2() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(30*1000);

            scheduleConnect(ss.getLocalSocketAddress(), 2000);

            Socket s = ss.accept();

            s.close();

        }

    }

    /**

     * Test timed accept where the accept times out

     */

    public void testTimedAccept3() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(2000);

            try {

                Socket s = ss.accept();

                s.close();

                assertTrue(false);

            } catch (SocketTimeoutException expected) {

                int timeout = ss.getSoTimeout();

            }

        }

    }

    /**

     * Test timed accept where a connection is established immediately after a

     * previous accept timed out.

     */

    public void testTimedAccept4() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(2000);

            try {

                Socket s = ss.accept();

                s.close();

                assertTrue(false);

            } catch (SocketTimeoutException expected) { }

            try (Socket s1 = new Socket()) {

                s1.connect(ss.getLocalSocketAddress());

                Socket s2 = ss.accept();

                s2.close();

            }

        }

    }

    /**

     * Test untimed accept where a connection is established after a previous

     * accept timed out

     */

    public void testTimedAccept5() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(2000);

            try {

                Socket s = ss.accept();

                s.close();

                assertTrue(false);

            } catch (SocketTimeoutException expected) { }

            ss.setSoTimeout(0);

            try (Socket s1 = new Socket()) {

                s1.connect(ss.getLocalSocketAddress());

                Socket s2 = ss.accept();

                s2.close();

            }

        }

    }

    /**

     * Test untimed accept where a connection is established after a previous

     * accept timed out and after a short delay

     */

    public void testTimedAccept6() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(2000);

            try {

                Socket s = ss.accept();

                s.close();

                assertTrue(false);

            } catch (SocketTimeoutException expected) { }

            ss.setSoTimeout(0);

            scheduleConnect(ss.getLocalSocketAddress(), 2000);

            Socket s = ss.accept();

            s.close();

        }

    }

    /**

     * Test async close of a timed accept

     */

    public void testTimedAccept7() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(30*1000);

            long delay = 2000;

            scheduleClose(ss, delay);

            try {

                ss.accept().close();

                assertTrue(false);

            } catch (SocketException expected) {

            }

        }

    }

    /**

     * Test timed accept with the thread interrupt status set.

     */

    public void testTimedAccept8() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(2000);

            Thread.currentThread().interrupt();

            try {

                Socket s = ss.accept();

                s.close();

                assertTrue(false);

            } catch (SocketTimeoutException expected) {

                // accept should have blocked for 2 seconds

                int timeout = ss.getSoTimeout();

                assertTrue(Thread.currentThread().isInterrupted());

            } finally {

                Thread.interrupted(); // clear interrupt status

            }

        }

    }

    /**

     * Test interrupt of thread blocked in timed accept.

     */

    public void testTimedAccept9() throws IOException {

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(4000);

            // interrupt thread after 1 second

            Future<?> interrupter = scheduleInterrupt(Thread.currentThread(), 1000);

            try {

                Socket s = ss.accept();   // should block for 4 seconds

                s.close();

                assertTrue(false);

            } catch (SocketTimeoutException expected) {

                // accept should have blocked for 4 seconds

                int timeout = ss.getSoTimeout();

                assertTrue(Thread.currentThread().isInterrupted());

            } finally {

                interrupter.cancel(true);

                Thread.interrupted(); // clear interrupt status

            }

        }

    }

    /**

     * Test two threads blocked in timed accept where no connection is established.

     */

    public void testTimedAccept10() throws Exception {

        ExecutorService pool = Executors.newFixedThreadPool(2);

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(4000);

            Future<Socket> result1 = pool.submit(ss::accept);

            Future<Socket> result2 = pool.submit(ss::accept);

            // both tasks should complete with SocketTimeoutException

            Throwable e = expectThrows(ExecutionException.class, result1::get);

            assertTrue(e.getCause() instanceof SocketTimeoutException);

            e = expectThrows(ExecutionException.class, result2::get);

            assertTrue(e.getCause() instanceof SocketTimeoutException);

            // should get here in 4 seconds, not 8 seconds

            int timeout = ss.getSoTimeout();

        } finally {

            pool.shutdown();

        }

    }

    /**

     * Test two threads blocked in timed accept where one connection is established.

     */

    public void testTimedAccept11() throws Exception {

        ExecutorService pool = Executors.newFixedThreadPool(2);

        try (ServerSocket ss = new ServerSocket(0)) {

            ss.setSoTimeout(4000);

            Future<Socket> result1 = pool.submit(ss::accept);

            Future<Socket> result2 = pool.submit(ss::accept);

            // establish connection after 2 seconds

            scheduleConnect(ss.getLocalSocketAddress(), 2000);

            // one task should have accepted the connection, the other should

            // have completed with SocketTimeoutException

            Socket s1 = null;

            try {

                s1 = result1.get();

                s1.close();

            } catch (ExecutionException e) {

                assertTrue(e.getCause() instanceof SocketTimeoutException);

            }

            Socket s2 = null;

            try {

                s2 = result2.get();

                s2.close();

            } catch (ExecutionException e) {

                assertTrue(e.getCause() instanceof SocketTimeoutException);

            }

            assertTrue((s1 != null) ^ (s2 != null));