xref: /freebsd/lib/libutil/kld.c (revision b111430e54537a66080df115c3a2950a2922da5a)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2006 Dag-Erling Coïdan Smørgrav
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer
12  *    in this position and unchanged.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  * $FreeBSD$
30  */
31 
32 #include <sys/param.h>
33 #include <sys/linker.h>
34 #include <sys/module.h>
35 
36 #include <errno.h>
37 #include <libutil.h>
38 #include <string.h>
39 
40 int
41 kld_isloaded(const char *name)
42 {
43 	struct kld_file_stat fstat;
44 	struct module_stat mstat;
45 	const char *ko;
46 	int fid, mid;
47 
48 	for (fid = kldnext(0); fid > 0; fid = kldnext(fid)) {
49 		fstat.version = sizeof(fstat);
50 		if (kldstat(fid, &fstat) != 0)
51 			continue;
52 		/* check if the file name matches the supplied name */
53 		if (strcmp(fstat.name, name) == 0)
54 			return (1);
55 		/* strip .ko and try again */
56 		if ((ko = strstr(fstat.name, ".ko")) != NULL &&
57 		    strlen(name) == (size_t)(ko - fstat.name) &&
58 		    strncmp(fstat.name, name, ko - fstat.name) == 0)
59 			return (1);
60 		/* look for a matching module within the file */
61 		for (mid = kldfirstmod(fid); mid > 0; mid = modfnext(mid)) {
62 			mstat.version = sizeof(mstat);
63 			if (modstat(mid, &mstat) != 0)
64 				continue;
65 			if (strcmp(mstat.name, name) == 0)
66 				return (1);
67 		}
68 	}
69 	return (0);
70 }
71 
72 int
73 kld_load(const char *name)
74 {
75 	if (kldload(name) == -1 && errno != EEXIST)
76 		return (-1);
77 	return (0);
78 }
79