d1ca1ad5f00a676ffae209602767db26b59b00d1
[WebKit-https.git] / Source / WebCore / platform / network / curl / CurlFormDataStream.h
1 /*
2  * Copyright (C) 2008 Apple Inc.  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  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. 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 APPLE INC. ``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 APPLE INC. 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 #include "FileSystem.h"
30 #include "FormData.h"
31 #include <wtf/Vector.h>
32
33 namespace WebCore {
34
35 class CurlFormDataStream {
36 public:
37     CurlFormDataStream(const FormData*);
38     WEBCORE_EXPORT ~CurlFormDataStream();
39
40     void clean();
41
42     size_t elementSize() { return m_formData ? m_formData->elements().size() : 0; }
43
44     std::optional<const Vector<char>&> getPostData();
45     bool shouldUseChunkTransfer();
46     unsigned long long totalSize();
47
48     std::optional<size_t> read(char*, size_t);
49     unsigned long long totalReadSize() { return m_totalReadSize; }
50
51 private:
52     void computeContentLength();
53
54     std::optional<size_t> readFromFile(const FormDataElement&, char*, size_t);
55     std::optional<size_t> readFromData(const FormDataElement&, char*, size_t);
56
57     RefPtr<FormData> m_formData;
58
59     std::unique_ptr<Vector<char>> m_postData;
60     bool m_isContentLengthUpdated { false };
61     bool m_shouldUseChunkTransfer { false };
62     unsigned long long m_totalSize { 0 };
63     unsigned long long m_totalReadSize { 0 };
64
65     size_t m_elementPosition { 0 };
66
67     FileSystem::PlatformFileHandle m_fileHandle { FileSystem::invalidPlatformFileHandle };
68     size_t m_dataOffset { 0 };
69 };
70
71 } // namespace WebCore