arc: enable Android tracing in verified-boot mode

chrome/browser/chromeos/arc/trace/arc_trace_bridge.cc

+// Copyright 2016 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 "chrome/browser/chromeos/arc/trace/arc_trace_bridge.h"
+
+#include <utility>
+
+#include "base/logging.h"
+#include "components/arc/arc_bridge_service.h"
+#include "components/arc/arc_service_manager.h"
+
+namespace arc {
+
+namespace {
+
+// This class is owned by ArcServiceManager so that it is safe to use this raw
+// pointer as the singleton reference.
+ArcTraceBridge* g_arc_trace_bridge = nullptr;
+
+// The prefix of the categories to be shown on the trace selection UI.
+constexpr char kCategoryPrefix[] = TRACE_DISABLED_BY_DEFAULT("android ");
+
+}  // namcespace

[Luis Héctor Chávez, 2016/12/27 19:15:58]

nit: s/namcespace/namespace/

[shunhsingou, 2016/12/28 09:15:15]

Done.

+
+ArcTraceBridge* ArcTraceBridge::Get() {
+  return g_arc_trace_bridge;
+}
+
+ArcTraceBridge::ArcTraceBridge(ArcBridgeService* bridge_service)
+    : ArcService(bridge_service), weak_ptr_factory_(this) {
+  arc_bridge_service()->trace()->AddObserver(this);
+  g_arc_trace_bridge = this;

[Luis Héctor Chávez, 2016/12/27 19:15:58]

DCHECK(!g_arc_trace_bridge);

[shunhsingou, 2016/12/28 09:15:15]

Done.

+}
+
+ArcTraceBridge::~ArcTraceBridge() {
+  arc_bridge_service()->trace()->RemoveObserver(this);
+  g_arc_trace_bridge = nullptr;
+}
+
+void ArcTraceBridge::OnInstanceReady() {
+  QueryAvailableCategories();
+}
+
+void ArcTraceBridge::QueryAvailableCategories() {
+  auto* trace_instance = arc_bridge_service()->trace()->GetInstanceForMethod(
+      "QueryAvailableCategories");
+  if (!trace_instance)
+    return;
+  trace_instance->QueryAvailableCategories(base::Bind(
+      &ArcTraceBridge::OnCategoriesReady, weak_ptr_factory_.GetWeakPtr()));
+}
+
+void ArcTraceBridge::OnCategoriesReady(
+    const std::vector<std::string>& categories) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+

[Luis Héctor Chávez, 2016/12/27 19:15:58]

Do you need |categories_.clear();| here?

[shunhsingou, 2016/12/28 09:15:15]

Done.

+  for (const auto& category : categories) {
+    categories_.push_back(
+        Category{category, std::string(kCategoryPrefix) + category});
+    // Show the category name in the selection UI.
+    base::trace_event::TraceLog::GetCategoryGroupEnabled(
+        categories_.back().full_name.c_str());
+  }
+}
+
+void ArcTraceBridge::StartTracing(
+    const base::trace_event::TraceConfig& trace_config) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  auto* trace_instance =
+      arc_bridge_service()->trace()->GetInstanceForMethod("StartTracing");
+  if (!trace_instance)
+    return;
+
+  std::vector<std::string> selected_categories;
+  for (const auto& category : categories_) {
+    if (trace_config.IsCategoryGroupEnabled(category.full_name.c_str()))
+      selected_categories.push_back(category.name);
+  }
+
+  trace_instance->StartTracing(selected_categories,
+                               base::Bind(&ArcTraceBridge::OnTraceStart,
+                                          weak_ptr_factory_.GetWeakPtr()));
+}
+
+void ArcTraceBridge::StopTracing() {
+  auto* trace_instance =
+      arc_bridge_service()->trace()->GetInstanceForMethod("StopTracing");
+  if (!trace_instance)
+    return;
+  trace_instance->StopTracing(
+      base::Bind(&ArcTraceBridge::OnTraceStop, weak_ptr_factory_.GetWeakPtr()));
+}
+
+void ArcTraceBridge::OnTraceStart(bool success) {
+  if (!success)
+    LOG(ERROR) << "Failed to start tracing in the container.";
+}
+
+void ArcTraceBridge::OnTraceStop(bool success) {
+  if (!success)
+    LOG(ERROR) << "Failed to stop tracing in the container.";
+}
+
+}  // namespace arc