Source/WebCore/platform/network/mac/ResourceRequestMac.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  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #import "config.h"
  27 #import "ResourceRequest.h"
  28
  29 #if !USE(CFNETWORK)
  30
  31 #import "WebCoreSystemInterface.h"
  32
  33 #import "FormDataStreamMac.h"
  34 #import "ResourceRequestCFNet.h"
  35 #import "WebCoreSystemInterface.h"
  36
  37 #import <Foundation/Foundation.h>
  38
  39 #ifdef BUILDING_ON_TIGER
  40 typedef unsigned NSUInteger;
  41 #endif
  42
  43 @interface NSURLRequest (WebNSURLRequestDetails)
  44 - (NSArray *)contentDispositionEncodingFallbackArray;
  45 + (void)setDefaultTimeoutInterval:(NSTimeInterval)seconds;
  46 @end
  47
  48 @interface NSMutableURLRequest (WebMutableNSURLRequestDetails)
  49 - (void)setContentDispositionEncodingFallbackArray:(NSArray *)theEncodingFallbackArray;
  50 @end
  51
  52 namespace WebCore {
  53
  54 NSURLRequest* ResourceRequest::nsURLRequest() const
  55 {
  56     updatePlatformRequest();
  57
  58     return [[m_nsRequest.get() retain] autorelease];
  59 }
  60
  61 void ResourceRequest::doUpdateResourceRequest()
  62 {
  63     m_url = [m_nsRequest.get() URL];
  64     m_cachePolicy = (ResourceRequestCachePolicy)[m_nsRequest.get() cachePolicy];
  65     m_timeoutInterval = [m_nsRequest.get() timeoutInterval];
  66     m_firstPartyForCookies = [m_nsRequest.get() mainDocumentURL];
  67
  68     if (NSString* method = [m_nsRequest.get() HTTPMethod])
  69         m_httpMethod = method;
  70     m_allowCookies = [m_nsRequest.get() HTTPShouldHandleCookies];
  71
  72 #if !defined(BUILDING_ON_TIGER) && !defined(BUILDING_ON_LEOPARD) && !defined(BUILDING_ON_SNOW_LEOPARD)
  73     if (isHTTPPipeliningEnabled() && !shouldForceHTTPPipeliningPriorityHigh())
  74         m_priority = mapHTTPPipeliningPriorityToResourceLoadPriority(wkGetHTTPPipeliningPriority(m_nsRequest.get()));
  75 #endif
  76
  77     NSDictionary *headers = [m_nsRequest.get() allHTTPHeaderFields];
  78     NSEnumerator *e = [headers keyEnumerator];
  79     NSString *name;
  80     m_httpHeaderFields.clear();
  81     while ((name = [e nextObject]))
  82         m_httpHeaderFields.set(name, [headers objectForKey:name]);
  83
  84     // The below check can be removed once we require a version of Foundation with -[NSURLRequest contentDispositionEncodingFallbackArray] method.
  85     static bool supportsContentDispositionEncodingFallbackArray = [NSURLRequest instancesRespondToSelector:@selector(contentDispositionEncodingFallbackArray)];
  86     if (supportsContentDispositionEncodingFallbackArray) {
  87         m_responseContentDispositionEncodingFallbackArray.clear();
  88         NSArray *encodingFallbacks = [m_nsRequest.get() contentDispositionEncodingFallbackArray];
  89         NSUInteger count = [encodingFallbacks count];
  90         for (NSUInteger i = 0; i < count; ++i) {
  91             CFStringEncoding encoding = CFStringConvertNSStringEncodingToEncoding([(NSNumber *)[encodingFallbacks objectAtIndex:i] unsignedLongValue]);
  92             if (encoding != kCFStringEncodingInvalidId)
  93                 m_responseContentDispositionEncodingFallbackArray.append(CFStringConvertEncodingToIANACharSetName(encoding));
  94         }
  95     }
  96
  97     if (NSData* bodyData = [m_nsRequest.get() HTTPBody])
  98         m_httpBody = FormData::create([bodyData bytes], [bodyData length]);
  99     else if (NSInputStream* bodyStream = [m_nsRequest.get() HTTPBodyStream])
 100         if (FormData* formData = httpBodyFromStream(bodyStream))
 101             m_httpBody = formData;
 102 }
 103
 104 void ResourceRequest::doUpdatePlatformRequest()
 105 {
 106     if (isNull()) {
 107         m_nsRequest = nil;
 108         return;
 109     }
 110
 111     NSMutableURLRequest* nsRequest = [m_nsRequest.get() mutableCopy];
 112
 113     if (nsRequest)
 114         [nsRequest setURL:url()];
 115     else
 116         nsRequest = [[NSMutableURLRequest alloc] initWithURL:url()];
 117
 118 #ifdef BUILDING_ON_TIGER
 119     wkSupportsMultipartXMixedReplace(nsRequest);
 120 #endif
 121
 122 #if !defined(BUILDING_ON_TIGER) && !defined(BUILDING_ON_LEOPARD) && !defined(BUILDING_ON_SNOW_LEOPARD)
 123     if (isHTTPPipeliningEnabled()) {
 124         int priority = mapResourceLoadPriorityToHTTPPipeliningPriority(m_priority);
 125         wkSetHTTPPipeliningPriority(nsRequest, shouldForceHTTPPipeliningPriorityHigh() ? 2 : priority);
 126     }
 127 #endif
 128
 129     [nsRequest setCachePolicy:(NSURLRequestCachePolicy)cachePolicy()];
 130
 131     double timeoutInterval = ResourceRequestBase::timeoutInterval();
 132     if (timeoutInterval)
 133         [nsRequest setTimeoutInterval:timeoutInterval];
 134     // Otherwise, respect NSURLRequest default timeout.
 135
 136     [nsRequest setMainDocumentURL:firstPartyForCookies()];
 137     if (!httpMethod().isEmpty())
 138         [nsRequest setHTTPMethod:httpMethod()];
 139     [nsRequest setHTTPShouldHandleCookies:allowCookies()];
 140
 141     // Cannot just use setAllHTTPHeaderFields here, because it does not remove headers.
 142     NSArray *oldHeaderFieldNames = [[nsRequest allHTTPHeaderFields] allKeys];
 143     for (unsigned i = [oldHeaderFieldNames count]; i != 0; --i)
 144         [nsRequest setValue:nil forHTTPHeaderField:[oldHeaderFieldNames objectAtIndex:i - 1]];
 145     HTTPHeaderMap::const_iterator end = httpHeaderFields().end();
 146     for (HTTPHeaderMap::const_iterator it = httpHeaderFields().begin(); it != end; ++it)
 147         [nsRequest setValue:it->second forHTTPHeaderField:it->first];
 148
 149     // The below check can be removed once we require a version of Foundation with -[NSMutableURLRequest setContentDispositionEncodingFallbackArray:] method.
 150     static bool supportsContentDispositionEncodingFallbackArray = [NSMutableURLRequest instancesRespondToSelector:@selector(setContentDispositionEncodingFallbackArray:)];
 151     if (supportsContentDispositionEncodingFallbackArray) {
 152         NSMutableArray *encodingFallbacks = [NSMutableArray array];
 153         unsigned count = m_responseContentDispositionEncodingFallbackArray.size();
 154         for (unsigned i = 0; i != count; ++i) {
 155             CFStringRef encodingName = m_responseContentDispositionEncodingFallbackArray[i].createCFString();
 156             unsigned long nsEncoding = CFStringConvertEncodingToNSStringEncoding(CFStringConvertIANACharSetNameToEncoding(encodingName));
 157             CFRelease(encodingName);
 158             if (nsEncoding != kCFStringEncodingInvalidId)
 159                 [encodingFallbacks addObject:[NSNumber numberWithUnsignedLong:nsEncoding]];
 160         }
 161         [nsRequest setContentDispositionEncodingFallbackArray:encodingFallbacks];
 162     }
 163
 164     RefPtr<FormData> formData = httpBody();
 165     if (formData && !formData->isEmpty())
 166         WebCore::setHTTPBody(nsRequest, formData);
 167
 168     m_nsRequest.adoptNS(nsRequest);
 169 }
 170
 171 void ResourceRequest::applyWebArchiveHackForMail()
 172 {
 173     // Hack because Mail checks for this property to detect data / archive loads
 174     [NSURLProtocol setProperty:@"" forKey:@"WebDataRequest" inRequest:(NSMutableURLRequest *)nsURLRequest()];
 175 }
 176
 177 #if USE(CFURLSTORAGESESSIONS)
 178
 179 void ResourceRequest::setStorageSession(CFURLStorageSessionRef storageSession)
 180 {
 181     m_nsRequest = wkCopyRequestWithStorageSession(storageSession, m_nsRequest.get());
 182 }
 183
 184 #endif
 185
 186 } // namespace WebCore
 187
 188 #endif // !USE(CFNETWORK)