[Storage] Cache storage inspection on all the frames!

Source/modules/serviceworkers/InspectorServiceWorkerCacheAgent.cpp

 // 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 "config.h"
 #include "modules/serviceworkers/InspectorServiceWorkerCacheAgent.h"
 
+#include "bindings/core/v8/ScriptState.h"
 #include "core/InspectorBackendDispatcher.h"
 #include "core/InspectorTypeBuilder.h"
+#include "core/dom/Document.h"
+#include "core/frame/DOMWindow.h"
+#include "core/frame/LocalFrame.h"
+#include "core/page/Page.h"
+#include "modules/serviceworkers/CacheStorage.h"
+#include "modules/serviceworkers/GlobalCacheStorage.h"
 #include "modules/serviceworkers/ServiceWorkerGlobalScope.h"
 #include "modules/serviceworkers/ServiceWorkerGlobalScopeClient.h"
 #include "platform/JSONValues.h"
 #include "platform/heap/Handle.h"
 #include "platform/weborigin/DatabaseIdentifier.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebServiceWorkerCache.h"
 #include "public/platform/WebServiceWorkerCacheError.h"
 #include "public/platform/WebServiceWorkerCacheStorage.h"
 #include "public/platform/WebServiceWorkerRequest.h"
 #include "public/platform/WebServiceWorkerResponse.h"
 #include "public/platform/WebString.h"
 #include "public/platform/WebURL.h"
 #include "public/platform/WebVector.h"
 #include "wtf/Noncopyable.h"
 #include "wtf/OwnPtr.h"
 #include "wtf/PassRefPtr.h"
 #include "wtf/RefCounted.h"
 #include "wtf/RefPtr.h"
 #include "wtf/Vector.h"
 #include "wtf/text/StringBuilder.h"
 
 #include <algorithm>
 
 using blink::TypeBuilder::Array;
-using blink::TypeBuilder::ServiceWorkerCache::DataEntry;
+using blink::TypeBuilder::CacheStorage::DataEntry;
 
