modutils/rmmod.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini rmmod implementation for busybox
   4  *
   5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15  * General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  20  *
  21  */
  22
  23 #include <stdio.h>
  24 #include <errno.h>
  25 #include <unistd.h>
  26 #include <stdlib.h>
  27 #include <getopt.h>
  28 #include <fcntl.h>
  29 #include <string.h>
  30 #include <sys/utsname.h>
  31 #include <sys/syscall.h>
  32 #include "busybox.h"
  33
  34 #ifdef CONFIG_FEATURE_2_6_MODULES
  35 static inline void filename2modname(char *modname, const char *afterslash)
  36 {
  37         unsigned int i;
  38
  39 #if ENABLE_FEATURE_2_4_MODULES
  40         int kr_chk = 1;
  41         if (get_linux_version_code() <= KERNEL_VERSION(2,6,0))
  42                 kr_chk = 0;
  43 #else
  44 #define kr_chk 1
  45 #endif
  46
  47         /* Convert to underscores, stop at first . */
  48         for (i = 0; afterslash[i] && afterslash[i] != '.'; i++) {
  49                 if (kr_chk && (afterslash[i] == '-'))
  50                         modname[i] = '_';
  51                 else
  52                         modname[i] = afterslash[i];
  53         }
  54         modname[i] = '\0';
  55 }
  56 #endif
  57
  58 int rmmod_main(int argc, char **argv)
  59 {
  60         int n, ret = EXIT_SUCCESS;
  61         unsigned int flags = O_NONBLOCK|O_EXCL;
  62 #ifdef CONFIG_FEATURE_QUERY_MODULE_INTERFACE
  63         /* bb_common_bufsiz1 hold the module names which we ignore
  64            but must get */
  65         size_t bufsize = sizeof(bb_common_bufsiz1);
  66 #endif
  67
  68         /* Parse command line. */
  69         n = bb_getopt_ulflags(argc, argv, "wfa");
  70         if((n & 1))     // --wait
  71                 flags &= ~O_NONBLOCK;
  72         if((n & 2))     // --force
  73                 flags |= O_TRUNC;
  74         if((n & 4)) {
  75                 /* Unload _all_ unused modules via NULL delete_module() call */
  76                 /* until the number of modules does not change */
  77                 size_t nmod = 0; /* number of modules */
  78                 size_t pnmod = -1; /* previous number of modules */
  79
  80                 while (nmod != pnmod) {
  81                         if (syscall(__NR_delete_module, NULL, flags) != 0) {
  82                                 if (errno==EFAULT)
  83                                         return(ret);
  84                                 bb_perror_msg_and_die("rmmod");
  85                         }
  86                         pnmod = nmod;
  87 #ifdef CONFIG_FEATURE_QUERY_MODULE_INTERFACE
  88                         /* 1 == QM_MODULES */
  89                         if (my_query_module(NULL, 1, &bb_common_bufsiz1, &bufsize, &nmod)) {
  90                                 bb_perror_msg_and_die("QM_MODULES");
  91                         }
  92 #endif
  93                 }
  94                 return EXIT_SUCCESS;
  95         }
  96
  97         if (optind == argc)
  98                 bb_show_usage();
  99
 100         for (n = optind; n < argc; n++) {
 101 #ifdef CONFIG_FEATURE_2_6_MODULES
 102                 const char *afterslash;
 103                 char *module_name;
 104
 105                 afterslash = strrchr(argv[n], '/');
 106                 if (!afterslash)
 107                         afterslash = argv[n];
 108                 else
 109                         afterslash++;
 110                 module_name = alloca(strlen(afterslash) + 1);
 111                 filename2modname(module_name, afterslash);
 112 #else
 113 #define module_name             argv[n]
 114 #endif
 115                 if (syscall(__NR_delete_module, module_name, flags) != 0) {
 116                         bb_perror_msg("%s", argv[n]);
 117                         ret = EXIT_FAILURE;
 118                 }
 119         }
 120
 121         return(ret);
 122 }