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