1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// SPDX-License-Identifier: GPL-2.0
#include <test_progs.h>
#include "fentry_kprobe_shared.skel.h"
static void trigger(struct fentry_kprobe_shared *skel)
{
LIBBPF_OPTS(bpf_test_run_opts, topts);
int err;
err = bpf_prog_test_run_opts(bpf_program__fd(skel->progs.fentry1), &topts);
ASSERT_OK(err, "test_run");
}
/*
* With a kprobe next to the trampoline, the ftrace site of the function
* reaches the trampoline through the list of ftrace callbacks, and so does
* the site of a trampoline that is modified to add or remove a program.
* With the fexit program, the trampoline is entered by a jmp.
*/
void test_fentry_kprobe_shared(void)
{
struct fentry_kprobe_shared *skel;
skel = fentry_kprobe_shared__open_and_load();
if (!ASSERT_OK_PTR(skel, "open_and_load"))
return;
skel->bss->pid = getpid();
skel->links.fentry1 = bpf_program__attach(skel->progs.fentry1);
if (!ASSERT_OK_PTR(skel->links.fentry1, "attach_fentry1"))
goto out;
skel->links.kprobe1 = bpf_program__attach(skel->progs.kprobe1);
if (!ASSERT_OK_PTR(skel->links.kprobe1, "attach_kprobe1"))
goto out;
trigger(skel);
ASSERT_EQ(skel->bss->fentry1_hits, 1, "fentry1_hits");
ASSERT_EQ(skel->bss->kprobe_hits, 1, "kprobe_hits");
skel->links.fentry2 = bpf_program__attach(skel->progs.fentry2);
if (!ASSERT_OK_PTR(skel->links.fentry2, "attach_fentry2"))
goto out;
trigger(skel);
ASSERT_EQ(skel->bss->fentry1_hits, 2, "fentry1_hits");
ASSERT_EQ(skel->bss->fentry2_hits, 1, "fentry2_hits");
ASSERT_EQ(skel->bss->kprobe_hits, 2, "kprobe_hits");
skel->links.fexit1 = bpf_program__attach(skel->progs.fexit1);
if (!ASSERT_OK_PTR(skel->links.fexit1, "attach_fexit1"))
goto out;
trigger(skel);
ASSERT_EQ(skel->bss->fentry1_hits, 3, "fentry1_hits");
ASSERT_EQ(skel->bss->fentry2_hits, 2, "fentry2_hits");
ASSERT_EQ(skel->bss->fexit_hits, 1, "fexit_hits");
ASSERT_EQ(skel->bss->kprobe_hits, 3, "kprobe_hits");
bpf_link__destroy(skel->links.kprobe1);
skel->links.kprobe1 = NULL;
trigger(skel);
ASSERT_EQ(skel->bss->fentry1_hits, 4, "fentry1_hits");
ASSERT_EQ(skel->bss->fentry2_hits, 3, "fentry2_hits");
ASSERT_EQ(skel->bss->fexit_hits, 2, "fexit_hits");
ASSERT_EQ(skel->bss->kprobe_hits, 3, "kprobe_hits");
out:
fentry_kprobe_shared__destroy(skel);
}
|
