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