makedevs: shrink by Vladimir
[oweals/busybox.git] / 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  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
12  */
13
14 #include "libbb.h"
15
16
17 #if !ENABLE_FEATURE_CHECK_TAINTED_MODULE
18 static void check_tainted(void) { bb_putchar('\n'); }
19 #else
20 #define TAINT_FILENAME                  "/proc/sys/kernel/tainted"
21 #define TAINT_PROPRIETORY_MODULE        (1<<0)
22 #define TAINT_FORCED_MODULE             (1<<1)
23 #define TAINT_UNSAFE_SMP                (1<<2)
24
25 static void check_tainted(void)
26 {
27         int tainted;
28         FILE *f;
29
30         tainted = 0;
31         f = fopen_for_read(TAINT_FILENAME);
32         if (f) {
33                 fscanf(f, "%d", &tainted);
34                 fclose(f);
35         }
36         if (tainted) {
37                 printf("    Tainted: %c%c%c\n",
38                                 tainted & TAINT_PROPRIETORY_MODULE      ? 'P' : 'G',
39                                 tainted & TAINT_FORCED_MODULE           ? 'F' : ' ',
40                                 tainted & TAINT_UNSAFE_SMP              ? 'S' : ' ');
41         } else {
42                 printf("    Not tainted\n");
43         }
44 }
45 #endif
46
47 #if ENABLE_FEATURE_QUERY_MODULE_INTERFACE
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, size_t *ret);
59
60 enum {
61 /* Values for query_module's which.  */
62         QM_MODULES = 1,
63         QM_DEPS = 2,
64         QM_REFS = 3,
65         QM_SYMBOLS = 4,
66         QM_INFO = 5,
67
68 /* Bits of module.flags.  */
69         NEW_MOD_RUNNING = 1,
70         NEW_MOD_DELETED = 2,
71         NEW_MOD_AUTOCLEAN = 4,
72         NEW_MOD_VISITED = 8,
73         NEW_MOD_USED_ONCE = 16,
74         NEW_MOD_INITIALIZING = 64
75 };
76
77 int lsmod_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
78 int lsmod_main(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
79 {
80         struct module_info info;
81         char *module_names, *mn, *deps, *dn;
82         size_t bufsize, depsize, nmod, count, i, j;
83
84         module_names = deps = NULL;
85         bufsize = depsize = 0;
86         while (query_module(NULL, QM_MODULES, module_names, bufsize, &nmod)) {
87                 if (errno != ENOSPC) bb_perror_msg_and_die("QM_MODULES");
88                 module_names = xmalloc(bufsize = nmod);
89         }
90
91         deps = xmalloc(depsize = 256);
92         printf("Module\t\t\tSize  Used by");
93         check_tainted();
94
95         for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
96                 if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
97                         if (errno == ENOENT) {
98                                 /* The module was removed out from underneath us. */
99                                 continue;
100                         }
101                         /* else choke */
102                         bb_perror_msg_and_die("module %s: QM_INFO", mn);
103                 }
104                 while (query_module(mn, QM_REFS, deps, depsize, &count)) {
105                         if (errno == ENOENT) {
106                                 /* The module was removed out from underneath us. */
107                                 continue;
108                         } else if (errno != ENOSPC)
109                                 bb_perror_msg_and_die("module %s: QM_REFS", mn);
110                         deps = xrealloc(deps, count);
111                 }
112                 printf("%-20s%8lu%4ld", mn, info.size, info.usecount);
113                 if (info.flags & NEW_MOD_DELETED)
114                         printf(" (deleted)");
115                 else if (info.flags & NEW_MOD_INITIALIZING)
116                         printf(" (initializing)");
117                 else if (!(info.flags & NEW_MOD_RUNNING))
118                         printf(" (uninitialized)");
119                 else {
120                         if (info.flags & NEW_MOD_AUTOCLEAN)
121                                 printf(" (autoclean) ");
122                         if (!(info.flags & NEW_MOD_USED_ONCE))
123                                 printf(" (unused)");
124                 }
125                 if (count)
126                         printf(" [");
127                 for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
128                         printf("%s%s", dn, (j==count-1)? "":" ");
129                 }
130                 if (count)
131                         bb_putchar(']');
132
133                 bb_putchar('\n');
134         }
135
136 #if ENABLE_FEATURE_CLEAN_UP
137         free(module_names);
138 #endif
139
140         return 0;
141 }
142
143 #else /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */
144
145 int lsmod_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
146 int lsmod_main(int argc UNUSED_PARAM, char **argv UNUSED_PARAM)
147 {
148         FILE *file = xfopen_for_read("/proc/modules");
149
150         printf("Module                  Size  Used by");
151         check_tainted();
152 #if ENABLE_FEATURE_LSMOD_PRETTY_2_6_OUTPUT
153         {
154                 char *line;
155                 while ((line = xmalloc_fgets(file)) != NULL) {
156                         char *tok;
157
158                         tok = strtok(line, " \t");
159                         printf("%-19s", tok);
160                         tok = strtok(NULL, " \t\n");
161                         printf(" %8s", tok);
162                         tok = strtok(NULL, " \t\n");
163                         /* Null if no module unloading support. */
164                         if (tok) {
165                                 printf("  %s", tok);
166                                 tok = strtok(NULL, "\n");
167                                 if (!tok)
168                                         tok = (char*)"";
169                                 /* New-style has commas, or -.  If so,
170                                 truncate (other fields might follow). */
171                                 else if (strchr(tok, ',')) {
172                                         tok = strtok(tok, "\t ");
173                                         /* Strip trailing comma. */
174                                         if (tok[strlen(tok)-1] == ',')
175                                                 tok[strlen(tok)-1] = '\0';
176                                 } else if (tok[0] == '-'
177                                  && (tok[1] == '\0' || isspace(tok[1]))
178                                 ) {
179                                         tok = (char*)"";
180                                 }
181                                 printf(" %s", tok);
182                         }
183                         bb_putchar('\n');
184                         free(line);
185                 }
186                 fclose(file);
187         }
188 #else
189         xprint_and_close_file(file);
190 #endif  /*  CONFIG_FEATURE_2_6_MODULES  */
191         return EXIT_SUCCESS;
192 }
193
194 #endif /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */