1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2021 Facebook */ 3 #include <test_progs.h> 4 #include "timer_mim.skel.h" 5 #include "timer_mim_reject.skel.h" 6 7 static int timer_mim(struct timer_mim *timer_skel) 8 { 9 __u32 duration = 0, retval; 10 __u64 cnt1, cnt2; 11 int err, prog_fd, key1 = 1; 12 13 err = timer_mim__attach(timer_skel); 14 if (!ASSERT_OK(err, "timer_attach")) 15 return err; 16 17 prog_fd = bpf_program__fd(timer_skel->progs.test1); 18 err = bpf_prog_test_run(prog_fd, 1, NULL, 0, 19 NULL, NULL, &retval, &duration); 20 ASSERT_OK(err, "test_run"); 21 ASSERT_EQ(retval, 0, "test_run"); 22 timer_mim__detach(timer_skel); 23 24 /* check that timer_cb[12] are incrementing 'cnt' */ 25 cnt1 = READ_ONCE(timer_skel->bss->cnt); 26 usleep(200); /* 100 times more than interval */ 27 cnt2 = READ_ONCE(timer_skel->bss->cnt); 28 ASSERT_GT(cnt2, cnt1, "cnt"); 29 30 ASSERT_EQ(timer_skel->bss->err, 0, "err"); 31 /* check that code paths completed */ 32 ASSERT_EQ(timer_skel->bss->ok, 1 | 2, "ok"); 33 34 close(bpf_map__fd(timer_skel->maps.inner_htab)); 35 err = bpf_map_delete_elem(bpf_map__fd(timer_skel->maps.outer_arr), &key1); 36 ASSERT_EQ(err, 0, "delete inner map"); 37 38 /* check that timer_cb[12] are no longer running */ 39 cnt1 = READ_ONCE(timer_skel->bss->cnt); 40 usleep(200); 41 cnt2 = READ_ONCE(timer_skel->bss->cnt); 42 ASSERT_EQ(cnt2, cnt1, "cnt"); 43 44 return 0; 45 } 46 47 void test_timer_mim(void) 48 { 49 struct timer_mim_reject *timer_reject_skel = NULL; 50 libbpf_print_fn_t old_print_fn = NULL; 51 struct timer_mim *timer_skel = NULL; 52 int err; 53 54 old_print_fn = libbpf_set_print(NULL); 55 timer_reject_skel = timer_mim_reject__open_and_load(); 56 libbpf_set_print(old_print_fn); 57 if (!ASSERT_ERR_PTR(timer_reject_skel, "timer_reject_skel_load")) 58 goto cleanup; 59 60 timer_skel = timer_mim__open_and_load(); 61 if (!ASSERT_OK_PTR(timer_skel, "timer_skel_load")) 62 goto cleanup; 63 64 err = timer_mim(timer_skel); 65 ASSERT_OK(err, "timer_mim"); 66 cleanup: 67 timer_mim__destroy(timer_skel); 68 timer_mim_reject__destroy(timer_reject_skel); 69 } 70