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 "agent_common.h" 28 #include "jvmti_tools.h" 29 30 #define PASSED 0 31 #define STATUS_FAILED 2 32 33 extern "C" { 34 35 /* ========================================================================== */ 36 37 /* scaffold objects */ 38 static jlong timeout = 0; 39 40 /* test objects */ 41 static jthread thread = NULL; 42 43 /* event counts */ 44 static int MethodEntryEventsCount = 0; 45 46 /* ========================================================================== */ 47 48 /** callback functions **/ 49 50 static void JNICALL 51 MethodEntry(jvmtiEnv *jvmti_env, JNIEnv *jni_env, 52 jthread thread, jmethodID method) { 53 char *name = NULL; 54 char *signature = NULL; 55 56 MethodEntryEventsCount++; 57 if (!NSK_JVMTI_VERIFY(jvmti_env->GetMethodName(method, &name, &signature, NULL))) { 58 nsk_jvmti_setFailStatus(); 59 return; 60 } 61 NSK_DISPLAY2("MethodEntry event: %s%s\n", name, signature); 62 if (name != NULL) 63 jvmti_env->Deallocate((unsigned char*)name); 64 if (signature != NULL) 65 jvmti_env->Deallocate((unsigned char*)signature); 66 } 67 68 /* ========================================================================== */ 69 70 static int prepare(jvmtiEnv* jvmti, JNIEnv* jni) { 71 const char* THREAD_NAME = "Debuggee Thread"; 72 jvmtiThreadInfo info; 73 jthread *threads = NULL; 74 jint threads_count = 0; 75 int i; 76 77 NSK_DISPLAY0("Prepare: find tested thread\n"); 78 79 /* get all live threads */ 80 if (!NSK_JVMTI_VERIFY(jvmti->GetAllThreads(&threads_count, &threads))) 81 return NSK_FALSE; 82 83 if (!NSK_VERIFY(threads_count > 0 && threads != NULL)) 84 return NSK_FALSE; 85 86 /* find tested thread */ 87 for (i = 0; i < threads_count; i++) { 88 if (!NSK_VERIFY(threads[i] != NULL)) 89 return NSK_FALSE; 90 91 /* get thread information */ 92 if (!NSK_JVMTI_VERIFY(jvmti->GetThreadInfo(threads[i], &info))) 93 return NSK_FALSE; 94 95 NSK_DISPLAY3(" thread #%d (%s): %p\n", i, info.name, threads[i]); 96 97 /* find by name */ 98 if (info.name != NULL && (strcmp(info.name, THREAD_NAME) == 0)) { 99 thread = threads[i]; 100 } 101 } 102 103 if (!NSK_JNI_VERIFY(jni, (thread = jni->NewGlobalRef(thread)) != NULL)) 104 return NSK_FALSE; 105 106 /* deallocate threads list */ 107 if (!NSK_JVMTI_VERIFY(jvmti->Deallocate((unsigned char*)threads))) 108 return NSK_FALSE; 109 110 return NSK_TRUE; 111 } 112 113 /* ========================================================================== */ 114 115 /** Agent algorithm. */ 116 static void JNICALL 117 agentProc(jvmtiEnv* jvmti, JNIEnv* jni, void* arg) { 118 119 if (!nsk_jvmti_waitForSync(timeout)) 120 return; 121 122 if (!prepare(jvmti, jni)) { 123 nsk_jvmti_setFailStatus(); 124 return; 125 } 126 127 if (!NSK_JVMTI_VERIFY(jvmti->SetEventNotificationMode(JVMTI_ENABLE, JVMTI_EVENT_METHOD_ENTRY, thread))) 128 nsk_jvmti_setFailStatus(); 129 130 /* resume debugee and wait for sync */ 131 if (!nsk_jvmti_resumeSync()) 132 return; 133 if (!nsk_jvmti_waitForSync(timeout)) 134 return; 135 136 NSK_DISPLAY1("MethodEntry events received: %d\n", 137 MethodEntryEventsCount); 138 if (!NSK_VERIFY(MethodEntryEventsCount != 0)) 139 nsk_jvmti_setFailStatus(); 140 141 NSK_TRACE(jni->DeleteGlobalRef(thread)); 142 143 if (!nsk_jvmti_resumeSync()) 144 return; 145 } 146 147 /* ========================================================================== */ 148 149 /** Agent library initialization. */ 150 #ifdef STATIC_BUILD 151 JNIEXPORT jint JNICALL Agent_OnLoad_ma10t002(JavaVM *jvm, char *options, void *reserved) { 152 return Agent_Initialize(jvm, options, reserved); 153 } 154 JNIEXPORT jint JNICALL Agent_OnAttach_ma10t002(JavaVM *jvm, char *options, void *reserved) { 155 return Agent_Initialize(jvm, options, reserved); 156 } 157 JNIEXPORT jint JNI_OnLoad_ma10t002(JavaVM *jvm, char *options, void *reserved) { 158 return JNI_VERSION_1_8; 159 } 160 #endif 161 jint Agent_Initialize(JavaVM *jvm, char *options, void *reserved) { 162 jvmtiEnv* jvmti = NULL; 163 jvmtiCapabilities caps; 164 jvmtiEventCallbacks callbacks; 165 166 NSK_DISPLAY0("Agent_OnLoad\n"); 167 168 if (!NSK_VERIFY(nsk_jvmti_parseOptions(options))) 169 return JNI_ERR; 170 171 timeout = nsk_jvmti_getWaitTime() * 60 * 1000; 172 173 if (!NSK_VERIFY((jvmti = 174 nsk_jvmti_createJVMTIEnv(jvm, reserved)) != NULL)) 175 return JNI_ERR; 176 177 if (!NSK_VERIFY(nsk_jvmti_setAgentProc(agentProc, NULL))) 178 return JNI_ERR; 179 180 memset(&caps, 0, sizeof(caps)); 181 caps.can_generate_method_entry_events = 1; 182 if (!NSK_JVMTI_VERIFY(jvmti->AddCapabilities(&caps))) { 183 return JNI_ERR; 184 } 185 186 memset(&callbacks, 0, sizeof(callbacks)); 187 callbacks.MethodEntry = &MethodEntry; 188 if (!NSK_VERIFY(nsk_jvmti_init_MA(&callbacks))) 189 return JNI_ERR; 190 191 return JNI_OK; 192 } 193 194 /* ========================================================================== */ 195 196 }