modprobe.c revision 07df56eb2434b04880e3c92be6ab63a4b4c8e4aa 
1/*
2 * kmod-modprobe - manage linux kernel modules using libkmod.
3 *
4 * Copyright (C) 2011-2012  ProFUSION embedded systems
5 *
6 * This program is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20#include <assert.h>
21#include <stdio.h>
22#include <stdlib.h>
23#include <stdbool.h>
24#include <getopt.h>
25#include <errno.h>
26#include <string.h>
27#include <sys/types.h>
28#include <sys/stat.h>
29#include <sys/utsname.h>
30#include <sys/wait.h>
31#include <unistd.h>
32#include <syslog.h>
33#include <limits.h>
34
35#include "libkmod.h"
36#include "libkmod-array.h"
37#include "macro.h"
38
39static int log_priority = LOG_CRIT;
40static int use_syslog = 0;
41
42#define DEFAULT_VERBOSE LOG_WARNING
43static int verbose = DEFAULT_VERBOSE;
44static int do_show = 0;
45static int dry_run = 0;
46static int ignore_loaded = 0;
47static int lookup_only = 0;
48static int first_time = 0;
49static int ignore_commands = 0;
50static int use_blacklist = 0;
51static int force = 0;
52static int strip_modversion = 0;
53static int strip_vermagic = 0;
54static int remove_dependencies = 0;
55static int quiet_inuse = 0;
56
57static const char cmdopts_s[] = "arRibfDcnC:d:S:sqvVh";
58static const struct option cmdopts[] = {
59	{"all", no_argument, 0, 'a'},
60	{"remove", no_argument, 0, 'r'},
61	{"remove-dependencies", no_argument, 0, 5},
62	{"resolve-alias", no_argument, 0, 'R'},
63	{"first-time", no_argument, 0, 3},
64	{"ignore-install", no_argument, 0, 'i'},
65	{"ignore-remove", no_argument, 0, 'i'},
66	{"use-blacklist", no_argument, 0, 'b'},
67	{"force", no_argument, 0, 'f'},
68	{"force-modversion", no_argument, 0, 2},
69	{"force-vermagic", no_argument, 0, 1},
70
71	{"show-depends", no_argument, 0, 'D'},
72	{"showconfig", no_argument, 0, 'c'},
73	{"show-config", no_argument, 0, 'c'},
74	{"show-modversions", no_argument, 0, 4},
75	{"dump-modversions", no_argument, 0, 4},
76
77	{"dry-run", no_argument, 0, 'n'},
78	{"show", no_argument, 0, 'n'},
79
80	{"config", required_argument, 0, 'C'},
81	{"dirname", required_argument, 0, 'd'},
82	{"set-version", required_argument, 0, 'S'},
83
84	{"syslog", no_argument, 0, 's'},
85	{"quiet", no_argument, 0, 'q'},
86	{"verbose", no_argument, 0, 'v'},
87	{"version", no_argument, 0, 'V'},
88	{"help", no_argument, 0, 'h'},
89	{NULL, 0, 0, 0}
90};
91
92static void help(const char *progname)
93{
94	fprintf(stderr,
95		"Usage:\n"
96		"\t%s [options] [-i] [-b] modulename\n"
97		"\t%s [options] -a [-i] [-b] modulename [modulename...]\n"
98		"\t%s [options] -r [-i] modulename\n"
99		"\t%s [options] -r -a [-i] modulename [modulename...]\n"
100		"\t%s [options] -c\n"
101		"\t%s [options] --dump-modversions filename\n"
102		"Management Options:\n"
103		"\t-a, --all                   Consider every non-argument to\n"
104		"\t                            be a module name to be inserted\n"
105		"\t                            or removed (-r)\n"
106		"\t-r, --remove                Remove modules instead of inserting\n"
107		"\t    --remove-dependencies   Also remove modules depending on it\n"
108		"\t-R, --resolve-alias         Only lookup and print alias and exit\n"
109		"\t    --first-time            Fail if module already inserted or removed\n"
110		"\t-i, --ignore-install        Ignore install commands\n"
111		"\t-i, --ignore-remove         Ignore remove commands\n"
112		"\t-b, --use-blacklist         Apply blacklist to resolved alias.\n"
113		"\t-f, --force                 Force module insertion or removal.\n"
114		"\t                            implies --force-modversions and\n"
115		"\t                            --force-vermagic\n"
116		"\t    --force-modversion      Ignore module's version\n"
117		"\t    --force-vermagic        Ignore module's version magic\n"
118		"\n"
119		"Query Options:\n"
120		"\t-D, --show-depends          Only print module dependencies and exit\n"
121		"\t-c, --showconfig            Print out known configuration and exit\n"
122		"\t-c, --show-config           Same as --showconfig\n"
123		"\t    --show-modversions      Dump module symbol version and exit\n"
124		"\t    --dump-modversions      Same as --show-modversions\n"
125		"\n"
126		"General Options:\n"
127		"\t-n, --dry-run               Do not execute operations, just print out\n"
128		"\t-n, --show                  Same as --dry-run\n"
129
130		"\t-C, --config=FILE           Use FILE instead of default search paths\n"
131		"\t-d, --dirname=DIR           Use DIR as filesystem root for /lib/modules\n"
132		"\t-S, --set-version=VERSION   Use VERSION instead of `uname -r`\n"
133
134		"\t-s, --syslog                print to syslog, not stderr\n"
135		"\t-q, --quiet                 disable messages\n"
136		"\t-v, --verbose               enables more messages\n"
137		"\t-V, --version               show version\n"
138		"\t-h, --help                  show this help\n",
139		progname, progname, progname, progname, progname, progname);
140}
141
142static inline void _show(const char *fmt, ...)
143{
144	va_list args;
145
146	if (!do_show && verbose <= DEFAULT_VERBOSE)
147		return;
148
149	va_start(args, fmt);
150	vfprintf(stdout, fmt, args);
151	fflush(stdout);
152	va_end(args);
153}
154
155static _always_inline_ const char *prio_to_str(int prio)
156{
157	const char *prioname;
158	char buf[32];
159
160	switch (prio) {
161	case LOG_CRIT:
162		prioname = "FATAL";
163		break;
164	case LOG_ERR:
165		prioname = "ERROR";
166		break;
167	case LOG_WARNING:
168		prioname = "WARNING";
169		break;
170	case LOG_NOTICE:
171		prioname = "NOTICE";
172		break;
173	case LOG_INFO:
174		prioname = "INFO";
175		break;
176	case LOG_DEBUG:
177		prioname = "DEBUG";
178		break;
179	default:
180		snprintf(buf, sizeof(buf), "LOG-%03d", prio);
181		prioname = buf;
182	}
183
184	return prioname;
185}
186
187static void log_modprobe(void *data, int priority, const char *file, int line,
188			const char *fn, const char *format, va_list args)
189{
190	const char *prioname = prio_to_str(priority);
191	char *str;
192
193	if (vasprintf(&str, format, args) < 0)
194		return;
195
196	if (use_syslog) {
197#ifdef ENABLE_DEBUG
198		syslog(priority, "%s: %s:%d %s() %s", prioname, file, line,
199		       fn, str);
200#else
201		syslog(priority, "%s: %s", prioname, str);
202#endif
203	} else {
204#ifdef ENABLE_DEBUG
205		fprintf(stderr, "modprobe: %s: %s:%d %s() %s", prioname, file,
206			line, fn, str);
207#else
208		fprintf(stderr, "modprobe: %s: %s", prioname, str);
209#endif
210	}
211
212	free(str);
213	(void)data;
214}
215
216static inline void _log(int prio, const char *fmt, ...)
217{
218	const char *prioname;
219	char *msg;
220	va_list args;
221
222	if (prio > verbose)
223		return;
224
225	va_start(args, fmt);
226	if (vasprintf(&msg, fmt, args) < 0)
227		msg = NULL;
228	va_end(args);
229	if (msg == NULL)
230		return;
231
232	prioname = prio_to_str(prio);
233
234	if (use_syslog)
235		syslog(LOG_NOTICE, "%s: %s", prioname, msg);
236	else
237		fprintf(stderr, "%s: %s", prioname, msg);
238	free(msg);
239
240	if (prio <= LOG_CRIT)
241		exit(EXIT_FAILURE);
242}
243#define ERR(...) _log(LOG_ERR, __VA_ARGS__)
244#define WRN(...) _log(LOG_WARNING, __VA_ARGS__)
245#define INF(...) _log(LOG_INFO, __VA_ARGS__)
246#define DBG(...) _log(LOG_DEBUG, __VA_ARGS__)
247#define LOG(...) _log(log_priority, __VA_ARGS__)
248#define SHOW(...) _show(__VA_ARGS__)
249
250static int show_config(struct kmod_ctx *ctx)
251{
252	struct config_iterators {
253		const char *name;
254		struct kmod_config_iter *(*get_iter)(const struct kmod_ctx *ctx);
255	} ci[] = {
256		{ "blacklist", kmod_config_get_blacklists },
257		{ "install", kmod_config_get_install_commands },
258		{ "remove", kmod_config_get_remove_commands },
259		{ "alias", kmod_config_get_aliases },
260		{ "options", kmod_config_get_options },
261		{ "softdep", kmod_config_get_softdeps },
262	};
263	size_t i;
264
265	for (i = 0;  i < ARRAY_SIZE(ci); i++) {
266		struct kmod_config_iter *iter = ci[i].get_iter(ctx);
267
268		if (iter == NULL)
269			continue;
270
271		while (kmod_config_iter_next(iter)) {
272			const char *val;
273
274			printf("%s %s", ci[i].name,
275					kmod_config_iter_get_key(iter));
276			val = kmod_config_iter_get_value(iter);
277			if (val != NULL) {
278				putchar(' ');
279				puts(val);
280			} else
281				putchar('\n');
282		}
283
284		kmod_config_iter_free_iter(iter);
285	}
286
287	puts("\n# End of configuration files. Dumping indexes now:\n");
288	fflush(stdout);
289
290	kmod_dump_index(ctx, KMOD_INDEX_MODULES_ALIAS, STDOUT_FILENO);
291	kmod_dump_index(ctx, KMOD_INDEX_MODULES_SYMBOL, STDOUT_FILENO);
292
293	return 0;
294}
295
296static int show_modversions(struct kmod_ctx *ctx, const char *filename)
297{
298	struct kmod_list *l, *list = NULL;
299	struct kmod_module *mod;
300	int err = kmod_module_new_from_path(ctx, filename, &mod);
301	if (err < 0) {
302		LOG("Module %s not found.\n", filename);
303		return err;
304	}
305
306	err = kmod_module_get_versions(mod, &list);
307	if (err < 0) {
308		LOG("could not get modversions of %s: %s\n",
309			filename, strerror(-err));
310		kmod_module_unref(mod);
311		return err;
312	}
313
314	kmod_list_foreach(l, list) {
315		const char *symbol = kmod_module_version_get_symbol(l);
316		uint64_t crc = kmod_module_version_get_crc(l);
317		printf("0x%08"PRIx64"\t%s\n", crc, symbol);
318	}
319	kmod_module_versions_free_list(list);
320	kmod_module_unref(mod);
321	return 0;
322}
323
324static int command_do(struct kmod_module *module, const char *type,
325				const char *command, const char *cmdline_opts)
326{
327	const char *modname = kmod_module_get_name(module);
328	char *p, *cmd = NULL;
329	size_t cmdlen, cmdline_opts_len, varlen;
330	int ret = 0;
331
332	if (cmdline_opts == NULL)
333		cmdline_opts = "";
334	cmdline_opts_len = strlen(cmdline_opts);
335
336	cmd = strdup(command);
337	if (cmd == NULL)
338		return -ENOMEM;
339	cmdlen = strlen(cmd);
340	varlen = sizeof("$CMDLINE_OPTS") - 1;
341	while ((p = strstr(cmd, "$CMDLINE_OPTS")) != NULL) {
342		size_t prefixlen = p - cmd;
343		size_t suffixlen = cmdlen - prefixlen - varlen;
344		size_t slen = cmdlen - varlen + cmdline_opts_len;
345		char *suffix = p + varlen;
346		char *s = malloc(slen + 1);
347		if (s == NULL) {
348			free(cmd);
349			return -ENOMEM;
350		}
351		memcpy(s, cmd, p - cmd);
352		memcpy(s + prefixlen, cmdline_opts, cmdline_opts_len);
353		memcpy(s + prefixlen + cmdline_opts_len, suffix, suffixlen);
354		s[slen] = '\0';
355
356		free(cmd);
357		cmd = s;
358		cmdlen = slen;
359	}
360
361	SHOW("%s %s\n", type, cmd);
362	if (dry_run)
363		goto end;
364
365	setenv("MODPROBE_MODULE", modname, 1);
366	ret = system(cmd);
367	unsetenv("MODPROBE_MODULE");
368	if (ret == -1 || WEXITSTATUS(ret)) {
369		LOG("Error running %s command for %s\n", type, modname);
370		if (ret != -1)
371			ret = -WEXITSTATUS(ret);
372	}
373
374end:
375	free(cmd);
376	return ret;
377}
378
379static int rmmod_do_remove_module(struct kmod_module *mod)
380{
381	const char *modname = kmod_module_get_name(mod);
382	struct kmod_list *deps, *itr;
383	int flags = 0, err;
384
385	SHOW("rmmod %s\n", kmod_module_get_name(mod));
386
387	if (dry_run)
388		return 0;
389
390	if (force)
391		flags |= KMOD_REMOVE_FORCE;
392
393	err = kmod_module_remove_module(mod, flags);
394	if (err == -EEXIST) {
395		if (!first_time)
396			err = 0;
397		else
398			LOG("Module %s is not in kernel.\n", modname);
399	}
400
401	deps = kmod_module_get_dependencies(mod);
402	if (deps != NULL) {
403		kmod_list_foreach(itr, deps) {
404			struct kmod_module *dep = kmod_module_get_module(itr);
405			if (kmod_module_get_refcnt(dep) == 0)
406				rmmod_do_remove_module(dep);
407			kmod_module_unref(dep);
408		}
409		kmod_module_unref_list(deps);
410	}
411
412	return err;
413}
414
415static int rmmod_do_module(struct kmod_module *mod, bool do_dependencies);
416
417static int rmmod_do_deps_list(struct kmod_list *list, bool stop_on_errors)
418{
419	struct kmod_list *l;
420
421	kmod_list_foreach_reverse(l, list) {
422		struct kmod_module *m = kmod_module_get_module(l);
423		int r = rmmod_do_module(m, false);
424		kmod_module_unref(m);
425
426		if (r < 0 && stop_on_errors)
427			return r;
428	}
429
430	return 0;
431}
432
433static int rmmod_do_module(struct kmod_module *mod, bool do_dependencies)
434{
435	const char *modname = kmod_module_get_name(mod);
436	struct kmod_list *pre = NULL, *post = NULL;
437	const char *cmd = NULL;
438	int err;
439
440	if (!ignore_commands) {
441		err = kmod_module_get_softdeps(mod, &pre, &post);
442		if (err < 0) {
443			WRN("could not get softdeps of '%s': %s\n",
444						modname, strerror(-err));
445			return err;
446		}
447
448		cmd = kmod_module_get_remove_commands(mod);
449	}
450
451	if (cmd == NULL && !ignore_loaded) {
452		int state = kmod_module_get_initstate(mod);
453
454		if (state < 0) {
455			if (first_time) {
456				LOG("Module %s is not in kernel.\n", modname);
457				err = -ENOENT;
458			} else {
459				err = 0;
460			}
461			goto error;
462		} else if (state == KMOD_MODULE_BUILTIN) {
463			LOG("Module %s is builtin.\n", modname);
464			err = -ENOENT;
465			goto error;
466		}
467	}
468
469	rmmod_do_deps_list(post, false);
470
471	if (do_dependencies && remove_dependencies) {
472		struct kmod_list *deps = kmod_module_get_dependencies(mod);
473
474		err = rmmod_do_deps_list(deps, true);
475		if (err < 0)
476			goto error;
477	}
478
479	if (!ignore_loaded) {
480		int usage = kmod_module_get_refcnt(mod);
481
482		if (usage > 0) {
483			if (!quiet_inuse)
484				LOG("Module %s is in use.\n", modname);
485
486			err = -EBUSY;
487			goto error;
488		}
489	}
490
491	if (cmd == NULL)
492		err = rmmod_do_remove_module(mod);
493	else
494		err = command_do(mod, "remove", cmd, NULL);
495
496	if (err < 0)
497		goto error;
498
499	rmmod_do_deps_list(pre, false);
500
501error:
502	kmod_module_unref_list(pre);
503	kmod_module_unref_list(post);
504
505	return err;
506}
507
508static int rmmod(struct kmod_ctx *ctx, const char *alias)
509{
510	struct kmod_list *l, *list = NULL;
511	int err;
512
513	err = kmod_module_new_from_lookup(ctx, alias, &list);
514	if (err < 0)
515		return err;
516
517	if (list == NULL) {
518		LOG("Module %s not found.\n", alias);
519		err = -ENOENT;
520	}
521
522	kmod_list_foreach(l, list) {
523		struct kmod_module *mod = kmod_module_get_module(l);
524		err = rmmod_do_module(mod, true);
525		kmod_module_unref(mod);
526		if (err < 0)
527			break;
528	}
529
530	kmod_module_unref_list(list);
531	return err;
532}
533
534static int rmmod_all(struct kmod_ctx *ctx, char **args, int nargs)
535{
536	int i, err = 0;
537
538	for (i = 0; i < nargs; i++) {
539		int r = rmmod(ctx, args[i]);
540		if (r < 0)
541			err = r;
542	}
543
544	return err;
545}
546
547static int handle_failed_lookup(struct kmod_ctx *ctx, const char *alias)
548{
549	struct kmod_module *mod;
550	int state, err;
551
552	DBG("lookup failed - trying to check if it's builtin\n");
553
554	err = kmod_module_new_from_name(ctx, alias, &mod);
555	if (err < 0)
556		return err;
557
558	state = kmod_module_get_initstate(mod);
559	kmod_module_unref(mod);
560
561	if (state != KMOD_MODULE_BUILTIN) {
562		LOG("Module %s not found.\n", alias);
563		return -ENOENT;
564	}
565
566	if (first_time) {
567		LOG("Module %s already in kernel (builtin).\n", alias);
568		return -ENOENT;
569	}
570
571	SHOW("builtin %s\n", alias);
572	return 0;
573}
574
575static void print_action(struct kmod_module *m, bool install,
576							const char *options)
577{
578	const char *path;
579
580	if (install) {
581		printf("install %s %s\n", kmod_module_get_install_commands(m),
582								options);
583		return;
584	}
585
586	path = kmod_module_get_path(m);
587
588	if (path == NULL) {
589		assert(kmod_module_get_initstate(m) == KMOD_MODULE_BUILTIN);
590		printf("builtin %s\n", kmod_module_get_name(m));
591	} else
592		printf("insmod %s %s\n", kmod_module_get_path(m), options);
593}
594
595static int insmod(struct kmod_ctx *ctx, const char *alias,
596						const char *extra_options)
597{
598	struct kmod_list *l, *list = NULL;
599	int err, flags = 0;
600
601	void (*show)(struct kmod_module *m, bool install,
602						const char *options) = NULL;
603
604	err = kmod_module_new_from_lookup(ctx, alias, &list);
605	if (err < 0)
606		return err;
607
608	if (list == NULL)
609		return handle_failed_lookup(ctx, alias);
610
611	if (strip_modversion || force)
612		flags |= KMOD_PROBE_FORCE_MODVERSION;
613	if (strip_vermagic || force)
614		flags |= KMOD_PROBE_FORCE_VERMAGIC;
615	if (ignore_commands)
616		flags |= KMOD_PROBE_IGNORE_COMMAND;
617	if (ignore_loaded)
618		flags |= KMOD_PROBE_IGNORE_LOADED;
619	if (dry_run)
620		flags |= KMOD_PROBE_DRY_RUN;
621	if (do_show || verbose > DEFAULT_VERBOSE)
622		show = &print_action;
623
624	flags |= KMOD_PROBE_APPLY_BLACKLIST_ALIAS_ONLY;
625
626	if (use_blacklist)
627		flags |= KMOD_PROBE_APPLY_BLACKLIST;
628	if (first_time)
629		flags |= KMOD_PROBE_FAIL_ON_LOADED;
630
631	kmod_list_foreach(l, list) {
632		struct kmod_module *mod = kmod_module_get_module(l);
633
634		if (lookup_only)
635			printf("%s\n", kmod_module_get_name(mod));
636		else {
637			err = kmod_module_probe_insert_module(mod, flags,
638					extra_options, NULL, NULL, show);
639		}
640
641		if (err >= 0)
642			/* ignore flag return values such as a mod being blacklisted */
643			err = 0;
644		else {
645			switch (err) {
646			case -EEXIST:
647				ERR("could not insert '%s': Module already in kernel\n",
648							kmod_module_get_name(mod));
649				break;
650			case -ENOENT:
651				ERR("could not insert '%s': Unknown symbol in module, "
652						"or unknown parameter (see dmesg)\n",
653						kmod_module_get_name(mod));
654				break;
655			default:
656				ERR("could not insert '%s': %s\n",
657						kmod_module_get_name(mod),
658						strerror(-err));
659				break;
660			}
661		}
662
663		kmod_module_unref(mod);
664	}
665
666	kmod_module_unref_list(list);
667	return err;
668}
669
670static int insmod_all(struct kmod_ctx *ctx, char **args, int nargs)
671{
672	int i, err = 0;
673
674	for (i = 0; i < nargs; i++) {
675		int r = insmod(ctx, args[i], NULL);
676		if (r < 0)
677			err = r;
678	}
679
680	return err;
681}
682
683static void env_modprobe_options_append(const char *value)
684{
685	const char *old = getenv("MODPROBE_OPTIONS");
686	char *env;
687
688	if (old == NULL) {
689		setenv("MODPROBE_OPTIONS", value, 1);
690		return;
691	}
692
693	if (asprintf(&env, "%s %s", old, value) < 0) {
694		ERR("could not append value to $MODPROBE_OPTIONS\n");
695		return;
696	}
697
698	if (setenv("MODPROBE_OPTIONS", env, 1) < 0)
699		ERR("could not setenv(MODPROBE_OPTIONS, \"%s\")\n", env);
700	free(env);
701}
702
703static int options_from_array(char **args, int nargs, char **output)
704{
705	char *opts = NULL;
706	size_t optslen = 0;
707	int i, err = 0;
708
709	for (i = 1; i < nargs; i++) {
710		size_t len = strlen(args[i]);
711		size_t qlen = 0;
712		const char *value;
713		void *tmp;
714
715		value = strchr(args[i], '=');
716		if (value) {
717			value++;
718			if (*value != '"' && *value != '\'') {
719				if (strchr(value, ' '))
720					qlen = 2;
721			}
722		}
723
724		tmp = realloc(opts, optslen + len + qlen + 2);
725		if (!tmp) {
726			err = -errno;
727			free(opts);
728			opts = NULL;
729			ERR("could not gather module options: out-of-memory\n");
730			break;
731		}
732		opts = tmp;
733		if (optslen > 0) {
734			opts[optslen] = ' ';
735			optslen++;
736		}
737		if (qlen == 0) {
738			memcpy(opts + optslen, args[i], len + 1);
739			optslen += len;
740		} else {
741			size_t keylen = value - args[i];
742			size_t valuelen = len - keylen;
743			memcpy(opts + optslen, args[i], keylen);
744			optslen += keylen;
745			opts[optslen] = '"';
746			optslen++;
747			memcpy(opts + optslen, value, valuelen);
748			optslen += valuelen;
749			opts[optslen] = '"';
750			optslen++;
751			opts[optslen] = '\0';
752		}
753	}
754
755	*output = opts;
756	return err;
757}
758
759static char **prepend_options_from_env(int *p_argc, char **orig_argv)
760{
761	const char *p, *env = getenv("MODPROBE_OPTIONS");
762	char **new_argv, *str_start, *str_end, *str, *s, *quote;
763	int i, argc = *p_argc;
764	size_t envlen, space_count = 0;
765
766	if (env == NULL)
767		return orig_argv;
768
769	for (p = env; *p != '\0'; p++) {
770		if (*p == ' ')
771			space_count++;
772	}
773
774	envlen = p - env;
775	new_argv = malloc(sizeof(char *) * (argc + space_count + 3 + envlen));
776	if (new_argv == NULL)
777		return NULL;
778
779	new_argv[0] = orig_argv[0];
780	str_start = str = (char *) (new_argv + argc + space_count + 3);
781	memcpy(str, env, envlen + 1);
782
783	str_end = str_start + envlen;
784
785	quote = NULL;
786	for (i = 1, s = str; *s != '\0'; s++) {
787		if (quote == NULL) {
788			if (*s == ' ') {
789				new_argv[i] = str;
790				i++;
791				*s = '\0';
792				str = s + 1;
793			} else if (*s == '"' || *s == '\'')
794				quote = s;
795		} else {
796			if (*s == *quote) {
797				if (quote == str) {
798					new_argv[i] = str + 1;
799					i++;
800					*s = '\0';
801					str = s + 1;
802				} else {
803					char *it;
804					for (it = quote; it < s - 1; it++)
805						it[0] = it[1];
806					for (it = s - 1; it < str_end - 2; it++)
807						it[0] = it[2];
808					str_end -= 2;
809					*str_end = '\0';
810					s -= 2;
811				}
812				quote = NULL;
813			}
814		}
815	}
816	if (str < s) {
817		new_argv[i] = str;
818		i++;
819	}
820
821	memcpy(new_argv + i, orig_argv + 1, sizeof(char *) * (argc - 1));
822	new_argv[i + argc] = NULL;
823	*p_argc = i + argc - 1;
824
825	return new_argv;
826}
827
828static int do_modprobe(int argc, char **orig_argv)
829{
830	struct kmod_ctx *ctx;
831	char **args = NULL, **argv;
832	const char **config_paths = NULL;
833	int nargs = 0, n_config_paths = 0;
834	char dirname_buf[PATH_MAX];
835	const char *dirname = NULL;
836	const char *root = NULL;
837	const char *kversion = NULL;
838	int use_all = 0;
839	int do_remove = 0;
840	int do_show_config = 0;
841	int do_show_modversions = 0;
842	int err;
843
844	argv = prepend_options_from_env(&argc, orig_argv);
845	if (argv == NULL) {
846		fputs("Error: could not prepend options from command line\n",
847			stderr);
848		return EXIT_FAILURE;
849	}
850
851	for (;;) {
852		int c, idx = 0;
853		c = getopt_long(argc, argv, cmdopts_s, cmdopts, &idx);
854		if (c == -1)
855			break;
856		switch (c) {
857		case 'a':
858			log_priority = LOG_WARNING;
859			use_all = 1;
860			break;
861		case 'r':
862			do_remove = 1;
863			break;
864		case 5:
865			remove_dependencies = 1;
866			break;
867		case 'R':
868			lookup_only = 1;
869			break;
870		case 3:
871			first_time = 1;
872			break;
873		case 'i':
874			ignore_commands = 1;
875			break;
876		case 'b':
877			use_blacklist = 1;
878			break;
879		case 'f':
880			force = 1;
881			break;
882		case 2:
883			strip_modversion = 1;
884			break;
885		case 1:
886			strip_vermagic = 1;
887			break;
888		case 'D':
889			ignore_loaded = 1;
890			dry_run = 1;
891			do_show = 1;
892			break;
893		case 'c':
894			do_show_config = 1;
895			break;
896		case 4:
897			do_show_modversions = 1;
898			break;
899		case 'n':
900			dry_run = 1;
901			break;
902		case 'C': {
903			size_t bytes = sizeof(char *) * (n_config_paths + 2);
904			void *tmp = realloc(config_paths, bytes);
905			if (!tmp) {
906				fputs("Error: out-of-memory\n", stderr);
907				goto cmdline_failed;
908			}
909			config_paths = tmp;
910			config_paths[n_config_paths] = optarg;
911			n_config_paths++;
912			config_paths[n_config_paths] = NULL;
913
914			env_modprobe_options_append("-C");
915			env_modprobe_options_append(optarg);
916			break;
917		}
918		case 'd':
919			root = optarg;
920			break;
921		case 'S':
922			kversion = optarg;
923			break;
924		case 's':
925			env_modprobe_options_append("-s");
926			use_syslog = 1;
927			break;
928		case 'q':
929			env_modprobe_options_append("-q");
930			verbose = LOG_EMERG;
931			break;
932		case 'v':
933			env_modprobe_options_append("-v");
934			verbose++;
935			break;
936		case 'V':
937			puts(PACKAGE " version " VERSION);
938			err = EXIT_SUCCESS;
939			goto done;
940		case 'h':
941			help(basename(argv[0]));
942			err = EXIT_SUCCESS;
943			goto done;
944		case '?':
945			goto cmdline_failed;
946		default:
947			fprintf(stderr, "Error: unexpected getopt_long() value '%c'.\n",
948									c);
949			goto cmdline_failed;
950		}
951	}
952
953	args = argv + optind;
954	nargs = argc - optind;
955
956	if (!do_show_config) {
957		if (nargs == 0) {
958			fputs("Error: missing parameters. See -h.\n", stderr);
959			goto cmdline_failed;
960		}
961	}
962
963	if (root != NULL || kversion != NULL) {
964		struct utsname u;
965		if (root == NULL)
966			root = "";
967		if (kversion == NULL) {
968			if (uname(&u) < 0) {
969				fprintf(stderr, "Error: uname() failed: %s\n",
970					strerror(errno));
971				goto cmdline_failed;
972			}
973			kversion = u.release;
974		}
975		snprintf(dirname_buf, sizeof(dirname_buf),
976				"%s/lib/modules/%s", root,
977				kversion);
978		dirname = dirname_buf;
979	}
980
981	ctx = kmod_new(dirname, config_paths);
982	if (!ctx) {
983		fputs("Error: kmod_new() failed!\n", stderr);
984		goto cmdline_failed;
985	}
986	kmod_load_resources(ctx);
987
988	kmod_set_log_priority(ctx, verbose);
989	kmod_set_log_fn(ctx, log_modprobe, NULL);
990	if (use_syslog)
991		openlog("modprobe", LOG_CONS, LOG_DAEMON);
992
993	if (do_show_config)
994		err = show_config(ctx);
995	else if (do_show_modversions)
996		err = show_modversions(ctx, args[0]);
997	else if (do_remove)
998		err = rmmod_all(ctx, args, nargs);
999	else if (use_all)
1000		err = insmod_all(ctx, args, nargs);
1001	else {
1002		char *opts;
1003		err = options_from_array(args, nargs, &opts);
1004		if (err == 0) {
1005			err = insmod(ctx, args[0], opts);
1006			free(opts);
1007		}
1008	}
1009
1010	kmod_unref(ctx);
1011
1012	if (use_syslog)
1013		closelog();
1014
1015done:
1016	if (argv != orig_argv)
1017		free(argv);
1018
1019	free(config_paths);
1020	return err >= 0 ? EXIT_SUCCESS : EXIT_FAILURE;
1021
1022cmdline_failed:
1023	if (argv != orig_argv)
1024		free(argv);
1025	free(config_paths);
1026	return EXIT_FAILURE;
1027}
1028
1029#include "kmod.h"
1030
1031const struct kmod_cmd kmod_cmd_compat_modprobe = {
1032	.name = "modprobe",
1033	.cmd = do_modprobe,
1034	.help = "compat modprobe command",
1035};
1036