busybox/modutils/modprobe.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.
 */

/* Note that unlike older versions of modules.dep/depmod (busybox and m-i-t),
 * we expect the full dependency list to be specified in modules.dep.
 * Older versions would only export the direct dependency list.
 */
#include "libbb.h"
#include "modutils.h"
#include <sys/utsname.h>
#include <fnmatch.h>

//#define DBG(fmt, ...) bb_error_msg("%s: " fmt, __func__, ## __VA_ARGS__)
#define DBG(...) ((void)0)

#define MODULE_FLAG_LOADED              0x0001
#define MODULE_FLAG_NEED_DEPS           0x0002
/* "was seen in modules.dep": */
#define MODULE_FLAG_FOUND_IN_MODDEP     0x0004
#define MODULE_FLAG_BLACKLISTED         0x0008

struct module_entry { /* I'll call it ME. */
	unsigned flags;
	char *modname; /* stripped of /path/, .ext and s/-/_/g */
	const char *probed_name; /* verbatim as seen on cmdline */
	char *options; /* 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 */
};

/* NB: INSMOD_OPT_SILENT bit suppresses ONLY non-existent modules,
 * not deleted ones (those are still listed in modules.dep).
 * module-init-tools version 3.4:
 * # modprobe bogus
 * FATAL: Module bogus not found. [exitcode 1]
 * # modprobe -q bogus            [silent, exitcode still 1]
 * but:
 * # rm kernel/drivers/net/dummy.ko
 * # modprobe -q dummy
 * FATAL: Could not open '/lib/modules/xxx/kernel/drivers/net/dummy.ko': No such file or directory
 * [exitcode 1]
 */
#define MODPROBE_OPTS  "acdlnrt:VC:" IF_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,
};

struct globals {
	llist_t *db; /* MEs of all modules ever seen (caching for speed) */
	llist_t *probes; /* MEs of module(s) requested on cmdline */
	char *cmdline_mopts; /* module options from cmdline */
	int num_unresolved_deps;
	/* bool. "Did we have 'symbol:FOO' requested on cmdline?" */
	smallint need_symbols;
};
#define G (*(struct globals*)&bb_common_bufsiz1)
#define INIT_G() do { } while (0)


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 (append) {
		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)
{
	char modname[MODULE_NAME_LEN];
	struct module_entry *e;
	llist_t *l;

	filename2modname(module, modname);
	for (l = G.db; l != NULL; l = l->link) {
		e = (struct module_entry *) l->data;
		if (strcmp(e->modname, modname) == 0)
			return e;
	}
	if (!create)
		return NULL;

	e = xzalloc(sizeof(*e));
	e->modname = xstrdup(modname);
	llist_add_to(&G.db, e);

	return e;
}
static struct module_entry *get_or_add_modentry(const char *module)
{
	return helper_get_module(module, 1);
}
static struct module_entry *get_modentry(const char *module)
{
	return helper_get_module(module, 0);
}

static void add_probe(const char *name)
{
	struct module_entry *m;

	m = get_or_add_modentry(name);
	if (!(option_mask32 & MODPROBE_OPT_REMOVE)
	 && (m->flags & MODULE_FLAG_LOADED)
	) {
		DBG("skipping %s, it is already loaded", name);
		return;
	}

	DBG("queuing %s", name);
	m->probed_name = name;
	m->flags |= MODULE_FLAG_NEED_DEPS;
	llist_add_to_end(&G.probes, m);
	G.num_unresolved_deps++;
	if (ENABLE_FEATURE_MODUTILS_SYMBOLS
	 && strncmp(m->modname, "symbol:", 7) == 0
	) {
		G.need_symbols = 1;
	}
}

