Mercurial > hg > orthanc-wsi
annotate Resources/Orthanc/Core/Toolbox.cpp @ 98:ff0ef01c332c
shared copyright with osimis
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Wed, 04 Jan 2017 16:43:25 +0100 |
| parents | 4b0e0f7c9957 |
| children | a3e8ac8b7256 |
| rev | line source |
|---|---|
| 1 | 1 /** |
| 2 * Orthanc - A Lightweight, RESTful DICOM Store | |
| 3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
| 4 * Department, University Hospital of Liege, Belgium | |
|
98
ff0ef01c332c
shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
92
diff
changeset
|
5 * Copyright (C) 2017 Osimis, Belgium |
| 1 | 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 #include "PrecompiledHeaders.h" | |
| 35 #include "Toolbox.h" | |
| 36 | |
| 37 #include "OrthancException.h" | |
| 38 #include "Logging.h" | |
| 39 | |
| 43 | 40 #include <boost/algorithm/string/replace.hpp> |
| 41 #include <boost/lexical_cast.hpp> | |
| 42 #include <boost/locale.hpp> | |
| 43 #include <boost/uuid/sha1.hpp> | |
| 44 | |
| 1 | 45 #include <string> |
| 46 #include <stdint.h> | |
| 47 #include <string.h> | |
| 48 #include <algorithm> | |
| 49 #include <ctype.h> | |
| 50 | |
| 51 #if BOOST_HAS_REGEX == 1 | |
| 43 | 52 # include <boost/regex.hpp> |
| 1 | 53 #endif |
| 54 | |
| 55 #if BOOST_HAS_LOCALE != 1 | |
| 43 | 56 # error Since version 0.7.6, Orthanc entirely relies on boost::locale |
| 1 | 57 #endif |
| 58 | |
| 41 | 59 #if ORTHANC_ENABLE_MD5 == 1 |
| 60 # include "../Resources/ThirdParty/md5/md5.h" | |
| 1 | 61 #endif |
| 62 | |
| 41 | 63 #if ORTHANC_ENABLE_BASE64 == 1 |
| 64 # include "../Resources/ThirdParty/base64/base64.h" | |
| 1 | 65 #endif |
| 66 | |
| 67 | |
| 68 #if defined(_MSC_VER) && (_MSC_VER < 1800) | |
| 69 // Patch for the missing "_strtoll" symbol when compiling with Visual Studio < 2013 | |
| 70 extern "C" | |
| 71 { | |
| 72 int64_t _strtoi64(const char *nptr, char **endptr, int base); | |
| 73 int64_t strtoll(const char *nptr, char **endptr, int base) | |
| 74 { | |
| 75 return _strtoi64(nptr, endptr, base); | |
| 76 } | |
| 77 } | |
| 78 #endif | |
| 79 | |
| 80 | |
| 54 | 81 #if defined(_WIN32) |
| 82 # include <windows.h> // For ::Sleep | |
| 43 | 83 #endif |
| 84 | |
| 85 | |
| 39 | 86 #if ORTHANC_ENABLE_PUGIXML == 1 |
| 43 | 87 # include "ChunkedBuffer.h" |
| 88 # include <pugixml.hpp> | |
| 1 | 89 #endif |
| 90 | |
| 91 | |
| 92 namespace Orthanc | |
| 93 { | |
| 94 void Toolbox::ToUpperCase(std::string& s) | |
| 95 { | |
| 96 std::transform(s.begin(), s.end(), s.begin(), toupper); | |
| 97 } | |
| 98 | |
| 99 | |
| 100 void Toolbox::ToLowerCase(std::string& s) | |
| 101 { | |
| 102 std::transform(s.begin(), s.end(), s.begin(), tolower); | |
| 103 } | |
| 104 | |
| 105 | |
| 106 void Toolbox::ToUpperCase(std::string& result, | |
| 107 const std::string& source) | |
| 108 { | |
| 109 result = source; | |
| 110 ToUpperCase(result); | |
| 111 } | |
| 112 | |
| 113 void Toolbox::ToLowerCase(std::string& result, | |
| 114 const std::string& source) | |
| 115 { | |
| 116 result = source; | |
| 117 ToLowerCase(result); | |
| 118 } | |
| 119 | |
| 120 | |
| 121 void Toolbox::SplitUriComponents(UriComponents& components, | |
| 122 const std::string& uri) | |
| 123 { | |
| 124 static const char URI_SEPARATOR = '/'; | |
| 125 | |
| 126 components.clear(); | |
| 127 | |
| 128 if (uri.size() == 0 || | |
| 129 uri[0] != URI_SEPARATOR) | |
| 130 { | |
| 131 throw OrthancException(ErrorCode_UriSyntax); | |
| 132 } | |
| 133 | |
| 134 // Count the number of slashes in the URI to make an assumption | |
| 135 // about the number of components in the URI | |
| 136 unsigned int estimatedSize = 0; | |
| 137 for (unsigned int i = 0; i < uri.size(); i++) | |
| 138 { | |
| 139 if (uri[i] == URI_SEPARATOR) | |
| 140 estimatedSize++; | |
| 141 } | |
| 142 | |
| 143 components.reserve(estimatedSize - 1); | |
| 144 | |
| 145 unsigned int start = 1; | |
| 146 unsigned int end = 1; | |
| 147 while (end < uri.size()) | |
| 148 { | |
| 149 // This is the loop invariant | |
| 150 assert(uri[start - 1] == '/' && (end >= start)); | |
| 151 | |
| 152 if (uri[end] == '/') | |
| 153 { | |
| 154 components.push_back(std::string(&uri[start], end - start)); | |
| 155 end++; | |
| 156 start = end; | |
| 157 } | |
| 158 else | |
| 159 { | |
| 160 end++; | |
| 161 } | |
| 162 } | |
| 163 | |
| 164 if (start < uri.size()) | |
| 165 { | |
| 166 components.push_back(std::string(&uri[start], end - start)); | |
| 167 } | |
| 168 | |
| 169 for (size_t i = 0; i < components.size(); i++) | |
| 170 { | |
| 171 if (components[i].size() == 0) | |
| 172 { | |
| 173 // Empty component, as in: "/coucou//e" | |
| 174 throw OrthancException(ErrorCode_UriSyntax); | |
| 175 } | |
| 176 } | |
| 177 } | |
| 178 | |
| 179 | |
| 180 void Toolbox::TruncateUri(UriComponents& target, | |
| 181 const UriComponents& source, | |
| 182 size_t fromLevel) | |
| 183 { | |
| 184 target.clear(); | |
| 185 | |
| 186 if (source.size() > fromLevel) | |
| 187 { | |
| 188 target.resize(source.size() - fromLevel); | |
| 189 | |
| 190 size_t j = 0; | |
| 191 for (size_t i = fromLevel; i < source.size(); i++, j++) | |
| 192 { | |
| 193 target[j] = source[i]; | |
| 194 } | |
| 195 | |
| 196 assert(j == target.size()); | |
| 197 } | |
| 198 } | |
| 199 | |
| 200 | |
| 201 | |
| 202 bool Toolbox::IsChildUri(const UriComponents& baseUri, | |
| 203 const UriComponents& testedUri) | |
| 204 { | |
| 205 if (testedUri.size() < baseUri.size()) | |
| 206 { | |
| 207 return false; | |
| 208 } | |
| 209 | |
| 210 for (size_t i = 0; i < baseUri.size(); i++) | |
| 211 { | |
| 212 if (baseUri[i] != testedUri[i]) | |
| 213 return false; | |
| 214 } | |
| 215 | |
| 216 return true; | |
| 217 } | |
| 218 | |
| 219 | |
| 220 std::string Toolbox::AutodetectMimeType(const std::string& path) | |
| 221 { | |
| 222 std::string contentType; | |
| 223 size_t lastDot = path.rfind('.'); | |
| 224 size_t lastSlash = path.rfind('/'); | |
| 225 | |
| 226 if (lastDot == std::string::npos || | |
| 227 (lastSlash != std::string::npos && lastDot < lastSlash)) | |
| 228 { | |
| 229 // No trailing dot, unable to detect the content type | |
| 230 } | |
| 231 else | |
| 232 { | |
| 233 const char* extension = &path[lastDot + 1]; | |
| 234 | |
| 235 // http://en.wikipedia.org/wiki/Mime_types | |
| 236 // Text types | |
| 237 if (!strcmp(extension, "txt")) | |
| 238 contentType = "text/plain"; | |
| 239 else if (!strcmp(extension, "html")) | |
| 240 contentType = "text/html"; | |
| 241 else if (!strcmp(extension, "xml")) | |
| 242 contentType = "text/xml"; | |
| 243 else if (!strcmp(extension, "css")) | |
| 244 contentType = "text/css"; | |
| 245 | |
| 246 // Application types | |
| 247 else if (!strcmp(extension, "js")) | |
| 248 contentType = "application/javascript"; | |
| 249 else if (!strcmp(extension, "json")) | |
| 250 contentType = "application/json"; | |
| 251 else if (!strcmp(extension, "pdf")) | |
| 252 contentType = "application/pdf"; | |
| 253 | |
| 254 // Images types | |
| 255 else if (!strcmp(extension, "jpg") || !strcmp(extension, "jpeg")) | |
| 256 contentType = "image/jpeg"; | |
| 257 else if (!strcmp(extension, "gif")) | |
| 258 contentType = "image/gif"; | |
| 259 else if (!strcmp(extension, "png")) | |
| 260 contentType = "image/png"; | |
| 261 } | |
| 262 | |
| 263 return contentType; | |
| 264 } | |
| 265 | |
| 266 | |
| 267 std::string Toolbox::FlattenUri(const UriComponents& components, | |
| 268 size_t fromLevel) | |
| 269 { | |
| 270 if (components.size() <= fromLevel) | |
| 271 { | |
| 272 return "/"; | |
| 273 } | |
| 274 else | |
| 275 { | |
| 276 std::string r; | |
| 277 | |
| 278 for (size_t i = fromLevel; i < components.size(); i++) | |
| 279 { | |
| 280 r += "/" + components[i]; | |
| 281 } | |
| 282 | |
| 283 return r; | |
| 284 } | |
| 285 } | |
| 286 | |
| 287 | |
| 41 | 288 #if ORTHANC_ENABLE_MD5 == 1 |
| 1 | 289 static char GetHexadecimalCharacter(uint8_t value) |
| 290 { | |
| 291 assert(value < 16); | |
| 292 | |
| 293 if (value < 10) | |
| 294 { | |
| 295 return value + '0'; | |
| 296 } | |
| 297 else | |
| 298 { | |
| 299 return (value - 10) + 'a'; | |
| 300 } | |
| 301 } | |
| 302 | |
| 303 | |
| 304 void Toolbox::ComputeMD5(std::string& result, | |
| 305 const std::string& data) | |
| 306 { | |
| 307 if (data.size() > 0) | |
| 308 { | |
| 309 ComputeMD5(result, &data[0], data.size()); | |
| 310 } | |
| 311 else | |
| 312 { | |
| 313 ComputeMD5(result, NULL, 0); | |
| 314 } | |
| 315 } | |
| 316 | |
| 317 | |
| 318 void Toolbox::ComputeMD5(std::string& result, | |
| 319 const void* data, | |
| 320 size_t size) | |
| 321 { | |
| 322 md5_state_s state; | |
| 323 md5_init(&state); | |
| 324 | |
| 325 if (size > 0) | |
| 326 { | |
| 327 md5_append(&state, | |
| 328 reinterpret_cast<const md5_byte_t*>(data), | |
| 329 static_cast<int>(size)); | |
| 330 } | |
| 331 | |
| 332 md5_byte_t actualHash[16]; | |
| 333 md5_finish(&state, actualHash); | |
| 334 | |
| 335 result.resize(32); | |
| 336 for (unsigned int i = 0; i < 16; i++) | |
| 337 { | |
| 338 result[2 * i] = GetHexadecimalCharacter(static_cast<uint8_t>(actualHash[i] / 16)); | |
| 339 result[2 * i + 1] = GetHexadecimalCharacter(static_cast<uint8_t>(actualHash[i] % 16)); | |
| 340 } | |
| 341 } | |
| 342 #endif | |
| 343 | |
| 344 | |
| 41 | 345 #if ORTHANC_ENABLE_BASE64 == 1 |
| 1 | 346 void Toolbox::EncodeBase64(std::string& result, |
| 347 const std::string& data) | |
| 348 { | |
| 349 result = base64_encode(data); | |
| 350 } | |
| 351 | |
| 352 void Toolbox::DecodeBase64(std::string& result, | |
| 353 const std::string& data) | |
| 354 { | |
| 355 for (size_t i = 0; i < data.length(); i++) | |
| 356 { | |
| 357 if (!isalnum(data[i]) && | |
| 358 data[i] != '+' && | |
| 359 data[i] != '/' && | |
| 360 data[i] != '=') | |
| 361 { | |
| 362 // This is not a valid character for a Base64 string | |
| 363 throw OrthancException(ErrorCode_BadFileFormat); | |
| 364 } | |
| 365 } | |
| 366 | |
| 367 result = base64_decode(data); | |
| 368 } | |
| 369 | |
| 370 | |
| 371 # if BOOST_HAS_REGEX == 1 | |
| 372 bool Toolbox::DecodeDataUriScheme(std::string& mime, | |
| 373 std::string& content, | |
| 374 const std::string& source) | |
| 375 { | |
| 376 boost::regex pattern("data:([^;]+);base64,([a-zA-Z0-9=+/]*)", | |
| 377 boost::regex::icase /* case insensitive search */); | |
| 378 | |
| 379 boost::cmatch what; | |
| 380 if (regex_match(source.c_str(), what, pattern)) | |
| 381 { | |
| 382 mime = what[1]; | |
| 383 DecodeBase64(content, what[2]); | |
| 384 return true; | |
| 385 } | |
| 386 else | |
| 387 { | |
| 388 return false; | |
| 389 } | |
| 390 } | |
| 391 # endif | |
| 392 | |
| 393 | |
| 394 void Toolbox::EncodeDataUriScheme(std::string& result, | |
| 395 const std::string& mime, | |
| 396 const std::string& content) | |
| 397 { | |
| 398 result = "data:" + mime + ";base64," + base64_encode(content); | |
| 399 } | |
| 400 | |
| 401 #endif | |
| 402 | |
| 403 | |
| 404 static const char* GetBoostLocaleEncoding(const Encoding sourceEncoding) | |
| 405 { | |
| 406 switch (sourceEncoding) | |
| 407 { | |
| 408 case Encoding_Utf8: | |
| 409 return "UTF-8"; | |
| 410 | |
| 411 case Encoding_Ascii: | |
| 412 return "ASCII"; | |
| 413 | |
| 414 case Encoding_Latin1: | |
| 415 return "ISO-8859-1"; | |
| 416 break; | |
| 417 | |
| 418 case Encoding_Latin2: | |
| 419 return "ISO-8859-2"; | |
| 420 break; | |
| 421 | |
| 422 case Encoding_Latin3: | |
| 423 return "ISO-8859-3"; | |
| 424 break; | |
| 425 | |
| 426 case Encoding_Latin4: | |
| 427 return "ISO-8859-4"; | |
| 428 break; | |
| 429 | |
| 430 case Encoding_Latin5: | |
| 431 return "ISO-8859-9"; | |
| 432 break; | |
| 433 | |
| 434 case Encoding_Cyrillic: | |
| 435 return "ISO-8859-5"; | |
| 436 break; | |
| 437 | |
| 438 case Encoding_Windows1251: | |
| 439 return "WINDOWS-1251"; | |
| 440 break; | |
| 441 | |
| 442 case Encoding_Arabic: | |
| 443 return "ISO-8859-6"; | |
| 444 break; | |
| 445 | |
| 446 case Encoding_Greek: | |
| 447 return "ISO-8859-7"; | |
| 448 break; | |
| 449 | |
| 450 case Encoding_Hebrew: | |
| 451 return "ISO-8859-8"; | |
| 452 break; | |
| 453 | |
| 454 case Encoding_Japanese: | |
| 455 return "SHIFT-JIS"; | |
| 456 break; | |
| 457 | |
| 458 case Encoding_Chinese: | |
| 459 return "GB18030"; | |
| 460 break; | |
| 461 | |
| 462 case Encoding_Thai: | |
| 463 return "TIS620.2533-0"; | |
| 464 break; | |
| 465 | |
| 466 default: | |
| 467 throw OrthancException(ErrorCode_NotImplemented); | |
| 468 } | |
| 469 } | |
| 470 | |
| 471 | |
| 472 std::string Toolbox::ConvertToUtf8(const std::string& source, | |
| 473 Encoding sourceEncoding) | |
| 474 { | |
| 475 if (sourceEncoding == Encoding_Utf8) | |
| 476 { | |
| 477 // Already in UTF-8: No conversion is required | |
| 478 return source; | |
| 479 } | |
| 480 | |
| 481 if (sourceEncoding == Encoding_Ascii) | |
| 482 { | |
| 483 return ConvertToAscii(source); | |
| 484 } | |
| 485 | |
| 486 const char* encoding = GetBoostLocaleEncoding(sourceEncoding); | |
| 487 | |
| 488 try | |
| 489 { | |
| 490 return boost::locale::conv::to_utf<char>(source, encoding); | |
| 491 } | |
| 492 catch (std::runtime_error&) | |
| 493 { | |
| 494 // Bad input string or bad encoding | |
| 495 return ConvertToAscii(source); | |
| 496 } | |
| 497 } | |
| 498 | |
| 499 | |
| 500 std::string Toolbox::ConvertFromUtf8(const std::string& source, | |
| 501 Encoding targetEncoding) | |
| 502 { | |
| 503 if (targetEncoding == Encoding_Utf8) | |
| 504 { | |
| 505 // Already in UTF-8: No conversion is required | |
| 506 return source; | |
| 507 } | |
| 508 | |
| 509 if (targetEncoding == Encoding_Ascii) | |
| 510 { | |
| 511 return ConvertToAscii(source); | |
| 512 } | |
| 513 | |
| 514 const char* encoding = GetBoostLocaleEncoding(targetEncoding); | |
| 515 | |
| 516 try | |
| 517 { | |
| 518 return boost::locale::conv::from_utf<char>(source, encoding); | |
| 519 } | |
| 520 catch (std::runtime_error&) | |
| 521 { | |
| 522 // Bad input string or bad encoding | |
| 523 return ConvertToAscii(source); | |
| 524 } | |
| 525 } | |
| 526 | |
| 527 | |
| 38 | 528 bool Toolbox::IsAsciiString(const void* data, |
| 529 size_t size) | |
| 530 { | |
| 531 const uint8_t* p = reinterpret_cast<const uint8_t*>(data); | |
| 532 | |
| 533 for (size_t i = 0; i < size; i++, p++) | |
| 534 { | |
| 535 if (*p > 127 || (*p != 0 && iscntrl(*p))) | |
| 536 { | |
| 537 return false; | |
| 538 } | |
| 539 } | |
| 540 | |
| 541 return true; | |
| 542 } | |
| 543 | |
| 544 | |
| 1 | 545 std::string Toolbox::ConvertToAscii(const std::string& source) |
| 546 { | |
| 547 std::string result; | |
| 548 | |
| 549 result.reserve(source.size() + 1); | |
| 550 for (size_t i = 0; i < source.size(); i++) | |
| 551 { | |
| 552 if (source[i] <= 127 && source[i] >= 0 && !iscntrl(source[i])) | |
| 553 { | |
| 554 result.push_back(source[i]); | |
| 555 } | |
| 556 } | |
| 557 | |
| 558 return result; | |
| 559 } | |
| 560 | |
| 561 | |
| 562 void Toolbox::ComputeSHA1(std::string& result, | |
| 563 const void* data, | |
| 564 size_t size) | |
| 565 { | |
| 566 boost::uuids::detail::sha1 sha1; | |
| 567 | |
| 568 if (size > 0) | |
| 569 { | |
| 570 sha1.process_bytes(data, size); | |
| 571 } | |
| 572 | |
| 573 unsigned int digest[5]; | |
| 574 | |
| 575 // Sanity check for the memory layout: A SHA-1 digest is 160 bits wide | |
| 576 assert(sizeof(unsigned int) == 4 && sizeof(digest) == (160 / 8)); | |
| 577 | |
| 578 sha1.get_digest(digest); | |
| 579 | |
| 580 result.resize(8 * 5 + 4); | |
| 581 sprintf(&result[0], "%08x-%08x-%08x-%08x-%08x", | |
| 582 digest[0], | |
| 583 digest[1], | |
| 584 digest[2], | |
| 585 digest[3], | |
| 586 digest[4]); | |
| 587 } | |
| 588 | |
| 589 void Toolbox::ComputeSHA1(std::string& result, | |
| 590 const std::string& data) | |
| 591 { | |
| 592 if (data.size() > 0) | |
| 593 { | |
| 594 ComputeSHA1(result, data.c_str(), data.size()); | |
| 595 } | |
| 596 else | |
| 597 { | |
| 598 ComputeSHA1(result, NULL, 0); | |
| 599 } | |
| 600 } | |
| 601 | |
| 602 | |
| 603 bool Toolbox::IsSHA1(const char* str, | |
| 604 size_t size) | |
| 605 { | |
| 606 if (size == 0) | |
| 607 { | |
| 608 return false; | |
| 609 } | |
| 610 | |
| 611 const char* start = str; | |
| 612 const char* end = str + size; | |
| 613 | |
| 614 // Trim the beginning of the string | |
| 615 while (start < end) | |
| 616 { | |
| 617 if (*start == '\0' || | |
| 618 isspace(*start)) | |
| 619 { | |
| 620 start++; | |
| 621 } | |
| 622 else | |
| 623 { | |
| 624 break; | |
| 625 } | |
| 626 } | |
| 627 | |
| 628 // Trim the trailing of the string | |
| 629 while (start < end) | |
| 630 { | |
| 631 if (*(end - 1) == '\0' || | |
| 632 isspace(*(end - 1))) | |
| 633 { | |
| 634 end--; | |
| 635 } | |
| 636 else | |
| 637 { | |
| 638 break; | |
| 639 } | |
| 640 } | |
| 641 | |
| 642 if (end - start != 44) | |
| 643 { | |
| 644 return false; | |
| 645 } | |
| 646 | |
| 647 for (unsigned int i = 0; i < 44; i++) | |
| 648 { | |
| 649 if (i == 8 || | |
| 650 i == 17 || | |
| 651 i == 26 || | |
| 652 i == 35) | |
| 653 { | |
| 654 if (start[i] != '-') | |
| 655 return false; | |
| 656 } | |
| 657 else | |
| 658 { | |
| 659 if (!isalnum(start[i])) | |
| 660 return false; | |
| 661 } | |
| 662 } | |
| 663 | |
| 664 return true; | |
| 665 } | |
| 666 | |
| 667 | |
| 668 bool Toolbox::IsSHA1(const std::string& s) | |
| 669 { | |
| 670 if (s.size() == 0) | |
| 671 { | |
| 672 return false; | |
| 673 } | |
| 674 else | |
| 675 { | |
| 676 return IsSHA1(s.c_str(), s.size()); | |
| 677 } | |
| 678 } | |
| 679 | |
| 680 | |
| 681 std::string Toolbox::StripSpaces(const std::string& source) | |
| 682 { | |
| 683 size_t first = 0; | |
| 684 | |
| 685 while (first < source.length() && | |
| 686 isspace(source[first])) | |
| 687 { | |
| 688 first++; | |
| 689 } | |
| 690 | |
| 691 if (first == source.length()) | |
| 692 { | |
| 693 // String containing only spaces | |
| 694 return ""; | |
| 695 } | |
| 696 | |
| 697 size_t last = source.length(); | |
| 698 while (last > first && | |
| 699 isspace(source[last - 1])) | |
| 700 { | |
| 701 last--; | |
| 702 } | |
| 703 | |
| 704 assert(first <= last); | |
| 705 return source.substr(first, last - first); | |
| 706 } | |
| 707 | |
| 708 | |
| 709 static char Hex2Dec(char c) | |
| 710 { | |
| 711 return ((c >= '0' && c <= '9') ? c - '0' : | |
| 712 ((c >= 'a' && c <= 'f') ? c - 'a' + 10 : c - 'A' + 10)); | |
| 713 } | |
| 714 | |
| 715 void Toolbox::UrlDecode(std::string& s) | |
| 716 { | |
| 717 // http://en.wikipedia.org/wiki/Percent-encoding | |
| 718 // http://www.w3schools.com/tags/ref_urlencode.asp | |
| 719 // http://stackoverflow.com/questions/154536/encode-decode-urls-in-c | |
| 720 | |
| 721 if (s.size() == 0) | |
| 722 { | |
| 723 return; | |
| 724 } | |
| 725 | |
| 726 size_t source = 0; | |
| 727 size_t target = 0; | |
| 728 | |
| 729 while (source < s.size()) | |
| 730 { | |
| 731 if (s[source] == '%' && | |
| 732 source + 2 < s.size() && | |
| 733 isalnum(s[source + 1]) && | |
| 734 isalnum(s[source + 2])) | |
| 735 { | |
| 736 s[target] = (Hex2Dec(s[source + 1]) << 4) | Hex2Dec(s[source + 2]); | |
| 737 source += 3; | |
| 738 target += 1; | |
| 739 } | |
| 740 else | |
| 741 { | |
| 742 if (s[source] == '+') | |
| 743 s[target] = ' '; | |
| 744 else | |
| 745 s[target] = s[source]; | |
| 746 | |
| 747 source++; | |
| 748 target++; | |
| 749 } | |
| 750 } | |
| 751 | |
| 752 s.resize(target); | |
| 753 } | |
| 754 | |
| 755 | |
| 756 Endianness Toolbox::DetectEndianness() | |
| 757 { | |
| 758 // http://sourceforge.net/p/predef/wiki/Endianness/ | |
| 759 | |
| 760 uint8_t buffer[4]; | |
| 761 | |
| 762 buffer[0] = 0x00; | |
| 763 buffer[1] = 0x01; | |
| 764 buffer[2] = 0x02; | |
| 765 buffer[3] = 0x03; | |
| 766 | |
| 767 switch (*((uint32_t *)buffer)) | |
| 768 { | |
| 769 case 0x00010203: | |
| 770 return Endianness_Big; | |
| 771 | |
| 772 case 0x03020100: | |
| 773 return Endianness_Little; | |
| 774 | |
| 775 default: | |
| 776 throw OrthancException(ErrorCode_NotImplemented); | |
| 777 } | |
| 778 } | |
| 779 | |
| 780 | |
| 781 #if BOOST_HAS_REGEX == 1 | |
| 782 std::string Toolbox::WildcardToRegularExpression(const std::string& source) | |
| 783 { | |
| 784 // TODO - Speed up this with a regular expression | |
| 785 | |
| 786 std::string result = source; | |
| 787 | |
| 788 // Escape all special characters | |
| 789 boost::replace_all(result, "\\", "\\\\"); | |
| 790 boost::replace_all(result, "^", "\\^"); | |
| 791 boost::replace_all(result, ".", "\\."); | |
| 792 boost::replace_all(result, "$", "\\$"); | |
| 793 boost::replace_all(result, "|", "\\|"); | |
| 794 boost::replace_all(result, "(", "\\("); | |
| 795 boost::replace_all(result, ")", "\\)"); | |
| 796 boost::replace_all(result, "[", "\\["); | |
| 797 boost::replace_all(result, "]", "\\]"); | |
| 798 boost::replace_all(result, "+", "\\+"); | |
| 799 boost::replace_all(result, "/", "\\/"); | |
| 800 boost::replace_all(result, "{", "\\{"); | |
| 801 boost::replace_all(result, "}", "\\}"); | |
| 802 | |
| 803 // Convert wildcards '*' and '?' to their regex equivalents | |
| 804 boost::replace_all(result, "?", "."); | |
| 805 boost::replace_all(result, "*", ".*"); | |
| 806 | |
| 807 return result; | |
| 808 } | |
| 809 #endif | |
| 810 | |
| 811 | |
| 812 | |
| 813 void Toolbox::TokenizeString(std::vector<std::string>& result, | |
| 814 const std::string& value, | |
| 815 char separator) | |
| 816 { | |
| 817 result.clear(); | |
| 818 | |
| 819 std::string currentItem; | |
| 820 | |
| 821 for (size_t i = 0; i < value.size(); i++) | |
| 822 { | |
| 823 if (value[i] == separator) | |
| 824 { | |
| 825 result.push_back(currentItem); | |
| 826 currentItem.clear(); | |
| 827 } | |
| 828 else | |
| 829 { | |
| 830 currentItem.push_back(value[i]); | |
| 831 } | |
| 832 } | |
| 833 | |
| 834 result.push_back(currentItem); | |
| 835 } | |
| 836 | |
| 837 | |
| 39 | 838 #if ORTHANC_ENABLE_PUGIXML == 1 |
| 1 | 839 class ChunkedBufferWriter : public pugi::xml_writer |
| 840 { | |
| 841 private: | |
| 842 ChunkedBuffer buffer_; | |
| 843 | |
| 844 public: | |
| 845 virtual void write(const void *data, size_t size) | |
| 846 { | |
| 847 if (size > 0) | |
| 848 { | |
| 849 buffer_.AddChunk(reinterpret_cast<const char*>(data), size); | |
| 850 } | |
| 851 } | |
| 852 | |
| 853 void Flatten(std::string& s) | |
| 854 { | |
| 855 buffer_.Flatten(s); | |
| 856 } | |
| 857 }; | |
| 858 | |
| 859 | |
| 860 static void JsonToXmlInternal(pugi::xml_node& target, | |
| 861 const Json::Value& source, | |
| 862 const std::string& arrayElement) | |
| 863 { | |
| 864 // http://jsoncpp.sourceforge.net/value_8h_source.html#l00030 | |
| 865 | |
| 866 switch (source.type()) | |
| 867 { | |
| 868 case Json::nullValue: | |
| 869 { | |
| 870 target.append_child(pugi::node_pcdata).set_value("null"); | |
| 871 break; | |
| 872 } | |
| 873 | |
| 874 case Json::intValue: | |
| 875 { | |
| 876 std::string s = boost::lexical_cast<std::string>(source.asInt()); | |
| 877 target.append_child(pugi::node_pcdata).set_value(s.c_str()); | |
| 878 break; | |
| 879 } | |
| 880 | |
| 881 case Json::uintValue: | |
| 882 { | |
| 883 std::string s = boost::lexical_cast<std::string>(source.asUInt()); | |
| 884 target.append_child(pugi::node_pcdata).set_value(s.c_str()); | |
| 885 break; | |
| 886 } | |
| 887 | |
| 888 case Json::realValue: | |
| 889 { | |
| 890 std::string s = boost::lexical_cast<std::string>(source.asFloat()); | |
| 891 target.append_child(pugi::node_pcdata).set_value(s.c_str()); | |
| 892 break; | |
| 893 } | |
| 894 | |
| 895 case Json::stringValue: | |
| 896 { | |
| 897 target.append_child(pugi::node_pcdata).set_value(source.asString().c_str()); | |
| 898 break; | |
| 899 } | |
| 900 | |
| 901 case Json::booleanValue: | |
| 902 { | |
| 903 target.append_child(pugi::node_pcdata).set_value(source.asBool() ? "true" : "false"); | |
| 904 break; | |
| 905 } | |
| 906 | |
| 907 case Json::arrayValue: | |
| 908 { | |
| 909 for (Json::Value::ArrayIndex i = 0; i < source.size(); i++) | |
| 910 { | |
| 911 pugi::xml_node node = target.append_child(); | |
| 912 node.set_name(arrayElement.c_str()); | |
| 913 JsonToXmlInternal(node, source[i], arrayElement); | |
| 914 } | |
| 915 break; | |
| 916 } | |
| 917 | |
| 918 case Json::objectValue: | |
| 919 { | |
| 920 Json::Value::Members members = source.getMemberNames(); | |
| 921 | |
| 922 for (size_t i = 0; i < members.size(); i++) | |
| 923 { | |
| 924 pugi::xml_node node = target.append_child(); | |
| 925 node.set_name(members[i].c_str()); | |
| 926 JsonToXmlInternal(node, source[members[i]], arrayElement); | |
| 927 } | |
| 928 | |
| 929 break; | |
| 930 } | |
| 931 | |
| 932 default: | |
| 933 throw OrthancException(ErrorCode_NotImplemented); | |
| 934 } | |
| 935 } | |
| 936 | |
| 937 | |
| 938 void Toolbox::JsonToXml(std::string& target, | |
| 939 const Json::Value& source, | |
| 940 const std::string& rootElement, | |
| 941 const std::string& arrayElement) | |
| 942 { | |
| 943 pugi::xml_document doc; | |
| 944 | |
| 945 pugi::xml_node n = doc.append_child(rootElement.c_str()); | |
| 946 JsonToXmlInternal(n, source, arrayElement); | |
| 947 | |
| 948 pugi::xml_node decl = doc.prepend_child(pugi::node_declaration); | |
| 949 decl.append_attribute("version").set_value("1.0"); | |
| 950 decl.append_attribute("encoding").set_value("utf-8"); | |
| 951 | |
| 952 ChunkedBufferWriter writer; | |
| 953 doc.save(writer, " ", pugi::format_default, pugi::encoding_utf8); | |
| 954 writer.Flatten(target); | |
| 955 } | |
| 956 | |
| 957 #endif | |
| 958 | |
| 959 | |
| 960 | |
| 961 bool Toolbox::IsInteger(const std::string& str) | |
| 962 { | |
| 963 std::string s = StripSpaces(str); | |
| 964 | |
| 965 if (s.size() == 0) | |
| 966 { | |
| 967 return false; | |
| 968 } | |
| 969 | |
| 970 size_t pos = 0; | |
| 971 if (s[0] == '-') | |
| 972 { | |
| 973 if (s.size() == 1) | |
| 974 { | |
| 975 return false; | |
| 976 } | |
| 977 | |
| 978 pos = 1; | |
| 979 } | |
| 980 | |
| 981 while (pos < s.size()) | |
| 982 { | |
| 983 if (!isdigit(s[pos])) | |
| 984 { | |
| 985 return false; | |
| 986 } | |
| 987 | |
| 988 pos++; | |
| 989 } | |
| 990 | |
| 991 return true; | |
| 992 } | |
| 993 | |
| 994 | |
| 995 void Toolbox::CopyJsonWithoutComments(Json::Value& target, | |
| 996 const Json::Value& source) | |
| 997 { | |
| 998 switch (source.type()) | |
| 999 { | |
| 1000 case Json::nullValue: | |
| 1001 target = Json::nullValue; | |
| 1002 break; | |
| 1003 | |
| 1004 case Json::intValue: | |
| 1005 target = source.asInt64(); | |
| 1006 break; | |
| 1007 | |
| 1008 case Json::uintValue: | |
| 1009 target = source.asUInt64(); | |
| 1010 break; | |
| 1011 | |
| 1012 case Json::realValue: | |
| 1013 target = source.asDouble(); | |
| 1014 break; | |
| 1015 | |
| 1016 case Json::stringValue: | |
| 1017 target = source.asString(); | |
| 1018 break; | |
| 1019 | |
| 1020 case Json::booleanValue: | |
| 1021 target = source.asBool(); | |
| 1022 break; | |
| 1023 | |
| 1024 case Json::arrayValue: | |
| 1025 { | |
| 1026 target = Json::arrayValue; | |
| 1027 for (Json::Value::ArrayIndex i = 0; i < source.size(); i++) | |
| 1028 { | |
| 1029 Json::Value& item = target.append(Json::nullValue); | |
| 1030 CopyJsonWithoutComments(item, source[i]); | |
| 1031 } | |
| 1032 | |
| 1033 break; | |
| 1034 } | |
| 1035 | |
| 1036 case Json::objectValue: | |
| 1037 { | |
| 1038 target = Json::objectValue; | |
| 1039 Json::Value::Members members = source.getMemberNames(); | |
| 1040 for (Json::Value::ArrayIndex i = 0; i < members.size(); i++) | |
| 1041 { | |
| 1042 const std::string item = members[i]; | |
| 1043 CopyJsonWithoutComments(target[item], source[item]); | |
| 1044 } | |
| 1045 | |
| 1046 break; | |
| 1047 } | |
| 1048 | |
| 1049 default: | |
| 1050 break; | |
| 1051 } | |
| 1052 } | |
| 1053 | |
| 1054 | |
| 1055 bool Toolbox::StartsWith(const std::string& str, | |
| 1056 const std::string& prefix) | |
| 1057 { | |
| 1058 if (str.size() < prefix.size()) | |
| 1059 { | |
| 1060 return false; | |
| 1061 } | |
| 1062 else | |
| 1063 { | |
| 1064 return str.compare(0, prefix.size(), prefix) == 0; | |
| 1065 } | |
| 1066 } | |
| 1067 | |
| 1068 | |
| 1069 static bool IsUnreservedCharacter(char c) | |
| 1070 { | |
| 1071 // This function checks whether "c" is an unserved character | |
| 1072 // wrt. an URI percent-encoding | |
| 1073 // https://en.wikipedia.org/wiki/Percent-encoding#Percent-encoding%5Fin%5Fa%5FURI | |
| 1074 | |
| 1075 return ((c >= 'A' && c <= 'Z') || | |
| 1076 (c >= 'a' && c <= 'z') || | |
| 1077 (c >= '0' && c <= '9') || | |
| 1078 c == '-' || | |
| 1079 c == '_' || | |
| 1080 c == '.' || | |
| 1081 c == '~'); | |
| 1082 } | |
| 1083 | |
| 1084 void Toolbox::UriEncode(std::string& target, | |
| 1085 const std::string& source) | |
| 1086 { | |
| 1087 // Estimate the length of the percent-encoded URI | |
| 1088 size_t length = 0; | |
| 1089 | |
| 1090 for (size_t i = 0; i < source.size(); i++) | |
| 1091 { | |
| 1092 if (IsUnreservedCharacter(source[i])) | |
| 1093 { | |
| 1094 length += 1; | |
| 1095 } | |
| 1096 else | |
| 1097 { | |
| 1098 // This character must be percent-encoded | |
| 1099 length += 3; | |
| 1100 } | |
| 1101 } | |
| 1102 | |
| 1103 target.clear(); | |
| 1104 target.reserve(length); | |
| 1105 | |
| 1106 for (size_t i = 0; i < source.size(); i++) | |
| 1107 { | |
| 1108 if (IsUnreservedCharacter(source[i])) | |
| 1109 { | |
| 1110 target.push_back(source[i]); | |
| 1111 } | |
| 1112 else | |
| 1113 { | |
| 1114 // This character must be percent-encoded | |
| 1115 uint8_t byte = static_cast<uint8_t>(source[i]); | |
| 1116 uint8_t a = byte >> 4; | |
| 1117 uint8_t b = byte & 0x0f; | |
| 1118 | |
| 1119 target.push_back('%'); | |
| 1120 target.push_back(a < 10 ? a + '0' : a - 10 + 'A'); | |
| 1121 target.push_back(b < 10 ? b + '0' : b - 10 + 'A'); | |
| 1122 } | |
| 1123 } | |
| 1124 } | |
| 1125 | |
| 1126 | |
| 1127 static bool HasField(const Json::Value& json, | |
| 1128 const std::string& key, | |
| 1129 Json::ValueType expectedType) | |
| 1130 { | |
| 1131 if (json.type() != Json::objectValue || | |
| 1132 !json.isMember(key)) | |
| 1133 { | |
| 1134 return false; | |
| 1135 } | |
| 1136 else if (json[key].type() == expectedType) | |
| 1137 { | |
| 1138 return true; | |
| 1139 } | |
| 1140 else | |
| 1141 { | |
| 1142 throw OrthancException(ErrorCode_BadParameterType); | |
| 1143 } | |
| 1144 } | |
| 1145 | |
| 1146 | |
| 1147 std::string Toolbox::GetJsonStringField(const Json::Value& json, | |
| 1148 const std::string& key, | |
| 1149 const std::string& defaultValue) | |
| 1150 { | |
| 1151 if (HasField(json, key, Json::stringValue)) | |
| 1152 { | |
| 1153 return json[key].asString(); | |
| 1154 } | |
| 1155 else | |
| 1156 { | |
| 1157 return defaultValue; | |
| 1158 } | |
| 1159 } | |
| 1160 | |
| 1161 | |
| 1162 bool Toolbox::GetJsonBooleanField(const ::Json::Value& json, | |
| 1163 const std::string& key, | |
| 1164 bool defaultValue) | |
| 1165 { | |
| 1166 if (HasField(json, key, Json::booleanValue)) | |
| 1167 { | |
| 1168 return json[key].asBool(); | |
| 1169 } | |
| 1170 else | |
| 1171 { | |
| 1172 return defaultValue; | |
| 1173 } | |
| 1174 } | |
| 1175 | |
| 1176 | |
| 1177 int Toolbox::GetJsonIntegerField(const ::Json::Value& json, | |
| 1178 const std::string& key, | |
| 1179 int defaultValue) | |
| 1180 { | |
| 1181 if (HasField(json, key, Json::intValue)) | |
| 1182 { | |
| 1183 return json[key].asInt(); | |
| 1184 } | |
| 1185 else | |
| 1186 { | |
| 1187 return defaultValue; | |
| 1188 } | |
| 1189 } | |
| 1190 | |
| 1191 | |
| 1192 unsigned int Toolbox::GetJsonUnsignedIntegerField(const ::Json::Value& json, | |
| 1193 const std::string& key, | |
| 1194 unsigned int defaultValue) | |
| 1195 { | |
| 1196 int v = GetJsonIntegerField(json, key, defaultValue); | |
| 1197 | |
| 1198 if (v < 0) | |
| 1199 { | |
| 1200 throw OrthancException(ErrorCode_ParameterOutOfRange); | |
| 1201 } | |
| 1202 else | |
| 1203 { | |
| 1204 return static_cast<unsigned int>(v); | |
| 1205 } | |
| 1206 } | |
| 43 | 1207 |
| 1208 | |
| 1209 bool Toolbox::IsUuid(const std::string& str) | |
| 1210 { | |
| 1211 if (str.size() != 36) | |
| 1212 { | |
| 1213 return false; | |
| 1214 } | |
| 1215 | |
| 1216 for (size_t i = 0; i < str.length(); i++) | |
| 1217 { | |
| 1218 if (i == 8 || i == 13 || i == 18 || i == 23) | |
| 1219 { | |
| 1220 if (str[i] != '-') | |
| 1221 return false; | |
| 1222 } | |
| 1223 else | |
| 1224 { | |
| 1225 if (!isalnum(str[i])) | |
| 1226 return false; | |
| 1227 } | |
| 1228 } | |
| 1229 | |
| 1230 return true; | |
| 1231 } | |
| 1232 | |
| 1233 | |
| 1234 bool Toolbox::StartsWithUuid(const std::string& str) | |
| 1235 { | |
| 1236 if (str.size() < 36) | |
| 1237 { | |
| 1238 return false; | |
| 1239 } | |
| 1240 | |
| 1241 if (str.size() == 36) | |
| 1242 { | |
| 1243 return IsUuid(str); | |
| 1244 } | |
| 1245 | |
| 1246 assert(str.size() > 36); | |
| 1247 if (!isspace(str[36])) | |
| 1248 { | |
| 1249 return false; | |
| 1250 } | |
| 1251 | |
| 1252 return IsUuid(str.substr(0, 36)); | |
| 1253 } | |
| 1 | 1254 } |