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