chiark / gitweb /
util: overflow hardening
[elogind.git] / src / shared / conf-files.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 <stdio.h>
28 #include <sys/stat.h>
29 #include <dirent.h>
30
31 #include "macro.h"
32 #include "util.h"
33 #include "missing.h"
34 #include "log.h"
35 #include "strv.h"
36 #include "path-util.h"
37 #include "hashmap.h"
38 #include "conf-files.h"
39
40 static int files_add(Hashmap *h, const char *path, const char *suffix) {
41         DIR *dir;
42         int r = 0;
43
44         dir = opendir(path);
45         if (!dir) {
46                 if (errno == ENOENT)
47                         return 0;
48                 return -errno;
49         }
50
51         for (;;) {
52                 struct dirent *de;
53                 union dirent_storage buf;
54                 int k;
55                 char *p;
56
57                 k = readdir_r(dir, &buf.de, &de);
58                 if (k != 0) {
59                         r = -k;
60                         goto finish;
61                 }
62
63                 if (!de)
64                         break;
65
66                 if (!dirent_is_file_with_suffix(de, suffix))
67                         continue;
68
69                 if (asprintf(&p, "%s/%s", path, de->d_name) < 0) {
70                         r = -ENOMEM;
71                         goto finish;
72                 }
73
74                 if (hashmap_put(h, path_get_file_name(p), p) <= 0) {
75                         log_debug("Skip overridden file: %s.", p);
76                         free(p);
77                 }
78         }
79
80 finish:
81         closedir(dir);
82         return r;
83 }
84
85 static int base_cmp(const void *a, const void *b) {
86         const char *s1, *s2;
87
88         s1 = *(char * const *)a;
89         s2 = *(char * const *)b;
90         return strcmp(path_get_file_name(s1), path_get_file_name(s2));
91 }
92
93 int conf_files_list_strv(char ***strv, const char *suffix, const char **dirs) {
94         Hashmap *fh = NULL;
95         char **files = NULL;
96         const char **p;
97         int r;
98
99         assert(dirs);
100
101         fh = hashmap_new(string_hash_func, string_compare_func);
102         if (!fh) {
103                 r = -ENOMEM;
104                 goto finish;
105         }
106
107         STRV_FOREACH(p, dirs) {
108                 r = files_add(fh, *p, suffix);
109                 if (r < 0)
110                         log_warning("Failed to search for files in %s: %s",
111                                     *p, strerror(-r));
112         }
113
114         files = hashmap_get_strv(fh);
115         if (files == NULL) {
116                 log_error("Failed to compose list of files.");
117                 r = -ENOMEM;
118                 goto finish;
119         }
120         qsort(files, hashmap_size(fh), sizeof(char *), base_cmp);
121         r = 0;
122
123 finish:
124         hashmap_free(fh);
125         *strv = files;
126         return r;
127 }
128
129 int conf_files_list(char ***strv, const char *suffix, const char *dir, ...) {
130         char **dirs = NULL;
131         va_list ap;
132         int r;
133
134         va_start(ap, dir);
135         dirs = strv_new_ap(dir, ap);
136         va_end(ap);
137         if (!dirs) {
138                 r = -ENOMEM;
139                 goto finish;
140         }
141
142         if (!path_strv_canonicalize(dirs)) {
143                 r = -ENOMEM;
144                 goto finish;
145         }
146         strv_uniq(dirs);
147
148         r = conf_files_list_strv(strv, suffix, (const char **)dirs);
149
150 finish:
151         strv_free(dirs);
152         return r;
153 }