Port WebSockets to Mojo IPC

content/child/websocket_dispatcher.cc

-// Copyright 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 <stdint.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),
-    weak_ptr_factory_(this) {}
-
-WebSocketDispatcher::~WebSocketDispatcher() {}
-
-bool WebSocketDispatcher::CanHandleMessage(const IPC::Message& msg) {
-  switch (msg.type()) {
-    case WebSocketMsg_AddChannelResponse::ID:
-    case WebSocketMsg_NotifyStartOpeningHandshake::ID:
-    case WebSocketMsg_NotifyFinishOpeningHandshake::ID:
-    case WebSocketMsg_NotifyFailure::ID:
-    case WebSocketMsg_SendFrame::ID:
-    case WebSocketMsg_FlowControl::ID:
-    case WebSocketMsg_DropChannel::ID:
-    case WebSocketMsg_NotifyClosing::ID:
-      return true;
-    default:
-      return false;
-  }
-}
-
-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 iter = bridges_.find(channel_id);
-  if (iter == bridges_.end()) {
-    DVLOG(1) << "Remove a non-existent bridge(" << channel_id << ")";
-    return;
-  }
-  bridges_.erase(iter);
-}
-
-bool WebSocketDispatcher::OnMessageReceived(const IPC::Message& msg) {
-  if (!CanHandleMessage(msg))
-    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_t type) {
-  std::map<int, WebSocketBridge*>::iterator iter = bridges_.find(channel_id);
-  if (iter == bridges_.end()) {
-    DVLOG(1) << "No bridge for channel_id=" << channel_id
-             << ", type=" << type;
-    return NULL;
-  }
-  return iter->second;
-}
-
-}  // namespace content