| /* |
| * 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 <stdlib.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; } |
| |
| static jvmtiEnv *jvmti; |
| static jint iGlobalStatus = 0; |
| static jvmtiCapabilities jvmti_caps; |
| static jvmtiEventCallbacks callbacks; |
| static int boot_class_count = 0; |
| |
| static const char* BOOT_CLASS = |
| "nsk/jvmti/unit/functions/AddToBootstrapClassLoaderSearch/Boot"; |
| |
| static const char* CLASSPATH = "java.class.path"; |
| |
| static char segment[3000] = "."; |
| |
| int printdump = 0; |
| |
| |
| void debug_printf(char *fmt, ...) { |
| va_list args; |
| |
| va_start(args, fmt); |
| if (printdump) { |
| vprintf(fmt, args); |
| } |
| va_end(args); |
| } |
| |
| |
| /* |
| * Check that it is not possible to add to the boot class path during the Start phase |
| */ |
| void JNICALL |
| vmStart(jvmtiEnv *jvmti, JNIEnv* jni) { |
| jvmtiError res; |
| |
| debug_printf("VMStart event done\n"); |
| |
| res = JVMTI_ENV_PTR(jvmti)->AddToBootstrapClassLoaderSearch(JVMTI_ENV_ARG(jvmti, segment)); |
| JVMTI_ERROR_CHECK_EXPECTED_ERROR_VOID("VMStart: AddToBootstrapClassLoaderSearch returned error ", |
| res, JVMTI_ERROR_WRONG_PHASE); |
| } |
| |
| /* |
| * Check that it is possible to add to the boot class path before VMDeath event return. |
| */ |
| void JNICALL |
| vmDeath(jvmtiEnv *jvmti, JNIEnv* jni) { |
| jvmtiError res; |
| |
| debug_printf("VMDeath event done\n"); |
| |
| res = JVMTI_ENV_PTR(jvmti)->AddToBootstrapClassLoaderSearch(JVMTI_ENV_ARG(jvmti, segment)); |
| /* In the live phase, anything other than an existing JAR file is an invalid path. |
| So, check that JVMTI_ERROR_ILLEGAL_ARGUMENT error is thrown. |
| */ |
| JVMTI_ERROR_CHECK_EXPECTED_ERROR_VOID("VMDeath: AddToBootstrapClassLoaderSearch returned error ", |
| res, JVMTI_ERROR_ILLEGAL_ARGUMENT); |
| } |
| |
| /* |
| * Check that it is possible to add to the boot class path during the Live phase |
| */ |
| void JNICALL vmInit(jvmtiEnv *jvmti_env, JNIEnv *env, jthread thread) { |
| jvmtiError res; |
| |
| debug_printf("VMInit event done\n"); |
| |
| res = JVMTI_ENV_PTR(jvmti)->AddToBootstrapClassLoaderSearch(JVMTI_ENV_ARG(jvmti, segment)); |
| /* In the live phase, anything other than an existing JAR file is an invalid path. |
| So, check that JVMTI_ERROR_ILLEGAL_ARGUMENT error is thrown. |
| */ |
| JVMTI_ERROR_CHECK_EXPECTED_ERROR_VOID("VMInit: AddToBootstrapClassLoaderSearch returned error ", |
| res, JVMTI_ERROR_ILLEGAL_ARGUMENT); |
| } |
| |
| /* |
| * Check that it is not possible to add to the boot class path during the Primordial phase |
| */ |
| void JNICALL |
| NativeMethodBind(jvmtiEnv* jvmti, JNIEnv *jni, |
| jthread thread, jmethodID method, |
| void* address, void** new_address_ptr) { |
| jvmtiPhase phase; |
| jvmtiError res; |
| |
| res = JVMTI_ENV_PTR(jvmti)->GetPhase(JVMTI_ENV_ARG(jvmti, &phase)); |
| JVMTI_ERROR_CHECK_VOID("GetPhase returned error", res); |
| |
| if (phase == JVMTI_PHASE_PRIMORDIAL) { |
| debug_printf("Primordial phase\n"); |
| |
| res = JVMTI_ENV_PTR(jvmti)->AddToBootstrapClassLoaderSearch(JVMTI_ENV_ARG(jvmti, segment)); |
| JVMTI_ERROR_CHECK_EXPECTED_ERROR_VOID("Primordial: AddToBootstrapClassLoaderSearch returned error ", |
| res, JVMTI_ERROR_WRONG_PHASE); |
| } |
| } |
| |
| |
| void JNICALL |
| classFileLoadEvent(jvmtiEnv *jvmti_env, JNIEnv *env, |
| jclass redefined_class, |
| jobject loader,const char* name, |
| jobject protection_domain, |
| jint class_data_len, |
| const unsigned char* class_data, |
| jint* new_class_data_len, |
| unsigned char** new_class_data) { |
| |
| if (name != NULL && (strcmp(name, BOOT_CLASS) == 0)) { |
| debug_printf("Received class file load hook event for class: \n\t%s\n", |
| name); |
| debug_printf("Received class loader: 0x%p \n", loader); |
| /* Check to make sure Boot class got loaded from bootstrap class path.*/ |
| |
| if (loader == NULL) { |
| boot_class_count++; |
| } |
| } |
| } |
| |
| |
| void init_callbacks() { |
| memset((void *)&callbacks, 0, sizeof(jvmtiEventCallbacks)); |
| |
| callbacks.VMInit = vmInit; |
| callbacks.VMStart = vmStart; |
| callbacks.VMDeath = vmDeath; |
| callbacks.NativeMethodBind = NativeMethodBind; |
| callbacks.ClassFileLoadHook = classFileLoadEvent; |
| } |
| |
| |
| #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; |
| char *idx; |
| |
| debug_printf("Agent_OnLoad event done\n"); |
| |
| if (options && strlen(options) > 0) { |
| if (strstr(options, "printdump")) { |
| printdump = 1; |
| } |
| |
| strncpy(segment, options, (size_t) sizeof(segment)/sizeof(char)); |
| segment[(size_t) sizeof(segment)/sizeof(char) - 1] = 0; |
| idx = strchr(segment, ','); |
| if (idx != NULL) *idx = 0; |
| } |
| |
| 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("GetAddCapabilities 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_START,NULL); |
| JVMTI_ERROR_CHECK("SetEventNotificationMode for VM_START 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_NATIVE_METHOD_BIND,NULL); |
| JVMTI_ERROR_CHECK("SetEventNotificationMode for NATIVE_METHOD_BIND 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 returned error", res); |
| |
| res = JVMTI_ENV_PTR(jvmti)->SetEventNotificationMode(JVMTI_ENV_ARG(jvmti,JVMTI_ENABLE),JVMTI_EVENT_CLASS_FILE_LOAD_HOOK,NULL); |
| JVMTI_ERROR_CHECK("SetEventNotificationMode CLASS_FILE_LOAD_HOOK returned error", res); |
| |
| strcat(segment, "/newclass"); |
| debug_printf("segment=%s\n", segment); |
| res = JVMTI_ENV_PTR(jvmti)->AddToBootstrapClassLoaderSearch(JVMTI_ENV_ARG(jvmti, segment)); |
| JVMTI_ERROR_CHECK("AddToBootStrapClassLoaderSearch returned error", res); |
| |
| return JNI_OK; |
| } |
| |
| JNIEXPORT jint JNICALL |
| Java_nsk_jvmti_unit_functions_AddToBootstrapClassLoaderSearch_JvmtiTest_GetResult(JNIEnv * env, jclass cls) { |
| |
| if (boot_class_count != 1) { |
| printf("Error: no ClassFileLoadHook event for Boot class loaded from bootstrap class path\n"); |
| iGlobalStatus = 2; |
| } |
| return iGlobalStatus; |
| } |
| |
| |
| #ifdef __cplusplus |
| } |
| #endif |