selftests/bpf: Improve recursion selftest
authorAlexei Starovoitov <ast@kernel.org>
Wed, 10 Feb 2021 03:36:32 +0000 (19:36 -0800)
committerDaniel Borkmann <daniel@iogearbox.net>
Thu, 11 Feb 2021 15:19:24 +0000 (16:19 +0100)
Since recursion_misses counter is available in bpf_prog_info
improve the selftest to make sure it's counting correctly.

Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
Link: https://lore.kernel.org/bpf/20210210033634.62081-8-alexei.starovoitov@gmail.com
tools/testing/selftests/bpf/prog_tests/recursion.c

index 863757461e3f0ad2177be9dafbebfb8e750bfee0..0e378d63fe18398cd4ac486a2825af0baa58b878 100644 (file)
@@ -5,6 +5,8 @@
 
 void test_recursion(void)
 {
+       struct bpf_prog_info prog_info = {};
+       __u32 prog_info_len = sizeof(prog_info);
        struct recursion *skel;
        int key = 0;
        int err;
@@ -28,6 +30,12 @@ void test_recursion(void)
        ASSERT_EQ(skel->bss->pass2, 1, "pass2 == 1");
        bpf_map_lookup_elem(bpf_map__fd(skel->maps.hash2), &key, 0);
        ASSERT_EQ(skel->bss->pass2, 2, "pass2 == 2");
+
+       err = bpf_obj_get_info_by_fd(bpf_program__fd(skel->progs.on_lookup),
+                                    &prog_info, &prog_info_len);
+       if (!ASSERT_OK(err, "get_prog_info"))
+               goto out;
+       ASSERT_EQ(prog_info.recursion_misses, 2, "recursion_misses");
 out:
        recursion__destroy(skel);
 }