mirror of
				https://github.com/torvalds/linux.git
				synced 2025-11-04 02:30:34 +02:00 
			
		
		
		
	Architectures that select ARCH_HAS_SYSCALL_WRAPPER pass a pointer to struct pt_regs to syscall handlers, others unpack it into individual function parameters. Introduce a macro to describe what a particular arch does. Signed-off-by: Ilya Leoshkevich <iii@linux.ibm.com> Signed-off-by: Andrii Nakryiko <andrii@kernel.org> Link: https://lore.kernel.org/bpf/20220209021745.2215452-3-iii@linux.ibm.com
		
			
				
	
	
		
			453 lines
		
	
	
	
		
			17 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			453 lines
		
	
	
	
		
			17 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause) */
 | 
						|
#ifndef __BPF_TRACING_H__
 | 
						|
#define __BPF_TRACING_H__
 | 
						|
 | 
						|
/* Scan the ARCH passed in from ARCH env variable (see Makefile) */
 | 
						|
#if defined(__TARGET_ARCH_x86)
 | 
						|
	#define bpf_target_x86
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__TARGET_ARCH_s390)
 | 
						|
	#define bpf_target_s390
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__TARGET_ARCH_arm)
 | 
						|
	#define bpf_target_arm
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__TARGET_ARCH_arm64)
 | 
						|
	#define bpf_target_arm64
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__TARGET_ARCH_mips)
 | 
						|
	#define bpf_target_mips
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__TARGET_ARCH_powerpc)
 | 
						|
	#define bpf_target_powerpc
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__TARGET_ARCH_sparc)
 | 
						|
	#define bpf_target_sparc
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__TARGET_ARCH_riscv)
 | 
						|
	#define bpf_target_riscv
 | 
						|
	#define bpf_target_defined
 | 
						|
#else
 | 
						|
 | 
						|
/* Fall back to what the compiler says */
 | 
						|
#if defined(__x86_64__)
 | 
						|
	#define bpf_target_x86
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__s390__)
 | 
						|
	#define bpf_target_s390
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__arm__)
 | 
						|
	#define bpf_target_arm
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__aarch64__)
 | 
						|
	#define bpf_target_arm64
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__mips__)
 | 
						|
	#define bpf_target_mips
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__powerpc__)
 | 
						|
	#define bpf_target_powerpc
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__sparc__)
 | 
						|
	#define bpf_target_sparc
 | 
						|
	#define bpf_target_defined
 | 
						|
#elif defined(__riscv) && __riscv_xlen == 64
 | 
						|
	#define bpf_target_riscv
 | 
						|
	#define bpf_target_defined
 | 
						|
#endif /* no compiler target */
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef __BPF_TARGET_MISSING
 | 
						|
#define __BPF_TARGET_MISSING "GCC error \"Must specify a BPF target arch via __TARGET_ARCH_xxx\""
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(bpf_target_x86)
 | 
						|
 | 
						|
#if defined(__KERNEL__) || defined(__VMLINUX_H__)
 | 
						|
 | 
						|
#define __PT_PARM1_REG di
 | 
						|
#define __PT_PARM2_REG si
 | 
						|
#define __PT_PARM3_REG dx
 | 
						|
#define __PT_PARM4_REG cx
 | 
						|
#define __PT_PARM4_REG_SYSCALL r10 /* syscall uses r10 */
 | 
						|
#define __PT_PARM5_REG r8
 | 
						|
#define __PT_RET_REG sp
 | 
						|
#define __PT_FP_REG bp
 | 
						|
#define __PT_RC_REG ax
 | 
						|
#define __PT_SP_REG sp
 | 
						|
#define __PT_IP_REG ip
 | 
						|
 | 
						|
#else
 | 
						|
 | 
						|
#ifdef __i386__
 | 
						|
 | 
						|
#define __PT_PARM1_REG eax
 | 
						|
#define __PT_PARM2_REG edx
 | 
						|
#define __PT_PARM3_REG ecx
 | 
						|
