From ca577d08f88b410bc1b87d93f70bf68c03083b96 Mon Sep 17 00:00:00 2001 From: Kuniyuki Iwashima Date: Thu, 8 Oct 2026 03:15:23 +0000 Subject: selftest: bpf: Use bpf_tcp_ops_set_flags() in bpf_tcp_ops_hdr.c. The next patch guards bpf_tcp_ops callbacks in the fast path with per-socket flags. Let's use bpf_tcp_ops_set_flags() to enable flags for parse_hdr, write_hdr_opt, and hdr_opt_len. Note that BPF_TCP_OPS_FLAG_PARSE_HDR_OPT_ALL and BPF_TCP_OPS_FLAG_PARSE_HDR_OPT_UNKNOWN are used on the server and client, respectively, for better coverage. Signed-off-by: Kuniyuki Iwashima Signed-off-by: Alexei Starovoitov Reviewed-by: Amery Hung Acked-by: Stanislav Fomichev Link: https://patch.msgid.link/20261008031604.256498-4-kuniyu@google.com --- tools/testing/selftests/bpf/progs/bpf_tcp_ops_hdr.c | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'tools') diff --git a/tools/testing/selftests/bpf/progs/bpf_tcp_ops_hdr.c b/tools/testing/selftests/bpf/progs/bpf_tcp_ops_hdr.c index f3e3dff13784..706e94d794e4 100644 --- a/tools/testing/selftests/bpf/progs/bpf_tcp_ops_hdr.c +++ b/tools/testing/selftests/bpf/progs/bpf_tcp_ops_hdr.c @@ -18,6 +18,24 @@ int found_cnt; __u8 found_d0; __u8 found_d1; +SEC("struct_ops") +void BPF_PROG(test_listen, struct sock *sk) +{ + bpf_tcp_ops_set_flags((struct tcp_sock *)sk, + BPF_TCP_OPS_FLAG_PARSE_HDR_OPT_ALL | + BPF_TCP_OPS_FLAG_WRITE_HDR_OPT, + 0); +} + +SEC("struct_ops") +void BPF_PROG(test_connect, struct sock *sk) +{ + bpf_tcp_ops_set_flags((struct tcp_sock *)sk, + BPF_TCP_OPS_FLAG_PARSE_HDR_OPT_UNKNOWN | + BPF_TCP_OPS_FLAG_WRITE_HDR_OPT, + 0); +} + SEC("struct_ops") void BPF_PROG(test_hdr_opt_len, struct sock *sk, struct sk_buff *skb, struct request_sock *req, struct sk_buff *syn_skb, @@ -78,6 +96,8 @@ void BPF_PROG(test_parse_hdr, struct sock *sk, struct sk_buff *skb) SEC(".struct_ops.link") struct bpf_tcp_ops test_hdr_ops = { + .listen = (void *)test_listen, + .connect = (void *)test_connect, .hdr_opt_len = (void *)test_hdr_opt_len, .write_hdr_opt = (void *)test_write_hdr_opt, .parse_hdr = (void *)test_parse_hdr, -- cgit v1.2.3