2 * Copyright (C) 2006, 2007 Apple Inc. All rights reserved.
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
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.
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.
27 #include "ResourceRequestCFNet.h"
29 #include "FormDataStreamCFNet.h"
30 #include "ResourceRequest.h"
32 #include <CFNetwork/CFURLRequestPriv.h>
36 CFURLRequestRef ResourceRequest::cfURLRequest() const
38 updatePlatformRequest();
40 return m_cfRequest.get();
43 static inline void addHeadersFromHashMap(CFMutableURLRequestRef request, const HTTPHeaderMap& requestHeaders)
45 if (!requestHeaders.size())
48 HTTPHeaderMap::const_iterator end = requestHeaders.end();
49 for (HTTPHeaderMap::const_iterator it = requestHeaders.begin(); it != end; ++it) {
50 CFStringRef key = it->first.createCFString();
51 CFStringRef value = it->second.createCFString();
52 CFURLRequestSetHTTPHeaderFieldValue(request, key, value);
58 void ResourceRequest::doUpdatePlatformRequest()
60 CFMutableURLRequestRef cfRequest;
62 RetainPtr<CFURLRef> url(AdoptCF, ResourceRequest::url().createCFURL());
63 RetainPtr<CFURLRef> mainDocumentURL(AdoptCF, ResourceRequest::mainDocumentURL().createCFURL());
65 cfRequest = CFURLRequestCreateMutableCopy(0, m_cfRequest.get());
66 CFURLRequestSetURL(cfRequest, url.get());
67 CFURLRequestSetMainDocumentURL(cfRequest, mainDocumentURL.get());
69 cfRequest = CFURLRequestCreateMutable(0, url.get(), (CFURLRequestCachePolicy)cachePolicy(), timeoutInterval(), mainDocumentURL.get());
72 RetainPtr<CFStringRef> requestMethod(AdoptCF, httpMethod().createCFString());
73 CFURLRequestSetHTTPRequestMethod(cfRequest, requestMethod.get());
75 addHeadersFromHashMap(cfRequest, httpHeaderFields());
76 WebCore::setHTTPBody(cfRequest, httpBody());
77 CFURLRequestSetShouldHandleHTTPCookies(cfRequest, allowHTTPCookies());
80 CFURLRequestSetHTTPCookieStorageAcceptPolicy(cfRequest, CFURLRequestGetHTTPCookieStorageAcceptPolicy(m_cfRequest.get()));
81 CFURLRequestSetSSLProperties(cfRequest, CFURLRequestGetSSLProperties(m_cfRequest.get()));
84 m_cfRequest.adoptCF(cfRequest);
87 void ResourceRequest::doUpdateResourceRequest()
89 m_url = CFURLRequestGetURL(m_cfRequest.get());
91 m_cachePolicy = (ResourceRequestCachePolicy)CFURLRequestGetCachePolicy(m_cfRequest.get());
92 m_timeoutInterval = CFURLRequestGetTimeoutInterval(m_cfRequest.get());
93 m_mainDocumentURL = CFURLRequestGetMainDocumentURL(m_cfRequest.get());
94 if (CFStringRef method = CFURLRequestCopyHTTPRequestMethod(m_cfRequest.get())) {
95 m_httpMethod = method;
98 m_allowHTTPCookies = CFURLRequestShouldHandleHTTPCookies(m_cfRequest.get());
100 if (CFDictionaryRef headers = CFURLRequestCopyAllHTTPHeaderFields(m_cfRequest.get())) {
101 CFIndex headerCount = CFDictionaryGetCount(headers);
102 Vector<const void*, 128> keys(headerCount);
103 Vector<const void*, 128> values(headerCount);
104 CFDictionaryGetKeysAndValues(headers, keys.data(), values.data());
105 for (int i = 0; i < headerCount; ++i)
106 m_httpHeaderFields.set((CFStringRef)keys[i], (CFStringRef)values[i]);
110 if (CFDataRef bodyData = CFURLRequestCopyHTTPRequestBody(m_cfRequest.get())) {
111 m_httpBody = new FormData(CFDataGetBytePtr(bodyData), CFDataGetLength(bodyData));
113 } else if (CFReadStreamRef bodyStream = CFURLRequestCopyHTTPRequestBodyStream(m_cfRequest.get())) {
114 if (FormData* formData = httpBodyFromStream(bodyStream))
115 m_httpBody = formData;
116 CFRelease(bodyStream);
118 // FIXME: what to do about arbitrary body streams?