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 
  28 #ifdef __cplusplus
  29 extern "C" {
  30 #endif
  31 
  32 #define STATUS_PASSED 0
  33 #define STATUS_FAILED 2
  34 #define TranslateError(err) "JVMTI error"
  35 
  36 static jint result = STATUS_PASSED;
  37 static jvmtiEnv *jvmti = NULL;
  38 
  39 #define DECL_TEST_FUNC(type, Type) \
  40 static void \
  41 test_##type(jthread thr, int depth, int slot, const char* exp_type) { \
  42   j##type val; \
  43   jvmtiError err = jvmti->GetLocal##Type(thr, depth, slot, &val); \
  44   \
  45   printf(" GetLocal%s: %s (%d)\n", #Type, TranslateError(err), err); \
  46   if (err != JVMTI_ERROR_NONE) { \
  47     printf(" FAIL: GetLocal%s failed to get value from a local %s\n", #Type, exp_type); \
  48     result = STATUS_FAILED; \
  49   } else { \
  50     printf(" GetLocal%s got value from a local %s as expected\n", #Type, exp_type); \
  51   } \
  52 }
  53 
  54 #define DECL_TEST_INV_SLOT_FUNC(type, Type) \
  55 static void \
  56 test_##type##_inv_slot(jthread thr, int depth, int slot, const char* exp_type) { \
  57   j##type val; \
  58   jvmtiError err = jvmti->GetLocal##Type(thr, depth, slot, &val); \
  59   \
  60   printf(" GetLocal%s: %s (%d)\n", #Type, TranslateError(err), err); \
  61   if (err != JVMTI_ERROR_INVALID_SLOT) { \
  62     printf(" FAIL: GetLocal%s failed to return JVMTI_ERROR_INVALID_SLOT for local %s\n", #Type, exp_type); \
  63     result = STATUS_FAILED; \
  64   } else { \
  65     printf(" GetLocal%s returned JVMTI_ERROR_INVALID_SLOT for local %s as expected\n", #Type, exp_type); \
  66   } \
  67 }
  68 
  69 #define DECL_TEST_TYPE_MISMATCH_FUNC(type, Type) \
  70 static void \
  71 test_##type##_type_mismatch(jthread thr, int depth, int slot, const char* exp_type) { \
  72   j##type val; \
  73   jvmtiError err = jvmti->GetLocal##Type(thr, depth, slot, &val); \
  74   \
  75   printf(" GetLocal%s: %s (%d)\n", #Type, TranslateError(err), err); \
  76   if (err != JVMTI_ERROR_TYPE_MISMATCH) { \
  77     printf(" FAIL: GetLocal%s failed to return JVMTI_ERROR_TYPE_MISMATCH for local %s\n", #Type, exp_type); \
  78     result = STATUS_FAILED; \
  79   } else { \
  80     printf(" GetLocal%s returned JVMTI_ERROR_TYPE_MISMATCH for local %s as expected\n", #Type, exp_type); \
  81   } \
  82 }
  83 
  84 DECL_TEST_FUNC(int, Int);
  85 DECL_TEST_FUNC(float, Float);
  86 DECL_TEST_FUNC(long, Long);
  87 DECL_TEST_FUNC(double, Double);
  88 DECL_TEST_FUNC(object, Object);
  89 
  90 DECL_TEST_INV_SLOT_FUNC(int, Int);
  91 DECL_TEST_INV_SLOT_FUNC(float, Float);
  92 
  93 DECL_TEST_TYPE_MISMATCH_FUNC(int, Int);
  94 DECL_TEST_TYPE_MISMATCH_FUNC(float, Float);
  95 DECL_TEST_TYPE_MISMATCH_FUNC(long, Long);
  96 DECL_TEST_TYPE_MISMATCH_FUNC(double, Double);
  97 DECL_TEST_TYPE_MISMATCH_FUNC(object, Object);
  98 
  99 static void
 100 test_local_byte(jthread thr, int depth, int slot) {
 101   printf("\n test_local_byte: BEGIN\n\n");
 102 
 103   test_int(thr, depth, slot, "byte");
 104   test_long_inv_slot(thr, depth, slot, "byte");
 105   test_float(thr, depth, slot, "byte");
 106   test_double_inv_slot(thr, depth, slot, "byte");
 107   test_object_type_mismatch(thr, depth, slot, "byte");
 108 
 109   printf("\n test_local_byte: END\n\n");
 110 }
 111 
 112 static void
 113 test_local_object(jthread thr, int depth, int slot) {
 114   printf("\n test_local_object: BEGIN\n\n");
 115 
 116   test_int_type_mismatch(thr, depth, slot, "object");
 117   test_long_type_mismatch(thr, depth, slot, "object");
 118   test_float_type_mismatch(thr, depth, slot, "object");
 119   test_double_type_mismatch(thr, depth, slot, "object");
 120   test_object(thr, depth, slot, "object");
 121 
 122   printf("\n test_local_object: END\n\n");
 123 }
 124 
 125 static void
 126 test_local_double(jthread thr, int depth, int slot) {
 127   printf("\n test_local_double: BEGIN\n\n");
 128 
 129   test_int(thr, depth, slot, "double");
 130   test_long(thr, depth, slot, "double");
 131   test_float(thr, depth, slot, "double");
 132   test_double(thr, depth, slot, "double");
 133   test_object_type_mismatch(thr, depth, slot, "double");
 134 
 135   printf("\n test_local_double: END\n\n");
 136 }
 137 
 138 static void
 139 test_local_integer(jthread thr, int depth, int slot) {
 140   printf("\n test_local_integer: BEGIN\n\n");
 141 
 142   test_int(thr, depth, slot, "int");
 143   test_float(thr, depth, slot, "int");
 144   test_object_type_mismatch(thr, depth, slot, "double");
 145 
 146   printf("\n test_local_integer: END\n\n");
 147 }
 148 
 149 static void
 150 test_local_invalid(jthread thr, int depth, int slot) {
 151   printf("\n test_local_invalid: BEGIN\n\n");
 152 
 153   test_int_inv_slot(thr, depth, slot, "invalid");
 154   test_long_inv_slot(thr, depth, slot, "invalid");
 155   test_float_inv_slot(thr, depth, slot, "invalid");
 156   test_double_inv_slot(thr, depth, slot, "invalid");
 157 
 158   printf("\n test_local_invalid: END\n\n");
 159 }
 160 
 161 jint Agent_Initialize(JavaVM *jvm, char *options, void *reserved) {
 162   jint res;
 163   jvmtiError err;
 164   static jvmtiCapabilities caps;
 165 
 166   res = jvm->GetEnv((void **) &jvmti, JVMTI_VERSION_9);
 167   if (res != JNI_OK || jvmti == NULL) {
 168     printf("Wrong result of a valid call to GetEnv!\n");
 169     return JNI_ERR;
 170   }
 171   caps.can_access_local_variables = 1;
 172 
 173   err = jvmti->AddCapabilities(&caps);
 174   if (err != JVMTI_ERROR_NONE) {
 175     printf("AddCapabilities: unexpected error: %s (%d)\n", TranslateError(err), err);
 176     return JNI_ERR;
 177   }
 178   err = jvmti->GetCapabilities(&caps);
 179   if (err != JVMTI_ERROR_NONE) {
 180     printf("GetCapabilities: unexpected error: %s (%d)\n", TranslateError(err), err);
 181     return JNI_ERR;
 182   }
 183   if (!caps.can_access_local_variables) {
 184     printf("Warning: Access to local variables is not implemented\n");
 185     return JNI_ERR;
 186   }
 187   return JNI_OK;
 188 }
 189 
 190 JNIEXPORT jint JNICALL
 191 Agent_OnLoad(JavaVM *jvm, char *options, void *reserved) {
 192   return Agent_Initialize(jvm, options, reserved);
 193 }
 194 
 195 JNIEXPORT jint JNICALL
 196 Agent_OnAttach(JavaVM *jvm, char *options, void *reserved) {
 197   return Agent_Initialize(jvm, options, reserved);
 198 }
 199 
 200 JNIEXPORT void JNICALL
 201 Java_GetLocalVars_testLocals(JNIEnv *env, jclass cls, jobject thread) {
 202   /*
 203    * We test the JVMTI GetLocal<Type> for locals of the method:
 204    *
 205    *  int staticMeth(byte byteArg, Object objArg, double dblArg, int intArg) {
 206    *      testLocals(Thread.currentThread());
 207    *      {
 208    *          int intLoc = 9999;
 209    *          intArg = intLoc;
 210    *      }
 211    *      return intArg;
 212    *  }
 213    */
 214   static const char* METHOD_NAME = "staticMeth";
 215   static const char* METHOD_SIGN = "(BLjava/lang/Object;DI)I";
 216   static const int Depth = 1;
 217   static const int ByteSlot = 0;
 218   static const int ObjSlot = 1;
 219   static const int DblSlot = 2;
 220   static const int IntSlot = 4;
 221   static const int InvalidSlot = 5;
 222 
 223   jmethodID mid = NULL;
 224 
 225   if (jvmti == NULL) {
 226     printf("JVMTI client was not properly loaded!\n");
 227     result = STATUS_FAILED;
 228     return;
 229   }
 230 
 231   mid = env->GetStaticMethodID(cls, METHOD_NAME, METHOD_SIGN);
 232   if (mid == NULL) {
 233     printf("Cannot find Method ID for %s%s\n", METHOD_NAME, METHOD_SIGN);
 234     result = STATUS_FAILED;
 235     return;
 236   }
 237 
 238   test_local_byte(thread, Depth, ByteSlot);
 239   test_local_object(thread, Depth, ObjSlot);
 240   test_local_double(thread, Depth, DblSlot);
 241   test_local_integer(thread, Depth, IntSlot);
 242   test_local_invalid(thread, Depth, InvalidSlot);
 243 }
 244 
 245 JNIEXPORT jint JNICALL
 246 Java_GetLocalVars_getStatus(JNIEnv *env, jclass cls) {
 247     return result;
 248 }
 249 
 250 #ifdef __cplusplus
 251 }
 252 #endif