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

   1 // RUN: %clang_cc1 -triple=x86_64-apple-darwin10 -emit-llvm -std=c++14 -fblocks -fobjc-arc -o - %s | FileCheck %s
   2
   3 // CHECK: %[[S:.*]] = type { i32 }
   4 // CHECK: %[[CLASS_ANON_2:.*]] = type { ptr }
   5 // CHECK: %[[CLASS_ANON_3:.*]] = type { %[[S]] }
   6
   7 // CHECK: %[[BLOCK_CAPTURED0:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr }>, ptr %[[BLOCK:.*]], i32 0, i32 5
   8 // CHECK: %[[V0:.*]] = getelementptr inbounds %[[LAMBDA_CLASS:.*]], ptr %[[THIS:.*]], i32 0, i32 0
   9 // CHECK: %[[V1:.*]] = load ptr, ptr %[[V0]], align 8
  10 // CHECK: store ptr %[[V1]], ptr %[[BLOCK_CAPTURED0]], align 8
  11 // CHECK: %[[BLOCK_CAPTURED1:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr }>, ptr %[[BLOCK]], i32 0, i32 6
  12 // CHECK: %[[V2:.*]] = getelementptr inbounds %[[LAMBDA_CLASS]], ptr %[[THIS]], i32 0, i32 1
  13 // CHECK: %[[V3:.*]] = load ptr, ptr %[[V2]], align 8
  14 // CHECK: store ptr %[[V3]], ptr %[[BLOCK_CAPTURED1]], align 8
  15
  16 void foo1(int &, int &);
  17
  18 void block_in_lambda(int &s1, int &s2) {
  19   auto lambda = [&s1, &s2]() {
  20     auto block = ^{
  21       foo1(s1, s2);
  22     };
  23     block();
  24   };
  25
  26   lambda();
  27 }
  28
  29 namespace CaptureByReference {
  30
  31 id getObj();
  32 void use(id);
  33
  34 // Block copy/dispose helpers aren't needed because 'a' is captured by
  35 // reference.
  36
  37 // CHECK-LABEL: define{{.*}} void @_ZN18CaptureByReference5test0Ev(
  38 // CHECK-LABEL: define internal void @"_ZZN18CaptureByReference5test0EvENK3$_0clEv"(
  39 // CHECK: %[[BLOCK_DESCRIPTOR:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 4
  40 // CHECK: store ptr @"__block_descriptor_40_e5_v8\01?0ls32l8", ptr %[[BLOCK_DESCRIPTOR]], align 8
  41
  42 void test0() {
  43   id a = getObj();
  44   [&]{ ^{ a = 0; }(); }();
  45 }
  46
  47 // Block copy/dispose helpers shouldn't have to retain/release 'a' because it
  48 // is captured by reference.
  49
  50 // CHECK-LABEL: define{{.*}} void @_ZN18CaptureByReference5test1Ev(
  51 // CHECK-LABEL: define internal void @"_ZZN18CaptureByReference5test1EvENK3$_0clEv"(
  52 // CHECK: %[[BLOCK_DESCRIPTOR:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 4
  53 // CHECK: store ptr @"__block_descriptor_56_8_32s40s_e5_v8\01?0l", ptr %[[BLOCK_DESCRIPTOR]], align 8
  54
  55 void test1() {
  56   id a = getObj(), b = getObj(), c = getObj();
  57   [&a, b, c]{ ^{ a = 0; use(b); use(c); }(); }();
  58 }
  59
  60 struct S {
  61   int val() const;
  62   int a;
  63   S();
  64   S(const S&);
  65   S &operator=(const S&);
  66   S(S&&);
  67   S &operator=(S&&);
  68 };
  69
  70 S getS();
  71
  72 // CHECK: define internal noundef i32 @"_ZZN18CaptureByReference5test2EvENK3$_0clIiEEDaT_"(ptr {{[^,]*}} %{{.*}}, i32 noundef %{{.*}})
  73 // CHECK: %[[BLOCK:.*]] = alloca <{ ptr, i32, i32, ptr, {{.*}}, ptr }>, align 8
  74 // CHECK: %[[BLOCK_CAPTURED:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, {{.*}}, ptr }>, ptr %[[BLOCK]], i32 0, i32 5
  75 // CHECK: %[[V0:.*]] = getelementptr inbounds %[[CLASS_ANON_2]], ptr %{{.*}}, i32 0, i32 0
  76 // CHECK: %[[V1:.*]] = load ptr, ptr %[[V0]], align 8
  77 // CHECK: store ptr %[[V1]], ptr %[[BLOCK_CAPTURED]], align 8
  78
  79 int test2() {
  80   S s;
  81   auto fn = [&](const auto a){
  82     return ^{
  83       return s.val();
  84     }();
  85   };
  86   return fn(123);
  87 }
  88
  89 // CHECK: define internal noundef i32 @"_ZZN18CaptureByReference5test3EvENK3$_0clIiEEDaT_"(ptr {{[^,]*}} %{{.*}}, i32 noundef %{{.*}})
  90 // CHECK: %[[BLOCK:.*]] = alloca <{ ptr, i32, i32, ptr, ptr, %[[S]] }>, align 8
  91 // CHECK: %[[BLOCK_CAPTURED:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, %[[S]] }>, ptr %[[BLOCK]], i32 0, i32 5
  92 // CHECK: %[[V0:.*]] = getelementptr inbounds %[[CLASS_ANON_3]], ptr %{{.*}}, i32 0, i32 0
  93 // CHECK: call void @_ZN18CaptureByReference1SC1ERKS0_(ptr {{[^,]*}} %[[BLOCK_CAPTURED]], ptr {{.*}} %[[V0]])
  94
  95 int test3() {
  96   const S &s = getS();
  97   auto fn = [=](const auto a){
  98     return ^{
  99       return s.val();
 100     }();
 101   };
 102   return fn(123);
 103 }
 104
 105 // CHECK-LABEL: define linkonce_odr hidden void @__copy_helper_block_8_32s40s(
 106 // CHECK-NOT: call void @llvm.objc.storeStrong(
 107 // CHECK: %[[V4:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 5
 108 // CHECK: %[[V5:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 5
 109 // CHECK: %[[BLOCKCOPY_SRC:.*]] = load ptr, ptr %[[V4]], align 8
 110 // CHECK: store ptr null, ptr %[[V5]], align 8
 111 // CHECK: call void @llvm.objc.storeStrong(ptr %[[V5]], ptr %[[BLOCKCOPY_SRC]])
 112 // CHECK: %[[V6:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 6
 113 // CHECK: %[[V7:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 6
 114 // CHECK: %[[BLOCKCOPY_SRC2:.*]] = load ptr, ptr %[[V6]], align 8
 115 // CHECK: store ptr null, ptr %[[V7]], align 8
 116 // CHECK: call void @llvm.objc.storeStrong(ptr %[[V7]], ptr %[[BLOCKCOPY_SRC2]])
 117 // CHECK-NOT: call void @llvm.objc.storeStrong(
 118 // CHECK: ret void
 119
 120 // CHECK-LABEL: define linkonce_odr hidden void @__destroy_helper_block_8_32s40s(
 121 // CHECK: %[[V2:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 5
 122 // CHECK: %[[V3:.*]] = getelementptr inbounds <{ ptr, i32, i32, ptr, ptr, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 6
 123 // CHECK-NOT: call void @llvm.objc.storeStrong(
 124 // CHECK: call void @llvm.objc.storeStrong(ptr %[[V3]], ptr null)
 125 // CHECK: call void @llvm.objc.storeStrong(ptr %[[V2]], ptr null)
 126 // CHECK-NOT: call void @llvm.objc.storeStrong(
 127 // CHECK: ret void
 128
 129 }