Query directly for cache partition names
[WebKit-https.git] / Source / WebKit2 / WebProcess / ResourceCache / WebResourceCacheManager.cpp
1 /*
2  * Copyright (C) 2011, 2013 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "WebResourceCacheManager.h"
28
29 #include "SecurityOriginData.h"
30 #include "WebCoreArgumentCoders.h"
31 #include "WebProcess.h"
32 #include "WebResourceCacheManagerMessages.h"
33 #include "WebResourceCacheManagerProxyMessages.h"
34 #include <WebCore/MemoryCache.h>
35 #include <WebCore/SecurityOrigin.h>
36 #include <WebCore/SecurityOriginHash.h>
37 #include <wtf/UnusedParam.h>
38
39 using namespace WebCore;
40
41 namespace WebKit {
42
43 const char* WebResourceCacheManager::supplementName()
44 {
45     return "WebResourceCacheManager";
46 }
47
48 WebResourceCacheManager::WebResourceCacheManager(WebProcess* process)
49     : m_process(process)
50 {
51     m_process->addMessageReceiver(Messages::WebResourceCacheManager::messageReceiverName(), this);
52 }
53
54 void WebResourceCacheManager::getCacheOrigins(uint64_t callbackID) const
55 {
56 #if USE(CFURLCACHE) && ENABLE(CACHE_PARTITIONING)
57     __block MemoryCache::SecurityOriginSet origins;
58 #else
59     MemoryCache::SecurityOriginSet origins;
60 #endif
61     memoryCache()->getOriginsWithCache(origins);
62
63 #if USE(CFURLCACHE)
64 #if ENABLE(CACHE_PARTITIONING)
65     cfURLCacheHostNamesWithCallback(^(RetainPtr<CFArrayRef> cfURLHosts) {
66 #else
67         RetainPtr<CFArrayRef> cfURLHosts = cfURLCacheHostNames();
68 #endif
69         CFIndex size = cfURLHosts ? CFArrayGetCount(cfURLHosts.get()) : 0;
70
71         String httpString("http");
72         for (CFIndex i = 0; i < size; ++i) {
73             CFStringRef host = static_cast<CFStringRef>(CFArrayGetValueAtIndex(cfURLHosts.get(), i));
74             origins.add(SecurityOrigin::create(httpString, host, 0));
75         }
76 #endif
77
78         returnCacheOrigins(callbackID, origins);
79
80 #if USE(CFURLCACHE) && ENABLE(CACHE_PARTITIONING)
81     });
82 #endif
83 }
84
85 void WebResourceCacheManager::returnCacheOrigins(uint64_t callbackID, const MemoryCache::SecurityOriginSet& origins) const
86 {
87     // Create a list with the origins in both of the caches.
88     Vector<SecurityOriginData> identifiers;
89     identifiers.reserveCapacity(origins.size());
90
91     MemoryCache::SecurityOriginSet::iterator end = origins.end();
92     for (MemoryCache::SecurityOriginSet::iterator it = origins.begin(); it != end; ++it) {
93         RefPtr<SecurityOrigin> origin = *it;
94         
95         SecurityOriginData originData;
96         originData.protocol = origin->protocol();
97         originData.host = origin->host();
98         originData.port = origin->port();
99
100         identifiers.uncheckedAppend(originData);
101     }
102
103     m_process->send(Messages::WebResourceCacheManagerProxy::DidGetCacheOrigins(identifiers, callbackID), 0);
104 }
105
106 void WebResourceCacheManager::clearCacheForOrigin(const SecurityOriginData& originData, uint32_t cachesToClear) const
107 {
108 #if USE(CFURLCACHE)
109     ResourceCachesToClear resourceCachesToClear = static_cast<ResourceCachesToClear>(cachesToClear);
110 #else
111     UNUSED_PARAM(cachesToClear);
112 #endif
113
114     RefPtr<SecurityOrigin> origin = SecurityOrigin::create(originData.protocol, originData.host, originData.port);
115     if (!origin)
116         return;
117
118     memoryCache()->removeResourcesWithOrigin(origin.get());
119
120 #if USE(CFURLCACHE)
121     if (resourceCachesToClear != InMemoryResourceCachesOnly) { 
122         RetainPtr<CFMutableArrayRef> hostArray(AdoptCF, CFArrayCreateMutable(0, 0, &kCFTypeArrayCallBacks));
123         CFArrayAppendValue(hostArray.get(), origin->host().createCFString().get());
124
125         clearCFURLCacheForHostNames(hostArray.get());
126     }
127 #endif
128 }
129
130 void WebResourceCacheManager::clearCacheForAllOrigins(uint32_t cachesToClear) const
131 {
132     ResourceCachesToClear resourceCachesToClear = static_cast<ResourceCachesToClear>(cachesToClear);
133     m_process->clearResourceCaches(resourceCachesToClear);
134 }
135
136 } // namespace WebKit