chiark / gitweb /
tmpfiles: use an enum instead of plain char for item type
[elogind.git] / src / tmpfiles.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, Kay Sievers
7
8   systemd is free software; you can redistribute it and/or modify it
9   under the terms of the GNU General Public License as published by
10   the Free Software Foundation; either version 2 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   General Public License for more details.
17
18   You should have received a copy of the GNU General Public License
19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
20 ***/
21
22 #include <unistd.h>
23 #include <fcntl.h>
24 #include <errno.h>
25 #include <string.h>
26 #include <sys/stat.h>
27 #include <limits.h>
28 #include <dirent.h>
29 #include <grp.h>
30 #include <pwd.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <stddef.h>
34 #include <getopt.h>
35 #include <stdbool.h>
36 #include <time.h>
37 #include <sys/types.h>
38 #include <sys/param.h>
39 #include <glob.h>
40 #include <fnmatch.h>
41
42 #include "log.h"
43 #include "util.h"
44 #include "strv.h"
45 #include "label.h"
46 #include "set.h"
47
48 /* This reads all files listed in /etc/tmpfiles.d/?*.conf and creates
49  * them in the file system. This is intended to be used to create
50  * properly owned directories beneath /tmp, /var/tmp, /run, which are
51  * volatile and hence need to be recreated on bootup. */
52
53 typedef enum ItemType {
54         /* These ones take file names */
55         CREATE_FILE = 'f',
56         TRUNCATE_FILE = 'F',
57         CREATE_DIRECTORY = 'd',
58         TRUNCATE_DIRECTORY = 'D',
59         CREATE_FIFO = 'p',
60
61         /* These ones take globs */
62         IGNORE_PATH = 'x',
63         REMOVE_PATH = 'r',
64         RECURSIVE_REMOVE_PATH = 'R'
65 } ItemType;
66
67 typedef struct Item {
68         ItemType type;
69
70         char *path;
71         uid_t uid;
72         gid_t gid;
73         mode_t mode;
74         usec_t age;
75
76         bool uid_set:1;
77         bool gid_set:1;
78         bool mode_set:1;
79         bool age_set:1;
80 } Item;
81
82 static Hashmap *items = NULL, *globs = NULL;
83 static Set *unix_sockets = NULL;
84
85 static bool arg_create = false;
86 static bool arg_clean = false;
87 static bool arg_remove = false;
88
89 static const char *arg_prefix = NULL;
90
91 #define MAX_DEPTH 256
92
93 static bool needs_glob(ItemType t) {
94         return t == IGNORE_PATH || t == REMOVE_PATH || t == RECURSIVE_REMOVE_PATH;
95 }
96
97 static struct Item* find_glob(Hashmap *h, const char *match) {
98         Item *j;
99         Iterator i;
100
101         HASHMAP_FOREACH(j, h, i)
102                 if (fnmatch(j->path, match, FNM_PATHNAME|FNM_PERIOD) == 0)
103                         return j;
104
105         return NULL;
106 }
107
108 static void load_unix_sockets(void) {
109         FILE *f = NULL;
110         char line[LINE_MAX];
111
112         if (unix_sockets)
113                 return;
114
115         /* We maintain a cache of the sockets we found in
116          * /proc/net/unix to speed things up a little. */
117
118         if (!(unix_sockets = set_new(string_hash_func, string_compare_func)))
119                 return;
120
121         if (!(f = fopen("/proc/net/unix", "re")))
122                 return;
123
124         if (!(fgets(line, sizeof(line), f)))
125                 goto fail;
126
127         for (;;) {
128                 char *p, *s;
129                 int k;
130
131                 if (!(fgets(line, sizeof(line), f)))
132                         break;
133
134                 truncate_nl(line);
135
136                 if (strlen(line) < 53)
137                         continue;
138
139                 p = line + 53;
140                 p += strspn(p, WHITESPACE);
141                 p += strcspn(p, WHITESPACE);
142                 p += strspn(p, WHITESPACE);
143
144                 if (*p != '/')
145                         continue;
146
147                 if (!(s = strdup(p)))
148                         goto fail;
149
150                 path_kill_slashes(s);
151
152                 if ((k = set_put(unix_sockets, s)) < 0) {
153                         free(s);
154
155                         if (k != -EEXIST)
156                                 goto fail;
157                 }
158         }
159
160         fclose(f);
161         return;
162
163 fail:
164         set_free_free(unix_sockets);
165         unix_sockets = NULL;
166
167         if (f)
168                 fclose(f);
169 }
170
171 static bool unix_socket_alive(const char *fn) {
172         assert(fn);
173
174         load_unix_sockets();
175
176         if (unix_sockets)
177                 return !!set_get(unix_sockets, (char*) fn);
178
179         /* We don't know, so assume yes */
180         return true;
181 }
182
183 static int dir_cleanup(
184                 const char *p,
185                 DIR *d,
186                 const struct stat *ds,
187                 usec_t cutoff,
188                 dev_t rootdev,
189                 bool mountpoint,
190                 int maxdepth)
191 {
192         struct dirent *dent;
193         struct timespec times[2];
194         bool deleted = false;
195         char *sub_path = NULL;
196         int r = 0;
197
198         while ((dent = readdir(d))) {
199                 struct stat s;
200                 usec_t age;
201
202                 if (streq(dent->d_name, ".") ||
203                     streq(dent->d_name, ".."))
204                         continue;
205
206                 if (fstatat(dirfd(d), dent->d_name, &s, AT_SYMLINK_NOFOLLOW) < 0) {
207
208                         if (errno != ENOENT) {
209                                 log_error("stat(%s/%s) failed: %m", p, dent->d_name);
210                                 r = -errno;
211                         }
212
213                         continue;
214                 }
215
216                 /* Stay on the same filesystem */
217                 if (s.st_dev != rootdev)
218                         continue;
219
220                 /* Do not delete read-only files owned by root */
221                 if (s.st_uid == 0 && !(s.st_mode & S_IWUSR))
222                         continue;
223
224                 free(sub_path);
225                 sub_path = NULL;
226
227                 if (asprintf(&sub_path, "%s/%s", p, dent->d_name) < 0) {
228                         log_error("Out of memory");
229                         r = -ENOMEM;
230                         goto finish;
231                 }
232
233                 /* Is there an item configured for this path? */
234                 if (hashmap_get(items, sub_path))
235                         continue;
236
237                 if (find_glob(globs, sub_path))
238                         continue;
239
240                 if (S_ISDIR(s.st_mode)) {
241
242                         if (mountpoint &&
243                             streq(dent->d_name, "lost+found") &&
244                             s.st_uid == 0)
245                                 continue;
246
247                         if (maxdepth <= 0)
248                                 log_warning("Reached max depth on %s.", sub_path);
249                         else {
250                                 DIR *sub_dir;
251                                 int q;
252
253                                 sub_dir = xopendirat(dirfd(d), dent->d_name, O_NOFOLLOW);
254                                 if (sub_dir == NULL) {
255                                         if (errno != ENOENT) {
256                                                 log_error("opendir(%s/%s) failed: %m", p, dent->d_name);
257                                                 r = -errno;
258                                         }
259
260                                         continue;
261                                 }
262
263                                 q = dir_cleanup(sub_path, sub_dir, &s, cutoff, rootdev, false, maxdepth-1);
264                                 closedir(sub_dir);
265
266                                 if (q < 0)
267                                         r = q;
268                         }
269
270                         /* Ignore ctime, we change it when deleting */
271                         age = MAX(timespec_load(&s.st_mtim),
272                                   timespec_load(&s.st_atim));
273                         if (age >= cutoff)
274                                 continue;
275
276                         log_debug("rmdir '%s'\n", sub_path);
277
278                         if (unlinkat(dirfd(d), dent->d_name, AT_REMOVEDIR) < 0) {
279                                 if (errno != ENOENT && errno != ENOTEMPTY) {
280                                         log_error("rmdir(%s): %m", sub_path);
281                                         r = -errno;
282                                 }
283                         }
284
285                 } else {
286                         /* Skip files for which the sticky bit is
287                          * set. These are semantics we define, and are
288                          * unknown elsewhere. See XDG_RUNTIME_DIR
289                          * specification for details. */
290                         if (s.st_mode & S_ISVTX)
291                                 continue;
292
293                         if (mountpoint && S_ISREG(s.st_mode)) {
294                                 if (streq(dent->d_name, ".journal") &&
295                                     s.st_uid == 0)
296                                         continue;
297
298                                 if (streq(dent->d_name, "aquota.user") ||
299                                     streq(dent->d_name, "aquota.group"))
300                                         continue;
301                         }
302
303                         /* Ignore sockets that are listed in /proc/net/unix */
304                         if (S_ISSOCK(s.st_mode) && unix_socket_alive(sub_path))
305                                 continue;
306
307                         /* Ignore device nodes */
308                         if (S_ISCHR(s.st_mode) || S_ISBLK(s.st_mode))
309                                 continue;
310
311                         age = MAX3(timespec_load(&s.st_mtim),
312                                    timespec_load(&s.st_atim),
313                                    timespec_load(&s.st_ctim));
314
315                         if (age >= cutoff)
316                                 continue;
317
318                         log_debug("unlink '%s'\n", sub_path);
319
320                         if (unlinkat(dirfd(d), dent->d_name, 0) < 0) {
321                                 if (errno != ENOENT) {
322                                         log_error("unlink(%s): %m", sub_path);
323                                         r = -errno;
324                                 }
325                         }
326
327                         deleted = true;
328                 }
329         }
330
331 finish:
332         if (deleted) {
333                 /* Restore original directory timestamps */
334                 times[0] = ds->st_atim;
335                 times[1] = ds->st_mtim;
336
337                 if (futimens(dirfd(d), times) < 0)
338                         log_error("utimensat(%s): %m", p);
339         }
340
341         free(sub_path);
342
343         return r;
344 }
345
346 static int clean_item(Item *i) {
347         DIR *d;
348         struct stat s, ps;
349         bool mountpoint;
350         int r;
351         usec_t cutoff, n;
352
353         assert(i);
354
355         if (i->type != CREATE_DIRECTORY &&
356             i->type != TRUNCATE_DIRECTORY &&
357             i->type != IGNORE_PATH)
358                 return 0;
359
360         if (!i->age_set || i->age <= 0)
361                 return 0;
362
363         n = now(CLOCK_REALTIME);
364         if (n < i->age)
365                 return 0;
366
367         cutoff = n - i->age;
368
369         d = opendir(i->path);
370         if (!d) {
371                 if (errno == ENOENT)
372                         return 0;
373
374                 log_error("Failed to open directory %s: %m", i->path);
375                 return -errno;
376         }
377
378         if (fstat(dirfd(d), &s) < 0) {
379                 log_error("stat(%s) failed: %m", i->path);
380                 r = -errno;
381                 goto finish;
382         }
383
384         if (!S_ISDIR(s.st_mode)) {
385                 log_error("%s is not a directory.", i->path);
386                 r = -ENOTDIR;
387                 goto finish;
388         }
389
390         if (fstatat(dirfd(d), "..", &ps, AT_SYMLINK_NOFOLLOW) != 0) {
391                 log_error("stat(%s/..) failed: %m", i->path);
392                 r = -errno;
393                 goto finish;
394         }
395
396         mountpoint = s.st_dev != ps.st_dev ||
397                      (s.st_dev == ps.st_dev && s.st_ino == ps.st_ino);
398
399         r = dir_cleanup(i->path, d, &s, cutoff, s.st_dev, mountpoint, MAX_DEPTH);
400
401 finish:
402         if (d)
403                 closedir(d);
404
405         return r;
406 }
407
408 static int create_item(Item *i) {
409         int fd = -1, r;
410         mode_t u;
411         struct stat st;
412
413         assert(i);
414
415         switch (i->type) {
416
417         case IGNORE_PATH:
418         case REMOVE_PATH:
419         case RECURSIVE_REMOVE_PATH:
420                 return 0;
421
422         case CREATE_FILE:
423         case TRUNCATE_FILE:
424
425                 u = umask(0);
426                 fd = open(i->path, O_CREAT|O_NDELAY|O_CLOEXEC|O_WRONLY|O_NOCTTY|O_NOFOLLOW|
427                           (i->type == TRUNCATE_FILE ? O_TRUNC : 0), i->mode);
428                 umask(u);
429
430                 if (fd < 0) {
431                         log_error("Failed to create file %s: %m", i->path);
432                         r = -errno;
433                         goto finish;
434                 }
435
436                 if (fstat(fd, &st) < 0) {
437                         log_error("stat(%s) failed: %m", i->path);
438                         r = -errno;
439                         goto finish;
440                 }
441
442                 if (!S_ISREG(st.st_mode)) {
443                         log_error("%s is not a file.", i->path);
444                         r = -EEXIST;
445                         goto finish;
446                 }
447
448                 if (i->mode_set)
449                         if (fchmod(fd, i->mode) < 0) {
450                                 log_error("chmod(%s) failed: %m", i->path);
451                                 r = -errno;
452                                 goto finish;
453                         }
454
455                 if (i->uid_set || i->gid_set)
456                         if (fchown(fd,
457                                    i->uid_set ? i->uid : (uid_t) -1,
458                                    i->gid_set ? i->gid : (gid_t) -1) < 0) {
459                                 log_error("chown(%s) failed: %m", i->path);
460                                 r = -errno;
461                                 goto finish;
462                         }
463
464                 break;
465
466         case TRUNCATE_DIRECTORY:
467         case CREATE_DIRECTORY:
468
469                 u = umask(0);
470                 mkdir_parents(i->path, 0755);
471                 r = mkdir(i->path, i->mode);
472                 umask(u);
473
474                 if (r < 0 && errno != EEXIST) {
475                         log_error("Failed to create directory %s: %m", i->path);
476                         r = -errno;
477                         goto finish;
478                 }
479
480                 if (stat(i->path, &st) < 0) {
481                         log_error("stat(%s) failed: %m", i->path);
482                         r = -errno;
483                         goto finish;
484                 }
485
486                 if (!S_ISDIR(st.st_mode)) {
487                         log_error("%s is not a directory.", i->path);
488                         r = -EEXIST;
489                         goto finish;
490                 }
491
492                 if (i->mode_set)
493                         if (chmod(i->path, i->mode) < 0) {
494                                 log_error("chmod(%s) failed: %m", i->path);
495                                 r = -errno;
496                                 goto finish;
497                         }
498
499                 if (i->uid_set || i->gid_set)
500                         if (chown(i->path,
501                                   i->uid_set ? i->uid : (uid_t) -1,
502                                   i->gid_set ? i->gid : (gid_t) -1) < 0) {
503
504                                 log_error("chown(%s) failed: %m", i->path);
505                                 r = -errno;
506                                 goto finish;
507                         }
508
509                 break;
510
511         case CREATE_FIFO:
512
513                 u = umask(0);
514                 r = mkfifo(i->path, i->mode);
515                 umask(u);
516
517                 if (r < 0 && errno != EEXIST) {
518                         log_error("Failed to create fifo %s: %m", i->path);
519                         r = -errno;
520                         goto finish;
521                 }
522
523                 if (stat(i->path, &st) < 0) {
524                         log_error("stat(%s) failed: %m", i->path);
525                         r = -errno;
526                         goto finish;
527                 }
528
529                 if (!S_ISFIFO(st.st_mode)) {
530                         log_error("%s is not a fifo.", i->path);
531                         r = -EEXIST;
532                         goto finish;
533                 }
534
535                 if (i->mode_set)
536                         if (chmod(i->path, i->mode) < 0) {
537                                 log_error("chmod(%s) failed: %m", i->path);
538                                 r = -errno;
539                                 goto finish;
540                         }
541
542                 if (i->uid_set || i->gid_set)
543                         if (chown(i->path,
544                                    i->uid_set ? i->uid : (uid_t) -1,
545                                    i->gid_set ? i->gid : (gid_t) -1) < 0) {
546                                 log_error("chown(%s) failed: %m", i->path);
547                                 r = -errno;
548                                 goto finish;
549                         }
550
551                 break;
552         }
553
554         if ((r = label_fix(i->path, false)) < 0)
555                 goto finish;
556
557         log_debug("%s created successfully.", i->path);
558
559 finish:
560         if (fd >= 0)
561                 close_nointr_nofail(fd);
562
563         return r;
564 }
565
566 static int remove_item(Item *i, const char *instance) {
567         int r;
568
569         assert(i);
570
571         switch (i->type) {
572
573         case CREATE_FILE:
574         case TRUNCATE_FILE:
575         case CREATE_DIRECTORY:
576         case CREATE_FIFO:
577         case IGNORE_PATH:
578                 break;
579
580         case REMOVE_PATH:
581                 if (remove(instance) < 0 && errno != ENOENT) {
582                         log_error("remove(%s): %m", instance);
583                         return -errno;
584                 }
585
586                 break;
587
588         case TRUNCATE_DIRECTORY:
589         case RECURSIVE_REMOVE_PATH:
590                 if ((r = rm_rf(instance, false, i->type == RECURSIVE_REMOVE_PATH, false)) < 0 &&
591                     r != -ENOENT) {
592                         log_error("rm_rf(%s): %s", instance, strerror(-r));
593                         return r;
594                 }
595
596                 break;
597         }
598
599         return 0;
600 }
601
602 static int remove_item_glob(Item *i) {
603         assert(i);
604
605         switch (i->type) {
606
607         case CREATE_FILE:
608         case TRUNCATE_FILE:
609         case CREATE_DIRECTORY:
610         case CREATE_FIFO:
611         case IGNORE_PATH:
612                 break;
613
614         case REMOVE_PATH:
615         case TRUNCATE_DIRECTORY:
616         case RECURSIVE_REMOVE_PATH: {
617                 int r = 0, k;
618                 glob_t g;
619                 char **fn;
620
621                 zero(g);
622
623                 errno = 0;
624                 if ((k = glob(i->path, GLOB_NOSORT|GLOB_BRACE, NULL, &g)) != 0) {
625
626                         if (k != GLOB_NOMATCH) {
627                                 if (errno != 0)
628                                         errno = EIO;
629
630                                 log_error("glob(%s) failed: %m", i->path);
631                                 return -errno;
632                         }
633                 }
634
635                 STRV_FOREACH(fn, g.gl_pathv)
636                         if ((k = remove_item(i, *fn)) < 0)
637                                 r = k;
638
639                 globfree(&g);
640                 return r;
641         }
642         }
643
644         return 0;
645 }
646
647 static int process_item(Item *i) {
648         int r, q, p;
649
650         assert(i);
651
652         r = arg_create ? create_item(i) : 0;
653         q = arg_remove ? remove_item_glob(i) : 0;
654         p = arg_clean ? clean_item(i) : 0;
655
656         if (r < 0)
657                 return r;
658
659         if (q < 0)
660                 return q;
661
662         return p;
663 }
664
665 static void item_free(Item *i) {
666         assert(i);
667
668         free(i->path);
669         free(i);
670 }
671
672 static bool item_equal(Item *a, Item *b) {
673         assert(a);
674         assert(b);
675
676         if (!streq_ptr(a->path, b->path))
677                 return false;
678
679         if (a->type != b->type)
680                 return false;
681
682         if (a->uid_set != b->uid_set ||
683             (a->uid_set && a->uid != b->uid))
684             return false;
685
686         if (a->gid_set != b->gid_set ||
687             (a->gid_set && a->gid != b->gid))
688             return false;
689
690         if (a->mode_set != b->mode_set ||
691             (a->mode_set && a->mode != b->mode))
692             return false;
693
694         if (a->age_set != b->age_set ||
695             (a->age_set && a->age != b->age))
696             return false;
697
698         return true;
699 }
700
701 static int parse_line(const char *fname, unsigned line, const char *buffer) {
702         Item *i, *existing;
703         char *mode = NULL, *user = NULL, *group = NULL, *age = NULL;
704         char type;
705         Hashmap *h;
706         int r;
707
708         assert(fname);
709         assert(line >= 1);
710         assert(buffer);
711
712         if (!(i = new0(Item, 1))) {
713                 log_error("Out of memory");
714                 return -ENOMEM;
715         }
716
717         if (sscanf(buffer,
718                    "%c "
719                    "%ms "
720                    "%ms "
721                    "%ms "
722                    "%ms "
723                    "%ms",
724                    &type,
725                    &i->path,
726                    &mode,
727                    &user,
728                    &group,
729                    &age) < 2) {
730                 log_error("[%s:%u] Syntax error.", fname, line);
731                 r = -EIO;
732                 goto finish;
733         }
734         i->type = type;
735
736         if (i->type != CREATE_FILE &&
737             i->type != TRUNCATE_FILE &&
738             i->type != CREATE_DIRECTORY &&
739             i->type != TRUNCATE_DIRECTORY &&
740             i->type != CREATE_FIFO &&
741             i->type != IGNORE_PATH &&
742             i->type != REMOVE_PATH &&
743             i->type != RECURSIVE_REMOVE_PATH) {
744                 log_error("[%s:%u] Unknown file type '%c'.", fname, line, i->type);
745                 r = -EBADMSG;
746                 goto finish;
747         }
748
749         if (!path_is_absolute(i->path)) {
750                 log_error("[%s:%u] Path '%s' not absolute.", fname, line, i->path);
751                 r = -EBADMSG;
752                 goto finish;
753         }
754
755         path_kill_slashes(i->path);
756
757         if (arg_prefix && !path_startswith(i->path, arg_prefix)) {
758                 r = 0;
759                 goto finish;
760         }
761
762         if (user && !streq(user, "-")) {
763                 const char *u = user;
764
765                 r = get_user_creds(&u, &i->uid, NULL, NULL);
766                 if (r < 0) {
767                         log_error("[%s:%u] Unknown user '%s'.", fname, line, user);
768                         goto finish;
769                 }
770
771                 i->uid_set = true;
772         }
773
774         if (group && !streq(group, "-")) {
775                 const char *g = group;
776
777                 r = get_group_creds(&g, &i->gid);
778                 if (r < 0) {
779                         log_error("[%s:%u] Unknown group '%s'.", fname, line, group);
780                         goto finish;
781                 }
782
783                 i->gid_set = true;
784         }
785
786         if (mode && !streq(mode, "-")) {
787                 unsigned m;
788
789                 if (sscanf(mode, "%o", &m) != 1) {
790                         log_error("[%s:%u] Invalid mode '%s'.", fname, line, mode);
791                         r = -ENOENT;
792                         goto finish;
793                 }
794
795                 i->mode = m;
796                 i->mode_set = true;
797         } else
798                 i->mode = i->type == CREATE_DIRECTORY ? 0755 : 0644;
799
800         if (age && !streq(age, "-")) {
801                 if (parse_usec(age, &i->age) < 0) {
802                         log_error("[%s:%u] Invalid age '%s'.", fname, line, age);
803                         r = -EBADMSG;
804                         goto finish;
805                 }
806
807                 i->age_set = true;
808         }
809
810         h = needs_glob(i->type) ? globs : items;
811
812         if ((existing = hashmap_get(h, i->path))) {
813
814                 /* Two identical items are fine */
815                 if (!item_equal(existing, i))
816                         log_warning("Two or more conflicting lines for %s configured, ignoring.", i->path);
817
818                 r = 0;
819                 goto finish;
820         }
821
822         if ((r = hashmap_put(h, i->path, i)) < 0) {
823                 log_error("Failed to insert item %s: %s", i->path, strerror(-r));
824                 goto finish;
825         }
826
827         i = NULL;
828         r = 0;
829
830 finish:
831         free(user);
832         free(group);
833         free(mode);
834         free(age);
835
836         if (i)
837                 item_free(i);
838
839         return r;
840 }
841
842 static int help(void) {
843
844         printf("%s [OPTIONS...] [CONFIGURATION FILE...]\n\n"
845                "Creates, deletes and cleans up volatile and temporary files and directories.\n\n"
846                "  -h --help             Show this help\n"
847                "     --create           Create marked files/directories\n"
848                "     --clean            Clean up marked directories\n"
849                "     --remove           Remove marked files/directories\n"
850                "     --prefix=PATH      Only apply rules that apply to paths with the specified prefix\n",
851                program_invocation_short_name);
852
853         return 0;
854 }
855
856 static int parse_argv(int argc, char *argv[]) {
857
858         enum {
859                 ARG_CREATE,
860                 ARG_CLEAN,
861                 ARG_REMOVE,
862                 ARG_PREFIX
863         };
864
865         static const struct option options[] = {
866                 { "help",      no_argument,       NULL, 'h'           },
867                 { "create",    no_argument,       NULL, ARG_CREATE    },
868                 { "clean",     no_argument,       NULL, ARG_CLEAN     },
869                 { "remove",    no_argument,       NULL, ARG_REMOVE    },
870                 { "prefix",    required_argument, NULL, ARG_PREFIX    },
871                 { NULL,        0,                 NULL, 0             }
872         };
873
874         int c;
875
876         assert(argc >= 0);
877         assert(argv);
878
879         while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0) {
880
881                 switch (c) {
882
883                 case 'h':
884                         help();
885                         return 0;
886
887                 case ARG_CREATE:
888                         arg_create = true;
889                         break;
890
891                 case ARG_CLEAN:
892                         arg_clean = true;
893                         break;
894
895                 case ARG_REMOVE:
896                         arg_remove = true;
897                         break;
898
899                 case ARG_PREFIX:
900                         arg_prefix = optarg;
901                         break;
902
903                 case '?':
904                         return -EINVAL;
905
906                 default:
907                         log_error("Unknown option code %c", c);
908                         return -EINVAL;
909                 }
910         }
911
912         if (!arg_clean && !arg_create && !arg_remove) {
913                 log_error("You need to specify at least one of --clean, --create or --remove.");
914                 return -EINVAL;
915         }
916
917         return 1;
918 }
919
920 static int read_config_file(const char *fn, bool ignore_enoent) {
921         FILE *f;
922         unsigned v = 0;
923         int r = 0;
924
925         assert(fn);
926
927         if (!(f = fopen(fn, "re"))) {
928
929                 if (ignore_enoent && errno == ENOENT)
930                         return 0;
931
932                 log_error("Failed to open %s: %m", fn);
933                 return -errno;
934         }
935
936         log_debug("apply: %s\n", fn);
937         for (;;) {
938                 char line[LINE_MAX], *l;
939                 int k;
940
941                 if (!(fgets(line, sizeof(line), f)))
942                         break;
943
944                 v++;
945
946                 l = strstrip(line);
947                 if (*l == '#' || *l == 0)
948                         continue;
949
950                 if ((k = parse_line(fn, v, l)) < 0)
951                         if (r == 0)
952                                 r = k;
953         }
954
955         if (ferror(f)) {
956                 log_error("Failed to read from file %s: %m", fn);
957                 if (r == 0)
958                         r = -EIO;
959         }
960
961         fclose(f);
962
963         return r;
964 }
965
966 int main(int argc, char *argv[]) {
967         int r;
968         Item *i;
969         Iterator iterator;
970
971         if ((r = parse_argv(argc, argv)) <= 0)
972                 return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS;
973
974         log_set_target(LOG_TARGET_AUTO);
975         log_parse_environment();
976         log_open();
977
978         umask(0022);
979
980         label_init();
981
982         items = hashmap_new(string_hash_func, string_compare_func);
983         globs = hashmap_new(string_hash_func, string_compare_func);
984
985         if (!items || !globs) {
986                 log_error("Out of memory");
987                 r = EXIT_FAILURE;
988                 goto finish;
989         }
990
991         r = EXIT_SUCCESS;
992
993         if (optind < argc) {
994                 int j;
995
996                 for (j = optind; j < argc; j++)
997                         if (read_config_file(argv[j], false) < 0)
998                                 r = EXIT_FAILURE;
999
1000         } else {
1001                 char **files, **f;
1002
1003                 r = conf_files_list(&files, ".conf",
1004                                     "/run/tmpfiles.d",
1005                                     "/etc/tmpfiles.d",
1006                                     "/usr/local/lib/tmpfiles.d",
1007                                     "/usr/lib/tmpfiles.d",
1008                                     NULL);
1009                 if (r < 0) {
1010                         r = EXIT_FAILURE;
1011                         log_error("Failed to enumerate tmpfiles.d files: %s", strerror(-r));
1012                         goto finish;
1013                 }
1014
1015                 STRV_FOREACH(f, files) {
1016                         if (read_config_file(*f, true) < 0)
1017                                 r = EXIT_FAILURE;
1018                 }
1019
1020                 strv_free(files);
1021         }
1022
1023         HASHMAP_FOREACH(i, globs, iterator)
1024                 process_item(i);
1025
1026         HASHMAP_FOREACH(i, items, iterator)
1027                 process_item(i);
1028
1029 finish:
1030         while ((i = hashmap_steal_first(items)))
1031                 item_free(i);
1032
1033         while ((i = hashmap_steal_first(globs)))
1034                 item_free(i);
1035
1036         hashmap_free(items);
1037         hashmap_free(globs);
1038
1039         set_free_free(unix_sockets);
1040
1041         label_finish();
1042
1043         return r;
1044 }