Home | History | Annotate | Line # | Download | only in hwasan
      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