mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2026-08-31 08:15:07 -04:00
8a20655749
Cover all three sleepable tracepoint types (tp_btf.s, raw_tp.s, tp.s) and sys_exit (via bpf_task_pt_regs) with functional tests using bpf_copy_from_user() on getcwd. Verify alias and bare SEC variants, bpf_prog_test_run_raw_tp() with BPF_F_TEST_RUN_ON_CPU rejection, attach-time rejection on non-faultable tracepoints, and load-time rejection for sleepable tp_btf on non-faultable tracepoints. Signed-off-by: Mykyta Yatsenko <yatsenko@meta.com> Acked-by: Kumar Kartikeya Dwivedi <memxor@gmail.com> Link: https://lore.kernel.org/bpf/20260422-sleepable_tracepoints-v13-6-99005dff21ef@meta.com Signed-off-by: Kumar Kartikeya Dwivedi <memxor@gmail.com>
113 lines
2.4 KiB
C
113 lines
2.4 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
/* Copyright (c) 2025 Meta Platforms, Inc. and affiliates. */
|
|
|
|
#include <vmlinux.h>
|
|
#include <asm/unistd.h>
|
|
#include <bpf/bpf_tracing.h>
|
|
#include <bpf/bpf_core_read.h>
|
|
#include <bpf/bpf_helpers.h>
|
|
|
|
char _license[] SEC("license") = "GPL";
|
|
|
|
int target_pid;
|
|
int prog_triggered;
|
|
long err;
|
|
char copied_byte;
|
|
|
|
static int copy_getcwd_arg(char *ubuf)
|
|
{
|
|
err = bpf_copy_from_user(&copied_byte, sizeof(copied_byte), ubuf);
|
|
if (err)
|
|
return err;
|
|
|
|
prog_triggered = 1;
|
|
return 0;
|
|
}
|
|
|
|
SEC("tp_btf.s/sys_enter")
|
|
int BPF_PROG(handle_sys_enter_tp_btf, struct pt_regs *regs, long id)
|
|
{
|
|
if ((bpf_get_current_pid_tgid() >> 32) != target_pid ||
|
|
id != __NR_getcwd)
|
|
return 0;
|
|
|
|
return copy_getcwd_arg((void *)PT_REGS_PARM1_SYSCALL(regs));
|
|
}
|
|
|
|
SEC("raw_tp.s/sys_enter")
|
|
int BPF_PROG(handle_sys_enter_raw_tp, struct pt_regs *regs, long id)
|
|
{
|
|
if ((bpf_get_current_pid_tgid() >> 32) != target_pid ||
|
|
id != __NR_getcwd)
|
|
return 0;
|
|
|
|
return copy_getcwd_arg((void *)PT_REGS_PARM1_CORE_SYSCALL(regs));
|
|
}
|
|
|
|
SEC("tp.s/syscalls/sys_enter_getcwd")
|
|
int handle_sys_enter_tp(struct syscall_trace_enter *args)
|
|
{
|
|
if ((bpf_get_current_pid_tgid() >> 32) != target_pid)
|
|
return 0;
|
|
|
|
return copy_getcwd_arg((void *)args->args[0]);
|
|
}
|
|
|
|
SEC("tp.s/syscalls/sys_exit_getcwd")
|
|
int handle_sys_exit_tp(struct syscall_trace_exit *args)
|
|
{
|
|
struct pt_regs *regs;
|
|
|
|
if ((bpf_get_current_pid_tgid() >> 32) != target_pid)
|
|
return 0;
|
|
|
|
regs = (struct pt_regs *)bpf_task_pt_regs(bpf_get_current_task_btf());
|
|
return copy_getcwd_arg((void *)PT_REGS_PARM1_CORE_SYSCALL(regs));
|
|
}
|
|
|
|
SEC("raw_tp.s")
|
|
int BPF_PROG(handle_raw_tp_bare, struct pt_regs *regs, long id)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
SEC("tp.s")
|
|
int handle_tp_bare(void *ctx)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
SEC("tracepoint.s/syscalls/sys_enter_getcwd")
|
|
int handle_sys_enter_tp_alias(struct syscall_trace_enter *args)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
SEC("raw_tracepoint.s/sys_enter")
|
|
int BPF_PROG(handle_sys_enter_raw_tp_alias, struct pt_regs *regs, long id)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
SEC("raw_tp.s/sys_enter")
|
|
int BPF_PROG(handle_test_run, struct pt_regs *regs, long id)
|
|
{
|
|
if ((__u64)regs == 0x1234ULL && (__u64)id == 0x5678ULL)
|
|
return (__u64)regs + (__u64)id;
|
|
|
|
return 0;
|
|
}
|
|
|
|
SEC("raw_tp.s/sched_switch")
|
|
int BPF_PROG(handle_raw_tp_non_faultable, bool preempt,
|
|
struct task_struct *prev, struct task_struct *next)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
SEC("tp.s/sched/sched_switch")
|
|
int handle_tp_non_syscall(void *ctx)
|
|
{
|
|
return 0;
|
|
}
|