More scaffolding, add class ServiceWorkerProviderHost.

content/browser/service_worker/service_worker_dispatcher_host.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/browser/service_worker/service_worker_dispatcher_host.h"
 
+#include "base/stl_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
 #include "content/common/service_worker_messages.h"
 #include "ipc/ipc_message_macros.h"
 #include "third_party/WebKit/public/platform/WebServiceWorkerError.h"
 #include "url/gurl.h"
 
 namespace content {
 
 namespace {
 
 const char kDisabledErrorMessage[] =
     "ServiceWorker is disabled";
 const char kDomainMismatchErrorMessage[] =
     "Scope and scripts do not have the same origin";
 
 // TODO(alecflett): Store the service_worker_id keyed by (domain+pattern,
 // script) so we don't always return a new service worker id.
 int64 NextWorkerId() {
   static int64 service_worker_id = 0;
   return service_worker_id++;
 }
 
 }  // namespace
 
 ServiceWorkerDispatcherHost::ServiceWorkerDispatcherHost(
-    int render_process_id) {
+    int render_process_id)
+    : render_process_id_(render_process_id) {
 }
 
 ServiceWorkerDispatcherHost::~ServiceWorkerDispatcherHost() {
+  STLDeleteValues(&providers_);
+  if (context_)
+    context_->DetachProviderHostMap(render_process_id_);
 }
 
 void ServiceWorkerDispatcherHost::Init(
     ServiceWorkerContextWrapper* context_wrapper) {
   if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) {
     BrowserThread::PostTask(
         BrowserThread::IO, FROM_HERE,
         base::Bind(&ServiceWorkerDispatcherHost::Init,
                     this, make_scoped_refptr(context_wrapper)));
       return;
   }
   context_ = context_wrapper->context()->AsWeakPtr();
+  if (context_)
+    context_->AttachProviderHostMap(render_process_id_, &providers_);
+}
+
+void ServiceWorkerDispatcherHost::OnDestruct() const {
+  BrowserThread::DeleteOnIOThread::Destruct(this);
 }
 
 bool ServiceWorkerDispatcherHost::OnMessageReceived(
     const IPC::Message& message,
     bool* message_was_ok) {
   if (IPC_MESSAGE_CLASS(message) != ServiceWorkerMsgStart)
     return false;
 
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP_EX(
     ServiceWorkerDispatcherHost, message, *message_was_ok)
     IPC_MESSAGE_HANDLER(ServiceWorkerHostMsg_RegisterServiceWorker,
                         OnRegisterServiceWorker)
     IPC_MESSAGE_HANDLER(ServiceWorkerHostMsg_UnregisterServiceWorker,
                         OnUnregisterServiceWorker)
+    IPC_MESSAGE_HANDLER(ServiceWorkerHostMsg_ProviderCreated,
+                        OnProviderCreated)
+    IPC_MESSAGE_HANDLER(ServiceWorkerHostMsg_ProviderDestroyed,
+                        OnProviderDestroyed)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled;
 }
 
 void ServiceWorkerDispatcherHost::OnRegisterServiceWorker(
     int32 thread_id,
     int32 request_id,
     const GURL& scope,
@@ skipping 34 matching lines @@
         thread_id,
         request_id,
         blink::WebServiceWorkerError::DisabledError,
         ASCIIToUTF16(kDisabledErrorMessage)));
     return;
   }
 
   Send(new ServiceWorkerMsg_ServiceWorkerUnregistered(thread_id, request_id));
 }
 
+void ServiceWorkerDispatcherHost::OnProviderCreated(int provider_id) {
+  ServiceWorkerProviderHostMap::iterator found = providers_.find(provider_id);
+  if (found != providers_.end()) {
+    BadMessageReceived();
+    return;
+  }
+  providers_[provider_id] = new ServiceWorkerProviderHost(provider_id);
+}
+
+void ServiceWorkerDispatcherHost::OnProviderDestroyed(int provider_id) {
+  ServiceWorkerProviderHostMap::iterator found = providers_.find(provider_id);
+  if (found == providers_.end()) {
+    BadMessageReceived();
+    return;
+  }
+  delete found->second;
+  providers_.erase(found);
+}
+
+ServiceWorkerProviderHost* ServiceWorkerDispatcherHost::GetProviderHost(
+    int provider_id) {
+  ServiceWorkerProviderHostMap::iterator found = providers_.find(provider_id);
+  if (found == providers_.end())
+    return NULL;
+  return found->second;
+}
+
 }  // namespace content