154 lines · plain
1; RUN: llc < %s -mtriple=arm-pc-windows-msvc | FileCheck %s -check-prefixes=CHECK,MSVC2; RUN: llc < %s -mtriple=arm-w64-windows-gnu | FileCheck %s -check-prefixes=CHECK,MINGW3; Control Flow Guard is currently only available on Windows4 5; Test that Control Flow Guard checks are correctly added when required.6 7 8declare i32 @target_func()9 10 11; Test that Control Flow Guard checks are not added on calls with the "guard_nocf" attribute.12define i32 @func_guard_nocf() #0 {13entry:14 %func_ptr = alloca ptr, align 815 store ptr @target_func, ptr %func_ptr, align 816 %0 = load ptr, ptr %func_ptr, align 817 %1 = call arm_aapcs_vfpcc i32 %0() #118 ret i32 %119 20 ; CHECK-LABEL: func_guard_nocf21 ; CHECK: movw r0, :lower16:target_func22 ; CHECK: movt r0, :upper16:target_func23 ; CHECK-NOT: __guard_check_icall_fptr24 ; CHECK: blx r025}26attributes #0 = { "target-cpu"="cortex-a9" "target-features"="+armv7-a,+dsp,+fp16,+neon,+strict-align,+thumb-mode,+vfp3"}27attributes #1 = { "guard_nocf" }28 29 30; Test that Control Flow Guard checks are added even at -O0.31define i32 @func_optnone_cf() #2 {32entry:33 %func_ptr = alloca ptr, align 834 store ptr @target_func, ptr %func_ptr, align 835 %0 = load ptr, ptr %func_ptr, align 836 %1 = call i32 %0()37 ret i32 %138 39 ; The call to __guard_check_icall_fptr should come immediately before the call to the target function.40 ; CHECK-LABEL: func_optnone_cf41 ; CHECK: movw r0, :lower16:target_func42 ; CHECK: movt r0, :upper16:target_func43 ; CHECK: str r0, [sp]44 ; CHECK: ldr r4, [sp]45 ; CHECK: movw r0, :lower16:__guard_check_icall_fptr46 ; CHECK: movt r0, :upper16:__guard_check_icall_fptr47 ; CHECK: ldr r1, [r0]48 ; CHECK: mov r0, r449 ; CHECK: blx r150 ; CHECK-NEXT: blx r451}52attributes #2 = { noinline optnone "target-cpu"="cortex-a9" "target-features"="+armv7-a,+dsp,+fp16,+neon,+strict-align,+thumb-mode,+vfp3"}53 54 55; Test that Control Flow Guard checks are correctly added in optimized code (common case).56define i32 @func_cf() #0 {57entry:58 %func_ptr = alloca ptr, align 859 store ptr @target_func, ptr %func_ptr, align 860 %0 = load ptr, ptr %func_ptr, align 861 %1 = call i32 %0()62 ret i32 %163 64 ; The call to __guard_check_icall_fptr should come immediately before the call to the target function.65 ; CHECK-LABEL: func_cf66 ; CHECK: movw r0, :lower16:__guard_check_icall_fptr67 ; CHECK: movt r0, :upper16:__guard_check_icall_fptr68 ; CHECK: ldr r1, [r0]69 ; CHECK: movw r4, :lower16:target_func70 ; CHECK: movt r4, :upper16:target_func71 ; CHECK: mov r0, r472 ; CHECK: blx r173 ; CHECK-NEXT: blx r474}75 76 77; Test that Control Flow Guard checks are correctly added on invoke instructions.78define i32 @func_cf_invoke() #0 personality ptr @h {79entry:80 %0 = alloca i32, align 481 %func_ptr = alloca ptr, align 882 store ptr @target_func, ptr %func_ptr, align 883 %1 = load ptr, ptr %func_ptr, align 884 %2 = invoke i32 %1()85 to label %invoke.cont unwind label %lpad86invoke.cont: ; preds = %entry87 ret i32 %288 89lpad: ; preds = %entry90 %tmp = landingpad { ptr, i32 }91 catch ptr null92 ret i32 -193 94 ; The call to __guard_check_icall_fptr should come immediately before the call to the target function.95 ; CHECK-LABEL: func_cf_invoke96 ; CHECK: movw r0, :lower16:__guard_check_icall_fptr97 ; CHECK: movt r0, :upper16:__guard_check_icall_fptr98 ; CHECK: ldr r1, [r0]99 ; CHECK: movw r4, :lower16:target_func100 ; CHECK: movt r4, :upper16:target_func101 ; CHECK: mov r0, r4102 ; CHECK: blx r1103 ; MSVC-NEXT: $Mtmp0:104 ; MINGW-NEXT: .Ltmp0:105 ; CHECK-NEXT: blx r4106 ; CHECK: @ %common.ret107 ; CHECK: @ %lpad108}109 110declare void @h()111 112 113; Test that longjmp targets have public labels and are included in the .gljmp section.114%struct._SETJMP_FLOAT128 = type { [2 x i64] }115@buf1 = internal global [16 x %struct._SETJMP_FLOAT128] zeroinitializer, align 16116 117define i32 @func_cf_setjmp() #0 {118 %1 = alloca i32, align 4119 %2 = alloca i32, align 4120 store i32 0, ptr %1, align 4121 store i32 -1, ptr %2, align 4122 %3 = call ptr @llvm.frameaddress(i32 0)123 %4 = call i32 @_setjmp(ptr @buf1, ptr %3) #3124 125 ; CHECK-LABEL: func_cf_setjmp126 ; CHECK: bl _setjmp127 ; CHECK-NEXT: $cfgsj_func_cf_setjmp0:128 129 %5 = call ptr @llvm.frameaddress(i32 0)130 %6 = call i32 @_setjmp(ptr @buf1, ptr %5) #3131 132 ; CHECK: bl _setjmp133 ; CHECK-NEXT: $cfgsj_func_cf_setjmp1:134 135 store i32 1, ptr %2, align 4136 %7 = load i32, ptr %2, align 4137 ret i32 %7138 139 ; CHECK: .section .gljmp$y,"dr"140 ; CHECK-NEXT: .symidx $cfgsj_func_cf_setjmp0141 ; CHECK-NEXT: .symidx $cfgsj_func_cf_setjmp1142}143 144declare ptr @llvm.frameaddress(i32)145 146; Function Attrs: returns_twice147declare dso_local i32 @_setjmp(ptr, ptr) #3148 149attributes #3 = { returns_twice }150 151 152!llvm.module.flags = !{!0}153!0 = !{i32 2, !"cfguard", i32 2}154