/* i386 kernel is built with -mregparm=3 */
 | 
						|
#define __PT_PARM4_REG __unsupported__
 | 
						|
#define __PT_PARM5_REG __unsupported__
 | 
						|
#define __PT_RET_REG esp
 | 
						|
#define __PT_FP_REG ebp
 | 
						|
#define __PT_RC_REG eax
 | 
						|
#define __PT_SP_REG esp
 | 
						|
#define __PT_IP_REG eip
 | 
						|
 | 
						|
#else /* __i386__ */
 | 
						|
 | 
						|
#define __PT_PARM1_REG rdi
 | 
						|
#define __PT_PARM2_REG rsi
 | 
						|
#define __PT_PARM3_REG rdx
 | 
						|
#define __PT_PARM4_REG rcx
 | 
						|
#define __PT_PARM4_REG_SYSCALL r10 /* syscall uses r10 */
 | 
						|
#define __PT_PARM5_REG r8
 | 
						|
#define __PT_RET_REG rsp
 | 
						|
#define __PT_FP_REG rbp
 | 
						|
#define __PT_RC_REG rax
 | 
						|
#define __PT_SP_REG rsp
 | 
						|
#define __PT_IP_REG rip
 | 
						|
 | 
						|
#endif /* __i386__ */
 | 
						|
 | 
						|
#endif /* __KERNEL__ || __VMLINUX_H__ */
 | 
						|
 | 
						|
#elif defined(bpf_target_s390)
 | 
						|
 | 
						|
/* s390 provides user_pt_regs instead of struct pt_regs to userspace */
 | 
						|
#define __PT_REGS_CAST(x) ((const user_pt_regs *)(x))
 | 
						|
#define __PT_PARM1_REG gprs[2]
 | 
						|
#define __PT_PARM2_REG gprs[3]
 | 
						|
#define __PT_PARM3_REG gprs[4]
 | 
						|
#define __PT_PARM4_REG gprs[5]
 | 
						|
#define __PT_PARM5_REG gprs[6]
 | 
						|
#define __PT_RET_REG grps[14]
 | 
						|
#define __PT_FP_REG gprs[11]	/* Works only with CONFIG_FRAME_POINTER */
 | 
						|
#define __PT_RC_REG gprs[2]
 | 
						|
#define __PT_SP_REG gprs[15]
 | 
						|
#define __PT_IP_REG psw.addr
 | 
						|
 | 
						|
#elif defined(bpf_target_arm)
 | 
						|
 | 
						|
#define __PT_PARM1_REG uregs[0]
 | 
						|
#define __PT_PARM2_REG uregs[1]
 | 
						|
#define __PT_PARM3_REG uregs[2]
 | 
						|
#define __PT_PARM4_REG uregs[3]
 | 
						|
#define __PT_PARM5_REG uregs[4]
 | 
						|
#define __PT_RET_REG uregs[14]
 | 
						|
#define __PT_FP_REG uregs[11]	/* Works only with CONFIG_FRAME_POINTER */
 | 
						|
#define __PT_RC_REG uregs[0]
 | 
						|
#define __PT_SP_REG uregs[13]
 | 
						|
#define __PT_IP_REG uregs[12]
 | 
						|
 | 
						|
#elif defined(bpf_target_arm64)
 | 
						|
 | 
						|
/* arm64 provides struct user_pt_regs instead of struct pt_regs to userspace */
 | 
						|
#define __PT_REGS_CAST(x) ((const struct user_pt_regs *)(x))
 | 
						|
#define __PT_PARM1_REG regs[0]
 | 
						|
#define __PT_PARM2_REG regs[1]
 | 
						|
#define __PT_PARM3_REG regs[2]
 | 
						|
#define __PT_PARM4_REG regs[3]
 | 
						|
#define __PT_PARM5_REG regs[4]
 | 
						|
#define __PT_RET_REG regs[30]
 | 
						|
