Hook up the Push API with GCM's new ability to own encryption keys.

chrome/browser/push_messaging/push_messaging_service_impl.cc

Index: chrome/browser/push_messaging/push_messaging_service_impl.cc
diff --git a/chrome/browser/push_messaging/push_messaging_service_impl.cc b/chrome/browser/push_messaging/push_messaging_service_impl.cc
index 48bc0d813ced6d859e455ec399fe6cbe089e9c0f..5e92c5dd0eb372efa83cbb2283f341c13757b60f 100644
--- a/chrome/browser/push_messaging/push_messaging_service_impl.cc
+++ b/chrome/browser/push_messaging/push_messaging_service_impl.cc
@@ -211,8 +211,7 @@ void PushMessagingServiceImpl::OnMessage(const std::string& app_id,
   std::string data;
   // TODO(peter): Message payloads are disabled pending mandatory encryption.
   // https://crbug.com/449184
-  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
-          switches::kEnablePushMessagePayload)) {
+  if (EnableMessagePayloads()) {
     gcm::MessageData::const_iterator it = message.data.find("data");
     if (it != message.data.end())
       data = it->second;
@@ -315,15 +314,37 @@ GURL PushMessagingServiceImpl::GetPushEndpoint() {
   return GURL(std::string(kPushMessagingEndpoint));
 }
 
-// GetPublicEncryptionKey method -----------------------------------------------
+// GetPublicEncryptionKey methods ----------------------------------------------
 
 void PushMessagingServiceImpl::GetPublicEncryptionKey(
     const GURL& origin,
     int64_t service_worker_registration_id,
     const PushMessagingService::PublicKeyCallback& callback) {
-  // TODO(peter): Get the public key from the GCM Driver. Right now we have to
-  // return success=true here, otherwise subscriptions would fail.
-  callback.Run(true /* success */, std::vector<uint8_t>());
+  // An empty public key will be returned if payloads are not enabled.
+  if (!EnableMessagePayloads()) {
+    callback.Run(true /* success */, std::vector<uint8_t>());
+    return;
+  }
+
+  PushMessagingAppIdentifier app_identifier =
+      PushMessagingAppIdentifier::FindByServiceWorker(
+          profile_, origin, service_worker_registration_id);
+
+  DCHECK(!app_identifier.is_null());
+
+  GetGCMDriver()->GetPublicEncryptionKey(
+      app_identifier.app_id(),
+      base::Bind(&PushMessagingServiceImpl::DidGetPublicEncryptionKey,
+                 weak_factory_.GetWeakPtr(), callback));
+}
+
+void PushMessagingServiceImpl::DidGetPublicEncryptionKey(
+    const PushMessagingService::PublicKeyCallback& callback,
+    const std::vector<uint8_t>& public_key) const {
+  // I/O errors might prevent the GCM Driver from retrieving a key-pair.
+  const bool success = public_key.size();
+
+  callback.Run(success, public_key);
 }
 
 // Subscribe and GetPermissionStatus methods -----------------------------------
@@ -449,17 +470,28 @@ void PushMessagingServiceImpl::DidSubscribe(
     gcm::GCMClient::Result result) {
   content::PushRegistrationStatus status =
       content::PUSH_REGISTRATION_STATUS_SERVICE_ERROR;
-  std::vector<uint8_t> curve25519dh;
 
   switch (result) {
     case gcm::GCMClient::SUCCESS:
-      status = content::PUSH_REGISTRATION_STATUS_SUCCESS_FROM_PUSH_SERVICE;
-      app_identifier.PersistToPrefs(profile_);
-
-      // TODO(peter): Hook up getting the keys from the GCM Driver.
+      // Do not get a certificate if message payloads have not been enabled.
+      if (!EnableMessagePayloads()) {
+        DidSubscribeWithPublicEncryptionKey(
+            app_identifier, callback, subscription_id,
+            std::vector<uint8_t>() /* public_key */);
+        return;
+      }
 
-      IncreasePushSubscriptionCount(1, false /* is_pending */);
-      break;
+      // Make sure that this subscription has associated encryption keys prior
+      // to returning it to the developer - they'll need this information in
+      // order to send payloads to the user.
+      GetGCMDriver()->GetPublicEncryptionKey(
+          app_identifier.app_id(),
+          base::Bind(
+              &PushMessagingServiceImpl::DidSubscribeWithPublicEncryptionKey,
+              weak_factory_.GetWeakPtr(), app_identifier, callback,
+              subscription_id));
+
+      return;
     case gcm::GCMClient::INVALID_PARAMETER:
     case gcm::GCMClient::GCM_DISABLED:
     case gcm::GCMClient::ASYNC_OPERATION_PENDING:
@@ -473,8 +505,27 @@ void PushMessagingServiceImpl::DidSubscribe(
       break;
   }
 
-  SubscribeEnd(callback, subscription_id, curve25519dh, status);
+  SubscribeEndWithError(callback, status);
+}
+
+void PushMessagingServiceImpl::DidSubscribeWithPublicEncryptionKey(
+    const PushMessagingAppIdentifier& app_identifier,
+    const content::PushMessagingService::RegisterCallback& callback,
+    const std::string& subscription_id,
+    const std::vector<uint8_t>& public_key) {
+  if (!public_key.size() && EnableMessagePayloads()) {
+    SubscribeEndWithError(
+        callback, content::PUSH_REGISTRATION_STATUS_PUBLIC_KEY_UNAVAILABLE);
+    return;
+  }
+
+  app_identifier.PersistToPrefs(profile_);
+
+  IncreasePushSubscriptionCount(1, false /* is_pending */);
   DecreasePushSubscriptionCount(1, true /* was_pending */);
+
+  SubscribeEnd(callback, subscription_id, public_key,
+               content::PUSH_REGISTRATION_STATUS_SUCCESS_FROM_PUSH_SERVICE);
 }
 
 void PushMessagingServiceImpl::DidRequestPermission(
@@ -672,6 +723,11 @@ bool PushMessagingServiceImpl::IsPermissionSet(const GURL& origin) {
          blink::WebPushPermissionStatusGranted;
 }
 
+bool PushMessagingServiceImpl::EnableMessagePayloads() const {
+  return base::CommandLine::ForCurrentProcess()->HasSwitch(
+      switches::kEnablePushMessagePayload);
+}
+
 gcm::GCMDriver* PushMessagingServiceImpl::GetGCMDriver() const {
   gcm::GCMProfileService* gcm_profile_service =
       gcm::GCMProfileServiceFactory::GetForProfile(profile_);