1 /* Copyright (C) 2016 The Android Open Source Project
2  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
3  *
4  * This file implements interfaces from the file jvmti.h. This implementation
5  * is licensed under the same terms as the file jvmti.h.  The
6  * copyright and license information for the file jvmti.h follows.
7  *
8  * Copyright (c) 2003, 2011, Oracle and/or its affiliates. All rights reserved.
9  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
10  *
11  * This code is free software; you can redistribute it and/or modify it
12  * under the terms of the GNU General Public License version 2 only, as
13  * published by the Free Software Foundation.  Oracle designates this
14  * particular file as subject to the "Classpath" exception as provided
15  * by Oracle in the LICENSE file that accompanied this code.
16  *
17  * This code is distributed in the hope that it will be useful, but WITHOUT
18  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
20  * version 2 for more details (a copy is included in the LICENSE file that
21  * accompanied this code).
22  *
23  * You should have received a copy of the GNU General Public License version
24  * 2 along with this work; if not, write to the Free Software Foundation,
25  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
26  *
27  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
28  * or visit www.oracle.com if you need additional information or have any
29  * questions.
30  */
31 
32 #include "ti_method.h"
33 
34 #include <initializer_list>
35 #include <type_traits>
36 #include <variant>
37 
38 #include "android-base/macros.h"
39 #include "arch/context.h"
40 #include "art_jvmti.h"
41 #include "art_method-inl.h"
42 #include "base/enums.h"
43 #include "base/globals.h"
44 #include "base/macros.h"
45 #include "base/mutex-inl.h"
46 #include "deopt_manager.h"
47 #include "dex/code_item_accessors-inl.h"
48 #include "dex/code_item_accessors.h"
49 #include "dex/dex_file_annotations.h"
50 #include "dex/dex_file_types.h"
51 #include "dex/dex_instruction.h"
52 #include "dex/dex_instruction_iterator.h"
53 #include "dex/modifiers.h"
54 #include "dex/primitive.h"
55 #include "events-inl.h"
56 #include "gc_root-inl.h"
57 #include "handle.h"
58 #include "jit/jit.h"
59 #include "jni/jni_internal.h"
60 #include "jvmti.h"
61 #include "mirror/class-inl.h"
62 #include "mirror/class_loader.h"
63 #include "mirror/object-inl.h"
64 #include "mirror/object_array-inl.h"
65 #include "nativehelper/scoped_local_ref.h"
66 #include "oat_file.h"
67 #include "obj_ptr.h"
68 #include "runtime.h"
69 #include "runtime_callbacks.h"
70 #include "scoped_thread_state_change-inl.h"
71 #include "scoped_thread_state_change.h"
72 #include "stack.h"
73 #include "thread-current-inl.h"
74 #include "thread.h"
75 #include "thread_list.h"
76 #include "ti_logging.h"
77 #include "ti_stack.h"
78 #include "ti_thread.h"
79 #include "ti_phase.h"
80 #include "verifier/register_line-inl.h"
81 #include "verifier/reg_type-inl.h"
82 #include "verifier/method_verifier-inl.h"
83 
84 namespace openjdkjvmti {
85 
86 struct TiMethodCallback : public art::MethodCallback {
RegisterNativeMethodopenjdkjvmti::TiMethodCallback87   void RegisterNativeMethod(art::ArtMethod* method,
88                             const void* cur_method,
89                             /*out*/void** new_method)
90       override REQUIRES_SHARED(art::Locks::mutator_lock_) {
91     if (event_handler->IsEventEnabledAnywhere(ArtJvmtiEvent::kNativeMethodBind)) {
92       art::Thread* thread = art::Thread::Current();
93       art::JNIEnvExt* jnienv = thread->GetJniEnv();
94       ScopedLocalRef<jthread> thread_jni(
95           jnienv, PhaseUtil::IsLivePhase() ? jnienv->AddLocalReference<jthread>(thread->GetPeer())
96                                            : nullptr);
97       jmethodID method_id = art::jni::EncodeArtMethod(method);
98       art::ScopedThreadSuspension sts(thread, art::ThreadState::kNative);
99       event_handler->DispatchEvent<ArtJvmtiEvent::kNativeMethodBind>(
100           thread,
101           static_cast<JNIEnv*>(jnienv),
102           thread_jni.get(),
103           method_id,
104           const_cast<void*>(cur_method),
105           new_method);
106     }
107   }
108 
109   EventHandler* event_handler = nullptr;
110 };
111 
112 TiMethodCallback gMethodCallback;
113 
Register(EventHandler * handler)114 void MethodUtil::Register(EventHandler* handler) {
115   gMethodCallback.event_handler = handler;
116   art::ScopedThreadStateChange stsc(art::Thread::Current(),
117                                     art::ThreadState::kWaitingForDebuggerToAttach);
118   art::ScopedSuspendAll ssa("Add method callback");
119   art::RuntimeCallbacks* callbacks = art::Runtime::Current()->GetRuntimeCallbacks();
120   callbacks->AddMethodCallback(&gMethodCallback);
121 }
122 
Unregister()123 void MethodUtil::Unregister() {
124   art::ScopedThreadStateChange stsc(art::Thread::Current(),
125                                     art::ThreadState::kWaitingForDebuggerToAttach);
126   art::ScopedSuspendAll ssa("Remove method callback");
127   art::RuntimeCallbacks* callbacks = art::Runtime::Current()->GetRuntimeCallbacks();
128   callbacks->RemoveMethodCallback(&gMethodCallback);
129 }
130 
GetBytecodes(jvmtiEnv * env,jmethodID method,jint * size_ptr,unsigned char ** bytecode_ptr)131 jvmtiError MethodUtil::GetBytecodes(jvmtiEnv* env,
132                                     jmethodID method,
133                                     jint* size_ptr,
134                                     unsigned char** bytecode_ptr) {
135   if (method == nullptr) {
136     return ERR(INVALID_METHODID);
137   }
138   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
139 
140   if (art_method->IsNative()) {
141     return ERR(NATIVE_METHOD);
142   }
143 
144   if (size_ptr == nullptr || bytecode_ptr == nullptr) {
145     return ERR(NULL_POINTER);
146   }
147 
148   art::ScopedObjectAccess soa(art::Thread::Current());
149   art::CodeItemInstructionAccessor accessor(art_method->DexInstructions());
150   if (!accessor.HasCodeItem()) {
151     *size_ptr = 0;
152     *bytecode_ptr = nullptr;
153     return OK;
154   }
155   // 2 bytes per instruction for dex code.
156   *size_ptr = accessor.InsnsSizeInCodeUnits() * 2;
157   jvmtiError err = env->Allocate(*size_ptr, bytecode_ptr);
158   if (err != OK) {
159     return err;
160   }
161   memcpy(*bytecode_ptr, accessor.Insns(), *size_ptr);
162   return OK;
163 }
164 
GetArgumentsSize(jvmtiEnv * env ATTRIBUTE_UNUSED,jmethodID method,jint * size_ptr)165 jvmtiError MethodUtil::GetArgumentsSize(jvmtiEnv* env ATTRIBUTE_UNUSED,
166                                         jmethodID method,
167                                         jint* size_ptr) {
168   if (method == nullptr) {
169     return ERR(INVALID_METHODID);
170   }
171   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
172 
173   if (art_method->IsNative()) {
174     return ERR(NATIVE_METHOD);
175   }
176 
177   if (size_ptr == nullptr) {
178     return ERR(NULL_POINTER);
179   }
180 
181   art::ScopedObjectAccess soa(art::Thread::Current());
182   if (art_method->IsProxyMethod() || art_method->IsAbstract()) {
183     // Use the shorty.
184     art::ArtMethod* base_method = art_method->GetInterfaceMethodIfProxy(art::kRuntimePointerSize);
185     size_t arg_count = art::ArtMethod::NumArgRegisters(base_method->GetShorty());
186     if (!base_method->IsStatic()) {
187       arg_count++;
188     }
189     *size_ptr = static_cast<jint>(arg_count);
190     return ERR(NONE);
191   }
192 
193   DCHECK(art_method->HasCodeItem());
194   DCHECK_NE(art_method->GetCodeItem(), nullptr);
195   *size_ptr = art_method->DexInstructionData().InsSize();
196 
197   return ERR(NONE);
198 }
199 
GetLocalVariableTable(jvmtiEnv * env,jmethodID method,jint * entry_count_ptr,jvmtiLocalVariableEntry ** table_ptr)200 jvmtiError MethodUtil::GetLocalVariableTable(jvmtiEnv* env,
201                                              jmethodID method,
202                                              jint* entry_count_ptr,
203                                              jvmtiLocalVariableEntry** table_ptr) {
204   if (method == nullptr) {
205     return ERR(INVALID_METHODID);
206   }
207   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
208 
209   if (art_method->IsNative()) {
210     return ERR(NATIVE_METHOD);
211   }
212 
213   if (entry_count_ptr == nullptr || table_ptr == nullptr) {
214     return ERR(NULL_POINTER);
215   }
216 
217   art::ScopedObjectAccess soa(art::Thread::Current());
218 
219   const art::DexFile* const dex_file = art_method->GetDexFile();
220   if (dex_file == nullptr) {
221     return ERR(ABSENT_INFORMATION);
222   }
223 
224   // TODO HasCodeItem == false means that the method is abstract (or native, but we check that
225   // earlier). We should check what is returned by the RI in this situation since it's not clear
226   // what the appropriate return value is from the spec.
227   art::CodeItemDebugInfoAccessor accessor(art_method->DexInstructionDebugInfo());
228   if (!accessor.HasCodeItem()) {
229     return ERR(ABSENT_INFORMATION);
230   }
231 
232   std::vector<jvmtiLocalVariableEntry> variables;
233   jvmtiError err = OK;
234 
235   auto release = [&](jint* out_entry_count_ptr, jvmtiLocalVariableEntry** out_table_ptr) {
236     jlong table_size = sizeof(jvmtiLocalVariableEntry) * variables.size();
237     if (err != OK ||
238         (err = env->Allocate(table_size,
239                               reinterpret_cast<unsigned char**>(out_table_ptr))) != OK) {
240       for (jvmtiLocalVariableEntry& e : variables) {
241         env->Deallocate(reinterpret_cast<unsigned char*>(e.name));
242         env->Deallocate(reinterpret_cast<unsigned char*>(e.signature));
243         env->Deallocate(reinterpret_cast<unsigned char*>(e.generic_signature));
244       }
245       return err;
246     }
247     *out_entry_count_ptr = variables.size();
248     memcpy(*out_table_ptr, variables.data(), table_size);
249     return OK;
250   };
251 
252   auto visitor = [&](const art::DexFile::LocalInfo& entry) {
253     if (err != OK) {
254       return;
255     }
256     JvmtiUniquePtr<char[]> name_str = CopyString(env, entry.name_, &err);
257     if (err != OK) {
258       return;
259     }
260     JvmtiUniquePtr<char[]> sig_str = CopyString(env, entry.descriptor_, &err);
261     if (err != OK) {
262       return;
263     }
264     JvmtiUniquePtr<char[]> generic_sig_str = CopyString(env, entry.signature_, &err);
265     if (err != OK) {
266       return;
267     }
268     variables.push_back({
269       .start_location = static_cast<jlocation>(entry.start_address_),
270       .length = static_cast<jint>(entry.end_address_ - entry.start_address_),
271       .name = name_str.release(),
272       .signature = sig_str.release(),
273       .generic_signature = generic_sig_str.release(),
274       .slot = entry.reg_,
275     });
276   };
277 
278   if (!accessor.DecodeDebugLocalInfo(art_method->IsStatic(),
279                                      art_method->GetDexMethodIndex(),
280                                      visitor)) {
281     // Something went wrong with decoding the debug information. It might as well not be there.
282     return ERR(ABSENT_INFORMATION);
283   }
284   return release(entry_count_ptr, table_ptr);
285 }
286 
GetMaxLocals(jvmtiEnv * env ATTRIBUTE_UNUSED,jmethodID method,jint * max_ptr)287 jvmtiError MethodUtil::GetMaxLocals(jvmtiEnv* env ATTRIBUTE_UNUSED,
288                                     jmethodID method,
289                                     jint* max_ptr) {
290   if (method == nullptr) {
291     return ERR(INVALID_METHODID);
292   }
293   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
294 
295   if (art_method->IsNative()) {
296     return ERR(NATIVE_METHOD);
297   }
298 
299   if (max_ptr == nullptr) {
300     return ERR(NULL_POINTER);
301   }
302 
303   art::ScopedObjectAccess soa(art::Thread::Current());
304   if (art_method->IsProxyMethod() || art_method->IsAbstract()) {
305     // This isn't specified as an error case, so return 0.
306     *max_ptr = 0;
307     return ERR(NONE);
308   }
309 
310   DCHECK(art_method->HasCodeItem());
311   DCHECK_NE(art_method->GetCodeItem(), nullptr);
312   *max_ptr = art_method->DexInstructionData().RegistersSize();
313 
314   return ERR(NONE);
315 }
316 
GetMethodName(jvmtiEnv * env,jmethodID method,char ** name_ptr,char ** signature_ptr,char ** generic_ptr)317 jvmtiError MethodUtil::GetMethodName(jvmtiEnv* env,
318                                      jmethodID method,
319                                      char** name_ptr,
320                                      char** signature_ptr,
321                                      char** generic_ptr) {
322   art::ScopedObjectAccess soa(art::Thread::Current());
323   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
324   art_method = art_method->GetInterfaceMethodIfProxy(art::kRuntimePointerSize);
325 
326   JvmtiUniquePtr<char[]> name_copy;
327   if (name_ptr != nullptr) {
328     const char* method_name = art_method->GetName();
329     if (method_name == nullptr) {
330       method_name = "<error>";
331     }
332     jvmtiError ret;
333     name_copy = CopyString(env, method_name, &ret);
334     if (name_copy == nullptr) {
335       return ret;
336     }
337     *name_ptr = name_copy.get();
338   }
339 
340   JvmtiUniquePtr<char[]> signature_copy;
341   if (signature_ptr != nullptr) {
342     const art::Signature sig = art_method->GetSignature();
343     std::string str = sig.ToString();
344     jvmtiError ret;
345     signature_copy = CopyString(env, str.c_str(), &ret);
346     if (signature_copy == nullptr) {
347       return ret;
348     }
349     *signature_ptr = signature_copy.get();
350   }
351 
352   if (generic_ptr != nullptr) {
353     *generic_ptr = nullptr;
354     if (!art_method->GetDeclaringClass()->IsProxyClass()) {
355       art::ObjPtr<art::mirror::ObjectArray<art::mirror::String>> str_array =
356           art::annotations::GetSignatureAnnotationForMethod(art_method);
357       if (str_array != nullptr) {
358         std::ostringstream oss;
359         for (auto str : str_array->Iterate()) {
360           oss << str->ToModifiedUtf8();
361         }
362         std::string output_string = oss.str();
363         jvmtiError ret;
364         JvmtiUniquePtr<char[]> generic_copy = CopyString(env, output_string.c_str(), &ret);
365         if (generic_copy == nullptr) {
366           return ret;
367         }
368         *generic_ptr = generic_copy.release();
369       } else if (soa.Self()->IsExceptionPending()) {
370         // TODO: Should we report an error here?
371         soa.Self()->ClearException();
372       }
373     }
374   }
375 
376   // Everything is fine, release the buffers.
377   name_copy.release();
378   signature_copy.release();
379 
380   return ERR(NONE);
381 }
382 
GetMethodDeclaringClass(jvmtiEnv * env ATTRIBUTE_UNUSED,jmethodID method,jclass * declaring_class_ptr)383 jvmtiError MethodUtil::GetMethodDeclaringClass(jvmtiEnv* env ATTRIBUTE_UNUSED,
384                                                jmethodID method,
385                                                jclass* declaring_class_ptr) {
386   if (declaring_class_ptr == nullptr) {
387     return ERR(NULL_POINTER);
388   }
389 
390   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
391   // Note: No GetInterfaceMethodIfProxy, we want to actual class.
392 
393   art::ScopedObjectAccess soa(art::Thread::Current());
394   art::ObjPtr<art::mirror::Class> klass = art_method->GetDeclaringClass();
395   *declaring_class_ptr = soa.AddLocalReference<jclass>(klass);
396 
397   return ERR(NONE);
398 }
399 
GetMethodLocation(jvmtiEnv * env ATTRIBUTE_UNUSED,jmethodID method,jlocation * start_location_ptr,jlocation * end_location_ptr)400 jvmtiError MethodUtil::GetMethodLocation(jvmtiEnv* env ATTRIBUTE_UNUSED,
401                                          jmethodID method,
402                                          jlocation* start_location_ptr,
403                                          jlocation* end_location_ptr) {
404   if (method == nullptr) {
405     return ERR(INVALID_METHODID);
406   }
407   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
408 
409   if (art_method->IsNative()) {
410     return ERR(NATIVE_METHOD);
411   }
412 
413   if (start_location_ptr == nullptr || end_location_ptr == nullptr) {
414     return ERR(NULL_POINTER);
415   }
416 
417   art::ScopedObjectAccess soa(art::Thread::Current());
418   if (art_method->IsProxyMethod() || art_method->IsAbstract()) {
419     // This isn't specified as an error case, so return -1/-1 as the RI does.
420     *start_location_ptr = -1;
421     *end_location_ptr = -1;
422     return ERR(NONE);
423   }
424 
425   DCHECK(art_method->HasCodeItem());
426   DCHECK_NE(art_method->GetCodeItem(), nullptr);
427   *start_location_ptr = 0;
428   *end_location_ptr = art_method->DexInstructions().InsnsSizeInCodeUnits() - 1;
429 
430   return ERR(NONE);
431 }
432 
GetMethodModifiers(jvmtiEnv * env ATTRIBUTE_UNUSED,jmethodID method,jint * modifiers_ptr)433 jvmtiError MethodUtil::GetMethodModifiers(jvmtiEnv* env ATTRIBUTE_UNUSED,
434                                           jmethodID method,
435                                           jint* modifiers_ptr) {
436   if (modifiers_ptr == nullptr) {
437     return ERR(NULL_POINTER);
438   }
439 
440   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
441   uint32_t modifiers = art_method->GetAccessFlags();
442 
443   // Note: Keep this code in sync with Executable.fixMethodFlags.
444   if ((modifiers & art::kAccAbstract) != 0) {
445     modifiers &= ~art::kAccNative;
446   }
447   modifiers &= ~art::kAccSynchronized;
448   if ((modifiers & art::kAccDeclaredSynchronized) != 0) {
449     modifiers |= art::kAccSynchronized;
450   }
451   modifiers &= art::kAccJavaFlagsMask;
452 
453   *modifiers_ptr = modifiers;
454   return ERR(NONE);
455 }
456 
GetLineNumberTable(jvmtiEnv * env,jmethodID method,jint * entry_count_ptr,jvmtiLineNumberEntry ** table_ptr)457 jvmtiError MethodUtil::GetLineNumberTable(jvmtiEnv* env,
458                                           jmethodID method,
459                                           jint* entry_count_ptr,
460                                           jvmtiLineNumberEntry** table_ptr) {
461   if (method == nullptr) {
462     return ERR(NULL_POINTER);
463   }
464   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
465   DCHECK(!art_method->IsRuntimeMethod());
466 
467   art::CodeItemDebugInfoAccessor accessor;
468   const art::DexFile* dex_file;
469   {
470     art::ScopedObjectAccess soa(art::Thread::Current());
471 
472     if (art_method->IsProxyMethod()) {
473       return ERR(ABSENT_INFORMATION);
474     }
475     if (art_method->IsNative()) {
476       return ERR(NATIVE_METHOD);
477     }
478     if (entry_count_ptr == nullptr || table_ptr == nullptr) {
479       return ERR(NULL_POINTER);
480     }
481 
482     accessor = art::CodeItemDebugInfoAccessor(art_method->DexInstructionDebugInfo());
483     dex_file = art_method->GetDexFile();
484     DCHECK(accessor.HasCodeItem()) << art_method->PrettyMethod() << " " << dex_file->GetLocation();
485   }
486 
487   std::vector<jvmtiLineNumberEntry> context;
488   bool success = accessor.DecodeDebugPositionInfo([&](const art::DexFile::PositionInfo& entry) {
489     context.push_back({static_cast<jlocation>(entry.address_), static_cast<jint>(entry.line_)});
490     return false;
491   });
492   if (!success) {
493     return ERR(ABSENT_INFORMATION);
494   }
495 
496   unsigned char* data;
497   jlong mem_size = context.size() * sizeof(jvmtiLineNumberEntry);
498   jvmtiError alloc_error = env->Allocate(mem_size, &data);
499   if (alloc_error != ERR(NONE)) {
500     return alloc_error;
501   }
502   *table_ptr = reinterpret_cast<jvmtiLineNumberEntry*>(data);
503   memcpy(*table_ptr, context.data(), mem_size);
504   *entry_count_ptr = static_cast<jint>(context.size());
505 
506   return ERR(NONE);
507 }
508 
509 template <typename T>
IsMethodT(jvmtiEnv * env ATTRIBUTE_UNUSED,jmethodID method,T test,jboolean * is_t_ptr)510 static jvmtiError IsMethodT(jvmtiEnv* env ATTRIBUTE_UNUSED,
511                             jmethodID method,
512                             T test,
513                             jboolean* is_t_ptr) {
514   if (method == nullptr) {
515     return ERR(INVALID_METHODID);
516   }
517   if (is_t_ptr == nullptr) {
518     return ERR(NULL_POINTER);
519   }
520 
521   art::ArtMethod* art_method = art::jni::DecodeArtMethod(method);
522   *is_t_ptr = test(art_method) ? JNI_TRUE : JNI_FALSE;
523 
524   return ERR(NONE);
525 }
526 
IsMethodNative(jvmtiEnv * env,jmethodID m,jboolean * is_native_ptr)527 jvmtiError MethodUtil::IsMethodNative(jvmtiEnv* env, jmethodID m, jboolean* is_native_ptr) {
528   auto test = [](art::ArtMethod* method) {
529     return method->IsNative();
530   };
531   return IsMethodT(env, m, test, is_native_ptr);
532 }
533 
IsMethodObsolete(jvmtiEnv * env,jmethodID m,jboolean * is_obsolete_ptr)534 jvmtiError MethodUtil::IsMethodObsolete(jvmtiEnv* env, jmethodID m, jboolean* is_obsolete_ptr) {
535   auto test = [](art::ArtMethod* method) {
536     return method->IsObsolete();
537   };
538   return IsMethodT(env, m, test, is_obsolete_ptr);
539 }
540 
IsMethodSynthetic(jvmtiEnv * env,jmethodID m,jboolean * is_synthetic_ptr)541 jvmtiError MethodUtil::IsMethodSynthetic(jvmtiEnv* env, jmethodID m, jboolean* is_synthetic_ptr) {
542   auto test = [](art::ArtMethod* method) {
543     return method->IsSynthetic();
544   };
545   return IsMethodT(env, m, test, is_synthetic_ptr);
546 }
547 
548 class CommonLocalVariableClosure : public art::Closure {
549  public:
550   // The verifier isn't always able to be as specific as the local-variable-table. We can only get
551   // 32-bit, 64-bit or reference.
552   enum class VerifierPrimitiveType {
553     k32BitValue,  // float, int, short, char, boolean, byte
554     k64BitValue,  // double, long
555     kReferenceValue,  // Object
556     kZeroValue,  // null or zero constant. Might be either k32BitValue or kReferenceValue
557   };
558 
559   using SlotType = std::variant<art::Primitive::Type, VerifierPrimitiveType>;
560 
CommonLocalVariableClosure(jvmtiEnv * jvmti,jint depth,jint slot)561   CommonLocalVariableClosure(jvmtiEnv* jvmti, jint depth, jint slot)
562       : jvmti_(jvmti), result_(ERR(INTERNAL)), depth_(depth), slot_(slot) {}
563 
Run(art::Thread * self)564   void Run(art::Thread* self) override REQUIRES_SHARED(art::Locks::mutator_lock_) {
565     art::Locks::mutator_lock_->AssertSharedHeld(art::Thread::Current());
566     bool needs_instrument;
567     {
568       art::ScopedAssertNoThreadSuspension sants("CommonLocalVariableClosure::Run");
569       std::unique_ptr<art::Context> context(art::Context::Create());
570       FindFrameAtDepthVisitor visitor(self, context.get(), depth_);
571       visitor.WalkStack();
572       if (!visitor.FoundFrame()) {
573         // Must have been a bad depth.
574         result_ = ERR(NO_MORE_FRAMES);
575         return;
576       }
577       art::ArtMethod* method = visitor.GetMethod();
578       // Native and 'art' proxy methods don't have registers.
579       if (method->IsNative() || method->IsProxyMethod()) {
580         // TODO It might be useful to fake up support for get at least on proxy frames.
581         result_ = ERR(OPAQUE_FRAME);
582         return;
583       } else if (slot_ >= method->DexInstructionData().RegistersSize() || slot_ < 0) {
584         result_ = ERR(INVALID_SLOT);
585         return;
586       }
587       needs_instrument = !visitor.IsShadowFrame();
588       uint32_t pc = visitor.GetDexPc(/*abort_on_failure=*/false);
589       if (pc == art::dex::kDexNoIndex) {
590         // Cannot figure out current PC.
591         result_ = ERR(OPAQUE_FRAME);
592         return;
593       }
594       std::string descriptor;
595       SlotType slot_type{ art::Primitive::kPrimVoid };
596       jvmtiError err = GetSlotType(method, pc, &descriptor, &slot_type);
597       if (err != OK) {
598         result_ = err;
599         return;
600       }
601 
602       err = GetTypeError(method, slot_type, descriptor);
603       if (err != OK) {
604         result_ = err;
605         return;
606       }
607       result_ = Execute(method, visitor);
608     }
609     if (needs_instrument) {
610       DeoptManager::Get()->DeoptimizeThread(self);
611     }
612   }
613 
GetResult()614   virtual jvmtiError GetResult() {
615     return result_;
616   }
617 
618  protected:
619   virtual jvmtiError Execute(art::ArtMethod* method, art::StackVisitor& visitor)
620       REQUIRES_SHARED(art::Locks::mutator_lock_) = 0;
621   virtual jvmtiError GetTypeError(art::ArtMethod* method,
622                                   SlotType type,
623                                   const std::string& descriptor)
624       REQUIRES_SHARED(art::Locks::mutator_lock_)  = 0;
625 
626   jvmtiError GetSlotType(art::ArtMethod* method,
627                          uint32_t dex_pc,
628                          /*out*/std::string* descriptor,
629                          /*out*/SlotType* type)
630       REQUIRES_SHARED(art::Locks::mutator_lock_);
631 
InferSlotTypeFromVerifier(art::ArtMethod * method,uint32_t dex_pc,std::string * descriptor,SlotType * type)632   jvmtiError InferSlotTypeFromVerifier(art::ArtMethod* method,
633                                        uint32_t dex_pc,
634                                        /*out*/ std::string* descriptor,
635                                        /*out*/ SlotType* type)
636       REQUIRES_SHARED(art::Locks::mutator_lock_) {
637     art::Thread* self = art::Thread::Current();
638     art::StackHandleScope<2> hs(self);
639     std::unique_ptr<art::verifier::MethodVerifier> verifier(
640         art::verifier::MethodVerifier::CalculateVerificationInfo(
641             self,
642             method,
643             hs.NewHandle(method->GetDexCache()),
644             hs.NewHandle(method->GetDeclaringClass()->GetClassLoader())));
645     if (verifier == nullptr) {
646       JVMTI_LOG(WARNING, jvmti_) << "Unable to extract verification information from "
647                                  << method->PrettyMethod() << " due to hard verification failures! "
648                                  << "How did this method even get loaded!";
649       return ERR(INTERNAL);
650     }
651     art::verifier::RegisterLine* line = verifier->GetRegLine(dex_pc);
652     if (line == nullptr) {
653       JVMTI_LOG(WARNING, jvmti_) << "Unable to determine register line at dex-pc " << dex_pc
654                                  << " for method " << method->PrettyMethod();
655       return ERR(OPAQUE_FRAME);
656     }
657     const art::verifier::RegType& rt = line->GetRegisterType(verifier.get(), slot_);
658     if (rt.IsUndefined()) {
659       return ERR(INVALID_SLOT);
660     } else if (rt.IsNonZeroReferenceTypes() || rt.IsNull()) {
661       *descriptor = (rt.HasClass() ? rt.GetDescriptor() : "Ljava/lang/Object;");
662       *type = VerifierPrimitiveType::kReferenceValue;
663       return OK;
664     } else if (rt.IsZero()) {
665       *descriptor = "I";
666       *type = VerifierPrimitiveType::kZeroValue;
667       return OK;
668     } else if (rt.IsCategory1Types()) {
669       *descriptor = "I";
670       *type = VerifierPrimitiveType::k32BitValue;
671       return OK;
672     } else if (rt.IsCategory2Types() && rt.IsLowHalf()) {
673       *descriptor = "J";
674       *type = VerifierPrimitiveType::k64BitValue;
675       return OK;
676     } else {
677       // The slot doesn't have a type. Must not be valid here.
678       return ERR(INVALID_SLOT);
679     }
680   }
681 
SquashType(SlotType t)682   constexpr VerifierPrimitiveType SquashType(SlotType t) {
683     if (std::holds_alternative<art::Primitive::Type>(t)) {
684       switch (std::get<art::Primitive::Type>(t)) {
685         // 32-bit primitives
686         case art::Primitive::kPrimByte:
687         case art::Primitive::kPrimChar:
688         case art::Primitive::kPrimInt:
689         case art::Primitive::kPrimShort:
690         case art::Primitive::kPrimBoolean:
691         case art::Primitive::kPrimFloat:
692           return VerifierPrimitiveType::k32BitValue;
693         // 64-bit primitives
694         case art::Primitive::kPrimLong:
695         case art::Primitive::kPrimDouble:
696           return VerifierPrimitiveType::k64BitValue;
697         case art::Primitive::kPrimNot:
698           return VerifierPrimitiveType::kReferenceValue;
699         case art::Primitive::kPrimVoid:
700           LOG(FATAL) << "Got kPrimVoid";
701           UNREACHABLE();
702       }
703     } else {
704       return std::get<VerifierPrimitiveType>(t);
705     }
706   }
707 
708   jvmtiEnv* jvmti_;
709   jvmtiError result_;
710   jint depth_;
711   jint slot_;
712 
713  private:
714   DISALLOW_COPY_AND_ASSIGN(CommonLocalVariableClosure);
715 };
716 
operator <<(std::ostream & os,CommonLocalVariableClosure::VerifierPrimitiveType state)717 std::ostream& operator<<(std::ostream& os,
718                          CommonLocalVariableClosure::VerifierPrimitiveType state) {
719   switch (state) {
720     case CommonLocalVariableClosure::VerifierPrimitiveType::k32BitValue:
721       return os << "32BitValue";
722     case CommonLocalVariableClosure::VerifierPrimitiveType::k64BitValue:
723       return os << "64BitValue";
724     case CommonLocalVariableClosure::VerifierPrimitiveType::kReferenceValue:
725       return os << "ReferenceValue";
726     case CommonLocalVariableClosure::VerifierPrimitiveType::kZeroValue:
727       return os << "ZeroValue";
728   }
729 }
730 
operator <<(std::ostream & os,CommonLocalVariableClosure::SlotType state)731 std::ostream& operator<<(std::ostream& os, CommonLocalVariableClosure::SlotType state) {
732   if (std::holds_alternative<art::Primitive::Type>(state)) {
733     return os << "Primitive::Type[" << std::get<art::Primitive::Type>(state) << "]";
734   } else {
735     return os << "VerifierPrimitiveType["
736               << std::get<CommonLocalVariableClosure::VerifierPrimitiveType>(state) << "]";
737   }
738 }
739 
GetSlotType(art::ArtMethod * method,uint32_t dex_pc,std::string * descriptor,SlotType * type)740 jvmtiError CommonLocalVariableClosure::GetSlotType(art::ArtMethod* method,
741                                                    uint32_t dex_pc,
742                                                    /*out*/ std::string* descriptor,
743                                                    /*out*/ SlotType* type) {
744   const art::DexFile* dex_file = method->GetDexFile();
745   if (dex_file == nullptr) {
746     return ERR(OPAQUE_FRAME);
747   }
748   art::CodeItemDebugInfoAccessor accessor(method->DexInstructionDebugInfo());
749   if (!accessor.HasCodeItem()) {
750     return ERR(OPAQUE_FRAME);
751   }
752   bool found = false;
753   *type = art::Primitive::kPrimVoid;
754   descriptor->clear();
755   auto visitor = [&](const art::DexFile::LocalInfo& entry) {
756     if (!found && entry.start_address_ <= dex_pc && entry.end_address_ > dex_pc &&
757         entry.reg_ == slot_) {
758       found = true;
759       *type = art::Primitive::GetType(entry.descriptor_[0]);
760       *descriptor = entry.descriptor_;
761     }
762   };
763   if (!accessor.DecodeDebugLocalInfo(method->IsStatic(), method->GetDexMethodIndex(), visitor) ||
764       !found) {
765     // Something went wrong with decoding the debug information. It might as well not be there.
766     // Try to find the type with the verifier.
767     // TODO This is very slow.
768     return InferSlotTypeFromVerifier(method, dex_pc, descriptor, type);
769   } else if (art::kIsDebugBuild) {
770     std::string type_unused;
771     SlotType verifier_type{ art::Primitive::kPrimVoid };
772     DCHECK_EQ(InferSlotTypeFromVerifier(method, dex_pc, &type_unused, &verifier_type), OK)
773         << method->PrettyMethod() << " failed to verify!";
774     if (*type == SlotType{ art::Primitive::kPrimNot }) {
775       // We cannot distinguish between a constant 0 and a null reference so we return that it is a
776       // 32bit value (Due to the way references are read by the interpreter this is safe even if
777       // it's modified, the value will remain null). This is not ideal since it prevents modifying
778       // locals in some circumstances but generally is not a big deal (since one can just modify it
779       // later once it's been determined to be a reference by a later instruction).
780       DCHECK(verifier_type == SlotType { VerifierPrimitiveType::kZeroValue } ||
781              verifier_type == SlotType { VerifierPrimitiveType::kReferenceValue })
782           << "Verifier disagrees on type of slot! debug: " << *type
783           << " verifier: " << verifier_type;
784     } else if (verifier_type == SlotType { VerifierPrimitiveType::kZeroValue }) {
785       DCHECK(VerifierPrimitiveType::k32BitValue == SquashType(*type) ||
786              VerifierPrimitiveType::kReferenceValue == SquashType(*type))
787           << "Verifier disagrees on type of slot! debug: " << *type
788           << " verifier: " << verifier_type;
789     } else {
790       DCHECK_EQ(SquashType(verifier_type), SquashType(*type))
791           << "Verifier disagrees on type of slot! debug: " << *type
792           << " verifier: " << verifier_type;
793     }
794   }
795   return OK;
796 }
797 
798 class GetLocalVariableClosure : public CommonLocalVariableClosure {
799  public:
GetLocalVariableClosure(jvmtiEnv * jvmti,jint depth,jint slot,art::Primitive::Type type,jvalue * val)800   GetLocalVariableClosure(jvmtiEnv* jvmti,
801                           jint depth,
802                           jint slot,
803                           art::Primitive::Type type,
804                           jvalue* val)
805       : CommonLocalVariableClosure(jvmti, depth, slot),
806         type_(type),
807         val_(val),
808         obj_val_(nullptr) {}
809 
GetResult()810   jvmtiError GetResult() override REQUIRES_SHARED(art::Locks::mutator_lock_) {
811     if (result_ == OK && type_ == art::Primitive::kPrimNot) {
812       if (obj_val_ == nullptr) {
813         val_->l = nullptr;
814       } else {
815         art::JNIEnvExt* jni = art::Thread::Current()->GetJniEnv();
816         val_->l = static_cast<JNIEnv*>(jni)->NewLocalRef(obj_val_);
817         jni->DeleteGlobalRef(obj_val_);
818         obj_val_ = nullptr;
819       }
820     }
821     return CommonLocalVariableClosure::GetResult();
822   }
823 
824  protected:
825   jvmtiError
GetTypeError(art::ArtMethod * method,SlotType slot_type,const std::string & descriptor)826   GetTypeError(art::ArtMethod* method, SlotType slot_type, const std::string& descriptor) override
827       REQUIRES_SHARED(art::Locks::mutator_lock_) {
828     jvmtiError res = GetTypeErrorInner(method, slot_type, descriptor);
829     if (res == ERR(TYPE_MISMATCH)) {
830       JVMTI_LOG(INFO, jvmti_) << "Unable to Get local variable in slot " << slot_ << ". Expected"
831                               << " slot to be of type compatible with " << SlotType { type_ }
832                               << " but slot is " << slot_type;
833     } else if (res != OK) {
834       JVMTI_LOG(INFO, jvmti_) << "Unable to get local variable in slot " << slot_ << ".";
835     }
836     return res;
837   }
838 
GetTypeErrorInner(art::ArtMethod * method ATTRIBUTE_UNUSED,SlotType slot_type,const std::string & descriptor ATTRIBUTE_UNUSED)839   jvmtiError GetTypeErrorInner(art::ArtMethod* method ATTRIBUTE_UNUSED,
840                                SlotType slot_type,
841                                const std::string& descriptor ATTRIBUTE_UNUSED)
842       REQUIRES_SHARED(art::Locks::mutator_lock_) {
843     switch (type_) {
844       case art::Primitive::kPrimFloat:
845       case art::Primitive::kPrimInt: {
846         if (std::holds_alternative<VerifierPrimitiveType>(slot_type)) {
847           return (slot_type == SlotType { VerifierPrimitiveType::k32BitValue } ||
848                   slot_type == SlotType { VerifierPrimitiveType::kZeroValue })
849                      ? OK
850                      : ERR(TYPE_MISMATCH);
851         } else if (type_ == art::Primitive::kPrimFloat ||
852                    slot_type == SlotType { art::Primitive::kPrimFloat }) {
853           // Check that we are actually a float.
854           return (SlotType { type_ } == slot_type) ? OK : ERR(TYPE_MISMATCH);
855         } else {
856           // Some smaller int type.
857           return SquashType(slot_type) == SquashType(SlotType { type_ }) ? OK : ERR(TYPE_MISMATCH);
858         }
859       }
860       case art::Primitive::kPrimLong:
861       case art::Primitive::kPrimDouble: {
862         // todo
863         if (std::holds_alternative<VerifierPrimitiveType>(slot_type)) {
864           return (slot_type == SlotType { VerifierPrimitiveType::k64BitValue })
865                      ? OK
866                      : ERR(TYPE_MISMATCH);
867         } else {
868           return slot_type == SlotType { type_ } ? OK : ERR(TYPE_MISMATCH);
869         }
870       }
871       case art::Primitive::kPrimNot:
872         return (SquashType(slot_type) == VerifierPrimitiveType::kReferenceValue ||
873                 SquashType(slot_type) == VerifierPrimitiveType::kZeroValue)
874                    ? OK
875                    : ERR(TYPE_MISMATCH);
876       case art::Primitive::kPrimShort:
877       case art::Primitive::kPrimChar:
878       case art::Primitive::kPrimByte:
879       case art::Primitive::kPrimBoolean:
880       case art::Primitive::kPrimVoid:
881         LOG(FATAL) << "Unexpected primitive type " << slot_type;
882         UNREACHABLE();
883     }
884   }
885 
Execute(art::ArtMethod * method,art::StackVisitor & visitor)886   jvmtiError Execute(art::ArtMethod* method, art::StackVisitor& visitor)
887       override REQUIRES_SHARED(art::Locks::mutator_lock_) {
888     switch (type_) {
889       case art::Primitive::kPrimNot: {
890         uint32_t ptr_val;
891         if (!visitor.GetVReg(method,
892                              static_cast<uint16_t>(slot_),
893                              art::kReferenceVReg,
894                              &ptr_val)) {
895           return ERR(OPAQUE_FRAME);
896         }
897         art::ObjPtr<art::mirror::Object> obj(reinterpret_cast<art::mirror::Object*>(ptr_val));
898         obj_val_ = art::Runtime::Current()->GetJavaVM()->AddGlobalRef(art::Thread::Current(), obj);
899         break;
900       }
901       case art::Primitive::kPrimInt:
902       case art::Primitive::kPrimFloat: {
903         if (!visitor.GetVReg(method,
904                              static_cast<uint16_t>(slot_),
905                              type_ == art::Primitive::kPrimFloat ? art::kFloatVReg : art::kIntVReg,
906                              reinterpret_cast<uint32_t*>(&val_->i))) {
907           return ERR(OPAQUE_FRAME);
908         }
909         break;
910       }
911       case art::Primitive::kPrimDouble:
912       case art::Primitive::kPrimLong: {
913         auto lo_type = type_ == art::Primitive::kPrimLong ? art::kLongLoVReg : art::kDoubleLoVReg;
914         auto high_type = type_ == art::Primitive::kPrimLong ? art::kLongHiVReg : art::kDoubleHiVReg;
915         if (!visitor.GetVRegPair(method,
916                                  static_cast<uint16_t>(slot_),
917                                  lo_type,
918                                  high_type,
919                                  reinterpret_cast<uint64_t*>(&val_->j))) {
920           return ERR(OPAQUE_FRAME);
921         }
922         break;
923       }
924       default: {
925         LOG(FATAL) << "unexpected register type " << type_;
926         UNREACHABLE();
927       }
928     }
929     return OK;
930   }
931 
932  private:
933   art::Primitive::Type type_;
934   jvalue* val_;
935   // A global reference to the return value. We use the global reference to safely transfer the
936   // value between threads.
937   jobject obj_val_;
938 };
939 
GetLocalVariableGeneric(jvmtiEnv * env,jthread thread,jint depth,jint slot,art::Primitive::Type type,jvalue * val)940 jvmtiError MethodUtil::GetLocalVariableGeneric(jvmtiEnv* env,
941                                                jthread thread,
942                                                jint depth,
943                                                jint slot,
944                                                art::Primitive::Type type,
945                                                jvalue* val) {
946   if (depth < 0) {
947     return ERR(ILLEGAL_ARGUMENT);
948   }
949   art::Thread* self = art::Thread::Current();
950   art::ScopedObjectAccess soa(self);
951   art::Locks::thread_list_lock_->ExclusiveLock(self);
952   art::Thread* target = nullptr;
953   jvmtiError err = ERR(INTERNAL);
954   if (!ThreadUtil::GetAliveNativeThread(thread, soa, &target, &err)) {
955     art::Locks::thread_list_lock_->ExclusiveUnlock(self);
956     return err;
957   }
958   GetLocalVariableClosure c(env, depth, slot, type, val);
959   // RequestSynchronousCheckpoint releases the thread_list_lock_ as a part of its execution.
960   if (!target->RequestSynchronousCheckpoint(&c)) {
961     return ERR(THREAD_NOT_ALIVE);
962   } else {
963     return c.GetResult();
964   }
965 }
966 
967 class SetLocalVariableClosure : public CommonLocalVariableClosure {
968  public:
SetLocalVariableClosure(jvmtiEnv * jvmti,art::Thread * caller,jint depth,jint slot,art::Primitive::Type type,jvalue val)969   SetLocalVariableClosure(jvmtiEnv* jvmti,
970                           art::Thread* caller,
971                           jint depth,
972                           jint slot,
973                           art::Primitive::Type type,
974                           jvalue val)
975       : CommonLocalVariableClosure(jvmti, depth, slot), caller_(caller), type_(type), val_(val) {}
976 
977  protected:
978   jvmtiError
GetTypeError(art::ArtMethod * method,SlotType slot_type,const std::string & descriptor)979   GetTypeError(art::ArtMethod* method, SlotType slot_type, const std::string& descriptor) override
980       REQUIRES_SHARED(art::Locks::mutator_lock_) {
981     jvmtiError res = GetTypeErrorInner(method, slot_type, descriptor);
982     if (res != OK) {
983       if (res == ERR(TYPE_MISMATCH)) {
984         std::ostringstream desc_exp;
985         std::ostringstream desc_set;
986         if (type_ == art::Primitive::kPrimNot) {
987           desc_exp << " (type: " << descriptor << ")";
988           art::ObjPtr<art::mirror::Object> new_val(art::Thread::Current()->DecodeJObject(val_.l));
989           desc_set << " (type: "
990                   << (new_val.IsNull() ? "NULL" : new_val->GetClass()->PrettyDescriptor()) << ")";
991         }
992         JVMTI_LOG(INFO, jvmti_) << "Unable to Set local variable in slot " << slot_ << ". Expected"
993                                 << " slot to be of type compatible with " << SlotType{ type_ }
994                                 << desc_set.str() << " but slot is " << slot_type << desc_exp.str();
995       } else {
996         JVMTI_LOG(INFO, jvmti_) << "Unable to set local variable in slot " << slot_ << ". "
997                                 << err_.str();
998       }
999     }
1000     return res;
1001   }
1002 
1003   jvmtiError
GetTypeErrorInner(art::ArtMethod * method,SlotType slot_type,const std::string & descriptor)1004   GetTypeErrorInner(art::ArtMethod* method, SlotType slot_type, const std::string& descriptor)
1005       REQUIRES_SHARED(art::Locks::mutator_lock_) {
1006     switch (SquashType(SlotType{ type_ })) {
1007       case VerifierPrimitiveType::k32BitValue: {
1008         if (slot_type == SlotType{ VerifierPrimitiveType::kZeroValue }) {
1009           if (val_.i == 0) {
1010             return OK;
1011           } else {
1012             err_ << "Cannot determine if slot " << slot_ << " is a null reference or 32bit "
1013                  << "constant. Cannot allow writing to slot.";
1014             return ERR(INTERNAL);
1015           }
1016         } else if (SquashType(slot_type) != VerifierPrimitiveType::k32BitValue) {
1017           return ERR(TYPE_MISMATCH);
1018         } else if (slot_type == SlotType { VerifierPrimitiveType::k32BitValue } ||
1019                    slot_type == SlotType { type_ }) {
1020           return OK;
1021         } else if (type_ == art::Primitive::kPrimFloat ||
1022                    slot_type == SlotType { art::Primitive::kPrimFloat }) {
1023           // we should have hit the get == type_ above
1024           return ERR(TYPE_MISMATCH);
1025         } else {
1026           // Some smaller type then int.
1027           return OK;
1028         }
1029       }
1030       case VerifierPrimitiveType::k64BitValue: {
1031         if (slot_type == SlotType { VerifierPrimitiveType::k64BitValue } ||
1032             slot_type == SlotType { type_ }) {
1033           return OK;
1034         } else {
1035           return ERR(TYPE_MISMATCH);
1036         }
1037       }
1038       case VerifierPrimitiveType::kReferenceValue: {
1039         if (SquashType(slot_type) != VerifierPrimitiveType::kReferenceValue &&
1040             SquashType(slot_type) != VerifierPrimitiveType::kZeroValue) {
1041           return ERR(TYPE_MISMATCH);
1042         } else if (val_.l == nullptr) {
1043           return OK;
1044         } else if (slot_type == SlotType { VerifierPrimitiveType::kZeroValue }) {
1045           err_ << "Cannot determine if slot " << slot_ << " is a null "
1046                << "reference or 32bit constant. Cannot allow writing to slot.";
1047           return ERR(INTERNAL);
1048         } else {
1049           art::ClassLinker* cl = art::Runtime::Current()->GetClassLinker();
1050           art::ObjPtr<art::mirror::Class> set_class = caller_->DecodeJObject(val_.l)->GetClass();
1051           art::ObjPtr<art::mirror::ClassLoader> loader =
1052               method->GetDeclaringClass()->GetClassLoader();
1053           art::ObjPtr<art::mirror::Class> slot_class =
1054               cl->LookupClass(caller_, descriptor.c_str(), loader);
1055           DCHECK(!slot_class.IsNull()) << descriptor << " slot: " << slot_type;
1056           return slot_class->IsAssignableFrom(set_class) ? OK : ERR(TYPE_MISMATCH);
1057         }
1058       }
1059       case VerifierPrimitiveType::kZeroValue: {
1060         LOG(FATAL) << "Illegal result from SquashType of art::Primitive::Type " << type_;
1061         UNREACHABLE();
1062       }
1063     }
1064   }
1065 
Execute(art::ArtMethod * method,art::StackVisitor & visitor)1066   jvmtiError Execute(art::ArtMethod* method, art::StackVisitor& visitor)
1067       override REQUIRES_SHARED(art::Locks::mutator_lock_) {
1068     switch (type_) {
1069       case art::Primitive::kPrimNot: {
1070         if (!visitor.SetVRegReference(method,
1071                                       static_cast<uint16_t>(slot_),
1072                                       caller_->DecodeJObject(val_.l))) {
1073           return ERR(OPAQUE_FRAME);
1074         }
1075         break;
1076       }
1077       case art::Primitive::kPrimInt:
1078       case art::Primitive::kPrimFloat: {
1079         if (!visitor.SetVReg(method,
1080                              static_cast<uint16_t>(slot_),
1081                              static_cast<uint32_t>(val_.i),
1082                              type_ == art::Primitive::kPrimFloat ? art::kFloatVReg
1083                                                                  : art::kIntVReg)) {
1084           return ERR(OPAQUE_FRAME);
1085         }
1086         break;
1087       }
1088       case art::Primitive::kPrimDouble:
1089       case art::Primitive::kPrimLong: {
1090         auto lo_type = type_ == art::Primitive::kPrimLong ? art::kLongLoVReg : art::kDoubleLoVReg;
1091         auto high_type = type_ == art::Primitive::kPrimLong ? art::kLongHiVReg : art::kDoubleHiVReg;
1092         if (!visitor.SetVRegPair(method,
1093                                  static_cast<uint16_t>(slot_),
1094                                  static_cast<uint64_t>(val_.j),
1095                                  lo_type,
1096                                  high_type)) {
1097           return ERR(OPAQUE_FRAME);
1098         }
1099         break;
1100       }
1101       default: {
1102         LOG(FATAL) << "unexpected register type " << type_;
1103         UNREACHABLE();
1104       }
1105     }
1106     return OK;
1107   }
1108 
1109  private:
1110   art::Thread* caller_;
1111   art::Primitive::Type type_;
1112   jvalue val_;
1113   std::ostringstream err_;
1114 };
1115 
SetLocalVariableGeneric(jvmtiEnv * env,jthread thread,jint depth,jint slot,art::Primitive::Type type,jvalue val)1116 jvmtiError MethodUtil::SetLocalVariableGeneric(jvmtiEnv* env,
1117                                                jthread thread,
1118                                                jint depth,
1119                                                jint slot,
1120                                                art::Primitive::Type type,
1121                                                jvalue val) {
1122   if (depth < 0) {
1123     return ERR(ILLEGAL_ARGUMENT);
1124   }
1125   // Make sure that we know not to do any OSR anymore.
1126   // TODO We should really keep track of this at the Frame granularity.
1127   DeoptManager::Get()->SetLocalsUpdated();
1128   art::Thread* self = art::Thread::Current();
1129   art::ScopedObjectAccess soa(self);
1130   art::Locks::thread_list_lock_->ExclusiveLock(self);
1131   art::Thread* target = nullptr;
1132   jvmtiError err = ERR(INTERNAL);
1133   if (!ThreadUtil::GetAliveNativeThread(thread, soa, &target, &err)) {
1134     art::Locks::thread_list_lock_->ExclusiveUnlock(self);
1135     return err;
1136   }
1137   SetLocalVariableClosure c(env, self, depth, slot, type, val);
1138   // RequestSynchronousCheckpoint releases the thread_list_lock_ as a part of its execution.
1139   if (!target->RequestSynchronousCheckpoint(&c)) {
1140     return ERR(THREAD_NOT_ALIVE);
1141   } else {
1142     return c.GetResult();
1143   }
1144 }
1145 
1146 class GetLocalInstanceClosure : public art::Closure {
1147  public:
GetLocalInstanceClosure(jint depth)1148   explicit GetLocalInstanceClosure(jint depth)
1149       : result_(ERR(INTERNAL)),
1150         depth_(depth),
1151         val_(nullptr) {}
1152 
Run(art::Thread * self)1153   void Run(art::Thread* self) override REQUIRES(art::Locks::mutator_lock_) {
1154     art::ScopedAssertNoThreadSuspension sants("GetLocalInstanceClosure::Run");
1155     art::Locks::mutator_lock_->AssertSharedHeld(art::Thread::Current());
1156     std::unique_ptr<art::Context> context(art::Context::Create());
1157     FindFrameAtDepthVisitor visitor(self, context.get(), depth_);
1158     visitor.WalkStack();
1159     if (!visitor.FoundFrame()) {
1160       // Must have been a bad depth.
1161       result_ = ERR(NO_MORE_FRAMES);
1162       return;
1163     }
1164     result_ = OK;
1165     val_ = art::GcRoot<art::mirror::Object>(visitor.GetThisObject());
1166   }
1167 
GetResult(jobject * data_out)1168   jvmtiError GetResult(jobject* data_out) REQUIRES_SHARED(art::Locks::mutator_lock_) {
1169     if (result_ == OK) {
1170       *data_out = val_.IsNull()
1171           ? nullptr
1172           : art::Thread::Current()->GetJniEnv()->AddLocalReference<jobject>(val_.Read());
1173     }
1174     return result_;
1175   }
1176 
1177  private:
1178   jvmtiError result_;
1179   jint depth_;
1180   art::GcRoot<art::mirror::Object> val_;
1181 };
1182 
GetLocalInstance(jvmtiEnv * env ATTRIBUTE_UNUSED,jthread thread,jint depth,jobject * data)1183 jvmtiError MethodUtil::GetLocalInstance(jvmtiEnv* env ATTRIBUTE_UNUSED,
1184                                         jthread thread,
1185                                         jint depth,
1186                                         jobject* data) {
1187   if (depth < 0) {
1188     return ERR(ILLEGAL_ARGUMENT);
1189   }
1190   art::Thread* self = art::Thread::Current();
1191   art::ScopedObjectAccess soa(self);
1192   art::Locks::thread_list_lock_->ExclusiveLock(self);
1193   art::Thread* target = nullptr;
1194   jvmtiError err = ERR(INTERNAL);
1195   if (!ThreadUtil::GetAliveNativeThread(thread, soa, &target, &err)) {
1196     art::Locks::thread_list_lock_->ExclusiveUnlock(self);
1197     return err;
1198   }
1199   art::ScopedAssertNoThreadSuspension sants("Performing GetLocalInstance");
1200   GetLocalInstanceClosure c(depth);
1201   // RequestSynchronousCheckpoint releases the thread_list_lock_ as a part of its execution.  We
1202   // need to avoid suspending as we wait for the checkpoint to occur since we are (potentially)
1203   // transfering a GcRoot across threads.
1204   if (!target->RequestSynchronousCheckpoint(&c, art::ThreadState::kRunnable)) {
1205     return ERR(THREAD_NOT_ALIVE);
1206   } else {
1207     return c.GetResult(data);
1208   }
1209 }
1210 
1211 #define FOR_JVMTI_JVALUE_TYPES(fn) \
1212     fn(jint, art::Primitive::kPrimInt, i) \
1213     fn(jlong, art::Primitive::kPrimLong, j) \
1214     fn(jfloat, art::Primitive::kPrimFloat, f) \
1215     fn(jdouble, art::Primitive::kPrimDouble, d) \
1216     fn(jobject, art::Primitive::kPrimNot, l)
1217 
1218 namespace impl {
1219 
1220 template<typename T> void WriteJvalue(T, jvalue*);
1221 template<typename T> void ReadJvalue(jvalue, T*);
1222 template<typename T> art::Primitive::Type GetJNIType();
1223 
1224 #define JNI_TYPE_CHAR(type, prim, id) \
1225 template<> art::Primitive::Type GetJNIType<type>() { \
1226   return prim; \
1227 }
1228 
1229 FOR_JVMTI_JVALUE_TYPES(JNI_TYPE_CHAR);
1230 
1231 #undef JNI_TYPE_CHAR
1232 
1233 #define RW_JVALUE(srctype, prim, id) \
1234     template<> void ReadJvalue<srctype>(jvalue in, std::add_pointer<srctype>::type out) { \
1235       *out = in.id; \
1236     } \
1237     template<> void WriteJvalue<srctype>(srctype in, jvalue* out) { \
1238       out->id = in; \
1239     }
1240 
1241 FOR_JVMTI_JVALUE_TYPES(RW_JVALUE);
1242 
1243 #undef RW_JVALUE
1244 
1245 }  // namespace impl
1246 
1247 template<typename T>
SetLocalVariable(jvmtiEnv * env,jthread thread,jint depth,jint slot,T data)1248 jvmtiError MethodUtil::SetLocalVariable(jvmtiEnv* env,
1249                                         jthread thread,
1250                                         jint depth,
1251                                         jint slot,
1252                                         T data) {
1253   jvalue v = {.j = 0};
1254   art::Primitive::Type type = impl::GetJNIType<T>();
1255   impl::WriteJvalue(data, &v);
1256   return SetLocalVariableGeneric(env, thread, depth, slot, type, v);
1257 }
1258 
1259 template<typename T>
GetLocalVariable(jvmtiEnv * env,jthread thread,jint depth,jint slot,T * data)1260 jvmtiError MethodUtil::GetLocalVariable(jvmtiEnv* env,
1261                                         jthread thread,
1262                                         jint depth,
1263                                         jint slot,
1264                                         T* data) {
1265   if (data == nullptr) {
1266     return ERR(NULL_POINTER);
1267   }
1268   jvalue v = {.j = 0};
1269   art::Primitive::Type type = impl::GetJNIType<T>();
1270   jvmtiError err = GetLocalVariableGeneric(env, thread, depth, slot, type, &v);
1271   if (err != OK) {
1272     return err;
1273   } else {
1274     impl::ReadJvalue(v, data);
1275     return OK;
1276   }
1277 }
1278 
1279 #define GET_SET_LV(srctype, prim, id) \
1280     template jvmtiError MethodUtil::GetLocalVariable<srctype>(jvmtiEnv*, \
1281                                                               jthread, \
1282                                                               jint, \
1283                                                               jint, \
1284                                                               std::add_pointer<srctype>::type); \
1285     template jvmtiError MethodUtil::SetLocalVariable<srctype>(jvmtiEnv*, \
1286                                                               jthread, \
1287                                                               jint, \
1288                                                               jint, \
1289                                                               srctype);
1290 
1291 FOR_JVMTI_JVALUE_TYPES(GET_SET_LV);
1292 
1293 #undef GET_SET_LV
1294 
1295 #undef FOR_JVMTI_JVALUE_TYPES
1296 
1297 }  // namespace openjdkjvmti
1298