Commit 80bea9ac authored by Magnus Karlsson's avatar Magnus Karlsson Committed by Alexei Starovoitov

selftests/xsk: add test when some packets are XDP_DROPed

Add a new test where some of the packets are not passed to the AF_XDP
socket and instead get a XDP_DROP verdict. This is important as it
tests the recycling mechanism of the buffer pool. If a packet is not
sent to the AF_XDP socket, the buffer the packet resides in is instead
recycled so it can be used again without the round-trip to user
space. The test introduces a new XDP program that drops every other
packet.
Signed-off-by: default avatarMagnus Karlsson <magnus.karlsson@intel.com>
Acked-by: default avatarMaciej Fijalkowski <maciej.fijalkowski@intel.com>
Link: https://lore.kernel.org/r/20230111093526.11682-13-magnus.karlsson@gmail.comSigned-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent f0a249df
...@@ -11,9 +11,20 @@ struct { ...@@ -11,9 +11,20 @@ struct {
__uint(value_size, sizeof(int)); __uint(value_size, sizeof(int));
} xsk SEC(".maps"); } xsk SEC(".maps");
static unsigned int idx;
SEC("xdp") int xsk_def_prog(struct xdp_md *xdp) SEC("xdp") int xsk_def_prog(struct xdp_md *xdp)
{ {
return bpf_redirect_map(&xsk, 0, XDP_DROP); return bpf_redirect_map(&xsk, 0, XDP_DROP);
} }
SEC("xdp") int xsk_xdp_drop(struct xdp_md *xdp)
{
/* Drop every other packet */
if (idx++ % 2)
return XDP_DROP;
return bpf_redirect_map(&xsk, 0, XDP_DROP);
}
char _license[] SEC("license") = "GPL"; char _license[] SEC("license") = "GPL";
...@@ -1646,6 +1646,34 @@ static void testapp_invalid_desc(struct test_spec *test) ...@@ -1646,6 +1646,34 @@ static void testapp_invalid_desc(struct test_spec *test)
pkt_stream_restore_default(test); pkt_stream_restore_default(test);
} }
static void testapp_xdp_drop(struct test_spec *test)
{
struct ifobject *ifobj = test->ifobj_rx;
int err;
test_spec_set_name(test, "XDP_DROP_HALF");
xsk_detach_xdp_program(ifobj->ifindex, ifobj->xdp_flags);
err = xsk_attach_xdp_program(ifobj->xdp_progs->progs.xsk_xdp_drop, ifobj->ifindex,
ifobj->xdp_flags);
if (err) {
printf("Error attaching XDP_DROP program\n");
test->fail = true;
return;
}
pkt_stream_receive_half(test);
testapp_validate_traffic(test);
pkt_stream_restore_default(test);
xsk_detach_xdp_program(ifobj->ifindex, ifobj->xdp_flags);
err = xsk_attach_xdp_program(ifobj->xdp_progs->progs.xsk_def_prog, ifobj->ifindex,
ifobj->xdp_flags);
if (err) {
printf("Error restoring default XDP program\n");
exit_with_error(-err);
}
}
static int xsk_load_xdp_programs(struct ifobject *ifobj) static int xsk_load_xdp_programs(struct ifobject *ifobj)
{ {
ifobj->xdp_progs = xsk_xdp_progs__open_and_load(); ifobj->xdp_progs = xsk_xdp_progs__open_and_load();
...@@ -1796,6 +1824,9 @@ static void run_pkt_test(struct test_spec *test, enum test_mode mode, enum test_ ...@@ -1796,6 +1824,9 @@ static void run_pkt_test(struct test_spec *test, enum test_mode mode, enum test_
case TEST_TYPE_HEADROOM: case TEST_TYPE_HEADROOM:
testapp_headroom(test); testapp_headroom(test);
break; break;
case TEST_TYPE_XDP_DROP_HALF:
testapp_xdp_drop(test);
break;
default: default:
break; break;
} }
......
...@@ -87,6 +87,7 @@ enum test_type { ...@@ -87,6 +87,7 @@ enum test_type {
TEST_TYPE_STATS_RX_FULL, TEST_TYPE_STATS_RX_FULL,
TEST_TYPE_STATS_FILL_EMPTY, TEST_TYPE_STATS_FILL_EMPTY,
TEST_TYPE_BPF_RES, TEST_TYPE_BPF_RES,
TEST_TYPE_XDP_DROP_HALF,
TEST_TYPE_MAX TEST_TYPE_MAX
}; };
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment