| 1 | // RUN: %clang_cc1 -triple=x86_64-apple-darwin10 -emit-llvm -std=c++11 -fblocks -fobjc-arc -o - %s | FileCheck %s |
| 2 | |
| 3 | // CHECK: %[[STRUCT_BLOCK_DESCRIPTOR:.*]] = type { i64, i64 } |
| 4 | |
| 5 | // CHECK: %[[BLOCK_CAPTURED0:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, i32*, i32* }>, <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, i32*, i32* }>* %[[BLOCK:.*]], i32 0, i32 5 |
| 6 | // CHECK: %[[V0:.*]] = getelementptr inbounds %[[LAMBDA_CLASS:.*]], %[[LAMBDA_CLASS]]* %[[THIS:.*]], i32 0, i32 0 |
| 7 | // CHECK: %[[V1:.*]] = load i32*, i32** %[[V0]], align 8 |
| 8 | // CHECK: store i32* %[[V1]], i32** %[[BLOCK_CAPTURED0]], align 8 |
| 9 | // CHECK: %[[BLOCK_CAPTURED1:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, i32*, i32* }>, <{ i8*, i32, i32, i8*, %struct.__block_descriptor*, i32*, i32* }>* %[[BLOCK]], i32 0, i32 6 |
| 10 | // CHECK: %[[V2:.*]] = getelementptr inbounds %[[LAMBDA_CLASS]], %[[LAMBDA_CLASS]]* %[[THIS]], i32 0, i32 1 |
| 11 | // CHECK: %[[V3:.*]] = load i32*, i32** %[[V2]], align 8 |
| 12 | // CHECK: store i32* %[[V3]], i32** %[[BLOCK_CAPTURED1]], align 8 |
| 13 | |
| 14 | void foo1(int &, int &); |
| 15 | |
| 16 | void block_in_lambda(int &s1, int &s2) { |
| 17 | auto lambda = [&s1, &s2]() { |
| 18 | auto block = ^{ |
| 19 | foo1(s1, s2); |
| 20 | }; |
| 21 | block(); |
| 22 | }; |
| 23 | |
| 24 | lambda(); |
| 25 | } |
| 26 | |
| 27 | namespace CaptureByReference { |
| 28 | |
| 29 | id getObj(); |
| 30 | void use(id); |
| 31 | |
| 32 | // Block copy/dispose helpers aren't needed because 'a' is captured by |
| 33 | // reference. |
| 34 | |
| 35 | // CHECK-LABEL: define void @_ZN18CaptureByReference5test0Ev( |
| 36 | // CHECK-LABEL: define internal void @"_ZZN18CaptureByReference5test0EvENK3$_1clEv"( |
| 37 | // CHECK: %[[BLOCK_DESCRIPTOR:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8** }>* %{{.*}}, i32 0, i32 4 |
| 38 | // CHECK: store %[[STRUCT_BLOCK_DESCRIPTOR]]* bitcast ({ i64, i64, i8*, i64 }* @"__block_descriptor_40_e5_v8\01?0ls32l8" to %[[STRUCT_BLOCK_DESCRIPTOR]]*), %[[STRUCT_BLOCK_DESCRIPTOR]]** %[[BLOCK_DESCRIPTOR]], align 8 |
| 39 | |
| 40 | void test0() { |
| 41 | id a = getObj(); |
| 42 | [&]{ ^{ a = 0; }(); }(); |
| 43 | } |
| 44 | |
| 45 | // Block copy/dispose helpers shouldn't have to retain/release 'a' because it |
| 46 | // is captured by reference. |
| 47 | |
| 48 | // CHECK-LABEL: define void @_ZN18CaptureByReference5test1Ev( |
| 49 | // CHECK-LABEL: define internal void @"_ZZN18CaptureByReference5test1EvENK3$_2clEv"( |
| 50 | // CHECK: %[[BLOCK_DESCRIPTOR:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>* %{{.*}}, i32 0, i32 4 |
| 51 | // CHECK: store %[[STRUCT_BLOCK_DESCRIPTOR]]* bitcast ({ i64, i64, i8*, i8*, i8*, i64 }* @"__block_descriptor_56_8_32s40s_e5_v8\01?0l" to %[[STRUCT_BLOCK_DESCRIPTOR]]*), %[[STRUCT_BLOCK_DESCRIPTOR]]** %[[BLOCK_DESCRIPTOR]], align 8 |
| 52 | |
| 53 | // CHECK-LABEL: define linkonce_odr hidden void @__copy_helper_block_8_32s40s( |
| 54 | // CHECK-NOT: call void @llvm.objc.storeStrong( |
| 55 | // CHECK: %[[V4:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>* %{{.*}}, i32 0, i32 5 |
| 56 | // CHECK: %[[V5:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>* %{{.*}}, i32 0, i32 5 |
| 57 | // CHECK: %[[BLOCKCOPY_SRC:.*]] = load i8*, i8** %[[V4]], align 8 |
| 58 | // CHECK: store i8* null, i8** %[[V5]], align 8 |
| 59 | // CHECK: call void @llvm.objc.storeStrong(i8** %[[V5]], i8* %[[BLOCKCOPY_SRC]]) |
| 60 | // CHECK: %[[V6:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>* %{{.*}}, i32 0, i32 6 |
| 61 | // CHECK: %[[V7:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>* %{{.*}}, i32 0, i32 6 |
| 62 | // CHECK: %[[BLOCKCOPY_SRC2:.*]] = load i8*, i8** %[[V6]], align 8 |
| 63 | // CHECK: store i8* null, i8** %[[V7]], align 8 |
| 64 | // CHECK: call void @llvm.objc.storeStrong(i8** %[[V7]], i8* %[[BLOCKCOPY_SRC2]]) |
| 65 | // CHECK-NOT: call void @llvm.objc.storeStrong( |
| 66 | // CHECK: ret void |
| 67 | |
| 68 | // CHECK-LABEL: define linkonce_odr hidden void @__destroy_helper_block_8_32s40s( |
| 69 | // CHECK: %[[V2:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>* %{{.*}}, i32 0, i32 5 |
| 70 | // CHECK: %[[V3:.*]] = getelementptr inbounds <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>, <{ i8*, i32, i32, i8*, %[[STRUCT_BLOCK_DESCRIPTOR]]*, i8*, i8*, i8** }>* %{{.*}}, i32 0, i32 6 |
| 71 | // CHECK-NOT: call void @llvm.objc.storeStrong( |
| 72 | // CHECK: call void @llvm.objc.storeStrong(i8** %[[V3]], i8* null) |
| 73 | // CHECK: call void @llvm.objc.storeStrong(i8** %[[V2]], i8* null) |
| 74 | // CHECK-NOT: call void @llvm.objc.storeStrong( |
| 75 | // CHECK: ret void |
| 76 | |
| 77 | void test1() { |
| 78 | id a = getObj(), b = getObj(), c = getObj(); |
| 79 | [&a, b, c]{ ^{ a = 0; use(b); use(c); }(); }(); |
| 80 | } |
| 81 | |
| 82 | } |
| 83 | |