| /* |
| * Copyright (C) 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "common_helper.h" |
| |
| #include <deque> |
| #include <map> |
| #include <stdio.h> |
| #include <sstream> |
| #include <string> |
| #include <vector> |
| |
| #include "jni.h" |
| #include "jvmti.h" |
| |
| #include "jvmti_helper.h" |
| #include "test_env.h" |
| |
| namespace art { |
| |
| static void SetupCommonRedefine(); |
| static void SetupCommonRetransform(); |
| static void SetupCommonTransform(); |
| template <bool is_redefine> |
| static void throwCommonRedefinitionError(jvmtiEnv* jvmti, |
| JNIEnv* env, |
| jint num_targets, |
| jclass* target, |
| jvmtiError res) { |
| std::stringstream err; |
| char* error = nullptr; |
| jvmti->GetErrorName(res, &error); |
| err << "Failed to " << (is_redefine ? "redefine" : "retransform") << " class"; |
| if (num_targets > 1) { |
| err << "es"; |
| } |
| err << " <"; |
| for (jint i = 0; i < num_targets; i++) { |
| char* signature = nullptr; |
| char* generic = nullptr; |
| jvmti->GetClassSignature(target[i], &signature, &generic); |
| if (i != 0) { |
| err << ", "; |
| } |
| err << signature; |
| jvmti->Deallocate(reinterpret_cast<unsigned char*>(signature)); |
| jvmti->Deallocate(reinterpret_cast<unsigned char*>(generic)); |
| } |
| err << "> due to " << error; |
| std::string message = err.str(); |
| jvmti->Deallocate(reinterpret_cast<unsigned char*>(error)); |
| env->ThrowNew(env->FindClass("java/lang/Exception"), message.c_str()); |
| } |
| |
| #define CONFIGURATION_COMMON_REDEFINE 0 |
| #define CONFIGURATION_COMMON_RETRANSFORM 1 |
| #define CONFIGURATION_COMMON_TRANSFORM 2 |
| |
| extern "C" JNIEXPORT void JNICALL Java_art_Redefinition_nativeSetTestConfiguration(JNIEnv*, |
| jclass, |
| jint type) { |
| switch (type) { |
| case CONFIGURATION_COMMON_REDEFINE: { |
| SetupCommonRedefine(); |
| return; |
| } |
| case CONFIGURATION_COMMON_RETRANSFORM: { |
| SetupCommonRetransform(); |
| return; |
| } |
| case CONFIGURATION_COMMON_TRANSFORM: { |
| SetupCommonTransform(); |
| return; |
| } |
| default: { |
| LOG(FATAL) << "Unknown test configuration: " << type; |
| } |
| } |
| } |
| |
| namespace common_redefine { |
| |
| static void throwRedefinitionError(jvmtiEnv* jvmti, |
| JNIEnv* env, |
| jint num_targets, |
| jclass* target, |
| jvmtiError res) { |
| return throwCommonRedefinitionError<true>(jvmti, env, num_targets, target, res); |
| } |
| |
| static void DoMultiClassRedefine(jvmtiEnv* jvmti_env, |
| JNIEnv* env, |
| jint num_redefines, |
| jclass* targets, |
| jbyteArray* class_file_bytes, |
| jbyteArray* dex_file_bytes) { |
| std::vector<jvmtiClassDefinition> defs; |
| for (jint i = 0; i < num_redefines; i++) { |
| jbyteArray desired_array = IsJVM() ? class_file_bytes[i] : dex_file_bytes[i]; |
| jint len = static_cast<jint>(env->GetArrayLength(desired_array)); |
| const unsigned char* redef_bytes = reinterpret_cast<const unsigned char*>( |
| env->GetByteArrayElements(desired_array, nullptr)); |
| defs.push_back({targets[i], static_cast<jint>(len), redef_bytes}); |
| } |
| jvmtiError res = jvmti_env->RedefineClasses(num_redefines, defs.data()); |
| if (res != JVMTI_ERROR_NONE) { |
| throwRedefinitionError(jvmti_env, env, num_redefines, targets, res); |
| } |
| } |
| |
| static void DoClassRedefine(jvmtiEnv* jvmti_env, |
| JNIEnv* env, |
| jclass target, |
| jbyteArray class_file_bytes, |
| jbyteArray dex_file_bytes) { |
| return DoMultiClassRedefine(jvmti_env, env, 1, &target, &class_file_bytes, &dex_file_bytes); |
| } |
| |
| // Magic JNI export that classes can use for redefining classes. |
| // To use classes should declare this as a native function with signature (Ljava/lang/Class;[B[B)V |
| extern "C" JNIEXPORT void JNICALL Java_art_Redefinition_doCommonClassRedefinition( |
| JNIEnv* env, jclass, jclass target, jbyteArray class_file_bytes, jbyteArray dex_file_bytes) { |
| DoClassRedefine(jvmti_env, env, target, class_file_bytes, dex_file_bytes); |
| } |
| |
| // Magic JNI export that classes can use for redefining classes. |
| // To use classes should declare this as a native function with signature |
| // ([Ljava/lang/Class;[[B[[B)V |
| extern "C" JNIEXPORT void JNICALL Java_art_Redefinition_doCommonMultiClassRedefinition( |
| JNIEnv* env, |
| jclass, |
| jobjectArray targets, |
| jobjectArray class_file_bytes, |
| jobjectArray dex_file_bytes) { |
| std::vector<jclass> classes; |
| std::vector<jbyteArray> class_files; |
| std::vector<jbyteArray> dex_files; |
| jint len = env->GetArrayLength(targets); |
| if (len != env->GetArrayLength(class_file_bytes) || len != env->GetArrayLength(dex_file_bytes)) { |
| env->ThrowNew(env->FindClass("java/lang/IllegalArgumentException"), |
| "the three array arguments passed to this function have different lengths!"); |
| return; |
| } |
| for (jint i = 0; i < len; i++) { |
| classes.push_back(static_cast<jclass>(env->GetObjectArrayElement(targets, i))); |
| dex_files.push_back(static_cast<jbyteArray>(env->GetObjectArrayElement(dex_file_bytes, i))); |
| class_files.push_back(static_cast<jbyteArray>(env->GetObjectArrayElement(class_file_bytes, i))); |
| } |
| return DoMultiClassRedefine(jvmti_env, |
| env, |
| len, |
| classes.data(), |
| class_files.data(), |
| dex_files.data()); |
| } |
| |
| // Get all capabilities except those related to retransformation. |
| jint OnLoad(JavaVM* vm, |
| char* options ATTRIBUTE_UNUSED, |
| void* reserved ATTRIBUTE_UNUSED) { |
| if (vm->GetEnv(reinterpret_cast<void**>(&jvmti_env), JVMTI_VERSION_1_0)) { |
| printf("Unable to get jvmti env!\n"); |
| return 1; |
| } |
| SetupCommonRedefine(); |
| return 0; |
| } |
| |
| } // namespace common_redefine |
| |
| namespace common_retransform { |
| |
| struct CommonTransformationResult { |
| std::vector<unsigned char> class_bytes; |
| std::vector<unsigned char> dex_bytes; |
| |
| CommonTransformationResult(size_t class_size, size_t dex_size) |
| : class_bytes(class_size), dex_bytes(dex_size) {} |
| |
| CommonTransformationResult() = default; |
| CommonTransformationResult(CommonTransformationResult&&) = default; |
| CommonTransformationResult(CommonTransformationResult&) = default; |
| }; |
| |
| // Map from class name to transformation result. |
| std::map<std::string, std::deque<CommonTransformationResult>> gTransformations; |
| bool gPopTransformations = true; |
| |
| extern "C" JNIEXPORT void JNICALL Java_art_Redefinition_addCommonTransformationResult( |
| JNIEnv* env, jclass, jstring class_name, jbyteArray class_array, jbyteArray dex_array) { |
| const char* name_chrs = env->GetStringUTFChars(class_name, nullptr); |
| std::string name_str(name_chrs); |
| env->ReleaseStringUTFChars(class_name, name_chrs); |
| CommonTransformationResult trans(env->GetArrayLength(class_array), |
| env->GetArrayLength(dex_array)); |
| if (env->ExceptionOccurred()) { |
| return; |
| } |
| env->GetByteArrayRegion(class_array, |
| 0, |
| env->GetArrayLength(class_array), |
| reinterpret_cast<jbyte*>(trans.class_bytes.data())); |
| if (env->ExceptionOccurred()) { |
| return; |
| } |
| env->GetByteArrayRegion(dex_array, |
| 0, |
| env->GetArrayLength(dex_array), |
| reinterpret_cast<jbyte*>(trans.dex_bytes.data())); |
| if (env->ExceptionOccurred()) { |
| return; |
| } |
| if (gTransformations.find(name_str) == gTransformations.end()) { |
| std::deque<CommonTransformationResult> list; |
| gTransformations[name_str] = std::move(list); |
| } |
| gTransformations[name_str].push_back(std::move(trans)); |
| } |
| |
| // The hook we are using. |
| void JNICALL CommonClassFileLoadHookRetransformable(jvmtiEnv* jvmti_env, |
| JNIEnv* jni_env ATTRIBUTE_UNUSED, |
| jclass class_being_redefined ATTRIBUTE_UNUSED, |
| jobject loader ATTRIBUTE_UNUSED, |
| const char* name, |
| jobject protection_domain ATTRIBUTE_UNUSED, |
| jint class_data_len ATTRIBUTE_UNUSED, |
| const unsigned char* class_dat ATTRIBUTE_UNUSED, |
| jint* new_class_data_len, |
| unsigned char** new_class_data) { |
| std::string name_str(name); |
| if (gTransformations.find(name_str) != gTransformations.end() && |
| gTransformations[name_str].size() > 0) { |
| CommonTransformationResult& res = gTransformations[name_str][0]; |
| const std::vector<unsigned char>& desired_array = IsJVM() ? res.class_bytes : res.dex_bytes; |
| unsigned char* new_data; |
| CHECK_EQ(JVMTI_ERROR_NONE, jvmti_env->Allocate(desired_array.size(), &new_data)); |
| memcpy(new_data, desired_array.data(), desired_array.size()); |
| *new_class_data = new_data; |
| *new_class_data_len = desired_array.size(); |
| if (gPopTransformations) { |
| gTransformations[name_str].pop_front(); |
| } |
| } |
| } |
| |
| extern "C" JNIEXPORT void Java_art_Redefinition_setPopRetransformations(JNIEnv*, |
| jclass, |
| jboolean enable) { |
| gPopTransformations = enable; |
| } |
| |
| extern "C" JNIEXPORT void Java_art_Redefinition_popTransformationFor(JNIEnv* env, |
| jclass, |
| jstring class_name) { |
| const char* name_chrs = env->GetStringUTFChars(class_name, nullptr); |
| std::string name_str(name_chrs); |
| env->ReleaseStringUTFChars(class_name, name_chrs); |
| if (gTransformations.find(name_str) != gTransformations.end() && |
| gTransformations[name_str].size() > 0) { |
| gTransformations[name_str].pop_front(); |
| } else { |
| std::stringstream err; |
| err << "No transformations found for class " << name_str; |
| std::string message = err.str(); |
| env->ThrowNew(env->FindClass("java/lang/Exception"), message.c_str()); |
| } |
| } |
| |
| extern "C" JNIEXPORT void Java_art_Redefinition_enableCommonRetransformation(JNIEnv* env, |
| jclass, |
| jboolean enable) { |
| jvmtiError res = jvmti_env->SetEventNotificationMode(enable ? JVMTI_ENABLE : JVMTI_DISABLE, |
| JVMTI_EVENT_CLASS_FILE_LOAD_HOOK, |
| nullptr); |
| if (res != JVMTI_ERROR_NONE) { |
| JvmtiErrorToException(env, jvmti_env, res); |
| } |
| } |
| |
| static void throwRetransformationError(jvmtiEnv* jvmti, |
| JNIEnv* env, |
| jint num_targets, |
| jclass* targets, |
| jvmtiError res) { |
| return throwCommonRedefinitionError<false>(jvmti, env, num_targets, targets, res); |
| } |
| |
| static void DoClassRetransformation(jvmtiEnv* jvmti_env, JNIEnv* env, jobjectArray targets) { |
| std::vector<jclass> classes; |
| jint len = env->GetArrayLength(targets); |
| for (jint i = 0; i < len; i++) { |
| classes.push_back(static_cast<jclass>(env->GetObjectArrayElement(targets, i))); |
| } |
| jvmtiError res = jvmti_env->RetransformClasses(len, classes.data()); |
| if (res != JVMTI_ERROR_NONE) { |
| throwRetransformationError(jvmti_env, env, len, classes.data(), res); |
| } |
| } |
| |
| extern "C" JNIEXPORT void JNICALL Java_art_Redefinition_doCommonClassRetransformation( |
| JNIEnv* env, jclass, jobjectArray targets) { |
| jvmtiCapabilities caps; |
| jvmtiError caps_err = jvmti_env->GetCapabilities(&caps); |
| if (caps_err != JVMTI_ERROR_NONE) { |
| env->ThrowNew(env->FindClass("java/lang/Exception"), |
| "Unable to get current jvmtiEnv capabilities"); |
| return; |
| } |
| |
| // Allocate a new environment if we don't have the can_retransform_classes capability needed to |
| // call the RetransformClasses function. |
| jvmtiEnv* real_env = nullptr; |
| if (caps.can_retransform_classes != 1) { |
| JavaVM* vm = nullptr; |
| if (env->GetJavaVM(&vm) != 0 || |
| vm->GetEnv(reinterpret_cast<void**>(&real_env), JVMTI_VERSION_1_0) != 0) { |
| env->ThrowNew(env->FindClass("java/lang/Exception"), |
| "Unable to create temporary jvmtiEnv for RetransformClasses call."); |
| return; |
| } |
| SetStandardCapabilities(real_env); |
| } else { |
| real_env = jvmti_env; |
| } |
| DoClassRetransformation(real_env, env, targets); |
| if (caps.can_retransform_classes != 1) { |
| real_env->DisposeEnvironment(); |
| } |
| } |
| |
| // Get all capabilities except those related to retransformation. |
| jint OnLoad(JavaVM* vm, |
| char* options ATTRIBUTE_UNUSED, |
| void* reserved ATTRIBUTE_UNUSED) { |
| if (vm->GetEnv(reinterpret_cast<void**>(&jvmti_env), JVMTI_VERSION_1_0)) { |
| printf("Unable to get jvmti env!\n"); |
| return 1; |
| } |
| SetupCommonRetransform(); |
| return 0; |
| } |
| |
| } // namespace common_retransform |
| |
| namespace common_transform { |
| |
| // Get all capabilities except those related to retransformation. |
| jint OnLoad(JavaVM* vm, |
| char* options ATTRIBUTE_UNUSED, |
| void* reserved ATTRIBUTE_UNUSED) { |
| if (vm->GetEnv(reinterpret_cast<void**>(&jvmti_env), JVMTI_VERSION_1_0)) { |
| printf("Unable to get jvmti env!\n"); |
| return 1; |
| } |
| SetupCommonTransform(); |
| return 0; |
| } |
| |
| } // namespace common_transform |
| |
| static void SetupCommonRedefine() { |
| jvmtiCapabilities caps = GetStandardCapabilities(); |
| caps.can_retransform_classes = 0; |
| caps.can_retransform_any_class = 0; |
| jvmti_env->AddCapabilities(&caps); |
| } |
| |
| static void SetupCommonRetransform() { |
| SetStandardCapabilities(jvmti_env); |
| jvmtiEventCallbacks cb; |
| memset(&cb, 0, sizeof(cb)); |
| cb.ClassFileLoadHook = common_retransform::CommonClassFileLoadHookRetransformable; |
| jvmtiError res = jvmti_env->SetEventCallbacks(&cb, sizeof(cb)); |
| CHECK_EQ(res, JVMTI_ERROR_NONE); |
| common_retransform::gTransformations.clear(); |
| } |
| |
| static void SetupCommonTransform() { |
| // Don't set the retransform caps |
| jvmtiCapabilities caps = GetStandardCapabilities(); |
| caps.can_retransform_classes = 0; |
| caps.can_retransform_any_class = 0; |
| jvmti_env->AddCapabilities(&caps); |
| |
| // Use the same callback as the retransform test. |
| jvmtiEventCallbacks cb; |
| memset(&cb, 0, sizeof(cb)); |
| cb.ClassFileLoadHook = common_retransform::CommonClassFileLoadHookRetransformable; |
| jvmtiError res = jvmti_env->SetEventCallbacks(&cb, sizeof(cb)); |
| CHECK_EQ(res, JVMTI_ERROR_NONE); |
| common_retransform::gTransformations.clear(); |
| } |
| |
| } // namespace art |