-typedef blink::InspectorBackendDispatcher::ServiceWorkerCacheCommandHandler::DeleteCacheCallback DeleteCacheCallback;
-typedef blink::InspectorBackendDispatcher::ServiceWorkerCacheCommandHandler::RequestCacheNamesCallback RequestCacheNamesCallback;
-typedef blink::InspectorBackendDispatcher::ServiceWorkerCacheCommandHandler::RequestEntriesCallback RequestEntriesCallback;
+typedef blink::InspectorBackendDispatcher::CacheStorageCommandHandler::DeleteCacheCallback DeleteCacheCallback;
+typedef blink::InspectorBackendDispatcher::CacheStorageCommandHandler::RequestCacheNamesCallback RequestCacheNamesCallback;
+typedef blink::InspectorBackendDispatcher::CacheStorageCommandHandler::RequestEntriesCallback RequestEntriesCallback;
 typedef blink::InspectorBackendDispatcher::CallbackBase RequestCallback;
 
 namespace blink {
 
 namespace {
 
-PassOwnPtr<WebServiceWorkerCacheStorage> assertCacheStorage(ErrorString* errorString, ServiceWorkerGlobalScope* globalScope)
+LocalFrame* findFrameWithSecurityOrigin(Page* page, const String& securityOrigin)

[pfeldman, 2015/04/02 21:18:58]

Sorry, did I miss a reply here? I was suggesting to remove this method.

 {
-    String identifier = createDatabaseIdentifierFromSecurityOrigin(globalScope->securityOrigin());
-    OwnPtr<WebServiceWorkerCacheStorage> caches = adoptPtr(Platform::current()->cacheStorage(identifier));
-    if (!caches) {
-        *errorString = "Cache Storage not available in global scope.";
+    for (Frame* frame = page->mainFrame(); frame; frame = frame->tree().traverseNext()) {
+        if (!frame->isLocalFrame())
+            continue;
+        RefPtr<SecurityOrigin> documentOrigin = toLocalFrame(frame)->document()->securityOrigin();
+        if (documentOrigin->toRawString() == securityOrigin)
+            return toLocalFrame(frame);
+    }
+    return 0;
+}
+
+DOMWindow* assertWindow(ErrorString* errorString, LocalFrame* frame)

[pfeldman, 2015/04/02 21:18:58]

And this method

+{
+    DOMWindow* window = frame ? frame->domWindow() : 0;
+
+    if (!window)
+        *errorString = "No window for given frame found.";
+
+    return window;
+}
+
+WebServiceWorkerCacheStorage* assertCacheStorage(ErrorString* errorString, DOMWindow& window)
+{
+    TrackExceptionState exceptionState;

[pfeldman, 2015/04/02 21:18:58]

And go straight to WebServiceWorkerCacheStorage here through the platform
accessor.

[dmurph, 2015/04/02 23:23:18]

Ah.  So calling Platform::current()->cacheStorage(securityOrigin) currently
check fails.  I'm not sure why.  I can add a todo for fixing that and rewiring
this (as well as indexeddb)

+    CacheStorage* cache = GlobalCacheStorage::caches(window, exceptionState);
+    if (exceptionState.hadException()) {
+        *errorString = "Exception while retrieving Cache Storage.";
         return nullptr;
     }
-    return caches.release();
+    if (!cache) {
+        *errorString = "Cache Storage not available.";
+        return nullptr;
+    }
+    WebServiceWorkerCacheStorage* webCache = cache->webCacheStorage();
+    if (!webCache) {
+        *errorString = "Web cache not found on Cache Storage.";
+        return nullptr;
+    }
+    return webCache;
+}
+
+WebServiceWorkerCacheStorage* assertCacheStorage(ErrorString* errorString, WorkerGlobalScope& scope)
+{
+    TrackExceptionState exceptionState;
+    CacheStorage* cache = GlobalCacheStorage::caches(scope, exceptionState);
+    if (exceptionState.hadException()) {
+        *errorString = "Exception while retrieving Cache Storage.";
+        return nullptr;
+    }
+    if (!cache) {
+        *errorString = "Cache Storage not available.";
+        return nullptr;
+    }
+    WebServiceWorkerCacheStorage* webCache = cache->webCacheStorage();
+    if (!webCache) {
+        *errorString = "Web cache not found on Cache Storage.";
+        return nullptr;
+    }
+    return webCache;
 }
 
 CString serviceWorkerCacheErrorString(WebServiceWorkerCacheError* error)
 {
     switch (*error) {
     case WebServiceWorkerCacheErrorNotImplemented:
         return CString("not implemented.");
         break;
     case WebServiceWorkerCacheErrorNotFound:
         return CString("not found.");
@@ skipping 218 matching lines @@
     {
         m_callback->sendFailure(String::format("Error requesting cache names: %s", serviceWorkerCacheErrorString(error).data()));
     }
 
 private:
     RefPtrWillBePersistent<DeleteCacheCallback> m_callback;
 };
 
 } // namespace
 
-InspectorServiceWorkerCacheAgent::InspectorServiceWorkerCacheAgent(ServiceWorkerGlobalScope* scope)
-    : InspectorBaseAgent<blink::InspectorServiceWorkerCacheAgent, InspectorFrontend::ServiceWorkerCache>("ServiceWorkerCache")
-    , m_globalScope(scope)
+InspectorServiceWorkerCacheAgent::InspectorServiceWorkerCacheAgent(Page* page)
+    : InspectorBaseAgent<InspectorServiceWorkerCacheAgent, InspectorFrontend::CacheStorage>("CacheStorage")
+    , m_page(page)
+    , m_serviceWorkerGlobalScope(nullptr)
 {
 }
 
+InspectorServiceWorkerCacheAgent::InspectorServiceWorkerCacheAgent(ServiceWorkerGlobalScope* scope)
+    : InspectorBaseAgent<InspectorServiceWorkerCacheAgent, InspectorFrontend::CacheStorage>("CacheStorage")
+    , m_page(nullptr)
+    , m_serviceWorkerGlobalScope(scope)
+{
+}
+
+
 InspectorServiceWorkerCacheAgent::~InspectorServiceWorkerCacheAgent() { }
 
 DEFINE_TRACE(InspectorServiceWorkerCacheAgent)
 {
     InspectorBaseAgent::trace(visitor);
 }
 
-void InspectorServiceWorkerCacheAgent::requestCacheNames(ErrorString* errorString, PassRefPtrWillBeRawPtr<RequestCacheNamesCallback> callback)
+void InspectorServiceWorkerCacheAgent::requestCacheNames(ErrorString* errorString, const String& securityOrigin, PassRefPtrWillBeRawPtr<RequestCacheNamesCallback> callback)
 {
-    OwnPtr<WebServiceWorkerCacheStorage> cache = assertCacheStorage(errorString, m_globalScope);
+    WebServiceWorkerCacheStorage* cache;
+    if (!m_serviceWorkerGlobalScope) {
+        LocalFrame* frame = findFrameWithSecurityOrigin(m_page, securityOrigin);
+        DOMWindow* window = assertWindow(errorString, frame);
+        if (!window) {
+            callback->sendFailure(*errorString);
+            return;
+        }
+        cache = assertCacheStorage(errorString, *window);
+    } else {
+        cache = assertCacheStorage(errorString, *m_serviceWorkerGlobalScope);
+    }
     if (!cache) {
         callback->sendFailure(*errorString);
         return;
     }
     cache->dispatchKeys(new RequestCacheNames(callback));
 }
 
-
-void InspectorServiceWorkerCacheAgent::requestEntries(ErrorString* errorString, const String& cacheName, int skipCount, int pageSize, PassRefPtrWillBeRawPtr<RequestEntriesCallback> callback)
+void InspectorServiceWorkerCacheAgent::requestEntries(ErrorString* errorString, const String& securityOrigin, const String& cacheName, int skipCount, int pageSize, PassRefPtrWillBeRawPtr<RequestEntriesCallback> callback)
 {
-    OwnPtr<WebServiceWorkerCacheStorage> cache = assertCacheStorage(errorString, m_globalScope);
+    WebServiceWorkerCacheStorage* cache;
+    if (!m_serviceWorkerGlobalScope) {
+        LocalFrame* frame = findFrameWithSecurityOrigin(m_page, securityOrigin);
+        DOMWindow* window = assertWindow(errorString, frame);
+        if (!window) {
+            callback->sendFailure(*errorString);
+            return;
+        }
+        cache = assertCacheStorage(errorString, *window);
+    } else {
+        cache = assertCacheStorage(errorString, *m_serviceWorkerGlobalScope);
+    }
     if (!cache) {
         callback->sendFailure(*errorString);
         return;
     }
     DataRequestParams params;
     params.cacheName = cacheName;
     params.pageSize = pageSize;
     params.skipCount = skipCount;
     cache->dispatchOpen(new GetCacheForRequestData(params, callback), WebString(cacheName));
 }
 
-void InspectorServiceWorkerCacheAgent::deleteCache(ErrorString* errorString, const String& cacheName, PassRefPtrWillBeRawPtr<DeleteCacheCallback> callback)
+void InspectorServiceWorkerCacheAgent::deleteCache(ErrorString* errorString, const String& securityOrigin, const String& cacheName, PassRefPtrWillBeRawPtr<DeleteCacheCallback> callback)
 {
-    OwnPtr<WebServiceWorkerCacheStorage> cache = assertCacheStorage(errorString, m_globalScope);
+    WebServiceWorkerCacheStorage* cache;
+    if (!m_serviceWorkerGlobalScope) {
+        LocalFrame* frame = findFrameWithSecurityOrigin(m_page, securityOrigin);
+        DOMWindow* window = assertWindow(errorString, frame);
+        if (!window) {
+            callback->sendFailure(*errorString);
+            return;
+        }
+        cache = assertCacheStorage(errorString, *window);
+    } else {
+        cache = assertCacheStorage(errorString, *m_serviceWorkerGlobalScope);
+    }
     if (!cache) {
         callback->sendFailure(*errorString);
         return;
     }
     cache->dispatchDelete(new DeleteCache(callback), WebString(cacheName));
 }
 
 } // namespace blink