blob: abb6083de423ea7860b9aedccfcd34e236a24f5b [file] [log] [blame]
Andreas Gampe27fa96c2016-10-07 15:05:24 -07001/*
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
Andreas Gampe8cf9cb32017-07-19 09:28:38 -070017#include <pthread.h>
18
19#include <cstdio>
Andreas Gampe27fa96c2016-10-07 15:05:24 -070020#include <iostream>
Igor Murashkin5573c372017-11-16 13:34:30 -080021#include <mutex>
Andreas Gampe27fa96c2016-10-07 15:05:24 -070022#include <vector>
23
Andreas Gampe027444b2017-03-31 12:49:07 -070024#include "android-base/logging.h"
Andreas Gampe115b4982017-04-12 19:06:12 -070025#include "android-base/stringprintf.h"
Andreas Gampe27fa96c2016-10-07 15:05:24 -070026#include "jni.h"
Andreas Gampe5e03a302017-03-13 13:10:00 -070027#include "jvmti.h"
Andreas Gampe027444b2017-03-31 12:49:07 -070028#include "scoped_local_ref.h"
29#include "scoped_utf_chars.h"
Andreas Gampe27fa96c2016-10-07 15:05:24 -070030
Andreas Gampe3f46c962017-03-30 10:26:59 -070031// Test infrastructure
32#include "jvmti_helper.h"
33#include "test_env.h"
34
Andreas Gampe27fa96c2016-10-07 15:05:24 -070035namespace art {
36namespace Test904ObjectAllocation {
37
Alex Lightd4102ba2018-04-30 12:47:22 -070038static JavaVM* vm;
39
Andreas Gampe27fa96c2016-10-07 15:05:24 -070040static std::string GetClassName(JNIEnv* jni_env, jclass cls) {
41 ScopedLocalRef<jclass> class_class(jni_env, jni_env->GetObjectClass(cls));
42 jmethodID mid = jni_env->GetMethodID(class_class.get(), "getName", "()Ljava/lang/String;");
43 ScopedLocalRef<jstring> str(
44 jni_env, reinterpret_cast<jstring>(jni_env->CallObjectMethod(cls, mid)));
45 ScopedUtfChars utf_chars(jni_env, str.get());
46 return utf_chars.c_str();
47}
48
Alex Lightd4102ba2018-04-30 12:47:22 -070049template <typename T>
50class ScopedGlobalRef {
51 public:
52 ScopedGlobalRef(JNIEnv* env, T obj) : obj_(env->NewGlobalRef(obj)) {}
53 ScopedGlobalRef(const ScopedGlobalRef<T>& src) noexcept
54 : obj_(GetEnv()->NewGlobalRef(src.obj_)) {}
55 ScopedGlobalRef(ScopedGlobalRef<T>&& src) noexcept : obj_(src.obj_) {
56 src.obj_ = nullptr;
57 }
58
59 ~ScopedGlobalRef() {
60 GetEnv()->DeleteGlobalRef(obj_);
61 }
62
63 T Get(JNIEnv* env) const {
Alex Light9d3b5ec2019-06-20 09:47:01 -070064 return reinterpret_cast<T>(env->NewLocalRef(obj_));
Alex Lightd4102ba2018-04-30 12:47:22 -070065 }
66
67 private:
68 JNIEnv* GetEnv() const {
69 JNIEnv* env = nullptr;
70 CHECK_EQ(vm->GetEnv(reinterpret_cast<void**>(&env), JNI_VERSION_1_6), 0);
71 return env;
72 }
73
74 jobject obj_;
75};
76
77struct EventLog {
Alex Light9d3b5ec2019-06-20 09:47:01 -070078 ScopedGlobalRef<jclass> object_klass;
79 ScopedGlobalRef<jclass> object_klass2;
80 jlong size;
Alex Lightd4102ba2018-04-30 12:47:22 -070081 ScopedGlobalRef<jthread> thr_;
82};
83
Andreas Gampe115b4982017-04-12 19:06:12 -070084static std::mutex gEventsMutex;
Alex Lightd4102ba2018-04-30 12:47:22 -070085static std::vector<EventLog> gEvents;
Andreas Gampe115b4982017-04-12 19:06:12 -070086
Andreas Gampe27fa96c2016-10-07 15:05:24 -070087static void JNICALL ObjectAllocated(jvmtiEnv* ti_env ATTRIBUTE_UNUSED,
88 JNIEnv* jni_env,
Alex Lightd4102ba2018-04-30 12:47:22 -070089 jthread thread,
Andreas Gampe27fa96c2016-10-07 15:05:24 -070090 jobject object,
91 jclass object_klass,
92 jlong size) {
Andreas Gampe27fa96c2016-10-07 15:05:24 -070093 ScopedLocalRef<jclass> object_klass2(jni_env, jni_env->GetObjectClass(object));
Andreas Gampe115b4982017-04-12 19:06:12 -070094 std::lock_guard<std::mutex> guard(gEventsMutex);
Alex Light9d3b5ec2019-06-20 09:47:01 -070095 gEvents.push_back({ScopedGlobalRef<jclass>(jni_env, object_klass),
96 ScopedGlobalRef<jclass>(jni_env, object_klass2.get()),
97 size,
Alex Lightd4102ba2018-04-30 12:47:22 -070098 ScopedGlobalRef<jthread>(jni_env, thread)});
Andreas Gampe27fa96c2016-10-07 15:05:24 -070099}
100
Andreas Gampe46651672017-04-07 09:00:04 -0700101extern "C" JNIEXPORT void JNICALL Java_art_Test904_setupObjectAllocCallback(
Andreas Gampe3f46c962017-03-30 10:26:59 -0700102 JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jboolean enable) {
Alex Lightd4102ba2018-04-30 12:47:22 -0700103 env->GetJavaVM(&vm);
Andreas Gampe27fa96c2016-10-07 15:05:24 -0700104 jvmtiEventCallbacks callbacks;
105 memset(&callbacks, 0, sizeof(jvmtiEventCallbacks));
Andreas Gampe2427aae2016-10-17 18:05:19 -0700106 callbacks.VMObjectAlloc = enable ? ObjectAllocated : nullptr;
Andreas Gampe27fa96c2016-10-07 15:05:24 -0700107
108 jvmtiError ret = jvmti_env->SetEventCallbacks(&callbacks, sizeof(callbacks));
Andreas Gampe3f46c962017-03-30 10:26:59 -0700109 JvmtiErrorToException(env, jvmti_env, ret);
Andreas Gampe27fa96c2016-10-07 15:05:24 -0700110}
111
Andreas Gampe46651672017-04-07 09:00:04 -0700112extern "C" JNIEXPORT void JNICALL Java_art_Test904_enableAllocationTracking(
113 JNIEnv* env, jclass, jthread thread, jboolean enable) {
Andreas Gampe27fa96c2016-10-07 15:05:24 -0700114 jvmtiError ret = jvmti_env->SetEventNotificationMode(
115 enable ? JVMTI_ENABLE : JVMTI_DISABLE,
116 JVMTI_EVENT_VM_OBJECT_ALLOC,
117 thread);
Andreas Gampe3f46c962017-03-30 10:26:59 -0700118 JvmtiErrorToException(env, jvmti_env, ret);
Andreas Gampe27fa96c2016-10-07 15:05:24 -0700119}
120
Andreas Gampe115b4982017-04-12 19:06:12 -0700121extern "C" JNIEXPORT jobjectArray JNICALL Java_art_Test904_getTrackingEventMessages(
Alex Lightd4102ba2018-04-30 12:47:22 -0700122 JNIEnv* env, jclass Main_klass ATTRIBUTE_UNUSED, jobjectArray threads) {
Andreas Gampe115b4982017-04-12 19:06:12 -0700123 std::lock_guard<std::mutex> guard(gEventsMutex);
Alex Lightd4102ba2018-04-30 12:47:22 -0700124 std::vector<std::string> real_events;
125 std::vector<jthread> thread_lst;
126 jint nthreads = env->GetArrayLength(threads);
127 {
128 env->PushLocalFrame(nthreads + 1);
129 for (jint i = 0; i < nthreads; i++) {
130 thread_lst.push_back(reinterpret_cast<jthread>(env->GetObjectArrayElement(threads, i)));
131 }
132 for (const EventLog& ev : gEvents) {
133 ScopedLocalRef<jthread> thr(env, ev.thr_.Get(env));
134 for (jthread req_thread : thread_lst) {
135 if (env->IsSameObject(req_thread, thr.get())) {
Alex Light9d3b5ec2019-06-20 09:47:01 -0700136 ScopedLocalRef<jclass> klass(env, ev.object_klass.Get(env));
137 ScopedLocalRef<jclass> klass2(env, ev.object_klass2.Get(env));
138 std::string object_klass_descriptor = GetClassName(env, klass.get());
139 std::string object_klass_descriptor2 = GetClassName(env, klass2.get());
140 std::string res(android::base::StringPrintf("ObjectAllocated type %s/%s size %zu",
141 object_klass_descriptor.c_str(),
142 object_klass_descriptor2.c_str(),
143 static_cast<size_t>(ev.size)));
144 real_events.push_back(res);
Alex Lightd4102ba2018-04-30 12:47:22 -0700145 break;
146 }
147 }
148 }
149 env->PopLocalFrame(nullptr);
150 }
Andreas Gampe115b4982017-04-12 19:06:12 -0700151 jobjectArray ret = CreateObjectArray(env,
Alex Lightd4102ba2018-04-30 12:47:22 -0700152 static_cast<jint>(real_events.size()),
Andreas Gampe115b4982017-04-12 19:06:12 -0700153 "java/lang/String",
154 [&](jint i) {
Alex Lightd4102ba2018-04-30 12:47:22 -0700155 return env->NewStringUTF(real_events[i].c_str());
Andreas Gampe115b4982017-04-12 19:06:12 -0700156 });
157 gEvents.clear();
158 return ret;
159}
160
Andreas Gampe27fa96c2016-10-07 15:05:24 -0700161} // namespace Test904ObjectAllocation
162} // namespace art