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