src/content/browser/worker_host/worker_message_filter.cc

   1 // Copyright (c) 2011 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/browser/worker_host/worker_message_filter.h"
   6
   7 #include "content/browser/message_port_message_filter.h"
   8 #include "content/browser/worker_host/worker_service_impl.h"
   9 #include "content/common/view_messages.h"
  10 #include "content/common/worker_messages.h"
  11 #include "content/public/browser/resource_context.h"
  12
  13 namespace content {
  14
  15 WorkerMessageFilter::WorkerMessageFilter(
  16     int render_process_id,
  17     ResourceContext* resource_context,
  18     const WorkerStoragePartition& partition,
  19     MessagePortMessageFilter* message_port_message_filter)
  20     : render_process_id_(render_process_id),
  21       resource_context_(resource_context),
  22       partition_(partition),
  23       message_port_message_filter_(message_port_message_filter) {
  24   // Note: This constructor is called on both IO or UI thread.
  25   DCHECK(resource_context);
  26 }
  27
  28 WorkerMessageFilter::~WorkerMessageFilter() {
  29   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  30 }
  31
  32 void WorkerMessageFilter::OnChannelClosing() {
  33   WorkerServiceImpl::GetInstance()->OnWorkerMessageFilterClosing(this);
  34 }
  35
  36 bool WorkerMessageFilter::OnMessageReceived(const IPC::Message& message,
  37                                             bool* message_was_ok) {
  38   bool handled = true;
  39   IPC_BEGIN_MESSAGE_MAP_EX(WorkerMessageFilter, message, *message_was_ok)
  40     // Worker messages.
  41     // Only sent from renderer for now, until we have nested workers.
  42     IPC_MESSAGE_HANDLER(ViewHostMsg_CreateWorker, OnCreateWorker)
  43     IPC_MESSAGE_HANDLER(ViewHostMsg_ForwardToWorker, OnForwardToWorker)
  44     // Only sent from renderer.
  45     IPC_MESSAGE_HANDLER(ViewHostMsg_DocumentDetached, OnDocumentDetached)
  46     IPC_MESSAGE_UNHANDLED(handled = false)
  47   IPC_END_MESSAGE_MAP_EX()
  48
  49   return handled;
  50 }
  51
  52 int WorkerMessageFilter::GetNextRoutingID() {
  53   return message_port_message_filter_->GetNextRoutingID();
  54 }
  55
  56 void WorkerMessageFilter::OnCreateWorker(
  57     const ViewHostMsg_CreateWorker_Params& params,
  58     int* route_id) {
  59   bool url_error = false;
  60   *route_id = GetNextRoutingID();
  61   WorkerServiceImpl::GetInstance()->CreateWorker(
  62       params, *route_id, this, resource_context_, partition_, &url_error);
  63   if (url_error)
  64     *route_id = MSG_ROUTING_NONE;
  65 }
  66
  67 void WorkerMessageFilter::OnForwardToWorker(const IPC::Message& message) {
  68   WorkerServiceImpl::GetInstance()->ForwardToWorker(message, this);
  69 }
  70
  71 void WorkerMessageFilter::OnDocumentDetached(unsigned long long document_id) {
  72   WorkerServiceImpl::GetInstance()->DocumentDetached(document_id, this);
  73 }
  74
  75 }  // namespace content