X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=src%2Fmodules-load%2Fmodules-load.c;h=6aeaf459aff0d65e837b4072ce6cfc006c959336;hp=0f2144c2ed84b853e6df14dfc037f3c74796afe2;hb=ab060556a9e1ebb2744719a29985e40919101a22;hpb=dce818b390a857a11f7dd634684500675cf79833 diff --git a/src/modules-load/modules-load.c b/src/modules-load/modules-load.c index 0f2144c2e..6aeaf459a 100644 --- a/src/modules-load/modules-load.c +++ b/src/modules-load/modules-load.c @@ -31,21 +31,148 @@ #include "log.h" #include "util.h" #include "strv.h" +#include "conf-files.h" +#include "virt.h" + +static char **arg_proc_cmdline_modules = NULL; #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wformat-nonliteral" static void systemd_kmod_log(void *data, int priority, const char *file, int line, const char *fn, const char *format, va_list args) { - log_meta(priority, file, line, fn, format, args); + log_metav(priority, file, line, fn, format, args); } #pragma GCC diagnostic pop +static int add_modules(const char *p) { + char **t, **k; + + k = strv_split(p, ","); + if (!k) + return log_oom(); + + t = strv_merge(arg_proc_cmdline_modules, k); + strv_free(k); + if (!t) + return log_oom(); + + strv_free(arg_proc_cmdline_modules); + arg_proc_cmdline_modules = t; + + return 0; +} + +static int parse_proc_cmdline(void) { + char *line, *w, *state; + int r; + size_t l; + + if (detect_container(NULL) > 0) + return 0; + + r = read_one_line_file("/proc/cmdline", &line); + if (r < 0) { + log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r)); + return 0; + } + + FOREACH_WORD_QUOTED(w, l, line, state) { + char *word; + + word = strndup(w, l); + if (!word) { + r = -ENOMEM; + goto finish; + } + + if (startswith(word, "modules-load=")) { + + r = add_modules(word + 13); + if (r < 0) + goto finish; + + } else if (startswith(word, "rd.modules-load=")) { + + if (in_initrd()) { + r = add_modules(word + 16); + if (r < 0) + goto finish; + } + + } + + free(word); + } + + r = 0; + +finish: + free(line); + return r; +} + +static int load_module(struct kmod_ctx *ctx, const char *m) { + const int probe_flags = KMOD_PROBE_APPLY_BLACKLIST; + struct kmod_list *itr, *modlist = NULL; + int r = 0; + + log_debug("load: %s\n", m); + + r = kmod_module_new_from_lookup(ctx, m, &modlist); + if (r < 0) { + log_error("Failed to lookup alias '%s': %s", m, strerror(-r)); + return r; + } + + if (!modlist) { + log_error("Failed to find module '%s'", m); + return -ENOENT; + } + + kmod_list_foreach(itr, modlist) { + struct kmod_module *mod; + int state, err; + + mod = kmod_module_get_module(itr); + state = kmod_module_get_initstate(mod); + + switch (state) { + case KMOD_MODULE_BUILTIN: + log_info("Module '%s' is builtin", kmod_module_get_name(mod)); + break; + + case KMOD_MODULE_LIVE: + log_info("Module '%s' is already loaded", kmod_module_get_name(mod)); + break; + + default: + err = kmod_module_probe_insert_module(mod, probe_flags, + NULL, NULL, NULL, NULL); + + if (err == 0) + log_info("Inserted module '%s'", kmod_module_get_name(mod)); + else if (err == KMOD_PROBE_APPLY_BLACKLIST) + log_info("Module '%s' is blacklisted", kmod_module_get_name(mod)); + else { + log_error("Failed to insert '%s': %s", kmod_module_get_name(mod), + strerror(-err)); + r = err; + } + } + + kmod_module_unref(mod); + } + + kmod_module_unref_list(modlist); + + return r; +} + int main(int argc, char *argv[]) { - int r = EXIT_FAILURE; - char **files, **fn; + int r = EXIT_FAILURE, k; + char **files, **fn, **i; struct kmod_ctx *ctx; - const int probe_flags = KMOD_PROBE_APPLY_BLACKLIST|KMOD_PROBE_IGNORE_LOADED; if (argc > 1) { log_error("This program takes no argument."); @@ -58,6 +185,9 @@ int main(int argc, char *argv[]) { umask(0022); + if (parse_proc_cmdline() < 0) + return EXIT_FAILURE; + ctx = kmod_new(NULL, NULL); if (!ctx) { log_error("Failed to allocate memory for kmod."); @@ -65,10 +195,17 @@ int main(int argc, char *argv[]) { } kmod_load_resources(ctx); - kmod_set_log_fn(ctx, systemd_kmod_log, NULL); - if (conf_files_list(&files, ".conf", + r = EXIT_SUCCESS; + + STRV_FOREACH(i, arg_proc_cmdline_modules) { + k = load_module(ctx, *i); + if (k < 0) + r = EXIT_FAILURE; + } + + k = conf_files_list(&files, ".conf", "/etc/modules-load.d", "/run/modules-load.d", "/usr/local/lib/modules-load.d", @@ -76,13 +213,13 @@ int main(int argc, char *argv[]) { #ifdef HAVE_SPLIT_USR "/lib/modules-load.d", #endif - NULL) < 0) { - log_error("Failed to enumerate modules-load.d files: %s", strerror(-r)); + NULL); + if (k < 0) { + log_error("Failed to enumerate modules-load.d files: %s", strerror(-k)); + r = EXIT_FAILURE; goto finish; } - r = EXIT_SUCCESS; - STRV_FOREACH(fn, files) { FILE *f; @@ -99,8 +236,6 @@ int main(int argc, char *argv[]) { log_debug("apply: %s\n", *fn); for (;;) { char line[LINE_MAX], *l; - struct kmod_list *itr, *modlist = NULL; - int err; if (!fgets(line, sizeof(line), f)) break; @@ -109,34 +244,9 @@ int main(int argc, char *argv[]) { if (*l == '#' || *l == 0) continue; - err = kmod_module_new_from_lookup(ctx, l, &modlist); - if (err < 0) { - log_error("Failed to lookup alias '%s'", l); + k = load_module(ctx, l); + if (k < 0) r = EXIT_FAILURE; - continue; - } - - kmod_list_foreach(itr, modlist) { - struct kmod_module *mod; - - mod = kmod_module_get_module(itr); - err = kmod_module_probe_insert_module(mod, probe_flags, - NULL, NULL, NULL, NULL); - - if (err == 0) - log_info("Inserted module '%s'", kmod_module_get_name(mod)); - else if (err == KMOD_PROBE_APPLY_BLACKLIST) - log_info("Module '%s' is blacklisted", kmod_module_get_name(mod)); - else { - log_error("Failed to insert '%s': %s", kmod_module_get_name(mod), - strerror(-err)); - r = EXIT_FAILURE; - } - - kmod_module_unref(mod); - } - - kmod_module_unref_list(modlist); } if (ferror(f)) { @@ -150,6 +260,7 @@ int main(int argc, char *argv[]) { finish: strv_free(files); kmod_unref(ctx); + strv_free(arg_proc_cmdline_modules); return r; }