chiark / gitweb /
util: modernize loop_read() and loop_write() a bit
[elogind.git] / src / shared / util.c
1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
2
3 /***
4   This file is part of systemd.
5
6   Copyright 2010 Lennart Poettering
7
8   systemd is free software; you can redistribute it and/or modify it
9   under the terms of the GNU Lesser General Public License as published by
10   the Free Software Foundation; either version 2.1 of the License, or
11   (at your option) any later version.
12
13   systemd is distributed in the hope that it will be useful, but
14   WITHOUT ANY WARRANTY; without even the implied warranty of
15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16   Lesser General Public License for more details.
17
18   You should have received a copy of the GNU Lesser General Public License
19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
20 ***/
21
22 #include <assert.h>
23 #include <string.h>
24 #include <unistd.h>
25 #include <errno.h>
26 #include <stdlib.h>
27 #include <signal.h>
28 #include <stdio.h>
29 #include <syslog.h>
30 #include <sched.h>
31 #include <sys/resource.h>
32 #include <linux/sched.h>
33 #include <sys/types.h>
34 #include <sys/stat.h>
35 #include <fcntl.h>
36 #include <dirent.h>
37 #include <sys/ioctl.h>
38 #include <linux/vt.h>
39 #include <linux/tiocl.h>
40 #include <termios.h>
41 #include <stdarg.h>
42 #include <sys/inotify.h>
43 #include <sys/poll.h>
44 #include <ctype.h>
45 #include <sys/prctl.h>
46 #include <sys/utsname.h>
47 #include <pwd.h>
48 #include <netinet/ip.h>
49 #include <linux/kd.h>
50 #include <dlfcn.h>
51 #include <sys/wait.h>
52 #include <sys/time.h>
53 #include <glob.h>
54 #include <grp.h>
55 #include <sys/mman.h>
56 #include <sys/vfs.h>
57 #include <linux/magic.h>
58 #include <limits.h>
59 #include <langinfo.h>
60 #include <locale.h>
61 #include <libgen.h>
62 #undef basename
63
64 #ifdef HAVE_SYS_AUXV_H
65 #include <sys/auxv.h>
66 #endif
67
68 #include "macro.h"
69 #include "util.h"
70 #include "ioprio.h"
71 #include "missing.h"
72 #include "log.h"
73 #include "strv.h"
74 #include "label.h"
75 #include "path-util.h"
76 #include "exit-status.h"
77 #include "hashmap.h"
78 #include "env-util.h"
79 #include "fileio.h"
80 #include "device-nodes.h"
81 #include "utf8.h"
82 #include "gunicode.h"
83 #include "virt.h"
84 #include "def.h"
85
86 int saved_argc = 0;
87 char **saved_argv = NULL;
88
89 static volatile unsigned cached_columns = 0;
90 static volatile unsigned cached_lines = 0;
91
92 size_t page_size(void) {
93         static thread_local size_t pgsz = 0;
94         long r;
95
96         if (_likely_(pgsz > 0))
97                 return pgsz;
98
99         r = sysconf(_SC_PAGESIZE);
100         assert(r > 0);
101
102         pgsz = (size_t) r;
103         return pgsz;
104 }
105
106 bool streq_ptr(const char *a, const char *b) {
107
108         /* Like streq(), but tries to make sense of NULL pointers */
109
110         if (a && b)
111                 return streq(a, b);
112
113         if (!a && !b)
114                 return true;
115
116         return false;
117 }
118
119 char* endswith(const char *s, const char *postfix) {
120         size_t sl, pl;
121
122         assert(s);
123         assert(postfix);
124
125         sl = strlen(s);
126         pl = strlen(postfix);
127
128         if (pl == 0)
129                 return (char*) s + sl;
130
131         if (sl < pl)
132                 return NULL;
133
134         if (memcmp(s + sl - pl, postfix, pl) != 0)
135                 return NULL;
136
137         return (char*) s + sl - pl;
138 }
139
140 bool first_word(const char *s, const char *word) {
141         size_t sl, wl;
142
143         assert(s);
144         assert(word);
145
146         sl = strlen(s);
147         wl = strlen(word);
148
149         if (sl < wl)
150                 return false;
151
152         if (wl == 0)
153                 return true;
154
155         if (memcmp(s, word, wl) != 0)
156                 return false;
157
158         return s[wl] == 0 ||
159                 strchr(WHITESPACE, s[wl]);
160 }
161
162 int close_nointr(int fd) {
163         int r;
164
165         assert(fd >= 0);
166         r = close(fd);
167
168         /* Just ignore EINTR; a retry loop is the wrong
169          * thing to do on Linux.
170          *
171          * http://lkml.indiana.edu/hypermail/linux/kernel/0509.1/0877.html
172          * https://bugzilla.gnome.org/show_bug.cgi?id=682819
173          * http://utcc.utoronto.ca/~cks/space/blog/unix/CloseEINTR
174          * https://sites.google.com/site/michaelsafyan/software-engineering/checkforeintrwheninvokingclosethinkagain
175          */
176         if (_unlikely_(r < 0 && errno == EINTR))
177                 return 0;
178         else if (r >= 0)
179                 return r;
180         else
181                 return -errno;
182 }
183
184 void close_nointr_nofail(int fd) {
185         PROTECT_ERRNO;
186
187         /* like close_nointr() but cannot fail, and guarantees errno
188          * is unchanged */
189
190         assert_se(close_nointr(fd) == 0);
191 }
192
193 void close_many(const int fds[], unsigned n_fd) {
194         unsigned i;
195
196         assert(fds || n_fd <= 0);
197
198         for (i = 0; i < n_fd; i++)
199                 close_nointr_nofail(fds[i]);
200 }
201
202 int unlink_noerrno(const char *path) {
203         PROTECT_ERRNO;
204         int r;
205
206         r = unlink(path);
207         if (r < 0)
208                 return -errno;
209
210         return 0;
211 }
212
213 int parse_boolean(const char *v) {
214         assert(v);
215
216         if (streq(v, "1") || v[0] == 'y' || v[0] == 'Y' || v[0] == 't' || v[0] == 'T' || strcaseeq(v, "on"))
217                 return 1;
218         else if (streq(v, "0") || v[0] == 'n' || v[0] == 'N' || v[0] == 'f' || v[0] == 'F' || strcaseeq(v, "off"))
219                 return 0;
220
221         return -EINVAL;
222 }
223
224 int parse_pid(const char *s, pid_t* ret_pid) {
225         unsigned long ul = 0;
226         pid_t pid;
227         int r;
228
229         assert(s);
230         assert(ret_pid);
231
232         r = safe_atolu(s, &ul);
233         if (r < 0)
234                 return r;
235
236         pid = (pid_t) ul;
237
238         if ((unsigned long) pid != ul)
239                 return -ERANGE;
240
241         if (pid <= 0)
242                 return -ERANGE;
243
244         *ret_pid = pid;
245         return 0;
246 }
247
248 int parse_uid(const char *s, uid_t* ret_uid) {
249         unsigned long ul = 0;
250         uid_t uid;
251         int r;
252
253         assert(s);
254         assert(ret_uid);
255
256         r = safe_atolu(s, &ul);
257         if (r < 0)
258                 return r;
259
260         uid = (uid_t) ul;
261
262         if ((unsigned long) uid != ul)
263                 return -ERANGE;
264
265         *ret_uid = uid;
266         return 0;
267 }
268
269 int safe_atou(const char *s, unsigned *ret_u) {
270         char *x = NULL;
271         unsigned long l;
272
273         assert(s);
274         assert(ret_u);
275
276         errno = 0;
277         l = strtoul(s, &x, 0);
278
279         if (!x || x == s || *x || errno)
280                 return errno > 0 ? -errno : -EINVAL;
281
282         if ((unsigned long) (unsigned) l != l)
283                 return -ERANGE;
284
285         *ret_u = (unsigned) l;
286         return 0;
287 }
288
289 int safe_atoi(const char *s, int *ret_i) {
290         char *x = NULL;
291         long l;
292
293         assert(s);
294         assert(ret_i);
295
296         errno = 0;
297         l = strtol(s, &x, 0);
298
299         if (!x || x == s || *x || errno)
300                 return errno > 0 ? -errno : -EINVAL;
301
302         if ((long) (int) l != l)
303                 return -ERANGE;
304
305         *ret_i = (int) l;
306         return 0;
307 }
308
309 int safe_atollu(const char *s, long long unsigned *ret_llu) {
310         char *x = NULL;
311         unsigned long long l;
312
313         assert(s);
314         assert(ret_llu);
315
316         errno = 0;
317         l = strtoull(s, &x, 0);
318
319         if (!x || x == s || *x || errno)
320                 return errno ? -errno : -EINVAL;
321
322         *ret_llu = l;
323         return 0;
324 }
325
326 int safe_atolli(const char *s, long long int *ret_lli) {
327         char *x = NULL;
328         long long l;
329
330         assert(s);
331         assert(ret_lli);
332
333         errno = 0;
334         l = strtoll(s, &x, 0);
335
336         if (!x || x == s || *x || errno)
337                 return errno ? -errno : -EINVAL;
338
339         *ret_lli = l;
340         return 0;
341 }
342
343 int safe_atod(const char *s, double *ret_d) {
344         char *x = NULL;
345         double d = 0;
346
347         assert(s);
348         assert(ret_d);
349
350         RUN_WITH_LOCALE(LC_NUMERIC_MASK, "C") {
351                 errno = 0;
352                 d = strtod(s, &x);
353         }
354
355         if (!x || x == s || *x || errno)
356                 return errno ? -errno : -EINVAL;
357
358         *ret_d = (double) d;
359         return 0;
360 }
361
362 static size_t strcspn_escaped(const char *s, const char *reject) {
363         bool escaped = false;
364         size_t n;
365
366         for (n=0; s[n]; n++) {
367                 if (escaped)
368                         escaped = false;
369                 else if (s[n] == '\\')
370                         escaped = true;
371                 else if (strchr(reject, s[n]))
372                         return n;
373         }
374         return n;
375 }
376
377 /* Split a string into words. */
378 char *split(const char *c, size_t *l, const char *separator, bool quoted, char **state) {
379         char *current;
380
381         current = *state ? *state : (char*) c;
382
383         if (!*current || *c == 0)
384                 return NULL;
385
386         current += strspn(current, separator);
387         if (!*current)
388                 return NULL;
389
390         if (quoted && strchr("\'\"", *current)) {
391                 char quotechar = *(current++);
392                 *l = strcspn_escaped(current, (char[]){quotechar, '\0'});
393                 *state = current+*l+1;
394         } else if (quoted) {
395                 *l = strcspn_escaped(current, separator);
396                 *state = current+*l;
397         } else {
398                 *l = strcspn(current, separator);
399                 *state = current+*l;
400         }
401
402         return (char*) current;
403 }
404
405 int get_parent_of_pid(pid_t pid, pid_t *_ppid) {
406         int r;
407         _cleanup_free_ char *line = NULL;
408         long unsigned ppid;
409         const char *p;
410
411         assert(pid >= 0);
412         assert(_ppid);
413
414         if (pid == 0) {
415                 *_ppid = getppid();
416                 return 0;
417         }
418
419         p = procfs_file_alloca(pid, "stat");
420         r = read_one_line_file(p, &line);
421         if (r < 0)
422                 return r;
423
424         /* Let's skip the pid and comm fields. The latter is enclosed
425          * in () but does not escape any () in its value, so let's
426          * skip over it manually */
427
428         p = strrchr(line, ')');
429         if (!p)
430                 return -EIO;
431
432         p++;
433
434         if (sscanf(p, " "
435                    "%*c "  /* state */
436                    "%lu ", /* ppid */
437                    &ppid) != 1)
438                 return -EIO;
439
440         if ((long unsigned) (pid_t) ppid != ppid)
441                 return -ERANGE;
442
443         *_ppid = (pid_t) ppid;
444
445         return 0;
446 }
447
448 int get_starttime_of_pid(pid_t pid, unsigned long long *st) {
449         int r;
450         _cleanup_free_ char *line = NULL;
451         const char *p;
452
453         assert(pid >= 0);
454         assert(st);
455
456         p = procfs_file_alloca(pid, "stat");
457         r = read_one_line_file(p, &line);
458         if (r < 0)
459                 return r;
460
461         /* Let's skip the pid and comm fields. The latter is enclosed
462          * in () but does not escape any () in its value, so let's
463          * skip over it manually */
464
465         p = strrchr(line, ')');
466         if (!p)
467                 return -EIO;
468
469         p++;
470
471         if (sscanf(p, " "
472                    "%*c "  /* state */
473                    "%*d "  /* ppid */
474                    "%*d "  /* pgrp */
475                    "%*d "  /* session */
476                    "%*d "  /* tty_nr */
477                    "%*d "  /* tpgid */
478                    "%*u "  /* flags */
479                    "%*u "  /* minflt */
480                    "%*u "  /* cminflt */
481                    "%*u "  /* majflt */
482                    "%*u "  /* cmajflt */
483                    "%*u "  /* utime */
484                    "%*u "  /* stime */
485                    "%*d "  /* cutime */
486                    "%*d "  /* cstime */
487                    "%*d "  /* priority */
488                    "%*d "  /* nice */
489                    "%*d "  /* num_threads */
490                    "%*d "  /* itrealvalue */
491                    "%llu "  /* starttime */,
492                    st) != 1)
493                 return -EIO;
494
495         return 0;
496 }
497
498 int fchmod_umask(int fd, mode_t m) {
499         mode_t u;
500         int r;
501
502         u = umask(0777);
503         r = fchmod(fd, m & (~u)) < 0 ? -errno : 0;
504         umask(u);
505
506         return r;
507 }
508
509 char *truncate_nl(char *s) {
510         assert(s);
511
512         s[strcspn(s, NEWLINE)] = 0;
513         return s;
514 }
515
516 int get_process_comm(pid_t pid, char **name) {
517         const char *p;
518         int r;
519
520         assert(name);
521         assert(pid >= 0);
522
523         p = procfs_file_alloca(pid, "comm");
524
525         r = read_one_line_file(p, name);
526         if (r == -ENOENT)
527                 return -ESRCH;
528
529         return r;
530 }
531
532 int get_process_cmdline(pid_t pid, size_t max_length, bool comm_fallback, char **line) {
533         _cleanup_fclose_ FILE *f = NULL;
534         char *r = NULL, *k;
535         const char *p;
536         int c;
537
538         assert(line);
539         assert(pid >= 0);
540
541         p = procfs_file_alloca(pid, "cmdline");
542
543         f = fopen(p, "re");
544         if (!f)
545                 return -errno;
546
547         if (max_length == 0) {
548                 size_t len = 0, allocated = 0;
549
550                 while ((c = getc(f)) != EOF) {
551
552                         if (!GREEDY_REALLOC(r, allocated, len+2)) {
553                                 free(r);
554                                 return -ENOMEM;
555                         }
556
557                         r[len++] = isprint(c) ? c : ' ';
558                 }
559
560                 if (len > 0)
561                         r[len-1] = 0;
562
563         } else {
564                 bool space = false;
565                 size_t left;
566
567                 r = new(char, max_length);
568                 if (!r)
569                         return -ENOMEM;
570
571                 k = r;
572                 left = max_length;
573                 while ((c = getc(f)) != EOF) {
574
575                         if (isprint(c)) {
576                                 if (space) {
577                                         if (left <= 4)
578                                                 break;
579
580                                         *(k++) = ' ';
581                                         left--;
582                                         space = false;
583                                 }
584
585                                 if (left <= 4)
586                                         break;
587
588                                 *(k++) = (char) c;
589                                 left--;
590                         }  else
591                                 space = true;
592                 }
593
594                 if (left <= 4) {
595                         size_t n = MIN(left-1, 3U);
596                         memcpy(k, "...", n);
597                         k[n] = 0;
598                 } else
599                         *k = 0;
600         }
601
602         /* Kernel threads have no argv[] */
603         if (r == NULL || r[0] == 0) {
604                 _cleanup_free_ char *t = NULL;
605                 int h;
606
607                 free(r);
608
609                 if (!comm_fallback)
610                         return -ENOENT;
611
612                 h = get_process_comm(pid, &t);
613                 if (h < 0)
614                         return h;
615
616                 r = strjoin("[", t, "]", NULL);
617                 if (!r)
618                         return -ENOMEM;
619         }
620
621         *line = r;
622         return 0;
623 }
624
625 int is_kernel_thread(pid_t pid) {
626         const char *p;
627         size_t count;
628         char c;
629         bool eof;
630         FILE *f;
631
632         if (pid == 0)
633                 return 0;
634
635         assert(pid > 0);
636
637         p = procfs_file_alloca(pid, "cmdline");
638         f = fopen(p, "re");
639         if (!f)
640                 return -errno;
641
642         count = fread(&c, 1, 1, f);
643         eof = feof(f);
644         fclose(f);
645
646         /* Kernel threads have an empty cmdline */
647
648         if (count <= 0)
649                 return eof ? 1 : -errno;
650
651         return 0;
652 }
653
654 int get_process_capeff(pid_t pid, char **capeff) {
655         const char *p;
656
657         assert(capeff);
658         assert(pid >= 0);
659
660         p = procfs_file_alloca(pid, "status");
661
662         return get_status_field(p, "\nCapEff:", capeff);
663 }
664
665 int get_process_exe(pid_t pid, char **name) {
666         const char *p;
667         char *d;
668         int r;
669
670         assert(pid >= 0);
671         assert(name);
672
673         p = procfs_file_alloca(pid, "exe");
674
675         r = readlink_malloc(p, name);
676         if (r < 0)
677                 return r == -ENOENT ? -ESRCH : r;
678
679         d = endswith(*name, " (deleted)");
680         if (d)
681                 *d = '\0';
682
683         return 0;
684 }
685
686 static int get_process_id(pid_t pid, const char *field, uid_t *uid) {
687         _cleanup_fclose_ FILE *f = NULL;
688         char line[LINE_MAX];
689         const char *p;
690
691         assert(field);
692         assert(uid);
693
694         if (pid == 0)
695                 return getuid();
696
697         p = procfs_file_alloca(pid, "status");
698         f = fopen(p, "re");
699         if (!f)
700                 return -errno;
701
702         FOREACH_LINE(line, f, return -errno) {
703                 char *l;
704
705                 l = strstrip(line);
706
707                 if (startswith(l, field)) {
708                         l += strlen(field);
709                         l += strspn(l, WHITESPACE);
710
711                         l[strcspn(l, WHITESPACE)] = 0;
712
713                         return parse_uid(l, uid);
714                 }
715         }
716
717         return -EIO;
718 }
719
720 int get_process_uid(pid_t pid, uid_t *uid) {
721         return get_process_id(pid, "Uid:", uid);
722 }
723
724 int get_process_gid(pid_t pid, gid_t *gid) {
725         assert_cc(sizeof(uid_t) == sizeof(gid_t));
726         return get_process_id(pid, "Gid:", gid);
727 }
728
729 char *strnappend(const char *s, const char *suffix, size_t b) {
730         size_t a;
731         char *r;
732
733         if (!s && !suffix)
734                 return strdup("");
735
736         if (!s)
737                 return strndup(suffix, b);
738
739         if (!suffix)
740                 return strdup(s);
741
742         assert(s);
743         assert(suffix);
744
745         a = strlen(s);
746         if (b > ((size_t) -1) - a)
747                 return NULL;
748
749         r = new(char, a+b+1);
750         if (!r)
751                 return NULL;
752
753         memcpy(r, s, a);
754         memcpy(r+a, suffix, b);
755         r[a+b] = 0;
756
757         return r;
758 }
759
760 char *strappend(const char *s, const char *suffix) {
761         return strnappend(s, suffix, suffix ? strlen(suffix) : 0);
762 }
763
764 int readlink_malloc(const char *p, char **r) {
765         size_t l = 100;
766
767         assert(p);
768         assert(r);
769
770         for (;;) {
771                 char *c;
772                 ssize_t n;
773
774                 if (!(c = new(char, l)))
775                         return -ENOMEM;
776
777                 if ((n = readlink(p, c, l-1)) < 0) {
778                         int ret = -errno;
779                         free(c);
780                         return ret;
781                 }
782
783                 if ((size_t) n < l-1) {
784                         c[n] = 0;
785                         *r = c;
786                         return 0;
787                 }
788
789                 free(c);
790                 l *= 2;
791         }
792 }
793
794 int readlink_and_make_absolute(const char *p, char **r) {
795         _cleanup_free_ char *target = NULL;
796         char *k;
797         int j;
798
799         assert(p);
800         assert(r);
801
802         j = readlink_malloc(p, &target);
803         if (j < 0)
804                 return j;
805
806         k = file_in_same_dir(p, target);
807         if (!k)
808                 return -ENOMEM;
809
810         *r = k;
811         return 0;
812 }
813
814 int readlink_and_canonicalize(const char *p, char **r) {
815         char *t, *s;
816         int j;
817
818         assert(p);
819         assert(r);
820
821         j = readlink_and_make_absolute(p, &t);
822         if (j < 0)
823                 return j;
824
825         s = canonicalize_file_name(t);
826         if (s) {
827                 free(t);
828                 *r = s;
829         } else
830                 *r = t;
831
832         path_kill_slashes(*r);
833
834         return 0;
835 }
836
837 int reset_all_signal_handlers(void) {
838         int sig;
839
840         for (sig = 1; sig < _NSIG; sig++) {
841                 struct sigaction sa = {
842                         .sa_handler = SIG_DFL,
843                         .sa_flags = SA_RESTART,
844                 };
845
846                 if (sig == SIGKILL || sig == SIGSTOP)
847                         continue;
848
849                 /* On Linux the first two RT signals are reserved by
850                  * glibc, and sigaction() will return EINVAL for them. */
851                 if ((sigaction(sig, &sa, NULL) < 0))
852                         if (errno != EINVAL)
853                                 return -errno;
854         }
855
856         return 0;
857 }
858
859 char *strstrip(char *s) {
860         char *e;
861
862         /* Drops trailing whitespace. Modifies the string in
863          * place. Returns pointer to first non-space character */
864
865         s += strspn(s, WHITESPACE);
866
867         for (e = strchr(s, 0); e > s; e --)
868                 if (!strchr(WHITESPACE, e[-1]))
869                         break;
870
871         *e = 0;
872
873         return s;
874 }
875
876 char *delete_chars(char *s, const char *bad) {
877         char *f, *t;
878
879         /* Drops all whitespace, regardless where in the string */
880
881         for (f = s, t = s; *f; f++) {
882                 if (strchr(bad, *f))
883                         continue;
884
885                 *(t++) = *f;
886         }
887
888         *t = 0;
889
890         return s;
891 }
892
893 bool in_charset(const char *s, const char* charset) {
894         const char *i;
895
896         assert(s);
897         assert(charset);
898
899         for (i = s; *i; i++)
900                 if (!strchr(charset, *i))
901                         return false;
902
903         return true;
904 }
905
906 char *file_in_same_dir(const char *path, const char *filename) {
907         char *e, *r;
908         size_t k;
909
910         assert(path);
911         assert(filename);
912
913         /* This removes the last component of path and appends
914          * filename, unless the latter is absolute anyway or the
915          * former isn't */
916
917         if (path_is_absolute(filename))
918                 return strdup(filename);
919
920         if (!(e = strrchr(path, '/')))
921                 return strdup(filename);
922
923         k = strlen(filename);
924         if (!(r = new(char, e-path+1+k+1)))
925                 return NULL;
926
927         memcpy(r, path, e-path+1);
928         memcpy(r+(e-path)+1, filename, k+1);
929
930         return r;
931 }
932
933 int rmdir_parents(const char *path, const char *stop) {
934         size_t l;
935         int r = 0;
936
937         assert(path);
938         assert(stop);
939
940         l = strlen(path);
941
942         /* Skip trailing slashes */
943         while (l > 0 && path[l-1] == '/')
944                 l--;
945
946         while (l > 0) {
947                 char *t;
948
949                 /* Skip last component */
950                 while (l > 0 && path[l-1] != '/')
951                         l--;
952
953                 /* Skip trailing slashes */
954                 while (l > 0 && path[l-1] == '/')
955                         l--;
956
957                 if (l <= 0)
958                         break;
959
960                 if (!(t = strndup(path, l)))
961                         return -ENOMEM;
962
963                 if (path_startswith(stop, t)) {
964                         free(t);
965                         return 0;
966                 }
967
968                 r = rmdir(t);
969                 free(t);
970
971                 if (r < 0)
972                         if (errno != ENOENT)
973                                 return -errno;
974         }
975
976         return 0;
977 }
978
979 char hexchar(int x) {
980         static const char table[16] = "0123456789abcdef";
981
982         return table[x & 15];
983 }
984
985 int unhexchar(char c) {
986
987         if (c >= '0' && c <= '9')
988                 return c - '0';
989
990         if (c >= 'a' && c <= 'f')
991                 return c - 'a' + 10;
992
993         if (c >= 'A' && c <= 'F')
994                 return c - 'A' + 10;
995
996         return -1;
997 }
998
999 char *hexmem(const void *p, size_t l) {
1000         char *r, *z;
1001         const uint8_t *x;
1002
1003         z = r = malloc(l * 2 + 1);
1004         if (!r)
1005                 return NULL;
1006
1007         for (x = p; x < (const uint8_t*) p + l; x++) {
1008                 *(z++) = hexchar(*x >> 4);
1009                 *(z++) = hexchar(*x & 15);
1010         }
1011
1012         *z = 0;
1013         return r;
1014 }
1015
1016 void *unhexmem(const char *p, size_t l) {
1017         uint8_t *r, *z;
1018         const char *x;
1019
1020         assert(p);
1021
1022         z = r = malloc((l + 1) / 2 + 1);
1023         if (!r)
1024                 return NULL;
1025
1026         for (x = p; x < p + l; x += 2) {
1027                 int a, b;
1028
1029                 a = unhexchar(x[0]);
1030                 if (x+1 < p + l)
1031                         b = unhexchar(x[1]);
1032                 else
1033                         b = 0;
1034
1035                 *(z++) = (uint8_t) a << 4 | (uint8_t) b;
1036         }
1037
1038         *z = 0;
1039         return r;
1040 }
1041
1042 char octchar(int x) {
1043         return '0' + (x & 7);
1044 }
1045
1046 int unoctchar(char c) {
1047
1048         if (c >= '0' && c <= '7')
1049                 return c - '0';
1050
1051         return -1;
1052 }
1053
1054 char decchar(int x) {
1055         return '0' + (x % 10);
1056 }
1057
1058 int undecchar(char c) {
1059
1060         if (c >= '0' && c <= '9')
1061                 return c - '0';
1062
1063         return -1;
1064 }
1065
1066 char *cescape(const char *s) {
1067         char *r, *t;
1068         const char *f;
1069
1070         assert(s);
1071
1072         /* Does C style string escaping. */
1073
1074         r = new(char, strlen(s)*4 + 1);
1075         if (!r)
1076                 return NULL;
1077
1078         for (f = s, t = r; *f; f++)
1079
1080                 switch (*f) {
1081
1082                 case '\a':
1083                         *(t++) = '\\';
1084                         *(t++) = 'a';
1085                         break;
1086                 case '\b':
1087                         *(t++) = '\\';
1088                         *(t++) = 'b';
1089                         break;
1090                 case '\f':
1091                         *(t++) = '\\';
1092                         *(t++) = 'f';
1093                         break;
1094                 case '\n':
1095                         *(t++) = '\\';
1096                         *(t++) = 'n';
1097                         break;
1098                 case '\r':
1099                         *(t++) = '\\';
1100                         *(t++) = 'r';
1101                         break;
1102                 case '\t':
1103                         *(t++) = '\\';
1104                         *(t++) = 't';
1105                         break;
1106                 case '\v':
1107                         *(t++) = '\\';
1108                         *(t++) = 'v';
1109                         break;
1110                 case '\\':
1111                         *(t++) = '\\';
1112                         *(t++) = '\\';
1113                         break;
1114                 case '"':
1115                         *(t++) = '\\';
1116                         *(t++) = '"';
1117                         break;
1118                 case '\'':
1119                         *(t++) = '\\';
1120                         *(t++) = '\'';
1121                         break;
1122
1123                 default:
1124                         /* For special chars we prefer octal over
1125                          * hexadecimal encoding, simply because glib's
1126                          * g_strescape() does the same */
1127                         if ((*f < ' ') || (*f >= 127)) {
1128                                 *(t++) = '\\';
1129                                 *(t++) = octchar((unsigned char) *f >> 6);
1130                                 *(t++) = octchar((unsigned char) *f >> 3);
1131                                 *(t++) = octchar((unsigned char) *f);
1132                         } else
1133                                 *(t++) = *f;
1134                         break;
1135                 }
1136
1137         *t = 0;
1138
1139         return r;
1140 }
1141
1142 char *cunescape_length_with_prefix(const char *s, size_t length, const char *prefix) {
1143         char *r, *t;
1144         const char *f;
1145         size_t pl;
1146
1147         assert(s);
1148
1149         /* Undoes C style string escaping, and optionally prefixes it. */
1150
1151         pl = prefix ? strlen(prefix) : 0;
1152
1153         r = new(char, pl+length+1);
1154         if (!r)
1155                 return r;
1156
1157         if (prefix)
1158                 memcpy(r, prefix, pl);
1159
1160         for (f = s, t = r + pl; f < s + length; f++) {
1161
1162                 if (*f != '\\') {
1163                         *(t++) = *f;
1164                         continue;
1165                 }
1166
1167                 f++;
1168
1169                 switch (*f) {
1170
1171                 case 'a':
1172                         *(t++) = '\a';
1173                         break;
1174                 case 'b':
1175                         *(t++) = '\b';
1176                         break;
1177                 case 'f':
1178                         *(t++) = '\f';
1179                         break;
1180                 case 'n':
1181                         *(t++) = '\n';
1182                         break;
1183                 case 'r':
1184                         *(t++) = '\r';
1185                         break;
1186                 case 't':
1187                         *(t++) = '\t';
1188                         break;
1189                 case 'v':
1190                         *(t++) = '\v';
1191                         break;
1192                 case '\\':
1193                         *(t++) = '\\';
1194                         break;
1195                 case '"':
1196                         *(t++) = '"';
1197                         break;
1198                 case '\'':
1199                         *(t++) = '\'';
1200                         break;
1201
1202                 case 's':
1203                         /* This is an extension of the XDG syntax files */
1204                         *(t++) = ' ';
1205                         break;
1206
1207                 case 'x': {
1208                         /* hexadecimal encoding */
1209                         int a, b;
1210
1211                         a = unhexchar(f[1]);
1212                         b = unhexchar(f[2]);
1213
1214                         if (a < 0 || b < 0) {
1215                                 /* Invalid escape code, let's take it literal then */
1216                                 *(t++) = '\\';
1217                                 *(t++) = 'x';
1218                         } else {
1219                                 *(t++) = (char) ((a << 4) | b);
1220                                 f += 2;
1221                         }
1222
1223                         break;
1224                 }
1225
1226                 case '0':
1227                 case '1':
1228                 case '2':
1229                 case '3':
1230                 case '4':
1231                 case '5':
1232                 case '6':
1233                 case '7': {
1234                         /* octal encoding */
1235                         int a, b, c;
1236
1237                         a = unoctchar(f[0]);
1238                         b = unoctchar(f[1]);
1239                         c = unoctchar(f[2]);
1240
1241                         if (a < 0 || b < 0 || c < 0) {
1242                                 /* Invalid escape code, let's take it literal then */
1243                                 *(t++) = '\\';
1244                                 *(t++) = f[0];
1245                         } else {
1246                                 *(t++) = (char) ((a << 6) | (b << 3) | c);
1247                                 f += 2;
1248                         }
1249
1250                         break;
1251                 }
1252
1253                 case 0:
1254                         /* premature end of string.*/
1255                         *(t++) = '\\';
1256                         goto finish;
1257
1258                 default:
1259                         /* Invalid escape code, let's take it literal then */
1260                         *(t++) = '\\';
1261                         *(t++) = *f;
1262                         break;
1263                 }
1264         }
1265
1266 finish:
1267         *t = 0;
1268         return r;
1269 }
1270
1271 char *cunescape_length(const char *s, size_t length) {
1272         return cunescape_length_with_prefix(s, length, NULL);
1273 }
1274
1275 char *cunescape(const char *s) {
1276         assert(s);
1277
1278         return cunescape_length(s, strlen(s));
1279 }
1280
1281 char *xescape(const char *s, const char *bad) {
1282         char *r, *t;
1283         const char *f;
1284
1285         /* Escapes all chars in bad, in addition to \ and all special
1286          * chars, in \xFF style escaping. May be reversed with
1287          * cunescape. */
1288
1289         r = new(char, strlen(s) * 4 + 1);
1290         if (!r)
1291                 return NULL;
1292
1293         for (f = s, t = r; *f; f++) {
1294
1295                 if ((*f < ' ') || (*f >= 127) ||
1296                     (*f == '\\') || strchr(bad, *f)) {
1297                         *(t++) = '\\';
1298                         *(t++) = 'x';
1299                         *(t++) = hexchar(*f >> 4);
1300                         *(t++) = hexchar(*f);
1301                 } else
1302                         *(t++) = *f;
1303         }
1304
1305         *t = 0;
1306
1307         return r;
1308 }
1309
1310 char *ascii_strlower(char *t) {
1311         char *p;
1312
1313         assert(t);
1314
1315         for (p = t; *p; p++)
1316                 if (*p >= 'A' && *p <= 'Z')
1317                         *p = *p - 'A' + 'a';
1318
1319         return t;
1320 }
1321
1322 _pure_ static bool ignore_file_allow_backup(const char *filename) {
1323         assert(filename);
1324
1325         return
1326                 filename[0] == '.' ||
1327                 streq(filename, "lost+found") ||
1328                 streq(filename, "aquota.user") ||
1329                 streq(filename, "aquota.group") ||
1330                 endswith(filename, ".rpmnew") ||
1331                 endswith(filename, ".rpmsave") ||
1332                 endswith(filename, ".rpmorig") ||
1333                 endswith(filename, ".dpkg-old") ||
1334                 endswith(filename, ".dpkg-new") ||
1335                 endswith(filename, ".swp");
1336 }
1337
1338 bool ignore_file(const char *filename) {
1339         assert(filename);
1340
1341         if (endswith(filename, "~"))
1342                 return false;
1343
1344         return ignore_file_allow_backup(filename);
1345 }
1346
1347 int fd_nonblock(int fd, bool nonblock) {
1348         int flags;
1349
1350         assert(fd >= 0);
1351
1352         if ((flags = fcntl(fd, F_GETFL, 0)) < 0)
1353                 return -errno;
1354
1355         if (nonblock)
1356                 flags |= O_NONBLOCK;
1357         else
1358                 flags &= ~O_NONBLOCK;
1359
1360         if (fcntl(fd, F_SETFL, flags) < 0)
1361                 return -errno;
1362
1363         return 0;
1364 }
1365
1366 int fd_cloexec(int fd, bool cloexec) {
1367         int flags;
1368
1369         assert(fd >= 0);
1370
1371         if ((flags = fcntl(fd, F_GETFD, 0)) < 0)
1372                 return -errno;
1373
1374         if (cloexec)
1375                 flags |= FD_CLOEXEC;
1376         else
1377                 flags &= ~FD_CLOEXEC;
1378
1379         if (fcntl(fd, F_SETFD, flags) < 0)
1380                 return -errno;
1381
1382         return 0;
1383 }
1384
1385 _pure_ static bool fd_in_set(int fd, const int fdset[], unsigned n_fdset) {
1386         unsigned i;
1387
1388         assert(n_fdset == 0 || fdset);
1389
1390         for (i = 0; i < n_fdset; i++)
1391                 if (fdset[i] == fd)
1392                         return true;
1393
1394         return false;
1395 }
1396
1397 int close_all_fds(const int except[], unsigned n_except) {
1398         DIR *d;
1399         struct dirent *de;
1400         int r = 0;
1401
1402         assert(n_except == 0 || except);
1403
1404         d = opendir("/proc/self/fd");
1405         if (!d) {
1406                 int fd;
1407                 struct rlimit rl;
1408
1409                 /* When /proc isn't available (for example in chroots)
1410                  * the fallback is brute forcing through the fd
1411                  * table */
1412
1413                 assert_se(getrlimit(RLIMIT_NOFILE, &rl) >= 0);
1414                 for (fd = 3; fd < (int) rl.rlim_max; fd ++) {
1415
1416                         if (fd_in_set(fd, except, n_except))
1417                                 continue;
1418
1419                         if (close_nointr(fd) < 0)
1420                                 if (errno != EBADF && r == 0)
1421                                         r = -errno;
1422                 }
1423
1424                 return r;
1425         }
1426
1427         while ((de = readdir(d))) {
1428                 int fd = -1;
1429
1430                 if (ignore_file(de->d_name))
1431                         continue;
1432
1433                 if (safe_atoi(de->d_name, &fd) < 0)
1434                         /* Let's better ignore this, just in case */
1435                         continue;
1436
1437                 if (fd < 3)
1438                         continue;
1439
1440                 if (fd == dirfd(d))
1441                         continue;
1442
1443                 if (fd_in_set(fd, except, n_except))
1444                         continue;
1445
1446                 if (close_nointr(fd) < 0) {
1447                         /* Valgrind has its own FD and doesn't want to have it closed */
1448                         if (errno != EBADF && r == 0)
1449                                 r = -errno;
1450                 }
1451         }
1452
1453         closedir(d);
1454         return r;
1455 }
1456
1457 bool chars_intersect(const char *a, const char *b) {
1458         const char *p;
1459
1460         /* Returns true if any of the chars in a are in b. */
1461         for (p = a; *p; p++)
1462                 if (strchr(b, *p))
1463                         return true;
1464
1465         return false;
1466 }
1467
1468 bool fstype_is_network(const char *fstype) {
1469         static const char table[] =
1470                 "cifs\0"
1471                 "smbfs\0"
1472                 "ncpfs\0"
1473                 "ncp\0"
1474                 "nfs\0"
1475                 "nfs4\0"
1476                 "gfs\0"
1477                 "gfs2\0";
1478
1479         return nulstr_contains(table, fstype);
1480 }
1481
1482 int chvt(int vt) {
1483         _cleanup_close_ int fd;
1484
1485         fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
1486         if (fd < 0)
1487                 return -errno;
1488
1489         if (vt < 0) {
1490                 int tiocl[2] = {
1491                         TIOCL_GETKMSGREDIRECT,
1492                         0
1493                 };
1494
1495                 if (ioctl(fd, TIOCLINUX, tiocl) < 0)
1496                         return -errno;
1497
1498                 vt = tiocl[0] <= 0 ? 1 : tiocl[0];
1499         }
1500
1501         if (ioctl(fd, VT_ACTIVATE, vt) < 0)
1502                 return -errno;
1503
1504         return 0;
1505 }
1506
1507 int read_one_char(FILE *f, char *ret, usec_t t, bool *need_nl) {
1508         struct termios old_termios, new_termios;
1509         char c;
1510         char line[LINE_MAX];
1511
1512         assert(f);
1513         assert(ret);
1514
1515         if (tcgetattr(fileno(f), &old_termios) >= 0) {
1516                 new_termios = old_termios;
1517
1518                 new_termios.c_lflag &= ~ICANON;
1519                 new_termios.c_cc[VMIN] = 1;
1520                 new_termios.c_cc[VTIME] = 0;
1521
1522                 if (tcsetattr(fileno(f), TCSADRAIN, &new_termios) >= 0) {
1523                         size_t k;
1524
1525                         if (t != (usec_t) -1) {
1526                                 if (fd_wait_for_event(fileno(f), POLLIN, t) <= 0) {
1527                                         tcsetattr(fileno(f), TCSADRAIN, &old_termios);
1528                                         return -ETIMEDOUT;
1529                                 }
1530                         }
1531
1532                         k = fread(&c, 1, 1, f);
1533
1534                         tcsetattr(fileno(f), TCSADRAIN, &old_termios);
1535
1536                         if (k <= 0)
1537                                 return -EIO;
1538
1539                         if (need_nl)
1540                                 *need_nl = c != '\n';
1541
1542                         *ret = c;
1543                         return 0;
1544                 }
1545         }
1546
1547         if (t != (usec_t) -1)
1548                 if (fd_wait_for_event(fileno(f), POLLIN, t) <= 0)
1549                         return -ETIMEDOUT;
1550
1551         if (!fgets(line, sizeof(line), f))
1552                 return -EIO;
1553
1554         truncate_nl(line);
1555
1556         if (strlen(line) != 1)
1557                 return -EBADMSG;
1558
1559         if (need_nl)
1560                 *need_nl = false;
1561
1562         *ret = line[0];
1563         return 0;
1564 }
1565
1566 int ask(char *ret, const char *replies, const char *text, ...) {
1567
1568         assert(ret);
1569         assert(replies);
1570         assert(text);
1571
1572         for (;;) {
1573                 va_list ap;
1574                 char c;
1575                 int r;
1576                 bool need_nl = true;
1577
1578                 if (on_tty())
1579                         fputs(ANSI_HIGHLIGHT_ON, stdout);
1580
1581                 va_start(ap, text);
1582                 vprintf(text, ap);
1583                 va_end(ap);
1584
1585                 if (on_tty())
1586                         fputs(ANSI_HIGHLIGHT_OFF, stdout);
1587
1588                 fflush(stdout);
1589
1590                 r = read_one_char(stdin, &c, (usec_t) -1, &need_nl);
1591                 if (r < 0) {
1592
1593                         if (r == -EBADMSG) {
1594                                 puts("Bad input, please try again.");
1595                                 continue;
1596                         }
1597
1598                         putchar('\n');
1599                         return r;
1600                 }
1601
1602                 if (need_nl)
1603                         putchar('\n');
1604
1605                 if (strchr(replies, c)) {
1606                         *ret = c;
1607                         return 0;
1608                 }
1609
1610                 puts("Read unexpected character, please try again.");
1611         }
1612 }
1613
1614 int reset_terminal_fd(int fd, bool switch_to_text) {
1615         struct termios termios;
1616         int r = 0;
1617
1618         /* Set terminal to some sane defaults */
1619
1620         assert(fd >= 0);
1621
1622         /* We leave locked terminal attributes untouched, so that
1623          * Plymouth may set whatever it wants to set, and we don't
1624          * interfere with that. */
1625
1626         /* Disable exclusive mode, just in case */
1627         ioctl(fd, TIOCNXCL);
1628
1629         /* Switch to text mode */
1630         if (switch_to_text)
1631                 ioctl(fd, KDSETMODE, KD_TEXT);
1632
1633         /* Enable console unicode mode */
1634         ioctl(fd, KDSKBMODE, K_UNICODE);
1635
1636         if (tcgetattr(fd, &termios) < 0) {
1637                 r = -errno;
1638                 goto finish;
1639         }
1640
1641         /* We only reset the stuff that matters to the software. How
1642          * hardware is set up we don't touch assuming that somebody
1643          * else will do that for us */
1644
1645         termios.c_iflag &= ~(IGNBRK | BRKINT | ISTRIP | INLCR | IGNCR | IUCLC);
1646         termios.c_iflag |= ICRNL | IMAXBEL | IUTF8;
1647         termios.c_oflag |= ONLCR;
1648         termios.c_cflag |= CREAD;
1649         termios.c_lflag = ISIG | ICANON | IEXTEN | ECHO | ECHOE | ECHOK | ECHOCTL | ECHOPRT | ECHOKE;
1650
1651         termios.c_cc[VINTR]    =   03;  /* ^C */
1652         termios.c_cc[VQUIT]    =  034;  /* ^\ */
1653         termios.c_cc[VERASE]   = 0177;
1654         termios.c_cc[VKILL]    =  025;  /* ^X */
1655         termios.c_cc[VEOF]     =   04;  /* ^D */
1656         termios.c_cc[VSTART]   =  021;  /* ^Q */
1657         termios.c_cc[VSTOP]    =  023;  /* ^S */
1658         termios.c_cc[VSUSP]    =  032;  /* ^Z */
1659         termios.c_cc[VLNEXT]   =  026;  /* ^V */
1660         termios.c_cc[VWERASE]  =  027;  /* ^W */
1661         termios.c_cc[VREPRINT] =  022;  /* ^R */
1662         termios.c_cc[VEOL]     =    0;
1663         termios.c_cc[VEOL2]    =    0;
1664
1665         termios.c_cc[VTIME]  = 0;
1666         termios.c_cc[VMIN]   = 1;
1667
1668         if (tcsetattr(fd, TCSANOW, &termios) < 0)
1669                 r = -errno;
1670
1671 finish:
1672         /* Just in case, flush all crap out */
1673         tcflush(fd, TCIOFLUSH);
1674
1675         return r;
1676 }
1677
1678 int reset_terminal(const char *name) {
1679         int fd, r;
1680
1681         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
1682         if (fd < 0)
1683                 return fd;
1684
1685         r = reset_terminal_fd(fd, true);
1686         close_nointr_nofail(fd);
1687
1688         return r;
1689 }
1690
1691 int open_terminal(const char *name, int mode) {
1692         int fd, r;
1693         unsigned c = 0;
1694
1695         /*
1696          * If a TTY is in the process of being closed opening it might
1697          * cause EIO. This is horribly awful, but unlikely to be
1698          * changed in the kernel. Hence we work around this problem by
1699          * retrying a couple of times.
1700          *
1701          * https://bugs.launchpad.net/ubuntu/+source/linux/+bug/554172/comments/245
1702          */
1703
1704         assert(!(mode & O_CREAT));
1705
1706         for (;;) {
1707                 fd = open(name, mode, 0);
1708                 if (fd >= 0)
1709                         break;
1710
1711                 if (errno != EIO)
1712                         return -errno;
1713
1714                 /* Max 1s in total */
1715                 if (c >= 20)
1716                         return -errno;
1717
1718                 usleep(50 * USEC_PER_MSEC);
1719                 c++;
1720         }
1721
1722         if (fd < 0)
1723                 return -errno;
1724
1725         r = isatty(fd);
1726         if (r < 0) {
1727                 close_nointr_nofail(fd);
1728                 return -errno;
1729         }
1730
1731         if (!r) {
1732                 close_nointr_nofail(fd);
1733                 return -ENOTTY;
1734         }
1735
1736         return fd;
1737 }
1738
1739 int flush_fd(int fd) {
1740         struct pollfd pollfd = {
1741                 .fd = fd,
1742                 .events = POLLIN,
1743         };
1744
1745         for (;;) {
1746                 char buf[LINE_MAX];
1747                 ssize_t l;
1748                 int r;
1749
1750                 r = poll(&pollfd, 1, 0);
1751                 if (r < 0) {
1752                         if (errno == EINTR)
1753                                 continue;
1754
1755                         return -errno;
1756
1757                 } else if (r == 0)
1758                         return 0;
1759
1760                 l = read(fd, buf, sizeof(buf));
1761                 if (l < 0) {
1762
1763                         if (errno == EINTR)
1764                                 continue;
1765
1766                         if (errno == EAGAIN)
1767                                 return 0;
1768
1769                         return -errno;
1770                 } else if (l == 0)
1771                         return 0;
1772         }
1773 }
1774
1775 int acquire_terminal(
1776                 const char *name,
1777                 bool fail,
1778                 bool force,
1779                 bool ignore_tiocstty_eperm,
1780                 usec_t timeout) {
1781
1782         int fd = -1, notify = -1, r = 0, wd = -1;
1783         usec_t ts = 0;
1784
1785         assert(name);
1786
1787         /* We use inotify to be notified when the tty is closed. We
1788          * create the watch before checking if we can actually acquire
1789          * it, so that we don't lose any event.
1790          *
1791          * Note: strictly speaking this actually watches for the
1792          * device being closed, it does *not* really watch whether a
1793          * tty loses its controlling process. However, unless some
1794          * rogue process uses TIOCNOTTY on /dev/tty *after* closing
1795          * its tty otherwise this will not become a problem. As long
1796          * as the administrator makes sure not configure any service
1797          * on the same tty as an untrusted user this should not be a
1798          * problem. (Which he probably should not do anyway.) */
1799
1800         if (timeout != (usec_t) -1)
1801                 ts = now(CLOCK_MONOTONIC);
1802
1803         if (!fail && !force) {
1804                 notify = inotify_init1(IN_CLOEXEC | (timeout != (usec_t) -1 ? IN_NONBLOCK : 0));
1805                 if (notify < 0) {
1806                         r = -errno;
1807                         goto fail;
1808                 }
1809
1810                 wd = inotify_add_watch(notify, name, IN_CLOSE);
1811                 if (wd < 0) {
1812                         r = -errno;
1813                         goto fail;
1814                 }
1815         }
1816
1817         for (;;) {
1818                 struct sigaction sa_old, sa_new = {
1819                         .sa_handler = SIG_IGN,
1820                         .sa_flags = SA_RESTART,
1821                 };
1822
1823                 if (notify >= 0) {
1824                         r = flush_fd(notify);
1825                         if (r < 0)
1826                                 goto fail;
1827                 }
1828
1829                 /* We pass here O_NOCTTY only so that we can check the return
1830                  * value TIOCSCTTY and have a reliable way to figure out if we
1831                  * successfully became the controlling process of the tty */
1832                 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
1833                 if (fd < 0)
1834                         return fd;
1835
1836                 /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
1837                  * if we already own the tty. */
1838                 assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
1839
1840                 /* First, try to get the tty */
1841                 if (ioctl(fd, TIOCSCTTY, force) < 0)
1842                         r = -errno;
1843
1844                 assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
1845
1846                 /* Sometimes it makes sense to ignore TIOCSCTTY
1847                  * returning EPERM, i.e. when very likely we already
1848                  * are have this controlling terminal. */
1849                 if (r < 0 && r == -EPERM && ignore_tiocstty_eperm)
1850                         r = 0;
1851
1852                 if (r < 0 && (force || fail || r != -EPERM)) {
1853                         goto fail;
1854                 }
1855
1856                 if (r >= 0)
1857                         break;
1858
1859                 assert(!fail);
1860                 assert(!force);
1861                 assert(notify >= 0);
1862
1863                 for (;;) {
1864                         uint8_t inotify_buffer[sizeof(struct inotify_event) + FILENAME_MAX];
1865                         ssize_t l;
1866                         struct inotify_event *e;
1867
1868                         if (timeout != (usec_t) -1) {
1869                                 usec_t n;
1870
1871                                 n = now(CLOCK_MONOTONIC);
1872                                 if (ts + timeout < n) {
1873                                         r = -ETIMEDOUT;
1874                                         goto fail;
1875                                 }
1876
1877                                 r = fd_wait_for_event(fd, POLLIN, ts + timeout - n);
1878                                 if (r < 0)
1879                                         goto fail;
1880
1881                                 if (r == 0) {
1882                                         r = -ETIMEDOUT;
1883                                         goto fail;
1884                                 }
1885                         }
1886
1887                         l = read(notify, inotify_buffer, sizeof(inotify_buffer));
1888                         if (l < 0) {
1889
1890                                 if (errno == EINTR || errno == EAGAIN)
1891                                         continue;
1892
1893                                 r = -errno;
1894                                 goto fail;
1895                         }
1896
1897                         e = (struct inotify_event*) inotify_buffer;
1898
1899                         while (l > 0) {
1900                                 size_t step;
1901
1902                                 if (e->wd != wd || !(e->mask & IN_CLOSE)) {
1903                                         r = -EIO;
1904                                         goto fail;
1905                                 }
1906
1907                                 step = sizeof(struct inotify_event) + e->len;
1908                                 assert(step <= (size_t) l);
1909
1910                                 e = (struct inotify_event*) ((uint8_t*) e + step);
1911                                 l -= step;
1912                         }
1913
1914                         break;
1915                 }
1916
1917                 /* We close the tty fd here since if the old session
1918                  * ended our handle will be dead. It's important that
1919                  * we do this after sleeping, so that we don't enter
1920                  * an endless loop. */
1921                 close_nointr_nofail(fd);
1922         }
1923
1924         if (notify >= 0)
1925                 close_nointr_nofail(notify);
1926
1927         r = reset_terminal_fd(fd, true);
1928         if (r < 0)
1929                 log_warning("Failed to reset terminal: %s", strerror(-r));
1930
1931         return fd;
1932
1933 fail:
1934         if (fd >= 0)
1935                 close_nointr_nofail(fd);
1936
1937         if (notify >= 0)
1938                 close_nointr_nofail(notify);
1939
1940         return r;
1941 }
1942
1943 int release_terminal(void) {
1944         int r = 0;
1945         struct sigaction sa_old, sa_new = {
1946                 .sa_handler = SIG_IGN,
1947                 .sa_flags = SA_RESTART,
1948         };
1949         _cleanup_close_ int fd;
1950
1951         fd = open("/dev/tty", O_RDWR|O_NOCTTY|O_NDELAY|O_CLOEXEC);
1952         if (fd < 0)
1953                 return -errno;
1954
1955         /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
1956          * by our own TIOCNOTTY */
1957         assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
1958
1959         if (ioctl(fd, TIOCNOTTY) < 0)
1960                 r = -errno;
1961
1962         assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
1963
1964         return r;
1965 }
1966
1967 int sigaction_many(const struct sigaction *sa, ...) {
1968         va_list ap;
1969         int r = 0, sig;
1970
1971         va_start(ap, sa);
1972         while ((sig = va_arg(ap, int)) > 0)
1973                 if (sigaction(sig, sa, NULL) < 0)
1974                         r = -errno;
1975         va_end(ap);
1976
1977         return r;
1978 }
1979
1980 int ignore_signals(int sig, ...) {
1981         struct sigaction sa = {
1982                 .sa_handler = SIG_IGN,
1983                 .sa_flags = SA_RESTART,
1984         };
1985         va_list ap;
1986         int r = 0;
1987
1988
1989         if (sigaction(sig, &sa, NULL) < 0)
1990                 r = -errno;
1991
1992         va_start(ap, sig);
1993         while ((sig = va_arg(ap, int)) > 0)
1994                 if (sigaction(sig, &sa, NULL) < 0)
1995                         r = -errno;
1996         va_end(ap);
1997
1998         return r;
1999 }
2000
2001 int default_signals(int sig, ...) {
2002         struct sigaction sa = {
2003                 .sa_handler = SIG_DFL,
2004                 .sa_flags = SA_RESTART,
2005         };
2006         va_list ap;
2007         int r = 0;
2008
2009         if (sigaction(sig, &sa, NULL) < 0)
2010                 r = -errno;
2011
2012         va_start(ap, sig);
2013         while ((sig = va_arg(ap, int)) > 0)
2014                 if (sigaction(sig, &sa, NULL) < 0)
2015                         r = -errno;
2016         va_end(ap);
2017
2018         return r;
2019 }
2020
2021 int close_pipe(int p[]) {
2022         int a = 0, b = 0;
2023
2024         assert(p);
2025
2026         if (p[0] >= 0) {
2027                 a = close_nointr(p[0]);
2028                 p[0] = -1;
2029         }
2030
2031         if (p[1] >= 0) {
2032                 b = close_nointr(p[1]);
2033                 p[1] = -1;
2034         }
2035
2036         return a < 0 ? a : b;
2037 }
2038
2039 ssize_t loop_read(int fd, void *buf, size_t nbytes, bool do_poll) {
2040         uint8_t *p = buf;
2041         ssize_t n = 0;
2042
2043         assert(fd >= 0);
2044         assert(buf);
2045
2046         while (nbytes > 0) {
2047                 ssize_t k;
2048
2049                 k = read(fd, p, nbytes);
2050                 if (k < 0 && errno == EINTR)
2051                         continue;
2052
2053                 if (k < 0 && errno == EAGAIN && do_poll) {
2054
2055                         /* We knowingly ignore any return value here,
2056                          * and expect that any error/EOF is reported
2057                          * via read() */
2058
2059                         fd_wait_for_event(fd, POLLIN, (usec_t) -1);
2060                         continue;
2061                 }
2062
2063                 if (k <= 0)
2064                         return n > 0 ? n : (k < 0 ? -errno : 0);
2065
2066                 p += k;
2067                 nbytes -= k;
2068                 n += k;
2069         }
2070
2071         return n;
2072 }
2073
2074 ssize_t loop_write(int fd, const void *buf, size_t nbytes, bool do_poll) {
2075         const uint8_t *p = buf;
2076         ssize_t n = 0;
2077
2078         assert(fd >= 0);
2079         assert(buf);
2080
2081         while (nbytes > 0) {
2082                 ssize_t k;
2083
2084                 k = write(fd, p, nbytes);
2085                 if (k < 0 && errno == EINTR)
2086                         continue;
2087
2088                 if (k < 0 && errno == EAGAIN && do_poll) {
2089
2090                         /* We knowingly ignore any return value here,
2091                          * and expect that any error/EOF is reported
2092                          * via write() */
2093
2094                         fd_wait_for_event(fd, POLLOUT, (usec_t) -1);
2095                         continue;
2096                 }
2097
2098                 if (k <= 0)
2099                         return n > 0 ? n : (k < 0 ? -errno : 0);
2100
2101                 p += k;
2102                 nbytes -= k;
2103                 n += k;
2104         }
2105
2106         return n;
2107 }
2108
2109 int parse_bytes(const char *t, off_t *bytes) {
2110         static const struct {
2111                 const char *suffix;
2112                 unsigned long long factor;
2113         } table[] = {
2114                 { "B", 1 },
2115                 { "K", 1024ULL },
2116                 { "M", 1024ULL*1024ULL },
2117                 { "G", 1024ULL*1024ULL*1024ULL },
2118                 { "T", 1024ULL*1024ULL*1024ULL*1024ULL },
2119                 { "P", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2120                 { "E", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2121                 { "", 1 },
2122         };
2123
2124         const char *p;
2125         unsigned long long r = 0;
2126
2127         assert(t);
2128         assert(bytes);
2129
2130         p = t;
2131         do {
2132                 long long l;
2133                 char *e;
2134                 unsigned i;
2135
2136                 errno = 0;
2137                 l = strtoll(p, &e, 10);
2138
2139                 if (errno > 0)
2140                         return -errno;
2141
2142                 if (l < 0)
2143                         return -ERANGE;
2144
2145                 if (e == p)
2146                         return -EINVAL;
2147
2148                 e += strspn(e, WHITESPACE);
2149
2150                 for (i = 0; i < ELEMENTSOF(table); i++)
2151                         if (startswith(e, table[i].suffix)) {
2152                                 unsigned long long tmp;
2153                                 if ((unsigned long long) l > ULLONG_MAX / table[i].factor)
2154                                         return -ERANGE;
2155                                 tmp = l * table[i].factor;
2156                                 if (tmp > ULLONG_MAX - r)
2157                                         return -ERANGE;
2158
2159                                 r += tmp;
2160                                 if ((unsigned long long) (off_t) r != r)
2161                                         return -ERANGE;
2162
2163                                 p = e + strlen(table[i].suffix);
2164                                 break;
2165                         }
2166
2167                 if (i >= ELEMENTSOF(table))
2168                         return -EINVAL;
2169
2170         } while (*p);
2171
2172         *bytes = r;
2173
2174         return 0;
2175 }
2176
2177 int make_stdio(int fd) {
2178         int r, s, t;
2179
2180         assert(fd >= 0);
2181
2182         r = dup3(fd, STDIN_FILENO, 0);
2183         s = dup3(fd, STDOUT_FILENO, 0);
2184         t = dup3(fd, STDERR_FILENO, 0);
2185
2186         if (fd >= 3)
2187                 close_nointr_nofail(fd);
2188
2189         if (r < 0 || s < 0 || t < 0)
2190                 return -errno;
2191
2192         /* We rely here that the new fd has O_CLOEXEC not set */
2193
2194         return 0;
2195 }
2196
2197 int make_null_stdio(void) {
2198         int null_fd;
2199
2200         null_fd = open("/dev/null", O_RDWR|O_NOCTTY);
2201         if (null_fd < 0)
2202                 return -errno;
2203
2204         return make_stdio(null_fd);
2205 }
2206
2207 bool is_device_path(const char *path) {
2208
2209         /* Returns true on paths that refer to a device, either in
2210          * sysfs or in /dev */
2211
2212         return
2213                 path_startswith(path, "/dev/") ||
2214                 path_startswith(path, "/sys/");
2215 }
2216
2217 int dir_is_empty(const char *path) {
2218         _cleanup_closedir_ DIR *d;
2219
2220         d = opendir(path);
2221         if (!d)
2222                 return -errno;
2223
2224         for (;;) {
2225                 struct dirent *de;
2226
2227                 errno = 0;
2228                 de = readdir(d);
2229                 if (!de && errno != 0)
2230                         return -errno;
2231
2232                 if (!de)
2233                         return 1;
2234
2235                 if (!ignore_file(de->d_name))
2236                         return 0;
2237         }
2238 }
2239
2240 char* dirname_malloc(const char *path) {
2241         char *d, *dir, *dir2;
2242
2243         d = strdup(path);
2244         if (!d)
2245                 return NULL;
2246         dir = dirname(d);
2247         assert(dir);
2248
2249         if (dir != d) {
2250                 dir2 = strdup(dir);
2251                 free(d);
2252                 return dir2;
2253         }
2254
2255         return dir;
2256 }
2257
2258 void random_bytes(void *p, size_t n) {
2259         static bool srand_called = false;
2260         _cleanup_close_ int fd;
2261         ssize_t k;
2262         uint8_t *q;
2263
2264         fd = open("/dev/urandom", O_RDONLY|O_CLOEXEC|O_NOCTTY);
2265         if (fd < 0)
2266                 goto fallback;
2267
2268         k = loop_read(fd, p, n, true);
2269         if (k < 0 || (size_t) k != n)
2270                 goto fallback;
2271
2272         return;
2273
2274 fallback:
2275
2276         if (!srand_called) {
2277
2278 #ifdef HAVE_SYS_AUXV_H
2279                 /* The kernel provides us with a bit of entropy in
2280                  * auxv, so let's try to make use of that to seed the
2281                  * pseudo-random generator. It's better than
2282                  * nothing... */
2283
2284                 void *auxv;
2285
2286                 auxv = (void*) getauxval(AT_RANDOM);
2287                 if (auxv)
2288                         srand(*(unsigned*) auxv);
2289                 else
2290 #endif
2291                         srand(time(NULL) + gettid());
2292
2293                 srand_called = true;
2294         }
2295
2296         /* If some idiot made /dev/urandom unavailable to us, he'll
2297          * get a PRNG instead. */
2298         for (q = p; q < (uint8_t*) p + n; q ++)
2299                 *q = rand();
2300 }
2301
2302 void rename_process(const char name[8]) {
2303         assert(name);
2304
2305         /* This is a like a poor man's setproctitle(). It changes the
2306          * comm field, argv[0], and also the glibc's internally used
2307          * name of the process. For the first one a limit of 16 chars
2308          * applies, to the second one usually one of 10 (i.e. length
2309          * of "/sbin/init"), to the third one one of 7 (i.e. length of
2310          * "systemd"). If you pass a longer string it will be
2311          * truncated */
2312
2313         prctl(PR_SET_NAME, name);
2314
2315         if (program_invocation_name)
2316                 strncpy(program_invocation_name, name, strlen(program_invocation_name));
2317
2318         if (saved_argc > 0) {
2319                 int i;
2320
2321                 if (saved_argv[0])
2322                         strncpy(saved_argv[0], name, strlen(saved_argv[0]));
2323
2324                 for (i = 1; i < saved_argc; i++) {
2325                         if (!saved_argv[i])
2326                                 break;
2327
2328                         memset(saved_argv[i], 0, strlen(saved_argv[i]));
2329                 }
2330         }
2331 }
2332
2333 void sigset_add_many(sigset_t *ss, ...) {
2334         va_list ap;
2335         int sig;
2336
2337         assert(ss);
2338
2339         va_start(ap, ss);
2340         while ((sig = va_arg(ap, int)) > 0)
2341                 assert_se(sigaddset(ss, sig) == 0);
2342         va_end(ap);
2343 }
2344
2345 char* gethostname_malloc(void) {
2346         struct utsname u;
2347
2348         assert_se(uname(&u) >= 0);
2349
2350         if (!isempty(u.nodename) && !streq(u.nodename, "(none)"))
2351                 return strdup(u.nodename);
2352
2353         return strdup(u.sysname);
2354 }
2355
2356 bool hostname_is_set(void) {
2357         struct utsname u;
2358
2359         assert_se(uname(&u) >= 0);
2360
2361         return !isempty(u.nodename) && !streq(u.nodename, "(none)");
2362 }
2363
2364 static char *lookup_uid(uid_t uid) {
2365         long bufsize;
2366         char *name;
2367         _cleanup_free_ char *buf = NULL;
2368         struct passwd pwbuf, *pw = NULL;
2369
2370         /* Shortcut things to avoid NSS lookups */
2371         if (uid == 0)
2372                 return strdup("root");
2373
2374         bufsize = sysconf(_SC_GETPW_R_SIZE_MAX);
2375         if (bufsize <= 0)
2376                 bufsize = 4096;
2377
2378         buf = malloc(bufsize);
2379         if (!buf)
2380                 return NULL;
2381
2382         if (getpwuid_r(uid, &pwbuf, buf, bufsize, &pw) == 0 && pw)
2383                 return strdup(pw->pw_name);
2384
2385         if (asprintf(&name, "%lu", (unsigned long) uid) < 0)
2386                 return NULL;
2387
2388         return name;
2389 }
2390
2391 char* getlogname_malloc(void) {
2392         uid_t uid;
2393         struct stat st;
2394
2395         if (isatty(STDIN_FILENO) && fstat(STDIN_FILENO, &st) >= 0)
2396                 uid = st.st_uid;
2397         else
2398                 uid = getuid();
2399
2400         return lookup_uid(uid);
2401 }
2402
2403 char *getusername_malloc(void) {
2404         const char *e;
2405
2406         e = getenv("USER");
2407         if (e)
2408                 return strdup(e);
2409
2410         return lookup_uid(getuid());
2411 }
2412
2413 int getttyname_malloc(int fd, char **r) {
2414         char path[PATH_MAX], *c;
2415         int k;
2416
2417         assert(r);
2418
2419         k = ttyname_r(fd, path, sizeof(path));
2420         if (k > 0)
2421                 return -k;
2422
2423         char_array_0(path);
2424
2425         c = strdup(startswith(path, "/dev/") ? path + 5 : path);
2426         if (!c)
2427                 return -ENOMEM;
2428
2429         *r = c;
2430         return 0;
2431 }
2432
2433 int getttyname_harder(int fd, char **r) {
2434         int k;
2435         char *s;
2436
2437         k = getttyname_malloc(fd, &s);
2438         if (k < 0)
2439                 return k;
2440
2441         if (streq(s, "tty")) {
2442                 free(s);
2443                 return get_ctty(0, NULL, r);
2444         }
2445
2446         *r = s;
2447         return 0;
2448 }
2449
2450 int get_ctty_devnr(pid_t pid, dev_t *d) {
2451         int r;
2452         _cleanup_free_ char *line = NULL;
2453         const char *p;
2454         unsigned long ttynr;
2455
2456         assert(pid >= 0);
2457
2458         p = procfs_file_alloca(pid, "stat");
2459         r = read_one_line_file(p, &line);
2460         if (r < 0)
2461                 return r;
2462
2463         p = strrchr(line, ')');
2464         if (!p)
2465                 return -EIO;
2466
2467         p++;
2468
2469         if (sscanf(p, " "
2470                    "%*c "  /* state */
2471                    "%*d "  /* ppid */
2472                    "%*d "  /* pgrp */
2473                    "%*d "  /* session */
2474                    "%lu ", /* ttynr */
2475                    &ttynr) != 1)
2476                 return -EIO;
2477
2478         if (major(ttynr) == 0 && minor(ttynr) == 0)
2479                 return -ENOENT;
2480
2481         if (d)
2482                 *d = (dev_t) ttynr;
2483
2484         return 0;
2485 }
2486
2487 int get_ctty(pid_t pid, dev_t *_devnr, char **r) {
2488         int k;
2489         char fn[sizeof("/dev/char/")-1 + 2*DECIMAL_STR_MAX(unsigned) + 1 + 1], *s, *b, *p;
2490         dev_t devnr;
2491
2492         assert(r);
2493
2494         k = get_ctty_devnr(pid, &devnr);
2495         if (k < 0)
2496                 return k;
2497
2498         snprintf(fn, sizeof(fn), "/dev/char/%u:%u", major(devnr), minor(devnr));
2499
2500         k = readlink_malloc(fn, &s);
2501         if (k < 0) {
2502
2503                 if (k != -ENOENT)
2504                         return k;
2505
2506                 /* This is an ugly hack */
2507                 if (major(devnr) == 136) {
2508                         if (asprintf(&b, "pts/%lu", (unsigned long) minor(devnr)) < 0)
2509                                 return -ENOMEM;
2510
2511                         *r = b;
2512                         if (_devnr)
2513                                 *_devnr = devnr;
2514
2515                         return 0;
2516                 }
2517
2518                 /* Probably something like the ptys which have no
2519                  * symlink in /dev/char. Let's return something
2520                  * vaguely useful. */
2521
2522                 b = strdup(fn + 5);
2523                 if (!b)
2524                         return -ENOMEM;
2525
2526                 *r = b;
2527                 if (_devnr)
2528                         *_devnr = devnr;
2529
2530                 return 0;
2531         }
2532
2533         if (startswith(s, "/dev/"))
2534                 p = s + 5;
2535         else if (startswith(s, "../"))
2536                 p = s + 3;
2537         else
2538                 p = s;
2539
2540         b = strdup(p);
2541         free(s);
2542
2543         if (!b)
2544                 return -ENOMEM;
2545
2546         *r = b;
2547         if (_devnr)
2548                 *_devnr = devnr;
2549
2550         return 0;
2551 }
2552
2553 int rm_rf_children_dangerous(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
2554         DIR *d;
2555         int ret = 0;
2556
2557         assert(fd >= 0);
2558
2559         /* This returns the first error we run into, but nevertheless
2560          * tries to go on. This closes the passed fd. */
2561
2562         d = fdopendir(fd);
2563         if (!d) {
2564                 close_nointr_nofail(fd);
2565
2566                 return errno == ENOENT ? 0 : -errno;
2567         }
2568
2569         for (;;) {
2570                 struct dirent *de;
2571                 bool is_dir, keep_around;
2572                 struct stat st;
2573                 int r;
2574
2575                 errno = 0;
2576                 de = readdir(d);
2577                 if (!de && errno != 0) {
2578                         if (ret == 0)
2579                                 ret = -errno;
2580                         break;
2581                 }
2582
2583                 if (!de)
2584                         break;
2585
2586                 if (streq(de->d_name, ".") || streq(de->d_name, ".."))
2587                         continue;
2588
2589                 if (de->d_type == DT_UNKNOWN ||
2590                     honour_sticky ||
2591                     (de->d_type == DT_DIR && root_dev)) {
2592                         if (fstatat(fd, de->d_name, &st, AT_SYMLINK_NOFOLLOW) < 0) {
2593                                 if (ret == 0 && errno != ENOENT)
2594                                         ret = -errno;
2595                                 continue;
2596                         }
2597
2598                         is_dir = S_ISDIR(st.st_mode);
2599                         keep_around =
2600                                 honour_sticky &&
2601                                 (st.st_uid == 0 || st.st_uid == getuid()) &&
2602                                 (st.st_mode & S_ISVTX);
2603                 } else {
2604                         is_dir = de->d_type == DT_DIR;
2605                         keep_around = false;
2606                 }
2607
2608                 if (is_dir) {
2609                         int subdir_fd;
2610
2611                         /* if root_dev is set, remove subdirectories only, if device is same as dir */
2612                         if (root_dev && st.st_dev != root_dev->st_dev)
2613                                 continue;
2614
2615                         subdir_fd = openat(fd, de->d_name,
2616                                            O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
2617                         if (subdir_fd < 0) {
2618                                 if (ret == 0 && errno != ENOENT)
2619                                         ret = -errno;
2620                                 continue;
2621                         }
2622
2623                         r = rm_rf_children_dangerous(subdir_fd, only_dirs, honour_sticky, root_dev);
2624                         if (r < 0 && ret == 0)
2625                                 ret = r;
2626
2627                         if (!keep_around)
2628                                 if (unlinkat(fd, de->d_name, AT_REMOVEDIR) < 0) {
2629                                         if (ret == 0 && errno != ENOENT)
2630                                                 ret = -errno;
2631                                 }
2632
2633                 } else if (!only_dirs && !keep_around) {
2634
2635                         if (unlinkat(fd, de->d_name, 0) < 0) {
2636                                 if (ret == 0 && errno != ENOENT)
2637                                         ret = -errno;
2638                         }
2639                 }
2640         }
2641
2642         closedir(d);
2643
2644         return ret;
2645 }
2646
2647 _pure_ static int is_temporary_fs(struct statfs *s) {
2648         assert(s);
2649
2650         return F_TYPE_EQUAL(s->f_type, TMPFS_MAGIC) ||
2651                F_TYPE_EQUAL(s->f_type, RAMFS_MAGIC);
2652 }
2653
2654 int rm_rf_children(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
2655         struct statfs s;
2656
2657         assert(fd >= 0);
2658
2659         if (fstatfs(fd, &s) < 0) {
2660                 close_nointr_nofail(fd);
2661                 return -errno;
2662         }
2663
2664         /* We refuse to clean disk file systems with this call. This
2665          * is extra paranoia just to be sure we never ever remove
2666          * non-state data */
2667         if (!is_temporary_fs(&s)) {
2668                 log_error("Attempted to remove disk file system, and we can't allow that.");
2669                 close_nointr_nofail(fd);
2670                 return -EPERM;
2671         }
2672
2673         return rm_rf_children_dangerous(fd, only_dirs, honour_sticky, root_dev);
2674 }
2675
2676 static int rm_rf_internal(const char *path, bool only_dirs, bool delete_root, bool honour_sticky, bool dangerous) {
2677         int fd, r;
2678         struct statfs s;
2679
2680         assert(path);
2681
2682         /* We refuse to clean the root file system with this
2683          * call. This is extra paranoia to never cause a really
2684          * seriously broken system. */
2685         if (path_equal(path, "/")) {
2686                 log_error("Attempted to remove entire root file system, and we can't allow that.");
2687                 return -EPERM;
2688         }
2689
2690         fd = open(path, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
2691         if (fd < 0) {
2692
2693                 if (errno != ENOTDIR)
2694                         return -errno;
2695
2696                 if (!dangerous) {
2697                         if (statfs(path, &s) < 0)
2698                                 return -errno;
2699
2700                         if (!is_temporary_fs(&s)) {
2701                                 log_error("Attempted to remove disk file system, and we can't allow that.");
2702                                 return -EPERM;
2703                         }
2704                 }
2705
2706                 if (delete_root && !only_dirs)
2707                         if (unlink(path) < 0 && errno != ENOENT)
2708                                 return -errno;
2709
2710                 return 0;
2711         }
2712
2713         if (!dangerous) {
2714                 if (fstatfs(fd, &s) < 0) {
2715                         close_nointr_nofail(fd);
2716                         return -errno;
2717                 }
2718
2719                 if (!is_temporary_fs(&s)) {
2720                         log_error("Attempted to remove disk file system, and we can't allow that.");
2721                         close_nointr_nofail(fd);
2722                         return -EPERM;
2723                 }
2724         }
2725
2726         r = rm_rf_children_dangerous(fd, only_dirs, honour_sticky, NULL);
2727         if (delete_root) {
2728
2729                 if (honour_sticky && file_is_priv_sticky(path) > 0)
2730                         return r;
2731
2732                 if (rmdir(path) < 0 && errno != ENOENT) {
2733                         if (r == 0)
2734                                 r = -errno;
2735                 }
2736         }
2737
2738         return r;
2739 }
2740
2741 int rm_rf(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
2742         return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, false);
2743 }
2744
2745 int rm_rf_dangerous(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
2746         return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, true);
2747 }
2748
2749 int chmod_and_chown(const char *path, mode_t mode, uid_t uid, gid_t gid) {
2750         assert(path);
2751
2752         /* Under the assumption that we are running privileged we
2753          * first change the access mode and only then hand out
2754          * ownership to avoid a window where access is too open. */
2755
2756         if (mode != (mode_t) -1)
2757                 if (chmod(path, mode) < 0)
2758                         return -errno;
2759
2760         if (uid != (uid_t) -1 || gid != (gid_t) -1)
2761                 if (chown(path, uid, gid) < 0)
2762                         return -errno;
2763
2764         return 0;
2765 }
2766
2767 int fchmod_and_fchown(int fd, mode_t mode, uid_t uid, gid_t gid) {
2768         assert(fd >= 0);
2769
2770         /* Under the assumption that we are running privileged we
2771          * first change the access mode and only then hand out
2772          * ownership to avoid a window where access is too open. */
2773
2774         if (mode != (mode_t) -1)
2775                 if (fchmod(fd, mode) < 0)
2776                         return -errno;
2777
2778         if (uid != (uid_t) -1 || gid != (gid_t) -1)
2779                 if (fchown(fd, uid, gid) < 0)
2780                         return -errno;
2781
2782         return 0;
2783 }
2784
2785 cpu_set_t* cpu_set_malloc(unsigned *ncpus) {
2786         cpu_set_t *r;
2787         unsigned n = 1024;
2788
2789         /* Allocates the cpuset in the right size */
2790
2791         for (;;) {
2792                 if (!(r = CPU_ALLOC(n)))
2793                         return NULL;
2794
2795                 if (sched_getaffinity(0, CPU_ALLOC_SIZE(n), r) >= 0) {
2796                         CPU_ZERO_S(CPU_ALLOC_SIZE(n), r);
2797
2798                         if (ncpus)
2799                                 *ncpus = n;
2800
2801                         return r;
2802                 }
2803
2804                 CPU_FREE(r);
2805
2806                 if (errno != EINVAL)
2807                         return NULL;
2808
2809                 n *= 2;
2810         }
2811 }
2812
2813 int status_vprintf(const char *status, bool ellipse, bool ephemeral, const char *format, va_list ap) {
2814         static const char status_indent[] = "         "; /* "[" STATUS "] " */
2815         _cleanup_free_ char *s = NULL;
2816         _cleanup_close_ int fd = -1;
2817         struct iovec iovec[6] = {};
2818         int n = 0;
2819         static bool prev_ephemeral;
2820
2821         assert(format);
2822
2823         /* This is independent of logging, as status messages are
2824          * optional and go exclusively to the console. */
2825
2826         if (vasprintf(&s, format, ap) < 0)
2827                 return log_oom();
2828
2829         fd = open_terminal("/dev/console", O_WRONLY|O_NOCTTY|O_CLOEXEC);
2830         if (fd < 0)
2831                 return fd;
2832
2833         if (ellipse) {
2834                 char *e;
2835                 size_t emax, sl;
2836                 int c;
2837
2838                 c = fd_columns(fd);
2839                 if (c <= 0)
2840                         c = 80;
2841
2842                 sl = status ? sizeof(status_indent)-1 : 0;
2843
2844                 emax = c - sl - 1;
2845                 if (emax < 3)
2846                         emax = 3;
2847
2848                 e = ellipsize(s, emax, 75);
2849                 if (e) {
2850                         free(s);
2851                         s = e;
2852                 }
2853         }
2854
2855         if (prev_ephemeral)
2856                 IOVEC_SET_STRING(iovec[n++], "\r" ANSI_ERASE_TO_END_OF_LINE);
2857         prev_ephemeral = ephemeral;
2858
2859         if (status) {
2860                 if (!isempty(status)) {
2861                         IOVEC_SET_STRING(iovec[n++], "[");
2862                         IOVEC_SET_STRING(iovec[n++], status);
2863                         IOVEC_SET_STRING(iovec[n++], "] ");
2864                 } else
2865                         IOVEC_SET_STRING(iovec[n++], status_indent);
2866         }
2867
2868         IOVEC_SET_STRING(iovec[n++], s);
2869         if (!ephemeral)
2870                 IOVEC_SET_STRING(iovec[n++], "\n");
2871
2872         if (writev(fd, iovec, n) < 0)
2873                 return -errno;
2874
2875         return 0;
2876 }
2877
2878 int status_printf(const char *status, bool ellipse, bool ephemeral, const char *format, ...) {
2879         va_list ap;
2880         int r;
2881
2882         assert(format);
2883
2884         va_start(ap, format);
2885         r = status_vprintf(status, ellipse, ephemeral, format, ap);
2886         va_end(ap);
2887
2888         return r;
2889 }
2890
2891 int status_welcome(void) {
2892         _cleanup_free_ char *pretty_name = NULL, *ansi_color = NULL;
2893         int r;
2894
2895         r = parse_env_file("/etc/os-release", NEWLINE,
2896                            "PRETTY_NAME", &pretty_name,
2897                            "ANSI_COLOR", &ansi_color,
2898                            NULL);
2899
2900         if (r < 0 && r != -ENOENT)
2901                 log_warning("Failed to read /etc/os-release: %s", strerror(-r));
2902
2903         return status_printf(NULL, false, false,
2904                              "\nWelcome to \x1B[%sm%s\x1B[0m!\n",
2905                              isempty(ansi_color) ? "1" : ansi_color,
2906                              isempty(pretty_name) ? "Linux" : pretty_name);
2907 }
2908
2909 char *replace_env(const char *format, char **env) {
2910         enum {
2911                 WORD,
2912                 CURLY,
2913                 VARIABLE
2914         } state = WORD;
2915
2916         const char *e, *word = format;
2917         char *r = NULL, *k;
2918
2919         assert(format);
2920
2921         for (e = format; *e; e ++) {
2922
2923                 switch (state) {
2924
2925                 case WORD:
2926                         if (*e == '$')
2927                                 state = CURLY;
2928                         break;
2929
2930                 case CURLY:
2931                         if (*e == '{') {
2932                                 if (!(k = strnappend(r, word, e-word-1)))
2933                                         goto fail;
2934
2935                                 free(r);
2936                                 r = k;
2937
2938                                 word = e-1;
2939                                 state = VARIABLE;
2940
2941                         } else if (*e == '$') {
2942                                 if (!(k = strnappend(r, word, e-word)))
2943                                         goto fail;
2944
2945                                 free(r);
2946                                 r = k;
2947
2948                                 word = e+1;
2949                                 state = WORD;
2950                         } else
2951                                 state = WORD;
2952                         break;
2953
2954                 case VARIABLE:
2955                         if (*e == '}') {
2956                                 const char *t;
2957
2958                                 t = strempty(strv_env_get_n(env, word+2, e-word-2));
2959
2960                                 k = strappend(r, t);
2961                                 if (!k)
2962                                         goto fail;
2963
2964                                 free(r);
2965                                 r = k;
2966
2967                                 word = e+1;
2968                                 state = WORD;
2969                         }
2970                         break;
2971                 }
2972         }
2973
2974         if (!(k = strnappend(r, word, e-word)))
2975                 goto fail;
2976
2977         free(r);
2978         return k;
2979
2980 fail:
2981         free(r);
2982         return NULL;
2983 }
2984
2985 char **replace_env_argv(char **argv, char **env) {
2986         char **r, **i;
2987         unsigned k = 0, l = 0;
2988
2989         l = strv_length(argv);
2990
2991         if (!(r = new(char*, l+1)))
2992                 return NULL;
2993
2994         STRV_FOREACH(i, argv) {
2995
2996                 /* If $FOO appears as single word, replace it by the split up variable */
2997                 if ((*i)[0] == '$' && (*i)[1] != '{') {
2998                         char *e;
2999                         char **w, **m;
3000                         unsigned q;
3001
3002                         e = strv_env_get(env, *i+1);
3003                         if (e) {
3004
3005                                 if (!(m = strv_split_quoted(e))) {
3006                                         r[k] = NULL;
3007                                         strv_free(r);
3008                                         return NULL;
3009                                 }
3010                         } else
3011                                 m = NULL;
3012
3013                         q = strv_length(m);
3014                         l = l + q - 1;
3015
3016                         if (!(w = realloc(r, sizeof(char*) * (l+1)))) {
3017                                 r[k] = NULL;
3018                                 strv_free(r);
3019                                 strv_free(m);
3020                                 return NULL;
3021                         }
3022
3023                         r = w;
3024                         if (m) {
3025                                 memcpy(r + k, m, q * sizeof(char*));
3026                                 free(m);
3027                         }
3028
3029                         k += q;
3030                         continue;
3031                 }
3032
3033                 /* If ${FOO} appears as part of a word, replace it by the variable as-is */
3034                 if (!(r[k++] = replace_env(*i, env))) {
3035                         strv_free(r);
3036                         return NULL;
3037                 }
3038         }
3039
3040         r[k] = NULL;
3041         return r;
3042 }
3043
3044 int fd_columns(int fd) {
3045         struct winsize ws = {};
3046
3047         if (ioctl(fd, TIOCGWINSZ, &ws) < 0)
3048                 return -errno;
3049
3050         if (ws.ws_col <= 0)
3051                 return -EIO;
3052
3053         return ws.ws_col;
3054 }
3055
3056 unsigned columns(void) {
3057         const char *e;
3058         int c;
3059
3060         if (_likely_(cached_columns > 0))
3061                 return cached_columns;
3062
3063         c = 0;
3064         e = getenv("COLUMNS");
3065         if (e)
3066                 safe_atoi(e, &c);
3067
3068         if (c <= 0)
3069                 c = fd_columns(STDOUT_FILENO);
3070
3071         if (c <= 0)
3072                 c = 80;
3073
3074         cached_columns = c;
3075         return c;
3076 }
3077
3078 int fd_lines(int fd) {
3079         struct winsize ws = {};
3080
3081         if (ioctl(fd, TIOCGWINSZ, &ws) < 0)
3082                 return -errno;
3083
3084         if (ws.ws_row <= 0)
3085                 return -EIO;
3086
3087         return ws.ws_row;
3088 }
3089
3090 unsigned lines(void) {
3091         const char *e;
3092         unsigned l;
3093
3094         if (_likely_(cached_lines > 0))
3095                 return cached_lines;
3096
3097         l = 0;
3098         e = getenv("LINES");
3099         if (e)
3100                 safe_atou(e, &l);
3101
3102         if (l <= 0)
3103                 l = fd_lines(STDOUT_FILENO);
3104
3105         if (l <= 0)
3106                 l = 24;
3107
3108         cached_lines = l;
3109         return cached_lines;
3110 }
3111
3112 /* intended to be used as a SIGWINCH sighandler */
3113 void columns_lines_cache_reset(int signum) {
3114         cached_columns = 0;
3115         cached_lines = 0;
3116 }
3117
3118 bool on_tty(void) {
3119         static int cached_on_tty = -1;
3120
3121         if (_unlikely_(cached_on_tty < 0))
3122                 cached_on_tty = isatty(STDOUT_FILENO) > 0;
3123
3124         return cached_on_tty;
3125 }
3126
3127 int running_in_chroot(void) {
3128         struct stat a = {}, b = {};
3129
3130         /* Only works as root */
3131         if (stat("/proc/1/root", &a) < 0)
3132                 return -errno;
3133
3134         if (stat("/", &b) < 0)
3135                 return -errno;
3136
3137         return
3138                 a.st_dev != b.st_dev ||
3139                 a.st_ino != b.st_ino;
3140 }
3141
3142 static char *ascii_ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3143         size_t x;
3144         char *r;
3145
3146         assert(s);
3147         assert(percent <= 100);
3148         assert(new_length >= 3);
3149
3150         if (old_length <= 3 || old_length <= new_length)
3151                 return strndup(s, old_length);
3152
3153         r = new0(char, new_length+1);
3154         if (!r)
3155                 return NULL;
3156
3157         x = (new_length * percent) / 100;
3158
3159         if (x > new_length - 3)
3160                 x = new_length - 3;
3161
3162         memcpy(r, s, x);
3163         r[x] = '.';
3164         r[x+1] = '.';
3165         r[x+2] = '.';
3166         memcpy(r + x + 3,
3167                s + old_length - (new_length - x - 3),
3168                new_length - x - 3);
3169
3170         return r;
3171 }
3172
3173 char *ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3174         size_t x;
3175         char *e;
3176         const char *i, *j;
3177         unsigned k, len, len2;
3178
3179         assert(s);
3180         assert(percent <= 100);
3181         assert(new_length >= 3);
3182
3183         /* if no multibyte characters use ascii_ellipsize_mem for speed */
3184         if (ascii_is_valid(s))
3185                 return ascii_ellipsize_mem(s, old_length, new_length, percent);
3186
3187         if (old_length <= 3 || old_length <= new_length)
3188                 return strndup(s, old_length);
3189
3190         x = (new_length * percent) / 100;
3191
3192         if (x > new_length - 3)
3193                 x = new_length - 3;
3194
3195         k = 0;
3196         for (i = s; k < x && i < s + old_length; i = utf8_next_char(i)) {
3197                 int c;
3198
3199                 c = utf8_encoded_to_unichar(i);
3200                 if (c < 0)
3201                         return NULL;
3202                 k += unichar_iswide(c) ? 2 : 1;
3203         }
3204
3205         if (k > x) /* last character was wide and went over quota */
3206                 x ++;
3207
3208         for (j = s + old_length; k < new_length && j > i; ) {
3209                 int c;
3210
3211                 j = utf8_prev_char(j);
3212                 c = utf8_encoded_to_unichar(j);
3213                 if (c < 0)
3214                         return NULL;
3215                 k += unichar_iswide(c) ? 2 : 1;
3216         }
3217         assert(i <= j);
3218
3219         /* we don't actually need to ellipsize */
3220         if (i == j)
3221                 return memdup(s, old_length + 1);
3222
3223         /* make space for ellipsis */
3224         j = utf8_next_char(j);
3225
3226         len = i - s;
3227         len2 = s + old_length - j;
3228         e = new(char, len + 3 + len2 + 1);
3229         if (!e)
3230                 return NULL;
3231
3232         /*
3233         printf("old_length=%zu new_length=%zu x=%zu len=%u len2=%u k=%u\n",
3234                old_length, new_length, x, len, len2, k);
3235         */
3236
3237         memcpy(e, s, len);
3238         e[len]   = 0xe2; /* tri-dot ellipsis: … */
3239         e[len + 1] = 0x80;
3240         e[len + 2] = 0xa6;
3241
3242         memcpy(e + len + 3, j, len2 + 1);
3243
3244         return e;
3245 }
3246
3247 char *ellipsize(const char *s, size_t length, unsigned percent) {
3248         return ellipsize_mem(s, strlen(s), length, percent);
3249 }
3250
3251 int touch(const char *path) {
3252         int fd;
3253
3254         assert(path);
3255
3256         /* This just opens the file for writing, ensuring it
3257          * exists. It doesn't call utimensat() the way /usr/bin/touch
3258          * does it. */
3259
3260         fd = open(path, O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY, 0644);
3261         if (fd < 0)
3262                 return -errno;
3263
3264         close_nointr_nofail(fd);
3265         return 0;
3266 }
3267
3268 char *unquote(const char *s, const char* quotes) {
3269         size_t l;
3270         assert(s);
3271
3272         /* This is rather stupid, simply removes the heading and
3273          * trailing quotes if there is one. Doesn't care about
3274          * escaping or anything. We should make this smarter one
3275          * day...*/
3276
3277         l = strlen(s);
3278         if (l < 2)
3279                 return strdup(s);
3280
3281         if (strchr(quotes, s[0]) && s[l-1] == s[0])
3282                 return strndup(s+1, l-2);
3283
3284         return strdup(s);
3285 }
3286
3287 char *normalize_env_assignment(const char *s) {
3288         _cleanup_free_ char *name = NULL, *value = NULL, *p = NULL;
3289         char *eq, *r;
3290
3291         eq = strchr(s, '=');
3292         if (!eq) {
3293                 char *t;
3294
3295                 r = strdup(s);
3296                 if (!r)
3297                         return NULL;
3298
3299                 t = strstrip(r);
3300                 if (t == r)
3301                         return r;
3302
3303                 memmove(r, t, strlen(t) + 1);
3304                 return r;
3305         }
3306
3307         name = strndup(s, eq - s);
3308         if (!name)
3309                 return NULL;
3310
3311         p = strdup(eq + 1);
3312         if (!p)
3313                 return NULL;
3314
3315         value = unquote(strstrip(p), QUOTES);
3316         if (!value)
3317                 return NULL;
3318
3319         if (asprintf(&r, "%s=%s", strstrip(name), value) < 0)
3320                 r = NULL;
3321
3322         return r;
3323 }
3324
3325 int wait_for_terminate(pid_t pid, siginfo_t *status) {
3326         siginfo_t dummy;
3327
3328         assert(pid >= 1);
3329
3330         if (!status)
3331                 status = &dummy;
3332
3333         for (;;) {
3334                 zero(*status);
3335
3336                 if (waitid(P_PID, pid, status, WEXITED) < 0) {
3337
3338                         if (errno == EINTR)
3339                                 continue;
3340
3341                         return -errno;
3342                 }
3343
3344                 return 0;
3345         }
3346 }
3347
3348 int wait_for_terminate_and_warn(const char *name, pid_t pid) {
3349         int r;
3350         siginfo_t status;
3351
3352         assert(name);
3353         assert(pid > 1);
3354
3355         r = wait_for_terminate(pid, &status);
3356         if (r < 0) {
3357                 log_warning("Failed to wait for %s: %s", name, strerror(-r));
3358                 return r;
3359         }
3360
3361         if (status.si_code == CLD_EXITED) {
3362                 if (status.si_status != 0) {
3363                         log_warning("%s failed with error code %i.", name, status.si_status);
3364                         return status.si_status;
3365                 }
3366
3367                 log_debug("%s succeeded.", name);
3368                 return 0;
3369
3370         } else if (status.si_code == CLD_KILLED ||
3371                    status.si_code == CLD_DUMPED) {
3372
3373                 log_warning("%s terminated by signal %s.", name, signal_to_string(status.si_status));
3374                 return -EPROTO;
3375         }
3376
3377         log_warning("%s failed due to unknown reason.", name);
3378         return -EPROTO;
3379 }
3380
3381 noreturn void freeze(void) {
3382
3383         /* Make sure nobody waits for us on a socket anymore */
3384         close_all_fds(NULL, 0);
3385
3386         sync();
3387
3388         for (;;)
3389                 pause();
3390 }
3391
3392 bool null_or_empty(struct stat *st) {
3393         assert(st);
3394
3395         if (S_ISREG(st->st_mode) && st->st_size <= 0)
3396                 return true;
3397
3398         if (S_ISCHR(st->st_mode) || S_ISBLK(st->st_mode))
3399                 return true;
3400
3401         return false;
3402 }
3403
3404 int null_or_empty_path(const char *fn) {
3405         struct stat st;
3406
3407         assert(fn);
3408
3409         if (stat(fn, &st) < 0)
3410                 return -errno;
3411
3412         return null_or_empty(&st);
3413 }
3414
3415 DIR *xopendirat(int fd, const char *name, int flags) {
3416         int nfd;
3417         DIR *d;
3418
3419         assert(!(flags & O_CREAT));
3420
3421         nfd = openat(fd, name, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|flags, 0);
3422         if (nfd < 0)
3423                 return NULL;
3424
3425         d = fdopendir(nfd);
3426         if (!d) {
3427                 close_nointr_nofail(nfd);
3428                 return NULL;
3429         }
3430
3431         return d;
3432 }
3433
3434 int signal_from_string_try_harder(const char *s) {
3435         int signo;
3436         assert(s);
3437
3438         signo = signal_from_string(s);
3439         if (signo <= 0)
3440                 if (startswith(s, "SIG"))
3441                         return signal_from_string(s+3);
3442
3443         return signo;
3444 }
3445
3446 static char *tag_to_udev_node(const char *tagvalue, const char *by) {
3447         _cleanup_free_ char *t = NULL, *u = NULL;
3448         char *dn;
3449         size_t enc_len;
3450
3451         u = unquote(tagvalue, "\"\'");
3452         if (u == NULL)
3453                 return NULL;
3454
3455         enc_len = strlen(u) * 4 + 1;
3456         t = new(char, enc_len);
3457         if (t == NULL)
3458                 return NULL;
3459
3460         if (encode_devnode_name(u, t, enc_len) < 0)
3461                 return NULL;
3462
3463         if (asprintf(&dn, "/dev/disk/by-%s/%s", by, t) < 0)
3464                 return NULL;
3465
3466         return dn;
3467 }
3468
3469 char *fstab_node_to_udev_node(const char *p) {
3470         assert(p);
3471
3472         if (startswith(p, "LABEL="))
3473                 return tag_to_udev_node(p+6, "label");
3474
3475         if (startswith(p, "UUID="))
3476                 return tag_to_udev_node(p+5, "uuid");
3477
3478         if (startswith(p, "PARTUUID="))
3479                 return tag_to_udev_node(p+9, "partuuid");
3480
3481         if (startswith(p, "PARTLABEL="))
3482                 return tag_to_udev_node(p+10, "partlabel");
3483
3484         return strdup(p);
3485 }
3486
3487 bool tty_is_vc(const char *tty) {
3488         assert(tty);
3489
3490         if (startswith(tty, "/dev/"))
3491                 tty += 5;
3492
3493         return vtnr_from_tty(tty) >= 0;
3494 }
3495
3496 bool tty_is_console(const char *tty) {
3497         assert(tty);
3498
3499         if (startswith(tty, "/dev/"))
3500                 tty += 5;
3501
3502         return streq(tty, "console");
3503 }
3504
3505 int vtnr_from_tty(const char *tty) {
3506         int i, r;
3507
3508         assert(tty);
3509
3510         if (startswith(tty, "/dev/"))
3511                 tty += 5;
3512
3513         if (!startswith(tty, "tty") )
3514                 return -EINVAL;
3515
3516         if (tty[3] < '0' || tty[3] > '9')
3517                 return -EINVAL;
3518
3519         r = safe_atoi(tty+3, &i);
3520         if (r < 0)
3521                 return r;
3522
3523         if (i < 0 || i > 63)
3524                 return -EINVAL;
3525
3526         return i;
3527 }
3528
3529 char *resolve_dev_console(char **active) {
3530         char *tty;
3531
3532         /* Resolve where /dev/console is pointing to, if /sys is actually ours
3533          * (i.e. not read-only-mounted which is a sign for container setups) */
3534
3535         if (path_is_read_only_fs("/sys") > 0)
3536                 return NULL;
3537
3538         if (read_one_line_file("/sys/class/tty/console/active", active) < 0)
3539                 return NULL;
3540
3541         /* If multiple log outputs are configured the last one is what
3542          * /dev/console points to */
3543         tty = strrchr(*active, ' ');
3544         if (tty)
3545                 tty++;
3546         else
3547                 tty = *active;
3548
3549         if (streq(tty, "tty0")) {
3550                 char *tmp;
3551
3552                 /* Get the active VC (e.g. tty1) */
3553                 if (read_one_line_file("/sys/class/tty/tty0/active", &tmp) >= 0) {
3554                         free(*active);
3555                         tty = *active = tmp;
3556                 }
3557         }
3558
3559         return tty;
3560 }
3561
3562 bool tty_is_vc_resolve(const char *tty) {
3563         _cleanup_free_ char *active = NULL;
3564
3565         assert(tty);
3566
3567         if (startswith(tty, "/dev/"))
3568                 tty += 5;
3569
3570         if (streq(tty, "console")) {
3571                 tty = resolve_dev_console(&active);
3572                 if (!tty)
3573                         return false;
3574         }
3575
3576         return tty_is_vc(tty);
3577 }
3578
3579 const char *default_term_for_tty(const char *tty) {
3580         assert(tty);
3581
3582         return tty_is_vc_resolve(tty) ? "TERM=linux" : "TERM=vt102";
3583 }
3584
3585 bool dirent_is_file(const struct dirent *de) {
3586         assert(de);
3587
3588         if (ignore_file(de->d_name))
3589                 return false;
3590
3591         if (de->d_type != DT_REG &&
3592             de->d_type != DT_LNK &&
3593             de->d_type != DT_UNKNOWN)
3594                 return false;
3595
3596         return true;
3597 }
3598
3599 bool dirent_is_file_with_suffix(const struct dirent *de, const char *suffix) {
3600         assert(de);
3601
3602         if (de->d_type != DT_REG &&
3603             de->d_type != DT_LNK &&
3604             de->d_type != DT_UNKNOWN)
3605                 return false;
3606
3607         if (ignore_file_allow_backup(de->d_name))
3608                 return false;
3609
3610         return endswith(de->d_name, suffix);
3611 }
3612
3613 void execute_directory(const char *directory, DIR *d, char *argv[]) {
3614         DIR *_d = NULL;
3615         struct dirent *de;
3616         Hashmap *pids = NULL;
3617
3618         assert(directory);
3619
3620         /* Executes all binaries in a directory in parallel and
3621          * waits for them to finish. */
3622
3623         if (!d) {
3624                 if (!(_d = opendir(directory))) {
3625
3626                         if (errno == ENOENT)
3627                                 return;
3628
3629                         log_error("Failed to enumerate directory %s: %m", directory);
3630                         return;
3631                 }
3632
3633                 d = _d;
3634         }
3635
3636         if (!(pids = hashmap_new(trivial_hash_func, trivial_compare_func))) {
3637                 log_error("Failed to allocate set.");
3638                 goto finish;
3639         }
3640
3641         while ((de = readdir(d))) {
3642                 char *path;
3643                 pid_t pid;
3644                 int k;
3645
3646                 if (!dirent_is_file(de))
3647                         continue;
3648
3649                 if (asprintf(&path, "%s/%s", directory, de->d_name) < 0) {
3650                         log_oom();
3651                         continue;
3652                 }
3653
3654                 if ((pid = fork()) < 0) {
3655                         log_error("Failed to fork: %m");
3656                         free(path);
3657                         continue;
3658                 }
3659
3660                 if (pid == 0) {
3661                         char *_argv[2];
3662                         /* Child */
3663
3664                         if (!argv) {
3665                                 _argv[0] = path;
3666                                 _argv[1] = NULL;
3667                                 argv = _argv;
3668                         } else
3669                                 argv[0] = path;
3670
3671                         execv(path, argv);
3672
3673                         log_error("Failed to execute %s: %m", path);
3674                         _exit(EXIT_FAILURE);
3675                 }
3676
3677                 log_debug("Spawned %s as %lu", path, (unsigned long) pid);
3678
3679                 if ((k = hashmap_put(pids, UINT_TO_PTR(pid), path)) < 0) {
3680                         log_error("Failed to add PID to set: %s", strerror(-k));
3681                         free(path);
3682                 }
3683         }
3684
3685         while (!hashmap_isempty(pids)) {
3686                 pid_t pid = PTR_TO_UINT(hashmap_first_key(pids));
3687                 siginfo_t si = {};
3688                 char *path;
3689
3690                 if (waitid(P_PID, pid, &si, WEXITED) < 0) {
3691
3692                         if (errno == EINTR)
3693                                 continue;
3694
3695                         log_error("waitid() failed: %m");
3696                         goto finish;
3697                 }
3698
3699                 if ((path = hashmap_remove(pids, UINT_TO_PTR(si.si_pid)))) {
3700                         if (!is_clean_exit(si.si_code, si.si_status, NULL)) {
3701                                 if (si.si_code == CLD_EXITED)
3702                                         log_error("%s exited with exit status %i.", path, si.si_status);
3703                                 else
3704                                         log_error("%s terminated by signal %s.", path, signal_to_string(si.si_status));
3705                         } else
3706                                 log_debug("%s exited successfully.", path);
3707
3708                         free(path);
3709                 }
3710         }
3711
3712 finish:
3713         if (_d)
3714                 closedir(_d);
3715
3716         if (pids)
3717                 hashmap_free_free(pids);
3718 }
3719
3720 int kill_and_sigcont(pid_t pid, int sig) {
3721         int r;
3722
3723         r = kill(pid, sig) < 0 ? -errno : 0;
3724
3725         if (r >= 0)
3726                 kill(pid, SIGCONT);
3727
3728         return r;
3729 }
3730
3731 bool nulstr_contains(const char*nulstr, const char *needle) {
3732         const char *i;
3733
3734         if (!nulstr)
3735                 return false;
3736
3737         NULSTR_FOREACH(i, nulstr)
3738                 if (streq(i, needle))
3739                         return true;
3740
3741         return false;
3742 }
3743
3744 bool plymouth_running(void) {
3745         return access("/run/plymouth/pid", F_OK) >= 0;
3746 }
3747
3748 char* strshorten(char *s, size_t l) {
3749         assert(s);
3750
3751         if (l < strlen(s))
3752                 s[l] = 0;
3753
3754         return s;
3755 }
3756
3757 static bool hostname_valid_char(char c) {
3758         return
3759                 (c >= 'a' && c <= 'z') ||
3760                 (c >= 'A' && c <= 'Z') ||
3761                 (c >= '0' && c <= '9') ||
3762                 c == '-' ||
3763                 c == '_' ||
3764                 c == '.';
3765 }
3766
3767 bool hostname_is_valid(const char *s) {
3768         const char *p;
3769         bool dot;
3770
3771         if (isempty(s))
3772                 return false;
3773
3774         for (p = s, dot = true; *p; p++) {
3775                 if (*p == '.') {
3776                         if (dot)
3777                                 return false;
3778
3779                         dot = true;
3780                 } else {
3781                         if (!hostname_valid_char(*p))
3782                                 return false;
3783
3784                         dot = false;
3785                 }
3786         }
3787
3788         if (dot)
3789                 return false;
3790
3791         if (p-s > HOST_NAME_MAX)
3792                 return false;
3793
3794         return true;
3795 }
3796
3797 char* hostname_cleanup(char *s, bool lowercase) {
3798         char *p, *d;
3799         bool dot;
3800
3801         for (p = s, d = s, dot = true; *p; p++) {
3802                 if (*p == '.') {
3803                         if (dot)
3804                                 continue;
3805
3806                         *(d++) = '.';
3807                         dot = true;
3808                 } else if (hostname_valid_char(*p)) {
3809                         *(d++) = lowercase ? tolower(*p) : *p;
3810                         dot = false;
3811                 }
3812
3813         }
3814
3815         if (dot && d > s)
3816                 d[-1] = 0;
3817         else
3818                 *d = 0;
3819
3820         strshorten(s, HOST_NAME_MAX);
3821
3822         return s;
3823 }
3824
3825 int pipe_eof(int fd) {
3826         int r;
3827         struct pollfd pollfd = {
3828                 .fd = fd,
3829                 .events = POLLIN|POLLHUP,
3830         };
3831
3832         r = poll(&pollfd, 1, 0);
3833         if (r < 0)
3834                 return -errno;
3835
3836         if (r == 0)
3837                 return 0;
3838
3839         return pollfd.revents & POLLHUP;
3840 }
3841
3842 int fd_wait_for_event(int fd, int event, usec_t t) {
3843
3844         struct pollfd pollfd = {
3845                 .fd = fd,
3846                 .events = event,
3847         };
3848
3849         struct timespec ts;
3850         int r;
3851
3852         r = ppoll(&pollfd, 1, t == (usec_t) -1 ? NULL : timespec_store(&ts, t), NULL);
3853         if (r < 0)
3854                 return -errno;
3855
3856         if (r == 0)
3857                 return 0;
3858
3859         return pollfd.revents;
3860 }
3861
3862 int fopen_temporary(const char *path, FILE **_f, char **_temp_path) {
3863         FILE *f;
3864         char *t;
3865         const char *fn;
3866         size_t k;
3867         int fd;
3868
3869         assert(path);
3870         assert(_f);
3871         assert(_temp_path);
3872
3873         t = new(char, strlen(path) + 1 + 6 + 1);
3874         if (!t)
3875                 return -ENOMEM;
3876
3877         fn = basename(path);
3878         k = fn - path;
3879         memcpy(t, path, k);
3880         t[k] = '.';
3881         stpcpy(stpcpy(t+k+1, fn), "XXXXXX");
3882
3883         fd = mkostemp(t, O_WRONLY|O_CLOEXEC);
3884         if (fd < 0) {
3885                 free(t);
3886                 return -errno;
3887         }
3888
3889         f = fdopen(fd, "we");
3890         if (!f) {
3891                 unlink(t);
3892                 free(t);
3893                 return -errno;
3894         }
3895
3896         *_f = f;
3897         *_temp_path = t;
3898
3899         return 0;
3900 }
3901
3902 int terminal_vhangup_fd(int fd) {
3903         assert(fd >= 0);
3904
3905         if (ioctl(fd, TIOCVHANGUP) < 0)
3906                 return -errno;
3907
3908         return 0;
3909 }
3910
3911 int terminal_vhangup(const char *name) {
3912         int fd, r;
3913
3914         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
3915         if (fd < 0)
3916                 return fd;
3917
3918         r = terminal_vhangup_fd(fd);
3919         close_nointr_nofail(fd);
3920
3921         return r;
3922 }
3923
3924 int vt_disallocate(const char *name) {
3925         int fd, r;
3926         unsigned u;
3927
3928         /* Deallocate the VT if possible. If not possible
3929          * (i.e. because it is the active one), at least clear it
3930          * entirely (including the scrollback buffer) */
3931
3932         if (!startswith(name, "/dev/"))
3933                 return -EINVAL;
3934
3935         if (!tty_is_vc(name)) {
3936                 /* So this is not a VT. I guess we cannot deallocate
3937                  * it then. But let's at least clear the screen */
3938
3939                 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
3940                 if (fd < 0)
3941                         return fd;
3942
3943                 loop_write(fd,
3944                            "\033[r"    /* clear scrolling region */
3945                            "\033[H"    /* move home */
3946                            "\033[2J",  /* clear screen */
3947                            10, false);
3948                 close_nointr_nofail(fd);
3949
3950                 return 0;
3951         }
3952
3953         if (!startswith(name, "/dev/tty"))
3954                 return -EINVAL;
3955
3956         r = safe_atou(name+8, &u);
3957         if (r < 0)
3958                 return r;
3959
3960         if (u <= 0)
3961                 return -EINVAL;
3962
3963         /* Try to deallocate */
3964         fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
3965         if (fd < 0)
3966                 return fd;
3967
3968         r = ioctl(fd, VT_DISALLOCATE, u);
3969         close_nointr_nofail(fd);
3970
3971         if (r >= 0)
3972                 return 0;
3973
3974         if (errno != EBUSY)
3975                 return -errno;
3976
3977         /* Couldn't deallocate, so let's clear it fully with
3978          * scrollback */
3979         fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
3980         if (fd < 0)
3981                 return fd;
3982
3983         loop_write(fd,
3984                    "\033[r"   /* clear scrolling region */
3985                    "\033[H"   /* move home */
3986                    "\033[3J", /* clear screen including scrollbac