aosp12/art/test/ti-agent/stack_trace_helper.cc

100 lines
3.4 KiB
C++

/*
* Copyright (C) 2017 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 "jni.h"
#include "jvmti.h"
#include "jvmti_helper.h"
#include "scoped_local_ref.h"
#include "test_env.h"
namespace art {
namespace common_stack_trace {
extern "C" JNIEXPORT jint JNICALL Java_art_StackTrace_GetStackDepth(
JNIEnv* env, jclass, jthread thr) {
jint ret;
JvmtiErrorToException(env, jvmti_env, jvmti_env->GetFrameCount(thr, &ret));
return ret;
}
extern "C" JNIEXPORT jobjectArray Java_art_StackTrace_nativeGetStackTrace(JNIEnv* env,
jclass,
jthread thr) {
jint depth;
ScopedLocalRef<jclass> klass(env, env->FindClass("art/StackTrace$StackFrameData"));
if (env->ExceptionCheck()) {
return nullptr;
}
jmethodID constructor = env->GetMethodID(
klass.get(), "<init>", "(Ljava/lang/Thread;Ljava/lang/reflect/Executable;JI)V");
if (env->ExceptionCheck()) {
return nullptr;
}
if (JvmtiErrorToException(env, jvmti_env, jvmti_env->GetFrameCount(thr, &depth))) {
return nullptr;
}
// Just give some extra space.
depth += 10;
jvmtiFrameInfo* frames;
if (JvmtiErrorToException(
env, jvmti_env, jvmti_env->Allocate(depth * sizeof(jvmtiFrameInfo),
reinterpret_cast<unsigned char**>(&frames)))) {
return nullptr;
}
jint nframes = 0;
if (JvmtiErrorToException(
env, jvmti_env, jvmti_env->GetStackTrace(thr, 0, depth, frames, &nframes))) {
jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(frames));
return nullptr;
}
jobjectArray frames_array = env->NewObjectArray(nframes, klass.get(), nullptr);
if (env->ExceptionCheck()) {
jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(frames));
return nullptr;
}
for (jint i = 0; i < nframes; i++) {
jobject jmethod = GetJavaMethod(jvmti_env, env, frames[i].method);
if (env->ExceptionCheck()) {
jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(frames));
return nullptr;
}
jobject frame_obj = env->NewObject(klass.get(),
constructor,
thr,
jmethod,
frames[i].location,
i);
if (env->ExceptionCheck()) {
jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(frames));
return nullptr;
}
env->SetObjectArrayElement(frames_array, i, frame_obj);
if (env->ExceptionCheck()) {
jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(frames));
return nullptr;
}
}
jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(frames));
return frames_array;
}
} // namespace common_stack_trace
} // namespace art