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