WebCore:
[WebKit-https.git] / WebCore / platform / network / cf / ResourceRequestCFNet.cpp
1 /*
2  * Copyright (C) 2006, 2007 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 #include "config.h"
27 #include "ResourceRequestCFNet.h"
28
29 #include "FormDataStreamCFNet.h"
30 #include "ResourceRequest.h"
31
32 #include <CFNetwork/CFURLRequestPriv.h>
33
34 namespace WebCore {
35
36 CFURLRequestRef ResourceRequest::cfURLRequest() const
37 {
38     updatePlatformRequest();
39
40     return m_cfRequest.get();
41 }
42
43 static inline void addHeadersFromHashMap(CFMutableURLRequestRef request, const HTTPHeaderMap& requestHeaders) 
44 {
45     if (!requestHeaders.size())
46         return;
47         
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);
53         CFRelease(key);
54         CFRelease(value);
55     }
56 }
57
58 void ResourceRequest::doUpdatePlatformRequest()
59 {
60     CFMutableURLRequestRef cfRequest;
61
62     if (m_cfRequest) 
63         cfRequest = CFURLRequestCreateMutableCopy(0, m_cfRequest.get());
64     else {
65         RetainPtr<CFURLRef> url(AdoptCF, ResourceRequest::url().createCFURL());
66         RetainPtr<CFURLRef> mainDocumentURL(AdoptCF, ResourceRequest::mainDocumentURL().createCFURL());
67
68         cfRequest = CFURLRequestCreateMutable(0, url.get(), (CFURLRequestCachePolicy)cachePolicy(), timeoutInterval(), mainDocumentURL.get());
69     }
70
71     RetainPtr<CFStringRef> requestMethod(AdoptCF, httpMethod().createCFString());
72     CFURLRequestSetHTTPRequestMethod(cfRequest, requestMethod.get());
73
74     addHeadersFromHashMap(cfRequest, httpHeaderFields());
75     WebCore::setHTTPBody(cfRequest, httpBody());
76     CFURLRequestSetShouldHandleHTTPCookies(cfRequest, allowHTTPCookies());
77
78     if (m_cfRequest) {
79         CFURLRequestSetHTTPCookieStorageAcceptPolicy(cfRequest, CFURLRequestGetHTTPCookieStorageAcceptPolicy(m_cfRequest.get()));
80         CFURLRequestSetSSLProperties(cfRequest, CFURLRequestGetSSLProperties(m_cfRequest.get()));
81     }
82
83     m_cfRequest.adoptCF(cfRequest);
84 }
85
86 void ResourceRequest::doUpdateResourceRequest()
87 {
88     m_url = CFURLRequestGetURL(m_cfRequest.get());
89
90     m_cachePolicy = (ResourceRequestCachePolicy)CFURLRequestGetCachePolicy(m_cfRequest.get());
91     m_timeoutInterval = CFURLRequestGetTimeoutInterval(m_cfRequest.get());
92     m_mainDocumentURL = CFURLRequestGetMainDocumentURL(m_cfRequest.get());
93     if (CFStringRef method = CFURLRequestCopyHTTPRequestMethod(m_cfRequest.get())) {
94         m_httpMethod = method;
95         CFRelease(method);
96     }
97     m_allowHTTPCookies = CFURLRequestShouldHandleHTTPCookies(m_cfRequest.get());
98
99     if (CFDictionaryRef headers = CFURLRequestCopyAllHTTPHeaderFields(m_cfRequest.get())) {
100         CFIndex headerCount = CFDictionaryGetCount(headers);
101         Vector<const void*, 128> keys(headerCount);
102         Vector<const void*, 128> values(headerCount);
103         CFDictionaryGetKeysAndValues(headers, keys.data(), values.data());
104         for (int i = 0; i < headerCount; ++i)
105             m_httpHeaderFields.set((CFStringRef)keys[i], (CFStringRef)values[i]);
106         CFRelease(headers);
107     }
108
109     if (CFDataRef bodyData = CFURLRequestCopyHTTPRequestBody(m_cfRequest.get())) {
110         m_httpBody = new FormData(CFDataGetBytePtr(bodyData), CFDataGetLength(bodyData));
111         CFRelease(bodyData);
112     } else if (CFReadStreamRef bodyStream = CFURLRequestCopyHTTPRequestBodyStream(m_cfRequest.get())) {
113         if (FormData* formData = httpBodyFromStream(bodyStream))
114             m_httpBody = formData;
115         CFRelease(bodyStream);
116     }
117     // FIXME: what to do about arbitrary body streams?
118 }
119
120 }