1 /*-
2 * Copyright (c) 2022 University of Cambridge
3 *
4 * This software was developed by Ararat River Consulting, LLC under
5 * sponsorship from the University of Cambridge Computer Laboratory
6 * (Department of Computer Science and Technology) and Google, Inc.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
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
30 #include <sys/param.h>
31 #include <sys/module.h>
32 #include <errno.h>
33 #include <atf-c.h>
34
35 ATF_TC_WITHOUT_HEAD(modfind);
ATF_TC_BODY(modfind,tc)36 ATF_TC_BODY(modfind, tc)
37 {
38 int modid;
39
40 /* This module is present in sys/kern/subr_bus.c. */
41 modid = modfind("rootbus");
42 ATF_REQUIRE(modid > 0);
43
44 modid = modfind("nonexistent_module");
45 ATF_REQUIRE(modid == -1);
46 ATF_REQUIRE(errno == ENOENT);
47 }
48
49 ATF_TC_WITHOUT_HEAD(modnext);
ATF_TC_BODY(modnext,tc)50 ATF_TC_BODY(modnext, tc)
51 {
52 int modid;
53
54 /* This assumes -1 is never used as a valid module id. */
55 modid = modnext(-1);
56 ATF_REQUIRE(modid == -1);
57 ATF_REQUIRE(errno == ENOENT);
58
59 modid = modnext(0);
60 ATF_REQUIRE(modid > 0);
61
62 for (;;) {
63 modid = modnext(modid);
64 ATF_REQUIRE(modid >= 0);
65 if (modid == 0)
66 break;
67 }
68 }
69
70 ATF_TC_WITHOUT_HEAD(modfnext);
ATF_TC_BODY(modfnext,tc)71 ATF_TC_BODY(modfnext, tc)
72 {
73 int modid;
74
75 /* This assumes -1 is never used as a valid module id. */
76 modid = modfnext(-1);
77 ATF_REQUIRE(modid == -1);
78 ATF_REQUIRE(errno == ENOENT);
79
80 modid = modfnext(0);
81 ATF_REQUIRE(modid == -1);
82 ATF_REQUIRE(errno == ENOENT);
83
84 modid = modnext(0);
85 ATF_REQUIRE(modid > 0);
86
87 for (;;) {
88 modid = modfnext(modid);
89 ATF_REQUIRE(modid >= 0);
90 if (modid == 0)
91 break;
92 }
93 }
94
95 ATF_TC_WITHOUT_HEAD(modstat);
ATF_TC_BODY(modstat,tc)96 ATF_TC_BODY(modstat, tc)
97 {
98 struct module_stat ms;
99 int modid;
100
101 ms.version = sizeof(ms);
102 ATF_REQUIRE(modstat(0, &ms) == -1);
103 ATF_REQUIRE(errno == ENOENT);
104
105 modid = modnext(0);
106 ATF_REQUIRE(modid > 0);
107
108 ATF_REQUIRE(modstat(modid, NULL) == -1);
109 ATF_REQUIRE(errno == EFAULT);
110
111 ms.version = 0;
112 ATF_REQUIRE(modstat(modid, &ms) == -1);
113 ATF_REQUIRE(errno == EINVAL);
114
115 ms.version = sizeof(ms);
116 ATF_REQUIRE(modstat(modid, &ms) == 0);
117 ATF_REQUIRE(ms.id == modid);
118 if (strnlen(ms.name, sizeof(ms.name)) < sizeof(ms.name))
119 ATF_REQUIRE(modfind(ms.name) == modid);
120 }
121
122 ATF_TC_WITHOUT_HEAD(modstat_v1);
ATF_TC_BODY(modstat_v1,tc)123 ATF_TC_BODY(modstat_v1, tc)
124 {
125 struct module_stat_v1 {
126 int version;
127 char name[32];
128 int refs;
129 int id;
130 } ms;
131 int modid;
132
133 ms.version = sizeof(ms);
134 ATF_REQUIRE(modstat(0, (struct module_stat *)&ms) == -1);
135 ATF_REQUIRE(errno == ENOENT);
136
137 modid = modnext(0);
138 ATF_REQUIRE(modid > 0);
139
140 ATF_REQUIRE(modstat(modid, NULL) == -1);
141 ATF_REQUIRE(errno == EFAULT);
142
143 ms.version = 0;
144 ATF_REQUIRE(modstat(modid, (struct module_stat *)&ms) == -1);
145 ATF_REQUIRE(errno == EINVAL);
146
147 ms.version = sizeof(ms);
148 ATF_REQUIRE(modstat(modid, (struct module_stat *)&ms) == 0);
149 ATF_REQUIRE(ms.id == modid);
150 if (strnlen(ms.name, sizeof(ms.name)) < sizeof(ms.name))
151 ATF_REQUIRE(modfind(ms.name) == modid);
152 }
153
154 ATF_TC_WITHOUT_HEAD(modstat_v2);
ATF_TC_BODY(modstat_v2,tc)155 ATF_TC_BODY(modstat_v2, tc)
156 {
157 struct module_stat_v2 {
158 int version;
159 char name[32];
160 int refs;
161 int id;
162 modspecific_t data;
163 } ms;
164 int modid;
165
166 ms.version = sizeof(ms);
167 ATF_REQUIRE(modstat(0, (struct module_stat *)&ms) == -1);
168 ATF_REQUIRE(errno == ENOENT);
169
170 modid = modnext(0);
171 ATF_REQUIRE(modid > 0);
172
173 ATF_REQUIRE(modstat(modid, NULL) == -1);
174 ATF_REQUIRE(errno == EFAULT);
175
176 ms.version = 0;
177 ATF_REQUIRE(modstat(modid, (struct module_stat *)&ms) == -1);
178 ATF_REQUIRE(errno == EINVAL);
179
180 ms.version = sizeof(ms);
181 ATF_REQUIRE(modstat(modid, (struct module_stat *)&ms) == 0);
182 ATF_REQUIRE(ms.id == modid);
183 if (strnlen(ms.name, sizeof(ms.name)) < sizeof(ms.name))
184 ATF_REQUIRE(modfind(ms.name) == modid);
185 }
186
ATF_TP_ADD_TCS(tp)187 ATF_TP_ADD_TCS(tp)
188 {
189 ATF_TP_ADD_TC(tp, modfind);
190 ATF_TP_ADD_TC(tp, modnext);
191 ATF_TP_ADD_TC(tp, modfnext);
192 ATF_TP_ADD_TC(tp, modstat);
193 ATF_TP_ADD_TC(tp, modstat_v1);
194 ATF_TP_ADD_TC(tp, modstat_v2);
195
196 return (atf_no_error());
197 }
198