1// RUN: %clang_cc1 -triple x86_64-apple-darwin10 -emit-llvm -fblocks -fobjc-arc -fobjc-runtime-has-weak -O2 -disable-llvm-optzns -o - %s | FileCheck %s
2
3void test0(_Bool cond) {
4  id test0_helper(void) __attribute__((ns_returns_retained));
5
6  // CHECK-LABEL:      define void @test0(
7  // CHECK:      [[COND:%.*]] = alloca i8,
8  // CHECK-NEXT: [[X:%.*]] = alloca i8*,
9  // CHECK-NEXT: [[RELVAL:%.*]] = alloca i8*
10  // CHECK-NEXT: [[RELCOND:%.*]] = alloca i1
11  // CHECK-NEXT: zext
12  // CHECK-NEXT: store
13  // CHECK-NEXT: [[T0:%.*]] = load i8* [[COND]]
14  // CHECK-NEXT: [[T1:%.*]] = trunc i8 [[T0]] to i1
15  // CHECK-NEXT: store i1 false, i1* [[RELCOND]]
16  // CHECK-NEXT: br i1 [[T1]],
17  // CHECK:      br label
18  // CHECK:      [[CALL:%.*]] = call i8* @test0_helper()
19  // CHECK-NEXT: store i8* [[CALL]], i8** [[RELVAL]]
20  // CHECK-NEXT: store i1 true, i1* [[RELCOND]]
21  // CHECK-NEXT: br label
22  // CHECK:      [[T0:%.*]] = phi i8* [ null, {{%.*}} ], [ [[CALL]], {{%.*}} ]
23  // CHECK-NEXT: [[T1:%.*]] = call i8* @objc_retain(i8* [[T0]]) [[NUW:#[0-9]+]]
24  // CHECK-NEXT: store i8* [[T1]], i8** [[X]],
25  // CHECK-NEXT: [[REL:%.*]] = load i1* [[RELCOND]]
26  // CHECK-NEXT: br i1 [[REL]],
27  // CHECK:      [[T0:%.*]] = load i8** [[RELVAL]]
28  // CHECK-NEXT: call void @objc_release(i8* [[T0]]) [[NUW]]
29  // CHECK-NEXT: br label
30  // CHECK:      [[T0:%.*]] = load i8** [[X]]
31  // CHECK-NEXT: call void @objc_release(i8* [[T0]]) [[NUW]]
32  // CHECK-NEXT: ret void
33  id x = (cond ? 0 : test0_helper());
34}
35
36void test1(int cond) {
37  __strong id strong;
38  __weak id weak;
39  extern void test1_sink(id *);
40  test1_sink(cond ? &strong : 0);
41  test1_sink(cond ? &weak : 0);
42
43  // CHECK-LABEL:    define void @test1(
44  // CHECK:      [[COND:%.*]] = alloca i32
45  // CHECK-NEXT: [[STRONG:%.*]] = alloca i8*
46  // CHECK-NEXT: [[WEAK:%.*]] = alloca i8*
47  // CHECK-NEXT: [[TEMP1:%.*]] = alloca i8*
48  // CHECK-NEXT: [[TEMP2:%.*]] = alloca i8*
49  // CHECK-NEXT: [[CONDCLEANUPSAVE:%.*]] = alloca i8*
50  // CHECK-NEXT: [[CONDCLEANUP:%.*]] = alloca i1
51  // CHECK-NEXT: store i32
52  // CHECK-NEXT: store i8* null, i8** [[STRONG]]
53  // CHECK-NEXT: call i8* @objc_initWeak(i8** [[WEAK]], i8* null)
54
55  // CHECK-NEXT: [[T0:%.*]] = load i32* [[COND]]
56  // CHECK-NEXT: [[T1:%.*]] = icmp ne i32 [[T0]], 0
57  // CHECK:      [[ARG:%.*]] = phi i8**
58  // CHECK-NEXT: [[T0:%.*]] = icmp eq i8** [[ARG]], null
59  // CHECK-NEXT: [[T1:%.*]] = select i1 [[T0]], i8** null, i8** [[TEMP1]]
60  // CHECK-NEXT: br i1 [[T0]],
61  // CHECK:      [[T0:%.*]] = load i8** [[ARG]]
62  // CHECK-NEXT: store i8* [[T0]], i8** [[TEMP1]]
63  // CHECK-NEXT: br label
64  // CHECK:      [[W:%.*]] = phi i8* [ [[T0]], {{%.*}} ], [ undef, {{%.*}} ]
65  // CHECK-NEXT: call void @test1_sink(i8** [[T1]])
66  // CHECK-NEXT: [[T0:%.*]] = icmp eq i8** [[ARG]], null
67  // CHECK-NEXT: br i1 [[T0]],
68  // CHECK:      [[T0:%.*]] = load i8** [[TEMP1]]
69  // CHECK-NEXT: [[T1:%.*]] = call i8* @objc_retain(i8* [[T0]])
70  // CHECK-NEXT: call void (...)* @clang.arc.use(i8* [[W]]) [[NUW]]
71  // CHECK-NEXT: [[T2:%.*]] = load i8** [[ARG]]
72  // CHECK-NEXT: store i8* [[T1]], i8** [[ARG]]
73  // CHECK-NEXT: call void @objc_release(i8* [[T2]])
74  // CHECK-NEXT: br label
75
76  // CHECK:      [[T0:%.*]] = load i32* [[COND]]
77  // CHECK-NEXT: [[T1:%.*]] = icmp ne i32 [[T0]], 0
78  // CHECK:      [[ARG:%.*]] = phi i8**
79  // CHECK-NEXT: [[T0:%.*]] = icmp eq i8** [[ARG]], null
80  // CHECK-NEXT: [[T1:%.*]] = select i1 [[T0]], i8** null, i8** [[TEMP2]]
81  // CHECK-NEXT: store i1 false, i1* [[CONDCLEANUP]]
82  // CHECK-NEXT: br i1 [[T0]],
83  // CHECK:      [[T0:%.*]] = call i8* @objc_loadWeakRetained(i8** [[ARG]])
84  // CHECK-NEXT: store i8* [[T0]], i8** [[CONDCLEANUPSAVE]]
85  // CHECK-NEXT: store i1 true, i1* [[CONDCLEANUP]]
86  // CHECK-NEXT: store i8* [[T0]], i8** [[TEMP2]]
87  // CHECK-NEXT: br label
88  // CHECK:      call void @test1_sink(i8** [[T1]])
89  // CHECK-NEXT: [[T0:%.*]] = icmp eq i8** [[ARG]], null
90  // CHECK-NEXT: br i1 [[T0]],
91  // CHECK:      [[T0:%.*]] = load i8** [[TEMP2]]
92  // CHECK-NEXT: call i8* @objc_storeWeak(i8** [[ARG]], i8* [[T0]])
93  // CHECK-NEXT: br label
94
95  // CHECK:      call void @objc_destroyWeak(i8** [[WEAK]])
96  // CHECK:      ret void
97}
98
99// rdar://13113981
100// Test that, when emitting an expression at +1 that we can't peephole,
101// we emit the retain inside the full-expression.  If we ever peephole
102// +1s of conditional expressions (which we probably ought to), we'll
103// need to find another example of something we need to do this for.
104void test2(int cond) {
105  extern id test2_producer(void);
106  for (id obj in cond ? test2_producer() : (void*) 0) {
107  }
108
109  // CHECK-LABEL:    define void @test2(
110  // CHECK:      [[COND:%.*]] = alloca i32,
111  // CHECK:      alloca i8*
112  // CHECK:      [[CLEANUP_SAVE:%.*]] = alloca i8*
113  // CHECK:      [[RUN_CLEANUP:%.*]] = alloca i1
114  //   Evaluate condition; cleanup disabled by default.
115  // CHECK:      [[T0:%.*]] = load i32* [[COND]],
116  // CHECK-NEXT: icmp ne i32 [[T0]], 0
117  // CHECK-NEXT: store i1 false, i1* [[RUN_CLEANUP]]
118  // CHECK-NEXT: br i1
119  //   Within true branch, cleanup enabled.
120  // CHECK:      [[T0:%.*]] = call i8* @test2_producer()
121  // CHECK-NEXT: [[T1:%.*]] = call i8* @objc_retainAutoreleasedReturnValue(i8* [[T0]])
122  // CHECK-NEXT: store i8* [[T1]], i8** [[CLEANUP_SAVE]]
123  // CHECK-NEXT: store i1 true, i1* [[RUN_CLEANUP]]
124  // CHECK-NEXT: br label
125  //   Join point for conditional operator; retain immediately.
126  // CHECK:      [[T0:%.*]] = phi i8* [ [[T1]], {{%.*}} ], [ null, {{%.*}} ]
127  // CHECK-NEXT: [[RESULT:%.*]] = call i8* @objc_retain(i8* [[T0]])
128  //   Leaving full-expression; run conditional cleanup.
129  // CHECK-NEXT: [[T0:%.*]] = load i1* [[RUN_CLEANUP]]
130  // CHECK-NEXT: br i1 [[T0]]
131  // CHECK:      [[T0:%.*]] = load i8** [[CLEANUP_SAVE]]
132  // CHECK-NEXT: call void @objc_release(i8* [[T0]])
133  // CHECK-NEXT: br label
134  //   And way down at the end of the loop:
135  // CHECK:      call void @objc_release(i8* [[RESULT]])
136}
137
138// CHECK: attributes [[NUW]] = { nounwind }
139