tools/modprobe.c

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