xref: /linux/tools/testing/selftests/bpf/prog_tests/test_dst_clear.c (revision 91a4855d6c03e770e42f17c798a36a3c46e63de2)
1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2026 Meta Platforms, Inc. and affiliates. */
3 
4 #include <sys/types.h>
5 #include <sys/socket.h>
6 #include <net/if.h>
7 
8 #include "test_progs.h"
9 #include "network_helpers.h"
10 #include "test_dst_clear.skel.h"
11 
12 #define IPV4_IFACE_ADDR "1.0.0.1"
13 #define UDP_TEST_PORT 7777
14 
15 void test_ns_dst_clear(void)
16 {
17 	LIBBPF_OPTS(bpf_tcx_opts, tcx_opts);
18 	struct test_dst_clear *skel;
19 	struct sockaddr_in addr;
20 	struct bpf_link *link;
21 	socklen_t addrlen;
22 	char buf[128] = {};
23 	int sockfd, err;
24 
25 	skel = test_dst_clear__open_and_load();
26 	if (!ASSERT_OK_PTR(skel, "skel open_and_load"))
27 		return;
28 
29 	SYS(fail, "ip addr add %s/8 dev lo", IPV4_IFACE_ADDR);
30 
31 	link = bpf_program__attach_tcx(skel->progs.dst_clear,
32 				       if_nametoindex("lo"), &tcx_opts);
33 	if (!ASSERT_OK_PTR(link, "attach_tcx"))
34 		goto fail;
35 	skel->links.dst_clear = link;
36 
37 	addrlen = sizeof(addr);
38 	err = make_sockaddr(AF_INET, IPV4_IFACE_ADDR, UDP_TEST_PORT,
39 			    (void *)&addr, &addrlen);
40 	if (!ASSERT_OK(err, "make_sockaddr"))
41 		goto fail;
42 	sockfd = socket(AF_INET, SOCK_DGRAM, 0);
43 	if (!ASSERT_NEQ(sockfd, -1, "socket"))
44 		goto fail;
45 	err = sendto(sockfd, buf, sizeof(buf), 0, (void *)&addr, addrlen);
46 	close(sockfd);
47 	if (!ASSERT_EQ(err, sizeof(buf), "send"))
48 		goto fail;
49 
50 	ASSERT_TRUE(skel->bss->had_dst, "had_dst");
51 	ASSERT_TRUE(skel->bss->dst_cleared, "dst_cleared");
52 
53 fail:
54 	test_dst_clear__destroy(skel);
55 }
56