llvm/test/CodeGen/AArch64/setjmp-bti.ll

   1 ; RUN: llc -mtriple=aarch64-none-linux-gnu < %s | FileCheck %s --check-prefix=BTI
   2 ; RUN: llc -mtriple=aarch64-none-linux-gnu -global-isel < %s | FileCheck %s --check-prefix=BTI
   3 ; RUN: llc -mtriple=aarch64-none-linux-gnu -fast-isel < %s | FileCheck %s --check-prefix=BTI
   4 ; RUN: llc -mtriple=aarch64-none-linux-gnu -mattr=+harden-sls-blr< %s | FileCheck %s --check-prefix=BTISLS
   5 ; RUN: llc -mtriple=aarch64-none-linux-gnu -global-isel -mattr=+harden-sls-blr< %s | FileCheck %s --check-prefix=BTISLS
   6 ; RUN: llc -mtriple=aarch64-none-linux-gnu -fast-isel   -mattr=+harden-sls-blr< %s | FileCheck %s --check-prefix=BTISLS
   7 ; RUN: llc -mtriple=aarch64-none-linux-gnu -mattr=+no-bti-at-return-twice < %s | \
   8 ; RUN: FileCheck %s --check-prefix=NOBTI
   9 ; RUN: llc -mtriple=aarch64-none-linux-gnu -global-isel -mattr=+no-bti-at-return-twice < %s | \
  10 ; RUN: FileCheck %s --check-prefix=NOBTI
  11 ; RUN: llc -mtriple=aarch64-none-linux-gnu -fast-isel -mattr=+no-bti-at-return-twice < %s | \
  12 ; RUN: FileCheck %s --check-prefix=NOBTI
  13
  14 ; C source
  15 ; --------
  16 ; extern int setjmp(ptr);
  17 ; extern void notsetjmp(void);
  18 ;
  19 ; void bbb(void) {
  20 ;   setjmp(0);
  21 ;   setjmp(0); // With the attributes removed.
  22 ;   int (*fnptr)(ptr) = setjmp;
  23 ;   fnptr(0); // With attributes added.
  24 ;   notsetjmp();
  25 ; }
  26
  27 define void @bbb() #1 {
  28 ; BTI-LABEL: bbb:
  29 ; BTI:       bl setjmp
  30 ; BTI-NEXT:  hint #36
  31 ; BTI:       bl setjmp
  32 ; BTI-NEXT:  hint #36
  33 ; BTI:       blr x{{[0-9]+}}
  34 ; BTI-NEXT:  hint #36
  35 ; BTI:       bl notsetjmp
  36 ; BTI-NOT:   hint #36
  37
  38 ; BTISLS-LABEL: bbb:
  39 ; BTISLS:       bl setjmp
  40 ; BTISLS-NEXT:  hint #36
  41 ; BTISLS:       bl setjmp
  42 ; BTISLS-NEXT:  hint #36
  43 ; BTISLS:       bl __llvm_slsblr_thunk_x{{[0-9]+}}
  44 ; BTISLS-NEXT:  hint #36
  45 ; BTISLS:       bl notsetjmp
  46 ; BTISLS-NOT:   hint #36
  47
  48 ; NOBTI-LABEL: bbb:
  49 ; NOBTI:     bl setjmp
  50 ; NOBTI-NOT: hint #36
  51 ; NOBTI:     bl setjmp
  52 ; NOBTI-NOT: hint #36
  53 ; NOBTI:     blr x{{[0-9]+}}
  54 ; NOBTI-NOT: hint #36
  55 ; NOBTI:     bl notsetjmp
  56 ; NOBTI-NOT: hint #36
  57 entry:
  58   %fnptr = alloca ptr, align 8
  59   ; The frontend may apply attributes to the call, but it doesn't have to. We
  60   ; should be looking at the call base, which looks past that to the called function.
  61   %call = call i32 @setjmp(ptr noundef null) #0
  62   %call1 = call i32 @setjmp(ptr noundef null)
  63   store ptr @setjmp, ptr %fnptr, align 8
  64   %0 = load ptr, ptr %fnptr, align 8
  65   ; Clang does not attach the attribute here but if it did, it should work.
  66   %call2 = call i32 %0(ptr noundef null) #0
  67   call void @notsetjmp()
  68   ret void
  69 }
  70
  71 declare i32 @setjmp(ptr noundef) #0
  72 declare void @notsetjmp()
  73
  74 attributes #0 = { returns_twice }
  75 attributes #1 = { "branch-target-enforcement" }