1 // SPDX-License-Identifier: GPL-2.0 2 #include <test_progs.h> 3 4 static void test_l4lb(const char *file) 5 { 6 unsigned int nr_cpus = bpf_num_possible_cpus(); 7 struct vip key = {.protocol = 6}; 8 struct vip_meta { 9 __u32 flags; 10 __u32 vip_num; 11 } value = {.vip_num = VIP_NUM}; 12 __u32 stats_key = VIP_NUM; 13 struct vip_stats { 14 __u64 bytes; 15 __u64 pkts; 16 } stats[nr_cpus]; 17 struct real_definition { 18 union { 19 __be32 dst; 20 __be32 dstv6[4]; 21 }; 22 __u8 flags; 23 } real_def = {.dst = MAGIC_VAL}; 24 __u32 ch_key = 11, real_num = 3; 25 __u32 duration, retval, size; 26 int err, i, prog_fd, map_fd; 27 __u64 bytes = 0, pkts = 0; 28 struct bpf_object *obj; 29 char buf[128]; 30 u32 *magic = (u32 *)buf; 31 32 err = bpf_prog_load(file, BPF_PROG_TYPE_SCHED_CLS, &obj, &prog_fd); 33 if (CHECK_FAIL(err)) 34 return; 35 36 map_fd = bpf_find_map(__func__, obj, "vip_map"); 37 if (map_fd < 0) 38 goto out; 39 bpf_map_update_elem(map_fd, &key, &value, 0); 40 41 map_fd = bpf_find_map(__func__, obj, "ch_rings"); 42 if (map_fd < 0) 43 goto out; 44 bpf_map_update_elem(map_fd, &ch_key, &real_num, 0); 45 46 map_fd = bpf_find_map(__func__, obj, "reals"); 47 if (map_fd < 0) 48 goto out; 49 bpf_map_update_elem(map_fd, &real_num, &real_def, 0); 50 51 err = bpf_prog_test_run(prog_fd, NUM_ITER, &pkt_v4, sizeof(pkt_v4), 52 buf, &size, &retval, &duration); 53 CHECK(err || retval != 7/*TC_ACT_REDIRECT*/ || size != 54 || 54 *magic != MAGIC_VAL, "ipv4", 55 "err %d errno %d retval %d size %d magic %x\n", 56 err, errno, retval, size, *magic); 57 58 err = bpf_prog_test_run(prog_fd, NUM_ITER, &pkt_v6, sizeof(pkt_v6), 59 buf, &size, &retval, &duration); 60 CHECK(err || retval != 7/*TC_ACT_REDIRECT*/ || size != 74 || 61 *magic != MAGIC_VAL, "ipv6", 62 "err %d errno %d retval %d size %d magic %x\n", 63 err, errno, retval, size, *magic); 64 65 map_fd = bpf_find_map(__func__, obj, "stats"); 66 if (map_fd < 0) 67 goto out; 68 bpf_map_lookup_elem(map_fd, &stats_key, stats); 69 for (i = 0; i < nr_cpus; i++) { 70 bytes += stats[i].bytes; 71 pkts += stats[i].pkts; 72 } 73 if (CHECK_FAIL(bytes != MAGIC_BYTES * NUM_ITER * 2 || 74 pkts != NUM_ITER * 2)) 75 printf("test_l4lb:FAIL:stats %lld %lld\n", bytes, pkts); 76 out: 77 bpf_object__close(obj); 78 } 79 80 void test_l4lb_all(void) 81 { 82 const char *file1 = "./test_l4lb.o"; 83 const char *file2 = "./test_l4lb_noinline.o"; 84 85 test_l4lb(file1); 86 test_l4lb(file2); 87 } 88