1/*
2 * Copyright (C) 2015 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#include "DownloadID.h"
29#include "NetworkDataTask.h"
30#include "NetworkLoadClient.h"
31#include "NetworkLoadParameters.h"
32#include <WebCore/AuthenticationChallenge.h>
33#include <wtf/CompletionHandler.h>
34#include <wtf/text/WTFString.h>
35
36namespace WebCore {
37class BlobRegistryImpl;
38}
39
40namespace WebKit {
41
42class NetworkProcess;
43
44class NetworkLoad final : private NetworkDataTaskClient {
45 WTF_MAKE_FAST_ALLOCATED;
46public:
47 NetworkLoad(NetworkLoadClient&, WebCore::BlobRegistryImpl*, NetworkLoadParameters&&, NetworkSession&);
48 ~NetworkLoad();
49
50 void cancel();
51
52 bool isAllowedToAskUserForCredentials() const;
53
54 const WebCore::ResourceRequest& currentRequest() const { return m_currentRequest; }
55 void updateRequestAfterRedirection(WebCore::ResourceRequest&) const;
56
57 const NetworkLoadParameters& parameters() const { return m_parameters; }
58
59 void continueWillSendRequest(WebCore::ResourceRequest&&);
60
61 void convertTaskToDownload(PendingDownload&, const WebCore::ResourceRequest&, const WebCore::ResourceResponse&, ResponseCompletionHandler&&);
62 void setPendingDownloadID(DownloadID);
63 void setSuggestedFilename(const String&);
64 void setPendingDownload(PendingDownload&);
65 DownloadID pendingDownloadID() { return m_task->pendingDownloadID(); }
66
67 bool shouldCaptureExtraNetworkLoadMetrics() const final;
68
69 String description() const;
70
71private:
72 void initialize(NetworkSession&, WebCore::BlobRegistryImpl*);
73
74 // NetworkDataTaskClient
75 void willPerformHTTPRedirection(WebCore::ResourceResponse&&, WebCore::ResourceRequest&&, RedirectCompletionHandler&&) final;
76 void didReceiveChallenge(WebCore::AuthenticationChallenge&&, ChallengeCompletionHandler&&) final;
77 void didReceiveResponse(WebCore::ResourceResponse&&, ResponseCompletionHandler&&) final;
78 void didReceiveData(Ref<WebCore::SharedBuffer>&&) final;
79 void didCompleteWithError(const WebCore::ResourceError&, const WebCore::NetworkLoadMetrics&) final;
80 void didSendData(uint64_t totalBytesSent, uint64_t totalBytesExpectedToSend) final;
81 void wasBlocked() final;
82 void cannotShowURL() final;
83 void wasBlockedByRestrictions() final;
84
85 void notifyDidReceiveResponse(WebCore::ResourceResponse&&, ResponseCompletionHandler&&);
86 void throttleDelayCompleted();
87
88 std::reference_wrapper<NetworkLoadClient> m_client;
89 Ref<NetworkProcess> m_networkProcess;
90 const NetworkLoadParameters m_parameters;
91 CompletionHandler<void(WebCore::ResourceRequest&&)> m_redirectCompletionHandler;
92 RefPtr<NetworkDataTask> m_task;
93
94 struct Throttle;
95 std::unique_ptr<Throttle> m_throttle;
96 Seconds m_loadThrottleLatency;
97
98 WebCore::ResourceRequest m_currentRequest; // Updated on redirects.
99};
100
101} // namespace WebKit
102