]>
Commit | Line | Data |
---|---|---|
1 | /* SPDX-License-Identifier: LGPL-2.1+ */ | |
2 | /*** | |
3 | Copyright 2013 Thomas H.P. Andersen | |
4 | ***/ | |
5 | ||
6 | #include <errno.h> | |
7 | #include <string.h> | |
8 | #include <sys/wait.h> | |
9 | #include <unistd.h> | |
10 | ||
11 | #include "def.h" | |
12 | #include "fileio.h" | |
13 | #include "fs-util.h" | |
14 | #include "parse-util.h" | |
15 | #include "process-util.h" | |
16 | #include "raw-clone.h" | |
17 | #include "rm-rf.h" | |
18 | #include "string-util.h" | |
19 | #include "util.h" | |
20 | ||
21 | static void test_align_power2(void) { | |
22 | unsigned long i, p2; | |
23 | ||
24 | assert_se(ALIGN_POWER2(0) == 0); | |
25 | assert_se(ALIGN_POWER2(1) == 1); | |
26 | assert_se(ALIGN_POWER2(2) == 2); | |
27 | assert_se(ALIGN_POWER2(3) == 4); | |
28 | assert_se(ALIGN_POWER2(12) == 16); | |
29 | ||
30 | assert_se(ALIGN_POWER2(ULONG_MAX) == 0); | |
31 | assert_se(ALIGN_POWER2(ULONG_MAX - 1) == 0); | |
32 | assert_se(ALIGN_POWER2(ULONG_MAX - 1024) == 0); | |
33 | assert_se(ALIGN_POWER2(ULONG_MAX / 2) == ULONG_MAX / 2 + 1); | |
34 | assert_se(ALIGN_POWER2(ULONG_MAX + 1) == 0); | |
35 | ||
36 | for (i = 1; i < 131071; ++i) { | |
37 | for (p2 = 1; p2 < i; p2 <<= 1) | |
38 | /* empty */ ; | |
39 | ||
40 | assert_se(ALIGN_POWER2(i) == p2); | |
41 | } | |
42 | ||
43 | for (i = ULONG_MAX - 1024; i < ULONG_MAX; ++i) { | |
44 | for (p2 = 1; p2 && p2 < i; p2 <<= 1) | |
45 | /* empty */ ; | |
46 | ||
47 | assert_se(ALIGN_POWER2(i) == p2); | |
48 | } | |
49 | } | |
50 | ||
51 | static void test_max(void) { | |
52 | static const struct { | |
53 | int a; | |
54 | int b[CONST_MAX(10, 100)]; | |
55 | } val1 = { | |
56 | .a = CONST_MAX(10, 100), | |
57 | }; | |
58 | int d = 0; | |
59 | ||
60 | assert_cc(sizeof(val1.b) == sizeof(int) * 100); | |
61 | ||
62 | /* CONST_MAX returns (void) instead of a value if the passed arguments | |
63 | * are not of the same type or not constant expressions. */ | |
64 | assert_cc(__builtin_types_compatible_p(typeof(CONST_MAX(1, 10)), int)); | |
65 | assert_cc(__builtin_types_compatible_p(typeof(CONST_MAX(1, 1U)), void)); | |
66 | ||
67 | assert_se(val1.a == 100); | |
68 | assert_se(MAX(++d, 0) == 1); | |
69 | assert_se(d == 1); | |
70 | ||
71 | assert_cc(MAXSIZE(char[3], uint16_t) == 3); | |
72 | assert_cc(MAXSIZE(char[3], uint32_t) == 4); | |
73 | assert_cc(MAXSIZE(char, long) == sizeof(long)); | |
74 | ||
75 | assert_se(MAX(-5, 5) == 5); | |
76 | assert_se(MAX(5, 5) == 5); | |
77 | assert_se(MAX(MAX(1, MAX(2, MAX(3, 4))), 5) == 5); | |
78 | assert_se(MAX(MAX(1, MAX(2, MAX(3, 2))), 1) == 3); | |
79 | assert_se(MAX(MIN(1, MIN(2, MIN(3, 4))), 5) == 5); | |
80 | assert_se(MAX(MAX(1, MIN(2, MIN(3, 2))), 1) == 2); | |
81 | assert_se(LESS_BY(8, 4) == 4); | |
82 | assert_se(LESS_BY(8, 8) == 0); | |
83 | assert_se(LESS_BY(4, 8) == 0); | |
84 | assert_se(LESS_BY(16, LESS_BY(8, 4)) == 12); | |
85 | assert_se(LESS_BY(4, LESS_BY(8, 4)) == 0); | |
86 | assert_se(CLAMP(-5, 0, 1) == 0); | |
87 | assert_se(CLAMP(5, 0, 1) == 1); | |
88 | assert_se(CLAMP(5, -10, 1) == 1); | |
89 | assert_se(CLAMP(5, -10, 10) == 5); | |
90 | assert_se(CLAMP(CLAMP(0, -10, 10), CLAMP(-5, 10, 20), CLAMP(100, -5, 20)) == 10); | |
91 | } | |
92 | ||
93 | #pragma GCC diagnostic push | |
94 | #ifdef __clang__ | |
95 | # pragma GCC diagnostic ignored "-Waddress-of-packed-member" | |
96 | #endif | |
97 | ||
98 | static void test_container_of(void) { | |
99 | struct mytype { | |
100 | uint8_t pad1[3]; | |
101 | uint64_t v1; | |
102 | uint8_t pad2[2]; | |
103 | uint32_t v2; | |
104 | } _packed_ myval = { }; | |
105 | ||
106 | assert_cc(sizeof(myval) == 17); | |
107 | assert_se(container_of(&myval.v1, struct mytype, v1) == &myval); | |
108 | assert_se(container_of(&myval.v2, struct mytype, v2) == &myval); | |
109 | assert_se(container_of(&container_of(&myval.v2, | |
110 | struct mytype, | |
111 | v2)->v1, | |
112 | struct mytype, | |
113 | v1) == &myval); | |
114 | } | |
115 | ||
116 | #pragma GCC diagnostic pop | |
117 | ||
118 | static void test_div_round_up(void) { | |
119 | int div; | |
120 | ||
121 | /* basic tests */ | |
122 | assert_se(DIV_ROUND_UP(0, 8) == 0); | |
123 | assert_se(DIV_ROUND_UP(1, 8) == 1); | |
124 | assert_se(DIV_ROUND_UP(8, 8) == 1); | |
125 | assert_se(DIV_ROUND_UP(12, 8) == 2); | |
126 | assert_se(DIV_ROUND_UP(16, 8) == 2); | |
127 | ||
128 | /* test multiple evaluation */ | |
129 | div = 0; | |
130 | assert_se(DIV_ROUND_UP(div++, 8) == 0 && div == 1); | |
131 | assert_se(DIV_ROUND_UP(++div, 8) == 1 && div == 2); | |
132 | assert_se(DIV_ROUND_UP(8, div++) == 4 && div == 3); | |
133 | assert_se(DIV_ROUND_UP(8, ++div) == 2 && div == 4); | |
134 | ||
135 | /* overflow test with exact division */ | |
136 | assert_se(sizeof(0U) == 4); | |
137 | assert_se(0xfffffffaU % 10U == 0U); | |
138 | assert_se(0xfffffffaU / 10U == 429496729U); | |
139 | assert_se(DIV_ROUND_UP(0xfffffffaU, 10U) == 429496729U); | |
140 | assert_se((0xfffffffaU + 10U - 1U) / 10U == 0U); | |
141 | assert_se(0xfffffffaU / 10U + !!(0xfffffffaU % 10U) == 429496729U); | |
142 | ||
143 | /* overflow test with rounded division */ | |
144 | assert_se(0xfffffffdU % 10U == 3U); | |
145 | assert_se(0xfffffffdU / 10U == 429496729U); | |
146 | assert_se(DIV_ROUND_UP(0xfffffffdU, 10U) == 429496730U); | |
147 | assert_se((0xfffffffdU + 10U - 1U) / 10U == 0U); | |
148 | assert_se(0xfffffffdU / 10U + !!(0xfffffffdU % 10U) == 429496730U); | |
149 | } | |
150 | ||
151 | static void test_u64log2(void) { | |
152 | assert_se(u64log2(0) == 0); | |
153 | assert_se(u64log2(8) == 3); | |
154 | assert_se(u64log2(9) == 3); | |
155 | assert_se(u64log2(15) == 3); | |
156 | assert_se(u64log2(16) == 4); | |
157 | assert_se(u64log2(1024*1024) == 20); | |
158 | assert_se(u64log2(1024*1024+5) == 20); | |
159 | } | |
160 | ||
161 | static void test_protect_errno(void) { | |
162 | errno = 12; | |
163 | { | |
164 | PROTECT_ERRNO; | |
165 | errno = 11; | |
166 | } | |
167 | assert_se(errno == 12); | |
168 | } | |
169 | ||
170 | static void test_in_set(void) { | |
171 | assert_se(IN_SET(1, 1)); | |
172 | assert_se(IN_SET(1, 1, 2, 3, 4)); | |
173 | assert_se(IN_SET(2, 1, 2, 3, 4)); | |
174 | assert_se(IN_SET(3, 1, 2, 3, 4)); | |
175 | assert_se(IN_SET(4, 1, 2, 3, 4)); | |
176 | assert_se(!IN_SET(0, 1)); | |
177 | assert_se(!IN_SET(0, 1, 2, 3, 4)); | |
178 | } | |
179 | ||
180 | static void test_log2i(void) { | |
181 | assert_se(log2i(1) == 0); | |
182 | assert_se(log2i(2) == 1); | |
183 | assert_se(log2i(3) == 1); | |
184 | assert_se(log2i(4) == 2); | |
185 | assert_se(log2i(32) == 5); | |
186 | assert_se(log2i(33) == 5); | |
187 | assert_se(log2i(63) == 5); | |
188 | assert_se(log2i(INT_MAX) == sizeof(int)*8-2); | |
189 | } | |
190 | ||
191 | static void test_raw_clone(void) { | |
192 | pid_t parent, pid, pid2; | |
193 | ||
194 | parent = getpid(); | |
195 | log_info("before clone: getpid()→"PID_FMT, parent); | |
196 | assert_se(raw_getpid() == parent); | |
197 | ||
198 | pid = raw_clone(0); | |
199 | assert_se(pid >= 0); | |
200 | ||
201 | pid2 = raw_getpid(); | |
202 | log_info("raw_clone: "PID_FMT" getpid()→"PID_FMT" raw_getpid()→"PID_FMT, | |
203 | pid, getpid(), pid2); | |
204 | if (pid == 0) { | |
205 | assert_se(pid2 != parent); | |
206 | _exit(EXIT_SUCCESS); | |
207 | } else { | |
208 | int status; | |
209 | ||
210 | assert_se(pid2 == parent); | |
211 | waitpid(pid, &status, __WCLONE); | |
212 | assert_se(WIFEXITED(status) && WEXITSTATUS(status) == EXIT_SUCCESS); | |
213 | } | |
214 | ||
215 | errno = 0; | |
216 | assert_se(raw_clone(CLONE_FS|CLONE_NEWNS) == -1); | |
217 | assert_se(errno == EINVAL); | |
218 | } | |
219 | ||
220 | static void test_physical_memory(void) { | |
221 | uint64_t p; | |
222 | char buf[FORMAT_BYTES_MAX]; | |
223 | ||
224 | p = physical_memory(); | |
225 | assert_se(p > 0); | |
226 | assert_se(p < UINT64_MAX); | |
227 | assert_se(p % page_size() == 0); | |
228 | ||
229 | log_info("Memory: %s (%" PRIu64 ")", format_bytes(buf, sizeof(buf), p), p); | |
230 | } | |
231 | ||
232 | static void test_physical_memory_scale(void) { | |
233 | uint64_t p; | |
234 | ||
235 | p = physical_memory(); | |
236 | ||
237 | assert_se(physical_memory_scale(0, 100) == 0); | |
238 | assert_se(physical_memory_scale(100, 100) == p); | |
239 | ||
240 | log_info("Memory original: %" PRIu64, physical_memory()); | |
241 | log_info("Memory scaled by 50%%: %" PRIu64, physical_memory_scale(50, 100)); | |
242 | log_info("Memory divided by 2: %" PRIu64, physical_memory() / 2); | |
243 | log_info("Page size: %zu", page_size()); | |
244 | ||
245 | /* There might be an uneven number of pages, hence permit these calculations to be half a page off... */ | |
246 | assert_se(page_size()/2 + physical_memory_scale(50, 100) - p/2 <= page_size()); | |
247 | assert_se(physical_memory_scale(200, 100) == p*2); | |
248 | ||
249 | assert_se(physical_memory_scale(0, 1) == 0); | |
250 | assert_se(physical_memory_scale(1, 1) == p); | |
251 | assert_se(physical_memory_scale(2, 1) == p*2); | |
252 | ||
253 | assert_se(physical_memory_scale(0, 2) == 0); | |
254 | ||
255 | assert_se(page_size()/2 + physical_memory_scale(1, 2) - p/2 <= page_size()); | |
256 | assert_se(physical_memory_scale(2, 2) == p); | |
257 | assert_se(physical_memory_scale(4, 2) == p*2); | |
258 | ||
259 | assert_se(physical_memory_scale(0, UINT32_MAX) == 0); | |
260 | assert_se(physical_memory_scale(UINT32_MAX, UINT32_MAX) == p); | |
261 | ||
262 | /* overflow */ | |
263 | assert_se(physical_memory_scale(UINT64_MAX/4, UINT64_MAX) == UINT64_MAX); | |
264 | } | |
265 | ||
266 | static void test_system_tasks_max(void) { | |
267 | uint64_t t; | |
268 | ||
269 | t = system_tasks_max(); | |
270 | assert_se(t > 0); | |
271 | assert_se(t < UINT64_MAX); | |
272 | ||
273 | log_info("Max tasks: %" PRIu64, t); | |
274 | } | |
275 | ||
276 | static void test_system_tasks_max_scale(void) { | |
277 | uint64_t t; | |
278 | ||
279 | t = system_tasks_max(); | |
280 | ||
281 | assert_se(system_tasks_max_scale(0, 100) == 0); | |
282 | assert_se(system_tasks_max_scale(100, 100) == t); | |
283 | ||
284 | assert_se(system_tasks_max_scale(0, 1) == 0); | |
285 | assert_se(system_tasks_max_scale(1, 1) == t); | |
286 | assert_se(system_tasks_max_scale(2, 1) == 2*t); | |
287 | ||
288 | assert_se(system_tasks_max_scale(0, 2) == 0); | |
289 | assert_se(system_tasks_max_scale(1, 2) == t/2); | |
290 | assert_se(system_tasks_max_scale(2, 2) == t); | |
291 | assert_se(system_tasks_max_scale(3, 2) == (3*t)/2); | |
292 | assert_se(system_tasks_max_scale(4, 2) == t*2); | |
293 | ||
294 | assert_se(system_tasks_max_scale(0, UINT32_MAX) == 0); | |
295 | assert_se(system_tasks_max_scale((UINT32_MAX-1)/2, UINT32_MAX-1) == t/2); | |
296 | assert_se(system_tasks_max_scale(UINT32_MAX, UINT32_MAX) == t); | |
297 | ||
298 | /* overflow */ | |
299 | ||
300 | assert_se(system_tasks_max_scale(UINT64_MAX/4, UINT64_MAX) == UINT64_MAX); | |
301 | } | |
302 | ||
303 | int main(int argc, char *argv[]) { | |
304 | log_parse_environment(); | |
305 | log_open(); | |
306 | ||
307 | test_align_power2(); | |
308 | test_max(); | |
309 | test_container_of(); | |
310 | test_div_round_up(); | |
311 | test_u64log2(); | |
312 | test_protect_errno(); | |
313 | test_in_set(); | |
314 | test_log2i(); | |
315 | test_raw_clone(); | |
316 | test_physical_memory(); | |
317 | test_physical_memory_scale(); | |
318 | test_system_tasks_max(); | |
319 | test_system_tasks_max_scale(); | |
320 | ||
321 | return 0; | |
322 | } |