Mercurial > hg > orthanc-authorization
annotate Plugin/ResourceHierarchyCache.cpp @ 76:d301047ee3c4
fix unchecked resource
| author | Alain Mazy <am@osimis.io> |
|---|---|
| date | Thu, 09 Mar 2023 14:37:52 +0100 |
| parents | 1a13c4fbc9a1 |
| children | 94a9484d7f8f |
| rev | line source |
|---|---|
| 1 | 1 /** |
| 2 * Advanced authorization plugin for Orthanc | |
| 68 | 3 * Copyright (C) 2017-2023 Osimis S.A., Belgium |
| 1 | 4 * |
| 5 * This program is free software: you can redistribute it and/or | |
| 6 * modify it under the terms of the GNU Affero General Public License | |
| 7 * as published by the Free Software Foundation, either version 3 of | |
| 8 * the License, or (at your option) any later version. | |
| 9 * | |
| 10 * This program is distributed in the hope that it will be useful, but | |
| 11 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 13 * Affero General Public License for more details. | |
| 14 * | |
| 15 * You should have received a copy of the GNU Affero General Public License | |
| 16 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 17 **/ | |
| 18 | |
| 19 #include "ResourceHierarchyCache.h" | |
| 20 | |
| 32 | 21 #include <Logging.h> |
| 22 #include <OrthancException.h> | |
| 1 | 23 |
| 24 #include <boost/lexical_cast.hpp> | |
| 25 | |
| 26 namespace OrthancPlugins | |
| 27 { | |
| 28 std::string ResourceHierarchyCache::ComputeKey(Orthanc::ResourceType level, | |
| 43 | 29 const std::string& identifier) const |
| 1 | 30 { |
| 31 return boost::lexical_cast<std::string>(level) + "|" + identifier; | |
| 32 } | |
| 33 | |
| 34 | |
| 35 void ResourceHierarchyCache::LinkParent(const OrthancResource& child, | |
| 36 const OrthancResource& parent) | |
| 37 { | |
| 38 LOG(INFO) << "Linking " << Orthanc::EnumerationToString(child.GetLevel()) | |
| 39 << " \"" << child.GetIdentifier() << "\" to its parent " | |
| 40 << Orthanc::EnumerationToString(parent.GetLevel()) | |
| 41 << " \"" << parent.GetIdentifier() << "\""; | |
| 42 | |
| 43 cache_->Store(ComputeKey(child), parent.GetIdentifier(), 0 /* no expiration */); | |
| 44 } | |
| 45 | |
| 46 | |
| 47 bool ResourceHierarchyCache::LookupParent(std::string& target, | |
| 48 const OrthancResource& resource) | |
| 49 { | |
| 50 std::string key = ComputeKey(resource); | |
| 51 | |
| 52 if (cache_->Retrieve(target, key)) | |
| 53 { | |
| 54 // The parent was already stored in the cache | |
| 55 return true; | |
| 56 } | |
| 57 | |
| 58 // Not in the cache, reading the resource from the Orthanc store | |
| 59 std::string dicomUid; | |
| 60 OrthancResource parent; | |
| 61 std::list<OrthancResource> children; | |
| 62 | |
|
29
bc0431cb6b8f
fix for compatibility with simplified OrthancPluginCppWrapper
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
63 if (!resource.GetHierarchy(dicomUid, parent, children)) |
| 1 | 64 { |
| 65 // The resource is non-existing (*) | |
| 66 return false; | |
| 67 } | |
| 68 | |
| 69 orthancToDicom_->Store(key, dicomUid, 0 /* no expiration */); | |
| 70 dicomToOrthanc_->Store(ComputeKey(resource.GetLevel(), dicomUid), | |
| 71 resource.GetIdentifier(), 0 /* no expiration */); | |
| 72 | |
| 73 for (std::list<OrthancResource>::const_iterator | |
| 74 it = children.begin(); it != children.end(); ++it) | |
| 75 { | |
| 76 // Cache the relation of the resource with its children | |
| 77 LinkParent(*it, resource); | |
| 78 } | |
| 79 | |
| 80 if (parent.IsValid()) | |
| 81 { | |
| 82 LinkParent(resource, parent); | |
| 83 target = parent.GetIdentifier(); | |
| 84 return true; | |
| 85 } | |
| 86 else | |
| 87 { | |
| 88 // We reached the patient level, or the resource was removed | |
| 89 // from Orthanc since (*) | |
| 90 return false; | |
| 91 } | |
| 92 } | |
| 93 | |
| 94 | |
|
29
bc0431cb6b8f
fix for compatibility with simplified OrthancPluginCppWrapper
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
95 ResourceHierarchyCache::ResourceHierarchyCache(ICacheFactory& factory) : |
| 1 | 96 cache_(factory.Create()), |
| 97 orthancToDicom_(factory.Create()), | |
| 98 dicomToOrthanc_(factory.Create()) | |
| 99 { | |
| 100 if (cache_.get() == NULL) | |
| 101 { | |
| 102 throw Orthanc::OrthancException(Orthanc::ErrorCode_InternalError); | |
| 103 } | |
| 104 } | |
| 105 | |
| 106 | |
| 107 void ResourceHierarchyCache::Invalidate(Orthanc::ResourceType level, | |
| 108 const std::string& identifier) | |
| 109 { | |
| 110 LOG(INFO) << "Invalidating " << Orthanc::EnumerationToString(level) | |
| 111 << " resource with ID: " << identifier; | |
| 112 | |
| 113 std::string key = ComputeKey(level, identifier); | |
| 114 cache_->Invalidate(key); | |
| 115 orthancToDicom_->Invalidate(key); | |
| 116 } | |
| 117 | |
| 118 | |
| 119 bool ResourceHierarchyCache::LookupSeries(std::string& patient, | |
| 120 std::string& study, | |
| 121 const std::string& series) | |
| 122 { | |
| 123 if (LookupParent(study, Orthanc::ResourceType_Series, series)) | |
| 124 { | |
| 125 return LookupStudy(patient, study); | |
| 126 } | |
| 127 else | |
| 128 { | |
| 129 return false; | |
| 130 } | |
| 131 } | |
| 132 | |
| 133 | |
| 134 bool ResourceHierarchyCache::LookupInstance(std::string& patient, | |
| 135 std::string& study, | |
| 136 std::string& series, | |
| 137 const std::string& instance) | |
| 138 { | |
| 139 if (LookupParent(series, Orthanc::ResourceType_Instance, instance)) | |
| 140 { | |
| 141 return LookupSeries(patient, study, series); | |
| 142 } | |
| 143 else | |
| 144 { | |
| 145 return false; | |
| 146 } | |
| 147 } | |
| 148 | |
| 149 | |
| 150 bool ResourceHierarchyCache::LookupDicomUid(std::string& target, | |
| 151 Orthanc::ResourceType level, | |
| 152 const std::string& orthancId) | |
| 153 { | |
| 154 std::string key = ComputeKey(level, orthancId); | |
| 155 | |
| 156 if (orthancToDicom_->Retrieve(target, key)) | |
| 157 { | |
| 158 return true; | |
| 159 } | |
| 160 | |
| 161 OrthancResource resource(level, orthancId); | |
| 162 | |
|
29
bc0431cb6b8f
fix for compatibility with simplified OrthancPluginCppWrapper
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
163 if (resource.GetDicomUid(target)) |
| 1 | 164 { |
| 165 orthancToDicom_->Store(key, target, 0 /* no expiration */); | |
| 166 return true; | |
| 167 } | |
| 168 else | |
| 169 { | |
| 170 return false; | |
| 171 } | |
| 172 } | |
| 173 | |
| 174 | |
| 175 bool ResourceHierarchyCache::LookupOrthancId(std::string& target, | |
| 176 Orthanc::ResourceType level, | |
| 177 const std::string& dicomUid) | |
| 178 { | |
| 179 std::string key = ComputeKey(level, dicomUid); | |
| 180 | |
| 181 if (dicomToOrthanc_->Retrieve(target, key)) | |
| 182 { | |
| 183 return true; | |
| 184 } | |
| 185 | |
|
29
bc0431cb6b8f
fix for compatibility with simplified OrthancPluginCppWrapper
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
22
diff
changeset
|
186 if (OrthancResource::LookupOrthancId(target, level, dicomUid)) |
| 1 | 187 { |
| 188 dicomToOrthanc_->Store(key, target, 0 /* no expiration */); | |
| 189 return true; | |
| 190 } | |
| 191 else | |
| 192 { | |
| 193 return false; | |
| 194 } | |
| 195 } | |
| 196 } |