bpf/prog_tests/timer_mim.c

0001 // SPDX-License-Identifier: GPL-2.0
0002 /* Copyright (c) 2021 Facebook */
0003 #include <test_progs.h>
0004 #include "timer_mim.skel.h"
0005 #include "timer_mim_reject.skel.h"
0006
0007 static int timer_mim(struct timer_mim *timer_skel)
0008 {
0009     __u64 cnt1, cnt2;
0010     int err, prog_fd, key1 = 1;
0011     LIBBPF_OPTS(bpf_test_run_opts, topts);
0012
0013     err = timer_mim__attach(timer_skel);
0014     if (!ASSERT_OK(err, "timer_attach"))
0015         return err;
0016
0017     prog_fd = bpf_program__fd(timer_skel->progs.test1);
0018     err = bpf_prog_test_run_opts(prog_fd, &topts);
0019     ASSERT_OK(err, "test_run");
0020     ASSERT_EQ(topts.retval, 0, "test_run");
0021     timer_mim__detach(timer_skel);
0022
0023     /* check that timer_cb[12] are incrementing 'cnt' */
0024     cnt1 = READ_ONCE(timer_skel->bss->cnt);
0025     for (int i = 0; i < 100; i++) {
0026         cnt2 = READ_ONCE(timer_skel->bss->cnt);
0027         if (cnt2 != cnt1)
0028             break;
0029         usleep(200); /* 100 times more than interval */
0030     }
0031     ASSERT_GT(cnt2, cnt1, "cnt");
0032
0033     ASSERT_EQ(timer_skel->bss->err, 0, "err");
0034     /* check that code paths completed */
0035     ASSERT_EQ(timer_skel->bss->ok, 1 | 2, "ok");
0036
0037     close(bpf_map__fd(timer_skel->maps.inner_htab));
0038     err = bpf_map__delete_elem(timer_skel->maps.outer_arr, &key1, sizeof(key1), 0);
0039     ASSERT_EQ(err, 0, "delete inner map");
0040
0041     /* check that timer_cb[12] are no longer running */
0042     cnt1 = READ_ONCE(timer_skel->bss->cnt);
0043     for (int i = 0; i < 100; i++) {
0044         usleep(200); /* 100 times more than interval */
0045         cnt2 = READ_ONCE(timer_skel->bss->cnt);
0046         if (cnt2 == cnt1)
0047             break;
0048     }
0049     ASSERT_EQ(cnt2, cnt1, "cnt");
0050
0051     return 0;
0052 }
0053
0054 void serial_test_timer_mim(void)
0055 {
0056     struct timer_mim_reject *timer_reject_skel = NULL;
0057     libbpf_print_fn_t old_print_fn = NULL;
0058     struct timer_mim *timer_skel = NULL;
0059     int err;
0060
0061     old_print_fn = libbpf_set_print(NULL);
0062     timer_reject_skel = timer_mim_reject__open_and_load();
0063     libbpf_set_print(old_print_fn);
0064     if (!ASSERT_ERR_PTR(timer_reject_skel, "timer_reject_skel_load"))
0065         goto cleanup;
0066
0067     timer_skel = timer_mim__open_and_load();
0068     if (!ASSERT_OK_PTR(timer_skel, "timer_skel_load"))
0069         goto cleanup;
0070
0071     err = timer_mim(timer_skel);
0072     ASSERT_OK(err, "timer_mim");
0073 cleanup:
0074     timer_mim__destroy(timer_skel);
0075     timer_mim_reject__destroy(timer_reject_skel);
0076 }