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