Move WebCore into Source
[WebKit-https.git] / Source / WebCore / platform / network / cf / CookieJarCFNet.cpp
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 #include "config.h"
27 #include "CookieJar.h"
28
29 #if USE(CFNETWORK)
30
31 #include "Cookie.h"
32 #include "CookieStorageCFNet.h"
33 #include "Document.h"
34 #include "KURL.h"
35 #include "PlatformString.h"
36 #include "ResourceHandle.h"
37 #include <CFNetwork/CFHTTPCookiesPriv.h>
38 #include <CoreFoundation/CoreFoundation.h>
39 #include <WebKitSystemInterface/WebKitSystemInterface.h>
40 #include <windows.h>
41
42 namespace WebCore {
43
44 static const CFStringRef s_setCookieKeyCF = CFSTR("Set-Cookie");
45 static const CFStringRef s_cookieCF = CFSTR("Cookie");
46
47 static RetainPtr<CFArrayRef> filterCookies(CFArrayRef unfilteredCookies)
48 {
49     CFIndex count = CFArrayGetCount(unfilteredCookies);
50     RetainPtr<CFMutableArrayRef> filteredCookies(AdoptCF, CFArrayCreateMutable(0, count, &kCFTypeArrayCallBacks));
51     for (CFIndex i = 0; i < count; ++i) {
52         CFHTTPCookieRef cookie = (CFHTTPCookieRef)CFArrayGetValueAtIndex(unfilteredCookies, i);
53
54         // <rdar://problem/5632883> CFHTTPCookieStorage would store an empty cookie,
55         // which would be sent as "Cookie: =". We have a workaround in setCookies() to prevent
56         // that, but we also need to avoid sending cookies that were previously stored, and
57         // there's no harm to doing this check because such a cookie is never valid.
58         if (!CFStringGetLength(CFHTTPCookieGetName(cookie)))
59             continue;
60
61         if (CFHTTPCookieIsHTTPOnly(cookie))
62             continue;
63
64         CFArrayAppendValue(filteredCookies.get(), cookie);
65     }
66     return filteredCookies;
67 }
68
69 void setCookies(Document* document, const KURL& url, const String& value)
70 {
71     // <rdar://problem/5632883> CFHTTPCookieStorage stores an empty cookie, which would be sent as "Cookie: =".
72     if (value.isEmpty())
73         return;
74
75     CFHTTPCookieStorageRef cookieStorage = currentCookieStorage();
76     if (!cookieStorage)
77         return;
78
79     RetainPtr<CFURLRef> urlCF(AdoptCF, url.createCFURL());
80     RetainPtr<CFURLRef> firstPartyForCookiesCF(AdoptCF, document->firstPartyForCookies().createCFURL());
81
82     // <http://bugs.webkit.org/show_bug.cgi?id=6531>, <rdar://4409034>
83     // cookiesWithResponseHeaderFields doesn't parse cookies without a value
84     String cookieString = value.contains('=') ? value : value + "=";
85
86     RetainPtr<CFStringRef> cookieStringCF(AdoptCF, cookieString.createCFString());
87     RetainPtr<CFDictionaryRef> headerFieldsCF(AdoptCF, CFDictionaryCreate(kCFAllocatorDefault,
88         (const void**)&s_setCookieKeyCF, (const void**)&cookieStringCF, 1,
89         &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks));
90
91     RetainPtr<CFArrayRef> cookiesCF(AdoptCF, CFHTTPCookieCreateWithResponseHeaderFields(kCFAllocatorDefault,
92         headerFieldsCF.get(), urlCF.get()));
93
94     CFHTTPCookieStorageSetCookies(cookieStorage, filterCookies(cookiesCF.get()).get(), urlCF.get(), firstPartyForCookiesCF.get());
95 }
96
97 String cookies(const Document* /*document*/, const KURL& url)
98 {
99     CFHTTPCookieStorageRef cookieStorage = currentCookieStorage();
100     if (!cookieStorage)
101         return String();
102
103     RetainPtr<CFURLRef> urlCF(AdoptCF, url.createCFURL());
104
105     bool secure = url.protocolIs("https");
106     RetainPtr<CFArrayRef> cookiesCF(AdoptCF, CFHTTPCookieStorageCopyCookiesForURL(cookieStorage, urlCF.get(), secure));
107     RetainPtr<CFDictionaryRef> headerCF(AdoptCF, CFHTTPCookieCopyRequestHeaderFields(kCFAllocatorDefault, filterCookies(cookiesCF.get()).get()));
108     return (CFStringRef)CFDictionaryGetValue(headerCF.get(), s_cookieCF);
109 }
110
111 String cookieRequestHeaderFieldValue(const Document* /*document*/, const KURL& url)
112 {
113     CFHTTPCookieStorageRef cookieStorage = currentCookieStorage();
114     if (!cookieStorage)
115         return String();
116
117     RetainPtr<CFURLRef> urlCF(AdoptCF, url.createCFURL());
118
119     bool secure = url.protocolIs("https");
120     RetainPtr<CFArrayRef> cookiesCF(AdoptCF, CFHTTPCookieStorageCopyCookiesForURL(cookieStorage, urlCF.get(), secure));
121     RetainPtr<CFDictionaryRef> headerCF(AdoptCF, CFHTTPCookieCopyRequestHeaderFields(kCFAllocatorDefault, cookiesCF.get()));
122     return (CFStringRef)CFDictionaryGetValue(headerCF.get(), s_cookieCF);
123 }
124
125 bool cookiesEnabled(const Document* /*document*/)
126 {
127     CFHTTPCookieStorageAcceptPolicy policy = CFHTTPCookieStorageAcceptPolicyOnlyFromMainDocumentDomain;
128     if (CFHTTPCookieStorageRef cookieStorage = currentCookieStorage())
129         policy = CFHTTPCookieStorageGetCookieAcceptPolicy(cookieStorage);
130     return policy == CFHTTPCookieStorageAcceptPolicyOnlyFromMainDocumentDomain || policy == CFHTTPCookieStorageAcceptPolicyAlways;
131 }
132
133 bool getRawCookies(const Document*, const KURL& url, Vector<Cookie>& rawCookies)
134 {
135     rawCookies.clear();
136     CFHTTPCookieStorageRef cookieStorage = currentCookieStorage();
137     if (!cookieStorage)
138         return false;
139
140     RetainPtr<CFURLRef> urlCF(AdoptCF, url.createCFURL());
141
142     bool sendSecureCookies = url.protocolIs("https");
143     RetainPtr<CFArrayRef> cookiesCF(AdoptCF, CFHTTPCookieStorageCopyCookiesForURL(cookieStorage, urlCF.get(), sendSecureCookies));
144
145     CFIndex count = CFArrayGetCount(cookiesCF.get());
146     rawCookies.reserveCapacity(count);
147
148     for (CFIndex i = 0; i < count; i++) {
149        CFHTTPCookieRef cookie = (CFHTTPCookieRef)CFArrayGetValueAtIndex(cookiesCF.get(), i);
150        String name = CFHTTPCookieGetName(cookie);
151        String value = CFHTTPCookieGetValue(cookie);
152        String domain = CFHTTPCookieGetDomain(cookie);
153        String path = CFHTTPCookieGetPath(cookie);
154
155        double expires = (CFDateGetAbsoluteTime(CFHTTPCookieGetExpiratonDate(cookie)) + kCFAbsoluteTimeIntervalSince1970) * 1000;
156
157        bool httpOnly = CFHTTPCookieIsHTTPOnly(cookie);
158        bool secure = CFHTTPCookieIsSecure(cookie);
159        bool session = false;    // FIXME: Need API for if a cookie is a session cookie.
160
161        rawCookies.uncheckedAppend(Cookie(name, value, domain, path, expires, httpOnly, secure, session));
162     }
163
164     return true;
165 }
166
167 void deleteCookie(const Document*, const KURL& url, const String& name)
168 {
169     CFHTTPCookieStorageRef cookieStorage = currentCookieStorage();
170     if (!cookieStorage)
171         return;
172
173     RetainPtr<CFURLRef> urlCF(AdoptCF, url.createCFURL());
174
175     bool sendSecureCookies = url.protocolIs("https");
176     RetainPtr<CFArrayRef> cookiesCF(AdoptCF, CFHTTPCookieStorageCopyCookiesForURL(cookieStorage, urlCF.get(), sendSecureCookies));
177
178     CFIndex count = CFArrayGetCount(cookiesCF.get());
179     for (CFIndex i = 0; i < count; i++) {
180         CFHTTPCookieRef cookie = (CFHTTPCookieRef)CFArrayGetValueAtIndex(cookiesCF.get(), i);
181         String cookieName = CFHTTPCookieGetName(cookie);
182         if (cookieName == name) {
183             CFHTTPCookieStorageDeleteCookie(cookieStorage, cookie);
184             break;
185         }
186     }
187 }
188
189 } // namespace WebCore
190
191 #endif // USE(CFNETWORK)