#define __PT_FP_REG regs[29]	/* Works only with CONFIG_FRAME_POINTER */
 | 
						|
#define __PT_RC_REG regs[0]
 | 
						|
#define __PT_SP_REG sp
 | 
						|
#define __PT_IP_REG pc
 | 
						|
 | 
						|
#elif defined(bpf_target_mips)
 | 
						|
 | 
						|
#define __PT_PARM1_REG regs[4]
 | 
						|
#define __PT_PARM2_REG regs[5]
 | 
						|
#define __PT_PARM3_REG regs[6]
 | 
						|
#define __PT_PARM4_REG regs[7]
 | 
						|
#define __PT_PARM5_REG regs[8]
 | 
						|
#define __PT_RET_REG regs[31]
 | 
						|
#define __PT_FP_REG regs[30]	/* Works only with CONFIG_FRAME_POINTER */
 | 
						|
#define __PT_RC_REG regs[2]
 | 
						|
#define __PT_SP_REG regs[29]
 | 
						|
#define __PT_IP_REG cp0_epc
 | 
						|
 | 
						|
#elif defined(bpf_target_powerpc)
 | 
						|
 | 
						|
#define __PT_PARM1_REG gpr[3]
 | 
						|
#define __PT_PARM2_REG gpr[4]
 | 
						|
#define __PT_PARM3_REG gpr[5]
 | 
						|
#define __PT_PARM4_REG gpr[6]
 | 
						|
#define __PT_PARM5_REG gpr[7]
 | 
						|
#define __PT_RET_REG regs[31]
 | 
						|
#define __PT_FP_REG __unsupported__
 | 
						|
#define __PT_RC_REG gpr[3]
 | 
						|
#define __PT_SP_REG sp
 | 
						|
#define __PT_IP_REG nip
 | 
						|
 | 
						|
#elif defined(bpf_target_sparc)
 | 
						|
 | 
						|
#define __PT_PARM1_REG u_regs[UREG_I0]
 | 
						|
#define __PT_PARM2_REG u_regs[UREG_I1]
 | 
						|
#define __PT_PARM3_REG u_regs[UREG_I2]
 | 
						|
#define __PT_PARM4_REG u_regs[UREG_I3]
 | 
						|
#define __PT_PARM5_REG u_regs[UREG_I4]
 | 
						|
#define __PT_RET_REG u_regs[UREG_I7]
 | 
						|
#define __PT_FP_REG __unsupported__
 | 
						|
#define __PT_RC_REG u_regs[UREG_I0]
 | 
						|
#define __PT_SP_REG u_regs[UREG_FP]
 | 
						|
/* Should this also be a bpf_target check for the sparc case? */
 | 
						|
#if defined(__arch64__)
 | 
						|
#define __PT_IP_REG tpc
 | 
						|
#else
 | 
						|
#define __PT_IP_REG pc
 | 
						|
#endif
 | 
						|
 | 
						|
#elif defined(bpf_target_riscv)
 | 
						|
 | 
						|
#define __PT_REGS_CAST(x) ((const struct user_regs_struct *)(x))
 | 
						|
#define __PT_PARM1_REG a0
 | 
						|
#define __PT_PARM2_REG a1
 | 
						|
#define __PT_PARM3_REG a2
 | 
						|
#define __PT_PARM4_REG a3
 | 
						|
#define __PT_PARM5_REG a4
 | 
						|
#define __PT_RET_REG ra
 | 
						|
#define __PT_FP_REG fp
 | 
						|
#define __PT_RC_REG a5
 | 
						|
#define __PT_SP_REG sp
 | 
						|
#define __PT_IP_REG epc
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
#if defined(bpf_target_defined)
 | 
						|
 | 
						|
struct pt_regs;
 | 
						|
 | 
						|
/* allow some architecutres to override `struct pt_regs` */
 | 
						|
#ifndef __PT_REGS_CAST
 | 
						|
#define __PT_REGS_CAST(x) (x)
 | 
						|
#endif
 | 
						|
 | 
						|
