OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "content/browser/tracing/trace_message_filter.h" | 5 #include "content/browser/tracing/trace_message_filter.h" |
6 | 6 |
7 #include "components/tracing/common/tracing_messages.h" | 7 #include "components/tracing/common/tracing_messages.h" |
8 #include "content/browser/tracing/background_tracing_manager_impl.h" | 8 #include "content/browser/tracing/background_tracing_manager_impl.h" |
9 #include "content/browser/tracing/tracing_controller_impl.h" | 9 #include "content/browser/tracing/tracing_controller_impl.h" |
10 #include "content/common/child_process_host_impl.h" | 10 #include "content/common/child_process_host_impl.h" |
11 | 11 |
12 namespace content { | 12 namespace content { |
13 | 13 |
14 TraceMessageFilter::TraceMessageFilter(int child_process_id) | 14 TraceMessageFilter::TraceMessageFilter(int child_process_id) |
15 : BrowserMessageFilter(TracingMsgStart), | 15 : BrowserMessageFilter(TracingMsgStart) {} |
16 has_child_(false), | |
17 tracing_process_id_( | |
18 ChildProcessHostImpl::ChildProcessUniqueIdToTracingProcessId( | |
19 child_process_id)), | |
20 is_awaiting_end_ack_(false), | |
21 is_awaiting_buffer_percent_full_ack_(false) { | |
22 } | |
23 | 16 |
24 TraceMessageFilter::~TraceMessageFilter() {} | 17 TraceMessageFilter::~TraceMessageFilter() {} |
25 | 18 |
26 void TraceMessageFilter::OnChannelClosing() { | |
27 if (has_child_) { | |
28 if (is_awaiting_end_ack_) | |
29 OnEndTracingAck(std::vector<std::string>()); | |
30 | |
31 if (is_awaiting_buffer_percent_full_ack_) | |
32 OnTraceLogStatusReply(base::trace_event::TraceLogStatus()); | |
33 | |
34 TracingControllerImpl::GetInstance()->RemoveTraceMessageFilter(this); | |
35 } | |
36 } | |
37 | |
38 bool TraceMessageFilter::OnMessageReceived(const IPC::Message& message) { | 19 bool TraceMessageFilter::OnMessageReceived(const IPC::Message& message) { |
39 // Always on IO thread (BrowserMessageFilter guarantee). | 20 // Always on IO thread (BrowserMessageFilter guarantee). |
40 bool handled = true; | 21 bool handled = true; |
41 IPC_BEGIN_MESSAGE_MAP(TraceMessageFilter, message) | 22 IPC_BEGIN_MESSAGE_MAP(TraceMessageFilter, message) |
42 IPC_MESSAGE_HANDLER(TracingHostMsg_ChildSupportsTracing, | |
43 OnChildSupportsTracing) | |
44 IPC_MESSAGE_HANDLER(TracingHostMsg_EndTracingAck, OnEndTracingAck) | |
45 IPC_MESSAGE_HANDLER(TracingHostMsg_TraceDataCollected, | |
46 OnTraceDataCollected) | |
47 IPC_MESSAGE_HANDLER(TracingHostMsg_TraceLogStatusReply, | |
48 OnTraceLogStatusReply) | |
49 IPC_MESSAGE_HANDLER(TracingHostMsg_TriggerBackgroundTrace, | 23 IPC_MESSAGE_HANDLER(TracingHostMsg_TriggerBackgroundTrace, |
50 OnTriggerBackgroundTrace) | 24 OnTriggerBackgroundTrace) |
51 IPC_MESSAGE_HANDLER(TracingHostMsg_AbortBackgroundTrace, | 25 IPC_MESSAGE_HANDLER(TracingHostMsg_AbortBackgroundTrace, |
52 OnAbortBackgroundTrace) | 26 OnAbortBackgroundTrace) |
53 IPC_MESSAGE_UNHANDLED(handled = false) | 27 IPC_MESSAGE_UNHANDLED(handled = false) |
54 IPC_END_MESSAGE_MAP() | 28 IPC_END_MESSAGE_MAP() |
55 return handled; | 29 return handled; |
56 } | 30 } |
57 | 31 |
58 void TraceMessageFilter::SendBeginTracing( | |
59 const base::trace_event::TraceConfig& trace_config) { | |
60 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
61 Send(new TracingMsg_BeginTracing( | |
62 trace_config.ToString(), base::TimeTicks::Now(), tracing_process_id_)); | |
63 } | |
64 | |
65 void TraceMessageFilter::SendEndTracing() { | |
66 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
67 DCHECK(!is_awaiting_end_ack_); | |
68 is_awaiting_end_ack_ = true; | |
69 Send(new TracingMsg_EndTracing); | |
70 } | |
71 | |
72 void TraceMessageFilter::SendCancelTracing() { | |
73 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
74 DCHECK(!is_awaiting_end_ack_); | |
75 is_awaiting_end_ack_ = true; | |
76 Send(new TracingMsg_CancelTracing); | |
77 } | |
78 | |
79 void TraceMessageFilter::SendGetTraceLogStatus() { | |
80 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
81 DCHECK(!is_awaiting_buffer_percent_full_ack_); | |
82 is_awaiting_buffer_percent_full_ack_ = true; | |
83 Send(new TracingMsg_GetTraceLogStatus); | |
84 } | |
85 | |
86 void TraceMessageFilter::OnChildSupportsTracing() { | |
87 has_child_ = true; | |
88 TracingControllerImpl::GetInstance()->AddTraceMessageFilter(this); | |
89 } | |
90 | |
91 void TraceMessageFilter::OnEndTracingAck( | |
92 const std::vector<std::string>& known_categories) { | |
93 // is_awaiting_end_ack_ should always be true here, but check in case the | |
94 // child process is compromised. | |
95 if (is_awaiting_end_ack_) { | |
96 is_awaiting_end_ack_ = false; | |
97 TracingControllerImpl::GetInstance()->OnStopTracingAcked( | |
98 this, known_categories); | |
99 } else { | |
100 NOTREACHED(); | |
101 } | |
102 } | |
103 | |
104 void TraceMessageFilter::OnTraceDataCollected(const std::string& data) { | |
105 scoped_refptr<base::RefCountedString> data_ptr(new base::RefCountedString()); | |
106 data_ptr->data() = data; | |
107 TracingControllerImpl::GetInstance()->OnTraceDataCollected(data_ptr); | |
108 } | |
109 | |
110 void TraceMessageFilter::OnTraceLogStatusReply( | |
111 const base::trace_event::TraceLogStatus& status) { | |
112 if (is_awaiting_buffer_percent_full_ack_) { | |
113 is_awaiting_buffer_percent_full_ack_ = false; | |
114 TracingControllerImpl::GetInstance()->OnTraceLogStatusReply(this, status); | |
115 } else { | |
116 NOTREACHED(); | |
117 } | |
118 } | |
119 | |
120 void TraceMessageFilter::OnTriggerBackgroundTrace(const std::string& name) { | 32 void TraceMessageFilter::OnTriggerBackgroundTrace(const std::string& name) { |
121 BackgroundTracingManagerImpl::GetInstance()->OnHistogramTrigger(name); | 33 BackgroundTracingManagerImpl::GetInstance()->OnHistogramTrigger(name); |
122 } | 34 } |
123 | 35 |
124 void TraceMessageFilter::OnAbortBackgroundTrace() { | 36 void TraceMessageFilter::OnAbortBackgroundTrace() { |
125 BackgroundTracingManagerImpl::GetInstance()->AbortScenario(); | 37 BackgroundTracingManagerImpl::GetInstance()->AbortScenario(); |
126 } | 38 } |
127 | 39 |
128 } // namespace content | 40 } // namespace content |
OLD | NEW |