ResourceRequestMac.mm [plain text]
/*
* Copyright (C) 2006, 2007, 2008 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 COMPUTER, INC. ``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 COMPUTER, INC. OR
* 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 "ResourceRequest.h"
#import "FormDataStreamMac.h"
#import "ResourceRequestCFNet.h"
#import "RuntimeApplicationChecks.h"
#import "WebCoreSystemInterface.h"
#import <wtf/text/CString.h>
#import <Foundation/Foundation.h>
@interface NSURLRequest (WebNSURLRequestDetails)
- (NSArray *)contentDispositionEncodingFallbackArray;
+ (void)setDefaultTimeoutInterval:(NSTimeInterval)seconds;
- (CFURLRequestRef)_CFURLRequest;
- (id)_initWithCFURLRequest:(CFURLRequestRef)request;
@end
@interface NSMutableURLRequest (WebMutableNSURLRequestDetails)
- (void)setContentDispositionEncodingFallbackArray:(NSArray *)theEncodingFallbackArray;
@end
namespace WebCore {
NSURLRequest *ResourceRequest::nsURLRequest(HTTPBodyUpdatePolicy bodyPolicy) const
{
updatePlatformRequest(bodyPolicy);
return [[m_nsRequest.get() retain] autorelease];
}
#if USE(CFNETWORK)
ResourceRequest::ResourceRequest(NSURLRequest *nsRequest)
: ResourceRequestBase()
, m_cfRequest([nsRequest _CFURLRequest])
, m_nsRequest(nsRequest)
{
}
void ResourceRequest::updateNSURLRequest()
{
if (m_cfRequest)
m_nsRequest = adoptNS([[NSURLRequest alloc] _initWithCFURLRequest:m_cfRequest.get()]);
}
#else
CFURLRequestRef ResourceRequest::cfURLRequest(HTTPBodyUpdatePolicy bodyPolicy) const
{
return [nsURLRequest(bodyPolicy) _CFURLRequest];
}
void ResourceRequest::doUpdateResourceRequest()
{
m_url = [m_nsRequest.get() URL];
m_cachePolicy = (ResourceRequestCachePolicy)[m_nsRequest.get() cachePolicy];
m_timeoutInterval = [m_nsRequest.get() timeoutInterval];
m_firstPartyForCookies = [m_nsRequest.get() mainDocumentURL];
if (NSString* method = [m_nsRequest.get() HTTPMethod])
m_httpMethod = method;
m_allowCookies = [m_nsRequest.get() HTTPShouldHandleCookies];
#if PLATFORM(IOS) || __MAC_OS_X_VERSION_MIN_REQUIRED >= 1070
if (ResourceRequest::httpPipeliningEnabled())
m_priority = toResourceLoadPriority(wkGetHTTPPipeliningPriority([m_nsRequest.get() _CFURLRequest]));
#endif
NSDictionary *headers = [m_nsRequest.get() allHTTPHeaderFields];
NSEnumerator *e = [headers keyEnumerator];
NSString *name;
m_httpHeaderFields.clear();
while ((name = [e nextObject]))
m_httpHeaderFields.set(name, [headers objectForKey:name]);
m_responseContentDispositionEncodingFallbackArray.clear();
NSArray *encodingFallbacks = [m_nsRequest.get() contentDispositionEncodingFallbackArray];
NSUInteger count = [encodingFallbacks count];
for (NSUInteger i = 0; i < count; ++i) {
CFStringEncoding encoding = CFStringConvertNSStringEncodingToEncoding([(NSNumber *)[encodingFallbacks objectAtIndex:i] unsignedLongValue]);
if (encoding != kCFStringEncodingInvalidId)
m_responseContentDispositionEncodingFallbackArray.append(CFStringConvertEncodingToIANACharSetName(encoding));
}
#if ENABLE(CACHE_PARTITIONING)
NSString* cachePartition = [NSURLProtocol propertyForKey:(NSString *)wkCachePartitionKey() inRequest:m_nsRequest.get()];
if (cachePartition)
m_cachePartition = cachePartition;
#endif
}
void ResourceRequest::doUpdateResourceHTTPBody()
{
if (NSData* bodyData = [m_nsRequest.get() HTTPBody])
m_httpBody = FormData::create([bodyData bytes], [bodyData length]);
else if (NSInputStream* bodyStream = [m_nsRequest.get() HTTPBodyStream]) {
FormData* formData = httpBodyFromStream(bodyStream);
// There is no FormData object if a client provided a custom data stream.
// We shouldn't be looking at http body after client callbacks.
ASSERT(formData);
if (formData)
m_httpBody = formData;
}
}
void ResourceRequest::doUpdatePlatformRequest()
{
if (isNull()) {
m_nsRequest = nil;
return;
}
NSMutableURLRequest *nsRequest = [m_nsRequest.get() mutableCopy];
if (nsRequest)
[nsRequest setURL:url()];
else
nsRequest = [[NSMutableURLRequest alloc] initWithURL:url()];
#if PLATFORM(IOS) || __MAC_OS_X_VERSION_MIN_REQUIRED >= 1070
if (ResourceRequest::httpPipeliningEnabled())
wkSetHTTPPipeliningPriority([nsRequest _CFURLRequest], toHTTPPipeliningPriority(m_priority));
#endif
[nsRequest setCachePolicy:(NSURLRequestCachePolicy)cachePolicy()];
#if PLATFORM(IOS) || __MAC_OS_X_VERSION_MIN_REQUIRED >= 1070
wkCFURLRequestAllowAllPostCaching([nsRequest _CFURLRequest]);
#endif
double timeoutInterval = ResourceRequestBase::timeoutInterval();
if (timeoutInterval)
[nsRequest setTimeoutInterval:timeoutInterval];
// Otherwise, respect NSURLRequest default timeout.
[nsRequest setMainDocumentURL:firstPartyForCookies()];
if (!httpMethod().isEmpty())
[nsRequest setHTTPMethod:httpMethod()];
[nsRequest setHTTPShouldHandleCookies:allowCookies()];
// Cannot just use setAllHTTPHeaderFields here, because it does not remove headers.
NSArray *oldHeaderFieldNames = [[nsRequest allHTTPHeaderFields] allKeys];
for (unsigned i = [oldHeaderFieldNames count]; i != 0; --i)
[nsRequest setValue:nil forHTTPHeaderField:[oldHeaderFieldNames objectAtIndex:i - 1]];
HTTPHeaderMap::const_iterator end = httpHeaderFields().end();
for (HTTPHeaderMap::const_iterator it = httpHeaderFields().begin(); it != end; ++it)
[nsRequest setValue:it->value forHTTPHeaderField:it->key];
#if __MAC_OS_X_VERSION_MIN_REQUIRED <= 1070
// Workaround for <rdar://problem/16848509>
if (url().host().contains('%'))
[nsRequest setValue:url().host() forHTTPHeaderField:@"Host"];
#endif
NSMutableArray *encodingFallbacks = [NSMutableArray array];
unsigned count = m_responseContentDispositionEncodingFallbackArray.size();
for (unsigned i = 0; i != count; ++i) {
RetainPtr<CFStringRef> encodingName = m_responseContentDispositionEncodingFallbackArray[i].createCFString();
unsigned long nsEncoding = CFStringConvertEncodingToNSStringEncoding(CFStringConvertIANACharSetNameToEncoding(encodingName.get()));
if (nsEncoding != kCFStringEncodingInvalidId)
[encodingFallbacks addObject:[NSNumber numberWithUnsignedLong:nsEncoding]];
}
[nsRequest setContentDispositionEncodingFallbackArray:encodingFallbacks];
#if ENABLE(CACHE_PARTITIONING)
String partition = cachePartition();
if (!partition.isNull() && !partition.isEmpty()) {
NSString *partitionValue = [NSString stringWithUTF8String:partition.utf8().data()];
[NSURLProtocol setProperty:partitionValue forKey:(NSString *)wkCachePartitionKey() inRequest:nsRequest];
}
#endif
m_nsRequest = adoptNS(nsRequest);
}
void ResourceRequest::doUpdatePlatformHTTPBody()
{
if (isNull()) {
ASSERT(!m_nsRequest);
return;
}
NSMutableURLRequest *nsRequest = [m_nsRequest.get() mutableCopy];
if (nsRequest)
[nsRequest setURL:url()];
else
nsRequest = [[NSMutableURLRequest alloc] initWithURL:url()];
RefPtr<FormData> formData = httpBody();
if (formData && !formData->isEmpty())
WebCore::setHTTPBody(nsRequest, formData);
if (NSInputStream *bodyStream = [nsRequest HTTPBodyStream]) {
// For streams, provide a Content-Length to avoid using chunked encoding, and to get accurate total length in callbacks.
NSString *lengthString = [bodyStream propertyForKey:(NSString *)formDataStreamLengthPropertyName()];
if (lengthString) {
[nsRequest setValue:lengthString forHTTPHeaderField:@"Content-Length"];
// Since resource request is already marked updated, we need to keep it up to date too.
ASSERT(m_resourceRequestUpdated);
m_httpHeaderFields.set("Content-Length", lengthString);
}
}
m_nsRequest = adoptNS(nsRequest);
}
void ResourceRequest::updateFromDelegatePreservingOldHTTPBody(const ResourceRequest& delegateProvidedRequest)
{
RefPtr<FormData> oldHTTPBody = httpBody();
*this = delegateProvidedRequest;
setHTTPBody(oldHTTPBody.release());
}
void ResourceRequest::applyWebArchiveHackForMail()
{
// Hack because Mail checks for this property to detect data / archive loads
[NSURLProtocol setProperty:@"" forKey:@"WebDataRequest" inRequest:(NSMutableURLRequest *)nsURLRequest(DoNotUpdateHTTPBody)];
}
void ResourceRequest::setStorageSession(CFURLStorageSessionRef storageSession)
{
updatePlatformRequest();
m_nsRequest = adoptNS(wkCopyRequestWithStorageSession(storageSession, m_nsRequest.get()));
}
#endif // USE(CFNETWORK)
static bool initQuickLookResourceCachingQuirks()
{
if (applicationIsSafari())
return false;
NSArray* frameworks = [NSBundle allFrameworks];
if (!frameworks)
return false;
for (unsigned int i = 0; i < [frameworks count]; i++) {
NSBundle* bundle = [frameworks objectAtIndex: i];
const char* bundleID = [[bundle bundleIdentifier] UTF8String];
if (bundleID && !strcasecmp(bundleID, "com.apple.QuickLookUIFramework"))
return true;
}
return false;
}
bool ResourceRequest::useQuickLookResourceCachingQuirks()
{
static bool flag = initQuickLookResourceCachingQuirks();
return flag;
}
} // namespace WebCore