blob: be772c7fa311a162e1dec446fcc1ded35f832159 [file] [log] [blame]
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001//===------- CGObjCGNU.cpp - Emit LLVM Code from ASTs for a Module --------===//
Chris Lattner0f984262008-03-01 08:50:34 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000010// This provides Objective-C code generation targetting the GNU runtime. The
11// class in this file generates structures used by the GNU Objective-C runtime
12// library. These structures are defined in objc/objc.h and objc/objc-api.h in
13// the GNU runtime distribution.
Chris Lattner0f984262008-03-01 08:50:34 +000014//
15//===----------------------------------------------------------------------===//
16
17#include "CGObjCRuntime.h"
Chris Lattnerdce14062008-06-26 04:19:03 +000018#include "CodeGenModule.h"
Daniel Dunbar8f2926b2008-08-23 03:46:30 +000019#include "CodeGenFunction.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000020
Chris Lattnerdce14062008-06-26 04:19:03 +000021#include "clang/AST/ASTContext.h"
Daniel Dunbare91593e2008-08-11 04:54:23 +000022#include "clang/AST/Decl.h"
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +000023#include "clang/AST/DeclObjC.h"
Anders Carlsson19cc4ab2009-07-18 19:43:29 +000024#include "clang/AST/RecordLayout.h"
Chris Lattner16f00492009-04-26 01:32:48 +000025#include "clang/AST/StmtObjC.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000026
27#include "llvm/Intrinsics.h"
Chris Lattner0f984262008-03-01 08:50:34 +000028#include "llvm/Module.h"
Chris Lattner0f984262008-03-01 08:50:34 +000029#include "llvm/ADT/SmallVector.h"
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000030#include "llvm/ADT/StringMap.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000031#include "llvm/Support/Compiler.h"
Daniel Dunbar7ded7f42008-08-15 22:20:32 +000032#include "llvm/Target/TargetData.h"
Chris Lattner5dc08672009-05-08 00:11:50 +000033
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000034#include <map>
Chris Lattnere160c9b2009-01-27 05:06:01 +000035
36
Chris Lattnerdce14062008-06-26 04:19:03 +000037using namespace clang;
Daniel Dunbar46f45b92008-09-09 01:06:48 +000038using namespace CodeGen;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000039using llvm::dyn_cast;
40
41// The version of the runtime that this class targets. Must match the version
42// in the runtime.
43static const int RuntimeVersion = 8;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +000044static const int NonFragileRuntimeVersion = 9;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000045static const int ProtocolVersion = 2;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000046static const int NonFragileProtocolVersion = 3;
Chris Lattner0f984262008-03-01 08:50:34 +000047
Chris Lattner0f984262008-03-01 08:50:34 +000048namespace {
Chris Lattnerdce14062008-06-26 04:19:03 +000049class CGObjCGNU : public CodeGen::CGObjCRuntime {
Chris Lattner0f984262008-03-01 08:50:34 +000050private:
Chris Lattnerdce14062008-06-26 04:19:03 +000051 CodeGen::CodeGenModule &CGM;
Chris Lattner0f984262008-03-01 08:50:34 +000052 llvm::Module &TheModule;
Chris Lattnere160c9b2009-01-27 05:06:01 +000053 const llvm::PointerType *SelectorTy;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000054 const llvm::IntegerType *Int8Ty;
Chris Lattnere160c9b2009-01-27 05:06:01 +000055 const llvm::PointerType *PtrToInt8Ty;
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +000056 const llvm::FunctionType *IMPTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000057 const llvm::PointerType *IdTy;
David Chisnall0f436562009-08-17 16:35:33 +000058 QualType ASTIdTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000059 const llvm::IntegerType *IntTy;
60 const llvm::PointerType *PtrTy;
61 const llvm::IntegerType *LongTy;
62 const llvm::PointerType *PtrToIntTy;
Daniel Dunbar5efccb12009-05-04 15:31:17 +000063 llvm::GlobalAlias *ClassPtrAlias;
64 llvm::GlobalAlias *MetaClassPtrAlias;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000065 std::vector<llvm::Constant*> Classes;
66 std::vector<llvm::Constant*> Categories;
67 std::vector<llvm::Constant*> ConstantStrings;
68 llvm::Function *LoadFunction;
69 llvm::StringMap<llvm::Constant*> ExistingProtocols;
70 typedef std::pair<std::string, std::string> TypedSelector;
71 std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
72 llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
73 // Some zeros used for GEPs in lots of places.
74 llvm::Constant *Zeros[2];
75 llvm::Constant *NULLPtr;
Owen Andersona1cf15f2009-07-14 23:10:40 +000076 llvm::LLVMContext &VMContext;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000077private:
78 llvm::Constant *GenerateIvarList(
79 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
80 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
81 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets);
82 llvm::Constant *GenerateMethodList(const std::string &ClassName,
83 const std::string &CategoryName,
Mike Stump1eb44332009-09-09 15:08:12 +000084 const llvm::SmallVectorImpl<Selector> &MethodSels,
85 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000086 bool isClassMethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +000087 llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000088 llvm::Constant *GeneratePropertyList(const ObjCImplementationDecl *OID,
89 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
90 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000091 llvm::Constant *GenerateProtocolList(
92 const llvm::SmallVectorImpl<std::string> &Protocols);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +000093 // To ensure that all protocols are seen by the runtime, we add a category on
94 // a class defined in the runtime, declaring no methods, but adopting the
95 // protocols.
96 void GenerateProtocolHolderCategory(void);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000097 llvm::Constant *GenerateClassStructure(
98 llvm::Constant *MetaClass,
99 llvm::Constant *SuperClass,
100 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000101 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000102 llvm::Constant *Version,
103 llvm::Constant *InstanceSize,
104 llvm::Constant *IVars,
105 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000106 llvm::Constant *Protocols,
107 llvm::Constant *IvarOffsets,
108 llvm::Constant *Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000109 llvm::Constant *GenerateProtocolMethodList(
110 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
111 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes);
112 llvm::Constant *MakeConstantString(const std::string &Str, const std::string
113 &Name="");
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000114 llvm::Constant *ExportUniqueString(const std::string &Str, const std::string
115 prefix);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000116 llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
117 std::vector<llvm::Constant*> &V, const std::string &Name="");
118 llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
119 std::vector<llvm::Constant*> &V, const std::string &Name="");
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000120 llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
121 const ObjCIvarDecl *Ivar);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000122 void EmitClassRef(const std::string &className);
Chris Lattner0f984262008-03-01 08:50:34 +0000123public:
Chris Lattnerdce14062008-06-26 04:19:03 +0000124 CGObjCGNU(CodeGen::CodeGenModule &cgm);
Steve Naroff33fdb732009-03-31 16:53:37 +0000125 virtual llvm::Constant *GenerateConstantString(const ObjCStringLiteral *);
Mike Stump1eb44332009-09-09 15:08:12 +0000126 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000127 GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000128 QualType ResultType,
129 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000130 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000131 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000132 const CallArgList &CallArgs,
133 const ObjCMethodDecl *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000134 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000135 GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000136 QualType ResultType,
137 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000138 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000139 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000140 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000141 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000142 const CallArgList &CallArgs,
143 const ObjCMethodDecl *Method);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000144 virtual llvm::Value *GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000145 const ObjCInterfaceDecl *OID);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000146 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000147 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
148 *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000149
150 virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000151 const ObjCContainerDecl *CD);
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000152 virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
153 virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000154 virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000155 const ObjCProtocolDecl *PD);
156 virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000157 virtual llvm::Function *ModuleInitFunction();
Daniel Dunbar49f66022008-09-24 03:38:44 +0000158 virtual llvm::Function *GetPropertyGetFunction();
159 virtual llvm::Function *GetPropertySetFunction();
Daniel Dunbar309a4362009-07-24 07:40:24 +0000160 virtual llvm::Constant *EnumerationMutationFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000161
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000162 virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
163 const Stmt &S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000164 virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
165 const ObjCAtThrowStmt &S);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000166 virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6dc23172008-11-18 21:45:40 +0000167 llvm::Value *AddrWeakObj);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000168 virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
169 llvm::Value *src, llvm::Value *dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000170 virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
171 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +0000172 virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +0000173 llvm::Value *src, llvm::Value *dest,
174 llvm::Value *ivarOffset);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000175 virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
176 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000177 virtual void EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +0000178 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000179 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +0000180 QualType Ty);
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000181 virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
182 QualType ObjectTy,
183 llvm::Value *BaseValue,
184 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000185 unsigned CVRQualifiers);
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000186 virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +0000187 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000188 const ObjCIvarDecl *Ivar);
Chris Lattner0f984262008-03-01 08:50:34 +0000189};
190} // end anonymous namespace
191
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000192
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000193/// Emits a reference to a dummy variable which is emitted with each class.
194/// This ensures that a linker error will be generated when trying to link
195/// together modules where a referenced class is not defined.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000196void CGObjCGNU::EmitClassRef(const std::string &className) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000197 std::string symbolRef = "__objc_class_ref_" + className;
198 // Don't emit two copies of the same symbol
Mike Stumpbb1c8602009-07-31 21:31:32 +0000199 if (TheModule.getGlobalVariable(symbolRef))
200 return;
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000201 std::string symbolName = "__objc_class_name_" + className;
202 llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
203 if (!ClassSymbol) {
Owen Anderson1c431b32009-07-08 19:05:04 +0000204 ClassSymbol = new llvm::GlobalVariable(TheModule, LongTy, false,
205 llvm::GlobalValue::ExternalLinkage, 0, symbolName);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000206 }
Owen Anderson1c431b32009-07-08 19:05:04 +0000207 new llvm::GlobalVariable(TheModule, ClassSymbol->getType(), true,
Chris Lattnerf35271b2009-08-05 05:25:18 +0000208 llvm::GlobalValue::WeakAnyLinkage, ClassSymbol, symbolRef);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000209}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000210
211static std::string SymbolNameForClass(const std::string &ClassName) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000212 return "_OBJC_CLASS_" + ClassName;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000213}
214
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000215static std::string SymbolNameForMethod(const std::string &ClassName, const
216 std::string &CategoryName, const std::string &MethodName, bool isClassMethod)
217{
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000218 return "_OBJC_METHOD_" + ClassName + "("+CategoryName+")"+
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000219 (isClassMethod ? "+" : "-") + MethodName;
220}
221
Chris Lattnerdce14062008-06-26 04:19:03 +0000222CGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000223 : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
Owen Andersona1cf15f2009-07-14 23:10:40 +0000224 MetaClassPtrAlias(0), VMContext(cgm.getLLVMContext()) {
Chris Lattnere160c9b2009-01-27 05:06:01 +0000225 IntTy = cast<llvm::IntegerType>(
226 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
227 LongTy = cast<llvm::IntegerType>(
228 CGM.getTypes().ConvertType(CGM.getContext().LongTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000229
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000230 Int8Ty = llvm::Type::getInt8Ty(VMContext);
231 // C string type. Used in lots of places.
232 PtrToInt8Ty = llvm::PointerType::getUnqual(Int8Ty);
233
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000234 Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000235 Zeros[1] = Zeros[0];
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000236 NULLPtr = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Chris Lattner391d77a2008-03-30 23:03:07 +0000237 // Get the selector Type.
Chris Lattnere160c9b2009-01-27 05:06:01 +0000238 SelectorTy = cast<llvm::PointerType>(
239 CGM.getTypes().ConvertType(CGM.getContext().getObjCSelType()));
240
Owen Anderson96e0fc72009-07-29 22:16:19 +0000241 PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
Chris Lattner391d77a2008-03-30 23:03:07 +0000242 PtrTy = PtrToInt8Ty;
Mike Stump1eb44332009-09-09 15:08:12 +0000243
Chris Lattner391d77a2008-03-30 23:03:07 +0000244 // Object type
David Chisnall0f436562009-08-17 16:35:33 +0000245 ASTIdTy = CGM.getContext().getObjCIdType();
246 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000247
Chris Lattner391d77a2008-03-30 23:03:07 +0000248 // IMP type
249 std::vector<const llvm::Type*> IMPArgs;
250 IMPArgs.push_back(IdTy);
251 IMPArgs.push_back(SelectorTy);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000252 IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000253}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000254
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000255// This has to perform the lookup every time, since posing and related
256// techniques can modify the name -> class mapping.
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000257llvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000258 const ObjCInterfaceDecl *OID) {
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000259 llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000260 EmitClassRef(OID->getNameAsString());
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000261 ClassName = Builder.CreateStructGEP(ClassName, 0);
262
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000263 std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000264 llvm::Constant *ClassLookupFn =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000265 CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000266 Params,
267 true),
268 "objc_lookup_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000269 return Builder.CreateCall(ClassLookupFn, ClassName);
Chris Lattner391d77a2008-03-30 23:03:07 +0000270}
271
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000272llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel) {
Chris Lattner077bf5e2008-11-24 03:33:13 +0000273 llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
Chris Lattner8e67b632008-06-26 04:37:12 +0000274 if (US == 0)
Owen Anderson96e0fc72009-07-29 22:16:19 +0000275 US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000276 llvm::GlobalValue::PrivateLinkage,
277 ".objc_untyped_selector_alias"+Sel.getAsString(),
Chris Lattner8e67b632008-06-26 04:37:12 +0000278 NULL, &TheModule);
Mike Stump1eb44332009-09-09 15:08:12 +0000279
Chris Lattner8e67b632008-06-26 04:37:12 +0000280 return Builder.CreateLoad(US);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000281}
282
283llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
284 *Method) {
285
286 std::string SelName = Method->getSelector().getAsString();
287 std::string SelTypes;
288 CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
289 // Typed selectors
290 TypedSelector Selector = TypedSelector(SelName,
291 SelTypes);
292
293 // If it's already cached, return it.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000294 if (TypedSelectors[Selector]) {
295 return Builder.CreateLoad(TypedSelectors[Selector]);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000296 }
297
298 // If it isn't, cache it.
299 llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000300 llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000301 llvm::GlobalValue::PrivateLinkage, ".objc_selector_alias" + SelName,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000302 NULL, &TheModule);
303 TypedSelectors[Selector] = Sel;
304
305 return Builder.CreateLoad(Sel);
Chris Lattner8e67b632008-06-26 04:37:12 +0000306}
307
Chris Lattner5e7dcc62008-06-26 04:44:19 +0000308llvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
309 const std::string &Name) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000310 llvm::Constant *ConstStr = CGM.GetAddrOfConstantCString(Str, Name.c_str());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000311 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000312}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000313llvm::Constant *CGObjCGNU::ExportUniqueString(const std::string &Str,
314 const std::string prefix) {
315 std::string name = prefix + Str;
316 llvm::Constant *ConstStr = TheModule.getGlobalVariable(name);
317 if (!ConstStr) {
318 llvm::Constant *value = llvm::ConstantArray::get(VMContext, Str, true);
319 ConstStr = new llvm::GlobalVariable(TheModule, value->getType(), true,
320 llvm::GlobalValue::LinkOnceODRLinkage, value, prefix + Str);
321 }
322 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
323}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000324
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000325llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
326 std::vector<llvm::Constant*> &V, const std::string &Name) {
Owen Anderson08e25242009-07-27 22:29:56 +0000327 llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000328 return new llvm::GlobalVariable(TheModule, Ty, false,
329 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000330}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000331
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000332llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
333 std::vector<llvm::Constant*> &V, const std::string &Name) {
Owen Anderson7db6d832009-07-28 18:33:04 +0000334 llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000335 return new llvm::GlobalVariable(TheModule, Ty, false,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000336 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000337}
338
339/// Generate an NSConstantString object.
340//TODO: In case there are any crazy people still using the GNU runtime without
341//an OpenStep implementation, this should let them select their own class for
342//constant strings.
Steve Naroff33fdb732009-03-31 16:53:37 +0000343llvm::Constant *CGObjCGNU::GenerateConstantString(const ObjCStringLiteral *SL) {
Mike Stump1eb44332009-09-09 15:08:12 +0000344 std::string Str(SL->getString()->getStrData(),
Steve Naroff33fdb732009-03-31 16:53:37 +0000345 SL->getString()->getByteLength());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000346 std::vector<llvm::Constant*> Ivars;
347 Ivars.push_back(NULLPtr);
Chris Lattner13fd7e52008-06-21 21:44:18 +0000348 Ivars.push_back(MakeConstantString(Str));
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000349 Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000350 llvm::Constant *ObjCStr = MakeGlobal(
Owen Anderson47a434f2009-08-05 23:18:46 +0000351 llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000352 Ivars, ".objc_str");
353 ConstantStrings.push_back(
Owen Anderson3c4972d2009-07-29 18:54:39 +0000354 llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000355 return ObjCStr;
356}
357
358///Generates a message send where the super is the receiver. This is a message
359///send to self with special delivery semantics indicating which class's method
360///should be called.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000361CodeGen::RValue
362CGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000363 QualType ResultType,
364 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000365 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000366 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000367 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000368 bool IsClassMessage,
Daniel Dunbard6c93d72009-09-17 04:01:22 +0000369 const CallArgList &CallArgs,
370 const ObjCMethodDecl *Method) {
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000371 llvm::Value *cmd = GetSelector(CGF.Builder, Sel);
372
373 CallArgList ActualArgs;
374
375 ActualArgs.push_back(
David Chisnall0f436562009-08-17 16:35:33 +0000376 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
377 ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000378 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
379 CGF.getContext().getObjCSelType()));
380 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
381
382 CodeGenTypes &Types = CGM.getTypes();
383 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
Daniel Dunbar67939662009-09-17 04:01:40 +0000384 const llvm::FunctionType *impType =
385 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000386
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000387 llvm::Value *ReceiverClass = 0;
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000388 if (isCategoryImpl) {
389 llvm::Constant *classLookupFunction = 0;
390 std::vector<const llvm::Type*> Params;
391 Params.push_back(PtrTy);
392 if (IsClassMessage) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000393 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000394 IdTy, Params, true), "objc_get_meta_class");
395 } else {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000396 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000397 IdTy, Params, true), "objc_get_class");
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000398 }
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000399 ReceiverClass = CGF.Builder.CreateCall(classLookupFunction,
400 MakeConstantString(Class->getNameAsString()));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000401 } else {
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000402 // Set up global aliases for the metaclass or class pointer if they do not
403 // already exist. These will are forward-references which will be set to
Mike Stumpbb1c8602009-07-31 21:31:32 +0000404 // pointers to the class and metaclass structure created for the runtime
405 // load function. To send a message to super, we look up the value of the
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000406 // super_class pointer from either the class or metaclass structure.
407 if (IsClassMessage) {
408 if (!MetaClassPtrAlias) {
409 MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
410 llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
411 Class->getNameAsString(), NULL, &TheModule);
412 }
413 ReceiverClass = MetaClassPtrAlias;
414 } else {
415 if (!ClassPtrAlias) {
416 ClassPtrAlias = new llvm::GlobalAlias(IdTy,
417 llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
418 Class->getNameAsString(), NULL, &TheModule);
419 }
420 ReceiverClass = ClassPtrAlias;
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000421 }
Chris Lattner71238f62009-04-25 23:19:45 +0000422 }
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000423 // Cast the pointer to a simplified version of the class structure
Mike Stump1eb44332009-09-09 15:08:12 +0000424 ReceiverClass = CGF.Builder.CreateBitCast(ReceiverClass,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000425 llvm::PointerType::getUnqual(
Owen Anderson47a434f2009-08-05 23:18:46 +0000426 llvm::StructType::get(VMContext, IdTy, IdTy, NULL)));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000427 // Get the superclass pointer
428 ReceiverClass = CGF.Builder.CreateStructGEP(ReceiverClass, 1);
429 // Load the superclass pointer
430 ReceiverClass = CGF.Builder.CreateLoad(ReceiverClass);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000431 // Construct the structure used to look up the IMP
Owen Anderson47a434f2009-08-05 23:18:46 +0000432 llvm::StructType *ObjCSuperTy = llvm::StructType::get(VMContext,
433 Receiver->getType(), IdTy, NULL);
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000434 llvm::Value *ObjCSuper = CGF.Builder.CreateAlloca(ObjCSuperTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000435
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000436 CGF.Builder.CreateStore(Receiver, CGF.Builder.CreateStructGEP(ObjCSuper, 0));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000437 CGF.Builder.CreateStore(ReceiverClass,
438 CGF.Builder.CreateStructGEP(ObjCSuper, 1));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000439
440 // Get the IMP
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000441 std::vector<const llvm::Type*> Params;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000442 Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000443 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000444 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000445 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
446 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000447 "objc_msg_lookup_super");
448
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000449 llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000450 llvm::Value *imp = CGF.Builder.CreateCall(lookupFunction, lookupArgs,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000451 lookupArgs+2);
452
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000453 return CGF.EmitCall(FnInfo, imp, ActualArgs);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000454}
455
Mike Stump1eb44332009-09-09 15:08:12 +0000456/// Generate code for a message send expression.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000457CodeGen::RValue
458CGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000459 QualType ResultType,
460 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000461 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000462 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000463 const CallArgList &CallArgs,
464 const ObjCMethodDecl *Method) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000465 CGBuilderTy &Builder = CGF.Builder;
David Chisnall0f436562009-08-17 16:35:33 +0000466 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000467 llvm::Value *cmd;
468 if (Method)
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000469 cmd = GetSelector(Builder, Method);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000470 else
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000471 cmd = GetSelector(Builder, Sel);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000472 CallArgList ActualArgs;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000473
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000474 Receiver = Builder.CreateBitCast(Receiver, IdTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000475 ActualArgs.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000476 std::make_pair(RValue::get(Receiver), ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000477 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
478 CGF.getContext().getObjCSelType()));
479 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
480
481 CodeGenTypes &Types = CGM.getTypes();
482 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
Daniel Dunbar67939662009-09-17 04:01:40 +0000483 const llvm::FunctionType *impType =
484 Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000485
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000486 llvm::Value *imp;
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000487 // For sender-aware dispatch, we pass the sender as the third argument to a
488 // lookup function. When sending messages from C code, the sender is nil.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000489 // objc_msg_lookup_sender(id *receiver, SEL selector, id sender);
490 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
491
492 std::vector<const llvm::Type*> Params;
493 llvm::Value *ReceiverPtr = CGF.CreateTempAlloca(Receiver->getType());
494 Builder.CreateStore(Receiver, ReceiverPtr);
495 Params.push_back(ReceiverPtr->getType());
496 Params.push_back(SelectorTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000497 llvm::Value *self;
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000498
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000499 if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
500 self = CGF.LoadObjCSelf();
501 } else {
Owen Anderson03e20502009-07-30 23:11:26 +0000502 self = llvm::ConstantPointerNull::get(IdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000503 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000504
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000505 Params.push_back(self->getType());
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000506
507 // The lookup function returns a slot, which can be safely cached.
508 llvm::Type *SlotTy = llvm::StructType::get(VMContext, PtrTy, PtrTy, PtrTy,
509 IntTy, llvm::PointerType::getUnqual(impType), NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000510 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000511 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000512 llvm::PointerType::getUnqual(SlotTy), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000513 "objc_msg_lookup_sender");
514
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000515 // The lookup function is guaranteed not to capture the receiver pointer.
516 if (llvm::Function *LookupFn = dyn_cast<llvm::Function>(lookupFunction)) {
517 LookupFn->setDoesNotCapture(1);
518 }
519
520 llvm::Value *slot =
521 Builder.CreateCall3(lookupFunction, ReceiverPtr, cmd, self);
522 imp = Builder.CreateLoad(Builder.CreateStructGEP(slot, 4));
523 // The lookup function may have changed the receiver, so make sure we use
524 // the new one.
525 ActualArgs[0] =
526 std::make_pair(RValue::get(Builder.CreateLoad(ReceiverPtr)), ASTIdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000527 } else {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000528 std::vector<const llvm::Type*> Params;
529 Params.push_back(Receiver->getType());
530 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000531 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000532 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
533 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000534 "objc_msg_lookup");
535
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000536 imp = Builder.CreateCall2(lookupFunction, Receiver, cmd);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000537 }
Chris Lattner3eae03e2008-05-06 00:56:42 +0000538
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000539 return CGF.EmitCall(FnInfo, imp, ActualArgs);
Chris Lattner0f984262008-03-01 08:50:34 +0000540}
541
Mike Stump1eb44332009-09-09 15:08:12 +0000542/// Generates a MethodList. Used in construction of a objc_class and
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000543/// objc_category structures.
544llvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
Mike Stump1eb44332009-09-09 15:08:12 +0000545 const std::string &CategoryName,
546 const llvm::SmallVectorImpl<Selector> &MethodSels,
547 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000548 bool isClassMethodList) {
David Chisnall0f436562009-08-17 16:35:33 +0000549 if (MethodSels.empty())
550 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000551 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000552 llvm::StructType *ObjCMethodTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000553 PtrToInt8Ty, // Really a selector, but the runtime creates it us.
554 PtrToInt8Ty, // Method types
Owen Anderson96e0fc72009-07-29 22:16:19 +0000555 llvm::PointerType::getUnqual(IMPTy), //Method pointer
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000556 NULL);
557 std::vector<llvm::Constant*> Methods;
558 std::vector<llvm::Constant*> Elements;
559 for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
560 Elements.clear();
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000561 if (llvm::Constant *Method =
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000562 TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
Chris Lattner077bf5e2008-11-24 03:33:13 +0000563 MethodSels[i].getAsString(),
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000564 isClassMethodList))) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000565 llvm::Constant *C = MakeConstantString(MethodSels[i].getAsString());
566 Elements.push_back(C);
567 Elements.push_back(MethodTypes[i]);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000568 Method = llvm::ConstantExpr::getBitCast(Method,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000569 llvm::PointerType::getUnqual(IMPTy));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000570 Elements.push_back(Method);
Owen Anderson08e25242009-07-27 22:29:56 +0000571 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000572 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000573 }
574
575 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000576 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000577 Methods.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000578 llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
Chris Lattnerfba67632008-06-26 04:52:29 +0000579 Methods);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000580
581 // Structure containing list pointer, array and array count
582 llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
Owen Anderson8c8f69e2009-08-13 23:27:53 +0000583 llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get(VMContext);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000584 llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
Owen Anderson47a434f2009-08-05 23:18:46 +0000585 llvm::StructType *ObjCMethodListTy = llvm::StructType::get(VMContext,
Mike Stump1eb44332009-09-09 15:08:12 +0000586 NextPtrTy,
587 IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000588 ObjCMethodArrayTy,
589 NULL);
590 // Refine next pointer type to concrete type
591 llvm::cast<llvm::OpaqueType>(
592 OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
593 ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
594
595 Methods.clear();
Owen Anderson03e20502009-07-30 23:11:26 +0000596 Methods.push_back(llvm::ConstantPointerNull::get(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000597 llvm::PointerType::getUnqual(ObjCMethodListTy)));
Owen Anderson0032b272009-08-13 21:57:51 +0000598 Methods.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000599 MethodTypes.size()));
600 Methods.push_back(MethodArray);
Mike Stump1eb44332009-09-09 15:08:12 +0000601
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000602 // Create an instance of the structure
603 return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
604}
605
606/// Generates an IvarList. Used in construction of a objc_class.
607llvm::Constant *CGObjCGNU::GenerateIvarList(
608 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
609 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
610 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets) {
David Chisnall18044632009-11-16 19:05:54 +0000611 if (IvarNames.size() == 0)
612 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000613 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000614 llvm::StructType *ObjCIvarTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000615 PtrToInt8Ty,
616 PtrToInt8Ty,
617 IntTy,
618 NULL);
619 std::vector<llvm::Constant*> Ivars;
620 std::vector<llvm::Constant*> Elements;
621 for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
622 Elements.clear();
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000623 Elements.push_back(IvarNames[i]);
624 Elements.push_back(IvarTypes[i]);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000625 Elements.push_back(IvarOffsets[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000626 Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000627 }
628
629 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000630 llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000631 IvarNames.size());
632
Mike Stump1eb44332009-09-09 15:08:12 +0000633
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000634 Elements.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000635 Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
Owen Anderson7db6d832009-07-28 18:33:04 +0000636 Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000637 // Structure containing array and array count
Owen Anderson47a434f2009-08-05 23:18:46 +0000638 llvm::StructType *ObjCIvarListTy = llvm::StructType::get(VMContext, IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000639 ObjCIvarArrayTy,
640 NULL);
641
642 // Create an instance of the structure
643 return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
644}
645
646/// Generate a class structure
647llvm::Constant *CGObjCGNU::GenerateClassStructure(
648 llvm::Constant *MetaClass,
649 llvm::Constant *SuperClass,
650 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000651 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000652 llvm::Constant *Version,
653 llvm::Constant *InstanceSize,
654 llvm::Constant *IVars,
655 llvm::Constant *Methods,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000656 llvm::Constant *Protocols,
657 llvm::Constant *IvarOffsets,
658 llvm::Constant *Properties) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000659 // Set up the class structure
660 // Note: Several of these are char*s when they should be ids. This is
661 // because the runtime performs this translation on load.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000662 //
663 // Fields marked New ABI are part of the GNUstep runtime. We emit them
664 // anyway; the classes will still work with the GNU runtime, they will just
665 // be ignored.
Owen Anderson47a434f2009-08-05 23:18:46 +0000666 llvm::StructType *ClassTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000667 PtrToInt8Ty, // class_pointer
668 PtrToInt8Ty, // super_class
669 PtrToInt8Ty, // name
670 LongTy, // version
671 LongTy, // info
672 LongTy, // instance_size
673 IVars->getType(), // ivars
674 Methods->getType(), // methods
Mike Stump1eb44332009-09-09 15:08:12 +0000675 // These are all filled in by the runtime, so we pretend
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000676 PtrTy, // dtable
677 PtrTy, // subclass_list
678 PtrTy, // sibling_class
679 PtrTy, // protocols
680 PtrTy, // gc_object_type
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000681 // New ABI:
682 LongTy, // abi_version
683 IvarOffsets->getType(), // ivar_offsets
684 Properties->getType(), // properties
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000685 NULL);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000686 llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000687 // Fill in the structure
688 std::vector<llvm::Constant*> Elements;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000689 Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000690 Elements.push_back(SuperClass);
Chris Lattnerd002cc62008-06-26 04:47:04 +0000691 Elements.push_back(MakeConstantString(Name, ".class_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000692 Elements.push_back(Zero);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000693 Elements.push_back(llvm::ConstantInt::get(LongTy, info));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000694 Elements.push_back(InstanceSize);
695 Elements.push_back(IVars);
696 Elements.push_back(Methods);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000697 Elements.push_back(NULLPtr);
698 Elements.push_back(NULLPtr);
699 Elements.push_back(NULLPtr);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000700 Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000701 Elements.push_back(NULLPtr);
702 Elements.push_back(Zero);
703 Elements.push_back(IvarOffsets);
704 Elements.push_back(Properties);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000705 // Create an instance of the structure
Chris Lattner1565e032008-07-21 06:31:05 +0000706 return MakeGlobal(ClassTy, Elements, SymbolNameForClass(Name));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000707}
708
709llvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
710 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
711 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes) {
Mike Stump1eb44332009-09-09 15:08:12 +0000712 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000713 llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000714 PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
715 PtrToInt8Ty,
716 NULL);
717 std::vector<llvm::Constant*> Methods;
718 std::vector<llvm::Constant*> Elements;
719 for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
720 Elements.clear();
Mike Stump1eb44332009-09-09 15:08:12 +0000721 Elements.push_back(MethodNames[i]);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000722 Elements.push_back(MethodTypes[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000723 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000724 }
Owen Anderson96e0fc72009-07-29 22:16:19 +0000725 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000726 MethodNames.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000727 llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000728 Methods);
Owen Anderson47a434f2009-08-05 23:18:46 +0000729 llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000730 IntTy, ObjCMethodArrayTy, NULL);
731 Methods.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000732 Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000733 Methods.push_back(Array);
734 return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
735}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000736
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000737// Create the protocol list structure used in classes, categories and so on
738llvm::Constant *CGObjCGNU::GenerateProtocolList(
739 const llvm::SmallVectorImpl<std::string> &Protocols) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000740 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000741 Protocols.size());
Owen Anderson47a434f2009-08-05 23:18:46 +0000742 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000743 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
744 LongTy,//FIXME: Should be size_t
745 ProtocolArrayTy,
746 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000747 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000748 for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
749 iter != endIter ; iter++) {
David Chisnallff80fab2009-11-20 14:50:59 +0000750 llvm::Constant *protocol = 0;
751 llvm::StringMap<llvm::Constant*>::iterator value =
752 ExistingProtocols.find(*iter);
753 if (value == ExistingProtocols.end()) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000754 protocol = GenerateEmptyProtocol(*iter);
David Chisnallff80fab2009-11-20 14:50:59 +0000755 } else {
756 protocol = value->getValue();
757 }
Owen Anderson3c4972d2009-07-29 18:54:39 +0000758 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(protocol,
Owen Andersona1cf15f2009-07-14 23:10:40 +0000759 PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000760 Elements.push_back(Ptr);
761 }
Owen Anderson7db6d832009-07-28 18:33:04 +0000762 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000763 Elements);
764 Elements.clear();
765 Elements.push_back(NULLPtr);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000766 Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000767 Elements.push_back(ProtocolArray);
768 return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
769}
770
Mike Stump1eb44332009-09-09 15:08:12 +0000771llvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000772 const ObjCProtocolDecl *PD) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000773 llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
Mike Stump1eb44332009-09-09 15:08:12 +0000774 const llvm::Type *T =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000775 CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
Owen Anderson96e0fc72009-07-29 22:16:19 +0000776 return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000777}
778
779llvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
780 const std::string &ProtocolName) {
781 llvm::SmallVector<std::string, 0> EmptyStringVector;
782 llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
783
784 llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000785 llvm::Constant *MethodList =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000786 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
787 // Protocols are objects containing lists of the methods implemented and
788 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000789 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000790 PtrToInt8Ty,
791 ProtocolList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000792 MethodList->getType(),
793 MethodList->getType(),
794 MethodList->getType(),
795 MethodList->getType(),
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000796 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000797 std::vector<llvm::Constant*> Elements;
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000798 // The isa pointer must be set to a magic number so the runtime knows it's
799 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000800 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
801 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000802 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000803 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000804 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
805 Elements.push_back(ProtocolList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000806 Elements.push_back(MethodList);
807 Elements.push_back(MethodList);
808 Elements.push_back(MethodList);
809 Elements.push_back(MethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000810 return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000811}
812
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000813void CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
814 ASTContext &Context = CGM.getContext();
Chris Lattner8ec03f52008-11-24 03:54:41 +0000815 std::string ProtocolName = PD->getNameAsString();
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000816 llvm::SmallVector<std::string, 16> Protocols;
817 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
818 E = PD->protocol_end(); PI != E; ++PI)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000819 Protocols.push_back((*PI)->getNameAsString());
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000820 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
821 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000822 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodNames;
823 llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodTypes;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000824 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
825 E = PD->instmeth_end(); iter != E; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000826 std::string TypeStr;
827 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000828 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
829 InstanceMethodNames.push_back(
830 MakeConstantString((*iter)->getSelector().getAsString()));
831 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
832 } else {
833 OptionalInstanceMethodNames.push_back(
834 MakeConstantString((*iter)->getSelector().getAsString()));
835 OptionalInstanceMethodTypes.push_back(MakeConstantString(TypeStr));
836 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000837 }
838 // Collect information about class methods:
839 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
840 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000841 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodNames;
842 llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +0000843 for (ObjCProtocolDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000844 iter = PD->classmeth_begin(), endIter = PD->classmeth_end();
845 iter != endIter ; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000846 std::string TypeStr;
847 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000848 if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
849 ClassMethodNames.push_back(
850 MakeConstantString((*iter)->getSelector().getAsString()));
851 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
852 } else {
853 OptionalClassMethodNames.push_back(
854 MakeConstantString((*iter)->getSelector().getAsString()));
855 OptionalClassMethodTypes.push_back(MakeConstantString(TypeStr));
856 }
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000857 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000858
859 llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
860 llvm::Constant *InstanceMethodList =
861 GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
862 llvm::Constant *ClassMethodList =
863 GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000864 llvm::Constant *OptionalInstanceMethodList =
865 GenerateProtocolMethodList(OptionalInstanceMethodNames,
866 OptionalInstanceMethodTypes);
867 llvm::Constant *OptionalClassMethodList =
868 GenerateProtocolMethodList(OptionalClassMethodNames,
869 OptionalClassMethodTypes);
870
871 // Property metadata: name, attributes, isSynthesized, setter name, setter
872 // types, getter name, getter types.
873 // The isSynthesized value is always set to 0 in a protocol. It exists to
874 // simplify the runtime library by allowing it to use the same data
875 // structures for protocol metadata everywhere.
876 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
877 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
878 PtrToInt8Ty, NULL);
879 std::vector<llvm::Constant*> Properties;
880 std::vector<llvm::Constant*> OptionalProperties;
881
882 // Add all of the property methods need adding to the method list and to the
883 // property metadata list.
884 for (ObjCContainerDecl::prop_iterator
885 iter = PD->prop_begin(), endIter = PD->prop_end();
886 iter != endIter ; iter++) {
887 std::vector<llvm::Constant*> Fields;
888 ObjCPropertyDecl *property = (*iter);
889
890 Fields.push_back(MakeConstantString(property->getNameAsString()));
891 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
892 property->getPropertyAttributes()));
893 Fields.push_back(llvm::ConstantInt::get(Int8Ty, 0));
894 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
895 std::string TypeStr;
896 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
897 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
898 InstanceMethodTypes.push_back(TypeEncoding);
899 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
900 Fields.push_back(TypeEncoding);
901 } else {
902 Fields.push_back(NULLPtr);
903 Fields.push_back(NULLPtr);
904 }
905 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
906 std::string TypeStr;
907 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
908 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
909 InstanceMethodTypes.push_back(TypeEncoding);
910 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
911 Fields.push_back(TypeEncoding);
912 } else {
913 Fields.push_back(NULLPtr);
914 Fields.push_back(NULLPtr);
915 }
916 if (property->getPropertyImplementation() == ObjCPropertyDecl::Optional) {
917 OptionalProperties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
918 } else {
919 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
920 }
921 }
922 llvm::Constant *PropertyArray = llvm::ConstantArray::get(
923 llvm::ArrayType::get(PropertyMetadataTy, Properties.size()), Properties);
924 llvm::Constant* PropertyListInitFields[] =
925 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
926
927 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000928 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000929 llvm::Constant *PropertyList = new llvm::GlobalVariable(TheModule,
930 PropertyListInit->getType(), false, llvm::GlobalValue::InternalLinkage,
931 PropertyListInit, ".objc_property_list");
932
933 llvm::Constant *OptionalPropertyArray =
934 llvm::ConstantArray::get(llvm::ArrayType::get(PropertyMetadataTy,
935 OptionalProperties.size()) , OptionalProperties);
936 llvm::Constant* OptionalPropertyListInitFields[] = {
937 llvm::ConstantInt::get(IntTy, OptionalProperties.size()), NULLPtr,
938 OptionalPropertyArray };
939
940 llvm::Constant *OptionalPropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +0000941 llvm::ConstantStruct::get(VMContext, OptionalPropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000942 llvm::Constant *OptionalPropertyList = new llvm::GlobalVariable(TheModule,
943 OptionalPropertyListInit->getType(), false,
944 llvm::GlobalValue::InternalLinkage, OptionalPropertyListInit,
945 ".objc_property_list");
946
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000947 // Protocols are objects containing lists of the methods implemented and
948 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000949 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000950 PtrToInt8Ty,
951 ProtocolList->getType(),
952 InstanceMethodList->getType(),
953 ClassMethodList->getType(),
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000954 OptionalInstanceMethodList->getType(),
955 OptionalClassMethodList->getType(),
956 PropertyList->getType(),
957 OptionalPropertyList->getType(),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000958 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000959 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000960 // The isa pointer must be set to a magic number so the runtime knows it's
961 // the correct layout.
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000962 int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
963 NonFragileProtocolVersion : ProtocolVersion;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000964 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000965 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000966 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
967 Elements.push_back(ProtocolList);
968 Elements.push_back(InstanceMethodList);
969 Elements.push_back(ClassMethodList);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000970 Elements.push_back(OptionalInstanceMethodList);
971 Elements.push_back(OptionalClassMethodList);
972 Elements.push_back(PropertyList);
973 Elements.push_back(OptionalPropertyList);
Mike Stump1eb44332009-09-09 15:08:12 +0000974 ExistingProtocols[ProtocolName] =
Owen Anderson3c4972d2009-07-29 18:54:39 +0000975 llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000976 ".objc_protocol"), IdTy);
977}
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +0000978void CGObjCGNU::GenerateProtocolHolderCategory(void) {
979 // Collect information about instance methods
980 llvm::SmallVector<Selector, 1> MethodSels;
981 llvm::SmallVector<llvm::Constant*, 1> MethodTypes;
982
983 std::vector<llvm::Constant*> Elements;
984 const std::string ClassName = "__ObjC_Protocol_Holder_Ugly_Hack";
985 const std::string CategoryName = "AnotherHack";
986 Elements.push_back(MakeConstantString(CategoryName));
987 Elements.push_back(MakeConstantString(ClassName));
988 // Instance method list
989 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
990 ClassName, CategoryName, MethodSels, MethodTypes, false), PtrTy));
991 // Class method list
992 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
993 ClassName, CategoryName, MethodSels, MethodTypes, true), PtrTy));
994 // Protocol list
995 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrTy,
996 ExistingProtocols.size());
997 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
998 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
999 LongTy,//FIXME: Should be size_t
1000 ProtocolArrayTy,
1001 NULL);
1002 std::vector<llvm::Constant*> ProtocolElements;
1003 for (llvm::StringMapIterator<llvm::Constant*> iter =
1004 ExistingProtocols.begin(), endIter = ExistingProtocols.end();
1005 iter != endIter ; iter++) {
1006 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(iter->getValue(),
1007 PtrTy);
1008 ProtocolElements.push_back(Ptr);
1009 }
1010 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
1011 ProtocolElements);
1012 ProtocolElements.clear();
1013 ProtocolElements.push_back(NULLPtr);
1014 ProtocolElements.push_back(llvm::ConstantInt::get(LongTy,
1015 ExistingProtocols.size()));
1016 ProtocolElements.push_back(ProtocolArray);
1017 Elements.push_back(llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolListTy,
1018 ProtocolElements, ".objc_protocol_list"), PtrTy));
1019 Categories.push_back(llvm::ConstantExpr::getBitCast(
1020 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
1021 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
1022}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001023
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001024void CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001025 std::string ClassName = OCD->getClassInterface()->getNameAsString();
1026 std::string CategoryName = OCD->getNameAsString();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001027 // Collect information about instance methods
1028 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1029 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001030 for (ObjCCategoryImplDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001031 iter = OCD->instmeth_begin(), endIter = OCD->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001032 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001033 InstanceMethodSels.push_back((*iter)->getSelector());
1034 std::string TypeStr;
1035 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001036 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001037 }
1038
1039 // Collect information about class methods
1040 llvm::SmallVector<Selector, 16> ClassMethodSels;
1041 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001042 for (ObjCCategoryImplDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001043 iter = OCD->classmeth_begin(), endIter = OCD->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001044 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001045 ClassMethodSels.push_back((*iter)->getSelector());
1046 std::string TypeStr;
1047 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001048 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001049 }
1050
1051 // Collect the names of referenced protocols
1052 llvm::SmallVector<std::string, 16> Protocols;
1053 const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
1054 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1055 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1056 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001057 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001058
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001059 std::vector<llvm::Constant*> Elements;
1060 Elements.push_back(MakeConstantString(CategoryName));
1061 Elements.push_back(MakeConstantString(ClassName));
Mike Stump1eb44332009-09-09 15:08:12 +00001062 // Instance method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001063 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001064 ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001065 false), PtrTy));
1066 // Class method list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001067 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +00001068 ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001069 PtrTy));
1070 // Protocol list
Owen Anderson3c4972d2009-07-29 18:54:39 +00001071 Elements.push_back(llvm::ConstantExpr::getBitCast(
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001072 GenerateProtocolList(Protocols), PtrTy));
Owen Anderson3c4972d2009-07-29 18:54:39 +00001073 Categories.push_back(llvm::ConstantExpr::getBitCast(
Mike Stump1eb44332009-09-09 15:08:12 +00001074 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
Owen Anderson47a434f2009-08-05 23:18:46 +00001075 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001076}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001077
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001078llvm::Constant *CGObjCGNU::GeneratePropertyList(const ObjCImplementationDecl *OID,
1079 llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
1080 llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes) {
1081 ASTContext &Context = CGM.getContext();
1082 //
1083 // Property metadata: name, attributes, isSynthesized, setter name, setter
1084 // types, getter name, getter types.
1085 llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1086 PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1087 PtrToInt8Ty, NULL);
1088 std::vector<llvm::Constant*> Properties;
1089
1090
1091 // Add all of the property methods need adding to the method list and to the
1092 // property metadata list.
1093 for (ObjCImplDecl::propimpl_iterator
1094 iter = OID->propimpl_begin(), endIter = OID->propimpl_end();
1095 iter != endIter ; iter++) {
1096 std::vector<llvm::Constant*> Fields;
1097 ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
1098
1099 Fields.push_back(MakeConstantString(property->getNameAsString()));
1100 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1101 property->getPropertyAttributes()));
1102 Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1103 (*iter)->getPropertyImplementation() ==
1104 ObjCPropertyImplDecl::Synthesize));
1105 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
1106 InstanceMethodSels.push_back(getter->getSelector());
1107 std::string TypeStr;
1108 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1109 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1110 InstanceMethodTypes.push_back(TypeEncoding);
1111 Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1112 Fields.push_back(TypeEncoding);
1113 } else {
1114 Fields.push_back(NULLPtr);
1115 Fields.push_back(NULLPtr);
1116 }
1117 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1118 InstanceMethodSels.push_back(setter->getSelector());
1119 std::string TypeStr;
1120 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1121 llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1122 InstanceMethodTypes.push_back(TypeEncoding);
1123 Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1124 Fields.push_back(TypeEncoding);
1125 } else {
1126 Fields.push_back(NULLPtr);
1127 Fields.push_back(NULLPtr);
1128 }
1129 Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1130 }
1131 llvm::ArrayType *PropertyArrayTy =
1132 llvm::ArrayType::get(PropertyMetadataTy, Properties.size());
1133 llvm::Constant *PropertyArray = llvm::ConstantArray::get(PropertyArrayTy,
1134 Properties);
1135 llvm::Constant* PropertyListInitFields[] =
1136 {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1137
1138 llvm::Constant *PropertyListInit =
Nick Lewycky0d36dd22009-09-19 20:00:52 +00001139 llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001140 return new llvm::GlobalVariable(TheModule, PropertyListInit->getType(), false,
1141 llvm::GlobalValue::InternalLinkage, PropertyListInit,
1142 ".objc_property_list");
1143}
1144
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001145void CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
1146 ASTContext &Context = CGM.getContext();
1147
1148 // Get the superclass name.
Mike Stump1eb44332009-09-09 15:08:12 +00001149 const ObjCInterfaceDecl * SuperClassDecl =
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001150 OID->getClassInterface()->getSuperClass();
Chris Lattner8ec03f52008-11-24 03:54:41 +00001151 std::string SuperClassName;
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001152 if (SuperClassDecl) {
Chris Lattner8ec03f52008-11-24 03:54:41 +00001153 SuperClassName = SuperClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001154 EmitClassRef(SuperClassName);
1155 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001156
1157 // Get the class name
Chris Lattner09dc6662009-04-01 02:00:48 +00001158 ObjCInterfaceDecl *ClassDecl =
1159 const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
Chris Lattner8ec03f52008-11-24 03:54:41 +00001160 std::string ClassName = ClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +00001161 // Emit the symbol that is used to generate linker errors if this class is
1162 // referenced in other modules but not declared.
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001163 std::string classSymbolName = "__objc_class_name_" + ClassName;
Mike Stump1eb44332009-09-09 15:08:12 +00001164 if (llvm::GlobalVariable *symbol =
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001165 TheModule.getGlobalVariable(classSymbolName)) {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001166 symbol->setInitializer(llvm::ConstantInt::get(LongTy, 0));
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001167 } else {
Owen Anderson1c431b32009-07-08 19:05:04 +00001168 new llvm::GlobalVariable(TheModule, LongTy, false,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001169 llvm::GlobalValue::ExternalLinkage, llvm::ConstantInt::get(LongTy, 0),
Owen Anderson1c431b32009-07-08 19:05:04 +00001170 classSymbolName);
Fariborz Jahanianc51db232009-07-03 15:10:14 +00001171 }
Mike Stump1eb44332009-09-09 15:08:12 +00001172
Daniel Dunbar2bebbf02009-05-03 10:46:44 +00001173 // Get the size of instances.
1174 int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001175
1176 // Collect information about instance variables.
1177 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
1178 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
1179 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
Mike Stump1eb44332009-09-09 15:08:12 +00001180
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001181 std::vector<llvm::Constant*> IvarOffsetValues;
1182
Mike Stump1eb44332009-09-09 15:08:12 +00001183 int superInstanceSize = !SuperClassDecl ? 0 :
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001184 Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
1185 // For non-fragile ivars, set the instance size to 0 - {the size of just this
1186 // class}. The runtime will then set this to the correct value on load.
1187 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
1188 instanceSize = 0 - (instanceSize - superInstanceSize);
1189 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001190 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1191 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1192 // Store the name
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001193 IvarNames.push_back(MakeConstantString((*iter)->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001194 // Get the type encoding for this ivar
1195 std::string TypeStr;
Daniel Dunbar0d504c12008-10-17 20:21:44 +00001196 Context.getObjCEncodingForType((*iter)->getType(), TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001197 IvarTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001198 // Get the offset
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001199 uint64_t BaseOffset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter);
David Chisnallaecbf242009-11-17 19:32:15 +00001200 uint64_t Offset = BaseOffset;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001201 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001202 Offset = BaseOffset - superInstanceSize;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001203 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001204 IvarOffsets.push_back(
Owen Anderson0032b272009-08-13 21:57:51 +00001205 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset));
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001206 IvarOffsetValues.push_back(new llvm::GlobalVariable(TheModule, IntTy,
1207 false, llvm::GlobalValue::ExternalLinkage,
1208 llvm::ConstantInt::get(IntTy, BaseOffset),
1209 "__objc_ivar_offset_value_" + ClassName +"." +
1210 (*iter)->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001211 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001212 llvm::Constant *IvarOffsetArrayInit =
1213 llvm::ConstantArray::get(llvm::ArrayType::get(PtrToIntTy,
1214 IvarOffsetValues.size()), IvarOffsetValues);
1215 llvm::GlobalVariable *IvarOffsetArray = new llvm::GlobalVariable(TheModule,
1216 IvarOffsetArrayInit->getType(), false,
1217 llvm::GlobalValue::InternalLinkage, IvarOffsetArrayInit,
1218 ".ivar.offsets");
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001219
1220 // Collect information about instance methods
1221 llvm::SmallVector<Selector, 16> InstanceMethodSels;
1222 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +00001223 for (ObjCImplementationDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001224 iter = OID->instmeth_begin(), endIter = OID->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001225 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001226 InstanceMethodSels.push_back((*iter)->getSelector());
1227 std::string TypeStr;
1228 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001229 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001230 }
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001231
1232 llvm::Constant *Properties = GeneratePropertyList(OID, InstanceMethodSels,
1233 InstanceMethodTypes);
1234
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001235
1236 // Collect information about class methods
1237 llvm::SmallVector<Selector, 16> ClassMethodSels;
1238 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +00001239 for (ObjCImplementationDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +00001240 iter = OID->classmeth_begin(), endIter = OID->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +00001241 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001242 ClassMethodSels.push_back((*iter)->getSelector());
1243 std::string TypeStr;
1244 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001245 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001246 }
1247 // Collect the names of referenced protocols
1248 llvm::SmallVector<std::string, 16> Protocols;
1249 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
1250 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
1251 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +00001252 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001253
1254
1255
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001256 // Get the superclass pointer.
1257 llvm::Constant *SuperClass;
Chris Lattner8ec03f52008-11-24 03:54:41 +00001258 if (!SuperClassName.empty()) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001259 SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
1260 } else {
Owen Anderson03e20502009-07-30 23:11:26 +00001261 SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001262 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001263 // Empty vector used to construct empty method lists
1264 llvm::SmallVector<llvm::Constant*, 1> empty;
1265 // Generate the method and instance variable lists
1266 llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001267 InstanceMethodSels, InstanceMethodTypes, false);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001268 llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +00001269 ClassMethodSels, ClassMethodTypes, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001270 llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
1271 IvarOffsets);
Mike Stump1eb44332009-09-09 15:08:12 +00001272 // Irrespective of whether we are compiling for a fragile or non-fragile ABI,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001273 // we emit a symbol containing the offset for each ivar in the class. This
1274 // allows code compiled for the non-Fragile ABI to inherit from code compiled
1275 // for the legacy ABI, without causing problems. The converse is also
1276 // possible, but causes all ivar accesses to be fragile.
1277 int i = 0;
1278 // Offset pointer for getting at the correct field in the ivar list when
1279 // setting up the alias. These are: The base address for the global, the
1280 // ivar array (second field), the ivar in this list (set for each ivar), and
1281 // the offset (third field in ivar structure)
1282 const llvm::Type *IndexTy = llvm::Type::getInt32Ty(VMContext);
1283 llvm::Constant *offsetPointerIndexes[] = {Zeros[0],
Mike Stump1eb44332009-09-09 15:08:12 +00001284 llvm::ConstantInt::get(IndexTy, 1), 0,
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001285 llvm::ConstantInt::get(IndexTy, 2) };
1286
1287 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
1288 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
1289 const std::string Name = "__objc_ivar_offset_" + ClassName + '.'
1290 +(*iter)->getNameAsString();
1291 offsetPointerIndexes[2] = llvm::ConstantInt::get(IndexTy, i++);
1292 // Get the correct ivar field
1293 llvm::Constant *offsetValue = llvm::ConstantExpr::getGetElementPtr(
1294 IvarList, offsetPointerIndexes, 4);
1295 // Get the existing alias, if one exists.
1296 llvm::GlobalVariable *offset = TheModule.getNamedGlobal(Name);
1297 if (offset) {
1298 offset->setInitializer(offsetValue);
1299 // If this is the real definition, change its linkage type so that
1300 // different modules will use this one, rather than their private
1301 // copy.
1302 offset->setLinkage(llvm::GlobalValue::ExternalLinkage);
1303 } else {
1304 // Add a new alias if there isn't one already.
1305 offset = new llvm::GlobalVariable(TheModule, offsetValue->getType(),
1306 false, llvm::GlobalValue::ExternalLinkage, offsetValue, Name);
1307 }
1308 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001309 //Generate metaclass for class methods
1310 llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
David Chisnall18044632009-11-16 19:05:54 +00001311 NULLPtr, 0x12L, ClassName.c_str(), 0, Zeros[0], GenerateIvarList(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001312 empty, empty, empty), ClassMethodList, NULLPtr, NULLPtr, NULLPtr);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001313
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001314 // Generate the class structure
Chris Lattner8ec03f52008-11-24 03:54:41 +00001315 llvm::Constant *ClassStruct =
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001316 GenerateClassStructure(MetaClassStruct, SuperClass, 0x11L,
Chris Lattner8ec03f52008-11-24 03:54:41 +00001317 ClassName.c_str(), 0,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001318 llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001319 MethodList, GenerateProtocolList(Protocols), IvarOffsetArray,
1320 Properties);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001321
1322 // Resolve the class aliases, if they exist.
1323 if (ClassPtrAlias) {
1324 ClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001325 llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001326 ClassPtrAlias = 0;
1327 }
1328 if (MetaClassPtrAlias) {
1329 MetaClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001330 llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001331 MetaClassPtrAlias = 0;
1332 }
1333
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001334 // Add class structure to list to be added to the symtab later
Owen Anderson3c4972d2009-07-29 18:54:39 +00001335 ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001336 Classes.push_back(ClassStruct);
1337}
1338
Fariborz Jahanianc38e9af2009-06-23 21:47:46 +00001339
Mike Stump1eb44332009-09-09 15:08:12 +00001340llvm::Function *CGObjCGNU::ModuleInitFunction() {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001341 // Only emit an ObjC load function if no Objective-C stuff has been called
1342 if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
1343 ExistingProtocols.empty() && TypedSelectors.empty() &&
Anton Korobeynikov5f58b912008-06-01 15:14:46 +00001344 UntypedSelectors.empty())
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001345 return NULL;
Eli Friedman1b8956e2008-06-01 16:00:02 +00001346
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001347 // Add all referenced protocols to a category.
1348 GenerateProtocolHolderCategory();
1349
Chris Lattnere160c9b2009-01-27 05:06:01 +00001350 const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1351 SelectorTy->getElementType());
1352 const llvm::Type *SelStructPtrTy = SelectorTy;
1353 bool isSelOpaque = false;
1354 if (SelStructTy == 0) {
Owen Anderson47a434f2009-08-05 23:18:46 +00001355 SelStructTy = llvm::StructType::get(VMContext, PtrToInt8Ty,
1356 PtrToInt8Ty, NULL);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001357 SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
Chris Lattnere160c9b2009-01-27 05:06:01 +00001358 isSelOpaque = true;
1359 }
1360
Eli Friedman1b8956e2008-06-01 16:00:02 +00001361 // Name the ObjC types to make the IR a bit easier to read
Chris Lattnere160c9b2009-01-27 05:06:01 +00001362 TheModule.addTypeName(".objc_selector", SelStructPtrTy);
Eli Friedman1b8956e2008-06-01 16:00:02 +00001363 TheModule.addTypeName(".objc_id", IdTy);
1364 TheModule.addTypeName(".objc_imp", IMPTy);
1365
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001366 std::vector<llvm::Constant*> Elements;
Chris Lattner71238f62009-04-25 23:19:45 +00001367 llvm::Constant *Statics = NULLPtr;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001368 // Generate statics list:
Chris Lattner71238f62009-04-25 23:19:45 +00001369 if (ConstantStrings.size()) {
Owen Anderson96e0fc72009-07-29 22:16:19 +00001370 llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Chris Lattner71238f62009-04-25 23:19:45 +00001371 ConstantStrings.size() + 1);
1372 ConstantStrings.push_back(NULLPtr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001373
Daniel Dunbar1b096952009-11-29 02:38:47 +00001374 llvm::StringRef StringClass = CGM.getLangOptions().ObjCConstantStringClass;
1375 if (StringClass.empty()) StringClass = "NXConstantString";
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001376 Elements.push_back(MakeConstantString(StringClass,
1377 ".objc_static_class_name"));
Owen Anderson7db6d832009-07-28 18:33:04 +00001378 Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
Chris Lattner71238f62009-04-25 23:19:45 +00001379 ConstantStrings));
Mike Stump1eb44332009-09-09 15:08:12 +00001380 llvm::StructType *StaticsListTy =
Owen Anderson47a434f2009-08-05 23:18:46 +00001381 llvm::StructType::get(VMContext, PtrToInt8Ty, StaticsArrayTy, NULL);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001382 llvm::Type *StaticsListPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001383 llvm::PointerType::getUnqual(StaticsListTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001384 Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
Mike Stump1eb44332009-09-09 15:08:12 +00001385 llvm::ArrayType *StaticsListArrayTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001386 llvm::ArrayType::get(StaticsListPtrTy, 2);
Chris Lattner71238f62009-04-25 23:19:45 +00001387 Elements.clear();
1388 Elements.push_back(Statics);
Owen Andersonc9c88b42009-07-31 20:28:54 +00001389 Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
Chris Lattner71238f62009-04-25 23:19:45 +00001390 Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001391 Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001392 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001393 // Array of classes, categories, and constant objects
Owen Anderson96e0fc72009-07-29 22:16:19 +00001394 llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001395 Classes.size() + Categories.size() + 2);
Mike Stump1eb44332009-09-09 15:08:12 +00001396 llvm::StructType *SymTabTy = llvm::StructType::get(VMContext,
Owen Anderson47a434f2009-08-05 23:18:46 +00001397 LongTy, SelStructPtrTy,
Owen Anderson0032b272009-08-13 21:57:51 +00001398 llvm::Type::getInt16Ty(VMContext),
1399 llvm::Type::getInt16Ty(VMContext),
Chris Lattner630404b2008-06-26 04:10:42 +00001400 ClassListTy, NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001401
1402 Elements.clear();
1403 // Pointer to an array of selectors used in this module.
1404 std::vector<llvm::Constant*> Selectors;
1405 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1406 iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
1407 iter != iterEnd ; ++iter) {
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001408 Elements.push_back(ExportUniqueString(iter->first.first, ".objc_sel_name"));
David Chisnalla7c5b082009-09-14 19:04:10 +00001409 Elements.push_back(MakeConstantString(iter->first.second,
Chris Lattner630404b2008-06-26 04:10:42 +00001410 ".objc_sel_types"));
Owen Anderson08e25242009-07-27 22:29:56 +00001411 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001412 Elements.clear();
1413 }
1414 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1415 iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
Chris Lattner630404b2008-06-26 04:10:42 +00001416 iter != iterEnd; ++iter) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001417 Elements.push_back(
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001418 ExportUniqueString(iter->getKeyData(), ".objc_sel_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001419 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001420 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001421 Elements.clear();
1422 }
1423 Elements.push_back(NULLPtr);
1424 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001425 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001426 Elements.clear();
1427 // Number of static selectors
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001428 Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001429 llvm::Constant *SelectorList = MakeGlobal(
Owen Anderson96e0fc72009-07-29 22:16:19 +00001430 llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001431 ".objc_selector_list");
Mike Stump1eb44332009-09-09 15:08:12 +00001432 Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
Chris Lattnere160c9b2009-01-27 05:06:01 +00001433 SelStructPtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001434
1435 // Now that all of the static selectors exist, create pointers to them.
1436 int index = 0;
1437 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1438 iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
1439 iter != iterEnd; ++iter) {
1440 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001441 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Owen Anderson1c431b32009-07-08 19:05:04 +00001442 llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
1443 true, llvm::GlobalValue::InternalLinkage,
Owen Anderson3c4972d2009-07-29 18:54:39 +00001444 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
Owen Anderson1c431b32009-07-08 19:05:04 +00001445 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001446 // If selectors are defined as an opaque type, cast the pointer to this
1447 // type.
1448 if (isSelOpaque) {
Owen Anderson3c4972d2009-07-29 18:54:39 +00001449 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001450 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001451 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001452 (*iter).second->setAliasee(SelPtr);
1453 }
1454 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1455 iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1456 iter != iterEnd; iter++) {
1457 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001458 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Mike Stumpbb1c8602009-07-31 21:31:32 +00001459 llvm::Constant *SelPtr = new llvm::GlobalVariable
Mike Stump1eb44332009-09-09 15:08:12 +00001460 (TheModule, SelStructPtrTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +00001461 true, llvm::GlobalValue::InternalLinkage,
1462 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1463 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001464 // If selectors are defined as an opaque type, cast the pointer to this
1465 // type.
1466 if (isSelOpaque) {
Owen Anderson3c4972d2009-07-29 18:54:39 +00001467 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001468 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001469 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001470 (*iter).second->setAliasee(SelPtr);
1471 }
1472 // Number of classes defined.
Mike Stump1eb44332009-09-09 15:08:12 +00001473 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001474 Classes.size()));
1475 // Number of categories defined
Mike Stump1eb44332009-09-09 15:08:12 +00001476 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001477 Categories.size()));
1478 // Create an array of classes, then categories, then static object instances
1479 Classes.insert(Classes.end(), Categories.begin(), Categories.end());
1480 // NULL-terminated list of static object instances (mainly constant strings)
1481 Classes.push_back(Statics);
1482 Classes.push_back(NULLPtr);
Owen Anderson7db6d832009-07-28 18:33:04 +00001483 llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001484 Elements.push_back(ClassList);
Mike Stump1eb44332009-09-09 15:08:12 +00001485 // Construct the symbol table
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001486 llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
1487
1488 // The symbol table is contained in a module which has some version-checking
1489 // constants
Owen Anderson47a434f2009-08-05 23:18:46 +00001490 llvm::StructType * ModuleTy = llvm::StructType::get(VMContext, LongTy, LongTy,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001491 PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001492 Elements.clear();
1493 // Runtime version used for compatibility checking.
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001494 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Mike Stump1eb44332009-09-09 15:08:12 +00001495 Elements.push_back(llvm::ConstantInt::get(LongTy,
Fariborz Jahaniand9a1db32009-09-10 21:48:21 +00001496 NonFragileRuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001497 } else {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001498 Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001499 }
Fariborz Jahanian91a0b512009-04-01 19:49:42 +00001500 // sizeof(ModuleTy)
1501 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001502 Elements.push_back(llvm::ConstantInt::get(LongTy,
Owen Andersona1cf15f2009-07-14 23:10:40 +00001503 td.getTypeSizeInBits(ModuleTy)/8));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001504 //FIXME: Should be the path to the file where this module was declared
1505 Elements.push_back(NULLPtr);
1506 Elements.push_back(SymTab);
1507 llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
1508
1509 // Create the load function calling the runtime entry point with the module
1510 // structure
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001511 llvm::Function * LoadFunction = llvm::Function::Create(
Owen Anderson0032b272009-08-13 21:57:51 +00001512 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), false),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001513 llvm::GlobalValue::InternalLinkage, ".objc_load_function",
1514 &TheModule);
Owen Anderson0032b272009-08-13 21:57:51 +00001515 llvm::BasicBlock *EntryBB =
1516 llvm::BasicBlock::Create(VMContext, "entry", LoadFunction);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001517 CGBuilderTy Builder(VMContext);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001518 Builder.SetInsertPoint(EntryBB);
Fariborz Jahanian26c82942009-03-30 18:02:14 +00001519
1520 std::vector<const llvm::Type*> Params(1,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001521 llvm::PointerType::getUnqual(ModuleTy));
1522 llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Owen Anderson0032b272009-08-13 21:57:51 +00001523 llvm::Type::getVoidTy(VMContext), Params, true), "__objc_exec_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001524 Builder.CreateCall(Register, Module);
1525 Builder.CreateRetVoid();
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001526
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001527 return LoadFunction;
1528}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001529
Fariborz Jahanian679a5022009-01-10 21:06:09 +00001530llvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
Mike Stump1eb44332009-09-09 15:08:12 +00001531 const ObjCContainerDecl *CD) {
1532 const ObjCCategoryImplDecl *OCD =
Steve Naroff3e0a5402009-01-08 19:41:02 +00001533 dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
Chris Lattner077bf5e2008-11-24 03:33:13 +00001534 std::string CategoryName = OCD ? OCD->getNameAsString() : "";
1535 std::string ClassName = OMD->getClassInterface()->getNameAsString();
1536 std::string MethodName = OMD->getSelector().getAsString();
Douglas Gregorf8d49f62009-01-09 17:18:27 +00001537 bool isClassMethod = !OMD->isInstanceMethod();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001538
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001539 CodeGenTypes &Types = CGM.getTypes();
Mike Stump1eb44332009-09-09 15:08:12 +00001540 const llvm::FunctionType *MethodTy =
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001541 Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001542 std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
1543 MethodName, isClassMethod);
1544
Daniel Dunbard6c93d72009-09-17 04:01:22 +00001545 llvm::Function *Method
Mike Stump1eb44332009-09-09 15:08:12 +00001546 = llvm::Function::Create(MethodTy,
1547 llvm::GlobalValue::InternalLinkage,
1548 FunctionName,
1549 &TheModule);
Chris Lattner391d77a2008-03-30 23:03:07 +00001550 return Method;
1551}
1552
Daniel Dunbar49f66022008-09-24 03:38:44 +00001553llvm::Function *CGObjCGNU::GetPropertyGetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001554 std::vector<const llvm::Type*> Params;
1555 const llvm::Type *BoolTy =
1556 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1557 Params.push_back(IdTy);
1558 Params.push_back(SelectorTy);
1559 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1560 Params.push_back(LongTy);
1561 Params.push_back(BoolTy);
1562 // void objc_getProperty (id, SEL, ptrdiff_t, bool)
1563 const llvm::FunctionType *FTy =
1564 llvm::FunctionType::get(IdTy, Params, false);
1565 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1566 "objc_getProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001567}
1568
1569llvm::Function *CGObjCGNU::GetPropertySetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001570 std::vector<const llvm::Type*> Params;
1571 const llvm::Type *BoolTy =
1572 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1573 Params.push_back(IdTy);
1574 Params.push_back(SelectorTy);
1575 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1576 Params.push_back(LongTy);
1577 Params.push_back(IdTy);
1578 Params.push_back(BoolTy);
1579 Params.push_back(BoolTy);
1580 // void objc_setProperty (id, SEL, ptrdiff_t, id, bool, bool)
1581 const llvm::FunctionType *FTy =
1582 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Params, false);
1583 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1584 "objc_setProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001585}
1586
Daniel Dunbar309a4362009-07-24 07:40:24 +00001587llvm::Constant *CGObjCGNU::EnumerationMutationFunction() {
1588 CodeGen::CodeGenTypes &Types = CGM.getTypes();
1589 ASTContext &Ctx = CGM.getContext();
1590 // void objc_enumerationMutation (id)
1591 llvm::SmallVector<QualType,16> Params;
David Chisnall0f436562009-08-17 16:35:33 +00001592 Params.push_back(ASTIdTy);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001593 const llvm::FunctionType *FTy =
1594 Types.GetFunctionType(Types.getFunctionInfo(Ctx.VoidTy, Params), false);
1595 return CGM.CreateRuntimeFunction(FTy, "objc_enumerationMutation");
Anders Carlsson2abd89c2008-08-31 04:05:03 +00001596}
1597
Fariborz Jahanianbd71be42008-11-21 00:49:24 +00001598void CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1599 const Stmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001600 // Pointer to the personality function
1601 llvm::Constant *Personality =
Owen Anderson0032b272009-08-13 21:57:51 +00001602 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getInt32Ty(VMContext),
Chris Lattnerb59761b2009-07-01 04:13:52 +00001603 true),
Chris Lattner5dc08672009-05-08 00:11:50 +00001604 "__gnu_objc_personality_v0");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001605 Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
Chris Lattner5dc08672009-05-08 00:11:50 +00001606 std::vector<const llvm::Type*> Params;
1607 Params.push_back(PtrTy);
1608 llvm::Value *RethrowFn =
Owen Anderson0032b272009-08-13 21:57:51 +00001609 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
Chris Lattner5dc08672009-05-08 00:11:50 +00001610 Params, false), "_Unwind_Resume_or_Rethrow");
1611
1612 bool isTry = isa<ObjCAtTryStmt>(S);
1613 llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
1614 llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
1615 llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
Chris Lattnerd6a99072009-05-11 18:16:28 +00001616 llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
Chris Lattner5dc08672009-05-08 00:11:50 +00001617 llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
1618 llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
1619 llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
1620
1621 // GNU runtime does not currently support @synchronized()
1622 if (!isTry) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001623 std::vector<const llvm::Type*> Args(1, IdTy);
1624 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001625 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001626 llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
Mike Stump1eb44332009-09-09 15:08:12 +00001627 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001628 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1629 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1630 CGF.Builder.CreateCall(SyncEnter, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001631 }
1632
Chris Lattnerd6a99072009-05-11 18:16:28 +00001633
Chris Lattner5dc08672009-05-08 00:11:50 +00001634 // Push an EH context entry, used for handling rethrows and jumps
1635 // through finally.
1636 CGF.PushCleanupBlock(FinallyBlock);
1637
1638 // Emit the statements in the @try {} block
1639 CGF.setInvokeDest(TryHandler);
1640
1641 CGF.EmitBlock(TryBlock);
Mike Stump1eb44332009-09-09 15:08:12 +00001642 CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
Chris Lattner5dc08672009-05-08 00:11:50 +00001643 : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
1644
1645 // Jump to @finally if there is no exception
1646 CGF.EmitBranchThroughCleanup(FinallyEnd);
1647
1648 // Emit the handlers
1649 CGF.EmitBlock(TryHandler);
1650
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001651 // Get the correct versions of the exception handling intrinsics
1652 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
Mike Stump1eb44332009-09-09 15:08:12 +00001653 llvm::Value *llvm_eh_exception =
Chris Lattner5dc08672009-05-08 00:11:50 +00001654 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
Duncan Sands790b20e2009-10-14 16:13:30 +00001655 llvm::Value *llvm_eh_selector =
1656 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector);
1657 llvm::Value *llvm_eh_typeid_for =
1658 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for);
Chris Lattner5dc08672009-05-08 00:11:50 +00001659
1660 // Exception object
1661 llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1662 llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
1663
1664 llvm::SmallVector<llvm::Value*, 8> ESelArgs;
1665 llvm::SmallVector<std::pair<const ParmVarDecl*, const Stmt*>, 8> Handlers;
1666
1667 ESelArgs.push_back(Exc);
1668 ESelArgs.push_back(Personality);
1669
1670 bool HasCatchAll = false;
1671 // Only @try blocks are allowed @catch blocks, but both can have @finally
1672 if (isTry) {
1673 if (const ObjCAtCatchStmt* CatchStmt =
1674 cast<ObjCAtTryStmt>(S).getCatchStmts()) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001675 CGF.setInvokeDest(CatchInCatch);
Chris Lattner5dc08672009-05-08 00:11:50 +00001676
1677 for (; CatchStmt; CatchStmt = CatchStmt->getNextCatchStmt()) {
1678 const ParmVarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
Mike Stumpbb1c8602009-07-31 21:31:32 +00001679 Handlers.push_back(std::make_pair(CatchDecl,
1680 CatchStmt->getCatchBody()));
Chris Lattner5dc08672009-05-08 00:11:50 +00001681
1682 // @catch() and @catch(id) both catch any ObjC exception
Steve Naroff14108da2009-07-10 23:34:53 +00001683 if (!CatchDecl || CatchDecl->getType()->isObjCIdType()
Chris Lattner5dc08672009-05-08 00:11:50 +00001684 || CatchDecl->getType()->isObjCQualifiedIdType()) {
1685 // Use i8* null here to signal this is a catch all, not a cleanup.
1686 ESelArgs.push_back(NULLPtr);
1687 HasCatchAll = true;
1688 // No further catches after this one will ever by reached
1689 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001690 }
Chris Lattner5dc08672009-05-08 00:11:50 +00001691
1692 // All other types should be Objective-C interface pointer types.
Mike Stump1eb44332009-09-09 15:08:12 +00001693 const ObjCObjectPointerType *OPT =
John McCall183700f2009-09-21 23:43:11 +00001694 CatchDecl->getType()->getAs<ObjCObjectPointerType>();
Steve Naroff14108da2009-07-10 23:34:53 +00001695 assert(OPT && "Invalid @catch type.");
Mike Stump1eb44332009-09-09 15:08:12 +00001696 const ObjCInterfaceType *IT =
John McCall183700f2009-09-21 23:43:11 +00001697 OPT->getPointeeType()->getAs<ObjCInterfaceType>();
Chris Lattner5dc08672009-05-08 00:11:50 +00001698 assert(IT && "Invalid @catch type.");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001699 llvm::Value *EHType =
1700 MakeConstantString(IT->getDecl()->getNameAsString());
Chris Lattner5dc08672009-05-08 00:11:50 +00001701 ESelArgs.push_back(EHType);
1702 }
1703 }
1704 }
1705
1706 // We use a cleanup unless there was already a catch all.
1707 if (!HasCatchAll) {
Owen Anderson0032b272009-08-13 21:57:51 +00001708 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattner5dc08672009-05-08 00:11:50 +00001709 Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
1710 }
1711
1712 // Find which handler was matched.
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001713 llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
Chris Lattner5dc08672009-05-08 00:11:50 +00001714 ESelArgs.begin(), ESelArgs.end(), "selector");
1715
1716 for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
1717 const ParmVarDecl *CatchParam = Handlers[i].first;
1718 const Stmt *CatchBody = Handlers[i].second;
1719
1720 llvm::BasicBlock *Next = 0;
1721
1722 // The last handler always matches.
1723 if (i + 1 != e) {
1724 assert(CatchParam && "Only last handler can be a catch all.");
1725
1726 // Test whether this block matches the type for the selector and branch
1727 // to Match if it does, or to the next BB if it doesn't.
1728 llvm::BasicBlock *Match = CGF.createBasicBlock("match");
1729 Next = CGF.createBasicBlock("catch.next");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001730 llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
Chris Lattner5dc08672009-05-08 00:11:50 +00001731 CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
1732 CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
1733 Next);
1734
1735 CGF.EmitBlock(Match);
1736 }
Mike Stump1eb44332009-09-09 15:08:12 +00001737
Chris Lattner5dc08672009-05-08 00:11:50 +00001738 if (CatchBody) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001739 llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
1740 CGF.ConvertType(CatchParam->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001741
Chris Lattner5dc08672009-05-08 00:11:50 +00001742 // Bind the catch parameter if it exists.
1743 if (CatchParam) {
1744 // CatchParam is a ParmVarDecl because of the grammar
1745 // construction used to handle this, but for codegen purposes
1746 // we treat this as a local decl.
1747 CGF.EmitLocalBlockVarDecl(*CatchParam);
1748 CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
1749 }
1750
1751 CGF.ObjCEHValueStack.push_back(ExcObject);
1752 CGF.EmitStmt(CatchBody);
1753 CGF.ObjCEHValueStack.pop_back();
1754
1755 CGF.EmitBranchThroughCleanup(FinallyEnd);
1756
1757 if (Next)
1758 CGF.EmitBlock(Next);
1759 } else {
1760 assert(!Next && "catchup should be last handler.");
1761
1762 CGF.Builder.CreateStore(Exc, RethrowPtr);
1763 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1764 }
1765 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001766 // The @finally block is a secondary landing pad for any exceptions thrown in
1767 // @catch() blocks
1768 CGF.EmitBlock(CatchInCatch);
1769 Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1770 ESelArgs.clear();
1771 ESelArgs.push_back(Exc);
1772 ESelArgs.push_back(Personality);
Owen Anderson0032b272009-08-13 21:57:51 +00001773 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattnerd6a99072009-05-11 18:16:28 +00001774 CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
1775 "selector");
1776 CGF.Builder.CreateCall(llvm_eh_typeid_for,
1777 CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
1778 CGF.Builder.CreateStore(Exc, RethrowPtr);
1779 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1780
Chris Lattner5dc08672009-05-08 00:11:50 +00001781 CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
1782
1783 CGF.setInvokeDest(PrevLandingPad);
1784
1785 CGF.EmitBlock(FinallyBlock);
1786
Chris Lattnerd6a99072009-05-11 18:16:28 +00001787
Chris Lattner5dc08672009-05-08 00:11:50 +00001788 if (isTry) {
Mike Stump1eb44332009-09-09 15:08:12 +00001789 if (const ObjCAtFinallyStmt* FinallyStmt =
Chris Lattner5dc08672009-05-08 00:11:50 +00001790 cast<ObjCAtTryStmt>(S).getFinallyStmt())
1791 CGF.EmitStmt(FinallyStmt->getFinallyBody());
1792 } else {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001793 // Emit 'objc_sync_exit(expr)' as finally's sole statement for
Chris Lattner5dc08672009-05-08 00:11:50 +00001794 // @synchronized.
Chris Lattnerd6a99072009-05-11 18:16:28 +00001795 std::vector<const llvm::Type*> Args(1, IdTy);
1796 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001797 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001798 llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
Mike Stump1eb44332009-09-09 15:08:12 +00001799 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001800 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1801 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1802 CGF.Builder.CreateCall(SyncExit, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001803 }
1804
1805 if (Info.SwitchBlock)
1806 CGF.EmitBlock(Info.SwitchBlock);
1807 if (Info.EndBlock)
1808 CGF.EmitBlock(Info.EndBlock);
1809
1810 // Branch around the rethrow code.
1811 CGF.EmitBranch(FinallyEnd);
1812
1813 CGF.EmitBlock(FinallyRethrow);
1814 CGF.Builder.CreateCall(RethrowFn, CGF.Builder.CreateLoad(RethrowPtr));
1815 CGF.Builder.CreateUnreachable();
Mike Stump1eb44332009-09-09 15:08:12 +00001816
Chris Lattner5dc08672009-05-08 00:11:50 +00001817 CGF.EmitBlock(FinallyEnd);
1818
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001819}
1820
1821void CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar49f66022008-09-24 03:38:44 +00001822 const ObjCAtThrowStmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001823 llvm::Value *ExceptionAsObject;
1824
1825 std::vector<const llvm::Type*> Args(1, IdTy);
1826 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001827 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Mike Stump1eb44332009-09-09 15:08:12 +00001828 llvm::Value *ThrowFn =
Chris Lattner5dc08672009-05-08 00:11:50 +00001829 CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
Mike Stump1eb44332009-09-09 15:08:12 +00001830
Chris Lattner5dc08672009-05-08 00:11:50 +00001831 if (const Expr *ThrowExpr = S.getThrowExpr()) {
1832 llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001833 ExceptionAsObject = Exception;
Chris Lattner5dc08672009-05-08 00:11:50 +00001834 } else {
Mike Stump1eb44332009-09-09 15:08:12 +00001835 assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
Chris Lattner5dc08672009-05-08 00:11:50 +00001836 "Unexpected rethrow outside @catch block.");
1837 ExceptionAsObject = CGF.ObjCEHValueStack.back();
1838 }
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001839 ExceptionAsObject =
1840 CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
Mike Stump1eb44332009-09-09 15:08:12 +00001841
Chris Lattner5dc08672009-05-08 00:11:50 +00001842 // Note: This may have to be an invoke, if we want to support constructs like:
1843 // @try {
1844 // @throw(obj);
1845 // }
1846 // @catch(id) ...
1847 //
1848 // This is effectively turning @throw into an incredibly-expensive goto, but
1849 // it may happen as a result of inlining followed by missed optimizations, or
1850 // as a result of stupidity.
1851 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
1852 if (!UnwindBB) {
1853 CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
1854 CGF.Builder.CreateUnreachable();
1855 } else {
1856 CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
1857 &ExceptionAsObject+1);
1858 }
1859 // Clear the insertion point to indicate we are in unreachable code.
1860 CGF.Builder.ClearInsertionPoint();
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001861}
1862
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001863llvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001864 llvm::Value *AddrWeakObj) {
Fariborz Jahanian6dc23172008-11-18 21:45:40 +00001865 return 0;
1866}
1867
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001868void CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001869 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001870 return;
1871}
1872
Fariborz Jahanian58626502008-11-19 00:59:10 +00001873void CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001874 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian58626502008-11-19 00:59:10 +00001875 return;
1876}
1877
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00001878void CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6c7a1f32009-09-24 22:25:38 +00001879 llvm::Value *src, llvm::Value *dst,
1880 llvm::Value *ivarOffset) {
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00001881 return;
1882}
1883
Fariborz Jahanian58626502008-11-19 00:59:10 +00001884void CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001885 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian58626502008-11-19 00:59:10 +00001886 return;
1887}
1888
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001889void CGObjCGNU::EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001890 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001891 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +00001892 QualType Ty) {
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001893 return;
1894}
1895
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001896llvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
1897 const ObjCInterfaceDecl *ID,
1898 const ObjCIvarDecl *Ivar) {
1899 const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
1900 + '.' + Ivar->getNameAsString();
1901 // Emit the variable and initialize it with what we think the correct value
1902 // is. This allows code compiled with non-fragile ivars to work correctly
1903 // when linked against code which isn't (most of the time).
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001904 llvm::GlobalVariable *IvarOffsetPointer = TheModule.getNamedGlobal(Name);
1905 if (!IvarOffsetPointer) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001906 uint64_t Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
1907 llvm::ConstantInt *OffsetGuess =
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001908 llvm::ConstantInt::get(LongTy, Offset, "ivar");
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001909 // Don't emit the guess in non-PIC code because the linker will not be able
1910 // to replace it with the real version for a library. In non-PIC code you
1911 // must compile with the fragile ABI if you want to use ivars from a
Mike Stump1eb44332009-09-09 15:08:12 +00001912 // GCC-compiled class.
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001913 if (CGM.getLangOptions().PICLevel) {
1914 llvm::GlobalVariable *IvarOffsetGV = new llvm::GlobalVariable(TheModule,
1915 llvm::Type::getInt32Ty(VMContext), false,
1916 llvm::GlobalValue::PrivateLinkage, OffsetGuess, Name+".guess");
1917 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
1918 IvarOffsetGV->getType(), false, llvm::GlobalValue::LinkOnceAnyLinkage,
1919 IvarOffsetGV, Name);
1920 } else {
1921 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
Benjamin Kramer3c0ef8c2009-10-13 10:07:13 +00001922 llvm::Type::getInt32PtrTy(VMContext), false,
1923 llvm::GlobalValue::ExternalLinkage, 0, Name);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001924 }
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001925 }
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001926 return IvarOffsetPointer;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001927}
1928
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00001929LValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
1930 QualType ObjectTy,
1931 llvm::Value *BaseValue,
1932 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00001933 unsigned CVRQualifiers) {
John McCall183700f2009-09-21 23:43:11 +00001934 const ObjCInterfaceDecl *ID = ObjectTy->getAs<ObjCInterfaceType>()->getDecl();
Daniel Dunbar97776872009-04-22 07:32:20 +00001935 return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
1936 EmitIvarOffset(CGF, ID, Ivar));
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00001937}
Mike Stumpbb1c8602009-07-31 21:31:32 +00001938
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001939static const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
1940 const ObjCInterfaceDecl *OID,
1941 const ObjCIvarDecl *OIVD) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001942 llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
Fariborz Jahanian8e6ac1d2009-06-04 01:19:09 +00001943 Context.ShallowCollectObjCIvars(OID, Ivars);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001944 for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
1945 if (OIVD == Ivars[k])
1946 return OID;
1947 }
Mike Stump1eb44332009-09-09 15:08:12 +00001948
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001949 // Otherwise check in the super class.
1950 if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
1951 return FindIvarInterface(Context, Super, OIVD);
Mike Stump1eb44332009-09-09 15:08:12 +00001952
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001953 return 0;
1954}
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00001955
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001956llvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +00001957 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001958 const ObjCIvarDecl *Ivar) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001959 if (CGM.getLangOptions().ObjCNonFragileABI) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001960 Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001961 return CGF.Builder.CreateLoad(CGF.Builder.CreateLoad(
1962 ObjCIvarOffsetVariable(Interface, Ivar), false, "ivar"));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001963 }
Daniel Dunbar97776872009-04-22 07:32:20 +00001964 uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001965 return llvm::ConstantInt::get(LongTy, Offset, "ivar");
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001966}
1967
Mike Stumpbb1c8602009-07-31 21:31:32 +00001968CodeGen::CGObjCRuntime *
1969CodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM) {
Chris Lattnerdce14062008-06-26 04:19:03 +00001970 return new CGObjCGNU(CGM);
Chris Lattner0f984262008-03-01 08:50:34 +00001971}