comparison OrthancFramework/Sources/HttpServer/MultipartStreamReader.h @ 4044:d25f4c0fa160 framework

splitting code into OrthancFramework and OrthancServer
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 10 Jun 2020 20:30:34 +0200
parents Core/HttpServer/MultipartStreamReader.h@f9863630ec7f
children bf7b9edf6b81
comparison
equal deleted inserted replaced
4043:6c6239aec462 4044:d25f4c0fa160
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
4 * Department, University Hospital of Liege, Belgium
5 * Copyright (C) 2017-2020 Osimis S.A., Belgium
6 *
7 * This program is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License as
9 * published by the Free Software Foundation, either version 3 of the
10 * License, or (at your option) any later version.
11 *
12 * In addition, as a special exception, the copyright holders of this
13 * program give permission to link the code of its release with the
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it
15 * that use the same license as the "OpenSSL" library), and distribute
16 * the linked executables. You must obey the GNU General Public License
17 * in all respects for all of the code used other than "OpenSSL". If you
18 * modify file(s) with this exception, you may extend this exception to
19 * your version of the file(s), but you are not obligated to do so. If
20 * you do not wish to do so, delete this exception statement from your
21 * version. If you delete this exception statement from all source files
22 * in the program, then also delete it here.
23 *
24 * This program is distributed in the hope that it will be useful, but
25 * WITHOUT ANY WARRANTY; without even the implied warranty of
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
27 * General Public License for more details.
28 *
29 * You should have received a copy of the GNU General Public License
30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
31 **/
32
33
34 #pragma once
35
36 #include "StringMatcher.h"
37 #include "../ChunkedBuffer.h"
38
39 #include <map>
40
41 namespace Orthanc
42 {
43 class ORTHANC_PUBLIC MultipartStreamReader : public boost::noncopyable
44 {
45 public:
46 typedef std::map<std::string, std::string> HttpHeaders;
47
48 class IHandler : public boost::noncopyable
49 {
50 public:
51 virtual ~IHandler()
52 {
53 }
54
55 virtual void HandlePart(const HttpHeaders& headers,
56 const void* part,
57 size_t size) = 0;
58 };
59
60 private:
61 enum State
62 {
63 State_UnusedArea,
64 State_Content,
65 State_Done
66 };
67
68 State state_;
69 IHandler* handler_;
70 StringMatcher headersMatcher_;
71 StringMatcher boundaryMatcher_;
72 ChunkedBuffer buffer_;
73 size_t blockSize_;
74
75 void ParseStream();
76
77 public:
78 MultipartStreamReader(const std::string& boundary);
79
80 void SetBlockSize(size_t size);
81
82 size_t GetBlockSize() const
83 {
84 return blockSize_;
85 }
86
87 void SetHandler(IHandler& handler)
88 {
89 handler_ = &handler;
90 }
91
92 void AddChunk(const void* chunk,
93 size_t size);
94
95 void AddChunk(const std::string& chunk);
96
97 void CloseStream();
98
99 static bool GetMainContentType(std::string& contentType,
100 const HttpHeaders& headers);
101
102 static bool ParseMultipartContentType(std::string& contentType,
103 std::string& subType, // Possibly empty
104 std::string& boundary,
105 const std::string& contentTypeHeader);
106 };
107 }