Wire provider_id into scriptable API provider to start listening events

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/strings/utf_string_conversions.h"
 #include "content/browser/service_worker/embedded_worker_registry.h"
 #include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
@@ skipping 62 matching lines @@
   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_HANDLER(ServiceWorkerHostMsg_AddScriptClient,
+                        OnAddScriptClient)
+    IPC_MESSAGE_HANDLER(ServiceWorkerHostMsg_RemoveScriptClient,
+                        OnRemoveScriptClient)
     IPC_MESSAGE_HANDLER(EmbeddedWorkerHostMsg_WorkerStarted,
                         OnWorkerStarted)
     IPC_MESSAGE_HANDLER(EmbeddedWorkerHostMsg_WorkerStopped,
                         OnWorkerStopped)
     IPC_MESSAGE_HANDLER(EmbeddedWorkerHostMsg_SendMessageToBrowser,
                         OnSendMessageToBrowser)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled;
@@ skipping 75 matching lines @@
 void ServiceWorkerDispatcherHost::OnProviderDestroyed(int provider_id) {
   if (!context_)
     return;
   if (!context_->GetProviderHost(render_process_id_, provider_id)) {
     BadMessageReceived();
     return;
   }
   context_->RemoveProviderHost(render_process_id_, provider_id);
 }
 
+void ServiceWorkerDispatcherHost::OnAddScriptClient(
+    int thread_id, int provider_id) {
+  if (!context_)
+    return;
+  ServiceWorkerProviderHost* provider_host =
+      context_->GetProviderHost(render_process_id_, provider_id);
+  if (!provider_host) {

[michaeln, 2014/03/07 02:24:09]

This might not be indicative of a bad message depending on blink thread shutdown
ordering? I don't know how much we can trust blink about this?

[kinuko, 2014/03/07 02:37:13]

Hmm, good point.  Changed to just return in null case.

+    BadMessageReceived();
+    return;
+  }
+  provider_host->AddScriptClient(thread_id);
+}
+
+void ServiceWorkerDispatcherHost::OnRemoveScriptClient(
+    int thread_id, int provider_id) {
+  if (!context_)
+    return;
+  ServiceWorkerProviderHost* provider_host =
+      context_->GetProviderHost(render_process_id_, provider_id);
+  if (!provider_host) {

[michaeln, 2014/03/07 02:24:09]

ditto can we trust blink on this?

+    BadMessageReceived();
+    return;
+  }
+  provider_host->RemoveScriptClient(thread_id);
+}
+
 void ServiceWorkerDispatcherHost::RegistrationComplete(
     int32 thread_id,
     int32 request_id,
     ServiceWorkerStatusCode status,
     int64 registration_id) {
   if (status != SERVICE_WORKER_OK) {
     SendRegistrationError(thread_id, request_id, status);
     return;
   }
 
@@ skipping 44 matching lines @@
     ServiceWorkerStatusCode status) {
   base::string16 error_message;
   blink::WebServiceWorkerError::ErrorType error_type;
   GetServiceWorkerRegistrationStatusResponse(
       status, &error_type, &error_message);
   Send(new ServiceWorkerMsg_ServiceWorkerRegistrationError(
       thread_id, request_id, error_type, error_message));
 }
 
 }  // namespace content