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     // Only sent from renderer for now, until we have nested workers.
  44     IPC_MESSAGE_HANDLER(ViewHostMsg_LookupSharedWorker, OnLookupSharedWorker)
  45     IPC_MESSAGE_HANDLER(ViewHostMsg_ForwardToWorker, OnForwardToWorker)
  46     // Only sent from renderer.
  47     IPC_MESSAGE_HANDLER(ViewHostMsg_DocumentDetached, OnDocumentDetached)
  48     IPC_MESSAGE_UNHANDLED(handled = false)
  49   IPC_END_MESSAGE_MAP_EX()
  50
  51   return handled;
  52 }
  53
  54 int WorkerMessageFilter::GetNextRoutingID() {
  55   return message_port_message_filter_->GetNextRoutingID();
  56 }
  57
  58 void WorkerMessageFilter::OnCreateWorker(
  59     const ViewHostMsg_CreateWorker_Params& params,
  60     int* route_id) {
  61   *route_id = params.route_id != MSG_ROUTING_NONE ?
  62       params.route_id : GetNextRoutingID();
  63   WorkerServiceImpl::GetInstance()->CreateWorker(
  64       params, *route_id, this, resource_context_, partition_);
  65 }
  66
  67 void WorkerMessageFilter::OnLookupSharedWorker(
  68     const ViewHostMsg_CreateWorker_Params& params,
  69     bool* exists,
  70     int* route_id,
  71     bool* url_error) {
  72   *route_id = GetNextRoutingID();
  73
  74   WorkerServiceImpl::GetInstance()->LookupSharedWorker(
  75       params, *route_id, this, resource_context_, partition_, exists,
  76       url_error);
  77 }
  78
  79 void WorkerMessageFilter::OnForwardToWorker(const IPC::Message& message) {
  80   WorkerServiceImpl::GetInstance()->ForwardToWorker(message, this);
  81 }
  82
  83 void WorkerMessageFilter::OnDocumentDetached(unsigned long long document_id) {
  84   WorkerServiceImpl::GetInstance()->DocumentDetached(document_id, this);
  85 }
  86
  87 }  // namespace content