further work on modprobe

function                                             old     new   delta
gather_options_str                                     -      75     +75
config_file_action                                   388     413     +25
load_modules_dep                                     175     192     +17
add_probe                                             67      72      +5
modprobe_main                                        494     488      -6
gather_options                                        89       -     -89
------------------------------------------------------------------------------
(add/remove: 1/1 grow/shrink: 3/1 up/down: 122/-95)            Total: 27 bytes
This commit is contained in:
Denis Vlasenko 2009-03-06 22:48:11 +00:00
parent 8517d6fc1f
commit a6b6f05379

View File

@ -13,18 +13,23 @@
#include <sys/utsname.h> #include <sys/utsname.h>
#include <fnmatch.h> #include <fnmatch.h>
//#define DBG(...) bb_error_msg(__VA_ARGS__)
#define DBG(...) ((void)0)
#define MODULE_FLAG_LOADED 0x0001 #define MODULE_FLAG_LOADED 0x0001
#define MODULE_FLAG_NEED_DEPS 0x0002 #define MODULE_FLAG_NEED_DEPS 0x0002
//Misnomer? Seems to mean "was seen in modules.dep": /* "was seen in modules.dep": */
#define MODULE_FLAG_EXISTS 0x0004 #define MODULE_FLAG_FOUND_IN_MODDEP 0x0004
#define MODULE_FLAG_BLACKLISTED 0x0008 #define MODULE_FLAG_BLACKLISTED 0x0008
struct module_entry { /* I'll call it ME. */ struct module_entry { /* I'll call it ME. */
unsigned flags; unsigned flags;
char *modname; /* stripped of /path/, .ext and s/-/_/g */ char *modname; /* stripped of /path/, .ext and s/-/_/g */
const char *probed_name; /* verbatim as seen on cmdline */ const char *probed_name; /* verbatim as seen on cmdline */
llist_t *aliases; /* strings. aliases from config files */ char *options; /* options from config files */
llist_t *options; /* strings. options from config files */ llist_t *realnames; /* strings. if this module is an alias, */
/* real module name is one of these. */
//Can there really be more than one? Example from real kernel?
llist_t *deps; /* strings. modules we depend on */ llist_t *deps; /* strings. modules we depend on */
}; };
@ -45,8 +50,8 @@ enum {
struct globals { struct globals {
llist_t *db; /* MEs of all modules ever seen (caching for speed) */ llist_t *db; /* MEs of all modules ever seen (caching for speed) */
llist_t *probes; /* MEs of module(s) requested on cmdline */ llist_t *probes; /* MEs of module(s) requested on cmdline */
llist_t *cmdline_mopts; /* strings. module options (from cmdline) */ char *cmdline_mopts; /* module options from cmdline */
int num_deps; /* what is this? */ int num_unresolved_deps;
/* bool. "Did we have 'symbol:FOO' requested on cmdline?" */ /* bool. "Did we have 'symbol:FOO' requested on cmdline?" */
smallint need_symbols; smallint need_symbols;
}; };
@ -56,6 +61,19 @@ struct globals {
static int read_config(const char *path); static int read_config(const char *path);
static char *gather_options_str(char *opts, const char *append)
{
/* Speed-optimized. We call gather_options_str many times. */
if (opts == NULL) {
opts = xstrdup(append);
} else {
int optlen = strlen(opts);
opts = xrealloc(opts, optlen + strlen(append) + 2);
sprintf(opts + optlen, " %s", append);
}
return opts;
}
static struct module_entry *helper_get_module(const char *module, int create) static struct module_entry *helper_get_module(const char *module, int create)
{ {
char modname[MODULE_NAME_LEN]; char modname[MODULE_NAME_LEN];
@ -91,14 +109,20 @@ static void add_probe(const char *name)
struct module_entry *m; struct module_entry *m;
m = get_or_add_modentry(name); m = get_or_add_modentry(name);
if (m->flags & MODULE_FLAG_LOADED) {
DBG("skipping %s, it is already loaded", name);
return;
}
m->probed_name = name; m->probed_name = name;
m->flags |= MODULE_FLAG_NEED_DEPS; m->flags |= MODULE_FLAG_NEED_DEPS;
llist_add_to(&G.probes, m); llist_add_to_end(&G.probes, m);
G.num_unresolved_deps++;
G.num_deps++;
if (ENABLE_FEATURE_MODUTILS_SYMBOLS if (ENABLE_FEATURE_MODUTILS_SYMBOLS
&& strncmp(m->modname, "symbol:", 7) == 0) && strncmp(m->modname, "symbol:", 7) == 0
) {
G.need_symbols = 1; G.need_symbols = 1;
}
} }
static int FAST_FUNC config_file_action(const char *filename, static int FAST_FUNC config_file_action(const char *filename,
@ -106,10 +130,8 @@ static int FAST_FUNC config_file_action(const char *filename,
void *userdata UNUSED_PARAM, void *userdata UNUSED_PARAM,
int depth UNUSED_PARAM) int depth UNUSED_PARAM)
{ {
RESERVE_CONFIG_BUFFER(modname, MODULE_NAME_LEN); char *tokens[3];
char *tokens[3], *rmod;
parser_t *p; parser_t *p;
llist_t *l;
struct module_entry *m; struct module_entry *m;
int rc = TRUE; int rc = TRUE;
@ -125,42 +147,51 @@ static int FAST_FUNC config_file_action(const char *filename,
while (config_read(p, tokens, 3, 2, "# \t", PARSE_NORMAL)) { while (config_read(p, tokens, 3, 2, "# \t", PARSE_NORMAL)) {
//Use index_in_strings? //Use index_in_strings?
if (strcmp(tokens[0], "alias") == 0) { if (strcmp(tokens[0], "alias") == 0) {
filename2modname(tokens[1], modname); /* alias <wildcard> <modulename> */
llist_t *l;
char wildcard[MODULE_NAME_LEN];
char *rmod;
if (tokens[2] == NULL) if (tokens[2] == NULL)
continue; continue;
filename2modname(tokens[1], wildcard);
for (l = G.probes; l != NULL; l = l->link) { for (l = G.probes; l != NULL; l = l->link) {
m = (struct module_entry *) l->data; m = (struct module_entry *) l->data;
if (fnmatch(modname, m->modname, 0) != 0) if (fnmatch(wildcard, m->modname, 0) != 0)
continue; continue;
rmod = filename2modname(tokens[2], NULL); rmod = filename2modname(tokens[2], NULL);
llist_add_to(&m->aliases, rmod); llist_add_to(&m->realnames, rmod);
if (m->flags & MODULE_FLAG_NEED_DEPS) { if (m->flags & MODULE_FLAG_NEED_DEPS) {
m->flags &= ~MODULE_FLAG_NEED_DEPS; m->flags &= ~MODULE_FLAG_NEED_DEPS;
G.num_deps--; G.num_unresolved_deps--;
} }
m = get_or_add_modentry(rmod); m = get_or_add_modentry(rmod);
m->flags |= MODULE_FLAG_NEED_DEPS; if (!(m->flags & MODULE_FLAG_NEED_DEPS)) {
G.num_deps++; m->flags |= MODULE_FLAG_NEED_DEPS;
G.num_unresolved_deps++;
}
} }
} else if (strcmp(tokens[0], "options") == 0) { } else if (strcmp(tokens[0], "options") == 0) {
/* options <modulename> <option...> */
if (tokens[2] == NULL) if (tokens[2] == NULL)
continue; continue;
m = get_or_add_modentry(tokens[1]); m = get_or_add_modentry(tokens[1]);
llist_add_to(&m->options, xstrdup(tokens[2])); m->options = gather_options_str(m->options, tokens[2]);
} else if (strcmp(tokens[0], "include") == 0) { } else if (strcmp(tokens[0], "include") == 0) {
/* include <filename> */
read_config(tokens[1]); read_config(tokens[1]);
} else if (ENABLE_FEATURE_MODPROBE_BLACKLIST } else if (ENABLE_FEATURE_MODPROBE_BLACKLIST
&& strcmp(tokens[0], "blacklist") == 0 && strcmp(tokens[0], "blacklist") == 0
) { ) {
/* blacklist <modulename> */
get_or_add_modentry(tokens[1])->flags |= MODULE_FLAG_BLACKLISTED; get_or_add_modentry(tokens[1])->flags |= MODULE_FLAG_BLACKLISTED;
} }
} }
config_close(p); config_close(p);
error: error:
RELEASE_CONFIG_BUFFER(modname);
return rc; return rc;
} }
@ -170,29 +201,17 @@ static int read_config(const char *path)
config_file_action, NULL, NULL, 1); config_file_action, NULL, NULL, 1);
} }
static char *gather_options(char *opts, llist_t *o)
{
int optlen;
if (opts == NULL)
opts = xstrdup("");
optlen = strlen(opts);
for (; o != NULL; o = o->link) {
opts = xrealloc(opts, optlen + strlen(o->data) + 2);
optlen += sprintf(opts + optlen, "%s ", o->data);
}
return opts;
}
static int do_modprobe(struct module_entry *m) static int do_modprobe(struct module_entry *m)
{ {
struct module_entry *m2; struct module_entry *m2;
char *fn, *options; char *fn, *options;
int rc = -1; int rc = -1;
if (!(m->flags & MODULE_FLAG_EXISTS)) if (!(m->flags & MODULE_FLAG_FOUND_IN_MODDEP)) {
DBG("skipping %s, not found in modules.dep", m->modname);
return -ENOENT; return -ENOENT;
}
DBG("do_modprob'ing %s", m->modname);
if (!(option_mask32 & MODPROBE_OPT_REMOVE)) if (!(option_mask32 & MODPROBE_OPT_REMOVE))
m->deps = llist_rev(m->deps); m->deps = llist_rev(m->deps);
@ -205,19 +224,23 @@ static int do_modprobe(struct module_entry *m)
if (bb_delete_module(m->modname, O_EXCL) != 0) if (bb_delete_module(m->modname, O_EXCL) != 0)
rc = errno; rc = errno;
} else if (!(m2->flags & MODULE_FLAG_LOADED)) { } else if (!(m2->flags & MODULE_FLAG_LOADED)) {
options = gather_options(NULL, m2->options); options = m2->options;
m2->options = NULL;
if (m == m2) if (m == m2)
options = gather_options(options, G.cmdline_mopts); options = gather_options_str(options, G.cmdline_mopts);
//TODO: looks like G.cmdline_mopts can contain either NULL or just one string, not more?
rc = bb_init_module(fn, options); rc = bb_init_module(fn, options);
DBG("loaded %s '%s', rc:%d", fn, options, rc);
if (rc == 0) if (rc == 0)
m2->flags |= MODULE_FLAG_LOADED; m2->flags |= MODULE_FLAG_LOADED;
free(options); free(options);
} else {
DBG("%s is already loaded, skipping", fn);
} }
free(fn); free(fn);
} }
//FIXME: what if rc < 0?
if (rc > 0 && !(option_mask32 & INSMOD_OPT_SILENT)) { if (rc > 0 && !(option_mask32 & INSMOD_OPT_SILENT)) {
bb_error_msg("failed to %sload module %s: %s", bb_error_msg("failed to %sload module %s: %s",
(option_mask32 & MODPROBE_OPT_REMOVE) ? "un" : "", (option_mask32 & MODPROBE_OPT_REMOVE) ? "un" : "",
@ -235,17 +258,15 @@ static void load_modules_dep(void)
char *colon, *tokens[2]; char *colon, *tokens[2];
parser_t *p; parser_t *p;
p = config_open2(CONFIG_DEFAULT_DEPMOD_FILE, xfopen_for_read);
//Still true?
/* Modprobe does not work at all without modprobe.dep, /* Modprobe does not work at all without modprobe.dep,
* even if the full module name is given. Returning error here * even if the full module name is given. Returning error here
* was making us later confuse user with this message: * was making us later confuse user with this message:
* "module /full/path/to/existing/file/module.ko not found". * "module /full/path/to/existing/file/module.ko not found".
* It's better to die immediately, with good message: */ * It's better to die immediately, with good message.
// if (p == NULL) * xfopen_for_read provides that. */
// bb_perror_msg_and_die("can't open '%s'", CONFIG_DEFAULT_DEPMOD_FILE); p = config_open2(CONFIG_DEFAULT_DEPMOD_FILE, xfopen_for_read);
while (G.num_deps while (G.num_unresolved_deps
&& config_read(p, tokens, 2, 1, "# \t", PARSE_NORMAL) && config_read(p, tokens, 2, 1, "# \t", PARSE_NORMAL)
) { ) {
colon = last_char_is(tokens[0], ':'); colon = last_char_is(tokens[0], ':');
@ -256,11 +277,18 @@ static void load_modules_dep(void)
m = get_modentry(tokens[0]); m = get_modentry(tokens[0]);
if (m == NULL) if (m == NULL)
continue; continue;
//Can we skip it too if it is MODULE_FLAG_LOADED?
m->flags |= MODULE_FLAG_EXISTS; /* Optimization... */
if ((m->flags & MODULE_FLAG_LOADED)
&& !(option_mask32 & MODPROBE_OPT_REMOVE)
) {
DBG("skip deps of %s, it's already loaded", tokens[0]);
continue;
}
m->flags |= MODULE_FLAG_FOUND_IN_MODDEP;
if ((m->flags & MODULE_FLAG_NEED_DEPS) && (m->deps == NULL)) { if ((m->flags & MODULE_FLAG_NEED_DEPS) && (m->deps == NULL)) {
G.num_deps--; G.num_unresolved_deps--;
llist_add_to(&m->deps, xstrdup(tokens[0])); llist_add_to(&m->deps, xstrdup(tokens[0]));
if (tokens[1]) if (tokens[1])
string_to_llist(tokens[1], &m->deps, " "); string_to_llist(tokens[1], &m->deps, " ");
@ -285,33 +313,22 @@ int modprobe_main(int argc UNUSED_PARAM, char **argv)
MODPROBE_OPT_SHOW_ONLY)) MODPROBE_OPT_SHOW_ONLY))
bb_error_msg_and_die("not supported"); bb_error_msg_and_die("not supported");
/* Goto modules location */
xchdir(CONFIG_DEFAULT_MODULES_DIR);
uname(&uts);
xchdir(uts.release);
if (!argv[0]) { if (!argv[0]) {
if (opt & MODPROBE_OPT_REMOVE) { if (opt & MODPROBE_OPT_REMOVE) {
/* TODO: comment here */ /* "modprobe -r" (w/o params).
* "If name is NULL, all unused modules marked
* autoclean will be removed".
*/
if (bb_delete_module(NULL, O_NONBLOCK|O_EXCL) != 0) if (bb_delete_module(NULL, O_NONBLOCK|O_EXCL) != 0)
bb_perror_msg_and_die("rmmod"); bb_perror_msg_and_die("rmmod");
} }
return EXIT_SUCCESS; return EXIT_SUCCESS;
} }
if (opt & MODPROBE_OPT_INSERT_ALL) { /* Goto modules location */
/* Each argument is a module name */ xchdir(CONFIG_DEFAULT_MODULES_DIR);
do { uname(&uts);
add_probe(*argv++); xchdir(uts.release);
} while (*argv);
G.probes = llist_rev(G.probes);
} else {
/* First argument is module name, rest are parameters */
add_probe(argv[0]);
//TODO: looks like G.cmdline_mopts can contain either NULL or just one string, not more?
//optimize it them
llist_add_to(&G.cmdline_mopts, parse_cmdline_module_options(argv));
}
/* Retrieve module names of already loaded modules */ /* Retrieve module names of already loaded modules */
{ {
@ -322,39 +339,58 @@ int modprobe_main(int argc UNUSED_PARAM, char **argv)
config_close(parser); config_close(parser);
} }
if (opt & MODPROBE_OPT_INSERT_ALL) {
/* Each argument is a module name */
do {
add_probe(*argv++);
} while (*argv);
} else {
/* First argument is module name, rest are parameters */
add_probe(argv[0]);
G.cmdline_mopts = parse_cmdline_module_options(argv);
}
/* Happens if all requested modules are already loaded */
if (G.probes == NULL)
return EXIT_SUCCESS;
read_config("/etc/modprobe.conf"); read_config("/etc/modprobe.conf");
read_config("/etc/modprobe.d"); read_config("/etc/modprobe.d");
if (ENABLE_FEATURE_MODUTILS_SYMBOLS && G.need_symbols) if (ENABLE_FEATURE_MODUTILS_SYMBOLS && G.need_symbols)
read_config("modules.symbols"); read_config("modules.symbols");
load_modules_dep(); load_modules_dep();
if (ENABLE_FEATURE_MODUTILS_ALIAS && G.num_deps) { if (ENABLE_FEATURE_MODUTILS_ALIAS && G.num_unresolved_deps) {
read_config("modules.alias"); read_config("modules.alias");
load_modules_dep(); load_modules_dep();
} }
while ((me = llist_pop(&G.probes)) != NULL) { while ((me = llist_pop(&G.probes)) != NULL) {
if (me->aliases == NULL) { if (me->realnames == NULL) {
/* Try if module by literal name is found; literal /* This is not an alias. Literal names are blacklisted
* names are blacklisted only if '-b' is given. */ * only if '-b' is given.
*/
if (!(opt & MODPROBE_OPT_BLACKLIST) if (!(opt & MODPROBE_OPT_BLACKLIST)
|| !(me->flags & MODULE_FLAG_BLACKLISTED) || !(me->flags & MODULE_FLAG_BLACKLISTED)
) { ) {
rc = do_modprobe(me); rc = do_modprobe(me);
//FIXME: what if rc > 0?
if (rc < 0 && !(opt & INSMOD_OPT_SILENT)) if (rc < 0 && !(opt & INSMOD_OPT_SILENT))
bb_error_msg("module %s not found", bb_error_msg("module %s not found",
me->probed_name); me->probed_name);
} }
} else { } else {
/* Probe all aliases */ /* Probe all realnames */
do { do {
char *realname = llist_pop(&me->aliases); char *realname = llist_pop(&me->realnames);
struct module_entry *m2; struct module_entry *m2;
DBG("probing %s by realname %s", me->modname, realname);
m2 = get_or_add_modentry(realname); m2 = get_or_add_modentry(realname);
if (!(m2->flags & MODULE_FLAG_BLACKLISTED)) if (!(m2->flags & MODULE_FLAG_BLACKLISTED))
do_modprobe(m2); do_modprobe(m2);
//FIXME: error check?
free(realname); free(realname);
} while (me->aliases != NULL); } while (me->realnames != NULL);
} }
} }