/*

 * 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

 * @bug 8220493

 * @modules java.base/java.net:+open java.base/sun.nio.ch:+open

 * @run testng/othervm SocketImplCombinations

 * @summary Test Socket and ServerSocket with combinations of SocketImpls

 */

import java.io.FileDescriptor;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.lang.reflect.Field;

import java.net.InetAddress;

import java.net.InetSocketAddress;

import java.net.Proxy;

import java.net.ServerSocket;

import java.net.Socket;

import java.net.SocketAddress;

import java.net.SocketImpl;

import java.net.SocketImplFactory;

import java.nio.channels.ServerSocketChannel;

import java.nio.channels.SocketChannel;

import java.util.function.BiConsumer;

import org.testng.annotations.Test;

import static org.testng.Assert.*;

@Test

public class SocketImplCombinations {

    /**

     * Test creating an unconnected Socket, it should be created with a platform SocketImpl.

     */

    public void testNewSocket1() throws IOException {

        try (Socket s = new Socket()) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(isSocksSocketImpl(si));

            SocketImpl delegate = getDelegate(si);

            assertTrue(isPlatformSocketImpl(delegate));

        }

    }

    /**

     * Test creating a connected Socket, it should be created with a platform SocketImpl.

     */

    public void testNewSocket2() throws IOException {

        try (ServerSocket ss = boundServerSocket()) {

            try (Socket s = new Socket(ss.getInetAddress(), ss.getLocalPort())) {

                SocketImpl si = getSocketImpl(s);

                assertTrue(isSocksSocketImpl(si));

                SocketImpl delegate = getDelegate(si);

                assertTrue(isPlatformSocketImpl(delegate));

            }

        }

    }

    /**

     * Test creating a Socket for a DIRECT connection, it should be created with a

     * platform SocketImpl.

     */

    public void testNewSocket3() throws IOException {

        try (Socket s = new Socket(Proxy.NO_PROXY)) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(isPlatformSocketImpl(si));

        }

    }

    /**

     * Test creating a Socket for a SOCKS connection, it should be created with a

     * SOCKS SocketImpl.

     */

    public void testNewSocket4() throws IOException {

        var address = new InetSocketAddress("127.0.0.1", 1080);

        var socksProxy = new Proxy(Proxy.Type.SOCKS, address);

        try (Socket s = new Socket(socksProxy)) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(isSocksSocketImpl(si));

            SocketImpl delegate = getDelegate(si);

            assertTrue(isPlatformSocketImpl(delegate));

        }

    }

    /**

     * Test creating a Socket for a HTTP proxy connection, it should be created with

     * a HTTP proxy SocketImpl.

     */

    public void testNewSocket5() throws IOException {

        var address = new InetSocketAddress("127.0.0.1", 8080);

        var httpProxy = new Proxy(Proxy.Type.HTTP, address);

        try (Socket s = new Socket(httpProxy)) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(isHttpConnectSocketImpl(si));

            SocketImpl delegate = getDelegate(si);

            assertTrue(isPlatformSocketImpl(delegate));

        }

    }

    /**

     * Test creating a Socket no SocketImpl. A platform SocketImpl should be

     * created lazily.

     */

    public void testNewSocket6() throws IOException {

        Socket s = new Socket((SocketImpl) null) { };

        try (s) {

            assertTrue(getSocketImpl(s) == null);

            s.bind(loopbackSocketAddress());   // force SocketImpl to be created

            SocketImpl si = getSocketImpl(s);

            assertTrue(isSocksSocketImpl(si));

            SocketImpl delegate = getDelegate(si);

            assertTrue(isPlatformSocketImpl(delegate));

        }

    }

    /**

     * Test creating a Socket with a custom SocketImpl.

     */

    public void testNewSocket7() throws IOException {

        Socket s = new Socket(new CustomSocketImpl(false)) { };

        try (s) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(si instanceof CustomSocketImpl);

        }

    }

    /**

     * Test creating a Socket when there is a SocketImplFactory set.

     */

    public void testNewSocket8() throws IOException {

        setSocketSocketImplFactory(() -> new CustomSocketImpl(false));

        try (Socket s = new Socket()) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(si instanceof CustomSocketImpl);

        } finally {

            setSocketSocketImplFactory(null);

        }

    }

    /**

     * Test creating a Socket for a DIRECT connection when there is a

     * SocketImplFactory set.

     */

    public void testNewSocket9() throws IOException {

        setSocketSocketImplFactory(() -> new CustomSocketImpl(false));

        try (Socket s = new Socket(Proxy.NO_PROXY)) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(si instanceof CustomSocketImpl);

        } finally {

            setSocketSocketImplFactory(null);

        }

    }

    /**

     * Test creating a Socket for a SOCKS connection when there is a

     * SocketImplFactory set.

     */

    public void testNewSocket10() throws IOException {

        var address = new InetSocketAddress("127.0.0.1", 1080);

        var socksProxy = new Proxy(Proxy.Type.SOCKS, address);

        setSocketSocketImplFactory(() -> new CustomSocketImpl(false));

        try (Socket s = new Socket(socksProxy)) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(isSocksSocketImpl(si));

            SocketImpl delegate = getDelegate(si);

            assertTrue(isPlatformSocketImpl(delegate));

        } finally {

            setSocketSocketImplFactory(null);

        }

    }

    /**

     * Test creating a Socket for a HTTP proxy connection when there is a

     * SocketImplFactory set.

     */

    public void testNewSocket11() throws IOException {

        var address = new InetSocketAddress("127.0.0.1", 8080);

        var httpProxy = new Proxy(Proxy.Type.HTTP, address);

        setSocketSocketImplFactory(() -> new CustomSocketImpl(false));

        try (Socket s = new Socket(httpProxy)) {

            SocketImpl si = getSocketImpl(s);

            assertTrue(isHttpConnectSocketImpl(si));

            SocketImpl delegate = getDelegate(si);

            assertTrue(isPlatformSocketImpl(delegate));

        } finally {

            setSocketSocketImplFactory(null);

        }

    }

    /**

     * Test creating a Socket no SocketImpl when there is a SocketImplFactory set.

     */

    public void testNewSocket12() throws IOException {

        setSocketSocketImplFactory(() -> new CustomSocketImpl(false));

        try {

            Socket s = new Socket((SocketImpl) null) { };

            try (s) {

                assertTrue(getSocketImpl(s) == null);

                s.bind(loopbackSocketAddress());   // force SocketImpl to be created

                assertTrue(getSocketImpl(s) instanceof CustomSocketImpl);

            }

        } finally {

            setSocketSocketImplFactory(null);

        }

    }

    /**

     * Test creating an unbound ServerSocket, it should be created with a platform

     * SocketImpl.

     */

    public void testNewServerSocket1() throws IOException {

        try (ServerSocket ss = new ServerSocket()) {

            SocketImpl si = getSocketImpl(ss);

            assertTrue(isPlatformSocketImpl(si));

        }

    }

    /**

     * Test creating a bound ServerSocket, it should be created with a platform

     * SocketImpl.

     */

    public void testNewServerSocket2() throws IOException {

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

            SocketImpl si = getSocketImpl(ss);

            assertTrue(isPlatformSocketImpl(si));

        }

    }

    /**

     * Test creating a ServerSocket with a custom SocketImpl.

     */

    public void testNewServerSocket3() throws IOException {

        ServerSocket ss = new ServerSocket(new CustomSocketImpl(true)) { };

        try (ss) {

            SocketImpl si = getSocketImpl(ss);

            assertTrue(si instanceof CustomSocketImpl);

        }

    }

    /**

     * Test creating an unbound ServerSocket when there is a SocketImplFactory set.

     */

    public void testNewServerSocket4() throws IOException {

        setServerSocketImplFactory(() -> new CustomSocketImpl(true));

        try (ServerSocket ss = new ServerSocket()) {

            SocketImpl si = getSocketImpl(ss);

            assertTrue(si instanceof CustomSocketImpl);

        } finally {

            setServerSocketImplFactory(null);

        }

    }

    /**

     * Test creating a bound ServerSocket when there is a SocketImplFactory set.

     */

    public void testNewServerSocket5() throws IOException {

        setServerSocketImplFactory(() -> new CustomSocketImpl(true));

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

            SocketImpl si = getSocketImpl(ss);

            assertTrue(si instanceof CustomSocketImpl);

        } finally {

            setServerSocketImplFactory(null);

        }

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a platform SocketImpl,

     * the Socket to accept is created with no SocketImpl.

     */

    public void testServerSocketAccept1() throws IOException {

        var socket = new Socket((SocketImpl) null) { };

        assertTrue(getSocketImpl(socket) == null);

        serverSocketAccept(socket, (ss, s) -> {

            assertTrue(isPlatformSocketImpl(getSocketImpl(ss)));

            assertTrue(s == socket);

            SocketImpl si = getSocketImpl(s);

            assertTrue(isPlatformSocketImpl(si));

            checkFields(si);

        });

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a platform SocketImpl,

     * the Socket to accept is created with no SocketImpl, and there is a custom

     * client SocketImplFactory set.

     */

    public void testServerSocketAccept2() throws IOException {

        var socket = new Socket((SocketImpl) null) { };

        assertTrue(getSocketImpl(socket) == null);

        serverSocketAccept(socket, () -> new CustomSocketImpl(false), (ss, s) -> {

            assertTrue(isPlatformSocketImpl(getSocketImpl(ss)));

            assertTrue(s == socket);

            SocketImpl si = getSocketImpl(s);

            assertTrue(isPlatformSocketImpl(si));

            checkFields(si);

        });

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a platform SocketImpl,

     * the Socket to accept is created with a SocketImpl that delegates to a

     * platform SocketImpl.

     */

    public void testServerSocketAccept3() throws IOException {

        var socket = new Socket();

        SocketImpl si = getSocketImpl(socket);

        assertTrue(isSocksSocketImpl(si));

        SocketImpl delegate = getDelegate(si);

        assertTrue(isPlatformSocketImpl(delegate));

        serverSocketAccept(socket, (ss, s) -> {

            assertTrue(isPlatformSocketImpl(getSocketImpl(ss)));

            assertTrue(s == socket);

            SocketImpl psi = getSocketImpl(socket);

            assertTrue(isPlatformSocketImpl(psi));

            checkFields(psi);

        });

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a platform SocketImpl,

     * the Socket to accept is created with a custom SocketImpl.

     */

    public void testServerSocketAccept4a() throws IOException {

        SocketImpl clientImpl = new CustomSocketImpl(false);

        Socket socket = new Socket(clientImpl) { };

        assertTrue(getSocketImpl(socket) == clientImpl);

        try (ServerSocket ss = serverSocketToAccept(socket)) {

            expectThrows(IOException.class, ss::accept);

        } finally {

            socket.close();

        }

    }

    public void testServerSocketAccept4b() throws IOException {

        SocketImpl clientImpl = new CustomSocketImpl(false);

        Socket socket = new Socket(clientImpl) { };

        assertTrue(getSocketImpl(socket) == clientImpl);

        setSocketSocketImplFactory(() -> new CustomSocketImpl(false));

        try (ServerSocket ss = serverSocketToAccept(socket)) {

            expectThrows(IOException.class, ss::accept);

        } finally {

            setSocketSocketImplFactory(null);

            socket.close();

        }

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a custom SocketImpl,

     * the Socket to accept is created no SocketImpl.

     */

    public void testServerSocketAccept5a() throws IOException {

        SocketImpl serverImpl = new CustomSocketImpl(true);

        try (ServerSocket ss = new ServerSocket(serverImpl) { }) {

            ss.bind(loopbackSocketAddress());

            expectThrows(IOException.class, ss::accept);

        }

    }

    public void testServerSocketAccept5b() throws IOException {

        var socket = new Socket((SocketImpl) null) { };

        assertTrue(getSocketImpl(socket) == null);

        SocketImpl serverImpl = new CustomSocketImpl(true);

        try (ServerSocket ss = serverSocketToAccept(serverImpl, socket)) {

            expectThrows(IOException.class, ss::accept);

        } finally {

            socket.close();

        }

    }

    public void testServerSocketAccept5c() throws IOException {

        setServerSocketImplFactory(() -> new CustomSocketImpl(true));

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

            expectThrows(IOException.class, ss::accept);

        } finally {

            setServerSocketImplFactory(null);

        }

    }

    public void testServerSocketAccept5d() throws IOException {

        var socket = new Socket((SocketImpl) null) { };

        assertTrue(getSocketImpl(socket) == null);

        setServerSocketImplFactory(() -> new CustomSocketImpl(true));

        try (ServerSocket ss = serverSocketToAccept(socket)) {

            expectThrows(IOException.class, ss::accept);

        } finally {

            setServerSocketImplFactory(null);

            socket.close();

        }

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a custom SocketImpl,

     * the Socket to accept is created with no SocketImpl, and there is a custom

     * client SocketImplFactory set.

     */

    public void testServerSocketAccept6() throws Exception {

        var socket = new Socket((SocketImpl) null) { };

        assertTrue(getSocketImpl(socket) == null);

        SocketImpl serverImpl = new CustomSocketImpl(true);

        SocketImplFactory clientFactory = () -> new CustomSocketImpl(false);

        serverSocketAccept(serverImpl, socket, clientFactory, (ss, s) -> {

            assertTrue(getSocketImpl(ss) == serverImpl);

            SocketImpl si = getSocketImpl(s);

            assertTrue(si instanceof CustomSocketImpl);

            checkFields(si);

        });

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a custom SocketImpl,

     * the Socket to accept is created with a SocketImpl that delegates to a

     * platform SocketImpl.

     */

    public void testServerSocketAccept7a() throws IOException {

        var socket = new Socket();

        SocketImpl si = getSocketImpl(socket);

        assertTrue(isSocksSocketImpl(si));

        SocketImpl delegate = getDelegate(si);

        assertTrue(isPlatformSocketImpl(delegate));

        SocketImpl serverImpl = new CustomSocketImpl(true);

        try (ServerSocket ss = serverSocketToAccept(serverImpl, socket)) {

            expectThrows(IOException.class, ss::accept);

        } finally {

            socket.close();

        }

    }

    public void testServerSocketAccept7b() throws IOException {

        var socket = new Socket();

        SocketImpl si = getSocketImpl(socket);

        assertTrue(isSocksSocketImpl(si));

        SocketImpl delegate = getDelegate(si);

        assertTrue(isPlatformSocketImpl(delegate));

        setServerSocketImplFactory(() -> new CustomSocketImpl(true));

        try (ServerSocket ss = serverSocketToAccept(socket)) {

            expectThrows(IOException.class, ss::accept);

        } finally {

            setServerSocketImplFactory(null);

            socket.close();

        }

    }

    /**

     * Test ServerSocket.accept. The ServerSocket uses a custom SocketImpl,

     * the Socket to accept is created with a custom SocketImpl.

     */

    public void testServerSocketAccept8() throws Exception {

        SocketImpl clientImpl = new CustomSocketImpl(false);

        Socket socket = new Socket(clientImpl) { };

        assertTrue(getSocketImpl(socket) == clientImpl);

        SocketImpl serverImpl = new CustomSocketImpl(true);

        SocketImplFactory clientFactory = () -> new CustomSocketImpl(false);

        serverSocketAccept(serverImpl, socket, clientFactory, (ss, s) -> {

            assertTrue(getSocketImpl(ss) == serverImpl);

            assertTrue(getSocketImpl(s) == clientImpl);

            checkFields(clientImpl);

        });

    }

    /**

     * Creates a ServerSocket that returns the given Socket from accept.

     * The consumer is invoked with the server socket and the accepted socket.

     */

    static void serverSocketAccept(Socket socket,

                                   BiConsumer<ServerSocket, Socket> consumer)

        throws IOException

    {

        Socket s1 = null;

        Socket s2 = null;

        try (ServerSocket ss = serverSocketToAccept(socket)) {

            s1 = new Socket(ss.getInetAddress(), ss.getLocalPort());

            s2 = ss.accept();

            consumer.accept(ss, s2);

        } finally {

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

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

        }

    }

    /**

     * Creates a ServerSocket that returns the given Socket from accept. The

     * given SocketImplFactory is set during the accept and the consumer is

     * invoked when the server socket and the accepted socket.

     */

    static void serverSocketAccept(Socket socket,

                                   SocketImplFactory factory,

                                   BiConsumer<ServerSocket, Socket> consumer)

        throws IOException

    {

        Socket s1 = null;

        Socket s2 = null;

        try (ServerSocket ss = serverSocketToAccept(socket)) {

            s1 = new Socket(ss.getInetAddress(), ss.getLocalPort());

            setSocketSocketImplFactory(factory);