Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 2 | /* |
| 3 | * Mini rmmod implementation for busybox |
| 4 | * |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame^] | 5 | * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org> |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 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 | |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 23 | #include <stdio.h> |
| 24 | #include <errno.h> |
| 25 | #include <unistd.h> |
Eric Andersen | ed3ef50 | 2001-01-27 08:24:39 +0000 | [diff] [blame] | 26 | #include <stdlib.h> |
Mark Whitley | f90c28d | 2001-03-09 21:49:12 +0000 | [diff] [blame] | 27 | #include <getopt.h> |
Eric Andersen | e704788 | 2003-12-11 01:42:13 +0000 | [diff] [blame] | 28 | #include <fcntl.h> |
| 29 | #include <sys/syscall.h> |
Eric Andersen | cbe31da | 2001-02-20 06:14:08 +0000 | [diff] [blame] | 30 | #include "busybox.h" |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 31 | |
Eric Andersen | e76c3b0 | 2001-04-05 03:14:39 +0000 | [diff] [blame] | 32 | extern int delete_module(const char * name); |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 33 | |
| 34 | |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 35 | extern int rmmod_main(int argc, char **argv) |
| 36 | { |
Mark Whitley | f90c28d | 2001-03-09 21:49:12 +0000 | [diff] [blame] | 37 | int n, ret = EXIT_SUCCESS; |
Tim Riker | cf93274 | 2002-12-14 01:58:59 +0000 | [diff] [blame] | 38 | size_t nmod = 0; /* number of modules */ |
| 39 | size_t pnmod = -1; /* previous number of modules */ |
| 40 | void *buf; /* hold the module names which we ignore but must get */ |
| 41 | size_t bufsize = 0; |
Eric Andersen | e704788 | 2003-12-11 01:42:13 +0000 | [diff] [blame] | 42 | unsigned int flags = O_NONBLOCK|O_EXCL; |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 43 | |
Mark Whitley | f90c28d | 2001-03-09 21:49:12 +0000 | [diff] [blame] | 44 | /* Parse command line. */ |
| 45 | while ((n = getopt(argc, argv, "a")) != EOF) { |
| 46 | switch (n) { |
Eric Andersen | e704788 | 2003-12-11 01:42:13 +0000 | [diff] [blame] | 47 | case 'w': // --wait |
| 48 | flags &= ~O_NONBLOCK; |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame^] | 49 | break; |
Eric Andersen | e704788 | 2003-12-11 01:42:13 +0000 | [diff] [blame] | 50 | case 'f': // --force |
| 51 | flags |= O_TRUNC; |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame^] | 52 | break; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 53 | case 'a': |
| 54 | /* Unload _all_ unused modules via NULL delete_module() call */ |
Tim Riker | cf93274 | 2002-12-14 01:58:59 +0000 | [diff] [blame] | 55 | /* until the number of modules does not change */ |
| 56 | buf = xmalloc(bufsize = 256); |
| 57 | while (nmod != pnmod) { |
| 58 | if (delete_module(NULL)) |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 59 | bb_perror_msg_and_die("rmmod"); |
Tim Riker | cf93274 | 2002-12-14 01:58:59 +0000 | [diff] [blame] | 60 | pnmod = nmod; |
| 61 | /* 1 == QM_MODULES */ |
| 62 | if (my_query_module(NULL, 1, &buf, &bufsize, &nmod)) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 63 | bb_perror_msg_and_die("QM_MODULES"); |
Tim Riker | cf93274 | 2002-12-14 01:58:59 +0000 | [diff] [blame] | 64 | } |
| 65 | } |
| 66 | #ifdef CONFIG_FEATURE_CLEAN_UP |
| 67 | free(buf); |
| 68 | #endif |
Matt Kraai | 3e856ce | 2000-12-01 02:55:13 +0000 | [diff] [blame] | 69 | return EXIT_SUCCESS; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 70 | default: |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 71 | bb_show_usage(); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 72 | } |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 73 | } |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 74 | |
Mark Whitley | f90c28d | 2001-03-09 21:49:12 +0000 | [diff] [blame] | 75 | if (optind == argc) |
Eric Andersen | 3b1a744 | 2003-12-24 20:30:45 +0000 | [diff] [blame] | 76 | bb_show_usage(); |
Mark Whitley | f90c28d | 2001-03-09 21:49:12 +0000 | [diff] [blame] | 77 | |
| 78 | for (n = optind; n < argc; n++) { |
Eric Andersen | e704788 | 2003-12-11 01:42:13 +0000 | [diff] [blame] | 79 | if (syscall(__NR_delete_module, argv[n], flags) < 0) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 80 | bb_perror_msg("%s", argv[n]); |
Matt Kraai | 3e856ce | 2000-12-01 02:55:13 +0000 | [diff] [blame] | 81 | ret = EXIT_FAILURE; |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 82 | } |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 83 | } |
Mark Whitley | f90c28d | 2001-03-09 21:49:12 +0000 | [diff] [blame] | 84 | |
Eric Andersen | a5716d3 | 2000-07-28 15:16:37 +0000 | [diff] [blame] | 85 | return(ret); |
Erik Andersen | 3d7e341 | 1999-12-16 23:04:20 +0000 | [diff] [blame] | 86 | } |