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 6extern void use(id); 7extern void use_block(void (^)(void)); 8 9struct NSFastEnumerationState; 10@interface NSArray 11- (unsigned long) countByEnumeratingWithState: (struct NSFastEnumerationState*) state 12 objects: (id*) buffer 13 count: (unsigned long) bufferSize; 14@end; 15 16void 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 @llvm.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* align 8 [[T0]], i8 0, i64 64, 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* @llvm.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: [[T2:%.*]] = call i8* @llvm.objc.retain(i8* [[T1]]) 72// CHECK-LP64-NEXT: store i8* [[T2]], i8** [[T0]] 73// CHECK-LP64-NEXT: [[BLOCK1:%.*]] = bitcast [[BLOCK_T]]* [[BLOCK]] 74// CHECK-LP64-NEXT: call void @use_block(void ()* [[BLOCK1]]) 75// CHECK-LP64-NEXT: call void @llvm.objc.storeStrong(i8** [[D0]], i8* null) 76// CHECK-LP64-NOT: call void (...) @llvm.objc.clang.arc.use(i8* [[CAPTURE]]) 77 78// CHECK-LP64-OPT: [[D0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i64 0, i32 5 79// CHECK-LP64-OPT: [[CAPTURE:%.*]] = load i8*, i8** [[D0]] 80// CHECK-LP64-OPT: call void (...) @llvm.objc.clang.arc.use(i8* [[CAPTURE]]) 81 82// CHECK-LP64: [[T0:%.*]] = load i8*, i8** @OBJC_SELECTOR_REFERENCES_ 83// CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[ARRAY_T]]* [[SAVED_ARRAY]] to i8* 84// 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) 85 86// Release the array. 87// CHECK-LP64: [[T0:%.*]] = bitcast [[ARRAY_T]]* [[SAVED_ARRAY]] to i8* 88// CHECK-LP64-NEXT: call void @llvm.objc.release(i8* [[T0]]) 89 90// Destroy 'array'. 91// CHECK-LP64: [[T0:%.*]] = bitcast [[ARRAY_T]]** [[ARRAY]] to i8** 92// CHECK-LP64-NEXT: call void @llvm.objc.storeStrong(i8** [[T0]], i8* null) 93// CHECK-LP64-NEXT: ret void 94 95// CHECK-LP64-LABEL: define internal void @__test0_block_invoke 96// CHECK-LP64: [[BLOCK:%.*]] = bitcast i8* {{%.*}} to [[BLOCK_T]]* 97// CHECK-LP64-NOT: ret 98// CHECK-LP64: [[T0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5 99// CHECK-LP64-NEXT: [[T2:%.*]] = load i8*, i8** [[T0]], align 8 100// CHECK-LP64-NEXT: call void @use(i8* [[T2]]) 101 102void test1(NSArray *array) { 103 for (__weak id x in array) { 104 use_block(^{ use(x); }); 105 } 106} 107 108// CHECK-LP64-LABEL: define void @test1( 109// CHECK-LP64: alloca [[ARRAY_T:%.*]]*, 110// CHECK-LP64-NEXT: [[X:%.*]] = alloca i8*, 111// CHECK-LP64-NEXT: [[STATE:%.*]] = alloca [[STATE_T:%.*]], 112// CHECK-LP64-NEXT: alloca [16 x i8*], align 8 113// CHECK-LP64-NEXT: [[BLOCK:%.*]] = alloca [[BLOCK_T:<{.*}>]], 114 115// CHECK-LP64: [[T0:%.*]] = getelementptr inbounds [[STATE_T]], [[STATE_T]]* [[STATE]], i32 0, i32 1 116// CHECK-LP64-NEXT: [[T1:%.*]] = load i8**, i8*** [[T0]] 117// CHECK-LP64-NEXT: [[T2:%.*]] = getelementptr i8*, i8** [[T1]], i64 118// CHECK-LP64-NEXT: [[T3:%.*]] = load i8*, i8** [[T2]] 119// CHECK-LP64-NEXT: call i8* @llvm.objc.initWeak(i8** [[X]], i8* [[T3]]) 120 121// CHECK-LP64: [[D0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5 122// CHECK-LP64: [[T0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i32 0, i32 5 123// CHECK-LP64-NEXT: call void @llvm.objc.copyWeak(i8** [[T0]], i8** [[X]]) 124// CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[BLOCK_T]]* [[BLOCK]] to 125// CHECK-LP64: call void @use_block 126// CHECK-LP64-NEXT: call void @llvm.objc.destroyWeak(i8** [[D0]]) 127// CHECK-LP64-NEXT: call void @llvm.objc.destroyWeak(i8** [[X]]) 128 129// rdar://problem/9817306 130@interface Test2 131- (NSArray *) array; 132@end 133void test2(Test2 *a) { 134 for (id x in a.array) { 135 use(x); 136 } 137} 138 139// CHECK-LP64-LABEL: define void @test2( 140// CHECK-LP64: [[T0:%.*]] = call [[ARRAY_T]]* bitcast (i8* (i8*, i8*, ...)* @objc_msgSend to [[ARRAY_T]]* (i8*, i8*)*)( 141// CHECK-LP64-NEXT: [[T1:%.*]] = bitcast [[ARRAY_T]]* [[T0]] to i8* 142// CHECK-LP64-NEXT: [[T2:%.*]] = notail call i8* @llvm.objc.retainAutoreleasedReturnValue(i8* [[T1]]) 143// CHECK-LP64-NEXT: [[COLL:%.*]] = bitcast i8* [[T2]] to [[ARRAY_T]]* 144 145// Make sure it's not immediately released before starting the iteration. 146// CHECK-LP64-NEXT: load i8*, i8** @OBJC_SELECTOR_REFERENCES_ 147// CHECK-LP64-NEXT: [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8* 148// CHECK-LP64-NEXT: @objc_msgSend 149 150// This bitcast is for the mutation check. 151// CHECK-LP64: [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8* 152// CHECK-LP64-NEXT: @objc_enumerationMutation 153 154// This bitcast is for the 'next' message send. 155// CHECK-LP64: [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8* 156// CHECK-LP64-NEXT: @objc_msgSend 157 158// This bitcast is for the final release. 159// CHECK-LP64: [[T0:%.*]] = bitcast [[ARRAY_T]]* [[COLL]] to i8* 160// CHECK-LP64-NEXT: call void @llvm.objc.release(i8* [[T0]]) 161 162 163// Check that the 'continue' label is positioned appropriately 164// relative to the collection clenaup. 165void test3(NSArray *array) { 166 for (id x in array) { 167 if (!x) continue; 168 use(x); 169 } 170 171 // CHECK-LP64-LABEL: define void @test3( 172 // CHECK-LP64: [[ARRAY:%.*]] = alloca [[ARRAY_T]]*, align 8 173 // CHECK-LP64-NEXT: [[X:%.*]] = alloca i8*, align 8 174 // CHECK-LP64: [[T0:%.*]] = load i8*, i8** [[X]], align 8 175 // CHECK-LP64-NEXT: [[T1:%.*]] = icmp ne i8* [[T0]], null 176 // CHECK-LP64-NEXT: br i1 [[T1]], 177 // CHECK-LP64: br label [[L:%[^ ]+]] 178 // CHECK-LP64: [[T0:%.*]] = load i8*, i8** [[X]], align 8 179 // CHECK-LP64-NEXT: call void @use(i8* [[T0]]) 180 // CHECK-LP64-NEXT: br label [[L]] 181} 182 183@interface NSObject @end 184 185@interface I1 : NSObject 186- (NSArray *) foo1:(void (^)(void))block; 187- (void) foo2; 188@end 189 190NSArray *array4; 191 192@implementation I1 : NSObject 193- (NSArray *) foo1:(void (^)(void))block { 194 block(); 195 return array4; 196} 197 198- (void) foo2 { 199 for (id x in [self foo1:^{ use(self); }]) { 200 use(x); 201 break; 202 } 203} 204@end 205 206// CHECK-LP64-LABEL: define internal void @"\01-[I1 foo2]"( 207// CHECK-LP64: [[SELF_ADDR:%.*]] = alloca [[TY:%.*]]*, 208// CHECK-LP64: [[BLOCK:%.*]] = alloca <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, [[TY]]* }>, 209// CHECK-LP64: store [[TY]]* %self, [[TY]]** [[SELF_ADDR]] 210// 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 211// 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 212// CHECK-LP64: [[T1:%.*]] = load [[TY]]*, [[TY]]** [[SELF_ADDR]] 213// CHECK-LP64: [[T2:%.*]] = bitcast [[TY]]* [[T1]] to i8* 214// CHECK-LP64: call i8* @llvm.objc.retain(i8* [[T2]]) 215 216// CHECK-LP64-OPT-LABEL: define internal void @"\01-[I1 foo2]"( 217// CHECK-LP64-OPT: [[TY:%.*]]* %self 218// CHECK-LP64-OPT: [[BLOCK:%.*]] = alloca [[BLOCK_T:<{.*}>]], 219// CHECK-LP64-OPT: [[T0:%.*]] = getelementptr inbounds [[BLOCK_T]], [[BLOCK_T]]* [[BLOCK]], i64 0, i32 5 220 221// CHECK-LP64: [[T5:%.*]] = bitcast [[TY]]** [[T0]] to i8** 222// CHECK-LP64: call void @llvm.objc.storeStrong(i8** [[T5]], i8* null) 223// CHECK-LP64-NOT: call void (...) @llvm.objc.clang.arc.use([[TY]]* [[T5]]) 224// CHECK-LP64: switch i32 {{%.*}}, label %[[UNREACHABLE:.*]] [ 225// CHECK-LP64-NEXT: i32 0, label %[[CLEANUP_CONT:.*]] 226// CHECK-LP64-NEXT: i32 2, label %[[FORCOLL_END:.*]] 227// CHECK-LP64-NEXT: ] 228 229// CHECK-LP64-OPT: [[T5:%.*]] = load [[TY]]*, [[TY]]** [[T0]] 230// CHECK-LP64-OPT: call void (...) @llvm.objc.clang.arc.use([[TY]]* [[T5]]) 231 232// CHECK-LP64: {{^|:}}[[CLEANUP_CONT]] 233// CHECK-LP64-NEXT: br label %[[FORCOLL_END]] 234 235// CHECK-LP64: {{^|:}}[[FORCOLL_END]] 236// CHECK-LP64-NEXT: ret void 237 238// CHECK-LP64: {{^|:}}[[UNREACHABLE]] 239// CHECK-LP64-NEXT: unreachable 240 241// CHECK-LP64: attributes [[NUW]] = { nounwind } 242