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 { "pyperf_global.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 52 53 /* full unroll by llvm */ 54 { "pyperf50.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 55 { "pyperf100.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 56 { "pyperf180.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 57 58 /* partial unroll. llvm will unroll loop ~150 times. 59 * C loop count -> 600. 60 * Asm loop count -> 4. 61 * 16k insns in loop body. 62 * Total of 5 such loops. Total program size ~82k insns. 63 */ 64 { "pyperf600.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 65 66 /* no unroll at all. 67 * C loop count -> 600. 68 * ASM loop count -> 600. 69 * ~110 insns in loop body. 70 * Total of 5 such loops. Total program size ~1500 insns. 71 */ 72 { "pyperf600_nounroll.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 73 74 { "loop1.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 75 { "loop2.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 76 { "loop4.o", BPF_PROG_TYPE_SCHED_CLS }, 77 { "loop5.o", BPF_PROG_TYPE_SCHED_CLS }, 78 79 /* partial unroll. 19k insn in a loop. 80 * Total program size 20.8k insn. 81 * ~350k processed_insns 82 */ 83 { "strobemeta.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 84 85 /* no unroll, tiny loops */ 86 { "strobemeta_nounroll1.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 87 { "strobemeta_nounroll2.o", BPF_PROG_TYPE_RAW_TRACEPOINT }, 88 89 { "test_sysctl_loop1.o", BPF_PROG_TYPE_CGROUP_SYSCTL }, 90 { "test_sysctl_loop2.o", BPF_PROG_TYPE_CGROUP_SYSCTL }, 91 92 { "test_xdp_loop.o", BPF_PROG_TYPE_XDP }, 93 { "test_seg6_loop.o", BPF_PROG_TYPE_LWT_SEG6LOCAL }, 94 }; 95 libbpf_print_fn_t old_print_fn = NULL; 96 int err, i; 97 98 if (env.verifier_stats) { 99 test__force_log(); 100 old_print_fn = libbpf_set_print(libbpf_debug_print); 101 } 102 103 for (i = 0; i < ARRAY_SIZE(tests); i++) { 104 const struct scale_test_def *test = &tests[i]; 105 106 if (!test__start_subtest(test->file)) 107 continue; 108 109 err = check_load(test->file, test->attach_type); 110 CHECK_FAIL(err && !test->fails); 111 } 112 113 if (env.verifier_stats) 114 libbpf_set_print(old_print_fn); 115 } 116