Delegate WebSocket message handlers to the loading task runner

content/child/websocket_message_filter.cc

Index: content/child/websocket_message_filter.cc
diff --git a/content/child/websocket_message_filter.cc b/content/child/websocket_message_filter.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b2e54c08f96b38eb027b9a7d8ea504fcdfdd5315
--- /dev/null
+++ b/content/child/websocket_message_filter.cc
@@ -0,0 +1,36 @@
+// Copyright 2016 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_message_filter.h"
+
+#include "base/bind.h"
+#include "base/single_thread_task_runner.h"
+#include "content/child/websocket_dispatcher.h"
+#include "ipc/ipc_message.h"
+
+namespace content {
+
+WebSocketMessageFilter::WebSocketMessageFilter(
+    WebSocketDispatcher* websocket_dispatcher)
+  : websocket_dispatcher_(websocket_dispatcher->GetWeakPtr()) {}
+
+WebSocketMessageFilter::~WebSocketMessageFilter() {}
+
+bool WebSocketMessageFilter::OnMessageReceived(const IPC::Message& message) {
+  if (!WebSocketDispatcher::CanHandleMessage(message))
+    return false;
+  loading_task_runner_->PostTask(FROM_HERE, base::Bind(
+      &WebSocketMessageFilter::OnMessageReceivedOnMainThread,
+      this, message));
+  return true;
+}
+
+void WebSocketMessageFilter::OnMessageReceivedOnMainThread(
+    const IPC::Message& message) {
+  if (!websocket_dispatcher_.get())
+    return;
+  websocket_dispatcher_->OnMessageReceived(message);
+}
+
+}  // namespace content