#define PT_REGS_PARM1(x) (__PT_REGS_CAST(x)->__PT_PARM1_REG)
 | 
						|
#define PT_REGS_PARM2(x) (__PT_REGS_CAST(x)->__PT_PARM2_REG)
 | 
						|
#define PT_REGS_PARM3(x) (__PT_REGS_CAST(x)->__PT_PARM3_REG)
 | 
						|
#define PT_REGS_PARM4(x) (__PT_REGS_CAST(x)->__PT_PARM4_REG)
 | 
						|
#define PT_REGS_PARM5(x) (__PT_REGS_CAST(x)->__PT_PARM5_REG)
 | 
						|
#define PT_REGS_RET(x) (__PT_REGS_CAST(x)->__PT_RET_REG)
 | 
						|
#define PT_REGS_FP(x) (__PT_REGS_CAST(x)->__PT_FP_REG)
 | 
						|
#define PT_REGS_RC(x) (__PT_REGS_CAST(x)->__PT_RC_REG)
 | 
						|
#define PT_REGS_SP(x) (__PT_REGS_CAST(x)->__PT_SP_REG)
 | 
						|
#define PT_REGS_IP(x) (__PT_REGS_CAST(x)->__PT_IP_REG)
 | 
						|
 | 
						|
#define PT_REGS_PARM1_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM1_REG)
 | 
						|
#define PT_REGS_PARM2_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM2_REG)
 | 
						|
#define PT_REGS_PARM3_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM3_REG)
 | 
						|
#define PT_REGS_PARM4_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM4_REG)
 | 
						|
#define PT_REGS_PARM5_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM5_REG)
 | 
						|
#define PT_REGS_RET_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_RET_REG)
 | 
						|
#define PT_REGS_FP_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_FP_REG)
 | 
						|
#define PT_REGS_RC_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_RC_REG)
 | 
						|
#define PT_REGS_SP_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_SP_REG)
 | 
						|
#define PT_REGS_IP_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_IP_REG)
 | 
						|
 | 
						|
#if defined(bpf_target_powerpc)
 | 
						|
 | 
						|
#define BPF_KPROBE_READ_RET_IP(ip, ctx)		({ (ip) = (ctx)->link; })
 | 
						|
#define BPF_KRETPROBE_READ_RET_IP		BPF_KPROBE_READ_RET_IP
 | 
						|
 | 
						|
#elif defined(bpf_target_sparc)
 | 
						|
 | 
						|
#define BPF_KPROBE_READ_RET_IP(ip, ctx)		({ (ip) = PT_REGS_RET(ctx); })
 | 
						|
#define BPF_KRETPROBE_READ_RET_IP		BPF_KPROBE_READ_RET_IP
 | 
						|
 | 
						|
#else
 | 
						|
 | 
						|
#define BPF_KPROBE_READ_RET_IP(ip, ctx)					    \
 | 
						|
	({ bpf_probe_read_kernel(&(ip), sizeof(ip), (void *)PT_REGS_RET(ctx)); })
 | 
						|
#define BPF_KRETPROBE_READ_RET_IP(ip, ctx)				    \
 | 
						|
	({ bpf_probe_read_kernel(&(ip), sizeof(ip), (void *)(PT_REGS_FP(ctx) + sizeof(ip))); })
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
#define PT_REGS_PARM1_SYSCALL(x) PT_REGS_PARM1(x)
 | 
						|
#define PT_REGS_PARM2_SYSCALL(x) PT_REGS_PARM2(x)
 | 
						|
#define PT_REGS_PARM3_SYSCALL(x) PT_REGS_PARM3(x)
 | 
						|
#ifdef __PT_PARM4_REG_SYSCALL
 | 
						|
#define PT_REGS_PARM4_SYSCALL(x) (__PT_REGS_CAST(x)->__PT_PARM4_REG_SYSCALL)
 | 
						|
#else /* __PT_PARM4_REG_SYSCALL */
 | 
						|
