1 /*
2 * Copyright (c) 1998, 2017, 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. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 #include "util.h"
27 #include "transport.h"
28 #include "debugLoop.h"
29 #include "debugDispatch.h"
30 #include "standardHandlers.h"
31 #include "inStream.h"
32 #include "outStream.h"
33 #include "threadControl.h"
34
35
36 static void JNICALL reader(jvmtiEnv* jvmti_env, JNIEnv* jni_env, void* arg);
37 static void enqueue(jdwpPacket *p);
38 static jboolean dequeue(jdwpPacket *p);
39 static void notifyTransportError(void);
40
41 struct PacketList {
42 jdwpPacket packet;
43 struct PacketList *next;
44 };
45
46 static volatile struct PacketList *cmdQueue;
47 static jrawMonitorID cmdQueueLock;
48 static jrawMonitorID vmDeathLock;
49 static jboolean transportError;
50
51 static jboolean
52 lastCommand(jdwpCmdPacket *cmd)
53 {
54 if ((cmd->cmdSet == JDWP_COMMAND_SET(VirtualMachine)) &&
55 ((cmd->cmd == JDWP_COMMAND(VirtualMachine, Dispose)) ||
56 (cmd->cmd == JDWP_COMMAND(VirtualMachine, Exit)))) {
57 return JNI_TRUE;
58 } else {
59 return JNI_FALSE;
60 }
61 }
62
63 void
64 debugLoop_initialize(void)
65 {
66 vmDeathLock = debugMonitorCreate("JDWP VM_DEATH Lock");
67 }
68
69 void
70 debugLoop_sync(void)
71 {
72 debugMonitorEnter(vmDeathLock);
73 debugMonitorExit(vmDeathLock);
74 }
75
76 /*
77 * This is where all the work gets done.
78 */
79
80 void
81 debugLoop_run(void)
82 {
83 jboolean shouldListen;
84 jdwpPacket p;
85 jvmtiStartFunction func;
86
87 /* Initialize all statics */
88 /* We may be starting a new connection after an error */
89 cmdQueue = NULL;
90 cmdQueueLock = debugMonitorCreate("JDWP Command Queue Lock");
91 transportError = JNI_FALSE;
92
93 shouldListen = JNI_TRUE;
94
95 func = &reader;
96 (void)spawnNewThread(func, NULL, "JDWP Command Reader");
97
98 standardHandlers_onConnect();
99 threadControl_onConnect();
100
101 /* Okay, start reading cmds! */
102 while (shouldListen) {
103 if (!dequeue(&p)) {
104 break;
105 }
106
107 if (p.type.cmd.flags & JDWPTRANSPORT_FLAGS_REPLY) {
108 /*
109 * Its a reply packet.
110 */
111 continue;
112 } else {
113 /*
114 * Its a cmd packet.
115 */
116 jdwpCmdPacket *cmd = &p.type.cmd;
117 PacketInputStream in;
118 PacketOutputStream out;
119 CommandHandler func;
120
121 /* Should reply be sent to sender.
122 * For error handling, assume yes, since
123 * only VM/exit does not reply
124 */
125 jboolean replyToSender = JNI_TRUE;
126
127 /*
128 * For all commands we hold the vmDeathLock
129 * while executing and replying to the command. This ensures
130 * that a command after VM_DEATH will be allowed to complete
131 * before the thread posting the VM_DEATH continues VM
132 * termination.
133 */
134 debugMonitorEnter(vmDeathLock);
135
136 /* Initialize the input and output streams */
137 inStream_init(&in, p);
138 outStream_initReply(&out, inStream_id(&in));
139
140 LOG_MISC(("Command set %d, command %d", cmd->cmdSet, cmd->cmd));
141
142 func = debugDispatch_getHandler(cmd->cmdSet,cmd->cmd);
143 if (func == NULL) {
144 /* we've never heard of this, so I guess we
145 * haven't implemented it.
146 * Handle gracefully for future expansion
147 * and platform / vendor expansion.
148 */
149 outStream_setError(&out, JDWP_ERROR(NOT_IMPLEMENTED));
150 } else if (gdata->vmDead &&
151 ((cmd->cmdSet) != JDWP_COMMAND_SET(VirtualMachine))) {
152 /* Protect the VM from calls while dead.
153 * VirtualMachine cmdSet quietly ignores some cmds
154 * after VM death, so, it sends it's own errors.
155 */
156 outStream_setError(&out, JDWP_ERROR(VM_DEAD));
157 } else {
158 /* Call the command handler */
159 replyToSender = func(&in, &out);
160 }
161
162 /* Reply to the sender */
163 if (replyToSender) {
164 if (inStream_error(&in)) {
165 outStream_setError(&out, inStream_error(&in));
166 }
167 outStream_sendReply(&out);
168 }
169
170 /*
171 * Release the vmDeathLock as the reply has been posted.
172 */
173 debugMonitorExit(vmDeathLock);
174
175 inStream_destroy(&in);
176 outStream_destroy(&out);
177
178 shouldListen = !lastCommand(cmd);
179 }
180 }
181 threadControl_onDisconnect();
182 standardHandlers_onDisconnect();
183
184 /*
185 * Cut off the transport immediately. This has the effect of
186 * cutting off any events that the eventHelper thread might
187 * be trying to send.
188 */
189 transport_close();
190 debugMonitorDestroy(cmdQueueLock);
191
192 /* Reset for a new connection to this VM if it's still alive */
193 if ( ! gdata->vmDead ) {
194 debugInit_reset(getEnv());
195 }
196 }
197
198 /* Command reader */
199 static void JNICALL
200 reader(jvmtiEnv* jvmti_env, JNIEnv* jni_env, void* arg)
201 {
202 jdwpPacket packet;
203 jdwpCmdPacket *cmd;
204 jboolean shouldListen = JNI_TRUE;
205
206 LOG_MISC(("Begin reader thread"));
207
208 while (shouldListen) {
209 jint rc;
210
211 rc = transport_receivePacket(&packet);
212
213 /* I/O error or EOF */
214 if (rc != 0 || (rc == 0 && packet.type.cmd.len == 0)) {
215 shouldListen = JNI_FALSE;
216 notifyTransportError();
217 } else if (packet.type.cmd.flags != JDWPTRANSPORT_FLAGS_NONE) {
218 /*
219 * Close the connection when we get a jdwpCmdPacket with an
220 * invalid flags field value. This is a protocol violation
221 * so we drop the connection. Also this could be a web
222 * browser generating an HTTP request that passes the JDWP
223 * handshake. HTTP requests requires that everything be in
224 * the ASCII printable range so a flags value of
225 * JDWPTRANSPORT_FLAGS_NONE(0) cannot be generated via HTTP.
226 */
227 ERROR_MESSAGE(("Received jdwpPacket with flags != 0x%d (actual=0x%x) when a jdwpCmdPacket was expected.",
228 JDWPTRANSPORT_FLAGS_NONE, packet.type.cmd.flags));
229 shouldListen = JNI_FALSE;
230 notifyTransportError();
231 } else {
232 cmd = &packet.type.cmd;
233
234 LOG_MISC(("Command set %d, command %d", cmd->cmdSet, cmd->cmd));
235
236 /*
237 * FIXME! We need to deal with high priority
238 * packets and queue flushes!
239 */
240 enqueue(&packet);
241
242 shouldListen = !lastCommand(cmd);
243 }
244 }
245 LOG_MISC(("End reader thread"));
246 }
247
248 /*
249 * The current system for queueing packets is highly
250 * inefficient, and should be rewritten! It'd be nice
251 * to avoid any additional memory allocations.
252 */
253
254 static void
255 enqueue(jdwpPacket *packet)
256 {
257 struct PacketList *pL;
258 struct PacketList *walker;
259
260 pL = jvmtiAllocate((jint)sizeof(struct PacketList));
261 if (pL == NULL) {
262 EXIT_ERROR(AGENT_ERROR_OUT_OF_MEMORY,"packet list");
263 }
264
265 pL->packet = *packet;
266 pL->next = NULL;
267
268 debugMonitorEnter(cmdQueueLock);
269
270 if (cmdQueue == NULL) {
271 cmdQueue = pL;
272 debugMonitorNotify(cmdQueueLock);
273 } else {
274 walker = (struct PacketList *)cmdQueue;
275 while (walker->next != NULL)
276 walker = walker->next;
277
278 walker->next = pL;
279 }
280
281 debugMonitorExit(cmdQueueLock);
282 }
283
284 static jboolean
285 dequeue(jdwpPacket *packet) {
286 struct PacketList *node = NULL;
287
288 debugMonitorEnter(cmdQueueLock);
289
290 while (!transportError && (cmdQueue == NULL)) {
291 debugMonitorWait(cmdQueueLock);
292 }
293
294 if (cmdQueue != NULL) {
295 node = (struct PacketList *)cmdQueue;
296 cmdQueue = node->next;
297 }
298 debugMonitorExit(cmdQueueLock);
299
300 if (node != NULL) {
301 *packet = node->packet;
302 jvmtiDeallocate(node);
303 }
304 return (node != NULL);
305 }
306
307 static void
308 notifyTransportError(void) {
309 debugMonitorEnter(cmdQueueLock);
310 transportError = JNI_TRUE;
311 debugMonitorNotify(cmdQueueLock);
312 debugMonitorExit(cmdQueueLock);
313 }