String cookies(const KURL& url)
{
#if USE(CFNETWORK)
CFHTTPCookieStorageRef defaultCookieStorage = wkGetDefaultHTTPCookieStorage();
if (!defaultCookieStorage)
return String();
String cookieString;
RetainPtr<CFURLRef> urlCF(AdoptCF, url.createCFURL());
bool secure = equalIgnoringCase(url.protocol(), "https");
RetainPtr<CFArrayRef> cookiesCF(AdoptCF, CFHTTPCookieStorageCopyCookiesForURL(defaultCookieStorage, urlCF.get(), secure));
RetainPtr<CFDictionaryRef> headerCF(AdoptCF, CFHTTPCookieCopyRequestHeaderFields(kCFAllocatorDefault, cookiesCF.get()));
return (CFStringRef)CFDictionaryGetValue(headerCF.get(), s_cookieCF);
#else
DeprecatedString str = url.url();
str.append((UChar)'\0');
DWORD count = str.length();
InternetGetCookie((UChar*)str.unicode(), 0, 0, &count);
if (count <= 1) // Null terminator counts as 1.
return String();
UChar* buffer = new UChar[count];
InternetGetCookie((UChar*)str.unicode(), 0, buffer, &count);
String& result = String(buffer, count-1); // Ignore the null terminator.
delete[] buffer;
return result;
#endif
}
void setCookies(const KURL& url, const KURL& policyURL, const String& value)
{
#if USE(CFNETWORK)
CFHTTPCookieStorageRef defaultCookieStorage = wkGetDefaultHTTPCookieStorage();
if (!defaultCookieStorage)
return;
RetainPtr<CFURLRef> urlCF(AdoptCF, url.createCFURL());
RetainPtr<CFURLRef> policyURLCF(AdoptCF, policyURL.createCFURL());
// <http://bugzilla.opendarwin.org/show_bug.cgi?id=6531>, <rdar://4409034>
// cookiesWithResponseHeaderFields doesn't parse cookies without a value
String cookieString = value.contains('=') ? value : value + "=";
RetainPtr<CFStringRef> cookieStringCF(AdoptCF, cookieString.createCFString());
RetainPtr<CFDictionaryRef> headerFieldsCF(AdoptCF, CFDictionaryCreate(kCFAllocatorDefault, (const void**)&s_setCookieKeyCF,
(const void**)&cookieStringCF, 1, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks));
RetainPtr<CFArrayRef> cookiesCF(AdoptCF, CFHTTPCookieCreateWithResponseHeaderFields(kCFAllocatorDefault,
headerFieldsCF.get(), urlCF.get()));
CFHTTPCookieStorageSetCookies(defaultCookieStorage, cookiesCF.get(), urlCF.get(), policyURLCF.get());
#else
// FIXME: Deal with the policy URL.
DeprecatedString str = url.url();
str.append((UChar)'\0');
DeprecatedString val = value.deprecatedString();
val.append((UChar)'\0');
InternetSetCookie((UChar*)str.unicode(), 0, (UChar*)val.unicode());
#endif
}
bool cookiesEnabled()
{
CFHTTPCookieStorageAcceptPolicy policy = CFHTTPCookieStorageAcceptPolicyOnlyFromMainDocumentDomain;
if (CFHTTPCookieStorageRef defaultCookieStorage = wkGetDefaultHTTPCookieStorage())
policy = CFHTTPCookieStorageGetCookieAcceptPolicy(defaultCookieStorage);
return policy == CFHTTPCookieStorageAcceptPolicyOnlyFromMainDocumentDomain || policy == CFHTTPCookieStorageAcceptPolicyAlways;
}
RetainPtr<CFHTTPCookieStorageRef> NetworkStorageSession::cookieStorage() const
{
if (m_platformSession)
return adoptCF(wkCopyHTTPCookieStorage(m_platformSession.get()));
#if USE(CFNETWORK)
return wkGetDefaultHTTPCookieStorage();
#else
// When using NSURLConnection, we also use its shared cookie storage.
return 0;
#endif
}
RetainPtr<CFHTTPCookieStorageRef> currentCFHTTPCookieStorage()
{
#if PLATFORM(WIN)
if (RetainPtr<CFHTTPCookieStorageRef>& override = cookieStorageOverride())
return override;
#endif
if (CFURLStorageSessionRef session = ResourceHandle::currentStorageSession())
return RetainPtr<CFHTTPCookieStorageRef>(AdoptCF, wkCopyHTTPCookieStorage(session));
#if USE(CFNETWORK)
return wkGetDefaultHTTPCookieStorage();
#else
// When using NSURLConnection, we also use its default cookie storage.
return 0;
#endif
}
CFHTTPCookieStorageRef currentCookieStorage()
{
if (s_cookieStorage)
return s_cookieStorage.get();
return wkGetDefaultHTTPCookieStorage();
}
