#include "config.h"
#include "LinkLoader.h"
#include "CSSStyleSheet.h"
#include "CachedCSSStyleSheet.h"
#include "CachedResourceLoader.h"
#include "CachedResourceRequest.h"
#include "ContainerNode.h"
#include "CrossOriginAccessControl.h"
#include "Document.h"
#include "Frame.h"
#include "FrameLoaderClient.h"
#include "FrameView.h"
#include "LinkRelAttribute.h"
#include "RuntimeEnabledFeatures.h"
#include "Settings.h"
#include "StyleResolver.h"
namespace WebCore {
LinkLoader::LinkLoader(LinkLoaderClient& client)
: m_client(client)
, m_linkLoadTimer(*this, &LinkLoader::linkLoadTimerFired)
, m_linkLoadingErrorTimer(*this, &LinkLoader::linkLoadingErrorTimerFired)
{
}
LinkLoader::~LinkLoader()
{
if (m_cachedLinkResource)
m_cachedLinkResource->removeClient(this);
}
void LinkLoader::linkLoadTimerFired()
{
m_client.linkLoaded();
}
void LinkLoader::linkLoadingErrorTimerFired()
{
m_client.linkLoadingErrored();
}
void LinkLoader::notifyFinished(CachedResource* resource)
{
ASSERT_UNUSED(resource, m_cachedLinkResource.get() == resource);
if (m_cachedLinkResource->errorOccurred())
m_linkLoadingErrorTimer.startOneShot(0);
else
m_linkLoadTimer.startOneShot(0);
m_cachedLinkResource->removeClient(this);
m_cachedLinkResource = nullptr;
}
Optional<CachedResource::Type> LinkLoader::resourceTypeFromAsAttribute(const String& as)
{
if (as.isEmpty())
return CachedResource::LinkPreload;
if (equalLettersIgnoringASCIICase(as, "image"))
return CachedResource::ImageResource;
if (equalLettersIgnoringASCIICase(as, "script"))
return CachedResource::Script;
if (equalLettersIgnoringASCIICase(as, "style"))
return CachedResource::CSSStyleSheet;
if (equalLettersIgnoringASCIICase(as, "media"))
return CachedResource::MediaResource;
if (equalLettersIgnoringASCIICase(as, "font"))
return CachedResource::FontResource;
#if ENABLE(VIDEO_TRACK)
if (equalLettersIgnoringASCIICase(as, "track"))
return CachedResource::TextTrackResource;
#endif
return Nullopt;
}
static void preloadIfNeeded(const LinkRelAttribute& relAttribute, const URL& href, Document& document, const String& as, const String& crossOriginMode)
{
if (!document.loader() || !relAttribute.isLinkPreload)
return;
ASSERT(RuntimeEnabledFeatures::sharedFeatures().linkPreloadEnabled());
if (!href.isValid()) {
document.addConsoleMessage(MessageSource::Other, MessageLevel::Error, String("<link rel=preload> has an invalid `href` value"));
return;
}
auto type = LinkLoader::resourceTypeFromAsAttribute(as);
if (!type) {
document.addConsoleMessage(MessageSource::Other, MessageLevel::Error, String("<link rel=preload> must have a valid `as` value"));
return;
}
ResourceRequest resourceRequest(document.completeURL(href));
CachedResourceRequest linkRequest(resourceRequest, CachedResource::defaultPriorityForResourceType(type.value()));
linkRequest.setInitiator("link");
if (!crossOriginMode.isNull()) {
ASSERT(document.securityOrigin());
StoredCredentials allowCredentials = equalLettersIgnoringASCIICase(crossOriginMode, "use-credentials") ? AllowStoredCredentials : DoNotAllowStoredCredentials;
updateRequestForAccessControl(linkRequest.mutableResourceRequest(), *document.securityOrigin(), allowCredentials);
}
linkRequest.setForPreload(true);
document.cachedResourceLoader().preload(type.value(), linkRequest, emptyString());
}
bool LinkLoader::loadLink(const LinkRelAttribute& relAttribute, const URL& href, const String& as, const String& crossOrigin, Document& document)
{
if (relAttribute.isDNSPrefetch) {
Settings* settings = document.settings();
if (settings && settings->dnsPrefetchingEnabled() && href.isValid() && !href.isEmpty() && document.frame())
document.frame()->loader().client().prefetchDNS(href.host());
}
if (m_client.shouldLoadLink())
preloadIfNeeded(relAttribute, href, document, as, crossOrigin);
#if ENABLE(LINK_PREFETCH)
if ((relAttribute.isLinkPrefetch || relAttribute.isLinkSubresource) && href.isValid() && document.frame()) {
if (!m_client.shouldLoadLink())
return false;
Optional<ResourceLoadPriority> priority;
CachedResource::Type type = CachedResource::LinkPrefetch;
if (relAttribute.isLinkSubresource) {
priority = ResourceLoadPriority::Low;
type = CachedResource::LinkSubresource;
}
CachedResourceRequest linkRequest(ResourceRequest(document.completeURL(href)), priority);
if (m_cachedLinkResource) {
m_cachedLinkResource->removeClient(this);
m_cachedLinkResource = nullptr;
}
m_cachedLinkResource = document.cachedResourceLoader().requestLinkResource(type, linkRequest);
if (m_cachedLinkResource)
m_cachedLinkResource->addClient(this);
}
#endif
return true;
}
}