| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/android/trace_event_binding.h" |
| |
| #include <jni.h> |
| |
| #include <set> |
| |
| #include "base/debug/trace_event.h" |
| #include "base/debug/trace_event_impl.h" |
| #include "base/lazy_instance.h" |
| #include "jni/TraceEvent_jni.h" |
| |
| namespace base { |
| namespace android { |
| |
| namespace { |
| |
| const char kJavaCategory[] = "Java"; |
| const char kToplevelCategory[] = "toplevel"; |
| const char kLooperDispatchMessage[] = "Looper.dispatchMessage"; |
| |
| // Boilerplate for safely converting Java data to TRACE_EVENT data. |
| class TraceEventDataConverter { |
| public: |
| TraceEventDataConverter(JNIEnv* env, |
| jstring jname, |
| jstring jarg) |
| : env_(env), |
| jname_(jname), |
| jarg_(jarg), |
| name_(env->GetStringUTFChars(jname, NULL)), |
| arg_(jarg ? env->GetStringUTFChars(jarg, NULL) : NULL) { |
| } |
| ~TraceEventDataConverter() { |
| env_->ReleaseStringUTFChars(jname_, name_); |
| if (jarg_) |
| env_->ReleaseStringUTFChars(jarg_, arg_); |
| } |
| |
| // Return saves values to pass to TRACE_EVENT macros. |
| const char* name() { return name_; } |
| const char* arg_name() { return arg_ ? "arg" : NULL; } |
| const char* arg() { return arg_; } |
| |
| private: |
| JNIEnv* env_; |
| jstring jname_; |
| jstring jarg_; |
| const char* name_; |
| const char* arg_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TraceEventDataConverter); |
| }; |
| |
| class TraceEnabledObserver : public debug::TraceLog::EnabledStateObserver { |
| public: |
| virtual void OnTraceLogEnabled() override { |
| JNIEnv* env = base::android::AttachCurrentThread(); |
| base::android::Java_TraceEvent_setEnabled(env, true); |
| } |
| virtual void OnTraceLogDisabled() override { |
| JNIEnv* env = base::android::AttachCurrentThread(); |
| base::android::Java_TraceEvent_setEnabled(env, false); |
| } |
| }; |
| |
| base::LazyInstance<TraceEnabledObserver>::Leaky g_trace_enabled_state_observer_; |
| |
| } // namespace |
| |
| static void RegisterEnabledObserver(JNIEnv* env, jclass clazz) { |
| bool enabled = debug::TraceLog::GetInstance()->IsEnabled(); |
| base::android::Java_TraceEvent_setEnabled(env, enabled); |
| debug::TraceLog::GetInstance()->AddEnabledStateObserver( |
| g_trace_enabled_state_observer_.Pointer()); |
| } |
| |
| static void StartATrace(JNIEnv* env, jclass clazz) { |
| base::debug::TraceLog::GetInstance()->StartATrace(); |
| } |
| |
| static void StopATrace(JNIEnv* env, jclass clazz) { |
| base::debug::TraceLog::GetInstance()->StopATrace(); |
| } |
| |
| static void Instant(JNIEnv* env, jclass clazz, |
| jstring jname, jstring jarg) { |
| TraceEventDataConverter converter(env, jname, jarg); |
| if (converter.arg()) { |
| TRACE_EVENT_COPY_INSTANT1(kJavaCategory, converter.name(), |
| TRACE_EVENT_SCOPE_THREAD, |
| converter.arg_name(), converter.arg()); |
| } else { |
| TRACE_EVENT_COPY_INSTANT0(kJavaCategory, converter.name(), |
| TRACE_EVENT_SCOPE_THREAD); |
| } |
| } |
| |
| static void Begin(JNIEnv* env, jclass clazz, |
| jstring jname, jstring jarg) { |
| TraceEventDataConverter converter(env, jname, jarg); |
| if (converter.arg()) { |
| TRACE_EVENT_COPY_BEGIN1(kJavaCategory, converter.name(), |
| converter.arg_name(), converter.arg()); |
| } else { |
| TRACE_EVENT_COPY_BEGIN0(kJavaCategory, converter.name()); |
| } |
| } |
| |
| static void End(JNIEnv* env, jclass clazz, |
| jstring jname, jstring jarg) { |
| TraceEventDataConverter converter(env, jname, jarg); |
| if (converter.arg()) { |
| TRACE_EVENT_COPY_END1(kJavaCategory, converter.name(), |
| converter.arg_name(), converter.arg()); |
| } else { |
| TRACE_EVENT_COPY_END0(kJavaCategory, converter.name()); |
| } |
| } |
| |
| static void BeginToplevel(JNIEnv* env, jclass clazz) { |
| TRACE_EVENT_BEGIN0(kToplevelCategory, kLooperDispatchMessage); |
| } |
| |
| static void EndToplevel(JNIEnv* env, jclass clazz) { |
| TRACE_EVENT_END0(kToplevelCategory, kLooperDispatchMessage); |
| } |
| |
| static void StartAsync(JNIEnv* env, jclass clazz, |
| jstring jname, jlong jid, jstring jarg) { |
| TraceEventDataConverter converter(env, jname, jarg); |
| if (converter.arg()) { |
| TRACE_EVENT_COPY_ASYNC_BEGIN1(kJavaCategory, |
| converter.name(), |
| jid, |
| converter.arg_name(), |
| converter.arg()); |
| } else { |
| TRACE_EVENT_COPY_ASYNC_BEGIN0(kJavaCategory, |
| converter.name(), |
| jid); |
| } |
| } |
| |
| static void FinishAsync(JNIEnv* env, jclass clazz, |
| jstring jname, jlong jid, jstring jarg) { |
| TraceEventDataConverter converter(env, jname, jarg); |
| if (converter.arg()) { |
| TRACE_EVENT_COPY_ASYNC_END1(kJavaCategory, |
| converter.name(), |
| jid, |
| converter.arg_name(), |
| converter.arg()); |
| } else { |
| TRACE_EVENT_COPY_ASYNC_END0(kJavaCategory, |
| converter.name(), |
| jid); |
| } |
| } |
| |
| bool RegisterTraceEvent(JNIEnv* env) { |
| return RegisterNativesImpl(env); |
| } |
| |
| } // namespace android |
| } // namespace base |