From nobody Wed Apr 1 22:11:48 2026 Received: from out-173.mta0.migadu.com (out-173.mta0.migadu.com [91.218.175.173]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id B43C236B04E for ; Wed, 1 Apr 2026 11:06:00 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=91.218.175.173 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1775041562; cv=none; b=EsFxQYr+BAu1GZp/yNb3dRjCMDl4JttGFKazGZq6m5s5xdOPgp3C33G6Lo5mY1jpzvJtiKVV+Fl2sSYeDOjKedh/bFTz4kPXiTSUy+S9Rvwq2JB72ZdPY7TnY0usjRM6f8+O1RQ9ufkwuQigdPY+OpkO19AxOKZPuBGQdJdw8t0= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1775041562; c=relaxed/simple; bh=cPBmDKGfQA/x2FNSfuO8SV/6uR6TOlLuz1aJaJ0nOoU=; h=From:To:Cc:Subject:Date:Message-ID:In-Reply-To:References: MIME-Version; b=DEPCPxGZymhc7DAhlxH1U4v0csL6ykMdCUUosyliEQoG/6IEaxIU4hHHYv6PXgTNtZc/qFkrj3u8V6yFs6jVawz4W0SjPeeNz8pDnYgygLDxChbo8ZQoq1clxYZeT9ydclnnX6K6E5n2fWC0D7Uj/4J87yTAJsX2TjxgjZUkkJc= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=linux.dev; spf=pass smtp.mailfrom=linux.dev; dkim=pass (1024-bit key) header.d=linux.dev header.i=@linux.dev header.b=SUVp+7fG; arc=none smtp.client-ip=91.218.175.173 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=none dis=none) header.from=linux.dev Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=linux.dev Authentication-Results: smtp.subspace.kernel.org; dkim=pass (1024-bit key) header.d=linux.dev header.i=@linux.dev header.b="SUVp+7fG" X-Report-Abuse: Please report any abuse attempt to abuse@migadu.com and include these headers. DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=linux.dev; s=key1; t=1775041558; h=from:from:reply-to:subject:subject:date:date:message-id:message-id: to:to:cc:cc:mime-version:mime-version: content-transfer-encoding:content-transfer-encoding: in-reply-to:in-reply-to:references:references; bh=UpPfSjlH3UMy2LiUOfTVIf8nD9bsIvAShICExqAy5Cs=; b=SUVp+7fGR2ZogNpHKxKV6haU5L6MxqD3YRNIcC8h9kzsbOrqQccBB+m54+itbvDL97l1sV vGvUR+mUy8S+IJnC1malfcNeXguVe5Mt2eQHDrEtqsw4z+JX8UmjNINzxyqASOCFKYvgRq 7vPR4Bj8Nski7GA9MZ6WN0+2IFBufCw= From: Jiayuan Chen To: bpf@vger.kernel.org Cc: Jiayuan Chen , Martin KaFai Lau , Daniel Borkmann , John Fastabend , Stanislav Fomichev , Alexei Starovoitov , Andrii Nakryiko , Eduard Zingerman , Song Liu , Yonghong Song , KP Singh , Hao Luo , Jiri Olsa , "David S. Miller" , Eric Dumazet , Jakub Kicinski , Paolo Abeni , Simon Horman , Shuah Khan , Kuniyuki Iwashima , netdev@vger.kernel.org, linux-kernel@vger.kernel.org, linux-kselftest@vger.kernel.org Subject: [PATCH bpf v5 2/2] selftests/bpf: Add protocol check test for bpf_sk_assign_tcp_reqsk() Date: Wed, 1 Apr 2026 19:04:50 +0800 Message-ID: <20260401110511.73355-3-jiayuan.chen@linux.dev> In-Reply-To: <20260401110511.73355-1-jiayuan.chen@linux.dev> References: <20260401110511.73355-1-jiayuan.chen@linux.dev> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable X-Migadu-Flow: FLOW_OUT Content-Type: text/plain; charset="utf-8" Add test_tcp_custom_syncookie_protocol_check to verify that bpf_sk_assign_tcp_reqsk() rejects non-TCP skbs. The test sends a UDP packet through TC ingress where a BPF program calls bpf_sk_assign_tcp_reqsk() on it and checks that the kfunc returns an error. A UDP server recv() is used as synchronization to ensure the BPF program has finished processing before checking the result. Without the fix in bpf_sk_assign_tcp_reqsk(), the kfunc succeeds and attaches a TCP reqsk to the UDP skb, which causes a null pointer dereference panic when the kernel processes it through the UDP receive path. Test result: ./test_progs -a tcp_custom_syncookie_protocol_check -v setup_netns:PASS:create netns 0 nsec setup_netns:PASS:ip 0 nsec write_sysctl:PASS:open sysctl 0 nsec write_sysctl:PASS:write sysctl 0 nsec setup_netns:PASS:write_sysctl 0 nsec test_tcp_custom_syncookie_protocol_check:PASS:open_and_load 0 nsec setup_tc:PASS:qdisc add dev lo clsact 0 nsec setup_tc:PASS:filter add dev lo ingress 0 nsec run_protocol_check:PASS:start tcp_server 0 nsec run_protocol_check:PASS:start udp_server 0 nsec run_protocol_check:PASS:connect udp_client 0 nsec run_protocol_check:PASS:send udp 0 nsec run_protocol_check:PASS:recv udp 0 nsec run_protocol_check:PASS:recv data 0 nsec run_protocol_check:PASS:udp_intercepted 0 nsec run_protocol_check:PASS:assign_ret 0 nsec #471/1 tcp_custom_syncookie_protocol_check/IPv4 TCP:OK run_protocol_check:PASS:start tcp_server 0 nsec run_protocol_check:PASS:start udp_server 0 nsec run_protocol_check:PASS:connect udp_client 0 nsec run_protocol_check:PASS:send udp 0 nsec run_protocol_check:PASS:recv udp 0 nsec run_protocol_check:PASS:recv data 0 nsec run_protocol_check:PASS:udp_intercepted 0 nsec run_protocol_check:PASS:assign_ret 0 nsec #471/2 tcp_custom_syncookie_protocol_check/IPv6 TCP:OK #471 tcp_custom_syncookie_protocol_check:OK Summary: 1/2 PASSED, 0 SKIPPED, 0 FAILED Cc: Jiayuan Chen Signed-off-by: Jiayuan Chen --- .../bpf/prog_tests/tcp_custom_syncookie.c | 93 ++++++++++++++- .../bpf/progs/test_tcp_custom_syncookie.c | 109 ++++++++++++++++++ 2 files changed, 198 insertions(+), 4 deletions(-) diff --git a/tools/testing/selftests/bpf/prog_tests/tcp_custom_syncookie.c = b/tools/testing/selftests/bpf/prog_tests/tcp_custom_syncookie.c index eaf441dc7e79..017836581669 100644 --- a/tools/testing/selftests/bpf/prog_tests/tcp_custom_syncookie.c +++ b/tools/testing/selftests/bpf/prog_tests/tcp_custom_syncookie.c @@ -5,6 +5,7 @@ #include #include #include +#include =20 #include "test_progs.h" #include "cgroup_helpers.h" @@ -47,11 +48,10 @@ static int setup_netns(void) return -1; } =20 -static int setup_tc(struct test_tcp_custom_syncookie *skel) +static int setup_tc(int prog_fd) { LIBBPF_OPTS(bpf_tc_hook, qdisc_lo, .attach_point =3D BPF_TC_INGRESS); - LIBBPF_OPTS(bpf_tc_opts, tc_attach, - .prog_fd =3D bpf_program__fd(skel->progs.tcp_custom_syncookie)); + LIBBPF_OPTS(bpf_tc_opts, tc_attach, .prog_fd =3D prog_fd); =20 qdisc_lo.ifindex =3D if_nametoindex("lo"); if (!ASSERT_OK(bpf_tc_hook_create(&qdisc_lo), "qdisc add dev lo clsact")) @@ -127,7 +127,7 @@ void test_tcp_custom_syncookie(void) if (!ASSERT_OK_PTR(skel, "open_and_load")) return; =20 - if (setup_tc(skel)) + if (setup_tc(bpf_program__fd(skel->progs.tcp_custom_syncookie))) goto destroy_skel; =20 for (i =3D 0; i < ARRAY_SIZE(test_cases); i++) { @@ -145,6 +145,91 @@ void test_tcp_custom_syncookie(void) =20 destroy_skel: system("tc qdisc del dev lo clsact"); + test_tcp_custom_syncookie__destroy(skel); +} + +/* Test: bpf_sk_assign_tcp_reqsk() should reject non-TCP skb. + * + * Send a UDP packet through TC ingress where a BPF program calls + * bpf_sk_assign_tcp_reqsk() on it. The kfunc should return an error + * because the skb carries UDP, not TCP. + * + * TCP and UDP servers share the same port. The BPF program intercepts + * the UDP packet, looks up the TCP listener via the dest port, and + * attempts to assign a TCP reqsk to the UDP skb. + */ +static void run_protocol_check(struct test_tcp_custom_syncookie *skel, + int family, const char *addr) +{ + int tcp_server, udp_server, udp_client; + char buf[] =3D "test"; + int port, ret; + + tcp_server =3D start_server(family, SOCK_STREAM, addr, 0, 0); + if (!ASSERT_NEQ(tcp_server, -1, "start tcp_server")) + return; + + port =3D ntohs(get_socket_local_port(tcp_server)); + + /* UDP server on same port for synchronization and port sharing */ + udp_server =3D start_server(family, SOCK_DGRAM, addr, port, 0); + if (!ASSERT_NEQ(udp_server, -1, "start udp_server")) + goto close_tcp; + + skel->bss->udp_intercepted =3D false; + skel->bss->assign_ret =3D 0; + + udp_client =3D connect_to_fd(udp_server, 0); + if (!ASSERT_NEQ(udp_client, -1, "connect udp_client")) + goto close_udp_server; + + ret =3D send(udp_client, buf, sizeof(buf), 0); + if (!ASSERT_EQ(ret, sizeof(buf), "send udp")) + goto close_udp_client; + + memset(buf, 0, sizeof(buf)); + + /* recv() ensures TC ingress BPF has processed the skb */ + ret =3D recv(udp_server, buf, sizeof(buf), 0); + if (!ASSERT_EQ(ret, sizeof(buf), "recv udp")) + goto close_udp_client; + + ASSERT_STREQ(buf, "test", "recv data"); + ASSERT_EQ(skel->bss->udp_intercepted, true, "udp_intercepted"); + ASSERT_EQ(skel->bss->assign_ret, -EINVAL, "assign_ret"); + +close_udp_client: + close(udp_client); +close_udp_server: + close(udp_server); +close_tcp: + close(tcp_server); +} + +void test_tcp_custom_syncookie_protocol_check(void) +{ + struct test_tcp_custom_syncookie *skel; + int i; + + if (setup_netns()) + return; + + skel =3D test_tcp_custom_syncookie__open_and_load(); + if (!ASSERT_OK_PTR(skel, "open_and_load")) + return; + + if (setup_tc(bpf_program__fd(skel->progs.tcp_custom_syncookie_badproto))) + goto destroy_skel; + + for (i =3D 0; i < ARRAY_SIZE(test_cases); i++) { + if (!test__start_subtest(test_cases[i].name)) + continue; =20 + run_protocol_check(skel, test_cases[i].family, + test_cases[i].addr); + } + +destroy_skel: + system("tc qdisc del dev lo clsact"); test_tcp_custom_syncookie__destroy(skel); } diff --git a/tools/testing/selftests/bpf/progs/test_tcp_custom_syncookie.c = b/tools/testing/selftests/bpf/progs/test_tcp_custom_syncookie.c index 7d5293de1952..331435c837cc 100644 --- a/tools/testing/selftests/bpf/progs/test_tcp_custom_syncookie.c +++ b/tools/testing/selftests/bpf/progs/test_tcp_custom_syncookie.c @@ -588,4 +588,113 @@ int tcp_custom_syncookie(struct __sk_buff *skb) return tcp_handle_ack(&ctx); } =20 +/* Test: call bpf_sk_assign_tcp_reqsk() on a UDP skb. + * The kfunc should reject it because the skb is not TCP. + * + * TCP and UDP servers share the same port. The BPF program intercepts + * UDP packets, looks up the TCP listener on the same port, and tries + * to assign a TCP reqsk to the UDP skb. + */ +int assign_ret; +bool udp_intercepted; + +static int badproto_lookup_assign(struct __sk_buff *skb, struct udphdr *ud= p, + struct bpf_sock_tuple *tuple, u32 tuple_size) +{ + struct bpf_tcp_req_attrs attrs =3D {}; + struct bpf_sock *skc; + struct sock *sk; + + skc =3D bpf_skc_lookup_tcp(skb, tuple, tuple_size, -1, 0); + if (!skc) + return TC_ACT_OK; + + if (skc->state !=3D TCP_LISTEN) { + bpf_sk_release(skc); + return TC_ACT_OK; + } + + sk =3D (struct sock *)bpf_skc_to_tcp_sock(skc); + if (!sk) { + bpf_sk_release(skc); + return TC_ACT_OK; + } + + attrs.mss =3D 1460; + attrs.wscale_ok =3D 1; + attrs.snd_wscale =3D 7; + attrs.rcv_wscale =3D 7; + attrs.sack_ok =3D 1; + + assign_ret =3D bpf_sk_assign_tcp_reqsk(skb, sk, &attrs, sizeof(attrs)); + + bpf_sk_release(skc); + return TC_ACT_OK; +} + +SEC("tc") +int tcp_custom_syncookie_badproto(struct __sk_buff *skb) +{ + void *data =3D (void *)(long)skb->data; + void *data_end =3D (void *)(long)skb->data_end; + struct bpf_sock_tuple tuple =3D {}; + struct ethhdr *eth; + struct iphdr *iph; + struct ipv6hdr *ip6h; + struct udphdr *udp; + + eth =3D (struct ethhdr *)data; + if (eth + 1 > data_end) + return TC_ACT_OK; + + switch (bpf_ntohs(eth->h_proto)) { + case ETH_P_IP: + iph =3D (struct iphdr *)(eth + 1); + if (iph + 1 > data_end) + return TC_ACT_OK; + + if (iph->protocol !=3D IPPROTO_UDP) + return TC_ACT_OK; + + udp =3D (struct udphdr *)(iph + 1); + if (udp + 1 > data_end) + return TC_ACT_OK; + + udp_intercepted =3D true; + + tuple.ipv4.saddr =3D iph->saddr; + tuple.ipv4.daddr =3D iph->daddr; + tuple.ipv4.sport =3D udp->source; + tuple.ipv4.dport =3D udp->dest; + + return badproto_lookup_assign(skb, udp, &tuple, + sizeof(tuple.ipv4)); + case ETH_P_IPV6: + ip6h =3D (struct ipv6hdr *)(eth + 1); + if (ip6h + 1 > data_end) + return TC_ACT_OK; + + if (ip6h->nexthdr !=3D IPPROTO_UDP) + return TC_ACT_OK; + + udp =3D (struct udphdr *)(ip6h + 1); + if (udp + 1 > data_end) + return TC_ACT_OK; + + udp_intercepted =3D true; + + __builtin_memcpy(tuple.ipv6.saddr, &ip6h->saddr, + sizeof(tuple.ipv6.saddr)); + __builtin_memcpy(tuple.ipv6.daddr, &ip6h->daddr, + sizeof(tuple.ipv6.daddr)); + tuple.ipv6.sport =3D udp->source; + tuple.ipv6.dport =3D udp->dest; + + return badproto_lookup_assign(skb, udp, &tuple, + sizeof(tuple.ipv6)); + default: + return TC_ACT_OK; + } +} + char _license[] SEC("license") =3D "GPL"; --=20 2.43.0