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