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 <iostream> 18 #include <mutex> 19 #include <pthread.h> 20 #include <stdio.h> 21 #include <vector> 22 23 #include "android-base/logging.h" 24 #include "android-base/stringprintf.h" 25 #include "jni.h" 26 #include "jvmti.h" 27 #include "scoped_local_ref.h" 28 #include "scoped_utf_chars.h" 29 30 // Test infrastructure 31 #include "jvmti_helper.h" 32 #include "test_env.h" 33 34 namespace art { 35 namespace Test904ObjectAllocation { 36 37 static std::string GetClassName(JNIEnv* jni_env, jclass cls) { 38 ScopedLocalRef<jclass> class_class(jni_env, jni_env->GetObjectClass(cls)); 39 jmethodID mid = jni_env->GetMethodID(class_class.get(), "getName", "()Ljava/lang/String;"); 40 ScopedLocalRef<jstring> str( 41 jni_env, reinterpret_cast<jstring>(jni_env->CallObjectMethod(cls, mid))); 42 ScopedUtfChars utf_chars(jni_env, str.get()); 43 return utf_chars.c_str(); 44 } 45 46 static std::mutex gEventsMutex; 47 static std::vector<std::string> gEvents; 48 49 static void JNICALL ObjectAllocated(jvmtiEnv* ti_env ATTRIBUTE_UNUSED, 50 JNIEnv* jni_env, 51 jthread thread ATTRIBUTE_UNUSED, 52 jobject object, 53 jclass object_klass, 54 jlong size) { 55 std::string object_klass_descriptor = GetClassName(jni_env, object_klass); 56 ScopedLocalRef<jclass> object_klass2(jni_env, jni_env->GetObjectClass(object)); 57 std::string object_klass_descriptor2 = GetClassName(jni_env, object_klass2.get()); 58 59 std::lock_guard<std::mutex> guard(gEventsMutex); 60 gEvents.push_back(android::base::StringPrintf("ObjectAllocated type %s/%s size %zu", 61 object_klass_descriptor.c_str(), 62 object_klass_descriptor2.c_str(), 63 static_cast<size_t>(size))); 64 } 65 66 extern "C" JNIEXPORT void JNICALL Java_art_Test904_setupObjectAllocCallback( 67 JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jboolean enable) { 68 jvmtiEventCallbacks callbacks; 69 memset(&callbacks, 0, sizeof(jvmtiEventCallbacks)); 70 callbacks.VMObjectAlloc = enable ? ObjectAllocated : nullptr; 71 72 jvmtiError ret = jvmti_env->SetEventCallbacks(&callbacks, sizeof(callbacks)); 73 JvmtiErrorToException(env, jvmti_env, ret); 74 } 75 76 extern "C" JNIEXPORT void JNICALL Java_art_Test904_enableAllocationTracking( 77 JNIEnv* env, jclass, jthread thread, jboolean enable) { 78 jvmtiError ret = jvmti_env->SetEventNotificationMode( 79 enable ? JVMTI_ENABLE : JVMTI_DISABLE, 80 JVMTI_EVENT_VM_OBJECT_ALLOC, 81 thread); 82 JvmtiErrorToException(env, jvmti_env, ret); 83 } 84 85 extern "C" JNIEXPORT jobjectArray JNICALL Java_art_Test904_getTrackingEventMessages( 86 JNIEnv* env, jclass Main_klass ATTRIBUTE_UNUSED) { 87 std::lock_guard<std::mutex> guard(gEventsMutex); 88 jobjectArray ret = CreateObjectArray(env, 89 static_cast<jint>(gEvents.size()), 90 "java/lang/String", 91 [&](jint i) { 92 return env->NewStringUTF(gEvents[i].c_str()); 93 }); 94 gEvents.clear(); 95 return ret; 96 } 97 98 } // namespace Test904ObjectAllocation 99 } // namespace art 100