modutils/lsmod.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini lsmod implementation for busybox
   4  *
   5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6  *
   7  * Modified by Alcove, Julien Gaulmin <julien.gaulmin@alcove.fr> and
   8  * Nicolas Ferre <nicolas.ferre@alcove.fr> to support pre 2.1 kernels
   9  * (which lack the query_module() interface).
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19  * General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program; if not, write to the Free Software
  23  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  24  *
  25  */
  26
  27 #include <stdlib.h>
  28 #include <stdio.h>
  29 #include <string.h>
  30 #include <stddef.h>
  31 #include <errno.h>
  32 #include <unistd.h>
  33 #include <dirent.h>
  34 #include <ctype.h>
  35 #include <assert.h>
  36 #include <getopt.h>
  37 #include <sys/utsname.h>
  38 #include <sys/file.h>
  39 #include "busybox.h"
  40
  41
  42 #ifndef CONFIG_FEATURE_CHECK_TAINTED_MODULE
  43 static inline void check_tainted(void) { printf("\n"); }
  44 #else
  45 #define TAINT_FILENAME                  "/proc/sys/kernel/tainted"
  46 #define TAINT_PROPRIETORY_MODULE        (1<<0)
  47 #define TAINT_FORCED_MODULE             (1<<1)
  48 #define TAINT_UNSAFE_SMP                (1<<2)
  49
  50 static void check_tainted(void)
  51 {
  52         int tainted;
  53         FILE *f;
  54
  55         tainted = 0;
  56         if ((f = fopen(TAINT_FILENAME, "r"))) {
  57                 fscanf(f, "%d", &tainted);
  58                 fclose(f);
  59         }
  60         if (f && tainted) {
  61                 printf("    Tainted: %c%c%c\n",
  62                                 tainted & TAINT_PROPRIETORY_MODULE      ? 'P' : 'G',
  63                                 tainted & TAINT_FORCED_MODULE           ? 'F' : ' ',
  64                                 tainted & TAINT_UNSAFE_SMP              ? 'S' : ' ');
  65         }
  66         else {
  67                 printf("    Not tainted\n");
  68         }
  69 }
  70 #endif
  71
  72 #ifdef CONFIG_FEATURE_QUERY_MODULE_INTERFACE
  73
  74 struct module_info
  75 {
  76         unsigned long addr;
  77         unsigned long size;
  78         unsigned long flags;
  79         long usecount;
  80 };
  81
  82
  83 int query_module(const char *name, int which, void *buf, size_t bufsize, size_t *ret);
  84
  85 enum {
  86 /* Values for query_module's which.  */
  87         QM_MODULES = 1,
  88         QM_DEPS = 2,
  89         QM_REFS = 3,
  90         QM_SYMBOLS = 4,
  91         QM_INFO = 5,
  92
  93 /* Bits of module.flags.  */
  94         NEW_MOD_RUNNING = 1,
  95         NEW_MOD_DELETED = 2,
  96         NEW_MOD_AUTOCLEAN = 4,
  97         NEW_MOD_VISITED = 8,
  98         NEW_MOD_USED_ONCE = 16,
  99         NEW_MOD_INITIALIZING = 64
 100 };
 101
 102 int lsmod_main(int argc, char **argv)
 103 {
 104         struct module_info info;
 105         char *module_names, *mn, *deps, *dn;
 106         size_t bufsize, depsize, nmod, count, i, j;
 107
 108         module_names = xmalloc(bufsize = 256);
 109         if (my_query_module(NULL, QM_MODULES, (void **)&module_names, &bufsize,
 110                                 &nmod)) {
 111                 bb_perror_msg_and_die("QM_MODULES");
 112         }
 113
 114         deps = xmalloc(depsize = 256);
 115         printf("Module                  Size  Used by");
 116         check_tainted();
 117
 118         for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
 119                 if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
 120                         if (errno == ENOENT) {
 121                                 /* The module was removed out from underneath us. */
 122                                 continue;
 123                         }
 124                         /* else choke */
 125                         bb_perror_msg_and_die("module %s: QM_INFO", mn);
 126                 }
 127                 if (my_query_module(mn, QM_REFS, (void **)&deps, &depsize, &count)) {
 128                         if (errno == ENOENT) {
 129                                 /* The module was removed out from underneath us. */
 130                                 continue;
 131                         }
 132                         bb_perror_msg_and_die("module %s: QM_REFS", mn);
 133                 }
 134                 printf("%-20s%8lu%4ld", mn, info.size, info.usecount);
 135                 if (info.flags & NEW_MOD_DELETED)
 136                         printf(" (deleted)");
 137                 else if (info.flags & NEW_MOD_INITIALIZING)
 138                         printf(" (initializing)");
 139                 else if (!(info.flags & NEW_MOD_RUNNING))
 140                         printf(" (uninitialized)");
 141                 else {
 142                         if (info.flags & NEW_MOD_AUTOCLEAN)
 143                                 printf(" (autoclean) ");
 144                         if (!(info.flags & NEW_MOD_USED_ONCE))
 145                                 printf(" (unused)");
 146                 }
 147                 if (count) printf(" [");
 148                 for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
 149                         printf("%s%s", dn, (j==count-1)? "":" ");
 150                 }
 151                 if (count) printf("]");
 152
 153                 printf("\n");
 154         }
 155
 156 #ifdef CONFIG_FEATURE_CLEAN_UP
 157         free(module_names);
 158 #endif
 159
 160         return( 0);
 161 }
 162
 163 #else /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */
 164
 165 int lsmod_main(int argc, char **argv)
 166 {
 167         printf("Module                  Size  Used by");
 168         check_tainted();
 169 #if defined(CONFIG_FEATURE_LSMOD_PRETTY_2_6_OUTPUT)
 170         {
 171           FILE *file;
 172           char line[4096];
 173
 174           file = bb_xfopen("/proc/modules", "r");
 175
 176           while (fgets(line, sizeof(line), file)) {
 177             char *tok;
 178
 179             tok = strtok(line, " \t");
 180             printf("%-19s", tok);
 181             tok = strtok(NULL, " \t\n");
 182             printf(" %8s", tok);
 183             tok = strtok(NULL, " \t\n");
 184             /* Null if no module unloading support. */
 185             if (tok) {
 186               printf("  %s", tok);
 187               tok = strtok(NULL, "\n");
 188               if (!tok)
 189                 tok = "";
 190               /* New-style has commas, or -.  If so,
 191                  truncate (other fields might follow). */
 192               else if (strchr(tok, ',')) {
 193                 tok = strtok(tok, "\t ");
 194                 /* Strip trailing comma. */
 195                 if (tok[strlen(tok)-1] == ',')
 196                   tok[strlen(tok)-1] = '\0';
 197               } else if (tok[0] == '-'
 198                          && (tok[1] == '\0' || isspace(tok[1])))
 199                 tok = "";
 200               printf(" %s", tok);
 201             }
 202             printf("\n");
 203           }
 204           fclose(file);
 205         }
 206         return 0;  /* Success  */
 207 #else
 208         if (bb_xprint_file_by_name("/proc/modules") < 0) {
 209                 return 0;
 210         }
 211 #endif  /*  CONFIG_FEATURE_2_6_MODULES  */
 212         return 1;
 213 }
 214
 215 #endif /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */