d492cd617e291d20f9195b2eed3b5f97d9873e47
[WebKit-https.git] / Source / WebCore / platform / network / curl / ResourceHandleCurlDelegate.h
1 /*
2  * Copyright (C) 2017 NAVER Corp. All rights reserved.
3  * Copyright (C) 2017 Sony Interactive Entertainment Inc.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *  * Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  *  * Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26
27 #pragma once
28
29 #if USE(CURL)
30
31 #include "Credential.h"
32 #include "CurlRequestClient.h"
33 #include "ResourceRequest.h"
34
35 namespace WebCore {
36
37 class CurlRequest;
38 class MultipartHandle;
39 class ResourceHandle;
40 class ResourceResponse;
41
42 class ResourceHandleCurlDelegate final : public ThreadSafeRefCounted<ResourceHandleCurlDelegate>, public CurlRequestClient {
43 public:
44     ResourceHandleCurlDelegate(ResourceHandle*);
45     ~ResourceHandleCurlDelegate();
46
47     bool hasHandle() const;
48     void releaseHandle();
49
50     void start();
51     void cancel();
52
53     void setDefersLoading(bool);
54     void setAuthentication(const String&, const String&);
55
56     void dispatchSynchronousJob();
57
58     void continueDidReceiveResponse();
59     void platformContinueSynchronousDidReceiveResponse();
60
61     void continueWillSendRequest(ResourceRequest&&);
62
63 private:
64     // Called from main thread.
65     ResourceResponse& response();
66
67     std::pair<String, String> getCredential(ResourceRequest&, bool);
68
69     bool cancelledOrClientless();
70
71     Ref<CurlRequest> createCurlRequest(ResourceRequest&);
72     void curlDidReceiveResponse(const CurlResponse&) override;
73     void curlDidReceiveBuffer(Ref<SharedBuffer>&&) override;
74     void curlDidComplete() override;
75     void curlDidFailWithError(const ResourceError&) override;
76
77     void continueAfterDidReceiveResponse();
78
79     bool shouldRedirectAsGET(const ResourceRequest&, bool crossOrigin);
80     void willSendRequest();
81     void continueAfterWillSendRequest(ResourceRequest&&);
82
83     void handleDataURL();
84
85     // Used by main thread.
86     ResourceHandle* m_handle;
87     std::unique_ptr<MultipartHandle> m_multipartHandle;
88     unsigned m_authFailureCount { 0 };
89     unsigned m_redirectCount { 0 };
90
91     ResourceRequest m_firstRequest;
92     ResourceRequest m_currentRequest;
93     bool m_shouldUseCredentialStorage;
94     String m_user;
95     String m_pass;
96     Credential m_initialCredential;
97     bool m_defersLoading;
98     bool m_addedCacheValidationHeaders { false };
99     RefPtr<CurlRequest> m_curlRequest;
100 };
101
102 } // namespace WebCore
103
104 #endif