Add Prerenderer, PrerenderHandle and a chromium interface for Prerendering.
[WebKit-https.git] / Source / WebCore / loader / LinkLoader.cpp
1 /*
2  * Copyright (C) 2011 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  *
30  */
31
32 #include "config.h"
33 #include "LinkLoader.h"
34
35 #include "CSSStyleSheet.h"
36 #include "CachedCSSStyleSheet.h"
37 #include "CachedResourceLoader.h"
38 #include "ContainerNode.h"
39 #include "DNS.h"
40 #include "Document.h"
41 #include "Frame.h"
42 #include "FrameView.h"
43 #include "LinkRelAttribute.h"
44 #include "Settings.h"
45 #include "StyleResolver.h"
46
47 #if ENABLE(LINK_PRERENDER)
48 #include "PrerenderHandle.h"
49 #include "Prerenderer.h"
50 #endif
51
52 namespace WebCore {
53
54 LinkLoader::LinkLoader(LinkLoaderClient* client)
55     : m_client(client)
56     , m_linkLoadTimer(this, &LinkLoader::linkLoadTimerFired)
57     , m_linkLoadingErrorTimer(this, &LinkLoader::linkLoadingErrorTimerFired)
58 {
59 }
60
61 LinkLoader::~LinkLoader()
62 {
63     if (m_cachedLinkResource)
64         m_cachedLinkResource->removeClient(this);
65 }
66
67 void LinkLoader::linkLoadTimerFired(Timer<LinkLoader>* timer)
68 {
69     ASSERT_UNUSED(timer, timer == &m_linkLoadTimer);
70     m_client->linkLoaded();
71 }
72
73 void LinkLoader::linkLoadingErrorTimerFired(Timer<LinkLoader>* timer)
74 {
75     ASSERT_UNUSED(timer, timer == &m_linkLoadingErrorTimer);
76     m_client->linkLoadingErrored();
77 }
78
79 void LinkLoader::notifyFinished(CachedResource* resource)
80 {
81     ASSERT_UNUSED(resource, m_cachedLinkResource.get() == resource);
82
83     if (m_cachedLinkResource->errorOccurred())
84         m_linkLoadingErrorTimer.startOneShot(0);
85     else 
86         m_linkLoadTimer.startOneShot(0);
87
88     m_cachedLinkResource->removeClient(this);
89     m_cachedLinkResource = 0;
90 }
91
92 bool LinkLoader::loadLink(const LinkRelAttribute& relAttribute, const String& type,
93                           const String& sizes, const KURL& href, Document* document)
94 {
95     // We'll record this URL per document, even if we later only use it in top level frames
96     if (relAttribute.m_iconType != InvalidIcon && href.isValid() && !href.isEmpty()) {
97         if (!m_client->shouldLoadLink()) 
98             return false;
99         document->addIconURL(href.string(), type, sizes, relAttribute.m_iconType);
100     }
101
102     if (relAttribute.m_isDNSPrefetch) {
103         Settings* settings = document->settings();
104         // FIXME: The href attribute of the link element can be in "//hostname" form, and we shouldn't attempt
105         // to complete that as URL <https://bugs.webkit.org/show_bug.cgi?id=48857>.
106         if (settings && settings->dnsPrefetchingEnabled() && href.isValid() && !href.isEmpty())
107             prefetchDNS(href.host());
108     }
109
110 #if ENABLE(LINK_PREFETCH)
111     if ((relAttribute.m_isLinkPrefetch || relAttribute.m_isLinkSubresource) && href.isValid() && document->frame()) {
112         if (!m_client->shouldLoadLink())
113             return false;
114         ResourceLoadPriority priority = ResourceLoadPriorityUnresolved;
115         CachedResource::Type type = CachedResource::LinkPrefetch;
116         // We only make one request to the cachedresourcelodaer if multiple rel types are
117         // specified, 
118         if (relAttribute.m_isLinkSubresource) {
119             priority = ResourceLoadPriorityLow;
120             type = CachedResource::LinkSubresource;
121         }
122         ResourceRequest linkRequest(document->completeURL(href));
123         
124         if (m_cachedLinkResource) {
125             m_cachedLinkResource->removeClient(this);
126             m_cachedLinkResource = 0;
127         }
128         m_cachedLinkResource = document->cachedResourceLoader()->requestLinkResource(type, linkRequest, priority);
129         if (m_cachedLinkResource)
130             m_cachedLinkResource->addClient(this);
131     }
132 #endif
133
134 #if ENABLE(LINK_PRERENDER)
135     if (relAttribute.m_isLinkPrerender) {
136         ASSERT(!m_prerenderHandle);
137         m_prerenderHandle = document->prerenderer()->render(href);
138     }
139 #endif
140     return true;
141 }
142
143 void LinkLoader::released()
144 {
145     // Only prerenders need treatment here; other links either use the CachedResource interface, or are notionally
146     // atomic (dns prefetch).
147 #if ENABLE(LINK_PRERENDER)
148     if (m_prerenderHandle) {
149         m_prerenderHandle->cancel();
150         m_prerenderHandle.clear();
151     }
152 #endif
153 }
154
155 }