1// RUN: %clang_cc1 -triple x86_64-apple-darwin10 -emit-llvm -fblocks -fobjc-arc -fobjc-runtime-has-weak -O2 -disable-llvm-passes -o - %s | FileCheck %s 2 3id g0, g1; 4 5void test0(_Bool cond) { 6 id test0_helper(void) __attribute__((ns_returns_retained)); 7 8 // CHECK-LABEL: define{{.*}} void @test0( 9 // CHECK: [[COND:%.*]] = alloca i8, 10 // CHECK-NEXT: [[X:%.*]] = alloca ptr, 11 // CHECK-NEXT: [[RELVAL:%.*]] = alloca ptr 12 // CHECK-NEXT: [[RELCOND:%.*]] = alloca i1 13 // CHECK-NEXT: zext 14 // CHECK-NEXT: store 15 // CHECK-NEXT: call void @llvm.lifetime.start.p0(i64 8, ptr [[X]]) 16 // CHECK-NEXT: [[T0:%.*]] = load i8, ptr [[COND]] 17 // CHECK-NEXT: [[T1:%.*]] = trunc i8 [[T0]] to i1 18 // CHECK-NEXT: store i1 false, ptr [[RELCOND]] 19 // CHECK-NEXT: br i1 [[T1]], 20 // CHECK: br label 21 // CHECK: [[CALL:%.*]] = call ptr @test0_helper() 22 // CHECK-NEXT: store ptr [[CALL]], ptr [[RELVAL]] 23 // CHECK-NEXT: store i1 true, ptr [[RELCOND]] 24 // CHECK-NEXT: br label 25 // CHECK: [[T0:%.*]] = phi ptr [ null, {{%.*}} ], [ [[CALL]], {{%.*}} ] 26 // CHECK-NEXT: [[T1:%.*]] = call ptr @llvm.objc.retain(ptr [[T0]]) [[NUW:#[0-9]+]] 27 // CHECK-NEXT: store ptr [[T1]], ptr [[X]], 28 // CHECK-NEXT: [[REL:%.*]] = load i1, ptr [[RELCOND]] 29 // CHECK-NEXT: br i1 [[REL]], 30 // CHECK: [[T0:%.*]] = load ptr, ptr [[RELVAL]] 31 // CHECK-NEXT: call void @llvm.objc.release(ptr [[T0]]) [[NUW]] 32 // CHECK-NEXT: br label 33 // CHECK: [[T0:%.*]] = load ptr, ptr [[X]] 34 // CHECK-NEXT: call void @llvm.objc.release(ptr [[T0]]) [[NUW]] 35 // CHECK-NEXT: call void @llvm.lifetime.end.p0(i64 8, ptr [[X]]) 36 // CHECK-NEXT: ret void 37 id x = (cond ? 0 : test0_helper()); 38} 39 40void test1(int cond) { 41 __strong id strong; 42 __weak id weak; 43 extern void test1_sink(id *); 44 test1_sink(cond ? &strong : 0); 45 test1_sink(cond ? &weak : 0); 46 47 // CHECK-LABEL: define{{.*}} void @test1( 48 // CHECK: [[COND:%.*]] = alloca i32 49 // CHECK-NEXT: [[STRONG:%.*]] = alloca ptr 50 // CHECK-NEXT: [[WEAK:%.*]] = alloca ptr 51 // CHECK-NEXT: [[TEMP1:%.*]] = alloca ptr 52 // CHECK-NEXT: [[TEMP2:%.*]] = alloca ptr 53 // CHECK-NEXT: [[CONDCLEANUPSAVE:%.*]] = alloca ptr 54 // CHECK-NEXT: [[CONDCLEANUP:%.*]] = alloca i1 55 // CHECK-NEXT: store i32 56 // CHECK-NEXT: call void @llvm.lifetime.start.p0(i64 8, ptr [[STRONG]]) 57 // CHECK-NEXT: store ptr null, ptr [[STRONG]] 58 // CHECK-NEXT: call void @llvm.lifetime.start.p0(i64 8, ptr [[WEAK]]) 59 // CHECK-NEXT: call ptr @llvm.objc.initWeak(ptr [[WEAK]], ptr null) 60 61 // CHECK-NEXT: [[T0:%.*]] = load i32, ptr [[COND]] 62 // CHECK-NEXT: [[T1:%.*]] = icmp ne i32 [[T0]], 0 63 // CHECK: [[ARG:%.*]] = phi ptr 64 // CHECK-NEXT: [[T0:%.*]] = icmp eq ptr [[ARG]], null 65 // CHECK-NEXT: [[T1:%.*]] = select i1 [[T0]], ptr null, ptr [[TEMP1]] 66 // CHECK-NEXT: br i1 [[T0]], 67 // CHECK: [[T0:%.*]] = load ptr, ptr [[ARG]] 68 // CHECK-NEXT: store ptr [[T0]], ptr [[TEMP1]] 69 // CHECK-NEXT: br label 70 // CHECK: [[W:%.*]] = phi ptr [ [[T0]], {{%.*}} ], [ poison, {{%.*}} ] 71 // CHECK-NEXT: call void @test1_sink(ptr noundef [[T1]]) 72 // CHECK-NEXT: [[T0:%.*]] = icmp eq ptr [[ARG]], null 73 // CHECK-NEXT: br i1 [[T0]], 74 // CHECK: [[T0:%.*]] = load ptr, ptr [[TEMP1]] 75 // CHECK-NEXT: [[T1:%.*]] = call ptr @llvm.objc.retain(ptr [[T0]]) 76 // CHECK-NEXT: call void (...) @llvm.objc.clang.arc.use(ptr [[W]]) [[NUW]] 77 // CHECK-NEXT: [[T2:%.*]] = load ptr, ptr [[ARG]] 78 // CHECK-NEXT: store ptr [[T1]], ptr [[ARG]] 79 // CHECK-NEXT: call void @llvm.objc.release(ptr [[T2]]) 80 // CHECK-NEXT: br label 81 82 // CHECK: [[T0:%.*]] = load i32, ptr [[COND]] 83 // CHECK-NEXT: [[T1:%.*]] = icmp ne i32 [[T0]], 0 84 // CHECK: [[ARG:%.*]] = phi ptr 85 // CHECK-NEXT: [[T0:%.*]] = icmp eq ptr [[ARG]], null 86 // CHECK-NEXT: [[T1:%.*]] = select i1 [[T0]], ptr null, ptr [[TEMP2]] 87 // CHECK-NEXT: store i1 false, ptr [[CONDCLEANUP]] 88 // CHECK-NEXT: br i1 [[T0]], 89 // CHECK: [[T0:%.*]] = call ptr @llvm.objc.loadWeakRetained(ptr [[ARG]]) 90 // CHECK-NEXT: store ptr [[T0]], ptr [[CONDCLEANUPSAVE]] 91 // CHECK-NEXT: store i1 true, ptr [[CONDCLEANUP]] 92 // CHECK-NEXT: store ptr [[T0]], ptr [[TEMP2]] 93 // CHECK-NEXT: br label 94 // CHECK: call void @test1_sink(ptr noundef [[T1]]) 95 // CHECK-NEXT: [[T0:%.*]] = icmp eq ptr [[ARG]], null 96 // CHECK-NEXT: br i1 [[T0]], 97 // CHECK: [[T0:%.*]] = load ptr, ptr [[TEMP2]] 98 // CHECK-NEXT: call ptr @llvm.objc.storeWeak(ptr [[ARG]], ptr [[T0]]) 99 // CHECK-NEXT: br label 100 101 // CHECK: call void @llvm.objc.destroyWeak(ptr [[WEAK]]) 102 // CHECK: call void @llvm.lifetime.end.p0(i64 8, ptr [[WEAK]]) 103 // CHECK: call void @llvm.lifetime.end.p0(i64 8, ptr [[STRONG]]) 104 // CHECK: ret void 105} 106 107// Test that, when emitting an expression at +1 that we can't peephole, 108// we emit the retain inside the full-expression. If we ever peephole 109// +1s of conditional expressions (which we probably ought to), we'll 110// need to find another example of something we need to do this for. 111void test2(int cond) { 112 extern id test2_producer(void); 113 for (id obj in cond ? test2_producer() : (void*) 0) { 114 } 115 116 // CHECK-LABEL: define{{.*}} void @test2( 117 // CHECK: [[COND:%.*]] = alloca i32, 118 // CHECK: alloca ptr 119 // CHECK: [[CLEANUP_SAVE:%.*]] = alloca ptr 120 // CHECK: [[RUN_CLEANUP:%.*]] = alloca i1 121 // Evaluate condition; cleanup disabled by default. 122 // CHECK: [[T0:%.*]] = load i32, ptr [[COND]], 123 // CHECK-NEXT: icmp ne i32 [[T0]], 0 124 // CHECK-NEXT: store i1 false, ptr [[RUN_CLEANUP]] 125 // CHECK-NEXT: br i1 126 // Within true branch, cleanup enabled. 127 // CHECK: [[T1:%.*]] = call ptr @test2_producer() [ "clang.arc.attachedcall"(ptr @llvm.objc.retainAutoreleasedReturnValue) ] 128 // CHECK-NEXT: call void (...) @llvm.objc.clang.arc.noop.use(ptr [[T1]]) 129 // CHECK-NEXT: store ptr [[T1]], ptr [[CLEANUP_SAVE]] 130 // CHECK-NEXT: store i1 true, ptr [[RUN_CLEANUP]] 131 // CHECK-NEXT: br label 132 // Join point for conditional operator; retain immediately. 133 // CHECK: [[T0:%.*]] = phi ptr [ [[T1]], {{%.*}} ], [ null, {{%.*}} ] 134 // CHECK-NEXT: [[RESULT:%.*]] = call ptr @llvm.objc.retain(ptr [[T0]]) 135 // Leaving full-expression; run conditional cleanup. 136 // CHECK-NEXT: [[T0:%.*]] = load i1, ptr [[RUN_CLEANUP]] 137 // CHECK-NEXT: br i1 [[T0]] 138 // CHECK: [[T0:%.*]] = load ptr, ptr [[CLEANUP_SAVE]] 139 // CHECK-NEXT: call void @llvm.objc.release(ptr [[T0]]) 140 // CHECK-NEXT: br label 141 // And way down at the end of the loop: 142 // CHECK: call void @llvm.objc.release(ptr [[RESULT]]) 143} 144 145void test3(int cond) { 146 __strong id *p = cond ? (__strong id[]){g0, g1} : (__strong id[]){g1, g0}; 147 test2(cond); 148 149 // CHECK: define{{.*}} void @test3( 150 // CHECK: %[[P:.*]] = alloca ptr, align 8 151 // CHECK: %[[_COMPOUNDLITERAL:.*]] = alloca [2 x ptr], align 8 152 // CHECK: %[[CLEANUP_COND:.*]] = alloca i1, align 1 153 // CHECK: %[[_COMPOUNDLITERAL1:.*]] = alloca [2 x ptr], align 8 154 // CHECK: %[[CLEANUP_COND4:.*]] = alloca i1, align 1 155 156 // CHECK: %[[V2:.*]] = load ptr, ptr @g0, align 8 157 // CHECK: %[[V3:.*]] = call ptr @llvm.objc.retain(ptr %[[V2]]) 158 // CHECK: store ptr %[[V3]], ptr %[[_COMPOUNDLITERAL]], align 8 159 // CHECK: %[[ARRAYINIT_ELEMENT:.*]] = getelementptr inbounds ptr, ptr %[[_COMPOUNDLITERAL]], i64 1 160 // CHECK: %[[V4:.*]] = load ptr, ptr @g1, align 8 161 // CHECK: %[[V5:.*]] = call ptr @llvm.objc.retain(ptr %[[V4]]) 162 // CHECK: store ptr %[[V5]], ptr %[[ARRAYINIT_ELEMENT]], align 8 163 // CHECK: store i1 true, ptr %[[CLEANUP_COND]], align 1 164 // CHECK: %[[ARRAYDECAY:.*]] = getelementptr inbounds [2 x ptr], ptr %[[_COMPOUNDLITERAL]], i64 0, i64 0 165 166 // CHECK: %[[V6:.*]] = load ptr, ptr @g1, align 8 167 // CHECK: %[[V7:.*]] = call ptr @llvm.objc.retain(ptr %[[V6]]) 168 // CHECK: store ptr %[[V7]], ptr %[[_COMPOUNDLITERAL1]], align 8 169 // CHECK: %[[ARRAYINIT_ELEMENT3:.*]] = getelementptr inbounds ptr, ptr %[[_COMPOUNDLITERAL1]], i64 1 170 // CHECK: %[[V8:.*]] = load ptr, ptr @g0, align 8 171 // CHECK: %[[V9:.*]] = call ptr @llvm.objc.retain(ptr %[[V8]]) 172 // CHECK: store ptr %[[V9]], ptr %[[ARRAYINIT_ELEMENT3]], align 8 173 // CHECK: store i1 true, ptr %[[CLEANUP_COND4]], align 1 174 // CHECK: %[[ARRAYDECAY5:.*]] = getelementptr inbounds [2 x ptr], ptr %[[_COMPOUNDLITERAL1]], i64 0, i64 0 175 176 // CHECK: %[[COND6:.*]] = phi ptr [ %[[ARRAYDECAY]], %{{.*}} ], [ %[[ARRAYDECAY5]], %{{.*}} ] 177 // CHECK: store ptr %[[COND6]], ptr %[[P]], align 8 178 // CHECK: call void @test2( 179 180 // CHECK: %[[ARRAY_BEGIN:.*]] = getelementptr inbounds [2 x ptr], ptr %[[_COMPOUNDLITERAL1]], i32 0, i32 0 181 // CHECK: %[[V11:.*]] = getelementptr inbounds ptr, ptr %[[ARRAY_BEGIN]], i64 2 182 183 // CHECK: %[[ARRAYDESTROY_ELEMENTPAST:.*]] = phi ptr [ %[[V11]], %{{.*}} ], [ %[[ARRAYDESTROY_ELEMENT:.*]], %{{.*}} ] 184 // CHECK: %[[ARRAYDESTROY_ELEMENT]] = getelementptr inbounds ptr, ptr %[[ARRAYDESTROY_ELEMENTPAST]], i64 -1 185 // CHECK: %[[V12:.*]] = load ptr, ptr %[[ARRAYDESTROY_ELEMENT]], align 8 186 // CHECK: call void @llvm.objc.release(ptr %[[V12]]) 187 188 // CHECK: %[[ARRAY_BEGIN10:.*]] = getelementptr inbounds [2 x ptr], ptr %[[_COMPOUNDLITERAL]], i32 0, i32 0 189 // CHECK: %[[V13:.*]] = getelementptr inbounds ptr, ptr %[[ARRAY_BEGIN10]], i64 2 190 191 // CHECK: %[[ARRAYDESTROY_ELEMENTPAST12:.*]] = phi ptr [ %[[V13]], %{{.*}} ], [ %[[ARRAYDESTROY_ELEMENT13:.*]], %{{.*}} ] 192 // CHECK: %[[ARRAYDESTROY_ELEMENT13]] = getelementptr inbounds ptr, ptr %[[ARRAYDESTROY_ELEMENTPAST12]], i64 -1 193 // CHECK: %[[V14:.*]] = load ptr, ptr %[[ARRAYDESTROY_ELEMENT13]], align 8 194 // CHECK: call void @llvm.objc.release(ptr %[[V14]]) 195} 196 197// CHECK: attributes [[NUW]] = { nounwind } 198