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

   1 ; RUN: llc -mtriple=aarch64-- -verify-machineinstrs < %s | FileCheck %s --check-prefix=ASM
   2 ; RUN: llc -mtriple=aarch64-- -verify-machineinstrs -stop-after=finalize-isel < %s | FileCheck %s --check-prefixes=MIR,ISEL
   3 ; RUN: llc -mtriple=aarch64-- -verify-machineinstrs -stop-after=kcfi < %s | FileCheck %s --check-prefixes=MIR,KCFI
   4
   5 ; ASM:       .word 12345678
   6 define void @f1(ptr noundef %x) !kcfi_type !2 {
   7 ; ASM-LABEL: f1:
   8 ; ASM:       // %bb.0:
   9 ; ASM:         ldur w16, [x0, #-4]
  10 ; ASM-NEXT:    movk w17, #24910
  11 ; ASM-NEXT:    movk w17, #188, lsl #16
  12 ; ASM-NEXT:    cmp w16, w17
  13 ; ASM-NEXT:    b.eq .Ltmp0
  14 ; ASM-NEXT:    brk #0x8220
  15 ; ASM-NEXT:  .Ltmp0:
  16 ; ASM-NEXT:    blr x0
  17
  18 ; MIR-LABEL: name: f1
  19 ; MIR: body:
  20
  21 ; ISEL: BLR %0, csr_aarch64_aapcs, implicit-def dead $lr, implicit $sp, implicit-def $sp, cfi-type 12345678
  22
  23 ; KCFI:       BUNDLE{{.*}} {
  24 ; KCFI-NEXT:    KCFI_CHECK $x0, 12345678, implicit-def $x9, implicit-def $x16, implicit-def $x17, implicit-def $nzcv
  25 ; KCFI-NEXT:    BLR killed $x0, csr_aarch64_aapcs, implicit-def dead $lr, implicit $sp, implicit-def $sp
  26 ; KCFI-NEXT:  }
  27
  28   call void %x() [ "kcfi"(i32 12345678) ]
  29   ret void
  30 }
  31
  32 ; ASM:       .word 12345678
  33 define void @f2(ptr noundef %x) !kcfi_type !2 {
  34 ; ASM-LABEL: f2:
  35 ; ASM:       // %bb.0:
  36 ; ASM:         ldur w16, [x0, #-4]
  37 ; ASM-NEXT:    movk w17, #24910
  38 ; ASM-NEXT:    movk w17, #188, lsl #16
  39 ; ASM-NEXT:    cmp w16, w17
  40 ; ASM-NEXT:    b.eq .Ltmp1
  41 ; ASM-NEXT:    brk #0x8220
  42 ; ASM-NEXT:  .Ltmp1:
  43 ; ASM-NEXT:    blr x0
  44
  45 ; MIR-LABEL: name: f2
  46 ; MIR: body:
  47
  48 ; ISEL: BLR_BTI %0, csr_aarch64_aapcs, implicit-def dead $lr, implicit $sp, implicit-def $sp, cfi-type 12345678
  49
  50 ; KCFI:       BUNDLE{{.*}} {
  51 ; KCFI-NEXT:    KCFI_CHECK $x0, 12345678, implicit-def $x9, implicit-def $x16, implicit-def $x17, implicit-def $nzcv
  52 ; KCFI-NEXT:    BLR killed $x0, implicit-def $lr, implicit $sp
  53 ; KCFI-NEXT:    HINT 36
  54 ; KCFI-NEXT:  }
  55
  56   call void %x() #0 [ "kcfi"(i32 12345678) ]
  57   ret void
  58 }
  59
  60 ; ASM-NOT: .word:
  61 define void @f3(ptr noundef %x) {
  62 ; ASM-LABEL: f3:
  63 ; ASM:       // %bb.0:
  64 ; ASM:         ldur w9, [x16, #-4]
  65 ; ASM-NEXT:    movk w17, #24910
  66 ; ASM-NEXT:    movk w17, #188, lsl #16
  67 ; ASM-NEXT:    cmp w9, w17
  68 ; ASM-NEXT:    b.eq .Ltmp2
  69 ; ASM-NEXT:    brk #0x8230
  70 ; ASM-NEXT:  .Ltmp2:
  71 ; ASM-NEXT:    br x16
  72
  73 ; MIR-LABEL: name: f3
  74 ; MIR: body:
  75
  76 ; ISEL: TCRETURNriBTI %1, 0, csr_aarch64_aapcs, implicit $sp, cfi-type 12345678
  77
  78 ; KCFI:       BUNDLE{{.*}} {
  79 ; KCFI-NEXT:    KCFI_CHECK $x16, 12345678, implicit-def $x9, implicit-def $x16, implicit-def $x17, implicit-def $nzcv
  80 ; KCFI-NEXT:    TCRETURNriBTI internal killed $x16, 0, csr_aarch64_aapcs, implicit $sp
  81 ; KCFI-NEXT:  }
  82
  83   tail call void %x() [ "kcfi"(i32 12345678) ]
  84   ret void
  85 }
  86
  87 attributes #0 = { returns_twice }
  88
  89 !llvm.module.flags = !{!0, !1}
  90 !0 = !{i32 8, !"branch-target-enforcement", i32 1}
  91 !1 = !{i32 4, !"kcfi", i32 1}
  92 !2 = !{i32 12345678}