mirror of
https://github.com/sheumann/hush.git
synced 2024-09-12 15:54:56 +00:00
290 lines
7.7 KiB
C
290 lines
7.7 KiB
C
/* vi: set sw=4 ts=4: */
|
|
/*
|
|
* Modprobe written from scratch for BusyBox
|
|
*
|
|
* Copyright (c) 2008 Timo Teras <timo.teras@iki.fi>
|
|
* Copyright (c) 2008 Vladimir Dronnikov
|
|
*
|
|
* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
|
|
*/
|
|
|
|
#include "libbb.h"
|
|
#include "modutils.h"
|
|
#include <sys/utsname.h>
|
|
#include <fnmatch.h>
|
|
|
|
struct modprobe_option {
|
|
char *module;
|
|
char *option;
|
|
};
|
|
|
|
struct modprobe_conf {
|
|
char probename[MODULE_NAME_LEN];
|
|
llist_t *options;
|
|
llist_t *aliases;
|
|
#if ENABLE_FEATURE_MODPROBE_BLACKLIST
|
|
#define add_to_blacklist(conf, name) llist_add_to(&conf->blacklist, name)
|
|
#define check_blacklist(conf, name) (llist_find(conf->blacklist, name) == NULL)
|
|
llist_t *blacklist;
|
|
#else
|
|
#define add_to_blacklist(conf, name) do {} while (0)
|
|
#define check_blacklist(conf, name) (1)
|
|
#endif
|
|
};
|
|
|
|
#define MODPROBE_OPTS "acdlnrt:VC:" USE_FEATURE_MODPROBE_BLACKLIST("b")
|
|
enum {
|
|
MODPROBE_OPT_INSERT_ALL = (INSMOD_OPT_UNUSED << 0), /* a */
|
|
MODPROBE_OPT_DUMP_ONLY = (INSMOD_OPT_UNUSED << 1), /* c */
|
|
MODPROBE_OPT_D = (INSMOD_OPT_UNUSED << 2), /* d */
|
|
MODPROBE_OPT_LIST_ONLY = (INSMOD_OPT_UNUSED << 3), /* l */
|
|
MODPROBE_OPT_SHOW_ONLY = (INSMOD_OPT_UNUSED << 4), /* n */
|
|
MODPROBE_OPT_REMOVE = (INSMOD_OPT_UNUSED << 5), /* r */
|
|
MODPROBE_OPT_RESTRICT = (INSMOD_OPT_UNUSED << 6), /* t */
|
|
MODPROBE_OPT_VERONLY = (INSMOD_OPT_UNUSED << 7), /* V */
|
|
MODPROBE_OPT_CONFIGFILE = (INSMOD_OPT_UNUSED << 8), /* C */
|
|
MODPROBE_OPT_BLACKLIST = (INSMOD_OPT_UNUSED << 9) * ENABLE_FEATURE_MODPROBE_BLACKLIST,
|
|
};
|
|
|
|
static llist_t *loaded;
|
|
|
|
static int read_config(struct modprobe_conf *conf, const char *path);
|
|
|
|
static void add_option(llist_t **all_opts, const char *module, const char *opts)
|
|
{
|
|
struct modprobe_option *o;
|
|
|
|
o = xzalloc(sizeof(struct modprobe_option));
|
|
if (module)
|
|
o->module = filename2modname(module, NULL);
|
|
o->option = xstrdup(opts);
|
|
llist_add_to(all_opts, o);
|
|
}
|
|
|
|
static int FAST_FUNC config_file_action(const char *filename,
|
|
struct stat *statbuf UNUSED_PARAM,
|
|
void *userdata,
|
|
int depth UNUSED_PARAM)
|
|
{
|
|
struct modprobe_conf *conf = (struct modprobe_conf *) userdata;
|
|
RESERVE_CONFIG_BUFFER(modname, MODULE_NAME_LEN);
|
|
char *tokens[3];
|
|
parser_t *p;
|
|
int rc = TRUE;
|
|
|
|
if (bb_basename(filename)[0] == '.')
|
|
goto error;
|
|
|
|
p = config_open2(filename, fopen_for_read);
|
|
if (p == NULL) {
|
|
rc = FALSE;
|
|
goto error;
|
|
}
|
|
|
|
while (config_read(p, tokens, 3, 2, "# \t", PARSE_NORMAL)) {
|
|
if (strcmp(tokens[0], "alias") == 0) {
|
|
filename2modname(tokens[1], modname);
|
|
if (tokens[2] &&
|
|
fnmatch(modname, conf->probename, 0) == 0)
|
|
llist_add_to(&conf->aliases,
|
|
filename2modname(tokens[2], NULL));
|
|
} else if (strcmp(tokens[0], "options") == 0) {
|
|
if (tokens[2])
|
|
add_option(&conf->options, tokens[1], tokens[2]);
|
|
} else if (strcmp(tokens[0], "include") == 0) {
|
|
read_config(conf, tokens[1]);
|
|
} else if (ENABLE_FEATURE_MODPROBE_BLACKLIST &&
|
|
strcmp(tokens[0], "blacklist") == 0) {
|
|
add_to_blacklist(conf, xstrdup(tokens[1]));
|
|
}
|
|
}
|
|
config_close(p);
|
|
error:
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
RELEASE_CONFIG_BUFFER(modname);
|
|
return rc;
|
|
}
|
|
|
|
static int read_config(struct modprobe_conf *conf, const char *path)
|
|
{
|
|
return recursive_action(path, ACTION_RECURSE | ACTION_QUIET,
|
|
config_file_action, NULL, conf, 1);
|
|
}
|
|
|
|
static char *gather_options(llist_t *first, const char *module, int usecmdline)
|
|
{
|
|
struct modprobe_option *opt;
|
|
llist_t *n;
|
|
char *opts = xstrdup("");
|
|
int optlen = 0;
|
|
|
|
for (n = first; n != NULL; n = n->link) {
|
|
opt = (struct modprobe_option *) n->data;
|
|
|
|
if (opt->module == NULL && !usecmdline)
|
|
continue;
|
|
if (opt->module != NULL && strcmp(opt->module, module) != 0)
|
|
continue;
|
|
|
|
opts = xrealloc(opts, optlen + strlen(opt->option) + 2);
|
|
optlen += sprintf(opts + optlen, "%s ", opt->option);
|
|
}
|
|
return opts;
|
|
}
|
|
|
|
static int do_modprobe(struct modprobe_conf *conf, const char *module)
|
|
{
|
|
RESERVE_CONFIG_BUFFER(modname, MODULE_NAME_LEN);
|
|
llist_t *deps = NULL;
|
|
char *fn, *options, *colon = NULL, *tokens[2];
|
|
parser_t *p;
|
|
int rc = -1;
|
|
|
|
p = config_open2(CONFIG_DEFAULT_DEPMOD_FILE, fopen_for_read);
|
|
if (p == NULL)
|
|
goto error;
|
|
|
|
while (config_read(p, tokens, 2, 1, "# \t", PARSE_NORMAL)) {
|
|
colon = last_char_is(tokens[0], ':');
|
|
if (colon == NULL)
|
|
continue;
|
|
|
|
filename2modname(tokens[0], modname);
|
|
if (strcmp(modname, module) == 0)
|
|
break;
|
|
|
|
colon = NULL;
|
|
}
|
|
if (colon == NULL)
|
|
goto error_not_found;
|
|
|
|
colon[0] = '\0';
|
|
llist_add_to(&deps, xstrdup(tokens[0]));
|
|
if (tokens[1])
|
|
string_to_llist(tokens[1], &deps, " ");
|
|
|
|
if (!(option_mask32 & MODPROBE_OPT_REMOVE))
|
|
deps = llist_rev(deps);
|
|
|
|
rc = 0;
|
|
while (deps && rc == 0) {
|
|
fn = llist_pop(&deps);
|
|
filename2modname(fn, modname);
|
|
if (option_mask32 & MODPROBE_OPT_REMOVE) {
|
|
if (bb_delete_module(modname, O_EXCL) != 0)
|
|
rc = errno;
|
|
} else if (llist_find(loaded, modname) == NULL) {
|
|
options = gather_options(conf->options, modname,
|
|
strcmp(modname, module) == 0);
|
|
rc = bb_init_module(fn, options);
|
|
if (rc == 0)
|
|
llist_add_to(&loaded, xstrdup(modname));
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
free(options);
|
|
}
|
|
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
free(fn);
|
|
}
|
|
|
|
error_not_found:
|
|
config_close(p);
|
|
error:
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
RELEASE_CONFIG_BUFFER(modname);
|
|
if (rc > 0 && !(option_mask32 & INSMOD_OPT_SILENT))
|
|
bb_error_msg("Failed to %sload module %s: %s.",
|
|
(option_mask32 & MODPROBE_OPT_REMOVE) ? "un" : "",
|
|
module, moderror(rc));
|
|
return rc;
|
|
}
|
|
|
|
int modprobe_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
|
|
int modprobe_main(int argc UNUSED_PARAM, char **argv)
|
|
{
|
|
struct utsname uts;
|
|
int rc;
|
|
unsigned opt;
|
|
llist_t *options = NULL;
|
|
|
|
opt_complementary = "q-v:v-q";
|
|
opt = getopt32(argv, INSMOD_OPTS MODPROBE_OPTS INSMOD_ARGS,
|
|
NULL, NULL);
|
|
argv += optind;
|
|
|
|
if (opt & (MODPROBE_OPT_DUMP_ONLY | MODPROBE_OPT_LIST_ONLY |
|
|
MODPROBE_OPT_SHOW_ONLY))
|
|
bb_error_msg_and_die("not supported");
|
|
|
|
/* goto modules location */
|
|
xchdir(CONFIG_DEFAULT_MODULES_DIR);
|
|
uname(&uts);
|
|
xchdir(uts.release);
|
|
|
|
if (!argv[0]) {
|
|
if (opt & MODPROBE_OPT_REMOVE) {
|
|
if (bb_delete_module(NULL, O_NONBLOCK|O_EXCL) != 0)
|
|
bb_perror_msg_and_die("rmmod");
|
|
}
|
|
return EXIT_SUCCESS;
|
|
}
|
|
if (!(opt & MODPROBE_OPT_INSERT_ALL)) {
|
|
/* If not -a, we have only one module name,
|
|
* the rest of parameters are options */
|
|
add_option(&options, NULL, parse_cmdline_module_options(argv));
|
|
argv[1] = NULL;
|
|
}
|
|
|
|
/* cache modules */
|
|
{
|
|
char *s;
|
|
parser_t *parser = config_open2("/proc/modules", fopen_for_read);
|
|
while (config_read(parser, &s, 1, 1, "# \t", PARSE_NORMAL & ~PARSE_GREEDY))
|
|
llist_add_to(&loaded, xstrdup(s));
|
|
config_close(parser);
|
|
}
|
|
|
|
while (*argv) {
|
|
const char *arg = *argv;
|
|
struct modprobe_conf *conf;
|
|
|
|
conf = xzalloc(sizeof(*conf));
|
|
conf->options = options;
|
|
filename2modname(arg, conf->probename);
|
|
read_config(conf, "/etc/modprobe.conf");
|
|
read_config(conf, "/etc/modprobe.d");
|
|
if (ENABLE_FEATURE_MODUTILS_SYMBOLS
|
|
&& conf->aliases == NULL
|
|
&& strncmp(arg, "symbol:", 7) == 0
|
|
) {
|
|
read_config(conf, "modules.symbols");
|
|
}
|
|
|
|
if (ENABLE_FEATURE_MODUTILS_ALIAS && conf->aliases == NULL)
|
|
read_config(conf, "modules.alias");
|
|
|
|
if (conf->aliases == NULL) {
|
|
/* Try if module by literal name is found; literal
|
|
* names are blacklist only if '-b' is given. */
|
|
if (!(opt & MODPROBE_OPT_BLACKLIST) ||
|
|
check_blacklist(conf, conf->probename)) {
|
|
rc = do_modprobe(conf, conf->probename);
|
|
if (rc < 0 && !(opt & INSMOD_OPT_SILENT))
|
|
bb_error_msg("Module %s not found.", arg);
|
|
}
|
|
} else {
|
|
/* Probe all aliases */
|
|
while (conf->aliases != NULL) {
|
|
char *realname = llist_pop(&conf->aliases);
|
|
if (check_blacklist(conf, realname))
|
|
do_modprobe(conf, realname);
|
|
if (ENABLE_FEATURE_CLEAN_UP)
|
|
free(realname);
|
|
}
|
|
}
|
|
argv++;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|