jdk/test/java/net/MulticastSocket/SetOutgoingIf.java
changeset 2 90ce3da70b43
child 3945 391194a1b78a
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/test/java/net/MulticastSocket/SetOutgoingIf.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,181 @@
+/*
+ * Copyright 2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+/*
+ * @test
+ * @bug 4742177
+ * @summary Re-test IPv6 (and specifically MulticastSocket) with latest Linux & USAGI code
+ */
+import java.net.*;
+import java.util.concurrent.*;
+import java.util.*;
+
+
+public class SetOutgoingIf {
+    private static int PORT = 9001;
+    private static String osname;
+
+    static boolean isWindows() {
+        if (osname == null)
+            osname = System.getProperty("os.name");
+        return osname.contains("Windows");
+    }
+
+    private static boolean hasIPv6() throws Exception {
+        List<NetworkInterface> nics = Collections.list(
+                                        NetworkInterface.getNetworkInterfaces());
+        for (NetworkInterface nic : nics) {
+            List<InetAddress> addrs = Collections.list(nic.getInetAddresses());
+            for (InetAddress addr : addrs) {
+                if (addr instanceof Inet6Address)
+                    return true;
+            }
+        }
+
+        return false;
+    }
+
+    public static void main(String[] args) throws Exception {
+        if (isWindows()) {
+            System.out.println("The test only run on non-Windows OS. Bye.");
+            return;
+        }
+
+        if (!hasIPv6()) {
+            System.out.println("No IPv6 available. Bye.");
+            return;
+        }
+
+        // We need 2 or more network interfaces to run the test
+        //
+        List<NetworkInterface> nics = new ArrayList<NetworkInterface>();
+        for (NetworkInterface nic : Collections.list(NetworkInterface.getNetworkInterfaces())) {
+            if (!nic.isLoopback())
+                nics.add(nic);
+        }
+        if (nics.size() <= 1) {
+            System.out.println("Need 2 or more network interfaces to run. Bye.");
+            return;
+        }
+
+        // We will send packets to one ipv4, one ipv4-mapped, and one ipv6
+        // multicast group using each network interface :-
+        //      224.1.1.1        --|
+        //      ::ffff:224.1.1.2 -----> using network interface #1
+        //      ff02::1:1        --|
+        //      224.1.2.1        --|
+        //      ::ffff:224.1.2.2 -----> using network interface #2
+        //      ff02::1:2        --|
+        // and so on.
+        //
+        List<InetAddress> groups = new ArrayList<InetAddress>();
+        for (int i = 0; i < nics.size(); i++) {
+            InetAddress groupv4 = InetAddress.getByName("224.1." + (i+1) + ".1");
+            InetAddress groupv4mapped = InetAddress.getByName("::ffff:224.1." + (i+1) + ".2");
+            InetAddress groupv6 = InetAddress.getByName("ff02::1:" + (i+1));
+            groups.add(groupv4);
+            groups.add(groupv4mapped);
+            groups.add(groupv6);
+
+            // use a separated thread to send to those 3 groups
+            Thread sender = new Thread(new Sender(nics.get(i), groupv4, groupv4mapped, groupv6, PORT));
+            sender.setDaemon(true); // we want sender to stop when main thread exits
+            sender.start();
+        }
+
+        // try to receive on each group, then check if the packet comes
+        // from the expected network interface
+        //
+        byte[] buf = new byte[1024];
+        for (InetAddress group : groups) {
+        MulticastSocket mcastsock = new MulticastSocket(PORT);
+        mcastsock.setSoTimeout(5000);   // 5 second
+            DatagramPacket packet = new DatagramPacket(buf, 0, buf.length);
+
+            mcastsock.joinGroup(new InetSocketAddress(group, PORT), nics.get(groups.indexOf(group) / 3));
+
+            try {
+                mcastsock.receive(packet);
+            } catch (Exception e) {
+                // test failed if any exception
+                throw new RuntimeException(e);
+            }
+
+            // now check which network interface this packet comes from
+            NetworkInterface from = NetworkInterface.getByInetAddress(packet.getAddress());
+            NetworkInterface shouldbe = nics.get(groups.indexOf(group) / 3);
+            if (!from.equals(shouldbe)) {
+                System.out.println("Packets on group "
+                                    + group + " should come from "
+                                    + shouldbe.getName() + ", but came from "
+                                    + from.getName());
+                //throw new RuntimeException("Test failed.");
+            }
+
+            mcastsock.leaveGroup(new InetSocketAddress(group, PORT), nics.get(groups.indexOf(group) / 3));
+        }
+    }
+}
+
+class Sender implements Runnable {
+    private NetworkInterface nic;
+    private InetAddress group1;
+    private InetAddress group2;
+    private InetAddress group3;
+    private int port;
+
+    public Sender(NetworkInterface nic,
+                    InetAddress groupv4, InetAddress groupv4mapped, InetAddress groupv6,
+                    int port) {
+        this.nic = nic;
+        group1 = groupv4;
+        group2 = groupv4mapped;
+        group3 = groupv6;
+        this.port = port;
+    }
+
+    public void run() {
+        try {
+            MulticastSocket mcastsock = new MulticastSocket();
+            mcastsock.setNetworkInterface(nic);
+
+            byte[] buf = "hello world".getBytes();
+            DatagramPacket packet1 = new DatagramPacket(buf, buf.length,
+                                        new InetSocketAddress(group1, port));
+            DatagramPacket packet2 = new DatagramPacket(buf, buf.length,
+                                        new InetSocketAddress(group2, port));
+            DatagramPacket packet3 = new DatagramPacket(buf, buf.length,
+                                        new InetSocketAddress(group3, port));
+
+            for (;;) {
+                mcastsock.send(packet1);
+                mcastsock.send(packet2);
+                mcastsock.send(packet3);
+
+                Thread.currentThread().sleep(1000);   // sleep 1 second
+            }
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+}