1*67e74705SXin Li // RUN: %clang_cc1 %s -emit-llvm -o - -fblocks -triple x86_64-apple-darwin10 | FileCheck %s
2*67e74705SXin Li
3*67e74705SXin Li // CHECK: [[AGG:%.*]] = type { i32 }
4*67e74705SXin Li typedef struct { int v; } Agg;
5*67e74705SXin Li Agg makeAgg(void);
6*67e74705SXin Li
7*67e74705SXin Li // When assigning into a __block variable, ensure that we compute that
8*67e74705SXin Li // address *after* evaluating the RHS when the RHS has the capacity to
9*67e74705SXin Li // cause a block copy. rdar://9309454
test0()10*67e74705SXin Li void test0() {
11*67e74705SXin Li __block Agg a = {100};
12*67e74705SXin Li
13*67e74705SXin Li a = makeAgg();
14*67e74705SXin Li }
15*67e74705SXin Li // CHECK-LABEL: define void @test0()
16*67e74705SXin Li // CHECK: [[A:%.*]] = alloca [[BYREF:%.*]], align 8
17*67e74705SXin Li // CHECK-NEXT: [[TEMP:%.*]] = alloca [[AGG]], align 4
18*67e74705SXin Li // CHECK: [[RESULT:%.*]] = call i32 @makeAgg()
19*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = getelementptr inbounds [[AGG]], [[AGG]]* [[TEMP]], i32 0, i32 0
20*67e74705SXin Li // CHECK-NEXT: store i32 [[RESULT]], i32* [[T0]]
21*67e74705SXin Li // Check that we properly assign into the forwarding pointer.
22*67e74705SXin Li // CHECK-NEXT: [[A_FORWARDING:%.*]] = getelementptr inbounds [[BYREF]], [[BYREF]]* [[A]], i32 0, i32 1
23*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = load [[BYREF]]*, [[BYREF]]** [[A_FORWARDING]]
24*67e74705SXin Li // CHECK-NEXT: [[T1:%.*]] = getelementptr inbounds [[BYREF]], [[BYREF]]* [[T0]], i32 0, i32 4
25*67e74705SXin Li // CHECK-NEXT: [[T2:%.*]] = bitcast [[AGG]]* [[T1]] to i8*
26*67e74705SXin Li // CHECK-NEXT: [[T3:%.*]] = bitcast [[AGG]]* [[TEMP]] to i8*
27*67e74705SXin Li // CHECK-NEXT: call void @llvm.memcpy.p0i8.p0i8.i64(i8* [[T2]], i8* [[T3]], i64 4, i32 4, i1 false)
28*67e74705SXin Li // Verify that there's nothing else significant in the function.
29*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = bitcast [[BYREF]]* [[A]] to i8*
30*67e74705SXin Li // CHECK-NEXT: call void @_Block_object_dispose(i8* [[T0]], i32 8)
31*67e74705SXin Li // CHECK-NEXT: ret void
32*67e74705SXin Li
33*67e74705SXin Li // When chaining assignments into __block variables, make sure we
34*67e74705SXin Li // propagate the actual value into the outer variable.
35*67e74705SXin Li // rdar://11757470
test1()36*67e74705SXin Li void test1() {
37*67e74705SXin Li __block Agg a, b;
38*67e74705SXin Li a = b = makeAgg();
39*67e74705SXin Li }
40*67e74705SXin Li // CHECK-LABEL: define void @test1()
41*67e74705SXin Li // CHECK: [[A:%.*]] = alloca [[A_BYREF:%.*]], align 8
42*67e74705SXin Li // CHECK-NEXT: [[B:%.*]] = alloca [[B_BYREF:%.*]], align 8
43*67e74705SXin Li // CHECK-NEXT: [[TEMP:%.*]] = alloca [[AGG]], align 4
44*67e74705SXin Li // CHECK: [[RESULT:%.*]] = call i32 @makeAgg()
45*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = getelementptr inbounds [[AGG]], [[AGG]]* [[TEMP]], i32 0, i32 0
46*67e74705SXin Li // CHECK-NEXT: store i32 [[RESULT]], i32* [[T0]]
47*67e74705SXin Li // Check that we properly assign into the forwarding pointer, first for b:
48*67e74705SXin Li // CHECK-NEXT: [[B_FORWARDING:%.*]] = getelementptr inbounds [[B_BYREF]], [[B_BYREF]]* [[B]], i32 0, i32 1
49*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = load [[B_BYREF]]*, [[B_BYREF]]** [[B_FORWARDING]]
50*67e74705SXin Li // CHECK-NEXT: [[T1:%.*]] = getelementptr inbounds [[B_BYREF]], [[B_BYREF]]* [[T0]], i32 0, i32 4
51*67e74705SXin Li // CHECK-NEXT: [[T2:%.*]] = bitcast [[AGG]]* [[T1]] to i8*
52*67e74705SXin Li // CHECK-NEXT: [[T3:%.*]] = bitcast [[AGG]]* [[TEMP]] to i8*
53*67e74705SXin Li // CHECK-NEXT: call void @llvm.memcpy.p0i8.p0i8.i64(i8* [[T2]], i8* [[T3]], i64 4, i32 4, i1 false)
54*67e74705SXin Li // Then for 'a':
55*67e74705SXin Li // CHECK-NEXT: [[A_FORWARDING:%.*]] = getelementptr inbounds [[A_BYREF]], [[A_BYREF]]* [[A]], i32 0, i32 1
56*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = load [[A_BYREF]]*, [[A_BYREF]]** [[A_FORWARDING]]
57*67e74705SXin Li // CHECK-NEXT: [[T1:%.*]] = getelementptr inbounds [[A_BYREF]], [[A_BYREF]]* [[T0]], i32 0, i32 4
58*67e74705SXin Li // CHECK-NEXT: [[T2:%.*]] = bitcast [[AGG]]* [[T1]] to i8*
59*67e74705SXin Li // CHECK-NEXT: [[T3:%.*]] = bitcast [[AGG]]* [[TEMP]] to i8*
60*67e74705SXin Li // CHECK-NEXT: call void @llvm.memcpy.p0i8.p0i8.i64(i8* [[T2]], i8* [[T3]], i64 4, i32 4, i1 false)
61*67e74705SXin Li // Verify that there's nothing else significant in the function.
62*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = bitcast [[B_BYREF]]* [[B]] to i8*
63*67e74705SXin Li // CHECK-NEXT: call void @_Block_object_dispose(i8* [[T0]], i32 8)
64*67e74705SXin Li // CHECK-NEXT: [[T0:%.*]] = bitcast [[A_BYREF]]* [[A]] to i8*
65*67e74705SXin Li // CHECK-NEXT: call void @_Block_object_dispose(i8* [[T0]], i32 8)
66*67e74705SXin Li // CHECK-NEXT: ret void
67