libTest8026328.c

/*
 * Copyright (c) 2013, 2018, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

#include <string.h>
#include <stdio.h>

#include "jvmti.h"

#define CLASS_NAME "Lcompiler/jsr292/cr8026328/Test8026328;"
#define METHOD_NAME "main"

static jvmtiEnv *jvmti = NULL;


void JNICALL
classprepare(jvmtiEnv* jvmti_env,
            JNIEnv* jni_env,
            jthread thread,
            jclass klass) {

    char* buf;
    (*jvmti)->GetClassSignature(jvmti,
                klass,
                &buf,
                NULL);
    if (strcmp(CLASS_NAME, buf) == 0) {
        jint nMethods;
        jmethodID* methods;
        int i;
        (*jvmti)->GetClassMethods(jvmti,
                    klass,
                    &nMethods,
                    &methods);
        for (i = 0; i < nMethods; i++) {
            char* name;
            (*jvmti)->GetMethodName(jvmti,
                        methods[i],
                        &name,
                        NULL,
                        NULL);
            if (strcmp(METHOD_NAME, name) == 0) {
                printf("Setting breakpoint\n");
                fflush(stdout);
                (*jvmti)->SetBreakpoint(jvmti, methods[i], 0);
            }
        }
    }
}


void JNICALL
breakpoint(jvmtiEnv* jvmti_env,
            JNIEnv* jni_env,
            jthread thread,
            jmethodID method,
            jlocation location) {

    jclass declaring_class;
    char* name;
    char* cname;
    (*jvmti)->GetMethodName(jvmti,
                method,
                &name,
                NULL,
                NULL);
    (*jvmti)->GetMethodDeclaringClass(jvmti,
                method,
                &declaring_class);
    (*jvmti)->GetClassSignature(jvmti,
                declaring_class,
                &cname,
                NULL);
    printf("Hit breakpoint at %s::%s:%d\n", cname, name, (int) location);
    fflush(stdout);
}


JNIEXPORT jint JNICALL
Agent_OnLoad(JavaVM* vm,
             char* options,
             void* reserved) {

    jvmtiCapabilities capa;
    jvmtiEventCallbacks cbs = {0};

    (*vm)->GetEnv(vm, (void**)&jvmti, JVMTI_VERSION_1_0);

    memset(&capa, 0, sizeof(capa));
    capa.can_generate_breakpoint_events = 1;
    capa.can_generate_single_step_events = 1;
    (*jvmti)->AddCapabilities(jvmti, &capa);

    cbs.ClassPrepare = classprepare;
    cbs.Breakpoint = breakpoint;
    (*jvmti)->SetEventCallbacks(jvmti, &cbs, sizeof(cbs));
    (*jvmti)->SetEventNotificationMode(jvmti, JVMTI_ENABLE, JVMTI_EVENT_CLASS_PREPARE, NULL);
    (*jvmti)->SetEventNotificationMode(jvmti, JVMTI_ENABLE, JVMTI_EVENT_BREAKPOINT, NULL);
    printf("Loaded agent\n");
    fflush(stdout);

    return 0;
}