blob: 2bfa49662576b77823d012bad4345a971359143a [file] [log] [blame]
Anders Carlsson55085182007-08-21 17:43:55 +00001//===---- CGBuiltin.cpp - Emit LLVM Code for builtins ---------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Anders Carlsson55085182007-08-21 17:43:55 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This contains code to emit Objective-C code as LLVM code.
11//
12//===----------------------------------------------------------------------===//
13
Devang Patelbcbd03a2011-01-19 01:36:36 +000014#include "CGDebugInfo.h"
Ted Kremenek2979ec72008-04-09 15:51:31 +000015#include "CGObjCRuntime.h"
Anders Carlsson55085182007-08-21 17:43:55 +000016#include "CodeGenFunction.h"
17#include "CodeGenModule.h"
Daniel Dunbar85c59ed2008-08-29 08:11:39 +000018#include "clang/AST/ASTContext.h"
Daniel Dunbarc4a1dea2008-08-11 05:35:13 +000019#include "clang/AST/DeclObjC.h"
Chris Lattner16f00492009-04-26 01:32:48 +000020#include "clang/AST/StmtObjC.h"
Daniel Dunbare66f4e32008-09-03 00:27:26 +000021#include "clang/Basic/Diagnostic.h"
Anders Carlsson3d8400d2008-08-30 19:51:14 +000022#include "llvm/ADT/STLExtras.h"
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +000023#include "llvm/Target/TargetData.h"
Anders Carlsson55085182007-08-21 17:43:55 +000024using namespace clang;
25using namespace CodeGen;
26
Chris Lattner8fdf3282008-06-24 17:04:18 +000027/// Emits an instance of NSConstantString representing the object.
Mike Stump1eb44332009-09-09 15:08:12 +000028llvm::Value *CodeGenFunction::EmitObjCStringLiteral(const ObjCStringLiteral *E)
Daniel Dunbar71fcec92008-11-25 21:53:21 +000029{
David Chisnall0d13f6f2010-01-23 02:40:42 +000030 llvm::Constant *C =
31 CGM.getObjCRuntime().GenerateConstantString(E->getString());
Daniel Dunbared7c6182008-08-20 00:28:19 +000032 // FIXME: This bitcast should just be made an invariant on the Runtime.
Owen Anderson3c4972d2009-07-29 18:54:39 +000033 return llvm::ConstantExpr::getBitCast(C, ConvertType(E->getType()));
Chris Lattner8fdf3282008-06-24 17:04:18 +000034}
35
36/// Emit a selector.
37llvm::Value *CodeGenFunction::EmitObjCSelectorExpr(const ObjCSelectorExpr *E) {
38 // Untyped selector.
39 // Note that this implementation allows for non-constant strings to be passed
40 // as arguments to @selector(). Currently, the only thing preventing this
41 // behaviour is the type checking in the front end.
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +000042 return CGM.getObjCRuntime().GetSelector(Builder, E->getSelector());
Chris Lattner8fdf3282008-06-24 17:04:18 +000043}
44
Daniel Dunbared7c6182008-08-20 00:28:19 +000045llvm::Value *CodeGenFunction::EmitObjCProtocolExpr(const ObjCProtocolExpr *E) {
46 // FIXME: This should pass the Decl not the name.
47 return CGM.getObjCRuntime().GenerateProtocolRef(Builder, E->getProtocol());
48}
Chris Lattner8fdf3282008-06-24 17:04:18 +000049
50
John McCallef072fd2010-05-22 01:48:05 +000051RValue CodeGenFunction::EmitObjCMessageExpr(const ObjCMessageExpr *E,
52 ReturnValueSlot Return) {
Chris Lattner8fdf3282008-06-24 17:04:18 +000053 // Only the lookup mechanism and first two arguments of the method
54 // implementation vary between runtimes. We can get the receiver and
55 // arguments in generic code.
Mike Stump1eb44332009-09-09 15:08:12 +000056
Daniel Dunbar208ff5e2008-08-11 18:12:00 +000057 CGObjCRuntime &Runtime = CGM.getObjCRuntime();
Chris Lattner8fdf3282008-06-24 17:04:18 +000058 bool isSuperMessage = false;
Daniel Dunbarf56f1912008-08-25 08:19:24 +000059 bool isClassMessage = false;
David Chisnallc6cd5fd2010-04-28 19:33:36 +000060 ObjCInterfaceDecl *OID = 0;
Chris Lattner8fdf3282008-06-24 17:04:18 +000061 // Find the receiver
Daniel Dunbar0b647a62010-04-22 03:17:06 +000062 llvm::Value *Receiver = 0;
Douglas Gregor04badcf2010-04-21 00:45:42 +000063 switch (E->getReceiverKind()) {
64 case ObjCMessageExpr::Instance:
65 Receiver = EmitScalarExpr(E->getInstanceReceiver());
66 break;
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +000067
Douglas Gregor04badcf2010-04-21 00:45:42 +000068 case ObjCMessageExpr::Class: {
John McCall3031c632010-05-17 20:12:43 +000069 const ObjCObjectType *ObjTy
70 = E->getClassReceiver()->getAs<ObjCObjectType>();
71 assert(ObjTy && "Invalid Objective-C class message send");
72 OID = ObjTy->getInterface();
73 assert(OID && "Invalid Objective-C class message send");
David Chisnallc6cd5fd2010-04-28 19:33:36 +000074 Receiver = Runtime.GetClass(Builder, OID);
Daniel Dunbarf56f1912008-08-25 08:19:24 +000075 isClassMessage = true;
Douglas Gregor04badcf2010-04-21 00:45:42 +000076 break;
77 }
78
79 case ObjCMessageExpr::SuperInstance:
Chris Lattner8fdf3282008-06-24 17:04:18 +000080 Receiver = LoadObjCSelf();
Douglas Gregor04badcf2010-04-21 00:45:42 +000081 isSuperMessage = true;
82 break;
83
84 case ObjCMessageExpr::SuperClass:
85 Receiver = LoadObjCSelf();
86 isSuperMessage = true;
87 isClassMessage = true;
88 break;
Chris Lattner8fdf3282008-06-24 17:04:18 +000089 }
90
Daniel Dunbar19cd87e2008-08-30 03:02:31 +000091 CallArgList Args;
Anders Carlsson131038e2009-04-18 20:29:27 +000092 EmitCallArgs(Args, E->getMethodDecl(), E->arg_begin(), E->arg_end());
Mike Stump1eb44332009-09-09 15:08:12 +000093
Anders Carlsson7e70fb22010-06-21 20:59:55 +000094 QualType ResultType =
95 E->getMethodDecl() ? E->getMethodDecl()->getResultType() : E->getType();
96
Chris Lattner8fdf3282008-06-24 17:04:18 +000097 if (isSuperMessage) {
Chris Lattner9384c762008-06-26 04:42:20 +000098 // super is only valid in an Objective-C method
99 const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000100 bool isCategoryImpl = isa<ObjCCategoryImplDecl>(OMD->getDeclContext());
Anders Carlsson7e70fb22010-06-21 20:59:55 +0000101 return Runtime.GenerateMessageSendSuper(*this, Return, ResultType,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000102 E->getSelector(),
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000103 OMD->getClassInterface(),
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000104 isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000105 Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000106 isClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000107 Args,
108 E->getMethodDecl());
Chris Lattner8fdf3282008-06-24 17:04:18 +0000109 }
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000110
Anders Carlsson7e70fb22010-06-21 20:59:55 +0000111 return Runtime.GenerateMessageSend(*this, Return, ResultType,
John McCallef072fd2010-05-22 01:48:05 +0000112 E->getSelector(),
David Chisnallc6cd5fd2010-04-28 19:33:36 +0000113 Receiver, Args, OID,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000114 E->getMethodDecl());
Anders Carlsson55085182007-08-21 17:43:55 +0000115}
116
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000117/// StartObjCMethod - Begin emission of an ObjCMethod. This generates
118/// the LLVM function and sets the other context used by
119/// CodeGenFunction.
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000120void CodeGenFunction::StartObjCMethod(const ObjCMethodDecl *OMD,
121 const ObjCContainerDecl *CD) {
John McCalld26bc762011-03-09 04:27:21 +0000122 FunctionArgList args;
Devang Patel4800ea62010-04-05 21:09:15 +0000123 // Check if we should generate debug info for this method.
Devang Patelaa112892011-03-07 18:45:56 +0000124 if (CGM.getModuleDebugInfo() && !OMD->hasAttr<NoDebugAttr>())
125 DebugInfo = CGM.getModuleDebugInfo();
Devang Patel4800ea62010-04-05 21:09:15 +0000126
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000127 llvm::Function *Fn = CGM.getObjCRuntime().GenerateMethod(OMD, CD);
Daniel Dunbarf80519b2008-09-04 23:41:35 +0000128
Daniel Dunbar0e4f40e2009-04-17 00:48:04 +0000129 const CGFunctionInfo &FI = CGM.getTypes().getFunctionInfo(OMD);
130 CGM.SetInternalFunctionAttributes(OMD, Fn, FI);
Chris Lattner41110242008-06-17 18:05:57 +0000131
John McCalld26bc762011-03-09 04:27:21 +0000132 args.push_back(OMD->getSelfDecl());
133 args.push_back(OMD->getCmdDecl());
Chris Lattner41110242008-06-17 18:05:57 +0000134
Chris Lattner89951a82009-02-20 18:43:26 +0000135 for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(),
136 E = OMD->param_end(); PI != E; ++PI)
John McCalld26bc762011-03-09 04:27:21 +0000137 args.push_back(*PI);
Chris Lattner41110242008-06-17 18:05:57 +0000138
Peter Collingbourne14110472011-01-13 18:57:25 +0000139 CurGD = OMD;
140
John McCalld26bc762011-03-09 04:27:21 +0000141 StartFunction(OMD, OMD->getResultType(), Fn, FI, args, OMD->getLocStart());
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000142}
Daniel Dunbarb7ec2462008-08-16 03:19:19 +0000143
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000144void CodeGenFunction::GenerateObjCGetterBody(ObjCIvarDecl *Ivar,
145 bool IsAtomic, bool IsStrong) {
146 LValue LV = EmitLValueForIvar(TypeOfSelfObject(), LoadObjCSelf(),
147 Ivar, 0);
148 llvm::Value *GetCopyStructFn =
149 CGM.getObjCRuntime().GetGetStructFunction();
150 CodeGenTypes &Types = CGM.getTypes();
151 // objc_copyStruct (ReturnValue, &structIvar,
152 // sizeof (Type of Ivar), isAtomic, false);
153 CallArgList Args;
154 RValue RV = RValue::get(Builder.CreateBitCast(ReturnValue,
155 Types.ConvertType(getContext().VoidPtrTy)));
156 Args.push_back(std::make_pair(RV, getContext().VoidPtrTy));
157 RV = RValue::get(Builder.CreateBitCast(LV.getAddress(),
158 Types.ConvertType(getContext().VoidPtrTy)));
159 Args.push_back(std::make_pair(RV, getContext().VoidPtrTy));
160 // sizeof (Type of Ivar)
161 CharUnits Size = getContext().getTypeSizeInChars(Ivar->getType());
162 llvm::Value *SizeVal =
163 llvm::ConstantInt::get(Types.ConvertType(getContext().LongTy),
164 Size.getQuantity());
165 Args.push_back(std::make_pair(RValue::get(SizeVal),
166 getContext().LongTy));
167 llvm::Value *isAtomic =
168 llvm::ConstantInt::get(Types.ConvertType(getContext().BoolTy),
169 IsAtomic ? 1 : 0);
170 Args.push_back(std::make_pair(RValue::get(isAtomic),
171 getContext().BoolTy));
172 llvm::Value *hasStrong =
173 llvm::ConstantInt::get(Types.ConvertType(getContext().BoolTy),
174 IsStrong ? 1 : 0);
175 Args.push_back(std::make_pair(RValue::get(hasStrong),
176 getContext().BoolTy));
177 EmitCall(Types.getFunctionInfo(getContext().VoidTy, Args,
178 FunctionType::ExtInfo()),
179 GetCopyStructFn, ReturnValueSlot(), Args);
180}
181
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000182/// Generate an Objective-C method. An Objective-C method is a C function with
Mike Stump1eb44332009-09-09 15:08:12 +0000183/// its pointer, name, and types registered in the class struture.
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000184void CodeGenFunction::GenerateObjCMethod(const ObjCMethodDecl *OMD) {
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000185 StartObjCMethod(OMD, OMD->getClassInterface());
Argyrios Kyrtzidis6fb0aee2009-06-30 02:35:26 +0000186 EmitStmt(OMD->getBody());
187 FinishFunction(OMD->getBodyRBrace());
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000188}
189
Mike Stumpf5408fe2009-05-16 07:57:57 +0000190// FIXME: I wasn't sure about the synthesis approach. If we end up generating an
191// AST for the whole body we can just fall back to having a GenerateFunction
192// which takes the body Stmt.
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000193
194/// GenerateObjCGetter - Generate an Objective-C property getter
Steve Naroff489034c2009-01-10 22:55:25 +0000195/// function. The given Decl must be an ObjCImplementationDecl. @synthesize
196/// is illegal within a category.
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000197void CodeGenFunction::GenerateObjCGetter(ObjCImplementationDecl *IMP,
198 const ObjCPropertyImplDecl *PID) {
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000199 ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000200 const ObjCPropertyDecl *PD = PID->getPropertyDecl();
Fariborz Jahanian15bd5882010-04-13 18:32:24 +0000201 bool IsAtomic =
202 !(PD->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_nonatomic);
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000203 ObjCMethodDecl *OMD = PD->getGetterMethodDecl();
204 assert(OMD && "Invalid call to generate getter (empty method)");
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000205 StartObjCMethod(OMD, IMP->getClassInterface());
Fariborz Jahanian15bd5882010-04-13 18:32:24 +0000206
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000207 // Determine if we should use an objc_getProperty call for
Fariborz Jahanian447d7ae2008-12-08 23:56:17 +0000208 // this. Non-atomic properties are directly evaluated.
209 // atomic 'copy' and 'retain' properties are also directly
210 // evaluated in gc-only mode.
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000211 if (CGM.getLangOptions().getGCMode() != LangOptions::GCOnly &&
Fariborz Jahanian15bd5882010-04-13 18:32:24 +0000212 IsAtomic &&
Fariborz Jahanian447d7ae2008-12-08 23:56:17 +0000213 (PD->getSetterKind() == ObjCPropertyDecl::Copy ||
214 PD->getSetterKind() == ObjCPropertyDecl::Retain)) {
Mike Stump1eb44332009-09-09 15:08:12 +0000215 llvm::Value *GetPropertyFn =
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000216 CGM.getObjCRuntime().GetPropertyGetFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000217
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000218 if (!GetPropertyFn) {
219 CGM.ErrorUnsupported(PID, "Obj-C getter requiring atomic copy");
220 FinishFunction();
221 return;
222 }
223
224 // Return (ivar-type) objc_getProperty((id) self, _cmd, offset, true).
225 // FIXME: Can't this be simpler? This might even be worse than the
226 // corresponding gcc code.
227 CodeGenTypes &Types = CGM.getTypes();
228 ValueDecl *Cmd = OMD->getCmdDecl();
229 llvm::Value *CmdVal = Builder.CreateLoad(LocalDeclMap[Cmd], "cmd");
230 QualType IdTy = getContext().getObjCIdType();
Mike Stump1eb44332009-09-09 15:08:12 +0000231 llvm::Value *SelfAsId =
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000232 Builder.CreateBitCast(LoadObjCSelf(), Types.ConvertType(IdTy));
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000233 llvm::Value *Offset = EmitIvarOffset(IMP->getClassInterface(), Ivar);
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000234 llvm::Value *True =
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000235 llvm::ConstantInt::get(Types.ConvertType(getContext().BoolTy), 1);
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000236 CallArgList Args;
237 Args.push_back(std::make_pair(RValue::get(SelfAsId), IdTy));
238 Args.push_back(std::make_pair(RValue::get(CmdVal), Cmd->getType()));
239 Args.push_back(std::make_pair(RValue::get(Offset), getContext().LongTy));
240 Args.push_back(std::make_pair(RValue::get(True), getContext().BoolTy));
Daniel Dunbare4be5a62009-02-03 23:43:59 +0000241 // FIXME: We shouldn't need to get the function info here, the
242 // runtime already should have computed it to build the function.
John McCall04a67a62010-02-05 21:31:56 +0000243 RValue RV = EmitCall(Types.getFunctionInfo(PD->getType(), Args,
Rafael Espindola264ba482010-03-30 20:24:48 +0000244 FunctionType::ExtInfo()),
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000245 GetPropertyFn, ReturnValueSlot(), Args);
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000246 // We need to fix the type here. Ivars with copy & retain are
247 // always objects so we don't need to worry about complex or
248 // aggregates.
Mike Stump1eb44332009-09-09 15:08:12 +0000249 RV = RValue::get(Builder.CreateBitCast(RV.getScalarVal(),
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000250 Types.ConvertType(PD->getType())));
251 EmitReturnOfRValue(RV, PD->getType());
252 } else {
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000253 const llvm::Triple &Triple = getContext().Target.getTriple();
254 QualType IVART = Ivar->getType();
255 if (IsAtomic &&
256 IVART->isScalarType() &&
257 (Triple.getArch() == llvm::Triple::arm ||
258 Triple.getArch() == llvm::Triple::thumb) &&
259 (getContext().getTypeSizeInChars(IVART)
260 > CharUnits::fromQuantity(4)) &&
261 CGM.getObjCRuntime().GetGetStructFunction()) {
262 GenerateObjCGetterBody(Ivar, true, false);
263 }
264 else if (IVART->isAnyComplexType()) {
Fariborz Jahanian97a73cd2010-05-06 15:45:36 +0000265 LValue LV = EmitLValueForIvar(TypeOfSelfObject(), LoadObjCSelf(),
266 Ivar, 0);
Fariborz Jahanian1b23fe62010-03-25 21:56:43 +0000267 ComplexPairTy Pair = LoadComplexFromAddr(LV.getAddress(),
268 LV.isVolatileQualified());
269 StoreComplexToAddr(Pair, ReturnValue, LV.isVolatileQualified());
270 }
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000271 else if (hasAggregateLLVMType(IVART)) {
Fariborz Jahanian15bd5882010-04-13 18:32:24 +0000272 bool IsStrong = false;
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000273 if ((IsAtomic || (IsStrong = IvarTypeWithAggrGCObjects(IVART)))
Fariborz Jahanian0b2bd472010-04-13 00:38:05 +0000274 && CurFnInfo->getReturnInfo().getKind() == ABIArgInfo::Indirect
David Chisnall8fac25d2010-12-26 22:13:16 +0000275 && CGM.getObjCRuntime().GetGetStructFunction()) {
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000276 GenerateObjCGetterBody(Ivar, IsAtomic, IsStrong);
Fariborz Jahanian0b2bd472010-04-13 00:38:05 +0000277 }
Fariborz Jahanian97a73cd2010-05-06 15:45:36 +0000278 else {
279 if (PID->getGetterCXXConstructor()) {
280 ReturnStmt *Stmt =
281 new (getContext()) ReturnStmt(SourceLocation(),
Douglas Gregor5077c382010-05-15 06:01:05 +0000282 PID->getGetterCXXConstructor(),
283 0);
Fariborz Jahanian97a73cd2010-05-06 15:45:36 +0000284 EmitReturnStmt(*Stmt);
285 }
286 else {
287 LValue LV = EmitLValueForIvar(TypeOfSelfObject(), LoadObjCSelf(),
288 Ivar, 0);
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000289 EmitAggregateCopy(ReturnValue, LV.getAddress(), IVART);
Fariborz Jahanian97a73cd2010-05-06 15:45:36 +0000290 }
291 }
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000292 }
293 else {
294 LValue LV = EmitLValueForIvar(TypeOfSelfObject(), LoadObjCSelf(),
Fariborz Jahanian97a73cd2010-05-06 15:45:36 +0000295 Ivar, 0);
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000296 CodeGenTypes &Types = CGM.getTypes();
297 RValue RV = EmitLoadOfLValue(LV, IVART);
298 RV = RValue::get(Builder.CreateBitCast(RV.getScalarVal(),
299 Types.ConvertType(PD->getType())));
300 EmitReturnOfRValue(RV, PD->getType());
Fariborz Jahanianed1d29d2009-03-03 18:49:40 +0000301 }
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000302 }
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000303
304 FinishFunction();
305}
306
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000307void CodeGenFunction::GenerateObjCAtomicSetterBody(ObjCMethodDecl *OMD,
308 ObjCIvarDecl *Ivar) {
309 // objc_copyStruct (&structIvar, &Arg,
310 // sizeof (struct something), true, false);
311 llvm::Value *GetCopyStructFn =
312 CGM.getObjCRuntime().GetSetStructFunction();
313 CodeGenTypes &Types = CGM.getTypes();
314 CallArgList Args;
315 LValue LV = EmitLValueForIvar(TypeOfSelfObject(), LoadObjCSelf(), Ivar, 0);
316 RValue RV =
317 RValue::get(Builder.CreateBitCast(LV.getAddress(),
318 Types.ConvertType(getContext().VoidPtrTy)));
319 Args.push_back(std::make_pair(RV, getContext().VoidPtrTy));
320 llvm::Value *Arg = LocalDeclMap[*OMD->param_begin()];
321 llvm::Value *ArgAsPtrTy =
322 Builder.CreateBitCast(Arg,
323 Types.ConvertType(getContext().VoidPtrTy));
324 RV = RValue::get(ArgAsPtrTy);
325 Args.push_back(std::make_pair(RV, getContext().VoidPtrTy));
326 // sizeof (Type of Ivar)
327 CharUnits Size = getContext().getTypeSizeInChars(Ivar->getType());
328 llvm::Value *SizeVal =
329 llvm::ConstantInt::get(Types.ConvertType(getContext().LongTy),
330 Size.getQuantity());
331 Args.push_back(std::make_pair(RValue::get(SizeVal),
332 getContext().LongTy));
333 llvm::Value *True =
334 llvm::ConstantInt::get(Types.ConvertType(getContext().BoolTy), 1);
335 Args.push_back(std::make_pair(RValue::get(True), getContext().BoolTy));
336 llvm::Value *False =
337 llvm::ConstantInt::get(Types.ConvertType(getContext().BoolTy), 0);
338 Args.push_back(std::make_pair(RValue::get(False), getContext().BoolTy));
339 EmitCall(Types.getFunctionInfo(getContext().VoidTy, Args,
340 FunctionType::ExtInfo()),
341 GetCopyStructFn, ReturnValueSlot(), Args);
342}
343
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000344/// GenerateObjCSetter - Generate an Objective-C property setter
Steve Naroff489034c2009-01-10 22:55:25 +0000345/// function. The given Decl must be an ObjCImplementationDecl. @synthesize
346/// is illegal within a category.
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000347void CodeGenFunction::GenerateObjCSetter(ObjCImplementationDecl *IMP,
348 const ObjCPropertyImplDecl *PID) {
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000349 ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000350 const ObjCPropertyDecl *PD = PID->getPropertyDecl();
351 ObjCMethodDecl *OMD = PD->getSetterMethodDecl();
352 assert(OMD && "Invalid call to generate setter (empty method)");
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000353 StartObjCMethod(OMD, IMP->getClassInterface());
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000354
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000355 bool IsCopy = PD->getSetterKind() == ObjCPropertyDecl::Copy;
Mike Stump1eb44332009-09-09 15:08:12 +0000356 bool IsAtomic =
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000357 !(PD->getPropertyAttributes() & ObjCPropertyDecl::OBJC_PR_nonatomic);
358
359 // Determine if we should use an objc_setProperty call for
360 // this. Properties with 'copy' semantics always use it, as do
361 // non-atomic properties with 'release' semantics as long as we are
362 // not in gc-only mode.
363 if (IsCopy ||
364 (CGM.getLangOptions().getGCMode() != LangOptions::GCOnly &&
365 PD->getSetterKind() == ObjCPropertyDecl::Retain)) {
Mike Stump1eb44332009-09-09 15:08:12 +0000366 llvm::Value *SetPropertyFn =
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000367 CGM.getObjCRuntime().GetPropertySetFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000368
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000369 if (!SetPropertyFn) {
370 CGM.ErrorUnsupported(PID, "Obj-C getter requiring atomic copy");
371 FinishFunction();
372 return;
373 }
Mike Stump1eb44332009-09-09 15:08:12 +0000374
375 // Emit objc_setProperty((id) self, _cmd, offset, arg,
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000376 // <is-atomic>, <is-copy>).
377 // FIXME: Can't this be simpler? This might even be worse than the
378 // corresponding gcc code.
379 CodeGenTypes &Types = CGM.getTypes();
380 ValueDecl *Cmd = OMD->getCmdDecl();
381 llvm::Value *CmdVal = Builder.CreateLoad(LocalDeclMap[Cmd], "cmd");
382 QualType IdTy = getContext().getObjCIdType();
Mike Stump1eb44332009-09-09 15:08:12 +0000383 llvm::Value *SelfAsId =
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000384 Builder.CreateBitCast(LoadObjCSelf(), Types.ConvertType(IdTy));
Fariborz Jahanianfef30b52008-12-09 20:23:04 +0000385 llvm::Value *Offset = EmitIvarOffset(IMP->getClassInterface(), Ivar);
Chris Lattner89951a82009-02-20 18:43:26 +0000386 llvm::Value *Arg = LocalDeclMap[*OMD->param_begin()];
Mike Stump1eb44332009-09-09 15:08:12 +0000387 llvm::Value *ArgAsId =
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000388 Builder.CreateBitCast(Builder.CreateLoad(Arg, "arg"),
389 Types.ConvertType(IdTy));
390 llvm::Value *True =
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000391 llvm::ConstantInt::get(Types.ConvertType(getContext().BoolTy), 1);
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000392 llvm::Value *False =
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000393 llvm::ConstantInt::get(Types.ConvertType(getContext().BoolTy), 0);
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000394 CallArgList Args;
395 Args.push_back(std::make_pair(RValue::get(SelfAsId), IdTy));
396 Args.push_back(std::make_pair(RValue::get(CmdVal), Cmd->getType()));
397 Args.push_back(std::make_pair(RValue::get(Offset), getContext().LongTy));
398 Args.push_back(std::make_pair(RValue::get(ArgAsId), IdTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000399 Args.push_back(std::make_pair(RValue::get(IsAtomic ? True : False),
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000400 getContext().BoolTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000401 Args.push_back(std::make_pair(RValue::get(IsCopy ? True : False),
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000402 getContext().BoolTy));
Mike Stumpf5408fe2009-05-16 07:57:57 +0000403 // FIXME: We shouldn't need to get the function info here, the runtime
404 // already should have computed it to build the function.
John McCall04a67a62010-02-05 21:31:56 +0000405 EmitCall(Types.getFunctionInfo(getContext().VoidTy, Args,
Rafael Espindola264ba482010-03-30 20:24:48 +0000406 FunctionType::ExtInfo()),
407 SetPropertyFn,
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000408 ReturnValueSlot(), Args);
Fariborz Jahanian0b2bd472010-04-13 00:38:05 +0000409 } else if (IsAtomic && hasAggregateLLVMType(Ivar->getType()) &&
410 !Ivar->getType()->isAnyComplexType() &&
411 IndirectObjCSetterArg(*CurFnInfo)
David Chisnall8fac25d2010-12-26 22:13:16 +0000412 && CGM.getObjCRuntime().GetSetStructFunction()) {
Fariborz Jahanian0b2bd472010-04-13 00:38:05 +0000413 // objc_copyStruct (&structIvar, &Arg,
414 // sizeof (struct something), true, false);
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000415 GenerateObjCAtomicSetterBody(OMD, Ivar);
Fariborz Jahanian97a73cd2010-05-06 15:45:36 +0000416 } else if (PID->getSetterCXXAssignment()) {
John McCall2a416372010-12-05 02:00:02 +0000417 EmitIgnoredExpr(PID->getSetterCXXAssignment());
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000418 } else {
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000419 const llvm::Triple &Triple = getContext().Target.getTriple();
420 QualType IVART = Ivar->getType();
421 if (IsAtomic &&
422 IVART->isScalarType() &&
423 (Triple.getArch() == llvm::Triple::arm ||
424 Triple.getArch() == llvm::Triple::thumb) &&
425 (getContext().getTypeSizeInChars(IVART)
426 > CharUnits::fromQuantity(4)) &&
427 CGM.getObjCRuntime().GetGetStructFunction()) {
428 GenerateObjCAtomicSetterBody(OMD, Ivar);
429 }
430 else {
431 // FIXME: Find a clean way to avoid AST node creation.
432 SourceLocation Loc = PD->getLocation();
433 ValueDecl *Self = OMD->getSelfDecl();
434 ObjCIvarDecl *Ivar = PID->getPropertyIvarDecl();
435 DeclRefExpr Base(Self, Self->getType(), VK_RValue, Loc);
436 ParmVarDecl *ArgDecl = *OMD->param_begin();
437 DeclRefExpr Arg(ArgDecl, ArgDecl->getType(), VK_LValue, Loc);
438 ObjCIvarRefExpr IvarRef(Ivar, Ivar->getType(), Loc, &Base, true, true);
Daniel Dunbar45e84232009-10-27 19:21:30 +0000439
Fariborz Jahanian2846b972011-02-18 19:15:13 +0000440 // The property type can differ from the ivar type in some situations with
441 // Objective-C pointer types, we can always bit cast the RHS in these cases.
442 if (getContext().getCanonicalType(Ivar->getType()) !=
443 getContext().getCanonicalType(ArgDecl->getType())) {
444 ImplicitCastExpr ArgCasted(ImplicitCastExpr::OnStack,
445 Ivar->getType(), CK_BitCast, &Arg,
446 VK_RValue);
447 BinaryOperator Assign(&IvarRef, &ArgCasted, BO_Assign,
448 Ivar->getType(), VK_RValue, OK_Ordinary, Loc);
449 EmitStmt(&Assign);
450 } else {
451 BinaryOperator Assign(&IvarRef, &Arg, BO_Assign,
452 Ivar->getType(), VK_RValue, OK_Ordinary, Loc);
453 EmitStmt(&Assign);
454 }
Daniel Dunbar45e84232009-10-27 19:21:30 +0000455 }
Daniel Dunbar86957eb2008-09-24 06:32:09 +0000456 }
Daniel Dunbaraf05bb92008-08-26 08:29:31 +0000457
458 FinishFunction();
Chris Lattner41110242008-06-17 18:05:57 +0000459}
460
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000461void CodeGenFunction::GenerateObjCCtorDtorMethod(ObjCImplementationDecl *IMP,
462 ObjCMethodDecl *MD,
463 bool ctor) {
Sean Huntcbb67482011-01-08 20:30:50 +0000464 llvm::SmallVector<CXXCtorInitializer *, 8> IvarInitializers;
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000465 MD->createImplicitParams(CGM.getContext(), IMP->getClassInterface());
466 StartObjCMethod(MD, IMP->getClassInterface());
467 for (ObjCImplementationDecl::init_const_iterator B = IMP->init_begin(),
468 E = IMP->init_end(); B != E; ++B) {
Sean Huntcbb67482011-01-08 20:30:50 +0000469 CXXCtorInitializer *Member = (*B);
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000470 IvarInitializers.push_back(Member);
471 }
472 if (ctor) {
473 for (unsigned I = 0, E = IvarInitializers.size(); I != E; ++I) {
Sean Huntcbb67482011-01-08 20:30:50 +0000474 CXXCtorInitializer *IvarInit = IvarInitializers[I];
Francois Pichet00eb3f92010-12-04 09:14:42 +0000475 FieldDecl *Field = IvarInit->getAnyMember();
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000476 ObjCIvarDecl *Ivar = cast<ObjCIvarDecl>(Field);
Fariborz Jahanian9b4d4fc2010-04-28 22:30:33 +0000477 LValue LV = EmitLValueForIvar(TypeOfSelfObject(),
478 LoadObjCSelf(), Ivar, 0);
John McCall558d2ab2010-09-15 10:14:12 +0000479 EmitAggExpr(IvarInit->getInit(), AggValueSlot::forLValue(LV, true));
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000480 }
481 // constructor returns 'self'.
482 CodeGenTypes &Types = CGM.getTypes();
483 QualType IdTy(CGM.getContext().getObjCIdType());
484 llvm::Value *SelfAsId =
485 Builder.CreateBitCast(LoadObjCSelf(), Types.ConvertType(IdTy));
486 EmitReturnOfRValue(RValue::get(SelfAsId), IdTy);
Chandler Carruthbc397cf2010-05-06 00:20:39 +0000487 } else {
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000488 // dtor
489 for (size_t i = IvarInitializers.size(); i > 0; --i) {
Francois Pichet00eb3f92010-12-04 09:14:42 +0000490 FieldDecl *Field = IvarInitializers[i - 1]->getAnyMember();
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000491 QualType FieldType = Field->getType();
Fariborz Jahanian9b4d4fc2010-04-28 22:30:33 +0000492 const ConstantArrayType *Array =
493 getContext().getAsConstantArrayType(FieldType);
494 if (Array)
495 FieldType = getContext().getBaseElementType(FieldType);
496
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000497 ObjCIvarDecl *Ivar = cast<ObjCIvarDecl>(Field);
498 LValue LV = EmitLValueForIvar(TypeOfSelfObject(),
499 LoadObjCSelf(), Ivar, 0);
500 const RecordType *RT = FieldType->getAs<RecordType>();
501 CXXRecordDecl *FieldClassDecl = cast<CXXRecordDecl>(RT->getDecl());
Douglas Gregor1d110e02010-07-01 14:13:13 +0000502 CXXDestructorDecl *Dtor = FieldClassDecl->getDestructor();
Chandler Carruthbc397cf2010-05-06 00:20:39 +0000503 if (!Dtor->isTrivial()) {
Fariborz Jahanian8b688ed2010-05-04 19:29:32 +0000504 if (Array) {
505 const llvm::Type *BasePtr = ConvertType(FieldType);
506 BasePtr = llvm::PointerType::getUnqual(BasePtr);
507 llvm::Value *BaseAddrPtr =
508 Builder.CreateBitCast(LV.getAddress(), BasePtr);
509 EmitCXXAggrDestructorCall(Dtor,
510 Array, BaseAddrPtr);
Chandler Carruthbc397cf2010-05-06 00:20:39 +0000511 } else {
Fariborz Jahanian8b688ed2010-05-04 19:29:32 +0000512 EmitCXXDestructorCall(Dtor,
513 Dtor_Complete, /*ForVirtualBase=*/false,
514 LV.getAddress());
Chandler Carruthbc397cf2010-05-06 00:20:39 +0000515 }
516 }
517 }
Fariborz Jahanian109dfc62010-04-28 21:28:56 +0000518 }
519 FinishFunction();
520}
521
Fariborz Jahanian0b2bd472010-04-13 00:38:05 +0000522bool CodeGenFunction::IndirectObjCSetterArg(const CGFunctionInfo &FI) {
523 CGFunctionInfo::const_arg_iterator it = FI.arg_begin();
524 it++; it++;
525 const ABIArgInfo &AI = it->info;
526 // FIXME. Is this sufficient check?
527 return (AI.getKind() == ABIArgInfo::Indirect);
528}
529
Fariborz Jahanian15bd5882010-04-13 18:32:24 +0000530bool CodeGenFunction::IvarTypeWithAggrGCObjects(QualType Ty) {
531 if (CGM.getLangOptions().getGCMode() == LangOptions::NonGC)
532 return false;
533 if (const RecordType *FDTTy = Ty.getTypePtr()->getAs<RecordType>())
534 return FDTTy->getDecl()->hasObjectMember();
535 return false;
536}
537
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000538llvm::Value *CodeGenFunction::LoadObjCSelf() {
Daniel Dunbarb7ec2462008-08-16 03:19:19 +0000539 const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
540 return Builder.CreateLoad(LocalDeclMap[OMD->getSelfDecl()], "self");
Chris Lattner41110242008-06-17 18:05:57 +0000541}
542
Fariborz Jahanian45012a72009-02-03 00:09:52 +0000543QualType CodeGenFunction::TypeOfSelfObject() {
544 const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CurFuncDecl);
545 ImplicitParamDecl *selfDecl = OMD->getSelfDecl();
Steve Naroff14108da2009-07-10 23:34:53 +0000546 const ObjCObjectPointerType *PTy = cast<ObjCObjectPointerType>(
547 getContext().getCanonicalType(selfDecl->getType()));
Fariborz Jahanian45012a72009-02-03 00:09:52 +0000548 return PTy->getPointeeType();
549}
550
John McCalle68b9842010-12-04 03:11:00 +0000551LValue
552CodeGenFunction::EmitObjCPropertyRefLValue(const ObjCPropertyRefExpr *E) {
553 // This is a special l-value that just issues sends when we load or
554 // store through it.
555
556 // For certain base kinds, we need to emit the base immediately.
557 llvm::Value *Base;
558 if (E->isSuperReceiver())
559 Base = LoadObjCSelf();
560 else if (E->isClassReceiver())
561 Base = CGM.getObjCRuntime().GetClass(Builder, E->getClassReceiver());
562 else
563 Base = EmitScalarExpr(E->getBase());
564 return LValue::MakePropertyRef(E, Base);
565}
566
567static RValue GenerateMessageSendSuper(CodeGenFunction &CGF,
568 ReturnValueSlot Return,
569 QualType ResultType,
570 Selector S,
571 llvm::Value *Receiver,
572 const CallArgList &CallArgs) {
573 const ObjCMethodDecl *OMD = cast<ObjCMethodDecl>(CGF.CurFuncDecl);
Fariborz Jahanianf4695572009-03-20 19:18:21 +0000574 bool isClassMessage = OMD->isClassMethod();
575 bool isCategoryImpl = isa<ObjCCategoryImplDecl>(OMD->getDeclContext());
John McCalle68b9842010-12-04 03:11:00 +0000576 return CGF.CGM.getObjCRuntime()
577 .GenerateMessageSendSuper(CGF, Return, ResultType,
578 S, OMD->getClassInterface(),
579 isCategoryImpl, Receiver,
580 isClassMessage, CallArgs);
Fariborz Jahanianf4695572009-03-20 19:18:21 +0000581}
582
John McCall119a1c62010-12-04 02:32:38 +0000583RValue CodeGenFunction::EmitLoadOfPropertyRefLValue(LValue LV,
584 ReturnValueSlot Return) {
585 const ObjCPropertyRefExpr *E = LV.getPropertyRefExpr();
John McCall12f78a62010-12-02 01:19:52 +0000586 QualType ResultType;
587 Selector S;
588 if (E->isExplicitProperty()) {
589 const ObjCPropertyDecl *Property = E->getExplicitProperty();
590 S = Property->getGetterName();
591 ResultType = E->getType();
Mike Stumpb3589f42009-07-30 22:28:39 +0000592 } else {
John McCall12f78a62010-12-02 01:19:52 +0000593 const ObjCMethodDecl *Getter = E->getImplicitPropertyGetter();
594 S = Getter->getSelector();
595 ResultType = Getter->getResultType(); // with reference!
Fariborz Jahanian43f44702008-11-22 22:30:21 +0000596 }
John McCall12f78a62010-12-02 01:19:52 +0000597
John McCall119a1c62010-12-04 02:32:38 +0000598 llvm::Value *Receiver = LV.getPropertyRefBaseAddr();
John McCalle68b9842010-12-04 03:11:00 +0000599
600 // Accesses to 'super' follow a different code path.
601 if (E->isSuperReceiver())
602 return GenerateMessageSendSuper(*this, Return, ResultType,
603 S, Receiver, CallArgList());
604
John McCall119a1c62010-12-04 02:32:38 +0000605 const ObjCInterfaceDecl *ReceiverClass
606 = (E->isClassReceiver() ? E->getClassReceiver() : 0);
John McCall12f78a62010-12-02 01:19:52 +0000607 return CGM.getObjCRuntime().
608 GenerateMessageSend(*this, Return, ResultType, S,
609 Receiver, CallArgList(), ReceiverClass);
Daniel Dunbar9c3fc702008-08-27 06:57:25 +0000610}
611
John McCall119a1c62010-12-04 02:32:38 +0000612void CodeGenFunction::EmitStoreThroughPropertyRefLValue(RValue Src,
613 LValue Dst) {
614 const ObjCPropertyRefExpr *E = Dst.getPropertyRefExpr();
John McCall12f78a62010-12-02 01:19:52 +0000615 Selector S = E->getSetterSelector();
616 QualType ArgType;
617 if (E->isImplicitProperty()) {
618 const ObjCMethodDecl *Setter = E->getImplicitPropertySetter();
619 ObjCMethodDecl::param_iterator P = Setter->param_begin();
620 ArgType = (*P)->getType();
621 } else {
622 ArgType = E->getType();
623 }
Fariborz Jahanianb19c76e2011-02-08 22:33:23 +0000624 // FIXME. Other than scalars, AST is not adequate for setter and
625 // getter type mismatches which require conversion.
626 if (Src.isScalar()) {
627 llvm::Value *SrcVal = Src.getScalarVal();
628 QualType DstType = getContext().getCanonicalType(ArgType);
629 const llvm::Type *DstTy = ConvertType(DstType);
630 if (SrcVal->getType() != DstTy)
631 Src =
632 RValue::get(EmitScalarConversion(SrcVal, E->getType(), DstType));
633 }
634
John McCalle68b9842010-12-04 03:11:00 +0000635 CallArgList Args;
636 Args.push_back(std::make_pair(Src, ArgType));
637
638 llvm::Value *Receiver = Dst.getPropertyRefBaseAddr();
639 QualType ResultType = getContext().VoidTy;
640
John McCall12f78a62010-12-02 01:19:52 +0000641 if (E->isSuperReceiver()) {
John McCalle68b9842010-12-04 03:11:00 +0000642 GenerateMessageSendSuper(*this, ReturnValueSlot(),
643 ResultType, S, Receiver, Args);
John McCall12f78a62010-12-02 01:19:52 +0000644 return;
645 }
646
John McCall119a1c62010-12-04 02:32:38 +0000647 const ObjCInterfaceDecl *ReceiverClass
648 = (E->isClassReceiver() ? E->getClassReceiver() : 0);
John McCall12f78a62010-12-02 01:19:52 +0000649
John McCall12f78a62010-12-02 01:19:52 +0000650 CGM.getObjCRuntime().GenerateMessageSend(*this, ReturnValueSlot(),
John McCalle68b9842010-12-04 03:11:00 +0000651 ResultType, S, Receiver, Args,
652 ReceiverClass);
Daniel Dunbar85c59ed2008-08-29 08:11:39 +0000653}
654
Chris Lattner74391b42009-03-22 21:03:39 +0000655void CodeGenFunction::EmitObjCForCollectionStmt(const ObjCForCollectionStmt &S){
Mike Stump1eb44332009-09-09 15:08:12 +0000656 llvm::Constant *EnumerationMutationFn =
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000657 CGM.getObjCRuntime().EnumerationMutationFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000658
Daniel Dunbarc1cf4a52008-09-24 04:04:31 +0000659 if (!EnumerationMutationFn) {
660 CGM.ErrorUnsupported(&S, "Obj-C fast enumeration for this runtime");
661 return;
662 }
663
John McCall57b3b6a2011-02-22 07:16:58 +0000664 // The local variable comes into scope immediately.
665 AutoVarEmission variable = AutoVarEmission::invalid();
666 if (const DeclStmt *SD = dyn_cast<DeclStmt>(S.getElement()))
667 variable = EmitAutoVarAlloca(*cast<VarDecl>(SD->getSingleDecl()));
668
Devang Patelbcbd03a2011-01-19 01:36:36 +0000669 CGDebugInfo *DI = getDebugInfo();
670 if (DI) {
671 DI->setLocation(S.getSourceRange().getBegin());
672 DI->EmitRegionStart(Builder);
673 }
674
John McCalld88687f2011-01-07 01:49:06 +0000675 JumpDest LoopEnd = getJumpDestInCurrentScope("forcoll.end");
676 JumpDest AfterBody = getJumpDestInCurrentScope("forcoll.next");
Mike Stump1eb44332009-09-09 15:08:12 +0000677
Anders Carlssonf484c312008-08-31 02:33:12 +0000678 // Fast enumeration state.
679 QualType StateTy = getContext().getObjCFastEnumerationStateType();
Daniel Dunbar195337d2010-02-09 02:48:28 +0000680 llvm::Value *StatePtr = CreateMemTemp(StateTy, "state.ptr");
Anders Carlsson1884eb02010-05-22 17:35:42 +0000681 EmitNullInitialization(StatePtr, StateTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000682
Anders Carlssonf484c312008-08-31 02:33:12 +0000683 // Number of elements in the items array.
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000684 static const unsigned NumItems = 16;
Mike Stump1eb44332009-09-09 15:08:12 +0000685
John McCalld88687f2011-01-07 01:49:06 +0000686 // Fetch the countByEnumeratingWithState:objects:count: selector.
Benjamin Kramerad468862010-03-30 11:36:44 +0000687 IdentifierInfo *II[] = {
688 &CGM.getContext().Idents.get("countByEnumeratingWithState"),
689 &CGM.getContext().Idents.get("objects"),
690 &CGM.getContext().Idents.get("count")
691 };
692 Selector FastEnumSel =
693 CGM.getContext().Selectors.getSelector(llvm::array_lengthof(II), &II[0]);
Anders Carlssonf484c312008-08-31 02:33:12 +0000694
695 QualType ItemsTy =
696 getContext().getConstantArrayType(getContext().getObjCIdType(),
Mike Stump1eb44332009-09-09 15:08:12 +0000697 llvm::APInt(32, NumItems),
Anders Carlssonf484c312008-08-31 02:33:12 +0000698 ArrayType::Normal, 0);
Daniel Dunbar195337d2010-02-09 02:48:28 +0000699 llvm::Value *ItemsPtr = CreateMemTemp(ItemsTy, "items.ptr");
Mike Stump1eb44332009-09-09 15:08:12 +0000700
John McCalld88687f2011-01-07 01:49:06 +0000701 // Emit the collection pointer.
Anders Carlssonf484c312008-08-31 02:33:12 +0000702 llvm::Value *Collection = EmitScalarExpr(S.getCollection());
Mike Stump1eb44332009-09-09 15:08:12 +0000703
John McCalld88687f2011-01-07 01:49:06 +0000704 // Send it our message:
Anders Carlssonf484c312008-08-31 02:33:12 +0000705 CallArgList Args;
John McCalld88687f2011-01-07 01:49:06 +0000706
707 // The first argument is a temporary of the enumeration-state type.
Mike Stump1eb44332009-09-09 15:08:12 +0000708 Args.push_back(std::make_pair(RValue::get(StatePtr),
Anders Carlssonf484c312008-08-31 02:33:12 +0000709 getContext().getPointerType(StateTy)));
Mike Stump1eb44332009-09-09 15:08:12 +0000710
John McCalld88687f2011-01-07 01:49:06 +0000711 // The second argument is a temporary array with space for NumItems
712 // pointers. We'll actually be loading elements from the array
713 // pointer written into the control state; this buffer is so that
714 // collections that *aren't* backed by arrays can still queue up
715 // batches of elements.
Mike Stump1eb44332009-09-09 15:08:12 +0000716 Args.push_back(std::make_pair(RValue::get(ItemsPtr),
Anders Carlssonf484c312008-08-31 02:33:12 +0000717 getContext().getPointerType(ItemsTy)));
Mike Stump1eb44332009-09-09 15:08:12 +0000718
John McCalld88687f2011-01-07 01:49:06 +0000719 // The third argument is the capacity of that temporary array.
Anders Carlssonf484c312008-08-31 02:33:12 +0000720 const llvm::Type *UnsignedLongLTy = ConvertType(getContext().UnsignedLongTy);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000721 llvm::Constant *Count = llvm::ConstantInt::get(UnsignedLongLTy, NumItems);
Mike Stump1eb44332009-09-09 15:08:12 +0000722 Args.push_back(std::make_pair(RValue::get(Count),
Daniel Dunbar46f45b92008-09-09 01:06:48 +0000723 getContext().UnsignedLongTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000724
John McCalld88687f2011-01-07 01:49:06 +0000725 // Start the enumeration.
Mike Stump1eb44332009-09-09 15:08:12 +0000726 RValue CountRV =
John McCallef072fd2010-05-22 01:48:05 +0000727 CGM.getObjCRuntime().GenerateMessageSend(*this, ReturnValueSlot(),
Anders Carlssonf484c312008-08-31 02:33:12 +0000728 getContext().UnsignedLongTy,
729 FastEnumSel,
David Chisnallc6cd5fd2010-04-28 19:33:36 +0000730 Collection, Args);
Anders Carlssonf484c312008-08-31 02:33:12 +0000731
John McCalld88687f2011-01-07 01:49:06 +0000732 // The initial number of objects that were returned in the buffer.
733 llvm::Value *initialBufferLimit = CountRV.getScalarVal();
Mike Stump1eb44332009-09-09 15:08:12 +0000734
John McCalld88687f2011-01-07 01:49:06 +0000735 llvm::BasicBlock *EmptyBB = createBasicBlock("forcoll.empty");
736 llvm::BasicBlock *LoopInitBB = createBasicBlock("forcoll.loopinit");
Mike Stump1eb44332009-09-09 15:08:12 +0000737
John McCalld88687f2011-01-07 01:49:06 +0000738 llvm::Value *zero = llvm::Constant::getNullValue(UnsignedLongLTy);
Anders Carlssonf484c312008-08-31 02:33:12 +0000739
John McCalld88687f2011-01-07 01:49:06 +0000740 // If the limit pointer was zero to begin with, the collection is
741 // empty; skip all this.
742 Builder.CreateCondBr(Builder.CreateICmpEQ(initialBufferLimit, zero, "iszero"),
743 EmptyBB, LoopInitBB);
Anders Carlssonf484c312008-08-31 02:33:12 +0000744
John McCalld88687f2011-01-07 01:49:06 +0000745 // Otherwise, initialize the loop.
746 EmitBlock(LoopInitBB);
Mike Stump1eb44332009-09-09 15:08:12 +0000747
John McCalld88687f2011-01-07 01:49:06 +0000748 // Save the initial mutations value. This is the value at an
749 // address that was written into the state object by
750 // countByEnumeratingWithState:objects:count:.
Mike Stump1eb44332009-09-09 15:08:12 +0000751 llvm::Value *StateMutationsPtrPtr =
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000752 Builder.CreateStructGEP(StatePtr, 2, "mutationsptr.ptr");
Mike Stump1eb44332009-09-09 15:08:12 +0000753 llvm::Value *StateMutationsPtr = Builder.CreateLoad(StateMutationsPtrPtr,
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000754 "mutationsptr");
Mike Stump1eb44332009-09-09 15:08:12 +0000755
John McCalld88687f2011-01-07 01:49:06 +0000756 llvm::Value *initialMutations =
757 Builder.CreateLoad(StateMutationsPtr, "forcoll.initial-mutations");
Mike Stump1eb44332009-09-09 15:08:12 +0000758
John McCalld88687f2011-01-07 01:49:06 +0000759 // Start looping. This is the point we return to whenever we have a
760 // fresh, non-empty batch of objects.
761 llvm::BasicBlock *LoopBodyBB = createBasicBlock("forcoll.loopbody");
762 EmitBlock(LoopBodyBB);
Mike Stump1eb44332009-09-09 15:08:12 +0000763
John McCalld88687f2011-01-07 01:49:06 +0000764 // The current index into the buffer.
765 llvm::PHINode *index = Builder.CreatePHI(UnsignedLongLTy, "forcoll.index");
766 index->addIncoming(zero, LoopInitBB);
Anders Carlssonf484c312008-08-31 02:33:12 +0000767
John McCalld88687f2011-01-07 01:49:06 +0000768 // The current buffer size.
769 llvm::PHINode *count = Builder.CreatePHI(UnsignedLongLTy, "forcoll.count");
770 count->addIncoming(initialBufferLimit, LoopInitBB);
Mike Stump1eb44332009-09-09 15:08:12 +0000771
John McCalld88687f2011-01-07 01:49:06 +0000772 // Check whether the mutations value has changed from where it was
773 // at start. StateMutationsPtr should actually be invariant between
774 // refreshes.
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000775 StateMutationsPtr = Builder.CreateLoad(StateMutationsPtrPtr, "mutationsptr");
John McCalld88687f2011-01-07 01:49:06 +0000776 llvm::Value *currentMutations
777 = Builder.CreateLoad(StateMutationsPtr, "statemutations");
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000778
John McCalld88687f2011-01-07 01:49:06 +0000779 llvm::BasicBlock *WasMutatedBB = createBasicBlock("forcoll.mutated");
Dan Gohman361cf982011-03-02 22:39:34 +0000780 llvm::BasicBlock *WasNotMutatedBB = createBasicBlock("forcoll.notmutated");
Mike Stump1eb44332009-09-09 15:08:12 +0000781
John McCalld88687f2011-01-07 01:49:06 +0000782 Builder.CreateCondBr(Builder.CreateICmpEQ(currentMutations, initialMutations),
783 WasNotMutatedBB, WasMutatedBB);
Mike Stump1eb44332009-09-09 15:08:12 +0000784
John McCalld88687f2011-01-07 01:49:06 +0000785 // If so, call the enumeration-mutation function.
786 EmitBlock(WasMutatedBB);
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000787 llvm::Value *V =
Mike Stump1eb44332009-09-09 15:08:12 +0000788 Builder.CreateBitCast(Collection,
Anders Carlsson2abd89c2008-08-31 04:05:03 +0000789 ConvertType(getContext().getObjCIdType()),
790 "tmp");
Daniel Dunbar2b2105e2009-02-03 23:55:40 +0000791 CallArgList Args2;
Mike Stump1eb44332009-09-09 15:08:12 +0000792 Args2.push_back(std::make_pair(RValue::get(V),
Daniel Dunbar2b2105e2009-02-03 23:55:40 +0000793 getContext().getObjCIdType()));
Mike Stumpf5408fe2009-05-16 07:57:57 +0000794 // FIXME: We shouldn't need to get the function info here, the runtime already
795 // should have computed it to build the function.
John McCall04a67a62010-02-05 21:31:56 +0000796 EmitCall(CGM.getTypes().getFunctionInfo(getContext().VoidTy, Args2,
Rafael Espindola264ba482010-03-30 20:24:48 +0000797 FunctionType::ExtInfo()),
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000798 EnumerationMutationFn, ReturnValueSlot(), Args2);
Mike Stump1eb44332009-09-09 15:08:12 +0000799
John McCalld88687f2011-01-07 01:49:06 +0000800 // Otherwise, or if the mutation function returns, just continue.
801 EmitBlock(WasNotMutatedBB);
Mike Stump1eb44332009-09-09 15:08:12 +0000802
John McCalld88687f2011-01-07 01:49:06 +0000803 // Initialize the element variable.
804 RunCleanupsScope elementVariableScope(*this);
John McCall57b3b6a2011-02-22 07:16:58 +0000805 bool elementIsVariable;
John McCalld88687f2011-01-07 01:49:06 +0000806 LValue elementLValue;
807 QualType elementType;
808 if (const DeclStmt *SD = dyn_cast<DeclStmt>(S.getElement())) {
John McCall57b3b6a2011-02-22 07:16:58 +0000809 // Initialize the variable, in case it's a __block variable or something.
810 EmitAutoVarInit(variable);
John McCalld88687f2011-01-07 01:49:06 +0000811
John McCall57b3b6a2011-02-22 07:16:58 +0000812 const VarDecl* D = cast<VarDecl>(SD->getSingleDecl());
John McCalld88687f2011-01-07 01:49:06 +0000813 DeclRefExpr tempDRE(const_cast<VarDecl*>(D), D->getType(),
814 VK_LValue, SourceLocation());
815 elementLValue = EmitLValue(&tempDRE);
816 elementType = D->getType();
John McCall57b3b6a2011-02-22 07:16:58 +0000817 elementIsVariable = true;
John McCalld88687f2011-01-07 01:49:06 +0000818 } else {
819 elementLValue = LValue(); // suppress warning
820 elementType = cast<Expr>(S.getElement())->getType();
John McCall57b3b6a2011-02-22 07:16:58 +0000821 elementIsVariable = false;
John McCalld88687f2011-01-07 01:49:06 +0000822 }
823 const llvm::Type *convertedElementType = ConvertType(elementType);
824
825 // Fetch the buffer out of the enumeration state.
826 // TODO: this pointer should actually be invariant between
827 // refreshes, which would help us do certain loop optimizations.
Mike Stump1eb44332009-09-09 15:08:12 +0000828 llvm::Value *StateItemsPtr =
Anders Carlssonf484c312008-08-31 02:33:12 +0000829 Builder.CreateStructGEP(StatePtr, 1, "stateitems.ptr");
John McCalld88687f2011-01-07 01:49:06 +0000830 llvm::Value *EnumStateItems =
831 Builder.CreateLoad(StateItemsPtr, "stateitems");
Anders Carlssonf484c312008-08-31 02:33:12 +0000832
John McCalld88687f2011-01-07 01:49:06 +0000833 // Fetch the value at the current index from the buffer.
Mike Stump1eb44332009-09-09 15:08:12 +0000834 llvm::Value *CurrentItemPtr =
John McCalld88687f2011-01-07 01:49:06 +0000835 Builder.CreateGEP(EnumStateItems, index, "currentitem.ptr");
836 llvm::Value *CurrentItem = Builder.CreateLoad(CurrentItemPtr);
Mike Stump1eb44332009-09-09 15:08:12 +0000837
John McCalld88687f2011-01-07 01:49:06 +0000838 // Cast that value to the right type.
839 CurrentItem = Builder.CreateBitCast(CurrentItem, convertedElementType,
840 "currentitem");
Mike Stump1eb44332009-09-09 15:08:12 +0000841
John McCalld88687f2011-01-07 01:49:06 +0000842 // Make sure we have an l-value. Yes, this gets evaluated every
843 // time through the loop.
John McCall57b3b6a2011-02-22 07:16:58 +0000844 if (!elementIsVariable)
John McCalld88687f2011-01-07 01:49:06 +0000845 elementLValue = EmitLValue(cast<Expr>(S.getElement()));
Mike Stump1eb44332009-09-09 15:08:12 +0000846
John McCalld88687f2011-01-07 01:49:06 +0000847 EmitStoreThroughLValue(RValue::get(CurrentItem), elementLValue, elementType);
Mike Stump1eb44332009-09-09 15:08:12 +0000848
John McCall57b3b6a2011-02-22 07:16:58 +0000849 // If we do have an element variable, this assignment is the end of
850 // its initialization.
851 if (elementIsVariable)
852 EmitAutoVarCleanups(variable);
853
John McCalld88687f2011-01-07 01:49:06 +0000854 // Perform the loop body, setting up break and continue labels.
Anders Carlssone4b6d342009-02-10 05:52:02 +0000855 BreakContinueStack.push_back(BreakContinue(LoopEnd, AfterBody));
John McCalld88687f2011-01-07 01:49:06 +0000856 {
857 RunCleanupsScope Scope(*this);
858 EmitStmt(S.getBody());
859 }
Anders Carlssonf484c312008-08-31 02:33:12 +0000860 BreakContinueStack.pop_back();
Mike Stump1eb44332009-09-09 15:08:12 +0000861
John McCalld88687f2011-01-07 01:49:06 +0000862 // Destroy the element variable now.
863 elementVariableScope.ForceCleanup();
864
865 // Check whether there are more elements.
John McCallff8e1152010-07-23 21:56:41 +0000866 EmitBlock(AfterBody.getBlock());
Mike Stump1eb44332009-09-09 15:08:12 +0000867
John McCalld88687f2011-01-07 01:49:06 +0000868 llvm::BasicBlock *FetchMoreBB = createBasicBlock("forcoll.refetch");
Fariborz Jahanianf0906c42009-01-06 18:56:31 +0000869
John McCalld88687f2011-01-07 01:49:06 +0000870 // First we check in the local buffer.
871 llvm::Value *indexPlusOne
872 = Builder.CreateAdd(index, llvm::ConstantInt::get(UnsignedLongLTy, 1));
Anders Carlssonf484c312008-08-31 02:33:12 +0000873
John McCalld88687f2011-01-07 01:49:06 +0000874 // If we haven't overrun the buffer yet, we can continue.
875 Builder.CreateCondBr(Builder.CreateICmpULT(indexPlusOne, count),
876 LoopBodyBB, FetchMoreBB);
877
878 index->addIncoming(indexPlusOne, AfterBody.getBlock());
879 count->addIncoming(count, AfterBody.getBlock());
880
881 // Otherwise, we have to fetch more elements.
882 EmitBlock(FetchMoreBB);
Mike Stump1eb44332009-09-09 15:08:12 +0000883
884 CountRV =
John McCallef072fd2010-05-22 01:48:05 +0000885 CGM.getObjCRuntime().GenerateMessageSend(*this, ReturnValueSlot(),
Anders Carlssonf484c312008-08-31 02:33:12 +0000886 getContext().UnsignedLongTy,
Mike Stump1eb44332009-09-09 15:08:12 +0000887 FastEnumSel,
David Chisnallc6cd5fd2010-04-28 19:33:36 +0000888 Collection, Args);
Mike Stump1eb44332009-09-09 15:08:12 +0000889
John McCalld88687f2011-01-07 01:49:06 +0000890 // If we got a zero count, we're done.
891 llvm::Value *refetchCount = CountRV.getScalarVal();
892
893 // (note that the message send might split FetchMoreBB)
894 index->addIncoming(zero, Builder.GetInsertBlock());
895 count->addIncoming(refetchCount, Builder.GetInsertBlock());
896
897 Builder.CreateCondBr(Builder.CreateICmpEQ(refetchCount, zero),
898 EmptyBB, LoopBodyBB);
Mike Stump1eb44332009-09-09 15:08:12 +0000899
Anders Carlssonf484c312008-08-31 02:33:12 +0000900 // No more elements.
John McCalld88687f2011-01-07 01:49:06 +0000901 EmitBlock(EmptyBB);
Anders Carlssonf484c312008-08-31 02:33:12 +0000902
John McCall57b3b6a2011-02-22 07:16:58 +0000903 if (!elementIsVariable) {
Anders Carlssonf484c312008-08-31 02:33:12 +0000904 // If the element was not a declaration, set it to be null.
905
John McCalld88687f2011-01-07 01:49:06 +0000906 llvm::Value *null = llvm::Constant::getNullValue(convertedElementType);
907 elementLValue = EmitLValue(cast<Expr>(S.getElement()));
908 EmitStoreThroughLValue(RValue::get(null), elementLValue, elementType);
Anders Carlssonf484c312008-08-31 02:33:12 +0000909 }
910
Devang Patelbcbd03a2011-01-19 01:36:36 +0000911 if (DI) {
912 DI->setLocation(S.getSourceRange().getEnd());
913 DI->EmitRegionEnd(Builder);
914 }
915
John McCallff8e1152010-07-23 21:56:41 +0000916 EmitBlock(LoopEnd.getBlock());
Anders Carlsson3d8400d2008-08-30 19:51:14 +0000917}
918
Mike Stump1eb44332009-09-09 15:08:12 +0000919void CodeGenFunction::EmitObjCAtTryStmt(const ObjCAtTryStmt &S) {
John McCallf1549f62010-07-06 01:34:17 +0000920 CGM.getObjCRuntime().EmitTryStmt(*this, S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000921}
922
Mike Stump1eb44332009-09-09 15:08:12 +0000923void CodeGenFunction::EmitObjCAtThrowStmt(const ObjCAtThrowStmt &S) {
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000924 CGM.getObjCRuntime().EmitThrowStmt(*this, S);
925}
926
Chris Lattner10cac6f2008-11-15 21:26:17 +0000927void CodeGenFunction::EmitObjCAtSynchronizedStmt(
Mike Stump1eb44332009-09-09 15:08:12 +0000928 const ObjCAtSynchronizedStmt &S) {
John McCallf1549f62010-07-06 01:34:17 +0000929 CGM.getObjCRuntime().EmitSynchronizedStmt(*this, S);
Chris Lattner10cac6f2008-11-15 21:26:17 +0000930}
931
Ted Kremenek2979ec72008-04-09 15:51:31 +0000932CGObjCRuntime::~CGObjCRuntime() {}