blob: 1d38ef9e2d2ff922a75340bf093c263fcf035382 [file] [log] [blame]
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001//===------- CGObjCGNU.cpp - Emit LLVM Code from ASTs for a Module --------===//
Chris Lattner0f984262008-03-01 08:50:34 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000010// This provides Objective-C code generation targetting the GNU runtime. The
11// class in this file generates structures used by the GNU Objective-C runtime
12// library. These structures are defined in objc/objc.h and objc/objc-api.h in
13// the GNU runtime distribution.
Chris Lattner0f984262008-03-01 08:50:34 +000014//
15//===----------------------------------------------------------------------===//
16
17#include "CGObjCRuntime.h"
Chris Lattnerdce14062008-06-26 04:19:03 +000018#include "CodeGenModule.h"
Daniel Dunbar8f2926b2008-08-23 03:46:30 +000019#include "CodeGenFunction.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000020
Chris Lattnerdce14062008-06-26 04:19:03 +000021#include "clang/AST/ASTContext.h"
Daniel Dunbare91593e2008-08-11 04:54:23 +000022#include "clang/AST/Decl.h"
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +000023#include "clang/AST/DeclObjC.h"
Anders Carlsson19cc4ab2009-07-18 19:43:29 +000024#include "clang/AST/RecordLayout.h"
Chris Lattner16f00492009-04-26 01:32:48 +000025#include "clang/AST/StmtObjC.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000026
27#include "llvm/Intrinsics.h"
Chris Lattner0f984262008-03-01 08:50:34 +000028#include "llvm/Module.h"
Chris Lattner0f984262008-03-01 08:50:34 +000029#include "llvm/ADT/SmallVector.h"
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000030#include "llvm/ADT/StringMap.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000031#include "llvm/Support/Compiler.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000032#include "llvm/Target/TargetData.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000033
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000034#include <map>
Chris Lattnere160c9b2009-01-27 05:06:01 +000035
36
Chris Lattnerdce14062008-06-26 04:19:03 +000037using namespace clang;
Daniel Dunbar46f45b92008-09-09 01:06:48 +000038using namespace CodeGen;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000039using llvm::dyn_cast;
40
41// The version of the runtime that this class targets. Must match the version
42// in the runtime.
43static const int RuntimeVersion = 8;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +000044static const int NonFragileRuntimeVersion = 9;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000045static const int ProtocolVersion = 2;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000046static const int NonFragileProtocolVersion = 3;
Chris Lattner0f984262008-03-01 08:50:34 +000047
Chris Lattner0f984262008-03-01 08:50:34 +000048namespace {
Chris Lattnerdce14062008-06-26 04:19:03 +000049class CGObjCGNU : public CodeGen::CGObjCRuntime {
Chris Lattner0f984262008-03-01 08:50:34 +000050private:
Chris Lattnerdce14062008-06-26 04:19:03 +000051 CodeGen::CodeGenModule &CGM;
Chris Lattner0f984262008-03-01 08:50:34 +000052 llvm::Module &TheModule;
Chris Lattnere160c9b2009-01-27 05:06:01 +000053 const llvm::PointerType *SelectorTy;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000054 const llvm::IntegerType *Int8Ty;
Chris Lattnere160c9b2009-01-27 05:06:01 +000055 const llvm::PointerType *PtrToInt8Ty;
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +000056 const llvm::FunctionType *IMPTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000057 const llvm::PointerType *IdTy;
David Chisnallef6e0f32010-02-03 15:59:02 +000058 const llvm::PointerType *PtrToIdTy;
David Chisnall0f436562009-08-17 16:35:33 +000059 QualType ASTIdTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000060 const llvm::IntegerType *IntTy;
61 const llvm::PointerType *PtrTy;
62 const llvm::IntegerType *LongTy;
63 const llvm::PointerType *PtrToIntTy;
Daniel Dunbar5efccb12009-05-04 15:31:17 +000064 llvm::GlobalAlias *ClassPtrAlias;
65 llvm::GlobalAlias *MetaClassPtrAlias;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000066 std::vector<llvm::Constant*> Classes;
67 std::vector<llvm::Constant*> Categories;
68 std::vector<llvm::Constant*> ConstantStrings;
David Chisnall48272a02010-01-27 12:49:23 +000069 llvm::StringMap<llvm::Constant*> ObjCStrings;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000070 llvm::Function *LoadFunction;
71 llvm::StringMap<llvm::Constant*> ExistingProtocols;
72 typedef std::pair<std::string, std::string> TypedSelector;
73 std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
74 llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
David Chisnallef6e0f32010-02-03 15:59:02 +000075 // Selectors that we don't emit in GC mode
76 Selector RetainSel, ReleaseSel, AutoreleaseSel;
77 // Functions used for GC.
78 llvm::Constant *IvarAssignFn, *StrongCastAssignFn, *MemMoveFn, *WeakReadFn,
79 *WeakAssignFn, *GlobalAssignFn;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000080 // Some zeros used for GEPs in lots of places.
81 llvm::Constant *Zeros[2];
82 llvm::Constant *NULLPtr;
Owen Andersona1cf15f2009-07-14 23:10:40 +000083 llvm::LLVMContext &VMContext;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000084private:
85 llvm::Constant *GenerateIvarList(
86 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
87 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
88 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets);
89 llvm::Constant *GenerateMethodList(const std::string &ClassName,
90 const std::string &CategoryName,
Mike Stump1eb44332009-09-09 15:08:12 +000091 const llvm::SmallVectorImpl<Selector> &MethodSels,
92 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000093 bool isClassMethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +000094 llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000095 llvm::Constant *GeneratePropertyList(const ObjCImplementationDecl *OID,
96 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
97 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000098 llvm::Constant *GenerateProtocolList(
99 const llvm::SmallVectorImpl<std::string> &Protocols);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000100 // To ensure that all protocols are seen by the runtime, we add a category on
101 // a class defined in the runtime, declaring no methods, but adopting the
102 // protocols.
103 void GenerateProtocolHolderCategory(void);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000104 llvm::Constant *GenerateClassStructure(
105 llvm::Constant *MetaClass,
106 llvm::Constant *SuperClass,
107 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000108 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000109 llvm::Constant *Version,
110 llvm::Constant *InstanceSize,
111 llvm::Constant *IVars,
112 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000113 llvm::Constant *Protocols,
114 llvm::Constant *IvarOffsets,
115 llvm::Constant *Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000116 llvm::Constant *GenerateProtocolMethodList(
117 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
118 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes);
119 llvm::Constant *MakeConstantString(const std::string &Str, const std::string
120 &Name="");
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000121 llvm::Constant *ExportUniqueString(const std::string &Str, const std::string
122 prefix);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000123 llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000124 std::vector<llvm::Constant*> &V, llvm::StringRef Name="",
David Chisnall41d63ed2010-01-08 00:14:31 +0000125 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000126 llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000127 std::vector<llvm::Constant*> &V, llvm::StringRef Name="",
David Chisnall41d63ed2010-01-08 00:14:31 +0000128 llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000129 llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
130 const ObjCIvarDecl *Ivar);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000131 void EmitClassRef(const std::string &className);
David Chisnallef6e0f32010-02-03 15:59:02 +0000132 llvm::Value* EnforceType(CGBuilderTy B, llvm::Value *V, const llvm::Type *Ty){
133 if (V->getType() == Ty) return V;
134 return B.CreateBitCast(V, Ty);
135 }
Chris Lattner0f984262008-03-01 08:50:34 +0000136public:
Chris Lattnerdce14062008-06-26 04:19:03 +0000137 CGObjCGNU(CodeGen::CodeGenModule &cgm);
David Chisnall0d13f6f2010-01-23 02:40:42 +0000138 virtual llvm::Constant *GenerateConstantString(const StringLiteral *);
Mike Stump1eb44332009-09-09 15:08:12 +0000139 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000140 GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000141 QualType ResultType,
142 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000143 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000144 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000145 const CallArgList &CallArgs,
146 const ObjCMethodDecl *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000147 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000148 GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000149 QualType ResultType,
150 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000151 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000152 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000153 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000154 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000155 const CallArgList &CallArgs,
156 const ObjCMethodDecl *Method);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000157 virtual llvm::Value *GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000158 const ObjCInterfaceDecl *OID);
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000159 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel);
160 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
161 *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000162
163 virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000164 const ObjCContainerDecl *CD);
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000165 virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
166 virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000167 virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000168 const ObjCProtocolDecl *PD);
169 virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000170 virtual llvm::Function *ModuleInitFunction();
Daniel Dunbar49f66022008-09-24 03:38:44 +0000171 virtual llvm::Function *GetPropertyGetFunction();
172 virtual llvm::Function *GetPropertySetFunction();
Daniel Dunbar309a4362009-07-24 07:40:24 +0000173 virtual llvm::Constant *EnumerationMutationFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000174
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000175 virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
176 const Stmt &S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000177 virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
178 const ObjCAtThrowStmt &S);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000179 virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6dc23172008-11-18 21:45:40 +0000180 llvm::Value *AddrWeakObj);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000181 virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
182 llvm::Value *src, llvm::Value *dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000183 virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
184 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +0000185 virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +0000186 llvm::Value *src, llvm::Value *dest,
187 llvm::Value *ivarOffset);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000188 virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
189 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000190 virtual void EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +0000191 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000192 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +0000193 QualType Ty);
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000194 virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
195 QualType ObjectTy,
196 llvm::Value *BaseValue,
197 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000198 unsigned CVRQualifiers);
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000199 virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +0000200 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000201 const ObjCIvarDecl *Ivar);
Chris Lattner0f984262008-03-01 08:50:34 +0000202};
203} // end anonymous namespace
204
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000205
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000206/// Emits a reference to a dummy variable which is emitted with each class.
207/// This ensures that a linker error will be generated when trying to link
208/// together modules where a referenced class is not defined.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000209void CGObjCGNU::EmitClassRef(const std::string &className) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000210 std::string symbolRef = "__objc_class_ref_" + className;
211 // Don't emit two copies of the same symbol
Mike Stumpbb1c8602009-07-31 21:31:32 +0000212 if (TheModule.getGlobalVariable(symbolRef))
213 return;
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000214 std::string symbolName = "__objc_class_name_" + className;
215 llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
216 if (!ClassSymbol) {
Owen Anderson1c431b32009-07-08 19:05:04 +0000217 ClassSymbol = new llvm::GlobalVariable(TheModule, LongTy, false,
218 llvm::GlobalValue::ExternalLinkage, 0, symbolName);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000219 }
Owen Anderson1c431b32009-07-08 19:05:04 +0000220 new llvm::GlobalVariable(TheModule, ClassSymbol->getType(), true,
Chris Lattnerf35271b2009-08-05 05:25:18 +0000221 llvm::GlobalValue::WeakAnyLinkage, ClassSymbol, symbolRef);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000222}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000223
224static std::string SymbolNameForClass(const std::string &ClassName) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000225 return "_OBJC_CLASS_" + ClassName;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000226}
227
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000228static std::string SymbolNameForMethod(const std::string &ClassName, const
229 std::string &CategoryName, const std::string &MethodName, bool isClassMethod)
230{
David Chisnalld3467362010-01-14 14:08:19 +0000231 std::string MethodNameColonStripped = MethodName;
232 std::replace(MethodNameColonStripped.begin(), MethodNameColonStripped.end(),
233 ':', '_');
234 return std::string(isClassMethod ? "_c_" : "_i_") + ClassName + "_" +
235 CategoryName + "_" + MethodNameColonStripped;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000236}
237
Chris Lattnerdce14062008-06-26 04:19:03 +0000238CGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000239 : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
Owen Andersona1cf15f2009-07-14 23:10:40 +0000240 MetaClassPtrAlias(0), VMContext(cgm.getLLVMContext()) {
Chris Lattnere160c9b2009-01-27 05:06:01 +0000241 IntTy = cast<llvm::IntegerType>(
242 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
243 LongTy = cast<llvm::IntegerType>(
244 CGM.getTypes().ConvertType(CGM.getContext().LongTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000245
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000246 Int8Ty = llvm::Type::getInt8Ty(VMContext);
247 // C string type. Used in lots of places.
248 PtrToInt8Ty = llvm::PointerType::getUnqual(Int8Ty);
249
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000250 Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000251 Zeros[1] = Zeros[0];
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000252 NULLPtr = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Chris Lattner391d77a2008-03-30 23:03:07 +0000253 // Get the selector Type.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000254 QualType selTy = CGM.getContext().getObjCSelType();
255 if (QualType() == selTy) {
256 SelectorTy = PtrToInt8Ty;
257 } else {
258 SelectorTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(selTy));
259 }
Chris Lattnere160c9b2009-01-27 05:06:01 +0000260
Owen Anderson96e0fc72009-07-29 22:16:19 +0000261 PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
Chris Lattner391d77a2008-03-30 23:03:07 +0000262 PtrTy = PtrToInt8Ty;
Mike Stump1eb44332009-09-09 15:08:12 +0000263
Chris Lattner391d77a2008-03-30 23:03:07 +0000264 // Object type
David Chisnall0f436562009-08-17 16:35:33 +0000265 ASTIdTy = CGM.getContext().getObjCIdType();
David Chisnall0d13f6f2010-01-23 02:40:42 +0000266 if (QualType() == ASTIdTy) {
267 IdTy = PtrToInt8Ty;
268 } else {
269 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
270 }
David Chisnallef6e0f32010-02-03 15:59:02 +0000271 PtrToIdTy = llvm::PointerType::getUnqual(IdTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000272
Chris Lattner391d77a2008-03-30 23:03:07 +0000273 // IMP type
274 std::vector<const llvm::Type*> IMPArgs;
275 IMPArgs.push_back(IdTy);
276 IMPArgs.push_back(SelectorTy);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000277 IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
David Chisnallef6e0f32010-02-03 15:59:02 +0000278
279 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
280 // Get selectors needed in GC mode
281 RetainSel = GetNullarySelector("retain", CGM.getContext());
282 ReleaseSel = GetNullarySelector("release", CGM.getContext());
283 AutoreleaseSel = GetNullarySelector("autorelease", CGM.getContext());
284
285 // Get functions needed in GC mode
286
287 // id objc_assign_ivar(id, id, ptrdiff_t);
288 std::vector<const llvm::Type*> Args(1, IdTy);
289 Args.push_back(PtrToIdTy);
290 // FIXME: ptrdiff_t
291 Args.push_back(LongTy);
292 llvm::FunctionType *FTy = llvm::FunctionType::get(IdTy, Args, false);
293 IvarAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_ivar");
294 // id objc_assign_strongCast (id, id*)
295 Args.pop_back();
296 FTy = llvm::FunctionType::get(IdTy, Args, false);
297 StrongCastAssignFn =
298 CGM.CreateRuntimeFunction(FTy, "objc_assign_strongCast");
299 // id objc_assign_global(id, id*);
300 FTy = llvm::FunctionType::get(IdTy, Args, false);
301 GlobalAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_global");
302 // id objc_assign_weak(id, id*);
303 FTy = llvm::FunctionType::get(IdTy, Args, false);
304 WeakAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_weak");
305 // id objc_read_weak(id*);
306 Args.clear();
307 Args.push_back(PtrToIdTy);
308 FTy = llvm::FunctionType::get(IdTy, Args, false);
309 WeakReadFn = CGM.CreateRuntimeFunction(FTy, "objc_read_weak");
310 // void *objc_memmove_collectable(void*, void *, size_t);
311 Args.clear();
312 Args.push_back(PtrToInt8Ty);
313 Args.push_back(PtrToInt8Ty);
314 // FIXME: size_t
315 Args.push_back(LongTy);
316 FTy = llvm::FunctionType::get(IdTy, Args, false);
317 MemMoveFn = CGM.CreateRuntimeFunction(FTy, "objc_memmove_collectable");
318 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000319}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000320
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000321// This has to perform the lookup every time, since posing and related
322// techniques can modify the name -> class mapping.
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000323llvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000324 const ObjCInterfaceDecl *OID) {
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000325 llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
David Chisnall41d63ed2010-01-08 00:14:31 +0000326 // With the incompatible ABI, this will need to be replaced with a direct
327 // reference to the class symbol. For the compatible nonfragile ABI we are
328 // still performing this lookup at run time but emitting the symbol for the
329 // class externally so that we can make the switch later.
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000330 EmitClassRef(OID->getNameAsString());
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000331 ClassName = Builder.CreateStructGEP(ClassName, 0);
332
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000333 std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000334 llvm::Constant *ClassLookupFn =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000335 CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000336 Params,
337 true),
338 "objc_lookup_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000339 return Builder.CreateCall(ClassLookupFn, ClassName);
Chris Lattner391d77a2008-03-30 23:03:07 +0000340}
341
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000342llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel) {
Chris Lattner077bf5e2008-11-24 03:33:13 +0000343 llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
Chris Lattner8e67b632008-06-26 04:37:12 +0000344 if (US == 0)
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000345 US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000346 llvm::GlobalValue::PrivateLinkage,
347 ".objc_untyped_selector_alias"+Sel.getAsString(),
Chris Lattner8e67b632008-06-26 04:37:12 +0000348 NULL, &TheModule);
Mike Stump1eb44332009-09-09 15:08:12 +0000349
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000350 return Builder.CreateLoad(US);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000351}
352
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000353llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000354 *Method) {
355
356 std::string SelName = Method->getSelector().getAsString();
357 std::string SelTypes;
358 CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
359 // Typed selectors
360 TypedSelector Selector = TypedSelector(SelName,
361 SelTypes);
362
363 // If it's already cached, return it.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000364 if (TypedSelectors[Selector]) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000365 return Builder.CreateLoad(TypedSelectors[Selector]);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000366 }
367
368 // If it isn't, cache it.
369 llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000370 llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000371 llvm::GlobalValue::PrivateLinkage, ".objc_selector_alias" + SelName,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000372 NULL, &TheModule);
373 TypedSelectors[Selector] = Sel;
374
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +0000375 return Builder.CreateLoad(Sel);
Chris Lattner8e67b632008-06-26 04:37:12 +0000376}
377
Chris Lattner5e7dcc62008-06-26 04:44:19 +0000378llvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
379 const std::string &Name) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000380 llvm::Constant *ConstStr = CGM.GetAddrOfConstantCString(Str, Name.c_str());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000381 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000382}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000383llvm::Constant *CGObjCGNU::ExportUniqueString(const std::string &Str,
384 const std::string prefix) {
385 std::string name = prefix + Str;
386 llvm::Constant *ConstStr = TheModule.getGlobalVariable(name);
387 if (!ConstStr) {
388 llvm::Constant *value = llvm::ConstantArray::get(VMContext, Str, true);
389 ConstStr = new llvm::GlobalVariable(TheModule, value->getType(), true,
390 llvm::GlobalValue::LinkOnceODRLinkage, value, prefix + Str);
391 }
392 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
393}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000394
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000395llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000396 std::vector<llvm::Constant*> &V, llvm::StringRef Name,
David Chisnall41d63ed2010-01-08 00:14:31 +0000397 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson08e25242009-07-27 22:29:56 +0000398 llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000399 return new llvm::GlobalVariable(TheModule, Ty, false,
400 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000401}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000402
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000403llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +0000404 std::vector<llvm::Constant*> &V, llvm::StringRef Name,
David Chisnall41d63ed2010-01-08 00:14:31 +0000405 llvm::GlobalValue::LinkageTypes linkage) {
Owen Anderson7db6d832009-07-28 18:33:04 +0000406 llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000407 return new llvm::GlobalVariable(TheModule, Ty, false,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000408 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000409}
410
411/// Generate an NSConstantString object.
David Chisnall0d13f6f2010-01-23 02:40:42 +0000412llvm::Constant *CGObjCGNU::GenerateConstantString(const StringLiteral *SL) {
David Chisnall48272a02010-01-27 12:49:23 +0000413
David Chisnall0d13f6f2010-01-23 02:40:42 +0000414 std::string Str(SL->getStrData(), SL->getByteLength());
415
David Chisnall48272a02010-01-27 12:49:23 +0000416 // Look for an existing one
417 llvm::StringMap<llvm::Constant*>::iterator old = ObjCStrings.find(Str);
418 if (old != ObjCStrings.end())
419 return old->getValue();
420
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000421 std::vector<llvm::Constant*> Ivars;
422 Ivars.push_back(NULLPtr);
Chris Lattner13fd7e52008-06-21 21:44:18 +0000423 Ivars.push_back(MakeConstantString(Str));
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000424 Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000425 llvm::Constant *ObjCStr = MakeGlobal(
Owen Anderson47a434f2009-08-05 23:18:46 +0000426 llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000427 Ivars, ".objc_str");
David Chisnall48272a02010-01-27 12:49:23 +0000428 ObjCStr = llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty);
429 ObjCStrings[Str] = ObjCStr;
430 ConstantStrings.push_back(ObjCStr);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000431 return ObjCStr;
432}
433
434///Generates a message send where the super is the receiver. This is a message
435///send to self with special delivery semantics indicating which class's method
436///should be called.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000437CodeGen::RValue
438CGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000439 QualType ResultType,
440 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000441 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000442 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000443 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000444 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000445 const CallArgList &CallArgs,
446 const ObjCMethodDecl *Method) {
David Chisnallef6e0f32010-02-03 15:59:02 +0000447 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
448 if (Sel == RetainSel || Sel == AutoreleaseSel) {
449 return RValue::get(Receiver);
450 }
451 if (Sel == ReleaseSel) {
452 return RValue::get(0);
453 }
454 }
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000455 llvm::Value *cmd = GetSelector(CGF.Builder, Sel);
456
457 CallArgList ActualArgs;
458
459 ActualArgs.push_back(
David Chisnall0f436562009-08-17 16:35:33 +0000460 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
461 ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000462 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
463 CGF.getContext().getObjCSelType()));
464 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
465
466 CodeGenTypes &Types = CGM.getTypes();
John McCall04a67a62010-02-05 21:31:56 +0000467 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs,
468 CC_Default, false);
Daniel Dunbar67939662009-09-17 04:01:40 +0000469 const llvm::FunctionType *impType =
470 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000471
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000472 llvm::Value *ReceiverClass = 0;
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000473 if (isCategoryImpl) {
474 llvm::Constant *classLookupFunction = 0;
475 std::vector<const llvm::Type*> Params;
476 Params.push_back(PtrTy);
477 if (IsClassMessage) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000478 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000479 IdTy, Params, true), "objc_get_meta_class");
480 } else {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000481 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000482 IdTy, Params, true), "objc_get_class");
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000483 }
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000484 ReceiverClass = CGF.Builder.CreateCall(classLookupFunction,
485 MakeConstantString(Class->getNameAsString()));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000486 } else {
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000487 // Set up global aliases for the metaclass or class pointer if they do not
488 // already exist. These will are forward-references which will be set to
Mike Stumpbb1c8602009-07-31 21:31:32 +0000489 // pointers to the class and metaclass structure created for the runtime
490 // load function. To send a message to super, we look up the value of the
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000491 // super_class pointer from either the class or metaclass structure.
492 if (IsClassMessage) {
493 if (!MetaClassPtrAlias) {
494 MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
495 llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
496 Class->getNameAsString(), NULL, &TheModule);
497 }
498 ReceiverClass = MetaClassPtrAlias;
499 } else {
500 if (!ClassPtrAlias) {
501 ClassPtrAlias = new llvm::GlobalAlias(IdTy,
502 llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
503 Class->getNameAsString(), NULL, &TheModule);
504 }
505 ReceiverClass = ClassPtrAlias;
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000506 }
Chris Lattner71238f62009-04-25 23:19:45 +0000507 }
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000508 // Cast the pointer to a simplified version of the class structure
Mike Stump1eb44332009-09-09 15:08:12 +0000509 ReceiverClass = CGF.Builder.CreateBitCast(ReceiverClass,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000510 llvm::PointerType::getUnqual(
Owen Anderson47a434f2009-08-05 23:18:46 +0000511 llvm::StructType::get(VMContext, IdTy, IdTy, NULL)));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000512 // Get the superclass pointer
513 ReceiverClass = CGF.Builder.CreateStructGEP(ReceiverClass, 1);
514 // Load the superclass pointer
515 ReceiverClass = CGF.Builder.CreateLoad(ReceiverClass);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000516 // Construct the structure used to look up the IMP
Owen Anderson47a434f2009-08-05 23:18:46 +0000517 llvm::StructType *ObjCSuperTy = llvm::StructType::get(VMContext,
518 Receiver->getType(), IdTy, NULL);
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000519 llvm::Value *ObjCSuper = CGF.Builder.CreateAlloca(ObjCSuperTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000520
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000521 CGF.Builder.CreateStore(Receiver, CGF.Builder.CreateStructGEP(ObjCSuper, 0));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000522 CGF.Builder.CreateStore(ReceiverClass,
523 CGF.Builder.CreateStructGEP(ObjCSuper, 1));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000524
525 // Get the IMP
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000526 std::vector<const llvm::Type*> Params;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000527 Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000528 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000529 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000530 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
531 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000532 "objc_msg_lookup_super");
533
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000534 llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000535 llvm::Value *imp = CGF.Builder.CreateCall(lookupFunction, lookupArgs,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000536 lookupArgs+2);
537
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000538 return CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000539}
540
Mike Stump1eb44332009-09-09 15:08:12 +0000541/// Generate code for a message send expression.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000542CodeGen::RValue
543CGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000544 QualType ResultType,
545 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000546 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000547 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000548 const CallArgList &CallArgs,
549 const ObjCMethodDecl *Method) {
David Chisnallef6e0f32010-02-03 15:59:02 +0000550 if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
551 if (Sel == RetainSel || Sel == AutoreleaseSel) {
552 return RValue::get(Receiver);
553 }
554 if (Sel == ReleaseSel) {
555 return RValue::get(0);
556 }
557 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000558 CGBuilderTy &Builder = CGF.Builder;
David Chisnall0f436562009-08-17 16:35:33 +0000559 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000560 llvm::Value *cmd;
561 if (Method)
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000562 cmd = GetSelector(Builder, Method);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000563 else
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000564 cmd = GetSelector(Builder, Sel);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000565 CallArgList ActualArgs;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000566
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000567 Receiver = Builder.CreateBitCast(Receiver, IdTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000568 ActualArgs.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000569 std::make_pair(RValue::get(Receiver), ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000570 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
571 CGF.getContext().getObjCSelType()));
572 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
573
574 CodeGenTypes &Types = CGM.getTypes();
John McCall04a67a62010-02-05 21:31:56 +0000575 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs,
576 CC_Default, false);
Daniel Dunbar67939662009-09-17 04:01:40 +0000577 const llvm::FunctionType *impType =
578 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000579
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000580 llvm::Value *imp;
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000581 // For sender-aware dispatch, we pass the sender as the third argument to a
582 // lookup function. When sending messages from C code, the sender is nil.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000583 // objc_msg_lookup_sender(id *receiver, SEL selector, id sender);
584 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
585
586 std::vector<const llvm::Type*> Params;
587 llvm::Value *ReceiverPtr = CGF.CreateTempAlloca(Receiver->getType());
588 Builder.CreateStore(Receiver, ReceiverPtr);
589 Params.push_back(ReceiverPtr->getType());
590 Params.push_back(SelectorTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000591 llvm::Value *self;
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000592
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000593 if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
594 self = CGF.LoadObjCSelf();
595 } else {
Owen Anderson03e20502009-07-30 23:11:26 +0000596 self = llvm::ConstantPointerNull::get(IdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000597 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000598
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000599 Params.push_back(self->getType());
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000600
601 // The lookup function returns a slot, which can be safely cached.
602 llvm::Type *SlotTy = llvm::StructType::get(VMContext, PtrTy, PtrTy, PtrTy,
603 IntTy, llvm::PointerType::getUnqual(impType), NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000604 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000605 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000606 llvm::PointerType::getUnqual(SlotTy), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000607 "objc_msg_lookup_sender");
608
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000609 // The lookup function is guaranteed not to capture the receiver pointer.
610 if (llvm::Function *LookupFn = dyn_cast<llvm::Function>(lookupFunction)) {
611 LookupFn->setDoesNotCapture(1);
612 }
613
614 llvm::Value *slot =
615 Builder.CreateCall3(lookupFunction, ReceiverPtr, cmd, self);
616 imp = Builder.CreateLoad(Builder.CreateStructGEP(slot, 4));
617 // The lookup function may have changed the receiver, so make sure we use
618 // the new one.
619 ActualArgs[0] =
620 std::make_pair(RValue::get(Builder.CreateLoad(ReceiverPtr)), ASTIdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000621 } else {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000622 std::vector<const llvm::Type*> Params;
623 Params.push_back(Receiver->getType());
624 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000625 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000626 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
627 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000628 "objc_msg_lookup");
629
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000630 imp = Builder.CreateCall2(lookupFunction, Receiver, cmd);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000631 }
Chris Lattner3eae03e2008-05-06 00:56:42 +0000632
Anders Carlssonf3c47c92009-12-24 19:25:24 +0000633 return CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
Chris Lattner0f984262008-03-01 08:50:34 +0000634}
635
Mike Stump1eb44332009-09-09 15:08:12 +0000636/// Generates a MethodList. Used in construction of a objc_class and
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000637/// objc_category structures.
638llvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
Mike Stump1eb44332009-09-09 15:08:12 +0000639 const std::string &CategoryName,
640 const llvm::SmallVectorImpl<Selector> &MethodSels,
641 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000642 bool isClassMethodList) {
David Chisnall0f436562009-08-17 16:35:33 +0000643 if (MethodSels.empty())
644 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000645 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000646 llvm::StructType *ObjCMethodTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000647 PtrToInt8Ty, // Really a selector, but the runtime creates it us.
648 PtrToInt8Ty, // Method types
Owen Anderson96e0fc72009-07-29 22:16:19 +0000649 llvm::PointerType::getUnqual(IMPTy), //Method pointer
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000650 NULL);
651 std::vector<llvm::Constant*> Methods;
652 std::vector<llvm::Constant*> Elements;
653 for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
654 Elements.clear();
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000655 if (llvm::Constant *Method =
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000656 TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
Chris Lattner077bf5e2008-11-24 03:33:13 +0000657 MethodSels[i].getAsString(),
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000658 isClassMethodList))) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000659 llvm::Constant *C = MakeConstantString(MethodSels[i].getAsString());
660 Elements.push_back(C);
661 Elements.push_back(MethodTypes[i]);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000662 Method = llvm::ConstantExpr::getBitCast(Method,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000663 llvm::PointerType::getUnqual(IMPTy));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000664 Elements.push_back(Method);
Owen Anderson08e25242009-07-27 22:29:56 +0000665 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000666 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000667 }
668
669 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000670 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000671 Methods.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000672 llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
Chris Lattnerfba67632008-06-26 04:52:29 +0000673 Methods);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000674
675 // Structure containing list pointer, array and array count
676 llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
Owen Anderson8c8f69e2009-08-13 23:27:53 +0000677 llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get(VMContext);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000678 llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
Owen Anderson47a434f2009-08-05 23:18:46 +0000679 llvm::StructType *ObjCMethodListTy = llvm::StructType::get(VMContext,
Mike Stump1eb44332009-09-09 15:08:12 +0000680 NextPtrTy,
681 IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000682 ObjCMethodArrayTy,
683 NULL);
684 // Refine next pointer type to concrete type
685 llvm::cast<llvm::OpaqueType>(
686 OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
687 ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
688
689 Methods.clear();
Owen Anderson03e20502009-07-30 23:11:26 +0000690 Methods.push_back(llvm::ConstantPointerNull::get(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000691 llvm::PointerType::getUnqual(ObjCMethodListTy)));
Owen Anderson0032b272009-08-13 21:57:51 +0000692 Methods.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000693 MethodTypes.size()));
694 Methods.push_back(MethodArray);
Mike Stump1eb44332009-09-09 15:08:12 +0000695
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000696 // Create an instance of the structure
697 return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
698}
699
700/// Generates an IvarList. Used in construction of a objc_class.
701llvm::Constant *CGObjCGNU::GenerateIvarList(
702 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
703 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
704 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets) {
David Chisnall18044632009-11-16 19:05:54 +0000705 if (IvarNames.size() == 0)
706 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000707 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000708 llvm::StructType *ObjCIvarTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000709 PtrToInt8Ty,
710 PtrToInt8Ty,
711 IntTy,
712 NULL);
713 std::vector<llvm::Constant*> Ivars;
714 std::vector<llvm::Constant*> Elements;
715 for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
716 Elements.clear();
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000717 Elements.push_back(IvarNames[i]);
718 Elements.push_back(IvarTypes[i]);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000719 Elements.push_back(IvarOffsets[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000720 Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000721 }
722
723 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000724 llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000725 IvarNames.size());
726
Mike Stump1eb44332009-09-09 15:08:12 +0000727
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000728 Elements.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000729 Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
Owen Anderson7db6d832009-07-28 18:33:04 +0000730 Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000731 // Structure containing array and array count
Owen Anderson47a434f2009-08-05 23:18:46 +0000732 llvm::StructType *ObjCIvarListTy = llvm::StructType::get(VMContext, IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000733 ObjCIvarArrayTy,
734 NULL);
735
736 // Create an instance of the structure
737 return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
738}
739
740/// Generate a class structure
741llvm::Constant *CGObjCGNU::GenerateClassStructure(
742 llvm::Constant *MetaClass,
743 llvm::Constant *SuperClass,
744 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000745 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000746 llvm::Constant *Version,
747 llvm::Constant *InstanceSize,
748 llvm::Constant *IVars,
749 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000750 llvm::Constant *Protocols,
751 llvm::Constant *IvarOffsets,
752 llvm::Constant *Properties) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000753 // Set up the class structure
754 // Note: Several of these are char*s when they should be ids. This is
755 // because the runtime performs this translation on load.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000756 //
757 // Fields marked New ABI are part of the GNUstep runtime. We emit them
758 // anyway; the classes will still work with the GNU runtime, they will just
759 // be ignored.
Owen Anderson47a434f2009-08-05 23:18:46 +0000760 llvm::StructType *ClassTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000761 PtrToInt8Ty, // class_pointer
762 PtrToInt8Ty, // super_class
763 PtrToInt8Ty, // name
764 LongTy, // version
765 LongTy, // info
766 LongTy, // instance_size
767 IVars->getType(), // ivars
768 Methods->getType(), // methods
Mike Stump1eb44332009-09-09 15:08:12 +0000769 // These are all filled in by the runtime, so we pretend
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000770 PtrTy, // dtable
771 PtrTy, // subclass_list
772 PtrTy, // sibling_class
773 PtrTy, // protocols
774 PtrTy, // gc_object_type
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000775 // New ABI:
776 LongTy, // abi_version
777 IvarOffsets->getType(), // ivar_offsets
778 Properties->getType(), // properties
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000779 NULL);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000780 llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000781 // Fill in the structure
782 std::vector<llvm::Constant*> Elements;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000783 Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000784 Elements.push_back(SuperClass);
Chris Lattnerd002cc62008-06-26 04:47:04 +0000785 Elements.push_back(MakeConstantString(Name, ".class_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000786 Elements.push_back(Zero);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000787 Elements.push_back(llvm::ConstantInt::get(LongTy, info));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000788 Elements.push_back(InstanceSize);
789 Elements.push_back(IVars);
790 Elements.push_back(Methods);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000791 Elements.push_back(NULLPtr);
792 Elements.push_back(NULLPtr);
793 Elements.push_back(NULLPtr);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000794 Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000795 Elements.push_back(NULLPtr);
796 Elements.push_back(Zero);
797 Elements.push_back(IvarOffsets);
798 Elements.push_back(Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000799 // Create an instance of the structure
David Chisnall41d63ed2010-01-08 00:14:31 +0000800 // This is now an externally visible symbol, so that we can speed up class
801 // messages in the next ABI.
802 return MakeGlobal(ClassTy, Elements, SymbolNameForClass(Name),
803 llvm::GlobalValue::ExternalLinkage);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000804}
805
806llvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
807 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
808 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes) {
Mike Stump1eb44332009-09-09 15:08:12 +0000809 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000810 llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000811 PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
812 PtrToInt8Ty,
813 NULL);
814 std::vector<llvm::Constant*> Methods;
815 std::vector<llvm::Constant*> Elements;
816 for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
817 Elements.clear();
Mike Stump1eb44332009-09-09 15:08:12 +0000818 Elements.push_back(MethodNames[i]);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000819 Elements.push_back(MethodTypes[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000820 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000821 }
Owen Anderson96e0fc72009-07-29 22:16:19 +0000822 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000823 MethodNames.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000824 llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000825 Methods);
Owen Anderson47a434f2009-08-05 23:18:46 +0000826 llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000827 IntTy, ObjCMethodArrayTy, NULL);
828 Methods.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000829 Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000830 Methods.push_back(Array);
831 return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
832}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000833
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000834// Create the protocol list structure used in classes, categories and so on
835llvm::Constant *CGObjCGNU::GenerateProtocolList(
836 const llvm::SmallVectorImpl<std::string> &Protocols) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000837 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000838 Protocols.size());
Owen Anderson47a434f2009-08-05 23:18:46 +0000839 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000840 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
841 LongTy,//FIXME: Should be size_t
842 ProtocolArrayTy,
843 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000844 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000845 for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
846 iter != endIter ; iter++) {
David Chisnallff80fab2009-11-20 14:50:59 +0000847 llvm::Constant *protocol = 0;
848 llvm::StringMap<llvm::Constant*>::iterator value =
849 ExistingProtocols.find(*iter);
850 if (value == ExistingProtocols.end()) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000851 protocol = GenerateEmptyProtocol(*iter);
David Chisnallff80fab2009-11-20 14:50:59 +0000852 } else {
853 protocol = value->getValue();
854 }
Owen Anderson3c4972d2009-07-29 18:54:39 +0000855 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(protocol,
Owen Andersona1cf15f2009-07-14 23:10:40 +0000856 PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000857 Elements.push_back(Ptr);
858 }
Owen Anderson7db6d832009-07-28 18:33:04 +0000859 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000860 Elements);
861 Elements.clear();
862 Elements.push_back(NULLPtr);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000863 Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000864 Elements.push_back(ProtocolArray);
865 return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
866}
867
Mike Stump1eb44332009-09-09 15:08:12 +0000868llvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000869 const ObjCProtocolDecl *PD) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000870 llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
Mike Stump1eb44332009-09-09 15:08:12 +0000871 const llvm::Type *T =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000872 CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
Owen Anderson96e0fc72009-07-29 22:16:19 +0000873 return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000874}
875
876llvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
877 const std::string &ProtocolName) {
878 llvm::SmallVector<std::string, 0> EmptyStringVector;
879 llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
880
881 llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000882 llvm::Constant *MethodList =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000883 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
884 // Protocols are objects containing lists of the methods implemented and
885 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000886 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000887 PtrToInt8Ty,
888 ProtocolList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000889 MethodList->getType(),
890 MethodList->getType(),
891 MethodList->getType(),
892 MethodList->getType(),
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000893 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000894 std::vector<llvm::Constant*> Elements;
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000895 // The isa pointer must be set to a magic number so the runtime knows it's
896 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000897 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
898 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000899 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000900 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000901 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
902 Elements.push_back(ProtocolList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000903 Elements.push_back(MethodList);
904 Elements.push_back(MethodList);
905 Elements.push_back(MethodList);
906 Elements.push_back(MethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000907 return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000908}
909
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000910void CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
911 ASTContext &Context = CGM.getContext();
Chris Lattner8ec03f52008-11-24 03:54:41 +0000912 std::string ProtocolName = PD->getNameAsString();
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000913 llvm::SmallVector<std::string, 16> Protocols;
914 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
915 E = PD->protocol_end(); PI != E; ++PI)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000916 Protocols.push_back((*PI)->getNameAsString());
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000917 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
918 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000919 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodNames;
920 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodTypes;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000921 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
922 E = PD->instmeth_end(); iter != E; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000923 std::string TypeStr;
924 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000925 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
926 InstanceMethodNames.push_back(
927 MakeConstantString((*iter)->getSelector().getAsString()));
928 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
929 } else {
930 OptionalInstanceMethodNames.push_back(
931 MakeConstantString((*iter)->getSelector().getAsString()));
932 OptionalInstanceMethodTypes.push_back(MakeConstantString(TypeStr));
933 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000934 }
935 // Collect information about class methods:
936 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
937 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000938 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodNames;
939 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +0000940 for (ObjCProtocolDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000941 iter = PD->classmeth_begin(), endIter = PD->classmeth_end();
942 iter != endIter ; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000943 std::string TypeStr;
944 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000945 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
946 ClassMethodNames.push_back(
947 MakeConstantString((*iter)->getSelector().getAsString()));
948 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
949 } else {
950 OptionalClassMethodNames.push_back(
951 MakeConstantString((*iter)->getSelector().getAsString()));
952 OptionalClassMethodTypes.push_back(MakeConstantString(TypeStr));
953 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000954 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000955
956 llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
957 llvm::Constant *InstanceMethodList =
958 GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
959 llvm::Constant *ClassMethodList =
960 GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000961 llvm::Constant *OptionalInstanceMethodList =
962 GenerateProtocolMethodList(OptionalInstanceMethodNames,
963 OptionalInstanceMethodTypes);
964 llvm::Constant *OptionalClassMethodList =
965 GenerateProtocolMethodList(OptionalClassMethodNames,
966 OptionalClassMethodTypes);
967
968 // Property metadata: name, attributes, isSynthesized, setter name, setter
969 // types, getter name, getter types.
970 // The isSynthesized value is always set to 0 in a protocol. It exists to
971 // simplify the runtime library by allowing it to use the same data
972 // structures for protocol metadata everywhere.
973 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
974 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
975 PtrToInt8Ty, NULL);
976 std::vector<llvm::Constant*> Properties;
977 std::vector<llvm::Constant*> OptionalProperties;
978
979 // Add all of the property methods need adding to the method list and to the
980 // property metadata list.
981 for (ObjCContainerDecl::prop_iterator
982 iter = PD->prop_begin(), endIter = PD->prop_end();
983 iter != endIter ; iter++) {
984 std::vector<llvm::Constant*> Fields;
985 ObjCPropertyDecl *property = (*iter);
986
987 Fields.push_back(MakeConstantString(property->getNameAsString()));
988 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
989 property->getPropertyAttributes()));
990 Fields.push_back(llvm::ConstantInt::get(Int8Ty, 0));
991 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
992 std::string TypeStr;
993 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
994 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
995 InstanceMethodTypes.push_back(TypeEncoding);
996 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
997 Fields.push_back(TypeEncoding);
998 } else {
999 Fields.push_back(NULLPtr);
1000 Fields.push_back(NULLPtr);
1001 }
1002 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1003 std::string TypeStr;
1004 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1005 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1006 InstanceMethodTypes.push_back(TypeEncoding);
1007 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1008 Fields.push_back(TypeEncoding);
1009 } else {
1010 Fields.push_back(NULLPtr);
1011 Fields.push_back(NULLPtr);
1012 }
1013 if (property->getPropertyImplementation() == ObjCPropertyDecl::Optional) {
1014 OptionalProperties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1015 } else {
1016 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1017 }
1018 }
1019 llvm::Constant *PropertyArray = llvm::ConstantArray::get(
1020 llvm::ArrayType::get(PropertyMetadataTy, Properties.size()), Properties);
1021 llvm::Constant* PropertyListInitFields[] =
1022 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1023
1024 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001025 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001026 llvm::Constant *PropertyList = new llvm::GlobalVariable(TheModule,
1027 PropertyListInit->getType(), false, llvm::GlobalValue::InternalLinkage,
1028 PropertyListInit, ".objc_property_list");
1029
1030 llvm::Constant *OptionalPropertyArray =
1031 llvm::ConstantArray::get(llvm::ArrayType::get(PropertyMetadataTy,
1032 OptionalProperties.size()) , OptionalProperties);
1033 llvm::Constant* OptionalPropertyListInitFields[] = {
1034 llvm::ConstantInt::get(IntTy, OptionalProperties.size()), NULLPtr,
1035 OptionalPropertyArray };
1036
1037 llvm::Constant *OptionalPropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001038 llvm::ConstantStruct::get(VMContext, OptionalPropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001039 llvm::Constant *OptionalPropertyList = new llvm::GlobalVariable(TheModule,
1040 OptionalPropertyListInit->getType(), false,
1041 llvm::GlobalValue::InternalLinkage, OptionalPropertyListInit,
1042 ".objc_property_list");
1043
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001044 // Protocols are objects containing lists of the methods implemented and
1045 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +00001046 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001047 PtrToInt8Ty,
1048 ProtocolList->getType(),
1049 InstanceMethodList->getType(),
1050 ClassMethodList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001051 OptionalInstanceMethodList->getType(),
1052 OptionalClassMethodList->getType(),
1053 PropertyList->getType(),
1054 OptionalPropertyList->getType(),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001055 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +00001056 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001057 // The isa pointer must be set to a magic number so the runtime knows it's
1058 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001059 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
1060 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +00001061 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001062 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001063 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
1064 Elements.push_back(ProtocolList);
1065 Elements.push_back(InstanceMethodList);
1066 Elements.push_back(ClassMethodList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001067 Elements.push_back(OptionalInstanceMethodList);
1068 Elements.push_back(OptionalClassMethodList);
1069 Elements.push_back(PropertyList);
1070 Elements.push_back(OptionalPropertyList);
Mike Stump1eb44332009-09-09 15:08:12 +00001071 ExistingProtocols[ProtocolName] =
Owen Anderson3c4972d2009-07-29 18:54:39 +00001072 llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001073 ".objc_protocol"), IdTy);
1074}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001075void CGObjCGNU::GenerateProtocolHolderCategory(void) {
1076 // Collect information about instance methods
1077 llvm::SmallVector<Selector, 1> MethodSels;
1078 llvm::SmallVector<llvm::Constant*, 1> MethodTypes;
1079
1080 std::vector<llvm::Constant*> Elements;
1081 const std::string ClassName = "__ObjC_Protocol_Holder_Ugly_Hack";
1082 const std::string CategoryName = "AnotherHack";
1083 Elements.push_back(MakeConstantString(CategoryName));
1084 Elements.push_back(MakeConstantString(ClassName));
1085 // Instance method list
1086 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1087 ClassName, CategoryName, MethodSels, MethodTypes, false), PtrTy));
1088 // Class method list
1089 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1090 ClassName, CategoryName, MethodSels, MethodTypes, true), PtrTy));
1091 // Protocol list
1092 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrTy,
1093 ExistingProtocols.size());
1094 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
1095 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
1096 LongTy,//FIXME: Should be size_t
1097 ProtocolArrayTy,
1098 NULL);
1099 std::vector<llvm::Constant*> ProtocolElements;
1100 for (llvm::StringMapIterator<llvm::Constant*> iter =
1101 ExistingProtocols.begin(), endIter = ExistingProtocols.end();
1102 iter != endIter ; iter++) {
1103 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(iter->getValue(),
1104 PtrTy);
1105 ProtocolElements.push_back(Ptr);
1106 }
1107 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
1108 ProtocolElements);
1109 ProtocolElements.clear();
1110 ProtocolElements.push_back(NULLPtr);
1111 ProtocolElements.push_back(llvm::ConstantInt::get(LongTy,
1112 ExistingProtocols.size()));
1113 ProtocolElements.push_back(ProtocolArray);
1114 Elements.push_back(llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolListTy,
1115 ProtocolElements, ".objc_protocol_list"), PtrTy));
1116 Categories.push_back(llvm::ConstantExpr::getBitCast(
1117 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
1118 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
1119}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001120
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001121void CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001122 std::string ClassName = OCD->getClassInterface()->getNameAsString();
1123 std::string CategoryName = OCD->getNameAsString();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001124 // Collect information about instance methods
1125 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1126 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001127 for (ObjCCategoryImplDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001128 iter = OCD->instmeth_begin(), endIter = OCD->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001129 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001130 InstanceMethodSels.push_back((*iter)->getSelector());
1131 std::string TypeStr;
1132 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001133 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001134 }
1135
1136 // Collect information about class methods
1137 llvm::SmallVector<Selector, 16> ClassMethodSels;
1138 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001139 for (ObjCCategoryImplDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001140 iter = OCD->classmeth_begin(), endIter = OCD->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001141 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001142 ClassMethodSels.push_back((*iter)->getSelector());
1143 std::string TypeStr;
1144 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001145 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001146 }
1147
1148 // Collect the names of referenced protocols
1149 llvm::SmallVector<std::string, 16> Protocols;
1150 const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
1151 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1152 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1153 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001154 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001155
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001156 std::vector<llvm::Constant*> Elements;
1157 Elements.push_back(MakeConstantString(CategoryName));
1158 Elements.push_back(MakeConstantString(ClassName));
Mike Stump1eb44332009-09-09 15:08:12 +00001159 // Instance method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001160 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001161 ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001162 false), PtrTy));
1163 // Class method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001164 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001165 ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001166 PtrTy));
1167 // Protocol list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001168 Elements.push_back(llvm::ConstantExpr::getBitCast(
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001169 GenerateProtocolList(Protocols), PtrTy));
Owen Anderson3c4972d2009-07-29 18:54:39 +00001170 Categories.push_back(llvm::ConstantExpr::getBitCast(
Mike Stump1eb44332009-09-09 15:08:12 +00001171 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
Owen Anderson47a434f2009-08-05 23:18:46 +00001172 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001173}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001174
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001175llvm::Constant *CGObjCGNU::GeneratePropertyList(const ObjCImplementationDecl *OID,
1176 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
1177 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes) {
1178 ASTContext &Context = CGM.getContext();
1179 //
1180 // Property metadata: name, attributes, isSynthesized, setter name, setter
1181 // types, getter name, getter types.
1182 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1183 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1184 PtrToInt8Ty, NULL);
1185 std::vector<llvm::Constant*> Properties;
1186
1187
1188 // Add all of the property methods need adding to the method list and to the
1189 // property metadata list.
1190 for (ObjCImplDecl::propimpl_iterator
1191 iter = OID->propimpl_begin(), endIter = OID->propimpl_end();
1192 iter != endIter ; iter++) {
1193 std::vector<llvm::Constant*> Fields;
1194 ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
1195
1196 Fields.push_back(MakeConstantString(property->getNameAsString()));
1197 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1198 property->getPropertyAttributes()));
1199 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1200 (*iter)->getPropertyImplementation() ==
1201 ObjCPropertyImplDecl::Synthesize));
1202 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
1203 InstanceMethodSels.push_back(getter->getSelector());
1204 std::string TypeStr;
1205 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1206 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1207 InstanceMethodTypes.push_back(TypeEncoding);
1208 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1209 Fields.push_back(TypeEncoding);
1210 } else {
1211 Fields.push_back(NULLPtr);
1212 Fields.push_back(NULLPtr);
1213 }
1214 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1215 InstanceMethodSels.push_back(setter->getSelector());
1216 std::string TypeStr;
1217 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1218 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1219 InstanceMethodTypes.push_back(TypeEncoding);
1220 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1221 Fields.push_back(TypeEncoding);
1222 } else {
1223 Fields.push_back(NULLPtr);
1224 Fields.push_back(NULLPtr);
1225 }
1226 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1227 }
1228 llvm::ArrayType *PropertyArrayTy =
1229 llvm::ArrayType::get(PropertyMetadataTy, Properties.size());
1230 llvm::Constant *PropertyArray = llvm::ConstantArray::get(PropertyArrayTy,
1231 Properties);
1232 llvm::Constant* PropertyListInitFields[] =
1233 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1234
1235 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001236 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001237 return new llvm::GlobalVariable(TheModule, PropertyListInit->getType(), false,
1238 llvm::GlobalValue::InternalLinkage, PropertyListInit,
1239 ".objc_property_list");
1240}
1241
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001242void CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
1243 ASTContext &Context = CGM.getContext();
1244
1245 // Get the superclass name.
Mike Stump1eb44332009-09-09 15:08:12 +00001246 const ObjCInterfaceDecl * SuperClassDecl =
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001247 OID->getClassInterface()->getSuperClass();
Chris Lattner8ec03f52008-11-24 03:54:41 +00001248 std::string SuperClassName;
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001249 if (SuperClassDecl) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001250 SuperClassName = SuperClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001251 EmitClassRef(SuperClassName);
1252 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001253
1254 // Get the class name
Chris Lattner09dc6662009-04-01 02:00:48 +00001255 ObjCInterfaceDecl *ClassDecl =
1256 const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
Chris Lattner8ec03f52008-11-24 03:54:41 +00001257 std::string ClassName = ClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001258 // Emit the symbol that is used to generate linker errors if this class is
1259 // referenced in other modules but not declared.
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001260 std::string classSymbolName = "__objc_class_name_" + ClassName;
Mike Stump1eb44332009-09-09 15:08:12 +00001261 if (llvm::GlobalVariable *symbol =
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001262 TheModule.getGlobalVariable(classSymbolName)) {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001263 symbol->setInitializer(llvm::ConstantInt::get(LongTy, 0));
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001264 } else {
Owen Anderson1c431b32009-07-08 19:05:04 +00001265 new llvm::GlobalVariable(TheModule, LongTy, false,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001266 llvm::GlobalValue::ExternalLinkage, llvm::ConstantInt::get(LongTy, 0),
Owen Anderson1c431b32009-07-08 19:05:04 +00001267 classSymbolName);
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001268 }
Mike Stump1eb44332009-09-09 15:08:12 +00001269
Daniel Dunbar2bebbf02009-05-03 10:46:44 +00001270 // Get the size of instances.
1271 int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001272
1273 // Collect information about instance variables.
1274 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
1275 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
1276 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
Mike Stump1eb44332009-09-09 15:08:12 +00001277
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001278 std::vector<llvm::Constant*> IvarOffsetValues;
1279
Mike Stump1eb44332009-09-09 15:08:12 +00001280 int superInstanceSize = !SuperClassDecl ? 0 :
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001281 Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
1282 // For non-fragile ivars, set the instance size to 0 - {the size of just this
1283 // class}. The runtime will then set this to the correct value on load.
1284 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
1285 instanceSize = 0 - (instanceSize - superInstanceSize);
1286 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001287 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1288 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1289 // Store the name
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001290 IvarNames.push_back(MakeConstantString((*iter)->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001291 // Get the type encoding for this ivar
1292 std::string TypeStr;
Daniel Dunbar0d504c12008-10-17 20:21:44 +00001293 Context.getObjCEncodingForType((*iter)->getType(), TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001294 IvarTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001295 // Get the offset
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001296 uint64_t BaseOffset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter);
David Chisnallaecbf242009-11-17 19:32:15 +00001297 uint64_t Offset = BaseOffset;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001298 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001299 Offset = BaseOffset - superInstanceSize;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001300 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001301 IvarOffsets.push_back(
Owen Anderson0032b272009-08-13 21:57:51 +00001302 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001303 IvarOffsetValues.push_back(new llvm::GlobalVariable(TheModule, IntTy,
1304 false, llvm::GlobalValue::ExternalLinkage,
1305 llvm::ConstantInt::get(IntTy, BaseOffset),
1306 "__objc_ivar_offset_value_" + ClassName +"." +
1307 (*iter)->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001308 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001309 llvm::Constant *IvarOffsetArrayInit =
1310 llvm::ConstantArray::get(llvm::ArrayType::get(PtrToIntTy,
1311 IvarOffsetValues.size()), IvarOffsetValues);
1312 llvm::GlobalVariable *IvarOffsetArray = new llvm::GlobalVariable(TheModule,
1313 IvarOffsetArrayInit->getType(), false,
1314 llvm::GlobalValue::InternalLinkage, IvarOffsetArrayInit,
1315 ".ivar.offsets");
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001316
1317 // Collect information about instance methods
1318 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1319 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001320 for (ObjCImplementationDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001321 iter = OID->instmeth_begin(), endIter = OID->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001322 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001323 InstanceMethodSels.push_back((*iter)->getSelector());
1324 std::string TypeStr;
1325 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001326 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001327 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001328
1329 llvm::Constant *Properties = GeneratePropertyList(OID, InstanceMethodSels,
1330 InstanceMethodTypes);
1331
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001332
1333 // Collect information about class methods
1334 llvm::SmallVector<Selector, 16> ClassMethodSels;
1335 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001336 for (ObjCImplementationDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001337 iter = OID->classmeth_begin(), endIter = OID->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001338 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001339 ClassMethodSels.push_back((*iter)->getSelector());
1340 std::string TypeStr;
1341 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001342 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001343 }
1344 // Collect the names of referenced protocols
1345 llvm::SmallVector<std::string, 16> Protocols;
1346 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1347 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1348 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001349 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001350
1351
1352
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001353 // Get the superclass pointer.
1354 llvm::Constant *SuperClass;
Chris Lattner8ec03f52008-11-24 03:54:41 +00001355 if (!SuperClassName.empty()) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001356 SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
1357 } else {
Owen Anderson03e20502009-07-30 23:11:26 +00001358 SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001359 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001360 // Empty vector used to construct empty method lists
1361 llvm::SmallVector<llvm::Constant*, 1> empty;
1362 // Generate the method and instance variable lists
1363 llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001364 InstanceMethodSels, InstanceMethodTypes, false);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001365 llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001366 ClassMethodSels, ClassMethodTypes, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001367 llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
1368 IvarOffsets);
Mike Stump1eb44332009-09-09 15:08:12 +00001369 // Irrespective of whether we are compiling for a fragile or non-fragile ABI,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001370 // we emit a symbol containing the offset for each ivar in the class. This
1371 // allows code compiled for the non-Fragile ABI to inherit from code compiled
1372 // for the legacy ABI, without causing problems. The converse is also
1373 // possible, but causes all ivar accesses to be fragile.
1374 int i = 0;
1375 // Offset pointer for getting at the correct field in the ivar list when
1376 // setting up the alias. These are: The base address for the global, the
1377 // ivar array (second field), the ivar in this list (set for each ivar), and
1378 // the offset (third field in ivar structure)
1379 const llvm::Type *IndexTy = llvm::Type::getInt32Ty(VMContext);
1380 llvm::Constant *offsetPointerIndexes[] = {Zeros[0],
Mike Stump1eb44332009-09-09 15:08:12 +00001381 llvm::ConstantInt::get(IndexTy, 1), 0,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001382 llvm::ConstantInt::get(IndexTy, 2) };
1383
1384 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1385 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1386 const std::string Name = "__objc_ivar_offset_" + ClassName + '.'
1387 +(*iter)->getNameAsString();
1388 offsetPointerIndexes[2] = llvm::ConstantInt::get(IndexTy, i++);
1389 // Get the correct ivar field
1390 llvm::Constant *offsetValue = llvm::ConstantExpr::getGetElementPtr(
1391 IvarList, offsetPointerIndexes, 4);
1392 // Get the existing alias, if one exists.
1393 llvm::GlobalVariable *offset = TheModule.getNamedGlobal(Name);
1394 if (offset) {
1395 offset->setInitializer(offsetValue);
1396 // If this is the real definition, change its linkage type so that
1397 // different modules will use this one, rather than their private
1398 // copy.
1399 offset->setLinkage(llvm::GlobalValue::ExternalLinkage);
1400 } else {
1401 // Add a new alias if there isn't one already.
1402 offset = new llvm::GlobalVariable(TheModule, offsetValue->getType(),
1403 false, llvm::GlobalValue::ExternalLinkage, offsetValue, Name);
1404 }
1405 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001406 //Generate metaclass for class methods
1407 llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
David Chisnall18044632009-11-16 19:05:54 +00001408 NULLPtr, 0x12L, ClassName.c_str(), 0, Zeros[0], GenerateIvarList(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001409 empty, empty, empty), ClassMethodList, NULLPtr, NULLPtr, NULLPtr);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001410
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001411 // Generate the class structure
Chris Lattner8ec03f52008-11-24 03:54:41 +00001412 llvm::Constant *ClassStruct =
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001413 GenerateClassStructure(MetaClassStruct, SuperClass, 0x11L,
Chris Lattner8ec03f52008-11-24 03:54:41 +00001414 ClassName.c_str(), 0,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001415 llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001416 MethodList, GenerateProtocolList(Protocols), IvarOffsetArray,
1417 Properties);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001418
1419 // Resolve the class aliases, if they exist.
1420 if (ClassPtrAlias) {
1421 ClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001422 llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001423 ClassPtrAlias = 0;
1424 }
1425 if (MetaClassPtrAlias) {
1426 MetaClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001427 llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001428 MetaClassPtrAlias = 0;
1429 }
1430
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001431 // Add class structure to list to be added to the symtab later
Owen Anderson3c4972d2009-07-29 18:54:39 +00001432 ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001433 Classes.push_back(ClassStruct);
1434}
1435
Fariborz Jahanianc38e9af2009-06-23 21:47:46 +00001436
Mike Stump1eb44332009-09-09 15:08:12 +00001437llvm::Function *CGObjCGNU::ModuleInitFunction() {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001438 // Only emit an ObjC load function if no Objective-C stuff has been called
1439 if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
1440 ExistingProtocols.empty() && TypedSelectors.empty() &&
Anton Korobeynikov5f58b912008-06-01 15:14:46 +00001441 UntypedSelectors.empty())
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001442 return NULL;
Eli Friedman1b8956e2008-06-01 16:00:02 +00001443
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001444 // Add all referenced protocols to a category.
1445 GenerateProtocolHolderCategory();
1446
Chris Lattnere160c9b2009-01-27 05:06:01 +00001447 const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1448 SelectorTy->getElementType());
1449 const llvm::Type *SelStructPtrTy = SelectorTy;
1450 bool isSelOpaque = false;
1451 if (SelStructTy == 0) {
Owen Anderson47a434f2009-08-05 23:18:46 +00001452 SelStructTy = llvm::StructType::get(VMContext, PtrToInt8Ty,
1453 PtrToInt8Ty, NULL);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001454 SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
Chris Lattnere160c9b2009-01-27 05:06:01 +00001455 isSelOpaque = true;
1456 }
1457
Eli Friedman1b8956e2008-06-01 16:00:02 +00001458 // Name the ObjC types to make the IR a bit easier to read
Chris Lattnere160c9b2009-01-27 05:06:01 +00001459 TheModule.addTypeName(".objc_selector", SelStructPtrTy);
Eli Friedman1b8956e2008-06-01 16:00:02 +00001460 TheModule.addTypeName(".objc_id", IdTy);
1461 TheModule.addTypeName(".objc_imp", IMPTy);
1462
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001463 std::vector<llvm::Constant*> Elements;
Chris Lattner71238f62009-04-25 23:19:45 +00001464 llvm::Constant *Statics = NULLPtr;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001465 // Generate statics list:
Chris Lattner71238f62009-04-25 23:19:45 +00001466 if (ConstantStrings.size()) {
Owen Anderson96e0fc72009-07-29 22:16:19 +00001467 llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Chris Lattner71238f62009-04-25 23:19:45 +00001468 ConstantStrings.size() + 1);
1469 ConstantStrings.push_back(NULLPtr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001470
Daniel Dunbar1b096952009-11-29 02:38:47 +00001471 llvm::StringRef StringClass = CGM.getLangOptions().ObjCConstantStringClass;
1472 if (StringClass.empty()) StringClass = "NXConstantString";
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001473 Elements.push_back(MakeConstantString(StringClass,
1474 ".objc_static_class_name"));
Owen Anderson7db6d832009-07-28 18:33:04 +00001475 Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
Chris Lattner71238f62009-04-25 23:19:45 +00001476 ConstantStrings));
Mike Stump1eb44332009-09-09 15:08:12 +00001477 llvm::StructType *StaticsListTy =
Owen Anderson47a434f2009-08-05 23:18:46 +00001478 llvm::StructType::get(VMContext, PtrToInt8Ty, StaticsArrayTy, NULL);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001479 llvm::Type *StaticsListPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001480 llvm::PointerType::getUnqual(StaticsListTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001481 Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
Mike Stump1eb44332009-09-09 15:08:12 +00001482 llvm::ArrayType *StaticsListArrayTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001483 llvm::ArrayType::get(StaticsListPtrTy, 2);
Chris Lattner71238f62009-04-25 23:19:45 +00001484 Elements.clear();
1485 Elements.push_back(Statics);
Owen Andersonc9c88b42009-07-31 20:28:54 +00001486 Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
Chris Lattner71238f62009-04-25 23:19:45 +00001487 Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001488 Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001489 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001490 // Array of classes, categories, and constant objects
Owen Anderson96e0fc72009-07-29 22:16:19 +00001491 llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001492 Classes.size() + Categories.size() + 2);
Mike Stump1eb44332009-09-09 15:08:12 +00001493 llvm::StructType *SymTabTy = llvm::StructType::get(VMContext,
Owen Anderson47a434f2009-08-05 23:18:46 +00001494 LongTy, SelStructPtrTy,
Owen Anderson0032b272009-08-13 21:57:51 +00001495 llvm::Type::getInt16Ty(VMContext),
1496 llvm::Type::getInt16Ty(VMContext),
Chris Lattner630404b2008-06-26 04:10:42 +00001497 ClassListTy, NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001498
1499 Elements.clear();
1500 // Pointer to an array of selectors used in this module.
1501 std::vector<llvm::Constant*> Selectors;
1502 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1503 iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
1504 iter != iterEnd ; ++iter) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001505 Elements.push_back(ExportUniqueString(iter->first.first, ".objc_sel_name"));
David Chisnalla7c5b082009-09-14 19:04:10 +00001506 Elements.push_back(MakeConstantString(iter->first.second,
Chris Lattner630404b2008-06-26 04:10:42 +00001507 ".objc_sel_types"));
Owen Anderson08e25242009-07-27 22:29:56 +00001508 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001509 Elements.clear();
1510 }
1511 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1512 iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
Chris Lattner630404b2008-06-26 04:10:42 +00001513 iter != iterEnd; ++iter) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001514 Elements.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001515 ExportUniqueString(iter->getKeyData(), ".objc_sel_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001516 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001517 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001518 Elements.clear();
1519 }
1520 Elements.push_back(NULLPtr);
1521 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001522 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001523 Elements.clear();
1524 // Number of static selectors
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001525 Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001526 llvm::Constant *SelectorList = MakeGlobal(
Owen Anderson96e0fc72009-07-29 22:16:19 +00001527 llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001528 ".objc_selector_list");
Mike Stump1eb44332009-09-09 15:08:12 +00001529 Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
Chris Lattnere160c9b2009-01-27 05:06:01 +00001530 SelStructPtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001531
1532 // Now that all of the static selectors exist, create pointers to them.
1533 int index = 0;
1534 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1535 iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
1536 iter != iterEnd; ++iter) {
1537 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001538 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001539 llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
1540 true, llvm::GlobalValue::InternalLinkage,
1541 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1542 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001543 // If selectors are defined as an opaque type, cast the pointer to this
1544 // type.
1545 if (isSelOpaque) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001546 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1547 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001548 }
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001549 (*iter).second->setAliasee(SelPtr);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001550 }
1551 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1552 iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1553 iter != iterEnd; iter++) {
1554 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001555 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001556 llvm::Constant *SelPtr = new llvm::GlobalVariable
1557 (TheModule, SelStructPtrTy,
1558 true, llvm::GlobalValue::InternalLinkage,
1559 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1560 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001561 // If selectors are defined as an opaque type, cast the pointer to this
1562 // type.
1563 if (isSelOpaque) {
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001564 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1565 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001566 }
Daniel Dunbar6d5a1c22010-02-03 20:11:42 +00001567 (*iter).second->setAliasee(SelPtr);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001568 }
1569 // Number of classes defined.
Mike Stump1eb44332009-09-09 15:08:12 +00001570 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001571 Classes.size()));
1572 // Number of categories defined
Mike Stump1eb44332009-09-09 15:08:12 +00001573 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001574 Categories.size()));
1575 // Create an array of classes, then categories, then static object instances
1576 Classes.insert(Classes.end(), Categories.begin(), Categories.end());
1577 // NULL-terminated list of static object instances (mainly constant strings)
1578 Classes.push_back(Statics);
1579 Classes.push_back(NULLPtr);
Owen Anderson7db6d832009-07-28 18:33:04 +00001580 llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001581 Elements.push_back(ClassList);
Mike Stump1eb44332009-09-09 15:08:12 +00001582 // Construct the symbol table
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001583 llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
1584
1585 // The symbol table is contained in a module which has some version-checking
1586 // constants
Owen Anderson47a434f2009-08-05 23:18:46 +00001587 llvm::StructType * ModuleTy = llvm::StructType::get(VMContext, LongTy, LongTy,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001588 PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001589 Elements.clear();
1590 // Runtime version used for compatibility checking.
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001591 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Mike Stump1eb44332009-09-09 15:08:12 +00001592 Elements.push_back(llvm::ConstantInt::get(LongTy,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001593 NonFragileRuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001594 } else {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001595 Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001596 }
Fariborz Jahanian91a0b512009-04-01 19:49:42 +00001597 // sizeof(ModuleTy)
Benjamin Kramer74a8bbf2010-02-09 19:31:24 +00001598 llvm::TargetData td(&TheModule);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001599 Elements.push_back(llvm::ConstantInt::get(LongTy,
Owen Andersona1cf15f2009-07-14 23:10:40 +00001600 td.getTypeSizeInBits(ModuleTy)/8));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001601 //FIXME: Should be the path to the file where this module was declared
1602 Elements.push_back(NULLPtr);
1603 Elements.push_back(SymTab);
1604 llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
1605
1606 // Create the load function calling the runtime entry point with the module
1607 // structure
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001608 llvm::Function * LoadFunction = llvm::Function::Create(
Owen Anderson0032b272009-08-13 21:57:51 +00001609 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), false),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001610 llvm::GlobalValue::InternalLinkage, ".objc_load_function",
1611 &TheModule);
Owen Anderson0032b272009-08-13 21:57:51 +00001612 llvm::BasicBlock *EntryBB =
1613 llvm::BasicBlock::Create(VMContext, "entry", LoadFunction);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001614 CGBuilderTy Builder(VMContext);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001615 Builder.SetInsertPoint(EntryBB);
Fariborz Jahanian26c82942009-03-30 18:02:14 +00001616
1617 std::vector<const llvm::Type*> Params(1,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001618 llvm::PointerType::getUnqual(ModuleTy));
1619 llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Owen Anderson0032b272009-08-13 21:57:51 +00001620 llvm::Type::getVoidTy(VMContext), Params, true), "__objc_exec_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001621 Builder.CreateCall(Register, Module);
1622 Builder.CreateRetVoid();
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001623
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001624 return LoadFunction;
1625}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001626
Fariborz Jahanian679a5022009-01-10 21:06:09 +00001627llvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
Mike Stump1eb44332009-09-09 15:08:12 +00001628 const ObjCContainerDecl *CD) {
1629 const ObjCCategoryImplDecl *OCD =
Steve Naroff3e0a5402009-01-08 19:41:02 +00001630 dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
Chris Lattner077bf5e2008-11-24 03:33:13 +00001631 std::string CategoryName = OCD ? OCD->getNameAsString() : "";
1632 std::string ClassName = OMD->getClassInterface()->getNameAsString();
1633 std::string MethodName = OMD->getSelector().getAsString();
Douglas Gregorf8d49f62009-01-09 17:18:27 +00001634 bool isClassMethod = !OMD->isInstanceMethod();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001635
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001636 CodeGenTypes &Types = CGM.getTypes();
Mike Stump1eb44332009-09-09 15:08:12 +00001637 const llvm::FunctionType *MethodTy =
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001638 Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001639 std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
1640 MethodName, isClassMethod);
1641
Daniel Dunbard6c93d72009-09-17 04:01:22 +00001642 llvm::Function *Method
Mike Stump1eb44332009-09-09 15:08:12 +00001643 = llvm::Function::Create(MethodTy,
1644 llvm::GlobalValue::InternalLinkage,
1645 FunctionName,
1646 &TheModule);
Chris Lattner391d77a2008-03-30 23:03:07 +00001647 return Method;
1648}
1649
Daniel Dunbar49f66022008-09-24 03:38:44 +00001650llvm::Function *CGObjCGNU::GetPropertyGetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001651 std::vector<const llvm::Type*> Params;
1652 const llvm::Type *BoolTy =
1653 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1654 Params.push_back(IdTy);
1655 Params.push_back(SelectorTy);
1656 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1657 Params.push_back(LongTy);
1658 Params.push_back(BoolTy);
1659 // void objc_getProperty (id, SEL, ptrdiff_t, bool)
1660 const llvm::FunctionType *FTy =
1661 llvm::FunctionType::get(IdTy, Params, false);
1662 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1663 "objc_getProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001664}
1665
1666llvm::Function *CGObjCGNU::GetPropertySetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001667 std::vector<const llvm::Type*> Params;
1668 const llvm::Type *BoolTy =
1669 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1670 Params.push_back(IdTy);
1671 Params.push_back(SelectorTy);
1672 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1673 Params.push_back(LongTy);
1674 Params.push_back(IdTy);
1675 Params.push_back(BoolTy);
1676 Params.push_back(BoolTy);
1677 // void objc_setProperty (id, SEL, ptrdiff_t, id, bool, bool)
1678 const llvm::FunctionType *FTy =
1679 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Params, false);
1680 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1681 "objc_setProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001682}
1683
Daniel Dunbar309a4362009-07-24 07:40:24 +00001684llvm::Constant *CGObjCGNU::EnumerationMutationFunction() {
1685 CodeGen::CodeGenTypes &Types = CGM.getTypes();
1686 ASTContext &Ctx = CGM.getContext();
1687 // void objc_enumerationMutation (id)
1688 llvm::SmallVector<QualType,16> Params;
David Chisnall0f436562009-08-17 16:35:33 +00001689 Params.push_back(ASTIdTy);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001690 const llvm::FunctionType *FTy =
John McCall04a67a62010-02-05 21:31:56 +00001691 Types.GetFunctionType(Types.getFunctionInfo(Ctx.VoidTy, Params,
1692 CC_Default, false), false);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001693 return CGM.CreateRuntimeFunction(FTy, "objc_enumerationMutation");
Anders Carlsson2abd89c2008-08-31 04:05:03 +00001694}
1695
Fariborz Jahanianbd71be42008-11-21 00:49:24 +00001696void CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1697 const Stmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001698 // Pointer to the personality function
1699 llvm::Constant *Personality =
Owen Anderson0032b272009-08-13 21:57:51 +00001700 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getInt32Ty(VMContext),
Chris Lattnerb59761b2009-07-01 04:13:52 +00001701 true),
Chris Lattner5dc08672009-05-08 00:11:50 +00001702 "__gnu_objc_personality_v0");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001703 Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
Chris Lattner5dc08672009-05-08 00:11:50 +00001704 std::vector<const llvm::Type*> Params;
1705 Params.push_back(PtrTy);
1706 llvm::Value *RethrowFn =
Owen Anderson0032b272009-08-13 21:57:51 +00001707 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
David Chisnall90665bd2010-01-06 18:02:59 +00001708 Params, false), "_Unwind_Resume");
Chris Lattner5dc08672009-05-08 00:11:50 +00001709
1710 bool isTry = isa<ObjCAtTryStmt>(S);
1711 llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
1712 llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
1713 llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
Chris Lattnerd6a99072009-05-11 18:16:28 +00001714 llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
Chris Lattner5dc08672009-05-08 00:11:50 +00001715 llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
1716 llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
1717 llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
1718
David Chisnall0faa5162009-12-24 02:26:34 +00001719 // @synchronized()
Chris Lattner5dc08672009-05-08 00:11:50 +00001720 if (!isTry) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001721 std::vector<const llvm::Type*> Args(1, IdTy);
1722 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001723 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001724 llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
Mike Stump1eb44332009-09-09 15:08:12 +00001725 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001726 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1727 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1728 CGF.Builder.CreateCall(SyncEnter, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001729 }
1730
Chris Lattnerd6a99072009-05-11 18:16:28 +00001731
Chris Lattner5dc08672009-05-08 00:11:50 +00001732 // Push an EH context entry, used for handling rethrows and jumps
1733 // through finally.
1734 CGF.PushCleanupBlock(FinallyBlock);
1735
1736 // Emit the statements in the @try {} block
1737 CGF.setInvokeDest(TryHandler);
1738
1739 CGF.EmitBlock(TryBlock);
Mike Stump1eb44332009-09-09 15:08:12 +00001740 CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
Chris Lattner5dc08672009-05-08 00:11:50 +00001741 : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
1742
1743 // Jump to @finally if there is no exception
1744 CGF.EmitBranchThroughCleanup(FinallyEnd);
1745
1746 // Emit the handlers
1747 CGF.EmitBlock(TryHandler);
1748
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001749 // Get the correct versions of the exception handling intrinsics
Mike Stump1eb44332009-09-09 15:08:12 +00001750 llvm::Value *llvm_eh_exception =
Chris Lattner5dc08672009-05-08 00:11:50 +00001751 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
Duncan Sands790b20e2009-10-14 16:13:30 +00001752 llvm::Value *llvm_eh_selector =
1753 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector);
1754 llvm::Value *llvm_eh_typeid_for =
1755 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for);
Chris Lattner5dc08672009-05-08 00:11:50 +00001756
1757 // Exception object
1758 llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1759 llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
1760
1761 llvm::SmallVector<llvm::Value*, 8> ESelArgs;
1762 llvm::SmallVector<std::pair<const ParmVarDecl*, const Stmt*>, 8> Handlers;
1763
1764 ESelArgs.push_back(Exc);
1765 ESelArgs.push_back(Personality);
1766
1767 bool HasCatchAll = false;
1768 // Only @try blocks are allowed @catch blocks, but both can have @finally
1769 if (isTry) {
1770 if (const ObjCAtCatchStmt* CatchStmt =
1771 cast<ObjCAtTryStmt>(S).getCatchStmts()) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001772 CGF.setInvokeDest(CatchInCatch);
Chris Lattner5dc08672009-05-08 00:11:50 +00001773
1774 for (; CatchStmt; CatchStmt = CatchStmt->getNextCatchStmt()) {
1775 const ParmVarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
Mike Stumpbb1c8602009-07-31 21:31:32 +00001776 Handlers.push_back(std::make_pair(CatchDecl,
1777 CatchStmt->getCatchBody()));
Chris Lattner5dc08672009-05-08 00:11:50 +00001778
1779 // @catch() and @catch(id) both catch any ObjC exception
Steve Naroff14108da2009-07-10 23:34:53 +00001780 if (!CatchDecl || CatchDecl->getType()->isObjCIdType()
Chris Lattner5dc08672009-05-08 00:11:50 +00001781 || CatchDecl->getType()->isObjCQualifiedIdType()) {
1782 // Use i8* null here to signal this is a catch all, not a cleanup.
1783 ESelArgs.push_back(NULLPtr);
1784 HasCatchAll = true;
1785 // No further catches after this one will ever by reached
1786 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001787 }
Chris Lattner5dc08672009-05-08 00:11:50 +00001788
1789 // All other types should be Objective-C interface pointer types.
Mike Stump1eb44332009-09-09 15:08:12 +00001790 const ObjCObjectPointerType *OPT =
John McCall183700f2009-09-21 23:43:11 +00001791 CatchDecl->getType()->getAs<ObjCObjectPointerType>();
Steve Naroff14108da2009-07-10 23:34:53 +00001792 assert(OPT && "Invalid @catch type.");
Mike Stump1eb44332009-09-09 15:08:12 +00001793 const ObjCInterfaceType *IT =
John McCall183700f2009-09-21 23:43:11 +00001794 OPT->getPointeeType()->getAs<ObjCInterfaceType>();
Chris Lattner5dc08672009-05-08 00:11:50 +00001795 assert(IT && "Invalid @catch type.");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001796 llvm::Value *EHType =
1797 MakeConstantString(IT->getDecl()->getNameAsString());
Chris Lattner5dc08672009-05-08 00:11:50 +00001798 ESelArgs.push_back(EHType);
1799 }
1800 }
1801 }
1802
1803 // We use a cleanup unless there was already a catch all.
1804 if (!HasCatchAll) {
Owen Anderson0032b272009-08-13 21:57:51 +00001805 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattner5dc08672009-05-08 00:11:50 +00001806 Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
1807 }
1808
1809 // Find which handler was matched.
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001810 llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
Chris Lattner5dc08672009-05-08 00:11:50 +00001811 ESelArgs.begin(), ESelArgs.end(), "selector");
1812
1813 for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
1814 const ParmVarDecl *CatchParam = Handlers[i].first;
1815 const Stmt *CatchBody = Handlers[i].second;
1816
1817 llvm::BasicBlock *Next = 0;
1818
1819 // The last handler always matches.
1820 if (i + 1 != e) {
1821 assert(CatchParam && "Only last handler can be a catch all.");
1822
1823 // Test whether this block matches the type for the selector and branch
1824 // to Match if it does, or to the next BB if it doesn't.
1825 llvm::BasicBlock *Match = CGF.createBasicBlock("match");
1826 Next = CGF.createBasicBlock("catch.next");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001827 llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
Chris Lattner5dc08672009-05-08 00:11:50 +00001828 CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
1829 CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
1830 Next);
1831
1832 CGF.EmitBlock(Match);
1833 }
Mike Stump1eb44332009-09-09 15:08:12 +00001834
Chris Lattner5dc08672009-05-08 00:11:50 +00001835 if (CatchBody) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001836 llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
1837 CGF.ConvertType(CatchParam->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001838
Chris Lattner5dc08672009-05-08 00:11:50 +00001839 // Bind the catch parameter if it exists.
1840 if (CatchParam) {
1841 // CatchParam is a ParmVarDecl because of the grammar
1842 // construction used to handle this, but for codegen purposes
1843 // we treat this as a local decl.
1844 CGF.EmitLocalBlockVarDecl(*CatchParam);
1845 CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
1846 }
1847
1848 CGF.ObjCEHValueStack.push_back(ExcObject);
1849 CGF.EmitStmt(CatchBody);
1850 CGF.ObjCEHValueStack.pop_back();
1851
1852 CGF.EmitBranchThroughCleanup(FinallyEnd);
1853
1854 if (Next)
1855 CGF.EmitBlock(Next);
1856 } else {
1857 assert(!Next && "catchup should be last handler.");
1858
1859 CGF.Builder.CreateStore(Exc, RethrowPtr);
1860 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1861 }
1862 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001863 // The @finally block is a secondary landing pad for any exceptions thrown in
1864 // @catch() blocks
1865 CGF.EmitBlock(CatchInCatch);
1866 Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1867 ESelArgs.clear();
1868 ESelArgs.push_back(Exc);
1869 ESelArgs.push_back(Personality);
David Chisnall0faa5162009-12-24 02:26:34 +00001870 // If there is a @catch or @finally clause in outside of this one then we
1871 // need to make sure that we catch and rethrow it.
1872 if (PrevLandingPad) {
1873 ESelArgs.push_back(NULLPtr);
1874 } else {
1875 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
1876 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001877 CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
1878 "selector");
1879 CGF.Builder.CreateCall(llvm_eh_typeid_for,
1880 CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
1881 CGF.Builder.CreateStore(Exc, RethrowPtr);
1882 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1883
Chris Lattner5dc08672009-05-08 00:11:50 +00001884 CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
1885
1886 CGF.setInvokeDest(PrevLandingPad);
1887
1888 CGF.EmitBlock(FinallyBlock);
1889
Chris Lattnerd6a99072009-05-11 18:16:28 +00001890
Chris Lattner5dc08672009-05-08 00:11:50 +00001891 if (isTry) {
Mike Stump1eb44332009-09-09 15:08:12 +00001892 if (const ObjCAtFinallyStmt* FinallyStmt =
Chris Lattner5dc08672009-05-08 00:11:50 +00001893 cast<ObjCAtTryStmt>(S).getFinallyStmt())
1894 CGF.EmitStmt(FinallyStmt->getFinallyBody());
1895 } else {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001896 // Emit 'objc_sync_exit(expr)' as finally's sole statement for
Chris Lattner5dc08672009-05-08 00:11:50 +00001897 // @synchronized.
Chris Lattnerd6a99072009-05-11 18:16:28 +00001898 std::vector<const llvm::Type*> Args(1, IdTy);
1899 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001900 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001901 llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
Mike Stump1eb44332009-09-09 15:08:12 +00001902 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001903 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1904 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1905 CGF.Builder.CreateCall(SyncExit, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001906 }
1907
1908 if (Info.SwitchBlock)
1909 CGF.EmitBlock(Info.SwitchBlock);
1910 if (Info.EndBlock)
1911 CGF.EmitBlock(Info.EndBlock);
1912
1913 // Branch around the rethrow code.
1914 CGF.EmitBranch(FinallyEnd);
1915
1916 CGF.EmitBlock(FinallyRethrow);
David Chisnall0faa5162009-12-24 02:26:34 +00001917
1918 llvm::Value *ExceptionObject = CGF.Builder.CreateLoad(RethrowPtr);
1919 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
1920 if (!UnwindBB) {
1921 CGF.Builder.CreateCall(RethrowFn, ExceptionObject);
1922 // Exception always thrown, next instruction is never reached.
1923 CGF.Builder.CreateUnreachable();
1924 } else {
1925 // If there is a @catch block outside this scope, we invoke instead of
1926 // calling because we may return to this function. This is very slow, but
1927 // some people still do it. It would be nice to add an optimised path for
1928 // this.
1929 CGF.Builder.CreateInvoke(RethrowFn, UnwindBB, UnwindBB, &ExceptionObject,
1930 &ExceptionObject+1);
1931 }
Mike Stump1eb44332009-09-09 15:08:12 +00001932
Chris Lattner5dc08672009-05-08 00:11:50 +00001933 CGF.EmitBlock(FinallyEnd);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001934}
1935
1936void CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar49f66022008-09-24 03:38:44 +00001937 const ObjCAtThrowStmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001938 llvm::Value *ExceptionAsObject;
1939
1940 std::vector<const llvm::Type*> Args(1, IdTy);
1941 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001942 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Mike Stump1eb44332009-09-09 15:08:12 +00001943 llvm::Value *ThrowFn =
Chris Lattner5dc08672009-05-08 00:11:50 +00001944 CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
Mike Stump1eb44332009-09-09 15:08:12 +00001945
Chris Lattner5dc08672009-05-08 00:11:50 +00001946 if (const Expr *ThrowExpr = S.getThrowExpr()) {
1947 llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001948 ExceptionAsObject = Exception;
Chris Lattner5dc08672009-05-08 00:11:50 +00001949 } else {
Mike Stump1eb44332009-09-09 15:08:12 +00001950 assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
Chris Lattner5dc08672009-05-08 00:11:50 +00001951 "Unexpected rethrow outside @catch block.");
1952 ExceptionAsObject = CGF.ObjCEHValueStack.back();
1953 }
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001954 ExceptionAsObject =
1955 CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
Mike Stump1eb44332009-09-09 15:08:12 +00001956
Chris Lattner5dc08672009-05-08 00:11:50 +00001957 // Note: This may have to be an invoke, if we want to support constructs like:
1958 // @try {
1959 // @throw(obj);
1960 // }
1961 // @catch(id) ...
1962 //
1963 // This is effectively turning @throw into an incredibly-expensive goto, but
1964 // it may happen as a result of inlining followed by missed optimizations, or
1965 // as a result of stupidity.
1966 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
1967 if (!UnwindBB) {
1968 CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
1969 CGF.Builder.CreateUnreachable();
1970 } else {
1971 CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
1972 &ExceptionAsObject+1);
1973 }
1974 // Clear the insertion point to indicate we are in unreachable code.
1975 CGF.Builder.ClearInsertionPoint();
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001976}
1977
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001978llvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001979 llvm::Value *AddrWeakObj) {
David Chisnallef6e0f32010-02-03 15:59:02 +00001980 CGBuilderTy B = CGF.Builder;
1981 AddrWeakObj = EnforceType(B, AddrWeakObj, IdTy);
1982 return B.CreateCall(WeakReadFn, AddrWeakObj);
Fariborz Jahanian6dc23172008-11-18 21:45:40 +00001983}
1984
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001985void CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001986 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00001987 CGBuilderTy B = CGF.Builder;
1988 src = EnforceType(B, src, IdTy);
1989 dst = EnforceType(B, dst, PtrToIdTy);
1990 B.CreateCall2(WeakAssignFn, src, dst);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001991}
1992
Fariborz Jahanian58626502008-11-19 00:59:10 +00001993void CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001994 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00001995 CGBuilderTy B = CGF.Builder;
1996 src = EnforceType(B, src, IdTy);
1997 dst = EnforceType(B, dst, PtrToIdTy);
1998 B.CreateCall2(GlobalAssignFn, src, dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +00001999}
2000
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00002001void CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +00002002 llvm::Value *src, llvm::Value *dst,
2003 llvm::Value *ivarOffset) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002004 CGBuilderTy B = CGF.Builder;
2005 src = EnforceType(B, src, IdTy);
2006 dst = EnforceType(B, dst, PtrToIdTy);
2007 B.CreateCall3(IvarAssignFn, src, dst, ivarOffset);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00002008}
2009
Fariborz Jahanian58626502008-11-19 00:59:10 +00002010void CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002011 llvm::Value *src, llvm::Value *dst) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002012 CGBuilderTy B = CGF.Builder;
2013 src = EnforceType(B, src, IdTy);
2014 dst = EnforceType(B, dst, PtrToIdTy);
2015 B.CreateCall2(StrongCastAssignFn, src, dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +00002016}
2017
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002018void CGObjCGNU::EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00002019 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002020 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +00002021 QualType Ty) {
David Chisnallef6e0f32010-02-03 15:59:02 +00002022 CGBuilderTy B = CGF.Builder;
2023 DestPtr = EnforceType(B, DestPtr, IdTy);
2024 SrcPtr = EnforceType(B, SrcPtr, PtrToIdTy);
2025
2026 std::pair<uint64_t, unsigned> TypeInfo = CGM.getContext().getTypeInfo(Ty);
2027 unsigned long size = TypeInfo.first/8;
2028 // FIXME: size_t
2029 llvm::Value *N = llvm::ConstantInt::get(LongTy, size);
2030
2031 B.CreateCall3(MemMoveFn, DestPtr, SrcPtr, N);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00002032}
2033
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002034llvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
2035 const ObjCInterfaceDecl *ID,
2036 const ObjCIvarDecl *Ivar) {
2037 const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
2038 + '.' + Ivar->getNameAsString();
2039 // Emit the variable and initialize it with what we think the correct value
2040 // is. This allows code compiled with non-fragile ivars to work correctly
2041 // when linked against code which isn't (most of the time).
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002042 llvm::GlobalVariable *IvarOffsetPointer = TheModule.getNamedGlobal(Name);
2043 if (!IvarOffsetPointer) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002044 uint64_t Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
2045 llvm::ConstantInt *OffsetGuess =
David Chisnallf9508372010-01-11 19:02:35 +00002046 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset, "ivar");
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002047 // Don't emit the guess in non-PIC code because the linker will not be able
2048 // to replace it with the real version for a library. In non-PIC code you
2049 // must compile with the fragile ABI if you want to use ivars from a
Mike Stump1eb44332009-09-09 15:08:12 +00002050 // GCC-compiled class.
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002051 if (CGM.getLangOptions().PICLevel) {
2052 llvm::GlobalVariable *IvarOffsetGV = new llvm::GlobalVariable(TheModule,
2053 llvm::Type::getInt32Ty(VMContext), false,
2054 llvm::GlobalValue::PrivateLinkage, OffsetGuess, Name+".guess");
2055 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
2056 IvarOffsetGV->getType(), false, llvm::GlobalValue::LinkOnceAnyLinkage,
2057 IvarOffsetGV, Name);
2058 } else {
2059 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
Benjamin Kramer3c0ef8c2009-10-13 10:07:13 +00002060 llvm::Type::getInt32PtrTy(VMContext), false,
2061 llvm::GlobalValue::ExternalLinkage, 0, Name);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002062 }
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002063 }
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002064 return IvarOffsetPointer;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002065}
2066
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00002067LValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
2068 QualType ObjectTy,
2069 llvm::Value *BaseValue,
2070 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00002071 unsigned CVRQualifiers) {
John McCall183700f2009-09-21 23:43:11 +00002072 const ObjCInterfaceDecl *ID = ObjectTy->getAs<ObjCInterfaceType>()->getDecl();
Daniel Dunbar97776872009-04-22 07:32:20 +00002073 return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
2074 EmitIvarOffset(CGF, ID, Ivar));
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00002075}
Mike Stumpbb1c8602009-07-31 21:31:32 +00002076
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002077static const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
2078 const ObjCInterfaceDecl *OID,
2079 const ObjCIvarDecl *OIVD) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002080 llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
Fariborz Jahanian8e6ac1d2009-06-04 01:19:09 +00002081 Context.ShallowCollectObjCIvars(OID, Ivars);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002082 for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
2083 if (OIVD == Ivars[k])
2084 return OID;
2085 }
Mike Stump1eb44332009-09-09 15:08:12 +00002086
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002087 // Otherwise check in the super class.
2088 if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
2089 return FindIvarInterface(Context, Super, OIVD);
Mike Stump1eb44332009-09-09 15:08:12 +00002090
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002091 return 0;
2092}
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00002093
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002094llvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +00002095 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002096 const ObjCIvarDecl *Ivar) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002097 if (CGM.getLangOptions().ObjCNonFragileABI) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002098 Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00002099 return CGF.Builder.CreateLoad(CGF.Builder.CreateLoad(
2100 ObjCIvarOffsetVariable(Interface, Ivar), false, "ivar"));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00002101 }
Daniel Dunbar97776872009-04-22 07:32:20 +00002102 uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00002103 return llvm::ConstantInt::get(LongTy, Offset, "ivar");
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00002104}
2105
Mike Stumpbb1c8602009-07-31 21:31:32 +00002106CodeGen::CGObjCRuntime *
2107CodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM) {
Chris Lattnerdce14062008-06-26 04:19:03 +00002108 return new CGObjCGNU(CGM);
Chris Lattner0f984262008-03-01 08:50:34 +00002109}