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
46struct module_info
47{
48	unsigned long addr;
49	unsigned long size;
50	unsigned long flags;
51	long usecount;
52};
53
54
55int query_module(const char *name, int which, void *buf, size_t bufsize, size_t *ret);
56
57/* Values for query_module's which.  */
58static const int QM_MODULES = 1;
59static const int QM_DEPS = 2;
60static const int QM_REFS = 3;
61static const int QM_SYMBOLS = 4;
62static const int QM_INFO = 5;
63
64/* Bits of module.flags.  */
65static const int NEW_MOD_RUNNING = 1;
66static const int NEW_MOD_DELETED = 2;
67static const int NEW_MOD_AUTOCLEAN = 4;
68static const int NEW_MOD_VISITED = 8;
69static const int NEW_MOD_USED_ONCE = 16;
70static const int NEW_MOD_INITIALIZING = 64;
71
72static 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
89extern 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
147extern 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*/
167