src/content/browser/service_worker/embedded_worker_registry.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/browser/service_worker/embedded_worker_registry.h"
   6
   7 #include "base/stl_util.h"
   8 #include "content/browser/service_worker/embedded_worker_instance.h"
   9 #include "content/browser/service_worker/service_worker_context_core.h"
  10 #include "content/common/service_worker/embedded_worker_messages.h"
  11 #include "content/common/service_worker/service_worker_messages.h"
  12 #include "ipc/ipc_message.h"
  13 #include "ipc/ipc_sender.h"
  14
  15 namespace content {
  16
  17 EmbeddedWorkerRegistry::EmbeddedWorkerRegistry(
  18     base::WeakPtr<ServiceWorkerContextCore> context)
  19     : context_(context),
  20       next_embedded_worker_id_(0) {}
  21
  22 scoped_ptr<EmbeddedWorkerInstance> EmbeddedWorkerRegistry::CreateWorker() {
  23   scoped_ptr<EmbeddedWorkerInstance> worker(
  24       new EmbeddedWorkerInstance(this, next_embedded_worker_id_));
  25   worker_map_[next_embedded_worker_id_++] = worker.get();
  26   return worker.Pass();
  27 }
  28
  29 ServiceWorkerStatusCode EmbeddedWorkerRegistry::StartWorker(
  30     int process_id,
  31     int embedded_worker_id,
  32     int64 service_worker_version_id,
  33     const GURL& script_url) {
  34   return Send(process_id,
  35               new EmbeddedWorkerMsg_StartWorker(embedded_worker_id,
  36                                                service_worker_version_id,
  37                                                script_url));
  38 }
  39
  40 ServiceWorkerStatusCode EmbeddedWorkerRegistry::StopWorker(
  41     int process_id, int embedded_worker_id) {
  42   return Send(process_id,
  43               new EmbeddedWorkerMsg_StopWorker(embedded_worker_id));
  44 }
  45
  46 void EmbeddedWorkerRegistry::OnWorkerStarted(
  47     int process_id, int thread_id, int embedded_worker_id) {
  48   DCHECK(!ContainsKey(worker_process_map_, process_id) ||
  49          worker_process_map_[process_id].count(embedded_worker_id) == 0);
  50   WorkerInstanceMap::iterator found = worker_map_.find(embedded_worker_id);
  51   if (found == worker_map_.end()) {
  52     LOG(ERROR) << "Worker " << embedded_worker_id << " not registered";
  53     return;
  54   }
  55   worker_process_map_[process_id].insert(embedded_worker_id);
  56   DCHECK_EQ(found->second->process_id(), process_id);
  57   found->second->OnStarted(thread_id);
  58 }
  59
  60 void EmbeddedWorkerRegistry::OnWorkerStopped(
  61     int process_id, int embedded_worker_id) {
  62   WorkerInstanceMap::iterator found = worker_map_.find(embedded_worker_id);
  63   if (found == worker_map_.end()) {
  64     LOG(ERROR) << "Worker " << embedded_worker_id << " not registered";
  65     return;
  66   }
  67   DCHECK_EQ(found->second->process_id(), process_id);
  68   worker_process_map_[process_id].erase(embedded_worker_id);
  69   found->second->OnStopped();
  70 }
  71
  72 void EmbeddedWorkerRegistry::OnSendMessageToBrowser(
  73     int embedded_worker_id, int request_id, const IPC::Message& message) {
  74   WorkerInstanceMap::iterator found = worker_map_.find(embedded_worker_id);
  75   if (found == worker_map_.end()) {
  76     LOG(ERROR) << "Worker " << embedded_worker_id << " not registered";
  77     return;
  78   }
  79   // Perform security check to filter out any unexpected (and non-test)
  80   // messages. This must list up all message types that can go through here.
  81   if (message.type() == ServiceWorkerHostMsg_InstallEventFinished::ID ||
  82       message.type() == ServiceWorkerHostMsg_FetchEventFinished::ID ||
  83       IPC_MESSAGE_CLASS(message) == TestMsgStart) {
  84     found->second->OnMessageReceived(request_id, message);
  85     return;
  86   }
  87   NOTREACHED() << "Got unexpected message: " << message.type();
  88 }
  89
  90 void EmbeddedWorkerRegistry::AddChildProcessSender(
  91     int process_id, IPC::Sender* sender) {
  92   process_sender_map_[process_id] = sender;
  93   DCHECK(!ContainsKey(worker_process_map_, process_id));
  94 }
  95
  96 void EmbeddedWorkerRegistry::RemoveChildProcessSender(int process_id) {
  97   process_sender_map_.erase(process_id);
  98   std::map<int, std::set<int> >::iterator found =
  99       worker_process_map_.find(process_id);
 100   if (found != worker_process_map_.end()) {
 101     const std::set<int>& worker_set = worker_process_map_[process_id];
 102     for (std::set<int>::const_iterator it = worker_set.begin();
 103          it != worker_set.end();
 104          ++it) {
 105       int embedded_worker_id = *it;
 106       DCHECK(ContainsKey(worker_map_, embedded_worker_id));
 107       worker_map_[embedded_worker_id]->OnStopped();
 108     }
 109     worker_process_map_.erase(found);
 110   }
 111 }
 112
 113 EmbeddedWorkerInstance* EmbeddedWorkerRegistry::GetWorker(
 114     int embedded_worker_id) {
 115   WorkerInstanceMap::iterator found = worker_map_.find(embedded_worker_id);
 116   if (found == worker_map_.end())
 117     return NULL;
 118   return found->second;
 119 }
 120
 121 EmbeddedWorkerRegistry::~EmbeddedWorkerRegistry() {}
 122
 123 ServiceWorkerStatusCode EmbeddedWorkerRegistry::Send(
 124     int process_id, IPC::Message* message) {
 125   if (!context_)
 126     return SERVICE_WORKER_ERROR_ABORT;
 127   ProcessToSenderMap::iterator found = process_sender_map_.find(process_id);
 128   if (found == process_sender_map_.end())
 129     return SERVICE_WORKER_ERROR_PROCESS_NOT_FOUND;
 130   if (!found->second->Send(message))
 131     return SERVICE_WORKER_ERROR_IPC_FAILED;
 132   return SERVICE_WORKER_OK;
 133 }
 134
 135 void EmbeddedWorkerRegistry::RemoveWorker(int process_id,
 136                                           int embedded_worker_id) {
 137   DCHECK(ContainsKey(worker_map_, embedded_worker_id));
 138   worker_map_.erase(embedded_worker_id);
 139   worker_process_map_.erase(process_id);
 140 }
 141
 142 }  // namespace content