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