| // 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/cache_storage/cache_storage_cache.h" |
| |
| #include <stddef.h> |
| #include <algorithm> |
| #include <limits> |
| #include <memory> |
| #include <string> |
| #include <utility> |
| |
| #include "base/barrier_closure.h" |
| #include "base/bind_helpers.h" |
| #include "base/files/file_path.h" |
| #include "base/guid.h" |
| #include "base/macros.h" |
| #include "base/memory/ptr_util.h" |
| #include "base/metrics/histogram_macros.h" |
| #include "base/strings/string_split.h" |
| #include "base/strings/string_util.h" |
| #include "base/threading/thread_task_runner_handle.h" |
| #include "content/browser/cache_storage/cache_storage.pb.h" |
| #include "content/browser/cache_storage/cache_storage_blob_to_disk_cache.h" |
| #include "content/browser/cache_storage/cache_storage_cache_handle.h" |
| #include "content/browser/cache_storage/cache_storage_cache_observer.h" |
| #include "content/browser/cache_storage/cache_storage_scheduler.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/common/referrer.h" |
| #include "net/base/completion_callback.h" |
| #include "net/base/io_buffer.h" |
| #include "net/base/net_errors.h" |
| #include "net/disk_cache/disk_cache.h" |
| #include "net/url_request/url_request_context_getter.h" |
| #include "storage/browser/blob/blob_data_builder.h" |
| #include "storage/browser/blob/blob_data_handle.h" |
| #include "storage/browser/blob/blob_storage_context.h" |
| #include "storage/browser/blob/blob_url_request_job_factory.h" |
| #include "storage/browser/quota/quota_manager_proxy.h" |
| #include "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorkerResponseType.h" |
| |
| namespace content { |
| |
| namespace { |
| |
| const size_t kMaxQueryCacheResultBytes = |
| 1024 * 1024 * 10; // 10MB query cache limit |
| |
| // This class ensures that the cache and the entry have a lifetime as long as |
| // the blob that is created to contain them. |
| class CacheStorageCacheDataHandle |
| : public storage::BlobDataBuilder::DataHandle { |
| public: |
| CacheStorageCacheDataHandle( |
| std::unique_ptr<CacheStorageCacheHandle> cache_handle, |
| disk_cache::ScopedEntryPtr entry) |
| : cache_handle_(std::move(cache_handle)), entry_(std::move(entry)) {} |
| |
| private: |
| ~CacheStorageCacheDataHandle() override {} |
| |
| std::unique_ptr<CacheStorageCacheHandle> cache_handle_; |
| disk_cache::ScopedEntryPtr entry_; |
| |
| DISALLOW_COPY_AND_ASSIGN(CacheStorageCacheDataHandle); |
| }; |
| |
| typedef base::Callback<void(std::unique_ptr<proto::CacheMetadata>)> |
| MetadataCallback; |
| |
| // The maximum size of each cache. Ultimately, cache size |
| // is controlled per-origin by the QuotaManager. |
| const int kMaxCacheBytes = std::numeric_limits<int>::max(); |
| |
| blink::WebServiceWorkerResponseType ProtoResponseTypeToWebResponseType( |
| proto::CacheResponse::ResponseType response_type) { |
| switch (response_type) { |
| case proto::CacheResponse::BASIC_TYPE: |
| return blink::WebServiceWorkerResponseTypeBasic; |
| case proto::CacheResponse::CORS_TYPE: |
| return blink::WebServiceWorkerResponseTypeCORS; |
| case proto::CacheResponse::DEFAULT_TYPE: |
| return blink::WebServiceWorkerResponseTypeDefault; |
| case proto::CacheResponse::ERROR_TYPE: |
| return blink::WebServiceWorkerResponseTypeError; |
| case proto::CacheResponse::OPAQUE_TYPE: |
| return blink::WebServiceWorkerResponseTypeOpaque; |
| case proto::CacheResponse::OPAQUE_REDIRECT_TYPE: |
| return blink::WebServiceWorkerResponseTypeOpaqueRedirect; |
| } |
| NOTREACHED(); |
| return blink::WebServiceWorkerResponseTypeOpaque; |
| } |
| |
| proto::CacheResponse::ResponseType WebResponseTypeToProtoResponseType( |
| blink::WebServiceWorkerResponseType response_type) { |
| switch (response_type) { |
| case blink::WebServiceWorkerResponseTypeBasic: |
| return proto::CacheResponse::BASIC_TYPE; |
| case blink::WebServiceWorkerResponseTypeCORS: |
| return proto::CacheResponse::CORS_TYPE; |
| case blink::WebServiceWorkerResponseTypeDefault: |
| return proto::CacheResponse::DEFAULT_TYPE; |
| case blink::WebServiceWorkerResponseTypeError: |
| return proto::CacheResponse::ERROR_TYPE; |
| case blink::WebServiceWorkerResponseTypeOpaque: |
| return proto::CacheResponse::OPAQUE_TYPE; |
| case blink::WebServiceWorkerResponseTypeOpaqueRedirect: |
| return proto::CacheResponse::OPAQUE_REDIRECT_TYPE; |
| } |
| NOTREACHED(); |
| return proto::CacheResponse::OPAQUE_TYPE; |
| } |
| |
| // Copy headers out of a cache entry and into a protobuf. The callback is |
| // guaranteed to be run. |
| void ReadMetadata(disk_cache::Entry* entry, const MetadataCallback& callback); |
| void ReadMetadataDidReadMetadata(disk_cache::Entry* entry, |
| const MetadataCallback& callback, |
| scoped_refptr<net::IOBufferWithSize> buffer, |
| int rv); |
| |
| bool VaryMatches(const ServiceWorkerHeaderMap& request, |
| const ServiceWorkerHeaderMap& cached_request, |
| const ServiceWorkerHeaderMap& response) { |
| ServiceWorkerHeaderMap::const_iterator vary_iter = response.find("vary"); |
| if (vary_iter == response.end()) |
| return true; |
| |
| for (const std::string& trimmed : |
| base::SplitString(vary_iter->second, ",", |
| base::TRIM_WHITESPACE, base::SPLIT_WANT_NONEMPTY)) { |
| if (trimmed == "*") |
| return false; |
| |
| ServiceWorkerHeaderMap::const_iterator request_iter = request.find(trimmed); |
| ServiceWorkerHeaderMap::const_iterator cached_request_iter = |
| cached_request.find(trimmed); |
| |
| // If the header exists in one but not the other, no match. |
| if ((request_iter == request.end()) != |
| (cached_request_iter == cached_request.end())) |
| return false; |
| |
| // If the header exists in one, it exists in both. Verify that the values |
| // are equal. |
| if (request_iter != request.end() && |
| request_iter->second != cached_request_iter->second) |
| return false; |
| } |
| |
| return true; |
| } |
| |
| GURL RemoveQueryParam(const GURL& url) { |
| url::Replacements<char> replacements; |
| replacements.ClearQuery(); |
| return url.ReplaceComponents(replacements); |
| } |
| |
| void ReadMetadata(disk_cache::Entry* entry, const MetadataCallback& callback) { |
| DCHECK(entry); |
| |
| scoped_refptr<net::IOBufferWithSize> buffer(new net::IOBufferWithSize( |
| entry->GetDataSize(CacheStorageCache::INDEX_HEADERS))); |
| |
| net::CompletionCallback read_header_callback = |
| base::Bind(ReadMetadataDidReadMetadata, entry, callback, buffer); |
| |
| int read_rv = |
| entry->ReadData(CacheStorageCache::INDEX_HEADERS, 0, buffer.get(), |
| buffer->size(), read_header_callback); |
| |
| if (read_rv != net::ERR_IO_PENDING) |
| read_header_callback.Run(read_rv); |
| } |
| |
| void ReadMetadataDidReadMetadata(disk_cache::Entry* entry, |
| const MetadataCallback& callback, |
| scoped_refptr<net::IOBufferWithSize> buffer, |
| int rv) { |
| if (rv != buffer->size()) { |
| callback.Run(std::unique_ptr<proto::CacheMetadata>()); |
| return; |
| } |
| |
| std::unique_ptr<proto::CacheMetadata> metadata(new proto::CacheMetadata()); |
| |
| if (!metadata->ParseFromArray(buffer->data(), buffer->size())) { |
| callback.Run(std::unique_ptr<proto::CacheMetadata>()); |
| return; |
| } |
| |
| callback.Run(std::move(metadata)); |
| } |
| |
| std::unique_ptr<ServiceWorkerFetchRequest> CreateRequest( |
| const proto::CacheMetadata& metadata, |
| const GURL& request_url) { |
| auto request = base::MakeUnique<ServiceWorkerFetchRequest>( |
| request_url, metadata.request().method(), ServiceWorkerHeaderMap(), |
| Referrer(), false); |
| |
| for (int i = 0; i < metadata.request().headers_size(); ++i) { |
| const proto::CacheHeaderMap header = metadata.request().headers(i); |
| DCHECK_EQ(std::string::npos, header.name().find('\0')); |
| DCHECK_EQ(std::string::npos, header.value().find('\0')); |
| request->headers.insert(std::make_pair(header.name(), header.value())); |
| } |
| return request; |
| } |
| |
| std::unique_ptr<ServiceWorkerResponse> CreateResponse( |
| const proto::CacheMetadata& metadata, |
| const std::string& cache_name) { |
| std::unique_ptr<std::vector<GURL>> url_list = |
| base::MakeUnique<std::vector<GURL>>(); |
| // From Chrome 57, proto::CacheMetadata's url field was deprecated. |
| UMA_HISTOGRAM_BOOLEAN("ServiceWorkerCache.Response.HasDeprecatedURL", |
| metadata.response().has_url()); |
| if (metadata.response().has_url()) { |
| url_list->push_back(GURL(metadata.response().url())); |
| } else { |
| url_list->reserve(metadata.response().url_list_size()); |
| for (int i = 0; i < metadata.response().url_list_size(); ++i) |
| url_list->push_back(GURL(metadata.response().url_list(i))); |
| } |
| |
| std::unique_ptr<ServiceWorkerHeaderMap> headers = |
| base::MakeUnique<ServiceWorkerHeaderMap>(); |
| for (int i = 0; i < metadata.response().headers_size(); ++i) { |
| const proto::CacheHeaderMap header = metadata.response().headers(i); |
| DCHECK_EQ(std::string::npos, header.name().find('\0')); |
| DCHECK_EQ(std::string::npos, header.value().find('\0')); |
| headers->insert(std::make_pair(header.name(), header.value())); |
| } |
| |
| return base::MakeUnique<ServiceWorkerResponse>( |
| std::move(url_list), metadata.response().status_code(), |
| metadata.response().status_text(), |
| ProtoResponseTypeToWebResponseType(metadata.response().response_type()), |
| std::move(headers), "", 0, GURL(), |
| blink::WebServiceWorkerResponseErrorUnknown, |
| base::Time::FromInternalValue(metadata.response().response_time()), |
| true /* is_in_cache_storage */, cache_name, |
| base::MakeUnique<ServiceWorkerHeaderList>( |
| metadata.response().cors_exposed_header_names().begin(), |
| metadata.response().cors_exposed_header_names().end())); |
| } |
| |
| } // namespace |
| |
| // The state needed to pass between CacheStorageCache::Put callbacks. |
| struct CacheStorageCache::PutContext { |
| PutContext(std::unique_ptr<ServiceWorkerFetchRequest> request, |
| std::unique_ptr<ServiceWorkerResponse> response, |
| std::unique_ptr<storage::BlobDataHandle> blob_data_handle, |
| const CacheStorageCache::ErrorCallback& callback) |
| : request(std::move(request)), |
| response(std::move(response)), |
| blob_data_handle(std::move(blob_data_handle)), |
| callback(callback) {} |
| |
| // Input parameters to the Put function. |
| std::unique_ptr<ServiceWorkerFetchRequest> request; |
| std::unique_ptr<ServiceWorkerResponse> response; |
| std::unique_ptr<storage::BlobDataHandle> blob_data_handle; |
| CacheStorageCache::ErrorCallback callback; |
| disk_cache::ScopedEntryPtr cache_entry; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(PutContext); |
| }; |
| |
| struct CacheStorageCache::QueryCacheResult { |
| explicit QueryCacheResult(base::Time entry_time) : entry_time(entry_time) {} |
| |
| std::unique_ptr<ServiceWorkerFetchRequest> request; |
| std::unique_ptr<ServiceWorkerResponse> response; |
| std::unique_ptr<storage::BlobDataHandle> blob_handle; |
| disk_cache::ScopedEntryPtr entry; |
| base::Time entry_time; |
| }; |
| |
| struct CacheStorageCache::QueryCacheContext { |
| QueryCacheContext(std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& options, |
| const QueryCacheCallback& callback) |
| : request(std::move(request)), |
| options(options), |
| callback(callback), |
| matches(base::MakeUnique<QueryCacheResults>()) {} |
| |
| ~QueryCacheContext() { |
| // If the CacheStorageCache is deleted before a backend operation to open |
| // an entry completes, the callback won't be run and the resulting entry |
| // will be leaked unless we close it here. |
| if (enumerated_entry) { |
| enumerated_entry->Close(); |
| enumerated_entry = nullptr; |
| } |
| } |
| |
| // Input to QueryCache |
| std::unique_ptr<ServiceWorkerFetchRequest> request; |
| CacheStorageCacheQueryParams options; |
| QueryCacheCallback callback; |
| QueryCacheType query_type; |
| size_t estimated_out_bytes = 0; |
| |
| // Iteration state |
| std::unique_ptr<disk_cache::Backend::Iterator> backend_iterator; |
| disk_cache::Entry* enumerated_entry = nullptr; |
| |
| // Output of QueryCache |
| std::unique_ptr<std::vector<QueryCacheResult>> matches; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(QueryCacheContext); |
| }; |
| |
| // static |
| std::unique_ptr<CacheStorageCache> CacheStorageCache::CreateMemoryCache( |
| const GURL& origin, |
| const std::string& cache_name, |
| CacheStorage* cache_storage, |
| scoped_refptr<net::URLRequestContextGetter> request_context_getter, |
| scoped_refptr<storage::QuotaManagerProxy> quota_manager_proxy, |
| base::WeakPtr<storage::BlobStorageContext> blob_context) { |
| CacheStorageCache* cache = new CacheStorageCache( |
| origin, cache_name, base::FilePath(), cache_storage, |
| std::move(request_context_getter), std::move(quota_manager_proxy), |
| blob_context, 0 /* cache_size */); |
| cache->SetObserver(cache_storage); |
| cache->InitBackend(); |
| return base::WrapUnique(cache); |
| } |
| |
| // static |
| std::unique_ptr<CacheStorageCache> CacheStorageCache::CreatePersistentCache( |
| const GURL& origin, |
| const std::string& cache_name, |
| CacheStorage* cache_storage, |
| const base::FilePath& path, |
| scoped_refptr<net::URLRequestContextGetter> request_context_getter, |
| scoped_refptr<storage::QuotaManagerProxy> quota_manager_proxy, |
| base::WeakPtr<storage::BlobStorageContext> blob_context, |
| int64_t cache_size) { |
| CacheStorageCache* cache = new CacheStorageCache( |
| origin, cache_name, path, cache_storage, |
| std::move(request_context_getter), std::move(quota_manager_proxy), |
| blob_context, cache_size); |
| cache->SetObserver(cache_storage), cache->InitBackend(); |
| return base::WrapUnique(cache); |
| } |
| |
| base::WeakPtr<CacheStorageCache> CacheStorageCache::AsWeakPtr() { |
| return weak_ptr_factory_.GetWeakPtr(); |
| } |
| |
| void CacheStorageCache::Match( |
| std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& match_params, |
| const ResponseCallback& callback) { |
| if (backend_state_ == BACKEND_CLOSED) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE, |
| std::unique_ptr<ServiceWorkerResponse>(), |
| std::unique_ptr<storage::BlobDataHandle>()); |
| return; |
| } |
| |
| scheduler_->ScheduleOperation( |
| base::Bind(&CacheStorageCache::MatchImpl, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(request)), match_params, |
| scheduler_->WrapCallbackToRunNext(callback))); |
| } |
| |
| void CacheStorageCache::MatchAll( |
| std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& match_params, |
| const ResponsesCallback& callback) { |
| if (backend_state_ == BACKEND_CLOSED) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE, std::unique_ptr<Responses>(), |
| std::unique_ptr<BlobDataHandles>()); |
| return; |
| } |
| |
| scheduler_->ScheduleOperation(base::Bind( |
| &CacheStorageCache::MatchAllImpl, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(request)), match_params, |
| scheduler_->WrapCallbackToRunNext(callback))); |
| } |
| |
| void CacheStorageCache::WriteSideData(const ErrorCallback& callback, |
| const GURL& url, |
| base::Time expected_response_time, |
| scoped_refptr<net::IOBuffer> buffer, |
| int buf_len) { |
| if (backend_state_ == BACKEND_CLOSED) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::Bind(callback, CACHE_STORAGE_ERROR_STORAGE)); |
| return; |
| } |
| |
| // GetUsageAndQuota is called before entering a scheduled operation since it |
| // can call Size, another scheduled operation. |
| quota_manager_proxy_->GetUsageAndQuota( |
| base::ThreadTaskRunnerHandle::Get().get(), origin_, |
| storage::kStorageTypeTemporary, |
| base::Bind(&CacheStorageCache::WriteSideDataDidGetQuota, |
| weak_ptr_factory_.GetWeakPtr(), callback, url, |
| expected_response_time, buffer, buf_len)); |
| } |
| |
| void CacheStorageCache::BatchOperation( |
| const std::vector<CacheStorageBatchOperation>& operations, |
| const ErrorCallback& callback) { |
| if (backend_state_ == BACKEND_CLOSED) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::Bind(callback, CACHE_STORAGE_ERROR_STORAGE)); |
| return; |
| } |
| |
| // Estimate the required size of the put operations. The size of the deletes |
| // is unknown and not considered. |
| int64_t space_required = 0; |
| for (const auto& operation : operations) { |
| if (operation.operation_type == CACHE_STORAGE_CACHE_OPERATION_TYPE_PUT) { |
| space_required += |
| operation.request.blob_size + operation.response.blob_size; |
| } |
| } |
| if (space_required > 0) { |
| // GetUsageAndQuota is called before entering a scheduled operation since it |
| // can call Size, another scheduled operation. This is racy. The decision |
| // to commit is made before the scheduled Put operation runs. By the time |
| // Put runs, the cache might already be full and the origin will be larger |
| // than it's supposed to be. |
| quota_manager_proxy_->GetUsageAndQuota( |
| base::ThreadTaskRunnerHandle::Get().get(), origin_, |
| storage::kStorageTypeTemporary, |
| base::Bind(&CacheStorageCache::BatchDidGetUsageAndQuota, |
| weak_ptr_factory_.GetWeakPtr(), operations, callback, |
| space_required)); |
| return; |
| } |
| |
| BatchDidGetUsageAndQuota(operations, callback, 0 /* space_required */, |
| storage::kQuotaStatusOk, 0 /* usage */, |
| 0 /* quota */); |
| } |
| |
| void CacheStorageCache::BatchDidGetUsageAndQuota( |
| const std::vector<CacheStorageBatchOperation>& operations, |
| const ErrorCallback& callback, |
| int64_t space_required, |
| storage::QuotaStatusCode status_code, |
| int64_t usage, |
| int64_t quota) { |
| if (status_code != storage::kQuotaStatusOk || |
| space_required > quota - usage) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::Bind(callback, CACHE_STORAGE_ERROR_QUOTA_EXCEEDED)); |
| return; |
| } |
| |
| std::unique_ptr<ErrorCallback> callback_copy(new ErrorCallback(callback)); |
| ErrorCallback* callback_ptr = callback_copy.get(); |
| base::Closure barrier_closure = base::BarrierClosure( |
| operations.size(), base::Bind(&CacheStorageCache::BatchDidAllOperations, |
| weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(callback_copy)))); |
| ErrorCallback completion_callback = |
| base::Bind(&CacheStorageCache::BatchDidOneOperation, |
| weak_ptr_factory_.GetWeakPtr(), barrier_closure, callback_ptr); |
| |
| for (const auto& operation : operations) { |
| switch (operation.operation_type) { |
| case CACHE_STORAGE_CACHE_OPERATION_TYPE_PUT: |
| Put(operation, completion_callback); |
| break; |
| case CACHE_STORAGE_CACHE_OPERATION_TYPE_DELETE: |
| DCHECK_EQ(1u, operations.size()); |
| Delete(operation, completion_callback); |
| break; |
| case CACHE_STORAGE_CACHE_OPERATION_TYPE_UNDEFINED: |
| NOTREACHED(); |
| // TODO(nhiroki): This should return "TypeError". |
| // http://crbug.com/425505 |
| completion_callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| break; |
| } |
| } |
| } |
| |
| void CacheStorageCache::BatchDidOneOperation( |
| const base::Closure& barrier_closure, |
| ErrorCallback* callback, |
| CacheStorageError error) { |
| if (callback->is_null() || error == CACHE_STORAGE_OK) { |
| barrier_closure.Run(); |
| return; |
| } |
| callback->Run(error); |
| callback->Reset(); // Only call the callback once. |
| |
| barrier_closure.Run(); |
| } |
| |
| void CacheStorageCache::BatchDidAllOperations( |
| std::unique_ptr<ErrorCallback> callback) { |
| if (callback->is_null()) |
| return; |
| callback->Run(CACHE_STORAGE_OK); |
| } |
| |
| void CacheStorageCache::Keys(std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& options, |
| const RequestsCallback& callback) { |
| if (backend_state_ == BACKEND_CLOSED) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE, std::unique_ptr<Requests>()); |
| return; |
| } |
| |
| scheduler_->ScheduleOperation( |
| base::Bind(&CacheStorageCache::KeysImpl, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(request)), options, |
| scheduler_->WrapCallbackToRunNext(callback))); |
| } |
| |
| void CacheStorageCache::Close(const base::Closure& callback) { |
| DCHECK_NE(BACKEND_CLOSED, backend_state_) |
| << "Was CacheStorageCache::Close() called twice?"; |
| |
| scheduler_->ScheduleOperation( |
| base::Bind(&CacheStorageCache::CloseImpl, weak_ptr_factory_.GetWeakPtr(), |
| scheduler_->WrapCallbackToRunNext(callback))); |
| } |
| |
| void CacheStorageCache::Size(const SizeCallback& callback) { |
| if (backend_state_ == BACKEND_CLOSED) { |
| // TODO(jkarlin): Delete caches that can't be initialized. |
| base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| base::Bind(callback, 0)); |
| return; |
| } |
| |
| scheduler_->ScheduleOperation( |
| base::Bind(&CacheStorageCache::SizeImpl, weak_ptr_factory_.GetWeakPtr(), |
| scheduler_->WrapCallbackToRunNext(callback))); |
| } |
| |
| void CacheStorageCache::GetSizeThenClose(const SizeCallback& callback) { |
| if (backend_state_ == BACKEND_CLOSED) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| base::Bind(callback, 0)); |
| return; |
| } |
| |
| scheduler_->ScheduleOperation( |
| base::Bind(&CacheStorageCache::SizeImpl, weak_ptr_factory_.GetWeakPtr(), |
| base::Bind(&CacheStorageCache::GetSizeThenCloseDidGetSize, |
| weak_ptr_factory_.GetWeakPtr(), |
| scheduler_->WrapCallbackToRunNext(callback)))); |
| } |
| |
| void CacheStorageCache::SetObserver(CacheStorageCacheObserver* observer) { |
| DCHECK((observer == nullptr) ^ (cache_observer_ == nullptr)); |
| cache_observer_ = observer; |
| } |
| |
| CacheStorageCache::~CacheStorageCache() { |
| quota_manager_proxy_->NotifyOriginNoLongerInUse(origin_); |
| } |
| |
| CacheStorageCache::CacheStorageCache( |
| const GURL& origin, |
| const std::string& cache_name, |
| const base::FilePath& path, |
| CacheStorage* cache_storage, |
| scoped_refptr<net::URLRequestContextGetter> request_context_getter, |
| scoped_refptr<storage::QuotaManagerProxy> quota_manager_proxy, |
| base::WeakPtr<storage::BlobStorageContext> blob_context, |
| int64_t cache_size) |
| : origin_(origin), |
| cache_name_(cache_name), |
| path_(path), |
| cache_storage_(cache_storage), |
| request_context_getter_(std::move(request_context_getter)), |
| quota_manager_proxy_(std::move(quota_manager_proxy)), |
| blob_storage_context_(blob_context), |
| scheduler_( |
| new CacheStorageScheduler(CacheStorageSchedulerClient::CLIENT_CACHE)), |
| cache_size_(cache_size), |
| max_query_size_bytes_(kMaxQueryCacheResultBytes), |
| cache_observer_(nullptr), |
| memory_only_(path.empty()), |
| weak_ptr_factory_(this) { |
| DCHECK(!origin_.is_empty()); |
| DCHECK(quota_manager_proxy_.get()); |
| |
| quota_manager_proxy_->NotifyOriginInUse(origin_); |
| } |
| |
| void CacheStorageCache::QueryCache( |
| std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& options, |
| QueryCacheType query_type, |
| const QueryCacheCallback& callback) { |
| DCHECK_NE(BACKEND_UNINITIALIZED, backend_state_); |
| if (backend_state_ != BACKEND_OPEN) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE, |
| std::unique_ptr<QueryCacheResults>()); |
| return; |
| } |
| |
| if (!options.ignore_method && request && !request->method.empty() && |
| request->method != "GET") { |
| callback.Run(CACHE_STORAGE_OK, base::MakeUnique<QueryCacheResults>()); |
| return; |
| } |
| |
| ServiceWorkerFetchRequest* request_ptr = request.get(); |
| std::unique_ptr<QueryCacheContext> query_cache_context( |
| new QueryCacheContext(std::move(request), options, callback)); |
| query_cache_context->query_type = query_type; |
| |
| if (query_cache_context->request && |
| !query_cache_context->request->url.is_empty() && !options.ignore_search) { |
| // There is no need to scan the entire backend, just open the exact |
| // URL. |
| disk_cache::Entry** entry_ptr = &query_cache_context->enumerated_entry; |
| net::CompletionCallback open_entry_callback = |
| base::Bind(&CacheStorageCache::QueryCacheDidOpenFastPath, |
| weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(query_cache_context))); |
| int rv = backend_->OpenEntry(request_ptr->url.spec(), entry_ptr, |
| open_entry_callback); |
| if (rv != net::ERR_IO_PENDING) |
| open_entry_callback.Run(rv); |
| return; |
| } |
| |
| query_cache_context->backend_iterator = backend_->CreateIterator(); |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| } |
| |
| void CacheStorageCache::QueryCacheDidOpenFastPath( |
| std::unique_ptr<QueryCacheContext> query_cache_context, |
| int rv) { |
| if (rv != net::OK) { |
| QueryCacheContext* results = query_cache_context.get(); |
| results->callback.Run(CACHE_STORAGE_OK, |
| std::move(query_cache_context->matches)); |
| return; |
| } |
| QueryCacheFilterEntry(std::move(query_cache_context), rv); |
| } |
| |
| void CacheStorageCache::QueryCacheOpenNextEntry( |
| std::unique_ptr<QueryCacheContext> query_cache_context) { |
| DCHECK_EQ(nullptr, query_cache_context->enumerated_entry); |
| |
| if (!query_cache_context->backend_iterator) { |
| // Iteration is complete. |
| std::sort(query_cache_context->matches->begin(), |
| query_cache_context->matches->end(), QueryCacheResultCompare); |
| |
| query_cache_context->callback.Run(CACHE_STORAGE_OK, |
| std::move(query_cache_context->matches)); |
| return; |
| } |
| |
| disk_cache::Backend::Iterator& iterator = |
| *query_cache_context->backend_iterator; |
| disk_cache::Entry** enumerated_entry = &query_cache_context->enumerated_entry; |
| net::CompletionCallback open_entry_callback = base::Bind( |
| &CacheStorageCache::QueryCacheFilterEntry, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(query_cache_context))); |
| |
| int rv = iterator.OpenNextEntry(enumerated_entry, open_entry_callback); |
| |
| if (rv != net::ERR_IO_PENDING) |
| open_entry_callback.Run(rv); |
| } |
| |
| void CacheStorageCache::QueryCacheFilterEntry( |
| std::unique_ptr<QueryCacheContext> query_cache_context, |
| int rv) { |
| if (rv == net::ERR_FAILED) { |
| // This is the indicator that iteration is complete. |
| query_cache_context->backend_iterator.reset(); |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| return; |
| } |
| |
| if (rv < 0) { |
| QueryCacheCallback callback = query_cache_context->callback; |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE, |
| std::move(query_cache_context->matches)); |
| return; |
| } |
| |
| disk_cache::ScopedEntryPtr entry(query_cache_context->enumerated_entry); |
| query_cache_context->enumerated_entry = nullptr; |
| |
| if (backend_state_ != BACKEND_OPEN) { |
| QueryCacheCallback callback = query_cache_context->callback; |
| callback.Run(CACHE_STORAGE_ERROR_NOT_FOUND, |
| std::move(query_cache_context->matches)); |
| return; |
| } |
| |
| if (query_cache_context->request && |
| !query_cache_context->request->url.is_empty()) { |
| GURL requestURL = query_cache_context->request->url; |
| GURL cachedURL = GURL(entry->GetKey()); |
| |
| if (query_cache_context->options.ignore_search) { |
| requestURL = RemoveQueryParam(requestURL); |
| cachedURL = RemoveQueryParam(cachedURL); |
| } |
| |
| if (cachedURL != requestURL) { |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| return; |
| } |
| } |
| |
| disk_cache::Entry* entry_ptr = entry.get(); |
| ReadMetadata(entry_ptr, |
| base::Bind(&CacheStorageCache::QueryCacheDidReadMetadata, |
| weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(query_cache_context)), |
| base::Passed(std::move(entry)))); |
| } |
| |
| void CacheStorageCache::QueryCacheDidReadMetadata( |
| std::unique_ptr<QueryCacheContext> query_cache_context, |
| disk_cache::ScopedEntryPtr entry, |
| std::unique_ptr<proto::CacheMetadata> metadata) { |
| if (!metadata) { |
| entry->Doom(); |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| return; |
| } |
| |
| // If the entry was created before we started adding entry times, then |
| // default to using the Response object's time for sorting purposes. |
| int64_t entry_time = metadata->has_entry_time() |
| ? metadata->entry_time() |
| : metadata->response().response_time(); |
| |
| query_cache_context->matches->push_back( |
| QueryCacheResult(base::Time::FromInternalValue(entry_time))); |
| QueryCacheResult* match = &query_cache_context->matches->back(); |
| match->request = CreateRequest(*metadata, GURL(entry->GetKey())); |
| match->response = CreateResponse(*metadata, cache_name_); |
| |
| if (query_cache_context->request && |
| !query_cache_context->options.ignore_vary && |
| !VaryMatches(query_cache_context->request->headers, |
| match->request->headers, match->response->headers)) { |
| query_cache_context->matches->pop_back(); |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| return; |
| } |
| |
| if (query_cache_context->query_type == QueryCacheType::CACHE_ENTRIES) { |
| match->request.reset(); |
| match->response.reset(); |
| match->entry = std::move(entry); |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| return; |
| } |
| |
| query_cache_context->estimated_out_bytes += |
| match->request->EstimatedStructSize(); |
| if (query_cache_context->estimated_out_bytes > max_query_size_bytes_) { |
| query_cache_context->callback.Run(CACHE_STORAGE_ERROR_QUERY_TOO_LARGE, |
| std::unique_ptr<QueryCacheResults>()); |
| return; |
| } |
| |
| if (query_cache_context->query_type == QueryCacheType::REQUESTS) { |
| match->response.reset(); |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| return; |
| } |
| |
| DCHECK_EQ(QueryCacheType::REQUESTS_AND_RESPONSES, |
| query_cache_context->query_type); |
| |
| query_cache_context->estimated_out_bytes += |
| match->response->EstimatedStructSize(); |
| if (query_cache_context->estimated_out_bytes > max_query_size_bytes_) { |
| query_cache_context->callback.Run(CACHE_STORAGE_ERROR_QUERY_TOO_LARGE, |
| std::unique_ptr<QueryCacheResults>()); |
| return; |
| } |
| |
| if (entry->GetDataSize(INDEX_RESPONSE_BODY) == 0) { |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| return; |
| } |
| |
| if (!blob_storage_context_) { |
| query_cache_context->callback.Run(CACHE_STORAGE_ERROR_STORAGE, |
| base::MakeUnique<QueryCacheResults>()); |
| return; |
| } |
| |
| std::unique_ptr<storage::BlobDataHandle> blob_data_handle = |
| PopulateResponseBody(std::move(entry), match->response.get()); |
| match->blob_handle = std::move(blob_data_handle); |
| |
| QueryCacheOpenNextEntry(std::move(query_cache_context)); |
| } |
| |
| // static |
| bool CacheStorageCache::QueryCacheResultCompare(const QueryCacheResult& lhs, |
| const QueryCacheResult& rhs) { |
| return lhs.entry_time < rhs.entry_time; |
| } |
| |
| void CacheStorageCache::MatchImpl( |
| std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& match_params, |
| const ResponseCallback& callback) { |
| MatchAllImpl(std::move(request), match_params, |
| base::Bind(&CacheStorageCache::MatchDidMatchAll, |
| weak_ptr_factory_.GetWeakPtr(), callback)); |
| } |
| |
| void CacheStorageCache::MatchDidMatchAll( |
| const ResponseCallback& callback, |
| CacheStorageError match_all_error, |
| std::unique_ptr<Responses> match_all_responses, |
| std::unique_ptr<BlobDataHandles> match_all_handles) { |
| if (match_all_error != CACHE_STORAGE_OK) { |
| callback.Run(match_all_error, std::unique_ptr<ServiceWorkerResponse>(), |
| std::unique_ptr<storage::BlobDataHandle>()); |
| return; |
| } |
| |
| if (match_all_responses->empty()) { |
| callback.Run(CACHE_STORAGE_ERROR_NOT_FOUND, |
| std::unique_ptr<ServiceWorkerResponse>(), |
| std::unique_ptr<storage::BlobDataHandle>()); |
| return; |
| } |
| |
| std::unique_ptr<ServiceWorkerResponse> response = |
| base::MakeUnique<ServiceWorkerResponse>(match_all_responses->at(0)); |
| |
| callback.Run(CACHE_STORAGE_OK, std::move(response), |
| std::move(match_all_handles->at(0))); |
| } |
| |
| void CacheStorageCache::MatchAllImpl( |
| std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& options, |
| const ResponsesCallback& callback) { |
| DCHECK_NE(BACKEND_UNINITIALIZED, backend_state_); |
| if (backend_state_ != BACKEND_OPEN) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE, std::unique_ptr<Responses>(), |
| std::unique_ptr<BlobDataHandles>()); |
| return; |
| } |
| |
| QueryCache(std::move(request), options, |
| QueryCacheType::REQUESTS_AND_RESPONSES, |
| base::Bind(&CacheStorageCache::MatchAllDidQueryCache, |
| weak_ptr_factory_.GetWeakPtr(), callback)); |
| } |
| |
| void CacheStorageCache::MatchAllDidQueryCache( |
| const ResponsesCallback& callback, |
| CacheStorageError error, |
| std::unique_ptr<QueryCacheResults> query_cache_results) { |
| if (error != CACHE_STORAGE_OK) { |
| callback.Run(error, std::unique_ptr<Responses>(), |
| std::unique_ptr<BlobDataHandles>()); |
| return; |
| } |
| |
| std::unique_ptr<Responses> out_responses = base::MakeUnique<Responses>(); |
| std::unique_ptr<BlobDataHandles> out_handles = |
| base::MakeUnique<BlobDataHandles>(); |
| out_responses->reserve(query_cache_results->size()); |
| out_handles->reserve(query_cache_results->size()); |
| |
| for (auto& result : *query_cache_results) { |
| out_responses->push_back(*result.response); |
| out_handles->push_back(std::move(result.blob_handle)); |
| } |
| |
| callback.Run(CACHE_STORAGE_OK, std::move(out_responses), |
| std::move(out_handles)); |
| } |
| |
| void CacheStorageCache::WriteSideDataDidGetQuota( |
| const ErrorCallback& callback, |
| const GURL& url, |
| base::Time expected_response_time, |
| scoped_refptr<net::IOBuffer> buffer, |
| int buf_len, |
| storage::QuotaStatusCode status_code, |
| int64_t usage, |
| int64_t quota) { |
| if (status_code != storage::kQuotaStatusOk || (buf_len > quota - usage)) { |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, base::Bind(callback, CACHE_STORAGE_ERROR_QUOTA_EXCEEDED)); |
| return; |
| } |
| |
| scheduler_->ScheduleOperation(base::Bind( |
| &CacheStorageCache::WriteSideDataImpl, weak_ptr_factory_.GetWeakPtr(), |
| scheduler_->WrapCallbackToRunNext(callback), url, expected_response_time, |
| buffer, buf_len)); |
| } |
| |
| void CacheStorageCache::WriteSideDataImpl(const ErrorCallback& callback, |
| const GURL& url, |
| base::Time expected_response_time, |
| scoped_refptr<net::IOBuffer> buffer, |
| int buf_len) { |
| DCHECK_NE(BACKEND_UNINITIALIZED, backend_state_); |
| if (backend_state_ != BACKEND_OPEN) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| std::unique_ptr<disk_cache::Entry*> scoped_entry_ptr( |
| new disk_cache::Entry*()); |
| disk_cache::Entry** entry_ptr = scoped_entry_ptr.get(); |
| net::CompletionCallback open_entry_callback = base::Bind( |
| &CacheStorageCache::WriteSideDataDidOpenEntry, |
| weak_ptr_factory_.GetWeakPtr(), callback, expected_response_time, buffer, |
| buf_len, base::Passed(std::move(scoped_entry_ptr))); |
| |
| int rv = backend_->OpenEntry(url.spec(), entry_ptr, open_entry_callback); |
| if (rv != net::ERR_IO_PENDING) |
| open_entry_callback.Run(rv); |
| } |
| |
| void CacheStorageCache::WriteSideDataDidOpenEntry( |
| const ErrorCallback& callback, |
| base::Time expected_response_time, |
| scoped_refptr<net::IOBuffer> buffer, |
| int buf_len, |
| std::unique_ptr<disk_cache::Entry*> entry_ptr, |
| int rv) { |
| if (rv != net::OK) { |
| callback.Run(CACHE_STORAGE_ERROR_NOT_FOUND); |
| return; |
| } |
| disk_cache::ScopedEntryPtr entry(*entry_ptr); |
| |
| ReadMetadata(*entry_ptr, |
| base::Bind(&CacheStorageCache::WriteSideDataDidReadMetaData, |
| weak_ptr_factory_.GetWeakPtr(), callback, |
| expected_response_time, buffer, buf_len, |
| base::Passed(std::move(entry)))); |
| } |
| |
| void CacheStorageCache::WriteSideDataDidReadMetaData( |
| const ErrorCallback& callback, |
| base::Time expected_response_time, |
| scoped_refptr<net::IOBuffer> buffer, |
| int buf_len, |
| disk_cache::ScopedEntryPtr entry, |
| std::unique_ptr<proto::CacheMetadata> headers) { |
| if (!headers || |
| headers->response().response_time() != |
| expected_response_time.ToInternalValue()) { |
| callback.Run(CACHE_STORAGE_ERROR_NOT_FOUND); |
| return; |
| } |
| // Get a temporary copy of the entry pointer before passing it in base::Bind. |
| disk_cache::Entry* temp_entry_ptr = entry.get(); |
| |
| net::CompletionCallback write_side_data_callback = base::Bind( |
| &CacheStorageCache::WriteSideDataDidWrite, weak_ptr_factory_.GetWeakPtr(), |
| callback, base::Passed(std::move(entry)), buf_len); |
| |
| int rv = temp_entry_ptr->WriteData( |
| INDEX_SIDE_DATA, 0 /* offset */, buffer.get(), buf_len, |
| write_side_data_callback, true /* truncate */); |
| |
| if (rv != net::ERR_IO_PENDING) |
| write_side_data_callback.Run(rv); |
| } |
| |
| void CacheStorageCache::WriteSideDataDidWrite(const ErrorCallback& callback, |
| disk_cache::ScopedEntryPtr entry, |
| int expected_bytes, |
| int rv) { |
| if (rv != expected_bytes) { |
| entry->Doom(); |
| UpdateCacheSize(base::Bind(callback, CACHE_STORAGE_ERROR_NOT_FOUND)); |
| return; |
| } |
| |
| UpdateCacheSize(base::Bind(callback, CACHE_STORAGE_OK)); |
| } |
| |
| void CacheStorageCache::Put(const CacheStorageBatchOperation& operation, |
| const ErrorCallback& callback) { |
| DCHECK(BACKEND_OPEN == backend_state_ || initializing_); |
| DCHECK_EQ(CACHE_STORAGE_CACHE_OPERATION_TYPE_PUT, operation.operation_type); |
| |
| std::unique_ptr<ServiceWorkerFetchRequest> request( |
| new ServiceWorkerFetchRequest( |
| operation.request.url, operation.request.method, |
| operation.request.headers, operation.request.referrer, |
| operation.request.is_reload)); |
| |
| // We don't support streaming for cache. |
| DCHECK(operation.response.stream_url.is_empty()); |
| // We don't support the body of redirect response. |
| DCHECK(!(operation.response.response_type == |
| blink::WebServiceWorkerResponseTypeOpaqueRedirect && |
| operation.response.blob_size)); |
| |
| std::unique_ptr<ServiceWorkerResponse> response = |
| base::MakeUnique<ServiceWorkerResponse>(operation.response); |
| std::unique_ptr<storage::BlobDataHandle> blob_data_handle; |
| |
| if (!response->blob_uuid.empty()) { |
| if (!blob_storage_context_) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| blob_data_handle = |
| blob_storage_context_->GetBlobDataFromUUID(response->blob_uuid); |
| if (!blob_data_handle) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| } |
| |
| UMA_HISTOGRAM_ENUMERATION( |
| "ServiceWorkerCache.Cache.AllWritesResponseType", |
| operation.response.response_type, |
| blink::WebServiceWorkerResponseType::WebServiceWorkerResponseTypeLast + |
| 1); |
| |
| std::unique_ptr<PutContext> put_context(new PutContext( |
| std::move(request), std::move(response), std::move(blob_data_handle), |
| scheduler_->WrapCallbackToRunNext(callback))); |
| |
| scheduler_->ScheduleOperation( |
| base::Bind(&CacheStorageCache::PutImpl, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(put_context)))); |
| } |
| |
| void CacheStorageCache::PutImpl(std::unique_ptr<PutContext> put_context) { |
| DCHECK_NE(BACKEND_UNINITIALIZED, backend_state_); |
| if (backend_state_ != BACKEND_OPEN) { |
| put_context->callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| std::string key = put_context->request->url.spec(); |
| |
| net::CompletionCallback callback = base::Bind( |
| &CacheStorageCache::PutDidDoomEntry, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(put_context))); |
| |
| int rv = backend_->DoomEntry(key, callback); |
| if (rv != net::ERR_IO_PENDING) |
| callback.Run(rv); |
| } |
| |
| void CacheStorageCache::PutDidDoomEntry(std::unique_ptr<PutContext> put_context, |
| int rv) { |
| if (backend_state_ != BACKEND_OPEN) { |
| put_context->callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| // |rv| is ignored as doom entry can fail if the entry doesn't exist. |
| |
| std::unique_ptr<disk_cache::Entry*> scoped_entry_ptr( |
| new disk_cache::Entry*()); |
| disk_cache::Entry** entry_ptr = scoped_entry_ptr.get(); |
| ServiceWorkerFetchRequest* request_ptr = put_context->request.get(); |
| disk_cache::Backend* backend_ptr = backend_.get(); |
| |
| net::CompletionCallback create_entry_callback = base::Bind( |
| &CacheStorageCache::PutDidCreateEntry, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(scoped_entry_ptr)), |
| base::Passed(std::move(put_context))); |
| |
| int create_rv = backend_ptr->CreateEntry(request_ptr->url.spec(), entry_ptr, |
| create_entry_callback); |
| |
| if (create_rv != net::ERR_IO_PENDING) |
| create_entry_callback.Run(create_rv); |
| } |
| |
| void CacheStorageCache::PutDidCreateEntry( |
| std::unique_ptr<disk_cache::Entry*> entry_ptr, |
| std::unique_ptr<PutContext> put_context, |
| int rv) { |
| put_context->cache_entry.reset(*entry_ptr); |
| |
| if (rv != net::OK) { |
| put_context->callback.Run(CACHE_STORAGE_ERROR_EXISTS); |
| return; |
| } |
| |
| proto::CacheMetadata metadata; |
| metadata.set_entry_time(base::Time::Now().ToInternalValue()); |
| proto::CacheRequest* request_metadata = metadata.mutable_request(); |
| request_metadata->set_method(put_context->request->method); |
| for (ServiceWorkerHeaderMap::const_iterator it = |
| put_context->request->headers.begin(); |
| it != put_context->request->headers.end(); ++it) { |
| DCHECK_EQ(std::string::npos, it->first.find('\0')); |
| DCHECK_EQ(std::string::npos, it->second.find('\0')); |
| proto::CacheHeaderMap* header_map = request_metadata->add_headers(); |
| header_map->set_name(it->first); |
| header_map->set_value(it->second); |
| } |
| |
| proto::CacheResponse* response_metadata = metadata.mutable_response(); |
| response_metadata->set_status_code(put_context->response->status_code); |
| response_metadata->set_status_text(put_context->response->status_text); |
| response_metadata->set_response_type( |
| WebResponseTypeToProtoResponseType(put_context->response->response_type)); |
| for (const auto& url : put_context->response->url_list) |
| response_metadata->add_url_list(url.spec()); |
| response_metadata->set_response_time( |
| put_context->response->response_time.ToInternalValue()); |
| for (ServiceWorkerHeaderMap::const_iterator it = |
| put_context->response->headers.begin(); |
| it != put_context->response->headers.end(); ++it) { |
| DCHECK_EQ(std::string::npos, it->first.find('\0')); |
| DCHECK_EQ(std::string::npos, it->second.find('\0')); |
| proto::CacheHeaderMap* header_map = response_metadata->add_headers(); |
| header_map->set_name(it->first); |
| header_map->set_value(it->second); |
| } |
| for (const auto& header : put_context->response->cors_exposed_header_names) |
| response_metadata->add_cors_exposed_header_names(header); |
| |
| std::unique_ptr<std::string> serialized(new std::string()); |
| if (!metadata.SerializeToString(serialized.get())) { |
| put_context->callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| scoped_refptr<net::StringIOBuffer> buffer( |
| new net::StringIOBuffer(std::move(serialized))); |
| |
| // Get a temporary copy of the entry pointer before passing it in base::Bind. |
| disk_cache::Entry* temp_entry_ptr = put_context->cache_entry.get(); |
| |
| net::CompletionCallback write_headers_callback = base::Bind( |
| &CacheStorageCache::PutDidWriteHeaders, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(put_context)), buffer->size()); |
| |
| rv = temp_entry_ptr->WriteData(INDEX_HEADERS, 0 /* offset */, buffer.get(), |
| buffer->size(), write_headers_callback, |
| true /* truncate */); |
| |
| if (rv != net::ERR_IO_PENDING) |
| write_headers_callback.Run(rv); |
| } |
| |
| void CacheStorageCache::PutDidWriteHeaders( |
| std::unique_ptr<PutContext> put_context, |
| int expected_bytes, |
| int rv) { |
| if (rv != expected_bytes) { |
| put_context->cache_entry->Doom(); |
| put_context->callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| // The metadata is written, now for the response content. The data is streamed |
| // from the blob into the cache entry. |
| |
| if (put_context->response->blob_uuid.empty()) { |
| UpdateCacheSize(base::Bind(put_context->callback, CACHE_STORAGE_OK)); |
| return; |
| } |
| |
| DCHECK(put_context->blob_data_handle); |
| |
| disk_cache::ScopedEntryPtr entry(std::move(put_context->cache_entry)); |
| put_context->cache_entry = NULL; |
| |
| auto blob_to_cache = base::MakeUnique<CacheStorageBlobToDiskCache>(); |
| CacheStorageBlobToDiskCache* blob_to_cache_raw = blob_to_cache.get(); |
| BlobToDiskCacheIDMap::KeyType blob_to_cache_key = |
| active_blob_to_disk_cache_writers_.Add(std::move(blob_to_cache)); |
| |
| std::unique_ptr<storage::BlobDataHandle> blob_data_handle = |
| std::move(put_context->blob_data_handle); |
| |
| blob_to_cache_raw->StreamBlobToCache( |
| std::move(entry), INDEX_RESPONSE_BODY, request_context_getter_.get(), |
| std::move(blob_data_handle), |
| base::Bind(&CacheStorageCache::PutDidWriteBlobToCache, |
| weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(put_context)), blob_to_cache_key)); |
| } |
| |
| void CacheStorageCache::PutDidWriteBlobToCache( |
| std::unique_ptr<PutContext> put_context, |
| BlobToDiskCacheIDMap::KeyType blob_to_cache_key, |
| disk_cache::ScopedEntryPtr entry, |
| bool success) { |
| DCHECK(entry); |
| put_context->cache_entry = std::move(entry); |
| |
| active_blob_to_disk_cache_writers_.Remove(blob_to_cache_key); |
| |
| if (!success) { |
| put_context->cache_entry->Doom(); |
| put_context->callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| UpdateCacheSize(base::Bind(put_context->callback, CACHE_STORAGE_OK)); |
| } |
| |
| void CacheStorageCache::UpdateCacheSize(const base::Closure& callback) { |
| if (backend_state_ != BACKEND_OPEN) |
| return; |
| |
| // Note that the callback holds a cache handle to keep the cache alive during |
| // the operation since this UpdateCacheSize is often run after an operation |
| // completes and runs its callback. |
| int rv = backend_->CalculateSizeOfAllEntries( |
| base::Bind(&CacheStorageCache::UpdateCacheSizeGotSize, |
| weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(CreateCacheHandle()), callback)); |
| |
| if (rv != net::ERR_IO_PENDING) |
| UpdateCacheSizeGotSize(CreateCacheHandle(), callback, rv); |
| } |
| |
| void CacheStorageCache::UpdateCacheSizeGotSize( |
| std::unique_ptr<CacheStorageCacheHandle> cache_handle, |
| const base::Closure& callback, |
| int current_cache_size) { |
| DCHECK_NE(current_cache_size, CacheStorage::kSizeUnknown); |
| int64_t old_cache_size = cache_size_; |
| cache_size_ = current_cache_size; |
| |
| int64_t size_delta = current_cache_size - old_cache_size; |
| |
| quota_manager_proxy_->NotifyStorageModified( |
| storage::QuotaClient::kServiceWorkerCache, origin_, |
| storage::kStorageTypeTemporary, size_delta); |
| |
| if (cache_observer_) |
| cache_observer_->CacheSizeUpdated(this, current_cache_size); |
| |
| callback.Run(); |
| } |
| |
| void CacheStorageCache::Delete(const CacheStorageBatchOperation& operation, |
| const ErrorCallback& callback) { |
| DCHECK(BACKEND_OPEN == backend_state_ || initializing_); |
| DCHECK_EQ(CACHE_STORAGE_CACHE_OPERATION_TYPE_DELETE, |
| operation.operation_type); |
| |
| std::unique_ptr<ServiceWorkerFetchRequest> request( |
| new ServiceWorkerFetchRequest( |
| operation.request.url, operation.request.method, |
| operation.request.headers, operation.request.referrer, |
| operation.request.is_reload)); |
| |
| scheduler_->ScheduleOperation( |
| base::Bind(&CacheStorageCache::DeleteImpl, weak_ptr_factory_.GetWeakPtr(), |
| base::Passed(std::move(request)), operation.match_params, |
| scheduler_->WrapCallbackToRunNext(callback))); |
| } |
| |
| void CacheStorageCache::DeleteImpl( |
| std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& match_params, |
| const ErrorCallback& callback) { |
| DCHECK_NE(BACKEND_UNINITIALIZED, backend_state_); |
| if (backend_state_ != BACKEND_OPEN) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| QueryCache(std::move(request), match_params, QueryCacheType::CACHE_ENTRIES, |
| base::Bind(&CacheStorageCache::DeleteDidQueryCache, |
| weak_ptr_factory_.GetWeakPtr(), callback)); |
| } |
| |
| void CacheStorageCache::DeleteDidQueryCache( |
| const ErrorCallback& callback, |
| CacheStorageError error, |
| std::unique_ptr<QueryCacheResults> query_cache_results) { |
| if (error != CACHE_STORAGE_OK) { |
| callback.Run(error); |
| return; |
| } |
| |
| if (query_cache_results->empty()) { |
| callback.Run(CACHE_STORAGE_ERROR_NOT_FOUND); |
| return; |
| } |
| |
| for (auto& result : *query_cache_results) { |
| disk_cache::ScopedEntryPtr entry = std::move(result.entry); |
| entry->Doom(); |
| } |
| |
| UpdateCacheSize(base::Bind(callback, CACHE_STORAGE_OK)); |
| } |
| |
| void CacheStorageCache::KeysImpl( |
| std::unique_ptr<ServiceWorkerFetchRequest> request, |
| const CacheStorageCacheQueryParams& options, |
| const RequestsCallback& callback) { |
| DCHECK_NE(BACKEND_UNINITIALIZED, backend_state_); |
| if (backend_state_ != BACKEND_OPEN) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE, std::unique_ptr<Requests>()); |
| return; |
| } |
| |
| QueryCache(std::move(request), options, QueryCacheType::REQUESTS, |
| base::Bind(&CacheStorageCache::KeysDidQueryCache, |
| weak_ptr_factory_.GetWeakPtr(), callback)); |
| } |
| |
| void CacheStorageCache::KeysDidQueryCache( |
| const RequestsCallback& callback, |
| CacheStorageError error, |
| std::unique_ptr<QueryCacheResults> query_cache_results) { |
| if (error != CACHE_STORAGE_OK) { |
| callback.Run(error, std::unique_ptr<Requests>()); |
| return; |
| } |
| |
| std::unique_ptr<Requests> out_requests = base::MakeUnique<Requests>(); |
| out_requests->reserve(query_cache_results->size()); |
| for (const auto& result : *query_cache_results) |
| out_requests->push_back(*result.request); |
| |
| callback.Run(CACHE_STORAGE_OK, std::move(out_requests)); |
| } |
| |
| void CacheStorageCache::CloseImpl(const base::Closure& callback) { |
| DCHECK_NE(BACKEND_CLOSED, backend_state_); |
| |
| backend_state_ = BACKEND_CLOSED; |
| backend_.reset(); |
| callback.Run(); |
| } |
| |
| void CacheStorageCache::SizeImpl(const SizeCallback& callback) { |
| DCHECK_NE(BACKEND_UNINITIALIZED, backend_state_); |
| |
| int64_t size = backend_state_ == BACKEND_OPEN ? cache_size_ : 0; |
| base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, |
| base::Bind(callback, size)); |
| } |
| |
| void CacheStorageCache::GetSizeThenCloseDidGetSize(const SizeCallback& callback, |
| int64_t cache_size) { |
| CloseImpl(base::Bind(callback, cache_size)); |
| } |
| |
| void CacheStorageCache::CreateBackend(const ErrorCallback& callback) { |
| DCHECK(!backend_); |
| |
| // Use APP_CACHE as opposed to DISK_CACHE to prevent cache eviction. |
| net::CacheType cache_type = memory_only_ ? net::MEMORY_CACHE : net::APP_CACHE; |
| |
| std::unique_ptr<ScopedBackendPtr> backend_ptr(new ScopedBackendPtr()); |
| |
| // Temporary pointer so that backend_ptr can be Pass()'d in Bind below. |
| ScopedBackendPtr* backend = backend_ptr.get(); |
| |
| net::CompletionCallback create_cache_callback = |
| base::Bind(&CacheStorageCache::CreateBackendDidCreate, |
| weak_ptr_factory_.GetWeakPtr(), callback, |
| base::Passed(std::move(backend_ptr))); |
| |
| // TODO(jkarlin): Use the cache task runner that ServiceWorkerCacheCore |
| // has for disk caches. |
| int rv = disk_cache::CreateCacheBackend( |
| cache_type, net::CACHE_BACKEND_SIMPLE, path_, kMaxCacheBytes, |
| false, /* force */ |
| BrowserThread::GetTaskRunnerForThread(BrowserThread::CACHE).get(), NULL, |
| backend, create_cache_callback); |
| if (rv != net::ERR_IO_PENDING) |
| create_cache_callback.Run(rv); |
| } |
| |
| void CacheStorageCache::CreateBackendDidCreate( |
| const CacheStorageCache::ErrorCallback& callback, |
| std::unique_ptr<ScopedBackendPtr> backend_ptr, |
| int rv) { |
| if (rv != net::OK) { |
| callback.Run(CACHE_STORAGE_ERROR_STORAGE); |
| return; |
| } |
| |
| backend_ = std::move(*backend_ptr); |
| callback.Run(CACHE_STORAGE_OK); |
| } |
| |
| void CacheStorageCache::InitBackend() { |
| DCHECK_EQ(BACKEND_UNINITIALIZED, backend_state_); |
| DCHECK(!initializing_); |
| DCHECK(!scheduler_->ScheduledOperations()); |
| initializing_ = true; |
| |
| scheduler_->ScheduleOperation(base::Bind( |
| &CacheStorageCache::CreateBackend, weak_ptr_factory_.GetWeakPtr(), |
| base::Bind( |
| &CacheStorageCache::InitDidCreateBackend, |
| weak_ptr_factory_.GetWeakPtr(), |
| scheduler_->WrapCallbackToRunNext(base::Bind(&base::DoNothing))))); |
| } |
| |
| void CacheStorageCache::InitDidCreateBackend( |
| const base::Closure& callback, |
| CacheStorageError cache_create_error) { |
| if (cache_create_error != CACHE_STORAGE_OK) { |
| InitGotCacheSize(callback, cache_create_error, 0); |
| return; |
| } |
| |
| int rv = backend_->CalculateSizeOfAllEntries( |
| base::Bind(&CacheStorageCache::InitGotCacheSize, |
| weak_ptr_factory_.GetWeakPtr(), callback, cache_create_error)); |
| |
| if (rv != net::ERR_IO_PENDING) |
| InitGotCacheSize(callback, cache_create_error, rv); |
| } |
| |
| void CacheStorageCache::InitGotCacheSize(const base::Closure& callback, |
| CacheStorageError cache_create_error, |
| int cache_size) { |
| // Now that we know the cache size either 1) the cache size should be unknown |
| // (which is why the size was calculated), or 2) it must match the current |
| // size. If the sizes aren't equal then there is a bug in how the cache size |
| // is saved in the store's index. |
| if (cache_size_ != CacheStorage::kSizeUnknown) { |
| LOG_IF(ERROR, cache_size_ != cache_size) |
| << "Cache size: " << cache_size |
| << " does not match size from index: " << cache_size_; |
| UMA_HISTOGRAM_COUNTS_10M("ServiceWorkerCache.IndexSizeDifference", |
| std::abs(cache_size_ - cache_size)); |
| // Disabled for crbug.com/681900. |
| // DCHECK_EQ(cache_size_, cache_size); |
| } |
| cache_size_ = cache_size; |
| initializing_ = false; |
| backend_state_ = (cache_create_error == CACHE_STORAGE_OK && backend_ && |
| backend_state_ == BACKEND_UNINITIALIZED) |
| ? BACKEND_OPEN |
| : BACKEND_CLOSED; |
| |
| UMA_HISTOGRAM_ENUMERATION("ServiceWorkerCache.InitBackendResult", |
| cache_create_error, CACHE_STORAGE_ERROR_LAST + 1); |
| |
| if (cache_observer_) |
| cache_observer_->CacheSizeUpdated(this, cache_size_); |
| |
| callback.Run(); |
| } |
| |
| std::unique_ptr<storage::BlobDataHandle> |
| CacheStorageCache::PopulateResponseBody(disk_cache::ScopedEntryPtr entry, |
| ServiceWorkerResponse* response) { |
| DCHECK(blob_storage_context_); |
| |
| // Create a blob with the response body data. |
| response->blob_size = entry->GetDataSize(INDEX_RESPONSE_BODY); |
| response->blob_uuid = base::GenerateGUID(); |
| storage::BlobDataBuilder blob_data(response->blob_uuid); |
| |
| disk_cache::Entry* temp_entry = entry.get(); |
| blob_data.AppendDiskCacheEntryWithSideData( |
| new CacheStorageCacheDataHandle(CreateCacheHandle(), std::move(entry)), |
| temp_entry, INDEX_RESPONSE_BODY, INDEX_SIDE_DATA); |
| return blob_storage_context_->AddFinishedBlob(&blob_data); |
| } |
| |
| std::unique_ptr<CacheStorageCacheHandle> |
| CacheStorageCache::CreateCacheHandle() { |
| return cache_storage_->CreateCacheHandle(this); |
| } |
| |
| } // namespace content |