/*
* Copyright (C) 2010, 2011 Apple 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.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. 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 INC. 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 "config.h"
#import "Download.h"
#if PLATFORM(MAC)
#import <WebCore/AuthenticationChallenge.h>
#import <WebCore/AuthenticationMac.h>
#import <WebCore/NotImplemented.h>
#import <WebCore/ResourceHandle.h>
#import <WebCore/ResourceHandleClient.h>
#import <WebCore/ResourceResponse.h>
#import "DataReference.h"
#import "WebPage.h"
@interface NSURLDownload (WebNSURLDownloadDetails)
+(id)_downloadWithLoadingConnection:(NSURLConnection *)connection
request:(NSURLRequest *)request
response:(NSURLResponse *)r
delegate:(id)delegate
proxy:(id)proxy;
@end
@interface WKDownloadAsDelegate : NSObject <NSURLDownloadDelegate> {
WebKit::Download* _download;
}
- (id)initWithDownload:(WebKit::Download*)download;
- (void)invalidate;
@end
using namespace WebCore;
namespace WebKit {
void Download::start()
{
ASSERT(!m_nsURLDownload);
ASSERT(!m_delegate);
m_delegate = adoptNS([[WKDownloadAsDelegate alloc] initWithDownload:this]);
m_nsURLDownload = adoptNS([[NSURLDownload alloc] initWithRequest:m_request.nsURLRequest(UpdateHTTPBody) delegate:m_delegate.get()]);
// FIXME: Allow this to be changed by the client.
[m_nsURLDownload setDeletesFileUponFailure:NO];
}
void Download::startWithHandle(ResourceHandle* handle, const ResourceResponse& response)
{
ASSERT(!m_nsURLDownload);
ASSERT(!m_delegate);
m_delegate = adoptNS([[WKDownloadAsDelegate alloc] initWithDownload:this]);
m_nsURLDownload = [NSURLDownload _downloadWithLoadingConnection:handle->connection()
request:m_request.nsURLRequest(UpdateHTTPBody)
response:response.nsURLResponse()
delegate:m_delegate.get()
proxy:nil];
// FIXME: Allow this to be changed by the client.
[m_nsURLDownload setDeletesFileUponFailure:NO];
}
void Download::cancel()
{
[m_nsURLDownload cancel];
RetainPtr<NSData> resumeData = [m_nsURLDownload resumeData];
didCancel(IPC::DataReference(reinterpret_cast<const uint8_t*>([resumeData bytes]), [resumeData length]));
}
void Download::platformInvalidate()
{
ASSERT(m_nsURLDownload);
ASSERT(m_delegate);
[m_delegate invalidate];
m_delegate = nullptr;
m_nsURLDownload = nullptr;
}
void Download::didDecideDestination(const String& destination, bool allowOverwrite)
{
}
void Download::platformDidFinish()
{
}
void Download::receivedCredential(const AuthenticationChallenge& authenticationChallenge, const Credential& credential)
{
[authenticationChallenge.sender() useCredential:mac(credential) forAuthenticationChallenge:authenticationChallenge.nsURLAuthenticationChallenge()];
}
void Download::receivedRequestToContinueWithoutCredential(const AuthenticationChallenge& authenticationChallenge)
{
[authenticationChallenge.sender() continueWithoutCredentialForAuthenticationChallenge:authenticationChallenge.nsURLAuthenticationChallenge()];
}
void Download::receivedCancellation(const AuthenticationChallenge& authenticationChallenge)
{
[authenticationChallenge.sender() cancelAuthenticationChallenge:authenticationChallenge.nsURLAuthenticationChallenge()];
}
void Download::receivedRequestToPerformDefaultHandling(const AuthenticationChallenge& authenticationChallenge)
{
[authenticationChallenge.sender() performDefaultHandlingForAuthenticationChallenge:authenticationChallenge.nsURLAuthenticationChallenge()];
}
void Download::receivedChallengeRejection(const AuthenticationChallenge& authenticationChallenge)
{
[authenticationChallenge.sender() rejectProtectionSpaceAndContinueWithChallenge:authenticationChallenge.nsURLAuthenticationChallenge()];
}
} // namespace WebKit
@implementation WKDownloadAsDelegate
// FIXME: It would be nice if these callbacks wouldn't have to be invoked on the main thread.
static void dispatchOnMainThread(void (^block)())
{
if (RunLoop::isMain()) {
block();
return;
}
dispatch_sync(dispatch_get_main_queue(), block);
}
- (id)initWithDownload:(WebKit::Download*)download
{
self = [super init];
if (!self)
return nil;
_download = download;
return self;
}
- (void)invalidate
{
_download = 0;
}
- (void)downloadDidBegin:(NSURLDownload *)download
{
dispatchOnMainThread(^{
if (_download)
_download->didStart();
});
}
- (NSURLRequest *)download:(NSURLDownload *)download willSendRequest:(NSURLRequest *)request redirectResponse:(NSURLResponse *)redirectResponse
{
return request;
}
- (BOOL)download:(NSURLDownload *)connection canAuthenticateAgainstProtectionSpace:(NSURLProtectionSpace *)protectionSpace
{
// FIXME: Implement.
notImplemented();
return NO;
}
- (void)download:(NSURLDownload *)download didReceiveAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge
{
dispatchOnMainThread(^{
if (_download)
_download->didReceiveAuthenticationChallenge(core(challenge));
});
}
- (void)download:(NSURLDownload *)download didCancelAuthenticationChallenge:(NSURLAuthenticationChallenge *)challenge
{
// FIXME: Implement.
notImplemented();
}
- (BOOL)downloadShouldUseCredentialStorage:(NSURLDownload *)download
{
return NO;
}
- (void)download:(NSURLDownload *)download didReceiveResponse:(NSURLResponse *)response
{
dispatchOnMainThread(^{
if (_download)
_download->didReceiveResponse(response);
});
}
- (void)download:(NSURLDownload *)download willResumeWithResponse:(NSURLResponse *)response fromByte:(long long)startingByte
{
// FIXME: Implement.
notImplemented();
}
- (void)download:(NSURLDownload *)download didReceiveDataOfLength:(NSUInteger)length
{
dispatchOnMainThread(^{
if (_download)
_download->didReceiveData(length);
});
}
- (BOOL)download:(NSURLDownload *)download shouldDecodeSourceDataOfMIMEType:(NSString *)encodingType
{
__block BOOL returnValue;
dispatchOnMainThread(^{
if (_download)
returnValue = _download->shouldDecodeSourceDataOfMIMEType(encodingType);
else
returnValue = YES;
});
return returnValue;;
}
- (void)download:(NSURLDownload *)download decideDestinationWithSuggestedFilename:(NSString *)filename
{
dispatchOnMainThread(^{
String destination;
bool allowOverwrite;
if (_download)
destination = _download->decideDestinationWithSuggestedFilename(filename, allowOverwrite);
if (!destination.isNull())
[download setDestination:destination allowOverwrite:allowOverwrite];
});
}
- (void)download:(NSURLDownload *)download didCreateDestination:(NSString *)path
{
dispatchOnMainThread(^{
if (_download)
_download->didCreateDestination(path);
});
}
- (void)downloadDidFinish:(NSURLDownload *)download
{
dispatchOnMainThread(^{
if (_download)
_download->didFinish();
});
}
- (void)download:(NSURLDownload *)download didFailWithError:(NSError *)error
{
dispatchOnMainThread(^{
if (!_download)
return;
RetainPtr<NSData> resumeData = [download resumeData];
IPC::DataReference dataReference(reinterpret_cast<const uint8_t*>([resumeData bytes]), [resumeData length]);
_download->didFail(error, dataReference);
});
}
@end
#endif // PLATFORM(MAC)