clang/test/CodeGen/blocks.c

   1 // RUN: %clang_cc1 -triple i386-unknown-unknown %s -emit-llvm -Wno-strict-prototypes -o - -fblocks | FileCheck --check-prefix=CHECK --check-prefix=SIG_STR %s
   2 // RUN: %clang_cc1 -triple i386-unknown-unknown %s -emit-llvm -Wno-strict-prototypes -o - -fblocks -fdisable-block-signature-string | FileCheck --check-prefix=CHECK --check-prefix=NO_SIG_STR %s
   3 // RUN: %clang_cc1 -triple s390x-unknown-unknown %s -emit-llvm -Wno-strict-prototypes -o - -fblocks | FileCheck --check-prefix=SYSTEMZ %s
   4
   5 // SIG_STR: @[[STR:.*]] = private unnamed_addr constant [6 x i8] c"v4@?0\00", align 1
   6 // SIG_STR: @{{.*}} = internal constant { ptr, i32, i32, ptr, ptr } { ptr @_NSConcreteGlobalBlock, i32 1342177280, i32 0, ptr @f_block_invoke, ptr @{{.*}} }, align 4
   7 // NO_SIG_STR: @{{.*}} = internal constant { ptr, i32, i32, ptr, ptr } { ptr @_NSConcreteGlobalBlock, i32 268435456, i32 0, ptr @f_block_invoke, ptr @{{.*}} }, align 4
   8
   9 // SIG_STR: @{{.*}} = internal constant { i32, i32, ptr, ptr, ptr, ptr } { i32 0, i32 24, ptr @__copy_helper_block_4_20r, ptr @__destroy_helper_block_4_20r, ptr @[[STR]], ptr null }, align 4
  10 // SIG_STR: @[[BLOCK_DESCRIPTOR_TMP21:.*]] = internal constant { i32, i32, ptr, ptr, ptr, ptr } { i32 0, i32 24, ptr @__copy_helper_block_4_20r, ptr @__destroy_helper_block_4_20r, ptr @[[STR]], ptr null }, align 4
  11 // NO_SIG_STR: @{{.*}} = internal constant { i32, i32, ptr, ptr, ptr, ptr } { i32 0, i32 24, ptr @__copy_helper_block_4_20r, ptr @__destroy_helper_block_4_20r, ptr null, ptr null }, align 4
  12 // NO_SIG_STR: @[[BLOCK_DESCRIPTOR_TMP21:.*]] = internal constant { i32, i32, ptr, ptr, ptr, ptr } { i32 0, i32 24, ptr @__copy_helper_block_4_20r, ptr @__destroy_helper_block_4_20r, ptr null, ptr null }, align 4
  13
  14 void (^f)(void) = ^{};
  15
  16 int f0(int (^a0)()) {
  17   return a0(1, 2, 3);
  18 }
  19
  20 // Verify that attributes on blocks are set correctly.
  21 typedef struct s0 T;
  22 struct s0 {
  23   int a[64];
  24 };
  25
  26 // CHECK: define internal void @__f2_block_invoke(ptr dead_on_unwind noalias writable sret(%struct.s0) align 4 {{%.*}}, ptr noundef {{%.*}}, ptr noundef byval(%struct.s0) align 4 {{.*}})
  27 struct s0 f2(struct s0 a0) {
  28   return ^(struct s0 a1){ return a1; }(a0);
  29 }
  30
  31 // This should not crash.
  32 void *P = ^{
  33   void *Q = __func__;
  34 };
  35
  36 void (^test1)(void) = ^(void) {
  37   __block int i;
  38   ^ { i = 1; }();
  39 };
  40
  41 // CHECK-LABEL: define linkonce_odr hidden void @__copy_helper_block_4_20r(ptr noundef %0, ptr noundef %1) unnamed_addr
  42 // CHECK: %[[_ADDR:.*]] = alloca ptr, align 4
  43 // CHECK-NEXT: %[[_ADDR1:.*]] = alloca ptr, align 4
  44 // CHECK-NEXT: store ptr %0, ptr %[[_ADDR]], align 4
  45 // CHECK-NEXT: store ptr %1, ptr %[[_ADDR1]], align 4
  46 // CHECK-NEXT: %[[V2:.*]] = load ptr, ptr %[[_ADDR1]], align 4
  47 // CHECK-NEXT: %[[V3:.*]] = load ptr, ptr %[[_ADDR]], align 4
  48 // CHECK-NEXT: %[[V4:.*]] = getelementptr inbounds nuw <{ ptr, i32, i32, ptr, ptr, ptr }>, ptr %[[V2]], i32 0, i32 5
  49 // CHECK-NEXT: %[[V5:.*]] = getelementptr inbounds nuw <{ ptr, i32, i32, ptr, ptr, ptr }>, ptr %[[V3]], i32 0, i32 5
  50 // CHECK-NEXT: %[[BLOCKCOPY_SRC:.*]] = load ptr, ptr %[[V4]], align 4
  51 // CHECK-NEXT: call void @_Block_object_assign(ptr %[[V5]], ptr %[[BLOCKCOPY_SRC]], i32 8)
  52 // CHECK-NEXT: ret void
  53
  54 // SYSTEMZ: declare void @_Block_object_assign(ptr noundef, ptr noundef, i32 noundef signext)
  55
  56 // CHECK-LABEL: define linkonce_odr hidden void @__destroy_helper_block_4_20r(ptr noundef %0) unnamed_addr
  57 // CHECK: %[[_ADDR:.*]] = alloca ptr, align 4
  58 // CHECK-NEXT: store ptr %0, ptr %[[_ADDR]], align 4
  59 // CHECK-NEXT: %[[V1:.*]] = load ptr, ptr %[[_ADDR]], align 4
  60 // CHECK-NEXT: %[[V2:.*]] = getelementptr inbounds nuw <{ ptr, i32, i32, ptr, ptr, ptr }>, ptr %[[V1]], i32 0, i32 5
  61 // CHECK-NEXT: %[[V3:.*]] = load ptr, ptr %[[V2]], align 4
  62 // CHECK-NEXT: call void @_Block_object_dispose(ptr %[[V3]], i32 8)
  63 // CHECK-NEXT: ret void
  64
  65 // SYSTEMZ: declare void @_Block_object_dispose(ptr noundef, i32 noundef signext)
  66
  67 typedef double ftype(double);
  68 // It's not clear that we *should* support this syntax, but until that decision
  69 // is made, we should support it properly and not crash.
  70 ftype ^test2 = ^ftype {
  71   return 0;
  72 };
  73
  74 void f3_helper(void (^)(void));
  75 void f3(void) {
  76   _Bool b = 0;
  77   f3_helper(^{ if (b) {} });
  78 }
  79
  80 // The bool can fill in between the header and the long long.
  81 // Add the appropriate amount of padding between them.
  82 void f4_helper(long long (^)(void));
  83 // CHECK-LABEL: define{{.*}} void @f4()
  84 void f4(void) {
  85   _Bool b = 0;
  86   long long ll = 0;
  87   // CHECK: alloca <{ ptr, i32, i32, ptr, ptr, i8, [3 x i8], i64 }>, align 8
  88   f4_helper(^{ if (b) return ll; return 0LL; });
  89 }
  90
  91 // The alignment after rounding up to the align of F5 is actually
  92 // greater than the required alignment.  Don't assert.
  93 struct F5 {
  94   char buffer[32] __attribute((aligned));
  95 };
  96 void f5_helper(void (^)(struct F5 *));
  97 // CHECK-LABEL: define{{.*}} void @f5()
  98 void f5(void) {
  99   struct F5 value;
 100   // CHECK: alloca <{ ptr, i32, i32, ptr, ptr, [12 x i8], [[F5:%.*]] }>, align 16
 101   f5_helper(^(struct F5 *slot) { *slot = value; });
 102 }
 103
 104 void (^b)() = ^{};
 105 int main(void) {
 106    (b?: ^{})();
 107 }
 108 // CHECK: [[ZERO:%.*]] = load ptr, ptr @b
 109 // CHECK-NEXT: [[TB:%.*]] = icmp ne ptr [[ZERO]], null
 110 // CHECK-NEXT: br i1 [[TB]], label [[CT:%.*]], label [[CF:%.*]]
 111 // CHECK:   br label [[CE:%.*]]
 112
 113 // Ensure that we don't emit helper code in copy/dispose routines for variables
 114 // that are const-captured.
 115 void testConstCaptureInCopyAndDestroyHelpers(void) {
 116   const int x = 0;
 117   __block int i;
 118   (^ { i = x; })();
 119 }
 120 // CHECK-LABEL: define{{.*}} void @testConstCaptureInCopyAndDestroyHelpers(
 121 // CHECK: %[[BLOCK_DESCRIPTOR:.*]] = getelementptr inbounds nuw <{ ptr, i32, i32, ptr, ptr, ptr }>, ptr %{{.*}}, i32 0, i32 4
 122 // CHECK: store ptr @[[BLOCK_DESCRIPTOR_TMP21]], ptr %[[BLOCK_DESCRIPTOR]], align 4
 123
 124 // CHECK-LABEL: define internal void @__testConstCaptureInCopyAndDestroyHelpers_block_invoke