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 | #include <WebCore/ResourceRequest.h> |
29 | #include <wtf/Deque.h> |
30 | #include <wtf/RefCounted.h> |
31 | |
32 | namespace WebCore { |
33 | class ResourceError; |
34 | class ResourceLoader; |
35 | class ResourceResponse; |
36 | } |
37 | |
38 | namespace WebKit { |
39 | |
40 | class WebURLSchemeHandlerProxy; |
41 | |
42 | class WebURLSchemeTaskProxy : public RefCounted<WebURLSchemeTaskProxy> { |
43 | public: |
44 | static Ref<WebURLSchemeTaskProxy> create(WebURLSchemeHandlerProxy& handler, WebCore::ResourceLoader& loader) |
45 | { |
46 | return adoptRef(*new WebURLSchemeTaskProxy(handler, loader)); |
47 | } |
48 | |
49 | const WebCore::ResourceRequest& request() const { return m_request; } |
50 | |
51 | void startLoading(); |
52 | void stopLoading(); |
53 | |
54 | void didPerformRedirection(WebCore::ResourceResponse&&, WebCore::ResourceRequest&&); |
55 | void didReceiveResponse(const WebCore::ResourceResponse&); |
56 | void didReceiveData(size_t, const uint8_t* data); |
57 | void didComplete(const WebCore::ResourceError&); |
58 | |
59 | unsigned long identifier() const { return m_identifier; } |
60 | |
61 | private: |
62 | WebURLSchemeTaskProxy(WebURLSchemeHandlerProxy&, WebCore::ResourceLoader&); |
63 | bool hasLoader(); |
64 | |
65 | void queueTask(Function<void()>&& task) { m_queuedTasks.append(WTFMove(task)); } |
66 | void processNextPendingTask(); |
67 | |
68 | WebURLSchemeHandlerProxy& m_urlSchemeHandler; |
69 | RefPtr<WebCore::ResourceLoader> m_coreLoader; |
70 | WebCore::ResourceRequest m_request; |
71 | unsigned long m_identifier; |
72 | bool m_waitingForCompletionHandler { false }; |
73 | Deque<Function<void()>> m_queuedTasks; |
74 | }; |
75 | |
76 | } // namespace WebKit |
77 | |