WebCookieManager.cpp [plain text]
#include "config.h"
#include "WebCookieManager.h"
#include "NetworkProcess.h"
#include "WebCookieManagerMessages.h"
#include "WebCookieManagerProxyMessages.h"
#include "WebCoreArgumentCoders.h"
#include <WebCore/Cookie.h>
#include <WebCore/CookieStorage.h>
#include <WebCore/NetworkStorageSession.h>
#include <wtf/MainThread.h>
#include <wtf/URL.h>
#include <wtf/text/StringHash.h>
#include <wtf/text/WTFString.h>
namespace WebKit {
using namespace WebCore;
const char* WebCookieManager::supplementName()
{
return "WebCookieManager";
}
WebCookieManager::WebCookieManager(NetworkProcess& process)
: m_process(process)
{
m_process.addMessageReceiver(Messages::WebCookieManager::messageReceiverName(), *this);
}
WebCookieManager::~WebCookieManager() = default;
void WebCookieManager::getHostnamesWithCookies(PAL::SessionID sessionID, CompletionHandler<void(Vector<String>&&)>&& completionHandler)
{
HashSet<String> hostnames;
if (auto* storageSession = m_process.storageSession(sessionID))
storageSession->getHostnamesWithCookies(hostnames);
completionHandler(copyToVector(hostnames));
}
void WebCookieManager::deleteCookiesForHostnames(PAL::SessionID sessionID, const Vector<String>& hostnames)
{
if (auto* storageSession = m_process.storageSession(sessionID))
storageSession->deleteCookiesForHostnames(hostnames);
}
void WebCookieManager::deleteAllCookies(PAL::SessionID sessionID)
{
if (auto* storageSession = m_process.storageSession(sessionID))
storageSession->deleteAllCookies();
}
void WebCookieManager::deleteCookie(PAL::SessionID sessionID, const Cookie& cookie, CompletionHandler<void()>&& completionHandler)
{
if (auto* storageSession = m_process.storageSession(sessionID))
storageSession->deleteCookie(cookie);
completionHandler();
}
void WebCookieManager::deleteAllCookiesModifiedSince(PAL::SessionID sessionID, WallTime time, CompletionHandler<void()>&& completionHandler)
{
if (auto* storageSession = m_process.storageSession(sessionID))
storageSession->deleteAllCookiesModifiedSince(time);
completionHandler();
}
void WebCookieManager::getAllCookies(PAL::SessionID sessionID, CompletionHandler<void(Vector<WebCore::Cookie>&&)>&& completionHandler)
{
Vector<Cookie> cookies;
if (auto* storageSession = m_process.storageSession(sessionID))
cookies = storageSession->getAllCookies();
completionHandler(WTFMove(cookies));
}
void WebCookieManager::getCookies(PAL::SessionID sessionID, const URL& url, CompletionHandler<void(Vector<WebCore::Cookie>&&)>&& completionHandler)
{
Vector<Cookie> cookies;
if (auto* storageSession = m_process.storageSession(sessionID))
cookies = storageSession->getCookies(url);
completionHandler(WTFMove(cookies));
}
void WebCookieManager::setCookie(PAL::SessionID sessionID, const Vector<Cookie>& cookies, CompletionHandler<void()>&& completionHandler)
{
if (auto* storageSession = m_process.storageSession(sessionID)) {
for (auto& cookie : cookies)
storageSession->setCookie(cookie);
}
completionHandler();
}
void WebCookieManager::setCookies(PAL::SessionID sessionID, const Vector<Cookie>& cookies, const URL& url, const URL& mainDocumentURL, CompletionHandler<void()>&& completionHandler)
{
if (auto* storageSession = m_process.storageSession(sessionID))
storageSession->setCookies(cookies, url, mainDocumentURL);
completionHandler();
}
void WebCookieManager::notifyCookiesDidChange(PAL::SessionID sessionID)
{
ASSERT(RunLoop::isMain());
m_process.send(Messages::WebCookieManagerProxy::CookiesDidChange(sessionID), 0);
}
void WebCookieManager::startObservingCookieChanges(PAL::SessionID sessionID)
{
if (auto* storageSession = m_process.storageSession(sessionID)) {
WebCore::startObservingCookieChanges(*storageSession, [this, sessionID] {
notifyCookiesDidChange(sessionID);
});
}
}
void WebCookieManager::stopObservingCookieChanges(PAL::SessionID sessionID)
{
if (auto* storageSession = m_process.storageSession(sessionID))
WebCore::stopObservingCookieChanges(*storageSession);
}
void WebCookieManager::setHTTPCookieAcceptPolicy(HTTPCookieAcceptPolicy policy, CompletionHandler<void()>&& completionHandler)
{
platformSetHTTPCookieAcceptPolicy(policy);
m_process.cookieAcceptPolicyChanged(policy);
completionHandler();
}
void WebCookieManager::getHTTPCookieAcceptPolicy(CompletionHandler<void(HTTPCookieAcceptPolicy)>&& completionHandler)
{
completionHandler(m_process.defaultStorageSession().cookieAcceptPolicy());
}
}