[Curl] Extract multipart handling from ResourceHandle to CurlRequest.
[WebKit-https.git] / Source / WebCore / platform / network / curl / CurlMultipartHandle.h
1 /*
2  * Copyright (C) 2013 University of Szeged
3  * Copyright (C) 2018 Sony Interactive Entertainment Inc.
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY UNIVERSITY OF SZEGED ``AS IS'' AND ANY
16  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL UNIVERSITY OF SZEGED OR
19  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
22  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
23  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27
28 #pragma once
29
30 #include <wtf/Vector.h>
31 #include <wtf/text/WTFString.h>
32
33 namespace WebCore {
34
35 class CurlMultipartHandleClient;
36 class CurlResponse;
37 class SharedBuffer;
38
39 class CurlMultipartHandle {
40 public:
41     static std::unique_ptr<CurlMultipartHandle> createIfNeeded(CurlMultipartHandleClient&, const CurlResponse&);
42
43     CurlMultipartHandle(CurlMultipartHandleClient&, const String&);
44     ~CurlMultipartHandle() { }
45
46     void didReceiveData(const SharedBuffer&);
47     void didComplete();
48
49 private:
50     enum class State {
51         CheckBoundary,
52         InBoundary,
53         InHeader,
54         InContent,
55         EndBoundary,
56         End
57     };
58
59     static std::optional<String> extractBoundary(const CurlResponse&);
60     static std::optional<String> extractBoundaryFromContentType(const String&);
61
62     bool processContent();
63     bool checkForBoundary(size_t& boundaryStartPosition, size_t& lastPartialMatchPosition);
64     size_t matchedLength(const char* data);
65     bool parseHeadersIfPossible();
66
67     CurlMultipartHandleClient& m_client;
68
69     String m_boundary;
70     Vector<char> m_buffer;
71     Vector<String> m_headers;
72
73     State m_state { State::CheckBoundary };
74 };
75
76 } // namespace WebCore