WebCookieCache.cpp [plain text]
#include "config.h"
#include "WebCookieCache.h"
#include "NetworkConnectionToWebProcessMessages.h"
#include "NetworkProcessConnection.h"
#include "WebProcess.h"
namespace WebKit {
using namespace WebCore;
bool WebCookieCache::isSupported()
{
#if HAVE(COOKIE_CHANGE_LISTENER_API)
return inMemoryStorageSession().supportsCookieChangeListenerAPI();
#else
return false;
#endif
}
String WebCookieCache::cookiesForDOM(const URL& firstParty, const SameSiteInfo& sameSiteInfo, const URL& url, FrameIdentifier frameID, PageIdentifier pageID, IncludeSecureCookies includeSecureCookies)
{
String host = url.host().toString();
if (!m_hostsWithInMemoryStorage.contains(host)) {
Vector<Cookie> cookies;
bool subscribeToCookieChangeNotifications = true;
if (!WebProcess::singleton().ensureNetworkProcessConnection().connection().sendSync(Messages::NetworkConnectionToWebProcess::DomCookiesForHost(url.host().toString(), subscribeToCookieChangeNotifications), Messages::NetworkConnectionToWebProcess::DomCookiesForHost::Reply(cookies), 0))
return { };
pruneCacheIfNecessary();
m_hostsWithInMemoryStorage.add(host);
for (auto& cookie : cookies)
inMemoryStorageSession().setCookie(cookie);
}
return inMemoryStorageSession().cookiesForDOM(firstParty, sameSiteInfo, url, frameID, pageID, includeSecureCookies, ShouldAskITP::No, ShouldRelaxThirdPartyCookieBlocking::No).first;
}
void WebCookieCache::setCookiesFromDOM(const URL& firstParty, const SameSiteInfo& sameSiteInfo, const URL& url, FrameIdentifier frameID, PageIdentifier pageID, const String& cookieString, ShouldRelaxThirdPartyCookieBlocking shouldRelaxThirdPartyCookieBlocking)
{
String host = url.host().toString();
if (m_hostsWithInMemoryStorage.contains(host))
inMemoryStorageSession().setCookiesFromDOM(firstParty, sameSiteInfo, url, frameID, pageID, ShouldAskITP::No, cookieString, shouldRelaxThirdPartyCookieBlocking);
}
void WebCookieCache::cookiesAdded(const String& host, const Vector<Cookie>& cookies)
{
if (!m_hostsWithInMemoryStorage.contains(host))
return;
for (auto& cookie : cookies)
inMemoryStorageSession().setCookie(cookie);
}
void WebCookieCache::cookiesDeleted(const String& host, const Vector<WebCore::Cookie>& cookies)
{
if (!m_hostsWithInMemoryStorage.contains(host))
return;
for (auto& cookie : cookies)
inMemoryStorageSession().deleteCookie(cookie);
}
void WebCookieCache::allCookiesDeleted()
{
clear();
}
void WebCookieCache::clear()
{
#if HAVE(COOKIE_CHANGE_LISTENER_API)
if (!m_hostsWithInMemoryStorage.isEmpty())
WebProcess::singleton().ensureNetworkProcessConnection().connection().send(Messages::NetworkConnectionToWebProcess::UnsubscribeFromCookieChangeNotifications(m_hostsWithInMemoryStorage), 0);
#endif
m_hostsWithInMemoryStorage.clear();
m_inMemoryStorageSession = nullptr;
}
void WebCookieCache::clearForHost(const String& host)
{
String removedHost = m_hostsWithInMemoryStorage.take(host);
if (removedHost.isNull())
return;
inMemoryStorageSession().deleteCookiesForHostnames(Vector<String> { removedHost });
#if HAVE(COOKIE_CHANGE_LISTENER_API)
WebProcess::singleton().ensureNetworkProcessConnection().connection().send(Messages::NetworkConnectionToWebProcess::UnsubscribeFromCookieChangeNotifications(HashSet<String> { removedHost }), 0);
#endif
}
void WebCookieCache::pruneCacheIfNecessary()
{
static const unsigned maxCachedHosts = 5;
while (m_hostsWithInMemoryStorage.size() >= maxCachedHosts)
clearForHost(*m_hostsWithInMemoryStorage.random());
}
#if !PLATFORM(COCOA)
NetworkStorageSession& WebCookieCache::inMemoryStorageSession()
{
ASSERT_NOT_IMPLEMENTED_YET();
return *m_inMemoryStorageSession;
}
#endif
}