Loading...
Note: File does not exist in v3.1.
1/* SPDX-License-Identifier: GPL-2.0 */
2#ifndef __TEST_PROGS_H
3#define __TEST_PROGS_H
4
5#include <stdio.h>
6#include <unistd.h>
7#include <errno.h>
8#include <string.h>
9#include <assert.h>
10#include <stdlib.h>
11#include <stdarg.h>
12#include <time.h>
13#include <signal.h>
14
15#include <linux/types.h>
16typedef __u16 __sum16;
17#include <arpa/inet.h>
18#include <linux/if_ether.h>
19#include <linux/if_packet.h>
20#include <linux/ip.h>
21#include <linux/ipv6.h>
22#include <linux/filter.h>
23#include <linux/perf_event.h>
24#include <linux/socket.h>
25#include <linux/unistd.h>
26
27#include <sys/ioctl.h>
28#include <sys/wait.h>
29#include <sys/types.h>
30#include <sys/time.h>
31#include <sys/param.h>
32#include <fcntl.h>
33#include <pthread.h>
34#include <linux/bpf.h>
35#include <linux/err.h>
36#include <bpf/bpf.h>
37#include <bpf/libbpf.h>
38
39#include "test_iptunnel_common.h"
40#include "bpf_util.h"
41#include <bpf/bpf_endian.h>
42#include "trace_helpers.h"
43#include "testing_helpers.h"
44
45enum verbosity {
46 VERBOSE_NONE,
47 VERBOSE_NORMAL,
48 VERBOSE_VERY,
49 VERBOSE_SUPER,
50};
51
52struct test_filter {
53 char *name;
54 char **subtests;
55 int subtest_cnt;
56};
57
58struct test_filter_set {
59 struct test_filter *tests;
60 int cnt;
61};
62
63struct test_selector {
64 struct test_filter_set whitelist;
65 struct test_filter_set blacklist;
66 bool *num_set;
67 int num_set_len;
68};
69
70struct subtest_state {
71 char *name;
72 size_t log_cnt;
73 char *log_buf;
74 int error_cnt;
75 bool skipped;
76 bool filtered;
77
78 FILE *stdout;
79};
80
81struct test_state {
82 bool tested;
83 bool force_log;
84
85 int error_cnt;
86 int skip_cnt;
87 int sub_succ_cnt;
88
89 struct subtest_state *subtest_states;
90 int subtest_num;
91
92 size_t log_cnt;
93 char *log_buf;
94
95 FILE *stdout;
96};
97
98struct test_env {
99 struct test_selector test_selector;
100 struct test_selector subtest_selector;
101 bool verifier_stats;
102 bool debug;
103 enum verbosity verbosity;
104
105 bool jit_enabled;
106 bool has_testmod;
107 bool get_test_cnt;
108 bool list_test_names;
109
110 struct prog_test_def *test; /* current running test */
111 struct test_state *test_state; /* current running test state */
112 struct subtest_state *subtest_state; /* current running subtest state */
113
114 FILE *stdout;
115 FILE *stderr;
116 int nr_cpus;
117
118 int succ_cnt; /* successful tests */
119 int sub_succ_cnt; /* successful sub-tests */
120 int fail_cnt; /* total failed tests + sub-tests */
121 int skip_cnt; /* skipped tests */
122
123 int saved_netns_fd;
124 int workers; /* number of worker process */
125 int worker_id; /* id number of current worker, main process is -1 */
126 pid_t *worker_pids; /* array of worker pids */
127 int *worker_socks; /* array of worker socks */
128 int *worker_current_test; /* array of current running test for each worker */
129};
130
131#define MAX_LOG_TRUNK_SIZE 8192
132#define MAX_SUBTEST_NAME 1024
133enum msg_type {
134 MSG_DO_TEST = 0,
135 MSG_TEST_DONE = 1,
136 MSG_TEST_LOG = 2,
137 MSG_SUBTEST_DONE = 3,
138 MSG_EXIT = 255,
139};
140struct msg {
141 enum msg_type type;
142 union {
143 struct {
144 int num;
145 } do_test;
146 struct {
147 int num;
148 int sub_succ_cnt;
149 int error_cnt;
150 int skip_cnt;
151 bool have_log;
152 int subtest_num;
153 } test_done;
154 struct {
155 char log_buf[MAX_LOG_TRUNK_SIZE + 1];
156 bool is_last;
157 } test_log;
158 struct {
159 int num;
160 char name[MAX_SUBTEST_NAME + 1];
161 int error_cnt;
162 bool skipped;
163 bool filtered;
164 bool have_log;
165 } subtest_done;
166 };
167};
168
169extern struct test_env env;
170
171void test__force_log(void);
172bool test__start_subtest(const char *name);
173void test__end_subtest(void);
174void test__skip(void);
175void test__fail(void);
176int test__join_cgroup(const char *path);
177
178#define PRINT_FAIL(format...) \
179 ({ \
180 test__fail(); \
181 fprintf(stdout, "%s:FAIL:%d ", __func__, __LINE__); \
182 fprintf(stdout, ##format); \
183 })
184
185#define _CHECK(condition, tag, duration, format...) ({ \
186 int __ret = !!(condition); \
187 int __save_errno = errno; \
188 if (__ret) { \
189 test__fail(); \
190 fprintf(stdout, "%s:FAIL:%s ", __func__, tag); \
191 fprintf(stdout, ##format); \
192 } else { \
193 fprintf(stdout, "%s:PASS:%s %d nsec\n", \
194 __func__, tag, duration); \
195 } \
196 errno = __save_errno; \
197 __ret; \
198})
199
200#define CHECK_FAIL(condition) ({ \
201 int __ret = !!(condition); \
202 int __save_errno = errno; \
203 if (__ret) { \
204 test__fail(); \
205 fprintf(stdout, "%s:FAIL:%d\n", __func__, __LINE__); \
206 } \
207 errno = __save_errno; \
208 __ret; \
209})
210
211#define CHECK(condition, tag, format...) \
212 _CHECK(condition, tag, duration, format)
213#define CHECK_ATTR(condition, tag, format...) \
214 _CHECK(condition, tag, tattr.duration, format)
215
216#define ASSERT_FAIL(fmt, args...) ({ \
217 static int duration = 0; \
218 CHECK(false, "", fmt"\n", ##args); \
219 false; \
220})
221
222#define ASSERT_TRUE(actual, name) ({ \
223 static int duration = 0; \
224 bool ___ok = (actual); \
225 CHECK(!___ok, (name), "unexpected %s: got FALSE\n", (name)); \
226 ___ok; \
227})
228
229#define ASSERT_FALSE(actual, name) ({ \
230 static int duration = 0; \
231 bool ___ok = !(actual); \
232 CHECK(!___ok, (name), "unexpected %s: got TRUE\n", (name)); \
233 ___ok; \
234})
235
236#define ASSERT_EQ(actual, expected, name) ({ \
237 static int duration = 0; \
238 typeof(actual) ___act = (actual); \
239 typeof(expected) ___exp = (expected); \
240 bool ___ok = ___act == ___exp; \
241 CHECK(!___ok, (name), \
242 "unexpected %s: actual %lld != expected %lld\n", \
243 (name), (long long)(___act), (long long)(___exp)); \
244 ___ok; \
245})
246
247#define ASSERT_NEQ(actual, expected, name) ({ \
248 static int duration = 0; \
249 typeof(actual) ___act = (actual); \
250 typeof(expected) ___exp = (expected); \
251 bool ___ok = ___act != ___exp; \
252 CHECK(!___ok, (name), \
253 "unexpected %s: actual %lld == expected %lld\n", \
254 (name), (long long)(___act), (long long)(___exp)); \
255 ___ok; \
256})
257
258#define ASSERT_LT(actual, expected, name) ({ \
259 static int duration = 0; \
260 typeof(actual) ___act = (actual); \
261 typeof(expected) ___exp = (expected); \
262 bool ___ok = ___act < ___exp; \
263 CHECK(!___ok, (name), \
264 "unexpected %s: actual %lld >= expected %lld\n", \
265 (name), (long long)(___act), (long long)(___exp)); \
266 ___ok; \
267})
268
269#define ASSERT_LE(actual, expected, name) ({ \
270 static int duration = 0; \
271 typeof(actual) ___act = (actual); \
272 typeof(expected) ___exp = (expected); \
273 bool ___ok = ___act <= ___exp; \
274 CHECK(!___ok, (name), \
275 "unexpected %s: actual %lld > expected %lld\n", \
276 (name), (long long)(___act), (long long)(___exp)); \
277 ___ok; \
278})
279
280#define ASSERT_GT(actual, expected, name) ({ \
281 static int duration = 0; \
282 typeof(actual) ___act = (actual); \
283 typeof(expected) ___exp = (expected); \
284 bool ___ok = ___act > ___exp; \
285 CHECK(!___ok, (name), \
286 "unexpected %s: actual %lld <= expected %lld\n", \
287 (name), (long long)(___act), (long long)(___exp)); \
288 ___ok; \
289})
290
291#define ASSERT_GE(actual, expected, name) ({ \
292 static int duration = 0; \
293 typeof(actual) ___act = (actual); \
294 typeof(expected) ___exp = (expected); \
295 bool ___ok = ___act >= ___exp; \
296 CHECK(!___ok, (name), \
297 "unexpected %s: actual %lld < expected %lld\n", \
298 (name), (long long)(___act), (long long)(___exp)); \
299 ___ok; \
300})
301
302#define ASSERT_STREQ(actual, expected, name) ({ \
303 static int duration = 0; \
304 const char *___act = actual; \
305 const char *___exp = expected; \
306 bool ___ok = strcmp(___act, ___exp) == 0; \
307 CHECK(!___ok, (name), \
308 "unexpected %s: actual '%s' != expected '%s'\n", \
309 (name), ___act, ___exp); \
310 ___ok; \
311})
312
313#define ASSERT_STRNEQ(actual, expected, len, name) ({ \
314 static int duration = 0; \
315 const char *___act = actual; \
316 const char *___exp = expected; \
317 int ___len = len; \
318 bool ___ok = strncmp(___act, ___exp, ___len) == 0; \
319 CHECK(!___ok, (name), \
320 "unexpected %s: actual '%.*s' != expected '%.*s'\n", \
321 (name), ___len, ___act, ___len, ___exp); \
322 ___ok; \
323})
324
325#define ASSERT_HAS_SUBSTR(str, substr, name) ({ \
326 static int duration = 0; \
327 const char *___str = str; \
328 const char *___substr = substr; \
329 bool ___ok = strstr(___str, ___substr) != NULL; \
330 CHECK(!___ok, (name), \
331 "unexpected %s: '%s' is not a substring of '%s'\n", \
332 (name), ___substr, ___str); \
333 ___ok; \
334})
335
336#define ASSERT_OK(res, name) ({ \
337 static int duration = 0; \
338 long long ___res = (res); \
339 bool ___ok = ___res == 0; \
340 CHECK(!___ok, (name), "unexpected error: %lld (errno %d)\n", \
341 ___res, errno); \
342 ___ok; \
343})
344
345#define ASSERT_ERR(res, name) ({ \
346 static int duration = 0; \
347 long long ___res = (res); \
348 bool ___ok = ___res < 0; \
349 CHECK(!___ok, (name), "unexpected success: %lld\n", ___res); \
350 ___ok; \
351})
352
353#define ASSERT_NULL(ptr, name) ({ \
354 static int duration = 0; \
355 const void *___res = (ptr); \
356 bool ___ok = !___res; \
357 CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res); \
358 ___ok; \
359})
360
361#define ASSERT_OK_PTR(ptr, name) ({ \
362 static int duration = 0; \
363 const void *___res = (ptr); \
364 int ___err = libbpf_get_error(___res); \
365 bool ___ok = ___err == 0; \
366 CHECK(!___ok, (name), "unexpected error: %d\n", ___err); \
367 ___ok; \
368})
369
370#define ASSERT_ERR_PTR(ptr, name) ({ \
371 static int duration = 0; \
372 const void *___res = (ptr); \
373 int ___err = libbpf_get_error(___res); \
374 bool ___ok = ___err != 0; \
375 CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res); \
376 ___ok; \
377})
378
379static inline __u64 ptr_to_u64(const void *ptr)
380{
381 return (__u64) (unsigned long) ptr;
382}
383
384static inline void *u64_to_ptr(__u64 ptr)
385{
386 return (void *) (unsigned long) ptr;
387}
388
389int bpf_find_map(const char *test, struct bpf_object *obj, const char *name);
390int compare_map_keys(int map1_fd, int map2_fd);
391int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len);
392int extract_build_id(char *build_id, size_t size);
393int kern_sync_rcu(void);
394int trigger_module_test_read(int read_sz);
395int trigger_module_test_write(int write_sz);
396int write_sysctl(const char *sysctl, const char *value);
397
398#ifdef __x86_64__
399#define SYS_NANOSLEEP_KPROBE_NAME "__x64_sys_nanosleep"
400#elif defined(__s390x__)
401#define SYS_NANOSLEEP_KPROBE_NAME "__s390x_sys_nanosleep"
402#elif defined(__aarch64__)
403#define SYS_NANOSLEEP_KPROBE_NAME "__arm64_sys_nanosleep"
404#else
405#define SYS_NANOSLEEP_KPROBE_NAME "sys_nanosleep"
406#endif
407
408#define BPF_TESTMOD_TEST_FILE "/sys/kernel/bpf_testmod"
409
410struct test_loader {
411 char *log_buf;
412 size_t log_buf_sz;
413
414 struct bpf_object *obj;
415};
416
417typedef const void *(*skel_elf_bytes_fn)(size_t *sz);
418
419extern void test_loader__run_subtests(struct test_loader *tester,
420 const char *skel_name,
421 skel_elf_bytes_fn elf_bytes_factory);
422
423extern void test_loader_fini(struct test_loader *tester);
424
425#define RUN_TESTS(skel) ({ \
426 struct test_loader tester = {}; \
427 \
428 test_loader__run_subtests(&tester, #skel, skel##__elf_bytes); \
429 test_loader_fini(&tester); \
430})
431
432#endif /* __TEST_PROGS_H */