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                 (void) safe_atoi(e, &c);
3277
3278         if (c <= 0)
3279                 c = fd_columns(STDOUT_FILENO);
3280
3281         if (c <= 0)
3282                 c = 80;
3283
3284         cached_columns = c;
3285         return c;
3286 }
3287
3288 int fd_lines(int fd) {
3289         struct winsize ws = {};
3290
3291         if (ioctl(fd, TIOCGWINSZ, &ws) < 0)
3292                 return -errno;
3293
3294         if (ws.ws_row <= 0)
3295                 return -EIO;
3296
3297         return ws.ws_row;
3298 }
3299
3300 unsigned lines(void) {
3301         const char *e;
3302         unsigned l;
3303
3304         if (_likely_(cached_lines > 0))
3305                 return cached_lines;
3306
3307         l = 0;
3308         e = getenv("LINES");
3309         if (e) {
3310                 int r;
3311
3312                 r = safe_atou(e, &l);
3313                 if (r < 0) {}
3314                         /* do nothing, we fall back to l = 0 */
3315         }
3316
3317         if (l <= 0)
3318                 l = fd_lines(STDOUT_FILENO);
3319
3320         if (l <= 0)
3321                 l = 24;
3322
3323         cached_lines = l;
3324         return cached_lines;
3325 }
3326
3327 /* intended to be used as a SIGWINCH sighandler */
3328 void columns_lines_cache_reset(int signum) {
3329         cached_columns = 0;
3330         cached_lines = 0;
3331 }
3332
3333 bool on_tty(void) {
3334         static int cached_on_tty = -1;
3335
3336         if (_unlikely_(cached_on_tty < 0))
3337                 cached_on_tty = isatty(STDOUT_FILENO) > 0;
3338
3339         return cached_on_tty;
3340 }
3341
3342 int files_same(const char *filea, const char *fileb) {
3343         struct stat a, b;
3344
3345         if (stat(filea, &a) < 0)
3346                 return -errno;
3347
3348         if (stat(fileb, &b) < 0)
3349                 return -errno;
3350
3351         return a.st_dev == b.st_dev &&
3352                a.st_ino == b.st_ino;
3353 }
3354
3355 int running_in_chroot(void) {
3356         int ret;
3357
3358         ret = files_same("/proc/1/root", "/");
3359         if (ret < 0)
3360                 return ret;
3361
3362         return ret == 0;
3363 }
3364
3365 static char *ascii_ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3366         size_t x;
3367         char *r;
3368
3369         assert(s);
3370         assert(percent <= 100);
3371         assert(new_length >= 3);
3372
3373         if (old_length <= 3 || old_length <= new_length)
3374                 return strndup(s, old_length);
3375
3376         r = new0(char, new_length+1);
3377         if (!r)
3378                 return NULL;
3379
3380         x = (new_length * percent) / 100;
3381
3382         if (x > new_length - 3)
3383                 x = new_length - 3;
3384
3385         memcpy(r, s, x);
3386         r[x] = '.';
3387         r[x+1] = '.';
3388         r[x+2] = '.';
3389         memcpy(r + x + 3,
3390                s + old_length - (new_length - x - 3),
3391                new_length - x - 3);
3392
3393         return r;
3394 }
3395
3396 char *ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3397         size_t x;
3398         char *e;
3399         const char *i, *j;
3400         unsigned k, len, len2;
3401
3402         assert(s);
3403         assert(percent <= 100);
3404         assert(new_length >= 3);
3405
3406         /* if no multibyte characters use ascii_ellipsize_mem for speed */
3407         if (ascii_is_valid(s))
3408                 return ascii_ellipsize_mem(s, old_length, new_length, percent);
3409
3410         if (old_length <= 3 || old_length <= new_length)
3411                 return strndup(s, old_length);
3412
3413         x = (new_length * percent) / 100;
3414
3415         if (x > new_length - 3)
3416                 x = new_length - 3;
3417
3418         k = 0;
3419         for (i = s; k < x && i < s + old_length; i = utf8_next_char(i)) {
3420                 int c;
3421
3422                 c = utf8_encoded_to_unichar(i);
3423                 if (c < 0)
3424                         return NULL;
3425                 k += unichar_iswide(c) ? 2 : 1;
3426         }
3427
3428         if (k > x) /* last character was wide and went over quota */
3429                 x ++;
3430
3431         for (j = s + old_length; k < new_length && j > i; ) {
3432                 int c;
3433
3434                 j = utf8_prev_char(j);
3435                 c = utf8_encoded_to_unichar(j);
3436                 if (c < 0)
3437                         return NULL;
3438                 k += unichar_iswide(c) ? 2 : 1;
3439         }
3440         assert(i <= j);
3441
3442         /* we don't actually need to ellipsize */
3443         if (i == j)
3444                 return memdup(s, old_length + 1);
3445
3446         /* make space for ellipsis */
3447         j = utf8_next_char(j);
3448
3449         len = i - s;
3450         len2 = s + old_length - j;
3451         e = new(char, len + 3 + len2 + 1);
3452         if (!e)
3453                 return NULL;
3454
3455         /*
3456         printf("old_length=%zu new_length=%zu x=%zu len=%u len2=%u k=%u\n",
3457                old_length, new_length, x, len, len2, k);
3458         */
3459
3460         memcpy(e, s, len);
3461         e[len]   = 0xe2; /* tri-dot ellipsis: … */
3462         e[len + 1] = 0x80;
3463         e[len + 2] = 0xa6;
3464
3465         memcpy(e + len + 3, j, len2 + 1);
3466
3467         return e;
3468 }
3469
3470 char *ellipsize(const char *s, size_t length, unsigned percent) {
3471         return ellipsize_mem(s, strlen(s), length, percent);
3472 }
3473
3474 int touch_file(const char *path, bool parents, usec_t stamp, uid_t uid, gid_t gid, mode_t mode) {
3475         _cleanup_close_ int fd;
3476         int r;
3477
3478         assert(path);
3479
3480         if (parents)
3481                 mkdir_parents(path, 0755);
3482
3483         fd = open(path, O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY, mode > 0 ? mode : 0644);
3484         if (fd < 0)
3485                 return -errno;
3486
3487         if (mode > 0) {
3488                 r = fchmod(fd, mode);
3489                 if (r < 0)
3490                         return -errno;
3491         }
3492
3493         if (uid != (uid_t) -1 || gid != (gid_t) -1) {
3494                 r = fchown(fd, uid, gid);
3495                 if (r < 0)
3496                         return -errno;
3497         }
3498
3499         if (stamp != USEC_INFINITY) {
3500                 struct timespec ts[2];
3501
3502                 timespec_store(&ts[0], stamp);
3503                 ts[1] = ts[0];
3504                 r = futimens(fd, ts);
3505         } else
3506                 r = futimens(fd, NULL);
3507         if (r < 0)
3508                 return -errno;
3509
3510         return 0;
3511 }
3512
3513 int touch(const char *path) {
3514         return touch_file(path, false, USEC_INFINITY, (uid_t) -1, (gid_t) -1, 0);
3515 }
3516
3517 char *unquote(const char *s, const char* quotes) {
3518         size_t l;
3519         assert(s);
3520
3521         /* This is rather stupid, simply removes the heading and
3522          * trailing quotes if there is one. Doesn't care about
3523          * escaping or anything. We should make this smarter one
3524          * day...*/
3525
3526         l = strlen(s);
3527         if (l < 2)
3528                 return strdup(s);
3529
3530         if (strchr(quotes, s[0]) && s[l-1] == s[0])
3531                 return strndup(s+1, l-2);
3532
3533         return strdup(s);
3534 }
3535
3536 char *normalize_env_assignment(const char *s) {
3537         _cleanup_free_ char *name = NULL, *value = NULL, *p = NULL;
3538         char *eq, *r;
3539
3540         eq = strchr(s, '=');
3541         if (!eq) {
3542                 char *t;
3543
3544                 r = strdup(s);
3545                 if (!r)
3546                         return NULL;
3547
3548                 t = strstrip(r);
3549                 if (t == r)
3550                         return r;
3551
3552                 memmove(r, t, strlen(t) + 1);
3553                 return r;
3554         }
3555
3556         name = strndup(s, eq - s);
3557         if (!name)
3558                 return NULL;
3559
3560         p = strdup(eq + 1);
3561         if (!p)
3562                 return NULL;
3563
3564         value = unquote(strstrip(p), QUOTES);
3565         if (!value)
3566                 return NULL;
3567
3568         if (asprintf(&r, "%s=%s", strstrip(name), value) < 0)
3569                 r = NULL;
3570
3571         return r;
3572 }
3573
3574 int wait_for_terminate(pid_t pid, siginfo_t *status) {
3575         siginfo_t dummy;
3576
3577         assert(pid >= 1);
3578
3579         if (!status)
3580                 status = &dummy;
3581
3582         for (;;) {
3583                 zero(*status);
3584
3585                 if (waitid(P_PID, pid, status, WEXITED) < 0) {
3586
3587                         if (errno == EINTR)
3588                                 continue;
3589
3590                         return -errno;
3591                 }
3592
3593                 return 0;
3594         }
3595 }
3596
3597 /*
3598  * Return values:
3599  * < 0 : wait_for_terminate() failed to get the state of the
3600  *       process, the process was terminated by a signal, or
3601  *       failed for an unknown reason.
3602  * >=0 : The process terminated normally, and its exit code is
3603  *       returned.
3604  *
3605  * That is, success is indicated by a return value of zero, and an
3606  * error is indicated by a non-zero value.
3607  */
3608 int wait_for_terminate_and_warn(const char *name, pid_t pid) {
3609         int r;
3610         siginfo_t status;
3611
3612         assert(name);
3613         assert(pid > 1);
3614
3615         r = wait_for_terminate(pid, &status);
3616         if (r < 0) {
3617                 log_warning("Failed to wait for %s: %s", name, strerror(-r));
3618                 return r;
3619         }
3620
3621         if (status.si_code == CLD_EXITED) {
3622                 if (status.si_status != 0) {
3623                         log_warning("%s failed with error code %i.", name, status.si_status);
3624                         return status.si_status;
3625                 }
3626
3627                 log_debug("%s succeeded.", name);
3628                 return 0;
3629
3630         } else if (status.si_code == CLD_KILLED ||
3631                    status.si_code == CLD_DUMPED) {
3632
3633                 log_warning("%s terminated by signal %s.", name, signal_to_string(status.si_status));
3634                 return -EPROTO;
3635         }
3636
3637         log_warning("%s failed due to unknown reason.", name);
3638         return -EPROTO;
3639 }
3640
3641 noreturn void freeze(void) {
3642
3643         /* Make sure nobody waits for us on a socket anymore */
3644         close_all_fds(NULL, 0);
3645
3646         sync();
3647
3648         for (;;)
3649                 pause();
3650 }
3651
3652 bool null_or_empty(struct stat *st) {
3653         assert(st);
3654
3655         if (S_ISREG(st->st_mode) && st->st_size <= 0)
3656                 return true;
3657
3658         if (S_ISCHR(st->st_mode) || S_ISBLK(st->st_mode))
3659                 return true;
3660
3661         return false;
3662 }
3663
3664 int null_or_empty_path(const char *fn) {
3665         struct stat st;
3666
3667         assert(fn);
3668
3669         if (stat(fn, &st) < 0)
3670                 return -errno;
3671
3672         return null_or_empty(&st);
3673 }
3674
3675 int null_or_empty_fd(int fd) {
3676         struct stat st;
3677
3678         assert(fd >= 0);
3679
3680         if (fstat(fd, &st) < 0)
3681                 return -errno;
3682
3683         return null_or_empty(&st);
3684 }
3685
3686 DIR *xopendirat(int fd, const char *name, int flags) {
3687         int nfd;
3688         DIR *d;
3689
3690         assert(!(flags & O_CREAT));
3691
3692         nfd = openat(fd, name, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|flags, 0);
3693         if (nfd < 0)
3694                 return NULL;
3695
3696         d = fdopendir(nfd);
3697         if (!d) {
3698                 safe_close(nfd);
3699                 return NULL;
3700         }
3701
3702         return d;
3703 }
3704
3705 int signal_from_string_try_harder(const char *s) {
3706         int signo;
3707         assert(s);
3708
3709         signo = signal_from_string(s);
3710         if (signo <= 0)
3711                 if (startswith(s, "SIG"))
3712                         return signal_from_string(s+3);
3713
3714         return signo;
3715 }
3716
3717 static char *tag_to_udev_node(const char *tagvalue, const char *by) {
3718         _cleanup_free_ char *t = NULL, *u = NULL;
3719         size_t enc_len;
3720
3721         u = unquote(tagvalue, "\"\'");
3722         if (!u)
3723                 return NULL;
3724
3725         enc_len = strlen(u) * 4 + 1;
3726         t = new(char, enc_len);
3727         if (!t)
3728                 return NULL;
3729
3730         if (encode_devnode_name(u, t, enc_len) < 0)
3731                 return NULL;
3732
3733         return strjoin("/dev/disk/by-", by, "/", t, NULL);
3734 }
3735
3736 char *fstab_node_to_udev_node(const char *p) {
3737         assert(p);
3738
3739         if (startswith(p, "LABEL="))
3740                 return tag_to_udev_node(p+6, "label");
3741
3742         if (startswith(p, "UUID="))
3743                 return tag_to_udev_node(p+5, "uuid");
3744
3745         if (startswith(p, "PARTUUID="))
3746                 return tag_to_udev_node(p+9, "partuuid");
3747
3748         if (startswith(p, "PARTLABEL="))
3749                 return tag_to_udev_node(p+10, "partlabel");
3750
3751         return strdup(p);
3752 }
3753
3754 bool tty_is_vc(const char *tty) {
3755         assert(tty);
3756
3757         return vtnr_from_tty(tty) >= 0;
3758 }
3759
3760 bool tty_is_console(const char *tty) {
3761         assert(tty);
3762
3763         if (startswith(tty, "/dev/"))
3764                 tty += 5;
3765
3766         return streq(tty, "console");
3767 }
3768
3769 int vtnr_from_tty(const char *tty) {
3770         int i, r;
3771
3772         assert(tty);
3773
3774         if (startswith(tty, "/dev/"))
3775                 tty += 5;
3776
3777         if (!startswith(tty, "tty") )
3778                 return -EINVAL;
3779
3780         if (tty[3] < '0' || tty[3] > '9')
3781                 return -EINVAL;
3782
3783         r = safe_atoi(tty+3, &i);
3784         if (r < 0)
3785                 return r;
3786
3787         if (i < 0 || i > 63)
3788                 return -EINVAL;
3789
3790         return i;
3791 }
3792
3793 char *resolve_dev_console(char **active) {
3794         char *tty;
3795
3796         /* Resolve where /dev/console is pointing to, if /sys is actually ours
3797          * (i.e. not read-only-mounted which is a sign for container setups) */
3798
3799         if (path_is_read_only_fs("/sys") > 0)
3800                 return NULL;
3801
3802         if (read_one_line_file("/sys/class/tty/console/active", active) < 0)
3803                 return NULL;
3804
3805         /* If multiple log outputs are configured the last one is what
3806          * /dev/console points to */
3807         tty = strrchr(*active, ' ');
3808         if (tty)
3809                 tty++;
3810         else
3811                 tty = *active;
3812
3813         if (streq(tty, "tty0")) {
3814                 char *tmp;
3815
3816                 /* Get the active VC (e.g. tty1) */
3817                 if (read_one_line_file("/sys/class/tty/tty0/active", &tmp) >= 0) {
3818                         free(*active);
3819                         tty = *active = tmp;
3820                 }
3821         }
3822
3823         return tty;
3824 }
3825
3826 bool tty_is_vc_resolve(const char *tty) {
3827         _cleanup_free_ char *active = NULL;
3828
3829         assert(tty);
3830
3831         if (startswith(tty, "/dev/"))
3832                 tty += 5;
3833
3834         if (streq(tty, "console")) {
3835                 tty = resolve_dev_console(&active);
3836                 if (!tty)
3837                         return false;
3838         }
3839
3840         return tty_is_vc(tty);
3841 }
3842
3843 const char *default_term_for_tty(const char *tty) {
3844         assert(tty);
3845
3846         return tty_is_vc_resolve(tty) ? "TERM=linux" : "TERM=vt102";
3847 }
3848
3849 bool dirent_is_file(const struct dirent *de) {
3850         assert(de);
3851
3852         if (ignore_file(de->d_name))
3853                 return false;
3854
3855         if (de->d_type != DT_REG &&
3856             de->d_type != DT_LNK &&
3857             de->d_type != DT_UNKNOWN)
3858                 return false;
3859
3860         return true;
3861 }
3862
3863 bool dirent_is_file_with_suffix(const struct dirent *de, const char *suffix) {
3864         assert(de);
3865
3866         if (de->d_type != DT_REG &&
3867             de->d_type != DT_LNK &&
3868             de->d_type != DT_UNKNOWN)
3869                 return false;
3870
3871         if (ignore_file_allow_backup(de->d_name))
3872                 return false;
3873
3874         return endswith(de->d_name, suffix);
3875 }
3876
3877 void execute_directory(const char *directory, DIR *d, usec_t timeout, char *argv[]) {
3878         pid_t executor_pid;
3879         int r;
3880
3881         assert(directory);
3882
3883         /* Executes all binaries in a directory in parallel and waits
3884          * for them to finish. Optionally a timeout is applied. */
3885
3886         executor_pid = fork();
3887         if (executor_pid < 0) {
3888                 log_error("Failed to fork: %m");
3889                 return;
3890
3891         } else if (executor_pid == 0) {
3892                 _cleanup_hashmap_free_free_ Hashmap *pids = NULL;
3893                 _cleanup_closedir_ DIR *_d = NULL;
3894                 struct dirent *de;
3895
3896                 /* We fork this all off from a child process so that
3897                  * we can somewhat cleanly make use of SIGALRM to set
3898                  * a time limit */
3899
3900                 reset_all_signal_handlers();
3901                 reset_signal_mask();
3902
3903                 assert_se(prctl(PR_SET_PDEATHSIG, SIGTERM) == 0);
3904
3905                 if (!d) {
3906                         d = _d = opendir(directory);
3907                         if (!d) {
3908                                 if (errno == ENOENT)
3909                                         _exit(EXIT_SUCCESS);
3910
3911                                 log_error("Failed to enumerate directory %s: %m", directory);
3912                                 _exit(EXIT_FAILURE);
3913                         }
3914                 }
3915
3916                 pids = hashmap_new(NULL);
3917                 if (!pids) {
3918                         log_oom();
3919                         _exit(EXIT_FAILURE);
3920                 }
3921
3922                 FOREACH_DIRENT(de, d, break) {
3923                         _cleanup_free_ char *path = NULL;
3924                         pid_t pid;
3925
3926                         if (!dirent_is_file(de))
3927                                 continue;
3928
3929                         path = strjoin(directory, "/", de->d_name, NULL);
3930                         if (!path) {
3931                                 log_oom();
3932                                 _exit(EXIT_FAILURE);
3933                         }
3934
3935                         pid = fork();
3936                         if (pid < 0) {
3937                                 log_error("Failed to fork: %m");
3938                                 continue;
3939                         } else if (pid == 0) {
3940                                 char *_argv[2];
3941
3942                                 assert_se(prctl(PR_SET_PDEATHSIG, SIGTERM) == 0);
3943
3944                                 if (!argv) {
3945                                         _argv[0] = path;
3946                                         _argv[1] = NULL;
3947                                         argv = _argv;
3948                                 } else
3949                                         argv[0] = path;
3950
3951                                 execv(path, argv);
3952                                 log_error("Failed to execute %s: %m", path);
3953                                 _exit(EXIT_FAILURE);
3954                         }
3955
3956                         log_debug("Spawned %s as " PID_FMT ".", path, pid);
3957
3958                         r = hashmap_put(pids, UINT_TO_PTR(pid), path);
3959                         if (r < 0) {
3960                                 log_oom();
3961                                 _exit(EXIT_FAILURE);
3962                         }
3963
3964                         path = NULL;
3965                 }
3966
3967                 /* Abort execution of this process after the
3968                  * timout. We simply rely on SIGALRM as default action
3969                  * terminating the process, and turn on alarm(). */
3970
3971                 if (timeout != USEC_INFINITY)
3972                         alarm((timeout + USEC_PER_SEC - 1) / USEC_PER_SEC);
3973
3974                 while (!hashmap_isempty(pids)) {
3975                         _cleanup_free_ char *path = NULL;
3976                         pid_t pid;
3977
3978                         pid = PTR_TO_UINT(hashmap_first_key(pids));
3979                         assert(pid > 0);
3980
3981                         path = hashmap_remove(pids, UINT_TO_PTR(pid));
3982                         assert(path);
3983
3984                         wait_for_terminate_and_warn(path, pid);
3985                 }
3986
3987                 _exit(EXIT_SUCCESS);
3988         }
3989
3990         wait_for_terminate_and_warn(directory, executor_pid);
3991 }
3992
3993 int kill_and_sigcont(pid_t pid, int sig) {
3994         int r;
3995
3996         r = kill(pid, sig) < 0 ? -errno : 0;
3997
3998         if (r >= 0)
3999                 kill(pid, SIGCONT);
4000
4001         return r;
4002 }
4003
4004 bool nulstr_contains(const char*nulstr, const char *needle) {
4005         const char *i;
4006
4007         if (!nulstr)
4008                 return false;
4009
4010         NULSTR_FOREACH(i, nulstr)
4011                 if (streq(i, needle))
4012                         return true;
4013
4014         return false;
4015 }
4016
4017 bool plymouth_running(void) {
4018         return access("/run/plymouth/pid", F_OK) >= 0;
4019 }
4020
4021 char* strshorten(char *s, size_t l) {
4022         assert(s);
4023
4024         if (l < strlen(s))
4025                 s[l] = 0;
4026
4027         return s;
4028 }
4029
4030 static bool hostname_valid_char(char c) {
4031         return
4032                 (c >= 'a' && c <= 'z') ||
4033                 (c >= 'A' && c <= 'Z') ||
4034                 (c >= '0' && c <= '9') ||
4035                 c == '-' ||
4036                 c == '_' ||
4037                 c == '.';
4038 }
4039
4040 bool hostname_is_valid(const char *s) {
4041         const char *p;
4042         bool dot;
4043
4044         if (isempty(s))
4045                 return false;
4046
4047         for (p = s, dot = true; *p; p++) {
4048                 if (*p == '.') {
4049                         if (dot)
4050                                 return false;
4051
4052                         dot = true;
4053                 } else {
4054                         if (!hostname_valid_char(*p))
4055                                 return false;
4056
4057                         dot = false;
4058                 }
4059         }
4060
4061         if (dot)
4062                 return false;
4063
4064         if (p-s > HOST_NAME_MAX)
4065                 return false;
4066
4067         return true;
4068 }
4069
4070 char* hostname_cleanup(char *s, bool lowercase) {
4071         char *p, *d;
4072         bool dot;
4073
4074         for (p = s, d = s, dot = true; *p; p++) {
4075                 if (*p == '.') {
4076                         if (dot)
4077                                 continue;
4078
4079                         *(d++) = '.';
4080                         dot = true;
4081                 } else if (hostname_valid_char(*p)) {
4082                         *(d++) = lowercase ? tolower(*p) : *p;
4083                         dot = false;
4084                 }
4085
4086         }
4087
4088         if (dot && d > s)
4089                 d[-1] = 0;
4090         else
4091                 *d = 0;
4092
4093         strshorten(s, HOST_NAME_MAX);
4094
4095         return s;
4096 }
4097
4098 bool machine_name_is_valid(const char *s) {
4099
4100         if (!hostname_is_valid(s))
4101                 return false;
4102
4103         /* Machine names should be useful hostnames, but also be
4104          * useful in unit names, hence we enforce a stricter length
4105          * limitation. */
4106
4107         if (strlen(s) > 64)
4108                 return false;
4109
4110         return true;
4111 }
4112
4113 int pipe_eof(int fd) {
4114         struct pollfd pollfd = {
4115                 .fd = fd,
4116                 .events = POLLIN|POLLHUP,
4117         };
4118
4119         int r;
4120
4121         r = poll(&pollfd, 1, 0);
4122         if (r < 0)
4123                 return -errno;
4124
4125         if (r == 0)
4126                 return 0;
4127
4128         return pollfd.revents & POLLHUP;
4129 }
4130
4131 int fd_wait_for_event(int fd, int event, usec_t t) {
4132
4133         struct pollfd pollfd = {
4134                 .fd = fd,
4135                 .events = event,
4136         };
4137
4138         struct timespec ts;
4139         int r;
4140
4141         r = ppoll(&pollfd, 1, t == USEC_INFINITY ? NULL : timespec_store(&ts, t), NULL);
4142         if (r < 0)
4143                 return -errno;
4144
4145         if (r == 0)
4146                 return 0;
4147
4148         return pollfd.revents;
4149 }
4150
4151 int fopen_temporary(const char *path, FILE **_f, char **_temp_path) {
4152         FILE *f;
4153         char *t;
4154         int fd;
4155
4156         assert(path);
4157         assert(_f);
4158         assert(_temp_path);
4159
4160         t = tempfn_xxxxxx(path);
4161         if (!t)
4162                 return -ENOMEM;
4163
4164         fd = mkostemp_safe(t, O_WRONLY|O_CLOEXEC);
4165         if (fd < 0) {
4166                 free(t);
4167                 return -errno;
4168         }
4169
4170         f = fdopen(fd, "we");
4171         if (!f) {
4172                 unlink(t);
4173                 free(t);
4174                 return -errno;
4175         }
4176
4177         *_f = f;
4178         *_temp_path = t;
4179
4180         return 0;
4181 }
4182
4183 int terminal_vhangup_fd(int fd) {
4184         assert(fd >= 0);
4185
4186         if (ioctl(fd, TIOCVHANGUP) < 0)
4187                 return -errno;
4188
4189         return 0;
4190 }
4191
4192 int terminal_vhangup(const char *name) {
4193         _cleanup_close_ int fd;
4194
4195         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4196         if (fd < 0)
4197                 return fd;
4198
4199         return terminal_vhangup_fd(fd);
4200 }
4201
4202 int vt_disallocate(const char *name) {
4203         int fd, r;
4204         unsigned u;
4205
4206         /* Deallocate the VT if possible. If not possible
4207          * (i.e. because it is the active one), at least clear it
4208          * entirely (including the scrollback buffer) */
4209
4210         if (!startswith(name, "/dev/"))
4211                 return -EINVAL;
4212
4213         if (!tty_is_vc(name)) {
4214                 /* So this is not a VT. I guess we cannot deallocate
4215                  * it then. But let's at least clear the screen */
4216
4217                 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4218                 if (fd < 0)
4219                         return fd;
4220
4221                 loop_write(fd,
4222                            "\033[r"    /* clear scrolling region */
4223                            "\033[H"    /* move home */
4224                            "\033[2J",  /* clear screen */
4225                            10, false);
4226                 safe_close(fd);
4227
4228                 return 0;
4229         }
4230
4231         if (!startswith(name, "/dev/tty"))
4232                 return -EINVAL;
4233
4234         r = safe_atou(name+8, &u);
4235         if (r < 0)
4236                 return r;
4237
4238         if (u <= 0)
4239                 return -EINVAL;
4240
4241         /* Try to deallocate */
4242         fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
4243         if (fd < 0)
4244                 return fd;
4245
4246         r = ioctl(fd, VT_DISALLOCATE, u);
4247         safe_close(fd);
4248
4249         if (r >= 0)
4250                 return 0;
4251
4252         if (errno != EBUSY)
4253                 return -errno;
4254
4255         /* Couldn't deallocate, so let's clear it fully with
4256          * scrollback */
4257         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4258         if (fd < 0)
4259                 return fd;
4260
4261         loop_write(fd,
4262                    "\033[r"   /* clear scrolling region */
4263                    "\033[H"   /* move home */
4264                    "\033[3J", /* clear screen including scrollback, requires Linux 2.6.40 */
4265                    10, false);
4266         safe_close(fd);
4267
4268         return 0;
4269 }
4270
4271 int symlink_atomic(const char *from, const char *to) {
4272         _cleanup_free_ char *t = NULL;
4273
4274         assert(from);
4275         assert(to);
4276
4277         t = tempfn_random(to);
4278         if (!t)
4279                 return -ENOMEM;
4280
4281         if (symlink(from, t) < 0)
4282                 return -errno;
4283
4284         if (rename(t, to) < 0) {
4285                 unlink_noerrno(t);
4286                 return -errno;
4287         }
4288
4289         return 0;
4290 }
4291
4292 int mknod_atomic(const char *path, mode_t mode, dev_t dev) {
4293         _cleanup_free_ char *t = NULL;
4294
4295         assert(path);
4296
4297         t = tempfn_random(path);
4298         if (!t)
4299                 return -ENOMEM;
4300
4301         if (mknod(t, mode, dev) < 0)
4302                 return -errno;
4303
4304         if (rename(t, path) < 0) {
4305                 unlink_noerrno(t);
4306                 return -errno;
4307         }
4308
4309         return 0;
4310 }
4311
4312 int mkfifo_atomic(const char *path, mode_t mode) {
4313         _cleanup_free_ char *t = NULL;
4314
4315         assert(path);
4316
4317         t = tempfn_random(path);
4318         if (!t)
4319                 return -ENOMEM;
4320
4321         if (mkfifo(t, mode) < 0)
4322                 return -errno;
4323
4324         if (rename(t, path) < 0) {
4325                 unlink_noerrno(t);
4326                 return -errno;
4327         }
4328
4329         return 0;
4330 }
4331
4332 bool display_is_local(const char *display) {
4333         assert(display);
4334
4335         return
4336                 display[0] == ':' &&
4337                 display[1] >= '0' &&
4338                 display[1] <= '9';
4339 }
4340
4341 int socket_from_display(const char *display, char **path) {
4342         size_t k;
4343         char *f, *c;
4344
4345         assert(display);
4346         assert(path);
4347
4348         if (!display_is_local(display))
4349                 return -EINVAL;
4350
4351         k = strspn(display+1, "0123456789");
4352
4353         f = new(char, strlen("/tmp/.X11-unix/X") + k + 1);
4354         if (!f)
4355                 return -ENOMEM;
4356
4357         c = stpcpy(f, "/tmp/.X11-unix/X");
4358         memcpy(c, display+1, k);
4359         c[k] = 0;
4360
4361         *path = f;
4362
4363         return 0;
4364 }
4365
4366 int get_user_creds(
4367                 const char **username,
4368                 uid_t *uid, gid_t *gid,
4369                 const char **home,
4370                 const char **shell) {
4371
4372         struct passwd *p;
4373         uid_t u;
4374
4375         assert(username);
4376         assert(*username);
4377
4378         /* We enforce some special rules for uid=0: in order to avoid
4379          * NSS lookups for root we hardcode its data. */
4380
4381         if (streq(*username, "root") || streq(*username, "0")) {
4382                 *username = "root";
4383
4384                 if (uid)
4385                         *uid = 0;
4386
4387                 if (gid)
4388                         *gid = 0;
4389
4390                 if (home)
4391                         *home = "/root";
4392
4393                 if (shell)
4394                         *shell = "/bin/sh";
4395
4396                 return 0;
4397         }
4398
4399         if (parse_uid(*username, &u) >= 0) {
4400                 errno = 0;
4401                 p = getpwuid(u);
4402
4403                 /* If there are multiple users with the same id, make
4404                  * sure to leave $USER to the configured value instead
4405                  * of the first occurrence in the database. However if
4406                  * the uid was configured by a numeric uid, then let's
4407                  * pick the real username from /etc/passwd. */
4408                 if (p)
4409                         *username = p->pw_name;
4410         } else {
4411                 errno = 0;
4412                 p = getpwnam(*username);
4413         }
4414
4415         if (!p)
4416                 return errno > 0 ? -errno : -ESRCH;
4417
4418         if (uid)
4419                 *uid = p->pw_uid;
4420
4421         if (gid)
4422                 *gid = p->pw_gid;
4423
4424         if (home)
4425                 *home = p->pw_dir;
4426
4427         if (shell)
4428                 *shell = p->pw_shell;
4429
4430         return 0;
4431 }
4432
4433 char* uid_to_name(uid_t uid) {
4434         struct passwd *p;
4435         char *r;
4436
4437         if (uid == 0)
4438                 return strdup("root");
4439
4440         p = getpwuid(uid);
4441         if (p)
4442                 return strdup(p->pw_name);
4443
4444         if (asprintf(&r, UID_FMT, uid) < 0)
4445                 return NULL;
4446
4447         return r;
4448 }
4449
4450 char* gid_to_name(gid_t gid) {
4451         struct group *p;
4452         char *r;
4453
4454         if (gid == 0)
4455                 return strdup("root");
4456
4457         p = getgrgid(gid);
4458         if (p)
4459                 return strdup(p->gr_name);
4460
4461         if (asprintf(&r, GID_FMT, gid) < 0)
4462                 return NULL;
4463
4464         return r;
4465 }
4466
4467 int get_group_creds(const char **groupname, gid_t *gid) {
4468         struct group *g;
4469         gid_t id;
4470
4471         assert(groupname);
4472
4473         /* We enforce some special rules for gid=0: in order to avoid
4474          * NSS lookups for root we hardcode its data. */
4475
4476         if (streq(*groupname, "root") || streq(*groupname, "0")) {
4477                 *groupname = "root";
4478
4479                 if (gid)
4480                         *gid = 0;
4481
4482                 return 0;
4483         }
4484
4485         if (parse_gid(*groupname, &id) >= 0) {
4486                 errno = 0;
4487                 g = getgrgid(id);
4488
4489                 if (g)
4490                         *groupname = g->gr_name;
4491         } else {
4492                 errno = 0;
4493                 g = getgrnam(*groupname);
4494         }
4495
4496         if (!g)
4497                 return errno > 0 ? -errno : -ESRCH;
4498
4499         if (gid)
4500                 *gid = g->gr_gid;
4501
4502         return 0;
4503 }
4504
4505 int in_gid(gid_t gid) {
4506         gid_t *gids;
4507         int ngroups_max, r, i;
4508
4509         if (getgid() == gid)
4510                 return 1;
4511
4512         if (getegid() == gid)
4513                 return 1;
4514
4515         ngroups_max = sysconf(_SC_NGROUPS_MAX);
4516         assert(ngroups_max > 0);
4517
4518         gids = alloca(sizeof(gid_t) * ngroups_max);
4519
4520         r = getgroups(ngroups_max, gids);
4521         if (r < 0)
4522                 return -errno;
4523
4524         for (i = 0; i < r; i++)
4525                 if (gids[i] == gid)
4526                         return 1;
4527
4528         return 0;
4529 }
4530
4531 int in_group(const char *name) {
4532         int r;
4533         gid_t gid;
4534
4535         r = get_group_creds(&name, &gid);
4536         if (r < 0)
4537                 return r;
4538
4539         return in_gid(gid);
4540 }
4541
4542 int glob_exists(const char *path) {
4543         _cleanup_globfree_ glob_t g = {};
4544         int k;
4545
4546         assert(path);
4547
4548         errno = 0;
4549         k = glob(path, GLOB_NOSORT|GLOB_BRACE, NULL, &g);
4550
4551         if (k == GLOB_NOMATCH)
4552                 return 0;
4553         else if (k == GLOB_NOSPACE)
4554                 return -ENOMEM;
4555         else if (k == 0)
4556                 return !strv_isempty(g.gl_pathv);
4557         else
4558                 return errno ? -errno : -EIO;
4559 }
4560
4561 int glob_extend(char ***strv, const char *path) {
4562         _cleanup_globfree_ glob_t g = {};
4563         int k;
4564         char **p;
4565
4566         errno = 0;
4567         k = glob(path, GLOB_NOSORT|GLOB_BRACE, NULL, &g);
4568
4569         if (k == GLOB_NOMATCH)
4570                 return -ENOENT;
4571         else if (k == GLOB_NOSPACE)
4572                 return -ENOMEM;
4573         else if (k != 0 || strv_isempty(g.gl_pathv))
4574                 return errno ? -errno : -EIO;
4575
4576         STRV_FOREACH(p, g.gl_pathv) {
4577                 k = strv_extend(strv, *p);
4578                 if (k < 0)
4579                         break;
4580         }
4581
4582         return k;
4583 }
4584
4585 int dirent_ensure_type(DIR *d, struct dirent *de) {
4586         struct stat st;
4587
4588         assert(d);
4589         assert(de);
4590
4591         if (de->d_type != DT_UNKNOWN)
4592                 return 0;
4593
4594         if (fstatat(dirfd(d), de->d_name, &st, AT_SYMLINK_NOFOLLOW) < 0)
4595                 return -errno;
4596
4597         de->d_type =
4598                 S_ISREG(st.st_mode)  ? DT_REG  :
4599                 S_ISDIR(st.st_mode)  ? DT_DIR  :
4600                 S_ISLNK(st.st_mode)  ? DT_LNK  :
4601                 S_ISFIFO(st.st_mode) ? DT_FIFO :
4602                 S_ISSOCK(st.st_mode) ? DT_SOCK :
4603                 S_ISCHR(st.st_mode)  ? DT_CHR  :
4604                 S_ISBLK(st.st_mode)  ? DT_BLK  :
4605                                        DT_UNKNOWN;
4606
4607         return 0;
4608 }
4609
4610 int get_files_in_directory(const char *path, char ***list) {
4611         _cleanup_closedir_ DIR *d = NULL;
4612         size_t bufsize = 0, n = 0;
4613         _cleanup_strv_free_ char **l = NULL;
4614
4615         assert(path);
4616
4617         /* Returns all files in a directory in *list, and the number
4618          * of files as return value. If list is NULL returns only the
4619          * number. */
4620
4621         d = opendir(path);
4622         if (!d)
4623                 return -errno;
4624
4625         for (;;) {
4626                 struct dirent *de;
4627
4628                 errno = 0;
4629                 de = readdir(d);
4630                 if (!de && errno != 0)
4631                         return -errno;
4632                 if (!de)
4633                         break;
4634
4635                 dirent_ensure_type(d, de);
4636
4637                 if (!dirent_is_file(de))
4638                         continue;
4639
4640                 if (list) {
4641                         /* one extra slot is needed for the terminating NULL */
4642                         if (!GREEDY_REALLOC(l, bufsize, n + 2))
4643                                 return -ENOMEM;
4644
4645                         l[n] = strdup(de->d_name);
4646                         if (!l[n])
4647                                 return -ENOMEM;
4648
4649                         l[++n] = NULL;
4650                 } else
4651                         n++;
4652         }
4653
4654         if (list) {
4655                 *list = l;
4656                 l = NULL; /* avoid freeing */
4657         }
4658
4659         return n;
4660 }
4661
4662 char *strjoin(const char *x, ...) {
4663         va_list ap;
4664         size_t l;
4665         char *r, *p;
4666
4667         va_start(ap, x);
4668
4669         if (x) {
4670                 l = strlen(x);
4671
4672                 for (;;) {
4673                         const char *t;
4674                         size_t n;
4675
4676                         t = va_arg(ap, const char *);
4677                         if (!t)
4678                                 break;
4679
4680                         n = strlen(t);
4681                         if (n > ((size_t) -1) - l) {
4682                                 va_end(ap);
4683                                 return NULL;
4684                         }
4685
4686                         l += n;
4687                 }
4688         } else
4689                 l = 0;
4690
4691         va_end(ap);
4692
4693         r = new(char, l+1);
4694         if (!r)
4695                 return NULL;
4696
4697         if (x) {
4698                 p = stpcpy(r, x);
4699
4700                 va_start(ap, x);
4701
4702                 for (;;) {
4703                         const char *t;
4704
4705                         t = va_arg(ap, const char *);
4706                         if (!t)
4707                                 break;
4708
4709                         p = stpcpy(p, t);
4710                 }
4711
4712                 va_end(ap);
4713         } else
4714                 r[0] = 0;
4715
4716         return r;
4717 }
4718
4719 bool is_main_thread(void) {
4720         static thread_local int cached = 0;
4721
4722         if (_unlikely_(cached == 0))
4723                 cached = getpid() == gettid() ? 1 : -1;
4724
4725         return cached > 0;
4726 }
4727
4728 int block_get_whole_disk(dev_t d, dev_t *ret) {
4729         char *p, *s;
4730         int r;
4731         unsigned n, m;
4732
4733         assert(ret);
4734
4735         /* If it has a queue this is good enough for us */
4736         if (asprintf(&p, "/sys/dev/block/%u:%u/queue", major(d), minor(d)) < 0)
4737                 return -ENOMEM;
4738
4739         r = access(p, F_OK);
4740         free(p);
4741
4742         if (r >= 0) {
4743                 *ret = d;
4744                 return 0;
4745         }
4746
4747         /* If it is a partition find the originating device */
4748         if (asprintf(&p, "/sys/dev/block/%u:%u/partition", major(d), minor(d)) < 0)
4749                 return -ENOMEM;
4750
4751         r = access(p, F_OK);
4752         free(p);
4753
4754         if (r < 0)
4755                 return -ENOENT;
4756
4757         /* Get parent dev_t */
4758         if (asprintf(&p, "/sys/dev/block/%u:%u/../dev", major(d), minor(d)) < 0)
4759                 return -ENOMEM;
4760
4761         r = read_one_line_file(p, &s);
4762         free(p);
4763
4764         if (r < 0)
4765                 return r;
4766
4767         r = sscanf(s, "%u:%u", &m, &n);
4768         free(s);
4769
4770         if (r != 2)
4771                 return -EINVAL;
4772
4773         /* Only return this if it is really good enough for us. */
4774         if (asprintf(&p, "/sys/dev/block/%u:%u/queue", m, n) < 0)
4775                 return -ENOMEM;
4776
4777         r = access(p, F_OK);
4778         free(p);
4779
4780         if (r >= 0) {
4781                 *ret = makedev(m, n);
4782                 return 0;
4783         }
4784
4785         return -ENOENT;
4786 }
4787
4788 int file_is_priv_sticky(const char *p) {
4789         struct stat st;
4790
4791         assert(p);
4792
4793         if (lstat(p, &st) < 0)
4794                 return -errno;
4795
4796         return
4797                 (st.st_uid == 0 || st.st_uid == getuid()) &&
4798                 (st.st_mode & S_ISVTX);
4799 }
4800
4801 static const char *const ioprio_class_table[] = {
4802         [IOPRIO_CLASS_NONE] = "none",
4803         [IOPRIO_CLASS_RT] = "realtime",
4804         [IOPRIO_CLASS_BE] = "best-effort",
4805         [IOPRIO_CLASS_IDLE] = "idle"
4806 };
4807
4808 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(ioprio_class, int, INT_MAX);
4809
4810 static const char *const sigchld_code_table[] = {
4811         [CLD_EXITED] = "exited",
4812         [CLD_KILLED] = "killed",
4813         [CLD_DUMPED] = "dumped",
4814         [CLD_TRAPPED] = "trapped",
4815         [CLD_STOPPED] = "stopped",
4816         [CLD_CONTINUED] = "continued",
4817 };
4818
4819 DEFINE_STRING_TABLE_LOOKUP(sigchld_code, int);
4820
4821 static const char *const log_facility_unshifted_table[LOG_NFACILITIES] = {
4822         [LOG_FAC(LOG_KERN)] = "kern",
4823         [LOG_FAC(LOG_USER)] = "user",
4824         [LOG_FAC(LOG_MAIL)] = "mail",
4825         [LOG_FAC(LOG_DAEMON)] = "daemon",
4826         [LOG_FAC(LOG_AUTH)] = "auth",
4827         [LOG_FAC(LOG_SYSLOG)] = "syslog",
4828         [LOG_FAC(LOG_LPR)] = "lpr",
4829         [LOG_FAC(LOG_NEWS)] = "news",
4830         [LOG_FAC(LOG_UUCP)] = "uucp",
4831         [LOG_FAC(LOG_CRON)] = "cron",
4832         [LOG_FAC(LOG_AUTHPRIV)] = "authpriv",
4833         [LOG_FAC(LOG_FTP)] = "ftp",
4834         [LOG_FAC(LOG_LOCAL0)] = "local0",
4835         [LOG_FAC(LOG_LOCAL1)] = "local1",
4836         [LOG_FAC(LOG_LOCAL2)] = "local2",
4837         [LOG_FAC(LOG_LOCAL3)] = "local3",
4838         [LOG_FAC(LOG_LOCAL4)] = "local4",
4839         [LOG_FAC(LOG_LOCAL5)] = "local5",
4840         [LOG_FAC(LOG_LOCAL6)] = "local6",
4841         [LOG_FAC(LOG_LOCAL7)] = "local7"
4842 };
4843
4844 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(log_facility_unshifted, int, LOG_FAC(~0));
4845
4846 static const char *const log_level_table[] = {
4847         [LOG_EMERG] = "emerg",
4848         [LOG_ALERT] = "alert",
4849         [LOG_CRIT] = "crit",
4850         [LOG_ERR] = "err",
4851         [LOG_WARNING] = "warning",
4852         [LOG_NOTICE] = "notice",
4853         [LOG_INFO] = "info",
4854         [LOG_DEBUG] = "debug"
4855 };
4856
4857 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(log_level, int, LOG_DEBUG);
4858
4859 static const char* const sched_policy_table[] = {
4860         [SCHED_OTHER] = "other",
4861         [SCHED_BATCH] = "batch",
4862         [SCHED_IDLE] = "idle",
4863         [SCHED_FIFO] = "fifo",
4864         [SCHED_RR] = "rr"
4865 };
4866
4867 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(sched_policy, int, INT_MAX);
4868
4869 static const char* const rlimit_table[_RLIMIT_MAX] = {
4870         [RLIMIT_CPU] = "LimitCPU",
4871         [RLIMIT_FSIZE] = "LimitFSIZE",
4872         [RLIMIT_DATA] = "LimitDATA",
4873         [RLIMIT_STACK] = "LimitSTACK",
4874         [RLIMIT_CORE] = "LimitCORE",
4875         [RLIMIT_RSS] = "LimitRSS",
4876         [RLIMIT_NOFILE] = "LimitNOFILE",
4877         [RLIMIT_AS] = "LimitAS",
4878         [RLIMIT_NPROC] = "LimitNPROC",
4879         [RLIMIT_MEMLOCK] = "LimitMEMLOCK",
4880         [RLIMIT_LOCKS] = "LimitLOCKS",
4881         [RLIMIT_SIGPENDING] = "LimitSIGPENDING",
4882         [RLIMIT_MSGQUEUE] = "LimitMSGQUEUE",
4883         [RLIMIT_NICE] = "LimitNICE",
4884         [RLIMIT_RTPRIO] = "LimitRTPRIO",
4885         [RLIMIT_RTTIME] = "LimitRTTIME"
4886 };
4887
4888 DEFINE_STRING_TABLE_LOOKUP(rlimit, int);
4889
4890 static const char* const ip_tos_table[] = {
4891         [IPTOS_LOWDELAY] = "low-delay",
4892         [IPTOS_THROUGHPUT] = "throughput",
4893         [IPTOS_RELIABILITY] = "reliability",
4894         [IPTOS_LOWCOST] = "low-cost",
4895 };
4896
4897 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(ip_tos, int, 0xff);
4898
4899 static const char *const __signal_table[] = {
4900         [SIGHUP] = "HUP",
4901         [SIGINT] = "INT",
4902         [SIGQUIT] = "QUIT",
4903         [SIGILL] = "ILL",
4904         [SIGTRAP] = "TRAP",
4905         [SIGABRT] = "ABRT",
4906         [SIGBUS] = "BUS",
4907         [SIGFPE] = "FPE",
4908         [SIGKILL] = "KILL",
4909         [SIGUSR1] = "USR1",
4910         [SIGSEGV] = "SEGV",
4911         [SIGUSR2] = "USR2",
4912         [SIGPIPE] = "PIPE",
4913         [SIGALRM] = "ALRM",
4914         [SIGTERM] = "TERM",
4915 #ifdef SIGSTKFLT
4916         [SIGSTKFLT] = "STKFLT",  /* Linux on SPARC doesn't know SIGSTKFLT */
4917 #endif
4918         [SIGCHLD] = "CHLD",
4919         [SIGCONT] = "CONT",
4920         [SIGSTOP] = "STOP",
4921         [SIGTSTP] = "TSTP",
4922         [SIGTTIN] = "TTIN",
4923         [SIGTTOU] = "TTOU",
4924         [SIGURG] = "URG",
4925         [SIGXCPU] = "XCPU",
4926         [SIGXFSZ] = "XFSZ",
4927         [SIGVTALRM] = "VTALRM",
4928         [SIGPROF] = "PROF",
4929         [SIGWINCH] = "WINCH",
4930         [SIGIO] = "IO",
4931         [SIGPWR] = "PWR",
4932         [SIGSYS] = "SYS"
4933 };
4934
4935 DEFINE_PRIVATE_STRING_TABLE_LOOKUP(__signal, int);
4936
4937 const char *signal_to_string(int signo) {
4938         static thread_local char buf[sizeof("RTMIN+")-1 + DECIMAL_STR_MAX(int) + 1];
4939         const char *name;
4940
4941         name = __signal_to_string(signo);
4942         if (name)
4943                 return name;
4944
4945         if (signo >= SIGRTMIN && signo <= SIGRTMAX)
4946                 snprintf(buf, sizeof(buf), "RTMIN+%d", signo - SIGRTMIN);
4947         else
4948                 snprintf(buf, sizeof(buf), "%d", signo);
4949
4950         return buf;
4951 }
4952
4953 int signal_from_string(const char *s) {
4954         int signo;
4955         int offset = 0;
4956         unsigned u;
4957
4958         signo = __signal_from_string(s);
4959         if (signo > 0)
4960                 return signo;
4961
4962         if (startswith(s, "RTMIN+")) {
4963                 s += 6;
4964                 offset = SIGRTMIN;
4965         }
4966         if (safe_atou(s, &u) >= 0) {
4967                 signo = (int) u + offset;
4968                 if (signo > 0 && signo < _NSIG)
4969                         return signo;
4970         }
4971         return -EINVAL;
4972 }
4973
4974 bool kexec_loaded(void) {
4975        bool loaded = false;
4976        char *s;
4977
4978        if (read_one_line_file("/sys/kernel/kexec_loaded", &s) >= 0) {
4979                if (s[0] == '1')
4980                        loaded = true;
4981                free(s);
4982        }
4983        return loaded;
4984 }
4985
4986 int prot_from_flags(int flags) {
4987
4988         switch (flags & O_ACCMODE) {
4989
4990         case O_RDONLY:
4991                 return PROT_READ;
4992
4993         case O_WRONLY:
4994                 return PROT_WRITE;
4995
4996         case O_RDWR:
4997                 return PROT_READ|PROT_WRITE;
4998
4999         default:
5000                 return -EINVAL;
5001         }
5002 }
5003
5004 char *format_bytes(char *buf, size_t l, off_t t) {
5005         unsigned i;
5006
5007         static const struct {
5008                 const char *suffix;
5009                 off_t factor;
5010         } table[] = {
5011                 { "E", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
5012                 { "P", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
5013                 { "T", 1024ULL*1024ULL*1024ULL*1024ULL },
5014                 { "G", 1024ULL*1024ULL*1024ULL },
5015                 { "M", 1024ULL*1024ULL },
5016                 { "K", 1024ULL },
5017         };
5018
5019         for (i = 0; i < ELEMENTSOF(table); i++) {
5020
5021                 if (t >= table[i].factor) {
5022                         snprintf(buf, l,
5023                                  "%llu.%llu%s",
5024                                  (unsigned long long) (t / table[i].factor),
5025                                  (unsigned long long) (((t*10ULL) / table[i].factor) % 10ULL),
5026                                  table[i].suffix);
5027
5028                         goto finish;
5029                 }
5030         }
5031
5032         snprintf(buf, l, "%lluB", (unsigned long long) t);
5033
5034 finish:
5035         buf[l-1] = 0;
5036         return buf;
5037
5038 }
5039
5040 void* memdup(const void *p, size_t l) {
5041         void *r;
5042
5043         assert(p);
5044
5045         r = malloc(l);
5046         if (!r)
5047                 return NULL;
5048
5049         memcpy(r, p, l);
5050         return r;
5051 }
5052
5053 int fd_inc_sndbuf(int fd, size_t n) {
5054         int r, value;
5055         socklen_t l = sizeof(value);
5056
5057         r = getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, &l);
5058         if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
5059                 return 0;
5060
5061         /* If we have the privileges we will ignore the kernel limit. */
5062
5063         value = (int) n;
5064         if (setsockopt(fd, SOL_SOCKET, SO_SNDBUFFORCE, &value, sizeof(value)) < 0)
5065                 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value)) < 0)
5066                         return -errno;
5067
5068         return 1;
5069 }
5070
5071 int fd_inc_rcvbuf(int fd, size_t n) {
5072         int r, value;
5073         socklen_t l = sizeof(value);
5074
5075         r = getsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, &l);
5076         if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
5077                 return 0;
5078
5079         /* If we have the privileges we will ignore the kernel limit. */
5080
5081         value = (int) n;
5082         if (setsockopt(fd, SOL_SOCKET, SO_RCVBUFFORCE, &value, sizeof(value)) < 0)
5083                 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value)) < 0)
5084                         return -errno;
5085         return 1;
5086 }
5087
5088 int fork_agent(pid_t *pid, const int except[], unsigned n_except, const char *path, ...) {
5089         bool stdout_is_tty, stderr_is_tty;
5090         pid_t parent_pid, agent_pid;
5091         sigset_t ss, saved_ss;
5092         unsigned n, i;
5093         va_list ap;
5094         char **l;
5095
5096         assert(pid);
5097         assert(path);
5098
5099         /* Spawns a temporary TTY agent, making sure it goes away when
5100          * we go away */
5101
5102         parent_pid = getpid();
5103
5104         /* First we temporarily block all signals, so that the new
5105          * child has them blocked initially. This way, we can be sure
5106          * that SIGTERMs are not lost we might send to the agent. */
5107         assert_se(sigfillset(&ss) >= 0);
5108         assert_se(sigprocmask(SIG_SETMASK, &ss, &saved_ss) >= 0);
5109
5110         agent_pid = fork();
5111         if (agent_pid < 0) {
5112                 assert_se(sigprocmask(SIG_SETMASK, &saved_ss, NULL) >= 0);
5113                 return -errno;
5114         }
5115
5116         if (agent_pid != 0) {
5117                 assert_se(sigprocmask(SIG_SETMASK, &saved_ss, NULL) >= 0);
5118                 *pid = agent_pid;
5119                 return 0;
5120         }
5121
5122         /* In the child:
5123          *
5124          * Make sure the agent goes away when the parent dies */
5125         if (prctl(PR_SET_PDEATHSIG, SIGTERM) < 0)
5126                 _exit(EXIT_FAILURE);
5127
5128         /* Make sure we actually can kill the agent, if we need to, in
5129          * case somebody invoked us from a shell script that trapped
5130          * SIGTERM or so... */
5131         reset_all_signal_handlers();
5132         reset_signal_mask();
5133
5134         /* Check whether our parent died before we were able
5135          * to set the death signal and unblock the signals */
5136         if (getppid() != parent_pid)
5137                 _exit(EXIT_SUCCESS);
5138
5139         /* Don't leak fds to the agent */
5140         close_all_fds(except, n_except);
5141
5142         stdout_is_tty = isatty(STDOUT_FILENO);
5143         stderr_is_tty = isatty(STDERR_FILENO);
5144
5145         if (!stdout_is_tty || !stderr_is_tty) {
5146                 int fd;
5147
5148                 /* Detach from stdout/stderr. and reopen
5149                  * /dev/tty for them. This is important to
5150                  * ensure that when systemctl is started via
5151                  * popen() or a similar call that expects to
5152                  * read EOF we actually do generate EOF and
5153                  * not delay this indefinitely by because we
5154                  * keep an unused copy of stdin around. */
5155                 fd = open("/dev/tty", O_WRONLY);
5156                 if (fd < 0) {
5157                         log_error("Failed to open /dev/tty: %m");
5158                         _exit(EXIT_FAILURE);
5159                 }
5160
5161                 if (!stdout_is_tty)
5162                         dup2(fd, STDOUT_FILENO);
5163
5164                 if (!stderr_is_tty)
5165                         dup2(fd, STDERR_FILENO);
5166
5167                 if (fd > 2)
5168                         close(fd);
5169         }
5170
5171         /* Count arguments */
5172         va_start(ap, path);
5173         for (n = 0; va_arg(ap, char*); n++)
5174                 ;
5175         va_end(ap);
5176
5177         /* Allocate strv */
5178         l = alloca(sizeof(char *) * (n + 1));
5179
5180         /* Fill in arguments */
5181         va_start(ap, path);
5182         for (i = 0; i <= n; i++)
5183                 l[i] = va_arg(ap, char*);
5184         va_end(ap);
5185
5186         execv(path, l);
5187         _exit(EXIT_FAILURE);
5188 }
5189
5190 int setrlimit_closest(int resource, const struct rlimit *rlim) {
5191         struct rlimit highest, fixed;
5192
5193         assert(rlim);
5194
5195         if (setrlimit(resource, rlim) >= 0)
5196                 return 0;
5197
5198         if (errno != EPERM)
5199                 return -errno;
5200
5201         /* So we failed to set the desired setrlimit, then let's try
5202          * to get as close as we can */
5203         assert_se(getrlimit(resource, &highest) == 0);
5204
5205         fixed.rlim_cur = MIN(rlim->rlim_cur, highest.rlim_max);
5206         fixed.rlim_max = MIN(rlim->rlim_max, highest.rlim_max);
5207
5208         if (setrlimit(resource, &fixed) < 0)
5209                 return -errno;
5210
5211         return 0;
5212 }
5213
5214 int getenv_for_pid(pid_t pid, const char *field, char **_value) {
5215         _cleanup_fclose_ FILE *f = NULL;
5216         char *value = NULL;
5217         int r;
5218         bool done = false;
5219         size_t l;
5220         const char *path;
5221
5222         assert(pid >= 0);
5223         assert(field);
5224         assert(_value);
5225
5226         path = procfs_file_alloca(pid, "environ");
5227
5228         f = fopen(path, "re");
5229         if (!f)
5230                 return -errno;
5231
5232         l = strlen(field);
5233         r = 0;
5234
5235         do {
5236                 char line[LINE_MAX];
5237                 unsigned i;
5238
5239                 for (i = 0; i < sizeof(line)-1; i++) {
5240                         int c;
5241
5242                         c = getc(f);
5243                         if (_unlikely_(c == EOF)) {
5244                                 done = true;
5245                                 break;
5246                         } else if (c == 0)
5247                                 break;
5248
5249                         line[i] = c;
5250                 }
5251                 line[i] = 0;
5252
5253                 if (memcmp(line, field, l) == 0 && line[l] == '=') {
5254                         value = strdup(line + l + 1);
5255                         if (!value)
5256                                 return -ENOMEM;
5257
5258                         r = 1;
5259                         break;
5260                 }
5261
5262         } while (!done);
5263
5264         *_value = value;
5265         return r;
5266 }
5267
5268 bool is_valid_documentation_url(const char *url) {
5269         assert(url);
5270
5271         if (startswith(url, "http://") && url[7])
5272                 return true;
5273
5274         if (startswith(url, "https://") && url[8])
5275                 return true;
5276
5277         if (startswith(url, "file:") && url[5])
5278                 return true;
5279
5280         if (startswith(url, "info:") && url[5])
5281                 return true;
5282
5283         if (startswith(url, "man:") && url[4])
5284                 return true;
5285
5286         return false;
5287 }
5288
5289 bool in_initrd(void) {
5290         static int saved = -1;
5291         struct statfs s;
5292
5293         if (saved >= 0)
5294                 return saved;
5295
5296         /* We make two checks here:
5297          *
5298          * 1. the flag file /etc/initrd-release must exist
5299          * 2. the root file system must be a memory file system
5300          *
5301          * The second check is extra paranoia, since misdetecting an
5302          * initrd can have bad bad consequences due the initrd
5303          * emptying when transititioning to the main systemd.
5304          */
5305
5306         saved = access("/etc/initrd-release", F_OK) >= 0 &&
5307                 statfs("/", &s) >= 0 &&
5308                 is_temporary_fs(&s);
5309
5310         return saved;
5311 }
5312
5313 void warn_melody(void) {
5314         _cleanup_close_ int fd = -1;
5315
5316         fd = open("/dev/console", O_WRONLY|O_CLOEXEC|O_NOCTTY);
5317         if (fd < 0)
5318                 return;
5319
5320         /* Yeah, this is synchronous. Kinda sucks. But well... */
5321
5322         ioctl(fd, KIOCSOUND, (int)(1193180/440));
5323         usleep(125*USEC_PER_MSEC);
5324
5325         ioctl(fd, KIOCSOUND, (int)(1193180/220));
5326         usleep(125*USEC_PER_MSEC);
5327
5328         ioctl(fd, KIOCSOUND, (int)(1193180/220));
5329         usleep(125*USEC_PER_MSEC);
5330
5331         ioctl(fd, KIOCSOUND, 0);
5332 }
5333
5334 int make_console_stdio(void) {
5335         int fd, r;
5336
5337         /* Make /dev/console the controlling terminal and stdin/stdout/stderr */
5338
5339         fd = acquire_terminal("/dev/console", false, true, true, USEC_INFINITY);
5340         if (fd < 0) {
5341                 log_error("Failed to acquire terminal: %s", strerror(-fd));
5342                 return fd;
5343         }
5344
5345         r = make_stdio(fd);
5346         if (r < 0) {
5347                 log_error("Failed to duplicate terminal fd: %s", strerror(-r));
5348                 return r;
5349         }
5350
5351         return 0;
5352 }
5353
5354 int get_home_dir(char **_h) {
5355         struct passwd *p;
5356         const char *e;
5357         char *h;
5358         uid_t u;
5359
5360         assert(_h);
5361
5362         /* Take the user specified one */
5363         e = secure_getenv("HOME");
5364         if (e && path_is_absolute(e)) {
5365                 h = strdup(e);
5366                 if (!h)
5367                         return -ENOMEM;
5368
5369                 *_h = h;
5370                 return 0;
5371         }
5372
5373         /* Hardcode home directory for root to avoid NSS */
5374         u = getuid();
5375         if (u == 0) {
5376                 h = strdup("/root");
5377                 if (!h)
5378                         return -ENOMEM;
5379
5380                 *_h = h;
5381                 return 0;
5382         }
5383
5384         /* Check the database... */
5385         errno = 0;
5386         p = getpwuid(u);
5387         if (!p)
5388                 return errno > 0 ? -errno : -ESRCH;
5389
5390         if (!path_is_absolute(p->pw_dir))
5391                 return -EINVAL;
5392
5393         h = strdup(p->pw_dir);
5394         if (!h)
5395                 return -ENOMEM;
5396
5397         *_h = h;
5398         return 0;
5399 }
5400
5401 int get_shell(char **_s) {
5402         struct passwd *p;
5403         const char *e;
5404         char *s;
5405         uid_t u;
5406
5407         assert(_s);
5408
5409         /* Take the user specified one */
5410         e = getenv("SHELL");
5411         if (e) {
5412                 s = strdup(e);
5413                 if (!s)
5414                         return -ENOMEM;
5415
5416                 *_s = s;
5417                 return 0;
5418         }
5419
5420         /* Hardcode home directory for root to avoid NSS */
5421         u = getuid();
5422         if (u == 0) {
5423                 s = strdup("/bin/sh");
5424                 if (!s)
5425                         return -ENOMEM;
5426
5427                 *_s = s;
5428                 return 0;
5429         }
5430
5431         /* Check the database... */
5432         errno = 0;
5433         p = getpwuid(u);
5434         if (!p)
5435                 return errno > 0 ? -errno : -ESRCH;
5436
5437         if (!path_is_absolute(p->pw_shell))
5438                 return -EINVAL;
5439
5440         s = strdup(p->pw_shell);
5441         if (!s)
5442                 return -ENOMEM;
5443
5444         *_s = s;
5445         return 0;
5446 }
5447
5448 bool filename_is_safe(const char *p) {
5449
5450         if (isempty(p))
5451                 return false;
5452
5453         if (strchr(p, '/'))
5454                 return false;
5455
5456         if (streq(p, "."))
5457                 return false;
5458
5459         if (streq(p, ".."))
5460                 return false;
5461
5462         if (strlen(p) > FILENAME_MAX)
5463                 return false;
5464
5465         return true;
5466 }
5467
5468 bool string_is_safe(const char *p) {
5469         const char *t;
5470
5471         if (!p)
5472                 return false;
5473
5474         for (t = p; *t; t++) {
5475                 if (*t > 0 && *t < ' ')
5476                         return false;
5477
5478                 if (strchr("\\\"\'\0x7f", *t))
5479                         return false;
5480         }
5481
5482         return true;
5483 }
5484
5485 /**
5486  * Check if a string contains control characters. If 'ok' is non-NULL
5487  * it may be a string containing additional CCs to be considered OK.
5488  */
5489 bool string_has_cc(const char *p, const char *ok) {
5490         const char *t;
5491
5492         assert(p);
5493
5494         for (t = p; *t; t++) {
5495                 if (ok && strchr(ok, *t))
5496                         continue;
5497
5498                 if (*t > 0 && *t < ' ')
5499                         return true;
5500
5501                 if (*t == 127)
5502                         return true;
5503         }
5504
5505         return false;
5506 }
5507
5508 bool path_is_safe(const char *p) {
5509
5510         if (isempty(p))
5511                 return false;
5512
5513         if (streq(p, "..") || startswith(p, "../") || endswith(p, "/..") || strstr(p, "/../"))
5514                 return false;
5515
5516         if (strlen(p) > PATH_MAX)
5517                 return false;
5518
5519         /* The following two checks are not really dangerous, but hey, they still are confusing */
5520         if (streq(p, ".") || startswith(p, "./") || endswith(p, "/.") || strstr(p, "/./"))
5521                 return false;
5522
5523         if (strstr(p, "//"))
5524                 return false;
5525
5526         return true;
5527 }
5528
5529 /* hey glibc, APIs with callbacks without a user pointer are so useless */
5530 void *xbsearch_r(const void *key, const void *base, size_t nmemb, size_t size,
5531                  int (*compar) (const void *, const void *, void *), void *arg) {
5532         size_t l, u, idx;
5533         const void *p;
5534         int comparison;
5535
5536         l = 0;
5537         u = nmemb;
5538         while (l < u) {
5539                 idx = (l + u) / 2;
5540                 p = (void *)(((const char *) base) + (idx * size));
5541                 comparison = compar(key, p, arg);
5542                 if (comparison < 0)
5543                         u = idx;
5544                 else if (comparison > 0)
5545                         l = idx + 1;
5546                 else
5547                         return (void *)p;
5548         }
5549         return NULL;
5550 }
5551
5552 bool is_locale_utf8(void) {
5553         const char *set;
5554         static int cached_answer = -1;
5555
5556         if (cached_answer >= 0)
5557                 goto out;
5558
5559         if (!setlocale(LC_ALL, "")) {
5560                 cached_answer = true;
5561                 goto out;
5562         }
5563
5564         set = nl_langinfo(CODESET);
5565         if (!set) {
5566                 cached_answer = true;
5567                 goto out;
5568         }
5569
5570         if (streq(set, "UTF-8")) {
5571                 cached_answer = true;
5572                 goto out;
5573         }
5574
5575         /* For LC_CTYPE=="C" return true, because CTYPE is effectly
5576          * unset and everything can do to UTF-8 nowadays. */
5577         set = setlocale(LC_CTYPE, NULL);
5578         if (!set) {
5579                 cached_answer = true;
5580                 goto out;
5581         }
5582
5583         /* Check result, but ignore the result if C was set
5584          * explicitly. */
5585         cached_answer =
5586                 streq(set, "C") &&
5587                 !getenv("LC_ALL") &&
5588                 !getenv("LC_CTYPE") &&
5589                 !getenv("LANG");
5590
5591 out:
5592         return (bool) cached_answer;
5593 }
5594
5595 const char *draw_special_char(DrawSpecialChar ch) {
5596         static const char *draw_table[2][_DRAW_SPECIAL_CHAR_MAX] = {
5597
5598                 /* UTF-8 */ {
5599                         [DRAW_TREE_VERTICAL]      = "\342\224\202 ",            /* │  */
5600                         [DRAW_TREE_BRANCH]        = "\342\224\234\342\224\200", /* ├─ */
5601                         [DRAW_TREE_RIGHT]         = "\342\224\224\342\224\200", /* └─ */
5602                         [DRAW_TREE_SPACE]         = "  ",                       /*    */
5603                         [DRAW_TRIANGULAR_BULLET]  = "\342\200\243",             /* ‣ */
5604                         [DRAW_BLACK_CIRCLE]       = "\342\227\217",             /* ● */
5605                         [DRAW_ARROW]              = "\342\206\222",             /* → */
5606                         [DRAW_DASH]               = "\342\200\223",             /* – */
5607                 },
5608
5609                 /* ASCII fallback */ {
5610                         [DRAW_TREE_VERTICAL]      = "| ",
5611                         [DRAW_TREE_BRANCH]        = "|-",
5612                         [DRAW_TREE_RIGHT]         = "`-",
5613                         [DRAW_TREE_SPACE]         = "  ",
5614                         [DRAW_TRIANGULAR_BULLET]  = ">",
5615                         [DRAW_BLACK_CIRCLE]       = "*",
5616                         [DRAW_ARROW]              = "->",
5617                         [DRAW_DASH]               = "-",
5618                 }
5619         };
5620
5621         return draw_table[!is_locale_utf8()][ch];
5622 }
5623
5624 char *strreplace(const char *text, const char *old_string, const char *new_string) {
5625         const char *f;
5626         char *t, *r;
5627         size_t l, old_len, new_len;
5628
5629         assert(text);
5630         assert(old_string);
5631         assert(new_string);
5632
5633         old_len = strlen(old_string);
5634         new_len = strlen(new_string);
5635
5636         l = strlen(text);
5637         r = new(char, l+1);
5638         if (!r)
5639                 return NULL;
5640
5641         f = text;
5642         t = r;
5643         while (*f) {
5644                 char *a;
5645                 size_t d, nl;
5646
5647                 if (!startswith(f, old_string)) {
5648                         *(t++) = *(f++);
5649                         continue;
5650                 }
5651
5652                 d = t - r;
5653                 nl = l - old_len + new_len;
5654                 a = realloc(r, nl + 1);
5655                 if (!a)
5656                         goto oom;
5657
5658                 l = nl;
5659                 r = a;
5660                 t = r + d;
5661
5662                 t = stpcpy(t, new_string);
5663                 f += old_len;
5664         }
5665
5666         *t = 0;
5667         return r;
5668
5669 oom:
5670         free(r);
5671         return NULL;
5672 }
5673
5674 char *strip_tab_ansi(char **ibuf, size_t *_isz) {
5675         const char *i, *begin = NULL;
5676         enum {
5677                 STATE_OTHER,
5678                 STATE_ESCAPE,
5679                 STATE_BRACKET
5680         } state = STATE_OTHER;
5681         char *obuf = NULL;
5682         size_t osz = 0, isz;
5683         FILE *f;
5684
5685         assert(ibuf);
5686         assert(*ibuf);
5687
5688         /* Strips ANSI color and replaces TABs by 8 spaces */
5689
5690         isz = _isz ? *_isz : strlen(*ibuf);
5691
5692         f = open_memstream(&obuf, &osz);
5693         if (!f)
5694                 return NULL;
5695
5696         for (i = *ibuf; i < *ibuf + isz + 1; i++) {
5697
5698                 switch (state) {
5699
5700                 case STATE_OTHER:
5701                         if (i >= *ibuf + isz) /* EOT */
5702                                 break;
5703                         else if (*i == '\x1B')
5704                                 state = STATE_ESCAPE;
5705                         else if (*i == '\t')
5706                                 fputs("        ", f);
5707                         else
5708                                 fputc(*i, f);
5709                         break;
5710
5711                 case STATE_ESCAPE:
5712                         if (i >= *ibuf + isz) { /* EOT */
5713                                 fputc('\x1B', f);
5714                                 break;
5715                         } else if (*i == '[') {
5716                                 state = STATE_BRACKET;
5717                                 begin = i + 1;
5718                         } else {
5719                                 fputc('\x1B', f);
5720                                 fputc(*i, f);
5721                                 state = STATE_OTHER;
5722                         }
5723
5724                         break;
5725
5726                 case STATE_BRACKET:
5727
5728                         if (i >= *ibuf + isz || /* EOT */
5729                             (!(*i >= '0' && *i <= '9') && *i != ';' && *i != 'm')) {
5730                                 fputc('\x1B', f);
5731                                 fputc('[', f);
5732                                 state = STATE_OTHER;
5733                                 i = begin-1;
5734                         } else if (*i == 'm')
5735                                 state = STATE_OTHER;
5736                         break;
5737                 }
5738         }
5739
5740         if (ferror(f)) {
5741                 fclose(f);
5742                 free(obuf);
5743                 return NULL;
5744         }
5745
5746         fclose(f);
5747
5748         free(*ibuf);
5749         *ibuf = obuf;
5750
5751         if (_isz)
5752                 *_isz = osz;
5753
5754         return obuf;
5755 }
5756
5757 int on_ac_power(void) {
5758         bool found_offline = false, found_online = false;
5759         _cleanup_closedir_ DIR *d = NULL;
5760
5761         d = opendir("/sys/class/power_supply");
5762         if (!d)
5763                 return -errno;
5764
5765         for (;;) {
5766                 struct dirent *de;
5767                 _cleanup_close_ int fd = -1, device = -1;
5768                 char contents[6];
5769                 ssize_t n;
5770
5771                 errno = 0;
5772                 de = readdir(d);
5773                 if (!de && errno != 0)
5774                         return -errno;
5775
5776                 if (!de)
5777                         break;
5778
5779                 if (ignore_file(de->d_name))
5780                         continue;
5781
5782                 device = openat(dirfd(d), de->d_name, O_DIRECTORY|O_RDONLY|O_CLOEXEC|O_NOCTTY);
5783                 if (device < 0) {
5784                         if (errno == ENOENT || errno == ENOTDIR)
5785                                 continue;
5786
5787                         return -errno;
5788                 }
5789
5790                 fd = openat(device, "type", O_RDONLY|O_CLOEXEC|O_NOCTTY);
5791                 if (fd < 0) {
5792                         if (errno == ENOENT)
5793                                 continue;
5794
5795                         return -errno;
5796                 }
5797
5798                 n = read(fd, contents, sizeof(contents));
5799                 if (n < 0)
5800                         return -errno;
5801
5802                 if (n != 6 || memcmp(contents, "Mains\n", 6))
5803                         continue;
5804
5805                 safe_close(fd);
5806                 fd = openat(device, "online", O_RDONLY|O_CLOEXEC|O_NOCTTY);
5807                 if (fd < 0) {
5808                         if (errno == ENOENT)
5809                                 continue;
5810
5811                         return -errno;
5812                 }
5813
5814                 n = read(fd, contents, sizeof(contents));
5815                 if (n < 0)
5816                         return -errno;
5817
5818                 if (n != 2 || contents[1] != '\n')
5819                         return -EIO;
5820
5821                 if (contents[0] == '1') {
5822                         found_online = true;
5823                         break;
5824                 } else if (contents[0] == '0')
5825                         found_offline = true;
5826                 else
5827                         return -EIO;
5828         }
5829
5830         return found_online || !found_offline;
5831 }
5832
5833 static int search_and_fopen_internal(const char *path, const char *mode, const char *root, char **search, FILE **_f) {
5834         char **i;
5835
5836         assert(path);
5837         assert(mode);
5838         assert(_f);
5839
5840         if (!path_strv_resolve_uniq(search, root))
5841                 return -ENOMEM;
5842
5843         STRV_FOREACH(i, search) {
5844                 _cleanup_free_ char *p = NULL;
5845                 FILE *f;
5846
5847                 if (root)
5848                         p = strjoin(root, *i, "/", path, NULL);
5849                 else
5850                         p = strjoin(*i, "/", path, NULL);
5851                 if (!p)
5852                         return -ENOMEM;
5853
5854                 f = fopen(p, mode);
5855                 if (f) {
5856                         *_f = f;
5857                         return 0;
5858                 }
5859
5860                 if (errno != ENOENT)
5861                         return -errno;
5862         }
5863
5864         return -ENOENT;
5865 }
5866
5867 int search_and_fopen(const char *path, const char *mode, const char *root, const char **search, FILE **_f) {
5868         _cleanup_strv_free_ char **copy = NULL;
5869
5870         assert(path);
5871         assert(mode);
5872         assert(_f);
5873
5874         if (path_is_absolute(path)) {
5875                 FILE *f;
5876
5877                 f = fopen(path, mode);
5878                 if (f) {
5879                         *_f = f;
5880                         return 0;
5881                 }
5882
5883                 return -errno;
5884         }
5885
5886         copy = strv_copy((char**) search);
5887         if (!copy)
5888                 return -ENOMEM;
5889
5890         return search_and_fopen_internal(path, mode, root, copy, _f);
5891 }
5892
5893 int search_and_fopen_nulstr(const char *path, const char *mode, const char *root, const char *search, FILE **_f) {
5894         _cleanup_strv_free_ char **s = NULL;
5895
5896         if (path_is_absolute(path)) {
5897                 FILE *f;
5898
5899                 f = fopen(path, mode);
5900                 if (f) {
5901                         *_f = f;
5902                         return 0;
5903                 }
5904
5905                 return -errno;
5906         }
5907
5908         s = strv_split_nulstr(search);
5909         if (!s)
5910                 return -ENOMEM;
5911
5912         return search_and_fopen_internal(path, mode, root, s, _f);
5913 }
5914
5915 char *strextend(char **x, ...) {
5916         va_list ap;
5917         size_t f, l;
5918         char *r, *p;
5919
5920         assert(x);
5921
5922         l = f = *x ? strlen(*x) : 0;
5923
5924         va_start(ap, x);
5925         for (;;) {
5926                 const char *t;
5927                 size_t n;
5928
5929                 t = va_arg(ap, const char *);
5930                 if (!t)
5931                         break;
5932
5933                 n = strlen(t);
5934                 if (n > ((size_t) -1) - l) {
5935                         va_end(ap);
5936                         return NULL;
5937                 }
5938
5939                 l += n;
5940         }
5941         va_end(ap);
5942
5943         r = realloc(*x, l+1);
5944         if (!r)
5945                 return NULL;
5946
5947         p = r + f;
5948
5949         va_start(ap, x);
5950         for (;;) {
5951                 const char *t;
5952
5953                 t = va_arg(ap, const char *);
5954                 if (!t)
5955                         break;
5956
5957                 p = stpcpy(p, t);
5958         }
5959         va_end(ap);
5960
5961         *p = 0;
5962         *x = r;
5963
5964         return r + l;
5965 }
5966
5967 char *strrep(const char *s, unsigned n) {
5968         size_t l;
5969         char *r, *p;
5970         unsigned i;
5971
5972         assert(s);
5973
5974         l = strlen(s);
5975         p = r = malloc(l * n + 1);
5976         if (!r)
5977                 return NULL;
5978
5979         for (i = 0; i < n; i++)
5980                 p = stpcpy(p, s);
5981
5982         *p = 0;
5983         return r;
5984 }
5985
5986 void* greedy_realloc(void **p, size_t *allocated, size_t need, size_t size) {
5987         size_t a, newalloc;
5988         void *q;
5989
5990         assert(p);
5991         assert(allocated);
5992
5993         if (*allocated >= need)
5994                 return *p;
5995
5996         newalloc = MAX(need * 2, 64u / size);
5997         a = newalloc * size;
5998
5999         /* check for overflows */
6000         if (a < size * need)
6001                 return NULL;
6002
6003         q = realloc(*p, a);
6004         if (!q)
6005                 return NULL;
6006
6007         *p = q;
6008         *allocated = newalloc;
6009         return q;
6010 }
6011
6012 void* greedy_realloc0(void **p, size_t *allocated, size_t need, size_t size) {
6013         size_t prev;
6014         uint8_t *q;
6015
6016         assert(p);
6017         assert(allocated);
6018
6019         prev = *allocated;
6020
6021         q = greedy_realloc(p, allocated, need, size);
6022         if (!q)
6023                 return NULL;
6024
6025         if (*allocated > prev)
6026                 memzero(q + prev * size, (*allocated - prev) * size);
6027
6028         return q;
6029 }
6030
6031 bool id128_is_valid(const char *s) {
6032         size_t i, l;
6033
6034         l = strlen(s);
6035         if (l == 32) {
6036
6037                 /* Simple formatted 128bit hex string */
6038
6039                 for (i = 0; i < l; i++) {
6040                         char c = s[i];
6041
6042                         if (!(c >= '0' && c <= '9') &&
6043                             !(c >= 'a' && c <= 'z') &&
6044                             !(c >= 'A' && c <= 'Z'))
6045                                 return false;
6046                 }
6047
6048         } else if (l == 36) {
6049
6050                 /* Formatted UUID */
6051
6052                 for (i = 0; i < l; i++) {
6053                         char c = s[i];
6054
6055                         if ((i == 8 || i == 13 || i == 18 || i == 23)) {
6056                                 if (c != '-')
6057                                         return false;
6058                         } else {
6059                                 if (!(c >= '0' && c <= '9') &&
6060                                     !(c >= 'a' && c <= 'z') &&
6061                                     !(c >= 'A' && c <= 'Z'))
6062                                         return false;
6063                         }
6064                 }
6065
6066         } else
6067                 return false;
6068
6069         return true;
6070 }
6071
6072 int split_pair(const char *s, const char *sep, char **l, char **r) {
6073         char *x, *a, *b;
6074
6075         assert(s);
6076         assert(sep);
6077         assert(l);
6078         assert(r);
6079
6080         if (isempty(sep))
6081                 return -EINVAL;
6082
6083         x = strstr(s, sep);
6084         if (!x)
6085                 return -EINVAL;
6086
6087         a = strndup(s, x - s);
6088         if (!a)
6089                 return -ENOMEM;
6090
6091         b = strdup(x + strlen(sep));
6092         if (!b) {
6093                 free(a);
6094                 return -ENOMEM;
6095         }
6096
6097         *l = a;
6098         *r = b;
6099
6100         return 0;
6101 }
6102
6103 int shall_restore_state(void) {
6104         _cleanup_free_ char *line = NULL;
6105         const char *word, *state;
6106         size_t l;
6107         int r;
6108
6109         r = proc_cmdline(&line);
6110         if (r < 0)
6111                 return r;
6112         if (r == 0) /* Container ... */
6113                 return 1;
6114
6115         r = 1;
6116
6117         FOREACH_WORD_QUOTED(word, l, line, state) {
6118                 const char *e;
6119                 char n[l+1];
6120                 int k;
6121
6122                 memcpy(n, word, l);
6123                 n[l] = 0;
6124
6125                 e = startswith(n, "systemd.restore_state=");
6126                 if (!e)
6127                         continue;
6128
6129                 k = parse_boolean(e);
6130                 if (k >= 0)
6131                         r = k;
6132         }
6133
6134         return r;
6135 }
6136
6137 int proc_cmdline(char **ret) {
6138         int r;
6139
6140         if (detect_container(NULL) > 0) {
6141                 char *buf = NULL, *p;
6142                 size_t sz = 0;
6143
6144                 r = read_full_file("/proc/1/cmdline", &buf, &sz);
6145                 if (r < 0)
6146                         return r;
6147
6148                 for (p = buf; p + 1 < buf + sz; p++)
6149                         if (*p == 0)
6150                                 *p = ' ';
6151
6152                 *p = 0;
6153                 *ret = buf;
6154                 return 1;
6155         }
6156
6157         r = read_one_line_file("/proc/cmdline", ret);
6158         if (r < 0)
6159                 return r;
6160
6161         return 1;
6162 }
6163
6164 int parse_proc_cmdline(int (*parse_item)(const char *key, const char *value)) {
6165         _cleanup_free_ char *line = NULL;
6166         const char *w, *state;
6167         size_t l;
6168         int r;
6169
6170         assert(parse_item);
6171
6172         r = proc_cmdline(&line);
6173         if (r < 0)
6174                 log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r));
6175         if (r <= 0)
6176                 return 0;
6177
6178         FOREACH_WORD_QUOTED(w, l, line, state) {
6179                 char word[l+1], *value;
6180
6181                 memcpy(word, w, l);
6182                 word[l] = 0;
6183
6184                 /* Filter out arguments that are intended only for the
6185                  * initrd */
6186                 if (!in_initrd() && startswith(word, "rd."))
6187                         continue;
6188
6189                 value = strchr(word, '=');
6190                 if (value)
6191                         *(value++) = 0;
6192
6193                 r = parse_item(word, value);
6194                 if (r < 0)
6195                         return r;
6196         }
6197
6198         return 0;
6199 }
6200
6201 int container_get_leader(const char *machine, pid_t *pid) {
6202         _cleanup_free_ char *s = NULL, *class = NULL;
6203         const char *p;
6204         pid_t leader;
6205         int r;
6206
6207         assert(machine);
6208         assert(pid);
6209
6210         p = strappenda("/run/systemd/machines/", machine);
6211         r = parse_env_file(p, NEWLINE, "LEADER", &s, "CLASS", &class, NULL);
6212         if (r == -ENOENT)
6213                 return -EHOSTDOWN;
6214         if (r < 0)
6215                 return r;
6216         if (!s)
6217                 return -EIO;
6218
6219         if (!streq_ptr(class, "container"))
6220                 return -EIO;
6221
6222         r = parse_pid(s, &leader);
6223         if (r < 0)
6224                 return r;
6225         if (leader <= 1)
6226                 return -EIO;
6227
6228         *pid = leader;
6229         return 0;
6230 }
6231
6232 int namespace_open(pid_t pid, int *pidns_fd, int *mntns_fd, int *netns_fd, int *root_fd) {
6233         _cleanup_close_ int pidnsfd = -1, mntnsfd = -1, netnsfd = -1;
6234         int rfd = -1;
6235
6236         assert(pid >= 0);
6237
6238         if (mntns_fd) {
6239                 const char *mntns;
6240
6241                 mntns = procfs_file_alloca(pid, "ns/mnt");
6242                 mntnsfd = open(mntns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6243                 if (mntnsfd < 0)
6244                         return -errno;
6245         }
6246
6247         if (pidns_fd) {
6248                 const char *pidns;
6249
6250                 pidns = procfs_file_alloca(pid, "ns/pid");
6251                 pidnsfd = open(pidns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6252                 if (pidnsfd < 0)
6253                         return -errno;
6254         }
6255
6256         if (netns_fd) {
6257                 const char *netns;
6258
6259                 netns = procfs_file_alloca(pid, "ns/net");
6260                 netnsfd = open(netns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6261                 if (netnsfd < 0)
6262                         return -errno;
6263         }
6264
6265         if (root_fd) {
6266                 const char *root;
6267
6268                 root = procfs_file_alloca(pid, "root");
6269                 rfd = open(root, O_RDONLY|O_NOCTTY|O_CLOEXEC|O_DIRECTORY);
6270                 if (rfd < 0)
6271                         return -errno;
6272         }
6273
6274         if (pidns_fd)
6275                 *pidns_fd = pidnsfd;
6276
6277         if (mntns_fd)
6278                 *mntns_fd = mntnsfd;
6279
6280         if (netns_fd)
6281                 *netns_fd = netnsfd;
6282
6283         if (root_fd)
6284                 *root_fd = rfd;
6285
6286         pidnsfd = mntnsfd = netnsfd = -1;
6287
6288         return 0;
6289 }
6290
6291 int namespace_enter(int pidns_fd, int mntns_fd, int netns_fd, int root_fd) {
6292
6293         if (pidns_fd >= 0)
6294                 if (setns(pidns_fd, CLONE_NEWPID) < 0)
6295                         return -errno;
6296
6297         if (mntns_fd >= 0)
6298                 if (setns(mntns_fd, CLONE_NEWNS) < 0)
6299                         return -errno;
6300
6301         if (netns_fd >= 0)
6302                 if (setns(netns_fd, CLONE_NEWNET) < 0)
6303                         return -errno;
6304
6305         if (root_fd >= 0) {
6306                 if (fchdir(root_fd) < 0)
6307                         return -errno;
6308
6309                 if (chroot(".") < 0)
6310                         return -errno;
6311         }
6312
6313         if (setresgid(0, 0, 0) < 0)
6314                 return -errno;
6315
6316         if (setgroups(0, NULL) < 0)
6317                 return -errno;
6318
6319         if (setresuid(0, 0, 0) < 0)
6320                 return -errno;
6321
6322         return 0;
6323 }
6324
6325 bool pid_is_unwaited(pid_t pid) {
6326         /* Checks whether a PID is still valid at all, including a zombie */
6327
6328         if (pid <= 0)
6329                 return false;
6330
6331         if (kill(pid, 0) >= 0)
6332                 return true;
6333
6334         return errno != ESRCH;
6335 }
6336
6337 bool pid_is_alive(pid_t pid) {
6338         int r;
6339
6340         /* Checks whether a PID is still valid and not a zombie */
6341
6342         if (pid <= 0)
6343                 return false;
6344
6345         r = get_process_state(pid);
6346         if (r == -ENOENT || r == 'Z')
6347                 return false;
6348
6349         return true;
6350 }
6351
6352 int getpeercred(int fd, struct ucred *ucred) {
6353         socklen_t n = sizeof(struct ucred);
6354         struct ucred u;
6355         int r;
6356
6357         assert(fd >= 0);
6358         assert(ucred);
6359
6360         r = getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &u, &n);
6361         if (r < 0)
6362                 return -errno;
6363
6364         if (n != sizeof(struct ucred))
6365                 return -EIO;
6366
6367         /* Check if the data is actually useful and not suppressed due
6368          * to namespacing issues */
6369         if (u.pid <= 0)
6370                 return -ENODATA;
6371
6372         *ucred = u;
6373         return 0;
6374 }
6375
6376 int getpeersec(int fd, char **ret) {
6377         socklen_t n = 64;
6378         char *s;
6379         int r;
6380
6381         assert(fd >= 0);
6382         assert(ret);
6383
6384         s = new0(char, n);
6385         if (!s)
6386                 return -ENOMEM;
6387
6388         r = getsockopt(fd, SOL_SOCKET, SO_PEERSEC, s, &n);
6389         if (r < 0) {
6390                 free(s);
6391
6392                 if (errno != ERANGE)
6393                         return -errno;
6394
6395                 s = new0(char, n);
6396                 if (!s)
6397                         return -ENOMEM;
6398
6399                 r = getsockopt(fd, SOL_SOCKET, SO_PEERSEC, s, &n);
6400                 if (r < 0) {
6401                         free(s);
6402                         return -errno;
6403                 }
6404         }
6405
6406         if (isempty(s)) {
6407                 free(s);
6408                 return -ENOTSUP;
6409         }
6410
6411         *ret = s;
6412         return 0;
6413 }
6414
6415 /* This is much like like mkostemp() but is subject to umask(). */
6416 int mkostemp_safe(char *pattern, int flags) {
6417         _cleanup_umask_ mode_t u;
6418         int fd;
6419
6420         assert(pattern);
6421
6422         u = umask(077);
6423
6424         fd = mkostemp(pattern, flags);
6425         if (fd < 0)
6426                 return -errno;
6427
6428         return fd;
6429 }
6430
6431 int open_tmpfile(const char *path, int flags) {
6432         char *p;
6433         int fd;
6434
6435         assert(path);
6436
6437 #ifdef O_TMPFILE
6438         /* Try O_TMPFILE first, if it is supported */
6439         fd = open(path, flags|O_TMPFILE, S_IRUSR|S_IWUSR);
6440         if (fd >= 0)
6441                 return fd;
6442 #endif
6443
6444         /* Fall back to unguessable name + unlinking */
6445         p = strappenda(path, "/systemd-tmp-XXXXXX");
6446
6447         fd = mkostemp_safe(p, flags);
6448         if (fd < 0)
6449                 return fd;
6450
6451         unlink(p);
6452         return fd;
6453 }
6454
6455 int fd_warn_permissions(const char *path, int fd) {
6456         struct stat st;
6457
6458         if (fstat(fd, &st) < 0)
6459                 return -errno;
6460
6461         if (st.st_mode & 0111)
6462                 log_warning("Configuration file %s is marked executable. Please remove executable permission bits. Proceeding anyway.", path);
6463
6464         if (st.st_mode & 0002)
6465                 log_warning("Configuration file %s is marked world-writable. Please remove world writability permission bits. Proceeding anyway.", path);
6466
6467         if (getpid() == 1 && (st.st_mode & 0044) != 0044)
6468                 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);
6469
6470         return 0;
6471 }
6472
6473 unsigned long personality_from_string(const char *p) {
6474
6475         /* Parse a personality specifier. We introduce our own
6476          * identifiers that indicate specific ABIs, rather than just
6477          * hints regarding the register size, since we want to keep
6478          * things open for multiple locally supported ABIs for the
6479          * same register size. We try to reuse the ABI identifiers
6480          * used by libseccomp. */
6481
6482 #if defined(__x86_64__)
6483
6484         if (streq(p, "x86"))
6485                 return PER_LINUX32;
6486
6487         if (streq(p, "x86-64"))
6488                 return PER_LINUX;
6489
6490 #elif defined(__i386__)
6491
6492         if (streq(p, "x86"))
6493                 return PER_LINUX;
6494 #endif
6495
6496         /* personality(7) documents that 0xffffffffUL is used for
6497          * querying the current personality, hence let's use that here
6498          * as error indicator. */
6499         return 0xffffffffUL;
6500 }
6501
6502 const char* personality_to_string(unsigned long p) {
6503
6504 #if defined(__x86_64__)
6505
6506         if (p == PER_LINUX32)
6507                 return "x86";
6508
6509         if (p == PER_LINUX)
6510                 return "x86-64";
6511
6512 #elif defined(__i386__)
6513
6514         if (p == PER_LINUX)
6515                 return "x86";
6516 #endif
6517
6518         return NULL;
6519 }
6520
6521 uint64_t physical_memory(void) {
6522         long mem;
6523
6524         /* We return this as uint64_t in case we are running as 32bit
6525          * process on a 64bit kernel with huge amounts of memory */
6526
6527         mem = sysconf(_SC_PHYS_PAGES);
6528         assert(mem > 0);
6529
6530         return (uint64_t) mem * (uint64_t) page_size();
6531 }
6532
6533 char* mount_test_option(const char *haystack, const char *needle) {
6534
6535         struct mntent me = {
6536                 .mnt_opts = (char*) haystack
6537         };
6538
6539         assert(needle);
6540
6541         /* Like glibc's hasmntopt(), but works on a string, not a
6542          * struct mntent */
6543
6544         if (!haystack)
6545                 return NULL;
6546
6547         return hasmntopt(&me, needle);
6548 }
6549
6550 void hexdump(FILE *f, const void *p, size_t s) {
6551         const uint8_t *b = p;
6552         unsigned n = 0;
6553
6554         assert(s == 0 || b);
6555
6556         while (s > 0) {
6557                 size_t i;
6558
6559                 fprintf(f, "%04x  ", n);
6560
6561                 for (i = 0; i < 16; i++) {
6562
6563                         if (i >= s)
6564                                 fputs("   ", f);
6565                         else
6566                                 fprintf(f, "%02x ", b[i]);
6567
6568                         if (i == 7)
6569                                 fputc(' ', f);
6570                 }
6571
6572                 fputc(' ', f);
6573
6574                 for (i = 0; i < 16; i++) {
6575
6576                         if (i >= s)
6577                                 fputc(' ', f);
6578                         else
6579                                 fputc(isprint(b[i]) ? (char) b[i] : '.', f);
6580                 }
6581
6582                 fputc('\n', f);
6583
6584                 if (s < 16)
6585                         break;
6586
6587                 n += 16;
6588                 b += 16;
6589                 s -= 16;
6590         }
6591 }
6592
6593 int update_reboot_param_file(const char *param) {
6594         int r = 0;
6595
6596         if (param) {
6597
6598                 r = write_string_file(REBOOT_PARAM_FILE, param);
6599                 if (r < 0)
6600                         log_error("Failed to write reboot param to "
6601                                   REBOOT_PARAM_FILE": %s", strerror(-r));
6602         } else
6603                 unlink(REBOOT_PARAM_FILE);
6604
6605         return r;
6606 }
6607
6608 int umount_recursive(const char *prefix, int flags) {
6609         bool again;
6610         int n = 0, r;
6611
6612         /* Try to umount everything recursively below a
6613          * directory. Also, take care of stacked mounts, and keep
6614          * unmounting them until they are gone. */
6615
6616         do {
6617                 _cleanup_fclose_ FILE *proc_self_mountinfo = NULL;
6618
6619                 again = false;
6620                 r = 0;
6621
6622                 proc_self_mountinfo = fopen("/proc/self/mountinfo", "re");
6623                 if (!proc_self_mountinfo)
6624                         return -errno;
6625
6626                 for (;;) {
6627                         _cleanup_free_ char *path = NULL, *p = NULL;
6628                         int k;
6629
6630                         k = fscanf(proc_self_mountinfo,
6631                                    "%*s "       /* (1) mount id */
6632                                    "%*s "       /* (2) parent id */
6633                                    "%*s "       /* (3) major:minor */
6634                                    "%*s "       /* (4) root */
6635                                    "%ms "       /* (5) mount point */
6636                                    "%*s"        /* (6) mount options */
6637                                    "%*[^-]"     /* (7) optional fields */
6638                                    "- "         /* (8) separator */
6639                                    "%*s "       /* (9) file system type */
6640                                    "%*s"        /* (10) mount source */
6641                                    "%*s"        /* (11) mount options 2 */
6642                                    "%*[^\n]",   /* some rubbish at the end */
6643                                    &path);
6644                         if (k != 1) {
6645                                 if (k == EOF)
6646                                         break;
6647
6648                                 continue;
6649                         }
6650
6651                         p = cunescape(path);
6652                         if (!p)
6653                                 return -ENOMEM;
6654
6655                         if (!path_startswith(p, prefix))
6656                                 continue;
6657
6658                         if (umount2(p, flags) < 0) {
6659                                 r = -errno;
6660                                 continue;
6661                         }
6662
6663                         again = true;
6664                         n++;
6665
6666                         break;
6667                 }
6668
6669         } while (again);
6670
6671         return r ? r : n;
6672 }
6673
6674 int bind_remount_recursive(const char *prefix, bool ro) {
6675         _cleanup_set_free_free_ Set *done = NULL;
6676         _cleanup_free_ char *cleaned = NULL;
6677         int r;
6678
6679         /* Recursively remount a directory (and all its submounts)
6680          * read-only or read-write. If the directory is already
6681          * mounted, we reuse the mount and simply mark it
6682          * MS_BIND|MS_RDONLY (or remove the MS_RDONLY for read-write
6683          * operation). If it isn't we first make it one. Afterwards we
6684          * apply MS_BIND|MS_RDONLY (or remove MS_RDONLY) to all
6685          * submounts we can access, too. When mounts are stacked on
6686          * the same mount point we only care for each individual
6687          * "top-level" mount on each point, as we cannot
6688          * influence/access the underlying mounts anyway. We do not
6689          * have any effect on future submounts that might get
6690          * propagated, they migt be writable. This includes future
6691          * submounts that have been triggered via autofs. */
6692
6693         cleaned = strdup(prefix);
6694         if (!cleaned)
6695                 return -ENOMEM;
6696
6697         path_kill_slashes(cleaned);
6698
6699         done = set_new(&string_hash_ops);
6700         if (!done)
6701                 return -ENOMEM;
6702
6703         for (;;) {
6704                 _cleanup_fclose_ FILE *proc_self_mountinfo = NULL;
6705                 _cleanup_set_free_free_ Set *todo = NULL;
6706                 bool top_autofs = false;
6707                 char *x;
6708
6709                 todo = set_new(&string_hash_ops);
6710                 if (!todo)
6711                         return -ENOMEM;
6712
6713                 proc_self_mountinfo = fopen("/proc/self/mountinfo", "re");
6714                 if (!proc_self_mountinfo)
6715                         return -errno;
6716
6717                 for (;;) {
6718                         _cleanup_free_ char *path = NULL, *p = NULL, *type = NULL;
6719                         int k;
6720
6721                         k = fscanf(proc_self_mountinfo,
6722                                    "%*s "       /* (1) mount id */
6723                                    "%*s "       /* (2) parent id */
6724                                    "%*s "       /* (3) major:minor */
6725                                    "%*s "       /* (4) root */
6726                                    "%ms "       /* (5) mount point */
6727                                    "%*s"        /* (6) mount options (superblock) */
6728                                    "%*[^-]"     /* (7) optional fields */
6729                                    "- "         /* (8) separator */
6730                                    "%ms "       /* (9) file system type */
6731                                    "%*s"        /* (10) mount source */
6732                                    "%*s"        /* (11) mount options (bind mount) */
6733                                    "%*[^\n]",   /* some rubbish at the end */
6734                                    &path,
6735                                    &type);
6736                         if (k != 2) {
6737                                 if (k == EOF)
6738                                         break;
6739
6740                                 continue;
6741                         }
6742
6743                         p = cunescape(path);
6744                         if (!p)
6745                                 return -ENOMEM;
6746
6747                         /* Let's ignore autofs mounts.  If they aren't
6748                          * triggered yet, we want to avoid triggering
6749                          * them, as we don't make any guarantees for
6750                          * future submounts anyway.  If they are
6751                          * already triggered, then we will find
6752                          * another entry for this. */
6753                         if (streq(type, "autofs")) {
6754                                 top_autofs = top_autofs || path_equal(cleaned, p);
6755                                 continue;
6756                         }
6757
6758                         if (path_startswith(p, cleaned) &&
6759                             !set_contains(done, p)) {
6760
6761                                 r = set_consume(todo, p);
6762                                 p = NULL;
6763
6764                                 if (r == -EEXIST)
6765                                         continue;
6766                                 if (r < 0)
6767                                         return r;
6768                         }
6769                 }
6770
6771                 /* If we have no submounts to process anymore and if
6772                  * the root is either already done, or an autofs, we
6773                  * are done */
6774                 if (set_isempty(todo) &&
6775                     (top_autofs || set_contains(done, cleaned)))
6776                         return 0;
6777
6778                 if (!set_contains(done, cleaned) &&
6779                     !set_contains(todo, cleaned)) {
6780                         /* The prefix directory itself is not yet a
6781                          * mount, make it one. */
6782                         if (mount(cleaned, cleaned, NULL, MS_BIND|MS_REC, NULL) < 0)
6783                                 return -errno;
6784
6785                         if (mount(NULL, prefix, NULL, MS_BIND|MS_REMOUNT|(ro ? MS_RDONLY : 0), NULL) < 0)
6786                                 return -errno;
6787
6788                         x = strdup(cleaned);
6789                         if (!x)
6790                                 return -ENOMEM;
6791
6792                         r = set_consume(done, x);
6793                         if (r < 0)
6794                                 return r;
6795                 }
6796
6797                 while ((x = set_steal_first(todo))) {
6798
6799                         r = set_consume(done, x);
6800                         if (r == -EEXIST)
6801                                 continue;
6802                         if (r < 0)
6803                                 return r;
6804
6805                         if (mount(NULL, x, NULL, MS_BIND|MS_REMOUNT|(ro ? MS_RDONLY : 0), NULL) < 0) {
6806
6807                                 /* Deal with mount points that are
6808                                  * obstructed by a later mount */
6809
6810                                 if (errno != ENOENT)
6811                                         return -errno;
6812                         }
6813
6814                 }
6815         }
6816 }
6817
6818 int fflush_and_check(FILE *f) {
6819         assert(f);
6820
6821         errno = 0;
6822         fflush(f);
6823
6824         if (ferror(f))
6825                 return errno ? -errno : -EIO;
6826
6827         return 0;
6828 }
6829
6830 char *tempfn_xxxxxx(const char *p) {
6831         const char *fn;
6832         char *t;
6833         size_t k;
6834
6835         assert(p);
6836
6837         t = new(char, strlen(p) + 1 + 6 + 1);
6838         if (!t)
6839                 return NULL;
6840
6841         fn = basename(p);
6842         k = fn - p;
6843
6844         strcpy(stpcpy(stpcpy(mempcpy(t, p, k), "."), fn), "XXXXXX");
6845
6846         return t;
6847 }
6848
6849 char *tempfn_random(const char *p) {
6850         const char *fn;
6851         char *t, *x;
6852         uint64_t u;
6853         size_t k;
6854         unsigned i;
6855
6856         assert(p);
6857
6858         t = new(char, strlen(p) + 1 + 16 + 1);
6859         if (!t)
6860                 return NULL;
6861
6862         fn = basename(p);
6863         k = fn - p;
6864
6865         x = stpcpy(stpcpy(mempcpy(t, p, k), "."), fn);
6866
6867         u = random_u64();
6868         for (i = 0; i < 16; i++) {
6869                 *(x++) = hexchar(u & 0xF);
6870                 u >>= 4;
6871         }
6872
6873         *x = 0;
6874
6875         return t;
6876 }
6877
6878 /* make sure the hostname is not "localhost" */
6879 bool is_localhost(const char *hostname) {
6880         assert(hostname);
6881
6882         /* This tries to identify local host and domain names
6883          * described in RFC6761 plus the redhatism of .localdomain */
6884
6885         return streq(hostname, "localhost") ||
6886                streq(hostname, "localhost.") ||
6887                streq(hostname, "localdomain.") ||
6888                streq(hostname, "localdomain") ||
6889                endswith(hostname, ".localhost") ||
6890                endswith(hostname, ".localhost.") ||
6891                endswith(hostname, ".localdomain") ||
6892                endswith(hostname, ".localdomain.");
6893 }
6894
6895 int take_password_lock(const char *root) {
6896
6897         struct flock flock = {
6898                 .l_type = F_WRLCK,
6899                 .l_whence = SEEK_SET,
6900                 .l_start = 0,
6901                 .l_len = 0,
6902         };
6903
6904         const char *path;
6905         int fd, r;
6906
6907         /* This is roughly the same as lckpwdf(), but not as awful. We
6908          * don't want to use alarm() and signals, hence we implement
6909          * our own trivial version of this.
6910          *
6911          * Note that shadow-utils also takes per-database locks in
6912          * addition to lckpwdf(). However, we don't given that they
6913          * are redundant as they they invoke lckpwdf() first and keep
6914          * it during everything they do. The per-database locks are
6915          * awfully racy, and thus we just won't do them. */
6916
6917         if (root)
6918                 path = strappenda(root, "/etc/.pwd.lock");
6919         else
6920                 path = "/etc/.pwd.lock";
6921
6922         fd = open(path, O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW, 0600);
6923         if (fd < 0)
6924                 return -errno;
6925
6926         r = fcntl(fd, F_SETLKW, &flock);
6927         if (r < 0) {
6928                 safe_close(fd);
6929                 return -errno;
6930         }
6931
6932         return fd;
6933 }
6934
6935 int is_symlink(const char *path) {
6936         struct stat info;
6937
6938         if (lstat(path, &info) < 0)
6939                 return -errno;
6940
6941         if (S_ISLNK(info.st_mode))
6942                 return 1;
6943
6944         return 0;
6945 }
6946
6947 int unquote_first_word(const char **p, char **ret) {
6948         _cleanup_free_ char *s = NULL;
6949         size_t allocated = 0, sz = 0;
6950
6951         enum {
6952                 START,
6953                 VALUE,
6954                 VALUE_ESCAPE,
6955                 SINGLE_QUOTE,
6956                 SINGLE_QUOTE_ESCAPE,
6957                 DOUBLE_QUOTE,
6958                 DOUBLE_QUOTE_ESCAPE,
6959                 SPACE,
6960         } state = START;
6961
6962         assert(p);
6963         assert(*p);
6964         assert(ret);
6965
6966         /* Parses the first word of a string, and returns it in
6967          * *ret. Removes all quotes in the process. When parsing fails
6968          * (because of an uneven number of quotes or similar), leaves
6969          * the pointer *p at the first invalid character. */
6970
6971         for (;;) {
6972                 char c = **p;
6973
6974                 switch (state) {
6975
6976                 case START:
6977                         if (c == 0)
6978                                 goto finish;
6979                         else if (strchr(WHITESPACE, c))
6980                                 break;
6981
6982                         state = VALUE;
6983                         /* fallthrough */
6984
6985                 case VALUE:
6986                         if (c == 0)
6987                                 goto finish;
6988                         else if (c == '\'')
6989                                 state = SINGLE_QUOTE;
6990                         else if (c == '\\')
6991                                 state = VALUE_ESCAPE;
6992                         else if (c == '\"')
6993                                 state = DOUBLE_QUOTE;
6994                         else if (strchr(WHITESPACE, c))
6995                                 state = SPACE;
6996                         else {
6997                                 if (!GREEDY_REALLOC(s, allocated, sz+2))
6998                                         return -ENOMEM;
6999
7000                                 s[sz++] = c;
7001                         }
7002
7003                         break;
7004
7005                 case VALUE_ESCAPE:
7006                         if (c == 0)
7007                                 return -EINVAL;
7008
7009                         if (!GREEDY_REALLOC(s, allocated, sz+2))
7010                                 return -ENOMEM;
7011
7012                         s[sz++] = c;
7013                         state = VALUE;
7014
7015                         break;
7016
7017                 case SINGLE_QUOTE:
7018                         if (c == 0)
7019                                 return -EINVAL;
7020                         else if (c == '\'')
7021                                 state = VALUE;
7022                         else if (c == '\\')
7023                                 state = SINGLE_QUOTE_ESCAPE;
7024                         else {
7025                                 if (!GREEDY_REALLOC(s, allocated, sz+2))
7026                                         return -ENOMEM;
7027
7028                                 s[sz++] = c;
7029                         }
7030
7031                         break;
7032
7033                 case SINGLE_QUOTE_ESCAPE:
7034                         if (c == 0)
7035                                 return -EINVAL;
7036
7037                         if (!GREEDY_REALLOC(s, allocated, sz+2))
7038                                 return -ENOMEM;
7039
7040                         s[sz++] = c;
7041                         state = SINGLE_QUOTE;
7042                         break;
7043
7044                 case DOUBLE_QUOTE:
7045                         if (c == 0)
7046                                 return -EINVAL;
7047                         else if (c == '\"')
7048                                 state = VALUE;
7049                         else if (c == '\\')
7050                                 state = DOUBLE_QUOTE_ESCAPE;
7051                         else {
7052                                 if (!GREEDY_REALLOC(s, allocated, sz+2))
7053                                         return -ENOMEM;
7054
7055                                 s[sz++] = c;
7056                         }
7057
7058                         break;
7059
7060                 case DOUBLE_QUOTE_ESCAPE:
7061                         if (c == 0)
7062                                 return -EINVAL;
7063
7064                         if (!GREEDY_REALLOC(s, allocated, sz+2))
7065                                 return -ENOMEM;
7066
7067                         s[sz++] = c;
7068                         state = DOUBLE_QUOTE;
7069                         break;
7070
7071                 case SPACE:
7072                         if (c == 0)
7073                                 goto finish;
7074                         if (!strchr(WHITESPACE, c))
7075                                 goto finish;
7076
7077                         break;
7078                 }
7079
7080                 (*p) ++;
7081         }
7082
7083 finish:
7084         if (!s) {
7085                 *ret = NULL;
7086                 return 0;
7087         }
7088
7089         s[sz] = 0;
7090         *ret = s;
7091         s = NULL;
7092
7093         return 1;
7094 }
7095
7096 int unquote_many_words(const char **p, ...) {
7097         va_list ap;
7098         char **l;
7099         int n = 0, i, c, r;
7100
7101         /* Parses a number of words from a string, stripping any
7102          * quotes if necessary. */
7103
7104         assert(p);
7105
7106         /* Count how many words are expected */
7107         va_start(ap, p);
7108         for (;;) {
7109                 if (!va_arg(ap, char **))
7110                         break;
7111                 n++;
7112         }
7113         va_end(ap);
7114
7115         if (n <= 0)
7116                 return 0;
7117
7118         /* Read all words into a temporary array */
7119         l = newa0(char*, n);
7120         for (c = 0; c < n; c++) {
7121
7122                 r = unquote_first_word(p, &l[c]);
7123                 if (r < 0) {
7124                         int j;
7125
7126                         for (j = 0; j < c; j++)
7127                                 free(l[j]);
7128
7129                         return r;
7130                 }
7131
7132                 if (r == 0)
7133                         break;
7134         }
7135
7136         /* If we managed to parse all words, return them in the passed
7137          * in parameters */
7138         va_start(ap, p);
7139         for (i = 0; i < n; i++) {
7140                 char **v;
7141
7142                 v = va_arg(ap, char **);
7143                 assert(v);
7144
7145                 *v = l[i];
7146         }
7147         va_end(ap);
7148
7149         return c;
7150 }
7151
7152 int free_and_strdup(char **p, const char *s) {
7153         char *t;
7154
7155         assert(p);
7156
7157         /* Replaces a string pointer with an strdup()ed new string,
7158          * possibly freeing the old one. */
7159
7160         if (s) {
7161                 t = strdup(s);
7162                 if (!t)
7163                         return -ENOMEM;
7164         } else
7165                 t = NULL;
7166
7167         free(*p);
7168         *p = t;
7169
7170         return 0;
7171 }