1/* 2 * Copyright (C) 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 "CookieStorageCFNet.h" 28 29#if USE(CFNETWORK) 30 31#include "LoaderRunLoopCF.h" 32#include "ResourceHandle.h" 33#include <CFNetwork/CFHTTPCookiesPriv.h> 34#include <WebKitSystemInterface/WebKitSystemInterface.h> 35#include <wtf/MainThread.h> 36#include <wtf/RetainPtr.h> 37 38#if USE(PLATFORM_STRATEGIES) 39#include "CookiesStrategy.h" 40#include "PlatformStrategies.h" 41#endif 42 43namespace WebCore { 44 45static RetainPtr<CFHTTPCookieStorageRef>& privateCookieStorage() 46{ 47 DEFINE_STATIC_LOCAL(RetainPtr<CFHTTPCookieStorageRef>, cookieStorage, ()); 48 return cookieStorage; 49} 50 51CFHTTPCookieStorageRef currentCookieStorage() 52{ 53 ASSERT(isMainThread()); 54 55 if (CFHTTPCookieStorageRef cookieStorage = privateCookieStorage().get()) 56 return cookieStorage; 57 return wkGetDefaultHTTPCookieStorage(); 58} 59 60CFHTTPCookieStorageRef privateBrowsingCookieStorage() 61{ 62 return privateCookieStorage().get(); 63} 64 65void setCurrentCookieStorage(CFHTTPCookieStorageRef cookieStorage) 66{ 67 ASSERT(isMainThread()); 68 69 privateCookieStorage().adoptCF(cookieStorage); 70} 71 72void setCookieStoragePrivateBrowsingEnabled(bool enabled) 73{ 74 ASSERT(isMainThread()); 75 76 if (!enabled) { 77 privateCookieStorage() = nullptr; 78 return; 79 } 80 81#if USE(CFURLSTORAGESESSIONS) 82 if (CFURLStorageSessionRef privateStorageSession = ResourceHandle::privateBrowsingStorageSession()) 83 privateCookieStorage().adoptCF(wkCopyHTTPCookieStorage(privateStorageSession)); 84 else 85#endif 86 privateCookieStorage().adoptCF(wkCreateInMemoryHTTPCookieStorage()); 87} 88 89CFHTTPCookieStorageRef defaultCookieStorage() 90{ 91 return wkGetDefaultHTTPCookieStorage(); 92} 93 94static void notifyCookiesChangedOnMainThread(void* context) 95{ 96 ASSERT(isMainThread()); 97 98#if USE(PLATFORM_STRATEGIES) 99 platformStrategies()->cookiesStrategy()->notifyCookiesChanged(); 100#endif 101} 102 103static void notifyCookiesChanged(CFHTTPCookieStorageRef inStorage, void *context) 104{ 105 callOnMainThread(notifyCookiesChangedOnMainThread, 0); 106} 107 108static inline CFRunLoopRef cookieStorageObserverRunLoop() 109{ 110 // We're using the loader run loop because we need a CFRunLoop to 111 // call the CFNetwork cookie storage APIs with. Re-using the loader 112 // run loop is less overhead than starting a new thread to just listen 113 // for changes in cookies. 114 115 // FIXME: The loaderRunLoop function name should be a little more generic. 116 return loaderRunLoop(); 117} 118 119void startObservingCookieChanges() 120{ 121 ASSERT(isMainThread()); 122 123 CFRunLoopRef runLoop = cookieStorageObserverRunLoop(); 124 ASSERT(runLoop); 125 126 CFHTTPCookieStorageRef cookieStorage = currentCookieStorage(); 127 ASSERT(cookieStorage); 128 129 CFHTTPCookieStorageScheduleWithRunLoop(cookieStorage, runLoop, kCFRunLoopCommonModes); 130 CFHTTPCookieStorageAddObserver(cookieStorage, runLoop, kCFRunLoopDefaultMode, notifyCookiesChanged, 0); 131} 132 133void stopObservingCookieChanges() 134{ 135 ASSERT(isMainThread()); 136 137 CFRunLoopRef runLoop = cookieStorageObserverRunLoop(); 138 ASSERT(runLoop); 139 140 CFHTTPCookieStorageRef cookieStorage = currentCookieStorage(); 141 ASSERT(cookieStorage); 142 143 CFHTTPCookieStorageRemoveObserver(cookieStorage, runLoop, kCFRunLoopDefaultMode, notifyCookiesChanged, 0); 144 CFHTTPCookieStorageUnscheduleFromRunLoop(cookieStorage, runLoop, kCFRunLoopCommonModes); 145} 146 147} // namespace WebCore 148 149#endif // USE(CFNETWORK) 150