ad35464e5133f188089402be1ee978313a87c7ae
[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
33 namespace WebCore {
34 class ContentSecurityPolicy;
35 class URL;
36 }
37
38 namespace WebKit {
39
40 class NetworkCORSPreflightChecker;
41
42 class PingLoad final : private NetworkDataTaskClient {
43 public:
44     explicit PingLoad(NetworkResourceLoadParameters&&);
45     
46 private:
47     ~PingLoad();
48
49     WebCore::ContentSecurityPolicy* contentSecurityPolicy() const;
50
51     void willPerformHTTPRedirection(WebCore::ResourceResponse&&, WebCore::ResourceRequest&&, RedirectCompletionHandler&&) final;
52     void didReceiveChallenge(const WebCore::AuthenticationChallenge&, ChallengeCompletionHandler&&) final;
53     void didReceiveResponseNetworkSession(WebCore::ResourceResponse&&, ResponseCompletionHandler&&) final;
54     void didReceiveData(Ref<WebCore::SharedBuffer>&&) final;
55     void didCompleteWithError(const WebCore::ResourceError&, const WebCore::NetworkLoadMetrics&) final;
56     void didSendData(uint64_t totalBytesSent, uint64_t totalBytesExpectedToSend) final;
57     void wasBlocked() final;
58     void cannotShowURL() final;
59     void timeoutTimerFired();
60
61     void loadRequest(WebCore::ResourceRequest&&);
62     bool isAllowedRedirect(const WebCore::URL&) const;
63     void makeCrossOriginAccessRequest(WebCore::ResourceRequest&&);
64     void makeSimpleCrossOriginAccessRequest(WebCore::ResourceRequest&&);
65     void makeCrossOriginAccessRequestWithPreflight(WebCore::ResourceRequest&&);
66     void preflightSuccess(WebCore::ResourceRequest&&);
67
68     WebCore::SecurityOrigin& securityOrigin() const;
69     const WebCore::HTTPHeaderMap& originalRequestHeaders() const; // Needed for CORS checks.
70     
71     NetworkResourceLoadParameters m_parameters;
72     RefPtr<NetworkDataTask> m_task;
73     WebCore::Timer m_timeoutTimer;
74     std::unique_ptr<NetworkCORSPreflightChecker> m_corsPreflightChecker;
75     RefPtr<WebCore::SecurityOrigin> m_origin;
76     bool m_isSameOriginRequest;
77     bool m_isSimpleRequest { true };
78     RedirectCompletionHandler m_redirectHandler;
79     mutable std::unique_ptr<WebCore::ContentSecurityPolicy> m_contentSecurityPolicy;
80 };
81
82 }
83
84 #endif // USE(NETWORK_SESSION)