test/CodeGenObjCXX/block-nested-in-lambda.mm

   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 }