1/*
2 * Copyright (C) 2016-2018 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 "NetworkDataTask.h"
29#include "NetworkResourceLoadParameters.h"
30#include <WebCore/ResourceError.h>
31#include <WebCore/ResourceResponse.h>
32#include <wtf/CompletionHandler.h>
33#include <wtf/UniqueRef.h>
34#include <wtf/WeakPtr.h>
35
36namespace WebKit {
37
38class NetworkConnectionToWebProcess;
39class NetworkLoadChecker;
40class NetworkProcess;
41
42class PingLoad final : public CanMakeWeakPtr<PingLoad>, private NetworkDataTaskClient {
43public:
44 PingLoad(NetworkConnectionToWebProcess&, NetworkProcess&, NetworkResourceLoadParameters&&, CompletionHandler<void(const WebCore::ResourceError&, const WebCore::ResourceResponse&)>&&);
45 PingLoad(NetworkProcess&, NetworkResourceLoadParameters&&, CompletionHandler<void(const WebCore::ResourceError&, const WebCore::ResourceResponse&)>&&);
46
47private:
48 ~PingLoad();
49 void initialize(NetworkProcess&);
50
51 const URL& currentURL() const;
52
53 void willPerformHTTPRedirection(WebCore::ResourceResponse&&, WebCore::ResourceRequest&&, RedirectCompletionHandler&&) final;
54 void didReceiveChallenge(WebCore::AuthenticationChallenge&&, ChallengeCompletionHandler&&) final;
55 void didReceiveResponse(WebCore::ResourceResponse&&, ResponseCompletionHandler&&) final;
56 void didReceiveData(Ref<WebCore::SharedBuffer>&&) final;
57 void didCompleteWithError(const WebCore::ResourceError&, const WebCore::NetworkLoadMetrics&) final;
58 void didSendData(uint64_t totalBytesSent, uint64_t totalBytesExpectedToSend) final;
59 void wasBlocked() final;
60 void cannotShowURL() final;
61 void wasBlockedByRestrictions() final;
62 void timeoutTimerFired();
63
64 void loadRequest(NetworkProcess&, WebCore::ResourceRequest&&);
65
66 void didFinish(const WebCore::ResourceError& = { }, const WebCore::ResourceResponse& response = { });
67
68 NetworkResourceLoadParameters m_parameters;
69 CompletionHandler<void(const WebCore::ResourceError&, const WebCore::ResourceResponse&)> m_completionHandler;
70 RefPtr<NetworkDataTask> m_task;
71 WebCore::Timer m_timeoutTimer;
72 UniqueRef<NetworkLoadChecker> m_networkLoadChecker;
73 Vector<RefPtr<WebCore::BlobDataFileReference>> m_blobFiles;
74};
75
76}
77