5c1044826f8ab7c192d604a2b1f79f2dfdb0f0d3
[WebKit-https.git] / Source / WebCore / loader / cache / CachedResourceLoader.h
1 /*
2     Copyright (C) 1998 Lars Knoll (knoll@mpi-hd.mpg.de)
3     Copyright (C) 2001 Dirk Mueller <mueller@kde.org>
4     Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011 Apple Inc. All rights reserved.
5     Copyright (C) 2009 Torch Mobile Inc. http://www.torchmobile.com/
6
7     This library is free software; you can redistribute it and/or
8     modify it under the terms of the GNU Library General Public
9     License as published by the Free Software Foundation; either
10     version 2 of the License, or (at your option) any later version.
11
12     This library is distributed in the hope that it will be useful,
13     but WITHOUT ANY WARRANTY; without even the implied warranty of
14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15     Library General Public License for more details.
16
17     You should have received a copy of the GNU Library General Public License
18     along with this library; see the file COPYING.LIB.  If not, write to
19     the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
20     Boston, MA 02110-1301, USA.
21
22     This class provides all functionality needed for loading images, style sheets and html
23     pages from the web. It has a memory cache for these objects.
24 */
25
26 #ifndef CachedResourceLoader_h
27 #define CachedResourceLoader_h
28
29 #include "CachePolicy.h"
30 #include "CachedResource.h"
31 #include "CachedResourceHandle.h"
32 #include "CachedResourceRequest.h"
33 #include "ResourceLoadPriority.h"
34 #include "Timer.h"
35 #include <wtf/Deque.h>
36 #include <wtf/HashMap.h>
37 #include <wtf/HashSet.h>
38 #include <wtf/ListHashSet.h>
39 #include <wtf/text/StringHash.h>
40
41 namespace WebCore {
42
43 class CachedCSSStyleSheet;
44 class CachedSVGDocument;
45 class CachedFont;
46 class CachedImage;
47 class CachedRawResource;
48 class CachedScript;
49 class CachedTextTrack;
50 class CachedXSLStyleSheet;
51 class Document;
52 class DocumentLoader;
53 class Frame;
54 class ImageLoader;
55 class URL;
56
57 // The CachedResourceLoader provides a per-context interface to the MemoryCache
58 // and enforces a bunch of security checks and rules for resource revalidation.
59 // Its lifetime is roughly per-DocumentLoader, in that it is generally created
60 // in the DocumentLoader constructor and loses its ability to generate network
61 // requests when the DocumentLoader is destroyed. Documents also hold a 
62 // RefPtr<CachedResourceLoader> for their lifetime (and will create one if they
63 // are initialized without a Frame), so a Document can keep a CachedResourceLoader
64 // alive past detach if scripts still reference the Document.
65 class CachedResourceLoader : public RefCounted<CachedResourceLoader> {
66     WTF_MAKE_NONCOPYABLE(CachedResourceLoader); WTF_MAKE_FAST_ALLOCATED;
67 friend class ImageLoader;
68 friend class ResourceCacheValidationSuppressor;
69
70 public:
71     static Ref<CachedResourceLoader> create(DocumentLoader* documentLoader) { return adoptRef(*new CachedResourceLoader(documentLoader)); }
72     ~CachedResourceLoader();
73
74     CachedResourceHandle<CachedImage> requestImage(CachedResourceRequest&);
75     CachedResourceHandle<CachedCSSStyleSheet> requestCSSStyleSheet(CachedResourceRequest&);
76     CachedResourceHandle<CachedCSSStyleSheet> requestUserCSSStyleSheet(CachedResourceRequest&);
77     CachedResourceHandle<CachedScript> requestScript(CachedResourceRequest&);
78     CachedResourceHandle<CachedFont> requestFont(CachedResourceRequest&, bool isSVG);
79     CachedResourceHandle<CachedRawResource> requestRawResource(CachedResourceRequest&);
80     CachedResourceHandle<CachedRawResource> requestMainResource(CachedResourceRequest&);
81     CachedResourceHandle<CachedSVGDocument> requestSVGDocument(CachedResourceRequest&);
82 #if ENABLE(XSLT)
83     CachedResourceHandle<CachedXSLStyleSheet> requestXSLStyleSheet(CachedResourceRequest&);
84 #endif
85 #if ENABLE(LINK_PREFETCH)
86     CachedResourceHandle<CachedResource> requestLinkResource(CachedResource::Type, CachedResourceRequest&);
87 #endif
88 #if ENABLE(VIDEO_TRACK)
89     CachedResourceHandle<CachedTextTrack> requestTextTrack(CachedResourceRequest&);
90 #endif
91
92     // Logs an access denied message to the console for the specified URL.
93     void printAccessDeniedMessage(const URL& url) const;
94
95     CachedResource* cachedResource(const String& url) const;
96     CachedResource* cachedResource(const URL& url) const;
97     
98     typedef HashMap<String, CachedResourceHandle<CachedResource>> DocumentResourceMap;
99     const DocumentResourceMap& allCachedResources() const { return m_documentResources; }
100
101     bool autoLoadImages() const { return m_autoLoadImages; }
102     void setAutoLoadImages(bool);
103
104     void setImagesEnabled(bool);
105
106     bool shouldDeferImageLoad(const URL&) const;
107     bool shouldPerformImageLoad(const URL&) const;
108     
109     CachePolicy cachePolicy(CachedResource::Type) const;
110     
111     Frame* frame() const; // Can be null
112     Document* document() const { return m_document; } // Can be null
113     void setDocument(Document* document) { m_document = document; }
114     void clearDocumentLoader() { m_documentLoader = nullptr; }
115     SessionID sessionID() const;
116
117     void removeCachedResource(CachedResource&);
118
119     void loadDone(CachedResource*, bool shouldPerformPostLoadActions = true);
120
121     WEBCORE_EXPORT void garbageCollectDocumentResources();
122     
123     void incrementRequestCount(const CachedResource*);
124     void decrementRequestCount(const CachedResource*);
125     int requestCount() const { return m_requestCount; }
126
127     WEBCORE_EXPORT bool isPreloaded(const String& urlString) const;
128     void clearPreloads();
129     void clearPendingPreloads();
130     void preload(CachedResource::Type, CachedResourceRequest&, const String& charset);
131     void checkForPendingPreloads();
132     void printPreloadStats();
133     bool canRequest(CachedResource::Type, const URL&, const ResourceLoaderOptions&, bool forPreload = false);
134
135     static const ResourceLoaderOptions& defaultCachedResourceOptions();
136
137     void documentDidFinishLoadEvent();
138
139 private:
140     explicit CachedResourceLoader(DocumentLoader*);
141
142     CachedResourceHandle<CachedResource> requestResource(CachedResource::Type, CachedResourceRequest&);
143     CachedResourceHandle<CachedResource> revalidateResource(const CachedResourceRequest&, CachedResource*);
144     CachedResourceHandle<CachedResource> loadResource(CachedResource::Type, CachedResourceRequest&);
145 #if ENABLE(RESOURCE_TIMING)
146     void storeResourceTimingInitiatorInformation(const CachedResourceHandle<CachedResource>&, const CachedResourceRequest&);
147 #endif
148     void requestPreload(CachedResource::Type, CachedResourceRequest&, const String& charset);
149
150     enum RevalidationPolicy { Use, Revalidate, Reload, Load };
151     RevalidationPolicy determineRevalidationPolicy(CachedResource::Type, ResourceRequest&, bool forPreload, CachedResource* existingResource, CachedResourceRequest::DeferOption) const;
152     
153     bool shouldContinueAfterNotifyingLoadedFromMemoryCache(const CachedResourceRequest&, CachedResource*);
154     bool checkInsecureContent(CachedResource::Type, const URL&) const;
155
156     void performPostLoadActions();
157
158     bool clientDefersImage(const URL&) const;
159     void reloadImagesIfNotDeferred();
160
161     bool canRequestInContentDispositionAttachmentSandbox(CachedResource::Type, const URL&) const;
162     
163     HashSet<String> m_validatedURLs;
164     mutable DocumentResourceMap m_documentResources;
165     Document* m_document;
166     DocumentLoader* m_documentLoader;
167     
168     int m_requestCount;
169     
170     std::unique_ptr<ListHashSet<CachedResource*>> m_preloads;
171     struct PendingPreload {
172         CachedResource::Type m_type;
173         CachedResourceRequest m_request;
174         String m_charset;
175     };
176     Deque<PendingPreload> m_pendingPreloads;
177
178     Timer m_garbageCollectDocumentResourcesTimer;
179
180 #if ENABLE(RESOURCE_TIMING)
181     struct InitiatorInfo {
182         AtomicString name;
183         double startTime;
184     };
185     HashMap<CachedResource*, InitiatorInfo> m_initiatorMap;
186 #endif
187
188     // 29 bits left
189     bool m_autoLoadImages : 1;
190     bool m_imagesEnabled : 1;
191     bool m_allowStaleResources : 1;
192 };
193
194 class ResourceCacheValidationSuppressor {
195     WTF_MAKE_NONCOPYABLE(ResourceCacheValidationSuppressor);
196     WTF_MAKE_FAST_ALLOCATED;
197 public:
198     ResourceCacheValidationSuppressor(CachedResourceLoader& loader)
199         : m_loader(loader)
200         , m_previousState(m_loader.m_allowStaleResources)
201     {
202         m_loader.m_allowStaleResources = true;
203     }
204     ~ResourceCacheValidationSuppressor()
205     {
206         m_loader.m_allowStaleResources = m_previousState;
207     }
208 private:
209     CachedResourceLoader& m_loader;
210     bool m_previousState;
211 };
212
213 } // namespace WebCore
214
215 #endif