1/* vi: set sw=4 ts=4: */
2/*
3 * Mini lsmod implementation for busybox
4 *
5 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
6 *
7 * Modified by Alcove, Julien Gaulmin <julien.gaulmin@alcove.fr> and
8 * Nicolas Ferre <nicolas.ferre@alcove.fr> to support pre 2.1 kernels
9 * (which lack the query_module() interface).
10 *
11 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
12 */
13
14#include "libbb.h"
15
16
17#if !ENABLE_FEATURE_CHECK_TAINTED_MODULE
18static void check_tainted(void) { puts(""); }
19#else
20#define TAINT_FILENAME                  "/proc/sys/kernel/tainted"
21#define TAINT_PROPRIETORY_MODULE        (1<<0)
22#define TAINT_FORCED_MODULE             (1<<1)
23#define TAINT_UNSAFE_SMP                (1<<2)
24
25static void check_tainted(void)
26{
27	int tainted;
28	FILE *f;
29
30	tainted = 0;
31	if ((f = fopen(TAINT_FILENAME, "r"))) {
32		fscanf(f, "%d", &tainted);
33		fclose(f);
34	}
35	if (f && tainted) {
36		printf("    Tainted: %c%c%c\n",
37				tainted & TAINT_PROPRIETORY_MODULE      ? 'P' : 'G',
38				tainted & TAINT_FORCED_MODULE           ? 'F' : ' ',
39				tainted & TAINT_UNSAFE_SMP              ? 'S' : ' ');
40	}
41	else {
42		printf("    Not tainted\n");
43	}
44}
45#endif
46
47#if ENABLE_FEATURE_QUERY_MODULE_INTERFACE
48
49struct module_info
50{
51	unsigned long addr;
52	unsigned long size;
53	unsigned long flags;
54	long usecount;
55};
56
57
58int query_module(const char *name, int which, void *buf, size_t bufsize, size_t *ret);
59
60enum {
61/* Values for query_module's which.  */
62	QM_MODULES = 1,
63	QM_DEPS = 2,
64	QM_REFS = 3,
65	QM_SYMBOLS = 4,
66	QM_INFO = 5,
67
68/* Bits of module.flags.  */
69	NEW_MOD_RUNNING = 1,
70	NEW_MOD_DELETED = 2,
71	NEW_MOD_AUTOCLEAN = 4,
72	NEW_MOD_VISITED = 8,
73	NEW_MOD_USED_ONCE = 16,
74	NEW_MOD_INITIALIZING = 64
75};
76
77int lsmod_main(int argc, char **argv);
78int lsmod_main(int argc, char **argv)
79{
80	struct module_info info;
81	char *module_names, *mn, *deps, *dn;
82	size_t bufsize, depsize, nmod, count, i, j;
83
84	module_names = deps = NULL;
85	bufsize = depsize = 0;
86	while (query_module(NULL, QM_MODULES, module_names, bufsize, &nmod)) {
87		if (errno != ENOSPC) bb_perror_msg_and_die("QM_MODULES");
88		module_names = xmalloc(bufsize = nmod);
89	}
90
91	deps = xmalloc(depsize = 256);
92	printf("Module\t\t\tSize  Used by");
93	check_tainted();
94
95	for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
96		if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
97			if (errno == ENOENT) {
98				/* The module was removed out from underneath us. */
99				continue;
100			}
101			/* else choke */
102			bb_perror_msg_and_die("module %s: QM_INFO", mn);
103		}
104		while (query_module(mn, QM_REFS, deps, depsize, &count)) {
105			if (errno == ENOENT) {
106				/* The module was removed out from underneath us. */
107				continue;
108			} else if (errno != ENOSPC)
109				bb_perror_msg_and_die("module %s: QM_REFS", mn);
110			deps = xrealloc(deps, count);
111		}
112		printf("%-20s%8lu%4ld", mn, info.size, info.usecount);
113		if (info.flags & NEW_MOD_DELETED)
114			printf(" (deleted)");
115		else if (info.flags & NEW_MOD_INITIALIZING)
116			printf(" (initializing)");
117		else if (!(info.flags & NEW_MOD_RUNNING))
118			printf(" (uninitialized)");
119		else {
120			if (info.flags & NEW_MOD_AUTOCLEAN)
121				printf(" (autoclean) ");
122			if (!(info.flags & NEW_MOD_USED_ONCE))
123				printf(" (unused)");
124		}
125		if (count) printf(" [");
126		for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
127			printf("%s%s", dn, (j==count-1)? "":" ");
128		}
129		if (count) printf("]");
130
131		puts("");
132	}
133
134#if ENABLE_FEATURE_CLEAN_UP
135	free(module_names);
136#endif
137
138	return 0;
139}
140
141#else /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */
142
143int lsmod_main(int argc, char **argv);
144int lsmod_main(int argc, char **argv)
145{
146	FILE *file = xfopen("/proc/modules", "r");
147
148	printf("Module                  Size  Used by");
149	check_tainted();
150#if defined(CONFIG_FEATURE_LSMOD_PRETTY_2_6_OUTPUT)
151	{
152		char *line;
153		while ((line = xmalloc_fgets(file)) != NULL) {
154			char *tok;
155
156			tok = strtok(line, " \t");
157			printf("%-19s", tok);
158			tok = strtok(NULL, " \t\n");
159			printf(" %8s", tok);
160			tok = strtok(NULL, " \t\n");
161			/* Null if no module unloading support. */
162			if (tok) {
163				printf("  %s", tok);
164				tok = strtok(NULL, "\n");
165				if (!tok)
166					tok = (char*)"";
167				/* New-style has commas, or -.  If so,
168				truncate (other fields might follow). */
169				else if (strchr(tok, ',')) {
170					tok = strtok(tok, "\t ");
171					/* Strip trailing comma. */
172					if (tok[strlen(tok)-1] == ',')
173						tok[strlen(tok)-1] = '\0';
174				} else if (tok[0] == '-'
175				 && (tok[1] == '\0' || isspace(tok[1]))
176				) {
177					tok = (char*)"";
178				}
179				printf(" %s", tok);
180			}
181			puts("");
182			free(line);
183		}
184		fclose(file);
185	}
186#else
187	xprint_and_close_file(file);
188#endif  /*  CONFIG_FEATURE_2_6_MODULES  */
189	return EXIT_SUCCESS;
190}
191
192#endif /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */
193