2e37557fde
The IR stack protector pass should insert stack checks before the tail calls not only the musttail calls. So that the attributes `ssqreq` and `tail call`, which are emited by llvm-opt, could be both enabled by llvm-llc. Reviewed By: compnerd Differential Revision: https://reviews.llvm.org/D133860
106 lines
2.7 KiB
LLVM
106 lines
2.7 KiB
LLVM
; RUN: llc -mtriple=x86_64-linux-gnu -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() sspreq {
|
|
; CHECK-LABEL: define void @caller1()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackprotector
|
|
|
|
; CHECK: [[GUARD:%.*]] = load volatile i8*, i8*
|
|
; 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() sspreq {
|
|
; CHECK-LABEL: define void @justret()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackprotector
|
|
|
|
; CHECK: [[GUARD:%.*]] = load volatile i8*, i8*
|
|
; 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() sspreq {
|
|
; CHECK-LABEL: define i8* @caller2()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackprotector
|
|
|
|
; CHECK: [[GUARD:%.*]] = load volatile i8*, i8*
|
|
; 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
|
|
}
|
|
|
|
define void @caller3() sspreq {
|
|
; CHECK-LABEL: define void @caller3()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackprotector
|
|
|
|
; CHECK: [[GUARD:%.*]] = load volatile i8*, i8*
|
|
; CHECK: [[TOKEN:%.*]] = load volatile i8*, i8** {{%.*}}
|
|
; CHECK: [[TST:%.*]] = icmp eq i8* [[GUARD]], [[TOKEN]]
|
|
; CHECK: br i1 [[TST]]
|
|
|
|
; CHECK: tail call void @callee()
|
|
; CHECK-NEXT: ret void
|
|
%var = alloca [2 x i64]
|
|
store [2 x i64]* %var, [2 x i64]** @var
|
|
tail call void @callee()
|
|
ret void
|
|
}
|
|
|
|
define i8* @caller4() sspreq {
|
|
; CHECK-LABEL: define i8* @caller4()
|
|
; Prologue:
|
|
; CHECK: @llvm.stackprotector
|
|
|
|
; CHECK: [[GUARD:%.*]] = load volatile i8*, i8*
|
|
; CHECK: [[TOKEN:%.*]] = load volatile i8*, i8** {{%.*}}
|
|
; CHECK: [[TST:%.*]] = icmp eq i8* [[GUARD]], [[TOKEN]]
|
|
; CHECK: br i1 [[TST]]
|
|
|
|
; CHECK: [[TMP:%.*]] = tail 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 = tail call i64* @callee2()
|
|
%res = bitcast i64* %tmp to i8*
|
|
ret i8* %res
|
|
}
|