1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
|
; 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 }
|
