1 // SPDX-License-Identifier: GPL-2.0
2 #include <uapi/linux/bpf.h>
3 #include <linux/if_link.h>
4 #include <test_progs.h>
5
6 #include "test_xdp_with_cpumap_frags_helpers.skel.h"
7 #include "test_xdp_with_cpumap_helpers.skel.h"
8
9 #define IFINDEX_LO 1
10
test_xdp_with_cpumap_helpers(void)11 static void test_xdp_with_cpumap_helpers(void)
12 {
13 struct test_xdp_with_cpumap_helpers *skel;
14 struct bpf_prog_info info = {};
15 __u32 len = sizeof(info);
16 struct bpf_cpumap_val val = {
17 .qsize = 192,
18 };
19 int err, prog_fd, map_fd;
20 __u32 idx = 0;
21
22 skel = test_xdp_with_cpumap_helpers__open_and_load();
23 if (!ASSERT_OK_PTR(skel, "test_xdp_with_cpumap_helpers__open_and_load"))
24 return;
25
26 prog_fd = bpf_program__fd(skel->progs.xdp_redir_prog);
27 err = bpf_xdp_attach(IFINDEX_LO, prog_fd, XDP_FLAGS_SKB_MODE, NULL);
28 if (!ASSERT_OK(err, "Generic attach of program with 8-byte CPUMAP"))
29 goto out_close;
30
31 err = bpf_xdp_detach(IFINDEX_LO, XDP_FLAGS_SKB_MODE, NULL);
32 ASSERT_OK(err, "XDP program detach");
33
34 prog_fd = bpf_program__fd(skel->progs.xdp_dummy_cm);
35 map_fd = bpf_map__fd(skel->maps.cpu_map);
36 err = bpf_prog_get_info_by_fd(prog_fd, &info, &len);
37 if (!ASSERT_OK(err, "bpf_prog_get_info_by_fd"))
38 goto out_close;
39
40 val.bpf_prog.fd = prog_fd;
41 err = bpf_map_update_elem(map_fd, &idx, &val, 0);
42 ASSERT_OK(err, "Add program to cpumap entry");
43
44 err = bpf_map_lookup_elem(map_fd, &idx, &val);
45 ASSERT_OK(err, "Read cpumap entry");
46 ASSERT_EQ(info.id, val.bpf_prog.id, "Match program id to cpumap entry prog_id");
47
48 /* can not attach BPF_XDP_CPUMAP program to a device */
49 err = bpf_xdp_attach(IFINDEX_LO, prog_fd, XDP_FLAGS_SKB_MODE, NULL);
50 if (!ASSERT_NEQ(err, 0, "Attach of BPF_XDP_CPUMAP program"))
51 bpf_xdp_detach(IFINDEX_LO, XDP_FLAGS_SKB_MODE, NULL);
52
53 val.qsize = 192;
54 val.bpf_prog.fd = bpf_program__fd(skel->progs.xdp_dummy_prog);
55 err = bpf_map_update_elem(map_fd, &idx, &val, 0);
56 ASSERT_NEQ(err, 0, "Add non-BPF_XDP_CPUMAP program to cpumap entry");
57
58 /* Try to attach BPF_XDP program with frags to cpumap when we have
59 * already loaded a BPF_XDP program on the map
60 */
61 idx = 1;
62 val.qsize = 192;
63 val.bpf_prog.fd = bpf_program__fd(skel->progs.xdp_dummy_cm_frags);
64 err = bpf_map_update_elem(map_fd, &idx, &val, 0);
65 ASSERT_NEQ(err, 0, "Add BPF_XDP program with frags to cpumap entry");
66
67 out_close:
68 test_xdp_with_cpumap_helpers__destroy(skel);
69 }
70
test_xdp_with_cpumap_frags_helpers(void)71 static void test_xdp_with_cpumap_frags_helpers(void)
72 {
73 struct test_xdp_with_cpumap_frags_helpers *skel;
74 struct bpf_prog_info info = {};
75 __u32 len = sizeof(info);
76 struct bpf_cpumap_val val = {
77 .qsize = 192,
78 };
79 int err, frags_prog_fd, map_fd;
80 __u32 idx = 0;
81
82 skel = test_xdp_with_cpumap_frags_helpers__open_and_load();
83 if (!ASSERT_OK_PTR(skel, "test_xdp_with_cpumap_helpers__open_and_load"))
84 return;
85
86 frags_prog_fd = bpf_program__fd(skel->progs.xdp_dummy_cm_frags);
87 map_fd = bpf_map__fd(skel->maps.cpu_map);
88 err = bpf_prog_get_info_by_fd(frags_prog_fd, &info, &len);
89 if (!ASSERT_OK(err, "bpf_prog_get_info_by_fd"))
90 goto out_close;
91
92 val.bpf_prog.fd = frags_prog_fd;
93 err = bpf_map_update_elem(map_fd, &idx, &val, 0);
94 ASSERT_OK(err, "Add program to cpumap entry");
95
96 err = bpf_map_lookup_elem(map_fd, &idx, &val);
97 ASSERT_OK(err, "Read cpumap entry");
98 ASSERT_EQ(info.id, val.bpf_prog.id,
99 "Match program id to cpumap entry prog_id");
100
101 /* Try to attach BPF_XDP program to cpumap when we have
102 * already loaded a BPF_XDP program with frags on the map
103 */
104 idx = 1;
105 val.qsize = 192;
106 val.bpf_prog.fd = bpf_program__fd(skel->progs.xdp_dummy_cm);
107 err = bpf_map_update_elem(map_fd, &idx, &val, 0);
108 ASSERT_NEQ(err, 0, "Add BPF_XDP program to cpumap entry");
109
110 out_close:
111 test_xdp_with_cpumap_frags_helpers__destroy(skel);
112 }
113
serial_test_xdp_cpumap_attach(void)114 void serial_test_xdp_cpumap_attach(void)
115 {
116 if (test__start_subtest("CPUMAP with programs in entries"))
117 test_xdp_with_cpumap_helpers();
118
119 if (test__start_subtest("CPUMAP with frags programs in entries"))
120 test_xdp_with_cpumap_frags_helpers();
121 }
122