libbpf: Fix and complete ARC spec in bpf_tracing.h
[platform/kernel/linux-starfive.git] / tools / lib / bpf / bpf_tracing.h
1 /* SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause) */
2 #ifndef __BPF_TRACING_H__
3 #define __BPF_TRACING_H__
4
5 #include <bpf/bpf_helpers.h>
6
7 /* Scan the ARCH passed in from ARCH env variable (see Makefile) */
8 #if defined(__TARGET_ARCH_x86)
9         #define bpf_target_x86
10         #define bpf_target_defined
11 #elif defined(__TARGET_ARCH_s390)
12         #define bpf_target_s390
13         #define bpf_target_defined
14 #elif defined(__TARGET_ARCH_arm)
15         #define bpf_target_arm
16         #define bpf_target_defined
17 #elif defined(__TARGET_ARCH_arm64)
18         #define bpf_target_arm64
19         #define bpf_target_defined
20 #elif defined(__TARGET_ARCH_mips)
21         #define bpf_target_mips
22         #define bpf_target_defined
23 #elif defined(__TARGET_ARCH_powerpc)
24         #define bpf_target_powerpc
25         #define bpf_target_defined
26 #elif defined(__TARGET_ARCH_sparc)
27         #define bpf_target_sparc
28         #define bpf_target_defined
29 #elif defined(__TARGET_ARCH_riscv)
30         #define bpf_target_riscv
31         #define bpf_target_defined
32 #elif defined(__TARGET_ARCH_arc)
33         #define bpf_target_arc
34         #define bpf_target_defined
35 #elif defined(__TARGET_ARCH_loongarch)
36         #define bpf_target_loongarch
37         #define bpf_target_defined
38 #else
39
40 /* Fall back to what the compiler says */
41 #if defined(__x86_64__)
42         #define bpf_target_x86
43         #define bpf_target_defined
44 #elif defined(__s390__)
45         #define bpf_target_s390
46         #define bpf_target_defined
47 #elif defined(__arm__)
48         #define bpf_target_arm
49         #define bpf_target_defined
50 #elif defined(__aarch64__)
51         #define bpf_target_arm64
52         #define bpf_target_defined
53 #elif defined(__mips__)
54         #define bpf_target_mips
55         #define bpf_target_defined
56 #elif defined(__powerpc__)
57         #define bpf_target_powerpc
58         #define bpf_target_defined
59 #elif defined(__sparc__)
60         #define bpf_target_sparc
61         #define bpf_target_defined
62 #elif defined(__riscv) && __riscv_xlen == 64
63         #define bpf_target_riscv
64         #define bpf_target_defined
65 #elif defined(__arc__)
66         #define bpf_target_arc
67         #define bpf_target_defined
68 #elif defined(__loongarch__)
69         #define bpf_target_loongarch
70         #define bpf_target_defined
71 #endif /* no compiler target */
72
73 #endif
74
75 #ifndef __BPF_TARGET_MISSING
76 #define __BPF_TARGET_MISSING "GCC error \"Must specify a BPF target arch via __TARGET_ARCH_xxx\""
77 #endif
78
79 #if defined(bpf_target_x86)
80
81 /*
82  * https://en.wikipedia.org/wiki/X86_calling_conventions#System_V_AMD64_ABI
83  */
84
85 #if defined(__KERNEL__) || defined(__VMLINUX_H__)
86
87 #define __PT_PARM1_REG di
88 #define __PT_PARM2_REG si
89 #define __PT_PARM3_REG dx
90 #define __PT_PARM4_REG cx
91 #define __PT_PARM5_REG r8
92 #define __PT_PARM6_REG r9
93 #define __PT_RET_REG sp
94 #define __PT_FP_REG bp
95 #define __PT_RC_REG ax
96 #define __PT_SP_REG sp
97 #define __PT_IP_REG ip
98 /* syscall uses r10 for PARM4 */
99 #define PT_REGS_PARM4_SYSCALL(x) ((x)->r10)
100 #define PT_REGS_PARM4_CORE_SYSCALL(x) BPF_CORE_READ(x, r10)
101
102 #else
103
104 #ifdef __i386__
105
106 /* i386 kernel is built with -mregparm=3 */
107 #define __PT_PARM1_REG eax
108 #define __PT_PARM2_REG edx
109 #define __PT_PARM3_REG ecx
110 #define __PT_RET_REG esp
111 #define __PT_FP_REG ebp
112 #define __PT_RC_REG eax
113 #define __PT_SP_REG esp
114 #define __PT_IP_REG eip
115
116 #else /* __i386__ */
117
118 #define __PT_PARM1_REG rdi
119 #define __PT_PARM2_REG rsi
120 #define __PT_PARM3_REG rdx
121 #define __PT_PARM4_REG rcx
122 #define __PT_PARM5_REG r8
123 #define __PT_PARM6_REG r9
124 #define __PT_RET_REG rsp
125 #define __PT_FP_REG rbp
126 #define __PT_RC_REG rax
127 #define __PT_SP_REG rsp
128 #define __PT_IP_REG rip
129 /* syscall uses r10 for PARM4 */
130 #define PT_REGS_PARM4_SYSCALL(x) ((x)->r10)
131 #define PT_REGS_PARM4_CORE_SYSCALL(x) BPF_CORE_READ(x, r10)
132
133 #endif /* __i386__ */
134
135 #endif /* __KERNEL__ || __VMLINUX_H__ */
136
137 #elif defined(bpf_target_s390)
138
139 struct pt_regs___s390 {
140         unsigned long orig_gpr2;
141 };
142
143 /* s390 provides user_pt_regs instead of struct pt_regs to userspace */
144 #define __PT_REGS_CAST(x) ((const user_pt_regs *)(x))
145 #define __PT_PARM1_REG gprs[2]
146 #define __PT_PARM2_REG gprs[3]
147 #define __PT_PARM3_REG gprs[4]
148 #define __PT_PARM4_REG gprs[5]
149 #define __PT_PARM5_REG gprs[6]
150 #define __PT_RET_REG gprs[14]
151 #define __PT_FP_REG gprs[11]    /* Works only with CONFIG_FRAME_POINTER */
152 #define __PT_RC_REG gprs[2]
153 #define __PT_SP_REG gprs[15]
154 #define __PT_IP_REG psw.addr
155 #define PT_REGS_PARM1_SYSCALL(x) PT_REGS_PARM1_CORE_SYSCALL(x)
156 #define PT_REGS_PARM1_CORE_SYSCALL(x) BPF_CORE_READ((const struct pt_regs___s390 *)(x), orig_gpr2)
157
158 #elif defined(bpf_target_arm)
159
160 /*
161  * https://github.com/ARM-software/abi-aa/blob/main/aapcs32/aapcs32.rst#machine-registers
162  */
163
164 #define __PT_PARM1_REG uregs[0]
165 #define __PT_PARM2_REG uregs[1]
166 #define __PT_PARM3_REG uregs[2]
167 #define __PT_PARM4_REG uregs[3]
168 #define __PT_RET_REG uregs[14]
169 #define __PT_FP_REG uregs[11]   /* Works only with CONFIG_FRAME_POINTER */
170 #define __PT_RC_REG uregs[0]
171 #define __PT_SP_REG uregs[13]
172 #define __PT_IP_REG uregs[12]
173
174 #elif defined(bpf_target_arm64)
175
176 /*
177  * https://github.com/ARM-software/abi-aa/blob/main/aapcs64/aapcs64.rst#machine-registers
178  */
179
180 struct pt_regs___arm64 {
181         unsigned long orig_x0;
182 };
183
184 /* arm64 provides struct user_pt_regs instead of struct pt_regs to userspace */
185 #define __PT_REGS_CAST(x) ((const struct user_pt_regs *)(x))
186 #define __PT_PARM1_REG regs[0]
187 #define __PT_PARM2_REG regs[1]
188 #define __PT_PARM3_REG regs[2]
189 #define __PT_PARM4_REG regs[3]
190 #define __PT_PARM5_REG regs[4]
191 #define __PT_PARM6_REG regs[5]
192 #define __PT_PARM7_REG regs[6]
193 #define __PT_PARM8_REG regs[7]
194 #define __PT_RET_REG regs[30]
195 #define __PT_FP_REG regs[29]    /* Works only with CONFIG_FRAME_POINTER */
196 #define __PT_RC_REG regs[0]
197 #define __PT_SP_REG sp
198 #define __PT_IP_REG pc
199 #define PT_REGS_PARM1_SYSCALL(x) PT_REGS_PARM1_CORE_SYSCALL(x)
200 #define PT_REGS_PARM1_CORE_SYSCALL(x) BPF_CORE_READ((const struct pt_regs___arm64 *)(x), orig_x0)
201
202 #elif defined(bpf_target_mips)
203
204 /*
205  * N64 ABI is assumed right now.
206  * https://en.wikipedia.org/wiki/MIPS_architecture#Calling_conventions
207  */
208
209 #define __PT_PARM1_REG regs[4]
210 #define __PT_PARM2_REG regs[5]
211 #define __PT_PARM3_REG regs[6]
212 #define __PT_PARM4_REG regs[7]
213 #define __PT_PARM5_REG regs[8]
214 #define __PT_PARM6_REG regs[9]
215 #define __PT_PARM7_REG regs[10]
216 #define __PT_PARM8_REG regs[11]
217 #define __PT_RET_REG regs[31]
218 #define __PT_FP_REG regs[30]    /* Works only with CONFIG_FRAME_POINTER */
219 #define __PT_RC_REG regs[2]
220 #define __PT_SP_REG regs[29]
221 #define __PT_IP_REG cp0_epc
222
223 #elif defined(bpf_target_powerpc)
224
225 /*
226  * http://refspecs.linux-foundation.org/elf/elfspec_ppc.pdf (page 3-14,
227  * section "Function Calling Sequence")
228  */
229
230 #define __PT_PARM1_REG gpr[3]
231 #define __PT_PARM2_REG gpr[4]
232 #define __PT_PARM3_REG gpr[5]
233 #define __PT_PARM4_REG gpr[6]
234 #define __PT_PARM5_REG gpr[7]
235 #define __PT_PARM6_REG gpr[8]
236 #define __PT_PARM7_REG gpr[9]
237 #define __PT_PARM8_REG gpr[10]
238 #define __PT_RET_REG regs[31]
239 #define __PT_FP_REG __unsupported__
240 #define __PT_RC_REG gpr[3]
241 #define __PT_SP_REG sp
242 #define __PT_IP_REG nip
243 /* powerpc does not select ARCH_HAS_SYSCALL_WRAPPER. */
244 #define PT_REGS_SYSCALL_REGS(ctx) ctx
245
246 #elif defined(bpf_target_sparc)
247
248 /*
249  * https://en.wikipedia.org/wiki/Calling_convention#SPARC
250  */
251
252 #define __PT_PARM1_REG u_regs[UREG_I0]
253 #define __PT_PARM2_REG u_regs[UREG_I1]
254 #define __PT_PARM3_REG u_regs[UREG_I2]
255 #define __PT_PARM4_REG u_regs[UREG_I3]
256 #define __PT_PARM5_REG u_regs[UREG_I4]
257 #define __PT_PARM6_REG u_regs[UREG_I5]
258 #define __PT_RET_REG u_regs[UREG_I7]
259 #define __PT_FP_REG __unsupported__
260 #define __PT_RC_REG u_regs[UREG_I0]
261 #define __PT_SP_REG u_regs[UREG_FP]
262 /* Should this also be a bpf_target check for the sparc case? */
263 #if defined(__arch64__)
264 #define __PT_IP_REG tpc
265 #else
266 #define __PT_IP_REG pc
267 #endif
268
269 #elif defined(bpf_target_riscv)
270
271 /*
272  * https://github.com/riscv-non-isa/riscv-elf-psabi-doc/blob/master/riscv-cc.adoc#risc-v-calling-conventions
273  */
274
275 #define __PT_REGS_CAST(x) ((const struct user_regs_struct *)(x))
276 #define __PT_PARM1_REG a0
277 #define __PT_PARM2_REG a1
278 #define __PT_PARM3_REG a2
279 #define __PT_PARM4_REG a3
280 #define __PT_PARM5_REG a4
281 #define __PT_PARM6_REG a5
282 #define __PT_PARM7_REG a6
283 #define __PT_PARM8_REG a7
284 #define __PT_RET_REG ra
285 #define __PT_FP_REG s0
286 #define __PT_RC_REG a0
287 #define __PT_SP_REG sp
288 #define __PT_IP_REG pc
289 /* riscv does not select ARCH_HAS_SYSCALL_WRAPPER. */
290 #define PT_REGS_SYSCALL_REGS(ctx) ctx
291
292 #elif defined(bpf_target_arc)
293
294 /*
295  * Section "Function Calling Sequence" (page 24):
296  * https://raw.githubusercontent.com/wiki/foss-for-synopsys-dwc-arc-processors/toolchain/files/ARCv2_ABI.pdf
297  */
298
299 /* arc provides struct user_pt_regs instead of struct pt_regs to userspace */
300 #define __PT_REGS_CAST(x) ((const struct user_regs_struct *)(x))
301 #define __PT_PARM1_REG scratch.r0
302 #define __PT_PARM2_REG scratch.r1
303 #define __PT_PARM3_REG scratch.r2
304 #define __PT_PARM4_REG scratch.r3
305 #define __PT_PARM5_REG scratch.r4
306 #define __PT_PARM6_REG scratch.r5
307 #define __PT_PARM7_REG scratch.r6
308 #define __PT_PARM8_REG scratch.r7
309 #define __PT_RET_REG scratch.blink
310 #define __PT_FP_REG scratch.fp
311 #define __PT_RC_REG scratch.r0
312 #define __PT_SP_REG scratch.sp
313 #define __PT_IP_REG scratch.ret
314 /* arc does not select ARCH_HAS_SYSCALL_WRAPPER. */
315 #define PT_REGS_SYSCALL_REGS(ctx) ctx
316
317 #elif defined(bpf_target_loongarch)
318
319 /* https://loongson.github.io/LoongArch-Documentation/LoongArch-ELF-ABI-EN.html */
320
321 #define __PT_PARM1_REG regs[4]
322 #define __PT_PARM2_REG regs[5]
323 #define __PT_PARM3_REG regs[6]
324 #define __PT_PARM4_REG regs[7]
325 #define __PT_PARM5_REG regs[8]
326 #define __PT_RET_REG regs[1]
327 #define __PT_FP_REG regs[22]
328 #define __PT_RC_REG regs[4]
329 #define __PT_SP_REG regs[3]
330 #define __PT_IP_REG csr_era
331 /* loongarch does not select ARCH_HAS_SYSCALL_WRAPPER. */
332 #define PT_REGS_SYSCALL_REGS(ctx) ctx
333
334 #endif
335
336 #if defined(bpf_target_defined)
337
338 struct pt_regs;
339
340 /* allow some architectures to override `struct pt_regs` */
341 #ifndef __PT_REGS_CAST
342 #define __PT_REGS_CAST(x) (x)
343 #endif
344
345 /*
346  * Different architectures support different number of arguments passed
347  * through registers. i386 supports just 3, some arches support up to 8.
348  */
349 #ifndef __PT_PARM4_REG
350 #define __PT_PARM4_REG __unsupported__
351 #endif
352 #ifndef __PT_PARM5_REG
353 #define __PT_PARM5_REG __unsupported__
354 #endif
355 #ifndef __PT_PARM6_REG
356 #define __PT_PARM6_REG __unsupported__
357 #endif
358 #ifndef __PT_PARM7_REG
359 #define __PT_PARM7_REG __unsupported__
360 #endif
361 #ifndef __PT_PARM8_REG
362 #define __PT_PARM8_REG __unsupported__
363 #endif
364
365 #define PT_REGS_PARM1(x) (__PT_REGS_CAST(x)->__PT_PARM1_REG)
366 #define PT_REGS_PARM2(x) (__PT_REGS_CAST(x)->__PT_PARM2_REG)
367 #define PT_REGS_PARM3(x) (__PT_REGS_CAST(x)->__PT_PARM3_REG)
368 #define PT_REGS_PARM4(x) (__PT_REGS_CAST(x)->__PT_PARM4_REG)
369 #define PT_REGS_PARM5(x) (__PT_REGS_CAST(x)->__PT_PARM5_REG)
370 #define PT_REGS_PARM6(x) (__PT_REGS_CAST(x)->__PT_PARM6_REG)
371 #define PT_REGS_PARM7(x) (__PT_REGS_CAST(x)->__PT_PARM7_REG)
372 #define PT_REGS_PARM8(x) (__PT_REGS_CAST(x)->__PT_PARM8_REG)
373 #define PT_REGS_RET(x) (__PT_REGS_CAST(x)->__PT_RET_REG)
374 #define PT_REGS_FP(x) (__PT_REGS_CAST(x)->__PT_FP_REG)
375 #define PT_REGS_RC(x) (__PT_REGS_CAST(x)->__PT_RC_REG)
376 #define PT_REGS_SP(x) (__PT_REGS_CAST(x)->__PT_SP_REG)
377 #define PT_REGS_IP(x) (__PT_REGS_CAST(x)->__PT_IP_REG)
378
379 #define PT_REGS_PARM1_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM1_REG)
380 #define PT_REGS_PARM2_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM2_REG)
381 #define PT_REGS_PARM3_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM3_REG)
382 #define PT_REGS_PARM4_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM4_REG)
383 #define PT_REGS_PARM5_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM5_REG)
384 #define PT_REGS_PARM6_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM6_REG)
385 #define PT_REGS_PARM7_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM7_REG)
386 #define PT_REGS_PARM8_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_PARM8_REG)
387 #define PT_REGS_RET_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_RET_REG)
388 #define PT_REGS_FP_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_FP_REG)
389 #define PT_REGS_RC_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_RC_REG)
390 #define PT_REGS_SP_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_SP_REG)
391 #define PT_REGS_IP_CORE(x) BPF_CORE_READ(__PT_REGS_CAST(x), __PT_IP_REG)
392
393 #if defined(bpf_target_powerpc)
394
395 #define BPF_KPROBE_READ_RET_IP(ip, ctx)         ({ (ip) = (ctx)->link; })
396 #define BPF_KRETPROBE_READ_RET_IP               BPF_KPROBE_READ_RET_IP
397
398 #elif defined(bpf_target_sparc)
399
400 #define BPF_KPROBE_READ_RET_IP(ip, ctx)         ({ (ip) = PT_REGS_RET(ctx); })
401 #define BPF_KRETPROBE_READ_RET_IP               BPF_KPROBE_READ_RET_IP
402
403 #else
404
405 #define BPF_KPROBE_READ_RET_IP(ip, ctx)                                     \
406         ({ bpf_probe_read_kernel(&(ip), sizeof(ip), (void *)PT_REGS_RET(ctx)); })
407 #define BPF_KRETPROBE_READ_RET_IP(ip, ctx)                                  \
408         ({ bpf_probe_read_kernel(&(ip), sizeof(ip), (void *)(PT_REGS_FP(ctx) + sizeof(ip))); })
409
410 #endif
411
412 #ifndef PT_REGS_PARM1_SYSCALL
413 #define PT_REGS_PARM1_SYSCALL(x) PT_REGS_PARM1(x)
414 #endif
415 #define PT_REGS_PARM2_SYSCALL(x) PT_REGS_PARM2(x)
416 #define PT_REGS_PARM3_SYSCALL(x) PT_REGS_PARM3(x)
417 #ifndef PT_REGS_PARM4_SYSCALL
418 #define PT_REGS_PARM4_SYSCALL(x) PT_REGS_PARM4(x)
419 #endif
420 #define PT_REGS_PARM5_SYSCALL(x) PT_REGS_PARM5(x)
421
422 #ifndef PT_REGS_PARM1_CORE_SYSCALL
423 #define PT_REGS_PARM1_CORE_SYSCALL(x) PT_REGS_PARM1_CORE(x)
424 #endif
425 #define PT_REGS_PARM2_CORE_SYSCALL(x) PT_REGS_PARM2_CORE(x)
426 #define PT_REGS_PARM3_CORE_SYSCALL(x) PT_REGS_PARM3_CORE(x)
427 #ifndef PT_REGS_PARM4_CORE_SYSCALL
428 #define PT_REGS_PARM4_CORE_SYSCALL(x) PT_REGS_PARM4_CORE(x)
429 #endif
430 #define PT_REGS_PARM5_CORE_SYSCALL(x) PT_REGS_PARM5_CORE(x)
431
432 #else /* defined(bpf_target_defined) */
433
434 #define PT_REGS_PARM1(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
435 #define PT_REGS_PARM2(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
436 #define PT_REGS_PARM3(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
437 #define PT_REGS_PARM4(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
438 #define PT_REGS_PARM5(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
439 #define PT_REGS_PARM6(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
440 #define PT_REGS_PARM7(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
441 #define PT_REGS_PARM8(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
442 #define PT_REGS_RET(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
443 #define PT_REGS_FP(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
444 #define PT_REGS_RC(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
445 #define PT_REGS_SP(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
446 #define PT_REGS_IP(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
447
448 #define PT_REGS_PARM1_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
449 #define PT_REGS_PARM2_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
450 #define PT_REGS_PARM3_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
451 #define PT_REGS_PARM4_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
452 #define PT_REGS_PARM5_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
453 #define PT_REGS_PARM6_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
454 #define PT_REGS_PARM7_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
455 #define PT_REGS_PARM8_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
456 #define PT_REGS_RET_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
457 #define PT_REGS_FP_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
458 #define PT_REGS_RC_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
459 #define PT_REGS_SP_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
460 #define PT_REGS_IP_CORE(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
461
462 #define BPF_KPROBE_READ_RET_IP(ip, ctx) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
463 #define BPF_KRETPROBE_READ_RET_IP(ip, ctx) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
464
465 #define PT_REGS_PARM1_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
466 #define PT_REGS_PARM2_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
467 #define PT_REGS_PARM3_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
468 #define PT_REGS_PARM4_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
469 #define PT_REGS_PARM5_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
470
471 #define PT_REGS_PARM1_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
472 #define PT_REGS_PARM2_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
473 #define PT_REGS_PARM3_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
474 #define PT_REGS_PARM4_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
475 #define PT_REGS_PARM5_CORE_SYSCALL(x) ({ _Pragma(__BPF_TARGET_MISSING); 0l; })
476
477 #endif /* defined(bpf_target_defined) */
478
479 /*
480  * When invoked from a syscall handler kprobe, returns a pointer to a
481  * struct pt_regs containing syscall arguments and suitable for passing to
482  * PT_REGS_PARMn_SYSCALL() and PT_REGS_PARMn_CORE_SYSCALL().
483  */
484 #ifndef PT_REGS_SYSCALL_REGS
485 /* By default, assume that the arch selects ARCH_HAS_SYSCALL_WRAPPER. */
486 #define PT_REGS_SYSCALL_REGS(ctx) ((struct pt_regs *)PT_REGS_PARM1(ctx))
487 #endif
488
489 #ifndef ___bpf_concat
490 #define ___bpf_concat(a, b) a ## b
491 #endif
492 #ifndef ___bpf_apply
493 #define ___bpf_apply(fn, n) ___bpf_concat(fn, n)
494 #endif
495 #ifndef ___bpf_nth
496 #define ___bpf_nth(_, _1, _2, _3, _4, _5, _6, _7, _8, _9, _a, _b, _c, N, ...) N
497 #endif
498 #ifndef ___bpf_narg
499 #define ___bpf_narg(...) ___bpf_nth(_, ##__VA_ARGS__, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0)
500 #endif
501
502 #define ___bpf_ctx_cast0()            ctx
503 #define ___bpf_ctx_cast1(x)           ___bpf_ctx_cast0(), (void *)ctx[0]
504 #define ___bpf_ctx_cast2(x, args...)  ___bpf_ctx_cast1(args), (void *)ctx[1]
505 #define ___bpf_ctx_cast3(x, args...)  ___bpf_ctx_cast2(args), (void *)ctx[2]
506 #define ___bpf_ctx_cast4(x, args...)  ___bpf_ctx_cast3(args), (void *)ctx[3]
507 #define ___bpf_ctx_cast5(x, args...)  ___bpf_ctx_cast4(args), (void *)ctx[4]
508 #define ___bpf_ctx_cast6(x, args...)  ___bpf_ctx_cast5(args), (void *)ctx[5]
509 #define ___bpf_ctx_cast7(x, args...)  ___bpf_ctx_cast6(args), (void *)ctx[6]
510 #define ___bpf_ctx_cast8(x, args...)  ___bpf_ctx_cast7(args), (void *)ctx[7]
511 #define ___bpf_ctx_cast9(x, args...)  ___bpf_ctx_cast8(args), (void *)ctx[8]
512 #define ___bpf_ctx_cast10(x, args...) ___bpf_ctx_cast9(args), (void *)ctx[9]
513 #define ___bpf_ctx_cast11(x, args...) ___bpf_ctx_cast10(args), (void *)ctx[10]
514 #define ___bpf_ctx_cast12(x, args...) ___bpf_ctx_cast11(args), (void *)ctx[11]
515 #define ___bpf_ctx_cast(args...)      ___bpf_apply(___bpf_ctx_cast, ___bpf_narg(args))(args)
516
517 /*
518  * BPF_PROG is a convenience wrapper for generic tp_btf/fentry/fexit and
519  * similar kinds of BPF programs, that accept input arguments as a single
520  * pointer to untyped u64 array, where each u64 can actually be a typed
521  * pointer or integer of different size. Instead of requring user to write
522  * manual casts and work with array elements by index, BPF_PROG macro
523  * allows user to declare a list of named and typed input arguments in the
524  * same syntax as for normal C function. All the casting is hidden and
525  * performed transparently, while user code can just assume working with
526  * function arguments of specified type and name.
527  *
528  * Original raw context argument is preserved as well as 'ctx' argument.
529  * This is useful when using BPF helpers that expect original context
530  * as one of the parameters (e.g., for bpf_perf_event_output()).
531  */
532 #define BPF_PROG(name, args...)                                             \
533 name(unsigned long long *ctx);                                              \
534 static __always_inline typeof(name(0))                                      \
535 ____##name(unsigned long long *ctx, ##args);                                \
536 typeof(name(0)) name(unsigned long long *ctx)                               \
537 {                                                                           \
538         _Pragma("GCC diagnostic push")                                      \
539         _Pragma("GCC diagnostic ignored \"-Wint-conversion\"")              \
540         return ____##name(___bpf_ctx_cast(args));                           \
541         _Pragma("GCC diagnostic pop")                                       \
542 }                                                                           \
543 static __always_inline typeof(name(0))                                      \
544 ____##name(unsigned long long *ctx, ##args)
545
546 #ifndef ___bpf_nth2
547 #define ___bpf_nth2(_, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13,  \
548                     _14, _15, _16, _17, _18, _19, _20, _21, _22, _23, _24, N, ...) N
549 #endif
550 #ifndef ___bpf_narg2
551 #define ___bpf_narg2(...)       \
552         ___bpf_nth2(_, ##__VA_ARGS__, 12, 12, 11, 11, 10, 10, 9, 9, 8, 8, 7, 7, \
553                     6, 6, 5, 5, 4, 4, 3, 3, 2, 2, 1, 1, 0)
554 #endif
555
556 #define ___bpf_treg_cnt(t) \
557         __builtin_choose_expr(sizeof(t) == 1, 1,        \
558         __builtin_choose_expr(sizeof(t) == 2, 1,        \
559         __builtin_choose_expr(sizeof(t) == 4, 1,        \
560         __builtin_choose_expr(sizeof(t) == 8, 1,        \
561         __builtin_choose_expr(sizeof(t) == 16, 2,       \
562                               (void)0)))))
563
564 #define ___bpf_reg_cnt0()               (0)
565 #define ___bpf_reg_cnt1(t, x)           (___bpf_reg_cnt0() + ___bpf_treg_cnt(t))
566 #define ___bpf_reg_cnt2(t, x, args...)  (___bpf_reg_cnt1(args) + ___bpf_treg_cnt(t))
567 #define ___bpf_reg_cnt3(t, x, args...)  (___bpf_reg_cnt2(args) + ___bpf_treg_cnt(t))
568 #define ___bpf_reg_cnt4(t, x, args...)  (___bpf_reg_cnt3(args) + ___bpf_treg_cnt(t))
569 #define ___bpf_reg_cnt5(t, x, args...)  (___bpf_reg_cnt4(args) + ___bpf_treg_cnt(t))
570 #define ___bpf_reg_cnt6(t, x, args...)  (___bpf_reg_cnt5(args) + ___bpf_treg_cnt(t))
571 #define ___bpf_reg_cnt7(t, x, args...)  (___bpf_reg_cnt6(args) + ___bpf_treg_cnt(t))
572 #define ___bpf_reg_cnt8(t, x, args...)  (___bpf_reg_cnt7(args) + ___bpf_treg_cnt(t))
573 #define ___bpf_reg_cnt9(t, x, args...)  (___bpf_reg_cnt8(args) + ___bpf_treg_cnt(t))
574 #define ___bpf_reg_cnt10(t, x, args...) (___bpf_reg_cnt9(args) + ___bpf_treg_cnt(t))
575 #define ___bpf_reg_cnt11(t, x, args...) (___bpf_reg_cnt10(args) + ___bpf_treg_cnt(t))
576 #define ___bpf_reg_cnt12(t, x, args...) (___bpf_reg_cnt11(args) + ___bpf_treg_cnt(t))
577 #define ___bpf_reg_cnt(args...)  ___bpf_apply(___bpf_reg_cnt, ___bpf_narg2(args))(args)
578
579 #define ___bpf_union_arg(t, x, n) \
580         __builtin_choose_expr(sizeof(t) == 1, ({ union { __u8 z[1]; t x; } ___t = { .z = {ctx[n]}}; ___t.x; }), \
581         __builtin_choose_expr(sizeof(t) == 2, ({ union { __u16 z[1]; t x; } ___t = { .z = {ctx[n]} }; ___t.x; }), \
582         __builtin_choose_expr(sizeof(t) == 4, ({ union { __u32 z[1]; t x; } ___t = { .z = {ctx[n]} }; ___t.x; }), \
583         __builtin_choose_expr(sizeof(t) == 8, ({ union { __u64 z[1]; t x; } ___t = {.z = {ctx[n]} }; ___t.x; }), \
584         __builtin_choose_expr(sizeof(t) == 16, ({ union { __u64 z[2]; t x; } ___t = {.z = {ctx[n], ctx[n + 1]} }; ___t.x; }), \
585                               (void)0)))))
586
587 #define ___bpf_ctx_arg0(n, args...)
588 #define ___bpf_ctx_arg1(n, t, x)                , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt1(t, x))
589 #define ___bpf_ctx_arg2(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt2(t, x, args)) ___bpf_ctx_arg1(n, args)
590 #define ___bpf_ctx_arg3(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt3(t, x, args)) ___bpf_ctx_arg2(n, args)
591 #define ___bpf_ctx_arg4(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt4(t, x, args)) ___bpf_ctx_arg3(n, args)
592 #define ___bpf_ctx_arg5(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt5(t, x, args)) ___bpf_ctx_arg4(n, args)
593 #define ___bpf_ctx_arg6(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt6(t, x, args)) ___bpf_ctx_arg5(n, args)
594 #define ___bpf_ctx_arg7(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt7(t, x, args)) ___bpf_ctx_arg6(n, args)
595 #define ___bpf_ctx_arg8(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt8(t, x, args)) ___bpf_ctx_arg7(n, args)
596 #define ___bpf_ctx_arg9(n, t, x, args...)       , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt9(t, x, args)) ___bpf_ctx_arg8(n, args)
597 #define ___bpf_ctx_arg10(n, t, x, args...)      , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt10(t, x, args)) ___bpf_ctx_arg9(n, args)
598 #define ___bpf_ctx_arg11(n, t, x, args...)      , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt11(t, x, args)) ___bpf_ctx_arg10(n, args)
599 #define ___bpf_ctx_arg12(n, t, x, args...)      , ___bpf_union_arg(t, x, n - ___bpf_reg_cnt12(t, x, args)) ___bpf_ctx_arg11(n, args)
600 #define ___bpf_ctx_arg(args...) ___bpf_apply(___bpf_ctx_arg, ___bpf_narg2(args))(___bpf_reg_cnt(args), args)
601
602 #define ___bpf_ctx_decl0()
603 #define ___bpf_ctx_decl1(t, x)                  , t x
604 #define ___bpf_ctx_decl2(t, x, args...)         , t x ___bpf_ctx_decl1(args)
605 #define ___bpf_ctx_decl3(t, x, args...)         , t x ___bpf_ctx_decl2(args)
606 #define ___bpf_ctx_decl4(t, x, args...)         , t x ___bpf_ctx_decl3(args)
607 #define ___bpf_ctx_decl5(t, x, args...)         , t x ___bpf_ctx_decl4(args)
608 #define ___bpf_ctx_decl6(t, x, args...)         , t x ___bpf_ctx_decl5(args)
609 #define ___bpf_ctx_decl7(t, x, args...)         , t x ___bpf_ctx_decl6(args)
610 #define ___bpf_ctx_decl8(t, x, args...)         , t x ___bpf_ctx_decl7(args)
611 #define ___bpf_ctx_decl9(t, x, args...)         , t x ___bpf_ctx_decl8(args)
612 #define ___bpf_ctx_decl10(t, x, args...)        , t x ___bpf_ctx_decl9(args)
613 #define ___bpf_ctx_decl11(t, x, args...)        , t x ___bpf_ctx_decl10(args)
614 #define ___bpf_ctx_decl12(t, x, args...)        , t x ___bpf_ctx_decl11(args)
615 #define ___bpf_ctx_decl(args...)        ___bpf_apply(___bpf_ctx_decl, ___bpf_narg2(args))(args)
616
617 /*
618  * BPF_PROG2 is an enhanced version of BPF_PROG in order to handle struct
619  * arguments. Since each struct argument might take one or two u64 values
620  * in the trampoline stack, argument type size is needed to place proper number
621  * of u64 values for each argument. Therefore, BPF_PROG2 has different
622  * syntax from BPF_PROG. For example, for the following BPF_PROG syntax:
623  *
624  *   int BPF_PROG(test2, int a, int b) { ... }
625  *
626  * the corresponding BPF_PROG2 syntax is:
627  *
628  *   int BPF_PROG2(test2, int, a, int, b) { ... }
629  *
630  * where type and the corresponding argument name are separated by comma.
631  *
632  * Use BPF_PROG2 macro if one of the arguments might be a struct/union larger
633  * than 8 bytes:
634  *
635  *   int BPF_PROG2(test_struct_arg, struct bpf_testmod_struct_arg_1, a, int, b,
636  *                 int, c, int, d, struct bpf_testmod_struct_arg_2, e, int, ret)
637  *   {
638  *        // access a, b, c, d, e, and ret directly
639  *        ...
640  *   }
641  */
642 #define BPF_PROG2(name, args...)                                                \
643 name(unsigned long long *ctx);                                                  \
644 static __always_inline typeof(name(0))                                          \
645 ____##name(unsigned long long *ctx ___bpf_ctx_decl(args));                      \
646 typeof(name(0)) name(unsigned long long *ctx)                                   \
647 {                                                                               \
648         return ____##name(ctx ___bpf_ctx_arg(args));                            \
649 }                                                                               \
650 static __always_inline typeof(name(0))                                          \
651 ____##name(unsigned long long *ctx ___bpf_ctx_decl(args))
652
653 struct pt_regs;
654
655 #define ___bpf_kprobe_args0()           ctx
656 #define ___bpf_kprobe_args1(x)          ___bpf_kprobe_args0(), (void *)PT_REGS_PARM1(ctx)
657 #define ___bpf_kprobe_args2(x, args...) ___bpf_kprobe_args1(args), (void *)PT_REGS_PARM2(ctx)
658 #define ___bpf_kprobe_args3(x, args...) ___bpf_kprobe_args2(args), (void *)PT_REGS_PARM3(ctx)
659 #define ___bpf_kprobe_args4(x, args...) ___bpf_kprobe_args3(args), (void *)PT_REGS_PARM4(ctx)
660 #define ___bpf_kprobe_args5(x, args...) ___bpf_kprobe_args4(args), (void *)PT_REGS_PARM5(ctx)
661 #define ___bpf_kprobe_args6(x, args...) ___bpf_kprobe_args5(args), (void *)PT_REGS_PARM6(ctx)
662 #define ___bpf_kprobe_args7(x, args...) ___bpf_kprobe_args6(args), (void *)PT_REGS_PARM7(ctx)
663 #define ___bpf_kprobe_args8(x, args...) ___bpf_kprobe_args7(args), (void *)PT_REGS_PARM8(ctx)
664 #define ___bpf_kprobe_args(args...)     ___bpf_apply(___bpf_kprobe_args, ___bpf_narg(args))(args)
665
666 /*
667  * BPF_KPROBE serves the same purpose for kprobes as BPF_PROG for
668  * tp_btf/fentry/fexit BPF programs. It hides the underlying platform-specific
669  * low-level way of getting kprobe input arguments from struct pt_regs, and
670  * provides a familiar typed and named function arguments syntax and
671  * semantics of accessing kprobe input paremeters.
672  *
673  * Original struct pt_regs* context is preserved as 'ctx' argument. This might
674  * be necessary when using BPF helpers like bpf_perf_event_output().
675  */
676 #define BPF_KPROBE(name, args...)                                           \
677 name(struct pt_regs *ctx);                                                  \
678 static __always_inline typeof(name(0))                                      \
679 ____##name(struct pt_regs *ctx, ##args);                                    \
680 typeof(name(0)) name(struct pt_regs *ctx)                                   \
681 {                                                                           \
682         _Pragma("GCC diagnostic push")                                      \
683         _Pragma("GCC diagnostic ignored \"-Wint-conversion\"")              \
684         return ____##name(___bpf_kprobe_args(args));                        \
685         _Pragma("GCC diagnostic pop")                                       \
686 }                                                                           \
687 static __always_inline typeof(name(0))                                      \
688 ____##name(struct pt_regs *ctx, ##args)
689
690 #define ___bpf_kretprobe_args0()       ctx
691 #define ___bpf_kretprobe_args1(x)      ___bpf_kretprobe_args0(), (void *)PT_REGS_RC(ctx)
692 #define ___bpf_kretprobe_args(args...) ___bpf_apply(___bpf_kretprobe_args, ___bpf_narg(args))(args)
693
694 /*
695  * BPF_KRETPROBE is similar to BPF_KPROBE, except, it only provides optional
696  * return value (in addition to `struct pt_regs *ctx`), but no input
697  * arguments, because they will be clobbered by the time probed function
698  * returns.
699  */
700 #define BPF_KRETPROBE(name, args...)                                        \
701 name(struct pt_regs *ctx);                                                  \
702 static __always_inline typeof(name(0))                                      \
703 ____##name(struct pt_regs *ctx, ##args);                                    \
704 typeof(name(0)) name(struct pt_regs *ctx)                                   \
705 {                                                                           \
706         _Pragma("GCC diagnostic push")                                      \
707         _Pragma("GCC diagnostic ignored \"-Wint-conversion\"")              \
708         return ____##name(___bpf_kretprobe_args(args));                     \
709         _Pragma("GCC diagnostic pop")                                       \
710 }                                                                           \
711 static __always_inline typeof(name(0)) ____##name(struct pt_regs *ctx, ##args)
712
713 /* If kernel has CONFIG_ARCH_HAS_SYSCALL_WRAPPER, read pt_regs directly */
714 #define ___bpf_syscall_args0()           ctx
715 #define ___bpf_syscall_args1(x)          ___bpf_syscall_args0(), (void *)PT_REGS_PARM1_SYSCALL(regs)
716 #define ___bpf_syscall_args2(x, args...) ___bpf_syscall_args1(args), (void *)PT_REGS_PARM2_SYSCALL(regs)
717 #define ___bpf_syscall_args3(x, args...) ___bpf_syscall_args2(args), (void *)PT_REGS_PARM3_SYSCALL(regs)
718 #define ___bpf_syscall_args4(x, args...) ___bpf_syscall_args3(args), (void *)PT_REGS_PARM4_SYSCALL(regs)
719 #define ___bpf_syscall_args5(x, args...) ___bpf_syscall_args4(args), (void *)PT_REGS_PARM5_SYSCALL(regs)
720 #define ___bpf_syscall_args(args...)     ___bpf_apply(___bpf_syscall_args, ___bpf_narg(args))(args)
721
722 /* If kernel doesn't have CONFIG_ARCH_HAS_SYSCALL_WRAPPER, we have to BPF_CORE_READ from pt_regs */
723 #define ___bpf_syswrap_args0()           ctx
724 #define ___bpf_syswrap_args1(x)          ___bpf_syswrap_args0(), (void *)PT_REGS_PARM1_CORE_SYSCALL(regs)
725 #define ___bpf_syswrap_args2(x, args...) ___bpf_syswrap_args1(args), (void *)PT_REGS_PARM2_CORE_SYSCALL(regs)
726 #define ___bpf_syswrap_args3(x, args...) ___bpf_syswrap_args2(args), (void *)PT_REGS_PARM3_CORE_SYSCALL(regs)
727 #define ___bpf_syswrap_args4(x, args...) ___bpf_syswrap_args3(args), (void *)PT_REGS_PARM4_CORE_SYSCALL(regs)
728 #define ___bpf_syswrap_args5(x, args...) ___bpf_syswrap_args4(args), (void *)PT_REGS_PARM5_CORE_SYSCALL(regs)
729 #define ___bpf_syswrap_args(args...)     ___bpf_apply(___bpf_syswrap_args, ___bpf_narg(args))(args)
730
731 /*
732  * BPF_KSYSCALL is a variant of BPF_KPROBE, which is intended for
733  * tracing syscall functions, like __x64_sys_close. It hides the underlying
734  * platform-specific low-level way of getting syscall input arguments from
735  * struct pt_regs, and provides a familiar typed and named function arguments
736  * syntax and semantics of accessing syscall input parameters.
737  *
738  * Original struct pt_regs * context is preserved as 'ctx' argument. This might
739  * be necessary when using BPF helpers like bpf_perf_event_output().
740  *
741  * At the moment BPF_KSYSCALL does not transparently handle all the calling
742  * convention quirks for the following syscalls:
743  *
744  * - mmap(): __ARCH_WANT_SYS_OLD_MMAP.
745  * - clone(): CONFIG_CLONE_BACKWARDS, CONFIG_CLONE_BACKWARDS2 and
746  *            CONFIG_CLONE_BACKWARDS3.
747  * - socket-related syscalls: __ARCH_WANT_SYS_SOCKETCALL.
748  * - compat syscalls.
749  *
750  * This may or may not change in the future. User needs to take extra measures
751  * to handle such quirks explicitly, if necessary.
752  *
753  * This macro relies on BPF CO-RE support and virtual __kconfig externs.
754  */
755 #define BPF_KSYSCALL(name, args...)                                         \
756 name(struct pt_regs *ctx);                                                  \
757 extern _Bool LINUX_HAS_SYSCALL_WRAPPER __kconfig;                           \
758 static __always_inline typeof(name(0))                                      \
759 ____##name(struct pt_regs *ctx, ##args);                                    \
760 typeof(name(0)) name(struct pt_regs *ctx)                                   \
761 {                                                                           \
762         struct pt_regs *regs = LINUX_HAS_SYSCALL_WRAPPER                    \
763                                ? (struct pt_regs *)PT_REGS_PARM1(ctx)       \
764                                : ctx;                                       \
765         _Pragma("GCC diagnostic push")                                      \
766         _Pragma("GCC diagnostic ignored \"-Wint-conversion\"")              \
767         if (LINUX_HAS_SYSCALL_WRAPPER)                                      \
768                 return ____##name(___bpf_syswrap_args(args));               \
769         else                                                                \
770                 return ____##name(___bpf_syscall_args(args));               \
771         _Pragma("GCC diagnostic pop")                                       \
772 }                                                                           \
773 static __always_inline typeof(name(0))                                      \
774 ____##name(struct pt_regs *ctx, ##args)
775
776 #define BPF_KPROBE_SYSCALL BPF_KSYSCALL
777
778 #endif