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