]>
Commit | Line | Data |
---|---|---|
663996b3 MS |
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 <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 <getopt.h> | |
30 | #include <libkmod.h> | |
31 | ||
32 | #include "log.h" | |
33 | #include "util.h" | |
34 | #include "strv.h" | |
35 | #include "conf-files.h" | |
663996b3 | 36 | #include "fileio.h" |
60f067b4 | 37 | #include "build.h" |
663996b3 MS |
38 | |
39 | static char **arg_proc_cmdline_modules = NULL; | |
40 | ||
41 | static const char conf_file_dirs[] = | |
42 | "/etc/modules-load.d\0" | |
43 | "/run/modules-load.d\0" | |
44 | "/usr/local/lib/modules-load.d\0" | |
45 | "/usr/lib/modules-load.d\0" | |
46 | #ifdef HAVE_SPLIT_USR | |
47 | "/lib/modules-load.d\0" | |
48 | #endif | |
49 | ; | |
50 | ||
663996b3 | 51 | static void systemd_kmod_log(void *data, int priority, const char *file, int line, |
60f067b4 JS |
52 | const char *fn, const char *format, va_list args) { |
53 | ||
54 | DISABLE_WARNING_FORMAT_NONLITERAL; | |
663996b3 | 55 | log_metav(priority, file, line, fn, format, args); |
60f067b4 | 56 | REENABLE_WARNING; |
663996b3 | 57 | } |
663996b3 MS |
58 | |
59 | static int add_modules(const char *p) { | |
663996b3 MS |
60 | _cleanup_strv_free_ char **k = NULL; |
61 | ||
62 | k = strv_split(p, ","); | |
63 | if (!k) | |
64 | return log_oom(); | |
65 | ||
60f067b4 | 66 | if (strv_extend_strv(&arg_proc_cmdline_modules, k) < 0) |
663996b3 MS |
67 | return log_oom(); |
68 | ||
663996b3 MS |
69 | return 0; |
70 | } | |
71 | ||
60f067b4 | 72 | static int parse_proc_cmdline_item(const char *key, const char *value) { |
663996b3 | 73 | int r; |
663996b3 | 74 | |
60f067b4 JS |
75 | if (STR_IN_SET(key, "modules-load", "rd.modules-load") && value) { |
76 | r = add_modules(value); | |
77 | if (r < 0) | |
78 | return r; | |
663996b3 MS |
79 | } |
80 | ||
81 | return 0; | |
82 | } | |
83 | ||
84 | static int load_module(struct kmod_ctx *ctx, const char *m) { | |
85 | const int probe_flags = KMOD_PROBE_APPLY_BLACKLIST; | |
86 | struct kmod_list *itr, *modlist = NULL; | |
87 | int r = 0; | |
88 | ||
60f067b4 | 89 | log_debug("load: %s", m); |
663996b3 MS |
90 | |
91 | r = kmod_module_new_from_lookup(ctx, m, &modlist); | |
92 | if (r < 0) { | |
93 | log_error("Failed to lookup alias '%s': %s", m, strerror(-r)); | |
94 | return r; | |
95 | } | |
96 | ||
97 | if (!modlist) { | |
98 | log_error("Failed to find module '%s'", m); | |
99 | return -ENOENT; | |
100 | } | |
101 | ||
102 | kmod_list_foreach(itr, modlist) { | |
103 | struct kmod_module *mod; | |
104 | int state, err; | |
105 | ||
106 | mod = kmod_module_get_module(itr); | |
107 | state = kmod_module_get_initstate(mod); | |
108 | ||
109 | switch (state) { | |
110 | case KMOD_MODULE_BUILTIN: | |
111 | log_info("Module '%s' is builtin", kmod_module_get_name(mod)); | |
112 | break; | |
113 | ||
114 | case KMOD_MODULE_LIVE: | |
115 | log_debug("Module '%s' is already loaded", kmod_module_get_name(mod)); | |
116 | break; | |
117 | ||
118 | default: | |
119 | err = kmod_module_probe_insert_module(mod, probe_flags, | |
120 | NULL, NULL, NULL, NULL); | |
121 | ||
122 | if (err == 0) | |
123 | log_info("Inserted module '%s'", kmod_module_get_name(mod)); | |
124 | else if (err == KMOD_PROBE_APPLY_BLACKLIST) | |
125 | log_info("Module '%s' is blacklisted", kmod_module_get_name(mod)); | |
126 | else { | |
127 | log_error("Failed to insert '%s': %s", kmod_module_get_name(mod), | |
128 | strerror(-err)); | |
129 | r = err; | |
130 | } | |
131 | } | |
132 | ||
133 | kmod_module_unref(mod); | |
134 | } | |
135 | ||
136 | kmod_module_unref_list(modlist); | |
137 | ||
138 | return r; | |
139 | } | |
140 | ||
141 | static int apply_file(struct kmod_ctx *ctx, const char *path, bool ignore_enoent) { | |
142 | _cleanup_fclose_ FILE *f = NULL; | |
143 | int r; | |
144 | ||
145 | assert(ctx); | |
146 | assert(path); | |
147 | ||
60f067b4 | 148 | r = search_and_fopen_nulstr(path, "re", NULL, conf_file_dirs, &f); |
663996b3 MS |
149 | if (r < 0) { |
150 | if (ignore_enoent && r == -ENOENT) | |
151 | return 0; | |
152 | ||
153 | log_error("Failed to open %s, ignoring: %s", path, strerror(-r)); | |
154 | return r; | |
155 | } | |
156 | ||
60f067b4 | 157 | log_debug("apply: %s", path); |
663996b3 MS |
158 | for (;;) { |
159 | char line[LINE_MAX], *l; | |
160 | int k; | |
161 | ||
162 | if (!fgets(line, sizeof(line), f)) { | |
163 | if (feof(f)) | |
164 | break; | |
165 | ||
166 | log_error("Failed to read file '%s', ignoring: %m", path); | |
167 | return -errno; | |
168 | } | |
169 | ||
170 | l = strstrip(line); | |
171 | if (!*l) | |
172 | continue; | |
173 | if (strchr(COMMENTS "\n", *l)) | |
174 | continue; | |
175 | ||
176 | k = load_module(ctx, l); | |
177 | if (k < 0 && r == 0) | |
178 | r = k; | |
179 | } | |
180 | ||
181 | return r; | |
182 | } | |
183 | ||
184 | static int help(void) { | |
185 | ||
186 | printf("%s [OPTIONS...] [CONFIGURATION FILE...]\n\n" | |
187 | "Loads statically configured kernel modules.\n\n" | |
60f067b4 JS |
188 | " -h --help Show this help\n" |
189 | " --version Show package version\n", | |
663996b3 MS |
190 | program_invocation_short_name); |
191 | ||
192 | return 0; | |
193 | } | |
194 | ||
195 | static int parse_argv(int argc, char *argv[]) { | |
196 | ||
60f067b4 JS |
197 | enum { |
198 | ARG_VERSION = 0x100, | |
199 | }; | |
200 | ||
663996b3 MS |
201 | static const struct option options[] = { |
202 | { "help", no_argument, NULL, 'h' }, | |
60f067b4 JS |
203 | { "version", no_argument, NULL, ARG_VERSION }, |
204 | {} | |
663996b3 MS |
205 | }; |
206 | ||
207 | int c; | |
208 | ||
209 | assert(argc >= 0); | |
210 | assert(argv); | |
211 | ||
212 | while ((c = getopt_long(argc, argv, "h", options, NULL)) >= 0) { | |
213 | ||
214 | switch (c) { | |
215 | ||
216 | case 'h': | |
60f067b4 JS |
217 | return help(); |
218 | ||
219 | case ARG_VERSION: | |
220 | puts(PACKAGE_STRING); | |
221 | puts(SYSTEMD_FEATURES); | |
663996b3 MS |
222 | return 0; |
223 | ||
224 | case '?': | |
225 | return -EINVAL; | |
226 | ||
227 | default: | |
60f067b4 | 228 | assert_not_reached("Unhandled option"); |
663996b3 MS |
229 | } |
230 | } | |
231 | ||
232 | return 1; | |
233 | } | |
234 | ||
235 | int main(int argc, char *argv[]) { | |
236 | int r, k; | |
237 | struct kmod_ctx *ctx; | |
238 | ||
239 | r = parse_argv(argc, argv); | |
240 | if (r <= 0) | |
241 | return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; | |
242 | ||
243 | log_set_target(LOG_TARGET_AUTO); | |
244 | log_parse_environment(); | |
245 | log_open(); | |
246 | ||
247 | umask(0022); | |
248 | ||
60f067b4 | 249 | if (parse_proc_cmdline(parse_proc_cmdline_item) < 0) |
663996b3 MS |
250 | return EXIT_FAILURE; |
251 | ||
252 | ctx = kmod_new(NULL, NULL); | |
253 | if (!ctx) { | |
254 | log_error("Failed to allocate memory for kmod."); | |
255 | goto finish; | |
256 | } | |
257 | ||
258 | kmod_load_resources(ctx); | |
259 | kmod_set_log_fn(ctx, systemd_kmod_log, NULL); | |
260 | ||
261 | r = 0; | |
262 | ||
263 | if (argc > optind) { | |
264 | int i; | |
265 | ||
266 | for (i = optind; i < argc; i++) { | |
267 | k = apply_file(ctx, argv[i], false); | |
268 | if (k < 0 && r == 0) | |
269 | r = k; | |
270 | } | |
271 | ||
272 | } else { | |
273 | _cleanup_free_ char **files = NULL; | |
274 | char **fn, **i; | |
275 | ||
276 | STRV_FOREACH(i, arg_proc_cmdline_modules) { | |
277 | k = load_module(ctx, *i); | |
60f067b4 JS |
278 | if (k < 0 && r == 0) |
279 | r = k; | |
663996b3 MS |
280 | } |
281 | ||
60f067b4 JS |
282 | k = conf_files_list_nulstr(&files, ".conf", NULL, conf_file_dirs); |
283 | if (k < 0) { | |
284 | log_error("Failed to enumerate modules-load.d files: %s", strerror(-k)); | |
285 | if (r == 0) | |
286 | r = k; | |
663996b3 MS |
287 | goto finish; |
288 | } | |
289 | ||
290 | STRV_FOREACH(fn, files) { | |
291 | k = apply_file(ctx, *fn, true); | |
292 | if (k < 0 && r == 0) | |
293 | r = k; | |
294 | } | |
295 | } | |
296 | ||
297 | finish: | |
298 | kmod_unref(ctx); | |
299 | strv_free(arg_proc_cmdline_modules); | |
300 | ||
301 | return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; | |
302 | } |