blob: a621c7e303cdbf94000c7328d639d316fa757e95 [file] [log] [blame]
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001//===------- CGObjCGNU.cpp - Emit LLVM Code from ASTs for a Module --------===//
Chris Lattner0f984262008-03-01 08:50:34 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000010// This provides Objective-C code generation targetting the GNU runtime. The
11// class in this file generates structures used by the GNU Objective-C runtime
12// library. These structures are defined in objc/objc.h and objc/objc-api.h in
13// the GNU runtime distribution.
Chris Lattner0f984262008-03-01 08:50:34 +000014//
15//===----------------------------------------------------------------------===//
16
17#include "CGObjCRuntime.h"
Chris Lattnerdce14062008-06-26 04:19:03 +000018#include "CodeGenModule.h"
Daniel Dunbar8f2926b2008-08-23 03:46:30 +000019#include "CodeGenFunction.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000020
Chris Lattnerdce14062008-06-26 04:19:03 +000021#include "clang/AST/ASTContext.h"
Daniel Dunbare91593e2008-08-11 04:54:23 +000022#include "clang/AST/Decl.h"
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +000023#include "clang/AST/DeclObjC.h"
Anders Carlsson19cc4ab2009-07-18 19:43:29 +000024#include "clang/AST/RecordLayout.h"
Chris Lattner16f00492009-04-26 01:32:48 +000025#include "clang/AST/StmtObjC.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000026
27#include "llvm/Intrinsics.h"
Chris Lattner0f984262008-03-01 08:50:34 +000028#include "llvm/Module.h"
Chris Lattner0f984262008-03-01 08:50:34 +000029#include "llvm/ADT/SmallVector.h"
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000030#include "llvm/ADT/StringMap.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000031#include "llvm/Support/Compiler.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000032#include "llvm/Target/TargetData.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000033
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000034#include <map>
Chris Lattnere160c9b2009-01-27 05:06:01 +000035
36
Chris Lattnerdce14062008-06-26 04:19:03 +000037using namespace clang;
Daniel Dunbar46f45b92008-09-09 01:06:48 +000038using namespace CodeGen;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000039using llvm::dyn_cast;
40
41// The version of the runtime that this class targets. Must match the version
42// in the runtime.
43static const int RuntimeVersion = 8;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +000044static const int NonFragileRuntimeVersion = 9;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000045static const int ProtocolVersion = 2;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000046static const int NonFragileProtocolVersion = 3;
Chris Lattner0f984262008-03-01 08:50:34 +000047
Chris Lattner0f984262008-03-01 08:50:34 +000048namespace {
Chris Lattnerdce14062008-06-26 04:19:03 +000049class CGObjCGNU : public CodeGen::CGObjCRuntime {
Chris Lattner0f984262008-03-01 08:50:34 +000050private:
Chris Lattnerdce14062008-06-26 04:19:03 +000051 CodeGen::CodeGenModule &CGM;
Chris Lattner0f984262008-03-01 08:50:34 +000052 llvm::Module &TheModule;
Chris Lattnere160c9b2009-01-27 05:06:01 +000053 const llvm::PointerType *SelectorTy;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000054 const llvm::IntegerType *Int8Ty;
Chris Lattnere160c9b2009-01-27 05:06:01 +000055 const llvm::PointerType *PtrToInt8Ty;
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +000056 const llvm::FunctionType *IMPTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000057 const llvm::PointerType *IdTy;
David Chisnallef6e0f32010-02-03 15:59:02 +000058 const llvm::PointerType *PtrToIdTy;
John McCallead608a2010-02-26 00:48:12 +000059 CanQualType ASTIdTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000060 const llvm::IntegerType *IntTy;
61 const llvm::PointerType *PtrTy;
62 const llvm::IntegerType *LongTy;
63 const llvm::PointerType *PtrToIntTy;
Daniel Dunbar5efccb12009-05-04 15:31:17 +000064 llvm::GlobalAlias *ClassPtrAlias;
65 llvm::GlobalAlias *MetaClassPtrAlias;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000066 std::vector<llvm::Constant*> Classes;
67 std::vector<llvm::Constant*> Categories;
68 std::vector<llvm::Constant*> ConstantStrings;
David Chisnall48272a02010-01-27 12:49:23 +000069 llvm::StringMap<llvm::Constant*> ObjCStrings;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000070 llvm::Function *LoadFunction;
71 llvm::StringMap<llvm::Constant*> ExistingProtocols;
72 typedef std::pair<std::string, std::string> TypedSelector;
73 std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
74 llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
David Chisnallef6e0f32010-02-03 15:59:02 +000075 // Selectors that we don't emit in GC mode
76 Selector RetainSel, ReleaseSel, AutoreleaseSel;
77 // Functions used for GC.
78 llvm::Constant *IvarAssignFn, *StrongCastAssignFn, *MemMoveFn, *WeakReadFn,
79 *WeakAssignFn, *GlobalAssignFn;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000080 // Some zeros used for GEPs in lots of places.
81 llvm::Constant *Zeros[2];
82 llvm::Constant *NULLPtr;
Owen Andersona1cf15f2009-07-14 23:10:40 +000083 llvm::LLVMContext &VMContext;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000084private:
85 llvm::Constant *GenerateIvarList(
86 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
87 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
88 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets);
89 llvm::Constant *GenerateMethodList(const std::string &ClassName,
90 const std::string &CategoryName,
Mike Stump1eb44332009-09-09 15:08:12 +000091 const llvm::SmallVectorImpl<Selector> &MethodSels,
92 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000093 bool isClassMethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +000094 llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000095 llvm::Constant *GeneratePropertyList(const ObjCImplementationDecl *OID,
96 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
97 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000098 llvm::Constant *GenerateProtocolList(
99 const llvm::SmallVectorImpl<std::string> &Protocols);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000100 // To ensure that all protocols are seen by the runtime, we add a category on
101 // a class defined in the runtime, declaring no methods, but adopting the
102 // protocols.
103 void GenerateProtocolHolderCategory(void);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000104 llvm::Constant *GenerateClassStructure(
105 llvm::Constant *MetaClass,
106 llvm::Constant *SuperClass,
107 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000108 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000109 llvm::Constant *Version,
110 llvm::Constant *InstanceSize,
111 llvm::Constant *IVars,
112 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000113 llvm::Constant *Protocols,
114 llvm::Constant *IvarOffsets,
115 llvm::Constant *Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000116 llvm::Constant *GenerateProtocolMethodList(
117 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
118 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes);
119 llvm::Constant *MakeConstantString(const std::string &Str, const std::string
120 &Name="");
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000121 llvm::Constant *ExportUniqueString(const std::string &Str, const std::string
122 prefix);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000123 llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000124 std::vector<llvm::Constant*> &V, llvm::StringRef Name="",
David Chisnall41d63ed2010-01-08 00:14:31 +0000125 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000126 llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000127 std::vector<llvm::Constant*> &V, llvm::StringRef Name="",
David Chisnall41d63ed2010-01-08 00:14:31 +0000128 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000129 llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
130 const ObjCIvarDecl *Ivar);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000131 void EmitClassRef(const std::string &className);
David Chisnallef6e0f32010-02-03 15:59:02 +0000132 llvm::Value* EnforceType(CGBuilderTy B, llvm::Value *V, const llvm::Type *Ty){
133 if (V->getType() == Ty) return V;
134 return B.CreateBitCast(V, Ty);
135 }
Chris Lattner0f984262008-03-01 08:50:34 +0000136public:
Chris Lattnerdce14062008-06-26 04:19:03 +0000137 CGObjCGNU(CodeGen::CodeGenModule &cgm);
David Chisnall0d13f6f2010-01-23 02:40:42 +0000138 virtual llvm::Constant *GenerateConstantString(const StringLiteral *);
Mike Stump1eb44332009-09-09 15:08:12 +0000139 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000140 GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000141 QualType ResultType,
142 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000143 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000144 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000145 const CallArgList &CallArgs,
146 const ObjCMethodDecl *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000147 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000148 GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000149 QualType ResultType,
150 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000151 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000152 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000153 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000154 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000155 const CallArgList &CallArgs,
156 const ObjCMethodDecl *Method);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000157 virtual llvm::Value *GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000158 const ObjCInterfaceDecl *OID);
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000159 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel);
160 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
161 *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000162
163 virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000164 const ObjCContainerDecl *CD);
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000165 virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
166 virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000167 virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000168 const ObjCProtocolDecl *PD);
169 virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000170 virtual llvm::Function *ModuleInitFunction();
Daniel Dunbar49f66022008-09-24 03:38:44 +0000171 virtual llvm::Function *GetPropertyGetFunction();
172 virtual llvm::Function *GetPropertySetFunction();
Fariborz Jahanian6cc59062010-04-12 18:18:10 +0000173 virtual llvm::Function *GetCopyStructFunction();
Daniel Dunbar309a4362009-07-24 07:40:24 +0000174 virtual llvm::Constant *EnumerationMutationFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000175
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000176 virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
177 const Stmt &S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000178 virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
179 const ObjCAtThrowStmt &S);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000180 virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6dc23172008-11-18 21:45:40 +0000181 llvm::Value *AddrWeakObj);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000182 virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
183 llvm::Value *src, llvm::Value *dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000184 virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
185 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +0000186 virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +0000187 llvm::Value *src, llvm::Value *dest,
188 llvm::Value *ivarOffset);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000189 virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
190 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000191 virtual void EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +0000192 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000193 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +0000194 QualType Ty);
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000195 virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
196 QualType ObjectTy,
197 llvm::Value *BaseValue,
198 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000199 unsigned CVRQualifiers);
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000200 virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +0000201 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000202 const ObjCIvarDecl *Ivar);
Chris Lattner0f984262008-03-01 08:50:34 +0000203};
204} // end anonymous namespace
205
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000206
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000207/// Emits a reference to a dummy variable which is emitted with each class.
208/// This ensures that a linker error will be generated when trying to link
209/// together modules where a referenced class is not defined.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000210void CGObjCGNU::EmitClassRef(const std::string &className) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000211 std::string symbolRef = "__objc_class_ref_" + className;
212 // Don't emit two copies of the same symbol
Mike Stumpbb1c8602009-07-31 21:31:32 +0000213 if (TheModule.getGlobalVariable(symbolRef))
214 return;
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000215 std::string symbolName = "__objc_class_name_" + className;
216 llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
217 if (!ClassSymbol) {
Owen Anderson1c431b32009-07-08 19:05:04 +0000218 ClassSymbol = new llvm::GlobalVariable(TheModule, LongTy, false,
219 llvm::GlobalValue::ExternalLinkage, 0, symbolName);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000220 }
Owen Anderson1c431b32009-07-08 19:05:04 +0000221 new llvm::GlobalVariable(TheModule, ClassSymbol->getType(), true,
Chris Lattnerf35271b2009-08-05 05:25:18 +0000222 llvm::GlobalValue::WeakAnyLinkage, ClassSymbol, symbolRef);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000223}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000224
225static std::string SymbolNameForClass(const std::string &ClassName) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000226 return "_OBJC_CLASS_" + ClassName;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000227}
228
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000229static std::string SymbolNameForMethod(const std::string &ClassName, const
230 std::string &CategoryName, const std::string &MethodName, bool isClassMethod)
231{
David Chisnalld3467362010-01-14 14:08:19 +0000232 std::string MethodNameColonStripped = MethodName;
233 std::replace(MethodNameColonStripped.begin(), MethodNameColonStripped.end(),
234 ':', '_');
235 return std::string(isClassMethod ? "_c_" : "_i_") + ClassName + "_" +
236 CategoryName + "_" + MethodNameColonStripped;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000237}
238
Chris Lattnerdce14062008-06-26 04:19:03 +0000239CGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000240 : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
Owen Andersona1cf15f2009-07-14 23:10:40 +0000241 MetaClassPtrAlias(0), VMContext(cgm.getLLVMContext()) {
Chris Lattnere160c9b2009-01-27 05:06:01 +0000242 IntTy = cast<llvm::IntegerType>(
243 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
244 LongTy = cast<llvm::IntegerType>(
245 CGM.getTypes().ConvertType(CGM.getContext().LongTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000246
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000247 Int8Ty = llvm::Type::getInt8Ty(VMContext);
248 // C string type. Used in lots of places.
249 PtrToInt8Ty = llvm::PointerType::getUnqual(Int8Ty);
250
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000251 Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000252 Zeros[1] = Zeros[0];
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000253 NULLPtr = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Chris Lattner391d77a2008-03-30 23:03:07 +0000254 // Get the selector Type.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000255 QualType selTy = CGM.getContext().getObjCSelType();
256 if (QualType() == selTy) {
257 SelectorTy = PtrToInt8Ty;
258 } else {
259 SelectorTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(selTy));
260 }
Chris Lattnere160c9b2009-01-27 05:06:01 +0000261
Owen Anderson96e0fc72009-07-29 22:16:19 +0000262 PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
Chris Lattner391d77a2008-03-30 23:03:07 +0000263 PtrTy = PtrToInt8Ty;
Mike Stump1eb44332009-09-09 15:08:12 +0000264
Chris Lattner391d77a2008-03-30 23:03:07 +0000265 // Object type
John McCallead608a2010-02-26 00:48:12 +0000266 ASTIdTy = CGM.getContext().getCanonicalType(CGM.getContext().getObjCIdType());
David Chisnall0d13f6f2010-01-23 02:40:42 +0000267 if (QualType() == ASTIdTy) {
268 IdTy = PtrToInt8Ty;
269 } else {
270 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
271 }
David Chisnallef6e0f32010-02-03 15:59:02 +0000272 PtrToIdTy = llvm::PointerType::getUnqual(IdTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000273
Chris Lattner391d77a2008-03-30 23:03:07 +0000274 // IMP type
275 std::vector<const llvm::Type*> IMPArgs;
276 IMPArgs.push_back(IdTy);
277 IMPArgs.push_back(SelectorTy);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000278 IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
David Chisnallef6e0f32010-02-03 15:59:02 +0000279
280 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
281 // Get selectors needed in GC mode
282 RetainSel = GetNullarySelector("retain", CGM.getContext());
283 ReleaseSel = GetNullarySelector("release", CGM.getContext());
284 AutoreleaseSel = GetNullarySelector("autorelease", CGM.getContext());
285
286 // Get functions needed in GC mode
287
288 // id objc_assign_ivar(id, id, ptrdiff_t);
289 std::vector<const llvm::Type*> Args(1, IdTy);
290 Args.push_back(PtrToIdTy);
291 // FIXME: ptrdiff_t
292 Args.push_back(LongTy);
293 llvm::FunctionType *FTy = llvm::FunctionType::get(IdTy, Args, false);
294 IvarAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_ivar");
295 // id objc_assign_strongCast (id, id*)
296 Args.pop_back();
297 FTy = llvm::FunctionType::get(IdTy, Args, false);
298 StrongCastAssignFn =
299 CGM.CreateRuntimeFunction(FTy, "objc_assign_strongCast");
300 // id objc_assign_global(id, id*);
301 FTy = llvm::FunctionType::get(IdTy, Args, false);
302 GlobalAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_global");
303 // id objc_assign_weak(id, id*);
304 FTy = llvm::FunctionType::get(IdTy, Args, false);
305 WeakAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_weak");
306 // id objc_read_weak(id*);
307 Args.clear();
308 Args.push_back(PtrToIdTy);
309 FTy = llvm::FunctionType::get(IdTy, Args, false);
310 WeakReadFn = CGM.CreateRuntimeFunction(FTy, "objc_read_weak");
311 // void *objc_memmove_collectable(void*, void *, size_t);
312 Args.clear();
313 Args.push_back(PtrToInt8Ty);
314 Args.push_back(PtrToInt8Ty);
315 // FIXME: size_t
316 Args.push_back(LongTy);
317 FTy = llvm::FunctionType::get(IdTy, Args, false);
318 MemMoveFn = CGM.CreateRuntimeFunction(FTy, "objc_memmove_collectable");
319 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000320}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000321
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000322// This has to perform the lookup every time, since posing and related
323// techniques can modify the name -> class mapping.
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000324llvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000325 const ObjCInterfaceDecl *OID) {
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000326 llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
David Chisnall41d63ed2010-01-08 00:14:31 +0000327 // With the incompatible ABI, this will need to be replaced with a direct
328 // reference to the class symbol. For the compatible nonfragile ABI we are
329 // still performing this lookup at run time but emitting the symbol for the
330 // class externally so that we can make the switch later.
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000331 EmitClassRef(OID->getNameAsString());
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000332 ClassName = Builder.CreateStructGEP(ClassName, 0);
333
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000334 std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000335 llvm::Constant *ClassLookupFn =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000336 CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000337 Params,
338 true),
339 "objc_lookup_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000340 return Builder.CreateCall(ClassLookupFn, ClassName);
Chris Lattner391d77a2008-03-30 23:03:07 +0000341}
342
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000343llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel) {
Chris Lattner077bf5e2008-11-24 03:33:13 +0000344 llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
Chris Lattner8e67b632008-06-26 04:37:12 +0000345 if (US == 0)
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000346 US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000347 llvm::GlobalValue::PrivateLinkage,
348 ".objc_untyped_selector_alias"+Sel.getAsString(),
Chris Lattner8e67b632008-06-26 04:37:12 +0000349 NULL, &TheModule);
Mike Stump1eb44332009-09-09 15:08:12 +0000350
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000351 return Builder.CreateLoad(US);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000352}
353
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000354llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000355 *Method) {
356
357 std::string SelName = Method->getSelector().getAsString();
358 std::string SelTypes;
359 CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
360 // Typed selectors
361 TypedSelector Selector = TypedSelector(SelName,
362 SelTypes);
363
364 // If it's already cached, return it.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000365 if (TypedSelectors[Selector]) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000366 return Builder.CreateLoad(TypedSelectors[Selector]);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000367 }
368
369 // If it isn't, cache it.
370 llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000371 llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000372 llvm::GlobalValue::PrivateLinkage, ".objc_selector_alias" + SelName,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000373 NULL, &TheModule);
374 TypedSelectors[Selector] = Sel;
375
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000376 return Builder.CreateLoad(Sel);
Chris Lattner8e67b632008-06-26 04:37:12 +0000377}
378
Chris Lattner5e7dcc62008-06-26 04:44:19 +0000379llvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
380 const std::string &Name) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000381 llvm::Constant *ConstStr = CGM.GetAddrOfConstantCString(Str, Name.c_str());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000382 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000383}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000384llvm::Constant *CGObjCGNU::ExportUniqueString(const std::string &Str,
385 const std::string prefix) {
386 std::string name = prefix + Str;
387 llvm::Constant *ConstStr = TheModule.getGlobalVariable(name);
388 if (!ConstStr) {
389 llvm::Constant *value = llvm::ConstantArray::get(VMContext, Str, true);
390 ConstStr = new llvm::GlobalVariable(TheModule, value->getType(), true,
391 llvm::GlobalValue::LinkOnceODRLinkage, value, prefix + Str);
392 }
393 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
394}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000395
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000396llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000397 std::vector<llvm::Constant*> &V, llvm::StringRef Name,
David Chisnall41d63ed2010-01-08 00:14:31 +0000398 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson08e25242009-07-27 22:29:56 +0000399 llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000400 return new llvm::GlobalVariable(TheModule, Ty, false,
401 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000402}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000403
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000404llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000405 std::vector<llvm::Constant*> &V, llvm::StringRef Name,
David Chisnall41d63ed2010-01-08 00:14:31 +0000406 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson7db6d832009-07-28 18:33:04 +0000407 llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000408 return new llvm::GlobalVariable(TheModule, Ty, false,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000409 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000410}
411
412/// Generate an NSConstantString object.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000413llvm::Constant *CGObjCGNU::GenerateConstantString(const StringLiteral *SL) {
David Chisnall48272a02010-01-27 12:49:23 +0000414
David Chisnall0d13f6f2010-01-23 02:40:42 +0000415 std::string Str(SL->getStrData(), SL->getByteLength());
416
David Chisnall48272a02010-01-27 12:49:23 +0000417 // Look for an existing one
418 llvm::StringMap<llvm::Constant*>::iterator old = ObjCStrings.find(Str);
419 if (old != ObjCStrings.end())
420 return old->getValue();
421
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000422 std::vector<llvm::Constant*> Ivars;
423 Ivars.push_back(NULLPtr);
Chris Lattner13fd7e52008-06-21 21:44:18 +0000424 Ivars.push_back(MakeConstantString(Str));
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000425 Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000426 llvm::Constant *ObjCStr = MakeGlobal(
Owen Anderson47a434f2009-08-05 23:18:46 +0000427 llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000428 Ivars, ".objc_str");
David Chisnall48272a02010-01-27 12:49:23 +0000429 ObjCStr = llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty);
430 ObjCStrings[Str] = ObjCStr;
431 ConstantStrings.push_back(ObjCStr);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000432 return ObjCStr;
433}
434
435///Generates a message send where the super is the receiver. This is a message
436///send to self with special delivery semantics indicating which class's method
437///should be called.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000438CodeGen::RValue
439CGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000440 QualType ResultType,
441 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000442 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000443 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000444 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000445 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000446 const CallArgList &CallArgs,
447 const ObjCMethodDecl *Method) {
David Chisnallef6e0f32010-02-03 15:59:02 +0000448 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
449 if (Sel == RetainSel || Sel == AutoreleaseSel) {
450 return RValue::get(Receiver);
451 }
452 if (Sel == ReleaseSel) {
453 return RValue::get(0);
454 }
455 }
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000456 llvm::Value *cmd = GetSelector(CGF.Builder, Sel);
457
458 CallArgList ActualArgs;
459
460 ActualArgs.push_back(
David Chisnall0f436562009-08-17 16:35:33 +0000461 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
462 ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000463 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
464 CGF.getContext().getObjCSelType()));
465 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
466
467 CodeGenTypes &Types = CGM.getTypes();
John McCall04a67a62010-02-05 21:31:56 +0000468 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs,
Rafael Espindola264ba482010-03-30 20:24:48 +0000469 FunctionType::ExtInfo());
Daniel Dunbar67939662009-09-17 04:01:40 +0000470 const llvm::FunctionType *impType =
471 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000472
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000473 llvm::Value *ReceiverClass = 0;
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000474 if (isCategoryImpl) {
475 llvm::Constant *classLookupFunction = 0;
476 std::vector<const llvm::Type*> Params;
477 Params.push_back(PtrTy);
478 if (IsClassMessage) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000479 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000480 IdTy, Params, true), "objc_get_meta_class");
481 } else {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000482 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000483 IdTy, Params, true), "objc_get_class");
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000484 }
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000485 ReceiverClass = CGF.Builder.CreateCall(classLookupFunction,
486 MakeConstantString(Class->getNameAsString()));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000487 } else {
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000488 // Set up global aliases for the metaclass or class pointer if they do not
489 // already exist. These will are forward-references which will be set to
Mike Stumpbb1c8602009-07-31 21:31:32 +0000490 // pointers to the class and metaclass structure created for the runtime
491 // load function. To send a message to super, we look up the value of the
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000492 // super_class pointer from either the class or metaclass structure.
493 if (IsClassMessage) {
494 if (!MetaClassPtrAlias) {
495 MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
496 llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
497 Class->getNameAsString(), NULL, &TheModule);
498 }
499 ReceiverClass = MetaClassPtrAlias;
500 } else {
501 if (!ClassPtrAlias) {
502 ClassPtrAlias = new llvm::GlobalAlias(IdTy,
503 llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
504 Class->getNameAsString(), NULL, &TheModule);
505 }
506 ReceiverClass = ClassPtrAlias;
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000507 }
Chris Lattner71238f62009-04-25 23:19:45 +0000508 }
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000509 // Cast the pointer to a simplified version of the class structure
Mike Stump1eb44332009-09-09 15:08:12 +0000510 ReceiverClass = CGF.Builder.CreateBitCast(ReceiverClass,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000511 llvm::PointerType::getUnqual(
Owen Anderson47a434f2009-08-05 23:18:46 +0000512 llvm::StructType::get(VMContext, IdTy, IdTy, NULL)));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000513 // Get the superclass pointer
514 ReceiverClass = CGF.Builder.CreateStructGEP(ReceiverClass, 1);
515 // Load the superclass pointer
516 ReceiverClass = CGF.Builder.CreateLoad(ReceiverClass);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000517 // Construct the structure used to look up the IMP
Owen Anderson47a434f2009-08-05 23:18:46 +0000518 llvm::StructType *ObjCSuperTy = llvm::StructType::get(VMContext,
519 Receiver->getType(), IdTy, NULL);
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000520 llvm::Value *ObjCSuper = CGF.Builder.CreateAlloca(ObjCSuperTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000521
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000522 CGF.Builder.CreateStore(Receiver, CGF.Builder.CreateStructGEP(ObjCSuper, 0));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000523 CGF.Builder.CreateStore(ReceiverClass,
524 CGF.Builder.CreateStructGEP(ObjCSuper, 1));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000525
526 // Get the IMP
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000527 std::vector<const llvm::Type*> Params;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000528 Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000529 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000530 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000531 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
532 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000533 "objc_msg_lookup_super");
534
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000535 llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000536 llvm::Value *imp = CGF.Builder.CreateCall(lookupFunction, lookupArgs,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000537 lookupArgs+2);
538
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000539 return CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000540}
541
Mike Stump1eb44332009-09-09 15:08:12 +0000542/// Generate code for a message send expression.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000543CodeGen::RValue
544CGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000545 QualType ResultType,
546 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000547 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000548 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000549 const CallArgList &CallArgs,
550 const ObjCMethodDecl *Method) {
David Chisnall664b7c72010-04-27 15:08:48 +0000551 // Strip out message sends to retain / release in GC mode
David Chisnallef6e0f32010-02-03 15:59:02 +0000552 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
553 if (Sel == RetainSel || Sel == AutoreleaseSel) {
554 return RValue::get(Receiver);
555 }
556 if (Sel == ReleaseSel) {
557 return RValue::get(0);
558 }
559 }
David Chisnall664b7c72010-04-27 15:08:48 +0000560
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000561 CGBuilderTy &Builder = CGF.Builder;
David Chisnall664b7c72010-04-27 15:08:48 +0000562
563 // If the return type is something that goes in an integer register, the
564 // runtime will handle 0 returns. For other cases, we fill in the 0 value
565 // ourselves.
566 //
567 // The language spec says the result of this kind of message send is
568 // undefined, but lots of people seem to have forgotten to read that
569 // paragraph and insist on sending messages to nil that have structure
570 // returns. With GCC, this generates a random return value (whatever happens
571 // to be on the stack / in those registers at the time) on most platforms,
572 // and generates a SegV on SPARC. With LLVM it corrupts the stack.
573 bool isPointerSizedReturn = false;
David Chisnall254e2d92010-04-27 20:33:30 +0000574 if (ResultType->isAnyPointerType() || ResultType->isIntegralType() ||
575 ResultType->isVoidType())
David Chisnall664b7c72010-04-27 15:08:48 +0000576 isPointerSizedReturn = true;
577
578 llvm::BasicBlock *startBB = 0;
579 llvm::BasicBlock *messageBB = 0;
580 llvm::BasicBlock *contiueBB = 0;
581
582 if (!isPointerSizedReturn) {
583 startBB = Builder.GetInsertBlock();
584 messageBB = CGF.createBasicBlock("msgSend");
585 contiueBB = CGF.createBasicBlock("continue");
586
587 llvm::Value *isNil = Builder.CreateICmpEQ(Receiver,
588 llvm::Constant::getNullValue(Receiver->getType()));
589 Builder.CreateCondBr(isNil, contiueBB, messageBB);
590 CGF.EmitBlock(messageBB);
591 }
592
David Chisnall0f436562009-08-17 16:35:33 +0000593 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000594 llvm::Value *cmd;
595 if (Method)
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000596 cmd = GetSelector(Builder, Method);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000597 else
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000598 cmd = GetSelector(Builder, Sel);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000599 CallArgList ActualArgs;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000600
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000601 Receiver = Builder.CreateBitCast(Receiver, IdTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000602 ActualArgs.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000603 std::make_pair(RValue::get(Receiver), ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000604 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
605 CGF.getContext().getObjCSelType()));
606 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
607
608 CodeGenTypes &Types = CGM.getTypes();
John McCall04a67a62010-02-05 21:31:56 +0000609 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs,
Rafael Espindola264ba482010-03-30 20:24:48 +0000610 FunctionType::ExtInfo());
Daniel Dunbar67939662009-09-17 04:01:40 +0000611 const llvm::FunctionType *impType =
612 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000613
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000614 llvm::Value *imp;
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000615 // For sender-aware dispatch, we pass the sender as the third argument to a
616 // lookup function. When sending messages from C code, the sender is nil.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000617 // objc_msg_lookup_sender(id *receiver, SEL selector, id sender);
618 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
619
620 std::vector<const llvm::Type*> Params;
621 llvm::Value *ReceiverPtr = CGF.CreateTempAlloca(Receiver->getType());
622 Builder.CreateStore(Receiver, ReceiverPtr);
623 Params.push_back(ReceiverPtr->getType());
624 Params.push_back(SelectorTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000625 llvm::Value *self;
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000626
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000627 if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
628 self = CGF.LoadObjCSelf();
629 } else {
Owen Anderson03e20502009-07-30 23:11:26 +0000630 self = llvm::ConstantPointerNull::get(IdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000631 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000632
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000633 Params.push_back(self->getType());
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000634
635 // The lookup function returns a slot, which can be safely cached.
636 llvm::Type *SlotTy = llvm::StructType::get(VMContext, PtrTy, PtrTy, PtrTy,
637 IntTy, llvm::PointerType::getUnqual(impType), NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000638 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000639 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000640 llvm::PointerType::getUnqual(SlotTy), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000641 "objc_msg_lookup_sender");
642
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000643 // The lookup function is guaranteed not to capture the receiver pointer.
644 if (llvm::Function *LookupFn = dyn_cast<llvm::Function>(lookupFunction)) {
645 LookupFn->setDoesNotCapture(1);
646 }
647
648 llvm::Value *slot =
649 Builder.CreateCall3(lookupFunction, ReceiverPtr, cmd, self);
650 imp = Builder.CreateLoad(Builder.CreateStructGEP(slot, 4));
651 // The lookup function may have changed the receiver, so make sure we use
652 // the new one.
653 ActualArgs[0] =
654 std::make_pair(RValue::get(Builder.CreateLoad(ReceiverPtr)), ASTIdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000655 } else {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000656 std::vector<const llvm::Type*> Params;
657 Params.push_back(Receiver->getType());
658 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000659 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000660 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
661 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000662 "objc_msg_lookup");
663
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000664 imp = Builder.CreateCall2(lookupFunction, Receiver, cmd);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000665 }
Chris Lattner3eae03e2008-05-06 00:56:42 +0000666
David Chisnall664b7c72010-04-27 15:08:48 +0000667 RValue msgRet =
668 CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
669
670 if (!isPointerSizedReturn) {
671 CGF.EmitBlock(contiueBB);
672 if (msgRet.isScalar()) {
673 llvm::Value *v = msgRet.getScalarVal();
674 llvm::PHINode *phi = Builder.CreatePHI(v->getType());
675 phi->addIncoming(v, messageBB);
676 phi->addIncoming(llvm::Constant::getNullValue(v->getType()), startBB);
677 msgRet = RValue::get(phi);
678 } else if (msgRet.isAggregate()) {
679 llvm::Value *v = msgRet.getAggregateAddr();
680 llvm::PHINode *phi = Builder.CreatePHI(v->getType());
681 const llvm::PointerType *RetTy = cast<llvm::PointerType>(v->getType());
682 llvm::AllocaInst *NullVal = CGF.CreateTempAlloca(RetTy, "null");
683 CGF.InitTempAlloca(NullVal,
684 llvm::Constant::getNullValue(RetTy->getElementType()));
685 phi->addIncoming(v, messageBB);
686 phi->addIncoming(NullVal, startBB);
687 msgRet = RValue::getAggregate(phi);
688 } else /* isComplex() */ {
689 std::pair<llvm::Value*,llvm::Value*> v = msgRet.getComplexVal();
690 llvm::PHINode *phi = Builder.CreatePHI(v.first->getType());
691 phi->addIncoming(v.first, messageBB);
692 phi->addIncoming(llvm::Constant::getNullValue(v.first->getType()),
693 startBB);
694 llvm::PHINode *phi2 = Builder.CreatePHI(v.second->getType());
695 phi2->addIncoming(v.second, messageBB);
696 phi2->addIncoming(llvm::Constant::getNullValue(v.second->getType()),
697 startBB);
698 msgRet = RValue::getComplex(phi, phi2);
699 }
700 }
701 return msgRet;
Chris Lattner0f984262008-03-01 08:50:34 +0000702}
703
Mike Stump1eb44332009-09-09 15:08:12 +0000704/// Generates a MethodList. Used in construction of a objc_class and
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000705/// objc_category structures.
706llvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
Mike Stump1eb44332009-09-09 15:08:12 +0000707 const std::string &CategoryName,
708 const llvm::SmallVectorImpl<Selector> &MethodSels,
709 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000710 bool isClassMethodList) {
David Chisnall0f436562009-08-17 16:35:33 +0000711 if (MethodSels.empty())
712 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000713 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000714 llvm::StructType *ObjCMethodTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000715 PtrToInt8Ty, // Really a selector, but the runtime creates it us.
716 PtrToInt8Ty, // Method types
Owen Anderson96e0fc72009-07-29 22:16:19 +0000717 llvm::PointerType::getUnqual(IMPTy), //Method pointer
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000718 NULL);
719 std::vector<llvm::Constant*> Methods;
720 std::vector<llvm::Constant*> Elements;
721 for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
722 Elements.clear();
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000723 if (llvm::Constant *Method =
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000724 TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
Chris Lattner077bf5e2008-11-24 03:33:13 +0000725 MethodSels[i].getAsString(),
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000726 isClassMethodList))) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000727 llvm::Constant *C = MakeConstantString(MethodSels[i].getAsString());
728 Elements.push_back(C);
729 Elements.push_back(MethodTypes[i]);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000730 Method = llvm::ConstantExpr::getBitCast(Method,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000731 llvm::PointerType::getUnqual(IMPTy));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000732 Elements.push_back(Method);
Owen Anderson08e25242009-07-27 22:29:56 +0000733 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000734 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000735 }
736
737 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000738 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000739 Methods.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000740 llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
Chris Lattnerfba67632008-06-26 04:52:29 +0000741 Methods);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000742
743 // Structure containing list pointer, array and array count
744 llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
Owen Anderson8c8f69e2009-08-13 23:27:53 +0000745 llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get(VMContext);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000746 llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
Owen Anderson47a434f2009-08-05 23:18:46 +0000747 llvm::StructType *ObjCMethodListTy = llvm::StructType::get(VMContext,
Mike Stump1eb44332009-09-09 15:08:12 +0000748 NextPtrTy,
749 IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000750 ObjCMethodArrayTy,
751 NULL);
752 // Refine next pointer type to concrete type
753 llvm::cast<llvm::OpaqueType>(
754 OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
755 ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
756
757 Methods.clear();
Owen Anderson03e20502009-07-30 23:11:26 +0000758 Methods.push_back(llvm::ConstantPointerNull::get(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000759 llvm::PointerType::getUnqual(ObjCMethodListTy)));
Owen Anderson0032b272009-08-13 21:57:51 +0000760 Methods.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000761 MethodTypes.size()));
762 Methods.push_back(MethodArray);
Mike Stump1eb44332009-09-09 15:08:12 +0000763
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000764 // Create an instance of the structure
765 return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
766}
767
768/// Generates an IvarList. Used in construction of a objc_class.
769llvm::Constant *CGObjCGNU::GenerateIvarList(
770 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
771 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
772 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets) {
David Chisnall18044632009-11-16 19:05:54 +0000773 if (IvarNames.size() == 0)
774 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000775 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000776 llvm::StructType *ObjCIvarTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000777 PtrToInt8Ty,
778 PtrToInt8Ty,
779 IntTy,
780 NULL);
781 std::vector<llvm::Constant*> Ivars;
782 std::vector<llvm::Constant*> Elements;
783 for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
784 Elements.clear();
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000785 Elements.push_back(IvarNames[i]);
786 Elements.push_back(IvarTypes[i]);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000787 Elements.push_back(IvarOffsets[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000788 Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000789 }
790
791 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000792 llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000793 IvarNames.size());
794
Mike Stump1eb44332009-09-09 15:08:12 +0000795
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000796 Elements.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000797 Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
Owen Anderson7db6d832009-07-28 18:33:04 +0000798 Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000799 // Structure containing array and array count
Owen Anderson47a434f2009-08-05 23:18:46 +0000800 llvm::StructType *ObjCIvarListTy = llvm::StructType::get(VMContext, IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000801 ObjCIvarArrayTy,
802 NULL);
803
804 // Create an instance of the structure
805 return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
806}
807
808/// Generate a class structure
809llvm::Constant *CGObjCGNU::GenerateClassStructure(
810 llvm::Constant *MetaClass,
811 llvm::Constant *SuperClass,
812 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000813 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000814 llvm::Constant *Version,
815 llvm::Constant *InstanceSize,
816 llvm::Constant *IVars,
817 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000818 llvm::Constant *Protocols,
819 llvm::Constant *IvarOffsets,
820 llvm::Constant *Properties) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000821 // Set up the class structure
822 // Note: Several of these are char*s when they should be ids. This is
823 // because the runtime performs this translation on load.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000824 //
825 // Fields marked New ABI are part of the GNUstep runtime. We emit them
826 // anyway; the classes will still work with the GNU runtime, they will just
827 // be ignored.
Owen Anderson47a434f2009-08-05 23:18:46 +0000828 llvm::StructType *ClassTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000829 PtrToInt8Ty, // class_pointer
830 PtrToInt8Ty, // super_class
831 PtrToInt8Ty, // name
832 LongTy, // version
833 LongTy, // info
834 LongTy, // instance_size
835 IVars->getType(), // ivars
836 Methods->getType(), // methods
Mike Stump1eb44332009-09-09 15:08:12 +0000837 // These are all filled in by the runtime, so we pretend
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000838 PtrTy, // dtable
839 PtrTy, // subclass_list
840 PtrTy, // sibling_class
841 PtrTy, // protocols
842 PtrTy, // gc_object_type
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000843 // New ABI:
844 LongTy, // abi_version
845 IvarOffsets->getType(), // ivar_offsets
846 Properties->getType(), // properties
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000847 NULL);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000848 llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000849 // Fill in the structure
850 std::vector<llvm::Constant*> Elements;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000851 Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000852 Elements.push_back(SuperClass);
Chris Lattnerd002cc62008-06-26 04:47:04 +0000853 Elements.push_back(MakeConstantString(Name, ".class_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000854 Elements.push_back(Zero);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000855 Elements.push_back(llvm::ConstantInt::get(LongTy, info));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000856 Elements.push_back(InstanceSize);
857 Elements.push_back(IVars);
858 Elements.push_back(Methods);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000859 Elements.push_back(NULLPtr);
860 Elements.push_back(NULLPtr);
861 Elements.push_back(NULLPtr);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000862 Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000863 Elements.push_back(NULLPtr);
864 Elements.push_back(Zero);
865 Elements.push_back(IvarOffsets);
866 Elements.push_back(Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000867 // Create an instance of the structure
David Chisnall41d63ed2010-01-08 00:14:31 +0000868 // This is now an externally visible symbol, so that we can speed up class
869 // messages in the next ABI.
870 return MakeGlobal(ClassTy, Elements, SymbolNameForClass(Name),
871 llvm::GlobalValue::ExternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000872}
873
874llvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
875 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
876 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes) {
Mike Stump1eb44332009-09-09 15:08:12 +0000877 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000878 llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000879 PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
880 PtrToInt8Ty,
881 NULL);
882 std::vector<llvm::Constant*> Methods;
883 std::vector<llvm::Constant*> Elements;
884 for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
885 Elements.clear();
Mike Stump1eb44332009-09-09 15:08:12 +0000886 Elements.push_back(MethodNames[i]);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000887 Elements.push_back(MethodTypes[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000888 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000889 }
Owen Anderson96e0fc72009-07-29 22:16:19 +0000890 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000891 MethodNames.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000892 llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000893 Methods);
Owen Anderson47a434f2009-08-05 23:18:46 +0000894 llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000895 IntTy, ObjCMethodArrayTy, NULL);
896 Methods.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000897 Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000898 Methods.push_back(Array);
899 return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
900}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000901
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000902// Create the protocol list structure used in classes, categories and so on
903llvm::Constant *CGObjCGNU::GenerateProtocolList(
904 const llvm::SmallVectorImpl<std::string> &Protocols) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000905 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000906 Protocols.size());
Owen Anderson47a434f2009-08-05 23:18:46 +0000907 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000908 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
909 LongTy,//FIXME: Should be size_t
910 ProtocolArrayTy,
911 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000912 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000913 for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
914 iter != endIter ; iter++) {
David Chisnallff80fab2009-11-20 14:50:59 +0000915 llvm::Constant *protocol = 0;
916 llvm::StringMap<llvm::Constant*>::iterator value =
917 ExistingProtocols.find(*iter);
918 if (value == ExistingProtocols.end()) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000919 protocol = GenerateEmptyProtocol(*iter);
David Chisnallff80fab2009-11-20 14:50:59 +0000920 } else {
921 protocol = value->getValue();
922 }
Owen Anderson3c4972d2009-07-29 18:54:39 +0000923 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(protocol,
Owen Andersona1cf15f2009-07-14 23:10:40 +0000924 PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000925 Elements.push_back(Ptr);
926 }
Owen Anderson7db6d832009-07-28 18:33:04 +0000927 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000928 Elements);
929 Elements.clear();
930 Elements.push_back(NULLPtr);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000931 Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000932 Elements.push_back(ProtocolArray);
933 return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
934}
935
Mike Stump1eb44332009-09-09 15:08:12 +0000936llvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000937 const ObjCProtocolDecl *PD) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000938 llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
Mike Stump1eb44332009-09-09 15:08:12 +0000939 const llvm::Type *T =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000940 CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
Owen Anderson96e0fc72009-07-29 22:16:19 +0000941 return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000942}
943
944llvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
945 const std::string &ProtocolName) {
946 llvm::SmallVector<std::string, 0> EmptyStringVector;
947 llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
948
949 llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000950 llvm::Constant *MethodList =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000951 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
952 // Protocols are objects containing lists of the methods implemented and
953 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000954 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000955 PtrToInt8Ty,
956 ProtocolList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000957 MethodList->getType(),
958 MethodList->getType(),
959 MethodList->getType(),
960 MethodList->getType(),
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000961 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000962 std::vector<llvm::Constant*> Elements;
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000963 // The isa pointer must be set to a magic number so the runtime knows it's
964 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000965 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
966 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000967 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000968 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000969 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
970 Elements.push_back(ProtocolList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000971 Elements.push_back(MethodList);
972 Elements.push_back(MethodList);
973 Elements.push_back(MethodList);
974 Elements.push_back(MethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000975 return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000976}
977
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000978void CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
979 ASTContext &Context = CGM.getContext();
Chris Lattner8ec03f52008-11-24 03:54:41 +0000980 std::string ProtocolName = PD->getNameAsString();
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000981 llvm::SmallVector<std::string, 16> Protocols;
982 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
983 E = PD->protocol_end(); PI != E; ++PI)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000984 Protocols.push_back((*PI)->getNameAsString());
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000985 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
986 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000987 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodNames;
988 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodTypes;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000989 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
990 E = PD->instmeth_end(); iter != E; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000991 std::string TypeStr;
992 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000993 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
994 InstanceMethodNames.push_back(
995 MakeConstantString((*iter)->getSelector().getAsString()));
996 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
997 } else {
998 OptionalInstanceMethodNames.push_back(
999 MakeConstantString((*iter)->getSelector().getAsString()));
1000 OptionalInstanceMethodTypes.push_back(MakeConstantString(TypeStr));
1001 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001002 }
1003 // Collect information about class methods:
1004 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
1005 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001006 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodNames;
1007 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001008 for (ObjCProtocolDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001009 iter = PD->classmeth_begin(), endIter = PD->classmeth_end();
1010 iter != endIter ; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001011 std::string TypeStr;
1012 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001013 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
1014 ClassMethodNames.push_back(
1015 MakeConstantString((*iter)->getSelector().getAsString()));
1016 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
1017 } else {
1018 OptionalClassMethodNames.push_back(
1019 MakeConstantString((*iter)->getSelector().getAsString()));
1020 OptionalClassMethodTypes.push_back(MakeConstantString(TypeStr));
1021 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001022 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001023
1024 llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
1025 llvm::Constant *InstanceMethodList =
1026 GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
1027 llvm::Constant *ClassMethodList =
1028 GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001029 llvm::Constant *OptionalInstanceMethodList =
1030 GenerateProtocolMethodList(OptionalInstanceMethodNames,
1031 OptionalInstanceMethodTypes);
1032 llvm::Constant *OptionalClassMethodList =
1033 GenerateProtocolMethodList(OptionalClassMethodNames,
1034 OptionalClassMethodTypes);
1035
1036 // Property metadata: name, attributes, isSynthesized, setter name, setter
1037 // types, getter name, getter types.
1038 // The isSynthesized value is always set to 0 in a protocol. It exists to
1039 // simplify the runtime library by allowing it to use the same data
1040 // structures for protocol metadata everywhere.
1041 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1042 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1043 PtrToInt8Ty, NULL);
1044 std::vector<llvm::Constant*> Properties;
1045 std::vector<llvm::Constant*> OptionalProperties;
1046
1047 // Add all of the property methods need adding to the method list and to the
1048 // property metadata list.
1049 for (ObjCContainerDecl::prop_iterator
1050 iter = PD->prop_begin(), endIter = PD->prop_end();
1051 iter != endIter ; iter++) {
1052 std::vector<llvm::Constant*> Fields;
1053 ObjCPropertyDecl *property = (*iter);
1054
1055 Fields.push_back(MakeConstantString(property->getNameAsString()));
1056 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1057 property->getPropertyAttributes()));
1058 Fields.push_back(llvm::ConstantInt::get(Int8Ty, 0));
1059 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
1060 std::string TypeStr;
1061 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1062 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1063 InstanceMethodTypes.push_back(TypeEncoding);
1064 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1065 Fields.push_back(TypeEncoding);
1066 } else {
1067 Fields.push_back(NULLPtr);
1068 Fields.push_back(NULLPtr);
1069 }
1070 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1071 std::string TypeStr;
1072 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1073 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1074 InstanceMethodTypes.push_back(TypeEncoding);
1075 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1076 Fields.push_back(TypeEncoding);
1077 } else {
1078 Fields.push_back(NULLPtr);
1079 Fields.push_back(NULLPtr);
1080 }
1081 if (property->getPropertyImplementation() == ObjCPropertyDecl::Optional) {
1082 OptionalProperties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1083 } else {
1084 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1085 }
1086 }
1087 llvm::Constant *PropertyArray = llvm::ConstantArray::get(
1088 llvm::ArrayType::get(PropertyMetadataTy, Properties.size()), Properties);
1089 llvm::Constant* PropertyListInitFields[] =
1090 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1091
1092 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001093 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001094 llvm::Constant *PropertyList = new llvm::GlobalVariable(TheModule,
1095 PropertyListInit->getType(), false, llvm::GlobalValue::InternalLinkage,
1096 PropertyListInit, ".objc_property_list");
1097
1098 llvm::Constant *OptionalPropertyArray =
1099 llvm::ConstantArray::get(llvm::ArrayType::get(PropertyMetadataTy,
1100 OptionalProperties.size()) , OptionalProperties);
1101 llvm::Constant* OptionalPropertyListInitFields[] = {
1102 llvm::ConstantInt::get(IntTy, OptionalProperties.size()), NULLPtr,
1103 OptionalPropertyArray };
1104
1105 llvm::Constant *OptionalPropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001106 llvm::ConstantStruct::get(VMContext, OptionalPropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001107 llvm::Constant *OptionalPropertyList = new llvm::GlobalVariable(TheModule,
1108 OptionalPropertyListInit->getType(), false,
1109 llvm::GlobalValue::InternalLinkage, OptionalPropertyListInit,
1110 ".objc_property_list");
1111
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001112 // Protocols are objects containing lists of the methods implemented and
1113 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +00001114 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001115 PtrToInt8Ty,
1116 ProtocolList->getType(),
1117 InstanceMethodList->getType(),
1118 ClassMethodList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001119 OptionalInstanceMethodList->getType(),
1120 OptionalClassMethodList->getType(),
1121 PropertyList->getType(),
1122 OptionalPropertyList->getType(),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001123 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +00001124 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001125 // The isa pointer must be set to a magic number so the runtime knows it's
1126 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001127 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
1128 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +00001129 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001130 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001131 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
1132 Elements.push_back(ProtocolList);
1133 Elements.push_back(InstanceMethodList);
1134 Elements.push_back(ClassMethodList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001135 Elements.push_back(OptionalInstanceMethodList);
1136 Elements.push_back(OptionalClassMethodList);
1137 Elements.push_back(PropertyList);
1138 Elements.push_back(OptionalPropertyList);
Mike Stump1eb44332009-09-09 15:08:12 +00001139 ExistingProtocols[ProtocolName] =
Owen Anderson3c4972d2009-07-29 18:54:39 +00001140 llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001141 ".objc_protocol"), IdTy);
1142}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001143void CGObjCGNU::GenerateProtocolHolderCategory(void) {
1144 // Collect information about instance methods
1145 llvm::SmallVector<Selector, 1> MethodSels;
1146 llvm::SmallVector<llvm::Constant*, 1> MethodTypes;
1147
1148 std::vector<llvm::Constant*> Elements;
1149 const std::string ClassName = "__ObjC_Protocol_Holder_Ugly_Hack";
1150 const std::string CategoryName = "AnotherHack";
1151 Elements.push_back(MakeConstantString(CategoryName));
1152 Elements.push_back(MakeConstantString(ClassName));
1153 // Instance method list
1154 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1155 ClassName, CategoryName, MethodSels, MethodTypes, false), PtrTy));
1156 // Class method list
1157 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1158 ClassName, CategoryName, MethodSels, MethodTypes, true), PtrTy));
1159 // Protocol list
1160 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrTy,
1161 ExistingProtocols.size());
1162 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
1163 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
1164 LongTy,//FIXME: Should be size_t
1165 ProtocolArrayTy,
1166 NULL);
1167 std::vector<llvm::Constant*> ProtocolElements;
1168 for (llvm::StringMapIterator<llvm::Constant*> iter =
1169 ExistingProtocols.begin(), endIter = ExistingProtocols.end();
1170 iter != endIter ; iter++) {
1171 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(iter->getValue(),
1172 PtrTy);
1173 ProtocolElements.push_back(Ptr);
1174 }
1175 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
1176 ProtocolElements);
1177 ProtocolElements.clear();
1178 ProtocolElements.push_back(NULLPtr);
1179 ProtocolElements.push_back(llvm::ConstantInt::get(LongTy,
1180 ExistingProtocols.size()));
1181 ProtocolElements.push_back(ProtocolArray);
1182 Elements.push_back(llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolListTy,
1183 ProtocolElements, ".objc_protocol_list"), PtrTy));
1184 Categories.push_back(llvm::ConstantExpr::getBitCast(
1185 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
1186 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
1187}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001188
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001189void CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001190 std::string ClassName = OCD->getClassInterface()->getNameAsString();
1191 std::string CategoryName = OCD->getNameAsString();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001192 // Collect information about instance methods
1193 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1194 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001195 for (ObjCCategoryImplDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001196 iter = OCD->instmeth_begin(), endIter = OCD->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001197 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001198 InstanceMethodSels.push_back((*iter)->getSelector());
1199 std::string TypeStr;
1200 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001201 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001202 }
1203
1204 // Collect information about class methods
1205 llvm::SmallVector<Selector, 16> ClassMethodSels;
1206 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001207 for (ObjCCategoryImplDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001208 iter = OCD->classmeth_begin(), endIter = OCD->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001209 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001210 ClassMethodSels.push_back((*iter)->getSelector());
1211 std::string TypeStr;
1212 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001213 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001214 }
1215
1216 // Collect the names of referenced protocols
1217 llvm::SmallVector<std::string, 16> Protocols;
David Chisnallad9e06d2010-03-13 22:20:45 +00001218 const ObjCCategoryDecl *CatDecl = OCD->getCategoryDecl();
1219 const ObjCList<ObjCProtocolDecl> &Protos = CatDecl->getReferencedProtocols();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001220 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1221 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001222 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001223
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001224 std::vector<llvm::Constant*> Elements;
1225 Elements.push_back(MakeConstantString(CategoryName));
1226 Elements.push_back(MakeConstantString(ClassName));
Mike Stump1eb44332009-09-09 15:08:12 +00001227 // Instance method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001228 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001229 ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001230 false), PtrTy));
1231 // Class method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001232 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001233 ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001234 PtrTy));
1235 // Protocol list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001236 Elements.push_back(llvm::ConstantExpr::getBitCast(
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001237 GenerateProtocolList(Protocols), PtrTy));
Owen Anderson3c4972d2009-07-29 18:54:39 +00001238 Categories.push_back(llvm::ConstantExpr::getBitCast(
Mike Stump1eb44332009-09-09 15:08:12 +00001239 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
Owen Anderson47a434f2009-08-05 23:18:46 +00001240 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001241}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001242
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001243llvm::Constant *CGObjCGNU::GeneratePropertyList(const ObjCImplementationDecl *OID,
1244 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
1245 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes) {
1246 ASTContext &Context = CGM.getContext();
1247 //
1248 // Property metadata: name, attributes, isSynthesized, setter name, setter
1249 // types, getter name, getter types.
1250 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1251 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1252 PtrToInt8Ty, NULL);
1253 std::vector<llvm::Constant*> Properties;
1254
1255
1256 // Add all of the property methods need adding to the method list and to the
1257 // property metadata list.
1258 for (ObjCImplDecl::propimpl_iterator
1259 iter = OID->propimpl_begin(), endIter = OID->propimpl_end();
1260 iter != endIter ; iter++) {
1261 std::vector<llvm::Constant*> Fields;
1262 ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
David Chisnall42ba04a2010-02-26 01:11:38 +00001263 ObjCPropertyImplDecl *propertyImpl = *iter;
1264 bool isSynthesized = (propertyImpl->getPropertyImplementation() ==
1265 ObjCPropertyImplDecl::Synthesize);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001266
1267 Fields.push_back(MakeConstantString(property->getNameAsString()));
1268 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1269 property->getPropertyAttributes()));
David Chisnall42ba04a2010-02-26 01:11:38 +00001270 Fields.push_back(llvm::ConstantInt::get(Int8Ty, isSynthesized));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001271 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001272 std::string TypeStr;
1273 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1274 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
David Chisnall42ba04a2010-02-26 01:11:38 +00001275 if (isSynthesized) {
1276 InstanceMethodTypes.push_back(TypeEncoding);
1277 InstanceMethodSels.push_back(getter->getSelector());
1278 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001279 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1280 Fields.push_back(TypeEncoding);
1281 } else {
1282 Fields.push_back(NULLPtr);
1283 Fields.push_back(NULLPtr);
1284 }
1285 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001286 std::string TypeStr;
1287 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1288 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
David Chisnall42ba04a2010-02-26 01:11:38 +00001289 if (isSynthesized) {
1290 InstanceMethodTypes.push_back(TypeEncoding);
1291 InstanceMethodSels.push_back(setter->getSelector());
1292 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001293 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1294 Fields.push_back(TypeEncoding);
1295 } else {
1296 Fields.push_back(NULLPtr);
1297 Fields.push_back(NULLPtr);
1298 }
1299 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1300 }
1301 llvm::ArrayType *PropertyArrayTy =
1302 llvm::ArrayType::get(PropertyMetadataTy, Properties.size());
1303 llvm::Constant *PropertyArray = llvm::ConstantArray::get(PropertyArrayTy,
1304 Properties);
1305 llvm::Constant* PropertyListInitFields[] =
1306 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1307
1308 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001309 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001310 return new llvm::GlobalVariable(TheModule, PropertyListInit->getType(), false,
1311 llvm::GlobalValue::InternalLinkage, PropertyListInit,
1312 ".objc_property_list");
1313}
1314
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001315void CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
1316 ASTContext &Context = CGM.getContext();
1317
1318 // Get the superclass name.
Mike Stump1eb44332009-09-09 15:08:12 +00001319 const ObjCInterfaceDecl * SuperClassDecl =
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001320 OID->getClassInterface()->getSuperClass();
Chris Lattner8ec03f52008-11-24 03:54:41 +00001321 std::string SuperClassName;
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001322 if (SuperClassDecl) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001323 SuperClassName = SuperClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001324 EmitClassRef(SuperClassName);
1325 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001326
1327 // Get the class name
Chris Lattner09dc6662009-04-01 02:00:48 +00001328 ObjCInterfaceDecl *ClassDecl =
1329 const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
Chris Lattner8ec03f52008-11-24 03:54:41 +00001330 std::string ClassName = ClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001331 // Emit the symbol that is used to generate linker errors if this class is
1332 // referenced in other modules but not declared.
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001333 std::string classSymbolName = "__objc_class_name_" + ClassName;
Mike Stump1eb44332009-09-09 15:08:12 +00001334 if (llvm::GlobalVariable *symbol =
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001335 TheModule.getGlobalVariable(classSymbolName)) {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001336 symbol->setInitializer(llvm::ConstantInt::get(LongTy, 0));
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001337 } else {
Owen Anderson1c431b32009-07-08 19:05:04 +00001338 new llvm::GlobalVariable(TheModule, LongTy, false,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001339 llvm::GlobalValue::ExternalLinkage, llvm::ConstantInt::get(LongTy, 0),
Owen Anderson1c431b32009-07-08 19:05:04 +00001340 classSymbolName);
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001341 }
Mike Stump1eb44332009-09-09 15:08:12 +00001342
Daniel Dunbar2bebbf02009-05-03 10:46:44 +00001343 // Get the size of instances.
1344 int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001345
1346 // Collect information about instance variables.
1347 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
1348 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
1349 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
Mike Stump1eb44332009-09-09 15:08:12 +00001350
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001351 std::vector<llvm::Constant*> IvarOffsetValues;
1352
Mike Stump1eb44332009-09-09 15:08:12 +00001353 int superInstanceSize = !SuperClassDecl ? 0 :
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001354 Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
1355 // For non-fragile ivars, set the instance size to 0 - {the size of just this
1356 // class}. The runtime will then set this to the correct value on load.
1357 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
1358 instanceSize = 0 - (instanceSize - superInstanceSize);
1359 }
David Chisnall7f63cb02010-04-19 00:45:34 +00001360
1361 // Collect declared and synthesized ivars.
1362 llvm::SmallVector<ObjCIvarDecl*, 16> OIvars;
1363 CGM.getContext().ShallowCollectObjCIvars(ClassDecl, OIvars);
1364
1365 for (unsigned i = 0, e = OIvars.size(); i != e; ++i) {
1366 ObjCIvarDecl *IVD = OIvars[i];
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001367 // Store the name
David Chisnall7f63cb02010-04-19 00:45:34 +00001368 IvarNames.push_back(MakeConstantString(IVD->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001369 // Get the type encoding for this ivar
1370 std::string TypeStr;
David Chisnall7f63cb02010-04-19 00:45:34 +00001371 Context.getObjCEncodingForType(IVD->getType(), TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001372 IvarTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001373 // Get the offset
David Chisnalld901da52010-04-19 01:37:25 +00001374 uint64_t BaseOffset = ComputeIvarBaseOffset(CGM, OID, IVD);
David Chisnallaecbf242009-11-17 19:32:15 +00001375 uint64_t Offset = BaseOffset;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001376 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001377 Offset = BaseOffset - superInstanceSize;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001378 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001379 IvarOffsets.push_back(
Owen Anderson0032b272009-08-13 21:57:51 +00001380 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001381 IvarOffsetValues.push_back(new llvm::GlobalVariable(TheModule, IntTy,
1382 false, llvm::GlobalValue::ExternalLinkage,
1383 llvm::ConstantInt::get(IntTy, BaseOffset),
1384 "__objc_ivar_offset_value_" + ClassName +"." +
David Chisnall7f63cb02010-04-19 00:45:34 +00001385 IVD->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001386 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001387 llvm::Constant *IvarOffsetArrayInit =
1388 llvm::ConstantArray::get(llvm::ArrayType::get(PtrToIntTy,
1389 IvarOffsetValues.size()), IvarOffsetValues);
1390 llvm::GlobalVariable *IvarOffsetArray = new llvm::GlobalVariable(TheModule,
1391 IvarOffsetArrayInit->getType(), false,
1392 llvm::GlobalValue::InternalLinkage, IvarOffsetArrayInit,
1393 ".ivar.offsets");
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001394
1395 // Collect information about instance methods
1396 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1397 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001398 for (ObjCImplementationDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001399 iter = OID->instmeth_begin(), endIter = OID->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001400 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001401 InstanceMethodSels.push_back((*iter)->getSelector());
1402 std::string TypeStr;
1403 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001404 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001405 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001406
1407 llvm::Constant *Properties = GeneratePropertyList(OID, InstanceMethodSels,
1408 InstanceMethodTypes);
1409
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001410
1411 // Collect information about class methods
1412 llvm::SmallVector<Selector, 16> ClassMethodSels;
1413 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001414 for (ObjCImplementationDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001415 iter = OID->classmeth_begin(), endIter = OID->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001416 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001417 ClassMethodSels.push_back((*iter)->getSelector());
1418 std::string TypeStr;
1419 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001420 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001421 }
1422 // Collect the names of referenced protocols
1423 llvm::SmallVector<std::string, 16> Protocols;
1424 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1425 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1426 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001427 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001428
1429
1430
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001431 // Get the superclass pointer.
1432 llvm::Constant *SuperClass;
Chris Lattner8ec03f52008-11-24 03:54:41 +00001433 if (!SuperClassName.empty()) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001434 SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
1435 } else {
Owen Anderson03e20502009-07-30 23:11:26 +00001436 SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001437 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001438 // Empty vector used to construct empty method lists
1439 llvm::SmallVector<llvm::Constant*, 1> empty;
1440 // Generate the method and instance variable lists
1441 llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001442 InstanceMethodSels, InstanceMethodTypes, false);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001443 llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001444 ClassMethodSels, ClassMethodTypes, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001445 llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
1446 IvarOffsets);
Mike Stump1eb44332009-09-09 15:08:12 +00001447 // Irrespective of whether we are compiling for a fragile or non-fragile ABI,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001448 // we emit a symbol containing the offset for each ivar in the class. This
1449 // allows code compiled for the non-Fragile ABI to inherit from code compiled
1450 // for the legacy ABI, without causing problems. The converse is also
1451 // possible, but causes all ivar accesses to be fragile.
1452 int i = 0;
1453 // Offset pointer for getting at the correct field in the ivar list when
1454 // setting up the alias. These are: The base address for the global, the
1455 // ivar array (second field), the ivar in this list (set for each ivar), and
1456 // the offset (third field in ivar structure)
1457 const llvm::Type *IndexTy = llvm::Type::getInt32Ty(VMContext);
1458 llvm::Constant *offsetPointerIndexes[] = {Zeros[0],
Mike Stump1eb44332009-09-09 15:08:12 +00001459 llvm::ConstantInt::get(IndexTy, 1), 0,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001460 llvm::ConstantInt::get(IndexTy, 2) };
1461
1462 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1463 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1464 const std::string Name = "__objc_ivar_offset_" + ClassName + '.'
1465 +(*iter)->getNameAsString();
1466 offsetPointerIndexes[2] = llvm::ConstantInt::get(IndexTy, i++);
1467 // Get the correct ivar field
1468 llvm::Constant *offsetValue = llvm::ConstantExpr::getGetElementPtr(
1469 IvarList, offsetPointerIndexes, 4);
1470 // Get the existing alias, if one exists.
1471 llvm::GlobalVariable *offset = TheModule.getNamedGlobal(Name);
1472 if (offset) {
1473 offset->setInitializer(offsetValue);
1474 // If this is the real definition, change its linkage type so that
1475 // different modules will use this one, rather than their private
1476 // copy.
1477 offset->setLinkage(llvm::GlobalValue::ExternalLinkage);
1478 } else {
1479 // Add a new alias if there isn't one already.
1480 offset = new llvm::GlobalVariable(TheModule, offsetValue->getType(),
1481 false, llvm::GlobalValue::ExternalLinkage, offsetValue, Name);
1482 }
1483 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001484 //Generate metaclass for class methods
1485 llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
David Chisnall18044632009-11-16 19:05:54 +00001486 NULLPtr, 0x12L, ClassName.c_str(), 0, Zeros[0], GenerateIvarList(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001487 empty, empty, empty), ClassMethodList, NULLPtr, NULLPtr, NULLPtr);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001488
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001489 // Generate the class structure
Chris Lattner8ec03f52008-11-24 03:54:41 +00001490 llvm::Constant *ClassStruct =
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001491 GenerateClassStructure(MetaClassStruct, SuperClass, 0x11L,
Chris Lattner8ec03f52008-11-24 03:54:41 +00001492 ClassName.c_str(), 0,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001493 llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001494 MethodList, GenerateProtocolList(Protocols), IvarOffsetArray,
1495 Properties);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001496
1497 // Resolve the class aliases, if they exist.
1498 if (ClassPtrAlias) {
1499 ClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001500 llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001501 ClassPtrAlias = 0;
1502 }
1503 if (MetaClassPtrAlias) {
1504 MetaClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001505 llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001506 MetaClassPtrAlias = 0;
1507 }
1508
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001509 // Add class structure to list to be added to the symtab later
Owen Anderson3c4972d2009-07-29 18:54:39 +00001510 ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001511 Classes.push_back(ClassStruct);
1512}
1513
Fariborz Jahanianc38e9af2009-06-23 21:47:46 +00001514
Mike Stump1eb44332009-09-09 15:08:12 +00001515llvm::Function *CGObjCGNU::ModuleInitFunction() {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001516 // Only emit an ObjC load function if no Objective-C stuff has been called
1517 if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
1518 ExistingProtocols.empty() && TypedSelectors.empty() &&
Anton Korobeynikov5f58b912008-06-01 15:14:46 +00001519 UntypedSelectors.empty())
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001520 return NULL;
Eli Friedman1b8956e2008-06-01 16:00:02 +00001521
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001522 // Add all referenced protocols to a category.
1523 GenerateProtocolHolderCategory();
1524
Chris Lattnere160c9b2009-01-27 05:06:01 +00001525 const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1526 SelectorTy->getElementType());
1527 const llvm::Type *SelStructPtrTy = SelectorTy;
1528 bool isSelOpaque = false;
1529 if (SelStructTy == 0) {
Owen Anderson47a434f2009-08-05 23:18:46 +00001530 SelStructTy = llvm::StructType::get(VMContext, PtrToInt8Ty,
1531 PtrToInt8Ty, NULL);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001532 SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
Chris Lattnere160c9b2009-01-27 05:06:01 +00001533 isSelOpaque = true;
1534 }
1535
Eli Friedman1b8956e2008-06-01 16:00:02 +00001536 // Name the ObjC types to make the IR a bit easier to read
Chris Lattnere160c9b2009-01-27 05:06:01 +00001537 TheModule.addTypeName(".objc_selector", SelStructPtrTy);
Eli Friedman1b8956e2008-06-01 16:00:02 +00001538 TheModule.addTypeName(".objc_id", IdTy);
1539 TheModule.addTypeName(".objc_imp", IMPTy);
1540
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001541 std::vector<llvm::Constant*> Elements;
Chris Lattner71238f62009-04-25 23:19:45 +00001542 llvm::Constant *Statics = NULLPtr;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001543 // Generate statics list:
Chris Lattner71238f62009-04-25 23:19:45 +00001544 if (ConstantStrings.size()) {
Owen Anderson96e0fc72009-07-29 22:16:19 +00001545 llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Chris Lattner71238f62009-04-25 23:19:45 +00001546 ConstantStrings.size() + 1);
1547 ConstantStrings.push_back(NULLPtr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001548
Daniel Dunbar1b096952009-11-29 02:38:47 +00001549 llvm::StringRef StringClass = CGM.getLangOptions().ObjCConstantStringClass;
1550 if (StringClass.empty()) StringClass = "NXConstantString";
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001551 Elements.push_back(MakeConstantString(StringClass,
1552 ".objc_static_class_name"));
Owen Anderson7db6d832009-07-28 18:33:04 +00001553 Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
Chris Lattner71238f62009-04-25 23:19:45 +00001554 ConstantStrings));
Mike Stump1eb44332009-09-09 15:08:12 +00001555 llvm::StructType *StaticsListTy =
Owen Anderson47a434f2009-08-05 23:18:46 +00001556 llvm::StructType::get(VMContext, PtrToInt8Ty, StaticsArrayTy, NULL);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001557 llvm::Type *StaticsListPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001558 llvm::PointerType::getUnqual(StaticsListTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001559 Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
Mike Stump1eb44332009-09-09 15:08:12 +00001560 llvm::ArrayType *StaticsListArrayTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001561 llvm::ArrayType::get(StaticsListPtrTy, 2);
Chris Lattner71238f62009-04-25 23:19:45 +00001562 Elements.clear();
1563 Elements.push_back(Statics);
Owen Andersonc9c88b42009-07-31 20:28:54 +00001564 Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
Chris Lattner71238f62009-04-25 23:19:45 +00001565 Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001566 Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001567 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001568 // Array of classes, categories, and constant objects
Owen Anderson96e0fc72009-07-29 22:16:19 +00001569 llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001570 Classes.size() + Categories.size() + 2);
Mike Stump1eb44332009-09-09 15:08:12 +00001571 llvm::StructType *SymTabTy = llvm::StructType::get(VMContext,
Owen Anderson47a434f2009-08-05 23:18:46 +00001572 LongTy, SelStructPtrTy,
Owen Anderson0032b272009-08-13 21:57:51 +00001573 llvm::Type::getInt16Ty(VMContext),
1574 llvm::Type::getInt16Ty(VMContext),
Chris Lattner630404b2008-06-26 04:10:42 +00001575 ClassListTy, NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001576
1577 Elements.clear();
1578 // Pointer to an array of selectors used in this module.
1579 std::vector<llvm::Constant*> Selectors;
1580 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1581 iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
1582 iter != iterEnd ; ++iter) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001583 Elements.push_back(ExportUniqueString(iter->first.first, ".objc_sel_name"));
David Chisnalla7c5b082009-09-14 19:04:10 +00001584 Elements.push_back(MakeConstantString(iter->first.second,
Chris Lattner630404b2008-06-26 04:10:42 +00001585 ".objc_sel_types"));
Owen Anderson08e25242009-07-27 22:29:56 +00001586 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001587 Elements.clear();
1588 }
1589 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1590 iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
Chris Lattner630404b2008-06-26 04:10:42 +00001591 iter != iterEnd; ++iter) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001592 Elements.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001593 ExportUniqueString(iter->getKeyData(), ".objc_sel_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001594 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001595 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001596 Elements.clear();
1597 }
1598 Elements.push_back(NULLPtr);
1599 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001600 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001601 Elements.clear();
1602 // Number of static selectors
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001603 Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001604 llvm::Constant *SelectorList = MakeGlobal(
Owen Anderson96e0fc72009-07-29 22:16:19 +00001605 llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001606 ".objc_selector_list");
Mike Stump1eb44332009-09-09 15:08:12 +00001607 Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
Chris Lattnere160c9b2009-01-27 05:06:01 +00001608 SelStructPtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001609
1610 // Now that all of the static selectors exist, create pointers to them.
1611 int index = 0;
1612 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1613 iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
1614 iter != iterEnd; ++iter) {
1615 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001616 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001617 llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
1618 true, llvm::GlobalValue::InternalLinkage,
1619 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1620 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001621 // If selectors are defined as an opaque type, cast the pointer to this
1622 // type.
1623 if (isSelOpaque) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001624 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1625 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001626 }
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001627 (*iter).second->setAliasee(SelPtr);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001628 }
1629 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1630 iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1631 iter != iterEnd; iter++) {
1632 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001633 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001634 llvm::Constant *SelPtr = new llvm::GlobalVariable
1635 (TheModule, SelStructPtrTy,
1636 true, llvm::GlobalValue::InternalLinkage,
1637 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1638 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001639 // If selectors are defined as an opaque type, cast the pointer to this
1640 // type.
1641 if (isSelOpaque) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001642 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1643 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001644 }
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001645 (*iter).second->setAliasee(SelPtr);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001646 }
1647 // Number of classes defined.
Mike Stump1eb44332009-09-09 15:08:12 +00001648 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001649 Classes.size()));
1650 // Number of categories defined
Mike Stump1eb44332009-09-09 15:08:12 +00001651 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001652 Categories.size()));
1653 // Create an array of classes, then categories, then static object instances
1654 Classes.insert(Classes.end(), Categories.begin(), Categories.end());
1655 // NULL-terminated list of static object instances (mainly constant strings)
1656 Classes.push_back(Statics);
1657 Classes.push_back(NULLPtr);
Owen Anderson7db6d832009-07-28 18:33:04 +00001658 llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001659 Elements.push_back(ClassList);
Mike Stump1eb44332009-09-09 15:08:12 +00001660 // Construct the symbol table
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001661 llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
1662
1663 // The symbol table is contained in a module which has some version-checking
1664 // constants
Owen Anderson47a434f2009-08-05 23:18:46 +00001665 llvm::StructType * ModuleTy = llvm::StructType::get(VMContext, LongTy, LongTy,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001666 PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001667 Elements.clear();
1668 // Runtime version used for compatibility checking.
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001669 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Mike Stump1eb44332009-09-09 15:08:12 +00001670 Elements.push_back(llvm::ConstantInt::get(LongTy,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001671 NonFragileRuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001672 } else {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001673 Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001674 }
Fariborz Jahanian91a0b512009-04-01 19:49:42 +00001675 // sizeof(ModuleTy)
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +00001676 llvm::TargetData td(&TheModule);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001677 Elements.push_back(llvm::ConstantInt::get(LongTy,
Owen Andersona1cf15f2009-07-14 23:10:40 +00001678 td.getTypeSizeInBits(ModuleTy)/8));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001679 //FIXME: Should be the path to the file where this module was declared
1680 Elements.push_back(NULLPtr);
1681 Elements.push_back(SymTab);
1682 llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
1683
1684 // Create the load function calling the runtime entry point with the module
1685 // structure
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001686 llvm::Function * LoadFunction = llvm::Function::Create(
Owen Anderson0032b272009-08-13 21:57:51 +00001687 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), false),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001688 llvm::GlobalValue::InternalLinkage, ".objc_load_function",
1689 &TheModule);
Owen Anderson0032b272009-08-13 21:57:51 +00001690 llvm::BasicBlock *EntryBB =
1691 llvm::BasicBlock::Create(VMContext, "entry", LoadFunction);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001692 CGBuilderTy Builder(VMContext);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001693 Builder.SetInsertPoint(EntryBB);
Fariborz Jahanian26c82942009-03-30 18:02:14 +00001694
1695 std::vector<const llvm::Type*> Params(1,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001696 llvm::PointerType::getUnqual(ModuleTy));
1697 llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Owen Anderson0032b272009-08-13 21:57:51 +00001698 llvm::Type::getVoidTy(VMContext), Params, true), "__objc_exec_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001699 Builder.CreateCall(Register, Module);
1700 Builder.CreateRetVoid();
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001701
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001702 return LoadFunction;
1703}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001704
Fariborz Jahanian679a5022009-01-10 21:06:09 +00001705llvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
Mike Stump1eb44332009-09-09 15:08:12 +00001706 const ObjCContainerDecl *CD) {
1707 const ObjCCategoryImplDecl *OCD =
Steve Naroff3e0a5402009-01-08 19:41:02 +00001708 dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
Chris Lattner077bf5e2008-11-24 03:33:13 +00001709 std::string CategoryName = OCD ? OCD->getNameAsString() : "";
David Chisnall4c8c8e92010-03-20 19:53:29 +00001710 std::string ClassName = CD->getName();
Chris Lattner077bf5e2008-11-24 03:33:13 +00001711 std::string MethodName = OMD->getSelector().getAsString();
Douglas Gregorf8d49f62009-01-09 17:18:27 +00001712 bool isClassMethod = !OMD->isInstanceMethod();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001713
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001714 CodeGenTypes &Types = CGM.getTypes();
Mike Stump1eb44332009-09-09 15:08:12 +00001715 const llvm::FunctionType *MethodTy =
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001716 Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001717 std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
1718 MethodName, isClassMethod);
1719
Daniel Dunbard6c93d72009-09-17 04:01:22 +00001720 llvm::Function *Method
Mike Stump1eb44332009-09-09 15:08:12 +00001721 = llvm::Function::Create(MethodTy,
1722 llvm::GlobalValue::InternalLinkage,
1723 FunctionName,
1724 &TheModule);
Chris Lattner391d77a2008-03-30 23:03:07 +00001725 return Method;
1726}
1727
Daniel Dunbar49f66022008-09-24 03:38:44 +00001728llvm::Function *CGObjCGNU::GetPropertyGetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001729 std::vector<const llvm::Type*> Params;
1730 const llvm::Type *BoolTy =
1731 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1732 Params.push_back(IdTy);
1733 Params.push_back(SelectorTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001734 Params.push_back(IntTy);
Mike Stump1eb44332009-09-09 15:08:12 +00001735 Params.push_back(BoolTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001736 // void objc_getProperty (id, SEL, int, bool)
Mike Stump1eb44332009-09-09 15:08:12 +00001737 const llvm::FunctionType *FTy =
1738 llvm::FunctionType::get(IdTy, Params, false);
1739 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1740 "objc_getProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001741}
1742
1743llvm::Function *CGObjCGNU::GetPropertySetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001744 std::vector<const llvm::Type*> Params;
1745 const llvm::Type *BoolTy =
1746 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1747 Params.push_back(IdTy);
1748 Params.push_back(SelectorTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001749 Params.push_back(IntTy);
Mike Stump1eb44332009-09-09 15:08:12 +00001750 Params.push_back(IdTy);
1751 Params.push_back(BoolTy);
1752 Params.push_back(BoolTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001753 // void objc_setProperty (id, SEL, int, id, bool, bool)
Mike Stump1eb44332009-09-09 15:08:12 +00001754 const llvm::FunctionType *FTy =
1755 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Params, false);
1756 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1757 "objc_setProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001758}
1759
Fariborz Jahanian6cc59062010-04-12 18:18:10 +00001760// FIXME. Implement this.
1761llvm::Function *CGObjCGNU::GetCopyStructFunction() {
1762 return 0;
1763}
1764
Daniel Dunbar309a4362009-07-24 07:40:24 +00001765llvm::Constant *CGObjCGNU::EnumerationMutationFunction() {
1766 CodeGen::CodeGenTypes &Types = CGM.getTypes();
1767 ASTContext &Ctx = CGM.getContext();
1768 // void objc_enumerationMutation (id)
John McCallead608a2010-02-26 00:48:12 +00001769 llvm::SmallVector<CanQualType,1> Params;
David Chisnall0f436562009-08-17 16:35:33 +00001770 Params.push_back(ASTIdTy);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001771 const llvm::FunctionType *FTy =
John McCall04a67a62010-02-05 21:31:56 +00001772 Types.GetFunctionType(Types.getFunctionInfo(Ctx.VoidTy, Params,
Rafael Espindola264ba482010-03-30 20:24:48 +00001773 FunctionType::ExtInfo()), false);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001774 return CGM.CreateRuntimeFunction(FTy, "objc_enumerationMutation");
Anders Carlsson2abd89c2008-08-31 04:05:03 +00001775}
1776
Fariborz Jahanianbd71be42008-11-21 00:49:24 +00001777void CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1778 const Stmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001779 // Pointer to the personality function
1780 llvm::Constant *Personality =
Owen Anderson0032b272009-08-13 21:57:51 +00001781 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getInt32Ty(VMContext),
Chris Lattnerb59761b2009-07-01 04:13:52 +00001782 true),
Chris Lattner5dc08672009-05-08 00:11:50 +00001783 "__gnu_objc_personality_v0");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001784 Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
Chris Lattner5dc08672009-05-08 00:11:50 +00001785 std::vector<const llvm::Type*> Params;
1786 Params.push_back(PtrTy);
1787 llvm::Value *RethrowFn =
Owen Anderson0032b272009-08-13 21:57:51 +00001788 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
David Chisnall90665bd2010-01-06 18:02:59 +00001789 Params, false), "_Unwind_Resume");
Chris Lattner5dc08672009-05-08 00:11:50 +00001790
1791 bool isTry = isa<ObjCAtTryStmt>(S);
1792 llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
1793 llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
1794 llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
Chris Lattnerd6a99072009-05-11 18:16:28 +00001795 llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
Chris Lattner5dc08672009-05-08 00:11:50 +00001796 llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
1797 llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
1798 llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
1799
David Chisnall0faa5162009-12-24 02:26:34 +00001800 // @synchronized()
Chris Lattner5dc08672009-05-08 00:11:50 +00001801 if (!isTry) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001802 std::vector<const llvm::Type*> Args(1, IdTy);
1803 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001804 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001805 llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
Mike Stump1eb44332009-09-09 15:08:12 +00001806 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001807 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1808 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1809 CGF.Builder.CreateCall(SyncEnter, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001810 }
1811
Chris Lattnerd6a99072009-05-11 18:16:28 +00001812
Chris Lattner5dc08672009-05-08 00:11:50 +00001813 // Push an EH context entry, used for handling rethrows and jumps
1814 // through finally.
1815 CGF.PushCleanupBlock(FinallyBlock);
1816
1817 // Emit the statements in the @try {} block
1818 CGF.setInvokeDest(TryHandler);
1819
1820 CGF.EmitBlock(TryBlock);
Mike Stump1eb44332009-09-09 15:08:12 +00001821 CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
Chris Lattner5dc08672009-05-08 00:11:50 +00001822 : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
1823
1824 // Jump to @finally if there is no exception
1825 CGF.EmitBranchThroughCleanup(FinallyEnd);
1826
1827 // Emit the handlers
1828 CGF.EmitBlock(TryHandler);
1829
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001830 // Get the correct versions of the exception handling intrinsics
Mike Stump1eb44332009-09-09 15:08:12 +00001831 llvm::Value *llvm_eh_exception =
Chris Lattner5dc08672009-05-08 00:11:50 +00001832 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
Duncan Sands790b20e2009-10-14 16:13:30 +00001833 llvm::Value *llvm_eh_selector =
1834 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector);
1835 llvm::Value *llvm_eh_typeid_for =
1836 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for);
Chris Lattner5dc08672009-05-08 00:11:50 +00001837
1838 // Exception object
1839 llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1840 llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
1841
1842 llvm::SmallVector<llvm::Value*, 8> ESelArgs;
Douglas Gregorc00d8e12010-04-26 16:46:50 +00001843 llvm::SmallVector<std::pair<const VarDecl*, const Stmt*>, 8> Handlers;
Chris Lattner5dc08672009-05-08 00:11:50 +00001844
1845 ESelArgs.push_back(Exc);
1846 ESelArgs.push_back(Personality);
1847
1848 bool HasCatchAll = false;
1849 // Only @try blocks are allowed @catch blocks, but both can have @finally
1850 if (isTry) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +00001851 if (cast<ObjCAtTryStmt>(S).getNumCatchStmts()) {
1852 const ObjCAtTryStmt &AtTry = cast<ObjCAtTryStmt>(S);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001853 CGF.setInvokeDest(CatchInCatch);
Chris Lattner5dc08672009-05-08 00:11:50 +00001854
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +00001855 for (unsigned I = 0, N = AtTry.getNumCatchStmts(); I != N; ++I) {
1856 const ObjCAtCatchStmt *CatchStmt = AtTry.getCatchStmt(I);
Douglas Gregorc00d8e12010-04-26 16:46:50 +00001857 const VarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
Mike Stumpbb1c8602009-07-31 21:31:32 +00001858 Handlers.push_back(std::make_pair(CatchDecl,
1859 CatchStmt->getCatchBody()));
Chris Lattner5dc08672009-05-08 00:11:50 +00001860
1861 // @catch() and @catch(id) both catch any ObjC exception
Steve Naroff14108da2009-07-10 23:34:53 +00001862 if (!CatchDecl || CatchDecl->getType()->isObjCIdType()
Chris Lattner5dc08672009-05-08 00:11:50 +00001863 || CatchDecl->getType()->isObjCQualifiedIdType()) {
1864 // Use i8* null here to signal this is a catch all, not a cleanup.
1865 ESelArgs.push_back(NULLPtr);
1866 HasCatchAll = true;
1867 // No further catches after this one will ever by reached
1868 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001869 }
Chris Lattner5dc08672009-05-08 00:11:50 +00001870
1871 // All other types should be Objective-C interface pointer types.
Mike Stump1eb44332009-09-09 15:08:12 +00001872 const ObjCObjectPointerType *OPT =
John McCall183700f2009-09-21 23:43:11 +00001873 CatchDecl->getType()->getAs<ObjCObjectPointerType>();
Steve Naroff14108da2009-07-10 23:34:53 +00001874 assert(OPT && "Invalid @catch type.");
Mike Stump1eb44332009-09-09 15:08:12 +00001875 const ObjCInterfaceType *IT =
John McCall183700f2009-09-21 23:43:11 +00001876 OPT->getPointeeType()->getAs<ObjCInterfaceType>();
Chris Lattner5dc08672009-05-08 00:11:50 +00001877 assert(IT && "Invalid @catch type.");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001878 llvm::Value *EHType =
1879 MakeConstantString(IT->getDecl()->getNameAsString());
Chris Lattner5dc08672009-05-08 00:11:50 +00001880 ESelArgs.push_back(EHType);
1881 }
1882 }
1883 }
1884
1885 // We use a cleanup unless there was already a catch all.
1886 if (!HasCatchAll) {
Owen Anderson0032b272009-08-13 21:57:51 +00001887 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattner5dc08672009-05-08 00:11:50 +00001888 Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
1889 }
1890
1891 // Find which handler was matched.
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001892 llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
Chris Lattner5dc08672009-05-08 00:11:50 +00001893 ESelArgs.begin(), ESelArgs.end(), "selector");
1894
1895 for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
Douglas Gregorc00d8e12010-04-26 16:46:50 +00001896 const VarDecl *CatchParam = Handlers[i].first;
Chris Lattner5dc08672009-05-08 00:11:50 +00001897 const Stmt *CatchBody = Handlers[i].second;
1898
1899 llvm::BasicBlock *Next = 0;
1900
1901 // The last handler always matches.
1902 if (i + 1 != e) {
1903 assert(CatchParam && "Only last handler can be a catch all.");
1904
1905 // Test whether this block matches the type for the selector and branch
1906 // to Match if it does, or to the next BB if it doesn't.
1907 llvm::BasicBlock *Match = CGF.createBasicBlock("match");
1908 Next = CGF.createBasicBlock("catch.next");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001909 llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
Chris Lattner5dc08672009-05-08 00:11:50 +00001910 CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
1911 CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
1912 Next);
1913
1914 CGF.EmitBlock(Match);
1915 }
Mike Stump1eb44332009-09-09 15:08:12 +00001916
Chris Lattner5dc08672009-05-08 00:11:50 +00001917 if (CatchBody) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001918 llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
1919 CGF.ConvertType(CatchParam->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001920
Chris Lattner5dc08672009-05-08 00:11:50 +00001921 // Bind the catch parameter if it exists.
1922 if (CatchParam) {
1923 // CatchParam is a ParmVarDecl because of the grammar
1924 // construction used to handle this, but for codegen purposes
1925 // we treat this as a local decl.
1926 CGF.EmitLocalBlockVarDecl(*CatchParam);
1927 CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
1928 }
1929
1930 CGF.ObjCEHValueStack.push_back(ExcObject);
1931 CGF.EmitStmt(CatchBody);
1932 CGF.ObjCEHValueStack.pop_back();
1933
1934 CGF.EmitBranchThroughCleanup(FinallyEnd);
1935
1936 if (Next)
1937 CGF.EmitBlock(Next);
1938 } else {
1939 assert(!Next && "catchup should be last handler.");
1940
1941 CGF.Builder.CreateStore(Exc, RethrowPtr);
1942 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1943 }
1944 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001945 // The @finally block is a secondary landing pad for any exceptions thrown in
1946 // @catch() blocks
1947 CGF.EmitBlock(CatchInCatch);
1948 Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1949 ESelArgs.clear();
1950 ESelArgs.push_back(Exc);
1951 ESelArgs.push_back(Personality);
David Chisnall0faa5162009-12-24 02:26:34 +00001952 // If there is a @catch or @finally clause in outside of this one then we
1953 // need to make sure that we catch and rethrow it.
1954 if (PrevLandingPad) {
1955 ESelArgs.push_back(NULLPtr);
1956 } else {
1957 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
1958 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001959 CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
1960 "selector");
1961 CGF.Builder.CreateCall(llvm_eh_typeid_for,
1962 CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
1963 CGF.Builder.CreateStore(Exc, RethrowPtr);
1964 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1965
Chris Lattner5dc08672009-05-08 00:11:50 +00001966 CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
1967
1968 CGF.setInvokeDest(PrevLandingPad);
1969
1970 CGF.EmitBlock(FinallyBlock);
1971
Chris Lattnerd6a99072009-05-11 18:16:28 +00001972
Chris Lattner5dc08672009-05-08 00:11:50 +00001973 if (isTry) {
Mike Stump1eb44332009-09-09 15:08:12 +00001974 if (const ObjCAtFinallyStmt* FinallyStmt =
Chris Lattner5dc08672009-05-08 00:11:50 +00001975 cast<ObjCAtTryStmt>(S).getFinallyStmt())
1976 CGF.EmitStmt(FinallyStmt->getFinallyBody());
1977 } else {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001978 // Emit 'objc_sync_exit(expr)' as finally's sole statement for
Chris Lattner5dc08672009-05-08 00:11:50 +00001979 // @synchronized.
Chris Lattnerd6a99072009-05-11 18:16:28 +00001980 std::vector<const llvm::Type*> Args(1, IdTy);
1981 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001982 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001983 llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
Mike Stump1eb44332009-09-09 15:08:12 +00001984 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001985 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1986 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1987 CGF.Builder.CreateCall(SyncExit, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001988 }
1989
1990 if (Info.SwitchBlock)
1991 CGF.EmitBlock(Info.SwitchBlock);
1992 if (Info.EndBlock)
1993 CGF.EmitBlock(Info.EndBlock);
1994
1995 // Branch around the rethrow code.
1996 CGF.EmitBranch(FinallyEnd);
1997
1998 CGF.EmitBlock(FinallyRethrow);
David Chisnall0faa5162009-12-24 02:26:34 +00001999
2000 llvm::Value *ExceptionObject = CGF.Builder.CreateLoad(RethrowPtr);
2001 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
2002 if (!UnwindBB) {
2003 CGF.Builder.CreateCall(RethrowFn, ExceptionObject);
2004 // Exception always thrown, next instruction is never reached.
2005 CGF.Builder.CreateUnreachable();
2006 } else {
2007 // If there is a @catch block outside this scope, we invoke instead of
2008 // calling because we may return to this function. This is very slow, but
2009 // some people still do it. It would be nice to add an optimised path for
2010 // this.
2011 CGF.Builder.CreateInvoke(RethrowFn, UnwindBB, UnwindBB, &ExceptionObject,
2012 &ExceptionObject+1);
2013 }
Mike Stump1eb44332009-09-09 15:08:12 +00002014
Chris Lattner5dc08672009-05-08 00:11:50 +00002015 CGF.EmitBlock(FinallyEnd);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00002016}
2017
2018void CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar49f66022008-09-24 03:38:44 +00002019 const ObjCAtThrowStmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00002020 llvm::Value *ExceptionAsObject;
2021
2022 std::vector<const llvm::Type*> Args(1, IdTy);
2023 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00002024 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Mike Stump1eb44332009-09-09 15:08:12 +00002025 llvm::Value *ThrowFn =
Chris Lattner5dc08672009-05-08 00:11:50 +00002026 CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
Mike Stump1eb44332009-09-09 15:08:12 +00002027
Chris Lattner5dc08672009-05-08 00:11:50 +00002028 if (const Expr *ThrowExpr = S.getThrowExpr()) {
2029 llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00002030 ExceptionAsObject = Exception;
Chris Lattner5dc08672009-05-08 00:11:50 +00002031 } else {
Mike Stump1eb44332009-09-09 15:08:12 +00002032 assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
Chris Lattner5dc08672009-05-08 00:11:50 +00002033 "Unexpected rethrow outside @catch block.");
2034 ExceptionAsObject = CGF.ObjCEHValueStack.back();
2035 }
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00002036 ExceptionAsObject =
2037 CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
Mike Stump1eb44332009-09-09 15:08:12 +00002038
Chris Lattner5dc08672009-05-08 00:11:50 +00002039 // Note: This may have to be an invoke, if we want to support constructs like:
2040 // @try {
2041 // @throw(obj);
2042 // }
2043 // @catch(id) ...
2044 //
2045 // This is effectively turning @throw into an incredibly-expensive goto, but
2046 // it may happen as a result of inlining followed by missed optimizations, or
2047 // as a result of stupidity.
2048 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
2049 if (!UnwindBB) {
2050 CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
2051 CGF.Builder.CreateUnreachable();
2052 } else {
2053 CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
2054 &ExceptionAsObject+1);
2055 }
2056 // Clear the insertion point to indicate we are in unreachable code.
2057 CGF.Builder.ClearInsertionPoint();
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00002058}
2059
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00002060llvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002061 llvm::Value *AddrWeakObj) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002062 CGBuilderTy B = CGF.Builder;
2063 AddrWeakObj = EnforceType(B, AddrWeakObj, IdTy);
2064 return B.CreateCall(WeakReadFn, AddrWeakObj);
Fariborz Jahanian6dc23172008-11-18 21:45:40 +00002065}
2066
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00002067void CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002068 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002069 CGBuilderTy B = CGF.Builder;
2070 src = EnforceType(B, src, IdTy);
2071 dst = EnforceType(B, dst, PtrToIdTy);
2072 B.CreateCall2(WeakAssignFn, src, dst);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00002073}
2074
Fariborz Jahanian58626502008-11-19 00:59:10 +00002075void CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002076 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002077 CGBuilderTy B = CGF.Builder;
2078 src = EnforceType(B, src, IdTy);
2079 dst = EnforceType(B, dst, PtrToIdTy);
2080 B.CreateCall2(GlobalAssignFn, src, dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +00002081}
2082
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00002083void CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +00002084 llvm::Value *src, llvm::Value *dst,
2085 llvm::Value *ivarOffset) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002086 CGBuilderTy B = CGF.Builder;
2087 src = EnforceType(B, src, IdTy);
2088 dst = EnforceType(B, dst, PtrToIdTy);
2089 B.CreateCall3(IvarAssignFn, src, dst, ivarOffset);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00002090}
2091
Fariborz Jahanian58626502008-11-19 00:59:10 +00002092void CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002093 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002094 CGBuilderTy B = CGF.Builder;
2095 src = EnforceType(B, src, IdTy);
2096 dst = EnforceType(B, dst, PtrToIdTy);
2097 B.CreateCall2(StrongCastAssignFn, src, dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +00002098}
2099
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002100void CGObjCGNU::EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002101 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002102 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +00002103 QualType Ty) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002104 CGBuilderTy B = CGF.Builder;
2105 DestPtr = EnforceType(B, DestPtr, IdTy);
2106 SrcPtr = EnforceType(B, SrcPtr, PtrToIdTy);
2107
2108 std::pair<uint64_t, unsigned> TypeInfo = CGM.getContext().getTypeInfo(Ty);
2109 unsigned long size = TypeInfo.first/8;
2110 // FIXME: size_t
2111 llvm::Value *N = llvm::ConstantInt::get(LongTy, size);
2112
2113 B.CreateCall3(MemMoveFn, DestPtr, SrcPtr, N);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002114}
2115
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002116llvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
2117 const ObjCInterfaceDecl *ID,
2118 const ObjCIvarDecl *Ivar) {
2119 const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
2120 + '.' + Ivar->getNameAsString();
2121 // Emit the variable and initialize it with what we think the correct value
2122 // is. This allows code compiled with non-fragile ivars to work correctly
2123 // when linked against code which isn't (most of the time).
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002124 llvm::GlobalVariable *IvarOffsetPointer = TheModule.getNamedGlobal(Name);
2125 if (!IvarOffsetPointer) {
David Chisnalld901da52010-04-19 01:37:25 +00002126 uint64_t Offset;
2127 if (ObjCImplementationDecl *OID =
Dan Gohmancb421fa2010-04-19 16:39:44 +00002128 CGM.getContext().getObjCImplementation(
2129 const_cast<ObjCInterfaceDecl *>(ID)))
David Chisnalld901da52010-04-19 01:37:25 +00002130 Offset = ComputeIvarBaseOffset(CGM, OID, Ivar);
2131 else
2132 Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
2133
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002134 llvm::ConstantInt *OffsetGuess =
David Chisnallf9508372010-01-11 19:02:35 +00002135 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset, "ivar");
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002136 // Don't emit the guess in non-PIC code because the linker will not be able
2137 // to replace it with the real version for a library. In non-PIC code you
2138 // must compile with the fragile ABI if you want to use ivars from a
Mike Stump1eb44332009-09-09 15:08:12 +00002139 // GCC-compiled class.
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002140 if (CGM.getLangOptions().PICLevel) {
2141 llvm::GlobalVariable *IvarOffsetGV = new llvm::GlobalVariable(TheModule,
2142 llvm::Type::getInt32Ty(VMContext), false,
2143 llvm::GlobalValue::PrivateLinkage, OffsetGuess, Name+".guess");
2144 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
2145 IvarOffsetGV->getType(), false, llvm::GlobalValue::LinkOnceAnyLinkage,
2146 IvarOffsetGV, Name);
2147 } else {
2148 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
Benjamin Kramer3c0ef8c2009-10-13 10:07:13 +00002149 llvm::Type::getInt32PtrTy(VMContext), false,
2150 llvm::GlobalValue::ExternalLinkage, 0, Name);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002151 }
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002152 }
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002153 return IvarOffsetPointer;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002154}
2155
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00002156LValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
2157 QualType ObjectTy,
2158 llvm::Value *BaseValue,
2159 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00002160 unsigned CVRQualifiers) {
John McCall183700f2009-09-21 23:43:11 +00002161 const ObjCInterfaceDecl *ID = ObjectTy->getAs<ObjCInterfaceType>()->getDecl();
Daniel Dunbar97776872009-04-22 07:32:20 +00002162 return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
2163 EmitIvarOffset(CGF, ID, Ivar));
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00002164}
Mike Stumpbb1c8602009-07-31 21:31:32 +00002165
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002166static const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
2167 const ObjCInterfaceDecl *OID,
2168 const ObjCIvarDecl *OIVD) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002169 llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
Fariborz Jahanian8e6ac1d2009-06-04 01:19:09 +00002170 Context.ShallowCollectObjCIvars(OID, Ivars);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002171 for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
2172 if (OIVD == Ivars[k])
2173 return OID;
2174 }
Mike Stump1eb44332009-09-09 15:08:12 +00002175
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002176 // Otherwise check in the super class.
2177 if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
2178 return FindIvarInterface(Context, Super, OIVD);
Mike Stump1eb44332009-09-09 15:08:12 +00002179
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002180 return 0;
2181}
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00002182
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002183llvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +00002184 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002185 const ObjCIvarDecl *Ivar) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002186 if (CGM.getLangOptions().ObjCNonFragileABI) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002187 Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002188 return CGF.Builder.CreateLoad(CGF.Builder.CreateLoad(
2189 ObjCIvarOffsetVariable(Interface, Ivar), false, "ivar"));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002190 }
Daniel Dunbar97776872009-04-22 07:32:20 +00002191 uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00002192 return llvm::ConstantInt::get(LongTy, Offset, "ivar");
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002193}
2194
Mike Stumpbb1c8602009-07-31 21:31:32 +00002195CodeGen::CGObjCRuntime *
2196CodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM) {
Chris Lattnerdce14062008-06-26 04:19:03 +00002197 return new CGObjCGNU(CGM);
Chris Lattner0f984262008-03-01 08:50:34 +00002198}