1/*
2 * Copyright (C) 2005 Apple Computer, 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 <WebKit/WebDownload.h>
30
31#import <Foundation/NSURLAuthenticationChallenge.h>
32#import <Foundation/NSURLDownload.h>
33#import <WebCore/AuthenticationMac.h>
34#import <WebCore/Credential.h>
35#import <WebCore/CredentialStorage.h>
36#import <WebCore/ProtectionSpace.h>
37#import <WebKit/WebPanelAuthenticationHandler.h>
38#import <wtf/Assertions.h>
39
40#import "WebTypesInternal.h"
41
42using namespace WebCore;
43
44@class NSURLConnectionDelegateProxy;
45
46// FIXME: The following are NSURLDownload SPI - it would be nice to not have to override them at
47// some point in the future
48@interface NSURLDownload (WebDownloadCapability)
49- (id)_initWithLoadingConnection:(NSURLConnection *)connection
50                         request:(NSURLRequest *)request
51                        response:(NSURLResponse *)response
52                        delegate:(id)delegate
53                           proxy:(NSURLConnectionDelegateProxy *)proxy;
54- (id)_initWithRequest:(NSURLRequest *)request
55              delegate:(id)delegate
56             directory:(NSString *)directory;
57@end
58
59@interface WebDownloadInternal : NSObject <NSURLDownloadDelegate>
60{
61@public
62    id realDelegate;
63}
64
65- (void)setRealDelegate:(id)rd;
66
67@end
68
69@implementation WebDownloadInternal
70
71- (void)dealloc
72{
73    [realDelegate release];
74    [super dealloc];
75}
76
77- (void)setRealDelegate:(id)rd
78{
79    [rd retain];
80    [realDelegate release];
81    realDelegate = rd;
82}
83
84- (BOOL)respondsToSelector:(SEL)selector
85{
86    if (selector == @selector(downloadDidBegin:) ||
87        selector == @selector(download:willSendRequest:redirectResponse:) ||
88        selector == @selector(download:didReceiveResponse:) ||
89        selector == @selector(download:didReceiveDataOfLength:) ||
90        selector == @selector(download:shouldDecodeSourceDataOfMIMEType:) ||
91        selector == @selector(download:decideDestinationWithSuggestedFilename:) ||
92        selector == @selector(download:didCreateDestination:) ||
93        selector == @selector(downloadDidFinish:) ||
94        selector == @selector(download:didFailWithError:) ||
95        selector == @selector(download:shouldBeginChildDownloadOfSource:delegate:) ||
96        selector == @selector(download:didBeginChildDownload:)) {
97        return [realDelegate respondsToSelector:selector];
98    }
99
100    return [super respondsToSelector:selector];
101}
102
103- (void)downloadDidBegin:(NSURLDownload *)download
104{
105    [realDelegate downloadDidBegin:download];
106}
107
108- (NSURLRequest *)download:(NSURLDownload *)download willSendRequest:(NSURLRequest *)request redirectResponse:(NSURLResponse *)redirectResponse
109{
110    return [realDelegate download:download willSendRequest:request redirectResponse:redirectResponse];
111}
112
113- (void)download:(NSURLDownload *)download didReceiveAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge
114{
115    // Try previously stored credential first.
116    if (![challenge previousFailureCount]) {
117        NSURLCredential *credential = mac(CredentialStorage::get(core([challenge protectionSpace])));
118        if (credential) {
119            [[challenge sender] useCredential:credential forAuthenticationChallenge:challenge];
120            return;
121        }
122    }
123
124    if ([realDelegate respondsToSelector:@selector(download:didReceiveAuthenticationChallenge:)]) {
125        [realDelegate download:download didReceiveAuthenticationChallenge:challenge];
126    } else {
127        NSWindow *window = nil;
128        if ([realDelegate respondsToSelector:@selector(downloadWindowForAuthenticationSheet:)]) {
129            window = [realDelegate downloadWindowForAuthenticationSheet:(WebDownload *)download];
130        }
131
132        [[WebPanelAuthenticationHandler sharedHandler] startAuthentication:challenge window:window];
133    }
134}
135
136- (void)download:(NSURLDownload *)download didCancelAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge
137{
138    if ([realDelegate respondsToSelector:@selector(download:didCancelAuthenticationChallenge:)]) {
139        [realDelegate download:download didCancelAuthenticationChallenge:challenge];
140    } else {
141        [[WebPanelAuthenticationHandler sharedHandler] cancelAuthentication:challenge];
142    }
143}
144
145- (void)download:(NSURLDownload *)download didReceiveResponse:(NSURLResponse *)response
146{
147    [realDelegate download:download didReceiveResponse:response];
148}
149
150- (void)download:(NSURLDownload *)download didReceiveDataOfLength:(NSUInteger)length
151{
152    [realDelegate download:download didReceiveDataOfLength:length];
153}
154
155- (BOOL)download:(NSURLDownload *)download shouldDecodeSourceDataOfMIMEType:(NSString *)encodingType
156{
157    return [realDelegate download:download shouldDecodeSourceDataOfMIMEType:encodingType];
158}
159
160- (void)download:(NSURLDownload *)download decideDestinationWithSuggestedFilename:(NSString *)filename
161{
162    [realDelegate download:download decideDestinationWithSuggestedFilename:filename];
163}
164
165- (void)download:(NSURLDownload *)download didCreateDestination:(NSString *)path
166{
167    [realDelegate download:download didCreateDestination:path];
168}
169
170- (void)downloadDidFinish:(NSURLDownload *)download
171{
172    [realDelegate downloadDidFinish:download];
173}
174
175- (void)download:(NSURLDownload *)download didFailWithError:(NSError *)error
176{
177    [realDelegate download:download didFailWithError:error];
178}
179
180- (NSURLRequest *)download:(NSURLDownload *)download shouldBeginChildDownloadOfSource:(NSURLRequest *)child delegate:(id *)childDelegate
181{
182    return [realDelegate download:download shouldBeginChildDownloadOfSource:child delegate:childDelegate];
183}
184
185- (void)download:(NSURLDownload *)parent didBeginChildDownload:(NSURLDownload *)child
186{
187    [realDelegate download:parent didBeginChildDownload:child];
188}
189
190@end
191
192@implementation WebDownload
193
194- (void)_setRealDelegate:(id)delegate
195{
196    if (_webInternal == nil) {
197        _webInternal = [[WebDownloadInternal alloc] init];
198        [_webInternal setRealDelegate:delegate];
199    } else {
200        ASSERT(_webInternal == delegate);
201    }
202}
203
204- (id)init
205{
206    self = [super init];
207    if (self != nil) {
208        // _webInternal can be set up before init by _setRealDelegate
209        if (_webInternal == nil) {
210            _webInternal = [[WebDownloadInternal alloc] init];
211        }
212    }
213    return self;
214}
215
216- (void)dealloc
217{
218    [_webInternal release];
219    [super dealloc];
220}
221
222- (id)initWithRequest:(NSURLRequest *)request delegate:(id<NSURLDownloadDelegate>)delegate
223{
224    [self _setRealDelegate:delegate];
225    return [super initWithRequest:request delegate:_webInternal];
226}
227
228- (id)_initWithLoadingConnection:(NSURLConnection *)connection
229                         request:(NSURLRequest *)request
230                        response:(NSURLResponse *)response
231                        delegate:(id)delegate
232                           proxy:(NSURLConnectionDelegateProxy *)proxy
233{
234    [self _setRealDelegate:delegate];
235    return [super _initWithLoadingConnection:connection request:request response:response delegate:_webInternal proxy:proxy];
236}
237
238- (id)_initWithRequest:(NSURLRequest *)request
239              delegate:(id)delegate
240             directory:(NSString *)directory
241{
242    [self _setRealDelegate:delegate];
243    return [super _initWithRequest:request delegate:_webInternal directory:directory];
244}
245
246- (void)connection:(NSURLConnection *)connection willStopBufferingData:(NSData *)data
247{
248    // NSURLConnection calls this method even if it is not implemented.
249    // This happens because NSURLConnection caches the results of respondsToSelector.
250    // Those results become invalid when the delegate of NSURLConnectionDelegateProxy is changed.
251    // This is a workaround since this problem needs to be fixed in NSURLConnectionDelegateProxy.
252    // <rdar://problem/3913270> NSURLConnection calls unimplemented delegate method in WebDownload
253}
254
255@end
256