[chromium] WebCore::Prerender::didStartPrerender depends on LinkLoader
[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 "CachedResourceRequest.h"
39 #include "ContainerNode.h"
40 #include "DNS.h"
41 #include "Document.h"
42 #include "Frame.h"
43 #include "FrameView.h"
44 #include "LinkRelAttribute.h"
45 #include "Settings.h"
46 #include "StyleResolver.h"
47
48 #if ENABLE(LINK_PRERENDER)
49 #include "PrerenderHandle.h"
50 #include "Prerenderer.h"
51 #endif
52
53 namespace WebCore {
54
55 LinkLoader::LinkLoader(LinkLoaderClient* client)
56     : m_client(client)
57     , m_linkLoadTimer(this, &LinkLoader::linkLoadTimerFired)
58     , m_linkLoadingErrorTimer(this, &LinkLoader::linkLoadingErrorTimerFired)
59 {
60 }
61
62 LinkLoader::~LinkLoader()
63 {
64     if (m_cachedLinkResource)
65         m_cachedLinkResource->removeClient(this);
66 #if ENABLE(LINK_PRERENDER)
67     if (m_prerenderHandle)
68         m_prerenderHandle->removeClient();
69 #endif
70 }
71
72 void LinkLoader::linkLoadTimerFired(Timer<LinkLoader>* timer)
73 {
74     ASSERT_UNUSED(timer, timer == &m_linkLoadTimer);
75     m_client->linkLoaded();
76 }
77
78 void LinkLoader::linkLoadingErrorTimerFired(Timer<LinkLoader>* timer)
79 {
80     ASSERT_UNUSED(timer, timer == &m_linkLoadingErrorTimer);
81     m_client->linkLoadingErrored();
82 }
83
84 void LinkLoader::notifyFinished(CachedResource* resource)
85 {
86     ASSERT_UNUSED(resource, m_cachedLinkResource.get() == resource);
87
88     if (m_cachedLinkResource->errorOccurred())
89         m_linkLoadingErrorTimer.startOneShot(0);
90     else 
91         m_linkLoadTimer.startOneShot(0);
92
93     m_cachedLinkResource->removeClient(this);
94     m_cachedLinkResource = 0;
95 }
96
97 #if ENABLE(LINK_PRERENDER)
98
99 void LinkLoader::didStartPrerender()
100 {
101     m_client->didStartLinkPrerender();
102 }
103
104 void LinkLoader::didStopPrerender()
105 {
106     m_client->didStopLinkPrerender();
107 }
108
109 void LinkLoader::didSendLoadForPrerender()
110 {
111     m_client->didSendLoadForLinkPrerender();
112 }
113
114 void LinkLoader::didSendDOMContentLoadedForPrerender()
115 {
116     m_client->didSendDOMContentLoadedForLinkPrerender();
117 }
118
119 #endif
120
121 bool LinkLoader::loadLink(const LinkRelAttribute& relAttribute, const String& type,
122                           const String& sizes, const KURL& href, Document* document)
123 {
124     // We'll record this URL per document, even if we later only use it in top level frames
125     if (relAttribute.m_iconType != InvalidIcon && href.isValid() && !href.isEmpty()) {
126         if (!m_client->shouldLoadLink()) 
127             return false;
128         document->addIconURL(href.string(), type, sizes, relAttribute.m_iconType);
129     }
130
131     if (relAttribute.m_isDNSPrefetch) {
132         Settings* settings = document->settings();
133         // FIXME: The href attribute of the link element can be in "//hostname" form, and we shouldn't attempt
134         // to complete that as URL <https://bugs.webkit.org/show_bug.cgi?id=48857>.
135         if (settings && settings->dnsPrefetchingEnabled() && href.isValid() && !href.isEmpty())
136             prefetchDNS(href.host());
137     }
138
139 #if ENABLE(LINK_PREFETCH)
140     if ((relAttribute.m_isLinkPrefetch || relAttribute.m_isLinkSubresource) && href.isValid() && document->frame()) {
141         if (!m_client->shouldLoadLink())
142             return false;
143         ResourceLoadPriority priority = ResourceLoadPriorityUnresolved;
144         CachedResource::Type type = CachedResource::LinkPrefetch;
145         // We only make one request to the cachedresourcelodaer if multiple rel types are
146         // specified, 
147         if (relAttribute.m_isLinkSubresource) {
148             priority = ResourceLoadPriorityLow;
149             type = CachedResource::LinkSubresource;
150         }
151         CachedResourceRequest linkRequest(ResourceRequest(document->completeURL(href)), priority);
152         
153         if (m_cachedLinkResource) {
154             m_cachedLinkResource->removeClient(this);
155             m_cachedLinkResource = 0;
156         }
157         m_cachedLinkResource = document->cachedResourceLoader()->requestLinkResource(type, linkRequest);
158         if (m_cachedLinkResource)
159             m_cachedLinkResource->addClient(this);
160     }
161 #endif
162
163 #if ENABLE(LINK_PRERENDER)
164     if (relAttribute.m_isLinkPrerender) {
165         ASSERT(!m_prerenderHandle);
166         m_prerenderHandle = document->prerenderer()->render(this, href);
167     }
168 #endif
169     return true;
170 }
171
172 void LinkLoader::released()
173 {
174     // Only prerenders need treatment here; other links either use the CachedResource interface, or are notionally
175     // atomic (dns prefetch).
176 #if ENABLE(LINK_PRERENDER)
177     if (m_prerenderHandle) {
178         m_prerenderHandle->cancel();
179         m_prerenderHandle->removeClient();
180         m_prerenderHandle.clear();
181     }
182 #endif
183 }
184
185 }