1 /*
2 * Copyright (c) 2003, 2014, 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 /lib/testlibrary
29 *
30 * @build jdk.testlibrary.ProcessTools jdk.testlibrary.JDKToolLauncher jdk.testlibrary.Utils
31 * @build VMConnection ExclusiveBind HelloWorld
32 * @run main ExclusiveBind
33 */
34 import java.net.ServerSocket;
35 import com.sun.jdi.Bootstrap;
36 import com.sun.jdi.VirtualMachine;
37 import com.sun.jdi.connect.Connector;
38 import com.sun.jdi.connect.AttachingConnector;
39
40 import java.util.ArrayList;
41 import java.util.Map;
42 import java.util.List;
43 import java.util.Iterator;
44 import java.util.concurrent.TimeUnit;
45
46 import jdk.testlibrary.ProcessTools;
47 import jdk.testlibrary.Utils;
48
49 public class ExclusiveBind {
50 /*
51 * Find a connector by name
52 */
53 private static Connector findConnector(String name) {
54 List connectors = Bootstrap.virtualMachineManager().allConnectors();
55 Iterator iter = connectors.iterator();
56 while (iter.hasNext()) {
57 Connector connector = (Connector)iter.next();
58 if (connector.name().equals(name)) {
59 return connector;
60 }
61 }
62 return null;
63 }
64
65 /*
66 * Launch (in server mode) a debuggee with the given address and
67 * suspend mode.
68 */
69 private static ProcessBuilder prepareLauncher(String address, boolean suspend, String class_name) throws Exception {
70 List<String> args = new ArrayList<>();
71 for(String dbgOption : VMConnection.getDebuggeeVMOptions().split(" ")) {
72 args.add(dbgOption);
73 }
74 String lib = "-agentlib:jdwp=transport=dt_socket,server=y,suspend=";
75 if (suspend) {
76 lib += "y";
77 } else {
78 lib += "n";
79 }
80 lib += ",address=" + address;
81
82 args.add(lib);
83 args.add(class_name);
84
85 return ProcessTools.createJavaProcessBuilder(args.toArray(new String[args.size()]));
86 }
87
88 /*
89 * - pick a TCP port
90 * - Launch a debuggee in server=y,suspend=y,address=${port}
91 * - Launch a second debuggee in server=y,suspend=n with the same port
92 * - Second debuggee should fail with an error (address already in use)
93 * - For clean-up we attach to the first debuggee and resume it.
94 */
95 public static void main(String args[]) throws Exception {
96 // find a free port
97 ServerSocket ss = new ServerSocket(0);
98 int port = ss.getLocalPort();
99 ss.close();
100
101 String address = String.valueOf(port);
102
103 // launch the first debuggee
104 ProcessBuilder process1 = prepareLauncher(address, true, "HelloWorld");
105 // start the debuggee and wait for the "ready" message
106 Process p = ProcessTools.startProcess(
107 "process1",
108 process1,
109 line -> line.equals("Listening for transport dt_socket at address: " + address),
110 Math.round(5000 * Utils.TIMEOUT_FACTOR),
111 TimeUnit.MILLISECONDS
112 );
113
114 // launch a second debuggee with the same address
115 ProcessBuilder process2 = prepareLauncher(address, false, "HelloWorld");
116
117 // get exit status from second debuggee
118 int exitCode = ProcessTools.startProcess("process2", process2).waitFor();
119
120 // clean-up - attach to first debuggee and resume it
121 AttachingConnector conn = (AttachingConnector)findConnector("com.sun.jdi.SocketAttach");
122 Map conn_args = conn.defaultArguments();
123 Connector.IntegerArgument port_arg =
124 (Connector.IntegerArgument)conn_args.get("port");
125 port_arg.setValue(port);
126 VirtualMachine vm = conn.attach(conn_args);
127 vm.resume();
128
129 // if the second debuggee ran to completion then we've got a problem
130 if (exitCode == 0) {
131 throw new RuntimeException("Test failed - second debuggee didn't fail to bind");
132 } else {
133 System.out.println("Test passed - second debuggee correctly failed to bind");
134 }
135 }
136 }