blob: 264e9d0215bf79ae954ddd80eb6301c8a4233187 [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;
Chris Lattner0f984262008-03-01 08:50:34 +000046
Chris Lattner0f984262008-03-01 08:50:34 +000047namespace {
Chris Lattnerdce14062008-06-26 04:19:03 +000048class CGObjCGNU : public CodeGen::CGObjCRuntime {
Chris Lattner0f984262008-03-01 08:50:34 +000049private:
Chris Lattnerdce14062008-06-26 04:19:03 +000050 CodeGen::CodeGenModule &CGM;
Chris Lattner0f984262008-03-01 08:50:34 +000051 llvm::Module &TheModule;
Chris Lattnere160c9b2009-01-27 05:06:01 +000052 const llvm::PointerType *SelectorTy;
53 const llvm::PointerType *PtrToInt8Ty;
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +000054 const llvm::FunctionType *IMPTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000055 const llvm::PointerType *IdTy;
David Chisnall0f436562009-08-17 16:35:33 +000056 QualType ASTIdTy;
Chris Lattnere160c9b2009-01-27 05:06:01 +000057 const llvm::IntegerType *IntTy;
58 const llvm::PointerType *PtrTy;
59 const llvm::IntegerType *LongTy;
60 const llvm::PointerType *PtrToIntTy;
Daniel Dunbar5efccb12009-05-04 15:31:17 +000061 llvm::GlobalAlias *ClassPtrAlias;
62 llvm::GlobalAlias *MetaClassPtrAlias;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000063 std::vector<llvm::Constant*> Classes;
64 std::vector<llvm::Constant*> Categories;
65 std::vector<llvm::Constant*> ConstantStrings;
66 llvm::Function *LoadFunction;
67 llvm::StringMap<llvm::Constant*> ExistingProtocols;
68 typedef std::pair<std::string, std::string> TypedSelector;
69 std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
70 llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
71 // Some zeros used for GEPs in lots of places.
72 llvm::Constant *Zeros[2];
73 llvm::Constant *NULLPtr;
Owen Andersona1cf15f2009-07-14 23:10:40 +000074 llvm::LLVMContext &VMContext;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000075private:
76 llvm::Constant *GenerateIvarList(
77 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
78 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
79 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets);
80 llvm::Constant *GenerateMethodList(const std::string &ClassName,
81 const std::string &CategoryName,
Mike Stump1eb44332009-09-09 15:08:12 +000082 const llvm::SmallVectorImpl<Selector> &MethodSels,
83 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000084 bool isClassMethodList);
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +000085 llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000086 llvm::Constant *GenerateProtocolList(
87 const llvm::SmallVectorImpl<std::string> &Protocols);
88 llvm::Constant *GenerateClassStructure(
89 llvm::Constant *MetaClass,
90 llvm::Constant *SuperClass,
91 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +000092 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +000093 llvm::Constant *Version,
94 llvm::Constant *InstanceSize,
95 llvm::Constant *IVars,
96 llvm::Constant *Methods,
97 llvm::Constant *Protocols);
98 llvm::Constant *GenerateProtocolMethodList(
99 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
100 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes);
101 llvm::Constant *MakeConstantString(const std::string &Str, const std::string
102 &Name="");
103 llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
104 std::vector<llvm::Constant*> &V, const std::string &Name="");
105 llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
106 std::vector<llvm::Constant*> &V, const std::string &Name="");
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000107 llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
108 const ObjCIvarDecl *Ivar);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000109 void EmitClassRef(const std::string &className);
Chris Lattner0f984262008-03-01 08:50:34 +0000110public:
Chris Lattnerdce14062008-06-26 04:19:03 +0000111 CGObjCGNU(CodeGen::CodeGenModule &cgm);
Steve Naroff33fdb732009-03-31 16:53:37 +0000112 virtual llvm::Constant *GenerateConstantString(const ObjCStringLiteral *);
Mike Stump1eb44332009-09-09 15:08:12 +0000113 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000114 GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000115 QualType ResultType,
116 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000117 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000118 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000119 const CallArgList &CallArgs,
120 const ObjCMethodDecl *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000121 virtual CodeGen::RValue
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000122 GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000123 QualType ResultType,
124 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000125 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000126 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000127 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000128 bool IsClassMessage,
129 const CallArgList &CallArgs);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000130 virtual llvm::Value *GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000131 const ObjCInterfaceDecl *OID);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000132 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000133 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
134 *Method);
Mike Stump1eb44332009-09-09 15:08:12 +0000135
136 virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
Fariborz Jahanian679a5022009-01-10 21:06:09 +0000137 const ObjCContainerDecl *CD);
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000138 virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
139 virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000140 virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000141 const ObjCProtocolDecl *PD);
142 virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000143 virtual llvm::Function *ModuleInitFunction();
Daniel Dunbar49f66022008-09-24 03:38:44 +0000144 virtual llvm::Function *GetPropertyGetFunction();
145 virtual llvm::Function *GetPropertySetFunction();
Daniel Dunbar309a4362009-07-24 07:40:24 +0000146 virtual llvm::Constant *EnumerationMutationFunction();
Mike Stump1eb44332009-09-09 15:08:12 +0000147
Fariborz Jahanianbd71be42008-11-21 00:49:24 +0000148 virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
149 const Stmt &S);
Anders Carlsson64d5d6c2008-09-09 10:04:29 +0000150 virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
151 const ObjCAtThrowStmt &S);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000152 virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian6dc23172008-11-18 21:45:40 +0000153 llvm::Value *AddrWeakObj);
Fariborz Jahanian3e283e32008-11-18 22:37:34 +0000154 virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
155 llvm::Value *src, llvm::Value *dst);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000156 virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
157 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian7eda8362008-11-20 19:23:36 +0000158 virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
159 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian58626502008-11-19 00:59:10 +0000160 virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
161 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000162 virtual void EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +0000163 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +0000164 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +0000165 QualType Ty);
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000166 virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
167 QualType ObjectTy,
168 llvm::Value *BaseValue,
169 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +0000170 unsigned CVRQualifiers);
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000171 virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +0000172 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +0000173 const ObjCIvarDecl *Ivar);
Chris Lattner0f984262008-03-01 08:50:34 +0000174};
175} // end anonymous namespace
176
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000177
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000178/// Emits a reference to a dummy variable which is emitted with each class.
179/// This ensures that a linker error will be generated when trying to link
180/// together modules where a referenced class is not defined.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000181void CGObjCGNU::EmitClassRef(const std::string &className) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000182 std::string symbolRef = "__objc_class_ref_" + className;
183 // Don't emit two copies of the same symbol
Mike Stumpbb1c8602009-07-31 21:31:32 +0000184 if (TheModule.getGlobalVariable(symbolRef))
185 return;
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000186 std::string symbolName = "__objc_class_name_" + className;
187 llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
188 if (!ClassSymbol) {
Owen Anderson1c431b32009-07-08 19:05:04 +0000189 ClassSymbol = new llvm::GlobalVariable(TheModule, LongTy, false,
190 llvm::GlobalValue::ExternalLinkage, 0, symbolName);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000191 }
Owen Anderson1c431b32009-07-08 19:05:04 +0000192 new llvm::GlobalVariable(TheModule, ClassSymbol->getType(), true,
Chris Lattnerf35271b2009-08-05 05:25:18 +0000193 llvm::GlobalValue::WeakAnyLinkage, ClassSymbol, symbolRef);
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000194}
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000195
196static std::string SymbolNameForClass(const std::string &ClassName) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000197 return "_OBJC_CLASS_" + ClassName;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000198}
199
Mike Stump1eb44332009-09-09 15:08:12 +0000200static std::string SymbolNameForMethod(const std::string &ClassName,
201 const std::string &CategoryName,
202 const std::string &MethodName,
203 bool isClassMethod) {
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000204 return "_OBJC_METHOD_" + ClassName + "("+CategoryName+")"+
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000205 (isClassMethod ? "+" : "-") + MethodName;
206}
207
Chris Lattnerdce14062008-06-26 04:19:03 +0000208CGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000209 : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
Owen Andersona1cf15f2009-07-14 23:10:40 +0000210 MetaClassPtrAlias(0), VMContext(cgm.getLLVMContext()) {
Chris Lattnere160c9b2009-01-27 05:06:01 +0000211 IntTy = cast<llvm::IntegerType>(
212 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
213 LongTy = cast<llvm::IntegerType>(
214 CGM.getTypes().ConvertType(CGM.getContext().LongTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000215
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000216 Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000217 Zeros[1] = Zeros[0];
Owen Anderson03e20502009-07-30 23:11:26 +0000218 NULLPtr = llvm::ConstantPointerNull::get(
Owen Anderson0032b272009-08-13 21:57:51 +0000219 llvm::PointerType::getUnqual(llvm::Type::getInt8Ty(VMContext)));
Chris Lattner391d77a2008-03-30 23:03:07 +0000220 // C string type. Used in lots of places.
Mike Stump1eb44332009-09-09 15:08:12 +0000221 PtrToInt8Ty =
Owen Anderson0032b272009-08-13 21:57:51 +0000222 llvm::PointerType::getUnqual(llvm::Type::getInt8Ty(VMContext));
Chris Lattner391d77a2008-03-30 23:03:07 +0000223 // Get the selector Type.
Chris Lattnere160c9b2009-01-27 05:06:01 +0000224 SelectorTy = cast<llvm::PointerType>(
225 CGM.getTypes().ConvertType(CGM.getContext().getObjCSelType()));
226
Owen Anderson96e0fc72009-07-29 22:16:19 +0000227 PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
Chris Lattner391d77a2008-03-30 23:03:07 +0000228 PtrTy = PtrToInt8Ty;
Mike Stump1eb44332009-09-09 15:08:12 +0000229
Chris Lattner391d77a2008-03-30 23:03:07 +0000230 // Object type
David Chisnall0f436562009-08-17 16:35:33 +0000231 ASTIdTy = CGM.getContext().getObjCIdType();
232 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Mike Stump1eb44332009-09-09 15:08:12 +0000233
Chris Lattner391d77a2008-03-30 23:03:07 +0000234 // IMP type
235 std::vector<const llvm::Type*> IMPArgs;
236 IMPArgs.push_back(IdTy);
237 IMPArgs.push_back(SelectorTy);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000238 IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000239}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000240
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000241// This has to perform the lookup every time, since posing and related
242// techniques can modify the name -> class mapping.
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000243llvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000244 const ObjCInterfaceDecl *OID) {
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000245 llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000246 EmitClassRef(OID->getNameAsString());
Daniel Dunbarddb2a3d2008-08-16 00:25:02 +0000247 ClassName = Builder.CreateStructGEP(ClassName, 0);
248
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000249 std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000250 llvm::Constant *ClassLookupFn =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000251 CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000252 Params,
253 true),
254 "objc_lookup_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000255 return Builder.CreateCall(ClassLookupFn, ClassName);
Chris Lattner391d77a2008-03-30 23:03:07 +0000256}
257
Daniel Dunbar45d196b2008-11-01 01:53:16 +0000258llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel) {
Chris Lattner077bf5e2008-11-24 03:33:13 +0000259 llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
Chris Lattner8e67b632008-06-26 04:37:12 +0000260 if (US == 0)
Owen Anderson96e0fc72009-07-29 22:16:19 +0000261 US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000262 llvm::GlobalValue::PrivateLinkage,
263 ".objc_untyped_selector_alias"+Sel.getAsString(),
Chris Lattner8e67b632008-06-26 04:37:12 +0000264 NULL, &TheModule);
Mike Stump1eb44332009-09-09 15:08:12 +0000265
Chris Lattner8e67b632008-06-26 04:37:12 +0000266 return Builder.CreateLoad(US);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000267}
268
269llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
270 *Method) {
271
272 std::string SelName = Method->getSelector().getAsString();
273 std::string SelTypes;
274 CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
275 // Typed selectors
276 TypedSelector Selector = TypedSelector(SelName,
277 SelTypes);
278
279 // If it's already cached, return it.
Mike Stumpbb1c8602009-07-31 21:31:32 +0000280 if (TypedSelectors[Selector]) {
281 return Builder.CreateLoad(TypedSelectors[Selector]);
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000282 }
283
284 // If it isn't, cache it.
285 llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000286 llvm::PointerType::getUnqual(SelectorTy),
David Chisnall0f436562009-08-17 16:35:33 +0000287 llvm::GlobalValue::PrivateLinkage, ".objc_selector_alias" + SelName,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000288 NULL, &TheModule);
289 TypedSelectors[Selector] = Sel;
290
291 return Builder.CreateLoad(Sel);
Chris Lattner8e67b632008-06-26 04:37:12 +0000292}
293
Chris Lattner5e7dcc62008-06-26 04:44:19 +0000294llvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
295 const std::string &Name) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000296 llvm::Constant *ConstStr = CGM.GetAddrOfConstantCString(Str, Name.c_str());
Owen Anderson3c4972d2009-07-29 18:54:39 +0000297 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000298}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000299
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000300llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
301 std::vector<llvm::Constant*> &V, const std::string &Name) {
Owen Anderson08e25242009-07-27 22:29:56 +0000302 llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000303 return new llvm::GlobalVariable(TheModule, Ty, false,
304 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000305}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000306
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000307llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
308 std::vector<llvm::Constant*> &V, const std::string &Name) {
Owen Anderson7db6d832009-07-28 18:33:04 +0000309 llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
Owen Anderson1c431b32009-07-08 19:05:04 +0000310 return new llvm::GlobalVariable(TheModule, Ty, false,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000311 llvm::GlobalValue::InternalLinkage, C, Name);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000312}
313
314/// Generate an NSConstantString object.
315//TODO: In case there are any crazy people still using the GNU runtime without
316//an OpenStep implementation, this should let them select their own class for
317//constant strings.
Steve Naroff33fdb732009-03-31 16:53:37 +0000318llvm::Constant *CGObjCGNU::GenerateConstantString(const ObjCStringLiteral *SL) {
Mike Stump1eb44332009-09-09 15:08:12 +0000319 std::string Str(SL->getString()->getStrData(),
Steve Naroff33fdb732009-03-31 16:53:37 +0000320 SL->getString()->getByteLength());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000321 std::vector<llvm::Constant*> Ivars;
322 Ivars.push_back(NULLPtr);
Chris Lattner13fd7e52008-06-21 21:44:18 +0000323 Ivars.push_back(MakeConstantString(Str));
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000324 Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000325 llvm::Constant *ObjCStr = MakeGlobal(
Owen Anderson47a434f2009-08-05 23:18:46 +0000326 llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000327 Ivars, ".objc_str");
328 ConstantStrings.push_back(
Owen Anderson3c4972d2009-07-29 18:54:39 +0000329 llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000330 return ObjCStr;
331}
332
333///Generates a message send where the super is the receiver. This is a message
334///send to self with special delivery semantics indicating which class's method
335///should be called.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000336CodeGen::RValue
337CGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000338 QualType ResultType,
339 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000340 const ObjCInterfaceDecl *Class,
Fariborz Jahanian7ce77922009-02-28 20:07:56 +0000341 bool isCategoryImpl,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000342 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000343 bool IsClassMessage,
344 const CallArgList &CallArgs) {
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000345 llvm::Value *cmd = GetSelector(CGF.Builder, Sel);
346
347 CallArgList ActualArgs;
348
349 ActualArgs.push_back(
David Chisnall0f436562009-08-17 16:35:33 +0000350 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
351 ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000352 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
353 CGF.getContext().getObjCSelType()));
354 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
355
356 CodeGenTypes &Types = CGM.getTypes();
357 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
358 const llvm::FunctionType *impType = Types.GetFunctionType(FnInfo, false);
359
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000360 llvm::Value *ReceiverClass = 0;
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000361 if (isCategoryImpl) {
362 llvm::Constant *classLookupFunction = 0;
363 std::vector<const llvm::Type*> Params;
364 Params.push_back(PtrTy);
365 if (IsClassMessage) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000366 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000367 IdTy, Params, true), "objc_get_meta_class");
368 } else {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000369 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000370 IdTy, Params, true), "objc_get_class");
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000371 }
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000372 ReceiverClass = CGF.Builder.CreateCall(classLookupFunction,
373 MakeConstantString(Class->getNameAsString()));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000374 } else {
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000375 // Set up global aliases for the metaclass or class pointer if they do not
376 // already exist. These will are forward-references which will be set to
Mike Stumpbb1c8602009-07-31 21:31:32 +0000377 // pointers to the class and metaclass structure created for the runtime
378 // load function. To send a message to super, we look up the value of the
Chris Lattner48e6e7e2009-05-08 15:39:58 +0000379 // super_class pointer from either the class or metaclass structure.
380 if (IsClassMessage) {
381 if (!MetaClassPtrAlias) {
382 MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
383 llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
384 Class->getNameAsString(), NULL, &TheModule);
385 }
386 ReceiverClass = MetaClassPtrAlias;
387 } else {
388 if (!ClassPtrAlias) {
389 ClassPtrAlias = new llvm::GlobalAlias(IdTy,
390 llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
391 Class->getNameAsString(), NULL, &TheModule);
392 }
393 ReceiverClass = ClassPtrAlias;
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000394 }
Chris Lattner71238f62009-04-25 23:19:45 +0000395 }
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000396 // Cast the pointer to a simplified version of the class structure
Mike Stump1eb44332009-09-09 15:08:12 +0000397 ReceiverClass = CGF.Builder.CreateBitCast(ReceiverClass,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000398 llvm::PointerType::getUnqual(
Owen Anderson47a434f2009-08-05 23:18:46 +0000399 llvm::StructType::get(VMContext, IdTy, IdTy, NULL)));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000400 // Get the superclass pointer
401 ReceiverClass = CGF.Builder.CreateStructGEP(ReceiverClass, 1);
402 // Load the superclass pointer
403 ReceiverClass = CGF.Builder.CreateLoad(ReceiverClass);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000404 // Construct the structure used to look up the IMP
Owen Anderson47a434f2009-08-05 23:18:46 +0000405 llvm::StructType *ObjCSuperTy = llvm::StructType::get(VMContext,
406 Receiver->getType(), IdTy, NULL);
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000407 llvm::Value *ObjCSuper = CGF.Builder.CreateAlloca(ObjCSuperTy);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000408
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000409 CGF.Builder.CreateStore(Receiver, CGF.Builder.CreateStructGEP(ObjCSuper, 0));
Daniel Dunbar5efccb12009-05-04 15:31:17 +0000410 CGF.Builder.CreateStore(ReceiverClass,
411 CGF.Builder.CreateStructGEP(ObjCSuper, 1));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000412
413 // Get the IMP
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000414 std::vector<const llvm::Type*> Params;
Owen Anderson96e0fc72009-07-29 22:16:19 +0000415 Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000416 Params.push_back(SelectorTy);
Mike Stump1eb44332009-09-09 15:08:12 +0000417 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000418 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
419 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000420 "objc_msg_lookup_super");
421
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000422 llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000423 llvm::Value *imp = CGF.Builder.CreateCall(lookupFunction, lookupArgs,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000424 lookupArgs+2);
425
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000426 return CGF.EmitCall(FnInfo, imp, ActualArgs);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000427}
428
Mike Stump1eb44332009-09-09 15:08:12 +0000429/// Generate code for a message send expression.
Daniel Dunbar8f2926b2008-08-23 03:46:30 +0000430CodeGen::RValue
431CGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar7f8ea5c2008-08-30 05:35:15 +0000432 QualType ResultType,
433 Selector Sel,
Daniel Dunbarf56f1912008-08-25 08:19:24 +0000434 llvm::Value *Receiver,
Daniel Dunbar19cd87e2008-08-30 03:02:31 +0000435 bool IsClassMessage,
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000436 const CallArgList &CallArgs,
437 const ObjCMethodDecl *Method) {
David Chisnall0f436562009-08-17 16:35:33 +0000438 IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
Fariborz Jahaniandf9ccc62009-05-05 21:36:57 +0000439 llvm::Value *cmd;
440 if (Method)
441 cmd = GetSelector(CGF.Builder, Method);
442 else
443 cmd = GetSelector(CGF.Builder, Sel);
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000444 CallArgList ActualArgs;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000445
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000446 ActualArgs.push_back(
David Chisnall0f436562009-08-17 16:35:33 +0000447 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)), ASTIdTy));
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000448 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
449 CGF.getContext().getObjCSelType()));
450 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
451
452 CodeGenTypes &Types = CGM.getTypes();
453 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
454 const llvm::FunctionType *impType = Types.GetFunctionType(FnInfo, false);
455
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000456 llvm::Value *imp;
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000457 std::vector<const llvm::Type*> Params;
458 Params.push_back(Receiver->getType());
459 Params.push_back(SelectorTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000460 // For sender-aware dispatch, we pass the sender as the third argument to a
461 // lookup function. When sending messages from C code, the sender is nil.
462 // objc_msg_lookup_sender(id receiver, SEL selector, id sender);
463 if (CGM.getContext().getLangOptions().ObjCSenderDispatch) {
464 llvm::Value *self;
Fariborz Jahanian26c82942009-03-30 18:02:14 +0000465
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000466 if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
467 self = CGF.LoadObjCSelf();
468 } else {
Owen Anderson03e20502009-07-30 23:11:26 +0000469 self = llvm::ConstantPointerNull::get(IdTy);
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000470 }
471 Params.push_back(self->getType());
Mike Stump1eb44332009-09-09 15:08:12 +0000472 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000473 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
474 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000475 "objc_msg_lookup_sender");
476
477 imp = CGF.Builder.CreateCall3(lookupFunction, Receiver, cmd, self);
478 } else {
Mike Stump1eb44332009-09-09 15:08:12 +0000479 llvm::Constant *lookupFunction =
Owen Anderson96e0fc72009-07-29 22:16:19 +0000480 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
481 llvm::PointerType::getUnqual(impType), Params, true),
Fariborz Jahanian34e65772009-05-22 20:17:16 +0000482 "objc_msg_lookup");
483
484 imp = CGF.Builder.CreateCall2(lookupFunction, Receiver, cmd);
485 }
Chris Lattner3eae03e2008-05-06 00:56:42 +0000486
Fariborz Jahanianb3716ef2009-02-04 20:31:19 +0000487 return CGF.EmitCall(FnInfo, imp, ActualArgs);
Chris Lattner0f984262008-03-01 08:50:34 +0000488}
489
Mike Stump1eb44332009-09-09 15:08:12 +0000490/// Generates a MethodList. Used in construction of a objc_class and
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000491/// objc_category structures.
492llvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
Mike Stump1eb44332009-09-09 15:08:12 +0000493 const std::string &CategoryName,
494 const llvm::SmallVectorImpl<Selector> &MethodSels,
495 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000496 bool isClassMethodList) {
David Chisnall0f436562009-08-17 16:35:33 +0000497 if (MethodSels.empty())
498 return NULLPtr;
Mike Stump1eb44332009-09-09 15:08:12 +0000499 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000500 llvm::StructType *ObjCMethodTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000501 PtrToInt8Ty, // Really a selector, but the runtime creates it us.
502 PtrToInt8Ty, // Method types
Owen Anderson96e0fc72009-07-29 22:16:19 +0000503 llvm::PointerType::getUnqual(IMPTy), //Method pointer
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000504 NULL);
505 std::vector<llvm::Constant*> Methods;
506 std::vector<llvm::Constant*> Elements;
507 for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
508 Elements.clear();
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000509 if (llvm::Constant *Method =
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000510 TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
Chris Lattner077bf5e2008-11-24 03:33:13 +0000511 MethodSels[i].getAsString(),
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000512 isClassMethodList))) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000513 llvm::Constant *C = MakeConstantString(MethodSels[i].getAsString());
514 Elements.push_back(C);
515 Elements.push_back(MethodTypes[i]);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000516 Method = llvm::ConstantExpr::getBitCast(Method,
Owen Anderson96e0fc72009-07-29 22:16:19 +0000517 llvm::PointerType::getUnqual(IMPTy));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000518 Elements.push_back(Method);
Owen Anderson08e25242009-07-27 22:29:56 +0000519 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000520 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000521 }
522
523 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000524 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000525 Methods.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000526 llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
Chris Lattnerfba67632008-06-26 04:52:29 +0000527 Methods);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000528
529 // Structure containing list pointer, array and array count
530 llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
Owen Anderson8c8f69e2009-08-13 23:27:53 +0000531 llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get(VMContext);
Owen Anderson96e0fc72009-07-29 22:16:19 +0000532 llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
Owen Anderson47a434f2009-08-05 23:18:46 +0000533 llvm::StructType *ObjCMethodListTy = llvm::StructType::get(VMContext,
Mike Stump1eb44332009-09-09 15:08:12 +0000534 NextPtrTy,
535 IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000536 ObjCMethodArrayTy,
537 NULL);
538 // Refine next pointer type to concrete type
539 llvm::cast<llvm::OpaqueType>(
540 OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
541 ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
542
543 Methods.clear();
Owen Anderson03e20502009-07-30 23:11:26 +0000544 Methods.push_back(llvm::ConstantPointerNull::get(
Owen Anderson96e0fc72009-07-29 22:16:19 +0000545 llvm::PointerType::getUnqual(ObjCMethodListTy)));
Owen Anderson0032b272009-08-13 21:57:51 +0000546 Methods.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000547 MethodTypes.size()));
548 Methods.push_back(MethodArray);
Mike Stump1eb44332009-09-09 15:08:12 +0000549
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000550 // Create an instance of the structure
551 return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
552}
553
554/// Generates an IvarList. Used in construction of a objc_class.
555llvm::Constant *CGObjCGNU::GenerateIvarList(
556 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
557 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
558 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets) {
Mike Stump1eb44332009-09-09 15:08:12 +0000559 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000560 llvm::StructType *ObjCIvarTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000561 PtrToInt8Ty,
562 PtrToInt8Ty,
563 IntTy,
564 NULL);
565 std::vector<llvm::Constant*> Ivars;
566 std::vector<llvm::Constant*> Elements;
567 for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
568 Elements.clear();
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000569 Elements.push_back(IvarNames[i]);
570 Elements.push_back(IvarTypes[i]);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000571 Elements.push_back(IvarOffsets[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000572 Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000573 }
574
575 // Array of method structures
Owen Anderson96e0fc72009-07-29 22:16:19 +0000576 llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000577 IvarNames.size());
578
Mike Stump1eb44332009-09-09 15:08:12 +0000579
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000580 Elements.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000581 Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
Owen Anderson7db6d832009-07-28 18:33:04 +0000582 Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000583 // Structure containing array and array count
Owen Anderson47a434f2009-08-05 23:18:46 +0000584 llvm::StructType *ObjCIvarListTy = llvm::StructType::get(VMContext, IntTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000585 ObjCIvarArrayTy,
586 NULL);
587
588 // Create an instance of the structure
589 return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
590}
591
592/// Generate a class structure
593llvm::Constant *CGObjCGNU::GenerateClassStructure(
594 llvm::Constant *MetaClass,
595 llvm::Constant *SuperClass,
596 unsigned info,
Chris Lattnerd002cc62008-06-26 04:47:04 +0000597 const char *Name,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000598 llvm::Constant *Version,
599 llvm::Constant *InstanceSize,
600 llvm::Constant *IVars,
601 llvm::Constant *Methods,
602 llvm::Constant *Protocols) {
603 // Set up the class structure
604 // Note: Several of these are char*s when they should be ids. This is
605 // because the runtime performs this translation on load.
Owen Anderson47a434f2009-08-05 23:18:46 +0000606 llvm::StructType *ClassTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000607 PtrToInt8Ty, // class_pointer
608 PtrToInt8Ty, // super_class
609 PtrToInt8Ty, // name
610 LongTy, // version
611 LongTy, // info
612 LongTy, // instance_size
613 IVars->getType(), // ivars
614 Methods->getType(), // methods
Mike Stump1eb44332009-09-09 15:08:12 +0000615 // These are all filled in by the runtime, so we pretend
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000616 PtrTy, // dtable
617 PtrTy, // subclass_list
618 PtrTy, // sibling_class
619 PtrTy, // protocols
620 PtrTy, // gc_object_type
621 NULL);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000622 llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000623 llvm::Constant *NullP =
Owen Anderson03e20502009-07-30 23:11:26 +0000624 llvm::ConstantPointerNull::get(PtrTy);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000625 // Fill in the structure
626 std::vector<llvm::Constant*> Elements;
Owen Anderson3c4972d2009-07-29 18:54:39 +0000627 Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000628 Elements.push_back(SuperClass);
Chris Lattnerd002cc62008-06-26 04:47:04 +0000629 Elements.push_back(MakeConstantString(Name, ".class_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000630 Elements.push_back(Zero);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000631 Elements.push_back(llvm::ConstantInt::get(LongTy, info));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000632 Elements.push_back(InstanceSize);
633 Elements.push_back(IVars);
634 Elements.push_back(Methods);
635 Elements.push_back(NullP);
636 Elements.push_back(NullP);
637 Elements.push_back(NullP);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000638 Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000639 Elements.push_back(NullP);
640 // Create an instance of the structure
Chris Lattner1565e032008-07-21 06:31:05 +0000641 return MakeGlobal(ClassTy, Elements, SymbolNameForClass(Name));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000642}
643
644llvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
645 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
646 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes) {
Mike Stump1eb44332009-09-09 15:08:12 +0000647 // Get the method structure type.
Owen Anderson47a434f2009-08-05 23:18:46 +0000648 llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000649 PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
650 PtrToInt8Ty,
651 NULL);
652 std::vector<llvm::Constant*> Methods;
653 std::vector<llvm::Constant*> Elements;
654 for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
655 Elements.clear();
Mike Stump1eb44332009-09-09 15:08:12 +0000656 Elements.push_back(MethodNames[i]);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000657 Elements.push_back(MethodTypes[i]);
Owen Anderson08e25242009-07-27 22:29:56 +0000658 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000659 }
Owen Anderson96e0fc72009-07-29 22:16:19 +0000660 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000661 MethodNames.size());
Owen Anderson7db6d832009-07-28 18:33:04 +0000662 llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +0000663 Methods);
Owen Anderson47a434f2009-08-05 23:18:46 +0000664 llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000665 IntTy, ObjCMethodArrayTy, NULL);
666 Methods.clear();
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000667 Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000668 Methods.push_back(Array);
669 return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
670}
Mike Stumpbb1c8602009-07-31 21:31:32 +0000671
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000672// Create the protocol list structure used in classes, categories and so on
673llvm::Constant *CGObjCGNU::GenerateProtocolList(
674 const llvm::SmallVectorImpl<std::string> &Protocols) {
Owen Anderson96e0fc72009-07-29 22:16:19 +0000675 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000676 Protocols.size());
Owen Anderson47a434f2009-08-05 23:18:46 +0000677 llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000678 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
679 LongTy,//FIXME: Should be size_t
680 ProtocolArrayTy,
681 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000682 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000683 for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
684 iter != endIter ; iter++) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000685 llvm::Constant *protocol = ExistingProtocols[*iter];
686 if (!protocol)
687 protocol = GenerateEmptyProtocol(*iter);
Owen Anderson3c4972d2009-07-29 18:54:39 +0000688 llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(protocol,
Owen Andersona1cf15f2009-07-14 23:10:40 +0000689 PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000690 Elements.push_back(Ptr);
691 }
Owen Anderson7db6d832009-07-28 18:33:04 +0000692 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000693 Elements);
694 Elements.clear();
695 Elements.push_back(NULLPtr);
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000696 Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000697 Elements.push_back(ProtocolArray);
698 return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
699}
700
Mike Stump1eb44332009-09-09 15:08:12 +0000701llvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000702 const ObjCProtocolDecl *PD) {
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000703 llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
Mike Stump1eb44332009-09-09 15:08:12 +0000704 const llvm::Type *T =
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000705 CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
Owen Anderson96e0fc72009-07-29 22:16:19 +0000706 return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000707}
708
709llvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
710 const std::string &ProtocolName) {
711 llvm::SmallVector<std::string, 0> EmptyStringVector;
712 llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
713
714 llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
715 llvm::Constant *InstanceMethodList =
716 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
717 llvm::Constant *ClassMethodList =
718 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
719 // Protocols are objects containing lists of the methods implemented and
720 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000721 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000722 PtrToInt8Ty,
723 ProtocolList->getType(),
724 InstanceMethodList->getType(),
725 ClassMethodList->getType(),
726 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000727 std::vector<llvm::Constant*> Elements;
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000728 // The isa pointer must be set to a magic number so the runtime knows it's
729 // the correct layout.
Owen Anderson3c4972d2009-07-29 18:54:39 +0000730 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Owen Anderson0032b272009-08-13 21:57:51 +0000731 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), ProtocolVersion), IdTy));
Fariborz Jahanianf8c4f542009-03-31 18:27:22 +0000732 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
733 Elements.push_back(ProtocolList);
734 Elements.push_back(InstanceMethodList);
735 Elements.push_back(ClassMethodList);
736 return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000737}
738
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000739void CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
740 ASTContext &Context = CGM.getContext();
Chris Lattner8ec03f52008-11-24 03:54:41 +0000741 std::string ProtocolName = PD->getNameAsString();
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000742 llvm::SmallVector<std::string, 16> Protocols;
743 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
744 E = PD->protocol_end(); PI != E; ++PI)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000745 Protocols.push_back((*PI)->getNameAsString());
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000746 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
747 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000748 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
749 E = PD->instmeth_end(); iter != E; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000750 std::string TypeStr;
751 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
752 InstanceMethodNames.push_back(
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000753 MakeConstantString((*iter)->getSelector().getAsString()));
754 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000755 }
756 // Collect information about class methods:
757 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
758 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +0000759 for (ObjCProtocolDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000760 iter = PD->classmeth_begin(), endIter = PD->classmeth_end();
761 iter != endIter ; iter++) {
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000762 std::string TypeStr;
763 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
764 ClassMethodNames.push_back(
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000765 MakeConstantString((*iter)->getSelector().getAsString()));
766 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbaraf2f62c2008-08-13 00:59:25 +0000767 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000768
769 llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
770 llvm::Constant *InstanceMethodList =
771 GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
772 llvm::Constant *ClassMethodList =
773 GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
774 // Protocols are objects containing lists of the methods implemented and
775 // protocols adopted.
Owen Anderson47a434f2009-08-05 23:18:46 +0000776 llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000777 PtrToInt8Ty,
778 ProtocolList->getType(),
779 InstanceMethodList->getType(),
780 ClassMethodList->getType(),
781 NULL);
Mike Stump1eb44332009-09-09 15:08:12 +0000782 std::vector<llvm::Constant*> Elements;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000783 // The isa pointer must be set to a magic number so the runtime knows it's
784 // the correct layout.
Owen Anderson3c4972d2009-07-29 18:54:39 +0000785 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
Owen Anderson0032b272009-08-13 21:57:51 +0000786 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), ProtocolVersion), IdTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000787 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
788 Elements.push_back(ProtocolList);
789 Elements.push_back(InstanceMethodList);
790 Elements.push_back(ClassMethodList);
Mike Stump1eb44332009-09-09 15:08:12 +0000791 ExistingProtocols[ProtocolName] =
Owen Anderson3c4972d2009-07-29 18:54:39 +0000792 llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000793 ".objc_protocol"), IdTy);
794}
795
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000796void CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
Chris Lattner8ec03f52008-11-24 03:54:41 +0000797 std::string ClassName = OCD->getClassInterface()->getNameAsString();
798 std::string CategoryName = OCD->getNameAsString();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000799 // Collect information about instance methods
800 llvm::SmallVector<Selector, 16> InstanceMethodSels;
801 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +0000802 for (ObjCCategoryImplDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000803 iter = OCD->instmeth_begin(), endIter = OCD->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +0000804 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000805 InstanceMethodSels.push_back((*iter)->getSelector());
806 std::string TypeStr;
807 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000808 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000809 }
810
811 // Collect information about class methods
812 llvm::SmallVector<Selector, 16> ClassMethodSels;
813 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +0000814 for (ObjCCategoryImplDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000815 iter = OCD->classmeth_begin(), endIter = OCD->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +0000816 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000817 ClassMethodSels.push_back((*iter)->getSelector());
818 std::string TypeStr;
819 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000820 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000821 }
822
823 // Collect the names of referenced protocols
824 llvm::SmallVector<std::string, 16> Protocols;
825 const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
826 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
827 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
828 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000829 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000830
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000831 std::vector<llvm::Constant*> Elements;
832 Elements.push_back(MakeConstantString(CategoryName));
833 Elements.push_back(MakeConstantString(ClassName));
Mike Stump1eb44332009-09-09 15:08:12 +0000834 // Instance method list
Owen Anderson3c4972d2009-07-29 18:54:39 +0000835 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +0000836 ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000837 false), PtrTy));
838 // Class method list
Owen Anderson3c4972d2009-07-29 18:54:39 +0000839 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattnera4210072008-06-26 05:08:00 +0000840 ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000841 PtrTy));
842 // Protocol list
Owen Anderson3c4972d2009-07-29 18:54:39 +0000843 Elements.push_back(llvm::ConstantExpr::getBitCast(
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000844 GenerateProtocolList(Protocols), PtrTy));
Owen Anderson3c4972d2009-07-29 18:54:39 +0000845 Categories.push_back(llvm::ConstantExpr::getBitCast(
Mike Stump1eb44332009-09-09 15:08:12 +0000846 MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
Owen Anderson47a434f2009-08-05 23:18:46 +0000847 PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000848}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000849
850void CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
851 ASTContext &Context = CGM.getContext();
852
853 // Get the superclass name.
Mike Stump1eb44332009-09-09 15:08:12 +0000854 const ObjCInterfaceDecl * SuperClassDecl =
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000855 OID->getClassInterface()->getSuperClass();
Chris Lattner8ec03f52008-11-24 03:54:41 +0000856 std::string SuperClassName;
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000857 if (SuperClassDecl) {
Chris Lattner8ec03f52008-11-24 03:54:41 +0000858 SuperClassName = SuperClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000859 EmitClassRef(SuperClassName);
860 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000861
862 // Get the class name
Chris Lattner09dc6662009-04-01 02:00:48 +0000863 ObjCInterfaceDecl *ClassDecl =
864 const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
Chris Lattner8ec03f52008-11-24 03:54:41 +0000865 std::string ClassName = ClassDecl->getNameAsString();
Chris Lattner2a8e4e12009-06-15 01:09:11 +0000866 // Emit the symbol that is used to generate linker errors if this class is
867 // referenced in other modules but not declared.
Fariborz Jahanianc51db232009-07-03 15:10:14 +0000868 std::string classSymbolName = "__objc_class_name_" + ClassName;
Mike Stump1eb44332009-09-09 15:08:12 +0000869 if (llvm::GlobalVariable *symbol =
Fariborz Jahanianc51db232009-07-03 15:10:14 +0000870 TheModule.getGlobalVariable(classSymbolName)) {
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000871 symbol->setInitializer(llvm::ConstantInt::get(LongTy, 0));
Fariborz Jahanianc51db232009-07-03 15:10:14 +0000872 } else {
Owen Anderson1c431b32009-07-08 19:05:04 +0000873 new llvm::GlobalVariable(TheModule, LongTy, false,
Owen Anderson4a28d5d2009-07-24 23:12:58 +0000874 llvm::GlobalValue::ExternalLinkage, llvm::ConstantInt::get(LongTy, 0),
Owen Anderson1c431b32009-07-08 19:05:04 +0000875 classSymbolName);
Fariborz Jahanianc51db232009-07-03 15:10:14 +0000876 }
Mike Stump1eb44332009-09-09 15:08:12 +0000877
Daniel Dunbar2bebbf02009-05-03 10:46:44 +0000878 // Get the size of instances.
879 int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000880
881 // Collect information about instance variables.
882 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
883 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
884 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
Mike Stump1eb44332009-09-09 15:08:12 +0000885
886 int superInstanceSize = !SuperClassDecl ? 0 :
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000887 Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
888 // For non-fragile ivars, set the instance size to 0 - {the size of just this
889 // class}. The runtime will then set this to the correct value on load.
890 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
891 instanceSize = 0 - (instanceSize - superInstanceSize);
892 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000893 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
894 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
895 // Store the name
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000896 IvarNames.push_back(MakeConstantString((*iter)->getNameAsString()));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000897 // Get the type encoding for this ivar
898 std::string TypeStr;
Daniel Dunbar0d504c12008-10-17 20:21:44 +0000899 Context.getObjCEncodingForType((*iter)->getType(), TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000900 IvarTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000901 // Get the offset
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000902 uint64_t Offset;
903 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000904 Offset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter) -
905 superInstanceSize;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +0000906 } else {
907 Offset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter);
908 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000909 IvarOffsets.push_back(
Owen Anderson0032b272009-08-13 21:57:51 +0000910 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000911 }
912
913 // Collect information about instance methods
914 llvm::SmallVector<Selector, 16> InstanceMethodSels;
915 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Mike Stump1eb44332009-09-09 15:08:12 +0000916 for (ObjCImplementationDecl::instmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000917 iter = OID->instmeth_begin(), endIter = OID->instmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +0000918 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000919 InstanceMethodSels.push_back((*iter)->getSelector());
920 std::string TypeStr;
921 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000922 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000923 }
Mike Stump1eb44332009-09-09 15:08:12 +0000924 for (ObjCImplDecl::propimpl_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000925 iter = OID->propimpl_begin(), endIter = OID->propimpl_end();
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000926 iter != endIter ; iter++) {
927 ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
928 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
929 InstanceMethodSels.push_back(getter->getSelector());
930 std::string TypeStr;
931 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000932 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000933 }
934 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
935 InstanceMethodSels.push_back(setter->getSelector());
936 std::string TypeStr;
937 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000938 InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
Fariborz Jahanian1e64a952009-05-17 16:49:27 +0000939 }
940 }
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000941
942 // Collect information about class methods
943 llvm::SmallVector<Selector, 16> ClassMethodSels;
944 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregor653f1b12009-04-23 01:02:12 +0000945 for (ObjCImplementationDecl::classmeth_iterator
Argyrios Kyrtzidis17945a02009-06-30 02:36:12 +0000946 iter = OID->classmeth_begin(), endIter = OID->classmeth_end();
Douglas Gregor653f1b12009-04-23 01:02:12 +0000947 iter != endIter ; iter++) {
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000948 ClassMethodSels.push_back((*iter)->getSelector());
949 std::string TypeStr;
950 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000951 ClassMethodTypes.push_back(MakeConstantString(TypeStr));
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000952 }
953 // Collect the names of referenced protocols
954 llvm::SmallVector<std::string, 16> Protocols;
955 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
956 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
957 E = Protos.end(); I != E; ++I)
Chris Lattnerd9d22dd2008-11-24 05:29:24 +0000958 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbar7ded7f42008-08-15 22:20:32 +0000959
960
961
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000962 // Get the superclass pointer.
963 llvm::Constant *SuperClass;
Chris Lattner8ec03f52008-11-24 03:54:41 +0000964 if (!SuperClassName.empty()) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000965 SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
966 } else {
Owen Anderson03e20502009-07-30 23:11:26 +0000967 SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000968 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000969 // Empty vector used to construct empty method lists
970 llvm::SmallVector<llvm::Constant*, 1> empty;
971 // Generate the method and instance variable lists
972 llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +0000973 InstanceMethodSels, InstanceMethodTypes, false);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000974 llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
Chris Lattnera4210072008-06-26 05:08:00 +0000975 ClassMethodSels, ClassMethodTypes, true);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +0000976 llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
977 IvarOffsets);
Mike Stump1eb44332009-09-09 15:08:12 +0000978 // Irrespective of whether we are compiling for a fragile or non-fragile ABI,
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000979 // we emit a symbol containing the offset for each ivar in the class. This
980 // allows code compiled for the non-Fragile ABI to inherit from code compiled
981 // for the legacy ABI, without causing problems. The converse is also
982 // possible, but causes all ivar accesses to be fragile.
983 int i = 0;
984 // Offset pointer for getting at the correct field in the ivar list when
985 // setting up the alias. These are: The base address for the global, the
986 // ivar array (second field), the ivar in this list (set for each ivar), and
987 // the offset (third field in ivar structure)
988 const llvm::Type *IndexTy = llvm::Type::getInt32Ty(VMContext);
989 llvm::Constant *offsetPointerIndexes[] = {Zeros[0],
Mike Stump1eb44332009-09-09 15:08:12 +0000990 llvm::ConstantInt::get(IndexTy, 1), 0,
David Chisnall8a5a9aa2009-08-31 16:41:57 +0000991 llvm::ConstantInt::get(IndexTy, 2) };
992
993 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
994 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
995 const std::string Name = "__objc_ivar_offset_" + ClassName + '.'
996 +(*iter)->getNameAsString();
997 offsetPointerIndexes[2] = llvm::ConstantInt::get(IndexTy, i++);
998 // Get the correct ivar field
999 llvm::Constant *offsetValue = llvm::ConstantExpr::getGetElementPtr(
1000 IvarList, offsetPointerIndexes, 4);
1001 // Get the existing alias, if one exists.
1002 llvm::GlobalVariable *offset = TheModule.getNamedGlobal(Name);
1003 if (offset) {
1004 offset->setInitializer(offsetValue);
1005 // If this is the real definition, change its linkage type so that
1006 // different modules will use this one, rather than their private
1007 // copy.
1008 offset->setLinkage(llvm::GlobalValue::ExternalLinkage);
1009 } else {
1010 // Add a new alias if there isn't one already.
1011 offset = new llvm::GlobalVariable(TheModule, offsetValue->getType(),
1012 false, llvm::GlobalValue::ExternalLinkage, offsetValue, Name);
1013 }
1014 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001015 //Generate metaclass for class methods
1016 llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
Chris Lattner1565e032008-07-21 06:31:05 +00001017 NULLPtr, 0x2L, /*name*/"", 0, Zeros[0], GenerateIvarList(
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001018 empty, empty, empty), ClassMethodList, NULLPtr);
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001019
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001020 // Generate the class structure
Chris Lattner8ec03f52008-11-24 03:54:41 +00001021 llvm::Constant *ClassStruct =
1022 GenerateClassStructure(MetaClassStruct, SuperClass, 0x1L,
1023 ClassName.c_str(), 0,
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001024 llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001025 MethodList, GenerateProtocolList(Protocols));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001026
1027 // Resolve the class aliases, if they exist.
1028 if (ClassPtrAlias) {
1029 ClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001030 llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001031 ClassPtrAlias = 0;
1032 }
1033 if (MetaClassPtrAlias) {
1034 MetaClassPtrAlias->setAliasee(
Owen Anderson3c4972d2009-07-29 18:54:39 +00001035 llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
Daniel Dunbar5efccb12009-05-04 15:31:17 +00001036 MetaClassPtrAlias = 0;
1037 }
1038
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001039 // Add class structure to list to be added to the symtab later
Owen Anderson3c4972d2009-07-29 18:54:39 +00001040 ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001041 Classes.push_back(ClassStruct);
1042}
1043
Fariborz Jahanianc38e9af2009-06-23 21:47:46 +00001044
Mike Stump1eb44332009-09-09 15:08:12 +00001045llvm::Function *CGObjCGNU::ModuleInitFunction() {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001046 // Only emit an ObjC load function if no Objective-C stuff has been called
1047 if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
1048 ExistingProtocols.empty() && TypedSelectors.empty() &&
Anton Korobeynikov5f58b912008-06-01 15:14:46 +00001049 UntypedSelectors.empty())
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001050 return NULL;
Eli Friedman1b8956e2008-06-01 16:00:02 +00001051
Chris Lattnere160c9b2009-01-27 05:06:01 +00001052 const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1053 SelectorTy->getElementType());
1054 const llvm::Type *SelStructPtrTy = SelectorTy;
1055 bool isSelOpaque = false;
1056 if (SelStructTy == 0) {
Owen Anderson47a434f2009-08-05 23:18:46 +00001057 SelStructTy = llvm::StructType::get(VMContext, PtrToInt8Ty,
1058 PtrToInt8Ty, NULL);
Owen Anderson96e0fc72009-07-29 22:16:19 +00001059 SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
Chris Lattnere160c9b2009-01-27 05:06:01 +00001060 isSelOpaque = true;
1061 }
1062
Eli Friedman1b8956e2008-06-01 16:00:02 +00001063 // Name the ObjC types to make the IR a bit easier to read
Chris Lattnere160c9b2009-01-27 05:06:01 +00001064 TheModule.addTypeName(".objc_selector", SelStructPtrTy);
Eli Friedman1b8956e2008-06-01 16:00:02 +00001065 TheModule.addTypeName(".objc_id", IdTy);
1066 TheModule.addTypeName(".objc_imp", IMPTy);
1067
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001068 std::vector<llvm::Constant*> Elements;
Chris Lattner71238f62009-04-25 23:19:45 +00001069 llvm::Constant *Statics = NULLPtr;
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001070 // Generate statics list:
Chris Lattner71238f62009-04-25 23:19:45 +00001071 if (ConstantStrings.size()) {
Owen Anderson96e0fc72009-07-29 22:16:19 +00001072 llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
Chris Lattner71238f62009-04-25 23:19:45 +00001073 ConstantStrings.size() + 1);
1074 ConstantStrings.push_back(NULLPtr);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001075
1076 const char *StringClass = CGM.getLangOptions().ObjCConstantStringClass;
1077 if (!StringClass) StringClass = "NXConstantString";
1078 Elements.push_back(MakeConstantString(StringClass,
1079 ".objc_static_class_name"));
Owen Anderson7db6d832009-07-28 18:33:04 +00001080 Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
Chris Lattner71238f62009-04-25 23:19:45 +00001081 ConstantStrings));
Mike Stump1eb44332009-09-09 15:08:12 +00001082 llvm::StructType *StaticsListTy =
Owen Anderson47a434f2009-08-05 23:18:46 +00001083 llvm::StructType::get(VMContext, PtrToInt8Ty, StaticsArrayTy, NULL);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001084 llvm::Type *StaticsListPtrTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001085 llvm::PointerType::getUnqual(StaticsListTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001086 Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
Mike Stump1eb44332009-09-09 15:08:12 +00001087 llvm::ArrayType *StaticsListArrayTy =
Owen Anderson96e0fc72009-07-29 22:16:19 +00001088 llvm::ArrayType::get(StaticsListPtrTy, 2);
Chris Lattner71238f62009-04-25 23:19:45 +00001089 Elements.clear();
1090 Elements.push_back(Statics);
Owen Andersonc9c88b42009-07-31 20:28:54 +00001091 Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
Chris Lattner71238f62009-04-25 23:19:45 +00001092 Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001093 Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
Chris Lattner71238f62009-04-25 23:19:45 +00001094 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001095 // Array of classes, categories, and constant objects
Owen Anderson96e0fc72009-07-29 22:16:19 +00001096 llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001097 Classes.size() + Categories.size() + 2);
Mike Stump1eb44332009-09-09 15:08:12 +00001098 llvm::StructType *SymTabTy = llvm::StructType::get(VMContext,
Owen Anderson47a434f2009-08-05 23:18:46 +00001099 LongTy, SelStructPtrTy,
Owen Anderson0032b272009-08-13 21:57:51 +00001100 llvm::Type::getInt16Ty(VMContext),
1101 llvm::Type::getInt16Ty(VMContext),
Chris Lattner630404b2008-06-26 04:10:42 +00001102 ClassListTy, NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001103
1104 Elements.clear();
1105 // Pointer to an array of selectors used in this module.
1106 std::vector<llvm::Constant*> Selectors;
1107 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1108 iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
1109 iter != iterEnd ; ++iter) {
Chris Lattner630404b2008-06-26 04:10:42 +00001110 Elements.push_back(MakeConstantString(iter->first.first, ".objc_sel_name"));
1111 Elements.push_back(MakeConstantString(iter->first.second,
1112 ".objc_sel_types"));
Owen Anderson08e25242009-07-27 22:29:56 +00001113 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001114 Elements.clear();
1115 }
1116 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1117 iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
Chris Lattner630404b2008-06-26 04:10:42 +00001118 iter != iterEnd; ++iter) {
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001119 Elements.push_back(
Chris Lattner630404b2008-06-26 04:10:42 +00001120 MakeConstantString(iter->getKeyData(), ".objc_sel_name"));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001121 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001122 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001123 Elements.clear();
1124 }
1125 Elements.push_back(NULLPtr);
1126 Elements.push_back(NULLPtr);
Owen Anderson08e25242009-07-27 22:29:56 +00001127 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001128 Elements.clear();
1129 // Number of static selectors
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001130 Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001131 llvm::Constant *SelectorList = MakeGlobal(
Owen Anderson96e0fc72009-07-29 22:16:19 +00001132 llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001133 ".objc_selector_list");
Mike Stump1eb44332009-09-09 15:08:12 +00001134 Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
Chris Lattnere160c9b2009-01-27 05:06:01 +00001135 SelStructPtrTy));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001136
1137 // Now that all of the static selectors exist, create pointers to them.
1138 int index = 0;
1139 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1140 iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
1141 iter != iterEnd; ++iter) {
1142 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001143 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Owen Anderson1c431b32009-07-08 19:05:04 +00001144 llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
1145 true, llvm::GlobalValue::InternalLinkage,
Owen Anderson3c4972d2009-07-29 18:54:39 +00001146 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
Owen Anderson1c431b32009-07-08 19:05:04 +00001147 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001148 // If selectors are defined as an opaque type, cast the pointer to this
1149 // type.
1150 if (isSelOpaque) {
Owen Anderson3c4972d2009-07-29 18:54:39 +00001151 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001152 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001153 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001154 (*iter).second->setAliasee(SelPtr);
1155 }
1156 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1157 iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1158 iter != iterEnd; iter++) {
1159 llvm::Constant *Idxs[] = {Zeros[0],
Owen Anderson0032b272009-08-13 21:57:51 +00001160 llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
Mike Stumpbb1c8602009-07-31 21:31:32 +00001161 llvm::Constant *SelPtr = new llvm::GlobalVariable
Mike Stump1eb44332009-09-09 15:08:12 +00001162 (TheModule, SelStructPtrTy,
Mike Stumpbb1c8602009-07-31 21:31:32 +00001163 true, llvm::GlobalValue::InternalLinkage,
1164 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1165 ".objc_sel_ptr");
Chris Lattnere160c9b2009-01-27 05:06:01 +00001166 // If selectors are defined as an opaque type, cast the pointer to this
1167 // type.
1168 if (isSelOpaque) {
Owen Anderson3c4972d2009-07-29 18:54:39 +00001169 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001170 llvm::PointerType::getUnqual(SelectorTy));
Chris Lattnere160c9b2009-01-27 05:06:01 +00001171 }
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001172 (*iter).second->setAliasee(SelPtr);
1173 }
1174 // Number of classes defined.
Mike Stump1eb44332009-09-09 15:08:12 +00001175 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001176 Classes.size()));
1177 // Number of categories defined
Mike Stump1eb44332009-09-09 15:08:12 +00001178 Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001179 Categories.size()));
1180 // Create an array of classes, then categories, then static object instances
1181 Classes.insert(Classes.end(), Categories.begin(), Categories.end());
1182 // NULL-terminated list of static object instances (mainly constant strings)
1183 Classes.push_back(Statics);
1184 Classes.push_back(NULLPtr);
Owen Anderson7db6d832009-07-28 18:33:04 +00001185 llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001186 Elements.push_back(ClassList);
Mike Stump1eb44332009-09-09 15:08:12 +00001187 // Construct the symbol table
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001188 llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
1189
1190 // The symbol table is contained in a module which has some version-checking
1191 // constants
Owen Anderson47a434f2009-08-05 23:18:46 +00001192 llvm::StructType * ModuleTy = llvm::StructType::get(VMContext, LongTy, LongTy,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001193 PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001194 Elements.clear();
1195 // Runtime version used for compatibility checking.
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001196 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
Mike Stump1eb44332009-09-09 15:08:12 +00001197 Elements.push_back(llvm::ConstantInt::get(LongTy,
1198 NonFragileRuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001199 } else {
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001200 Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001201 }
Fariborz Jahanian91a0b512009-04-01 19:49:42 +00001202 // sizeof(ModuleTy)
1203 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001204 Elements.push_back(llvm::ConstantInt::get(LongTy,
Owen Andersona1cf15f2009-07-14 23:10:40 +00001205 td.getTypeSizeInBits(ModuleTy)/8));
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001206 //FIXME: Should be the path to the file where this module was declared
1207 Elements.push_back(NULLPtr);
1208 Elements.push_back(SymTab);
1209 llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
1210
1211 // Create the load function calling the runtime entry point with the module
1212 // structure
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001213 llvm::Function * LoadFunction = llvm::Function::Create(
Owen Anderson0032b272009-08-13 21:57:51 +00001214 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), false),
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001215 llvm::GlobalValue::InternalLinkage, ".objc_load_function",
1216 &TheModule);
Owen Anderson0032b272009-08-13 21:57:51 +00001217 llvm::BasicBlock *EntryBB =
1218 llvm::BasicBlock::Create(VMContext, "entry", LoadFunction);
Owen Andersona1cf15f2009-07-14 23:10:40 +00001219 CGBuilderTy Builder(VMContext);
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001220 Builder.SetInsertPoint(EntryBB);
Fariborz Jahanian26c82942009-03-30 18:02:14 +00001221
1222 std::vector<const llvm::Type*> Params(1,
Owen Anderson96e0fc72009-07-29 22:16:19 +00001223 llvm::PointerType::getUnqual(ModuleTy));
1224 llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
Owen Anderson0032b272009-08-13 21:57:51 +00001225 llvm::Type::getVoidTy(VMContext), Params, true), "__objc_exec_class");
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001226 Builder.CreateCall(Register, Module);
1227 Builder.CreateRetVoid();
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001228
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001229 return LoadFunction;
1230}
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001231
Fariborz Jahanian679a5022009-01-10 21:06:09 +00001232llvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
Mike Stump1eb44332009-09-09 15:08:12 +00001233 const ObjCContainerDecl *CD) {
1234 const ObjCCategoryImplDecl *OCD =
Steve Naroff3e0a5402009-01-08 19:41:02 +00001235 dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
Chris Lattner077bf5e2008-11-24 03:33:13 +00001236 std::string CategoryName = OCD ? OCD->getNameAsString() : "";
1237 std::string ClassName = OMD->getClassInterface()->getNameAsString();
1238 std::string MethodName = OMD->getSelector().getAsString();
Douglas Gregorf8d49f62009-01-09 17:18:27 +00001239 bool isClassMethod = !OMD->isInstanceMethod();
Daniel Dunbar7ded7f42008-08-15 22:20:32 +00001240
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001241 CodeGenTypes &Types = CGM.getTypes();
Mike Stump1eb44332009-09-09 15:08:12 +00001242 const llvm::FunctionType *MethodTy =
Daniel Dunbar541b63b2009-02-02 23:23:47 +00001243 Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
Anton Korobeynikov20ff3102008-06-01 14:13:53 +00001244 std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
1245 MethodName, isClassMethod);
1246
Mike Stump1eb44332009-09-09 15:08:12 +00001247 llvm::Function *Method
1248 = llvm::Function::Create(MethodTy,
1249 llvm::GlobalValue::InternalLinkage,
1250 FunctionName,
1251 &TheModule);
Chris Lattner391d77a2008-03-30 23:03:07 +00001252 return Method;
1253}
1254
Daniel Dunbar49f66022008-09-24 03:38:44 +00001255llvm::Function *CGObjCGNU::GetPropertyGetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001256 std::vector<const llvm::Type*> Params;
1257 const llvm::Type *BoolTy =
1258 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1259 Params.push_back(IdTy);
1260 Params.push_back(SelectorTy);
1261 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1262 Params.push_back(LongTy);
1263 Params.push_back(BoolTy);
1264 // void objc_getProperty (id, SEL, ptrdiff_t, bool)
1265 const llvm::FunctionType *FTy =
1266 llvm::FunctionType::get(IdTy, Params, false);
1267 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1268 "objc_getProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001269}
1270
1271llvm::Function *CGObjCGNU::GetPropertySetFunction() {
Mike Stump1eb44332009-09-09 15:08:12 +00001272 std::vector<const llvm::Type*> Params;
1273 const llvm::Type *BoolTy =
1274 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1275 Params.push_back(IdTy);
1276 Params.push_back(SelectorTy);
1277 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1278 Params.push_back(LongTy);
1279 Params.push_back(IdTy);
1280 Params.push_back(BoolTy);
1281 Params.push_back(BoolTy);
1282 // void objc_setProperty (id, SEL, ptrdiff_t, id, bool, bool)
1283 const llvm::FunctionType *FTy =
1284 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Params, false);
1285 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1286 "objc_setProperty"));
Daniel Dunbar49f66022008-09-24 03:38:44 +00001287}
1288
Daniel Dunbar309a4362009-07-24 07:40:24 +00001289llvm::Constant *CGObjCGNU::EnumerationMutationFunction() {
1290 CodeGen::CodeGenTypes &Types = CGM.getTypes();
1291 ASTContext &Ctx = CGM.getContext();
1292 // void objc_enumerationMutation (id)
1293 llvm::SmallVector<QualType,16> Params;
David Chisnall0f436562009-08-17 16:35:33 +00001294 Params.push_back(ASTIdTy);
Daniel Dunbar309a4362009-07-24 07:40:24 +00001295 const llvm::FunctionType *FTy =
1296 Types.GetFunctionType(Types.getFunctionInfo(Ctx.VoidTy, Params), false);
1297 return CGM.CreateRuntimeFunction(FTy, "objc_enumerationMutation");
Anders Carlsson2abd89c2008-08-31 04:05:03 +00001298}
1299
Fariborz Jahanianbd71be42008-11-21 00:49:24 +00001300void CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1301 const Stmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001302 // Pointer to the personality function
1303 llvm::Constant *Personality =
Owen Anderson0032b272009-08-13 21:57:51 +00001304 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getInt32Ty(VMContext),
Chris Lattnerb59761b2009-07-01 04:13:52 +00001305 true),
Chris Lattner5dc08672009-05-08 00:11:50 +00001306 "__gnu_objc_personality_v0");
Owen Anderson3c4972d2009-07-29 18:54:39 +00001307 Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
Chris Lattner5dc08672009-05-08 00:11:50 +00001308 std::vector<const llvm::Type*> Params;
1309 Params.push_back(PtrTy);
1310 llvm::Value *RethrowFn =
Owen Anderson0032b272009-08-13 21:57:51 +00001311 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
Chris Lattner5dc08672009-05-08 00:11:50 +00001312 Params, false), "_Unwind_Resume_or_Rethrow");
1313
1314 bool isTry = isa<ObjCAtTryStmt>(S);
1315 llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
1316 llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
1317 llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
Chris Lattnerd6a99072009-05-11 18:16:28 +00001318 llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
Chris Lattner5dc08672009-05-08 00:11:50 +00001319 llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
1320 llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
1321 llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
1322
1323 // GNU runtime does not currently support @synchronized()
1324 if (!isTry) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001325 std::vector<const llvm::Type*> Args(1, IdTy);
1326 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001327 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001328 llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
Mike Stump1eb44332009-09-09 15:08:12 +00001329 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001330 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1331 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1332 CGF.Builder.CreateCall(SyncEnter, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001333 }
1334
Chris Lattnerd6a99072009-05-11 18:16:28 +00001335
Chris Lattner5dc08672009-05-08 00:11:50 +00001336 // Push an EH context entry, used for handling rethrows and jumps
1337 // through finally.
1338 CGF.PushCleanupBlock(FinallyBlock);
1339
1340 // Emit the statements in the @try {} block
1341 CGF.setInvokeDest(TryHandler);
1342
1343 CGF.EmitBlock(TryBlock);
Mike Stump1eb44332009-09-09 15:08:12 +00001344 CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
Chris Lattner5dc08672009-05-08 00:11:50 +00001345 : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
1346
1347 // Jump to @finally if there is no exception
1348 CGF.EmitBranchThroughCleanup(FinallyEnd);
1349
1350 // Emit the handlers
1351 CGF.EmitBlock(TryHandler);
1352
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001353 // Get the correct versions of the exception handling intrinsics
1354 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
1355 int PointerWidth = td.getTypeSizeInBits(PtrTy);
1356 assert((PointerWidth == 32 || PointerWidth == 64) &&
1357 "Can't yet handle exceptions if pointers are not 32 or 64 bits");
Mike Stump1eb44332009-09-09 15:08:12 +00001358 llvm::Value *llvm_eh_exception =
Chris Lattner5dc08672009-05-08 00:11:50 +00001359 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001360 llvm::Value *llvm_eh_selector = PointerWidth == 32 ?
1361 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector_i32) :
1362 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector_i64);
1363 llvm::Value *llvm_eh_typeid_for = PointerWidth == 32 ?
1364 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for_i32) :
1365 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for_i64);
Chris Lattner5dc08672009-05-08 00:11:50 +00001366
1367 // Exception object
1368 llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1369 llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
1370
1371 llvm::SmallVector<llvm::Value*, 8> ESelArgs;
1372 llvm::SmallVector<std::pair<const ParmVarDecl*, const Stmt*>, 8> Handlers;
1373
1374 ESelArgs.push_back(Exc);
1375 ESelArgs.push_back(Personality);
1376
1377 bool HasCatchAll = false;
1378 // Only @try blocks are allowed @catch blocks, but both can have @finally
1379 if (isTry) {
1380 if (const ObjCAtCatchStmt* CatchStmt =
1381 cast<ObjCAtTryStmt>(S).getCatchStmts()) {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001382 CGF.setInvokeDest(CatchInCatch);
Chris Lattner5dc08672009-05-08 00:11:50 +00001383
1384 for (; CatchStmt; CatchStmt = CatchStmt->getNextCatchStmt()) {
1385 const ParmVarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
Mike Stumpbb1c8602009-07-31 21:31:32 +00001386 Handlers.push_back(std::make_pair(CatchDecl,
1387 CatchStmt->getCatchBody()));
Chris Lattner5dc08672009-05-08 00:11:50 +00001388
1389 // @catch() and @catch(id) both catch any ObjC exception
Steve Naroff14108da2009-07-10 23:34:53 +00001390 if (!CatchDecl || CatchDecl->getType()->isObjCIdType()
Chris Lattner5dc08672009-05-08 00:11:50 +00001391 || CatchDecl->getType()->isObjCQualifiedIdType()) {
1392 // Use i8* null here to signal this is a catch all, not a cleanup.
1393 ESelArgs.push_back(NULLPtr);
1394 HasCatchAll = true;
1395 // No further catches after this one will ever by reached
1396 break;
Mike Stump1eb44332009-09-09 15:08:12 +00001397 }
Chris Lattner5dc08672009-05-08 00:11:50 +00001398
1399 // All other types should be Objective-C interface pointer types.
Mike Stump1eb44332009-09-09 15:08:12 +00001400 const ObjCObjectPointerType *OPT =
Steve Naroff14108da2009-07-10 23:34:53 +00001401 CatchDecl->getType()->getAsObjCObjectPointerType();
1402 assert(OPT && "Invalid @catch type.");
Mike Stump1eb44332009-09-09 15:08:12 +00001403 const ObjCInterfaceType *IT =
Steve Naroff14108da2009-07-10 23:34:53 +00001404 OPT->getPointeeType()->getAsObjCInterfaceType();
Chris Lattner5dc08672009-05-08 00:11:50 +00001405 assert(IT && "Invalid @catch type.");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001406 llvm::Value *EHType =
1407 MakeConstantString(IT->getDecl()->getNameAsString());
Chris Lattner5dc08672009-05-08 00:11:50 +00001408 ESelArgs.push_back(EHType);
1409 }
1410 }
1411 }
1412
1413 // We use a cleanup unless there was already a catch all.
1414 if (!HasCatchAll) {
Owen Anderson0032b272009-08-13 21:57:51 +00001415 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattner5dc08672009-05-08 00:11:50 +00001416 Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
1417 }
1418
1419 // Find which handler was matched.
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001420 llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
Chris Lattner5dc08672009-05-08 00:11:50 +00001421 ESelArgs.begin(), ESelArgs.end(), "selector");
1422
1423 for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
1424 const ParmVarDecl *CatchParam = Handlers[i].first;
1425 const Stmt *CatchBody = Handlers[i].second;
1426
1427 llvm::BasicBlock *Next = 0;
1428
1429 // The last handler always matches.
1430 if (i + 1 != e) {
1431 assert(CatchParam && "Only last handler can be a catch all.");
1432
1433 // Test whether this block matches the type for the selector and branch
1434 // to Match if it does, or to the next BB if it doesn't.
1435 llvm::BasicBlock *Match = CGF.createBasicBlock("match");
1436 Next = CGF.createBasicBlock("catch.next");
Chris Lattnerbb422ad2009-05-08 17:36:08 +00001437 llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
Chris Lattner5dc08672009-05-08 00:11:50 +00001438 CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
1439 CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
1440 Next);
1441
1442 CGF.EmitBlock(Match);
1443 }
Mike Stump1eb44332009-09-09 15:08:12 +00001444
Chris Lattner5dc08672009-05-08 00:11:50 +00001445 if (CatchBody) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001446 llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
1447 CGF.ConvertType(CatchParam->getType()));
Mike Stump1eb44332009-09-09 15:08:12 +00001448
Chris Lattner5dc08672009-05-08 00:11:50 +00001449 // Bind the catch parameter if it exists.
1450 if (CatchParam) {
1451 // CatchParam is a ParmVarDecl because of the grammar
1452 // construction used to handle this, but for codegen purposes
1453 // we treat this as a local decl.
1454 CGF.EmitLocalBlockVarDecl(*CatchParam);
1455 CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
1456 }
1457
1458 CGF.ObjCEHValueStack.push_back(ExcObject);
1459 CGF.EmitStmt(CatchBody);
1460 CGF.ObjCEHValueStack.pop_back();
1461
1462 CGF.EmitBranchThroughCleanup(FinallyEnd);
1463
1464 if (Next)
1465 CGF.EmitBlock(Next);
1466 } else {
1467 assert(!Next && "catchup should be last handler.");
1468
1469 CGF.Builder.CreateStore(Exc, RethrowPtr);
1470 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1471 }
1472 }
Chris Lattnerd6a99072009-05-11 18:16:28 +00001473 // The @finally block is a secondary landing pad for any exceptions thrown in
1474 // @catch() blocks
1475 CGF.EmitBlock(CatchInCatch);
1476 Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1477 ESelArgs.clear();
1478 ESelArgs.push_back(Exc);
1479 ESelArgs.push_back(Personality);
Owen Anderson0032b272009-08-13 21:57:51 +00001480 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
Chris Lattnerd6a99072009-05-11 18:16:28 +00001481 CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
1482 "selector");
1483 CGF.Builder.CreateCall(llvm_eh_typeid_for,
1484 CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
1485 CGF.Builder.CreateStore(Exc, RethrowPtr);
1486 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1487
Chris Lattner5dc08672009-05-08 00:11:50 +00001488 CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
1489
1490 CGF.setInvokeDest(PrevLandingPad);
1491
1492 CGF.EmitBlock(FinallyBlock);
1493
Chris Lattnerd6a99072009-05-11 18:16:28 +00001494
Chris Lattner5dc08672009-05-08 00:11:50 +00001495 if (isTry) {
Mike Stump1eb44332009-09-09 15:08:12 +00001496 if (const ObjCAtFinallyStmt* FinallyStmt =
Chris Lattner5dc08672009-05-08 00:11:50 +00001497 cast<ObjCAtTryStmt>(S).getFinallyStmt())
1498 CGF.EmitStmt(FinallyStmt->getFinallyBody());
1499 } else {
Chris Lattnerd6a99072009-05-11 18:16:28 +00001500 // Emit 'objc_sync_exit(expr)' as finally's sole statement for
Chris Lattner5dc08672009-05-08 00:11:50 +00001501 // @synchronized.
Chris Lattnerd6a99072009-05-11 18:16:28 +00001502 std::vector<const llvm::Type*> Args(1, IdTy);
1503 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001504 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Chris Lattnerd6a99072009-05-11 18:16:28 +00001505 llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
Mike Stump1eb44332009-09-09 15:08:12 +00001506 llvm::Value *SyncArg =
Chris Lattnerd6a99072009-05-11 18:16:28 +00001507 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1508 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1509 CGF.Builder.CreateCall(SyncExit, SyncArg);
Chris Lattner5dc08672009-05-08 00:11:50 +00001510 }
1511
1512 if (Info.SwitchBlock)
1513 CGF.EmitBlock(Info.SwitchBlock);
1514 if (Info.EndBlock)
1515 CGF.EmitBlock(Info.EndBlock);
1516
1517 // Branch around the rethrow code.
1518 CGF.EmitBranch(FinallyEnd);
1519
1520 CGF.EmitBlock(FinallyRethrow);
1521 CGF.Builder.CreateCall(RethrowFn, CGF.Builder.CreateLoad(RethrowPtr));
1522 CGF.Builder.CreateUnreachable();
Mike Stump1eb44332009-09-09 15:08:12 +00001523
Chris Lattner5dc08672009-05-08 00:11:50 +00001524 CGF.EmitBlock(FinallyEnd);
1525
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001526}
1527
1528void CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar49f66022008-09-24 03:38:44 +00001529 const ObjCAtThrowStmt &S) {
Chris Lattner5dc08672009-05-08 00:11:50 +00001530 llvm::Value *ExceptionAsObject;
1531
1532 std::vector<const llvm::Type*> Args(1, IdTy);
1533 llvm::FunctionType *FTy =
Owen Anderson0032b272009-08-13 21:57:51 +00001534 llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
Mike Stump1eb44332009-09-09 15:08:12 +00001535 llvm::Value *ThrowFn =
Chris Lattner5dc08672009-05-08 00:11:50 +00001536 CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
Mike Stump1eb44332009-09-09 15:08:12 +00001537
Chris Lattner5dc08672009-05-08 00:11:50 +00001538 if (const Expr *ThrowExpr = S.getThrowExpr()) {
1539 llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001540 ExceptionAsObject = Exception;
Chris Lattner5dc08672009-05-08 00:11:50 +00001541 } else {
Mike Stump1eb44332009-09-09 15:08:12 +00001542 assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
Chris Lattner5dc08672009-05-08 00:11:50 +00001543 "Unexpected rethrow outside @catch block.");
1544 ExceptionAsObject = CGF.ObjCEHValueStack.back();
1545 }
Fariborz Jahanian1e64a952009-05-17 16:49:27 +00001546 ExceptionAsObject =
1547 CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
Mike Stump1eb44332009-09-09 15:08:12 +00001548
Chris Lattner5dc08672009-05-08 00:11:50 +00001549 // Note: This may have to be an invoke, if we want to support constructs like:
1550 // @try {
1551 // @throw(obj);
1552 // }
1553 // @catch(id) ...
1554 //
1555 // This is effectively turning @throw into an incredibly-expensive goto, but
1556 // it may happen as a result of inlining followed by missed optimizations, or
1557 // as a result of stupidity.
1558 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
1559 if (!UnwindBB) {
1560 CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
1561 CGF.Builder.CreateUnreachable();
1562 } else {
1563 CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
1564 &ExceptionAsObject+1);
1565 }
1566 // Clear the insertion point to indicate we are in unreachable code.
1567 CGF.Builder.ClearInsertionPoint();
Anders Carlsson64d5d6c2008-09-09 10:04:29 +00001568}
1569
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001570llvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001571 llvm::Value *AddrWeakObj) {
Fariborz Jahanian6dc23172008-11-18 21:45:40 +00001572 return 0;
1573}
1574
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001575void CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001576 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian3e283e32008-11-18 22:37:34 +00001577 return;
1578}
1579
Fariborz Jahanian58626502008-11-19 00:59:10 +00001580void CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001581 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian58626502008-11-19 00:59:10 +00001582 return;
1583}
1584
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00001585void CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001586 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian7eda8362008-11-20 19:23:36 +00001587 return;
1588}
1589
Fariborz Jahanian58626502008-11-19 00:59:10 +00001590void CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001591 llvm::Value *src, llvm::Value *dst) {
Fariborz Jahanian58626502008-11-19 00:59:10 +00001592 return;
1593}
1594
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001595void CGObjCGNU::EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
Mike Stump1eb44332009-09-09 15:08:12 +00001596 llvm::Value *DestPtr,
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001597 llvm::Value *SrcPtr,
Fariborz Jahanian08c32132009-08-31 19:33:16 +00001598 QualType Ty) {
Fariborz Jahanian082b02e2009-07-08 01:18:33 +00001599 return;
1600}
1601
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001602llvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
1603 const ObjCInterfaceDecl *ID,
1604 const ObjCIvarDecl *Ivar) {
1605 const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
1606 + '.' + Ivar->getNameAsString();
1607 // Emit the variable and initialize it with what we think the correct value
1608 // is. This allows code compiled with non-fragile ivars to work correctly
1609 // when linked against code which isn't (most of the time).
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001610 llvm::GlobalVariable *IvarOffsetPointer = TheModule.getNamedGlobal(Name);
1611 if (!IvarOffsetPointer) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001612 uint64_t Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
1613 llvm::ConstantInt *OffsetGuess =
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001614 llvm::ConstantInt::get(LongTy, Offset, "ivar");
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001615 // Don't emit the guess in non-PIC code because the linker will not be able
1616 // to replace it with the real version for a library. In non-PIC code you
1617 // must compile with the fragile ABI if you want to use ivars from a
Mike Stump1eb44332009-09-09 15:08:12 +00001618 // GCC-compiled class.
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001619 if (CGM.getLangOptions().PICLevel) {
1620 llvm::GlobalVariable *IvarOffsetGV = new llvm::GlobalVariable(TheModule,
1621 llvm::Type::getInt32Ty(VMContext), false,
1622 llvm::GlobalValue::PrivateLinkage, OffsetGuess, Name+".guess");
1623 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
1624 IvarOffsetGV->getType(), false, llvm::GlobalValue::LinkOnceAnyLinkage,
1625 IvarOffsetGV, Name);
1626 } else {
1627 IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
1628 llvm::PointerType::getUnqual(llvm::Type::getInt32Ty(VMContext)),
1629 false, llvm::GlobalValue::ExternalLinkage, 0, Name);
1630 }
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001631 }
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001632 return IvarOffsetPointer;
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001633}
1634
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00001635LValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
1636 QualType ObjectTy,
1637 llvm::Value *BaseValue,
1638 const ObjCIvarDecl *Ivar,
Fariborz Jahanian598d3f62009-02-03 19:03:09 +00001639 unsigned CVRQualifiers) {
Daniel Dunbar525c9b72009-04-21 01:19:28 +00001640 const ObjCInterfaceDecl *ID = ObjectTy->getAsObjCInterfaceType()->getDecl();
Daniel Dunbar97776872009-04-22 07:32:20 +00001641 return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
1642 EmitIvarOffset(CGF, ID, Ivar));
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00001643}
Mike Stumpbb1c8602009-07-31 21:31:32 +00001644
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001645static const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
1646 const ObjCInterfaceDecl *OID,
1647 const ObjCIvarDecl *OIVD) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001648 llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
Fariborz Jahanian8e6ac1d2009-06-04 01:19:09 +00001649 Context.ShallowCollectObjCIvars(OID, Ivars);
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001650 for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
1651 if (OIVD == Ivars[k])
1652 return OID;
1653 }
Mike Stump1eb44332009-09-09 15:08:12 +00001654
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001655 // Otherwise check in the super class.
1656 if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
1657 return FindIvarInterface(Context, Super, OIVD);
Mike Stump1eb44332009-09-09 15:08:12 +00001658
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001659 return 0;
1660}
Fariborz Jahanian0bb20362009-02-02 20:02:29 +00001661
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001662llvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar2a031922009-04-22 05:08:15 +00001663 const ObjCInterfaceDecl *Interface,
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001664 const ObjCIvarDecl *Ivar) {
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001665 if (CGM.getLangOptions().ObjCNonFragileABI) {
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001666 Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
David Chisnall8a5a9aa2009-08-31 16:41:57 +00001667 return CGF.Builder.CreateLoad(CGF.Builder.CreateLoad(
1668 ObjCIvarOffsetVariable(Interface, Ivar), false, "ivar"));
Fariborz Jahanian9cd96ff2009-05-20 18:41:51 +00001669 }
Daniel Dunbar97776872009-04-22 07:32:20 +00001670 uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
Owen Anderson4a28d5d2009-07-24 23:12:58 +00001671 return llvm::ConstantInt::get(LongTy, Offset, "ivar");
Fariborz Jahanianf63aa3f2009-02-10 19:02:04 +00001672}
1673
Mike Stumpbb1c8602009-07-31 21:31:32 +00001674CodeGen::CGObjCRuntime *
1675CodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM) {
Chris Lattnerdce14062008-06-26 04:19:03 +00001676 return new CGObjCGNU(CGM);
Chris Lattner0f984262008-03-01 08:50:34 +00001677}