Chromium Code Reviews| Index: content/child/websocket_dispatcher.cc |
| diff --git a/content/child/websocket_dispatcher.cc b/content/child/websocket_dispatcher.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..e120e82bcc5b0094fbf66860936ce648b63da82c |
| --- /dev/null |
| +++ b/content/child/websocket_dispatcher.cc |
| @@ -0,0 +1,63 @@ |
| +// Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "content/child/websocket_dispatcher.h" |
| + |
| +#include <map> |
| + |
| +#include "base/logging.h" |
| +#include "content/child/websocket_bridge.h" |
| +#include "content/common/websocket_messages.h" |
| +#include "ipc/ipc_message.h" |
| +#include "url/gurl.h" |
| + |
| +namespace content { |
| + |
| +WebSocketDispatcher::WebSocketDispatcher() : channel_id_max_(0) {} |
| + |
| +WebSocketDispatcher::~WebSocketDispatcher() {} |
| + |
| +int WebSocketDispatcher::AddBridge(WebSocketBridge* bridge) { |
| + ++channel_id_max_; |
| + bridges_.insert(std::make_pair(channel_id_max_, bridge)); |
| + return channel_id_max_; |
| +} |
| + |
| +void WebSocketDispatcher::RemoveBridge(int channel_id) { |
| + std::map<int, WebSocketBridge*>::iterator i = bridges_.find(channel_id); |
|
Tom Sepez
2013/09/20 16:39:30
nit: pickier reviewers would insist that a longer
yhirano
2013/09/23 20:38:05
Done.
|
| + if (i == bridges_.end()) { |
| + DVLOG(1) << "Remove a non-existent bridge(" << channel_id << ")"; |
| + return; |
| + } |
| + bridges_.erase(i); |
| +} |
| + |
| +bool WebSocketDispatcher::OnMessageReceived(const IPC::Message& msg) { |
| + switch (msg.type()) { |
| + case WebSocketMsg_AddChannelResponse::ID: |
| + case WebSocketMsg_SendFrame::ID: |
| + case WebSocketMsg_FlowControl::ID: |
| + case WebSocketMsg_DropChannel::ID: |
| + break; |
| + default: |
| + return false; |
| + } |
| + |
| + WebSocketBridge* bridge = GetBridge(msg.routing_id(), msg.type()); |
| + if (!bridge) |
| + return true; |
| + return bridge->OnMessageReceived(msg); |
| +} |
| + |
| +WebSocketBridge* WebSocketDispatcher::GetBridge(int channel_id, uint32 type) { |
| + std::map<int, WebSocketBridge*>::iterator i = bridges_.find(channel_id); |
| + if (i == bridges_.end()) { |
| + DVLOG(1) << "No bridge for channel_id=" << channel_id |
| + << ", type=" << type; |
| + return NULL; |
| + } |
| + return i->second; |
| +} |
| + |
| +} // namespace content |