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