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