test/CodeGen/AArch64/cfguard-module-flag.ll

; RUN: sed -e s/.tableonly:// %s | llc -mtriple=aarch64-pc-windows-msvc | FileCheck %s --check-prefixes=CHECK,TABLEONLY
; RUN: sed -e s/.tableonly:// %s | llc -mtriple=aarch64-w64-windows-gnu | FileCheck %s --check-prefixes=CHECK,TABLEONLY
; RUN: sed -e s/.normal:// %s | llc -mtriple=aarch64-pc-windows-msvc | FileCheck %s --check-prefixes=CHECK,USECHECK
; RUN: sed -e s/.normal:// %s | llc -mtriple=aarch64-w64-windows-gnu | FileCheck %s --check-prefixes=CHECK,USECHECK
; RUN: sed -e s/.normal:// %s | llc -mtriple=arm64ec-pc-windows-msvc 2>&1 | FileCheck %s --check-prefixes=CHECK,EC,NOECWARN
; RUN: sed -e s/.check:// %s | llc -mtriple=aarch64-pc-windows-msvc | FileCheck %s --check-prefixes=CHECK,USECHECK
; RUN: sed -e s/.check:// %s | llc -mtriple=arm64ec-pc-windows-msvc 2>&1 | FileCheck %s --check-prefixes=CHECK,EC,NOECWARN
; RUN: sed -e s/.dispatch:// %s | llc -mtriple=aarch64-pc-windows-msvc | FileCheck %s --check-prefixes=CHECK,USEDISPATCH
; RUN: sed -e s/.dispatch:// %s | llc -mtriple=arm64ec-pc-windows-msvc 2>&1 | FileCheck %s --check-prefixes=CHECK,EC,ECWARN
; Control Flow Guard is currently only available on Windows

; NOECWARN-NOT: warning:
; ECWARN: warning: only the Check Control Flow Guard mechanism is supported for Arm64EC

; If no checks were inserted then the GuardCF bit shouldn't be set in @feat.00.
; TABLEONLY:    "@feat.00" = 0
; USECHECK:     "@feat.00" = 2048
; USEDISPATCH:  "@feat.00" = 2048
; EC:           "@feat.00" = 2048

declare void @target_func()

define void @func_in_module_without_cfguard() #0 {
entry:
  %func_ptr = alloca ptr, align 8
  store ptr @target_func, ptr %func_ptr, align 8
  %0 = load ptr, ptr %func_ptr, align 8

  call void %0()
  ret void

  ; CHECK:            adrp

  ; USECHECK-SAME:    __guard_check_icall_fptr
  ; USECHECK-NOT:     __guard_dispatch_icall_fptr

  ; USEDISPATCH-SAME: __guard_dispatch_icall_fptr
  ; USEDISPATCH-NOT:  __guard_check_icall_fptr

  ; TABLEONLY-SAME:   target_func
  ; TABLEONLY-NOT:    __guard_dispatch_icall_fptr
  ; TABLEONLY-NOT:    __guard_check_icall_fptr

  ; Arm64EC Always uses check
  ; EC-SAME:    __os_arm64x_check_icall_cfg
  ; EC-NOT:     _dispatch_icall_
}

; CHECK: .section        .gfids$y,"dr"

!0 = !{i32 2, !"cfguard", i32 1}
!1 = !{i32 2, !"cfguard", i32 2}
!2 = !{i32 2, !"cfguard-mechanism", i32 1}
!3 = !{i32 2, !"cfguard-mechanism", i32 2}
;tableonly: !llvm.module.flags = !{!0}
;normal:    !llvm.module.flags = !{!1}
;check:     !llvm.module.flags = !{!1, !2}
;dispatch:  !llvm.module.flags = !{!1, !3}