| 1 | /* SPDX-License-Identifier: GPL-2.0 */ |
| 2 | #ifndef _ASM_STATIC_CALL_H |
| 3 | #define _ASM_STATIC_CALL_H |
| 4 | |
| 5 | #include <asm/text-patching.h> |
| 6 | |
| 7 | /* |
| 8 | * For CONFIG_HAVE_STATIC_CALL_INLINE, this is a temporary trampoline which |
| 9 | * uses the current value of the key->func pointer to do an indirect jump to |
| 10 | * the function. This trampoline is only used during boot, before the call |
| 11 | * sites get patched by static_call_update(). The name of this trampoline has |
| 12 | * a magical aspect: objtool uses it to find static call sites so it can create |
| 13 | * the .static_call_sites section. |
| 14 | * |
| 15 | * For CONFIG_HAVE_STATIC_CALL, this is a permanent trampoline which |
| 16 | * does a direct jump to the function. The direct jump gets patched by |
| 17 | * static_call_update(). |
| 18 | * |
| 19 | * Having the trampoline in a special section forces GCC to emit a JMP.d32 when |
| 20 | * it does tail-call optimization on the call; since you cannot compute the |
| 21 | * relative displacement across sections. |
| 22 | */ |
| 23 | |
| 24 | /* |
| 25 | * The trampoline is 8 bytes and of the general form: |
| 26 | * |
| 27 | * jmp.d32 \func |
| 28 | * ud1 %esp, %ecx |
| 29 | * |
| 30 | * That trailing #UD provides both a speculation stop and serves as a unique |
| 31 | * 3 byte signature identifying static call trampolines. Also see tramp_ud[] |
| 32 | * and __static_call_fixup(). |
| 33 | */ |
| 34 | #define __ARCH_DEFINE_STATIC_CALL_TRAMP(name, insns) \ |
| 35 | asm(".pushsection .static_call.text, \"ax\" \n" \ |
| 36 | ".align 4 \n" \ |
| 37 | ".globl " STATIC_CALL_TRAMP_STR(name) " \n" \ |
| 38 | STATIC_CALL_TRAMP_STR(name) ": \n" \ |
| 39 | ANNOTATE_NOENDBR " \n" \ |
| 40 | insns " \n" \ |
| 41 | ".byte 0x0f, 0xb9, 0xcc \n" \ |
| 42 | ".type " STATIC_CALL_TRAMP_STR(name) ", @function \n" \ |
| 43 | ".size " STATIC_CALL_TRAMP_STR(name) ", . - " STATIC_CALL_TRAMP_STR(name) " \n" \ |
| 44 | ".popsection \n") |
| 45 | |
| 46 | #define ARCH_DEFINE_STATIC_CALL_TRAMP(name, func) \ |
| 47 | __ARCH_DEFINE_STATIC_CALL_TRAMP(name, ".byte 0xe9; .long " #func " - (. + 4)") |
| 48 | |
| 49 | #ifdef CONFIG_MITIGATION_RETHUNK |
| 50 | #define ARCH_DEFINE_STATIC_CALL_NULL_TRAMP(name) \ |
| 51 | __ARCH_DEFINE_STATIC_CALL_TRAMP(name, "jmp __x86_return_thunk") |
| 52 | #else |
| 53 | #define ARCH_DEFINE_STATIC_CALL_NULL_TRAMP(name) \ |
| 54 | __ARCH_DEFINE_STATIC_CALL_TRAMP(name, "ret; int3; nop; nop; nop") |
| 55 | #endif |
| 56 | |
| 57 | #define ARCH_DEFINE_STATIC_CALL_RET0_TRAMP(name) \ |
| 58 | ARCH_DEFINE_STATIC_CALL_TRAMP(name, __static_call_return0) |
| 59 | |
| 60 | #define ARCH_ADD_TRAMP_KEY(name) \ |
| 61 | asm(".pushsection .static_call_tramp_key, \"a\" \n" \ |
| 62 | ".long " STATIC_CALL_TRAMP_STR(name) " - . \n" \ |
| 63 | ".long " STATIC_CALL_KEY_STR(name) " - . \n" \ |
| 64 | ".popsection \n") |
| 65 | |
| 66 | extern bool __static_call_fixup(void *tramp, u8 op, void *dest); |
| 67 | |
| 68 | extern void __static_call_update_early(void *tramp, void *func); |
| 69 | |
| 70 | #define static_call_update_early(name, _func) \ |
| 71 | ({ \ |
| 72 | typeof(&STATIC_CALL_TRAMP(name)) __F = (_func); \ |
| 73 | if (static_call_initialized) { \ |
| 74 | __static_call_update(&STATIC_CALL_KEY(name), \ |
| 75 | STATIC_CALL_TRAMP_ADDR(name), __F);\ |
| 76 | } else { \ |
| 77 | WRITE_ONCE(STATIC_CALL_KEY(name).func, _func); \ |
| 78 | __static_call_update_early(STATIC_CALL_TRAMP_ADDR(name),\ |
| 79 | __F); \ |
| 80 | } \ |
| 81 | }) |
| 82 | |
| 83 | #endif /* _ASM_STATIC_CALL_H */ |
| 84 | |