tools/testing/selftests/bpf/progs/test_urandom_usdt.c

   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")
  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")
  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")
  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")
  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";