5e3d9fcc3a
The IR stack protector pass must insert stack checks before the call instead of between it and the return. Similarly, SDAG one should recognize that ADJCALLFRAME instructions could be part of the terminal sequence of a tail call. In this case because such call frames cannot be nested in LLVM the stack protection code must skip over the whole sequence (or risk clobbering argument registers).
67 lines
1.7 KiB
LLVM
67 lines
1.7 KiB
LLVM
; RUN: llc -mtriple=arm64-apple-macosx -fast-isel %s -o - -start-before=stack-protector -stop-after=stack-protector | FileCheck %s
|
|
|
|
@var = global [2 x i64]* null
|
|
|
|
declare void @callee()
|
|
|
|
define void @caller1() ssp {
|
|
; CHECK-LABEL: define void @caller1()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackguard
|
|
|
|
; CHECK: [[GUARD:%.*]] = call i8* @llvm.stackguard()
|
|
; CHECK: [[TOKEN:%.*]] = load volatile i8*, i8** {{%.*}}
|
|
; CHECK: [[TST:%.*]] = icmp eq i8* [[GUARD]], [[TOKEN]]
|
|
; CHECK: br i1 [[TST]]
|
|
|
|
; CHECK: musttail call void @callee()
|
|
; CHECK-NEXT: ret void
|
|
%var = alloca [2 x i64]
|
|
store [2 x i64]* %var, [2 x i64]** @var
|
|
musttail call void @callee()
|
|
ret void
|
|
}
|
|
|
|
define void @justret() ssp {
|
|
; CHECK-LABEL: define void @justret()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackguard
|
|
|
|
; CHECK: [[GUARD:%.*]] = call i8* @llvm.stackguard()
|
|
; CHECK: [[TOKEN:%.*]] = load volatile i8*, i8** {{%.*}}
|
|
; CHECK: [[TST:%.*]] = icmp eq i8* [[GUARD]], [[TOKEN]]
|
|
; CHECK: br i1 [[TST]]
|
|
|
|
; CHECK: ret void
|
|
%var = alloca [2 x i64]
|
|
store [2 x i64]* %var, [2 x i64]** @var
|
|
br label %retblock
|
|
|
|
retblock:
|
|
ret void
|
|
}
|
|
|
|
|
|
declare i64* @callee2()
|
|
|
|
define i8* @caller2() ssp {
|
|
; CHECK-LABEL: define i8* @caller2()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackguard
|
|
|
|
; CHECK: [[GUARD:%.*]] = call i8* @llvm.stackguard()
|
|
; CHECK: [[TOKEN:%.*]] = load volatile i8*, i8** {{%.*}}
|
|
; CHECK: [[TST:%.*]] = icmp eq i8* [[GUARD]], [[TOKEN]]
|
|
; CHECK: br i1 [[TST]]
|
|
|
|
; CHECK: [[TMP:%.*]] = musttail call i64* @callee2()
|
|
; CHECK-NEXT: [[RES:%.*]] = bitcast i64* [[TMP]] to i8*
|
|
; CHECK-NEXT: ret i8* [[RES]]
|
|
|
|
%var = alloca [2 x i64]
|
|
store [2 x i64]* %var, [2 x i64]** @var
|
|
%tmp = musttail call i64* @callee2()
|
|
%res = bitcast i64* %tmp to i8*
|
|
ret i8* %res
|
|
}
|