Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright (c) 2013 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 "content/child/websocket_dispatcher.h" | |
| 6 | |
| 7 #include <map> | |
| 8 | |
| 9 #include "base/logging.h" | |
| 10 #include "content/child/websocket_bridge.h" | |
| 11 #include "content/common/websocket_messages.h" | |
| 12 #include "ipc/ipc_message.h" | |
| 13 #include "url/gurl.h" | |
| 14 | |
| 15 namespace content { | |
| 16 | |
| 17 WebSocketDispatcher::WebSocketDispatcher() : channel_id_max_(0) {} | |
| 18 | |
| 19 WebSocketDispatcher::~WebSocketDispatcher() {} | |
| 20 | |
| 21 int WebSocketDispatcher::AddBridge(WebSocketBridge* bridge) { | |
| 22 ++channel_id_max_; | |
| 23 bridges_.insert(std::make_pair(channel_id_max_, bridge)); | |
| 24 return channel_id_max_; | |
| 25 } | |
| 26 | |
| 27 void WebSocketDispatcher::RemoveBridge(int channel_id) { | |
| 28 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.
| |
| 29 if (i == bridges_.end()) { | |
| 30 DVLOG(1) << "Remove a non-existent bridge(" << channel_id << ")"; | |
| 31 return; | |
| 32 } | |
| 33 bridges_.erase(i); | |
| 34 } | |
| 35 | |
| 36 bool WebSocketDispatcher::OnMessageReceived(const IPC::Message& msg) { | |
| 37 switch (msg.type()) { | |
| 38 case WebSocketMsg_AddChannelResponse::ID: | |
| 39 case WebSocketMsg_SendFrame::ID: | |
| 40 case WebSocketMsg_FlowControl::ID: | |
| 41 case WebSocketMsg_DropChannel::ID: | |
| 42 break; | |
| 43 default: | |
| 44 return false; | |
| 45 } | |
| 46 | |
| 47 WebSocketBridge* bridge = GetBridge(msg.routing_id(), msg.type()); | |
| 48 if (!bridge) | |
| 49 return true; | |
| 50 return bridge->OnMessageReceived(msg); | |
| 51 } | |
| 52 | |
| 53 WebSocketBridge* WebSocketDispatcher::GetBridge(int channel_id, uint32 type) { | |
| 54 std::map<int, WebSocketBridge*>::iterator i = bridges_.find(channel_id); | |
| 55 if (i == bridges_.end()) { | |
| 56 DVLOG(1) << "No bridge for channel_id=" << channel_id | |
| 57 << ", type=" << type; | |
| 58 return NULL; | |
| 59 } | |
| 60 return i->second; | |
| 61 } | |
| 62 | |
| 63 } // namespace content | |
| OLD | NEW |