Set the close-on-exec flag, just to be saf
[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",
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");
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         printf("\n");
131
132         for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
133                 if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
134                         if (errno == ENOENT) {
135                                 /* The module was removed out from underneath us. */
136                                 continue;
137                         }
138                         /* else choke */
139                         perror_msg_and_die("module %s: QM_INFO", mn);
140                 }
141                 if (my_query_module(mn, QM_REFS, (void **)&deps, &depsize, &count)) {
142                         if (errno == ENOENT) {
143                                 /* The module was removed out from underneath us. */
144                                 continue;
145                         }
146                         perror_msg_and_die("module %s: QM_REFS", mn);
147                 }
148                 printf("%-20s%8lu%4ld ", mn, info.size, info.usecount);
149                 if (info.flags & NEW_MOD_DELETED)
150                         printf("(deleted)");
151                 else if (info.flags & NEW_MOD_INITIALIZING)
152                         printf("(initializing)");
153                 else if (!(info.flags & NEW_MOD_RUNNING))
154                         printf("(uninitialized)");
155                 else {
156                         if (info.flags & NEW_MOD_AUTOCLEAN)
157                                 printf("(autoclean) ");
158                         if (!(info.flags & NEW_MOD_USED_ONCE))
159                                 printf("(unused)");
160                 }
161                 if (count) printf("[");
162                 for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
163                         printf("%s%s", dn, (j==count-1)? "":" ");
164                 }
165                 if (count) printf("] ");
166
167                 printf("\n");
168         }
169
170
171         return( 0);
172 }
173
174 #else /*CONFIG_FEATURE_OLD_MODULE_INTERFACE*/
175
176 extern int lsmod_main(int argc, char **argv)
177 {
178         int fd, i;
179         char line[128];
180
181         printf("Module                  Size  Used by");
182         check_tainted();
183         fflush(stdout);
184
185         if ((fd = open("/proc/modules", O_RDONLY)) >= 0 ) {
186                 while ((i = read(fd, line, sizeof(line))) > 0) {
187                         write(fileno(stdout), line, i);
188                 }
189                 close(fd);
190                 return 0;
191         }
192         perror_msg_and_die("/proc/modules");
193         return 1;
194 }
195
196 #endif /*CONFIG_FEATURE_OLD_MODULE_INTERFACE*/