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