1e5e7a8f2SAlexei Starovoitov // SPDX-License-Identifier: GPL-2.0
2e5e7a8f2SAlexei Starovoitov // Copyright (c) 2019 Facebook
3e5e7a8f2SAlexei Starovoitov #include <test_progs.h>
4e5e7a8f2SAlexei Starovoitov static int libbpf_debug_print(enum libbpf_print_level level,
5e5e7a8f2SAlexei Starovoitov 			      const char *format, va_list args)
6e5e7a8f2SAlexei Starovoitov {
7e5e7a8f2SAlexei Starovoitov 	if (level != LIBBPF_DEBUG)
8e5e7a8f2SAlexei Starovoitov 		return 0;
9e5e7a8f2SAlexei Starovoitov 
10e5e7a8f2SAlexei Starovoitov 	if (!strstr(format, "verifier log"))
11e5e7a8f2SAlexei Starovoitov 		return 0;
12e5e7a8f2SAlexei Starovoitov 	return vfprintf(stderr, "%s", args);
13e5e7a8f2SAlexei Starovoitov }
14e5e7a8f2SAlexei Starovoitov 
157c944106SAlexei Starovoitov static int check_load(const char *file, enum bpf_prog_type type)
16e5e7a8f2SAlexei Starovoitov {
17e5e7a8f2SAlexei Starovoitov 	struct bpf_prog_load_attr attr;
1807b61991SLorenz Bauer 	struct bpf_object *obj = NULL;
19e5e7a8f2SAlexei Starovoitov 	int err, prog_fd;
20e5e7a8f2SAlexei Starovoitov 
21e5e7a8f2SAlexei Starovoitov 	memset(&attr, 0, sizeof(struct bpf_prog_load_attr));
22e5e7a8f2SAlexei Starovoitov 	attr.file = file;
237c944106SAlexei Starovoitov 	attr.prog_type = type;
24e5e7a8f2SAlexei Starovoitov 	attr.log_level = 4;
25e5e7a8f2SAlexei Starovoitov 	err = bpf_prog_load_xattr(&attr, &obj, &prog_fd);
26e5e7a8f2SAlexei Starovoitov 	bpf_object__close(obj);
27e5e7a8f2SAlexei Starovoitov 	if (err)
28e5e7a8f2SAlexei Starovoitov 		error_cnt++;
29e5e7a8f2SAlexei Starovoitov 	return err;
30e5e7a8f2SAlexei Starovoitov }
31e5e7a8f2SAlexei Starovoitov 
32e5e7a8f2SAlexei Starovoitov void test_bpf_verif_scale(void)
33e5e7a8f2SAlexei Starovoitov {
347c944106SAlexei Starovoitov 	const char *scale[] = {
357c944106SAlexei Starovoitov 		"./test_verif_scale1.o", "./test_verif_scale2.o", "./test_verif_scale3.o"
367c944106SAlexei Starovoitov 	};
377c944106SAlexei Starovoitov 	const char *pyperf[] = {
387c944106SAlexei Starovoitov 		"./pyperf50.o",	"./pyperf100.o", "./pyperf180.o"
397c944106SAlexei Starovoitov 	};
407c944106SAlexei Starovoitov 	int err, i;
41e5e7a8f2SAlexei Starovoitov 
42e5e7a8f2SAlexei Starovoitov 	if (verifier_stats)
43e5e7a8f2SAlexei Starovoitov 		libbpf_set_print(libbpf_debug_print);
44e5e7a8f2SAlexei Starovoitov 
457c944106SAlexei Starovoitov 	for (i = 0; i < ARRAY_SIZE(scale); i++) {
467c944106SAlexei Starovoitov 		err = check_load(scale[i], BPF_PROG_TYPE_SCHED_CLS);
477c944106SAlexei Starovoitov 		printf("test_scale:%s:%s\n", scale[i], err ? "FAIL" : "OK");
487c944106SAlexei Starovoitov 	}
497c944106SAlexei Starovoitov 
507c944106SAlexei Starovoitov 	for (i = 0; i < ARRAY_SIZE(pyperf); i++) {
517c944106SAlexei Starovoitov 		err = check_load(pyperf[i], BPF_PROG_TYPE_RAW_TRACEPOINT);
527c944106SAlexei Starovoitov 		printf("test_scale:%s:%s\n", pyperf[i], err ? "FAIL" : "OK");
537c944106SAlexei Starovoitov 	}
54e5e7a8f2SAlexei Starovoitov }
55