src/shared/util.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   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.
  12
  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.
  17
  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/>.
  20 ***/
  21
  22 #include <assert.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25 #include <errno.h>
  26 #include <stdlib.h>
  27 #include <signal.h>
  28 #include <stdio.h>
  29 #include <syslog.h>
  30 #include <sched.h>
  31 #include <sys/resource.h>
  32 #include <linux/sched.h>
  33 #include <sys/types.h>
  34 #include <sys/stat.h>
  35 #include <fcntl.h>
  36 #include <dirent.h>
  37 #include <sys/ioctl.h>
  38 #include <linux/vt.h>
  39 #include <linux/tiocl.h>
  40 #include <termios.h>
  41 #include <stdarg.h>
  42 #include <sys/inotify.h>
  43 #include <sys/poll.h>
  44 #include <ctype.h>
  45 #include <sys/prctl.h>
  46 #include <sys/utsname.h>
  47 #include <pwd.h>
  48 #include <netinet/ip.h>
  49 #include <linux/kd.h>
  50 #include <dlfcn.h>
  51 #include <sys/wait.h>
  52 #include <sys/time.h>
  53 #include <glob.h>
  54 #include <grp.h>
  55 #include <sys/mman.h>
  56 #include <sys/vfs.h>
  57 #include <sys/mount.h>
  58 #include <linux/magic.h>
  59 #include <limits.h>
  60 #include <langinfo.h>
  61 #include <locale.h>
  62 #include <sys/personality.h>
  63 #include <libgen.h>
  64 #undef basename
  65
  66 #ifdef HAVE_SYS_AUXV_H
  67 #include <sys/auxv.h>
  68 #endif
  69
  70 #include "macro.h"
  71 #include "util.h"
  72 #include "ioprio.h"
  73 #include "missing.h"
  74 #include "log.h"
  75 #include "strv.h"
  76 #include "label.h"
  77 #include "mkdir.h"
  78 #include "path-util.h"
  79 #include "exit-status.h"
  80 #include "hashmap.h"
  81 #include "env-util.h"
  82 #include "fileio.h"
  83 #include "device-nodes.h"
  84 #include "utf8.h"
  85 #include "gunicode.h"
  86 #include "virt.h"
  87 #include "def.h"
  88
  89 int saved_argc = 0;
  90 char **saved_argv = NULL;
  91
  92 static volatile unsigned cached_columns = 0;
  93 static volatile unsigned cached_lines = 0;
  94
  95 size_t page_size(void) {
  96         static thread_local size_t pgsz = 0;
  97         long r;
  98
  99         if (_likely_(pgsz > 0))
 100                 return pgsz;
 101
 102         r = sysconf(_SC_PAGESIZE);
 103         assert(r > 0);
 104
 105         pgsz = (size_t) r;
 106         return pgsz;
 107 }
 108
 109 bool streq_ptr(const char *a, const char *b) {
 110
 111         /* Like streq(), but tries to make sense of NULL pointers */
 112
 113         if (a && b)
 114                 return streq(a, b);
 115
 116         if (!a && !b)
 117                 return true;
 118
 119         return false;
 120 }
 121
 122 char* endswith(const char *s, const char *postfix) {
 123         size_t sl, pl;
 124
 125         assert(s);
 126         assert(postfix);
 127
 128         sl = strlen(s);
 129         pl = strlen(postfix);
 130
 131         if (pl == 0)
 132                 return (char*) s + sl;
 133
 134         if (sl < pl)
 135                 return NULL;
 136
 137         if (memcmp(s + sl - pl, postfix, pl) != 0)
 138                 return NULL;
 139
 140         return (char*) s + sl - pl;
 141 }
 142
 143 char* first_word(const char *s, const char *word) {
 144         size_t sl, wl;
 145         const char *p;
 146
 147         assert(s);
 148         assert(word);
 149
 150         /* Checks if the string starts with the specified word, either
 151          * followed by NUL or by whitespace. Returns a pointer to the
 152          * NUL or the first character after the whitespace. */
 153
 154         sl = strlen(s);
 155         wl = strlen(word);
 156
 157         if (sl < wl)
 158                 return NULL;
 159
 160         if (wl == 0)
 161                 return (char*) s;
 162
 163         if (memcmp(s, word, wl) != 0)
 164                 return NULL;
 165
 166         p = s + wl;
 167         if (*p == 0)
 168                 return (char*) p;
 169
 170         if (!strchr(WHITESPACE, *p))
 171                 return NULL;
 172
 173         p += strspn(p, WHITESPACE);
 174         return (char*) p;
 175 }
 176
 177 int close_nointr(int fd) {
 178         assert(fd >= 0);
 179
 180         if (close(fd) >= 0)
 181                 return 0;
 182
 183         /*
 184          * Just ignore EINTR; a retry loop is the wrong thing to do on
 185          * Linux.
 186          *
 187          * http://lkml.indiana.edu/hypermail/linux/kernel/0509.1/0877.html
 188          * https://bugzilla.gnome.org/show_bug.cgi?id=682819
 189          * http://utcc.utoronto.ca/~cks/space/blog/unix/CloseEINTR
 190          * https://sites.google.com/site/michaelsafyan/software-engineering/checkforeintrwheninvokingclosethinkagain
 191          */
 192         if (errno == EINTR)
 193                 return 0;
 194
 195         return -errno;
 196 }
 197
 198 int safe_close(int fd) {
 199
 200         /*
 201          * Like close_nointr() but cannot fail. Guarantees errno is
 202          * unchanged. Is a NOP with negative fds passed, and returns
 203          * -1, so that it can be used in this syntax:
 204          *
 205          * fd = safe_close(fd);
 206          */
 207
 208         if (fd >= 0) {
 209                 PROTECT_ERRNO;
 210
 211                 /* The kernel might return pretty much any error code
 212                  * via close(), but the fd will be closed anyway. The
 213                  * only condition we want to check for here is whether
 214                  * the fd was invalid at all... */
 215
 216                 assert_se(close_nointr(fd) != -EBADF);
 217         }
 218
 219         return -1;
 220 }
 221
 222 void close_many(const int fds[], unsigned n_fd) {
 223         unsigned i;
 224
 225         assert(fds || n_fd <= 0);
 226
 227         for (i = 0; i < n_fd; i++)
 228                 safe_close(fds[i]);
 229 }
 230
 231 int unlink_noerrno(const char *path) {
 232         PROTECT_ERRNO;
 233         int r;
 234
 235         r = unlink(path);
 236         if (r < 0)
 237                 return -errno;
 238
 239         return 0;
 240 }
 241
 242 int parse_boolean(const char *v) {
 243         assert(v);
 244
 245         if (streq(v, "1") || strcaseeq(v, "yes") || strcaseeq(v, "y") || strcaseeq(v, "true") || strcaseeq(v, "t") || strcaseeq(v, "on"))
 246                 return 1;
 247         else if (streq(v, "0") || strcaseeq(v, "no") || strcaseeq(v, "n") || strcaseeq(v, "false") || strcaseeq(v, "f") || strcaseeq(v, "off"))
 248                 return 0;
 249
 250         return -EINVAL;
 251 }
 252
 253 int parse_pid(const char *s, pid_t* ret_pid) {
 254         unsigned long ul = 0;
 255         pid_t pid;
 256         int r;
 257
 258         assert(s);
 259         assert(ret_pid);
 260
 261         r = safe_atolu(s, &ul);
 262         if (r < 0)
 263                 return r;
 264
 265         pid = (pid_t) ul;
 266
 267         if ((unsigned long) pid != ul)
 268                 return -ERANGE;
 269
 270         if (pid <= 0)
 271                 return -ERANGE;
 272
 273         *ret_pid = pid;
 274         return 0;
 275 }
 276
 277 int parse_uid(const char *s, uid_t* ret_uid) {
 278         unsigned long ul = 0;
 279         uid_t uid;
 280         int r;
 281
 282         assert(s);
 283         assert(ret_uid);
 284
 285         r = safe_atolu(s, &ul);
 286         if (r < 0)
 287                 return r;
 288
 289         uid = (uid_t) ul;
 290
 291         if ((unsigned long) uid != ul)
 292                 return -ERANGE;
 293
 294         /* Some libc APIs use (uid_t) -1 as special placeholder */
 295         if (uid == (uid_t) 0xFFFFFFFF)
 296                 return -ENXIO;
 297
 298         /* A long time ago UIDs where 16bit, hence explicitly avoid the 16bit -1 too */
 299         if (uid == (uid_t) 0xFFFF)
 300                 return -ENXIO;
 301
 302         *ret_uid = uid;
 303         return 0;
 304 }
 305
 306 int safe_atou(const char *s, unsigned *ret_u) {
 307         char *x = NULL;
 308         unsigned long l;
 309
 310         assert(s);
 311         assert(ret_u);
 312
 313         errno = 0;
 314         l = strtoul(s, &x, 0);
 315
 316         if (!x || x == s || *x || errno)
 317                 return errno > 0 ? -errno : -EINVAL;
 318
 319         if ((unsigned long) (unsigned) l != l)
 320                 return -ERANGE;
 321
 322         *ret_u = (unsigned) l;
 323         return 0;
 324 }
 325
 326 int safe_atoi(const char *s, int *ret_i) {
 327         char *x = NULL;
 328         long l;
 329
 330         assert(s);
 331         assert(ret_i);
 332
 333         errno = 0;
 334         l = strtol(s, &x, 0);
 335
 336         if (!x || x == s || *x || errno)
 337                 return errno > 0 ? -errno : -EINVAL;
 338
 339         if ((long) (int) l != l)
 340                 return -ERANGE;
 341
 342         *ret_i = (int) l;
 343         return 0;
 344 }
 345
 346 int safe_atou8(const char *s, uint8_t *ret) {
 347         char *x = NULL;
 348         unsigned long l;
 349
 350         assert(s);
 351         assert(ret);
 352
 353         errno = 0;
 354         l = strtoul(s, &x, 0);
 355
 356         if (!x || x == s || *x || errno)
 357                 return errno > 0 ? -errno : -EINVAL;
 358
 359         if ((unsigned long) (uint8_t) l != l)
 360                 return -ERANGE;
 361
 362         *ret = (uint8_t) l;
 363         return 0;
 364 }
 365
 366 int safe_atollu(const char *s, long long unsigned *ret_llu) {
 367         char *x = NULL;
 368         unsigned long long l;
 369
 370         assert(s);
 371         assert(ret_llu);
 372
 373         errno = 0;
 374         l = strtoull(s, &x, 0);
 375
 376         if (!x || x == s || *x || errno)
 377                 return errno ? -errno : -EINVAL;
 378
 379         *ret_llu = l;
 380         return 0;
 381 }
 382
 383 int safe_atolli(const char *s, long long int *ret_lli) {
 384         char *x = NULL;
 385         long long l;
 386
 387         assert(s);
 388         assert(ret_lli);
 389
 390         errno = 0;
 391         l = strtoll(s, &x, 0);
 392
 393         if (!x || x == s || *x || errno)
 394                 return errno ? -errno : -EINVAL;
 395
 396         *ret_lli = l;
 397         return 0;
 398 }
 399
 400 int safe_atod(const char *s, double *ret_d) {
 401         char *x = NULL;
 402         double d = 0;
 403
 404         assert(s);
 405         assert(ret_d);
 406
 407         RUN_WITH_LOCALE(LC_NUMERIC_MASK, "C") {
 408                 errno = 0;
 409                 d = strtod(s, &x);
 410         }
 411
 412         if (!x || x == s || *x || errno)
 413                 return errno ? -errno : -EINVAL;
 414
 415         *ret_d = (double) d;
 416         return 0;
 417 }
 418
 419 static size_t strcspn_escaped(const char *s, const char *reject) {
 420         bool escaped = false;
 421         size_t n;
 422
 423         for (n=0; s[n]; n++) {
 424                 if (escaped)
 425                         escaped = false;
 426                 else if (s[n] == '\\')
 427                         escaped = true;
 428                 else if (strchr(reject, s[n]))
 429                         break;
 430         }
 431         /* if s ends in \, return index of previous char */
 432         return n - escaped;
 433 }
 434
 435 /* Split a string into words. */
 436 const char* split(const char **state, size_t *l, const char *separator, bool quoted) {
 437         const char *current;
 438
 439         current = *state;
 440
 441         if (!*current) {
 442                 assert(**state == '\0');
 443                 return NULL;
 444         }
 445
 446         current += strspn(current, separator);
 447         if (!*current) {
 448                 *state = current;
 449                 return NULL;
 450         }
 451
 452         if (quoted && strchr("\'\"", *current)) {
 453                 char quotechars[2] = {*current, '\0'};
 454
 455                 *l = strcspn_escaped(current + 1, quotechars);
 456                 if (current[*l + 1] == '\0' ||
 457                     (current[*l + 2] && !strchr(separator, current[*l + 2]))) {
 458                         /* right quote missing or garbage at the end*/
 459                         *state = current;
 460                         return NULL;
 461                 }
 462                 assert(current[*l + 1] == quotechars[0]);
 463                 *state = current++ + *l + 2;
 464         } else if (quoted) {
 465                 *l = strcspn_escaped(current, separator);
 466                 *state = current + *l;
 467         } else {
 468                 *l = strcspn(current, separator);
 469                 *state = current + *l;
 470         }
 471
 472         return current;
 473 }
 474
 475 int get_parent_of_pid(pid_t pid, pid_t *_ppid) {
 476         int r;
 477         _cleanup_free_ char *line = NULL;
 478         long unsigned ppid;
 479         const char *p;
 480
 481         assert(pid >= 0);
 482         assert(_ppid);
 483
 484         if (pid == 0) {
 485                 *_ppid = getppid();
 486                 return 0;
 487         }
 488
 489         p = procfs_file_alloca(pid, "stat");
 490         r = read_one_line_file(p, &line);
 491         if (r < 0)
 492                 return r;
 493
 494         /* Let's skip the pid and comm fields. The latter is enclosed
 495          * in () but does not escape any () in its value, so let's
 496          * skip over it manually */
 497
 498         p = strrchr(line, ')');
 499         if (!p)
 500                 return -EIO;
 501
 502         p++;
 503
 504         if (sscanf(p, " "
 505                    "%*c "  /* state */
 506                    "%lu ", /* ppid */
 507                    &ppid) != 1)
 508                 return -EIO;
 509
 510         if ((long unsigned) (pid_t) ppid != ppid)
 511                 return -ERANGE;
 512
 513         *_ppid = (pid_t) ppid;
 514
 515         return 0;
 516 }
 517
 518 int get_starttime_of_pid(pid_t pid, unsigned long long *st) {
 519         int r;
 520         _cleanup_free_ char *line = NULL;
 521         const char *p;
 522
 523         assert(pid >= 0);
 524         assert(st);
 525
 526         p = procfs_file_alloca(pid, "stat");
 527         r = read_one_line_file(p, &line);
 528         if (r < 0)
 529                 return r;
 530
 531         /* Let's skip the pid and comm fields. The latter is enclosed
 532          * in () but does not escape any () in its value, so let's
 533          * skip over it manually */
 534
 535         p = strrchr(line, ')');
 536         if (!p)
 537                 return -EIO;
 538
 539         p++;
 540
 541         if (sscanf(p, " "
 542                    "%*c "  /* state */
 543                    "%*d "  /* ppid */
 544                    "%*d "  /* pgrp */
 545                    "%*d "  /* session */
 546                    "%*d "  /* tty_nr */
 547                    "%*d "  /* tpgid */
 548                    "%*u "  /* flags */
 549                    "%*u "  /* minflt */
 550                    "%*u "  /* cminflt */
 551                    "%*u "  /* majflt */
 552                    "%*u "  /* cmajflt */
 553                    "%*u "  /* utime */
 554                    "%*u "  /* stime */
 555                    "%*d "  /* cutime */
 556                    "%*d "  /* cstime */
 557                    "%*d "  /* priority */
 558                    "%*d "  /* nice */
 559                    "%*d "  /* num_threads */
 560                    "%*d "  /* itrealvalue */
 561                    "%llu "  /* starttime */,
 562                    st) != 1)
 563                 return -EIO;
 564
 565         return 0;
 566 }
 567
 568 int fchmod_umask(int fd, mode_t m) {
 569         mode_t u;
 570         int r;
 571
 572         u = umask(0777);
 573         r = fchmod(fd, m & (~u)) < 0 ? -errno : 0;
 574         umask(u);
 575
 576         return r;
 577 }
 578
 579 char *truncate_nl(char *s) {
 580         assert(s);
 581
 582         s[strcspn(s, NEWLINE)] = 0;
 583         return s;
 584 }
 585
 586 int get_process_state(pid_t pid) {
 587         const char *p;
 588         char state;
 589         int r;
 590         _cleanup_free_ char *line = NULL;
 591
 592         assert(pid >= 0);
 593
 594         p = procfs_file_alloca(pid, "stat");
 595         r = read_one_line_file(p, &line);
 596         if (r < 0)
 597                 return r;
 598
 599         p = strrchr(line, ')');
 600         if (!p)
 601                 return -EIO;
 602
 603         p++;
 604
 605         if (sscanf(p, " %c", &state) != 1)
 606                 return -EIO;
 607
 608         return (unsigned char) state;
 609 }
 610
 611 int get_process_comm(pid_t pid, char **name) {
 612         const char *p;
 613         int r;
 614
 615         assert(name);
 616         assert(pid >= 0);
 617
 618         p = procfs_file_alloca(pid, "comm");
 619
 620         r = read_one_line_file(p, name);
 621         if (r == -ENOENT)
 622                 return -ESRCH;
 623
 624         return r;
 625 }
 626
 627 int get_process_cmdline(pid_t pid, size_t max_length, bool comm_fallback, char **line) {
 628         _cleanup_fclose_ FILE *f = NULL;
 629         char *r = NULL, *k;
 630         const char *p;
 631         int c;
 632
 633         assert(line);
 634         assert(pid >= 0);
 635
 636         p = procfs_file_alloca(pid, "cmdline");
 637
 638         f = fopen(p, "re");
 639         if (!f)
 640                 return -errno;
 641
 642         if (max_length == 0) {
 643                 size_t len = 0, allocated = 0;
 644
 645                 while ((c = getc(f)) != EOF) {
 646
 647                         if (!GREEDY_REALLOC(r, allocated, len+2)) {
 648                                 free(r);
 649                                 return -ENOMEM;
 650                         }
 651
 652                         r[len++] = isprint(c) ? c : ' ';
 653                 }
 654
 655                 if (len > 0)
 656                         r[len-1] = 0;
 657
 658         } else {
 659                 bool space = false;
 660                 size_t left;
 661
 662                 r = new(char, max_length);
 663                 if (!r)
 664                         return -ENOMEM;
 665
 666                 k = r;
 667                 left = max_length;
 668                 while ((c = getc(f)) != EOF) {
 669
 670                         if (isprint(c)) {
 671                                 if (space) {
 672                                         if (left <= 4)
 673                                                 break;
 674
 675                                         *(k++) = ' ';
 676                                         left--;
 677                                         space = false;
 678                                 }
 679
 680                                 if (left <= 4)
 681                                         break;
 682
 683                                 *(k++) = (char) c;
 684                                 left--;
 685                         }  else
 686                                 space = true;
 687                 }
 688
 689                 if (left <= 4) {
 690                         size_t n = MIN(left-1, 3U);
 691                         memcpy(k, "...", n);
 692                         k[n] = 0;
 693                 } else
 694                         *k = 0;
 695         }
 696
 697         /* Kernel threads have no argv[] */
 698         if (r == NULL || r[0] == 0) {
 699                 _cleanup_free_ char *t = NULL;
 700                 int h;
 701
 702                 free(r);
 703
 704                 if (!comm_fallback)
 705                         return -ENOENT;
 706
 707                 h = get_process_comm(pid, &t);
 708                 if (h < 0)
 709                         return h;
 710
 711                 r = strjoin("[", t, "]", NULL);
 712                 if (!r)
 713                         return -ENOMEM;
 714         }
 715
 716         *line = r;
 717         return 0;
 718 }
 719
 720 int is_kernel_thread(pid_t pid) {
 721         const char *p;
 722         size_t count;
 723         char c;
 724         bool eof;
 725         FILE *f;
 726
 727         if (pid == 0)
 728                 return 0;
 729
 730         assert(pid > 0);
 731
 732         p = procfs_file_alloca(pid, "cmdline");
 733         f = fopen(p, "re");
 734         if (!f)
 735                 return -errno;
 736
 737         count = fread(&c, 1, 1, f);
 738         eof = feof(f);
 739         fclose(f);
 740
 741         /* Kernel threads have an empty cmdline */
 742
 743         if (count <= 0)
 744                 return eof ? 1 : -errno;
 745
 746         return 0;
 747 }
 748
 749 int get_process_capeff(pid_t pid, char **capeff) {
 750         const char *p;
 751
 752         assert(capeff);
 753         assert(pid >= 0);
 754
 755         p = procfs_file_alloca(pid, "status");
 756
 757         return get_status_field(p, "\nCapEff:", capeff);
 758 }
 759
 760 int get_process_exe(pid_t pid, char **name) {
 761         const char *p;
 762         char *d;
 763         int r;
 764
 765         assert(pid >= 0);
 766         assert(name);
 767
 768         p = procfs_file_alloca(pid, "exe");
 769
 770         r = readlink_malloc(p, name);
 771         if (r < 0)
 772                 return r == -ENOENT ? -ESRCH : r;
 773
 774         d = endswith(*name, " (deleted)");
 775         if (d)
 776                 *d = '\0';
 777
 778         return 0;
 779 }
 780
 781 static int get_process_id(pid_t pid, const char *field, uid_t *uid) {
 782         _cleanup_fclose_ FILE *f = NULL;
 783         char line[LINE_MAX];
 784         const char *p;
 785
 786         assert(field);
 787         assert(uid);
 788
 789         if (pid == 0)
 790                 return getuid();
 791
 792         p = procfs_file_alloca(pid, "status");
 793         f = fopen(p, "re");
 794         if (!f)
 795                 return -errno;
 796
 797         FOREACH_LINE(line, f, return -errno) {
 798                 char *l;
 799
 800                 l = strstrip(line);
 801
 802                 if (startswith(l, field)) {
 803                         l += strlen(field);
 804                         l += strspn(l, WHITESPACE);
 805
 806                         l[strcspn(l, WHITESPACE)] = 0;
 807
 808                         return parse_uid(l, uid);
 809                 }
 810         }
 811
 812         return -EIO;
 813 }
 814
 815 int get_process_uid(pid_t pid, uid_t *uid) {
 816         return get_process_id(pid, "Uid:", uid);
 817 }
 818
 819 int get_process_gid(pid_t pid, gid_t *gid) {
 820         assert_cc(sizeof(uid_t) == sizeof(gid_t));
 821         return get_process_id(pid, "Gid:", gid);
 822 }
 823
 824 char *strnappend(const char *s, const char *suffix, size_t b) {
 825         size_t a;
 826         char *r;
 827
 828         if (!s && !suffix)
 829                 return strdup("");
 830
 831         if (!s)
 832                 return strndup(suffix, b);
 833
 834         if (!suffix)
 835                 return strdup(s);
 836
 837         assert(s);
 838         assert(suffix);
 839
 840         a = strlen(s);
 841         if (b > ((size_t) -1) - a)
 842                 return NULL;
 843
 844         r = new(char, a+b+1);
 845         if (!r)
 846                 return NULL;
 847
 848         memcpy(r, s, a);
 849         memcpy(r+a, suffix, b);
 850         r[a+b] = 0;
 851
 852         return r;
 853 }
 854
 855 char *strappend(const char *s, const char *suffix) {
 856         return strnappend(s, suffix, suffix ? strlen(suffix) : 0);
 857 }
 858
 859 int readlinkat_malloc(int fd, const char *p, char **ret) {
 860         size_t l = 100;
 861         int r;
 862
 863         assert(p);
 864         assert(ret);
 865
 866         for (;;) {
 867                 char *c;
 868                 ssize_t n;
 869
 870                 c = new(char, l);
 871                 if (!c)
 872                         return -ENOMEM;
 873
 874                 n = readlinkat(fd, p, c, l-1);
 875                 if (n < 0) {
 876                         r = -errno;
 877                         free(c);
 878                         return r;
 879                 }
 880
 881                 if ((size_t) n < l-1) {
 882                         c[n] = 0;
 883                         *ret = c;
 884                         return 0;
 885                 }
 886
 887                 free(c);
 888                 l *= 2;
 889         }
 890 }
 891
 892 int readlink_malloc(const char *p, char **ret) {
 893         return readlinkat_malloc(AT_FDCWD, p, ret);
 894 }
 895
 896 int readlink_and_make_absolute(const char *p, char **r) {
 897         _cleanup_free_ char *target = NULL;
 898         char *k;
 899         int j;
 900
 901         assert(p);
 902         assert(r);
 903
 904         j = readlink_malloc(p, &target);
 905         if (j < 0)
 906                 return j;
 907
 908         k = file_in_same_dir(p, target);
 909         if (!k)
 910                 return -ENOMEM;
 911
 912         *r = k;
 913         return 0;
 914 }
 915
 916 int readlink_and_canonicalize(const char *p, char **r) {
 917         char *t, *s;
 918         int j;
 919
 920         assert(p);
 921         assert(r);
 922
 923         j = readlink_and_make_absolute(p, &t);
 924         if (j < 0)
 925                 return j;
 926
 927         s = canonicalize_file_name(t);
 928         if (s) {
 929                 free(t);
 930                 *r = s;
 931         } else
 932                 *r = t;
 933
 934         path_kill_slashes(*r);
 935
 936         return 0;
 937 }
 938
 939 int reset_all_signal_handlers(void) {
 940         int sig, r = 0;
 941
 942         for (sig = 1; sig < _NSIG; sig++) {
 943                 struct sigaction sa = {
 944                         .sa_handler = SIG_DFL,
 945                         .sa_flags = SA_RESTART,
 946                 };
 947
 948                 /* These two cannot be caught... */
 949                 if (sig == SIGKILL || sig == SIGSTOP)
 950                         continue;
 951
 952                 /* On Linux the first two RT signals are reserved by
 953                  * glibc, and sigaction() will return EINVAL for them. */
 954                 if ((sigaction(sig, &sa, NULL) < 0))
 955                         if (errno != EINVAL && r == 0)
 956                                 r = -errno;
 957         }
 958
 959         return r;
 960 }
 961
 962 int reset_signal_mask(void) {
 963         sigset_t ss;
 964
 965         if (sigemptyset(&ss) < 0)
 966                 return -errno;
 967
 968         if (sigprocmask(SIG_SETMASK, &ss, NULL) < 0)
 969                 return -errno;
 970
 971         return 0;
 972 }
 973
 974 char *strstrip(char *s) {
 975         char *e;
 976
 977         /* Drops trailing whitespace. Modifies the string in
 978          * place. Returns pointer to first non-space character */
 979
 980         s += strspn(s, WHITESPACE);
 981
 982         for (e = strchr(s, 0); e > s; e --)
 983                 if (!strchr(WHITESPACE, e[-1]))
 984                         break;
 985
 986         *e = 0;
 987
 988         return s;
 989 }
 990
 991 char *delete_chars(char *s, const char *bad) {
 992         char *f, *t;
 993
 994         /* Drops all whitespace, regardless where in the string */
 995
 996         for (f = s, t = s; *f; f++) {
 997                 if (strchr(bad, *f))
 998                         continue;
 999
1000                 *(t++) = *f;
1001         }
1002
1003         *t = 0;
1004
1005         return s;
1006 }
1007
1008 char *file_in_same_dir(const char *path, const char *filename) {
1009         char *e, *r;
1010         size_t k;
1011
1012         assert(path);
1013         assert(filename);
1014
1015         /* This removes the last component of path and appends
1016          * filename, unless the latter is absolute anyway or the
1017          * former isn't */
1018
1019         if (path_is_absolute(filename))
1020                 return strdup(filename);
1021
1022         if (!(e = strrchr(path, '/')))
1023                 return strdup(filename);
1024
1025         k = strlen(filename);
1026         if (!(r = new(char, e-path+1+k+1)))
1027                 return NULL;
1028
1029         memcpy(r, path, e-path+1);
1030         memcpy(r+(e-path)+1, filename, k+1);
1031
1032         return r;
1033 }
1034
1035 int rmdir_parents(const char *path, const char *stop) {
1036         size_t l;
1037         int r = 0;
1038
1039         assert(path);
1040         assert(stop);
1041
1042         l = strlen(path);
1043
1044         /* Skip trailing slashes */
1045         while (l > 0 && path[l-1] == '/')
1046                 l--;
1047
1048         while (l > 0) {
1049                 char *t;
1050
1051                 /* Skip last component */
1052                 while (l > 0 && path[l-1] != '/')
1053                         l--;
1054
1055                 /* Skip trailing slashes */
1056                 while (l > 0 && path[l-1] == '/')
1057                         l--;
1058
1059                 if (l <= 0)
1060                         break;
1061
1062                 if (!(t = strndup(path, l)))
1063                         return -ENOMEM;
1064
1065                 if (path_startswith(stop, t)) {
1066                         free(t);
1067                         return 0;
1068                 }
1069
1070                 r = rmdir(t);
1071                 free(t);
1072
1073                 if (r < 0)
1074                         if (errno != ENOENT)
1075                                 return -errno;
1076         }
1077
1078         return 0;
1079 }
1080
1081 char hexchar(int x) {
1082         static const char table[16] = "0123456789abcdef";
1083
1084         return table[x & 15];
1085 }
1086
1087 int unhexchar(char c) {
1088
1089         if (c >= '0' && c <= '9')
1090                 return c - '0';
1091
1092         if (c >= 'a' && c <= 'f')
1093                 return c - 'a' + 10;
1094
1095         if (c >= 'A' && c <= 'F')
1096                 return c - 'A' + 10;
1097
1098         return -EINVAL;
1099 }
1100
1101 char *hexmem(const void *p, size_t l) {
1102         char *r, *z;
1103         const uint8_t *x;
1104
1105         z = r = malloc(l * 2 + 1);
1106         if (!r)
1107                 return NULL;
1108
1109         for (x = p; x < (const uint8_t*) p + l; x++) {
1110                 *(z++) = hexchar(*x >> 4);
1111                 *(z++) = hexchar(*x & 15);
1112         }
1113
1114         *z = 0;
1115         return r;
1116 }
1117
1118 void *unhexmem(const char *p, size_t l) {
1119         uint8_t *r, *z;
1120         const char *x;
1121
1122         assert(p);
1123
1124         z = r = malloc((l + 1) / 2 + 1);
1125         if (!r)
1126                 return NULL;
1127
1128         for (x = p; x < p + l; x += 2) {
1129                 int a, b;
1130
1131                 a = unhexchar(x[0]);
1132                 if (x+1 < p + l)
1133                         b = unhexchar(x[1]);
1134                 else
1135                         b = 0;
1136
1137                 *(z++) = (uint8_t) a << 4 | (uint8_t) b;
1138         }
1139
1140         *z = 0;
1141         return r;
1142 }
1143
1144 char octchar(int x) {
1145         return '0' + (x & 7);
1146 }
1147
1148 int unoctchar(char c) {
1149
1150         if (c >= '0' && c <= '7')
1151                 return c - '0';
1152
1153         return -EINVAL;
1154 }
1155
1156 char decchar(int x) {
1157         return '0' + (x % 10);
1158 }
1159
1160 int undecchar(char c) {
1161
1162         if (c >= '0' && c <= '9')
1163                 return c - '0';
1164
1165         return -EINVAL;
1166 }
1167
1168 char *cescape(const char *s) {
1169         char *r, *t;
1170         const char *f;
1171
1172         assert(s);
1173
1174         /* Does C style string escaping. */
1175
1176         r = new(char, strlen(s)*4 + 1);
1177         if (!r)
1178                 return NULL;
1179
1180         for (f = s, t = r; *f; f++)
1181
1182                 switch (*f) {
1183
1184                 case '\a':
1185                         *(t++) = '\\';
1186                         *(t++) = 'a';
1187                         break;
1188                 case '\b':
1189                         *(t++) = '\\';
1190                         *(t++) = 'b';
1191                         break;
1192                 case '\f':
1193                         *(t++) = '\\';
1194                         *(t++) = 'f';
1195                         break;
1196                 case '\n':
1197                         *(t++) = '\\';
1198                         *(t++) = 'n';
1199                         break;
1200                 case '\r':
1201                         *(t++) = '\\';
1202                         *(t++) = 'r';
1203                         break;
1204                 case '\t':
1205                         *(t++) = '\\';
1206                         *(t++) = 't';
1207                         break;
1208                 case '\v':
1209                         *(t++) = '\\';
1210                         *(t++) = 'v';
1211                         break;
1212                 case '\\':
1213                         *(t++) = '\\';
1214                         *(t++) = '\\';
1215                         break;
1216                 case '"':
1217                         *(t++) = '\\';
1218                         *(t++) = '"';
1219                         break;
1220                 case '\'':
1221                         *(t++) = '\\';
1222                         *(t++) = '\'';
1223                         break;
1224
1225                 default:
1226                         /* For special chars we prefer octal over
1227                          * hexadecimal encoding, simply because glib's
1228                          * g_strescape() does the same */
1229                         if ((*f < ' ') || (*f >= 127)) {
1230                                 *(t++) = '\\';
1231                                 *(t++) = octchar((unsigned char) *f >> 6);
1232                                 *(t++) = octchar((unsigned char) *f >> 3);
1233                                 *(t++) = octchar((unsigned char) *f);
1234                         } else
1235                                 *(t++) = *f;
1236                         break;
1237                 }
1238
1239         *t = 0;
1240
1241         return r;
1242 }
1243
1244 char *cunescape_length_with_prefix(const char *s, size_t length, const char *prefix) {
1245         char *r, *t;
1246         const char *f;
1247         size_t pl;
1248
1249         assert(s);
1250
1251         /* Undoes C style string escaping, and optionally prefixes it. */
1252
1253         pl = prefix ? strlen(prefix) : 0;
1254
1255         r = new(char, pl+length+1);
1256         if (!r)
1257                 return NULL;
1258
1259         if (prefix)
1260                 memcpy(r, prefix, pl);
1261
1262         for (f = s, t = r + pl; f < s + length; f++) {
1263
1264                 if (*f != '\\') {
1265                         *(t++) = *f;
1266                         continue;
1267                 }
1268
1269                 f++;
1270
1271                 switch (*f) {
1272
1273                 case 'a':
1274                         *(t++) = '\a';
1275                         break;
1276                 case 'b':
1277                         *(t++) = '\b';
1278                         break;
1279                 case 'f':
1280                         *(t++) = '\f';
1281                         break;
1282                 case 'n':
1283                         *(t++) = '\n';
1284                         break;
1285                 case 'r':
1286                         *(t++) = '\r';
1287                         break;
1288                 case 't':
1289                         *(t++) = '\t';
1290                         break;
1291                 case 'v':
1292                         *(t++) = '\v';
1293                         break;
1294                 case '\\':
1295                         *(t++) = '\\';
1296                         break;
1297                 case '"':
1298                         *(t++) = '"';
1299                         break;
1300                 case '\'':
1301                         *(t++) = '\'';
1302                         break;
1303
1304                 case 's':
1305                         /* This is an extension of the XDG syntax files */
1306                         *(t++) = ' ';
1307                         break;
1308
1309                 case 'x': {
1310                         /* hexadecimal encoding */
1311                         int a, b;
1312
1313                         a = unhexchar(f[1]);
1314                         b = unhexchar(f[2]);
1315
1316                         if (a < 0 || b < 0 || (a == 0 && b == 0)) {
1317                                 /* Invalid escape code, let's take it literal then */
1318                                 *(t++) = '\\';
1319                                 *(t++) = 'x';
1320                         } else {
1321                                 *(t++) = (char) ((a << 4) | b);
1322                                 f += 2;
1323                         }
1324
1325                         break;
1326                 }
1327
1328                 case '0':
1329                 case '1':
1330                 case '2':
1331                 case '3':
1332                 case '4':
1333                 case '5':
1334                 case '6':
1335                 case '7': {
1336                         /* octal encoding */
1337                         int a, b, c;
1338
1339                         a = unoctchar(f[0]);
1340                         b = unoctchar(f[1]);
1341                         c = unoctchar(f[2]);
1342
1343                         if (a < 0 || b < 0 || c < 0 || (a == 0 && b == 0 && c == 0)) {
1344                                 /* Invalid escape code, let's take it literal then */
1345                                 *(t++) = '\\';
1346                                 *(t++) = f[0];
1347                         } else {
1348                                 *(t++) = (char) ((a << 6) | (b << 3) | c);
1349                                 f += 2;
1350                         }
1351
1352                         break;
1353                 }
1354
1355                 case 0:
1356                         /* premature end of string.*/
1357                         *(t++) = '\\';
1358                         goto finish;
1359
1360                 default:
1361                         /* Invalid escape code, let's take it literal then */
1362                         *(t++) = '\\';
1363                         *(t++) = *f;
1364                         break;
1365                 }
1366         }
1367
1368 finish:
1369         *t = 0;
1370         return r;
1371 }
1372
1373 char *cunescape_length(const char *s, size_t length) {
1374         return cunescape_length_with_prefix(s, length, NULL);
1375 }
1376
1377 char *cunescape(const char *s) {
1378         assert(s);
1379
1380         return cunescape_length(s, strlen(s));
1381 }
1382
1383 char *xescape(const char *s, const char *bad) {
1384         char *r, *t;
1385         const char *f;
1386
1387         /* Escapes all chars in bad, in addition to \ and all special
1388          * chars, in \xFF style escaping. May be reversed with
1389          * cunescape. */
1390
1391         r = new(char, strlen(s) * 4 + 1);
1392         if (!r)
1393                 return NULL;
1394
1395         for (f = s, t = r; *f; f++) {
1396
1397                 if ((*f < ' ') || (*f >= 127) ||
1398                     (*f == '\\') || strchr(bad, *f)) {
1399                         *(t++) = '\\';
1400                         *(t++) = 'x';
1401                         *(t++) = hexchar(*f >> 4);
1402                         *(t++) = hexchar(*f);
1403                 } else
1404                         *(t++) = *f;
1405         }
1406
1407         *t = 0;
1408
1409         return r;
1410 }
1411
1412 char *ascii_strlower(char *t) {
1413         char *p;
1414
1415         assert(t);
1416
1417         for (p = t; *p; p++)
1418                 if (*p >= 'A' && *p <= 'Z')
1419                         *p = *p - 'A' + 'a';
1420
1421         return t;
1422 }
1423
1424 _pure_ static bool ignore_file_allow_backup(const char *filename) {
1425         assert(filename);
1426
1427         return
1428                 filename[0] == '.' ||
1429                 streq(filename, "lost+found") ||
1430                 streq(filename, "aquota.user") ||
1431                 streq(filename, "aquota.group") ||
1432                 endswith(filename, ".rpmnew") ||
1433                 endswith(filename, ".rpmsave") ||
1434                 endswith(filename, ".rpmorig") ||
1435                 endswith(filename, ".dpkg-old") ||
1436                 endswith(filename, ".dpkg-new") ||
1437                 endswith(filename, ".dpkg-tmp") ||
1438                 endswith(filename, ".swp");
1439 }
1440
1441 bool ignore_file(const char *filename) {
1442         assert(filename);
1443
1444         if (endswith(filename, "~"))
1445                 return true;
1446
1447         return ignore_file_allow_backup(filename);
1448 }
1449
1450 int fd_nonblock(int fd, bool nonblock) {
1451         int flags, nflags;
1452
1453         assert(fd >= 0);
1454
1455         flags = fcntl(fd, F_GETFL, 0);
1456         if (flags < 0)
1457                 return -errno;
1458
1459         if (nonblock)
1460                 nflags = flags | O_NONBLOCK;
1461         else
1462                 nflags = flags & ~O_NONBLOCK;
1463
1464         if (nflags == flags)
1465                 return 0;
1466
1467         if (fcntl(fd, F_SETFL, nflags) < 0)
1468                 return -errno;
1469
1470         return 0;
1471 }
1472
1473 int fd_cloexec(int fd, bool cloexec) {
1474         int flags, nflags;
1475
1476         assert(fd >= 0);
1477
1478         flags = fcntl(fd, F_GETFD, 0);
1479         if (flags < 0)
1480                 return -errno;
1481
1482         if (cloexec)
1483                 nflags = flags | FD_CLOEXEC;
1484         else
1485                 nflags = flags & ~FD_CLOEXEC;
1486
1487         if (nflags == flags)
1488                 return 0;
1489
1490         if (fcntl(fd, F_SETFD, nflags) < 0)
1491                 return -errno;
1492
1493         return 0;
1494 }
1495
1496 _pure_ static bool fd_in_set(int fd, const int fdset[], unsigned n_fdset) {
1497         unsigned i;
1498
1499         assert(n_fdset == 0 || fdset);
1500
1501         for (i = 0; i < n_fdset; i++)
1502                 if (fdset[i] == fd)
1503                         return true;
1504
1505         return false;
1506 }
1507
1508 int close_all_fds(const int except[], unsigned n_except) {
1509         _cleanup_closedir_ DIR *d = NULL;
1510         struct dirent *de;
1511         int r = 0;
1512
1513         assert(n_except == 0 || except);
1514
1515         d = opendir("/proc/self/fd");
1516         if (!d) {
1517                 int fd;
1518                 struct rlimit rl;
1519
1520                 /* When /proc isn't available (for example in chroots)
1521                  * the fallback is brute forcing through the fd
1522                  * table */
1523
1524                 assert_se(getrlimit(RLIMIT_NOFILE, &rl) >= 0);
1525                 for (fd = 3; fd < (int) rl.rlim_max; fd ++) {
1526
1527                         if (fd_in_set(fd, except, n_except))
1528                                 continue;
1529
1530                         if (close_nointr(fd) < 0)
1531                                 if (errno != EBADF && r == 0)
1532                                         r = -errno;
1533                 }
1534
1535                 return r;
1536         }
1537
1538         while ((de = readdir(d))) {
1539                 int fd = -1;
1540
1541                 if (ignore_file(de->d_name))
1542                         continue;
1543
1544                 if (safe_atoi(de->d_name, &fd) < 0)
1545                         /* Let's better ignore this, just in case */
1546                         continue;
1547
1548                 if (fd < 3)
1549                         continue;
1550
1551                 if (fd == dirfd(d))
1552                         continue;
1553
1554                 if (fd_in_set(fd, except, n_except))
1555                         continue;
1556
1557                 if (close_nointr(fd) < 0) {
1558                         /* Valgrind has its own FD and doesn't want to have it closed */
1559                         if (errno != EBADF && r == 0)
1560                                 r = -errno;
1561                 }
1562         }
1563
1564         return r;
1565 }
1566
1567 bool chars_intersect(const char *a, const char *b) {
1568         const char *p;
1569
1570         /* Returns true if any of the chars in a are in b. */
1571         for (p = a; *p; p++)
1572                 if (strchr(b, *p))
1573                         return true;
1574
1575         return false;
1576 }
1577
1578 bool fstype_is_network(const char *fstype) {
1579         static const char table[] =
1580                 "cifs\0"
1581                 "smbfs\0"
1582                 "sshfs\0"
1583                 "ncpfs\0"
1584                 "ncp\0"
1585                 "nfs\0"
1586                 "nfs4\0"
1587                 "gfs\0"
1588                 "gfs2\0"
1589                 "glusterfs\0";
1590
1591         const char *x;
1592
1593         x = startswith(fstype, "fuse.");
1594         if (x)
1595                 fstype = x;
1596
1597         return nulstr_contains(table, fstype);
1598 }
1599
1600 int chvt(int vt) {
1601         _cleanup_close_ int fd;
1602
1603         fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
1604         if (fd < 0)
1605                 return -errno;
1606
1607         if (vt < 0) {
1608                 int tiocl[2] = {
1609                         TIOCL_GETKMSGREDIRECT,
1610                         0
1611                 };
1612
1613                 if (ioctl(fd, TIOCLINUX, tiocl) < 0)
1614                         return -errno;
1615
1616                 vt = tiocl[0] <= 0 ? 1 : tiocl[0];
1617         }
1618
1619         if (ioctl(fd, VT_ACTIVATE, vt) < 0)
1620                 return -errno;
1621
1622         return 0;
1623 }
1624
1625 int read_one_char(FILE *f, char *ret, usec_t t, bool *need_nl) {
1626         struct termios old_termios, new_termios;
1627         char c, line[LINE_MAX];
1628
1629         assert(f);
1630         assert(ret);
1631
1632         if (tcgetattr(fileno(f), &old_termios) >= 0) {
1633                 new_termios = old_termios;
1634
1635                 new_termios.c_lflag &= ~ICANON;
1636                 new_termios.c_cc[VMIN] = 1;
1637                 new_termios.c_cc[VTIME] = 0;
1638
1639                 if (tcsetattr(fileno(f), TCSADRAIN, &new_termios) >= 0) {
1640                         size_t k;
1641
1642                         if (t != USEC_INFINITY) {
1643                                 if (fd_wait_for_event(fileno(f), POLLIN, t) <= 0) {
1644                                         tcsetattr(fileno(f), TCSADRAIN, &old_termios);
1645                                         return -ETIMEDOUT;
1646                                 }
1647                         }
1648
1649                         k = fread(&c, 1, 1, f);
1650
1651                         tcsetattr(fileno(f), TCSADRAIN, &old_termios);
1652
1653                         if (k <= 0)
1654                                 return -EIO;
1655
1656                         if (need_nl)
1657                                 *need_nl = c != '\n';
1658
1659                         *ret = c;
1660                         return 0;
1661                 }
1662         }
1663
1664         if (t != USEC_INFINITY) {
1665                 if (fd_wait_for_event(fileno(f), POLLIN, t) <= 0)
1666                         return -ETIMEDOUT;
1667         }
1668
1669         errno = 0;
1670         if (!fgets(line, sizeof(line), f))
1671                 return errno ? -errno : -EIO;
1672
1673         truncate_nl(line);
1674
1675         if (strlen(line) != 1)
1676                 return -EBADMSG;
1677
1678         if (need_nl)
1679                 *need_nl = false;
1680
1681         *ret = line[0];
1682         return 0;
1683 }
1684
1685 int ask_char(char *ret, const char *replies, const char *text, ...) {
1686         int r;
1687
1688         assert(ret);
1689         assert(replies);
1690         assert(text);
1691
1692         for (;;) {
1693                 va_list ap;
1694                 char c;
1695                 bool need_nl = true;
1696
1697                 if (on_tty())
1698                         fputs(ANSI_HIGHLIGHT_ON, stdout);
1699
1700                 va_start(ap, text);
1701                 vprintf(text, ap);
1702                 va_end(ap);
1703
1704                 if (on_tty())
1705                         fputs(ANSI_HIGHLIGHT_OFF, stdout);
1706
1707                 fflush(stdout);
1708
1709                 r = read_one_char(stdin, &c, USEC_INFINITY, &need_nl);
1710                 if (r < 0) {
1711
1712                         if (r == -EBADMSG) {
1713                                 puts("Bad input, please try again.");
1714                                 continue;
1715                         }
1716
1717                         putchar('\n');
1718                         return r;
1719                 }
1720
1721                 if (need_nl)
1722                         putchar('\n');
1723
1724                 if (strchr(replies, c)) {
1725                         *ret = c;
1726                         return 0;
1727                 }
1728
1729                 puts("Read unexpected character, please try again.");
1730         }
1731 }
1732
1733 int ask_string(char **ret, const char *text, ...) {
1734         assert(ret);
1735         assert(text);
1736
1737         for (;;) {
1738                 char line[LINE_MAX];
1739                 va_list ap;
1740
1741                 if (on_tty())
1742                         fputs(ANSI_HIGHLIGHT_ON, stdout);
1743
1744                 va_start(ap, text);
1745                 vprintf(text, ap);
1746                 va_end(ap);
1747
1748                 if (on_tty())
1749                         fputs(ANSI_HIGHLIGHT_OFF, stdout);
1750
1751                 fflush(stdout);
1752
1753                 errno = 0;
1754                 if (!fgets(line, sizeof(line), stdin))
1755                         return errno ? -errno : -EIO;
1756
1757                 if (!endswith(line, "\n"))
1758                         putchar('\n');
1759                 else {
1760                         char *s;
1761
1762                         if (isempty(line))
1763                                 continue;
1764
1765                         truncate_nl(line);
1766                         s = strdup(line);
1767                         if (!s)
1768                                 return -ENOMEM;
1769
1770                         *ret = s;
1771                         return 0;
1772                 }
1773         }
1774 }
1775
1776 int reset_terminal_fd(int fd, bool switch_to_text) {
1777         struct termios termios;
1778         int r = 0;
1779
1780         /* Set terminal to some sane defaults */
1781
1782         assert(fd >= 0);
1783
1784         /* We leave locked terminal attributes untouched, so that
1785          * Plymouth may set whatever it wants to set, and we don't
1786          * interfere with that. */
1787
1788         /* Disable exclusive mode, just in case */
1789         ioctl(fd, TIOCNXCL);
1790
1791         /* Switch to text mode */
1792         if (switch_to_text)
1793                 ioctl(fd, KDSETMODE, KD_TEXT);
1794
1795         /* Enable console unicode mode */
1796         ioctl(fd, KDSKBMODE, K_UNICODE);
1797
1798         if (tcgetattr(fd, &termios) < 0) {
1799                 r = -errno;
1800                 goto finish;
1801         }
1802
1803         /* We only reset the stuff that matters to the software. How
1804          * hardware is set up we don't touch assuming that somebody
1805          * else will do that for us */
1806
1807         termios.c_iflag &= ~(IGNBRK | BRKINT | ISTRIP | INLCR | IGNCR | IUCLC);
1808         termios.c_iflag |= ICRNL | IMAXBEL | IUTF8;
1809         termios.c_oflag |= ONLCR;
1810         termios.c_cflag |= CREAD;
1811         termios.c_lflag = ISIG | ICANON | IEXTEN | ECHO | ECHOE | ECHOK | ECHOCTL | ECHOPRT | ECHOKE;
1812
1813         termios.c_cc[VINTR]    =   03;  /* ^C */
1814         termios.c_cc[VQUIT]    =  034;  /* ^\ */
1815         termios.c_cc[VERASE]   = 0177;
1816         termios.c_cc[VKILL]    =  025;  /* ^X */
1817         termios.c_cc[VEOF]     =   04;  /* ^D */
1818         termios.c_cc[VSTART]   =  021;  /* ^Q */
1819         termios.c_cc[VSTOP]    =  023;  /* ^S */
1820         termios.c_cc[VSUSP]    =  032;  /* ^Z */
1821         termios.c_cc[VLNEXT]   =  026;  /* ^V */
1822         termios.c_cc[VWERASE]  =  027;  /* ^W */
1823         termios.c_cc[VREPRINT] =  022;  /* ^R */
1824         termios.c_cc[VEOL]     =    0;
1825         termios.c_cc[VEOL2]    =    0;
1826
1827         termios.c_cc[VTIME]  = 0;
1828         termios.c_cc[VMIN]   = 1;
1829
1830         if (tcsetattr(fd, TCSANOW, &termios) < 0)
1831                 r = -errno;
1832
1833 finish:
1834         /* Just in case, flush all crap out */
1835         tcflush(fd, TCIOFLUSH);
1836
1837         return r;
1838 }
1839
1840 int reset_terminal(const char *name) {
1841         _cleanup_close_ int fd = -1;
1842
1843         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
1844         if (fd < 0)
1845                 return fd;
1846
1847         return reset_terminal_fd(fd, true);
1848 }
1849
1850 int open_terminal(const char *name, int mode) {
1851         int fd, r;
1852         unsigned c = 0;
1853
1854         /*
1855          * If a TTY is in the process of being closed opening it might
1856          * cause EIO. This is horribly awful, but unlikely to be
1857          * changed in the kernel. Hence we work around this problem by
1858          * retrying a couple of times.
1859          *
1860          * https://bugs.launchpad.net/ubuntu/+source/linux/+bug/554172/comments/245
1861          */
1862
1863         assert(!(mode & O_CREAT));
1864
1865         for (;;) {
1866                 fd = open(name, mode, 0);
1867                 if (fd >= 0)
1868                         break;
1869
1870                 if (errno != EIO)
1871                         return -errno;
1872
1873                 /* Max 1s in total */
1874                 if (c >= 20)
1875                         return -errno;
1876
1877                 usleep(50 * USEC_PER_MSEC);
1878                 c++;
1879         }
1880
1881         r = isatty(fd);
1882         if (r < 0) {
1883                 safe_close(fd);
1884                 return -errno;
1885         }
1886
1887         if (!r) {
1888                 safe_close(fd);
1889                 return -ENOTTY;
1890         }
1891
1892         return fd;
1893 }
1894
1895 int flush_fd(int fd) {
1896         struct pollfd pollfd = {
1897                 .fd = fd,
1898                 .events = POLLIN,
1899         };
1900
1901         for (;;) {
1902                 char buf[LINE_MAX];
1903                 ssize_t l;
1904                 int r;
1905
1906                 r = poll(&pollfd, 1, 0);
1907                 if (r < 0) {
1908                         if (errno == EINTR)
1909                                 continue;
1910
1911                         return -errno;
1912
1913                 } else if (r == 0)
1914                         return 0;
1915
1916                 l = read(fd, buf, sizeof(buf));
1917                 if (l < 0) {
1918
1919                         if (errno == EINTR)
1920                                 continue;
1921
1922                         if (errno == EAGAIN)
1923                                 return 0;
1924
1925                         return -errno;
1926                 } else if (l == 0)
1927                         return 0;
1928         }
1929 }
1930
1931 int acquire_terminal(
1932                 const char *name,
1933                 bool fail,
1934                 bool force,
1935                 bool ignore_tiocstty_eperm,
1936                 usec_t timeout) {
1937
1938         int fd = -1, notify = -1, r = 0, wd = -1;
1939         usec_t ts = 0;
1940
1941         assert(name);
1942
1943         /* We use inotify to be notified when the tty is closed. We
1944          * create the watch before checking if we can actually acquire
1945          * it, so that we don't lose any event.
1946          *
1947          * Note: strictly speaking this actually watches for the
1948          * device being closed, it does *not* really watch whether a
1949          * tty loses its controlling process. However, unless some
1950          * rogue process uses TIOCNOTTY on /dev/tty *after* closing
1951          * its tty otherwise this will not become a problem. As long
1952          * as the administrator makes sure not configure any service
1953          * on the same tty as an untrusted user this should not be a
1954          * problem. (Which he probably should not do anyway.) */
1955
1956         if (timeout != USEC_INFINITY)
1957                 ts = now(CLOCK_MONOTONIC);
1958
1959         if (!fail && !force) {
1960                 notify = inotify_init1(IN_CLOEXEC | (timeout != USEC_INFINITY ? IN_NONBLOCK : 0));
1961                 if (notify < 0) {
1962                         r = -errno;
1963                         goto fail;
1964                 }
1965
1966                 wd = inotify_add_watch(notify, name, IN_CLOSE);
1967                 if (wd < 0) {
1968                         r = -errno;
1969                         goto fail;
1970                 }
1971         }
1972
1973         for (;;) {
1974                 struct sigaction sa_old, sa_new = {
1975                         .sa_handler = SIG_IGN,
1976                         .sa_flags = SA_RESTART,
1977                 };
1978
1979                 if (notify >= 0) {
1980                         r = flush_fd(notify);
1981                         if (r < 0)
1982                                 goto fail;
1983                 }
1984
1985                 /* We pass here O_NOCTTY only so that we can check the return
1986                  * value TIOCSCTTY and have a reliable way to figure out if we
1987                  * successfully became the controlling process of the tty */
1988                 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
1989                 if (fd < 0)
1990                         return fd;
1991
1992                 /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
1993                  * if we already own the tty. */
1994                 assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
1995
1996                 /* First, try to get the tty */
1997                 if (ioctl(fd, TIOCSCTTY, force) < 0)
1998                         r = -errno;
1999
2000                 assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
2001
2002                 /* Sometimes it makes sense to ignore TIOCSCTTY
2003                  * returning EPERM, i.e. when very likely we already
2004                  * are have this controlling terminal. */
2005                 if (r < 0 && r == -EPERM && ignore_tiocstty_eperm)
2006                         r = 0;
2007
2008                 if (r < 0 && (force || fail || r != -EPERM)) {
2009                         goto fail;
2010                 }
2011
2012                 if (r >= 0)
2013                         break;
2014
2015                 assert(!fail);
2016                 assert(!force);
2017                 assert(notify >= 0);
2018
2019                 for (;;) {
2020                         uint8_t inotify_buffer[sizeof(struct inotify_event) + FILENAME_MAX];
2021                         ssize_t l;
2022                         struct inotify_event *e;
2023
2024                         if (timeout != USEC_INFINITY) {
2025                                 usec_t n;
2026
2027                                 n = now(CLOCK_MONOTONIC);
2028                                 if (ts + timeout < n) {
2029                                         r = -ETIMEDOUT;
2030                                         goto fail;
2031                                 }
2032
2033                                 r = fd_wait_for_event(fd, POLLIN, ts + timeout - n);
2034                                 if (r < 0)
2035                                         goto fail;
2036
2037                                 if (r == 0) {
2038                                         r = -ETIMEDOUT;
2039                                         goto fail;
2040                                 }
2041                         }
2042
2043                         l = read(notify, inotify_buffer, sizeof(inotify_buffer));
2044                         if (l < 0) {
2045
2046                                 if (errno == EINTR || errno == EAGAIN)
2047                                         continue;
2048
2049                                 r = -errno;
2050                                 goto fail;
2051                         }
2052
2053                         e = (struct inotify_event*) inotify_buffer;
2054
2055                         while (l > 0) {
2056                                 size_t step;
2057
2058                                 if (e->wd != wd || !(e->mask & IN_CLOSE)) {
2059                                         r = -EIO;
2060                                         goto fail;
2061                                 }
2062
2063                                 step = sizeof(struct inotify_event) + e->len;
2064                                 assert(step <= (size_t) l);
2065
2066                                 e = (struct inotify_event*) ((uint8_t*) e + step);
2067                                 l -= step;
2068                         }
2069
2070                         break;
2071                 }
2072
2073                 /* We close the tty fd here since if the old session
2074                  * ended our handle will be dead. It's important that
2075                  * we do this after sleeping, so that we don't enter
2076                  * an endless loop. */
2077                 safe_close(fd);
2078         }
2079
2080         safe_close(notify);
2081
2082         r = reset_terminal_fd(fd, true);
2083         if (r < 0)
2084                 log_warning("Failed to reset terminal: %s", strerror(-r));
2085
2086         return fd;
2087
2088 fail:
2089         safe_close(fd);
2090         safe_close(notify);
2091
2092         return r;
2093 }
2094
2095 int release_terminal(void) {
2096         static const struct sigaction sa_new = {
2097                 .sa_handler = SIG_IGN,
2098                 .sa_flags = SA_RESTART,
2099         };
2100
2101         _cleanup_close_ int fd = -1;
2102         struct sigaction sa_old;
2103         int r = 0;
2104
2105         fd = open("/dev/tty", O_RDWR|O_NOCTTY|O_NDELAY|O_CLOEXEC);
2106         if (fd < 0)
2107                 return -errno;
2108
2109         /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
2110          * by our own TIOCNOTTY */
2111         assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
2112
2113         if (ioctl(fd, TIOCNOTTY) < 0)
2114                 r = -errno;
2115
2116         assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
2117
2118         return r;
2119 }
2120
2121 int sigaction_many(const struct sigaction *sa, ...) {
2122         va_list ap;
2123         int r = 0, sig;
2124
2125         va_start(ap, sa);
2126         while ((sig = va_arg(ap, int)) > 0)
2127                 if (sigaction(sig, sa, NULL) < 0)
2128                         r = -errno;
2129         va_end(ap);
2130
2131         return r;
2132 }
2133
2134 int ignore_signals(int sig, ...) {
2135         struct sigaction sa = {
2136                 .sa_handler = SIG_IGN,
2137                 .sa_flags = SA_RESTART,
2138         };
2139         va_list ap;
2140         int r = 0;
2141
2142         if (sigaction(sig, &sa, NULL) < 0)
2143                 r = -errno;
2144
2145         va_start(ap, sig);
2146         while ((sig = va_arg(ap, int)) > 0)
2147                 if (sigaction(sig, &sa, NULL) < 0)
2148                         r = -errno;
2149         va_end(ap);
2150
2151         return r;
2152 }
2153
2154 int default_signals(int sig, ...) {
2155         struct sigaction sa = {
2156                 .sa_handler = SIG_DFL,
2157                 .sa_flags = SA_RESTART,
2158         };
2159         va_list ap;
2160         int r = 0;
2161
2162         if (sigaction(sig, &sa, NULL) < 0)
2163                 r = -errno;
2164
2165         va_start(ap, sig);
2166         while ((sig = va_arg(ap, int)) > 0)
2167                 if (sigaction(sig, &sa, NULL) < 0)
2168                         r = -errno;
2169         va_end(ap);
2170
2171         return r;
2172 }
2173
2174 void safe_close_pair(int p[]) {
2175         assert(p);
2176
2177         if (p[0] == p[1]) {
2178                 /* Special case pairs which use the same fd in both
2179                  * directions... */
2180                 p[0] = p[1] = safe_close(p[0]);
2181                 return;
2182         }
2183
2184         p[0] = safe_close(p[0]);
2185         p[1] = safe_close(p[1]);
2186 }
2187
2188 ssize_t loop_read(int fd, void *buf, size_t nbytes, bool do_poll) {
2189         uint8_t *p = buf;
2190         ssize_t n = 0;
2191
2192         assert(fd >= 0);
2193         assert(buf);
2194
2195         while (nbytes > 0) {
2196                 ssize_t k;
2197
2198                 k = read(fd, p, nbytes);
2199                 if (k < 0 && errno == EINTR)
2200                         continue;
2201
2202                 if (k < 0 && errno == EAGAIN && do_poll) {
2203
2204                         /* We knowingly ignore any return value here,
2205                          * and expect that any error/EOF is reported
2206                          * via read() */
2207
2208                         fd_wait_for_event(fd, POLLIN, USEC_INFINITY);
2209                         continue;
2210                 }
2211
2212                 if (k <= 0)
2213                         return n > 0 ? n : (k < 0 ? -errno : 0);
2214
2215                 p += k;
2216                 nbytes -= k;
2217                 n += k;
2218         }
2219
2220         return n;
2221 }
2222
2223 ssize_t loop_write(int fd, const void *buf, size_t nbytes, bool do_poll) {
2224         const uint8_t *p = buf;
2225         ssize_t n = 0;
2226
2227         assert(fd >= 0);
2228         assert(buf);
2229
2230         while (nbytes > 0) {
2231                 ssize_t k;
2232
2233                 k = write(fd, p, nbytes);
2234                 if (k < 0 && errno == EINTR)
2235                         continue;
2236
2237                 if (k < 0 && errno == EAGAIN && do_poll) {
2238
2239                         /* We knowingly ignore any return value here,
2240                          * and expect that any error/EOF is reported
2241                          * via write() */
2242
2243                         fd_wait_for_event(fd, POLLOUT, USEC_INFINITY);
2244                         continue;
2245                 }
2246
2247                 if (k <= 0)
2248                         return n > 0 ? n : (k < 0 ? -errno : 0);
2249
2250                 p += k;
2251                 nbytes -= k;
2252                 n += k;
2253         }
2254
2255         return n;
2256 }
2257
2258 int parse_size(const char *t, off_t base, off_t *size) {
2259
2260         /* Soo, sometimes we want to parse IEC binary suffxies, and
2261          * sometimes SI decimal suffixes. This function can parse
2262          * both. Which one is the right way depends on the
2263          * context. Wikipedia suggests that SI is customary for
2264          * hardrware metrics and network speeds, while IEC is
2265          * customary for most data sizes used by software and volatile
2266          * (RAM) memory. Hence be careful which one you pick!
2267          *
2268          * In either case we use just K, M, G as suffix, and not Ki,
2269          * Mi, Gi or so (as IEC would suggest). That's because that's
2270          * frickin' ugly. But this means you really need to make sure
2271          * to document which base you are parsing when you use this
2272          * call. */
2273
2274         struct table {
2275                 const char *suffix;
2276                 unsigned long long factor;
2277         };
2278
2279         static const struct table iec[] = {
2280                 { "E", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2281                 { "P", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2282                 { "T", 1024ULL*1024ULL*1024ULL*1024ULL },
2283                 { "G", 1024ULL*1024ULL*1024ULL },
2284                 { "M", 1024ULL*1024ULL },
2285                 { "K", 1024ULL },
2286                 { "B", 1 },
2287                 { "", 1 },
2288         };
2289
2290         static const struct table si[] = {
2291                 { "E", 1000ULL*1000ULL*1000ULL*1000ULL*1000ULL*1000ULL },
2292                 { "P", 1000ULL*1000ULL*1000ULL*1000ULL*1000ULL },
2293                 { "T", 1000ULL*1000ULL*1000ULL*1000ULL },
2294                 { "G", 1000ULL*1000ULL*1000ULL },
2295                 { "M", 1000ULL*1000ULL },
2296                 { "K", 1000ULL },
2297                 { "B", 1 },
2298                 { "", 1 },
2299         };
2300
2301         const struct table *table;
2302         const char *p;
2303         unsigned long long r = 0;
2304         unsigned n_entries, start_pos = 0;
2305
2306         assert(t);
2307         assert(base == 1000 || base == 1024);
2308         assert(size);
2309
2310         if (base == 1000) {
2311                 table = si;
2312                 n_entries = ELEMENTSOF(si);
2313         } else {
2314                 table = iec;
2315                 n_entries = ELEMENTSOF(iec);
2316         }
2317
2318         p = t;
2319         do {
2320                 long long l;
2321                 unsigned long long l2;
2322                 double frac = 0;
2323                 char *e;
2324                 unsigned i;
2325
2326                 errno = 0;
2327                 l = strtoll(p, &e, 10);
2328
2329                 if (errno > 0)
2330                         return -errno;
2331
2332                 if (l < 0)
2333                         return -ERANGE;
2334
2335                 if (e == p)
2336                         return -EINVAL;
2337
2338                 if (*e == '.') {
2339                         e++;
2340                         if (*e >= '0' && *e <= '9') {
2341                                 char *e2;
2342
2343                                 /* strotoull itself would accept space/+/- */
2344                                 l2 = strtoull(e, &e2, 10);
2345
2346                                 if (errno == ERANGE)
2347                                         return -errno;
2348
2349                                 /* Ignore failure. E.g. 10.M is valid */
2350                                 frac = l2;
2351                                 for (; e < e2; e++)
2352                                         frac /= 10;
2353                         }
2354                 }
2355
2356                 e += strspn(e, WHITESPACE);
2357
2358                 for (i = start_pos; i < n_entries; i++)
2359                         if (startswith(e, table[i].suffix)) {
2360                                 unsigned long long tmp;
2361                                 if ((unsigned long long) l + (frac > 0) > ULLONG_MAX / table[i].factor)
2362                                         return -ERANGE;
2363                                 tmp = l * table[i].factor + (unsigned long long) (frac * table[i].factor);
2364                                 if (tmp > ULLONG_MAX - r)
2365                                         return -ERANGE;
2366
2367                                 r += tmp;
2368                                 if ((unsigned long long) (off_t) r != r)
2369                                         return -ERANGE;
2370
2371                                 p = e + strlen(table[i].suffix);
2372
2373                                 start_pos = i + 1;
2374                                 break;
2375                         }
2376
2377                 if (i >= n_entries)
2378                         return -EINVAL;
2379
2380         } while (*p);
2381
2382         *size = r;
2383
2384         return 0;
2385 }
2386
2387 int make_stdio(int fd) {
2388         int r, s, t;
2389
2390         assert(fd >= 0);
2391
2392         r = dup3(fd, STDIN_FILENO, 0);
2393         s = dup3(fd, STDOUT_FILENO, 0);
2394         t = dup3(fd, STDERR_FILENO, 0);
2395
2396         if (fd >= 3)
2397                 safe_close(fd);
2398
2399         if (r < 0 || s < 0 || t < 0)
2400                 return -errno;
2401
2402         /* We rely here that the new fd has O_CLOEXEC not set */
2403
2404         return 0;
2405 }
2406
2407 int make_null_stdio(void) {
2408         int null_fd;
2409
2410         null_fd = open("/dev/null", O_RDWR|O_NOCTTY);
2411         if (null_fd < 0)
2412                 return -errno;
2413
2414         return make_stdio(null_fd);
2415 }
2416
2417 bool is_device_path(const char *path) {
2418
2419         /* Returns true on paths that refer to a device, either in
2420          * sysfs or in /dev */
2421
2422         return
2423                 path_startswith(path, "/dev/") ||
2424                 path_startswith(path, "/sys/");
2425 }
2426
2427 int dir_is_empty(const char *path) {
2428         _cleanup_closedir_ DIR *d;
2429
2430         d = opendir(path);
2431         if (!d)
2432                 return -errno;
2433
2434         for (;;) {
2435                 struct dirent *de;
2436
2437                 errno = 0;
2438                 de = readdir(d);
2439                 if (!de && errno != 0)
2440                         return -errno;
2441
2442                 if (!de)
2443                         return 1;
2444
2445                 if (!ignore_file(de->d_name))
2446                         return 0;
2447         }
2448 }
2449
2450 char* dirname_malloc(const char *path) {
2451         char *d, *dir, *dir2;
2452
2453         d = strdup(path);
2454         if (!d)
2455                 return NULL;
2456         dir = dirname(d);
2457         assert(dir);
2458
2459         if (dir != d) {
2460                 dir2 = strdup(dir);
2461                 free(d);
2462                 return dir2;
2463         }
2464
2465         return dir;
2466 }
2467
2468 int dev_urandom(void *p, size_t n) {
2469         _cleanup_close_ int fd;
2470         ssize_t k;
2471
2472         fd = open("/dev/urandom", O_RDONLY|O_CLOEXEC|O_NOCTTY);
2473         if (fd < 0)
2474                 return errno == ENOENT ? -ENOSYS : -errno;
2475
2476         k = loop_read(fd, p, n, true);
2477         if (k < 0)
2478                 return (int) k;
2479         if ((size_t) k != n)
2480                 return -EIO;
2481
2482         return 0;
2483 }
2484
2485 void random_bytes(void *p, size_t n) {
2486         static bool srand_called = false;
2487         uint8_t *q;
2488         int r;
2489
2490         r = dev_urandom(p, n);
2491         if (r >= 0)
2492                 return;
2493
2494         /* If some idiot made /dev/urandom unavailable to us, he'll
2495          * get a PRNG instead. */
2496
2497         if (!srand_called) {
2498                 unsigned x = 0;
2499
2500 #ifdef HAVE_SYS_AUXV_H
2501                 /* The kernel provides us with a bit of entropy in
2502                  * auxv, so let's try to make use of that to seed the
2503                  * pseudo-random generator. It's better than
2504                  * nothing... */
2505
2506                 void *auxv;
2507
2508                 auxv = (void*) getauxval(AT_RANDOM);
2509                 if (auxv)
2510                         x ^= *(unsigned*) auxv;
2511 #endif
2512
2513                 x ^= (unsigned) now(CLOCK_REALTIME);
2514                 x ^= (unsigned) gettid();
2515
2516                 srand(x);
2517                 srand_called = true;
2518         }
2519
2520         for (q = p; q < (uint8_t*) p + n; q ++)
2521                 *q = rand();
2522 }
2523
2524 void rename_process(const char name[8]) {
2525         assert(name);
2526
2527         /* This is a like a poor man's setproctitle(). It changes the
2528          * comm field, argv[0], and also the glibc's internally used
2529          * name of the process. For the first one a limit of 16 chars
2530          * applies, to the second one usually one of 10 (i.e. length
2531          * of "/sbin/init"), to the third one one of 7 (i.e. length of
2532          * "systemd"). If you pass a longer string it will be
2533          * truncated */
2534
2535         prctl(PR_SET_NAME, name);
2536
2537         if (program_invocation_name)
2538                 strncpy(program_invocation_name, name, strlen(program_invocation_name));
2539
2540         if (saved_argc > 0) {
2541                 int i;
2542
2543                 if (saved_argv[0])
2544                         strncpy(saved_argv[0], name, strlen(saved_argv[0]));
2545
2546                 for (i = 1; i < saved_argc; i++) {
2547                         if (!saved_argv[i])
2548                                 break;
2549
2550                         memzero(saved_argv[i], strlen(saved_argv[i]));
2551                 }
2552         }
2553 }
2554
2555 void sigset_add_many(sigset_t *ss, ...) {
2556         va_list ap;
2557         int sig;
2558
2559         assert(ss);
2560
2561         va_start(ap, ss);
2562         while ((sig = va_arg(ap, int)) > 0)
2563                 assert_se(sigaddset(ss, sig) == 0);
2564         va_end(ap);
2565 }
2566
2567 int sigprocmask_many(int how, ...) {
2568         va_list ap;
2569         sigset_t ss;
2570         int sig;
2571
2572         assert_se(sigemptyset(&ss) == 0);
2573
2574         va_start(ap, how);
2575         while ((sig = va_arg(ap, int)) > 0)
2576                 assert_se(sigaddset(&ss, sig) == 0);
2577         va_end(ap);
2578
2579         if (sigprocmask(how, &ss, NULL) < 0)
2580                 return -errno;
2581
2582         return 0;
2583 }
2584
2585 char* gethostname_malloc(void) {
2586         struct utsname u;
2587
2588         assert_se(uname(&u) >= 0);
2589
2590         if (!isempty(u.nodename) && !streq(u.nodename, "(none)"))
2591                 return strdup(u.nodename);
2592
2593         return strdup(u.sysname);
2594 }
2595
2596 bool hostname_is_set(void) {
2597         struct utsname u;
2598
2599         assert_se(uname(&u) >= 0);
2600
2601         return !isempty(u.nodename) && !streq(u.nodename, "(none)");
2602 }
2603
2604 char *lookup_uid(uid_t uid) {
2605         long bufsize;
2606         char *name;
2607         _cleanup_free_ char *buf = NULL;
2608         struct passwd pwbuf, *pw = NULL;
2609
2610         /* Shortcut things to avoid NSS lookups */
2611         if (uid == 0)
2612                 return strdup("root");
2613
2614         bufsize = sysconf(_SC_GETPW_R_SIZE_MAX);
2615         if (bufsize <= 0)
2616                 bufsize = 4096;
2617
2618         buf = malloc(bufsize);
2619         if (!buf)
2620                 return NULL;
2621
2622         if (getpwuid_r(uid, &pwbuf, buf, bufsize, &pw) == 0 && pw)
2623                 return strdup(pw->pw_name);
2624
2625         if (asprintf(&name, UID_FMT, uid) < 0)
2626                 return NULL;
2627
2628         return name;
2629 }
2630
2631 char* getlogname_malloc(void) {
2632         uid_t uid;
2633         struct stat st;
2634
2635         if (isatty(STDIN_FILENO) && fstat(STDIN_FILENO, &st) >= 0)
2636                 uid = st.st_uid;
2637         else
2638                 uid = getuid();
2639
2640         return lookup_uid(uid);
2641 }
2642
2643 char *getusername_malloc(void) {
2644         const char *e;
2645
2646         e = getenv("USER");
2647         if (e)
2648                 return strdup(e);
2649
2650         return lookup_uid(getuid());
2651 }
2652
2653 int getttyname_malloc(int fd, char **r) {
2654         char path[PATH_MAX], *c;
2655         int k;
2656
2657         assert(r);
2658
2659         k = ttyname_r(fd, path, sizeof(path));
2660         if (k > 0)
2661                 return -k;
2662
2663         char_array_0(path);
2664
2665         c = strdup(startswith(path, "/dev/") ? path + 5 : path);
2666         if (!c)
2667                 return -ENOMEM;
2668
2669         *r = c;
2670         return 0;
2671 }
2672
2673 int getttyname_harder(int fd, char **r) {
2674         int k;
2675         char *s;
2676
2677         k = getttyname_malloc(fd, &s);
2678         if (k < 0)
2679                 return k;
2680
2681         if (streq(s, "tty")) {
2682                 free(s);
2683                 return get_ctty(0, NULL, r);
2684         }
2685
2686         *r = s;
2687         return 0;
2688 }
2689
2690 int get_ctty_devnr(pid_t pid, dev_t *d) {
2691         int r;
2692         _cleanup_free_ char *line = NULL;
2693         const char *p;
2694         unsigned long ttynr;
2695
2696         assert(pid >= 0);
2697
2698         p = procfs_file_alloca(pid, "stat");
2699         r = read_one_line_file(p, &line);
2700         if (r < 0)
2701                 return r;
2702
2703         p = strrchr(line, ')');
2704         if (!p)
2705                 return -EIO;
2706
2707         p++;
2708
2709         if (sscanf(p, " "
2710                    "%*c "  /* state */
2711                    "%*d "  /* ppid */
2712                    "%*d "  /* pgrp */
2713                    "%*d "  /* session */
2714                    "%lu ", /* ttynr */
2715                    &ttynr) != 1)
2716                 return -EIO;
2717
2718         if (major(ttynr) == 0 && minor(ttynr) == 0)
2719                 return -ENOENT;
2720
2721         if (d)
2722                 *d = (dev_t) ttynr;
2723
2724         return 0;
2725 }
2726
2727 int get_ctty(pid_t pid, dev_t *_devnr, char **r) {
2728         char fn[sizeof("/dev/char/")-1 + 2*DECIMAL_STR_MAX(unsigned) + 1 + 1], *b = NULL;
2729         _cleanup_free_ char *s = NULL;
2730         const char *p;
2731         dev_t devnr;
2732         int k;
2733
2734         assert(r);
2735
2736         k = get_ctty_devnr(pid, &devnr);
2737         if (k < 0)
2738                 return k;
2739
2740         snprintf(fn, sizeof(fn), "/dev/char/%u:%u", major(devnr), minor(devnr));
2741
2742         k = readlink_malloc(fn, &s);
2743         if (k < 0) {
2744
2745                 if (k != -ENOENT)
2746                         return k;
2747
2748                 /* This is an ugly hack */
2749                 if (major(devnr) == 136) {
2750                         asprintf(&b, "pts/%u", minor(devnr));
2751                         goto finish;
2752                 }
2753
2754                 /* Probably something like the ptys which have no
2755                  * symlink in /dev/char. Let's return something
2756                  * vaguely useful. */
2757
2758                 b = strdup(fn + 5);
2759                 goto finish;
2760         }
2761
2762         if (startswith(s, "/dev/"))
2763                 p = s + 5;
2764         else if (startswith(s, "../"))
2765                 p = s + 3;
2766         else
2767                 p = s;
2768
2769         b = strdup(p);
2770
2771 finish:
2772         if (!b)
2773                 return -ENOMEM;
2774
2775         *r = b;
2776         if (_devnr)
2777                 *_devnr = devnr;
2778
2779         return 0;
2780 }
2781
2782 int rm_rf_children_dangerous(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
2783         _cleanup_closedir_ DIR *d = NULL;
2784         int ret = 0;
2785
2786         assert(fd >= 0);
2787
2788         /* This returns the first error we run into, but nevertheless
2789          * tries to go on. This closes the passed fd. */
2790
2791         d = fdopendir(fd);
2792         if (!d) {
2793                 safe_close(fd);
2794
2795                 return errno == ENOENT ? 0 : -errno;
2796         }
2797
2798         for (;;) {
2799                 struct dirent *de;
2800                 bool is_dir, keep_around;
2801                 struct stat st;
2802                 int r;
2803
2804                 errno = 0;
2805                 de = readdir(d);
2806                 if (!de) {
2807                         if (errno != 0 && ret == 0)
2808                                 ret = -errno;
2809                         return ret;
2810                 }
2811
2812                 if (streq(de->d_name, ".") || streq(de->d_name, ".."))
2813                         continue;
2814
2815                 if (de->d_type == DT_UNKNOWN ||
2816                     honour_sticky ||
2817                     (de->d_type == DT_DIR && root_dev)) {
2818                         if (fstatat(fd, de->d_name, &st, AT_SYMLINK_NOFOLLOW) < 0) {
2819                                 if (ret == 0 && errno != ENOENT)
2820                                         ret = -errno;
2821                                 continue;
2822                         }
2823
2824                         is_dir = S_ISDIR(st.st_mode);
2825                         keep_around =
2826                                 honour_sticky &&
2827                                 (st.st_uid == 0 || st.st_uid == getuid()) &&
2828                                 (st.st_mode & S_ISVTX);
2829                 } else {
2830                         is_dir = de->d_type == DT_DIR;
2831                         keep_around = false;
2832                 }
2833
2834                 if (is_dir) {
2835                         int subdir_fd;
2836
2837                         /* if root_dev is set, remove subdirectories only, if device is same as dir */
2838                         if (root_dev && st.st_dev != root_dev->st_dev)
2839                                 continue;
2840
2841                         subdir_fd = openat(fd, de->d_name,
2842                                            O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
2843                         if (subdir_fd < 0) {
2844                                 if (ret == 0 && errno != ENOENT)
2845                                         ret = -errno;
2846                                 continue;
2847                         }
2848
2849                         r = rm_rf_children_dangerous(subdir_fd, only_dirs, honour_sticky, root_dev);
2850                         if (r < 0 && ret == 0)
2851                                 ret = r;
2852
2853                         if (!keep_around)
2854                                 if (unlinkat(fd, de->d_name, AT_REMOVEDIR) < 0) {
2855                                         if (ret == 0 && errno != ENOENT)
2856                                                 ret = -errno;
2857                                 }
2858
2859                 } else if (!only_dirs && !keep_around) {
2860
2861                         if (unlinkat(fd, de->d_name, 0) < 0) {
2862                                 if (ret == 0 && errno != ENOENT)
2863                                         ret = -errno;
2864                         }
2865                 }
2866         }
2867 }
2868
2869 _pure_ static int is_temporary_fs(struct statfs *s) {
2870         assert(s);
2871
2872         return F_TYPE_EQUAL(s->f_type, TMPFS_MAGIC) ||
2873                F_TYPE_EQUAL(s->f_type, RAMFS_MAGIC);
2874 }
2875
2876 int rm_rf_children(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
2877         struct statfs s;
2878
2879         assert(fd >= 0);
2880
2881         if (fstatfs(fd, &s) < 0) {
2882                 safe_close(fd);
2883                 return -errno;
2884         }
2885
2886         /* We refuse to clean disk file systems with this call. This
2887          * is extra paranoia just to be sure we never ever remove
2888          * non-state data */
2889         if (!is_temporary_fs(&s)) {
2890                 log_error("Attempted to remove disk file system, and we can't allow that.");
2891                 safe_close(fd);
2892                 return -EPERM;
2893         }
2894
2895         return rm_rf_children_dangerous(fd, only_dirs, honour_sticky, root_dev);
2896 }
2897
2898 static int rm_rf_internal(const char *path, bool only_dirs, bool delete_root, bool honour_sticky, bool dangerous) {
2899         int fd, r;
2900         struct statfs s;
2901
2902         assert(path);
2903
2904         /* We refuse to clean the root file system with this
2905          * call. This is extra paranoia to never cause a really
2906          * seriously broken system. */
2907         if (path_equal(path, "/")) {
2908                 log_error("Attempted to remove entire root file system, and we can't allow that.");
2909                 return -EPERM;
2910         }
2911
2912         fd = open(path, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
2913         if (fd < 0) {
2914
2915                 if (errno != ENOTDIR)
2916                         return -errno;
2917
2918                 if (!dangerous) {
2919                         if (statfs(path, &s) < 0)
2920                                 return -errno;
2921
2922                         if (!is_temporary_fs(&s)) {
2923                                 log_error("Attempted to remove disk file system, and we can't allow that.");
2924                                 return -EPERM;
2925                         }
2926                 }
2927
2928                 if (delete_root && !only_dirs)
2929                         if (unlink(path) < 0 && errno != ENOENT)
2930                                 return -errno;
2931
2932                 return 0;
2933         }
2934
2935         if (!dangerous) {
2936                 if (fstatfs(fd, &s) < 0) {
2937                         safe_close(fd);
2938                         return -errno;
2939                 }
2940
2941                 if (!is_temporary_fs(&s)) {
2942                         log_error("Attempted to remove disk file system, and we can't allow that.");
2943                         safe_close(fd);
2944                         return -EPERM;
2945                 }
2946         }
2947
2948         r = rm_rf_children_dangerous(fd, only_dirs, honour_sticky, NULL);
2949         if (delete_root) {
2950
2951                 if (honour_sticky && file_is_priv_sticky(path) > 0)
2952                         return r;
2953
2954                 if (rmdir(path) < 0 && errno != ENOENT) {
2955                         if (r == 0)
2956                                 r = -errno;
2957                 }
2958         }
2959
2960         return r;
2961 }
2962
2963 int rm_rf(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
2964         return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, false);
2965 }
2966
2967 int rm_rf_dangerous(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
2968         return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, true);
2969 }
2970
2971 int chmod_and_chown(const char *path, mode_t mode, uid_t uid, gid_t gid) {
2972         assert(path);
2973
2974         /* Under the assumption that we are running privileged we
2975          * first change the access mode and only then hand out
2976          * ownership to avoid a window where access is too open. */
2977
2978         if (mode != (mode_t) -1)
2979                 if (chmod(path, mode) < 0)
2980                         return -errno;
2981
2982         if (uid != (uid_t) -1 || gid != (gid_t) -1)
2983                 if (chown(path, uid, gid) < 0)
2984                         return -errno;
2985
2986         return 0;
2987 }
2988
2989 int fchmod_and_fchown(int fd, mode_t mode, uid_t uid, gid_t gid) {
2990         assert(fd >= 0);
2991
2992         /* Under the assumption that we are running privileged we
2993          * first change the access mode and only then hand out
2994          * ownership to avoid a window where access is too open. */
2995
2996         if (mode != (mode_t) -1)
2997                 if (fchmod(fd, mode) < 0)
2998                         return -errno;
2999
3000         if (uid != (uid_t) -1 || gid != (gid_t) -1)
3001                 if (fchown(fd, uid, gid) < 0)
3002                         return -errno;
3003
3004         return 0;
3005 }
3006
3007 cpu_set_t* cpu_set_malloc(unsigned *ncpus) {
3008         cpu_set_t *r;
3009         unsigned n = 1024;
3010
3011         /* Allocates the cpuset in the right size */
3012
3013         for (;;) {
3014                 if (!(r = CPU_ALLOC(n)))
3015                         return NULL;
3016
3017                 if (sched_getaffinity(0, CPU_ALLOC_SIZE(n), r) >= 0) {
3018                         CPU_ZERO_S(CPU_ALLOC_SIZE(n), r);
3019
3020                         if (ncpus)
3021                                 *ncpus = n;
3022
3023                         return r;
3024                 }
3025
3026                 CPU_FREE(r);
3027
3028                 if (errno != EINVAL)
3029                         return NULL;
3030
3031                 n *= 2;
3032         }
3033 }
3034
3035 int status_vprintf(const char *status, bool ellipse, bool ephemeral, const char *format, va_list ap) {
3036         static const char status_indent[] = "         "; /* "[" STATUS "] " */
3037         _cleanup_free_ char *s = NULL;
3038         _cleanup_close_ int fd = -1;
3039         struct iovec iovec[6] = {};
3040         int n = 0;
3041         static bool prev_ephemeral;
3042
3043         assert(format);
3044
3045         /* This is independent of logging, as status messages are
3046          * optional and go exclusively to the console. */
3047
3048         if (vasprintf(&s, format, ap) < 0)
3049                 return log_oom();
3050
3051         fd = open_terminal("/dev/console", O_WRONLY|O_NOCTTY|O_CLOEXEC);
3052         if (fd < 0)
3053                 return fd;
3054
3055         if (ellipse) {
3056                 char *e;
3057                 size_t emax, sl;
3058                 int c;
3059
3060                 c = fd_columns(fd);
3061                 if (c <= 0)
3062                         c = 80;
3063
3064                 sl = status ? sizeof(status_indent)-1 : 0;
3065
3066                 emax = c - sl - 1;
3067                 if (emax < 3)
3068                         emax = 3;
3069
3070                 e = ellipsize(s, emax, 50);
3071                 if (e) {
3072                         free(s);
3073                         s = e;
3074                 }
3075         }
3076
3077         if (prev_ephemeral)
3078                 IOVEC_SET_STRING(iovec[n++], "\r" ANSI_ERASE_TO_END_OF_LINE);
3079         prev_ephemeral = ephemeral;
3080
3081         if (status) {
3082                 if (!isempty(status)) {
3083                         IOVEC_SET_STRING(iovec[n++], "[");
3084                         IOVEC_SET_STRING(iovec[n++], status);
3085                         IOVEC_SET_STRING(iovec[n++], "] ");
3086                 } else
3087                         IOVEC_SET_STRING(iovec[n++], status_indent);
3088         }
3089
3090         IOVEC_SET_STRING(iovec[n++], s);
3091         if (!ephemeral)
3092                 IOVEC_SET_STRING(iovec[n++], "\n");
3093
3094         if (writev(fd, iovec, n) < 0)
3095                 return -errno;
3096
3097         return 0;
3098 }
3099
3100 int status_printf(const char *status, bool ellipse, bool ephemeral, const char *format, ...) {
3101         va_list ap;
3102         int r;
3103
3104         assert(format);
3105
3106         va_start(ap, format);
3107         r = status_vprintf(status, ellipse, ephemeral, format, ap);
3108         va_end(ap);
3109
3110         return r;
3111 }
3112
3113 char *replace_env(const char *format, char **env) {
3114         enum {
3115                 WORD,
3116                 CURLY,
3117                 VARIABLE
3118         } state = WORD;
3119
3120         const char *e, *word = format;
3121         char *r = NULL, *k;
3122
3123         assert(format);
3124
3125         for (e = format; *e; e ++) {
3126
3127                 switch (state) {
3128
3129                 case WORD:
3130                         if (*e == '$')
3131                                 state = CURLY;
3132                         break;
3133
3134                 case CURLY:
3135                         if (*e == '{') {
3136                                 if (!(k = strnappend(r, word, e-word-1)))
3137                                         goto fail;
3138
3139                                 free(r);
3140                                 r = k;
3141
3142                                 word = e-1;
3143                                 state = VARIABLE;
3144
3145                         } else if (*e == '$') {
3146                                 if (!(k = strnappend(r, word, e-word)))
3147                                         goto fail;
3148
3149                                 free(r);
3150                                 r = k;
3151
3152                                 word = e+1;
3153                                 state = WORD;
3154                         } else
3155                                 state = WORD;
3156                         break;
3157
3158                 case VARIABLE:
3159                         if (*e == '}') {
3160                                 const char *t;
3161
3162                                 t = strempty(strv_env_get_n(env, word+2, e-word-2));
3163
3164                                 k = strappend(r, t);
3165                                 if (!k)
3166                                         goto fail;
3167
3168                                 free(r);
3169                                 r = k;
3170
3171                                 word = e+1;
3172                                 state = WORD;
3173                         }
3174                         break;
3175                 }
3176         }
3177
3178         if (!(k = strnappend(r, word, e-word)))
3179                 goto fail;
3180
3181         free(r);
3182         return k;
3183
3184 fail:
3185         free(r);
3186         return NULL;
3187 }
3188
3189 char **replace_env_argv(char **argv, char **env) {
3190         char **ret, **i;
3191         unsigned k = 0, l = 0;
3192
3193         l = strv_length(argv);
3194
3195         ret = new(char*, l+1);
3196         if (!ret)
3197                 return NULL;
3198
3199         STRV_FOREACH(i, argv) {
3200
3201                 /* If $FOO appears as single word, replace it by the split up variable */
3202                 if ((*i)[0] == '$' && (*i)[1] != '{') {
3203                         char *e;
3204                         char **w, **m;
3205                         unsigned q;
3206
3207                         e = strv_env_get(env, *i+1);
3208                         if (e) {
3209                                 int r;
3210
3211                                 r = strv_split_quoted(&m, e);
3212                                 if (r < 0) {
3213                                         ret[k] = NULL;
3214                                         strv_free(ret);
3215                                         return NULL;
3216                                 }
3217                         } else
3218                                 m = NULL;
3219
3220                         q = strv_length(m);
3221                         l = l + q - 1;
3222
3223                         w = realloc(ret, sizeof(char*) * (l+1));
3224                         if (!w) {
3225                                 ret[k] = NULL;
3226                                 strv_free(ret);
3227                                 strv_free(m);
3228                                 return NULL;
3229                         }
3230
3231                         ret = w;
3232                         if (m) {
3233                                 memcpy(ret + k, m, q * sizeof(char*));
3234                                 free(m);
3235                         }
3236
3237                         k += q;
3238                         continue;
3239                 }
3240
3241                 /* If ${FOO} appears as part of a word, replace it by the variable as-is */
3242                 ret[k] = replace_env(*i, env);
3243                 if (!ret[k]) {
3244                         strv_free(ret);
3245                         return NULL;
3246                 }
3247                 k++;
3248         }
3249
3250         ret[k] = NULL;
3251         return ret;
3252 }
3253
3254 int fd_columns(int fd) {
3255         struct winsize ws = {};
3256
3257         if (ioctl(fd, TIOCGWINSZ, &ws) < 0)
3258                 return -errno;
3259
3260         if (ws.ws_col <= 0)
3261                 return -EIO;
3262
3263         return ws.ws_col;
3264 }
3265
3266 unsigned columns(void) {
3267         const char *e;
3268         int c;
3269
3270         if (_likely_(cached_columns > 0))
3271                 return cached_columns;
3272
3273         c = 0;
3274         e = getenv("COLUMNS");
3275         if (e) {
3276                 int r;
3277
3278                 r = safe_atoi(e, &c);
3279                 if (r < 0) {}
3280                         /* do nothing, we fall back to c = 0 */
3281         }
3282
3283         if (c <= 0)
3284                 c = fd_columns(STDOUT_FILENO);
3285
3286         if (c <= 0)
3287                 c = 80;
3288
3289         cached_columns = c;
3290         return c;
3291 }
3292
3293 int fd_lines(int fd) {
3294         struct winsize ws = {};
3295
3296         if (ioctl(fd, TIOCGWINSZ, &ws) < 0)
3297                 return -errno;
3298
3299         if (ws.ws_row <= 0)
3300                 return -EIO;
3301
3302         return ws.ws_row;
3303 }
3304
3305 unsigned lines(void) {
3306         const char *e;
3307         unsigned l;
3308
3309         if (_likely_(cached_lines > 0))
3310                 return cached_lines;
3311
3312         l = 0;
3313         e = getenv("LINES");
3314         if (e) {
3315                 int r;
3316
3317                 r = safe_atou(e, &l);
3318                 if (r < 0) {}
3319                         /* do nothing, we fall back to l = 0 */
3320         }
3321
3322         if (l <= 0)
3323                 l = fd_lines(STDOUT_FILENO);
3324
3325         if (l <= 0)
3326                 l = 24;
3327
3328         cached_lines = l;
3329         return cached_lines;
3330 }
3331
3332 /* intended to be used as a SIGWINCH sighandler */
3333 void columns_lines_cache_reset(int signum) {
3334         cached_columns = 0;
3335         cached_lines = 0;
3336 }
3337
3338 bool on_tty(void) {
3339         static int cached_on_tty = -1;
3340
3341         if (_unlikely_(cached_on_tty < 0))
3342                 cached_on_tty = isatty(STDOUT_FILENO) > 0;
3343
3344         return cached_on_tty;
3345 }
3346
3347 int files_same(const char *filea, const char *fileb) {
3348         struct stat a, b;
3349
3350         if (stat(filea, &a) < 0)
3351                 return -errno;
3352
3353         if (stat(fileb, &b) < 0)
3354                 return -errno;
3355
3356         return a.st_dev == b.st_dev &&
3357                a.st_ino == b.st_ino;
3358 }
3359
3360 int running_in_chroot(void) {
3361         int ret;
3362
3363         ret = files_same("/proc/1/root", "/");
3364         if (ret < 0)
3365                 return ret;
3366
3367         return ret == 0;
3368 }
3369
3370 static char *ascii_ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3371         size_t x;
3372         char *r;
3373
3374         assert(s);
3375         assert(percent <= 100);
3376         assert(new_length >= 3);
3377
3378         if (old_length <= 3 || old_length <= new_length)
3379                 return strndup(s, old_length);
3380
3381         r = new0(char, new_length+1);
3382         if (!r)
3383                 return NULL;
3384
3385         x = (new_length * percent) / 100;
3386
3387         if (x > new_length - 3)
3388                 x = new_length - 3;
3389
3390         memcpy(r, s, x);
3391         r[x] = '.';
3392         r[x+1] = '.';
3393         r[x+2] = '.';
3394         memcpy(r + x + 3,
3395                s + old_length - (new_length - x - 3),
3396                new_length - x - 3);
3397
3398         return r;
3399 }
3400
3401 char *ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3402         size_t x;
3403         char *e;
3404         const char *i, *j;
3405         unsigned k, len, len2;
3406
3407         assert(s);
3408         assert(percent <= 100);
3409         assert(new_length >= 3);
3410
3411         /* if no multibyte characters use ascii_ellipsize_mem for speed */
3412         if (ascii_is_valid(s))
3413                 return ascii_ellipsize_mem(s, old_length, new_length, percent);
3414
3415         if (old_length <= 3 || old_length <= new_length)
3416                 return strndup(s, old_length);
3417
3418         x = (new_length * percent) / 100;
3419
3420         if (x > new_length - 3)
3421                 x = new_length - 3;
3422
3423         k = 0;
3424         for (i = s; k < x && i < s + old_length; i = utf8_next_char(i)) {
3425                 int c;
3426
3427                 c = utf8_encoded_to_unichar(i);
3428                 if (c < 0)
3429                         return NULL;
3430                 k += unichar_iswide(c) ? 2 : 1;
3431         }
3432
3433         if (k > x) /* last character was wide and went over quota */
3434                 x ++;
3435
3436         for (j = s + old_length; k < new_length && j > i; ) {
3437                 int c;
3438
3439                 j = utf8_prev_char(j);
3440                 c = utf8_encoded_to_unichar(j);
3441                 if (c < 0)
3442                         return NULL;
3443                 k += unichar_iswide(c) ? 2 : 1;
3444         }
3445         assert(i <= j);
3446
3447         /* we don't actually need to ellipsize */
3448         if (i == j)
3449                 return memdup(s, old_length + 1);
3450
3451         /* make space for ellipsis */
3452         j = utf8_next_char(j);
3453
3454         len = i - s;
3455         len2 = s + old_length - j;
3456         e = new(char, len + 3 + len2 + 1);
3457         if (!e)
3458                 return NULL;
3459
3460         /*
3461         printf("old_length=%zu new_length=%zu x=%zu len=%u len2=%u k=%u\n",
3462                old_length, new_length, x, len, len2, k);
3463         */
3464
3465         memcpy(e, s, len);
3466         e[len]   = 0xe2; /* tri-dot ellipsis: … */
3467         e[len + 1] = 0x80;
3468         e[len + 2] = 0xa6;
3469
3470         memcpy(e + len + 3, j, len2 + 1);
3471
3472         return e;
3473 }
3474
3475 char *ellipsize(const char *s, size_t length, unsigned percent) {
3476         return ellipsize_mem(s, strlen(s), length, percent);
3477 }
3478
3479 int touch_file(const char *path, bool parents, usec_t stamp, uid_t uid, gid_t gid, mode_t mode) {
3480         _cleanup_close_ int fd;
3481         int r;
3482
3483         assert(path);
3484
3485         if (parents)
3486                 mkdir_parents(path, 0755);
3487
3488         fd = open(path, O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY, mode > 0 ? mode : 0644);
3489         if (fd < 0)
3490                 return -errno;
3491
3492         if (mode > 0) {
3493                 r = fchmod(fd, mode);
3494                 if (r < 0)
3495                         return -errno;
3496         }
3497
3498         if (uid != (uid_t) -1 || gid != (gid_t) -1) {
3499                 r = fchown(fd, uid, gid);
3500                 if (r < 0)
3501                         return -errno;
3502         }
3503
3504         if (stamp != USEC_INFINITY) {
3505                 struct timespec ts[2];
3506
3507                 timespec_store(&ts[0], stamp);
3508                 ts[1] = ts[0];
3509                 r = futimens(fd, ts);
3510         } else
3511                 r = futimens(fd, NULL);
3512         if (r < 0)
3513                 return -errno;
3514
3515         return 0;
3516 }
3517
3518 int touch(const char *path) {
3519         return touch_file(path, false, USEC_INFINITY, (uid_t) -1, (gid_t) -1, 0);
3520 }
3521
3522 char *unquote(const char *s, const char* quotes) {
3523         size_t l;
3524         assert(s);
3525
3526         /* This is rather stupid, simply removes the heading and
3527          * trailing quotes if there is one. Doesn't care about
3528          * escaping or anything. We should make this smarter one
3529          * day...*/
3530
3531         l = strlen(s);
3532         if (l < 2)
3533                 return strdup(s);
3534
3535         if (strchr(quotes, s[0]) && s[l-1] == s[0])
3536                 return strndup(s+1, l-2);
3537
3538         return strdup(s);
3539 }
3540
3541 char *normalize_env_assignment(const char *s) {
3542         _cleanup_free_ char *name = NULL, *value = NULL, *p = NULL;
3543         char *eq, *r;
3544
3545         eq = strchr(s, '=');
3546         if (!eq) {
3547                 char *t;
3548
3549                 r = strdup(s);
3550                 if (!r)
3551                         return NULL;
3552
3553                 t = strstrip(r);
3554                 if (t == r)
3555                         return r;
3556
3557                 memmove(r, t, strlen(t) + 1);
3558                 return r;
3559         }
3560
3561         name = strndup(s, eq - s);
3562         if (!name)
3563                 return NULL;
3564
3565         p = strdup(eq + 1);
3566         if (!p)
3567                 return NULL;
3568
3569         value = unquote(strstrip(p), QUOTES);
3570         if (!value)
3571                 return NULL;
3572
3573         if (asprintf(&r, "%s=%s", strstrip(name), value) < 0)
3574                 r = NULL;
3575
3576         return r;
3577 }
3578
3579 int wait_for_terminate(pid_t pid, siginfo_t *status) {
3580         siginfo_t dummy;
3581
3582         assert(pid >= 1);
3583
3584         if (!status)
3585                 status = &dummy;
3586
3587         for (;;) {
3588                 zero(*status);
3589
3590                 if (waitid(P_PID, pid, status, WEXITED) < 0) {
3591
3592                         if (errno == EINTR)
3593                                 continue;
3594
3595                         return -errno;
3596                 }
3597
3598                 return 0;
3599         }
3600 }
3601
3602 /*
3603  * Return values:
3604  * < 0 : wait_for_terminate() failed to get the state of the
3605  *       process, the process was terminated by a signal, or
3606  *       failed for an unknown reason.
3607  * >=0 : The process terminated normally, and its exit code is
3608  *       returned.
3609  *
3610  * That is, success is indicated by a return value of zero, and an
3611  * error is indicated by a non-zero value.
3612  */
3613 int wait_for_terminate_and_warn(const char *name, pid_t pid) {
3614         int r;
3615         siginfo_t status;
3616
3617         assert(name);
3618         assert(pid > 1);
3619
3620         r = wait_for_terminate(pid, &status);
3621         if (r < 0) {
3622                 log_warning("Failed to wait for %s: %s", name, strerror(-r));
3623                 return r;
3624         }
3625
3626         if (status.si_code == CLD_EXITED) {
3627                 if (status.si_status != 0) {
3628                         log_warning("%s failed with error code %i.", name, status.si_status);
3629                         return status.si_status;
3630                 }
3631
3632                 log_debug("%s succeeded.", name);
3633                 return 0;
3634
3635         } else if (status.si_code == CLD_KILLED ||
3636                    status.si_code == CLD_DUMPED) {
3637
3638                 log_warning("%s terminated by signal %s.", name, signal_to_string(status.si_status));
3639                 return -EPROTO;
3640         }
3641
3642         log_warning("%s failed due to unknown reason.", name);
3643         return -EPROTO;
3644 }
3645
3646 noreturn void freeze(void) {
3647
3648         /* Make sure nobody waits for us on a socket anymore */
3649         close_all_fds(NULL, 0);
3650
3651         sync();
3652
3653         for (;;)
3654                 pause();
3655 }
3656
3657 bool null_or_empty(struct stat *st) {
3658         assert(st);
3659
3660         if (S_ISREG(st->st_mode) && st->st_size <= 0)
3661                 return true;
3662
3663         if (S_ISCHR(st->st_mode) || S_ISBLK(st->st_mode))
3664                 return true;
3665
3666         return false;
3667 }
3668
3669 int null_or_empty_path(const char *fn) {
3670         struct stat st;
3671
3672         assert(fn);
3673
3674         if (stat(fn, &st) < 0)
3675                 return -errno;
3676
3677         return null_or_empty(&st);
3678 }
3679
3680 int null_or_empty_fd(int fd) {
3681         struct stat st;
3682
3683         assert(fd >= 0);
3684
3685         if (fstat(fd, &st) < 0)
3686                 return -errno;
3687
3688         return null_or_empty(&st);
3689 }
3690
3691 DIR *xopendirat(int fd, const char *name, int flags) {
3692         int nfd;
3693         DIR *d;
3694
3695         assert(!(flags & O_CREAT));
3696
3697         nfd = openat(fd, name, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|flags, 0);
3698         if (nfd < 0)
3699                 return NULL;
3700
3701         d = fdopendir(nfd);
3702         if (!d) {
3703                 safe_close(nfd);
3704                 return NULL;
3705         }
3706
3707         return d;
3708 }
3709
3710 int signal_from_string_try_harder(const char *s) {
3711         int signo;
3712         assert(s);
3713
3714         signo = signal_from_string(s);
3715         if (signo <= 0)
3716                 if (startswith(s, "SIG"))
3717                         return signal_from_string(s+3);
3718
3719         return signo;
3720 }
3721
3722 static char *tag_to_udev_node(const char *tagvalue, const char *by) {
3723         _cleanup_free_ char *t = NULL, *u = NULL;
3724         size_t enc_len;
3725
3726         u = unquote(tagvalue, "\"\'");
3727         if (!u)
3728                 return NULL;
3729
3730         enc_len = strlen(u) * 4 + 1;
3731         t = new(char, enc_len);
3732         if (!t)
3733                 return NULL;
3734
3735         if (encode_devnode_name(u, t, enc_len) < 0)
3736                 return NULL;
3737
3738         return strjoin("/dev/disk/by-", by, "/", t, NULL);
3739 }
3740
3741 char *fstab_node_to_udev_node(const char *p) {
3742         assert(p);
3743
3744         if (startswith(p, "LABEL="))
3745                 return tag_to_udev_node(p+6, "label");
3746
3747         if (startswith(p, "UUID="))
3748                 return tag_to_udev_node(p+5, "uuid");
3749
3750         if (startswith(p, "PARTUUID="))
3751                 return tag_to_udev_node(p+9, "partuuid");
3752
3753         if (startswith(p, "PARTLABEL="))
3754                 return tag_to_udev_node(p+10, "partlabel");
3755
3756         return strdup(p);
3757 }
3758
3759 bool tty_is_vc(const char *tty) {
3760         assert(tty);
3761
3762         return vtnr_from_tty(tty) >= 0;
3763 }
3764
3765 bool tty_is_console(const char *tty) {
3766         assert(tty);
3767
3768         if (startswith(tty, "/dev/"))
3769                 tty += 5;
3770
3771         return streq(tty, "console");
3772 }
3773
3774 int vtnr_from_tty(const char *tty) {
3775         int i, r;
3776
3777         assert(tty);
3778
3779         if (startswith(tty, "/dev/"))
3780                 tty += 5;
3781
3782         if (!startswith(tty, "tty") )
3783                 return -EINVAL;
3784
3785         if (tty[3] < '0' || tty[3] > '9')
3786                 return -EINVAL;
3787
3788         r = safe_atoi(tty+3, &i);
3789         if (r < 0)
3790                 return r;
3791
3792         if (i < 0 || i > 63)
3793                 return -EINVAL;
3794
3795         return i;
3796 }
3797
3798 char *resolve_dev_console(char **active) {
3799         char *tty;
3800
3801         /* Resolve where /dev/console is pointing to, if /sys is actually ours
3802          * (i.e. not read-only-mounted which is a sign for container setups) */
3803
3804         if (path_is_read_only_fs("/sys") > 0)
3805                 return NULL;
3806
3807         if (read_one_line_file("/sys/class/tty/console/active", active) < 0)
3808                 return NULL;
3809
3810         /* If multiple log outputs are configured the last one is what
3811          * /dev/console points to */
3812         tty = strrchr(*active, ' ');
3813         if (tty)
3814                 tty++;
3815         else
3816                 tty = *active;
3817
3818         if (streq(tty, "tty0")) {
3819                 char *tmp;
3820
3821                 /* Get the active VC (e.g. tty1) */
3822                 if (read_one_line_file("/sys/class/tty/tty0/active", &tmp) >= 0) {
3823                         free(*active);
3824                         tty = *active = tmp;
3825                 }
3826         }
3827
3828         return tty;
3829 }
3830
3831 bool tty_is_vc_resolve(const char *tty) {
3832         _cleanup_free_ char *active = NULL;
3833
3834         assert(tty);
3835
3836         if (startswith(tty, "/dev/"))
3837                 tty += 5;
3838
3839         if (streq(tty, "console")) {
3840                 tty = resolve_dev_console(&active);
3841                 if (!tty)
3842                         return false;
3843         }
3844
3845         return tty_is_vc(tty);
3846 }
3847
3848 const char *default_term_for_tty(const char *tty) {
3849         assert(tty);
3850
3851         return tty_is_vc_resolve(tty) ? "TERM=linux" : "TERM=vt102";
3852 }
3853
3854 bool dirent_is_file(const struct dirent *de) {
3855         assert(de);
3856
3857         if (ignore_file(de->d_name))
3858                 return false;
3859
3860         if (de->d_type != DT_REG &&
3861             de->d_type != DT_LNK &&
3862             de->d_type != DT_UNKNOWN)
3863                 return false;
3864
3865         return true;
3866 }
3867
3868 bool dirent_is_file_with_suffix(const struct dirent *de, const char *suffix) {
3869         assert(de);
3870
3871         if (de->d_type != DT_REG &&
3872             de->d_type != DT_LNK &&
3873             de->d_type != DT_UNKNOWN)
3874                 return false;
3875
3876         if (ignore_file_allow_backup(de->d_name))
3877                 return false;
3878
3879         return endswith(de->d_name, suffix);
3880 }
3881
3882 void execute_directory(const char *directory, DIR *d, usec_t timeout, char *argv[]) {
3883         pid_t executor_pid;
3884         int r;
3885
3886         assert(directory);
3887
3888         /* Executes all binaries in a directory in parallel and waits
3889          * for them to finish. Optionally a timeout is applied. */
3890
3891         executor_pid = fork();
3892         if (executor_pid < 0) {
3893                 log_error("Failed to fork: %m");
3894                 return;
3895
3896         } else if (executor_pid == 0) {
3897                 _cleanup_hashmap_free_free_ Hashmap *pids = NULL;
3898                 _cleanup_closedir_ DIR *_d = NULL;
3899                 struct dirent *de;
3900
3901                 /* We fork this all off from a child process so that
3902                  * we can somewhat cleanly make use of SIGALRM to set
3903                  * a time limit */
3904
3905                 reset_all_signal_handlers();
3906                 reset_signal_mask();
3907
3908                 assert_se(prctl(PR_SET_PDEATHSIG, SIGTERM) == 0);
3909
3910                 if (!d) {
3911                         d = _d = opendir(directory);
3912                         if (!d) {
3913                                 if (errno == ENOENT)
3914                                         _exit(EXIT_SUCCESS);
3915
3916                                 log_error("Failed to enumerate directory %s: %m", directory);
3917                                 _exit(EXIT_FAILURE);
3918                         }
3919                 }
3920
3921                 pids = hashmap_new(NULL);
3922                 if (!pids) {
3923                         log_oom();
3924                         _exit(EXIT_FAILURE);
3925                 }
3926
3927                 FOREACH_DIRENT(de, d, break) {
3928                         _cleanup_free_ char *path = NULL;
3929                         pid_t pid;
3930
3931                         if (!dirent_is_file(de))
3932                                 continue;
3933
3934                         path = strjoin(directory, "/", de->d_name, NULL);
3935                         if (!path) {
3936                                 log_oom();
3937                                 _exit(EXIT_FAILURE);
3938                         }
3939
3940                         pid = fork();
3941                         if (pid < 0) {
3942                                 log_error("Failed to fork: %m");
3943                                 continue;
3944                         } else if (pid == 0) {
3945                                 char *_argv[2];
3946
3947                                 assert_se(prctl(PR_SET_PDEATHSIG, SIGTERM) == 0);
3948
3949                                 if (!argv) {
3950                                         _argv[0] = path;
3951                                         _argv[1] = NULL;
3952                                         argv = _argv;
3953                                 } else
3954                                         argv[0] = path;
3955
3956                                 execv(path, argv);
3957                                 log_error("Failed to execute %s: %m", path);
3958                                 _exit(EXIT_FAILURE);
3959                         }
3960
3961                         log_debug("Spawned %s as " PID_FMT ".", path, pid);
3962
3963                         r = hashmap_put(pids, UINT_TO_PTR(pid), path);
3964                         if (r < 0) {
3965                                 log_oom();
3966                                 _exit(EXIT_FAILURE);
3967                         }
3968
3969                         path = NULL;
3970                 }
3971
3972                 /* Abort execution of this process after the
3973                  * timout. We simply rely on SIGALRM as default action
3974                  * terminating the process, and turn on alarm(). */
3975
3976                 if (timeout != USEC_INFINITY)
3977                         alarm((timeout + USEC_PER_SEC - 1) / USEC_PER_SEC);
3978
3979                 while (!hashmap_isempty(pids)) {
3980                         _cleanup_free_ char *path = NULL;
3981                         pid_t pid;
3982
3983                         pid = PTR_TO_UINT(hashmap_first_key(pids));
3984                         assert(pid > 0);
3985
3986                         path = hashmap_remove(pids, UINT_TO_PTR(pid));
3987                         assert(path);
3988
3989                         wait_for_terminate_and_warn(path, pid);
3990                 }
3991
3992                 _exit(EXIT_SUCCESS);
3993         }
3994
3995         wait_for_terminate_and_warn(directory, executor_pid);
3996 }
3997
3998 int kill_and_sigcont(pid_t pid, int sig) {
3999         int r;
4000
4001         r = kill(pid, sig) < 0 ? -errno : 0;
4002
4003         if (r >= 0)
4004                 kill(pid, SIGCONT);
4005
4006         return r;
4007 }
4008
4009 bool nulstr_contains(const char*nulstr, const char *needle) {
4010         const char *i;
4011
4012         if (!nulstr)
4013                 return false;
4014
4015         NULSTR_FOREACH(i, nulstr)
4016                 if (streq(i, needle))
4017                         return true;
4018
4019         return false;
4020 }
4021
4022 bool plymouth_running(void) {
4023         return access("/run/plymouth/pid", F_OK) >= 0;
4024 }
4025
4026 char* strshorten(char *s, size_t l) {
4027         assert(s);
4028
4029         if (l < strlen(s))
4030                 s[l] = 0;
4031
4032         return s;
4033 }
4034
4035 static bool hostname_valid_char(char c) {
4036         return
4037                 (c >= 'a' && c <= 'z') ||
4038                 (c >= 'A' && c <= 'Z') ||
4039                 (c >= '0' && c <= '9') ||
4040                 c == '-' ||
4041                 c == '_' ||
4042                 c == '.';
4043 }
4044
4045 bool hostname_is_valid(const char *s) {
4046         const char *p;
4047         bool dot;
4048
4049         if (isempty(s))
4050                 return false;
4051
4052         for (p = s, dot = true; *p; p++) {
4053                 if (*p == '.') {
4054                         if (dot)
4055                                 return false;
4056
4057                         dot = true;
4058                 } else {
4059                         if (!hostname_valid_char(*p))
4060                                 return false;
4061
4062                         dot = false;
4063                 }
4064         }
4065
4066         if (dot)
4067                 return false;
4068
4069         if (p-s > HOST_NAME_MAX)
4070                 return false;
4071
4072         return true;
4073 }
4074
4075 char* hostname_cleanup(char *s, bool lowercase) {
4076         char *p, *d;
4077         bool dot;
4078
4079         for (p = s, d = s, dot = true; *p; p++) {
4080                 if (*p == '.') {
4081                         if (dot)
4082                                 continue;
4083
4084                         *(d++) = '.';
4085                         dot = true;
4086                 } else if (hostname_valid_char(*p)) {
4087                         *(d++) = lowercase ? tolower(*p) : *p;
4088                         dot = false;
4089                 }
4090
4091         }
4092
4093         if (dot && d > s)
4094                 d[-1] = 0;
4095         else
4096                 *d = 0;
4097
4098         strshorten(s, HOST_NAME_MAX);
4099
4100         return s;
4101 }
4102
4103 bool machine_name_is_valid(const char *s) {
4104
4105         if (!hostname_is_valid(s))
4106                 return false;
4107
4108         /* Machine names should be useful hostnames, but also be
4109          * useful in unit names, hence we enforce a stricter length
4110          * limitation. */
4111
4112         if (strlen(s) > 64)
4113                 return false;
4114
4115         return true;
4116 }
4117
4118 int pipe_eof(int fd) {
4119         struct pollfd pollfd = {
4120                 .fd = fd,
4121                 .events = POLLIN|POLLHUP,
4122         };
4123
4124         int r;
4125
4126         r = poll(&pollfd, 1, 0);
4127         if (r < 0)
4128                 return -errno;
4129
4130         if (r == 0)
4131                 return 0;
4132
4133         return pollfd.revents & POLLHUP;
4134 }
4135
4136 int fd_wait_for_event(int fd, int event, usec_t t) {
4137
4138         struct pollfd pollfd = {
4139                 .fd = fd,
4140                 .events = event,
4141         };
4142
4143         struct timespec ts;
4144         int r;
4145
4146         r = ppoll(&pollfd, 1, t == USEC_INFINITY ? NULL : timespec_store(&ts, t), NULL);
4147         if (r < 0)
4148                 return -errno;
4149
4150         if (r == 0)
4151                 return 0;
4152
4153         return pollfd.revents;
4154 }
4155
4156 int fopen_temporary(const char *path, FILE **_f, char **_temp_path) {
4157         FILE *f;
4158         char *t;
4159         int fd;
4160
4161         assert(path);
4162         assert(_f);
4163         assert(_temp_path);
4164
4165         t = tempfn_xxxxxx(path);
4166         if (!t)
4167                 return -ENOMEM;
4168
4169         fd = mkostemp_safe(t, O_WRONLY|O_CLOEXEC);
4170         if (fd < 0) {
4171                 free(t);
4172                 return -errno;
4173         }
4174
4175         f = fdopen(fd, "we");
4176         if (!f) {
4177                 unlink(t);
4178                 free(t);
4179                 return -errno;
4180         }
4181
4182         *_f = f;
4183         *_temp_path = t;
4184
4185         return 0;
4186 }
4187
4188 int terminal_vhangup_fd(int fd) {
4189         assert(fd >= 0);
4190
4191         if (ioctl(fd, TIOCVHANGUP) < 0)
4192                 return -errno;
4193
4194         return 0;
4195 }
4196
4197 int terminal_vhangup(const char *name) {
4198         _cleanup_close_ int fd;
4199
4200         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4201         if (fd < 0)
4202                 return fd;
4203
4204         return terminal_vhangup_fd(fd);
4205 }
4206
4207 int vt_disallocate(const char *name) {
4208         int fd, r;
4209         unsigned u;
4210
4211         /* Deallocate the VT if possible. If not possible
4212          * (i.e. because it is the active one), at least clear it
4213          * entirely (including the scrollback buffer) */
4214
4215         if (!startswith(name, "/dev/"))
4216                 return -EINVAL;
4217
4218         if (!tty_is_vc(name)) {
4219                 /* So this is not a VT. I guess we cannot deallocate
4220                  * it then. But let's at least clear the screen */
4221
4222                 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4223                 if (fd < 0)
4224                         return fd;
4225
4226                 loop_write(fd,
4227                            "\033[r"    /* clear scrolling region */
4228                            "\033[H"    /* move home */
4229                            "\033[2J",  /* clear screen */
4230                            10, false);
4231                 safe_close(fd);
4232
4233                 return 0;
4234         }
4235
4236         if (!startswith(name, "/dev/tty"))
4237                 return -EINVAL;
4238
4239         r = safe_atou(name+8, &u);
4240         if (r < 0)
4241                 return r;
4242
4243         if (u <= 0)
4244                 return -EINVAL;
4245
4246         /* Try to deallocate */
4247         fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
4248         if (fd < 0)
4249                 return fd;
4250
4251         r = ioctl(fd, VT_DISALLOCATE, u);
4252         safe_close(fd);
4253
4254         if (r >= 0)
4255                 return 0;
4256
4257         if (errno != EBUSY)
4258                 return -errno;
4259
4260         /* Couldn't deallocate, so let's clear it fully with
4261          * scrollback */
4262         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4263         if (fd < 0)
4264                 return fd;
4265
4266         loop_write(fd,
4267                    "\033[r"   /* clear scrolling region */
4268                    "\033[H"   /* move home */
4269                    "\033[3J", /* clear screen including scrollback, requires Linux 2.6.40 */
4270                    10, false);
4271         safe_close(fd);
4272
4273         return 0;
4274 }
4275
4276 int symlink_atomic(const char *from, const char *to) {
4277         _cleanup_free_ char *t = NULL;
4278
4279         assert(from);
4280         assert(to);
4281
4282         t = tempfn_random(to);
4283         if (!t)
4284                 return -ENOMEM;
4285
4286         if (symlink(from, t) < 0)
4287                 return -errno;
4288
4289         if (rename(t, to) < 0) {
4290                 unlink_noerrno(t);
4291                 return -errno;
4292         }
4293
4294         return 0;
4295 }
4296
4297 int mknod_atomic(const char *path, mode_t mode, dev_t dev) {
4298         _cleanup_free_ char *t = NULL;
4299
4300         assert(path);
4301
4302         t = tempfn_random(path);
4303         if (!t)
4304                 return -ENOMEM;
4305
4306         if (mknod(t, mode, dev) < 0)
4307                 return -errno;
4308
4309         if (rename(t, path) < 0) {
4310                 unlink_noerrno(t);
4311                 return -errno;
4312         }
4313
4314         return 0;
4315 }
4316
4317 int mkfifo_atomic(const char *path, mode_t mode) {
4318         _cleanup_free_ char *t = NULL;
4319
4320         assert(path);
4321
4322         t = tempfn_random(path);
4323         if (!t)
4324                 return -ENOMEM;
4325
4326         if (mkfifo(t, mode) < 0)
4327                 return -errno;
4328
4329         if (rename(t, path) < 0) {
4330                 unlink_noerrno(t);
4331                 return -errno;
4332         }
4333
4334         return 0;
4335 }
4336
4337 bool display_is_local(const char *display) {
4338         assert(display);
4339
4340         return
4341                 display[0] == ':' &&
4342                 display[1] >= '0' &&
4343                 display[1] <= '9';
4344 }
4345
4346 int socket_from_display(const char *display, char **path) {
4347         size_t k;
4348         char *f, *c;
4349
4350         assert(display);
4351         assert(path);
4352
4353         if (!display_is_local(display))
4354                 return -EINVAL;
4355
4356         k = strspn(display+1, "0123456789");
4357
4358         f = new(char, strlen("/tmp/.X11-unix/X") + k + 1);
4359         if (!f)
4360                 return -ENOMEM;
4361
4362         c = stpcpy(f, "/tmp/.X11-unix/X");
4363         memcpy(c, display+1, k);
4364         c[k] = 0;
4365
4366         *path = f;
4367
4368         return 0;
4369 }
4370
4371 int get_user_creds(
4372                 const char **username,
4373                 uid_t *uid, gid_t *gid,
4374                 const char **home,
4375                 const char **shell) {
4376
4377         struct passwd *p;
4378         uid_t u;
4379
4380         assert(username);
4381         assert(*username);
4382
4383         /* We enforce some special rules for uid=0: in order to avoid
4384          * NSS lookups for root we hardcode its data. */
4385
4386         if (streq(*username, "root") || streq(*username, "0")) {
4387                 *username = "root";
4388
4389                 if (uid)
4390                         *uid = 0;
4391
4392                 if (gid)
4393                         *gid = 0;
4394
4395                 if (home)
4396                         *home = "/root";
4397
4398                 if (shell)
4399                         *shell = "/bin/sh";
4400
4401                 return 0;
4402         }
4403
4404         if (parse_uid(*username, &u) >= 0) {
4405                 errno = 0;
4406                 p = getpwuid(u);
4407
4408                 /* If there are multiple users with the same id, make
4409                  * sure to leave $USER to the configured value instead
4410                  * of the first occurrence in the database. However if
4411                  * the uid was configured by a numeric uid, then let's
4412                  * pick the real username from /etc/passwd. */
4413                 if (p)
4414                         *username = p->pw_name;
4415         } else {
4416                 errno = 0;
4417                 p = getpwnam(*username);
4418         }
4419
4420         if (!p)
4421                 return errno > 0 ? -errno : -ESRCH;
4422
4423         if (uid)
4424                 *uid = p->pw_uid;
4425
4426         if (gid)
4427                 *gid = p->pw_gid;
4428
4429         if (home)
4430                 *home = p->pw_dir;
4431
4432         if (shell)
4433                 *shell = p->pw_shell;
4434
4435         return 0;
4436 }
4437
4438 char* uid_to_name(uid_t uid) {
4439         struct passwd *p;
4440         char *r;
4441
4442         if (uid == 0)
4443                 return strdup("root");
4444
4445         p = getpwuid(uid);
4446         if (p)
4447                 return strdup(p->pw_name);
4448
4449         if (asprintf(&r, UID_FMT, uid) < 0)
4450                 return NULL;
4451
4452         return r;
4453 }
4454
4455 char* gid_to_name(gid_t gid) {
4456         struct group *p;
4457         char *r;
4458
4459         if (gid == 0)
4460                 return strdup("root");
4461
4462         p = getgrgid(gid);
4463         if (p)
4464                 return strdup(p->gr_name);
4465
4466         if (asprintf(&r, GID_FMT, gid) < 0)
4467                 return NULL;
4468
4469         return r;
4470 }
4471
4472 int get_group_creds(const char **groupname, gid_t *gid) {
4473         struct group *g;
4474         gid_t id;
4475
4476         assert(groupname);
4477
4478         /* We enforce some special rules for gid=0: in order to avoid
4479          * NSS lookups for root we hardcode its data. */
4480
4481         if (streq(*groupname, "root") || streq(*groupname, "0")) {
4482                 *groupname = "root";
4483
4484                 if (gid)
4485                         *gid = 0;
4486
4487                 return 0;
4488         }
4489
4490         if (parse_gid(*groupname, &id) >= 0) {
4491                 errno = 0;
4492                 g = getgrgid(id);
4493
4494                 if (g)
4495                         *groupname = g->gr_name;
4496         } else {
4497                 errno = 0;
4498                 g = getgrnam(*groupname);
4499         }
4500
4501         if (!g)
4502                 return errno > 0 ? -errno : -ESRCH;
4503
4504         if (gid)
4505                 *gid = g->gr_gid;
4506
4507         return 0;
4508 }
4509
4510 int in_gid(gid_t gid) {
4511         gid_t *gids;
4512         int ngroups_max, r, i;
4513
4514         if (getgid() == gid)
4515                 return 1;
4516
4517         if (getegid() == gid)
4518                 return 1;
4519
4520         ngroups_max = sysconf(_SC_NGROUPS_MAX);
4521         assert(ngroups_max > 0);
4522
4523         gids = alloca(sizeof(gid_t) * ngroups_max);
4524
4525         r = getgroups(ngroups_max, gids);
4526         if (r < 0)
4527                 return -errno;
4528
4529         for (i = 0; i < r; i++)
4530                 if (gids[i] == gid)
4531                         return 1;
4532
4533         return 0;
4534 }
4535
4536 int in_group(const char *name) {
4537         int r;
4538         gid_t gid;
4539
4540         r = get_group_creds(&name, &gid);
4541         if (r < 0)
4542                 return r;
4543
4544         return in_gid(gid);
4545 }
4546
4547 int glob_exists(const char *path) {
4548         _cleanup_globfree_ glob_t g = {};
4549         int k;
4550
4551         assert(path);
4552
4553         errno = 0;
4554         k = glob(path, GLOB_NOSORT|GLOB_BRACE, NULL, &g);
4555
4556         if (k == GLOB_NOMATCH)
4557                 return 0;
4558         else if (k == GLOB_NOSPACE)
4559                 return -ENOMEM;
4560         else if (k == 0)
4561                 return !strv_isempty(g.gl_pathv);
4562         else
4563                 return errno ? -errno : -EIO;
4564 }
4565
4566 int glob_extend(char ***strv, const char *path) {
4567         _cleanup_globfree_ glob_t g = {};
4568         int k;
4569         char **p;
4570
4571         errno = 0;
4572         k = glob(path, GLOB_NOSORT|GLOB_BRACE, NULL, &g);
4573
4574         if (k == GLOB_NOMATCH)
4575                 return -ENOENT;
4576         else if (k == GLOB_NOSPACE)
4577                 return -ENOMEM;
4578         else if (k != 0 || strv_isempty(g.gl_pathv))
4579                 return errno ? -errno : -EIO;
4580
4581         STRV_FOREACH(p, g.gl_pathv) {
4582                 k = strv_extend(strv, *p);
4583                 if (k < 0)
4584                         break;
4585         }
4586
4587         return k;
4588 }
4589
4590 int dirent_ensure_type(DIR *d, struct dirent *de) {
4591         struct stat st;
4592
4593         assert(d);
4594         assert(de);
4595
4596         if (de->d_type != DT_UNKNOWN)
4597                 return 0;
4598
4599         if (fstatat(dirfd(d), de->d_name, &st, AT_SYMLINK_NOFOLLOW) < 0)
4600                 return -errno;
4601
4602         de->d_type =
4603                 S_ISREG(st.st_mode)  ? DT_REG  :
4604                 S_ISDIR(st.st_mode)  ? DT_DIR  :
4605                 S_ISLNK(st.st_mode)  ? DT_LNK  :
4606                 S_ISFIFO(st.st_mode) ? DT_FIFO :
4607                 S_ISSOCK(st.st_mode) ? DT_SOCK :
4608                 S_ISCHR(st.st_mode)  ? DT_CHR  :
4609                 S_ISBLK(st.st_mode)  ? DT_BLK  :
4610                                        DT_UNKNOWN;
4611
4612         return 0;
4613 }
4614
4615 int get_files_in_directory(const char *path, char ***list) {
4616         _cleanup_closedir_ DIR *d = NULL;
4617         size_t bufsize = 0, n = 0;
4618         _cleanup_strv_free_ char **l = NULL;
4619
4620         assert(path);
4621
4622         /* Returns all files in a directory in *list, and the number
4623          * of files as return value. If list is NULL returns only the
4624          * number. */
4625
4626         d = opendir(path);
4627         if (!d)
4628                 return -errno;
4629
4630         for (;;) {
4631                 struct dirent *de;
4632
4633                 errno = 0;
4634                 de = readdir(d);
4635                 if (!de && errno != 0)
4636                         return -errno;
4637                 if (!de)
4638                         break;
4639
4640                 dirent_ensure_type(d, de);
4641
4642                 if (!dirent_is_file(de))
4643                         continue;
4644
4645                 if (list) {
4646                         /* one extra slot is needed for the terminating NULL */
4647                         if (!GREEDY_REALLOC(l, bufsize, n + 2))
4648                                 return -ENOMEM;
4649
4650                         l[n] = strdup(de->d_name);
4651                         if (!l[n])
4652                                 return -ENOMEM;
4653
4654                         l[++n] = NULL;
4655                 } else
4656                         n++;
4657         }
4658
4659         if (list) {
4660                 *list = l;
4661                 l = NULL; /* avoid freeing */
4662         }
4663
4664         return n;
4665 }
4666
4667 char *strjoin(const char *x, ...) {
4668         va_list ap;
4669         size_t l;
4670         char *r, *p;
4671
4672         va_start(ap, x);
4673
4674         if (x) {
4675                 l = strlen(x);
4676
4677                 for (;;) {
4678                         const char *t;
4679                         size_t n;
4680
4681                         t = va_arg(ap, const char *);
4682                         if (!t)
4683                                 break;
4684
4685                         n = strlen(t);
4686                         if (n > ((size_t) -1) - l) {
4687                                 va_end(ap);
4688                                 return NULL;
4689                         }
4690
4691                         l += n;
4692                 }
4693         } else
4694                 l = 0;
4695
4696         va_end(ap);
4697
4698         r = new(char, l+1);
4699         if (!r)
4700                 return NULL;
4701
4702         if (x) {
4703                 p = stpcpy(r, x);
4704
4705                 va_start(ap, x);
4706
4707                 for (;;) {
4708                         const char *t;
4709
4710                         t = va_arg(ap, const char *);
4711                         if (!t)
4712                                 break;
4713
4714                         p = stpcpy(p, t);
4715                 }
4716
4717                 va_end(ap);
4718         } else
4719                 r[0] = 0;
4720
4721         return r;
4722 }
4723
4724 bool is_main_thread(void) {
4725         static thread_local int cached = 0;
4726
4727         if (_unlikely_(cached == 0))
4728                 cached = getpid() == gettid() ? 1 : -1;
4729
4730         return cached > 0;
4731 }
4732
4733 int block_get_whole_disk(dev_t d, dev_t *ret) {
4734         char *p, *s;
4735         int r;
4736         unsigned n, m;
4737
4738         assert(ret);
4739
4740         /* If it has a queue this is good enough for us */
4741         if (asprintf(&p, "/sys/dev/block/%u:%u/queue", major(d), minor(d)) < 0)
4742                 return -ENOMEM;
4743
4744         r = access(p, F_OK);
4745         free(p);
4746
4747         if (r >= 0) {
4748                 *ret = d;
4749                 return 0;
4750         }
4751
4752         /* If it is a partition find the originating device */
4753         if (asprintf(&p, "/sys/dev/block/%u:%u/partition", major(d), minor(d)) < 0)
4754                 return -ENOMEM;
4755
4756         r = access(p, F_OK);
4757         free(p);
4758
4759         if (r < 0)
4760                 return -ENOENT;
4761
4762         /* Get parent dev_t */
4763         if (asprintf(&p, "/sys/dev/block/%u:%u/../dev", major(d), minor(d)) < 0)
4764                 return -ENOMEM;
4765
4766         r = read_one_line_file(p, &s);
4767         free(p);
4768
4769         if (r < 0)
4770                 return r;
4771
4772         r = sscanf(s, "%u:%u", &m, &n);
4773         free(s);
4774
4775         if (r != 2)
4776                 return -EINVAL;
4777
4778         /* Only return this if it is really good enough for us. */
4779         if (asprintf(&p, "/sys/dev/block/%u:%u/queue", m, n) < 0)
4780                 return -ENOMEM;
4781
4782         r = access(p, F_OK);
4783         free(p);
4784
4785         if (r >= 0) {
4786                 *ret = makedev(m, n);
4787                 return 0;
4788         }
4789
4790         return -ENOENT;
4791 }
4792
4793 int file_is_priv_sticky(const char *p) {
4794         struct stat st;
4795
4796         assert(p);
4797
4798         if (lstat(p, &st) < 0)
4799                 return -errno;
4800
4801         return
4802                 (st.st_uid == 0 || st.st_uid == getuid()) &&
4803                 (st.st_mode & S_ISVTX);
4804 }
4805
4806 static const char *const ioprio_class_table[] = {
4807         [IOPRIO_CLASS_NONE] = "none",
4808         [IOPRIO_CLASS_RT] = "realtime",
4809         [IOPRIO_CLASS_BE] = "best-effort",
4810         [IOPRIO_CLASS_IDLE] = "idle"
4811 };
4812
4813 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(ioprio_class, int, INT_MAX);
4814
4815 static const char *const sigchld_code_table[] = {
4816         [CLD_EXITED] = "exited",
4817         [CLD_KILLED] = "killed",
4818         [CLD_DUMPED] = "dumped",
4819         [CLD_TRAPPED] = "trapped",
4820         [CLD_STOPPED] = "stopped",
4821         [CLD_CONTINUED] = "continued",
4822 };
4823
4824 DEFINE_STRING_TABLE_LOOKUP(sigchld_code, int);
4825
4826 static const char *const log_facility_unshifted_table[LOG_NFACILITIES] = {
4827         [LOG_FAC(LOG_KERN)] = "kern",
4828         [LOG_FAC(LOG_USER)] = "user",
4829         [LOG_FAC(LOG_MAIL)] = "mail",
4830         [LOG_FAC(LOG_DAEMON)] = "daemon",
4831         [LOG_FAC(LOG_AUTH)] = "auth",
4832         [LOG_FAC(LOG_SYSLOG)] = "syslog",
4833         [LOG_FAC(LOG_LPR)] = "lpr",
4834         [LOG_FAC(LOG_NEWS)] = "news",
4835         [LOG_FAC(LOG_UUCP)] = "uucp",
4836         [LOG_FAC(LOG_CRON)] = "cron",
4837         [LOG_FAC(LOG_AUTHPRIV)] = "authpriv",
4838         [LOG_FAC(LOG_FTP)] = "ftp",
4839         [LOG_FAC(LOG_LOCAL0)] = "local0",
4840         [LOG_FAC(LOG_LOCAL1)] = "local1",
4841         [LOG_FAC(LOG_LOCAL2)] = "local2",
4842         [LOG_FAC(LOG_LOCAL3)] = "local3",
4843         [LOG_FAC(LOG_LOCAL4)] = "local4",
4844         [LOG_FAC(LOG_LOCAL5)] = "local5",
4845         [LOG_FAC(LOG_LOCAL6)] = "local6",
4846         [LOG_FAC(LOG_LOCAL7)] = "local7"
4847 };
4848
4849 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(log_facility_unshifted, int, LOG_FAC(~0));
4850
4851 static const char *const log_level_table[] = {
4852         [LOG_EMERG] = "emerg",
4853         [LOG_ALERT] = "alert",
4854         [LOG_CRIT] = "crit",
4855         [LOG_ERR] = "err",
4856         [LOG_WARNING] = "warning",
4857         [LOG_NOTICE] = "notice",
4858         [LOG_INFO] = "info",
4859         [LOG_DEBUG] = "debug"
4860 };
4861
4862 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(log_level, int, LOG_DEBUG);
4863
4864 static const char* const sched_policy_table[] = {
4865         [SCHED_OTHER] = "other",
4866         [SCHED_BATCH] = "batch",
4867         [SCHED_IDLE] = "idle",
4868         [SCHED_FIFO] = "fifo",
4869         [SCHED_RR] = "rr"
4870 };
4871
4872 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(sched_policy, int, INT_MAX);
4873
4874 static const char* const rlimit_table[_RLIMIT_MAX] = {
4875         [RLIMIT_CPU] = "LimitCPU",
4876         [RLIMIT_FSIZE] = "LimitFSIZE",
4877         [RLIMIT_DATA] = "LimitDATA",
4878         [RLIMIT_STACK] = "LimitSTACK",
4879         [RLIMIT_CORE] = "LimitCORE",
4880         [RLIMIT_RSS] = "LimitRSS",
4881         [RLIMIT_NOFILE] = "LimitNOFILE",
4882         [RLIMIT_AS] = "LimitAS",
4883         [RLIMIT_NPROC] = "LimitNPROC",
4884         [RLIMIT_MEMLOCK] = "LimitMEMLOCK",
4885         [RLIMIT_LOCKS] = "LimitLOCKS",
4886         [RLIMIT_SIGPENDING] = "LimitSIGPENDING",
4887         [RLIMIT_MSGQUEUE] = "LimitMSGQUEUE",
4888         [RLIMIT_NICE] = "LimitNICE",
4889         [RLIMIT_RTPRIO] = "LimitRTPRIO",
4890         [RLIMIT_RTTIME] = "LimitRTTIME"
4891 };
4892
4893 DEFINE_STRING_TABLE_LOOKUP(rlimit, int);
4894
4895 static const char* const ip_tos_table[] = {
4896         [IPTOS_LOWDELAY] = "low-delay",
4897         [IPTOS_THROUGHPUT] = "throughput",
4898         [IPTOS_RELIABILITY] = "reliability",
4899         [IPTOS_LOWCOST] = "low-cost",
4900 };
4901
4902 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(ip_tos, int, 0xff);
4903
4904 static const char *const __signal_table[] = {
4905         [SIGHUP] = "HUP",
4906         [SIGINT] = "INT",
4907         [SIGQUIT] = "QUIT",
4908         [SIGILL] = "ILL",
4909         [SIGTRAP] = "TRAP",
4910         [SIGABRT] = "ABRT",
4911         [SIGBUS] = "BUS",
4912         [SIGFPE] = "FPE",
4913         [SIGKILL] = "KILL",
4914         [SIGUSR1] = "USR1",
4915         [SIGSEGV] = "SEGV",
4916         [SIGUSR2] = "USR2",
4917         [SIGPIPE] = "PIPE",
4918         [SIGALRM] = "ALRM",
4919         [SIGTERM] = "TERM",
4920 #ifdef SIGSTKFLT
4921         [SIGSTKFLT] = "STKFLT",  /* Linux on SPARC doesn't know SIGSTKFLT */
4922 #endif
4923         [SIGCHLD] = "CHLD",
4924         [SIGCONT] = "CONT",
4925         [SIGSTOP] = "STOP",
4926         [SIGTSTP] = "TSTP",
4927         [SIGTTIN] = "TTIN",
4928         [SIGTTOU] = "TTOU",
4929         [SIGURG] = "URG",
4930         [SIGXCPU] = "XCPU",
4931         [SIGXFSZ] = "XFSZ",
4932         [SIGVTALRM] = "VTALRM",
4933         [SIGPROF] = "PROF",
4934         [SIGWINCH] = "WINCH",
4935         [SIGIO] = "IO",
4936         [SIGPWR] = "PWR",
4937         [SIGSYS] = "SYS"
4938 };
4939
4940 DEFINE_PRIVATE_STRING_TABLE_LOOKUP(__signal, int);
4941
4942 const char *signal_to_string(int signo) {
4943         static thread_local char buf[sizeof("RTMIN+")-1 + DECIMAL_STR_MAX(int) + 1];
4944         const char *name;
4945
4946         name = __signal_to_string(signo);
4947         if (name)
4948                 return name;
4949
4950         if (signo >= SIGRTMIN && signo <= SIGRTMAX)
4951                 snprintf(buf, sizeof(buf), "RTMIN+%d", signo - SIGRTMIN);
4952         else
4953                 snprintf(buf, sizeof(buf), "%d", signo);
4954
4955         return buf;
4956 }
4957
4958 int signal_from_string(const char *s) {
4959         int signo;
4960         int offset = 0;
4961         unsigned u;
4962
4963         signo = __signal_from_string(s);
4964         if (signo > 0)
4965                 return signo;
4966
4967         if (startswith(s, "RTMIN+")) {
4968                 s += 6;
4969                 offset = SIGRTMIN;
4970         }
4971         if (safe_atou(s, &u) >= 0) {
4972                 signo = (int) u + offset;
4973                 if (signo > 0 && signo < _NSIG)
4974                         return signo;
4975         }
4976         return -EINVAL;
4977 }
4978
4979 bool kexec_loaded(void) {
4980        bool loaded = false;
4981        char *s;
4982
4983        if (read_one_line_file("/sys/kernel/kexec_loaded", &s) >= 0) {
4984                if (s[0] == '1')
4985                        loaded = true;
4986                free(s);
4987        }
4988        return loaded;
4989 }
4990
4991 int prot_from_flags(int flags) {
4992
4993         switch (flags & O_ACCMODE) {
4994
4995         case O_RDONLY:
4996                 return PROT_READ;
4997
4998         case O_WRONLY:
4999                 return PROT_WRITE;
5000
5001         case O_RDWR:
5002                 return PROT_READ|PROT_WRITE;
5003
5004         default:
5005                 return -EINVAL;
5006         }
5007 }
5008
5009 char *format_bytes(char *buf, size_t l, off_t t) {
5010         unsigned i;
5011
5012         static const struct {
5013                 const char *suffix;
5014                 off_t factor;
5015         } table[] = {
5016                 { "E", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
5017                 { "P", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
5018                 { "T", 1024ULL*1024ULL*1024ULL*1024ULL },
5019                 { "G", 1024ULL*1024ULL*1024ULL },
5020                 { "M", 1024ULL*1024ULL },
5021                 { "K", 1024ULL },
5022         };
5023
5024         for (i = 0; i < ELEMENTSOF(table); i++) {
5025
5026                 if (t >= table[i].factor) {
5027                         snprintf(buf, l,
5028                                  "%llu.%llu%s",
5029                                  (unsigned long long) (t / table[i].factor),
5030                                  (unsigned long long) (((t*10ULL) / table[i].factor) % 10ULL),
5031                                  table[i].suffix);
5032
5033                         goto finish;
5034                 }
5035         }
5036
5037         snprintf(buf, l, "%lluB", (unsigned long long) t);
5038
5039 finish:
5040         buf[l-1] = 0;
5041         return buf;
5042
5043 }
5044
5045 void* memdup(const void *p, size_t l) {
5046         void *r;
5047
5048         assert(p);
5049
5050         r = malloc(l);
5051         if (!r)
5052                 return NULL;
5053
5054         memcpy(r, p, l);
5055         return r;
5056 }
5057
5058 int fd_inc_sndbuf(int fd, size_t n) {
5059         int r, value;
5060         socklen_t l = sizeof(value);
5061
5062         r = getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, &l);
5063         if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
5064                 return 0;
5065
5066         /* If we have the privileges we will ignore the kernel limit. */
5067
5068         value = (int) n;
5069         if (setsockopt(fd, SOL_SOCKET, SO_SNDBUFFORCE, &value, sizeof(value)) < 0)
5070                 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value)) < 0)
5071                         return -errno;
5072
5073         return 1;
5074 }
5075
5076 int fd_inc_rcvbuf(int fd, size_t n) {
5077         int r, value;
5078         socklen_t l = sizeof(value);
5079
5080         r = getsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, &l);
5081         if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
5082                 return 0;
5083
5084         /* If we have the privileges we will ignore the kernel limit. */
5085
5086         value = (int) n;
5087         if (setsockopt(fd, SOL_SOCKET, SO_RCVBUFFORCE, &value, sizeof(value)) < 0)
5088                 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value)) < 0)
5089                         return -errno;
5090         return 1;
5091 }
5092
5093 int fork_agent(pid_t *pid, const int except[], unsigned n_except, const char *path, ...) {
5094         bool stdout_is_tty, stderr_is_tty;
5095         pid_t parent_pid, agent_pid;
5096         sigset_t ss, saved_ss;
5097         unsigned n, i;
5098         va_list ap;
5099         char **l;
5100
5101         assert(pid);
5102         assert(path);
5103
5104         /* Spawns a temporary TTY agent, making sure it goes away when
5105          * we go away */
5106
5107         parent_pid = getpid();
5108
5109         /* First we temporarily block all signals, so that the new
5110          * child has them blocked initially. This way, we can be sure
5111          * that SIGTERMs are not lost we might send to the agent. */
5112         assert_se(sigfillset(&ss) >= 0);
5113         assert_se(sigprocmask(SIG_SETMASK, &ss, &saved_ss) >= 0);
5114
5115         agent_pid = fork();
5116         if (agent_pid < 0) {
5117                 assert_se(sigprocmask(SIG_SETMASK, &saved_ss, NULL) >= 0);
5118                 return -errno;
5119         }
5120
5121         if (agent_pid != 0) {
5122                 assert_se(sigprocmask(SIG_SETMASK, &saved_ss, NULL) >= 0);
5123                 *pid = agent_pid;
5124                 return 0;
5125         }
5126
5127         /* In the child:
5128          *
5129          * Make sure the agent goes away when the parent dies */
5130         if (prctl(PR_SET_PDEATHSIG, SIGTERM) < 0)
5131                 _exit(EXIT_FAILURE);
5132
5133         /* Make sure we actually can kill the agent, if we need to, in
5134          * case somebody invoked us from a shell script that trapped
5135          * SIGTERM or so... */
5136         reset_all_signal_handlers();
5137         reset_signal_mask();
5138
5139         /* Check whether our parent died before we were able
5140          * to set the death signal and unblock the signals */
5141         if (getppid() != parent_pid)
5142                 _exit(EXIT_SUCCESS);
5143
5144         /* Don't leak fds to the agent */
5145         close_all_fds(except, n_except);
5146
5147         stdout_is_tty = isatty(STDOUT_FILENO);
5148         stderr_is_tty = isatty(STDERR_FILENO);
5149
5150         if (!stdout_is_tty || !stderr_is_tty) {
5151                 int fd;
5152
5153                 /* Detach from stdout/stderr. and reopen
5154                  * /dev/tty for them. This is important to
5155                  * ensure that when systemctl is started via
5156                  * popen() or a similar call that expects to
5157                  * read EOF we actually do generate EOF and
5158                  * not delay this indefinitely by because we
5159                  * keep an unused copy of stdin around. */
5160                 fd = open("/dev/tty", O_WRONLY);
5161                 if (fd < 0) {
5162                         log_error("Failed to open /dev/tty: %m");
5163                         _exit(EXIT_FAILURE);
5164                 }
5165
5166                 if (!stdout_is_tty)
5167                         dup2(fd, STDOUT_FILENO);
5168
5169                 if (!stderr_is_tty)
5170                         dup2(fd, STDERR_FILENO);
5171
5172                 if (fd > 2)
5173                         close(fd);
5174         }
5175
5176         /* Count arguments */
5177         va_start(ap, path);
5178         for (n = 0; va_arg(ap, char*); n++)
5179                 ;
5180         va_end(ap);
5181
5182         /* Allocate strv */
5183         l = alloca(sizeof(char *) * (n + 1));
5184
5185         /* Fill in arguments */
5186         va_start(ap, path);
5187         for (i = 0; i <= n; i++)
5188                 l[i] = va_arg(ap, char*);
5189         va_end(ap);
5190
5191         execv(path, l);
5192         _exit(EXIT_FAILURE);
5193 }
5194
5195 int setrlimit_closest(int resource, const struct rlimit *rlim) {
5196         struct rlimit highest, fixed;
5197
5198         assert(rlim);
5199
5200         if (setrlimit(resource, rlim) >= 0)
5201                 return 0;
5202
5203         if (errno != EPERM)
5204                 return -errno;
5205
5206         /* So we failed to set the desired setrlimit, then let's try
5207          * to get as close as we can */
5208         assert_se(getrlimit(resource, &highest) == 0);
5209
5210         fixed.rlim_cur = MIN(rlim->rlim_cur, highest.rlim_max);
5211         fixed.rlim_max = MIN(rlim->rlim_max, highest.rlim_max);
5212
5213         if (setrlimit(resource, &fixed) < 0)
5214                 return -errno;
5215
5216         return 0;
5217 }
5218
5219 int getenv_for_pid(pid_t pid, const char *field, char **_value) {
5220         _cleanup_fclose_ FILE *f = NULL;
5221         char *value = NULL;
5222         int r;
5223         bool done = false;
5224         size_t l;
5225         const char *path;
5226
5227         assert(pid >= 0);
5228         assert(field);
5229         assert(_value);
5230
5231         path = procfs_file_alloca(pid, "environ");
5232
5233         f = fopen(path, "re");
5234         if (!f)
5235                 return -errno;
5236
5237         l = strlen(field);
5238         r = 0;
5239
5240         do {
5241                 char line[LINE_MAX];
5242                 unsigned i;
5243
5244                 for (i = 0; i < sizeof(line)-1; i++) {
5245                         int c;
5246
5247                         c = getc(f);
5248                         if (_unlikely_(c == EOF)) {
5249                                 done = true;
5250                                 break;
5251                         } else if (c == 0)
5252                                 break;
5253
5254                         line[i] = c;
5255                 }
5256                 line[i] = 0;
5257
5258                 if (memcmp(line, field, l) == 0 && line[l] == '=') {
5259                         value = strdup(line + l + 1);
5260                         if (!value)
5261                                 return -ENOMEM;
5262
5263                         r = 1;
5264                         break;
5265                 }
5266
5267         } while (!done);
5268
5269         *_value = value;
5270         return r;
5271 }
5272
5273 bool is_valid_documentation_url(const char *url) {
5274         assert(url);
5275
5276         if (startswith(url, "http://") && url[7])
5277                 return true;
5278
5279         if (startswith(url, "https://") && url[8])
5280                 return true;
5281
5282         if (startswith(url, "file:") && url[5])
5283                 return true;
5284
5285         if (startswith(url, "info:") && url[5])
5286                 return true;
5287
5288         if (startswith(url, "man:") && url[4])
5289                 return true;
5290
5291         return false;
5292 }
5293
5294 bool in_initrd(void) {
5295         static int saved = -1;
5296         struct statfs s;
5297
5298         if (saved >= 0)
5299                 return saved;
5300
5301         /* We make two checks here:
5302          *
5303          * 1. the flag file /etc/initrd-release must exist
5304          * 2. the root file system must be a memory file system
5305          *
5306          * The second check is extra paranoia, since misdetecting an
5307          * initrd can have bad bad consequences due the initrd
5308          * emptying when transititioning to the main systemd.
5309          */
5310
5311         saved = access("/etc/initrd-release", F_OK) >= 0 &&
5312                 statfs("/", &s) >= 0 &&
5313                 is_temporary_fs(&s);
5314
5315         return saved;
5316 }
5317
5318 void warn_melody(void) {
5319         _cleanup_close_ int fd = -1;
5320
5321         fd = open("/dev/console", O_WRONLY|O_CLOEXEC|O_NOCTTY);
5322         if (fd < 0)
5323                 return;
5324
5325         /* Yeah, this is synchronous. Kinda sucks. But well... */
5326
5327         ioctl(fd, KIOCSOUND, (int)(1193180/440));
5328         usleep(125*USEC_PER_MSEC);
5329
5330         ioctl(fd, KIOCSOUND, (int)(1193180/220));
5331         usleep(125*USEC_PER_MSEC);
5332
5333         ioctl(fd, KIOCSOUND, (int)(1193180/220));
5334         usleep(125*USEC_PER_MSEC);
5335
5336         ioctl(fd, KIOCSOUND, 0);
5337 }
5338
5339 int make_console_stdio(void) {
5340         int fd, r;
5341
5342         /* Make /dev/console the controlling terminal and stdin/stdout/stderr */
5343
5344         fd = acquire_terminal("/dev/console", false, true, true, USEC_INFINITY);
5345         if (fd < 0) {
5346                 log_error("Failed to acquire terminal: %s", strerror(-fd));
5347                 return fd;
5348         }
5349
5350         r = make_stdio(fd);
5351         if (r < 0) {
5352                 log_error("Failed to duplicate terminal fd: %s", strerror(-r));
5353                 return r;
5354         }
5355
5356         return 0;
5357 }
5358
5359 int get_home_dir(char **_h) {
5360         struct passwd *p;
5361         const char *e;
5362         char *h;
5363         uid_t u;
5364
5365         assert(_h);
5366
5367         /* Take the user specified one */
5368         e = secure_getenv("HOME");
5369         if (e && path_is_absolute(e)) {
5370                 h = strdup(e);
5371                 if (!h)
5372                         return -ENOMEM;
5373
5374                 *_h = h;
5375                 return 0;
5376         }
5377
5378         /* Hardcode home directory for root to avoid NSS */
5379         u = getuid();
5380         if (u == 0) {
5381                 h = strdup("/root");
5382                 if (!h)
5383                         return -ENOMEM;
5384
5385                 *_h = h;
5386                 return 0;
5387         }
5388
5389         /* Check the database... */
5390         errno = 0;
5391         p = getpwuid(u);
5392         if (!p)
5393                 return errno > 0 ? -errno : -ESRCH;
5394
5395         if (!path_is_absolute(p->pw_dir))
5396                 return -EINVAL;
5397
5398         h = strdup(p->pw_dir);
5399         if (!h)
5400                 return -ENOMEM;
5401
5402         *_h = h;
5403         return 0;
5404 }
5405
5406 int get_shell(char **_s) {
5407         struct passwd *p;
5408         const char *e;
5409         char *s;
5410         uid_t u;
5411
5412         assert(_s);
5413
5414         /* Take the user specified one */
5415         e = getenv("SHELL");
5416         if (e) {
5417                 s = strdup(e);
5418                 if (!s)
5419                         return -ENOMEM;
5420
5421                 *_s = s;
5422                 return 0;
5423         }
5424
5425         /* Hardcode home directory for root to avoid NSS */
5426         u = getuid();
5427         if (u == 0) {
5428                 s = strdup("/bin/sh");
5429                 if (!s)
5430                         return -ENOMEM;
5431
5432                 *_s = s;
5433                 return 0;
5434         }
5435
5436         /* Check the database... */
5437         errno = 0;
5438         p = getpwuid(u);
5439         if (!p)
5440                 return errno > 0 ? -errno : -ESRCH;
5441
5442         if (!path_is_absolute(p->pw_shell))
5443                 return -EINVAL;
5444
5445         s = strdup(p->pw_shell);
5446         if (!s)
5447                 return -ENOMEM;
5448
5449         *_s = s;
5450         return 0;
5451 }
5452
5453 bool filename_is_safe(const char *p) {
5454
5455         if (isempty(p))
5456                 return false;
5457
5458         if (strchr(p, '/'))
5459                 return false;
5460
5461         if (streq(p, "."))
5462                 return false;
5463
5464         if (streq(p, ".."))
5465                 return false;
5466
5467         if (strlen(p) > FILENAME_MAX)
5468                 return false;
5469
5470         return true;
5471 }
5472
5473 bool string_is_safe(const char *p) {
5474         const char *t;
5475
5476         if (!p)
5477                 return false;
5478
5479         for (t = p; *t; t++) {
5480                 if (*t > 0 && *t < ' ')
5481                         return false;
5482
5483                 if (strchr("\\\"\'\0x7f", *t))
5484                         return false;
5485         }
5486
5487         return true;
5488 }
5489
5490 /**
5491  * Check if a string contains control characters. If 'ok' is non-NULL
5492  * it may be a string containing additional CCs to be considered OK.
5493  */
5494 bool string_has_cc(const char *p, const char *ok) {
5495         const char *t;
5496
5497         assert(p);
5498
5499         for (t = p; *t; t++) {
5500                 if (ok && strchr(ok, *t))
5501                         continue;
5502
5503                 if (*t > 0 && *t < ' ')
5504                         return true;
5505
5506                 if (*t == 127)
5507                         return true;
5508         }
5509
5510         return false;
5511 }
5512
5513 bool path_is_safe(const char *p) {
5514
5515         if (isempty(p))
5516                 return false;
5517
5518         if (streq(p, "..") || startswith(p, "../") || endswith(p, "/..") || strstr(p, "/../"))
5519                 return false;
5520
5521         if (strlen(p) > PATH_MAX)
5522                 return false;
5523
5524         /* The following two checks are not really dangerous, but hey, they still are confusing */
5525         if (streq(p, ".") || startswith(p, "./") || endswith(p, "/.") || strstr(p, "/./"))
5526                 return false;
5527
5528         if (strstr(p, "//"))
5529                 return false;
5530
5531         return true;
5532 }
5533
5534 /* hey glibc, APIs with callbacks without a user pointer are so useless */
5535 void *xbsearch_r(const void *key, const void *base, size_t nmemb, size_t size,
5536                  int (*compar) (const void *, const void *, void *), void *arg) {
5537         size_t l, u, idx;
5538         const void *p;
5539         int comparison;
5540
5541         l = 0;
5542         u = nmemb;
5543         while (l < u) {
5544                 idx = (l + u) / 2;
5545                 p = (void *)(((const char *) base) + (idx * size));
5546                 comparison = compar(key, p, arg);
5547                 if (comparison < 0)
5548                         u = idx;
5549                 else if (comparison > 0)
5550                         l = idx + 1;
5551                 else
5552                         return (void *)p;
5553         }
5554         return NULL;
5555 }
5556
5557 bool is_locale_utf8(void) {
5558         const char *set;
5559         static int cached_answer = -1;
5560
5561         if (cached_answer >= 0)
5562                 goto out;
5563
5564         if (!setlocale(LC_ALL, "")) {
5565                 cached_answer = true;
5566                 goto out;
5567         }
5568
5569         set = nl_langinfo(CODESET);
5570         if (!set) {
5571                 cached_answer = true;
5572                 goto out;
5573         }
5574
5575         if (streq(set, "UTF-8")) {
5576                 cached_answer = true;
5577                 goto out;
5578         }
5579
5580         /* For LC_CTYPE=="C" return true, because CTYPE is effectly
5581          * unset and everything can do to UTF-8 nowadays. */
5582         set = setlocale(LC_CTYPE, NULL);
5583         if (!set) {
5584                 cached_answer = true;
5585                 goto out;
5586         }
5587
5588         /* Check result, but ignore the result if C was set
5589          * explicitly. */
5590         cached_answer =
5591                 streq(set, "C") &&
5592                 !getenv("LC_ALL") &&
5593                 !getenv("LC_CTYPE") &&
5594                 !getenv("LANG");
5595
5596 out:
5597         return (bool) cached_answer;
5598 }
5599
5600 const char *draw_special_char(DrawSpecialChar ch) {
5601         static const char *draw_table[2][_DRAW_SPECIAL_CHAR_MAX] = {
5602
5603                 /* UTF-8 */ {
5604                         [DRAW_TREE_VERTICAL]      = "\342\224\202 ",            /* │  */
5605                         [DRAW_TREE_BRANCH]        = "\342\224\234\342\224\200", /* ├─ */
5606                         [DRAW_TREE_RIGHT]         = "\342\224\224\342\224\200", /* └─ */
5607                         [DRAW_TREE_SPACE]         = "  ",                       /*    */
5608                         [DRAW_TRIANGULAR_BULLET]  = "\342\200\243",             /* ‣ */
5609                         [DRAW_BLACK_CIRCLE]       = "\342\227\217",             /* ● */
5610                         [DRAW_ARROW]              = "\342\206\222",             /* → */
5611                         [DRAW_DASH]               = "\342\200\223",             /* – */
5612                 },
5613
5614                 /* ASCII fallback */ {
5615                         [DRAW_TREE_VERTICAL]      = "| ",
5616                         [DRAW_TREE_BRANCH]        = "|-",
5617                         [DRAW_TREE_RIGHT]         = "`-",
5618                         [DRAW_TREE_SPACE]         = "  ",
5619                         [DRAW_TRIANGULAR_BULLET]  = ">",
5620                         [DRAW_BLACK_CIRCLE]       = "*",
5621                         [DRAW_ARROW]              = "->",
5622                         [DRAW_DASH]               = "-",
5623                 }
5624         };
5625
5626         return draw_table[!is_locale_utf8()][ch];
5627 }
5628
5629 char *strreplace(const char *text, const char *old_string, const char *new_string) {
5630         const char *f;
5631         char *t, *r;
5632         size_t l, old_len, new_len;
5633
5634         assert(text);
5635         assert(old_string);
5636         assert(new_string);
5637
5638         old_len = strlen(old_string);
5639         new_len = strlen(new_string);
5640
5641         l = strlen(text);
5642         r = new(char, l+1);
5643         if (!r)
5644                 return NULL;
5645
5646         f = text;
5647         t = r;
5648         while (*f) {
5649                 char *a;
5650                 size_t d, nl;
5651
5652                 if (!startswith(f, old_string)) {
5653                         *(t++) = *(f++);
5654                         continue;
5655                 }
5656
5657                 d = t - r;
5658                 nl = l - old_len + new_len;
5659                 a = realloc(r, nl + 1);
5660                 if (!a)
5661                         goto oom;
5662
5663                 l = nl;
5664                 r = a;
5665                 t = r + d;
5666
5667                 t = stpcpy(t, new_string);
5668                 f += old_len;
5669         }
5670
5671         *t = 0;
5672         return r;
5673
5674 oom:
5675         free(r);
5676         return NULL;
5677 }
5678
5679 char *strip_tab_ansi(char **ibuf, size_t *_isz) {
5680         const char *i, *begin = NULL;
5681         enum {
5682                 STATE_OTHER,
5683                 STATE_ESCAPE,
5684                 STATE_BRACKET
5685         } state = STATE_OTHER;
5686         char *obuf = NULL;
5687         size_t osz = 0, isz;
5688         FILE *f;
5689
5690         assert(ibuf);
5691         assert(*ibuf);
5692
5693         /* Strips ANSI color and replaces TABs by 8 spaces */
5694
5695         isz = _isz ? *_isz : strlen(*ibuf);
5696
5697         f = open_memstream(&obuf, &osz);
5698         if (!f)
5699                 return NULL;
5700
5701         for (i = *ibuf; i < *ibuf + isz + 1; i++) {
5702
5703                 switch (state) {
5704
5705                 case STATE_OTHER:
5706                         if (i >= *ibuf + isz) /* EOT */
5707                                 break;
5708                         else if (*i == '\x1B')
5709                                 state = STATE_ESCAPE;
5710                         else if (*i == '\t')
5711                                 fputs("        ", f);
5712                         else
5713                                 fputc(*i, f);
5714                         break;
5715
5716                 case STATE_ESCAPE:
5717                         if (i >= *ibuf + isz) { /* EOT */
5718                                 fputc('\x1B', f);
5719                                 break;
5720                         } else if (*i == '[') {
5721                                 state = STATE_BRACKET;
5722                                 begin = i + 1;
5723                         } else {
5724                                 fputc('\x1B', f);
5725                                 fputc(*i, f);
5726                                 state = STATE_OTHER;
5727                         }
5728
5729                         break;
5730
5731                 case STATE_BRACKET:
5732
5733                         if (i >= *ibuf + isz || /* EOT */
5734                             (!(*i >= '0' && *i <= '9') && *i != ';' && *i != 'm')) {
5735                                 fputc('\x1B', f);
5736                                 fputc('[', f);
5737                                 state = STATE_OTHER;
5738                                 i = begin-1;
5739                         } else if (*i == 'm')
5740                                 state = STATE_OTHER;
5741                         break;
5742                 }
5743         }
5744
5745         if (ferror(f)) {
5746                 fclose(f);
5747                 free(obuf);
5748                 return NULL;
5749         }
5750
5751         fclose(f);
5752
5753         free(*ibuf);
5754         *ibuf = obuf;
5755
5756         if (_isz)
5757                 *_isz = osz;
5758
5759         return obuf;
5760 }
5761
5762 int on_ac_power(void) {
5763         bool found_offline = false, found_online = false;
5764         _cleanup_closedir_ DIR *d = NULL;
5765
5766         d = opendir("/sys/class/power_supply");
5767         if (!d)
5768                 return -errno;
5769
5770         for (;;) {
5771                 struct dirent *de;
5772                 _cleanup_close_ int fd = -1, device = -1;
5773                 char contents[6];
5774                 ssize_t n;
5775
5776                 errno = 0;
5777                 de = readdir(d);
5778                 if (!de && errno != 0)
5779                         return -errno;
5780
5781                 if (!de)
5782                         break;
5783
5784                 if (ignore_file(de->d_name))
5785                         continue;
5786
5787                 device = openat(dirfd(d), de->d_name, O_DIRECTORY|O_RDONLY|O_CLOEXEC|O_NOCTTY);
5788                 if (device < 0) {
5789                         if (errno == ENOENT || errno == ENOTDIR)
5790                                 continue;
5791
5792                         return -errno;
5793                 }
5794
5795                 fd = openat(device, "type", O_RDONLY|O_CLOEXEC|O_NOCTTY);
5796                 if (fd < 0) {
5797                         if (errno == ENOENT)
5798                                 continue;
5799
5800                         return -errno;
5801                 }
5802
5803                 n = read(fd, contents, sizeof(contents));
5804                 if (n < 0)
5805                         return -errno;
5806
5807                 if (n != 6 || memcmp(contents, "Mains\n", 6))
5808                         continue;
5809
5810                 safe_close(fd);
5811                 fd = openat(device, "online", O_RDONLY|O_CLOEXEC|O_NOCTTY);
5812                 if (fd < 0) {
5813                         if (errno == ENOENT)
5814                                 continue;
5815
5816                         return -errno;
5817                 }
5818
5819                 n = read(fd, contents, sizeof(contents));
5820                 if (n < 0)
5821                         return -errno;
5822
5823                 if (n != 2 || contents[1] != '\n')
5824                         return -EIO;
5825
5826                 if (contents[0] == '1') {
5827                         found_online = true;
5828                         break;
5829                 } else if (contents[0] == '0')
5830                         found_offline = true;
5831                 else
5832                         return -EIO;
5833         }
5834
5835         return found_online || !found_offline;
5836 }
5837
5838 static int search_and_fopen_internal(const char *path, const char *mode, const char *root, char **search, FILE **_f) {
5839         char **i;
5840
5841         assert(path);
5842         assert(mode);
5843         assert(_f);
5844
5845         if (!path_strv_resolve_uniq(search, root))
5846                 return -ENOMEM;
5847
5848         STRV_FOREACH(i, search) {
5849                 _cleanup_free_ char *p = NULL;
5850                 FILE *f;
5851
5852                 if (root)
5853                         p = strjoin(root, *i, "/", path, NULL);
5854                 else
5855                         p = strjoin(*i, "/", path, NULL);
5856                 if (!p)
5857                         return -ENOMEM;
5858
5859                 f = fopen(p, mode);
5860                 if (f) {
5861                         *_f = f;
5862                         return 0;
5863                 }
5864
5865                 if (errno != ENOENT)
5866                         return -errno;
5867         }
5868
5869         return -ENOENT;
5870 }
5871
5872 int search_and_fopen(const char *path, const char *mode, const char *root, const char **search, FILE **_f) {
5873         _cleanup_strv_free_ char **copy = NULL;
5874
5875         assert(path);
5876         assert(mode);
5877         assert(_f);
5878
5879         if (path_is_absolute(path)) {
5880                 FILE *f;
5881
5882                 f = fopen(path, mode);
5883                 if (f) {
5884                         *_f = f;
5885                         return 0;
5886                 }
5887
5888                 return -errno;
5889         }
5890
5891         copy = strv_copy((char**) search);
5892         if (!copy)
5893                 return -ENOMEM;
5894
5895         return search_and_fopen_internal(path, mode, root, copy, _f);
5896 }
5897
5898 int search_and_fopen_nulstr(const char *path, const char *mode, const char *root, const char *search, FILE **_f) {
5899         _cleanup_strv_free_ char **s = NULL;
5900
5901         if (path_is_absolute(path)) {
5902                 FILE *f;
5903
5904                 f = fopen(path, mode);
5905                 if (f) {
5906                         *_f = f;
5907                         return 0;
5908                 }
5909
5910                 return -errno;
5911         }
5912
5913         s = strv_split_nulstr(search);
5914         if (!s)
5915                 return -ENOMEM;
5916
5917         return search_and_fopen_internal(path, mode, root, s, _f);
5918 }
5919
5920 char *strextend(char **x, ...) {
5921         va_list ap;
5922         size_t f, l;
5923         char *r, *p;
5924
5925         assert(x);
5926
5927         l = f = *x ? strlen(*x) : 0;
5928
5929         va_start(ap, x);
5930         for (;;) {
5931                 const char *t;
5932                 size_t n;
5933
5934                 t = va_arg(ap, const char *);
5935                 if (!t)
5936                         break;
5937
5938                 n = strlen(t);
5939                 if (n > ((size_t) -1) - l) {
5940                         va_end(ap);
5941                         return NULL;
5942                 }
5943
5944                 l += n;
5945         }
5946         va_end(ap);
5947
5948         r = realloc(*x, l+1);
5949         if (!r)
5950                 return NULL;
5951
5952         p = r + f;
5953
5954         va_start(ap, x);
5955         for (;;) {
5956                 const char *t;
5957
5958                 t = va_arg(ap, const char *);
5959                 if (!t)
5960                         break;
5961
5962                 p = stpcpy(p, t);
5963         }
5964         va_end(ap);
5965
5966         *p = 0;
5967         *x = r;
5968
5969         return r + l;
5970 }
5971
5972 char *strrep(const char *s, unsigned n) {
5973         size_t l;
5974         char *r, *p;
5975         unsigned i;
5976
5977         assert(s);
5978
5979         l = strlen(s);
5980         p = r = malloc(l * n + 1);
5981         if (!r)
5982                 return NULL;
5983
5984         for (i = 0; i < n; i++)
5985                 p = stpcpy(p, s);
5986
5987         *p = 0;
5988         return r;
5989 }
5990
5991 void* greedy_realloc(void **p, size_t *allocated, size_t need, size_t size) {
5992         size_t a, newalloc;
5993         void *q;
5994
5995         assert(p);
5996         assert(allocated);
5997
5998         if (*allocated >= need)
5999                 return *p;
6000
6001         newalloc = MAX(need * 2, 64u / size);
6002         a = newalloc * size;
6003
6004         /* check for overflows */
6005         if (a < size * need)
6006                 return NULL;
6007
6008         q = realloc(*p, a);
6009         if (!q)
6010                 return NULL;
6011
6012         *p = q;
6013         *allocated = newalloc;
6014         return q;
6015 }
6016
6017 void* greedy_realloc0(void **p, size_t *allocated, size_t need, size_t size) {
6018         size_t prev;
6019         uint8_t *q;
6020
6021         assert(p);
6022         assert(allocated);
6023
6024         prev = *allocated;
6025
6026         q = greedy_realloc(p, allocated, need, size);
6027         if (!q)
6028                 return NULL;
6029
6030         if (*allocated > prev)
6031                 memzero(q + prev * size, (*allocated - prev) * size);
6032
6033         return q;
6034 }
6035
6036 bool id128_is_valid(const char *s) {
6037         size_t i, l;
6038
6039         l = strlen(s);
6040         if (l == 32) {
6041
6042                 /* Simple formatted 128bit hex string */
6043
6044                 for (i = 0; i < l; i++) {
6045                         char c = s[i];
6046
6047                         if (!(c >= '0' && c <= '9') &&
6048                             !(c >= 'a' && c <= 'z') &&
6049                             !(c >= 'A' && c <= 'Z'))
6050                                 return false;
6051                 }
6052
6053         } else if (l == 36) {
6054
6055                 /* Formatted UUID */
6056
6057                 for (i = 0; i < l; i++) {
6058                         char c = s[i];
6059
6060                         if ((i == 8 || i == 13 || i == 18 || i == 23)) {
6061                                 if (c != '-')
6062                                         return false;
6063                         } else {
6064                                 if (!(c >= '0' && c <= '9') &&
6065                                     !(c >= 'a' && c <= 'z') &&
6066                                     !(c >= 'A' && c <= 'Z'))
6067                                         return false;
6068                         }
6069                 }
6070
6071         } else
6072                 return false;
6073
6074         return true;
6075 }
6076
6077 int split_pair(const char *s, const char *sep, char **l, char **r) {
6078         char *x, *a, *b;
6079
6080         assert(s);
6081         assert(sep);
6082         assert(l);
6083         assert(r);
6084
6085         if (isempty(sep))
6086                 return -EINVAL;
6087
6088         x = strstr(s, sep);
6089         if (!x)
6090                 return -EINVAL;
6091
6092         a = strndup(s, x - s);
6093         if (!a)
6094                 return -ENOMEM;
6095
6096         b = strdup(x + strlen(sep));
6097         if (!b) {
6098                 free(a);
6099                 return -ENOMEM;
6100         }
6101
6102         *l = a;
6103         *r = b;
6104
6105         return 0;
6106 }
6107
6108 int shall_restore_state(void) {
6109         _cleanup_free_ char *line = NULL;
6110         const char *word, *state;
6111         size_t l;
6112         int r;
6113
6114         r = proc_cmdline(&line);
6115         if (r < 0)
6116                 return r;
6117         if (r == 0) /* Container ... */
6118                 return 1;
6119
6120         r = 1;
6121
6122         FOREACH_WORD_QUOTED(word, l, line, state) {
6123                 const char *e;
6124                 char n[l+1];
6125                 int k;
6126
6127                 memcpy(n, word, l);
6128                 n[l] = 0;
6129
6130                 e = startswith(n, "systemd.restore_state=");
6131                 if (!e)
6132                         continue;
6133
6134                 k = parse_boolean(e);
6135                 if (k >= 0)
6136                         r = k;
6137         }
6138
6139         return r;
6140 }
6141
6142 int proc_cmdline(char **ret) {
6143         int r;
6144
6145         if (detect_container(NULL) > 0) {
6146                 char *buf = NULL, *p;
6147                 size_t sz = 0;
6148
6149                 r = read_full_file("/proc/1/cmdline", &buf, &sz);
6150                 if (r < 0)
6151                         return r;
6152
6153                 for (p = buf; p + 1 < buf + sz; p++)
6154                         if (*p == 0)
6155                                 *p = ' ';
6156
6157                 *p = 0;
6158                 *ret = buf;
6159                 return 1;
6160         }
6161
6162         r = read_one_line_file("/proc/cmdline", ret);
6163         if (r < 0)
6164                 return r;
6165
6166         return 1;
6167 }
6168
6169 int parse_proc_cmdline(int (*parse_item)(const char *key, const char *value)) {
6170         _cleanup_free_ char *line = NULL;
6171         const char *w, *state;
6172         size_t l;
6173         int r;
6174
6175         assert(parse_item);
6176
6177         r = proc_cmdline(&line);
6178         if (r < 0)
6179                 log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r));
6180         if (r <= 0)
6181                 return 0;
6182
6183         FOREACH_WORD_QUOTED(w, l, line, state) {
6184                 char word[l+1], *value;
6185
6186                 memcpy(word, w, l);
6187                 word[l] = 0;
6188
6189                 /* Filter out arguments that are intended only for the
6190                  * initrd */
6191                 if (!in_initrd() && startswith(word, "rd."))
6192                         continue;
6193
6194                 value = strchr(word, '=');
6195                 if (value)
6196                         *(value++) = 0;
6197
6198                 r = parse_item(word, value);
6199                 if (r < 0)
6200                         return r;
6201         }
6202
6203         return 0;
6204 }
6205
6206 int container_get_leader(const char *machine, pid_t *pid) {
6207         _cleanup_free_ char *s = NULL, *class = NULL;
6208         const char *p;
6209         pid_t leader;
6210         int r;
6211
6212         assert(machine);
6213         assert(pid);
6214
6215         p = strappenda("/run/systemd/machines/", machine);
6216         r = parse_env_file(p, NEWLINE, "LEADER", &s, "CLASS", &class, NULL);
6217         if (r == -ENOENT)
6218                 return -EHOSTDOWN;
6219         if (r < 0)
6220                 return r;
6221         if (!s)
6222                 return -EIO;
6223
6224         if (!streq_ptr(class, "container"))
6225                 return -EIO;
6226
6227         r = parse_pid(s, &leader);
6228         if (r < 0)
6229                 return r;
6230         if (leader <= 1)
6231                 return -EIO;
6232
6233         *pid = leader;
6234         return 0;
6235 }
6236
6237 int namespace_open(pid_t pid, int *pidns_fd, int *mntns_fd, int *netns_fd, int *root_fd) {
6238         _cleanup_close_ int pidnsfd = -1, mntnsfd = -1, netnsfd = -1;
6239         int rfd = -1;
6240
6241         assert(pid >= 0);
6242
6243         if (mntns_fd) {
6244                 const char *mntns;
6245
6246                 mntns = procfs_file_alloca(pid, "ns/mnt");
6247                 mntnsfd = open(mntns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6248                 if (mntnsfd < 0)
6249                         return -errno;
6250         }
6251
6252         if (pidns_fd) {
6253                 const char *pidns;
6254
6255                 pidns = procfs_file_alloca(pid, "ns/pid");
6256                 pidnsfd = open(pidns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6257                 if (pidnsfd < 0)
6258                         return -errno;
6259         }
6260
6261         if (netns_fd) {
6262                 const char *netns;
6263
6264                 netns = procfs_file_alloca(pid, "ns/net");
6265                 netnsfd = open(netns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6266                 if (netnsfd < 0)
6267                         return -errno;
6268         }
6269
6270         if (root_fd) {
6271                 const char *root;
6272
6273                 root = procfs_file_alloca(pid, "root");
6274                 rfd = open(root, O_RDONLY|O_NOCTTY|O_CLOEXEC|O_DIRECTORY);
6275                 if (rfd < 0)
6276                         return -errno;
6277         }
6278
6279         if (pidns_fd)
6280                 *pidns_fd = pidnsfd;
6281
6282         if (mntns_fd)
6283                 *mntns_fd = mntnsfd;
6284
6285         if (netns_fd)
6286                 *netns_fd = netnsfd;
6287
6288         if (root_fd)
6289                 *root_fd = rfd;
6290
6291         pidnsfd = mntnsfd = netnsfd = -1;
6292
6293         return 0;
6294 }
6295
6296 int namespace_enter(int pidns_fd, int mntns_fd, int netns_fd, int root_fd) {
6297
6298         if (pidns_fd >= 0)
6299                 if (setns(pidns_fd, CLONE_NEWPID) < 0)
6300                         return -errno;
6301
6302         if (mntns_fd >= 0)
6303                 if (setns(mntns_fd, CLONE_NEWNS) < 0)
6304                         return -errno;
6305
6306         if (netns_fd >= 0)
6307                 if (setns(netns_fd, CLONE_NEWNET) < 0)
6308                         return -errno;
6309
6310         if (root_fd >= 0) {
6311                 if (fchdir(root_fd) < 0)
6312                         return -errno;
6313
6314                 if (chroot(".") < 0)
6315                         return -errno;
6316         }
6317
6318         if (setresgid(0, 0, 0) < 0)
6319                 return -errno;
6320
6321         if (setgroups(0, NULL) < 0)
6322                 return -errno;
6323
6324         if (setresuid(0, 0, 0) < 0)
6325                 return -errno;
6326
6327         return 0;
6328 }
6329
6330 bool pid_is_unwaited(pid_t pid) {
6331         /* Checks whether a PID is still valid at all, including a zombie */
6332
6333         if (pid <= 0)
6334                 return false;
6335
6336         if (kill(pid, 0) >= 0)
6337                 return true;
6338
6339         return errno != ESRCH;
6340 }
6341
6342 bool pid_is_alive(pid_t pid) {
6343         int r;
6344
6345         /* Checks whether a PID is still valid and not a zombie */
6346
6347         if (pid <= 0)
6348                 return false;
6349
6350         r = get_process_state(pid);
6351         if (r == -ENOENT || r == 'Z')
6352                 return false;
6353
6354         return true;
6355 }
6356
6357 int getpeercred(int fd, struct ucred *ucred) {
6358         socklen_t n = sizeof(struct ucred);
6359         struct ucred u;
6360         int r;
6361
6362         assert(fd >= 0);
6363         assert(ucred);
6364
6365         r = getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &u, &n);
6366         if (r < 0)
6367                 return -errno;
6368
6369         if (n != sizeof(struct ucred))
6370                 return -EIO;
6371
6372         /* Check if the data is actually useful and not suppressed due
6373          * to namespacing issues */
6374         if (u.pid <= 0)
6375                 return -ENODATA;
6376
6377         *ucred = u;
6378         return 0;
6379 }
6380
6381 int getpeersec(int fd, char **ret) {
6382         socklen_t n = 64;
6383         char *s;
6384         int r;
6385
6386         assert(fd >= 0);
6387         assert(ret);
6388
6389         s = new0(char, n);
6390         if (!s)
6391                 return -ENOMEM;
6392
6393         r = getsockopt(fd, SOL_SOCKET, SO_PEERSEC, s, &n);
6394         if (r < 0) {
6395                 free(s);
6396
6397                 if (errno != ERANGE)
6398                         return -errno;
6399
6400                 s = new0(char, n);
6401                 if (!s)
6402                         return -ENOMEM;
6403
6404                 r = getsockopt(fd, SOL_SOCKET, SO_PEERSEC, s, &n);
6405                 if (r < 0) {
6406                         free(s);
6407                         return -errno;
6408                 }
6409         }
6410
6411         if (isempty(s)) {
6412                 free(s);
6413                 return -ENOTSUP;
6414         }
6415
6416         *ret = s;
6417         return 0;
6418 }
6419
6420 /* This is much like like mkostemp() but is subject to umask(). */
6421 int mkostemp_safe(char *pattern, int flags) {
6422         _cleanup_umask_ mode_t u;
6423         int fd;
6424
6425         assert(pattern);
6426
6427         u = umask(077);
6428
6429         fd = mkostemp(pattern, flags);
6430         if (fd < 0)
6431                 return -errno;
6432
6433         return fd;
6434 }
6435
6436 int open_tmpfile(const char *path, int flags) {
6437         char *p;
6438         int fd;
6439
6440         assert(path);
6441
6442 #ifdef O_TMPFILE
6443         /* Try O_TMPFILE first, if it is supported */
6444         fd = open(path, flags|O_TMPFILE, S_IRUSR|S_IWUSR);
6445         if (fd >= 0)
6446                 return fd;
6447 #endif
6448
6449         /* Fall back to unguessable name + unlinking */
6450         p = strappenda(path, "/systemd-tmp-XXXXXX");
6451
6452         fd = mkostemp_safe(p, flags);
6453         if (fd < 0)
6454                 return fd;
6455
6456         unlink(p);
6457         return fd;
6458 }
6459
6460 int fd_warn_permissions(const char *path, int fd) {
6461         struct stat st;
6462
6463         if (fstat(fd, &st) < 0)
6464                 return -errno;
6465
6466         if (st.st_mode & 0111)
6467                 log_warning("Configuration file %s is marked executable. Please remove executable permission bits. Proceeding anyway.", path);
6468
6469         if (st.st_mode & 0002)
6470                 log_warning("Configuration file %s is marked world-writable. Please remove world writability permission bits. Proceeding anyway.", path);
6471
6472         if (getpid() == 1 && (st.st_mode & 0044) != 0044)
6473                 log_warning("Configuration file %s is marked world-inaccessible. This has no effect as configuration data is accessible via APIs without restrictions. Proceeding anyway.", path);
6474
6475         return 0;
6476 }
6477
6478 unsigned long personality_from_string(const char *p) {
6479
6480         /* Parse a personality specifier. We introduce our own
6481          * identifiers that indicate specific ABIs, rather than just
6482          * hints regarding the register size, since we want to keep
6483          * things open for multiple locally supported ABIs for the
6484          * same register size. We try to reuse the ABI identifiers
6485          * used by libseccomp. */
6486
6487 #if defined(__x86_64__)
6488
6489         if (streq(p, "x86"))
6490                 return PER_LINUX32;
6491
6492         if (streq(p, "x86-64"))
6493                 return PER_LINUX;
6494
6495 #elif defined(__i386__)
6496
6497         if (streq(p, "x86"))
6498                 return PER_LINUX;
6499 #endif
6500
6501         /* personality(7) documents that 0xffffffffUL is used for
6502          * querying the current personality, hence let's use that here
6503          * as error indicator. */
6504         return 0xffffffffUL;
6505 }
6506
6507 const char* personality_to_string(unsigned long p) {
6508
6509 #if defined(__x86_64__)
6510
6511         if (p == PER_LINUX32)
6512                 return "x86";
6513
6514         if (p == PER_LINUX)
6515                 return "x86-64";
6516
6517 #elif defined(__i386__)
6518
6519         if (p == PER_LINUX)
6520                 return "x86";
6521 #endif
6522
6523         return NULL;
6524 }
6525
6526 uint64_t physical_memory(void) {
6527         long mem;
6528
6529         /* We return this as uint64_t in case we are running as 32bit
6530          * process on a 64bit kernel with huge amounts of memory */
6531
6532         mem = sysconf(_SC_PHYS_PAGES);
6533         assert(mem > 0);
6534
6535         return (uint64_t) mem * (uint64_t) page_size();
6536 }
6537
6538 char* mount_test_option(const char *haystack, const char *needle) {
6539
6540         struct mntent me = {
6541                 .mnt_opts = (char*) haystack
6542         };
6543
6544         assert(needle);
6545
6546         /* Like glibc's hasmntopt(), but works on a string, not a
6547          * struct mntent */
6548
6549         if (!haystack)
6550                 return NULL;
6551
6552         return hasmntopt(&me, needle);
6553 }
6554
6555 void hexdump(FILE *f, const void *p, size_t s) {
6556         const uint8_t *b = p;
6557         unsigned n = 0;
6558
6559         assert(s == 0 || b);
6560
6561         while (s > 0) {
6562                 size_t i;
6563
6564                 fprintf(f, "%04x  ", n);
6565
6566                 for (i = 0; i < 16; i++) {
6567
6568                         if (i >= s)
6569                                 fputs("   ", f);
6570                         else
6571                                 fprintf(f, "%02x ", b[i]);
6572
6573                         if (i == 7)
6574                                 fputc(' ', f);
6575                 }
6576
6577                 fputc(' ', f);
6578
6579                 for (i = 0; i < 16; i++) {
6580
6581                         if (i >= s)
6582                                 fputc(' ', f);
6583                         else
6584                                 fputc(isprint(b[i]) ? (char) b[i] : '.', f);
6585                 }
6586
6587                 fputc('\n', f);
6588
6589                 if (s < 16)
6590                         break;
6591
6592                 n += 16;
6593                 b += 16;
6594                 s -= 16;
6595         }
6596 }
6597
6598 int update_reboot_param_file(const char *param) {
6599         int r = 0;
6600
6601         if (param) {
6602
6603                 r = write_string_file(REBOOT_PARAM_FILE, param);
6604                 if (r < 0)
6605                         log_error("Failed to write reboot param to "
6606                                   REBOOT_PARAM_FILE": %s", strerror(-r));
6607         } else
6608                 unlink(REBOOT_PARAM_FILE);
6609
6610         return r;
6611 }
6612
6613 int umount_recursive(const char *prefix, int flags) {
6614         bool again;
6615         int n = 0, r;
6616
6617         /* Try to umount everything recursively below a
6618          * directory. Also, take care of stacked mounts, and keep
6619          * unmounting them until they are gone. */
6620
6621         do {
6622                 _cleanup_fclose_ FILE *proc_self_mountinfo = NULL;
6623
6624                 again = false;
6625                 r = 0;
6626
6627                 proc_self_mountinfo = fopen("/proc/self/mountinfo", "re");
6628                 if (!proc_self_mountinfo)
6629                         return -errno;
6630
6631                 for (;;) {
6632                         _cleanup_free_ char *path = NULL, *p = NULL;
6633                         int k;
6634
6635                         k = fscanf(proc_self_mountinfo,
6636                                    "%*s "       /* (1) mount id */
6637                                    "%*s "       /* (2) parent id */
6638                                    "%*s "       /* (3) major:minor */
6639                                    "%*s "       /* (4) root */
6640                                    "%ms "       /* (5) mount point */
6641                                    "%*s"        /* (6) mount options */
6642                                    "%*[^-]"     /* (7) optional fields */
6643                                    "- "         /* (8) separator */
6644                                    "%*s "       /* (9) file system type */
6645                                    "%*s"        /* (10) mount source */
6646                                    "%*s"        /* (11) mount options 2 */
6647                                    "%*[^\n]",   /* some rubbish at the end */
6648                                    &path);
6649                         if (k != 1) {
6650                                 if (k == EOF)
6651                                         break;
6652
6653                                 continue;
6654                         }
6655
6656                         p = cunescape(path);
6657                         if (!p)
6658                                 return -ENOMEM;
6659
6660                         if (!path_startswith(p, prefix))
6661                                 continue;
6662
6663                         if (umount2(p, flags) < 0) {
6664                                 r = -errno;
6665                                 continue;
6666                         }
6667
6668                         again = true;
6669                         n++;
6670
6671                         break;
6672                 }
6673
6674         } while (again);
6675
6676         return r ? r : n;
6677 }
6678
6679 int bind_remount_recursive(const char *prefix, bool ro) {
6680         _cleanup_set_free_free_ Set *done = NULL;
6681         _cleanup_free_ char *cleaned = NULL;
6682         int r;
6683
6684         /* Recursively remount a directory (and all its submounts)
6685          * read-only or read-write. If the directory is already
6686          * mounted, we reuse the mount and simply mark it
6687          * MS_BIND|MS_RDONLY (or remove the MS_RDONLY for read-write
6688          * operation). If it isn't we first make it one. Afterwards we
6689          * apply MS_BIND|MS_RDONLY (or remove MS_RDONLY) to all
6690          * submounts we can access, too. When mounts are stacked on
6691          * the same mount point we only care for each individual
6692          * "top-level" mount on each point, as we cannot
6693          * influence/access the underlying mounts anyway. We do not
6694          * have any effect on future submounts that might get
6695          * propagated, they migt be writable. This includes future
6696          * submounts that have been triggered via autofs. */
6697
6698         cleaned = strdup(prefix);
6699         if (!cleaned)
6700                 return -ENOMEM;
6701
6702         path_kill_slashes(cleaned);
6703
6704         done = set_new(&string_hash_ops);
6705         if (!done)
6706                 return -ENOMEM;
6707
6708         for (;;) {
6709                 _cleanup_fclose_ FILE *proc_self_mountinfo = NULL;
6710                 _cleanup_set_free_free_ Set *todo = NULL;
6711                 bool top_autofs = false;
6712                 char *x;
6713
6714                 todo = set_new(&string_hash_ops);
6715                 if (!todo)
6716                         return -ENOMEM;
6717
6718                 proc_self_mountinfo = fopen("/proc/self/mountinfo", "re");
6719                 if (!proc_self_mountinfo)
6720                         return -errno;
6721
6722                 for (;;) {
6723                         _cleanup_free_ char *path = NULL, *p = NULL, *type = NULL;
6724                         int k;
6725
6726                         k = fscanf(proc_self_mountinfo,
6727                                    "%*s "       /* (1) mount id */
6728                                    "%*s "       /* (2) parent id */
6729                                    "%*s "       /* (3) major:minor */
6730                                    "%*s "       /* (4) root */
6731                                    "%ms "       /* (5) mount point */
6732                                    "%*s"        /* (6) mount options (superblock) */
6733                                    "%*[^-]"     /* (7) optional fields */
6734                                    "- "         /* (8) separator */
6735                                    "%ms "       /* (9) file system type */
6736                                    "%*s"        /* (10) mount source */
6737                                    "%*s"        /* (11) mount options (bind mount) */
6738                                    "%*[^\n]",   /* some rubbish at the end */
6739                                    &path,
6740                                    &type);
6741                         if (k != 2) {
6742                                 if (k == EOF)
6743                                         break;
6744
6745                                 continue;
6746                         }
6747
6748                         p = cunescape(path);
6749                         if (!p)
6750                                 return -ENOMEM;
6751
6752                         /* Let's ignore autofs mounts.  If they aren't
6753                          * triggered yet, we want to avoid triggering
6754                          * them, as we don't make any guarantees for
6755                          * future submounts anyway.  If they are
6756                          * already triggered, then we will find
6757                          * another entry for this. */
6758                         if (streq(type, "autofs")) {
6759                                 top_autofs = top_autofs || path_equal(cleaned, p);
6760                                 continue;
6761                         }
6762
6763                         if (path_startswith(p, cleaned) &&
6764                             !set_contains(done, p)) {
6765
6766                                 r = set_consume(todo, p);
6767                                 p = NULL;
6768
6769                                 if (r == -EEXIST)
6770                                         continue;
6771                                 if (r < 0)
6772                                         return r;
6773                         }
6774                 }
6775
6776                 /* If we have no submounts to process anymore and if
6777                  * the root is either already done, or an autofs, we
6778                  * are done */
6779                 if (set_isempty(todo) &&
6780                     (top_autofs || set_contains(done, cleaned)))
6781                         return 0;
6782
6783                 if (!set_contains(done, cleaned) &&
6784                     !set_contains(todo, cleaned)) {
6785                         /* The prefix directory itself is not yet a
6786                          * mount, make it one. */
6787                         if (mount(cleaned, cleaned, NULL, MS_BIND|MS_REC, NULL) < 0)
6788                                 return -errno;
6789
6790                         if (mount(NULL, prefix, NULL, MS_BIND|MS_REMOUNT|(ro ? MS_RDONLY : 0), NULL) < 0)
6791                                 return -errno;
6792
6793                         x = strdup(cleaned);
6794                         if (!x)
6795                                 return -ENOMEM;
6796
6797                         r = set_consume(done, x);
6798                         if (r < 0)
6799                                 return r;
6800                 }
6801
6802                 while ((x = set_steal_first(todo))) {
6803
6804                         r = set_consume(done, x);
6805                         if (r == -EEXIST)
6806                                 continue;
6807                         if (r < 0)
6808                                 return r;
6809
6810                         if (mount(NULL, x, NULL, MS_BIND|MS_REMOUNT|(ro ? MS_RDONLY : 0), NULL) < 0) {
6811
6812                                 /* Deal with mount points that are
6813                                  * obstructed by a later mount */
6814
6815                                 if (errno != ENOENT)
6816                                         return -errno;
6817                         }
6818
6819                 }
6820         }
6821 }
6822
6823 int fflush_and_check(FILE *f) {
6824         assert(f);
6825
6826         errno = 0;
6827         fflush(f);
6828
6829         if (ferror(f))
6830                 return errno ? -errno : -EIO;
6831
6832         return 0;
6833 }
6834
6835 char *tempfn_xxxxxx(const char *p) {
6836         const char *fn;
6837         char *t;
6838         size_t k;
6839
6840         assert(p);
6841
6842         t = new(char, strlen(p) + 1 + 6 + 1);
6843         if (!t)
6844                 return NULL;
6845
6846         fn = basename(p);
6847         k = fn - p;
6848
6849         strcpy(stpcpy(stpcpy(mempcpy(t, p, k), "."), fn), "XXXXXX");
6850
6851         return t;
6852 }
6853
6854 char *tempfn_random(const char *p) {
6855         const char *fn;
6856         char *t, *x;
6857         uint64_t u;
6858         size_t k;
6859         unsigned i;
6860
6861         assert(p);
6862
6863         t = new(char, strlen(p) + 1 + 16 + 1);
6864         if (!t)
6865                 return NULL;
6866
6867         fn = basename(p);
6868         k = fn - p;
6869
6870         x = stpcpy(stpcpy(mempcpy(t, p, k), "."), fn);
6871
6872         u = random_u64();
6873         for (i = 0; i < 16; i++) {
6874                 *(x++) = hexchar(u & 0xF);
6875                 u >>= 4;
6876         }
6877
6878         *x = 0;
6879
6880         return t;
6881 }
6882
6883 /* make sure the hostname is not "localhost" */
6884 bool is_localhost(const char *hostname) {
6885         assert(hostname);
6886
6887         /* This tries to identify local host and domain names
6888          * described in RFC6761 plus the redhatism of .localdomain */
6889
6890         return streq(hostname, "localhost") ||
6891                streq(hostname, "localhost.") ||
6892                streq(hostname, "localdomain.") ||
6893                streq(hostname, "localdomain") ||
6894                endswith(hostname, ".localhost") ||
6895                endswith(hostname, ".localhost.") ||
6896                endswith(hostname, ".localdomain") ||
6897                endswith(hostname, ".localdomain.");
6898 }
6899
6900 int take_password_lock(const char *root) {
6901
6902         struct flock flock = {
6903                 .l_type = F_WRLCK,
6904                 .l_whence = SEEK_SET,
6905                 .l_start = 0,
6906                 .l_len = 0,
6907         };
6908
6909         const char *path;
6910         int fd, r;
6911
6912         /* This is roughly the same as lckpwdf(), but not as awful. We
6913          * don't want to use alarm() and signals, hence we implement
6914          * our own trivial version of this.
6915          *
6916          * Note that shadow-utils also takes per-database locks in
6917          * addition to lckpwdf(). However, we don't given that they
6918          * are redundant as they they invoke lckpwdf() first and keep
6919          * it during everything they do. The per-database locks are
6920          * awfully racy, and thus we just won't do them. */
6921
6922         if (root)
6923                 path = strappenda(root, "/etc/.pwd.lock");
6924         else
6925                 path = "/etc/.pwd.lock";
6926
6927         fd = open(path, O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW, 0600);
6928         if (fd < 0)
6929                 return -errno;
6930
6931         r = fcntl(fd, F_SETLKW, &flock);
6932         if (r < 0) {
6933                 safe_close(fd);
6934                 return -errno;
6935         }
6936
6937         return fd;
6938 }
6939
6940 int is_symlink(const char *path) {
6941         struct stat info;
6942
6943         if (lstat(path, &info) < 0)
6944                 return -errno;
6945
6946         if (S_ISLNK(info.st_mode))
6947                 return 1;
6948
6949         return 0;
6950 }
6951
6952 int unquote_first_word(const char **p, char **ret) {
6953         _cleanup_free_ char *s = NULL;
6954         size_t allocated = 0, sz = 0;
6955
6956         enum {
6957                 START,
6958                 VALUE,
6959                 VALUE_ESCAPE,
6960                 SINGLE_QUOTE,
6961                 SINGLE_QUOTE_ESCAPE,
6962                 DOUBLE_QUOTE,
6963                 DOUBLE_QUOTE_ESCAPE,
6964                 SPACE,
6965         } state = START;
6966
6967         assert(p);
6968         assert(*p);
6969         assert(ret);
6970
6971         /* Parses the first word of a string, and returns it in
6972          * *ret. Removes all quotes in the process. When parsing fails
6973          * (because of an uneven number of quotes or similar), leaves
6974          * the pointer *p at the first invalid character. */
6975
6976         for (;;) {
6977                 char c = **p;
6978
6979                 switch (state) {
6980
6981                 case START:
6982                         if (c == 0)
6983                                 goto finish;
6984                         else if (strchr(WHITESPACE, c))
6985                                 break;
6986
6987                         state = VALUE;
6988                         /* fallthrough */
6989
6990                 case VALUE:
6991                         if (c == 0)
6992                                 goto finish;
6993                         else if (c == '\'')
6994                                 state = SINGLE_QUOTE;
6995                         else if (c == '\\')
6996                                 state = VALUE_ESCAPE;
6997                         else if (c == '\"')
6998                                 state = DOUBLE_QUOTE;
6999                         else if (strchr(WHITESPACE, c))
7000                                 state = SPACE;
7001                         else {
7002                                 if (!GREEDY_REALLOC(s, allocated, sz+2))
7003                                         return -ENOMEM;
7004
7005                                 s[sz++] = c;
7006                         }
7007
7008                         break;
7009
7010                 case VALUE_ESCAPE:
7011                         if (c == 0)
7012                                 return -EINVAL;
7013
7014                         if (!GREEDY_REALLOC(s, allocated, sz+2))
7015                                 return -ENOMEM;
7016
7017                         s[sz++] = c;
7018                         state = VALUE;
7019
7020                         break;
7021
7022                 case SINGLE_QUOTE:
7023                         if (c == 0)
7024                                 return -EINVAL;
7025                         else if (c == '\'')
7026                                 state = VALUE;
7027                         else if (c == '\\')
7028                                 state = SINGLE_QUOTE_ESCAPE;
7029                         else {
7030                                 if (!GREEDY_REALLOC(s, allocated, sz+2))
7031                                         return -ENOMEM;
7032
7033                                 s[sz++] = c;
7034                         }
7035
7036                         break;
7037
7038                 case SINGLE_QUOTE_ESCAPE:
7039                         if (c == 0)
7040                                 return -EINVAL;
7041
7042                         if (!GREEDY_REALLOC(s, allocated, sz+2))
7043                                 return -ENOMEM;
7044
7045                         s[sz++] = c;
7046                         state = SINGLE_QUOTE;
7047                         break;
7048
7049                 case DOUBLE_QUOTE:
7050                         if (c == 0)
7051                                 return -EINVAL;
7052                         else if (c == '\"')
7053                                 state = VALUE;
7054                         else if (c == '\\')
7055                                 state = DOUBLE_QUOTE_ESCAPE;
7056                         else {
7057                                 if (!GREEDY_REALLOC(s, allocated, sz+2))
7058                                         return -ENOMEM;
7059
7060                                 s[sz++] = c;
7061                         }
7062
7063                         break;
7064
7065                 case DOUBLE_QUOTE_ESCAPE:
7066                         if (c == 0)
7067                                 return -EINVAL;
7068
7069                         if (!GREEDY_REALLOC(s, allocated, sz+2))
7070                                 return -ENOMEM;
7071
7072                         s[sz++] = c;
7073                         state = DOUBLE_QUOTE;
7074                         break;
7075
7076                 case SPACE:
7077                         if (c == 0)
7078                                 goto finish;
7079                         if (!strchr(WHITESPACE, c))
7080                                 goto finish;
7081
7082                         break;
7083                 }
7084
7085                 (*p) ++;
7086         }
7087
7088 finish:
7089         if (!s) {
7090                 *ret = NULL;
7091                 return 0;
7092         }
7093
7094         s[sz] = 0;
7095         *ret = s;
7096         s = NULL;
7097
7098         return 1;
7099 }
7100
7101 int unquote_many_words(const char **p, ...) {
7102         va_list ap;
7103         char **l;
7104         int n = 0, i, c, r;
7105
7106         /* Parses a number of words from a string, stripping any
7107          * quotes if necessary. */
7108
7109         assert(p);
7110
7111         /* Count how many words are expected */
7112         va_start(ap, p);
7113         for (;;) {
7114                 if (!va_arg(ap, char **))
7115                         break;
7116                 n++;
7117         }
7118         va_end(ap);
7119
7120         if (n <= 0)
7121                 return 0;
7122
7123         /* Read all words into a temporary array */
7124         l = newa0(char*, n);
7125         for (c = 0; c < n; c++) {
7126
7127                 r = unquote_first_word(p, &l[c]);
7128                 if (r < 0) {
7129                         int j;
7130
7131                         for (j = 0; j < c; j++)
7132                                 free(l[j]);
7133
7134                         return r;
7135                 }
7136
7137                 if (r == 0)
7138                         break;
7139         }
7140
7141         /* If we managed to parse all words, return them in the passed
7142          * in parameters */
7143         va_start(ap, p);
7144         for (i = 0; i < n; i++) {
7145                 char **v;
7146
7147                 v = va_arg(ap, char **);
7148                 assert(v);
7149
7150                 *v = l[i];
7151         }
7152         va_end(ap);
7153
7154         return c;
7155 }
7156
7157 int free_and_strdup(char **p, const char *s) {
7158         char *t;
7159
7160         assert(p);
7161
7162         /* Replaces a string pointer with an strdup()ed new string,
7163          * possibly freeing the old one. */
7164
7165         if (s) {
7166                 t = strdup(s);
7167                 if (!t)
7168                         return -ENOMEM;
7169         } else
7170                 t = NULL;
7171
7172         free(*p);
7173         *p = t;
7174
7175         return 0;
7176 }