tools/modprobe.c

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