base/profiler/stack_sampling_profiler_java_test_util.cc

   1 // Copyright 2019 The Chromium Authors
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/profiler/stack_sampling_profiler_java_test_util.h"
   6
   7 #include "base/base_profiler_test_support_jni/TestSupport_jni.h"
   8 #include "base/location.h"
   9
  10 namespace base {
  11
  12 namespace {
  13
  14 struct UnwinderJavaTestSupportParams {
  15   OnceClosure closure;
  16   FunctionAddressRange range;
  17 };
  18
  19 }  // namespace
  20
  21 void JNI_TestSupport_InvokeCallbackFunction(JNIEnv* env, jlong context) {
  22   const void* start_program_counter = GetProgramCounter();
  23
  24   UnwinderJavaTestSupportParams* params =
  25       reinterpret_cast<UnwinderJavaTestSupportParams*>(context);
  26   if (!params->closure.is_null()) {
  27     std::move(params->closure).Run();
  28   }
  29
  30   // Volatile to prevent a tail call to GetProgramCounter().
  31   const void* volatile end_program_counter = GetProgramCounter();
  32
  33   params->range = {start_program_counter, end_program_counter};
  34 }
  35
  36 FunctionAddressRange callWithJavaFunction(OnceClosure closure) {
  37   JNIEnv* env = base::android::AttachCurrentThread();
  38   UnwinderJavaTestSupportParams params{std::move(closure), {}};
  39   base::Java_TestSupport_callWithJavaFunction(
  40       env, reinterpret_cast<uintptr_t>(&params));
  41   return params.range;
  42 }
  43
  44 }  // namespace base