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