1/*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2021 The FreeBSD Foundation
5 *
6 * This software was developed by Mark Johnston under sponsorship from
7 * the FreeBSD Foundation.
8 */
9
10#define _WANT_FREEBSD_BITSET
11
12#include <sys/types.h>
13#include <sys/_bitset.h>
14#include <sys/bitset.h>
15#include <stdio.h>
16#include <stdlib.h>
17
18#include <atf-c.h>
19
20BITSET_DEFINE(bs256, 256);
21
22ATF_TC_WITHOUT_HEAD(bit_foreach);
23ATF_TC_BODY(bit_foreach, tc)
24{
25	struct bs256 bs0, bs1, bsrand;
26	int setc, clrc, i;
27
28#define	_BIT_FOREACH_COUNT(s, bs) do {					\
29	int prev = -1;							\
30	setc = clrc = 0;						\
31	BIT_FOREACH_ISSET((s), i, (bs)) {				\
32		ATF_REQUIRE_MSG(prev < i, "incorrect bit ordering");	\
33		ATF_REQUIRE_MSG(BIT_ISSET((s), i, (bs)),		\
34		    "bit %d is not set", i);				\
35		setc++;							\
36		prev = i;						\
37	}								\
38	prev = -1;							\
39	BIT_FOREACH_ISCLR((s), i, (bs)) {				\
40		ATF_REQUIRE_MSG(prev < i, "incorrect bit ordering");	\
41		ATF_REQUIRE_MSG(!BIT_ISSET((s), i, (bs)),		\
42		    "bit %d is set", i);				\
43		clrc++;							\
44		prev = i;						\
45	}								\
46} while (0)
47
48	/*
49	 * Create several bitsets, and for each one count the number
50	 * of set and clear bits and make sure they match what we expect.
51	 */
52
53	BIT_FILL(256, &bs1);
54	_BIT_FOREACH_COUNT(256, &bs1);
55	ATF_REQUIRE_MSG(setc == 256, "incorrect set count %d", setc);
56	ATF_REQUIRE_MSG(clrc == 0, "incorrect clear count %d", clrc);
57
58	BIT_ZERO(256, &bs0);
59	_BIT_FOREACH_COUNT(256, &bs0);
60	ATF_REQUIRE_MSG(setc == 0, "incorrect set count %d", setc);
61	ATF_REQUIRE_MSG(clrc == 256, "incorrect clear count %d", clrc);
62
63	BIT_ZERO(256, &bsrand);
64	for (i = 0; i < 256; i++)
65		if (random() % 2 != 0)
66			BIT_SET(256, i, &bsrand);
67	_BIT_FOREACH_COUNT(256, &bsrand);
68	ATF_REQUIRE_MSG(setc + clrc == 256, "incorrect counts %d, %d",
69	    setc, clrc);
70
71	/*
72	 * Try to verify that we can safely clear bits in the set while
73	 * iterating.
74	 */
75	BIT_FOREACH_ISSET(256, i, &bsrand) {
76		ATF_REQUIRE(setc-- > 0);
77		BIT_CLR(256, i, &bsrand);
78	}
79	_BIT_FOREACH_COUNT(256, &bsrand);
80	ATF_REQUIRE_MSG(setc == 0, "incorrect set count %d", setc);
81	ATF_REQUIRE_MSG(clrc == 256, "incorrect clear count %d", clrc);
82
83#undef _BIT_FOREACH_COUNT
84}
85
86ATF_TP_ADD_TCS(tp)
87{
88	ATF_TP_ADD_TC(tp, bit_foreach);
89	return (atf_no_error());
90}
91