| // 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_context_wrapper.h" |
| |
| #include <map> |
| #include <set> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "base/barrier_closure.h" |
| #include "base/bind.h" |
| #include "base/bind_helpers.h" |
| #include "base/feature_list.h" |
| #include "base/files/file_path.h" |
| #include "base/guid.h" |
| #include "base/lazy_instance.h" |
| #include "base/location.h" |
| #include "base/logging.h" |
| #include "base/single_thread_task_runner.h" |
| #include "base/task/post_task.h" |
| #include "base/threading/thread_task_runner_handle.h" |
| #include "content/browser/blob_storage/chrome_blob_storage_context.h" |
| #include "content/browser/service_worker/embedded_worker_status.h" |
| #include "content/browser/service_worker/service_worker_process_manager.h" |
| #include "content/browser/service_worker/service_worker_quota_client.h" |
| #include "content/browser/service_worker/service_worker_version.h" |
| #include "content/browser/storage_partition_impl.h" |
| #include "content/common/service_worker/service_worker_utils.h" |
| #include "content/public/browser/browser_context.h" |
| #include "content/public/browser/browser_task_traits.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/service_worker_context_observer.h" |
| #include "content/public/browser/storage_usage_info.h" |
| #include "content/public/common/content_features.h" |
| #include "net/base/url_util.h" |
| #include "storage/browser/quota/quota_manager_proxy.h" |
| #include "storage/browser/quota/special_storage_policy.h" |
| #include "third_party/blink/public/common/service_worker/service_worker_status_code.h" |
| #include "third_party/blink/public/mojom/service_worker/service_worker_registration.mojom.h" |
| #include "url/gurl.h" |
| |
| namespace content { |
| |
| namespace { |
| |
| // Value used to set the timeout when starting a long running ServiceWorker. See |
| // ServiceWorkerContextWrapper::StartServiceWorkerAndDispatchLongRunningMessage. |
| const int kActiveWorkerTimeoutDays = 999; |
| |
| void WorkerStarted(ServiceWorkerContextWrapper::StatusCallback callback, |
| blink::ServiceWorkerStatusCode status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), status)); |
| } |
| |
| void StartActiveWorkerOnIO( |
| ServiceWorkerContextWrapper::StatusCallback callback, |
| blink::ServiceWorkerStatusCode status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (status == blink::ServiceWorkerStatusCode::kOk) { |
| // Pass the reference of |registration| to WorkerStarted callback to prevent |
| // it from being deleted while starting the worker. If the refcount of |
| // |registration| is 1, it will be deleted after WorkerStarted is called. |
| registration->active_version()->StartWorker( |
| ServiceWorkerMetrics::EventType::UNKNOWN, |
| base::BindOnce(WorkerStarted, std::move(callback))); |
| return; |
| } |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorNotFound)); |
| } |
| |
| void SkipWaitingWorkerOnIO( |
| blink::ServiceWorkerStatusCode status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (status != blink::ServiceWorkerStatusCode::kOk || |
| !registration->waiting_version()) |
| return; |
| |
| registration->waiting_version()->set_skip_waiting(true); |
| registration->ActivateWaitingVersionWhenReady(); |
| } |
| |
| void DidStartWorker(scoped_refptr<ServiceWorkerVersion> version, |
| ServiceWorkerContext::StartWorkerCallback info_callback, |
| base::OnceClosure error_callback, |
| blink::ServiceWorkerStatusCode start_worker_status) { |
| if (start_worker_status != blink::ServiceWorkerStatusCode::kOk) { |
| std::move(error_callback).Run(); |
| return; |
| } |
| EmbeddedWorkerInstance* instance = version->embedded_worker(); |
| std::move(info_callback) |
| .Run(version->version_id(), instance->process_id(), |
| instance->thread_id()); |
| } |
| |
| void FoundRegistrationForStartWorker( |
| ServiceWorkerContext::StartWorkerCallback info_callback, |
| base::OnceClosure failure_callback, |
| blink::ServiceWorkerStatusCode service_worker_status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (service_worker_status != blink::ServiceWorkerStatusCode::kOk) { |
| std::move(failure_callback).Run(); |
| return; |
| } |
| |
| ServiceWorkerVersion* version_ptr = registration->active_version() |
| ? registration->active_version() |
| : registration->installing_version(); |
| // Since FindRegistrationForScope returned |
| // blink::ServiceWorkerStatusCode::kOk, there must be either: - |
| // an active version, which optionally might have activated from a waiting |
| // version (as DidFindRegistrationForFindImpl will activate any waiting |
| // version). |
| // - or an installing version. |
| DCHECK(version_ptr); |
| |
| // Note: There might be a remote possibility that |registration|'s |version| |
| // might change between here and DidStartWorker, so bind |version| to |
| // RunAfterStartWorker. |
| scoped_refptr<ServiceWorkerVersion> version = |
| base::WrapRefCounted(version_ptr); |
| version->RunAfterStartWorker( |
| ServiceWorkerMetrics::EventType::EXTERNAL_REQUEST, |
| base::BindOnce(&DidStartWorker, version, std::move(info_callback), |
| std::move(failure_callback))); |
| } |
| |
| void StatusCodeToBoolCallbackAdapter( |
| ServiceWorkerContext::ResultCallback callback, |
| blink::ServiceWorkerStatusCode code) { |
| std::move(callback).Run(code == blink::ServiceWorkerStatusCode::kOk); |
| } |
| |
| void FinishRegistrationOnIO(ServiceWorkerContext::ResultCallback callback, |
| blink::ServiceWorkerStatusCode status, |
| const std::string& status_message, |
| int64_t registration_id) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), |
| status == blink::ServiceWorkerStatusCode::kOk)); |
| } |
| |
| void FinishUnregistrationOnIO(ServiceWorkerContext::ResultCallback callback, |
| blink::ServiceWorkerStatusCode status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), |
| status == blink::ServiceWorkerStatusCode::kOk)); |
| } |
| |
| void MessageFinishedSending(ServiceWorkerContext::ResultCallback callback, |
| blink::ServiceWorkerStatusCode status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| std::move(callback).Run(status == blink::ServiceWorkerStatusCode::kOk); |
| } |
| |
| void RunOnceClosure(scoped_refptr<ServiceWorkerContextWrapper> ref_holder, |
| base::OnceClosure task) { |
| std::move(task).Run(); |
| } |
| |
| } // namespace |
| |
| // static |
| bool ServiceWorkerContext::ScopeMatches(const GURL& scope, const GURL& url) { |
| return ServiceWorkerUtils::ScopeMatches(scope, url); |
| } |
| |
| // static |
| void ServiceWorkerContext::RunTask( |
| scoped_refptr<base::SequencedTaskRunner> task_runner, |
| const base::Location& from_here, |
| ServiceWorkerContext* service_worker_context, |
| base::OnceClosure task) { |
| auto ref = base::WrapRefCounted( |
| static_cast<ServiceWorkerContextWrapper*>(service_worker_context)); |
| task_runner->PostTask( |
| from_here, |
| base::BindOnce(&RunOnceClosure, std::move(ref), std::move(task))); |
| } |
| |
| ServiceWorkerContextWrapper::ServiceWorkerContextWrapper( |
| BrowserContext* browser_context) |
| : core_observer_list_( |
| base::MakeRefCounted<ServiceWorkerContextObserverList>()), |
| process_manager_( |
| std::make_unique<ServiceWorkerProcessManager>(browser_context)) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| |
| // Add this object as an observer of the wrapped |context_core_|. This lets us |
| // forward observer methods to observers outside of content. |
| core_observer_list_->AddObserver(this); |
| } |
| |
| void ServiceWorkerContextWrapper::Init( |
| const base::FilePath& user_data_directory, |
| storage::QuotaManagerProxy* quota_manager_proxy, |
| storage::SpecialStoragePolicy* special_storage_policy, |
| ChromeBlobStorageContext* blob_context, |
| URLLoaderFactoryGetter* loader_factory_getter) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| |
| is_incognito_ = user_data_directory.empty(); |
| // The database task runner is BLOCK_SHUTDOWN in order to support |
| // ClearSessionOnlyOrigins() (called due to the "clear on browser exit" |
| // content setting). |
| // TODO(falken): Only block shutdown for that particular task, when someday |
| // task runners support mixing task shutdown behaviors. |
| scoped_refptr<base::SequencedTaskRunner> database_task_runner = |
| base::CreateSequencedTaskRunnerWithTraits( |
| {base::MayBlock(), base::TaskShutdownBehavior::BLOCK_SHUTDOWN}); |
| InitInternal(user_data_directory, std::move(database_task_runner), |
| quota_manager_proxy, special_storage_policy, blob_context, |
| loader_factory_getter); |
| } |
| |
| void ServiceWorkerContextWrapper::Shutdown() { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| |
| storage_partition_ = nullptr; |
| process_manager_->Shutdown(); |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::ShutdownOnIO, this)); |
| } |
| |
| void ServiceWorkerContextWrapper::InitializeResourceContext( |
| ResourceContext* resource_context) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| resource_context_ = resource_context; |
| } |
| |
| void ServiceWorkerContextWrapper::DeleteAndStartOver() { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| // The context could be null due to system shutdown or restart failure. In |
| // either case, we should not have to recover the system, so just return |
| // here. |
| return; |
| } |
| context_core_->DeleteAndStartOver(base::BindOnce( |
| &ServiceWorkerContextWrapper::DidDeleteAndStartOver, this)); |
| } |
| |
| StoragePartitionImpl* ServiceWorkerContextWrapper::storage_partition() const { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| return storage_partition_; |
| } |
| |
| void ServiceWorkerContextWrapper::set_storage_partition( |
| StoragePartitionImpl* storage_partition) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| storage_partition_ = storage_partition; |
| process_manager_->set_storage_partition(storage_partition_); |
| } |
| |
| ResourceContext* ServiceWorkerContextWrapper::resource_context() { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| return resource_context_; |
| } |
| |
| void ServiceWorkerContextWrapper::OnRegistrationCompleted( |
| int64_t registration_id, |
| const GURL& scope) { |
| for (auto& observer : observer_list_) |
| observer.OnRegistrationCompleted(scope); |
| } |
| |
| void ServiceWorkerContextWrapper::OnRegistrationStored(int64_t registration_id, |
| const GURL& scope) { |
| for (auto& observer : observer_list_) |
| observer.OnRegistrationStored(registration_id, scope); |
| } |
| |
| void ServiceWorkerContextWrapper::OnReportConsoleMessage( |
| int64_t version_id, |
| const ConsoleMessage& message) { |
| for (auto& observer : observer_list_) |
| observer.OnReportConsoleMessage(version_id, message); |
| } |
| |
| void ServiceWorkerContextWrapper::OnNoControllees(int64_t version_id, |
| const GURL& scope) { |
| for (auto& observer : observer_list_) |
| observer.OnNoControllees(version_id, scope); |
| } |
| |
| void ServiceWorkerContextWrapper::OnRunningStateChanged( |
| int64_t version_id, |
| EmbeddedWorkerStatus running_status) { |
| // When |running_status| is RUNNING/STOPPING, the service |
| // worker is doing some actual tasks, so we consider the service worker is |
| // in a running status. |
| // |
| // Although the service work does have some tasks being processed in the stage |
| // of STARTING, some resources of it may have not been allocated yet, |
| // e.g. process id. If STARTING is considered as running, then the observer of |
| // ServiceWorkerContextWrapper may get the unexpected information of this |
| // service worker after being notified due to the absence of the resoureces. |
| // |
| // TODO(minggang): Create a new observer to listen to the events when the |
| // process of the service worker is allocated/released, instead of using the |
| // running status of the embedded worker. |
| if (running_status == EmbeddedWorkerStatus::RUNNING) { |
| running_service_workers_.emplace(version_id); |
| for (auto& observer : observer_list_) { |
| observer.OnVersionRunningStatusChanged(this, version_id, |
| true /* is_running */); |
| } |
| return; |
| } |
| |
| if (running_service_workers_.find(version_id) != |
| running_service_workers_.end() && |
| running_status == EmbeddedWorkerStatus::STOPPED) { |
| running_service_workers_.erase(version_id); |
| for (auto& observer : observer_list_) { |
| observer.OnVersionRunningStatusChanged(this, version_id, |
| false /* is_running */); |
| } |
| } |
| } |
| |
| void ServiceWorkerContextWrapper::OnVersionStateChanged( |
| int64_t version_id, |
| const GURL& scope, |
| ServiceWorkerVersion::Status status) { |
| if (status == ServiceWorkerVersion::Status::ACTIVATED) { |
| for (auto& observer : observer_list_) |
| observer.OnVersionActivated(version_id, scope); |
| } else if (status == ServiceWorkerVersion::Status::REDUNDANT) { |
| for (auto& observer : observer_list_) |
| observer.OnVersionRedundant(version_id, scope); |
| } |
| } |
| |
| void ServiceWorkerContextWrapper::AddObserver( |
| ServiceWorkerContextObserver* observer) { |
| observer_list_.AddObserver(observer); |
| } |
| |
| void ServiceWorkerContextWrapper::RemoveObserver( |
| ServiceWorkerContextObserver* observer) { |
| observer_list_.RemoveObserver(observer); |
| } |
| |
| void ServiceWorkerContextWrapper::RegisterServiceWorker( |
| const GURL& script_url, |
| const blink::mojom::ServiceWorkerRegistrationOptions& options, |
| ResultCallback callback) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::RegisterServiceWorker, |
| this, script_url, options, std::move(callback))); |
| return; |
| } |
| if (!context_core_) { |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), false)); |
| return; |
| } |
| blink::mojom::ServiceWorkerRegistrationOptions options_to_pass( |
| net::SimplifyUrlForRequest(options.scope), options.type, |
| options.update_via_cache); |
| context()->RegisterServiceWorker( |
| net::SimplifyUrlForRequest(script_url), options_to_pass, |
| base::BindOnce(&FinishRegistrationOnIO, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::UnregisterServiceWorker( |
| const GURL& scope, |
| ResultCallback callback) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::UnregisterServiceWorker, |
| this, scope, std::move(callback))); |
| return; |
| } |
| if (!context_core_) { |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), false)); |
| return; |
| } |
| |
| context()->UnregisterServiceWorker( |
| net::SimplifyUrlForRequest(scope), |
| base::BindOnce(&FinishUnregistrationOnIO, std::move(callback))); |
| } |
| |
| bool ServiceWorkerContextWrapper::StartingExternalRequest( |
| int64_t service_worker_version_id, |
| const std::string& request_uuid) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context()) |
| return false; |
| ServiceWorkerVersion* version = |
| context()->GetLiveVersion(service_worker_version_id); |
| if (!version) |
| return false; |
| return version->StartExternalRequest(request_uuid); |
| } |
| |
| bool ServiceWorkerContextWrapper::FinishedExternalRequest( |
| int64_t service_worker_version_id, |
| const std::string& request_uuid) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context()) |
| return false; |
| ServiceWorkerVersion* version = |
| context()->GetLiveVersion(service_worker_version_id); |
| if (!version) |
| return false; |
| return version->FinishExternalRequest(request_uuid); |
| } |
| |
| void ServiceWorkerContextWrapper::CountExternalRequestsForTest( |
| const GURL& origin, |
| CountExternalRequestsCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::CountExternalRequests, this, |
| origin, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::GetAllOriginsInfo( |
| GetUsageInfoCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(std::move(callback), std::vector<StorageUsageInfo>())); |
| return; |
| } |
| context()->storage()->GetAllRegistrationsInfos(base::BindOnce( |
| &ServiceWorkerContextWrapper::DidGetAllRegistrationsForGetAllOrigins, |
| this, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::DeleteForOrigin(const GURL& origin, |
| ResultCallback callback) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::DeleteForOrigin, this, |
| origin, std::move(callback))); |
| return; |
| } |
| if (!context_core_) { |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(std::move(callback), false)); |
| return; |
| } |
| context()->DeleteForOrigin( |
| origin.GetOrigin(), |
| base::BindOnce(&StatusCodeToBoolCallbackAdapter, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::PerformStorageCleanup( |
| base::OnceClosure callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| std::move(callback).Run(); |
| return; |
| } |
| context()->PerformStorageCleanup(std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::CheckHasServiceWorker( |
| const GURL& url, |
| CheckHasServiceWorkerCallback callback) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::CheckHasServiceWorker, |
| this, url, std::move(callback))); |
| return; |
| } |
| if (!context_core_) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), |
| ServiceWorkerCapability::NO_SERVICE_WORKER)); |
| return; |
| } |
| context()->CheckHasServiceWorker( |
| net::SimplifyUrlForRequest(url), |
| base::BindOnce(&ServiceWorkerContextWrapper::DidCheckHasServiceWorker, |
| this, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::ClearAllServiceWorkersForTest( |
| base::OnceClosure callback) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::ClearAllServiceWorkersForTest, this, |
| std::move(callback))); |
| return; |
| } |
| if (!context_core_) { |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| std::move(callback)); |
| return; |
| } |
| context_core_->ClearAllServiceWorkersForTest(std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::StartWorkerForScope( |
| const GURL& scope, |
| StartWorkerCallback info_callback, |
| base::OnceClosure failure_callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| FindRegistrationForScope( |
| scope, |
| base::BindOnce(&FoundRegistrationForStartWorker, std::move(info_callback), |
| std::move(failure_callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::StartServiceWorkerAndDispatchMessage( |
| const GURL& scope, |
| blink::TransferableMessage message, |
| ResultCallback result_callback) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| |
| if (!context_core_) { |
| std::move(result_callback).Run(false); |
| return; |
| } |
| |
| FindReadyRegistrationForScope( |
| net::SimplifyUrlForRequest(scope), |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidFindRegistrationForMessageDispatch, |
| this, std::move(message), scope, std::move(result_callback), |
| false /* is_long_running_message */)); |
| } |
| |
| void ServiceWorkerContextWrapper:: |
| StartServiceWorkerAndDispatchLongRunningMessage( |
| const GURL& scope, |
| blink::TransferableMessage message, |
| ResultCallback result_callback) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| |
| if (!base::FeatureList::IsEnabled( |
| features::kServiceWorkerLongRunningMessage)) { |
| std::move(result_callback).Run(false); |
| return; |
| } |
| |
| if (!context_core_) { |
| std::move(result_callback).Run(false); |
| return; |
| } |
| |
| FindReadyRegistrationForScope( |
| net::SimplifyUrlForRequest(scope), |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidFindRegistrationForMessageDispatch, |
| this, std::move(message), scope, std::move(result_callback), |
| true /* is_long_running_message */)); |
| } |
| |
| void ServiceWorkerContextWrapper::DidFindRegistrationForMessageDispatch( |
| blink::TransferableMessage message, |
| const GURL& source_origin, |
| ResultCallback result_callback, |
| bool is_long_running_message, |
| blink::ServiceWorkerStatusCode service_worker_status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| if (service_worker_status != blink::ServiceWorkerStatusCode::kOk) { |
| LOG(WARNING) << "No registration available, status: " |
| << static_cast<int>(service_worker_status); |
| std::move(result_callback).Run(false); |
| return; |
| } |
| registration->active_version()->StartWorker( |
| is_long_running_message |
| ? ServiceWorkerMetrics::EventType::LONG_RUNNING_MESSAGE |
| : ServiceWorkerMetrics::EventType::MESSAGE, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidStartServiceWorkerForMessageDispatch, |
| this, std::move(message), source_origin, registration, |
| std::move(result_callback), is_long_running_message)); |
| } |
| |
| void ServiceWorkerContextWrapper::DidStartServiceWorkerForMessageDispatch( |
| blink::TransferableMessage message, |
| const GURL& source_origin, |
| scoped_refptr<ServiceWorkerRegistration> registration, |
| ResultCallback result_callback, |
| bool is_long_running_message, |
| blink::ServiceWorkerStatusCode status) { |
| if (status != blink::ServiceWorkerStatusCode::kOk) { |
| std::move(result_callback).Run(false); |
| return; |
| } |
| |
| scoped_refptr<ServiceWorkerVersion> version = registration->active_version(); |
| |
| blink::mojom::ExtendableMessageEventPtr event = |
| blink::mojom::ExtendableMessageEvent::New(); |
| event->message = std::move(message); |
| event->source_origin = url::Origin::Create(source_origin); |
| event->source_info_for_service_worker = |
| version->provider_host() |
| ->GetOrCreateServiceWorkerObjectHost(version) |
| ->CreateCompleteObjectInfoToSend(); |
| |
| if (is_long_running_message) { |
| int request_id = version->StartRequestWithCustomTimeout( |
| ServiceWorkerMetrics::EventType::LONG_RUNNING_MESSAGE, |
| base::BindOnce(&MessageFinishedSending, std::move(result_callback)), |
| base::TimeDelta::FromDays(kActiveWorkerTimeoutDays), |
| ServiceWorkerVersion::CONTINUE_ON_TIMEOUT); |
| version->endpoint()->DispatchExtendableMessageEventWithCustomTimeout( |
| std::move(event), base::TimeDelta::FromDays(kActiveWorkerTimeoutDays), |
| version->CreateSimpleEventCallback(request_id)); |
| } else { |
| int request_id = version->StartRequest( |
| ServiceWorkerMetrics::EventType::MESSAGE, |
| base::BindOnce(&MessageFinishedSending, std::move(result_callback))); |
| version->endpoint()->DispatchExtendableMessageEvent( |
| std::move(event), version->CreateSimpleEventCallback(request_id)); |
| } |
| } |
| |
| void ServiceWorkerContextWrapper::StartServiceWorkerForNavigationHint( |
| const GURL& document_url, |
| StartServiceWorkerForNavigationHintCallback callback) { |
| TRACE_EVENT1("ServiceWorker", "StartServiceWorkerForNavigationHint", |
| "document_url", document_url.spec()); |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::StartServiceWorkerForNavigationHintOnIO, |
| this, document_url, |
| base::BindOnce(&ServiceWorkerContextWrapper:: |
| RecordStartServiceWorkerForNavigationHintResult, |
| this, std::move(callback)))); |
| } |
| |
| void ServiceWorkerContextWrapper::StopAllServiceWorkersForOrigin( |
| const GURL& origin) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::StopAllServiceWorkersForOrigin, this, |
| origin)); |
| return; |
| } |
| if (!context_core_.get()) { |
| return; |
| } |
| std::vector<ServiceWorkerVersionInfo> live_versions = GetAllLiveVersionInfo(); |
| for (const ServiceWorkerVersionInfo& info : live_versions) { |
| ServiceWorkerVersion* version = GetLiveVersion(info.version_id); |
| if (version && version->scope().GetOrigin() == origin) |
| version->StopWorker(base::DoNothing()); |
| } |
| } |
| |
| void ServiceWorkerContextWrapper::StopAllServiceWorkers( |
| base::OnceClosure callback) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::StopAllServiceWorkersOnIO, |
| this, std::move(callback), |
| base::ThreadTaskRunnerHandle::Get())); |
| return; |
| } |
| StopAllServiceWorkersOnIO(std::move(callback), |
| base::ThreadTaskRunnerHandle::Get()); |
| } |
| |
| void ServiceWorkerContextWrapper::GetAllServiceWorkerRunningInfos( |
| GetAllServiceWorkerRunningInfosCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::GetAllServiceWorkerRunningInfosOnIO, |
| this, std::move(callback), base::ThreadTaskRunnerHandle::Get())); |
| } |
| |
| void ServiceWorkerContextWrapper::GetServiceWorkerRunningInfo( |
| int64_t version_id, |
| GetServiceWorkerRunningInfoCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::GetServiceWorkerRunningInfoOnIO, this, |
| version_id, std::move(callback), |
| base::ThreadTaskRunnerHandle::Get())); |
| } |
| |
| ServiceWorkerRegistration* ServiceWorkerContextWrapper::GetLiveRegistration( |
| int64_t registration_id) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) |
| return nullptr; |
| return context_core_->GetLiveRegistration(registration_id); |
| } |
| |
| ServiceWorkerVersion* ServiceWorkerContextWrapper::GetLiveVersion( |
| int64_t version_id) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) |
| return nullptr; |
| return context_core_->GetLiveVersion(version_id); |
| } |
| |
| std::vector<ServiceWorkerRegistrationInfo> |
| ServiceWorkerContextWrapper::GetAllLiveRegistrationInfo() { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) |
| return std::vector<ServiceWorkerRegistrationInfo>(); |
| return context_core_->GetAllLiveRegistrationInfo(); |
| } |
| |
| std::vector<ServiceWorkerVersionInfo> |
| ServiceWorkerContextWrapper::GetAllLiveVersionInfo() { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) |
| return std::vector<ServiceWorkerVersionInfo>(); |
| return context_core_->GetAllLiveVersionInfo(); |
| } |
| |
| void ServiceWorkerContextWrapper::HasMainFrameProviderHost( |
| const GURL& origin, |
| BoolCallback callback) const { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), false)); |
| return; |
| } |
| context_core_->HasMainFrameProviderHost(origin, std::move(callback)); |
| } |
| |
| std::unique_ptr<std::vector<GlobalFrameRoutingId>> |
| ServiceWorkerContextWrapper::GetProviderHostIds(const GURL& origin) const { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| std::unique_ptr<std::vector<GlobalFrameRoutingId>> provider_host_ids( |
| new std::vector<GlobalFrameRoutingId>()); |
| if (!context_core_) |
| return provider_host_ids; |
| |
| for (std::unique_ptr<ServiceWorkerContextCore::ProviderHostIterator> it = |
| context_core_->GetClientProviderHostIterator( |
| origin, false /* include_reserved_clients */); |
| !it->IsAtEnd(); it->Advance()) { |
| ServiceWorkerProviderHost* provider_host = it->GetProviderHost(); |
| provider_host_ids->push_back(GlobalFrameRoutingId( |
| provider_host->process_id(), provider_host->frame_id())); |
| } |
| |
| return provider_host_ids; |
| } |
| |
| void ServiceWorkerContextWrapper::FindReadyRegistrationForDocument( |
| const GURL& document_url, |
| FindRegistrationCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| // FindRegistrationForDocument() can run the callback synchronously. |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kErrorAbort, |
| nullptr); |
| return; |
| } |
| context_core_->storage()->FindRegistrationForDocument( |
| net::SimplifyUrlForRequest(document_url), |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidFindRegistrationForFindReady, this, |
| std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::FindReadyRegistrationForScope( |
| const GURL& scope, |
| FindRegistrationCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| FindRegistrationForScopeImpl(scope, false /* include_installing_version */, |
| std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::FindRegistrationForScope( |
| const GURL& scope, |
| FindRegistrationCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| FindRegistrationForScopeImpl(scope, true /* include_installing_version */, |
| std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::FindReadyRegistrationForId( |
| int64_t registration_id, |
| const GURL& origin, |
| FindRegistrationCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| // FindRegistrationForId() can run the callback synchronously. |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kErrorAbort, |
| nullptr); |
| return; |
| } |
| context_core_->storage()->FindRegistrationForId( |
| registration_id, origin.GetOrigin(), |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidFindRegistrationForFindReady, this, |
| std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::FindReadyRegistrationForIdOnly( |
| int64_t registration_id, |
| FindRegistrationCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| // FindRegistrationForIdOnly() can run the callback synchronously. |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kErrorAbort, |
| nullptr); |
| return; |
| } |
| context_core_->storage()->FindRegistrationForIdOnly( |
| registration_id, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidFindRegistrationForFindReady, this, |
| std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::GetAllRegistrations( |
| GetRegistrationsInfosCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorAbort, |
| std::vector<ServiceWorkerRegistrationInfo>())); |
| return; |
| } |
| context_core_->storage()->GetAllRegistrationsInfos(std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::GetRegistrationsForOrigin( |
| const url::Origin& origin, |
| GetRegistrationsCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::BindOnce( |
| std::move(callback), blink::ServiceWorkerStatusCode::kErrorAbort, |
| std::vector<scoped_refptr<ServiceWorkerRegistration>>())); |
| return; |
| } |
| context_core_->storage()->GetRegistrationsForOrigin(origin.GetURL(), |
| std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::GetRegistrationUserData( |
| int64_t registration_id, |
| const std::vector<std::string>& keys, |
| GetUserDataCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::BindOnce(std::move(callback), std::vector<std::string>(), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->GetUserData(registration_id, keys, |
| std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::GetRegistrationUserDataByKeyPrefix( |
| int64_t registration_id, |
| const std::string& key_prefix, |
| GetUserDataCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::BindOnce(std::move(callback), std::vector<std::string>(), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->GetUserDataByKeyPrefix(registration_id, key_prefix, |
| std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::GetRegistrationUserKeysAndDataByKeyPrefix( |
| int64_t registration_id, |
| const std::string& key_prefix, |
| GetUserKeysAndDataCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), |
| base::flat_map<std::string, std::string>(), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->GetUserKeysAndDataByKeyPrefix( |
| registration_id, key_prefix, std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::StoreRegistrationUserData( |
| int64_t registration_id, |
| const GURL& origin, |
| const std::vector<std::pair<std::string, std::string>>& key_value_pairs, |
| StatusCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->StoreUserData(registration_id, origin.GetOrigin(), |
| key_value_pairs, std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::ClearRegistrationUserData( |
| int64_t registration_id, |
| const std::vector<std::string>& keys, |
| StatusCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->ClearUserData(registration_id, keys, |
| std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::ClearRegistrationUserDataByKeyPrefixes( |
| int64_t registration_id, |
| const std::vector<std::string>& key_prefixes, |
| StatusCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->ClearUserDataByKeyPrefixes( |
| registration_id, key_prefixes, std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::GetUserDataForAllRegistrations( |
| const std::string& key, |
| GetUserDataForAllRegistrationsCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::BindOnce(std::move(callback), |
| std::vector<std::pair<int64_t, std::string>>(), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->GetUserDataForAllRegistrations(key, |
| std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::GetUserDataForAllRegistrationsByKeyPrefix( |
| const std::string& key_prefix, |
| GetUserDataForAllRegistrationsCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::BindOnce(std::move(callback), |
| std::vector<std::pair<int64_t, std::string>>(), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->GetUserDataForAllRegistrationsByKeyPrefix( |
| key_prefix, std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::ClearUserDataForAllRegistrationsByKeyPrefix( |
| const std::string& key_prefix, |
| StatusCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->ClearUserDataForAllRegistrationsByKeyPrefix( |
| key_prefix, std::move(callback)); |
| } |
| |
| void ServiceWorkerContextWrapper::StartServiceWorker(const GURL& scope, |
| StatusCallback callback) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::StartServiceWorker, this, |
| scope, std::move(callback))); |
| return; |
| } |
| if (!context_core_) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorAbort)); |
| return; |
| } |
| context_core_->storage()->FindRegistrationForScope( |
| net::SimplifyUrlForRequest(scope), |
| base::BindOnce(&StartActiveWorkerOnIO, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::SkipWaitingWorker(const GURL& scope) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::SkipWaitingWorker, this, |
| scope)); |
| return; |
| } |
| if (!context_core_) |
| return; |
| context_core_->storage()->FindRegistrationForScope( |
| net::SimplifyUrlForRequest(scope), |
| base::BindOnce(&SkipWaitingWorkerOnIO)); |
| } |
| |
| void ServiceWorkerContextWrapper::UpdateRegistration(const GURL& scope) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::UpdateRegistration, this, |
| scope)); |
| return; |
| } |
| if (!context_core_) |
| return; |
| context_core_->storage()->FindRegistrationForScope( |
| net::SimplifyUrlForRequest(scope), |
| base::BindOnce(&ServiceWorkerContextWrapper::DidFindRegistrationForUpdate, |
| this)); |
| } |
| |
| void ServiceWorkerContextWrapper::SetForceUpdateOnPageLoad( |
| bool force_update_on_page_load) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::SetForceUpdateOnPageLoad, |
| this, force_update_on_page_load)); |
| return; |
| } |
| if (!context_core_) |
| return; |
| context_core_->set_force_update_on_page_load(force_update_on_page_load); |
| } |
| |
| void ServiceWorkerContextWrapper::AddObserver( |
| ServiceWorkerContextCoreObserver* observer) { |
| core_observer_list_->AddObserver(observer); |
| } |
| |
| void ServiceWorkerContextWrapper::RemoveObserver( |
| ServiceWorkerContextCoreObserver* observer) { |
| core_observer_list_->RemoveObserver(observer); |
| } |
| |
| base::WeakPtr<ServiceWorkerProviderHost> |
| ServiceWorkerContextWrapper::PreCreateHostForSharedWorker( |
| int process_id, |
| blink::mojom::ServiceWorkerProviderInfoForWorkerPtr* out_provider_info) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| if (!context_core_) |
| return nullptr; |
| return ServiceWorkerProviderHost::PreCreateForSharedWorker( |
| context_core_->AsWeakPtr(), process_id, out_provider_info); |
| } |
| |
| ServiceWorkerContextWrapper::~ServiceWorkerContextWrapper() { |
| for (auto& observer : observer_list_) |
| observer.OnDestruct(static_cast<ServiceWorkerContext*>(this)); |
| |
| // Explicitly remove this object as an observer to avoid use-after-frees in |
| // tests where this object is not guaranteed to outlive the |
| // ServiceWorkerContextCore it wraps. |
| core_observer_list_->RemoveObserver(this); |
| DCHECK(!resource_context_); |
| } |
| |
| void ServiceWorkerContextWrapper::InitInternal( |
| const base::FilePath& user_data_directory, |
| scoped_refptr<base::SequencedTaskRunner> database_task_runner, |
| storage::QuotaManagerProxy* quota_manager_proxy, |
| storage::SpecialStoragePolicy* special_storage_policy, |
| ChromeBlobStorageContext* blob_context, |
| URLLoaderFactoryGetter* loader_factory_getter) { |
| if (!BrowserThread::CurrentlyOn(BrowserThread::IO)) { |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ServiceWorkerContextWrapper::InitInternal, this, |
| user_data_directory, std::move(database_task_runner), |
| base::RetainedRef(quota_manager_proxy), |
| base::RetainedRef(special_storage_policy), |
| base::RetainedRef(blob_context), |
| base::RetainedRef(loader_factory_getter))); |
| return; |
| } |
| DCHECK(!context_core_); |
| if (quota_manager_proxy) { |
| quota_manager_proxy->RegisterClient(new ServiceWorkerQuotaClient(this)); |
| } |
| |
| context_core_ = std::make_unique<ServiceWorkerContextCore>( |
| user_data_directory, std::move(database_task_runner), quota_manager_proxy, |
| special_storage_policy, loader_factory_getter, core_observer_list_.get(), |
| this); |
| } |
| |
| void ServiceWorkerContextWrapper::FindRegistrationForScopeImpl( |
| const GURL& scope, |
| bool include_installing_version, |
| FindRegistrationCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::BindOnce(std::move(callback), |
| blink::ServiceWorkerStatusCode::kErrorAbort, nullptr)); |
| return; |
| } |
| context_core_->storage()->FindRegistrationForScope( |
| net::SimplifyUrlForRequest(scope), |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidFindRegistrationForFindImpl, this, |
| include_installing_version, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::ShutdownOnIO() { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| resource_context_ = nullptr; |
| context_core_.reset(); |
| } |
| |
| void ServiceWorkerContextWrapper::DidFindRegistrationForFindImpl( |
| bool include_installing_version, |
| FindRegistrationCallback callback, |
| blink::ServiceWorkerStatusCode status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (status != blink::ServiceWorkerStatusCode::kOk) { |
| std::move(callback).Run(status, nullptr); |
| return; |
| } |
| |
| // Attempt to activate the waiting version because the registration retrieved |
| // from the disk might have only the waiting version. |
| if (registration->waiting_version()) |
| registration->ActivateWaitingVersionWhenReady(); |
| |
| scoped_refptr<ServiceWorkerVersion> active_version = |
| registration->active_version(); |
| if (active_version) { |
| if (active_version->status() == ServiceWorkerVersion::ACTIVATING) { |
| // Wait until the version is activated. |
| active_version->RegisterStatusChangeCallback(base::BindOnce( |
| &ServiceWorkerContextWrapper::OnStatusChangedForFindReadyRegistration, |
| this, std::move(callback), std::move(registration))); |
| return; |
| } |
| DCHECK_EQ(ServiceWorkerVersion::ACTIVATED, active_version->status()); |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kOk, |
| std::move(registration)); |
| return; |
| } |
| |
| if (include_installing_version && registration->installing_version()) { |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kOk, |
| std::move(registration)); |
| return; |
| } |
| |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kErrorNotFound, |
| nullptr); |
| } |
| |
| void ServiceWorkerContextWrapper::DidFindRegistrationForFindReady( |
| FindRegistrationCallback callback, |
| blink::ServiceWorkerStatusCode status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| DidFindRegistrationForFindImpl(false /* include_installing_version */, |
| std::move(callback), status, |
| std::move(registration)); |
| } |
| |
| void ServiceWorkerContextWrapper::OnStatusChangedForFindReadyRegistration( |
| FindRegistrationCallback callback, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| scoped_refptr<ServiceWorkerVersion> active_version = |
| registration->active_version(); |
| if (!active_version || |
| active_version->status() != ServiceWorkerVersion::ACTIVATED) { |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kErrorNotFound, |
| nullptr); |
| return; |
| } |
| std::move(callback).Run(blink::ServiceWorkerStatusCode::kOk, registration); |
| } |
| |
| void ServiceWorkerContextWrapper::DidDeleteAndStartOver( |
| blink::ServiceWorkerStatusCode status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (status != blink::ServiceWorkerStatusCode::kOk) { |
| context_core_.reset(); |
| return; |
| } |
| context_core_.reset(new ServiceWorkerContextCore(context_core_.get(), this)); |
| DVLOG(1) << "Restarted ServiceWorkerContextCore successfully."; |
| context_core_->OnStorageWiped(); |
| } |
| |
| void ServiceWorkerContextWrapper::DidGetAllRegistrationsForGetAllOrigins( |
| GetUsageInfoCallback callback, |
| blink::ServiceWorkerStatusCode status, |
| const std::vector<ServiceWorkerRegistrationInfo>& registrations) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| std::vector<StorageUsageInfo> usage_infos; |
| |
| std::map<GURL, StorageUsageInfo> origins; |
| for (const auto& registration_info : registrations) { |
| GURL origin = registration_info.scope.GetOrigin(); |
| |
| auto it = origins.find(origin); |
| if (it == origins.end()) { |
| origins[origin] = StorageUsageInfo( |
| url::Origin::Create(origin), |
| registration_info.stored_version_size_bytes, base::Time()); |
| } else { |
| it->second.total_size_bytes += |
| registration_info.stored_version_size_bytes; |
| } |
| } |
| |
| for (const auto& origin_info_pair : origins) { |
| usage_infos.push_back(origin_info_pair.second); |
| } |
| std::move(callback).Run(usage_infos); |
| } |
| |
| void ServiceWorkerContextWrapper::DidCheckHasServiceWorker( |
| CheckHasServiceWorkerCallback callback, |
| ServiceWorkerCapability capability) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), capability)); |
| } |
| |
| void ServiceWorkerContextWrapper::DidFindRegistrationForUpdate( |
| blink::ServiceWorkerStatusCode status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| if (status != blink::ServiceWorkerStatusCode::kOk) |
| return; |
| if (!context_core_) |
| return; |
| DCHECK(registration); |
| // TODO(jungkees): |force_bypass_cache| is set to true because the call stack |
| // is initiated by an update button on DevTools that expects the cache is |
| // bypassed. However, in order to provide options for callers to choose the |
| // cache bypass mode, plumb |force_bypass_cache| through to |
| // UpdateRegistration(). |
| context_core_->UpdateServiceWorker(registration.get(), |
| true /* force_bypass_cache */); |
| } |
| |
| void ServiceWorkerContextWrapper::CountExternalRequests( |
| const GURL& origin, |
| CountExternalRequestsCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| std::vector<ServiceWorkerVersionInfo> live_version_info = |
| GetAllLiveVersionInfo(); |
| size_t pending_external_request_count = 0; |
| for (const ServiceWorkerVersionInfo& info : live_version_info) { |
| ServiceWorkerVersion* version = GetLiveVersion(info.version_id); |
| if (version && version->scope().GetOrigin() == origin) { |
| pending_external_request_count = |
| version->GetExternalRequestCountForTest(); |
| break; |
| } |
| } |
| |
| base::PostTaskWithTraits( |
| FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), pending_external_request_count)); |
| } |
| |
| void ServiceWorkerContextWrapper::StartServiceWorkerForNavigationHintOnIO( |
| const GURL& document_url, |
| StartServiceWorkerForNavigationHintCallback callback) { |
| TRACE_EVENT1("ServiceWorker", "StartServiceWorkerForNavigationHintOnIO", |
| "document_url", document_url.spec()); |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_) { |
| std::move(callback).Run(StartServiceWorkerForNavigationHintResult::FAILED); |
| return; |
| } |
| context_core_->storage()->FindRegistrationForDocument( |
| net::SimplifyUrlForRequest(document_url), |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidFindRegistrationForNavigationHint, |
| this, std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::DidFindRegistrationForNavigationHint( |
| StartServiceWorkerForNavigationHintCallback callback, |
| blink::ServiceWorkerStatusCode status, |
| scoped_refptr<ServiceWorkerRegistration> registration) { |
| TRACE_EVENT1("ServiceWorker", "DidFindRegistrationForNavigationHint", |
| "status", blink::ServiceWorkerStatusToString(status)); |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!registration) { |
| DCHECK_NE(status, blink::ServiceWorkerStatusCode::kOk); |
| std::move(callback).Run(StartServiceWorkerForNavigationHintResult:: |
| NO_SERVICE_WORKER_REGISTRATION); |
| return; |
| } |
| if (!registration->active_version()) { |
| std::move(callback).Run(StartServiceWorkerForNavigationHintResult:: |
| NO_ACTIVE_SERVICE_WORKER_VERSION); |
| return; |
| } |
| if (registration->active_version()->fetch_handler_existence() == |
| ServiceWorkerVersion::FetchHandlerExistence::DOES_NOT_EXIST) { |
| std::move(callback).Run( |
| StartServiceWorkerForNavigationHintResult::NO_FETCH_HANDLER); |
| return; |
| } |
| if (registration->active_version()->running_status() == |
| EmbeddedWorkerStatus::RUNNING) { |
| std::move(callback).Run( |
| StartServiceWorkerForNavigationHintResult::ALREADY_RUNNING); |
| return; |
| } |
| |
| registration->active_version()->StartWorker( |
| ServiceWorkerMetrics::EventType::NAVIGATION_HINT, |
| base::BindOnce( |
| &ServiceWorkerContextWrapper::DidStartServiceWorkerForNavigationHint, |
| this, registration->scope(), std::move(callback))); |
| } |
| |
| void ServiceWorkerContextWrapper::DidStartServiceWorkerForNavigationHint( |
| const GURL& scope, |
| StartServiceWorkerForNavigationHintCallback callback, |
| blink::ServiceWorkerStatusCode code) { |
| TRACE_EVENT2("ServiceWorker", "DidStartServiceWorkerForNavigationHint", "url", |
| scope.spec(), "code", blink::ServiceWorkerStatusToString(code)); |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| std::move(callback).Run( |
| code == blink::ServiceWorkerStatusCode::kOk |
| ? StartServiceWorkerForNavigationHintResult::STARTED |
| : StartServiceWorkerForNavigationHintResult::FAILED); |
| } |
| |
| void ServiceWorkerContextWrapper:: |
| RecordStartServiceWorkerForNavigationHintResult( |
| StartServiceWorkerForNavigationHintCallback callback, |
| StartServiceWorkerForNavigationHintResult result) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| ServiceWorkerMetrics::RecordStartServiceWorkerForNavigationHintResult(result); |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), result)); |
| } |
| |
| void ServiceWorkerContextWrapper::StopAllServiceWorkersOnIO( |
| base::OnceClosure callback, |
| scoped_refptr<base::SingleThreadTaskRunner> task_runner_for_callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| if (!context_core_.get()) { |
| task_runner_for_callback->PostTask(FROM_HERE, std::move(callback)); |
| return; |
| } |
| std::vector<ServiceWorkerVersionInfo> live_versions = GetAllLiveVersionInfo(); |
| base::RepeatingClosure barrier = base::BarrierClosure( |
| live_versions.size(), |
| base::BindOnce( |
| base::IgnoreResult(&base::SingleThreadTaskRunner::PostTask), |
| std::move(task_runner_for_callback), FROM_HERE, std::move(callback))); |
| for (const ServiceWorkerVersionInfo& info : live_versions) { |
| ServiceWorkerVersion* version = GetLiveVersion(info.version_id); |
| DCHECK(version); |
| version->StopWorker(base::BindOnce(barrier)); |
| } |
| } |
| |
| ServiceWorkerContextCore* ServiceWorkerContextWrapper::context() { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| return context_core_.get(); |
| } |
| |
| void ServiceWorkerContextWrapper::GetAllServiceWorkerRunningInfosOnIO( |
| GetAllServiceWorkerRunningInfosCallback callback, |
| scoped_refptr<base::SingleThreadTaskRunner> task_runner_for_callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| std::vector<ServiceWorkerVersionInfo> live_versions = GetAllLiveVersionInfo(); |
| std::vector<ServiceWorkerRunningInfo> live_versions_running_info; |
| for (const auto& version_info : live_versions) { |
| if (IsRunningStatus(version_info.running_status)) { |
| live_versions_running_info.emplace_back( |
| ExtractServiceWorkerRunningInfoFromVersionInfo(version_info)); |
| } |
| } |
| |
| task_runner_for_callback->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), |
| static_cast<ServiceWorkerContext*>(this), |
| std::move(live_versions_running_info))); |
| } |
| |
| void ServiceWorkerContextWrapper::GetServiceWorkerRunningInfoOnIO( |
| int64_t version_id, |
| GetServiceWorkerRunningInfoCallback callback, |
| scoped_refptr<base::SingleThreadTaskRunner> task_runner_for_callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| ServiceWorkerVersion* version = GetLiveVersion(version_id); |
| if (version && IsRunningStatus(version->running_status())) { |
| ServiceWorkerVersionInfo info = version->GetInfo(); |
| task_runner_for_callback->PostTask( |
| FROM_HERE, |
| base::BindOnce(std::move(callback), |
| static_cast<ServiceWorkerContext*>(this), |
| ExtractServiceWorkerRunningInfoFromVersionInfo(info))); |
| } else { |
| task_runner_for_callback->PostTask( |
| FROM_HERE, base::BindOnce(std::move(callback), |
| static_cast<ServiceWorkerContext*>(this), |
| ServiceWorkerRunningInfo())); |
| } |
| } |
| |
| ServiceWorkerRunningInfo |
| ServiceWorkerContextWrapper::ExtractServiceWorkerRunningInfoFromVersionInfo( |
| const ServiceWorkerVersionInfo& version_info) { |
| return ServiceWorkerRunningInfo(version_info.script_url, |
| version_info.version_id, |
| version_info.process_id); |
| } |
| |
| bool ServiceWorkerContextWrapper::IsRunningStatus(EmbeddedWorkerStatus status) { |
| return status == EmbeddedWorkerStatus::RUNNING || |
| status == EmbeddedWorkerStatus::STOPPING; |
| } |
| |
| } // namespace content |
