| /* |
| * Copyright (c) 2003, 2018, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| #include <stdio.h> |
| #include <string.h> |
| #include "jvmti.h" |
| #include "agent_common.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| #ifndef JNI_ENV_ARG |
| |
| #ifdef __cplusplus |
| #define JNI_ENV_ARG(x, y) y |
| #define JNI_ENV_ARG1(x) |
| #define JNI_ENV_PTR(x) x |
| #else |
| #define JNI_ENV_ARG(x,y) x, y |
| #define JNI_ENV_ARG1(x) x |
| #define JNI_ENV_PTR(x) (*x) |
| #endif |
| |
| #endif |
| |
| #define JVMTI_ENV_ARG JNI_ENV_ARG |
| #define JVMTI_ENV_ARG1 JNI_ENV_ARG1 |
| #define JVMTI_ENV_PTR JNI_ENV_PTR |
| |
| #define JVMTI_ERROR_CHECK(str,res) if ( res != JVMTI_ERROR_NONE) { printf(str); printf("%d\n",res); return res;} |
| #define JVMTI_ERROR_CHECK_EXPECTED_ERROR(str,res,err) if ( res != err) { printf(str); printf("unexpected error %d\n",res); return res;} |
| |
| #define JVMTI_ERROR_CHECK_VOID(str,res) if ( res != JVMTI_ERROR_NONE) { printf(str); printf("%d\n",res); iGlobalStatus = 2; } |
| |
| #define JVMTI_ERROR_CHECK_EXPECTED_ERROR_VOID(str,res,err) if ( res != err) { printf(str); printf("unexpected error %d\n",res); iGlobalStatus = 2; } |
| |
| #define THREADS_LIMIT 2000 |
| |
| |
| jvmtiEnv *jvmti; |
| jint iGlobalStatus = 0; |
| jthread susp_thrd[THREADS_LIMIT]; |
| static jvmtiEventCallbacks callbacks; |
| static jvmtiCapabilities jvmti_caps; |
| jrawMonitorID jraw_monitor[20]; |
| int process_once = 0; |
| |
| /* forward decl */ |
| void print_method_name(jmethodID mid); |
| |
| int printdump = 0; |
| |
| |
| void debug_printf(char *fmt, ...) { |
| va_list args; |
| |
| va_start(args, fmt); |
| if (printdump) { |
| vprintf(fmt, args); |
| } |
| va_end(args); |
| } |
| |
| void JNICALL vmInit(jvmtiEnv *jvmti_env, JNIEnv *env, jthread thread) { |
| debug_printf("VMInit event\n"); |
| } |
| |
| void JNICALL vmExit(jvmtiEnv *jvmti_env, JNIEnv *env) { |
| debug_printf("VMDeath event\n"); |
| } |
| |
| typedef jclass (*findLoadClass_type) (JNIEnv *env, jobject loader, jstring name); |
| |
| findLoadClass_type findLoadedClass_func; |
| |
| JNIEXPORT jclass JNICALL |
| my_findLoadedClass(JNIEnv *env, jobject loader, jstring name) { |
| const char* sname = (*env)->GetStringUTFChars(env, name, NULL); |
| debug_printf("Intercepted findLoadedClass, name = %s\n", sname); |
| return (*findLoadedClass_func)(env, loader, name); |
| } |
| |
| void JNICALL testNativeMethodBind(jvmtiEnv* jvmti_env, JNIEnv *jni_env, |
| jthread thread, jmethodID mid, void* func, void** func_ptr) { |
| jvmtiPhase phase; |
| char *mname; |
| char *signature; |
| jint ret; |
| |
| ret = (*jvmti_env)->GetPhase(jvmti_env, &phase); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: GetPhase %d\n", ret); |
| iGlobalStatus = 2; |
| return; |
| } |
| |
| if (phase != JVMTI_PHASE_START && phase != JVMTI_PHASE_LIVE) |
| return; |
| |
| debug_printf("bind event: \n"); |
| print_method_name(mid); |
| |
| ret = (*jvmti_env)->GetMethodName(jvmti_env, mid, &mname, &signature, NULL); |
| if (ret == JVMTI_ERROR_NONE) { |
| if (strcmp(mname, "findLoadedClass") == 0) { |
| findLoadedClass_func = (findLoadClass_type)func; |
| *func_ptr = (void*)my_findLoadedClass; |
| debug_printf("REDIRECTED findLoadedClass\n"); |
| } |
| } |
| } |
| |
| |
| |
| void init_callbacks() { |
| memset((void *)&callbacks, 0, sizeof(jvmtiEventCallbacks)); |
| callbacks.VMInit = vmInit; |
| callbacks.VMDeath = vmExit; |
| callbacks.NativeMethodBind = testNativeMethodBind; |
| } |
| |
| |
| #ifdef STATIC_BUILD |
| JNIEXPORT jint JNICALL Agent_OnLoad_JvmtiTest(JavaVM *jvm, char *options, void *reserved) { |
| return Agent_Initialize(jvm, options, reserved); |
| } |
| JNIEXPORT jint JNICALL Agent_OnAttach_JvmtiTest(JavaVM *jvm, char *options, void *reserved) { |
| return Agent_Initialize(jvm, options, reserved); |
| } |
| JNIEXPORT jint JNI_OnLoad_JvmtiTest(JavaVM *jvm, char *options, void *reserved) { |
| return JNI_VERSION_1_8; |
| } |
| #endif |
| jint Agent_Initialize(JavaVM * jvm, char *options, void *reserved) { |
| jint res; |
| |
| if (options && strlen(options) > 0) { |
| if (strstr(options, "printdump")) { |
| printdump = 1; |
| } |
| } |
| |
| res = JNI_ENV_PTR(jvm)-> |
| GetEnv(JNI_ENV_ARG(jvm, (void **) &jvmti), JVMTI_VERSION_1_1); |
| if (res < 0) { |
| printf("Wrong result of a valid call to GetEnv!\n"); |
| return JNI_ERR; |
| } |
| |
| |
| /* Add capabilities */ |
| res = JVMTI_ENV_PTR(jvmti)->GetPotentialCapabilities(JVMTI_ENV_ARG(jvmti, &jvmti_caps)); |
| JVMTI_ERROR_CHECK("GetPotentialCapabilities returned error", res); |
| |
| res = JVMTI_ENV_PTR(jvmti)->AddCapabilities(JVMTI_ENV_ARG(jvmti, &jvmti_caps)); |
| JVMTI_ERROR_CHECK("AddCapabilities returned error", res); |
| |
| /* Enable events */ |
| init_callbacks(); |
| res = JVMTI_ENV_PTR(jvmti)->SetEventCallbacks(JVMTI_ENV_ARG(jvmti, &callbacks), sizeof(callbacks)); |
| JVMTI_ERROR_CHECK("SetEventCallbacks returned error", res); |
| |
| res = JVMTI_ENV_PTR(jvmti)->SetEventNotificationMode(JVMTI_ENV_ARG(jvmti,JVMTI_ENABLE),JVMTI_EVENT_VM_INIT,NULL); |
| JVMTI_ERROR_CHECK("SetEventNotificationMode for VM_INIT returned error", res); |
| |
| res = JVMTI_ENV_PTR(jvmti)->SetEventNotificationMode(JVMTI_ENV_ARG(jvmti,JVMTI_ENABLE),JVMTI_EVENT_VM_DEATH,NULL); |
| JVMTI_ERROR_CHECK("SetEventNotificationMode for vm death event returned error", res); |
| |
| res = JVMTI_ENV_PTR(jvmti)->SetEventNotificationMode(JVMTI_ENV_ARG(jvmti,JVMTI_ENABLE),JVMTI_EVENT_NATIVE_METHOD_BIND,NULL); |
| JVMTI_ERROR_CHECK("SetEventNotificationMode for native method bind event returned error", res); |
| |
| return JNI_OK; |
| } |
| |
| |
| JNIEXPORT jint JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_GetResult(JNIEnv * env, jclass cls) { |
| return iGlobalStatus; |
| } |
| |
| |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_CreateRawMonitor(JNIEnv * env, jclass class, jint i) { |
| jvmtiError ret; |
| char sz[128]; |
| |
| sprintf(sz, "Rawmonitor-%d",i); |
| debug_printf("jvmti create raw monitor \n"); |
| ret = JVMTI_ENV_PTR(jvmti)->CreateRawMonitor(JVMTI_ENV_ARG1(jvmti),sz, &jraw_monitor[i]); |
| |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: ForceGarbageCollection %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| } |
| |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_RawMonitorEnter(JNIEnv * env, jclass cls, jint i) { |
| jvmtiError ret; |
| |
| debug_printf("jvmti Raw monitor enter \n"); |
| ret = JVMTI_ENV_PTR(jvmti)->RawMonitorEnter(JVMTI_ENV_ARG1(jvmti),jraw_monitor[i]); |
| |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: Raw monitor enter %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| } |
| |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_RawMonitorExit(JNIEnv * env, jclass cls, jint i) { |
| jvmtiError ret; |
| |
| debug_printf("jvmti raw monitor exit \n"); |
| ret = JVMTI_ENV_PTR(jvmti)->RawMonitorExit(JVMTI_ENV_ARG1(jvmti), jraw_monitor[i]); |
| |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: RawMonitorExit %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| } |
| |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_RawMonitorWait(JNIEnv * env, jclass cls, jint i) { |
| jvmtiError ret; |
| |
| debug_printf("jvmti RawMonitorWait \n"); |
| ret = JVMTI_ENV_PTR(jvmti)->RawMonitorWait(JVMTI_ENV_ARG1(jvmti),jraw_monitor[i],-1); |
| |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: RawMonitorWait %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| } |
| |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_RawMonitorNotify(JNIEnv * env, jclass cls, jint i) { |
| jvmtiError ret; |
| |
| debug_printf("jvmti RawMonitorNotify \n"); |
| ret = JVMTI_ENV_PTR(jvmti)->RawMonitorNotifyAll(JVMTI_ENV_ARG1(jvmti),jraw_monitor[i]); |
| |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: RawMonitorNotify %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| } |
| |
| JNIEXPORT int JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_GetFrameCount(JNIEnv * env, jclass cls, jobject thr) { |
| jvmtiError ret; |
| jint count; |
| |
| debug_printf("jvmti GetFrameCount \n"); |
| ret = JVMTI_ENV_PTR(jvmti)->GetFrameCount(JVMTI_ENV_ARG1(jvmti), (jthread)thr, &count); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: GetFrameCount returned %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| return count; |
| } |
| |
| void |
| print_method_name(jmethodID mid) { |
| jvmtiError ret; |
| jclass class; |
| char *mname; |
| char *signature; |
| char *clname = "unknown"; |
| ret = JVMTI_ENV_PTR(jvmti)->GetMethodDeclaringClass(JVMTI_ENV_ARG1(jvmti), mid, &class); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: GetMethodDeclaringClass %d \n", ret); |
| iGlobalStatus = 2; |
| return; |
| } |
| |
| ret = JVMTI_ENV_PTR(jvmti)->GetClassSignature(JVMTI_ENV_ARG1(jvmti), class, &clname, NULL); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: GetClassSignature %d \n", ret); |
| iGlobalStatus = 2; |
| return; |
| } |
| |
| ret = JVMTI_ENV_PTR(jvmti)->GetMethodName(JVMTI_ENV_ARG1(jvmti), mid, &mname, &signature, NULL); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: GetMethodName %d \n", ret); |
| iGlobalStatus = 2; |
| return; |
| } |
| |
| debug_printf("%s::%s(%s)\n", clname, mname, signature); |
| } |
| |
| |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_GetStackTrace(JNIEnv * env, jclass cls, jobject thr) { |
| jvmtiError ret; |
| jvmtiFrameInfo *stack_buffer = NULL; |
| jint max_count = 20; |
| jint count; |
| |
| |
| ret = JVMTI_ENV_PTR(jvmti)->Allocate(JVMTI_ENV_ARG1(jvmti), sizeof(jvmtiFrameInfo) * max_count, (unsigned char**)&stack_buffer); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: Allocate failed with %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| |
| ret = JVMTI_ENV_PTR(jvmti)->GetStackTrace(JVMTI_ENV_ARG1(jvmti), thr, 0, max_count , stack_buffer, &count); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: GetStackTrace %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| |
| ret = JVMTI_ENV_PTR(jvmti)->Deallocate(JVMTI_ENV_ARG1(jvmti), (unsigned char *)stack_buffer); |
| if (ret != JVMTI_ERROR_NONE) { |
| printf("Error: Deallocate failed with %d \n", ret); |
| iGlobalStatus = 2; |
| } |
| |
| |
| } |
| |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_unit_MethodBind_JvmtiTest_SaveThreadInfo(JNIEnv * env, jclass cls, jobject oobj) { |
| |
| } |
| |
| #ifdef __cplusplus |
| } |
| #endif |