1; Test the ASan's stack layout.
2; More tests in tests/Transforms/Utils/ASanStackFrameLayoutTest.cpp
3; RUN: opt < %s -asan -asan-module -asan-stack-dynamic-alloca=0 -asan-use-after-scope -S \
4; RUN:     | FileCheck %s --check-prefixes=CHECK,CHECK-STATIC
5; RUN: opt < %s -asan -asan-module -asan-stack-dynamic-alloca=1 -asan-use-after-scope -S \
6; RUN:     | FileCheck %s --check-prefixes=CHECK,CHECK-DYNAMIC
7
8target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
9target triple = "x86_64-unknown-linux-gnu"
10
11declare void @Use(i8*)
12declare void @llvm.lifetime.start.p0i8(i64, i8* nocapture) nounwind
13declare void @llvm.lifetime.end.p0i8(i64, i8* nocapture) nounwind
14
15; CHECK: private unnamed_addr constant{{.*}}3 32 10 3 XXX 64 20 3 YYY 128 30 3 ZZZ\0
16; CHECK: private unnamed_addr constant{{.*}}3 32 5 3 AAA 64 55 3 BBB 160 555 3 CCC\0
17; CHECK: private unnamed_addr constant{{.*}}3 256 128 3 CCC 448 128 3 BBB 608 128 3 AAA\0
18; CHECK: private unnamed_addr constant{{.*}}2 32 4 3 AAA 48 4 5 BBB:7\0
19
20define void @Func1() sanitize_address {
21entry:
22; CHECK-LABEL: Func1
23
24; CHECK-STATIC: alloca [192 x i8]
25; CHECK-STATIC: %asan_local_stack_base = alloca i64
26; CHECK-DYNAMIC: alloca i8, i64 192
27
28; CHECK-NOT: alloca
29; CHECK: ret void
30  %XXX = alloca [10 x i8], align 1
31  %YYY = alloca [20 x i8], align 1
32  %ZZZ = alloca [30 x i8], align 1
33  %arr1.ptr = bitcast [10 x i8]* %XXX to i8*
34  store volatile i8 0, i8* %arr1.ptr
35  %arr2.ptr = bitcast [20 x i8]* %YYY to i8*
36  store volatile i8 0, i8* %arr2.ptr
37  %arr3.ptr = bitcast [30 x i8]* %ZZZ to i8*
38  store volatile i8 0, i8* %arr3.ptr
39  ret void
40}
41
42define void @Func2() sanitize_address {
43entry:
44; CHECK-LABEL: Func2
45
46; CHECK-STATIC: alloca [864 x i8]
47; CHECK-STATIC: %asan_local_stack_base = alloca i64
48; CHECK-DYNAMIC: alloca i8, i64 864
49
50; CHECK-NOT: alloca
51; CHECK: ret void
52  %AAA = alloca [5 x i8], align 1
53  %BBB = alloca [55 x i8], align 1
54  %CCC = alloca [555 x i8], align 1
55  %arr1.ptr = bitcast [5 x i8]* %AAA to i8*
56  store volatile i8 0, i8* %arr1.ptr
57  %arr2.ptr = bitcast [55 x i8]* %BBB to i8*
58  store volatile i8 0, i8* %arr2.ptr
59  %arr3.ptr = bitcast [555 x i8]* %CCC to i8*
60  store volatile i8 0, i8* %arr3.ptr
61  ret void
62}
63
64; Check that we reorder vars according to alignment and handle large alignments.
65define void @Func3() sanitize_address {
66entry:
67; CHECK-LABEL: Func3
68
69; CHECK-STATIC: alloca [768 x i8]
70; CHECK-STATIC: %asan_local_stack_base = alloca i64
71; CHECK-DYNAMIC: alloca i8, i64 768
72
73; CHECK-NOT: alloca
74; CHECK: ret void
75  %AAA = alloca [128 x i8], align 16
76  %BBB = alloca [128 x i8], align 64
77  %CCC = alloca [128 x i8], align 256
78  %arr1.ptr = bitcast [128 x i8]* %AAA to i8*
79  store volatile i8 0, i8* %arr1.ptr
80  %arr2.ptr = bitcast [128 x i8]* %BBB to i8*
81  store volatile i8 0, i8* %arr2.ptr
82  %arr3.ptr = bitcast [128 x i8]* %CCC to i8*
83  store volatile i8 0, i8* %arr3.ptr
84  ret void
85}
86
87; Check that line numbers are attached to variable names if variable
88; in the same file as a function.
89define void @Func5() sanitize_address #0 !dbg !11 {
90  %AAA = alloca i32, align 4  ; File is not the same as !11
91  %BBB = alloca i32, align 4  ; File is the same as !11
92  %BBB.ptr = bitcast i32* %BBB to i8*
93  call void @llvm.lifetime.start.p0i8(i64 4, i8* nonnull %BBB.ptr), !dbg !12
94  store volatile i32 5, i32* %BBB, align 4
95  %AAA.ptr = bitcast i32* %AAA to i8*
96  call void @llvm.lifetime.start.p0i8(i64 4, i8* nonnull %AAA.ptr), !dbg !14
97  store volatile i32 3, i32* %AAA, align 4
98  call void @llvm.lifetime.end.p0i8(i64 4, i8* nonnull %AAA.ptr), !dbg !17
99  call void @llvm.lifetime.end.p0i8(i64 4, i8* nonnull %BBB.ptr), !dbg !18
100  ret void
101}
102
103!llvm.dbg.cu = !{!0}
104!llvm.module.flags = !{!3, !4}
105!3 = !{i32 2, !"Dwarf Version", i32 4}
106!4 = !{i32 2, !"Debug Info Version", i32 3}
107
108!0 = distinct !DICompileUnit(language: DW_LANG_C99, file: !1)
109!1 = !DIFile(filename: "../file1.c", directory: "/")
110!11 = distinct !DISubprogram(name: "Func5", scope: !1, file: !1, line: 6, unit: !0)
111!12 = !DILocation(line: 7, column: 3, scope: !11)
112!18 = !DILocation(line: 10, column: 1, scope: !11)
113
114!21 = !DIFile(filename: "../file2.c", directory: "/")
115!6 = distinct !DISubprogram(name: "Func4", scope: !1, file: !21, line: 2, unit: !0)
116!15 = distinct !DILocation(line: 8, column: 3, scope: !11)
117!14 = !DILocation(line: 3, column: 3, scope: !6, inlinedAt: !15)
118!17 = !DILocation(line: 4, column: 1, scope: !6, inlinedAt: !15)
119