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/trace_message_filter.h" | 5 #include "content/browser/trace_message_filter.h" |
6 | 6 |
7 #include "content/browser/trace_controller_impl.h" | 7 #include "content/browser/trace_controller_impl.h" |
8 #include "content/common/child_process_messages.h" | 8 #include "content/common/child_process_messages.h" |
9 | 9 |
10 using content::BrowserMessageFilter; | 10 using content::BrowserMessageFilter; |
11 using content::BrowserThread; | 11 using content::BrowserThread; |
12 using content::TraceControllerImpl; | 12 using content::TraceControllerImpl; |
13 | 13 |
14 TraceMessageFilter::TraceMessageFilter() : | 14 TraceMessageFilter::TraceMessageFilter() : |
15 has_child_(false), | 15 has_child_(false), |
16 is_awaiting_end_ack_(false), | 16 is_awaiting_end_ack_(false), |
17 is_awaiting_bpf_ack_(false) { | 17 is_awaiting_buffer_percent_full_ack_(false) { |
18 } | 18 } |
19 | 19 |
20 void TraceMessageFilter::OnFilterAdded(IPC::Channel* channel) { | 20 void TraceMessageFilter::OnFilterAdded(IPC::Channel* channel) { |
21 // Always on IO thread (BrowserMessageFilter guarantee). | 21 // Always on IO thread (BrowserMessageFilter guarantee). |
22 BrowserMessageFilter::OnFilterAdded(channel); | 22 BrowserMessageFilter::OnFilterAdded(channel); |
23 } | 23 } |
24 | 24 |
25 void TraceMessageFilter::OnChannelClosing() { | 25 void TraceMessageFilter::OnChannelClosing() { |
26 // Always on IO thread (BrowserMessageFilter guarantee). | 26 // Always on IO thread (BrowserMessageFilter guarantee). |
27 BrowserMessageFilter::OnChannelClosing(); | 27 BrowserMessageFilter::OnChannelClosing(); |
28 | 28 |
29 if (has_child_) { | 29 if (has_child_) { |
30 if (is_awaiting_bpf_ack_) | 30 if (is_awaiting_end_ack_) |
31 OnEndTracingAck(std::vector<std::string>()); | 31 OnEndTracingAck(std::vector<std::string>()); |
32 | 32 |
33 if (is_awaiting_end_ack_) | 33 if (is_awaiting_buffer_percent_full_ack_) |
34 OnTraceBufferPercentFullReply(0.0f); | 34 OnTraceBufferPercentFullReply(0.0f); |
35 | 35 |
36 TraceControllerImpl::GetInstance()->RemoveFilter(this); | 36 TraceControllerImpl::GetInstance()->RemoveFilter(this); |
37 } | 37 } |
38 } | 38 } |
39 | 39 |
40 bool TraceMessageFilter::OnMessageReceived(const IPC::Message& message, | 40 bool TraceMessageFilter::OnMessageReceived(const IPC::Message& message, |
41 bool* message_was_ok) { | 41 bool* message_was_ok) { |
42 // Always on IO thread (BrowserMessageFilter guarantee). | 42 // Always on IO thread (BrowserMessageFilter guarantee). |
43 bool handled = true; | 43 bool handled = true; |
(...skipping 22 matching lines...) Expand all Loading... | |
66 | 66 |
67 void TraceMessageFilter::SendEndTracing() { | 67 void TraceMessageFilter::SendEndTracing() { |
68 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 68 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
69 DCHECK(!is_awaiting_end_ack_); | 69 DCHECK(!is_awaiting_end_ack_); |
70 is_awaiting_end_ack_ = true; | 70 is_awaiting_end_ack_ = true; |
71 Send(new ChildProcessMsg_EndTracing); | 71 Send(new ChildProcessMsg_EndTracing); |
72 } | 72 } |
73 | 73 |
74 void TraceMessageFilter::SendGetTraceBufferPercentFull() { | 74 void TraceMessageFilter::SendGetTraceBufferPercentFull() { |
75 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 75 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
76 DCHECK(!is_awaiting_bpf_ack_); | 76 DCHECK(!is_awaiting_buffer_percent_full_ack_); |
77 is_awaiting_bpf_ack_ = true; | 77 is_awaiting_buffer_percent_full_ack_ = true; |
78 Send(new ChildProcessMsg_GetTraceBufferPercentFull); | 78 Send(new ChildProcessMsg_GetTraceBufferPercentFull); |
79 } | 79 } |
80 | 80 |
81 TraceMessageFilter::~TraceMessageFilter() {} | 81 TraceMessageFilter::~TraceMessageFilter() {} |
82 | 82 |
83 void TraceMessageFilter::OnChildSupportsTracing() { | 83 void TraceMessageFilter::OnChildSupportsTracing() { |
84 has_child_ = true; | 84 has_child_ = true; |
85 TraceControllerImpl::GetInstance()->AddFilter(this); | 85 TraceControllerImpl::GetInstance()->AddFilter(this); |
86 } | 86 } |
87 | 87 |
88 void TraceMessageFilter::OnEndTracingAck( | 88 void TraceMessageFilter::OnEndTracingAck( |
89 const std::vector<std::string>& known_categories) { | 89 const std::vector<std::string>& known_categories) { |
90 // is_awaiting_end_ack_ should always be true here, but check in case the | 90 // is_awaiting_end_ack_ should always be true here, but check in case the |
91 // child process is compromised. | 91 // child process is compromised. |
92 DCHECK(is_awaiting_end_ack_); | |
sky
2012/06/06 20:16:32
How about an else NOTREACHED instead. Or, remove t
jbates
2012/06/06 21:22:17
Done.
| |
92 if (is_awaiting_end_ack_) { | 93 if (is_awaiting_end_ack_) { |
93 is_awaiting_end_ack_ = false; | 94 is_awaiting_end_ack_ = false; |
94 TraceControllerImpl::GetInstance()->OnEndTracingAck(known_categories); | 95 TraceControllerImpl::GetInstance()->OnEndTracingAck(known_categories); |
95 } | 96 } |
96 } | 97 } |
97 | 98 |
98 void TraceMessageFilter::OnTraceDataCollected(const std::string& data) { | 99 void TraceMessageFilter::OnTraceDataCollected(const std::string& data) { |
99 scoped_refptr<base::RefCountedString> data_ptr(new base::RefCountedString()); | 100 scoped_refptr<base::RefCountedString> data_ptr(new base::RefCountedString()); |
100 data_ptr->data() = data; | 101 data_ptr->data() = data; |
101 TraceControllerImpl::GetInstance()->OnTraceDataCollected(data_ptr); | 102 TraceControllerImpl::GetInstance()->OnTraceDataCollected(data_ptr); |
102 } | 103 } |
103 | 104 |
104 void TraceMessageFilter::OnTraceBufferFull() { | 105 void TraceMessageFilter::OnTraceBufferFull() { |
105 TraceControllerImpl::GetInstance()->OnTraceBufferFull(); | 106 TraceControllerImpl::GetInstance()->OnTraceBufferFull(); |
106 } | 107 } |
107 | 108 |
108 void TraceMessageFilter::OnTraceBufferPercentFullReply(float percent_full) { | 109 void TraceMessageFilter::OnTraceBufferPercentFullReply(float percent_full) { |
109 if (is_awaiting_bpf_ack_) { | 110 DCHECK(is_awaiting_buffer_percent_full_ack_); |
110 is_awaiting_bpf_ack_ = false; | 111 if (is_awaiting_buffer_percent_full_ack_) { |
112 is_awaiting_buffer_percent_full_ack_ = false; | |
111 TraceControllerImpl::GetInstance()->OnTraceBufferPercentFullReply( | 113 TraceControllerImpl::GetInstance()->OnTraceBufferPercentFullReply( |
112 percent_full); | 114 percent_full); |
113 } | 115 } |
114 } | 116 } |
115 | 117 |
OLD | NEW |