]> git.ipfire.org Git - thirdparty/systemd.git/blob - src/shared/util.c
projects / thirdparty / systemd.git / blob
? search:
summary | shortlog | log | commit | commitdiff | tree
blame | history | raw | HEAD
util: fix minimal race where we might miss SIGTERMs when forking off an agent
[thirdparty/systemd.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 if (fd < 0)
1882 return -errno;
1883
1884 r = isatty(fd);
1885 if (r < 0) {
1886 safe_close(fd);
1887 return -errno;
1888 }
1889
1890 if (!r) {
1891 safe_close(fd);
1892 return -ENOTTY;
1893 }
1894
1895 return fd;
1896 }
1897
1898 int flush_fd(int fd) {
1899 struct pollfd pollfd = {
1900 .fd = fd,
1901 .events = POLLIN,
1902 };
1903
1904 for (;;) {
1905 char buf[LINE_MAX];
1906 ssize_t l;
1907 int r;
1908
1909 r = poll(&pollfd, 1, 0);
1910 if (r < 0) {
1911 if (errno == EINTR)
1912 continue;
1913
1914 return -errno;
1915
1916 } else if (r == 0)
1917 return 0;
1918
1919 l = read(fd, buf, sizeof(buf));
1920 if (l < 0) {
1921
1922 if (errno == EINTR)
1923 continue;
1924
1925 if (errno == EAGAIN)
1926 return 0;
1927
1928 return -errno;
1929 } else if (l == 0)
1930 return 0;
1931 }
1932 }
1933
1934 int acquire_terminal(
1935 const char *name,
1936 bool fail,
1937 bool force,
1938 bool ignore_tiocstty_eperm,
1939 usec_t timeout) {
1940
1941 int fd = -1, notify = -1, r = 0, wd = -1;
1942 usec_t ts = 0;
1943
1944 assert(name);
1945
1946 /* We use inotify to be notified when the tty is closed. We
1947 * create the watch before checking if we can actually acquire
1948 * it, so that we don't lose any event.
1949 *
1950 * Note: strictly speaking this actually watches for the
1951 * device being closed, it does *not* really watch whether a
1952 * tty loses its controlling process. However, unless some
1953 * rogue process uses TIOCNOTTY on /dev/tty *after* closing
1954 * its tty otherwise this will not become a problem. As long
1955 * as the administrator makes sure not configure any service
1956 * on the same tty as an untrusted user this should not be a
1957 * problem. (Which he probably should not do anyway.) */
1958
1959 if (timeout != USEC_INFINITY)
1960 ts = now(CLOCK_MONOTONIC);
1961
1962 if (!fail && !force) {
1963 notify = inotify_init1(IN_CLOEXEC | (timeout != USEC_INFINITY ? IN_NONBLOCK : 0));
1964 if (notify < 0) {
1965 r = -errno;
1966 goto fail;
1967 }
1968
1969 wd = inotify_add_watch(notify, name, IN_CLOSE);
1970 if (wd < 0) {
1971 r = -errno;
1972 goto fail;
1973 }
1974 }
1975
1976 for (;;) {
1977 struct sigaction sa_old, sa_new = {
1978 .sa_handler = SIG_IGN,
1979 .sa_flags = SA_RESTART,
1980 };
1981
1982 if (notify >= 0) {
1983 r = flush_fd(notify);
1984 if (r < 0)
1985 goto fail;
1986 }
1987
1988 /* We pass here O_NOCTTY only so that we can check the return
1989 * value TIOCSCTTY and have a reliable way to figure out if we
1990 * successfully became the controlling process of the tty */
1991 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
1992 if (fd < 0)
1993 return fd;
1994
1995 /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
1996 * if we already own the tty. */
1997 assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
1998
1999 /* First, try to get the tty */
2000 if (ioctl(fd, TIOCSCTTY, force) < 0)
2001 r = -errno;
2002
2003 assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
2004
2005 /* Sometimes it makes sense to ignore TIOCSCTTY
2006 * returning EPERM, i.e. when very likely we already
2007 * are have this controlling terminal. */
2008 if (r < 0 && r == -EPERM && ignore_tiocstty_eperm)
2009 r = 0;
2010
2011 if (r < 0 && (force || fail || r != -EPERM)) {
2012 goto fail;
2013 }
2014
2015 if (r >= 0)
2016 break;
2017
2018 assert(!fail);
2019 assert(!force);
2020 assert(notify >= 0);
2021
2022 for (;;) {
2023 uint8_t inotify_buffer[sizeof(struct inotify_event) + FILENAME_MAX];
2024 ssize_t l;
2025 struct inotify_event *e;
2026
2027 if (timeout != USEC_INFINITY) {
2028 usec_t n;
2029
2030 n = now(CLOCK_MONOTONIC);
2031 if (ts + timeout < n) {
2032 r = -ETIMEDOUT;
2033 goto fail;
2034 }
2035
2036 r = fd_wait_for_event(fd, POLLIN, ts + timeout - n);
2037 if (r < 0)
2038 goto fail;
2039
2040 if (r == 0) {
2041 r = -ETIMEDOUT;
2042 goto fail;
2043 }
2044 }
2045
2046 l = read(notify, inotify_buffer, sizeof(inotify_buffer));
2047 if (l < 0) {
2048
2049 if (errno == EINTR || errno == EAGAIN)
2050 continue;
2051
2052 r = -errno;
2053 goto fail;
2054 }
2055
2056 e = (struct inotify_event*) inotify_buffer;
2057
2058 while (l > 0) {
2059 size_t step;
2060
2061 if (e->wd != wd || !(e->mask & IN_CLOSE)) {
2062 r = -EIO;
2063 goto fail;
2064 }
2065
2066 step = sizeof(struct inotify_event) + e->len;
2067 assert(step <= (size_t) l);
2068
2069 e = (struct inotify_event*) ((uint8_t*) e + step);
2070 l -= step;
2071 }
2072
2073 break;
2074 }
2075
2076 /* We close the tty fd here since if the old session
2077 * ended our handle will be dead. It's important that
2078 * we do this after sleeping, so that we don't enter
2079 * an endless loop. */
2080 safe_close(fd);
2081 }
2082
2083 safe_close(notify);
2084
2085 r = reset_terminal_fd(fd, true);
2086 if (r < 0)
2087 log_warning("Failed to reset terminal: %s", strerror(-r));
2088
2089 return fd;
2090
2091 fail:
2092 safe_close(fd);
2093 safe_close(notify);
2094
2095 return r;
2096 }
2097
2098 int release_terminal(void) {
2099 static const struct sigaction sa_new = {
2100 .sa_handler = SIG_IGN,
2101 .sa_flags = SA_RESTART,
2102 };
2103
2104 _cleanup_close_ int fd = -1;
2105 struct sigaction sa_old;
2106 int r = 0;
2107
2108 fd = open("/dev/tty", O_RDWR|O_NOCTTY|O_NDELAY|O_CLOEXEC);
2109 if (fd < 0)
2110 return -errno;
2111
2112 /* Temporarily ignore SIGHUP, so that we don't get SIGHUP'ed
2113 * by our own TIOCNOTTY */
2114 assert_se(sigaction(SIGHUP, &sa_new, &sa_old) == 0);
2115
2116 if (ioctl(fd, TIOCNOTTY) < 0)
2117 r = -errno;
2118
2119 assert_se(sigaction(SIGHUP, &sa_old, NULL) == 0);
2120
2121 return r;
2122 }
2123
2124 int sigaction_many(const struct sigaction *sa, ...) {
2125 va_list ap;
2126 int r = 0, sig;
2127
2128 va_start(ap, sa);
2129 while ((sig = va_arg(ap, int)) > 0)
2130 if (sigaction(sig, sa, NULL) < 0)
2131 r = -errno;
2132 va_end(ap);
2133
2134 return r;
2135 }
2136
2137 int ignore_signals(int sig, ...) {
2138 struct sigaction sa = {
2139 .sa_handler = SIG_IGN,
2140 .sa_flags = SA_RESTART,
2141 };
2142 va_list ap;
2143 int r = 0;
2144
2145 if (sigaction(sig, &sa, NULL) < 0)
2146 r = -errno;
2147
2148 va_start(ap, sig);
2149 while ((sig = va_arg(ap, int)) > 0)
2150 if (sigaction(sig, &sa, NULL) < 0)
2151 r = -errno;
2152 va_end(ap);
2153
2154 return r;
2155 }
2156
2157 int default_signals(int sig, ...) {
2158 struct sigaction sa = {
2159 .sa_handler = SIG_DFL,
2160 .sa_flags = SA_RESTART,
2161 };
2162 va_list ap;
2163 int r = 0;
2164
2165 if (sigaction(sig, &sa, NULL) < 0)
2166 r = -errno;
2167
2168 va_start(ap, sig);
2169 while ((sig = va_arg(ap, int)) > 0)
2170 if (sigaction(sig, &sa, NULL) < 0)
2171 r = -errno;
2172 va_end(ap);
2173
2174 return r;
2175 }
2176
2177 void safe_close_pair(int p[]) {
2178 assert(p);
2179
2180 if (p[0] == p[1]) {
2181 /* Special case pairs which use the same fd in both
2182 * directions... */
2183 p[0] = p[1] = safe_close(p[0]);
2184 return;
2185 }
2186
2187 p[0] = safe_close(p[0]);
2188 p[1] = safe_close(p[1]);
2189 }
2190
2191 ssize_t loop_read(int fd, void *buf, size_t nbytes, bool do_poll) {
2192 uint8_t *p = buf;
2193 ssize_t n = 0;
2194
2195 assert(fd >= 0);
2196 assert(buf);
2197
2198 while (nbytes > 0) {
2199 ssize_t k;
2200
2201 k = read(fd, p, nbytes);
2202 if (k < 0 && errno == EINTR)
2203 continue;
2204
2205 if (k < 0 && errno == EAGAIN && do_poll) {
2206
2207 /* We knowingly ignore any return value here,
2208 * and expect that any error/EOF is reported
2209 * via read() */
2210
2211 fd_wait_for_event(fd, POLLIN, USEC_INFINITY);
2212 continue;
2213 }
2214
2215 if (k <= 0)
2216 return n > 0 ? n : (k < 0 ? -errno : 0);
2217
2218 p += k;
2219 nbytes -= k;
2220 n += k;
2221 }
2222
2223 return n;
2224 }
2225
2226 ssize_t loop_write(int fd, const void *buf, size_t nbytes, bool do_poll) {
2227 const uint8_t *p = buf;
2228 ssize_t n = 0;
2229
2230 assert(fd >= 0);
2231 assert(buf);
2232
2233 while (nbytes > 0) {
2234 ssize_t k;
2235
2236 k = write(fd, p, nbytes);
2237 if (k < 0 && errno == EINTR)
2238 continue;
2239
2240 if (k < 0 && errno == EAGAIN && do_poll) {
2241
2242 /* We knowingly ignore any return value here,
2243 * and expect that any error/EOF is reported
2244 * via write() */
2245
2246 fd_wait_for_event(fd, POLLOUT, USEC_INFINITY);
2247 continue;
2248 }
2249
2250 if (k <= 0)
2251 return n > 0 ? n : (k < 0 ? -errno : 0);
2252
2253 p += k;
2254 nbytes -= k;
2255 n += k;
2256 }
2257
2258 return n;
2259 }
2260
2261 int parse_size(const char *t, off_t base, off_t *size) {
2262
2263 /* Soo, sometimes we want to parse IEC binary suffxies, and
2264 * sometimes SI decimal suffixes. This function can parse
2265 * both. Which one is the right way depends on the
2266 * context. Wikipedia suggests that SI is customary for
2267 * hardrware metrics and network speeds, while IEC is
2268 * customary for most data sizes used by software and volatile
2269 * (RAM) memory. Hence be careful which one you pick!
2270 *
2271 * In either case we use just K, M, G as suffix, and not Ki,
2272 * Mi, Gi or so (as IEC would suggest). That's because that's
2273 * frickin' ugly. But this means you really need to make sure
2274 * to document which base you are parsing when you use this
2275 * call. */
2276
2277 struct table {
2278 const char *suffix;
2279 unsigned long long factor;
2280 };
2281
2282 static const struct table iec[] = {
2283 { "E", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2284 { "P", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
2285 { "T", 1024ULL*1024ULL*1024ULL*1024ULL },
2286 { "G", 1024ULL*1024ULL*1024ULL },
2287 { "M", 1024ULL*1024ULL },
2288 { "K", 1024ULL },
2289 { "B", 1 },
2290 { "", 1 },
2291 };
2292
2293 static const struct table si[] = {
2294 { "E", 1000ULL*1000ULL*1000ULL*1000ULL*1000ULL*1000ULL },
2295 { "P", 1000ULL*1000ULL*1000ULL*1000ULL*1000ULL },
2296 { "T", 1000ULL*1000ULL*1000ULL*1000ULL },
2297 { "G", 1000ULL*1000ULL*1000ULL },
2298 { "M", 1000ULL*1000ULL },
2299 { "K", 1000ULL },
2300 { "B", 1 },
2301 { "", 1 },
2302 };
2303
2304 const struct table *table;
2305 const char *p;
2306 unsigned long long r = 0;
2307 unsigned n_entries, start_pos = 0;
2308
2309 assert(t);
2310 assert(base == 1000 || base == 1024);
2311 assert(size);
2312
2313 if (base == 1000) {
2314 table = si;
2315 n_entries = ELEMENTSOF(si);
2316 } else {
2317 table = iec;
2318 n_entries = ELEMENTSOF(iec);
2319 }
2320
2321 p = t;
2322 do {
2323 long long l;
2324 unsigned long long l2;
2325 double frac = 0;
2326 char *e;
2327 unsigned i;
2328
2329 errno = 0;
2330 l = strtoll(p, &e, 10);
2331
2332 if (errno > 0)
2333 return -errno;
2334
2335 if (l < 0)
2336 return -ERANGE;
2337
2338 if (e == p)
2339 return -EINVAL;
2340
2341 if (*e == '.') {
2342 e++;
2343 if (*e >= '0' && *e <= '9') {
2344 char *e2;
2345
2346 /* strotoull itself would accept space/+/- */
2347 l2 = strtoull(e, &e2, 10);
2348
2349 if (errno == ERANGE)
2350 return -errno;
2351
2352 /* Ignore failure. E.g. 10.M is valid */
2353 frac = l2;
2354 for (; e < e2; e++)
2355 frac /= 10;
2356 }
2357 }
2358
2359 e += strspn(e, WHITESPACE);
2360
2361 for (i = start_pos; i < n_entries; i++)
2362 if (startswith(e, table[i].suffix)) {
2363 unsigned long long tmp;
2364 if ((unsigned long long) l + (frac > 0) > ULLONG_MAX / table[i].factor)
2365 return -ERANGE;
2366 tmp = l * table[i].factor + (unsigned long long) (frac * table[i].factor);
2367 if (tmp > ULLONG_MAX - r)
2368 return -ERANGE;
2369
2370 r += tmp;
2371 if ((unsigned long long) (off_t) r != r)
2372 return -ERANGE;
2373
2374 p = e + strlen(table[i].suffix);
2375
2376 start_pos = i + 1;
2377 break;
2378 }
2379
2380 if (i >= n_entries)
2381 return -EINVAL;
2382
2383 } while (*p);
2384
2385 *size = r;
2386
2387 return 0;
2388 }
2389
2390 int make_stdio(int fd) {
2391 int r, s, t;
2392
2393 assert(fd >= 0);
2394
2395 r = dup3(fd, STDIN_FILENO, 0);
2396 s = dup3(fd, STDOUT_FILENO, 0);
2397 t = dup3(fd, STDERR_FILENO, 0);
2398
2399 if (fd >= 3)
2400 safe_close(fd);
2401
2402 if (r < 0 || s < 0 || t < 0)
2403 return -errno;
2404
2405 /* We rely here that the new fd has O_CLOEXEC not set */
2406
2407 return 0;
2408 }
2409
2410 int make_null_stdio(void) {
2411 int null_fd;
2412
2413 null_fd = open("/dev/null", O_RDWR|O_NOCTTY);
2414 if (null_fd < 0)
2415 return -errno;
2416
2417 return make_stdio(null_fd);
2418 }
2419
2420 bool is_device_path(const char *path) {
2421
2422 /* Returns true on paths that refer to a device, either in
2423 * sysfs or in /dev */
2424
2425 return
2426 path_startswith(path, "/dev/") ||
2427 path_startswith(path, "/sys/");
2428 }
2429
2430 int dir_is_empty(const char *path) {
2431 _cleanup_closedir_ DIR *d;
2432
2433 d = opendir(path);
2434 if (!d)
2435 return -errno;
2436
2437 for (;;) {
2438 struct dirent *de;
2439
2440 errno = 0;
2441 de = readdir(d);
2442 if (!de && errno != 0)
2443 return -errno;
2444
2445 if (!de)
2446 return 1;
2447
2448 if (!ignore_file(de->d_name))
2449 return 0;
2450 }
2451 }
2452
2453 char* dirname_malloc(const char *path) {
2454 char *d, *dir, *dir2;
2455
2456 d = strdup(path);
2457 if (!d)
2458 return NULL;
2459 dir = dirname(d);
2460 assert(dir);
2461
2462 if (dir != d) {
2463 dir2 = strdup(dir);
2464 free(d);
2465 return dir2;
2466 }
2467
2468 return dir;
2469 }
2470
2471 int dev_urandom(void *p, size_t n) {
2472 _cleanup_close_ int fd;
2473 ssize_t k;
2474
2475 fd = open("/dev/urandom", O_RDONLY|O_CLOEXEC|O_NOCTTY);
2476 if (fd < 0)
2477 return errno == ENOENT ? -ENOSYS : -errno;
2478
2479 k = loop_read(fd, p, n, true);
2480 if (k < 0)
2481 return (int) k;
2482 if ((size_t) k != n)
2483 return -EIO;
2484
2485 return 0;
2486 }
2487
2488 void random_bytes(void *p, size_t n) {
2489 static bool srand_called = false;
2490 uint8_t *q;
2491 int r;
2492
2493 r = dev_urandom(p, n);
2494 if (r >= 0)
2495 return;
2496
2497 /* If some idiot made /dev/urandom unavailable to us, he'll
2498 * get a PRNG instead. */
2499
2500 if (!srand_called) {
2501 unsigned x = 0;
2502
2503 #ifdef HAVE_SYS_AUXV_H
2504 /* The kernel provides us with a bit of entropy in
2505 * auxv, so let's try to make use of that to seed the
2506 * pseudo-random generator. It's better than
2507 * nothing... */
2508
2509 void *auxv;
2510
2511 auxv = (void*) getauxval(AT_RANDOM);
2512 if (auxv)
2513 x ^= *(unsigned*) auxv;
2514 #endif
2515
2516 x ^= (unsigned) now(CLOCK_REALTIME);
2517 x ^= (unsigned) gettid();
2518
2519 srand(x);
2520 srand_called = true;
2521 }
2522
2523 for (q = p; q < (uint8_t*) p + n; q ++)
2524 *q = rand();
2525 }
2526
2527 void rename_process(const char name[8]) {
2528 assert(name);
2529
2530 /* This is a like a poor man's setproctitle(). It changes the
2531 * comm field, argv[0], and also the glibc's internally used
2532 * name of the process. For the first one a limit of 16 chars
2533 * applies, to the second one usually one of 10 (i.e. length
2534 * of "/sbin/init"), to the third one one of 7 (i.e. length of
2535 * "systemd"). If you pass a longer string it will be
2536 * truncated */
2537
2538 prctl(PR_SET_NAME, name);
2539
2540 if (program_invocation_name)
2541 strncpy(program_invocation_name, name, strlen(program_invocation_name));
2542
2543 if (saved_argc > 0) {
2544 int i;
2545
2546 if (saved_argv[0])
2547 strncpy(saved_argv[0], name, strlen(saved_argv[0]));
2548
2549 for (i = 1; i < saved_argc; i++) {
2550 if (!saved_argv[i])
2551 break;
2552
2553 memzero(saved_argv[i], strlen(saved_argv[i]));
2554 }
2555 }
2556 }
2557
2558 void sigset_add_many(sigset_t *ss, ...) {
2559 va_list ap;
2560 int sig;
2561
2562 assert(ss);
2563
2564 va_start(ap, ss);
2565 while ((sig = va_arg(ap, int)) > 0)
2566 assert_se(sigaddset(ss, sig) == 0);
2567 va_end(ap);
2568 }
2569
2570 int sigprocmask_many(int how, ...) {
2571 va_list ap;
2572 sigset_t ss;
2573 int sig;
2574
2575 assert_se(sigemptyset(&ss) == 0);
2576
2577 va_start(ap, how);
2578 while ((sig = va_arg(ap, int)) > 0)
2579 assert_se(sigaddset(&ss, sig) == 0);
2580 va_end(ap);
2581
2582 if (sigprocmask(how, &ss, NULL) < 0)
2583 return -errno;
2584
2585 return 0;
2586 }
2587
2588 char* gethostname_malloc(void) {
2589 struct utsname u;
2590
2591 assert_se(uname(&u) >= 0);
2592
2593 if (!isempty(u.nodename) && !streq(u.nodename, "(none)"))
2594 return strdup(u.nodename);
2595
2596 return strdup(u.sysname);
2597 }
2598
2599 bool hostname_is_set(void) {
2600 struct utsname u;
2601
2602 assert_se(uname(&u) >= 0);
2603
2604 return !isempty(u.nodename) && !streq(u.nodename, "(none)");
2605 }
2606
2607 char *lookup_uid(uid_t uid) {
2608 long bufsize;
2609 char *name;
2610 _cleanup_free_ char *buf = NULL;
2611 struct passwd pwbuf, *pw = NULL;
2612
2613 /* Shortcut things to avoid NSS lookups */
2614 if (uid == 0)
2615 return strdup("root");
2616
2617 bufsize = sysconf(_SC_GETPW_R_SIZE_MAX);
2618 if (bufsize <= 0)
2619 bufsize = 4096;
2620
2621 buf = malloc(bufsize);
2622 if (!buf)
2623 return NULL;
2624
2625 if (getpwuid_r(uid, &pwbuf, buf, bufsize, &pw) == 0 && pw)
2626 return strdup(pw->pw_name);
2627
2628 if (asprintf(&name, UID_FMT, uid) < 0)
2629 return NULL;
2630
2631 return name;
2632 }
2633
2634 char* getlogname_malloc(void) {
2635 uid_t uid;
2636 struct stat st;
2637
2638 if (isatty(STDIN_FILENO) && fstat(STDIN_FILENO, &st) >= 0)
2639 uid = st.st_uid;
2640 else
2641 uid = getuid();
2642
2643 return lookup_uid(uid);
2644 }
2645
2646 char *getusername_malloc(void) {
2647 const char *e;
2648
2649 e = getenv("USER");
2650 if (e)
2651 return strdup(e);
2652
2653 return lookup_uid(getuid());
2654 }
2655
2656 int getttyname_malloc(int fd, char **r) {
2657 char path[PATH_MAX], *c;
2658 int k;
2659
2660 assert(r);
2661
2662 k = ttyname_r(fd, path, sizeof(path));
2663 if (k > 0)
2664 return -k;
2665
2666 char_array_0(path);
2667
2668 c = strdup(startswith(path, "/dev/") ? path + 5 : path);
2669 if (!c)
2670 return -ENOMEM;
2671
2672 *r = c;
2673 return 0;
2674 }
2675
2676 int getttyname_harder(int fd, char **r) {
2677 int k;
2678 char *s;
2679
2680 k = getttyname_malloc(fd, &s);
2681 if (k < 0)
2682 return k;
2683
2684 if (streq(s, "tty")) {
2685 free(s);
2686 return get_ctty(0, NULL, r);
2687 }
2688
2689 *r = s;
2690 return 0;
2691 }
2692
2693 int get_ctty_devnr(pid_t pid, dev_t *d) {
2694 int r;
2695 _cleanup_free_ char *line = NULL;
2696 const char *p;
2697 unsigned long ttynr;
2698
2699 assert(pid >= 0);
2700
2701 p = procfs_file_alloca(pid, "stat");
2702 r = read_one_line_file(p, &line);
2703 if (r < 0)
2704 return r;
2705
2706 p = strrchr(line, ')');
2707 if (!p)
2708 return -EIO;
2709
2710 p++;
2711
2712 if (sscanf(p, " "
2713 "%*c " /* state */
2714 "%*d " /* ppid */
2715 "%*d " /* pgrp */
2716 "%*d " /* session */
2717 "%lu ", /* ttynr */
2718 &ttynr) != 1)
2719 return -EIO;
2720
2721 if (major(ttynr) == 0 && minor(ttynr) == 0)
2722 return -ENOENT;
2723
2724 if (d)
2725 *d = (dev_t) ttynr;
2726
2727 return 0;
2728 }
2729
2730 int get_ctty(pid_t pid, dev_t *_devnr, char **r) {
2731 char fn[sizeof("/dev/char/")-1 + 2*DECIMAL_STR_MAX(unsigned) + 1 + 1], *b = NULL;
2732 _cleanup_free_ char *s = NULL;
2733 const char *p;
2734 dev_t devnr;
2735 int k;
2736
2737 assert(r);
2738
2739 k = get_ctty_devnr(pid, &devnr);
2740 if (k < 0)
2741 return k;
2742
2743 snprintf(fn, sizeof(fn), "/dev/char/%u:%u", major(devnr), minor(devnr));
2744
2745 k = readlink_malloc(fn, &s);
2746 if (k < 0) {
2747
2748 if (k != -ENOENT)
2749 return k;
2750
2751 /* This is an ugly hack */
2752 if (major(devnr) == 136) {
2753 asprintf(&b, "pts/%u", minor(devnr));
2754 goto finish;
2755 }
2756
2757 /* Probably something like the ptys which have no
2758 * symlink in /dev/char. Let's return something
2759 * vaguely useful. */
2760
2761 b = strdup(fn + 5);
2762 goto finish;
2763 }
2764
2765 if (startswith(s, "/dev/"))
2766 p = s + 5;
2767 else if (startswith(s, "../"))
2768 p = s + 3;
2769 else
2770 p = s;
2771
2772 b = strdup(p);
2773
2774 finish:
2775 if (!b)
2776 return -ENOMEM;
2777
2778 *r = b;
2779 if (_devnr)
2780 *_devnr = devnr;
2781
2782 return 0;
2783 }
2784
2785 int rm_rf_children_dangerous(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
2786 _cleanup_closedir_ DIR *d = NULL;
2787 int ret = 0;
2788
2789 assert(fd >= 0);
2790
2791 /* This returns the first error we run into, but nevertheless
2792 * tries to go on. This closes the passed fd. */
2793
2794 d = fdopendir(fd);
2795 if (!d) {
2796 safe_close(fd);
2797
2798 return errno == ENOENT ? 0 : -errno;
2799 }
2800
2801 for (;;) {
2802 struct dirent *de;
2803 bool is_dir, keep_around;
2804 struct stat st;
2805 int r;
2806
2807 errno = 0;
2808 de = readdir(d);
2809 if (!de) {
2810 if (errno != 0 && ret == 0)
2811 ret = -errno;
2812 return ret;
2813 }
2814
2815 if (streq(de->d_name, ".") || streq(de->d_name, ".."))
2816 continue;
2817
2818 if (de->d_type == DT_UNKNOWN ||
2819 honour_sticky ||
2820 (de->d_type == DT_DIR && root_dev)) {
2821 if (fstatat(fd, de->d_name, &st, AT_SYMLINK_NOFOLLOW) < 0) {
2822 if (ret == 0 && errno != ENOENT)
2823 ret = -errno;
2824 continue;
2825 }
2826
2827 is_dir = S_ISDIR(st.st_mode);
2828 keep_around =
2829 honour_sticky &&
2830 (st.st_uid == 0 || st.st_uid == getuid()) &&
2831 (st.st_mode & S_ISVTX);
2832 } else {
2833 is_dir = de->d_type == DT_DIR;
2834 keep_around = false;
2835 }
2836
2837 if (is_dir) {
2838 int subdir_fd;
2839
2840 /* if root_dev is set, remove subdirectories only, if device is same as dir */
2841 if (root_dev && st.st_dev != root_dev->st_dev)
2842 continue;
2843
2844 subdir_fd = openat(fd, de->d_name,
2845 O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
2846 if (subdir_fd < 0) {
2847 if (ret == 0 && errno != ENOENT)
2848 ret = -errno;
2849 continue;
2850 }
2851
2852 r = rm_rf_children_dangerous(subdir_fd, only_dirs, honour_sticky, root_dev);
2853 if (r < 0 && ret == 0)
2854 ret = r;
2855
2856 if (!keep_around)
2857 if (unlinkat(fd, de->d_name, AT_REMOVEDIR) < 0) {
2858 if (ret == 0 && errno != ENOENT)
2859 ret = -errno;
2860 }
2861
2862 } else if (!only_dirs && !keep_around) {
2863
2864 if (unlinkat(fd, de->d_name, 0) < 0) {
2865 if (ret == 0 && errno != ENOENT)
2866 ret = -errno;
2867 }
2868 }
2869 }
2870 }
2871
2872 _pure_ static int is_temporary_fs(struct statfs *s) {
2873 assert(s);
2874
2875 return F_TYPE_EQUAL(s->f_type, TMPFS_MAGIC) ||
2876 F_TYPE_EQUAL(s->f_type, RAMFS_MAGIC);
2877 }
2878
2879 int rm_rf_children(int fd, bool only_dirs, bool honour_sticky, struct stat *root_dev) {
2880 struct statfs s;
2881
2882 assert(fd >= 0);
2883
2884 if (fstatfs(fd, &s) < 0) {
2885 safe_close(fd);
2886 return -errno;
2887 }
2888
2889 /* We refuse to clean disk file systems with this call. This
2890 * is extra paranoia just to be sure we never ever remove
2891 * non-state data */
2892 if (!is_temporary_fs(&s)) {
2893 log_error("Attempted to remove disk file system, and we can't allow that.");
2894 safe_close(fd);
2895 return -EPERM;
2896 }
2897
2898 return rm_rf_children_dangerous(fd, only_dirs, honour_sticky, root_dev);
2899 }
2900
2901 static int rm_rf_internal(const char *path, bool only_dirs, bool delete_root, bool honour_sticky, bool dangerous) {
2902 int fd, r;
2903 struct statfs s;
2904
2905 assert(path);
2906
2907 /* We refuse to clean the root file system with this
2908 * call. This is extra paranoia to never cause a really
2909 * seriously broken system. */
2910 if (path_equal(path, "/")) {
2911 log_error("Attempted to remove entire root file system, and we can't allow that.");
2912 return -EPERM;
2913 }
2914
2915 fd = open(path, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|O_NOFOLLOW|O_NOATIME);
2916 if (fd < 0) {
2917
2918 if (errno != ENOTDIR)
2919 return -errno;
2920
2921 if (!dangerous) {
2922 if (statfs(path, &s) < 0)
2923 return -errno;
2924
2925 if (!is_temporary_fs(&s)) {
2926 log_error("Attempted to remove disk file system, and we can't allow that.");
2927 return -EPERM;
2928 }
2929 }
2930
2931 if (delete_root && !only_dirs)
2932 if (unlink(path) < 0 && errno != ENOENT)
2933 return -errno;
2934
2935 return 0;
2936 }
2937
2938 if (!dangerous) {
2939 if (fstatfs(fd, &s) < 0) {
2940 safe_close(fd);
2941 return -errno;
2942 }
2943
2944 if (!is_temporary_fs(&s)) {
2945 log_error("Attempted to remove disk file system, and we can't allow that.");
2946 safe_close(fd);
2947 return -EPERM;
2948 }
2949 }
2950
2951 r = rm_rf_children_dangerous(fd, only_dirs, honour_sticky, NULL);
2952 if (delete_root) {
2953
2954 if (honour_sticky && file_is_priv_sticky(path) > 0)
2955 return r;
2956
2957 if (rmdir(path) < 0 && errno != ENOENT) {
2958 if (r == 0)
2959 r = -errno;
2960 }
2961 }
2962
2963 return r;
2964 }
2965
2966 int rm_rf(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
2967 return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, false);
2968 }
2969
2970 int rm_rf_dangerous(const char *path, bool only_dirs, bool delete_root, bool honour_sticky) {
2971 return rm_rf_internal(path, only_dirs, delete_root, honour_sticky, true);
2972 }
2973
2974 int chmod_and_chown(const char *path, mode_t mode, uid_t uid, gid_t gid) {
2975 assert(path);
2976
2977 /* Under the assumption that we are running privileged we
2978 * first change the access mode and only then hand out
2979 * ownership to avoid a window where access is too open. */
2980
2981 if (mode != (mode_t) -1)
2982 if (chmod(path, mode) < 0)
2983 return -errno;
2984
2985 if (uid != (uid_t) -1 || gid != (gid_t) -1)
2986 if (chown(path, uid, gid) < 0)
2987 return -errno;
2988
2989 return 0;
2990 }
2991
2992 int fchmod_and_fchown(int fd, mode_t mode, uid_t uid, gid_t gid) {
2993 assert(fd >= 0);
2994
2995 /* Under the assumption that we are running privileged we
2996 * first change the access mode and only then hand out
2997 * ownership to avoid a window where access is too open. */
2998
2999 if (mode != (mode_t) -1)
3000 if (fchmod(fd, mode) < 0)
3001 return -errno;
3002
3003 if (uid != (uid_t) -1 || gid != (gid_t) -1)
3004 if (fchown(fd, uid, gid) < 0)
3005 return -errno;
3006
3007 return 0;
3008 }
3009
3010 cpu_set_t* cpu_set_malloc(unsigned *ncpus) {
3011 cpu_set_t *r;
3012 unsigned n = 1024;
3013
3014 /* Allocates the cpuset in the right size */
3015
3016 for (;;) {
3017 if (!(r = CPU_ALLOC(n)))
3018 return NULL;
3019
3020 if (sched_getaffinity(0, CPU_ALLOC_SIZE(n), r) >= 0) {
3021 CPU_ZERO_S(CPU_ALLOC_SIZE(n), r);
3022
3023 if (ncpus)
3024 *ncpus = n;
3025
3026 return r;
3027 }
3028
3029 CPU_FREE(r);
3030
3031 if (errno != EINVAL)
3032 return NULL;
3033
3034 n *= 2;
3035 }
3036 }
3037
3038 int status_vprintf(const char *status, bool ellipse, bool ephemeral, const char *format, va_list ap) {
3039 static const char status_indent[] = " "; /* "[" STATUS "] " */
3040 _cleanup_free_ char *s = NULL;
3041 _cleanup_close_ int fd = -1;
3042 struct iovec iovec[6] = {};
3043 int n = 0;
3044 static bool prev_ephemeral;
3045
3046 assert(format);
3047
3048 /* This is independent of logging, as status messages are
3049 * optional and go exclusively to the console. */
3050
3051 if (vasprintf(&s, format, ap) < 0)
3052 return log_oom();
3053
3054 fd = open_terminal("/dev/console", O_WRONLY|O_NOCTTY|O_CLOEXEC);
3055 if (fd < 0)
3056 return fd;
3057
3058 if (ellipse) {
3059 char *e;
3060 size_t emax, sl;
3061 int c;
3062
3063 c = fd_columns(fd);
3064 if (c <= 0)
3065 c = 80;
3066
3067 sl = status ? sizeof(status_indent)-1 : 0;
3068
3069 emax = c - sl - 1;
3070 if (emax < 3)
3071 emax = 3;
3072
3073 e = ellipsize(s, emax, 50);
3074 if (e) {
3075 free(s);
3076 s = e;
3077 }
3078 }
3079
3080 if (prev_ephemeral)
3081 IOVEC_SET_STRING(iovec[n++], "\r" ANSI_ERASE_TO_END_OF_LINE);
3082 prev_ephemeral = ephemeral;
3083
3084 if (status) {
3085 if (!isempty(status)) {
3086 IOVEC_SET_STRING(iovec[n++], "[");
3087 IOVEC_SET_STRING(iovec[n++], status);
3088 IOVEC_SET_STRING(iovec[n++], "] ");
3089 } else
3090 IOVEC_SET_STRING(iovec[n++], status_indent);
3091 }
3092
3093 IOVEC_SET_STRING(iovec[n++], s);
3094 if (!ephemeral)
3095 IOVEC_SET_STRING(iovec[n++], "\n");
3096
3097 if (writev(fd, iovec, n) < 0)
3098 return -errno;
3099
3100 return 0;
3101 }
3102
3103 int status_printf(const char *status, bool ellipse, bool ephemeral, const char *format, ...) {
3104 va_list ap;
3105 int r;
3106
3107 assert(format);
3108
3109 va_start(ap, format);
3110 r = status_vprintf(status, ellipse, ephemeral, format, ap);
3111 va_end(ap);
3112
3113 return r;
3114 }
3115
3116 char *replace_env(const char *format, char **env) {
3117 enum {
3118 WORD,
3119 CURLY,
3120 VARIABLE
3121 } state = WORD;
3122
3123 const char *e, *word = format;
3124 char *r = NULL, *k;
3125
3126 assert(format);
3127
3128 for (e = format; *e; e ++) {
3129
3130 switch (state) {
3131
3132 case WORD:
3133 if (*e == '$')
3134 state = CURLY;
3135 break;
3136
3137 case CURLY:
3138 if (*e == '{') {
3139 if (!(k = strnappend(r, word, e-word-1)))
3140 goto fail;
3141
3142 free(r);
3143 r = k;
3144
3145 word = e-1;
3146 state = VARIABLE;
3147
3148 } else if (*e == '$') {
3149 if (!(k = strnappend(r, word, e-word)))
3150 goto fail;
3151
3152 free(r);
3153 r = k;
3154
3155 word = e+1;
3156 state = WORD;
3157 } else
3158 state = WORD;
3159 break;
3160
3161 case VARIABLE:
3162 if (*e == '}') {
3163 const char *t;
3164
3165 t = strempty(strv_env_get_n(env, word+2, e-word-2));
3166
3167 k = strappend(r, t);
3168 if (!k)
3169 goto fail;
3170
3171 free(r);
3172 r = k;
3173
3174 word = e+1;
3175 state = WORD;
3176 }
3177 break;
3178 }
3179 }
3180
3181 if (!(k = strnappend(r, word, e-word)))
3182 goto fail;
3183
3184 free(r);
3185 return k;
3186
3187 fail:
3188 free(r);
3189 return NULL;
3190 }
3191
3192 char **replace_env_argv(char **argv, char **env) {
3193 char **ret, **i;
3194 unsigned k = 0, l = 0;
3195
3196 l = strv_length(argv);
3197
3198 ret = new(char*, l+1);
3199 if (!ret)
3200 return NULL;
3201
3202 STRV_FOREACH(i, argv) {
3203
3204 /* If $FOO appears as single word, replace it by the split up variable */
3205 if ((*i)[0] == '$' && (*i)[1] != '{') {
3206 char *e;
3207 char **w, **m;
3208 unsigned q;
3209
3210 e = strv_env_get(env, *i+1);
3211 if (e) {
3212 int r;
3213
3214 r = strv_split_quoted(&m, e);
3215 if (r < 0) {
3216 ret[k] = NULL;
3217 strv_free(ret);
3218 return NULL;
3219 }
3220 } else
3221 m = NULL;
3222
3223 q = strv_length(m);
3224 l = l + q - 1;
3225
3226 w = realloc(ret, sizeof(char*) * (l+1));
3227 if (!w) {
3228 ret[k] = NULL;
3229 strv_free(ret);
3230 strv_free(m);
3231 return NULL;
3232 }
3233
3234 ret = w;
3235 if (m) {
3236 memcpy(ret + k, m, q * sizeof(char*));
3237 free(m);
3238 }
3239
3240 k += q;
3241 continue;
3242 }
3243
3244 /* If ${FOO} appears as part of a word, replace it by the variable as-is */
3245 ret[k] = replace_env(*i, env);
3246 if (!ret[k]) {
3247 strv_free(ret);
3248 return NULL;
3249 }
3250 k++;
3251 }
3252
3253 ret[k] = NULL;
3254 return ret;
3255 }
3256
3257 int fd_columns(int fd) {
3258 struct winsize ws = {};
3259
3260 if (ioctl(fd, TIOCGWINSZ, &ws) < 0)
3261 return -errno;
3262
3263 if (ws.ws_col <= 0)
3264 return -EIO;
3265
3266 return ws.ws_col;
3267 }
3268
3269 unsigned columns(void) {
3270 const char *e;
3271 int c;
3272
3273 if (_likely_(cached_columns > 0))
3274 return cached_columns;
3275
3276 c = 0;
3277 e = getenv("COLUMNS");
3278 if (e)
3279 safe_atoi(e, &c);
3280
3281 if (c <= 0)
3282 c = fd_columns(STDOUT_FILENO);
3283
3284 if (c <= 0)
3285 c = 80;
3286
3287 cached_columns = c;
3288 return c;
3289 }
3290
3291 int fd_lines(int fd) {
3292 struct winsize ws = {};
3293
3294 if (ioctl(fd, TIOCGWINSZ, &ws) < 0)
3295 return -errno;
3296
3297 if (ws.ws_row <= 0)
3298 return -EIO;
3299
3300 return ws.ws_row;
3301 }
3302
3303 unsigned lines(void) {
3304 const char *e;
3305 unsigned l;
3306
3307 if (_likely_(cached_lines > 0))
3308 return cached_lines;
3309
3310 l = 0;
3311 e = getenv("LINES");
3312 if (e)
3313 safe_atou(e, &l);
3314
3315 if (l <= 0)
3316 l = fd_lines(STDOUT_FILENO);
3317
3318 if (l <= 0)
3319 l = 24;
3320
3321 cached_lines = l;
3322 return cached_lines;
3323 }
3324
3325 /* intended to be used as a SIGWINCH sighandler */
3326 void columns_lines_cache_reset(int signum) {
3327 cached_columns = 0;
3328 cached_lines = 0;
3329 }
3330
3331 bool on_tty(void) {
3332 static int cached_on_tty = -1;
3333
3334 if (_unlikely_(cached_on_tty < 0))
3335 cached_on_tty = isatty(STDOUT_FILENO) > 0;
3336
3337 return cached_on_tty;
3338 }
3339
3340 int files_same(const char *filea, const char *fileb) {
3341 struct stat a, b;
3342
3343 if (stat(filea, &a) < 0)
3344 return -errno;
3345
3346 if (stat(fileb, &b) < 0)
3347 return -errno;
3348
3349 return a.st_dev == b.st_dev &&
3350 a.st_ino == b.st_ino;
3351 }
3352
3353 int running_in_chroot(void) {
3354 int ret;
3355
3356 ret = files_same("/proc/1/root", "/");
3357 if (ret < 0)
3358 return ret;
3359
3360 return ret == 0;
3361 }
3362
3363 static char *ascii_ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3364 size_t x;
3365 char *r;
3366
3367 assert(s);
3368 assert(percent <= 100);
3369 assert(new_length >= 3);
3370
3371 if (old_length <= 3 || old_length <= new_length)
3372 return strndup(s, old_length);
3373
3374 r = new0(char, new_length+1);
3375 if (!r)
3376 return NULL;
3377
3378 x = (new_length * percent) / 100;
3379
3380 if (x > new_length - 3)
3381 x = new_length - 3;
3382
3383 memcpy(r, s, x);
3384 r[x] = '.';
3385 r[x+1] = '.';
3386 r[x+2] = '.';
3387 memcpy(r + x + 3,
3388 s + old_length - (new_length - x - 3),
3389 new_length - x - 3);
3390
3391 return r;
3392 }
3393
3394 char *ellipsize_mem(const char *s, size_t old_length, size_t new_length, unsigned percent) {
3395 size_t x;
3396 char *e;
3397 const char *i, *j;
3398 unsigned k, len, len2;
3399
3400 assert(s);
3401 assert(percent <= 100);
3402 assert(new_length >= 3);
3403
3404 /* if no multibyte characters use ascii_ellipsize_mem for speed */
3405 if (ascii_is_valid(s))
3406 return ascii_ellipsize_mem(s, old_length, new_length, percent);
3407
3408 if (old_length <= 3 || old_length <= new_length)
3409 return strndup(s, old_length);
3410
3411 x = (new_length * percent) / 100;
3412
3413 if (x > new_length - 3)
3414 x = new_length - 3;
3415
3416 k = 0;
3417 for (i = s; k < x && i < s + old_length; i = utf8_next_char(i)) {
3418 int c;
3419
3420 c = utf8_encoded_to_unichar(i);
3421 if (c < 0)
3422 return NULL;
3423 k += unichar_iswide(c) ? 2 : 1;
3424 }
3425
3426 if (k > x) /* last character was wide and went over quota */
3427 x ++;
3428
3429 for (j = s + old_length; k < new_length && j > i; ) {
3430 int c;
3431
3432 j = utf8_prev_char(j);
3433 c = utf8_encoded_to_unichar(j);
3434 if (c < 0)
3435 return NULL;
3436 k += unichar_iswide(c) ? 2 : 1;
3437 }
3438 assert(i <= j);
3439
3440 /* we don't actually need to ellipsize */
3441 if (i == j)
3442 return memdup(s, old_length + 1);
3443
3444 /* make space for ellipsis */
3445 j = utf8_next_char(j);
3446
3447 len = i - s;
3448 len2 = s + old_length - j;
3449 e = new(char, len + 3 + len2 + 1);
3450 if (!e)
3451 return NULL;
3452
3453 /*
3454 printf("old_length=%zu new_length=%zu x=%zu len=%u len2=%u k=%u\n",
3455 old_length, new_length, x, len, len2, k);
3456 */
3457
3458 memcpy(e, s, len);
3459 e[len] = 0xe2; /* tri-dot ellipsis: … */
3460 e[len + 1] = 0x80;
3461 e[len + 2] = 0xa6;
3462
3463 memcpy(e + len + 3, j, len2 + 1);
3464
3465 return e;
3466 }
3467
3468 char *ellipsize(const char *s, size_t length, unsigned percent) {
3469 return ellipsize_mem(s, strlen(s), length, percent);
3470 }
3471
3472 int touch_file(const char *path, bool parents, usec_t stamp, uid_t uid, gid_t gid, mode_t mode) {
3473 _cleanup_close_ int fd;
3474 int r;
3475
3476 assert(path);
3477
3478 if (parents)
3479 mkdir_parents(path, 0755);
3480
3481 fd = open(path, O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY, mode > 0 ? mode : 0644);
3482 if (fd < 0)
3483 return -errno;
3484
3485 if (mode > 0) {
3486 r = fchmod(fd, mode);
3487 if (r < 0)
3488 return -errno;
3489 }
3490
3491 if (uid != (uid_t) -1 || gid != (gid_t) -1) {
3492 r = fchown(fd, uid, gid);
3493 if (r < 0)
3494 return -errno;
3495 }
3496
3497 if (stamp != USEC_INFINITY) {
3498 struct timespec ts[2];
3499
3500 timespec_store(&ts[0], stamp);
3501 ts[1] = ts[0];
3502 r = futimens(fd, ts);
3503 } else
3504 r = futimens(fd, NULL);
3505 if (r < 0)
3506 return -errno;
3507
3508 return 0;
3509 }
3510
3511 int touch(const char *path) {
3512 return touch_file(path, false, USEC_INFINITY, (uid_t) -1, (gid_t) -1, 0);
3513 }
3514
3515 char *unquote(const char *s, const char* quotes) {
3516 size_t l;
3517 assert(s);
3518
3519 /* This is rather stupid, simply removes the heading and
3520 * trailing quotes if there is one. Doesn't care about
3521 * escaping or anything. We should make this smarter one
3522 * day...*/
3523
3524 l = strlen(s);
3525 if (l < 2)
3526 return strdup(s);
3527
3528 if (strchr(quotes, s[0]) && s[l-1] == s[0])
3529 return strndup(s+1, l-2);
3530
3531 return strdup(s);
3532 }
3533
3534 char *normalize_env_assignment(const char *s) {
3535 _cleanup_free_ char *name = NULL, *value = NULL, *p = NULL;
3536 char *eq, *r;
3537
3538 eq = strchr(s, '=');
3539 if (!eq) {
3540 char *t;
3541
3542 r = strdup(s);
3543 if (!r)
3544 return NULL;
3545
3546 t = strstrip(r);
3547 if (t == r)
3548 return r;
3549
3550 memmove(r, t, strlen(t) + 1);
3551 return r;
3552 }
3553
3554 name = strndup(s, eq - s);
3555 if (!name)
3556 return NULL;
3557
3558 p = strdup(eq + 1);
3559 if (!p)
3560 return NULL;
3561
3562 value = unquote(strstrip(p), QUOTES);
3563 if (!value)
3564 return NULL;
3565
3566 if (asprintf(&r, "%s=%s", strstrip(name), value) < 0)
3567 r = NULL;
3568
3569 return r;
3570 }
3571
3572 int wait_for_terminate(pid_t pid, siginfo_t *status) {
3573 siginfo_t dummy;
3574
3575 assert(pid >= 1);
3576
3577 if (!status)
3578 status = &dummy;
3579
3580 for (;;) {
3581 zero(*status);
3582
3583 if (waitid(P_PID, pid, status, WEXITED) < 0) {
3584
3585 if (errno == EINTR)
3586 continue;
3587
3588 return -errno;
3589 }
3590
3591 return 0;
3592 }
3593 }
3594
3595 /*
3596 * Return values:
3597 * < 0 : wait_for_terminate() failed to get the state of the
3598 * process, the process was terminated by a signal, or
3599 * failed for an unknown reason.
3600 * >=0 : The process terminated normally, and its exit code is
3601 * returned.
3602 *
3603 * That is, success is indicated by a return value of zero, and an
3604 * error is indicated by a non-zero value.
3605 */
3606 int wait_for_terminate_and_warn(const char *name, pid_t pid) {
3607 int r;
3608 siginfo_t status;
3609
3610 assert(name);
3611 assert(pid > 1);
3612
3613 r = wait_for_terminate(pid, &status);
3614 if (r < 0) {
3615 log_warning("Failed to wait for %s: %s", name, strerror(-r));
3616 return r;
3617 }
3618
3619 if (status.si_code == CLD_EXITED) {
3620 if (status.si_status != 0) {
3621 log_warning("%s failed with error code %i.", name, status.si_status);
3622 return status.si_status;
3623 }
3624
3625 log_debug("%s succeeded.", name);
3626 return 0;
3627
3628 } else if (status.si_code == CLD_KILLED ||
3629 status.si_code == CLD_DUMPED) {
3630
3631 log_warning("%s terminated by signal %s.", name, signal_to_string(status.si_status));
3632 return -EPROTO;
3633 }
3634
3635 log_warning("%s failed due to unknown reason.", name);
3636 return -EPROTO;
3637 }
3638
3639 noreturn void freeze(void) {
3640
3641 /* Make sure nobody waits for us on a socket anymore */
3642 close_all_fds(NULL, 0);
3643
3644 sync();
3645
3646 for (;;)
3647 pause();
3648 }
3649
3650 bool null_or_empty(struct stat *st) {
3651 assert(st);
3652
3653 if (S_ISREG(st->st_mode) && st->st_size <= 0)
3654 return true;
3655
3656 if (S_ISCHR(st->st_mode) || S_ISBLK(st->st_mode))
3657 return true;
3658
3659 return false;
3660 }
3661
3662 int null_or_empty_path(const char *fn) {
3663 struct stat st;
3664
3665 assert(fn);
3666
3667 if (stat(fn, &st) < 0)
3668 return -errno;
3669
3670 return null_or_empty(&st);
3671 }
3672
3673 int null_or_empty_fd(int fd) {
3674 struct stat st;
3675
3676 assert(fd >= 0);
3677
3678 if (fstat(fd, &st) < 0)
3679 return -errno;
3680
3681 return null_or_empty(&st);
3682 }
3683
3684 DIR *xopendirat(int fd, const char *name, int flags) {
3685 int nfd;
3686 DIR *d;
3687
3688 assert(!(flags & O_CREAT));
3689
3690 nfd = openat(fd, name, O_RDONLY|O_NONBLOCK|O_DIRECTORY|O_CLOEXEC|flags, 0);
3691 if (nfd < 0)
3692 return NULL;
3693
3694 d = fdopendir(nfd);
3695 if (!d) {
3696 safe_close(nfd);
3697 return NULL;
3698 }
3699
3700 return d;
3701 }
3702
3703 int signal_from_string_try_harder(const char *s) {
3704 int signo;
3705 assert(s);
3706
3707 signo = signal_from_string(s);
3708 if (signo <= 0)
3709 if (startswith(s, "SIG"))
3710 return signal_from_string(s+3);
3711
3712 return signo;
3713 }
3714
3715 static char *tag_to_udev_node(const char *tagvalue, const char *by) {
3716 _cleanup_free_ char *t = NULL, *u = NULL;
3717 size_t enc_len;
3718
3719 u = unquote(tagvalue, "\"\'");
3720 if (!u)
3721 return NULL;
3722
3723 enc_len = strlen(u) * 4 + 1;
3724 t = new(char, enc_len);
3725 if (!t)
3726 return NULL;
3727
3728 if (encode_devnode_name(u, t, enc_len) < 0)
3729 return NULL;
3730
3731 return strjoin("/dev/disk/by-", by, "/", t, NULL);
3732 }
3733
3734 char *fstab_node_to_udev_node(const char *p) {
3735 assert(p);
3736
3737 if (startswith(p, "LABEL="))
3738 return tag_to_udev_node(p+6, "label");
3739
3740 if (startswith(p, "UUID="))
3741 return tag_to_udev_node(p+5, "uuid");
3742
3743 if (startswith(p, "PARTUUID="))
3744 return tag_to_udev_node(p+9, "partuuid");
3745
3746 if (startswith(p, "PARTLABEL="))
3747 return tag_to_udev_node(p+10, "partlabel");
3748
3749 return strdup(p);
3750 }
3751
3752 bool tty_is_vc(const char *tty) {
3753 assert(tty);
3754
3755 return vtnr_from_tty(tty) >= 0;
3756 }
3757
3758 bool tty_is_console(const char *tty) {
3759 assert(tty);
3760
3761 if (startswith(tty, "/dev/"))
3762 tty += 5;
3763
3764 return streq(tty, "console");
3765 }
3766
3767 int vtnr_from_tty(const char *tty) {
3768 int i, r;
3769
3770 assert(tty);
3771
3772 if (startswith(tty, "/dev/"))
3773 tty += 5;
3774
3775 if (!startswith(tty, "tty") )
3776 return -EINVAL;
3777
3778 if (tty[3] < '0' || tty[3] > '9')
3779 return -EINVAL;
3780
3781 r = safe_atoi(tty+3, &i);
3782 if (r < 0)
3783 return r;
3784
3785 if (i < 0 || i > 63)
3786 return -EINVAL;
3787
3788 return i;
3789 }
3790
3791 char *resolve_dev_console(char **active) {
3792 char *tty;
3793
3794 /* Resolve where /dev/console is pointing to, if /sys is actually ours
3795 * (i.e. not read-only-mounted which is a sign for container setups) */
3796
3797 if (path_is_read_only_fs("/sys") > 0)
3798 return NULL;
3799
3800 if (read_one_line_file("/sys/class/tty/console/active", active) < 0)
3801 return NULL;
3802
3803 /* If multiple log outputs are configured the last one is what
3804 * /dev/console points to */
3805 tty = strrchr(*active, ' ');
3806 if (tty)
3807 tty++;
3808 else
3809 tty = *active;
3810
3811 if (streq(tty, "tty0")) {
3812 char *tmp;
3813
3814 /* Get the active VC (e.g. tty1) */
3815 if (read_one_line_file("/sys/class/tty/tty0/active", &tmp) >= 0) {
3816 free(*active);
3817 tty = *active = tmp;
3818 }
3819 }
3820
3821 return tty;
3822 }
3823
3824 bool tty_is_vc_resolve(const char *tty) {
3825 _cleanup_free_ char *active = NULL;
3826
3827 assert(tty);
3828
3829 if (startswith(tty, "/dev/"))
3830 tty += 5;
3831
3832 if (streq(tty, "console")) {
3833 tty = resolve_dev_console(&active);
3834 if (!tty)
3835 return false;
3836 }
3837
3838 return tty_is_vc(tty);
3839 }
3840
3841 const char *default_term_for_tty(const char *tty) {
3842 assert(tty);
3843
3844 return tty_is_vc_resolve(tty) ? "TERM=linux" : "TERM=vt102";
3845 }
3846
3847 bool dirent_is_file(const struct dirent *de) {
3848 assert(de);
3849
3850 if (ignore_file(de->d_name))
3851 return false;
3852
3853 if (de->d_type != DT_REG &&
3854 de->d_type != DT_LNK &&
3855 de->d_type != DT_UNKNOWN)
3856 return false;
3857
3858 return true;
3859 }
3860
3861 bool dirent_is_file_with_suffix(const struct dirent *de, const char *suffix) {
3862 assert(de);
3863
3864 if (de->d_type != DT_REG &&
3865 de->d_type != DT_LNK &&
3866 de->d_type != DT_UNKNOWN)
3867 return false;
3868
3869 if (ignore_file_allow_backup(de->d_name))
3870 return false;
3871
3872 return endswith(de->d_name, suffix);
3873 }
3874
3875 void execute_directory(const char *directory, DIR *d, usec_t timeout, char *argv[]) {
3876 pid_t executor_pid;
3877 int r;
3878
3879 assert(directory);
3880
3881 /* Executes all binaries in a directory in parallel and waits
3882 * for them to finish. Optionally a timeout is applied. */
3883
3884 executor_pid = fork();
3885 if (executor_pid < 0) {
3886 log_error("Failed to fork: %m");
3887 return;
3888
3889 } else if (executor_pid == 0) {
3890 _cleanup_hashmap_free_free_ Hashmap *pids = NULL;
3891 _cleanup_closedir_ DIR *_d = NULL;
3892 struct dirent *de;
3893
3894 /* We fork this all off from a child process so that
3895 * we can somewhat cleanly make use of SIGALRM to set
3896 * a time limit */
3897
3898 reset_all_signal_handlers();
3899 reset_signal_mask();
3900
3901 assert_se(prctl(PR_SET_PDEATHSIG, SIGTERM) == 0);
3902
3903 if (!d) {
3904 d = _d = opendir(directory);
3905 if (!d) {
3906 if (errno == ENOENT)
3907 _exit(EXIT_SUCCESS);
3908
3909 log_error("Failed to enumerate directory %s: %m", directory);
3910 _exit(EXIT_FAILURE);
3911 }
3912 }
3913
3914 pids = hashmap_new(NULL, NULL);
3915 if (!pids) {
3916 log_oom();
3917 _exit(EXIT_FAILURE);
3918 }
3919
3920 FOREACH_DIRENT(de, d, break) {
3921 _cleanup_free_ char *path = NULL;
3922 pid_t pid;
3923
3924 if (!dirent_is_file(de))
3925 continue;
3926
3927 path = strjoin(directory, "/", de->d_name, NULL);
3928 if (!path) {
3929 log_oom();
3930 _exit(EXIT_FAILURE);
3931 }
3932
3933 pid = fork();
3934 if (pid < 0) {
3935 log_error("Failed to fork: %m");
3936 continue;
3937 } else if (pid == 0) {
3938 char *_argv[2];
3939
3940 assert_se(prctl(PR_SET_PDEATHSIG, SIGTERM) == 0);
3941
3942 if (!argv) {
3943 _argv[0] = path;
3944 _argv[1] = NULL;
3945 argv = _argv;
3946 } else
3947 argv[0] = path;
3948
3949 execv(path, argv);
3950 log_error("Failed to execute %s: %m", path);
3951 _exit(EXIT_FAILURE);
3952 }
3953
3954 log_debug("Spawned %s as " PID_FMT ".", path, pid);
3955
3956 r = hashmap_put(pids, UINT_TO_PTR(pid), path);
3957 if (r < 0) {
3958 log_oom();
3959 _exit(EXIT_FAILURE);
3960 }
3961
3962 path = NULL;
3963 }
3964
3965 /* Abort execution of this process after the
3966 * timout. We simply rely on SIGALRM as default action
3967 * terminating the process, and turn on alarm(). */
3968
3969 if (timeout != USEC_INFINITY)
3970 alarm((timeout + USEC_PER_SEC - 1) / USEC_PER_SEC);
3971
3972 while (!hashmap_isempty(pids)) {
3973 _cleanup_free_ char *path = NULL;
3974 pid_t pid;
3975
3976 pid = PTR_TO_UINT(hashmap_first_key(pids));
3977 assert(pid > 0);
3978
3979 path = hashmap_remove(pids, UINT_TO_PTR(pid));
3980 assert(path);
3981
3982 wait_for_terminate_and_warn(path, pid);
3983 }
3984
3985 _exit(EXIT_SUCCESS);
3986 }
3987
3988 wait_for_terminate_and_warn(directory, executor_pid);
3989 }
3990
3991 int kill_and_sigcont(pid_t pid, int sig) {
3992 int r;
3993
3994 r = kill(pid, sig) < 0 ? -errno : 0;
3995
3996 if (r >= 0)
3997 kill(pid, SIGCONT);
3998
3999 return r;
4000 }
4001
4002 bool nulstr_contains(const char*nulstr, const char *needle) {
4003 const char *i;
4004
4005 if (!nulstr)
4006 return false;
4007
4008 NULSTR_FOREACH(i, nulstr)
4009 if (streq(i, needle))
4010 return true;
4011
4012 return false;
4013 }
4014
4015 bool plymouth_running(void) {
4016 return access("/run/plymouth/pid", F_OK) >= 0;
4017 }
4018
4019 char* strshorten(char *s, size_t l) {
4020 assert(s);
4021
4022 if (l < strlen(s))
4023 s[l] = 0;
4024
4025 return s;
4026 }
4027
4028 static bool hostname_valid_char(char c) {
4029 return
4030 (c >= 'a' && c <= 'z') ||
4031 (c >= 'A' && c <= 'Z') ||
4032 (c >= '0' && c <= '9') ||
4033 c == '-' ||
4034 c == '_' ||
4035 c == '.';
4036 }
4037
4038 bool hostname_is_valid(const char *s) {
4039 const char *p;
4040 bool dot;
4041
4042 if (isempty(s))
4043 return false;
4044
4045 for (p = s, dot = true; *p; p++) {
4046 if (*p == '.') {
4047 if (dot)
4048 return false;
4049
4050 dot = true;
4051 } else {
4052 if (!hostname_valid_char(*p))
4053 return false;
4054
4055 dot = false;
4056 }
4057 }
4058
4059 if (dot)
4060 return false;
4061
4062 if (p-s > HOST_NAME_MAX)
4063 return false;
4064
4065 return true;
4066 }
4067
4068 char* hostname_cleanup(char *s, bool lowercase) {
4069 char *p, *d;
4070 bool dot;
4071
4072 for (p = s, d = s, dot = true; *p; p++) {
4073 if (*p == '.') {
4074 if (dot)
4075 continue;
4076
4077 *(d++) = '.';
4078 dot = true;
4079 } else if (hostname_valid_char(*p)) {
4080 *(d++) = lowercase ? tolower(*p) : *p;
4081 dot = false;
4082 }
4083
4084 }
4085
4086 if (dot && d > s)
4087 d[-1] = 0;
4088 else
4089 *d = 0;
4090
4091 strshorten(s, HOST_NAME_MAX);
4092
4093 return s;
4094 }
4095
4096 bool machine_name_is_valid(const char *s) {
4097
4098 if (!hostname_is_valid(s))
4099 return false;
4100
4101 /* Machine names should be useful hostnames, but also be
4102 * useful in unit names, hence we enforce a stricter length
4103 * limitation. */
4104
4105 if (strlen(s) > 64)
4106 return false;
4107
4108 return true;
4109 }
4110
4111 int pipe_eof(int fd) {
4112 struct pollfd pollfd = {
4113 .fd = fd,
4114 .events = POLLIN|POLLHUP,
4115 };
4116
4117 int r;
4118
4119 r = poll(&pollfd, 1, 0);
4120 if (r < 0)
4121 return -errno;
4122
4123 if (r == 0)
4124 return 0;
4125
4126 return pollfd.revents & POLLHUP;
4127 }
4128
4129 int fd_wait_for_event(int fd, int event, usec_t t) {
4130
4131 struct pollfd pollfd = {
4132 .fd = fd,
4133 .events = event,
4134 };
4135
4136 struct timespec ts;
4137 int r;
4138
4139 r = ppoll(&pollfd, 1, t == USEC_INFINITY ? NULL : timespec_store(&ts, t), NULL);
4140 if (r < 0)
4141 return -errno;
4142
4143 if (r == 0)
4144 return 0;
4145
4146 return pollfd.revents;
4147 }
4148
4149 int fopen_temporary(const char *path, FILE **_f, char **_temp_path) {
4150 FILE *f;
4151 char *t;
4152 int fd;
4153
4154 assert(path);
4155 assert(_f);
4156 assert(_temp_path);
4157
4158 t = tempfn_xxxxxx(path);
4159 if (!t)
4160 return -ENOMEM;
4161
4162 fd = mkostemp_safe(t, O_WRONLY|O_CLOEXEC);
4163 if (fd < 0) {
4164 free(t);
4165 return -errno;
4166 }
4167
4168 f = fdopen(fd, "we");
4169 if (!f) {
4170 unlink(t);
4171 free(t);
4172 return -errno;
4173 }
4174
4175 *_f = f;
4176 *_temp_path = t;
4177
4178 return 0;
4179 }
4180
4181 int terminal_vhangup_fd(int fd) {
4182 assert(fd >= 0);
4183
4184 if (ioctl(fd, TIOCVHANGUP) < 0)
4185 return -errno;
4186
4187 return 0;
4188 }
4189
4190 int terminal_vhangup(const char *name) {
4191 _cleanup_close_ int fd;
4192
4193 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4194 if (fd < 0)
4195 return fd;
4196
4197 return terminal_vhangup_fd(fd);
4198 }
4199
4200 int vt_disallocate(const char *name) {
4201 int fd, r;
4202 unsigned u;
4203
4204 /* Deallocate the VT if possible. If not possible
4205 * (i.e. because it is the active one), at least clear it
4206 * entirely (including the scrollback buffer) */
4207
4208 if (!startswith(name, "/dev/"))
4209 return -EINVAL;
4210
4211 if (!tty_is_vc(name)) {
4212 /* So this is not a VT. I guess we cannot deallocate
4213 * it then. But let's at least clear the screen */
4214
4215 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4216 if (fd < 0)
4217 return fd;
4218
4219 loop_write(fd,
4220 "\033[r" /* clear scrolling region */
4221 "\033[H" /* move home */
4222 "\033[2J", /* clear screen */
4223 10, false);
4224 safe_close(fd);
4225
4226 return 0;
4227 }
4228
4229 if (!startswith(name, "/dev/tty"))
4230 return -EINVAL;
4231
4232 r = safe_atou(name+8, &u);
4233 if (r < 0)
4234 return r;
4235
4236 if (u <= 0)
4237 return -EINVAL;
4238
4239 /* Try to deallocate */
4240 fd = open_terminal("/dev/tty0", O_RDWR|O_NOCTTY|O_CLOEXEC);
4241 if (fd < 0)
4242 return fd;
4243
4244 r = ioctl(fd, VT_DISALLOCATE, u);
4245 safe_close(fd);
4246
4247 if (r >= 0)
4248 return 0;
4249
4250 if (errno != EBUSY)
4251 return -errno;
4252
4253 /* Couldn't deallocate, so let's clear it fully with
4254 * scrollback */
4255 fd = open_terminal(name, O_RDWR|O_NOCTTY|O_CLOEXEC);
4256 if (fd < 0)
4257 return fd;
4258
4259 loop_write(fd,
4260 "\033[r" /* clear scrolling region */
4261 "\033[H" /* move home */
4262 "\033[3J", /* clear screen including scrollback, requires Linux 2.6.40 */
4263 10, false);
4264 safe_close(fd);
4265
4266 return 0;
4267 }
4268
4269 int symlink_atomic(const char *from, const char *to) {
4270 _cleanup_free_ char *t = NULL;
4271
4272 assert(from);
4273 assert(to);
4274
4275 t = tempfn_random(to);
4276 if (!t)
4277 return -ENOMEM;
4278
4279 if (symlink(from, t) < 0)
4280 return -errno;
4281
4282 if (rename(t, to) < 0) {
4283 unlink_noerrno(t);
4284 return -errno;
4285 }
4286
4287 return 0;
4288 }
4289
4290 int mknod_atomic(const char *path, mode_t mode, dev_t dev) {
4291 _cleanup_free_ char *t = NULL;
4292
4293 assert(path);
4294
4295 t = tempfn_random(path);
4296 if (!t)
4297 return -ENOMEM;
4298
4299 if (mknod(t, mode, dev) < 0)
4300 return -errno;
4301
4302 if (rename(t, path) < 0) {
4303 unlink_noerrno(t);
4304 return -errno;
4305 }
4306
4307 return 0;
4308 }
4309
4310 int mkfifo_atomic(const char *path, mode_t mode) {
4311 _cleanup_free_ char *t = NULL;
4312
4313 assert(path);
4314
4315 t = tempfn_random(path);
4316 if (!t)
4317 return -ENOMEM;
4318
4319 if (mkfifo(t, mode) < 0)
4320 return -errno;
4321
4322 if (rename(t, path) < 0) {
4323 unlink_noerrno(t);
4324 return -errno;
4325 }
4326
4327 return 0;
4328 }
4329
4330 bool display_is_local(const char *display) {
4331 assert(display);
4332
4333 return
4334 display[0] == ':' &&
4335 display[1] >= '0' &&
4336 display[1] <= '9';
4337 }
4338
4339 int socket_from_display(const char *display, char **path) {
4340 size_t k;
4341 char *f, *c;
4342
4343 assert(display);
4344 assert(path);
4345
4346 if (!display_is_local(display))
4347 return -EINVAL;
4348
4349 k = strspn(display+1, "0123456789");
4350
4351 f = new(char, strlen("/tmp/.X11-unix/X") + k + 1);
4352 if (!f)
4353 return -ENOMEM;
4354
4355 c = stpcpy(f, "/tmp/.X11-unix/X");
4356 memcpy(c, display+1, k);
4357 c[k] = 0;
4358
4359 *path = f;
4360
4361 return 0;
4362 }
4363
4364 int get_user_creds(
4365 const char **username,
4366 uid_t *uid, gid_t *gid,
4367 const char **home,
4368 const char **shell) {
4369
4370 struct passwd *p;
4371 uid_t u;
4372
4373 assert(username);
4374 assert(*username);
4375
4376 /* We enforce some special rules for uid=0: in order to avoid
4377 * NSS lookups for root we hardcode its data. */
4378
4379 if (streq(*username, "root") || streq(*username, "0")) {
4380 *username = "root";
4381
4382 if (uid)
4383 *uid = 0;
4384
4385 if (gid)
4386 *gid = 0;
4387
4388 if (home)
4389 *home = "/root";
4390
4391 if (shell)
4392 *shell = "/bin/sh";
4393
4394 return 0;
4395 }
4396
4397 if (parse_uid(*username, &u) >= 0) {
4398 errno = 0;
4399 p = getpwuid(u);
4400
4401 /* If there are multiple users with the same id, make
4402 * sure to leave $USER to the configured value instead
4403 * of the first occurrence in the database. However if
4404 * the uid was configured by a numeric uid, then let's
4405 * pick the real username from /etc/passwd. */
4406 if (p)
4407 *username = p->pw_name;
4408 } else {
4409 errno = 0;
4410 p = getpwnam(*username);
4411 }
4412
4413 if (!p)
4414 return errno > 0 ? -errno : -ESRCH;
4415
4416 if (uid)
4417 *uid = p->pw_uid;
4418
4419 if (gid)
4420 *gid = p->pw_gid;
4421
4422 if (home)
4423 *home = p->pw_dir;
4424
4425 if (shell)
4426 *shell = p->pw_shell;
4427
4428 return 0;
4429 }
4430
4431 char* uid_to_name(uid_t uid) {
4432 struct passwd *p;
4433 char *r;
4434
4435 if (uid == 0)
4436 return strdup("root");
4437
4438 p = getpwuid(uid);
4439 if (p)
4440 return strdup(p->pw_name);
4441
4442 if (asprintf(&r, UID_FMT, uid) < 0)
4443 return NULL;
4444
4445 return r;
4446 }
4447
4448 char* gid_to_name(gid_t gid) {
4449 struct group *p;
4450 char *r;
4451
4452 if (gid == 0)
4453 return strdup("root");
4454
4455 p = getgrgid(gid);
4456 if (p)
4457 return strdup(p->gr_name);
4458
4459 if (asprintf(&r, GID_FMT, gid) < 0)
4460 return NULL;
4461
4462 return r;
4463 }
4464
4465 int get_group_creds(const char **groupname, gid_t *gid) {
4466 struct group *g;
4467 gid_t id;
4468
4469 assert(groupname);
4470
4471 /* We enforce some special rules for gid=0: in order to avoid
4472 * NSS lookups for root we hardcode its data. */
4473
4474 if (streq(*groupname, "root") || streq(*groupname, "0")) {
4475 *groupname = "root";
4476
4477 if (gid)
4478 *gid = 0;
4479
4480 return 0;
4481 }
4482
4483 if (parse_gid(*groupname, &id) >= 0) {
4484 errno = 0;
4485 g = getgrgid(id);
4486
4487 if (g)
4488 *groupname = g->gr_name;
4489 } else {
4490 errno = 0;
4491 g = getgrnam(*groupname);
4492 }
4493
4494 if (!g)
4495 return errno > 0 ? -errno : -ESRCH;
4496
4497 if (gid)
4498 *gid = g->gr_gid;
4499
4500 return 0;
4501 }
4502
4503 int in_gid(gid_t gid) {
4504 gid_t *gids;
4505 int ngroups_max, r, i;
4506
4507 if (getgid() == gid)
4508 return 1;
4509
4510 if (getegid() == gid)
4511 return 1;
4512
4513 ngroups_max = sysconf(_SC_NGROUPS_MAX);
4514 assert(ngroups_max > 0);
4515
4516 gids = alloca(sizeof(gid_t) * ngroups_max);
4517
4518 r = getgroups(ngroups_max, gids);
4519 if (r < 0)
4520 return -errno;
4521
4522 for (i = 0; i < r; i++)
4523 if (gids[i] == gid)
4524 return 1;
4525
4526 return 0;
4527 }
4528
4529 int in_group(const char *name) {
4530 int r;
4531 gid_t gid;
4532
4533 r = get_group_creds(&name, &gid);
4534 if (r < 0)
4535 return r;
4536
4537 return in_gid(gid);
4538 }
4539
4540 int glob_exists(const char *path) {
4541 _cleanup_globfree_ glob_t g = {};
4542 int k;
4543
4544 assert(path);
4545
4546 errno = 0;
4547 k = glob(path, GLOB_NOSORT|GLOB_BRACE, NULL, &g);
4548
4549 if (k == GLOB_NOMATCH)
4550 return 0;
4551 else if (k == GLOB_NOSPACE)
4552 return -ENOMEM;
4553 else if (k == 0)
4554 return !strv_isempty(g.gl_pathv);
4555 else
4556 return errno ? -errno : -EIO;
4557 }
4558
4559 int glob_extend(char ***strv, const char *path) {
4560 _cleanup_globfree_ glob_t g = {};
4561 int k;
4562 char **p;
4563
4564 errno = 0;
4565 k = glob(path, GLOB_NOSORT|GLOB_BRACE, NULL, &g);
4566
4567 if (k == GLOB_NOMATCH)
4568 return -ENOENT;
4569 else if (k == GLOB_NOSPACE)
4570 return -ENOMEM;
4571 else if (k != 0 || strv_isempty(g.gl_pathv))
4572 return errno ? -errno : -EIO;
4573
4574 STRV_FOREACH(p, g.gl_pathv) {
4575 k = strv_extend(strv, *p);
4576 if (k < 0)
4577 break;
4578 }
4579
4580 return k;
4581 }
4582
4583 int dirent_ensure_type(DIR *d, struct dirent *de) {
4584 struct stat st;
4585
4586 assert(d);
4587 assert(de);
4588
4589 if (de->d_type != DT_UNKNOWN)
4590 return 0;
4591
4592 if (fstatat(dirfd(d), de->d_name, &st, AT_SYMLINK_NOFOLLOW) < 0)
4593 return -errno;
4594
4595 de->d_type =
4596 S_ISREG(st.st_mode) ? DT_REG :
4597 S_ISDIR(st.st_mode) ? DT_DIR :
4598 S_ISLNK(st.st_mode) ? DT_LNK :
4599 S_ISFIFO(st.st_mode) ? DT_FIFO :
4600 S_ISSOCK(st.st_mode) ? DT_SOCK :
4601 S_ISCHR(st.st_mode) ? DT_CHR :
4602 S_ISBLK(st.st_mode) ? DT_BLK :
4603 DT_UNKNOWN;
4604
4605 return 0;
4606 }
4607
4608 int get_files_in_directory(const char *path, char ***list) {
4609 _cleanup_closedir_ DIR *d = NULL;
4610 size_t bufsize = 0, n = 0;
4611 _cleanup_strv_free_ char **l = NULL;
4612
4613 assert(path);
4614
4615 /* Returns all files in a directory in *list, and the number
4616 * of files as return value. If list is NULL returns only the
4617 * number. */
4618
4619 d = opendir(path);
4620 if (!d)
4621 return -errno;
4622
4623 for (;;) {
4624 struct dirent *de;
4625
4626 errno = 0;
4627 de = readdir(d);
4628 if (!de && errno != 0)
4629 return -errno;
4630 if (!de)
4631 break;
4632
4633 dirent_ensure_type(d, de);
4634
4635 if (!dirent_is_file(de))
4636 continue;
4637
4638 if (list) {
4639 /* one extra slot is needed for the terminating NULL */
4640 if (!GREEDY_REALLOC(l, bufsize, n + 2))
4641 return -ENOMEM;
4642
4643 l[n] = strdup(de->d_name);
4644 if (!l[n])
4645 return -ENOMEM;
4646
4647 l[++n] = NULL;
4648 } else
4649 n++;
4650 }
4651
4652 if (list) {
4653 *list = l;
4654 l = NULL; /* avoid freeing */
4655 }
4656
4657 return n;
4658 }
4659
4660 char *strjoin(const char *x, ...) {
4661 va_list ap;
4662 size_t l;
4663 char *r, *p;
4664
4665 va_start(ap, x);
4666
4667 if (x) {
4668 l = strlen(x);
4669
4670 for (;;) {
4671 const char *t;
4672 size_t n;
4673
4674 t = va_arg(ap, const char *);
4675 if (!t)
4676 break;
4677
4678 n = strlen(t);
4679 if (n > ((size_t) -1) - l) {
4680 va_end(ap);
4681 return NULL;
4682 }
4683
4684 l += n;
4685 }
4686 } else
4687 l = 0;
4688
4689 va_end(ap);
4690
4691 r = new(char, l+1);
4692 if (!r)
4693 return NULL;
4694
4695 if (x) {
4696 p = stpcpy(r, x);
4697
4698 va_start(ap, x);
4699
4700 for (;;) {
4701 const char *t;
4702
4703 t = va_arg(ap, const char *);
4704 if (!t)
4705 break;
4706
4707 p = stpcpy(p, t);
4708 }
4709
4710 va_end(ap);
4711 } else
4712 r[0] = 0;
4713
4714 return r;
4715 }
4716
4717 bool is_main_thread(void) {
4718 static thread_local int cached = 0;
4719
4720 if (_unlikely_(cached == 0))
4721 cached = getpid() == gettid() ? 1 : -1;
4722
4723 return cached > 0;
4724 }
4725
4726 int block_get_whole_disk(dev_t d, dev_t *ret) {
4727 char *p, *s;
4728 int r;
4729 unsigned n, m;
4730
4731 assert(ret);
4732
4733 /* If it has a queue this is good enough for us */
4734 if (asprintf(&p, "/sys/dev/block/%u:%u/queue", major(d), minor(d)) < 0)
4735 return -ENOMEM;
4736
4737 r = access(p, F_OK);
4738 free(p);
4739
4740 if (r >= 0) {
4741 *ret = d;
4742 return 0;
4743 }
4744
4745 /* If it is a partition find the originating device */
4746 if (asprintf(&p, "/sys/dev/block/%u:%u/partition", major(d), minor(d)) < 0)
4747 return -ENOMEM;
4748
4749 r = access(p, F_OK);
4750 free(p);
4751
4752 if (r < 0)
4753 return -ENOENT;
4754
4755 /* Get parent dev_t */
4756 if (asprintf(&p, "/sys/dev/block/%u:%u/../dev", major(d), minor(d)) < 0)
4757 return -ENOMEM;
4758
4759 r = read_one_line_file(p, &s);
4760 free(p);
4761
4762 if (r < 0)
4763 return r;
4764
4765 r = sscanf(s, "%u:%u", &m, &n);
4766 free(s);
4767
4768 if (r != 2)
4769 return -EINVAL;
4770
4771 /* Only return this if it is really good enough for us. */
4772 if (asprintf(&p, "/sys/dev/block/%u:%u/queue", m, n) < 0)
4773 return -ENOMEM;
4774
4775 r = access(p, F_OK);
4776 free(p);
4777
4778 if (r >= 0) {
4779 *ret = makedev(m, n);
4780 return 0;
4781 }
4782
4783 return -ENOENT;
4784 }
4785
4786 int file_is_priv_sticky(const char *p) {
4787 struct stat st;
4788
4789 assert(p);
4790
4791 if (lstat(p, &st) < 0)
4792 return -errno;
4793
4794 return
4795 (st.st_uid == 0 || st.st_uid == getuid()) &&
4796 (st.st_mode & S_ISVTX);
4797 }
4798
4799 static const char *const ioprio_class_table[] = {
4800 [IOPRIO_CLASS_NONE] = "none",
4801 [IOPRIO_CLASS_RT] = "realtime",
4802 [IOPRIO_CLASS_BE] = "best-effort",
4803 [IOPRIO_CLASS_IDLE] = "idle"
4804 };
4805
4806 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(ioprio_class, int, INT_MAX);
4807
4808 static const char *const sigchld_code_table[] = {
4809 [CLD_EXITED] = "exited",
4810 [CLD_KILLED] = "killed",
4811 [CLD_DUMPED] = "dumped",
4812 [CLD_TRAPPED] = "trapped",
4813 [CLD_STOPPED] = "stopped",
4814 [CLD_CONTINUED] = "continued",
4815 };
4816
4817 DEFINE_STRING_TABLE_LOOKUP(sigchld_code, int);
4818
4819 static const char *const log_facility_unshifted_table[LOG_NFACILITIES] = {
4820 [LOG_FAC(LOG_KERN)] = "kern",
4821 [LOG_FAC(LOG_USER)] = "user",
4822 [LOG_FAC(LOG_MAIL)] = "mail",
4823 [LOG_FAC(LOG_DAEMON)] = "daemon",
4824 [LOG_FAC(LOG_AUTH)] = "auth",
4825 [LOG_FAC(LOG_SYSLOG)] = "syslog",
4826 [LOG_FAC(LOG_LPR)] = "lpr",
4827 [LOG_FAC(LOG_NEWS)] = "news",
4828 [LOG_FAC(LOG_UUCP)] = "uucp",
4829 [LOG_FAC(LOG_CRON)] = "cron",
4830 [LOG_FAC(LOG_AUTHPRIV)] = "authpriv",
4831 [LOG_FAC(LOG_FTP)] = "ftp",
4832 [LOG_FAC(LOG_LOCAL0)] = "local0",
4833 [LOG_FAC(LOG_LOCAL1)] = "local1",
4834 [LOG_FAC(LOG_LOCAL2)] = "local2",
4835 [LOG_FAC(LOG_LOCAL3)] = "local3",
4836 [LOG_FAC(LOG_LOCAL4)] = "local4",
4837 [LOG_FAC(LOG_LOCAL5)] = "local5",
4838 [LOG_FAC(LOG_LOCAL6)] = "local6",
4839 [LOG_FAC(LOG_LOCAL7)] = "local7"
4840 };
4841
4842 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(log_facility_unshifted, int, LOG_FAC(~0));
4843
4844 static const char *const log_level_table[] = {
4845 [LOG_EMERG] = "emerg",
4846 [LOG_ALERT] = "alert",
4847 [LOG_CRIT] = "crit",
4848 [LOG_ERR] = "err",
4849 [LOG_WARNING] = "warning",
4850 [LOG_NOTICE] = "notice",
4851 [LOG_INFO] = "info",
4852 [LOG_DEBUG] = "debug"
4853 };
4854
4855 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(log_level, int, LOG_DEBUG);
4856
4857 static const char* const sched_policy_table[] = {
4858 [SCHED_OTHER] = "other",
4859 [SCHED_BATCH] = "batch",
4860 [SCHED_IDLE] = "idle",
4861 [SCHED_FIFO] = "fifo",
4862 [SCHED_RR] = "rr"
4863 };
4864
4865 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(sched_policy, int, INT_MAX);
4866
4867 static const char* const rlimit_table[_RLIMIT_MAX] = {
4868 [RLIMIT_CPU] = "LimitCPU",
4869 [RLIMIT_FSIZE] = "LimitFSIZE",
4870 [RLIMIT_DATA] = "LimitDATA",
4871 [RLIMIT_STACK] = "LimitSTACK",
4872 [RLIMIT_CORE] = "LimitCORE",
4873 [RLIMIT_RSS] = "LimitRSS",
4874 [RLIMIT_NOFILE] = "LimitNOFILE",
4875 [RLIMIT_AS] = "LimitAS",
4876 [RLIMIT_NPROC] = "LimitNPROC",
4877 [RLIMIT_MEMLOCK] = "LimitMEMLOCK",
4878 [RLIMIT_LOCKS] = "LimitLOCKS",
4879 [RLIMIT_SIGPENDING] = "LimitSIGPENDING",
4880 [RLIMIT_MSGQUEUE] = "LimitMSGQUEUE",
4881 [RLIMIT_NICE] = "LimitNICE",
4882 [RLIMIT_RTPRIO] = "LimitRTPRIO",
4883 [RLIMIT_RTTIME] = "LimitRTTIME"
4884 };
4885
4886 DEFINE_STRING_TABLE_LOOKUP(rlimit, int);
4887
4888 static const char* const ip_tos_table[] = {
4889 [IPTOS_LOWDELAY] = "low-delay",
4890 [IPTOS_THROUGHPUT] = "throughput",
4891 [IPTOS_RELIABILITY] = "reliability",
4892 [IPTOS_LOWCOST] = "low-cost",
4893 };
4894
4895 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(ip_tos, int, 0xff);
4896
4897 static const char *const __signal_table[] = {
4898 [SIGHUP] = "HUP",
4899 [SIGINT] = "INT",
4900 [SIGQUIT] = "QUIT",
4901 [SIGILL] = "ILL",
4902 [SIGTRAP] = "TRAP",
4903 [SIGABRT] = "ABRT",
4904 [SIGBUS] = "BUS",
4905 [SIGFPE] = "FPE",
4906 [SIGKILL] = "KILL",
4907 [SIGUSR1] = "USR1",
4908 [SIGSEGV] = "SEGV",
4909 [SIGUSR2] = "USR2",
4910 [SIGPIPE] = "PIPE",
4911 [SIGALRM] = "ALRM",
4912 [SIGTERM] = "TERM",
4913 #ifdef SIGSTKFLT
4914 [SIGSTKFLT] = "STKFLT", /* Linux on SPARC doesn't know SIGSTKFLT */
4915 #endif
4916 [SIGCHLD] = "CHLD",
4917 [SIGCONT] = "CONT",
4918 [SIGSTOP] = "STOP",
4919 [SIGTSTP] = "TSTP",
4920 [SIGTTIN] = "TTIN",
4921 [SIGTTOU] = "TTOU",
4922 [SIGURG] = "URG",
4923 [SIGXCPU] = "XCPU",
4924 [SIGXFSZ] = "XFSZ",
4925 [SIGVTALRM] = "VTALRM",
4926 [SIGPROF] = "PROF",
4927 [SIGWINCH] = "WINCH",
4928 [SIGIO] = "IO",
4929 [SIGPWR] = "PWR",
4930 [SIGSYS] = "SYS"
4931 };
4932
4933 DEFINE_PRIVATE_STRING_TABLE_LOOKUP(__signal, int);
4934
4935 const char *signal_to_string(int signo) {
4936 static thread_local char buf[sizeof("RTMIN+")-1 + DECIMAL_STR_MAX(int) + 1];
4937 const char *name;
4938
4939 name = __signal_to_string(signo);
4940 if (name)
4941 return name;
4942
4943 if (signo >= SIGRTMIN && signo <= SIGRTMAX)
4944 snprintf(buf, sizeof(buf), "RTMIN+%d", signo - SIGRTMIN);
4945 else
4946 snprintf(buf, sizeof(buf), "%d", signo);
4947
4948 return buf;
4949 }
4950
4951 int signal_from_string(const char *s) {
4952 int signo;
4953 int offset = 0;
4954 unsigned u;
4955
4956 signo = __signal_from_string(s);
4957 if (signo > 0)
4958 return signo;
4959
4960 if (startswith(s, "RTMIN+")) {
4961 s += 6;
4962 offset = SIGRTMIN;
4963 }
4964 if (safe_atou(s, &u) >= 0) {
4965 signo = (int) u + offset;
4966 if (signo > 0 && signo < _NSIG)
4967 return signo;
4968 }
4969 return -EINVAL;
4970 }
4971
4972 bool kexec_loaded(void) {
4973 bool loaded = false;
4974 char *s;
4975
4976 if (read_one_line_file("/sys/kernel/kexec_loaded", &s) >= 0) {
4977 if (s[0] == '1')
4978 loaded = true;
4979 free(s);
4980 }
4981 return loaded;
4982 }
4983
4984 int strdup_or_null(const char *a, char **b) {
4985 char *c;
4986
4987 assert(b);
4988
4989 if (!a) {
4990 *b = NULL;
4991 return 0;
4992 }
4993
4994 c = strdup(a);
4995 if (!c)
4996 return -ENOMEM;
4997
4998 *b = c;
4999 return 0;
5000 }
5001
5002 int prot_from_flags(int flags) {
5003
5004 switch (flags & O_ACCMODE) {
5005
5006 case O_RDONLY:
5007 return PROT_READ;
5008
5009 case O_WRONLY:
5010 return PROT_WRITE;
5011
5012 case O_RDWR:
5013 return PROT_READ|PROT_WRITE;
5014
5015 default:
5016 return -EINVAL;
5017 }
5018 }
5019
5020 char *format_bytes(char *buf, size_t l, off_t t) {
5021 unsigned i;
5022
5023 static const struct {
5024 const char *suffix;
5025 off_t factor;
5026 } table[] = {
5027 { "E", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
5028 { "P", 1024ULL*1024ULL*1024ULL*1024ULL*1024ULL },
5029 { "T", 1024ULL*1024ULL*1024ULL*1024ULL },
5030 { "G", 1024ULL*1024ULL*1024ULL },
5031 { "M", 1024ULL*1024ULL },
5032 { "K", 1024ULL },
5033 };
5034
5035 for (i = 0; i < ELEMENTSOF(table); i++) {
5036
5037 if (t >= table[i].factor) {
5038 snprintf(buf, l,
5039 "%llu.%llu%s",
5040 (unsigned long long) (t / table[i].factor),
5041 (unsigned long long) (((t*10ULL) / table[i].factor) % 10ULL),
5042 table[i].suffix);
5043
5044 goto finish;
5045 }
5046 }
5047
5048 snprintf(buf, l, "%lluB", (unsigned long long) t);
5049
5050 finish:
5051 buf[l-1] = 0;
5052 return buf;
5053
5054 }
5055
5056 void* memdup(const void *p, size_t l) {
5057 void *r;
5058
5059 assert(p);
5060
5061 r = malloc(l);
5062 if (!r)
5063 return NULL;
5064
5065 memcpy(r, p, l);
5066 return r;
5067 }
5068
5069 int fd_inc_sndbuf(int fd, size_t n) {
5070 int r, value;
5071 socklen_t l = sizeof(value);
5072
5073 r = getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, &l);
5074 if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
5075 return 0;
5076
5077 /* If we have the privileges we will ignore the kernel limit. */
5078
5079 value = (int) n;
5080 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUFFORCE, &value, sizeof(value)) < 0)
5081 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value)) < 0)
5082 return -errno;
5083
5084 return 1;
5085 }
5086
5087 int fd_inc_rcvbuf(int fd, size_t n) {
5088 int r, value;
5089 socklen_t l = sizeof(value);
5090
5091 r = getsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, &l);
5092 if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
5093 return 0;
5094
5095 /* If we have the privileges we will ignore the kernel limit. */
5096
5097 value = (int) n;
5098 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUFFORCE, &value, sizeof(value)) < 0)
5099 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value)) < 0)
5100 return -errno;
5101 return 1;
5102 }
5103
5104 int fork_agent(pid_t *pid, const int except[], unsigned n_except, const char *path, ...) {
5105 bool stdout_is_tty, stderr_is_tty;
5106 pid_t parent_pid, agent_pid;
5107 sigset_t ss, saved_ss;
5108 unsigned n, i;
5109 va_list ap;
5110 char **l;
5111
5112 assert(pid);
5113 assert(path);
5114
5115 /* Spawns a temporary TTY agent, making sure it goes away when
5116 * we go away */
5117
5118 parent_pid = getpid();
5119
5120 /* First we temporarily block all signals, so that the new
5121 * child has them blocked initially. This way, we can be sure
5122 * that SIGTERMs are not lost we might send to the agent. */
5123 assert_se(sigfillset(&ss) >= 0);
5124 assert_se(sigprocmask(SIG_SETMASK, &ss, &saved_ss) >= 0);
5125
5126 agent_pid = fork();
5127 if (agent_pid < 0) {
5128 assert_se(sigprocmask(SIG_SETMASK, &saved_ss, NULL) >= 0);
5129 return -errno;
5130 }
5131
5132 if (agent_pid != 0) {
5133 assert_se(sigprocmask(SIG_SETMASK, &saved_ss, NULL) >= 0);
5134 *pid = agent_pid;
5135 return 0;
5136 }
5137
5138 /* In the child:
5139 *
5140 * Make sure the agent goes away when the parent dies */
5141 if (prctl(PR_SET_PDEATHSIG, SIGTERM) < 0)
5142 _exit(EXIT_FAILURE);
5143
5144 /* Make sure we actually can kill the agent, if we need to, in
5145 * case somebody invoked us from a shell script that trapped
5146 * SIGTERM or so... */
5147 reset_all_signal_handlers();
5148 reset_signal_mask();
5149
5150 /* Check whether our parent died before we were able
5151 * to set the death signal and unblock the signals */
5152 if (getppid() != parent_pid)
5153 _exit(EXIT_SUCCESS);
5154
5155 /* Don't leak fds to the agent */
5156 close_all_fds(except, n_except);
5157
5158 stdout_is_tty = isatty(STDOUT_FILENO);
5159 stderr_is_tty = isatty(STDERR_FILENO);
5160
5161 if (!stdout_is_tty || !stderr_is_tty) {
5162 int fd;
5163
5164 /* Detach from stdout/stderr. and reopen
5165 * /dev/tty for them. This is important to
5166 * ensure that when systemctl is started via
5167 * popen() or a similar call that expects to
5168 * read EOF we actually do generate EOF and
5169 * not delay this indefinitely by because we
5170 * keep an unused copy of stdin around. */
5171 fd = open("/dev/tty", O_WRONLY);
5172 if (fd < 0) {
5173 log_error("Failed to open /dev/tty: %m");
5174 _exit(EXIT_FAILURE);
5175 }
5176
5177 if (!stdout_is_tty)
5178 dup2(fd, STDOUT_FILENO);
5179
5180 if (!stderr_is_tty)
5181 dup2(fd, STDERR_FILENO);
5182
5183 if (fd > 2)
5184 close(fd);
5185 }
5186
5187 /* Count arguments */
5188 va_start(ap, path);
5189 for (n = 0; va_arg(ap, char*); n++)
5190 ;
5191 va_end(ap);
5192
5193 /* Allocate strv */
5194 l = alloca(sizeof(char *) * (n + 1));
5195
5196 /* Fill in arguments */
5197 va_start(ap, path);
5198 for (i = 0; i <= n; i++)
5199 l[i] = va_arg(ap, char*);
5200 va_end(ap);
5201
5202 execv(path, l);
5203 _exit(EXIT_FAILURE);
5204 }
5205
5206 int setrlimit_closest(int resource, const struct rlimit *rlim) {
5207 struct rlimit highest, fixed;
5208
5209 assert(rlim);
5210
5211 if (setrlimit(resource, rlim) >= 0)
5212 return 0;
5213
5214 if (errno != EPERM)
5215 return -errno;
5216
5217 /* So we failed to set the desired setrlimit, then let's try
5218 * to get as close as we can */
5219 assert_se(getrlimit(resource, &highest) == 0);
5220
5221 fixed.rlim_cur = MIN(rlim->rlim_cur, highest.rlim_max);
5222 fixed.rlim_max = MIN(rlim->rlim_max, highest.rlim_max);
5223
5224 if (setrlimit(resource, &fixed) < 0)
5225 return -errno;
5226
5227 return 0;
5228 }
5229
5230 int getenv_for_pid(pid_t pid, const char *field, char **_value) {
5231 _cleanup_fclose_ FILE *f = NULL;
5232 char *value = NULL;
5233 int r;
5234 bool done = false;
5235 size_t l;
5236 const char *path;
5237
5238 assert(pid >= 0);
5239 assert(field);
5240 assert(_value);
5241
5242 path = procfs_file_alloca(pid, "environ");
5243
5244 f = fopen(path, "re");
5245 if (!f)
5246 return -errno;
5247
5248 l = strlen(field);
5249 r = 0;
5250
5251 do {
5252 char line[LINE_MAX];
5253 unsigned i;
5254
5255 for (i = 0; i < sizeof(line)-1; i++) {
5256 int c;
5257
5258 c = getc(f);
5259 if (_unlikely_(c == EOF)) {
5260 done = true;
5261 break;
5262 } else if (c == 0)
5263 break;
5264
5265 line[i] = c;
5266 }
5267 line[i] = 0;
5268
5269 if (memcmp(line, field, l) == 0 && line[l] == '=') {
5270 value = strdup(line + l + 1);
5271 if (!value)
5272 return -ENOMEM;
5273
5274 r = 1;
5275 break;
5276 }
5277
5278 } while (!done);
5279
5280 *_value = value;
5281 return r;
5282 }
5283
5284 bool is_valid_documentation_url(const char *url) {
5285 assert(url);
5286
5287 if (startswith(url, "http://") && url[7])
5288 return true;
5289
5290 if (startswith(url, "https://") && url[8])
5291 return true;
5292
5293 if (startswith(url, "file:") && url[5])
5294 return true;
5295
5296 if (startswith(url, "info:") && url[5])
5297 return true;
5298
5299 if (startswith(url, "man:") && url[4])
5300 return true;
5301
5302 return false;
5303 }
5304
5305 bool in_initrd(void) {
5306 static int saved = -1;
5307 struct statfs s;
5308
5309 if (saved >= 0)
5310 return saved;
5311
5312 /* We make two checks here:
5313 *
5314 * 1. the flag file /etc/initrd-release must exist
5315 * 2. the root file system must be a memory file system
5316 *
5317 * The second check is extra paranoia, since misdetecting an
5318 * initrd can have bad bad consequences due the initrd
5319 * emptying when transititioning to the main systemd.
5320 */
5321
5322 saved = access("/etc/initrd-release", F_OK) >= 0 &&
5323 statfs("/", &s) >= 0 &&
5324 is_temporary_fs(&s);
5325
5326 return saved;
5327 }
5328
5329 void warn_melody(void) {
5330 _cleanup_close_ int fd = -1;
5331
5332 fd = open("/dev/console", O_WRONLY|O_CLOEXEC|O_NOCTTY);
5333 if (fd < 0)
5334 return;
5335
5336 /* Yeah, this is synchronous. Kinda sucks. But well... */
5337
5338 ioctl(fd, KIOCSOUND, (int)(1193180/440));
5339 usleep(125*USEC_PER_MSEC);
5340
5341 ioctl(fd, KIOCSOUND, (int)(1193180/220));
5342 usleep(125*USEC_PER_MSEC);
5343
5344 ioctl(fd, KIOCSOUND, (int)(1193180/220));
5345 usleep(125*USEC_PER_MSEC);
5346
5347 ioctl(fd, KIOCSOUND, 0);
5348 }
5349
5350 int make_console_stdio(void) {
5351 int fd, r;
5352
5353 /* Make /dev/console the controlling terminal and stdin/stdout/stderr */
5354
5355 fd = acquire_terminal("/dev/console", false, true, true, USEC_INFINITY);
5356 if (fd < 0) {
5357 log_error("Failed to acquire terminal: %s", strerror(-fd));
5358 return fd;
5359 }
5360
5361 r = make_stdio(fd);
5362 if (r < 0) {
5363 log_error("Failed to duplicate terminal fd: %s", strerror(-r));
5364 return r;
5365 }
5366
5367 return 0;
5368 }
5369
5370 int get_home_dir(char **_h) {
5371 struct passwd *p;
5372 const char *e;
5373 char *h;
5374 uid_t u;
5375
5376 assert(_h);
5377
5378 /* Take the user specified one */
5379 e = secure_getenv("HOME");
5380 if (e && path_is_absolute(e)) {
5381 h = strdup(e);
5382 if (!h)
5383 return -ENOMEM;
5384
5385 *_h = h;
5386 return 0;
5387 }
5388
5389 /* Hardcode home directory for root to avoid NSS */
5390 u = getuid();
5391 if (u == 0) {
5392 h = strdup("/root");
5393 if (!h)
5394 return -ENOMEM;
5395
5396 *_h = h;
5397 return 0;
5398 }
5399
5400 /* Check the database... */
5401 errno = 0;
5402 p = getpwuid(u);
5403 if (!p)
5404 return errno > 0 ? -errno : -ESRCH;
5405
5406 if (!path_is_absolute(p->pw_dir))
5407 return -EINVAL;
5408
5409 h = strdup(p->pw_dir);
5410 if (!h)
5411 return -ENOMEM;
5412
5413 *_h = h;
5414 return 0;
5415 }
5416
5417 int get_shell(char **_s) {
5418 struct passwd *p;
5419 const char *e;
5420 char *s;
5421 uid_t u;
5422
5423 assert(_s);
5424
5425 /* Take the user specified one */
5426 e = getenv("SHELL");
5427 if (e) {
5428 s = strdup(e);
5429 if (!s)
5430 return -ENOMEM;
5431
5432 *_s = s;
5433 return 0;
5434 }
5435
5436 /* Hardcode home directory for root to avoid NSS */
5437 u = getuid();
5438 if (u == 0) {
5439 s = strdup("/bin/sh");
5440 if (!s)
5441 return -ENOMEM;
5442
5443 *_s = s;
5444 return 0;
5445 }
5446
5447 /* Check the database... */
5448 errno = 0;
5449 p = getpwuid(u);
5450 if (!p)
5451 return errno > 0 ? -errno : -ESRCH;
5452
5453 if (!path_is_absolute(p->pw_shell))
5454 return -EINVAL;
5455
5456 s = strdup(p->pw_shell);
5457 if (!s)
5458 return -ENOMEM;
5459
5460 *_s = s;
5461 return 0;
5462 }
5463
5464 bool filename_is_safe(const char *p) {
5465
5466 if (isempty(p))
5467 return false;
5468
5469 if (strchr(p, '/'))
5470 return false;
5471
5472 if (streq(p, "."))
5473 return false;
5474
5475 if (streq(p, ".."))
5476 return false;
5477
5478 if (strlen(p) > FILENAME_MAX)
5479 return false;
5480
5481 return true;
5482 }
5483
5484 bool string_is_safe(const char *p) {
5485 const char *t;
5486
5487 if (!p)
5488 return false;
5489
5490 for (t = p; *t; t++) {
5491 if (*t > 0 && *t < ' ')
5492 return false;
5493
5494 if (strchr("\\\"\'\0x7f", *t))
5495 return false;
5496 }
5497
5498 return true;
5499 }
5500
5501 /**
5502 * Check if a string contains control characters. If 'ok' is non-NULL
5503 * it may be a string containing additional CCs to be considered OK.
5504 */
5505 bool string_has_cc(const char *p, const char *ok) {
5506 const char *t;
5507
5508 assert(p);
5509
5510 for (t = p; *t; t++) {
5511 if (ok && strchr(ok, *t))
5512 continue;
5513
5514 if (*t > 0 && *t < ' ')
5515 return true;
5516
5517 if (*t == 127)
5518 return true;
5519 }
5520
5521 return false;
5522 }
5523
5524 bool path_is_safe(const char *p) {
5525
5526 if (isempty(p))
5527 return false;
5528
5529 if (streq(p, "..") || startswith(p, "../") || endswith(p, "/..") || strstr(p, "/../"))
5530 return false;
5531
5532 if (strlen(p) > PATH_MAX)
5533 return false;
5534
5535 /* The following two checks are not really dangerous, but hey, they still are confusing */
5536 if (streq(p, ".") || startswith(p, "./") || endswith(p, "/.") || strstr(p, "/./"))
5537 return false;
5538
5539 if (strstr(p, "//"))
5540 return false;
5541
5542 return true;
5543 }
5544
5545 /* hey glibc, APIs with callbacks without a user pointer are so useless */
5546 void *xbsearch_r(const void *key, const void *base, size_t nmemb, size_t size,
5547 int (*compar) (const void *, const void *, void *), void *arg) {
5548 size_t l, u, idx;
5549 const void *p;
5550 int comparison;
5551
5552 l = 0;
5553 u = nmemb;
5554 while (l < u) {
5555 idx = (l + u) / 2;
5556 p = (void *)(((const char *) base) + (idx * size));
5557 comparison = compar(key, p, arg);
5558 if (comparison < 0)
5559 u = idx;
5560 else if (comparison > 0)
5561 l = idx + 1;
5562 else
5563 return (void *)p;
5564 }
5565 return NULL;
5566 }
5567
5568 bool is_locale_utf8(void) {
5569 const char *set;
5570 static int cached_answer = -1;
5571
5572 if (cached_answer >= 0)
5573 goto out;
5574
5575 if (!setlocale(LC_ALL, "")) {
5576 cached_answer = true;
5577 goto out;
5578 }
5579
5580 set = nl_langinfo(CODESET);
5581 if (!set) {
5582 cached_answer = true;
5583 goto out;
5584 }
5585
5586 if (streq(set, "UTF-8")) {
5587 cached_answer = true;
5588 goto out;
5589 }
5590
5591 /* For LC_CTYPE=="C" return true, because CTYPE is effectly
5592 * unset and everything can do to UTF-8 nowadays. */
5593 set = setlocale(LC_CTYPE, NULL);
5594 if (!set) {
5595 cached_answer = true;
5596 goto out;
5597 }
5598
5599 /* Check result, but ignore the result if C was set
5600 * explicitly. */
5601 cached_answer =
5602 streq(set, "C") &&
5603 !getenv("LC_ALL") &&
5604 !getenv("LC_CTYPE") &&
5605 !getenv("LANG");
5606
5607 out:
5608 return (bool) cached_answer;
5609 }
5610
5611 const char *draw_special_char(DrawSpecialChar ch) {
5612 static const char *draw_table[2][_DRAW_SPECIAL_CHAR_MAX] = {
5613
5614 /* UTF-8 */ {
5615 [DRAW_TREE_VERTICAL] = "\342\224\202 ", /* │ */
5616 [DRAW_TREE_BRANCH] = "\342\224\234\342\224\200", /* ├─ */
5617 [DRAW_TREE_RIGHT] = "\342\224\224\342\224\200", /* └─ */
5618 [DRAW_TREE_SPACE] = " ", /* */
5619 [DRAW_TRIANGULAR_BULLET] = "\342\200\243", /* ‣ */
5620 [DRAW_BLACK_CIRCLE] = "\342\227\217", /* ● */
5621 [DRAW_ARROW] = "\342\206\222", /* → */
5622 [DRAW_DASH] = "\342\200\223", /* – */
5623 },
5624
5625 /* ASCII fallback */ {
5626 [DRAW_TREE_VERTICAL] = "| ",
5627 [DRAW_TREE_BRANCH] = "|-",
5628 [DRAW_TREE_RIGHT] = "`-",
5629 [DRAW_TREE_SPACE] = " ",
5630 [DRAW_TRIANGULAR_BULLET] = ">",
5631 [DRAW_BLACK_CIRCLE] = "*",
5632 [DRAW_ARROW] = "->",
5633 [DRAW_DASH] = "-",
5634 }
5635 };
5636
5637 return draw_table[!is_locale_utf8()][ch];
5638 }
5639
5640 char *strreplace(const char *text, const char *old_string, const char *new_string) {
5641 const char *f;
5642 char *t, *r;
5643 size_t l, old_len, new_len;
5644
5645 assert(text);
5646 assert(old_string);
5647 assert(new_string);
5648
5649 old_len = strlen(old_string);
5650 new_len = strlen(new_string);
5651
5652 l = strlen(text);
5653 r = new(char, l+1);
5654 if (!r)
5655 return NULL;
5656
5657 f = text;
5658 t = r;
5659 while (*f) {
5660 char *a;
5661 size_t d, nl;
5662
5663 if (!startswith(f, old_string)) {
5664 *(t++) = *(f++);
5665 continue;
5666 }
5667
5668 d = t - r;
5669 nl = l - old_len + new_len;
5670 a = realloc(r, nl + 1);
5671 if (!a)
5672 goto oom;
5673
5674 l = nl;
5675 r = a;
5676 t = r + d;
5677
5678 t = stpcpy(t, new_string);
5679 f += old_len;
5680 }
5681
5682 *t = 0;
5683 return r;
5684
5685 oom:
5686 free(r);
5687 return NULL;
5688 }
5689
5690 char *strip_tab_ansi(char **ibuf, size_t *_isz) {
5691 const char *i, *begin = NULL;
5692 enum {
5693 STATE_OTHER,
5694 STATE_ESCAPE,
5695 STATE_BRACKET
5696 } state = STATE_OTHER;
5697 char *obuf = NULL;
5698 size_t osz = 0, isz;
5699 FILE *f;
5700
5701 assert(ibuf);
5702 assert(*ibuf);
5703
5704 /* Strips ANSI color and replaces TABs by 8 spaces */
5705
5706 isz = _isz ? *_isz : strlen(*ibuf);
5707
5708 f = open_memstream(&obuf, &osz);
5709 if (!f)
5710 return NULL;
5711
5712 for (i = *ibuf; i < *ibuf + isz + 1; i++) {
5713
5714 switch (state) {
5715
5716 case STATE_OTHER:
5717 if (i >= *ibuf + isz) /* EOT */
5718 break;
5719 else if (*i == '\x1B')
5720 state = STATE_ESCAPE;
5721 else if (*i == '\t')
5722 fputs(" ", f);
5723 else
5724 fputc(*i, f);
5725 break;
5726
5727 case STATE_ESCAPE:
5728 if (i >= *ibuf + isz) { /* EOT */
5729 fputc('\x1B', f);
5730 break;
5731 } else if (*i == '[') {
5732 state = STATE_BRACKET;
5733 begin = i + 1;
5734 } else {
5735 fputc('\x1B', f);
5736 fputc(*i, f);
5737 state = STATE_OTHER;
5738 }
5739
5740 break;
5741
5742 case STATE_BRACKET:
5743
5744 if (i >= *ibuf + isz || /* EOT */
5745 (!(*i >= '0' && *i <= '9') && *i != ';' && *i != 'm')) {
5746 fputc('\x1B', f);
5747 fputc('[', f);
5748 state = STATE_OTHER;
5749 i = begin-1;
5750 } else if (*i == 'm')
5751 state = STATE_OTHER;
5752 break;
5753 }
5754 }
5755
5756 if (ferror(f)) {
5757 fclose(f);
5758 free(obuf);
5759 return NULL;
5760 }
5761
5762 fclose(f);
5763
5764 free(*ibuf);
5765 *ibuf = obuf;
5766
5767 if (_isz)
5768 *_isz = osz;
5769
5770 return obuf;
5771 }
5772
5773 int on_ac_power(void) {
5774 bool found_offline = false, found_online = false;
5775 _cleanup_closedir_ DIR *d = NULL;
5776
5777 d = opendir("/sys/class/power_supply");
5778 if (!d)
5779 return -errno;
5780
5781 for (;;) {
5782 struct dirent *de;
5783 _cleanup_close_ int fd = -1, device = -1;
5784 char contents[6];
5785 ssize_t n;
5786
5787 errno = 0;
5788 de = readdir(d);
5789 if (!de && errno != 0)
5790 return -errno;
5791
5792 if (!de)
5793 break;
5794
5795 if (ignore_file(de->d_name))
5796 continue;
5797
5798 device = openat(dirfd(d), de->d_name, O_DIRECTORY|O_RDONLY|O_CLOEXEC|O_NOCTTY);
5799 if (device < 0) {
5800 if (errno == ENOENT || errno == ENOTDIR)
5801 continue;
5802
5803 return -errno;
5804 }
5805
5806 fd = openat(device, "type", O_RDONLY|O_CLOEXEC|O_NOCTTY);
5807 if (fd < 0) {
5808 if (errno == ENOENT)
5809 continue;
5810
5811 return -errno;
5812 }
5813
5814 n = read(fd, contents, sizeof(contents));
5815 if (n < 0)
5816 return -errno;
5817
5818 if (n != 6 || memcmp(contents, "Mains\n", 6))
5819 continue;
5820
5821 safe_close(fd);
5822 fd = openat(device, "online", O_RDONLY|O_CLOEXEC|O_NOCTTY);
5823 if (fd < 0) {
5824 if (errno == ENOENT)
5825 continue;
5826
5827 return -errno;
5828 }
5829
5830 n = read(fd, contents, sizeof(contents));
5831 if (n < 0)
5832 return -errno;
5833
5834 if (n != 2 || contents[1] != '\n')
5835 return -EIO;
5836
5837 if (contents[0] == '1') {
5838 found_online = true;
5839 break;
5840 } else if (contents[0] == '0')
5841 found_offline = true;
5842 else
5843 return -EIO;
5844 }
5845
5846 return found_online || !found_offline;
5847 }
5848
5849 static int search_and_fopen_internal(const char *path, const char *mode, const char *root, char **search, FILE **_f) {
5850 char **i;
5851
5852 assert(path);
5853 assert(mode);
5854 assert(_f);
5855
5856 if (!path_strv_resolve_uniq(search, root))
5857 return -ENOMEM;
5858
5859 STRV_FOREACH(i, search) {
5860 _cleanup_free_ char *p = NULL;
5861 FILE *f;
5862
5863 if (root)
5864 p = strjoin(root, *i, "/", path, NULL);
5865 else
5866 p = strjoin(*i, "/", path, NULL);
5867 if (!p)
5868 return -ENOMEM;
5869
5870 f = fopen(p, mode);
5871 if (f) {
5872 *_f = f;
5873 return 0;
5874 }
5875
5876 if (errno != ENOENT)
5877 return -errno;
5878 }
5879
5880 return -ENOENT;
5881 }
5882
5883 int search_and_fopen(const char *path, const char *mode, const char *root, const char **search, FILE **_f) {
5884 _cleanup_strv_free_ char **copy = NULL;
5885
5886 assert(path);
5887 assert(mode);
5888 assert(_f);
5889
5890 if (path_is_absolute(path)) {
5891 FILE *f;
5892
5893 f = fopen(path, mode);
5894 if (f) {
5895 *_f = f;
5896 return 0;
5897 }
5898
5899 return -errno;
5900 }
5901
5902 copy = strv_copy((char**) search);
5903 if (!copy)
5904 return -ENOMEM;
5905
5906 return search_and_fopen_internal(path, mode, root, copy, _f);
5907 }
5908
5909 int search_and_fopen_nulstr(const char *path, const char *mode, const char *root, const char *search, FILE **_f) {
5910 _cleanup_strv_free_ char **s = NULL;
5911
5912 if (path_is_absolute(path)) {
5913 FILE *f;
5914
5915 f = fopen(path, mode);
5916 if (f) {
5917 *_f = f;
5918 return 0;
5919 }
5920
5921 return -errno;
5922 }
5923
5924 s = strv_split_nulstr(search);
5925 if (!s)
5926 return -ENOMEM;
5927
5928 return search_and_fopen_internal(path, mode, root, s, _f);
5929 }
5930
5931 char *strextend(char **x, ...) {
5932 va_list ap;
5933 size_t f, l;
5934 char *r, *p;
5935
5936 assert(x);
5937
5938 l = f = *x ? strlen(*x) : 0;
5939
5940 va_start(ap, x);
5941 for (;;) {
5942 const char *t;
5943 size_t n;
5944
5945 t = va_arg(ap, const char *);
5946 if (!t)
5947 break;
5948
5949 n = strlen(t);
5950 if (n > ((size_t) -1) - l) {
5951 va_end(ap);
5952 return NULL;
5953 }
5954
5955 l += n;
5956 }
5957 va_end(ap);
5958
5959 r = realloc(*x, l+1);
5960 if (!r)
5961 return NULL;
5962
5963 p = r + f;
5964
5965 va_start(ap, x);
5966 for (;;) {
5967 const char *t;
5968
5969 t = va_arg(ap, const char *);
5970 if (!t)
5971 break;
5972
5973 p = stpcpy(p, t);
5974 }
5975 va_end(ap);
5976
5977 *p = 0;
5978 *x = r;
5979
5980 return r + l;
5981 }
5982
5983 char *strrep(const char *s, unsigned n) {
5984 size_t l;
5985 char *r, *p;
5986 unsigned i;
5987
5988 assert(s);
5989
5990 l = strlen(s);
5991 p = r = malloc(l * n + 1);
5992 if (!r)
5993 return NULL;
5994
5995 for (i = 0; i < n; i++)
5996 p = stpcpy(p, s);
5997
5998 *p = 0;
5999 return r;
6000 }
6001
6002 void* greedy_realloc(void **p, size_t *allocated, size_t need, size_t size) {
6003 size_t a, newalloc;
6004 void *q;
6005
6006 assert(p);
6007 assert(allocated);
6008
6009 if (*allocated >= need)
6010 return *p;
6011
6012 newalloc = MAX(need * 2, 64u / size);
6013 a = newalloc * size;
6014
6015 /* check for overflows */
6016 if (a < size * need)
6017 return NULL;
6018
6019 q = realloc(*p, a);
6020 if (!q)
6021 return NULL;
6022
6023 *p = q;
6024 *allocated = newalloc;
6025 return q;
6026 }
6027
6028 void* greedy_realloc0(void **p, size_t *allocated, size_t need, size_t size) {
6029 size_t prev;
6030 uint8_t *q;
6031
6032 assert(p);
6033 assert(allocated);
6034
6035 prev = *allocated;
6036
6037 q = greedy_realloc(p, allocated, need, size);
6038 if (!q)
6039 return NULL;
6040
6041 if (*allocated > prev)
6042 memzero(q + prev * size, (*allocated - prev) * size);
6043
6044 return q;
6045 }
6046
6047 bool id128_is_valid(const char *s) {
6048 size_t i, l;
6049
6050 l = strlen(s);
6051 if (l == 32) {
6052
6053 /* Simple formatted 128bit hex string */
6054
6055 for (i = 0; i < l; i++) {
6056 char c = s[i];
6057
6058 if (!(c >= '0' && c <= '9') &&
6059 !(c >= 'a' && c <= 'z') &&
6060 !(c >= 'A' && c <= 'Z'))
6061 return false;
6062 }
6063
6064 } else if (l == 36) {
6065
6066 /* Formatted UUID */
6067
6068 for (i = 0; i < l; i++) {
6069 char c = s[i];
6070
6071 if ((i == 8 || i == 13 || i == 18 || i == 23)) {
6072 if (c != '-')
6073 return false;
6074 } else {
6075 if (!(c >= '0' && c <= '9') &&
6076 !(c >= 'a' && c <= 'z') &&
6077 !(c >= 'A' && c <= 'Z'))
6078 return false;
6079 }
6080 }
6081
6082 } else
6083 return false;
6084
6085 return true;
6086 }
6087
6088 int split_pair(const char *s, const char *sep, char **l, char **r) {
6089 char *x, *a, *b;
6090
6091 assert(s);
6092 assert(sep);
6093 assert(l);
6094 assert(r);
6095
6096 if (isempty(sep))
6097 return -EINVAL;
6098
6099 x = strstr(s, sep);
6100 if (!x)
6101 return -EINVAL;
6102
6103 a = strndup(s, x - s);
6104 if (!a)
6105 return -ENOMEM;
6106
6107 b = strdup(x + strlen(sep));
6108 if (!b) {
6109 free(a);
6110 return -ENOMEM;
6111 }
6112
6113 *l = a;
6114 *r = b;
6115
6116 return 0;
6117 }
6118
6119 int shall_restore_state(void) {
6120 _cleanup_free_ char *line = NULL;
6121 const char *word, *state;
6122 size_t l;
6123 int r;
6124
6125 r = proc_cmdline(&line);
6126 if (r < 0)
6127 return r;
6128 if (r == 0) /* Container ... */
6129 return 1;
6130
6131 r = 1;
6132
6133 FOREACH_WORD_QUOTED(word, l, line, state) {
6134 const char *e;
6135 char n[l+1];
6136 int k;
6137
6138 memcpy(n, word, l);
6139 n[l] = 0;
6140
6141 e = startswith(n, "systemd.restore_state=");
6142 if (!e)
6143 continue;
6144
6145 k = parse_boolean(e);
6146 if (k >= 0)
6147 r = k;
6148 }
6149
6150 return r;
6151 }
6152
6153 int proc_cmdline(char **ret) {
6154 int r;
6155
6156 if (detect_container(NULL) > 0) {
6157 char *buf = NULL, *p;
6158 size_t sz = 0;
6159
6160 r = read_full_file("/proc/1/cmdline", &buf, &sz);
6161 if (r < 0)
6162 return r;
6163
6164 for (p = buf; p + 1 < buf + sz; p++)
6165 if (*p == 0)
6166 *p = ' ';
6167
6168 *p = 0;
6169 *ret = buf;
6170 return 1;
6171 }
6172
6173 r = read_one_line_file("/proc/cmdline", ret);
6174 if (r < 0)
6175 return r;
6176
6177 return 1;
6178 }
6179
6180 int parse_proc_cmdline(int (*parse_item)(const char *key, const char *value)) {
6181 _cleanup_free_ char *line = NULL;
6182 const char *w, *state;
6183 size_t l;
6184 int r;
6185
6186 assert(parse_item);
6187
6188 r = proc_cmdline(&line);
6189 if (r < 0)
6190 log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r));
6191 if (r <= 0)
6192 return 0;
6193
6194 FOREACH_WORD_QUOTED(w, l, line, state) {
6195 char word[l+1], *value;
6196
6197 memcpy(word, w, l);
6198 word[l] = 0;
6199
6200 /* Filter out arguments that are intended only for the
6201 * initrd */
6202 if (!in_initrd() && startswith(word, "rd."))
6203 continue;
6204
6205 value = strchr(word, '=');
6206 if (value)
6207 *(value++) = 0;
6208
6209 r = parse_item(word, value);
6210 if (r < 0)
6211 return r;
6212 }
6213
6214 return 0;
6215 }
6216
6217 int container_get_leader(const char *machine, pid_t *pid) {
6218 _cleanup_free_ char *s = NULL, *class = NULL;
6219 const char *p;
6220 pid_t leader;
6221 int r;
6222
6223 assert(machine);
6224 assert(pid);
6225
6226 p = strappenda("/run/systemd/machines/", machine);
6227 r = parse_env_file(p, NEWLINE, "LEADER", &s, "CLASS", &class, NULL);
6228 if (r == -ENOENT)
6229 return -EHOSTDOWN;
6230 if (r < 0)
6231 return r;
6232 if (!s)
6233 return -EIO;
6234
6235 if (!streq_ptr(class, "container"))
6236 return -EIO;
6237
6238 r = parse_pid(s, &leader);
6239 if (r < 0)
6240 return r;
6241 if (leader <= 1)
6242 return -EIO;
6243
6244 *pid = leader;
6245 return 0;
6246 }
6247
6248 int namespace_open(pid_t pid, int *pidns_fd, int *mntns_fd, int *netns_fd, int *root_fd) {
6249 _cleanup_close_ int pidnsfd = -1, mntnsfd = -1, netnsfd = -1;
6250 int rfd = -1;
6251
6252 assert(pid >= 0);
6253
6254 if (mntns_fd) {
6255 const char *mntns;
6256
6257 mntns = procfs_file_alloca(pid, "ns/mnt");
6258 mntnsfd = open(mntns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6259 if (mntnsfd < 0)
6260 return -errno;
6261 }
6262
6263 if (pidns_fd) {
6264 const char *pidns;
6265
6266 pidns = procfs_file_alloca(pid, "ns/pid");
6267 pidnsfd = open(pidns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6268 if (pidnsfd < 0)
6269 return -errno;
6270 }
6271
6272 if (netns_fd) {
6273 const char *netns;
6274
6275 netns = procfs_file_alloca(pid, "ns/net");
6276 netnsfd = open(netns, O_RDONLY|O_NOCTTY|O_CLOEXEC);
6277 if (netnsfd < 0)
6278 return -errno;
6279 }
6280
6281 if (root_fd) {
6282 const char *root;
6283
6284 root = procfs_file_alloca(pid, "root");
6285 rfd = open(root, O_RDONLY|O_NOCTTY|O_CLOEXEC|O_DIRECTORY);
6286 if (rfd < 0)
6287 return -errno;
6288 }
6289
6290 if (pidns_fd)
6291 *pidns_fd = pidnsfd;
6292
6293 if (mntns_fd)
6294 *mntns_fd = mntnsfd;
6295
6296 if (netns_fd)
6297 *netns_fd = netnsfd;
6298
6299 if (root_fd)
6300 *root_fd = rfd;
6301
6302 pidnsfd = mntnsfd = netnsfd = -1;
6303
6304 return 0;
6305 }
6306
6307 int namespace_enter(int pidns_fd, int mntns_fd, int netns_fd, int root_fd) {
6308
6309 if (pidns_fd >= 0)
6310 if (setns(pidns_fd, CLONE_NEWPID) < 0)
6311 return -errno;
6312
6313 if (mntns_fd >= 0)
6314 if (setns(mntns_fd, CLONE_NEWNS) < 0)
6315 return -errno;
6316
6317 if (netns_fd >= 0)
6318 if (setns(netns_fd, CLONE_NEWNET) < 0)
6319 return -errno;
6320
6321 if (root_fd >= 0) {
6322 if (fchdir(root_fd) < 0)
6323 return -errno;
6324
6325 if (chroot(".") < 0)
6326 return -errno;
6327 }
6328
6329 if (setresgid(0, 0, 0) < 0)
6330 return -errno;
6331
6332 if (setgroups(0, NULL) < 0)
6333 return -errno;
6334
6335 if (setresuid(0, 0, 0) < 0)
6336 return -errno;
6337
6338 return 0;
6339 }
6340
6341 bool pid_is_unwaited(pid_t pid) {
6342 /* Checks whether a PID is still valid at all, including a zombie */
6343
6344 if (pid <= 0)
6345 return false;
6346
6347 if (kill(pid, 0) >= 0)
6348 return true;
6349
6350 return errno != ESRCH;
6351 }
6352
6353 bool pid_is_alive(pid_t pid) {
6354 int r;
6355
6356 /* Checks whether a PID is still valid and not a zombie */
6357
6358 if (pid <= 0)
6359 return false;
6360
6361 r = get_process_state(pid);
6362 if (r == -ENOENT || r == 'Z')
6363 return false;
6364
6365 return true;
6366 }
6367
6368 int getpeercred(int fd, struct ucred *ucred) {
6369 socklen_t n = sizeof(struct ucred);
6370 struct ucred u;
6371 int r;
6372
6373 assert(fd >= 0);
6374 assert(ucred);
6375
6376 r = getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &u, &n);
6377 if (r < 0)
6378 return -errno;
6379
6380 if (n != sizeof(struct ucred))
6381 return -EIO;
6382
6383 /* Check if the data is actually useful and not suppressed due
6384 * to namespacing issues */
6385 if (u.pid <= 0)
6386 return -ENODATA;
6387
6388 *ucred = u;
6389 return 0;
6390 }
6391
6392 int getpeersec(int fd, char **ret) {
6393 socklen_t n = 64;
6394 char *s;
6395 int r;
6396
6397 assert(fd >= 0);
6398 assert(ret);
6399
6400 s = new0(char, n);
6401 if (!s)
6402 return -ENOMEM;
6403
6404 r = getsockopt(fd, SOL_SOCKET, SO_PEERSEC, s, &n);
6405 if (r < 0) {
6406 free(s);
6407
6408 if (errno != ERANGE)
6409 return -errno;
6410
6411 s = new0(char, n);
6412 if (!s)
6413 return -ENOMEM;
6414
6415 r = getsockopt(fd, SOL_SOCKET, SO_PEERSEC, s, &n);
6416 if (r < 0) {
6417 free(s);
6418 return -errno;
6419 }
6420 }
6421
6422 if (isempty(s)) {
6423 free(s);
6424 return -ENOTSUP;
6425 }
6426
6427 *ret = s;
6428 return 0;
6429 }
6430
6431 /* This is much like like mkostemp() but is subject to umask(). */
6432 int mkostemp_safe(char *pattern, int flags) {
6433 _cleanup_umask_ mode_t u;
6434 int fd;
6435
6436 assert(pattern);
6437
6438 u = umask(077);
6439
6440 fd = mkostemp(pattern, flags);
6441 if (fd < 0)
6442 return -errno;
6443
6444 return fd;
6445 }
6446
6447 int open_tmpfile(const char *path, int flags) {
6448 char *p;
6449 int fd;
6450
6451 assert(path);
6452
6453 #ifdef O_TMPFILE
6454 /* Try O_TMPFILE first, if it is supported */
6455 fd = open(path, flags|O_TMPFILE, S_IRUSR|S_IWUSR);
6456 if (fd >= 0)
6457 return fd;
6458 #endif
6459
6460 /* Fall back to unguessable name + unlinking */
6461 p = strappenda(path, "/systemd-tmp-XXXXXX");
6462
6463 fd = mkostemp_safe(p, flags);
6464 if (fd < 0)
6465 return fd;
6466
6467 unlink(p);
6468 return fd;
6469 }
6470
6471 int fd_warn_permissions(const char *path, int fd) {
6472 struct stat st;
6473
6474 if (fstat(fd, &st) < 0)
6475 return -errno;
6476
6477 if (st.st_mode & 0111)
6478 log_warning("Configuration file %s is marked executable. Please remove executable permission bits. Proceeding anyway.", path);
6479
6480 if (st.st_mode & 0002)
6481 log_warning("Configuration file %s is marked world-writable. Please remove world writability permission bits. Proceeding anyway.", path);
6482
6483 if (getpid() == 1 && (st.st_mode & 0044) != 0044)
6484 log_warning("Configuration file %s is marked world-inaccessible. This has no effect as configuration data is accessible via APIs without restrictions. Proceeding anyway.", path);
6485
6486 return 0;
6487 }
6488
6489 unsigned long personality_from_string(const char *p) {
6490
6491 /* Parse a personality specifier. We introduce our own
6492 * identifiers that indicate specific ABIs, rather than just
6493 * hints regarding the register size, since we want to keep
6494 * things open for multiple locally supported ABIs for the
6495 * same register size. We try to reuse the ABI identifiers
6496 * used by libseccomp. */
6497
6498 #if defined(__x86_64__)
6499
6500 if (streq(p, "x86"))
6501 return PER_LINUX32;
6502
6503 if (streq(p, "x86-64"))
6504 return PER_LINUX;
6505
6506 #elif defined(__i386__)
6507
6508 if (streq(p, "x86"))
6509 return PER_LINUX;
6510 #endif
6511
6512 /* personality(7) documents that 0xffffffffUL is used for
6513 * querying the current personality, hence let's use that here
6514 * as error indicator. */
6515 return 0xffffffffUL;
6516 }
6517
6518 const char* personality_to_string(unsigned long p) {
6519
6520 #if defined(__x86_64__)
6521
6522 if (p == PER_LINUX32)
6523 return "x86";
6524
6525 if (p == PER_LINUX)
6526 return "x86-64";
6527
6528 #elif defined(__i386__)
6529
6530 if (p == PER_LINUX)
6531 return "x86";
6532 #endif
6533
6534 return NULL;
6535 }
6536
6537 uint64_t physical_memory(void) {
6538 long mem;
6539
6540 /* We return this as uint64_t in case we are running as 32bit
6541 * process on a 64bit kernel with huge amounts of memory */
6542
6543 mem = sysconf(_SC_PHYS_PAGES);
6544 assert(mem > 0);
6545
6546 return (uint64_t) mem * (uint64_t) page_size();
6547 }
6548
6549 char* mount_test_option(const char *haystack, const char *needle) {
6550
6551 struct mntent me = {
6552 .mnt_opts = (char*) haystack
6553 };
6554
6555 assert(needle);
6556
6557 /* Like glibc's hasmntopt(), but works on a string, not a
6558 * struct mntent */
6559
6560 if (!haystack)
6561 return NULL;
6562
6563 return hasmntopt(&me, needle);
6564 }
6565
6566 void hexdump(FILE *f, const void *p, size_t s) {
6567 const uint8_t *b = p;
6568 unsigned n = 0;
6569
6570 assert(s == 0 || b);
6571
6572 while (s > 0) {
6573 size_t i;
6574
6575 fprintf(f, "%04x ", n);
6576
6577 for (i = 0; i < 16; i++) {
6578
6579 if (i >= s)
6580 fputs(" ", f);
6581 else
6582 fprintf(f, "%02x ", b[i]);
6583
6584 if (i == 7)
6585 fputc(' ', f);
6586 }
6587
6588 fputc(' ', f);
6589
6590 for (i = 0; i < 16; i++) {
6591
6592 if (i >= s)
6593 fputc(' ', f);
6594 else
6595 fputc(isprint(b[i]) ? (char) b[i] : '.', f);
6596 }
6597
6598 fputc('\n', f);
6599
6600 if (s < 16)
6601 break;
6602
6603 n += 16;
6604 b += 16;
6605 s -= 16;
6606 }
6607 }
6608
6609 int update_reboot_param_file(const char *param) {
6610 int r = 0;
6611
6612 if (param) {
6613
6614 r = write_string_file(REBOOT_PARAM_FILE, param);
6615 if (r < 0)
6616 log_error("Failed to write reboot param to "
6617 REBOOT_PARAM_FILE": %s", strerror(-r));
6618 } else
6619 unlink(REBOOT_PARAM_FILE);
6620
6621 return r;
6622 }
6623
6624 int umount_recursive(const char *prefix, int flags) {
6625 bool again;
6626 int n = 0, r;
6627
6628 /* Try to umount everything recursively below a
6629 * directory. Also, take care of stacked mounts, and keep
6630 * unmounting them until they are gone. */
6631
6632 do {
6633 _cleanup_fclose_ FILE *proc_self_mountinfo = NULL;
6634
6635 again = false;
6636 r = 0;
6637
6638 proc_self_mountinfo = fopen("/proc/self/mountinfo", "re");
6639 if (!proc_self_mountinfo)
6640 return -errno;
6641
6642 for (;;) {
6643 _cleanup_free_ char *path = NULL, *p = NULL;
6644 int k;
6645
6646 k = fscanf(proc_self_mountinfo,
6647 "%*s " /* (1) mount id */
6648 "%*s " /* (2) parent id */
6649 "%*s " /* (3) major:minor */
6650 "%*s " /* (4) root */
6651 "%ms " /* (5) mount point */
6652 "%*s" /* (6) mount options */
6653 "%*[^-]" /* (7) optional fields */
6654 "- " /* (8) separator */
6655 "%*s " /* (9) file system type */
6656 "%*s" /* (10) mount source */
6657 "%*s" /* (11) mount options 2 */
6658 "%*[^\n]", /* some rubbish at the end */
6659 &path);
6660 if (k != 1) {
6661 if (k == EOF)
6662 break;
6663
6664 continue;
6665 }
6666
6667 p = cunescape(path);
6668 if (!p)
6669 return -ENOMEM;
6670
6671 if (!path_startswith(p, prefix))
6672 continue;
6673
6674 if (umount2(p, flags) < 0) {
6675 r = -errno;
6676 continue;
6677 }
6678
6679 again = true;
6680 n++;
6681
6682 break;
6683 }
6684
6685 } while (again);
6686
6687 return r ? r : n;
6688 }
6689
6690 int bind_remount_recursive(const char *prefix, bool ro) {
6691 _cleanup_set_free_free_ Set *done = NULL;
6692 _cleanup_free_ char *cleaned = NULL;
6693 int r;
6694
6695 /* Recursively remount a directory (and all its submounts)
6696 * read-only or read-write. If the directory is already
6697 * mounted, we reuse the mount and simply mark it
6698 * MS_BIND|MS_RDONLY (or remove the MS_RDONLY for read-write
6699 * operation). If it isn't we first make it one. Afterwards we
6700 * apply MS_BIND|MS_RDONLY (or remove MS_RDONLY) to all
6701 * submounts we can access, too. When mounts are stacked on
6702 * the same mount point we only care for each individual
6703 * "top-level" mount on each point, as we cannot
6704 * influence/access the underlying mounts anyway. We do not
6705 * have any effect on future submounts that might get
6706 * propagated, they migt be writable. This includes future
6707 * submounts that have been triggered via autofs. */
6708
6709 cleaned = strdup(prefix);
6710 if (!cleaned)
6711 return -ENOMEM;
6712
6713 path_kill_slashes(cleaned);
6714
6715 done = set_new(string_hash_func, string_compare_func);
6716 if (!done)
6717 return -ENOMEM;
6718
6719 for (;;) {
6720 _cleanup_fclose_ FILE *proc_self_mountinfo = NULL;
6721 _cleanup_set_free_free_ Set *todo = NULL;
6722 bool top_autofs = false;
6723 char *x;
6724
6725 todo = set_new(string_hash_func, string_compare_func);
6726 if (!todo)
6727 return -ENOMEM;
6728
6729 proc_self_mountinfo = fopen("/proc/self/mountinfo", "re");
6730 if (!proc_self_mountinfo)
6731 return -errno;
6732
6733 for (;;) {
6734 _cleanup_free_ char *path = NULL, *p = NULL, *type = NULL;
6735 int k;
6736
6737 k = fscanf(proc_self_mountinfo,
6738 "%*s " /* (1) mount id */
6739 "%*s " /* (2) parent id */
6740 "%*s " /* (3) major:minor */
6741 "%*s " /* (4) root */
6742 "%ms " /* (5) mount point */
6743 "%*s" /* (6) mount options (superblock) */
6744 "%*[^-]" /* (7) optional fields */
6745 "- " /* (8) separator */
6746 "%ms " /* (9) file system type */
6747 "%*s" /* (10) mount source */
6748 "%*s" /* (11) mount options (bind mount) */
6749 "%*[^\n]", /* some rubbish at the end */
6750 &path,
6751 &type);
6752 if (k != 2) {
6753 if (k == EOF)
6754 break;
6755
6756 continue;
6757 }
6758
6759 p = cunescape(path);
6760 if (!p)
6761 return -ENOMEM;
6762
6763 /* Let's ignore autofs mounts. If they aren't
6764 * triggered yet, we want to avoid triggering
6765 * them, as we don't make any guarantees for
6766 * future submounts anyway. If they are
6767 * already triggered, then we will find
6768 * another entry for this. */
6769 if (streq(type, "autofs")) {
6770 top_autofs = top_autofs || path_equal(cleaned, p);
6771 continue;
6772 }
6773
6774 if (path_startswith(p, cleaned) &&
6775 !set_contains(done, p)) {
6776
6777 r = set_consume(todo, p);
6778 p = NULL;
6779
6780 if (r == -EEXIST)
6781 continue;
6782 if (r < 0)
6783 return r;
6784 }
6785 }
6786
6787 /* If we have no submounts to process anymore and if
6788 * the root is either already done, or an autofs, we
6789 * are done */
6790 if (set_isempty(todo) &&
6791 (top_autofs || set_contains(done, cleaned)))
6792 return 0;
6793
6794 if (!set_contains(done, cleaned) &&
6795 !set_contains(todo, cleaned)) {
6796 /* The prefix directory itself is not yet a
6797 * mount, make it one. */
6798 if (mount(cleaned, cleaned, NULL, MS_BIND|MS_REC, NULL) < 0)
6799 return -errno;
6800
6801 if (mount(NULL, prefix, NULL, MS_BIND|MS_REMOUNT|(ro ? MS_RDONLY : 0), NULL) < 0)
6802 return -errno;
6803
6804 x = strdup(cleaned);
6805 if (!x)
6806 return -ENOMEM;
6807
6808 r = set_consume(done, x);
6809 if (r < 0)
6810 return r;
6811 }
6812
6813 while ((x = set_steal_first(todo))) {
6814
6815 r = set_consume(done, x);
6816 if (r == -EEXIST)
6817 continue;
6818 if (r < 0)
6819 return r;
6820
6821 if (mount(NULL, x, NULL, MS_BIND|MS_REMOUNT|(ro ? MS_RDONLY : 0), NULL) < 0) {
6822
6823 /* Deal with mount points that are
6824 * obstructed by a later mount */
6825
6826 if (errno != ENOENT)
6827 return -errno;
6828 }
6829
6830 }
6831 }
6832 }
6833
6834 int fflush_and_check(FILE *f) {
6835 assert(f);
6836
6837 errno = 0;
6838 fflush(f);
6839
6840 if (ferror(f))
6841 return errno ? -errno : -EIO;
6842
6843 return 0;
6844 }
6845
6846 char *tempfn_xxxxxx(const char *p) {
6847 const char *fn;
6848 char *t;
6849 size_t k;
6850
6851 assert(p);
6852
6853 t = new(char, strlen(p) + 1 + 6 + 1);
6854 if (!t)
6855 return NULL;
6856
6857 fn = basename(p);
6858 k = fn - p;
6859
6860 strcpy(stpcpy(stpcpy(mempcpy(t, p, k), "."), fn), "XXXXXX");
6861
6862 return t;
6863 }
6864
6865 char *tempfn_random(const char *p) {
6866 const char *fn;
6867 char *t, *x;
6868 uint64_t u;
6869 size_t k;
6870 unsigned i;
6871
6872 assert(p);
6873
6874 t = new(char, strlen(p) + 1 + 16 + 1);
6875 if (!t)
6876 return NULL;
6877
6878 fn = basename(p);
6879 k = fn - p;
6880
6881 x = stpcpy(stpcpy(mempcpy(t, p, k), "."), fn);
6882
6883 u = random_u64();
6884 for (i = 0; i < 16; i++) {
6885 *(x++) = hexchar(u & 0xF);
6886 u >>= 4;
6887 }
6888
6889 *x = 0;
6890
6891 return t;
6892 }
6893
6894 /* make sure the hostname is not "localhost" */
6895 bool is_localhost(const char *hostname) {
6896 assert(hostname);
6897
6898 /* This tries to identify local host and domain names
6899 * described in RFC6761 plus the redhatism of .localdomain */
6900
6901 return streq(hostname, "localhost") ||
6902 streq(hostname, "localhost.") ||
6903 streq(hostname, "localdomain.") ||
6904 streq(hostname, "localdomain") ||
6905 endswith(hostname, ".localhost") ||
6906 endswith(hostname, ".localhost.") ||
6907 endswith(hostname, ".localdomain") ||
6908 endswith(hostname, ".localdomain.");
6909 }
6910
6911 int take_password_lock(const char *root) {
6912
6913 struct flock flock = {
6914 .l_type = F_WRLCK,
6915 .l_whence = SEEK_SET,
6916 .l_start = 0,
6917 .l_len = 0,
6918 };
6919
6920 const char *path;
6921 int fd, r;
6922
6923 /* This is roughly the same as lckpwdf(), but not as awful. We
6924 * don't want to use alarm() and signals, hence we implement
6925 * our own trivial version of this.
6926 *
6927 * Note that shadow-utils also takes per-database locks in
6928 * addition to lckpwdf(). However, we don't given that they
6929 * are redundant as they they invoke lckpwdf() first and keep
6930 * it during everything they do. The per-database locks are
6931 * awfully racy, and thus we just won't do them. */
6932
6933 if (root)
6934 path = strappenda(root, "/etc/.pwd.lock");
6935 else
6936 path = "/etc/.pwd.lock";
6937
6938 fd = open(path, O_WRONLY|O_CREAT|O_CLOEXEC|O_NOCTTY|O_NOFOLLOW, 0600);
6939 if (fd < 0)
6940 return -errno;
6941
6942 r = fcntl(fd, F_SETLKW, &flock);
6943 if (r < 0) {
6944 safe_close(fd);
6945 return -errno;
6946 }
6947
6948 return fd;
6949 }
6950
6951 int is_symlink(const char *path) {
6952 struct stat info;
6953
6954 if (lstat(path, &info) < 0)
6955 return -errno;
6956
6957 if (S_ISLNK(info.st_mode))
6958 return 1;
6959
6960 return 0;
6961 }
6962
6963 int unquote_first_word(const char **p, char **ret) {
6964 _cleanup_free_ char *s = NULL;
6965 size_t allocated = 0, sz = 0;
6966
6967 enum {
6968 START,
6969 VALUE,
6970 VALUE_ESCAPE,
6971 SINGLE_QUOTE,
6972 SINGLE_QUOTE_ESCAPE,
6973 DOUBLE_QUOTE,
6974 DOUBLE_QUOTE_ESCAPE,
6975 SPACE,
6976 } state = START;
6977
6978 assert(p);
6979 assert(*p);
6980 assert(ret);
6981
6982 /* Parses the first word of a string, and returns it in
6983 * *ret. Removes all quotes in the process. When parsing fails
6984 * (because of an uneven number of quotes or similar), leaves
6985 * the pointer *p at the first invalid character. */
6986
6987 for (;;) {
6988 char c = **p;
6989
6990 switch (state) {
6991
6992 case START:
6993 if (c == 0)
6994 goto finish;
6995 else if (strchr(WHITESPACE, c))
6996 break;
6997
6998 state = VALUE;
6999 /* fallthrough */
7000
7001 case VALUE:
7002 if (c == 0)
7003 goto finish;
7004 else if (c == '\'')
7005 state = SINGLE_QUOTE;
7006 else if (c == '\\')
7007 state = VALUE_ESCAPE;
7008 else if (c == '\"')
7009 state = DOUBLE_QUOTE;
7010 else if (strchr(WHITESPACE, c))
7011 state = SPACE;
7012 else {
7013 if (!GREEDY_REALLOC(s, allocated, sz+2))
7014 return -ENOMEM;
7015
7016 s[sz++] = c;
7017 }
7018
7019 break;
7020
7021 case VALUE_ESCAPE:
7022 if (c == 0)
7023 return -EINVAL;
7024
7025 if (!GREEDY_REALLOC(s, allocated, sz+2))
7026 return -ENOMEM;
7027
7028 s[sz++] = c;
7029 state = VALUE;
7030
7031 break;
7032
7033 case SINGLE_QUOTE:
7034 if (c == 0)
7035 return -EINVAL;
7036 else if (c == '\'')
7037 state = VALUE;
7038 else if (c == '\\')
7039 state = SINGLE_QUOTE_ESCAPE;
7040 else {
7041 if (!GREEDY_REALLOC(s, allocated, sz+2))
7042 return -ENOMEM;
7043
7044 s[sz++] = c;
7045 }
7046
7047 break;
7048
7049 case SINGLE_QUOTE_ESCAPE:
7050 if (c == 0)
7051 return -EINVAL;
7052
7053 if (!GREEDY_REALLOC(s, allocated, sz+2))
7054 return -ENOMEM;
7055
7056 s[sz++] = c;
7057 state = SINGLE_QUOTE;
7058 break;
7059
7060 case DOUBLE_QUOTE:
7061 if (c == 0)
7062 return -EINVAL;
7063 else if (c == '\"')
7064 state = VALUE;
7065 else if (c == '\\')
7066 state = DOUBLE_QUOTE_ESCAPE;
7067 else {
7068 if (!GREEDY_REALLOC(s, allocated, sz+2))
7069 return -ENOMEM;
7070
7071 s[sz++] = c;
7072 }
7073
7074 break;
7075
7076 case DOUBLE_QUOTE_ESCAPE:
7077 if (c == 0)
7078 return -EINVAL;
7079
7080 if (!GREEDY_REALLOC(s, allocated, sz+2))
7081 return -ENOMEM;
7082
7083 s[sz++] = c;
7084 state = DOUBLE_QUOTE;
7085 break;
7086
7087 case SPACE:
7088 if (c == 0)
7089 goto finish;
7090 if (!strchr(WHITESPACE, c))
7091 goto finish;
7092
7093 break;
7094 }
7095
7096 (*p) ++;
7097 }
7098
7099 finish:
7100 if (!s) {
7101 *ret = NULL;
7102 return 0;
7103 }
7104
7105 s[sz] = 0;
7106 *ret = s;
7107 s = NULL;
7108
7109 return 1;
7110 }
7111
7112 int unquote_many_words(const char **p, ...) {
7113 va_list ap;
7114 char **l;
7115 int n = 0, i, c, r;
7116
7117 /* Parses a number of words from a string, stripping any
7118 * quotes if necessary. */
7119
7120 assert(p);
7121
7122 /* Count how many words are expected */
7123 va_start(ap, p);
7124 for (;;) {
7125 if (!va_arg(ap, char **))
7126 break;
7127 n++;
7128 }
7129 va_end(ap);
7130
7131 if (n <= 0)
7132 return 0;
7133
7134 /* Read all words into a temporary array */
7135 l = newa0(char*, n);
7136 for (c = 0; c < n; c++) {
7137
7138 r = unquote_first_word(p, &l[c]);
7139 if (r < 0) {
7140 int j;
7141
7142 for (j = 0; j < c; j++)
7143 free(l[j]);
7144
7145 return r;
7146 }
7147
7148 if (r == 0)
7149 break;
7150 }
7151
7152 /* If we managed to parse all words, return them in the passed
7153 * in parameters */
7154 va_start(ap, p);
7155 for (i = 0; i < n; i++) {
7156 char **v;
7157
7158 v = va_arg(ap, char **);
7159 assert(v);
7160
7161 *v = l[i];
7162 }
7163 va_end(ap);
7164
7165 return c;
7166 }
7167
7168 int free_and_strdup(char **p, const char *s) {
7169 char *t;
7170
7171 assert(p);
7172
7173 /* Replaces a string pointer with an strdup()ed new string,
7174 * possibly freeing the old one. */
7175
7176 if (s) {
7177 t = strdup(s);
7178 if (!t)
7179 return -ENOMEM;
7180 } else
7181 t = NULL;
7182
7183 free(*p);
7184 *p = t;
7185
7186 return 0;
7187 }
Unnamed repository; edit this file 'description' to name the repository.