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