Use sendWithAsyncReply for cookie getting/clearing functions
[WebKit.git] / Source / WebKit / NetworkProcess / Cookies / WebCookieManager.cpp
1 /*
2  * Copyright (C) 2011, 2013 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. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "WebCookieManager.h"
28
29 #include "NetworkProcess.h"
30 #include "WebCookieManagerMessages.h"
31 #include "WebCookieManagerProxyMessages.h"
32 #include "WebCoreArgumentCoders.h"
33 #include <WebCore/Cookie.h>
34 #include <WebCore/CookieStorage.h>
35 #include <WebCore/NetworkStorageSession.h>
36 #include <wtf/MainThread.h>
37 #include <wtf/URL.h>
38 #include <wtf/text/StringHash.h>
39 #include <wtf/text/WTFString.h>
40
41 namespace WebKit {
42 using namespace WebCore;
43
44 const char* WebCookieManager::supplementName()
45 {
46     return "WebCookieManager";
47 }
48
49 WebCookieManager::WebCookieManager(NetworkProcess& process)
50     : m_process(process)
51 {
52     m_process.addMessageReceiver(Messages::WebCookieManager::messageReceiverName(), *this);
53 }
54
55 WebCookieManager::~WebCookieManager() = default;
56
57 void WebCookieManager::getHostnamesWithCookies(PAL::SessionID sessionID, CompletionHandler<void(Vector<String>&&)>&& completionHandler)
58 {
59     HashSet<String> hostnames;
60     if (auto* storageSession = m_process.storageSession(sessionID))
61         storageSession->getHostnamesWithCookies(hostnames);
62     completionHandler(copyToVector(hostnames));
63 }
64
65 void WebCookieManager::deleteCookiesForHostnames(PAL::SessionID sessionID, const Vector<String>& hostnames)
66 {
67     if (auto* storageSession = m_process.storageSession(sessionID))
68         storageSession->deleteCookiesForHostnames(hostnames);
69 }
70
71 void WebCookieManager::deleteAllCookies(PAL::SessionID sessionID)
72 {
73     if (auto* storageSession = m_process.storageSession(sessionID))
74         storageSession->deleteAllCookies();
75 }
76
77 void WebCookieManager::deleteCookie(PAL::SessionID sessionID, const Cookie& cookie, CompletionHandler<void()>&& completionHandler)
78 {
79     if (auto* storageSession = m_process.storageSession(sessionID))
80         storageSession->deleteCookie(cookie);
81     completionHandler();
82 }
83
84 void WebCookieManager::deleteAllCookiesModifiedSince(PAL::SessionID sessionID, WallTime time, CompletionHandler<void()>&& completionHandler)
85 {
86     if (auto* storageSession = m_process.storageSession(sessionID))
87         storageSession->deleteAllCookiesModifiedSince(time);
88     completionHandler();
89 }
90
91 void WebCookieManager::getAllCookies(PAL::SessionID sessionID, CompletionHandler<void(Vector<WebCore::Cookie>&&)>&& completionHandler)
92 {
93     Vector<Cookie> cookies;
94     if (auto* storageSession = m_process.storageSession(sessionID))
95         cookies = storageSession->getAllCookies();
96     completionHandler(WTFMove(cookies));
97 }
98
99 void WebCookieManager::getCookies(PAL::SessionID sessionID, const URL& url, CompletionHandler<void(Vector<WebCore::Cookie>&&)>&& completionHandler)
100 {
101     Vector<Cookie> cookies;
102     if (auto* storageSession = m_process.storageSession(sessionID))
103         cookies = storageSession->getCookies(url);
104     completionHandler(WTFMove(cookies));
105 }
106
107 void WebCookieManager::setCookie(PAL::SessionID sessionID, const Vector<Cookie>& cookies, CompletionHandler<void()>&& completionHandler)
108 {
109     if (auto* storageSession = m_process.storageSession(sessionID)) {
110         for (auto& cookie : cookies)
111             storageSession->setCookie(cookie);
112     }
113     completionHandler();
114 }
115
116 void WebCookieManager::setCookies(PAL::SessionID sessionID, const Vector<Cookie>& cookies, const URL& url, const URL& mainDocumentURL, CompletionHandler<void()>&& completionHandler)
117 {
118     if (auto* storageSession = m_process.storageSession(sessionID))
119         storageSession->setCookies(cookies, url, mainDocumentURL);
120     completionHandler();
121 }
122
123 void WebCookieManager::notifyCookiesDidChange(PAL::SessionID sessionID)
124 {
125     ASSERT(RunLoop::isMain());
126     m_process.send(Messages::WebCookieManagerProxy::CookiesDidChange(sessionID), 0);
127 }
128
129 void WebCookieManager::startObservingCookieChanges(PAL::SessionID sessionID)
130 {
131     if (auto* storageSession = m_process.storageSession(sessionID)) {
132         WebCore::startObservingCookieChanges(*storageSession, [this, sessionID] {
133             notifyCookiesDidChange(sessionID);
134         });
135     }
136 }
137
138 void WebCookieManager::stopObservingCookieChanges(PAL::SessionID sessionID)
139 {
140     if (auto* storageSession = m_process.storageSession(sessionID))
141         WebCore::stopObservingCookieChanges(*storageSession);
142 }
143
144 void WebCookieManager::setHTTPCookieAcceptPolicy(HTTPCookieAcceptPolicy policy, CompletionHandler<void()>&& completionHandler)
145 {
146     platformSetHTTPCookieAcceptPolicy(policy);
147
148     completionHandler();
149 }
150
151 void WebCookieManager::getHTTPCookieAcceptPolicy(CompletionHandler<void(HTTPCookieAcceptPolicy)>&& completionHandler)
152 {
153     completionHandler(platformGetHTTPCookieAcceptPolicy());
154 }
155
156 } // namespace WebKit