Mercurial > hg > orthanc-webviewer
annotate Orthanc/Core/Toolbox.cpp @ 141:9362080e5e3d
sync
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 15 Jul 2016 12:00:11 +0200 |
parents | d73006baca3f |
children | d850500b8ca6 |
rev | line source |
---|---|
78 | 1 /** |
2 * Orthanc - A Lightweight, RESTful DICOM Store | |
122 | 3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics |
78 | 4 * Department, University Hospital of Liege, Belgium |
5 * | |
6 * This program is free software: you can redistribute it and/or | |
7 * modify it under the terms of the GNU General Public License as | |
8 * published by the Free Software Foundation, either version 3 of the | |
9 * License, or (at your option) any later version. | |
10 * | |
11 * In addition, as a special exception, the copyright holders of this | |
12 * program give permission to link the code of its release with the | |
13 * OpenSSL project's "OpenSSL" library (or with modified versions of it | |
14 * that use the same license as the "OpenSSL" library), and distribute | |
15 * the linked executables. You must obey the GNU General Public License | |
16 * in all respects for all of the code used other than "OpenSSL". If you | |
17 * modify file(s) with this exception, you may extend this exception to | |
18 * your version of the file(s), but you are not obligated to do so. If | |
19 * you do not wish to do so, delete this exception statement from your | |
20 * version. If you delete this exception statement from all source files | |
21 * in the program, then also delete it here. | |
22 * | |
23 * This program is distributed in the hope that it will be useful, but | |
24 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
25 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
26 * General Public License for more details. | |
27 * | |
28 * You should have received a copy of the GNU General Public License | |
29 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
30 **/ | |
31 | |
32 | |
33 #include "PrecompiledHeaders.h" | |
34 #include "Toolbox.h" | |
35 | |
36 #include "OrthancException.h" | |
37 #include "Logging.h" | |
38 | |
39 #include <string> | |
40 #include <stdint.h> | |
41 #include <string.h> | |
42 #include <boost/filesystem.hpp> | |
43 #include <boost/filesystem/fstream.hpp> | |
44 #include <boost/uuid/sha1.hpp> | |
45 #include <boost/lexical_cast.hpp> | |
46 #include <algorithm> | |
47 #include <ctype.h> | |
48 | |
49 #if BOOST_HAS_DATE_TIME == 1 | |
50 #include <boost/date_time/posix_time/posix_time.hpp> | |
51 #endif | |
52 | |
53 #if BOOST_HAS_REGEX == 1 | |
54 #include <boost/regex.hpp> | |
55 #endif | |
56 | |
57 #if defined(_WIN32) | |
58 #include <windows.h> | |
59 #include <process.h> // For "_spawnvp()" and "_getpid()" | |
60 #else | |
61 #include <unistd.h> // For "execvp()" | |
62 #include <sys/wait.h> // For "waitpid()" | |
63 #endif | |
64 | |
65 #if defined(__APPLE__) && defined(__MACH__) | |
66 #include <mach-o/dyld.h> /* _NSGetExecutablePath */ | |
67 #include <limits.h> /* PATH_MAX */ | |
68 #endif | |
69 | |
128
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
70 #if defined(__linux__) || defined(__FreeBSD_kernel__) || defined(__FreeBSD__) |
78 | 71 #include <limits.h> /* PATH_MAX */ |
72 #include <signal.h> | |
73 #include <unistd.h> | |
74 #endif | |
75 | |
76 #if BOOST_HAS_LOCALE != 1 | |
77 #error Since version 0.7.6, Orthanc entirely relies on boost::locale | |
78 #endif | |
79 | |
80 #include <boost/locale.hpp> | |
81 | |
82 | |
83 #if !defined(ORTHANC_ENABLE_MD5) || ORTHANC_ENABLE_MD5 == 1 | |
84 #include "../Resources/ThirdParty/md5/md5.h" | |
85 #endif | |
86 | |
87 | |
88 #if !defined(ORTHANC_ENABLE_BASE64) || ORTHANC_ENABLE_BASE64 == 1 | |
89 #include "../Resources/ThirdParty/base64/base64.h" | |
90 #endif | |
91 | |
92 | |
93 #if defined(_MSC_VER) && (_MSC_VER < 1800) | |
94 // Patch for the missing "_strtoll" symbol when compiling with Visual Studio < 2013 | |
95 extern "C" | |
96 { | |
97 int64_t _strtoi64(const char *nptr, char **endptr, int base); | |
98 int64_t strtoll(const char *nptr, char **endptr, int base) | |
99 { | |
100 return _strtoi64(nptr, endptr, base); | |
101 } | |
102 } | |
103 #endif | |
104 | |
105 | |
106 #if ORTHANC_PUGIXML_ENABLED == 1 | |
107 #include "ChunkedBuffer.h" | |
108 #include <pugixml.hpp> | |
109 #endif | |
110 | |
111 | |
112 namespace Orthanc | |
113 { | |
141 | 114 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
134 | 115 static bool finish_; |
116 static ServerBarrierEvent barrierEvent_; | |
78 | 117 |
118 #if defined(_WIN32) | |
119 static BOOL WINAPI ConsoleControlHandler(DWORD dwCtrlType) | |
120 { | |
121 // http://msdn.microsoft.com/en-us/library/ms683242(v=vs.85).aspx | |
134 | 122 finish_ = true; |
78 | 123 return true; |
124 } | |
125 #else | |
134 | 126 static void SignalHandler(int signal) |
78 | 127 { |
134 | 128 if (signal == SIGHUP) |
129 { | |
130 barrierEvent_ = ServerBarrierEvent_Reload; | |
131 } | |
132 | |
133 finish_ = true; | |
78 | 134 } |
135 #endif | |
136 | |
125 | 137 |
78 | 138 void Toolbox::USleep(uint64_t microSeconds) |
139 { | |
140 #if defined(_WIN32) | |
141 ::Sleep(static_cast<DWORD>(microSeconds / static_cast<uint64_t>(1000))); | |
128
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
142 #elif defined(__linux__) || defined(__APPLE__) || defined(__FreeBSD_kernel__) || defined(__FreeBSD__) |
78 | 143 usleep(microSeconds); |
144 #else | |
145 #error Support your platform here | |
146 #endif | |
147 } | |
148 | |
149 | |
134 | 150 static ServerBarrierEvent ServerBarrierInternal(const bool* stopFlag) |
78 | 151 { |
152 #if defined(_WIN32) | |
153 SetConsoleCtrlHandler(ConsoleControlHandler, true); | |
154 #else | |
155 signal(SIGINT, SignalHandler); | |
156 signal(SIGQUIT, SignalHandler); | |
157 signal(SIGTERM, SignalHandler); | |
134 | 158 signal(SIGHUP, SignalHandler); |
78 | 159 #endif |
160 | |
161 // Active loop that awakens every 100ms | |
134 | 162 finish_ = false; |
163 barrierEvent_ = ServerBarrierEvent_Stop; | |
164 while (!(*stopFlag || finish_)) | |
78 | 165 { |
166 Toolbox::USleep(100 * 1000); | |
167 } | |
168 | |
169 #if defined(_WIN32) | |
170 SetConsoleCtrlHandler(ConsoleControlHandler, false); | |
171 #else | |
172 signal(SIGINT, NULL); | |
173 signal(SIGQUIT, NULL); | |
174 signal(SIGTERM, NULL); | |
134 | 175 signal(SIGHUP, NULL); |
78 | 176 #endif |
134 | 177 |
178 return barrierEvent_; | |
78 | 179 } |
180 | |
181 | |
134 | 182 ServerBarrierEvent Toolbox::ServerBarrier(const bool& stopFlag) |
78 | 183 { |
134 | 184 return ServerBarrierInternal(&stopFlag); |
78 | 185 } |
186 | |
134 | 187 ServerBarrierEvent Toolbox::ServerBarrier() |
78 | 188 { |
189 const bool stopFlag = false; | |
134 | 190 return ServerBarrierInternal(&stopFlag); |
78 | 191 } |
141 | 192 #endif /* ORTHANC_SANDBOXED */ |
78 | 193 |
194 | |
195 void Toolbox::ToUpperCase(std::string& s) | |
196 { | |
197 std::transform(s.begin(), s.end(), s.begin(), toupper); | |
198 } | |
199 | |
200 | |
201 void Toolbox::ToLowerCase(std::string& s) | |
202 { | |
203 std::transform(s.begin(), s.end(), s.begin(), tolower); | |
204 } | |
205 | |
206 | |
207 void Toolbox::ToUpperCase(std::string& result, | |
208 const std::string& source) | |
209 { | |
210 result = source; | |
211 ToUpperCase(result); | |
212 } | |
213 | |
214 void Toolbox::ToLowerCase(std::string& result, | |
215 const std::string& source) | |
216 { | |
217 result = source; | |
218 ToLowerCase(result); | |
219 } | |
220 | |
221 | |
128
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
222 static std::streamsize GetStreamSize(std::istream& f) |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
223 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
224 // http://www.cplusplus.com/reference/iostream/istream/tellg/ |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
225 f.seekg(0, std::ios::end); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
226 std::streamsize size = f.tellg(); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
227 f.seekg(0, std::ios::beg); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
228 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
229 return size; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
230 } |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
231 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
232 |
141 | 233 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 234 void Toolbox::ReadFile(std::string& content, |
235 const std::string& path) | |
236 { | |
125 | 237 if (!IsRegularFile(path)) |
78 | 238 { |
89 | 239 LOG(ERROR) << std::string("The path does not point to a regular file: ") << path; |
78 | 240 throw OrthancException(ErrorCode_RegularFileExpected); |
241 } | |
242 | |
243 boost::filesystem::ifstream f; | |
244 f.open(path, std::ifstream::in | std::ifstream::binary); | |
245 if (!f.good()) | |
246 { | |
247 throw OrthancException(ErrorCode_InexistentFile); | |
248 } | |
249 | |
128
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
250 std::streamsize size = GetStreamSize(f); |
78 | 251 content.resize(size); |
252 if (size != 0) | |
253 { | |
254 f.read(reinterpret_cast<char*>(&content[0]), size); | |
255 } | |
256 | |
257 f.close(); | |
258 } | |
141 | 259 #endif |
78 | 260 |
261 | |
141 | 262 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
128
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
263 bool Toolbox::ReadHeader(std::string& header, |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
264 const std::string& path, |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
265 size_t headerSize) |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
266 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
267 if (!IsRegularFile(path)) |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
268 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
269 LOG(ERROR) << std::string("The path does not point to a regular file: ") << path; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
270 throw OrthancException(ErrorCode_RegularFileExpected); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
271 } |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
272 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
273 boost::filesystem::ifstream f; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
274 f.open(path, std::ifstream::in | std::ifstream::binary); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
275 if (!f.good()) |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
276 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
277 throw OrthancException(ErrorCode_InexistentFile); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
278 } |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
279 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
280 bool full = true; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
281 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
282 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
283 std::streamsize size = GetStreamSize(f); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
284 if (size <= 0) |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
285 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
286 headerSize = 0; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
287 full = false; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
288 } |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
289 else if (static_cast<size_t>(size) < headerSize) |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
290 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
291 headerSize = size; // Truncate to the size of the file |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
292 full = false; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
293 } |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
294 } |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
295 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
296 header.resize(headerSize); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
297 if (headerSize != 0) |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
298 { |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
299 f.read(reinterpret_cast<char*>(&header[0]), headerSize); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
300 } |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
301 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
302 f.close(); |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
303 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
304 return full; |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
305 } |
141 | 306 #endif |
128
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
307 |
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
308 |
141 | 309 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 310 void Toolbox::WriteFile(const void* content, |
311 size_t size, | |
312 const std::string& path) | |
313 { | |
314 boost::filesystem::ofstream f; | |
315 f.open(path, std::ofstream::binary); | |
316 if (!f.good()) | |
317 { | |
318 throw OrthancException(ErrorCode_CannotWriteFile); | |
319 } | |
320 | |
321 if (size != 0) | |
322 { | |
323 f.write(reinterpret_cast<const char*>(content), size); | |
324 } | |
325 | |
326 f.close(); | |
327 } | |
141 | 328 #endif |
78 | 329 |
330 | |
141 | 331 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 332 void Toolbox::WriteFile(const std::string& content, |
333 const std::string& path) | |
334 { | |
335 WriteFile(content.size() > 0 ? content.c_str() : NULL, | |
336 content.size(), path); | |
337 } | |
141 | 338 #endif |
78 | 339 |
340 | |
141 | 341 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 342 void Toolbox::RemoveFile(const std::string& path) |
343 { | |
344 if (boost::filesystem::exists(path)) | |
345 { | |
125 | 346 if (IsRegularFile(path)) |
78 | 347 { |
348 boost::filesystem::remove(path); | |
349 } | |
350 else | |
351 { | |
352 throw OrthancException(ErrorCode_RegularFileExpected); | |
353 } | |
354 } | |
355 } | |
141 | 356 #endif |
78 | 357 |
358 | |
359 void Toolbox::SplitUriComponents(UriComponents& components, | |
360 const std::string& uri) | |
361 { | |
362 static const char URI_SEPARATOR = '/'; | |
363 | |
364 components.clear(); | |
365 | |
366 if (uri.size() == 0 || | |
367 uri[0] != URI_SEPARATOR) | |
368 { | |
369 throw OrthancException(ErrorCode_UriSyntax); | |
370 } | |
371 | |
372 // Count the number of slashes in the URI to make an assumption | |
373 // about the number of components in the URI | |
374 unsigned int estimatedSize = 0; | |
375 for (unsigned int i = 0; i < uri.size(); i++) | |
376 { | |
377 if (uri[i] == URI_SEPARATOR) | |
378 estimatedSize++; | |
379 } | |
380 | |
381 components.reserve(estimatedSize - 1); | |
382 | |
383 unsigned int start = 1; | |
384 unsigned int end = 1; | |
385 while (end < uri.size()) | |
386 { | |
387 // This is the loop invariant | |
388 assert(uri[start - 1] == '/' && (end >= start)); | |
389 | |
390 if (uri[end] == '/') | |
391 { | |
392 components.push_back(std::string(&uri[start], end - start)); | |
393 end++; | |
394 start = end; | |
395 } | |
396 else | |
397 { | |
398 end++; | |
399 } | |
400 } | |
401 | |
402 if (start < uri.size()) | |
403 { | |
404 components.push_back(std::string(&uri[start], end - start)); | |
405 } | |
406 | |
407 for (size_t i = 0; i < components.size(); i++) | |
408 { | |
409 if (components[i].size() == 0) | |
410 { | |
411 // Empty component, as in: "/coucou//e" | |
412 throw OrthancException(ErrorCode_UriSyntax); | |
413 } | |
414 } | |
415 } | |
416 | |
417 | |
418 void Toolbox::TruncateUri(UriComponents& target, | |
419 const UriComponents& source, | |
420 size_t fromLevel) | |
421 { | |
422 target.clear(); | |
423 | |
424 if (source.size() > fromLevel) | |
425 { | |
426 target.resize(source.size() - fromLevel); | |
427 | |
428 size_t j = 0; | |
429 for (size_t i = fromLevel; i < source.size(); i++, j++) | |
430 { | |
431 target[j] = source[i]; | |
432 } | |
433 | |
434 assert(j == target.size()); | |
435 } | |
436 } | |
437 | |
438 | |
439 | |
440 bool Toolbox::IsChildUri(const UriComponents& baseUri, | |
441 const UriComponents& testedUri) | |
442 { | |
443 if (testedUri.size() < baseUri.size()) | |
444 { | |
445 return false; | |
446 } | |
447 | |
448 for (size_t i = 0; i < baseUri.size(); i++) | |
449 { | |
450 if (baseUri[i] != testedUri[i]) | |
451 return false; | |
452 } | |
453 | |
454 return true; | |
455 } | |
456 | |
457 | |
458 std::string Toolbox::AutodetectMimeType(const std::string& path) | |
459 { | |
460 std::string contentType; | |
461 size_t lastDot = path.rfind('.'); | |
462 size_t lastSlash = path.rfind('/'); | |
463 | |
464 if (lastDot == std::string::npos || | |
465 (lastSlash != std::string::npos && lastDot < lastSlash)) | |
466 { | |
467 // No trailing dot, unable to detect the content type | |
468 } | |
469 else | |
470 { | |
471 const char* extension = &path[lastDot + 1]; | |
472 | |
473 // http://en.wikipedia.org/wiki/Mime_types | |
474 // Text types | |
475 if (!strcmp(extension, "txt")) | |
476 contentType = "text/plain"; | |
477 else if (!strcmp(extension, "html")) | |
478 contentType = "text/html"; | |
479 else if (!strcmp(extension, "xml")) | |
480 contentType = "text/xml"; | |
481 else if (!strcmp(extension, "css")) | |
482 contentType = "text/css"; | |
483 | |
484 // Application types | |
485 else if (!strcmp(extension, "js")) | |
486 contentType = "application/javascript"; | |
487 else if (!strcmp(extension, "json")) | |
488 contentType = "application/json"; | |
489 else if (!strcmp(extension, "pdf")) | |
490 contentType = "application/pdf"; | |
491 | |
492 // Images types | |
493 else if (!strcmp(extension, "jpg") || !strcmp(extension, "jpeg")) | |
494 contentType = "image/jpeg"; | |
495 else if (!strcmp(extension, "gif")) | |
496 contentType = "image/gif"; | |
497 else if (!strcmp(extension, "png")) | |
498 contentType = "image/png"; | |
499 } | |
500 | |
501 return contentType; | |
502 } | |
503 | |
504 | |
505 std::string Toolbox::FlattenUri(const UriComponents& components, | |
506 size_t fromLevel) | |
507 { | |
508 if (components.size() <= fromLevel) | |
509 { | |
510 return "/"; | |
511 } | |
512 else | |
513 { | |
514 std::string r; | |
515 | |
516 for (size_t i = fromLevel; i < components.size(); i++) | |
517 { | |
518 r += "/" + components[i]; | |
519 } | |
520 | |
521 return r; | |
522 } | |
523 } | |
524 | |
525 | |
526 | |
141 | 527 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 528 uint64_t Toolbox::GetFileSize(const std::string& path) |
529 { | |
530 try | |
531 { | |
532 return static_cast<uint64_t>(boost::filesystem::file_size(path)); | |
533 } | |
534 catch (boost::filesystem::filesystem_error&) | |
535 { | |
536 throw OrthancException(ErrorCode_InexistentFile); | |
537 } | |
538 } | |
141 | 539 #endif |
78 | 540 |
541 | |
542 #if !defined(ORTHANC_ENABLE_MD5) || ORTHANC_ENABLE_MD5 == 1 | |
543 static char GetHexadecimalCharacter(uint8_t value) | |
544 { | |
545 assert(value < 16); | |
546 | |
547 if (value < 10) | |
82 | 548 { |
78 | 549 return value + '0'; |
82 | 550 } |
78 | 551 else |
82 | 552 { |
78 | 553 return (value - 10) + 'a'; |
82 | 554 } |
78 | 555 } |
556 | |
557 | |
558 void Toolbox::ComputeMD5(std::string& result, | |
559 const std::string& data) | |
560 { | |
561 if (data.size() > 0) | |
562 { | |
563 ComputeMD5(result, &data[0], data.size()); | |
564 } | |
565 else | |
566 { | |
567 ComputeMD5(result, NULL, 0); | |
568 } | |
569 } | |
570 | |
571 | |
572 void Toolbox::ComputeMD5(std::string& result, | |
573 const void* data, | |
90 | 574 size_t size) |
78 | 575 { |
576 md5_state_s state; | |
577 md5_init(&state); | |
578 | |
90 | 579 if (size > 0) |
78 | 580 { |
581 md5_append(&state, | |
582 reinterpret_cast<const md5_byte_t*>(data), | |
90 | 583 static_cast<int>(size)); |
78 | 584 } |
585 | |
586 md5_byte_t actualHash[16]; | |
587 md5_finish(&state, actualHash); | |
588 | |
589 result.resize(32); | |
590 for (unsigned int i = 0; i < 16; i++) | |
591 { | |
82 | 592 result[2 * i] = GetHexadecimalCharacter(static_cast<uint8_t>(actualHash[i] / 16)); |
593 result[2 * i + 1] = GetHexadecimalCharacter(static_cast<uint8_t>(actualHash[i] % 16)); | |
78 | 594 } |
595 } | |
596 #endif | |
597 | |
598 | |
599 #if !defined(ORTHANC_ENABLE_BASE64) || ORTHANC_ENABLE_BASE64 == 1 | |
600 void Toolbox::EncodeBase64(std::string& result, | |
601 const std::string& data) | |
602 { | |
603 result = base64_encode(data); | |
604 } | |
605 | |
606 void Toolbox::DecodeBase64(std::string& result, | |
607 const std::string& data) | |
608 { | |
129 | 609 for (size_t i = 0; i < data.length(); i++) |
610 { | |
611 if (!isalnum(data[i]) && | |
612 data[i] != '+' && | |
613 data[i] != '/' && | |
614 data[i] != '=') | |
615 { | |
616 // This is not a valid character for a Base64 string | |
617 throw OrthancException(ErrorCode_BadFileFormat); | |
618 } | |
619 } | |
620 | |
78 | 621 result = base64_decode(data); |
622 } | |
79 | 623 |
624 | |
625 # if BOOST_HAS_REGEX == 1 | |
129 | 626 bool Toolbox::DecodeDataUriScheme(std::string& mime, |
79 | 627 std::string& content, |
628 const std::string& source) | |
629 { | |
630 boost::regex pattern("data:([^;]+);base64,([a-zA-Z0-9=+/]*)", | |
631 boost::regex::icase /* case insensitive search */); | |
632 | |
633 boost::cmatch what; | |
634 if (regex_match(source.c_str(), what, pattern)) | |
635 { | |
636 mime = what[1]; | |
637 DecodeBase64(content, what[2]); | |
129 | 638 return true; |
79 | 639 } |
640 else | |
641 { | |
129 | 642 return false; |
79 | 643 } |
644 } | |
645 # endif | |
646 | |
90 | 647 |
648 void Toolbox::EncodeDataUriScheme(std::string& result, | |
649 const std::string& mime, | |
650 const std::string& content) | |
651 { | |
652 result = "data:" + mime + ";base64," + base64_encode(content); | |
653 } | |
654 | |
78 | 655 #endif |
656 | |
657 | |
658 | |
659 #if defined(_WIN32) | |
660 static std::string GetPathToExecutableInternal() | |
661 { | |
662 // Yes, this is ugly, but there is no simple way to get the | |
663 // required buffer size, so we use a big constant | |
664 std::vector<char> buffer(32768); | |
665 /*int bytes =*/ GetModuleFileNameA(NULL, &buffer[0], static_cast<DWORD>(buffer.size() - 1)); | |
666 return std::string(&buffer[0]); | |
667 } | |
668 | |
128
e8cfda4c8a2f
Sync + support of JPEG2000 lossless images with YBR_RCT photometric interpretation
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
125
diff
changeset
|
669 #elif defined(__linux__) || defined(__FreeBSD_kernel__) || defined(__FreeBSD__) |
78 | 670 static std::string GetPathToExecutableInternal() |
671 { | |
672 std::vector<char> buffer(PATH_MAX + 1); | |
673 ssize_t bytes = readlink("/proc/self/exe", &buffer[0], buffer.size() - 1); | |
674 if (bytes == 0) | |
675 { | |
676 throw OrthancException(ErrorCode_PathToExecutable); | |
677 } | |
678 | |
679 return std::string(&buffer[0]); | |
680 } | |
681 | |
682 #elif defined(__APPLE__) && defined(__MACH__) | |
683 static std::string GetPathToExecutableInternal() | |
684 { | |
685 char pathbuf[PATH_MAX + 1]; | |
686 unsigned int bufsize = static_cast<int>(sizeof(pathbuf)); | |
687 | |
688 _NSGetExecutablePath( pathbuf, &bufsize); | |
689 | |
690 return std::string(pathbuf); | |
691 } | |
692 | |
141 | 693 #elif defined(ORTHANC_SANDBOXED) && ORTHANC_SANDBOXED == 1 |
694 // Sandboxed Orthanc, no access to the executable | |
695 | |
78 | 696 #else |
697 #error Support your platform here | |
698 #endif | |
699 | |
700 | |
141 | 701 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 702 std::string Toolbox::GetPathToExecutable() |
703 { | |
704 boost::filesystem::path p(GetPathToExecutableInternal()); | |
705 return boost::filesystem::absolute(p).string(); | |
706 } | |
707 | |
708 | |
709 std::string Toolbox::GetDirectoryOfExecutable() | |
710 { | |
711 boost::filesystem::path p(GetPathToExecutableInternal()); | |
712 return boost::filesystem::absolute(p.parent_path()).string(); | |
713 } | |
141 | 714 #endif |
78 | 715 |
716 | |
717 static const char* GetBoostLocaleEncoding(const Encoding sourceEncoding) | |
718 { | |
719 switch (sourceEncoding) | |
720 { | |
721 case Encoding_Utf8: | |
722 return "UTF-8"; | |
723 | |
724 case Encoding_Ascii: | |
725 return "ASCII"; | |
726 | |
727 case Encoding_Latin1: | |
728 return "ISO-8859-1"; | |
729 break; | |
730 | |
731 case Encoding_Latin2: | |
732 return "ISO-8859-2"; | |
733 break; | |
734 | |
735 case Encoding_Latin3: | |
736 return "ISO-8859-3"; | |
737 break; | |
738 | |
739 case Encoding_Latin4: | |
740 return "ISO-8859-4"; | |
741 break; | |
742 | |
743 case Encoding_Latin5: | |
744 return "ISO-8859-9"; | |
745 break; | |
746 | |
747 case Encoding_Cyrillic: | |
748 return "ISO-8859-5"; | |
749 break; | |
750 | |
751 case Encoding_Windows1251: | |
752 return "WINDOWS-1251"; | |
753 break; | |
754 | |
755 case Encoding_Arabic: | |
756 return "ISO-8859-6"; | |
757 break; | |
758 | |
759 case Encoding_Greek: | |
760 return "ISO-8859-7"; | |
761 break; | |
762 | |
763 case Encoding_Hebrew: | |
764 return "ISO-8859-8"; | |
765 break; | |
766 | |
767 case Encoding_Japanese: | |
768 return "SHIFT-JIS"; | |
769 break; | |
770 | |
771 case Encoding_Chinese: | |
772 return "GB18030"; | |
773 break; | |
774 | |
775 case Encoding_Thai: | |
776 return "TIS620.2533-0"; | |
777 break; | |
778 | |
779 default: | |
780 throw OrthancException(ErrorCode_NotImplemented); | |
781 } | |
782 } | |
783 | |
784 | |
785 std::string Toolbox::ConvertToUtf8(const std::string& source, | |
786 Encoding sourceEncoding) | |
787 { | |
788 if (sourceEncoding == Encoding_Utf8) | |
789 { | |
790 // Already in UTF-8: No conversion is required | |
791 return source; | |
792 } | |
793 | |
794 if (sourceEncoding == Encoding_Ascii) | |
795 { | |
796 return ConvertToAscii(source); | |
797 } | |
798 | |
799 const char* encoding = GetBoostLocaleEncoding(sourceEncoding); | |
800 | |
801 try | |
802 { | |
803 return boost::locale::conv::to_utf<char>(source, encoding); | |
804 } | |
805 catch (std::runtime_error&) | |
806 { | |
807 // Bad input string or bad encoding | |
808 return ConvertToAscii(source); | |
809 } | |
810 } | |
811 | |
812 | |
813 std::string Toolbox::ConvertFromUtf8(const std::string& source, | |
814 Encoding targetEncoding) | |
815 { | |
816 if (targetEncoding == Encoding_Utf8) | |
817 { | |
818 // Already in UTF-8: No conversion is required | |
819 return source; | |
820 } | |
821 | |
822 if (targetEncoding == Encoding_Ascii) | |
823 { | |
824 return ConvertToAscii(source); | |
825 } | |
826 | |
827 const char* encoding = GetBoostLocaleEncoding(targetEncoding); | |
828 | |
829 try | |
830 { | |
831 return boost::locale::conv::from_utf<char>(source, encoding); | |
832 } | |
833 catch (std::runtime_error&) | |
834 { | |
835 // Bad input string or bad encoding | |
836 return ConvertToAscii(source); | |
837 } | |
838 } | |
839 | |
840 | |
841 std::string Toolbox::ConvertToAscii(const std::string& source) | |
842 { | |
843 std::string result; | |
844 | |
845 result.reserve(source.size() + 1); | |
846 for (size_t i = 0; i < source.size(); i++) | |
847 { | |
848 if (source[i] <= 127 && source[i] >= 0 && !iscntrl(source[i])) | |
849 { | |
850 result.push_back(source[i]); | |
851 } | |
852 } | |
853 | |
854 return result; | |
855 } | |
856 | |
90 | 857 |
78 | 858 void Toolbox::ComputeSHA1(std::string& result, |
90 | 859 const void* data, |
860 size_t size) | |
78 | 861 { |
862 boost::uuids::detail::sha1 sha1; | |
863 | |
90 | 864 if (size > 0) |
78 | 865 { |
90 | 866 sha1.process_bytes(data, size); |
78 | 867 } |
868 | |
869 unsigned int digest[5]; | |
870 | |
871 // Sanity check for the memory layout: A SHA-1 digest is 160 bits wide | |
872 assert(sizeof(unsigned int) == 4 && sizeof(digest) == (160 / 8)); | |
873 | |
874 sha1.get_digest(digest); | |
875 | |
876 result.resize(8 * 5 + 4); | |
877 sprintf(&result[0], "%08x-%08x-%08x-%08x-%08x", | |
878 digest[0], | |
879 digest[1], | |
880 digest[2], | |
881 digest[3], | |
882 digest[4]); | |
883 } | |
884 | |
90 | 885 void Toolbox::ComputeSHA1(std::string& result, |
886 const std::string& data) | |
887 { | |
888 if (data.size() > 0) | |
889 { | |
890 ComputeSHA1(result, data.c_str(), data.size()); | |
891 } | |
892 else | |
893 { | |
894 ComputeSHA1(result, NULL, 0); | |
895 } | |
896 } | |
897 | |
898 | |
78 | 899 bool Toolbox::IsSHA1(const char* str, |
900 size_t size) | |
901 { | |
902 if (size == 0) | |
903 { | |
904 return false; | |
905 } | |
906 | |
907 const char* start = str; | |
908 const char* end = str + size; | |
909 | |
910 // Trim the beginning of the string | |
911 while (start < end) | |
912 { | |
913 if (*start == '\0' || | |
914 isspace(*start)) | |
915 { | |
916 start++; | |
917 } | |
918 else | |
919 { | |
920 break; | |
921 } | |
922 } | |
923 | |
924 // Trim the trailing of the string | |
925 while (start < end) | |
926 { | |
927 if (*(end - 1) == '\0' || | |
928 isspace(*(end - 1))) | |
929 { | |
930 end--; | |
931 } | |
932 else | |
933 { | |
934 break; | |
935 } | |
936 } | |
937 | |
938 if (end - start != 44) | |
939 { | |
940 return false; | |
941 } | |
942 | |
943 for (unsigned int i = 0; i < 44; i++) | |
944 { | |
945 if (i == 8 || | |
946 i == 17 || | |
947 i == 26 || | |
948 i == 35) | |
949 { | |
950 if (start[i] != '-') | |
951 return false; | |
952 } | |
953 else | |
954 { | |
955 if (!isalnum(start[i])) | |
956 return false; | |
957 } | |
958 } | |
959 | |
960 return true; | |
961 } | |
962 | |
963 | |
964 bool Toolbox::IsSHA1(const std::string& s) | |
965 { | |
966 if (s.size() == 0) | |
967 { | |
968 return false; | |
969 } | |
970 else | |
971 { | |
972 return IsSHA1(s.c_str(), s.size()); | |
973 } | |
974 } | |
975 | |
976 | |
977 #if BOOST_HAS_DATE_TIME == 1 | |
978 std::string Toolbox::GetNowIsoString() | |
979 { | |
980 boost::posix_time::ptime now = boost::posix_time::second_clock::local_time(); | |
981 return boost::posix_time::to_iso_string(now); | |
982 } | |
983 | |
984 void Toolbox::GetNowDicom(std::string& date, | |
985 std::string& time) | |
986 { | |
987 boost::posix_time::ptime now = boost::posix_time::second_clock::local_time(); | |
988 tm tm = boost::posix_time::to_tm(now); | |
989 | |
990 char s[32]; | |
991 sprintf(s, "%04d%02d%02d", tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday); | |
992 date.assign(s); | |
993 | |
994 // TODO milliseconds | |
995 sprintf(s, "%02d%02d%02d.%06d", tm.tm_hour, tm.tm_min, tm.tm_sec, 0); | |
996 time.assign(s); | |
997 } | |
998 #endif | |
999 | |
1000 | |
1001 std::string Toolbox::StripSpaces(const std::string& source) | |
1002 { | |
1003 size_t first = 0; | |
1004 | |
1005 while (first < source.length() && | |
1006 isspace(source[first])) | |
1007 { | |
1008 first++; | |
1009 } | |
1010 | |
1011 if (first == source.length()) | |
1012 { | |
1013 // String containing only spaces | |
1014 return ""; | |
1015 } | |
1016 | |
1017 size_t last = source.length(); | |
1018 while (last > first && | |
1019 isspace(source[last - 1])) | |
1020 { | |
1021 last--; | |
1022 } | |
1023 | |
1024 assert(first <= last); | |
1025 return source.substr(first, last - first); | |
1026 } | |
1027 | |
1028 | |
1029 static char Hex2Dec(char c) | |
1030 { | |
1031 return ((c >= '0' && c <= '9') ? c - '0' : | |
1032 ((c >= 'a' && c <= 'f') ? c - 'a' + 10 : c - 'A' + 10)); | |
1033 } | |
1034 | |
1035 void Toolbox::UrlDecode(std::string& s) | |
1036 { | |
1037 // http://en.wikipedia.org/wiki/Percent-encoding | |
1038 // http://www.w3schools.com/tags/ref_urlencode.asp | |
1039 // http://stackoverflow.com/questions/154536/encode-decode-urls-in-c | |
1040 | |
1041 if (s.size() == 0) | |
1042 { | |
1043 return; | |
1044 } | |
1045 | |
1046 size_t source = 0; | |
1047 size_t target = 0; | |
1048 | |
1049 while (source < s.size()) | |
1050 { | |
1051 if (s[source] == '%' && | |
1052 source + 2 < s.size() && | |
1053 isalnum(s[source + 1]) && | |
1054 isalnum(s[source + 2])) | |
1055 { | |
1056 s[target] = (Hex2Dec(s[source + 1]) << 4) | Hex2Dec(s[source + 2]); | |
1057 source += 3; | |
1058 target += 1; | |
1059 } | |
1060 else | |
1061 { | |
1062 if (s[source] == '+') | |
1063 s[target] = ' '; | |
1064 else | |
1065 s[target] = s[source]; | |
1066 | |
1067 source++; | |
1068 target++; | |
1069 } | |
1070 } | |
1071 | |
1072 s.resize(target); | |
1073 } | |
1074 | |
1075 | |
1076 Endianness Toolbox::DetectEndianness() | |
1077 { | |
1078 // http://sourceforge.net/p/predef/wiki/Endianness/ | |
1079 | |
1080 uint8_t buffer[4]; | |
1081 | |
1082 buffer[0] = 0x00; | |
1083 buffer[1] = 0x01; | |
1084 buffer[2] = 0x02; | |
1085 buffer[3] = 0x03; | |
1086 | |
1087 switch (*((uint32_t *)buffer)) | |
1088 { | |
1089 case 0x00010203: | |
1090 return Endianness_Big; | |
1091 | |
1092 case 0x03020100: | |
1093 return Endianness_Little; | |
1094 | |
1095 default: | |
1096 throw OrthancException(ErrorCode_NotImplemented); | |
1097 } | |
1098 } | |
1099 | |
1100 | |
1101 #if BOOST_HAS_REGEX == 1 | |
1102 std::string Toolbox::WildcardToRegularExpression(const std::string& source) | |
1103 { | |
1104 // TODO - Speed up this with a regular expression | |
1105 | |
1106 std::string result = source; | |
1107 | |
1108 // Escape all special characters | |
1109 boost::replace_all(result, "\\", "\\\\"); | |
1110 boost::replace_all(result, "^", "\\^"); | |
1111 boost::replace_all(result, ".", "\\."); | |
1112 boost::replace_all(result, "$", "\\$"); | |
1113 boost::replace_all(result, "|", "\\|"); | |
1114 boost::replace_all(result, "(", "\\("); | |
1115 boost::replace_all(result, ")", "\\)"); | |
1116 boost::replace_all(result, "[", "\\["); | |
1117 boost::replace_all(result, "]", "\\]"); | |
1118 boost::replace_all(result, "+", "\\+"); | |
1119 boost::replace_all(result, "/", "\\/"); | |
1120 boost::replace_all(result, "{", "\\{"); | |
1121 boost::replace_all(result, "}", "\\}"); | |
1122 | |
1123 // Convert wildcards '*' and '?' to their regex equivalents | |
1124 boost::replace_all(result, "?", "."); | |
1125 boost::replace_all(result, "*", ".*"); | |
1126 | |
1127 return result; | |
1128 } | |
1129 #endif | |
1130 | |
1131 | |
1132 | |
1133 void Toolbox::TokenizeString(std::vector<std::string>& result, | |
1134 const std::string& value, | |
1135 char separator) | |
1136 { | |
1137 result.clear(); | |
1138 | |
1139 std::string currentItem; | |
1140 | |
1141 for (size_t i = 0; i < value.size(); i++) | |
1142 { | |
1143 if (value[i] == separator) | |
1144 { | |
1145 result.push_back(currentItem); | |
1146 currentItem.clear(); | |
1147 } | |
1148 else | |
1149 { | |
1150 currentItem.push_back(value[i]); | |
1151 } | |
1152 } | |
1153 | |
1154 result.push_back(currentItem); | |
1155 } | |
1156 | |
1157 | |
141 | 1158 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 1159 void Toolbox::MakeDirectory(const std::string& path) |
1160 { | |
1161 if (boost::filesystem::exists(path)) | |
1162 { | |
1163 if (!boost::filesystem::is_directory(path)) | |
1164 { | |
1165 throw OrthancException(ErrorCode_DirectoryOverFile); | |
1166 } | |
1167 } | |
1168 else | |
1169 { | |
1170 if (!boost::filesystem::create_directories(path)) | |
1171 { | |
1172 throw OrthancException(ErrorCode_MakeDirectory); | |
1173 } | |
1174 } | |
1175 } | |
141 | 1176 #endif |
78 | 1177 |
1178 | |
141 | 1179 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 1180 bool Toolbox::IsExistingFile(const std::string& path) |
1181 { | |
1182 return boost::filesystem::exists(path); | |
1183 } | |
141 | 1184 #endif |
78 | 1185 |
1186 | |
1187 #if ORTHANC_PUGIXML_ENABLED == 1 | |
1188 class ChunkedBufferWriter : public pugi::xml_writer | |
1189 { | |
1190 private: | |
1191 ChunkedBuffer buffer_; | |
1192 | |
1193 public: | |
1194 virtual void write(const void *data, size_t size) | |
1195 { | |
1196 if (size > 0) | |
1197 { | |
1198 buffer_.AddChunk(reinterpret_cast<const char*>(data), size); | |
1199 } | |
1200 } | |
1201 | |
1202 void Flatten(std::string& s) | |
1203 { | |
1204 buffer_.Flatten(s); | |
1205 } | |
1206 }; | |
1207 | |
1208 | |
1209 static void JsonToXmlInternal(pugi::xml_node& target, | |
1210 const Json::Value& source, | |
1211 const std::string& arrayElement) | |
1212 { | |
1213 // http://jsoncpp.sourceforge.net/value_8h_source.html#l00030 | |
1214 | |
1215 switch (source.type()) | |
1216 { | |
1217 case Json::nullValue: | |
1218 { | |
1219 target.append_child(pugi::node_pcdata).set_value("null"); | |
1220 break; | |
1221 } | |
1222 | |
1223 case Json::intValue: | |
1224 { | |
1225 std::string s = boost::lexical_cast<std::string>(source.asInt()); | |
1226 target.append_child(pugi::node_pcdata).set_value(s.c_str()); | |
1227 break; | |
1228 } | |
1229 | |
1230 case Json::uintValue: | |
1231 { | |
1232 std::string s = boost::lexical_cast<std::string>(source.asUInt()); | |
1233 target.append_child(pugi::node_pcdata).set_value(s.c_str()); | |
1234 break; | |
1235 } | |
1236 | |
1237 case Json::realValue: | |
1238 { | |
1239 std::string s = boost::lexical_cast<std::string>(source.asFloat()); | |
1240 target.append_child(pugi::node_pcdata).set_value(s.c_str()); | |
1241 break; | |
1242 } | |
1243 | |
1244 case Json::stringValue: | |
1245 { | |
1246 target.append_child(pugi::node_pcdata).set_value(source.asString().c_str()); | |
1247 break; | |
1248 } | |
1249 | |
1250 case Json::booleanValue: | |
1251 { | |
1252 target.append_child(pugi::node_pcdata).set_value(source.asBool() ? "true" : "false"); | |
1253 break; | |
1254 } | |
1255 | |
1256 case Json::arrayValue: | |
1257 { | |
1258 for (Json::Value::ArrayIndex i = 0; i < source.size(); i++) | |
1259 { | |
1260 pugi::xml_node node = target.append_child(); | |
1261 node.set_name(arrayElement.c_str()); | |
1262 JsonToXmlInternal(node, source[i], arrayElement); | |
1263 } | |
1264 break; | |
1265 } | |
1266 | |
1267 case Json::objectValue: | |
1268 { | |
1269 Json::Value::Members members = source.getMemberNames(); | |
1270 | |
1271 for (size_t i = 0; i < members.size(); i++) | |
1272 { | |
1273 pugi::xml_node node = target.append_child(); | |
1274 node.set_name(members[i].c_str()); | |
1275 JsonToXmlInternal(node, source[members[i]], arrayElement); | |
1276 } | |
1277 | |
1278 break; | |
1279 } | |
1280 | |
1281 default: | |
1282 throw OrthancException(ErrorCode_NotImplemented); | |
1283 } | |
1284 } | |
1285 | |
1286 | |
1287 void Toolbox::JsonToXml(std::string& target, | |
1288 const Json::Value& source, | |
1289 const std::string& rootElement, | |
1290 const std::string& arrayElement) | |
1291 { | |
1292 pugi::xml_document doc; | |
1293 | |
1294 pugi::xml_node n = doc.append_child(rootElement.c_str()); | |
1295 JsonToXmlInternal(n, source, arrayElement); | |
1296 | |
1297 pugi::xml_node decl = doc.prepend_child(pugi::node_declaration); | |
1298 decl.append_attribute("version").set_value("1.0"); | |
1299 decl.append_attribute("encoding").set_value("utf-8"); | |
1300 | |
1301 ChunkedBufferWriter writer; | |
1302 doc.save(writer, " ", pugi::format_default, pugi::encoding_utf8); | |
1303 writer.Flatten(target); | |
1304 } | |
1305 | |
1306 #endif | |
1307 | |
1308 | |
141 | 1309 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
78 | 1310 void Toolbox::ExecuteSystemCommand(const std::string& command, |
1311 const std::vector<std::string>& arguments) | |
1312 { | |
1313 // Convert the arguments as a C array | |
1314 std::vector<char*> args(arguments.size() + 2); | |
1315 | |
1316 args.front() = const_cast<char*>(command.c_str()); | |
1317 | |
1318 for (size_t i = 0; i < arguments.size(); i++) | |
1319 { | |
1320 args[i + 1] = const_cast<char*>(arguments[i].c_str()); | |
1321 } | |
1322 | |
1323 args.back() = NULL; | |
1324 | |
1325 int status; | |
1326 | |
1327 #if defined(_WIN32) | |
1328 // http://msdn.microsoft.com/en-us/library/275khfab.aspx | |
1329 status = static_cast<int>(_spawnvp(_P_OVERLAY, command.c_str(), &args[0])); | |
1330 | |
1331 #else | |
1332 int pid = fork(); | |
1333 | |
1334 if (pid == -1) | |
1335 { | |
1336 // Error in fork() | |
1337 #if ORTHANC_ENABLE_LOGGING == 1 | |
1338 LOG(ERROR) << "Cannot fork a child process"; | |
1339 #endif | |
1340 | |
1341 throw OrthancException(ErrorCode_SystemCommand); | |
1342 } | |
1343 else if (pid == 0) | |
1344 { | |
1345 // Execute the system command in the child process | |
1346 execvp(command.c_str(), &args[0]); | |
1347 | |
1348 // We should never get here | |
1349 _exit(1); | |
1350 } | |
1351 else | |
1352 { | |
1353 // Wait for the system command to exit | |
1354 waitpid(pid, &status, 0); | |
1355 } | |
1356 #endif | |
1357 | |
1358 if (status != 0) | |
1359 { | |
1360 #if ORTHANC_ENABLE_LOGGING == 1 | |
1361 LOG(ERROR) << "System command failed with status code " << status; | |
1362 #endif | |
1363 | |
1364 throw OrthancException(ErrorCode_SystemCommand); | |
1365 } | |
1366 } | |
141 | 1367 #endif |
78 | 1368 |
1369 | |
1370 bool Toolbox::IsInteger(const std::string& str) | |
1371 { | |
1372 std::string s = StripSpaces(str); | |
1373 | |
1374 if (s.size() == 0) | |
1375 { | |
1376 return false; | |
1377 } | |
1378 | |
1379 size_t pos = 0; | |
1380 if (s[0] == '-') | |
1381 { | |
1382 if (s.size() == 1) | |
1383 { | |
1384 return false; | |
1385 } | |
1386 | |
1387 pos = 1; | |
1388 } | |
1389 | |
1390 while (pos < s.size()) | |
1391 { | |
1392 if (!isdigit(s[pos])) | |
1393 { | |
1394 return false; | |
1395 } | |
1396 | |
1397 pos++; | |
1398 } | |
1399 | |
1400 return true; | |
1401 } | |
1402 | |
1403 | |
1404 void Toolbox::CopyJsonWithoutComments(Json::Value& target, | |
1405 const Json::Value& source) | |
1406 { | |
1407 switch (source.type()) | |
1408 { | |
1409 case Json::nullValue: | |
1410 target = Json::nullValue; | |
1411 break; | |
1412 | |
1413 case Json::intValue: | |
1414 target = source.asInt64(); | |
1415 break; | |
1416 | |
1417 case Json::uintValue: | |
1418 target = source.asUInt64(); | |
1419 break; | |
1420 | |
1421 case Json::realValue: | |
1422 target = source.asDouble(); | |
1423 break; | |
1424 | |
1425 case Json::stringValue: | |
1426 target = source.asString(); | |
1427 break; | |
1428 | |
1429 case Json::booleanValue: | |
1430 target = source.asBool(); | |
1431 break; | |
1432 | |
1433 case Json::arrayValue: | |
1434 { | |
1435 target = Json::arrayValue; | |
1436 for (Json::Value::ArrayIndex i = 0; i < source.size(); i++) | |
1437 { | |
1438 Json::Value& item = target.append(Json::nullValue); | |
1439 CopyJsonWithoutComments(item, source[i]); | |
1440 } | |
1441 | |
1442 break; | |
1443 } | |
1444 | |
1445 case Json::objectValue: | |
1446 { | |
1447 target = Json::objectValue; | |
1448 Json::Value::Members members = source.getMemberNames(); | |
1449 for (Json::Value::ArrayIndex i = 0; i < members.size(); i++) | |
1450 { | |
1451 const std::string item = members[i]; | |
1452 CopyJsonWithoutComments(target[item], source[item]); | |
1453 } | |
1454 | |
1455 break; | |
1456 } | |
1457 | |
1458 default: | |
1459 break; | |
1460 } | |
1461 } | |
1462 | |
1463 | |
1464 bool Toolbox::StartsWith(const std::string& str, | |
1465 const std::string& prefix) | |
1466 { | |
1467 if (str.size() < prefix.size()) | |
1468 { | |
1469 return false; | |
1470 } | |
1471 else | |
1472 { | |
1473 return str.compare(0, prefix.size(), prefix) == 0; | |
1474 } | |
1475 } | |
1476 | |
1477 | |
1478 int Toolbox::GetProcessId() | |
1479 { | |
1480 #if defined(_WIN32) | |
1481 return static_cast<int>(_getpid()); | |
1482 #else | |
1483 return static_cast<int>(getpid()); | |
1484 #endif | |
1485 } | |
125 | 1486 |
1487 | |
141 | 1488 #if !defined(ORTHANC_SANDBOXED) || ORTHANC_SANDBOXED != 1 |
125 | 1489 bool Toolbox::IsRegularFile(const std::string& path) |
1490 { | |
1491 namespace fs = boost::filesystem; | |
1492 | |
1493 try | |
1494 { | |
1495 if (fs::exists(path)) | |
1496 { | |
1497 fs::file_status status = fs::status(path); | |
1498 return (status.type() == boost::filesystem::regular_file || | |
1499 status.type() == boost::filesystem::reparse_file); // Fix BitBucket issue #11 | |
1500 } | |
1501 } | |
1502 catch (fs::filesystem_error&) | |
1503 { | |
1504 } | |
1505 | |
1506 return false; | |
1507 } | |
141 | 1508 #endif |
1509 | |
1510 | |
1511 FILE* Toolbox::OpenFile(const std::string& path, | |
1512 FileMode mode) | |
1513 { | |
1514 #if defined(_WIN32) | |
1515 // TODO Deal with special characters by converting to the current locale | |
1516 #endif | |
1517 | |
1518 const char* m; | |
1519 switch (mode) | |
1520 { | |
1521 case FileMode_ReadBinary: | |
1522 m = "rb"; | |
1523 break; | |
1524 | |
1525 case FileMode_WriteBinary: | |
1526 m = "wb"; | |
1527 break; | |
1528 | |
1529 default: | |
1530 throw OrthancException(ErrorCode_ParameterOutOfRange); | |
1531 } | |
1532 | |
1533 return fopen(path.c_str(), m); | |
1534 } | |
1535 | |
1536 | |
1537 | |
1538 static bool IsUnreservedCharacter(char c) | |
1539 { | |
1540 // This function checks whether "c" is an unserved character | |
1541 // wrt. an URI percent-encoding | |
1542 // https://en.wikipedia.org/wiki/Percent-encoding#Percent-encoding%5Fin%5Fa%5FURI | |
1543 | |
1544 return ((c >= 'A' && c <= 'Z') || | |
1545 (c >= 'a' && c <= 'z') || | |
1546 (c >= '0' && c <= '9') || | |
1547 c == '-' || | |
1548 c == '_' || | |
1549 c == '.' || | |
1550 c == '~'); | |
1551 } | |
1552 | |
1553 void Toolbox::UriEncode(std::string& target, | |
1554 const std::string& source) | |
1555 { | |
1556 // Estimate the length of the percent-encoded URI | |
1557 size_t length = 0; | |
1558 | |
1559 for (size_t i = 0; i < source.size(); i++) | |
1560 { | |
1561 if (IsUnreservedCharacter(source[i])) | |
1562 { | |
1563 length += 1; | |
1564 } | |
1565 else | |
1566 { | |
1567 // This character must be percent-encoded | |
1568 length += 3; | |
1569 } | |
1570 } | |
1571 | |
1572 target.clear(); | |
1573 target.reserve(length); | |
1574 | |
1575 for (size_t i = 0; i < source.size(); i++) | |
1576 { | |
1577 if (IsUnreservedCharacter(source[i])) | |
1578 { | |
1579 target.push_back(source[i]); | |
1580 } | |
1581 else | |
1582 { | |
1583 // This character must be percent-encoded | |
1584 uint8_t byte = static_cast<uint8_t>(source[i]); | |
1585 uint8_t a = byte >> 4; | |
1586 uint8_t b = byte & 0x0f; | |
1587 | |
1588 target.push_back('%'); | |
1589 target.push_back(a < 10 ? a + '0' : a - 10 + 'A'); | |
1590 target.push_back(b < 10 ? b + '0' : b - 10 + 'A'); | |
1591 } | |
1592 } | |
1593 } | |
1594 | |
1595 | |
1596 static bool HasField(const Json::Value& json, | |
1597 const std::string& key, | |
1598 Json::ValueType expectedType) | |
1599 { | |
1600 if (json.type() != Json::objectValue || | |
1601 !json.isMember(key)) | |
1602 { | |
1603 return false; | |
1604 } | |
1605 else if (json[key].type() == expectedType) | |
1606 { | |
1607 return true; | |
1608 } | |
1609 else | |
1610 { | |
1611 throw OrthancException(ErrorCode_BadParameterType); | |
1612 } | |
1613 } | |
1614 | |
1615 | |
1616 std::string Toolbox::GetJsonStringField(const Json::Value& json, | |
1617 const std::string& key, | |
1618 const std::string& defaultValue) | |
1619 { | |
1620 if (HasField(json, key, Json::stringValue)) | |
1621 { | |
1622 return json[key].asString(); | |
1623 } | |
1624 else | |
1625 { | |
1626 return defaultValue; | |
1627 } | |
1628 } | |
1629 | |
1630 | |
1631 bool Toolbox::GetJsonBooleanField(const ::Json::Value& json, | |
1632 const std::string& key, | |
1633 bool defaultValue) | |
1634 { | |
1635 if (HasField(json, key, Json::booleanValue)) | |
1636 { | |
1637 return json[key].asBool(); | |
1638 } | |
1639 else | |
1640 { | |
1641 return defaultValue; | |
1642 } | |
1643 } | |
1644 | |
1645 | |
1646 int Toolbox::GetJsonIntegerField(const ::Json::Value& json, | |
1647 const std::string& key, | |
1648 int defaultValue) | |
1649 { | |
1650 if (HasField(json, key, Json::intValue)) | |
1651 { | |
1652 return json[key].asInt(); | |
1653 } | |
1654 else | |
1655 { | |
1656 return defaultValue; | |
1657 } | |
1658 } | |
1659 | |
1660 | |
1661 unsigned int Toolbox::GetJsonUnsignedIntegerField(const ::Json::Value& json, | |
1662 const std::string& key, | |
1663 unsigned int defaultValue) | |
1664 { | |
1665 int v = GetJsonIntegerField(json, key, defaultValue); | |
1666 | |
1667 if (v < 0) | |
1668 { | |
1669 throw OrthancException(ErrorCode_ParameterOutOfRange); | |
1670 } | |
1671 else | |
1672 { | |
1673 return static_cast<unsigned int>(v); | |
1674 } | |
1675 } | |
78 | 1676 } |