| /* |
| * Copyright (c) 2004, 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 <stdlib.h> |
| #include <string.h> |
| #include "jni_tools.h" |
| #include "agent_common.h" |
| #include "jvmti_tools.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| #define MAX_DEPTH 1024 |
| |
| /* ========================================================================== */ |
| |
| static const int NUMBER_OF_SAMPLES = 1000; |
| static const int DISPLAYING_FREQUENCY = 100; |
| static const jlong SAMPLING_INTERVAL = 10; |
| |
| /* scaffold objects */ |
| static jlong timeout = 0; |
| |
| /* test objects */ |
| static jthread thread = NULL; |
| static jrawMonitorID waitLock = NULL; |
| static jrawMonitorID frameLock = NULL; |
| static int sampleCount = 0; |
| static volatile int depth = 0; |
| static jvmtiFrameInfo sampleStack[MAX_DEPTH]; |
| static jint frameCount = 0; |
| static jvmtiFrameInfo frameBuffer[MAX_DEPTH]; |
| |
| /* ========================================================================== */ |
| |
| static int prepare(jvmtiEnv* jvmti, JNIEnv* jni) { |
| const char* THREAD_NAME = "Debuggee Thread"; |
| jvmtiThreadInfo info; |
| jthread *threads = NULL; |
| jint threads_count = 0; |
| int i; |
| |
| NSK_DISPLAY0("Prepare: find tested thread\n"); |
| |
| /* get all live threads */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(GetAllThreads, jvmti, &threads_count, &threads))) |
| return NSK_FALSE; |
| |
| if (!NSK_VERIFY(threads_count > 0 && threads != NULL)) |
| return NSK_FALSE; |
| |
| /* find tested thread */ |
| for (i = 0; i < threads_count; i++) { |
| if (!NSK_VERIFY(threads[i] != NULL)) |
| return NSK_FALSE; |
| |
| /* get thread information */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(GetThreadInfo, jvmti, threads[i], &info))) |
| return NSK_FALSE; |
| |
| NSK_DISPLAY3(" thread #%d (%s): %p\n", i, info.name, threads[i]); |
| |
| /* find by name */ |
| if (info.name != NULL && (strcmp(info.name, THREAD_NAME) == 0)) { |
| thread = threads[i]; |
| } |
| |
| if (info.name != NULL) { |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2( |
| Deallocate, jvmti, (unsigned char*)info.name))) |
| return NSK_FALSE; |
| } |
| } |
| |
| /* deallocate threads list */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)threads))) |
| return NSK_FALSE; |
| |
| if (thread == NULL) { |
| NSK_COMPLAIN0("Debuggee thread not found"); |
| return NSK_FALSE; |
| } |
| |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(CreateRawMonitor, jvmti, "waitLock", &waitLock))) |
| return NSK_FALSE; |
| |
| return NSK_TRUE; |
| } |
| |
| /* ============================================================================= */ |
| |
| static int wait_for(jvmtiEnv* jvmti, jlong millis) { |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorEnter, jvmti, waitLock))) |
| return NSK_FALSE; |
| |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(RawMonitorWait, jvmti, waitLock, millis))) |
| nsk_jvmti_setFailStatus(); |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorExit, jvmti, waitLock))) |
| return NSK_FALSE; |
| |
| return NSK_TRUE; |
| } |
| |
| static int displayFrameInfo(jvmtiEnv* jvmti, jint i) { |
| char buffer[32]; |
| char *name = NULL; |
| char *signature = NULL; |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB5(GetMethodName, jvmti, |
| frameBuffer[frameCount-1-i].method, &name, &signature, NULL))) |
| return NSK_FALSE; |
| |
| NSK_DISPLAY4(" got[%d] method: %s%s, location: %s\n", i, name, |
| signature, jlong_to_string(frameBuffer[frameCount-1-i].location, buffer)); |
| if (name != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)name); |
| if (signature != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)signature); |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB5(GetMethodName, jvmti, |
| sampleStack[i].method, &name, &signature, NULL))) |
| return NSK_FALSE; |
| |
| NSK_DISPLAY4(" exp[%d] method: %s%s, location: %s\n", i, name, |
| signature, jlong_to_string(sampleStack[i].location, buffer)); |
| if (name != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)name); |
| if (signature != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)signature); |
| |
| return NSK_TRUE; |
| } |
| |
| static int complainFrameInfo(jvmtiEnv* jvmti, jint i) { |
| char buffer[32]; |
| char *name = NULL; |
| char *signature = NULL; |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB5(GetMethodName, jvmti, |
| frameBuffer[frameCount-1-i].method, &name, &signature, NULL))) |
| return NSK_FALSE; |
| |
| NSK_COMPLAIN3(" got: method=%s%s, location=%s\n", name, signature, |
| jlong_to_string(frameBuffer[frameCount-1-i].location, buffer)); |
| if (name != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)name); |
| if (signature != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)signature); |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB5(GetMethodName, jvmti, |
| sampleStack[i].method, &name, &signature, NULL))) |
| return NSK_FALSE; |
| |
| NSK_COMPLAIN3(" expected: method=%s%s, location=%s\n", name, signature, |
| jlong_to_string(sampleStack[i].location, buffer)); |
| if (name != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)name); |
| if (signature != NULL) |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)signature); |
| |
| return NSK_TRUE; |
| } |
| |
| static int checkStackTrace(jvmtiEnv* jvmti, JNIEnv* jni) { |
| jint i; |
| int res = NSK_TRUE; |
| int displayFlag = |
| (nsk_getVerboseMode() && (sampleCount % DISPLAYING_FREQUENCY) == 0); |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorEnter, jvmti, frameLock))) |
| return NSK_FALSE; |
| |
| /* get stack trace */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB6(GetStackTrace, jvmti, thread, |
| 0, MAX_DEPTH, frameBuffer, &frameCount))) { |
| res = NSK_FALSE; |
| } else { |
| if (displayFlag) { |
| NSK_DISPLAY3("Sample #%d, frameCount: %d, depth: %d\n", |
| sampleCount, frameCount, depth); |
| } |
| if (!NSK_VERIFY(frameCount >= depth)) { |
| NSK_COMPLAIN3("Sample #%d, wrong frameCount: %d, expected >= %d\n", |
| sampleCount, frameCount, depth); |
| res = NSK_FALSE; |
| } else { |
| for (i = 0; i < depth; i++) { |
| if (displayFlag && !displayFrameInfo(jvmti, i)) |
| res = NSK_FALSE; |
| if (!NSK_VERIFY(sampleStack[i].method == |
| frameBuffer[frameCount-1-i].method) || |
| !NSK_VERIFY(sampleStack[i].location == |
| frameBuffer[frameCount-1-i].location)) { |
| |
| NSK_COMPLAIN3("Sample #%d, depth=%d, wrong frame [%d]:\n", |
| sampleCount, depth, i); |
| complainFrameInfo(jvmti, i); |
| res = NSK_FALSE; |
| } |
| } |
| } |
| } |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorExit, jvmti, frameLock))) |
| return NSK_FALSE; |
| |
| return res; |
| } |
| |
| /* ========================================================================== */ |
| |
| /* agent algorithm */ |
| static void JNICALL |
| agentProc(jvmtiEnv* jvmti, JNIEnv* jni, void* arg) { |
| |
| /* wait for initial sync */ |
| if (!nsk_jvmti_waitForSync(timeout)) |
| return; |
| |
| if (!prepare(jvmti, jni)) { |
| nsk_jvmti_setFailStatus(); |
| return; |
| } |
| |
| /* sample stack of the tested thread */ |
| for (sampleCount = 0; |
| sampleCount < NUMBER_OF_SAMPLES && !nsk_jvmti_isFailStatus(); |
| sampleCount++) { |
| wait_for(jvmti, SAMPLING_INTERVAL); |
| if (!checkStackTrace(jvmti, jni)) |
| nsk_jvmti_setFailStatus(); |
| } |
| |
| /* resume debugee after last sync */ |
| if (!nsk_jvmti_resumeSync()) |
| return; |
| } |
| |
| /* ========================================================================== */ |
| |
| JNIEXPORT jint JNICALL |
| Java_nsk_jvmti_scenarios_sampling_SP07_sp07t001Thread_wrapper(JNIEnv* jni, |
| jobject obj, jint i) { |
| jint result = 0; |
| jclass klass = NULL; |
| jmethodID method = NULL; |
| jvmtiEnv* jvmti = nsk_jvmti_getAgentJVMTIEnv(); |
| |
| if (!NSK_VERIFY(depth < MAX_DEPTH)) { |
| nsk_jvmti_setFailStatus(); |
| return 0; |
| } |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorEnter, jvmti, frameLock))) |
| return NSK_FALSE; |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB5(GetFrameLocation, jvmti, NULL, 1, |
| &sampleStack[depth].method, &sampleStack[depth].location))) { |
| nsk_jvmti_setFailStatus(); |
| return 0; |
| } |
| |
| depth++; |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB5(GetFrameLocation, jvmti, NULL, 0, |
| &sampleStack[depth].method, &sampleStack[depth].location))) { |
| nsk_jvmti_setFailStatus(); |
| return 0; |
| } |
| |
| depth++; |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorExit, jvmti, frameLock))) |
| return NSK_FALSE; |
| |
| if (!NSK_JNI_VERIFY(jni, (klass = NSK_CPP_STUB2(GetObjectClass, |
| jni, obj)) != NULL)) { |
| nsk_jvmti_setFailStatus(); |
| return 0; |
| } |
| |
| if (!NSK_JNI_VERIFY(jni, (method = NSK_CPP_STUB4(GetMethodID, |
| jni, klass, "fibonacci", "(I)I")) != NULL)) { |
| nsk_jvmti_setFailStatus(); |
| return 0; |
| } |
| |
| result = NSK_CPP_STUB4(CallIntMethod, jni, obj, method, i); |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorEnter, jvmti, frameLock))) |
| return NSK_FALSE; |
| |
| depth--; |
| depth--; |
| |
| if (!NSK_JVMTI_VERIFY(NSK_CPP_STUB2(RawMonitorExit, jvmti, frameLock))) |
| return NSK_FALSE; |
| |
| return result; |
| } |
| |
| /* ========================================================================== */ |
| |
| /* agent library initialization */ |
| #ifdef STATIC_BUILD |
| JNIEXPORT jint JNICALL Agent_OnLoad_sp07t001(JavaVM *jvm, char *options, void *reserved) { |
| return Agent_Initialize(jvm, options, reserved); |
| } |
| JNIEXPORT jint JNICALL Agent_OnAttach_sp07t001(JavaVM *jvm, char *options, void *reserved) { |
| return Agent_Initialize(jvm, options, reserved); |
| } |
| JNIEXPORT jint JNI_OnLoad_sp07t001(JavaVM *jvm, char *options, void *reserved) { |
| return JNI_VERSION_1_8; |
| } |
| #endif |
| jint Agent_Initialize(JavaVM *jvm, char *options, void *reserved) { |
| jvmtiEnv* jvmti = NULL; |
| |
| /* init framework and parse options */ |
| if (!NSK_VERIFY(nsk_jvmti_parseOptions(options))) |
| return JNI_ERR; |
| |
| timeout = nsk_jvmti_getWaitTime() * 60000; |
| NSK_DISPLAY1("Timeout: %d msc\n", (int)timeout); |
| |
| /* create JVMTI environment */ |
| if (!NSK_VERIFY((jvmti = |
| nsk_jvmti_createJVMTIEnv(jvm, reserved)) != NULL)) |
| return JNI_ERR; |
| |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(CreateRawMonitor, jvmti, "frameLock", &frameLock))) |
| return NSK_FALSE; |
| |
| /* register agent proc and arg */ |
| if (!NSK_VERIFY(nsk_jvmti_setAgentProc(agentProc, NULL))) |
| return JNI_ERR; |
| |
| return JNI_OK; |
| } |
| |
| /* ========================================================================== */ |
| |
| #ifdef __cplusplus |
| } |
| #endif |
