2011-02-05 Patrick Gansterer <paroga@webkit.org>
[WebKit-https.git] / Source / WebCore / html / HTMLLinkElement.cpp
1 /*
2  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
3  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
4  *           (C) 2001 Dirk Mueller (mueller@kde.org)
5  * Copyright (C) 2003, 2006, 2007, 2008, 2009, 2010 Apple Inc. All rights reserved.
6  * Copyright (C) 2009 Rob Buis (rwlbuis@gmail.com)
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
24 #include "config.h"
25 #include "HTMLLinkElement.h"
26
27 #include "Attribute.h"
28 #include "CachedCSSStyleSheet.h"
29 #include "CachedResourceLoader.h"
30 #include "CSSStyleSelector.h"
31 #include "Document.h"
32 #include "Frame.h"
33 #include "FrameLoader.h"
34 #include "FrameLoaderClient.h"
35 #include "FrameTree.h"
36 #include "FrameView.h"
37 #include "HTMLNames.h"
38 #include "HTMLParserIdioms.h"
39 #include "MediaList.h"
40 #include "MediaQueryEvaluator.h"
41 #include "Page.h"
42 #include "ResourceHandle.h"
43 #include "ScriptEventListener.h"
44 #include "Settings.h"
45 #include <wtf/StdLibExtras.h>
46
47 namespace WebCore {
48
49 using namespace HTMLNames;
50
51 inline HTMLLinkElement::HTMLLinkElement(const QualifiedName& tagName, Document* document, bool createdByParser)
52     : HTMLElement(tagName, document)
53 #if ENABLE(LINK_PREFETCH)
54     , m_onloadTimer(this, &HTMLLinkElement::onloadTimerFired)
55 #endif
56     , m_disabledState(Unset)
57     , m_loading(false)
58     , m_createdByParser(createdByParser)
59     , m_pendingSheetType(None)
60 {
61     ASSERT(hasTagName(linkTag));
62 }
63
64 PassRefPtr<HTMLLinkElement> HTMLLinkElement::create(const QualifiedName& tagName, Document* document, bool createdByParser)
65 {
66     return adoptRef(new HTMLLinkElement(tagName, document, createdByParser));
67 }
68
69 HTMLLinkElement::~HTMLLinkElement()
70 {
71     if (m_sheet)
72         m_sheet->clearOwnerNode();
73
74     if (m_cachedSheet) {
75         m_cachedSheet->removeClient(this);    
76         removePendingSheet();
77     }
78     
79 #if ENABLE(LINK_PREFETCH)
80     if (m_cachedLinkPrefetch)
81         m_cachedLinkPrefetch->removeClient(this);
82 #endif
83 }
84
85 void HTMLLinkElement::setDisabledState(bool _disabled)
86 {
87     DisabledState oldDisabledState = m_disabledState;
88     m_disabledState = _disabled ? Disabled : EnabledViaScript;
89     if (oldDisabledState != m_disabledState) {
90         // If we change the disabled state while the sheet is still loading, then we have to
91         // perform three checks:
92         if (isLoading()) {
93             // Check #1: The sheet becomes disabled while loading.
94             if (m_disabledState == Disabled)
95                 removePendingSheet();
96
97             // Check #2: An alternate sheet becomes enabled while it is still loading.
98             if (m_relAttribute.m_isAlternate && m_disabledState == EnabledViaScript)
99                 addPendingSheet(Blocking);
100
101             // Check #3: A main sheet becomes enabled while it was still loading and
102             // after it was disabled via script.  It takes really terrible code to make this
103             // happen (a double toggle for no reason essentially).  This happens on
104             // virtualplastic.net, which manages to do about 12 enable/disables on only 3
105             // sheets. :)
106             if (!m_relAttribute.m_isAlternate && m_disabledState == EnabledViaScript && oldDisabledState == Disabled)
107                 addPendingSheet(Blocking);
108
109             // If the sheet is already loading just bail.
110             return;
111         }
112
113         // Load the sheet, since it's never been loaded before.
114         if (!m_sheet && m_disabledState == EnabledViaScript)
115             process();
116         else
117             document()->styleSelectorChanged(DeferRecalcStyle); // Update the style selector.
118     }
119 }
120
121 StyleSheet* HTMLLinkElement::sheet() const
122 {
123     return m_sheet.get();
124 }
125
126 void HTMLLinkElement::parseMappedAttribute(Attribute* attr)
127 {
128     if (attr->name() == relAttr) {
129         tokenizeRelAttribute(attr->value(), m_relAttribute);
130         process();
131     } else if (attr->name() == hrefAttr) {
132         m_url = document()->completeURL(stripLeadingAndTrailingHTMLSpaces(attr->value()));
133         process();
134     } else if (attr->name() == typeAttr) {
135         m_type = attr->value();
136         process();
137     } else if (attr->name() == mediaAttr) {
138         m_media = attr->value().string().lower();
139         process();
140     } else if (attr->name() == disabledAttr)
141         setDisabledState(!attr->isNull());
142     else if (attr->name() == onbeforeloadAttr)
143         setAttributeEventListener(eventNames().beforeloadEvent, createAttributeEventListener(this, attr));
144 #if ENABLE(LINK_PREFETCH)
145     else if (attr->name() == onloadAttr)
146         setAttributeEventListener(eventNames().loadEvent, createAttributeEventListener(this, attr));
147 #endif
148     else {
149         if (attr->name() == titleAttr && m_sheet)
150             m_sheet->setTitle(attr->value());
151         HTMLElement::parseMappedAttribute(attr);
152     }
153 }
154
155 void HTMLLinkElement::tokenizeRelAttribute(const AtomicString& rel, RelAttribute& relAttribute)
156 {
157     relAttribute.m_isStyleSheet = false;
158     relAttribute.m_isIcon = false;
159     relAttribute.m_isAlternate = false;
160     relAttribute.m_isDNSPrefetch = false;
161 #if ENABLE(LINK_PREFETCH)
162     relAttribute.m_isLinkPrefetch = false;
163 #endif
164     if (equalIgnoringCase(rel, "stylesheet"))
165         relAttribute.m_isStyleSheet = true;
166     else if (equalIgnoringCase(rel, "icon") || equalIgnoringCase(rel, "shortcut icon"))
167         relAttribute.m_isIcon = true;
168     else if (equalIgnoringCase(rel, "dns-prefetch"))
169         relAttribute.m_isDNSPrefetch = true;
170 #if ENABLE(LINK_PREFETCH)
171     else if (equalIgnoringCase(rel, "prefetch"))
172         relAttribute.m_isLinkPrefetch = true;
173 #endif
174     else if (equalIgnoringCase(rel, "alternate stylesheet") || equalIgnoringCase(rel, "stylesheet alternate")) {
175         relAttribute.m_isStyleSheet = true;
176         relAttribute.m_isAlternate = true;
177     } else {
178         // Tokenize the rel attribute and set bits based on specific keywords that we find.
179         String relString = rel.string();
180         relString.replace('\n', ' ');
181         Vector<String> list;
182         relString.split(' ', list);
183         Vector<String>::const_iterator end = list.end();
184         for (Vector<String>::const_iterator it = list.begin(); it != end; ++it) {
185             if (equalIgnoringCase(*it, "stylesheet"))
186                 relAttribute.m_isStyleSheet = true;
187             else if (equalIgnoringCase(*it, "alternate"))
188                 relAttribute.m_isAlternate = true;
189             else if (equalIgnoringCase(*it, "icon"))
190                 relAttribute.m_isIcon = true;
191         }
192     }
193 }
194
195 void HTMLLinkElement::process()
196 {
197     if (!inDocument()) {
198         ASSERT(!m_sheet);
199         return;
200     }
201
202     String type = m_type.lower();
203
204     // IE extension: location of small icon for locationbar / bookmarks
205     // We'll record this URL per document, even if we later only use it in top level frames
206     if (m_relAttribute.m_isIcon && m_url.isValid() && !m_url.isEmpty())
207         document()->setIconURL(m_url.string(), type);
208
209     if (m_relAttribute.m_isDNSPrefetch) {
210         Settings* settings = document()->settings();
211         // FIXME: The href attribute of the link element can be in "//hostname" form, and we shouldn't attempt
212         // to complete that as URL <https://bugs.webkit.org/show_bug.cgi?id=48857>.
213         if (settings && settings->dnsPrefetchingEnabled() && m_url.isValid() && !m_url.isEmpty())
214             ResourceHandle::prepareForURL(m_url);
215     }
216
217 #if ENABLE(LINK_PREFETCH)
218     if (m_relAttribute.m_isLinkPrefetch && m_url.isValid() && document()->frame()) {
219         m_cachedLinkPrefetch = document()->cachedResourceLoader()->requestLinkPrefetch(m_url);
220         if (m_cachedLinkPrefetch)
221             m_cachedLinkPrefetch->addClient(this);
222     }
223 #endif
224
225     bool acceptIfTypeContainsTextCSS = document()->page() && document()->page()->settings() && document()->page()->settings()->treatsAnyTextCSSLinkAsStylesheet();
226     
227     if (m_disabledState != Disabled && (m_relAttribute.m_isStyleSheet || (acceptIfTypeContainsTextCSS && type.contains("text/css"))) 
228         && document()->frame() && m_url.isValid()) {
229         
230         String charset = getAttribute(charsetAttr);
231         if (charset.isEmpty() && document()->frame())
232             charset = document()->charset();
233         
234         if (m_cachedSheet) {
235             removePendingSheet();
236             m_cachedSheet->removeClient(this);
237             m_cachedSheet = 0;
238         }
239
240         RefPtr<Document> originalDocument = document();
241         if (!dispatchBeforeLoadEvent(m_url))
242             return;
243         // A beforeload handler might have removed us from the document or changed the document.
244         if (!inDocument() || document() != originalDocument)
245             return;
246
247         m_loading = true;
248
249         bool mediaQueryMatches = true;
250         if (!m_media.isEmpty()) {
251             RefPtr<RenderStyle> documentStyle = CSSStyleSelector::styleForDocument(document());
252             RefPtr<MediaList> media = MediaList::createAllowingDescriptionSyntax(m_media);
253             MediaQueryEvaluator evaluator(document()->frame()->view()->mediaType(), document()->frame(), documentStyle.get());
254             mediaQueryMatches = evaluator.eval(media.get());
255         }
256
257         // Don't hold up render tree construction and script execution on stylesheets
258         // that are not needed for the rendering at the moment.
259         bool blocking = mediaQueryMatches && !isAlternate();
260         addPendingSheet(blocking ? Blocking : NonBlocking);
261
262         // Load stylesheets that are not needed for the rendering immediately with low priority.
263         ResourceLoadPriority priority = blocking ? ResourceLoadPriorityUnresolved : ResourceLoadPriorityVeryLow;
264         m_cachedSheet = document()->cachedResourceLoader()->requestCSSStyleSheet(m_url, charset, priority);
265         
266         if (m_cachedSheet)
267             m_cachedSheet->addClient(this);
268         else {
269             // The request may have been denied if (for example) the stylesheet is local and the document is remote.
270             m_loading = false;
271             removePendingSheet();
272         }
273     } else if (m_sheet) {
274         // we no longer contain a stylesheet, e.g. perhaps rel or type was changed
275         m_sheet = 0;
276         document()->styleSelectorChanged(DeferRecalcStyle);
277     }
278 }
279
280 void HTMLLinkElement::insertedIntoDocument()
281 {
282     HTMLElement::insertedIntoDocument();
283     document()->addStyleSheetCandidateNode(this, m_createdByParser);
284
285     process();
286 }
287
288 void HTMLLinkElement::removedFromDocument()
289 {
290     HTMLElement::removedFromDocument();
291
292     document()->removeStyleSheetCandidateNode(this);
293
294     if (m_sheet) {
295         ASSERT(m_sheet->ownerNode() == this);
296         m_sheet->clearOwnerNode();
297         m_sheet = 0;
298     }
299
300     if (document()->renderer())
301         document()->styleSelectorChanged(DeferRecalcStyle);
302 }
303
304 void HTMLLinkElement::finishParsingChildren()
305 {
306     m_createdByParser = false;
307     HTMLElement::finishParsingChildren();
308 }
309
310 void HTMLLinkElement::setCSSStyleSheet(const String& href, const KURL& baseURL, const String& charset, const CachedCSSStyleSheet* sheet)
311 {
312     if (!inDocument()) {
313         ASSERT(!m_sheet);
314         return;
315     }
316
317     m_sheet = CSSStyleSheet::create(this, href, baseURL, charset);
318
319     bool strictParsing = !document()->inQuirksMode();
320     bool enforceMIMEType = strictParsing;
321     bool crossOriginCSS = false;
322     bool validMIMEType = false;
323     bool needsSiteSpecificQuirks = document()->page() && document()->page()->settings()->needsSiteSpecificQuirks();
324
325     // Check to see if we should enforce the MIME type of the CSS resource in strict mode.
326     // Running in iWeb 2 is one example of where we don't want to - <rdar://problem/6099748>
327     if (enforceMIMEType && document()->page() && !document()->page()->settings()->enforceCSSMIMETypeInNoQuirksMode())
328         enforceMIMEType = false;
329
330 #if defined(BUILDING_ON_TIGER) || defined(BUILDING_ON_LEOPARD)
331     if (enforceMIMEType && needsSiteSpecificQuirks) {
332         // Covers both http and https, with or without "www."
333         if (baseURL.string().contains("mcafee.com/japan/", false))
334             enforceMIMEType = false;
335     }
336 #endif
337
338     String sheetText = sheet->sheetText(enforceMIMEType, &validMIMEType);
339     m_sheet->parseString(sheetText, strictParsing);
340
341     // If we're loading a stylesheet cross-origin, and the MIME type is not
342     // standard, require the CSS to at least start with a syntactically
343     // valid CSS rule.
344     // This prevents an attacker playing games by injecting CSS strings into
345     // HTML, XML, JSON, etc. etc.
346     if (!document()->securityOrigin()->canRequest(baseURL))
347         crossOriginCSS = true;
348
349     if (crossOriginCSS && !validMIMEType && !m_sheet->hasSyntacticallyValidCSSHeader())
350         m_sheet = CSSStyleSheet::create(this, href, baseURL, charset);
351
352     if (strictParsing && needsSiteSpecificQuirks) {
353         // Work around <https://bugs.webkit.org/show_bug.cgi?id=28350>.
354         DEFINE_STATIC_LOCAL(const String, slashKHTMLFixesDotCss, ("/KHTMLFixes.css"));
355         DEFINE_STATIC_LOCAL(const String, mediaWikiKHTMLFixesStyleSheet, ("/* KHTML fix stylesheet */\n/* work around the horizontal scrollbars */\n#column-content { margin-left: 0; }\n\n"));
356         // There are two variants of KHTMLFixes.css. One is equal to mediaWikiKHTMLFixesStyleSheet,
357         // while the other lacks the second trailing newline.
358         if (baseURL.string().endsWith(slashKHTMLFixesDotCss) && !sheetText.isNull() && mediaWikiKHTMLFixesStyleSheet.startsWith(sheetText)
359                 && sheetText.length() >= mediaWikiKHTMLFixesStyleSheet.length() - 1) {
360             ASSERT(m_sheet->length() == 1);
361             ExceptionCode ec;
362             m_sheet->deleteRule(0, ec);
363         }
364     }
365
366     m_sheet->setTitle(title());
367
368     RefPtr<MediaList> media = MediaList::createAllowingDescriptionSyntax(m_media);
369     m_sheet->setMedia(media.get());
370
371     m_loading = false;
372     m_sheet->checkLoaded();
373 }
374
375 bool HTMLLinkElement::isLoading() const
376 {
377     if (m_loading)
378         return true;
379     if (!m_sheet)
380         return false;
381     return static_cast<CSSStyleSheet *>(m_sheet.get())->isLoading();
382 }
383
384 #if ENABLE(LINK_PREFETCH)
385 void HTMLLinkElement::onloadTimerFired(Timer<HTMLLinkElement>* timer)
386 {
387     ASSERT_UNUSED(timer, timer == &m_onloadTimer);
388     dispatchEvent(Event::create(eventNames().loadEvent, false, false));
389 }
390
391 void HTMLLinkElement::notifyFinished(CachedResource* resource)
392 {
393     m_onloadTimer.startOneShot(0);
394     if (m_cachedLinkPrefetch.get() == resource) {
395         m_cachedLinkPrefetch->removeClient(this);
396         m_cachedLinkPrefetch = 0;
397     }
398 }
399 #endif
400
401 bool HTMLLinkElement::sheetLoaded()
402 {
403     if (!isLoading()) {
404         removePendingSheet();
405         return true;
406     }
407     return false;
408 }
409
410 bool HTMLLinkElement::isURLAttribute(Attribute *attr) const
411 {
412     return attr->name() == hrefAttr;
413 }
414
415 KURL HTMLLinkElement::href() const
416 {
417     return document()->completeURL(getAttribute(hrefAttr));
418 }
419
420 String HTMLLinkElement::rel() const
421 {
422     return getAttribute(relAttr);
423 }
424
425 String HTMLLinkElement::target() const
426 {
427     return getAttribute(targetAttr);
428 }
429
430 String HTMLLinkElement::type() const
431 {
432     return getAttribute(typeAttr);
433 }
434
435 void HTMLLinkElement::addSubresourceAttributeURLs(ListHashSet<KURL>& urls) const
436 {
437     HTMLElement::addSubresourceAttributeURLs(urls);
438
439     // Favicons are handled by a special case in LegacyWebArchive::create()
440     if (m_relAttribute.m_isIcon)
441         return;
442
443     if (!m_relAttribute.m_isStyleSheet)
444         return;
445     
446     // Append the URL of this link element.
447     addSubresourceURL(urls, href());
448     
449     // Walk the URLs linked by the linked-to stylesheet.
450     if (StyleSheet* styleSheet = const_cast<HTMLLinkElement*>(this)->sheet())
451         styleSheet->addSubresourceStyleURLs(urls);
452 }
453
454 void HTMLLinkElement::addPendingSheet(PendingSheetType type)
455 {
456     if (type <= m_pendingSheetType)
457         return;
458     m_pendingSheetType = type;
459
460     if (m_pendingSheetType == NonBlocking)
461         return;
462     document()->addPendingSheet();
463 }
464
465 void HTMLLinkElement::removePendingSheet()
466 {
467     PendingSheetType type = m_pendingSheetType;
468     m_pendingSheetType = None;
469
470     if (type == None)
471         return;
472     if (type == NonBlocking) {
473         // Document::removePendingSheet() triggers the style selector recalc for blocking sheets.
474         document()->styleSelectorChanged(RecalcStyleImmediately);
475         return;
476     }
477     document()->removePendingSheet();
478 }
479
480 }