Contents of /trunk/mkinitrd-magellan/busybox/modutils/modprobe.c
Parent Directory | Revision Log
Revision 984 -
(show annotations)
(download)
Sun May 30 11:32:42 2010 UTC (14 years, 3 months ago) by niro
File MIME type: text/plain
File size: 12641 byte(s)
Sun May 30 11:32:42 2010 UTC (14 years, 3 months ago) by niro
File MIME type: text/plain
File size: 12641 byte(s)
-updated to busybox-1.16.1 and enabled blkid/uuid support in default config
1 | /* vi: set sw=4 ts=4: */ |
2 | /* |
3 | * Modprobe written from scratch for BusyBox |
4 | * |
5 | * Copyright (c) 2008 Timo Teras <timo.teras@iki.fi> |
6 | * Copyright (c) 2008 Vladimir Dronnikov |
7 | * |
8 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
9 | */ |
10 | |
11 | /* Note that unlike older versions of modules.dep/depmod (busybox and m-i-t), |
12 | * we expect the full dependency list to be specified in modules.dep. |
13 | * Older versions would only export the direct dependency list. |
14 | */ |
15 | #include "libbb.h" |
16 | #include "modutils.h" |
17 | #include <sys/utsname.h> |
18 | #include <fnmatch.h> |
19 | |
20 | //#define DBG(fmt, ...) bb_error_msg("%s: " fmt, __func__, ## __VA_ARGS__) |
21 | #define DBG(...) ((void)0) |
22 | |
23 | #define MODULE_FLAG_LOADED 0x0001 |
24 | #define MODULE_FLAG_NEED_DEPS 0x0002 |
25 | /* "was seen in modules.dep": */ |
26 | #define MODULE_FLAG_FOUND_IN_MODDEP 0x0004 |
27 | #define MODULE_FLAG_BLACKLISTED 0x0008 |
28 | |
29 | struct module_entry { /* I'll call it ME. */ |
30 | unsigned flags; |
31 | char *modname; /* stripped of /path/, .ext and s/-/_/g */ |
32 | const char *probed_name; /* verbatim as seen on cmdline */ |
33 | char *options; /* options from config files */ |
34 | llist_t *realnames; /* strings. if this module is an alias, */ |
35 | /* real module name is one of these. */ |
36 | //Can there really be more than one? Example from real kernel? |
37 | llist_t *deps; /* strings. modules we depend on */ |
38 | }; |
39 | |
40 | /* NB: INSMOD_OPT_SILENT bit suppresses ONLY non-existent modules, |
41 | * not deleted ones (those are still listed in modules.dep). |
42 | * module-init-tools version 3.4: |
43 | * # modprobe bogus |
44 | * FATAL: Module bogus not found. [exitcode 1] |
45 | * # modprobe -q bogus [silent, exitcode still 1] |
46 | * but: |
47 | * # rm kernel/drivers/net/dummy.ko |
48 | * # modprobe -q dummy |
49 | * FATAL: Could not open '/lib/modules/xxx/kernel/drivers/net/dummy.ko': No such file or directory |
50 | * [exitcode 1] |
51 | */ |
52 | #define MODPROBE_OPTS "acdlnrt:VC:" IF_FEATURE_MODPROBE_BLACKLIST("b") |
53 | enum { |
54 | MODPROBE_OPT_INSERT_ALL = (INSMOD_OPT_UNUSED << 0), /* a */ |
55 | MODPROBE_OPT_DUMP_ONLY = (INSMOD_OPT_UNUSED << 1), /* c */ |
56 | MODPROBE_OPT_D = (INSMOD_OPT_UNUSED << 2), /* d */ |
57 | MODPROBE_OPT_LIST_ONLY = (INSMOD_OPT_UNUSED << 3), /* l */ |
58 | MODPROBE_OPT_SHOW_ONLY = (INSMOD_OPT_UNUSED << 4), /* n */ |
59 | MODPROBE_OPT_REMOVE = (INSMOD_OPT_UNUSED << 5), /* r */ |
60 | MODPROBE_OPT_RESTRICT = (INSMOD_OPT_UNUSED << 6), /* t */ |
61 | MODPROBE_OPT_VERONLY = (INSMOD_OPT_UNUSED << 7), /* V */ |
62 | MODPROBE_OPT_CONFIGFILE = (INSMOD_OPT_UNUSED << 8), /* C */ |
63 | MODPROBE_OPT_BLACKLIST = (INSMOD_OPT_UNUSED << 9) * ENABLE_FEATURE_MODPROBE_BLACKLIST, |
64 | }; |
65 | |
66 | struct globals { |
67 | llist_t *db; /* MEs of all modules ever seen (caching for speed) */ |
68 | llist_t *probes; /* MEs of module(s) requested on cmdline */ |
69 | char *cmdline_mopts; /* module options from cmdline */ |
70 | int num_unresolved_deps; |
71 | /* bool. "Did we have 'symbol:FOO' requested on cmdline?" */ |
72 | smallint need_symbols; |
73 | }; |
74 | #define G (*(struct globals*)&bb_common_bufsiz1) |
75 | #define INIT_G() do { } while (0) |
76 | |
77 | |
78 | static int read_config(const char *path); |
79 | |
80 | static char *gather_options_str(char *opts, const char *append) |
81 | { |
82 | /* Speed-optimized. We call gather_options_str many times. */ |
83 | if (append) { |
84 | if (opts == NULL) { |
85 | opts = xstrdup(append); |
86 | } else { |
87 | int optlen = strlen(opts); |
88 | opts = xrealloc(opts, optlen + strlen(append) + 2); |
89 | sprintf(opts + optlen, " %s", append); |
90 | } |
91 | } |
92 | return opts; |
93 | } |
94 | |
95 | static struct module_entry *helper_get_module(const char *module, int create) |
96 | { |
97 | char modname[MODULE_NAME_LEN]; |
98 | struct module_entry *e; |
99 | llist_t *l; |
100 | |
101 | filename2modname(module, modname); |
102 | for (l = G.db; l != NULL; l = l->link) { |
103 | e = (struct module_entry *) l->data; |
104 | if (strcmp(e->modname, modname) == 0) |
105 | return e; |
106 | } |
107 | if (!create) |
108 | return NULL; |
109 | |
110 | e = xzalloc(sizeof(*e)); |
111 | e->modname = xstrdup(modname); |
112 | llist_add_to(&G.db, e); |
113 | |
114 | return e; |
115 | } |
116 | static struct module_entry *get_or_add_modentry(const char *module) |
117 | { |
118 | return helper_get_module(module, 1); |
119 | } |
120 | static struct module_entry *get_modentry(const char *module) |
121 | { |
122 | return helper_get_module(module, 0); |
123 | } |
124 | |
125 | static void add_probe(const char *name) |
126 | { |
127 | struct module_entry *m; |
128 | |
129 | m = get_or_add_modentry(name); |
130 | if (!(option_mask32 & MODPROBE_OPT_REMOVE) |
131 | && (m->flags & MODULE_FLAG_LOADED) |
132 | ) { |
133 | DBG("skipping %s, it is already loaded", name); |
134 | return; |
135 | } |
136 | |
137 | DBG("queuing %s", name); |
138 | m->probed_name = name; |
139 | m->flags |= MODULE_FLAG_NEED_DEPS; |
140 | llist_add_to_end(&G.probes, m); |
141 | G.num_unresolved_deps++; |
142 | if (ENABLE_FEATURE_MODUTILS_SYMBOLS |
143 | && strncmp(m->modname, "symbol:", 7) == 0 |
144 | ) { |
145 | G.need_symbols = 1; |
146 | } |
147 | } |
148 | |
149 | static int FAST_FUNC config_file_action(const char *filename, |
150 | struct stat *statbuf UNUSED_PARAM, |
151 | void *userdata UNUSED_PARAM, |
152 | int depth UNUSED_PARAM) |
153 | { |
154 | char *tokens[3]; |
155 | parser_t *p; |
156 | struct module_entry *m; |
157 | int rc = TRUE; |
158 | |
159 | if (bb_basename(filename)[0] == '.') |
160 | goto error; |
161 | |
162 | p = config_open2(filename, fopen_for_read); |
163 | if (p == NULL) { |
164 | rc = FALSE; |
165 | goto error; |
166 | } |
167 | |
168 | while (config_read(p, tokens, 3, 2, "# \t", PARSE_NORMAL)) { |
169 | //Use index_in_strings? |
170 | if (strcmp(tokens[0], "alias") == 0) { |
171 | /* alias <wildcard> <modulename> */ |
172 | llist_t *l; |
173 | char wildcard[MODULE_NAME_LEN]; |
174 | char *rmod; |
175 | |
176 | if (tokens[2] == NULL) |
177 | continue; |
178 | filename2modname(tokens[1], wildcard); |
179 | |
180 | for (l = G.probes; l != NULL; l = l->link) { |
181 | m = (struct module_entry *) l->data; |
182 | if (fnmatch(wildcard, m->modname, 0) != 0) |
183 | continue; |
184 | rmod = filename2modname(tokens[2], NULL); |
185 | llist_add_to(&m->realnames, rmod); |
186 | |
187 | if (m->flags & MODULE_FLAG_NEED_DEPS) { |
188 | m->flags &= ~MODULE_FLAG_NEED_DEPS; |
189 | G.num_unresolved_deps--; |
190 | } |
191 | |
192 | m = get_or_add_modentry(rmod); |
193 | if (!(m->flags & MODULE_FLAG_NEED_DEPS)) { |
194 | m->flags |= MODULE_FLAG_NEED_DEPS; |
195 | G.num_unresolved_deps++; |
196 | } |
197 | } |
198 | } else if (strcmp(tokens[0], "options") == 0) { |
199 | /* options <modulename> <option...> */ |
200 | if (tokens[2] == NULL) |
201 | continue; |
202 | m = get_or_add_modentry(tokens[1]); |
203 | m->options = gather_options_str(m->options, tokens[2]); |
204 | } else if (strcmp(tokens[0], "include") == 0) { |
205 | /* include <filename> */ |
206 | read_config(tokens[1]); |
207 | } else if (ENABLE_FEATURE_MODPROBE_BLACKLIST |
208 | && strcmp(tokens[0], "blacklist") == 0 |
209 | ) { |
210 | /* blacklist <modulename> */ |
211 | get_or_add_modentry(tokens[1])->flags |= MODULE_FLAG_BLACKLISTED; |
212 | } |
213 | } |
214 | config_close(p); |
215 | error: |
216 | return rc; |
217 | } |
218 | |
219 | static int read_config(const char *path) |
220 | { |
221 | return recursive_action(path, ACTION_RECURSE | ACTION_QUIET, |
222 | config_file_action, NULL, NULL, 1); |
223 | } |
224 | |
225 | static const char *humanly_readable_name(struct module_entry *m) |
226 | { |
227 | /* probed_name may be NULL. modname always exists. */ |
228 | return m->probed_name ? m->probed_name : m->modname; |
229 | } |
230 | |
231 | /* Return: similar to bb_init_module: |
232 | * 0 on success, |
233 | * -errno on open/read error, |
234 | * errno on init_module() error |
235 | */ |
236 | static int do_modprobe(struct module_entry *m) |
237 | { |
238 | struct module_entry *m2 = m2; /* for compiler */ |
239 | char *fn, *options; |
240 | int rc, first; |
241 | llist_t *l; |
242 | |
243 | if (!(m->flags & MODULE_FLAG_FOUND_IN_MODDEP)) { |
244 | if (!(option_mask32 & INSMOD_OPT_SILENT)) |
245 | bb_error_msg("module %s not found in modules.dep", |
246 | humanly_readable_name(m)); |
247 | return -ENOENT; |
248 | } |
249 | DBG("do_modprob'ing %s", m->modname); |
250 | |
251 | if (!(option_mask32 & MODPROBE_OPT_REMOVE)) |
252 | m->deps = llist_rev(m->deps); |
253 | |
254 | for (l = m->deps; l != NULL; l = l->link) |
255 | DBG("dep: %s", l->data); |
256 | |
257 | first = 1; |
258 | rc = 0; |
259 | while (m->deps) { |
260 | rc = 0; |
261 | fn = llist_pop(&m->deps); /* we leak it */ |
262 | m2 = get_or_add_modentry(fn); |
263 | |
264 | if (option_mask32 & MODPROBE_OPT_REMOVE) { |
265 | /* modprobe -r */ |
266 | if (m2->flags & MODULE_FLAG_LOADED) { |
267 | rc = bb_delete_module(m2->modname, O_EXCL); |
268 | if (rc) { |
269 | if (first) { |
270 | bb_error_msg("failed to unload module %s: %s", |
271 | humanly_readable_name(m2), |
272 | moderror(rc)); |
273 | break; |
274 | } |
275 | } else { |
276 | m2->flags &= ~MODULE_FLAG_LOADED; |
277 | } |
278 | } |
279 | /* do not error out if *deps* fail to unload */ |
280 | first = 0; |
281 | continue; |
282 | } |
283 | |
284 | if (m2->flags & MODULE_FLAG_LOADED) { |
285 | DBG("%s is already loaded, skipping", fn); |
286 | continue; |
287 | } |
288 | |
289 | options = m2->options; |
290 | m2->options = NULL; |
291 | if (m == m2) |
292 | options = gather_options_str(options, G.cmdline_mopts); |
293 | rc = bb_init_module(fn, options); |
294 | DBG("loaded %s '%s', rc:%d", fn, options, rc); |
295 | if (rc == EEXIST) |
296 | rc = 0; |
297 | free(options); |
298 | if (rc) { |
299 | bb_error_msg("failed to load module %s (%s): %s", |
300 | humanly_readable_name(m2), |
301 | fn, |
302 | moderror(rc) |
303 | ); |
304 | break; |
305 | } |
306 | m2->flags |= MODULE_FLAG_LOADED; |
307 | } |
308 | |
309 | return rc; |
310 | } |
311 | |
312 | static void load_modules_dep(void) |
313 | { |
314 | struct module_entry *m; |
315 | char *colon, *tokens[2]; |
316 | parser_t *p; |
317 | |
318 | /* Modprobe does not work at all without modules.dep, |
319 | * even if the full module name is given. Returning error here |
320 | * was making us later confuse user with this message: |
321 | * "module /full/path/to/existing/file/module.ko not found". |
322 | * It's better to die immediately, with good message. |
323 | * xfopen_for_read provides that. */ |
324 | p = config_open2(CONFIG_DEFAULT_DEPMOD_FILE, xfopen_for_read); |
325 | |
326 | while (G.num_unresolved_deps |
327 | && config_read(p, tokens, 2, 1, "# \t", PARSE_NORMAL) |
328 | ) { |
329 | colon = last_char_is(tokens[0], ':'); |
330 | if (colon == NULL) |
331 | continue; |
332 | *colon = 0; |
333 | |
334 | m = get_modentry(tokens[0]); |
335 | if (m == NULL) |
336 | continue; |
337 | |
338 | /* Optimization... */ |
339 | if ((m->flags & MODULE_FLAG_LOADED) |
340 | && !(option_mask32 & MODPROBE_OPT_REMOVE) |
341 | ) { |
342 | DBG("skip deps of %s, it's already loaded", tokens[0]); |
343 | continue; |
344 | } |
345 | |
346 | m->flags |= MODULE_FLAG_FOUND_IN_MODDEP; |
347 | if ((m->flags & MODULE_FLAG_NEED_DEPS) && (m->deps == NULL)) { |
348 | G.num_unresolved_deps--; |
349 | llist_add_to(&m->deps, xstrdup(tokens[0])); |
350 | if (tokens[1]) |
351 | string_to_llist(tokens[1], &m->deps, " \t"); |
352 | } else |
353 | DBG("skipping dep line"); |
354 | } |
355 | config_close(p); |
356 | } |
357 | |
358 | int modprobe_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
359 | int modprobe_main(int argc UNUSED_PARAM, char **argv) |
360 | { |
361 | struct utsname uts; |
362 | int rc; |
363 | unsigned opt; |
364 | struct module_entry *me; |
365 | |
366 | opt_complementary = "q-v:v-q"; |
367 | opt = getopt32(argv, INSMOD_OPTS MODPROBE_OPTS INSMOD_ARGS, NULL, NULL); |
368 | argv += optind; |
369 | |
370 | if (opt & (MODPROBE_OPT_DUMP_ONLY | MODPROBE_OPT_LIST_ONLY | |
371 | MODPROBE_OPT_SHOW_ONLY)) |
372 | bb_error_msg_and_die("not supported"); |
373 | |
374 | if (!argv[0]) { |
375 | if (opt & MODPROBE_OPT_REMOVE) { |
376 | /* "modprobe -r" (w/o params). |
377 | * "If name is NULL, all unused modules marked |
378 | * autoclean will be removed". |
379 | */ |
380 | if (bb_delete_module(NULL, O_NONBLOCK | O_EXCL) != 0) |
381 | bb_perror_msg_and_die("rmmod"); |
382 | } |
383 | return EXIT_SUCCESS; |
384 | } |
385 | |
386 | /* Goto modules location */ |
387 | xchdir(CONFIG_DEFAULT_MODULES_DIR); |
388 | uname(&uts); |
389 | xchdir(uts.release); |
390 | |
391 | /* Retrieve module names of already loaded modules */ |
392 | { |
393 | char *s; |
394 | parser_t *parser = config_open2("/proc/modules", fopen_for_read); |
395 | while (config_read(parser, &s, 1, 1, "# \t", PARSE_NORMAL & ~PARSE_GREEDY)) |
396 | get_or_add_modentry(s)->flags |= MODULE_FLAG_LOADED; |
397 | config_close(parser); |
398 | } |
399 | |
400 | if (opt & (MODPROBE_OPT_INSERT_ALL | MODPROBE_OPT_REMOVE)) { |
401 | /* Each argument is a module name */ |
402 | do { |
403 | DBG("adding module %s", *argv); |
404 | add_probe(*argv++); |
405 | } while (*argv); |
406 | } else { |
407 | /* First argument is module name, rest are parameters */ |
408 | DBG("probing just module %s", *argv); |
409 | add_probe(argv[0]); |
410 | G.cmdline_mopts = parse_cmdline_module_options(argv); |
411 | } |
412 | |
413 | /* Happens if all requested modules are already loaded */ |
414 | if (G.probes == NULL) |
415 | return EXIT_SUCCESS; |
416 | |
417 | read_config("/etc/modprobe.conf"); |
418 | read_config("/etc/modprobe.d"); |
419 | if (ENABLE_FEATURE_MODUTILS_SYMBOLS && G.need_symbols) |
420 | read_config("modules.symbols"); |
421 | load_modules_dep(); |
422 | if (ENABLE_FEATURE_MODUTILS_ALIAS && G.num_unresolved_deps) { |
423 | read_config("modules.alias"); |
424 | load_modules_dep(); |
425 | } |
426 | |
427 | rc = 0; |
428 | while ((me = llist_pop(&G.probes)) != NULL) { |
429 | if (me->realnames == NULL) { |
430 | DBG("probing by module name"); |
431 | /* This is not an alias. Literal names are blacklisted |
432 | * only if '-b' is given. |
433 | */ |
434 | if (!(opt & MODPROBE_OPT_BLACKLIST) |
435 | || !(me->flags & MODULE_FLAG_BLACKLISTED) |
436 | ) { |
437 | rc |= do_modprobe(me); |
438 | } |
439 | continue; |
440 | } |
441 | |
442 | /* Probe all real names for the alias */ |
443 | do { |
444 | char *realname = llist_pop(&me->realnames); |
445 | struct module_entry *m2; |
446 | |
447 | DBG("probing alias %s by realname %s", me->modname, realname); |
448 | m2 = get_or_add_modentry(realname); |
449 | if (!(m2->flags & MODULE_FLAG_BLACKLISTED) |
450 | && (!(m2->flags & MODULE_FLAG_LOADED) |
451 | || (opt & MODPROBE_OPT_REMOVE)) |
452 | ) { |
453 | //TODO: we can pass "me" as 2nd param to do_modprobe, |
454 | //and make do_modprobe emit more meaningful error messages |
455 | //with alias name included, not just module name alias resolves to. |
456 | rc |= do_modprobe(m2); |
457 | } |
458 | free(realname); |
459 | } while (me->realnames != NULL); |
460 | } |
461 | |
462 | return (rc != 0); |
463 | } |