1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */ 3 4 #include <sys/types.h> 5 #include <net/if.h> 6 7 #include "test_progs.h" 8 #include "network_helpers.h" 9 #include "fib_lookup.skel.h" 10 11 #define NS_TEST "fib_lookup_ns" 12 #define IPV6_IFACE_ADDR "face::face" 13 #define IPV6_NUD_FAILED_ADDR "face::1" 14 #define IPV6_NUD_STALE_ADDR "face::2" 15 #define IPV4_IFACE_ADDR "10.0.0.254" 16 #define IPV4_NUD_FAILED_ADDR "10.0.0.1" 17 #define IPV4_NUD_STALE_ADDR "10.0.0.2" 18 #define DMAC "11:11:11:11:11:11" 19 #define DMAC_INIT { 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, } 20 21 struct fib_lookup_test { 22 const char *desc; 23 const char *daddr; 24 int expected_ret; 25 int lookup_flags; 26 __u8 dmac[6]; 27 }; 28 29 static const struct fib_lookup_test tests[] = { 30 { .desc = "IPv6 failed neigh", 31 .daddr = IPV6_NUD_FAILED_ADDR, .expected_ret = BPF_FIB_LKUP_RET_NO_NEIGH, }, 32 { .desc = "IPv6 stale neigh", 33 .daddr = IPV6_NUD_STALE_ADDR, .expected_ret = BPF_FIB_LKUP_RET_SUCCESS, 34 .dmac = DMAC_INIT, }, 35 { .desc = "IPv6 skip neigh", 36 .daddr = IPV6_NUD_FAILED_ADDR, .expected_ret = BPF_FIB_LKUP_RET_SUCCESS, 37 .lookup_flags = BPF_FIB_LOOKUP_SKIP_NEIGH, }, 38 { .desc = "IPv4 failed neigh", 39 .daddr = IPV4_NUD_FAILED_ADDR, .expected_ret = BPF_FIB_LKUP_RET_NO_NEIGH, }, 40 { .desc = "IPv4 stale neigh", 41 .daddr = IPV4_NUD_STALE_ADDR, .expected_ret = BPF_FIB_LKUP_RET_SUCCESS, 42 .dmac = DMAC_INIT, }, 43 { .desc = "IPv4 skip neigh", 44 .daddr = IPV4_NUD_FAILED_ADDR, .expected_ret = BPF_FIB_LKUP_RET_SUCCESS, 45 .lookup_flags = BPF_FIB_LOOKUP_SKIP_NEIGH, }, 46 }; 47 48 static int ifindex; 49 50 static int setup_netns(void) 51 { 52 int err; 53 54 SYS(fail, "ip link add veth1 type veth peer name veth2"); 55 SYS(fail, "ip link set dev veth1 up"); 56 57 SYS(fail, "ip addr add %s/64 dev veth1 nodad", IPV6_IFACE_ADDR); 58 SYS(fail, "ip neigh add %s dev veth1 nud failed", IPV6_NUD_FAILED_ADDR); 59 SYS(fail, "ip neigh add %s dev veth1 lladdr %s nud stale", IPV6_NUD_STALE_ADDR, DMAC); 60 61 SYS(fail, "ip addr add %s/24 dev veth1 nodad", IPV4_IFACE_ADDR); 62 SYS(fail, "ip neigh add %s dev veth1 nud failed", IPV4_NUD_FAILED_ADDR); 63 SYS(fail, "ip neigh add %s dev veth1 lladdr %s nud stale", IPV4_NUD_STALE_ADDR, DMAC); 64 65 err = write_sysctl("/proc/sys/net/ipv4/conf/veth1/forwarding", "1"); 66 if (!ASSERT_OK(err, "write_sysctl(net.ipv4.conf.veth1.forwarding)")) 67 goto fail; 68 69 err = write_sysctl("/proc/sys/net/ipv6/conf/veth1/forwarding", "1"); 70 if (!ASSERT_OK(err, "write_sysctl(net.ipv6.conf.veth1.forwarding)")) 71 goto fail; 72 73 return 0; 74 fail: 75 return -1; 76 } 77 78 static int set_lookup_params(struct bpf_fib_lookup *params, const char *daddr) 79 { 80 int ret; 81 82 memset(params, 0, sizeof(*params)); 83 84 params->l4_protocol = IPPROTO_TCP; 85 params->ifindex = ifindex; 86 87 if (inet_pton(AF_INET6, daddr, params->ipv6_dst) == 1) { 88 params->family = AF_INET6; 89 ret = inet_pton(AF_INET6, IPV6_IFACE_ADDR, params->ipv6_src); 90 if (!ASSERT_EQ(ret, 1, "inet_pton(IPV6_IFACE_ADDR)")) 91 return -1; 92 return 0; 93 } 94 95 ret = inet_pton(AF_INET, daddr, ¶ms->ipv4_dst); 96 if (!ASSERT_EQ(ret, 1, "convert IP[46] address")) 97 return -1; 98 params->family = AF_INET; 99 ret = inet_pton(AF_INET, IPV4_IFACE_ADDR, ¶ms->ipv4_src); 100 if (!ASSERT_EQ(ret, 1, "inet_pton(IPV4_IFACE_ADDR)")) 101 return -1; 102 103 return 0; 104 } 105 106 static void mac_str(char *b, const __u8 *mac) 107 { 108 sprintf(b, "%02X:%02X:%02X:%02X:%02X:%02X", 109 mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]); 110 } 111 112 void test_fib_lookup(void) 113 { 114 struct bpf_fib_lookup *fib_params; 115 struct nstoken *nstoken = NULL; 116 struct __sk_buff skb = { }; 117 struct fib_lookup *skel; 118 int prog_fd, err, ret, i; 119 120 /* The test does not use the skb->data, so 121 * use pkt_v6 for both v6 and v4 test. 122 */ 123 LIBBPF_OPTS(bpf_test_run_opts, run_opts, 124 .data_in = &pkt_v6, 125 .data_size_in = sizeof(pkt_v6), 126 .ctx_in = &skb, 127 .ctx_size_in = sizeof(skb), 128 ); 129 130 skel = fib_lookup__open_and_load(); 131 if (!ASSERT_OK_PTR(skel, "skel open_and_load")) 132 return; 133 prog_fd = bpf_program__fd(skel->progs.fib_lookup); 134 135 SYS(fail, "ip netns add %s", NS_TEST); 136 137 nstoken = open_netns(NS_TEST); 138 if (!ASSERT_OK_PTR(nstoken, "open_netns")) 139 goto fail; 140 141 if (setup_netns()) 142 goto fail; 143 144 ifindex = if_nametoindex("veth1"); 145 skb.ifindex = ifindex; 146 fib_params = &skel->bss->fib_params; 147 148 for (i = 0; i < ARRAY_SIZE(tests); i++) { 149 printf("Testing %s\n", tests[i].desc); 150 151 if (set_lookup_params(fib_params, tests[i].daddr)) 152 continue; 153 skel->bss->fib_lookup_ret = -1; 154 skel->bss->lookup_flags = BPF_FIB_LOOKUP_OUTPUT | 155 tests[i].lookup_flags; 156 157 err = bpf_prog_test_run_opts(prog_fd, &run_opts); 158 if (!ASSERT_OK(err, "bpf_prog_test_run_opts")) 159 continue; 160 161 ASSERT_EQ(tests[i].expected_ret, skel->bss->fib_lookup_ret, 162 "fib_lookup_ret"); 163 164 ret = memcmp(tests[i].dmac, fib_params->dmac, sizeof(tests[i].dmac)); 165 if (!ASSERT_EQ(ret, 0, "dmac not match")) { 166 char expected[18], actual[18]; 167 168 mac_str(expected, tests[i].dmac); 169 mac_str(actual, fib_params->dmac); 170 printf("dmac expected %s actual %s\n", expected, actual); 171 } 172 } 173 174 fail: 175 if (nstoken) 176 close_netns(nstoken); 177 SYS_NOFAIL("ip netns del " NS_TEST " &> /dev/null"); 178 fib_lookup__destroy(skel); 179 } 180