Add ChildMessageFilter, a base class for renderer/worker cross-thread MessageFilter

content/child/quota_message_filter.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/quota_message_filter.h"
 
-#include "base/bind.h"
-#include "base/location.h"
 #include "base/message_loop/message_loop_proxy.h"
-#include "base/pickle.h"
 #include "content/child/quota_dispatcher.h"
 #include "content/child/thread_safe_sender.h"
+#include "content/child/worker_thread_task_runner.h"
 #include "content/common/quota_messages.h"
-#include "webkit/child/worker_task_runner.h"
-
-using webkit_glue::WorkerTaskRunner;
 
 namespace content {
 
 QuotaMessageFilter::QuotaMessageFilter(
     ThreadSafeSender* thread_safe_sender)
     : main_thread_loop_proxy_(base::MessageLoopProxy::current()),
       thread_safe_sender_(thread_safe_sender),
       next_request_id_(0) {
 }
 
-bool QuotaMessageFilter::OnMessageReceived(const IPC::Message& msg) {
-  if (IPC_MESSAGE_CLASS(msg) != QuotaMsgStart)
-    return false;
-  int request_id = -1;
-  bool result = PickleIterator(msg).ReadInt(&request_id);
-  DCHECK(result);
-  base::Closure closure = base::Bind(
-      &QuotaMessageFilter::DispatchMessage, this, msg);
-  int thread_id = 0;
-  {
-    base::AutoLock lock(request_id_map_lock_);
-    RequestIdToThreadId::iterator found = request_id_map_.find(request_id);
-    if (found != request_id_map_.end()) {
-      thread_id = found->second;
-      request_id_map_.erase(found);
-    }
-  }
-  if (!thread_id) {
-    main_thread_loop_proxy_->PostTask(FROM_HERE, closure);
-    return true;
-  }
-  WorkerTaskRunner::Instance()->PostTask(thread_id, closure);
-  return true;
-}
+QuotaMessageFilter::~QuotaMessageFilter() {}
 
 int QuotaMessageFilter::GenerateRequestID(int thread_id) {
   base::AutoLock lock(request_id_map_lock_);
   request_id_map_[next_request_id_] = thread_id;
   return next_request_id_++;
 }
 
 void QuotaMessageFilter::ClearThreadRequests(int thread_id) {
   base::AutoLock lock(request_id_map_lock_);
   for (RequestIdToThreadId::iterator iter = request_id_map_.begin();
        iter != request_id_map_.end();) {
     if (iter->second == thread_id)
       request_id_map_.erase(iter++);
     else
       iter++;
   }
 }
 
-QuotaMessageFilter::~QuotaMessageFilter() {}
+base::TaskRunner* QuotaMessageFilter::OverrideTaskRunnerForMessage(
+    const IPC::Message& msg) {
+  if (IPC_MESSAGE_CLASS(msg) != QuotaMsgStart)
+    return NULL;
 
-void QuotaMessageFilter::DispatchMessage(const IPC::Message& msg) {
+  int request_id = -1, thread_id = 0;
+  const bool success = PickleIterator(msg).ReadInt(&request_id);
+  DCHECK(success);
+
+  {
+    base::AutoLock lock(request_id_map_lock_);
+    RequestIdToThreadId::iterator found = request_id_map_.find(request_id);
+    if (found != request_id_map_.end()) {
+      thread_id = found->second;
+      request_id_map_.erase(found);
+    }
+  }
+
+  if (!thread_id)
+    return main_thread_loop_proxy_.get();
+  return new WorkerThreadTaskRunner(thread_id);
+}
+
+bool QuotaMessageFilter::OnMessageReceived(const IPC::Message& msg) {
+  if (IPC_MESSAGE_CLASS(msg) != QuotaMsgStart)
+    return false;
   QuotaDispatcher::ThreadSpecificInstance(thread_safe_sender_.get(), this)
       ->OnMessageReceived(msg);
+  return true;
 }
 
 }  // namespace content