llvm/llvm/test/CodeGen/AArch64/stack-protector-musttail.ll

; RUN: llc -mtriple=arm64-apple-macosx -fast-isel %s -o - -start-before=stack-protector -stop-after=stack-protector  | FileCheck %s

@var = global ptr null

declare void @callee()

define void @caller1() ssp {
; CHECK-LABEL: define void @caller1()
; Prologue:
; CHECK: @llvm.stackguard

; CHECK: [[GUARD:%.*]] = call ptr @llvm.stackguard()
; CHECK: [[TOKEN:%.*]] = load volatile ptr, ptr {{%.*}}
; CHECK: [[TST:%.*]] = icmp eq ptr [[GUARD]], [[TOKEN]]
; CHECK: br i1 [[TST]]

; CHECK: musttail call void @callee()
; CHECK-NEXT: ret void
  %var = alloca [2 x i64]
  store ptr %var, ptr @var
  musttail call void @callee()
  ret void
}

define void @justret() ssp {
; CHECK-LABEL: define void @justret()
; Prologue:
; CHECK: @llvm.stackguard

; CHECK: [[GUARD:%.*]] = call ptr @llvm.stackguard()
; CHECK: [[TOKEN:%.*]] = load volatile ptr, ptr {{%.*}}
; CHECK: [[TST:%.*]] = icmp eq ptr [[GUARD]], [[TOKEN]]
; CHECK: br i1 [[TST]]

; CHECK: ret void
  %var = alloca [2 x i64]
  store ptr %var, ptr @var
  br label %retblock

retblock:
  ret void
}


declare ptr @callee2()

define ptr @caller2() ssp {
; CHECK-LABEL: define ptr @caller2()
; Prologue:
; CHECK: @llvm.stackguard

; CHECK: [[GUARD:%.*]] = call ptr @llvm.stackguard()
; CHECK: [[TOKEN:%.*]] = load volatile ptr, ptr {{%.*}}
; CHECK: [[TST:%.*]] = icmp eq ptr [[GUARD]], [[TOKEN]]
; CHECK: br i1 [[TST]]

; CHECK: [[TMP:%.*]] = musttail call ptr @callee2()
; CHECK-NEXT: ret ptr [[TMP]]

  %var = alloca [2 x i64]
  store ptr %var, ptr @var
  %tmp = musttail call ptr @callee2()
  ret ptr %tmp
}

define void @caller3() ssp {
; CHECK-LABEL: define void @caller3()
; Prologue:
; CHECK: @llvm.stackguard

; CHECK: [[GUARD:%.*]] = call ptr @llvm.stackguard()
; CHECK: [[TOKEN:%.*]] = load volatile ptr, ptr {{%.*}}
; CHECK: [[TST:%.*]] = icmp eq ptr [[GUARD]], [[TOKEN]]
; CHECK: br i1 [[TST]]

; CHECK: tail call void @callee()
; CHECK-NEXT: ret void
  %var = alloca [2 x i64]
  store ptr %var, ptr @var
  tail call void @callee()
  ret void
}

define ptr @caller4() ssp {
; CHECK-LABEL: define ptr @caller4()
; Prologue:
; CHECK: @llvm.stackguard

; CHECK: [[GUARD:%.*]] = call ptr @llvm.stackguard()
; CHECK: [[TOKEN:%.*]] = load volatile ptr, ptr {{%.*}}
; CHECK: [[TST:%.*]] = icmp eq ptr [[GUARD]], [[TOKEN]]
; CHECK: br i1 [[TST]]

; CHECK: [[TMP:%.*]] = tail call ptr @callee2()
; CHECK-NEXT: ret ptr [[TMP]]

  %var = alloca [2 x i64]
  store ptr %var, ptr @var
  %tmp = tail call ptr @callee2()
  ret ptr %tmp
}