1 /*
2 * Copyright (c) 2004, 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 #include <stdlib.h>
25 #include <string.h>
26 #include "jni_tools.h"
27 #include "jvmti_tools.h"
28 #include "Injector.h"
29 #include "agent_common.h"
30
31 #define PASSED 0
32
33 extern "C" {
34
35 /* ========================================================================== */
36
37 #define DEFAULT_MAX_NUMBER_OF_CLASSES 100
38 #define DEFAULT_NUMBER_OF_SAMPLES 10
39 #define DEFAULT_SAMPLING_INTERVAL 100
40 #define DEFAULT_PACKAGE_NAME "nsk/jvmti/scenarios/hotswap"
41 #define PROFILE_CLASS_NAME "nsk/share/jvmti/ProfileCollector"
42
43 enum {
44 VM_MODE_COMPILED = 0,
45 VM_MODE_INTERPRETED = 1,
46 VM_MODE_MIXED = 2
47 };
48
49 /* scaffold objects */
50 static jlong timeout = 0;
51
52 /* test options */
53 static int number_of_samples;
54 static jlong sampling_interval;
55 static const char* package_name;
56 static size_t package_name_length;
57 static int vm_mode = VM_MODE_COMPILED;
58 static int bci_mode = BCI_MODE_EMCP;
59 static int sync_freq = 0;
60
61 static jclass profile_klass = NULL;
62 static jfieldID count_field = NULL;
63
64 /* test objects */
65 static int max_classes;
66 static char** names = NULL;
67 static jvmtiClassDefinition* old_class_def = NULL;
68 static jvmtiClassDefinition* new_class_def = NULL;
69 static int classCount = 0;
70 static int newFlag = NSK_FALSE;
71
72 /* ========================================================================== */
73
redefine(jvmtiEnv * jvmti,jvmtiClassDefinition * class_def)74 static int redefine(jvmtiEnv* jvmti, jvmtiClassDefinition* class_def) {
75
76 if (!NSK_VERIFY(classCount != 0))
77 return NSK_FALSE;
78
79 NSK_DISPLAY1("Redefining %d classes...\n", classCount);
80
81 if (!NSK_JVMTI_VERIFY(jvmti->RedefineClasses(classCount, class_def)))
82 return NSK_FALSE;
83
84 return NSK_TRUE;
85 }
86
87 /* ========================================================================== */
88
89 /** callback functions **/
90
91 static void JNICALL
ClassFileLoadHook(jvmtiEnv * jvmti_env,JNIEnv * jni_env,jclass class_being_redefined,jobject loader,const char * name,jobject protection_domain,jint class_data_len,const unsigned char * class_data,jint * new_class_data_len,unsigned char ** new_class_data)92 ClassFileLoadHook(jvmtiEnv *jvmti_env, JNIEnv *jni_env,
93 jclass class_being_redefined, jobject loader,
94 const char* name, jobject protection_domain,
95 jint class_data_len, const unsigned char* class_data,
96 jint *new_class_data_len, unsigned char** new_class_data) {
97 jint name_len;
98
99 if (name != NULL && classCount < max_classes &&
100 class_being_redefined == NULL &&
101 (strcmp(name, PROFILE_CLASS_NAME) != 0) &&
102 (strncmp(name, package_name, package_name_length) == 0)) {
103 NSK_DISPLAY1("ClassFileLoadHook: %s\n", name);
104 name_len = (jint) strlen(name) + 1;
105 if (!NSK_JVMTI_VERIFY(jvmti_env->Allocate(name_len, (unsigned char**) &names[classCount]))) {
106 nsk_jvmti_setFailStatus();
107 return;
108 }
109 memcpy(names[classCount], name, name_len);
110 if (!NSK_JVMTI_VERIFY(jvmti_env->Allocate(class_data_len, (unsigned char**)
111 &old_class_def[classCount].class_bytes))) {
112 nsk_jvmti_setFailStatus();
113 return;
114 }
115 memcpy((unsigned char*) old_class_def[classCount].class_bytes,
116 class_data, class_data_len);
117 old_class_def[classCount].class_byte_count = class_data_len;
118 classCount++;
119 }
120 }
121
122 static int CompiledMethodLoadEventsCount = 0;
123
124 static void JNICALL
CompiledMethodLoad(jvmtiEnv * jvmti_env,jmethodID method,jint code_size,const void * code_addr,jint map_length,const jvmtiAddrLocationMap * map,const void * compile_info)125 CompiledMethodLoad(jvmtiEnv *jvmti_env, jmethodID method,
126 jint code_size, const void* code_addr, jint map_length,
127 const jvmtiAddrLocationMap* map, const void* compile_info) {
128 char *name = NULL;
129 char *signature = NULL;
130
131 CompiledMethodLoadEventsCount++;
132
133 if (!NSK_JVMTI_VERIFY(jvmti_env->GetMethodName(method, &name, &signature, NULL))) {
134 nsk_jvmti_setFailStatus();
135 return;
136 }
137 NSK_DISPLAY3("CompiledMethodLoad event: %s%s (0x%p)\n",
138 name, signature, code_addr);
139 if (name != NULL)
140 jvmti_env->Deallocate((unsigned char*)name);
141 if (signature != NULL)
142 jvmti_env->Deallocate((unsigned char*)signature);
143 }
144
145 static int SingleStepEventsCount = 0;
146
147 static void JNICALL
SingleStep(jvmtiEnv * jvmti_env,JNIEnv * jni_env,jthread thread,jmethodID method,jlocation location)148 SingleStep(jvmtiEnv *jvmti_env, JNIEnv* jni_env, jthread thread,
149 jmethodID method, jlocation location) {
150
151 SingleStepEventsCount++;
152 }
153
154 static int ExceptionEventsCount = 0;
155
156 static void JNICALL
Exception(jvmtiEnv * jvmti_env,JNIEnv * jni_env,jthread thread,jmethodID method,jlocation location,jobject exception,jmethodID catch_method,jlocation catch_location)157 Exception(jvmtiEnv *jvmti_env, JNIEnv *jni_env, jthread thread,
158 jmethodID method, jlocation location, jobject exception,
159 jmethodID catch_method, jlocation catch_location) {
160
161 if (sync_freq && ((ExceptionEventsCount % sync_freq) == 0)) {
162
163 if (nsk_getVerboseMode()) {
164 jclass klass = NULL;
165 char *signature = NULL;
166
167 if (!NSK_JNI_VERIFY(jni_env, (klass = jni_env->GetObjectClass(exception)) != NULL)) {
168 nsk_jvmti_setFailStatus();
169 return;
170 }
171 if (!NSK_JVMTI_VERIFY(jvmti_env->GetClassSignature(klass, &signature, NULL))) {
172 nsk_jvmti_setFailStatus();
173 return;
174 }
175 NSK_DISPLAY2("Exception event %d: %s\n",
176 ExceptionEventsCount, signature);
177 if (signature != NULL)
178 jvmti_env->Deallocate((unsigned char*)signature);
179 }
180
181 if (!redefine(jvmti_env, (bci_mode != BCI_MODE_EMCP && newFlag) ?
182 new_class_def : old_class_def))
183 nsk_jvmti_setFailStatus();
184
185 NSK_DISPLAY1("SingleStepEventsCount: %d\n", SingleStepEventsCount);
186 if (vm_mode == VM_MODE_MIXED) {
187 if (!NSK_JVMTI_VERIFY(jvmti_env->SetEventNotificationMode(
188 ((newFlag) ? JVMTI_DISABLE : JVMTI_ENABLE),
189 JVMTI_EVENT_SINGLE_STEP, NULL)))
190 nsk_jvmti_setFailStatus();
191 }
192
193 if (nsk_getVerboseMode() && bci_mode != BCI_MODE_EMCP) {
194 jint profileCount = jni_env->GetStaticIntField(profile_klass, count_field);
195 NSK_DISPLAY1("profileCount: %d\n", profileCount);
196 }
197
198 newFlag = (newFlag) ? NSK_FALSE : NSK_TRUE;
199 }
200
201 ExceptionEventsCount++;
202 }
203
204 /* ========================================================================== */
205
206 static jrawMonitorID waitLock = NULL;
207
prepare(jvmtiEnv * jvmti,JNIEnv * jni)208 static int prepare(jvmtiEnv* jvmti, JNIEnv* jni) {
209 int i;
210
211 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_DISABLE, JVMTI_EVENT_CLASS_FILE_LOAD_HOOK, NULL)))
212 return NSK_FALSE;
213
214 if (vm_mode != VM_MODE_COMPILED) {
215 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_ENABLE, JVMTI_EVENT_SINGLE_STEP, NULL)))
216 return NSK_FALSE;
217 }
218
219 if (!NSK_JVMTI_VERIFY(jvmti->CreateRawMonitor("waitLock", &waitLock)))
220 return NSK_FALSE;
221
222 for (i = 0; i < classCount; i++) {
223 NSK_DISPLAY1("Find class: %s\n", names[i]);
224 if (!NSK_JNI_VERIFY(jni, (old_class_def[i].klass = jni->FindClass(names[i])) != NULL))
225 return NSK_FALSE;
226
227 if (!NSK_JNI_VERIFY(jni, (old_class_def[i].klass = (jclass)
228 jni->NewGlobalRef(old_class_def[i].klass)) != NULL))
229 return NSK_FALSE;
230 }
231
232 if (bci_mode != BCI_MODE_EMCP) {
233 NSK_DISPLAY1("Find class: %s\n", PROFILE_CLASS_NAME);
234 if (!NSK_JNI_VERIFY(jni, (profile_klass = jni->FindClass(PROFILE_CLASS_NAME)) != NULL))
235 return NSK_FALSE;
236
237 if (!NSK_JNI_VERIFY(jni, (profile_klass = (jclass)
238 jni->NewGlobalRef(profile_klass)) != NULL))
239 return NSK_FALSE;
240
241 if (!NSK_JNI_VERIFY(jni, (count_field =
242 jni->GetStaticFieldID(profile_klass,
243 (bci_mode == BCI_MODE_CALL) ? "callCount" : "allocCount",
244 "I")) != NULL))
245 return NSK_FALSE;
246
247 if (!NSK_JVMTI_VERIFY(jvmti->Allocate(classCount * sizeof(jvmtiClassDefinition),
248 (unsigned char**) &new_class_def)))
249 return NSK_FALSE;
250
251 for (i = 0; i < classCount; i++) {
252 new_class_def[i].klass = old_class_def[i].klass;
253 if (!Inject(old_class_def[i].class_bytes,
254 old_class_def[i].class_byte_count,
255 (unsigned char**) &new_class_def[i].class_bytes,
256 &new_class_def[i].class_byte_count, bci_mode))
257 return NSK_FALSE;
258 }
259 }
260
261 if (sync_freq) {
262 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_ENABLE, JVMTI_EVENT_EXCEPTION, NULL)))
263 return NSK_FALSE;
264 }
265
266 return NSK_TRUE;
267 }
268
269 /* ========================================================================== */
270
wait_for(jvmtiEnv * jvmti,jlong millis)271 static int wait_for(jvmtiEnv* jvmti, jlong millis) {
272
273 if (!NSK_JVMTI_VERIFY(jvmti->RawMonitorEnter(waitLock)))
274 return NSK_FALSE;
275
276 if (!NSK_JVMTI_VERIFY(jvmti->RawMonitorWait(waitLock, millis)))
277 nsk_jvmti_setFailStatus();
278
279 if (!NSK_JVMTI_VERIFY(jvmti->RawMonitorExit(waitLock)))
280 return NSK_FALSE;
281
282 return NSK_TRUE;
283 }
284
285 /* ========================================================================== */
286
287 /** Agent algorithm. */
288 static void JNICALL
agentProc(jvmtiEnv * jvmti,JNIEnv * jni,void * arg)289 agentProc(jvmtiEnv* jvmti, JNIEnv* jni, void* arg) {
290 int i;
291
292 if (!nsk_jvmti_waitForSync(timeout))
293 return;
294
295 if (!prepare(jvmti, jni)) {
296 nsk_jvmti_setFailStatus();
297 return;
298 }
299
300 /* resume debugee and wait for sync */
301 if (!nsk_jvmti_resumeSync())
302 return;
303 if (!nsk_jvmti_waitForSync(timeout))
304 return;
305
306 if (sync_freq) {
307 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_DISABLE, JVMTI_EVENT_EXCEPTION, NULL)))
308 nsk_jvmti_setFailStatus();
309 } else {
310
311 for (i = 0; i < number_of_samples && !nsk_jvmti_isFailStatus(); i++) {
312 wait_for(jvmti, sampling_interval);
313
314 if (!redefine(jvmti, (bci_mode != BCI_MODE_EMCP && newFlag) ?
315 new_class_def : old_class_def))
316 nsk_jvmti_setFailStatus();
317
318 NSK_DISPLAY1("SingleStepEventsCount: %d\n", SingleStepEventsCount);
319 if (vm_mode == VM_MODE_MIXED) {
320 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(
321 (((i % 2) == 0) ? JVMTI_DISABLE : JVMTI_ENABLE),
322 JVMTI_EVENT_SINGLE_STEP, NULL)))
323 nsk_jvmti_setFailStatus();
324 }
325
326 if (nsk_getVerboseMode() && bci_mode != BCI_MODE_EMCP) {
327 jint profileCount = jni->GetStaticIntField(profile_klass, count_field);
328 NSK_DISPLAY1("profileCount: %d\n", profileCount);
329 }
330
331 newFlag = (newFlag) ? NSK_FALSE : NSK_TRUE;
332 }
333
334 }
335
336 if (vm_mode != VM_MODE_COMPILED) {
337 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_DISABLE, JVMTI_EVENT_SINGLE_STEP, NULL)))
338 nsk_jvmti_setFailStatus();
339 }
340
341 if (!nsk_jvmti_resumeSync())
342 return;
343 }
344
345 /* ========================================================================== */
346
347 /** Agent library initialization. */
Agent_Initialize(JavaVM * jvm,char * options,void * reserved)348 jint Agent_Initialize(JavaVM *jvm, char *options, void *reserved) {
349 jvmtiEnv* jvmti = NULL;
350 jvmtiCapabilities caps;
351 jvmtiEventCallbacks callbacks;
352 const char* optValue;
353
354 NSK_DISPLAY0("Agent_OnLoad\n");
355
356 /* init framework and parse options */
357 if (!NSK_VERIFY(nsk_jvmti_parseOptions(options)))
358 return JNI_ERR;
359
360 timeout = nsk_jvmti_getWaitTime() * 60 * 1000;
361
362 /* get options */
363 number_of_samples = nsk_jvmti_findOptionIntValue("samples",
364 DEFAULT_NUMBER_OF_SAMPLES);
365 if (!NSK_VERIFY(number_of_samples > 0))
366 return JNI_ERR;
367 NSK_DISPLAY1("samples: %d\n", number_of_samples);
368
369 sampling_interval = nsk_jvmti_findOptionIntValue("interval",
370 DEFAULT_SAMPLING_INTERVAL);
371 if (!NSK_VERIFY(sampling_interval > 0))
372 return JNI_ERR;
373 NSK_DISPLAY1("interval: %d\n", sampling_interval);
374
375 package_name = nsk_jvmti_findOptionStringValue("package",
376 DEFAULT_PACKAGE_NAME);
377 if (!NSK_VERIFY(package_name != NULL))
378 return JNI_ERR;
379 NSK_DISPLAY1("package: %s\n", package_name);
380
381 package_name_length = strlen(package_name);
382 if (!NSK_VERIFY(package_name_length > 0))
383 return JNI_ERR;
384
385 max_classes = nsk_jvmti_findOptionIntValue("classes",
386 DEFAULT_MAX_NUMBER_OF_CLASSES);
387 if (!NSK_VERIFY(max_classes > 0))
388 return JNI_ERR;
389 NSK_DISPLAY1("classes: %d\n", max_classes);
390
391 optValue = nsk_jvmti_findOptionValue("mode");
392 if (optValue != NULL) {
393 if (strcmp(optValue, "compiled") == 0)
394 vm_mode = VM_MODE_COMPILED;
395 else if (strcmp(optValue, "interpreted") == 0)
396 vm_mode = VM_MODE_INTERPRETED;
397 else if (strcmp(optValue, "mixed") == 0)
398 vm_mode = VM_MODE_MIXED;
399 else {
400 NSK_COMPLAIN1("Unknown option value: mode=%s\n", optValue);
401 return JNI_ERR;
402 }
403 }
404
405 optValue = nsk_jvmti_findOptionValue("bci");
406 if (optValue != NULL) {
407 if (strcmp(optValue, "emcp") == 0)
408 bci_mode = BCI_MODE_EMCP;
409 else if (strcmp(optValue, "call") == 0)
410 bci_mode = BCI_MODE_CALL;
411 else if (strcmp(optValue, "alloc") == 0)
412 bci_mode = BCI_MODE_ALLOC;
413 else {
414 NSK_COMPLAIN1("Unknown option value: bci=%s\n", optValue);
415 return JNI_ERR;
416 }
417 }
418
419 sync_freq = nsk_jvmti_findOptionIntValue("sync", 0);
420 if (!NSK_VERIFY(sync_freq >= 0))
421 return JNI_ERR;
422 NSK_DISPLAY1("sync: %d\n", sync_freq);
423
424 /* create JVMTI environment */
425 if (!NSK_VERIFY((jvmti =
426 nsk_jvmti_createJVMTIEnv(jvm, reserved)) != NULL))
427 return JNI_ERR;
428
429 /* allocate tables for classes */
430 if (!NSK_JVMTI_VERIFY(jvmti->Allocate(max_classes * sizeof(char*), (unsigned char**) &names)))
431 return JNI_ERR;
432
433 if (!NSK_JVMTI_VERIFY(jvmti->Allocate(max_classes * sizeof(jvmtiClassDefinition),
434 (unsigned char**) &old_class_def)))
435 return JNI_ERR;
436
437 /* add capabilities */
438 memset(&caps, 0, sizeof(caps));
439 caps.can_redefine_classes = 1;
440 caps.can_generate_compiled_method_load_events = 1;
441 if (vm_mode != VM_MODE_COMPILED) {
442 caps.can_generate_single_step_events = 1;
443 }
444 if (sync_freq) {
445 caps.can_generate_exception_events = 1;
446 }
447 if (!NSK_JVMTI_VERIFY(jvmti->AddCapabilities(&caps)))
448 return JNI_ERR;
449
450 if (!NSK_VERIFY(nsk_jvmti_setAgentProc(agentProc, NULL)))
451 return JNI_ERR;
452
453 /* set event callbacks */
454 memset(&callbacks, 0, sizeof(callbacks));
455 callbacks.ClassFileLoadHook = &ClassFileLoadHook;
456 callbacks.CompiledMethodLoad = &CompiledMethodLoad;
457 if (vm_mode != VM_MODE_COMPILED) {
458 callbacks.SingleStep = &SingleStep;
459 }
460 if (sync_freq) {
461 callbacks.Exception = &Exception;
462 }
463 if (!NSK_JVMTI_VERIFY(jvmti->SetEventCallbacks(&callbacks, sizeof(callbacks))))
464 return JNI_ERR;
465
466 /* enable events */
467 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_ENABLE, JVMTI_EVENT_CLASS_FILE_LOAD_HOOK, NULL)))
468 return JNI_ERR;
469 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_ENABLE, JVMTI_EVENT_COMPILED_METHOD_LOAD, NULL)))
470 return JNI_ERR;
471
472 return JNI_OK;
473 }
474
475 /* ========================================================================== */
476
477 }
478