1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2010 Lennart Poettering
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
31 #include <sys/resource.h>
32 #include <linux/sched.h>
33 #include <sys/types.h>
37 #include <sys/ioctl.h>
39 #include <linux/tiocl.h>
44 #include <sys/prctl.h>
45 #include <sys/utsname.h>
47 #include <netinet/ip.h>
56 #include <sys/mount.h>
57 #include <linux/magic.h>
61 #include <sys/personality.h>
62 #include <sys/xattr.h>
64 #include <sys/statvfs.h>
68 #ifdef HAVE_SYS_AUXV_H
80 #include "path-util.h"
81 #include "exit-status.h"
85 #include "device-nodes.h"
90 #include "sparse-endian.h"
93 char **saved_argv = NULL;
95 static volatile unsigned cached_columns = 0;
96 static volatile unsigned cached_lines = 0;
98 size_t page_size(void) {
99 static thread_local size_t pgsz = 0;
102 if (_likely_(pgsz > 0))
105 r = sysconf(_SC_PAGESIZE);
112 bool streq_ptr(const char *a, const char *b) {
114 /* Like streq(), but tries to make sense of NULL pointers */
125 char* endswith(const char *s, const char *postfix) {
132 pl = strlen(postfix);
135 return (char*) s + sl;
140 if (memcmp(s + sl - pl, postfix, pl) != 0)
143 return (char*) s + sl - pl;
146 char* first_word(const char *s, const char *word) {
153 /* Checks if the string starts with the specified word, either
154 * followed by NUL or by whitespace. Returns a pointer to the
155 * NUL or the first character after the whitespace. */
166 if (memcmp(s, word, wl) != 0)
173 if (!strchr(WHITESPACE, *p))
176 p += strspn(p, WHITESPACE);
180 static size_t cescape_char(char c, char *buf) {
181 char * buf_old = buf;
227 /* For special chars we prefer octal over
228 * hexadecimal encoding, simply because glib's
229 * g_strescape() does the same */
230 if ((c < ' ') || (c >= 127)) {
232 *(buf++) = octchar((unsigned char) c >> 6);
233 *(buf++) = octchar((unsigned char) c >> 3);
234 *(buf++) = octchar((unsigned char) c);
240 return buf - buf_old;
243 int close_nointr(int fd) {
250 * Just ignore EINTR; a retry loop is the wrong thing to do on
253 * http://lkml.indiana.edu/hypermail/linux/kernel/0509.1/0877.html
254 * https://bugzilla.gnome.org/show_bug.cgi?id=682819
255 * http://utcc.utoronto.ca/~cks/space/blog/unix/CloseEINTR
256 * https://sites.google.com/site/michaelsafyan/software-engineering/checkforeintrwheninvokingclosethinkagain
264 int safe_close(int fd) {
267 * Like close_nointr() but cannot fail. Guarantees errno is
268 * unchanged. Is a NOP with negative fds passed, and returns
269 * -1, so that it can be used in this syntax:
271 * fd = safe_close(fd);
277 /* The kernel might return pretty much any error code
278 * via close(), but the fd will be closed anyway. The
279 * only condition we want to check for here is whether
280 * the fd was invalid at all... */
282 assert_se(close_nointr(fd) != -EBADF);
288 void close_many(const int fds[], unsigned n_fd) {
291 assert(fds || n_fd <= 0);
293 for (i = 0; i < n_fd; i++)
297 int unlink_noerrno(const char *path) {
308 int parse_boolean(const char *v) {
311 if (streq(v, "1") || strcaseeq(v, "yes") || strcaseeq(v, "y") || strcaseeq(v, "true") || strcaseeq(v, "t") || strcaseeq(v, "on"))
313 else if (streq(v, "0") || strcaseeq(v, "no") || strcaseeq(v, "n") || strcaseeq(v, "false") || strcaseeq(v, "f") || strcaseeq(v, "off"))
319 int parse_pid(const char *s, pid_t* ret_pid) {
320 unsigned long ul = 0;
327 r = safe_atolu(s, &ul);
333 if ((unsigned long) pid != ul)
343 int parse_uid(const char *s, uid_t* ret_uid) {
344 unsigned long ul = 0;
351 r = safe_atolu(s, &ul);
357 if ((unsigned long) uid != ul)
360 /* Some libc APIs use UID_INVALID as special placeholder */
361 if (uid == (uid_t) 0xFFFFFFFF)
364 /* A long time ago UIDs where 16bit, hence explicitly avoid the 16bit -1 too */
365 if (uid == (uid_t) 0xFFFF)
372 int safe_atou(const char *s, unsigned *ret_u) {
380 l = strtoul(s, &x, 0);
382 if (!x || x == s || *x || errno)
383 return errno > 0 ? -errno : -EINVAL;
385 if ((unsigned long) (unsigned) l != l)
388 *ret_u = (unsigned) l;
392 int safe_atoi(const char *s, int *ret_i) {
400 l = strtol(s, &x, 0);
402 if (!x || x == s || *x || errno)
403 return errno > 0 ? -errno : -EINVAL;
405 if ((long) (int) l != l)
412 int safe_atou8(const char *s, uint8_t *ret) {
420 l = strtoul(s, &x, 0);
422 if (!x || x == s || *x || errno)
423 return errno > 0 ? -errno : -EINVAL;
425 if ((unsigned long) (uint8_t) l != l)
432 int safe_atou16(const char *s, uint16_t *ret) {
440 l = strtoul(s, &x, 0);
442 if (!x || x == s || *x || errno)
443 return errno > 0 ? -errno : -EINVAL;
445 if ((unsigned long) (uint16_t) l != l)
452 int safe_atoi16(const char *s, int16_t *ret) {
460 l = strtol(s, &x, 0);
462 if (!x || x == s || *x || errno)
463 return errno > 0 ? -errno : -EINVAL;
465 if ((long) (int16_t) l != l)
472 int safe_atollu(const char *s, long long unsigned *ret_llu) {
474 unsigned long long l;
480 l = strtoull(s, &x, 0);
482 if (!x || x == s || *x || errno)
483 return errno ? -errno : -EINVAL;
489 int safe_atolli(const char *s, long long int *ret_lli) {
497 l = strtoll(s, &x, 0);
499 if (!x || x == s || *x || errno)
500 return errno ? -errno : -EINVAL;
506 int safe_atod(const char *s, double *ret_d) {
513 RUN_WITH_LOCALE(LC_NUMERIC_MASK, "C") {
518 if (!x || x == s || *x || errno)
519 return errno ? -errno : -EINVAL;
525 static size_t strcspn_escaped(const char *s, const char *reject) {
526 bool escaped = false;
529 for (n=0; s[n]; n++) {
532 else if (s[n] == '\\')
534 else if (strchr(reject, s[n]))
538 /* if s ends in \, return index of previous char */
542 /* Split a string into words. */
543 const char* split(const char **state, size_t *l, const char *separator, bool quoted) {
549 assert(**state == '\0');
553 current += strspn(current, separator);
559 if (quoted && strchr("\'\"", *current)) {
560 char quotechars[2] = {*current, '\0'};
562 *l = strcspn_escaped(current + 1, quotechars);
563 if (current[*l + 1] == '\0' ||
564 (current[*l + 2] && !strchr(separator, current[*l + 2]))) {
565 /* right quote missing or garbage at the end */
569 assert(current[*l + 1] == quotechars[0]);
570 *state = current++ + *l + 2;
572 *l = strcspn_escaped(current, separator);
573 if (current[*l] && !strchr(separator, current[*l])) {
574 /* unfinished escape */
578 *state = current + *l;
580 *l = strcspn(current, separator);
581 *state = current + *l;
587 int get_parent_of_pid(pid_t pid, pid_t *_ppid) {
589 _cleanup_free_ char *line = NULL;
601 p = procfs_file_alloca(pid, "stat");
602 r = read_one_line_file(p, &line);
606 /* Let's skip the pid and comm fields. The latter is enclosed
607 * in () but does not escape any () in its value, so let's
608 * skip over it manually */
610 p = strrchr(line, ')');
622 if ((long unsigned) (pid_t) ppid != ppid)
625 *_ppid = (pid_t) ppid;
630 int fchmod_umask(int fd, mode_t m) {
635 r = fchmod(fd, m & (~u)) < 0 ? -errno : 0;
641 char *truncate_nl(char *s) {
644 s[strcspn(s, NEWLINE)] = 0;
648 int get_process_state(pid_t pid) {
652 _cleanup_free_ char *line = NULL;
656 p = procfs_file_alloca(pid, "stat");
657 r = read_one_line_file(p, &line);
661 p = strrchr(line, ')');
667 if (sscanf(p, " %c", &state) != 1)
670 return (unsigned char) state;
673 int get_process_comm(pid_t pid, char **name) {
680 p = procfs_file_alloca(pid, "comm");
682 r = read_one_line_file(p, name);
689 int get_process_cmdline(pid_t pid, size_t max_length, bool comm_fallback, char **line) {
690 _cleanup_fclose_ FILE *f = NULL;
698 p = procfs_file_alloca(pid, "cmdline");
704 if (max_length == 0) {
705 size_t len = 0, allocated = 0;
707 while ((c = getc(f)) != EOF) {
709 if (!GREEDY_REALLOC(r, allocated, len+2)) {
714 r[len++] = isprint(c) ? c : ' ';
724 r = new(char, max_length);
730 while ((c = getc(f)) != EOF) {
752 size_t n = MIN(left-1, 3U);
759 /* Kernel threads have no argv[] */
761 _cleanup_free_ char *t = NULL;
769 h = get_process_comm(pid, &t);
773 r = strjoin("[", t, "]", NULL);
782 int is_kernel_thread(pid_t pid) {
794 p = procfs_file_alloca(pid, "cmdline");
799 count = fread(&c, 1, 1, f);
803 /* Kernel threads have an empty cmdline */
806 return eof ? 1 : -errno;
811 int get_process_capeff(pid_t pid, char **capeff) {
817 p = procfs_file_alloca(pid, "status");
819 return get_status_field(p, "\nCapEff:", capeff);
822 static int get_process_link_contents(const char *proc_file, char **name) {
828 r = readlink_malloc(proc_file, name);
830 return r == -ENOENT ? -ESRCH : r;
835 int get_process_exe(pid_t pid, char **name) {
842 p = procfs_file_alloca(pid, "exe");
843 r = get_process_link_contents(p, name);
847 d = endswith(*name, " (deleted)");
854 static int get_process_id(pid_t pid, const char *field, uid_t *uid) {
855 _cleanup_fclose_ FILE *f = NULL;
865 p = procfs_file_alloca(pid, "status");
870 FOREACH_LINE(line, f, return -errno) {
875 if (startswith(l, field)) {
877 l += strspn(l, WHITESPACE);
879 l[strcspn(l, WHITESPACE)] = 0;
881 return parse_uid(l, uid);
888 int get_process_uid(pid_t pid, uid_t *uid) {
889 return get_process_id(pid, "Uid:", uid);
892 int get_process_gid(pid_t pid, gid_t *gid) {
893 assert_cc(sizeof(uid_t) == sizeof(gid_t));
894 return get_process_id(pid, "Gid:", gid);
897 int get_process_cwd(pid_t pid, char **cwd) {
902 p = procfs_file_alloca(pid, "cwd");
904 return get_process_link_contents(p, cwd);
907 int get_process_root(pid_t pid, char **root) {
912 p = procfs_file_alloca(pid, "root");
914 return get_process_link_contents(p, root);
917 int get_process_environ(pid_t pid, char **env) {
918 _cleanup_fclose_ FILE *f = NULL;
919 _cleanup_free_ char *outcome = NULL;
922 size_t allocated = 0, sz = 0;
927 p = procfs_file_alloca(pid, "environ");
933 while ((c = fgetc(f)) != EOF) {
934 if (!GREEDY_REALLOC(outcome, allocated, sz + 5))
938 outcome[sz++] = '\n';
940 sz += cescape_char(c, outcome + sz);
950 char *strnappend(const char *s, const char *suffix, size_t b) {
958 return strndup(suffix, b);
967 if (b > ((size_t) -1) - a)
970 r = new(char, a+b+1);
975 memcpy(r+a, suffix, b);
981 char *strappend(const char *s, const char *suffix) {
982 return strnappend(s, suffix, suffix ? strlen(suffix) : 0);
985 int readlinkat_malloc(int fd, const char *p, char **ret) {
1000 n = readlinkat(fd, p, c, l-1);
1007 if ((size_t) n < l-1) {
1018 int readlink_malloc(const char *p, char **ret) {
1019 return readlinkat_malloc(AT_FDCWD, p, ret);
1022 int readlink_value(const char *p, char **ret) {
1023 _cleanup_free_ char *link = NULL;
1027 r = readlink_malloc(p, &link);
1031 value = basename(link);
1035 value = strdup(value);
1044 int readlink_and_make_absolute(const char *p, char **r) {
1045 _cleanup_free_ char *target = NULL;
1052 j = readlink_malloc(p, &target);
1056 k = file_in_same_dir(p, target);
1064 int readlink_and_canonicalize(const char *p, char **r) {
1071 j = readlink_and_make_absolute(p, &t);
1075 s = canonicalize_file_name(t);
1082 path_kill_slashes(*r);
1087 int reset_all_signal_handlers(void) {
1090 for (sig = 1; sig < _NSIG; sig++) {
1091 struct sigaction sa = {
1092 .sa_handler = SIG_DFL,
1093 .sa_flags = SA_RESTART,
1096 /* These two cannot be caught... */
1097 if (sig == SIGKILL || sig == SIGSTOP)
1100 /* On Linux the first two RT signals are reserved by
1101 * glibc, and sigaction() will return EINVAL for them. */
1102 if ((sigaction(sig, &sa, NULL) < 0))
1103 if (errno != EINVAL && r == 0)
1110 int reset_signal_mask(void) {
1113 if (sigemptyset(&ss) < 0)
1116 if (sigprocmask(SIG_SETMASK, &ss, NULL) < 0)
1122 char *strstrip(char *s) {
1125 /* Drops trailing whitespace. Modifies the string in
1126 * place. Returns pointer to first non-space character */
1128 s += strspn(s, WHITESPACE);
1130 for (e = strchr(s, 0); e > s; e --)
1131 if (!strchr(WHITESPACE, e[-1]))
1139 char *delete_chars(char *s, const char *bad) {
1142 /* Drops all whitespace, regardless where in the string */
1144 for (f = s, t = s; *f; f++) {
1145 if (strchr(bad, *f))
1156 char *file_in_same_dir(const char *path, const char *filename) {
1163 /* This removes the last component of path and appends
1164 * filename, unless the latter is absolute anyway or the
1167 if (path_is_absolute(filename))
1168 return strdup(filename);
1170 e = strrchr(path, '/');
1172 return strdup(filename);
1174 k = strlen(filename);
1175 ret = new(char, (e + 1 - path) + k + 1);
1179 memcpy(mempcpy(ret, path, e + 1 - path), filename, k + 1);
1183 int rmdir_parents(const char *path, const char *stop) {
1192 /* Skip trailing slashes */
1193 while (l > 0 && path[l-1] == '/')
1199 /* Skip last component */
1200 while (l > 0 && path[l-1] != '/')
1203 /* Skip trailing slashes */
1204 while (l > 0 && path[l-1] == '/')
1210 if (!(t = strndup(path, l)))
1213 if (path_startswith(stop, t)) {
1222 if (errno != ENOENT)
1229 char hexchar(int x) {
1230 static const char table[16] = "0123456789abcdef";
1232 return table[x & 15];
1235 int unhexchar(char c) {
1237 if (c >= '0' && c <= '9')
1240 if (c >= 'a' && c <= 'f')
1241 return c - 'a' + 10;
1243 if (c >= 'A' && c <= 'F')
1244 return c - 'A' + 10;
1249 char *hexmem(const void *p, size_t l) {
1253 z = r = malloc(l * 2 + 1);
1257 for (x = p; x < (const uint8_t*) p + l; x++) {
1258 *(z++) = hexchar(*x >> 4);
1259 *(z++) = hexchar(*x & 15);
1266 void *unhexmem(const char *p, size_t l) {
1272 z = r = malloc((l + 1) / 2 + 1);
1276 for (x = p; x < p + l; x += 2) {
1279 a = unhexchar(x[0]);
1281 b = unhexchar(x[1]);
1285 *(z++) = (uint8_t) a << 4 | (uint8_t) b;
1292 char octchar(int x) {
1293 return '0' + (x & 7);
1296 int unoctchar(char c) {
1298 if (c >= '0' && c <= '7')
1304 char decchar(int x) {
1305 return '0' + (x % 10);
1308 int undecchar(char c) {
1310 if (c >= '0' && c <= '9')
1316 char *cescape(const char *s) {
1322 /* Does C style string escaping. */
1324 r = new(char, strlen(s)*4 + 1);
1328 for (f = s, t = r; *f; f++)
1329 t += cescape_char(*f, t);
1336 char *cunescape_length_with_prefix(const char *s, size_t length, const char *prefix) {
1343 /* Undoes C style string escaping, and optionally prefixes it. */
1345 pl = prefix ? strlen(prefix) : 0;
1347 r = new(char, pl+length+1);
1352 memcpy(r, prefix, pl);
1354 for (f = s, t = r + pl; f < s + length; f++) {
1355 size_t remaining = s + length - f;
1356 assert(remaining > 0);
1358 if (*f != '\\') { /* a literal literal */
1363 if (--remaining == 0) { /* copy trailing backslash verbatim */
1404 /* This is an extension of the XDG syntax files */
1409 /* hexadecimal encoding */
1412 if (remaining >= 2) {
1413 a = unhexchar(f[1]);
1414 b = unhexchar(f[2]);
1417 if (a < 0 || b < 0 || (a == 0 && b == 0)) {
1418 /* Invalid escape code, let's take it literal then */
1422 *(t++) = (char) ((a << 4) | b);
1437 /* octal encoding */
1438 int a = -1, b = -1, c = -1;
1440 if (remaining >= 3) {
1441 a = unoctchar(f[0]);
1442 b = unoctchar(f[1]);
1443 c = unoctchar(f[2]);
1446 if (a < 0 || b < 0 || c < 0 || (a == 0 && b == 0 && c == 0)) {
1447 /* Invalid escape code, let's take it literal then */
1451 *(t++) = (char) ((a << 6) | (b << 3) | c);
1459 /* Invalid escape code, let's take it literal then */
1470 char *cunescape_length(const char *s, size_t length) {
1471 return cunescape_length_with_prefix(s, length, NULL);
1474 char *cunescape(const char *s) {
1477 return cunescape_length(s, strlen(s));
1480 char *xescape(const char *s, const char *bad) {
1484 /* Escapes all chars in bad, in addition to \ and all special
1485 * chars, in \xFF style escaping. May be reversed with
1488 r = new(char, strlen(s) * 4 + 1);
1492 for (f = s, t = r; *f; f++) {
1494 if ((*f < ' ') || (*f >= 127) ||
1495 (*f == '\\') || strchr(bad, *f)) {
1498 *(t++) = hexchar(*f >> 4);
1499 *(t++) = hexchar(*f);
1509 char *ascii_strlower(char *t) {
1514 for (p = t; *p; p++)
1515 if (*p >= 'A' && *p <= 'Z')
1516 *p = *p - 'A' + 'a';
1521 _pure_ static bool hidden_file_allow_backup(const char *filename) {
1525 filename[0] == '.' ||
1526 streq(filename, "lost+found") ||
1527 streq(filename, "aquota.user") ||
1528 streq(filename, "aquota.group") ||
1529 endswith(filename, ".rpmnew") ||
1530 endswith(filename, ".rpmsave") ||
1531 endswith(filename, ".rpmorig") ||
1532 endswith(filename, ".dpkg-old") ||
1533 endswith(filename, ".dpkg-new") ||
1534 endswith(filename, ".dpkg-tmp") ||
1535 endswith(filename, ".swp");
1538 bool hidden_file(const char *filename) {
1541 if (endswith(filename, "~"))
1544 return hidden_file_allow_backup(filename);
1547 int fd_nonblock(int fd, bool nonblock) {
1552 flags = fcntl(fd, F_GETFL, 0);
1557 nflags = flags | O_NONBLOCK;
1559 nflags = flags & ~O_NONBLOCK;
1561 if (nflags == flags)
1564 if (fcntl(fd, F_SETFL, nflags) < 0)
1570 int fd_cloexec(int fd, bool cloexec) {
1575 flags = fcntl(fd, F_GETFD, 0);
1580 nflags = flags | FD_CLOEXEC;
1582 nflags = flags & ~FD_CLOEXEC;
1584 if (nflags == flags)
1587 if (fcntl(fd, F_SETFD, nflags) < 0)
1593 _pure_ static bool fd_in_set(int fd, const int fdset[], unsigned n_fdset) {
1596 assert(n_fdset == 0 || fdset);
1598 for (i = 0; i < n_fdset; i++)
1605 int close_all_fds(const int except[], unsigned n_except) {
1606 _cleanup_closedir_ DIR *d = NULL;
1610 assert(n_except == 0 || except);
1612 d = opendir("/proc/self/fd");
1617 /* When /proc isn't available (for example in chroots)
1618 * the fallback is brute forcing through the fd
1621 assert_se(getrlimit(RLIMIT_NOFILE, &rl) >= 0);
1622 for (fd = 3; fd < (int) rl.rlim_max; fd ++) {
1624 if (fd_in_set(fd, except, n_except))
1627 if (close_nointr(fd) < 0)
1628 if (errno != EBADF && r == 0)
1635 while ((de = readdir(d))) {
1638 if (hidden_file(de->d_name))
1641 if (safe_atoi(de->d_name, &fd) < 0)
1642 /* Let's better ignore this, just in case */
1651 if (fd_in_set(fd, except, n_except))
1654 if (close_nointr(fd) < 0) {
1655 /* Valgrind has its own FD and doesn't want to have it closed */
1656 if (errno != EBADF && r == 0)
1664 bool chars_intersect(const char *a, const char *b) {
1667 /* Returns true if any of the chars in a are in b. */
1668 for (p = a; *p; p++)
1675 bool fstype_is_network(const char *fstype) {
1676 static const char table[] =
1690 x = startswith(fstype, "fuse.");
1694 return nulstr_contains(table, fstype);
1698 _cleanup_close_ int fd;
1700 fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
1706 TIOCL_GETKMSGREDIRECT,
1710 if (ioctl(fd, TIOCLINUX, tiocl) < 0)
1713 vt = tiocl[0] <= 0 ? 1 : tiocl[0];
1716 if (ioctl(fd, VT_ACTIVATE, vt) < 0)
1722 int read_one_char(FILE *f, char *ret, usec_t t, bool *need_nl) {
1723 struct termios old_termios, new_termios;
1724 char c, line[LINE_MAX];
1729 if (tcgetattr(fileno(f), &old_termios) >= 0) {
1730 new_termios = old_termios;
1732 new_termios.c_lflag &= ~ICANON;
1733 new_termios.c_cc[VMIN] = 1;
1734 new_termios.c_cc[VTIME] = 0;
1736 if (tcsetattr(fileno(f), TCSADRAIN, &new_termios) >= 0) {
1739 if (t != USEC_INFINITY) {
1740 if (fd_wait_for_event(fileno(f), POLLIN, t) <= 0) {
1741 tcsetattr(fileno(f), TCSADRAIN, &old_termios);
1746 k = fread(&c, 1, 1, f);
1748 tcsetattr(fileno(f), TCSADRAIN, &old_termios);
1754 *need_nl = c != '\n';
1761 if (t != USEC_INFINITY) {
1762 if (fd_wait_for_event(fileno(f), POLLIN, t) <= 0)
1767 if (!fgets(line, sizeof(line), f))
1768 return errno ? -errno : -EIO;
1772 if (strlen(line) != 1)
1782 int ask_char(char *ret, const char *replies, const char *text, ...) {
1792 bool need_nl = true;
1795 fputs(ANSI_HIGHLIGHT_ON, stdout);
1802 fputs(ANSI_HIGHLIGHT_OFF, stdout);
1806 r = read_one_char(stdin, &c, USEC_INFINITY, &need_nl);
1809 if (r == -EBADMSG) {
1810 puts("Bad input, please try again.");
1821 if (strchr(replies, c)) {
1826 puts("Read unexpected character, please try again.");
1830 int ask_string(char **ret, const char *text, ...) {
1835 char line[LINE_MAX];
1839 fputs(ANSI_HIGHLIGHT_ON, stdout);
1846 fputs(ANSI_HIGHLIGHT_OFF, stdout);
1851 if (!fgets(line, sizeof(line), stdin))
1852 return errno ? -errno : -EIO;
1854 if (!endswith(line, "\n"))
1873 int reset_terminal_fd(int fd, bool switch_to_text) {
1874 struct termios termios;
1877 /* Set terminal to some sane defaults */
1881 /* We leave locked terminal attributes untouched, so that
1882 * Plymouth may set whatever it wants to set, and we don't
1883 * interfere with that. */
1885 /* Disable exclusive mode, just in case */
1886 ioctl(fd, TIOCNXCL);
1888 /* Switch to text mode */
1890 ioctl(fd, KDSETMODE, KD_TEXT);
1892 /* Enable console unicode mode */
1893 ioctl(fd, KDSKBMODE, K_UNICODE);
1895 if (tcgetattr(fd, &termios) < 0) {
1900 /* We only reset the stuff that matters to the software. How
1901 * hardware is set up we don't touch assuming that somebody
1902 * else will do that for us */
1904 termios.c_iflag &= ~(IGNBRK | BRKINT | ISTRIP | INLCR | IGNCR | IUCLC);
1905 termios.c_iflag |= ICRNL | IMAXBEL | IUTF8;
1906 termios.c_oflag |= ONLCR;
1907 termios.c_cflag |= CREAD;
1908 termios.c_lflag = ISIG | ICANON | IEXTEN | ECHO | ECHOE | ECHOK | ECHOCTL | ECHOPRT | ECHOKE;
1910 termios.c_cc[VINTR] = 03; /* ^C */
1911 termios.c_cc[VQUIT] = 034; /* ^\ */
1912 termios.c_cc[VERASE] = 0177;
1913 termios.c_cc[VKILL] = 025; /* ^X */
1914 termios.c_cc[VEOF] = 04; /* ^D */
1915 termios.c_cc[VSTART] = 021; /* ^Q */
1916 termios.c_cc[VSTOP] = 023; /* ^S */
1917 termios.c_cc[VSUSP] = 032; /* ^Z */
1918 termios.c_cc[VLNEXT] = 026; /* ^V */
1919 termios.c_cc[VWERASE] = 027; /* ^W */
1920 termios.c_cc[VREPRINT] = 022; /* ^R */
1921 termios.c_cc[VEOL] = 0;
1922 termios.c_cc[VEOL2] = 0;
1924 termios.c_cc[VTIME] = 0;
1925 termios.c_cc[VMIN] = 1;
1927 if (tcsetattr(fd, TCSANOW, &termios) < 0)
1931 /* Just in case, flush all crap out */
1932 tcflush(fd, TCIOFLUSH);
1937 int reset_terminal(const char *name) {
1938 _cleanup_close_ int fd = -1;
1940 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
1944 return reset_terminal_fd(fd, true);
1947 int open_terminal(const char *name, int mode) {
1952 * If a TTY is in the process of being closed opening it might
1953 * cause EIO. This is horribly awful, but unlikely to be
1954 * changed in the kernel. Hence we work around this problem by
1955 * retrying a couple of times.
1957 * https://bugs.launchpad.net/ubuntu/+source/linux/+bug/554172/comments/245
1960 assert(!(mode & O_CREAT));
1963 fd = open(name, mode, 0);
1970 /* Max 1s in total */
1974 usleep(50 * USEC_PER_MSEC);
1992 int flush_fd(int fd) {
1993 struct pollfd pollfd = {
2003 r = poll(&pollfd, 1, 0);
2013 l = read(fd, buf, sizeof(buf));
2019 if (errno == EAGAIN)
2028 int acquire_terminal(
2032 bool ignore_tiocstty_eperm,
2035 int fd = -1, notify = -1, r = 0, wd = -1;
2040 /* We use inotify to be notified when the tty is closed. We
2041 * create the watch before checking if we can actually acquire
2042 * it, so that we don't lose any event.
2044 * Note: strictly speaking this actually watches for the
2045 * device being closed, it does *not* really watch whether a
2046 * tty loses its controlling process. However, unless some
2047 * rogue process uses TIOCNOTTY on /dev/tty *after* closing
2048 * its tty otherwise this will not become a problem. As long
2049 * as the administrator makes sure not configure any service
2050 * on the same tty as an untrusted user this should not be a
2051 * problem. (Which he probably should not do anyway.) */
2053 if (timeout != USEC_INFINITY)
2054 ts = now(CLOCK_MONOTONIC);
2056 if (!fail && !force) {
2057 notify = inotify_init1(IN_CLOEXEC | (timeout != USEC_INFINITY ? IN_NONBLOCK : 0));
2063 wd = inotify_add_watch(notify, name, IN_CLOSE);
2071 struct sigaction sa_old, sa_new = {
2072 .sa_handler = SIG_IGN,
2073 .sa_flags = SA_RESTART,
2077 r = flush_fd(notify);
2082 /* We pass here O_NOCTTY only so that we can check the return
2083 * value TIOCSCTTY and have a reliable way to figure out if we
2084 * successfully became the controlling process of the tty */
2085 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
2089 /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
2090 * if we already own the tty. */
2091 assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
2093 /* First, try to get the tty */
2094 if (ioctl(fd, TIOCSCTTY, force) < 0)
2097 assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
2099 /* Sometimes it makes sense to ignore TIOCSCTTY
2100 * returning EPERM, i.e. when very likely we already
2101 * are have this controlling terminal. */
2102 if (r < 0 && r == -EPERM && ignore_tiocstty_eperm)
2105 if (r < 0 && (force || fail || r != -EPERM)) {
2114 assert(notify >= 0);
2117 union inotify_event_buffer buffer;
2118 struct inotify_event *e;
2121 if (timeout != USEC_INFINITY) {
2124 n = now(CLOCK_MONOTONIC);
2125 if (ts + timeout < n) {
2130 r = fd_wait_for_event(fd, POLLIN, ts + timeout - n);
2140 l = read(notify, &buffer, sizeof(buffer));
2142 if (errno == EINTR || errno == EAGAIN)
2149 FOREACH_INOTIFY_EVENT(e, buffer, l) {
2150 if (e->wd != wd || !(e->mask & IN_CLOSE)) {
2159 /* We close the tty fd here since if the old session
2160 * ended our handle will be dead. It's important that
2161 * we do this after sleeping, so that we don't enter
2162 * an endless loop. */
2163 fd = safe_close(fd);
2168 r = reset_terminal_fd(fd, true);
2170 log_warning_errno(r, "Failed to reset terminal: %m");
2181 int release_terminal(void) {
2182 static const struct sigaction sa_new = {
2183 .sa_handler = SIG_IGN,
2184 .sa_flags = SA_RESTART,
2187 _cleanup_close_ int fd = -1;
2188 struct sigaction sa_old;
2191 fd = open("/dev/tty", O_RDWR|O_NOCTTY|O_NDELAY|O_CLOEXEC);
2195 /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
2196 * by our own TIOCNOTTY */
2197 assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
2199 if (ioctl(fd, TIOCNOTTY) < 0)
2202 assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
2207 int sigaction_many(const struct sigaction *sa, ...) {
2212 while ((sig = va_arg(ap, int)) > 0)
2213 if (sigaction(sig, sa, NULL) < 0)
2220 int ignore_signals(int sig, ...) {
2221 struct sigaction sa = {
2222 .sa_handler = SIG_IGN,
2223 .sa_flags = SA_RESTART,
2228 if (sigaction(sig, &sa, NULL) < 0)
2232 while ((sig = va_arg(ap, int)) > 0)
2233 if (sigaction(sig, &sa, NULL) < 0)
2240 int default_signals(int sig, ...) {
2241 struct sigaction sa = {
2242 .sa_handler = SIG_DFL,
2243 .sa_flags = SA_RESTART,
2248 if (sigaction(sig, &sa, NULL) < 0)
2252 while ((sig = va_arg(ap, int)) > 0)
2253 if (sigaction(sig, &sa, NULL) < 0)
2260 void safe_close_pair(int p[]) {
2264 /* Special case pairs which use the same fd in both
2266 p[0] = p[1] = safe_close(p[0]);
2270 p[0] = safe_close(p[0]);
2271 p[1] = safe_close(p[1]);
2274 ssize_t loop_read(int fd, void *buf, size_t nbytes, bool do_poll) {
2281 while (nbytes > 0) {
2284 k = read(fd, p, nbytes);
2289 if (errno == EAGAIN && do_poll) {
2291 /* We knowingly ignore any return value here,
2292 * and expect that any error/EOF is reported
2295 fd_wait_for_event(fd, POLLIN, USEC_INFINITY);
2299 return n > 0 ? n : -errno;
2313 int loop_write(int fd, const void *buf, size_t nbytes, bool do_poll) {
2314 const uint8_t *p = buf;
2321 while (nbytes > 0) {
2324 k = write(fd, p, nbytes);
2329 if (errno == EAGAIN && do_poll) {
2330 /* We knowingly ignore any return value here,
2331 * and expect that any error/EOF is reported
2334 fd_wait_for_event(fd, POLLOUT, USEC_INFINITY);
2341 if (k == 0) /* Can't really happen */
2351 int parse_size(const char *t, off_t base, off_t *size) {
2353 /* Soo, sometimes we want to parse IEC binary suffxies, and
2354 * sometimes SI decimal suffixes. This function can parse
2355 * both. Which one is the right way depends on the
2356 * context. Wikipedia suggests that SI is customary for
2357 * hardrware metrics and network speeds, while IEC is
2358 * customary for most data sizes used by software and volatile
2359 * (RAM) memory. Hence be careful which one you pick!
2361 * In either case we use just K, M, G as suffix, and not Ki,
2362 * Mi, Gi or so (as IEC would suggest). That's because that's
2363 * frickin' ugly. But this means you really need to make sure
2364 * to document which base you are parsing when you use this
2369 unsigned long long factor;
2372 static const struct table iec[] = {
2373 { "E", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2374 { "P", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2375 { "T", 1024ULL*1024ULL*1024ULL*1024ULL },
2376 { "G", 1024ULL*1024ULL*1024ULL },
2377 { "M", 1024ULL*1024ULL },
2383 static const struct table si[] = {
2384 { "E", 1000ULL*1000ULL*1000ULL*1000ULL*1000ULL*1000ULL },
2385 { "P", 1000ULL*1000ULL*1000ULL*1000ULL*1000ULL },
2386 { "T", 1000ULL*1000ULL*1000ULL*1000ULL },
2387 { "G", 1000ULL*1000ULL*1000ULL },
2388 { "M", 1000ULL*1000ULL },
2394 const struct table *table;
2396 unsigned long long r = 0;
2397 unsigned n_entries, start_pos = 0;
2400 assert(base == 1000 || base == 1024);
2405 n_entries = ELEMENTSOF(si);
2408 n_entries = ELEMENTSOF(iec);
2414 unsigned long long l2;
2420 l = strtoll(p, &e, 10);
2433 if (*e >= '0' && *e <= '9') {
2436 /* strotoull itself would accept space/+/- */
2437 l2 = strtoull(e, &e2, 10);
2439 if (errno == ERANGE)
2442 /* Ignore failure. E.g. 10.M is valid */
2449 e += strspn(e, WHITESPACE);
2451 for (i = start_pos; i < n_entries; i++)
2452 if (startswith(e, table[i].suffix)) {
2453 unsigned long long tmp;
2454 if ((unsigned long long) l + (frac > 0) > ULLONG_MAX / table[i].factor)
2456 tmp = l * table[i].factor + (unsigned long long) (frac * table[i].factor);
2457 if (tmp > ULLONG_MAX - r)
2461 if ((unsigned long long) (off_t) r != r)
2464 p = e + strlen(table[i].suffix);
2480 int make_stdio(int fd) {
2485 r = dup2(fd, STDIN_FILENO);
2486 s = dup2(fd, STDOUT_FILENO);
2487 t = dup2(fd, STDERR_FILENO);
2492 if (r < 0 || s < 0 || t < 0)
2495 /* Explicitly unset O_CLOEXEC, since if fd was < 3, then
2496 * dup2() was a NOP and the bit hence possibly set. */
2497 fd_cloexec(STDIN_FILENO, false);
2498 fd_cloexec(STDOUT_FILENO, false);
2499 fd_cloexec(STDERR_FILENO, false);
2504 int make_null_stdio(void) {
2507 null_fd = open("/dev/null", O_RDWR|O_NOCTTY);
2511 return make_stdio(null_fd);
2514 bool is_device_path(const char *path) {
2516 /* Returns true on paths that refer to a device, either in
2517 * sysfs or in /dev */
2520 path_startswith(path, "/dev/") ||
2521 path_startswith(path, "/sys/");
2524 int dir_is_empty(const char *path) {
2525 _cleanup_closedir_ DIR *d;
2536 if (!de && errno != 0)
2542 if (!hidden_file(de->d_name))
2547 char* dirname_malloc(const char *path) {
2548 char *d, *dir, *dir2;
2565 int dev_urandom(void *p, size_t n) {
2566 static int have_syscall = -1;
2570 /* Gathers some randomness from the kernel. This call will
2571 * never block, and will always return some data from the
2572 * kernel, regardless if the random pool is fully initialized
2573 * or not. It thus makes no guarantee for the quality of the
2574 * returned entropy, but is good enough for or usual usecases
2575 * of seeding the hash functions for hashtable */
2577 /* Use the getrandom() syscall unless we know we don't have
2578 * it, or when the requested size is too large for it. */
2579 if (have_syscall != 0 || (size_t) (int) n != n) {
2580 r = getrandom(p, n, GRND_NONBLOCK);
2582 have_syscall = true;
2587 if (errno == ENOSYS)
2588 /* we lack the syscall, continue with
2589 * reading from /dev/urandom */
2590 have_syscall = false;
2591 else if (errno == EAGAIN)
2592 /* not enough entropy for now. Let's
2593 * remember to use the syscall the
2594 * next time, again, but also read
2595 * from /dev/urandom for now, which
2596 * doesn't care about the current
2597 * amount of entropy. */
2598 have_syscall = true;
2602 /* too short read? */
2606 fd = open("/dev/urandom", O_RDONLY|O_CLOEXEC|O_NOCTTY);
2608 return errno == ENOENT ? -ENOSYS : -errno;
2610 k = loop_read(fd, p, n, true);
2615 if ((size_t) k != n)
2621 void initialize_srand(void) {
2622 static bool srand_called = false;
2624 #ifdef HAVE_SYS_AUXV_H
2633 #ifdef HAVE_SYS_AUXV_H
2634 /* The kernel provides us with a bit of entropy in auxv, so
2635 * let's try to make use of that to seed the pseudo-random
2636 * generator. It's better than nothing... */
2638 auxv = (void*) getauxval(AT_RANDOM);
2640 x ^= *(unsigned*) auxv;
2643 x ^= (unsigned) now(CLOCK_REALTIME);
2644 x ^= (unsigned) gettid();
2647 srand_called = true;
2650 void random_bytes(void *p, size_t n) {
2654 r = dev_urandom(p, n);
2658 /* If some idiot made /dev/urandom unavailable to us, he'll
2659 * get a PRNG instead. */
2663 for (q = p; q < (uint8_t*) p + n; q ++)
2667 void rename_process(const char name[8]) {
2670 /* This is a like a poor man's setproctitle(). It changes the
2671 * comm field, argv[0], and also the glibc's internally used
2672 * name of the process. For the first one a limit of 16 chars
2673 * applies, to the second one usually one of 10 (i.e. length
2674 * of "/sbin/init"), to the third one one of 7 (i.e. length of
2675 * "systemd"). If you pass a longer string it will be
2678 prctl(PR_SET_NAME, name);
2680 if (program_invocation_name)
2681 strncpy(program_invocation_name, name, strlen(program_invocation_name));
2683 if (saved_argc > 0) {
2687 strncpy(saved_argv[0], name, strlen(saved_argv[0]));
2689 for (i = 1; i < saved_argc; i++) {
2693 memzero(saved_argv[i], strlen(saved_argv[i]));
2698 void sigset_add_many(sigset_t *ss, ...) {
2705 while ((sig = va_arg(ap, int)) > 0)
2706 assert_se(sigaddset(ss, sig) == 0);
2710 int sigprocmask_many(int how, ...) {
2715 assert_se(sigemptyset(&ss) == 0);
2718 while ((sig = va_arg(ap, int)) > 0)
2719 assert_se(sigaddset(&ss, sig) == 0);
2722 if (sigprocmask(how, &ss, NULL) < 0)
2728 char* gethostname_malloc(void) {
2731 assert_se(uname(&u) >= 0);
2733 if (!isempty(u.nodename) && !streq(u.nodename, "(none)"))
2734 return strdup(u.nodename);
2736 return strdup(u.sysname);
2739 bool hostname_is_set(void) {
2742 assert_se(uname(&u) >= 0);
2744 return !isempty(u.nodename) && !streq(u.nodename, "(none)");
2747 char *lookup_uid(uid_t uid) {
2750 _cleanup_free_ char *buf = NULL;
2751 struct passwd pwbuf, *pw = NULL;
2753 /* Shortcut things to avoid NSS lookups */
2755 return strdup("root");
2757 bufsize = sysconf(_SC_GETPW_R_SIZE_MAX);
2761 buf = malloc(bufsize);
2765 if (getpwuid_r(uid, &pwbuf, buf, bufsize, &pw) == 0 && pw)
2766 return strdup(pw->pw_name);
2768 if (asprintf(&name, UID_FMT, uid) < 0)
2774 char* getlogname_malloc(void) {
2778 if (isatty(STDIN_FILENO) && fstat(STDIN_FILENO, &st) >= 0)
2783 return lookup_uid(uid);
2786 char *getusername_malloc(void) {
2793 return lookup_uid(getuid());
2796 int getttyname_malloc(int fd, char **ret) {
2806 r = ttyname_r(fd, path, sizeof(path));
2811 p = startswith(path, "/dev/");
2812 c = strdup(p ?: path);
2829 int getttyname_harder(int fd, char **r) {
2833 k = getttyname_malloc(fd, &s);
2837 if (streq(s, "tty")) {
2839 return get_ctty(0, NULL, r);
2846 int get_ctty_devnr(pid_t pid, dev_t *d) {
2848 _cleanup_free_ char *line = NULL;
2850 unsigned long ttynr;
2854 p = procfs_file_alloca(pid, "stat");
2855 r = read_one_line_file(p, &line);
2859 p = strrchr(line, ')');
2869 "%*d " /* session */
2874 if (major(ttynr) == 0 && minor(ttynr) == 0)
2883 int get_ctty(pid_t pid, dev_t *_devnr, char **r) {
2884 char fn[sizeof("/dev/char/")-1 + 2*DECIMAL_STR_MAX(unsigned) + 1 + 1], *b = NULL;
2885 _cleanup_free_ char *s = NULL;
2892 k = get_ctty_devnr(pid, &devnr);
2896 sprintf(fn, "/dev/char/%u:%u", major(devnr), minor(devnr));
2898 k = readlink_malloc(fn, &s);
2904 /* This is an ugly hack */
2905 if (major(devnr) == 136) {
2906 asprintf(&b, "pts/%u", minor(devnr));
2910 /* Probably something like the ptys which have no
2911 * symlink in /dev/char. Let's return something
2912 * vaguely useful. */
2918 if (startswith(s, "/dev/"))
2920 else if (startswith(s, "../"))
2938 int rm_rf_children_dangerous(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
2939 _cleanup_closedir_ DIR *d = NULL;
2944 /* This returns the first error we run into, but nevertheless
2945 * tries to go on. This closes the passed fd. */
2951 return errno == ENOENT ? 0 : -errno;
2956 bool is_dir, keep_around;
2963 if (errno != 0 && ret == 0)
2968 if (streq(de->d_name, ".") || streq(de->d_name, ".."))
2971 if (de->d_type == DT_UNKNOWN ||
2973 (de->d_type == DT_DIR && root_dev)) {
2974 if (fstatat(fd, de->d_name, &st, AT_SYMLINK_NOFOLLOW) < 0) {
2975 if (ret == 0 && errno != ENOENT)
2980 is_dir = S_ISDIR(st.st_mode);
2983 (st.st_uid == 0 || st.st_uid == getuid()) &&
2984 (st.st_mode & S_ISVTX);
2986 is_dir = de->d_type == DT_DIR;
2987 keep_around = false;
2993 /* if root_dev is set, remove subdirectories only, if device is same as dir */
2994 if (root_dev && st.st_dev != root_dev->st_dev)
2997 subdir_fd = openat(fd, de->d_name,
2998 O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
2999 if (subdir_fd < 0) {
3000 if (ret == 0 && errno != ENOENT)
3005 r = rm_rf_children_dangerous(subdir_fd, only_dirs, honour_sticky, root_dev);
3006 if (r < 0 && ret == 0)
3010 if (unlinkat(fd, de->d_name, AT_REMOVEDIR) < 0) {
3011 if (ret == 0 && errno != ENOENT)
3015 } else if (!only_dirs && !keep_around) {
3017 if (unlinkat(fd, de->d_name, 0) < 0) {
3018 if (ret == 0 && errno != ENOENT)
3025 _pure_ static int is_temporary_fs(struct statfs *s) {
3028 return F_TYPE_EQUAL(s->f_type, TMPFS_MAGIC) ||
3029 F_TYPE_EQUAL(s->f_type, RAMFS_MAGIC);
3032 int is_fd_on_temporary_fs(int fd) {
3035 if (fstatfs(fd, &s) < 0)
3038 return is_temporary_fs(&s);
3041 int rm_rf_children(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
3046 if (fstatfs(fd, &s) < 0) {
3051 /* We refuse to clean disk file systems with this call. This
3052 * is extra paranoia just to be sure we never ever remove
3054 if (!is_temporary_fs(&s)) {
3055 log_error("Attempted to remove disk file system, and we can't allow that.");
3060 return rm_rf_children_dangerous(fd, only_dirs, honour_sticky, root_dev);
3063 static int file_is_priv_sticky(const char *p) {
3068 if (lstat(p, &st) < 0)
3072 (st.st_uid == 0 || st.st_uid == getuid()) &&
3073 (st.st_mode & S_ISVTX);
3076 static int rm_rf_internal(const char *path, bool only_dirs, bool delete_root, bool honour_sticky, bool dangerous) {
3082 /* We refuse to clean the root file system with this
3083 * call. This is extra paranoia to never cause a really
3084 * seriously broken system. */
3085 if (path_equal(path, "/")) {
3086 log_error("Attempted to remove entire root file system, and we can't allow that.");
3090 fd = open(path, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
3093 if (errno != ENOTDIR && errno != ELOOP)
3097 if (statfs(path, &s) < 0)
3100 if (!is_temporary_fs(&s)) {
3101 log_error("Attempted to remove disk file system, and we can't allow that.");
3106 if (delete_root && !only_dirs)
3107 if (unlink(path) < 0 && errno != ENOENT)
3114 if (fstatfs(fd, &s) < 0) {
3119 if (!is_temporary_fs(&s)) {
3120 log_error("Attempted to remove disk file system, and we can't allow that.");
3126 r = rm_rf_children_dangerous(fd, only_dirs, honour_sticky, NULL);
3129 if (honour_sticky && file_is_priv_sticky(path) > 0)
3132 if (rmdir(path) < 0 && errno != ENOENT) {
3141 int rm_rf(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
3142 return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, false);
3145 int rm_rf_dangerous(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
3146 return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, true);
3149 int chmod_and_chown(const char *path, mode_t mode, uid_t uid, gid_t gid) {
3152 /* Under the assumption that we are running privileged we
3153 * first change the access mode and only then hand out
3154 * ownership to avoid a window where access is too open. */
3156 if (mode != MODE_INVALID)
3157 if (chmod(path, mode) < 0)
3160 if (uid != UID_INVALID || gid != GID_INVALID)
3161 if (chown(path, uid, gid) < 0)
3167 int fchmod_and_fchown(int fd, mode_t mode, uid_t uid, gid_t gid) {
3170 /* Under the assumption that we are running privileged we
3171 * first change the access mode and only then hand out
3172 * ownership to avoid a window where access is too open. */
3174 if (mode != MODE_INVALID)
3175 if (fchmod(fd, mode) < 0)
3178 if (uid != UID_INVALID || gid != GID_INVALID)
3179 if (fchown(fd, uid, gid) < 0)
3185 cpu_set_t* cpu_set_malloc(unsigned *ncpus) {
3189 /* Allocates the cpuset in the right size */
3192 if (!(r = CPU_ALLOC(n)))
3195 if (sched_getaffinity(0, CPU_ALLOC_SIZE(n), r) >= 0) {
3196 CPU_ZERO_S(CPU_ALLOC_SIZE(n), r);
3206 if (errno != EINVAL)
3213 int status_vprintf(const char *status, bool ellipse, bool ephemeral, const char *format, va_list ap) {
3214 static const char status_indent[] = " "; /* "[" STATUS "] " */
3215 _cleanup_free_ char *s = NULL;
3216 _cleanup_close_ int fd = -1;
3217 struct iovec iovec[6] = {};
3219 static bool prev_ephemeral;
3223 /* This is independent of logging, as status messages are
3224 * optional and go exclusively to the console. */
3226 if (vasprintf(&s, format, ap) < 0)
3229 fd = open_terminal("/dev/console", O_WRONLY|O_NOCTTY|O_CLOEXEC);
3242 sl = status ? sizeof(status_indent)-1 : 0;
3248 e = ellipsize(s, emax, 50);
3256 IOVEC_SET_STRING(iovec[n++], "\r" ANSI_ERASE_TO_END_OF_LINE);
3257 prev_ephemeral = ephemeral;
3260 if (!isempty(status)) {
3261 IOVEC_SET_STRING(iovec[n++], "[");
3262 IOVEC_SET_STRING(iovec[n++], status);
3263 IOVEC_SET_STRING(iovec[n++], "] ");
3265 IOVEC_SET_STRING(iovec[n++], status_indent);
3268 IOVEC_SET_STRING(iovec[n++], s);
3270 IOVEC_SET_STRING(iovec[n++], "\n");
3272 if (writev(fd, iovec, n) < 0)
3278 int status_printf(const char *status, bool ellipse, bool ephemeral, const char *format, ...) {
3284 va_start(ap, format);
3285 r = status_vprintf(status, ellipse, ephemeral, format, ap);
3291 char *replace_env(const char *format, char **env) {
3298 const char *e, *word = format;
3303 for (e = format; *e; e ++) {
3314 k = strnappend(r, word, e-word-1);
3324 } else if (*e == '$') {
3325 k = strnappend(r, word, e-word);
3342 t = strempty(strv_env_get_n(env, word+2, e-word-2));
3344 k = strappend(r, t);
3358 k = strnappend(r, word, e-word);
3370 char **replace_env_argv(char **argv, char **env) {
3372 unsigned k = 0, l = 0;
3374 l = strv_length(argv);
3376 ret = new(char*, l+1);
3380 STRV_FOREACH(i, argv) {
3382 /* If $FOO appears as single word, replace it by the split up variable */
3383 if ((*i)[0] == '$' && (*i)[1] != '{') {
3388 e = strv_env_get(env, *i+1);
3392 r = strv_split_quoted(&m, e, true);