1/* SPDX-License-Identifier: GPL-2.0-only */
2/*
3 * Trampoline to trace irqs off. (otherwise CALLER_ADDR1 might crash)
4 * Copyright 2008 by Steven Rostedt, Red Hat, Inc
5 *  (inspired by Andi Kleen's thunk_64.S)
6 */
7	#include <linux/linkage.h>
8	#include <asm/asm.h>
9	#include <asm/export.h>
10
11	/* put return address in eax (arg1) */
12	.macro THUNK name, func, put_ret_addr_in_eax=0
13SYM_CODE_START_NOALIGN(\name)
14	pushl %eax
15	pushl %ecx
16	pushl %edx
17
18	.if \put_ret_addr_in_eax
19	/* Place EIP in the arg1 */
20	movl 3*4(%esp), %eax
21	.endif
22
23	call \func
24	popl %edx
25	popl %ecx
26	popl %eax
27	ret
28	_ASM_NOKPROBE(\name)
29SYM_CODE_END(\name)
30	.endm
31
32#ifdef CONFIG_PREEMPTION
33	THUNK preempt_schedule_thunk, preempt_schedule
34	THUNK preempt_schedule_notrace_thunk, preempt_schedule_notrace
35	EXPORT_SYMBOL(preempt_schedule_thunk)
36	EXPORT_SYMBOL(preempt_schedule_notrace_thunk)
37#endif
38
39