Skip to content

Commit 8996841

Browse files
committed
[CodeGen][RISCV] Add test for global stack canary
Make sure the code generated for a global stack guard canary references __stack_chk_guard before and after intervening function code. Signed-off-by: Keith Packard <[email protected]>
1 parent b096174 commit 8996841

File tree

1 file changed

+42
-0
lines changed

1 file changed

+42
-0
lines changed
Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
2+
; RUN: llc -mtriple=riscv64 -verify-machineinstrs < %s \
3+
; RUN: | FileCheck %s
4+
5+
define void @foo(i64 %t) sspstrong nounwind {
6+
; CHECK-LABEL: foo:
7+
; CHECK: # %bb.0:
8+
; CHECK-NEXT: addi sp, sp, -32
9+
; CHECK-NEXT: sd ra, 24(sp) # 8-byte Folded Spill
10+
; CHECK-NEXT: sd s0, 16(sp) # 8-byte Folded Spill
11+
; CHECK-NEXT: sd s1, 8(sp) # 8-byte Folded Spill
12+
; CHECK-NEXT: addi s0, sp, 32
13+
; CHECK-NEXT: lui s1, %hi(__stack_chk_guard)
14+
; CHECK-NEXT: ld a1, %lo(__stack_chk_guard)(s1)
15+
; CHECK-NEXT: sd a1, -32(s0)
16+
; CHECK-NEXT: slli a0, a0, 2
17+
; CHECK-NEXT: addi a0, a0, 15
18+
; CHECK-NEXT: andi a0, a0, -16
19+
; CHECK-NEXT: sub a0, sp, a0
20+
; CHECK-NEXT: mv sp, a0
21+
; CHECK-NEXT: call baz
22+
; CHECK-NEXT: ld a0, %lo(__stack_chk_guard)(s1)
23+
; CHECK-NEXT: ld a1, -32(s0)
24+
; CHECK-NEXT: bne a0, a1, .LBB0_2
25+
; CHECK-NEXT: # %bb.1:
26+
; CHECK-NEXT: addi sp, s0, -32
27+
; CHECK-NEXT: ld ra, 24(sp) # 8-byte Folded Reload
28+
; CHECK-NEXT: ld s0, 16(sp) # 8-byte Folded Reload
29+
; CHECK-NEXT: ld s1, 8(sp) # 8-byte Folded Reload
30+
; CHECK-NEXT: addi sp, sp, 32
31+
; CHECK-NEXT: ret
32+
; CHECK-NEXT: .LBB0_2:
33+
; CHECK-NEXT: call __stack_chk_fail
34+
%vla = alloca i32, i64 %t, align 4
35+
call void @baz(ptr %vla)
36+
ret void
37+
}
38+
39+
declare void @baz(ptr)
40+
41+
!llvm.module.flags = !{!1}
42+
!1 = !{i32 2, !"stack-protector-guard", !"global"}

0 commit comments

Comments
 (0)