blob: 11f5b1a220c51a895397e2b1138bde6479386237 [file] [log] [blame]
/*
* Copyright (C) 2005-2017 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of Apple Inc. ("Apple") nor the names of
* its contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#pragma once
#include "FrameLoaderTypes.h"
#include "ResourceLoader.h"
#include <wtf/CompletionHandler.h>
#include <wtf/text/WTFString.h>
namespace WebCore {
class CachedResource;
class CachedResourceLoader;
class NetworkLoadMetrics;
class ResourceRequest;
class SecurityOrigin;
class SubresourceLoader final : public ResourceLoader {
public:
WEBCORE_EXPORT static void create(Frame&, CachedResource&, ResourceRequest&&, const ResourceLoaderOptions&, CompletionHandler<void(RefPtr<SubresourceLoader>&&)>&&);
virtual ~SubresourceLoader();
void cancelIfNotFinishing();
bool isSubresourceLoader() const override;
CachedResource* cachedResource();
WEBCORE_EXPORT const HTTPHeaderMap* originalHeaders() const;
SecurityOrigin* origin() { return m_origin.get(); }
#if PLATFORM(IOS_FAMILY)
void startLoading() override;
// FIXME: What is an "iOS" original request? Why is it necessary?
const ResourceRequest& iOSOriginalRequest() const override { return m_iOSOriginalRequest; }
#endif
unsigned redirectCount() const { return m_redirectCount; }
void markInAsyncResponsePolicyCheck() { m_inAsyncResponsePolicyCheck = true; }
void didReceiveResponsePolicy();
private:
SubresourceLoader(Frame&, CachedResource&, const ResourceLoaderOptions&);
void init(ResourceRequest&&, CompletionHandler<void(bool)>&&) override;
void willSendRequestInternal(ResourceRequest&&, const ResourceResponse& redirectResponse, CompletionHandler<void(ResourceRequest&&)>&&) override;
void didSendData(unsigned long long bytesSent, unsigned long long totalBytesToBeSent) override;
void didReceiveResponse(const ResourceResponse&, CompletionHandler<void()>&& policyCompletionHandler) override;
void didReceiveData(const char*, unsigned, long long encodedDataLength, DataPayloadType) override;
void didReceiveBuffer(Ref<SharedBuffer>&&, long long encodedDataLength, DataPayloadType) override;
void didFinishLoading(const NetworkLoadMetrics&) override;
void didFail(const ResourceError&) override;
void willCancel(const ResourceError&) override;
void didCancel(const ResourceError&) override;
void updateReferrerPolicy(const String&);
#if PLATFORM(COCOA)
void willCacheResponseAsync(ResourceHandle*, NSCachedURLResponse*, CompletionHandler<void(NSCachedURLResponse *)>&&) override;
#endif
void releaseResources() override;
bool checkForHTTPStatusCodeError();
Expected<void, String> checkResponseCrossOriginAccessControl(const ResourceResponse&);
Expected<void, String> checkRedirectionCrossOriginAccessControl(const ResourceRequest& previousRequest, const ResourceResponse&, ResourceRequest& newRequest);
void didReceiveDataOrBuffer(const char*, int, RefPtr<SharedBuffer>&&, long long encodedDataLength, DataPayloadType);
void notifyDone(LoadCompletionType);
void reportResourceTiming(const NetworkLoadMetrics&);
#if USE(QUICK_LOOK)
bool shouldCreatePreviewLoaderForResponse(const ResourceResponse&) const;
void didReceivePreviewResponse(const ResourceResponse&) override;
#endif
enum SubresourceLoaderState {
Uninitialized,
Initialized,
Finishing,
#if PLATFORM(IOS_FAMILY)
CancelledWhileInitializing
#endif
};
class RequestCountTracker {
#if !COMPILER(MSVC)
WTF_MAKE_FAST_ALLOCATED;
#endif
public:
RequestCountTracker(CachedResourceLoader&, const CachedResource&);
~RequestCountTracker();
private:
CachedResourceLoader& m_cachedResourceLoader;
const CachedResource& m_resource;
};
#if PLATFORM(IOS_FAMILY)
ResourceRequest m_iOSOriginalRequest;
#endif
CachedResource* m_resource;
SubresourceLoaderState m_state;
Optional<RequestCountTracker> m_requestCountTracker;
RefPtr<SecurityOrigin> m_origin;
CompletionHandler<void()> m_policyForResponseCompletionHandler;
unsigned m_redirectCount { 0 };
bool m_loadingMultipartContent { false };
bool m_inAsyncResponsePolicyCheck { false };
};
}
SPECIALIZE_TYPE_TRAITS_BEGIN(WebCore::SubresourceLoader)
static bool isType(const WebCore::ResourceLoader& loader) { return loader.isSubresourceLoader(); }
SPECIALIZE_TYPE_TRAITS_END()