/*

 * Copyright (c) 2001, 2018, 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 4476378

 * @summary Check the specific behaviour of the setReuseAddress(boolean)

 * @run main Basic

 * @run main/othervm -Dsun.net.useExclusiveBind Basic

 * @run main/othervm -Djava.net.preferIPv4Stack=true Basic

 */

import java.net.*;

public class Basic {

    static int testCount = 0;

    static int failures = 0;

    void test(String msg) {

        testCount++;

        System.out.println("***************************************");

        System.out.println("Test " + testCount + ": " + msg);

    }

    void passed() {

        System.out.println("Test passed.");

    }

    void failed() {

        failures++;

        System.out.println("Test failed.");

    }

    void check(boolean pass) {

        if (pass) {

            passed();

        } else {

            failed();

        }

    }

    void SocketTests() throws Exception {

        Socket s1 = new Socket();

        test("Socket should be created with SO_REUSEADDR disabled");

        check(!s1.getReuseAddress());

        test("Socket.setReuseAddress(true)");

        s1.setReuseAddress(true);

        check(s1.getReuseAddress());

        test("Socket.setReuseAddress(false)");

        s1.setReuseAddress(false);

        check(!s1.getReuseAddress() );

        /* bind to any port */

        s1.bind( new InetSocketAddress(0) );

        test("Binding Socket to port already in use should throw " +

             "a BindException");

        Socket s2 = new Socket();

        try {

            s2.bind( new InetSocketAddress(s1.getLocalPort()) );

            failed();

        } catch (BindException e) {

            passed();

        }

        s2.close();

        s1.close();

    }

    void ServerSocketTests() throws Exception {

        ServerSocket s1 = new ServerSocket();

        test("ServerSocket.setReuseAddress(true)");

        s1.setReuseAddress(true);

        check(s1.getReuseAddress());

        test("Socket.setReuseAddress(false)");

        s1.setReuseAddress(false);

        check(!s1.getReuseAddress() );

        /* bind to any port */

        s1.bind( new InetSocketAddress(0) );

        test("Binding ServerSocket to port already in use should throw " +

                "a BindException");

        ServerSocket s2 = new ServerSocket();

        try {

            s2.bind( new InetSocketAddress(s1.getLocalPort()) );

            failed();

        } catch (BindException e) {

            passed();

        }

        s2.close();

        s1.close();

    }

    void DatagramSocketTests() throws Exception {

        DatagramSocket s1 = new DatagramSocket(null);

        test("DatagramSocket should be created with SO_REUSEADDR disabled");

        check(!s1.getReuseAddress());

        test("DatagramSocket.setReuseAddress(true)");

        s1.setReuseAddress(true);

        check(s1.getReuseAddress());

        test("DatagramSocket.setReuseAddress(false)");

        s1.setReuseAddress(false);

        check(!s1.getReuseAddress() );

        /* bind to any port */

        s1.bind( new InetSocketAddress(0) );

        test("Binding datagram socket to port already in use should throw " +

             "a BindException");

        DatagramSocket s2 = new DatagramSocket(null);

        try {

            s2.bind( new InetSocketAddress(s1.getLocalPort()) );

            failed();

        } catch (BindException e) {

            passed();

        }

        s2.close();

        s1.close();

        // bind with SO_REUSEADDR enabled

        s1 = new DatagramSocket(null);

        s1.setReuseAddress(true);

        s1.bind( new InetSocketAddress(0) );

        test("Bind 2 datagram sockets to the same port - second " +

             "bind doesn't have SO_REUSEADDR enabled");

        s2 = new DatagramSocket(null);

        try {

            s2.bind( new InetSocketAddress(s1.getLocalPort()) );

            failed();

        } catch (BindException e) {

            passed();

        }

        s2.close();

        test("Bind 2 datagram sockets to the same port - both have " +

             "SO_REUSEADDR enabled");

        s2 = new DatagramSocket(null);

        s2.setReuseAddress(true);

        try {

            s2.bind( new InetSocketAddress(s1.getLocalPort()) );

            passed();

        } catch (BindException e) {

            if (System.getProperty("sun.net.useExclusiveBind") != null) {

                // exclusive bind enabled - expected result

                passed();

            } else {

                failed();

            }

        }

        s2.close();

        s1.close();

    }

    void MulticastSocketTests() throws Exception {

        test("Check SO_REUSEADDR is enabled in MulticastSocket()");

        MulticastSocket s1 = new MulticastSocket();

        check(s1.getReuseAddress());

        s1.close();

        test("Check that SO_REUSEADDR is not disabled by " +

             "MulticastSocket.bind()");

        s1 = new MulticastSocket(null);

        // bind to specific address

        InetSocketAddress isa = new InetSocketAddress(

                                   InetAddress.getLocalHost(), 0);

        s1.bind(isa);

        check(s1.getReuseAddress());

        s1.close();

    }

    Basic() throws Exception {

        SocketTests();

        ServerSocketTests();

        DatagramSocketTests();

        MulticastSocketTests();

        System.out.println("***************************************");

        System.out.println(testCount + " test(s) executed, " +

                           failures + " failure(s).");

        if (failures > 0) {

            throw new Exception(failures + " test(s) failed");

        }

    }

    public static void main(String args[]) throws Exception {

        new Basic();

    }

}