src/content/child/quota_message_filter.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/child/quota_message_filter.h"
   6
   7 #include "base/bind.h"
   8 #include "base/location.h"
   9 #include "base/message_loop/message_loop_proxy.h"
  10 #include "base/pickle.h"
  11 #include "content/child/quota_dispatcher.h"
  12 #include "content/child/thread_safe_sender.h"
  13 #include "content/common/quota_messages.h"
  14 #include "webkit/child/worker_task_runner.h"
  15
  16 using webkit_glue::WorkerTaskRunner;
  17
  18 namespace content {
  19
  20 QuotaMessageFilter::QuotaMessageFilter(
  21     ThreadSafeSender* thread_safe_sender)
  22     : main_thread_loop_proxy_(base::MessageLoopProxy::current()),
  23       thread_safe_sender_(thread_safe_sender),
  24       next_request_id_(0) {
  25 }
  26
  27 bool QuotaMessageFilter::OnMessageReceived(const IPC::Message& msg) {
  28   if (IPC_MESSAGE_CLASS(msg) != QuotaMsgStart)
  29     return false;
  30   int request_id = -1;
  31   bool result = PickleIterator(msg).ReadInt(&request_id);
  32   DCHECK(result);
  33   base::Closure closure = base::Bind(
  34       &QuotaMessageFilter::DispatchMessage, this, msg);
  35   int thread_id = 0;
  36   {
  37     base::AutoLock lock(request_id_map_lock_);
  38     RequestIdToThreadId::iterator found = request_id_map_.find(request_id);
  39     if (found != request_id_map_.end()) {
  40       thread_id = found->second;
  41       request_id_map_.erase(found);
  42     }
  43   }
  44   if (!thread_id) {
  45     main_thread_loop_proxy_->PostTask(FROM_HERE, closure);
  46     return true;
  47   }
  48   WorkerTaskRunner::Instance()->PostTask(thread_id, closure);
  49   return true;
  50 }
  51
  52 int QuotaMessageFilter::GenerateRequestID(int thread_id) {
  53   base::AutoLock lock(request_id_map_lock_);
  54   request_id_map_[next_request_id_] = thread_id;
  55   return next_request_id_++;
  56 }
  57
  58 void QuotaMessageFilter::ClearThreadRequests(int thread_id) {
  59   base::AutoLock lock(request_id_map_lock_);
  60   for (RequestIdToThreadId::iterator iter = request_id_map_.begin();
  61        iter != request_id_map_.end();) {
  62     if (iter->second == thread_id)
  63       request_id_map_.erase(iter++);
  64     else
  65       iter++;
  66   }
  67 }
  68
  69 QuotaMessageFilter::~QuotaMessageFilter() {}
  70
  71 void QuotaMessageFilter::DispatchMessage(const IPC::Message& msg) {
  72   QuotaDispatcher::ThreadSpecificInstance(thread_safe_sender_.get(), this)
  73       ->OnMessageReceived(msg);
  74 }
  75
  76 }  // namespace content