llvm/test/CodeGen/ARM/setjmp-bti-basic.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
   2 ; RUN: llc -mtriple=thumbv8.1m.main-arm-none-eabi < %s | FileCheck %s --check-prefix=BTI
   3 ; RUN: llc -mtriple=thumbv8.1m.main-arm-none-eabi -mattr=+no-bti-at-return-twice < %s | \
   4 ; RUN: FileCheck %s --check-prefix=NOBTI
   5
   6 ; C source
   7 ; --------
   8 ; jmp_buf buf;
   9 ;
  10 ; extern void bar(int x);
  11 ;
  12 ; int foo(int x) {
  13 ;   if (setjmp(buf))
  14 ;     x = 0;
  15 ;   else
  16 ;     bar(x);
  17 ;   return x;
  18 ; }
  19
  20 @buf = global [20 x i64] zeroinitializer, align 8
  21
  22 define i32 @foo(i32 %x)  "branch-target-enforcement" {
  23 ; BTI-LABEL: foo:
  24 ; BTI:       @ %bb.0: @ %entry
  25 ; BTI-NEXT:    bti
  26 ; BTI-NEXT:    .save {r4, lr}
  27 ; BTI-NEXT:    push {r4, lr}
  28 ; BTI-NEXT:    mov r4, r0
  29 ; BTI-NEXT:    movw r0, :lower16:buf
  30 ; BTI-NEXT:    movt r0, :upper16:buf
  31 ; BTI-NEXT:    bl setjmp
  32 ; BTI-NEXT:    bti
  33 ; BTI-NEXT:    cmp r0, #0
  34 ; BTI-NEXT:    itt ne
  35 ; BTI-NEXT:    movne r0, #0
  36 ; BTI-NEXT:    popne {r4, pc}
  37 ; BTI-NEXT:  .LBB0_1: @ %if.else
  38 ; BTI-NEXT:    mov r0, r4
  39 ; BTI-NEXT:    bl bar
  40 ; BTI-NEXT:    mov r0, r4
  41 ; BTI-NEXT:    pop {r4, pc}
  42 ;
  43 ; NOBTI-LABEL: foo:
  44 ; NOBTI:       @ %bb.0: @ %entry
  45 ; NOBTI-NEXT:    bti
  46 ; NOBTI-NEXT:    .save {r4, lr}
  47 ; NOBTI-NEXT:    push {r4, lr}
  48 ; NOBTI-NEXT:    mov r4, r0
  49 ; NOBTI-NEXT:    movw r0, :lower16:buf
  50 ; NOBTI-NEXT:    movt r0, :upper16:buf
  51 ; NOBTI-NEXT:    bl setjmp
  52 ; NOBTI-NEXT:    cmp r0, #0
  53 ; NOBTI-NEXT:    itt ne
  54 ; NOBTI-NEXT:    movne r0, #0
  55 ; NOBTI-NEXT:    popne {r4, pc}
  56 ; NOBTI-NEXT:  .LBB0_1: @ %if.else
  57 ; NOBTI-NEXT:    mov r0, r4
  58 ; NOBTI-NEXT:    bl bar
  59 ; NOBTI-NEXT:    mov r0, r4
  60 ; NOBTI-NEXT:    pop {r4, pc}
  61
  62 entry:
  63   %call = call i32 @setjmp(ptr @buf) #0
  64   %tobool.not = icmp eq i32 %call, 0
  65   br i1 %tobool.not, label %if.else, label %if.end
  66
  67 if.else:                                          ; preds = %entry
  68   call void @bar(i32 %x)
  69   br label %if.end
  70
  71 if.end:                                           ; preds = %entry, %if.else
  72   %x.addr.0 = phi i32 [ %x, %if.else ], [ 0, %entry ]
  73   ret i32 %x.addr.0
  74 }
  75
  76 ;; Check that the BL to setjmp correctly clobbers LR
  77
  78 define i32 @baz() "branch-target-enforcement" {
  79 ; BTI-LABEL: baz:
  80 ; BTI:       @ %bb.0: @ %entry
  81 ; BTI-NEXT:    bti
  82 ; BTI-NEXT:    .save {r7, lr}
  83 ; BTI-NEXT:    push {r7, lr}
  84 ; BTI-NEXT:    .pad #160
  85 ; BTI-NEXT:    sub sp, #160
  86 ; BTI-NEXT:    mov r0, sp
  87 ; BTI-NEXT:    bl setjmp
  88 ; BTI-NEXT:    bti
  89 ; BTI-NEXT:    movs r0, #0
  90 ; BTI-NEXT:    add sp, #160
  91 ; BTI-NEXT:    pop {r7, pc}
  92 ;
  93 ; NOBTI-LABEL: baz:
  94 ; NOBTI:       @ %bb.0: @ %entry
  95 ; NOBTI-NEXT:    bti
  96 ; NOBTI-NEXT:    .save {r7, lr}
  97 ; NOBTI-NEXT:    push {r7, lr}
  98 ; NOBTI-NEXT:    .pad #160
  99 ; NOBTI-NEXT:    sub sp, #160
 100 ; NOBTI-NEXT:    mov r0, sp
 101 ; NOBTI-NEXT:    bl setjmp
 102 ; NOBTI-NEXT:    movs r0, #0
 103 ; NOBTI-NEXT:    add sp, #160
 104 ; NOBTI-NEXT:    pop {r7, pc}
 105 entry:
 106   %outgoing_jb = alloca [20 x i64], align 8
 107   %call = call i32 @setjmp(ptr %outgoing_jb) returns_twice
 108   ret i32 0
 109 }
 110
 111 declare void @bar(i32)
 112 declare i32 @setjmp(ptr) #0
 113
 114 attributes #0 = { returns_twice }
 115