Add a mechanism to let TracingController invoke JavaScript via TracingUI

content/browser/tracing/tracing_controller_impl.cc

Index: content/browser/tracing/tracing_controller_impl.cc
diff --git a/content/browser/tracing/tracing_controller_impl.cc b/content/browser/tracing/tracing_controller_impl.cc
index 8bca4a0c2d3ca535d5d529b8e7f559855a3f0a08..185eb988d5c7eefb1711ee97891e404a3232880c 100644
--- a/content/browser/tracing/tracing_controller_impl.cc
+++ b/content/browser/tracing/tracing_controller_impl.cc
@@ -10,6 +10,7 @@
 #include "base/json/string_escape.h"
 #include "base/strings/string_number_conversions.h"
 #include "content/browser/tracing/trace_message_filter.h"
+#include "content/browser/tracing/tracing_ui.h"
 #include "content/common/child_process_messages.h"
 #include "content/public/browser/browser_message_filter.h"
 #include "content/public/common/content_switches.h"
@@ -291,7 +292,7 @@ bool TracingControllerImpl::EnableMonitoring(
 
   if (!can_enable_monitoring())
     return false;
-  is_monitoring_ = true;
+  OnMonitoringStateChanged(true);
 
 #if defined(OS_ANDROID)
   TraceLog::GetInstance()->AddClockSyncMetadataEvent();
@@ -353,7 +354,7 @@ void TracingControllerImpl::OnDisableMonitoringDone(
     const DisableMonitoringDoneCallback& callback) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
-  is_monitoring_ = false;
+  OnMonitoringStateChanged(false);
 
   // Notify all child processes.
   for (TraceMessageFilterSet::iterator it = trace_message_filters_.begin();
@@ -782,4 +783,32 @@ void TracingControllerImpl::OnWatchEventMatched() {
     watch_event_callback_.Run();
 }
 
+void TracingControllerImpl::RegisterTracingUI(TracingUI* tracing_ui)
+{
+  DCHECK(tracing_uis_.find(tracing_ui) == tracing_uis_.end());
+  tracing_uis_.insert(tracing_ui);
+  tracing_ui->OnMonitoringStateChanged(is_monitoring_);
+}
+
+void TracingControllerImpl::UnregisterTracingUI(TracingUI* tracing_ui)
+{
+  std::set<TracingUI*>::iterator it = tracing_uis_.find(tracing_ui);
+  DCHECK(it != tracing_uis_.end());
+  tracing_uis_.erase(it);
+}
+
+void TracingControllerImpl::OnMonitoringStateChanged(bool is_monitoring)
+{
+  if (is_monitoring_ == is_monitoring)
+    return;
+
+  is_monitoring_ = is_monitoring;
+#if !defined(OS_ANDROID)
+  for (std::set<TracingUI*>::iterator it = tracing_uis_.begin();
+       it != tracing_uis_.end(); it++) {
+    (*it)->OnMonitoringStateChanged(is_monitoring);
+  }
+#endif
+}
+
 }  // namespace content