Push API: Store push registration IDs for GetRegistration etc

content/browser/push_messaging/push_messaging_message_filter.cc

Index: content/browser/push_messaging/push_messaging_message_filter.cc
diff --git a/content/browser/push_messaging/push_messaging_message_filter.cc b/content/browser/push_messaging/push_messaging_message_filter.cc
index 415d8401b6c747dc26d0c2081199abcfb8cfdf1f..69b6ab0cff66494272306235ec86883102ff638a 100644
--- a/content/browser/push_messaging/push_messaging_message_filter.cc
+++ b/content/browser/push_messaging/push_messaging_message_filter.cc
@@ -11,7 +11,9 @@
 #include "base/metrics/histogram.h"
 #include "base/strings/string_number_conversions.h"
 #include "content/browser/renderer_host/render_process_host_impl.h"
+#include "content/browser/service_worker/service_worker_context_core.h"
 #include "content/browser/service_worker/service_worker_context_wrapper.h"
+#include "content/browser/service_worker/service_worker_storage.h"
 #include "content/common/push_messaging_messages.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
@@ -23,6 +25,8 @@ namespace content {
 namespace {
 
 void RecordRegistrationStatus(PushRegistrationStatus status) {
+  // Called from both UI and IO threads. Slightly racy, but acceptable, see
+  // https://groups.google.com/a/chromium.org/d/msg/chromium-dev/FNzZRJtN2aw/Aw0CWAXJJ1kJ
   UMA_HISTOGRAM_ENUMERATION("PushMessaging.RegistrationStatus",
                             status,
                             PUSH_REGISTRATION_STATUS_LAST + 1);
@@ -40,6 +44,9 @@ bool PushMessagingMessageFilter::RegisterData::FromDocument() const {
   return render_frame_id != ChildProcessHost::kInvalidUniqueID;
 }
 
+static const char kPushRegistrationIdServiceWorkerKey[] =
+    "push_registration_id";
+
 PushMessagingMessageFilter::PushMessagingMessageFilter(
     int render_process_id,
     ServiceWorkerContextWrapper* service_worker_context)
@@ -47,6 +54,7 @@ PushMessagingMessageFilter::PushMessagingMessageFilter(
       render_process_id_(render_process_id),
       service_worker_context_(service_worker_context),
       service_(NULL),
+      weak_factory_io_to_io_(this),
       weak_factory_ui_to_ui_(this) {
 }
 
@@ -231,14 +239,44 @@ void PushMessagingMessageFilter::DidRegister(
     PushRegistrationStatus status) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   if (status == PUSH_REGISTRATION_STATUS_SUCCESS) {
-    SendRegisterSuccess(data, push_registration_id);
+    BrowserThread::PostTask(
+        BrowserThread::IO, FROM_HERE,
+        base::Bind(&PushMessagingMessageFilter::DoPersistRegistration,
+                   this, data, push_registration_id));
   } else {
     SendRegisterError(data, status);
   }
 }
 
+void PushMessagingMessageFilter::DoPersistRegistration(
+    const RegisterData& data,
+    const std::string& push_registration_id) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  service_worker_context_->context()->storage()->StoreUserData(
+      data.service_worker_registration_id,
+      data.requesting_origin,
+      kPushRegistrationIdServiceWorkerKey,
+      push_registration_id,
+      base::Bind(&PushMessagingMessageFilter::DidPersistRegistration,
+                 weak_factory_io_to_io_.GetWeakPtr(),
+                 data, push_registration_id));
+}
+
+void PushMessagingMessageFilter::DidPersistRegistration(
+    const RegisterData& data,
+    const std::string& push_registration_id,
+    ServiceWorkerStatusCode service_worker_status) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  if (service_worker_status == SERVICE_WORKER_OK) {
+    SendRegisterSuccess(data, push_registration_id);
+  } else {
+    SendRegisterError(data, PUSH_REGISTRATION_STATUS_STORAGE_ERROR);
+  }
+}
+
 void PushMessagingMessageFilter::SendRegisterError(
     const RegisterData& data, PushRegistrationStatus status) {
+  // Called from both UI and IO threads.
   if (data.FromDocument()) {
     Send(new PushMessagingMsg_RegisterFromDocumentError(
       data.render_frame_id, data.request_id, status));
@@ -251,6 +289,7 @@ void PushMessagingMessageFilter::SendRegisterError(
 
 void PushMessagingMessageFilter::SendRegisterSuccess(
     const RegisterData& data, const std::string& push_registration_id) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
   GURL push_endpoint(service()->PushEndpoint());
   if (data.FromDocument()) {
     Send(new PushMessagingMsg_RegisterFromDocumentSuccess(