[safestack] Fix handling of array allocas.
[oota-llvm.git] / test / Transforms / SafeStack / struct.ll
1 ; RUN: opt -safe-stack -S -mtriple=i386-pc-linux-gnu < %s -o - | FileCheck %s
2 ; RUN: opt -safe-stack -S -mtriple=x86_64-pc-linux-gnu < %s -o - | FileCheck %s
3
4 %struct.foo = type { [16 x i8] }
5
6 @.str = private unnamed_addr constant [4 x i8] c"%s\0A\00", align 1
7
8 ; struct { [16 x i8] }
9
10 define void @foo(i8* %a) nounwind uwtable safestack {
11 entry:
12   ; CHECK: %[[USP:.*]] = load i8*, i8** @__safestack_unsafe_stack_ptr
13
14   ; CHECK: %[[USST:.*]] = getelementptr i8, i8* %[[USP]], i32 -16
15
16   ; CHECK: store i8* %[[USST]], i8** @__safestack_unsafe_stack_ptr
17
18   ; CHECK: %[[AADDR:.*]] = alloca i8*, align 8
19   %a.addr = alloca i8*, align 8
20
21   ; CHECK: %[[BUFPTR:.*]] = getelementptr i8, i8* %[[USP]], i32 -16
22   ; CHECK: %[[BUFPTR2:.*]] = bitcast i8* %[[BUFPTR]] to %struct.foo*
23   %buf = alloca %struct.foo, align 1
24
25   ; CHECK: store i8* {{.*}}, i8** %[[AADDR]], align 8
26   store i8* %a, i8** %a.addr, align 8
27
28   ; CHECK: %[[GEP:.*]] = getelementptr inbounds %struct.foo, %struct.foo* %[[BUFPTR2]], i32 0, i32 0, i32 0
29   %gep = getelementptr inbounds %struct.foo, %struct.foo* %buf, i32 0, i32 0, i32 0
30
31   ; CHECK: %[[A:.*]] = load i8*, i8** %[[AADDR]], align 8
32   %a2 = load i8*, i8** %a.addr, align 8
33
34   ; CHECK: call i8* @strcpy(i8* %[[GEP]], i8* %[[A]])
35   %call = call i8* @strcpy(i8* %gep, i8* %a2)
36
37   ; CHECK: store i8* %[[USP]], i8** @__safestack_unsafe_stack_ptr
38   ret void
39 }
40
41 declare i8* @strcpy(i8*, i8*)