tools/testing/selftests/bpf/prog_tests/bpf_verif_scale.c

   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 }