CookieManager only works with the default session
[WebKit-https.git] / Source / WebCore / platform / network / cf / CookieStorageCFNet.cpp
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 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 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 "CookieStorage.h"
28
29 #include "NetworkStorageSession.h"
30 #include <wtf/HashMap.h>
31 #include <wtf/MainThread.h>
32 #include <wtf/NeverDestroyed.h>
33
34 #if PLATFORM(COCOA)
35 #include "WebCoreSystemInterface.h"
36 #elif PLATFORM(WIN)
37 #include "LoaderRunLoopCF.h"
38 #include <CFNetwork/CFHTTPCookiesPriv.h>
39 #include <WebKitSystemInterface/WebKitSystemInterface.h>
40 #endif
41
42 namespace WebCore {
43
44 #if PLATFORM(WIN)
45
46 static HashMap<CFHTTPCookieStorageRef, std::function<void ()>>& cookieChangeCallbackMap()
47 {
48     static NeverDestroyed<HashMap<CFHTTPCookieStorageRef, std::function<void ()>>> map;
49     return map;
50 }
51
52 static void notifyCookiesChanged(CFHTTPCookieStorageRef cookieStorage, void *)
53 {
54     callOnMainThread([cookieStorage] {
55         if (auto callback = cookieChangeCallbackMap().get(cookieStorage))
56             callback();
57     });
58 }
59
60 static inline CFRunLoopRef cookieStorageObserverRunLoop()
61 {
62     // We're using the loader run loop because we need a CFRunLoop to 
63     // call the CFNetwork cookie storage APIs with. Re-using the loader
64     // run loop is less overhead than starting a new thread to just listen
65     // for changes in cookies.
66     
67     // FIXME: The loaderRunLoop function name should be a little more generic.
68     return loaderRunLoop();
69 }
70
71 void startObservingCookieChanges(const NetworkStorageSession& storageSession, std::function<void ()>&& callback)
72 {
73     ASSERT(isMainThread());
74
75     CFRunLoopRef runLoop = cookieStorageObserverRunLoop();
76     ASSERT(runLoop);
77
78     RetainPtr<CFHTTPCookieStorageRef> cookieStorage = storageSession.cookieStorage();
79     ASSERT(cookieStorage);
80
81     ASSERT(cookieChangeCallbackMap().contains(cookieStorage.get()));
82     cookieChangeCallbackMap().add(cookieStorage.get(), WTFMove(callback));
83
84     CFHTTPCookieStorageScheduleWithRunLoop(cookieStorage.get(), runLoop, kCFRunLoopCommonModes);
85     CFHTTPCookieStorageAddObserver(cookieStorage.get(), runLoop, kCFRunLoopDefaultMode, notifyCookiesChanged, 0);
86 }
87
88 void stopObservingCookieChanges(const NetworkStorageSession& storageSession)
89 {
90     ASSERT(isMainThread());
91
92     CFRunLoopRef runLoop = cookieStorageObserverRunLoop();
93     ASSERT(runLoop);
94
95     RetainPtr<CFHTTPCookieStorageRef> cookieStorage = storageSession.cookieStorage();
96     ASSERT(cookieStorage);
97
98     cookieChangeCallbackMap().remove(cookieStorage.get());
99
100     CFHTTPCookieStorageRemoveObserver(cookieStorage.get(), runLoop, kCFRunLoopDefaultMode, notifyCookiesChanged, 0);
101     CFHTTPCookieStorageUnscheduleFromRunLoop(cookieStorage.get(), runLoop, kCFRunLoopCommonModes);
102 }
103
104 #endif // PLATFORM(WIN)
105
106 } // namespace WebCore