1; Test hanlding of llvm.lifetime intrinsics.
2; RUN: opt < %s -asan -asan-module -asan-check-lifetime -asan-use-after-return=0 -S | FileCheck %s
3
4target 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"
5target triple = "x86_64-unknown-linux-gnu"
6
7declare void @llvm.lifetime.start(i64, i8* nocapture) nounwind
8declare void @llvm.lifetime.end(i64, i8* nocapture) nounwind
9
10define void @lifetime_no_size() sanitize_address {
11entry:
12  %i = alloca i32, align 4
13  %i.ptr = bitcast i32* %i to i8*
14  call void @llvm.lifetime.start(i64 -1, i8* %i.ptr)
15  call void @llvm.lifetime.end(i64 -1, i8* %i.ptr)
16
17; Check that lifetime with no size are ignored.
18; CHECK-LABEL: define void @lifetime_no_size()
19; CHECK-NOT: @__asan_poison_stack_memory
20; CHECK-NOT: @__asan_unpoison_stack_memory
21; CHECK: ret void
22  ret void
23}
24
25; Generic case of lifetime analysis.
26define void @lifetime() sanitize_address {
27  ; CHECK-LABEL: define void @lifetime()
28
29  ; Regular variable lifetime intrinsics.
30  %i = alloca i32, align 4
31  %i.ptr = bitcast i32* %i to i8*
32  call void @llvm.lifetime.start(i64 3, i8* %i.ptr)
33  ; Memory is unpoisoned at llvm.lifetime.start
34  ; CHECK: %[[VAR:[^ ]*]] = ptrtoint i32* %{{[^ ]+}} to i64
35  ; CHECK-NEXT: call void @__asan_unpoison_stack_memory(i64 %[[VAR]], i64 3)
36  call void @llvm.lifetime.end(i64 4, i8* %i.ptr)
37  call void @llvm.lifetime.end(i64 2, i8* %i.ptr)
38  ; Memory is poisoned at every call to llvm.lifetime.end
39  ; CHECK: call void @__asan_poison_stack_memory(i64 %{{[^ ]+}}, i64 4)
40  ; CHECK: call void @__asan_poison_stack_memory(i64 %{{[^ ]+}}, i64 2)
41
42  ; Lifetime intrinsics for array.
43  %arr = alloca [10 x i32], align 16
44  %arr.ptr = bitcast [10 x i32]* %arr to i8*
45  call void @llvm.lifetime.start(i64 40, i8* %arr.ptr)
46  ; CHECK: call void @__asan_unpoison_stack_memory(i64 %{{[^ ]+}}, i64 40)
47  call void @llvm.lifetime.end(i64 40, i8* %arr.ptr)
48  ; CHECK: call void @__asan_poison_stack_memory(i64 %{{[^ ]+}}, i64 40)
49
50  ; One more lifetime start/end for the same variable %i.
51  call void @llvm.lifetime.start(i64 4, i8* %i.ptr)
52  ; CHECK: call void @__asan_unpoison_stack_memory(i64 %{{[^ ]+}}, i64 4)
53  call void @llvm.lifetime.end(i64 4, i8* %i.ptr)
54  ; CHECK: call void @__asan_poison_stack_memory(i64 %{{[^ ]+}}, i64 4)
55
56  ; Memory is unpoisoned at function exit (only once).
57  ; CHECK: call void @__asan_unpoison_stack_memory(i64 %{{[^ ]+}}, i64 {{.*}})
58  ; CHECK-NOT: @__asan_unpoison_stack_memory
59  ; CHECK: ret void
60  ret void
61}
62
63; Check that arguments of lifetime may come from phi nodes.
64define void @phi_args(i1 %x) sanitize_address {
65  ; CHECK-LABEL: define void @phi_args(i1 %x)
66
67entry:
68  %i = alloca i64, align 4
69  %i.ptr = bitcast i64* %i to i8*
70  call void @llvm.lifetime.start(i64 8, i8* %i.ptr)
71  ; CHECK: __asan_unpoison_stack_memory
72  br i1 %x, label %bb0, label %bb1
73
74bb0:
75  %i.ptr2 = bitcast i64* %i to i8*
76  br label %bb1
77
78bb1:
79  %i.phi = phi i8* [ %i.ptr, %entry ], [ %i.ptr2, %bb0 ]
80  call void @llvm.lifetime.end(i64 8, i8* %i.phi)
81  ; CHECK: __asan_poison_stack_memory
82  ; CHECK: ret void
83  ret void
84}
85