Prefetch @import files from CSS files.

Source/core/fetch/CSSStyleSheetResource.cpp

 /*
     Copyright (C) 1998 Lars Knoll (knoll@mpi-hd.mpg.de)
     Copyright (C) 2001 Dirk Mueller (mueller@kde.org)
     Copyright (C) 2002 Waldo Bastian (bastian@kde.org)
     Copyright (C) 2006 Samuel Weinig (sam.weinig@gmail.com)
     Copyright (C) 2004, 2005, 2006 Apple Computer, Inc.
 
     This library is free software; you can redistribute it and/or
     modify it under the terms of the GNU Library General Public
     License as published by the Free Software Foundation; either
@@ skipping 12 matching lines @@
     This class provides all functionality needed for loading images, style sheets and html
     pages from the web. It has a memory cache for these objects.
 */
 
 #include "config.h"
 #include "core/fetch/CSSStyleSheetResource.h"
 
 #include "core/css/StyleSheetContents.h"
 #include "core/fetch/ResourceClientWalker.h"
 #include "core/fetch/StyleSheetResourceClient.h"
+#include "core/html/parser/CompactHTMLToken.h"
 #include "core/html/parser/TextResourceDecoder.h"
 #include "platform/SharedBuffer.h"
 #include "platform/network/HTTPParsers.h"
 #include "wtf/CurrentTime.h"
 #include "wtf/Vector.h"
 
 namespace WebCore {
 
-CSSStyleSheetResource::CSSStyleSheetResource(const ResourceRequest& resourceRequest, const String& charset)
+CSSStyleSheetResource::CSSStyleSheetResource(const ResourceRequest& resourceRequest, const String& charset, Document* document)
     : StyleSheetResource(resourceRequest, CSSStyleSheet)
     , m_decoder(TextResourceDecoder::create("text/css", charset))
+    , m_tokenizer(HTMLTokenizer::create(m_options))
+    , m_preloadScanner(adoptPtr(new CSSPreloadScanner))
+    , m_input(adoptPtr(new BackgroundHTMLInputStream))
+    , m_token(adoptPtr(new HTMLToken))
+    , m_preloader(adoptPtr(new HTMLResourcePreloader(document)))
+    , m_prescan(true)
 {
     DEFINE_STATIC_LOCAL(const AtomicString, acceptCSS, ("text/css,*/*;q=0.1", AtomicString::ConstructFromLiteral));
 
     // Prefer text/css but accept any type (dell.com serves a stylesheet
     // as text/html; see <http://bugs.webkit.org/show_bug.cgi?id=11451>).
     setAccept(acceptCSS);
+    m_tokenizer->setState(HTMLTokenizer::RAWTEXTState);
 }
 
 CSSStyleSheetResource::~CSSStyleSheetResource()
 {
     if (m_parsedStyleSheetCache)
         m_parsedStyleSheetCache->removedFromMemoryCache();
 }
 
 void CSSStyleSheetResource::didAddClient(ResourceClient* c)
 {
@@ skipping 41 matching lines @@
         m_decodedSheetText = m_decodedSheetText + m_decoder->flush();
     }
 
     ResourceClientWalker<StyleSheetResourceClient> w(m_clients);
     while (StyleSheetResourceClient* c = w.next())
         c->setCSSStyleSheet(m_resourceRequest.url(), m_response.url(), m_decoder->encoding().name(), this);
     // Clear the decoded text as it is unlikely to be needed immediately again and is cheap to regenerate.
     m_decodedSheetText = String();
 }
 
+void CSSStyleSheetResource::appendData(const char* data, int length)
+{
+    Resource::appendData(data, length);
+    if (m_prescan) {
+        // FIXME: Allow scanning entire CSS file for background images and prefetch according to applied rules.
+        m_prescan = false;
+        prescan(data);
+    }
+}
+
+void CSSStyleSheetResource::prescan(const char* data)
+{
+    PreloadRequestStream pendingPreloads;
+    m_input->append(data);
+
+    while (true) {
+        if (!m_tokenizer->nextToken(m_input->current(), *m_token))
+            break; // We've reached the end of our current input.
+
+        CompactHTMLToken token(m_token.get(), TextPosition(m_input->current().currentLine(), m_input->current().currentColumn()));
+        m_preloadScanner->scan(token.data(), m_input->current(), pendingPreloads);

[shatch, 2014/02/14 16:20:17]

What's the advantage to running this through the tokenizer before sending it to
the preload scanner? Otherwise, seems like it would be cleaner to just send the
data directly.

+        m_token->clear();
+    }
+    m_preloader->takeAndPreload(pendingPreloads);
+}
+
 bool CSSStyleSheetResource::canUseSheet(bool enforceMIMEType, bool* hasValidMIMEType) const
 {
     if (errorOccurred())
         return false;
 
     if (!enforceMIMEType && !hasValidMIMEType)
         return true;
 
     // This check exactly matches Firefox. Note that we grab the Content-Type
     // header directly because we want to see what the value is BEFORE content
@@ skipping 53 matching lines @@
 
     if (m_parsedStyleSheetCache)
         m_parsedStyleSheetCache->removedFromMemoryCache();
     m_parsedStyleSheetCache = sheet;
     m_parsedStyleSheetCache->addedToMemoryCache();
 
     setDecodedSize(m_parsedStyleSheetCache->estimatedSizeInBytes());
 }
 
 }