GCC Code Coverage Report

Directory: cvmfs/
File: cvmfs/util/posix.cc
Date: 2026-11-01 02:41:19
Exec Total Coverage
Lines: 904 1160 77.9%
Branches: 671 1265 53.0%
Line Branch Exec Source
1 /**
2 * This file is part of the CernVM File System.
3 *
4 * Some common functions.
5 */
6
7 #include <cctype>
8 #include <stdlib.h>
9 #include <cstdlib>
10 #include <dirent.h>
11 #include <sys/file.h>
12 #include <string.h>
13 #include <stdio.h>
14 #include <sched.h>
15 #include <sys/select.h>
16 #include <sys/uio.h>
17
18 #include "posix.h"
19
20 #include <arpa/inet.h>
21 #include <errno.h>
22 #include <fcntl.h>
23 #include <grp.h>
24 #include <inttypes.h>
25 #include <netinet/in.h>
26 #include <pthread.h>
27 #include <pwd.h>
28 #include <signal.h>
29 #include <stdint.h>
30 #include <sys/resource.h>
31 #include <sys/socket.h>
32 #include <sys/stat.h>
33 #ifdef __APPLE__
34 #include <sys/mount.h> // for statfs()
35 #else
36 #include <sys/statfs.h>
37 #endif
38 #include <sys/time.h>
39 #include <sys/types.h>
40 #include <sys/un.h>
41 #include <sys/utsname.h>
42 #include <sys/wait.h>
43 #include <unistd.h>
44 // If valgrind headers are present on the build system, then we can detect
45 // valgrind at runtime.
46 #ifdef HAS_VALGRIND_HEADERS
47 #include <valgrind/valgrind.h>
48 #endif
49
50 #include <algorithm>
51 #include <cassert>
52 #include <cstdio>
53 #include <cstring>
54 #include <map>
55 #include <set>
56 #include <string>
57 #include <vector>
58
59 #include "util/algorithm.h"
60 #include "util/capabilities.h"
61 #include "util/exception.h"
62 #include "util/fs_traversal.h"
63 #include "util/logging.h"
64 #include "util/pipe.h"
65 #include "util/string.h"
66 #include "util/smalloc.h"
67 #include "util/smalloc.h"
68 #include "util/smalloc.h"
69 #include "util/mutex.h"
70 #include "util/mutex.h"
71 #include "util/mutex.h"
72
73 // using namespace std; // NOLINT
74
75 #ifndef ST_RDONLY
76 // On Linux, this is in sys/statvfs.h
77 // On macOS, this flag is called MNT_RDONLY /usr/include/sys/mount.h
78 #define ST_RDONLY 1
79 #endif
80
81 // Older Linux glibc versions do not provide the f_flags member in struct statfs
82 #define CVMFS_HAS_STATFS_F_FLAGS
83 #ifndef __APPLE__
84 #ifdef __GLIBC_MINOR__
85 #if __GLIBC_MINOR__ < 12
86 #undef CVMFS_HAS_STATFS_F_FLAGS
87 #endif
88 #endif
89 #endif
90
91 // Work around missing clearenv()
92 #ifdef __APPLE__
93 extern "C" {
94 extern char **environ;
95 }
96 #endif
97
98 #ifdef CVMFS_NAMESPACE_GUARD
99 namespace CVMFS_NAMESPACE_GUARD {
100 #endif
101
102 static pthread_mutex_t getumask_mutex = PTHREAD_MUTEX_INITIALIZER;
103
104
105 /**
106 * Removes a trailing "/" from a path.
107 */
108 12965 std::string MakeCanonicalPath(const std::string &path) {
109
2/2
✓ Branch 1 taken 16 times.
✓ Branch 2 taken 12949 times.
 12965 if (path.length() == 0)
110 16 return path;
111
112
2/2
✓ Branch 2 taken 214 times.
✓ Branch 3 taken 12735 times.
 12949 if (path[path.length() - 1] == '/') {
113 214 return path.substr(0, path.length() - 1);
114 } else {
115 12735 return path;
116 }
117 }
118
119 /**
120 * Return both the file and directory name for a given path.
121 *
122 * NOTE: If only a filename is given, the directory is returned as "."
123 */
124 1132 void SplitPath(const std::string &path,
125 std::string *dirname,
126 std::string *filename) {
127 1132 const size_t dir_sep = path.rfind('/');
128
2/2
✓ Branch 0 taken 970 times.
✓ Branch 1 taken 162 times.
 1132 if (dir_sep != std::string::npos) {
129 970 *dirname = path.substr(0, dir_sep);
130 970 *filename = path.substr(dir_sep + 1);
131 } else {
132 162 *dirname = ".";
133 162 *filename = path;
134 }
135 1132 }
136
137
138 /**
139 * Gets the directory part of a path.
140 */
141 11095 std::string GetParentPath(const std::string &path) {
142 11095 const std::string::size_type idx = path.find_last_of('/');
143
2/2
✓ Branch 0 taken 10958 times.
✓ Branch 1 taken 137 times.
 11095 if (idx != std::string::npos) {
144 10958 return path.substr(0, idx);
145 } else {
146
1/2
✓ Branch 2 taken 137 times.
✗ Branch 3 not taken.
 137 return "";
147 }
148 }
149
150
151 /**
152 * Gets the file name part of a path.
153 */
154 349115 std::string GetFileName(const std::string &path) {
155 349115 const std::string::size_type idx = path.find_last_of('/');
156
2/2
✓ Branch 0 taken 348698 times.
✓ Branch 1 taken 417 times.
 349115 if (idx != std::string::npos) {
157 348698 return path.substr(idx + 1);
158 } else {
159 417 return path;
160 }
161 }
162
163
164 262 bool IsAbsolutePath(const std::string &path) {
165
4/4
✓ Branch 1 taken 261 times.
✓ Branch 2 taken 1 times.
✓ Branch 4 taken 14 times.
✓ Branch 5 taken 247 times.
 262 return (!path.empty() && path[0] == '/');
166 }
167
168
169 259 std::string GetAbsolutePath(const std::string &path) {
170
2/2
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 246 times.
 259 if (IsAbsolutePath(path))
171 13 return path;
172
173
2/4
✓ Branch 2 taken 246 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 246 times.
✗ Branch 6 not taken.
 492 return GetCurrentWorkingDirectory() + "/" + path;
174 }
175
176
177 2511 bool IsHttpUrl(const std::string &path) {
178
2/2
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 2507 times.
 2511 if (path.length() < 7) {
179 4 return false;
180 }
181
182
1/2
✓ Branch 1 taken 2507 times.
✗ Branch 2 not taken.
 2507 std::string prefix = path.substr(0, 8);
183 2507 std::transform(prefix.begin(), prefix.end(), prefix.begin(), ::tolower);
184
185
6/12
✓ Branch 1 taken 2507 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 2497 times.
✓ Branch 5 taken 10 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 10 times.
✓ Branch 8 taken 2487 times.
✓ Branch 9 taken 2507 times.
✗ Branch 10 not taken.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
 2507 return prefix.substr(0, 7) == "http://" || prefix == "https://";
186 2507 }
187
188
189 6013 FileSystemInfo GetFileSystemInfo(const std::string &path) {
190 6013 FileSystemInfo result;
191
192 struct statfs info;
193 6013 const int retval = statfs(path.c_str(), &info);
194
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6013 times.
 6013 if (retval != 0)
195 ✗ return result;
196
197
2/6
✗ Branch 0 not taken.
✗ Branch 1 not taken.
✓ Branch 2 taken 12 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 6001 times.
 6013 switch (info.f_type) {
198 ✗ case kFsTypeAutofs:
199 ✗ result.type = kFsTypeAutofs;
200 ✗ break;
201 ✗ case kFsTypeNFS:
202 ✗ result.type = kFsTypeNFS;
203 ✗ break;
204 12 case kFsTypeProc:
205 12 result.type = kFsTypeProc;
206 12 break;
207 ✗ case kFsTypeBeeGFS:
208 ✗ result.type = kFsTypeBeeGFS;
209 ✗ break;
210 ✗ case kFsTypeTmpfs:
211 ✗ result.type = kFsTypeTmpfs;
212 ✗ break;
213 6001 default:
214 6001 result.type = kFsTypeUnknown;
215 }
216
217 #ifdef CVMFS_HAS_STATFS_F_FLAGS
218
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6013 times.
 6013 if (info.f_flags & ST_RDONLY)
219 ✗ result.is_rdonly = true;
220 #else
221 // On old Linux systems, fall back to access()
222 retval = access(path.c_str(), W_OK);
223 result.is_rdonly = (retval != 0);
224 #endif
225
226
227 6013 return result;
228 }
229
230
231 773 std::string ReadSymlink(const std::string &path) {
232 // TODO(jblomer): avoid PATH_MAX
233 char buf[PATH_MAX + 1];
234 773 const ssize_t nchars = readlink(path.c_str(), buf, PATH_MAX);
235
2/2
✓ Branch 0 taken 731 times.
✓ Branch 1 taken 42 times.
 773 if (nchars >= 0) {
236 731 buf[nchars] = '\0';
237
1/2
✓ Branch 2 taken 731 times.
✗ Branch 3 not taken.
 731 return std::string(buf);
238 }
239
1/2
✓ Branch 2 taken 42 times.
✗ Branch 3 not taken.
 42 return "";
240 }
241
242
243 /**
244 * Follow all symlinks if possible. Equivalent to
245 * `readlink --canonicalize-missing`
246 */
247 101 std::string ResolvePath(const std::string &path) {
248
6/6
✓ Branch 1 taken 69 times.
✓ Branch 2 taken 32 times.
✓ Branch 4 taken 15 times.
✓ Branch 5 taken 54 times.
✓ Branch 6 taken 47 times.
✓ Branch 7 taken 54 times.
 101 if (path.empty() || (path == "/"))
249
1/2
✓ Branch 2 taken 47 times.
✗ Branch 3 not taken.
 47 return "/";
250
1/2
✓ Branch 1 taken 54 times.
✗ Branch 2 not taken.
 54 const std::string name = GetFileName(path);
251
1/2
✓ Branch 1 taken 54 times.
✗ Branch 2 not taken.
 54 std::string result = name;
252
2/2
✓ Branch 1 taken 52 times.
✓ Branch 2 taken 2 times.
 54 if (name != path) {
253 // There is a parent path of 'path'
254
2/4
✓ Branch 1 taken 52 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 52 times.
✗ Branch 5 not taken.
 52 const std::string parent = ResolvePath(GetParentPath(path));
255
4/6
✓ Branch 1 taken 21 times.
✓ Branch 2 taken 31 times.
✓ Branch 4 taken 52 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 52 times.
✗ Branch 8 not taken.
 52 result = parent + (parent == "/" ? "" : "/") + name;
256 52 }
257 54 char *real_result = realpath(result.c_str(), NULL);
258
2/2
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 46 times.
 54 if (real_result) {
259
1/2
✓ Branch 1 taken 8 times.
✗ Branch 2 not taken.
 8 result = real_result;
260 8 free(real_result);
261 }
262
3/4
✓ Branch 1 taken 54 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 53 times.
 54 if (SymlinkExists(result)) {
263 char buf[PATH_MAX + 1];
264 1 const ssize_t nchars = readlink(result.c_str(), buf, PATH_MAX);
265
1/2
✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
 1 if (nchars >= 0) {
266 1 buf[nchars] = '\0';
267
1/2
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
 1 result = buf;
268 }
269 }
270 54 return result;
271 54 }
272
273
274 42 bool IsMountPoint(const std::string &path) {
275
1/2
✓ Branch 1 taken 42 times.
✗ Branch 2 not taken.
 42 std::vector<std::string> mount_list = platform_mountlist();
276
1/2
✓ Branch 1 taken 42 times.
✗ Branch 2 not taken.
 42 const std::string resolved_path = ResolvePath(path);
277
2/2
✓ Branch 1 taken 3444 times.
✓ Branch 2 taken 14 times.
 3458 for (unsigned i = 0; i < mount_list.size(); ++i) {
278
2/2
✓ Branch 2 taken 28 times.
✓ Branch 3 taken 3416 times.
 3444 if (mount_list[i] == resolved_path)
279 28 return true;
280 }
281 14 return false;
282 42 }
283
284
285 /**
286 * By default PANIC(NULL) on failure
287 */
288 2746 void CreateFile(const std::string &path,
289 const int mode,
290 const bool ignore_failure) {
291 2746 const int fd = open(path.c_str(), O_CREAT, mode);
292
2/2
✓ Branch 0 taken 2556 times.
✓ Branch 1 taken 190 times.
 2746 if (fd >= 0) {
293 2556 close(fd);
294 2556 return;
295 }
296
2/2
✓ Branch 0 taken 178 times.
✓ Branch 1 taken 12 times.
 190 if (ignore_failure)
297 178 return;
298 12 PANIC(NULL);
299 }
300
301
302 /**
303 * Symlinks /tmp/cvmfs.XYZ/l --> ParentPath(path) to make it shorter
304 */
305 71 static std::string MakeShortSocketLink(const std::string &path) {
306 //NOLINTNEXTLINE(misc-const-correctness)
307 struct sockaddr_un sock_addr;
308 71 const unsigned max_length = sizeof(sock_addr.sun_path);
309
310 71 std::string result;
311
2/4
✓ Branch 2 taken 71 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 71 times.
✗ Branch 6 not taken.
 142 const std::string tmp_path = CreateTempDir("/tmp/cvmfs");
312
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 71 times.
 71 if (tmp_path.empty())
313 ✗ return "";
314
1/2
✓ Branch 1 taken 71 times.
✗ Branch 2 not taken.
 71 const std::string link = tmp_path + "/l";
315
3/6
✓ Branch 1 taken 71 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 71 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 71 times.
✗ Branch 8 not taken.
 71 result = link + "/" + GetFileName(path);
316
2/2
✓ Branch 1 taken 28 times.
✓ Branch 2 taken 43 times.
 71 if (result.length() >= max_length) {
317 28 rmdir(tmp_path.c_str());
318
1/2
✓ Branch 2 taken 28 times.
✗ Branch 3 not taken.
 28 return "";
319 }
320
1/2
✓ Branch 2 taken 43 times.
✗ Branch 3 not taken.
 43 const int retval = symlink(GetParentPath(path).c_str(), link.c_str());
321
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 43 times.
 43 if (retval != 0) {
322 ✗ rmdir(tmp_path.c_str());
323 ✗ return "";
324 }
325 43 return result;
326 71 }
327
328 43 static void RemoveShortSocketLink(const std::string &short_path) {
329
1/2
✓ Branch 1 taken 43 times.
✗ Branch 2 not taken.
 43 const std::string link = GetParentPath(short_path);
330 43 unlink(link.c_str());
331
1/2
✓ Branch 1 taken 43 times.
✗ Branch 2 not taken.
 43 rmdir(GetParentPath(link).c_str());
332 43 }
333
334
335 /**
336 * Creates and binds to a named socket.
337 */
338 2053 int MakeSocket(const std::string &path, const int mode) {
339
1/2
✓ Branch 1 taken 2053 times.
✗ Branch 2 not taken.
 2053 std::string short_path(path);
340 struct sockaddr_un sock_addr;
341
2/2
✓ Branch 1 taken 41 times.
✓ Branch 2 taken 2012 times.
 2053 if (path.length() >= sizeof(sock_addr.sun_path)) {
342 // Socket paths are limited to 108 bytes (on some systems to 92 bytes),
343 // try working around
344
1/2
✓ Branch 1 taken 41 times.
✗ Branch 2 not taken.
 41 short_path = MakeShortSocketLink(path);
345
2/2
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 28 times.
 41 if (short_path.empty())
346 13 return -1;
347 }
348 2040 sock_addr.sun_family = AF_UNIX;
349 2040 strncpy(sock_addr.sun_path, short_path.c_str(), sizeof(sock_addr.sun_path));
350
351 2040 const int socket_fd = socket(AF_UNIX, SOCK_STREAM, 0);
352
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2040 times.
 2040 assert(socket_fd != -1);
353
354 #ifndef __APPLE__
355 // fchmod on a socket is not allowed under Mac OS X
356 // using default 0770 here
357
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 2040 times.
 2040 if (fchmod(socket_fd, mode) != 0)
358 ✗ goto make_socket_failure;
359 #endif
360
361 2040 if (bind(socket_fd, reinterpret_cast<struct sockaddr *>(&sock_addr),
362 sizeof(sock_addr.sun_family) + sizeof(sock_addr.sun_path))
363
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 2027 times.
 2040 < 0) {
364
3/6
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
✓ Branch 4 taken 13 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 13 times.
✗ Branch 7 not taken.
 13 if ((errno == EADDRINUSE) && (unlink(path.c_str()) == 0)) {
365 // Second try, perhaps the file was left over
366 13 if (bind(socket_fd, reinterpret_cast<struct sockaddr *>(&sock_addr),
367 sizeof(sock_addr.sun_family) + sizeof(sock_addr.sun_path))
368
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
 13 < 0) {
369 ✗ LogCvmfs(kLogCvmfs, kLogDebug, "binding socket failed (%d)", errno);
370 ✗ goto make_socket_failure;
371 }
372 } else {
373 ✗ LogCvmfs(kLogCvmfs, kLogDebug, "binding socket failed (%d)", errno);
374 ✗ goto make_socket_failure;
375 }
376 }
377
378
2/2
✓ Branch 1 taken 28 times.
✓ Branch 2 taken 2012 times.
 2040 if (short_path != path)
379
1/2
✓ Branch 1 taken 28 times.
✗ Branch 2 not taken.
 28 RemoveShortSocketLink(short_path);
380
381 2040 return socket_fd;
382
383 ✗ make_socket_failure:
384 ✗ close(socket_fd);
385 ✗ if (short_path != path)
386 ✗ RemoveShortSocketLink(short_path);
387 ✗ return -1;
388 2053 }
389
390
391 /**
392 * Creates and binds a TCP/IPv4 socket. An empty address binds to the "any"
393 * address.
394 */
395 1021 int MakeTcpEndpoint(const std::string &ipv4_address, int portno) {
396 1021 const int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
397
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1021 times.
 1021 assert(socket_fd != -1);
398 1021 const int on = 1;
399 1021 int retval = setsockopt(socket_fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
400
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1021 times.
 1021 assert(retval == 0);
401
402 struct sockaddr_in endpoint_addr;
403 1021 memset(&endpoint_addr, 0, sizeof(endpoint_addr));
404 1021 endpoint_addr.sin_family = AF_INET;
405
2/2
✓ Branch 1 taken 994 times.
✓ Branch 2 taken 27 times.
 1021 if (ipv4_address.empty()) {
406 994 endpoint_addr.sin_addr.s_addr = INADDR_ANY;
407 } else {
408 27 retval = inet_aton(ipv4_address.c_str(), &(endpoint_addr.sin_addr));
409
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 18 times.
 27 if (retval == 0) {
410
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 LogCvmfs(kLogCvmfs, kLogDebug, "invalid IPv4 address");
411
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 close(socket_fd);
412 9 return -1;
413 }
414 }
415 1012 endpoint_addr.sin_port = htons(portno);
416
417 1012 retval = bind(socket_fd, reinterpret_cast<struct sockaddr *>(&endpoint_addr),
418 sizeof(endpoint_addr));
419
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 1003 times.
 1012 if (retval < 0) {
420
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 LogCvmfs(kLogCvmfs, kLogDebug, "binding TCP endpoint failed (%d)", errno);
421
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 close(socket_fd);
422 9 return -1;
423 }
424 1003 return socket_fd;
425 }
426
427
428 /**
429 * Connects to a named socket.
430 *
431 * \return socket file descriptor on success, -1 else
432 */
433 678 int ConnectSocket(const std::string &path) {
434
1/2
✓ Branch 1 taken 678 times.
✗ Branch 2 not taken.
 678 std::string short_path(path);
435 struct sockaddr_un sock_addr;
436
2/2
✓ Branch 1 taken 30 times.
✓ Branch 2 taken 648 times.
 678 if (path.length() >= sizeof(sock_addr.sun_path)) {
437 // Socket paths are limited to 108 bytes (on some systems to 92 bytes),
438 // try working around
439
1/2
✓ Branch 1 taken 30 times.
✗ Branch 2 not taken.
 30 short_path = MakeShortSocketLink(path);
440
2/2
✓ Branch 1 taken 15 times.
✓ Branch 2 taken 15 times.
 30 if (short_path.empty())
441 15 return -1;
442 }
443 663 sock_addr.sun_family = AF_UNIX;
444 663 strncpy(sock_addr.sun_path, short_path.c_str(), sizeof(sock_addr.sun_path));
445
446 663 const int socket_fd = socket(AF_UNIX, SOCK_STREAM, 0);
447
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 663 times.
 663 assert(socket_fd != -1);
448
449
1/2
✓ Branch 1 taken 663 times.
✗ Branch 2 not taken.
 663 const int retval = connect(
450 socket_fd, reinterpret_cast<struct sockaddr *>(&sock_addr),
451 sizeof(sock_addr.sun_family) + sizeof(sock_addr.sun_path));
452
2/2
✓ Branch 1 taken 15 times.
✓ Branch 2 taken 648 times.
 663 if (short_path != path)
453
1/2
✓ Branch 1 taken 15 times.
✗ Branch 2 not taken.
 15 RemoveShortSocketLink(short_path);
454
455
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 648 times.
 663 if (retval < 0) {
456
1/2
✓ Branch 1 taken 15 times.
✗ Branch 2 not taken.
 15 close(socket_fd);
457 15 return -1;
458 }
459
460 648 return socket_fd;
461 678 }
462
463
464 /**
465 * Connects to a (remote) TCP server
466 */
467 53 int ConnectTcpEndpoint(const std::string &ipv4_address, int portno) {
468 53 const int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
469
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 53 times.
 53 assert(socket_fd != -1);
470
471 struct sockaddr_in endpoint_addr;
472 53 memset(&endpoint_addr, 0, sizeof(endpoint_addr));
473 53 endpoint_addr.sin_family = AF_INET;
474 53 int retval = inet_aton(ipv4_address.c_str(), &(endpoint_addr.sin_addr));
475
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 44 times.
 53 if (retval == 0) {
476
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 LogCvmfs(kLogCvmfs, kLogDebug, "invalid IPv4 address");
477
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 close(socket_fd);
478 9 return -1;
479 }
480 44 endpoint_addr.sin_port = htons(portno);
481
482
1/2
✓ Branch 1 taken 44 times.
✗ Branch 2 not taken.
 44 retval = connect(socket_fd,
483 reinterpret_cast<struct sockaddr *>(&endpoint_addr),
484 sizeof(endpoint_addr));
485
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 35 times.
 44 if (retval != 0) {
486 9 LogCvmfs(kLogCvmfs, kLogDebug, "failed to connect to TCP endpoint (%d)",
487
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 errno);
488
1/2
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
 9 close(socket_fd);
489 9 return -1;
490 }
491 35 return socket_fd;
492 }
493
494
495 /**
496 * Creating a pipe should always succeed.
497 */
498 39119 void MakePipe(int pipe_fd[2]) {
499 39119 const int retval = pipe(pipe_fd);
500
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 39119 times.
 39119 assert(retval == 0);
501 39119 }
502
503
504 /**
505 * Writes to a pipe should always succeed.
506 */
507 15909444 void WritePipe(int fd, const void *buf, size_t nbyte) {
508 ssize_t num_bytes;
509 do {
510 15909444 num_bytes = write(fd, buf, nbyte);
511
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 15909444 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 15909444 } while ((num_bytes < 0) && (errno == EINTR));
512
2/4
✓ Branch 0 taken 15909444 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 15909444 times.
✗ Branch 3 not taken.
 15909444 assert((num_bytes >= 0) && (static_cast<size_t>(num_bytes) == nbyte));
513 15909444 }
514
515
516 /**
517 * Reads from a pipe should always succeed.
518 */
519 81470703 void ReadPipe(int fd, void *buf, size_t nbyte) {
520 ssize_t num_bytes;
521 do {
522 81470703 num_bytes = read(fd, buf, nbyte);
523
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 81470703 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 81470703 } while ((num_bytes < 0) && (errno == EINTR));
524
2/4
✓ Branch 0 taken 81470703 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 81470703 times.
✗ Branch 3 not taken.
 81470703 assert((num_bytes >= 0) && (static_cast<size_t>(num_bytes) == nbyte));
525 81470703 }
526
527
528 /**
529 * Reads from a pipe where writer's end is not yet necessarily connected
530 */
531 4506178 bool ReadHalfPipe(int fd, void *buf, size_t nbyte, unsigned timeout_ms) {
532 ssize_t num_bytes;
533 4506178 unsigned i = 0;
534 4506178 unsigned backoff_ms = 1;
535 4506178 uint64_t duration_ms = 0;
536 4506178 uint64_t timestamp = 0;
537
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 4506167 times.
 4506178 if (timeout_ms != 0)
538 11 timestamp = platform_monotonic_time_ns();
539
540 4506178 const unsigned max_backoff_ms = 256;
541 do {
542 // When the writer is not connected, this takes ~200-300ns per call as per
543 // micro benchmarks
544 4506178 num_bytes = read(fd, buf, nbyte);
545
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4506178 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4506178 if ((num_bytes < 0) && (errno == EINTR))
546 ✗ continue;
547 4506178 i++;
548 // Start backing off when the busy loop reaches the ballpark of 1ms
549
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 4506178 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 4506178 if ((i > 3000) && (num_bytes == 0)) {
550 // The BackoffThrottle would pull in too many dependencies
551 ✗ SafeSleepMs(backoff_ms);
552 ✗ if (backoff_ms < max_backoff_ms)
553 ✗ backoff_ms *= 2;
554 }
555
3/4
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 4506167 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 11 times.
 4506178 if ((timeout_ms != 0) && (num_bytes == 0)) {
556 ✗ duration_ms = (platform_monotonic_time_ns() - timestamp)
557 / (1000UL * 1000UL);
558 ✗ if (duration_ms > timeout_ms)
559 ✗ return false;
560 }
561
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4506178 times.
 4506178 } while (num_bytes == 0);
562
2/4
✓ Branch 0 taken 4506178 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4506178 times.
✗ Branch 3 not taken.
 4506178 assert((num_bytes >= 0) && (static_cast<size_t>(num_bytes) == nbyte));
563 4506178 return true;
564 }
565
566
567 /**
568 * Closes both ends of a pipe
569 */
570 34431 void ClosePipe(int pipe_fd[2]) {
571 34431 close(pipe_fd[0]);
572 34431 close(pipe_fd[1]);
573 34431 }
574
575
576 /**
577 * Compares two directory trees on the meta-data level. Returns true iff the
578 * trees have identical content.
579 */
580 166243 bool DiffTree(const std::string &path_a, const std::string &path_b) {
581 int retval;
582 166243 std::vector<std::string> ls_a;
583 166243 std::vector<std::string> ls_b;
584 166243 std::vector<std::string> subdirs;
585
586
1/2
✓ Branch 2 taken 166243 times.
✗ Branch 3 not taken.
 166243 DIR *dirp_a = opendir(path_a.c_str());
587
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 166243 times.
 166243 if (dirp_a == NULL)
588 ✗ return false;
589
1/2
✓ Branch 2 taken 166243 times.
✗ Branch 3 not taken.
 166243 DIR *dirp_b = opendir(path_b.c_str());
590
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 166243 times.
 166243 if (dirp_b == NULL) {
591 ✗ closedir(dirp_a);
592 ✗ return false;
593 }
594
595 platform_dirent64 *dirent;
596
3/4
✓ Branch 1 taken 672804 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 506561 times.
✓ Branch 4 taken 166243 times.
 672804 while ((dirent = platform_readdir(dirp_a))) {
597
1/2
✓ Branch 2 taken 506561 times.
✗ Branch 3 not taken.
 506561 const std::string name(dirent->d_name);
598
6/6
✓ Branch 1 taken 340318 times.
✓ Branch 2 taken 166243 times.
✓ Branch 4 taken 166243 times.
✓ Branch 5 taken 174075 times.
✓ Branch 6 taken 332486 times.
✓ Branch 7 taken 174075 times.
 506561 if ((name == ".") || (name == ".."))
599 332486 continue;
600
2/4
✓ Branch 1 taken 174075 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 174075 times.
✗ Branch 5 not taken.
 174075 const std::string path = path_a + "/" + name;
601
1/2
✓ Branch 1 taken 174075 times.
✗ Branch 2 not taken.
 174075 ls_a.push_back(path);
602
603 platform_stat64 info;
604 174075 retval = platform_lstat(path.c_str(), &info);
605
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 174075 times.
 174075 if (retval != 0) {
606 ✗ closedir(dirp_a);
607 ✗ closedir(dirp_b);
608 ✗ return false;
609 }
610
2/2
✓ Branch 0 taken 166727 times.
✓ Branch 1 taken 7348 times.
 174075 if (S_ISDIR(info.st_mode))
611
1/2
✓ Branch 1 taken 166727 times.
✗ Branch 2 not taken.
 166727 subdirs.push_back(name);
612
3/5
✓ Branch 1 taken 174075 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 174075 times.
✓ Branch 5 taken 332486 times.
✗ Branch 6 not taken.
 506561 }
613
3/4
✓ Branch 1 taken 672254 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 506011 times.
✓ Branch 4 taken 166243 times.
 672254 while ((dirent = platform_readdir(dirp_b))) {
614
1/2
✓ Branch 2 taken 506011 times.
✗ Branch 3 not taken.
 506011 const std::string name(dirent->d_name);
615
6/6
✓ Branch 1 taken 339768 times.
✓ Branch 2 taken 166243 times.
✓ Branch 4 taken 166243 times.
✓ Branch 5 taken 173525 times.
✓ Branch 6 taken 332486 times.
✓ Branch 7 taken 173525 times.
 506011 if ((name == ".") || (name == ".."))
616 332486 continue;
617
2/4
✓ Branch 1 taken 173525 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 173525 times.
✗ Branch 5 not taken.
 173525 const std::string path = path_b + "/" + name;
618
1/2
✓ Branch 1 taken 173525 times.
✗ Branch 2 not taken.
 173525 ls_b.push_back(path);
619
2/2
✓ Branch 2 taken 173525 times.
✓ Branch 3 taken 332486 times.
 506011 }
620
1/2
✓ Branch 1 taken 166243 times.
✗ Branch 2 not taken.
 166243 closedir(dirp_a);
621
1/2
✓ Branch 1 taken 166243 times.
✗ Branch 2 not taken.
 166243 closedir(dirp_b);
622
623
1/2
✓ Branch 3 taken 166243 times.
✗ Branch 4 not taken.
 166243 sort(ls_a.begin(), ls_a.end());
624
1/2
✓ Branch 3 taken 166243 times.
✗ Branch 4 not taken.
 166243 sort(ls_b.begin(), ls_b.end());
625
2/2
✓ Branch 2 taken 11 times.
✓ Branch 3 taken 166232 times.
 166243 if (ls_a.size() != ls_b.size())
626 11 return false;
627
2/2
✓ Branch 1 taken 173294 times.
✓ Branch 2 taken 166232 times.
 339526 for (unsigned i = 0; i < ls_a.size(); ++i) {
628
3/7
✓ Branch 2 taken 173294 times.
✗ Branch 3 not taken.
✓ Branch 6 taken 173294 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✗ Branch 11 not taken.
✓ Branch 12 taken 173294 times.
 173294 if (GetFileName(ls_a[i]) != GetFileName(ls_b[i]))
629 ✗ return false;
630 platform_stat64 info_a;
631 platform_stat64 info_b;
632 173294 retval = platform_lstat(ls_a[i].c_str(), &info_a);
633
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 173294 times.
 173294 if (retval != 0)
634 ✗ return false;
635 173294 retval = platform_lstat(ls_b[i].c_str(), &info_b);
636
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 173294 times.
 173294 if (retval != 0)
637 ✗ return false;
638
2/4
✓ Branch 0 taken 173294 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 173294 times.
✗ Branch 3 not taken.
 173294 if ((info_a.st_mode != info_b.st_mode) || (info_a.st_uid != info_b.st_uid)
639
1/2
✓ Branch 0 taken 173294 times.
✗ Branch 1 not taken.
 173294 || (info_a.st_gid != info_b.st_gid)
640
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 173294 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 173294 || ((info_a.st_size != info_b.st_size) && !S_ISDIR(info_a.st_mode))) {
641 ✗ return false;
642 }
643 }
644
645
2/2
✓ Branch 1 taken 166210 times.
✓ Branch 2 taken 166232 times.
 332442 for (unsigned i = 0; i < subdirs.size(); ++i) {
646
3/6
✓ Branch 2 taken 166210 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 166210 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 166210 times.
✗ Branch 9 not taken.
 166210 const bool retval_subtree = DiffTree(path_a + "/" + subdirs[i],
647
2/4
✓ Branch 2 taken 166210 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 166210 times.
✗ Branch 6 not taken.
 332420 path_b + "/" + subdirs[i]);
648
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 166210 times.
 166210 if (!retval_subtree)
649 ✗ return false;
650 }
651
652 166232 return true;
653 166243 }
654
655
656 /**
657 * Changes a non-blocking file descriptor to a blocking one.
658 */
659 194 void Nonblock2Block(int filedes) {
660 194 const int flags = fcntl(filedes, F_GETFL);
661
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 194 times.
 194 assert(flags != -1);
662 194 const int retval = fcntl(filedes, F_SETFL, flags & ~O_NONBLOCK);
663
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 194 times.
 194 assert(retval != -1);
664 194 }
665
666
667 /**
668 * Changes a blocking file descriptor to a non-blocking one.
669 */
670 186 void Block2Nonblock(int filedes) {
671 186 const int flags = fcntl(filedes, F_GETFL);
672
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 186 times.
 186 assert(flags != -1);
673 186 const int retval = fcntl(filedes, F_SETFL, flags | O_NONBLOCK);
674
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 186 times.
 186 assert(retval != -1);
675 186 }
676
677
678 /**
679 * Drops the characters of string to a socket. It doesn't matter
680 * if the other side has hung up.
681 */
682 389 void SendMsg2Socket(const int fd, const std::string &msg) {
683 389 (void)send(fd, &msg[0], msg.length(), MSG_NOSIGNAL);
684 389 }
685
686 /**
687 * Sends the file descriptor passing_fd to the socket socket_fd. Can be used
688 * to transfer an open file descriptor from one process to another. Use
689 * ConnectSocket() to get the socket_fd.
690 */
691 10 bool SendFd2Socket(int socket_fd, int passing_fd) {
692 union {
693 // Make sure that ctrl_msg is properly aligned.
694 struct cmsghdr align;
695 // Buffer large enough to store the file descriptor (ancillary data)
696 unsigned char buf[CMSG_SPACE(sizeof(int))];
697 } ctrl_msg;
698
699 10 memset(ctrl_msg.buf, 0, sizeof(ctrl_msg.buf));
700
701 struct msghdr msgh;
702 10 msgh.msg_name = NULL;
703 10 msgh.msg_namelen = 0;
704
705 10 unsigned char dummy = 0;
706 struct iovec iov;
707 10 iov.iov_base = &dummy;
708 10 iov.iov_len = 1;
709 10 msgh.msg_iov = &iov;
710 10 msgh.msg_iovlen = 1;
711
712 10 msgh.msg_control = ctrl_msg.buf;
713 10 msgh.msg_controllen = sizeof(ctrl_msg.buf);
714
1/2
✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
 10 struct cmsghdr *cmsgp = CMSG_FIRSTHDR(&msgh);
715 10 cmsgp->cmsg_len = CMSG_LEN(sizeof(int));
716 10 cmsgp->cmsg_level = SOL_SOCKET;
717 10 cmsgp->cmsg_type = SCM_RIGHTS;
718 10 memcpy(CMSG_DATA(cmsgp), &passing_fd, sizeof(int));
719
720
1/2
✓ Branch 1 taken 10 times.
✗ Branch 2 not taken.
 10 const ssize_t retval = sendmsg(socket_fd, &msgh, 0);
721 10 return (retval != -1);
722 }
723
724
725 /**
726 * Returns the file descriptor that has been sent with SendFd2Socket. The
727 * msg_fd file descriptor needs to come from a call to accept() on the socket
728 * where the passing file descriptor has been sent to.
729 * Returns -errno on error.
730 */
731 ✗ int RecvFdFromSocket(int msg_fd) {
732 union {
733 // Make sure that ctrl_msg is properly aligned.
734 struct cmsghdr align;
735 // Buffer large enough to store the file descriptor (ancillary data)
736 unsigned char buf[CMSG_SPACE(sizeof(int))];
737 } ctrl_msg;
738
739 ✗ memset(ctrl_msg.buf, 0, sizeof(ctrl_msg.buf));
740
741 struct msghdr msgh;
742 ✗ msgh.msg_name = NULL;
743 ✗ msgh.msg_namelen = 0;
744
745 unsigned char dummy;
746 struct iovec iov;
747 ✗ iov.iov_base = &dummy;
748 ✗ iov.iov_len = 1;
749 ✗ msgh.msg_iov = &iov;
750 ✗ msgh.msg_iovlen = 1;
751
752 ✗ msgh.msg_control = ctrl_msg.buf;
753 ✗ msgh.msg_controllen = sizeof(ctrl_msg.buf);
754
755 ✗ const ssize_t retval = recvmsg(msg_fd, &msgh, 0);
756 ✗ if (retval == -1)
757 ✗ return -errno;
758
759 ✗ struct cmsghdr *cmsgp = CMSG_FIRSTHDR(&msgh);
760 ✗ assert(cmsgp != NULL);
761 ✗ if (cmsgp->cmsg_len != CMSG_LEN(sizeof(int)))
762 ✗ return -ERANGE;
763 ✗ assert(cmsgp->cmsg_level == SOL_SOCKET);
764 ✗ assert(cmsgp->cmsg_type == SCM_RIGHTS);
765
766 int passing_fd;
767 ✗ memcpy(&passing_fd, CMSG_DATA(cmsgp), sizeof(int));
768 ✗ assert(passing_fd >= 0);
769 ✗ return passing_fd;
770 }
771
772
773 15 std::string GetHostname() {
774 char name[HOST_NAME_MAX + 1];
775 15 const int retval = gethostname(name, HOST_NAME_MAX);
776
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
 15 assert(retval == 0);
777
1/2
✓ Branch 2 taken 15 times.
✗ Branch 3 not taken.
 15 return name;
778 }
779
780
781 /**
782 * set(e){g/u}id wrapper.
783 * If temporarily is true, reserve the ability to switch back.
784 *
785 * avoid_mutexes: if true, means that we are a forked child and must not use
786 * mutexes.
787 */
788 16 bool SwitchCredentials(const uid_t uid, const gid_t gid, const bool temporarily,
789 const bool avoid_mutexes) {
790
1/2
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
 16 if (!avoid_mutexes) {
791 16 LogCvmfs(kLogCvmfs, kLogDebug,
792 "current credentials uid %d gid %d "
793 "euid %d egid %d, switching to %d %d (temp: %d)",
794 getuid(), getgid(), geteuid(), getegid(), uid, gid, temporarily);
795 }
796 16 int retval = 0;
797
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
 16 if (temporarily) {
798 ✗ if (gid != getegid())
799 ✗ retval = setegid(gid);
800 ✗ if ((retval == 0) && (uid != geteuid()))
801 ✗ retval = seteuid(uid);
802 } else {
803 // If effective uid is not root, we probably need setuid+setgid
804 // capabilities first
805
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 if (geteuid() != 0) {
806 16 retval = ObtainSetuidgidCapabilities(avoid_mutexes);
807
2/4
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
 16 if (!retval && !avoid_mutexes) {
808 16 LogCvmfs(kLogCvmfs, kLogDebug,
809 "unable to obtain setuid & setgid capabilities first");
810 }
811 // The setuid below will drop the capability again unless
812 // switching to root or keepcaps has been enabled.
813 }
814
1/4
✗ Branch 1 not taken.
✓ Branch 2 taken 16 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 16 retval = setgid(gid) || setuid(uid);
815 }
816
1/2
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
 16 if (!avoid_mutexes) {
817 16 LogCvmfs(kLogCvmfs, kLogDebug, "switch credentials result %d (%d)", retval,
818 16 errno);
819 }
820 16 return retval == 0;
821 }
822
823
824 /**
825 * Checks if the regular file path exists.
826 */
827 132365 bool FileExists(const std::string &path) {
828 platform_stat64 info;
829
4/4
✓ Branch 2 taken 61418 times.
✓ Branch 3 taken 70947 times.
✓ Branch 4 taken 59732 times.
✓ Branch 5 taken 1686 times.
 132365 return ((platform_lstat(path.c_str(), &info) == 0) && S_ISREG(info.st_mode));
830 }
831
832
833 /**
834 * Returns -1 on failure.
835 */
836 14859989 int64_t GetFileSize(const std::string &path) {
837 platform_stat64 info;
838 14859989 const int retval = platform_stat(path.c_str(), &info);
839
2/2
✓ Branch 0 taken 14788527 times.
✓ Branch 1 taken 109502 times.
 14898029 if (retval != 0)
840 14788527 return -1;
841 109502 return info.st_size;
842 }
843
844
845 /**
846 * Checks if the directory (not symlink) path exists.
847 */
848 2388 bool DirectoryExists(const std::string &path) {
849 platform_stat64 info;
850
3/4
✓ Branch 2 taken 2297 times.
✓ Branch 3 taken 91 times.
✓ Branch 4 taken 2297 times.
✗ Branch 5 not taken.
 2388 return ((platform_lstat(path.c_str(), &info) == 0) && S_ISDIR(info.st_mode));
851 }
852
853
854 /**
855 * Checks if the symlink file path exists.
856 */
857 1774 bool SymlinkExists(const std::string &path) {
858 platform_stat64 info;
859
4/4
✓ Branch 2 taken 1728 times.
✓ Branch 3 taken 46 times.
✓ Branch 4 taken 1720 times.
✓ Branch 5 taken 8 times.
 1774 return ((platform_lstat(path.c_str(), &info) == 0) && S_ISLNK(info.st_mode));
860 }
861
862
863 /**
864 * Equivalent of `ln -sf $src $dest`
865 */
866 100 bool SymlinkForced(const std::string &src, const std::string &dest) {
867 100 int retval = unlink(dest.c_str());
868
3/4
✓ Branch 0 taken 86 times.
✓ Branch 1 taken 14 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 86 times.
 100 if ((retval != 0) && (errno != ENOENT))
869 ✗ return false;
870 100 retval = symlink(src.c_str(), dest.c_str());
871 100 return retval == 0;
872 }
873
874
875 /**
876 * The mkdir -p command. Additionally checks if the directory is writable
877 * if it exists.
878 */
879 2208314 bool MkdirDeep(const std::string &path,
880 const mode_t mode,
881 bool verify_writable) {
882
2/2
✓ Branch 1 taken 36 times.
✓ Branch 2 taken 2208278 times.
 2208314 if (path == "")
883 36 return false;
884
885 2208278 int retval = mkdir(path.c_str(), mode);
886
2/2
✓ Branch 0 taken 2197333 times.
✓ Branch 1 taken 10945 times.
 2208278 if (retval == 0)
887 2197333 return true;
888
889 21890 if ((errno == ENOENT)
890
9/15
✓ Branch 0 taken 3950 times.
✓ Branch 1 taken 6995 times.
✓ Branch 3 taken 3950 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✓ Branch 6 taken 3950 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 3950 times.
✗ Branch 9 not taken.
✓ Branch 10 taken 3950 times.
✓ Branch 11 taken 6995 times.
✓ Branch 13 taken 3950 times.
✓ Branch 14 taken 6995 times.
✗ Branch 15 not taken.
✗ Branch 16 not taken.
 10945 && (MkdirDeep(GetParentPath(path), mode, verify_writable))) {
891 3950 return MkdirDeep(path, mode, verify_writable);
892 }
893
894
1/2
✓ Branch 0 taken 6995 times.
✗ Branch 1 not taken.
 6995 if (errno == EEXIST) {
895 platform_stat64 info;
896
5/6
✓ Branch 2 taken 6995 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 6935 times.
✓ Branch 5 taken 60 times.
✓ Branch 6 taken 6935 times.
✓ Branch 7 taken 60 times.
 6995 if ((platform_stat(path.c_str(), &info) == 0) && S_ISDIR(info.st_mode)) {
897
2/2
✓ Branch 0 taken 376 times.
✓ Branch 1 taken 6559 times.
 6935 if (verify_writable) {
898 376 retval = utimes(path.c_str(), NULL);
899
1/2
✓ Branch 0 taken 376 times.
✗ Branch 1 not taken.
 376 if (retval == 0)
900 6935 return true;
901 } else {
902 6559 return true;
903 }
904 }
905 }
906
907 60 return false;
908 }
909
910
911 /**
912 * Creates the "hash cache" directory structure in path.
913 */
914 7505 bool MakeCacheDirectories(const std::string &path, const mode_t mode) {
915
1/2
✓ Branch 1 taken 7505 times.
✗ Branch 2 not taken.
 7505 const std::string canonical_path = MakeCanonicalPath(path);
916
917
1/2
✓ Branch 1 taken 7505 times.
✗ Branch 2 not taken.
 7505 std::string this_path = canonical_path + "/quarantaine";
918
2/4
✓ Branch 1 taken 7505 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 7505 times.
 7505 if (!MkdirDeep(this_path, mode, false))
919 ✗ return false;
920
921
1/2
✓ Branch 1 taken 7505 times.
✗ Branch 2 not taken.
 7505 this_path = canonical_path + "/ff";
922
923 platform_stat64 stat_info;
924
2/2
✓ Branch 2 taken 7282 times.
✓ Branch 3 taken 223 times.
 7505 if (platform_stat(this_path.c_str(), &stat_info) != 0) {
925
1/2
✓ Branch 1 taken 7282 times.
✗ Branch 2 not taken.
 7282 this_path = canonical_path + "/txn";
926
2/4
✓ Branch 1 taken 7282 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 7282 times.
 7282 if (!MkdirDeep(this_path, mode, false))
927 ✗ return false;
928
2/2
✓ Branch 0 taken 1864192 times.
✓ Branch 1 taken 7282 times.
 1871474 for (int i = 0; i <= 0xff; i++) {
929 char hex[4];
930 1864192 snprintf(hex, sizeof(hex), "%02x", i);
931
3/6
✓ Branch 2 taken 1864192 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1864192 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1864192 times.
✗ Branch 9 not taken.
 1864192 this_path = canonical_path + "/" + std::string(hex);
932
2/4
✓ Branch 1 taken 1864192 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 1864192 times.
 1864192 if (!MkdirDeep(this_path, mode, false))
933 ✗ return false;
934 }
935 }
936 7505 return true;
937 7505 }
938
939
940 /**
941 * Tries to locks file path, return an error if file is already locked.
942 * Creates path if required.
943 *
944 * \return file descriptor, -1 on error, -2 if it would block
945 */
946 3424 int TryLockFile(const std::string &path) {
947 3424 const int fd_lockfile = open(path.c_str(), O_RDONLY | O_CREAT, 0600);
948
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 3409 times.
 3424 if (fd_lockfile < 0)
949 15 return -1;
950
951
2/2
✓ Branch 1 taken 17 times.
✓ Branch 2 taken 3392 times.
 3409 if (flock(fd_lockfile, LOCK_EX | LOCK_NB) != 0) {
952 17 close(fd_lockfile);
953
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 17 times.
 17 if (errno != EWOULDBLOCK)
954 ✗ return -1;
955 17 return -2;
956 }
957
958 3392 return fd_lockfile;
959 }
960
961
962 /**
963 * Tries to write the process id in a /var/run/progname.pid like file. Returns
964 * the same as TryLockFile.
965 *
966 * \return file descriptor, -1 on error, -2 if it would block
967 */
968 20 int WritePidFile(const std::string &path) {
969
1/2
✓ Branch 2 taken 20 times.
✗ Branch 3 not taken.
 20 const int fd = open(path.c_str(), O_CREAT | O_RDWR, 0600);
970
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 15 times.
 20 if (fd < 0)
971 5 return -1;
972
2/2
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 10 times.
 15 if (flock(fd, LOCK_EX | LOCK_NB) != 0) {
973
1/2
✓ Branch 1 taken 5 times.
✗ Branch 2 not taken.
 5 close(fd);
974
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
 5 if (errno != EWOULDBLOCK)
975 ✗ return -1;
976 5 return -2;
977 }
978
979 // Don't leak the file descriptor to exec'd children
980
1/2
✓ Branch 1 taken 10 times.
✗ Branch 2 not taken.
 10 int flags = fcntl(fd, F_GETFD);
981
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 assert(flags != -1);
982 10 flags |= FD_CLOEXEC;
983
1/2
✓ Branch 1 taken 10 times.
✗ Branch 2 not taken.
 10 flags = fcntl(fd, F_SETFD, flags);
984
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 assert(flags != -1);
985
986 char buf[64];
987
988 10 snprintf(buf, sizeof(buf), "%" PRId64 "\n", static_cast<uint64_t>(getpid()));
989 10 const bool retval = (ftruncate(fd, 0) == 0)
990
3/6
✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 10 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 10 times.
✗ Branch 6 not taken.
 10 && SafeWrite(fd, buf, strlen(buf));
991
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 10 times.
 10 if (!retval) {
992 ✗ UnlockFile(fd);
993 ✗ return -1;
994 }
995 10 return fd;
996 }
997
998
999 /**
1000 * Locks file path, blocks if file is already locked. Creates path if required.
1001 *
1002 * \return file descriptor, -1 on error
1003 */
1004 4688 int LockFile(const std::string &path) {
1005 4688 const int fd_lockfile = open(path.c_str(), O_RDONLY | O_CREAT, 0600);
1006
2/2
✓ Branch 0 taken 105 times.
✓ Branch 1 taken 4583 times.
 4688 if (fd_lockfile < 0)
1007 105 return -1;
1008
1009
1010
2/2
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 4569 times.
 4583 if (flock(fd_lockfile, LOCK_EX | LOCK_NB) != 0) {
1011
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
 14 if (errno != EWOULDBLOCK) {
1012 ✗ close(fd_lockfile);
1013 ✗ return -1;
1014 }
1015 14 LogCvmfs(kLogCvmfs, kLogSyslog, "another process holds %s, waiting.",
1016 path.c_str());
1017
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 14 times.
 14 if (flock(fd_lockfile, LOCK_EX) != 0) {
1018 ✗ close(fd_lockfile);
1019 ✗ return -1;
1020 }
1021 14 LogCvmfs(kLogCvmfs, kLogSyslog, "lock %s acquired", path.c_str());
1022 }
1023
1024 4583 return fd_lockfile;
1025 }
1026
1027
1028 7937 void UnlockFile(const int filedes) {
1029 7937 const int retval = flock(filedes, LOCK_UN);
1030
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7937 times.
 7937 assert(retval == 0);
1031 7937 close(filedes);
1032 7937 }
1033
1034
1035 /**
1036 * Wrapper around mkstemp.
1037 */
1038 67198 FILE *CreateTempFile(const std::string &path_prefix, const int mode,
1039 const char *open_flags, std::string *final_path) {
1040 67198 *final_path = path_prefix + ".XXXXXX";
1041 67198 char *tmp_file = strdupa(final_path->c_str());
1042 67198 const int tmp_fd = mkstemp(tmp_file);
1043
2/2
✓ Branch 0 taken 65 times.
✓ Branch 1 taken 67133 times.
 67198 if (tmp_fd < 0) {
1044 65 return NULL;
1045 }
1046
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 67133 times.
 67133 if (fchmod(tmp_fd, mode) != 0) {
1047 ✗ close(tmp_fd);
1048 ✗ return NULL;
1049 }
1050
1051 67133 *final_path = tmp_file;
1052 67133 FILE *tmp_fp = fdopen(tmp_fd, open_flags);
1053
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 67133 times.
 67133 if (!tmp_fp) {
1054 ✗ close(tmp_fd);
1055 ✗ unlink(tmp_file);
1056 ✗ return NULL;
1057 }
1058
1059 67133 return tmp_fp;
1060 }
1061
1062
1063 /**
1064 * Create the file but don't open. Use only in non-public tmp directories.
1065 */
1066 18604 std::string CreateTempPath(const std::string &path_prefix, const int mode) {
1067 18604 std::string result;
1068
1/2
✓ Branch 1 taken 18604 times.
✗ Branch 2 not taken.
 18604 FILE *f = CreateTempFile(path_prefix, mode, "w", &result);
1069
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 18589 times.
 18604 if (!f)
1070
1/2
✓ Branch 2 taken 15 times.
✗ Branch 3 not taken.
 15 return "";
1071
1/2
✓ Branch 1 taken 18589 times.
✗ Branch 2 not taken.
 18589 fclose(f);
1072 18589 return result;
1073 18604 }
1074
1075
1076 /**
1077 * Create a directory with a unique name.
1078 */
1079 7151 std::string CreateTempDir(const std::string &path_prefix) {
1080
1/2
✓ Branch 1 taken 7151 times.
✗ Branch 2 not taken.
 7151 const std::string dir = path_prefix + ".XXXXXX";
1081 7151 char *tmp_dir = strdupa(dir.c_str());
1082 7151 tmp_dir = mkdtemp(tmp_dir);
1083
2/2
✓ Branch 0 taken 11 times.
✓ Branch 1 taken 7140 times.
 7151 if (tmp_dir == NULL)
1084
1/2
✓ Branch 2 taken 11 times.
✗ Branch 3 not taken.
 11 return "";
1085
1/2
✓ Branch 2 taken 7140 times.
✗ Branch 3 not taken.
 7140 return std::string(tmp_dir);
1086 7151 }
1087
1088
1089 /**
1090 * Get the current working directory of the running process
1091 */
1092 4985 std::string GetCurrentWorkingDirectory() {
1093 char cwd[PATH_MAX];
1094
3/9
✓ Branch 1 taken 4985 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 4985 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 4985 times.
✗ Branch 9 not taken.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
 4985 return (getcwd(cwd, sizeof(cwd)) != NULL) ? std::string(cwd) : std::string();
1095 }
1096
1097
1098 /**
1099 * Helper class that provides callback functions for the file system traversal.
1100 */
1101 class RemoveTreeHelper {
1102 public:
1103 bool success;
1104 15970 RemoveTreeHelper() { success = true; }
1105 99273 void RemoveFile(const std::string &parent_path, const std::string &name) {
1106
1/2
✓ Branch 2 taken 99273 times.
✗ Branch 3 not taken.
 99273 const int retval = unlink((parent_path + "/" + name).c_str());
1107
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 99273 times.
 99273 if (retval != 0)
1108 ✗ success = false;
1109 99273 }
1110 45636 void RemoveDir(const std::string &parent_path, const std::string &name) {
1111
1/2
✓ Branch 2 taken 45636 times.
✗ Branch 3 not taken.
 45636 const int retval = rmdir((parent_path + "/" + name).c_str());
1112
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 45636 times.
 45636 if (retval != 0)
1113 ✗ success = false;
1114 45636 }
1115 2041594 bool TryRemoveDir(const std::string &parent_path, const std::string &name) {
1116
1/2
✓ Branch 2 taken 2041594 times.
✗ Branch 3 not taken.
 2041594 const int retval = rmdir((parent_path + "/" + name).c_str());
1117 2041594 return (retval != 0);
1118 }
1119 };
1120
1121
1122 /**
1123 * Does rm -rf on path.
1124 */
1125 17276 bool RemoveTree(const std::string &path) {
1126 platform_stat64 info;
1127 17276 const int retval = platform_lstat(path.c_str(), &info);
1128
2/2
✓ Branch 0 taken 1306 times.
✓ Branch 1 taken 15970 times.
 17276 if (retval != 0)
1129 1306 return errno == ENOENT;
1130
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15970 times.
 15970 if (!S_ISDIR(info.st_mode))
1131 ✗ return false;
1132
1133
1/2
✓ Branch 1 taken 15970 times.
✗ Branch 2 not taken.
 15970 RemoveTreeHelper *remove_tree_helper = new RemoveTreeHelper();
1134
2/4
✓ Branch 2 taken 15970 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 15970 times.
✗ Branch 6 not taken.
 31940 FileSystemTraversal<RemoveTreeHelper> traversal(remove_tree_helper, "", true);
1135 15970 traversal.fn_new_file = &RemoveTreeHelper::RemoveFile;
1136 15970 traversal.fn_new_character_dev = &RemoveTreeHelper::RemoveFile;
1137 15970 traversal.fn_new_symlink = &RemoveTreeHelper::RemoveFile;
1138 15970 traversal.fn_new_socket = &RemoveTreeHelper::RemoveFile;
1139 15970 traversal.fn_new_fifo = &RemoveTreeHelper::RemoveFile;
1140 15970 traversal.fn_leave_dir = &RemoveTreeHelper::RemoveDir;
1141 15970 traversal.fn_new_dir_prefix = &RemoveTreeHelper::TryRemoveDir;
1142
1/2
✓ Branch 1 taken 15970 times.
✗ Branch 2 not taken.
 15970 traversal.Recurse(path);
1143 15970 const bool result = remove_tree_helper->success;
1144
1/2
✓ Branch 0 taken 15970 times.
✗ Branch 1 not taken.
 15970 delete remove_tree_helper;
1145
1146 15970 return result;
1147 15970 }
1148
1149
1150 /**
1151 * Returns ls $dir/GLOB$suffix
1152 */
1153 599 std::vector<std::string> FindFilesBySuffix(const std::string &dir,
1154 const std::string &suffix) {
1155 599 std::vector<std::string> result;
1156
1/2
✓ Branch 2 taken 599 times.
✗ Branch 3 not taken.
 599 DIR *dirp = opendir(dir.c_str());
1157
2/2
✓ Branch 0 taken 502 times.
✓ Branch 1 taken 97 times.
 599 if (!dirp)
1158 502 return result;
1159
1160 platform_dirent64 *dirent;
1161
3/4
✓ Branch 1 taken 499 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 402 times.
✓ Branch 4 taken 97 times.
 499 while ((dirent = platform_readdir(dirp))) {
1162
1/2
✓ Branch 2 taken 402 times.
✗ Branch 3 not taken.
 402 const std::string name(dirent->d_name);
1163 402 if ((name.length() >= suffix.length())
1164
9/14
✓ Branch 1 taken 324 times.
✓ Branch 2 taken 78 times.
✓ Branch 6 taken 324 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✓ Branch 9 taken 207 times.
✓ Branch 10 taken 117 times.
✓ Branch 11 taken 324 times.
✓ Branch 12 taken 78 times.
✗ Branch 13 not taken.
✓ Branch 14 taken 207 times.
✓ Branch 15 taken 195 times.
✗ Branch 16 not taken.
✗ Branch 17 not taken.
 804 && (name.substr(name.length() - suffix.length()) == suffix)) {
1165
3/6
✓ Branch 1 taken 207 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 207 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 207 times.
✗ Branch 8 not taken.
 207 result.push_back(dir + "/" + name);
1166 }
1167 402 }
1168
1/2
✓ Branch 1 taken 97 times.
✗ Branch 2 not taken.
 97 closedir(dirp);
1169
1/2
✓ Branch 3 taken 97 times.
✗ Branch 4 not taken.
 97 std::sort(result.begin(), result.end());
1170 97 return result;
1171 }
1172
1173
1174 /**
1175 * Returns ls $dir/$prefixGLOB
1176 */
1177 65 std::vector<std::string> FindFilesByPrefix(const std::string &dir,
1178 const std::string &prefix) {
1179 65 std::vector<std::string> result;
1180
1/2
✓ Branch 2 taken 65 times.
✗ Branch 3 not taken.
 65 DIR *dirp = opendir(dir.c_str());
1181
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 52 times.
 65 if (!dirp)
1182 13 return result;
1183
1184 platform_dirent64 *dirent;
1185
3/4
✓ Branch 1 taken 364 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 312 times.
✓ Branch 4 taken 52 times.
 364 while ((dirent = platform_readdir(dirp))) {
1186
1/2
✓ Branch 2 taken 312 times.
✗ Branch 3 not taken.
 312 const std::string name(dirent->d_name);
1187 312 if ((name.length() >= prefix.length())
1188
9/14
✓ Branch 1 taken 260 times.
✓ Branch 2 taken 52 times.
✓ Branch 5 taken 260 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 143 times.
✓ Branch 9 taken 117 times.
✓ Branch 10 taken 260 times.
✓ Branch 11 taken 52 times.
✗ Branch 12 not taken.
✓ Branch 13 taken 143 times.
✓ Branch 14 taken 169 times.
✗ Branch 15 not taken.
✗ Branch 16 not taken.
 624 && (name.substr(0, prefix.length()) == prefix)) {
1189
3/6
✓ Branch 1 taken 143 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 143 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 143 times.
✗ Branch 8 not taken.
 143 result.push_back(dir + "/" + name);
1190 }
1191 312 }
1192
1/2
✓ Branch 1 taken 52 times.
✗ Branch 2 not taken.
 52 closedir(dirp);
1193
1/2
✓ Branch 3 taken 52 times.
✗ Branch 4 not taken.
 52 std::sort(result.begin(), result.end());
1194 52 return result;
1195 }
1196
1197
1198 /**
1199 * Finds all direct subdirectories under parent_dir (except ., ..). Used,
1200 * for instance, to parse /etc/cvmfs/repositories.d/<reponoame>
1201 */
1202 4 std::vector<std::string> FindDirectories(const std::string &parent_dir) {
1203 4 std::vector<std::string> result;
1204
1/2
✓ Branch 2 taken 4 times.
✗ Branch 3 not taken.
 4 DIR *dirp = opendir(parent_dir.c_str());
1205
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (!dirp)
1206 ✗ return result;
1207
1208 platform_dirent64 *dirent;
1209
3/4
✓ Branch 1 taken 21 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 17 times.
✓ Branch 4 taken 4 times.
 21 while ((dirent = platform_readdir(dirp))) {
1210
1/2
✓ Branch 2 taken 17 times.
✗ Branch 3 not taken.
 17 const std::string name(dirent->d_name);
1211
6/6
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 4 times.
✓ Branch 4 taken 4 times.
✓ Branch 5 taken 9 times.
✓ Branch 6 taken 8 times.
✓ Branch 7 taken 9 times.
 17 if ((name == ".") || (name == ".."))
1212 8 continue;
1213
2/4
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 9 times.
✗ Branch 5 not taken.
 9 const std::string path = parent_dir + "/" + name;
1214
1215 platform_stat64 info;
1216 9 const int retval = platform_stat(path.c_str(), &info);
1217
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
 9 if (retval != 0)
1218 ✗ continue;
1219
2/2
✓ Branch 0 taken 7 times.
✓ Branch 1 taken 2 times.
 9 if (S_ISDIR(info.st_mode))
1220
1/2
✓ Branch 1 taken 7 times.
✗ Branch 2 not taken.
 7 result.push_back(path);
1221
3/4
✓ Branch 1 taken 9 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 9 times.
✓ Branch 5 taken 8 times.
 17 }
1222
1/2
✓ Branch 1 taken 4 times.
✗ Branch 2 not taken.
 4 closedir(dirp);
1223
1/2
✓ Branch 3 taken 4 times.
✗ Branch 4 not taken.
 4 sort(result.begin(), result.end());
1224 4 return result;
1225 }
1226
1227
1228 /**
1229 * Finds all files and direct subdirectories under directory (except ., ..).
1230 */
1231 130 bool ListDirectory(const std::string &directory,
1232 std::vector<std::string> *names,
1233 std::vector<mode_t> *modes) {
1234 130 DIR *dirp = opendir(directory.c_str());
1235
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 104 times.
 130 if (!dirp)
1236 26 return false;
1237
1238 platform_dirent64 *dirent;
1239
2/2
✓ Branch 1 taken 1599 times.
✓ Branch 2 taken 104 times.
 1703 while ((dirent = platform_readdir(dirp))) {
1240
1/2
✓ Branch 2 taken 1599 times.
✗ Branch 3 not taken.
 1599 const std::string name(dirent->d_name);
1241
6/6
✓ Branch 1 taken 1495 times.
✓ Branch 2 taken 104 times.
✓ Branch 4 taken 104 times.
✓ Branch 5 taken 1391 times.
✓ Branch 6 taken 208 times.
✓ Branch 7 taken 1391 times.
 1599 if ((name == ".") || (name == ".."))
1242 208 continue;
1243
2/4
✓ Branch 1 taken 1391 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1391 times.
✗ Branch 5 not taken.
 1391 const std::string path = directory + "/" + name;
1244
1245 platform_stat64 info;
1246 1391 const int retval = platform_lstat(path.c_str(), &info);
1247
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1391 times.
 1391 if (retval != 0) {
1248 ✗ closedir(dirp);
1249 ✗ return false;
1250 }
1251
1252
1/2
✓ Branch 1 taken 1391 times.
✗ Branch 2 not taken.
 1391 names->push_back(name);
1253
1/2
✓ Branch 1 taken 1391 times.
✗ Branch 2 not taken.
 1391 modes->push_back(info.st_mode);
1254
3/5
✓ Branch 1 taken 1391 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1391 times.
✓ Branch 5 taken 208 times.
✗ Branch 6 not taken.
 1599 }
1255 104 closedir(dirp);
1256
1257 104 SortTeam(names, modes);
1258 104 return true;
1259 }
1260
1261
1262 /**
1263 * Looks whether exe is an executable file. If exe is not an absolute path,
1264 * searches the PATH environment.
1265 */
1266 33 std::string FindExecutable(const std::string &exe) {
1267
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 33 times.
 33 if (exe.empty())
1268 ✗ return "";
1269
1270 33 std::vector<std::string> search_paths;
1271
2/2
✓ Branch 1 taken 11 times.
✓ Branch 2 taken 22 times.
 33 if (exe[0] == '/') {
1272
2/4
✓ Branch 1 taken 11 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 11 times.
✗ Branch 5 not taken.
 11 search_paths.push_back(GetParentPath(exe));
1273 } else {
1274 22 char *path_env = getenv("PATH");
1275
1/2
✓ Branch 0 taken 22 times.
✗ Branch 1 not taken.
 22 if (path_env) {
1276
2/4
✓ Branch 2 taken 22 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 22 times.
✗ Branch 6 not taken.
 22 search_paths = SplitString(path_env, ':');
1277 }
1278 }
1279
1280
2/2
✓ Branch 1 taken 121 times.
✓ Branch 2 taken 11 times.
 132 for (unsigned i = 0; i < search_paths.size(); ++i) {
1281
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 121 times.
 121 if (search_paths[i].empty())
1282 99 continue;
1283
2/4
✓ Branch 2 taken 121 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 121 times.
 121 if (search_paths[i][0] != '/')
1284 ✗ continue;
1285
1286
3/6
✓ Branch 1 taken 121 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 121 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 121 times.
✗ Branch 9 not taken.
 242 std::string path = search_paths[i] + "/" + GetFileName(exe);
1287 platform_stat64 info;
1288 121 int retval = platform_stat(path.c_str(), &info);
1289
2/2
✓ Branch 0 taken 99 times.
✓ Branch 1 taken 22 times.
 121 if (retval != 0)
1290 99 continue;
1291
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22 times.
 22 if (!S_ISREG(info.st_mode))
1292 ✗ continue;
1293 22 retval = access(path.c_str(), X_OK);
1294
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 22 times.
 22 if (retval != 0)
1295 ✗ continue;
1296
1297 22 return path;
1298
2/2
✓ Branch 1 taken 99 times.
✓ Branch 2 taken 22 times.
 121 }
1299
1300
1/2
✓ Branch 2 taken 11 times.
✗ Branch 3 not taken.
 11 return "";
1301 33 }
1302
1303
1304 1 std::string GetUserName() {
1305 struct passwd pwd;
1306 1 struct passwd *result = NULL;
1307 1 int bufsize = 16 * 1024;
1308 1 char *buf = static_cast<char *>(smalloc(bufsize));
1309
2/4
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 1 times.
 1 while (getpwuid_r(geteuid(), &pwd, buf, bufsize, &result) == ERANGE) {
1310 ✗ bufsize *= 2;
1311 ✗ buf = static_cast<char *>(srealloc(buf, bufsize));
1312 }
1313
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 if (result == NULL) {
1314 ✗ free(buf);
1315 ✗ return "";
1316 }
1317
1/2
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
 1 std::string user_name = pwd.pw_name;
1318 1 free(buf);
1319 1 return user_name;
1320 1 }
1321
1322 ✗ std::string GetShell() {
1323 struct passwd pwd;
1324 ✗ struct passwd *result = NULL;
1325 ✗ int bufsize = 16 * 1024;
1326 ✗ char *buf = static_cast<char *>(smalloc(bufsize));
1327 ✗ while (getpwuid_r(geteuid(), &pwd, buf, bufsize, &result) == ERANGE) {
1328 ✗ bufsize *= 2;
1329 ✗ buf = static_cast<char *>(srealloc(buf, bufsize));
1330 }
1331 ✗ if (result == NULL) {
1332 ✗ free(buf);
1333 ✗ return "";
1334 }
1335 ✗ std::string shell = pwd.pw_shell;
1336 ✗ free(buf);
1337 ✗ return shell;
1338 }
1339
1340 /**
1341 * UID -> Name from passwd database
1342 */
1343 18 bool GetUserNameOf(uid_t uid, std::string *username) {
1344 struct passwd pwd;
1345 18 struct passwd *result = NULL;
1346 18 int bufsize = 16 * 1024;
1347 18 char *buf = static_cast<char *>(smalloc(bufsize));
1348
2/4
✓ Branch 1 taken 18 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 18 times.
 18 while (getpwuid_r(uid, &pwd, buf, bufsize, &result) == ERANGE) {
1349 ✗ bufsize *= 2;
1350 ✗ buf = static_cast<char *>(srealloc(buf, bufsize));
1351 }
1352
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 if (result == NULL) {
1353 ✗ free(buf);
1354 ✗ return false;
1355 }
1356
1/2
✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
 18 if (username)
1357
1/2
✓ Branch 1 taken 18 times.
✗ Branch 2 not taken.
 18 *username = result->pw_name;
1358 18 free(buf);
1359 18 return true;
1360 }
1361
1362
1363 /**
1364 * Name -> UID from passwd database
1365 */
1366 263 bool GetUidOf(const std::string &username, uid_t *uid, gid_t *main_gid) {
1367 struct passwd pwd;
1368 263 struct passwd *result = NULL;
1369 263 int bufsize = 16 * 1024;
1370 263 char *buf = static_cast<char *>(smalloc(bufsize));
1371
2/4
✓ Branch 2 taken 263 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 263 times.
 263 while (getpwnam_r(username.c_str(), &pwd, buf, bufsize, &result) == ERANGE) {
1372 ✗ bufsize *= 2;
1373 ✗ buf = static_cast<char *>(srealloc(buf, bufsize));
1374 }
1375
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 254 times.
 263 if (result == NULL) {
1376 9 free(buf);
1377 9 return false;
1378 }
1379 254 *uid = result->pw_uid;
1380 254 *main_gid = result->pw_gid;
1381 254 free(buf);
1382 254 return true;
1383 }
1384
1385
1386 /**
1387 * Name -> GID from groups database
1388 */
1389 18 bool GetGidOf(const std::string &groupname, gid_t *gid) {
1390 struct group grp;
1391 18 struct group *result = NULL;
1392 18 int bufsize = 16 * 1024;
1393 18 char *buf = static_cast<char *>(smalloc(bufsize));
1394
2/4
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 18 times.
 18 while (getgrnam_r(groupname.c_str(), &grp, buf, bufsize, &result) == ERANGE) {
1395 ✗ bufsize *= 2;
1396 ✗ buf = static_cast<char *>(srealloc(buf, bufsize));
1397 }
1398
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 9 times.
 18 if (result == NULL) {
1399 9 free(buf);
1400 9 return false;
1401 }
1402 9 *gid = result->gr_gid;
1403 9 free(buf);
1404 9 return true;
1405 }
1406
1407 /**
1408 * read the current umask of this process
1409 * Note: umask query is guarded by a global mutex. Hence, always use
1410 * this function and beware of scalability bottlenecks
1411 */
1412 3361 mode_t GetUmask() {
1413 3361 const MutexLockGuard m(&getumask_mutex);
1414 3361 const mode_t my_umask = umask(0);
1415 3361 umask(my_umask);
1416 6722 return my_umask;
1417 3361 }
1418
1419
1420 /**
1421 * Adds gid to the list of supplementary groups
1422 */
1423 ✗ bool AddGroup2Persona(const gid_t gid) {
1424 ✗ const int ngroups = getgroups(0, NULL);
1425 ✗ if (ngroups < 0)
1426 ✗ return false;
1427 ✗ gid_t *groups = static_cast<gid_t *>(smalloc((ngroups + 1) * sizeof(gid_t)));
1428 ✗ int retval = getgroups(ngroups, groups);
1429 ✗ if (retval < 0) {
1430 ✗ free(groups);
1431 ✗ return false;
1432 }
1433 ✗ for (int i = 0; i < ngroups; ++i) {
1434 ✗ if (groups[i] == gid) {
1435 ✗ free(groups);
1436 ✗ return true;
1437 }
1438 }
1439 ✗ groups[ngroups] = gid;
1440 ✗ retval = setgroups(ngroups + 1, groups);
1441 ✗ free(groups);
1442 ✗ return retval == 0;
1443 }
1444
1445
1446 26 std::string GetHomeDirectory() {
1447 26 const uid_t uid = getuid();
1448 struct passwd pwd;
1449 26 struct passwd *result = NULL;
1450 26 int bufsize = 16 * 1024;
1451 26 char *buf = static_cast<char *>(smalloc(bufsize));
1452
2/4
✓ Branch 1 taken 26 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 26 times.
 26 while (getpwuid_r(uid, &pwd, buf, bufsize, &result) == ERANGE) {
1453 ✗ bufsize *= 2;
1454 ✗ buf = static_cast<char *>(srealloc(buf, bufsize));
1455 }
1456
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 26 times.
 26 if (result == NULL) {
1457 ✗ free(buf);
1458 ✗ return "";
1459 }
1460
1/2
✓ Branch 2 taken 26 times.
✗ Branch 3 not taken.
 26 std::string home_dir = result->pw_dir;
1461 26 free(buf);
1462 26 return home_dir;
1463 26 }
1464
1465 /**
1466 * Returns the output of `uname -m`
1467 */
1468 1 std::string GetArch() {
1469 struct utsname info;
1470 1 const int retval = uname(&info);
1471
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 assert(retval == 0);
1472
1/2
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
 1 return info.machine;
1473 }
1474
1475
1476 /**
1477 * Sets soft and hard limit for maximum number of open file descriptors.
1478 * Returns 0 on success, -1 on failure, -2 if running under valgrind.
1479 */
1480 183 int SetLimitNoFile(unsigned limit_nofile) {
1481 struct rlimit rpl;
1482 183 memset(&rpl, 0, sizeof(rpl));
1483 183 getrlimit(RLIMIT_NOFILE, &rpl);
1484
2/2
✓ Branch 0 taken 123 times.
✓ Branch 1 taken 60 times.
 183 if (rpl.rlim_max < limit_nofile)
1485 123 rpl.rlim_max = limit_nofile;
1486 183 rpl.rlim_cur = limit_nofile;
1487 183 const int retval = setrlimit(RLIMIT_NOFILE, &rpl);
1488
2/2
✓ Branch 0 taken 60 times.
✓ Branch 1 taken 123 times.
 183 if (retval == 0)
1489 60 return 0;
1490
1491 #ifdef HAS_VALGRIND_HEADERS
1492
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 123 times.
 123 return RUNNING_ON_VALGRIND ? -2 : -1;
1493 #else
1494 return -1;
1495 #endif
1496 }
1497
1498
1499 /**
1500 * Get the file descriptor limits
1501 */
1502 836 void GetLimitNoFile(unsigned *soft_limit, unsigned *hard_limit) {
1503 836 *soft_limit = 0;
1504 836 *hard_limit = 0;
1505
1506 struct rlimit rpl;
1507 836 memset(&rpl, 0, sizeof(rpl));
1508 836 getrlimit(RLIMIT_NOFILE, &rpl);
1509 836 *soft_limit = rpl.rlim_cur;
1510
1511 #ifdef __APPLE__
1512 int value = sysconf(_SC_OPEN_MAX);
1513 assert(value > 0);
1514 *hard_limit = value;
1515 #else
1516 836 *hard_limit = rpl.rlim_max;
1517 #endif
1518 836 }
1519
1520
1521 /**
1522 * Sets soft and hard limit for maximum core size
1523 */
1524 ✗ bool SetLimitCore(unsigned limit_core) {
1525 struct rlimit rpl;
1526 ✗ memset(&rpl, 0, sizeof(rpl));
1527 ✗ rpl.rlim_max = limit_core;
1528 ✗ rpl.rlim_cur = limit_core;
1529 ✗ return (setrlimit(RLIMIT_CORE, &rpl) == 0);
1530 }
1531
1532
1533 13 std::vector<LsofEntry> Lsof(const std::string &path) {
1534 13 std::vector<LsofEntry> result;
1535
1536 13 std::vector<std::string> proc_names;
1537 13 std::vector<mode_t> proc_modes;
1538
2/4
✓ Branch 2 taken 13 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 13 times.
✗ Branch 6 not taken.
 13 ListDirectory("/proc", &proc_names, &proc_modes);
1539
1540
2/2
✓ Branch 1 taken 845 times.
✓ Branch 2 taken 13 times.
 858 for (unsigned i = 0; i < proc_names.size(); ++i) {
1541
2/2
✓ Branch 1 taken 637 times.
✓ Branch 2 taken 208 times.
 845 if (!S_ISDIR(proc_modes[i]))
1542 767 continue;
1543
2/2
✓ Branch 2 taken 130 times.
✓ Branch 3 taken 78 times.
 208 if (proc_names[i].find_first_not_of("1234567890") != std::string::npos)
1544 130 continue;
1545
1546 78 std::vector<std::string> fd_names;
1547 78 std::vector<mode_t> fd_modes;
1548
1/2
✓ Branch 2 taken 78 times.
✗ Branch 3 not taken.
 78 const std::string proc_dir = "/proc/" + proc_names[i];
1549
1/2
✓ Branch 1 taken 78 times.
✗ Branch 2 not taken.
 78 const std::string fd_dir = proc_dir + "/fd";
1550
1/2
✓ Branch 1 taken 78 times.
✗ Branch 2 not taken.
 78 const bool rvb = ListDirectory(fd_dir, &fd_names, &fd_modes);
1551 78 uid_t proc_uid = 0;
1552
1553 // The working directory of the process requires special handling
1554
2/2
✓ Branch 0 taken 65 times.
✓ Branch 1 taken 13 times.
 78 if (rvb) {
1555 platform_stat64 info;
1556 65 platform_stat(proc_dir.c_str(), &info);
1557 65 proc_uid = info.st_uid;
1558
1559
2/4
✓ Branch 1 taken 65 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 65 times.
✗ Branch 5 not taken.
 65 const std::string cwd = ReadSymlink(proc_dir + "/cwd");
1560
5/9
✓ Branch 1 taken 65 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 65 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 65 times.
✗ Branch 8 not taken.
✓ Branch 11 taken 13 times.
✓ Branch 12 taken 52 times.
 65 if (HasPrefix(cwd + "/", path + "/", false /* ignore_case */)) {
1561 13 LsofEntry entry;
1562
1/2
✓ Branch 2 taken 13 times.
✗ Branch 3 not taken.
 13 entry.pid = static_cast<pid_t>(String2Uint64(proc_names[i]));
1563 13 entry.owner = proc_uid;
1564 13 entry.read_only = true; // A bit sloppy but good enough for the moment
1565
2/4
✓ Branch 1 taken 13 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 13 times.
✗ Branch 5 not taken.
 13 entry.executable = ReadSymlink(proc_dir + "/exe");
1566
1/2
✓ Branch 1 taken 13 times.
✗ Branch 2 not taken.
 13 entry.path = cwd;
1567
1/2
✓ Branch 1 taken 13 times.
✗ Branch 2 not taken.
 13 result.push_back(entry);
1568 13 }
1569 65 }
1570
1571
2/2
✓ Branch 1 taken 494 times.
✓ Branch 2 taken 78 times.
 572 for (unsigned j = 0; j < fd_names.size(); ++j) {
1572
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 494 times.
 494 if (!S_ISLNK(fd_modes[j]))
1573 351 continue;
1574
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 494 times.
 494 if (fd_names[j].find_first_not_of("1234567890") != std::string::npos)
1575 ✗ continue;
1576
1577
3/6
✓ Branch 2 taken 494 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 494 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 494 times.
✗ Branch 9 not taken.
 988 const std::string target = ReadSymlink(fd_dir + "/" + fd_names[j]);
1578
5/9
✓ Branch 1 taken 494 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 494 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 494 times.
✗ Branch 8 not taken.
✓ Branch 11 taken 351 times.
✓ Branch 12 taken 143 times.
 494 if (!HasPrefix(target + "/", path + "/", false /* ignore_case */))
1579 351 continue;
1580
1581 143 LsofEntry entry;
1582
1/2
✓ Branch 2 taken 143 times.
✗ Branch 3 not taken.
 143 entry.pid = static_cast<pid_t>(String2Uint64(proc_names[i]));
1583 143 entry.owner = proc_uid;
1584 143 entry.read_only = !((fd_modes[j] & S_IWUSR) == S_IWUSR);
1585
2/4
✓ Branch 1 taken 143 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 143 times.
✗ Branch 5 not taken.
 143 entry.executable = ReadSymlink(proc_dir + "/exe");
1586
1/2
✓ Branch 1 taken 143 times.
✗ Branch 2 not taken.
 143 entry.path = target;
1587
1/2
✓ Branch 1 taken 143 times.
✗ Branch 2 not taken.
 143 result.push_back(entry);
1588
2/2
✓ Branch 2 taken 143 times.
✓ Branch 3 taken 351 times.
 494 }
1589 78 }
1590
1591 26 return result;
1592 13 }
1593
1594
1595 39 bool ProcessExists(pid_t pid) {
1596
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 39 times.
 39 assert(pid > 0);
1597 39 const int retval = kill(pid, 0);
1598
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 13 times.
 39 if (retval == 0)
1599 26 return true;
1600 13 return (errno != ESRCH);
1601 }
1602
1603
1604 /**
1605 * Blocks a signal for the calling thread.
1606 */
1607 12 void BlockSignal(int signum) {
1608 sigset_t sigset;
1609 12 int retval = sigemptyset(&sigset);
1610
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
 12 assert(retval == 0);
1611 12 retval = sigaddset(&sigset, signum);
1612
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
 12 assert(retval == 0);
1613 12 retval = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
1614
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
 12 assert(retval == 0);
1615 12 }
1616
1617
1618 /**
1619 * Waits for a signal. The signal should be blocked before for all threads.
1620 * Threads inherit their parent's signal mask.
1621 */
1622 5 void WaitForSignal(int signum) {
1623 int retval;
1624 do {
1625 5 retval = platform_sigwait(signum);
1626
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 5 } while ((retval != signum) && (errno == EINTR));
1627
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
 5 assert(retval == signum);
1628 5 }
1629
1630
1631 /**
1632 * Returns -1 if the child crashed or the exit code otherwise.
1633 * @param pid Process identifier.
1634 * @param sig_ok List of signals that are still considered a successful
1635 * termination.
1636 */
1637 106 int WaitForChild(pid_t pid, const std::vector<int> &sig_ok) {
1638
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 106 times.
 106 assert(pid > 0);
1639 int statloc;
1640 while (true) {
1641
1/2
✓ Branch 1 taken 106 times.
✗ Branch 2 not taken.
 106 const pid_t retval = waitpid(pid, &statloc, 0);
1642
2/2
✓ Branch 0 taken 15 times.
✓ Branch 1 taken 91 times.
 106 if (retval == -1) {
1643
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
 15 if (errno == EINTR)
1644 ✗ continue;
1645 15 PANIC(kLogSyslogErr | kLogDebug, "waitpid failed with errno %d", errno);
1646 }
1647
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 91 times.
 91 assert(retval == pid);
1648 91 break;
1649 }
1650
2/2
✓ Branch 0 taken 76 times.
✓ Branch 1 taken 15 times.
 91 if (WIFEXITED(statloc))
1651 76 return WEXITSTATUS(statloc);
1652 30 if (WIFSIGNALED(statloc)
1653
3/6
✓ Branch 0 taken 15 times.
✗ Branch 1 not taken.
✓ Branch 5 taken 15 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 15 times.
 45 && (std::find(sig_ok.begin(), sig_ok.end(), WTERMSIG(statloc))
1654
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 15 times.
 45 != sig_ok.end()))
1655 ✗ return 0;
1656 15 return -1;
1657 }
1658
1659 /**
1660 * Exec a command as a daemon.
1661 */
1662
1663 ✗ bool ExecAsDaemon(const std::vector<std::string> &command_line,
1664 pid_t *child_pid) {
1665 ✗ assert(command_line.size() >= 1);
1666
1667 ✗ Pipe<kPipeDetachedChild> pipe_fork;
1668 ✗ const pid_t pid = fork();
1669 ✗ assert(pid >= 0);
1670 ✗ if (pid == 0) {
1671 pid_t pid_grand_child;
1672
1673 ✗ const char *argv[command_line.size() + 1];
1674 ✗ for (unsigned i = 0; i < command_line.size(); ++i)
1675 ✗ argv[i] = command_line[i].c_str();
1676 ✗ argv[command_line.size()] = NULL;
1677 ✗ int retval = setsid();
1678 ✗ assert(retval != -1);
1679
1680 ✗ pid_grand_child = fork();
1681 ✗ assert(pid_grand_child >= 0);
1682
1683 ✗ if (pid_grand_child != 0) {
1684 ✗ pipe_fork.Write<pid_t>(pid_grand_child);
1685 ✗ _exit(0);
1686 } else {
1687 ✗ const int null_read = open("/dev/null", O_RDONLY);
1688 ✗ const int null_write = open("/dev/null", O_WRONLY);
1689 ✗ assert((null_read >= 0) && (null_write >= 0));
1690 ✗ retval = dup2(null_read, 0);
1691 ✗ assert(retval == 0);
1692 ✗ retval = dup2(null_write, 1);
1693 ✗ assert(retval == 1);
1694 ✗ retval = dup2(null_write, 2);
1695 ✗ assert(retval == 2);
1696 ✗ close(null_read);
1697 ✗ close(null_write);
1698
1699 ✗ execvp(command_line[0].c_str(), const_cast<char **>(argv));
1700
1701 ✗ pipe_fork.CloseWriteFd();
1702 }
1703 }
1704 int statloc;
1705 ✗ waitpid(pid, &statloc, 0);
1706 ✗ pid_t buf_child_pid = 0;
1707 ✗ pipe_fork.Read(&buf_child_pid);
1708 ✗ if (child_pid != NULL)
1709 ✗ *child_pid = buf_child_pid;
1710 ✗ pipe_fork.CloseReadFd();
1711
1712 ✗ LogCvmfs(kLogCvmfs, kLogDebug, "exec'd as daemon %s (PID: %d)",
1713 ✗ command_line[0].c_str(), static_cast<int>(*child_pid));
1714 ✗ return true;
1715 }
1716
1717
1718 /**
1719 * Makes a daemon. The daemon() call is deprecated on OS X
1720 */
1721 ✗ void Daemonize() {
1722 pid_t pid;
1723 int statloc;
1724 ✗ if ((pid = fork()) == 0) {
1725 ✗ int retval = setsid();
1726 ✗ assert(retval != -1);
1727 ✗ if ((pid = fork()) == 0) {
1728
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 const int null_read = open("/dev/null", O_RDONLY);
1729
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 const int null_write = open("/dev/null", O_WRONLY);
1730
2/4
✓ Branch 0 taken 49 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 49 times.
✗ Branch 3 not taken.
 49 assert((null_read >= 0) && (null_write >= 0));
1731 49 retval = dup2(null_read, 0);
1732
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
 49 assert(retval == 0);
1733 49 retval = dup2(null_write, 1);
1734
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
 49 assert(retval == 1);
1735 49 retval = dup2(null_write, 2);
1736
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 49 times.
 49 assert(retval == 2);
1737
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 close(null_read);
1738
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 close(null_write);
1739
1/2
✓ Branch 1 taken 49 times.
✗ Branch 2 not taken.
 49 LogCvmfs(kLogCvmfs, kLogDebug, "daemonized");
1740 } else {
1741 ✗ assert(pid > 0);
1742 ✗ _exit(0);
1743 }
1744 } else {
1745 ✗ assert(pid > 0);
1746 ✗ waitpid(pid, &statloc, 0);
1747 ✗ _exit(0);
1748 }
1749 49 }
1750
1751
1752 818 bool ExecuteBinary(int *fd_stdin,
1753 int *fd_stdout,
1754 int *fd_stderr,
1755 const std::string &binary_path,
1756 const std::vector<std::string> &argv,
1757 const bool double_fork,
1758 pid_t *child_pid) {
1759 int pipe_stdin[2];
1760 int pipe_stdout[2];
1761 int pipe_stderr[2];
1762
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 MakePipe(pipe_stdin);
1763
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 MakePipe(pipe_stdout);
1764
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 MakePipe(pipe_stderr);
1765
1766 818 std::set<int> preserve_fildes;
1767
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 preserve_fildes.insert(0);
1768
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 preserve_fildes.insert(1);
1769
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 preserve_fildes.insert(2);
1770 818 std::map<int, int> map_fildes;
1771
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 map_fildes[pipe_stdin[0]] = 0; // Reading end of pipe_stdin
1772
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 map_fildes[pipe_stdout[1]] = 1; // Writing end of pipe_stdout
1773
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 map_fildes[pipe_stderr[1]] = 2; // Writing end of pipe_stderr
1774 818 std::vector<std::string> cmd_line;
1775
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 cmd_line.push_back(binary_path);
1776
1/2
✓ Branch 5 taken 818 times.
✗ Branch 6 not taken.
 818 cmd_line.insert(cmd_line.end(), argv.begin(), argv.end());
1777
1778
2/4
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 818 times.
 818 if (!ManagedExec(cmd_line,
1779 preserve_fildes,
1780 map_fildes,
1781 true /* drop_credentials */,
1782 false /* clear_env */,
1783 double_fork,
1784 child_pid)) {
1785 ✗ ClosePipe(pipe_stdin);
1786 ✗ ClosePipe(pipe_stdout);
1787 ✗ ClosePipe(pipe_stderr);
1788 ✗ return false;
1789 }
1790
1791
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 close(pipe_stdin[0]);
1792
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 close(pipe_stdout[1]);
1793
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 close(pipe_stderr[1]);
1794 818 *fd_stdin = pipe_stdin[1];
1795 818 *fd_stdout = pipe_stdout[0];
1796 818 *fd_stderr = pipe_stderr[0];
1797 818 return true;
1798 818 }
1799
1800
1801 /**
1802 * Opens /bin/sh and provides file descriptors to write into stdin and
1803 * read from stdout. Quit shell simply by closing stderr, stdout, and stdin.
1804 */
1805 787 bool Shell(int *fd_stdin, int *fd_stdout, int *fd_stderr) {
1806 787 const bool double_fork = true;
1807
2/4
✓ Branch 2 taken 787 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 787 times.
✗ Branch 6 not taken.
 1574 return ExecuteBinary(fd_stdin, fd_stdout, fd_stderr, "/bin/sh",
1808 2361 std::vector<std::string>(), double_fork);
1809 }
1810
1811 struct ForkFailures { // TODO(rmeusel): C++11 (type safe enum)
1812 enum Names {
1813 kSendPid,
1814 kUnknown,
1815 kFailDupFd,
1816 kFailCloseFds,
1817 kFailGetFdFlags,
1818 kFailSetFdFlags,
1819 kFailDropCredentials,
1820 kFailExec,
1821 };
1822
1823 ✗ static std::string ToString(const Names name) {
1824 ✗ switch (name) {
1825 ✗ case kSendPid:
1826 ✗ return "Sending PID";
1827
1828 ✗ default:
1829 case kUnknown:
1830 ✗ return "Unknown Status";
1831 ✗ case kFailDupFd:
1832 ✗ return "Duplicate File Descriptor";
1833 ✗ case kFailCloseFds:
1834 ✗ return "Close File Descriptors";
1835 ✗ case kFailGetFdFlags:
1836 ✗ return "Read File Descriptor Flags";
1837 ✗ case kFailSetFdFlags:
1838 ✗ return "Set File Descriptor Flags";
1839 ✗ case kFailDropCredentials:
1840 ✗ return "Lower User Permissions";
1841 ✗ case kFailExec:
1842 ✗ return "Invoking execvp()";
1843 }
1844 }
1845 };
1846
1847 /**
1848 * Loop through all possible FDs and close them.
1849 */
1850 23 static bool CloseAllFildesUntilMaxFD(const std::set<int> &preserve_fildes,
1851 long max_fd) { // NOLINT(runtime/int)
1852
2/2
✓ Branch 0 taken 11500 times.
✓ Branch 1 taken 23 times.
 11523 for (long fd = 0; fd < max_fd; fd++) { // NOLINT(runtime/int)
1853
3/4
✓ Branch 1 taken 11500 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 11387 times.
✓ Branch 4 taken 113 times.
 11500 if (preserve_fildes.count(static_cast<int>(fd)) == 0) {
1854 11387 close(static_cast<int>(fd));
1855 }
1856 }
1857
1858 23 return true;
1859 }
1860
1861 /**
1862 * Enumerate currently open file descriptors via a procfs-like directory
1863 * (/proc/self/fd on Linux, /dev/fd on macOS) and close the ones not in
1864 * preserve_fildes.
1865 */
1866 28 static bool CloseAllFildesInFdDir(const char *fd_dir,
1867 const std::set<int> &preserve_fildes) {
1868 28 DIR *dirp = opendir(fd_dir);
1869
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
 28 if (!dirp)
1870 ✗ return false;
1871
1872 platform_dirent64 *dirent;
1873
1874
2/2
✓ Branch 1 taken 501 times.
✓ Branch 2 taken 28 times.
 529 while ((dirent = platform_readdir(dirp))) {
1875
1/2
✓ Branch 2 taken 501 times.
✗ Branch 3 not taken.
 501 const std::string name(dirent->d_name);
1876 uint64_t name_uint64;
1877
1878 // Make sure the dir name is digits only (skips ".", ".." and similar).
1879
3/4
✓ Branch 1 taken 501 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 56 times.
✓ Branch 4 taken 445 times.
 501 if (!String2Uint64Parse(name, &name_uint64)) {
1880 56 continue;
1881 }
1882
1883 445 const int fd = static_cast<int>(name_uint64);
1884
3/4
✓ Branch 1 taken 445 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 140 times.
✓ Branch 4 taken 305 times.
 445 if (preserve_fildes.count(fd)) {
1885 140 continue;
1886 }
1887
1888
1/2
✓ Branch 1 taken 305 times.
✗ Branch 2 not taken.
 305 close(fd);
1889
2/2
✓ Branch 1 taken 305 times.
✓ Branch 2 taken 196 times.
 501 }
1890
1891 28 closedir(dirp);
1892
1893 28 return true;
1894 }
1895
1896 /**
1897 * Closes all file descriptors except the ones in preserve_fildes.
1898 * To be used after fork but before exec.
1899 */
1900 51 bool CloseAllFildes(const std::set<int> &preserve_fildes) {
1901 51 const long max_fd = sysconf(_SC_OPEN_MAX); // NOLINT(runtime/int)
1902
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 51 times.
 51 if (max_fd < 0) {
1903 ✗ return false;
1904 }
1905
1906 // When RLIMIT_NOFILE is unbounded (sysconf returns LONG_MAX) or simply very
1907 // large, iterating [0, max_fd) is impractical. Enumerate the actually open
1908 // fds via the kernel's fd directory instead.
1909
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 23 times.
 51 if (max_fd > 100000) {
1910 #ifdef __APPLE__
1911 return CloseAllFildesInFdDir("/dev/fd", preserve_fildes);
1912 #else
1913 28 return CloseAllFildesInFdDir("/proc/self/fd", preserve_fildes);
1914 #endif
1915 }
1916
1917 23 return CloseAllFildesUntilMaxFD(preserve_fildes, max_fd);
1918 }
1919
1920 /**
1921 * Execve to the given command line, preserving the given file descriptors.
1922 * If stdin, stdout, stderr should be preserved, add 0, 1, 2.
1923 * File descriptors from the parent process can also be mapped to the new
1924 * process (dup2) using map_fildes. Can be useful for
1925 * stdout/in/err redirection.
1926 * NOTE: The destination fildes have to be preserved!
1927 * NOTE: For forked child, held mutexes have invalid state, avoid operations.
1928 * Does a double fork to detach child.
1929 * The command_line parameter contains the binary at index 0 and the arguments
1930 * in the rest of the vector.
1931 * Using the optional parameter *pid it is possible to retrieve the process ID
1932 * of the spawned process.
1933 */
1934 877 bool ManagedExec(const std::vector<std::string> &command_line,
1935 const std::set<int> &preserve_fildes,
1936 const std::map<int, int> &map_fildes,
1937 const bool drop_credentials,
1938 const bool clear_env,
1939 const bool double_fork,
1940 pid_t *child_pid) {
1941
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 877 times.
 877 assert(command_line.size() >= 1);
1942
1943
1/2
✓ Branch 1 taken 877 times.
✗ Branch 2 not taken.
 877 Pipe<kPipeDetachedChild> pipe_fork;
1944 877 const pid_t pid = fork();
1945
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 879 times.
 879 assert(pid >= 0);
1946
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 877 times.
 879 if (pid == 0) {
1947 pid_t pid_grand_child;
1948 int fd_flags;
1949 2 ForkFailures::Names failed = ForkFailures::kUnknown;
1950
1951 2 std::set<int> skip_fds = preserve_fildes;
1952
1/2
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
 2 skip_fds.insert(pipe_fork.GetWriteFd());
1953
1954
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (clear_env) {
1955 #ifdef __APPLE__
1956 environ = NULL;
1957 #else
1958 ✗ const int retval = clearenv();
1959 ✗ assert(retval == 0);
1960 #endif
1961 }
1962
1963 2 const char *argv[command_line.size() + 1];
1964
2/2
✓ Branch 1 taken 3 times.
✓ Branch 2 taken 2 times.
 5 for (unsigned i = 0; i < command_line.size(); ++i)
1965 3 argv[i] = command_line[i].c_str();
1966 2 argv[command_line.size()] = NULL;
1967
1968 // Child, map file descriptors
1969 4 for (std::map<int, int>::const_iterator i = map_fildes.begin(),
1970 2 iEnd = map_fildes.end();
1971
2/2
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 2 times.
 8 i != iEnd;
1972 6 ++i) {
1973 6 const int retval = dup2(i->first, i->second);
1974
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (retval == -1) {
1975 ✗ failed = ForkFailures::kFailDupFd;
1976 ✗ goto fork_failure;
1977 }
1978 }
1979
1980 // Child, close file descriptors
1981
2/4
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 2 times.
 2 if (!CloseAllFildes(skip_fds)) {
1982 ✗ failed = ForkFailures::kFailCloseFds;
1983 ✗ goto fork_failure;
1984 }
1985
1986 // Double fork to disconnect from parent
1987
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (double_fork) {
1988 ✗ pid_grand_child = fork();
1989
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
 28 assert(pid_grand_child >= 0);
1990
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
 28 if (pid_grand_child != 0)
1991 ✗ _exit(0);
1992 }
1993
1994
1/2
✓ Branch 2 taken 30 times.
✗ Branch 3 not taken.
 30 fd_flags = fcntl(pipe_fork.GetWriteFd(), F_GETFD);
1995
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 30 times.
 30 if (fd_flags < 0) {
1996 ✗ failed = ForkFailures::kFailGetFdFlags;
1997 ✗ goto fork_failure;
1998 }
1999 30 fd_flags |= FD_CLOEXEC;
2000
2/4
✓ Branch 2 taken 30 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 30 times.
 30 if (fcntl(pipe_fork.GetWriteFd(), F_SETFD, fd_flags) < 0) {
2001 ✗ failed = ForkFailures::kFailSetFdFlags;
2002 ✗ goto fork_failure;
2003 }
2004
2005 #ifdef DEBUGMSG
2006
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 30 times.
 30 assert(setenv("__CVMFS_DEBUG_MODE__", "yes", 1) == 0);
2007 #endif
2008
5/8
✓ Branch 0 taken 29 times.
✓ Branch 1 taken 1 times.
✓ Branch 3 taken 29 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✓ Branch 6 taken 29 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 30 times.
 30 if (drop_credentials && SetuidCapabilityPermitted() &&
2009 ✗ !SwitchCredentials(geteuid(),
2010 getegid(),
2011 false /* temporarily */,
2012 true /* avoid_mutexes */)) {
2013 ✗ failed = ForkFailures::kFailDropCredentials;
2014 ✗ goto fork_failure;
2015 }
2016
2017 // retrieve the PID of the new (grand) child process and send it to the
2018 // grand father
2019 30 pid_grand_child = getpid();
2020 30 failed = ForkFailures::kSendPid;
2021
1/2
✓ Branch 1 taken 30 times.
✗ Branch 2 not taken.
 30 pipe_fork.Write<ForkFailures::Names>(failed);
2022
1/2
✓ Branch 1 taken 30 times.
✗ Branch 2 not taken.
 30 pipe_fork.Write<pid_t>(pid_grand_child);
2023
2024 30 execvp(command_line[0].c_str(), const_cast<char **>(argv));
2025
2026 ✗ failed = ForkFailures::kFailExec;
2027
2028 ✗ fork_failure:
2029 ✗ pipe_fork.Write<ForkFailures::Names>(failed);
2030 ✗ _exit(1);
2031
1/2
✓ Branch 1 taken 2 times.
✗ Branch 2 not taken.
 2 }
2032
2/2
✓ Branch 0 taken 861 times.
✓ Branch 1 taken 16 times.
 877 if (double_fork) {
2033 int statloc;
2034
1/2
✓ Branch 1 taken 861 times.
✗ Branch 2 not taken.
 861 waitpid(pid, &statloc, 0);
2035 }
2036
2037
1/2
✓ Branch 1 taken 877 times.
✗ Branch 2 not taken.
 877 pipe_fork.CloseWriteFd();
2038
2039 // Either the PID or a return value is sent
2040 ForkFailures::Names status_code;
2041
1/2
✓ Branch 1 taken 877 times.
✗ Branch 2 not taken.
 877 pipe_fork.Read<ForkFailures::Names>(&status_code);
2042
2043
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 877 times.
 877 if (status_code != ForkFailures::kSendPid) {
2044 ✗ pipe_fork.CloseReadFd();
2045 ✗ LogCvmfs(kLogCvmfs, kLogDebug, "managed execve failed (%s)",
2046 ✗ ForkFailures::ToString(status_code).c_str());
2047 ✗ return false;
2048 }
2049
2050 // read the PID of the spawned process if requested
2051 // (the actual read needs to be done in any case!)
2052 877 pid_t buf_child_pid = 0;
2053
1/2
✓ Branch 1 taken 877 times.
✗ Branch 2 not taken.
 877 pipe_fork.Read(&buf_child_pid);
2054
2/2
✓ Branch 0 taken 90 times.
✓ Branch 1 taken 787 times.
 877 if (child_pid != NULL)
2055 90 *child_pid = buf_child_pid;
2056
1/2
✓ Branch 1 taken 877 times.
✗ Branch 2 not taken.
 877 pipe_fork.CloseReadFd();
2057
1/2
✓ Branch 2 taken 877 times.
✗ Branch 3 not taken.
 877 LogCvmfs(kLogCvmfs, kLogDebug, "execve'd %s (PID: %d)",
2058 877 command_line[0].c_str(), static_cast<int>(buf_child_pid));
2059 877 return true;
2060 877 }
2061
2062
2063 /**
2064 * Sleeps using select. This is without signals and doesn't interfere with
2065 * other uses of the ALRM signal.
2066 */
2067 2293 void SafeSleepMs(const unsigned ms) {
2068 struct timeval wait_for;
2069 2293 wait_for.tv_sec = ms / 1000;
2070 2293 wait_for.tv_usec = (ms % 1000) * 1000;
2071
2/2
✓ Branch 1 taken 2245 times.
✓ Branch 2 taken 48 times.
 2293 select(0, NULL, NULL, NULL, &wait_for);
2072 2245 }
2073
2074
2075 /**
2076 * Deal with EINTR and partial writes.
2077 */
2078 8323 bool SafeWrite(int fd, const void *buf, size_t nbyte) {
2079
2/2
✓ Branch 0 taken 8323 times.
✓ Branch 1 taken 8254 times.
 16577 while (nbyte) {
2080 8323 const ssize_t retval = write(fd, buf, nbyte);
2081
2/2
✓ Branch 0 taken 69 times.
✓ Branch 1 taken 8254 times.
 8323 if (retval < 0) {
2082
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 69 times.
 69 if (errno == EINTR)
2083 ✗ continue;
2084 69 return false;
2085 }
2086
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8254 times.
 8254 assert(static_cast<size_t>(retval) <= nbyte);
2087 8254 buf = reinterpret_cast<const char *>(buf) + retval;
2088 8254 nbyte -= retval;
2089 }
2090 8254 return true;
2091 }
2092
2093 /**
2094 * The contents of the iov vector might be modified by the function.
2095 */
2096 233359 bool SafeWriteV(int fd, struct iovec *iov, unsigned iovcnt) {
2097 233359 unsigned nbytes = 0;
2098
2/2
✓ Branch 0 taken 657482 times.
✓ Branch 1 taken 233359 times.
 890841 for (unsigned i = 0; i < iovcnt; ++i)
2099 657482 nbytes += iov[i].iov_len;
2100 233359 unsigned iov_idx = 0;
2101
2102
2/2
✓ Branch 0 taken 233359 times.
✓ Branch 1 taken 33 times.
 233392 while (nbytes) {
2103 466586 const ssize_t retval = writev(fd, &iov[iov_idx],
2104 233359 static_cast<int>(iovcnt - iov_idx));
2105
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 233227 times.
 233227 if (retval < 0) {
2106 ✗ if (errno == EINTR)
2107 ✗ continue;
2108 ✗ return false;
2109 }
2110
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 233227 times.
 233227 assert(static_cast<size_t>(retval) <= nbytes);
2111 233227 nbytes -= retval;
2112
2113 233227 unsigned sum_written_blocks = 0;
2114 233227 while ((sum_written_blocks + iov[iov_idx].iov_len)
2115
2/2
✓ Branch 0 taken 656954 times.
✓ Branch 1 taken 33 times.
 656987 <= static_cast<size_t>(retval)) {
2116 656954 sum_written_blocks += iov[iov_idx].iov_len;
2117 656954 iov_idx++;
2118
2/2
✓ Branch 0 taken 233194 times.
✓ Branch 1 taken 423760 times.
 656954 if (iov_idx == iovcnt) {
2119
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 233194 times.
 233194 assert(sum_written_blocks == static_cast<size_t>(retval));
2120 233194 return true;
2121 }
2122 }
2123 33 const unsigned offset = retval - sum_written_blocks;
2124 33 iov[iov_idx].iov_len -= offset;
2125 33 iov[iov_idx].iov_base = reinterpret_cast<char *>(iov[iov_idx].iov_base)
2126 33 + offset;
2127 }
2128
2129 33 return true;
2130 }
2131
2132
2133 /**
2134 * Deal with EINTR and partial reads.
2135 */
2136 20675240 ssize_t SafeRead(int fd, void *buf, size_t nbyte) {
2137 20675240 ssize_t total_bytes = 0;
2138
2/2
✓ Branch 0 taken 20942828 times.
✓ Branch 1 taken 20649563 times.
 41592391 while (nbyte) {
2139 20942828 const ssize_t retval = read(fd, buf, nbyte);
2140
2/2
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 20968262 times.
 20968282 if (retval < 0) {
2141
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
 20 if (errno == EINTR)
2142 ✗ continue;
2143 20 return -1;
2144
2/2
✓ Branch 0 taken 51111 times.
✓ Branch 1 taken 20917151 times.
 20968262 } else if (retval == 0) {
2145 51111 return total_bytes;
2146 }
2147
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 20917151 times.
 20917151 assert(static_cast<size_t>(retval) <= nbyte);
2148 20917151 buf = reinterpret_cast<char *>(buf) + retval;
2149 20917151 nbyte -= retval;
2150 20917151 total_bytes += retval;
2151 }
2152 20649563 return total_bytes;
2153 }
2154
2155
2156 /**
2157 * Pull file contents into a string
2158 */
2159 1047 bool SafeReadToString(int fd, std::string *final_result) {
2160
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1047 times.
 1047 if (!final_result) {
2161 ✗ return false;
2162 }
2163
2164 1047 std::string tmp_result;
2165 static const int buf_size = 4096;
2166 char buf[4096];
2167 1047 ssize_t total_bytes = -1;
2168 do {
2169
1/2
✓ Branch 1 taken 11307 times.
✗ Branch 2 not taken.
 11307 total_bytes = SafeRead(fd, buf, buf_size);
2170
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 11297 times.
 11307 if (total_bytes < 0) {
2171 10 return false;
2172 }
2173
1/2
✓ Branch 1 taken 11297 times.
✗ Branch 2 not taken.
 11297 tmp_result.append(buf, total_bytes);
2174
2/2
✓ Branch 0 taken 10260 times.
✓ Branch 1 taken 1037 times.
 11297 } while (total_bytes == buf_size);
2175 1037 final_result->swap(tmp_result);
2176 1037 return true;
2177 1047 }
2178
2179 6165 bool SafeWriteToFile(const std::string &content,
2180 const std::string &path,
2181 int mode) {
2182 6165 const int fd = open(path.c_str(), O_WRONLY | O_CREAT | O_TRUNC, mode);
2183
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6165 times.
 6165 if (fd < 0)
2184 ✗ return false;
2185 6165 const bool retval = SafeWrite(fd, content.data(), content.size());
2186 6165 close(fd);
2187 6165 return retval;
2188 }
2189
2190
2191 #ifdef CVMFS_NAMESPACE_GUARD
2192 } // namespace CVMFS_NAMESPACE_GUARD
2193 #endif
2194