e1fb02cfacfff02ea218c22bfa868794481afd15
[WebKit-https.git] / Source / WebCore / loader / cache / CachedCSSStyleSheet.cpp
1 /*
2     Copyright (C) 1998 Lars Knoll (knoll@mpi-hd.mpg.de)
3     Copyright (C) 2001 Dirk Mueller (mueller@kde.org)
4     Copyright (C) 2002 Waldo Bastian (bastian@kde.org)
5     Copyright (C) 2006 Samuel Weinig (sam.weinig@gmail.com)
6     Copyright (C) 2004, 2005, 2006 Apple Computer, Inc.
7
8     This library is free software; you can redistribute it and/or
9     modify it under the terms of the GNU Library General Public
10     License as published by the Free Software Foundation; either
11     version 2 of the License, or (at your option) any later version.
12
13     This library is distributed in the hope that it will be useful,
14     but WITHOUT ANY WARRANTY; without even the implied warranty of
15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16     Library General Public License for more details.
17
18     You should have received a copy of the GNU Library General Public License
19     along with this library; see the file COPYING.LIB.  If not, write to
20     the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
21     Boston, MA 02110-1301, USA.
22
23     This class provides all functionality needed for loading images, style sheets and html
24     pages from the web. It has a memory cache for these objects.
25 */
26
27 #include "config.h"
28 #include "CachedCSSStyleSheet.h"
29
30 #include "CSSStyleSheet.h"
31 #include "CachedResourceClientWalker.h"
32 #include "CachedStyleSheetClient.h"
33 #include "HTTPParsers.h"
34 #include "MemoryCache.h"
35 #include "SharedBuffer.h"
36 #include "StyleSheetContents.h"
37 #include "TextResourceDecoder.h"
38 #include <wtf/CurrentTime.h>
39 #include <wtf/Vector.h>
40
41 namespace WebCore {
42
43 CachedCSSStyleSheet::CachedCSSStyleSheet(const ResourceRequest& resourceRequest, const String& charset)
44     : CachedResource(resourceRequest, CSSStyleSheet)
45     , m_decoder(TextResourceDecoder::create("text/css", charset))
46 {
47     // Prefer text/css but accept any type (dell.com serves a stylesheet
48     // as text/html; see <http://bugs.webkit.org/show_bug.cgi?id=11451>).
49     setAccept("text/css,*/*;q=0.1");
50 }
51
52 CachedCSSStyleSheet::~CachedCSSStyleSheet()
53 {
54     if (m_parsedStyleSheetCache)
55         m_parsedStyleSheetCache->removedFromMemoryCache();
56 }
57
58 void CachedCSSStyleSheet::didAddClient(CachedResourceClient* c)
59 {
60     ASSERT(c->resourceClientType() == CachedStyleSheetClient::expectedType());
61     if (!isLoading())
62         static_cast<CachedStyleSheetClient*>(c)->setCSSStyleSheet(m_resourceRequest.url(), m_response.url(), m_decoder->encoding().name(), this);
63
64     CachedResource::didAddClient(c);
65 }
66
67 void CachedCSSStyleSheet::setEncoding(const String& chs)
68 {
69     m_decoder->setEncoding(chs, TextResourceDecoder::EncodingFromHTTPHeader);
70 }
71
72 String CachedCSSStyleSheet::encoding() const
73 {
74     return m_decoder->encoding().name();
75 }
76     
77 const String CachedCSSStyleSheet::sheetText(bool enforceMIMEType, bool* hasValidMIMEType) const 
78
79     ASSERT(!isPurgeable());
80
81     if (!m_data || m_data->isEmpty() || !canUseSheet(enforceMIMEType, hasValidMIMEType))
82         return String();
83     
84     if (!m_decodedSheetText.isNull())
85         return m_decodedSheetText;
86     
87     // Don't cache the decoded text, regenerating is cheap and it can use quite a bit of memory
88     String sheetText = m_decoder->decode(m_data->data(), m_data->size());
89     sheetText += m_decoder->flush();
90     return sheetText;
91 }
92
93 void CachedCSSStyleSheet::data(PassRefPtr<SharedBuffer> data, bool allDataReceived)
94 {
95     if (!allDataReceived)
96         return;
97
98     m_data = data;
99     setEncodedSize(m_data.get() ? m_data->size() : 0);
100     // Decode the data to find out the encoding and keep the sheet text around during checkNotify()
101     if (m_data) {
102         m_decodedSheetText = m_decoder->decode(m_data->data(), m_data->size());
103         m_decodedSheetText += m_decoder->flush();
104     }
105     setLoading(false);
106     checkNotify();
107     // Clear the decoded text as it is unlikely to be needed immediately again and is cheap to regenerate.
108     m_decodedSheetText = String();
109 }
110
111 void CachedCSSStyleSheet::checkNotify()
112 {
113     if (isLoading())
114         return;
115
116     CachedResourceClientWalker<CachedStyleSheetClient> w(m_clients);
117     while (CachedStyleSheetClient* c = w.next())
118         c->setCSSStyleSheet(m_resourceRequest.url(), m_response.url(), m_decoder->encoding().name(), this);
119 }
120
121 void CachedCSSStyleSheet::error(CachedResource::Status status)
122 {
123     setStatus(status);
124     ASSERT(errorOccurred());
125     setLoading(false);
126     checkNotify();
127 }
128
129 bool CachedCSSStyleSheet::canUseSheet(bool enforceMIMEType, bool* hasValidMIMEType) const
130 {
131     if (errorOccurred())
132         return false;
133         
134     if (!enforceMIMEType && !hasValidMIMEType)
135         return true;
136
137     // This check exactly matches Firefox.  Note that we grab the Content-Type
138     // header directly because we want to see what the value is BEFORE content
139     // sniffing.  Firefox does this by setting a "type hint" on the channel.
140     // This implementation should be observationally equivalent.
141     //
142     // This code defaults to allowing the stylesheet for non-HTTP protocols so
143     // folks can use standards mode for local HTML documents.
144     String mimeType = extractMIMETypeFromMediaType(response().httpHeaderField("Content-Type"));
145     bool typeOK = mimeType.isEmpty() || equalIgnoringCase(mimeType, "text/css") || equalIgnoringCase(mimeType, "application/x-unknown-content-type");
146     if (hasValidMIMEType)
147         *hasValidMIMEType = typeOK;
148     if (!enforceMIMEType)
149         return true;
150     return typeOK;
151 }
152
153 void CachedCSSStyleSheet::destroyDecodedData()
154 {
155     if (!m_parsedStyleSheetCache)
156         return;
157
158     m_parsedStyleSheetCache->removedFromMemoryCache();
159     m_parsedStyleSheetCache.clear();
160
161     setDecodedSize(0);
162
163     if (!MemoryCache::shouldMakeResourcePurgeableOnEviction() && isSafeToMakePurgeable())
164         makePurgeable(true);
165 }
166
167 PassRefPtr<StyleSheetContents> CachedCSSStyleSheet::restoreParsedStyleSheet(const CSSParserContext& context)
168 {
169     if (!m_parsedStyleSheetCache)
170         return 0;
171     ASSERT(m_parsedStyleSheetCache->isCacheable());
172     ASSERT(m_parsedStyleSheetCache->isInMemoryCache());
173
174     // Contexts must be identical so we know we would get the same exact result if we parsed again.
175     if (m_parsedStyleSheetCache->parserContext() != context)
176         return 0;
177
178     didAccessDecodedData(currentTime());
179
180     return m_parsedStyleSheetCache;
181 }
182
183 void CachedCSSStyleSheet::saveParsedStyleSheet(PassRefPtr<StyleSheetContents> sheet)
184 {
185     ASSERT(sheet && sheet->isCacheable());
186
187     if (m_parsedStyleSheetCache)
188         m_parsedStyleSheetCache->removedFromMemoryCache();
189     m_parsedStyleSheetCache = sheet;
190     m_parsedStyleSheetCache->addedToMemoryCache();
191
192     setDecodedSize(m_parsedStyleSheetCache->estimatedSizeInBytes());
193 }
194
195 }