llvm/llvm/test/CodeGen/AArch64/machine-outliner-retaddr-sign-regsave.mir

# RUN: llc -mtriple=aarch64 -run-pass=prologepilog -run-pass=machine-outliner \
# RUN: -verify-machineinstrs -run-pass=aarch64-ptrauth %s -o - | FileCheck %s

# Check that we save LR to a callee-saved register when possible.
# foo() should use a callee-saved register. However, bar() should not.
--- |

  define void @foo() #0 {
    ret void
  }

  define void @bar() #1 {
    ret void
  }

  attributes #0 = { nounwind         "sign-return-address"="non-leaf" "sign-return-address-key"="b_key" minsize noinline noredzone "frame-pointer"="all" }
  attributes #1 = { nounwind uwtable "sign-return-address"="non-leaf" "sign-return-address-key"="b_key" minsize noinline noredzone "frame-pointer"="all" }
...
---
# CHECK-LABEL:   name:            foo
# CHECK:         bb.0:
# CHECK:            frame-setup EMITBKEY
# CHECK-NEXT:       frame-setup PACIBSP
# CHECK-NOT:        frame-setup CFI_INSTRUCTION negate_ra_sign_state
# CHECK:        bb.1:
# CHECK:            BL @[[OUTLINED_FUNCTION:OUTLINED_FUNCTION_[0-9]+]]
# CHECK:        bb.2:
# CHECK:            BL @[[OUTLINED_FUNCTION]]
# CHECK:        bb.3:
# CHECK:            BL @[[OUTLINED_FUNCTION]]
# CHECK:        bb.4:
# CHECK:            BL @[[OUTLINED_FUNCTION]]
# CHECK:        bb.5:
# CHECK:            frame-destroy AUTIBSP
# CHECK-NEXT:       RET
name:            foo
tracksRegLiveness: true
fixedStack:
body:             |
  bb.0:
    $x25 = ORRXri $xzr, 1
    $lr = ORRXri $xzr, 1
  bb.1:
    liveins: $lr, $w9
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 2
  bb.2:
    liveins: $lr, $w9
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 2
  bb.3:
    liveins: $lr, $w9
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 2
  bb.4:
    liveins: $lr, $w9
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 1
    $w9 = ORRWri $wzr, 2
  bb.5:
    liveins: $w9
    RET undef $lr

...
---
# CHECK:         name:            bar
# CHECK:          bb.0:
# CHECK:            frame-setup EMITBKEY
# CHECK-NEXT:       frame-setup PACIBSP implicit-def $lr, implicit $lr, implicit $sp
# CHECK:            frame-setup CFI_INSTRUCTION negate_ra_sign_state
# CHECK-NEXT:       frame-setup CFI_INSTRUCTION
# CHECK-NOT:        OUTLINED_FUNCTION_
# CHECK:          bb.1:
# CHECK-NOT:        OUTLINED_FUNCTION_
# CHECK:          bb.2:
# CHECK-NOT:        OUTLINED_FUNCTION_
# CHECK:          bb.3:
# CHECK-NOT:        OUTLINED_FUNCTION_
# CHECK:            RET
name:            bar
tracksRegLiveness: true
body:             |
  bb.0:
    liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w12 = ORRWri $wzr, 2
  bb.1:
    liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w12 = ORRWri $wzr, 2
  bb.2:
    liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w10 = ORRWri $wzr, 1
    $w12 = ORRWri $wzr, 2
  bb.3:
    liveins: $lr, $x0, $x1, $x2, $x3, $x4, $x5, $x6, $x7, $x8, $x9, $x10, $x11, $x12, $x13, $x14, $x15, $x19, $x20, $x21, $x22, $x23, $x20, $x21, $x22, $x23, $x24, $x25, $x26, $x27, $x28
    RET undef $lr

# CHECK:         name:            [[OUTLINED_FUNCTION]]
# CHECK:         body:
# CHECK-NEXT:     bb.0:
# CHECK-NOT:        frame-setup EMITBKEY
# CHECK-NOT:        frame-setup PACI{{[A,B]}}SP
# CHECK-NOT:        frame-setup CFI_INSTRUCTION negate_ra_sign_state
# CHECK-NOT:        frame-destroy AUTI{{[A,B]}}SP