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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2020 Facebook */
#include <test_progs.h>
#include <time.h>
#include "test_vmlinux.skel.h"
#define MY_TV_NSEC 1337
static void nsleep()
{
struct timespec ts = { .tv_nsec = MY_TV_NSEC };
(void)syscall(__NR_nanosleep, &ts, NULL);
}
static const char *hrtimer_func = "hrtimer_start_range_ns";
static int setup_hrtimer_progs(struct test_vmlinux *skel)
{
int err;
if (libbpf_find_vmlinux_btf_id("hrtimer_start_range_ns_user", BPF_TRACE_FENTRY) > 0)
hrtimer_func = "hrtimer_start_range_ns_user";
err = bpf_program__set_attach_target(skel->progs.handle__fentry, 0, hrtimer_func);
if (err)
return err;
/*
* Bare SEC("kprobe") has no target function, so attach it manually
* later after selecting the hrtimer function to probe.
*/
bpf_program__set_autoattach(skel->progs.handle__kprobe, false);
return 0;
}
void test_vmlinux(void)
{
int err;
struct test_vmlinux* skel;
struct test_vmlinux__bss *bss;
struct bpf_link *kprobe_link = NULL;
skel = test_vmlinux__open();
if (!ASSERT_OK_PTR(skel, "test_vmlinux__open"))
return;
err = setup_hrtimer_progs(skel);
if (!ASSERT_OK(err, "setup_hrtimer_progs"))
goto cleanup;
err = test_vmlinux__load(skel);
if (!ASSERT_OK(err, "test_vmlinux__load"))
goto cleanup;
bss = skel->bss;
err = test_vmlinux__attach(skel);
if (!ASSERT_OK(err, "test_vmlinux__attach"))
goto cleanup;
/* manually attach kprobe with the selected function */
if (hrtimer_func) {
kprobe_link = bpf_program__attach_kprobe(skel->progs.handle__kprobe,
false /* retprobe */, hrtimer_func);
if (!ASSERT_OK_PTR(kprobe_link, "bpf_program__attach_kprobe"))
goto cleanup;
}
/* trigger everything */
nsleep();
ASSERT_TRUE(bss->tp_called, "tp");
ASSERT_TRUE(bss->raw_tp_called, "raw_tp");
ASSERT_TRUE(bss->tp_btf_called, "tp_btf");
ASSERT_TRUE(bss->kprobe_called, "kprobe");
ASSERT_TRUE(bss->fentry_called, "fentry");
cleanup:
bpf_link__destroy(kprobe_link);
test_vmlinux__destroy(skel);
}
|
