1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
3 
4 #include <sys/types.h>
5 #include <sys/socket.h>
6 #include <net/if.h>
7 #include <linux/in6.h>
8 
9 #include "test_progs.h"
10 #include "network_helpers.h"
11 #include "decap_sanity.skel.h"
12 
13 #define SYS(fmt, ...)						\
14 	({							\
15 		char cmd[1024];					\
16 		snprintf(cmd, sizeof(cmd), fmt, ##__VA_ARGS__);	\
17 		if (!ASSERT_OK(system(cmd), cmd))		\
18 			goto fail;				\
19 	})
20 
21 #define NS_TEST "decap_sanity_ns"
22 #define IPV6_IFACE_ADDR "face::1"
23 #define UDP_TEST_PORT 7777
24 
25 void test_decap_sanity(void)
26 {
27 	LIBBPF_OPTS(bpf_tc_hook, qdisc_hook, .attach_point = BPF_TC_EGRESS);
28 	LIBBPF_OPTS(bpf_tc_opts, tc_attach);
29 	struct nstoken *nstoken = NULL;
30 	struct decap_sanity *skel;
31 	struct sockaddr_in6 addr;
32 	socklen_t addrlen;
33 	char buf[128] = {};
34 	int sockfd, err;
35 
36 	skel = decap_sanity__open_and_load();
37 	if (!ASSERT_OK_PTR(skel, "skel open_and_load"))
38 		return;
39 
40 	SYS("ip netns add %s", NS_TEST);
41 	SYS("ip -net %s -6 addr add %s/128 dev lo nodad", NS_TEST, IPV6_IFACE_ADDR);
42 	SYS("ip -net %s link set dev lo up", NS_TEST);
43 
44 	nstoken = open_netns(NS_TEST);
45 	if (!ASSERT_OK_PTR(nstoken, "open_netns"))
46 		goto fail;
47 
48 	qdisc_hook.ifindex = if_nametoindex("lo");
49 	if (!ASSERT_GT(qdisc_hook.ifindex, 0, "if_nametoindex lo"))
50 		goto fail;
51 
52 	err = bpf_tc_hook_create(&qdisc_hook);
53 	if (!ASSERT_OK(err, "create qdisc hook"))
54 		goto fail;
55 
56 	tc_attach.prog_fd = bpf_program__fd(skel->progs.decap_sanity);
57 	err = bpf_tc_attach(&qdisc_hook, &tc_attach);
58 	if (!ASSERT_OK(err, "attach filter"))
59 		goto fail;
60 
61 	addrlen = sizeof(addr);
62 	err = make_sockaddr(AF_INET6, IPV6_IFACE_ADDR, UDP_TEST_PORT,
63 			    (void *)&addr, &addrlen);
64 	if (!ASSERT_OK(err, "make_sockaddr"))
65 		goto fail;
66 	sockfd = socket(AF_INET6, SOCK_DGRAM, 0);
67 	if (!ASSERT_NEQ(sockfd, -1, "socket"))
68 		goto fail;
69 	err = sendto(sockfd, buf, sizeof(buf), 0, (void *)&addr, addrlen);
70 	close(sockfd);
71 	if (!ASSERT_EQ(err, sizeof(buf), "send"))
72 		goto fail;
73 
74 	ASSERT_TRUE(skel->bss->init_csum_partial, "init_csum_partial");
75 	ASSERT_TRUE(skel->bss->final_csum_none, "final_csum_none");
76 	ASSERT_FALSE(skel->bss->broken_csum_start, "broken_csum_start");
77 
78 fail:
79 	if (nstoken) {
80 		bpf_tc_hook_destroy(&qdisc_hook);
81 		close_netns(nstoken);
82 	}
83 	system("ip netns del " NS_TEST " &> /dev/null");
84 	decap_sanity__destroy(skel);
85 }
86