1 1.1 mrg //===-- hwasan_setjmp_x86_64.S --------------------------------------------===// 2 1.1 mrg // 3 1.1 mrg // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 1.1 mrg // See https://llvm.org/LICENSE.txt for license information. 5 1.1 mrg // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 1.1 mrg // 7 1.1 mrg //===----------------------------------------------------------------------===// 8 1.1 mrg // 9 1.1 mrg // setjmp interceptor for x86_64. 10 1.1 mrg // 11 1.1 mrg //===----------------------------------------------------------------------===// 12 1.1 mrg 13 1.1 mrg #include "sanitizer_common/sanitizer_asm.h" 14 1.1 mrg 15 1.1 mrg #if HWASAN_WITH_INTERCEPTORS && defined(__x86_64__) 16 1.1 mrg #include "sanitizer_common/sanitizer_platform.h" 17 1.1 mrg 18 1.1 mrg // We want to save the context of the calling function. 19 1.1 mrg // That requires 20 1.1 mrg // 1) No modification of the return address by this function. 21 1.1 mrg // 2) No modification of the stack pointer by this function. 22 1.1 mrg // 3) (no modification of any other saved register, but that's not really going 23 1.1 mrg // to occur, and hence isn't as much of a worry). 24 1.1 mrg // 25 1.1 mrg // There's essentially no way to ensure that the compiler will not modify the 26 1.1 mrg // stack pointer when compiling a C function. 27 1.1 mrg // Hence we have to write this function in assembly. 28 1.1 mrg // 29 1.1 mrg // TODO: Handle Intel CET. 30 1.1 mrg 31 1.1 mrg .section .text 32 1.1 mrg .file "hwasan_setjmp_x86_64.S" 33 1.1 mrg 34 1.1.1.2 mrg .global ASM_WRAPPER_NAME(setjmp) 35 1.1.1.2 mrg ASM_TYPE_FUNCTION(ASM_WRAPPER_NAME(setjmp)) 36 1.1.1.2 mrg ASM_WRAPPER_NAME(setjmp): 37 1.1 mrg CFI_STARTPROC 38 1.1 mrg _CET_ENDBR 39 1.1 mrg xorl %esi, %esi 40 1.1.1.2 mrg jmp .Linterceptor_sigsetjmp 41 1.1 mrg CFI_ENDPROC 42 1.1.1.2 mrg ASM_SIZE(ASM_WRAPPER_NAME(setjmp)) 43 1.1 mrg 44 1.1.1.2 mrg .global ASM_WRAPPER_NAME(sigsetjmp) 45 1.1.1.2 mrg ASM_TYPE_FUNCTION(ASM_WRAPPER_NAME(sigsetjmp)) 46 1.1.1.2 mrg ASM_WRAPPER_NAME(sigsetjmp): 47 1.1.1.2 mrg .Linterceptor_sigsetjmp: 48 1.1 mrg CFI_STARTPROC 49 1.1 mrg _CET_ENDBR 50 1.1 mrg 51 1.1 mrg // Save callee save registers. 52 1.1 mrg mov %rbx, (0*8)(%rdi) 53 1.1 mrg mov %rbp, (1*8)(%rdi) 54 1.1 mrg mov %r12, (2*8)(%rdi) 55 1.1 mrg mov %r13, (3*8)(%rdi) 56 1.1 mrg mov %r14, (4*8)(%rdi) 57 1.1 mrg mov %r15, (5*8)(%rdi) 58 1.1 mrg 59 1.1 mrg // Save SP as it was in caller's frame. 60 1.1 mrg lea 8(%rsp), %rdx 61 1.1 mrg mov %rdx, (6*8)(%rdi) 62 1.1 mrg 63 1.1 mrg // Save return address. 64 1.1 mrg mov (%rsp), %rax 65 1.1 mrg mov %rax, (7*8)(%rdi) 66 1.1 mrg 67 1.1 mrg jmp __sigjmp_save 68 1.1 mrg 69 1.1 mrg CFI_ENDPROC 70 1.1.1.2 mrg ASM_SIZE(ASM_WRAPPER_NAME(sigsetjmp)) 71 1.1 mrg 72 1.1.1.2 mrg ASM_INTERCEPTOR_TRAMPOLINE(sigsetjmp) 73 1.1.1.2 mrg ASM_TRAMPOLINE_ALIAS(__sigsetjmp, sigsetjmp) 74 1.1.1.2 mrg ASM_INTERCEPTOR_TRAMPOLINE(setjmp) 75 1.1.1.2 mrg ASM_TRAMPOLINE_ALIAS(_setjmp, setjmp) 76 1.1 mrg #endif 77 1.1 mrg 78 1.1 mrg // We do not need executable stack. 79 1.1 mrg NO_EXEC_STACK_DIRECTIVE 80