test/jdk/com/sun/jdi/JdwpListenTest.java
changeset 54884 8a6093c186a6
child 58876 1a8d65e71a66
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/com/sun/jdi/JdwpListenTest.java	Wed May 15 11:06:33 2019 -0700
@@ -0,0 +1,177 @@
+/*
+ * 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.
+ */
+
+import com.sun.jdi.Bootstrap;
+import com.sun.jdi.VirtualMachine;
+import com.sun.jdi.connect.AttachingConnector;
+import com.sun.jdi.connect.Connector;
+import com.sun.jdi.connect.IllegalConnectorArgumentsException;
+import lib.jdb.Debuggee;
+
+import java.io.IOException;
+import java.net.Inet4Address;
+import java.net.Inet6Address;
+import java.net.InetAddress;
+import java.net.NetworkInterface;
+import java.net.SocketException;
+import java.util.Enumeration;
+import java.util.Iterator;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+
+/*
+ * @test
+ * @bug 8184770
+ * @summary Tests for JDWP agent listen functionality (including IPv6 support)
+ * @library /test/lib
+ *
+ * @build HelloWorld JdwpListenTest
+ * @run main/othervm JdwpListenTest
+ */
+public class JdwpListenTest {
+
+    public static void main(String[] args) throws Exception {
+        List<InetAddress> addresses = getAddresses();
+
+        boolean ipv4EnclosedTested = false;
+        boolean ipv6EnclosedTested = false;
+        for (InetAddress listen: addresses) {
+            for (InetAddress attach: addresses) {
+                // can connect only from the same address
+                // IPv6 cannot connect to IPv4 (::1 to 127.0.0.1) and vice versa.
+                listenTest(listen.getHostAddress(), attach.getHostAddress(), attach.equals(listen));
+            }
+            // test that addresses enclosed in square brackets are supported.
+            if (listen instanceof Inet4Address && !ipv4EnclosedTested) {
+                listenTest("[" + listen.getHostAddress() + "]", "[" + listen.getHostAddress() + "]", true);
+                ipv4EnclosedTested = true;
+            }
+            if (listen instanceof Inet6Address && !ipv6EnclosedTested) {
+                listenTest("[" + listen.getHostAddress() + "]", "[" + listen.getHostAddress() + "]", true);
+                ipv6EnclosedTested = true;
+            }
+        }
+        // listen on "*" - should be accessible from any address
+        for (InetAddress attach: addresses) {
+            listenTest("*", attach.getHostAddress(), true);
+        }
+    }
+
+    private static void listenTest(String listenAddress, String connectAddress, boolean expectedResult)
+            throws IOException {
+        log("Starting listening debuggee at " + listenAddress);
+        try (Debuggee debuggee = Debuggee.launcher("HelloWorld").setAddress(listenAddress + ":0").launch()) {
+            log("Debuggee is listening on " + listenAddress + ":" + debuggee.getAddress());
+            log("Connecting from " + connectAddress + ", expected: " + (expectedResult ? "SUCCESS" : "FAILURE"));
+            try {
+                VirtualMachine vm = attach(connectAddress, debuggee.getAddress());
+                vm.dispose();
+                if (!expectedResult) {
+                    throw new RuntimeException("ERROR: attached successfully");
+                }
+            } catch (IOException ex) {
+                if (expectedResult) {
+                    throw new RuntimeException("ERROR: failed to attach", ex);
+                }
+            }
+        }
+    }
+
+    private static List<InetAddress> getAddresses() {
+        List<InetAddress> result = new LinkedList<>();
+        try {
+            Enumeration<NetworkInterface> networkInterfaces = NetworkInterface.getNetworkInterfaces();
+            while (networkInterfaces.hasMoreElements()) {
+                NetworkInterface iface = networkInterfaces.nextElement();
+                try {
+                    if (iface.isUp()) {
+                        Enumeration<InetAddress> addresses = iface.getInetAddresses();
+                        while (addresses.hasMoreElements()) {
+                            InetAddress addr = addresses.nextElement();
+                            // Java reports link local addresses with named scope,
+                            // but Windows sockets routines support only numeric scope id.
+                            // skip such addresses.
+                            if (addr instanceof Inet6Address) {
+                                Inet6Address addr6 = (Inet6Address)addr;
+                                if (addr6.getScopedInterface() != null) {
+                                    continue;
+                                }
+                            }
+                            log(" - (" + addr.getClass().getSimpleName() + ") " + addr.getHostAddress());
+                            result.add(addr);
+                        }
+                    }
+                } catch (SocketException e) {
+                    log("Interface " + iface.getDisplayName() + ": failed to get addresses");
+                }
+            }
+        } catch (SocketException e) {
+            log("Interface enumeration error: " + e);
+        }
+        return result;
+    }
+
+    private static String ATTACH_CONNECTOR = "com.sun.jdi.SocketAttach";
+    // cache socket attaching connector
+    private static AttachingConnector attachingConnector;
+
+    private static VirtualMachine attach(String address, String port) throws IOException {
+        if (attachingConnector == null) {
+            attachingConnector = (AttachingConnector)getConnector(ATTACH_CONNECTOR);
+        }
+        Map<String, Connector.Argument> args = attachingConnector.defaultArguments();
+        setConnectorArg(args, "hostname", address);
+        setConnectorArg(args, "port", port);
+        try {
+            return attachingConnector.attach(args);
+        } catch (IllegalConnectorArgumentsException e) {
+            // unexpected.. wrap in RuntimeException
+            throw new RuntimeException(e);
+        }
+    }
+
+    private static Connector getConnector(String name) {
+        List<Connector> connectors = Bootstrap.virtualMachineManager().allConnectors();
+        for (Iterator<Connector> iter = connectors.iterator(); iter.hasNext(); ) {
+            Connector connector = iter.next();
+            if (connector.name().equalsIgnoreCase(name)) {
+                return connector;
+            }
+        }
+        throw new IllegalArgumentException("Connector " + name + " not found");
+    }
+
+    private static void setConnectorArg(Map<String, Connector.Argument> args, String name, String value) {
+        Connector.Argument arg = args.get(name);
+        if (arg == null) {
+            throw new IllegalArgumentException("Argument " + name + " is not defined");
+        }
+        arg.setValue(value);
+    }
+
+    private static void log(Object o) {
+        System.out.println(String.valueOf(o));
+    }
+
+}