xref: /freebsd/contrib/netbsd-tests/modules/k_helper3/k_helper3.c (revision a0ee8cc636cd5c2374ec44ca71226564ea0bca95)
1 /*	$NetBSD: k_helper3.c,v 1.3 2012/04/17 21:39:19 martin Exp $ */
2 
3 /*-
4  * Copyright (c) 2012 The NetBSD Foundation, Inc.
5  * All rights reserved.
6  *
7  * This code is derived from software contributed to The NetBSD Foundation
8  * by Jukka Ruohonen.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  */
31 #include <sys/cdefs.h>
32 __RCSID("$NetBSD: k_helper3.c,v 1.3 2012/04/17 21:39:19 martin Exp $");
33 
34 #include <sys/module.h>
35 
36 #include <assert.h>
37 #include <errno.h>
38 #include <stdarg.h>
39 #include <stdio.h>
40 #include <stdlib.h>
41 #include <string.h>
42 
43 #include <prop/proplib.h>
44 
45 static int load(const char *, ...);
46 
47 /*
48  * A program that loads a module and returns the errno(2) from modctl(8).
49  */
50 int
51 main(int argc, char *argv[])
52 {
53 	assert(argc == 3);
54 
55 	return load(argv[1], argv[2]);
56 }
57 
58 static __printflike(1, 2) int
59 load(const char *fmt, ...)
60 {
61 	char filename[MAXPATHLEN], *propsstr, *shortname, *dot;
62 	prop_dictionary_t props;
63 	modctl_load_t ml;
64 	int serrno, rv;
65 	va_list ap;
66 
67 	props = prop_dictionary_create();
68 	propsstr = prop_dictionary_externalize(props);
69 	assert(propsstr != NULL);
70 	prop_object_release(props);
71 
72 	va_start(ap, fmt);
73 	(void)vsnprintf(filename, sizeof(filename), fmt, ap);
74 	va_end(ap);
75 
76 	ml.ml_filename = filename;
77 	ml.ml_flags = 0;
78 	ml.ml_props = propsstr;
79 	ml.ml_propslen = strlen(propsstr);
80 
81 	printf("Loading module %s\n", filename);
82 	errno = serrno = 0;
83 
84 	rv = modctl(MODCTL_LOAD, &ml);
85 
86 	if (rv != 0)
87 		serrno = errno;
88 
89 	shortname = strrchr(filename, '/');
90 	if (shortname != NULL)
91 		shortname++;
92 	else
93 		shortname = filename;
94 	dot = strrchr(shortname, '.');
95 	if (dot)
96 		*dot = 0;
97 	(void)modctl(MODCTL_UNLOAD, shortname);
98 
99 	free(propsstr);
100 
101 	return serrno;
102 }
103