; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5

; RUN: llc -mtriple arm64e-apple-darwin \
; RUN:   -aarch64-enable-collect-loh=false \
; RUN:   -aarch64-min-jump-table-entries=1 -aarch64-enable-atomic-cfg-tidy=0 \
; RUN:   -o - %s | FileCheck %s --check-prefix=MACHO

; RUN: llc -mtriple arm64e-apple-darwin \
; RUN:   -fast-isel \
; RUN:   -aarch64-enable-collect-loh=false \
; RUN:   -aarch64-min-jump-table-entries=1 -aarch64-enable-atomic-cfg-tidy=0 \
; RUN:   -o - %s | FileCheck %s --check-prefix=MACHO

; RUN: llc -mtriple arm64e-apple-darwin \
; RUN:   -global-isel -global-isel-abort=1 -verify-machineinstrs \
; RUN:   -aarch64-enable-collect-loh=false \
; RUN:   -aarch64-min-jump-table-entries=1 -aarch64-enable-atomic-cfg-tidy=0 \
; RUN:   -o - %s | FileCheck %s --check-prefix=MACHO

; RUN: llc -mtriple aarch64-elf -mattr=+pauth \
; RUN:   -aarch64-min-jump-table-entries=1 -aarch64-enable-atomic-cfg-tidy=0 \
; RUN:   -o - %s | FileCheck %s --check-prefix=ELF

; RUN: llc -mtriple aarch64-elf -mattr=+pauth \
; RUN:   -fast-isel \
; RUN:   -aarch64-min-jump-table-entries=1 -aarch64-enable-atomic-cfg-tidy=0 \
; RUN:   -o - %s | FileCheck %s --check-prefix=ELF

; RUN: llc -mtriple aarch64-elf -mattr=+pauth \
; RUN:   -global-isel -global-isel-abort=1 -verify-machineinstrs \
; RUN:   -aarch64-min-jump-table-entries=1 -aarch64-enable-atomic-cfg-tidy=0 \
; RUN:   -o - %s | FileCheck %s --check-prefix=ELF

;; The discriminator is the same for all blockaddresses in the function.
;; ptrauth_string_discriminator("test_indirectbr blockaddress") == 34947

define i32 @test_indirectbr() #0 {
; MACHO-LABEL: test_indirectbr:
; MACHO:       ; %bb.0: ; %entry
; MACHO-NEXT:    stp x29, x30, [sp, #-16]! ; 16-byte Folded Spill
; MACHO-NEXT:    adrp x16, Ltmp0@PAGE
; MACHO-NEXT:    add x16, x16, Ltmp0@PAGEOFF
; MACHO-NEXT:    mov x17, #34947 ; =0x8883
; MACHO-NEXT:    pacia x16, x17
; MACHO-NEXT:    mov x0, x16
; MACHO-NEXT:    adrp x16, Ltmp1@PAGE
; MACHO-NEXT:    add x16, x16, Ltmp1@PAGEOFF
; MACHO-NEXT:    mov x17, #34947 ; =0x8883
; MACHO-NEXT:    pacia x16, x17
; MACHO-NEXT:    mov x1, x16
; MACHO-NEXT:    bl _dummy_choose
; MACHO-NEXT:    mov x17, #34947 ; =0x8883
; MACHO-NEXT:    braa x0, x17
; MACHO-NEXT:  Ltmp0: ; Block address taken
; MACHO-NEXT:  LBB0_1: ; %bb1
; MACHO-NEXT:    mov w0, #1 ; =0x1
; MACHO-NEXT:    ldp x29, x30, [sp], #16 ; 16-byte Folded Reload
; MACHO-NEXT:    ret
; MACHO-NEXT:  Ltmp1: ; Block address taken
; MACHO-NEXT:  LBB0_2: ; %bb2
; MACHO-NEXT:    mov w0, #2 ; =0x2
; MACHO-NEXT:    ldp x29, x30, [sp], #16 ; 16-byte Folded Reload
; MACHO-NEXT:    ret
;
; ELF-LABEL: test_indirectbr:
; ELF:       // %bb.0: // %entry
; ELF-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
; ELF-NEXT:    adrp x16, .Ltmp0
; ELF-NEXT:    add x16, x16, :lo12:.Ltmp0
; ELF-NEXT:    mov x17, #34947 // =0x8883
; ELF-NEXT:    pacia x16, x17
; ELF-NEXT:    mov x0, x16
; ELF-NEXT:    adrp x16, .Ltmp1
; ELF-NEXT:    add x16, x16, :lo12:.Ltmp1
; ELF-NEXT:    mov x17, #34947 // =0x8883
; ELF-NEXT:    pacia x16, x17
; ELF-NEXT:    mov x1, x16
; ELF-NEXT:    bl dummy_choose
; ELF-NEXT:    mov x17, #34947 // =0x8883
; ELF-NEXT:    braa x0, x17
; ELF-NEXT:  .Ltmp0: // Block address taken
; ELF-NEXT:  .LBB0_1: // %bb1
; ELF-NEXT:    mov w0, #1 // =0x1
; ELF-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
; ELF-NEXT:    ret
; ELF-NEXT:  .Ltmp1: // Block address taken
; ELF-NEXT:  .LBB0_2: // %bb2
; ELF-NEXT:    mov w0, #2 // =0x2
; ELF-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
; ELF-NEXT:    ret
entry:
  %tmp0 = call ptr @dummy_choose(ptr blockaddress(@test_indirectbr, %bb1), ptr blockaddress(@test_indirectbr, %bb2))
  indirectbr ptr %tmp0, [label %bb1, label %bb2]

bb1:
  ret i32 1

bb2:
  ret i32 2
}