#define PT_REGS_PARM4_SYSCALL(x) PT_REGS_PARM4(x)
 | 
						|
#endif
 | 
						|
#define PT_REGS_PARM5_SYSCALL(x) PT_REGS_PARM5(x)
 | 
						|
 | 
						|
#define PT_REGS_PARM1_CORE_SYSCALL(x) PT_REGS_PARM1_CORE(x)
 | 
						|
#define PT_REGS_PARM2_CORE_SYSCALL(x) PT_REGS_PARM2_CORE(x)
 | 
						|
#define PT_REGS_PARM3_CORE_SYSCALL(x) PT_REGS_PARM3_CORE(x)
 | 
						|
#ifdef __PT_PARM4_REG_SYSCALL
 | 
						|
#define PT_REGS_PARM4_CORE_SYSCALL(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM4_REG_SYSCALL)
 | 
						|
#else /* __PT_PARM4_REG_SYSCALL */
 | 
						|
#define PT_REGS_PARM4_CORE_SYSCALL(x) PT_REGS_PARM4_CORE(x)
 | 
						|
#endif
 | 
						|
#define PT_REGS_PARM5_CORE_SYSCALL(x) PT_REGS_PARM5_CORE(x)
 | 
						|
 | 
						|
#else /* defined(bpf_target_defined) */
 | 
						|
 | 
						|
#define PT_REGS_PARM1(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM2(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM3(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM4(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM5(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_RET(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_FP(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_RC(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_SP(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_IP(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
 | 
						|
#define PT_REGS_PARM1_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM2_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM3_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM4_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM5_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_RET_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_FP_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_RC_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_SP_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_IP_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
 | 
						|
#define BPF_KPROBE_READ_RET_IP(ip, ctx) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define BPF_KRETPROBE_READ_RET_IP(ip, ctx) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
 | 
						|
#define PT_REGS_PARM1_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM2_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM3_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM4_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM5_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
 | 
						|
#define PT_REGS_PARM1_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM2_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM3_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM4_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
#define PT_REGS_PARM5_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
 | 
						|
 | 
						|
#endif /* defined(bpf_target_defined) */
 | 
						|
 | 
						|
/*
 | 
						|
 * When invoked from a syscall handler kprobe, returns a pointer to a
 | 
						|
 * struct pt_regs containing syscall arguments and suitable for passing to
 | 
						|
 * PT_REGS_PARMn_SYSCALL() and PT_REGS_PARMn_CORE_SYSCALL().
 | 
						|
 */
 | 
						|
#ifndef PT_REGS_SYSCALL_REGS
 | 
						|
/* By default, assume that the arch selects ARCH_HAS_SYSCALL_WRAPPER. */
 | 
						|
#define PT_REGS_SYSCALL_REGS(ctx) ((struct pt_regs *)PT_REGS_PARM1(ctx))
 | 
						|
#endif
 | 
						|
 | 
						|
#ifndef ___bpf_concat
 | 
						|
#define ___bpf_concat(a, b) a ## b
 | 
						|
#endif
 | 
						|
#ifndef ___bpf_apply
 | 
						|
#define ___bpf_apply(fn, n) ___bpf_concat(fn, n)
 | 
						|
#endif
 | 
						|
#ifndef ___bpf_nth
 | 
						|
#define ___bpf_nth(_, _1, _2, _3, _4, _5, _6, _7, _8, _9, _a, _b, _c, N, ...) N
 | 
						|
#endif
 | 
						|
#ifndef ___bpf_narg
 | 
						|
#define ___bpf_narg(...) ___bpf_nth(_, ##__VA_ARGS__, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)
 | 
						|
#endif
 | 
						|
 | 
						|
#define ___bpf_ctx_cast0()            ctx
 | 
						|
#define ___bpf_ctx_cast1(x)           ___bpf_ctx_cast0(), (void *)ctx[0]
 | 
						|
#define ___bpf_ctx_cast2(x, args...)  ___bpf_ctx_cast1(args), (void *)ctx[1]
 | 
						|
