0N/A/*
2362N/A * Copyright (c) 2004, Oracle and/or its affiliates. All rights reserved.
0N/A *
0N/A * Redistribution and use in source and binary forms, with or without
0N/A * modification, are permitted provided that the following conditions
0N/A * are met:
0N/A *
0N/A * - Redistributions of source code must retain the above copyright
0N/A * notice, this list of conditions and the following disclaimer.
0N/A *
0N/A * - Redistributions in binary form must reproduce the above copyright
0N/A * notice, this list of conditions and the following disclaimer in the
0N/A * documentation and/or other materials provided with the distribution.
0N/A *
2362N/A * - Neither the name of Oracle nor the names of its
0N/A * contributors may be used to endorse or promote products derived
0N/A * from this software without specific prior written permission.
0N/A *
0N/A * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
0N/A * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
0N/A * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
0N/A * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
0N/A * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
0N/A * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
0N/A * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
0N/A * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
0N/A * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
0N/A * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
0N/A * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
0N/A */
0N/A
4378N/A/*
4378N/A * This source code is provided to illustrate the usage of a given feature
4378N/A * or technique and has been deliberately simplified. Additional steps
4378N/A * required for a production-quality application, such as security checks,
4378N/A * input validation and proper error handling, might not be present in
4378N/A * this sample code.
4378N/A */
4378N/A
4378N/A
0N/A/* Example of using JVMTI events:
0N/A * JVMTI_EVENT_VM_INIT
0N/A * JVMTI_EVENT_VM_DEATH
0N/A * JVMTI_EVENT_THREAD_START
0N/A * JVMTI_EVENT_THREAD_END
0N/A * JVMTI_EVENT_MONITOR_CONTENDED_ENTER
0N/A * JVMTI_EVENT_MONITOR_WAIT
0N/A * JVMTI_EVENT_MONITOR_WAITED
0N/A * JVMTI_EVENT_OBJECT_FREE
0N/A */
0N/A
0N/A#include <stdio.h>
0N/A#include <stdlib.h>
0N/A#include <string.h>
0N/A
0N/A#include "jni.h"
0N/A#include "jvmti.h"
0N/A
0N/A#include "agent_util.h"
0N/A
0N/A#include "Monitor.hpp"
0N/A#include "Thread.hpp"
0N/A#include "Agent.hpp"
0N/A
0N/Astatic jrawMonitorID vm_death_lock;
0N/Astatic jboolean vm_death_active;
0N/A
0N/A/* Given a jvmtiEnv*, return the C++ Agent class instance */
0N/Astatic Agent *
0N/Aget_agent(jvmtiEnv *jvmti)
0N/A{
0N/A jvmtiError err;
0N/A Agent *agent;
0N/A
0N/A agent = NULL;
0N/A err = jvmti->GetEnvironmentLocalStorage((void**)&agent);
0N/A check_jvmti_error(jvmti, err, "get env local storage");
0N/A if ( agent == NULL ) {
0N/A /* This should never happen, but we should check */
0N/A fatal_error("ERROR: GetEnvironmentLocalStorage() returned NULL");
0N/A }
0N/A return agent;
0N/A}
0N/A
0N/A/* Enter raw monitor */
0N/Astatic void
0N/Amenter(jvmtiEnv *jvmti, jrawMonitorID rmon)
0N/A{
0N/A jvmtiError err;
0N/A
0N/A err = jvmti->RawMonitorEnter(rmon);
0N/A check_jvmti_error(jvmti, err, "raw monitor enter");
0N/A}
0N/A
0N/A/* Exit raw monitor */
0N/Astatic void
0N/Amexit(jvmtiEnv *jvmti, jrawMonitorID rmon)
0N/A{
0N/A jvmtiError err;
0N/A
0N/A err = jvmti->RawMonitorExit(rmon);
0N/A check_jvmti_error(jvmti, err, "raw monitor exit");
0N/A}
0N/A
0N/A
0N/A/* All callbacks need to be extern "C" */
0N/Aextern "C" {
0N/A static void JNICALL
0N/A vm_init(jvmtiEnv *jvmti, JNIEnv *env, jthread thread)
0N/A {
0N/A jvmtiError err;
0N/A Agent *agent;
0N/A
0N/A /* Create raw monitor to protect against threads running after death */
0N/A err = jvmti->CreateRawMonitor("Waiters vm_death lock", &vm_death_lock);
0N/A check_jvmti_error(jvmti, err, "create raw monitor");
0N/A vm_death_active = JNI_FALSE;
0N/A
0N/A /* Create an Agent instance, set JVMTI Local Storage */
0N/A agent = new Agent(jvmti, env, thread);
0N/A err = jvmti->SetEnvironmentLocalStorage((const void*)agent);
0N/A check_jvmti_error(jvmti, err, "set env local storage");
0N/A
0N/A /* Enable all other events we want */
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_VM_DEATH, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_THREAD_START, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_THREAD_END, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_MONITOR_CONTENDED_ENTER, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_MONITOR_CONTENDED_ENTERED, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_MONITOR_WAIT, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_MONITOR_WAITED, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_OBJECT_FREE, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A }
0N/A static void JNICALL
0N/A vm_death(jvmtiEnv *jvmti, JNIEnv *env)
0N/A {
0N/A jvmtiError err;
0N/A Agent *agent;
0N/A
0N/A /* Block all callbacks */
0N/A menter(jvmti, vm_death_lock); {
0N/A /* Set flag for other callbacks */
0N/A vm_death_active = JNI_TRUE;
0N/A
0N/A /* Inform Agent instance of VM_DEATH */
0N/A agent = get_agent(jvmti);
0N/A agent->vm_death(jvmti, env);
0N/A
0N/A /* Reclaim space of Agent */
0N/A err = jvmti->SetEnvironmentLocalStorage((const void*)NULL);
0N/A check_jvmti_error(jvmti, err, "set env local storage");
0N/A delete agent;
0N/A } mexit(jvmti, vm_death_lock);
0N/A
0N/A }
0N/A static void JNICALL
0N/A thread_start(jvmtiEnv *jvmti, JNIEnv *env, jthread thread)
0N/A {
0N/A menter(jvmti, vm_death_lock); {
0N/A if ( !vm_death_active ) {
0N/A get_agent(jvmti)->thread_start(jvmti, env, thread);
0N/A }
0N/A } mexit(jvmti, vm_death_lock);
0N/A }
0N/A static void JNICALL
0N/A thread_end(jvmtiEnv *jvmti, JNIEnv *env, jthread thread)
0N/A {
0N/A menter(jvmti, vm_death_lock); {
0N/A if ( !vm_death_active ) {
0N/A get_agent(jvmti)->thread_end(jvmti, env, thread);
0N/A }
0N/A } mexit(jvmti, vm_death_lock);
0N/A }
0N/A static void JNICALL
0N/A monitor_contended_enter(jvmtiEnv* jvmti, JNIEnv *env,
0N/A jthread thread, jobject object)
0N/A {
0N/A menter(jvmti, vm_death_lock); {
0N/A if ( !vm_death_active ) {
0N/A get_agent(jvmti)->monitor_contended_enter(jvmti, env,
0N/A thread, object);
0N/A }
0N/A } mexit(jvmti, vm_death_lock);
0N/A }
0N/A static void JNICALL
0N/A monitor_contended_entered(jvmtiEnv* jvmti, JNIEnv *env,
0N/A jthread thread, jobject object)
0N/A {
0N/A menter(jvmti, vm_death_lock); {
0N/A if ( !vm_death_active ) {
0N/A get_agent(jvmti)->monitor_contended_entered(jvmti, env,
0N/A thread, object);
0N/A }
0N/A } mexit(jvmti, vm_death_lock);
0N/A }
0N/A static void JNICALL
0N/A monitor_wait(jvmtiEnv* jvmti, JNIEnv *env,
0N/A jthread thread, jobject object, jlong timeout)
0N/A {
0N/A menter(jvmti, vm_death_lock); {
0N/A if ( !vm_death_active ) {
0N/A get_agent(jvmti)->monitor_wait(jvmti, env, thread,
0N/A object, timeout);
0N/A }
0N/A } mexit(jvmti, vm_death_lock);
0N/A }
0N/A static void JNICALL
0N/A monitor_waited(jvmtiEnv* jvmti, JNIEnv *env,
0N/A jthread thread, jobject object, jboolean timed_out)
0N/A {
0N/A menter(jvmti, vm_death_lock); {
0N/A if ( !vm_death_active ) {
0N/A get_agent(jvmti)->monitor_waited(jvmti, env, thread,
0N/A object, timed_out);
0N/A }
0N/A } mexit(jvmti, vm_death_lock);
0N/A }
0N/A static void JNICALL
0N/A object_free(jvmtiEnv* jvmti, jlong tag)
0N/A {
0N/A menter(jvmti, vm_death_lock); {
0N/A if ( !vm_death_active ) {
0N/A get_agent(jvmti)->object_free(jvmti, tag);
0N/A }
0N/A } mexit(jvmti, vm_death_lock);
0N/A }
0N/A
0N/A /* Agent_OnLoad() is called first, we prepare for a VM_INIT event here. */
0N/A JNIEXPORT jint JNICALL
0N/A Agent_OnLoad(JavaVM *vm, char *options, void *reserved)
0N/A {
0N/A jvmtiEnv *jvmti;
0N/A jint rc;
0N/A jvmtiError err;
0N/A jvmtiCapabilities capabilities;
0N/A jvmtiEventCallbacks callbacks;
0N/A
0N/A /* Get JVMTI environment */
0N/A rc = vm->GetEnv((void **)&jvmti, JVMTI_VERSION);
0N/A if (rc != JNI_OK) {
0N/A fatal_error("ERROR: Unable to create jvmtiEnv, GetEnv failed, error=%d\n", rc);
0N/A return -1;
0N/A }
0N/A
0N/A /* Get/Add JVMTI capabilities */
0N/A (void)memset(&capabilities, 0, sizeof(capabilities));
0N/A capabilities.can_generate_monitor_events = 1;
0N/A capabilities.can_get_monitor_info = 1;
0N/A capabilities.can_tag_objects = 1;
0N/A capabilities.can_generate_object_free_events = 1;
0N/A err = jvmti->AddCapabilities(&capabilities);
0N/A check_jvmti_error(jvmti, err, "add capabilities");
0N/A
0N/A /* Set all callbacks and enable VM_INIT event notification */
0N/A memset(&callbacks, 0, sizeof(callbacks));
0N/A callbacks.VMInit = &vm_init;
0N/A callbacks.VMDeath = &vm_death;
0N/A callbacks.ThreadStart = &thread_start;
0N/A callbacks.ThreadEnd = &thread_end;
0N/A callbacks.MonitorContendedEnter = &monitor_contended_enter;
0N/A callbacks.MonitorContendedEntered = &monitor_contended_entered;
0N/A callbacks.MonitorWait = &monitor_wait;
0N/A callbacks.MonitorWaited = &monitor_waited;
0N/A callbacks.ObjectFree = &object_free;
0N/A err = jvmti->SetEventCallbacks(&callbacks, (jint)sizeof(callbacks));
0N/A check_jvmti_error(jvmti, err, "set event callbacks");
0N/A err = jvmti->SetEventNotificationMode(JVMTI_ENABLE,
0N/A JVMTI_EVENT_VM_INIT, NULL);
0N/A check_jvmti_error(jvmti, err, "set event notify");
0N/A return 0;
0N/A }
0N/A
0N/A /* Agent_OnUnload() is called last */
0N/A JNIEXPORT void JNICALL
0N/A Agent_OnUnload(JavaVM *vm)
0N/A {
0N/A }
0N/A
0N/A} /* of extern "C" */