#include "config.h"
#include "CookieJar.h"
#include "CookieRequestHeaderFieldProxy.h"
#include "CookiesStrategy.h"
#include "Document.h"
#include "DocumentLoader.h"
#include "Frame.h"
#include "FrameLoader.h"
#include "FrameLoaderClient.h"
#include "NetworkStorageSession.h"
#include "NetworkingContext.h"
#include "PlatformStrategies.h"
#include "SameSiteInfo.h"
#include <wtf/SystemTracing.h>
namespace WebCore {
static NetworkingContext* networkingContext(const Document& document)
{
Frame* frame = document.frame();
if (!frame)
return nullptr;
return frame->loader().networkingContext();
}
inline NetworkStorageSession& storageSession(const Document& document)
{
NetworkingContext* context = networkingContext(document);
return context ? context->storageSession() : NetworkStorageSession::defaultStorageSession();
}
static IncludeSecureCookies shouldIncludeSecureCookies(const Document& document, const URL& url)
{
return (url.protocolIs("https") && !document.foundMixedContent().contains(SecurityContext::MixedContentType::Active)) ? IncludeSecureCookies::Yes : IncludeSecureCookies::No;
}
static inline SameSiteInfo sameSiteInfo(const Document& document)
{
if (auto* loader = document.loader())
return SameSiteInfo::create(loader->request());
return { };
}
String cookies(Document& document, const URL& url)
{
TraceScope scope(FetchCookiesStart, FetchCookiesEnd);
auto includeSecureCookies = shouldIncludeSecureCookies(document, url);
std::pair<String, bool> result;
auto frame = document.frame();
if (frame)
result = platformStrategies()->cookiesStrategy()->cookiesForDOM(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, frame->loader().client().frameID(), frame->loader().client().pageID(), includeSecureCookies);
else
result = platformStrategies()->cookiesStrategy()->cookiesForDOM(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, WTF::nullopt, WTF::nullopt, includeSecureCookies);
if (result.second)
document.setSecureCookiesAccessed();
return result.first;
}
CookieRequestHeaderFieldProxy cookieRequestHeaderFieldProxy(const Document& document, const URL& url)
{
TraceScope scope(FetchCookiesStart, FetchCookiesEnd);
CookieRequestHeaderFieldProxy proxy;
proxy.sessionID = storageSession(document).sessionID();
proxy.firstParty = document.firstPartyForCookies();
proxy.sameSiteInfo = sameSiteInfo(document);
proxy.url = url;
proxy.includeSecureCookies = shouldIncludeSecureCookies(document, url);
if (auto* frame = document.frame()) {
proxy.frameID = frame->loader().client().frameID();
proxy.pageID = frame->loader().client().pageID();
}
return proxy;
}
void setCookies(Document& document, const URL& url, const String& cookieString)
{
auto frame = document.frame();
if (frame)
platformStrategies()->cookiesStrategy()->setCookiesFromDOM(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, frame->loader().client().frameID(), frame->loader().client().pageID(), cookieString);
else
platformStrategies()->cookiesStrategy()->setCookiesFromDOM(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, WTF::nullopt, WTF::nullopt, cookieString);
}
bool cookiesEnabled(const Document& document)
{
return platformStrategies()->cookiesStrategy()->cookiesEnabled(storageSession(document));
}
String cookieRequestHeaderFieldValue(Document& document, const URL& url)
{
auto includeSecureCookies = shouldIncludeSecureCookies(document, url);
std::pair<String, bool> result;
auto frame = document.frame();
if (frame)
result = platformStrategies()->cookiesStrategy()->cookieRequestHeaderFieldValue(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, frame->loader().client().frameID(), frame->loader().client().pageID(), includeSecureCookies);
else
result = platformStrategies()->cookiesStrategy()->cookieRequestHeaderFieldValue(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, WTF::nullopt, WTF::nullopt, includeSecureCookies);
if (result.second)
document.setSecureCookiesAccessed();
return result.first;
}
bool getRawCookies(const Document& document, const URL& url, Vector<Cookie>& cookies)
{
auto frame = document.frame();
if (frame)
return platformStrategies()->cookiesStrategy()->getRawCookies(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, frame->loader().client().frameID(), frame->loader().client().pageID(), cookies);
return platformStrategies()->cookiesStrategy()->getRawCookies(storageSession(document), document.firstPartyForCookies(), sameSiteInfo(document), url, WTF::nullopt, WTF::nullopt, cookies);
}
void deleteCookie(const Document& document, const URL& url, const String& cookieName)
{
platformStrategies()->cookiesStrategy()->deleteCookie(storageSession(document), url, cookieName);
}
}