src/components/tracing/child_trace_message_filter.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   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 "components/tracing/child_trace_message_filter.h"
   6
   7 #include "base/debug/trace_event.h"
   8 #include "base/message_loop/message_loop_proxy.h"
   9 #include "components/tracing/tracing_messages.h"
  10 #include "ipc/ipc_channel.h"
  11
  12 using base::debug::TraceLog;
  13
  14 namespace tracing {
  15
  16 ChildTraceMessageFilter::ChildTraceMessageFilter(
  17     base::MessageLoopProxy* ipc_message_loop)
  18     : sender_(NULL),
  19       ipc_message_loop_(ipc_message_loop) {}
  20
  21 void ChildTraceMessageFilter::OnFilterAdded(IPC::Sender* sender) {
  22   sender_ = sender;
  23   sender_->Send(new TracingHostMsg_ChildSupportsTracing());
  24 }
  25
  26 void ChildTraceMessageFilter::OnFilterRemoved() {
  27   sender_ = NULL;
  28 }
  29
  30 bool ChildTraceMessageFilter::OnMessageReceived(const IPC::Message& message) {
  31   bool handled = true;
  32   IPC_BEGIN_MESSAGE_MAP(ChildTraceMessageFilter, message)
  33     IPC_MESSAGE_HANDLER(TracingMsg_BeginTracing, OnBeginTracing)
  34     IPC_MESSAGE_HANDLER(TracingMsg_EndTracing, OnEndTracing)
  35     IPC_MESSAGE_HANDLER(TracingMsg_EnableMonitoring, OnEnableMonitoring)
  36     IPC_MESSAGE_HANDLER(TracingMsg_DisableMonitoring, OnDisableMonitoring)
  37     IPC_MESSAGE_HANDLER(TracingMsg_CaptureMonitoringSnapshot,
  38                         OnCaptureMonitoringSnapshot)
  39     IPC_MESSAGE_HANDLER(TracingMsg_GetTraceBufferPercentFull,
  40                         OnGetTraceBufferPercentFull)
  41     IPC_MESSAGE_HANDLER(TracingMsg_SetWatchEvent, OnSetWatchEvent)
  42     IPC_MESSAGE_HANDLER(TracingMsg_CancelWatchEvent, OnCancelWatchEvent)
  43     IPC_MESSAGE_UNHANDLED(handled = false)
  44   IPC_END_MESSAGE_MAP()
  45   return handled;
  46 }
  47
  48 ChildTraceMessageFilter::~ChildTraceMessageFilter() {}
  49
  50 void ChildTraceMessageFilter::OnBeginTracing(
  51     const std::string& category_filter_str,
  52     base::TimeTicks browser_time,
  53     const std::string& options) {
  54 #if defined(__native_client__)
  55   // NaCl and system times are offset by a bit, so subtract some time from
  56   // the captured timestamps. The value might be off by a bit due to messaging
  57   // latency.
  58   base::TimeDelta time_offset = base::TimeTicks::NowFromSystemTraceTime() -
  59       browser_time;
  60   TraceLog::GetInstance()->SetTimeOffset(time_offset);
  61 #endif
  62   base::debug::TraceOptions trace_options;
  63   trace_options.SetFromString(options);
  64   TraceLog::GetInstance()->SetEnabled(
  65       base::debug::CategoryFilter(category_filter_str),
  66       base::debug::TraceLog::RECORDING_MODE,
  67       trace_options);
  68 }
  69
  70 void ChildTraceMessageFilter::OnEndTracing() {
  71   TraceLog::GetInstance()->SetDisabled();
  72
  73   // Flush will generate one or more callbacks to OnTraceDataCollected
  74   // synchronously or asynchronously. EndTracingAck will be sent in the last
  75   // OnTraceDataCollected. We are already on the IO thread, so the
  76   // OnTraceDataCollected calls will not be deferred.
  77   TraceLog::GetInstance()->Flush(
  78       base::Bind(&ChildTraceMessageFilter::OnTraceDataCollected, this));
  79 }
  80
  81 void ChildTraceMessageFilter::OnEnableMonitoring(
  82     const std::string& category_filter_str,
  83     base::TimeTicks browser_time,
  84     const std::string& options) {
  85   base::debug::TraceOptions trace_options;
  86   trace_options.SetFromString(options);
  87   TraceLog::GetInstance()->SetEnabled(
  88       base::debug::CategoryFilter(category_filter_str),
  89       base::debug::TraceLog::MONITORING_MODE,
  90       trace_options);
  91 }
  92
  93 void ChildTraceMessageFilter::OnDisableMonitoring() {
  94   TraceLog::GetInstance()->SetDisabled();
  95 }
  96
  97 void ChildTraceMessageFilter::OnCaptureMonitoringSnapshot() {
  98   // Flush will generate one or more callbacks to
  99   // OnMonitoringTraceDataCollected. It's important that the last
 100   // OnMonitoringTraceDataCollected gets called before
 101   // CaptureMonitoringSnapshotAck below. We are already on the IO thread,
 102   // so the OnMonitoringTraceDataCollected calls will not be deferred.
 103   TraceLog::GetInstance()->FlushButLeaveBufferIntact(
 104       base::Bind(&ChildTraceMessageFilter::OnMonitoringTraceDataCollected,
 105                  this));
 106 }
 107
 108 void ChildTraceMessageFilter::OnGetTraceBufferPercentFull() {
 109   float bpf = TraceLog::GetInstance()->GetBufferPercentFull();
 110
 111   sender_->Send(new TracingHostMsg_TraceBufferPercentFullReply(bpf));
 112 }
 113
 114 void ChildTraceMessageFilter::OnSetWatchEvent(const std::string& category_name,
 115                                               const std::string& event_name) {
 116   TraceLog::GetInstance()->SetWatchEvent(
 117       category_name, event_name,
 118       base::Bind(&ChildTraceMessageFilter::OnWatchEventMatched, this));
 119 }
 120
 121 void ChildTraceMessageFilter::OnCancelWatchEvent() {
 122   TraceLog::GetInstance()->CancelWatchEvent();
 123 }
 124
 125 void ChildTraceMessageFilter::OnWatchEventMatched() {
 126   if (!ipc_message_loop_->BelongsToCurrentThread()) {
 127     ipc_message_loop_->PostTask(FROM_HERE,
 128         base::Bind(&ChildTraceMessageFilter::OnWatchEventMatched, this));
 129     return;
 130   }
 131   sender_->Send(new TracingHostMsg_WatchEventMatched);
 132 }
 133
 134 void ChildTraceMessageFilter::OnTraceDataCollected(
 135     const scoped_refptr<base::RefCountedString>& events_str_ptr,
 136     bool has_more_events) {
 137   if (!ipc_message_loop_->BelongsToCurrentThread()) {
 138     ipc_message_loop_->PostTask(FROM_HERE,
 139         base::Bind(&ChildTraceMessageFilter::OnTraceDataCollected, this,
 140                    events_str_ptr, has_more_events));
 141     return;
 142   }
 143   if (events_str_ptr->data().size()) {
 144     sender_->Send(new TracingHostMsg_TraceDataCollected(
 145         events_str_ptr->data()));
 146   }
 147   if (!has_more_events) {
 148     std::vector<std::string> category_groups;
 149     TraceLog::GetInstance()->GetKnownCategoryGroups(&category_groups);
 150     sender_->Send(new TracingHostMsg_EndTracingAck(category_groups));
 151   }
 152 }
 153
 154 void ChildTraceMessageFilter::OnMonitoringTraceDataCollected(
 155      const scoped_refptr<base::RefCountedString>& events_str_ptr,
 156      bool has_more_events) {
 157   if (!ipc_message_loop_->BelongsToCurrentThread()) {
 158     ipc_message_loop_->PostTask(FROM_HERE,
 159         base::Bind(&ChildTraceMessageFilter::
 160                    OnMonitoringTraceDataCollected,
 161                    this,
 162                    events_str_ptr,
 163                    has_more_events));
 164     return;
 165   }
 166   sender_->Send(new TracingHostMsg_MonitoringTraceDataCollected(
 167       events_str_ptr->data()));
 168
 169   if (!has_more_events)
 170     sender_->Send(new TracingHostMsg_CaptureMonitoringSnapshotAck());
 171 }
 172
 173 }  // namespace tracing