Skip to content

Commit 93bc28d

Browse files
ThinkerYzu1Martin KaFai Lau
ThinkerYzu1
authored and
Martin KaFai Lau
committed
selftests/bpf: Test struct_ops maps with a large number of struct_ops program.
Create and load a struct_ops map with a large number of struct_ops programs to generate trampolines taking a size over multiple pages. The map includes 40 programs. Their trampolines takes 6.6k+, more than 1.5 pages, on x86. Signed-off-by: Kui-Feng Lee <thinker.li@gmail.com> Link: https://lore.kernel.org/r/20240224223418.526631-4-thinker.li@gmail.com Signed-off-by: Martin KaFai Lau <martin.lau@kernel.org>
1 parent 187e2af commit 93bc28d

3 files changed

Lines changed: 176 additions & 0 deletions

File tree

‎tools/testing/selftests/bpf/bpf_testmod/bpf_testmod.h‎

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,50 @@ struct bpf_testmod_ops {
4343
int b;
4444
} unsupported;
4545
int data;
46+
47+
/* The following pointers are used to test the maps having multiple
48+
* pages of trampolines.
49+
*/
50+
int (*tramp_1)(int value);
51+
int (*tramp_2)(int value);
52+
int (*tramp_3)(int value);
53+
int (*tramp_4)(int value);
54+
int (*tramp_5)(int value);
55+
int (*tramp_6)(int value);
56+
int (*tramp_7)(int value);
57+
int (*tramp_8)(int value);
58+
int (*tramp_9)(int value);
59+
int (*tramp_10)(int value);
60+
int (*tramp_11)(int value);
61+
int (*tramp_12)(int value);
62+
int (*tramp_13)(int value);
63+
int (*tramp_14)(int value);
64+
int (*tramp_15)(int value);
65+
int (*tramp_16)(int value);
66+
int (*tramp_17)(int value);
67+
int (*tramp_18)(int value);
68+
int (*tramp_19)(int value);
69+
int (*tramp_20)(int value);
70+
int (*tramp_21)(int value);
71+
int (*tramp_22)(int value);
72+
int (*tramp_23)(int value);
73+
int (*tramp_24)(int value);
74+
int (*tramp_25)(int value);
75+
int (*tramp_26)(int value);
76+
int (*tramp_27)(int value);
77+
int (*tramp_28)(int value);
78+
int (*tramp_29)(int value);
79+
int (*tramp_30)(int value);
80+
int (*tramp_31)(int value);
81+
int (*tramp_32)(int value);
82+
int (*tramp_33)(int value);
83+
int (*tramp_34)(int value);
84+
int (*tramp_35)(int value);
85+
int (*tramp_36)(int value);
86+
int (*tramp_37)(int value);
87+
int (*tramp_38)(int value);
88+
int (*tramp_39)(int value);
89+
int (*tramp_40)(int value);
4690
};
4791

4892
#endif /* _BPF_TESTMOD_H */

‎tools/testing/selftests/bpf/prog_tests/test_struct_ops_multi_pages.c‎

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
// SPDX-License-Identifier: GPL-2.0
2+
/* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */
3+
#include <test_progs.h>
4+
5+
#include "struct_ops_multi_pages.skel.h"
6+
7+
static void do_struct_ops_multi_pages(void)
8+
{
9+
struct struct_ops_multi_pages *skel;
10+
struct bpf_link *link;
11+
12+
/* The size of all trampolines of skel->maps.multi_pages should be
13+
* over 1 page (at least for x86).
14+
*/
15+
skel = struct_ops_multi_pages__open_and_load();
16+
if (!ASSERT_OK_PTR(skel, "struct_ops_multi_pages_open_and_load"))
17+
return;
18+
19+
link = bpf_map__attach_struct_ops(skel->maps.multi_pages);
20+
ASSERT_OK_PTR(link, "attach_multi_pages");
21+
22+
bpf_link__destroy(link);
23+
struct_ops_multi_pages__destroy(skel);
24+
}
25+
26+
void test_struct_ops_multi_pages(void)
27+
{
28+
if (test__start_subtest("multi_pages"))
29+
do_struct_ops_multi_pages();
30+
}

‎tools/testing/selftests/bpf/progs/struct_ops_multi_pages.c‎

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,102 @@
1+
// SPDX-License-Identifier: GPL-2.0
2+
/* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */
3+
#include <vmlinux.h>
4+
#include <bpf/bpf_helpers.h>
5+
#include <bpf/bpf_tracing.h>
6+
#include "../bpf_testmod/bpf_testmod.h"
7+
8+
char _license[] SEC("license") = "GPL";
9+
10+
#define TRAMP(x) \
11+
SEC("struct_ops/tramp_" #x) \
12+
int BPF_PROG(tramp_ ## x, int a) \
13+
{ \
14+
return a; \
15+
}
16+
17+
TRAMP(1)
18+
TRAMP(2)
19+
TRAMP(3)
20+
TRAMP(4)
21+
TRAMP(5)
22+
TRAMP(6)
23+
TRAMP(7)
24+
TRAMP(8)
25+
TRAMP(9)
26+
TRAMP(10)
27+
TRAMP(11)
28+
TRAMP(12)
29+
TRAMP(13)
30+
TRAMP(14)
31+
TRAMP(15)
32+
TRAMP(16)
33+
TRAMP(17)
34+
TRAMP(18)
35+
TRAMP(19)
36+
TRAMP(20)
37+
TRAMP(21)
38+
TRAMP(22)
39+
TRAMP(23)
40+
TRAMP(24)
41+
TRAMP(25)
42+
TRAMP(26)
43+
TRAMP(27)
44+
TRAMP(28)
45+
TRAMP(29)
46+
TRAMP(30)
47+
TRAMP(31)
48+
TRAMP(32)
49+
TRAMP(33)
50+
TRAMP(34)
51+
TRAMP(35)
52+
TRAMP(36)
53+
TRAMP(37)
54+
TRAMP(38)
55+
TRAMP(39)
56+
TRAMP(40)
57+
58+
#define F_TRAMP(x) .tramp_ ## x = (void *)tramp_ ## x
59+
60+
SEC(".struct_ops.link")
61+
struct bpf_testmod_ops multi_pages = {
62+
F_TRAMP(1),
63+
F_TRAMP(2),
64+
F_TRAMP(3),
65+
F_TRAMP(4),
66+
F_TRAMP(5),
67+
F_TRAMP(6),
68+
F_TRAMP(7),
69+
F_TRAMP(8),
70+
F_TRAMP(9),
71+
F_TRAMP(10),
72+
F_TRAMP(11),
73+
F_TRAMP(12),
74+
F_TRAMP(13),
75+
F_TRAMP(14),
76+
F_TRAMP(15),
77+
F_TRAMP(16),
78+
F_TRAMP(17),
79+
F_TRAMP(18),
80+
F_TRAMP(19),
81+
F_TRAMP(20),
82+
F_TRAMP(21),
83+
F_TRAMP(22),
84+
F_TRAMP(23),
85+
F_TRAMP(24),
86+
F_TRAMP(25),
87+
F_TRAMP(26),
88+
F_TRAMP(27),
89+
F_TRAMP(28),
90+
F_TRAMP(29),
91+
F_TRAMP(30),
92+
F_TRAMP(31),
93+
F_TRAMP(32),
94+
F_TRAMP(33),
95+
F_TRAMP(34),
96+
F_TRAMP(35),
97+
F_TRAMP(36),
98+
F_TRAMP(37),
99+
F_TRAMP(38),
100+
F_TRAMP(39),
101+
F_TRAMP(40),
102+
};

0 commit comments

Comments
 (0)