blob: f2532debfbdd21abd3bad7a2ff3694bec14bce6c [file] [log] [blame]
/*
* 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 <iomanip>
#include <iostream>
#include <pthread.h>
#include <sstream>
#include <stdio.h>
#include <vector>
#include "android-base/stringprintf.h"
#include "base/logging.h"
#include "jni.h"
#include "jvmti.h"
#include "ScopedPrimitiveArray.h"
#include "ti-agent/common_helper.h"
#include "ti-agent/common_load.h"
#include "utf.h"
namespace art {
namespace Test906IterateHeap {
class IterationConfig {
public:
IterationConfig() {}
virtual ~IterationConfig() {}
virtual jint Handle(jlong class_tag, jlong size, jlong* tag_ptr, jint length) = 0;
};
static jint JNICALL HeapIterationCallback(jlong class_tag,
jlong size,
jlong* tag_ptr,
jint length,
void* user_data) {
IterationConfig* config = reinterpret_cast<IterationConfig*>(user_data);
return config->Handle(class_tag, size, tag_ptr, length);
}
static bool Run(jint heap_filter, jclass klass_filter, IterationConfig* config) {
jvmtiHeapCallbacks callbacks;
memset(&callbacks, 0, sizeof(jvmtiHeapCallbacks));
callbacks.heap_iteration_callback = HeapIterationCallback;
jvmtiError ret = jvmti_env->IterateThroughHeap(heap_filter,
klass_filter,
&callbacks,
config);
if (ret != JVMTI_ERROR_NONE) {
char* err;
jvmti_env->GetErrorName(ret, &err);
printf("Failure running IterateThroughHeap: %s\n", err);
jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(err));
return false;
}
return true;
}
extern "C" JNIEXPORT jint JNICALL Java_Main_iterateThroughHeapCount(JNIEnv* env ATTRIBUTE_UNUSED,
jclass klass ATTRIBUTE_UNUSED,
jint heap_filter,
jclass klass_filter,
jint stop_after) {
class CountIterationConfig : public IterationConfig {
public:
CountIterationConfig(jint _counter, jint _stop_after)
: counter(_counter),
stop_after(_stop_after) {
}
jint Handle(jlong class_tag ATTRIBUTE_UNUSED,
jlong size ATTRIBUTE_UNUSED,
jlong* tag_ptr ATTRIBUTE_UNUSED,
jint length ATTRIBUTE_UNUSED) OVERRIDE {
counter++;
if (counter == stop_after) {
return JVMTI_VISIT_ABORT;
}
return 0;
}
jint counter;
const jint stop_after;
};
CountIterationConfig config(0, stop_after);
Run(heap_filter, klass_filter, &config);
if (config.counter > config.stop_after) {
printf("Error: more objects visited than signaled.");
}
return config.counter;
}
extern "C" JNIEXPORT jint JNICALL Java_Main_iterateThroughHeapData(JNIEnv* env,
jclass klass ATTRIBUTE_UNUSED,
jint heap_filter,
jclass klass_filter,
jlongArray class_tags,
jlongArray sizes,
jlongArray tags,
jintArray lengths) {
class DataIterationConfig : public IterationConfig {
public:
jint Handle(jlong class_tag, jlong size, jlong* tag_ptr, jint length) OVERRIDE {
class_tags_.push_back(class_tag);
sizes_.push_back(size);
tags_.push_back(*tag_ptr);
lengths_.push_back(length);
return 0; // Continue.
}
std::vector<jlong> class_tags_;
std::vector<jlong> sizes_;
std::vector<jlong> tags_;
std::vector<jint> lengths_;
};
DataIterationConfig config;
if (!Run(heap_filter, klass_filter, &config)) {
return -1;
}
ScopedLongArrayRW s_class_tags(env, class_tags);
ScopedLongArrayRW s_sizes(env, sizes);
ScopedLongArrayRW s_tags(env, tags);
ScopedIntArrayRW s_lengths(env, lengths);
for (size_t i = 0; i != config.class_tags_.size(); ++i) {
s_class_tags[i] = config.class_tags_[i];
s_sizes[i] = config.sizes_[i];
s_tags[i] = config.tags_[i];
s_lengths[i] = config.lengths_[i];
}
return static_cast<jint>(config.class_tags_.size());
}
extern "C" JNIEXPORT void JNICALL Java_Main_iterateThroughHeapAdd(JNIEnv* env ATTRIBUTE_UNUSED,
jclass klass ATTRIBUTE_UNUSED,
jint heap_filter,
jclass klass_filter) {
class AddIterationConfig : public IterationConfig {
public:
AddIterationConfig() {}
jint Handle(jlong class_tag ATTRIBUTE_UNUSED,
jlong size ATTRIBUTE_UNUSED,
jlong* tag_ptr,
jint length ATTRIBUTE_UNUSED) OVERRIDE {
jlong current_tag = *tag_ptr;
if (current_tag != 0) {
*tag_ptr = current_tag + 10;
}
return 0;
}
};
AddIterationConfig config;
Run(heap_filter, klass_filter, &config);
}
extern "C" JNIEXPORT jstring JNICALL Java_Main_iterateThroughHeapString(
JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jlong tag) {
struct FindStringCallbacks {
explicit FindStringCallbacks(jlong t) : tag_to_find(t) {}
static jint JNICALL HeapIterationCallback(jlong class_tag ATTRIBUTE_UNUSED,
jlong size ATTRIBUTE_UNUSED,
jlong* tag_ptr ATTRIBUTE_UNUSED,
jint length ATTRIBUTE_UNUSED,
void* user_data ATTRIBUTE_UNUSED) {
return 0;
}
static jint JNICALL StringValueCallback(jlong class_tag,
jlong size,
jlong* tag_ptr,
const jchar* value,
jint value_length,
void* user_data) {
FindStringCallbacks* p = reinterpret_cast<FindStringCallbacks*>(user_data);
if (*tag_ptr == p->tag_to_find) {
size_t utf_byte_count = CountUtf8Bytes(value, value_length);
std::unique_ptr<char[]> mod_utf(new char[utf_byte_count + 1]);
memset(mod_utf.get(), 0, utf_byte_count + 1);
ConvertUtf16ToModifiedUtf8(mod_utf.get(), utf_byte_count, value, value_length);
if (!p->data.empty()) {
p->data += "\n";
}
p->data += android::base::StringPrintf("%" PRId64 "@%" PRId64 " (%" PRId64 ", '%s')",
*tag_ptr,
class_tag,
size,
mod_utf.get());
// Update the tag to test whether that works.
*tag_ptr = *tag_ptr + 1;
}
return 0;
}
std::string data;
const jlong tag_to_find;
};
jvmtiHeapCallbacks callbacks;
memset(&callbacks, 0, sizeof(jvmtiHeapCallbacks));
callbacks.heap_iteration_callback = FindStringCallbacks::HeapIterationCallback;
callbacks.string_primitive_value_callback = FindStringCallbacks::StringValueCallback;
FindStringCallbacks fsc(tag);
jvmtiError ret = jvmti_env->IterateThroughHeap(0, nullptr, &callbacks, &fsc);
if (JvmtiErrorToException(env, ret)) {
return nullptr;
}
return env->NewStringUTF(fsc.data.c_str());
}
extern "C" JNIEXPORT jstring JNICALL Java_Main_iterateThroughHeapPrimitiveArray(
JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jlong tag) {
struct FindArrayCallbacks {
explicit FindArrayCallbacks(jlong t) : tag_to_find(t) {}
static jint JNICALL HeapIterationCallback(jlong class_tag ATTRIBUTE_UNUSED,
jlong size ATTRIBUTE_UNUSED,
jlong* tag_ptr ATTRIBUTE_UNUSED,
jint length ATTRIBUTE_UNUSED,
void* user_data ATTRIBUTE_UNUSED) {
return 0;
}
static jint JNICALL ArrayValueCallback(jlong class_tag,
jlong size,
jlong* tag_ptr,
jint element_count,
jvmtiPrimitiveType element_type,
const void* elements,
void* user_data) {
FindArrayCallbacks* p = reinterpret_cast<FindArrayCallbacks*>(user_data);
if (*tag_ptr == p->tag_to_find) {
std::ostringstream oss;
oss << *tag_ptr
<< '@'
<< class_tag
<< " ("
<< size
<< ", "
<< element_count
<< "x"
<< static_cast<char>(element_type)
<< " '";
size_t element_size;
switch (element_type) {
case JVMTI_PRIMITIVE_TYPE_BOOLEAN:
case JVMTI_PRIMITIVE_TYPE_BYTE:
element_size = 1;
break;
case JVMTI_PRIMITIVE_TYPE_CHAR:
case JVMTI_PRIMITIVE_TYPE_SHORT:
element_size = 2;
break;
case JVMTI_PRIMITIVE_TYPE_INT:
case JVMTI_PRIMITIVE_TYPE_FLOAT:
element_size = 4;
break;
case JVMTI_PRIMITIVE_TYPE_LONG:
case JVMTI_PRIMITIVE_TYPE_DOUBLE:
element_size = 8;
break;
default:
LOG(FATAL) << "Unknown type " << static_cast<size_t>(element_type);
UNREACHABLE();
}
const uint8_t* data = reinterpret_cast<const uint8_t*>(elements);
for (size_t i = 0; i != element_size * element_count; ++i) {
oss << android::base::StringPrintf("%02x", data[i]);
}
oss << "')";
if (!p->data.empty()) {
p->data += "\n";
}
p->data += oss.str();
// Update the tag to test whether that works.
*tag_ptr = *tag_ptr + 1;
}
return 0;
}
std::string data;
const jlong tag_to_find;
};
jvmtiHeapCallbacks callbacks;
memset(&callbacks, 0, sizeof(jvmtiHeapCallbacks));
callbacks.heap_iteration_callback = FindArrayCallbacks::HeapIterationCallback;
callbacks.array_primitive_value_callback = FindArrayCallbacks::ArrayValueCallback;
FindArrayCallbacks fac(tag);
jvmtiError ret = jvmti_env->IterateThroughHeap(0, nullptr, &callbacks, &fac);
if (JvmtiErrorToException(env, ret)) {
return nullptr;
}
return env->NewStringUTF(fac.data.c_str());
}
static constexpr const char* GetPrimitiveTypeName(jvmtiPrimitiveType type) {
switch (type) {
case JVMTI_PRIMITIVE_TYPE_BOOLEAN:
return "boolean";
case JVMTI_PRIMITIVE_TYPE_BYTE:
return "byte";
case JVMTI_PRIMITIVE_TYPE_CHAR:
return "char";
case JVMTI_PRIMITIVE_TYPE_SHORT:
return "short";
case JVMTI_PRIMITIVE_TYPE_INT:
return "int";
case JVMTI_PRIMITIVE_TYPE_FLOAT:
return "float";
case JVMTI_PRIMITIVE_TYPE_LONG:
return "long";
case JVMTI_PRIMITIVE_TYPE_DOUBLE:
return "double";
}
LOG(FATAL) << "Unknown type " << static_cast<size_t>(type);
UNREACHABLE();
}
extern "C" JNIEXPORT jstring JNICALL Java_Main_iterateThroughHeapPrimitiveFields(
JNIEnv* env, jclass klass ATTRIBUTE_UNUSED, jlong tag) {
struct FindFieldCallbacks {
explicit FindFieldCallbacks(jlong t) : tag_to_find(t) {}
static jint JNICALL HeapIterationCallback(jlong class_tag ATTRIBUTE_UNUSED,
jlong size ATTRIBUTE_UNUSED,
jlong* tag_ptr ATTRIBUTE_UNUSED,
jint length ATTRIBUTE_UNUSED,
void* user_data ATTRIBUTE_UNUSED) {
return 0;
}
static jint JNICALL PrimitiveFieldValueCallback(jvmtiHeapReferenceKind kind,
const jvmtiHeapReferenceInfo* info,
jlong class_tag,
jlong* tag_ptr,
jvalue value,
jvmtiPrimitiveType value_type,
void* user_data) {
FindFieldCallbacks* p = reinterpret_cast<FindFieldCallbacks*>(user_data);
if (*tag_ptr >= p->tag_to_find) {
std::ostringstream oss;
oss << *tag_ptr
<< '@'
<< class_tag
<< " ("
<< (kind == JVMTI_HEAP_REFERENCE_FIELD ? "instance, " : "static, ")
<< GetPrimitiveTypeName(value_type)
<< ", index="
<< info->field.index
<< ") ";
// Be lazy, always print eight bytes.
static_assert(sizeof(jvalue) == sizeof(uint64_t), "Unexpected jvalue size");
uint64_t val;
memcpy(&val, &value, sizeof(uint64_t)); // To avoid undefined behavior.
oss << android::base::StringPrintf("%016" PRIx64, val);
if (!p->data.empty()) {
p->data += "\n";
}
p->data += oss.str();
*tag_ptr = *tag_ptr + 1;
}
return 0;
}
std::string data;
const jlong tag_to_find;
};
jvmtiHeapCallbacks callbacks;
memset(&callbacks, 0, sizeof(jvmtiHeapCallbacks));
callbacks.heap_iteration_callback = FindFieldCallbacks::HeapIterationCallback;
callbacks.primitive_field_callback = FindFieldCallbacks::PrimitiveFieldValueCallback;
FindFieldCallbacks ffc(tag);
jvmtiError ret = jvmti_env->IterateThroughHeap(0, nullptr, &callbacks, &ffc);
if (JvmtiErrorToException(env, ret)) {
return nullptr;
}
return env->NewStringUTF(ffc.data.c_str());
}
} // namespace Test906IterateHeap
} // namespace art