1 /*
2 * Copyright (c) 2005, 2015, 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 import java.net.Socket;
25
26 import com.sun.jdi.Bootstrap;
27 import com.sun.jdi.VirtualMachine;
28 import com.sun.jdi.event.*;
29 import com.sun.jdi.connect.Connector;
30 import com.sun.jdi.connect.AttachingConnector;
31 import com.sun.jdi.connect.Connector.Argument;
32
33 import java.util.Map;
34 import java.util.List;
35 import java.util.Iterator;
36 import java.util.concurrent.TimeUnit;
37 import java.util.concurrent.atomic.AtomicBoolean;
38
39 import jdk.testlibrary.Utils;
40 import jdk.testlibrary.ProcessTools;
41
42 /* @test
43 * @bug 6306165 6432567
44 * @summary Check that a bad handshake doesn't cause a debuggee to abort
45 * @library /lib/testlibrary
46 *
47 * @modules java.management
48 * jdk.jdi
49 * @build jdk.testlibrary.* VMConnection BadHandshakeTest Exit0
50 * @run driver BadHandshakeTest
51 */
52 public class BadHandshakeTest {
53
54 /*
55 * Find a connector by name
56 */
57 private static Connector findConnector(String name) {
58 List<Connector> connectors = Bootstrap.virtualMachineManager().allConnectors();
59 Iterator<Connector> iter = connectors.iterator();
60 while (iter.hasNext()) {
61 Connector connector = (Connector)iter.next();
62 if (connector.name().equals(name)) {
63 return connector;
64 }
65 }
66 return null;
67 }
68
69 /*
70 * Launch a server debuggee with the given address
71 */
72 private static LaunchResult launch(String address, String class_name) throws Exception {
73 String[] args = VMConnection.insertDebuggeeVMOptions(new String[] {
74 "-agentlib:jdwp=transport=dt_socket" +
75 ",server=y" + ",suspend=y" + ",address=" + address,
76 class_name
77 });
78
79 ProcessBuilder pb = ProcessTools.createJavaProcessBuilder(args);
80
81 final AtomicBoolean success = new AtomicBoolean();
82 final AtomicBoolean bindFailed = new AtomicBoolean();
83 Process p = ProcessTools.startProcess(
84 class_name,
85 pb,
86 (line) -> {
87 // The first thing that will get read is
88 // Listening for transport dt_socket at address: xxxxx
89 // which shows the debuggee is ready to accept connections.
90 success.set(line.contains("Listening for transport dt_socket at address:"));
91 // If the first line contains 'Address already in use'
92 // that means the debuggee has failed to start due to busy port
93 bindFailed.set(line.contains("Address already in use"));
94 return true;
95 },
96 Integer.MAX_VALUE,
97 TimeUnit.MILLISECONDS
98 );
99
100 return new LaunchResult(success.get() ? p : null,
101 bindFailed.get());
102 }
103
104 /*
105 * - pick a TCP port
106 * - Launch a server debuggee: server=y,suspend=y,address=${port}
107 * - run it to VM death
108 * - verify we saw no error
109 */
110 public static void main(String args[]) throws Exception {
111 // Launch the server debuggee
112 int port = 0;
113 Process process = null;
114 while (process == null) {
115 port = Utils.getFreePort();
116 String address = String.valueOf(port);
117 LaunchResult launchResult = launch(address, "Exit0");
118 process = launchResult.getProcess();
119 if (launchResult.isBindFailed()) {
120 System.out.println("Port " + port + " already in use. Trying to restart debuggee with a new one...");
121 Thread.sleep(100);
122 } else if (process == null ) {
123 throw new RuntimeException("Unable to start debugee");
124 }
125 }
126
127 // Connect to the debuggee and handshake with garbage
128 Socket s = new Socket("localhost", port);
129 s.getOutputStream().write("Here's a poke in the eye".getBytes("UTF-8"));
130 s.close();
131
132 // Re-connect and to a partial handshake - don't disconnect
133 s = new Socket("localhost", port);
134 s.getOutputStream().write("JDWP-".getBytes("UTF-8"));
135
136
137 // Attach to server debuggee and resume it so it can exit
138 AttachingConnector conn = (AttachingConnector)findConnector("com.sun.jdi.SocketAttach");
139 Map<String, Argument> conn_args = conn.defaultArguments();
140 Connector.IntegerArgument port_arg =
141 (Connector.IntegerArgument)conn_args.get("port");
142 port_arg.setValue(port);
143 VirtualMachine vm = conn.attach(conn_args);
144
145 // The first event is always a VMStartEvent, and it is always in
146 // an EventSet by itself. Wait for it.
147 EventSet evtSet = vm.eventQueue().remove();
148 for (Event event: evtSet) {
149 if (event instanceof VMStartEvent) {
150 break;
151 }
152 throw new RuntimeException("Test failed - debuggee did not start properly");
153 }
154
155 vm.eventRequestManager().deleteAllBreakpoints();
156 vm.resume();
157
158 process.waitFor();
159 }
160
161 private static class LaunchResult {
162
163 private final Process p;
164 private final boolean bindFailed;
165
166 public LaunchResult(Process p, boolean bindFailed) {
167 this.p = p;
168 this.bindFailed = bindFailed;
169 }
170
171 public Process getProcess() {
172 return p;
173 }
174
175 public boolean isBindFailed() {
176 return bindFailed;
177 }
178
179 }
180
181 }