lsmod.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini lsmod implementation for busybox
   4  *
   5  * Copyright (C) 1999,2000 by Lineo, inc.
   6  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
   7  *
   8  * Modified by Alcove, Julien Gaulmin <julien.gaulmin@alcove.fr> and
   9  * Nicolas Ferre <nicolas.ferre@alcove.fr> to support pre 2.1 kernels
  10  * (which lack the query_module() interface).
  11  *
  12  * This program is free software; you can redistribute it and/or modify
  13  * it under the terms of the GNU General Public License as published by
  14  * the Free Software Foundation; either version 2 of the License, or
  15  * (at your option) any later version.
  16  *
  17  * This program is distributed in the hope that it will be useful,
  18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  20  * General Public License for more details.
  21  *
  22  * You should have received a copy of the GNU General Public License
  23  * along with this program; if not, write to the Free Software
  24  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  25  *
  26  */
  27
  28 #include "busybox.h"
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include <stddef.h>
  32 #include <errno.h>
  33 #include <unistd.h>
  34 #include <dirent.h>
  35 #include <ctype.h>
  36 #include <assert.h>
  37 #include <getopt.h>
  38 #include <sys/utsname.h>
  39 #include <sys/file.h>
  40
  41
  42
  43 #if !defined(BB_FEATURE_LSMOD_NEW_KERNEL) && !defined(BB_FEATURE_LSMOD_OLD_KERNEL)
  44 #error "Must have ether BB_FEATURE_LSMOD_NEW_KERNEL or BB_FEATURE_LSMOD_OLD_KERNEL defined"
  45 #endif
  46
  47 #ifdef BB_FEATURE_LSMOD_NEW_KERNEL
  48
  49 struct module_info
  50 {
  51         unsigned long addr;
  52         unsigned long size;
  53         unsigned long flags;
  54         long usecount;
  55 };
  56
  57
  58 int query_module(const char *name, int which, void *buf, size_t bufsize,
  59                  size_t *ret);
  60
  61 /* Values for query_module's which.  */
  62 #define QM_MODULES      1
  63 #define QM_DEPS         2
  64 #define QM_REFS         3
  65 #define QM_SYMBOLS      4
  66 #define QM_INFO         5
  67
  68 /* Bits of module.flags.  */
  69 #define NEW_MOD_RUNNING         1
  70 #define NEW_MOD_DELETED         2
  71 #define NEW_MOD_AUTOCLEAN       4
  72 #define NEW_MOD_VISITED         8
  73 #define NEW_MOD_USED_ONCE       16
  74 #define NEW_MOD_INITIALIZING    64
  75
  76
  77 extern int lsmod_main(int argc, char **argv)
  78 {
  79         struct module_info info;
  80         char *module_names, *mn, *deps, *dn;
  81         size_t bufsize, nmod, count, i, j;
  82
  83         module_names = xmalloc(bufsize = 256);
  84         deps = xmalloc(bufsize);
  85         if (query_module(NULL, QM_MODULES, module_names, bufsize, &nmod)) {
  86                 perror_msg_and_die("QM_MODULES");
  87         }
  88
  89         printf("Module                  Size  Used by\n");
  90         for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
  91                 if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
  92                         if (errno == ENOENT) {
  93                                 /* The module was removed out from underneath us. */
  94                                 continue;
  95                         }
  96                         /* else choke */
  97                         perror_msg_and_die("module %s: QM_INFO", mn);
  98                 }
  99                 while (query_module(mn, QM_REFS, deps, bufsize, &count)) {
 100                         if (errno == ENOENT) {
 101                                 /* The module was removed out from underneath us. */
 102                                 continue;
 103                         }
 104                         if (errno != ENOSPC) {
 105                                 error_msg_and_die("module %s: QM_REFS", mn);
 106                         }
 107                         deps = xrealloc(deps, bufsize = count);
 108                 }
 109                 printf("%-20s%8lu%4ld ", mn, info.size, info.usecount);
 110                 if (count) printf("[");
 111                 for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
 112                         printf("%s%s", dn, (j==count-1)? "":" ");
 113                 }
 114                 if (count) printf("] ");
 115
 116                 if (info.flags & NEW_MOD_DELETED)
 117                         printf("(deleted)");
 118                 else if (info.flags & NEW_MOD_INITIALIZING)
 119                         printf("(initializing)");
 120                 else if (!(info.flags & NEW_MOD_RUNNING))
 121                         printf("(uninitialized)");
 122                 else {
 123                         if (info.flags & NEW_MOD_AUTOCLEAN)
 124                                 printf("(autoclean) ");
 125                         if (!(info.flags & NEW_MOD_USED_ONCE))
 126                                 printf("(unused)");
 127                 }
 128                 printf("\n");
 129         }
 130
 131
 132         return( 0);
 133 }
 134
 135 #else /*BB_FEATURE_LSMOD_OLD_KERNEL*/
 136
 137 #if ! defined BB_FEATURE_USE_PROCFS
 138 #error Sorry, I depend on the /proc filesystem right now.
 139 #endif
 140
 141 extern int lsmod_main(int argc, char **argv)
 142 {
 143         int fd, i;
 144         char line[128];
 145
 146         puts("Module                  Size  Used by");
 147         fflush(stdout);
 148
 149         if ((fd = open("/proc/modules", O_RDONLY)) >= 0 ) {
 150                 while ((i = read(fd, line, sizeof(line))) > 0) {
 151                         write(fileno(stdout), line, i);
 152                 }
 153                 close(fd);
 154                 return 0;
 155         }
 156         perror_msg_and_die("/proc/modules");
 157         return 1;
 158 }
 159
 160 #endif /*BB_FEATURE_LSMOD_OLD_KERNEL*/