llvm/test/CodeGen/X86/kcfi.ll

   1 ; RUN: llc -mtriple=x86_64-unknown-linux-gnu -verify-machineinstrs < %s | FileCheck %s --check-prefix=ASM
   2 ; RUN: llc -mtriple=x86_64-unknown-linux-gnu -verify-machineinstrs -stop-after=finalize-isel < %s | FileCheck %s --check-prefixes=MIR,ISEL
   3 ; RUN: llc -mtriple=x86_64-unknown-linux-gnu -verify-machineinstrs -stop-after=kcfi < %s | FileCheck %s --check-prefixes=MIR,KCFI
   4
   5 ; ASM:       .p2align 4
   6 ; ASM:       .type __cfi_f1,@function
   7 ; ASM-LABEL: __cfi_f1:
   8 ; ASM-NEXT:    nop
   9 ; ASM-NEXT:    nop
  10 ; ASM-NEXT:    nop
  11 ; ASM-NEXT:    nop
  12 ; ASM-NEXT:    nop
  13 ; ASM-NEXT:    nop
  14 ; ASM-NEXT:    nop
  15 ; ASM-NEXT:    nop
  16 ; ASM-NEXT:    nop
  17 ; ASM-NEXT:    nop
  18 ; ASM-NEXT:    nop
  19 ; ASM-NEXT:    movl $12345678, %eax
  20 ; ASM-LABEL: .Lcfi_func_end0:
  21 ; ASM-NEXT:  .size   __cfi_f1, .Lcfi_func_end0-__cfi_f1
  22 define void @f1(ptr noundef %x) !kcfi_type !1 {
  23 ; ASM-LABEL: f1:
  24 ; ASM:       # %bb.0:
  25 ; ASM:         movl $4282621618, %r10d # imm = 0xFF439EB2
  26 ; ASM-NEXT:    addl -4(%rdi), %r10d
  27 ; ASM-NEXT:    je .Ltmp0
  28 ; ASM-NEXT:  .Ltmp1:
  29 ; ASM-NEXT:    ud2
  30 ; ASM-NEXT:    .section .kcfi_traps,"ao",@progbits,.text
  31 ; ASM-NEXT:  .Ltmp2:
  32 ; ASM-NEXT:    .long .Ltmp1-.Ltmp2
  33 ; ASM-NEXT:    .text
  34 ; ASM-NEXT:  .Ltmp0:
  35 ; ASM-NEXT:    callq *%rdi
  36
  37 ; MIR-LABEL: name: f1
  38 ; MIR: body:
  39 ; ISEL: CALL64r %0, csr_64, implicit $rsp, implicit $ssp, implicit-def $rsp, implicit-def $ssp, cfi-type 12345678
  40 ; KCFI:       BUNDLE{{.*}} {
  41 ; KCFI-NEXT:    KCFI_CHECK $rdi, 12345678, implicit-def $r10, implicit-def $r11, implicit-def $eflags
  42 ; KCFI-NEXT:    CALL64r killed $rdi, csr_64, implicit $rsp, implicit $ssp, implicit-def $rsp, implicit-def $ssp
  43 ; KCFI-NEXT:  }
  44   call void %x() [ "kcfi"(i32 12345678) ]
  45   ret void
  46 }
  47
  48 ; ASM-NOT: __cfi_f2:
  49 define void @f2(ptr noundef %x) {
  50 ; ASM-LABEL: f2:
  51
  52 ; MIR-LABEL: name: f2
  53 ; MIR: body:
  54 ; ISEL: TCRETURNri64 %0, 0, csr_64, implicit $rsp, implicit $ssp, cfi-type 12345678
  55 ; KCFI:       BUNDLE{{.*}} {
  56 ; KCFI-NEXT:    KCFI_CHECK $rdi, 12345678, implicit-def $r10, implicit-def $r11, implicit-def $eflags
  57 ; KCFI-NEXT:    TAILJMPr64 killed $rdi, csr_64, implicit $rsp, implicit $ssp, implicit $rsp, implicit $ssp
  58 ; KCFI-NEXT:  }
  59   tail call void %x() [ "kcfi"(i32 12345678) ]
  60   ret void
  61 }
  62
  63 ; ASM-NOT: __cfi_f3:
  64 define void @f3(ptr noundef %x) #0 {
  65 ; ASM-LABEL: f3:
  66 ; MIR-LABEL: name: f3
  67 ; MIR: body:
  68 ; ISEL: CALL64pcrel32 &__llvm_retpoline_r11, csr_64, implicit $rsp, implicit $ssp, implicit-def $rsp, implicit-def $ssp, implicit killed $r11, cfi-type 12345678
  69 ; KCFI:       BUNDLE{{.*}} {
  70 ; KCFI-NEXT:    KCFI_CHECK $r11, 12345678, implicit-def $r10, implicit-def $r11, implicit-def $eflags
  71 ; KCFI-NEXT:    CALL64pcrel32 &__llvm_retpoline_r11, csr_64, implicit $rsp, implicit $ssp, implicit-def $rsp, implicit-def $ssp, implicit internal killed $r11
  72 ; KCFI-NEXT:  }
  73   call void %x() [ "kcfi"(i32 12345678) ]
  74   ret void
  75 }
  76
  77 ; ASM-NOT: __cfi_f4:
  78 define void @f4(ptr noundef %x) #0 {
  79 ; ASM-LABEL: f4:
  80 ; MIR-LABEL: name: f4
  81 ; MIR: body:
  82 ; ISEL: TCRETURNdi64 &__llvm_retpoline_r11, 0, csr_64, implicit $rsp, implicit $ssp, implicit killed $r11, cfi-type 12345678
  83 ; KCFI:       BUNDLE{{.*}} {
  84 ; KCFI-NEXT:    KCFI_CHECK $r11, 12345678, implicit-def $r10, implicit-def $r11, implicit-def $eflags
  85 ; KCFI-NEXT:    TAILJMPd64 &__llvm_retpoline_r11, csr_64, implicit $rsp, implicit $ssp, implicit $rsp, implicit $ssp, implicit internal killed $r11
  86 ; KCFI-NEXT:  }
  87   tail call void %x() [ "kcfi"(i32 12345678) ]
  88   ret void
  89 }
  90
  91 ;; Ensure we emit Value + 1 for unwanted values (e.g. endbr64 == 4196274163).
  92 ; ASM-LABEL: __cfi_f5:
  93 ; ASM: movl $4196274164, %eax # imm = 0xFA1E0FF4
  94 define void @f5(ptr noundef %x) !kcfi_type !2 {
  95 ; ASM-LABEL: f5:
  96 ; ASM: movl $98693132, %r10d # imm = 0x5E1F00C
  97   tail call void %x() [ "kcfi"(i32 4196274163) ]
  98   ret void
  99 }
 100
 101 ;; Ensure we emit Value + 1 for unwanted values (e.g. -endbr64 == 98693133).
 102 ; ASM-LABEL: __cfi_f6:
 103 ; ASM: movl $98693134, %eax # imm = 0x5E1F00E
 104 define void @f6(ptr noundef %x) !kcfi_type !3 {
 105 ; ASM-LABEL: f6:
 106 ; ASM: movl $4196274162, %r10d # imm = 0xFA1E0FF2
 107   tail call void %x() [ "kcfi"(i32 98693133) ]
 108   ret void
 109 }
 110
 111 @g = external local_unnamed_addr global ptr, align 8
 112
 113 define void @f7() {
 114 ; MIR-LABEL: name: f7
 115 ; MIR: body:
 116 ; ISEL: TCRETURNmi64 killed %0, 1, $noreg, 0, $noreg, 0, csr_64, implicit $rsp, implicit $ssp, cfi-type 12345678
 117 ; KCFI: $r11 = MOV64rm killed renamable $rax, 1, $noreg, 0, $noreg
 118 ; KCFI-NEXT:  BUNDLE{{.*}} {
 119 ; KCFI-NEXT:    KCFI_CHECK $r11, 12345678, implicit-def $r10, implicit-def $r11, implicit-def $eflags
 120 ; KCFI-NEXT:    TAILJMPr64 internal $r11, csr_64, implicit $rsp, implicit $ssp, implicit $rsp, implicit $ssp
 121 ; KCFI-NEXT:  }
 122   %1 = load ptr, ptr @g, align 8
 123   tail call void %1() [ "kcfi"(i32 12345678) ]
 124   ret void
 125 }
 126
 127 define void @f8() {
 128 ; MIR-LABEL: name: f8
 129 ; MIR: body:
 130 ; ISEL: CALL64m killed %0, 1, $noreg, 0, $noreg, csr_64, implicit $rsp, implicit $ssp, implicit-def $rsp, implicit-def $ssp, cfi-type 12345678
 131 ; KCFI: $r11 = MOV64rm killed renamable $rax, 1, $noreg, 0, $noreg
 132 ; KCFI-NEXT:  BUNDLE{{.*}} {
 133 ; KCFI-NEXT:    KCFI_CHECK $r11, 12345678, implicit-def $r10, implicit-def $r11, implicit-def $eflags
 134 ; KCFI-NEXT:    CALL64r internal $r11, csr_64, implicit $rsp, implicit $ssp, implicit-def $rsp, implicit-def $ssp
 135 ; KCFI-NEXT:  }
 136   %1 = load ptr, ptr @g, align 8
 137   call void %1() [ "kcfi"(i32 12345678) ]
 138   ret void
 139 }
 140
 141 attributes #0 = { "target-features"="+retpoline-indirect-branches,+retpoline-indirect-calls" }
 142
 143 !llvm.module.flags = !{!0}
 144 !0 = !{i32 4, !"kcfi", i32 1}
 145 !1 = !{i32 12345678}
 146 !2 = !{i32 4196274163}
 147 !3 = !{i32 98693133}