/*
|
* Copyright (C) 2016 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 "901-hello-ti-agent/basics.h"
|
|
#include <thread>
|
|
#include <jni.h>
|
#include <stdio.h>
|
#include <string.h>
|
#include "android-base/macros.h"
|
#include "jvmti.h"
|
|
// Test infrastructure
|
#include "jvmti_helper.h"
|
#include "test_env.h"
|
|
namespace art {
|
namespace Test901HelloTi {
|
|
static void EnableEvent(jvmtiEnv* env, jvmtiEvent evt) {
|
jvmtiError error = env->SetEventNotificationMode(JVMTI_ENABLE, evt, nullptr);
|
if (error != JVMTI_ERROR_NONE) {
|
printf("Failed to enable event");
|
}
|
}
|
|
static jvmtiPhase getPhase(jvmtiEnv* jenv) {
|
jvmtiPhase out = static_cast<jvmtiPhase>(-1);
|
jenv->GetPhase(&out);
|
return out;
|
}
|
|
static void JNICALL VMStartCallback(jvmtiEnv *jenv, JNIEnv* jni_env ATTRIBUTE_UNUSED) {
|
printf("VMStart (phase %d)\n", getPhase(jenv));
|
fsync(1);
|
}
|
|
static void JNICALL VMInitCallback(jvmtiEnv *jvmti_env,
|
JNIEnv* jni_env ATTRIBUTE_UNUSED,
|
jthread thread ATTRIBUTE_UNUSED) {
|
printf("VMInit (phase %d)\n", getPhase(jvmti_env));
|
fsync(1);
|
}
|
|
static void JNICALL VMDeathCallback(jvmtiEnv *jenv, JNIEnv* jni_env) {
|
printf("VMDeath (phase %d)\n", getPhase(jenv));
|
fsync(1);
|
jthread cur_thr;
|
CHECK_EQ(jenv->GetCurrentThread(&cur_thr), JVMTI_ERROR_NONE);
|
CHECK(cur_thr != nullptr);
|
jni_env->DeleteLocalRef(cur_thr);
|
}
|
|
|
static void InstallVMEvents(jvmtiEnv* env) {
|
jvmtiEventCallbacks callbacks;
|
memset(&callbacks, 0, sizeof(jvmtiEventCallbacks));
|
callbacks.VMStart = VMStartCallback;
|
callbacks.VMInit = VMInitCallback;
|
callbacks.VMDeath = VMDeathCallback;
|
jvmtiError ret = env->SetEventCallbacks(&callbacks, sizeof(callbacks));
|
if (ret != JVMTI_ERROR_NONE) {
|
printf("Failed to install callbacks");
|
}
|
|
EnableEvent(env, JVMTI_EVENT_VM_START);
|
EnableEvent(env, JVMTI_EVENT_VM_INIT);
|
EnableEvent(env, JVMTI_EVENT_VM_DEATH);
|
}
|
|
jint OnLoad(JavaVM* vm,
|
char* options ATTRIBUTE_UNUSED,
|
void* reserved ATTRIBUTE_UNUSED) {
|
printf("Loaded Agent for test 901-hello-ti-agent\n");
|
fsync(1);
|
jvmtiEnv* env = nullptr;
|
jvmtiEnv* env2 = nullptr;
|
|
#define CHECK_CALL_SUCCESS(c) \
|
do { \
|
if ((c) != JNI_OK) { \
|
printf("call " #c " did not succeed\n"); \
|
return -1; \
|
} \
|
} while (false)
|
|
CHECK_CALL_SUCCESS(vm->GetEnv(reinterpret_cast<void**>(&env), JVMTI_VERSION_1_0));
|
CHECK_CALL_SUCCESS(vm->GetEnv(reinterpret_cast<void**>(&env2), JVMTI_VERSION_1_0));
|
if (env == env2) {
|
printf("GetEnv returned same environment twice!\n");
|
return -1;
|
}
|
unsigned char* local_data = nullptr;
|
CHECK_CALL_SUCCESS(env->Allocate(8, &local_data));
|
strcpy(reinterpret_cast<char*>(local_data), "hello!!");
|
CHECK_CALL_SUCCESS(env->SetEnvironmentLocalStorage(local_data));
|
unsigned char* get_data = nullptr;
|
CHECK_CALL_SUCCESS(env->GetEnvironmentLocalStorage(reinterpret_cast<void**>(&get_data)));
|
if (get_data != local_data) {
|
printf("Got different data from local storage then what was set!\n");
|
return -1;
|
}
|
CHECK_CALL_SUCCESS(env2->GetEnvironmentLocalStorage(reinterpret_cast<void**>(&get_data)));
|
if (get_data != nullptr) {
|
printf("env2 did not have nullptr local storage.\n");
|
return -1;
|
}
|
CHECK_CALL_SUCCESS(env->Deallocate(local_data));
|
jint version = 0;
|
CHECK_CALL_SUCCESS(env->GetVersionNumber(&version));
|
if ((version & JVMTI_VERSION_1) != JVMTI_VERSION_1) {
|
printf("Unexpected version number!\n");
|
return -1;
|
}
|
|
InstallVMEvents(env);
|
InstallVMEvents(env2);
|
|
CHECK_CALL_SUCCESS(env->DisposeEnvironment());
|
CHECK_CALL_SUCCESS(env2->DisposeEnvironment());
|
#undef CHECK_CALL_SUCCESS
|
|
if (vm->GetEnv(reinterpret_cast<void**>(&jvmti_env), JVMTI_VERSION_1_0)) {
|
printf("Unable to get jvmti env!\n");
|
return 1;
|
}
|
SetStandardCapabilities(jvmti_env);
|
|
jvmtiPhase current_phase;
|
jvmtiError phase_result = jvmti_env->GetPhase(¤t_phase);
|
if (phase_result != JVMTI_ERROR_NONE) {
|
printf("Could not get phase");
|
return 1;
|
}
|
if (current_phase != JVMTI_PHASE_ONLOAD) {
|
printf("Wrong phase");
|
return 1;
|
}
|
|
InstallVMEvents(jvmti_env);
|
|
return JNI_OK;
|
}
|
|
extern "C" JNIEXPORT void JNICALL Java_art_Test901_setVerboseFlag(
|
JNIEnv* env, jclass Main_klass ATTRIBUTE_UNUSED, jint iflag, jboolean val) {
|
jvmtiVerboseFlag flag = static_cast<jvmtiVerboseFlag>(iflag);
|
jvmtiError result = jvmti_env->SetVerboseFlag(flag, val);
|
JvmtiErrorToException(env, jvmti_env, result);
|
}
|
|
extern "C" JNIEXPORT jboolean JNICALL Java_art_Test901_checkLivePhase(
|
JNIEnv* env, jclass Main_klass ATTRIBUTE_UNUSED) {
|
jvmtiPhase current_phase;
|
jvmtiError phase_result = jvmti_env->GetPhase(¤t_phase);
|
if (JvmtiErrorToException(env, jvmti_env, phase_result)) {
|
return JNI_FALSE;
|
}
|
return (current_phase == JVMTI_PHASE_LIVE) ? JNI_TRUE : JNI_FALSE;
|
}
|
|
static void CallJvmtiFunction(jvmtiEnv* env, jclass klass, jvmtiError* err) {
|
jint n;
|
jmethodID* methods = nullptr;
|
*err = env->GetClassMethods(klass, &n, &methods);
|
}
|
|
extern "C" JNIEXPORT jboolean JNICALL Java_art_Test901_checkUnattached(
|
JNIEnv* env ATTRIBUTE_UNUSED, jclass Main_klass) {
|
jvmtiError res = JVMTI_ERROR_NONE;
|
std::thread t1(CallJvmtiFunction, jvmti_env, Main_klass, &res);
|
t1.join();
|
return res == JVMTI_ERROR_UNATTACHED_THREAD;
|
}
|
|
extern "C" JNIEXPORT jstring JNICALL Java_art_Test901_getErrorName(
|
JNIEnv* env, jclass Main_klass ATTRIBUTE_UNUSED, jint error) {
|
char* name;
|
jvmtiError res = jvmti_env->GetErrorName(static_cast<jvmtiError>(error), &name);
|
if (JvmtiErrorToException(env, jvmti_env, res)) {
|
return nullptr;
|
}
|
|
jstring ret_string = env->NewStringUTF(name);
|
jvmtiError dealloc = jvmti_env->Deallocate(reinterpret_cast<unsigned char*>(name));
|
if (JvmtiErrorToException(env, jvmti_env, dealloc)) {
|
return nullptr;
|
}
|
|
return ret_string;
|
}
|
|
} // namespace Test901HelloTi
|
} // namespace art
|
