<http://webkit.org/b/53192> Add experimental support for HTTP pipelining in CFNetwork
[WebKit-https.git] / 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 (WebCoreContentDispositionEncoding)
44 - (NSArray *)contentDispositionEncodingFallbackArray;
45 @end
46
47 @interface NSMutableURLRequest (WebCoreContentDispositionEncoding)
48 - (void)setContentDispositionEncodingFallbackArray:(NSArray *)theEncodingFallbackArray;
49 @end
50
51 namespace WebCore {
52
53 NSURLRequest* ResourceRequest::nsURLRequest() const
54
55     updatePlatformRequest();
56     
57     return [[m_nsRequest.get() retain] autorelease]; 
58 }
59
60 void ResourceRequest::doUpdateResourceRequest()
61 {
62     m_url = [m_nsRequest.get() URL];
63     m_cachePolicy = (ResourceRequestCachePolicy)[m_nsRequest.get() cachePolicy];
64     m_timeoutInterval = [m_nsRequest.get() timeoutInterval];
65     m_firstPartyForCookies = [m_nsRequest.get() mainDocumentURL];
66     
67     if (NSString* method = [m_nsRequest.get() HTTPMethod])
68         m_httpMethod = method;
69     m_allowCookies = [m_nsRequest.get() HTTPShouldHandleCookies];
70
71 #if !defined(BUILDING_ON_TIGER) && !defined(BUILDING_ON_LEOPARD) && !defined(BUILDING_ON_SNOW_LEOPARD)
72     if (isHTTPPipeliningEnabled() && !shouldForceHTTPPipeliningPriorityHigh())
73         m_priority = mapHTTPPipeliningPriorityToResourceLoadPriority(wkGetHTTPPipeliningPriority(m_nsRequest.get()));
74 #endif
75
76     NSDictionary *headers = [m_nsRequest.get() allHTTPHeaderFields];
77     NSEnumerator *e = [headers keyEnumerator];
78     NSString *name;
79     m_httpHeaderFields.clear();
80     while ((name = [e nextObject]))
81         m_httpHeaderFields.set(name, [headers objectForKey:name]);
82
83     // The below check can be removed once we require a version of Foundation with -[NSURLRequest contentDispositionEncodingFallbackArray] method.
84     static bool supportsContentDispositionEncodingFallbackArray = [NSURLRequest instancesRespondToSelector:@selector(contentDispositionEncodingFallbackArray)];
85     if (supportsContentDispositionEncodingFallbackArray) {
86         m_responseContentDispositionEncodingFallbackArray.clear();
87         NSArray *encodingFallbacks = [m_nsRequest.get() contentDispositionEncodingFallbackArray];
88         NSUInteger count = [encodingFallbacks count];
89         for (NSUInteger i = 0; i < count; ++i) {
90             CFStringEncoding encoding = CFStringConvertNSStringEncodingToEncoding([(NSNumber *)[encodingFallbacks objectAtIndex:i] unsignedLongValue]);
91             if (encoding != kCFStringEncodingInvalidId)
92                 m_responseContentDispositionEncodingFallbackArray.append(CFStringConvertEncodingToIANACharSetName(encoding));
93         }
94     }
95
96     if (NSData* bodyData = [m_nsRequest.get() HTTPBody])
97         m_httpBody = FormData::create([bodyData bytes], [bodyData length]);
98     else if (NSInputStream* bodyStream = [m_nsRequest.get() HTTPBodyStream])
99         if (FormData* formData = httpBodyFromStream(bodyStream))
100             m_httpBody = formData;
101 }
102
103 void ResourceRequest::doUpdatePlatformRequest()
104 {
105     if (isNull()) {
106         m_nsRequest = nil;
107         return;
108     }
109     
110     NSMutableURLRequest* nsRequest = [m_nsRequest.get() mutableCopy];
111
112     if (nsRequest)
113         [nsRequest setURL:url()];
114     else
115         nsRequest = [[NSMutableURLRequest alloc] initWithURL:url()];
116
117 #ifdef BUILDING_ON_TIGER
118     wkSupportsMultipartXMixedReplace(nsRequest);
119 #endif
120
121 #if !defined(BUILDING_ON_TIGER) && !defined(BUILDING_ON_LEOPARD) && !defined(BUILDING_ON_SNOW_LEOPARD)
122     if (isHTTPPipeliningEnabled()) {
123         int priority = mapResourceLoadPriorityToHTTPPipeliningPriority(m_priority);
124         wkSetHTTPPipeliningPriority(nsRequest, shouldForceHTTPPipeliningPriorityHigh() ? 2 : priority);
125     }
126 #endif
127
128     [nsRequest setCachePolicy:(NSURLRequestCachePolicy)cachePolicy()];
129     if (timeoutInterval() != unspecifiedTimeoutInterval)
130         [nsRequest setTimeoutInterval:timeoutInterval()];
131     [nsRequest setMainDocumentURL:firstPartyForCookies()];
132     if (!httpMethod().isEmpty())
133         [nsRequest setHTTPMethod:httpMethod()];
134     [nsRequest setHTTPShouldHandleCookies:allowCookies()];
135
136     // Cannot just use setAllHTTPHeaderFields here, because it does not remove headers.
137     NSArray *oldHeaderFieldNames = [[nsRequest allHTTPHeaderFields] allKeys];
138     for (unsigned i = [oldHeaderFieldNames count]; i != 0; --i)
139         [nsRequest setValue:nil forHTTPHeaderField:[oldHeaderFieldNames objectAtIndex:i - 1]];
140     HTTPHeaderMap::const_iterator end = httpHeaderFields().end();
141     for (HTTPHeaderMap::const_iterator it = httpHeaderFields().begin(); it != end; ++it)
142         [nsRequest setValue:it->second forHTTPHeaderField:it->first];
143
144     // The below check can be removed once we require a version of Foundation with -[NSMutableURLRequest setContentDispositionEncodingFallbackArray:] method.
145     static bool supportsContentDispositionEncodingFallbackArray = [NSMutableURLRequest instancesRespondToSelector:@selector(setContentDispositionEncodingFallbackArray:)];
146     if (supportsContentDispositionEncodingFallbackArray) {
147         NSMutableArray *encodingFallbacks = [NSMutableArray array];
148         unsigned count = m_responseContentDispositionEncodingFallbackArray.size();
149         for (unsigned i = 0; i != count; ++i) {
150             CFStringRef encodingName = m_responseContentDispositionEncodingFallbackArray[i].createCFString();
151             unsigned long nsEncoding = CFStringConvertEncodingToNSStringEncoding(CFStringConvertIANACharSetNameToEncoding(encodingName));
152             CFRelease(encodingName);
153             if (nsEncoding != kCFStringEncodingInvalidId)
154                 [encodingFallbacks addObject:[NSNumber numberWithUnsignedLong:nsEncoding]];
155         }
156         [nsRequest setContentDispositionEncodingFallbackArray:encodingFallbacks];
157     }
158
159     RefPtr<FormData> formData = httpBody();
160     if (formData && !formData->isEmpty())
161         WebCore::setHTTPBody(nsRequest, formData);
162     
163     m_nsRequest.adoptNS(nsRequest);
164 }
165
166 void ResourceRequest::applyWebArchiveHackForMail()
167 {
168     // Hack because Mail checks for this property to detect data / archive loads
169     [NSURLProtocol setProperty:@"" forKey:@"WebDataRequest" inRequest:(NSMutableURLRequest *)nsURLRequest()];
170 }
171
172 } // namespace WebCore
173
174 #endif // !USE(CFNETWORK)