Mercurial > hg > orthanc-stone
annotate Framework/Messaging/MessagingToolbox.cpp @ 22:b01d46e5a2b3
support for sandboxed environments
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Wed, 23 Nov 2016 12:08:15 +0100 |
| parents | ff1e935768e7 |
| children | 9aace933cb64 |
| rev | line source |
|---|---|
| 0 | 1 /** |
| 2 * Stone of Orthanc | |
| 3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
| 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 "MessagingToolbox.h" | |
| 34 | |
| 16 | 35 #include "../../Resources/Orthanc/Core/Images/Image.h" |
| 36 #include "../../Resources/Orthanc/Core/Images/ImageProcessing.h" | |
| 37 #include "../../Resources/Orthanc/Core/Images/JpegReader.h" | |
| 38 #include "../../Resources/Orthanc/Core/Images/PngReader.h" | |
| 39 #include "../../Resources/Orthanc/Core/OrthancException.h" | |
| 40 #include "../../Resources/Orthanc/Core/Toolbox.h" | |
| 41 #include "../../Resources/Orthanc/Core/Logging.h" | |
| 0 | 42 |
| 43 #include <boost/lexical_cast.hpp> | |
| 44 #include <json/reader.h> | |
| 45 | |
| 46 #if defined(__native_client__) | |
| 47 # include <boost/math/special_functions/round.hpp> | |
| 48 #else | |
|
22
b01d46e5a2b3
support for sandboxed environments
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
16
diff
changeset
|
49 # include <boost/date_time/posix_time/posix_time.hpp> |
| 0 | 50 # include <boost/date_time/microsec_time_clock.hpp> |
| 51 #endif | |
| 52 | |
| 53 namespace OrthancStone | |
| 54 { | |
| 55 namespace MessagingToolbox | |
| 56 { | |
| 57 #if defined(__native_client__) | |
| 58 static pp::Core* core_ = NULL; | |
| 59 | |
| 60 void Timestamp::Initialize(pp::Core* core) | |
| 61 { | |
| 62 if (core == NULL) | |
| 63 { | |
| 64 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
| 65 } | |
| 66 | |
| 67 core_ = core; | |
| 68 } | |
| 69 | |
| 70 Timestamp::Timestamp() | |
| 71 { | |
| 72 if (core_ == NULL) | |
| 73 { | |
| 74 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
| 75 } | |
| 76 | |
| 77 time_ = core_->GetTimeTicks(); | |
| 78 } | |
| 79 | |
| 80 int Timestamp::GetMillisecondsSince(const Timestamp& other) | |
| 81 { | |
| 82 double difference = time_ - other.time_; | |
| 83 return static_cast<int>(boost::math::iround(difference * 1000.0)); | |
| 84 } | |
| 85 #else | |
| 86 Timestamp::Timestamp() | |
| 87 { | |
| 88 time_ = boost::posix_time::microsec_clock::local_time(); | |
| 89 } | |
| 90 | |
| 91 int Timestamp::GetMillisecondsSince(const Timestamp& other) | |
| 92 { | |
| 93 boost::posix_time::time_duration difference = time_ - other.time_; | |
| 94 return static_cast<int>(difference.total_milliseconds()); | |
| 95 } | |
| 96 #endif | |
| 97 | |
| 98 static bool ParseVersion(std::string& version, | |
| 99 unsigned int& major, | |
| 100 unsigned int& minor, | |
| 101 unsigned int& patch, | |
| 102 const Json::Value& info) | |
| 103 { | |
| 104 if (info.type() != Json::objectValue || | |
| 105 !info.isMember("Version") || | |
| 106 info["Version"].type() != Json::stringValue) | |
| 107 { | |
| 108 return false; | |
| 109 } | |
| 110 | |
| 111 version = info["Version"].asString(); | |
| 112 if (version == "mainline") | |
| 113 { | |
| 114 // Some arbitrary high values Orthanc versions will never reach ;) | |
| 115 major = 999; | |
| 116 minor = 999; | |
| 117 patch = 999; | |
| 118 return true; | |
| 119 } | |
| 120 | |
| 121 std::vector<std::string> tokens; | |
| 122 Orthanc::Toolbox::TokenizeString(tokens, version, '.'); | |
| 123 | |
| 124 if (tokens.size() != 2 && | |
| 125 tokens.size() != 3) | |
| 126 { | |
| 127 return false; | |
| 128 } | |
| 129 | |
| 130 int a, b, c; | |
| 131 try | |
| 132 { | |
| 133 a = boost::lexical_cast<int>(tokens[0]); | |
| 134 b = boost::lexical_cast<int>(tokens[1]); | |
| 135 | |
| 136 if (tokens.size() == 3) | |
| 137 { | |
| 138 c = boost::lexical_cast<int>(tokens[2]); | |
| 139 } | |
| 140 else | |
| 141 { | |
| 142 c = 0; | |
| 143 } | |
| 144 } | |
| 145 catch (boost::bad_lexical_cast&) | |
| 146 { | |
| 147 return false; | |
| 148 } | |
| 149 | |
| 150 if (a < 0 || | |
| 151 b < 0 || | |
| 152 c < 0) | |
| 153 { | |
| 154 return false; | |
| 155 } | |
| 156 else | |
| 157 { | |
| 158 major = static_cast<unsigned int>(a); | |
| 159 minor = static_cast<unsigned int>(b); | |
| 160 patch = static_cast<unsigned int>(c); | |
| 161 return true; | |
| 162 } | |
| 163 } | |
| 164 | |
| 165 void ParseJson(Json::Value& target, | |
| 166 const std::string& source) | |
| 167 { | |
| 168 Json::Reader reader; | |
| 169 if (!reader.parse(source, target)) | |
| 170 { | |
| 171 throw Orthanc::OrthancException(Orthanc::ErrorCode_BadFileFormat); | |
| 172 } | |
| 173 } | |
| 174 | |
| 175 void RestApiGet(Json::Value& target, | |
| 176 IOrthancConnection& orthanc, | |
| 177 const std::string& uri) | |
| 178 { | |
| 179 std::string tmp; | |
| 180 orthanc.RestApiGet(tmp, uri); | |
| 181 ParseJson(target, tmp); | |
| 182 } | |
| 183 | |
| 184 | |
| 185 bool HasWebViewerInstalled(IOrthancConnection& orthanc) | |
| 186 { | |
| 187 try | |
| 188 { | |
| 189 Json::Value json; | |
| 190 RestApiGet(json, orthanc, "/plugins/web-viewer"); | |
| 191 return json.type() == Json::objectValue; | |
| 192 } | |
| 193 catch (Orthanc::OrthancException&) | |
| 194 { | |
| 195 return false; | |
| 196 } | |
| 197 } | |
| 198 | |
| 199 | |
| 200 bool CheckOrthancVersion(IOrthancConnection& orthanc) | |
| 201 { | |
| 202 Json::Value json; | |
| 203 std::string version; | |
| 204 unsigned int major, minor, patch; | |
| 205 | |
| 206 try | |
| 207 { | |
| 208 RestApiGet(json, orthanc, "/system"); | |
| 209 } | |
| 210 catch (Orthanc::OrthancException&) | |
| 211 { | |
| 212 LOG(ERROR) << "Cannot connect to your Orthanc server"; | |
| 213 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 214 } | |
| 215 | |
| 216 if (!ParseVersion(version, major, minor, patch, json)) | |
| 217 { | |
| 218 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 219 } | |
| 220 | |
| 221 LOG(WARNING) << "Version of the Orthanc core (must be above 1.1.0): " << version; | |
| 222 | |
| 223 // Stone is only compatible with Orthanc >= 1.1.0, otherwise deadlocks might occur | |
| 224 if (major < 1 || | |
| 225 (major == 1 && minor < 1)) | |
| 226 { | |
| 227 return false; | |
| 228 } | |
| 229 | |
| 230 try | |
| 231 { | |
| 232 RestApiGet(json, orthanc, "/plugins/web-viewer"); | |
| 233 } | |
| 234 catch (Orthanc::OrthancException&) | |
| 235 { | |
| 236 // The Web viewer is not installed, this is OK | |
| 237 LOG(WARNING) << "The Web viewer plugin is not installed, progressive download is disabled"; | |
| 238 return true; | |
| 239 } | |
| 240 | |
| 241 if (!ParseVersion(version, major, minor, patch, json)) | |
| 242 { | |
| 243 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 244 } | |
| 245 | |
| 246 LOG(WARNING) << "Version of the Web viewer plugin (must be above 2.2): " << version; | |
| 247 | |
| 248 return (major >= 3 || | |
| 249 (major == 2 && minor >= 2)); | |
| 250 } | |
| 251 | |
| 252 | |
| 253 Orthanc::ImageAccessor* DecodeFrame(IOrthancConnection& orthanc, | |
| 254 const std::string& instance, | |
| 255 unsigned int frame, | |
| 256 Orthanc::PixelFormat targetFormat) | |
| 257 { | |
| 258 std::string uri = ("instances/" + instance + "/frames/" + | |
| 259 boost::lexical_cast<std::string>(frame)); | |
| 260 | |
| 261 std::string compressed; | |
| 262 | |
| 263 switch (targetFormat) | |
| 264 { | |
| 265 case Orthanc::PixelFormat_RGB24: | |
| 266 orthanc.RestApiGet(compressed, uri + "/preview"); | |
| 267 break; | |
| 268 | |
| 269 case Orthanc::PixelFormat_Grayscale16: | |
| 270 orthanc.RestApiGet(compressed, uri + "/image-uint16"); | |
| 271 break; | |
| 272 | |
| 273 case Orthanc::PixelFormat_SignedGrayscale16: | |
| 274 orthanc.RestApiGet(compressed, uri + "/image-int16"); | |
| 275 break; | |
| 276 | |
| 277 default: | |
| 278 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
| 279 } | |
| 280 | |
| 281 std::auto_ptr<Orthanc::PngReader> result(new Orthanc::PngReader); | |
| 282 result->ReadFromMemory(compressed); | |
| 283 | |
| 284 if (targetFormat == Orthanc::PixelFormat_SignedGrayscale16) | |
| 285 { | |
| 286 if (result->GetFormat() == Orthanc::PixelFormat_Grayscale16) | |
| 287 { | |
| 288 result->SetFormat(Orthanc::PixelFormat_SignedGrayscale16); | |
| 289 } | |
| 290 else | |
| 291 { | |
| 292 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 293 } | |
| 294 } | |
| 295 | |
| 296 return result.release(); | |
| 297 } | |
| 298 | |
| 299 | |
| 300 Orthanc::ImageAccessor* DecodeJpegFrame(IOrthancConnection& orthanc, | |
| 301 const std::string& instance, | |
| 302 unsigned int frame, | |
| 303 unsigned int quality, | |
| 304 Orthanc::PixelFormat targetFormat) | |
| 305 { | |
| 306 if (quality <= 0 || | |
| 307 quality > 100) | |
| 308 { | |
| 309 throw Orthanc::OrthancException(Orthanc::ErrorCode_ParameterOutOfRange); | |
| 310 } | |
| 311 | |
| 312 // This requires the official Web viewer plugin to be installed! | |
| 313 std::string uri = ("web-viewer/instances/jpeg" + | |
| 314 boost::lexical_cast<std::string>(quality) + | |
| 315 "-" + instance + "_" + | |
| 316 boost::lexical_cast<std::string>(frame)); | |
| 317 | |
| 318 Json::Value encoded; | |
| 319 RestApiGet(encoded, orthanc, uri); | |
| 320 | |
| 321 if (encoded.type() != Json::objectValue || | |
| 322 !encoded.isMember("Orthanc") || | |
| 323 encoded["Orthanc"].type() != Json::objectValue) | |
| 324 { | |
| 325 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 326 } | |
| 327 | |
| 328 Json::Value& info = encoded["Orthanc"]; | |
| 329 if (!info.isMember("PixelData") || | |
| 330 !info.isMember("Stretched") || | |
| 331 !info.isMember("Compression") || | |
| 332 info["Compression"].type() != Json::stringValue || | |
| 333 info["PixelData"].type() != Json::stringValue || | |
| 334 info["Stretched"].type() != Json::booleanValue || | |
| 335 info["Compression"].asString() != "Jpeg") | |
| 336 { | |
| 337 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 338 } | |
| 339 | |
| 340 bool isSigned = false; | |
| 341 bool isStretched = info["Stretched"].asBool(); | |
| 342 | |
| 343 if (info.isMember("IsSigned")) | |
| 344 { | |
| 345 if (info["IsSigned"].type() != Json::booleanValue) | |
| 346 { | |
| 347 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 348 } | |
| 349 else | |
| 350 { | |
| 351 isSigned = info["IsSigned"].asBool(); | |
| 352 } | |
| 353 } | |
| 354 | |
| 355 std::string jpeg; | |
| 356 Orthanc::Toolbox::DecodeBase64(jpeg, info["PixelData"].asString()); | |
| 357 | |
| 358 std::auto_ptr<Orthanc::JpegReader> reader(new Orthanc::JpegReader); | |
| 359 reader->ReadFromMemory(jpeg); | |
| 360 | |
| 361 if (reader->GetFormat() == Orthanc::PixelFormat_RGB24) // This is a color image | |
| 362 { | |
| 363 if (targetFormat != Orthanc::PixelFormat_RGB24) | |
| 364 { | |
| 365 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 366 } | |
| 367 | |
| 368 if (isSigned || isStretched) | |
| 369 { | |
| 370 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
| 371 } | |
| 372 else | |
| 373 { | |
| 374 return reader.release(); | |
| 375 } | |
| 376 } | |
| 377 | |
| 378 if (reader->GetFormat() != Orthanc::PixelFormat_Grayscale8) | |
| 379 { | |
| 380 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
| 381 } | |
| 382 | |
| 383 if (!isStretched) | |
| 384 { | |
| 385 if (targetFormat != reader->GetFormat()) | |
| 386 { | |
| 387 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 388 } | |
| 389 | |
| 390 return reader.release(); | |
| 391 } | |
| 392 | |
| 393 int32_t stretchLow = 0; | |
| 394 int32_t stretchHigh = 0; | |
| 395 | |
| 396 if (!info.isMember("StretchLow") || | |
| 397 !info.isMember("StretchHigh") || | |
| 398 info["StretchLow"].type() != Json::intValue || | |
| 399 info["StretchHigh"].type() != Json::intValue) | |
| 400 { | |
| 401 throw Orthanc::OrthancException(Orthanc::ErrorCode_NetworkProtocol); | |
| 402 } | |
| 403 | |
| 404 stretchLow = info["StretchLow"].asInt(); | |
| 405 stretchHigh = info["StretchHigh"].asInt(); | |
| 406 | |
| 407 if (stretchLow < -32768 || | |
| 408 stretchHigh > 65535 || | |
| 409 (stretchLow < 0 && stretchHigh > 32767)) | |
| 410 { | |
| 411 // This range cannot be represented with a uint16_t or an int16_t | |
| 412 throw Orthanc::OrthancException(Orthanc::ErrorCode_NotImplemented); | |
| 413 } | |
| 414 | |
| 415 // Decode a grayscale JPEG 8bpp image coming from the Web viewer | |
| 416 std::auto_ptr<Orthanc::ImageAccessor> image | |
| 11 | 417 (new Orthanc::Image(targetFormat, reader->GetWidth(), reader->GetHeight(), false)); |
| 0 | 418 |
| 419 float scaling = static_cast<float>(stretchHigh - stretchLow) / 255.0f; | |
| 420 float offset = static_cast<float>(stretchLow) / scaling; | |
| 421 | |
| 422 Orthanc::ImageProcessing::Convert(*image, *reader); | |
| 423 Orthanc::ImageProcessing::ShiftScale(*image, offset, scaling); | |
| 424 | |
| 425 #if 0 | |
| 426 /*info.removeMember("PixelData"); | |
| 427 std::cout << info.toStyledString();*/ | |
| 428 | |
| 429 int64_t a, b; | |
| 430 Orthanc::ImageProcessing::GetMinMaxValue(a, b, *image); | |
| 431 std::cout << stretchLow << "->" << stretchHigh << " = " << a << "->" << b << std::endl; | |
| 432 #endif | |
| 433 | |
| 434 return image.release(); | |
| 435 } | |
| 436 } | |
| 437 } |