annotate Orthanc/Core/SystemToolbox.cpp @ 179:8fe42ddf0807

sync
author Sebastien Jodogne <s.jodogne@gmail.com>
date Thu, 04 Jan 2018 09:50:04 +0100
parents dd1ad819ca33
children ab66a6baf2ae
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
1 /**
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
2 * Orthanc - A Lightweight, RESTful DICOM Store
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
4 * Department, University Hospital of Liege, Belgium
175
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 167
diff changeset
5 * Copyright (C) 2017-2018 Osimis S.A., Belgium
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
6 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
7 * This program is free software: you can redistribute it and/or
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
8 * modify it under the terms of the GNU General Public License as
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
9 * published by the Free Software Foundation, either version 3 of the
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
10 * License, or (at your option) any later version.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
11 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
12 * In addition, as a special exception, the copyright holders of this
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
13 * program give permission to link the code of its release with the
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
15 * that use the same license as the "OpenSSL" library), and distribute
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
16 * the linked executables. You must obey the GNU General Public License
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
17 * in all respects for all of the code used other than "OpenSSL". If you
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
18 * modify file(s) with this exception, you may extend this exception to
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
19 * your version of the file(s), but you are not obligated to do so. If
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
20 * you do not wish to do so, delete this exception statement from your
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
21 * version. If you delete this exception statement from all source files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
22 * in the program, then also delete it here.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
23 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
24 * This program is distributed in the hope that it will be useful, but
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
25 * WITHOUT ANY WARRANTY; without even the implied warranty of
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
27 * General Public License for more details.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
28 *
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
29 * You should have received a copy of the GNU General Public License
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
30 * along with this program. If not, see <http://www.gnu.org/licenses/>.
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
31 **/
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
32
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
33
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
34 #include "PrecompiledHeaders.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
35 #include "SystemToolbox.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
36
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
37
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
38 #if defined(_WIN32)
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
39 # include <windows.h>
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
40 # include <process.h> // For "_spawnvp()" and "_getpid()"
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
41 #else
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
42 # include <unistd.h> // For "execvp()"
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
43 # include <sys/wait.h> // For "waitpid()"
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
44 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
45
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
46
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
47 #if defined(__APPLE__) && defined(__MACH__)
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
48 # include <mach-o/dyld.h> /* _NSGetExecutablePath */
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
49 # include <limits.h> /* PATH_MAX */
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
50 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
51
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
52
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
53 #if defined(__linux__) || defined(__FreeBSD_kernel__) || defined(__FreeBSD__)
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
54 # include <limits.h> /* PATH_MAX */
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
55 # include <signal.h>
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
56 # include <unistd.h>
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
57 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
58
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
59
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
60 #if defined(__OpenBSD__)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
61 # include <sys/sysctl.h> // For "sysctl", "CTL_KERN" and "KERN_PROC_ARGS"
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
62 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
63
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
64
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
65 // Inclusions for UUID
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
66 // http://stackoverflow.com/a/1626302
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
67
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
68 extern "C"
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
69 {
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
70 #if defined(_WIN32)
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
71 # include <rpc.h>
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
72 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
73 # include <uuid/uuid.h>
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
74 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
75 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
76
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
77
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
78 #include "Logging.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
79 #include "OrthancException.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
80 #include "Toolbox.h"
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
81
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
82 #include <boost/filesystem.hpp>
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
83 #include <boost/filesystem/fstream.hpp>
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
84 #include <boost/date_time/posix_time/posix_time.hpp>
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
85
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
86
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
87 namespace Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
88 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
89 static bool finish_;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
90 static ServerBarrierEvent barrierEvent_;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
91
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
92 #if defined(_WIN32)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
93 static BOOL WINAPI ConsoleControlHandler(DWORD dwCtrlType)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
94 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
95 // http://msdn.microsoft.com/en-us/library/ms683242(v=vs.85).aspx
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
96 finish_ = true;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
97 return true;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
98 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
99 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
100 static void SignalHandler(int signal)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
101 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
102 if (signal == SIGHUP)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
103 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
104 barrierEvent_ = ServerBarrierEvent_Reload;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
105 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
106
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
107 finish_ = true;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
108 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
109 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
110
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
111
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
112 static ServerBarrierEvent ServerBarrierInternal(const bool* stopFlag)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
113 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
114 #if defined(_WIN32)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
115 SetConsoleCtrlHandler(ConsoleControlHandler, true);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
116 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
117 signal(SIGINT, SignalHandler);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
118 signal(SIGQUIT, SignalHandler);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
119 signal(SIGTERM, SignalHandler);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
120 signal(SIGHUP, SignalHandler);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
121 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
122
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
123 // Active loop that awakens every 100ms
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
124 finish_ = false;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
125 barrierEvent_ = ServerBarrierEvent_Stop;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
126 while (!(*stopFlag || finish_))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
127 {
159
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
128 SystemToolbox::USleep(100 * 1000);
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
129 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
130
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
131 #if defined(_WIN32)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
132 SetConsoleCtrlHandler(ConsoleControlHandler, false);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
133 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
134 signal(SIGINT, NULL);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
135 signal(SIGQUIT, NULL);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
136 signal(SIGTERM, NULL);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
137 signal(SIGHUP, NULL);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
138 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
139
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
140 return barrierEvent_;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
141 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
142
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
143
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
144 ServerBarrierEvent SystemToolbox::ServerBarrier(const bool& stopFlag)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
145 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
146 return ServerBarrierInternal(&stopFlag);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
147 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
148
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
149
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
150 ServerBarrierEvent SystemToolbox::ServerBarrier()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
151 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
152 const bool stopFlag = false;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
153 return ServerBarrierInternal(&stopFlag);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
154 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
156
159
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
157 void SystemToolbox::USleep(uint64_t microSeconds)
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
158 {
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
159 #if defined(_WIN32)
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
160 ::Sleep(static_cast<DWORD>(microSeconds / static_cast<uint64_t>(1000)));
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
161 #elif defined(__linux__) || defined(__APPLE__) || defined(__FreeBSD_kernel__) || defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__native_client__)
159
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
162 usleep(microSeconds);
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
163 #else
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
164 #error Support your platform here
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
165 #endif
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
166 }
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
167
5dc54316d68b shared copyright with osimis
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 155
diff changeset
168
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
169 static std::streamsize GetStreamSize(std::istream& f)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
170 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
171 // http://www.cplusplus.com/reference/iostream/istream/tellg/
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
172 f.seekg(0, std::ios::end);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
173 std::streamsize size = f.tellg();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
174 f.seekg(0, std::ios::beg);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
175
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
176 return size;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
177 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
178
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
179
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
180 void SystemToolbox::ReadFile(std::string& content,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
181 const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
182 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
183 if (!IsRegularFile(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
184 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
185 LOG(ERROR) << std::string("The path does not point to a regular file: ") << path;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
186 throw OrthancException(ErrorCode_RegularFileExpected);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
187 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
188
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
189 boost::filesystem::ifstream f;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
190 f.open(path, std::ifstream::in | std::ifstream::binary);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
191 if (!f.good())
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
192 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
193 throw OrthancException(ErrorCode_InexistentFile);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
194 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
195
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
196 std::streamsize size = GetStreamSize(f);
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
197 content.resize(static_cast<size_t>(size));
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
198 if (size != 0)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
199 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
200 f.read(reinterpret_cast<char*>(&content[0]), size);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
201 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
202
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
203 f.close();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
204 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
205
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
206
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
207 bool SystemToolbox::ReadHeader(std::string& header,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
208 const std::string& path,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
209 size_t headerSize)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
210 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
211 if (!IsRegularFile(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
212 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
213 LOG(ERROR) << std::string("The path does not point to a regular file: ") << path;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
214 throw OrthancException(ErrorCode_RegularFileExpected);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
215 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
216
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
217 boost::filesystem::ifstream f;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
218 f.open(path, std::ifstream::in | std::ifstream::binary);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
219 if (!f.good())
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
220 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
221 throw OrthancException(ErrorCode_InexistentFile);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
222 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
223
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
224 bool full = true;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
225
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
226 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
227 std::streamsize size = GetStreamSize(f);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
228 if (size <= 0)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
229 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
230 headerSize = 0;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
231 full = false;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
232 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
233 else if (static_cast<size_t>(size) < headerSize)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
234 {
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
235 headerSize = static_cast<size_t>(size); // Truncate to the size of the file
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
236 full = false;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
237 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
238 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
239
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
240 header.resize(headerSize);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
241 if (headerSize != 0)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
242 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
243 f.read(reinterpret_cast<char*>(&header[0]), headerSize);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
244 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
245
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
246 f.close();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
247
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
248 return full;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
249 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
250
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
251
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
252 void SystemToolbox::WriteFile(const void* content,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
253 size_t size,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
254 const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
255 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
256 boost::filesystem::ofstream f;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
257 f.open(path, std::ofstream::out | std::ofstream::binary);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
258 if (!f.good())
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
259 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
260 throw OrthancException(ErrorCode_CannotWriteFile);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
261 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
262
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
263 if (size != 0)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
264 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
265 f.write(reinterpret_cast<const char*>(content), size);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
266
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
267 if (!f.good())
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
268 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
269 f.close();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
270 throw OrthancException(ErrorCode_FileStorageCannotWrite);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
271 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
272 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
273
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
274 f.close();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
275 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
276
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
277
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
278 void SystemToolbox::WriteFile(const std::string& content,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
279 const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
280 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
281 WriteFile(content.size() > 0 ? content.c_str() : NULL,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
282 content.size(), path);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
283 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
284
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
285
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
286 void SystemToolbox::RemoveFile(const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
287 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
288 if (boost::filesystem::exists(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
289 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
290 if (IsRegularFile(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
291 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
292 boost::filesystem::remove(path);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
293 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
294 else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
295 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
296 throw OrthancException(ErrorCode_RegularFileExpected);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
297 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
298 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
299 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
300
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
301
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
302 uint64_t SystemToolbox::GetFileSize(const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
303 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
304 try
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
305 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
306 return static_cast<uint64_t>(boost::filesystem::file_size(path));
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
307 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
308 catch (boost::filesystem::filesystem_error&)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
309 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
310 throw OrthancException(ErrorCode_InexistentFile);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
311 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
312 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
313
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
314
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
315 void SystemToolbox::MakeDirectory(const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
316 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
317 if (boost::filesystem::exists(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
318 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
319 if (!boost::filesystem::is_directory(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
320 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
321 throw OrthancException(ErrorCode_DirectoryOverFile);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
322 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
323 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
324 else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
325 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
326 if (!boost::filesystem::create_directories(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
327 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
328 throw OrthancException(ErrorCode_MakeDirectory);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
329 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
330 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
331 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
332
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
333
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
334 bool SystemToolbox::IsExistingFile(const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
335 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
336 return boost::filesystem::exists(path);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
337 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
338
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
339
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
340 #if defined(_WIN32)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
341 static std::string GetPathToExecutableInternal()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
342 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
343 // Yes, this is ugly, but there is no simple way to get the
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
344 // required buffer size, so we use a big constant
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
345 std::vector<char> buffer(32768);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
346 /*int bytes =*/ GetModuleFileNameA(NULL, &buffer[0], static_cast<DWORD>(buffer.size() - 1));
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
347 return std::string(&buffer[0]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
348 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
349
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
350 #elif defined(__linux__) || defined(__FreeBSD_kernel__) || defined(__FreeBSD__)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
351 static std::string GetPathToExecutableInternal()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
352 {
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
353 // NOTE: For FreeBSD, using KERN_PROC_PATHNAME might be a better alternative
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
354
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
355 std::vector<char> buffer(PATH_MAX + 1);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
356 ssize_t bytes = readlink("/proc/self/exe", &buffer[0], buffer.size() - 1);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
357 if (bytes == 0)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
358 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
359 throw OrthancException(ErrorCode_PathToExecutable);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
360 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
361
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
362 return std::string(&buffer[0]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
363 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
364
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
365 #elif defined(__APPLE__) && defined(__MACH__)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
366 static std::string GetPathToExecutableInternal()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
367 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
368 char pathbuf[PATH_MAX + 1];
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
369 unsigned int bufsize = static_cast<int>(sizeof(pathbuf));
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
370
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
371 _NSGetExecutablePath( pathbuf, &bufsize);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
372
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
373 return std::string(pathbuf);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
374 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
375
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
376 #elif defined(__OpenBSD__)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
377 static std::string GetPathToExecutableInternal()
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
378 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
379 // This is an adapted version of the patch proposed in issue #64
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
380 // without an explicit call to "malloc()" to prevent memory leak
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
381 // https://bitbucket.org/sjodogne/orthanc/issues/64/add-openbsd-support
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
382 // https://stackoverflow.com/q/31494901/881731
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
383
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
384 const int mib[4] = { CTL_KERN, KERN_PROC_ARGS, getpid(), KERN_PROC_ARGV };
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
385
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
386 size_t len;
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
387 if (sysctl(mib, 4, NULL, &len, NULL, 0) == -1)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
388 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
389 throw OrthancException(ErrorCode_PathToExecutable);
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
390 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
391
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
392 std::string tmp;
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
393 tmp.resize(len);
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
394
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
395 char** buffer = reinterpret_cast<char**>(&tmp[0]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
396
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
397 if (sysctl(mib, 4, buffer, &len, NULL, 0) == -1)
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
398 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
399 throw OrthancException(ErrorCode_PathToExecutable);
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
400 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
401 else
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
402 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
403 return std::string(buffer[0]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
404 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
405 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
406
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
407 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
408 #error Support your platform here
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
409 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
410
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
411
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
412 std::string SystemToolbox::GetPathToExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
413 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
414 boost::filesystem::path p(GetPathToExecutableInternal());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
415 return boost::filesystem::absolute(p).string();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
416 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
417
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
418
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
419 std::string SystemToolbox::GetDirectoryOfExecutable()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
420 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
421 boost::filesystem::path p(GetPathToExecutableInternal());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
422 return boost::filesystem::absolute(p.parent_path()).string();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
423 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
424
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
425
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
426 void SystemToolbox::ExecuteSystemCommand(const std::string& command,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
427 const std::vector<std::string>& arguments)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
428 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
429 // Convert the arguments as a C array
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
430 std::vector<char*> args(arguments.size() + 2);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
431
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
432 args.front() = const_cast<char*>(command.c_str());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
433
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
434 for (size_t i = 0; i < arguments.size(); i++)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
435 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
436 args[i + 1] = const_cast<char*>(arguments[i].c_str());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
437 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
438
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
439 args.back() = NULL;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
440
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
441 int status;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
442
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
443 #if defined(_WIN32)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
444 // http://msdn.microsoft.com/en-us/library/275khfab.aspx
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
445 status = static_cast<int>(_spawnvp(_P_OVERLAY, command.c_str(), &args[0]));
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
446
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
447 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
448 int pid = fork();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
449
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
450 if (pid == -1)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
451 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
452 // Error in fork()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
453 #if ORTHANC_ENABLE_LOGGING == 1
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
454 LOG(ERROR) << "Cannot fork a child process";
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
455 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
456
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
457 throw OrthancException(ErrorCode_SystemCommand);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
458 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
459 else if (pid == 0)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
460 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
461 // Execute the system command in the child process
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
462 execvp(command.c_str(), &args[0]);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
463
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
464 // We should never get here
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
465 _exit(1);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
466 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
467 else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
468 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
469 // Wait for the system command to exit
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
470 waitpid(pid, &status, 0);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
471 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
472 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
473
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
474 if (status != 0)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
475 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
476 #if ORTHANC_ENABLE_LOGGING == 1
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
477 LOG(ERROR) << "System command failed with status code " << status;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
478 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
479
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
480 throw OrthancException(ErrorCode_SystemCommand);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
481 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
482 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
483
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
484
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
485 int SystemToolbox::GetProcessId()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
486 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
487 #if defined(_WIN32)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
488 return static_cast<int>(_getpid());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
489 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
490 return static_cast<int>(getpid());
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
491 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
492 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
493
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
494
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
495 bool SystemToolbox::IsRegularFile(const std::string& path)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
496 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
497 namespace fs = boost::filesystem;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
498
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
499 try
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
500 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
501 if (fs::exists(path))
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
502 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
503 fs::file_status status = fs::status(path);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
504 return (status.type() == boost::filesystem::regular_file ||
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
505 status.type() == boost::filesystem::reparse_file); // Fix BitBucket issue #11
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
506 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
507 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
508 catch (fs::filesystem_error&)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
509 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
510 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
511
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
512 return false;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
513 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
514
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
515
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
516 FILE* SystemToolbox::OpenFile(const std::string& path,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
517 FileMode mode)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
518 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
519 #if defined(_WIN32)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
520 // TODO Deal with special characters by converting to the current locale
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
521 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
522
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
523 const char* m;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
524 switch (mode)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
525 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
526 case FileMode_ReadBinary:
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
527 m = "rb";
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
528 break;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
529
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
530 case FileMode_WriteBinary:
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
531 m = "wb";
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
532 break;
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
533
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
534 default:
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
535 throw OrthancException(ErrorCode_ParameterOutOfRange);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
536 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
537
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
538 return fopen(path.c_str(), m);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
539 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
540
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
541
155
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
542 std::string SystemToolbox::GenerateUuid()
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
543 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
544 #ifdef WIN32
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
545 UUID uuid;
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
546 UuidCreate ( &uuid );
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
547
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
548 unsigned char * str;
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
549 UuidToStringA ( &uuid, &str );
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
550
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
551 std::string s( ( char* ) str );
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
552
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
553 RpcStringFreeA ( &str );
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
554 #else
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
555 uuid_t uuid;
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
556 uuid_generate_random ( uuid );
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
557 char s[37];
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
558 uuid_unparse ( uuid, s );
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
559 #endif
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
560 return s;
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
561 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
562
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 150
diff changeset
563
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
564 std::string SystemToolbox::GetNowIsoString()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
565 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
566 boost::posix_time::ptime now = boost::posix_time::second_clock::local_time();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
567 return boost::posix_time::to_iso_string(now);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
568 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
569
167
Sebastien Jodogne <s.jodogne@gmail.com>
parents: 159
diff changeset
570
150
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
571 void SystemToolbox::GetNowDicom(std::string& date,
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
572 std::string& time)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
573 {
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
574 boost::posix_time::ptime now = boost::posix_time::second_clock::local_time();
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
575 tm tm = boost::posix_time::to_tm(now);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
576
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
577 char s[32];
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
578 sprintf(s, "%04d%02d%02d", tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
579 date.assign(s);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
580
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
581 // TODO milliseconds
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
582 sprintf(s, "%02d%02d%02d.%06d", tm.tm_hour, tm.tm_min, tm.tm_sec, 0);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
583 time.assign(s);
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
584 }
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff changeset
585 }