1 /* 2 * Copyright (C) 2013 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include <pthread.h> 18 19 #include <cstdio> 20 #include <iostream> 21 #include <mutex> 22 #include <vector> 23 24 #include "android-base/logging.h" 25 #include "android-base/stringprintf.h" 26 #include "jni.h" 27 #include "jvmti.h" 28 #include "scoped_local_ref.h" 29 #include "scoped_utf_chars.h" 30 31 // Test infrastructure 32 #include "jvmti_helper.h" 33 #include "test_env.h" 34 35 namespace art { 36 namespace Test904ObjectAllocation { 37 38 static std::string GetClassName(JNIEnv* jni_env, jclass cls) { 39 ScopedLocalRef<jclass> class_class(jni_env, jni_env->GetObjectClass(cls)); 40 jmethodID mid = jni_env->GetMethodID(class_class.get(), "getName", "()Ljava/lang/String;"); 41 ScopedLocalRef<jstring> str( 42 jni_env, reinterpret_cast<jstring>(jni_env->CallObjectMethod(cls, mid))); 43 ScopedUtfChars utf_chars(jni_env, str.get()); 44 return utf_chars.c_str(); 45 } 46 47 static std::mutex gEventsMutex; 48 static std::vector<std::string> gEvents; 49 50 static void JNICALL ObjectAllocated(jvmtiEnv* ti_env ATTRIBUTE_UNUSED, 51 JNIEnv* jni_env, 52 jthread thread ATTRIBUTE_UNUSED, 53 jobject object, 54 jclass object_klass, 55 jlong size) { 56 std::string object_klass_descriptor = GetClassName(jni_env, object_klass); 57 ScopedLocalRef<jclass> object_klass2(jni_env, jni_env->GetObjectClass(object)); 58 std::string object_klass_descriptor2 = GetClassName(jni_env, object_klass2.get()); 59 60 std::lock_guard<std::mutex> guard(gEventsMutex); 61 gEvents.push_back(android::base::StringPrintf("ObjectAllocated type %s/%s size %zu", 62 object_klass_descriptor.c_str(), 63 object_klass_descriptor2.c_str(), 64 static_cast<size_t>(size))); 65 } 66 67 extern "C" JNIEXPORT void JNICALL Java_art_Test904_setupObjectAllocCallback( 68 JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jboolean enable) { 69 jvmtiEventCallbacks callbacks; 70 memset(&callbacks, 0, sizeof(jvmtiEventCallbacks)); 71 callbacks.VMObjectAlloc = enable ? ObjectAllocated : nullptr; 72 73 jvmtiError ret = jvmti_env->SetEventCallbacks(&callbacks, sizeof(callbacks)); 74 JvmtiErrorToException(env, jvmti_env, ret); 75 } 76 77 extern "C" JNIEXPORT void JNICALL Java_art_Test904_enableAllocationTracking( 78 JNIEnv* env, jclass, jthread thread, jboolean enable) { 79 jvmtiError ret = jvmti_env->SetEventNotificationMode( 80 enable ? JVMTI_ENABLE : JVMTI_DISABLE, 81 JVMTI_EVENT_VM_OBJECT_ALLOC, 82 thread); 83 JvmtiErrorToException(env, jvmti_env, ret); 84 } 85 86 extern "C" JNIEXPORT jobjectArray JNICALL Java_art_Test904_getTrackingEventMessages( 87 JNIEnv* env, jclass Main_klass ATTRIBUTE_UNUSED) { 88 std::lock_guard<std::mutex> guard(gEventsMutex); 89 jobjectArray ret = CreateObjectArray(env, 90 static_cast<jint>(gEvents.size()), 91 "java/lang/String", 92 [&](jint i) { 93 return env->NewStringUTF(gEvents[i].c_str()); 94 }); 95 gEvents.clear(); 96 return ret; 97 } 98 99 } // namespace Test904ObjectAllocation 100 } // namespace art 101
