| 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 12 matching lines...) Expand all Loading... |
| 23 namespace content { | 23 namespace content { |
| 24 | 24 |
| 25 class BrowserMessageFilter::Internal : public IPC::MessageFilter { | 25 class BrowserMessageFilter::Internal : public IPC::MessageFilter { |
| 26 public: | 26 public: |
| 27 explicit Internal(BrowserMessageFilter* filter) : filter_(filter) {} | 27 explicit Internal(BrowserMessageFilter* filter) : filter_(filter) {} |
| 28 | 28 |
| 29 private: | 29 private: |
| 30 virtual ~Internal() {} | 30 virtual ~Internal() {} |
| 31 | 31 |
| 32 // IPC::MessageFilter implementation: | 32 // IPC::MessageFilter implementation: |
| 33 virtual void OnFilterAdded(IPC::Sender* sender) OVERRIDE { | 33 virtual void OnFilterAdded(IPC::Sender* sender) override { |
| 34 filter_->sender_ = sender; | 34 filter_->sender_ = sender; |
| 35 filter_->OnFilterAdded(sender); | 35 filter_->OnFilterAdded(sender); |
| 36 } | 36 } |
| 37 | 37 |
| 38 virtual void OnFilterRemoved() OVERRIDE { | 38 virtual void OnFilterRemoved() override { |
| 39 filter_->OnFilterRemoved(); | 39 filter_->OnFilterRemoved(); |
| 40 } | 40 } |
| 41 | 41 |
| 42 virtual void OnChannelClosing() OVERRIDE { | 42 virtual void OnChannelClosing() override { |
| 43 filter_->sender_ = NULL; | 43 filter_->sender_ = NULL; |
| 44 filter_->OnChannelClosing(); | 44 filter_->OnChannelClosing(); |
| 45 } | 45 } |
| 46 | 46 |
| 47 virtual void OnChannelConnected(int32 peer_pid) OVERRIDE { | 47 virtual void OnChannelConnected(int32 peer_pid) override { |
| 48 filter_->peer_pid_ = peer_pid; | 48 filter_->peer_pid_ = peer_pid; |
| 49 filter_->OnChannelConnected(peer_pid); | 49 filter_->OnChannelConnected(peer_pid); |
| 50 } | 50 } |
| 51 | 51 |
| 52 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE { | 52 virtual bool OnMessageReceived(const IPC::Message& message) override { |
| 53 BrowserThread::ID thread = BrowserThread::IO; | 53 BrowserThread::ID thread = BrowserThread::IO; |
| 54 filter_->OverrideThreadForMessage(message, &thread); | 54 filter_->OverrideThreadForMessage(message, &thread); |
| 55 | 55 |
| 56 if (thread == BrowserThread::IO) { | 56 if (thread == BrowserThread::IO) { |
| 57 scoped_refptr<base::TaskRunner> runner = | 57 scoped_refptr<base::TaskRunner> runner = |
| 58 filter_->OverrideTaskRunnerForMessage(message); | 58 filter_->OverrideTaskRunnerForMessage(message); |
| 59 if (runner.get()) { | 59 if (runner.get()) { |
| 60 runner->PostTask( | 60 runner->PostTask( |
| 61 FROM_HERE, | 61 FROM_HERE, |
| 62 base::Bind( | 62 base::Bind( |
| 63 base::IgnoreResult(&Internal::DispatchMessage), this, message)); | 63 base::IgnoreResult(&Internal::DispatchMessage), this, message)); |
| 64 return true; | 64 return true; |
| 65 } | 65 } |
| 66 return DispatchMessage(message); | 66 return DispatchMessage(message); |
| 67 } | 67 } |
| 68 | 68 |
| 69 if (thread == BrowserThread::UI && | 69 if (thread == BrowserThread::UI && |
| 70 !BrowserMessageFilter::CheckCanDispatchOnUI(message, filter_.get())) { | 70 !BrowserMessageFilter::CheckCanDispatchOnUI(message, filter_.get())) { |
| 71 return true; | 71 return true; |
| 72 } | 72 } |
| 73 | 73 |
| 74 BrowserThread::PostTask( | 74 BrowserThread::PostTask( |
| 75 thread, FROM_HERE, | 75 thread, FROM_HERE, |
| 76 base::Bind( | 76 base::Bind( |
| 77 base::IgnoreResult(&Internal::DispatchMessage), this, message)); | 77 base::IgnoreResult(&Internal::DispatchMessage), this, message)); |
| 78 return true; | 78 return true; |
| 79 } | 79 } |
| 80 | 80 |
| 81 virtual bool GetSupportedMessageClasses( | 81 virtual bool GetSupportedMessageClasses( |
| 82 std::vector<uint32>* supported_message_classes) const OVERRIDE { | 82 std::vector<uint32>* supported_message_classes) const override { |
| 83 supported_message_classes->assign( | 83 supported_message_classes->assign( |
| 84 filter_->message_classes_to_filter().begin(), | 84 filter_->message_classes_to_filter().begin(), |
| 85 filter_->message_classes_to_filter().end()); | 85 filter_->message_classes_to_filter().end()); |
| 86 return true; | 86 return true; |
| 87 } | 87 } |
| 88 | 88 |
| 89 // Dispatches a message to the derived class. | 89 // Dispatches a message to the derived class. |
| 90 bool DispatchMessage(const IPC::Message& message) { | 90 bool DispatchMessage(const IPC::Message& message) { |
| 91 bool rv = filter_->OnMessageReceived(message); | 91 bool rv = filter_->OnMessageReceived(message); |
| 92 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO) || rv) << | 92 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO) || rv) << |
| (...skipping 125 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 218 | 218 |
| 219 IPC::MessageFilter* BrowserMessageFilter::GetFilter() { | 219 IPC::MessageFilter* BrowserMessageFilter::GetFilter() { |
| 220 // We create this on demand so that if a filter is used in a unit test but | 220 // We create this on demand so that if a filter is used in a unit test but |
| 221 // never attached to a channel, we don't leak Internal and this; | 221 // never attached to a channel, we don't leak Internal and this; |
| 222 DCHECK(!internal_) << "Should only be called once."; | 222 DCHECK(!internal_) << "Should only be called once."; |
| 223 internal_ = new Internal(this); | 223 internal_ = new Internal(this); |
| 224 return internal_; | 224 return internal_; |
| 225 } | 225 } |
| 226 | 226 |
| 227 } // namespace content | 227 } // namespace content |
| OLD | NEW |