| Peter Collingbourne | 82437bf | 2015-06-15 21:07:11 +0000 | [diff] [blame] | 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 | ; array [4 x i8] |
| 5 | ; Requires protector. |
| 6 | |
| 7 | define void @foo(i8* %a) nounwind uwtable safestack { |
| 8 | entry: |
| 9 | ; CHECK: %[[USP:.*]] = load i8*, i8** @__safestack_unsafe_stack_ptr |
| 10 | |
| 11 | ; CHECK: %[[USST:.*]] = getelementptr i8, i8* %[[USP]], i32 -16 |
| 12 | |
| 13 | ; CHECK: store i8* %[[USST]], i8** @__safestack_unsafe_stack_ptr |
| 14 | |
| 15 | ; CHECK: %[[AADDR:.*]] = alloca i8*, align 8 |
| 16 | %a.addr = alloca i8*, align 8 |
| 17 | |
| 18 | ; CHECK: %[[BUFPTR:.*]] = getelementptr i8, i8* %[[USP]], i32 -4 |
| 19 | ; CHECK: %[[BUFPTR2:.*]] = bitcast i8* %[[BUFPTR]] to [4 x i8]* |
| 20 | %buf = alloca [4 x i8], align 1 |
| 21 | |
| 22 | ; CHECK: store i8* {{.*}}, i8** %[[AADDR]], align 8 |
| 23 | store i8* %a, i8** %a.addr, align 8 |
| 24 | |
| 25 | ; CHECK: %[[GEP:.*]] = getelementptr inbounds [4 x i8], [4 x i8]* %[[BUFPTR2]], i32 0, i32 0 |
| 26 | %gep = getelementptr inbounds [4 x i8], [4 x i8]* %buf, i32 0, i32 0 |
| 27 | |
| 28 | ; CHECK: %[[A2:.*]] = load i8*, i8** %[[AADDR]], align 8 |
| 29 | %a2 = load i8*, i8** %a.addr, align 8 |
| 30 | |
| 31 | ; CHECK: call i8* @strcpy(i8* %[[GEP]], i8* %[[A2]]) |
| 32 | %call = call i8* @strcpy(i8* %gep, i8* %a2) |
| 33 | |
| 34 | ; CHECK: store i8* %[[USP]], i8** @__safestack_unsafe_stack_ptr |
| 35 | ret void |
| 36 | } |
| 37 | |
| Evgeniy Stepanov | a4ac3f4 | 2015-12-01 00:06:13 +0000 | [diff] [blame^] | 38 | ; Load from an array at a fixed offset, no overflow. |
| 39 | define i8 @StaticArrayFixedSafe() nounwind uwtable safestack { |
| 40 | entry: |
| 41 | ; CHECK-LABEL: define i8 @StaticArrayFixedSafe( |
| 42 | ; CHECK-NOT: __safestack_unsafe_stack_ptr |
| 43 | ; CHECK: ret i8 |
| 44 | %buf = alloca i8, i32 4, align 1 |
| 45 | %gep = getelementptr inbounds i8, i8* %buf, i32 2 |
| 46 | %x = load i8, i8* %gep, align 1 |
| 47 | ret i8 %x |
| 48 | } |
| 49 | |
| 50 | ; Load from an array at a fixed offset with overflow. |
| 51 | define i8 @StaticArrayFixedUnsafe() nounwind uwtable safestack { |
| 52 | entry: |
| 53 | ; CHECK-LABEL: define i8 @StaticArrayFixedUnsafe( |
| 54 | ; CHECK: __safestack_unsafe_stack_ptr |
| 55 | ; CHECK: ret i8 |
| 56 | %buf = alloca i8, i32 4, align 1 |
| 57 | %gep = getelementptr inbounds i8, i8* %buf, i32 5 |
| 58 | %x = load i8, i8* %gep, align 1 |
| 59 | ret i8 %x |
| 60 | } |
| 61 | |
| 62 | ; Load from an array at an unknown offset. |
| 63 | define i8 @StaticArrayVariableUnsafe(i32 %ofs) nounwind uwtable safestack { |
| 64 | entry: |
| 65 | ; CHECK-LABEL: define i8 @StaticArrayVariableUnsafe( |
| 66 | ; CHECK: __safestack_unsafe_stack_ptr |
| 67 | ; CHECK: ret i8 |
| 68 | %buf = alloca i8, i32 4, align 1 |
| 69 | %gep = getelementptr inbounds i8, i8* %buf, i32 %ofs |
| 70 | %x = load i8, i8* %gep, align 1 |
| 71 | ret i8 %x |
| 72 | } |
| 73 | |
| 74 | ; Load from an array of an unknown size. |
| 75 | define i8 @DynamicArrayUnsafe(i32 %sz) nounwind uwtable safestack { |
| 76 | entry: |
| 77 | ; CHECK-LABEL: define i8 @DynamicArrayUnsafe( |
| 78 | ; CHECK: __safestack_unsafe_stack_ptr |
| 79 | ; CHECK: ret i8 |
| 80 | %buf = alloca i8, i32 %sz, align 1 |
| 81 | %gep = getelementptr inbounds i8, i8* %buf, i32 2 |
| 82 | %x = load i8, i8* %gep, align 1 |
| 83 | ret i8 %x |
| 84 | } |
| 85 | |
| Peter Collingbourne | 82437bf | 2015-06-15 21:07:11 +0000 | [diff] [blame] | 86 | declare i8* @strcpy(i8*, i8*) |