#define ___bpf_ctx_cast3(x, args...)  ___bpf_ctx_cast2(args), (void *)ctx[2]
 | 
						|
#define ___bpf_ctx_cast4(x, args...)  ___bpf_ctx_cast3(args), (void *)ctx[3]
 | 
						|
#define ___bpf_ctx_cast5(x, args...)  ___bpf_ctx_cast4(args), (void *)ctx[4]
 | 
						|
#define ___bpf_ctx_cast6(x, args...)  ___bpf_ctx_cast5(args), (void *)ctx[5]
 | 
						|
#define ___bpf_ctx_cast7(x, args...)  ___bpf_ctx_cast6(args), (void *)ctx[6]
 | 
						|
#define ___bpf_ctx_cast8(x, args...)  ___bpf_ctx_cast7(args), (void *)ctx[7]
 | 
						|
#define ___bpf_ctx_cast9(x, args...)  ___bpf_ctx_cast8(args), (void *)ctx[8]
 | 
						|
#define ___bpf_ctx_cast10(x, args...) ___bpf_ctx_cast9(args), (void *)ctx[9]
 | 
						|
#define ___bpf_ctx_cast11(x, args...) ___bpf_ctx_cast10(args), (void *)ctx[10]
 | 
						|
#define ___bpf_ctx_cast12(x, args...) ___bpf_ctx_cast11(args), (void *)ctx[11]
 | 
						|
#define ___bpf_ctx_cast(args...)      ___bpf_apply(___bpf_ctx_cast, ___bpf_narg(args))(args)
 | 
						|
 | 
						|
/*
 | 
						|
 * BPF_PROG is a convenience wrapper for generic tp_btf/fentry/fexit and
 | 
						|
 * similar kinds of BPF programs, that accept input arguments as a single
 | 
						|
 * pointer to untyped u64 array, where each u64 can actually be a typed
 | 
						|
 * pointer or integer of different size. Instead of requring user to write
 | 
						|
 * manual casts and work with array elements by index, BPF_PROG macro
 | 
						|
 * allows user to declare a list of named and typed input arguments in the
 | 
						|
 * same syntax as for normal C function. All the casting is hidden and
 | 
						|
 * performed transparently, while user code can just assume working with
 | 
						|
 * function arguments of specified type and name.
 | 
						|
 *
 | 
						|
 * Original raw context argument is preserved as well as 'ctx' argument.
 | 
						|
 * This is useful when using BPF helpers that expect original context
 | 
						|
 * as one of the parameters (e.g., for bpf_perf_event_output()).
 | 
						|
 */
 | 
						|
#define BPF_PROG(name, args...)						    \
 | 
						|
name(unsigned long long *ctx);						    \
 | 
						|
static __attribute__((always_inline)) typeof(name(0))			    \
 | 
						|
____##name(unsigned long long *ctx, ##args);				    \
 | 
						|
typeof(name(0)) name(unsigned long long *ctx)				    \
 | 
						|
{									    \
 | 
						|
	_Pragma("GCC diagnostic push")					    \
 | 
						|
	_Pragma("GCC diagnostic ignored \"-Wint-conversion\"")		    \
 | 
						|
	return ____##name(___bpf_ctx_cast(args));			    \
 | 
						|
	_Pragma("GCC diagnostic pop")					    \
 | 
						|
}									    \
 | 
						|
static __attribute__((always_inline)) typeof(name(0))			    \
 | 
						|
____##name(unsigned long long *ctx, ##args)
 | 
						|
 | 
						|
struct pt_regs;
 | 
						|
 | 
						|
#define ___bpf_kprobe_args0()           ctx
 | 
						|
#define ___bpf_kprobe_args1(x)          ___bpf_kprobe_args0(), (void *)PT_REGS_PARM1(ctx)
 | 
						|
#define ___bpf_kprobe_args2(x, args...) ___bpf_kprobe_args1(args), (void *)PT_REGS_PARM2(ctx)
 | 
						|
