1/*
2 * Copyright (C) 2017 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 ENABLE(SERVICE_WORKER)
29
30#include "DataReference.h"
31#include "FormDataReference.h"
32#include "MessageReceiver.h"
33#include "MessageSender.h"
34#include <WebCore/FetchIdentifier.h>
35#include <WebCore/ResourceError.h>
36#include <WebCore/ResourceLoader.h>
37#include <wtf/CompletionHandler.h>
38
39namespace WebKit {
40
41class WebSWClientConnection;
42class WebServiceWorkerProvider;
43
44class ServiceWorkerClientFetch final : public RefCounted<ServiceWorkerClientFetch>, public IPC::MessageReceiver {
45public:
46 enum class Result { Succeeded, Cancelled, Unhandled };
47 using Callback = WTF::CompletionHandler<void(Result)>;
48
49 static Ref<ServiceWorkerClientFetch> create(WebServiceWorkerProvider&, Ref<WebCore::ResourceLoader>&&, WebCore::FetchIdentifier, Ref<WebSWClientConnection>&&, bool shouldClearReferrerOnHTTPSToHTTPRedirect, Callback&&);
50 ~ServiceWorkerClientFetch();
51
52 void start();
53
54 void didReceiveMessage(IPC::Connection&, IPC::Decoder&) final;
55 void cancel();
56
57 bool isOngoing() const { return !!m_callback; }
58
59private:
60 ServiceWorkerClientFetch(WebServiceWorkerProvider&, Ref<WebCore::ResourceLoader>&&, WebCore::FetchIdentifier, Ref<WebSWClientConnection>&&, bool shouldClearReferrerOnHTTPSToHTTPRedirect, Callback&&);
61
62 Optional<WebCore::ResourceError> validateResponse(const WebCore::ResourceResponse&);
63
64 void didReceiveResponse(WebCore::ResourceResponse&&, bool needsContinueDidReceiveResponseMessage);
65 void didReceiveRedirectResponse(WebCore::ResourceResponse&&);
66 void didReceiveData(const IPC::DataReference&, int64_t encodedDataLength);
67 void didReceiveFormData(const IPC::FormDataReference&);
68 void didFinish();
69 void didFail(WebCore::ResourceError&&);
70 void didNotHandle();
71
72 WebServiceWorkerProvider& m_serviceWorkerProvider;
73 RefPtr<WebCore::ResourceLoader> m_loader;
74 WebCore::FetchIdentifier m_identifier;
75 Ref<WebSWClientConnection> m_connection;
76 Callback m_callback;
77 bool m_shouldClearReferrerOnHTTPSToHTTPRedirect { true };
78 int64_t m_encodedDataLength { 0 };
79 bool m_didFail { false };
80 WebCore::ServiceWorkerRegistrationIdentifier m_serviceWorkerRegistrationIdentifier;
81};
82
83} // namespace WebKit
84
85#endif // ENABLE(SERVICE_WORKER)
86