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     if (m_parsedStyleSheetCache->hasFailedOrCanceledSubresources()) {
 172         m_parsedStyleSheetCache->removedFromMemoryCache();
 173         m_parsedStyleSheetCache.clear();
 174         return 0;
 175     }
 176
 177     ASSERT(m_parsedStyleSheetCache->isCacheable());
 178     ASSERT(m_parsedStyleSheetCache->isInMemoryCache());
 179
 180     // Contexts must be identical so we know we would get the same exact result if we parsed again.
 181     if (m_parsedStyleSheetCache->parserContext() != context)
 182         return 0;
 183
 184     didAccessDecodedData(currentTime());
 185
 186     return m_parsedStyleSheetCache;
 187 }
 188
 189 void CachedCSSStyleSheet::saveParsedStyleSheet(PassRefPtr<StyleSheetContents> sheet)
 190 {
 191     ASSERT(sheet && sheet->isCacheable());
 192
 193     if (m_parsedStyleSheetCache)
 194         m_parsedStyleSheetCache->removedFromMemoryCache();
 195     m_parsedStyleSheetCache = sheet;
 196     m_parsedStyleSheetCache->addedToMemoryCache();
 197
 198     setDecodedSize(m_parsedStyleSheetCache->estimatedSizeInBytes());
 199 }
 200
 201 }