1 /*
   2  * Copyright (c) 2003, 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 <stdio.h>
  25 #include <string.h>
  26 #include "jvmti.h"
  27 #include "agent_common.h"
  28 
  29 extern "C" {
  30 
  31 
  32 #define JVMTI_ERROR_CHECK(str,res) if ( res != JVMTI_ERROR_NONE) { printf(str); printf("%d\n",res); return res;}
  33 #define JVMTI_ERROR_CHECK_EXPECTED_ERROR(str,res,err) if ( res != err) { printf(str); printf("unexpected error %d\n",res); return res;}
  34 
  35 #define JVMTI_ERROR_CHECK_VOID(str,res) if ( res != JVMTI_ERROR_NONE) { printf(str); printf("%d\n",res); iGlobalStatus = 2; }
  36 
  37 #define JVMTI_ERROR_CHECK_EXPECTED_ERROR_VOID(str,res,err) if ( res != err) { printf(str); printf("unexpected error %d\n",res); iGlobalStatus = 2; }
  38 
  39 #define THREADS_LIMIT 2000
  40 
  41 
  42 jvmtiEnv *jvmti;
  43 jint iGlobalStatus = 0;
  44 jthread susp_thrd[THREADS_LIMIT];
  45 static jvmtiEventCallbacks callbacks;
  46 static jvmtiCapabilities jvmti_caps;
  47 jrawMonitorID jraw_monitor[20];
  48 
  49 int process_once = 0;
  50 
  51 
  52 
  53 int printdump = 0;
  54 
  55 
  56 void debug_printf(const char *fmt, ...) {
  57     va_list args;
  58 
  59     va_start(args, fmt);
  60     if (printdump) {
  61         vprintf(fmt, args);
  62     }
  63     va_end(args);
  64 }
  65 
  66 
  67 void JNICALL vmInit(jvmtiEnv *jvmti_env, JNIEnv *env, jthread thread) {
  68 
  69     debug_printf("VMInit event  done\n");
  70 
  71 }
  72 
  73 void JNICALL vmExit(jvmtiEnv *jvmti_env, JNIEnv *env) {
  74     debug_printf("------------ JVMTI_EVENT_VM_DEATH ------------\n");
  75 }
  76 
  77 void JNICALL classFileLoadEvent(jvmtiEnv *jvmti_env, JNIEnv *env,
  78                         jclass class_being_redifined,
  79                         jobject loader, const char* name,
  80                         jobject protection_domain,
  81                         jint class_data_len,
  82                         const unsigned char* class_data,
  83                         jint* new_class_data_len,
  84                         unsigned char** new_class_data) {
  85 
  86 }
  87 
  88 
  89 
  90 void init_callbacks() {
  91     memset((void *)&callbacks, 0, sizeof(jvmtiEventCallbacks));
  92     callbacks.VMInit = vmInit;
  93     callbacks.VMDeath = vmExit;
  94     callbacks.ClassFileLoadHook = classFileLoadEvent;
  95 }
  96 
  97 
  98 #ifdef STATIC_BUILD
  99 JNIEXPORT jint JNICALL Agent_OnLoad_JvmtiTest(JavaVM *jvm, char *options, void *reserved) {
 100     return Agent_Initialize(jvm, options, reserved);
 101 }
 102 JNIEXPORT jint JNICALL Agent_OnAttach_JvmtiTest(JavaVM *jvm, char *options, void *reserved) {
 103     return Agent_Initialize(jvm, options, reserved);
 104 }
 105 JNIEXPORT jint JNI_OnLoad_JvmtiTest(JavaVM *jvm, char *options, void *reserved) {
 106     return JNI_VERSION_1_8;
 107 }
 108 #endif
 109 jint Agent_Initialize(JavaVM * jvm, char *options, void *reserved) {
 110     jint res;
 111 
 112     if (options && strlen(options) > 0) {
 113         if (strstr(options, "printdump")) {
 114             printdump = 1;
 115         }
 116     }
 117 
 118     res = jvm->GetEnv((void **) &jvmti, JVMTI_VERSION_1_1);
 119     if (res < 0) {
 120         debug_printf("Wrong result of a valid call to GetEnv!\n");
 121         return JNI_ERR;
 122     }
 123 
 124 
 125     /* Add capabilities */
 126     res = jvmti->GetPotentialCapabilities(&jvmti_caps);
 127     JVMTI_ERROR_CHECK("GetPotentialCapabilities returned error", res);
 128 
 129     res = jvmti->AddCapabilities(&jvmti_caps);
 130     JVMTI_ERROR_CHECK("GetPotentialCapabilities returned error", res);
 131 
 132     /* Enable events */
 133     init_callbacks();
 134     res = jvmti->SetEventCallbacks(&callbacks, sizeof(callbacks));
 135     JVMTI_ERROR_CHECK("SetEventCallbacks returned error", res);
 136 
 137     res = jvmti->SetEventNotificationMode(JVMTI_ENABLE, JVMTI_EVENT_VM_INIT, NULL);
 138     JVMTI_ERROR_CHECK("SetEventNotificationMode for VM_INIT returned error", res);
 139 
 140     res = jvmti->SetEventNotificationMode(JVMTI_ENABLE, JVMTI_EVENT_VM_DEATH, NULL);
 141     JVMTI_ERROR_CHECK("SetEventNotificationMode for vm death event returned error", res);
 142 
 143     return JNI_OK;
 144 }
 145 
 146 
 147 JNIEXPORT jint JNICALL
 148 Java_nsk_jvmti_unit_functions_nosuspendMonitorInfo_JvmtiTest_GetResult(JNIEnv * env, jclass cls) {
 149     return iGlobalStatus;
 150 }
 151 
 152 JNIEXPORT void JNICALL
 153 Java_nsk_jvmti_unit_functions_nosuspendMonitorInfo_JvmtiTest_CheckMonitorInfo(JNIEnv * env, jclass cls, jthread thr, jobject oobj, jint expected_count) {
 154 
 155   jvmtiError ret;
 156   jint count;
 157   jobject *owned_monitor;
 158 
 159   debug_printf(" jvmti GetMonitorInfo \n");
 160 
 161 
 162     ret = jvmti->GetOwnedMonitorInfo(thr, &count, &owned_monitor);
 163     if (ret != JVMTI_ERROR_NONE) {
 164         printf("Error: GetMonitorInfo %d \n", ret);
 165         iGlobalStatus = 2;
 166     }
 167 
 168     if (expected_count != 0 && expected_count != count) {
 169         printf("Error: GetMonitorInfo expected count %d but got %d \n", expected_count, count);
 170         iGlobalStatus = 2;
 171     }
 172 
 173     if (expected_count !=0) {
 174 
 175         ret = jvmti->GetCurrentContendedMonitor(thr, owned_monitor);
 176         if (ret != JVMTI_ERROR_NONE) {
 177             printf("Error: GetContendedMonitorInfo %d \n", ret);
 178             iGlobalStatus = 2;
 179         }
 180     }
 181 
 182 }
 183 
 184 }