src/shared/util.c

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