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