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 
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_obj_get_info_by_fd(prog_fd, &info, &len);
37 	if (!ASSERT_OK(err, "bpf_obj_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 
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_obj_get_info_by_fd(frags_prog_fd, &info, &len);
89 	if (!ASSERT_OK(err, "bpf_obj_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 
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