mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2026-08-30 02:59:05 -04:00
304841967c
Adding test that attaches bpf program on usdt probe in 2 scenarios; - attach program on top of usdt_1, which is single nop instruction, so the probe stays on nop instruction and is not optimized. - attach program on top of usdt_2 which is probe defined on top of nop,nop5 combo, so the probe is placed on top of nop5 and is optimized. Signed-off-by: Jiri Olsa <jolsa@kernel.org> Link: https://lore.kernel.org/r/20260224103915.1369690-5-jolsa@kernel.org Signed-off-by: Alexei Starovoitov <ast@kernel.org>
154 lines
3.2 KiB
C
154 lines
3.2 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
|
|
|
|
#include "vmlinux.h"
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/usdt.bpf.h>
|
|
|
|
int my_pid;
|
|
|
|
int usdt0_called;
|
|
u64 usdt0_cookie;
|
|
int usdt0_arg_cnt;
|
|
int usdt0_arg_ret;
|
|
int usdt0_arg_size;
|
|
|
|
SEC("usdt")
|
|
int usdt0(struct pt_regs *ctx)
|
|
{
|
|
long tmp;
|
|
|
|
if (my_pid != (bpf_get_current_pid_tgid() >> 32))
|
|
return 0;
|
|
|
|
__sync_fetch_and_add(&usdt0_called, 1);
|
|
|
|
usdt0_cookie = bpf_usdt_cookie(ctx);
|
|
usdt0_arg_cnt = bpf_usdt_arg_cnt(ctx);
|
|
/* should return -ENOENT for any arg_num */
|
|
usdt0_arg_ret = bpf_usdt_arg(ctx, bpf_get_prandom_u32(), &tmp);
|
|
usdt0_arg_size = bpf_usdt_arg_size(ctx, bpf_get_prandom_u32());
|
|
return 0;
|
|
}
|
|
|
|
int usdt3_called;
|
|
u64 usdt3_cookie;
|
|
int usdt3_arg_cnt;
|
|
int usdt3_arg_rets[3];
|
|
u64 usdt3_args[3];
|
|
int usdt3_arg_sizes[3];
|
|
|
|
SEC("usdt//proc/self/exe:test:usdt3")
|
|
int usdt3(struct pt_regs *ctx)
|
|
{
|
|
long tmp;
|
|
|
|
if (my_pid != (bpf_get_current_pid_tgid() >> 32))
|
|
return 0;
|
|
|
|
__sync_fetch_and_add(&usdt3_called, 1);
|
|
|
|
usdt3_cookie = bpf_usdt_cookie(ctx);
|
|
usdt3_arg_cnt = bpf_usdt_arg_cnt(ctx);
|
|
|
|
usdt3_arg_rets[0] = bpf_usdt_arg(ctx, 0, &tmp);
|
|
usdt3_args[0] = (int)tmp;
|
|
usdt3_arg_sizes[0] = bpf_usdt_arg_size(ctx, 0);
|
|
|
|
usdt3_arg_rets[1] = bpf_usdt_arg(ctx, 1, &tmp);
|
|
usdt3_args[1] = (long)tmp;
|
|
usdt3_arg_sizes[1] = bpf_usdt_arg_size(ctx, 1);
|
|
|
|
usdt3_arg_rets[2] = bpf_usdt_arg(ctx, 2, &tmp);
|
|
usdt3_args[2] = (uintptr_t)tmp;
|
|
usdt3_arg_sizes[2] = bpf_usdt_arg_size(ctx, 2);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int usdt12_called;
|
|
u64 usdt12_cookie;
|
|
int usdt12_arg_cnt;
|
|
u64 usdt12_args[12];
|
|
int usdt12_arg_sizes[12];
|
|
|
|
SEC("usdt//proc/self/exe:test:usdt12")
|
|
int BPF_USDT(usdt12, int a1, int a2, long a3, long a4, unsigned a5,
|
|
long a6, __u64 a7, uintptr_t a8, int a9, short a10,
|
|
short a11, signed char a12)
|
|
{
|
|
int i;
|
|
|
|
if (my_pid != (bpf_get_current_pid_tgid() >> 32))
|
|
return 0;
|
|
|
|
__sync_fetch_and_add(&usdt12_called, 1);
|
|
|
|
usdt12_cookie = bpf_usdt_cookie(ctx);
|
|
usdt12_arg_cnt = bpf_usdt_arg_cnt(ctx);
|
|
|
|
usdt12_args[0] = a1;
|
|
usdt12_args[1] = a2;
|
|
usdt12_args[2] = a3;
|
|
usdt12_args[3] = a4;
|
|
usdt12_args[4] = a5;
|
|
usdt12_args[5] = a6;
|
|
usdt12_args[6] = a7;
|
|
usdt12_args[7] = a8;
|
|
usdt12_args[8] = a9;
|
|
usdt12_args[9] = a10;
|
|
usdt12_args[10] = a11;
|
|
usdt12_args[11] = a12;
|
|
|
|
bpf_for(i, 0, 12) {
|
|
usdt12_arg_sizes[i] = bpf_usdt_arg_size(ctx, i);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int usdt_sib_called;
|
|
u64 usdt_sib_cookie;
|
|
int usdt_sib_arg_cnt;
|
|
int usdt_sib_arg_ret;
|
|
short usdt_sib_arg;
|
|
int usdt_sib_arg_size;
|
|
|
|
/*
|
|
* usdt_sib is only tested on x86-related architectures, so it requires
|
|
* manual attach since auto-attach will panic tests under other architectures
|
|
*/
|
|
SEC("usdt")
|
|
int usdt_sib(struct pt_regs *ctx)
|
|
{
|
|
long tmp;
|
|
|
|
if (my_pid != (bpf_get_current_pid_tgid() >> 32))
|
|
return 0;
|
|
|
|
__sync_fetch_and_add(&usdt_sib_called, 1);
|
|
|
|
usdt_sib_cookie = bpf_usdt_cookie(ctx);
|
|
usdt_sib_arg_cnt = bpf_usdt_arg_cnt(ctx);
|
|
|
|
usdt_sib_arg_ret = bpf_usdt_arg(ctx, 0, &tmp);
|
|
usdt_sib_arg = (short)tmp;
|
|
usdt_sib_arg_size = bpf_usdt_arg_size(ctx, 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
#ifdef __TARGET_ARCH_x86
|
|
int executed;
|
|
unsigned long expected_ip;
|
|
|
|
SEC("usdt")
|
|
int usdt_executed(struct pt_regs *ctx)
|
|
{
|
|
if (expected_ip == ctx->ip)
|
|
executed++;
|
|
return 0;
|
|
}
|
|
#endif
|
|
char _license[] SEC("license") = "GPL";
|