Delegate WebSocket message handlers to the loading task runner

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) {}
+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) {
-  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:
-      break;
-    default:
-      return false;
-  }
-
+  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