/*

 * 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 8216417

 * @summary cleanup of IPv6 scope-id handling

 * @library /test/lib

 * @build jdk.test.lib.NetworkConfiguration

 * @run main/othervm Scoping

 */

import java.io.IOException;

import java.net.*;

import java.nio.ByteBuffer;

import java.nio.channels.DatagramChannel;

import java.nio.channels.ServerSocketChannel;

import java.nio.channels.SocketChannel;

import java.util.Enumeration;

import java.util.LinkedList;

import java.util.List;

import java.util.stream.Collectors;

import jdk.test.lib.NetworkConfiguration;

public class Scoping {

    interface ThrowingConsumer<T> {

        public void accept(T t) throws Exception;

    }

    static List<ThrowingConsumer<InetSocketAddress>> targets = List.of(

        /* 0 */  (a) -> {Socket s = new Socket(); s.bind(a);s.close();},

        /* 1 */  (a) -> {ServerSocket s = new ServerSocket(); s.bind(a);s.close();},

        /* 2 */  (a) -> {DatagramSocket s = new DatagramSocket(null); s.bind(a); s.close();},

        /* 3 */  (a) -> {MulticastSocket s = new MulticastSocket(null); s.bind(a); s.close();},

        /* 4 */  (a) -> {SocketChannel s = SocketChannel.open(); s.bind(a);s.close();},

        /* 5 */  (a) -> {ServerSocketChannel s = ServerSocketChannel.open(); s.bind(a);s.close();},

        /* 6 */  (a) -> {DatagramChannel s = DatagramChannel.open(); s.bind(a); s.close();},

        /* 7 */  (a) -> {SocketChannel s = SocketChannel.open(); s.socket().bind(a);s.close();},

        /* 8 */  (a) -> {ServerSocketChannel s = ServerSocketChannel.open(); s.socket().bind(a);s.close();},

        /* 9 */  (a) -> {DatagramChannel s = DatagramChannel.open(); s.socket().bind(a); s.close();},

        /* 10 */ (a) -> {socketTest(a);},

        /* 11 */ (a) -> {dgSocketTest(a, false);},

        /* 12 */ (a) -> {dgSocketTest(a, true);},

        /* 13 */ (a) -> {dgChannelTest(a, false);},

        /* 14 */ (a) -> {dgChannelTest(a, true);}

    );

    static List<Inet6Address> getLinkLocalAddrs() throws IOException {

        return NetworkConfiguration.probe()

                   .ip6Addresses()

                   .filter(Inet6Address::isLinkLocalAddress)

                   .collect(Collectors.toList());

    }

    static void compare(InetSocketAddress a, InetSocketAddress b) {

        Inet6Address a1 = (Inet6Address)a.getAddress();

        Inet6Address b1 = (Inet6Address)b.getAddress();

        compare (a1, b1);

    }

    static void compare(InetAddress a, InetAddress b) {

        Inet6Address a1 = (Inet6Address)a;

        Inet6Address b1 = (Inet6Address)b;

        if (!a1.equals(b1)) {

            System.out.printf("%s != %s\n", a1, b1);

            throw new RuntimeException("Addresses not the same");

        }

        if (!a1.getHostAddress().equals(b1.getHostAddress())) {

            System.out.printf("%s != %s\n", a1, b1);

            if (a1.getScopeId() != b1.getScopeId())

                throw new RuntimeException("Scope ids not the same");

        }

    }

    static void socketTest(InetSocketAddress a) throws Exception {

        System.out.printf("socketTest: address %s\n", a);

        try (ServerSocket server = new ServerSocket(0, 5, a.getAddress())) {

            InetAddress saddr = server.getInetAddress();

            int port = server.getLocalPort();

            Socket client = new Socket(saddr, port);

            compare(client.getInetAddress(), saddr);

            try {

                client.close();

            } catch (IOException e) {}

        }

    }

    static void dgSocketTest(InetSocketAddress a, boolean connect) throws Exception {

        try (DatagramSocket s = new DatagramSocket(null)) {

            System.out.println("dgSocketTest: " + a);

            s.bind(a);

            String t = "Hello world";

            DatagramPacket rx = new DatagramPacket(new byte[128], 128);

            int port = s.getLocalPort();

            InetSocketAddress dest = new InetSocketAddress(a.getAddress(), port);

            DatagramPacket tx = new DatagramPacket(t.getBytes("ISO8859_1"), t.length(), dest);

            if (connect) {

                s.connect(dest);

                System.out.println("dgSocketTest: connect remote addr = " + s.getRemoteSocketAddress());

                compare(a, (InetSocketAddress)s.getRemoteSocketAddress());

            }

            s.send(tx);

            s.receive(rx);

            String t1 = new String(rx.getData(), rx.getOffset(), rx.getLength(), "ISO8859_1");

            if (!t1.equals(t))

                throw new RuntimeException("Packets not equal");

        }

    }

    static void dgChannelTest(InetSocketAddress a, boolean connect) throws Exception {

        try (DatagramChannel s = DatagramChannel.open()) {

            System.out.println("dgChannelTest: " + a);

            s.bind(a);

            String t = "Hello world";

            ByteBuffer rx = ByteBuffer.allocate(128);

            int port = ((InetSocketAddress)s.getLocalAddress()).getPort();

            InetSocketAddress dest = new InetSocketAddress(a.getAddress(), port);

            ByteBuffer tx = ByteBuffer.wrap(t.getBytes("ISO8859_1"));

            if (connect) {

                s.connect(dest);

                System.out.println("dgChannelTest: connect remote addr = " + s.getRemoteAddress());

                compare(a, (InetSocketAddress)s.getRemoteAddress());

            }

            s.send(tx, dest);

            s.receive(rx);

            rx.flip();

            String t1 = new String(rx.array(), 0, rx.limit(), "ISO8859_1");

            System.out.printf("rx : %s, data: %s\n", rx, t1);

            if (!t1.equals(t))

                throw new RuntimeException("Packets not equal");

        }

    }

    static Inet6Address stripScope(Inet6Address address) {

        byte[] bytes = address.getAddress();

        InetAddress result = null;

        try {

            result = InetAddress.getByAddress(bytes);

        } catch (UnknownHostException e) {

            assert false;

        }

        return (Inet6Address)result;

    }

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

        for (Inet6Address address : getLinkLocalAddrs()) {

            Inet6Address stripped = stripScope(address);

            InetSocketAddress s1 = new InetSocketAddress(address, 0);

            InetSocketAddress s2 = new InetSocketAddress(stripped, 0);

            System.out.println("Trying: " + address);

            int count = 0, success = 0;

            for (ThrowingConsumer<InetSocketAddress> target : targets) {

                try {

                    target.accept(s1);

                    System.out.println("target " + count + " OK");

                    // if that succeeds try s2 (the actual test)

                    try {

                        target.accept(s2);

                        success++;

                    } catch (IOException ee) {

                        String msg = "Failed: " + s2.toString() + "count: " + Integer.toString(count);

                        throw new RuntimeException (msg);

                    }

                } catch (IOException e) {

                    System.out.println(e.getMessage());

                    // OK

                }

                count++;

            }

            System.out.println("Succeeded with " + success + " binds");

        }

    }

}