259 lines
		
	
	
		
			8.8 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			259 lines
		
	
	
		
			8.8 KiB
		
	
	
	
		
			C++
		
	
	
	
| /*
 | |
|  * Copyright (C) 2013 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 <inttypes.h>
 | |
| 
 | |
| #include <cstdio>
 | |
| #include <memory>
 | |
| 
 | |
| #include "android-base/logging.h"
 | |
| #include "android-base/stringprintf.h"
 | |
| 
 | |
| #include "jni.h"
 | |
| #include "jvmti.h"
 | |
| #include "scoped_local_ref.h"
 | |
| 
 | |
| // Test infrastructure
 | |
| #include "jni_binder.h"
 | |
| #include "jni_helper.h"
 | |
| #include "jvmti_helper.h"
 | |
| #include "test_env.h"
 | |
| #include "ti_macros.h"
 | |
| 
 | |
| namespace art {
 | |
| namespace Test911GetStackTrace {
 | |
| 
 | |
| using android::base::StringPrintf;
 | |
| 
 | |
| static jint FindLineNumber(jint line_number_count,
 | |
|                            jvmtiLineNumberEntry* line_number_table,
 | |
|                            jlocation location) {
 | |
|   if (line_number_table == nullptr) {
 | |
|     return -2;
 | |
|   }
 | |
| 
 | |
|   jint line_number = -1;
 | |
|   for (jint i = 0; i != line_number_count; ++i) {
 | |
|     if (line_number_table[i].start_location > location) {
 | |
|       return line_number;
 | |
|     }
 | |
|     line_number = line_number_table[i].line_number;
 | |
|   }
 | |
|   return line_number;
 | |
| }
 | |
| 
 | |
| static jobjectArray TranslateJvmtiFrameInfoArray(JNIEnv* env,
 | |
|                                                  jvmtiFrameInfo* frames,
 | |
|                                                  jint count) {
 | |
|   auto callback = [&](jint method_index) -> jobjectArray {
 | |
|     char* name;
 | |
|     char* sig;
 | |
|     char* gen;
 | |
|     {
 | |
|       jvmtiError result2 = jvmti_env->GetMethodName(frames[method_index].method, &name, &sig, &gen);
 | |
|       if (JvmtiErrorToException(env, jvmti_env, result2)) {
 | |
|         return nullptr;
 | |
|       }
 | |
|     }
 | |
| 
 | |
|     jint line_number_count;
 | |
|     jvmtiLineNumberEntry* line_number_table;
 | |
|     {
 | |
|       jvmtiError line_result = jvmti_env->GetLineNumberTable(frames[method_index].method,
 | |
|                                                              &line_number_count,
 | |
|                                                              &line_number_table);
 | |
|       if (line_result != JVMTI_ERROR_NONE) {
 | |
|         // Accept absent info and native method errors.
 | |
|         if (line_result != JVMTI_ERROR_ABSENT_INFORMATION &&
 | |
|             line_result != JVMTI_ERROR_NATIVE_METHOD) {
 | |
|           JvmtiErrorToException(env, jvmti_env, line_result);
 | |
|           return nullptr;
 | |
|         }
 | |
|         line_number_table = nullptr;
 | |
|         line_number_count = 0;
 | |
|       }
 | |
|     }
 | |
| 
 | |
|     auto inner_callback = [&](jint component_index) -> jstring {
 | |
|       switch (component_index) {
 | |
|         case 0:
 | |
|           return (name == nullptr) ? nullptr : env->NewStringUTF(name);
 | |
|         case 1:
 | |
|           return (sig == nullptr) ? nullptr : env->NewStringUTF(sig);
 | |
|         case 2:
 | |
|           return env->NewStringUTF(StringPrintf("%" PRId64, frames[method_index].location).c_str());
 | |
|         case 3: {
 | |
|           jint line_number = FindLineNumber(line_number_count,
 | |
|                                             line_number_table,
 | |
|                                             frames[method_index].location);
 | |
|           return env->NewStringUTF(StringPrintf("%d", line_number).c_str());
 | |
|         }
 | |
|       }
 | |
|       LOG(FATAL) << "Unreachable";
 | |
|       UNREACHABLE();
 | |
|     };
 | |
|     jobjectArray inner_array = CreateObjectArray(env, 4, "java/lang/String", inner_callback);
 | |
| 
 | |
|     if (name != nullptr) {
 | |
|       jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(name));
 | |
|     }
 | |
|     if (sig != nullptr) {
 | |
|       jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(sig));
 | |
|     }
 | |
|     if (gen != nullptr) {
 | |
|       jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(gen));
 | |
|     }
 | |
|     if (line_number_table != nullptr) {
 | |
|       jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(line_number_table));
 | |
|     }
 | |
| 
 | |
|     return inner_array;
 | |
|   };
 | |
|   return CreateObjectArray(env, count, "[Ljava/lang/String;", callback);
 | |
| }
 | |
| 
 | |
| extern "C" JNIEXPORT jobjectArray JNICALL Java_art_PrintThread_getStackTrace(
 | |
|     JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jthread thread, jint start, jint max) {
 | |
|   std::unique_ptr<jvmtiFrameInfo[]> frames(new jvmtiFrameInfo[max]);
 | |
| 
 | |
|   jint count;
 | |
|   {
 | |
|     jvmtiError result = jvmti_env->GetStackTrace(thread, start, max, frames.get(), &count);
 | |
|     if (JvmtiErrorToException(env, jvmti_env, result)) {
 | |
|       return nullptr;
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   return TranslateJvmtiFrameInfoArray(env, frames.get(), count);
 | |
| }
 | |
| 
 | |
| extern "C" JNIEXPORT jobjectArray JNICALL Java_art_AllTraces_getAllStackTraces(
 | |
|     JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jint max) {
 | |
|   jint thread_count;
 | |
|   jvmtiStackInfo* stack_infos;
 | |
|   {
 | |
|     jvmtiError result = jvmti_env->GetAllStackTraces(max, &stack_infos, &thread_count);
 | |
|     if (JvmtiErrorToException(env, jvmti_env, result)) {
 | |
|       return nullptr;
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   auto callback = [&](jint thread_index) -> jobject {
 | |
|     auto inner_callback = [&](jint index) -> jobject {
 | |
|       if (index == 0) {
 | |
|         return stack_infos[thread_index].thread;
 | |
|       } else {
 | |
|         return TranslateJvmtiFrameInfoArray(env,
 | |
|                                             stack_infos[thread_index].frame_buffer,
 | |
|                                             stack_infos[thread_index].frame_count);
 | |
|       }
 | |
|     };
 | |
|     return CreateObjectArray(env, 2, "java/lang/Object", inner_callback);
 | |
|   };
 | |
|   jobjectArray ret = CreateObjectArray(env, thread_count, "[Ljava/lang/Object;", callback);
 | |
|   jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(stack_infos));
 | |
|   return ret;
 | |
| }
 | |
| 
 | |
| extern "C" JNIEXPORT jobjectArray JNICALL Java_art_ThreadListTraces_getThreadListStackTraces(
 | |
|     JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jobjectArray jthreads, jint max) {
 | |
|   jint thread_count = env->GetArrayLength(jthreads);
 | |
|   std::unique_ptr<jthread[]> threads(new jthread[thread_count]);
 | |
|   for (jint i = 0; i != thread_count; ++i) {
 | |
|     threads[i] = env->GetObjectArrayElement(jthreads, i);
 | |
|   }
 | |
| 
 | |
|   jvmtiStackInfo* stack_infos;
 | |
|   {
 | |
|     jvmtiError result = jvmti_env->GetThreadListStackTraces(thread_count,
 | |
|                                                             threads.get(),
 | |
|                                                             max,
 | |
|                                                             &stack_infos);
 | |
|     if (JvmtiErrorToException(env, jvmti_env, result)) {
 | |
|       return nullptr;
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   auto callback = [&](jint thread_index) -> jobject {
 | |
|     auto inner_callback = [&](jint index) -> jobject {
 | |
|       if (index == 0) {
 | |
|         return stack_infos[thread_index].thread;
 | |
|       } else {
 | |
|         return TranslateJvmtiFrameInfoArray(env,
 | |
|                                             stack_infos[thread_index].frame_buffer,
 | |
|                                             stack_infos[thread_index].frame_count);
 | |
|       }
 | |
|     };
 | |
|     return CreateObjectArray(env, 2, "java/lang/Object", inner_callback);
 | |
|   };
 | |
|   jobjectArray ret = CreateObjectArray(env, thread_count, "[Ljava/lang/Object;", callback);
 | |
|   jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(stack_infos));
 | |
|   return ret;
 | |
| }
 | |
| 
 | |
| extern "C" JNIEXPORT jint JNICALL Java_art_Frames_getFrameCount(
 | |
|     JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jthread thread) {
 | |
|   jint count;
 | |
|   jvmtiError result = jvmti_env->GetFrameCount(thread, &count);
 | |
|   if (JvmtiErrorToException(env, jvmti_env, result)) {
 | |
|     return -1;
 | |
|   }
 | |
|   return count;
 | |
| }
 | |
| 
 | |
| extern "C" JNIEXPORT jobjectArray JNICALL Java_art_Frames_getFrameLocation(
 | |
|     JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jthread thread, jint depth) {
 | |
|   jmethodID method;
 | |
|   jlocation location;
 | |
| 
 | |
|   jvmtiError result = jvmti_env->GetFrameLocation(thread, depth, &method, &location);
 | |
|   if (JvmtiErrorToException(env, jvmti_env, result)) {
 | |
|     return nullptr;
 | |
|   }
 | |
| 
 | |
|   auto callback = [&](jint index) -> jobject {
 | |
|     switch (index) {
 | |
|       case 0:
 | |
|       {
 | |
|         jclass decl_class;
 | |
|         jvmtiError class_result = jvmti_env->GetMethodDeclaringClass(method, &decl_class);
 | |
|         if (JvmtiErrorToException(env, jvmti_env, class_result)) {
 | |
|           return nullptr;
 | |
|         }
 | |
|         jint modifiers;
 | |
|         jvmtiError mod_result = jvmti_env->GetMethodModifiers(method, &modifiers);
 | |
|         if (JvmtiErrorToException(env, jvmti_env, mod_result)) {
 | |
|           return nullptr;
 | |
|         }
 | |
|         constexpr jint kStatic = 0x8;
 | |
|         return env->ToReflectedMethod(decl_class,
 | |
|                                       method,
 | |
|                                       (modifiers & kStatic) != 0 ? JNI_TRUE : JNI_FALSE);
 | |
|       }
 | |
|       case 1:
 | |
|         return env->NewStringUTF(
 | |
|             android::base::StringPrintf("%x", static_cast<uint32_t>(location)).c_str());
 | |
|     }
 | |
|     LOG(FATAL) << "Unreachable";
 | |
|     UNREACHABLE();
 | |
|   };
 | |
|   jobjectArray ret = CreateObjectArray(env, 2, "java/lang/Object", callback);
 | |
|   return ret;
 | |
| }
 | |
| 
 | |
| }  // namespace Test911GetStackTrace
 | |
| }  // namespace art
 |