llvm/test/CodeGen/AArch64/sign-return-address.ll

; RUN: llc -mtriple=aarch64-none-eabi              < %s | FileCheck %s
; RUN: llc -mtriple=aarch64-none-eabi -mattr=v8.3a < %s | FileCheck --check-prefix CHECK-V83A %s

; CHECK-LABEL: @leaf
; CHECK-NOT: paci{{[a,b]}}sp
; CHECK-NOT: auti{{[a,b]}}sp
define i32 @leaf(i32 %x) {
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_none
; CHECK-NOT: paci{{[a,b]}}sp
; CHECK-NOT: auti{{[a,b]}}sp
define i32 @leaf_sign_none(i32 %x) "sign-return-address"="none"  {
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_non_leaf
; CHECK-NOT: paci{{[a,b]}}sp
; CHECK-NOT: auti{{[a,b]}}sp
define i32 @leaf_sign_non_leaf(i32 %x) "sign-return-address"="non-leaf"  {
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_all
; CHECK:       hint #25
; CHECK-NEXT:  .cfi_negate_ra_state
; CHECK:       hint #29
; CHECK-NEXT:  .cfi_negate_ra_state
; CHECK:       ret
; CHECK-V83A:  pacia x30, sp
; CHECK-V83A:  retaa
define i32 @leaf_sign_all(i32 %x) "sign-return-address"="all" {
  ret i32 %x
}

; CHECK:             @leaf_clobbers_lr
; CHECK:             hint #25
; CHECK-NEXT:        .cfi_negate_ra_state
; CHECK-V83A:        pacia x30, sp
; CHECK, CHECK-V83A: str x30, [sp, #-16]!
; CHECK, CHECK-V83A: ldr x30, [sp], #16
; CHECK:             hint #29
; CHECK-NEXT:        .cfi_negate_ra_state
; CHECK:             ret
; CHECK-V32A-NEXT:   retaa
define i64 @leaf_clobbers_lr(i64 %x) "sign-return-address"="non-leaf"  {
  call void asm sideeffect "mov x30, $0", "r,~{lr}"(i64 %x) #1
  ret i64 %x
}

declare i32 @foo(i32)

; CHECK:      @non_leaf_sign_all
; CHECK:      hint #25
; CHECK-NEXT: .cfi_negate_ra_state
; CHECK:      hint #29
; CHECK-NEXT: .cfi_negate_ra_state
; CHECK:      ret
; CHECK-V83A: pacia x30, sp
; CHECK-V83A: retaa
define i32 @non_leaf_sign_all(i32 %x) "sign-return-address"="all" {
  %call = call i32 @foo(i32 %x)
  ret i32 %call
}

; CHECK:             @non_leaf_sign_non_leaf
; CHECK:             hint #25
; CHECK-NEXT:        .cfi_negate_ra_state
; CHECK-V83A:        pacia x30, sp
; CHECK, CHECK-V83A: str x30, [sp, #-16]!
; CHECK, CHECK-V83A: ldr x30, [sp], #16
; CHECK:             hint #29
; CHECK-NEXT:        .cfi_negate_ra_state
; CHECK:             ret
; CHECK-V83A:        retaa
define i32 @non_leaf_sign_non_leaf(i32 %x) "sign-return-address"="non-leaf"  {
  %call = call i32 @foo(i32 %x)
  ret i32 %call
}

; CHECK-LABEL: @leaf_sign_all_v83
; CHECK:        pacia x30, sp
; CHECK-NEXT:  .cfi_negate_ra_state
; CHECK-NOT:   ret
; CHECK:       retaa
; CHECK-NOT:   ret
define i32 @leaf_sign_all_v83(i32 %x) "sign-return-address"="all" "target-features"="+v8.3a" {
  ret i32 %x
}

declare fastcc i64 @bar(i64)

; CHECK-LABEL:       @spill_lr_and_tail_call
; CHECK:             hint #25
; CHECK-NEXT:        .cfi_negate_ra_state
; CHECK-V83A:        pacia x30, sp
; CHECK-V83A-NEXT:  .cfi_negate_ra_state
; CHECK, CHECK-V83A: str x30, [sp, #-16]!
; CHECK, CHECK-V83A: ldr x30, [sp], #16
; CHECK-V83A:        autiasp
; CHECK:             hint #29
; CHECK-NEXT:        .cfi_negate_ra_state
; CHECK:             b bar
define fastcc void @spill_lr_and_tail_call(i64 %x) "sign-return-address"="all" {
  call void asm sideeffect "mov x30, $0", "r,~{lr}"(i64 %x) #1
  tail call fastcc i64 @bar(i64 %x)
  ret void
}

; CHECK-LABEL: @leaf_sign_all_a_key
; CHECK:            hint #25
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK:            hint #29
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK-V83A:       pacia x30, sp
; CHECK-V83A-NEXT:  .cfi_negate_ra_state
; CHECK-V83A:       retaa
define i32 @leaf_sign_all_a_key(i32 %x) "sign-return-address"="all" "sign-return-address-key"="a_key" {
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_all_b_key
; CHECK:            hint #27
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK:            hint #31
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK-V83A:       pacib x30, sp
; CHECK-V83A-NEXT:  .cfi_negate_ra_state
; CHECK-V83A:       retab
define i32 @leaf_sign_all_b_key(i32 %x) "sign-return-address"="all" "sign-return-address-key"="b_key" {
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_all_v83_b_key
; CHECK:       pacib x30, sp
; CHECK-NEXT:  .cfi_negate_ra_state
; CHECK-NOT:   ret
; CHECK:       retab
; CHECK-NOT:   ret
define i32 @leaf_sign_all_v83_b_key(i32 %x) "sign-return-address"="all" "target-features"="+v8.3a" "sign-return-address-key"="b_key" {
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_all_a_key_bti
; CHECK-NOT:        hint #34
; CHECK:            hint #25
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK:            hint #29
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK-V83A:       pacia x30, sp
; CHECK-V83A-NEXT:  .cfi_negate_ra_state
; CHECK-V83A:       retaa
define i32 @leaf_sign_all_a_key_bti(i32 %x) "sign-return-address"="all" "sign-return-address-key"="a_key" "branch-target-enforcement"="true"{
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_all_b_key_bti
; CHECK-NOT:        hint #34
; CHECK:            hint #27
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK:            hint #31
; CHECK-NEXT:       .cfi_negate_ra_state
; CHECK-V83A:       pacib x30, sp
; CHECK-V83A-NEXT:  .cfi_negate_ra_state
; CHECK-V83A:       retab
define i32 @leaf_sign_all_b_key_bti(i32 %x) "sign-return-address"="all" "sign-return-address-key"="b_key" "branch-target-enforcement"="true"{
  ret i32 %x
}

; CHECK-LABEL: @leaf_sign_all_v83_b_key_bti
; CHECK:       pacib x30, sp
; CHECK-NEXT:  .cfi_negate_ra_state
; CHECK-NOT:   ret
; CHECK:       retab
; CHECK-NOT:   ret
define i32 @leaf_sign_all_v83_b_key_bti(i32 %x) "sign-return-address"="all" "target-features"="+v8.3a" "sign-return-address-key"="b_key" "branch-target-enforcement"="true" {
  ret i32 %x
}