1 // SPDX-License-Identifier: GPL-2.0
2 
3 #include <test_progs.h>
4 #include <network_helpers.h>
5 #include "test_parse_tcp_hdr_opt.skel.h"
6 #include "test_parse_tcp_hdr_opt_dynptr.skel.h"
7 #include "test_tcp_hdr_options.h"
8 
9 struct test_pkt {
10 	struct ipv6_packet pk6_v6;
11 	u8 options[16];
12 } __packed;
13 
14 struct test_pkt pkt = {
15 	.pk6_v6.eth.h_proto = __bpf_constant_htons(ETH_P_IPV6),
16 	.pk6_v6.iph.nexthdr = IPPROTO_TCP,
17 	.pk6_v6.iph.payload_len = __bpf_constant_htons(MAGIC_BYTES),
18 	.pk6_v6.tcp.urg_ptr = 123,
19 	.pk6_v6.tcp.doff = 9, /* 16 bytes of options */
20 
21 	.options = {
22 		TCPOPT_MSS, 4, 0x05, 0xB4, TCPOPT_NOP, TCPOPT_NOP,
23 		0, 6, 0xBB, 0xBB, 0xBB, 0xBB, TCPOPT_EOL
24 	},
25 };
26 
test_parse_opt(void)27 static void test_parse_opt(void)
28 {
29 	struct test_parse_tcp_hdr_opt *skel;
30 	struct bpf_program *prog;
31 	char buf[128];
32 	int err;
33 
34 	LIBBPF_OPTS(bpf_test_run_opts, topts,
35 		    .data_in = &pkt,
36 		    .data_size_in = sizeof(pkt),
37 		    .data_out = buf,
38 		    .data_size_out = sizeof(buf),
39 		    .repeat = 3,
40 	);
41 
42 	skel = test_parse_tcp_hdr_opt__open_and_load();
43 	if (!ASSERT_OK_PTR(skel, "skel_open_and_load"))
44 		return;
45 
46 	pkt.options[6] = skel->rodata->tcp_hdr_opt_kind_tpr;
47 	prog = skel->progs.xdp_ingress_v6;
48 
49 	err = bpf_prog_test_run_opts(bpf_program__fd(prog), &topts);
50 	ASSERT_OK(err, "ipv6 test_run");
51 	ASSERT_EQ(topts.retval, XDP_PASS, "ipv6 test_run retval");
52 	ASSERT_EQ(skel->bss->server_id, 0xBBBBBBBB, "server id");
53 
54 	test_parse_tcp_hdr_opt__destroy(skel);
55 }
56 
test_parse_opt_dynptr(void)57 static void test_parse_opt_dynptr(void)
58 {
59 	struct test_parse_tcp_hdr_opt_dynptr *skel;
60 	struct bpf_program *prog;
61 	char buf[128];
62 	int err;
63 
64 	LIBBPF_OPTS(bpf_test_run_opts, topts,
65 		    .data_in = &pkt,
66 		    .data_size_in = sizeof(pkt),
67 		    .data_out = buf,
68 		    .data_size_out = sizeof(buf),
69 		    .repeat = 3,
70 	);
71 
72 	skel = test_parse_tcp_hdr_opt_dynptr__open_and_load();
73 	if (!ASSERT_OK_PTR(skel, "skel_open_and_load"))
74 		return;
75 
76 	pkt.options[6] = skel->rodata->tcp_hdr_opt_kind_tpr;
77 	prog = skel->progs.xdp_ingress_v6;
78 
79 	err = bpf_prog_test_run_opts(bpf_program__fd(prog), &topts);
80 	ASSERT_OK(err, "ipv6 test_run");
81 	ASSERT_EQ(topts.retval, XDP_PASS, "ipv6 test_run retval");
82 	ASSERT_EQ(skel->bss->server_id, 0xBBBBBBBB, "server id");
83 
84 	test_parse_tcp_hdr_opt_dynptr__destroy(skel);
85 }
86 
test_parse_tcp_hdr_opt(void)87 void test_parse_tcp_hdr_opt(void)
88 {
89 	if (test__start_subtest("parse_tcp_hdr_opt"))
90 		test_parse_opt();
91 	if (test__start_subtest("parse_tcp_hdr_opt_dynptr"))
92 		test_parse_opt_dynptr();
93 }
94