xref: /llvm-project/llvm/test/CodeGen/X86/localescape.ll (revision 2f448bf509432c1a19ec46ab8cbc7353c03c6280)
1; RUN: llc -mtriple=x86_64-windows-msvc < %s | FileCheck %s --check-prefix=X64
2; RUN: llc -mtriple=i686-windows-msvc < %s | FileCheck %s --check-prefix=X86
3
4declare ptr @llvm.frameaddress(i32)
5declare void @llvm.localescape(...)
6declare ptr @llvm.localaddress()
7declare ptr @llvm.localrecover(ptr, ptr, i32)
8declare i32 @printf(ptr, ...)
9
10@str = internal constant [10 x i8] c"asdf: %d\0A\00"
11
12define void @print_framealloc_from_fp(ptr %fp) {
13  %a.i8 = call ptr @llvm.localrecover(ptr @alloc_func, ptr %fp, i32 0)
14  %a.val = load i32, ptr %a.i8
15  call i32 (ptr, ...) @printf(ptr @str, i32 %a.val)
16  %b.i8 = call ptr @llvm.localrecover(ptr @alloc_func, ptr %fp, i32 1)
17  %b.val = load i32, ptr %b.i8
18  call i32 (ptr, ...) @printf(ptr @str, i32 %b.val)
19  store i32 42, ptr %b.i8
20  %b2 = getelementptr i32, ptr %b.i8, i32 1
21  %b2.val = load i32, ptr %b2
22  call i32 (ptr, ...) @printf(ptr @str, i32 %b2.val)
23  ret void
24}
25
26; X64-LABEL: print_framealloc_from_fp:
27; X64: movq %rcx, %[[parent_fp:[a-z]+]]
28; X64: movl .Lalloc_func$frame_escape_0(%rcx), %edx
29; X64: leaq {{.*}}(%rip), %[[str:[a-z]+]]
30; X64: movq %[[str]], %rcx
31; X64: callq printf
32; X64: movl .Lalloc_func$frame_escape_1(%[[parent_fp]]), %edx
33; X64: movq %[[str]], %rcx
34; X64: callq printf
35; X64: movl    $42, .Lalloc_func$frame_escape_1(%[[parent_fp]])
36; X64: retq
37
38; X86-LABEL: print_framealloc_from_fp:
39; X86: pushl   %esi
40; X86: movl    8(%esp), %esi
41; X86: pushl   Lalloc_func$frame_escape_0(%esi)
42; X86: pushl   $_str
43; X86: calll   _printf
44; X86: addl    $8, %esp
45; X86: pushl   Lalloc_func$frame_escape_1(%esi)
46; X86: pushl   $_str
47; X86: calll   _printf
48; X86: addl    $8, %esp
49; X86: movl    $42, Lalloc_func$frame_escape_1(%esi)
50; X86: movl    $4, %eax
51; X86: pushl   Lalloc_func$frame_escape_1(%esi,%eax)
52; X86: pushl   $_str
53; X86: calll   _printf
54; X86: addl    $8, %esp
55; X86: popl    %esi
56; X86: retl
57
58define void @alloc_func(i32 %n) {
59  %a = alloca i32
60  %b = alloca i32, i32 2
61  call void (...) @llvm.localescape(ptr %a, ptr %b)
62  store i32 42, ptr %a
63  store i32 13, ptr %b
64
65  ; Force usage of EBP with a dynamic alloca.
66  alloca i8, i32 %n
67
68  %lp = call ptr @llvm.localaddress()
69  call void @print_framealloc_from_fp(ptr %lp)
70  ret void
71}
72
73; X64-LABEL: alloc_func:
74; X64: pushq   %rbp
75; X64: subq    $16, %rsp
76; X64: .seh_stackalloc 16
77; X64: leaq    16(%rsp), %rbp
78; X64: .seh_setframe %rbp, 16
79; X64: .set .Lalloc_func$frame_escape_0, -4
80; X64: .set .Lalloc_func$frame_escape_1, -12
81; X64: movl $42, -4(%rbp)
82; X64: movl $13, -12(%rbp)
83; X64: movq 	%rbp, %rcx
84; X64: callq print_framealloc_from_fp
85; X64: retq
86
87; X86-LABEL: alloc_func:
88; X86: pushl   %ebp
89; X86: movl    %esp, %ebp
90; X86: subl    $12, %esp
91; X86: .set Lalloc_func$frame_escape_0, -4
92; X86: .set Lalloc_func$frame_escape_1, -12
93; X86: movl    $42, -4(%ebp)
94; X86: movl    $13, -12(%ebp)
95; X86: pushl   %ebp
96; X86: calll   _print_framealloc_from_fp
97; X86: movl    %ebp, %esp
98; X86: popl    %ebp
99; X86: retl
100
101; Helper to make this a complete program so it can be compiled and tested.
102define i32 @main() {
103  call void @alloc_func(i32 3)
104  ret i32 0
105}
106
107define void @alloc_func_no_frameaddr() {
108  %a = alloca i32
109  %b = alloca i32
110  call void (...) @llvm.localescape(ptr %a, ptr %b)
111  store i32 42, ptr %a
112  store i32 13, ptr %b
113  call void @print_framealloc_from_fp(ptr null)
114  ret void
115}
116
117; X64-LABEL: alloc_func_no_frameaddr:
118; X64: subq    $40, %rsp
119; X64: .seh_stackalloc 40
120; X64: .seh_endprologue
121; X64: .set .Lalloc_func_no_frameaddr$frame_escape_0, 36
122; X64: .set .Lalloc_func_no_frameaddr$frame_escape_1, 32
123; X64: movl $42, 36(%rsp)
124; X64: movl $13, 32(%rsp)
125; X64: xorl %ecx, %ecx
126; X64: callq print_framealloc_from_fp
127; X64: addq $40, %rsp
128; X64: retq
129
130; X86-LABEL: alloc_func_no_frameaddr:
131; X86: subl    $8, %esp
132; X86: .set Lalloc_func_no_frameaddr$frame_escape_0, 4
133; X86: .set Lalloc_func_no_frameaddr$frame_escape_1, 0
134; X86: movl $42, 4(%esp)
135; X86: movl $13, (%esp)
136; X86: pushl $0
137; X86: calll _print_framealloc_from_fp
138; X86: addl    $12, %esp
139; X86: retl
140