1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _ASM_X86_CFI_H 3 #define _ASM_X86_CFI_H 4 5 /* 6 * Clang Control Flow Integrity (CFI) support. 7 * 8 * Copyright (C) 2022 Google LLC 9 */ 10 #include <linux/bug.h> 11 #include <asm/ibt.h> 12 13 /* 14 * An overview of the various calling conventions... 15 * 16 * Traditional: 17 * 18 * foo: 19 * ... code here ... 20 * ret 21 * 22 * direct caller: 23 * call foo 24 * 25 * indirect caller: 26 * lea foo(%rip), %r11 27 * ... 28 * call *%r11 29 * 30 * 31 * IBT: 32 * 33 * foo: 34 * endbr64 35 * ... code here ... 36 * ret 37 * 38 * direct caller: 39 * call foo / call foo+4 40 * 41 * indirect caller: 42 * lea foo(%rip), %r11 43 * ... 44 * call *%r11 45 * 46 * 47 * kCFI: 48 * 49 * __cfi_foo: 50 * movl $0x12345678, %eax 51 * # 11 nops when CONFIG_CALL_PADDING 52 * foo: 53 * endbr64 # when IBT 54 * ... code here ... 55 * ret 56 * 57 * direct call: 58 * call foo # / call foo+4 when IBT 59 * 60 * indirect call: 61 * lea foo(%rip), %r11 62 * ... 63 * movl $(-0x12345678), %r10d 64 * addl -4(%r11), %r10d # -15 when CONFIG_CALL_PADDING 65 * jz 1f 66 * ud2 67 * 1:call *%r11 68 * 69 * 70 * FineIBT (builds as kCFI + CALL_PADDING + IBT + RETPOLINE and runtime patches into): 71 * 72 * __cfi_foo: 73 * endbr64 74 * subl 0x12345678, %eax 75 * jne.32,pn foo+3 76 * foo: 77 * nopl -42(%rax) # was endbr64 78 * ... code here ... 79 * ret 80 * 81 * direct caller: 82 * call foo / call foo+4 83 * 84 * indirect caller: 85 * lea foo(%rip), %r11 86 * ... 87 * movl $0x12345678, %eax 88 * lea -0x10(%r11), %r11 89 * nop5 90 * call *%r11 91 * 92 */ 93 enum cfi_mode { 94 CFI_AUTO, /* FineIBT if hardware has IBT, otherwise kCFI */ 95 CFI_OFF, /* Taditional / IBT depending on .config */ 96 CFI_KCFI, /* Optionally CALL_PADDING, IBT, RETPOLINE */ 97 CFI_FINEIBT, /* see arch/x86/kernel/alternative.c */ 98 }; 99 100 extern enum cfi_mode cfi_mode; 101 102 #ifdef CONFIG_FINEIBT_BHI 103 extern bool cfi_bhi; 104 #else 105 #define cfi_bhi (0) 106 #endif 107 108 typedef u8 bhi_thunk[32]; 109 extern bhi_thunk __bhi_args[]; 110 extern bhi_thunk __bhi_args_end[]; 111 112 struct pt_regs; 113 114 #ifdef CONFIG_CFI 115 enum bug_trap_type handle_cfi_failure(struct pt_regs *regs); 116 #define __bpfcall 117 118 static inline int cfi_get_offset(void) 119 { 120 switch (cfi_mode) { 121 case CFI_FINEIBT: 122 return 16; 123 case CFI_KCFI: 124 if (IS_ENABLED(CONFIG_CALL_PADDING)) 125 return 16; 126 return 5; 127 default: 128 return 0; 129 } 130 } 131 #define cfi_get_offset cfi_get_offset 132 133 extern u32 cfi_get_func_hash(void *func); 134 #define cfi_get_func_hash cfi_get_func_hash 135 136 extern int cfi_get_func_arity(void *func); 137 138 #ifdef CONFIG_FINEIBT 139 extern bool decode_fineibt_insn(struct pt_regs *regs, unsigned long *target, u32 *type); 140 #else 141 static inline bool 142 decode_fineibt_insn(struct pt_regs *regs, unsigned long *target, u32 *type) 143 { 144 return false; 145 } 146 147 #endif 148 149 #else 150 static inline enum bug_trap_type handle_cfi_failure(struct pt_regs *regs) 151 { 152 return BUG_TRAP_TYPE_NONE; 153 } 154 static inline int cfi_get_func_arity(void *func) 155 { 156 return 0; 157 } 158 #endif /* CONFIG_CFI */ 159 160 #if HAS_KERNEL_IBT == 1 161 #define CFI_NOSEAL(x) asm(IBT_NOSEAL(__stringify(x))) 162 #endif 163 164 #endif /* _ASM_X86_CFI_H */ 165