define ptr @test_indirectbr_other_function() #0 {
; MACHO-LABEL: test_indirectbr_other_function:
; MACHO:       ; %bb.0:
; MACHO-NEXT:    adrp x16, Ltmp0@PAGE
; MACHO-NEXT:    add x16, x16, Ltmp0@PAGEOFF
; MACHO-NEXT:    mov x17, #34947 ; =0x8883
; MACHO-NEXT:    pacia x16, x17
; MACHO-NEXT:    mov x0, x16
; MACHO-NEXT:    ret
;
; ELF-LABEL: test_indirectbr_other_function:
; ELF:       // %bb.0:
; ELF-NEXT:    adrp x16, .Ltmp0
; ELF-NEXT:    add x16, x16, :lo12:.Ltmp0
; ELF-NEXT:    mov x17, #34947 // =0x8883
; ELF-NEXT:    pacia x16, x17
; ELF-NEXT:    mov x0, x16
; ELF-NEXT:    ret
  ret ptr blockaddress(@test_indirectbr, %bb1)
}

;; Test another function to compare the discriminator.
;; ptrauth_string_discriminator("test_indirectbr_2 blockaddress") == 40224

define i32 @test_indirectbr_2() #0 {
; MACHO-LABEL: test_indirectbr_2:
; MACHO:       ; %bb.0: ; %entry
; MACHO-NEXT:    stp x29, x30, [sp, #-16]! ; 16-byte Folded Spill
; MACHO-NEXT:    adrp x16, Ltmp2@PAGE
; MACHO-NEXT:    add x16, x16, Ltmp2@PAGEOFF
; MACHO-NEXT:    mov x17, #40224 ; =0x9d20
; MACHO-NEXT:    pacia x16, x17
; MACHO-NEXT:    mov x0, x16
; MACHO-NEXT:    adrp x16, Ltmp3@PAGE
; MACHO-NEXT:    add x16, x16, Ltmp3@PAGEOFF
; MACHO-NEXT:    mov x17, #40224 ; =0x9d20
; MACHO-NEXT:    pacia x16, x17
; MACHO-NEXT:    mov x1, x16
; MACHO-NEXT:    bl _dummy_choose
; MACHO-NEXT:    mov x17, #40224 ; =0x9d20
; MACHO-NEXT:    braa x0, x17
; MACHO-NEXT:  Ltmp2: ; Block address taken
; MACHO-NEXT:  LBB2_1: ; %bb1
; MACHO-NEXT:    mov w0, #1 ; =0x1
; MACHO-NEXT:    ldp x29, x30, [sp], #16 ; 16-byte Folded Reload
; MACHO-NEXT:    ret
; MACHO-NEXT:  Ltmp3: ; Block address taken
; MACHO-NEXT:  LBB2_2: ; %bb2
; MACHO-NEXT:    mov w0, #2 ; =0x2
; MACHO-NEXT:    ldp x29, x30, [sp], #16 ; 16-byte Folded Reload
; MACHO-NEXT:    ret
;
; ELF-LABEL: test_indirectbr_2:
; ELF:       // %bb.0: // %entry
; ELF-NEXT:    str x30, [sp, #-16]! // 8-byte Folded Spill
; ELF-NEXT:    adrp x16, .Ltmp2
; ELF-NEXT:    add x16, x16, :lo12:.Ltmp2
; ELF-NEXT:    mov x17, #40224 // =0x9d20
; ELF-NEXT:    pacia x16, x17
; ELF-NEXT:    mov x0, x16
; ELF-NEXT:    adrp x16, .Ltmp3
; ELF-NEXT:    add x16, x16, :lo12:.Ltmp3
; ELF-NEXT:    mov x17, #40224 // =0x9d20
; ELF-NEXT:    pacia x16, x17
; ELF-NEXT:    mov x1, x16
; ELF-NEXT:    bl dummy_choose
; ELF-NEXT:    mov x17, #40224 // =0x9d20
; ELF-NEXT:    braa x0, x17
; ELF-NEXT:  .Ltmp2: // Block address taken
; ELF-NEXT:  .LBB2_1: // %bb1
; ELF-NEXT:    mov w0, #1 // =0x1
; ELF-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
; ELF-NEXT:    ret
; ELF-NEXT:  .Ltmp3: // Block address taken
; ELF-NEXT:  .LBB2_2: // %bb2
; ELF-NEXT:    mov w0, #2 // =0x2
; ELF-NEXT:    ldr x30, [sp], #16 // 8-byte Folded Reload
; ELF-NEXT:    ret
entry:
  %tmp0 = call ptr @dummy_choose(ptr blockaddress(@test_indirectbr_2, %bb1), ptr blockaddress(@test_indirectbr_2, %bb2))
  indirectbr ptr %tmp0, [label %bb1, label %bb2]

bb1:
  ret i32 1

bb2:
  ret i32 2
}

;; Check we don't interfere with jump-table BRIND lowering.

; MACHO-LABEL: test_jumptable:
; MACHO:        adrp x9, LJTI3_0@PAGE
; MACHO-NEXT:   add x9, x9, LJTI3_0@PAGEOFF
; MACHO-NEXT:   adr x10, LBB3_2
; MACHO-NEXT:   ldrb w11, [x9, x8]
; MACHO-NEXT:   add x10, x10, x11, lsl #2
; MACHO-NEXT:   br x10

; ELF-LABEL: test_jumptable:
; ELF:        adrp x9, .LJTI3_0
; ELF-NEXT:   add x9, x9, :lo12:.LJTI3_0
; ELF-NEXT:   adr x10, .LBB3_2
; ELF-NEXT:   ldrb w11, [x9, x8]
; ELF-NEXT:   add x10, x10, x11, lsl #2
; ELF-NEXT:   br x10
define i32 @test_jumptable(i32 %in) #0 {
  switch i32 %in, label %def [
    i32 0, label %lbl1
    i32 1, label %lbl2
  ]

def:
  ret i32 0

lbl1:
  ret i32 1

lbl2:
  ret i32 2
}

