blob: 0835e52b527221560aa3c316c5611629d6325aaf [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"
David Chisnallc6cd5fd2010-04-28 19:33:36 +000029#include "llvm/LLVMContext.h"
Chris Lattner0f984262008-03-01 08:50:34 +000030#include "llvm/ADT/SmallVector.h"
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000031#include "llvm/ADT/StringMap.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000032#include "llvm/Support/Compiler.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000033#include "llvm/Target/TargetData.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000034
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000035#include <map>
Chris Lattnere160c9b2009-01-27 05:06:01 +000036
37
Chris Lattnerdce14062008-06-26 04:19:03 +000038using namespace clang;
Daniel Dunbar46f45b92008-09-09 01:06:48 +000039using namespace CodeGen;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000040using llvm::dyn_cast;
41
42// The version of the runtime that this class targets. Must match the version
43// in the runtime.
44static const int RuntimeVersion = 8;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +000045static const int NonFragileRuntimeVersion = 9;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000046static const int ProtocolVersion = 2;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000047static const int NonFragileProtocolVersion = 3;
Chris Lattner0f984262008-03-01 08:50:34 +000048
Chris Lattner0f984262008-03-01 08:50:34 +000049namespace {
Chris Lattnerdce14062008-06-26 04:19:03 +000050class CGObjCGNU : public CodeGen::CGObjCRuntime {
Chris Lattner0f984262008-03-01 08:50:34 +000051private:
Chris Lattnerdce14062008-06-26 04:19:03 +000052 CodeGen::CodeGenModule &CGM;
Chris Lattner0f984262008-03-01 08:50:34 +000053 llvm::Module &TheModule;
Chris Lattnere160c9b2009-01-27 05:06:01 +000054 const llvm::PointerType *SelectorTy;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000055 const llvm::IntegerType *Int8Ty;
Chris Lattnere160c9b2009-01-27 05:06:01 +000056 const llvm::PointerType *PtrToInt8Ty;
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +000057 const llvm::FunctionType *IMPTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000058 const llvm::PointerType *IdTy;
David Chisnallef6e0f32010-02-03 15:59:02 +000059 const llvm::PointerType *PtrToIdTy;
John McCallead608a2010-02-26 00:48:12 +000060 CanQualType ASTIdTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000061 const llvm::IntegerType *IntTy;
62 const llvm::PointerType *PtrTy;
63 const llvm::IntegerType *LongTy;
64 const llvm::PointerType *PtrToIntTy;
Daniel Dunbar5efccb12009-05-04 15:31:17 +000065 llvm::GlobalAlias *ClassPtrAlias;
66 llvm::GlobalAlias *MetaClassPtrAlias;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000067 std::vector<llvm::Constant*> Classes;
68 std::vector<llvm::Constant*> Categories;
69 std::vector<llvm::Constant*> ConstantStrings;
David Chisnall48272a02010-01-27 12:49:23 +000070 llvm::StringMap<llvm::Constant*> ObjCStrings;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000071 llvm::Function *LoadFunction;
72 llvm::StringMap<llvm::Constant*> ExistingProtocols;
73 typedef std::pair<std::string, std::string> TypedSelector;
74 std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
75 llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
David Chisnallef6e0f32010-02-03 15:59:02 +000076 // Selectors that we don't emit in GC mode
77 Selector RetainSel, ReleaseSel, AutoreleaseSel;
78 // Functions used for GC.
79 llvm::Constant *IvarAssignFn, *StrongCastAssignFn, *MemMoveFn, *WeakReadFn,
80 *WeakAssignFn, *GlobalAssignFn;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000081 // Some zeros used for GEPs in lots of places.
82 llvm::Constant *Zeros[2];
83 llvm::Constant *NULLPtr;
Owen Andersona1cf15f2009-07-14 23:10:40 +000084 llvm::LLVMContext &VMContext;
David Chisnallc6cd5fd2010-04-28 19:33:36 +000085 /// Metadata kind used to tie method lookups to message sends.
86 unsigned msgSendMDKind;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000087private:
88 llvm::Constant *GenerateIvarList(
89 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
90 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
91 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets);
92 llvm::Constant *GenerateMethodList(const std::string &ClassName,
93 const std::string &CategoryName,
Mike Stump1eb44332009-09-09 15:08:12 +000094 const llvm::SmallVectorImpl<Selector> &MethodSels,
95 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000096 bool isClassMethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +000097 llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000098 llvm::Constant *GeneratePropertyList(const ObjCImplementationDecl *OID,
99 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
100 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000101 llvm::Constant *GenerateProtocolList(
102 const llvm::SmallVectorImpl<std::string> &Protocols);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000103 // To ensure that all protocols are seen by the runtime, we add a category on
104 // a class defined in the runtime, declaring no methods, but adopting the
105 // protocols.
106 void GenerateProtocolHolderCategory(void);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000107 llvm::Constant *GenerateClassStructure(
108 llvm::Constant *MetaClass,
109 llvm::Constant *SuperClass,
110 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000111 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000112 llvm::Constant *Version,
113 llvm::Constant *InstanceSize,
114 llvm::Constant *IVars,
115 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000116 llvm::Constant *Protocols,
117 llvm::Constant *IvarOffsets,
David Chisnall8c757f92010-04-28 14:29:56 +0000118 llvm::Constant *Properties,
119 bool isMeta=false);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000120 llvm::Constant *GenerateProtocolMethodList(
121 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
122 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes);
123 llvm::Constant *MakeConstantString(const std::string &Str, const std::string
124 &Name="");
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000125 llvm::Constant *ExportUniqueString(const std::string &Str, const std::string
126 prefix);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000127 llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000128 std::vector<llvm::Constant*> &V, llvm::StringRef Name="",
David Chisnall41d63ed2010-01-08 00:14:31 +0000129 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000130 llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000131 std::vector<llvm::Constant*> &V, llvm::StringRef Name="",
David Chisnall41d63ed2010-01-08 00:14:31 +0000132 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000133 llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
134 const ObjCIvarDecl *Ivar);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000135 void EmitClassRef(const std::string &className);
David Chisnallef6e0f32010-02-03 15:59:02 +0000136 llvm::Value* EnforceType(CGBuilderTy B, llvm::Value *V, const llvm::Type *Ty){
137 if (V->getType() == Ty) return V;
138 return B.CreateBitCast(V, Ty);
139 }
Chris Lattner0f984262008-03-01 08:50:34 +0000140public:
Chris Lattnerdce14062008-06-26 04:19:03 +0000141 CGObjCGNU(CodeGen::CodeGenModule &cgm);
David Chisnall0d13f6f2010-01-23 02:40:42 +0000142 virtual llvm::Constant *GenerateConstantString(const StringLiteral *);
Mike Stump1eb44332009-09-09 15:08:12 +0000143 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000144 GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
John McCallef072fd2010-05-22 01:48:05 +0000145 ReturnValueSlot Return,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000146 QualType ResultType,
147 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000148 llvm::Value *Receiver,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000149 const CallArgList &CallArgs,
David Chisnallc6cd5fd2010-04-28 19:33:36 +0000150 const ObjCInterfaceDecl *Class,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000151 const ObjCMethodDecl *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000152 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000153 GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
John McCallef072fd2010-05-22 01:48:05 +0000154 ReturnValueSlot Return,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000155 QualType ResultType,
156 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000157 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000158 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000159 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000160 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000161 const CallArgList &CallArgs,
162 const ObjCMethodDecl *Method);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000163 virtual llvm::Value *GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000164 const ObjCInterfaceDecl *OID);
Fariborz Jahanian03b29602010-06-17 19:56:20 +0000165 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel,
166 bool lval = false);
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000167 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
168 *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000169
170 virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000171 const ObjCContainerDecl *CD);
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000172 virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
173 virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000174 virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000175 const ObjCProtocolDecl *PD);
176 virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000177 virtual llvm::Function *ModuleInitFunction();
Daniel Dunbar49f66022008-09-24 03:38:44 +0000178 virtual llvm::Function *GetPropertyGetFunction();
179 virtual llvm::Function *GetPropertySetFunction();
Fariborz Jahanian6cc59062010-04-12 18:18:10 +0000180 virtual llvm::Function *GetCopyStructFunction();
Daniel Dunbar309a4362009-07-24 07:40:24 +0000181 virtual llvm::Constant *EnumerationMutationFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000182
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000183 virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
184 const Stmt &S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000185 virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
186 const ObjCAtThrowStmt &S);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000187 virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6dc23172008-11-18 21:45:40 +0000188 llvm::Value *AddrWeakObj);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000189 virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
190 llvm::Value *src, llvm::Value *dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000191 virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
192 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +0000193 virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +0000194 llvm::Value *src, llvm::Value *dest,
195 llvm::Value *ivarOffset);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000196 virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
197 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000198 virtual void EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +0000199 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000200 llvm::Value *SrcPtr,
Fariborz Jahanian55bcace2010-06-15 22:44:06 +0000201 llvm::Value *Size);
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000202 virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
203 QualType ObjectTy,
204 llvm::Value *BaseValue,
205 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000206 unsigned CVRQualifiers);
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000207 virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +0000208 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000209 const ObjCIvarDecl *Ivar);
Chris Lattner0f984262008-03-01 08:50:34 +0000210};
211} // end anonymous namespace
212
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000213
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000214/// Emits a reference to a dummy variable which is emitted with each class.
215/// This ensures that a linker error will be generated when trying to link
216/// together modules where a referenced class is not defined.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000217void CGObjCGNU::EmitClassRef(const std::string &className) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000218 std::string symbolRef = "__objc_class_ref_" + className;
219 // Don't emit two copies of the same symbol
Mike Stumpbb1c8602009-07-31 21:31:32 +0000220 if (TheModule.getGlobalVariable(symbolRef))
221 return;
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000222 std::string symbolName = "__objc_class_name_" + className;
223 llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
224 if (!ClassSymbol) {
Owen Anderson1c431b32009-07-08 19:05:04 +0000225 ClassSymbol = new llvm::GlobalVariable(TheModule, LongTy, false,
226 llvm::GlobalValue::ExternalLinkage, 0, symbolName);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000227 }
Owen Anderson1c431b32009-07-08 19:05:04 +0000228 new llvm::GlobalVariable(TheModule, ClassSymbol->getType(), true,
Chris Lattnerf35271b2009-08-05 05:25:18 +0000229 llvm::GlobalValue::WeakAnyLinkage, ClassSymbol, symbolRef);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000230}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000231
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000232static std::string SymbolNameForMethod(const std::string &ClassName, const
233 std::string &CategoryName, const std::string &MethodName, bool isClassMethod)
234{
David Chisnalld3467362010-01-14 14:08:19 +0000235 std::string MethodNameColonStripped = MethodName;
236 std::replace(MethodNameColonStripped.begin(), MethodNameColonStripped.end(),
237 ':', '_');
238 return std::string(isClassMethod ? "_c_" : "_i_") + ClassName + "_" +
239 CategoryName + "_" + MethodNameColonStripped;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000240}
David Chisnall87935a82010-05-08 20:58:05 +0000241static std::string MangleSelectorTypes(const std::string &TypeString) {
242 std::string Mangled = TypeString;
David Chisnall22b88272010-05-09 01:01:43 +0000243 // Simple mangling to avoid breaking when we mix JIT / static code.
244 // Not part of the ABI, subject to change without notice.
David Chisnall87935a82010-05-08 20:58:05 +0000245 std::replace(Mangled.begin(), Mangled.end(), '@', '_');
David Chisnall22b88272010-05-09 01:01:43 +0000246 std::replace(Mangled.begin(), Mangled.end(), ':', 'J');
247 std::replace(Mangled.begin(), Mangled.end(), '*', 'e');
248 std::replace(Mangled.begin(), Mangled.end(), '#', 'E');
249 std::replace(Mangled.begin(), Mangled.end(), ':', 'j');
250 std::replace(Mangled.begin(), Mangled.end(), '(', 'g');
251 std::replace(Mangled.begin(), Mangled.end(), ')', 'G');
252 std::replace(Mangled.begin(), Mangled.end(), '[', 'h');
253 std::replace(Mangled.begin(), Mangled.end(), ']', 'H');
David Chisnall87935a82010-05-08 20:58:05 +0000254 return Mangled;
255}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000256
Chris Lattnerdce14062008-06-26 04:19:03 +0000257CGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000258 : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
Owen Andersona1cf15f2009-07-14 23:10:40 +0000259 MetaClassPtrAlias(0), VMContext(cgm.getLLVMContext()) {
David Chisnallc6cd5fd2010-04-28 19:33:36 +0000260
261 msgSendMDKind = VMContext.getMDKindID("GNUObjCMessageSend");
262
Chris Lattnere160c9b2009-01-27 05:06:01 +0000263 IntTy = cast<llvm::IntegerType>(
264 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
265 LongTy = cast<llvm::IntegerType>(
266 CGM.getTypes().ConvertType(CGM.getContext().LongTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000267
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000268 Int8Ty = llvm::Type::getInt8Ty(VMContext);
269 // C string type. Used in lots of places.
270 PtrToInt8Ty = llvm::PointerType::getUnqual(Int8Ty);
271
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000272 Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000273 Zeros[1] = Zeros[0];
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000274 NULLPtr = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Chris Lattner391d77a2008-03-30 23:03:07 +0000275 // Get the selector Type.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000276 QualType selTy = CGM.getContext().getObjCSelType();
277 if (QualType() == selTy) {
278 SelectorTy = PtrToInt8Ty;
279 } else {
280 SelectorTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(selTy));
281 }
Chris Lattnere160c9b2009-01-27 05:06:01 +0000282
Owen Anderson96e0fc72009-07-29 22:16:19 +0000283 PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
Chris Lattner391d77a2008-03-30 23:03:07 +0000284 PtrTy = PtrToInt8Ty;
Mike Stump1eb44332009-09-09 15:08:12 +0000285
Chris Lattner391d77a2008-03-30 23:03:07 +0000286 // Object type
John McCallead608a2010-02-26 00:48:12 +0000287 ASTIdTy = CGM.getContext().getCanonicalType(CGM.getContext().getObjCIdType());
David Chisnall0d13f6f2010-01-23 02:40:42 +0000288 if (QualType() == ASTIdTy) {
289 IdTy = PtrToInt8Ty;
290 } else {
291 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
292 }
David Chisnallef6e0f32010-02-03 15:59:02 +0000293 PtrToIdTy = llvm::PointerType::getUnqual(IdTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000294
Chris Lattner391d77a2008-03-30 23:03:07 +0000295 // IMP type
296 std::vector<const llvm::Type*> IMPArgs;
297 IMPArgs.push_back(IdTy);
298 IMPArgs.push_back(SelectorTy);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000299 IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
David Chisnallef6e0f32010-02-03 15:59:02 +0000300
301 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
302 // Get selectors needed in GC mode
303 RetainSel = GetNullarySelector("retain", CGM.getContext());
304 ReleaseSel = GetNullarySelector("release", CGM.getContext());
305 AutoreleaseSel = GetNullarySelector("autorelease", CGM.getContext());
306
307 // Get functions needed in GC mode
308
309 // id objc_assign_ivar(id, id, ptrdiff_t);
310 std::vector<const llvm::Type*> Args(1, IdTy);
311 Args.push_back(PtrToIdTy);
312 // FIXME: ptrdiff_t
313 Args.push_back(LongTy);
314 llvm::FunctionType *FTy = llvm::FunctionType::get(IdTy, Args, false);
315 IvarAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_ivar");
316 // id objc_assign_strongCast (id, id*)
317 Args.pop_back();
318 FTy = llvm::FunctionType::get(IdTy, Args, false);
319 StrongCastAssignFn =
320 CGM.CreateRuntimeFunction(FTy, "objc_assign_strongCast");
321 // id objc_assign_global(id, id*);
322 FTy = llvm::FunctionType::get(IdTy, Args, false);
323 GlobalAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_global");
324 // id objc_assign_weak(id, id*);
325 FTy = llvm::FunctionType::get(IdTy, Args, false);
326 WeakAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_weak");
327 // id objc_read_weak(id*);
328 Args.clear();
329 Args.push_back(PtrToIdTy);
330 FTy = llvm::FunctionType::get(IdTy, Args, false);
331 WeakReadFn = CGM.CreateRuntimeFunction(FTy, "objc_read_weak");
332 // void *objc_memmove_collectable(void*, void *, size_t);
333 Args.clear();
334 Args.push_back(PtrToInt8Ty);
335 Args.push_back(PtrToInt8Ty);
336 // FIXME: size_t
337 Args.push_back(LongTy);
338 FTy = llvm::FunctionType::get(IdTy, Args, false);
339 MemMoveFn = CGM.CreateRuntimeFunction(FTy, "objc_memmove_collectable");
340 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000341}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000342
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000343// This has to perform the lookup every time, since posing and related
344// techniques can modify the name -> class mapping.
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000345llvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000346 const ObjCInterfaceDecl *OID) {
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000347 llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
David Chisnall41d63ed2010-01-08 00:14:31 +0000348 // With the incompatible ABI, this will need to be replaced with a direct
349 // reference to the class symbol. For the compatible nonfragile ABI we are
350 // still performing this lookup at run time but emitting the symbol for the
351 // class externally so that we can make the switch later.
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000352 EmitClassRef(OID->getNameAsString());
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000353 ClassName = Builder.CreateStructGEP(ClassName, 0);
354
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000355 std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000356 llvm::Constant *ClassLookupFn =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000357 CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000358 Params,
359 true),
360 "objc_lookup_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000361 return Builder.CreateCall(ClassLookupFn, ClassName);
Chris Lattner391d77a2008-03-30 23:03:07 +0000362}
363
Fariborz Jahanian03b29602010-06-17 19:56:20 +0000364llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel,
365 bool lval) {
Chris Lattner077bf5e2008-11-24 03:33:13 +0000366 llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
Chris Lattner8e67b632008-06-26 04:37:12 +0000367 if (US == 0)
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000368 US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000369 llvm::GlobalValue::PrivateLinkage,
370 ".objc_untyped_selector_alias"+Sel.getAsString(),
Chris Lattner8e67b632008-06-26 04:37:12 +0000371 NULL, &TheModule);
Fariborz Jahanian03b29602010-06-17 19:56:20 +0000372 if (lval)
373 return US;
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000374 return Builder.CreateLoad(US);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000375}
376
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000377llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000378 *Method) {
379
380 std::string SelName = Method->getSelector().getAsString();
381 std::string SelTypes;
382 CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
383 // Typed selectors
384 TypedSelector Selector = TypedSelector(SelName,
385 SelTypes);
386
387 // If it's already cached, return it.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000388 if (TypedSelectors[Selector]) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000389 return Builder.CreateLoad(TypedSelectors[Selector]);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000390 }
391
392 // If it isn't, cache it.
393 llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000394 llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000395 llvm::GlobalValue::PrivateLinkage, ".objc_selector_alias" + SelName,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000396 NULL, &TheModule);
397 TypedSelectors[Selector] = Sel;
398
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000399 return Builder.CreateLoad(Sel);
Chris Lattner8e67b632008-06-26 04:37:12 +0000400}
401
Chris Lattner5e7dcc62008-06-26 04:44:19 +0000402llvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
403 const std::string &Name) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000404 llvm::Constant *ConstStr = CGM.GetAddrOfConstantCString(Str, Name.c_str());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000405 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000406}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000407llvm::Constant *CGObjCGNU::ExportUniqueString(const std::string &Str,
408 const std::string prefix) {
409 std::string name = prefix + Str;
410 llvm::Constant *ConstStr = TheModule.getGlobalVariable(name);
411 if (!ConstStr) {
412 llvm::Constant *value = llvm::ConstantArray::get(VMContext, Str, true);
413 ConstStr = new llvm::GlobalVariable(TheModule, value->getType(), true,
414 llvm::GlobalValue::LinkOnceODRLinkage, value, prefix + Str);
415 }
416 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
417}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000418
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000419llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000420 std::vector<llvm::Constant*> &V, llvm::StringRef Name,
David Chisnall41d63ed2010-01-08 00:14:31 +0000421 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson08e25242009-07-27 22:29:56 +0000422 llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000423 return new llvm::GlobalVariable(TheModule, Ty, false,
424 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000425}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000426
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000427llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000428 std::vector<llvm::Constant*> &V, llvm::StringRef Name,
David Chisnall41d63ed2010-01-08 00:14:31 +0000429 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson7db6d832009-07-28 18:33:04 +0000430 llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000431 return new llvm::GlobalVariable(TheModule, Ty, false,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000432 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000433}
434
435/// Generate an NSConstantString object.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000436llvm::Constant *CGObjCGNU::GenerateConstantString(const StringLiteral *SL) {
David Chisnall48272a02010-01-27 12:49:23 +0000437
David Chisnall0d13f6f2010-01-23 02:40:42 +0000438 std::string Str(SL->getStrData(), SL->getByteLength());
439
David Chisnall48272a02010-01-27 12:49:23 +0000440 // Look for an existing one
441 llvm::StringMap<llvm::Constant*>::iterator old = ObjCStrings.find(Str);
442 if (old != ObjCStrings.end())
443 return old->getValue();
444
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000445 std::vector<llvm::Constant*> Ivars;
446 Ivars.push_back(NULLPtr);
Chris Lattner13fd7e52008-06-21 21:44:18 +0000447 Ivars.push_back(MakeConstantString(Str));
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000448 Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000449 llvm::Constant *ObjCStr = MakeGlobal(
Owen Anderson47a434f2009-08-05 23:18:46 +0000450 llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000451 Ivars, ".objc_str");
David Chisnall48272a02010-01-27 12:49:23 +0000452 ObjCStr = llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty);
453 ObjCStrings[Str] = ObjCStr;
454 ConstantStrings.push_back(ObjCStr);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000455 return ObjCStr;
456}
457
458///Generates a message send where the super is the receiver. This is a message
459///send to self with special delivery semantics indicating which class's method
460///should be called.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000461CodeGen::RValue
462CGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
John McCallef072fd2010-05-22 01:48:05 +0000463 ReturnValueSlot Return,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000464 QualType ResultType,
465 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000466 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000467 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000468 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000469 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000470 const CallArgList &CallArgs,
471 const ObjCMethodDecl *Method) {
David Chisnallef6e0f32010-02-03 15:59:02 +0000472 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
473 if (Sel == RetainSel || Sel == AutoreleaseSel) {
474 return RValue::get(Receiver);
475 }
476 if (Sel == ReleaseSel) {
477 return RValue::get(0);
478 }
479 }
David Chisnalldb831942010-05-01 12:37:16 +0000480
481 CGBuilderTy &Builder = CGF.Builder;
482 llvm::Value *cmd = GetSelector(Builder, Sel);
483
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000484
485 CallArgList ActualArgs;
486
487 ActualArgs.push_back(
David Chisnalldb831942010-05-01 12:37:16 +0000488 std::make_pair(RValue::get(Builder.CreateBitCast(Receiver, IdTy)),
David Chisnall0f436562009-08-17 16:35:33 +0000489 ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000490 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
491 CGF.getContext().getObjCSelType()));
492 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
493
494 CodeGenTypes &Types = CGM.getTypes();
John McCall04a67a62010-02-05 21:31:56 +0000495 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs,
Rafael Espindola264ba482010-03-30 20:24:48 +0000496 FunctionType::ExtInfo());
Daniel Dunbar67939662009-09-17 04:01:40 +0000497 const llvm::FunctionType *impType =
498 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000499
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000500 llvm::Value *ReceiverClass = 0;
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000501 if (isCategoryImpl) {
502 llvm::Constant *classLookupFunction = 0;
503 std::vector<const llvm::Type*> Params;
504 Params.push_back(PtrTy);
505 if (IsClassMessage) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000506 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000507 IdTy, Params, true), "objc_get_meta_class");
508 } else {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000509 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000510 IdTy, Params, true), "objc_get_class");
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000511 }
David Chisnalldb831942010-05-01 12:37:16 +0000512 ReceiverClass = Builder.CreateCall(classLookupFunction,
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000513 MakeConstantString(Class->getNameAsString()));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000514 } else {
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000515 // Set up global aliases for the metaclass or class pointer if they do not
516 // already exist. These will are forward-references which will be set to
Mike Stumpbb1c8602009-07-31 21:31:32 +0000517 // pointers to the class and metaclass structure created for the runtime
518 // load function. To send a message to super, we look up the value of the
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000519 // super_class pointer from either the class or metaclass structure.
520 if (IsClassMessage) {
521 if (!MetaClassPtrAlias) {
522 MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
523 llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
524 Class->getNameAsString(), NULL, &TheModule);
525 }
526 ReceiverClass = MetaClassPtrAlias;
527 } else {
528 if (!ClassPtrAlias) {
529 ClassPtrAlias = new llvm::GlobalAlias(IdTy,
530 llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
531 Class->getNameAsString(), NULL, &TheModule);
532 }
533 ReceiverClass = ClassPtrAlias;
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000534 }
Chris Lattner71238f62009-04-25 23:19:45 +0000535 }
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000536 // Cast the pointer to a simplified version of the class structure
David Chisnalldb831942010-05-01 12:37:16 +0000537 ReceiverClass = Builder.CreateBitCast(ReceiverClass,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000538 llvm::PointerType::getUnqual(
Owen Anderson47a434f2009-08-05 23:18:46 +0000539 llvm::StructType::get(VMContext, IdTy, IdTy, NULL)));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000540 // Get the superclass pointer
David Chisnalldb831942010-05-01 12:37:16 +0000541 ReceiverClass = Builder.CreateStructGEP(ReceiverClass, 1);
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000542 // Load the superclass pointer
David Chisnalldb831942010-05-01 12:37:16 +0000543 ReceiverClass = Builder.CreateLoad(ReceiverClass);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000544 // Construct the structure used to look up the IMP
Owen Anderson47a434f2009-08-05 23:18:46 +0000545 llvm::StructType *ObjCSuperTy = llvm::StructType::get(VMContext,
546 Receiver->getType(), IdTy, NULL);
David Chisnalldb831942010-05-01 12:37:16 +0000547 llvm::Value *ObjCSuper = Builder.CreateAlloca(ObjCSuperTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000548
David Chisnalldb831942010-05-01 12:37:16 +0000549 Builder.CreateStore(Receiver, Builder.CreateStructGEP(ObjCSuper, 0));
550 Builder.CreateStore(ReceiverClass, Builder.CreateStructGEP(ObjCSuper, 1));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000551
552 // Get the IMP
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000553 std::vector<const llvm::Type*> Params;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000554 Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000555 Params.push_back(SelectorTy);
David Chisnalldb831942010-05-01 12:37:16 +0000556
557 llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
558 llvm::Value *imp;
559
560 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
561 // The lookup function returns a slot, which can be safely cached.
562 llvm::Type *SlotTy = llvm::StructType::get(VMContext, PtrTy, PtrTy, PtrTy,
563 IntTy, llvm::PointerType::getUnqual(impType), NULL);
564
565 llvm::Constant *lookupFunction =
566 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
567 llvm::PointerType::getUnqual(SlotTy), Params, true),
568 "objc_slot_lookup_super");
569
570 llvm::CallInst *slot = Builder.CreateCall(lookupFunction, lookupArgs,
571 lookupArgs+2);
572 slot->setOnlyReadsMemory();
573
574 imp = Builder.CreateLoad(Builder.CreateStructGEP(slot, 4));
575 } else {
Mike Stump1eb44332009-09-09 15:08:12 +0000576 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000577 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
578 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000579 "objc_msg_lookup_super");
David Chisnalldb831942010-05-01 12:37:16 +0000580 imp = Builder.CreateCall(lookupFunction, lookupArgs, lookupArgs+2);
581 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000582
David Chisnalldd5c98f2010-05-01 11:15:56 +0000583 llvm::Value *impMD[] = {
584 llvm::MDString::get(VMContext, Sel.getAsString()),
585 llvm::MDString::get(VMContext, Class->getSuperClass()->getNameAsString()),
586 llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext), IsClassMessage)
587 };
588 llvm::MDNode *node = llvm::MDNode::get(VMContext, impMD, 3);
589
David Chisnall4b02afc2010-05-02 13:41:58 +0000590 llvm::Instruction *call;
John McCallef072fd2010-05-22 01:48:05 +0000591 RValue msgRet = CGF.EmitCall(FnInfo, imp, Return, ActualArgs,
David Chisnall4b02afc2010-05-02 13:41:58 +0000592 0, &call);
593 call->setMetadata(msgSendMDKind, node);
594 return msgRet;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000595}
596
Mike Stump1eb44332009-09-09 15:08:12 +0000597/// Generate code for a message send expression.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000598CodeGen::RValue
599CGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
John McCallef072fd2010-05-22 01:48:05 +0000600 ReturnValueSlot Return,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000601 QualType ResultType,
602 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000603 llvm::Value *Receiver,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000604 const CallArgList &CallArgs,
David Chisnallc6cd5fd2010-04-28 19:33:36 +0000605 const ObjCInterfaceDecl *Class,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000606 const ObjCMethodDecl *Method) {
David Chisnall664b7c72010-04-27 15:08:48 +0000607 // Strip out message sends to retain / release in GC mode
David Chisnallef6e0f32010-02-03 15:59:02 +0000608 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
609 if (Sel == RetainSel || Sel == AutoreleaseSel) {
610 return RValue::get(Receiver);
611 }
612 if (Sel == ReleaseSel) {
613 return RValue::get(0);
614 }
615 }
David Chisnall664b7c72010-04-27 15:08:48 +0000616
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000617 CGBuilderTy &Builder = CGF.Builder;
David Chisnall664b7c72010-04-27 15:08:48 +0000618
619 // If the return type is something that goes in an integer register, the
620 // runtime will handle 0 returns. For other cases, we fill in the 0 value
621 // ourselves.
622 //
623 // The language spec says the result of this kind of message send is
624 // undefined, but lots of people seem to have forgotten to read that
625 // paragraph and insist on sending messages to nil that have structure
626 // returns. With GCC, this generates a random return value (whatever happens
627 // to be on the stack / in those registers at the time) on most platforms,
628 // and generates a SegV on SPARC. With LLVM it corrupts the stack.
629 bool isPointerSizedReturn = false;
Douglas Gregor9d3347a2010-06-16 00:35:25 +0000630 if (ResultType->isAnyPointerType() ||
631 ResultType->isIntegralOrEnumerationType() || ResultType->isVoidType())
David Chisnall664b7c72010-04-27 15:08:48 +0000632 isPointerSizedReturn = true;
633
634 llvm::BasicBlock *startBB = 0;
635 llvm::BasicBlock *messageBB = 0;
David Chisnalla54da052010-05-20 13:45:48 +0000636 llvm::BasicBlock *continueBB = 0;
David Chisnall664b7c72010-04-27 15:08:48 +0000637
638 if (!isPointerSizedReturn) {
639 startBB = Builder.GetInsertBlock();
640 messageBB = CGF.createBasicBlock("msgSend");
David Chisnalla54da052010-05-20 13:45:48 +0000641 continueBB = CGF.createBasicBlock("continue");
David Chisnall664b7c72010-04-27 15:08:48 +0000642
643 llvm::Value *isNil = Builder.CreateICmpEQ(Receiver,
644 llvm::Constant::getNullValue(Receiver->getType()));
David Chisnalla54da052010-05-20 13:45:48 +0000645 Builder.CreateCondBr(isNil, continueBB, messageBB);
David Chisnall664b7c72010-04-27 15:08:48 +0000646 CGF.EmitBlock(messageBB);
647 }
648
David Chisnall0f436562009-08-17 16:35:33 +0000649 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000650 llvm::Value *cmd;
651 if (Method)
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000652 cmd = GetSelector(Builder, Method);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000653 else
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000654 cmd = GetSelector(Builder, Sel);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000655 CallArgList ActualArgs;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000656
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000657 Receiver = Builder.CreateBitCast(Receiver, IdTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000658 ActualArgs.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000659 std::make_pair(RValue::get(Receiver), ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000660 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
661 CGF.getContext().getObjCSelType()));
662 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
663
664 CodeGenTypes &Types = CGM.getTypes();
John McCall04a67a62010-02-05 21:31:56 +0000665 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs,
Rafael Espindola264ba482010-03-30 20:24:48 +0000666 FunctionType::ExtInfo());
Daniel Dunbar67939662009-09-17 04:01:40 +0000667 const llvm::FunctionType *impType =
668 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000669
David Chisnall63e742b2010-05-01 12:56:56 +0000670 llvm::Value *impMD[] = {
671 llvm::MDString::get(VMContext, Sel.getAsString()),
672 llvm::MDString::get(VMContext, Class ? Class->getNameAsString() :""),
673 llvm::ConstantInt::get(llvm::Type::getInt1Ty(VMContext), Class!=0)
674 };
675 llvm::MDNode *node = llvm::MDNode::get(VMContext, impMD, 3);
676
677
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000678 llvm::Value *imp;
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000679 // For sender-aware dispatch, we pass the sender as the third argument to a
680 // lookup function. When sending messages from C code, the sender is nil.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000681 // objc_msg_lookup_sender(id *receiver, SEL selector, id sender);
682 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
683
684 std::vector<const llvm::Type*> Params;
685 llvm::Value *ReceiverPtr = CGF.CreateTempAlloca(Receiver->getType());
686 Builder.CreateStore(Receiver, ReceiverPtr);
687 Params.push_back(ReceiverPtr->getType());
688 Params.push_back(SelectorTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000689 llvm::Value *self;
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000690
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000691 if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
692 self = CGF.LoadObjCSelf();
693 } else {
Owen Anderson03e20502009-07-30 23:11:26 +0000694 self = llvm::ConstantPointerNull::get(IdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000695 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000696
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000697 Params.push_back(self->getType());
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000698
699 // The lookup function returns a slot, which can be safely cached.
700 llvm::Type *SlotTy = llvm::StructType::get(VMContext, PtrTy, PtrTy, PtrTy,
701 IntTy, llvm::PointerType::getUnqual(impType), NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000702 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000703 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000704 llvm::PointerType::getUnqual(SlotTy), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000705 "objc_msg_lookup_sender");
706
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000707 // The lookup function is guaranteed not to capture the receiver pointer.
708 if (llvm::Function *LookupFn = dyn_cast<llvm::Function>(lookupFunction)) {
709 LookupFn->setDoesNotCapture(1);
710 }
711
David Chisnall866163b2010-04-30 13:36:12 +0000712 llvm::CallInst *slot =
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000713 Builder.CreateCall3(lookupFunction, ReceiverPtr, cmd, self);
David Chisnall866163b2010-04-30 13:36:12 +0000714 slot->setOnlyReadsMemory();
David Chisnall63e742b2010-05-01 12:56:56 +0000715 slot->setMetadata(msgSendMDKind, node);
David Chisnall866163b2010-04-30 13:36:12 +0000716
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000717 imp = Builder.CreateLoad(Builder.CreateStructGEP(slot, 4));
David Chisnallc6cd5fd2010-04-28 19:33:36 +0000718
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000719 // The lookup function may have changed the receiver, so make sure we use
720 // the new one.
721 ActualArgs[0] =
722 std::make_pair(RValue::get(Builder.CreateLoad(ReceiverPtr)), ASTIdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000723 } else {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000724 std::vector<const llvm::Type*> Params;
725 Params.push_back(Receiver->getType());
726 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000727 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000728 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
729 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000730 "objc_msg_lookup");
731
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000732 imp = Builder.CreateCall2(lookupFunction, Receiver, cmd);
David Chisnall63e742b2010-05-01 12:56:56 +0000733 cast<llvm::CallInst>(imp)->setMetadata(msgSendMDKind, node);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000734 }
David Chisnall4b02afc2010-05-02 13:41:58 +0000735 llvm::Instruction *call;
John McCallef072fd2010-05-22 01:48:05 +0000736 RValue msgRet = CGF.EmitCall(FnInfo, imp, Return, ActualArgs,
David Chisnall4b02afc2010-05-02 13:41:58 +0000737 0, &call);
738 call->setMetadata(msgSendMDKind, node);
David Chisnall664b7c72010-04-27 15:08:48 +0000739
David Chisnalla54da052010-05-20 13:45:48 +0000740
David Chisnall664b7c72010-04-27 15:08:48 +0000741 if (!isPointerSizedReturn) {
David Chisnalla54da052010-05-20 13:45:48 +0000742 messageBB = CGF.Builder.GetInsertBlock();
743 CGF.Builder.CreateBr(continueBB);
744 CGF.EmitBlock(continueBB);
David Chisnall664b7c72010-04-27 15:08:48 +0000745 if (msgRet.isScalar()) {
746 llvm::Value *v = msgRet.getScalarVal();
747 llvm::PHINode *phi = Builder.CreatePHI(v->getType());
748 phi->addIncoming(v, messageBB);
749 phi->addIncoming(llvm::Constant::getNullValue(v->getType()), startBB);
750 msgRet = RValue::get(phi);
751 } else if (msgRet.isAggregate()) {
752 llvm::Value *v = msgRet.getAggregateAddr();
753 llvm::PHINode *phi = Builder.CreatePHI(v->getType());
754 const llvm::PointerType *RetTy = cast<llvm::PointerType>(v->getType());
David Chisnall866163b2010-04-30 13:36:12 +0000755 llvm::AllocaInst *NullVal =
756 CGF.CreateTempAlloca(RetTy->getElementType(), "null");
David Chisnall664b7c72010-04-27 15:08:48 +0000757 CGF.InitTempAlloca(NullVal,
758 llvm::Constant::getNullValue(RetTy->getElementType()));
759 phi->addIncoming(v, messageBB);
760 phi->addIncoming(NullVal, startBB);
761 msgRet = RValue::getAggregate(phi);
762 } else /* isComplex() */ {
763 std::pair<llvm::Value*,llvm::Value*> v = msgRet.getComplexVal();
764 llvm::PHINode *phi = Builder.CreatePHI(v.first->getType());
765 phi->addIncoming(v.first, messageBB);
766 phi->addIncoming(llvm::Constant::getNullValue(v.first->getType()),
767 startBB);
768 llvm::PHINode *phi2 = Builder.CreatePHI(v.second->getType());
769 phi2->addIncoming(v.second, messageBB);
770 phi2->addIncoming(llvm::Constant::getNullValue(v.second->getType()),
771 startBB);
772 msgRet = RValue::getComplex(phi, phi2);
773 }
774 }
775 return msgRet;
Chris Lattner0f984262008-03-01 08:50:34 +0000776}
777
Mike Stump1eb44332009-09-09 15:08:12 +0000778/// Generates a MethodList. Used in construction of a objc_class and
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000779/// objc_category structures.
780llvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
Mike Stump1eb44332009-09-09 15:08:12 +0000781 const std::string &CategoryName,
782 const llvm::SmallVectorImpl<Selector> &MethodSels,
783 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000784 bool isClassMethodList) {
David Chisnall0f436562009-08-17 16:35:33 +0000785 if (MethodSels.empty())
786 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000787 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000788 llvm::StructType *ObjCMethodTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000789 PtrToInt8Ty, // Really a selector, but the runtime creates it us.
790 PtrToInt8Ty, // Method types
Owen Anderson96e0fc72009-07-29 22:16:19 +0000791 llvm::PointerType::getUnqual(IMPTy), //Method pointer
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000792 NULL);
793 std::vector<llvm::Constant*> Methods;
794 std::vector<llvm::Constant*> Elements;
795 for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
796 Elements.clear();
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000797 if (llvm::Constant *Method =
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000798 TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
Chris Lattner077bf5e2008-11-24 03:33:13 +0000799 MethodSels[i].getAsString(),
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000800 isClassMethodList))) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000801 llvm::Constant *C = MakeConstantString(MethodSels[i].getAsString());
802 Elements.push_back(C);
803 Elements.push_back(MethodTypes[i]);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000804 Method = llvm::ConstantExpr::getBitCast(Method,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000805 llvm::PointerType::getUnqual(IMPTy));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000806 Elements.push_back(Method);
Owen Anderson08e25242009-07-27 22:29:56 +0000807 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000808 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000809 }
810
811 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000812 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000813 Methods.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000814 llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
Chris Lattnerfba67632008-06-26 04:52:29 +0000815 Methods);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000816
817 // Structure containing list pointer, array and array count
818 llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
Owen Anderson8c8f69e2009-08-13 23:27:53 +0000819 llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get(VMContext);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000820 llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
Owen Anderson47a434f2009-08-05 23:18:46 +0000821 llvm::StructType *ObjCMethodListTy = llvm::StructType::get(VMContext,
Mike Stump1eb44332009-09-09 15:08:12 +0000822 NextPtrTy,
823 IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000824 ObjCMethodArrayTy,
825 NULL);
826 // Refine next pointer type to concrete type
827 llvm::cast<llvm::OpaqueType>(
828 OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
829 ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
830
831 Methods.clear();
Owen Anderson03e20502009-07-30 23:11:26 +0000832 Methods.push_back(llvm::ConstantPointerNull::get(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000833 llvm::PointerType::getUnqual(ObjCMethodListTy)));
Owen Anderson0032b272009-08-13 21:57:51 +0000834 Methods.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000835 MethodTypes.size()));
836 Methods.push_back(MethodArray);
Mike Stump1eb44332009-09-09 15:08:12 +0000837
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000838 // Create an instance of the structure
839 return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
840}
841
842/// Generates an IvarList. Used in construction of a objc_class.
843llvm::Constant *CGObjCGNU::GenerateIvarList(
844 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
845 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
846 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets) {
David Chisnall18044632009-11-16 19:05:54 +0000847 if (IvarNames.size() == 0)
848 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000849 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000850 llvm::StructType *ObjCIvarTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000851 PtrToInt8Ty,
852 PtrToInt8Ty,
853 IntTy,
854 NULL);
855 std::vector<llvm::Constant*> Ivars;
856 std::vector<llvm::Constant*> Elements;
857 for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
858 Elements.clear();
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000859 Elements.push_back(IvarNames[i]);
860 Elements.push_back(IvarTypes[i]);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000861 Elements.push_back(IvarOffsets[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000862 Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000863 }
864
865 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000866 llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000867 IvarNames.size());
868
Mike Stump1eb44332009-09-09 15:08:12 +0000869
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000870 Elements.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000871 Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
Owen Anderson7db6d832009-07-28 18:33:04 +0000872 Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000873 // Structure containing array and array count
Owen Anderson47a434f2009-08-05 23:18:46 +0000874 llvm::StructType *ObjCIvarListTy = llvm::StructType::get(VMContext, IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000875 ObjCIvarArrayTy,
876 NULL);
877
878 // Create an instance of the structure
879 return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
880}
881
882/// Generate a class structure
883llvm::Constant *CGObjCGNU::GenerateClassStructure(
884 llvm::Constant *MetaClass,
885 llvm::Constant *SuperClass,
886 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000887 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000888 llvm::Constant *Version,
889 llvm::Constant *InstanceSize,
890 llvm::Constant *IVars,
891 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000892 llvm::Constant *Protocols,
893 llvm::Constant *IvarOffsets,
David Chisnall8c757f92010-04-28 14:29:56 +0000894 llvm::Constant *Properties,
895 bool isMeta) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000896 // Set up the class structure
897 // Note: Several of these are char*s when they should be ids. This is
898 // because the runtime performs this translation on load.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000899 //
900 // Fields marked New ABI are part of the GNUstep runtime. We emit them
901 // anyway; the classes will still work with the GNU runtime, they will just
902 // be ignored.
Owen Anderson47a434f2009-08-05 23:18:46 +0000903 llvm::StructType *ClassTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000904 PtrToInt8Ty, // class_pointer
905 PtrToInt8Ty, // super_class
906 PtrToInt8Ty, // name
907 LongTy, // version
908 LongTy, // info
909 LongTy, // instance_size
910 IVars->getType(), // ivars
911 Methods->getType(), // methods
Mike Stump1eb44332009-09-09 15:08:12 +0000912 // These are all filled in by the runtime, so we pretend
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000913 PtrTy, // dtable
914 PtrTy, // subclass_list
915 PtrTy, // sibling_class
916 PtrTy, // protocols
917 PtrTy, // gc_object_type
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000918 // New ABI:
919 LongTy, // abi_version
920 IvarOffsets->getType(), // ivar_offsets
921 Properties->getType(), // properties
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000922 NULL);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000923 llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000924 // Fill in the structure
925 std::vector<llvm::Constant*> Elements;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000926 Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000927 Elements.push_back(SuperClass);
Chris Lattnerd002cc62008-06-26 04:47:04 +0000928 Elements.push_back(MakeConstantString(Name, ".class_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000929 Elements.push_back(Zero);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000930 Elements.push_back(llvm::ConstantInt::get(LongTy, info));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000931 Elements.push_back(InstanceSize);
932 Elements.push_back(IVars);
933 Elements.push_back(Methods);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000934 Elements.push_back(NULLPtr);
935 Elements.push_back(NULLPtr);
936 Elements.push_back(NULLPtr);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000937 Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000938 Elements.push_back(NULLPtr);
939 Elements.push_back(Zero);
940 Elements.push_back(IvarOffsets);
941 Elements.push_back(Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000942 // Create an instance of the structure
David Chisnall41d63ed2010-01-08 00:14:31 +0000943 // This is now an externally visible symbol, so that we can speed up class
944 // messages in the next ABI.
David Chisnall8c757f92010-04-28 14:29:56 +0000945 return MakeGlobal(ClassTy, Elements, (isMeta ? "_OBJC_METACLASS_":
946 "_OBJC_CLASS_") + std::string(Name), llvm::GlobalValue::ExternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000947}
948
949llvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
950 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
951 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes) {
Mike Stump1eb44332009-09-09 15:08:12 +0000952 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000953 llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000954 PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
955 PtrToInt8Ty,
956 NULL);
957 std::vector<llvm::Constant*> Methods;
958 std::vector<llvm::Constant*> Elements;
959 for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
960 Elements.clear();
Mike Stump1eb44332009-09-09 15:08:12 +0000961 Elements.push_back(MethodNames[i]);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000962 Elements.push_back(MethodTypes[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000963 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000964 }
Owen Anderson96e0fc72009-07-29 22:16:19 +0000965 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000966 MethodNames.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000967 llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000968 Methods);
Owen Anderson47a434f2009-08-05 23:18:46 +0000969 llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000970 IntTy, ObjCMethodArrayTy, NULL);
971 Methods.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000972 Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000973 Methods.push_back(Array);
974 return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
975}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000976
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000977// Create the protocol list structure used in classes, categories and so on
978llvm::Constant *CGObjCGNU::GenerateProtocolList(
979 const llvm::SmallVectorImpl<std::string> &Protocols) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000980 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000981 Protocols.size());
Owen Anderson47a434f2009-08-05 23:18:46 +0000982 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000983 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
984 LongTy,//FIXME: Should be size_t
985 ProtocolArrayTy,
986 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000987 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000988 for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
989 iter != endIter ; iter++) {
David Chisnallff80fab2009-11-20 14:50:59 +0000990 llvm::Constant *protocol = 0;
991 llvm::StringMap<llvm::Constant*>::iterator value =
992 ExistingProtocols.find(*iter);
993 if (value == ExistingProtocols.end()) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000994 protocol = GenerateEmptyProtocol(*iter);
David Chisnallff80fab2009-11-20 14:50:59 +0000995 } else {
996 protocol = value->getValue();
997 }
Owen Anderson3c4972d2009-07-29 18:54:39 +0000998 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(protocol,
Owen Andersona1cf15f2009-07-14 23:10:40 +0000999 PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001000 Elements.push_back(Ptr);
1001 }
Owen Anderson7db6d832009-07-28 18:33:04 +00001002 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001003 Elements);
1004 Elements.clear();
1005 Elements.push_back(NULLPtr);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001006 Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001007 Elements.push_back(ProtocolArray);
1008 return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
1009}
1010
Mike Stump1eb44332009-09-09 15:08:12 +00001011llvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001012 const ObjCProtocolDecl *PD) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001013 llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
Mike Stump1eb44332009-09-09 15:08:12 +00001014 const llvm::Type *T =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001015 CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
Owen Anderson96e0fc72009-07-29 22:16:19 +00001016 return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001017}
1018
1019llvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
1020 const std::string &ProtocolName) {
1021 llvm::SmallVector<std::string, 0> EmptyStringVector;
1022 llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
1023
1024 llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001025 llvm::Constant *MethodList =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001026 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
1027 // Protocols are objects containing lists of the methods implemented and
1028 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +00001029 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001030 PtrToInt8Ty,
1031 ProtocolList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001032 MethodList->getType(),
1033 MethodList->getType(),
1034 MethodList->getType(),
1035 MethodList->getType(),
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001036 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +00001037 std::vector<llvm::Constant*> Elements;
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001038 // The isa pointer must be set to a magic number so the runtime knows it's
1039 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001040 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
1041 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +00001042 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001043 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001044 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
1045 Elements.push_back(ProtocolList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001046 Elements.push_back(MethodList);
1047 Elements.push_back(MethodList);
1048 Elements.push_back(MethodList);
1049 Elements.push_back(MethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +00001050 return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001051}
1052
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001053void CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
1054 ASTContext &Context = CGM.getContext();
Chris Lattner8ec03f52008-11-24 03:54:41 +00001055 std::string ProtocolName = PD->getNameAsString();
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001056 llvm::SmallVector<std::string, 16> Protocols;
1057 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
1058 E = PD->protocol_end(); PI != E; ++PI)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001059 Protocols.push_back((*PI)->getNameAsString());
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001060 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
1061 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001062 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodNames;
1063 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodTypes;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001064 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
1065 E = PD->instmeth_end(); iter != E; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001066 std::string TypeStr;
1067 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001068 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
1069 InstanceMethodNames.push_back(
1070 MakeConstantString((*iter)->getSelector().getAsString()));
1071 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
1072 } else {
1073 OptionalInstanceMethodNames.push_back(
1074 MakeConstantString((*iter)->getSelector().getAsString()));
1075 OptionalInstanceMethodTypes.push_back(MakeConstantString(TypeStr));
1076 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001077 }
1078 // Collect information about class methods:
1079 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
1080 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001081 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodNames;
1082 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001083 for (ObjCProtocolDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001084 iter = PD->classmeth_begin(), endIter = PD->classmeth_end();
1085 iter != endIter ; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001086 std::string TypeStr;
1087 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001088 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
1089 ClassMethodNames.push_back(
1090 MakeConstantString((*iter)->getSelector().getAsString()));
1091 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
1092 } else {
1093 OptionalClassMethodNames.push_back(
1094 MakeConstantString((*iter)->getSelector().getAsString()));
1095 OptionalClassMethodTypes.push_back(MakeConstantString(TypeStr));
1096 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +00001097 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001098
1099 llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
1100 llvm::Constant *InstanceMethodList =
1101 GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
1102 llvm::Constant *ClassMethodList =
1103 GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001104 llvm::Constant *OptionalInstanceMethodList =
1105 GenerateProtocolMethodList(OptionalInstanceMethodNames,
1106 OptionalInstanceMethodTypes);
1107 llvm::Constant *OptionalClassMethodList =
1108 GenerateProtocolMethodList(OptionalClassMethodNames,
1109 OptionalClassMethodTypes);
1110
1111 // Property metadata: name, attributes, isSynthesized, setter name, setter
1112 // types, getter name, getter types.
1113 // The isSynthesized value is always set to 0 in a protocol. It exists to
1114 // simplify the runtime library by allowing it to use the same data
1115 // structures for protocol metadata everywhere.
1116 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1117 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1118 PtrToInt8Ty, NULL);
1119 std::vector<llvm::Constant*> Properties;
1120 std::vector<llvm::Constant*> OptionalProperties;
1121
1122 // Add all of the property methods need adding to the method list and to the
1123 // property metadata list.
1124 for (ObjCContainerDecl::prop_iterator
1125 iter = PD->prop_begin(), endIter = PD->prop_end();
1126 iter != endIter ; iter++) {
1127 std::vector<llvm::Constant*> Fields;
1128 ObjCPropertyDecl *property = (*iter);
1129
1130 Fields.push_back(MakeConstantString(property->getNameAsString()));
1131 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1132 property->getPropertyAttributes()));
1133 Fields.push_back(llvm::ConstantInt::get(Int8Ty, 0));
1134 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
1135 std::string TypeStr;
1136 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1137 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1138 InstanceMethodTypes.push_back(TypeEncoding);
1139 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1140 Fields.push_back(TypeEncoding);
1141 } else {
1142 Fields.push_back(NULLPtr);
1143 Fields.push_back(NULLPtr);
1144 }
1145 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1146 std::string TypeStr;
1147 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1148 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1149 InstanceMethodTypes.push_back(TypeEncoding);
1150 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1151 Fields.push_back(TypeEncoding);
1152 } else {
1153 Fields.push_back(NULLPtr);
1154 Fields.push_back(NULLPtr);
1155 }
1156 if (property->getPropertyImplementation() == ObjCPropertyDecl::Optional) {
1157 OptionalProperties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1158 } else {
1159 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1160 }
1161 }
1162 llvm::Constant *PropertyArray = llvm::ConstantArray::get(
1163 llvm::ArrayType::get(PropertyMetadataTy, Properties.size()), Properties);
1164 llvm::Constant* PropertyListInitFields[] =
1165 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1166
1167 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001168 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001169 llvm::Constant *PropertyList = new llvm::GlobalVariable(TheModule,
1170 PropertyListInit->getType(), false, llvm::GlobalValue::InternalLinkage,
1171 PropertyListInit, ".objc_property_list");
1172
1173 llvm::Constant *OptionalPropertyArray =
1174 llvm::ConstantArray::get(llvm::ArrayType::get(PropertyMetadataTy,
1175 OptionalProperties.size()) , OptionalProperties);
1176 llvm::Constant* OptionalPropertyListInitFields[] = {
1177 llvm::ConstantInt::get(IntTy, OptionalProperties.size()), NULLPtr,
1178 OptionalPropertyArray };
1179
1180 llvm::Constant *OptionalPropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001181 llvm::ConstantStruct::get(VMContext, OptionalPropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001182 llvm::Constant *OptionalPropertyList = new llvm::GlobalVariable(TheModule,
1183 OptionalPropertyListInit->getType(), false,
1184 llvm::GlobalValue::InternalLinkage, OptionalPropertyListInit,
1185 ".objc_property_list");
1186
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001187 // Protocols are objects containing lists of the methods implemented and
1188 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +00001189 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001190 PtrToInt8Ty,
1191 ProtocolList->getType(),
1192 InstanceMethodList->getType(),
1193 ClassMethodList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001194 OptionalInstanceMethodList->getType(),
1195 OptionalClassMethodList->getType(),
1196 PropertyList->getType(),
1197 OptionalPropertyList->getType(),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001198 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +00001199 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001200 // The isa pointer must be set to a magic number so the runtime knows it's
1201 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001202 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
1203 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +00001204 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001205 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001206 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
1207 Elements.push_back(ProtocolList);
1208 Elements.push_back(InstanceMethodList);
1209 Elements.push_back(ClassMethodList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001210 Elements.push_back(OptionalInstanceMethodList);
1211 Elements.push_back(OptionalClassMethodList);
1212 Elements.push_back(PropertyList);
1213 Elements.push_back(OptionalPropertyList);
Mike Stump1eb44332009-09-09 15:08:12 +00001214 ExistingProtocols[ProtocolName] =
Owen Anderson3c4972d2009-07-29 18:54:39 +00001215 llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001216 ".objc_protocol"), IdTy);
1217}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001218void CGObjCGNU::GenerateProtocolHolderCategory(void) {
1219 // Collect information about instance methods
1220 llvm::SmallVector<Selector, 1> MethodSels;
1221 llvm::SmallVector<llvm::Constant*, 1> MethodTypes;
1222
1223 std::vector<llvm::Constant*> Elements;
1224 const std::string ClassName = "__ObjC_Protocol_Holder_Ugly_Hack";
1225 const std::string CategoryName = "AnotherHack";
1226 Elements.push_back(MakeConstantString(CategoryName));
1227 Elements.push_back(MakeConstantString(ClassName));
1228 // Instance method list
1229 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1230 ClassName, CategoryName, MethodSels, MethodTypes, false), PtrTy));
1231 // Class method list
1232 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1233 ClassName, CategoryName, MethodSels, MethodTypes, true), PtrTy));
1234 // Protocol list
1235 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrTy,
1236 ExistingProtocols.size());
1237 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
1238 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
1239 LongTy,//FIXME: Should be size_t
1240 ProtocolArrayTy,
1241 NULL);
1242 std::vector<llvm::Constant*> ProtocolElements;
1243 for (llvm::StringMapIterator<llvm::Constant*> iter =
1244 ExistingProtocols.begin(), endIter = ExistingProtocols.end();
1245 iter != endIter ; iter++) {
1246 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(iter->getValue(),
1247 PtrTy);
1248 ProtocolElements.push_back(Ptr);
1249 }
1250 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
1251 ProtocolElements);
1252 ProtocolElements.clear();
1253 ProtocolElements.push_back(NULLPtr);
1254 ProtocolElements.push_back(llvm::ConstantInt::get(LongTy,
1255 ExistingProtocols.size()));
1256 ProtocolElements.push_back(ProtocolArray);
1257 Elements.push_back(llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolListTy,
1258 ProtocolElements, ".objc_protocol_list"), PtrTy));
1259 Categories.push_back(llvm::ConstantExpr::getBitCast(
1260 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
1261 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
1262}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001263
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001264void CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001265 std::string ClassName = OCD->getClassInterface()->getNameAsString();
1266 std::string CategoryName = OCD->getNameAsString();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001267 // Collect information about instance methods
1268 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1269 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001270 for (ObjCCategoryImplDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001271 iter = OCD->instmeth_begin(), endIter = OCD->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001272 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001273 InstanceMethodSels.push_back((*iter)->getSelector());
1274 std::string TypeStr;
1275 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001276 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001277 }
1278
1279 // Collect information about class methods
1280 llvm::SmallVector<Selector, 16> ClassMethodSels;
1281 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001282 for (ObjCCategoryImplDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001283 iter = OCD->classmeth_begin(), endIter = OCD->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001284 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001285 ClassMethodSels.push_back((*iter)->getSelector());
1286 std::string TypeStr;
1287 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001288 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001289 }
1290
1291 // Collect the names of referenced protocols
1292 llvm::SmallVector<std::string, 16> Protocols;
David Chisnallad9e06d2010-03-13 22:20:45 +00001293 const ObjCCategoryDecl *CatDecl = OCD->getCategoryDecl();
1294 const ObjCList<ObjCProtocolDecl> &Protos = CatDecl->getReferencedProtocols();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001295 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1296 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001297 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001298
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001299 std::vector<llvm::Constant*> Elements;
1300 Elements.push_back(MakeConstantString(CategoryName));
1301 Elements.push_back(MakeConstantString(ClassName));
Mike Stump1eb44332009-09-09 15:08:12 +00001302 // Instance method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001303 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001304 ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001305 false), PtrTy));
1306 // Class method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001307 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001308 ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001309 PtrTy));
1310 // Protocol list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001311 Elements.push_back(llvm::ConstantExpr::getBitCast(
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001312 GenerateProtocolList(Protocols), PtrTy));
Owen Anderson3c4972d2009-07-29 18:54:39 +00001313 Categories.push_back(llvm::ConstantExpr::getBitCast(
Mike Stump1eb44332009-09-09 15:08:12 +00001314 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
Owen Anderson47a434f2009-08-05 23:18:46 +00001315 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001316}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001317
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001318llvm::Constant *CGObjCGNU::GeneratePropertyList(const ObjCImplementationDecl *OID,
1319 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
1320 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes) {
1321 ASTContext &Context = CGM.getContext();
1322 //
1323 // Property metadata: name, attributes, isSynthesized, setter name, setter
1324 // types, getter name, getter types.
1325 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1326 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1327 PtrToInt8Ty, NULL);
1328 std::vector<llvm::Constant*> Properties;
1329
1330
1331 // Add all of the property methods need adding to the method list and to the
1332 // property metadata list.
1333 for (ObjCImplDecl::propimpl_iterator
1334 iter = OID->propimpl_begin(), endIter = OID->propimpl_end();
1335 iter != endIter ; iter++) {
1336 std::vector<llvm::Constant*> Fields;
1337 ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
David Chisnall42ba04a2010-02-26 01:11:38 +00001338 ObjCPropertyImplDecl *propertyImpl = *iter;
1339 bool isSynthesized = (propertyImpl->getPropertyImplementation() ==
1340 ObjCPropertyImplDecl::Synthesize);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001341
1342 Fields.push_back(MakeConstantString(property->getNameAsString()));
1343 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1344 property->getPropertyAttributes()));
David Chisnall42ba04a2010-02-26 01:11:38 +00001345 Fields.push_back(llvm::ConstantInt::get(Int8Ty, isSynthesized));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001346 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001347 std::string TypeStr;
1348 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1349 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
David Chisnall42ba04a2010-02-26 01:11:38 +00001350 if (isSynthesized) {
1351 InstanceMethodTypes.push_back(TypeEncoding);
1352 InstanceMethodSels.push_back(getter->getSelector());
1353 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001354 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1355 Fields.push_back(TypeEncoding);
1356 } else {
1357 Fields.push_back(NULLPtr);
1358 Fields.push_back(NULLPtr);
1359 }
1360 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001361 std::string TypeStr;
1362 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1363 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
David Chisnall42ba04a2010-02-26 01:11:38 +00001364 if (isSynthesized) {
1365 InstanceMethodTypes.push_back(TypeEncoding);
1366 InstanceMethodSels.push_back(setter->getSelector());
1367 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001368 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1369 Fields.push_back(TypeEncoding);
1370 } else {
1371 Fields.push_back(NULLPtr);
1372 Fields.push_back(NULLPtr);
1373 }
1374 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1375 }
1376 llvm::ArrayType *PropertyArrayTy =
1377 llvm::ArrayType::get(PropertyMetadataTy, Properties.size());
1378 llvm::Constant *PropertyArray = llvm::ConstantArray::get(PropertyArrayTy,
1379 Properties);
1380 llvm::Constant* PropertyListInitFields[] =
1381 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1382
1383 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001384 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001385 return new llvm::GlobalVariable(TheModule, PropertyListInit->getType(), false,
1386 llvm::GlobalValue::InternalLinkage, PropertyListInit,
1387 ".objc_property_list");
1388}
1389
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001390void CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
1391 ASTContext &Context = CGM.getContext();
1392
1393 // Get the superclass name.
Mike Stump1eb44332009-09-09 15:08:12 +00001394 const ObjCInterfaceDecl * SuperClassDecl =
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001395 OID->getClassInterface()->getSuperClass();
Chris Lattner8ec03f52008-11-24 03:54:41 +00001396 std::string SuperClassName;
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001397 if (SuperClassDecl) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001398 SuperClassName = SuperClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001399 EmitClassRef(SuperClassName);
1400 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001401
1402 // Get the class name
Chris Lattner09dc6662009-04-01 02:00:48 +00001403 ObjCInterfaceDecl *ClassDecl =
1404 const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
Chris Lattner8ec03f52008-11-24 03:54:41 +00001405 std::string ClassName = ClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001406 // Emit the symbol that is used to generate linker errors if this class is
1407 // referenced in other modules but not declared.
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001408 std::string classSymbolName = "__objc_class_name_" + ClassName;
Mike Stump1eb44332009-09-09 15:08:12 +00001409 if (llvm::GlobalVariable *symbol =
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001410 TheModule.getGlobalVariable(classSymbolName)) {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001411 symbol->setInitializer(llvm::ConstantInt::get(LongTy, 0));
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001412 } else {
Owen Anderson1c431b32009-07-08 19:05:04 +00001413 new llvm::GlobalVariable(TheModule, LongTy, false,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001414 llvm::GlobalValue::ExternalLinkage, llvm::ConstantInt::get(LongTy, 0),
Owen Anderson1c431b32009-07-08 19:05:04 +00001415 classSymbolName);
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001416 }
Mike Stump1eb44332009-09-09 15:08:12 +00001417
Daniel Dunbar2bebbf02009-05-03 10:46:44 +00001418 // Get the size of instances.
1419 int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001420
1421 // Collect information about instance variables.
1422 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
1423 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
1424 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
Mike Stump1eb44332009-09-09 15:08:12 +00001425
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001426 std::vector<llvm::Constant*> IvarOffsetValues;
1427
Mike Stump1eb44332009-09-09 15:08:12 +00001428 int superInstanceSize = !SuperClassDecl ? 0 :
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001429 Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
1430 // For non-fragile ivars, set the instance size to 0 - {the size of just this
1431 // class}. The runtime will then set this to the correct value on load.
1432 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
1433 instanceSize = 0 - (instanceSize - superInstanceSize);
1434 }
David Chisnall7f63cb02010-04-19 00:45:34 +00001435
1436 // Collect declared and synthesized ivars.
1437 llvm::SmallVector<ObjCIvarDecl*, 16> OIvars;
1438 CGM.getContext().ShallowCollectObjCIvars(ClassDecl, OIvars);
1439
1440 for (unsigned i = 0, e = OIvars.size(); i != e; ++i) {
1441 ObjCIvarDecl *IVD = OIvars[i];
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001442 // Store the name
David Chisnall7f63cb02010-04-19 00:45:34 +00001443 IvarNames.push_back(MakeConstantString(IVD->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001444 // Get the type encoding for this ivar
1445 std::string TypeStr;
David Chisnall7f63cb02010-04-19 00:45:34 +00001446 Context.getObjCEncodingForType(IVD->getType(), TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001447 IvarTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001448 // Get the offset
David Chisnalld901da52010-04-19 01:37:25 +00001449 uint64_t BaseOffset = ComputeIvarBaseOffset(CGM, OID, IVD);
David Chisnallaecbf242009-11-17 19:32:15 +00001450 uint64_t Offset = BaseOffset;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001451 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001452 Offset = BaseOffset - superInstanceSize;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001453 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001454 IvarOffsets.push_back(
Owen Anderson0032b272009-08-13 21:57:51 +00001455 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001456 IvarOffsetValues.push_back(new llvm::GlobalVariable(TheModule, IntTy,
1457 false, llvm::GlobalValue::ExternalLinkage,
1458 llvm::ConstantInt::get(IntTy, BaseOffset),
1459 "__objc_ivar_offset_value_" + ClassName +"." +
David Chisnall7f63cb02010-04-19 00:45:34 +00001460 IVD->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001461 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001462 llvm::Constant *IvarOffsetArrayInit =
1463 llvm::ConstantArray::get(llvm::ArrayType::get(PtrToIntTy,
1464 IvarOffsetValues.size()), IvarOffsetValues);
1465 llvm::GlobalVariable *IvarOffsetArray = new llvm::GlobalVariable(TheModule,
1466 IvarOffsetArrayInit->getType(), false,
1467 llvm::GlobalValue::InternalLinkage, IvarOffsetArrayInit,
1468 ".ivar.offsets");
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001469
1470 // Collect information about instance methods
1471 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1472 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001473 for (ObjCImplementationDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001474 iter = OID->instmeth_begin(), endIter = OID->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001475 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001476 InstanceMethodSels.push_back((*iter)->getSelector());
1477 std::string TypeStr;
1478 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001479 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001480 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001481
1482 llvm::Constant *Properties = GeneratePropertyList(OID, InstanceMethodSels,
1483 InstanceMethodTypes);
1484
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001485
1486 // Collect information about class methods
1487 llvm::SmallVector<Selector, 16> ClassMethodSels;
1488 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001489 for (ObjCImplementationDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001490 iter = OID->classmeth_begin(), endIter = OID->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001491 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001492 ClassMethodSels.push_back((*iter)->getSelector());
1493 std::string TypeStr;
1494 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001495 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001496 }
1497 // Collect the names of referenced protocols
1498 llvm::SmallVector<std::string, 16> Protocols;
1499 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1500 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1501 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001502 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001503
1504
1505
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001506 // Get the superclass pointer.
1507 llvm::Constant *SuperClass;
Chris Lattner8ec03f52008-11-24 03:54:41 +00001508 if (!SuperClassName.empty()) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001509 SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
1510 } else {
Owen Anderson03e20502009-07-30 23:11:26 +00001511 SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001512 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001513 // Empty vector used to construct empty method lists
1514 llvm::SmallVector<llvm::Constant*, 1> empty;
1515 // Generate the method and instance variable lists
1516 llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001517 InstanceMethodSels, InstanceMethodTypes, false);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001518 llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001519 ClassMethodSels, ClassMethodTypes, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001520 llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
1521 IvarOffsets);
Mike Stump1eb44332009-09-09 15:08:12 +00001522 // Irrespective of whether we are compiling for a fragile or non-fragile ABI,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001523 // we emit a symbol containing the offset for each ivar in the class. This
1524 // allows code compiled for the non-Fragile ABI to inherit from code compiled
1525 // for the legacy ABI, without causing problems. The converse is also
1526 // possible, but causes all ivar accesses to be fragile.
1527 int i = 0;
1528 // Offset pointer for getting at the correct field in the ivar list when
1529 // setting up the alias. These are: The base address for the global, the
1530 // ivar array (second field), the ivar in this list (set for each ivar), and
1531 // the offset (third field in ivar structure)
1532 const llvm::Type *IndexTy = llvm::Type::getInt32Ty(VMContext);
1533 llvm::Constant *offsetPointerIndexes[] = {Zeros[0],
Mike Stump1eb44332009-09-09 15:08:12 +00001534 llvm::ConstantInt::get(IndexTy, 1), 0,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001535 llvm::ConstantInt::get(IndexTy, 2) };
1536
1537 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1538 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1539 const std::string Name = "__objc_ivar_offset_" + ClassName + '.'
1540 +(*iter)->getNameAsString();
1541 offsetPointerIndexes[2] = llvm::ConstantInt::get(IndexTy, i++);
1542 // Get the correct ivar field
1543 llvm::Constant *offsetValue = llvm::ConstantExpr::getGetElementPtr(
1544 IvarList, offsetPointerIndexes, 4);
1545 // Get the existing alias, if one exists.
1546 llvm::GlobalVariable *offset = TheModule.getNamedGlobal(Name);
1547 if (offset) {
1548 offset->setInitializer(offsetValue);
1549 // If this is the real definition, change its linkage type so that
1550 // different modules will use this one, rather than their private
1551 // copy.
1552 offset->setLinkage(llvm::GlobalValue::ExternalLinkage);
1553 } else {
1554 // Add a new alias if there isn't one already.
1555 offset = new llvm::GlobalVariable(TheModule, offsetValue->getType(),
1556 false, llvm::GlobalValue::ExternalLinkage, offsetValue, Name);
1557 }
1558 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001559 //Generate metaclass for class methods
1560 llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
David Chisnall18044632009-11-16 19:05:54 +00001561 NULLPtr, 0x12L, ClassName.c_str(), 0, Zeros[0], GenerateIvarList(
David Chisnall8c757f92010-04-28 14:29:56 +00001562 empty, empty, empty), ClassMethodList, NULLPtr, NULLPtr, NULLPtr, true);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001563
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001564 // Generate the class structure
Chris Lattner8ec03f52008-11-24 03:54:41 +00001565 llvm::Constant *ClassStruct =
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001566 GenerateClassStructure(MetaClassStruct, SuperClass, 0x11L,
Chris Lattner8ec03f52008-11-24 03:54:41 +00001567 ClassName.c_str(), 0,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001568 llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001569 MethodList, GenerateProtocolList(Protocols), IvarOffsetArray,
1570 Properties);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001571
1572 // Resolve the class aliases, if they exist.
1573 if (ClassPtrAlias) {
1574 ClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001575 llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001576 ClassPtrAlias = 0;
1577 }
1578 if (MetaClassPtrAlias) {
1579 MetaClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001580 llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001581 MetaClassPtrAlias = 0;
1582 }
1583
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001584 // Add class structure to list to be added to the symtab later
Owen Anderson3c4972d2009-07-29 18:54:39 +00001585 ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001586 Classes.push_back(ClassStruct);
1587}
1588
Fariborz Jahanianc38e9af2009-06-23 21:47:46 +00001589
Mike Stump1eb44332009-09-09 15:08:12 +00001590llvm::Function *CGObjCGNU::ModuleInitFunction() {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001591 // Only emit an ObjC load function if no Objective-C stuff has been called
1592 if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
1593 ExistingProtocols.empty() && TypedSelectors.empty() &&
Anton Korobeynikov5f58b912008-06-01 15:14:46 +00001594 UntypedSelectors.empty())
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001595 return NULL;
Eli Friedman1b8956e2008-06-01 16:00:02 +00001596
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001597 // Add all referenced protocols to a category.
1598 GenerateProtocolHolderCategory();
1599
Chris Lattnere160c9b2009-01-27 05:06:01 +00001600 const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1601 SelectorTy->getElementType());
1602 const llvm::Type *SelStructPtrTy = SelectorTy;
1603 bool isSelOpaque = false;
1604 if (SelStructTy == 0) {
Owen Anderson47a434f2009-08-05 23:18:46 +00001605 SelStructTy = llvm::StructType::get(VMContext, PtrToInt8Ty,
1606 PtrToInt8Ty, NULL);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001607 SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
Chris Lattnere160c9b2009-01-27 05:06:01 +00001608 isSelOpaque = true;
1609 }
1610
Eli Friedman1b8956e2008-06-01 16:00:02 +00001611 // Name the ObjC types to make the IR a bit easier to read
Chris Lattnere160c9b2009-01-27 05:06:01 +00001612 TheModule.addTypeName(".objc_selector", SelStructPtrTy);
Eli Friedman1b8956e2008-06-01 16:00:02 +00001613 TheModule.addTypeName(".objc_id", IdTy);
1614 TheModule.addTypeName(".objc_imp", IMPTy);
1615
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001616 std::vector<llvm::Constant*> Elements;
Chris Lattner71238f62009-04-25 23:19:45 +00001617 llvm::Constant *Statics = NULLPtr;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001618 // Generate statics list:
Chris Lattner71238f62009-04-25 23:19:45 +00001619 if (ConstantStrings.size()) {
Owen Anderson96e0fc72009-07-29 22:16:19 +00001620 llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Chris Lattner71238f62009-04-25 23:19:45 +00001621 ConstantStrings.size() + 1);
1622 ConstantStrings.push_back(NULLPtr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001623
Daniel Dunbar1b096952009-11-29 02:38:47 +00001624 llvm::StringRef StringClass = CGM.getLangOptions().ObjCConstantStringClass;
1625 if (StringClass.empty()) StringClass = "NXConstantString";
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001626 Elements.push_back(MakeConstantString(StringClass,
1627 ".objc_static_class_name"));
Owen Anderson7db6d832009-07-28 18:33:04 +00001628 Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
Chris Lattner71238f62009-04-25 23:19:45 +00001629 ConstantStrings));
Mike Stump1eb44332009-09-09 15:08:12 +00001630 llvm::StructType *StaticsListTy =
Owen Anderson47a434f2009-08-05 23:18:46 +00001631 llvm::StructType::get(VMContext, PtrToInt8Ty, StaticsArrayTy, NULL);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001632 llvm::Type *StaticsListPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001633 llvm::PointerType::getUnqual(StaticsListTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001634 Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
Mike Stump1eb44332009-09-09 15:08:12 +00001635 llvm::ArrayType *StaticsListArrayTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001636 llvm::ArrayType::get(StaticsListPtrTy, 2);
Chris Lattner71238f62009-04-25 23:19:45 +00001637 Elements.clear();
1638 Elements.push_back(Statics);
Owen Andersonc9c88b42009-07-31 20:28:54 +00001639 Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
Chris Lattner71238f62009-04-25 23:19:45 +00001640 Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001641 Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001642 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001643 // Array of classes, categories, and constant objects
Owen Anderson96e0fc72009-07-29 22:16:19 +00001644 llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001645 Classes.size() + Categories.size() + 2);
Mike Stump1eb44332009-09-09 15:08:12 +00001646 llvm::StructType *SymTabTy = llvm::StructType::get(VMContext,
Owen Anderson47a434f2009-08-05 23:18:46 +00001647 LongTy, SelStructPtrTy,
Owen Anderson0032b272009-08-13 21:57:51 +00001648 llvm::Type::getInt16Ty(VMContext),
1649 llvm::Type::getInt16Ty(VMContext),
Chris Lattner630404b2008-06-26 04:10:42 +00001650 ClassListTy, NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001651
1652 Elements.clear();
1653 // Pointer to an array of selectors used in this module.
1654 std::vector<llvm::Constant*> Selectors;
1655 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1656 iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
1657 iter != iterEnd ; ++iter) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001658 Elements.push_back(ExportUniqueString(iter->first.first, ".objc_sel_name"));
David Chisnalla7c5b082009-09-14 19:04:10 +00001659 Elements.push_back(MakeConstantString(iter->first.second,
Chris Lattner630404b2008-06-26 04:10:42 +00001660 ".objc_sel_types"));
Owen Anderson08e25242009-07-27 22:29:56 +00001661 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001662 Elements.clear();
1663 }
1664 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1665 iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
Chris Lattner630404b2008-06-26 04:10:42 +00001666 iter != iterEnd; ++iter) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001667 Elements.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001668 ExportUniqueString(iter->getKeyData(), ".objc_sel_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001669 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001670 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001671 Elements.clear();
1672 }
1673 Elements.push_back(NULLPtr);
1674 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001675 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001676 Elements.clear();
1677 // Number of static selectors
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001678 Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001679 llvm::Constant *SelectorList = MakeGlobal(
Owen Anderson96e0fc72009-07-29 22:16:19 +00001680 llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001681 ".objc_selector_list");
Mike Stump1eb44332009-09-09 15:08:12 +00001682 Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
Chris Lattnere160c9b2009-01-27 05:06:01 +00001683 SelStructPtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001684
1685 // Now that all of the static selectors exist, create pointers to them.
1686 int index = 0;
1687 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1688 iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
1689 iter != iterEnd; ++iter) {
1690 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001691 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001692 llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
David Chisnall22b88272010-05-09 01:01:43 +00001693 true, llvm::GlobalValue::LinkOnceODRLinkage,
1694 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1695 MangleSelectorTypes(".objc_sel_ptr"+iter->first.first+"."+
1696 iter->first.second));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001697 // If selectors are defined as an opaque type, cast the pointer to this
1698 // type.
1699 if (isSelOpaque) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001700 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1701 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001702 }
David Chisnall87935a82010-05-08 20:58:05 +00001703 (*iter).second->replaceAllUsesWith(SelPtr);
1704 (*iter).second->eraseFromParent();
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001705 }
1706 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1707 iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1708 iter != iterEnd; iter++) {
1709 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001710 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
David Chisnall87935a82010-05-08 20:58:05 +00001711 llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
David Chisnall22b88272010-05-09 01:01:43 +00001712 true, llvm::GlobalValue::LinkOnceODRLinkage,
1713 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1714 MangleSelectorTypes(std::string(".objc_sel_ptr")+iter->getKey().str()));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001715 // If selectors are defined as an opaque type, cast the pointer to this
1716 // type.
1717 if (isSelOpaque) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001718 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1719 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001720 }
David Chisnall87935a82010-05-08 20:58:05 +00001721 (*iter).second->replaceAllUsesWith(SelPtr);
1722 (*iter).second->eraseFromParent();
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001723 }
1724 // Number of classes defined.
Mike Stump1eb44332009-09-09 15:08:12 +00001725 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001726 Classes.size()));
1727 // Number of categories defined
Mike Stump1eb44332009-09-09 15:08:12 +00001728 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001729 Categories.size()));
1730 // Create an array of classes, then categories, then static object instances
1731 Classes.insert(Classes.end(), Categories.begin(), Categories.end());
1732 // NULL-terminated list of static object instances (mainly constant strings)
1733 Classes.push_back(Statics);
1734 Classes.push_back(NULLPtr);
Owen Anderson7db6d832009-07-28 18:33:04 +00001735 llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001736 Elements.push_back(ClassList);
Mike Stump1eb44332009-09-09 15:08:12 +00001737 // Construct the symbol table
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001738 llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
1739
1740 // The symbol table is contained in a module which has some version-checking
1741 // constants
Owen Anderson47a434f2009-08-05 23:18:46 +00001742 llvm::StructType * ModuleTy = llvm::StructType::get(VMContext, LongTy, LongTy,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001743 PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001744 Elements.clear();
1745 // Runtime version used for compatibility checking.
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001746 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Mike Stump1eb44332009-09-09 15:08:12 +00001747 Elements.push_back(llvm::ConstantInt::get(LongTy,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001748 NonFragileRuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001749 } else {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001750 Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001751 }
Fariborz Jahanian91a0b512009-04-01 19:49:42 +00001752 // sizeof(ModuleTy)
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +00001753 llvm::TargetData td(&TheModule);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001754 Elements.push_back(llvm::ConstantInt::get(LongTy,
Owen Andersona1cf15f2009-07-14 23:10:40 +00001755 td.getTypeSizeInBits(ModuleTy)/8));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001756 //FIXME: Should be the path to the file where this module was declared
1757 Elements.push_back(NULLPtr);
1758 Elements.push_back(SymTab);
1759 llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
1760
1761 // Create the load function calling the runtime entry point with the module
1762 // structure
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001763 llvm::Function * LoadFunction = llvm::Function::Create(
Owen Anderson0032b272009-08-13 21:57:51 +00001764 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), false),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001765 llvm::GlobalValue::InternalLinkage, ".objc_load_function",
1766 &TheModule);
Owen Anderson0032b272009-08-13 21:57:51 +00001767 llvm::BasicBlock *EntryBB =
1768 llvm::BasicBlock::Create(VMContext, "entry", LoadFunction);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001769 CGBuilderTy Builder(VMContext);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001770 Builder.SetInsertPoint(EntryBB);
Fariborz Jahanian26c82942009-03-30 18:02:14 +00001771
1772 std::vector<const llvm::Type*> Params(1,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001773 llvm::PointerType::getUnqual(ModuleTy));
1774 llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Owen Anderson0032b272009-08-13 21:57:51 +00001775 llvm::Type::getVoidTy(VMContext), Params, true), "__objc_exec_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001776 Builder.CreateCall(Register, Module);
1777 Builder.CreateRetVoid();
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001778
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001779 return LoadFunction;
1780}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001781
Fariborz Jahanian679a5022009-01-10 21:06:09 +00001782llvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
Mike Stump1eb44332009-09-09 15:08:12 +00001783 const ObjCContainerDecl *CD) {
1784 const ObjCCategoryImplDecl *OCD =
Steve Naroff3e0a5402009-01-08 19:41:02 +00001785 dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
Chris Lattner077bf5e2008-11-24 03:33:13 +00001786 std::string CategoryName = OCD ? OCD->getNameAsString() : "";
David Chisnall4c8c8e92010-03-20 19:53:29 +00001787 std::string ClassName = CD->getName();
Chris Lattner077bf5e2008-11-24 03:33:13 +00001788 std::string MethodName = OMD->getSelector().getAsString();
Douglas Gregorf8d49f62009-01-09 17:18:27 +00001789 bool isClassMethod = !OMD->isInstanceMethod();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001790
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001791 CodeGenTypes &Types = CGM.getTypes();
Mike Stump1eb44332009-09-09 15:08:12 +00001792 const llvm::FunctionType *MethodTy =
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001793 Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001794 std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
1795 MethodName, isClassMethod);
1796
Daniel Dunbard6c93d72009-09-17 04:01:22 +00001797 llvm::Function *Method
Mike Stump1eb44332009-09-09 15:08:12 +00001798 = llvm::Function::Create(MethodTy,
1799 llvm::GlobalValue::InternalLinkage,
1800 FunctionName,
1801 &TheModule);
Chris Lattner391d77a2008-03-30 23:03:07 +00001802 return Method;
1803}
1804
Daniel Dunbar49f66022008-09-24 03:38:44 +00001805llvm::Function *CGObjCGNU::GetPropertyGetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001806 std::vector<const llvm::Type*> Params;
1807 const llvm::Type *BoolTy =
1808 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1809 Params.push_back(IdTy);
1810 Params.push_back(SelectorTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001811 Params.push_back(IntTy);
Mike Stump1eb44332009-09-09 15:08:12 +00001812 Params.push_back(BoolTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001813 // void objc_getProperty (id, SEL, int, bool)
Mike Stump1eb44332009-09-09 15:08:12 +00001814 const llvm::FunctionType *FTy =
1815 llvm::FunctionType::get(IdTy, Params, false);
1816 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1817 "objc_getProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001818}
1819
1820llvm::Function *CGObjCGNU::GetPropertySetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001821 std::vector<const llvm::Type*> Params;
1822 const llvm::Type *BoolTy =
1823 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1824 Params.push_back(IdTy);
1825 Params.push_back(SelectorTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001826 Params.push_back(IntTy);
Mike Stump1eb44332009-09-09 15:08:12 +00001827 Params.push_back(IdTy);
1828 Params.push_back(BoolTy);
1829 Params.push_back(BoolTy);
David Chisnall7f63cb02010-04-19 00:45:34 +00001830 // void objc_setProperty (id, SEL, int, id, bool, bool)
Mike Stump1eb44332009-09-09 15:08:12 +00001831 const llvm::FunctionType *FTy =
1832 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Params, false);
1833 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1834 "objc_setProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001835}
1836
Fariborz Jahanian6cc59062010-04-12 18:18:10 +00001837// FIXME. Implement this.
1838llvm::Function *CGObjCGNU::GetCopyStructFunction() {
1839 return 0;
1840}
1841
Daniel Dunbar309a4362009-07-24 07:40:24 +00001842llvm::Constant *CGObjCGNU::EnumerationMutationFunction() {
1843 CodeGen::CodeGenTypes &Types = CGM.getTypes();
1844 ASTContext &Ctx = CGM.getContext();
1845 // void objc_enumerationMutation (id)
John McCallead608a2010-02-26 00:48:12 +00001846 llvm::SmallVector<CanQualType,1> Params;
David Chisnall0f436562009-08-17 16:35:33 +00001847 Params.push_back(ASTIdTy);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001848 const llvm::FunctionType *FTy =
John McCall04a67a62010-02-05 21:31:56 +00001849 Types.GetFunctionType(Types.getFunctionInfo(Ctx.VoidTy, Params,
Rafael Espindola264ba482010-03-30 20:24:48 +00001850 FunctionType::ExtInfo()), false);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001851 return CGM.CreateRuntimeFunction(FTy, "objc_enumerationMutation");
Anders Carlsson2abd89c2008-08-31 04:05:03 +00001852}
1853
Fariborz Jahanianbd71be42008-11-21 00:49:24 +00001854void CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1855 const Stmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001856 // Pointer to the personality function
1857 llvm::Constant *Personality =
Owen Anderson0032b272009-08-13 21:57:51 +00001858 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getInt32Ty(VMContext),
Chris Lattnerb59761b2009-07-01 04:13:52 +00001859 true),
Chris Lattner5dc08672009-05-08 00:11:50 +00001860 "__gnu_objc_personality_v0");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001861 Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
Chris Lattner5dc08672009-05-08 00:11:50 +00001862 std::vector<const llvm::Type*> Params;
1863 Params.push_back(PtrTy);
1864 llvm::Value *RethrowFn =
Owen Anderson0032b272009-08-13 21:57:51 +00001865 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
David Chisnall90665bd2010-01-06 18:02:59 +00001866 Params, false), "_Unwind_Resume");
Chris Lattner5dc08672009-05-08 00:11:50 +00001867
1868 bool isTry = isa<ObjCAtTryStmt>(S);
1869 llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
1870 llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
1871 llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
Chris Lattnerd6a99072009-05-11 18:16:28 +00001872 llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
Chris Lattner5dc08672009-05-08 00:11:50 +00001873 llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
1874 llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
1875 llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
1876
David Chisnall0faa5162009-12-24 02:26:34 +00001877 // @synchronized()
Chris Lattner5dc08672009-05-08 00:11:50 +00001878 if (!isTry) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001879 std::vector<const llvm::Type*> Args(1, IdTy);
1880 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001881 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001882 llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
Mike Stump1eb44332009-09-09 15:08:12 +00001883 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001884 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1885 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1886 CGF.Builder.CreateCall(SyncEnter, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001887 }
1888
Chris Lattnerd6a99072009-05-11 18:16:28 +00001889
Chris Lattner5dc08672009-05-08 00:11:50 +00001890 // Push an EH context entry, used for handling rethrows and jumps
1891 // through finally.
1892 CGF.PushCleanupBlock(FinallyBlock);
1893
1894 // Emit the statements in the @try {} block
1895 CGF.setInvokeDest(TryHandler);
1896
1897 CGF.EmitBlock(TryBlock);
Mike Stump1eb44332009-09-09 15:08:12 +00001898 CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
Chris Lattner5dc08672009-05-08 00:11:50 +00001899 : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
1900
1901 // Jump to @finally if there is no exception
1902 CGF.EmitBranchThroughCleanup(FinallyEnd);
1903
1904 // Emit the handlers
1905 CGF.EmitBlock(TryHandler);
1906
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001907 // Get the correct versions of the exception handling intrinsics
Mike Stump1eb44332009-09-09 15:08:12 +00001908 llvm::Value *llvm_eh_exception =
Chris Lattner5dc08672009-05-08 00:11:50 +00001909 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
Duncan Sands790b20e2009-10-14 16:13:30 +00001910 llvm::Value *llvm_eh_selector =
1911 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector);
1912 llvm::Value *llvm_eh_typeid_for =
1913 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for);
Chris Lattner5dc08672009-05-08 00:11:50 +00001914
1915 // Exception object
1916 llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1917 llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
1918
1919 llvm::SmallVector<llvm::Value*, 8> ESelArgs;
Douglas Gregorc00d8e12010-04-26 16:46:50 +00001920 llvm::SmallVector<std::pair<const VarDecl*, const Stmt*>, 8> Handlers;
Chris Lattner5dc08672009-05-08 00:11:50 +00001921
1922 ESelArgs.push_back(Exc);
1923 ESelArgs.push_back(Personality);
1924
1925 bool HasCatchAll = false;
1926 // Only @try blocks are allowed @catch blocks, but both can have @finally
1927 if (isTry) {
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +00001928 if (cast<ObjCAtTryStmt>(S).getNumCatchStmts()) {
1929 const ObjCAtTryStmt &AtTry = cast<ObjCAtTryStmt>(S);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001930 CGF.setInvokeDest(CatchInCatch);
Chris Lattner5dc08672009-05-08 00:11:50 +00001931
Douglas Gregor8f5e3dd2010-04-23 22:50:49 +00001932 for (unsigned I = 0, N = AtTry.getNumCatchStmts(); I != N; ++I) {
1933 const ObjCAtCatchStmt *CatchStmt = AtTry.getCatchStmt(I);
Douglas Gregorc00d8e12010-04-26 16:46:50 +00001934 const VarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
Mike Stumpbb1c8602009-07-31 21:31:32 +00001935 Handlers.push_back(std::make_pair(CatchDecl,
1936 CatchStmt->getCatchBody()));
Chris Lattner5dc08672009-05-08 00:11:50 +00001937
1938 // @catch() and @catch(id) both catch any ObjC exception
Steve Naroff14108da2009-07-10 23:34:53 +00001939 if (!CatchDecl || CatchDecl->getType()->isObjCIdType()
Chris Lattner5dc08672009-05-08 00:11:50 +00001940 || CatchDecl->getType()->isObjCQualifiedIdType()) {
1941 // Use i8* null here to signal this is a catch all, not a cleanup.
1942 ESelArgs.push_back(NULLPtr);
1943 HasCatchAll = true;
1944 // No further catches after this one will ever by reached
1945 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001946 }
Chris Lattner5dc08672009-05-08 00:11:50 +00001947
1948 // All other types should be Objective-C interface pointer types.
Mike Stump1eb44332009-09-09 15:08:12 +00001949 const ObjCObjectPointerType *OPT =
John McCall183700f2009-09-21 23:43:11 +00001950 CatchDecl->getType()->getAs<ObjCObjectPointerType>();
Steve Naroff14108da2009-07-10 23:34:53 +00001951 assert(OPT && "Invalid @catch type.");
John McCall506b57e2010-05-17 21:00:27 +00001952 const ObjCInterfaceDecl *IDecl =
1953 OPT->getObjectType()->getInterface();
1954 assert(IDecl && "Invalid @catch type.");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001955 llvm::Value *EHType =
John McCall506b57e2010-05-17 21:00:27 +00001956 MakeConstantString(IDecl->getNameAsString());
Chris Lattner5dc08672009-05-08 00:11:50 +00001957 ESelArgs.push_back(EHType);
1958 }
1959 }
1960 }
1961
1962 // We use a cleanup unless there was already a catch all.
1963 if (!HasCatchAll) {
Owen Anderson0032b272009-08-13 21:57:51 +00001964 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattner5dc08672009-05-08 00:11:50 +00001965 Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
1966 }
1967
1968 // Find which handler was matched.
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001969 llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
Chris Lattner5dc08672009-05-08 00:11:50 +00001970 ESelArgs.begin(), ESelArgs.end(), "selector");
1971
1972 for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
Douglas Gregorc00d8e12010-04-26 16:46:50 +00001973 const VarDecl *CatchParam = Handlers[i].first;
Chris Lattner5dc08672009-05-08 00:11:50 +00001974 const Stmt *CatchBody = Handlers[i].second;
1975
1976 llvm::BasicBlock *Next = 0;
1977
1978 // The last handler always matches.
1979 if (i + 1 != e) {
1980 assert(CatchParam && "Only last handler can be a catch all.");
1981
1982 // Test whether this block matches the type for the selector and branch
1983 // to Match if it does, or to the next BB if it doesn't.
1984 llvm::BasicBlock *Match = CGF.createBasicBlock("match");
1985 Next = CGF.createBasicBlock("catch.next");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001986 llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
Chris Lattner5dc08672009-05-08 00:11:50 +00001987 CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
1988 CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
1989 Next);
1990
1991 CGF.EmitBlock(Match);
1992 }
Mike Stump1eb44332009-09-09 15:08:12 +00001993
Chris Lattner5dc08672009-05-08 00:11:50 +00001994 if (CatchBody) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001995 llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
1996 CGF.ConvertType(CatchParam->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001997
Chris Lattner5dc08672009-05-08 00:11:50 +00001998 // Bind the catch parameter if it exists.
1999 if (CatchParam) {
2000 // CatchParam is a ParmVarDecl because of the grammar
2001 // construction used to handle this, but for codegen purposes
2002 // we treat this as a local decl.
2003 CGF.EmitLocalBlockVarDecl(*CatchParam);
2004 CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
2005 }
2006
2007 CGF.ObjCEHValueStack.push_back(ExcObject);
2008 CGF.EmitStmt(CatchBody);
2009 CGF.ObjCEHValueStack.pop_back();
2010
2011 CGF.EmitBranchThroughCleanup(FinallyEnd);
2012
2013 if (Next)
2014 CGF.EmitBlock(Next);
2015 } else {
2016 assert(!Next && "catchup should be last handler.");
2017
2018 CGF.Builder.CreateStore(Exc, RethrowPtr);
2019 CGF.EmitBranchThroughCleanup(FinallyRethrow);
2020 }
2021 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00002022 // The @finally block is a secondary landing pad for any exceptions thrown in
2023 // @catch() blocks
2024 CGF.EmitBlock(CatchInCatch);
2025 Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
2026 ESelArgs.clear();
2027 ESelArgs.push_back(Exc);
2028 ESelArgs.push_back(Personality);
David Chisnall0faa5162009-12-24 02:26:34 +00002029 // If there is a @catch or @finally clause in outside of this one then we
2030 // need to make sure that we catch and rethrow it.
2031 if (PrevLandingPad) {
2032 ESelArgs.push_back(NULLPtr);
2033 } else {
2034 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
2035 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00002036 CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
2037 "selector");
2038 CGF.Builder.CreateCall(llvm_eh_typeid_for,
2039 CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
2040 CGF.Builder.CreateStore(Exc, RethrowPtr);
2041 CGF.EmitBranchThroughCleanup(FinallyRethrow);
2042
Chris Lattner5dc08672009-05-08 00:11:50 +00002043 CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
2044
2045 CGF.setInvokeDest(PrevLandingPad);
2046
2047 CGF.EmitBlock(FinallyBlock);
2048
Chris Lattnerd6a99072009-05-11 18:16:28 +00002049
Chris Lattner5dc08672009-05-08 00:11:50 +00002050 if (isTry) {
Mike Stump1eb44332009-09-09 15:08:12 +00002051 if (const ObjCAtFinallyStmt* FinallyStmt =
Chris Lattner5dc08672009-05-08 00:11:50 +00002052 cast<ObjCAtTryStmt>(S).getFinallyStmt())
2053 CGF.EmitStmt(FinallyStmt->getFinallyBody());
2054 } else {
Chris Lattnerd6a99072009-05-11 18:16:28 +00002055 // Emit 'objc_sync_exit(expr)' as finally's sole statement for
Chris Lattner5dc08672009-05-08 00:11:50 +00002056 // @synchronized.
Chris Lattnerd6a99072009-05-11 18:16:28 +00002057 std::vector<const llvm::Type*> Args(1, IdTy);
2058 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00002059 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00002060 llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
Mike Stump1eb44332009-09-09 15:08:12 +00002061 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00002062 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
2063 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
2064 CGF.Builder.CreateCall(SyncExit, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00002065 }
2066
2067 if (Info.SwitchBlock)
2068 CGF.EmitBlock(Info.SwitchBlock);
2069 if (Info.EndBlock)
2070 CGF.EmitBlock(Info.EndBlock);
2071
2072 // Branch around the rethrow code.
2073 CGF.EmitBranch(FinallyEnd);
2074
2075 CGF.EmitBlock(FinallyRethrow);
David Chisnall0faa5162009-12-24 02:26:34 +00002076
2077 llvm::Value *ExceptionObject = CGF.Builder.CreateLoad(RethrowPtr);
2078 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
2079 if (!UnwindBB) {
2080 CGF.Builder.CreateCall(RethrowFn, ExceptionObject);
2081 // Exception always thrown, next instruction is never reached.
2082 CGF.Builder.CreateUnreachable();
2083 } else {
2084 // If there is a @catch block outside this scope, we invoke instead of
2085 // calling because we may return to this function. This is very slow, but
2086 // some people still do it. It would be nice to add an optimised path for
2087 // this.
2088 CGF.Builder.CreateInvoke(RethrowFn, UnwindBB, UnwindBB, &ExceptionObject,
2089 &ExceptionObject+1);
2090 }
Mike Stump1eb44332009-09-09 15:08:12 +00002091
Chris Lattner5dc08672009-05-08 00:11:50 +00002092 CGF.EmitBlock(FinallyEnd);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00002093}
2094
2095void CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar49f66022008-09-24 03:38:44 +00002096 const ObjCAtThrowStmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00002097 llvm::Value *ExceptionAsObject;
2098
2099 std::vector<const llvm::Type*> Args(1, IdTy);
2100 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00002101 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Mike Stump1eb44332009-09-09 15:08:12 +00002102 llvm::Value *ThrowFn =
Chris Lattner5dc08672009-05-08 00:11:50 +00002103 CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
Mike Stump1eb44332009-09-09 15:08:12 +00002104
Chris Lattner5dc08672009-05-08 00:11:50 +00002105 if (const Expr *ThrowExpr = S.getThrowExpr()) {
2106 llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00002107 ExceptionAsObject = Exception;
Chris Lattner5dc08672009-05-08 00:11:50 +00002108 } else {
Mike Stump1eb44332009-09-09 15:08:12 +00002109 assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
Chris Lattner5dc08672009-05-08 00:11:50 +00002110 "Unexpected rethrow outside @catch block.");
2111 ExceptionAsObject = CGF.ObjCEHValueStack.back();
2112 }
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00002113 ExceptionAsObject =
2114 CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
Mike Stump1eb44332009-09-09 15:08:12 +00002115
Chris Lattner5dc08672009-05-08 00:11:50 +00002116 // Note: This may have to be an invoke, if we want to support constructs like:
2117 // @try {
2118 // @throw(obj);
2119 // }
2120 // @catch(id) ...
2121 //
2122 // This is effectively turning @throw into an incredibly-expensive goto, but
2123 // it may happen as a result of inlining followed by missed optimizations, or
2124 // as a result of stupidity.
2125 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
2126 if (!UnwindBB) {
2127 CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
2128 CGF.Builder.CreateUnreachable();
2129 } else {
2130 CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
2131 &ExceptionAsObject+1);
2132 }
2133 // Clear the insertion point to indicate we are in unreachable code.
2134 CGF.Builder.ClearInsertionPoint();
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00002135}
2136
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00002137llvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002138 llvm::Value *AddrWeakObj) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002139 CGBuilderTy B = CGF.Builder;
2140 AddrWeakObj = EnforceType(B, AddrWeakObj, IdTy);
2141 return B.CreateCall(WeakReadFn, AddrWeakObj);
Fariborz Jahanian6dc23172008-11-18 21:45:40 +00002142}
2143
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00002144void CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002145 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002146 CGBuilderTy B = CGF.Builder;
2147 src = EnforceType(B, src, IdTy);
2148 dst = EnforceType(B, dst, PtrToIdTy);
2149 B.CreateCall2(WeakAssignFn, src, dst);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00002150}
2151
Fariborz Jahanian58626502008-11-19 00:59:10 +00002152void CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002153 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002154 CGBuilderTy B = CGF.Builder;
2155 src = EnforceType(B, src, IdTy);
2156 dst = EnforceType(B, dst, PtrToIdTy);
2157 B.CreateCall2(GlobalAssignFn, src, dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +00002158}
2159
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00002160void CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +00002161 llvm::Value *src, llvm::Value *dst,
2162 llvm::Value *ivarOffset) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002163 CGBuilderTy B = CGF.Builder;
2164 src = EnforceType(B, src, IdTy);
2165 dst = EnforceType(B, dst, PtrToIdTy);
2166 B.CreateCall3(IvarAssignFn, src, dst, ivarOffset);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00002167}
2168
Fariborz Jahanian58626502008-11-19 00:59:10 +00002169void CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002170 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002171 CGBuilderTy B = CGF.Builder;
2172 src = EnforceType(B, src, IdTy);
2173 dst = EnforceType(B, dst, PtrToIdTy);
2174 B.CreateCall2(StrongCastAssignFn, src, dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +00002175}
2176
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002177void CGObjCGNU::EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002178 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002179 llvm::Value *SrcPtr,
Fariborz Jahanian55bcace2010-06-15 22:44:06 +00002180 llvm::Value *Size) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002181 CGBuilderTy B = CGF.Builder;
2182 DestPtr = EnforceType(B, DestPtr, IdTy);
2183 SrcPtr = EnforceType(B, SrcPtr, PtrToIdTy);
2184
Fariborz Jahanian55bcace2010-06-15 22:44:06 +00002185 B.CreateCall3(MemMoveFn, DestPtr, SrcPtr, Size);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002186}
2187
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002188llvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
2189 const ObjCInterfaceDecl *ID,
2190 const ObjCIvarDecl *Ivar) {
2191 const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
2192 + '.' + Ivar->getNameAsString();
2193 // Emit the variable and initialize it with what we think the correct value
2194 // is. This allows code compiled with non-fragile ivars to work correctly
2195 // when linked against code which isn't (most of the time).
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002196 llvm::GlobalVariable *IvarOffsetPointer = TheModule.getNamedGlobal(Name);
2197 if (!IvarOffsetPointer) {
David Chisnalld901da52010-04-19 01:37:25 +00002198 uint64_t Offset;
2199 if (ObjCImplementationDecl *OID =
Dan Gohmancb421fa2010-04-19 16:39:44 +00002200 CGM.getContext().getObjCImplementation(
2201 const_cast<ObjCInterfaceDecl *>(ID)))
David Chisnalld901da52010-04-19 01:37:25 +00002202 Offset = ComputeIvarBaseOffset(CGM, OID, Ivar);
2203 else
2204 Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
2205
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002206 llvm::ConstantInt *OffsetGuess =
David Chisnallf9508372010-01-11 19:02:35 +00002207 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset, "ivar");
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002208 // Don't emit the guess in non-PIC code because the linker will not be able
2209 // to replace it with the real version for a library. In non-PIC code you
2210 // must compile with the fragile ABI if you want to use ivars from a
Mike Stump1eb44332009-09-09 15:08:12 +00002211 // GCC-compiled class.
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002212 if (CGM.getLangOptions().PICLevel) {
2213 llvm::GlobalVariable *IvarOffsetGV = new llvm::GlobalVariable(TheModule,
2214 llvm::Type::getInt32Ty(VMContext), false,
2215 llvm::GlobalValue::PrivateLinkage, OffsetGuess, Name+".guess");
2216 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
2217 IvarOffsetGV->getType(), false, llvm::GlobalValue::LinkOnceAnyLinkage,
2218 IvarOffsetGV, Name);
2219 } else {
2220 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
Benjamin Kramer3c0ef8c2009-10-13 10:07:13 +00002221 llvm::Type::getInt32PtrTy(VMContext), false,
2222 llvm::GlobalValue::ExternalLinkage, 0, Name);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002223 }
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002224 }
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002225 return IvarOffsetPointer;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002226}
2227
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00002228LValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
2229 QualType ObjectTy,
2230 llvm::Value *BaseValue,
2231 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00002232 unsigned CVRQualifiers) {
John McCallc12c5bb2010-05-15 11:32:37 +00002233 const ObjCInterfaceDecl *ID =
2234 ObjectTy->getAs<ObjCObjectType>()->getInterface();
Daniel Dunbar97776872009-04-22 07:32:20 +00002235 return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
2236 EmitIvarOffset(CGF, ID, Ivar));
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00002237}
Mike Stumpbb1c8602009-07-31 21:31:32 +00002238
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002239static const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
2240 const ObjCInterfaceDecl *OID,
2241 const ObjCIvarDecl *OIVD) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002242 llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
Fariborz Jahanian8e6ac1d2009-06-04 01:19:09 +00002243 Context.ShallowCollectObjCIvars(OID, Ivars);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002244 for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
2245 if (OIVD == Ivars[k])
2246 return OID;
2247 }
Mike Stump1eb44332009-09-09 15:08:12 +00002248
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002249 // Otherwise check in the super class.
2250 if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
2251 return FindIvarInterface(Context, Super, OIVD);
Mike Stump1eb44332009-09-09 15:08:12 +00002252
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002253 return 0;
2254}
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00002255
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002256llvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +00002257 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002258 const ObjCIvarDecl *Ivar) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002259 if (CGM.getLangOptions().ObjCNonFragileABI) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002260 Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002261 return CGF.Builder.CreateLoad(CGF.Builder.CreateLoad(
2262 ObjCIvarOffsetVariable(Interface, Ivar), false, "ivar"));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002263 }
Daniel Dunbar97776872009-04-22 07:32:20 +00002264 uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00002265 return llvm::ConstantInt::get(LongTy, Offset, "ivar");
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002266}
2267
Mike Stumpbb1c8602009-07-31 21:31:32 +00002268CodeGen::CGObjCRuntime *
2269CodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM) {
Chris Lattnerdce14062008-06-26 04:19:03 +00002270 return new CGObjCGNU(CGM);
Chris Lattner0f984262008-03-01 08:50:34 +00002271}