1*e1cc1f39SNamhyung Kim // SPDX-License-Identifier: GPL-2.0
2*e1cc1f39SNamhyung Kim #include <test_progs.h>
3*e1cc1f39SNamhyung Kim #include "stacktrace_map_skip.skel.h"
4*e1cc1f39SNamhyung Kim
5*e1cc1f39SNamhyung Kim #define TEST_STACK_DEPTH 2
6*e1cc1f39SNamhyung Kim
test_stacktrace_map_skip(void)7*e1cc1f39SNamhyung Kim void test_stacktrace_map_skip(void)
8*e1cc1f39SNamhyung Kim {
9*e1cc1f39SNamhyung Kim struct stacktrace_map_skip *skel;
10*e1cc1f39SNamhyung Kim int stackid_hmap_fd, stackmap_fd, stack_amap_fd;
11*e1cc1f39SNamhyung Kim int err, stack_trace_len;
12*e1cc1f39SNamhyung Kim
13*e1cc1f39SNamhyung Kim skel = stacktrace_map_skip__open_and_load();
14*e1cc1f39SNamhyung Kim if (!ASSERT_OK_PTR(skel, "skel_open_and_load"))
15*e1cc1f39SNamhyung Kim return;
16*e1cc1f39SNamhyung Kim
17*e1cc1f39SNamhyung Kim /* find map fds */
18*e1cc1f39SNamhyung Kim stackid_hmap_fd = bpf_map__fd(skel->maps.stackid_hmap);
19*e1cc1f39SNamhyung Kim if (!ASSERT_GE(stackid_hmap_fd, 0, "stackid_hmap fd"))
20*e1cc1f39SNamhyung Kim goto out;
21*e1cc1f39SNamhyung Kim
22*e1cc1f39SNamhyung Kim stackmap_fd = bpf_map__fd(skel->maps.stackmap);
23*e1cc1f39SNamhyung Kim if (!ASSERT_GE(stackmap_fd, 0, "stackmap fd"))
24*e1cc1f39SNamhyung Kim goto out;
25*e1cc1f39SNamhyung Kim
26*e1cc1f39SNamhyung Kim stack_amap_fd = bpf_map__fd(skel->maps.stack_amap);
27*e1cc1f39SNamhyung Kim if (!ASSERT_GE(stack_amap_fd, 0, "stack_amap fd"))
28*e1cc1f39SNamhyung Kim goto out;
29*e1cc1f39SNamhyung Kim
30*e1cc1f39SNamhyung Kim skel->bss->pid = getpid();
31*e1cc1f39SNamhyung Kim
32*e1cc1f39SNamhyung Kim err = stacktrace_map_skip__attach(skel);
33*e1cc1f39SNamhyung Kim if (!ASSERT_OK(err, "skel_attach"))
34*e1cc1f39SNamhyung Kim goto out;
35*e1cc1f39SNamhyung Kim
36*e1cc1f39SNamhyung Kim /* give some time for bpf program run */
37*e1cc1f39SNamhyung Kim sleep(1);
38*e1cc1f39SNamhyung Kim
39*e1cc1f39SNamhyung Kim /* disable stack trace collection */
40*e1cc1f39SNamhyung Kim skel->bss->control = 1;
41*e1cc1f39SNamhyung Kim
42*e1cc1f39SNamhyung Kim /* for every element in stackid_hmap, we can find a corresponding one
43*e1cc1f39SNamhyung Kim * in stackmap, and vise versa.
44*e1cc1f39SNamhyung Kim */
45*e1cc1f39SNamhyung Kim err = compare_map_keys(stackid_hmap_fd, stackmap_fd);
46*e1cc1f39SNamhyung Kim if (!ASSERT_OK(err, "compare_map_keys stackid_hmap vs. stackmap"))
47*e1cc1f39SNamhyung Kim goto out;
48*e1cc1f39SNamhyung Kim
49*e1cc1f39SNamhyung Kim err = compare_map_keys(stackmap_fd, stackid_hmap_fd);
50*e1cc1f39SNamhyung Kim if (!ASSERT_OK(err, "compare_map_keys stackmap vs. stackid_hmap"))
51*e1cc1f39SNamhyung Kim goto out;
52*e1cc1f39SNamhyung Kim
53*e1cc1f39SNamhyung Kim stack_trace_len = TEST_STACK_DEPTH * sizeof(__u64);
54*e1cc1f39SNamhyung Kim err = compare_stack_ips(stackmap_fd, stack_amap_fd, stack_trace_len);
55*e1cc1f39SNamhyung Kim if (!ASSERT_OK(err, "compare_stack_ips stackmap vs. stack_amap"))
56*e1cc1f39SNamhyung Kim goto out;
57*e1cc1f39SNamhyung Kim
58*e1cc1f39SNamhyung Kim if (!ASSERT_EQ(skel->bss->failed, 0, "skip_failed"))
59*e1cc1f39SNamhyung Kim goto out;
60*e1cc1f39SNamhyung Kim
61*e1cc1f39SNamhyung Kim out:
62*e1cc1f39SNamhyung Kim stacktrace_map_skip__destroy(skel);
63*e1cc1f39SNamhyung Kim }
64