; MACHO-LABEL: .globl _test_indirectbr_array
; MACHO-NEXT:  .p2align 4
; MACHO-NEXT:  _test_indirectbr_array:
; MACHO-NEXT:   .quad Ltmp0@AUTH(ia,34947)
; MACHO-NEXT:   .quad Ltmp1@AUTH(ia,34947)
; MACHO-NEXT:   .quad Ltmp2@AUTH(ia,40224)
; MACHO-NEXT:   .quad Ltmp3@AUTH(ia,40224)

; ELF-LABEL: .globl test_indirectbr_array
; ELF-NEXT:  .p2align 4, 0x0
; ELF-NEXT:  test_indirectbr_array:
; ELF-NEXT:   .xword .Ltmp0@AUTH(ia,34947)
; ELF-NEXT:   .xword .Ltmp1@AUTH(ia,34947)
; ELF-NEXT:   .xword .Ltmp2@AUTH(ia,40224)
; ELF-NEXT:   .xword .Ltmp3@AUTH(ia,40224)
; ELF-NEXT:   .size test_indirectbr_array, 32

@test_indirectbr_array = constant [4 x ptr] [
  ptr blockaddress(@test_indirectbr, %bb1), ptr blockaddress(@test_indirectbr, %bb2),
  ptr blockaddress(@test_indirectbr_2, %bb1), ptr blockaddress(@test_indirectbr_2, %bb2)
]

declare ptr @dummy_choose(ptr, ptr)

attributes #0 = { "ptrauth-indirect-gotos" nounwind }