llvm/test/CodeGen/X86/patchable-prologue.ll

   1 ; RUN: llc -verify-machineinstrs -filetype=obj -o - -mtriple=x86_64-apple-macosx < %s | llvm-objdump --no-print-imm-hex --triple=x86_64-apple-macosx -d - | FileCheck %s
   2 ; RUN: llc -verify-machineinstrs -mtriple=x86_64-apple-macosx < %s | FileCheck %s --check-prefix=CHECK-ALIGN
   3 ; RUN: llc -verify-machineinstrs -show-mc-encoding -mtriple=i386 < %s | FileCheck %s --check-prefixes=32,32CFI,XCHG
   4 ; RUN: llc -verify-machineinstrs -show-mc-encoding -mtriple=i386-windows-msvc < %s | FileCheck %s --check-prefixes=32,MOV
   5 ; RUN: llc -verify-machineinstrs -show-mc-encoding -mtriple=i386-windows-msvc -mcpu=pentium3 < %s | FileCheck %s --check-prefixes=32,MOV
   6 ; RUN: llc -verify-machineinstrs -show-mc-encoding -mtriple=i386-windows-msvc -mcpu=pentium4 < %s | FileCheck %s --check-prefixes=32,XCHG
   7 ; RUN: llc -verify-machineinstrs -show-mc-encoding -mtriple=x86_64-windows-msvc < %s | FileCheck %s --check-prefix=64
   8
   9 declare void @callee(ptr)
  10
  11 define void @f0() "patchable-function"="prologue-short-redirect" {
  12 ; CHECK-LABEL: _f0{{>?}}:
  13 ; CHECK-NEXT:  66 90    nop
  14
  15 ; CHECK-ALIGN:  .p2align        4, 0x90
  16 ; CHECK-ALIGN: _f0:
  17
  18 ; 32: f0:
  19 ; 32CFI-NEXT: .cfi_startproc
  20 ; 32-NEXT: # %bb.0:
  21 ; XCHG-NEXT: xchgw   %ax, %ax                # encoding: [0x66,0x90]
  22 ; MOV-NEXT: movl    %edi, %edi              # encoding: [0x8b,0xff]
  23 ; 32-NEXT: retl
  24
  25 ; 64: f0:
  26 ; 64-NEXT: # %bb.0:
  27 ; 64-NEXT: xchgw   %ax, %ax                # encoding: [0x66,0x90]
  28 ; 64-NEXT: retq
  29
  30   ret void
  31 }
  32
  33 define void @f1() "patchable-function"="prologue-short-redirect" "frame-pointer"="all" {
  34 ; CHECK-LABEL: _f1
  35 ; CHECK-NEXT: ff f5     pushq   %rbp
  36
  37 ; CHECK-ALIGN:  .p2align        4, 0x90
  38 ; CHECK-ALIGN: _f1:
  39
  40 ; 32: f1:
  41 ; 32CFI-NEXT: .cfi_startproc
  42 ; 32-NEXT: # %bb.0:
  43 ; XCHG-NEXT: xchgw   %ax, %ax                # encoding: [0x66,0x90]
  44 ; MOV-NEXT: movl    %edi, %edi              # encoding: [0x8b,0xff]
  45 ; 32-NEXT: pushl   %ebp
  46
  47 ; 64: f1:
  48 ; 64-NEXT: .seh_proc f1
  49 ; 64-NEXT: # %bb.0:
  50 ; 64-NEXT: pushq   %rbp
  51
  52   ret void
  53 }
  54
  55 define void @f2() "patchable-function"="prologue-short-redirect" {
  56 ; CHECK-LABEL: _f2
  57 ; CHECK-NEXT: 48 81 ec a8 00 00 00      subq    $168, %rsp
  58
  59 ; CHECK-ALIGN:  .p2align        4, 0x90
  60 ; CHECK-ALIGN: _f2:
  61
  62 ; 32: f2:
  63 ; 32CFI-NEXT: .cfi_startproc
  64 ; 32-NEXT: # %bb.0:
  65 ; XCHG-NEXT: xchgw   %ax, %ax                # encoding: [0x66,0x90]
  66 ; MOV-NEXT: movl    %edi, %edi              # encoding: [0x8b,0xff]
  67 ; 32-NEXT: pushl   %ebp
  68
  69 ; 64: f2:
  70 ; 64-NEXT: .seh_proc f2
  71 ; 64-NEXT: # %bb.0:
  72 ; 64-NEXT: subq    $200, %rsp
  73
  74   %ptr = alloca i64, i32 20
  75   call void @callee(ptr %ptr)
  76   ret void
  77 }
  78
  79 define void @f3() "patchable-function"="prologue-short-redirect" optsize {
  80 ; CHECK-LABEL: _f3
  81 ; CHECK-NEXT: 66 90     nop
  82
  83 ; CHECK-ALIGN:  .p2align        4, 0x90
  84 ; CHECK-ALIGN: _f3:
  85
  86 ; 32: f3:
  87 ; 32CFI-NEXT: .cfi_startproc
  88 ; 32-NEXT: # %bb.0:
  89 ; XCHG-NEXT: xchgw   %ax, %ax
  90 ; MOV-NEXT: movl   %edi, %edi
  91 ; 32-NEXT: retl
  92
  93 ; 64: f3:
  94 ; 64-NEXT: # %bb.0:
  95 ; 64-NEXT: xchgw   %ax, %ax
  96 ; 64-NEXT: retq
  97
  98   ret void
  99 }
 100
 101 ; This testcase happens to produce a KILL instruction at the beginning of the
 102 ; first basic block. In this case the 2nd instruction should be turned into a
 103 ; patchable one.
 104 ; CHECK-LABEL: f4{{>?}}:
 105 ; CHECK-NEXT: 8b 0c 37  movl  (%rdi,%rsi), %ecx
 106 ; 32: f4:
 107 ; 32CFI-NEXT: .cfi_startproc
 108 ; 32-NEXT: # %bb.0:
 109 ; XCHG-NEXT: xchgw   %ax, %ax
 110 ; MOV-NEXT: movl   %edi, %edi
 111 ; 32-NEXT: pushl   %ebx
 112
 113 ; 64: f4:
 114 ; 64-NEXT: # %bb.0:
 115 ; 64-NOT: xchgw   %ax, %ax
 116
 117 define i32 @f4(ptr %arg1, i64 %arg2, i32 %arg3) "patchable-function"="prologue-short-redirect" {
 118 bb:
 119   %tmp10 = getelementptr i8, ptr %arg1, i64 %arg2
 120   %tmp12 = load i32, ptr %tmp10, align 4
 121   fence acquire
 122   %tmp13 = add i32 %tmp12, %arg3
 123   %tmp14 = cmpxchg ptr %tmp10, i32 %tmp12, i32 %tmp13 seq_cst monotonic
 124   %tmp15 = extractvalue { i32, i1 } %tmp14, 1
 125   br i1 %tmp15, label %bb21, label %bb16
 126
 127 bb16:
 128   br label %bb21
 129
 130 bb21:
 131   %tmp22 = phi i32 [ %tmp12, %bb ], [ %arg3, %bb16 ]
 132   ret i32 %tmp22
 133 }
 134
 135 ; This testcase produces an empty function (not even a ret on some targets).
 136 ; This scenario can happen with undefined behavior.
 137 ; Ensure that the "patchable-function" pass supports this case.
 138 ; CHECK-LABEL: _emptyfunc
 139 ; CHECK-NEXT: 0f 0b     ud2
 140
 141 ; CHECK-ALIGN:  .p2align        4, 0x90
 142 ; CHECK-ALIGN: _emptyfunc:
 143
 144 ; 32: emptyfunc:
 145 ; 32CFI-NEXT: .cfi_startproc
 146 ; 32-NEXT: # %bb.0:
 147 ; XCHG-NEXT: xchgw   %ax, %ax
 148 ; MOV-NEXT: movl   %edi, %edi
 149
 150 ; 64: emptyfunc:
 151 ; 64-NEXT: # %bb.0:
 152 ; 64-NEXT: xchgw   %ax, %ax
 153
 154 ; From code: int emptyfunc() {}
 155 define i32 @emptyfunc() "patchable-function"="prologue-short-redirect" {
 156   unreachable
 157 }
 158
 159
 160 ; Hotpatch feature must ensure no jump within the function goes to the first instruction.
 161 ; From code:
 162 ; void jmp_to_start(char *b) {
 163 ;   do {
 164 ;   } while ((++(*b++)));
 165 ; }
 166
 167 ; CHECK-ALIGN:  .p2align        4, 0x90
 168 ; CHECK-ALIGN: _jmp_to_start:
 169
 170 ; 32: jmp_to_start:
 171 ; 32CFI-NEXT: .cfi_startproc
 172 ; 32-NEXT: # %bb.0:
 173 ; XCHG-NEXT: xchgw   %ax, %ax
 174 ; MOV-NEXT: movl   %edi, %edi
 175
 176 ; 64: jmp_to_start:
 177 ; 64-NEXT: # %bb.0:
 178 ; 64-NEXT: xchgw   %ax, %ax
 179
 180 define dso_local void @jmp_to_start(ptr inreg nocapture noundef %b) "patchable-function"="prologue-short-redirect" {
 181 entry:
 182   br label %do.body
 183 do.body:                                          ; preds = %do.body, %entry
 184   %b.addr.0 = phi ptr [ %b, %entry ], [ %incdec.ptr, %do.body ]
 185   %incdec.ptr = getelementptr inbounds i8, ptr %b.addr.0, i64 1
 186   %0 = load i8, ptr %b.addr.0, align 1
 187   %inc = add i8 %0, 1
 188   store i8 %inc, ptr %b.addr.0, align 1
 189   %tobool.not = icmp eq i8 %inc, 0
 190   br i1 %tobool.not, label %do.end, label %do.body
 191 do.end:                                           ; preds = %do.body
 192   ret void
 193 }