1 /* 2 * Copyright (c) 2003, 2018, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* @test 25 * @bug 4531526 26 * @summary Test that more than one debuggee cannot bind to same port 27 * at the same time. 28 * @library /test/lib 29 * 30 * @modules java.management 31 * jdk.jdi 32 * @build VMConnection ExclusiveBind HelloWorld 33 * @run driver ExclusiveBind 34 */ 35 import java.net.ServerSocket; 36 import com.sun.jdi.Bootstrap; 37 import com.sun.jdi.VirtualMachine; 38 import com.sun.jdi.connect.Connector; 39 import com.sun.jdi.connect.AttachingConnector; 40 41 import java.util.ArrayList; 42 import java.util.Map; 43 import java.util.List; 44 import java.util.Iterator; 45 import java.util.concurrent.TimeUnit; 46 47 import jdk.test.lib.process.ProcessTools; 48 import jdk.test.lib.Utils; 49 50 public class ExclusiveBind { 51 /* 52 * Find a connector by name 53 */ 54 private static Connector findConnector(String name) { 55 List connectors = Bootstrap.virtualMachineManager().allConnectors(); 56 Iterator iter = connectors.iterator(); 57 while (iter.hasNext()) { 58 Connector connector = (Connector)iter.next(); 59 if (connector.name().equals(name)) { 60 return connector; 61 } 62 } 63 return null; 64 } 65 66 /* 67 * Launch (in server mode) a debuggee with the given address and 68 * suspend mode. 69 */ 70 private static ProcessBuilder prepareLauncher(String address, boolean suspend, String class_name) throws Exception { 71 List<String> args = new ArrayList<>(); 72 for(String dbgOption : VMConnection.getDebuggeeVMOptions().split(" ")) { 73 if (!dbgOption.trim().isEmpty()) { 74 args.add(dbgOption); 75 } 76 } 77 String lib = "-agentlib:jdwp=transport=dt_socket,server=y,suspend="; 78 if (suspend) { 79 lib += "y"; 80 } else { 81 lib += "n"; 82 } 83 lib += ",address=" + address; 84 85 args.add(lib); 86 args.add(class_name); 87 88 return ProcessTools.createJavaProcessBuilder(args.toArray(new String[args.size()])); 89 } 90 91 /* 92 * - pick a TCP port 93 * - Launch a debuggee in server=y,suspend=y,address=${port} 94 * - Launch a second debuggee in server=y,suspend=n with the same port 95 * - Second debuggee should fail with an error (address already in use) 96 * - For clean-up we attach to the first debuggee and resume it. 97 */ 98 public static void main(String args[]) throws Exception { 99 // find a free port 100 ServerSocket ss = new ServerSocket(0); 101 int port = ss.getLocalPort(); 102 ss.close(); 103 104 String address = String.valueOf(port); 105 106 // launch the first debuggee 107 ProcessBuilder process1 = prepareLauncher(address, true, "HelloWorld"); 108 // start the debuggee and wait for the "ready" message 109 Process p = ProcessTools.startProcess( 110 "process1", 111 process1, 112 line -> line.equals("Listening for transport dt_socket at address: " + address), 113 Utils.adjustTimeout(5000), 114 TimeUnit.MILLISECONDS 115 ); 116 117 // launch a second debuggee with the same address 118 ProcessBuilder process2 = prepareLauncher(address, false, "HelloWorld"); 119 120 // get exit status from second debuggee 121 int exitCode = ProcessTools.startProcess("process2", process2).waitFor(); 122 123 // clean-up - attach to first debuggee and resume it 124 AttachingConnector conn = (AttachingConnector)findConnector("com.sun.jdi.SocketAttach"); 125 Map conn_args = conn.defaultArguments(); 126 Connector.IntegerArgument port_arg = 127 (Connector.IntegerArgument)conn_args.get("port"); 128 port_arg.setValue(port); 129 VirtualMachine vm = conn.attach(conn_args); 130 vm.resume(); 131 132 // if the second debuggee ran to completion then we've got a problem 133 if (exitCode == 0) { 134 throw new RuntimeException("Test failed - second debuggee didn't fail to bind"); 135 } else { 136 System.out.println("Test passed - second debuggee correctly failed to bind"); 137 } 138 } 139 }