Skip to content

Commit 1206770

Browse files
olsajiriKernel Patches Daemon
authored and
Kernel Patches Daemon
committed
selftests/bpf: Add uprobe_multi test program
Adding uprobe_multi test program that defines 50k uprobe_multi_func_* functions and will serve as attach point for uprobe_multi bench test in following patch. Signed-off-by: Jiri Olsa <jolsa@kernel.org>
1 parent c2095ba commit 1206770

File tree

2 files changed

+72
-0
lines changed

2 files changed

+72
-0
lines changed

tools/testing/selftests/bpf/Makefile

+5
Original file line numberDiff line numberDiff line change
@@ -585,6 +585,7 @@ TRUNNER_EXTRA_FILES := $(OUTPUT)/urandom_read $(OUTPUT)/bpf_testmod.ko \
585585
$(OUTPUT)/liburandom_read.so \
586586
$(OUTPUT)/xdp_synproxy \
587587
$(OUTPUT)/sign-file \
588+
$(OUTPUT)/uprobe_multi \
588589
ima_setup.sh \
589590
verify_sig_setup.sh \
590591
$(wildcard progs/btf_dump_test_case_*.c) \
@@ -698,6 +699,10 @@ $(OUTPUT)/veristat: $(OUTPUT)/veristat.o
698699
$(call msg,BINARY,,$@)
699700
$(Q)$(CC) $(CFLAGS) $(LDFLAGS) $(filter %.a %.o,$^) $(LDLIBS) -o $@
700701

702+
$(OUTPUT)/uprobe_multi: uprobe_multi.c
703+
$(call msg,BINARY,,$@)
704+
$(Q)$(CC) $(CFLAGS) $(LDFLAGS) $^ $(LDLIBS) -o $@
705+
701706
EXTRA_CLEAN := $(TEST_CUSTOM_PROGS) $(SCRATCH_DIR) $(HOST_SCRATCH_DIR) \
702707
prog_tests/tests.h map_tests/tests.h verifier/tests.h \
703708
feature bpftool \
+67
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,67 @@
1+
// SPDX-License-Identifier: GPL-2.0
2+
3+
#include <stdio.h>
4+
#include <string.h>
5+
6+
#define __PASTE(a, b) a##b
7+
#define PASTE(a, b) __PASTE(a, b)
8+
9+
#define NAME(name, idx) PASTE(name, idx)
10+
11+
#define DEF(name, idx) int NAME(name, idx)(void) { return 0; }
12+
#define CALL(name, idx) NAME(name, idx)();
13+
14+
#define F(body, name, idx) body(name, idx)
15+
16+
#define F10(body, name, idx) \
17+
F(body, PASTE(name, idx), 0) F(body, PASTE(name, idx), 1) F(body, PASTE(name, idx), 2) \
18+
F(body, PASTE(name, idx), 3) F(body, PASTE(name, idx), 4) F(body, PASTE(name, idx), 5) \
19+
F(body, PASTE(name, idx), 6) F(body, PASTE(name, idx), 7) F(body, PASTE(name, idx), 8) \
20+
F(body, PASTE(name, idx), 9)
21+
22+
#define F100(body, name, idx) \
23+
F10(body, PASTE(name, idx), 0) F10(body, PASTE(name, idx), 1) F10(body, PASTE(name, idx), 2) \
24+
F10(body, PASTE(name, idx), 3) F10(body, PASTE(name, idx), 4) F10(body, PASTE(name, idx), 5) \
25+
F10(body, PASTE(name, idx), 6) F10(body, PASTE(name, idx), 7) F10(body, PASTE(name, idx), 8) \
26+
F10(body, PASTE(name, idx), 9)
27+
28+
#define F1000(body, name, idx) \
29+
F100(body, PASTE(name, idx), 0) F100(body, PASTE(name, idx), 1) F100(body, PASTE(name, idx), 2) \
30+
F100(body, PASTE(name, idx), 3) F100(body, PASTE(name, idx), 4) F100(body, PASTE(name, idx), 5) \
31+
F100(body, PASTE(name, idx), 6) F100(body, PASTE(name, idx), 7) F100(body, PASTE(name, idx), 8) \
32+
F100(body, PASTE(name, idx), 9)
33+
34+
#define F10000(body, name, idx) \
35+
F1000(body, PASTE(name, idx), 0) F1000(body, PASTE(name, idx), 1) F1000(body, PASTE(name, idx), 2) \
36+
F1000(body, PASTE(name, idx), 3) F1000(body, PASTE(name, idx), 4) F1000(body, PASTE(name, idx), 5) \
37+
F1000(body, PASTE(name, idx), 6) F1000(body, PASTE(name, idx), 7) F1000(body, PASTE(name, idx), 8) \
38+
F1000(body, PASTE(name, idx), 9)
39+
40+
F10000(DEF, uprobe_multi_func_, 0)
41+
F10000(DEF, uprobe_multi_func_, 1)
42+
F10000(DEF, uprobe_multi_func_, 2)
43+
F10000(DEF, uprobe_multi_func_, 3)
44+
F10000(DEF, uprobe_multi_func_, 4)
45+
46+
static int bench(void)
47+
{
48+
F10000(CALL, uprobe_multi_func_, 0)
49+
F10000(CALL, uprobe_multi_func_, 1)
50+
F10000(CALL, uprobe_multi_func_, 2)
51+
F10000(CALL, uprobe_multi_func_, 3)
52+
F10000(CALL, uprobe_multi_func_, 4)
53+
return 0;
54+
}
55+
56+
int main(int argc, char **argv)
57+
{
58+
if (argc != 2)
59+
goto error;
60+
61+
if (!strcmp("bench", argv[1]))
62+
return bench();
63+
64+
error:
65+
fprintf(stderr, "usage: %s <bench>\n", argv[0]);
66+
return -1;
67+
}

0 commit comments

Comments
 (0)