21b27a5108550752934faafd9dc2915e48c43bbf
[WebKit-https.git] / WebKit / mac / WebCoreSupport / WebFrameLoaderClient.mm
1 /*
2  * Copyright (C) 2006, 2007, 2008 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  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer. 
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution. 
13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission. 
16  *
17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #import "WebFrameLoaderClient.h"
30
31 // Terrible hack; lets us get at the WebFrame private structure.
32 #define private public
33 #import "WebFrame.h"
34 #undef private
35
36 #import "DOMElementInternal.h"
37 #import "WebBackForwardList.h"
38 #import "WebCachedPagePlatformData.h"
39 #import "WebChromeClient.h"
40 #import "WebDataSourceInternal.h"
41 #import "WebDocumentInternal.h"
42 #import "WebDocumentLoaderMac.h"
43 #import "WebDownloadInternal.h"
44 #import "WebDynamicScrollBarsView.h"
45 #import "WebElementDictionary.h"
46 #import "WebFormDelegate.h"
47 #import "WebFrameBridge.h"
48 #import "WebFrameInternal.h"
49 #import "WebFrameLoadDelegate.h"
50 #import "WebFrameViewInternal.h"
51 #import "WebHTMLRepresentationPrivate.h"
52 #import "WebHTMLViewInternal.h"
53 #import "WebHistoryItemInternal.h"
54 #import "WebHistoryInternal.h"
55 #import "WebIconDatabaseInternal.h"
56 #import "WebKitErrorsPrivate.h"
57 #import "WebKitLogging.h"
58 #import "WebKitNSStringExtras.h"
59 #import "WebNSURLExtras.h"
60 #import "WebNetscapePluginEmbeddedView.h"
61 #import "WebNetscapePluginPackage.h"
62 #import "WebNullPluginView.h"
63 #import "WebPanelAuthenticationHandler.h"
64 #import "WebPluginController.h"
65 #import "WebPluginPackage.h"
66 #import "WebPluginViewFactoryPrivate.h"
67 #import "WebPolicyDelegate.h"
68 #import "WebPolicyDelegatePrivate.h"
69 #import "WebPreferences.h"
70 #import "WebResourceLoadDelegate.h"
71 #import "WebResourcePrivate.h"
72 #import "WebScriptDebugServerPrivate.h"
73 #import "WebUIDelegate.h"
74 #import "WebUIDelegatePrivate.h"
75 #import "WebViewInternal.h"
76 #import <WebKitSystemInterface.h>
77 #import <WebCore/AuthenticationMac.h>
78 #import <WebCore/BlockExceptions.h>
79 #import <WebCore/CachedPage.h>
80 #import <WebCore/Chrome.h>
81 #import <WebCore/Document.h>
82 #import <WebCore/DocumentLoader.h>
83 #import <WebCore/EventHandler.h>
84 #import <WebCore/FormState.h>
85 #import <WebCore/Frame.h>
86 #import <WebCore/FrameLoader.h>
87 #import <WebCore/FrameLoaderTypes.h>
88 #import <WebCore/FrameTree.h>
89 #import <WebCore/FrameView.h>
90 #import <WebCore/HTMLFormElement.h>
91 #import <WebCore/HistoryItem.h>
92 #import <WebCore/HitTestResult.h>
93 #import <WebCore/IconDatabase.h>
94 #import <WebCore/LoaderNSURLExtras.h>
95 #import <WebCore/MIMETypeRegistry.h>
96 #import <WebCore/MouseEvent.h>
97 #import <WebCore/Page.h>
98 #import <WebCore/PlatformString.h>
99 #import <WebCore/ResourceError.h>
100 #import <WebCore/ResourceHandle.h>
101 #import <WebCore/ResourceLoader.h>
102 #import <WebCore/ResourceRequest.h>
103 #import <WebCore/SharedBuffer.h>
104 #import <WebCore/WebCoreFrameBridge.h>
105 #import <WebCore/WebCoreObjCExtras.h>
106 #import <WebCore/Widget.h>
107 #import <WebKit/DOMElement.h>
108 #import <WebKit/DOMHTMLFormElement.h>
109 #import <wtf/PassRefPtr.h>
110
111 using namespace WebCore;
112
113 // SPI for NSURLDownload
114 // Needed for <rdar://problem/5121850> 
115 @interface NSURLDownload (WebNSURLDownloadDetails)
116 - (void)_setOriginatingURL:(NSURL *)originatingURL;
117 @end
118
119 // For backwards compatibility with older WebKit plug-ins.
120 NSString *WebPluginBaseURLKey = @"WebPluginBaseURL";
121 NSString *WebPluginAttributesKey = @"WebPluginAttributes";
122 NSString *WebPluginContainerKey = @"WebPluginContainer";
123
124 @interface WebFramePolicyListener : NSObject <WebPolicyDecisionListener, WebFormSubmissionListener> {
125     Frame* m_frame;
126 }
127 - (id)initWithWebCoreFrame:(Frame*)frame;
128 - (void)invalidate;
129 @end
130
131 static inline WebDataSource *dataSource(DocumentLoader* loader)
132 {
133     return loader ? static_cast<WebDocumentLoaderMac*>(loader)->dataSource() : nil;
134 }
135
136 WebFrameLoaderClient::WebFrameLoaderClient(WebFrame *webFrame)
137     : m_webFrame(webFrame)
138     , m_policyFunction(0)
139     , m_archivedResourcesDeliveryTimer(this, &WebFrameLoaderClient::deliverArchivedResources)
140 {
141 }
142
143 void WebFrameLoaderClient::frameLoaderDestroyed()
144 {
145     delete this;
146 }
147
148 bool WebFrameLoaderClient::hasWebView() const
149 {
150     return [m_webFrame.get() webView] != nil;
151 }
152
153 bool WebFrameLoaderClient::hasFrameView() const
154 {
155     return m_webFrame->_private->webFrameView != nil;
156 }
157
158 void WebFrameLoaderClient::makeRepresentation(DocumentLoader* loader)
159 {
160     [dataSource(loader) _makeRepresentation];
161 }
162
163 bool WebFrameLoaderClient::hasHTMLView() const
164 {
165     NSView <WebDocumentView> *view = [m_webFrame->_private->webFrameView documentView];
166     return [view isKindOfClass:[WebHTMLView class]];
167 }
168
169 void WebFrameLoaderClient::forceLayout()
170 {
171     NSView <WebDocumentView> *view = [m_webFrame->_private->webFrameView documentView];
172     if ([view isKindOfClass:[WebHTMLView class]])
173         [(WebHTMLView *)view setNeedsToApplyStyles:YES];
174     [view setNeedsLayout:YES];
175     [view layout];
176 }
177
178 void WebFrameLoaderClient::forceLayoutForNonHTML()
179 {
180     WebFrameView *thisView = m_webFrame->_private->webFrameView;
181     NSView <WebDocumentView> *thisDocumentView = [thisView documentView];
182     ASSERT(thisDocumentView != nil);
183     
184     // Tell the just loaded document to layout.  This may be necessary
185     // for non-html content that needs a layout message.
186     if (!([[m_webFrame.get() _dataSource] _isDocumentHTML])) {
187         [thisDocumentView setNeedsLayout:YES];
188         [thisDocumentView layout];
189         [thisDocumentView setNeedsDisplay:YES];
190     }
191 }
192
193 void WebFrameLoaderClient::setCopiesOnScroll()
194 {
195     [[[m_webFrame->_private->webFrameView _scrollView] contentView] setCopiesOnScroll:YES];
196 }
197
198 void WebFrameLoaderClient::detachedFromParent2()
199 {
200     [m_webFrame->_private->webFrameView _setWebFrame:nil]; // needed for now to be compatible w/ old behavior
201 }
202
203 void WebFrameLoaderClient::detachedFromParent3()
204 {
205     [m_webFrame->_private->webFrameView release];
206     m_webFrame->_private->webFrameView = nil;
207 }
208
209 void WebFrameLoaderClient::detachedFromParent4()
210 {
211     m_webFrame->_private->bridge = nil;
212 }
213
214 void WebFrameLoaderClient::download(ResourceHandle* handle, const ResourceRequest& request, const ResourceRequest& initialRequest, const ResourceResponse& response)
215 {
216     id proxy = handle->releaseProxy();
217     ASSERT(proxy);
218     
219     WebView *webView = getWebView(m_webFrame.get());
220     WebDownload *download = [WebDownload _downloadWithLoadingConnection:handle->connection()
221                                                                 request:request.nsURLRequest()
222                                                                response:response.nsURLResponse()
223                                                                delegate:[webView downloadDelegate]
224                                                                   proxy:proxy];
225     
226     setOriginalURLForDownload(download, initialRequest);    
227 }
228
229 void WebFrameLoaderClient::setOriginalURLForDownload(WebDownload *download, const ResourceRequest& initialRequest) const
230 {
231     NSURLRequest *initialURLRequest = initialRequest.nsURLRequest();
232     NSURL *originalURL = nil;
233     
234     // If there was no referrer, don't traverse the back/forward history
235     // since this download was initiated directly. <rdar://problem/5294691>
236     if ([initialURLRequest valueForHTTPHeaderField:@"Referer"]) {
237         // find the first item in the history that was originated by the user
238         WebView *webView = getWebView(m_webFrame.get());
239         WebBackForwardList *history = [webView backForwardList];
240         int backListCount = [history backListCount];
241         for (int backIndex = 0; backIndex <= backListCount && !originalURL; backIndex++) {
242             // FIXME: At one point we had code here to check a "was user gesture" flag.
243             // Do we need to restore that logic?
244             originalURL = [[history itemAtIndex:-backIndex] URL];
245         }
246     }
247
248     if (!originalURL)
249         originalURL = [initialURLRequest URL];
250
251     if ([download respondsToSelector:@selector(_setOriginatingURL:)]) {
252         NSString *scheme = [originalURL scheme];
253         NSString *host = [originalURL host];
254         if (scheme && host && [scheme length] && [host length]) {
255             NSNumber *port = [originalURL port];
256             if (port && [port intValue] < 0)
257                 port = nil;
258             NSString *hostOnlyURLString;
259             if (port)
260                 hostOnlyURLString = [[NSString alloc] initWithFormat:@"%@://%@:%d", scheme, host, [port intValue]];
261             else
262                 hostOnlyURLString = [[NSString alloc] initWithFormat:@"%@://%@", scheme, host];
263             NSURL *hostOnlyURL = [[NSURL alloc] initWithString:hostOnlyURLString];
264             [hostOnlyURLString release];
265             [download _setOriginatingURL:hostOnlyURL];
266             [hostOnlyURL release];
267         }
268     }
269 }
270
271 bool WebFrameLoaderClient::dispatchDidLoadResourceFromMemoryCache(DocumentLoader* loader, const ResourceRequest& request, const ResourceResponse& response, int length)
272 {
273     WebView *webView = getWebView(m_webFrame.get());
274     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
275     if (!implementations->didLoadResourceFromMemoryCacheFunc)
276         return false;
277
278     CallResourceLoadDelegate(implementations->didLoadResourceFromMemoryCacheFunc, webView, @selector(webView:didLoadResourceFromMemoryCache:response:length:fromDataSource:), request.nsURLRequest(), response.nsURLResponse(), length, dataSource(loader));
279     return true;
280 }
281
282 void WebFrameLoaderClient::assignIdentifierToInitialRequest(unsigned long identifier, DocumentLoader* loader, const ResourceRequest& request)
283 {
284     WebView *webView = getWebView(m_webFrame.get());
285     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
286
287     id object = nil;
288     BOOL shouldRelease = NO;
289     if (implementations->identifierForRequestFunc)
290         object = CallResourceLoadDelegate(implementations->identifierForRequestFunc, webView, @selector(webView:identifierForInitialRequest:fromDataSource:), request.nsURLRequest(), dataSource(loader));
291     else {
292         object = [[NSObject alloc] init];
293         shouldRelease = YES;
294     }
295
296     [webView _addObject:object forIdentifier:identifier];
297
298     if (shouldRelease)
299         [object release];
300 }
301
302 void WebFrameLoaderClient::dispatchWillSendRequest(DocumentLoader* loader, unsigned long identifier, ResourceRequest& request, const ResourceResponse& redirectResponse)
303 {
304     WebView *webView = getWebView(m_webFrame.get());
305     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
306
307     if (redirectResponse.isNull())
308         static_cast<WebDocumentLoaderMac*>(loader)->increaseLoadCount(identifier);
309
310     if (implementations->willSendRequestFunc)
311         request = (NSURLRequest *)CallResourceLoadDelegate(implementations->willSendRequestFunc, webView, @selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:), [webView _objectForIdentifier:identifier], request.nsURLRequest(), redirectResponse.nsURLResponse(), dataSource(loader));
312 }
313
314 void WebFrameLoaderClient::dispatchDidReceiveAuthenticationChallenge(DocumentLoader* loader, unsigned long identifier, const AuthenticationChallenge& challenge)
315 {
316     WebView *webView = getWebView(m_webFrame.get());
317     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
318
319     NSURLAuthenticationChallenge *webChallenge = mac(challenge);
320
321     if (implementations->didReceiveAuthenticationChallengeFunc) {
322         if (id resource = [webView _objectForIdentifier:identifier]) {
323             CallResourceLoadDelegate(implementations->didReceiveAuthenticationChallengeFunc, webView, @selector(webView:resource:didReceiveAuthenticationChallenge:fromDataSource:), resource, webChallenge, dataSource(loader));
324             return;
325         }
326     }
327
328     NSWindow *window = [webView hostWindow] ? [webView hostWindow] : [webView window];
329     [[WebPanelAuthenticationHandler sharedHandler] startAuthentication:webChallenge window:window];
330 }
331
332 void WebFrameLoaderClient::dispatchDidCancelAuthenticationChallenge(DocumentLoader* loader, unsigned long identifier, const AuthenticationChallenge&challenge)
333 {
334     WebView *webView = getWebView(m_webFrame.get());
335     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
336     NSURLAuthenticationChallenge *webChallenge = mac(challenge);
337
338     if (implementations->didCancelAuthenticationChallengeFunc) {
339         if (id resource = [webView _objectForIdentifier:identifier]) {
340             CallResourceLoadDelegate(implementations->didCancelAuthenticationChallengeFunc, webView, @selector(webView:resource:didCancelAuthenticationChallenge:fromDataSource:), resource, webChallenge, dataSource(loader));
341             return;
342         }
343     }
344
345     [(WebPanelAuthenticationHandler *)[WebPanelAuthenticationHandler sharedHandler] cancelAuthentication:webChallenge];
346 }
347
348 void WebFrameLoaderClient::dispatchDidReceiveResponse(DocumentLoader* loader, unsigned long identifier, const ResourceResponse& response)
349 {
350     WebView *webView = getWebView(m_webFrame.get());
351     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
352     if (implementations->didReceiveResponseFunc) {
353         if (id resource = [webView _objectForIdentifier:identifier])
354             CallResourceLoadDelegate(implementations->didReceiveResponseFunc, webView, @selector(webView:resource:didReceiveResponse:fromDataSource:), resource, response.nsURLResponse(), dataSource(loader));
355     }
356 }
357
358 NSCachedURLResponse* WebFrameLoaderClient::willCacheResponse(DocumentLoader* loader, unsigned long identifier, NSCachedURLResponse* response) const
359 {
360     WebView *webView = getWebView(m_webFrame.get());
361     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
362
363     if (implementations->willCacheResponseFunc) {
364         if (id resource = [webView _objectForIdentifier:identifier])
365             return CallResourceLoadDelegate(implementations->willCacheResponseFunc, webView, @selector(webView:resource:willCacheResponse:fromDataSource:), resource, response, dataSource(loader));
366     }
367
368     return response;
369 }
370
371 void WebFrameLoaderClient::dispatchDidReceiveContentLength(DocumentLoader* loader, unsigned long identifier, int lengthReceived)
372 {
373     WebView *webView = getWebView(m_webFrame.get());
374     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
375     if (implementations->didReceiveContentLengthFunc) {
376         if (id resource = [webView _objectForIdentifier:identifier])
377             CallResourceLoadDelegate(implementations->didReceiveContentLengthFunc, webView, @selector(webView:resource:didReceiveContentLength:fromDataSource:), resource, (NSInteger)lengthReceived, dataSource(loader));
378     }
379 }
380
381 void WebFrameLoaderClient::dispatchDidFinishLoading(DocumentLoader* loader, unsigned long identifier)
382 {
383     WebView *webView = getWebView(m_webFrame.get());
384     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
385
386     if (implementations->didFinishLoadingFromDataSourceFunc) {
387         if (id resource = [webView _objectForIdentifier:identifier])
388             CallResourceLoadDelegate(implementations->didFinishLoadingFromDataSourceFunc, webView, @selector(webView:resource:didFinishLoadingFromDataSource:), resource, dataSource(loader));
389     }
390
391     [webView _removeObjectForIdentifier:identifier];
392
393     static_cast<WebDocumentLoaderMac*>(loader)->decreaseLoadCount(identifier);
394 }
395
396 void WebFrameLoaderClient::dispatchDidFailLoading(DocumentLoader* loader, unsigned long identifier, const ResourceError& error)
397 {
398     WebView *webView = getWebView(m_webFrame.get());
399     WebResourceDelegateImplementationCache* implementations = WebViewGetResourceLoadDelegateImplementations(webView);
400
401     if (implementations->didFailLoadingWithErrorFromDataSourceFunc) {
402         if (id resource = [webView _objectForIdentifier:identifier])
403             CallResourceLoadDelegate(implementations->didFailLoadingWithErrorFromDataSourceFunc, webView, @selector(webView:resource:didFailLoadingWithError:fromDataSource:), resource, (NSError *)error, dataSource(loader));
404     }
405
406     [webView _removeObjectForIdentifier:identifier];
407
408     static_cast<WebDocumentLoaderMac*>(loader)->decreaseLoadCount(identifier);
409 }
410
411 void WebFrameLoaderClient::dispatchDidHandleOnloadEvents()
412 {
413     WebView *webView = getWebView(m_webFrame.get());
414     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
415     if (implementations->didHandleOnloadEventsForFrameFunc)
416         CallFrameLoadDelegate(implementations->didHandleOnloadEventsForFrameFunc, webView, @selector(webView:didHandleOnloadEventsForFrame:), m_webFrame.get());
417 }
418
419 void WebFrameLoaderClient::dispatchDidReceiveServerRedirectForProvisionalLoad()
420 {
421     WebView *webView = getWebView(m_webFrame.get());
422     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
423     if (implementations->didReceiveServerRedirectForProvisionalLoadForFrameFunc)
424         CallFrameLoadDelegate(implementations->didReceiveServerRedirectForProvisionalLoadForFrameFunc, webView, @selector(webView:didReceiveServerRedirectForProvisionalLoadForFrame:), m_webFrame.get());
425 }
426
427 void WebFrameLoaderClient::dispatchDidCancelClientRedirect()
428 {
429     WebView *webView = getWebView(m_webFrame.get());
430     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
431     if (implementations->didCancelClientRedirectForFrameFunc)
432         CallFrameLoadDelegate(implementations->didCancelClientRedirectForFrameFunc, webView, @selector(webView:didCancelClientRedirectForFrame:), m_webFrame.get());
433 }
434
435 void WebFrameLoaderClient::dispatchWillPerformClientRedirect(const KURL& url, double delay, double fireDate)
436 {
437     WebView *webView = getWebView(m_webFrame.get());
438     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
439     if (implementations->willPerformClientRedirectToURLDelayFireDateForFrameFunc) {
440         NSURL *cocoaURL = url;
441         CallFrameLoadDelegate(implementations->willPerformClientRedirectToURLDelayFireDateForFrameFunc, webView, @selector(webView:willPerformClientRedirectToURL:delay:fireDate:forFrame:), cocoaURL, delay, [NSDate dateWithTimeIntervalSince1970:fireDate], m_webFrame.get());
442     }
443 }
444
445 void WebFrameLoaderClient::dispatchDidChangeLocationWithinPage()
446 {
447     WebView *webView = getWebView(m_webFrame.get());
448     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
449     if (implementations->didChangeLocationWithinPageForFrameFunc)
450         CallFrameLoadDelegate(implementations->didChangeLocationWithinPageForFrameFunc, webView, @selector(webView:didChangeLocationWithinPageForFrame:), m_webFrame.get());
451 }
452
453 void WebFrameLoaderClient::dispatchWillClose()
454 {
455     WebView *webView = getWebView(m_webFrame.get());   
456     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
457     if (implementations->willCloseFrameFunc)
458         CallFrameLoadDelegate(implementations->willCloseFrameFunc, webView, @selector(webView:willCloseFrame:), m_webFrame.get());
459 }
460
461 void WebFrameLoaderClient::dispatchDidReceiveIcon()
462 {
463     ASSERT([m_webFrame.get() _isMainFrame]);
464     WebView *webView = getWebView(m_webFrame.get());   
465
466     [webView _dispatchDidReceiveIconFromWebFrame:m_webFrame.get()];
467 }
468
469 void WebFrameLoaderClient::dispatchDidStartProvisionalLoad()
470 {
471     WebView *webView = getWebView(m_webFrame.get());   
472     [webView _didStartProvisionalLoadForFrame:m_webFrame.get()];
473
474     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
475     if (implementations->didStartProvisionalLoadForFrameFunc)
476         CallFrameLoadDelegate(implementations->didStartProvisionalLoadForFrameFunc, webView, @selector(webView:didStartProvisionalLoadForFrame:), m_webFrame.get());
477 }
478
479 void WebFrameLoaderClient::dispatchDidReceiveTitle(const String& title)
480 {
481     WebView *webView = getWebView(m_webFrame.get());   
482     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
483     if (implementations->didReceiveTitleForFrameFunc)
484         CallFrameLoadDelegate(implementations->didReceiveTitleForFrameFunc, webView, @selector(webView:didReceiveTitle:forFrame:), (NSString *)title, m_webFrame.get());
485 }
486
487 void WebFrameLoaderClient::dispatchDidCommitLoad()
488 {
489     // Tell the client we've committed this URL.
490     ASSERT([m_webFrame->_private->webFrameView documentView] != nil);
491     
492     WebView *webView = getWebView(m_webFrame.get());   
493     [webView _didCommitLoadForFrame:m_webFrame.get()];
494
495     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
496     if (implementations->didCommitLoadForFrameFunc)
497         CallFrameLoadDelegate(implementations->didCommitLoadForFrameFunc, webView, @selector(webView:didCommitLoadForFrame:), m_webFrame.get());
498 }
499
500 void WebFrameLoaderClient::dispatchDidFailProvisionalLoad(const ResourceError& error)
501 {
502     WebView *webView = getWebView(m_webFrame.get());   
503     [webView _didFailProvisionalLoadWithError:error forFrame:m_webFrame.get()];
504
505     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
506     if (implementations->didFailProvisionalLoadWithErrorForFrameFunc)
507         CallFrameLoadDelegate(implementations->didFailProvisionalLoadWithErrorForFrameFunc, webView, @selector(webView:didFailProvisionalLoadWithError:forFrame:), (NSError *)error, m_webFrame.get());
508
509     [m_webFrame->_private->internalLoadDelegate webFrame:m_webFrame.get() didFinishLoadWithError:error];
510 }
511
512 void WebFrameLoaderClient::dispatchDidFailLoad(const ResourceError& error)
513 {
514     WebView *webView = getWebView(m_webFrame.get());   
515     [webView _didFailLoadWithError:error forFrame:m_webFrame.get()];
516
517     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
518     if (implementations->didFailLoadWithErrorForFrameFunc)
519         CallFrameLoadDelegate(implementations->didFailLoadWithErrorForFrameFunc, webView, @selector(webView:didFailLoadWithError:forFrame:), (NSError *)error, m_webFrame.get());
520
521     [m_webFrame->_private->internalLoadDelegate webFrame:m_webFrame.get() didFinishLoadWithError:error];
522 }
523
524 void WebFrameLoaderClient::dispatchDidFinishDocumentLoad()
525 {
526     WebView *webView = getWebView(m_webFrame.get());
527     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
528     if (implementations->didFinishDocumentLoadForFrameFunc)
529         CallFrameLoadDelegate(implementations->didFinishDocumentLoadForFrameFunc, webView, @selector(webView:didFinishDocumentLoadForFrame:), m_webFrame.get());
530 }
531
532 void WebFrameLoaderClient::dispatchDidFinishLoad()
533 {
534     WebView *webView = getWebView(m_webFrame.get());   
535     [webView _didFinishLoadForFrame:m_webFrame.get()];
536
537     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
538     if (implementations->didFinishLoadForFrameFunc)
539         CallFrameLoadDelegate(implementations->didFinishLoadForFrameFunc, webView, @selector(webView:didFinishLoadForFrame:), m_webFrame.get());
540
541     [m_webFrame->_private->internalLoadDelegate webFrame:m_webFrame.get() didFinishLoadWithError:nil];
542 }
543
544 void WebFrameLoaderClient::dispatchDidFirstLayout()
545 {
546     WebView *webView = getWebView(m_webFrame.get());
547     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
548     if (implementations->didFirstLayoutInFrameFunc)
549         CallFrameLoadDelegate(implementations->didFirstLayoutInFrameFunc, webView, @selector(webView:didFirstLayoutInFrame:), m_webFrame.get());
550 }
551
552 Frame* WebFrameLoaderClient::dispatchCreatePage()
553 {
554     WebView *currentWebView = getWebView(m_webFrame.get());
555     NSDictionary *features = [[NSDictionary alloc] init];
556     WebView *newWebView = [[currentWebView _UIDelegateForwarder] webView:currentWebView 
557                                                 createWebViewWithRequest:nil
558                                                           windowFeatures:features];
559     [features release];
560     return core([newWebView mainFrame]);
561 }
562
563 void WebFrameLoaderClient::dispatchShow()
564 {
565     WebView *webView = getWebView(m_webFrame.get());
566     [[webView _UIDelegateForwarder] webViewShow:webView];
567 }
568
569 void WebFrameLoaderClient::dispatchDecidePolicyForMIMEType(FramePolicyFunction function,
570     const String& MIMEType, const ResourceRequest& request)
571 {
572     WebView *webView = getWebView(m_webFrame.get());
573
574     [[webView _policyDelegateForwarder] webView:webView
575                         decidePolicyForMIMEType:MIMEType
576                                         request:request.nsURLRequest()
577                                           frame:m_webFrame.get()
578                                decisionListener:setUpPolicyListener(function).get()];
579 }
580
581 void WebFrameLoaderClient::dispatchDecidePolicyForNewWindowAction(FramePolicyFunction function,
582     const NavigationAction& action, const ResourceRequest& request, const String& frameName)
583 {
584     WebView *webView = getWebView(m_webFrame.get());
585     [[webView _policyDelegateForwarder] webView:webView
586             decidePolicyForNewWindowAction:actionDictionary(action)
587                                    request:request.nsURLRequest()
588                               newFrameName:frameName
589                           decisionListener:setUpPolicyListener(function).get()];
590 }
591
592 void WebFrameLoaderClient::dispatchDecidePolicyForNavigationAction(FramePolicyFunction function,
593     const NavigationAction& action, const ResourceRequest& request)
594 {
595     WebView *webView = getWebView(m_webFrame.get());
596     [[webView _policyDelegateForwarder] webView:webView
597                 decidePolicyForNavigationAction:actionDictionary(action)
598                                         request:request.nsURLRequest()
599                                           frame:m_webFrame.get()
600                                decisionListener:setUpPolicyListener(function).get()];
601 }
602
603 void WebFrameLoaderClient::cancelPolicyCheck()
604 {
605     [m_policyListener.get() invalidate];
606     m_policyListener = nil;
607     m_policyFunction = 0;
608 }
609
610 void WebFrameLoaderClient::dispatchUnableToImplementPolicy(const ResourceError& error)
611 {
612     WebView *webView = getWebView(m_webFrame.get());
613     [[webView _policyDelegateForwarder] webView:webView unableToImplementPolicyWithError:error frame:m_webFrame.get()];    
614 }
615
616 void WebFrameLoaderClient::dispatchWillSubmitForm(FramePolicyFunction function, PassRefPtr<FormState> formState)
617 {
618     id <WebFormDelegate> formDelegate = [getWebView(m_webFrame.get()) _formDelegate];
619     if (!formDelegate) {
620         (core(m_webFrame.get())->loader()->*function)(PolicyUse);
621         return;
622     }
623
624     NSMutableDictionary *dictionary = [[NSMutableDictionary alloc] initWithCapacity:formState->values().size()];
625     HashMap<String, String>::const_iterator end = formState->values().end();
626     for (HashMap<String, String>::const_iterator it = formState->values().begin(); it != end; ++it)
627         [dictionary setObject:it->second forKey:it->first];
628
629     CallFormDelegate(getWebView(m_webFrame.get()), @selector(frame:sourceFrame:willSubmitForm:withValues:submissionListener:), m_webFrame.get(), kit(formState->sourceFrame()), kit(formState->form()), dictionary, setUpPolicyListener(function).get());
630
631     [dictionary release];
632 }
633
634 void WebFrameLoaderClient::dispatchDidLoadMainResource(DocumentLoader* loader)
635 {
636     if ([WebScriptDebugServer listenerCount])
637         [[WebScriptDebugServer sharedScriptDebugServer] webView:getWebView(m_webFrame.get()) didLoadMainResourceForDataSource:dataSource(loader)];
638 }
639
640 void WebFrameLoaderClient::revertToProvisionalState(DocumentLoader* loader)
641 {
642     [dataSource(loader) _revertToProvisionalState];
643 }
644
645 void WebFrameLoaderClient::setMainDocumentError(DocumentLoader* loader, const ResourceError& error)
646 {
647     [dataSource(loader) _setMainDocumentError:error];
648 }
649
650 void WebFrameLoaderClient::clearUnarchivingState(DocumentLoader* loader)
651 {
652     [dataSource(loader) _clearUnarchivingState];
653 }
654
655 void WebFrameLoaderClient::willChangeEstimatedProgress()
656 {
657     [getWebView(m_webFrame.get()) _willChangeValueForKey:_WebEstimatedProgressKey];
658 }
659
660 void WebFrameLoaderClient::didChangeEstimatedProgress()
661 {
662     [getWebView(m_webFrame.get()) _didChangeValueForKey:_WebEstimatedProgressKey];
663 }
664
665 void WebFrameLoaderClient::postProgressStartedNotification()
666 {
667     [[NSNotificationCenter defaultCenter] postNotificationName:WebViewProgressStartedNotification object:getWebView(m_webFrame.get())];
668 }
669
670 void WebFrameLoaderClient::postProgressEstimateChangedNotification()
671 {
672     [[NSNotificationCenter defaultCenter] postNotificationName:WebViewProgressEstimateChangedNotification object:getWebView(m_webFrame.get())];
673 }
674
675 void WebFrameLoaderClient::postProgressFinishedNotification()
676 {
677     [[NSNotificationCenter defaultCenter] postNotificationName:WebViewProgressFinishedNotification object:getWebView(m_webFrame.get())];
678 }
679
680 void WebFrameLoaderClient::setMainFrameDocumentReady(bool ready)
681 {
682     [getWebView(m_webFrame.get()) setMainFrameDocumentReady:ready];
683 }
684
685 void WebFrameLoaderClient::startDownload(const ResourceRequest& request)
686 {
687     // FIXME: Should download full request.
688     WebDownload *download = [getWebView(m_webFrame.get()) _downloadURL:request.url()];
689     
690     setOriginalURLForDownload(download, request);
691 }
692
693 void WebFrameLoaderClient::willChangeTitle(DocumentLoader* loader)
694 {
695     // FIXME: Should do this only in main frame case, right?
696     [getWebView(m_webFrame.get()) _willChangeValueForKey:_WebMainFrameTitleKey];
697 }
698
699 void WebFrameLoaderClient::didChangeTitle(DocumentLoader* loader)
700 {
701     // FIXME: Should do this only in main frame case, right?
702     [getWebView(m_webFrame.get()) _didChangeValueForKey:_WebMainFrameTitleKey];
703 }
704
705 void WebFrameLoaderClient::committedLoad(DocumentLoader* loader, const char* data, int length)
706 {
707     NSData *nsData = [[NSData alloc] initWithBytesNoCopy:(void*)data length:length freeWhenDone:NO];
708     [dataSource(loader) _receivedData:nsData];
709     [nsData release];
710 }
711
712 void WebFrameLoaderClient::finishedLoading(DocumentLoader* loader)
713 {
714     [dataSource(loader) _finishedLoading];
715 }
716
717 void WebFrameLoaderClient::finalSetupForReplace(DocumentLoader* loader)
718 {
719     [dataSource(loader) _clearUnarchivingState];
720 }
721
722 void WebFrameLoaderClient::updateGlobalHistory(const KURL& url)
723 {
724     NSURL *cocoaURL = url;
725     const String& pageTitle = core(m_webFrame.get())->loader()->documentLoader()->title();
726     [[WebHistory optionalSharedHistory] _addItemForURL:cocoaURL title:pageTitle];
727 }
728  
729 bool WebFrameLoaderClient::shouldGoToHistoryItem(HistoryItem* item) const
730 {
731     WebView* view = getWebView(m_webFrame.get());
732     WebHistoryItem *webItem = kit(item);
733     
734     return [[view _policyDelegateForwarder] webView:view shouldGoToHistoryItem:webItem];
735 }
736
737 ResourceError WebFrameLoaderClient::cancelledError(const ResourceRequest& request)
738 {
739     return [NSError _webKitErrorWithDomain:NSURLErrorDomain code:NSURLErrorCancelled URL:request.url()];
740 }
741     
742 ResourceError WebFrameLoaderClient::blockedError(const ResourceRequest& request)
743 {
744     return [NSError _webKitErrorWithDomain:WebKitErrorDomain code:WebKitErrorCannotUseRestrictedPort URL:request.url()];
745 }
746
747 ResourceError WebFrameLoaderClient::cannotShowURLError(const ResourceRequest& request)
748 {
749     return [NSError _webKitErrorWithDomain:WebKitErrorDomain code:WebKitErrorCannotShowURL URL:request.url()];
750 }
751
752 ResourceError WebFrameLoaderClient::interruptForPolicyChangeError(const ResourceRequest& request)
753 {
754     return [NSError _webKitErrorWithDomain:WebKitErrorDomain code:WebKitErrorFrameLoadInterruptedByPolicyChange URL:request.url()];
755 }
756
757 ResourceError WebFrameLoaderClient::cannotShowMIMETypeError(const ResourceResponse& response)
758 {
759     return [NSError _webKitErrorWithDomain:NSURLErrorDomain code:WebKitErrorCannotShowMIMEType URL:response.url()];
760 }
761
762 ResourceError WebFrameLoaderClient::fileDoesNotExistError(const ResourceResponse& response)
763 {
764     return [NSError _webKitErrorWithDomain:NSURLErrorDomain code:NSURLErrorFileDoesNotExist URL:response.url()];    
765 }
766
767 bool WebFrameLoaderClient::shouldFallBack(const ResourceError& error)
768 {
769     // FIXME: Needs to check domain.
770     // FIXME: WebKitErrorPlugInWillHandleLoad is a workaround for the cancel we do to prevent
771     // loading plugin content twice.  See <rdar://problem/4258008>
772     return error.errorCode() != NSURLErrorCancelled && error.errorCode() != WebKitErrorPlugInWillHandleLoad;
773 }
774
775 void WebFrameLoaderClient::setDefersLoading(bool defers)
776 {
777     if (!defers)
778         deliverArchivedResourcesAfterDelay();
779 }
780
781 bool WebFrameLoaderClient::willUseArchive(ResourceLoader* loader, const ResourceRequest& request, const KURL& originalURL) const
782 {
783     if (request.url() != originalURL)
784         return false;
785
786     WebResource *resource = [dataSource(core(m_webFrame.get())->loader()->activeDocumentLoader()) _archivedSubresourceForURL:originalURL];
787     if (!resource)
788         return false;
789
790     m_pendingArchivedResources.set(loader, resource);
791     // Deliver the resource after a delay because callers don't expect to receive callbacks while calling this method.
792     deliverArchivedResourcesAfterDelay();
793
794     return true;
795 }
796
797 bool WebFrameLoaderClient::isArchiveLoadPending(ResourceLoader* loader) const
798 {
799     return m_pendingArchivedResources.contains(loader);
800 }
801
802 void WebFrameLoaderClient::cancelPendingArchiveLoad(ResourceLoader* loader)
803 {
804     if (m_pendingArchivedResources.isEmpty())
805         return;
806     m_pendingArchivedResources.remove(loader);
807     if (m_pendingArchivedResources.isEmpty())
808         m_archivedResourcesDeliveryTimer.stop();
809 }
810
811 void WebFrameLoaderClient::clearArchivedResources()
812 {
813     m_pendingArchivedResources.clear();
814     m_archivedResourcesDeliveryTimer.stop();
815 }
816
817 bool WebFrameLoaderClient::canHandleRequest(const ResourceRequest& request) const
818 {
819     return [WebView _canHandleRequest:request.nsURLRequest()];
820 }
821
822 bool WebFrameLoaderClient::canShowMIMEType(const String& MIMEType) const
823 {
824     return [WebView canShowMIMEType:MIMEType];
825 }
826
827 bool WebFrameLoaderClient::representationExistsForURLScheme(const String& URLScheme) const
828 {
829     return [WebView _representationExistsForURLScheme:URLScheme];
830 }
831
832 String WebFrameLoaderClient::generatedMIMETypeForURLScheme(const String& URLScheme) const
833 {
834     return [WebView _generatedMIMETypeForURLScheme:URLScheme];
835 }
836
837 void WebFrameLoaderClient::frameLoadCompleted()
838 {
839     // Note: Can be called multiple times.
840     // Even if already complete, we might have set a previous item on a frame that
841     // didn't do any data loading on the past transaction. Make sure to clear these out.
842     NSScrollView *sv = [m_webFrame->_private->webFrameView _scrollView];
843     if ([getWebView(m_webFrame.get()) drawsBackground])
844         [sv setDrawsBackground:YES];
845     core(m_webFrame.get())->loader()->setPreviousHistoryItem(0);
846 }
847
848
849 void WebFrameLoaderClient::saveViewStateToItem(HistoryItem* item)
850 {
851     if (!item)
852         return;
853     
854     NSView <WebDocumentView> *docView = [m_webFrame->_private->webFrameView documentView];
855
856     // we might already be detached when this is called from detachFromParent, in which
857     // case we don't want to override real data earlier gathered with (0,0)
858     if ([docView superview] && [docView conformsToProtocol:@protocol(_WebDocumentViewState)])
859         item->setViewState([(id <_WebDocumentViewState>)docView viewState]);
860 }
861
862 void WebFrameLoaderClient::restoreViewState()
863 {
864     HistoryItem* currentItem = core(m_webFrame.get())->loader()->currentHistoryItem();
865     ASSERT(currentItem);
866
867     // FIXME: As the ASSERT attests, it seems we should always have a currentItem here.
868     // One counterexample is <rdar://problem/4917290>
869     // For now, to cover this issue in release builds, there is no technical harm to returning
870     // early and from a user standpoint - as in the above radar - the previous page load failed 
871     // so there *is* no scroll state to restore!
872     if (!currentItem)
873         return;
874     
875     NSView <WebDocumentView> *docView = [m_webFrame->_private->webFrameView documentView];
876     if ([docView conformsToProtocol:@protocol(_WebDocumentViewState)]) {        
877         id state = currentItem->viewState();
878         if (state) {
879             [(id <_WebDocumentViewState>)docView setViewState:state];
880         }
881     }
882 }
883
884 void WebFrameLoaderClient::provisionalLoadStarted()
885 {    
886     // FIXME: This is OK as long as no one resizes the window,
887     // but in the case where someone does, it means garbage outside
888     // the occupied part of the scroll view.
889     [[m_webFrame->_private->webFrameView _scrollView] setDrawsBackground:NO];
890 }
891
892 void WebFrameLoaderClient::didFinishLoad()
893 {
894     [m_webFrame->_private->internalLoadDelegate webFrame:m_webFrame.get() didFinishLoadWithError:nil];    
895 }
896
897 void WebFrameLoaderClient::prepareForDataSourceReplacement()
898 {
899     if (![m_webFrame.get() _dataSource]) {
900         ASSERT(!core(m_webFrame.get())->tree()->childCount());
901         return;
902     }
903     
904     // Make sure that any work that is triggered by resigning first reponder can get done.
905     // The main example where this came up is the textDidEndEditing that is sent to the
906     // FormsDelegate (3223413).  We need to do this before _detachChildren, since that will
907     // remove the views as a side-effect of freeing the bridge, at which point we can't
908     // post the FormDelegate messages.
909     //
910     // Note that this can also take FirstResponder away from a child of our frameView that
911     // is not in a child frame's view.  This is OK because we are in the process
912     // of loading new content, which will blow away all editors in this top frame, and if
913     // a non-editor is firstReponder it will not be affected by endEditingFor:.
914     // Potentially one day someone could write a DocView whose editors were not all
915     // replaced by loading new content, but that does not apply currently.
916     NSView *frameView = m_webFrame->_private->webFrameView;
917     NSWindow *window = [frameView window];
918     NSResponder *firstResp = [window firstResponder];
919     if ([firstResp isKindOfClass:[NSView class]] && [(NSView *)firstResp isDescendantOf:frameView])
920         [window endEditingFor:firstResp];
921 }
922
923 PassRefPtr<DocumentLoader> WebFrameLoaderClient::createDocumentLoader(const ResourceRequest& request, const SubstituteData& substituteData)
924 {
925     RefPtr<WebDocumentLoaderMac> loader = new WebDocumentLoaderMac(request, substituteData);
926
927     WebDataSource *dataSource = [[WebDataSource alloc] _initWithDocumentLoader:loader.get()];
928     loader->setDataSource(dataSource, getWebView(m_webFrame.get()));
929     [dataSource release];
930
931     return loader.release();
932 }
933
934 // FIXME: <rdar://problem/4880065> - Push Global History into WebCore
935 // Once that task is complete, this will go away
936 void WebFrameLoaderClient::setTitle(const String& title, const KURL& URL)
937 {
938     NSURL* nsURL = URL;
939     nsURL = [nsURL _webkit_canonicalize];
940     if(!nsURL)
941         return;
942     NSString *titleNSString = title;
943     [[[WebHistory optionalSharedHistory] itemForURL:nsURL] setTitle:titleNSString];
944 }
945
946 void WebFrameLoaderClient::deliverArchivedResourcesAfterDelay() const
947 {
948     if (m_pendingArchivedResources.isEmpty())
949         return;
950     if (core(m_webFrame.get())->page()->defersLoading())
951         return;
952     if (!m_archivedResourcesDeliveryTimer.isActive())
953         m_archivedResourcesDeliveryTimer.startOneShot(0);
954 }
955
956 void WebFrameLoaderClient::deliverArchivedResources(Timer<WebFrameLoaderClient>*)
957 {
958     if (m_pendingArchivedResources.isEmpty())
959         return;
960     if (core(m_webFrame.get())->page()->defersLoading())
961         return;
962
963     const ResourceMap copy = m_pendingArchivedResources;
964     m_pendingArchivedResources.clear();
965
966     ResourceMap::const_iterator end = copy.end();
967     for (ResourceMap::const_iterator it = copy.begin(); it != end; ++it) {
968         RefPtr<ResourceLoader> loader = it->first;
969         WebResource *resource = it->second.get();
970         NSData *data = [[resource data] retain];
971         loader->didReceiveResponse([resource _response]);
972         loader->didReceiveData((const char*)[data bytes], [data length], [data length], true);
973         [data release];
974         loader->didFinishLoading();
975     }
976 }
977
978 void WebFrameLoaderClient::savePlatformDataToCachedPage(CachedPage* cachedPage)
979 {
980     WebCachedPagePlatformData* webPlatformData = new WebCachedPagePlatformData([m_webFrame->_private->webFrameView documentView]);
981     cachedPage->setCachedPagePlatformData(webPlatformData);
982 }
983
984 void WebFrameLoaderClient::transitionToCommittedFromCachedPage(CachedPage* cachedPage)
985 {
986     WebCachedPagePlatformData* platformData = reinterpret_cast<WebCachedPagePlatformData*>(cachedPage->cachedPagePlatformData());
987     NSView <WebDocumentView> *cachedView = platformData->webDocumentView();
988     ASSERT(cachedView != nil);
989     ASSERT(cachedPage->documentLoader());
990     [cachedView setDataSource:dataSource(cachedPage->documentLoader())];
991     [m_webFrame->_private->webFrameView _setDocumentView:cachedView];
992 }
993
994 void WebFrameLoaderClient::transitionToCommittedForNewPage()
995 {
996     WebFrameView *v = m_webFrame->_private->webFrameView;
997     WebDataSource *ds = [m_webFrame.get() _dataSource];
998
999     bool willProduceHTMLView = [[WebFrameView class] _viewClassForMIMEType:[ds _responseMIMEType]] == [WebHTMLView class];
1000     bool canSkipCreation = [m_webFrame.get() _frameLoader]->committingFirstRealLoad() && willProduceHTMLView;
1001     if (canSkipCreation) {
1002         [[v documentView] setDataSource:ds];
1003         return;
1004     }
1005
1006     // Don't suppress scrollbars before the view creation if we're making the view for a non-HTML view.
1007     if (!willProduceHTMLView)
1008         [[v _scrollView] setScrollBarsSuppressed:NO repaintOnUnsuppress:NO];
1009     
1010     NSView <WebDocumentView> *documentView = [v _makeDocumentViewForDataSource:ds];
1011     if (!documentView)
1012         return;
1013
1014     // FIXME: Could we skip some of this work for a top-level view that is not a WebHTMLView?
1015
1016     // If we own the view, delete the old one - otherwise the render m_frame will take care of deleting the view.
1017     Frame* coreFrame = core(m_webFrame.get());
1018     coreFrame->setView(0);
1019     FrameView* coreView = new FrameView(coreFrame);
1020     coreFrame->setView(coreView);
1021     coreView->deref(); // FIXME: Eliminate this crazy refcounting!
1022     coreView->setView(documentView);
1023     int marginWidth = [v _marginWidth];
1024     if (marginWidth >= 0)
1025         coreView->setMarginWidth(marginWidth);
1026     int marginHeight = [v _marginHeight];
1027     if (marginHeight >= 0)
1028         coreView->setMarginHeight(marginHeight);
1029
1030     [m_webFrame.get() _updateBackground];
1031
1032     [v _install];
1033
1034     // Call setDataSource on the document view after it has been placed in the view hierarchy.
1035     // This what we for the top-level view, so should do this for views in subframes as well.
1036     [documentView setDataSource:ds];
1037 }
1038
1039 RetainPtr<WebFramePolicyListener> WebFrameLoaderClient::setUpPolicyListener(FramePolicyFunction function)
1040 {
1041     // FIXME: <rdar://5634381> We need to support multiple active policy listeners.
1042
1043     [m_policyListener.get() invalidate];
1044
1045     WebFramePolicyListener *listener = [[WebFramePolicyListener alloc] initWithWebCoreFrame:core(m_webFrame.get())];
1046     m_policyListener = listener;
1047     [listener release];
1048     m_policyFunction = function;
1049
1050     return listener;
1051 }
1052
1053 void WebFrameLoaderClient::receivedPolicyDecison(PolicyAction action)
1054 {
1055     ASSERT(m_policyListener);
1056     ASSERT(m_policyFunction);
1057
1058     FramePolicyFunction function = m_policyFunction;
1059
1060     m_policyListener = nil;
1061     m_policyFunction = 0;
1062
1063     (core(m_webFrame.get())->loader()->*function)(action);
1064 }
1065
1066 String WebFrameLoaderClient::userAgent(const KURL& url)
1067 {
1068     return [getWebView(m_webFrame.get()) _userAgentForURL:url];
1069 }
1070
1071 static const MouseEvent* findMouseEvent(const Event* event)
1072 {
1073     for (const Event* e = event; e; e = e->underlyingEvent())
1074         if (e->isMouseEvent())
1075             return static_cast<const MouseEvent*>(e);
1076     return 0;
1077 }
1078
1079 NSDictionary *WebFrameLoaderClient::actionDictionary(const NavigationAction& action) const
1080 {
1081     unsigned modifierFlags = 0;
1082     const Event* event = action.event();
1083     if (const UIEventWithKeyState* keyStateEvent = findEventWithKeyState(const_cast<Event*>(event))) {
1084         if (keyStateEvent->ctrlKey())
1085             modifierFlags |= NSControlKeyMask;
1086         if (keyStateEvent->altKey())
1087             modifierFlags |= NSAlternateKeyMask;
1088         if (keyStateEvent->shiftKey())
1089             modifierFlags |= NSShiftKeyMask;
1090         if (keyStateEvent->metaKey())
1091             modifierFlags |= NSCommandKeyMask;
1092     }
1093     NSURL *originalURL = action.url();
1094     if (const MouseEvent* mouseEvent = findMouseEvent(event)) {
1095         IntPoint point(mouseEvent->pageX(), mouseEvent->pageY());
1096         WebElementDictionary *element = [[WebElementDictionary alloc]
1097             initWithHitTestResult:core(m_webFrame.get())->eventHandler()->hitTestResultAtPoint(point, false)];
1098         NSDictionary *result = [NSDictionary dictionaryWithObjectsAndKeys:
1099             [NSNumber numberWithInt:action.type()], WebActionNavigationTypeKey,
1100             element, WebActionElementKey,
1101             [NSNumber numberWithInt:mouseEvent->button()], WebActionButtonKey,
1102             [NSNumber numberWithInt:modifierFlags], WebActionModifierFlagsKey,
1103             originalURL, WebActionOriginalURLKey,
1104             nil];
1105         [element release];
1106         return result;
1107     }
1108     return [NSDictionary dictionaryWithObjectsAndKeys:
1109         [NSNumber numberWithInt:action.type()], WebActionNavigationTypeKey,
1110         [NSNumber numberWithInt:modifierFlags], WebActionModifierFlagsKey,
1111         originalURL, WebActionOriginalURLKey,
1112         nil];
1113 }
1114
1115 bool WebFrameLoaderClient::canCachePage() const
1116 {
1117     // We can only cache HTML pages right now
1118     return [[[m_webFrame.get() _dataSource] representation] isKindOfClass:[WebHTMLRepresentation class]];
1119 }
1120
1121 PassRefPtr<Frame> WebFrameLoaderClient::createFrame(const KURL& url, const String& name, HTMLFrameOwnerElement* ownerElement,
1122     const String& referrer, bool allowsScrolling, int marginWidth, int marginHeight)
1123 {
1124     BEGIN_BLOCK_OBJC_EXCEPTIONS;
1125     
1126     ASSERT(m_webFrame);
1127     
1128     WebFrameView *childView = [[WebFrameView alloc] init];
1129     [childView setAllowsScrolling:allowsScrolling];
1130     [childView _setMarginWidth:marginWidth];
1131     [childView _setMarginHeight:marginHeight];
1132
1133     WebFrameBridge *newBridge = [[WebFrameBridge alloc] initSubframeWithOwnerElement:ownerElement frameName:name frameView:childView];
1134     [childView release];
1135
1136     if (!newBridge)
1137         return 0;
1138
1139     [m_webFrame.get() _addChild:[newBridge webFrame]];
1140     [newBridge release];
1141
1142     RefPtr<Frame> newFrame = [newBridge _frame];
1143     
1144     [m_webFrame.get() _loadURL:url referrer:referrer intoChild:kit(newFrame.get())];
1145
1146     // The frame's onload handler may have removed it from the document.
1147     if (!newFrame->tree()->parent())
1148         return 0;
1149
1150     return newFrame.get();
1151
1152     END_BLOCK_OBJC_EXCEPTIONS;
1153
1154     return 0;
1155 }
1156
1157 ObjectContentType WebFrameLoaderClient::objectContentType(const KURL& url, const String& mimeType)
1158 {
1159     BEGIN_BLOCK_OBJC_EXCEPTIONS;
1160
1161     // This is a quirk that ensures Tiger Mail's WebKit plug-in will load during layout
1162     // and not attach time. (5520541)
1163     static BOOL isTigerMail = WKAppVersionCheckLessThan(@"com.apple.mail", -1, 3.0);
1164     if (isTigerMail && mimeType == "application/x-apple-msg-attachment")
1165         return ObjectContentNetscapePlugin;
1166
1167     String type = mimeType;
1168
1169     if (type.isEmpty()) {
1170         // Try to guess the MIME type based off the extension.
1171         NSURL *URL = url;
1172         NSString *extension = [[URL path] pathExtension];
1173         if ([extension length] > 0) {
1174             type = WKGetMIMETypeForExtension(extension);
1175             if (type.isEmpty()) {
1176                 // If no MIME type is specified, use a plug-in if we have one that can handle the extension.
1177                 if (WebBasePluginPackage *package = [getWebView(m_webFrame.get()) _pluginForExtension:extension]) {
1178                     if ([package isKindOfClass:[WebPluginPackage class]]) 
1179                         return ObjectContentOtherPlugin;
1180 #ifndef __LP64__
1181                     else {
1182                         ASSERT([package isKindOfClass:[WebNetscapePluginPackage class]]);
1183                         return ObjectContentNetscapePlugin;
1184                     }
1185 #endif
1186                 }
1187             }
1188         }
1189     }
1190
1191     if (type.isEmpty())
1192         return ObjectContentFrame; // Go ahead and hope that we can display the content.
1193
1194     if (MIMETypeRegistry::isSupportedImageMIMEType(type))
1195         return ObjectContentImage;
1196
1197     if (WebBasePluginPackage *package = [getWebView(m_webFrame.get()) _pluginForMIMEType:type]) {
1198 #ifndef __LP64__
1199         if ([package isKindOfClass:[WebNetscapePluginPackage class]])
1200             return ObjectContentNetscapePlugin;
1201 #endif
1202         ASSERT([package isKindOfClass:[WebPluginPackage class]]);
1203         return ObjectContentOtherPlugin;
1204     }
1205
1206     if ([WebFrameView _viewClassForMIMEType:type])
1207         return ObjectContentFrame;
1208     
1209     return ObjectContentNone;
1210
1211     END_BLOCK_OBJC_EXCEPTIONS;
1212
1213     return ObjectContentNone;
1214 }
1215
1216 static NSMutableArray* kit(const Vector<String>& vector)
1217 {
1218     unsigned len = vector.size();
1219     NSMutableArray* array = [NSMutableArray arrayWithCapacity:len];
1220     for (unsigned x = 0; x < len; x++)
1221         [array addObject:vector[x]];
1222     return array;
1223 }
1224
1225 static String parameterValue(const Vector<String>& paramNames, const Vector<String>& paramValues, const String& name)
1226 {
1227     size_t size = paramNames.size();
1228     ASSERT(size == paramValues.size());
1229     for (size_t i = 0; i < size; ++i) {
1230         if (equalIgnoringCase(paramNames[i], name))
1231             return paramValues[i];
1232     }
1233     return String();
1234 }
1235
1236 static NSView *pluginView(WebFrame *frame, WebPluginPackage *pluginPackage,
1237     NSArray *attributeNames, NSArray *attributeValues, NSURL *baseURL,
1238     DOMElement *element, BOOL loadManually)
1239 {
1240     WebHTMLView *docView = (WebHTMLView *)[[frame frameView] documentView];
1241     ASSERT([docView isKindOfClass:[WebHTMLView class]]);
1242         
1243     WebPluginController *pluginController = [docView _pluginController];
1244     
1245     // Store attributes in a dictionary so they can be passed to WebPlugins.
1246     NSMutableDictionary *attributes = [[NSMutableDictionary alloc] initWithObjects:attributeValues forKeys:attributeNames];
1247     
1248     [pluginPackage load];
1249     Class viewFactory = [pluginPackage viewFactory];
1250     
1251     NSView *view = nil;
1252     NSDictionary *arguments = nil;
1253     
1254     if ([viewFactory respondsToSelector:@selector(plugInViewWithArguments:)]) {
1255         arguments = [NSDictionary dictionaryWithObjectsAndKeys:
1256             baseURL, WebPlugInBaseURLKey,
1257             attributes, WebPlugInAttributesKey,
1258             pluginController, WebPlugInContainerKey,
1259             [NSNumber numberWithInt:loadManually ? WebPlugInModeFull : WebPlugInModeEmbed], WebPlugInModeKey,
1260             [NSNumber numberWithBool:!loadManually], WebPlugInShouldLoadMainResourceKey,
1261             element, WebPlugInContainingElementKey,
1262             nil];
1263         LOG(Plugins, "arguments:\n%@", arguments);
1264     } else if ([viewFactory respondsToSelector:@selector(pluginViewWithArguments:)]) {
1265         arguments = [NSDictionary dictionaryWithObjectsAndKeys:
1266             baseURL, WebPluginBaseURLKey,
1267             attributes, WebPluginAttributesKey,
1268             pluginController, WebPluginContainerKey,
1269             element, WebPlugInContainingElementKey,
1270             nil];
1271         LOG(Plugins, "arguments:\n%@", arguments);
1272     }
1273
1274     view = [WebPluginController plugInViewWithArguments:arguments fromPluginPackage:pluginPackage];
1275     [attributes release];
1276     return view;
1277 }
1278
1279 Widget* WebFrameLoaderClient::createPlugin(const IntSize& size, Element* element, const KURL& url,
1280     const Vector<String>& paramNames, const Vector<String>& paramValues, const String& mimeType, bool loadManually)
1281 {
1282     BEGIN_BLOCK_OBJC_EXCEPTIONS;
1283
1284     ASSERT(paramNames.size() == paramValues.size());
1285
1286     int errorCode = 0;
1287
1288     WebView *webView = getWebView(m_webFrame.get());
1289     SEL selector = @selector(webView:plugInViewWithArguments:);
1290     NSURL *URL = url;
1291
1292     if ([[webView UIDelegate] respondsToSelector:selector]) {
1293         NSMutableDictionary *attributes = [[NSMutableDictionary alloc] initWithObjects:kit(paramNames) forKeys:kit(paramValues)];
1294         NSDictionary *arguments = [[NSDictionary alloc] initWithObjectsAndKeys:
1295             attributes, WebPlugInAttributesKey,
1296             [NSNumber numberWithInt:loadManually ? WebPlugInModeFull : WebPlugInModeEmbed], WebPlugInModeKey,
1297             [NSNumber numberWithBool:!loadManually], WebPlugInShouldLoadMainResourceKey,
1298             element, WebPlugInContainingElementKey,
1299             URL, WebPlugInBaseURLKey, // URL might be nil, so add it last
1300             nil];
1301
1302         NSView *view = CallUIDelegate(webView, selector, arguments);
1303
1304         [attributes release];
1305         [arguments release];
1306
1307         if (view)
1308             return new Widget(view);
1309     }
1310
1311     NSString *MIMEType;
1312     WebBasePluginPackage *pluginPackage;
1313     if (mimeType.isEmpty()) {
1314         MIMEType = nil;
1315         pluginPackage = nil;
1316     } else {
1317         MIMEType = mimeType;
1318         pluginPackage = [webView _pluginForMIMEType:mimeType];
1319     }
1320     
1321     NSString *extension = [[URL path] pathExtension];
1322     if (!pluginPackage && [extension length] != 0) {
1323         pluginPackage = [webView _pluginForExtension:extension];
1324         if (pluginPackage) {
1325             NSString *newMIMEType = [pluginPackage MIMETypeForExtension:extension];
1326             if ([newMIMEType length] != 0)
1327                 MIMEType = newMIMEType;
1328         }
1329     }
1330
1331     NSView *view = nil;
1332
1333     Document* document = core(m_webFrame.get())->document();
1334     NSURL *baseURL = document->baseURL();
1335     if (pluginPackage) {
1336         if ([pluginPackage isKindOfClass:[WebPluginPackage class]])
1337             view = pluginView(m_webFrame.get(), (WebPluginPackage *)pluginPackage, kit(paramNames), kit(paramValues), baseURL, kit(element), loadManually);
1338             
1339 #ifndef __LP64__
1340         else if ([pluginPackage isKindOfClass:[WebNetscapePluginPackage class]]) {
1341             WebNetscapePluginEmbeddedView *embeddedView = [[[WebNetscapePluginEmbeddedView alloc]
1342                 initWithFrame:NSMakeRect(0, 0, size.width(), size.height())
1343                 pluginPackage:(WebNetscapePluginPackage *)pluginPackage
1344                 URL:URL
1345                 baseURL:baseURL
1346                 MIMEType:MIMEType
1347                 attributeKeys:kit(paramNames)
1348                 attributeValues:kit(paramValues)
1349                 loadManually:loadManually
1350                 DOMElement:kit(element)] autorelease];
1351             view = embeddedView;
1352         } 
1353 #endif
1354     } else
1355         errorCode = WebKitErrorCannotFindPlugIn;
1356
1357     if (!errorCode && !view)
1358         errorCode = WebKitErrorCannotLoadPlugIn;
1359
1360     if (errorCode) {
1361         NSError *error = [[NSError alloc] _initWithPluginErrorCode:errorCode
1362             contentURL:URL
1363             pluginPageURL:document->completeURL(parseURL(parameterValue(paramNames, paramValues, "pluginspage")))
1364             pluginName:[pluginPackage name]
1365             MIMEType:MIMEType];
1366         WebNullPluginView *nullView = [[[WebNullPluginView alloc] initWithFrame:NSMakeRect(0, 0, size.width(), size.height())
1367             error:error DOMElement:kit(element)] autorelease];
1368         view = nullView;
1369         [error release];
1370     }
1371     
1372     ASSERT(view);
1373     return new Widget(view);
1374
1375     END_BLOCK_OBJC_EXCEPTIONS;
1376
1377     return 0;
1378 }
1379
1380 void WebFrameLoaderClient::redirectDataToPlugin(Widget* pluginWidget)
1381 {
1382     BEGIN_BLOCK_OBJC_EXCEPTIONS;
1383
1384     WebHTMLRepresentation *representation = (WebHTMLRepresentation *)[[m_webFrame.get() _dataSource] representation];
1385
1386     NSView *pluginView = pluginWidget->getView();
1387
1388 #ifndef __LP64__
1389     if ([pluginView isKindOfClass:[WebNetscapePluginEmbeddedView class]])
1390         [representation _redirectDataToManualLoader:(WebNetscapePluginEmbeddedView *)pluginView forPluginView:pluginView];
1391     else {
1392 #else
1393     {
1394 #endif
1395         WebHTMLView *documentView = (WebHTMLView *)[[m_webFrame.get() frameView] documentView];
1396         ASSERT([documentView isKindOfClass:[WebHTMLView class]]);
1397         [representation _redirectDataToManualLoader:[documentView _pluginController] forPluginView:pluginView];
1398     }
1399
1400     END_BLOCK_OBJC_EXCEPTIONS;
1401 }
1402
1403 Widget* WebFrameLoaderClient::createJavaAppletWidget(const IntSize& size, Element* element, const KURL& baseURL, 
1404     const Vector<String>& paramNames, const Vector<String>& paramValues)
1405 {
1406     BEGIN_BLOCK_OBJC_EXCEPTIONS;
1407
1408     NSView *view = nil;
1409
1410     NSString *MIMEType = @"application/x-java-applet";
1411     
1412     WebView *webView = getWebView(m_webFrame.get());
1413
1414     WebBasePluginPackage *pluginPackage = [webView _pluginForMIMEType:MIMEType];
1415
1416     if (pluginPackage) {
1417         if ([pluginPackage isKindOfClass:[WebPluginPackage class]]) {
1418             // For some reason, the Java plug-in requires that we pass the dimension of the plug-in as attributes.
1419             NSMutableArray *names = kit(paramNames);
1420             NSMutableArray *values = kit(paramValues);
1421             if (parameterValue(paramNames, paramValues, "width").isNull()) {
1422                 [names addObject:@"width"];
1423                 [values addObject:[NSString stringWithFormat:@"%d", size.width()]];
1424             }
1425             if (parameterValue(paramNames, paramValues, "height").isNull()) {
1426                 [names addObject:@"height"];
1427                 [values addObject:[NSString stringWithFormat:@"%d", size.width()]];
1428             }
1429             view = pluginView(m_webFrame.get(), (WebPluginPackage *)pluginPackage, names, values, baseURL, kit(element), NO);
1430         } 
1431 #ifndef __LP64__
1432         else if ([pluginPackage isKindOfClass:[WebNetscapePluginPackage class]]) {
1433             view = [[[WebNetscapePluginEmbeddedView alloc] initWithFrame:NSMakeRect(0, 0, size.width(), size.height())
1434                 pluginPackage:(WebNetscapePluginPackage *)pluginPackage
1435                 URL:nil
1436                 baseURL:baseURL
1437                 MIMEType:MIMEType
1438                 attributeKeys:kit(paramNames)
1439                 attributeValues:kit(paramValues)
1440                 loadManually:NO
1441                 DOMElement:kit(element)] autorelease];
1442         } else {
1443             ASSERT_NOT_REACHED();
1444         }
1445 #endif
1446     }
1447
1448     if (!view) {
1449         NSError *error = [[NSError alloc] _initWithPluginErrorCode:WebKitErrorJavaUnavailable
1450             contentURL:nil
1451             pluginPageURL:nil
1452             pluginName:[pluginPackage name]
1453             MIMEType:MIMEType];
1454         view = [[[WebNullPluginView alloc] initWithFrame:NSMakeRect(0, 0, size.width(), size.height())
1455             error:error DOMElement:kit(element)] autorelease];
1456         [error release];
1457     }
1458
1459     ASSERT(view);
1460     return new Widget(view);
1461
1462     END_BLOCK_OBJC_EXCEPTIONS;
1463     
1464     return new Widget;
1465 }
1466
1467 String WebFrameLoaderClient::overrideMediaType() const
1468 {
1469     NSString* overrideType = [getWebView(m_webFrame.get()) mediaStyle];
1470     if (overrideType)
1471         return overrideType;
1472     return String();
1473 }
1474
1475 void WebFrameLoaderClient::windowObjectCleared()
1476 {
1477     Frame *frame = core(m_webFrame.get());
1478     WebView *webView = getWebView(m_webFrame.get());
1479     WebFrameLoadDelegateImplementationCache* implementations = WebViewGetFrameLoadDelegateImplementations(webView);
1480     if (implementations->didClearWindowObjectForFrameFunc) {
1481         CallFrameLoadDelegate(implementations->didClearWindowObjectForFrameFunc, webView, @selector(webView:didClearWindowObject:forFrame:),
1482             frame->windowScriptObject(), m_webFrame.get());
1483     } else if (implementations->windowScriptObjectAvailableFunc) {
1484         CallFrameLoadDelegate(implementations->windowScriptObjectAvailableFunc, webView, @selector(webView:windowScriptObjectAvailable:),
1485             frame->windowScriptObject());
1486     }
1487
1488     if ([webView scriptDebugDelegate] || [WebScriptDebugServer listenerCount]) {
1489         [m_webFrame.get() _detachScriptDebugger];
1490         [m_webFrame.get() _attachScriptDebugger];
1491     }
1492 }
1493
1494 void WebFrameLoaderClient::registerForIconNotification(bool listen)
1495 {
1496     [[m_webFrame.get() webView] _registerForIconNotification:listen];
1497 }
1498
1499 void WebFrameLoaderClient::didPerformFirstNavigation() const
1500 {
1501     WebPreferences *preferences = [[m_webFrame.get() webView] preferences];
1502     if ([preferences automaticallyDetectsCacheModel] && [preferences cacheModel] < WebCacheModelDocumentBrowser)
1503         [preferences setCacheModel:WebCacheModelDocumentBrowser];
1504 }
1505
1506 @implementation WebFramePolicyListener
1507
1508 #ifndef BUILDING_ON_TIGER
1509 + (void)initialize
1510 {
1511     WebCoreObjCFinalizeOnMainThread(self);
1512 }
1513 #endif
1514
1515 - (id)initWithWebCoreFrame:(Frame*)frame
1516 {
1517     self = [self init];
1518     if (!self)
1519         return nil;
1520     frame->ref();
1521     m_frame = frame;
1522     return self;
1523 }
1524
1525 - (void)invalidate
1526 {
1527     if (m_frame) {
1528         m_frame->deref();
1529         m_frame = 0;
1530     }
1531 }
1532
1533 - (void)dealloc
1534 {
1535     if (m_frame)
1536         m_frame->deref();
1537     [super dealloc];
1538 }
1539
1540 - (void)finalize
1541 {
1542     ASSERT_MAIN_THREAD();
1543     if (m_frame)
1544         m_frame->deref();
1545     [super finalize];
1546 }
1547
1548 - (void)receivedPolicyDecision:(PolicyAction)action
1549 {
1550     RefPtr<Frame> frame = adoptRef(m_frame);
1551     m_frame = 0;
1552     if (frame)
1553         static_cast<WebFrameLoaderClient*>(frame->loader()->client())->receivedPolicyDecison(action);
1554 }
1555
1556 - (void)ignore
1557 {
1558     [self receivedPolicyDecision:PolicyIgnore];
1559 }
1560
1561 - (void)download
1562 {
1563     [self receivedPolicyDecision:PolicyDownload];
1564 }
1565
1566 - (void)use
1567 {
1568     [self receivedPolicyDecision:PolicyUse];
1569 }
1570
1571 - (void)continue
1572 {
1573     [self receivedPolicyDecision:PolicyUse];
1574 }
1575
1576 @end