summaryrefslogtreecommitdiff
path: root/modutils/depmod.c
blob: 9faf9c302ecc09b571943fe6e2d2e1426daa093c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
/* vi: set sw=4 ts=4: */
/*
 * depmod - generate modules.dep
 * Copyright (c) 2008 Bernhard Fischer
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

#undef _GNU_SOURCE
#define _GNU_SOURCE
#include <libbb.h>
#include <sys/utsname.h> /* uname() */

/*
 * Theory of operation:
 * - iterate over all modules and record their full path
 * - iterate over all modules looking for "depends=" entries
 *   for each depends, look through our list of full paths and emit if found
 */
struct globals {
	llist_t *lst;
};
#define G (*(struct globals*)&bb_common_bufsiz1)
/* We have to zero it out because of NOEXEC */
#define INIT_G() memset(&G, 0, sizeof(G))

static int fill_lst(const char *modulename, struct stat ATTRIBUTE_UNUSED *sb,
					void ATTRIBUTE_UNUSED *data, int ATTRIBUTE_UNUSED depth)
{
	llist_add_to(&G.lst, xstrdup(modulename));
	return TRUE;
}

static int fileAction(const char *fname, struct stat *sb,
					void *data, int ATTRIBUTE_UNUSED depth)
{
	size_t len = sb->st_size;
	void *the_module, *ptr;
	int fd;
	char *depends, *deps;

/*XXX: FIXME: does not handle compressed modules!
 * There should be a function that looks at the extension and sets up
 * open_transformer for us.
 */
	if (last_char_is(fname, 'o') == NULL) /* not a module */
		goto skip;

	fd = xopen(fname, O_RDONLY);
	the_module = mmap(NULL, len, PROT_READ, MAP_SHARED
#if defined MAP_POPULATE
						|MAP_POPULATE
#endif
						, fd, 0);
	close(fd);
	if (the_module == MAP_FAILED)
		bb_perror_msg_and_die("mmap");

	ptr = the_module;

	fprintf((FILE*)data, "\n%s:", fname);
//bb_info_msg("fname='%s'", fname);
	do {
		/* search for a 'd' */
		ptr = memchr(ptr, 'd', len);
		if (ptr == NULL) /* no d left, done */
			goto done;
		if (memcmp(ptr, "depends=", sizeof("depends=")-1) == 0)
			break;
		len -= ++ptr - the_module;
	} while (1);
	deps = depends = xstrdup (ptr + sizeof("depends=")-1);
//bb_info_msg(" depends='%s'", depends);
	while (*deps) {
		llist_t * _lst = G.lst;
		char *buf1;

		ptr = strchr(deps, ',');
		if (ptr != NULL)
			*(char*)ptr = '\0';
		/* remember the length of the current dependency plus eventual 0 byte */
		len = strlen(deps) + (ptr != NULL);
		buf1 = xmalloc(len + 3);
		sprintf(buf1, "/%s.", deps); /* make sure we match the correct file */
		while (_lst) {
			ptr = strstr(_lst->data, buf1);
			if (ptr != NULL)
				break; /* found it */
			_lst = _lst->link;
		}
		free(buf1);
		if (_lst /*&& _lst->data*/) {
//bb_info_msg("[%s] -> '%s'", deps, _lst->data);
			fprintf((FILE*)data, " %s", _lst->data);
			deps += len;
		}
	}
	free(depends);
done:
	munmap(the_module, sb->st_size);
skip:
	return TRUE;
}

int depmod_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int depmod_main(int ATTRIBUTE_UNUSED argc, char **argv)
{
	int retval = EXIT_SUCCESS;
	char *moddir_base = NULL, *moddir, *system_map, *chp;
	FILE *filedes = stdout;
	enum {
		ARG_a = (1<<0), /* All modules, ignore mods in argv */
		ARG_A = (1<<1), /* Only emit .ko that are newer than modules.dep file */
		ARG_b = (1<<2), /* not /lib/modules/$(uname -r)/ but this base-dir */
		ARG_e = (1<<3), /* with -F, print unresolved symbols */
		ARG_F = (1<<4), /* System.map that contains the symbols */
		ARG_n = (1<<5)  /* dry-run, print to stdout only */
	};

	getopt32(argv, "aAb:eF:n", &moddir_base, &system_map);
	argv += optind;
	if (*argv) {/* got a version to use? */
		chp = *argv++;
//		if (memchr(chp, '/', strlen(chp)) != NULL) /* XXX: drop this */
//			bb_show_usage();
	} else {
		struct utsname uts;
		if (uname(&uts) < 0)
			bb_simple_perror_msg_and_die("uname");
		chp = xstrdup(uts.release);
	}
	/* if no modules are given on the command-line, -a is on per default */
	option_mask32 |= *argv == NULL;

	moddir = concat_path_file(CONFIG_DEFAULT_MODULES_DIR, chp);
//	if (ENABLE_FEATURE_CLEAN_UP)
//		free(chp);
	if (option_mask32 & ARG_b) {
		char *old_moddir = moddir;
		moddir = concat_path_file(moddir_base, moddir);
		if (ENABLE_FEATURE_CLEAN_UP)
			free(old_moddir);
	}

	if (!(option_mask32 & ARG_n)) { /* --dry-run */
		chp = concat_path_file(moddir, CONFIG_DEFAULT_DEPMOD_FILE);
		filedes = xfopen(chp, "w");
		if (ENABLE_FEATURE_CLEAN_UP)
			free(chp);
	}
	/* have to do a full walk to collect all needed data */
	if (!recursive_action(moddir,
			ACTION_RECURSE, /* flags */
			fill_lst, /* file action */
			NULL, /* dir action */
			NULL, /* user data */
			0)) {
		retval = EXIT_FAILURE;
	} else
	do {
		chp = concat_path_file(option_mask32 & ARG_a ?  moddir : moddir_base,
				option_mask32 & ARG_a ? "" : *argv++);

		if (!recursive_action(chp,
				ACTION_RECURSE, /* flags */
				fileAction, /* file action */
				NULL, /* dir action */
				(void*)filedes, /* user data */
				0)) { /* depth */
			retval = EXIT_FAILURE;
		}
		if (ENABLE_FEATURE_CLEAN_UP)
			free(chp);
	} while (!(option_mask32 & ARG_a) && *argv);

	if (ENABLE_FEATURE_CLEAN_UP) {
		fclose(filedes);
		llist_free(G.lst, free);
		free(moddir);
	}
	return retval;
}