Enable tracing in SkyShell

sky/shell/tracing_controller.cc

+// Copyright 2015 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 "sky/shell/tracing_controller.h"
+
+#include "base/android/jni_android.h"
+#include "base/android/jni_string.h"
+#include "base/files/file_util.h"
+#include "base/macros.h"
+#include "base/trace_event/trace_event.h"
+#include "jni/TracingController_jni.h"
+
+namespace sky {
+namespace shell {
+namespace {
+
+const char kStart[] = "{\"traceEvents\":[";
+const char kEnd[] = "]}";
+
+static FILE* g_file = NULL;
+
+void Write(const std::string& data) {
+  ignore_result(fwrite(data.data(), data.length(), 1, g_file));
+}
+
+void HandleChunk(const scoped_refptr<base::RefCountedString>& chunk,
+                 bool has_more_events) {
+  Write(chunk->data());
+  if (has_more_events)
+    Write(",");
+
+  if (!has_more_events) {
+    Write(kEnd);
+    base::CloseFile(g_file);
+    g_file = NULL;
+
+    LOG(INFO) << "Trace complete";
+  }
+}
+
+}  // namespace
+
+static void StartTracing(JNIEnv* env, jclass clazz) {
+  LOG(INFO) << "Starting trace";
+
+  base::trace_event::TraceLog::GetInstance()->SetEnabled(
+      base::trace_event::CategoryFilter("*"),
+      base::trace_event::TraceLog::RECORDING_MODE,
+      base::trace_event::TraceOptions(base::trace_event::RECORD_UNTIL_FULL));
+}
+
+static void StopTracing(JNIEnv* env, jclass clazz, jstring path) {
+  base::trace_event::TraceLog::GetInstance()->SetDisabled();
+
+  base::FilePath file_path(base::android::ConvertJavaStringToUTF8(env, path));
+  g_file = base::OpenFile(file_path, "w");
+  CHECK(g_file) << "Failed to open file " << file_path.LossyDisplayName();
+
+  LOG(INFO) << "Saving trace to " << file_path.LossyDisplayName();
+  Write(kStart);
+  base::trace_event::TraceLog::GetInstance()->Flush(base::Bind(&HandleChunk));
+}
+
+bool RegisterTracingController(JNIEnv* env) {
+  return RegisterNativesImpl(env);
+}
+
+}  // namespace shell
+}  // namespace sky