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> 16 typedef __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 45 enum verbosity { 46 VERBOSE_NONE, 47 VERBOSE_NORMAL, 48 VERBOSE_VERY, 49 VERBOSE_SUPER, 50 }; 51 52 struct test_filter { 53 char *name; 54 char **subtests; 55 int subtest_cnt; 56 }; 57 58 struct test_filter_set { 59 struct test_filter *tests; 60 int cnt; 61 }; 62 63 struct 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 70 struct 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 bool should_tmon; 78 79 FILE *stdout_saved; 80 }; 81 82 struct test_state { 83 bool tested; 84 bool force_log; 85 86 int error_cnt; 87 int skip_cnt; 88 int sub_succ_cnt; 89 90 struct subtest_state *subtest_states; 91 int subtest_num; 92 93 size_t log_cnt; 94 char *log_buf; 95 96 FILE *stdout_saved; 97 }; 98 99 extern int env_verbosity; 100 101 struct test_env { 102 struct test_selector test_selector; 103 struct test_selector subtest_selector; 104 struct test_selector tmon_selector; 105 bool verifier_stats; 106 bool debug; 107 enum verbosity verbosity; 108 109 bool jit_enabled; 110 bool has_testmod; 111 bool get_test_cnt; 112 bool list_test_names; 113 114 struct prog_test_def *test; /* current running test */ 115 struct test_state *test_state; /* current running test state */ 116 struct subtest_state *subtest_state; /* current running subtest state */ 117 118 FILE *stdout_saved; 119 FILE *stderr_saved; 120 int nr_cpus; 121 FILE *json; 122 123 int succ_cnt; /* successful tests */ 124 int sub_succ_cnt; /* successful sub-tests */ 125 int fail_cnt; /* total failed tests + sub-tests */ 126 int skip_cnt; /* skipped tests */ 127 128 int saved_netns_fd; 129 int workers; /* number of worker process */ 130 int worker_id; /* id number of current worker, main process is -1 */ 131 pid_t *worker_pids; /* array of worker pids */ 132 int *worker_socks; /* array of worker socks */ 133 int *worker_current_test; /* array of current running test for each worker */ 134 }; 135 136 #define MAX_LOG_TRUNK_SIZE 8192 137 #define MAX_SUBTEST_NAME 1024 138 enum msg_type { 139 MSG_DO_TEST = 0, 140 MSG_TEST_DONE = 1, 141 MSG_TEST_LOG = 2, 142 MSG_SUBTEST_DONE = 3, 143 MSG_EXIT = 255, 144 }; 145 struct msg { 146 enum msg_type type; 147 union { 148 struct { 149 int num; 150 } do_test; 151 struct { 152 int num; 153 int sub_succ_cnt; 154 int error_cnt; 155 int skip_cnt; 156 bool have_log; 157 int subtest_num; 158 } test_done; 159 struct { 160 char log_buf[MAX_LOG_TRUNK_SIZE + 1]; 161 bool is_last; 162 } test_log; 163 struct { 164 int num; 165 char name[MAX_SUBTEST_NAME + 1]; 166 int error_cnt; 167 bool skipped; 168 bool filtered; 169 bool have_log; 170 } subtest_done; 171 }; 172 }; 173 174 extern struct test_env env; 175 176 void test__force_log(void); 177 bool test__start_subtest(const char *name); 178 void test__end_subtest(void); 179 void test__skip(void); 180 void test__fail(void); 181 int test__join_cgroup(const char *path); 182 183 #define PRINT_FAIL(format...) \ 184 ({ \ 185 test__fail(); \ 186 fprintf(stdout, "%s:FAIL:%d ", __func__, __LINE__); \ 187 fprintf(stdout, ##format); \ 188 }) 189 190 #define _CHECK(condition, tag, duration, format...) ({ \ 191 int __ret = !!(condition); \ 192 int __save_errno = errno; \ 193 if (__ret) { \ 194 test__fail(); \ 195 fprintf(stdout, "%s:FAIL:%s ", __func__, tag); \ 196 fprintf(stdout, ##format); \ 197 } else { \ 198 fprintf(stdout, "%s:PASS:%s %d nsec\n", \ 199 __func__, tag, duration); \ 200 } \ 201 errno = __save_errno; \ 202 __ret; \ 203 }) 204 205 #define CHECK_FAIL(condition) ({ \ 206 int __ret = !!(condition); \ 207 int __save_errno = errno; \ 208 if (__ret) { \ 209 test__fail(); \ 210 fprintf(stdout, "%s:FAIL:%d\n", __func__, __LINE__); \ 211 } \ 212 errno = __save_errno; \ 213 __ret; \ 214 }) 215 216 #define CHECK(condition, tag, format...) \ 217 _CHECK(condition, tag, duration, format) 218 #define CHECK_ATTR(condition, tag, format...) \ 219 _CHECK(condition, tag, tattr.duration, format) 220 221 #define ASSERT_FAIL(fmt, args...) ({ \ 222 static int duration = 0; \ 223 CHECK(false, "", fmt"\n", ##args); \ 224 false; \ 225 }) 226 227 #define ASSERT_TRUE(actual, name) ({ \ 228 static int duration = 0; \ 229 bool ___ok = (actual); \ 230 CHECK(!___ok, (name), "unexpected %s: got FALSE\n", (name)); \ 231 ___ok; \ 232 }) 233 234 #define ASSERT_FALSE(actual, name) ({ \ 235 static int duration = 0; \ 236 bool ___ok = !(actual); \ 237 CHECK(!___ok, (name), "unexpected %s: got TRUE\n", (name)); \ 238 ___ok; \ 239 }) 240 241 #define ASSERT_EQ(actual, expected, name) ({ \ 242 static int duration = 0; \ 243 typeof(actual) ___act = (actual); \ 244 typeof(expected) ___exp = (expected); \ 245 bool ___ok = ___act == ___exp; \ 246 CHECK(!___ok, (name), \ 247 "unexpected %s: actual %lld != expected %lld\n", \ 248 (name), (long long)(___act), (long long)(___exp)); \ 249 ___ok; \ 250 }) 251 252 #define ASSERT_NEQ(actual, expected, name) ({ \ 253 static int duration = 0; \ 254 typeof(actual) ___act = (actual); \ 255 typeof(expected) ___exp = (expected); \ 256 bool ___ok = ___act != ___exp; \ 257 CHECK(!___ok, (name), \ 258 "unexpected %s: actual %lld == expected %lld\n", \ 259 (name), (long long)(___act), (long long)(___exp)); \ 260 ___ok; \ 261 }) 262 263 #define ASSERT_LT(actual, expected, name) ({ \ 264 static int duration = 0; \ 265 typeof(actual) ___act = (actual); \ 266 typeof(expected) ___exp = (expected); \ 267 bool ___ok = ___act < ___exp; \ 268 CHECK(!___ok, (name), \ 269 "unexpected %s: actual %lld >= expected %lld\n", \ 270 (name), (long long)(___act), (long long)(___exp)); \ 271 ___ok; \ 272 }) 273 274 #define ASSERT_LE(actual, expected, name) ({ \ 275 static int duration = 0; \ 276 typeof(actual) ___act = (actual); \ 277 typeof(expected) ___exp = (expected); \ 278 bool ___ok = ___act <= ___exp; \ 279 CHECK(!___ok, (name), \ 280 "unexpected %s: actual %lld > expected %lld\n", \ 281 (name), (long long)(___act), (long long)(___exp)); \ 282 ___ok; \ 283 }) 284 285 #define ASSERT_GT(actual, expected, name) ({ \ 286 static int duration = 0; \ 287 typeof(actual) ___act = (actual); \ 288 typeof(expected) ___exp = (expected); \ 289 bool ___ok = ___act > ___exp; \ 290 CHECK(!___ok, (name), \ 291 "unexpected %s: actual %lld <= expected %lld\n", \ 292 (name), (long long)(___act), (long long)(___exp)); \ 293 ___ok; \ 294 }) 295 296 #define ASSERT_GE(actual, expected, name) ({ \ 297 static int duration = 0; \ 298 typeof(actual) ___act = (actual); \ 299 typeof(expected) ___exp = (expected); \ 300 bool ___ok = ___act >= ___exp; \ 301 CHECK(!___ok, (name), \ 302 "unexpected %s: actual %lld < expected %lld\n", \ 303 (name), (long long)(___act), (long long)(___exp)); \ 304 ___ok; \ 305 }) 306 307 #define ASSERT_STREQ(actual, expected, name) ({ \ 308 static int duration = 0; \ 309 const char *___act = actual; \ 310 const char *___exp = expected; \ 311 bool ___ok = strcmp(___act, ___exp) == 0; \ 312 CHECK(!___ok, (name), \ 313 "unexpected %s: actual '%s' != expected '%s'\n", \ 314 (name), ___act, ___exp); \ 315 ___ok; \ 316 }) 317 318 #define ASSERT_STRNEQ(actual, expected, len, name) ({ \ 319 static int duration = 0; \ 320 const char *___act = actual; \ 321 const char *___exp = expected; \ 322 int ___len = len; \ 323 bool ___ok = strncmp(___act, ___exp, ___len) == 0; \ 324 CHECK(!___ok, (name), \ 325 "unexpected %s: actual '%.*s' != expected '%.*s'\n", \ 326 (name), ___len, ___act, ___len, ___exp); \ 327 ___ok; \ 328 }) 329 330 #define ASSERT_HAS_SUBSTR(str, substr, name) ({ \ 331 static int duration = 0; \ 332 const char *___str = str; \ 333 const char *___substr = substr; \ 334 bool ___ok = strstr(___str, ___substr) != NULL; \ 335 CHECK(!___ok, (name), \ 336 "unexpected %s: '%s' is not a substring of '%s'\n", \ 337 (name), ___substr, ___str); \ 338 ___ok; \ 339 }) 340 341 #define ASSERT_OK(res, name) ({ \ 342 static int duration = 0; \ 343 long long ___res = (res); \ 344 bool ___ok = ___res == 0; \ 345 CHECK(!___ok, (name), "unexpected error: %lld (errno %d)\n", \ 346 ___res, errno); \ 347 ___ok; \ 348 }) 349 350 #define ASSERT_ERR(res, name) ({ \ 351 static int duration = 0; \ 352 long long ___res = (res); \ 353 bool ___ok = ___res < 0; \ 354 CHECK(!___ok, (name), "unexpected success: %lld\n", ___res); \ 355 ___ok; \ 356 }) 357 358 #define ASSERT_NULL(ptr, name) ({ \ 359 static int duration = 0; \ 360 const void *___res = (ptr); \ 361 bool ___ok = !___res; \ 362 CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res); \ 363 ___ok; \ 364 }) 365 366 #define ASSERT_OK_PTR(ptr, name) ({ \ 367 static int duration = 0; \ 368 const void *___res = (ptr); \ 369 int ___err = libbpf_get_error(___res); \ 370 bool ___ok = ___err == 0; \ 371 CHECK(!___ok, (name), "unexpected error: %d\n", ___err); \ 372 ___ok; \ 373 }) 374 375 #define ASSERT_ERR_PTR(ptr, name) ({ \ 376 static int duration = 0; \ 377 const void *___res = (ptr); \ 378 int ___err = libbpf_get_error(___res); \ 379 bool ___ok = ___err != 0; \ 380 CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res); \ 381 ___ok; \ 382 }) 383 384 #define ASSERT_OK_FD(fd, name) ({ \ 385 static int duration = 0; \ 386 int ___fd = (fd); \ 387 bool ___ok = ___fd >= 0; \ 388 CHECK(!___ok, (name), "unexpected fd: %d (errno %d)\n", \ 389 ___fd, errno); \ 390 ___ok; \ 391 }) 392 393 #define SYS(goto_label, fmt, ...) \ 394 ({ \ 395 char cmd[1024]; \ 396 snprintf(cmd, sizeof(cmd), fmt, ##__VA_ARGS__); \ 397 if (!ASSERT_OK(system(cmd), cmd)) \ 398 goto goto_label; \ 399 }) 400 401 #define ALL_TO_DEV_NULL " >/dev/null 2>&1" 402 403 #define SYS_NOFAIL(fmt, ...) \ 404 ({ \ 405 char cmd[1024]; \ 406 int n; \ 407 n = snprintf(cmd, sizeof(cmd), fmt, ##__VA_ARGS__); \ 408 if (n < sizeof(cmd) && sizeof(cmd) - n >= sizeof(ALL_TO_DEV_NULL)) \ 409 strcat(cmd, ALL_TO_DEV_NULL); \ 410 system(cmd); \ 411 }) 412 413 int start_libbpf_log_capture(void); 414 char *stop_libbpf_log_capture(void); 415 ptr_to_u64(const void * ptr)416 static inline __u64 ptr_to_u64(const void *ptr) 417 { 418 return (__u64) (unsigned long) ptr; 419 } 420 u64_to_ptr(__u64 ptr)421 static inline void *u64_to_ptr(__u64 ptr) 422 { 423 return (void *) (unsigned long) ptr; 424 } 425 426 int bpf_find_map(const char *test, struct bpf_object *obj, const char *name); 427 int compare_map_keys(int map1_fd, int map2_fd); 428 int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len); 429 int trigger_module_test_read(int read_sz); 430 int trigger_module_test_write(int write_sz); 431 int write_sysctl(const char *sysctl, const char *value); 432 int get_bpf_max_tramp_links_from(struct btf *btf); 433 int get_bpf_max_tramp_links(void); 434 435 struct netns_obj; 436 struct netns_obj *netns_new(const char *name, bool open); 437 void netns_free(struct netns_obj *netns); 438 439 #ifdef __x86_64__ 440 #define SYS_NANOSLEEP_KPROBE_NAME "__x64_sys_nanosleep" 441 #elif defined(__s390x__) 442 #define SYS_NANOSLEEP_KPROBE_NAME "__s390x_sys_nanosleep" 443 #elif defined(__aarch64__) 444 #define SYS_NANOSLEEP_KPROBE_NAME "__arm64_sys_nanosleep" 445 #elif defined(__riscv) 446 #define SYS_NANOSLEEP_KPROBE_NAME "__riscv_sys_nanosleep" 447 #else 448 #define SYS_NANOSLEEP_KPROBE_NAME "sys_nanosleep" 449 #endif 450 451 #define BPF_TESTMOD_TEST_FILE "/sys/kernel/bpf_testmod" 452 453 typedef int (*pre_execution_cb)(struct bpf_object *obj); 454 455 struct test_loader { 456 char *log_buf; 457 size_t log_buf_sz; 458 pre_execution_cb pre_execution_cb; 459 460 struct bpf_object *obj; 461 }; 462 test_loader__set_pre_execution_cb(struct test_loader * tester,pre_execution_cb cb)463 static inline void test_loader__set_pre_execution_cb(struct test_loader *tester, 464 pre_execution_cb cb) 465 { 466 tester->pre_execution_cb = cb; 467 } 468 469 typedef const void *(*skel_elf_bytes_fn)(size_t *sz); 470 471 extern void test_loader__run_subtests(struct test_loader *tester, 472 const char *skel_name, 473 skel_elf_bytes_fn elf_bytes_factory); 474 475 extern void test_loader_fini(struct test_loader *tester); 476 477 #define RUN_TESTS(skel) ({ \ 478 struct test_loader tester = {}; \ 479 \ 480 test_loader__run_subtests(&tester, #skel, skel##__elf_bytes); \ 481 test_loader_fini(&tester); \ 482 }) 483 484 #endif /* __TEST_PROGS_H */ 485