; 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 }
Codebase Browser
llvm