blob: 77be9fb58231caa64c3d6d1bd214a8d545cc20ca [file] [log] [blame]
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001//===------- CGObjCGNU.cpp - Emit LLVM Code from ASTs for a Module --------===//
Chris Lattner0f984262008-03-01 08:50:34 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000010// This provides Objective-C code generation targetting the GNU runtime. The
11// class in this file generates structures used by the GNU Objective-C runtime
12// library. These structures are defined in objc/objc.h and objc/objc-api.h in
13// the GNU runtime distribution.
Chris Lattner0f984262008-03-01 08:50:34 +000014//
15//===----------------------------------------------------------------------===//
16
17#include "CGObjCRuntime.h"
Chris Lattnerdce14062008-06-26 04:19:03 +000018#include "CodeGenModule.h"
Daniel Dunbar8f2926b2008-08-23 03:46:30 +000019#include "CodeGenFunction.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000020
Chris Lattnerdce14062008-06-26 04:19:03 +000021#include "clang/AST/ASTContext.h"
Daniel Dunbare91593e2008-08-11 04:54:23 +000022#include "clang/AST/Decl.h"
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +000023#include "clang/AST/DeclObjC.h"
Anders Carlsson19cc4ab2009-07-18 19:43:29 +000024#include "clang/AST/RecordLayout.h"
Chris Lattner16f00492009-04-26 01:32:48 +000025#include "clang/AST/StmtObjC.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000026
27#include "llvm/Intrinsics.h"
Chris Lattner0f984262008-03-01 08:50:34 +000028#include "llvm/Module.h"
Chris Lattner0f984262008-03-01 08:50:34 +000029#include "llvm/ADT/SmallVector.h"
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000030#include "llvm/ADT/StringMap.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000031#include "llvm/Support/Compiler.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000032#include "llvm/Target/TargetData.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000033
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000034#include <map>
Chris Lattnere160c9b2009-01-27 05:06:01 +000035
36
Chris Lattnerdce14062008-06-26 04:19:03 +000037using namespace clang;
Daniel Dunbar46f45b92008-09-09 01:06:48 +000038using namespace CodeGen;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000039using llvm::dyn_cast;
40
41// The version of the runtime that this class targets. Must match the version
42// in the runtime.
43static const int RuntimeVersion = 8;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +000044static const int NonFragileRuntimeVersion = 9;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000045static const int ProtocolVersion = 2;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000046static const int NonFragileProtocolVersion = 3;
Chris Lattner0f984262008-03-01 08:50:34 +000047
Chris Lattner0f984262008-03-01 08:50:34 +000048namespace {
Chris Lattnerdce14062008-06-26 04:19:03 +000049class CGObjCGNU : public CodeGen::CGObjCRuntime {
Chris Lattner0f984262008-03-01 08:50:34 +000050private:
Chris Lattnerdce14062008-06-26 04:19:03 +000051 CodeGen::CodeGenModule &CGM;
Chris Lattner0f984262008-03-01 08:50:34 +000052 llvm::Module &TheModule;
Chris Lattnere160c9b2009-01-27 05:06:01 +000053 const llvm::PointerType *SelectorTy;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000054 const llvm::IntegerType *Int8Ty;
Chris Lattnere160c9b2009-01-27 05:06:01 +000055 const llvm::PointerType *PtrToInt8Ty;
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +000056 const llvm::FunctionType *IMPTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000057 const llvm::PointerType *IdTy;
David Chisnall0f436562009-08-17 16:35:33 +000058 QualType ASTIdTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000059 const llvm::IntegerType *IntTy;
60 const llvm::PointerType *PtrTy;
61 const llvm::IntegerType *LongTy;
62 const llvm::PointerType *PtrToIntTy;
Daniel Dunbar5efccb12009-05-04 15:31:17 +000063 llvm::GlobalAlias *ClassPtrAlias;
64 llvm::GlobalAlias *MetaClassPtrAlias;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000065 std::vector<llvm::Constant*> Classes;
66 std::vector<llvm::Constant*> Categories;
67 std::vector<llvm::Constant*> ConstantStrings;
68 llvm::Function *LoadFunction;
69 llvm::StringMap<llvm::Constant*> ExistingProtocols;
70 typedef std::pair<std::string, std::string> TypedSelector;
71 std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
72 llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
73 // Some zeros used for GEPs in lots of places.
74 llvm::Constant *Zeros[2];
75 llvm::Constant *NULLPtr;
Owen Andersona1cf15f2009-07-14 23:10:40 +000076 llvm::LLVMContext &VMContext;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000077private:
78 llvm::Constant *GenerateIvarList(
79 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
80 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
81 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets);
82 llvm::Constant *GenerateMethodList(const std::string &ClassName,
83 const std::string &CategoryName,
Mike Stump1eb44332009-09-09 15:08:12 +000084 const llvm::SmallVectorImpl<Selector> &MethodSels,
85 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000086 bool isClassMethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +000087 llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000088 llvm::Constant *GeneratePropertyList(const ObjCImplementationDecl *OID,
89 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
90 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000091 llvm::Constant *GenerateProtocolList(
92 const llvm::SmallVectorImpl<std::string> &Protocols);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000093 // To ensure that all protocols are seen by the runtime, we add a category on
94 // a class defined in the runtime, declaring no methods, but adopting the
95 // protocols.
96 void GenerateProtocolHolderCategory(void);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000097 llvm::Constant *GenerateClassStructure(
98 llvm::Constant *MetaClass,
99 llvm::Constant *SuperClass,
100 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000101 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000102 llvm::Constant *Version,
103 llvm::Constant *InstanceSize,
104 llvm::Constant *IVars,
105 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000106 llvm::Constant *Protocols,
107 llvm::Constant *IvarOffsets,
108 llvm::Constant *Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000109 llvm::Constant *GenerateProtocolMethodList(
110 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
111 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes);
112 llvm::Constant *MakeConstantString(const std::string &Str, const std::string
113 &Name="");
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000114 llvm::Constant *ExportUniqueString(const std::string &Str, const std::string
115 prefix);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000116 llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
David Chisnall41d63ed2010-01-08 00:14:31 +0000117 std::vector<llvm::Constant*> &V, const std::string &Name="",
118 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000119 llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
David Chisnall41d63ed2010-01-08 00:14:31 +0000120 std::vector<llvm::Constant*> &V, const std::string &Name="",
121 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000122 llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
123 const ObjCIvarDecl *Ivar);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000124 void EmitClassRef(const std::string &className);
Chris Lattner0f984262008-03-01 08:50:34 +0000125public:
Chris Lattnerdce14062008-06-26 04:19:03 +0000126 CGObjCGNU(CodeGen::CodeGenModule &cgm);
David Chisnall0d13f6f2010-01-23 02:40:42 +0000127 virtual llvm::Constant *GenerateConstantString(const StringLiteral *);
Mike Stump1eb44332009-09-09 15:08:12 +0000128 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000129 GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000130 QualType ResultType,
131 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000132 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000133 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000134 const CallArgList &CallArgs,
135 const ObjCMethodDecl *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000136 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000137 GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000138 QualType ResultType,
139 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000140 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000141 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000142 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000143 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000144 const CallArgList &CallArgs,
145 const ObjCMethodDecl *Method);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000146 virtual llvm::Value *GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000147 const ObjCInterfaceDecl *OID);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000148 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000149 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
150 *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000151
152 virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000153 const ObjCContainerDecl *CD);
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000154 virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
155 virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000156 virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000157 const ObjCProtocolDecl *PD);
158 virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000159 virtual llvm::Function *ModuleInitFunction();
Daniel Dunbar49f66022008-09-24 03:38:44 +0000160 virtual llvm::Function *GetPropertyGetFunction();
161 virtual llvm::Function *GetPropertySetFunction();
Daniel Dunbar309a4362009-07-24 07:40:24 +0000162 virtual llvm::Constant *EnumerationMutationFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000163
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000164 virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
165 const Stmt &S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000166 virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
167 const ObjCAtThrowStmt &S);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000168 virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6dc23172008-11-18 21:45:40 +0000169 llvm::Value *AddrWeakObj);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000170 virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
171 llvm::Value *src, llvm::Value *dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000172 virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
173 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +0000174 virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +0000175 llvm::Value *src, llvm::Value *dest,
176 llvm::Value *ivarOffset);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000177 virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
178 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000179 virtual void EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +0000180 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000181 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +0000182 QualType Ty);
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000183 virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
184 QualType ObjectTy,
185 llvm::Value *BaseValue,
186 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000187 unsigned CVRQualifiers);
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000188 virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +0000189 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000190 const ObjCIvarDecl *Ivar);
Chris Lattner0f984262008-03-01 08:50:34 +0000191};
192} // end anonymous namespace
193
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000194
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000195/// Emits a reference to a dummy variable which is emitted with each class.
196/// This ensures that a linker error will be generated when trying to link
197/// together modules where a referenced class is not defined.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000198void CGObjCGNU::EmitClassRef(const std::string &className) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000199 std::string symbolRef = "__objc_class_ref_" + className;
200 // Don't emit two copies of the same symbol
Mike Stumpbb1c8602009-07-31 21:31:32 +0000201 if (TheModule.getGlobalVariable(symbolRef))
202 return;
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000203 std::string symbolName = "__objc_class_name_" + className;
204 llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
205 if (!ClassSymbol) {
Owen Anderson1c431b32009-07-08 19:05:04 +0000206 ClassSymbol = new llvm::GlobalVariable(TheModule, LongTy, false,
207 llvm::GlobalValue::ExternalLinkage, 0, symbolName);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000208 }
Owen Anderson1c431b32009-07-08 19:05:04 +0000209 new llvm::GlobalVariable(TheModule, ClassSymbol->getType(), true,
Chris Lattnerf35271b2009-08-05 05:25:18 +0000210 llvm::GlobalValue::WeakAnyLinkage, ClassSymbol, symbolRef);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000211}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000212
213static std::string SymbolNameForClass(const std::string &ClassName) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000214 return "_OBJC_CLASS_" + ClassName;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000215}
216
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000217static std::string SymbolNameForMethod(const std::string &ClassName, const
218 std::string &CategoryName, const std::string &MethodName, bool isClassMethod)
219{
David Chisnalld3467362010-01-14 14:08:19 +0000220 std::string MethodNameColonStripped = MethodName;
221 std::replace(MethodNameColonStripped.begin(), MethodNameColonStripped.end(),
222 ':', '_');
223 return std::string(isClassMethod ? "_c_" : "_i_") + ClassName + "_" +
224 CategoryName + "_" + MethodNameColonStripped;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000225}
226
Chris Lattnerdce14062008-06-26 04:19:03 +0000227CGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000228 : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
Owen Andersona1cf15f2009-07-14 23:10:40 +0000229 MetaClassPtrAlias(0), VMContext(cgm.getLLVMContext()) {
Chris Lattnere160c9b2009-01-27 05:06:01 +0000230 IntTy = cast<llvm::IntegerType>(
231 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
232 LongTy = cast<llvm::IntegerType>(
233 CGM.getTypes().ConvertType(CGM.getContext().LongTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000234
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000235 Int8Ty = llvm::Type::getInt8Ty(VMContext);
236 // C string type. Used in lots of places.
237 PtrToInt8Ty = llvm::PointerType::getUnqual(Int8Ty);
238
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000239 Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000240 Zeros[1] = Zeros[0];
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000241 NULLPtr = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Chris Lattner391d77a2008-03-30 23:03:07 +0000242 // Get the selector Type.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000243 QualType selTy = CGM.getContext().getObjCSelType();
244 if (QualType() == selTy) {
245 SelectorTy = PtrToInt8Ty;
246 } else {
247 SelectorTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(selTy));
248 }
Chris Lattnere160c9b2009-01-27 05:06:01 +0000249
Owen Anderson96e0fc72009-07-29 22:16:19 +0000250 PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
Chris Lattner391d77a2008-03-30 23:03:07 +0000251 PtrTy = PtrToInt8Ty;
Mike Stump1eb44332009-09-09 15:08:12 +0000252
Chris Lattner391d77a2008-03-30 23:03:07 +0000253 // Object type
David Chisnall0f436562009-08-17 16:35:33 +0000254 ASTIdTy = CGM.getContext().getObjCIdType();
David Chisnall0d13f6f2010-01-23 02:40:42 +0000255 if (QualType() == ASTIdTy) {
256 IdTy = PtrToInt8Ty;
257 } else {
258 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
259 }
Mike Stump1eb44332009-09-09 15:08:12 +0000260
Chris Lattner391d77a2008-03-30 23:03:07 +0000261 // IMP type
262 std::vector<const llvm::Type*> IMPArgs;
263 IMPArgs.push_back(IdTy);
264 IMPArgs.push_back(SelectorTy);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000265 IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000266}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000267
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000268// This has to perform the lookup every time, since posing and related
269// techniques can modify the name -> class mapping.
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000270llvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000271 const ObjCInterfaceDecl *OID) {
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000272 llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
David Chisnall41d63ed2010-01-08 00:14:31 +0000273 // With the incompatible ABI, this will need to be replaced with a direct
274 // reference to the class symbol. For the compatible nonfragile ABI we are
275 // still performing this lookup at run time but emitting the symbol for the
276 // class externally so that we can make the switch later.
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000277 EmitClassRef(OID->getNameAsString());
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000278 ClassName = Builder.CreateStructGEP(ClassName, 0);
279
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000280 std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000281 llvm::Constant *ClassLookupFn =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000282 CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000283 Params,
284 true),
285 "objc_lookup_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000286 return Builder.CreateCall(ClassLookupFn, ClassName);
Chris Lattner391d77a2008-03-30 23:03:07 +0000287}
288
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000289llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel) {
Chris Lattner077bf5e2008-11-24 03:33:13 +0000290 llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
Chris Lattner8e67b632008-06-26 04:37:12 +0000291 if (US == 0)
Owen Anderson96e0fc72009-07-29 22:16:19 +0000292 US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000293 llvm::GlobalValue::PrivateLinkage,
294 ".objc_untyped_selector_alias"+Sel.getAsString(),
Chris Lattner8e67b632008-06-26 04:37:12 +0000295 NULL, &TheModule);
Mike Stump1eb44332009-09-09 15:08:12 +0000296
Chris Lattner8e67b632008-06-26 04:37:12 +0000297 return Builder.CreateLoad(US);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000298}
299
300llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
301 *Method) {
302
303 std::string SelName = Method->getSelector().getAsString();
304 std::string SelTypes;
305 CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
306 // Typed selectors
307 TypedSelector Selector = TypedSelector(SelName,
308 SelTypes);
309
310 // If it's already cached, return it.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000311 if (TypedSelectors[Selector]) {
312 return Builder.CreateLoad(TypedSelectors[Selector]);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000313 }
314
315 // If it isn't, cache it.
316 llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000317 llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000318 llvm::GlobalValue::PrivateLinkage, ".objc_selector_alias" + SelName,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000319 NULL, &TheModule);
320 TypedSelectors[Selector] = Sel;
321
322 return Builder.CreateLoad(Sel);
Chris Lattner8e67b632008-06-26 04:37:12 +0000323}
324
Chris Lattner5e7dcc62008-06-26 04:44:19 +0000325llvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
326 const std::string &Name) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000327 llvm::Constant *ConstStr = CGM.GetAddrOfConstantCString(Str, Name.c_str());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000328 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000329}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000330llvm::Constant *CGObjCGNU::ExportUniqueString(const std::string &Str,
331 const std::string prefix) {
332 std::string name = prefix + Str;
333 llvm::Constant *ConstStr = TheModule.getGlobalVariable(name);
334 if (!ConstStr) {
335 llvm::Constant *value = llvm::ConstantArray::get(VMContext, Str, true);
336 ConstStr = new llvm::GlobalVariable(TheModule, value->getType(), true,
337 llvm::GlobalValue::LinkOnceODRLinkage, value, prefix + Str);
338 }
339 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
340}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000341
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000342llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
David Chisnall41d63ed2010-01-08 00:14:31 +0000343 std::vector<llvm::Constant*> &V, const std::string &Name,
344 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson08e25242009-07-27 22:29:56 +0000345 llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000346 return new llvm::GlobalVariable(TheModule, Ty, false,
347 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000348}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000349
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000350llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
David Chisnall41d63ed2010-01-08 00:14:31 +0000351 std::vector<llvm::Constant*> &V, const std::string &Name,
352 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson7db6d832009-07-28 18:33:04 +0000353 llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000354 return new llvm::GlobalVariable(TheModule, Ty, false,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000355 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000356}
357
358/// Generate an NSConstantString object.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000359llvm::Constant *CGObjCGNU::GenerateConstantString(const StringLiteral *SL) {
360 std::string Str(SL->getStrData(), SL->getByteLength());
361
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000362 std::vector<llvm::Constant*> Ivars;
363 Ivars.push_back(NULLPtr);
Chris Lattner13fd7e52008-06-21 21:44:18 +0000364 Ivars.push_back(MakeConstantString(Str));
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000365 Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000366 llvm::Constant *ObjCStr = MakeGlobal(
Owen Anderson47a434f2009-08-05 23:18:46 +0000367 llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000368 Ivars, ".objc_str");
369 ConstantStrings.push_back(
Owen Anderson3c4972d2009-07-29 18:54:39 +0000370 llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000371 return ObjCStr;
372}
373
374///Generates a message send where the super is the receiver. This is a message
375///send to self with special delivery semantics indicating which class's method
376///should be called.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000377CodeGen::RValue
378CGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000379 QualType ResultType,
380 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000381 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000382 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000383 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000384 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000385 const CallArgList &CallArgs,
386 const ObjCMethodDecl *Method) {
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000387 llvm::Value *cmd = GetSelector(CGF.Builder, Sel);
388
389 CallArgList ActualArgs;
390
391 ActualArgs.push_back(
David Chisnall0f436562009-08-17 16:35:33 +0000392 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
393 ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000394 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
395 CGF.getContext().getObjCSelType()));
396 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
397
398 CodeGenTypes &Types = CGM.getTypes();
399 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
Daniel Dunbar67939662009-09-17 04:01:40 +0000400 const llvm::FunctionType *impType =
401 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000402
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000403 llvm::Value *ReceiverClass = 0;
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000404 if (isCategoryImpl) {
405 llvm::Constant *classLookupFunction = 0;
406 std::vector<const llvm::Type*> Params;
407 Params.push_back(PtrTy);
408 if (IsClassMessage) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000409 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000410 IdTy, Params, true), "objc_get_meta_class");
411 } else {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000412 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000413 IdTy, Params, true), "objc_get_class");
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000414 }
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000415 ReceiverClass = CGF.Builder.CreateCall(classLookupFunction,
416 MakeConstantString(Class->getNameAsString()));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000417 } else {
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000418 // Set up global aliases for the metaclass or class pointer if they do not
419 // already exist. These will are forward-references which will be set to
Mike Stumpbb1c8602009-07-31 21:31:32 +0000420 // pointers to the class and metaclass structure created for the runtime
421 // load function. To send a message to super, we look up the value of the
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000422 // super_class pointer from either the class or metaclass structure.
423 if (IsClassMessage) {
424 if (!MetaClassPtrAlias) {
425 MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
426 llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
427 Class->getNameAsString(), NULL, &TheModule);
428 }
429 ReceiverClass = MetaClassPtrAlias;
430 } else {
431 if (!ClassPtrAlias) {
432 ClassPtrAlias = new llvm::GlobalAlias(IdTy,
433 llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
434 Class->getNameAsString(), NULL, &TheModule);
435 }
436 ReceiverClass = ClassPtrAlias;
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000437 }
Chris Lattner71238f62009-04-25 23:19:45 +0000438 }
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000439 // Cast the pointer to a simplified version of the class structure
Mike Stump1eb44332009-09-09 15:08:12 +0000440 ReceiverClass = CGF.Builder.CreateBitCast(ReceiverClass,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000441 llvm::PointerType::getUnqual(
Owen Anderson47a434f2009-08-05 23:18:46 +0000442 llvm::StructType::get(VMContext, IdTy, IdTy, NULL)));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000443 // Get the superclass pointer
444 ReceiverClass = CGF.Builder.CreateStructGEP(ReceiverClass, 1);
445 // Load the superclass pointer
446 ReceiverClass = CGF.Builder.CreateLoad(ReceiverClass);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000447 // Construct the structure used to look up the IMP
Owen Anderson47a434f2009-08-05 23:18:46 +0000448 llvm::StructType *ObjCSuperTy = llvm::StructType::get(VMContext,
449 Receiver->getType(), IdTy, NULL);
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000450 llvm::Value *ObjCSuper = CGF.Builder.CreateAlloca(ObjCSuperTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000451
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000452 CGF.Builder.CreateStore(Receiver, CGF.Builder.CreateStructGEP(ObjCSuper, 0));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000453 CGF.Builder.CreateStore(ReceiverClass,
454 CGF.Builder.CreateStructGEP(ObjCSuper, 1));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000455
456 // Get the IMP
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000457 std::vector<const llvm::Type*> Params;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000458 Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000459 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000460 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000461 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
462 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000463 "objc_msg_lookup_super");
464
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000465 llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000466 llvm::Value *imp = CGF.Builder.CreateCall(lookupFunction, lookupArgs,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000467 lookupArgs+2);
468
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000469 return CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000470}
471
Mike Stump1eb44332009-09-09 15:08:12 +0000472/// Generate code for a message send expression.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000473CodeGen::RValue
474CGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000475 QualType ResultType,
476 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000477 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000478 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000479 const CallArgList &CallArgs,
480 const ObjCMethodDecl *Method) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000481 CGBuilderTy &Builder = CGF.Builder;
David Chisnall0f436562009-08-17 16:35:33 +0000482 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000483 llvm::Value *cmd;
484 if (Method)
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000485 cmd = GetSelector(Builder, Method);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000486 else
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000487 cmd = GetSelector(Builder, Sel);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000488 CallArgList ActualArgs;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000489
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000490 Receiver = Builder.CreateBitCast(Receiver, IdTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000491 ActualArgs.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000492 std::make_pair(RValue::get(Receiver), ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000493 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
494 CGF.getContext().getObjCSelType()));
495 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
496
497 CodeGenTypes &Types = CGM.getTypes();
498 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
Daniel Dunbar67939662009-09-17 04:01:40 +0000499 const llvm::FunctionType *impType =
500 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000501
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000502 llvm::Value *imp;
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000503 // For sender-aware dispatch, we pass the sender as the third argument to a
504 // lookup function. When sending messages from C code, the sender is nil.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000505 // objc_msg_lookup_sender(id *receiver, SEL selector, id sender);
506 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
507
508 std::vector<const llvm::Type*> Params;
509 llvm::Value *ReceiverPtr = CGF.CreateTempAlloca(Receiver->getType());
510 Builder.CreateStore(Receiver, ReceiverPtr);
511 Params.push_back(ReceiverPtr->getType());
512 Params.push_back(SelectorTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000513 llvm::Value *self;
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000514
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000515 if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
516 self = CGF.LoadObjCSelf();
517 } else {
Owen Anderson03e20502009-07-30 23:11:26 +0000518 self = llvm::ConstantPointerNull::get(IdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000519 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000520
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000521 Params.push_back(self->getType());
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000522
523 // The lookup function returns a slot, which can be safely cached.
524 llvm::Type *SlotTy = llvm::StructType::get(VMContext, PtrTy, PtrTy, PtrTy,
525 IntTy, llvm::PointerType::getUnqual(impType), NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000526 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000527 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000528 llvm::PointerType::getUnqual(SlotTy), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000529 "objc_msg_lookup_sender");
530
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000531 // The lookup function is guaranteed not to capture the receiver pointer.
532 if (llvm::Function *LookupFn = dyn_cast<llvm::Function>(lookupFunction)) {
533 LookupFn->setDoesNotCapture(1);
534 }
535
536 llvm::Value *slot =
537 Builder.CreateCall3(lookupFunction, ReceiverPtr, cmd, self);
538 imp = Builder.CreateLoad(Builder.CreateStructGEP(slot, 4));
539 // The lookup function may have changed the receiver, so make sure we use
540 // the new one.
541 ActualArgs[0] =
542 std::make_pair(RValue::get(Builder.CreateLoad(ReceiverPtr)), ASTIdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000543 } else {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000544 std::vector<const llvm::Type*> Params;
545 Params.push_back(Receiver->getType());
546 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000547 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000548 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
549 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000550 "objc_msg_lookup");
551
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000552 imp = Builder.CreateCall2(lookupFunction, Receiver, cmd);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000553 }
Chris Lattner3eae03e2008-05-06 00:56:42 +0000554
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000555 return CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
Chris Lattner0f984262008-03-01 08:50:34 +0000556}
557
Mike Stump1eb44332009-09-09 15:08:12 +0000558/// Generates a MethodList. Used in construction of a objc_class and
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000559/// objc_category structures.
560llvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
Mike Stump1eb44332009-09-09 15:08:12 +0000561 const std::string &CategoryName,
562 const llvm::SmallVectorImpl<Selector> &MethodSels,
563 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000564 bool isClassMethodList) {
David Chisnall0f436562009-08-17 16:35:33 +0000565 if (MethodSels.empty())
566 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000567 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000568 llvm::StructType *ObjCMethodTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000569 PtrToInt8Ty, // Really a selector, but the runtime creates it us.
570 PtrToInt8Ty, // Method types
Owen Anderson96e0fc72009-07-29 22:16:19 +0000571 llvm::PointerType::getUnqual(IMPTy), //Method pointer
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000572 NULL);
573 std::vector<llvm::Constant*> Methods;
574 std::vector<llvm::Constant*> Elements;
575 for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
576 Elements.clear();
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000577 if (llvm::Constant *Method =
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000578 TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
Chris Lattner077bf5e2008-11-24 03:33:13 +0000579 MethodSels[i].getAsString(),
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000580 isClassMethodList))) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000581 llvm::Constant *C = MakeConstantString(MethodSels[i].getAsString());
582 Elements.push_back(C);
583 Elements.push_back(MethodTypes[i]);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000584 Method = llvm::ConstantExpr::getBitCast(Method,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000585 llvm::PointerType::getUnqual(IMPTy));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000586 Elements.push_back(Method);
Owen Anderson08e25242009-07-27 22:29:56 +0000587 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000588 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000589 }
590
591 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000592 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000593 Methods.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000594 llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
Chris Lattnerfba67632008-06-26 04:52:29 +0000595 Methods);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000596
597 // Structure containing list pointer, array and array count
598 llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
Owen Anderson8c8f69e2009-08-13 23:27:53 +0000599 llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get(VMContext);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000600 llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
Owen Anderson47a434f2009-08-05 23:18:46 +0000601 llvm::StructType *ObjCMethodListTy = llvm::StructType::get(VMContext,
Mike Stump1eb44332009-09-09 15:08:12 +0000602 NextPtrTy,
603 IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000604 ObjCMethodArrayTy,
605 NULL);
606 // Refine next pointer type to concrete type
607 llvm::cast<llvm::OpaqueType>(
608 OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
609 ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
610
611 Methods.clear();
Owen Anderson03e20502009-07-30 23:11:26 +0000612 Methods.push_back(llvm::ConstantPointerNull::get(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000613 llvm::PointerType::getUnqual(ObjCMethodListTy)));
Owen Anderson0032b272009-08-13 21:57:51 +0000614 Methods.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000615 MethodTypes.size()));
616 Methods.push_back(MethodArray);
Mike Stump1eb44332009-09-09 15:08:12 +0000617
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000618 // Create an instance of the structure
619 return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
620}
621
622/// Generates an IvarList. Used in construction of a objc_class.
623llvm::Constant *CGObjCGNU::GenerateIvarList(
624 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
625 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
626 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets) {
David Chisnall18044632009-11-16 19:05:54 +0000627 if (IvarNames.size() == 0)
628 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000629 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000630 llvm::StructType *ObjCIvarTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000631 PtrToInt8Ty,
632 PtrToInt8Ty,
633 IntTy,
634 NULL);
635 std::vector<llvm::Constant*> Ivars;
636 std::vector<llvm::Constant*> Elements;
637 for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
638 Elements.clear();
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000639 Elements.push_back(IvarNames[i]);
640 Elements.push_back(IvarTypes[i]);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000641 Elements.push_back(IvarOffsets[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000642 Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000643 }
644
645 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000646 llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000647 IvarNames.size());
648
Mike Stump1eb44332009-09-09 15:08:12 +0000649
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000650 Elements.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000651 Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
Owen Anderson7db6d832009-07-28 18:33:04 +0000652 Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000653 // Structure containing array and array count
Owen Anderson47a434f2009-08-05 23:18:46 +0000654 llvm::StructType *ObjCIvarListTy = llvm::StructType::get(VMContext, IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000655 ObjCIvarArrayTy,
656 NULL);
657
658 // Create an instance of the structure
659 return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
660}
661
662/// Generate a class structure
663llvm::Constant *CGObjCGNU::GenerateClassStructure(
664 llvm::Constant *MetaClass,
665 llvm::Constant *SuperClass,
666 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000667 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000668 llvm::Constant *Version,
669 llvm::Constant *InstanceSize,
670 llvm::Constant *IVars,
671 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000672 llvm::Constant *Protocols,
673 llvm::Constant *IvarOffsets,
674 llvm::Constant *Properties) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000675 // Set up the class structure
676 // Note: Several of these are char*s when they should be ids. This is
677 // because the runtime performs this translation on load.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000678 //
679 // Fields marked New ABI are part of the GNUstep runtime. We emit them
680 // anyway; the classes will still work with the GNU runtime, they will just
681 // be ignored.
Owen Anderson47a434f2009-08-05 23:18:46 +0000682 llvm::StructType *ClassTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000683 PtrToInt8Ty, // class_pointer
684 PtrToInt8Ty, // super_class
685 PtrToInt8Ty, // name
686 LongTy, // version
687 LongTy, // info
688 LongTy, // instance_size
689 IVars->getType(), // ivars
690 Methods->getType(), // methods
Mike Stump1eb44332009-09-09 15:08:12 +0000691 // These are all filled in by the runtime, so we pretend
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000692 PtrTy, // dtable
693 PtrTy, // subclass_list
694 PtrTy, // sibling_class
695 PtrTy, // protocols
696 PtrTy, // gc_object_type
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000697 // New ABI:
698 LongTy, // abi_version
699 IvarOffsets->getType(), // ivar_offsets
700 Properties->getType(), // properties
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000701 NULL);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000702 llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000703 // Fill in the structure
704 std::vector<llvm::Constant*> Elements;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000705 Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000706 Elements.push_back(SuperClass);
Chris Lattnerd002cc62008-06-26 04:47:04 +0000707 Elements.push_back(MakeConstantString(Name, ".class_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000708 Elements.push_back(Zero);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000709 Elements.push_back(llvm::ConstantInt::get(LongTy, info));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000710 Elements.push_back(InstanceSize);
711 Elements.push_back(IVars);
712 Elements.push_back(Methods);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000713 Elements.push_back(NULLPtr);
714 Elements.push_back(NULLPtr);
715 Elements.push_back(NULLPtr);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000716 Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000717 Elements.push_back(NULLPtr);
718 Elements.push_back(Zero);
719 Elements.push_back(IvarOffsets);
720 Elements.push_back(Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000721 // Create an instance of the structure
David Chisnall41d63ed2010-01-08 00:14:31 +0000722 // This is now an externally visible symbol, so that we can speed up class
723 // messages in the next ABI.
724 return MakeGlobal(ClassTy, Elements, SymbolNameForClass(Name),
725 llvm::GlobalValue::ExternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000726}
727
728llvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
729 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
730 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes) {
Mike Stump1eb44332009-09-09 15:08:12 +0000731 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000732 llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000733 PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
734 PtrToInt8Ty,
735 NULL);
736 std::vector<llvm::Constant*> Methods;
737 std::vector<llvm::Constant*> Elements;
738 for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
739 Elements.clear();
Mike Stump1eb44332009-09-09 15:08:12 +0000740 Elements.push_back(MethodNames[i]);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000741 Elements.push_back(MethodTypes[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000742 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000743 }
Owen Anderson96e0fc72009-07-29 22:16:19 +0000744 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000745 MethodNames.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000746 llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000747 Methods);
Owen Anderson47a434f2009-08-05 23:18:46 +0000748 llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000749 IntTy, ObjCMethodArrayTy, NULL);
750 Methods.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000751 Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000752 Methods.push_back(Array);
753 return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
754}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000755
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000756// Create the protocol list structure used in classes, categories and so on
757llvm::Constant *CGObjCGNU::GenerateProtocolList(
758 const llvm::SmallVectorImpl<std::string> &Protocols) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000759 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000760 Protocols.size());
Owen Anderson47a434f2009-08-05 23:18:46 +0000761 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000762 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
763 LongTy,//FIXME: Should be size_t
764 ProtocolArrayTy,
765 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000766 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000767 for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
768 iter != endIter ; iter++) {
David Chisnallff80fab2009-11-20 14:50:59 +0000769 llvm::Constant *protocol = 0;
770 llvm::StringMap<llvm::Constant*>::iterator value =
771 ExistingProtocols.find(*iter);
772 if (value == ExistingProtocols.end()) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000773 protocol = GenerateEmptyProtocol(*iter);
David Chisnallff80fab2009-11-20 14:50:59 +0000774 } else {
775 protocol = value->getValue();
776 }
Owen Anderson3c4972d2009-07-29 18:54:39 +0000777 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(protocol,
Owen Andersona1cf15f2009-07-14 23:10:40 +0000778 PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000779 Elements.push_back(Ptr);
780 }
Owen Anderson7db6d832009-07-28 18:33:04 +0000781 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000782 Elements);
783 Elements.clear();
784 Elements.push_back(NULLPtr);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000785 Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000786 Elements.push_back(ProtocolArray);
787 return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
788}
789
Mike Stump1eb44332009-09-09 15:08:12 +0000790llvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000791 const ObjCProtocolDecl *PD) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000792 llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
Mike Stump1eb44332009-09-09 15:08:12 +0000793 const llvm::Type *T =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000794 CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
Owen Anderson96e0fc72009-07-29 22:16:19 +0000795 return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000796}
797
798llvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
799 const std::string &ProtocolName) {
800 llvm::SmallVector<std::string, 0> EmptyStringVector;
801 llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
802
803 llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000804 llvm::Constant *MethodList =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000805 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
806 // Protocols are objects containing lists of the methods implemented and
807 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000808 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000809 PtrToInt8Ty,
810 ProtocolList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000811 MethodList->getType(),
812 MethodList->getType(),
813 MethodList->getType(),
814 MethodList->getType(),
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000815 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000816 std::vector<llvm::Constant*> Elements;
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000817 // The isa pointer must be set to a magic number so the runtime knows it's
818 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000819 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
820 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000821 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000822 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000823 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
824 Elements.push_back(ProtocolList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000825 Elements.push_back(MethodList);
826 Elements.push_back(MethodList);
827 Elements.push_back(MethodList);
828 Elements.push_back(MethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000829 return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000830}
831
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000832void CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
833 ASTContext &Context = CGM.getContext();
Chris Lattner8ec03f52008-11-24 03:54:41 +0000834 std::string ProtocolName = PD->getNameAsString();
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000835 llvm::SmallVector<std::string, 16> Protocols;
836 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
837 E = PD->protocol_end(); PI != E; ++PI)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000838 Protocols.push_back((*PI)->getNameAsString());
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000839 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
840 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000841 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodNames;
842 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodTypes;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000843 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
844 E = PD->instmeth_end(); iter != E; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000845 std::string TypeStr;
846 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000847 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
848 InstanceMethodNames.push_back(
849 MakeConstantString((*iter)->getSelector().getAsString()));
850 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
851 } else {
852 OptionalInstanceMethodNames.push_back(
853 MakeConstantString((*iter)->getSelector().getAsString()));
854 OptionalInstanceMethodTypes.push_back(MakeConstantString(TypeStr));
855 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000856 }
857 // Collect information about class methods:
858 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
859 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000860 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodNames;
861 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +0000862 for (ObjCProtocolDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000863 iter = PD->classmeth_begin(), endIter = PD->classmeth_end();
864 iter != endIter ; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000865 std::string TypeStr;
866 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000867 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
868 ClassMethodNames.push_back(
869 MakeConstantString((*iter)->getSelector().getAsString()));
870 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
871 } else {
872 OptionalClassMethodNames.push_back(
873 MakeConstantString((*iter)->getSelector().getAsString()));
874 OptionalClassMethodTypes.push_back(MakeConstantString(TypeStr));
875 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000876 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000877
878 llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
879 llvm::Constant *InstanceMethodList =
880 GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
881 llvm::Constant *ClassMethodList =
882 GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000883 llvm::Constant *OptionalInstanceMethodList =
884 GenerateProtocolMethodList(OptionalInstanceMethodNames,
885 OptionalInstanceMethodTypes);
886 llvm::Constant *OptionalClassMethodList =
887 GenerateProtocolMethodList(OptionalClassMethodNames,
888 OptionalClassMethodTypes);
889
890 // Property metadata: name, attributes, isSynthesized, setter name, setter
891 // types, getter name, getter types.
892 // The isSynthesized value is always set to 0 in a protocol. It exists to
893 // simplify the runtime library by allowing it to use the same data
894 // structures for protocol metadata everywhere.
895 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
896 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
897 PtrToInt8Ty, NULL);
898 std::vector<llvm::Constant*> Properties;
899 std::vector<llvm::Constant*> OptionalProperties;
900
901 // Add all of the property methods need adding to the method list and to the
902 // property metadata list.
903 for (ObjCContainerDecl::prop_iterator
904 iter = PD->prop_begin(), endIter = PD->prop_end();
905 iter != endIter ; iter++) {
906 std::vector<llvm::Constant*> Fields;
907 ObjCPropertyDecl *property = (*iter);
908
909 Fields.push_back(MakeConstantString(property->getNameAsString()));
910 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
911 property->getPropertyAttributes()));
912 Fields.push_back(llvm::ConstantInt::get(Int8Ty, 0));
913 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
914 std::string TypeStr;
915 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
916 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
917 InstanceMethodTypes.push_back(TypeEncoding);
918 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
919 Fields.push_back(TypeEncoding);
920 } else {
921 Fields.push_back(NULLPtr);
922 Fields.push_back(NULLPtr);
923 }
924 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
925 std::string TypeStr;
926 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
927 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
928 InstanceMethodTypes.push_back(TypeEncoding);
929 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
930 Fields.push_back(TypeEncoding);
931 } else {
932 Fields.push_back(NULLPtr);
933 Fields.push_back(NULLPtr);
934 }
935 if (property->getPropertyImplementation() == ObjCPropertyDecl::Optional) {
936 OptionalProperties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
937 } else {
938 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
939 }
940 }
941 llvm::Constant *PropertyArray = llvm::ConstantArray::get(
942 llvm::ArrayType::get(PropertyMetadataTy, Properties.size()), Properties);
943 llvm::Constant* PropertyListInitFields[] =
944 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
945
946 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000947 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000948 llvm::Constant *PropertyList = new llvm::GlobalVariable(TheModule,
949 PropertyListInit->getType(), false, llvm::GlobalValue::InternalLinkage,
950 PropertyListInit, ".objc_property_list");
951
952 llvm::Constant *OptionalPropertyArray =
953 llvm::ConstantArray::get(llvm::ArrayType::get(PropertyMetadataTy,
954 OptionalProperties.size()) , OptionalProperties);
955 llvm::Constant* OptionalPropertyListInitFields[] = {
956 llvm::ConstantInt::get(IntTy, OptionalProperties.size()), NULLPtr,
957 OptionalPropertyArray };
958
959 llvm::Constant *OptionalPropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000960 llvm::ConstantStruct::get(VMContext, OptionalPropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000961 llvm::Constant *OptionalPropertyList = new llvm::GlobalVariable(TheModule,
962 OptionalPropertyListInit->getType(), false,
963 llvm::GlobalValue::InternalLinkage, OptionalPropertyListInit,
964 ".objc_property_list");
965
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000966 // Protocols are objects containing lists of the methods implemented and
967 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000968 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000969 PtrToInt8Ty,
970 ProtocolList->getType(),
971 InstanceMethodList->getType(),
972 ClassMethodList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000973 OptionalInstanceMethodList->getType(),
974 OptionalClassMethodList->getType(),
975 PropertyList->getType(),
976 OptionalPropertyList->getType(),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000977 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000978 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000979 // The isa pointer must be set to a magic number so the runtime knows it's
980 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000981 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
982 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000983 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000984 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000985 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
986 Elements.push_back(ProtocolList);
987 Elements.push_back(InstanceMethodList);
988 Elements.push_back(ClassMethodList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000989 Elements.push_back(OptionalInstanceMethodList);
990 Elements.push_back(OptionalClassMethodList);
991 Elements.push_back(PropertyList);
992 Elements.push_back(OptionalPropertyList);
Mike Stump1eb44332009-09-09 15:08:12 +0000993 ExistingProtocols[ProtocolName] =
Owen Anderson3c4972d2009-07-29 18:54:39 +0000994 llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000995 ".objc_protocol"), IdTy);
996}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000997void CGObjCGNU::GenerateProtocolHolderCategory(void) {
998 // Collect information about instance methods
999 llvm::SmallVector<Selector, 1> MethodSels;
1000 llvm::SmallVector<llvm::Constant*, 1> MethodTypes;
1001
1002 std::vector<llvm::Constant*> Elements;
1003 const std::string ClassName = "__ObjC_Protocol_Holder_Ugly_Hack";
1004 const std::string CategoryName = "AnotherHack";
1005 Elements.push_back(MakeConstantString(CategoryName));
1006 Elements.push_back(MakeConstantString(ClassName));
1007 // Instance method list
1008 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1009 ClassName, CategoryName, MethodSels, MethodTypes, false), PtrTy));
1010 // Class method list
1011 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1012 ClassName, CategoryName, MethodSels, MethodTypes, true), PtrTy));
1013 // Protocol list
1014 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrTy,
1015 ExistingProtocols.size());
1016 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
1017 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
1018 LongTy,//FIXME: Should be size_t
1019 ProtocolArrayTy,
1020 NULL);
1021 std::vector<llvm::Constant*> ProtocolElements;
1022 for (llvm::StringMapIterator<llvm::Constant*> iter =
1023 ExistingProtocols.begin(), endIter = ExistingProtocols.end();
1024 iter != endIter ; iter++) {
1025 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(iter->getValue(),
1026 PtrTy);
1027 ProtocolElements.push_back(Ptr);
1028 }
1029 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
1030 ProtocolElements);
1031 ProtocolElements.clear();
1032 ProtocolElements.push_back(NULLPtr);
1033 ProtocolElements.push_back(llvm::ConstantInt::get(LongTy,
1034 ExistingProtocols.size()));
1035 ProtocolElements.push_back(ProtocolArray);
1036 Elements.push_back(llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolListTy,
1037 ProtocolElements, ".objc_protocol_list"), PtrTy));
1038 Categories.push_back(llvm::ConstantExpr::getBitCast(
1039 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
1040 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
1041}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001042
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001043void CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001044 std::string ClassName = OCD->getClassInterface()->getNameAsString();
1045 std::string CategoryName = OCD->getNameAsString();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001046 // Collect information about instance methods
1047 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1048 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001049 for (ObjCCategoryImplDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001050 iter = OCD->instmeth_begin(), endIter = OCD->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001051 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001052 InstanceMethodSels.push_back((*iter)->getSelector());
1053 std::string TypeStr;
1054 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001055 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001056 }
1057
1058 // Collect information about class methods
1059 llvm::SmallVector<Selector, 16> ClassMethodSels;
1060 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001061 for (ObjCCategoryImplDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001062 iter = OCD->classmeth_begin(), endIter = OCD->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001063 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001064 ClassMethodSels.push_back((*iter)->getSelector());
1065 std::string TypeStr;
1066 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001067 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001068 }
1069
1070 // Collect the names of referenced protocols
1071 llvm::SmallVector<std::string, 16> Protocols;
1072 const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
1073 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1074 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1075 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001076 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001077
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001078 std::vector<llvm::Constant*> Elements;
1079 Elements.push_back(MakeConstantString(CategoryName));
1080 Elements.push_back(MakeConstantString(ClassName));
Mike Stump1eb44332009-09-09 15:08:12 +00001081 // Instance method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001082 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001083 ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001084 false), PtrTy));
1085 // Class method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001086 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001087 ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001088 PtrTy));
1089 // Protocol list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001090 Elements.push_back(llvm::ConstantExpr::getBitCast(
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001091 GenerateProtocolList(Protocols), PtrTy));
Owen Anderson3c4972d2009-07-29 18:54:39 +00001092 Categories.push_back(llvm::ConstantExpr::getBitCast(
Mike Stump1eb44332009-09-09 15:08:12 +00001093 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
Owen Anderson47a434f2009-08-05 23:18:46 +00001094 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001095}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001096
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001097llvm::Constant *CGObjCGNU::GeneratePropertyList(const ObjCImplementationDecl *OID,
1098 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
1099 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes) {
1100 ASTContext &Context = CGM.getContext();
1101 //
1102 // Property metadata: name, attributes, isSynthesized, setter name, setter
1103 // types, getter name, getter types.
1104 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1105 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1106 PtrToInt8Ty, NULL);
1107 std::vector<llvm::Constant*> Properties;
1108
1109
1110 // Add all of the property methods need adding to the method list and to the
1111 // property metadata list.
1112 for (ObjCImplDecl::propimpl_iterator
1113 iter = OID->propimpl_begin(), endIter = OID->propimpl_end();
1114 iter != endIter ; iter++) {
1115 std::vector<llvm::Constant*> Fields;
1116 ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
1117
1118 Fields.push_back(MakeConstantString(property->getNameAsString()));
1119 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1120 property->getPropertyAttributes()));
1121 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1122 (*iter)->getPropertyImplementation() ==
1123 ObjCPropertyImplDecl::Synthesize));
1124 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
1125 InstanceMethodSels.push_back(getter->getSelector());
1126 std::string TypeStr;
1127 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1128 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1129 InstanceMethodTypes.push_back(TypeEncoding);
1130 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1131 Fields.push_back(TypeEncoding);
1132 } else {
1133 Fields.push_back(NULLPtr);
1134 Fields.push_back(NULLPtr);
1135 }
1136 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1137 InstanceMethodSels.push_back(setter->getSelector());
1138 std::string TypeStr;
1139 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1140 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1141 InstanceMethodTypes.push_back(TypeEncoding);
1142 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1143 Fields.push_back(TypeEncoding);
1144 } else {
1145 Fields.push_back(NULLPtr);
1146 Fields.push_back(NULLPtr);
1147 }
1148 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1149 }
1150 llvm::ArrayType *PropertyArrayTy =
1151 llvm::ArrayType::get(PropertyMetadataTy, Properties.size());
1152 llvm::Constant *PropertyArray = llvm::ConstantArray::get(PropertyArrayTy,
1153 Properties);
1154 llvm::Constant* PropertyListInitFields[] =
1155 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1156
1157 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001158 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001159 return new llvm::GlobalVariable(TheModule, PropertyListInit->getType(), false,
1160 llvm::GlobalValue::InternalLinkage, PropertyListInit,
1161 ".objc_property_list");
1162}
1163
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001164void CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
1165 ASTContext &Context = CGM.getContext();
1166
1167 // Get the superclass name.
Mike Stump1eb44332009-09-09 15:08:12 +00001168 const ObjCInterfaceDecl * SuperClassDecl =
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001169 OID->getClassInterface()->getSuperClass();
Chris Lattner8ec03f52008-11-24 03:54:41 +00001170 std::string SuperClassName;
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001171 if (SuperClassDecl) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001172 SuperClassName = SuperClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001173 EmitClassRef(SuperClassName);
1174 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001175
1176 // Get the class name
Chris Lattner09dc6662009-04-01 02:00:48 +00001177 ObjCInterfaceDecl *ClassDecl =
1178 const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
Chris Lattner8ec03f52008-11-24 03:54:41 +00001179 std::string ClassName = ClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001180 // Emit the symbol that is used to generate linker errors if this class is
1181 // referenced in other modules but not declared.
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001182 std::string classSymbolName = "__objc_class_name_" + ClassName;
Mike Stump1eb44332009-09-09 15:08:12 +00001183 if (llvm::GlobalVariable *symbol =
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001184 TheModule.getGlobalVariable(classSymbolName)) {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001185 symbol->setInitializer(llvm::ConstantInt::get(LongTy, 0));
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001186 } else {
Owen Anderson1c431b32009-07-08 19:05:04 +00001187 new llvm::GlobalVariable(TheModule, LongTy, false,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001188 llvm::GlobalValue::ExternalLinkage, llvm::ConstantInt::get(LongTy, 0),
Owen Anderson1c431b32009-07-08 19:05:04 +00001189 classSymbolName);
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001190 }
Mike Stump1eb44332009-09-09 15:08:12 +00001191
Daniel Dunbar2bebbf02009-05-03 10:46:44 +00001192 // Get the size of instances.
1193 int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001194
1195 // Collect information about instance variables.
1196 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
1197 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
1198 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
Mike Stump1eb44332009-09-09 15:08:12 +00001199
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001200 std::vector<llvm::Constant*> IvarOffsetValues;
1201
Mike Stump1eb44332009-09-09 15:08:12 +00001202 int superInstanceSize = !SuperClassDecl ? 0 :
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001203 Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
1204 // For non-fragile ivars, set the instance size to 0 - {the size of just this
1205 // class}. The runtime will then set this to the correct value on load.
1206 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
1207 instanceSize = 0 - (instanceSize - superInstanceSize);
1208 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001209 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1210 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1211 // Store the name
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001212 IvarNames.push_back(MakeConstantString((*iter)->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001213 // Get the type encoding for this ivar
1214 std::string TypeStr;
Daniel Dunbar0d504c12008-10-17 20:21:44 +00001215 Context.getObjCEncodingForType((*iter)->getType(), TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001216 IvarTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001217 // Get the offset
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001218 uint64_t BaseOffset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter);
David Chisnallaecbf242009-11-17 19:32:15 +00001219 uint64_t Offset = BaseOffset;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001220 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001221 Offset = BaseOffset - superInstanceSize;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001222 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001223 IvarOffsets.push_back(
Owen Anderson0032b272009-08-13 21:57:51 +00001224 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001225 IvarOffsetValues.push_back(new llvm::GlobalVariable(TheModule, IntTy,
1226 false, llvm::GlobalValue::ExternalLinkage,
1227 llvm::ConstantInt::get(IntTy, BaseOffset),
1228 "__objc_ivar_offset_value_" + ClassName +"." +
1229 (*iter)->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001230 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001231 llvm::Constant *IvarOffsetArrayInit =
1232 llvm::ConstantArray::get(llvm::ArrayType::get(PtrToIntTy,
1233 IvarOffsetValues.size()), IvarOffsetValues);
1234 llvm::GlobalVariable *IvarOffsetArray = new llvm::GlobalVariable(TheModule,
1235 IvarOffsetArrayInit->getType(), false,
1236 llvm::GlobalValue::InternalLinkage, IvarOffsetArrayInit,
1237 ".ivar.offsets");
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001238
1239 // Collect information about instance methods
1240 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1241 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001242 for (ObjCImplementationDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001243 iter = OID->instmeth_begin(), endIter = OID->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001244 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001245 InstanceMethodSels.push_back((*iter)->getSelector());
1246 std::string TypeStr;
1247 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001248 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001249 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001250
1251 llvm::Constant *Properties = GeneratePropertyList(OID, InstanceMethodSels,
1252 InstanceMethodTypes);
1253
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001254
1255 // Collect information about class methods
1256 llvm::SmallVector<Selector, 16> ClassMethodSels;
1257 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001258 for (ObjCImplementationDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001259 iter = OID->classmeth_begin(), endIter = OID->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001260 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001261 ClassMethodSels.push_back((*iter)->getSelector());
1262 std::string TypeStr;
1263 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001264 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001265 }
1266 // Collect the names of referenced protocols
1267 llvm::SmallVector<std::string, 16> Protocols;
1268 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1269 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1270 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001271 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001272
1273
1274
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001275 // Get the superclass pointer.
1276 llvm::Constant *SuperClass;
Chris Lattner8ec03f52008-11-24 03:54:41 +00001277 if (!SuperClassName.empty()) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001278 SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
1279 } else {
Owen Anderson03e20502009-07-30 23:11:26 +00001280 SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001281 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001282 // Empty vector used to construct empty method lists
1283 llvm::SmallVector<llvm::Constant*, 1> empty;
1284 // Generate the method and instance variable lists
1285 llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001286 InstanceMethodSels, InstanceMethodTypes, false);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001287 llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001288 ClassMethodSels, ClassMethodTypes, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001289 llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
1290 IvarOffsets);
Mike Stump1eb44332009-09-09 15:08:12 +00001291 // Irrespective of whether we are compiling for a fragile or non-fragile ABI,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001292 // we emit a symbol containing the offset for each ivar in the class. This
1293 // allows code compiled for the non-Fragile ABI to inherit from code compiled
1294 // for the legacy ABI, without causing problems. The converse is also
1295 // possible, but causes all ivar accesses to be fragile.
1296 int i = 0;
1297 // Offset pointer for getting at the correct field in the ivar list when
1298 // setting up the alias. These are: The base address for the global, the
1299 // ivar array (second field), the ivar in this list (set for each ivar), and
1300 // the offset (third field in ivar structure)
1301 const llvm::Type *IndexTy = llvm::Type::getInt32Ty(VMContext);
1302 llvm::Constant *offsetPointerIndexes[] = {Zeros[0],
Mike Stump1eb44332009-09-09 15:08:12 +00001303 llvm::ConstantInt::get(IndexTy, 1), 0,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001304 llvm::ConstantInt::get(IndexTy, 2) };
1305
1306 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1307 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1308 const std::string Name = "__objc_ivar_offset_" + ClassName + '.'
1309 +(*iter)->getNameAsString();
1310 offsetPointerIndexes[2] = llvm::ConstantInt::get(IndexTy, i++);
1311 // Get the correct ivar field
1312 llvm::Constant *offsetValue = llvm::ConstantExpr::getGetElementPtr(
1313 IvarList, offsetPointerIndexes, 4);
1314 // Get the existing alias, if one exists.
1315 llvm::GlobalVariable *offset = TheModule.getNamedGlobal(Name);
1316 if (offset) {
1317 offset->setInitializer(offsetValue);
1318 // If this is the real definition, change its linkage type so that
1319 // different modules will use this one, rather than their private
1320 // copy.
1321 offset->setLinkage(llvm::GlobalValue::ExternalLinkage);
1322 } else {
1323 // Add a new alias if there isn't one already.
1324 offset = new llvm::GlobalVariable(TheModule, offsetValue->getType(),
1325 false, llvm::GlobalValue::ExternalLinkage, offsetValue, Name);
1326 }
1327 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001328 //Generate metaclass for class methods
1329 llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
David Chisnall18044632009-11-16 19:05:54 +00001330 NULLPtr, 0x12L, ClassName.c_str(), 0, Zeros[0], GenerateIvarList(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001331 empty, empty, empty), ClassMethodList, NULLPtr, NULLPtr, NULLPtr);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001332
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001333 // Generate the class structure
Chris Lattner8ec03f52008-11-24 03:54:41 +00001334 llvm::Constant *ClassStruct =
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001335 GenerateClassStructure(MetaClassStruct, SuperClass, 0x11L,
Chris Lattner8ec03f52008-11-24 03:54:41 +00001336 ClassName.c_str(), 0,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001337 llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001338 MethodList, GenerateProtocolList(Protocols), IvarOffsetArray,
1339 Properties);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001340
1341 // Resolve the class aliases, if they exist.
1342 if (ClassPtrAlias) {
1343 ClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001344 llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001345 ClassPtrAlias = 0;
1346 }
1347 if (MetaClassPtrAlias) {
1348 MetaClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001349 llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001350 MetaClassPtrAlias = 0;
1351 }
1352
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001353 // Add class structure to list to be added to the symtab later
Owen Anderson3c4972d2009-07-29 18:54:39 +00001354 ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001355 Classes.push_back(ClassStruct);
1356}
1357
Fariborz Jahanianc38e9af2009-06-23 21:47:46 +00001358
Mike Stump1eb44332009-09-09 15:08:12 +00001359llvm::Function *CGObjCGNU::ModuleInitFunction() {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001360 // Only emit an ObjC load function if no Objective-C stuff has been called
1361 if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
1362 ExistingProtocols.empty() && TypedSelectors.empty() &&
Anton Korobeynikov5f58b912008-06-01 15:14:46 +00001363 UntypedSelectors.empty())
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001364 return NULL;
Eli Friedman1b8956e2008-06-01 16:00:02 +00001365
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001366 // Add all referenced protocols to a category.
1367 GenerateProtocolHolderCategory();
1368
Chris Lattnere160c9b2009-01-27 05:06:01 +00001369 const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1370 SelectorTy->getElementType());
1371 const llvm::Type *SelStructPtrTy = SelectorTy;
1372 bool isSelOpaque = false;
1373 if (SelStructTy == 0) {
Owen Anderson47a434f2009-08-05 23:18:46 +00001374 SelStructTy = llvm::StructType::get(VMContext, PtrToInt8Ty,
1375 PtrToInt8Ty, NULL);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001376 SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
Chris Lattnere160c9b2009-01-27 05:06:01 +00001377 isSelOpaque = true;
1378 }
1379
Eli Friedman1b8956e2008-06-01 16:00:02 +00001380 // Name the ObjC types to make the IR a bit easier to read
Chris Lattnere160c9b2009-01-27 05:06:01 +00001381 TheModule.addTypeName(".objc_selector", SelStructPtrTy);
Eli Friedman1b8956e2008-06-01 16:00:02 +00001382 TheModule.addTypeName(".objc_id", IdTy);
1383 TheModule.addTypeName(".objc_imp", IMPTy);
1384
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001385 std::vector<llvm::Constant*> Elements;
Chris Lattner71238f62009-04-25 23:19:45 +00001386 llvm::Constant *Statics = NULLPtr;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001387 // Generate statics list:
Chris Lattner71238f62009-04-25 23:19:45 +00001388 if (ConstantStrings.size()) {
Owen Anderson96e0fc72009-07-29 22:16:19 +00001389 llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Chris Lattner71238f62009-04-25 23:19:45 +00001390 ConstantStrings.size() + 1);
1391 ConstantStrings.push_back(NULLPtr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001392
Daniel Dunbar1b096952009-11-29 02:38:47 +00001393 llvm::StringRef StringClass = CGM.getLangOptions().ObjCConstantStringClass;
1394 if (StringClass.empty()) StringClass = "NXConstantString";
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001395 Elements.push_back(MakeConstantString(StringClass,
1396 ".objc_static_class_name"));
Owen Anderson7db6d832009-07-28 18:33:04 +00001397 Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
Chris Lattner71238f62009-04-25 23:19:45 +00001398 ConstantStrings));
Mike Stump1eb44332009-09-09 15:08:12 +00001399 llvm::StructType *StaticsListTy =
Owen Anderson47a434f2009-08-05 23:18:46 +00001400 llvm::StructType::get(VMContext, PtrToInt8Ty, StaticsArrayTy, NULL);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001401 llvm::Type *StaticsListPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001402 llvm::PointerType::getUnqual(StaticsListTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001403 Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
Mike Stump1eb44332009-09-09 15:08:12 +00001404 llvm::ArrayType *StaticsListArrayTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001405 llvm::ArrayType::get(StaticsListPtrTy, 2);
Chris Lattner71238f62009-04-25 23:19:45 +00001406 Elements.clear();
1407 Elements.push_back(Statics);
Owen Andersonc9c88b42009-07-31 20:28:54 +00001408 Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
Chris Lattner71238f62009-04-25 23:19:45 +00001409 Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001410 Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001411 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001412 // Array of classes, categories, and constant objects
Owen Anderson96e0fc72009-07-29 22:16:19 +00001413 llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001414 Classes.size() + Categories.size() + 2);
Mike Stump1eb44332009-09-09 15:08:12 +00001415 llvm::StructType *SymTabTy = llvm::StructType::get(VMContext,
Owen Anderson47a434f2009-08-05 23:18:46 +00001416 LongTy, SelStructPtrTy,
Owen Anderson0032b272009-08-13 21:57:51 +00001417 llvm::Type::getInt16Ty(VMContext),
1418 llvm::Type::getInt16Ty(VMContext),
Chris Lattner630404b2008-06-26 04:10:42 +00001419 ClassListTy, NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001420
1421 Elements.clear();
1422 // Pointer to an array of selectors used in this module.
1423 std::vector<llvm::Constant*> Selectors;
1424 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1425 iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
1426 iter != iterEnd ; ++iter) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001427 Elements.push_back(ExportUniqueString(iter->first.first, ".objc_sel_name"));
David Chisnalla7c5b082009-09-14 19:04:10 +00001428 Elements.push_back(MakeConstantString(iter->first.second,
Chris Lattner630404b2008-06-26 04:10:42 +00001429 ".objc_sel_types"));
Owen Anderson08e25242009-07-27 22:29:56 +00001430 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001431 Elements.clear();
1432 }
1433 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1434 iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
Chris Lattner630404b2008-06-26 04:10:42 +00001435 iter != iterEnd; ++iter) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001436 Elements.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001437 ExportUniqueString(iter->getKeyData(), ".objc_sel_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001438 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001439 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001440 Elements.clear();
1441 }
1442 Elements.push_back(NULLPtr);
1443 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001444 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001445 Elements.clear();
1446 // Number of static selectors
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001447 Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001448 llvm::Constant *SelectorList = MakeGlobal(
Owen Anderson96e0fc72009-07-29 22:16:19 +00001449 llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001450 ".objc_selector_list");
Mike Stump1eb44332009-09-09 15:08:12 +00001451 Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
Chris Lattnere160c9b2009-01-27 05:06:01 +00001452 SelStructPtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001453
1454 // Now that all of the static selectors exist, create pointers to them.
1455 int index = 0;
1456 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1457 iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
1458 iter != iterEnd; ++iter) {
1459 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001460 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Owen Anderson1c431b32009-07-08 19:05:04 +00001461 llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
1462 true, llvm::GlobalValue::InternalLinkage,
Owen Anderson3c4972d2009-07-29 18:54:39 +00001463 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
Owen Anderson1c431b32009-07-08 19:05:04 +00001464 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001465 // If selectors are defined as an opaque type, cast the pointer to this
1466 // type.
1467 if (isSelOpaque) {
Owen Anderson3c4972d2009-07-29 18:54:39 +00001468 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001469 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001470 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001471 (*iter).second->setAliasee(SelPtr);
1472 }
1473 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1474 iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1475 iter != iterEnd; iter++) {
1476 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001477 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Mike Stumpbb1c8602009-07-31 21:31:32 +00001478 llvm::Constant *SelPtr = new llvm::GlobalVariable
Mike Stump1eb44332009-09-09 15:08:12 +00001479 (TheModule, SelStructPtrTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +00001480 true, llvm::GlobalValue::InternalLinkage,
1481 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1482 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001483 // If selectors are defined as an opaque type, cast the pointer to this
1484 // type.
1485 if (isSelOpaque) {
Owen Anderson3c4972d2009-07-29 18:54:39 +00001486 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001487 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001488 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001489 (*iter).second->setAliasee(SelPtr);
1490 }
1491 // Number of classes defined.
Mike Stump1eb44332009-09-09 15:08:12 +00001492 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001493 Classes.size()));
1494 // Number of categories defined
Mike Stump1eb44332009-09-09 15:08:12 +00001495 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001496 Categories.size()));
1497 // Create an array of classes, then categories, then static object instances
1498 Classes.insert(Classes.end(), Categories.begin(), Categories.end());
1499 // NULL-terminated list of static object instances (mainly constant strings)
1500 Classes.push_back(Statics);
1501 Classes.push_back(NULLPtr);
Owen Anderson7db6d832009-07-28 18:33:04 +00001502 llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001503 Elements.push_back(ClassList);
Mike Stump1eb44332009-09-09 15:08:12 +00001504 // Construct the symbol table
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001505 llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
1506
1507 // The symbol table is contained in a module which has some version-checking
1508 // constants
Owen Anderson47a434f2009-08-05 23:18:46 +00001509 llvm::StructType * ModuleTy = llvm::StructType::get(VMContext, LongTy, LongTy,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001510 PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001511 Elements.clear();
1512 // Runtime version used for compatibility checking.
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001513 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Mike Stump1eb44332009-09-09 15:08:12 +00001514 Elements.push_back(llvm::ConstantInt::get(LongTy,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001515 NonFragileRuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001516 } else {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001517 Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001518 }
Fariborz Jahanian91a0b512009-04-01 19:49:42 +00001519 // sizeof(ModuleTy)
1520 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001521 Elements.push_back(llvm::ConstantInt::get(LongTy,
Owen Andersona1cf15f2009-07-14 23:10:40 +00001522 td.getTypeSizeInBits(ModuleTy)/8));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001523 //FIXME: Should be the path to the file where this module was declared
1524 Elements.push_back(NULLPtr);
1525 Elements.push_back(SymTab);
1526 llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
1527
1528 // Create the load function calling the runtime entry point with the module
1529 // structure
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001530 llvm::Function * LoadFunction = llvm::Function::Create(
Owen Anderson0032b272009-08-13 21:57:51 +00001531 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), false),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001532 llvm::GlobalValue::InternalLinkage, ".objc_load_function",
1533 &TheModule);
Owen Anderson0032b272009-08-13 21:57:51 +00001534 llvm::BasicBlock *EntryBB =
1535 llvm::BasicBlock::Create(VMContext, "entry", LoadFunction);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001536 CGBuilderTy Builder(VMContext);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001537 Builder.SetInsertPoint(EntryBB);
Fariborz Jahanian26c82942009-03-30 18:02:14 +00001538
1539 std::vector<const llvm::Type*> Params(1,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001540 llvm::PointerType::getUnqual(ModuleTy));
1541 llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Owen Anderson0032b272009-08-13 21:57:51 +00001542 llvm::Type::getVoidTy(VMContext), Params, true), "__objc_exec_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001543 Builder.CreateCall(Register, Module);
1544 Builder.CreateRetVoid();
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001545
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001546 return LoadFunction;
1547}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001548
Fariborz Jahanian679a5022009-01-10 21:06:09 +00001549llvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
Mike Stump1eb44332009-09-09 15:08:12 +00001550 const ObjCContainerDecl *CD) {
1551 const ObjCCategoryImplDecl *OCD =
Steve Naroff3e0a5402009-01-08 19:41:02 +00001552 dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
Chris Lattner077bf5e2008-11-24 03:33:13 +00001553 std::string CategoryName = OCD ? OCD->getNameAsString() : "";
1554 std::string ClassName = OMD->getClassInterface()->getNameAsString();
1555 std::string MethodName = OMD->getSelector().getAsString();
Douglas Gregorf8d49f62009-01-09 17:18:27 +00001556 bool isClassMethod = !OMD->isInstanceMethod();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001557
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001558 CodeGenTypes &Types = CGM.getTypes();
Mike Stump1eb44332009-09-09 15:08:12 +00001559 const llvm::FunctionType *MethodTy =
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001560 Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001561 std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
1562 MethodName, isClassMethod);
1563
Daniel Dunbard6c93d72009-09-17 04:01:22 +00001564 llvm::Function *Method
Mike Stump1eb44332009-09-09 15:08:12 +00001565 = llvm::Function::Create(MethodTy,
1566 llvm::GlobalValue::InternalLinkage,
1567 FunctionName,
1568 &TheModule);
Chris Lattner391d77a2008-03-30 23:03:07 +00001569 return Method;
1570}
1571
Daniel Dunbar49f66022008-09-24 03:38:44 +00001572llvm::Function *CGObjCGNU::GetPropertyGetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001573 std::vector<const llvm::Type*> Params;
1574 const llvm::Type *BoolTy =
1575 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1576 Params.push_back(IdTy);
1577 Params.push_back(SelectorTy);
1578 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1579 Params.push_back(LongTy);
1580 Params.push_back(BoolTy);
1581 // void objc_getProperty (id, SEL, ptrdiff_t, bool)
1582 const llvm::FunctionType *FTy =
1583 llvm::FunctionType::get(IdTy, Params, false);
1584 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1585 "objc_getProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001586}
1587
1588llvm::Function *CGObjCGNU::GetPropertySetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001589 std::vector<const llvm::Type*> Params;
1590 const llvm::Type *BoolTy =
1591 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1592 Params.push_back(IdTy);
1593 Params.push_back(SelectorTy);
1594 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1595 Params.push_back(LongTy);
1596 Params.push_back(IdTy);
1597 Params.push_back(BoolTy);
1598 Params.push_back(BoolTy);
1599 // void objc_setProperty (id, SEL, ptrdiff_t, id, bool, bool)
1600 const llvm::FunctionType *FTy =
1601 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Params, false);
1602 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1603 "objc_setProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001604}
1605
Daniel Dunbar309a4362009-07-24 07:40:24 +00001606llvm::Constant *CGObjCGNU::EnumerationMutationFunction() {
1607 CodeGen::CodeGenTypes &Types = CGM.getTypes();
1608 ASTContext &Ctx = CGM.getContext();
1609 // void objc_enumerationMutation (id)
1610 llvm::SmallVector<QualType,16> Params;
David Chisnall0f436562009-08-17 16:35:33 +00001611 Params.push_back(ASTIdTy);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001612 const llvm::FunctionType *FTy =
1613 Types.GetFunctionType(Types.getFunctionInfo(Ctx.VoidTy, Params), false);
1614 return CGM.CreateRuntimeFunction(FTy, "objc_enumerationMutation");
Anders Carlsson2abd89c2008-08-31 04:05:03 +00001615}
1616
Fariborz Jahanianbd71be42008-11-21 00:49:24 +00001617void CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1618 const Stmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001619 // Pointer to the personality function
1620 llvm::Constant *Personality =
Owen Anderson0032b272009-08-13 21:57:51 +00001621 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getInt32Ty(VMContext),
Chris Lattnerb59761b2009-07-01 04:13:52 +00001622 true),
Chris Lattner5dc08672009-05-08 00:11:50 +00001623 "__gnu_objc_personality_v0");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001624 Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
Chris Lattner5dc08672009-05-08 00:11:50 +00001625 std::vector<const llvm::Type*> Params;
1626 Params.push_back(PtrTy);
1627 llvm::Value *RethrowFn =
Owen Anderson0032b272009-08-13 21:57:51 +00001628 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
David Chisnall90665bd2010-01-06 18:02:59 +00001629 Params, false), "_Unwind_Resume");
Chris Lattner5dc08672009-05-08 00:11:50 +00001630
1631 bool isTry = isa<ObjCAtTryStmt>(S);
1632 llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
1633 llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
1634 llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
Chris Lattnerd6a99072009-05-11 18:16:28 +00001635 llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
Chris Lattner5dc08672009-05-08 00:11:50 +00001636 llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
1637 llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
1638 llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
1639
David Chisnall0faa5162009-12-24 02:26:34 +00001640 // @synchronized()
Chris Lattner5dc08672009-05-08 00:11:50 +00001641 if (!isTry) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001642 std::vector<const llvm::Type*> Args(1, IdTy);
1643 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001644 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001645 llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
Mike Stump1eb44332009-09-09 15:08:12 +00001646 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001647 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1648 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1649 CGF.Builder.CreateCall(SyncEnter, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001650 }
1651
Chris Lattnerd6a99072009-05-11 18:16:28 +00001652
Chris Lattner5dc08672009-05-08 00:11:50 +00001653 // Push an EH context entry, used for handling rethrows and jumps
1654 // through finally.
1655 CGF.PushCleanupBlock(FinallyBlock);
1656
1657 // Emit the statements in the @try {} block
1658 CGF.setInvokeDest(TryHandler);
1659
1660 CGF.EmitBlock(TryBlock);
Mike Stump1eb44332009-09-09 15:08:12 +00001661 CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
Chris Lattner5dc08672009-05-08 00:11:50 +00001662 : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
1663
1664 // Jump to @finally if there is no exception
1665 CGF.EmitBranchThroughCleanup(FinallyEnd);
1666
1667 // Emit the handlers
1668 CGF.EmitBlock(TryHandler);
1669
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001670 // Get the correct versions of the exception handling intrinsics
1671 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
Mike Stump1eb44332009-09-09 15:08:12 +00001672 llvm::Value *llvm_eh_exception =
Chris Lattner5dc08672009-05-08 00:11:50 +00001673 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
Duncan Sands790b20e2009-10-14 16:13:30 +00001674 llvm::Value *llvm_eh_selector =
1675 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector);
1676 llvm::Value *llvm_eh_typeid_for =
1677 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for);
Chris Lattner5dc08672009-05-08 00:11:50 +00001678
1679 // Exception object
1680 llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1681 llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
1682
1683 llvm::SmallVector<llvm::Value*, 8> ESelArgs;
1684 llvm::SmallVector<std::pair<const ParmVarDecl*, const Stmt*>, 8> Handlers;
1685
1686 ESelArgs.push_back(Exc);
1687 ESelArgs.push_back(Personality);
1688
1689 bool HasCatchAll = false;
1690 // Only @try blocks are allowed @catch blocks, but both can have @finally
1691 if (isTry) {
1692 if (const ObjCAtCatchStmt* CatchStmt =
1693 cast<ObjCAtTryStmt>(S).getCatchStmts()) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001694 CGF.setInvokeDest(CatchInCatch);
Chris Lattner5dc08672009-05-08 00:11:50 +00001695
1696 for (; CatchStmt; CatchStmt = CatchStmt->getNextCatchStmt()) {
1697 const ParmVarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
Mike Stumpbb1c8602009-07-31 21:31:32 +00001698 Handlers.push_back(std::make_pair(CatchDecl,
1699 CatchStmt->getCatchBody()));
Chris Lattner5dc08672009-05-08 00:11:50 +00001700
1701 // @catch() and @catch(id) both catch any ObjC exception
Steve Naroff14108da2009-07-10 23:34:53 +00001702 if (!CatchDecl || CatchDecl->getType()->isObjCIdType()
Chris Lattner5dc08672009-05-08 00:11:50 +00001703 || CatchDecl->getType()->isObjCQualifiedIdType()) {
1704 // Use i8* null here to signal this is a catch all, not a cleanup.
1705 ESelArgs.push_back(NULLPtr);
1706 HasCatchAll = true;
1707 // No further catches after this one will ever by reached
1708 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001709 }
Chris Lattner5dc08672009-05-08 00:11:50 +00001710
1711 // All other types should be Objective-C interface pointer types.
Mike Stump1eb44332009-09-09 15:08:12 +00001712 const ObjCObjectPointerType *OPT =
John McCall183700f2009-09-21 23:43:11 +00001713 CatchDecl->getType()->getAs<ObjCObjectPointerType>();
Steve Naroff14108da2009-07-10 23:34:53 +00001714 assert(OPT && "Invalid @catch type.");
Mike Stump1eb44332009-09-09 15:08:12 +00001715 const ObjCInterfaceType *IT =
John McCall183700f2009-09-21 23:43:11 +00001716 OPT->getPointeeType()->getAs<ObjCInterfaceType>();
Chris Lattner5dc08672009-05-08 00:11:50 +00001717 assert(IT && "Invalid @catch type.");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001718 llvm::Value *EHType =
1719 MakeConstantString(IT->getDecl()->getNameAsString());
Chris Lattner5dc08672009-05-08 00:11:50 +00001720 ESelArgs.push_back(EHType);
1721 }
1722 }
1723 }
1724
1725 // We use a cleanup unless there was already a catch all.
1726 if (!HasCatchAll) {
Owen Anderson0032b272009-08-13 21:57:51 +00001727 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattner5dc08672009-05-08 00:11:50 +00001728 Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
1729 }
1730
1731 // Find which handler was matched.
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001732 llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
Chris Lattner5dc08672009-05-08 00:11:50 +00001733 ESelArgs.begin(), ESelArgs.end(), "selector");
1734
1735 for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
1736 const ParmVarDecl *CatchParam = Handlers[i].first;
1737 const Stmt *CatchBody = Handlers[i].second;
1738
1739 llvm::BasicBlock *Next = 0;
1740
1741 // The last handler always matches.
1742 if (i + 1 != e) {
1743 assert(CatchParam && "Only last handler can be a catch all.");
1744
1745 // Test whether this block matches the type for the selector and branch
1746 // to Match if it does, or to the next BB if it doesn't.
1747 llvm::BasicBlock *Match = CGF.createBasicBlock("match");
1748 Next = CGF.createBasicBlock("catch.next");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001749 llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
Chris Lattner5dc08672009-05-08 00:11:50 +00001750 CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
1751 CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
1752 Next);
1753
1754 CGF.EmitBlock(Match);
1755 }
Mike Stump1eb44332009-09-09 15:08:12 +00001756
Chris Lattner5dc08672009-05-08 00:11:50 +00001757 if (CatchBody) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001758 llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
1759 CGF.ConvertType(CatchParam->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001760
Chris Lattner5dc08672009-05-08 00:11:50 +00001761 // Bind the catch parameter if it exists.
1762 if (CatchParam) {
1763 // CatchParam is a ParmVarDecl because of the grammar
1764 // construction used to handle this, but for codegen purposes
1765 // we treat this as a local decl.
1766 CGF.EmitLocalBlockVarDecl(*CatchParam);
1767 CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
1768 }
1769
1770 CGF.ObjCEHValueStack.push_back(ExcObject);
1771 CGF.EmitStmt(CatchBody);
1772 CGF.ObjCEHValueStack.pop_back();
1773
1774 CGF.EmitBranchThroughCleanup(FinallyEnd);
1775
1776 if (Next)
1777 CGF.EmitBlock(Next);
1778 } else {
1779 assert(!Next && "catchup should be last handler.");
1780
1781 CGF.Builder.CreateStore(Exc, RethrowPtr);
1782 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1783 }
1784 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001785 // The @finally block is a secondary landing pad for any exceptions thrown in
1786 // @catch() blocks
1787 CGF.EmitBlock(CatchInCatch);
1788 Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1789 ESelArgs.clear();
1790 ESelArgs.push_back(Exc);
1791 ESelArgs.push_back(Personality);
David Chisnall0faa5162009-12-24 02:26:34 +00001792 // If there is a @catch or @finally clause in outside of this one then we
1793 // need to make sure that we catch and rethrow it.
1794 if (PrevLandingPad) {
1795 ESelArgs.push_back(NULLPtr);
1796 } else {
1797 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
1798 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001799 CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
1800 "selector");
1801 CGF.Builder.CreateCall(llvm_eh_typeid_for,
1802 CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
1803 CGF.Builder.CreateStore(Exc, RethrowPtr);
1804 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1805
Chris Lattner5dc08672009-05-08 00:11:50 +00001806 CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
1807
1808 CGF.setInvokeDest(PrevLandingPad);
1809
1810 CGF.EmitBlock(FinallyBlock);
1811
Chris Lattnerd6a99072009-05-11 18:16:28 +00001812
Chris Lattner5dc08672009-05-08 00:11:50 +00001813 if (isTry) {
Mike Stump1eb44332009-09-09 15:08:12 +00001814 if (const ObjCAtFinallyStmt* FinallyStmt =
Chris Lattner5dc08672009-05-08 00:11:50 +00001815 cast<ObjCAtTryStmt>(S).getFinallyStmt())
1816 CGF.EmitStmt(FinallyStmt->getFinallyBody());
1817 } else {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001818 // Emit 'objc_sync_exit(expr)' as finally's sole statement for
Chris Lattner5dc08672009-05-08 00:11:50 +00001819 // @synchronized.
Chris Lattnerd6a99072009-05-11 18:16:28 +00001820 std::vector<const llvm::Type*> Args(1, IdTy);
1821 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001822 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001823 llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
Mike Stump1eb44332009-09-09 15:08:12 +00001824 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001825 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1826 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1827 CGF.Builder.CreateCall(SyncExit, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001828 }
1829
1830 if (Info.SwitchBlock)
1831 CGF.EmitBlock(Info.SwitchBlock);
1832 if (Info.EndBlock)
1833 CGF.EmitBlock(Info.EndBlock);
1834
1835 // Branch around the rethrow code.
1836 CGF.EmitBranch(FinallyEnd);
1837
1838 CGF.EmitBlock(FinallyRethrow);
David Chisnall0faa5162009-12-24 02:26:34 +00001839
1840 llvm::Value *ExceptionObject = CGF.Builder.CreateLoad(RethrowPtr);
1841 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
1842 if (!UnwindBB) {
1843 CGF.Builder.CreateCall(RethrowFn, ExceptionObject);
1844 // Exception always thrown, next instruction is never reached.
1845 CGF.Builder.CreateUnreachable();
1846 } else {
1847 // If there is a @catch block outside this scope, we invoke instead of
1848 // calling because we may return to this function. This is very slow, but
1849 // some people still do it. It would be nice to add an optimised path for
1850 // this.
1851 CGF.Builder.CreateInvoke(RethrowFn, UnwindBB, UnwindBB, &ExceptionObject,
1852 &ExceptionObject+1);
1853 }
Mike Stump1eb44332009-09-09 15:08:12 +00001854
Chris Lattner5dc08672009-05-08 00:11:50 +00001855 CGF.EmitBlock(FinallyEnd);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001856}
1857
1858void CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar49f66022008-09-24 03:38:44 +00001859 const ObjCAtThrowStmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001860 llvm::Value *ExceptionAsObject;
1861
1862 std::vector<const llvm::Type*> Args(1, IdTy);
1863 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001864 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Mike Stump1eb44332009-09-09 15:08:12 +00001865 llvm::Value *ThrowFn =
Chris Lattner5dc08672009-05-08 00:11:50 +00001866 CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
Mike Stump1eb44332009-09-09 15:08:12 +00001867
Chris Lattner5dc08672009-05-08 00:11:50 +00001868 if (const Expr *ThrowExpr = S.getThrowExpr()) {
1869 llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001870 ExceptionAsObject = Exception;
Chris Lattner5dc08672009-05-08 00:11:50 +00001871 } else {
Mike Stump1eb44332009-09-09 15:08:12 +00001872 assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
Chris Lattner5dc08672009-05-08 00:11:50 +00001873 "Unexpected rethrow outside @catch block.");
1874 ExceptionAsObject = CGF.ObjCEHValueStack.back();
1875 }
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001876 ExceptionAsObject =
1877 CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
Mike Stump1eb44332009-09-09 15:08:12 +00001878
Chris Lattner5dc08672009-05-08 00:11:50 +00001879 // Note: This may have to be an invoke, if we want to support constructs like:
1880 // @try {
1881 // @throw(obj);
1882 // }
1883 // @catch(id) ...
1884 //
1885 // This is effectively turning @throw into an incredibly-expensive goto, but
1886 // it may happen as a result of inlining followed by missed optimizations, or
1887 // as a result of stupidity.
1888 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
1889 if (!UnwindBB) {
1890 CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
1891 CGF.Builder.CreateUnreachable();
1892 } else {
1893 CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
1894 &ExceptionAsObject+1);
1895 }
1896 // Clear the insertion point to indicate we are in unreachable code.
1897 CGF.Builder.ClearInsertionPoint();
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001898}
1899
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001900llvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001901 llvm::Value *AddrWeakObj) {
Fariborz Jahanian6dc23172008-11-18 21:45:40 +00001902 return 0;
1903}
1904
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001905void CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001906 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001907 return;
1908}
1909
Fariborz Jahanian58626502008-11-19 00:59:10 +00001910void CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001911 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian58626502008-11-19 00:59:10 +00001912 return;
1913}
1914
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00001915void CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +00001916 llvm::Value *src, llvm::Value *dst,
1917 llvm::Value *ivarOffset) {
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00001918 return;
1919}
1920
Fariborz Jahanian58626502008-11-19 00:59:10 +00001921void CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001922 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian58626502008-11-19 00:59:10 +00001923 return;
1924}
1925
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001926void CGObjCGNU::EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001927 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001928 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +00001929 QualType Ty) {
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001930 return;
1931}
1932
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001933llvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
1934 const ObjCInterfaceDecl *ID,
1935 const ObjCIvarDecl *Ivar) {
1936 const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
1937 + '.' + Ivar->getNameAsString();
1938 // Emit the variable and initialize it with what we think the correct value
1939 // is. This allows code compiled with non-fragile ivars to work correctly
1940 // when linked against code which isn't (most of the time).
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001941 llvm::GlobalVariable *IvarOffsetPointer = TheModule.getNamedGlobal(Name);
1942 if (!IvarOffsetPointer) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001943 uint64_t Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
1944 llvm::ConstantInt *OffsetGuess =
David Chisnallf9508372010-01-11 19:02:35 +00001945 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset, "ivar");
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001946 // Don't emit the guess in non-PIC code because the linker will not be able
1947 // to replace it with the real version for a library. In non-PIC code you
1948 // must compile with the fragile ABI if you want to use ivars from a
Mike Stump1eb44332009-09-09 15:08:12 +00001949 // GCC-compiled class.
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001950 if (CGM.getLangOptions().PICLevel) {
1951 llvm::GlobalVariable *IvarOffsetGV = new llvm::GlobalVariable(TheModule,
1952 llvm::Type::getInt32Ty(VMContext), false,
1953 llvm::GlobalValue::PrivateLinkage, OffsetGuess, Name+".guess");
1954 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
1955 IvarOffsetGV->getType(), false, llvm::GlobalValue::LinkOnceAnyLinkage,
1956 IvarOffsetGV, Name);
1957 } else {
1958 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
Benjamin Kramer3c0ef8c2009-10-13 10:07:13 +00001959 llvm::Type::getInt32PtrTy(VMContext), false,
1960 llvm::GlobalValue::ExternalLinkage, 0, Name);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001961 }
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001962 }
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001963 return IvarOffsetPointer;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001964}
1965
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00001966LValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
1967 QualType ObjectTy,
1968 llvm::Value *BaseValue,
1969 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00001970 unsigned CVRQualifiers) {
John McCall183700f2009-09-21 23:43:11 +00001971 const ObjCInterfaceDecl *ID = ObjectTy->getAs<ObjCInterfaceType>()->getDecl();
Daniel Dunbar97776872009-04-22 07:32:20 +00001972 return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
1973 EmitIvarOffset(CGF, ID, Ivar));
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00001974}
Mike Stumpbb1c8602009-07-31 21:31:32 +00001975
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001976static const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
1977 const ObjCInterfaceDecl *OID,
1978 const ObjCIvarDecl *OIVD) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001979 llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
Fariborz Jahanian8e6ac1d2009-06-04 01:19:09 +00001980 Context.ShallowCollectObjCIvars(OID, Ivars);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001981 for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
1982 if (OIVD == Ivars[k])
1983 return OID;
1984 }
Mike Stump1eb44332009-09-09 15:08:12 +00001985
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001986 // Otherwise check in the super class.
1987 if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
1988 return FindIvarInterface(Context, Super, OIVD);
Mike Stump1eb44332009-09-09 15:08:12 +00001989
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001990 return 0;
1991}
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00001992
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001993llvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +00001994 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001995 const ObjCIvarDecl *Ivar) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001996 if (CGM.getLangOptions().ObjCNonFragileABI) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001997 Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001998 return CGF.Builder.CreateLoad(CGF.Builder.CreateLoad(
1999 ObjCIvarOffsetVariable(Interface, Ivar), false, "ivar"));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002000 }
Daniel Dunbar97776872009-04-22 07:32:20 +00002001 uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00002002 return llvm::ConstantInt::get(LongTy, Offset, "ivar");
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002003}
2004
Mike Stumpbb1c8602009-07-31 21:31:32 +00002005CodeGen::CGObjCRuntime *
2006CodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM) {
Chris Lattnerdce14062008-06-26 04:19:03 +00002007 return new CGObjCGNU(CGM);
Chris Lattner0f984262008-03-01 08:50:34 +00002008}