nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 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 |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #include "ipc/ipc_forwarding_message_filter.h" |
| 6 | |
| 7 | #include "base/bind.h" |
| 8 | #include "base/location.h" |
dmichael@chromium.org | c0c370e | 2014-04-25 09:07:30 +0900 | [diff] [blame] | 9 | #include "ipc/ipc_message.h" |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 10 | |
| 11 | namespace IPC { |
| 12 | |
| 13 | ForwardingMessageFilter::ForwardingMessageFilter( |
| 14 | const uint32* message_ids_to_filter, |
| 15 | size_t num_message_ids_to_filter, |
nduca@chromium.org | a6d7cf6 | 2012-08-14 08:24:35 +0900 | [diff] [blame] | 16 | base::TaskRunner* target_task_runner) |
| 17 | : target_task_runner_(target_task_runner) { |
rsleevi@chromium.org | 23b6623 | 2013-06-01 13:11:27 +0900 | [diff] [blame] | 18 | DCHECK(target_task_runner_.get()); |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 19 | for (size_t i = 0; i < num_message_ids_to_filter; i++) |
| 20 | message_ids_to_filter_.insert(message_ids_to_filter[i]); |
| 21 | } |
| 22 | |
nduca@chromium.org | a6d7cf6 | 2012-08-14 08:24:35 +0900 | [diff] [blame] | 23 | void ForwardingMessageFilter::AddRoute(int routing_id, const Handler& handler) { |
| 24 | DCHECK(!handler.is_null()); |
| 25 | base::AutoLock locked(handlers_lock_); |
| 26 | handlers_.insert(std::make_pair(routing_id, handler)); |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 27 | } |
| 28 | |
| 29 | void ForwardingMessageFilter::RemoveRoute(int routing_id) { |
nduca@chromium.org | a6d7cf6 | 2012-08-14 08:24:35 +0900 | [diff] [blame] | 30 | base::AutoLock locked(handlers_lock_); |
| 31 | handlers_.erase(routing_id); |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 32 | } |
| 33 | |
| 34 | bool ForwardingMessageFilter::OnMessageReceived(const Message& message) { |
| 35 | if (message_ids_to_filter_.find(message.type()) == |
| 36 | message_ids_to_filter_.end()) |
| 37 | return false; |
| 38 | |
nduca@chromium.org | a6d7cf6 | 2012-08-14 08:24:35 +0900 | [diff] [blame] | 39 | |
| 40 | Handler handler; |
| 41 | |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 42 | { |
nduca@chromium.org | a6d7cf6 | 2012-08-14 08:24:35 +0900 | [diff] [blame] | 43 | base::AutoLock locked(handlers_lock_); |
| 44 | std::map<int, Handler>::iterator it = handlers_.find(message.routing_id()); |
| 45 | if (it == handlers_.end()) |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 46 | return false; |
nduca@chromium.org | a6d7cf6 | 2012-08-14 08:24:35 +0900 | [diff] [blame] | 47 | handler = it->second; |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 48 | } |
| 49 | |
nduca@chromium.org | a6d7cf6 | 2012-08-14 08:24:35 +0900 | [diff] [blame] | 50 | target_task_runner_->PostTask(FROM_HERE, base::Bind(handler, message)); |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 51 | return true; |
| 52 | } |
| 53 | |
| 54 | ForwardingMessageFilter::~ForwardingMessageFilter() { |
| 55 | } |
| 56 | |
nduca@chromium.org | bf47b67 | 2012-07-21 05:14:27 +0900 | [diff] [blame] | 57 | } // namespace IPC |