a2a582389ca0860d23a5a2c2416655f30e452fc0
[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,2000 by Lineo, inc. and Erik Andersen
6  * Copyright (C) 1999,2000,2001 by Erik Andersen <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 #define TAINT_FILENAME                  "/proc/sys/kernel/tainted"
45 #define TAINT_PROPRIETORY_MODULE        (1<<0)
46 #define TAINT_FORCED_MODULE             (1<<1)
47 #define TAINT_UNSAFE_SMP                (1<<2)
48
49 void check_tainted(void) 
50 {
51         int tainted;
52         FILE *f;
53
54         tainted = 0;
55         if ((f = fopen(TAINT_FILENAME, "r"))) {
56                 fscanf(f, "%d", &tainted);
57                 fclose(f);
58         }
59         if (f && tainted) {
60                 printf("    Tainted: %c%c%c\n",
61                                 tainted & TAINT_PROPRIETORY_MODULE      ? 'P' : 'G',
62                                 tainted & TAINT_FORCED_MODULE           ? 'F' : ' ',
63                                 tainted & TAINT_UNSAFE_SMP              ? 'S' : ' ');
64         }
65         else {
66                 printf("    Not tainted\n");
67         }
68 }
69
70 #ifdef CONFIG_FEATURE_QUERY_MODULE_INTERFACE
71
72 struct module_info
73 {
74         unsigned long addr;
75         unsigned long size;
76         unsigned long flags;
77         long usecount;
78 };
79
80
81 int query_module(const char *name, int which, void *buf, size_t bufsize, size_t *ret);
82
83 /* Values for query_module's which.  */
84 static const int QM_MODULES = 1;
85 static const int QM_DEPS = 2;
86 static const int QM_REFS = 3;
87 static const int QM_SYMBOLS = 4;
88 static const int QM_INFO = 5;
89
90 /* Bits of module.flags.  */
91 static const int NEW_MOD_RUNNING = 1;
92 static const int NEW_MOD_DELETED = 2;
93 static const int NEW_MOD_AUTOCLEAN = 4;
94 static const int NEW_MOD_VISITED = 8;
95 static const int NEW_MOD_USED_ONCE = 16;
96 static const int NEW_MOD_INITIALIZING = 64;
97
98 static int my_query_module(const char *name, int which, void **buf,
99                 size_t *bufsize, size_t *ret)
100 {
101         int my_ret;
102
103         my_ret = query_module(name, which, *buf, *bufsize, ret);
104
105         if (my_ret == -1 && errno == ENOSPC) {
106                 *buf = xrealloc(*buf, *ret);
107                 *bufsize = *ret;
108
109                 my_ret = query_module(name, which, *buf, *bufsize, ret);
110         }
111
112         return my_ret;
113 }
114
115 extern int lsmod_main(int argc, char **argv)
116 {
117         struct module_info info;
118         char *module_names, *mn, *deps, *dn;
119         size_t bufsize, depsize, nmod, count, i, j;
120
121         module_names = xmalloc(bufsize = 256);
122         if (my_query_module(NULL, QM_MODULES, (void **)&module_names, &bufsize,
123                                 &nmod)) {
124                 perror_msg_and_die("QM_MODULES");
125         }
126
127         deps = xmalloc(depsize = 256);
128         printf("Module                  Size  Used by");
129         check_tainted();
130
131         for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
132                 if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
133                         if (errno == ENOENT) {
134                                 /* The module was removed out from underneath us. */
135                                 continue;
136                         }
137                         /* else choke */
138                         perror_msg_and_die("module %s: QM_INFO", mn);
139                 }
140                 if (my_query_module(mn, QM_REFS, (void **)&deps, &depsize, &count)) {
141                         if (errno == ENOENT) {
142                                 /* The module was removed out from underneath us. */
143                                 continue;
144                         }
145                         perror_msg_and_die("module %s: QM_REFS", mn);
146                 }
147                 printf("%-20s%8lu%4ld ", mn, info.size, info.usecount);
148                 if (info.flags & NEW_MOD_DELETED)
149                         printf("(deleted)");
150                 else if (info.flags & NEW_MOD_INITIALIZING)
151                         printf("(initializing)");
152                 else if (!(info.flags & NEW_MOD_RUNNING))
153                         printf("(uninitialized)");
154                 else {
155                         if (info.flags & NEW_MOD_AUTOCLEAN)
156                                 printf("(autoclean) ");
157                         if (!(info.flags & NEW_MOD_USED_ONCE))
158                                 printf("(unused)");
159                 }
160                 if (count) printf("[");
161                 for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
162                         printf("%s%s", dn, (j==count-1)? "":" ");
163                 }
164                 if (count) printf("] ");
165
166                 printf("\n");
167         }
168
169
170         return( 0);
171 }
172
173 #else /*CONFIG_FEATURE_OLD_MODULE_INTERFACE*/
174
175 extern int lsmod_main(int argc, char **argv)
176 {
177         int fd, i;
178         char line[128];
179
180         printf("Module                  Size  Used by");
181         check_tainted();
182         fflush(stdout);
183
184         if ((fd = open("/proc/modules", O_RDONLY)) >= 0 ) {
185                 while ((i = read(fd, line, sizeof(line))) > 0) {
186                         write(fileno(stdout), line, i);
187                 }
188                 close(fd);
189                 return 0;
190         }
191         perror_msg_and_die("/proc/modules");
192         return 1;
193 }
194
195 #endif /*CONFIG_FEATURE_OLD_MODULE_INTERFACE*/