1 /*
2  * Copyright (c) 2003, 2018, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.
8  *
9  * This code is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12  * version 2 for more details (a copy is included in the LICENSE file that
13  * accompanied this code).
14  *
15  * You should have received a copy of the GNU General Public License version
16  * 2 along with this work; if not, write to the Free Software Foundation,
17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20  * or visit www.oracle.com if you need additional information or have any
21  * questions.
22  */
23 
24 #include <stdio.h>
25 #include <string.h>
26 #include "jvmti.h"
27 #include "agent_common.h"
28 #include "JVMTITools.h"
29 
30 extern "C" {
31 
32 
33 #define PASSED 0
34 #define STATUS_FAILED 2
35 
36 static jvmtiEnv *jvmti = NULL;
37 static jint result = PASSED;
38 static jboolean printdump = JNI_FALSE;
39 
40 #ifdef STATIC_BUILD
Agent_OnLoad_methloc002(JavaVM * jvm,char * options,void * reserved)41 JNIEXPORT jint JNICALL Agent_OnLoad_methloc002(JavaVM *jvm, char *options, void *reserved) {
42     return Agent_Initialize(jvm, options, reserved);
43 }
Agent_OnAttach_methloc002(JavaVM * jvm,char * options,void * reserved)44 JNIEXPORT jint JNICALL Agent_OnAttach_methloc002(JavaVM *jvm, char *options, void *reserved) {
45     return Agent_Initialize(jvm, options, reserved);
46 }
JNI_OnLoad_methloc002(JavaVM * jvm,char * options,void * reserved)47 JNIEXPORT jint JNI_OnLoad_methloc002(JavaVM *jvm, char *options, void *reserved) {
48     return JNI_VERSION_1_8;
49 }
50 #endif
Agent_Initialize(JavaVM * jvm,char * options,void * reserved)51 jint Agent_Initialize(JavaVM *jvm, char *options, void *reserved) {
52     jint res;
53 
54     if (options != NULL && strcmp(options, "printdump") == 0) {
55         printdump = JNI_TRUE;
56     }
57 
58     res = jvm->GetEnv((void **) &jvmti, JVMTI_VERSION_1_1);
59     if (res != JNI_OK || jvmti == NULL) {
60         printf("Wrong result of a valid call to GetEnv!\n");
61         return JNI_ERR;
62     }
63 
64     return JNI_OK;
65 }
66 
67 JNIEXPORT jint JNICALL
Java_nsk_jvmti_GetMethodLocation_methloc002_check(JNIEnv * env,jclass cls)68 Java_nsk_jvmti_GetMethodLocation_methloc002_check(JNIEnv *env, jclass cls) {
69     jvmtiError err;
70     jmethodID mid;
71     jlocation startLocation;
72     jlocation endLocation;
73 
74     if (jvmti == NULL) {
75         printf("JVMTI client was not properly loaded!\n");
76         return STATUS_FAILED;
77     }
78 
79     mid = env->GetMethodID(cls, "<init>", "()V");
80     if (mid == NULL) {
81         printf("Cannot get method ID for \"<init>\"!\n");
82         return STATUS_FAILED;
83     }
84 
85     if (printdump == JNI_TRUE) {
86         printf(">>> invalid method check ...\n");
87     }
88     err = jvmti->GetMethodLocation(NULL, &startLocation, &endLocation);
89     if (err != JVMTI_ERROR_INVALID_METHODID) {
90         printf("Error expected: JVMTI_ERROR_INVALID_METHODID,\n");
91         printf("\tactual: %s (%d)\n", TranslateError(err), err);
92         result = STATUS_FAILED;
93     }
94 
95     if (printdump == JNI_TRUE) {
96         printf(">>> (startLocationPtr) null pointer check ...\n");
97     }
98     err = jvmti->GetMethodLocation(mid, NULL, &endLocation);
99     if (err != JVMTI_ERROR_NULL_POINTER) {
100         printf("(startLocationPtr) error expected: JVMTI_ERROR_NULL_POINTER,\n");
101         printf("\tactual: %s (%d)\n", TranslateError(err), err);
102         result = STATUS_FAILED;
103     }
104 
105     if (printdump == JNI_TRUE) {
106         printf(">>> (endLocationPtr) null pointer check ...\n");
107     }
108     err = jvmti->GetMethodLocation(mid, &startLocation, NULL);
109     if (err != JVMTI_ERROR_NULL_POINTER) {
110         printf("(endLocationPtr) error expected: JVMTI_ERROR_NULL_POINTER,\n");
111         printf("\tactual: %s (%d)\n", TranslateError(err), err);
112         result = STATUS_FAILED;
113     }
114 
115     mid = env->GetStaticMethodID(cls, "check", "()I");
116     if (mid == NULL) {
117         printf("Cannot get method ID for \"check\"!\n");
118         return STATUS_FAILED;
119     }
120 
121     if (printdump == JNI_TRUE) {
122         printf(">>> native method check ...\n");
123     }
124     err = jvmti->GetMethodLocation(mid, &startLocation, &endLocation);
125     if (err != JVMTI_ERROR_NATIVE_METHOD) {
126         printf("Error expected: JVMTI_ERROR_NATIVE_METHOD,\n");
127         printf("\tactual: %s (%d)\n", TranslateError(err), err);
128         result = STATUS_FAILED;
129     }
130 
131     if (printdump == JNI_TRUE) {
132         printf(">>> ... done\n");
133     }
134 
135     return result;
136 }
137 
138 }
139