| /* |
| * 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 <string.h> |
| #include "jvmti.h" |
| #include "agent_common.h" |
| #include "jni_tools.h" |
| #include "jvmti_tools.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| /* ============================================================================= */ |
| |
| /* scaffold objects */ |
| static JNIEnv* jni = NULL; |
| static jvmtiEnv *jvmti = NULL; |
| static jlong timeout = 0; |
| |
| /* number of tested threads */ |
| #define THREADS_KINDS 6 |
| #define DEFAULT_THREADS_NUMBER 1 |
| |
| /* names of tested threads */ |
| static const char* threadsName[THREADS_KINDS] = { |
| "ThreadRunning", |
| "ThreadEntering", |
| "ThreadWaiting", |
| "ThreadSleeping", |
| "ThreadRunningInterrupted", |
| "ThreadRunningNative" |
| }; |
| |
| /* expected states of tested threads */ |
| #define JVMTI_THREAD_STATE_NOT_STARTED 0 |
| static jint threadsState[THREADS_KINDS] = { |
| JVMTI_THREAD_STATE_RUNNABLE, |
| JVMTI_THREAD_STATE_BLOCKED_ON_MONITOR_ENTER, |
| JVMTI_THREAD_STATE_IN_OBJECT_WAIT, |
| JVMTI_THREAD_STATE_SLEEPING, |
| JVMTI_THREAD_STATE_RUNNABLE, |
| JVMTI_THREAD_STATE_RUNNABLE |
| }; |
| |
| /* references to tested threads */ |
| static jthread* threadsList[THREADS_KINDS]; |
| static int threadsCounts[THREADS_KINDS]; |
| static int threadsCount = 0; |
| |
| /* indexes of known threads */ |
| static const int interruptedThreadIndex = THREADS_KINDS - 2; |
| static const int nativeThreadIndex = THREADS_KINDS - 1; |
| |
| /* ============================================================================= */ |
| |
| /* testcase(s) */ |
| static int prepare(); |
| static int checkThreads(int suspended, const char* kind, jlong timeout); |
| static int suspendThreadsList(int suspend); |
| static int suspendThreadsIndividually(int suspend); |
| static int clean(); |
| |
| /* ============================================================================= */ |
| |
| /** Agent algorithm. */ |
| static void JNICALL |
| agentProc(jvmtiEnv* jvmti, JNIEnv* agentJNI, void* arg) { |
| jni = agentJNI; |
| |
| /* wait for initial sync */ |
| if (!nsk_jvmti_waitForSync(timeout)) |
| return; |
| |
| /* perform testcase(s) */ |
| { |
| /* prepare data: find threads */ |
| if (!prepare()) { |
| nsk_jvmti_setFailStatus(); |
| return; |
| } |
| |
| /* testcase #1: check not suspended threads */ |
| NSK_DISPLAY0("Testcase #1: check state of not suspended threads\n"); |
| if (!checkThreads(NSK_FALSE, "not suspended", timeout)) |
| return; |
| |
| /* suspend threads */ |
| NSK_DISPLAY0("Suspend threads list\n"); |
| if (!suspendThreadsList(NSK_TRUE)) |
| return; |
| |
| /* testcase #2: check suspended threads */ |
| NSK_DISPLAY0("Testcase #2: check state of suspended threads\n"); |
| if (!checkThreads(NSK_TRUE, "suspended", 0)) |
| return; |
| |
| /* resume threads */ |
| NSK_DISPLAY0("Resume threads list\n"); |
| if (!suspendThreadsList(NSK_FALSE)) |
| return; |
| |
| /* testcase #3: check resumed threads */ |
| NSK_DISPLAY0("Testcase #3: check state of resumed threads\n"); |
| if (!checkThreads(NSK_FALSE, "resumed", 0)) |
| return; |
| |
| /* clean date: delete threads references */ |
| if (!clean()) { |
| nsk_jvmti_setFailStatus(); |
| return; |
| } |
| } |
| |
| /* resume debugee after last sync */ |
| if (!nsk_jvmti_resumeSync()) |
| return; |
| } |
| |
| /* ============================================================================= */ |
| |
| /** |
| * Prepare data: |
| * - clean threads list |
| * - get all live threads |
| * - get threads name |
| * - find tested threads |
| * - make global refs |
| */ |
| static int prepare() { |
| jthread *allThreadsList = NULL; |
| jint allThreadsCount = 0; |
| int notfound = 0; |
| int i, j; |
| |
| NSK_DISPLAY1("Prepare: find tested threads: %d kinds\n", THREADS_KINDS); |
| |
| /* allocate and clean threads list */ |
| for (i = 0; i < THREADS_KINDS; i++) { |
| threadsCounts[i] = 0; |
| threadsList[i] = NULL; |
| |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(Allocate, jvmti, (threadsCount * sizeof(jthread)), |
| (unsigned char**)&threadsList[i]))) |
| return NSK_FALSE; |
| |
| for (j = 0; j < threadsCount; j++) { |
| threadsList[i][j] = NULL; |
| } |
| } |
| |
| /* get all live threads */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(GetAllThreads, jvmti, &allThreadsCount, &allThreadsList))) |
| return NSK_FALSE; |
| |
| if (!NSK_VERIFY(allThreadsCount > 0 && allThreadsList != NULL)) |
| return NSK_FALSE; |
| |
| /* find tested threads */ |
| for (i = 0; i < allThreadsCount; i++) { |
| jvmtiThreadInfo threadInfo; |
| |
| if (!NSK_VERIFY(allThreadsList[i] != NULL)) |
| return NSK_FALSE; |
| |
| /* get thread name (info) */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(GetThreadInfo, jvmti, allThreadsList[i], &threadInfo))) |
| return NSK_FALSE; |
| |
| /* find by name */ |
| if (threadInfo.name != NULL) { |
| for (j = 0; j < THREADS_KINDS; j++) { |
| if (strcmp(threadInfo.name, threadsName[j]) == 0) { |
| int k = threadsCounts[j]; |
| if (k < threadsCount) |
| threadsList[j][k] = allThreadsList[i]; |
| threadsCounts[j]++; |
| } |
| } |
| } |
| } |
| |
| /* deallocate all threads list */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)allThreadsList))) |
| return NSK_FALSE; |
| |
| /* check if all tested threads found */ |
| notfound = 0; |
| for (i = 0; i < THREADS_KINDS; i++) { |
| if (threadsCounts[i] != threadsCount) { |
| NSK_COMPLAIN3("Found unexpected number of tested threads (%s):\n" |
| "# found: %d\n" |
| "# expected: %d\n", |
| threadsName[i], threadsCounts[i], threadsCount); |
| nsk_jvmti_setFailStatus(); |
| notfound++; |
| } |
| } |
| |
| if (notfound > 0) |
| return NSK_FALSE; |
| |
| /* make global refs */ |
| for (i = 0; i < THREADS_KINDS; i++) { |
| for (j = 0; j < threadsCount; j++) { |
| if (!NSK_JNI_VERIFY(jni, (threadsList[i][j] = |
| NSK_CPP_STUB2(NewGlobalRef, jni, threadsList[i][j])) != NULL)) |
| return NSK_FALSE; |
| } |
| } |
| |
| return NSK_TRUE; |
| } |
| |
| /** |
| * Suspend or resume tested threads list. |
| */ |
| static int suspendThreadsList(int suspend) { |
| jlong resultsSize = threadsCount * sizeof(jvmtiError); |
| jvmtiError* results = NULL; |
| const char* kind = (suspend ? "suspending" : "resuming"); |
| int i, j; |
| |
| /* allocate results array */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(Allocate, jvmti, resultsSize, (unsigned char**)&results))) { |
| nsk_jvmti_setFailStatus(); |
| return NSK_FALSE; |
| } |
| |
| for (i = 0; i < THREADS_KINDS; i++) { |
| /* suspend or resume threads list */ |
| if (suspend) { |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB4(SuspendThreadList, jvmti, threadsCount, |
| threadsList[i], results))) |
| nsk_jvmti_setFailStatus(); |
| } else { |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB4(ResumeThreadList, jvmti, threadsCount, |
| threadsList[i], results))) |
| nsk_jvmti_setFailStatus(); |
| } |
| |
| /* check results */ |
| for (j = 0; j < threadsCount; j++) { |
| if (results[j] != JVMTI_ERROR_NONE) { |
| NSK_COMPLAIN5("Unexpected result of %s thread #%d (%s):\n" |
| "# got result: %s (%d)\n", |
| kind, j, threadsName[i], |
| TranslateError(results[j]), (int)results[j]); |
| nsk_jvmti_setFailStatus(); |
| } |
| } |
| } |
| |
| /* deallocate results array */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)results))) { |
| nsk_jvmti_setFailStatus(); |
| } |
| |
| return NSK_TRUE; |
| } |
| |
| /** |
| * Suspend or resume tested threads individually. |
| */ |
| static int suspendThreadsIndividually(int suspend) { |
| int i, j; |
| |
| for (i = 0; i < THREADS_KINDS; i++) { |
| for (j = 0; j < threadsCount; j++) { |
| if (suspend) { |
| NSK_DISPLAY2(" suspend thread #%d (%s)\n", j, threadsName[i]); |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB2(SuspendThread, jvmti, threadsList[i][j]))) |
| nsk_jvmti_setFailStatus(); |
| } else { |
| NSK_DISPLAY2(" resume thread #%d (%s)\n", j, threadsName[i]); |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB2(ResumeThread, jvmti, threadsList[i][j]))) |
| nsk_jvmti_setFailStatus(); |
| } |
| } |
| } |
| return NSK_TRUE; |
| } |
| |
| /** |
| * Testcase: check tested threads |
| * - get thread state and state flag |
| * - wait for WAITIME if state is not expected |
| * - check if thread state is as expected |
| * |
| * Returns NSK_TRUE if test may continue; or NSK_FALSE for test break. |
| */ |
| static int checkThreads(int suspended, const char* kind, jlong timeout) { |
| int i, j; |
| |
| /* check each thread */ |
| for (i = 0; i < THREADS_KINDS; i++) { |
| for (j = 0; j < threadsCount; j++) { |
| jint state = JVMTI_THREAD_STATE_NOT_STARTED; |
| jlong t = 0; |
| |
| NSK_DISPLAY2(" thread #%d (%s):\n", j, threadsName[i]); |
| |
| /* get thread state */ |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB3(GetThreadState, jvmti, threadsList[i][j], &state))) { |
| nsk_jvmti_setFailStatus(); |
| return NSK_FALSE; |
| } |
| |
| NSK_DISPLAY2(" state = %s (%d)\n", |
| TranslateState(state), (int)state); |
| |
| /* check SUSPENDED state */ |
| if (suspended) { |
| if (!(state & JVMTI_THREAD_STATE_SUSPENDED)) { |
| NSK_COMPLAIN5("No SUSPENDED state for %s thread #%d (%s):\n" |
| "# got flags: %s (%d)\n", |
| kind, j, threadsName[i], |
| TranslateState(state), (int)state); |
| nsk_jvmti_setFailStatus(); |
| } |
| } else { |
| if (state & JVMTI_THREAD_STATE_SUSPENDED) { |
| NSK_COMPLAIN5("Unexpected SUSPENDED state for %s thread #%d (%s):\n" |
| "# got flags: %s (%d)\n", |
| kind, j, threadsName[i], |
| TranslateState(state), (int)state); |
| nsk_jvmti_setFailStatus(); |
| } |
| } |
| } |
| } |
| |
| /* test may continue */ |
| return NSK_TRUE; |
| } |
| |
| /** |
| * Clean data: |
| * - dispose global references to tested threads |
| */ |
| static int clean() { |
| int i, j; |
| |
| /* dispose global references to threads */ |
| for (i = 0; i < THREADS_KINDS; i++) { |
| for (j = 0; j < threadsCount; j++) { |
| NSK_TRACE(NSK_CPP_STUB2(DeleteGlobalRef, jni, threadsList[i][j])); |
| } |
| } |
| |
| /* deallocate memory */ |
| for (i = 0; i < THREADS_KINDS; i++) { |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB2(Deallocate, jvmti, (unsigned char*)threadsList[i]))) |
| return NSK_FALSE; |
| threadsList[i] = NULL; |
| } |
| |
| return NSK_TRUE; |
| } |
| |
| /* ============================================================================= */ |
| |
| static volatile int testedThreadRunning = NSK_FALSE; |
| static volatile int testedThreadShouldFinish = NSK_FALSE; |
| |
| /** Native running method in tested thread. */ |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_scenarios_sampling_SP03_sp03t001ThreadRunningNative_nativeMethod(JNIEnv* jni, |
| jobject obj) { |
| volatile int i = 0, n = 1000; |
| |
| /* run in a loop */ |
| testedThreadRunning = NSK_TRUE; |
| while (!testedThreadShouldFinish) { |
| if (n <= 0) |
| n = 1000; |
| if (i >= n) |
| i = 0; |
| i++; |
| } |
| testedThreadRunning = NSK_FALSE; |
| } |
| |
| /** Wait for native method is running. */ |
| JNIEXPORT jboolean JNICALL |
| Java_nsk_jvmti_scenarios_sampling_SP03_sp03t001ThreadRunningNative_checkReady(JNIEnv* jni, |
| jobject obj) { |
| while (!testedThreadRunning) { |
| nsk_jvmti_sleep(1000); |
| } |
| return testedThreadRunning ? JNI_TRUE : JNI_FALSE; |
| } |
| |
| /* Let native method to finish. */ |
| JNIEXPORT void JNICALL |
| Java_nsk_jvmti_scenarios_sampling_SP03_sp03t001ThreadRunningNative_letFinish(JNIEnv* jni, |
| jobject obj) { |
| testedThreadShouldFinish = NSK_TRUE; |
| } |
| |
| /* ============================================================================= */ |
| |
| /** Agent library initialization. */ |
| #ifdef STATIC_BUILD |
| JNIEXPORT jint JNICALL Agent_OnLoad_sp03t001(JavaVM *jvm, char *options, void *reserved) { |
| return Agent_Initialize(jvm, options, reserved); |
| } |
| JNIEXPORT jint JNICALL Agent_OnAttach_sp03t001(JavaVM *jvm, char *options, void *reserved) { |
| return Agent_Initialize(jvm, options, reserved); |
| } |
| JNIEXPORT jint JNI_OnLoad_sp03t001(JavaVM *jvm, char *options, void *reserved) { |
| return JNI_VERSION_1_8; |
| } |
| #endif |
| jint Agent_Initialize(JavaVM *jvm, char *options, void *reserved) { |
| |
| /* init framework and parse options */ |
| if (!NSK_VERIFY(nsk_jvmti_parseOptions(options))) |
| return JNI_ERR; |
| |
| timeout = nsk_jvmti_getWaitTime() * 60 * 1000; |
| |
| /* get number of threads for each kind */ |
| threadsCount = nsk_jvmti_findOptionIntValue("threads", DEFAULT_THREADS_NUMBER); |
| if (!NSK_VERIFY(threadsCount > 0)) |
| return JNI_ERR; |
| |
| /* create JVMTI environment */ |
| if (!NSK_VERIFY((jvmti = |
| nsk_jvmti_createJVMTIEnv(jvm, reserved)) != NULL)) |
| return JNI_ERR; |
| |
| /* add specific capabilities for suspending thread */ |
| { |
| jvmtiCapabilities suspendCaps; |
| memset(&suspendCaps, 0, sizeof(suspendCaps)); |
| suspendCaps.can_suspend = 1; |
| if (!NSK_JVMTI_VERIFY( |
| NSK_CPP_STUB2(AddCapabilities, jvmti, &suspendCaps))) |
| return JNI_ERR; |
| } |
| |
| /* register agent proc and arg */ |
| if (!NSK_VERIFY(nsk_jvmti_setAgentProc(agentProc, NULL))) |
| return JNI_ERR; |
| |
| return JNI_OK; |
| } |
| |
| /* ============================================================================= */ |
| |
| #ifdef __cplusplus |
| } |
| #endif |