static int FAST_FUNC config_file_action(const char *filename,
					struct stat *statbuf UNUSED_PARAM,
					void *userdata UNUSED_PARAM,
					int depth UNUSED_PARAM)
{
	char *tokens[3];
	parser_t *p;
	struct module_entry *m;
	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)) {
//Use index_in_strings?
		if (strcmp(tokens[0], "alias") == 0) {
			/* alias <wildcard> <modulename> */
			llist_t *l;
			char wildcard[MODULE_NAME_LEN];
			char *rmod;

			if (tokens[2] == NULL)
				continue;
			filename2modname(tokens[1], wildcard);

			for (l = G.probes; l != NULL; l = l->link) {
				m = (struct module_entry *) l->data;
				if (fnmatch(wildcard, m->modname, 0) != 0)
					continue;
				rmod = filename2modname(tokens[2], NULL);
				llist_add_to(&m->realnames, rmod);

				if (m->flags & MODULE_FLAG_NEED_DEPS) {
					m->flags &= ~MODULE_FLAG_NEED_DEPS;
					G.num_unresolved_deps--;
				}

				m = get_or_add_modentry(rmod);
				if (!(m->flags & MODULE_FLAG_NEED_DEPS)) {
					m->flags |= MODULE_FLAG_NEED_DEPS;
					G.num_unresolved_deps++;
				}
			}
		} else if (strcmp(tokens[0], "options") == 0) {
			/* options <modulename> <option...> */
			if (tokens[2] == NULL)
				continue;
			m = get_or_add_modentry(tokens[1]);
			m->options = gather_options_str(m->options, tokens[2]);
		} else if (strcmp(tokens[0], "include") == 0) {
			/* include <filename> */
			read_config(tokens[1]);
		} else if (ENABLE_FEATURE_MODPROBE_BLACKLIST
		 && strcmp(tokens[0], "blacklist") == 0
		) {
			/* blacklist <modulename> */
			get_or_add_modentry(tokens[1])->flags |= MODULE_FLAG_BLACKLISTED;
		}
	}
	config_close(p);
 error:
	return rc;
}

static int read_config(const char *path)
{
	return recursive_action(path, ACTION_RECURSE | ACTION_QUIET,
				config_file_action, NULL, NULL, 1);
}

static const char *humanly_readable_name(struct module_entry *m)
{
	/* probed_name may be NULL. modname always exists. */
	return m->probed_name ? m->probed_name : m->modname;
}

/* Return: similar to bb_init_module:
 * 0 on success,
 * -errno on open/read error,
 * errno on init_module() error
 */
static int do_modprobe(struct module_entry *m)
{
	struct module_entry *m2 = m2; /* for compiler */
	char *fn, *options;
	int rc, first;
	llist_t *l;

	if (!(m->flags & MODULE_FLAG_FOUND_IN_MODDEP)) {
		if (!(option_mask32 & INSMOD_OPT_SILENT))
			bb_error_msg("module %s not found in modules.dep",
				humanly_readable_name(m));
		return -ENOENT;
	}
	DBG("do_modprob'ing %s", m->modname);

	if (!(option_mask32 & MODPROBE_OPT_REMOVE))
		m->deps = llist_rev(m->deps);

	for (l = m->deps; l != NULL; l = l->link)
		DBG("dep: %s", l->data);

	first = 1;
	rc = 0;
	while (m->deps) {
		rc = 0;
		fn = llist_pop(&m->deps); /* we leak it */
		m2 = get_or_add_modentry(fn);

		if (option_mask32 & MODPROBE_OPT_REMOVE) {
			/* modprobe -r */
			if (m2->flags & MODULE_FLAG_LOADED) {
				rc = bb_delete_module(m2->modname, O_EXCL);
				if (rc) {
					if (first) {
						bb_error_msg("failed to unload module %s: %s",
							humanly_readable_name(m2),
							moderror(rc));
						break;
					}
				} else {
					m2->flags &= ~MODULE_FLAG_LOADED;
				}
			}
			/* do not error out if *deps* fail to unload */
			first = 0;
			continue;
		}

		if (m2->flags & MODULE_FLAG_LOADED) {
			DBG("%s is already loaded, skipping", fn);
			continue;
		}

		options = m2->options;
		m2->options = NULL;
		if (m == m2)
			options = gather_options_str(options, G.cmdline_mopts);
		rc = bb_init_module(fn, options);
		DBG("loaded %s '%s', rc:%d", fn, options, rc);
		if (rc == EEXIST)
			rc = 0;
		free(options);
		if (rc) {
			bb_error_msg("failed to load module %s (%s): %s",
				humanly_readable_name(m2),
				fn,
				moderror(rc)
			);
			break;
		}
		m2->flags |= MODULE_FLAG_LOADED;
	}

	return rc;
}

