Delegate WebSocket message handlers to the loading task runner

content/child/websocket_message_filter.cc

+// 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) {}
+
+WebSocketMessageFilter::~WebSocketMessageFilter() {}
+
+bool WebSocketMessageFilter::OnMessageReceived(const IPC::Message& message) {
+  if (!websocket_dispatcher_->CanHandleMessage(message))
+    return false;
+  // |websocket_dispatcher_| is owned by a RenderThreadImpl and lives when the
+  // renderer process lives.
+  loading_task_runner_->PostTask(FROM_HERE, base::Bind(
+      base::IgnoreResult(&WebSocketDispatcher::OnMessageReceived),
+      base::Unretained(websocket_dispatcher_),
+      message));
+  return true;
+}
+
+}  // namespace content