4 * Copyright (c) 2008 The NetBSD Foundation, Inc.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
29 #include <sys/cdefs.h>
30 __COPYRIGHT("@(#) Copyright (c) 2008\
31 The NetBSD Foundation, inc. All rights reserved.");
34 #include <sys/types.h>
35 #include <sys/queue.h>
36 #include <sys/extent.h>
44 #include "../h_macros.h"
47 static struct extent
*ex
;
49 #define h_create(name, start, end, flags) \
50 ATF_REQUIRE((ex = extent_create(name, \
51 start, end, 0, 0, 0, flags)) != NULL);
53 #define h_alloc_region(start, size) \
54 ATF_REQUIRE_EQ_MSG(ret = extent_alloc_region(ex, \
55 start, size, 0), 0, "%s", strerror(ret));
57 #define h_free(start, size) \
58 ATF_REQUIRE_EQ_MSG(ret = extent_free(ex, \
59 start, size, 0), 0, "%s", strerror(ret));
62 h_alloc_subregion(u_long substart
, u_long subend
, u_long size
,
63 u_long alignment
, u_long boundary
, int expret
, u_long expres
)
67 #define FAIL(fmt, ...) \
68 atf_tc_fail("extent_alloc_subregion1(ex, %#lx, %#lx, %#lx, %#lx, 0, " \
69 "%#lx, 0, &result): " fmt, substart, subend, size, alignment, \
70 boundary, ##__VA_ARGS__)
72 ret
= extent_alloc_subregion1(ex
, substart
, subend
, size
,
73 alignment
, 0, boundary
, 0, &result
);
76 FAIL("%s", strerror(errno
));
78 if (expret
== 0 && result
!= expres
)
79 FAIL("result should be: %#lx, got: %#lx", expres
, result
);
84 h_require(const char *name
, u_long start
,
85 u_long end
, u_long flags
, const char *exp
)
88 struct extent_region
*rp
;
91 ATF_REQUIRE_STREQ_MSG(ex
->ex_name
, name
,
92 "expected: \"%s\", got: \"%s\"", name
, ex
->ex_name
);
93 ATF_REQUIRE_EQ_MSG(ex
->ex_start
, start
,
94 "expected: %#lx, got: %#lx", start
, ex
->ex_start
);
95 ATF_REQUIRE_EQ_MSG(ex
->ex_end
, end
,
96 "expected: %#lx, got: %#lx", end
, ex
->ex_end
);
97 ATF_REQUIRE_EQ_MSG(ex
->ex_flags
, flags
,
98 "expected: %#lx, got: %#lx", flags
, ex
->ex_flags
);
100 (void)memset(buf
, 0, sizeof(buf
));
101 LIST_FOREACH(rp
, &ex
->ex_regions
, er_link
)
102 n
+= snprintf(buf
+ n
, sizeof(buf
) - n
,
103 "0x%lx - 0x%lx\n", rp
->er_start
, rp
->er_end
);
105 if (strcmp(buf
, exp
) == 0)
108 printf("Incorrect extent map\n");
109 printf("Expected:\n%s\n", exp
);
110 printf("Got:\n%s\n", buf
);
111 atf_tc_fail("incorrect extent map");
115 ATF_TC_HEAD(coalesce
, tc
)
117 atf_tc_set_md_var(tc
, "descr", "Checks coalescing of regions");
119 ATF_TC_BODY(coalesce
, tc
)
121 h_create("test1", 0, 0x4f, 0);
123 h_alloc_region(0x00, 0x10);
124 h_alloc_region(0x20, 0x10);
125 h_alloc_region(0x40, 0x10);
126 h_alloc_region(0x10, 0x10);
127 h_alloc_subregion(0, 0x4f, 0x10, EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0x30);
129 h_require("test1", 0x00, 0x4f, 0x00,
136 ATF_TC_HEAD(subregion1
, tc
)
138 atf_tc_set_md_var(tc
, "descr",
139 "Checks that subregions work (PR kern/7539)");
141 ATF_TC_BODY(subregion1
, tc
)
143 h_create("test2", 0, 0x2f, EX_NOCOALESCE
);
145 h_alloc_region(0x00, 0x10);
146 h_alloc_subregion(0x20, 0x30, 0x10, EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0x20);
148 h_require("test2", 0x00, 0x2f, 0x2,
156 ATF_TC_HEAD(subregion2
, tc
)
158 atf_tc_set_md_var(tc
, "descr",
159 "Checks that subregion allocations don't overlap with existing "
160 "ones (fixed in 1.25)");
162 ATF_TC_BODY(subregion2
, tc
)
164 h_create("test3", 0, 0x3f, EX_NOCOALESCE
);
166 h_alloc_region(0x00, 0x20);
167 h_alloc_region(0x30, 0x10);
168 h_alloc_subregion(0x10, 0x3f, 0x10,
169 EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0x20);
171 h_require("test3", 0x00, 0x3f, 0x2,
180 ATF_TC_HEAD(bound1
, tc
)
182 atf_tc_set_md_var(tc
, "descr",
183 "Checks for overflow in boundary check, before an allocated region "
186 ATF_TC_BODY(bound1
, tc
)
188 h_create("test4", 0xf0000000, 0xffffffff, 0);
190 h_alloc_region(0xf1000000, 0x1);
191 h_alloc_subregion(0xf0000000, 0xffffffff, 0x1,
192 EX_NOALIGN
, 0x20000000, 0, 0xf0000000);
194 h_require("test4", 0xf0000000, 0xffffffff, 0x0,
195 "0xf0000000 - 0xf0000000\n"
196 "0xf1000000 - 0xf1000000\n");
202 ATF_TC_HEAD(bound2
, tc
)
204 atf_tc_set_md_var(tc
, "descr",
205 "Checks for overflow in boundary checks, before the subregion end "
208 ATF_TC_BODY(bound2
, tc
)
210 h_create("test5", 0xf0000000, 0xffffffff, 0);
212 h_alloc_subregion(0xf0000000, 0xffffffff, 0x1,
213 EX_NOALIGN
, 0x20000000, 0, 0xf0000000);
215 h_require("test5", 0xf0000000, 0xffffffff, 0x0,
216 "0xf0000000 - 0xf0000000\n");
222 ATF_TC_HEAD(bound3
, tc
)
224 atf_tc_set_md_var(tc
, "descr",
225 "Checks allocation beyond last boundary line: last two "
226 "allocations should succeed without boundary \"fixups\"");
228 ATF_TC_BODY(bound3
, tc
)
230 h_create("test6", 0, 11, 0);
232 h_alloc_subregion(0, 11, 8, EX_NOALIGN
, 8, 0, 0);
233 h_alloc_subregion(0, 11, 2, EX_NOALIGN
, 8, 0, 0x8);
234 h_alloc_subregion(0, 11, 2, EX_NOALIGN
, 8, 0, 0xa);
236 h_require("test6", 0x0, 0xb, 0x0, "0x0 - 0xb\n");
242 ATF_TC_HEAD(bound4
, tc
)
244 atf_tc_set_md_var(tc
, "descr",
245 "Checks allocation beyond last boundary line: last allocation "
246 "should be bumped to the next boundary and exactly fit the "
249 ATF_TC_BODY(bound4
, tc
)
251 h_create("test7", 0, 11, 0);
253 h_alloc_subregion(0, 11, 7, EX_NOALIGN
, 8, 0, 0);
254 h_alloc_subregion(0, 11, 4, EX_NOALIGN
, 8, 0, 8);
256 h_require("test7", 0x0, 0xb, 0x0,
264 ATF_TC_HEAD(subregion3
, tc
)
266 atf_tc_set_md_var(tc
, "descr",
267 "Checks that we don't allocate a region pasts the end of "
268 "subregion (i.e., the second alloc_subregion should fail). "
269 "subr_extent.c prior to rev. 1.43 allocated region starting "
272 ATF_TC_BODY(subregion3
, tc
)
274 h_create("test8", 0, 0x4f, EX_NOCOALESCE
);
276 h_alloc_region(0x30, 0x10);
277 h_alloc_subregion(0, 0xf, 0x10, EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0);
278 h_alloc_subregion(0, 0xf, 0x10, EX_NOALIGN
, EX_NOBOUNDARY
, EAGAIN
, 0);
280 h_require("test8", 0x0, 0x4f, 0x2,
288 ATF_TC_HEAD(bound5
, tc
)
290 atf_tc_set_md_var(tc
, "descr",
291 "When allocating a region with a boundary constraint, checks "
292 "proper detection of overflaps once the candidate region has "
293 "been aligned. subr_extent.c prior 1.45 could corrupt the extent "
294 "map in this situation");
296 ATF_TC_BODY(bound5
, tc
)
298 h_create("test9", 0, 0x4f, 0);
300 h_alloc_subregion(0, 0x10, 4, EX_NOALIGN
, 0, 0, 0);
301 h_alloc_subregion(0xd, 0x20, 2, EX_NOALIGN
, 0, 0, 0xd);
302 h_alloc_subregion(0, 0x4f, 8, EX_NOALIGN
, 8, 0, 0x10);
304 h_require("test9", 0x0, 0x4f, 0x0,
313 ATF_TC_HEAD(free
, tc
)
315 atf_tc_set_md_var(tc
, "descr", "Checks extent_free()");
317 ATF_TC_BODY(free
, tc
)
319 h_create("test10", 0xc0002000, 0xffffe000, EX_BOUNDZERO
);
321 h_alloc_subregion(0xc0002000, 0xffffe000, 0x2000,
322 0x10000, 0x10000, 0, 0xc0010000);
323 h_alloc_subregion(0xc0002000, 0xffffe000, 0x2000,
324 0x10000, 0x10000, 0, 0xc0020000);
326 h_require("test10", 0xc0002000, 0xffffe000, 0x0,
327 "0xc0010000 - 0xc0011fff\n"
328 "0xc0020000 - 0xc0021fff\n");
330 h_free(0xc0020000, 0x2000);
331 h_require("test10", 0xc0002000, 0xffffe000, 0x0,
332 "0xc0010000 - 0xc0011fff\n");
334 h_alloc_subregion(0xc0002000, 0xffffe000, 0x10000,
335 0x10000, 0x10000, 0, 0xc0022000);
337 h_require("test10", 0xc0002000, 0xffffe000, 0x0,
338 "0xc0010000 - 0xc0011fff\n"
339 "0xc0022000 - 0xc0031fff\n");
345 ATF_TC_HEAD(subregion4
, tc
)
347 atf_tc_set_md_var(tc
, "descr",
348 "Checks for off-by-one bug which would cause a region at the end "
349 "of the extent to be allocated multiple times (fixed in 1.51)");
351 ATF_TC_BODY(subregion4
, tc
)
353 h_create("test11", 0x10, 0x20, EX_NOCOALESCE
);
355 h_alloc_subregion(0x10, 0x13, 0x4, EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0x10);
356 h_alloc_subregion(0x1e, 0x1f, 0x2, EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0x1e);
357 h_alloc_subregion(0x20, 0x20, 0x1, EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0x20);
358 h_alloc_subregion(0x20, 0x20, 0x1, EX_NOALIGN
, EX_NOBOUNDARY
, EAGAIN
, 0);
359 h_alloc_subregion(0x10, 0x20, 0x1, EX_NOALIGN
, EX_NOBOUNDARY
, 0, 0x14);
361 h_require("test11", 0x10, 0x20, 0x2,
372 ATF_TP_ADD_TC(tp
, coalesce
);
373 ATF_TP_ADD_TC(tp
, subregion1
);
374 ATF_TP_ADD_TC(tp
, subregion2
);
375 ATF_TP_ADD_TC(tp
, bound1
);
376 ATF_TP_ADD_TC(tp
, bound2
);
377 ATF_TP_ADD_TC(tp
, bound3
);
378 ATF_TP_ADD_TC(tp
, bound4
);
379 ATF_TP_ADD_TC(tp
, subregion3
);
380 ATF_TP_ADD_TC(tp
, bound5
);
381 ATF_TP_ADD_TC(tp
, free
);
382 ATF_TP_ADD_TC(tp
, subregion4
);
384 return atf_no_error();