| // Copyright (c) 2012 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 "components/tracing/child_trace_message_filter.h" |
| |
| #include "base/debug/trace_event.h" |
| #include "base/message_loop/message_loop_proxy.h" |
| #include "components/tracing/tracing_messages.h" |
| |
| using base::debug::TraceLog; |
| |
| namespace tracing { |
| |
| ChildTraceMessageFilter::ChildTraceMessageFilter( |
| base::MessageLoopProxy* ipc_message_loop) |
| : channel_(NULL), |
| ipc_message_loop_(ipc_message_loop) {} |
| |
| void ChildTraceMessageFilter::OnFilterAdded(IPC::Channel* channel) { |
| channel_ = channel; |
| channel_->Send(new TracingHostMsg_ChildSupportsTracing()); |
| } |
| |
| void ChildTraceMessageFilter::OnFilterRemoved() { |
| channel_ = NULL; |
| } |
| |
| bool ChildTraceMessageFilter::OnMessageReceived(const IPC::Message& message) { |
| bool handled = true; |
| IPC_BEGIN_MESSAGE_MAP(ChildTraceMessageFilter, message) |
| IPC_MESSAGE_HANDLER(TracingMsg_BeginTracing, OnBeginTracing) |
| IPC_MESSAGE_HANDLER(TracingMsg_EndTracing, OnEndTracing) |
| IPC_MESSAGE_HANDLER(TracingMsg_EnableMonitoring, OnEnableMonitoring) |
| IPC_MESSAGE_HANDLER(TracingMsg_DisableMonitoring, OnDisableMonitoring) |
| IPC_MESSAGE_HANDLER(TracingMsg_CaptureMonitoringSnapshot, |
| OnCaptureMonitoringSnapshot) |
| IPC_MESSAGE_HANDLER(TracingMsg_GetTraceBufferPercentFull, |
| OnGetTraceBufferPercentFull) |
| IPC_MESSAGE_HANDLER(TracingMsg_SetWatchEvent, OnSetWatchEvent) |
| IPC_MESSAGE_HANDLER(TracingMsg_CancelWatchEvent, OnCancelWatchEvent) |
| IPC_MESSAGE_UNHANDLED(handled = false) |
| IPC_END_MESSAGE_MAP() |
| return handled; |
| } |
| |
| ChildTraceMessageFilter::~ChildTraceMessageFilter() {} |
| |
| void ChildTraceMessageFilter::OnBeginTracing( |
| const std::string& category_filter_str, |
| base::TimeTicks browser_time, |
| int options) { |
| #if defined(__native_client__) |
| // NaCl and system times are offset by a bit, so subtract some time from |
| // the captured timestamps. The value might be off by a bit due to messaging |
| // latency. |
| base::TimeDelta time_offset = base::TimeTicks::NowFromSystemTraceTime() - |
| browser_time; |
| TraceLog::GetInstance()->SetTimeOffset(time_offset); |
| #endif |
| |
| TraceLog::GetInstance()->SetEnabled( |
| base::debug::CategoryFilter(category_filter_str), |
| static_cast<base::debug::TraceLog::Options>(options)); |
| } |
| |
| void ChildTraceMessageFilter::OnEndTracing() { |
| TraceLog::GetInstance()->SetDisabled(); |
| |
| // Flush will generate one or more callbacks to OnTraceDataCollected |
| // synchronously or asynchronously. EndTracingAck will be sent in the last |
| // OnTraceDataCollected. We are already on the IO thread, so the |
| // OnTraceDataCollected calls will not be deferred. |
| TraceLog::GetInstance()->Flush( |
| base::Bind(&ChildTraceMessageFilter::OnTraceDataCollected, this)); |
| } |
| |
| void ChildTraceMessageFilter::OnEnableMonitoring( |
| const std::string& category_filter_str, |
| base::TimeTicks browser_time, |
| int options) { |
| TraceLog::GetInstance()->SetEnabled( |
| base::debug::CategoryFilter(category_filter_str), |
| static_cast<base::debug::TraceLog::Options>(options)); |
| } |
| |
| void ChildTraceMessageFilter::OnDisableMonitoring() { |
| TraceLog::GetInstance()->SetDisabled(); |
| } |
| |
| void ChildTraceMessageFilter::OnCaptureMonitoringSnapshot() { |
| // Flush will generate one or more callbacks to |
| // OnMonitoringTraceDataCollected. It's important that the last |
| // OnMonitoringTraceDataCollected gets called before |
| // CaptureMonitoringSnapshotAck below. We are already on the IO thread, |
| // so the OnMonitoringTraceDataCollected calls will not be deferred. |
| TraceLog::GetInstance()->FlushButLeaveBufferIntact( |
| base::Bind(&ChildTraceMessageFilter::OnMonitoringTraceDataCollected, |
| this)); |
| } |
| |
| void ChildTraceMessageFilter::OnGetTraceBufferPercentFull() { |
| float bpf = TraceLog::GetInstance()->GetBufferPercentFull(); |
| |
| channel_->Send(new TracingHostMsg_TraceBufferPercentFullReply(bpf)); |
| } |
| |
| void ChildTraceMessageFilter::OnSetWatchEvent(const std::string& category_name, |
| const std::string& event_name) { |
| TraceLog::GetInstance()->SetWatchEvent( |
| category_name, event_name, |
| base::Bind(&ChildTraceMessageFilter::OnWatchEventMatched, this)); |
| } |
| |
| void ChildTraceMessageFilter::OnCancelWatchEvent() { |
| TraceLog::GetInstance()->CancelWatchEvent(); |
| } |
| |
| void ChildTraceMessageFilter::OnWatchEventMatched() { |
| if (!ipc_message_loop_->BelongsToCurrentThread()) { |
| ipc_message_loop_->PostTask(FROM_HERE, |
| base::Bind(&ChildTraceMessageFilter::OnWatchEventMatched, this)); |
| return; |
| } |
| channel_->Send(new TracingHostMsg_WatchEventMatched); |
| } |
| |
| void ChildTraceMessageFilter::OnTraceDataCollected( |
| const scoped_refptr<base::RefCountedString>& events_str_ptr, |
| bool has_more_events) { |
| if (!ipc_message_loop_->BelongsToCurrentThread()) { |
| ipc_message_loop_->PostTask(FROM_HERE, |
| base::Bind(&ChildTraceMessageFilter::OnTraceDataCollected, this, |
| events_str_ptr, has_more_events)); |
| return; |
| } |
| if (events_str_ptr->data().size()) { |
| channel_->Send(new TracingHostMsg_TraceDataCollected( |
| events_str_ptr->data())); |
| } |
| if (!has_more_events) { |
| std::vector<std::string> category_groups; |
| TraceLog::GetInstance()->GetKnownCategoryGroups(&category_groups); |
| channel_->Send(new TracingHostMsg_EndTracingAck(category_groups)); |
| } |
| } |
| |
| void ChildTraceMessageFilter::OnMonitoringTraceDataCollected( |
| const scoped_refptr<base::RefCountedString>& events_str_ptr, |
| bool has_more_events) { |
| if (!ipc_message_loop_->BelongsToCurrentThread()) { |
| ipc_message_loop_->PostTask(FROM_HERE, |
| base::Bind(&ChildTraceMessageFilter:: |
| OnMonitoringTraceDataCollected, |
| this, |
| events_str_ptr, |
| has_more_events)); |
| return; |
| } |
| channel_->Send(new TracingHostMsg_MonitoringTraceDataCollected( |
| events_str_ptr->data())); |
| |
| if (!has_more_events) |
| channel_->Send(new TracingHostMsg_CaptureMonitoringSnapshotAck()); |
| } |
| |
| } // namespace tracing |