1*4a58963dSCong Wang // SPDX-License-Identifier: GPL-2.0
2*4a58963dSCong Wang #include <error.h>
3*4a58963dSCong Wang #include <test_progs.h>
4*4a58963dSCong Wang #include <linux/pkt_cls.h>
5*4a58963dSCong Wang
6*4a58963dSCong Wang #include "test_tc_change_tail.skel.h"
7*4a58963dSCong Wang #include "socket_helpers.h"
8*4a58963dSCong Wang
9*4a58963dSCong Wang #define LO_IFINDEX 1
10*4a58963dSCong Wang
test_tc_change_tail(void)11*4a58963dSCong Wang void test_tc_change_tail(void)
12*4a58963dSCong Wang {
13*4a58963dSCong Wang LIBBPF_OPTS(bpf_tcx_opts, tcx_opts);
14*4a58963dSCong Wang struct test_tc_change_tail *skel = NULL;
15*4a58963dSCong Wang struct bpf_link *link;
16*4a58963dSCong Wang int c1, p1;
17*4a58963dSCong Wang char buf[2];
18*4a58963dSCong Wang int ret;
19*4a58963dSCong Wang
20*4a58963dSCong Wang skel = test_tc_change_tail__open_and_load();
21*4a58963dSCong Wang if (!ASSERT_OK_PTR(skel, "test_tc_change_tail__open_and_load"))
22*4a58963dSCong Wang return;
23*4a58963dSCong Wang
24*4a58963dSCong Wang link = bpf_program__attach_tcx(skel->progs.change_tail, LO_IFINDEX,
25*4a58963dSCong Wang &tcx_opts);
26*4a58963dSCong Wang if (!ASSERT_OK_PTR(link, "bpf_program__attach_tcx"))
27*4a58963dSCong Wang goto destroy;
28*4a58963dSCong Wang
29*4a58963dSCong Wang skel->links.change_tail = link;
30*4a58963dSCong Wang ret = create_pair(AF_INET, SOCK_DGRAM, &c1, &p1);
31*4a58963dSCong Wang if (!ASSERT_OK(ret, "create_pair"))
32*4a58963dSCong Wang goto destroy;
33*4a58963dSCong Wang
34*4a58963dSCong Wang ret = xsend(p1, "Tr", 2, 0);
35*4a58963dSCong Wang ASSERT_EQ(ret, 2, "xsend(p1)");
36*4a58963dSCong Wang ret = recv(c1, buf, 2, 0);
37*4a58963dSCong Wang ASSERT_EQ(ret, 2, "recv(c1)");
38*4a58963dSCong Wang ASSERT_EQ(skel->data->change_tail_ret, 0, "change_tail_ret");
39*4a58963dSCong Wang
40*4a58963dSCong Wang ret = xsend(p1, "G", 1, 0);
41*4a58963dSCong Wang ASSERT_EQ(ret, 1, "xsend(p1)");
42*4a58963dSCong Wang ret = recv(c1, buf, 2, 0);
43*4a58963dSCong Wang ASSERT_EQ(ret, 1, "recv(c1)");
44*4a58963dSCong Wang ASSERT_EQ(skel->data->change_tail_ret, 0, "change_tail_ret");
45*4a58963dSCong Wang
46*4a58963dSCong Wang ret = xsend(p1, "E", 1, 0);
47*4a58963dSCong Wang ASSERT_EQ(ret, 1, "xsend(p1)");
48*4a58963dSCong Wang ret = recv(c1, buf, 1, 0);
49*4a58963dSCong Wang ASSERT_EQ(ret, 1, "recv(c1)");
50*4a58963dSCong Wang ASSERT_EQ(skel->data->change_tail_ret, -EINVAL, "change_tail_ret");
51*4a58963dSCong Wang
52*4a58963dSCong Wang ret = xsend(p1, "Z", 1, 0);
53*4a58963dSCong Wang ASSERT_EQ(ret, 1, "xsend(p1)");
54*4a58963dSCong Wang ret = recv(c1, buf, 1, 0);
55*4a58963dSCong Wang ASSERT_EQ(ret, 1, "recv(c1)");
56*4a58963dSCong Wang ASSERT_EQ(skel->data->change_tail_ret, -EINVAL, "change_tail_ret");
57*4a58963dSCong Wang
58*4a58963dSCong Wang close(c1);
59*4a58963dSCong Wang close(p1);
60*4a58963dSCong Wang destroy:
61*4a58963dSCong Wang test_tc_change_tail__destroy(skel);
62*4a58963dSCong Wang }
63