97 lines · c
1// SPDX-License-Identifier: GPL-2.02/* 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 8int my_pid;9 10int usdt0_called;11u64 usdt0_cookie;12int usdt0_arg_cnt;13int usdt0_arg_ret;14 15SEC("usdt")16int 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 32int usdt3_called;33u64 usdt3_cookie;34int usdt3_arg_cnt;35int usdt3_arg_rets[3];36u64 usdt3_args[3];37 38SEC("usdt//proc/self/exe:test:usdt3")39int 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 63int usdt12_called;64u64 usdt12_cookie;65int usdt12_arg_cnt;66u64 usdt12_args[12];67 68SEC("usdt//proc/self/exe:test:usdt12")69int 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 96char _license[] SEC("license") = "GPL";97