150
|
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 @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
|
|
102 void 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
|
|
133 void 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.
|
|
165 void 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
|
|
190 NSArray *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 }
|