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