1*90a039fdSAlan Maguire // SPDX-License-Identifier: GPL-2.0
2*90a039fdSAlan Maguire /* Copyright (c) 2022, Oracle and/or its affiliates. */
3*90a039fdSAlan Maguire
4*90a039fdSAlan Maguire #include "vmlinux.h"
5*90a039fdSAlan Maguire
6*90a039fdSAlan Maguire #include <bpf/bpf_helpers.h>
7*90a039fdSAlan Maguire #include <bpf/bpf_tracing.h>
8*90a039fdSAlan Maguire #include "bpf_misc.h"
9*90a039fdSAlan Maguire
10*90a039fdSAlan Maguire __u32 perfbuf_val = 0;
11*90a039fdSAlan Maguire __u32 ringbuf_val = 0;
12*90a039fdSAlan Maguire
13*90a039fdSAlan Maguire int test_pid;
14*90a039fdSAlan Maguire
15*90a039fdSAlan Maguire struct {
16*90a039fdSAlan Maguire __uint(type, BPF_MAP_TYPE_ARRAY);
17*90a039fdSAlan Maguire __uint(max_entries, 1);
18*90a039fdSAlan Maguire __type(key, __u32);
19*90a039fdSAlan Maguire __type(value, __u32);
20*90a039fdSAlan Maguire } array SEC(".maps");
21*90a039fdSAlan Maguire
22*90a039fdSAlan Maguire struct {
23*90a039fdSAlan Maguire __uint(type, BPF_MAP_TYPE_PERCPU_ARRAY);
24*90a039fdSAlan Maguire __uint(max_entries, 1);
25*90a039fdSAlan Maguire __type(key, __u32);
26*90a039fdSAlan Maguire __type(value, __u32);
27*90a039fdSAlan Maguire } percpu_array SEC(".maps");
28*90a039fdSAlan Maguire
29*90a039fdSAlan Maguire struct {
30*90a039fdSAlan Maguire __uint(type, BPF_MAP_TYPE_HASH);
31*90a039fdSAlan Maguire __uint(max_entries, 1);
32*90a039fdSAlan Maguire __type(key, __u32);
33*90a039fdSAlan Maguire __type(value, __u32);
34*90a039fdSAlan Maguire } hash SEC(".maps");
35*90a039fdSAlan Maguire
36*90a039fdSAlan Maguire struct {
37*90a039fdSAlan Maguire __uint(type, BPF_MAP_TYPE_PERCPU_HASH);
38*90a039fdSAlan Maguire __uint(max_entries, 1);
39*90a039fdSAlan Maguire __type(key, __u32);
40*90a039fdSAlan Maguire __type(value, __u32);
41*90a039fdSAlan Maguire } percpu_hash SEC(".maps");
42*90a039fdSAlan Maguire
43*90a039fdSAlan Maguire struct {
44*90a039fdSAlan Maguire __uint(type, BPF_MAP_TYPE_PERF_EVENT_ARRAY);
45*90a039fdSAlan Maguire __type(key, __u32);
46*90a039fdSAlan Maguire __type(value, __u32);
47*90a039fdSAlan Maguire } perfbuf SEC(".maps");
48*90a039fdSAlan Maguire
49*90a039fdSAlan Maguire struct {
50*90a039fdSAlan Maguire __uint(type, BPF_MAP_TYPE_RINGBUF);
51*90a039fdSAlan Maguire __uint(max_entries, 1 << 12);
52*90a039fdSAlan Maguire } ringbuf SEC(".maps");
53*90a039fdSAlan Maguire
54*90a039fdSAlan Maguire struct {
55*90a039fdSAlan Maguire __uint(type, BPF_MAP_TYPE_PROG_ARRAY);
56*90a039fdSAlan Maguire __uint(max_entries, 1);
57*90a039fdSAlan Maguire __uint(key_size, sizeof(__u32));
58*90a039fdSAlan Maguire __uint(value_size, sizeof(__u32));
59*90a039fdSAlan Maguire } prog_array SEC(".maps");
60*90a039fdSAlan Maguire
61*90a039fdSAlan Maguire SEC("fentry/" SYS_PREFIX "sys_nanosleep")
sys_nanosleep_enter(void * ctx)62*90a039fdSAlan Maguire int sys_nanosleep_enter(void *ctx)
63*90a039fdSAlan Maguire {
64*90a039fdSAlan Maguire int cur_pid;
65*90a039fdSAlan Maguire
66*90a039fdSAlan Maguire cur_pid = bpf_get_current_pid_tgid() >> 32;
67*90a039fdSAlan Maguire
68*90a039fdSAlan Maguire if (cur_pid != test_pid)
69*90a039fdSAlan Maguire return 0;
70*90a039fdSAlan Maguire
71*90a039fdSAlan Maguire bpf_perf_event_output(ctx, &perfbuf, BPF_F_CURRENT_CPU, &perfbuf_val, sizeof(perfbuf_val));
72*90a039fdSAlan Maguire bpf_ringbuf_output(&ringbuf, &ringbuf_val, sizeof(ringbuf_val), 0);
73*90a039fdSAlan Maguire
74*90a039fdSAlan Maguire return 0;
75*90a039fdSAlan Maguire }
76*90a039fdSAlan Maguire
77*90a039fdSAlan Maguire SEC("perf_event")
handle_perf_event(void * ctx)78*90a039fdSAlan Maguire int handle_perf_event(void *ctx)
79*90a039fdSAlan Maguire {
80*90a039fdSAlan Maguire return 0;
81*90a039fdSAlan Maguire }
82*90a039fdSAlan Maguire
83*90a039fdSAlan Maguire char _license[] SEC("license") = "GPL";
84