#include "config.h"
#include "PingLoader.h"
#include "CachedResourceLoader.h"
#include "CachedResourceRequest.h"
#include "ContentRuleListResults.h"
#include "ContentSecurityPolicy.h"
#include "Document.h"
#include "Frame.h"
#include "FrameLoader.h"
#include "FrameLoaderClient.h"
#include "HTTPHeaderValues.h"
#include "InspectorInstrumentation.h"
#include "LoaderStrategy.h"
#include "NetworkLoadMetrics.h"
#include "Page.h"
#include "PlatformStrategies.h"
#include "ProgressTracker.h"
#include "ResourceError.h"
#include "ResourceHandle.h"
#include "ResourceLoadInfo.h"
#include "ResourceRequest.h"
#include "ResourceResponse.h"
#include "SecurityOrigin.h"
#include "SecurityPolicy.h"
#include "UserContentController.h"
#include <wtf/text/CString.h>
namespace WebCore {
#if ENABLE(CONTENT_EXTENSIONS)
static bool processContentRuleListsForLoad(const Frame& frame, ResourceRequest& request, OptionSet<ContentExtensions::ResourceType> resourceType)
{
auto* documentLoader = frame.loader().documentLoader();
if (!documentLoader)
return false;
auto* page = frame.page();
if (!page)
return false;
auto results = page->userContentProvider().processContentRuleListsForLoad(request.url(), resourceType, *documentLoader);
bool result = results.summary.blockedLoad;
ContentExtensions::applyResultsToRequest(WTFMove(results), page, request);
return result;
}
#endif
void PingLoader::loadImage(Frame& frame, const URL& url)
{
ASSERT(frame.document());
auto& document = *frame.document();
if (!document.securityOrigin().canDisplay(url)) {
FrameLoader::reportLocalLoadFailed(&frame, url);
return;
}
ResourceRequest request(url);
#if ENABLE(CONTENT_EXTENSIONS)
if (processContentRuleListsForLoad(frame, request, ContentExtensions::ResourceType::Image))
return;
#endif
document.contentSecurityPolicy()->upgradeInsecureRequestIfNeeded(request, ContentSecurityPolicy::InsecureRequestType::Load);
request.setHTTPHeaderField(HTTPHeaderName::CacheControl, HTTPHeaderValues::maxAge0());
HTTPHeaderMap originalRequestHeader = request.httpHeaderFields();
String referrer = SecurityPolicy::generateReferrerHeader(document.referrerPolicy(), request.url(), frame.loader().outgoingReferrer());
if (!referrer.isEmpty())
request.setHTTPReferrer(referrer);
frame.loader().addExtraFieldsToSubresourceRequest(request);
startPingLoad(frame, request, WTFMove(originalRequestHeader), ShouldFollowRedirects::Yes, ContentSecurityPolicyImposition::DoPolicyCheck, ReferrerPolicy::EmptyString);
}
void PingLoader::sendPing(Frame& frame, const URL& pingURL, const URL& destinationURL)
{
ASSERT(frame.document());
if (!pingURL.protocolIsInHTTPFamily())
return;
ResourceRequest request(pingURL);
#if ENABLE(CONTENT_EXTENSIONS)
if (processContentRuleListsForLoad(frame, request, { ContentExtensions::ResourceType::Raw, ContentExtensions::ResourceType::Ping }))
return;
#endif
auto& document = *frame.document();
document.contentSecurityPolicy()->upgradeInsecureRequestIfNeeded(request, ContentSecurityPolicy::InsecureRequestType::Load);
request.setHTTPMethod("POST");
request.setHTTPContentType("text/ping");
request.setHTTPBody(FormData::create("PING"));
request.setHTTPHeaderField(HTTPHeaderName::CacheControl, HTTPHeaderValues::maxAge0());
HTTPHeaderMap originalRequestHeader = request.httpHeaderFields();
frame.loader().addExtraFieldsToSubresourceRequest(request);
auto& sourceOrigin = document.securityOrigin();
FrameLoader::addHTTPOriginIfNeeded(request, sourceOrigin.toString());
request.setHTTPHeaderField(HTTPHeaderName::PingTo, destinationURL);
if (!SecurityPolicy::shouldHideReferrer(pingURL, frame.loader().outgoingReferrer())) {
request.setHTTPHeaderField(HTTPHeaderName::PingFrom, document.url());
if (!sourceOrigin.isSameSchemeHostPort(SecurityOrigin::create(pingURL).get())) {
String referrer = SecurityPolicy::generateReferrerHeader(document.referrerPolicy(), pingURL, frame.loader().outgoingReferrer());
if (!referrer.isEmpty())
request.setHTTPReferrer(referrer);
}
}
startPingLoad(frame, request, WTFMove(originalRequestHeader), ShouldFollowRedirects::Yes, ContentSecurityPolicyImposition::DoPolicyCheck, request.httpReferrer().isEmpty() ? ReferrerPolicy::NoReferrer : ReferrerPolicy::UnsafeUrl);
}
void PingLoader::sendViolationReport(Frame& frame, const URL& reportURL, Ref<FormData>&& report, ViolationReportType reportType)
{
ASSERT(frame.document());
ResourceRequest request(reportURL);
#if ENABLE(CONTENT_EXTENSIONS)
if (processContentRuleListsForLoad(frame, request, ContentExtensions::ResourceType::Raw))
return;
#endif
auto& document = *frame.document();
document.contentSecurityPolicy()->upgradeInsecureRequestIfNeeded(request, ContentSecurityPolicy::InsecureRequestType::Load);
request.setHTTPMethod("POST"_s);
request.setHTTPBody(WTFMove(report));
switch (reportType) {
case ViolationReportType::ContentSecurityPolicy:
request.setHTTPContentType("application/csp-report"_s);
break;
case ViolationReportType::XSSAuditor:
request.setHTTPContentType("application/json"_s);
break;
}
bool removeCookies = true;
if (document.securityOrigin().isSameSchemeHostPort(SecurityOrigin::create(reportURL).get()))
removeCookies = false;
if (removeCookies)
request.setAllowCookies(false);
HTTPHeaderMap originalRequestHeader = request.httpHeaderFields();
frame.loader().addExtraFieldsToSubresourceRequest(request);
String referrer = SecurityPolicy::generateReferrerHeader(document.referrerPolicy(), reportURL, frame.loader().outgoingReferrer());
if (!referrer.isEmpty())
request.setHTTPReferrer(referrer);
startPingLoad(frame, request, WTFMove(originalRequestHeader), ShouldFollowRedirects::No, ContentSecurityPolicyImposition::SkipPolicyCheck, ReferrerPolicy::EmptyString);
}
void PingLoader::startPingLoad(Frame& frame, ResourceRequest& request, HTTPHeaderMap&& originalRequestHeaders, ShouldFollowRedirects shouldFollowRedirects, ContentSecurityPolicyImposition policyCheck, ReferrerPolicy referrerPolicy)
{
unsigned long identifier = frame.page()->progress().createUniqueIdentifier();
bool shouldUseCredentialStorage = frame.loader().client().shouldUseCredentialStorage(frame.loader().activeDocumentLoader(), identifier);
ResourceLoaderOptions options;
options.credentials = shouldUseCredentialStorage ? FetchOptions::Credentials::Include : FetchOptions::Credentials::Omit;
options.redirect = shouldFollowRedirects == ShouldFollowRedirects::Yes ? FetchOptions::Redirect::Follow : FetchOptions::Redirect::Error;
options.keepAlive = true;
options.contentSecurityPolicyImposition = policyCheck;
options.referrerPolicy = referrerPolicy;
options.sendLoadCallbacks = SendCallbackPolicy::SendCallbacks;
options.cache = FetchOptions::Cache::NoCache;
if (platformStrategies()->loaderStrategy()->usePingLoad()) {
InspectorInstrumentation::willSendRequestOfType(&frame, identifier, frame.loader().activeDocumentLoader(), request, InspectorInstrumentation::LoadType::Ping);
platformStrategies()->loaderStrategy()->startPingLoad(frame, request, WTFMove(originalRequestHeaders), options, policyCheck, [protectedFrame = makeRef(frame), identifier] (const ResourceError& error, const ResourceResponse& response) {
if (!response.isNull())
InspectorInstrumentation::didReceiveResourceResponse(protectedFrame, identifier, protectedFrame->loader().activeDocumentLoader(), response, nullptr);
if (!error.isNull()) {
InspectorInstrumentation::didFailLoading(protectedFrame.ptr(), protectedFrame->loader().activeDocumentLoader(), identifier, error);
return;
}
InspectorInstrumentation::didFinishLoading(protectedFrame.ptr(), protectedFrame->loader().activeDocumentLoader(), identifier, { }, nullptr);
});
return;
}
CachedResourceRequest cachedResourceRequest { ResourceRequest { request }, options };
frame.document()->cachedResourceLoader().requestPingResource(WTFMove(cachedResourceRequest));
}
}