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/public/browser/browser_message_filter.h" | 5 #include "content/public/browser/browser_message_filter.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
9 #include "base/command_line.h" | 9 #include "base/command_line.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
69 return true; | 69 return true; |
70 } | 70 } |
71 | 71 |
72 BrowserThread::PostTask( | 72 BrowserThread::PostTask( |
73 thread, FROM_HERE, | 73 thread, FROM_HERE, |
74 base::Bind( | 74 base::Bind( |
75 base::IgnoreResult(&Internal::DispatchMessage), this, message)); | 75 base::IgnoreResult(&Internal::DispatchMessage), this, message)); |
76 return true; | 76 return true; |
77 } | 77 } |
78 | 78 |
79 virtual bool GetSupportedMessageClasses( | |
80 std::vector<uint32>* supported_message_classes) const OVERRIDE { | |
81 supported_message_classes->assign( | |
82 filter_->message_classes_to_filter().begin(), | |
83 filter_->message_classes_to_filter().end()); | |
84 return true; | |
85 } | |
86 | |
87 // Dispatches a message to the derived class. | 79 // Dispatches a message to the derived class. |
88 bool DispatchMessage(const IPC::Message& message) { | 80 bool DispatchMessage(const IPC::Message& message) { |
89 bool message_was_ok = true; | 81 bool message_was_ok = true; |
90 bool rv = filter_->OnMessageReceived(message, &message_was_ok); | 82 bool rv = filter_->OnMessageReceived(message, &message_was_ok); |
91 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO) || rv) << | 83 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO) || rv) << |
92 "Must handle messages that were dispatched to another thread!"; | 84 "Must handle messages that were dispatched to another thread!"; |
93 if (!message_was_ok) { | 85 if (!message_was_ok) { |
94 content::RecordAction( | 86 content::RecordAction( |
95 base::UserMetricsAction("BadMessageTerminate_BMF")); | 87 base::UserMetricsAction("BadMessageTerminate_BMF")); |
96 filter_->BadMessageReceived(); | 88 filter_->BadMessageReceived(); |
97 } | 89 } |
98 | 90 |
99 return rv; | 91 return rv; |
100 } | 92 } |
101 | 93 |
102 scoped_refptr<BrowserMessageFilter> filter_; | 94 scoped_refptr<BrowserMessageFilter> filter_; |
103 | 95 |
104 DISALLOW_COPY_AND_ASSIGN(Internal); | 96 DISALLOW_COPY_AND_ASSIGN(Internal); |
105 }; | 97 }; |
106 | 98 |
107 BrowserMessageFilter::BrowserMessageFilter(uint32 message_class_to_filter) | 99 BrowserMessageFilter::BrowserMessageFilter() |
108 : internal_(NULL), | 100 : internal_(NULL), channel_(NULL), |
109 channel_(NULL), | |
110 #if defined(OS_WIN) | 101 #if defined(OS_WIN) |
111 peer_handle_(base::kNullProcessHandle), | 102 peer_handle_(base::kNullProcessHandle), |
112 #endif | 103 #endif |
113 peer_pid_(base::kNullProcessId), | 104 peer_pid_(base::kNullProcessId) { |
114 message_classes_to_filter_(1, message_class_to_filter) {} | |
115 | |
116 BrowserMessageFilter::BrowserMessageFilter( | |
117 const uint32* message_classes_to_filter, | |
118 size_t num_message_classes_to_filter) | |
119 : internal_(NULL), | |
120 channel_(NULL), | |
121 #if defined(OS_WIN) | |
122 peer_handle_(base::kNullProcessHandle), | |
123 #endif | |
124 peer_pid_(base::kNullProcessId), | |
125 message_classes_to_filter_( | |
126 message_classes_to_filter, | |
127 message_classes_to_filter + num_message_classes_to_filter) { | |
128 DCHECK(num_message_classes_to_filter); | |
129 } | 105 } |
130 | 106 |
131 base::ProcessHandle BrowserMessageFilter::PeerHandle() { | 107 base::ProcessHandle BrowserMessageFilter::PeerHandle() { |
132 #if defined(OS_WIN) | 108 #if defined(OS_WIN) |
133 base::AutoLock lock(peer_handle_lock_); | 109 base::AutoLock lock(peer_handle_lock_); |
134 if (peer_handle_ == base::kNullProcessHandle) | 110 if (peer_handle_ == base::kNullProcessHandle) |
135 base::OpenPrivilegedProcessHandle(peer_pid_, &peer_handle_); | 111 base::OpenPrivilegedProcessHandle(peer_pid_, &peer_handle_); |
136 | 112 |
137 return peer_handle_; | 113 return peer_handle_; |
138 #else | 114 #else |
(...skipping 82 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
221 | 197 |
222 IPC::ChannelProxy::MessageFilter* BrowserMessageFilter::GetFilter() { | 198 IPC::ChannelProxy::MessageFilter* BrowserMessageFilter::GetFilter() { |
223 // We create this on demand so that if a filter is used in a unit test but | 199 // We create this on demand so that if a filter is used in a unit test but |
224 // never attached to a channel, we don't leak Internal and this; | 200 // never attached to a channel, we don't leak Internal and this; |
225 DCHECK(!internal_) << "Should only be called once."; | 201 DCHECK(!internal_) << "Should only be called once."; |
226 internal_ = new Internal(this); | 202 internal_ = new Internal(this); |
227 return internal_; | 203 return internal_; |
228 } | 204 } |
229 | 205 |
230 } // namespace content | 206 } // namespace content |
OLD | NEW |