1 /*
2  * libkmod - interface to kernel module operations
3  *
4  * Copyright (C) 2011-2013  ProFUSION embedded systems
5  * Copyright (C) 2013  Intel Corporation. All rights reserved.
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
19  */
20 
21 #include <ctype.h>
22 #include <dirent.h>
23 #include <errno.h>
24 #include <stdarg.h>
25 #include <stddef.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <unistd.h>
30 #include <sys/stat.h>
31 #include <sys/types.h>
32 
33 #include <shared/util.h>
34 
35 #include "libkmod.h"
36 #include "libkmod-internal.h"
37 
38 struct kmod_alias {
39 	char *name;
40 	char modname[];
41 };
42 
43 struct kmod_options {
44 	char *options;
45 	char modname[];
46 };
47 
48 struct kmod_command {
49 	char *command;
50 	char modname[];
51 };
52 
53 struct kmod_softdep {
54 	char *name;
55 	const char **pre;
56 	const char **post;
57 	unsigned int n_pre;
58 	unsigned int n_post;
59 };
60 
kmod_blacklist_get_modname(const struct kmod_list * l)61 const char *kmod_blacklist_get_modname(const struct kmod_list *l)
62 {
63 	return l->data;
64 }
65 
kmod_alias_get_name(const struct kmod_list * l)66 const char *kmod_alias_get_name(const struct kmod_list *l) {
67 	const struct kmod_alias *alias = l->data;
68 	return alias->name;
69 }
70 
kmod_alias_get_modname(const struct kmod_list * l)71 const char *kmod_alias_get_modname(const struct kmod_list *l) {
72 	const struct kmod_alias *alias = l->data;
73 	return alias->modname;
74 }
75 
kmod_option_get_options(const struct kmod_list * l)76 const char *kmod_option_get_options(const struct kmod_list *l) {
77 	const struct kmod_options *alias = l->data;
78 	return alias->options;
79 }
80 
kmod_option_get_modname(const struct kmod_list * l)81 const char *kmod_option_get_modname(const struct kmod_list *l) {
82 	const struct kmod_options *alias = l->data;
83 	return alias->modname;
84 }
85 
kmod_command_get_command(const struct kmod_list * l)86 const char *kmod_command_get_command(const struct kmod_list *l) {
87 	const struct kmod_command *alias = l->data;
88 	return alias->command;
89 }
90 
kmod_command_get_modname(const struct kmod_list * l)91 const char *kmod_command_get_modname(const struct kmod_list *l) {
92 	const struct kmod_command *alias = l->data;
93 	return alias->modname;
94 }
95 
kmod_softdep_get_name(const struct kmod_list * l)96 const char *kmod_softdep_get_name(const struct kmod_list *l) {
97 	const struct kmod_softdep *dep = l->data;
98 	return dep->name;
99 }
100 
kmod_softdep_get_pre(const struct kmod_list * l,unsigned int * count)101 const char * const *kmod_softdep_get_pre(const struct kmod_list *l, unsigned int *count) {
102 	const struct kmod_softdep *dep = l->data;
103 	*count = dep->n_pre;
104 	return dep->pre;
105 }
106 
kmod_softdep_get_post(const struct kmod_list * l,unsigned int * count)107 const char * const *kmod_softdep_get_post(const struct kmod_list *l, unsigned int *count) {
108 	const struct kmod_softdep *dep = l->data;
109 	*count = dep->n_post;
110 	return dep->post;
111 }
112 
kmod_config_add_command(struct kmod_config * config,const char * modname,const char * command,const char * command_name,struct kmod_list ** list)113 static int kmod_config_add_command(struct kmod_config *config,
114 						const char *modname,
115 						const char *command,
116 						const char *command_name,
117 						struct kmod_list **list)
118 {
119 	_cleanup_free_ struct kmod_command *cmd;
120 	struct kmod_list *l;
121 	size_t modnamelen = strlen(modname) + 1;
122 	size_t commandlen = strlen(command) + 1;
123 
124 	DBG(config->ctx, "modname='%s' cmd='%s %s'\n", modname, command_name,
125 								command);
126 
127 	cmd = malloc(sizeof(*cmd) + modnamelen + commandlen);
128 	if (!cmd)
129 		return -ENOMEM;
130 
131 	cmd->command = sizeof(*cmd) + modnamelen + (char *)cmd;
132 	memcpy(cmd->modname, modname, modnamelen);
133 	memcpy(cmd->command, command, commandlen);
134 
135 	l = kmod_list_append(*list, cmd);
136 	if (!l)
137 		return -ENOMEM;
138 
139 	*list = l;
140 	cmd = NULL;
141 	return 0;
142 }
143 
kmod_config_free_command(struct kmod_config * config,struct kmod_list * l,struct kmod_list ** list)144 static void kmod_config_free_command(struct kmod_config *config,
145 					struct kmod_list *l,
146 					struct kmod_list **list)
147 {
148 	struct kmod_command *cmd = l->data;
149 
150 	free(cmd);
151 	*list = kmod_list_remove(l);
152 }
153 
kmod_config_add_options(struct kmod_config * config,const char * modname,const char * options)154 static int kmod_config_add_options(struct kmod_config *config,
155 				const char *modname, const char *options)
156 {
157 	_cleanup_free_ struct kmod_options *opt;
158 	struct kmod_list *list;
159 	size_t modnamelen = strlen(modname) + 1;
160 	size_t optionslen = strlen(options) + 1;
161 
162 	DBG(config->ctx, "modname='%s' options='%s'\n", modname, options);
163 
164 	opt = malloc(sizeof(*opt) + modnamelen + optionslen);
165 	if (!opt)
166 		return -ENOMEM;
167 
168 	opt->options = sizeof(*opt) + modnamelen + (char *)opt;
169 
170 	memcpy(opt->modname, modname, modnamelen);
171 	memcpy(opt->options, options, optionslen);
172 	strchr_replace(opt->options, '\t', ' ');
173 
174 	list = kmod_list_append(config->options, opt);
175 	if (!list)
176 		return -ENOMEM;
177 
178 	opt = NULL;
179 	config->options = list;
180 	return 0;
181 }
182 
kmod_config_free_options(struct kmod_config * config,struct kmod_list * l)183 static void kmod_config_free_options(struct kmod_config *config,
184 							struct kmod_list *l)
185 {
186 	struct kmod_options *opt = l->data;
187 
188 	free(opt);
189 
190 	config->options = kmod_list_remove(l);
191 }
192 
kmod_config_add_alias(struct kmod_config * config,const char * name,const char * modname)193 static int kmod_config_add_alias(struct kmod_config *config,
194 					const char *name, const char *modname)
195 {
196 	_cleanup_free_ struct kmod_alias *alias;
197 	struct kmod_list *list;
198 	size_t namelen = strlen(name) + 1, modnamelen = strlen(modname) + 1;
199 
200 	DBG(config->ctx, "name=%s modname=%s\n", name, modname);
201 
202 	alias = malloc(sizeof(*alias) + namelen + modnamelen);
203 	if (!alias)
204 		return -ENOMEM;
205 
206 	alias->name = sizeof(*alias) + modnamelen + (char *)alias;
207 
208 	memcpy(alias->modname, modname, modnamelen);
209 	memcpy(alias->name, name, namelen);
210 
211 	list = kmod_list_append(config->aliases, alias);
212 	if (!list)
213 		return -ENOMEM;
214 
215 	alias = NULL;
216 	config->aliases = list;
217 	return 0;
218 }
219 
kmod_config_free_alias(struct kmod_config * config,struct kmod_list * l)220 static void kmod_config_free_alias(struct kmod_config *config,
221 							struct kmod_list *l)
222 {
223 	struct kmod_alias *alias = l->data;
224 
225 	free(alias);
226 
227 	config->aliases = kmod_list_remove(l);
228 }
229 
kmod_config_add_blacklist(struct kmod_config * config,const char * modname)230 static int kmod_config_add_blacklist(struct kmod_config *config,
231 							const char *modname)
232 {
233 	_cleanup_free_ char *p;
234 	struct kmod_list *list;
235 
236 	DBG(config->ctx, "modname=%s\n", modname);
237 
238 	p = strdup(modname);
239 	if (!p)
240 		return -ENOMEM;
241 
242 	list = kmod_list_append(config->blacklists, p);
243 	if (!list)
244 		return -ENOMEM;
245 
246 	p = NULL;
247 	config->blacklists = list;
248 	return 0;
249 }
250 
kmod_config_free_blacklist(struct kmod_config * config,struct kmod_list * l)251 static void kmod_config_free_blacklist(struct kmod_config *config,
252 							struct kmod_list *l)
253 {
254 	free(l->data);
255 	config->blacklists = kmod_list_remove(l);
256 }
257 
kmod_config_add_softdep(struct kmod_config * config,const char * modname,const char * line)258 static int kmod_config_add_softdep(struct kmod_config *config,
259 							const char *modname,
260 							const char *line)
261 {
262 	struct kmod_list *list;
263 	struct kmod_softdep *dep;
264 	const char *s, *p;
265 	char *itr;
266 	unsigned int n_pre = 0, n_post = 0;
267 	size_t modnamelen = strlen(modname) + 1;
268 	size_t buflen = 0;
269 	bool was_space = false;
270 	enum { S_NONE, S_PRE, S_POST } mode = S_NONE;
271 
272 	DBG(config->ctx, "modname=%s\n", modname);
273 
274 	/* analyze and count */
275 	for (p = s = line; ; s++) {
276 		size_t plen;
277 
278 		if (*s != '\0') {
279 			if (!isspace(*s)) {
280 				was_space = false;
281 				continue;
282 			}
283 
284 			if (was_space) {
285 				p = s + 1;
286 				continue;
287 			}
288 			was_space = true;
289 
290 			if (p >= s)
291 				continue;
292 		}
293 		plen = s - p;
294 
295 		if (plen == sizeof("pre:") - 1 &&
296 				memcmp(p, "pre:", sizeof("pre:") - 1) == 0)
297 			mode = S_PRE;
298 		else if (plen == sizeof("post:") - 1 &&
299 				memcmp(p, "post:", sizeof("post:") - 1) == 0)
300 			mode = S_POST;
301 		else if (*s != '\0' || (*s == '\0' && !was_space)) {
302 			if (mode == S_PRE) {
303 				buflen += plen + 1;
304 				n_pre++;
305 			} else if (mode == S_POST) {
306 				buflen += plen + 1;
307 				n_post++;
308 			}
309 		}
310 		p = s + 1;
311 		if (*s == '\0')
312 			break;
313 	}
314 
315 	DBG(config->ctx, "%u pre, %u post\n", n_pre, n_post);
316 
317 	dep = malloc(sizeof(struct kmod_softdep) + modnamelen +
318 		     n_pre * sizeof(const char *) +
319 		     n_post * sizeof(const char *) +
320 		     buflen);
321 	if (dep == NULL) {
322 		ERR(config->ctx, "out-of-memory modname=%s\n", modname);
323 		return -ENOMEM;
324 	}
325 	dep->n_pre = n_pre;
326 	dep->n_post = n_post;
327 	dep->pre = (const char **)((char *)dep + sizeof(struct kmod_softdep));
328 	dep->post = dep->pre + n_pre;
329 	dep->name = (char *)(dep->post + n_post);
330 
331 	memcpy(dep->name, modname, modnamelen);
332 
333 	/* copy strings */
334 	itr = dep->name + modnamelen;
335 	n_pre = 0;
336 	n_post = 0;
337 	mode = S_NONE;
338 	was_space = false;
339 	for (p = s = line; ; s++) {
340 		size_t plen;
341 
342 		if (*s != '\0') {
343 			if (!isspace(*s)) {
344 				was_space = false;
345 				continue;
346 			}
347 
348 			if (was_space) {
349 				p = s + 1;
350 				continue;
351 			}
352 			was_space = true;
353 
354 			if (p >= s)
355 				continue;
356 		}
357 		plen = s - p;
358 
359 		if (plen == sizeof("pre:") - 1 &&
360 				memcmp(p, "pre:", sizeof("pre:") - 1) == 0)
361 			mode = S_PRE;
362 		else if (plen == sizeof("post:") - 1 &&
363 				memcmp(p, "post:", sizeof("post:") - 1) == 0)
364 			mode = S_POST;
365 		else if (*s != '\0' || (*s == '\0' && !was_space)) {
366 			if (mode == S_PRE) {
367 				dep->pre[n_pre] = itr;
368 				memcpy(itr, p, plen);
369 				itr[plen] = '\0';
370 				itr += plen + 1;
371 				n_pre++;
372 			} else if (mode == S_POST) {
373 				dep->post[n_post] = itr;
374 				memcpy(itr, p, plen);
375 				itr[plen] = '\0';
376 				itr += plen + 1;
377 				n_post++;
378 			}
379 		}
380 		p = s + 1;
381 		if (*s == '\0')
382 			break;
383 	}
384 
385 	list = kmod_list_append(config->softdeps, dep);
386 	if (list == NULL) {
387 		free(dep);
388 		return -ENOMEM;
389 	}
390 	config->softdeps = list;
391 
392 	return 0;
393 }
394 
softdep_to_char(struct kmod_softdep * dep)395 static char *softdep_to_char(struct kmod_softdep *dep) {
396 	const size_t sz_preprefix = sizeof("pre: ") - 1;
397 	const size_t sz_postprefix = sizeof("post: ") - 1;
398 	size_t sz = 1; /* at least '\0' */
399 	size_t sz_pre, sz_post;
400 	const char *start, *end;
401 	char *s, *itr;
402 
403 	/*
404 	 * Rely on the fact that dep->pre[] and dep->post[] are strv's that
405 	 * point to a contiguous buffer
406 	 */
407 	if (dep->n_pre > 0) {
408 		start = dep->pre[0];
409 		end = dep->pre[dep->n_pre - 1]
410 					+ strlen(dep->pre[dep->n_pre - 1]);
411 		sz_pre = end - start;
412 		sz += sz_pre + sz_preprefix;
413 	} else
414 		sz_pre = 0;
415 
416 	if (dep->n_post > 0) {
417 		start = dep->post[0];
418 		end = dep->post[dep->n_post - 1]
419 					+ strlen(dep->post[dep->n_post - 1]);
420 		sz_post = end - start;
421 		sz += sz_post + sz_postprefix;
422 	} else
423 		sz_post = 0;
424 
425 	itr = s = malloc(sz);
426 	if (s == NULL)
427 		return NULL;
428 
429 	if (sz_pre) {
430 		char *p;
431 
432 		memcpy(itr, "pre: ", sz_preprefix);
433 		itr += sz_preprefix;
434 
435 		/* include last '\0' */
436 		memcpy(itr, dep->pre[0], sz_pre + 1);
437 		for (p = itr; p < itr + sz_pre; p++) {
438 			if (*p == '\0')
439 				*p = ' ';
440 		}
441 		itr = p;
442 	}
443 
444 	if (sz_post) {
445 		char *p;
446 
447 		memcpy(itr, "post: ", sz_postprefix);
448 		itr += sz_postprefix;
449 
450 		/* include last '\0' */
451 		memcpy(itr, dep->post[0], sz_post + 1);
452 		for (p = itr; p < itr + sz_post; p++) {
453 			if (*p == '\0')
454 				*p = ' ';
455 		}
456 		itr = p;
457 	}
458 
459 	*itr = '\0';
460 
461 	return s;
462 }
463 
kmod_config_free_softdep(struct kmod_config * config,struct kmod_list * l)464 static void kmod_config_free_softdep(struct kmod_config *config,
465 							struct kmod_list *l)
466 {
467 	free(l->data);
468 	config->softdeps = kmod_list_remove(l);
469 }
470 
kcmdline_parse_result(struct kmod_config * config,char * modname,char * param,char * value)471 static void kcmdline_parse_result(struct kmod_config *config, char *modname,
472 						char *param, char *value)
473 {
474 	if (modname == NULL || param == NULL)
475 		return;
476 
477 	DBG(config->ctx, "%s %s\n", modname, param);
478 
479 	if (streq(modname, "modprobe") && !strncmp(param, "blacklist=", 10)) {
480 		for (;;) {
481 			char *t = strsep(&value, ",");
482 			if (t == NULL)
483 				break;
484 
485 			kmod_config_add_blacklist(config, t);
486 		}
487 	} else {
488 		if (underscores(modname) < 0) {
489 			ERR(config->ctx, "Ignoring bad option on kernel command line while parsing module name: '%s'\n",
490 			    modname);
491 		} else {
492 			kmod_config_add_options(config, modname, param);
493 		}
494 	}
495 }
496 
kmod_config_parse_kcmdline(struct kmod_config * config)497 static int kmod_config_parse_kcmdline(struct kmod_config *config)
498 {
499 	char buf[KCMD_LINE_SIZE];
500 	int fd, err;
501 	char *p, *modname,  *param = NULL, *value = NULL;
502 	bool is_quoted = false, is_module = true;
503 
504 	fd = open("/proc/cmdline", O_RDONLY|O_CLOEXEC);
505 	if (fd < 0) {
506 		err = -errno;
507 		DBG(config->ctx, "could not open '/proc/cmdline' for reading: %m\n");
508 		return err;
509 	}
510 
511 	err = read_str_safe(fd, buf, sizeof(buf));
512 	close(fd);
513 	if (err < 0) {
514 		ERR(config->ctx, "could not read from '/proc/cmdline': %s\n",
515 							strerror(-err));
516 		return err;
517 	}
518 
519 	for (p = buf, modname = buf; *p != '\0' && *p != '\n'; p++) {
520 		if (*p == '"') {
521 			is_quoted = !is_quoted;
522 
523 			if (is_quoted) {
524 				/* don't consider a module until closing quotes */
525 				is_module = false;
526 			} else if (param != NULL && value != NULL) {
527 				/*
528 				 * If we are indeed expecting a value and
529 				 * closing quotes, then this can be considered
530 				 * a valid option for a module
531 				 */
532 				is_module = true;
533 			}
534 
535 			continue;
536 		}
537 		if (is_quoted)
538 			continue;
539 
540 		switch (*p) {
541 		case ' ':
542 			*p = '\0';
543 			if (is_module)
544 				kcmdline_parse_result(config, modname, param, value);
545 			param = value = NULL;
546 			modname = p + 1;
547 			is_module = true;
548 			break;
549 		case '.':
550 			if (param == NULL) {
551 				*p = '\0';
552 				param = p + 1;
553 			}
554 			break;
555 		case '=':
556 			if (param != NULL)
557 				value = p + 1;
558 			else
559 				is_module = false;
560 			break;
561 		}
562 	}
563 
564 	*p = '\0';
565 	if (is_module)
566 		kcmdline_parse_result(config, modname, param, value);
567 
568 	return 0;
569 }
570 
571 /*
572  * Take an fd and own it. It will be closed on return. filename is used only
573  * for debug messages
574  */
kmod_config_parse(struct kmod_config * config,int fd,const char * filename)575 static int kmod_config_parse(struct kmod_config *config, int fd,
576 							const char *filename)
577 {
578 	struct kmod_ctx *ctx = config->ctx;
579 	char *line;
580 	FILE *fp;
581 	unsigned int linenum = 0;
582 	int err;
583 
584 	fp = fdopen(fd, "r");
585 	if (fp == NULL) {
586 		err = -errno;
587 		ERR(config->ctx, "fd %d: %m\n", fd);
588 		close(fd);
589 		return err;
590 	}
591 
592 	while ((line = freadline_wrapped(fp, &linenum)) != NULL) {
593 		char *cmd, *saveptr;
594 
595 		if (line[0] == '\0' || line[0] == '#')
596 			goto done_next;
597 
598 		cmd = strtok_r(line, "\t ", &saveptr);
599 		if (cmd == NULL)
600 			goto done_next;
601 
602 		if (streq(cmd, "alias")) {
603 			char *alias = strtok_r(NULL, "\t ", &saveptr);
604 			char *modname = strtok_r(NULL, "\t ", &saveptr);
605 
606 			if (underscores(alias) < 0 || underscores(modname) < 0)
607 				goto syntax_error;
608 
609 			kmod_config_add_alias(config, alias, modname);
610 		} else if (streq(cmd, "blacklist")) {
611 			char *modname = strtok_r(NULL, "\t ", &saveptr);
612 
613 			if (underscores(modname) < 0)
614 				goto syntax_error;
615 
616 			kmod_config_add_blacklist(config, modname);
617 		} else if (streq(cmd, "options")) {
618 			char *modname = strtok_r(NULL, "\t ", &saveptr);
619 			char *options = strtok_r(NULL, "\0", &saveptr);
620 
621 			if (underscores(modname) < 0 || options == NULL)
622 				goto syntax_error;
623 
624 			kmod_config_add_options(config, modname, options);
625 		} else if (streq(cmd, "install")) {
626 			char *modname = strtok_r(NULL, "\t ", &saveptr);
627 			char *installcmd = strtok_r(NULL, "\0", &saveptr);
628 
629 			if (underscores(modname) < 0 || installcmd == NULL)
630 				goto syntax_error;
631 
632 			kmod_config_add_command(config, modname, installcmd,
633 					cmd, &config->install_commands);
634 		} else if (streq(cmd, "remove")) {
635 			char *modname = strtok_r(NULL, "\t ", &saveptr);
636 			char *removecmd = strtok_r(NULL, "\0", &saveptr);
637 
638 			if (underscores(modname) < 0 || removecmd == NULL)
639 				goto syntax_error;
640 
641 			kmod_config_add_command(config, modname, removecmd,
642 					cmd, &config->remove_commands);
643 		} else if (streq(cmd, "softdep")) {
644 			char *modname = strtok_r(NULL, "\t ", &saveptr);
645 			char *softdeps = strtok_r(NULL, "\0", &saveptr);
646 
647 			if (underscores(modname) < 0 || softdeps == NULL)
648 				goto syntax_error;
649 
650 			kmod_config_add_softdep(config, modname, softdeps);
651 		} else if (streq(cmd, "include")
652 				|| streq(cmd, "config")) {
653 			ERR(ctx, "%s: command %s is deprecated and not parsed anymore\n",
654 								filename, cmd);
655 		} else {
656 syntax_error:
657 			ERR(ctx, "%s line %u: ignoring bad line starting with '%s'\n",
658 						filename, linenum, cmd);
659 		}
660 
661 done_next:
662 		free(line);
663 	}
664 
665 	fclose(fp);
666 
667 	return 0;
668 }
669 
kmod_config_free(struct kmod_config * config)670 void kmod_config_free(struct kmod_config *config)
671 {
672 	while (config->aliases)
673 		kmod_config_free_alias(config, config->aliases);
674 
675 	while (config->blacklists)
676 		kmod_config_free_blacklist(config, config->blacklists);
677 
678 	while (config->options)
679 		kmod_config_free_options(config, config->options);
680 
681 	while (config->install_commands) {
682 		kmod_config_free_command(config, config->install_commands,
683 						&config->install_commands);
684 	}
685 
686 	while (config->remove_commands) {
687 		kmod_config_free_command(config, config->remove_commands,
688 						&config->remove_commands);
689 	}
690 
691 	while (config->softdeps)
692 		kmod_config_free_softdep(config, config->softdeps);
693 
694 	for (; config->paths != NULL;
695 				config->paths = kmod_list_remove(config->paths))
696 		free(config->paths->data);
697 
698 	free(config);
699 }
700 
conf_files_filter_out(struct kmod_ctx * ctx,DIR * d,const char * path,const char * fn)701 static bool conf_files_filter_out(struct kmod_ctx *ctx, DIR *d,
702 					const char *path, const char *fn)
703 {
704 	size_t len = strlen(fn);
705 	struct stat st;
706 
707 	if (fn[0] == '.')
708 		return true;
709 
710 	if (len < 6 || (!streq(&fn[len - 5], ".conf")
711 				&& !streq(&fn[len - 6], ".alias")))
712 		return true;
713 
714 	fstatat(dirfd(d), fn, &st, 0);
715 
716 	if (S_ISDIR(st.st_mode)) {
717 		ERR(ctx, "Directories inside directories are not supported: "
718 							"%s/%s\n", path, fn);
719 		return true;
720 	}
721 
722 	return false;
723 }
724 
725 struct conf_file {
726 	const char *path;
727 	bool is_single;
728 	char name[];
729 };
730 
conf_files_insert_sorted(struct kmod_ctx * ctx,struct kmod_list ** list,const char * path,const char * name)731 static int conf_files_insert_sorted(struct kmod_ctx *ctx,
732 					struct kmod_list **list,
733 					const char *path, const char *name)
734 {
735 	struct kmod_list *lpos, *tmp;
736 	struct conf_file *cf;
737 	size_t namelen;
738 	int cmp = -1;
739 	bool is_single = false;
740 
741 	if (name == NULL) {
742 		name = basename(path);
743 		is_single = true;
744 	}
745 
746 	kmod_list_foreach(lpos, *list) {
747 		cf = lpos->data;
748 
749 		if ((cmp = strcmp(name, cf->name)) <= 0)
750 			break;
751 	}
752 
753 	if (cmp == 0) {
754 		DBG(ctx, "Ignoring duplicate config file: %s/%s\n", path,
755 									name);
756 		return -EEXIST;
757 	}
758 
759 	namelen = strlen(name);
760 	cf = malloc(sizeof(*cf) + namelen + 1);
761 	if (cf == NULL)
762 		return -ENOMEM;
763 
764 	memcpy(cf->name, name, namelen + 1);
765 	cf->path = path;
766 	cf->is_single = is_single;
767 
768 	if (lpos == NULL)
769 		tmp = kmod_list_append(*list, cf);
770 	else if (lpos == *list)
771 		tmp = kmod_list_prepend(*list, cf);
772 	else
773 		tmp = kmod_list_insert_before(lpos, cf);
774 
775 	if (tmp == NULL) {
776 		free(cf);
777 		return -ENOMEM;
778 	}
779 
780 	if (lpos == NULL || lpos == *list)
781 		*list = tmp;
782 
783 	return 0;
784 }
785 
786 /*
787  * Insert configuration files in @list, ignoring duplicates
788  */
conf_files_list(struct kmod_ctx * ctx,struct kmod_list ** list,const char * path,unsigned long long * path_stamp)789 static int conf_files_list(struct kmod_ctx *ctx, struct kmod_list **list,
790 						const char *path,
791 						unsigned long long *path_stamp)
792 {
793 	DIR *d;
794 	int err;
795 	struct stat st;
796 	struct dirent *dent;
797 
798 	if (stat(path, &st) != 0) {
799 		err = -errno;
800 		DBG(ctx, "could not stat '%s': %m\n", path);
801 		return err;
802 	}
803 
804 	*path_stamp = stat_mstamp(&st);
805 
806 	if (!S_ISDIR(st.st_mode)) {
807 		conf_files_insert_sorted(ctx, list, path, NULL);
808 		return 0;
809 	}
810 
811 	d = opendir(path);
812 	if (d == NULL) {
813 		ERR(ctx, "opendir(%s): %m\n", path);
814 		return -EINVAL;
815 	}
816 
817 	for (dent = readdir(d); dent != NULL; dent = readdir(d)) {
818 		if (conf_files_filter_out(ctx, d, path, dent->d_name))
819 			continue;
820 
821 		conf_files_insert_sorted(ctx, list, path, dent->d_name);
822 	}
823 
824 	closedir(d);
825 	return 0;
826 }
827 
kmod_config_new(struct kmod_ctx * ctx,struct kmod_config ** p_config,const char * const * config_paths)828 int kmod_config_new(struct kmod_ctx *ctx, struct kmod_config **p_config,
829 					const char * const *config_paths)
830 {
831 	struct kmod_config *config;
832 	struct kmod_list *list = NULL;
833 	struct kmod_list *path_list = NULL;
834 	size_t i;
835 
836 	conf_files_insert_sorted(ctx, &list, kmod_get_dirname(ctx), "modules.softdep");
837 
838 	for (i = 0; config_paths[i] != NULL; i++) {
839 		const char *path = config_paths[i];
840 		unsigned long long path_stamp = 0;
841 		size_t pathlen;
842 		struct kmod_list *tmp;
843 		struct kmod_config_path *cf;
844 
845 		if (conf_files_list(ctx, &list, path, &path_stamp) < 0)
846 			continue;
847 
848 		pathlen = strlen(path) + 1;
849 		cf = malloc(sizeof(*cf) + pathlen);
850 		if (cf == NULL)
851 			goto oom;
852 
853 		cf->stamp = path_stamp;
854 		memcpy(cf->path, path, pathlen);
855 
856 		tmp = kmod_list_append(path_list, cf);
857 		if (tmp == NULL)
858 			goto oom;
859 		path_list = tmp;
860 	}
861 
862 	*p_config = config = calloc(1, sizeof(struct kmod_config));
863 	if (config == NULL)
864 		goto oom;
865 
866 	config->paths = path_list;
867 	config->ctx = ctx;
868 
869 	for (; list != NULL; list = kmod_list_remove(list)) {
870 		char buf[PATH_MAX];
871 		const char *fn = buf;
872 		struct conf_file *cf = list->data;
873 		int fd;
874 
875 		if (cf->is_single) {
876 			fn = cf->path;
877 		} else if (snprintf(buf, sizeof(buf), "%s/%s",
878 				    cf->path, cf->name) >= (int)sizeof(buf)) {
879 			ERR(ctx, "Error parsing %s/%s: path too long\n",
880 			    cf->path, cf->name);
881 			free(cf);
882 			continue;
883 		}
884 
885 		fd = open(fn, O_RDONLY|O_CLOEXEC);
886 		DBG(ctx, "parsing file '%s' fd=%d\n", fn, fd);
887 
888 		if (fd >= 0)
889 			kmod_config_parse(config, fd, fn);
890 
891 		free(cf);
892 	}
893 
894 	kmod_config_parse_kcmdline(config);
895 
896 	return 0;
897 
898 oom:
899 	for (; list != NULL; list = kmod_list_remove(list))
900 		free(list->data);
901 
902 	for (; path_list != NULL; path_list = kmod_list_remove(path_list))
903 		free(path_list->data);
904 
905 	return -ENOMEM;
906 }
907 
908 /**********************************************************************
909  * struct kmod_config_iter functions
910  **********************************************************************/
911 
912 enum config_type {
913 	CONFIG_TYPE_BLACKLIST = 0,
914 	CONFIG_TYPE_INSTALL,
915 	CONFIG_TYPE_REMOVE,
916 	CONFIG_TYPE_ALIAS,
917 	CONFIG_TYPE_OPTION,
918 	CONFIG_TYPE_SOFTDEP,
919 };
920 
921 struct kmod_config_iter {
922 	enum config_type type;
923 	bool intermediate;
924 	const struct kmod_list *list;
925 	const struct kmod_list *curr;
926 	void *data;
927 	const char *(*get_key)(const struct kmod_list *l);
928 	const char *(*get_value)(const struct kmod_list *l);
929 };
930 
softdep_get_plain_softdep(const struct kmod_list * l)931 static const char *softdep_get_plain_softdep(const struct kmod_list *l)
932 {
933 	char *s = softdep_to_char(l->data);
934 	return s;
935 }
936 
kmod_config_iter_new(const struct kmod_ctx * ctx,enum config_type type)937 static struct kmod_config_iter *kmod_config_iter_new(const struct kmod_ctx* ctx,
938 							enum config_type type)
939 {
940 	struct kmod_config_iter *iter = calloc(1, sizeof(*iter));
941 	const struct kmod_config *config = kmod_get_config(ctx);
942 
943 	if (iter == NULL)
944 		return NULL;
945 
946 	iter->type = type;
947 
948 	switch (type) {
949 	case CONFIG_TYPE_BLACKLIST:
950 		iter->list = config->blacklists;
951 		iter->get_key = kmod_blacklist_get_modname;
952 		break;
953 	case CONFIG_TYPE_INSTALL:
954 		iter->list = config->install_commands;
955 		iter->get_key = kmod_command_get_modname;
956 		iter->get_value = kmod_command_get_command;
957 		break;
958 	case CONFIG_TYPE_REMOVE:
959 		iter->list = config->remove_commands;
960 		iter->get_key = kmod_command_get_modname;
961 		iter->get_value = kmod_command_get_command;
962 		break;
963 	case CONFIG_TYPE_ALIAS:
964 		iter->list = config->aliases;
965 		iter->get_key = kmod_alias_get_name;
966 		iter->get_value = kmod_alias_get_modname;
967 		break;
968 	case CONFIG_TYPE_OPTION:
969 		iter->list = config->options;
970 		iter->get_key = kmod_option_get_modname;
971 		iter->get_value = kmod_option_get_options;
972 		break;
973 	case CONFIG_TYPE_SOFTDEP:
974 		iter->list = config->softdeps;
975 		iter->get_key = kmod_softdep_get_name;
976 		iter->get_value = softdep_get_plain_softdep;
977 		iter->intermediate = true;
978 		break;
979 	}
980 
981 	return iter;
982 }
983 
984 /**
985  * SECTION:libkmod-config
986  * @short_description: retrieve current libkmod configuration
987  */
988 
989 /**
990  * kmod_config_get_blacklists:
991  * @ctx: kmod library context
992  *
993  * Retrieve an iterator to deal with the blacklist maintained inside the
994  * library. See kmod_config_iter_get_key(), kmod_config_iter_get_value() and
995  * kmod_config_iter_next(). At least one call to kmod_config_iter_next() must
996  * be made to initialize the iterator and check if it's valid.
997  *
998  * Returns: a new iterator over the blacklists or NULL on failure. Free it
999  * with kmod_config_iter_free_iter().
1000  */
kmod_config_get_blacklists(const struct kmod_ctx * ctx)1001 KMOD_EXPORT struct kmod_config_iter *kmod_config_get_blacklists(const struct kmod_ctx *ctx)
1002 {
1003 	if (ctx == NULL)
1004 		return NULL;;
1005 
1006 	return kmod_config_iter_new(ctx, CONFIG_TYPE_BLACKLIST);
1007 }
1008 
1009 /**
1010  * kmod_config_get_install_commands:
1011  * @ctx: kmod library context
1012  *
1013  * Retrieve an iterator to deal with the install commands maintained inside the
1014  * library. See kmod_config_iter_get_key(), kmod_config_iter_get_value() and
1015  * kmod_config_iter_next(). At least one call to kmod_config_iter_next() must
1016  * be made to initialize the iterator and check if it's valid.
1017  *
1018  * Returns: a new iterator over the install commands or NULL on failure. Free
1019  * it with kmod_config_iter_free_iter().
1020  */
kmod_config_get_install_commands(const struct kmod_ctx * ctx)1021 KMOD_EXPORT struct kmod_config_iter *kmod_config_get_install_commands(const struct kmod_ctx *ctx)
1022 {
1023 	if (ctx == NULL)
1024 		return NULL;;
1025 
1026 	return kmod_config_iter_new(ctx, CONFIG_TYPE_INSTALL);
1027 }
1028 
1029 /**
1030  * kmod_config_get_remove_commands:
1031  * @ctx: kmod library context
1032  *
1033  * Retrieve an iterator to deal with the remove commands maintained inside the
1034  * library. See kmod_config_iter_get_key(), kmod_config_iter_get_value() and
1035  * kmod_config_iter_next(). At least one call to kmod_config_iter_next() must
1036  * be made to initialize the iterator and check if it's valid.
1037  *
1038  * Returns: a new iterator over the remove commands or NULL on failure. Free
1039  * it with kmod_config_iter_free_iter().
1040  */
kmod_config_get_remove_commands(const struct kmod_ctx * ctx)1041 KMOD_EXPORT struct kmod_config_iter *kmod_config_get_remove_commands(const struct kmod_ctx *ctx)
1042 {
1043 	if (ctx == NULL)
1044 		return NULL;;
1045 
1046 	return kmod_config_iter_new(ctx, CONFIG_TYPE_REMOVE);
1047 }
1048 
1049 /**
1050  * kmod_config_get_aliases:
1051  * @ctx: kmod library context
1052  *
1053  * Retrieve an iterator to deal with the aliases maintained inside the
1054  * library. See kmod_config_iter_get_key(), kmod_config_iter_get_value() and
1055  * kmod_config_iter_next(). At least one call to kmod_config_iter_next() must
1056  * be made to initialize the iterator and check if it's valid.
1057  *
1058  * Returns: a new iterator over the aliases or NULL on failure. Free it with
1059  * kmod_config_iter_free_iter().
1060  */
kmod_config_get_aliases(const struct kmod_ctx * ctx)1061 KMOD_EXPORT struct kmod_config_iter *kmod_config_get_aliases(const struct kmod_ctx *ctx)
1062 {
1063 	if (ctx == NULL)
1064 		return NULL;;
1065 
1066 	return kmod_config_iter_new(ctx, CONFIG_TYPE_ALIAS);
1067 }
1068 
1069 /**
1070  * kmod_config_get_options:
1071  * @ctx: kmod library context
1072  *
1073  * Retrieve an iterator to deal with the options maintained inside the
1074  * library. See kmod_config_iter_get_key(), kmod_config_iter_get_value() and
1075  * kmod_config_iter_next(). At least one call to kmod_config_iter_next() must
1076  * be made to initialize the iterator and check if it's valid.
1077  *
1078  * Returns: a new iterator over the options or NULL on failure. Free it with
1079  * kmod_config_iter_free_iter().
1080  */
kmod_config_get_options(const struct kmod_ctx * ctx)1081 KMOD_EXPORT struct kmod_config_iter *kmod_config_get_options(const struct kmod_ctx *ctx)
1082 {
1083 	if (ctx == NULL)
1084 		return NULL;;
1085 
1086 	return kmod_config_iter_new(ctx, CONFIG_TYPE_OPTION);
1087 }
1088 
1089 /**
1090  * kmod_config_get_softdeps:
1091  * @ctx: kmod library context
1092  *
1093  * Retrieve an iterator to deal with the softdeps maintained inside the
1094  * library. See kmod_config_iter_get_key(), kmod_config_iter_get_value() and
1095  * kmod_config_iter_next(). At least one call to kmod_config_iter_next() must
1096  * be made to initialize the iterator and check if it's valid.
1097  *
1098  * Returns: a new iterator over the softdeps or NULL on failure. Free it with
1099  * kmod_config_iter_free_iter().
1100  */
kmod_config_get_softdeps(const struct kmod_ctx * ctx)1101 KMOD_EXPORT struct kmod_config_iter *kmod_config_get_softdeps(const struct kmod_ctx *ctx)
1102 {
1103 	if (ctx == NULL)
1104 		return NULL;;
1105 
1106 	return kmod_config_iter_new(ctx, CONFIG_TYPE_SOFTDEP);
1107 }
1108 
1109 /**
1110  * kmod_config_iter_get_key:
1111  * @iter: iterator over a certain configuration
1112  *
1113  * When using a new allocated iterator, user must perform a call to
1114  * kmod_config_iter_next() to initialize iterator's position and check if it's
1115  * valid.
1116  *
1117  * Returns: the key of the current configuration pointed by @iter.
1118  */
kmod_config_iter_get_key(const struct kmod_config_iter * iter)1119 KMOD_EXPORT const char *kmod_config_iter_get_key(const struct kmod_config_iter *iter)
1120 {
1121 	if (iter == NULL || iter->curr == NULL)
1122 		return NULL;
1123 
1124 	return iter->get_key(iter->curr);
1125 }
1126 
1127 /**
1128  * kmod_config_iter_get_value:
1129  * @iter: iterator over a certain configuration
1130  *
1131  * When using a new allocated iterator, user must perform a call to
1132  * kmod_config_iter_next() to initialize iterator's position and check if it's
1133  * valid.
1134  *
1135  * Returns: the value of the current configuration pointed by @iter.
1136  */
kmod_config_iter_get_value(const struct kmod_config_iter * iter)1137 KMOD_EXPORT const char *kmod_config_iter_get_value(const struct kmod_config_iter *iter)
1138 {
1139 	const char *s;
1140 
1141 	if (iter == NULL || iter->curr == NULL)
1142 		return NULL;
1143 
1144 	if (iter->get_value == NULL)
1145 		return NULL;
1146 
1147 	if (iter->intermediate) {
1148 		struct kmod_config_iter *i = (struct kmod_config_iter *)iter;
1149 
1150 		free(i->data);
1151 		s = i->data = (void *) iter->get_value(iter->curr);
1152 	} else
1153 		s = iter->get_value(iter->curr);
1154 
1155 	return s;
1156 }
1157 
1158 /**
1159  * kmod_config_iter_next:
1160  * @iter: iterator over a certain configuration
1161  *
1162  * Make @iter point to the next item of a certain configuration. It's an
1163  * automatically recycling iterator. When it reaches the end, false is
1164  * returned; then if user wants to iterate again, it's sufficient to call this
1165  * function once more.
1166  *
1167  * Returns: true if next position of @iter is valid or false if its end is
1168  * reached.
1169  */
kmod_config_iter_next(struct kmod_config_iter * iter)1170 KMOD_EXPORT bool kmod_config_iter_next(struct kmod_config_iter *iter)
1171 {
1172 	if (iter == NULL)
1173 		return false;
1174 
1175 	if (iter->curr == NULL) {
1176 		iter->curr = iter->list;
1177 		return iter->curr != NULL;
1178 	}
1179 
1180 	iter->curr = kmod_list_next(iter->list, iter->curr);
1181 
1182 	return iter->curr != NULL;
1183 }
1184 
1185 /**
1186  * kmod_config_iter_free_iter:
1187  * @iter: iterator over a certain configuration
1188  *
1189  * Free resources used by the iterator.
1190  */
kmod_config_iter_free_iter(struct kmod_config_iter * iter)1191 KMOD_EXPORT void kmod_config_iter_free_iter(struct kmod_config_iter *iter)
1192 {
1193 	free(iter->data);
1194 	free(iter);
1195 }
1196