1 // SPDX-License-Identifier: GPL-2.0
2 #include <test_progs.h>
3 #include <network_helpers.h>
4 
5 void test_xdp_adjust_tail_shrink(void)
6 {
7 	const char *file = "./test_xdp_adjust_tail_shrink.o";
8 	__u32 duration, retval, size, expect_sz;
9 	struct bpf_object *obj;
10 	int err, prog_fd;
11 	char buf[128];
12 
13 	err = bpf_prog_load(file, BPF_PROG_TYPE_XDP, &obj, &prog_fd);
14 	if (CHECK_FAIL(err))
15 		return;
16 
17 	err = bpf_prog_test_run(prog_fd, 1, &pkt_v4, sizeof(pkt_v4),
18 				buf, &size, &retval, &duration);
19 
20 	CHECK(err || retval != XDP_DROP,
21 	      "ipv4", "err %d errno %d retval %d size %d\n",
22 	      err, errno, retval, size);
23 
24 	expect_sz = sizeof(pkt_v6) - 20;  /* Test shrink with 20 bytes */
25 	err = bpf_prog_test_run(prog_fd, 1, &pkt_v6, sizeof(pkt_v6),
26 				buf, &size, &retval, &duration);
27 	CHECK(err || retval != XDP_TX || size != expect_sz,
28 	      "ipv6", "err %d errno %d retval %d size %d expect-size %d\n",
29 	      err, errno, retval, size, expect_sz);
30 	bpf_object__close(obj);
31 }
32 
33 void test_xdp_adjust_tail_grow(void)
34 {
35 	const char *file = "./test_xdp_adjust_tail_grow.o";
36 	struct bpf_object *obj;
37 	char buf[4096]; /* avoid segfault: large buf to hold grow results */
38 	__u32 duration, retval, size, expect_sz;
39 	int err, prog_fd;
40 
41 	err = bpf_prog_load(file, BPF_PROG_TYPE_XDP, &obj, &prog_fd);
42 	if (CHECK_FAIL(err))
43 		return;
44 
45 	err = bpf_prog_test_run(prog_fd, 1, &pkt_v4, sizeof(pkt_v4),
46 				buf, &size, &retval, &duration);
47 	CHECK(err || retval != XDP_DROP,
48 	      "ipv4", "err %d errno %d retval %d size %d\n",
49 	      err, errno, retval, size);
50 
51 	expect_sz = sizeof(pkt_v6) + 40; /* Test grow with 40 bytes */
52 	err = bpf_prog_test_run(prog_fd, 1, &pkt_v6, sizeof(pkt_v6) /* 74 */,
53 				buf, &size, &retval, &duration);
54 	CHECK(err || retval != XDP_TX || size != expect_sz,
55 	      "ipv6", "err %d errno %d retval %d size %d expect-size %d\n",
56 	      err, errno, retval, size, expect_sz);
57 
58 	bpf_object__close(obj);
59 }
60 
61 void test_xdp_adjust_tail_grow2(void)
62 {
63 	const char *file = "./test_xdp_adjust_tail_grow.o";
64 	char buf[4096]; /* avoid segfault: large buf to hold grow results */
65 	int tailroom = 320; /* SKB_DATA_ALIGN(sizeof(struct skb_shared_info))*/;
66 	struct bpf_object *obj;
67 	int err, cnt, i;
68 	int max_grow;
69 
70 	struct bpf_prog_test_run_attr tattr = {
71 		.repeat		= 1,
72 		.data_in	= &buf,
73 		.data_out	= &buf,
74 		.data_size_in	= 0, /* Per test */
75 		.data_size_out	= 0, /* Per test */
76 	};
77 
78 	err = bpf_prog_load(file, BPF_PROG_TYPE_XDP, &obj, &tattr.prog_fd);
79 	if (CHECK_ATTR(err, "load", "err %d errno %d\n", err, errno))
80 		return;
81 
82 	/* Test case-64 */
83 	memset(buf, 1, sizeof(buf));
84 	tattr.data_size_in  =  64; /* Determine test case via pkt size */
85 	tattr.data_size_out = 128; /* Limit copy_size */
86 	/* Kernel side alloc packet memory area that is zero init */
87 	err = bpf_prog_test_run_xattr(&tattr);
88 
89 	CHECK_ATTR(errno != ENOSPC /* Due limit copy_size in bpf_test_finish */
90 		   || tattr.retval != XDP_TX
91 		   || tattr.data_size_out != 192, /* Expected grow size */
92 		   "case-64",
93 		   "err %d errno %d retval %d size %d\n",
94 		   err, errno, tattr.retval, tattr.data_size_out);
95 
96 	/* Extra checks for data contents */
97 	CHECK_ATTR(tattr.data_size_out != 192
98 		   || buf[0]   != 1 ||  buf[63]  != 1  /*  0-63  memset to 1 */
99 		   || buf[64]  != 0 ||  buf[127] != 0  /* 64-127 memset to 0 */
100 		   || buf[128] != 1 ||  buf[191] != 1, /*128-191 memset to 1 */
101 		   "case-64-data",
102 		   "err %d errno %d retval %d size %d\n",
103 		   err, errno, tattr.retval, tattr.data_size_out);
104 
105 	/* Test case-128 */
106 	memset(buf, 2, sizeof(buf));
107 	tattr.data_size_in  = 128; /* Determine test case via pkt size */
108 	tattr.data_size_out = sizeof(buf);   /* Copy everything */
109 	err = bpf_prog_test_run_xattr(&tattr);
110 
111 	max_grow = 4096 - XDP_PACKET_HEADROOM -	tailroom; /* 3520 */
112 	CHECK_ATTR(err
113 		   || tattr.retval != XDP_TX
114 		   || tattr.data_size_out != max_grow,/* Expect max grow size */
115 		   "case-128",
116 		   "err %d errno %d retval %d size %d expect-size %d\n",
117 		   err, errno, tattr.retval, tattr.data_size_out, max_grow);
118 
119 	/* Extra checks for data content: Count grow size, will contain zeros */
120 	for (i = 0, cnt = 0; i < sizeof(buf); i++) {
121 		if (buf[i] == 0)
122 			cnt++;
123 	}
124 	CHECK_ATTR((cnt != (max_grow - tattr.data_size_in)) /* Grow increase */
125 		   || tattr.data_size_out != max_grow, /* Total grow size */
126 		   "case-128-data",
127 		   "err %d errno %d retval %d size %d grow-size %d\n",
128 		   err, errno, tattr.retval, tattr.data_size_out, cnt);
129 
130 	bpf_object__close(obj);
131 }
132 
133 void test_xdp_adjust_tail(void)
134 {
135 	if (test__start_subtest("xdp_adjust_tail_shrink"))
136 		test_xdp_adjust_tail_shrink();
137 	if (test__start_subtest("xdp_adjust_tail_grow"))
138 		test_xdp_adjust_tail_grow();
139 	if (test__start_subtest("xdp_adjust_tail_grow2"))
140 		test_xdp_adjust_tail_grow2();
141 }
142