src/shared/util.c

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