Introduce Platform::cacheMetadataInCacheStorage() to store V8 code cache in CacheStorage

third_party/WebKit/Source/core/fetch/Resource.cpp

 /*
     Copyright (C) 1998 Lars Knoll (knoll@mpi-hd.mpg.de)
     Copyright (C) 2001 Dirk Mueller (mueller@kde.org)
     Copyright (C) 2002 Waldo Bastian (bastian@kde.org)
     Copyright (C) 2006 Samuel Weinig (sam.weinig@gmail.com)
     Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011 Apple Inc. All rights reserved.
 
     This library is free software; you can redistribute it and/or
     modify it under the terms of the GNU Library General Public
     License as published by the Free Software Foundation; either
@@ skipping 23 matching lines @@
 #include "core/inspector/InspectorInstrumentation.h"
 #include "core/inspector/InstanceCounters.h"
 #include "platform/Logging.h"
 #include "platform/SharedBuffer.h"
 #include "platform/TraceEvent.h"
 #include "platform/network/HTTPParsers.h"
 #include "platform/weborigin/KURL.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebProcessMemoryDump.h"
 #include "public/platform/WebScheduler.h"
+#include "public/platform/WebSecurityOrigin.h"
 #include "wtf/CurrentTime.h"
 #include "wtf/MathExtras.h"
 #include "wtf/StdLibExtras.h"
 #include "wtf/Vector.h"
 #include "wtf/text/CString.h"
 #include <algorithm>
 
 namespace blink {
 
 // These response headers are not copied from a revalidated response to the
@@ skipping 31 matching lines @@
         if (equalIgnoringCase(header, headersToIgnoreAfterRevalidation[i]))
             return false;
     }
     for (size_t i = 0; i < WTF_ARRAY_LENGTH(headerPrefixesToIgnoreAfterRevalidation); i++) {
         if (header.startsWith(headerPrefixesToIgnoreAfterRevalidation[i], TextCaseInsensitive))
             return false;
     }
     return true;
 }
 
-class Resource::CacheHandler : public CachedMetadataHandler {
+class Resource::CachedMetadataHandlerImpl : public CachedMetadataHandler {
 public:
-    static CacheHandler* create(Resource* resource)
+    static Resource::CachedMetadataHandlerImpl* create(Resource* resource)
     {
-        return new CacheHandler(resource);
+        return new CachedMetadataHandlerImpl(resource);
     }
-    ~CacheHandler() override { }
+    ~CachedMetadataHandlerImpl() override {}
     DECLARE_VIRTUAL_TRACE();
     void setCachedMetadata(unsigned, const char*, size_t, CacheType) override;
     void clearCachedMetadata(CacheType) override;
     CachedMetadata* cachedMetadata(unsigned) const override;
     String encoding() const override;
+    // Sets the serialized metadata retrieved from the platform's cache.
+    void setSerializedCachedMetadata(const char*, size_t);
+
+protected:
+    explicit CachedMetadataHandlerImpl(Resource*);
+    virtual void sendToPlatform();
+    const ResourceResponse& response() const { return m_resource->response(); }
+
+    RefPtr<CachedMetadata> m_cachedMetadata;
 
 private:
-    explicit CacheHandler(Resource*);
     Member<Resource> m_resource;
 };
 
-Resource::CacheHandler::CacheHandler(Resource* resource)
+Resource::CachedMetadataHandlerImpl::CachedMetadataHandlerImpl(Resource* resource)
     : m_resource(resource)
 {
 }
 
-DEFINE_TRACE(Resource::CacheHandler)
+DEFINE_TRACE(Resource::CachedMetadataHandlerImpl)
 {
     visitor->trace(m_resource);
     CachedMetadataHandler::trace(visitor);
 }
 
-void Resource::CacheHandler::setCachedMetadata(unsigned dataTypeID, const char* data, size_t size, CacheType type)
+void Resource::CachedMetadataHandlerImpl::setCachedMetadata(unsigned dataTypeID, const char* data, size_t size, CachedMetadataHandler::CacheType cacheType)
 {
-    m_resource->setCachedMetadata(dataTypeID, data, size, type);
+    // Currently, only one type of cached metadata per resource is supported.
+    // If the need arises for multiple types of metadata per resource this could
+    // be enhanced to store types of metadata in a map.
+    ASSERT(!m_cachedMetadata);
+    m_cachedMetadata = CachedMetadata::create(dataTypeID, data, size);
+    if (cacheType == CachedMetadataHandler::SendToPlatform)
+        sendToPlatform();
 }
 
-void Resource::CacheHandler::clearCachedMetadata(CacheType type)
+void Resource::CachedMetadataHandlerImpl::clearCachedMetadata(CachedMetadataHandler::CacheType cacheType)
 {
-    m_resource->clearCachedMetadata(type);
+    m_cachedMetadata.clear();
+    if (cacheType == CachedMetadataHandler::SendToPlatform)
+        sendToPlatform();
 }
 
-CachedMetadata* Resource::CacheHandler::cachedMetadata(unsigned dataTypeID) const
+CachedMetadata* Resource::CachedMetadataHandlerImpl::cachedMetadata(unsigned dataTypeID) const
 {
-    return m_resource->cachedMetadata(dataTypeID);
+    if (!m_cachedMetadata || m_cachedMetadata->dataTypeID() != dataTypeID)
+        return nullptr;
+    return m_cachedMetadata.get();
 }
 
-String Resource::CacheHandler::encoding() const
+String Resource::CachedMetadataHandlerImpl::encoding() const
 {
     return m_resource->encoding();
 }
 
+void Resource::CachedMetadataHandlerImpl::setSerializedCachedMetadata(const char* data, size_t size)
+{
+    // We only expect to receive cached metadata from the platform once.
+    // If this triggers, it indicates an efficiency problem which is most
+    // likely unexpected in code designed to improve performance.
+    ASSERT(!m_cachedMetadata);
+    m_cachedMetadata = CachedMetadata::deserialize(data, size);
+}
+
+void Resource::CachedMetadataHandlerImpl::sendToPlatform()
+{
+    if (m_cachedMetadata) {
+        const Vector<char>& serializedData = m_cachedMetadata->serialize();
+        Platform::current()->cacheMetadata(response().url(), response().responseTime(), serializedData.data(), serializedData.size());
+    } else {
+        Platform::current()->cacheMetadata(response().url(), response().responseTime(), nullptr, 0);
+    }
+}
+
+class Resource::ServiceWorkerResponseCachedMetadataHandler : public Resource::CachedMetadataHandlerImpl {
+public:
+    static Resource::CachedMetadataHandlerImpl* create(Resource* resource, SecurityOrigin* securityOrigin)
+    {
+        return new ServiceWorkerResponseCachedMetadataHandler(resource, securityOrigin);
+    }
+    ~ServiceWorkerResponseCachedMetadataHandler() override {}
+    DECLARE_VIRTUAL_TRACE();
+
+protected:
+    void sendToPlatform() override;
+
+private:
+    explicit ServiceWorkerResponseCachedMetadataHandler(Resource*, SecurityOrigin*);
+    String m_cacheStorageCacheName;
+    RefPtr<SecurityOrigin> m_securityOrigin;
+};
+
+Resource::ServiceWorkerResponseCachedMetadataHandler::ServiceWorkerResponseCachedMetadataHandler(Resource* resource, SecurityOrigin* securityOrigin)
+    : CachedMetadataHandlerImpl(resource)
+    , m_securityOrigin(securityOrigin)
+{
+}
+
+DEFINE_TRACE(Resource::ServiceWorkerResponseCachedMetadataHandler)
+{
+    CachedMetadataHandlerImpl::trace(visitor);
+}
+
+void Resource::ServiceWorkerResponseCachedMetadataHandler::sendToPlatform()
+{
+    // We don't support sending the metadata to the platform when the response
+    // was directly fetched via a ServiceWorker
+    // (eg: FetchEvent.respondWith(fetch(FetchEvent.request))) to prevent an
+    // attacker's Service Worker from poisoning the metadata cache of HTTPCache.
+    if (response().cacheStorageCacheName().isNull())
+        return;
+
+    if (m_cachedMetadata) {
+        const Vector<char>& serializedData = m_cachedMetadata->serialize();
+        Platform::current()->cacheMetadataInCacheStorage(response().url(), response().responseTime(), serializedData.data(), serializedData.size(), WebSecurityOrigin(m_securityOrigin), response().cacheStorageCacheName());
+    } else {
+        Platform::current()->cacheMetadataInCacheStorage(response().url(), response().responseTime(), nullptr, 0, WebSecurityOrigin(m_securityOrigin), response().cacheStorageCacheName());
+    }
+}
+
 class Resource::ResourceCallback final : public GarbageCollectedFinalized<ResourceCallback> {
 public:
     static ResourceCallback& callbackHandler();
     DECLARE_TRACE();
     void schedule(Resource*);
     void cancel(Resource*);
     bool isScheduled(Resource*) const;
 private:
     ResourceCallback();
 
@@ skipping 70 matching lines @@
     , m_type(type)
     , m_status(NotStarted)
     , m_needsSynchronousCacheHit(false)
     , m_linkPreload(false)
 {
     ASSERT(m_type == unsigned(type)); // m_type is a bitfield, so this tests careless updates of the enum.
     InstanceCounters::incrementCounter(InstanceCounters::ResourceCounter);
 
     // Currently we support the metadata caching only for HTTP family.
     if (m_resourceRequest.url().protocolIsInHTTPFamily())
-        m_cacheHandler = CacheHandler::create(this);
+        m_cacheHandler = CachedMetadataHandlerImpl::create(this);
 }
 
 Resource::~Resource()
 {
     InstanceCounters::decrementCounter(InstanceCounters::ResourceCounter);
 }
 
 void Resource::removedFromMemoryCache()
 {
     InspectorInstrumentation::removedResourceFromMemoryCache(this);
@@ skipping 13 matching lines @@
         return;
 
     RELEASE_ASSERT(!m_loader);
     ASSERT(stillNeedsLoad());
     m_status = Pending;
 
     ResourceRequest& request(m_revalidatingRequest.isNull() ? m_resourceRequest : m_revalidatingRequest);
     KURL url = request.url();
     request.setAllowStoredCredentials(m_options.allowCredentials == AllowStoredCredentials);
 
+    m_fetcherSecurityOrigin = fetcher->context().getSecurityOrigin();
     m_loader = ResourceLoader::create(fetcher, this);
     m_loader->start(request);
     // If the request reference is null (i.e., a synchronous revalidation will
     // null the request), don't make the request non-null by setting the url.
     if (!request.isNull())
         request.setURL(url);
 }
 
 void Resource::checkNotify()
 {
@@ skipping 186 matching lines @@
     m_revalidatingRequest = request;
     m_status = NotStarted;
 }
 
 void Resource::willFollowRedirect(ResourceRequest& newRequest, const ResourceResponse& redirectResponse)
 {
     newRequest.setAllowStoredCredentials(m_options.allowCredentials == AllowStoredCredentials);
     m_redirectChain.append(RedirectPair(newRequest, redirectResponse));
 }
 
+void Resource::setResponse(const ResourceResponse& response)
+{
+    m_response = response;
+    if (m_response.wasFetchedViaServiceWorker())
+        m_cacheHandler = ServiceWorkerResponseCachedMetadataHandler::create(this, m_fetcherSecurityOrigin.get());
+}
+
 bool Resource::unlock()
 {
     if (!m_data)
         return false;
 
     if (!m_data->isLocked())
         return true;
 
     if (!memoryCache()->contains(this) || hasClientsOrObservers() || !m_revalidatingRequest.isNull() || !m_loadFinishTime || !isSafeToUnlock())
         return false;
@@ skipping 14 matching lines @@
         revalidationFailed();
     }
     setResponse(response);
     String encoding = response.textEncodingName();
     if (!encoding.isNull())
         setEncoding(encoding);
 }
 
 void Resource::setSerializedCachedMetadata(const char* data, size_t size)
 {
-    // We only expect to receive cached metadata from the platform once.
-    // If this triggers, it indicates an efficiency problem which is most
-    // likely unexpected in code designed to improve performance.
-    ASSERT(!m_cachedMetadata);
     ASSERT(m_revalidatingRequest.isNull());
-
-    m_cachedMetadata = CachedMetadata::deserialize(data, size);
+    ASSERT(!m_response.isNull());
+    if (m_cacheHandler)
+        m_cacheHandler->setSerializedCachedMetadata(data, size);
 }
 
 CachedMetadataHandler* Resource::cacheHandler()
 {
     return m_cacheHandler.get();
 }
 
-void Resource::setCachedMetadata(unsigned dataTypeID, const char* data, size_t size, CachedMetadataHandler::CacheType cacheType)
-{
-    // Currently, only one type of cached metadata per resource is supported.
-    // If the need arises for multiple types of metadata per resource this could
-    // be enhanced to store types of metadata in a map.
-    ASSERT(!m_cachedMetadata);
-
-    m_cachedMetadata = CachedMetadata::create(dataTypeID, data, size);
-
-    // We don't support sending the metadata to the platform when the response
-    // was fetched via a ServiceWorker to prevent an attacker's Service Worker
-    // from poisoning the metadata cache.
-    // FIXME: Support sending the metadata even if the response was fetched via
-    // a ServiceWorker. https://crbug.com/448706
-    if (cacheType == CachedMetadataHandler::SendToPlatform && !m_response.wasFetchedViaServiceWorker()) {
-        const Vector<char>& serializedData = m_cachedMetadata->serialize();
-        Platform::current()->cacheMetadata(m_response.url(), m_response.responseTime(), serializedData.data(), serializedData.size());
-    }
-}
-
-void Resource::clearCachedMetadata(CachedMetadataHandler::CacheType cacheType)
-{
-    m_cachedMetadata.clear();
-
-    if (cacheType == CachedMetadataHandler::SendToPlatform)
-        Platform::current()->cacheMetadata(m_response.url(), m_response.responseTime(), 0, 0);
-}
 
 String Resource::reasonNotDeletable() const
 {
     StringBuilder builder;
     if (hasClientsOrObservers()) {
         builder.append("hasClients(");
         builder.appendNumber(m_clients.size());
         if (!m_clientsAwaitingCallback.isEmpty()) {
             builder.append(", AwaitingCallback=");
             builder.appendNumber(m_clientsAwaitingCallback.size());
@@ skipping 17 matching lines @@
         builder.append(")");
     }
     if (memoryCache()->contains(this)) {
         if (!builder.isEmpty())
             builder.append(' ');
         builder.append("in_memory_cache");
     }
     return builder.toString();
 }
 
-CachedMetadata* Resource::cachedMetadata(unsigned dataTypeID) const
-{
-    if (!m_cachedMetadata || m_cachedMetadata->dataTypeID() != dataTypeID)
-        return nullptr;
-    return m_cachedMetadata.get();
-}
-
 void Resource::clearLoader()
 {
     m_loader = nullptr;
 }
 
 void Resource::didAddClient(ResourceClient* c)
 {
     if (isLoaded()) {
         c->notifyFinished(this);
         if (m_clients.contains(c)) {
@@ skipping 265 matching lines @@
     m_resourceRequest = m_revalidatingRequest;
     m_revalidatingRequest = ResourceRequest();
 }
 
 void Resource::revalidationFailed()
 {
     m_resourceRequest = m_revalidatingRequest;
     m_revalidatingRequest = ResourceRequest();
     m_redirectChain.clear();
     m_data.clear();
-    m_cachedMetadata.clear();
+    m_cacheHandler.clear();
     destroyDecodedDataForFailedRevalidation();
 }
 
 bool Resource::canReuseRedirectChain()
 {
     for (auto& redirect : m_redirectChain) {
         if (!canUseResponse(redirect.m_redirectResponse, m_responseTimestamp))
             return false;
         if (redirect.m_request.cacheControlContainsNoCache() || redirect.m_request.cacheControlContainsNoStore())
             return false;
@@ skipping 185 matching lines @@
     case Resource::Media:
         return "Media";
     case Resource::Manifest:
         return "Manifest";
     }
     ASSERT_NOT_REACHED();
     return "Unknown";
 }
 
 } // namespace blink