#define ___bpf_kprobe_args3(x, args...) ___bpf_kprobe_args2(args), (void *)PT_REGS_PARM3(ctx)
 | 
						|
#define ___bpf_kprobe_args4(x, args...) ___bpf_kprobe_args3(args), (void *)PT_REGS_PARM4(ctx)
 | 
						|
#define ___bpf_kprobe_args5(x, args...) ___bpf_kprobe_args4(args), (void *)PT_REGS_PARM5(ctx)
 | 
						|
#define ___bpf_kprobe_args(args...)     ___bpf_apply(___bpf_kprobe_args, ___bpf_narg(args))(args)
 | 
						|
 | 
						|
/*
 | 
						|
 * BPF_KPROBE serves the same purpose for kprobes as BPF_PROG for
 | 
						|
 * tp_btf/fentry/fexit BPF programs. It hides the underlying platform-specific
 | 
						|
 * low-level way of getting kprobe input arguments from struct pt_regs, and
 | 
						|
 * provides a familiar typed and named function arguments syntax and
 | 
						|
 * semantics of accessing kprobe input paremeters.
 | 
						|
 *
 | 
						|
 * Original struct pt_regs* context is preserved as 'ctx' argument. This might
 | 
						|
 * be necessary when using BPF helpers like bpf_perf_event_output().
 | 
						|
 */
 | 
						|
#define BPF_KPROBE(name, args...)					    \
 | 
						|
name(struct pt_regs *ctx);						    \
 | 
						|
static __attribute__((always_inline)) typeof(name(0))			    \
 | 
						|
____##name(struct pt_regs *ctx, ##args);				    \
 | 
						|
typeof(name(0)) name(struct pt_regs *ctx)				    \
 | 
						|
{									    \
 | 
						|
	_Pragma("GCC diagnostic push")					    \
 | 
						|
	_Pragma("GCC diagnostic ignored \"-Wint-conversion\"")		    \
 | 
						|
	return ____##name(___bpf_kprobe_args(args));			    \
 | 
						|
	_Pragma("GCC diagnostic pop")					    \
 | 
						|
}									    \
 | 
						|
static __attribute__((always_inline)) typeof(name(0))			    \
 | 
						|
____##name(struct pt_regs *ctx, ##args)
 | 
						|
 | 
						|
#define ___bpf_kretprobe_args0()       ctx
 | 
						|
#define ___bpf_kretprobe_args1(x)      ___bpf_kretprobe_args0(), (void *)PT_REGS_RC(ctx)
 | 
						|
#define ___bpf_kretprobe_args(args...) ___bpf_apply(___bpf_kretprobe_args, ___bpf_narg(args))(args)
 | 
						|
 | 
						|
/*
 | 
						|
 * BPF_KRETPROBE is similar to BPF_KPROBE, except, it only provides optional
 | 
						|
 * return value (in addition to `struct pt_regs *ctx`), but no input
 | 
						|
 * arguments, because they will be clobbered by the time probed function
 | 
						|
 * returns.
 | 
						|
 */
 | 
						|
#define BPF_KRETPROBE(name, args...)					    \
 | 
						|
name(struct pt_regs *ctx);						    \
 | 
						|
static __attribute__((always_inline)) typeof(name(0))			    \
 | 
						|
____##name(struct pt_regs *ctx, ##args);				    \
 | 
						|
typeof(name(0)) name(struct pt_regs *ctx)				    \
 | 
						|
{									    \
 | 
						|
	_Pragma("GCC diagnostic push")					    \
 | 
						|
	_Pragma("GCC diagnostic ignored \"-Wint-conversion\"")		    \
 | 
						|
	return ____##name(___bpf_kretprobe_args(args));			    \
 | 
						|
	_Pragma("GCC diagnostic pop")					    \
 | 
						|
}									    \
 | 
						|
static __always_inline typeof(name(0)) ____##name(struct pt_regs *ctx, ##args)
 | 
						|
 | 
						|
#endif
 |