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