1 /* SPDX-License-Identifier: GPL-2.0 */ 2 /* 3 * GCC stack protector support. 4 * 5 * Stack protector works by putting predefined pattern at the start of 6 * the stack frame and verifying that it hasn't been overwritten when 7 * returning from the function. The pattern is called stack canary 8 * and unfortunately gcc requires it to be at a fixed offset from %gs. 9 * On x86_64, the offset is 40 bytes and on x86_32 20 bytes. x86_64 10 * and x86_32 use segment registers differently and thus handles this 11 * requirement differently. 12 * 13 * On x86_64, %gs is shared by percpu area and stack canary. All 14 * percpu symbols are zero based and %gs points to the base of percpu 15 * area. The first occupant of the percpu area is always 16 * fixed_percpu_data which contains stack_canary at offset 40. Userland 17 * %gs is always saved and restored on kernel entry and exit using 18 * swapgs, so stack protector doesn't add any complexity there. 19 * 20 * On x86_32, it's slightly more complicated. As in x86_64, %gs is 21 * used for userland TLS. Unfortunately, some processors are much 22 * slower at loading segment registers with different value when 23 * entering and leaving the kernel, so the kernel uses %fs for percpu 24 * area and manages %gs lazily so that %gs is switched only when 25 * necessary, usually during task switch. 26 * 27 * As gcc requires the stack canary at %gs:20, %gs can't be managed 28 * lazily if stack protector is enabled, so the kernel saves and 29 * restores userland %gs on kernel entry and exit. This behavior is 30 * controlled by CONFIG_X86_32_LAZY_GS and accessors are defined in 31 * system.h to hide the details. 32 */ 33 34 #ifndef _ASM_STACKPROTECTOR_H 35 #define _ASM_STACKPROTECTOR_H 1 36 37 #ifdef CONFIG_STACKPROTECTOR 38 39 #include <asm/tsc.h> 40 #include <asm/processor.h> 41 #include <asm/percpu.h> 42 #include <asm/desc.h> 43 44 #include <linux/random.h> 45 #include <linux/sched.h> 46 47 /* 48 * 24 byte read-only segment initializer for stack canary. Linker 49 * can't handle the address bit shifting. Address will be set in 50 * head_32 for boot CPU and setup_per_cpu_areas() for others. 51 */ 52 #define GDT_STACK_CANARY_INIT \ 53 [GDT_ENTRY_STACK_CANARY] = GDT_ENTRY_INIT(0x4090, 0, 0x18), 54 55 /* 56 * Initialize the stackprotector canary value. 57 * 58 * NOTE: this must only be called from functions that never return 59 * and it must always be inlined. 60 * 61 * In addition, it should be called from a compilation unit for which 62 * stack protector is disabled. Alternatively, the caller should not end 63 * with a function call which gets tail-call optimized as that would 64 * lead to checking a modified canary value. 65 */ 66 static __always_inline void boot_init_stack_canary(void) 67 { 68 u64 canary; 69 u64 tsc; 70 71 #ifdef CONFIG_X86_64 72 BUILD_BUG_ON(offsetof(struct fixed_percpu_data, stack_canary) != 40); 73 #endif 74 /* 75 * We both use the random pool and the current TSC as a source 76 * of randomness. The TSC only matters for very early init, 77 * there it already has some randomness on most systems. Later 78 * on during the bootup the random pool has true entropy too. 79 */ 80 get_random_bytes(&canary, sizeof(canary)); 81 tsc = rdtsc(); 82 canary += tsc + (tsc << 32UL); 83 canary &= CANARY_MASK; 84 85 current->stack_canary = canary; 86 #ifdef CONFIG_X86_64 87 this_cpu_write(fixed_percpu_data.stack_canary, canary); 88 #else 89 this_cpu_write(stack_canary.canary, canary); 90 #endif 91 } 92 93 static inline void cpu_init_stack_canary(int cpu, struct task_struct *idle) 94 { 95 #ifdef CONFIG_X86_64 96 per_cpu(fixed_percpu_data.stack_canary, cpu) = idle->stack_canary; 97 #else 98 per_cpu(stack_canary.canary, cpu) = idle->stack_canary; 99 #endif 100 } 101 102 static inline void setup_stack_canary_segment(int cpu) 103 { 104 #ifdef CONFIG_X86_32 105 unsigned long canary = (unsigned long)&per_cpu(stack_canary, cpu); 106 struct desc_struct *gdt_table = get_cpu_gdt_rw(cpu); 107 struct desc_struct desc; 108 109 desc = gdt_table[GDT_ENTRY_STACK_CANARY]; 110 set_desc_base(&desc, canary); 111 write_gdt_entry(gdt_table, GDT_ENTRY_STACK_CANARY, &desc, DESCTYPE_S); 112 #endif 113 } 114 115 static inline void load_stack_canary_segment(void) 116 { 117 #ifdef CONFIG_X86_32 118 asm("mov %0, %%gs" : : "r" (__KERNEL_STACK_CANARY) : "memory"); 119 #endif 120 } 121 122 #else /* STACKPROTECTOR */ 123 124 #define GDT_STACK_CANARY_INIT 125 126 /* dummy boot_init_stack_canary() is defined in linux/stackprotector.h */ 127 128 static inline void setup_stack_canary_segment(int cpu) 129 { } 130 131 static inline void cpu_init_stack_canary(int cpu, struct task_struct *idle) 132 { } 133 134 static inline void load_stack_canary_segment(void) 135 { 136 #ifdef CONFIG_X86_32 137 asm volatile ("mov %0, %%gs" : : "r" (0)); 138 #endif 139 } 140 141 #endif /* STACKPROTECTOR */ 142 #endif /* _ASM_STACKPROTECTOR_H */ 143