1 | // SPDX-License-Identifier: GPL-2.0 |
2 | /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */ |
3 | |
4 | #include "vmlinux.h" |
5 | #include <bpf/bpf_helpers.h> |
6 | #include <bpf/usdt.bpf.h> |
7 | |
8 | int my_pid; |
9 | |
10 | int usdt0_called; |
11 | u64 usdt0_cookie; |
12 | int usdt0_arg_cnt; |
13 | int usdt0_arg_ret; |
14 | |
15 | SEC("usdt" ) |
16 | int usdt0(struct pt_regs *ctx) |
17 | { |
18 | long tmp; |
19 | |
20 | if (my_pid != (bpf_get_current_pid_tgid() >> 32)) |
21 | return 0; |
22 | |
23 | __sync_fetch_and_add(&usdt0_called, 1); |
24 | |
25 | usdt0_cookie = bpf_usdt_cookie(ctx); |
26 | usdt0_arg_cnt = bpf_usdt_arg_cnt(ctx); |
27 | /* should return -ENOENT for any arg_num */ |
28 | usdt0_arg_ret = bpf_usdt_arg(ctx, bpf_get_prandom_u32(), &tmp); |
29 | return 0; |
30 | } |
31 | |
32 | int usdt3_called; |
33 | u64 usdt3_cookie; |
34 | int usdt3_arg_cnt; |
35 | int usdt3_arg_rets[3]; |
36 | u64 usdt3_args[3]; |
37 | |
38 | SEC("usdt//proc/self/exe:test:usdt3" ) |
39 | int usdt3(struct pt_regs *ctx) |
40 | { |
41 | long tmp; |
42 | |
43 | if (my_pid != (bpf_get_current_pid_tgid() >> 32)) |
44 | return 0; |
45 | |
46 | __sync_fetch_and_add(&usdt3_called, 1); |
47 | |
48 | usdt3_cookie = bpf_usdt_cookie(ctx); |
49 | usdt3_arg_cnt = bpf_usdt_arg_cnt(ctx); |
50 | |
51 | usdt3_arg_rets[0] = bpf_usdt_arg(ctx, 0, &tmp); |
52 | usdt3_args[0] = (int)tmp; |
53 | |
54 | usdt3_arg_rets[1] = bpf_usdt_arg(ctx, 1, &tmp); |
55 | usdt3_args[1] = (long)tmp; |
56 | |
57 | usdt3_arg_rets[2] = bpf_usdt_arg(ctx, 2, &tmp); |
58 | usdt3_args[2] = (uintptr_t)tmp; |
59 | |
60 | return 0; |
61 | } |
62 | |
63 | int usdt12_called; |
64 | u64 usdt12_cookie; |
65 | int usdt12_arg_cnt; |
66 | u64 usdt12_args[12]; |
67 | |
68 | SEC("usdt//proc/self/exe:test:usdt12" ) |
69 | int BPF_USDT(usdt12, int a1, int a2, long a3, long a4, unsigned a5, |
70 | long a6, __u64 a7, uintptr_t a8, int a9, short a10, |
71 | short a11, signed char a12) |
72 | { |
73 | if (my_pid != (bpf_get_current_pid_tgid() >> 32)) |
74 | return 0; |
75 | |
76 | __sync_fetch_and_add(&usdt12_called, 1); |
77 | |
78 | usdt12_cookie = bpf_usdt_cookie(ctx); |
79 | usdt12_arg_cnt = bpf_usdt_arg_cnt(ctx); |
80 | |
81 | usdt12_args[0] = a1; |
82 | usdt12_args[1] = a2; |
83 | usdt12_args[2] = a3; |
84 | usdt12_args[3] = a4; |
85 | usdt12_args[4] = a5; |
86 | usdt12_args[5] = a6; |
87 | usdt12_args[6] = a7; |
88 | usdt12_args[7] = a8; |
89 | usdt12_args[8] = a9; |
90 | usdt12_args[9] = a10; |
91 | usdt12_args[10] = a11; |
92 | usdt12_args[11] = a12; |
93 | return 0; |
94 | } |
95 | |
96 | char _license[] SEC("license" ) = "GPL" ; |
97 | |