static void load_modules_dep(void)
{
	struct module_entry *m;
	char *colon, *tokens[2];
	parser_t *p;

	/* Modprobe does not work at all without modules.dep,
	 * even if the full module name is given. Returning error here
	 * was making us later confuse user with this message:
	 * "module /full/path/to/existing/file/module.ko not found".
	 * It's better to die immediately, with good message.
	 * xfopen_for_read provides that. */
	p = config_open2(CONFIG_DEFAULT_DEPMOD_FILE, xfopen_for_read);

	while (G.num_unresolved_deps
	 && config_read(p, tokens, 2, 1, "# \t", PARSE_NORMAL)
	) {
		colon = last_char_is(tokens[0], ':');
		if (colon == NULL)
			continue;
		*colon = 0;

		m = get_modentry(tokens[0]);
		if (m == NULL)
			continue;

		/* 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)) {
			G.num_unresolved_deps--;
			llist_add_to(&m->deps, xstrdup(tokens[0]));
			if (tokens[1])
				string_to_llist(tokens[1], &m->deps, " \t");
		} else
			DBG("skipping dep line");
	}
	config_close(p);
}

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;
	struct module_entry *me;

	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");

	if (!argv[0]) {
		if (opt & MODPROBE_OPT_REMOVE) {
			/* "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)
				bb_perror_msg_and_die("rmmod");
		}
		return EXIT_SUCCESS;
	}

	/* Goto modules location */
	xchdir(CONFIG_DEFAULT_MODULES_DIR);
	uname(&uts);
	xchdir(uts.release);

	/* Retrieve module names of already loaded 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))
			get_or_add_modentry(s)->flags |= MODULE_FLAG_LOADED;
		config_close(parser);
	}

	if (opt & (MODPROBE_OPT_INSERT_ALL | MODPROBE_OPT_REMOVE)) {
		/* Each argument is a module name */
		do {
			DBG("adding module %s", *argv);
			add_probe(*argv++);
		} while (*argv);
	} else {
		/* First argument is module name, rest are parameters */
		DBG("probing just module %s", *argv);
		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.d");
	if (ENABLE_FEATURE_MODUTILS_SYMBOLS && G.need_symbols)
		read_config("modules.symbols");
	load_modules_dep();
	if (ENABLE_FEATURE_MODUTILS_ALIAS && G.num_unresolved_deps) {
		read_config("modules.alias");
		load_modules_dep();
	}

	rc = 0;
	while ((me = llist_pop(&G.probes)) != NULL) {
		if (me->realnames == NULL) {
			DBG("probing by module name");
			/* This is not an alias. Literal names are blacklisted
			 * only if '-b' is given.
			 */
			if (!(opt & MODPROBE_OPT_BLACKLIST)
			 || !(me->flags & MODULE_FLAG_BLACKLISTED)
			) {
				rc |= do_modprobe(me);
			}
			continue;
		}

		/* Probe all real names for the alias */
		do {
			char *realname = llist_pop(&me->realnames);
			struct module_entry *m2;

			DBG("probing alias %s by realname %s", me->modname, realname);
			m2 = get_or_add_modentry(realname);
			if (!(m2->flags & MODULE_FLAG_BLACKLISTED)
			 && (!(m2->flags & MODULE_FLAG_LOADED)
			    || (opt & MODPROBE_OPT_REMOVE))
			) {
//TODO: we can pass "me" as 2nd param to do_modprobe,
//and make do_modprobe emit more meaningful error messages
//with alias name included, not just module name alias resolves to.
				rc |= do_modprobe(m2);
			}
			free(realname);
		} while (me->realnames != NULL);
	}

	return (rc != 0);
}
1	niro	532	/* vi: set sw=4 ts=4: */
2			/*
3			* Modprobe written from scratch for BusyBox
4			*
5	niro	816	* Copyright (c) 2008 Timo Teras <timo.teras@iki.fi>
6			* Copyright (c) 2008 Vladimir Dronnikov
7	niro	532	*
8			* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
9	niro	816	*/
10	niro	532
11	niro	984	/* 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	niro	816	#include "libbb.h"
16			#include "modutils.h"
17	niro	532	#include <sys/utsname.h>
18			#include <fnmatch.h>
19
20	niro	984	//#define DBG(fmt, ...) bb_error_msg("%s: " fmt, __func__, ## __VA_ARGS__)
21			#define DBG(...) ((void)0)
22	niro	532
23	niro	984	#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	niro	532	};
39
40	niro	984	/* 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	niro	816	enum {
54	niro	984	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	niro	532	};
65
66	niro	984	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	niro	532
77
78	niro	984	static int read_config(const char *path);
79
80			static char gather_options_str(char opts, const char *append)
81	niro	816	{
82	niro	984	/* 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	niro	532
95	niro	984	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	niro	816	}
116	niro	984	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	niro	532
125	niro	984	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	niro	816	static int FAST_FUNC config_file_action(const char *filename,
150			struct stat *statbuf UNUSED_PARAM,
151	niro	984	void *userdata UNUSED_PARAM,
152	niro	816	int depth UNUSED_PARAM)
153	niro	532	{
154	niro	816	char *tokens[3];
155			parser_t *p;
156	niro	984	struct module_entry *m;
157	niro	816	int rc = TRUE;
158	niro	532
159	niro	816	if (bb_basename(filename)[0] == '.')
160			goto error;
161	niro	532
162	niro	816	p = config_open2(filename, fopen_for_read);
163			if (p == NULL) {
164			rc = FALSE;
165			goto error;
166			}
167	niro	532
168	niro	816	while (config_read(p, tokens, 3, 2, "# \t", PARSE_NORMAL)) {
169	niro	984	//Use index_in_strings?
170	niro	816	if (strcmp(tokens[0], "alias") == 0) {
171	niro	984	/* 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	niro	816	} else if (strcmp(tokens[0], "options") == 0) {
199	niro	984	/* 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	niro	816	} else if (strcmp(tokens[0], "include") == 0) {
205	niro	984	/* 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	niro	532	}
213			}
214	niro	816	config_close(p);
215	niro	984	error:
216	niro	816	return rc;
217	niro	532	}
218
219	niro	984	static int read_config(const char *path)
220	niro	532	{
221	niro	816	return recursive_action(path, ACTION_RECURSE \| ACTION_QUIET,
222	niro	984	config_file_action, NULL, NULL, 1);
223	niro	532	}
224
225	niro	984	static const char humanly_readable_name(struct module_entry m)
226	niro	532	{
227	niro	984	/* probed_name may be NULL. modname always exists. */
228			return m->probed_name ? m->probed_name : m->modname;
229			}
230	niro	532
231	niro	984	/* 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	niro	532
243	niro	984	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	niro	532	continue;
282	niro	984	}
283
284			if (m2->flags & MODULE_FLAG_LOADED) {
285			DBG("%s is already loaded, skipping", fn);
286	niro	816	continue;
287	niro	984	}
288	niro	532
289	niro	984	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	niro	532	}
308	niro	984
309			return rc;
310	niro	532	}
311
312	niro	984	static void load_modules_dep(void)
313	niro	532	{
314	niro	984	struct module_entry *m;
315			char colon, tokens[2];
316	niro	816	parser_t *p;
317	niro	532
318	niro	984	/* 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	niro	532
326	niro	984	while (G.num_unresolved_deps
327			&& config_read(p, tokens, 2, 1, "# \t", PARSE_NORMAL)
328			) {
329	niro	816	colon = last_char_is(tokens[0], ':');
330			if (colon == NULL)
331	niro	532	continue;
332	niro	984	*colon = 0;
333	niro	532
334	niro	984	m = get_modentry(tokens[0]);
335			if (m == NULL)
336			continue;
337	niro	532
338	niro	984	/* 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	niro	532	}
345
346	niro	984	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	niro	532	}
355	niro	816	config_close(p);
356	niro	532	}
357
358	niro	816	int modprobe_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
359			int modprobe_main(int argc UNUSED_PARAM, char **argv)
360	niro	532	{
361	niro	816	struct utsname uts;
362			int rc;
363			unsigned opt;
364	niro	984	struct module_entry *me;
365	niro	532
366	niro	816	opt_complementary = "q-v:v-q";
367	niro	984	opt = getopt32(argv, INSMOD_OPTS MODPROBE_OPTS INSMOD_ARGS, NULL, NULL);
368	niro	816	argv += optind;
369	niro	532
370	niro	816	if (opt & (MODPROBE_OPT_DUMP_ONLY \| MODPROBE_OPT_LIST_ONLY \|
371			MODPROBE_OPT_SHOW_ONLY))
372			bb_error_msg_and_die("not supported");
373	niro	532
374	niro	816	if (!argv[0]) {
375			if (opt & MODPROBE_OPT_REMOVE) {
376	niro	984	/* "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	niro	816	bb_perror_msg_and_die("rmmod");
382	niro	532	}
383	niro	816	return EXIT_SUCCESS;
384	niro	532	}
385
386	niro	984	/* 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	niro	816	{
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	niro	984	get_or_add_modentry(s)->flags \|= MODULE_FLAG_LOADED;
397	niro	816	config_close(parser);
398	niro	532	}
399
400	niro	984	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	niro	532
413	niro	984	/* 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	niro	532	}
441
442	niro	984	/* Probe all real names for the alias */
443			do {
444			char *realname = llist_pop(&me->realnames);
445			struct module_entry *m2;
446	niro	532
447	niro	984	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	niro	532	}
458	niro	984	free(realname);
459			} while (me->realnames != NULL);
460	niro	532	}
461
462	niro	984	return (rc != 0);
463	niro	532	}