lsmod.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini lsmod implementation for busybox
   4  *
   5  * Copyright (C) 1999,2000,2001 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 <stdlib.h>
  29 #include <stdio.h>
  30 #include <string.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 #include "busybox.h"
  41
  42
  43
  44 #ifdef BB_FEATURE_NEW_MODULE_INTERFACE
  45
  46 struct module_info
  47 {
  48         unsigned long addr;
  49         unsigned long size;
  50         unsigned long flags;
  51         long usecount;
  52 };
  53
  54
  55 int query_module(const char *name, int which, void *buf, size_t bufsize,
  56                  size_t *ret);
  57
  58 /* Values for query_module's which.  */
  59 static const int QM_MODULES = 1;
  60 static const int QM_DEPS = 2;
  61 static const int QM_REFS = 3;
  62 static const int QM_SYMBOLS = 4;
  63 static const int QM_INFO = 5;
  64
  65 /* Bits of module.flags.  */
  66 static const int NEW_MOD_RUNNING = 1;
  67 static const int NEW_MOD_DELETED = 2;
  68 static const int NEW_MOD_AUTOCLEAN = 4;
  69 static const int NEW_MOD_VISITED = 8;
  70 static const int NEW_MOD_USED_ONCE = 16;
  71 static const int NEW_MOD_INITIALIZING = 64;
  72
  73
  74 extern int lsmod_main(int argc, char **argv)
  75 {
  76         struct module_info info;
  77         char *module_names, *mn, *deps, *dn;
  78         size_t bufsize, nmod, count, i, j;
  79
  80         module_names = xmalloc(bufsize = 256);
  81         deps = xmalloc(bufsize);
  82         if (query_module(NULL, QM_MODULES, module_names, bufsize, &nmod)) {
  83                 perror_msg_and_die("QM_MODULES");
  84         }
  85
  86         printf("Module                  Size  Used by\n");
  87         for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
  88                 if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
  89                         if (errno == ENOENT) {
  90                                 /* The module was removed out from underneath us. */
  91                                 continue;
  92                         }
  93                         /* else choke */
  94                         perror_msg_and_die("module %s: QM_INFO", mn);
  95                 }
  96                 while (query_module(mn, QM_REFS, deps, bufsize, &count)) {
  97                         if (errno == ENOENT) {
  98                                 /* The module was removed out from underneath us. */
  99                                 continue;
 100                         }
 101                         if (errno != ENOSPC) {
 102                                 error_msg_and_die("module %s: QM_REFS", mn);
 103                         }
 104                         deps = xrealloc(deps, bufsize = count);
 105                 }
 106                 printf("%-20s%8lu%4ld ", mn, info.size, info.usecount);
 107                 if (info.flags & NEW_MOD_DELETED)
 108                         printf("(deleted)");
 109                 else if (info.flags & NEW_MOD_INITIALIZING)
 110                         printf("(initializing)");
 111                 else if (!(info.flags & NEW_MOD_RUNNING))
 112                         printf("(uninitialized)");
 113                 else {
 114                         if (info.flags & NEW_MOD_AUTOCLEAN)
 115                                 printf("(autoclean) ");
 116                         if (!(info.flags & NEW_MOD_USED_ONCE))
 117                                 printf("(unused)");
 118                 }
 119                 if (count) printf("[");
 120                 for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
 121                         printf("%s%s", dn, (j==count-1)? "":" ");
 122                 }
 123                 if (count) printf("] ");
 124
 125                 printf("\n");
 126         }
 127
 128
 129         return( 0);
 130 }
 131
 132 #else /*BB_FEATURE_OLD_MODULE_INTERFACE*/
 133
 134 extern int lsmod_main(int argc, char **argv)
 135 {
 136         int fd, i;
 137         char line[128];
 138
 139         puts("Module                  Size  Used by");
 140         fflush(stdout);
 141
 142         if ((fd = open("/proc/modules", O_RDONLY)) >= 0 ) {
 143                 while ((i = read(fd, line, sizeof(line))) > 0) {
 144                         write(fileno(stdout), line, i);
 145                 }
 146                 close(fd);
 147                 return 0;
 148         }
 149         perror_msg_and_die("/proc/modules");
 150         return 1;
 151 }
 152
 153 #endif /*BB_FEATURE_OLD_MODULE_INTERFACE*/