test/Transforms/SafeStack/array.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 ; 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
  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
  86 declare i8* @strcpy(i8*, i8*)