#include "config.h"
#include "WebCookieJar.h"
#include "NetworkConnectionToWebProcessMessages.h"
#include "NetworkProcessConnection.h"
#include "WebProcess.h"
#include <WebCore/CookieRequestHeaderFieldProxy.h>
#include <WebCore/Document.h>
#include <WebCore/Frame.h>
#include <WebCore/FrameLoader.h>
#include <WebCore/FrameLoaderClient.h>
#include <WebCore/StorageSessionProvider.h>
namespace WebKit {
using namespace WebCore;
class WebStorageSessionProvider : public WebCore::StorageSessionProvider {
WebCore::NetworkStorageSession* storageSession() const final { return nullptr; }
};
WebCookieJar::WebCookieJar()
: WebCore::CookieJar(adoptRef(*new WebStorageSessionProvider)) { }
String WebCookieJar::cookies(WebCore::Document& document, const URL& url) const
{
Optional<FrameIdentifier> frameID;
Optional<PageIdentifier> pageID;
if (auto* frame = document.frame()) {
frameID = frame->loader().client().frameID();
pageID = frame->loader().client().pageID();
}
String cookieString;
bool secureCookiesAccessed = false;
if (!WebProcess::singleton().ensureNetworkProcessConnection().connection().sendSync(Messages::NetworkConnectionToWebProcess::CookiesForDOM(document.firstPartyForCookies(), sameSiteInfo(document), url, frameID, pageID, shouldIncludeSecureCookies(document, url)), Messages::NetworkConnectionToWebProcess::CookiesForDOM::Reply(cookieString, secureCookiesAccessed), 0))
return { };
return cookieString;
}
void WebCookieJar::setCookies(WebCore::Document& document, const URL& url, const String& cookieString)
{
Optional<FrameIdentifier> frameID;
Optional<PageIdentifier> pageID;
if (auto* frame = document.frame()) {
frameID = frame->loader().client().frameID();
pageID = frame->loader().client().pageID();
}
WebProcess::singleton().ensureNetworkProcessConnection().connection().send(Messages::NetworkConnectionToWebProcess::SetCookiesFromDOM(document.firstPartyForCookies(), sameSiteInfo(document), url, frameID, pageID, cookieString), 0);
}
bool WebCookieJar::cookiesEnabled(const WebCore::Document& document) const
{
bool result = false;
if (!WebProcess::singleton().ensureNetworkProcessConnection().connection().sendSync(Messages::NetworkConnectionToWebProcess::CookiesEnabled(), Messages::NetworkConnectionToWebProcess::CookiesEnabled::Reply(result), 0))
return false;
return result;
}
std::pair<String, WebCore::SecureCookiesAccessed> WebCookieJar::cookieRequestHeaderFieldValue(const URL& firstParty, const WebCore::SameSiteInfo& sameSiteInfo, const URL& url, Optional<FrameIdentifier> frameID, Optional<PageIdentifier> pageID, WebCore::IncludeSecureCookies includeSecureCookies) const
{
String cookieString;
bool secureCookiesAccessed = false;
if (!WebProcess::singleton().ensureNetworkProcessConnection().connection().sendSync(Messages::NetworkConnectionToWebProcess::CookieRequestHeaderFieldValue(firstParty, sameSiteInfo, url, frameID, pageID, includeSecureCookies), Messages::NetworkConnectionToWebProcess::CookieRequestHeaderFieldValue::Reply(cookieString, secureCookiesAccessed), 0))
return { };
return { cookieString, secureCookiesAccessed ? WebCore::SecureCookiesAccessed::Yes : WebCore::SecureCookiesAccessed::No };
}
bool WebCookieJar::getRawCookies(const WebCore::Document& document, const URL& url, Vector<WebCore::Cookie>& rawCookies) const
{
Optional<FrameIdentifier> frameID;
Optional<PageIdentifier> pageID;
if (auto* frame = document.frame()) {
frameID = frame->loader().client().frameID();
pageID = frame->loader().client().pageID();
}
if (!WebProcess::singleton().ensureNetworkProcessConnection().connection().sendSync(Messages::NetworkConnectionToWebProcess::GetRawCookies(document.firstPartyForCookies(), sameSiteInfo(document), url, frameID, pageID), Messages::NetworkConnectionToWebProcess::GetRawCookies::Reply(rawCookies), 0))
return false;
return true;
}
void WebCookieJar::deleteCookie(const WebCore::Document& document, const URL& url, const String& cookieName)
{
WebProcess::singleton().ensureNetworkProcessConnection().connection().send(Messages::NetworkConnectionToWebProcess::DeleteCookie(url, cookieName), 0);
}
}