1 // SPDX-License-Identifier: GPL-2.0 2 // Copyright (c) 2019 Facebook 3 #include <test_progs.h> 4 static int libbpf_debug_print(enum libbpf_print_level level, 5 const char *format, va_list args) 6 { 7 if (level != LIBBPF_DEBUG) { 8 vprintf(format, args); 9 return 0; 10 } 11 12 if (!strstr(format, "verifier log")) 13 return 0; 14 vprintf("%s", args); 15 return 0; 16 } 17 18 extern int extra_prog_load_log_flags; 19 20 static int check_load(const char *file, enum bpf_prog_type type) 21 { 22 struct bpf_prog_load_attr attr; 23 struct bpf_object *obj = NULL; 24 int err, prog_fd; 25 26 memset(&attr, 0, sizeof(struct bpf_prog_load_attr)); 27 attr.file = file; 28 attr.prog_type = type; 29 attr.log_level = 4 | extra_prog_load_log_flags; 30 attr.prog_flags = BPF_F_TEST_RND_HI32; 31 err = bpf_prog_load_xattr(&attr, &obj, &prog_fd); 32 bpf_object__close(obj); 33 return err; 34 } 35 36 struct scale_test_def { 37 const char *file; 38 enum bpf_prog_type attach_type; 39 bool fails; 40 }; 41 42 void test_bpf_verif_scale(void) 43 { 44 struct scale_test_def tests[] = { 45 { "loop3.o", BPF_PROG_TYPE_RAW_TRACEPOINT, true /* fails */ }, 46 47 { "test_verif_scale1.o", BPF_PROG_TYPE_SCHED_CLS }, 48 { "test_verif_scale2.o", BPF_PROG_TYPE_SCHED_CLS }, 49 { "test_verif_scale3.o", BPF_PROG_TYPE_SCHED_CLS }, 50 51 /* full unroll by llvm */ 52 { "pyperf50.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 53 { "pyperf100.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 54 { "pyperf180.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 55 56 /* partial unroll. llvm will unroll loop ~150 times. 57 * C loop count -> 600. 58 * Asm loop count -> 4. 59 * 16k insns in loop body. 60 * Total of 5 such loops. Total program size ~82k insns. 61 */ 62 { "pyperf600.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 63 64 /* no unroll at all. 65 * C loop count -> 600. 66 * ASM loop count -> 600. 67 * ~110 insns in loop body. 68 * Total of 5 such loops. Total program size ~1500 insns. 69 */ 70 { "pyperf600_nounroll.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 71 72 { "loop1.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 73 { "loop2.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 74 { "loop4.o", BPF_PROG_TYPE_SCHED_CLS }, 75 { "loop5.o", BPF_PROG_TYPE_SCHED_CLS }, 76 77 /* partial unroll. 19k insn in a loop. 78 * Total program size 20.8k insn. 79 * ~350k processed_insns 80 */ 81 { "strobemeta.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 82 83 /* no unroll, tiny loops */ 84 { "strobemeta_nounroll1.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 85 { "strobemeta_nounroll2.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 86 87 { "test_sysctl_loop1.o", BPF_PROG_TYPE_CGROUP_SYSCTL }, 88 { "test_sysctl_loop2.o", BPF_PROG_TYPE_CGROUP_SYSCTL }, 89 90 { "test_xdp_loop.o", BPF_PROG_TYPE_XDP }, 91 { "test_seg6_loop.o", BPF_PROG_TYPE_LWT_SEG6LOCAL }, 92 }; 93 libbpf_print_fn_t old_print_fn = NULL; 94 int err, i; 95 96 if (env.verifier_stats) { 97 test__force_log(); 98 old_print_fn = libbpf_set_print(libbpf_debug_print); 99 } 100 101 for (i = 0; i < ARRAY_SIZE(tests); i++) { 102 const struct scale_test_def *test = &tests[i]; 103 104 if (!test__start_subtest(test->file)) 105 continue; 106 107 err = check_load(test->file, test->attach_type); 108 CHECK_FAIL(err && !test->fails); 109 } 110 111 if (env.verifier_stats) 112 libbpf_set_print(old_print_fn); 113 } 114