77a2b1938fedecd4a76d5283030db50f07491697
[WebKit.git] / Source / WebKit2 / UIProcess / WebFrameProxy.cpp
1 /*
2  * Copyright (C) 2010 Apple 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
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "WebFrameProxy.h"
27
28 #include "WebCertificateInfo.h"
29 #include "WebContext.h"
30 #include "WebFormSubmissionListenerProxy.h"
31 #include "WebFramePolicyListenerProxy.h"
32 #include "WebPageProxy.h"
33 #include <WebCore/DOMImplementation.h>
34 #include <WebCore/Image.h>
35 #include <wtf/text/WTFString.h>
36
37 using namespace WebCore;
38 using namespace std;
39
40 namespace WebKit {
41
42 WebFrameProxy::WebFrameProxy(WebPageProxy* page, uint64_t frameID)
43     : m_page(page)
44     , m_parentFrame(0)
45     , m_nextSibling(0)
46     , m_previousSibling(0)
47     , m_firstChild(0)
48     , m_lastChild(0)
49     , m_loadState(LoadStateFinished)
50     , m_isFrameSet(false)
51     , m_frameID(frameID)
52 {
53     WebContext::statistics().wkFrameCount++;
54 }
55
56 WebFrameProxy::~WebFrameProxy()
57 {
58     WebContext::statistics().wkFrameCount--;
59 }
60
61 void WebFrameProxy::disconnect()
62 {
63     m_page = 0;
64     m_parentFrame = 0;
65     m_nextSibling = 0;
66     m_previousSibling = 0;
67     m_firstChild = 0;
68     m_lastChild = 0;
69
70     if (m_activeListener) {
71         m_activeListener->invalidate();
72         m_activeListener = 0;
73     }
74 }
75
76 bool WebFrameProxy::isMainFrame() const
77 {
78     if (!m_page)
79         return false;
80
81     return this == m_page->mainFrame();
82 }
83
84 bool WebFrameProxy::canProvideSource() const
85 {
86     return isDisplayingMarkupDocument();
87 }
88
89 bool WebFrameProxy::canShowMIMEType(const String& mimeType) const
90 {
91     if (!m_page)
92         return false;
93
94     if (m_page->canShowMIMEType(mimeType))
95         return true;
96
97 #if PLATFORM(MAC)
98     // On Mac, we can show PDFs in the main frame.
99     if (isMainFrame() && !mimeType.isEmpty())
100         return WebContext::pdfAndPostScriptMIMETypes().contains(mimeType);
101 #endif
102
103     return false;
104 }
105
106 bool WebFrameProxy::isDisplayingStandaloneImageDocument() const
107 {
108     return Image::supportsType(m_MIMEType);
109 }
110
111 bool WebFrameProxy::isDisplayingMarkupDocument() const
112 {
113     // FIXME: This check should be moved to somewhere in WebCore.
114     // FIXME: This returns false when displaying a web archive.
115     return m_MIMEType == "text/html" || m_MIMEType == "image/svg+xml" || DOMImplementation::isXMLMIMEType(m_MIMEType);
116 }
117
118 void WebFrameProxy::didStartProvisionalLoad(const String& url)
119 {
120     ASSERT(m_loadState == LoadStateFinished);
121     ASSERT(m_provisionalURL.isEmpty());
122     m_loadState = LoadStateProvisional;
123     m_provisionalURL = url;
124 }
125
126 void WebFrameProxy::didReceiveServerRedirectForProvisionalLoad(const String& url)
127 {
128     ASSERT(!url.isEmpty());
129     ASSERT(m_loadState == LoadStateProvisional);
130     ASSERT(!m_provisionalURL.isEmpty());
131     m_provisionalURL = url;
132 }
133
134 void WebFrameProxy::didFailProvisionalLoad()
135 {
136     ASSERT(m_loadState == LoadStateProvisional);
137     ASSERT(!m_provisionalURL.isEmpty());
138     m_loadState = LoadStateFinished;
139     m_provisionalURL = String();
140 }
141
142 void WebFrameProxy::didCommitLoad(const String& contentType, const PlatformCertificateInfo& certificateInfo)
143 {
144     ASSERT(m_loadState == LoadStateProvisional);
145     m_loadState = LoadStateCommitted;
146     m_url = m_provisionalURL;
147     m_provisionalURL = String();
148     m_title = String();
149     m_MIMEType = contentType;
150     m_isFrameSet = false;
151     m_certificateInfo = WebCertificateInfo::create(certificateInfo);
152 }
153
154 void WebFrameProxy::didFinishLoad()
155 {
156     ASSERT(m_loadState == LoadStateCommitted);
157     ASSERT(m_provisionalURL.isEmpty());
158     m_loadState = LoadStateFinished;
159 }
160
161 void WebFrameProxy::didFailLoad()
162 {
163     ASSERT(m_loadState == LoadStateCommitted);
164     ASSERT(m_provisionalURL.isEmpty());
165     ASSERT(!m_url.isEmpty());
166     m_loadState = LoadStateFinished;
167     m_title = String();
168 }
169
170 void WebFrameProxy::didSameDocumentNavigation(const String& url)
171 {
172     m_url = url;
173 }
174
175 void WebFrameProxy::didChangeTitle(const String& title)
176 {
177     m_title = title;
178 }
179
180 void WebFrameProxy::appendChild(WebFrameProxy* child)
181 {
182     ASSERT(child->page() == page());
183     ASSERT(!child->m_parentFrame);
184     ASSERT(!child->m_nextSibling);
185     ASSERT(!child->m_previousSibling);
186
187     child->m_parentFrame = this;
188
189     WebFrameProxy* oldLast = m_lastChild;
190     m_lastChild = child;
191
192     if (oldLast) {
193         ASSERT(!oldLast->m_nextSibling);
194         child->m_previousSibling = oldLast;
195         oldLast->m_nextSibling = child;
196     } else
197         m_firstChild = child;
198 }
199
200 void WebFrameProxy::removeChild(WebFrameProxy* child)
201 {
202     child->m_parentFrame = 0;
203
204     WebFrameProxy*& newLocationForNext = m_firstChild == child ? m_firstChild : child->m_previousSibling->m_nextSibling;
205     WebFrameProxy*& newLocationForPrevious = m_lastChild == child ? m_lastChild : child->m_nextSibling->m_previousSibling;
206     swap(newLocationForNext, child->m_nextSibling);
207     swap(newLocationForPrevious, child->m_previousSibling);
208     child->m_previousSibling = 0;
209     child->m_nextSibling = 0;
210 }
211
212 bool WebFrameProxy::isDescendantOf(const WebFrameProxy* ancestor) const
213 {
214     if (!ancestor)
215         return false;
216
217     if (m_page != ancestor->m_page)
218         return false;
219
220     for (const WebFrameProxy* frame = this; frame; frame = frame->m_parentFrame) {
221         if (frame == ancestor)
222             return true;
223     }
224
225     return false;
226 }
227
228 void WebFrameProxy::dumpFrameTreeToSTDOUT(unsigned indent)
229 {
230     if (!indent && m_parentFrame)
231         printf("NOTE: Printing subtree.\n");
232
233     for (unsigned i = 0; i < indent; ++i)
234         printf(" ");
235     printf("| FRAME %d %s\n", (int)m_frameID, m_url.utf8().data());
236
237     for (WebFrameProxy* child = m_firstChild; child; child = child->m_nextSibling)
238         child->dumpFrameTreeToSTDOUT(indent + 4);
239 }
240
241 void WebFrameProxy::didRemoveFromHierarchy()
242 {
243     if (m_parentFrame)
244         m_parentFrame->removeChild(this);
245 }
246
247 PassRefPtr<ImmutableArray> WebFrameProxy::childFrames()
248 {
249     if (!m_firstChild)
250         return ImmutableArray::create();
251
252     Vector<RefPtr<APIObject> > vector;
253     for (WebFrameProxy* child = m_firstChild; child; child = child->m_nextSibling)
254         vector.append(child);
255
256     return ImmutableArray::adopt(vector);
257 }
258
259 void WebFrameProxy::receivedPolicyDecision(WebCore::PolicyAction action, uint64_t listenerID)
260 {
261     if (!m_page)
262         return;
263
264     ASSERT(m_activeListener);
265     ASSERT(m_activeListener->listenerID() == listenerID);
266     m_page->receivedPolicyDecision(action, this, listenerID);
267 }
268
269 WebFramePolicyListenerProxy* WebFrameProxy::setUpPolicyListenerProxy(uint64_t listenerID)
270 {
271     if (m_activeListener)
272         m_activeListener->invalidate();
273     m_activeListener = WebFramePolicyListenerProxy::create(this, listenerID);
274     return static_cast<WebFramePolicyListenerProxy*>(m_activeListener.get());
275 }
276
277 WebFormSubmissionListenerProxy* WebFrameProxy::setUpFormSubmissionListenerProxy(uint64_t listenerID)
278 {
279     if (m_activeListener)
280         m_activeListener->invalidate();
281     m_activeListener = WebFormSubmissionListenerProxy::create(this, listenerID);
282     return static_cast<WebFormSubmissionListenerProxy*>(m_activeListener.get());
283 }
284
285 void WebFrameProxy::getWebArchive(PassRefPtr<DataCallback> callback)
286 {
287     if (!m_page) {
288         callback->invalidate();
289         return;
290     }
291
292     m_page->getWebArchiveOfFrame(this, callback);
293 }
294
295 void WebFrameProxy::getMainResourceData(PassRefPtr<DataCallback> callback)
296 {
297     if (!m_page) {
298         callback->invalidate();
299         return;
300     }
301
302     m_page->getMainResourceDataOfFrame(this, callback);
303 }
304
305 void WebFrameProxy::getResourceData(WebURL* resourceURL, PassRefPtr<DataCallback> callback)
306 {
307     if (!m_page) {
308         callback->invalidate();
309         return;
310     }
311
312     m_page->getResourceDataFromFrame(this, resourceURL, callback);
313 }
314
315 } // namespace WebKit