| /* |
| * Copyright (C) 2005 Apple Computer, Inc. All rights reserved. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of |
| * its contributors may be used to endorse or promote products derived |
| * from this software without specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY |
| * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE |
| * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY |
| * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
| * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
| * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
| * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
| * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| */ |
| |
| #import <WebKit/WebDownload.h> |
| |
| #import <Foundation/NSURLAuthenticationChallenge.h> |
| #import <Foundation/NSURLDownload.h> |
| #import <WebCore/AuthenticationMac.h> |
| #import <WebCore/Credential.h> |
| #import <WebCore/CredentialStorage.h> |
| #import <WebCore/ProtectionSpace.h> |
| #import <WebKit/WebPanelAuthenticationHandler.h> |
| #import <wtf/Assertions.h> |
| |
| #import "WebTypesInternal.h" |
| |
| using namespace WebCore; |
| |
| @class NSURLConnectionDelegateProxy; |
| |
| // FIXME: The following are NSURLDownload SPI - it would be nice to not have to override them at |
| // some point in the future |
| @interface NSURLDownload (WebDownloadCapability) |
| - (id)_initWithLoadingConnection:(NSURLConnection *)connection |
| request:(NSURLRequest *)request |
| response:(NSURLResponse *)response |
| delegate:(id)delegate |
| proxy:(NSURLConnectionDelegateProxy *)proxy; |
| - (id)_initWithRequest:(NSURLRequest *)request |
| delegate:(id)delegate |
| directory:(NSString *)directory; |
| @end |
| |
| @interface WebDownloadInternal : NSObject <NSURLDownloadDelegate> |
| { |
| @public |
| id realDelegate; |
| } |
| |
| - (void)setRealDelegate:(id)rd; |
| |
| @end |
| |
| @implementation WebDownloadInternal |
| |
| - (void)dealloc |
| { |
| [realDelegate release]; |
| [super dealloc]; |
| } |
| |
| - (void)setRealDelegate:(id)rd |
| { |
| [rd retain]; |
| [realDelegate release]; |
| realDelegate = rd; |
| } |
| |
| - (BOOL)respondsToSelector:(SEL)selector |
| { |
| if (selector == @selector(downloadDidBegin:) || |
| selector == @selector(download:willSendRequest:redirectResponse:) || |
| selector == @selector(download:didReceiveResponse:) || |
| selector == @selector(download:didReceiveDataOfLength:) || |
| selector == @selector(download:shouldDecodeSourceDataOfMIMEType:) || |
| selector == @selector(download:decideDestinationWithSuggestedFilename:) || |
| selector == @selector(download:didCreateDestination:) || |
| selector == @selector(downloadDidFinish:) || |
| selector == @selector(download:didFailWithError:) || |
| selector == @selector(download:shouldBeginChildDownloadOfSource:delegate:) || |
| selector == @selector(download:didBeginChildDownload:)) { |
| return [realDelegate respondsToSelector:selector]; |
| } |
| |
| return [super respondsToSelector:selector]; |
| } |
| |
| - (void)downloadDidBegin:(NSURLDownload *)download |
| { |
| [realDelegate downloadDidBegin:download]; |
| } |
| |
| - (NSURLRequest *)download:(NSURLDownload *)download willSendRequest:(NSURLRequest *)request redirectResponse:(NSURLResponse *)redirectResponse |
| { |
| return [realDelegate download:download willSendRequest:request redirectResponse:redirectResponse]; |
| } |
| |
| - (void)download:(NSURLDownload *)download didReceiveAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge |
| { |
| // Try previously stored credential first. |
| if (![challenge previousFailureCount]) { |
| NSURLCredential *credential = mac(CredentialStorage::get(core([challenge protectionSpace]))); |
| if (credential) { |
| [[challenge sender] useCredential:credential forAuthenticationChallenge:challenge]; |
| return; |
| } |
| } |
| |
| if ([realDelegate respondsToSelector:@selector(download:didReceiveAuthenticationChallenge:)]) { |
| [realDelegate download:download didReceiveAuthenticationChallenge:challenge]; |
| } else { |
| NSWindow *window = nil; |
| if ([realDelegate respondsToSelector:@selector(downloadWindowForAuthenticationSheet:)]) { |
| window = [realDelegate downloadWindowForAuthenticationSheet:(WebDownload *)download]; |
| } |
| |
| [[WebPanelAuthenticationHandler sharedHandler] startAuthentication:challenge window:window]; |
| } |
| } |
| |
| - (void)download:(NSURLDownload *)download didCancelAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge |
| { |
| if ([realDelegate respondsToSelector:@selector(download:didCancelAuthenticationChallenge:)]) { |
| [realDelegate download:download didCancelAuthenticationChallenge:challenge]; |
| } else { |
| [[WebPanelAuthenticationHandler sharedHandler] cancelAuthentication:challenge]; |
| } |
| } |
| |
| - (void)download:(NSURLDownload *)download didReceiveResponse:(NSURLResponse *)response |
| { |
| [realDelegate download:download didReceiveResponse:response]; |
| } |
| |
| - (void)download:(NSURLDownload *)download didReceiveDataOfLength:(NSUInteger)length |
| { |
| [realDelegate download:download didReceiveDataOfLength:length]; |
| } |
| |
| - (BOOL)download:(NSURLDownload *)download shouldDecodeSourceDataOfMIMEType:(NSString *)encodingType |
| { |
| return [realDelegate download:download shouldDecodeSourceDataOfMIMEType:encodingType]; |
| } |
| |
| - (void)download:(NSURLDownload *)download decideDestinationWithSuggestedFilename:(NSString *)filename |
| { |
| [realDelegate download:download decideDestinationWithSuggestedFilename:filename]; |
| } |
| |
| - (void)download:(NSURLDownload *)download didCreateDestination:(NSString *)path |
| { |
| [realDelegate download:download didCreateDestination:path]; |
| } |
| |
| - (void)downloadDidFinish:(NSURLDownload *)download |
| { |
| [realDelegate downloadDidFinish:download]; |
| } |
| |
| - (void)download:(NSURLDownload *)download didFailWithError:(NSError *)error |
| { |
| [realDelegate download:download didFailWithError:error]; |
| } |
| |
| - (NSURLRequest *)download:(NSURLDownload *)download shouldBeginChildDownloadOfSource:(NSURLRequest *)child delegate:(id *)childDelegate |
| { |
| return [realDelegate download:download shouldBeginChildDownloadOfSource:child delegate:childDelegate]; |
| } |
| |
| - (void)download:(NSURLDownload *)parent didBeginChildDownload:(NSURLDownload *)child |
| { |
| [realDelegate download:parent didBeginChildDownload:child]; |
| } |
| |
| @end |
| |
| @implementation WebDownload |
| |
| - (void)_setRealDelegate:(id)delegate |
| { |
| if (_webInternal == nil) { |
| _webInternal = [[WebDownloadInternal alloc] init]; |
| [_webInternal setRealDelegate:delegate]; |
| } else { |
| ASSERT(_webInternal == delegate); |
| } |
| } |
| |
| - (id)init |
| { |
| self = [super init]; |
| if (self != nil) { |
| // _webInternal can be set up before init by _setRealDelegate |
| if (_webInternal == nil) { |
| _webInternal = [[WebDownloadInternal alloc] init]; |
| } |
| } |
| return self; |
| } |
| |
| - (void)dealloc |
| { |
| [_webInternal release]; |
| [super dealloc]; |
| } |
| |
| - (id)initWithRequest:(NSURLRequest *)request delegate:(id<NSURLDownloadDelegate>)delegate |
| { |
| [self _setRealDelegate:delegate]; |
| return [super initWithRequest:request delegate:_webInternal]; |
| } |
| |
| - (id)_initWithLoadingConnection:(NSURLConnection *)connection |
| request:(NSURLRequest *)request |
| response:(NSURLResponse *)response |
| delegate:(id)delegate |
| proxy:(NSURLConnectionDelegateProxy *)proxy |
| { |
| [self _setRealDelegate:delegate]; |
| return [super _initWithLoadingConnection:connection request:request response:response delegate:_webInternal proxy:proxy]; |
| } |
| |
| - (id)_initWithRequest:(NSURLRequest *)request |
| delegate:(id)delegate |
| directory:(NSString *)directory |
| { |
| [self _setRealDelegate:delegate]; |
| return [super _initWithRequest:request delegate:_webInternal directory:directory]; |
| } |
| |
| - (void)connection:(NSURLConnection *)connection willStopBufferingData:(NSData *)data |
| { |
| // NSURLConnection calls this method even if it is not implemented. |
| // This happens because NSURLConnection caches the results of respondsToSelector. |
| // Those results become invalid when the delegate of NSURLConnectionDelegateProxy is changed. |
| // This is a workaround since this problem needs to be fixed in NSURLConnectionDelegateProxy. |
| // <rdar://problem/3913270> NSURLConnection calls unimplemented delegate method in WebDownload |
| } |
| |
| @end |