test/CodeGenObjC/arc-foreach.m

   1 // RUN: %clang_cc1 -triple x86_64-apple-darwin -fblocks -fobjc-arc -fobjc-runtime-has-weak -emit-llvm %s -o - | FileCheck -check-prefix CHECK-LP64 %s
   2 // RUN: %clang_cc1 -triple x86_64-apple-darwin -O1 -fblocks -fobjc-arc -fobjc-runtime-has-weak -emit-llvm %s -o - | FileCheck -check-prefix CHECK-LP64-OPT %s
   3 // rdar://9503326
   4 // rdar://9606600
   5
   6 extern void use(id);
   7 extern void use_block(void (^)(void));
   8
   9 struct NSFastEnumerationState;
  10 @interface NSArray
  11 - (unsigned long) countByEnumeratingWithState: (struct NSFastEnumerationState*) state
  12                   objects: (id*) buffer
  13                   count: (unsigned long) bufferSize;
  14 @end;
  15
  16 void test0(NSArray *array) {
  17   // 'x' should be initialized without a retain.
  18   // We should actually do a non-constant capture, and that
  19   // capture should require a retain.
  20   for (id x in array) {
  21     use_block(^{ use(x); });
  22   }
  23 }
  24
  25 // CHECK-LP64-LABEL:    define void @test0(
  26 // CHECK-LP64:      [[ARRAY:%.*]] = alloca [[ARRAY_T:%.*]]*,
  27 // CHECK-LP64-NEXT: [[X:%.*]] = alloca i8*,
  28 // CHECK-LP64-NEXT: [[STATE:%.*]] = alloca [[STATE_T:%.*]],
  29 // CHECK-LP64-NEXT: [[BUFFER:%.*]] = alloca [16 x i8*], align 8
  30 // CHECK-LP64-NEXT: [[BLOCK:%.*]] = alloca [[BLOCK_T:<{.*}>]],
  31
  32 // CHECK-LP64-OPT-LABEL: define void @test0
  33 // CHECK-LP64-OPT: [[STATE:%.*]] = alloca [[STATE_T:%.*]], align 8
  34 // CHECK-LP64-OPT-NEXT: [[BUFFER:%.*]] = alloca [16 x i8*], align 8
  35 // CHECK-LP64-OPT-NEXT: [[BLOCK:%.*]] = alloca [[BLOCK_T:<{.*}>]], align 8
  36
  37 // Initialize 'array'.
  38 // CHECK-LP64-NEXT: store [[ARRAY_T]]* null, [[ARRAY_T]]** [[ARRAY]]
  39 // CHECK-LP64-NEXT: [[ZERO:%.*]] = bitcast [[ARRAY_T]]** [[ARRAY]] to i8**
  40 // CHECK-LP64-NEXT: [[ONE:%.*]] = bitcast [[ARRAY_T]]* {{%.*}} to i8*
  41 // CHECK-LP64-NEXT: call void @objc_storeStrong(i8** [[ZERO]], i8* [[ONE]]) [[NUW:#[0-9]+]]
  42
  43 // Initialize the fast enumaration state.
  44 // CHECK-LP64-NEXT: [[T0:%.*]] = bitcast [[STATE_T]]* [[STATE]] to i8*
  45 // CHECK-LP64-NEXT: call void @llvm.memset.p0i8.i64(i8* [[T0]], i8 0, i64 64, i32 8, i1 false)
  46
  47 // Evaluate the collection expression and retain.
  48 // CHECK-LP64-NEXT: [[T0:%.*]] = load [[ARRAY_T]]*, [[ARRAY_T]]** [[ARRAY]], align 8
  49 // CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[ARRAY_T]]* [[T0]] to i8*
  50 // CHECK-LP64-NEXT: [[T2:%.*]] = call i8* @objc_retain(i8* [[T1]])
  51 // CHECK-LP64-NEXT: [[SAVED_ARRAY:%.*]] = bitcast i8* [[T2]] to [[ARRAY_T]]*
  52
  53 // Call the enumeration method.
  54 // CHECK-LP64-NEXT: [[T0:%.*]] = load i8*, i8** @OBJC_SELECTOR_REFERENCES_
  55 // CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[ARRAY_T]]* [[SAVED_ARRAY]] to i8*
  56 // CHECK-LP64-NEXT: [[SIZE:%.*]] = call i64 bitcast (i8* (i8*, i8*, ...)* @objc_msgSend to i64 (i8*, i8*, [[STATE_T]]*, [16 x i8*]*, i64)*)(i8* [[T1]], i8* [[T0]], [[STATE_T]]* [[STATE]], [16 x i8*]* [[BUFFER]], i64 16)
  57
  58 // Check for a nonzero result.
  59 // CHECK-LP64-NEXT: [[T0:%.*]] = icmp eq i64 [[SIZE]], 0
  60 // CHECK-LP64-NEXT: br i1 [[T0]]
  61
  62 // CHECK-LP64:      [[T0:%.*]] = getelementptr inbounds [[STATE_T]], [[STATE_T]]* [[STATE]], i32 0, i32 1
  63 // CHECK-LP64-NEXT: [[T1:%.*]] = load i8**, i8*** [[T0]]
  64 // CHECK-LP64-NEXT: [[T2:%.*]] = getelementptr i8*, i8** [[T1]], i64
  65 // CHECK-LP64-NEXT: [[T3:%.*]] = load i8*, i8** [[T2]]
  66 // CHECK-LP64-NEXT: store i8* [[T3]], i8** [[X]]
  67
  68 // CHECK-LP64:      [[D0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5
  69 // CHECK-LP64:      [[T0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5
  70 // CHECK-LP64-NEXT: [[T1:%.*]] = load i8*, i8** [[X]]
  71 // CHECK-LP64-NEXT: store i8* [[T1]], i8** [[T0]]
  72 // CHECK-LP64-NEXT: [[BLOCK1:%.*]] = bitcast [[BLOCK_T]]* [[BLOCK]]
  73 // CHECK-LP64-NEXT: call void @use_block(void ()* [[BLOCK1]])
  74 // CHECK-LP64-NEXT: call void @objc_storeStrong(i8** [[D0]], i8* null)
  75 // CHECK-LP64-NOT:  call void (...) @clang.arc.use(i8* [[CAPTURE]])
  76
  77 // CHECK-LP64-OPT: [[D0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i64 0, i32 5
  78 // CHECK-LP64-OPT: [[CAPTURE:%.*]] = load i8*, i8** [[D0]]
  79 // CHECK-LP64-OPT: call void (...) @clang.arc.use(i8* [[CAPTURE]])
  80
  81 // CHECK-LP64:      [[T0:%.*]] = load i8*, i8** @OBJC_SELECTOR_REFERENCES_
  82 // CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[ARRAY_T]]* [[SAVED_ARRAY]] to i8*
  83 // CHECK-LP64-NEXT: [[SIZE:%.*]] = call i64 bitcast (i8* (i8*, i8*, ...)* @objc_msgSend to i64 (i8*, i8*, [[STATE_T]]*, [16 x i8*]*, i64)*)(i8* [[T1]], i8* [[T0]], [[STATE_T]]* [[STATE]], [16 x i8*]* [[BUFFER]], i64 16)
  84
  85 // Release the array.
  86 // CHECK-LP64:      [[T0:%.*]] = bitcast [[ARRAY_T]]* [[SAVED_ARRAY]] to i8*
  87 // CHECK-LP64-NEXT: call void @objc_release(i8* [[T0]])
  88
  89 // Destroy 'array'.
  90 // CHECK-LP64:      [[T0:%.*]] = bitcast [[ARRAY_T]]** [[ARRAY]] to i8**
  91 // CHECK-LP64-NEXT: call void @objc_storeStrong(i8** [[T0]], i8* null)
  92 // CHECK-LP64-NEXT: ret void
  93
  94 // CHECK-LP64-LABEL:    define internal void @__test0_block_invoke
  95 // CHECK-LP64:      [[BLOCK:%.*]] = bitcast i8* {{%.*}} to [[BLOCK_T]]*
  96 // CHECK-LP64-NOT:  ret
  97 // CHECK-LP64:      [[T0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5
  98 // CHECK-LP64-NEXT: [[T2:%.*]] = load i8*, i8** [[T0]], align 8
  99 // CHECK-LP64-NEXT: call void @use(i8* [[T2]])
 100
 101 void test1(NSArray *array) {
 102   for (__weak id x in array) {
 103     use_block(^{ use(x); });
 104   }
 105 }
 106
 107 // CHECK-LP64-LABEL:    define void @test1(
 108 // CHECK-LP64:      alloca [[ARRAY_T:%.*]]*,
 109 // CHECK-LP64-NEXT: [[X:%.*]] = alloca i8*,
 110 // CHECK-LP64-NEXT: [[STATE:%.*]] = alloca [[STATE_T:%.*]],
 111 // CHECK-LP64-NEXT: alloca [16 x i8*], align 8
 112 // CHECK-LP64-NEXT: [[BLOCK:%.*]] = alloca [[BLOCK_T:<{.*}>]],
 113
 114 // CHECK-LP64:      [[T0:%.*]] = getelementptr inbounds [[STATE_T]], [[STATE_T]]* [[STATE]], i32 0, i32 1
 115 // CHECK-LP64-NEXT: [[T1:%.*]] = load i8**, i8*** [[T0]]
 116 // CHECK-LP64-NEXT: [[T2:%.*]] = getelementptr i8*, i8** [[T1]], i64
 117 // CHECK-LP64-NEXT: [[T3:%.*]] = load i8*, i8** [[T2]]
 118 // CHECK-LP64-NEXT: call i8* @objc_initWeak(i8** [[X]], i8* [[T3]])
 119
 120 // CHECK-LP64:      [[D0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5
 121 // CHECK-LP64:      [[T0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5
 122 // CHECK-LP64-NEXT: call void @objc_copyWeak(i8** [[T0]], i8** [[X]])
 123 // CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[BLOCK_T]]* [[BLOCK]] to
 124 // CHECK-LP64: call void @use_block
 125 // CHECK-LP64-NEXT: call void @objc_destroyWeak(i8** [[D0]])
 126 // CHECK-LP64-NEXT: call void @objc_destroyWeak(i8** [[X]])
 127
 128 // rdar://problem/9817306
 129 @interface Test2
 130 - (NSArray *) array;
 131 @end
 132 void test2(Test2 *a) {
 133   for (id x in a.array) {
 134     use(x);
 135   }
 136 }
 137
 138 // CHECK-LP64-LABEL:    define void @test2(
 139 // CHECK-LP64:      [[T0:%.*]] = call [[ARRAY_T]]* bitcast (i8* (i8*, i8*, ...)* @objc_msgSend to [[ARRAY_T]]* (i8*, i8*)*)(
 140 // CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[ARRAY_T]]* [[T0]] to i8*
 141 // CHECK-LP64-NEXT: [[T2:%.*]] = call i8* @objc_retainAutoreleasedReturnValue(i8* [[T1]])
 142 // CHECK-LP64-NEXT: [[COLL:%.*]] = bitcast i8* [[T2]] to [[ARRAY_T]]*
 143
 144 // Make sure it's not immediately released before starting the iteration.
 145 // CHECK-LP64-NEXT: load i8*, i8** @OBJC_SELECTOR_REFERENCES_
 146 // CHECK-LP64-NEXT: [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8*
 147 // CHECK-LP64-NEXT: @objc_msgSend
 148
 149 // This bitcast is for the mutation check.
 150 // CHECK-LP64:      [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8*
 151 // CHECK-LP64-NEXT: @objc_enumerationMutation
 152
 153 // This bitcast is for the 'next' message send.
 154 // CHECK-LP64:      [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8*
 155 // CHECK-LP64-NEXT: @objc_msgSend
 156
 157 // This bitcast is for the final release.
 158 // CHECK-LP64:      [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8*
 159 // CHECK-LP64-NEXT: call void @objc_release(i8* [[T0]])
 160
 161
 162 // Check that the 'continue' label is positioned appropriately
 163 // relative to the collection clenaup.
 164 void test3(NSArray *array) {
 165   for (id x in array) {
 166     if (!x) continue;
 167     use(x);
 168   }
 169
 170   // CHECK-LP64-LABEL:    define void @test3(
 171   // CHECK-LP64:      [[ARRAY:%.*]] = alloca [[ARRAY_T]]*, align 8
 172   // CHECK-LP64-NEXT: [[X:%.*]] = alloca i8*, align 8
 173   // CHECK-LP64:      [[T0:%.*]] = load i8*, i8** [[X]], align 8
 174   // CHECK-LP64-NEXT: [[T1:%.*]] = icmp ne i8* [[T0]], null
 175   // CHECK-LP64-NEXT: br i1 [[T1]],
 176   // CHECK-LP64:      br label [[L:%[^ ]+]]
 177   // CHECK-LP64:      [[T0:%.*]] = load i8*, i8** [[X]], align 8
 178   // CHECK-LP64-NEXT: call void @use(i8* [[T0]])
 179   // CHECK-LP64-NEXT: br label [[L]]
 180 }
 181
 182 @interface NSObject @end
 183
 184 @interface I1 : NSObject
 185 - (NSArray *) foo1:(void (^)(void))block;
 186 - (void) foo2;
 187 @end
 188
 189 NSArray *array4;
 190
 191 @implementation I1 : NSObject
 192 - (NSArray *) foo1:(void (^)(void))block {
 193   block();
 194   return array4;
 195 }
 196
 197 - (void) foo2 {
 198   for (id x in [self foo1:^{ use(self); }]) {
 199     use(x);
 200     break;
 201   }
 202 }
 203 @end
 204
 205 // CHECK-LP64-LABEL: define internal void @"\01-[I1 foo2]"(
 206 // CHECK-LP64:         [[SELF_ADDR:%.*]] = alloca [[TY:%.*]]*,
 207 // CHECK-LP64:         [[BLOCK:%.*]] = alloca <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, [[TY]]* }>,
 208 // CHECK-LP64:         store [[TY]]* %self, [[TY]]** [[SELF_ADDR]]
 209 // CHECK-LP64:         [[T0:%.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, [[TY]]* }>, <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, [[TY]]* }>* [[BLOCK]], i32 0, i32 5
 210 // CHECK-LP64:         [[BC:%.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, [[TY]]* }>, <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, [[TY]]* }>* [[BLOCK]], i32 0, i32 5
 211 // CHECK-LP64:         [[T1:%.*]] = load [[TY]]*, [[TY]]** [[SELF_ADDR]]
 212 // CHECK-LP64:         store [[TY]]* [[T1]], [[TY]]** [[BC]], align 8
 213
 214 // CHECK-LP64-OPT-LABEL: define internal void @"\01-[I1 foo2]"(
 215 // CHECK-LP64-OPT: [[TY:%.*]]* %self
 216 // CHECK-LP64-OPT: [[BLOCK:%.*]] = alloca [[BLOCK_T:<{.*}>]],
 217 // CHECK-LP64-OPT: [[T0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i64 0, i32 5
 218
 219 // CHECK-LP64:         [[T5:%.*]] = bitcast [[TY]]** [[T0]] to i8**
 220 // CHECK-LP64:         call void @objc_storeStrong(i8** [[T5]], i8* null)
 221 // CHECK-LP64-NOT:     call void (...) @clang.arc.use([[TY]]* [[T5]])
 222 // CHECK-LP64:         switch i32 {{%.*}}, label %[[UNREACHABLE:.*]] [
 223 // CHECK-LP64-NEXT:      i32 0, label %[[CLEANUP_CONT:.*]]
 224 // CHECK-LP64-NEXT:      i32 2, label %[[FORCOLL_END:.*]]
 225 // CHECK-LP64-NEXT:    ]
 226
 227 // CHECK-LP64-OPT: [[T5:%.*]] = load [[TY]]*, [[TY]]** [[T0]]
 228 // CHECK-LP64-OPT: call void (...) @clang.arc.use([[TY]]* [[T5]])
 229
 230 // CHECK-LP64:       {{^|:}}[[CLEANUP_CONT]]
 231 // CHECK-LP64-NEXT:    br label %[[FORCOLL_END]]
 232
 233 // CHECK-LP64:       {{^|:}}[[FORCOLL_END]]
 234 // CHECK-LP64-NEXT:    ret void
 235
 236 // CHECK-LP64:       {{^|:}}[[UNREACHABLE]]
 237 // CHECK-LP64-NEXT:    unreachable
 238
 239 // CHECK-LP64: attributes [[NUW]] = { nounwind }