src/shared/util.c

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