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 urand_pid;
9
10 int urand_read_without_sema_call_cnt;
11 int urand_read_without_sema_buf_sz_sum;
12
13 SEC("usdt/./urandom_read:urand:read_without_sema")
BPF_USDT(urand_read_without_sema,int iter_num,int iter_cnt,int buf_sz)14 int BPF_USDT(urand_read_without_sema, int iter_num, int iter_cnt, int buf_sz)
15 {
16 if (urand_pid != (bpf_get_current_pid_tgid() >> 32))
17 return 0;
18
19 __sync_fetch_and_add(&urand_read_without_sema_call_cnt, 1);
20 __sync_fetch_and_add(&urand_read_without_sema_buf_sz_sum, buf_sz);
21
22 return 0;
23 }
24
25 int urand_read_with_sema_call_cnt;
26 int urand_read_with_sema_buf_sz_sum;
27
28 SEC("usdt/./urandom_read:urand:read_with_sema")
BPF_USDT(urand_read_with_sema,int iter_num,int iter_cnt,int buf_sz)29 int BPF_USDT(urand_read_with_sema, int iter_num, int iter_cnt, int buf_sz)
30 {
31 if (urand_pid != (bpf_get_current_pid_tgid() >> 32))
32 return 0;
33
34 __sync_fetch_and_add(&urand_read_with_sema_call_cnt, 1);
35 __sync_fetch_and_add(&urand_read_with_sema_buf_sz_sum, buf_sz);
36
37 return 0;
38 }
39
40 int urandlib_read_without_sema_call_cnt;
41 int urandlib_read_without_sema_buf_sz_sum;
42
43 SEC("usdt/./liburandom_read.so:urandlib:read_without_sema")
BPF_USDT(urandlib_read_without_sema,int iter_num,int iter_cnt,int buf_sz)44 int BPF_USDT(urandlib_read_without_sema, int iter_num, int iter_cnt, int buf_sz)
45 {
46 if (urand_pid != (bpf_get_current_pid_tgid() >> 32))
47 return 0;
48
49 __sync_fetch_and_add(&urandlib_read_without_sema_call_cnt, 1);
50 __sync_fetch_and_add(&urandlib_read_without_sema_buf_sz_sum, buf_sz);
51
52 return 0;
53 }
54
55 int urandlib_read_with_sema_call_cnt;
56 int urandlib_read_with_sema_buf_sz_sum;
57
58 SEC("usdt/./liburandom_read.so:urandlib:read_with_sema")
BPF_USDT(urandlib_read_with_sema,int iter_num,int iter_cnt,int buf_sz)59 int BPF_USDT(urandlib_read_with_sema, int iter_num, int iter_cnt, int buf_sz)
60 {
61 if (urand_pid != (bpf_get_current_pid_tgid() >> 32))
62 return 0;
63
64 __sync_fetch_and_add(&urandlib_read_with_sema_call_cnt, 1);
65 __sync_fetch_and_add(&urandlib_read_with_sema_buf_sz_sum, buf_sz);
66
67 return 0;
68 }
69
70 char _license[] SEC("license") = "GPL";
71