/
niceSOFT
/
kmod
Обзор
Документация
Войти
/
niceSOFT
/
kmod
Код
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
v21
tools/modprobe.c
920 строк
21 KB
Lucas De Marchi
tools: display features in --version
09 июн 2015, 04:38
09 июн 2015, 04:38
655de27
Код
Авторство
О чём код?
/* * kmod-modprobe - manage linux kernel modules using libkmod. * * Copyright (C) 2011-2013 ProFUSION embedded systems * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ #include <assert.h> #include <errno.h> #include <getopt.h> #include <limits.h> #include <stdbool.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <sys/stat.h> #include <sys/types.h> #include <sys/utsname.h> #include <sys/wait.h> #include <shared/array.h> #include <shared/macro.h> #include <libkmod/libkmod.h> #include "kmod.h" static int log_priority = LOG_CRIT; static int use_syslog = 0; #define LOG(...) log_printf(log_priority, __VA_ARGS__) #define DEFAULT_VERBOSE LOG_WARNING static int verbose = DEFAULT_VERBOSE; static int do_show = 0; static int dry_run = 0; static int ignore_loaded = 0; static int lookup_only = 0; static int first_time = 0; static int ignore_commands = 0; static int use_blacklist = 0; static int force = 0; static int strip_modversion = 0; static int strip_vermagic = 0; static int remove_dependencies = 0; static int quiet_inuse = 0; static const char cmdopts_s[] = "arRibfDcnC:d:S:sqvVh"; static const struct option cmdopts[] = { {"all", no_argument, 0, 'a'}, {"remove", no_argument, 0, 'r'}, {"remove-dependencies", no_argument, 0, 5}, {"resolve-alias", no_argument, 0, 'R'}, {"first-time", no_argument, 0, 3}, {"ignore-install", no_argument, 0, 'i'}, {"ignore-remove", no_argument, 0, 'i'}, {"use-blacklist", no_argument, 0, 'b'}, {"force", no_argument, 0, 'f'}, {"force-modversion", no_argument, 0, 2}, {"force-vermagic", no_argument, 0, 1}, {"show-depends", no_argument, 0, 'D'}, {"showconfig", no_argument, 0, 'c'}, {"show-config", no_argument, 0, 'c'}, {"show-modversions", no_argument, 0, 4}, {"dump-modversions", no_argument, 0, 4}, {"dry-run", no_argument, 0, 'n'}, {"show", no_argument, 0, 'n'}, {"config", required_argument, 0, 'C'}, {"dirname", required_argument, 0, 'd'}, {"set-version", required_argument, 0, 'S'}, {"syslog", no_argument, 0, 's'}, {"quiet", no_argument, 0, 'q'}, {"verbose", no_argument, 0, 'v'}, {"version", no_argument, 0, 'V'}, {"help", no_argument, 0, 'h'}, {NULL, 0, 0, 0} }; static void help(void) { printf("Usage:\n" "\t%s [options] [-i] [-b] modulename\n" "\t%s [options] -a [-i] [-b] modulename [modulename...]\n" "\t%s [options] -r [-i] modulename\n" "\t%s [options] -r -a [-i] modulename [modulename...]\n" "\t%s [options] -c\n" "\t%s [options] --dump-modversions filename\n" "Management Options:\n" "\t-a, --all Consider every non-argument to\n" "\t be a module name to be inserted\n" "\t or removed (-r)\n" "\t-r, --remove Remove modules instead of inserting\n" "\t --remove-dependencies Also remove modules depending on it\n" "\t-R, --resolve-alias Only lookup and print alias and exit\n" "\t --first-time Fail if module already inserted or removed\n" "\t-i, --ignore-install Ignore install commands\n" "\t-i, --ignore-remove Ignore remove commands\n" "\t-b, --use-blacklist Apply blacklist to resolved alias.\n" "\t-f, --force Force module insertion or removal.\n" "\t implies --force-modversions and\n" "\t --force-vermagic\n" "\t --force-modversion Ignore module's version\n" "\t --force-vermagic Ignore module's version magic\n" "\n" "Query Options:\n" "\t-D, --show-depends Only print module dependencies and exit\n" "\t-c, --showconfig Print out known configuration and exit\n" "\t-c, --show-config Same as --showconfig\n" "\t --show-modversions Dump module symbol version and exit\n" "\t --dump-modversions Same as --show-modversions\n" "\n" "General Options:\n" "\t-n, --dry-run Do not execute operations, just print out\n" "\t-n, --show Same as --dry-run\n" "\t-C, --config=FILE Use FILE instead of default search paths\n" "\t-d, --dirname=DIR Use DIR as filesystem root for /lib/modules\n" "\t-S, --set-version=VERSION Use VERSION instead of `uname -r`\n" "\t-s, --syslog print to syslog, not stderr\n" "\t-q, --quiet disable messages\n" "\t-v, --verbose enables more messages\n" "\t-V, --version show version\n" "\t-h, --help show this help\n", program_invocation_short_name, program_invocation_short_name, program_invocation_short_name, program_invocation_short_name, program_invocation_short_name, program_invocation_short_name); } _printf_format_(1, 2) static inline void _show(const char *fmt, ...) { va_list args; if (!do_show && verbose <= DEFAULT_VERBOSE) return; va_start(args, fmt); vfprintf(stdout, fmt, args); fflush(stdout); va_end(args); } #define SHOW(...) _show(__VA_ARGS__) static int show_config(struct kmod_ctx *ctx) { struct config_iterators { const char *name; struct kmod_config_iter *(*get_iter)(const struct kmod_ctx *ctx); } ci[] = { { "blacklist", kmod_config_get_blacklists }, { "install", kmod_config_get_install_commands }, { "remove", kmod_config_get_remove_commands }, { "alias", kmod_config_get_aliases }, { "options", kmod_config_get_options }, { "softdep", kmod_config_get_softdeps }, }; size_t i; for (i = 0; i < ARRAY_SIZE(ci); i++) { struct kmod_config_iter *iter = ci[i].get_iter(ctx); if (iter == NULL) continue; while (kmod_config_iter_next(iter)) { const char *val; printf("%s %s", ci[i].name, kmod_config_iter_get_key(iter)); val = kmod_config_iter_get_value(iter); if (val != NULL) { putchar(' '); puts(val); } else putchar('\n'); } kmod_config_iter_free_iter(iter); } puts("\n# End of configuration files. Dumping indexes now:\n"); fflush(stdout); kmod_dump_index(ctx, KMOD_INDEX_MODULES_ALIAS, STDOUT_FILENO); kmod_dump_index(ctx, KMOD_INDEX_MODULES_SYMBOL, STDOUT_FILENO); return 0; } static int show_modversions(struct kmod_ctx *ctx, const char *filename) { struct kmod_list *l, *list = NULL; struct kmod_module *mod; int err = kmod_module_new_from_path(ctx, filename, &mod); if (err < 0) { LOG("Module %s not found.\n", filename); return err; } err = kmod_module_get_versions(mod, &list); if (err < 0) { LOG("could not get modversions of %s: %s\n", filename, strerror(-err)); kmod_module_unref(mod); return err; } kmod_list_foreach(l, list) { const char *symbol = kmod_module_version_get_symbol(l); uint64_t crc = kmod_module_version_get_crc(l); printf("0x%08"PRIx64"\t%s\n", crc, symbol); } kmod_module_versions_free_list(list); kmod_module_unref(mod); return 0; } static int command_do(struct kmod_module *module, const char *type, const char *command, const char *cmdline_opts) { const char *modname = kmod_module_get_name(module); char *p, *cmd = NULL; size_t cmdlen, cmdline_opts_len, varlen; int ret = 0; if (cmdline_opts == NULL) cmdline_opts = ""; cmdline_opts_len = strlen(cmdline_opts); cmd = strdup(command); if (cmd == NULL) return -ENOMEM; cmdlen = strlen(cmd); varlen = sizeof("$CMDLINE_OPTS") - 1; while ((p = strstr(cmd, "$CMDLINE_OPTS")) != NULL) { size_t prefixlen = p - cmd; size_t suffixlen = cmdlen - prefixlen - varlen; size_t slen = cmdlen - varlen + cmdline_opts_len; char *suffix = p + varlen; char *s = malloc(slen + 1); if (s == NULL) { free(cmd); return -ENOMEM; } memcpy(s, cmd, p - cmd); memcpy(s + prefixlen, cmdline_opts, cmdline_opts_len); memcpy(s + prefixlen + cmdline_opts_len, suffix, suffixlen); s[slen] = '\0'; free(cmd); cmd = s; cmdlen = slen; } SHOW("%s %s\n", type, cmd); if (dry_run) goto end; setenv("MODPROBE_MODULE", modname, 1); ret = system(cmd); unsetenv("MODPROBE_MODULE"); if (ret == -1 || WEXITSTATUS(ret)) { LOG("Error running %s command for %s\n", type, modname); if (ret != -1) ret = -WEXITSTATUS(ret); } end: free(cmd); return ret; } static int rmmod_do_remove_module(struct kmod_module *mod) { const char *modname = kmod_module_get_name(mod); struct kmod_list *deps, *itr; int flags = 0, err; SHOW("rmmod %s\n", kmod_module_get_name(mod)); if (dry_run) return 0; if (force) flags |= KMOD_REMOVE_FORCE; err = kmod_module_remove_module(mod, flags); if (err == -EEXIST) { if (!first_time) err = 0; else LOG("Module %s is not in kernel.\n", modname); } deps = kmod_module_get_dependencies(mod); if (deps != NULL) { kmod_list_foreach(itr, deps) { struct kmod_module *dep = kmod_module_get_module(itr); if (kmod_module_get_refcnt(dep) == 0) rmmod_do_remove_module(dep); kmod_module_unref(dep); } kmod_module_unref_list(deps); } return err; } static int rmmod_do_module(struct kmod_module *mod, bool do_dependencies); static int rmmod_do_deps_list(struct kmod_list *list, bool stop_on_errors) { struct kmod_list *l; kmod_list_foreach_reverse(l, list) { struct kmod_module *m = kmod_module_get_module(l); int r = rmmod_do_module(m, false); kmod_module_unref(m); if (r < 0 && stop_on_errors) return r; } return 0; } static int rmmod_do_module(struct kmod_module *mod, bool do_dependencies) { const char *modname = kmod_module_get_name(mod); struct kmod_list *pre = NULL, *post = NULL; const char *cmd = NULL; int err; if (!ignore_commands) { err = kmod_module_get_softdeps(mod, &pre, &post); if (err < 0) { WRN("could not get softdeps of '%s': %s\n", modname, strerror(-err)); return err; } cmd = kmod_module_get_remove_commands(mod); } if (cmd == NULL && !ignore_loaded) { int state = kmod_module_get_initstate(mod); if (state < 0) { if (first_time) { LOG("Module %s is not in kernel.\n", modname); err = -ENOENT; } else { err = 0; } goto error; } else if (state == KMOD_MODULE_BUILTIN) { LOG("Module %s is builtin.\n", modname); err = -ENOENT; goto error; } } rmmod_do_deps_list(post, false); if (do_dependencies && remove_dependencies) { struct kmod_list *deps = kmod_module_get_dependencies(mod); err = rmmod_do_deps_list(deps, true); if (err < 0) goto error; } if (!ignore_loaded && !cmd) { int usage = kmod_module_get_refcnt(mod); if (usage > 0) { if (!quiet_inuse) LOG("Module %s is in use.\n", modname); err = -EBUSY; goto error; } } if (cmd == NULL) err = rmmod_do_remove_module(mod); else err = command_do(mod, "remove", cmd, NULL); if (err < 0) goto error; rmmod_do_deps_list(pre, false); error: kmod_module_unref_list(pre); kmod_module_unref_list(post); return err; } static int rmmod(struct kmod_ctx *ctx, const char *alias) { struct kmod_list *l, *list = NULL; int err; err = kmod_module_new_from_lookup(ctx, alias, &list); if (err < 0) return err; if (list == NULL) { LOG("Module %s not found.\n", alias); err = -ENOENT; } kmod_list_foreach(l, list) { struct kmod_module *mod = kmod_module_get_module(l); err = rmmod_do_module(mod, true); kmod_module_unref(mod); if (err < 0) break; } kmod_module_unref_list(list); return err; } static int rmmod_all(struct kmod_ctx *ctx, char **args, int nargs) { int i, err = 0; for (i = 0; i < nargs; i++) { int r = rmmod(ctx, args[i]); if (r < 0) err = r; } return err; } static void print_action(struct kmod_module *m, bool install, const char *options) { const char *path; if (install) { printf("install %s %s\n", kmod_module_get_install_commands(m), options); return; } path = kmod_module_get_path(m); if (path == NULL) { /* * Either a builtin module, or an alias, print only for * builtin */ if (kmod_module_get_initstate(m) == KMOD_MODULE_BUILTIN) printf("builtin %s\n", kmod_module_get_name(m)); } else printf("insmod %s %s\n", kmod_module_get_path(m), options); } static int insmod(struct kmod_ctx *ctx, const char *alias, const char *extra_options) { struct kmod_list *l, *list = NULL; int err, flags = 0; void (*show)(struct kmod_module *m, bool install, const char *options) = NULL; err = kmod_module_new_from_lookup(ctx, alias, &list); if (err < 0) return err; if (list == NULL) { LOG("Module %s not found.\n", alias); return -ENOENT; } if (strip_modversion || force) flags |= KMOD_PROBE_FORCE_MODVERSION; if (strip_vermagic || force) flags |= KMOD_PROBE_FORCE_VERMAGIC; if (ignore_commands) flags |= KMOD_PROBE_IGNORE_COMMAND; if (ignore_loaded) flags |= KMOD_PROBE_IGNORE_LOADED; if (dry_run) flags |= KMOD_PROBE_DRY_RUN; if (do_show || verbose > DEFAULT_VERBOSE) show = &print_action; flags |= KMOD_PROBE_APPLY_BLACKLIST_ALIAS_ONLY; if (use_blacklist) flags |= KMOD_PROBE_APPLY_BLACKLIST; if (first_time) flags |= KMOD_PROBE_FAIL_ON_LOADED; kmod_list_foreach(l, list) { struct kmod_module *mod = kmod_module_get_module(l); if (lookup_only) printf("%s\n", kmod_module_get_name(mod)); else { err = kmod_module_probe_insert_module(mod, flags, extra_options, NULL, NULL, show); } if (err >= 0) /* ignore flag return values such as a mod being blacklisted */ err = 0; else { switch (err) { case -EEXIST: ERR("could not insert '%s': Module already in kernel\n", kmod_module_get_name(mod)); break; case -ENOENT: ERR("could not insert '%s': Unknown symbol in module, " "or unknown parameter (see dmesg)\n", kmod_module_get_name(mod)); break; default: ERR("could not insert '%s': %s\n", kmod_module_get_name(mod), strerror(-err)); break; } } kmod_module_unref(mod); } kmod_module_unref_list(list); return err; } static int insmod_all(struct kmod_ctx *ctx, char **args, int nargs) { int i, err = 0; for (i = 0; i < nargs; i++) { int r = insmod(ctx, args[i], NULL); if (r < 0) err = r; } return err; } static void env_modprobe_options_append(const char *value) { const char *old = getenv("MODPROBE_OPTIONS"); char *env; if (old == NULL) { setenv("MODPROBE_OPTIONS", value, 1); return; } if (asprintf(&env, "%s %s", old, value) < 0) { ERR("could not append value to $MODPROBE_OPTIONS\n"); return; } if (setenv("MODPROBE_OPTIONS", env, 1) < 0) ERR("could not setenv(MODPROBE_OPTIONS, \"%s\")\n", env); free(env); } static int options_from_array(char **args, int nargs, char **output) { char *opts = NULL; size_t optslen = 0; int i, err = 0; for (i = 1; i < nargs; i++) { size_t len = strlen(args[i]); size_t qlen = 0; const char *value; void *tmp; value = strchr(args[i], '='); if (value) { value++; if (*value != '"' && *value != '\'') { if (strchr(value, ' ')) qlen = 2; } } tmp = realloc(opts, optslen + len + qlen + 2); if (!tmp) { err = -errno; free(opts); opts = NULL; ERR("could not gather module options: out-of-memory\n"); break; } opts = tmp; if (optslen > 0) { opts[optslen] = ' '; optslen++; } if (qlen == 0) { memcpy(opts + optslen, args[i], len + 1); optslen += len; } else { size_t keylen = value - args[i]; size_t valuelen = len - keylen; memcpy(opts + optslen, args[i], keylen); optslen += keylen; opts[optslen] = '"'; optslen++; memcpy(opts + optslen, value, valuelen); optslen += valuelen; opts[optslen] = '"'; optslen++; opts[optslen] = '\0'; } } *output = opts; return err; } static char **prepend_options_from_env(int *p_argc, char **orig_argv) { const char *p, *env = getenv("MODPROBE_OPTIONS"); char **new_argv, *str_start, *str_end, *str, *s, *quote; int i, argc = *p_argc; size_t envlen, space_count = 0; if (env == NULL) return orig_argv; for (p = env; *p != '\0'; p++) { if (*p == ' ') space_count++; } envlen = p - env; new_argv = malloc(sizeof(char *) * (argc + space_count + 3 + envlen)); if (new_argv == NULL) return NULL; new_argv[0] = orig_argv[0]; str_start = str = (char *) (new_argv + argc + space_count + 3); memcpy(str, env, envlen + 1); str_end = str_start + envlen; quote = NULL; for (i = 1, s = str; *s != '\0'; s++) { if (quote == NULL) { if (*s == ' ') { new_argv[i] = str; i++; *s = '\0'; str = s + 1; } else if (*s == '"' || *s == '\'') quote = s; } else { if (*s == *quote) { if (quote == str) { new_argv[i] = str + 1; i++; *s = '\0'; str = s + 1; } else { char *it; for (it = quote; it < s - 1; it++) it[0] = it[1]; for (it = s - 1; it < str_end - 2; it++) it[0] = it[2]; str_end -= 2; *str_end = '\0'; s -= 2; } quote = NULL; } } } if (str < s) { new_argv[i] = str; i++; } memcpy(new_argv + i, orig_argv + 1, sizeof(char *) * (argc - 1)); new_argv[i + argc] = NULL; *p_argc = i + argc - 1; return new_argv; } static int do_modprobe(int argc, char **orig_argv) { struct kmod_ctx *ctx; char **args = NULL, **argv; const char **config_paths = NULL; int nargs = 0, n_config_paths = 0; char dirname_buf[PATH_MAX]; const char *dirname = NULL; const char *root = NULL; const char *kversion = NULL; int use_all = 0; int do_remove = 0; int do_show_config = 0; int do_show_modversions = 0; int err; argv = prepend_options_from_env(&argc, orig_argv); if (argv == NULL) { ERR("Could not prepend options from command line\n"); return EXIT_FAILURE; } for (;;) { int c, idx = 0; c = getopt_long(argc, argv, cmdopts_s, cmdopts, &idx); if (c == -1) break; switch (c) { case 'a': log_priority = LOG_WARNING; use_all = 1; break; case 'r': do_remove = 1; break; case 5: remove_dependencies = 1; break; case 'R': lookup_only = 1; break; case 3: first_time = 1; break; case 'i': ignore_commands = 1; break; case 'b': use_blacklist = 1; break; case 'f': force = 1; break; case 2: strip_modversion = 1; break; case 1: strip_vermagic = 1; break; case 'D': ignore_loaded = 1; dry_run = 1; do_show = 1; break; case 'c': do_show_config = 1; break; case 4: do_show_modversions = 1; break; case 'n': dry_run = 1; break; case 'C': { size_t bytes = sizeof(char *) * (n_config_paths + 2); void *tmp = realloc(config_paths, bytes); if (!tmp) { ERR("out-of-memory\n"); err = -1; goto done; } config_paths = tmp; config_paths[n_config_paths] = optarg; n_config_paths++; config_paths[n_config_paths] = NULL; env_modprobe_options_append("-C"); env_modprobe_options_append(optarg); break; } case 'd': root = optarg; break; case 'S': kversion = optarg; break; case 's': env_modprobe_options_append("-s"); use_syslog = 1; break; case 'q': env_modprobe_options_append("-q"); verbose = LOG_EMERG; break; case 'v': env_modprobe_options_append("-v"); verbose++; break; case 'V': puts(PACKAGE " version " VERSION); puts(KMOD_FEATURES); err = 0; goto done; case 'h': help(); err = 0; goto done; case '?': err = -1; goto done; default: ERR("unexpected getopt_long() value '%c'.\n", c); err = -1; goto done; } } args = argv + optind; nargs = argc - optind; log_open(use_syslog); if (!do_show_config) { if (nargs == 0) { ERR("missing parameters. See -h.\n"); err = -1; goto done; } } if (root != NULL || kversion != NULL) { struct utsname u; if (root == NULL) root = ""; if (kversion == NULL) { if (uname(&u) < 0) { ERR("uname() failed: %m\n"); err = -1; goto done; } kversion = u.release; } snprintf(dirname_buf, sizeof(dirname_buf), "%s/lib/modules/%s", root, kversion); dirname = dirname_buf; } ctx = kmod_new(dirname, config_paths); if (!ctx) { ERR("kmod_new() failed!\n"); err = -1; goto done; } log_setup_kmod_log(ctx, verbose); kmod_load_resources(ctx); if (do_show_config) err = show_config(ctx); else if (do_show_modversions) err = show_modversions(ctx, args[0]); else if (do_remove) err = rmmod_all(ctx, args, nargs); else if (use_all) err = insmod_all(ctx, args, nargs); else { char *opts; err = options_from_array(args, nargs, &opts); if (err == 0) { err = insmod(ctx, args[0], opts); free(opts); } } kmod_unref(ctx); done: log_close(); if (argv != orig_argv) free(argv); free(config_paths); return err >= 0 ? EXIT_SUCCESS : EXIT_FAILURE; } const struct kmod_cmd kmod_cmd_compat_modprobe = { .name = "modprobe", .cmd = do_modprobe, .help = "compat modprobe command", };