GCC Code Coverage Report

Directory: cvmfs/
File: cvmfs/util/posix.cc
Date: 2026-08-30 02:40:36
Exec Total Coverage
Lines: 899 1160 77.5%
Branches: 669 1265 52.9%
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 11772 std::string MakeCanonicalPath(const std::string &path) {
109
2/2
✓ Branch 1 taken 25 times.
✓ Branch 2 taken 11747 times.
 11772 if (path.length() == 0)
110 25 return path;
111
112
2/2
✓ Branch 2 taken 171 times.
✓ Branch 3 taken 11576 times.
 11747 if (path[path.length() - 1] == '/') {
113 171 return path.substr(0, path.length() - 1);
114 } else {
115 11576 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 867 void SplitPath(const std::string &path,
125 std::string *dirname,
126 std::string *filename) {
127 867 const size_t dir_sep = path.rfind('/');
128
2/2
✓ Branch 0 taken 730 times.
✓ Branch 1 taken 137 times.
 867 if (dir_sep != std::string::npos) {
129 730 *dirname = path.substr(0, dir_sep);
130 730 *filename = path.substr(dir_sep + 1);
131 } else {
132 137 *dirname = ".";
133 137 *filename = path;
134 }
135 867 }
136
137
138 /**
139 * Gets the directory part of a path.
140 */
141 13056 std::string GetParentPath(const std::string &path) {
142 13056 const std::string::size_type idx = path.find_last_of('/');
143
2/2
✓ Branch 0 taken 12852 times.
✓ Branch 1 taken 204 times.
 13056 if (idx != std::string::npos) {
144 12852 return path.substr(0, idx);
145 } else {
146
1/2
✓ Branch 2 taken 204 times.
✗ Branch 3 not taken.
 204 return "";
147 }
148 }
149
150
151 /**
152 * Gets the file name part of a path.
153 */
154 344100 std::string GetFileName(const std::string &path) {
155 344100 const std::string::size_type idx = path.find_last_of('/');
156
2/2
✓ Branch 0 taken 343508 times.
✓ Branch 1 taken 592 times.
 344100 if (idx != std::string::npos) {
157 343508 return path.substr(idx + 1);
158 } else {
159 592 return path;
160 }
161 }
162
163
164 196 bool IsAbsolutePath(const std::string &path) {
165
4/4
✓ Branch 1 taken 184 times.
✓ Branch 2 taken 12 times.
✓ Branch 4 taken 25 times.
✓ Branch 5 taken 159 times.
 196 return (!path.empty() && path[0] == '/');
166 }
167
168
169 160 std::string GetAbsolutePath(const std::string &path) {
170
2/2
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 147 times.
 160 if (IsAbsolutePath(path))
171 13 return path;
172
173
2/4
✓ Branch 2 taken 147 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 147 times.
✗ Branch 6 not taken.
 294 return GetCurrentWorkingDirectory() + "/" + path;
174 }
175
176
177 2028 bool IsHttpUrl(const std::string &path) {
178
2/2
✓ Branch 1 taken 42 times.
✓ Branch 2 taken 1986 times.
 2028 if (path.length() < 7) {
179 42 return false;
180 }
181
182
1/2
✓ Branch 1 taken 1986 times.
✗ Branch 2 not taken.
 1986 std::string prefix = path.substr(0, 8);
183 1986 std::transform(prefix.begin(), prefix.end(), prefix.begin(), ::tolower);
184
185
6/12
✓ Branch 1 taken 1986 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 1881 times.
✓ Branch 5 taken 105 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 105 times.
✓ Branch 8 taken 1776 times.
✓ Branch 9 taken 1986 times.
✗ Branch 10 not taken.
✗ Branch 12 not taken.
✗ Branch 13 not taken.
 1986 return prefix.substr(0, 7) == "http://" || prefix == "https://";
186 1986 }
187
188
189 4631 FileSystemInfo GetFileSystemInfo(const std::string &path) {
190 4631 FileSystemInfo result;
191
192 struct statfs info;
193 4631 const int retval = statfs(path.c_str(), &info);
194
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4631 times.
 4631 if (retval != 0)
195 ✗ return result;
196
197
2/6
✗ Branch 0 not taken.
✗ Branch 1 not taken.
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 4615 times.
 4631 switch (info.f_type) {
198 ✗ case kFsTypeAutofs:
199 ✗ result.type = kFsTypeAutofs;
200 ✗ break;
201 ✗ case kFsTypeNFS:
202 ✗ result.type = kFsTypeNFS;
203 ✗ break;
204 16 case kFsTypeProc:
205 16 result.type = kFsTypeProc;
206 16 break;
207 ✗ case kFsTypeBeeGFS:
208 ✗ result.type = kFsTypeBeeGFS;
209 ✗ break;
210 ✗ case kFsTypeTmpfs:
211 ✗ result.type = kFsTypeTmpfs;
212 ✗ break;
213 4615 default:
214 4615 result.type = kFsTypeUnknown;
215 }
216
217 #ifdef CVMFS_HAS_STATFS_F_FLAGS
218
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4631 times.
 4631 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 4631 return result;
228 }
229
230
231 996 std::string ReadSymlink(const std::string &path) {
232 // TODO(jblomer): avoid PATH_MAX
233 char buf[PATH_MAX + 1];
234 996 const ssize_t nchars = readlink(path.c_str(), buf, PATH_MAX);
235
2/2
✓ Branch 0 taken 954 times.
✓ Branch 1 taken 42 times.
 996 if (nchars >= 0) {
236 954 buf[nchars] = '\0';
237
1/2
✓ Branch 2 taken 954 times.
✗ Branch 3 not taken.
 954 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 434 std::string ResolvePath(const std::string &path) {
248
6/6
✓ Branch 1 taken 316 times.
✓ Branch 2 taken 118 times.
✓ Branch 4 taken 43 times.
✓ Branch 5 taken 273 times.
✓ Branch 6 taken 161 times.
✓ Branch 7 taken 273 times.
 434 if (path.empty() || (path == "/"))
249
1/2
✓ Branch 2 taken 161 times.
✗ Branch 3 not taken.
 161 return "/";
250
1/2
✓ Branch 1 taken 273 times.
✗ Branch 2 not taken.
 273 const std::string name = GetFileName(path);
251
1/2
✓ Branch 1 taken 273 times.
✗ Branch 2 not taken.
 273 std::string result = name;
252
2/2
✓ Branch 1 taken 241 times.
✓ Branch 2 taken 32 times.
 273 if (name != path) {
253 // There is a parent path of 'path'
254
2/4
✓ Branch 1 taken 241 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 241 times.
✗ Branch 5 not taken.
 241 const std::string parent = ResolvePath(GetParentPath(path));
255
4/6
✓ Branch 1 taken 139 times.
✓ Branch 2 taken 102 times.
✓ Branch 4 taken 241 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 241 times.
✗ Branch 8 not taken.
 241 result = parent + (parent == "/" ? "" : "/") + name;
256 241 }
257 273 char *real_result = realpath(result.c_str(), NULL);
258
2/2
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 145 times.
 273 if (real_result) {
259
1/2
✓ Branch 1 taken 128 times.
✗ Branch 2 not taken.
 128 result = real_result;
260 128 free(real_result);
261 }
262
3/4
✓ Branch 1 taken 273 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 16 times.
✓ Branch 4 taken 257 times.
 273 if (SymlinkExists(result)) {
263 char buf[PATH_MAX + 1];
264 16 const ssize_t nchars = readlink(result.c_str(), buf, PATH_MAX);
265
1/2
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
 16 if (nchars >= 0) {
266 16 buf[nchars] = '\0';
267
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 result = buf;
268 }
269 }
270 273 return result;
271 273 }
272
273
274 81 bool IsMountPoint(const std::string &path) {
275
1/2
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
 81 std::vector<std::string> mount_list = platform_mountlist();
276
1/2
✓ Branch 1 taken 81 times.
✗ Branch 2 not taken.
 81 const std::string resolved_path = ResolvePath(path);
277
2/2
✓ Branch 1 taken 6642 times.
✓ Branch 2 taken 27 times.
 6669 for (unsigned i = 0; i < mount_list.size(); ++i) {
278
2/2
✓ Branch 2 taken 54 times.
✓ Branch 3 taken 6588 times.
 6642 if (mount_list[i] == resolved_path)
279 54 return true;
280 }
281 27 return false;
282 81 }
283
284
285 /**
286 * By default PANIC(NULL) on failure
287 */
288 2874 void CreateFile(const std::string &path,
289 const int mode,
290 const bool ignore_failure) {
291 2874 const int fd = open(path.c_str(), O_CREAT, mode);
292
2/2
✓ Branch 0 taken 2661 times.
✓ Branch 1 taken 213 times.
 2874 if (fd >= 0) {
293 2661 close(fd);
294 2661 return;
295 }
296
2/2
✓ Branch 0 taken 197 times.
✓ Branch 1 taken 16 times.
 213 if (ignore_failure)
297 197 return;
298 16 PANIC(NULL);
299 }
300
301
302 /**
303 * Symlinks /tmp/cvmfs.XYZ/l --> ParentPath(path) to make it shorter
304 */
305 110 static std::string MakeShortSocketLink(const std::string &path) {
306 //NOLINTNEXTLINE(misc-const-correctness)
307 struct sockaddr_un sock_addr;
308 110 const unsigned max_length = sizeof(sock_addr.sun_path);
309
310 110 std::string result;
311
2/4
✓ Branch 2 taken 110 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 110 times.
✗ Branch 6 not taken.
 220 const std::string tmp_path = CreateTempDir("/tmp/cvmfs");
312
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 110 times.
 110 if (tmp_path.empty())
313 ✗ return "";
314
1/2
✓ Branch 1 taken 110 times.
✗ Branch 2 not taken.
 110 const std::string link = tmp_path + "/l";
315
3/6
✓ Branch 1 taken 110 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 110 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 110 times.
✗ Branch 8 not taken.
 110 result = link + "/" + GetFileName(path);
316
2/2
✓ Branch 1 taken 42 times.
✓ Branch 2 taken 68 times.
 110 if (result.length() >= max_length) {
317 42 rmdir(tmp_path.c_str());
318
1/2
✓ Branch 2 taken 42 times.
✗ Branch 3 not taken.
 42 return "";
319 }
320
1/2
✓ Branch 2 taken 68 times.
✗ Branch 3 not taken.
 68 const int retval = symlink(GetParentPath(path).c_str(), link.c_str());
321
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 68 times.
 68 if (retval != 0) {
322 ✗ rmdir(tmp_path.c_str());
323 ✗ return "";
324 }
325 68 return result;
326 110 }
327
328 68 static void RemoveShortSocketLink(const std::string &short_path) {
329
1/2
✓ Branch 1 taken 68 times.
✗ Branch 2 not taken.
 68 const std::string link = GetParentPath(short_path);
330 68 unlink(link.c_str());
331
1/2
✓ Branch 1 taken 68 times.
✗ Branch 2 not taken.
 68 rmdir(GetParentPath(link).c_str());
332 68 }
333
334
335 /**
336 * Creates and binds to a named socket.
337 */
338 2708 int MakeSocket(const std::string &path, const int mode) {
339
1/2
✓ Branch 1 taken 2708 times.
✗ Branch 2 not taken.
 2708 std::string short_path(path);
340 struct sockaddr_un sock_addr;
341
2/2
✓ Branch 1 taken 58 times.
✓ Branch 2 taken 2650 times.
 2708 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 58 times.
✗ Branch 2 not taken.
 58 short_path = MakeShortSocketLink(path);
345
2/2
✓ Branch 1 taken 16 times.
✓ Branch 2 taken 42 times.
 58 if (short_path.empty())
346 16 return -1;
347 }
348 2692 sock_addr.sun_family = AF_UNIX;
349 2692 strncpy(sock_addr.sun_path, short_path.c_str(), sizeof(sock_addr.sun_path));
350
351 2692 const int socket_fd = socket(AF_UNIX, SOCK_STREAM, 0);
352
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2692 times.
 2692 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 2692 times.
 2692 if (fchmod(socket_fd, mode) != 0)
358 ✗ goto make_socket_failure;
359 #endif
360
361 2692 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 16 times.
✓ Branch 1 taken 2676 times.
 2692 < 0) {
364
3/6
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
✓ Branch 4 taken 16 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 16 times.
✗ Branch 7 not taken.
 16 if ((errno == EADDRINUSE) && (unlink(path.c_str()) == 0)) {
365 // Second try, perhaps the file was left over
366 16 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 16 times.
 16 < 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 42 times.
✓ Branch 2 taken 2650 times.
 2692 if (short_path != path)
379
1/2
✓ Branch 1 taken 42 times.
✗ Branch 2 not taken.
 42 RemoveShortSocketLink(short_path);
380
381 2692 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 2708 }
389
390
391 /**
392 * Creates and binds a TCP/IPv4 socket. An empty address binds to the "any"
393 * address.
394 */
395 566 int MakeTcpEndpoint(const std::string &ipv4_address, int portno) {
396 566 const int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
397
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 566 times.
 566 assert(socket_fd != -1);
398 566 const int on = 1;
399 566 int retval = setsockopt(socket_fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
400
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 566 times.
 566 assert(retval == 0);
401
402 struct sockaddr_in endpoint_addr;
403 566 memset(&endpoint_addr, 0, sizeof(endpoint_addr));
404 566 endpoint_addr.sin_family = AF_INET;
405
2/2
✓ Branch 1 taken 518 times.
✓ Branch 2 taken 48 times.
 566 if (ipv4_address.empty()) {
406 518 endpoint_addr.sin_addr.s_addr = INADDR_ANY;
407 } else {
408 48 retval = inet_aton(ipv4_address.c_str(), &(endpoint_addr.sin_addr));
409
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 32 times.
 48 if (retval == 0) {
410
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 LogCvmfs(kLogCvmfs, kLogDebug, "invalid IPv4 address");
411
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 close(socket_fd);
412 16 return -1;
413 }
414 }
415 550 endpoint_addr.sin_port = htons(portno);
416
417 550 retval = bind(socket_fd, reinterpret_cast<struct sockaddr *>(&endpoint_addr),
418 sizeof(endpoint_addr));
419
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 534 times.
 550 if (retval < 0) {
420
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 LogCvmfs(kLogCvmfs, kLogDebug, "binding TCP endpoint failed (%d)", errno);
421
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 close(socket_fd);
422 16 return -1;
423 }
424 534 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 844 int ConnectSocket(const std::string &path) {
434
1/2
✓ Branch 1 taken 844 times.
✗ Branch 2 not taken.
 844 std::string short_path(path);
435 struct sockaddr_un sock_addr;
436
2/2
✓ Branch 1 taken 52 times.
✓ Branch 2 taken 792 times.
 844 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 52 times.
✗ Branch 2 not taken.
 52 short_path = MakeShortSocketLink(path);
440
2/2
✓ Branch 1 taken 26 times.
✓ Branch 2 taken 26 times.
 52 if (short_path.empty())
441 26 return -1;
442 }
443 818 sock_addr.sun_family = AF_UNIX;
444 818 strncpy(sock_addr.sun_path, short_path.c_str(), sizeof(sock_addr.sun_path));
445
446 818 const int socket_fd = socket(AF_UNIX, SOCK_STREAM, 0);
447
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 818 times.
 818 assert(socket_fd != -1);
448
449
1/2
✓ Branch 1 taken 818 times.
✗ Branch 2 not taken.
 818 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 26 times.
✓ Branch 2 taken 792 times.
 818 if (short_path != path)
453
1/2
✓ Branch 1 taken 26 times.
✗ Branch 2 not taken.
 26 RemoveShortSocketLink(short_path);
454
455
2/2
✓ Branch 0 taken 26 times.
✓ Branch 1 taken 792 times.
 818 if (retval < 0) {
456
1/2
✓ Branch 1 taken 26 times.
✗ Branch 2 not taken.
 26 close(socket_fd);
457 26 return -1;
458 }
459
460 792 return socket_fd;
461 844 }
462
463
464 /**
465 * Connects to a (remote) TCP server
466 */
467 74 int ConnectTcpEndpoint(const std::string &ipv4_address, int portno) {
468 74 const int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
469
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 74 times.
 74 assert(socket_fd != -1);
470
471 struct sockaddr_in endpoint_addr;
472 74 memset(&endpoint_addr, 0, sizeof(endpoint_addr));
473 74 endpoint_addr.sin_family = AF_INET;
474 74 int retval = inet_aton(ipv4_address.c_str(), &(endpoint_addr.sin_addr));
475
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 58 times.
 74 if (retval == 0) {
476
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 LogCvmfs(kLogCvmfs, kLogDebug, "invalid IPv4 address");
477
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 close(socket_fd);
478 16 return -1;
479 }
480 58 endpoint_addr.sin_port = htons(portno);
481
482
1/2
✓ Branch 1 taken 58 times.
✗ Branch 2 not taken.
 58 retval = connect(socket_fd,
483 reinterpret_cast<struct sockaddr *>(&endpoint_addr),
484 sizeof(endpoint_addr));
485
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 42 times.
 58 if (retval != 0) {
486 16 LogCvmfs(kLogCvmfs, kLogDebug, "failed to connect to TCP endpoint (%d)",
487
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 errno);
488
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 close(socket_fd);
489 16 return -1;
490 }
491 42 return socket_fd;
492 }
493
494
495 /**
496 * Creating a pipe should always succeed.
497 */
498 30366 void MakePipe(int pipe_fd[2]) {
499 30366 const int retval = pipe(pipe_fd);
500
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 30366 times.
 30366 assert(retval == 0);
501 30366 }
502
503
504 /**
505 * Writes to a pipe should always succeed.
506 */
507 13432952 void WritePipe(int fd, const void *buf, size_t nbyte) {
508 ssize_t num_bytes;
509 do {
510 13432952 num_bytes = write(fd, buf, nbyte);
511
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 13432952 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 13432952 } while ((num_bytes < 0) && (errno == EINTR));
512
2/4
✓ Branch 0 taken 13432952 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 13432952 times.
✗ Branch 3 not taken.
 13432952 assert((num_bytes >= 0) && (static_cast<size_t>(num_bytes) == nbyte));
513 13432952 }
514
515
516 /**
517 * Reads from a pipe should always succeed.
518 */
519 74805611 void ReadPipe(int fd, void *buf, size_t nbyte) {
520 ssize_t num_bytes;
521 do {
522 74805611 num_bytes = read(fd, buf, nbyte);
523
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 74805611 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 74805611 } while ((num_bytes < 0) && (errno == EINTR));
524
2/4
✓ Branch 0 taken 74805611 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 74805611 times.
✗ Branch 3 not taken.
 74805611 assert((num_bytes >= 0) && (static_cast<size_t>(num_bytes) == nbyte));
525 74805611 }
526
527
528 /**
529 * Reads from a pipe where writer's end is not yet necessarily connected
530 */
531 3805099 bool ReadHalfPipe(int fd, void *buf, size_t nbyte, unsigned timeout_ms) {
532 ssize_t num_bytes;
533 3805099 unsigned i = 0;
534 3805099 unsigned backoff_ms = 1;
535 3805099 uint64_t duration_ms = 0;
536 3805099 uint64_t timestamp = 0;
537
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 3805087 times.
 3805099 if (timeout_ms != 0)
538 12 timestamp = platform_monotonic_time_ns();
539
540 3805099 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 3805099 num_bytes = read(fd, buf, nbyte);
545
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3805099 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 3805099 if ((num_bytes < 0) && (errno == EINTR))
546 ✗ continue;
547 3805099 i++;
548 // Start backing off when the busy loop reaches the ballpark of 1ms
549
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3805099 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 3805099 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 12 times.
✓ Branch 1 taken 3805087 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 12 times.
 3805099 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 3805099 times.
 3805099 } while (num_bytes == 0);
562
2/4
✓ Branch 0 taken 3805099 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 3805099 times.
✗ Branch 3 not taken.
 3805099 assert((num_bytes >= 0) && (static_cast<size_t>(num_bytes) == nbyte));
563 3805099 return true;
564 }
565
566
567 /**
568 * Closes both ends of a pipe
569 */
570 25445 void ClosePipe(int pipe_fd[2]) {
571 25445 close(pipe_fd[0]);
572 25445 close(pipe_fd[1]);
573 25445 }
574
575
576 /**
577 * Compares two directory trees on the meta-data level. Returns true iff the
578 * trees have identical content.
579 */
580 163164 bool DiffTree(const std::string &path_a, const std::string &path_b) {
581 int retval;
582 163164 std::vector<std::string> ls_a;
583 163164 std::vector<std::string> ls_b;
584 163164 std::vector<std::string> subdirs;
585
586
1/2
✓ Branch 2 taken 163164 times.
✗ Branch 3 not taken.
 163164 DIR *dirp_a = opendir(path_a.c_str());
587
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 163164 times.
 163164 if (dirp_a == NULL)
588 ✗ return false;
589
1/2
✓ Branch 2 taken 163164 times.
✗ Branch 3 not taken.
 163164 DIR *dirp_b = opendir(path_b.c_str());
590
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 163164 times.
 163164 if (dirp_b == NULL) {
591 ✗ closedir(dirp_a);
592 ✗ return false;
593 }
594
595 platform_dirent64 *dirent;
596
3/4
✓ Branch 1 taken 660924 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 497760 times.
✓ Branch 4 taken 163164 times.
 660924 while ((dirent = platform_readdir(dirp_a))) {
597
1/2
✓ Branch 2 taken 497760 times.
✗ Branch 3 not taken.
 497760 const std::string name(dirent->d_name);
598
6/6
✓ Branch 1 taken 334596 times.
✓ Branch 2 taken 163164 times.
✓ Branch 4 taken 163164 times.
✓ Branch 5 taken 171432 times.
✓ Branch 6 taken 326328 times.
✓ Branch 7 taken 171432 times.
 497760 if ((name == ".") || (name == ".."))
599 326328 continue;
600
2/4
✓ Branch 1 taken 171432 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 171432 times.
✗ Branch 5 not taken.
 171432 const std::string path = path_a + "/" + name;
601
1/2
✓ Branch 1 taken 171432 times.
✗ Branch 2 not taken.
 171432 ls_a.push_back(path);
602
603 platform_stat64 info;
604 171432 retval = platform_lstat(path.c_str(), &info);
605
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 171432 times.
 171432 if (retval != 0) {
606 ✗ closedir(dirp_a);
607 ✗ closedir(dirp_b);
608 ✗ return false;
609 }
610
2/2
✓ Branch 0 taken 163764 times.
✓ Branch 1 taken 7668 times.
 171432 if (S_ISDIR(info.st_mode))
611
1/2
✓ Branch 1 taken 163764 times.
✗ Branch 2 not taken.
 163764 subdirs.push_back(name);
612
3/5
✓ Branch 1 taken 171432 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 171432 times.
✓ Branch 5 taken 326328 times.
✗ Branch 6 not taken.
 497760 }
613
3/4
✓ Branch 1 taken 660288 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 497124 times.
✓ Branch 4 taken 163164 times.
 660288 while ((dirent = platform_readdir(dirp_b))) {
614
1/2
✓ Branch 2 taken 497124 times.
✗ Branch 3 not taken.
 497124 const std::string name(dirent->d_name);
615
6/6
✓ Branch 1 taken 333960 times.
✓ Branch 2 taken 163164 times.
✓ Branch 4 taken 163164 times.
✓ Branch 5 taken 170796 times.
✓ Branch 6 taken 326328 times.
✓ Branch 7 taken 170796 times.
 497124 if ((name == ".") || (name == ".."))
616 326328 continue;
617
2/4
✓ Branch 1 taken 170796 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 170796 times.
✗ Branch 5 not taken.
 170796 const std::string path = path_b + "/" + name;
618
1/2
✓ Branch 1 taken 170796 times.
✗ Branch 2 not taken.
 170796 ls_b.push_back(path);
619
2/2
✓ Branch 2 taken 170796 times.
✓ Branch 3 taken 326328 times.
 497124 }
620
1/2
✓ Branch 1 taken 163164 times.
✗ Branch 2 not taken.
 163164 closedir(dirp_a);
621
1/2
✓ Branch 1 taken 163164 times.
✗ Branch 2 not taken.
 163164 closedir(dirp_b);
622
623
1/2
✓ Branch 3 taken 163164 times.
✗ Branch 4 not taken.
 163164 sort(ls_a.begin(), ls_a.end());
624
1/2
✓ Branch 3 taken 163164 times.
✗ Branch 4 not taken.
 163164 sort(ls_b.begin(), ls_b.end());
625
2/2
✓ Branch 2 taken 12 times.
✓ Branch 3 taken 163152 times.
 163164 if (ls_a.size() != ls_b.size())
626 12 return false;
627
2/2
✓ Branch 1 taken 170544 times.
✓ Branch 2 taken 163152 times.
 333696 for (unsigned i = 0; i < ls_a.size(); ++i) {
628
3/7
✓ Branch 2 taken 170544 times.
✗ Branch 3 not taken.
✓ Branch 6 taken 170544 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✗ Branch 11 not taken.
✓ Branch 12 taken 170544 times.
 170544 if (GetFileName(ls_a[i]) != GetFileName(ls_b[i]))
629 ✗ return false;
630 platform_stat64 info_a;
631 platform_stat64 info_b;
632 170544 retval = platform_lstat(ls_a[i].c_str(), &info_a);
633
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 170544 times.
 170544 if (retval != 0)
634 ✗ return false;
635 170544 retval = platform_lstat(ls_b[i].c_str(), &info_b);
636
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 170544 times.
 170544 if (retval != 0)
637 ✗ return false;
638
2/4
✓ Branch 0 taken 170544 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 170544 times.
✗ Branch 3 not taken.
 170544 if ((info_a.st_mode != info_b.st_mode) || (info_a.st_uid != info_b.st_uid)
639
1/2
✓ Branch 0 taken 170544 times.
✗ Branch 1 not taken.
 170544 || (info_a.st_gid != info_b.st_gid)
640
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 170544 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 170544 || ((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 163128 times.
✓ Branch 2 taken 163152 times.
 326280 for (unsigned i = 0; i < subdirs.size(); ++i) {
646
3/6
✓ Branch 2 taken 163128 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 163128 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 163128 times.
✗ Branch 9 not taken.
 163128 const bool retval_subtree = DiffTree(path_a + "/" + subdirs[i],
647
2/4
✓ Branch 2 taken 163128 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 163128 times.
✗ Branch 6 not taken.
 326256 path_b + "/" + subdirs[i]);
648
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 163128 times.
 163128 if (!retval_subtree)
649 ✗ return false;
650 }
651
652 163152 return true;
653 163164 }
654
655
656 /**
657 * Changes a non-blocking file descriptor to a blocking one.
658 */
659 170 void Nonblock2Block(int filedes) {
660 170 const int flags = fcntl(filedes, F_GETFL);
661
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 170 times.
 170 assert(flags != -1);
662 170 const int retval = fcntl(filedes, F_SETFL, flags & ~O_NONBLOCK);
663
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 170 times.
 170 assert(retval != -1);
664 170 }
665
666
667 /**
668 * Changes a blocking file descriptor to a non-blocking one.
669 */
670 168 void Block2Nonblock(int filedes) {
671 168 const int flags = fcntl(filedes, F_GETFL);
672
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 168 times.
 168 assert(flags != -1);
673 168 const int retval = fcntl(filedes, F_SETFL, flags | O_NONBLOCK);
674
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 168 times.
 168 assert(retval != -1);
675 168 }
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 67 void SendMsg2Socket(const int fd, const std::string &msg) {
683 67 (void)send(fd, &msg[0], msg.length(), MSG_NOSIGNAL);
684 67 }
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 27 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 27 memset(ctrl_msg.buf, 0, sizeof(ctrl_msg.buf));
700
701 struct msghdr msgh;
702 27 msgh.msg_name = NULL;
703 27 msgh.msg_namelen = 0;
704
705 27 unsigned char dummy = 0;
706 struct iovec iov;
707 27 iov.iov_base = &dummy;
708 27 iov.iov_len = 1;
709 27 msgh.msg_iov = &iov;
710 27 msgh.msg_iovlen = 1;
711
712 27 msgh.msg_control = ctrl_msg.buf;
713 27 msgh.msg_controllen = sizeof(ctrl_msg.buf);
714
1/2
✓ Branch 0 taken 27 times.
✗ Branch 1 not taken.
 27 struct cmsghdr *cmsgp = CMSG_FIRSTHDR(&msgh);
715 27 cmsgp->cmsg_len = CMSG_LEN(sizeof(int));
716 27 cmsgp->cmsg_level = SOL_SOCKET;
717 27 cmsgp->cmsg_type = SCM_RIGHTS;
718 27 memcpy(CMSG_DATA(cmsgp), &passing_fd, sizeof(int));
719
720
1/2
✓ Branch 1 taken 27 times.
✗ Branch 2 not taken.
 27 const ssize_t retval = sendmsg(socket_fd, &msgh, 0);
721 27 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 27 std::string GetHostname() {
774 char name[HOST_NAME_MAX + 1];
775 27 const int retval = gethostname(name, HOST_NAME_MAX);
776
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 27 times.
 27 assert(retval == 0);
777
1/2
✓ Branch 2 taken 27 times.
✗ Branch 3 not taken.
 27 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 18 bool SwitchCredentials(const uid_t uid, const gid_t gid, const bool temporarily,
789 const bool avoid_mutexes) {
790
1/2
✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
 18 if (!avoid_mutexes) {
791 18 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 18 int retval = 0;
797
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
 18 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 18 times.
✗ Branch 2 not taken.
 18 if (geteuid() != 0) {
806 18 retval = ObtainSetuidgidCapabilities(avoid_mutexes);
807
2/4
✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
 18 if (!retval && !avoid_mutexes) {
808 18 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 18 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
 18 retval = setgid(gid) || setuid(uid);
815 }
816
1/2
✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
 18 if (!avoid_mutexes) {
817 18 LogCvmfs(kLogCvmfs, kLogDebug, "switch credentials result %d (%d)", retval,
818 18 errno);
819 }
820 18 return retval == 0;
821 }
822
823
824 /**
825 * Checks if the regular file path exists.
826 */
827 102923 bool FileExists(const std::string &path) {
828 platform_stat64 info;
829
4/4
✓ Branch 2 taken 52060 times.
✓ Branch 3 taken 50863 times.
✓ Branch 4 taken 50433 times.
✓ Branch 5 taken 1627 times.
 102923 return ((platform_lstat(path.c_str(), &info) == 0) && S_ISREG(info.st_mode));
830 }
831
832
833 /**
834 * Returns -1 on failure.
835 */
836 2751910 int64_t GetFileSize(const std::string &path) {
837 platform_stat64 info;
838 2751910 const int retval = platform_stat(path.c_str(), &info);
839
2/2
✓ Branch 0 taken 2661858 times.
✓ Branch 1 taken 97017 times.
 2758875 if (retval != 0)
840 2661858 return -1;
841 97017 return info.st_size;
842 }
843
844
845 /**
846 * Checks if the directory (not symlink) path exists.
847 */
848 3825 bool DirectoryExists(const std::string &path) {
849 platform_stat64 info;
850
3/4
✓ Branch 2 taken 3757 times.
✓ Branch 3 taken 68 times.
✓ Branch 4 taken 3757 times.
✗ Branch 5 not taken.
 3825 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 1976 bool SymlinkExists(const std::string &path) {
858 platform_stat64 info;
859
4/4
✓ Branch 2 taken 1830 times.
✓ Branch 3 taken 146 times.
✓ Branch 4 taken 1702 times.
✓ Branch 5 taken 128 times.
 1976 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 131 bool SymlinkForced(const std::string &src, const std::string &dest) {
867 131 int retval = unlink(dest.c_str());
868
3/4
✓ Branch 0 taken 114 times.
✓ Branch 1 taken 17 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 114 times.
 131 if ((retval != 0) && (errno != ENOENT))
869 ✗ return false;
870 131 retval = symlink(src.c_str(), dest.c_str());
871 131 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 1879022 bool MkdirDeep(const std::string &path,
880 const mode_t mode,
881 bool verify_writable) {
882
2/2
✓ Branch 1 taken 50 times.
✓ Branch 2 taken 1878972 times.
 1879022 if (path == "")
883 50 return false;
884
885 1878972 int retval = mkdir(path.c_str(), mode);
886
2/2
✓ Branch 0 taken 1869356 times.
✓ Branch 1 taken 9616 times.
 1878972 if (retval == 0)
887 1869356 return true;
888
889 19232 if ((errno == ENOENT)
890
9/15
✓ Branch 0 taken 3835 times.
✓ Branch 1 taken 5781 times.
✓ Branch 3 taken 3835 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✓ Branch 6 taken 3835 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 3835 times.
✗ Branch 9 not taken.
✓ Branch 10 taken 3835 times.
✓ Branch 11 taken 5781 times.
✓ Branch 13 taken 3835 times.
✓ Branch 14 taken 5781 times.
✗ Branch 15 not taken.
✗ Branch 16 not taken.
 9616 && (MkdirDeep(GetParentPath(path), mode, verify_writable))) {
891 3835 return MkdirDeep(path, mode, verify_writable);
892 }
893
894
1/2
✓ Branch 0 taken 5781 times.
✗ Branch 1 not taken.
 5781 if (errno == EEXIST) {
895 platform_stat64 info;
896
5/6
✓ Branch 2 taken 5781 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 5759 times.
✓ Branch 5 taken 22 times.
✓ Branch 6 taken 5759 times.
✓ Branch 7 taken 22 times.
 5781 if ((platform_stat(path.c_str(), &info) == 0) && S_ISDIR(info.st_mode)) {
897
2/2
✓ Branch 0 taken 417 times.
✓ Branch 1 taken 5342 times.
 5759 if (verify_writable) {
898 417 retval = utimes(path.c_str(), NULL);
899
1/2
✓ Branch 0 taken 417 times.
✗ Branch 1 not taken.
 417 if (retval == 0)
900 5759 return true;
901 } else {
902 5342 return true;
903 }
904 }
905 }
906
907 22 return false;
908 }
909
910
911 /**
912 * Creates the "hash cache" directory structure in path.
913 */
914 6347 bool MakeCacheDirectories(const std::string &path, const mode_t mode) {
915
1/2
✓ Branch 1 taken 6347 times.
✗ Branch 2 not taken.
 6347 const std::string canonical_path = MakeCanonicalPath(path);
916
917
1/2
✓ Branch 1 taken 6347 times.
✗ Branch 2 not taken.
 6347 std::string this_path = canonical_path + "/quarantaine";
918
2/4
✓ Branch 1 taken 6347 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 6347 times.
 6347 if (!MkdirDeep(this_path, mode, false))
919 ✗ return false;
920
921
1/2
✓ Branch 1 taken 6347 times.
✗ Branch 2 not taken.
 6347 this_path = canonical_path + "/ff";
922
923 platform_stat64 stat_info;
924
2/2
✓ Branch 2 taken 6174 times.
✓ Branch 3 taken 173 times.
 6347 if (platform_stat(this_path.c_str(), &stat_info) != 0) {
925
1/2
✓ Branch 1 taken 6174 times.
✗ Branch 2 not taken.
 6174 this_path = canonical_path + "/txn";
926
2/4
✓ Branch 1 taken 6174 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 6174 times.
 6174 if (!MkdirDeep(this_path, mode, false))
927 ✗ return false;
928
2/2
✓ Branch 0 taken 1580544 times.
✓ Branch 1 taken 6174 times.
 1586718 for (int i = 0; i <= 0xff; i++) {
929 char hex[4];
930 1580544 snprintf(hex, sizeof(hex), "%02x", i);
931
3/6
✓ Branch 2 taken 1580544 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 1580544 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 1580544 times.
✗ Branch 9 not taken.
 1580544 this_path = canonical_path + "/" + std::string(hex);
932
2/4
✓ Branch 1 taken 1580544 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 1580544 times.
 1580544 if (!MkdirDeep(this_path, mode, false))
933 ✗ return false;
934 }
935 }
936 6347 return true;
937 6347 }
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 3829 int TryLockFile(const std::string &path) {
947 3829 const int fd_lockfile = open(path.c_str(), O_RDONLY | O_CREAT, 0600);
948
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 3804 times.
 3829 if (fd_lockfile < 0)
949 25 return -1;
950
951
2/2
✓ Branch 1 taken 42 times.
✓ Branch 2 taken 3762 times.
 3804 if (flock(fd_lockfile, LOCK_EX | LOCK_NB) != 0) {
952 42 close(fd_lockfile);
953
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
 42 if (errno != EWOULDBLOCK)
954 ✗ return -1;
955 42 return -2;
956 }
957
958 3762 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 64 int WritePidFile(const std::string &path) {
969
1/2
✓ Branch 2 taken 64 times.
✗ Branch 3 not taken.
 64 const int fd = open(path.c_str(), O_CREAT | O_RDWR, 0600);
970
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 48 times.
 64 if (fd < 0)
971 16 return -1;
972
2/2
✓ Branch 1 taken 16 times.
✓ Branch 2 taken 32 times.
 48 if (flock(fd, LOCK_EX | LOCK_NB) != 0) {
973
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 close(fd);
974
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
 16 if (errno != EWOULDBLOCK)
975 ✗ return -1;
976 16 return -2;
977 }
978
979 // Don't leak the file descriptor to exec'd children
980
1/2
✓ Branch 1 taken 32 times.
✗ Branch 2 not taken.
 32 int flags = fcntl(fd, F_GETFD);
981
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32 times.
 32 assert(flags != -1);
982 32 flags |= FD_CLOEXEC;
983
1/2
✓ Branch 1 taken 32 times.
✗ Branch 2 not taken.
 32 flags = fcntl(fd, F_SETFD, flags);
984
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32 times.
 32 assert(flags != -1);
985
986 char buf[64];
987
988 32 snprintf(buf, sizeof(buf), "%" PRId64 "\n", static_cast<uint64_t>(getpid()));
989 32 const bool retval = (ftruncate(fd, 0) == 0)
990
3/6
✓ Branch 0 taken 32 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 32 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 32 times.
✗ Branch 6 not taken.
 32 && SafeWrite(fd, buf, strlen(buf));
991
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32 times.
 32 if (!retval) {
992 ✗ UnlockFile(fd);
993 ✗ return -1;
994 }
995 32 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 4335 int LockFile(const std::string &path) {
1005 4335 const int fd_lockfile = open(path.c_str(), O_RDONLY | O_CREAT, 0600);
1006
2/2
✓ Branch 0 taken 117 times.
✓ Branch 1 taken 4218 times.
 4335 if (fd_lockfile < 0)
1007 117 return -1;
1008
1009
1010
2/2
✓ Branch 1 taken 25 times.
✓ Branch 2 taken 4193 times.
 4218 if (flock(fd_lockfile, LOCK_EX | LOCK_NB) != 0) {
1011
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 25 times.
 25 if (errno != EWOULDBLOCK) {
1012 ✗ close(fd_lockfile);
1013 ✗ return -1;
1014 }
1015 25 LogCvmfs(kLogCvmfs, kLogSyslog, "another process holds %s, waiting.",
1016 path.c_str());
1017
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 25 times.
 25 if (flock(fd_lockfile, LOCK_EX) != 0) {
1018 ✗ close(fd_lockfile);
1019 ✗ return -1;
1020 }
1021 25 LogCvmfs(kLogCvmfs, kLogSyslog, "lock %s acquired", path.c_str());
1022 }
1023
1024 4218 return fd_lockfile;
1025 }
1026
1027
1028 7917 void UnlockFile(const int filedes) {
1029 7917 const int retval = flock(filedes, LOCK_UN);
1030
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7917 times.
 7917 assert(retval == 0);
1031 7917 close(filedes);
1032 7917 }
1033
1034
1035 /**
1036 * Wrapper around mkstemp.
1037 */
1038 65507 FILE *CreateTempFile(const std::string &path_prefix, const int mode,
1039 const char *open_flags, std::string *final_path) {
1040 65507 *final_path = path_prefix + ".XXXXXX";
1041 65507 char *tmp_file = strdupa(final_path->c_str());
1042 65507 const int tmp_fd = mkstemp(tmp_file);
1043
2/2
✓ Branch 0 taken 94 times.
✓ Branch 1 taken 65413 times.
 65507 if (tmp_fd < 0) {
1044 94 return NULL;
1045 }
1046
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 65413 times.
 65413 if (fchmod(tmp_fd, mode) != 0) {
1047 ✗ close(tmp_fd);
1048 ✗ return NULL;
1049 }
1050
1051 65413 *final_path = tmp_file;
1052 65413 FILE *tmp_fp = fdopen(tmp_fd, open_flags);
1053
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 65413 times.
 65413 if (!tmp_fp) {
1054 ✗ close(tmp_fd);
1055 ✗ unlink(tmp_file);
1056 ✗ return NULL;
1057 }
1058
1059 65413 return tmp_fp;
1060 }
1061
1062
1063 /**
1064 * Create the file but don't open. Use only in non-public tmp directories.
1065 */
1066 18854 std::string CreateTempPath(const std::string &path_prefix, const int mode) {
1067 18854 std::string result;
1068
1/2
✓ Branch 1 taken 18854 times.
✗ Branch 2 not taken.
 18854 FILE *f = CreateTempFile(path_prefix, mode, "w", &result);
1069
2/2
✓ Branch 0 taken 25 times.
✓ Branch 1 taken 18829 times.
 18854 if (!f)
1070
1/2
✓ Branch 2 taken 25 times.
✗ Branch 3 not taken.
 25 return "";
1071
1/2
✓ Branch 1 taken 18829 times.
✗ Branch 2 not taken.
 18829 fclose(f);
1072 18829 return result;
1073 18854 }
1074
1075
1076 /**
1077 * Create a directory with a unique name.
1078 */
1079 7210 std::string CreateTempDir(const std::string &path_prefix) {
1080
1/2
✓ Branch 1 taken 7210 times.
✗ Branch 2 not taken.
 7210 const std::string dir = path_prefix + ".XXXXXX";
1081 7210 char *tmp_dir = strdupa(dir.c_str());
1082 7210 tmp_dir = mkdtemp(tmp_dir);
1083
2/2
✓ Branch 0 taken 18 times.
✓ Branch 1 taken 7192 times.
 7210 if (tmp_dir == NULL)
1084
1/2
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
 18 return "";
1085
1/2
✓ Branch 2 taken 7192 times.
✗ Branch 3 not taken.
 7192 return std::string(tmp_dir);
1086 7210 }
1087
1088
1089 /**
1090 * Get the current working directory of the running process
1091 */
1092 5944 std::string GetCurrentWorkingDirectory() {
1093 char cwd[PATH_MAX];
1094
3/9
✓ Branch 1 taken 5944 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 5944 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 5944 times.
✗ Branch 9 not taken.
✗ Branch 11 not taken.
✗ Branch 12 not taken.
 5944 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 13662 RemoveTreeHelper() { success = true; }
1105 85957 void RemoveFile(const std::string &parent_path, const std::string &name) {
1106
1/2
✓ Branch 2 taken 85957 times.
✗ Branch 3 not taken.
 85957 const int retval = unlink((parent_path + "/" + name).c_str());
1107
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 85957 times.
 85957 if (retval != 0)
1108 ✗ success = false;
1109 85957 }
1110 39147 void RemoveDir(const std::string &parent_path, const std::string &name) {
1111
1/2
✓ Branch 2 taken 39147 times.
✗ Branch 3 not taken.
 39147 const int retval = rmdir((parent_path + "/" + name).c_str());
1112
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 39147 times.
 39147 if (retval != 0)
1113 ✗ success = false;
1114 39147 }
1115 1663829 bool TryRemoveDir(const std::string &parent_path, const std::string &name) {
1116
1/2
✓ Branch 2 taken 1663829 times.
✗ Branch 3 not taken.
 1663829 const int retval = rmdir((parent_path + "/" + name).c_str());
1117 1663829 return (retval != 0);
1118 }
1119 };
1120
1121
1122 /**
1123 * Does rm -rf on path.
1124 */
1125 15132 bool RemoveTree(const std::string &path) {
1126 platform_stat64 info;
1127 15132 const int retval = platform_lstat(path.c_str(), &info);
1128
2/2
✓ Branch 0 taken 1470 times.
✓ Branch 1 taken 13662 times.
 15132 if (retval != 0)
1129 1470 return errno == ENOENT;
1130
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13662 times.
 13662 if (!S_ISDIR(info.st_mode))
1131 ✗ return false;
1132
1133
1/2
✓ Branch 1 taken 13662 times.
✗ Branch 2 not taken.
 13662 RemoveTreeHelper *remove_tree_helper = new RemoveTreeHelper();
1134
2/4
✓ Branch 2 taken 13662 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 13662 times.
✗ Branch 6 not taken.
 27324 FileSystemTraversal<RemoveTreeHelper> traversal(remove_tree_helper, "", true);
1135 13662 traversal.fn_new_file = &RemoveTreeHelper::RemoveFile;
1136 13662 traversal.fn_new_character_dev = &RemoveTreeHelper::RemoveFile;
1137 13662 traversal.fn_new_symlink = &RemoveTreeHelper::RemoveFile;
1138 13662 traversal.fn_new_socket = &RemoveTreeHelper::RemoveFile;
1139 13662 traversal.fn_new_fifo = &RemoveTreeHelper::RemoveFile;
1140 13662 traversal.fn_leave_dir = &RemoveTreeHelper::RemoveDir;
1141 13662 traversal.fn_new_dir_prefix = &RemoveTreeHelper::TryRemoveDir;
1142
1/2
✓ Branch 1 taken 13662 times.
✗ Branch 2 not taken.
 13662 traversal.Recurse(path);
1143 13662 const bool result = remove_tree_helper->success;
1144
1/2
✓ Branch 0 taken 13662 times.
✗ Branch 1 not taken.
 13662 delete remove_tree_helper;
1145
1146 13662 return result;
1147 13662 }
1148
1149
1150 /**
1151 * Returns ls $dir/GLOB$suffix
1152 */
1153 737 std::vector<std::string> FindFilesBySuffix(const std::string &dir,
1154 const std::string &suffix) {
1155 737 std::vector<std::string> result;
1156
1/2
✓ Branch 2 taken 737 times.
✗ Branch 3 not taken.
 737 DIR *dirp = opendir(dir.c_str());
1157
2/2
✓ Branch 0 taken 615 times.
✓ Branch 1 taken 122 times.
 737 if (!dirp)
1158 615 return result;
1159
1160 platform_dirent64 *dirent;
1161
3/4
✓ Branch 1 taken 702 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 580 times.
✓ Branch 4 taken 122 times.
 702 while ((dirent = platform_readdir(dirp))) {
1162
1/2
✓ Branch 2 taken 580 times.
✗ Branch 3 not taken.
 580 const std::string name(dirent->d_name);
1163 580 if ((name.length() >= suffix.length())
1164
9/14
✓ Branch 1 taken 454 times.
✓ Branch 2 taken 126 times.
✓ Branch 6 taken 454 times.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✓ Branch 9 taken 265 times.
✓ Branch 10 taken 189 times.
✓ Branch 11 taken 454 times.
✓ Branch 12 taken 126 times.
✗ Branch 13 not taken.
✓ Branch 14 taken 265 times.
✓ Branch 15 taken 315 times.
✗ Branch 16 not taken.
✗ Branch 17 not taken.
 1160 && (name.substr(name.length() - suffix.length()) == suffix)) {
1165
3/6
✓ Branch 1 taken 265 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 265 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 265 times.
✗ Branch 8 not taken.
 265 result.push_back(dir + "/" + name);
1166 }
1167 580 }
1168
1/2
✓ Branch 1 taken 122 times.
✗ Branch 2 not taken.
 122 closedir(dirp);
1169
1/2
✓ Branch 3 taken 122 times.
✗ Branch 4 not taken.
 122 std::sort(result.begin(), result.end());
1170 122 return result;
1171 }
1172
1173
1174 /**
1175 * Returns ls $dir/$prefixGLOB
1176 */
1177 90 std::vector<std::string> FindFilesByPrefix(const std::string &dir,
1178 const std::string &prefix) {
1179 90 std::vector<std::string> result;
1180
1/2
✓ Branch 2 taken 90 times.
✗ Branch 3 not taken.
 90 DIR *dirp = opendir(dir.c_str());
1181
2/2
✓ Branch 0 taken 18 times.
✓ Branch 1 taken 72 times.
 90 if (!dirp)
1182 18 return result;
1183
1184 platform_dirent64 *dirent;
1185
3/4
✓ Branch 1 taken 504 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 432 times.
✓ Branch 4 taken 72 times.
 504 while ((dirent = platform_readdir(dirp))) {
1186
1/2
✓ Branch 2 taken 432 times.
✗ Branch 3 not taken.
 432 const std::string name(dirent->d_name);
1187 432 if ((name.length() >= prefix.length())
1188
9/14
✓ Branch 1 taken 360 times.
✓ Branch 2 taken 72 times.
✓ Branch 5 taken 360 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 198 times.
✓ Branch 9 taken 162 times.
✓ Branch 10 taken 360 times.
✓ Branch 11 taken 72 times.
✗ Branch 12 not taken.
✓ Branch 13 taken 198 times.
✓ Branch 14 taken 234 times.
✗ Branch 15 not taken.
✗ Branch 16 not taken.
 864 && (name.substr(0, prefix.length()) == prefix)) {
1189
3/6
✓ Branch 1 taken 198 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 198 times.
✗ Branch 5 not taken.
✓ Branch 7 taken 198 times.
✗ Branch 8 not taken.
 198 result.push_back(dir + "/" + name);
1190 }
1191 432 }
1192
1/2
✓ Branch 1 taken 72 times.
✗ Branch 2 not taken.
 72 closedir(dirp);
1193
1/2
✓ Branch 3 taken 72 times.
✗ Branch 4 not taken.
 72 std::sort(result.begin(), result.end());
1194 72 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 60 std::vector<std::string> FindDirectories(const std::string &parent_dir) {
1203 60 std::vector<std::string> result;
1204
1/2
✓ Branch 2 taken 60 times.
✗ Branch 3 not taken.
 60 DIR *dirp = opendir(parent_dir.c_str());
1205
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 60 times.
 60 if (!dirp)
1206 ✗ return result;
1207
1208 platform_dirent64 *dirent;
1209
3/4
✓ Branch 1 taken 315 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 255 times.
✓ Branch 4 taken 60 times.
 315 while ((dirent = platform_readdir(dirp))) {
1210
1/2
✓ Branch 2 taken 255 times.
✗ Branch 3 not taken.
 255 const std::string name(dirent->d_name);
1211
6/6
✓ Branch 1 taken 195 times.
✓ Branch 2 taken 60 times.
✓ Branch 4 taken 60 times.
✓ Branch 5 taken 135 times.
✓ Branch 6 taken 120 times.
✓ Branch 7 taken 135 times.
 255 if ((name == ".") || (name == ".."))
1212 120 continue;
1213
2/4
✓ Branch 1 taken 135 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 135 times.
✗ Branch 5 not taken.
 135 const std::string path = parent_dir + "/" + name;
1214
1215 platform_stat64 info;
1216 135 const int retval = platform_stat(path.c_str(), &info);
1217
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 135 times.
 135 if (retval != 0)
1218 ✗ continue;
1219
2/2
✓ Branch 0 taken 105 times.
✓ Branch 1 taken 30 times.
 135 if (S_ISDIR(info.st_mode))
1220
1/2
✓ Branch 1 taken 105 times.
✗ Branch 2 not taken.
 105 result.push_back(path);
1221
3/4
✓ Branch 1 taken 135 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 135 times.
✓ Branch 5 taken 120 times.
 255 }
1222
1/2
✓ Branch 1 taken 60 times.
✗ Branch 2 not taken.
 60 closedir(dirp);
1223
1/2
✓ Branch 3 taken 60 times.
✗ Branch 4 not taken.
 60 sort(result.begin(), result.end());
1224 60 return result;
1225 }
1226
1227
1228 /**
1229 * Finds all files and direct subdirectories under directory (except ., ..).
1230 */
1231 150 bool ListDirectory(const std::string &directory,
1232 std::vector<std::string> *names,
1233 std::vector<mode_t> *modes) {
1234 150 DIR *dirp = opendir(directory.c_str());
1235
2/2
✓ Branch 0 taken 34 times.
✓ Branch 1 taken 116 times.
 150 if (!dirp)
1236 34 return false;
1237
1238 platform_dirent64 *dirent;
1239
2/2
✓ Branch 1 taken 1968 times.
✓ Branch 2 taken 116 times.
 2084 while ((dirent = platform_readdir(dirp))) {
1240
1/2
✓ Branch 2 taken 1968 times.
✗ Branch 3 not taken.
 1968 const std::string name(dirent->d_name);
1241
6/6
✓ Branch 1 taken 1852 times.
✓ Branch 2 taken 116 times.
✓ Branch 4 taken 116 times.
✓ Branch 5 taken 1736 times.
✓ Branch 6 taken 232 times.
✓ Branch 7 taken 1736 times.
 1968 if ((name == ".") || (name == ".."))
1242 232 continue;
1243
2/4
✓ Branch 1 taken 1736 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1736 times.
✗ Branch 5 not taken.
 1736 const std::string path = directory + "/" + name;
1244
1245 platform_stat64 info;
1246 1736 const int retval = platform_lstat(path.c_str(), &info);
1247
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1736 times.
 1736 if (retval != 0) {
1248 ✗ closedir(dirp);
1249 ✗ return false;
1250 }
1251
1252
1/2
✓ Branch 1 taken 1736 times.
✗ Branch 2 not taken.
 1736 names->push_back(name);
1253
1/2
✓ Branch 1 taken 1736 times.
✗ Branch 2 not taken.
 1736 modes->push_back(info.st_mode);
1254
3/5
✓ Branch 1 taken 1736 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 1736 times.
✓ Branch 5 taken 232 times.
✗ Branch 6 not taken.
 1968 }
1255 116 closedir(dirp);
1256
1257 116 SortTeam(names, modes);
1258 116 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 48 std::string FindExecutable(const std::string &exe) {
1267
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 48 times.
 48 if (exe.empty())
1268 ✗ return "";
1269
1270 48 std::vector<std::string> search_paths;
1271
2/2
✓ Branch 1 taken 16 times.
✓ Branch 2 taken 32 times.
 48 if (exe[0] == '/') {
1272
2/4
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 16 times.
✗ Branch 5 not taken.
 16 search_paths.push_back(GetParentPath(exe));
1273 } else {
1274 32 char *path_env = getenv("PATH");
1275
1/2
✓ Branch 0 taken 32 times.
✗ Branch 1 not taken.
 32 if (path_env) {
1276
2/4
✓ Branch 2 taken 32 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 32 times.
✗ Branch 6 not taken.
 32 search_paths = SplitString(path_env, ':');
1277 }
1278 }
1279
1280
2/2
✓ Branch 1 taken 176 times.
✓ Branch 2 taken 16 times.
 192 for (unsigned i = 0; i < search_paths.size(); ++i) {
1281
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 176 times.
 176 if (search_paths[i].empty())
1282 144 continue;
1283
2/4
✓ Branch 2 taken 176 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 176 times.
 176 if (search_paths[i][0] != '/')
1284 ✗ continue;
1285
1286
3/6
✓ Branch 1 taken 176 times.
✗ Branch 2 not taken.
✓ Branch 5 taken 176 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 176 times.
✗ Branch 9 not taken.
 352 std::string path = search_paths[i] + "/" + GetFileName(exe);
1287 platform_stat64 info;
1288 176 int retval = platform_stat(path.c_str(), &info);
1289
2/2
✓ Branch 0 taken 144 times.
✓ Branch 1 taken 32 times.
 176 if (retval != 0)
1290 144 continue;
1291
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32 times.
 32 if (!S_ISREG(info.st_mode))
1292 ✗ continue;
1293 32 retval = access(path.c_str(), X_OK);
1294
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32 times.
 32 if (retval != 0)
1295 ✗ continue;
1296
1297 32 return path;
1298
2/2
✓ Branch 1 taken 144 times.
✓ Branch 2 taken 32 times.
 176 }
1299
1300
1/2
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
 16 return "";
1301 48 }
1302
1303
1304 16 std::string GetUserName() {
1305 struct passwd pwd;
1306 16 struct passwd *result = NULL;
1307 16 int bufsize = 16 * 1024;
1308 16 char *buf = static_cast<char *>(smalloc(bufsize));
1309
2/4
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 16 times.
 16 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 16 times.
 16 if (result == NULL) {
1314 ✗ free(buf);
1315 ✗ return "";
1316 }
1317
1/2
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
 16 std::string user_name = pwd.pw_name;
1318 16 free(buf);
1319 16 return user_name;
1320 16 }
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 16 bool GetUserNameOf(uid_t uid, std::string *username) {
1344 struct passwd pwd;
1345 16 struct passwd *result = NULL;
1346 16 int bufsize = 16 * 1024;
1347 16 char *buf = static_cast<char *>(smalloc(bufsize));
1348
2/4
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 16 times.
 16 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 16 times.
 16 if (result == NULL) {
1353 ✗ free(buf);
1354 ✗ return false;
1355 }
1356
1/2
✓ Branch 0 taken 16 times.
✗ Branch 1 not taken.
 16 if (username)
1357
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 *username = result->pw_name;
1358 16 free(buf);
1359 16 return true;
1360 }
1361
1362
1363 /**
1364 * Name -> UID from passwd database
1365 */
1366 260 bool GetUidOf(const std::string &username, uid_t *uid, gid_t *main_gid) {
1367 struct passwd pwd;
1368 260 struct passwd *result = NULL;
1369 260 int bufsize = 16 * 1024;
1370 260 char *buf = static_cast<char *>(smalloc(bufsize));
1371
2/4
✓ Branch 2 taken 260 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 260 times.
 260 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 20 times.
✓ Branch 1 taken 240 times.
 260 if (result == NULL) {
1376 20 free(buf);
1377 20 return false;
1378 }
1379 240 *uid = result->pw_uid;
1380 240 *main_gid = result->pw_gid;
1381 240 free(buf);
1382 240 return true;
1383 }
1384
1385
1386 /**
1387 * Name -> GID from groups database
1388 */
1389 54 bool GetGidOf(const std::string &groupname, gid_t *gid) {
1390 struct group grp;
1391 54 struct group *result = NULL;
1392 54 int bufsize = 16 * 1024;
1393 54 char *buf = static_cast<char *>(smalloc(bufsize));
1394
2/4
✓ Branch 2 taken 54 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 54 times.
 54 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 27 times.
✓ Branch 1 taken 27 times.
 54 if (result == NULL) {
1399 27 free(buf);
1400 27 return false;
1401 }
1402 27 *gid = result->gr_gid;
1403 27 free(buf);
1404 27 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 3708 mode_t GetUmask() {
1413 3708 const MutexLockGuard m(&getumask_mutex);
1414 3708 const mode_t my_umask = umask(0);
1415 3708 umask(my_umask);
1416 7416 return my_umask;
1417 3708 }
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 58 std::string GetHomeDirectory() {
1447 58 const uid_t uid = getuid();
1448 struct passwd pwd;
1449 58 struct passwd *result = NULL;
1450 58 int bufsize = 16 * 1024;
1451 58 char *buf = static_cast<char *>(smalloc(bufsize));
1452
2/4
✓ Branch 1 taken 58 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 58 times.
 58 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 58 times.
 58 if (result == NULL) {
1457 ✗ free(buf);
1458 ✗ return "";
1459 }
1460
1/2
✓ Branch 2 taken 58 times.
✗ Branch 3 not taken.
 58 std::string home_dir = result->pw_dir;
1461 58 free(buf);
1462 58 return home_dir;
1463 58 }
1464
1465 /**
1466 * Returns the output of `uname -m`
1467 */
1468 16 std::string GetArch() {
1469 struct utsname info;
1470 16 const int retval = uname(&info);
1471
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
 16 assert(retval == 0);
1472
1/2
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
 16 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 244 int SetLimitNoFile(unsigned limit_nofile) {
1481 struct rlimit rpl;
1482 244 memset(&rpl, 0, sizeof(rpl));
1483 244 getrlimit(RLIMIT_NOFILE, &rpl);
1484
2/2
✓ Branch 0 taken 166 times.
✓ Branch 1 taken 78 times.
 244 if (rpl.rlim_max < limit_nofile)
1485 166 rpl.rlim_max = limit_nofile;
1486 244 rpl.rlim_cur = limit_nofile;
1487 244 const int retval = setrlimit(RLIMIT_NOFILE, &rpl);
1488
2/2
✓ Branch 0 taken 78 times.
✓ Branch 1 taken 166 times.
 244 if (retval == 0)
1489 78 return 0;
1490
1491 #ifdef HAS_VALGRIND_HEADERS
1492
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 166 times.
 166 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 872 void GetLimitNoFile(unsigned *soft_limit, unsigned *hard_limit) {
1503 872 *soft_limit = 0;
1504 872 *hard_limit = 0;
1505
1506 struct rlimit rpl;
1507 872 memset(&rpl, 0, sizeof(rpl));
1508 872 getrlimit(RLIMIT_NOFILE, &rpl);
1509 872 *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 872 *hard_limit = rpl.rlim_max;
1517 #endif
1518 872 }
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 16 std::vector<LsofEntry> Lsof(const std::string &path) {
1534 16 std::vector<LsofEntry> result;
1535
1536 16 std::vector<std::string> proc_names;
1537 16 std::vector<mode_t> proc_modes;
1538
2/4
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 16 times.
✗ Branch 6 not taken.
 16 ListDirectory("/proc", &proc_names, &proc_modes);
1539
1540
2/2
✓ Branch 1 taken 1024 times.
✓ Branch 2 taken 16 times.
 1040 for (unsigned i = 0; i < proc_names.size(); ++i) {
1541
2/2
✓ Branch 1 taken 784 times.
✓ Branch 2 taken 240 times.
 1024 if (!S_ISDIR(proc_modes[i]))
1542 944 continue;
1543
2/2
✓ Branch 2 taken 160 times.
✓ Branch 3 taken 80 times.
 240 if (proc_names[i].find_first_not_of("1234567890") != std::string::npos)
1544 160 continue;
1545
1546 80 std::vector<std::string> fd_names;
1547 80 std::vector<mode_t> fd_modes;
1548
1/2
✓ Branch 2 taken 80 times.
✗ Branch 3 not taken.
 80 const std::string proc_dir = "/proc/" + proc_names[i];
1549
1/2
✓ Branch 1 taken 80 times.
✗ Branch 2 not taken.
 80 const std::string fd_dir = proc_dir + "/fd";
1550
1/2
✓ Branch 1 taken 80 times.
✗ Branch 2 not taken.
 80 const bool rvb = ListDirectory(fd_dir, &fd_names, &fd_modes);
1551 80 uid_t proc_uid = 0;
1552
1553 // The working directory of the process requires special handling
1554
2/2
✓ Branch 0 taken 64 times.
✓ Branch 1 taken 16 times.
 80 if (rvb) {
1555 platform_stat64 info;
1556 64 platform_stat(proc_dir.c_str(), &info);
1557 64 proc_uid = info.st_uid;
1558
1559
2/4
✓ Branch 1 taken 64 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 64 times.
✗ Branch 5 not taken.
 64 const std::string cwd = ReadSymlink(proc_dir + "/cwd");
1560
5/9
✓ Branch 1 taken 64 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 64 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 64 times.
✗ Branch 8 not taken.
✓ Branch 11 taken 16 times.
✓ Branch 12 taken 48 times.
 64 if (HasPrefix(cwd + "/", path + "/", false /* ignore_case */)) {
1561 16 LsofEntry entry;
1562
1/2
✓ Branch 2 taken 16 times.
✗ Branch 3 not taken.
 16 entry.pid = static_cast<pid_t>(String2Uint64(proc_names[i]));
1563 16 entry.owner = proc_uid;
1564 16 entry.read_only = true; // A bit sloppy but good enough for the moment
1565
2/4
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 16 times.
✗ Branch 5 not taken.
 16 entry.executable = ReadSymlink(proc_dir + "/exe");
1566
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 entry.path = cwd;
1567
1/2
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
 16 result.push_back(entry);
1568 16 }
1569 64 }
1570
1571
2/2
✓ Branch 1 taken 640 times.
✓ Branch 2 taken 80 times.
 720 for (unsigned j = 0; j < fd_names.size(); ++j) {
1572
1/2
✗ Branch 1 not taken.
✓ Branch 2 taken 640 times.
 640 if (!S_ISLNK(fd_modes[j]))
1573 448 continue;
1574
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 640 times.
 640 if (fd_names[j].find_first_not_of("1234567890") != std::string::npos)
1575 ✗ continue;
1576
1577
3/6
✓ Branch 2 taken 640 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 640 times.
✗ Branch 6 not taken.
✓ Branch 8 taken 640 times.
✗ Branch 9 not taken.
 1280 const std::string target = ReadSymlink(fd_dir + "/" + fd_names[j]);
1578
5/9
✓ Branch 1 taken 640 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 640 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 640 times.
✗ Branch 8 not taken.
✓ Branch 11 taken 448 times.
✓ Branch 12 taken 192 times.
 640 if (!HasPrefix(target + "/", path + "/", false /* ignore_case */))
1579 448 continue;
1580
1581 192 LsofEntry entry;
1582
1/2
✓ Branch 2 taken 192 times.
✗ Branch 3 not taken.
 192 entry.pid = static_cast<pid_t>(String2Uint64(proc_names[i]));
1583 192 entry.owner = proc_uid;
1584 192 entry.read_only = !((fd_modes[j] & S_IWUSR) == S_IWUSR);
1585
2/4
✓ Branch 1 taken 192 times.
✗ Branch 2 not taken.
✓ Branch 4 taken 192 times.
✗ Branch 5 not taken.
 192 entry.executable = ReadSymlink(proc_dir + "/exe");
1586
1/2
✓ Branch 1 taken 192 times.
✗ Branch 2 not taken.
 192 entry.path = target;
1587
1/2
✓ Branch 1 taken 192 times.
✗ Branch 2 not taken.
 192 result.push_back(entry);
1588
2/2
✓ Branch 2 taken 192 times.
✓ Branch 3 taken 448 times.
 640 }
1589 80 }
1590
1591 32 return result;
1592 16 }
1593
1594
1595 42 bool ProcessExists(pid_t pid) {
1596
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
 42 assert(pid > 0);
1597 42 const int retval = kill(pid, 0);
1598
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 14 times.
 42 if (retval == 0)
1599 28 return true;
1600 14 return (errno != ESRCH);
1601 }
1602
1603
1604 /**
1605 * Blocks a signal for the calling thread.
1606 */
1607 35 void BlockSignal(int signum) {
1608 sigset_t sigset;
1609 35 int retval = sigemptyset(&sigset);
1610
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35 times.
 35 assert(retval == 0);
1611 35 retval = sigaddset(&sigset, signum);
1612
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35 times.
 35 assert(retval == 0);
1613 35 retval = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
1614
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 35 times.
 35 assert(retval == 0);
1615 35 }
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 16 void WaitForSignal(int signum) {
1623 int retval;
1624 do {
1625 16 retval = platform_sigwait(signum);
1626
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 16 } while ((retval != signum) && (errno == EINTR));
1627
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
 16 assert(retval == signum);
1628 16 }
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 157 int WaitForChild(pid_t pid, const std::vector<int> &sig_ok) {
1638
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 157 times.
 157 assert(pid > 0);
1639 int statloc;
1640 while (true) {
1641
1/2
✓ Branch 1 taken 157 times.
✗ Branch 2 not taken.
 157 const pid_t retval = waitpid(pid, &statloc, 0);
1642
2/2
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 137 times.
 157 if (retval == -1) {
1643
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
 20 if (errno == EINTR)
1644 ✗ continue;
1645 20 PANIC(kLogSyslogErr | kLogDebug, "waitpid failed with errno %d", errno);
1646 }
1647
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 137 times.
 137 assert(retval == pid);
1648 137 break;
1649 }
1650
2/2
✓ Branch 0 taken 117 times.
✓ Branch 1 taken 20 times.
 137 if (WIFEXITED(statloc))
1651 117 return WEXITSTATUS(statloc);
1652 40 if (WIFSIGNALED(statloc)
1653
3/6
✓ Branch 0 taken 20 times.
✗ Branch 1 not taken.
✓ Branch 5 taken 20 times.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✓ Branch 8 taken 20 times.
 60 && (std::find(sig_ok.begin(), sig_ok.end(), WTERMSIG(statloc))
1654
1/2
✗ Branch 2 not taken.
✓ Branch 3 taken 20 times.
 60 != sig_ok.end()))
1655 ✗ return 0;
1656 20 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 782 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 782 times.
✗ Branch 2 not taken.
 782 MakePipe(pipe_stdin);
1763
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 MakePipe(pipe_stdout);
1764
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 MakePipe(pipe_stderr);
1765
1766 782 std::set<int> preserve_fildes;
1767
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 preserve_fildes.insert(0);
1768
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 preserve_fildes.insert(1);
1769
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 preserve_fildes.insert(2);
1770 782 std::map<int, int> map_fildes;
1771
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 map_fildes[pipe_stdin[0]] = 0; // Reading end of pipe_stdin
1772
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 map_fildes[pipe_stdout[1]] = 1; // Writing end of pipe_stdout
1773
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 map_fildes[pipe_stderr[1]] = 2; // Writing end of pipe_stderr
1774 782 std::vector<std::string> cmd_line;
1775
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 cmd_line.push_back(binary_path);
1776
1/2
✓ Branch 5 taken 782 times.
✗ Branch 6 not taken.
 782 cmd_line.insert(cmd_line.end(), argv.begin(), argv.end());
1777
1778
2/4
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 782 times.
 782 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 782 times.
✗ Branch 2 not taken.
 782 close(pipe_stdin[0]);
1792
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 close(pipe_stdout[1]);
1793
1/2
✓ Branch 1 taken 782 times.
✗ Branch 2 not taken.
 782 close(pipe_stderr[1]);
1794 782 *fd_stdin = pipe_stdin[1];
1795 782 *fd_stdout = pipe_stdout[0];
1796 782 *fd_stderr = pipe_stderr[0];
1797 782 return true;
1798 782 }
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 738 bool Shell(int *fd_stdin, int *fd_stdout, int *fd_stderr) {
1806 738 const bool double_fork = true;
1807
2/4
✓ Branch 2 taken 738 times.
✗ Branch 3 not taken.
✓ Branch 5 taken 738 times.
✗ Branch 6 not taken.
 1476 return ExecuteBinary(fd_stdin, fd_stdout, fd_stderr, "/bin/sh",
1808 2214 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 31 static bool CloseAllFildesUntilMaxFD(const std::set<int> &preserve_fildes,
1851 long max_fd) { // NOLINT(runtime/int)
1852
2/2
✓ Branch 0 taken 15500 times.
✓ Branch 1 taken 31 times.
 15531 for (long fd = 0; fd < max_fd; fd++) { // NOLINT(runtime/int)
1853
3/4
✓ Branch 1 taken 15500 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 15347 times.
✓ Branch 4 taken 153 times.
 15500 if (preserve_fildes.count(static_cast<int>(fd)) == 0) {
1854 15347 close(static_cast<int>(fd));
1855 }
1856 }
1857
1858 31 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 20 static bool CloseAllFildesInFdDir(const char *fd_dir,
1867 const std::set<int> &preserve_fildes) {
1868 20 DIR *dirp = opendir(fd_dir);
1869
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 20 times.
 20 if (!dirp)
1870 ✗ return false;
1871
1872 platform_dirent64 *dirent;
1873
1874
2/2
✓ Branch 1 taken 305 times.
✓ Branch 2 taken 20 times.
 325 while ((dirent = platform_readdir(dirp))) {
1875
1/2
✓ Branch 2 taken 305 times.
✗ Branch 3 not taken.
 305 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 305 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 40 times.
✓ Branch 4 taken 265 times.
 305 if (!String2Uint64Parse(name, &name_uint64)) {
1880 40 continue;
1881 }
1882
1883 265 const int fd = static_cast<int>(name_uint64);
1884
3/4
✓ Branch 1 taken 265 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 100 times.
✓ Branch 4 taken 165 times.
 265 if (preserve_fildes.count(fd)) {
1885 100 continue;
1886 }
1887
1888
1/2
✓ Branch 1 taken 165 times.
✗ Branch 2 not taken.
 165 close(fd);
1889
2/2
✓ Branch 1 taken 165 times.
✓ Branch 2 taken 140 times.
 305 }
1890
1891 20 closedir(dirp);
1892
1893 20 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 20 times.
✓ Branch 1 taken 31 times.
 51 if (max_fd > 100000) {
1910 #ifdef __APPLE__
1911 return CloseAllFildesInFdDir("/dev/fd", preserve_fildes);
1912 #else
1913 20 return CloseAllFildesInFdDir("/proc/self/fd", preserve_fildes);
1914 #endif
1915 }
1916
1917 31 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 860 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 860 times.
 860 assert(command_line.size() >= 1);
1942
1943
1/2
✓ Branch 1 taken 860 times.
✗ Branch 2 not taken.
 860 Pipe<kPipeDetachedChild> pipe_fork;
1944 860 const pid_t pid = fork();
1945
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 862 times.
 862 assert(pid >= 0);
1946
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 860 times.
 862 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 832 times.
✓ Branch 1 taken 28 times.
 860 if (double_fork) {
2033 int statloc;
2034
1/2
✓ Branch 1 taken 832 times.
✗ Branch 2 not taken.
 832 waitpid(pid, &statloc, 0);
2035 }
2036
2037
1/2
✓ Branch 1 taken 860 times.
✗ Branch 2 not taken.
 860 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 860 times.
✗ Branch 2 not taken.
 860 pipe_fork.Read<ForkFailures::Names>(&status_code);
2042
2043
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 860 times.
 860 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 860 pid_t buf_child_pid = 0;
2053
1/2
✓ Branch 1 taken 860 times.
✗ Branch 2 not taken.
 860 pipe_fork.Read(&buf_child_pid);
2054
2/2
✓ Branch 0 taken 122 times.
✓ Branch 1 taken 738 times.
 860 if (child_pid != NULL)
2055 122 *child_pid = buf_child_pid;
2056
1/2
✓ Branch 1 taken 860 times.
✗ Branch 2 not taken.
 860 pipe_fork.CloseReadFd();
2057
1/2
✓ Branch 2 taken 860 times.
✗ Branch 3 not taken.
 860 LogCvmfs(kLogCvmfs, kLogDebug, "execve'd %s (PID: %d)",
2058 860 command_line[0].c_str(), static_cast<int>(buf_child_pid));
2059 860 return true;
2060 860 }
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 898 void SafeSleepMs(const unsigned ms) {
2068 struct timeval wait_for;
2069 898 wait_for.tv_sec = ms / 1000;
2070 898 wait_for.tv_usec = (ms % 1000) * 1000;
2071
2/2
✓ Branch 1 taken 896 times.
✓ Branch 2 taken 2 times.
 898 select(0, NULL, NULL, NULL, &wait_for);
2072 896 }
2073
2074
2075 /**
2076 * Deal with EINTR and partial writes.
2077 */
2078 7972 bool SafeWrite(int fd, const void *buf, size_t nbyte) {
2079
2/2
✓ Branch 0 taken 7972 times.
✓ Branch 1 taken 7842 times.
 15814 while (nbyte) {
2080 7972 const ssize_t retval = write(fd, buf, nbyte);
2081
2/2
✓ Branch 0 taken 130 times.
✓ Branch 1 taken 7842 times.
 7972 if (retval < 0) {
2082
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 130 times.
 130 if (errno == EINTR)
2083 ✗ continue;
2084 130 return false;
2085 }
2086
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7842 times.
 7842 assert(static_cast<size_t>(retval) <= nbyte);
2087 7842 buf = reinterpret_cast<const char *>(buf) + retval;
2088 7842 nbyte -= retval;
2089 }
2090 7842 return true;
2091 }
2092
2093 /**
2094 * The contents of the iov vector might be modified by the function.
2095 */
2096 267845 bool SafeWriteV(int fd, struct iovec *iov, unsigned iovcnt) {
2097 267845 unsigned nbytes = 0;
2098
2/2
✓ Branch 0 taken 754622 times.
✓ Branch 1 taken 267845 times.
 1022467 for (unsigned i = 0; i < iovcnt; ++i)
2099 754622 nbytes += iov[i].iov_len;
2100 267845 unsigned iov_idx = 0;
2101
2102
1/2
✓ Branch 0 taken 267845 times.
✗ Branch 1 not taken.
 267845 while (nbytes) {
2103 535690 const ssize_t retval = writev(fd, &iov[iov_idx],
2104 267845 static_cast<int>(iovcnt - iov_idx));
2105
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 267845 times.
 267845 if (retval < 0) {
2106 ✗ if (errno == EINTR)
2107 ✗ continue;
2108 ✗ return false;
2109 }
2110
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 267845 times.
 267845 assert(static_cast<size_t>(retval) <= nbytes);
2111 267845 nbytes -= retval;
2112
2113 267845 unsigned sum_written_blocks = 0;
2114 267845 while ((sum_written_blocks + iov[iov_idx].iov_len)
2115
1/2
✓ Branch 0 taken 754622 times.
✗ Branch 1 not taken.
 754622 <= static_cast<size_t>(retval)) {
2116 754622 sum_written_blocks += iov[iov_idx].iov_len;
2117 754622 iov_idx++;
2118
2/2
✓ Branch 0 taken 267845 times.
✓ Branch 1 taken 486777 times.
 754622 if (iov_idx == iovcnt) {
2119
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 267845 times.
 267845 assert(sum_written_blocks == static_cast<size_t>(retval));
2120 267845 return true;
2121 }
2122 }
2123 ✗ const unsigned offset = retval - sum_written_blocks;
2124 ✗ iov[iov_idx].iov_len -= offset;
2125 ✗ iov[iov_idx].iov_base = reinterpret_cast<char *>(iov[iov_idx].iov_base)
2126 ✗ + offset;
2127 }
2128
2129 ✗ return true;
2130 }
2131
2132
2133 /**
2134 * Deal with EINTR and partial reads.
2135 */
2136 18365068 ssize_t SafeRead(int fd, void *buf, size_t nbyte) {
2137 18365068 ssize_t total_bytes = 0;
2138
2/2
✓ Branch 0 taken 18694801 times.
✓ Branch 1 taken 18343804 times.
 37038605 while (nbyte) {
2139 18694801 const ssize_t retval = read(fd, buf, nbyte);
2140
2/2
✓ Branch 0 taken 32 times.
✓ Branch 1 taken 18718485 times.
 18718517 if (retval < 0) {
2141
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32 times.
 32 if (errno == EINTR)
2142 ✗ continue;
2143 32 return -1;
2144
2/2
✓ Branch 0 taken 44948 times.
✓ Branch 1 taken 18673537 times.
 18718485 } else if (retval == 0) {
2145 44948 return total_bytes;
2146 }
2147
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 18673537 times.
 18673537 assert(static_cast<size_t>(retval) <= nbyte);
2148 18673537 buf = reinterpret_cast<char *>(buf) + retval;
2149 18673537 nbyte -= retval;
2150 18673537 total_bytes += retval;
2151 }
2152 18343804 return total_bytes;
2153 }
2154
2155
2156 /**
2157 * Pull file contents into a string
2158 */
2159 804 bool SafeReadToString(int fd, std::string *final_result) {
2160
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 804 times.
 804 if (!final_result) {
2161 ✗ return false;
2162 }
2163
2164 804 std::string tmp_result;
2165 static const int buf_size = 4096;
2166 char buf[4096];
2167 804 ssize_t total_bytes = -1;
2168 do {
2169
1/2
✓ Branch 1 taken 8172 times.
✗ Branch 2 not taken.
 8172 total_bytes = SafeRead(fd, buf, buf_size);
2170
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 8156 times.
 8172 if (total_bytes < 0) {
2171 16 return false;
2172 }
2173
1/2
✓ Branch 1 taken 8156 times.
✗ Branch 2 not taken.
 8156 tmp_result.append(buf, total_bytes);
2174
2/2
✓ Branch 0 taken 7368 times.
✓ Branch 1 taken 788 times.
 8156 } while (total_bytes == buf_size);
2175 788 final_result->swap(tmp_result);
2176 788 return true;
2177 804 }
2178
2179 6437 bool SafeWriteToFile(const std::string &content,
2180 const std::string &path,
2181 int mode) {
2182 6437 const int fd = open(path.c_str(), O_WRONLY | O_CREAT | O_TRUNC, mode);
2183
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6437 times.
 6437 if (fd < 0)
2184 ✗ return false;
2185 6437 const bool retval = SafeWrite(fd, content.data(), content.size());
2186 6437 close(fd);
2187 6437 return retval;
2188 }
2189
2190
2191 #ifdef CVMFS_NAMESPACE_GUARD
2192 } // namespace CVMFS_NAMESPACE_GUARD
2193 #endif
2194