c28c0149dac728ecb97466d28b0f00b14f443dfd
[WebKit-https.git] / Source / WebKit / NetworkProcess / PingLoad.h
1 /*
2  * Copyright (C) 2016 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #pragma once
27
28 #if USE(NETWORK_SESSION)
29
30 #include "NetworkDataTask.h"
31 #include "NetworkResourceLoadParameters.h"
32 #include <WebCore/ContentExtensionsBackend.h>
33 #include <WebCore/ResourceError.h>
34 #include <wtf/CompletionHandler.h>
35
36 namespace WebCore {
37 class ContentSecurityPolicy;
38 class HTTPHeaderMap;
39 class URL;
40 }
41
42 namespace WebKit {
43
44 class NetworkCORSPreflightChecker;
45
46 class PingLoad final : private NetworkDataTaskClient {
47 public:
48     PingLoad(NetworkResourceLoadParameters&&, WebCore::HTTPHeaderMap&& originalRequestHeaders, WTF::CompletionHandler<void(const WebCore::ResourceError&)>&&);
49     
50 private:
51     ~PingLoad();
52
53     WebCore::ContentSecurityPolicy* contentSecurityPolicy() const;
54
55     void willPerformHTTPRedirection(WebCore::ResourceResponse&&, WebCore::ResourceRequest&&, RedirectCompletionHandler&&) final;
56     void didReceiveChallenge(const WebCore::AuthenticationChallenge&, ChallengeCompletionHandler&&) final;
57     void didReceiveResponseNetworkSession(WebCore::ResourceResponse&&, ResponseCompletionHandler&&) final;
58     void didReceiveData(Ref<WebCore::SharedBuffer>&&) final;
59     void didCompleteWithError(const WebCore::ResourceError&, const WebCore::NetworkLoadMetrics&) final;
60     void didSendData(uint64_t totalBytesSent, uint64_t totalBytesExpectedToSend) final;
61     void wasBlocked() final;
62     void cannotShowURL() final;
63     void timeoutTimerFired();
64
65     void loadRequest(WebCore::ResourceRequest&&);
66     bool isAllowedRedirect(const WebCore::URL&) const;
67     void makeCrossOriginAccessRequest(WebCore::ResourceRequest&&);
68     void makeSimpleCrossOriginAccessRequest(WebCore::ResourceRequest&&);
69     void makeCrossOriginAccessRequestWithPreflight(WebCore::ResourceRequest&&);
70     void preflightSuccess(WebCore::ResourceRequest&&);
71
72 #if ENABLE(CONTENT_EXTENSIONS)
73     WebCore::ContentExtensions::ContentExtensionsBackend& contentExtensionsBackend();
74     WebCore::ContentExtensions::BlockedStatus processContentExtensionRulesForLoad(WebCore::ResourceRequest&);
75 #endif
76
77     WebCore::SecurityOrigin& securityOrigin() const;
78
79     const WebCore::ResourceRequest& currentRequest() const;
80     void didFinish(const WebCore::ResourceError& = { });
81     
82     NetworkResourceLoadParameters m_parameters;
83     WebCore::HTTPHeaderMap m_originalRequestHeaders; // Needed for CORS checks.
84     WTF::CompletionHandler<void(const WebCore::ResourceError&)> m_completionHandler;
85     RefPtr<NetworkDataTask> m_task;
86     WebCore::Timer m_timeoutTimer;
87     std::unique_ptr<NetworkCORSPreflightChecker> m_corsPreflightChecker;
88     RefPtr<WebCore::SecurityOrigin> m_origin;
89     bool m_isSameOriginRequest;
90     bool m_isSimpleRequest { true };
91     RedirectCompletionHandler m_redirectHandler;
92     mutable std::unique_ptr<WebCore::ContentSecurityPolicy> m_contentSecurityPolicy;
93 #if ENABLE(CONTENT_EXTENSIONS)
94     std::unique_ptr<WebCore::ContentExtensions::ContentExtensionsBackend> m_contentExtensionsBackend;
95 #endif
96     std::optional<WebCore::ResourceRequest> m_lastRedirectionRequest;
97 };
98
99 }
100
101 #endif // USE(NETWORK_SESSION)