| // Copyright 2014 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/push_messaging/push_messaging_manager.h" |
| |
| #include <string> |
| #include <vector> |
| |
| #include "base/bind.h" |
| #include "base/bind_helpers.h" |
| #include "base/command_line.h" |
| #include "base/logging.h" |
| #include "base/macros.h" |
| #include "base/metrics/histogram_macros.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.mojom.h" |
| #include "content/public/browser/browser_context.h" |
| #include "content/public/browser/permission_manager.h" |
| #include "content/public/browser/permission_type.h" |
| #include "content/public/browser/push_messaging_service.h" |
| #include "content/public/browser/render_frame_host.h" |
| #include "content/public/browser/web_contents.h" |
| #include "content/public/common/child_process_host.h" |
| #include "content/public/common/console_message_level.h" |
| #include "content/public/common/content_switches.h" |
| #include "content/public/common/push_messaging_status.mojom.h" |
| #include "third_party/blink/public/platform/modules/permissions/permission_status.mojom.h" |
| |
| namespace content { |
| |
| // Service Worker database keys. If a registration ID is stored, the stored |
| // sender ID must be the one used to register. Unfortunately, this isn't always |
| // true of pre-InstanceID registrations previously stored in the database, but |
| // fortunately it's less important for their sender ID to be accurate. |
| const char kPushSenderIdServiceWorkerKey[] = "push_sender_id"; |
| const char kPushRegistrationIdServiceWorkerKey[] = "push_registration_id"; |
| |
| namespace { |
| |
| // Chrome currently does not support the Push API in incognito. |
| const char kIncognitoPushUnsupportedMessage[] = |
| "Chrome currently does not support the Push API in incognito mode " |
| "(https://crbug.com/401439). There is deliberately no way to " |
| "feature-detect this, since incognito mode needs to be undetectable by " |
| "websites."; |
| |
| // These UMA methods are called from the IO and/or UI threads. Racey but ok, see |
| // https://groups.google.com/a/chromium.org/d/msg/chromium-dev/FNzZRJtN2aw/Aw0CWAXJJ1kJ |
| void RecordRegistrationStatus(mojom::PushRegistrationStatus status) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO) || |
| BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| UMA_HISTOGRAM_ENUMERATION("PushMessaging.RegistrationStatus", status); |
| } |
| void RecordUnregistrationStatus(mojom::PushUnregistrationStatus status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| UMA_HISTOGRAM_ENUMERATION("PushMessaging.UnregistrationStatus", status); |
| } |
| void RecordGetRegistrationStatus(mojom::PushGetRegistrationStatus status) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO) || |
| BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| UMA_HISTOGRAM_ENUMERATION("PushMessaging.GetRegistrationStatus", status); |
| } |
| |
| const char* PushUnregistrationStatusToString( |
| mojom::PushUnregistrationStatus status) { |
| switch (status) { |
| case mojom::PushUnregistrationStatus::SUCCESS_UNREGISTERED: |
| return "Unregistration successful - from push service"; |
| |
| case mojom::PushUnregistrationStatus::SUCCESS_WAS_NOT_REGISTERED: |
| return "Unregistration successful - was not registered"; |
| |
| case mojom::PushUnregistrationStatus::PENDING_NETWORK_ERROR: |
| return "Unregistration pending - a network error occurred, but it will " |
| "be retried until it succeeds"; |
| |
| case mojom::PushUnregistrationStatus::NO_SERVICE_WORKER: |
| return "Unregistration failed - no Service Worker"; |
| |
| case mojom::PushUnregistrationStatus::SERVICE_NOT_AVAILABLE: |
| return "Unregistration failed - push service not available"; |
| |
| case mojom::PushUnregistrationStatus::PENDING_SERVICE_ERROR: |
| return "Unregistration pending - a push service error occurred, but it " |
| "will be retried until it succeeds"; |
| |
| case mojom::PushUnregistrationStatus::STORAGE_ERROR: |
| return "Unregistration failed - storage error"; |
| |
| case mojom::PushUnregistrationStatus::NETWORK_ERROR: |
| return "Unregistration failed - could not connect to push server"; |
| } |
| NOTREACHED(); |
| return ""; |
| } |
| |
| // Returns whether |sender_info| contains a valid application server key, that |
| // is, a NIST P-256 public key in uncompressed format. |
| bool IsApplicationServerKey(const std::string& sender_info) { |
| return sender_info.size() == 65 && sender_info[0] == 0x04; |
| } |
| |
| // Returns sender_info if non-empty, otherwise checks if stored_sender_id |
| // may be used as a fallback and if so, returns stored_sender_id instead. |
| // |
| // This is in order to support the legacy way of subscribing from a service |
| // worker (first subscribe from the document using a gcm_sender_id set in the |
| // manifest, and then subscribe from the service worker with no key). |
| // |
| // An empty string will be returned if sender_info is empty and the fallback |
| // is not a numeric gcm sender id. |
| std::string FixSenderInfo(const std::string& sender_info, |
| const std::string& stored_sender_id) { |
| if (!sender_info.empty()) |
| return sender_info; |
| if (base::ContainsOnlyChars(stored_sender_id, "0123456789")) |
| return stored_sender_id; |
| return std::string(); |
| } |
| |
| } // namespace |
| |
| struct PushMessagingManager::RegisterData { |
| RegisterData(); |
| RegisterData(RegisterData&& other) = default; |
| |
| bool FromDocument() const; |
| |
| GURL requesting_origin; |
| int64_t service_worker_registration_id; |
| std::string existing_subscription_id; |
| PushSubscriptionOptions options; |
| SubscribeCallback callback; |
| |
| // The following member should only be read if FromDocument() is true. |
| int render_frame_id; |
| |
| // True if the call to register was made with a user gesture. |
| bool user_gesture; |
| }; |
| |
| // Inner core of the PushMessagingManager which lives on the UI thread. |
| class PushMessagingManager::Core { |
| public: |
| Core(const base::WeakPtr<PushMessagingManager>& io_parent, |
| int render_process_id); |
| |
| // Public Register methods on UI thread -------------------------------------- |
| |
| // Called via PostTask from IO thread. |
| void RegisterOnUI(RegisterData data); |
| |
| // Public Unregister methods on UI thread ------------------------------------ |
| |
| // Called via PostTask from IO thread. |
| void UnregisterFromService(UnsubscribeCallback callback, |
| int64_t service_worker_registration_id, |
| const GURL& requesting_origin, |
| const std::string& sender_id); |
| |
| // Public GetSubscription methods on UI thread ------------------------------- |
| |
| // Callback called on UI thread. |
| void GetSubscriptionDidGetInfoOnUI(GetSubscriptionCallback callback, |
| const GURL& origin, |
| int64_t service_worker_registration_id, |
| const GURL& endpoint, |
| const std::string& sender_info, |
| bool is_valid, |
| const std::vector<uint8_t>& p256dh, |
| const std::vector<uint8_t>& auth); |
| |
| // Callback called on UI thread. |
| void GetSubscriptionDidUnsubscribe( |
| GetSubscriptionCallback callback, |
| mojom::PushGetRegistrationStatus get_status, |
| mojom::PushUnregistrationStatus unsubscribe_status); |
| |
| // Public helper methods on UI thread ---------------------------------------- |
| |
| // Called via PostTask from IO thread. |callback| will be run on UI thread. |
| void GetSubscriptionInfoOnUI( |
| const GURL& origin, |
| int64_t service_worker_registration_id, |
| const std::string& sender_id, |
| const std::string& push_subscription_id, |
| PushMessagingService::SubscriptionInfoCallback callback); |
| |
| // Called (directly) from both the UI and IO threads. |
| bool is_incognito() const { return is_incognito_; } |
| |
| // Returns a push messaging service. May return null. |
| PushMessagingService* service(); |
| |
| // Returns a weak ptr. Must only be called on the UI thread (and hence can |
| // only be called from the outer class's constructor). |
| base::WeakPtr<Core> GetWeakPtrFromIOParentConstructor(); |
| |
| private: |
| friend struct BrowserThread::DeleteOnThread<BrowserThread::UI>; |
| friend class base::DeleteHelper<Core>; |
| |
| ~Core(); |
| |
| // Private Register methods on UI thread ------------------------------------- |
| |
| void DidRequestPermissionInIncognito(RegisterData data, |
| blink::mojom::PermissionStatus status); |
| |
| void DidRegister(RegisterData data, |
| const std::string& push_subscription_id, |
| const std::vector<uint8_t>& p256dh, |
| const std::vector<uint8_t>& auth, |
| mojom::PushRegistrationStatus status); |
| |
| // Private Unregister methods on UI thread ----------------------------------- |
| |
| void DidUnregisterFromService( |
| UnsubscribeCallback callback, |
| int64_t service_worker_registration_id, |
| mojom::PushUnregistrationStatus unregistration_status); |
| |
| // Outer part of the PushMessagingManager which lives on the IO thread. |
| base::WeakPtr<PushMessagingManager> io_parent_; |
| |
| int render_process_id_; |
| |
| bool is_incognito_; |
| |
| base::WeakPtrFactory<Core> weak_factory_ui_to_ui_; |
| |
| DISALLOW_COPY_AND_ASSIGN(Core); |
| }; |
| |
| PushMessagingManager::RegisterData::RegisterData() |
| : service_worker_registration_id(0), |
| render_frame_id(ChildProcessHost::kInvalidUniqueID) {} |
| |
| bool PushMessagingManager::RegisterData::FromDocument() const { |
| return render_frame_id != ChildProcessHost::kInvalidUniqueID; |
| } |
| |
| PushMessagingManager::Core::Core( |
| const base::WeakPtr<PushMessagingManager>& io_parent, |
| int render_process_id) |
| : io_parent_(io_parent), |
| render_process_id_(render_process_id), |
| weak_factory_ui_to_ui_(this) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| RenderProcessHost* process_host = |
| RenderProcessHost::FromID(render_process_id_); // Can't be null yet. |
| is_incognito_ = process_host->GetBrowserContext()->IsOffTheRecord(); |
| } |
| |
| PushMessagingManager::Core::~Core() {} |
| |
| PushMessagingManager::PushMessagingManager( |
| int render_process_id, |
| ServiceWorkerContextWrapper* service_worker_context) |
| : service_worker_context_(service_worker_context), |
| weak_factory_io_to_io_(this) { |
| // Although this class is used only on the IO thread, it is constructed on UI. |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| // Normally, it would be unsafe to obtain a weak pointer from the UI thread, |
| // but it's ok in the constructor since we can't be destroyed before our |
| // constructor finishes. |
| ui_core_.reset( |
| new Core(weak_factory_io_to_io_.GetWeakPtr(), render_process_id)); |
| ui_core_weak_ptr_ = ui_core_->GetWeakPtrFromIOParentConstructor(); |
| |
| PushMessagingService* service = ui_core_->service(); |
| service_available_ = !!service; |
| |
| if (service_available_) { |
| default_endpoint_ = service->GetEndpoint(false /* standard_protocol */); |
| web_push_protocol_endpoint_ = |
| service->GetEndpoint(true /* standard_protocol */); |
| } |
| } |
| |
| PushMessagingManager::~PushMessagingManager() {} |
| |
| void PushMessagingManager::BindRequest( |
| mojom::PushMessagingRequest request) { |
| bindings_.AddBinding(this, std::move(request)); |
| } |
| |
| // Subscribe methods on both IO and UI threads, merged in order of use from |
| // PushMessagingManager and Core. |
| // ----------------------------------------------------------------------------- |
| |
| void PushMessagingManager::Subscribe(int32_t render_frame_id, |
| int64_t service_worker_registration_id, |
| const PushSubscriptionOptions& options, |
| bool user_gesture, |
| SubscribeCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| // TODO(mvanouwerkerk): Validate arguments? |
| RegisterData data; |
| |
| // Will be ChildProcessHost::kInvalidUniqueID in requests from Service Worker. |
| data.render_frame_id = render_frame_id; |
| |
| data.service_worker_registration_id = service_worker_registration_id; |
| data.callback = std::move(callback); |
| data.options = options; |
| data.user_gesture = user_gesture; |
| |
| ServiceWorkerRegistration* service_worker_registration = |
| service_worker_context_->GetLiveRegistration( |
| data.service_worker_registration_id); |
| if (!service_worker_registration || |
| !service_worker_registration->active_version()) { |
| SendSubscriptionError(std::move(data), |
| mojom::PushRegistrationStatus::NO_SERVICE_WORKER); |
| return; |
| } |
| data.requesting_origin = service_worker_registration->pattern().GetOrigin(); |
| |
| DCHECK(!(data.options.sender_info.empty() && data.FromDocument())); |
| |
| int64_t registration_id = data.service_worker_registration_id; |
| service_worker_context_->GetRegistrationUserData( |
| registration_id, |
| {kPushRegistrationIdServiceWorkerKey, kPushSenderIdServiceWorkerKey}, |
| base::BindOnce(&PushMessagingManager::DidCheckForExistingRegistration, |
| weak_factory_io_to_io_.GetWeakPtr(), std::move(data))); |
| } |
| |
| void PushMessagingManager::DidCheckForExistingRegistration( |
| RegisterData data, |
| const std::vector<std::string>& subscription_id_and_sender_id, |
| blink::ServiceWorkerStatusCode service_worker_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| // Validate the stored subscription against the subscription request made by |
| // the developer. The authorized entity must match. |
| if (service_worker_status == blink::ServiceWorkerStatusCode::kOk) { |
| DCHECK_EQ(2u, subscription_id_and_sender_id.size()); |
| |
| const std::string& subscription_id = subscription_id_and_sender_id[0]; |
| const std::string& stored_sender_id = subscription_id_and_sender_id[1]; |
| |
| std::string fixed_sender_id = |
| FixSenderInfo(data.options.sender_info, stored_sender_id); |
| if (fixed_sender_id.empty()) { |
| SendSubscriptionError(std::move(data), |
| mojom::PushRegistrationStatus::NO_SENDER_ID); |
| return; |
| } |
| |
| if (fixed_sender_id != stored_sender_id) { |
| SendSubscriptionError(std::move(data), |
| mojom::PushRegistrationStatus::SENDER_ID_MISMATCH); |
| return; |
| } |
| |
| data.existing_subscription_id = subscription_id; |
| } |
| |
| // TODO(peter): Handle failures other than |
| // blink::ServiceWorkerStatusCode::kErrorNotFound by rejecting |
| // the subscription algorithm instead of trying to subscribe. |
| |
| if (!data.options.sender_info.empty()) { |
| BrowserThread::PostTask( |
| BrowserThread::UI, FROM_HERE, |
| base::BindOnce(&Core::RegisterOnUI, base::Unretained(ui_core_.get()), |
| std::move(data))); |
| } else { |
| // No |sender_info| was provided by the developer. Fall back to checking |
| // whether a previous subscription did identify a sender. |
| int64_t registration_id = data.service_worker_registration_id; |
| service_worker_context_->GetRegistrationUserData( |
| registration_id, {kPushSenderIdServiceWorkerKey}, |
| base::BindOnce(&PushMessagingManager::DidGetSenderIdFromStorage, |
| weak_factory_io_to_io_.GetWeakPtr(), std::move(data))); |
| } |
| } |
| |
| void PushMessagingManager::DidGetSenderIdFromStorage( |
| RegisterData data, |
| const std::vector<std::string>& stored_sender_id, |
| blink::ServiceWorkerStatusCode service_worker_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (service_worker_status != blink::ServiceWorkerStatusCode::kOk) { |
| SendSubscriptionError(std::move(data), |
| mojom::PushRegistrationStatus::NO_SENDER_ID); |
| return; |
| } |
| DCHECK_EQ(1u, stored_sender_id.size()); |
| // We should only be here because no sender info was supplied to subscribe(). |
| DCHECK(data.options.sender_info.empty()); |
| std::string fixed_sender_id = |
| FixSenderInfo(data.options.sender_info, stored_sender_id[0]); |
| if (fixed_sender_id.empty()) { |
| SendSubscriptionError(std::move(data), |
| mojom::PushRegistrationStatus::NO_SENDER_ID); |
| return; |
| } |
| data.options.sender_info = fixed_sender_id; |
| BrowserThread::PostTask( |
| BrowserThread::UI, FROM_HERE, |
| base::BindOnce(&Core::RegisterOnUI, base::Unretained(ui_core_.get()), |
| std::move(data))); |
| } |
| |
| void PushMessagingManager::Core::RegisterOnUI( |
| PushMessagingManager::RegisterData data) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| PushMessagingService* push_service = service(); |
| if (!push_service) { |
| if (!is_incognito()) { |
| // This might happen if InstanceIDProfileService::IsInstanceIDEnabled |
| // returns false because the Instance ID kill switch was enabled. |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce(&PushMessagingManager::SendSubscriptionError, |
| io_parent_, std::move(data), |
| mojom::PushRegistrationStatus::SERVICE_NOT_AVAILABLE)); |
| } else { |
| // Prevent websites from detecting incognito mode, by emulating what would |
| // have happened if we had a PushMessagingService available. |
| if (!data.FromDocument() || !data.options.user_visible_only) { |
| // Throw a permission denied error under the same circumstances. |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce( |
| &PushMessagingManager::SendSubscriptionError, io_parent_, |
| std::move(data), |
| mojom::PushRegistrationStatus::INCOGNITO_PERMISSION_DENIED)); |
| } else { |
| RenderFrameHost* render_frame_host = |
| RenderFrameHost::FromID(render_process_id_, data.render_frame_id); |
| WebContents* web_contents = |
| WebContents::FromRenderFrameHost(render_frame_host); |
| if (web_contents) { |
| web_contents->GetMainFrame()->AddMessageToConsole( |
| CONSOLE_MESSAGE_LEVEL_ERROR, kIncognitoPushUnsupportedMessage); |
| |
| BrowserContext* browser_context = web_contents->GetBrowserContext(); |
| |
| // It's valid for embedders to return a null permission manager. |
| // Immediately reject the permission request when this happens. |
| if (!browser_context->GetPermissionManager()) { |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce(&PushMessagingManager::SendSubscriptionError, |
| io_parent_, std::move(data), |
| mojom::PushRegistrationStatus:: |
| INCOGNITO_PERMISSION_DENIED)); |
| |
| return; |
| } |
| |
| // Request notifications permission (which will fail, since |
| // notifications aren't supported in incognito), so the website can't |
| // detect whether incognito is active. |
| GURL requesting_origin = data.requesting_origin; |
| browser_context->GetPermissionManager()->RequestPermission( |
| PermissionType::NOTIFICATIONS, render_frame_host, |
| requesting_origin, data.user_gesture, |
| base::Bind( |
| &PushMessagingManager::Core::DidRequestPermissionInIncognito, |
| weak_factory_ui_to_ui_.GetWeakPtr(), base::Passed(&data))); |
| } |
| } |
| } |
| return; |
| } |
| |
| int64_t registration_id = data.service_worker_registration_id; |
| GURL requesting_origin = data.requesting_origin; |
| PushSubscriptionOptions options = data.options; |
| int render_frame_id = data.render_frame_id; |
| if (data.FromDocument()) { |
| push_service->SubscribeFromDocument( |
| requesting_origin, registration_id, render_process_id_, render_frame_id, |
| options, data.user_gesture, |
| base::Bind(&Core::DidRegister, weak_factory_ui_to_ui_.GetWeakPtr(), |
| base::Passed(&data))); |
| } else { |
| push_service->SubscribeFromWorker( |
| requesting_origin, registration_id, options, |
| base::Bind(&Core::DidRegister, weak_factory_ui_to_ui_.GetWeakPtr(), |
| base::Passed(&data))); |
| } |
| } |
| |
| void PushMessagingManager::Core::DidRequestPermissionInIncognito( |
| RegisterData data, |
| blink::mojom::PermissionStatus status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| // Notification permission should always be denied in incognito. |
| DCHECK_EQ(blink::mojom::PermissionStatus::DENIED, status); |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce( |
| &PushMessagingManager::SendSubscriptionError, io_parent_, |
| std::move(data), |
| mojom::PushRegistrationStatus::INCOGNITO_PERMISSION_DENIED)); |
| } |
| |
| void PushMessagingManager::Core::DidRegister( |
| RegisterData data, |
| const std::string& push_subscription_id, |
| const std::vector<uint8_t>& p256dh, |
| const std::vector<uint8_t>& auth, |
| mojom::PushRegistrationStatus status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| |
| // TODO(nator): Handle the case where |push_subscription_id| and |
| // |data.existing_subscription_id| are not the same. Right now we just |
| // override the old subscription ID and encryption information. |
| const bool subscription_changed = |
| push_subscription_id != data.existing_subscription_id; |
| |
| if (status == mojom::PushRegistrationStatus::SUCCESS_FROM_PUSH_SERVICE) { |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce( |
| &PushMessagingManager::PersistRegistrationOnIO, io_parent_, |
| std::move(data), push_subscription_id, p256dh, auth, |
| subscription_changed |
| ? mojom::PushRegistrationStatus:: |
| SUCCESS_NEW_SUBSCRIPTION_FROM_PUSH_SERVICE |
| : mojom::PushRegistrationStatus::SUCCESS_FROM_PUSH_SERVICE)); |
| } else { |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce(&PushMessagingManager::SendSubscriptionError, io_parent_, |
| std::move(data), status)); |
| } |
| } |
| |
| void PushMessagingManager::PersistRegistrationOnIO( |
| RegisterData data, |
| const std::string& push_subscription_id, |
| const std::vector<uint8_t>& p256dh, |
| const std::vector<uint8_t>& auth, |
| mojom::PushRegistrationStatus status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| GURL requesting_origin = data.requesting_origin; |
| int64_t registration_id = data.service_worker_registration_id; |
| std::string sender_info = data.options.sender_info; |
| |
| service_worker_context_->StoreRegistrationUserData( |
| registration_id, requesting_origin, |
| {{kPushRegistrationIdServiceWorkerKey, push_subscription_id}, |
| {kPushSenderIdServiceWorkerKey, sender_info}}, |
| base::BindOnce(&PushMessagingManager::DidPersistRegistrationOnIO, |
| weak_factory_io_to_io_.GetWeakPtr(), std::move(data), |
| push_subscription_id, p256dh, auth, status)); |
| } |
| |
| void PushMessagingManager::DidPersistRegistrationOnIO( |
| RegisterData data, |
| const std::string& push_subscription_id, |
| const std::vector<uint8_t>& p256dh, |
| const std::vector<uint8_t>& auth, |
| mojom::PushRegistrationStatus push_registration_status, |
| blink::ServiceWorkerStatusCode service_worker_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (service_worker_status == blink::ServiceWorkerStatusCode::kOk) { |
| SendSubscriptionSuccess(std::move(data), push_registration_status, |
| push_subscription_id, p256dh, auth); |
| } else { |
| // TODO(johnme): Unregister, so PushMessagingServiceImpl can decrease count. |
| SendSubscriptionError(std::move(data), |
| mojom::PushRegistrationStatus::STORAGE_ERROR); |
| } |
| } |
| |
| void PushMessagingManager::SendSubscriptionError( |
| RegisterData data, |
| mojom::PushRegistrationStatus status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| std::move(data.callback) |
| .Run(status, base::nullopt /* endpoint */, base::nullopt /* options */, |
| base::nullopt /* p256dh */, base::nullopt /* auth */); |
| RecordRegistrationStatus(status); |
| } |
| |
| void PushMessagingManager::SendSubscriptionSuccess( |
| RegisterData data, |
| mojom::PushRegistrationStatus status, |
| const std::string& push_subscription_id, |
| const std::vector<uint8_t>& p256dh, |
| const std::vector<uint8_t>& auth) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!service_available_) { |
| // This shouldn't be possible in incognito mode, since we've already checked |
| // that we have an existing registration. Hence it's ok to throw an error. |
| DCHECK(!ui_core_->is_incognito()); |
| SendSubscriptionError(std::move(data), |
| mojom::PushRegistrationStatus::SERVICE_NOT_AVAILABLE); |
| return; |
| } |
| |
| const GURL endpoint = CreateEndpoint( |
| IsApplicationServerKey(data.options.sender_info), push_subscription_id); |
| |
| std::move(data.callback).Run(status, endpoint, data.options, p256dh, auth); |
| |
| RecordRegistrationStatus(status); |
| } |
| |
| // Unsubscribe methods on both IO and UI threads, merged in order of use from |
| // PushMessagingManager and Core. |
| // ----------------------------------------------------------------------------- |
| |
| void PushMessagingManager::Unsubscribe(int64_t service_worker_registration_id, |
| UnsubscribeCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| ServiceWorkerRegistration* service_worker_registration = |
| service_worker_context_->GetLiveRegistration( |
| service_worker_registration_id); |
| if (!service_worker_registration) { |
| DidUnregister(std::move(callback), |
| mojom::PushUnregistrationStatus::NO_SERVICE_WORKER); |
| return; |
| } |
| |
| service_worker_context_->GetRegistrationUserData( |
| service_worker_registration_id, {kPushSenderIdServiceWorkerKey}, |
| base::BindOnce(&PushMessagingManager::UnsubscribeHavingGottenSenderId, |
| weak_factory_io_to_io_.GetWeakPtr(), std::move(callback), |
| service_worker_registration_id, |
| service_worker_registration->pattern().GetOrigin())); |
| } |
| |
| void PushMessagingManager::UnsubscribeHavingGottenSenderId( |
| UnsubscribeCallback callback, |
| int64_t service_worker_registration_id, |
| const GURL& requesting_origin, |
| const std::vector<std::string>& sender_ids, |
| blink::ServiceWorkerStatusCode service_worker_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| std::string sender_id; |
| if (service_worker_status == blink::ServiceWorkerStatusCode::kOk) { |
| DCHECK_EQ(1u, sender_ids.size()); |
| sender_id = sender_ids[0]; |
| } |
| BrowserThread::PostTask( |
| BrowserThread::UI, FROM_HERE, |
| base::BindOnce(&Core::UnregisterFromService, |
| base::Unretained(ui_core_.get()), std::move(callback), |
| service_worker_registration_id, requesting_origin, |
| sender_id)); |
| } |
| |
| void PushMessagingManager::Core::UnregisterFromService( |
| UnsubscribeCallback callback, |
| int64_t service_worker_registration_id, |
| const GURL& requesting_origin, |
| const std::string& sender_id) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| PushMessagingService* push_service = service(); |
| if (!push_service) { |
| // This shouldn't be possible in incognito mode, since we've already checked |
| // that we have an existing registration. Hence it's ok to throw an error. |
| DCHECK(!is_incognito()); |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce(&PushMessagingManager::DidUnregister, io_parent_, |
| std::move(callback), |
| mojom::PushUnregistrationStatus::SERVICE_NOT_AVAILABLE)); |
| return; |
| } |
| |
| push_service->Unsubscribe( |
| mojom::PushUnregistrationReason::JAVASCRIPT_API, requesting_origin, |
| service_worker_registration_id, sender_id, |
| base::Bind(&Core::DidUnregisterFromService, |
| weak_factory_ui_to_ui_.GetWeakPtr(), base::Passed(&callback), |
| service_worker_registration_id)); |
| } |
| |
| void PushMessagingManager::Core::DidUnregisterFromService( |
| UnsubscribeCallback callback, |
| int64_t service_worker_registration_id, |
| mojom::PushUnregistrationStatus unregistration_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce(&PushMessagingManager::DidUnregister, io_parent_, |
| std::move(callback), unregistration_status)); |
| } |
| |
| void PushMessagingManager::DidUnregister( |
| UnsubscribeCallback callback, |
| mojom::PushUnregistrationStatus unregistration_status) { |
| // Only called from IO thread, but would be safe to call from UI thread. |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| switch (unregistration_status) { |
| case mojom::PushUnregistrationStatus::SUCCESS_UNREGISTERED: |
| case mojom::PushUnregistrationStatus::PENDING_NETWORK_ERROR: |
| case mojom::PushUnregistrationStatus::PENDING_SERVICE_ERROR: |
| std::move(callback).Run(blink::WebPushError::kErrorTypeNone, |
| true /* did_unsubscribe */, |
| base::nullopt /* error_message */); |
| break; |
| case mojom::PushUnregistrationStatus::SUCCESS_WAS_NOT_REGISTERED: |
| std::move(callback).Run(blink::WebPushError::kErrorTypeNone, |
| false /* did_unsubscribe */, |
| base::nullopt /* error_message */); |
| break; |
| case mojom::PushUnregistrationStatus::NO_SERVICE_WORKER: |
| case mojom::PushUnregistrationStatus::SERVICE_NOT_AVAILABLE: |
| case mojom::PushUnregistrationStatus::STORAGE_ERROR: |
| std::move(callback).Run(blink::WebPushError::kErrorTypeAbort, false, |
| std::string(PushUnregistrationStatusToString( |
| unregistration_status)) /* error_message */); |
| break; |
| case mojom::PushUnregistrationStatus::NETWORK_ERROR: |
| NOTREACHED(); |
| break; |
| } |
| RecordUnregistrationStatus(unregistration_status); |
| } |
| |
| // GetSubscription methods on both IO and UI threads, merged in order of use |
| // from PushMessagingManager and Core. |
| // ----------------------------------------------------------------------------- |
| |
| void PushMessagingManager::GetSubscription( |
| int64_t service_worker_registration_id, |
| GetSubscriptionCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| // TODO(johnme): Validate arguments? |
| service_worker_context_->GetRegistrationUserData( |
| service_worker_registration_id, |
| {kPushRegistrationIdServiceWorkerKey, kPushSenderIdServiceWorkerKey}, |
| base::BindOnce(&PushMessagingManager::DidGetSubscription, |
| weak_factory_io_to_io_.GetWeakPtr(), std::move(callback), |
| service_worker_registration_id)); |
| } |
| |
| void PushMessagingManager::DidGetSubscription( |
| GetSubscriptionCallback callback, |
| int64_t service_worker_registration_id, |
| const std::vector<std::string>& push_subscription_id_and_sender_info, |
| blink::ServiceWorkerStatusCode service_worker_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| mojom::PushGetRegistrationStatus get_status = |
| mojom::PushGetRegistrationStatus::STORAGE_ERROR; |
| switch (service_worker_status) { |
| case blink::ServiceWorkerStatusCode::kOk: { |
| DCHECK_EQ(2u, push_subscription_id_and_sender_info.size()); |
| const std::string& push_subscription_id = |
| push_subscription_id_and_sender_info[0]; |
| const std::string& sender_info = push_subscription_id_and_sender_info[1]; |
| |
| if (!service_available_) { |
| // Return not found in incognito mode, so websites can't detect it. |
| get_status = |
| ui_core_->is_incognito() |
| ? mojom::PushGetRegistrationStatus:: |
| INCOGNITO_REGISTRATION_NOT_FOUND |
| : mojom::PushGetRegistrationStatus::SERVICE_NOT_AVAILABLE; |
| break; |
| } |
| |
| ServiceWorkerRegistration* registration = |
| service_worker_context_->GetLiveRegistration( |
| service_worker_registration_id); |
| if (!registration) { |
| get_status = mojom::PushGetRegistrationStatus::NO_LIVE_SERVICE_WORKER; |
| break; |
| } |
| |
| const GURL origin = registration->pattern().GetOrigin(); |
| |
| const bool uses_standard_protocol = IsApplicationServerKey(sender_info); |
| const GURL endpoint = |
| CreateEndpoint(uses_standard_protocol, push_subscription_id); |
| |
| BrowserThread::PostTask( |
| BrowserThread::UI, FROM_HERE, |
| base::BindOnce(&Core::GetSubscriptionInfoOnUI, |
| base::Unretained(ui_core_.get()), origin, |
| service_worker_registration_id, sender_info, |
| push_subscription_id, |
| base::Bind(&Core::GetSubscriptionDidGetInfoOnUI, |
| ui_core_weak_ptr_, base::Passed(&callback), |
| origin, service_worker_registration_id, |
| endpoint, sender_info))); |
| |
| return; |
| } |
| case blink::ServiceWorkerStatusCode::kErrorNotFound: { |
| get_status = mojom::PushGetRegistrationStatus::REGISTRATION_NOT_FOUND; |
| break; |
| } |
| case blink::ServiceWorkerStatusCode::kErrorFailed: { |
| get_status = mojom::PushGetRegistrationStatus::STORAGE_ERROR; |
| break; |
| } |
| case blink::ServiceWorkerStatusCode::kErrorAbort: |
| case blink::ServiceWorkerStatusCode::kErrorStartWorkerFailed: |
| case blink::ServiceWorkerStatusCode::kErrorProcessNotFound: |
| case blink::ServiceWorkerStatusCode::kErrorExists: |
| case blink::ServiceWorkerStatusCode::kErrorInstallWorkerFailed: |
| case blink::ServiceWorkerStatusCode::kErrorActivateWorkerFailed: |
| case blink::ServiceWorkerStatusCode::kErrorIpcFailed: |
| case blink::ServiceWorkerStatusCode::kErrorNetwork: |
| case blink::ServiceWorkerStatusCode::kErrorSecurity: |
| case blink::ServiceWorkerStatusCode::kErrorEventWaitUntilRejected: |
| case blink::ServiceWorkerStatusCode::kErrorState: |
| case blink::ServiceWorkerStatusCode::kErrorTimeout: |
| case blink::ServiceWorkerStatusCode::kErrorScriptEvaluateFailed: |
| case blink::ServiceWorkerStatusCode::kErrorDiskCache: |
| case blink::ServiceWorkerStatusCode::kErrorRedundant: |
| case blink::ServiceWorkerStatusCode::kErrorDisallowed: { |
| NOTREACHED() << "Got unexpected error code: " |
| << static_cast<uint32_t>(service_worker_status) << " " |
| << blink::ServiceWorkerStatusToString(service_worker_status); |
| get_status = mojom::PushGetRegistrationStatus::STORAGE_ERROR; |
| break; |
| } |
| } |
| std::move(callback).Run(get_status, base::nullopt /* endpoint */, |
| base::nullopt /* options */, |
| base::nullopt /* p256dh */, base::nullopt /* auth */); |
| RecordGetRegistrationStatus(get_status); |
| } |
| |
| void PushMessagingManager::Core::GetSubscriptionDidGetInfoOnUI( |
| GetSubscriptionCallback callback, |
| const GURL& origin, |
| int64_t service_worker_registration_id, |
| const GURL& endpoint, |
| const std::string& sender_info, |
| bool is_valid, |
| const std::vector<uint8_t>& p256dh, |
| const std::vector<uint8_t>& auth) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| if (is_valid) { |
| PushSubscriptionOptions options; |
| // Chrome rejects subscription requests with userVisibleOnly false, so it |
| // must have been true. TODO(harkness): If Chrome starts accepting silent |
| // push subscriptions with userVisibleOnly false, the bool will need to be |
| // stored. |
| options.user_visible_only = true; |
| options.sender_info = sender_info; |
| |
| mojom::PushGetRegistrationStatus status = |
| mojom::PushGetRegistrationStatus::SUCCESS; |
| |
| BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, |
| base::BindOnce(std::move(callback), status, |
| endpoint, options, p256dh, auth)); |
| |
| RecordGetRegistrationStatus(status); |
| } else { |
| PushMessagingService* push_service = service(); |
| if (!push_service) { |
| // Shouldn't be possible to have a stored push subscription in a profile |
| // with no push service, but this case can occur when the renderer is |
| // shutting down. |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce(std::move(callback), |
| mojom::PushGetRegistrationStatus::RENDERER_SHUTDOWN, |
| base::nullopt /* endpoint */, |
| base::nullopt /* options */, |
| base::nullopt /* p256dh */, base::nullopt /* auth */)); |
| return; |
| } |
| |
| // Uh-oh! Although there was a cached subscription in the Service Worker |
| // database, it did not have matching counterparts in the |
| // PushMessagingAppIdentifier map and/or GCM Store. Unsubscribe to fix this |
| // inconsistency. |
| mojom::PushGetRegistrationStatus status = |
| mojom::PushGetRegistrationStatus::STORAGE_CORRUPT; |
| |
| push_service->Unsubscribe( |
| mojom::PushUnregistrationReason::GET_SUBSCRIPTION_STORAGE_CORRUPT, |
| origin, service_worker_registration_id, sender_info, |
| base::Bind(&Core::GetSubscriptionDidUnsubscribe, |
| weak_factory_ui_to_ui_.GetWeakPtr(), base::Passed(&callback), |
| status)); |
| |
| RecordGetRegistrationStatus(status); |
| } |
| } |
| |
| void PushMessagingManager::Core::GetSubscriptionDidUnsubscribe( |
| GetSubscriptionCallback callback, |
| mojom::PushGetRegistrationStatus get_status, |
| mojom::PushUnregistrationStatus unsubscribe_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| BrowserThread::PostTask( |
| BrowserThread::IO, FROM_HERE, |
| base::BindOnce(std::move(callback), get_status, |
| base::nullopt /* endpoint */, base::nullopt /* options */, |
| base::nullopt /* p256dh */, base::nullopt /* auth */)); |
| } |
| |
| // Helper methods on both IO and UI threads, merged from |
| // PushMessagingManager and Core. |
| // ----------------------------------------------------------------------------- |
| |
| void PushMessagingManager::Core::GetSubscriptionInfoOnUI( |
| const GURL& origin, |
| int64_t service_worker_registration_id, |
| const std::string& sender_id, |
| const std::string& push_subscription_id, |
| PushMessagingService::SubscriptionInfoCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| PushMessagingService* push_service = service(); |
| if (!push_service) { |
| std::move(callback).Run(false /* is_valid */, |
| std::vector<uint8_t>() /* p256dh */, |
| std::vector<uint8_t>() /* auth */); |
| return; |
| } |
| |
| push_service->GetSubscriptionInfo(origin, service_worker_registration_id, |
| sender_id, push_subscription_id, |
| std::move(callback)); |
| } |
| |
| GURL PushMessagingManager::CreateEndpoint( |
| bool standard_protocol, |
| const std::string& subscription_id) const { |
| const GURL& base = |
| standard_protocol ? web_push_protocol_endpoint_ : default_endpoint_; |
| |
| return GURL(base.spec() + subscription_id); |
| } |
| |
| PushMessagingService* PushMessagingManager::Core::service() { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| RenderProcessHost* process_host = |
| RenderProcessHost::FromID(render_process_id_); |
| return process_host |
| ? process_host->GetBrowserContext()->GetPushMessagingService() |
| : nullptr; |
| } |
| |
| base::WeakPtr<PushMessagingManager::Core> |
| PushMessagingManager::Core::GetWeakPtrFromIOParentConstructor() { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| return weak_factory_ui_to_ui_.GetWeakPtr(); |
| } |
| |
| } // namespace content |
