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