Fix an _old_ interface problem with a never-used feature.
[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,2001 by Lineo, inc.
6  * Written by Erik Andersen <andersen@lineo.com>, <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 #ifdef BB_FEATURE_NEW_MODULE_INTERFACE
45
46 struct module_info
47 {
48         unsigned long addr;
49         unsigned long size;
50         unsigned long flags;
51         long usecount;
52 };
53
54
55 int query_module(const char *name, int which, void *buf, size_t bufsize, size_t *ret);
56
57 /* Values for query_module's which.  */
58 static const int QM_MODULES = 1;
59 static const int QM_DEPS = 2;
60 static const int QM_REFS = 3;
61 static const int QM_SYMBOLS = 4;
62 static const int QM_INFO = 5;
63
64 /* Bits of module.flags.  */
65 static const int NEW_MOD_RUNNING = 1;
66 static const int NEW_MOD_DELETED = 2;
67 static const int NEW_MOD_AUTOCLEAN = 4;
68 static const int NEW_MOD_VISITED = 8;
69 static const int NEW_MOD_USED_ONCE = 16;
70 static const int NEW_MOD_INITIALIZING = 64;
71
72 static int my_query_module(const char *name, int which, void **buf,
73                 size_t *bufsize, size_t *ret)
74 {
75         int my_ret;
76
77         my_ret = query_module(name, which, *buf, *bufsize, ret);
78
79         if (my_ret == -1 && errno == ENOSPC) {
80                 *buf = xrealloc(*buf, *ret);
81                 *bufsize = *ret;
82
83                 my_ret = query_module(name, which, *buf, *bufsize, ret);
84         }
85
86         return my_ret;
87 }
88
89 extern int lsmod_main(int argc, char **argv)
90 {
91         struct module_info info;
92         char *module_names, *mn, *deps, *dn;
93         size_t bufsize, depsize, nmod, count, i, j;
94
95         module_names = xmalloc(bufsize = 256);
96         if (my_query_module(NULL, QM_MODULES, (void **)&module_names, &bufsize,
97                                 &nmod)) {
98                 perror_msg_and_die("QM_MODULES");
99         }
100
101         deps = xmalloc(depsize = 256);
102         printf("Module                  Size  Used by\n");
103         for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
104                 if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
105                         if (errno == ENOENT) {
106                                 /* The module was removed out from underneath us. */
107                                 continue;
108                         }
109                         /* else choke */
110                         perror_msg_and_die("module %s: QM_INFO", mn);
111                 }
112                 if (my_query_module(mn, QM_REFS, (void **)&deps, &depsize, &count)) {
113                         if (errno == ENOENT) {
114                                 /* The module was removed out from underneath us. */
115                                 continue;
116                         }
117                         perror_msg_and_die("module %s: QM_REFS", mn);
118                 }
119                 printf("%-20s%8lu%4ld ", mn, info.size, info.usecount);
120                 if (info.flags & NEW_MOD_DELETED)
121                         printf("(deleted)");
122                 else if (info.flags & NEW_MOD_INITIALIZING)
123                         printf("(initializing)");
124                 else if (!(info.flags & NEW_MOD_RUNNING))
125                         printf("(uninitialized)");
126                 else {
127                         if (info.flags & NEW_MOD_AUTOCLEAN)
128                                 printf("(autoclean) ");
129                         if (!(info.flags & NEW_MOD_USED_ONCE))
130                                 printf("(unused)");
131                 }
132                 if (count) printf("[");
133                 for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
134                         printf("%s%s", dn, (j==count-1)? "":" ");
135                 }
136                 if (count) printf("] ");
137
138                 printf("\n");
139         }
140
141
142         return( 0);
143 }
144
145 #else /*BB_FEATURE_OLD_MODULE_INTERFACE*/
146
147 extern int lsmod_main(int argc, char **argv)
148 {
149         int fd, i;
150         char line[128];
151
152         puts("Module                  Size  Used by");
153         fflush(stdout);
154
155         if ((fd = open("/proc/modules", O_RDONLY)) >= 0 ) {
156                 while ((i = read(fd, line, sizeof(line))) > 0) {
157                         write(fileno(stdout), line, i);
158                 }
159                 close(fd);
160                 return 0;
161         }
162         perror_msg_and_die("/proc/modules");
163         return 1;
164 }
165
166 #endif /*BB_FEATURE_OLD_MODULE_INTERFACE*/