247 lines · c
1// SPDX-License-Identifier: GPL-2.02#include <arpa/inet.h>3#include <uapi/linux/bpf.h>4#include <linux/if_link.h>5#include <network_helpers.h>6#include <net/if.h>7#include <test_progs.h>8 9#include "test_xdp_devmap_helpers.skel.h"10#include "test_xdp_with_devmap_frags_helpers.skel.h"11#include "test_xdp_with_devmap_helpers.skel.h"12 13#define IFINDEX_LO 114#define TEST_NS "devmap_attach_ns"15 16static void test_xdp_with_devmap_helpers(void)17{18 struct test_xdp_with_devmap_helpers *skel = NULL;19 struct bpf_prog_info info = {};20 struct bpf_devmap_val val = {21 .ifindex = IFINDEX_LO,22 };23 __u32 len = sizeof(info);24 int err, dm_fd, dm_fd_redir, map_fd;25 struct nstoken *nstoken = NULL;26 char data[10] = {};27 __u32 idx = 0;28 29 SYS(out_close, "ip netns add %s", TEST_NS);30 nstoken = open_netns(TEST_NS);31 if (!ASSERT_OK_PTR(nstoken, "open_netns"))32 goto out_close;33 SYS(out_close, "ip link set dev lo up");34 35 skel = test_xdp_with_devmap_helpers__open_and_load();36 if (!ASSERT_OK_PTR(skel, "test_xdp_with_devmap_helpers__open_and_load"))37 goto out_close;38 39 dm_fd_redir = bpf_program__fd(skel->progs.xdp_redir_prog);40 err = bpf_xdp_attach(IFINDEX_LO, dm_fd_redir, XDP_FLAGS_SKB_MODE, NULL);41 if (!ASSERT_OK(err, "Generic attach of program with 8-byte devmap"))42 goto out_close;43 44 dm_fd = bpf_program__fd(skel->progs.xdp_dummy_dm);45 map_fd = bpf_map__fd(skel->maps.dm_ports);46 err = bpf_prog_get_info_by_fd(dm_fd, &info, &len);47 if (!ASSERT_OK(err, "bpf_prog_get_info_by_fd"))48 goto out_close;49 50 val.bpf_prog.fd = dm_fd;51 err = bpf_map_update_elem(map_fd, &idx, &val, 0);52 ASSERT_OK(err, "Add program to devmap entry");53 54 err = bpf_map_lookup_elem(map_fd, &idx, &val);55 ASSERT_OK(err, "Read devmap entry");56 ASSERT_EQ(info.id, val.bpf_prog.id, "Match program id to devmap entry prog_id");57 58 /* send a packet to trigger any potential bugs in there */59 DECLARE_LIBBPF_OPTS(bpf_test_run_opts, opts,60 .data_in = &data,61 .data_size_in = 10,62 .flags = BPF_F_TEST_XDP_LIVE_FRAMES,63 .repeat = 1,64 );65 err = bpf_prog_test_run_opts(dm_fd_redir, &opts);66 ASSERT_OK(err, "XDP test run");67 68 /* wait for the packets to be flushed */69 kern_sync_rcu();70 71 err = bpf_xdp_detach(IFINDEX_LO, XDP_FLAGS_SKB_MODE, NULL);72 ASSERT_OK(err, "XDP program detach");73 74 /* can not attach BPF_XDP_DEVMAP program to a device */75 err = bpf_xdp_attach(IFINDEX_LO, dm_fd, XDP_FLAGS_SKB_MODE, NULL);76 if (!ASSERT_NEQ(err, 0, "Attach of BPF_XDP_DEVMAP program"))77 bpf_xdp_detach(IFINDEX_LO, XDP_FLAGS_SKB_MODE, NULL);78 79 val.ifindex = 1;80 val.bpf_prog.fd = bpf_program__fd(skel->progs.xdp_dummy_prog);81 err = bpf_map_update_elem(map_fd, &idx, &val, 0);82 ASSERT_NEQ(err, 0, "Add non-BPF_XDP_DEVMAP program to devmap entry");83 84 /* Try to attach BPF_XDP program with frags to devmap when we have85 * already loaded a BPF_XDP program on the map86 */87 idx = 1;88 val.ifindex = 1;89 val.bpf_prog.fd = bpf_program__fd(skel->progs.xdp_dummy_dm_frags);90 err = bpf_map_update_elem(map_fd, &idx, &val, 0);91 ASSERT_NEQ(err, 0, "Add BPF_XDP program with frags to devmap entry");92 93out_close:94 close_netns(nstoken);95 SYS_NOFAIL("ip netns del %s", TEST_NS);96 test_xdp_with_devmap_helpers__destroy(skel);97}98 99static void test_neg_xdp_devmap_helpers(void)100{101 struct test_xdp_devmap_helpers *skel;102 103 skel = test_xdp_devmap_helpers__open_and_load();104 if (!ASSERT_EQ(skel, NULL,105 "Load of XDP program accessing egress ifindex without attach type")) {106 test_xdp_devmap_helpers__destroy(skel);107 }108}109 110static void test_xdp_with_devmap_frags_helpers(void)111{112 struct test_xdp_with_devmap_frags_helpers *skel;113 struct bpf_prog_info info = {};114 struct bpf_devmap_val val = {115 .ifindex = IFINDEX_LO,116 };117 __u32 len = sizeof(info);118 int err, dm_fd_frags, map_fd;119 __u32 idx = 0;120 121 skel = test_xdp_with_devmap_frags_helpers__open_and_load();122 if (!ASSERT_OK_PTR(skel, "test_xdp_with_devmap_helpers__open_and_load"))123 return;124 125 dm_fd_frags = bpf_program__fd(skel->progs.xdp_dummy_dm_frags);126 map_fd = bpf_map__fd(skel->maps.dm_ports);127 err = bpf_prog_get_info_by_fd(dm_fd_frags, &info, &len);128 if (!ASSERT_OK(err, "bpf_prog_get_info_by_fd"))129 goto out_close;130 131 val.bpf_prog.fd = dm_fd_frags;132 err = bpf_map_update_elem(map_fd, &idx, &val, 0);133 ASSERT_OK(err, "Add frags program to devmap entry");134 135 err = bpf_map_lookup_elem(map_fd, &idx, &val);136 ASSERT_OK(err, "Read devmap entry");137 ASSERT_EQ(info.id, val.bpf_prog.id,138 "Match program id to devmap entry prog_id");139 140 /* Try to attach BPF_XDP program to devmap when we have141 * already loaded a BPF_XDP program with frags on the map142 */143 idx = 1;144 val.ifindex = 1;145 val.bpf_prog.fd = bpf_program__fd(skel->progs.xdp_dummy_dm);146 err = bpf_map_update_elem(map_fd, &idx, &val, 0);147 ASSERT_NEQ(err, 0, "Add BPF_XDP program to devmap entry");148 149out_close:150 test_xdp_with_devmap_frags_helpers__destroy(skel);151}152 153static void test_xdp_with_devmap_helpers_veth(void)154{155 struct test_xdp_with_devmap_helpers *skel = NULL;156 struct bpf_prog_info info = {};157 struct bpf_devmap_val val = {};158 struct nstoken *nstoken = NULL;159 __u32 len = sizeof(info);160 int err, dm_fd, dm_fd_redir, map_fd, ifindex_dst;161 char data[10] = {};162 __u32 idx = 0;163 164 SYS(out_close, "ip netns add %s", TEST_NS);165 nstoken = open_netns(TEST_NS);166 if (!ASSERT_OK_PTR(nstoken, "open_netns"))167 goto out_close;168 169 SYS(out_close, "ip link add veth_src type veth peer name veth_dst");170 SYS(out_close, "ip link set dev veth_src up");171 SYS(out_close, "ip link set dev veth_dst up");172 173 val.ifindex = if_nametoindex("veth_src");174 ifindex_dst = if_nametoindex("veth_dst");175 if (!ASSERT_NEQ(val.ifindex, 0, "val.ifindex") ||176 !ASSERT_NEQ(ifindex_dst, 0, "ifindex_dst"))177 goto out_close;178 179 skel = test_xdp_with_devmap_helpers__open_and_load();180 if (!ASSERT_OK_PTR(skel, "test_xdp_with_devmap_helpers__open_and_load"))181 goto out_close;182 183 dm_fd_redir = bpf_program__fd(skel->progs.xdp_redir_prog);184 err = bpf_xdp_attach(val.ifindex, dm_fd_redir, XDP_FLAGS_DRV_MODE, NULL);185 if (!ASSERT_OK(err, "Attach of program with 8-byte devmap"))186 goto out_close;187 188 dm_fd = bpf_program__fd(skel->progs.xdp_dummy_dm);189 map_fd = bpf_map__fd(skel->maps.dm_ports);190 err = bpf_prog_get_info_by_fd(dm_fd, &info, &len);191 if (!ASSERT_OK(err, "bpf_prog_get_info_by_fd"))192 goto out_close;193 194 val.bpf_prog.fd = dm_fd;195 err = bpf_map_update_elem(map_fd, &idx, &val, 0);196 ASSERT_OK(err, "Add program to devmap entry");197 198 err = bpf_map_lookup_elem(map_fd, &idx, &val);199 ASSERT_OK(err, "Read devmap entry");200 ASSERT_EQ(info.id, val.bpf_prog.id, "Match program id to devmap entry prog_id");201 202 /* attach dummy to other side to enable reception */203 dm_fd = bpf_program__fd(skel->progs.xdp_dummy_prog);204 err = bpf_xdp_attach(ifindex_dst, dm_fd, XDP_FLAGS_DRV_MODE, NULL);205 if (!ASSERT_OK(err, "Attach of dummy XDP"))206 goto out_close;207 208 /* send a packet to trigger any potential bugs in there */209 DECLARE_LIBBPF_OPTS(bpf_test_run_opts, opts,210 .data_in = &data,211 .data_size_in = 10,212 .flags = BPF_F_TEST_XDP_LIVE_FRAMES,213 .repeat = 1,214 );215 err = bpf_prog_test_run_opts(dm_fd_redir, &opts);216 ASSERT_OK(err, "XDP test run");217 218 /* wait for the packets to be flushed */219 kern_sync_rcu();220 221 err = bpf_xdp_detach(val.ifindex, XDP_FLAGS_DRV_MODE, NULL);222 ASSERT_OK(err, "XDP program detach");223 224 err = bpf_xdp_detach(ifindex_dst, XDP_FLAGS_DRV_MODE, NULL);225 ASSERT_OK(err, "XDP program detach");226 227out_close:228 close_netns(nstoken);229 SYS_NOFAIL("ip netns del %s", TEST_NS);230 test_xdp_with_devmap_helpers__destroy(skel);231}232 233void serial_test_xdp_devmap_attach(void)234{235 if (test__start_subtest("DEVMAP with programs in entries"))236 test_xdp_with_devmap_helpers();237 238 if (test__start_subtest("DEVMAP with frags programs in entries"))239 test_xdp_with_devmap_frags_helpers();240 241 if (test__start_subtest("Verifier check of DEVMAP programs"))242 test_neg_xdp_devmap_helpers();243 244 if (test__start_subtest("DEVMAP with programs in entries on veth"))245 test_xdp_with_devmap_helpers_veth();246}247