1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
|
/* SPDX-License-Identifier: LGPL-2.1-or-later */
#include <stddef.h>
#include "bitfield.h"
#include "log.h"
#include "tests.h"
#define TEST_BITS(bits, v, ...) \
({ \
assert_se((!!BITS_SET(bits, ##__VA_ARGS__)) == v); \
assert_se((!!BITS_SET(~(bits), ##__VA_ARGS__)) == !v); \
})
#define TEST_BIT(bits, v, i) \
({ \
assert_se((!!BIT_SET(bits, i)) == v); \
assert_se((!!BIT_SET(~(bits), i)) == !v); \
TEST_BITS(bits, v, i); \
})
#define TEST_BIT_SET(bits, i) TEST_BIT(bits, 1, i)
#define TEST_BIT_CLEAR(bits, i) TEST_BIT(bits, 0, i)
#define TEST_BITS_SET(bits, ...) TEST_BITS(bits, 1, ##__VA_ARGS__)
#define TEST_BITS_CLEAR(bits, ...) TEST_BITS(bits, 0, ##__VA_ARGS__)
TEST(bits) {
int count;
/* Test uint8_t */
TEST_BIT_SET(0x81, 0);
TEST_BIT_SET(0x81, 7);
TEST_BITS_SET(0x81, 0, 7);
TEST_BIT_CLEAR(0x81, 4);
TEST_BIT_CLEAR(0x81, 6);
TEST_BITS_CLEAR(0x81, 1, 2, 3, 4, 5, 6);
uint8_t expected8 = 0;
BIT_FOREACH(i, 0x81)
expected8 |= UINT8_C(1) << i;
assert_se(expected8 == 0x81);
uint8_t u8 = 0x91;
TEST_BIT_SET(u8, 4);
TEST_BITS_SET(u8, 0, 4, 7);
TEST_BIT_CLEAR(u8, 2);
TEST_BITS_CLEAR(u8, 1, 2, 3, 5, 6);
SET_BIT(u8, 1);
TEST_BITS_SET(u8, 0, 1, 4, 7);
TEST_BITS_CLEAR(u8, 2, 3, 5, 6);
SET_BITS(u8, 3, 5);
TEST_BITS_SET(u8, 0, 1, 3, 4, 5, 7);
TEST_BITS_CLEAR(u8, 2, 6);
CLEAR_BIT(u8, 4);
TEST_BITS_SET(u8, 0, 1, 3, 5, 7);
TEST_BITS_CLEAR(u8, 2, 4, 6);
CLEAR_BITS(u8, 1);
CLEAR_BITS(u8, 0, 7);
TEST_BITS_SET(u8, 3, 5);
TEST_BITS_CLEAR(u8, 0, 1, 2, 4, 6, 7);
expected8 = 0;
BIT_FOREACH(i, u8)
expected8 |= UINT8_C(1) << i;
assert_se(expected8 == u8);
u8 = 0;
TEST_BITS_CLEAR(u8, 0, 1, 2, 3, 4, 5, 6, 7);
BIT_FOREACH(i, u8)
assert_se(0);
u8 = ~u8;
TEST_BITS_SET(u8, 0, 1, 2, 3, 4, 5, 6, 7);
count = 0;
BIT_FOREACH(i, u8)
count++;
assert_se(count == 8);
uint8_t _u8 = u8;
SET_BITS(u8);
assert_se(_u8 == u8);
CLEAR_BITS(u8);
assert_se(_u8 == u8);
/* Test uint16_t */
TEST_BIT_SET(0x1f81, 10);
TEST_BITS_SET(0x1f81, 0, 7, 8, 9, 10, 11, 12);
TEST_BIT_CLEAR(0x1f81, 13);
TEST_BITS_CLEAR(0x1f81, 1, 2, 3, 4, 5, 6, 13, 14, 15);
uint16_t expected16 = 0;
BIT_FOREACH(i, 0x1f81)
expected16 |= UINT16_C(1) << i;
assert_se(expected16 == 0x1f81);
uint16_t u16 = 0xf060;
TEST_BIT_SET(u16, 12);
TEST_BITS_SET(u16, 5, 6, 12, 13, 14, 15);
TEST_BIT_CLEAR(u16, 9);
TEST_BITS_CLEAR(u16, 0, 1, 2, 3, 4, 7, 8, 9, 10, 11);
SET_BITS(u16, 1, 8);
TEST_BITS_SET(u16, 1, 5, 6, 8, 12, 13, 14, 15);
TEST_BITS_CLEAR(u16, 0, 2, 3, 4, 7, 9, 10, 11);
CLEAR_BITS(u16, 13, 14);
TEST_BITS_SET(u16, 1, 5, 6, 8, 12, 15);
TEST_BITS_CLEAR(u16, 0, 2, 3, 4, 7, 9, 10, 11, 13, 14);
expected16 = 0;
BIT_FOREACH(i, u16)
expected16 |= UINT16_C(1) << i;
assert_se(expected16 == u16);
u16 = 0;
TEST_BITS_CLEAR(u16, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
BIT_FOREACH(i, u16)
assert_se(0);
u16 = ~u16;
TEST_BITS_SET(u16, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
count = 0;
BIT_FOREACH(i, u16)
count++;
assert_se(count == 16);
uint16_t _u16 = u16;
SET_BITS(u16);
assert_se(_u16 == u16);
CLEAR_BITS(u16);
assert_se(_u16 == u16);
/* Test uint32_t */
TEST_BIT_SET(0x80224f10, 11);
TEST_BITS_SET(0x80224f10, 4, 8, 9, 10, 11, 14, 17, 21, 31);
TEST_BIT_CLEAR(0x80224f10, 28);
TEST_BITS_CLEAR(0x80224f10, 0, 1, 2, 3, 5, 6, 7, 12, 13, 15, 16, 18, 19, 20, 22, 23, 24, 25, 26, 27, 28, 29, 30);
uint32_t expected32 = 0;
BIT_FOREACH(i, 0x80224f10)
expected32 |= UINT32_C(1) << i;
assert_se(expected32 == 0x80224f10);
uint32_t u32 = 0x605e0388;
TEST_BIT_SET(u32, 3);
TEST_BIT_SET(u32, 30);
TEST_BITS_SET(u32, 3, 7, 8, 9, 17, 18, 19, 20, 22, 29, 30);
TEST_BIT_CLEAR(u32, 0);
TEST_BIT_CLEAR(u32, 31);
TEST_BITS_CLEAR(u32, 0, 1, 2, 4, 5, 6, 10, 11, 12, 13, 14, 15, 16, 21, 23, 24, 25, 26, 27, 28, 31);
SET_BITS(u32, 1, 25, 26);
TEST_BITS_SET(u32, 1, 3, 7, 8, 9, 17, 18, 19, 20, 22, 25, 26, 29, 30);
TEST_BITS_CLEAR(u32, 0, 2, 4, 5, 6, 10, 11, 12, 13, 14, 15, 16, 21, 23, 24, 27, 28, 31);
CLEAR_BITS(u32, 29, 17, 1);
TEST_BITS_SET(u32, 3, 7, 8, 9, 18, 19, 20, 22, 25, 26, 30);
TEST_BITS_CLEAR(u32, 0, 1, 2, 4, 5, 6, 10, 11, 12, 13, 14, 15, 16, 17, 21, 23, 24, 27, 28, 29, 31);
expected32 = 0;
BIT_FOREACH(i, u32)
expected32 |= UINT32_C(1) << i;
assert_se(expected32 == u32);
u32 = 0;
TEST_BITS_CLEAR(u32, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31);
BIT_FOREACH(i, u32)
assert_se(0);
u32 = ~u32;
TEST_BITS_SET(u32, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31);
count = 0;
BIT_FOREACH(i, u32)
count++;
assert_se(count == 32);
uint32_t _u32 = u32;
SET_BITS(u32);
assert_se(_u32 == u32);
CLEAR_BITS(u32);
assert_se(_u32 == u32);
/* Test uint64_t */
TEST_BIT_SET(0x18ba1400f4857460, 60);
TEST_BITS_SET(0x18ba1400f4857460, 5, 6, 10, 12, 13, 14, 16, 18, 23, 26, 28, 29, 30, 31, 42, 44, 49, 51, 52, 53, 55, 59, 60);
TEST_BIT_CLEAR(UINT64_C(0x18ba1400f4857460), 0);
TEST_BIT_CLEAR(UINT64_C(0x18ba1400f4857460), 63);
TEST_BITS_CLEAR(UINT64_C(0x18ba1400f4857460), 0, 1, 2, 3, 4, 7, 8, 9, 11, 15, 17, 19, 20, 21, 22, 24, 25, 27, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 43, 45, 46, 47, 48, 50, 54, 56, 57, 58, 61, 62, 63);
uint64_t expected64 = 0;
BIT_FOREACH(i, 0x18ba1400f4857460)
expected64 |= UINT64_C(1) << i;
assert_se(expected64 == 0x18ba1400f4857460);
uint64_t u64 = 0xa90e2d8507a65739;
TEST_BIT_SET(u64, 0);
TEST_BIT_SET(u64, 63);
TEST_BITS_SET(u64, 0, 3, 4, 5, 8, 9, 10, 12, 14, 17, 18, 21, 23, 24, 25, 26, 32, 34, 39, 40, 42, 43, 45, 49, 50, 51, 56, 59, 61, 63);
TEST_BIT_CLEAR(u64, 1);
TEST_BITS_CLEAR(u64, 1, 2, 6, 7, 11, 13, 15, 16, 19, 20, 22, 27, 28, 29, 30, 31, 33, 35, 36, 37, 38, 41, 44, 46, 47, 48, 52, 53, 54, 55, 57, 58, 60, 62);
SET_BIT(u64, 1);
TEST_BITS_SET(u64, 0, 1, 3, 4, 5, 8, 9, 10, 12, 14, 17, 18, 21, 23, 24, 25, 26, 32, 34, 39, 40, 42, 43, 45, 49, 50, 51, 56, 59, 61, 63);
TEST_BITS_CLEAR(u64, 2, 6, 7, 11, 13, 15, 16, 19, 20, 22, 27, 28, 29, 30, 31, 33, 35, 36, 37, 38, 41, 44, 46, 47, 48, 52, 53, 54, 55, 57, 58, 60, 62);
CLEAR_BIT(u64, 63);
TEST_BITS_SET(u64, 0, 1, 3, 4, 5, 8, 9, 10, 12, 14, 17, 18, 21, 23, 24, 25, 26, 32, 34, 39, 40, 42, 43, 45, 49, 50, 51, 56, 59, 61);
TEST_BITS_CLEAR(u64, 2, 6, 7, 11, 13, 15, 16, 19, 20, 22, 27, 28, 29, 30, 31, 33, 35, 36, 37, 38, 41, 44, 46, 47, 48, 52, 53, 54, 55, 57, 58, 60, 62, 63);
SET_BIT(u64, 62);
TEST_BITS_SET(u64, 0, 1, 3, 4, 5, 8, 9, 10, 12, 14, 17, 18, 21, 23, 24, 25, 26, 32, 34, 39, 40, 42, 43, 45, 49, 50, 51, 56, 59, 61, 62);
TEST_BITS_CLEAR(u64, 2, 6, 7, 11, 13, 15, 16, 19, 20, 22, 27, 28, 29, 30, 31, 33, 35, 36, 37, 38, 41, 44, 46, 47, 48, 52, 53, 54, 55, 57, 58, 60, 63);
SET_BITS(u64, 63, 62, 7, 13, 38, 40);
TEST_BITS_SET(u64, 0, 1, 3, 4, 5, 7, 8, 9, 10, 12, 13, 14, 17, 18, 21, 23, 24, 25, 26, 32, 34, 38, 39, 40, 42, 43, 45, 49, 50, 51, 56, 59, 61, 62, 63);
TEST_BITS_CLEAR(u64, 2, 6, 11, 15, 16, 19, 20, 22, 27, 28, 29, 30, 31, 33, 35, 36, 37, 41, 44, 46, 47, 48, 52, 53, 54, 55, 57, 58, 60);
CLEAR_BIT(u64, 32);
TEST_BITS_SET(u64, 0, 1, 3, 4, 5, 7, 8, 9, 10, 12, 13, 14, 17, 18, 21, 23, 24, 25, 26, 34, 38, 39, 40, 42, 43, 45, 49, 50, 51, 56, 59, 61, 62, 63);
TEST_BITS_CLEAR(u64, 2, 6, 11, 15, 16, 19, 20, 22, 27, 28, 29, 30, 31, 32, 33, 35, 36, 37, 41, 44, 46, 47, 48, 52, 53, 54, 55, 57, 58, 60);
CLEAR_BITS(u64, 0, 2, 11, 63, 32, 58);
TEST_BITS_SET(u64, 1, 3, 4, 5, 7, 8, 9, 10, 12, 13, 14, 17, 18, 21, 23, 24, 25, 26, 34, 38, 39, 40, 42, 43, 45, 49, 50, 51, 56, 59, 61, 62);
TEST_BITS_CLEAR(u64, 0, 2, 6, 11, 15, 16, 19, 20, 22, 27, 28, 29, 30, 31, 32, 33, 35, 36, 37, 41, 44, 46, 47, 48, 52, 53, 54, 55, 57, 58, 60, 63);
expected64 = 0;
BIT_FOREACH(i, u64)
expected64 |= UINT64_C(1) << i;
assert_se(expected64 == u64);
u64 = 0;
TEST_BITS_CLEAR(u64, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63);
BIT_FOREACH(i, u64)
assert_se(0);
u64 = ~u64;
TEST_BITS_SET(u64, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63);
count = 0;
BIT_FOREACH(i, u64)
count++;
assert_se(count == 64);
uint64_t _u64 = u64;
SET_BITS(u64);
assert_se(_u64 == u64);
CLEAR_BITS(u64);
assert_se(_u64 == u64);
/* Verify these use cases are constant-folded. */
assert_cc(__builtin_constant_p(INDEX_TO_MASK(uint8_t, 1)));
assert_cc(__builtin_constant_p(INDEX_TO_MASK(uint16_t, 1)));
assert_cc(__builtin_constant_p(INDEX_TO_MASK(uint32_t, 1)));
assert_cc(__builtin_constant_p(INDEX_TO_MASK(uint64_t, 1)));
assert_cc(__builtin_constant_p(BIT_SET((uint8_t)2, 1)));
assert_cc(__builtin_constant_p(BIT_SET((uint16_t)2, 1)));
assert_cc(__builtin_constant_p(BIT_SET((uint32_t)2, 1)));
assert_cc(__builtin_constant_p(BIT_SET((uint64_t)2, 1)));
}
DEFINE_TEST_MAIN(LOG_INFO);
|
