blob: 912479f0b0ed04c52c54f4bd8233192e63c554c4 [file] [log] [blame]
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001//===------- CGObjCGNU.cpp - Emit LLVM Code from ASTs for a Module --------===//
Chris Lattnera0fd5ee2008-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 Korobeynikovcd5d08d2008-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 Lattnera0fd5ee2008-03-01 08:50:34 +000014//
15//===----------------------------------------------------------------------===//
16
17#include "CGObjCRuntime.h"
Chris Lattner547907c2008-06-26 04:19:03 +000018#include "CodeGenModule.h"
Daniel Dunbara04840b2008-08-23 03:46:30 +000019#include "CodeGenFunction.h"
Chris Lattnerfee3eb02009-05-08 00:11:50 +000020
Chris Lattner547907c2008-06-26 04:19:03 +000021#include "clang/AST/ASTContext.h"
Daniel Dunbarde300732008-08-11 04:54:23 +000022#include "clang/AST/Decl.h"
Daniel Dunbar84bb85f2008-08-13 00:59:25 +000023#include "clang/AST/DeclObjC.h"
Daniel Dunbar07ddfa92009-05-03 10:46:44 +000024#include "clang/AST/RecordLayout.h"
Chris Lattner4a9e9272009-04-26 01:32:48 +000025#include "clang/AST/StmtObjC.h"
Chris Lattnerfee3eb02009-05-08 00:11:50 +000026
27#include "llvm/Intrinsics.h"
Chris Lattnera0fd5ee2008-03-01 08:50:34 +000028#include "llvm/Module.h"
Chris Lattnera0fd5ee2008-03-01 08:50:34 +000029#include "llvm/ADT/SmallVector.h"
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000030#include "llvm/ADT/StringMap.h"
Daniel Dunbarac93e472008-08-15 22:20:32 +000031#include "llvm/Support/Compiler.h"
Daniel Dunbarac93e472008-08-15 22:20:32 +000032#include "llvm/Target/TargetData.h"
Chris Lattnerfee3eb02009-05-08 00:11:50 +000033
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000034#include <map>
Chris Lattner431dc382009-01-27 05:06:01 +000035
36
Chris Lattner547907c2008-06-26 04:19:03 +000037using namespace clang;
Daniel Dunbar0a2da0f2008-09-09 01:06:48 +000038using namespace CodeGen;
Anton Korobeynikovcd5d08d2008-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 Jahanianf25adf12009-05-20 18:41:51 +000044static const int NonFragileRuntimeVersion = 9;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000045static const int ProtocolVersion = 2;
Chris Lattnera0fd5ee2008-03-01 08:50:34 +000046
Chris Lattnera0fd5ee2008-03-01 08:50:34 +000047namespace {
Chris Lattner547907c2008-06-26 04:19:03 +000048class CGObjCGNU : public CodeGen::CGObjCRuntime {
Chris Lattnera0fd5ee2008-03-01 08:50:34 +000049private:
Chris Lattner547907c2008-06-26 04:19:03 +000050 CodeGen::CodeGenModule &CGM;
Chris Lattnera0fd5ee2008-03-01 08:50:34 +000051 llvm::Module &TheModule;
Chris Lattner431dc382009-01-27 05:06:01 +000052 const llvm::PointerType *SelectorTy;
53 const llvm::PointerType *PtrToInt8Ty;
Fariborz Jahanianbb284742009-02-04 20:31:19 +000054 const llvm::FunctionType *IMPTy;
Chris Lattner431dc382009-01-27 05:06:01 +000055 const llvm::PointerType *IdTy;
56 const llvm::IntegerType *IntTy;
57 const llvm::PointerType *PtrTy;
58 const llvm::IntegerType *LongTy;
59 const llvm::PointerType *PtrToIntTy;
Daniel Dunbarc4532f52009-05-04 15:31:17 +000060 llvm::GlobalAlias *ClassPtrAlias;
61 llvm::GlobalAlias *MetaClassPtrAlias;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000062 std::vector<llvm::Constant*> Classes;
63 std::vector<llvm::Constant*> Categories;
64 std::vector<llvm::Constant*> ConstantStrings;
65 llvm::Function *LoadFunction;
66 llvm::StringMap<llvm::Constant*> ExistingProtocols;
67 typedef std::pair<std::string, std::string> TypedSelector;
68 std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
69 llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
70 // Some zeros used for GEPs in lots of places.
71 llvm::Constant *Zeros[2];
72 llvm::Constant *NULLPtr;
73private:
74 llvm::Constant *GenerateIvarList(
75 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
76 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
77 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets);
78 llvm::Constant *GenerateMethodList(const std::string &ClassName,
79 const std::string &CategoryName,
Chris Lattner578279d2008-06-26 05:08:00 +000080 const llvm::SmallVectorImpl<Selector> &MethodSels,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000081 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
82 bool isClassMethodList);
Fariborz Jahanian926c3712009-03-31 18:27:22 +000083 llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000084 llvm::Constant *GenerateProtocolList(
85 const llvm::SmallVectorImpl<std::string> &Protocols);
86 llvm::Constant *GenerateClassStructure(
87 llvm::Constant *MetaClass,
88 llvm::Constant *SuperClass,
89 unsigned info,
Chris Lattnerad9c3f32008-06-26 04:47:04 +000090 const char *Name,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +000091 llvm::Constant *Version,
92 llvm::Constant *InstanceSize,
93 llvm::Constant *IVars,
94 llvm::Constant *Methods,
95 llvm::Constant *Protocols);
96 llvm::Constant *GenerateProtocolMethodList(
97 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
98 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes);
99 llvm::Constant *MakeConstantString(const std::string &Str, const std::string
100 &Name="");
101 llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
102 std::vector<llvm::Constant*> &V, const std::string &Name="");
103 llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
104 std::vector<llvm::Constant*> &V, const std::string &Name="");
Fariborz Jahanianf25adf12009-05-20 18:41:51 +0000105 llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
106 const ObjCIvarDecl *Ivar);
Chris Lattner9c445ae2009-06-15 01:09:11 +0000107 void EmitClassRef(const std::string &className);
Chris Lattnera0fd5ee2008-03-01 08:50:34 +0000108public:
Chris Lattner547907c2008-06-26 04:19:03 +0000109 CGObjCGNU(CodeGen::CodeGenModule &cgm);
Steve Naroff2c8a08e2009-03-31 16:53:37 +0000110 virtual llvm::Constant *GenerateConstantString(const ObjCStringLiteral *);
Daniel Dunbara04840b2008-08-23 03:46:30 +0000111 virtual CodeGen::RValue
112 GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbardd851282008-08-30 05:35:15 +0000113 QualType ResultType,
114 Selector Sel,
Daniel Dunbarb1ee5d62008-08-25 08:19:24 +0000115 llvm::Value *Receiver,
Daniel Dunbar0ed60b02008-08-30 03:02:31 +0000116 bool IsClassMessage,
Fariborz Jahanianc8007152009-05-05 21:36:57 +0000117 const CallArgList &CallArgs,
118 const ObjCMethodDecl *Method);
Daniel Dunbara04840b2008-08-23 03:46:30 +0000119 virtual CodeGen::RValue
120 GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbardd851282008-08-30 05:35:15 +0000121 QualType ResultType,
122 Selector Sel,
Daniel Dunbarb1ee5d62008-08-25 08:19:24 +0000123 const ObjCInterfaceDecl *Class,
Fariborz Jahanian17636fa2009-02-28 20:07:56 +0000124 bool isCategoryImpl,
Daniel Dunbarb1ee5d62008-08-25 08:19:24 +0000125 llvm::Value *Receiver,
Daniel Dunbar0ed60b02008-08-30 03:02:31 +0000126 bool IsClassMessage,
127 const CallArgList &CallArgs);
Daniel Dunbard916e6e2008-11-01 01:53:16 +0000128 virtual llvm::Value *GetClass(CGBuilderTy &Builder,
Daniel Dunbar434627a2008-08-16 00:25:02 +0000129 const ObjCInterfaceDecl *OID);
Daniel Dunbard916e6e2008-11-01 01:53:16 +0000130 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel);
Fariborz Jahanianc8007152009-05-05 21:36:57 +0000131 virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
132 *Method);
Chris Lattnerd71288e2008-06-26 04:37:12 +0000133
Fariborz Jahanian0adaa8a2009-01-10 21:06:09 +0000134 virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
135 const ObjCContainerDecl *CD);
Daniel Dunbarac93e472008-08-15 22:20:32 +0000136 virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
137 virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
Daniel Dunbard916e6e2008-11-01 01:53:16 +0000138 virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000139 const ObjCProtocolDecl *PD);
140 virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000141 virtual llvm::Function *ModuleInitFunction();
Daniel Dunbarf7103722008-09-24 03:38:44 +0000142 virtual llvm::Function *GetPropertyGetFunction();
143 virtual llvm::Function *GetPropertySetFunction();
Anders Carlsson58d16242008-08-31 04:05:03 +0000144 virtual llvm::Function *EnumerationMutationFunction();
Anders Carlssonb01a2112008-09-09 10:04:29 +0000145
Fariborz Jahanianfbeda7b2008-11-21 00:49:24 +0000146 virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
147 const Stmt &S);
Anders Carlssonb01a2112008-09-09 10:04:29 +0000148 virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
149 const ObjCAtThrowStmt &S);
Fariborz Jahanian252d87f2008-11-18 22:37:34 +0000150 virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian3305ad32008-11-18 21:45:40 +0000151 llvm::Value *AddrWeakObj);
Fariborz Jahanian252d87f2008-11-18 22:37:34 +0000152 virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
153 llvm::Value *src, llvm::Value *dst);
Fariborz Jahanian17958902008-11-19 00:59:10 +0000154 virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
155 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanianf310b592008-11-20 19:23:36 +0000156 virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
157 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanian17958902008-11-19 00:59:10 +0000158 virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
159 llvm::Value *src, llvm::Value *dest);
Fariborz Jahanianc912eb72009-02-03 19:03:09 +0000160 virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
161 QualType ObjectTy,
162 llvm::Value *BaseValue,
163 const ObjCIvarDecl *Ivar,
Fariborz Jahanianc912eb72009-02-03 19:03:09 +0000164 unsigned CVRQualifiers);
Fariborz Jahanian27cc6662009-02-10 19:02:04 +0000165 virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar61e14a62009-04-22 05:08:15 +0000166 const ObjCInterfaceDecl *Interface,
Fariborz Jahanian27cc6662009-02-10 19:02:04 +0000167 const ObjCIvarDecl *Ivar);
Chris Lattnera0fd5ee2008-03-01 08:50:34 +0000168};
169} // end anonymous namespace
170
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000171
Chris Lattner9c445ae2009-06-15 01:09:11 +0000172/// Emits a reference to a dummy variable which is emitted with each class.
173/// This ensures that a linker error will be generated when trying to link
174/// together modules where a referenced class is not defined.
175void CGObjCGNU::EmitClassRef(const std::string &className){
176 std::string symbolRef = "__objc_class_ref_" + className;
177 // Don't emit two copies of the same symbol
178 if (TheModule.getGlobalVariable(symbolRef)) return;
179 std::string symbolName = "__objc_class_name_" + className;
180 llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
181 if (!ClassSymbol) {
182 ClassSymbol = new llvm::GlobalVariable(LongTy, false,
183 llvm::GlobalValue::ExternalLinkage, 0, symbolName, &TheModule);
184 }
185 new llvm::GlobalVariable(ClassSymbol->getType(), true,
186 llvm::GlobalValue::CommonLinkage, ClassSymbol, symbolRef, &TheModule);
187}
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000188
189static std::string SymbolNameForClass(const std::string &ClassName) {
Chris Lattner9c445ae2009-06-15 01:09:11 +0000190 return "_OBJC_CLASS_" + ClassName;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000191}
192
193static std::string SymbolNameForMethod(const std::string &ClassName, const
194 std::string &CategoryName, const std::string &MethodName, bool isClassMethod)
195{
Chris Lattner9c445ae2009-06-15 01:09:11 +0000196 return "_OBJC_METHOD_" + ClassName + "("+CategoryName+")"+
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000197 (isClassMethod ? "+" : "-") + MethodName;
198}
199
Chris Lattner547907c2008-06-26 04:19:03 +0000200CGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000201 : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
202 MetaClassPtrAlias(0) {
Chris Lattner431dc382009-01-27 05:06:01 +0000203 IntTy = cast<llvm::IntegerType>(
204 CGM.getTypes().ConvertType(CGM.getContext().IntTy));
205 LongTy = cast<llvm::IntegerType>(
206 CGM.getTypes().ConvertType(CGM.getContext().LongTy));
Chris Lattner547907c2008-06-26 04:19:03 +0000207
Sebastian Redle9e21b32008-12-04 00:10:55 +0000208 Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000209 Zeros[1] = Zeros[0];
210 NULLPtr = llvm::ConstantPointerNull::get(
211 llvm::PointerType::getUnqual(llvm::Type::Int8Ty));
Chris Lattnerb326b172008-03-30 23:03:07 +0000212 // C string type. Used in lots of places.
213 PtrToInt8Ty =
214 llvm::PointerType::getUnqual(llvm::Type::Int8Ty);
215 // Get the selector Type.
Chris Lattner431dc382009-01-27 05:06:01 +0000216 SelectorTy = cast<llvm::PointerType>(
217 CGM.getTypes().ConvertType(CGM.getContext().getObjCSelType()));
218
Chris Lattnerb326b172008-03-30 23:03:07 +0000219 PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
220 PtrTy = PtrToInt8Ty;
221
222 // Object type
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000223 IdTy = cast<llvm::PointerType>(
224 CGM.getTypes().ConvertType(CGM.getContext().getObjCIdType()));
Chris Lattnerb326b172008-03-30 23:03:07 +0000225
226 // IMP type
227 std::vector<const llvm::Type*> IMPArgs;
228 IMPArgs.push_back(IdTy);
229 IMPArgs.push_back(SelectorTy);
230 IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000231}
232// This has to perform the lookup every time, since posing and related
233// techniques can modify the name -> class mapping.
Daniel Dunbard916e6e2008-11-01 01:53:16 +0000234llvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
Daniel Dunbar434627a2008-08-16 00:25:02 +0000235 const ObjCInterfaceDecl *OID) {
Chris Lattner271d4c22008-11-24 05:29:24 +0000236 llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
Chris Lattner9c445ae2009-06-15 01:09:11 +0000237 EmitClassRef(OID->getNameAsString());
Daniel Dunbar434627a2008-08-16 00:25:02 +0000238 ClassName = Builder.CreateStructGEP(ClassName, 0);
239
Fariborz Jahanian6158e692009-03-30 18:02:14 +0000240 std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000241 llvm::Constant *ClassLookupFn =
Fariborz Jahanian6158e692009-03-30 18:02:14 +0000242 CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
243 Params,
244 true),
245 "objc_lookup_class");
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000246 return Builder.CreateCall(ClassLookupFn, ClassName);
Chris Lattnerb326b172008-03-30 23:03:07 +0000247}
248
Daniel Dunbard916e6e2008-11-01 01:53:16 +0000249llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel) {
Chris Lattner3a8f2942008-11-24 03:33:13 +0000250 llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
Chris Lattnerd71288e2008-06-26 04:37:12 +0000251 if (US == 0)
252 US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
253 llvm::GlobalValue::InternalLinkage,
254 ".objc_untyped_selector_alias",
255 NULL, &TheModule);
256
257 return Builder.CreateLoad(US);
Fariborz Jahanianc8007152009-05-05 21:36:57 +0000258}
259
260llvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
261 *Method) {
262
263 std::string SelName = Method->getSelector().getAsString();
264 std::string SelTypes;
265 CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
266 // Typed selectors
267 TypedSelector Selector = TypedSelector(SelName,
268 SelTypes);
269
270 // If it's already cached, return it.
271 if (TypedSelectors[Selector])
272 {
273 return Builder.CreateLoad(TypedSelectors[Selector]);
274 }
275
276 // If it isn't, cache it.
277 llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
278 llvm::PointerType::getUnqual(SelectorTy),
279 llvm::GlobalValue::InternalLinkage, SelName,
280 NULL, &TheModule);
281 TypedSelectors[Selector] = Sel;
282
283 return Builder.CreateLoad(Sel);
Chris Lattnerd71288e2008-06-26 04:37:12 +0000284}
285
Chris Lattnera5b18882008-06-26 04:44:19 +0000286llvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
287 const std::string &Name) {
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000288 llvm::Constant * ConstStr = llvm::ConstantArray::get(Str);
289 ConstStr = new llvm::GlobalVariable(ConstStr->getType(), true,
290 llvm::GlobalValue::InternalLinkage,
291 ConstStr, Name, &TheModule);
292 return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
293}
294llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
295 std::vector<llvm::Constant*> &V, const std::string &Name) {
296 llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
297 return new llvm::GlobalVariable(Ty, false,
298 llvm::GlobalValue::InternalLinkage, C, Name, &TheModule);
299}
300llvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
301 std::vector<llvm::Constant*> &V, const std::string &Name) {
302 llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
303 return new llvm::GlobalVariable(Ty, false,
304 llvm::GlobalValue::InternalLinkage, C, Name, &TheModule);
305}
306
307/// Generate an NSConstantString object.
308//TODO: In case there are any crazy people still using the GNU runtime without
309//an OpenStep implementation, this should let them select their own class for
310//constant strings.
Steve Naroff2c8a08e2009-03-31 16:53:37 +0000311llvm::Constant *CGObjCGNU::GenerateConstantString(const ObjCStringLiteral *SL) {
312 std::string Str(SL->getString()->getStrData(),
313 SL->getString()->getByteLength());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000314 std::vector<llvm::Constant*> Ivars;
315 Ivars.push_back(NULLPtr);
Chris Lattnerbac12452008-06-21 21:44:18 +0000316 Ivars.push_back(MakeConstantString(Str));
Daniel Dunbardaf4ad42008-08-12 00:12:39 +0000317 Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000318 llvm::Constant *ObjCStr = MakeGlobal(
319 llvm::StructType::get(PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
320 Ivars, ".objc_str");
321 ConstantStrings.push_back(
322 llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty));
323 return ObjCStr;
324}
325
326///Generates a message send where the super is the receiver. This is a message
327///send to self with special delivery semantics indicating which class's method
328///should be called.
Daniel Dunbara04840b2008-08-23 03:46:30 +0000329CodeGen::RValue
330CGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
Daniel Dunbardd851282008-08-30 05:35:15 +0000331 QualType ResultType,
332 Selector Sel,
Daniel Dunbarb1ee5d62008-08-25 08:19:24 +0000333 const ObjCInterfaceDecl *Class,
Fariborz Jahanian17636fa2009-02-28 20:07:56 +0000334 bool isCategoryImpl,
Daniel Dunbarb1ee5d62008-08-25 08:19:24 +0000335 llvm::Value *Receiver,
Daniel Dunbar0ed60b02008-08-30 03:02:31 +0000336 bool IsClassMessage,
337 const CallArgList &CallArgs) {
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000338 llvm::Value *cmd = GetSelector(CGF.Builder, Sel);
339
340 CallArgList ActualArgs;
341
342 ActualArgs.push_back(
343 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
344 CGF.getContext().getObjCIdType()));
345 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
346 CGF.getContext().getObjCSelType()));
347 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
348
349 CodeGenTypes &Types = CGM.getTypes();
350 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
351 const llvm::FunctionType *impType = Types.GetFunctionType(FnInfo, false);
352
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000353 llvm::Value *ReceiverClass = 0;
Chris Lattner3a3c2962009-05-08 15:39:58 +0000354 if (isCategoryImpl) {
355 llvm::Constant *classLookupFunction = 0;
356 std::vector<const llvm::Type*> Params;
357 Params.push_back(PtrTy);
358 if (IsClassMessage) {
359 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
360 IdTy, Params, true), "objc_get_meta_class");
361 } else {
362 classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
363 IdTy, Params, true), "objc_get_class");
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000364 }
Chris Lattner3a3c2962009-05-08 15:39:58 +0000365 ReceiverClass = CGF.Builder.CreateCall(classLookupFunction,
366 MakeConstantString(Class->getNameAsString()));
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000367 } else {
Chris Lattner3a3c2962009-05-08 15:39:58 +0000368 // Set up global aliases for the metaclass or class pointer if they do not
369 // already exist. These will are forward-references which will be set to
370 // pointers to the class and metaclass structure created for the runtime load
371 // function. To send a message to super, we look up the value of the
372 // super_class pointer from either the class or metaclass structure.
373 if (IsClassMessage) {
374 if (!MetaClassPtrAlias) {
375 MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
376 llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
377 Class->getNameAsString(), NULL, &TheModule);
378 }
379 ReceiverClass = MetaClassPtrAlias;
380 } else {
381 if (!ClassPtrAlias) {
382 ClassPtrAlias = new llvm::GlobalAlias(IdTy,
383 llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
384 Class->getNameAsString(), NULL, &TheModule);
385 }
386 ReceiverClass = ClassPtrAlias;
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000387 }
Chris Lattner7eaa9f02009-04-25 23:19:45 +0000388 }
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000389 // Cast the pointer to a simplified version of the class structure
390 ReceiverClass = CGF.Builder.CreateBitCast(ReceiverClass,
391 llvm::PointerType::getUnqual(llvm::StructType::get(IdTy, IdTy, NULL)));
392 // Get the superclass pointer
393 ReceiverClass = CGF.Builder.CreateStructGEP(ReceiverClass, 1);
394 // Load the superclass pointer
395 ReceiverClass = CGF.Builder.CreateLoad(ReceiverClass);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000396 // Construct the structure used to look up the IMP
397 llvm::StructType *ObjCSuperTy = llvm::StructType::get(Receiver->getType(),
398 IdTy, NULL);
Daniel Dunbara04840b2008-08-23 03:46:30 +0000399 llvm::Value *ObjCSuper = CGF.Builder.CreateAlloca(ObjCSuperTy);
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000400
Daniel Dunbara04840b2008-08-23 03:46:30 +0000401 CGF.Builder.CreateStore(Receiver, CGF.Builder.CreateStructGEP(ObjCSuper, 0));
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000402 CGF.Builder.CreateStore(ReceiverClass,
403 CGF.Builder.CreateStructGEP(ObjCSuper, 1));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000404
405 // Get the IMP
Fariborz Jahanian6158e692009-03-30 18:02:14 +0000406 std::vector<const llvm::Type*> Params;
407 Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
408 Params.push_back(SelectorTy);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000409 llvm::Constant *lookupFunction =
Fariborz Jahanian6158e692009-03-30 18:02:14 +0000410 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
411 llvm::PointerType::getUnqual(impType), Params, true),
412 "objc_msg_lookup_super");
413
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000414 llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
Daniel Dunbara04840b2008-08-23 03:46:30 +0000415 llvm::Value *imp = CGF.Builder.CreateCall(lookupFunction, lookupArgs,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000416 lookupArgs+2);
417
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000418 return CGF.EmitCall(FnInfo, imp, ActualArgs);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000419}
420
421/// Generate code for a message send expression.
Daniel Dunbara04840b2008-08-23 03:46:30 +0000422CodeGen::RValue
423CGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
Daniel Dunbardd851282008-08-30 05:35:15 +0000424 QualType ResultType,
425 Selector Sel,
Daniel Dunbarb1ee5d62008-08-25 08:19:24 +0000426 llvm::Value *Receiver,
Daniel Dunbar0ed60b02008-08-30 03:02:31 +0000427 bool IsClassMessage,
Fariborz Jahanianc8007152009-05-05 21:36:57 +0000428 const CallArgList &CallArgs,
429 const ObjCMethodDecl *Method) {
430 llvm::Value *cmd;
431 if (Method)
432 cmd = GetSelector(CGF.Builder, Method);
433 else
434 cmd = GetSelector(CGF.Builder, Sel);
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000435 CallArgList ActualArgs;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000436
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000437 ActualArgs.push_back(
Fariborz Jahanianfefd5e82009-05-22 20:17:16 +0000438 std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
439 CGF.getContext().getObjCIdType()));
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000440 ActualArgs.push_back(std::make_pair(RValue::get(cmd),
441 CGF.getContext().getObjCSelType()));
442 ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
443
444 CodeGenTypes &Types = CGM.getTypes();
445 const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
446 const llvm::FunctionType *impType = Types.GetFunctionType(FnInfo, false);
447
Fariborz Jahanianfefd5e82009-05-22 20:17:16 +0000448 llvm::Value *imp;
Fariborz Jahanian6158e692009-03-30 18:02:14 +0000449 std::vector<const llvm::Type*> Params;
450 Params.push_back(Receiver->getType());
451 Params.push_back(SelectorTy);
Fariborz Jahanianfefd5e82009-05-22 20:17:16 +0000452 // For sender-aware dispatch, we pass the sender as the third argument to a
453 // lookup function. When sending messages from C code, the sender is nil.
454 // objc_msg_lookup_sender(id receiver, SEL selector, id sender);
455 if (CGM.getContext().getLangOptions().ObjCSenderDispatch) {
456 llvm::Value *self;
Fariborz Jahanian6158e692009-03-30 18:02:14 +0000457
Fariborz Jahanianfefd5e82009-05-22 20:17:16 +0000458 if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
459 self = CGF.LoadObjCSelf();
460 } else {
461 self = llvm::ConstantPointerNull::get(IdTy);
462 }
463 Params.push_back(self->getType());
464 llvm::Constant *lookupFunction =
465 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
466 llvm::PointerType::getUnqual(impType), Params, true),
467 "objc_msg_lookup_sender");
468
469 imp = CGF.Builder.CreateCall3(lookupFunction, Receiver, cmd, self);
470 } else {
471 llvm::Constant *lookupFunction =
472 CGM.CreateRuntimeFunction(llvm::FunctionType::get(
473 llvm::PointerType::getUnqual(impType), Params, true),
474 "objc_msg_lookup");
475
476 imp = CGF.Builder.CreateCall2(lookupFunction, Receiver, cmd);
477 }
Chris Lattner7db5e942008-05-06 00:56:42 +0000478
Fariborz Jahanianbb284742009-02-04 20:31:19 +0000479 return CGF.EmitCall(FnInfo, imp, ActualArgs);
Chris Lattnera0fd5ee2008-03-01 08:50:34 +0000480}
481
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000482/// Generates a MethodList. Used in construction of a objc_class and
483/// objc_category structures.
484llvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
Chris Lattner578279d2008-06-26 05:08:00 +0000485 const std::string &CategoryName,
486 const llvm::SmallVectorImpl<Selector> &MethodSels,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000487 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
488 bool isClassMethodList) {
489 // Get the method structure type.
490 llvm::StructType *ObjCMethodTy = llvm::StructType::get(
491 PtrToInt8Ty, // Really a selector, but the runtime creates it us.
492 PtrToInt8Ty, // Method types
493 llvm::PointerType::getUnqual(IMPTy), //Method pointer
494 NULL);
495 std::vector<llvm::Constant*> Methods;
496 std::vector<llvm::Constant*> Elements;
497 for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
498 Elements.clear();
Fariborz Jahanianf3ab5642009-05-17 16:49:27 +0000499 if (llvm::Constant *Method =
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000500 TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
Chris Lattner3a8f2942008-11-24 03:33:13 +0000501 MethodSels[i].getAsString(),
Fariborz Jahanianf3ab5642009-05-17 16:49:27 +0000502 isClassMethodList))) {
503 llvm::Constant *C =
504 CGM.GetAddrOfConstantCString(MethodSels[i].getAsString());
505 Elements.push_back(llvm::ConstantExpr::getGetElementPtr(C, Zeros, 2));
506 Elements.push_back(
507 llvm::ConstantExpr::getGetElementPtr(MethodTypes[i], Zeros, 2));
508 Method = llvm::ConstantExpr::getBitCast(Method,
509 llvm::PointerType::getUnqual(IMPTy));
510 Elements.push_back(Method);
511 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
512 }
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000513 }
514
515 // Array of method structures
516 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
Fariborz Jahanianf3ab5642009-05-17 16:49:27 +0000517 Methods.size());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000518 llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
Chris Lattnerbcb3e862008-06-26 04:52:29 +0000519 Methods);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000520
521 // Structure containing list pointer, array and array count
522 llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
523 llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get();
524 llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
525 llvm::StructType *ObjCMethodListTy = llvm::StructType::get(NextPtrTy,
526 IntTy,
527 ObjCMethodArrayTy,
528 NULL);
529 // Refine next pointer type to concrete type
530 llvm::cast<llvm::OpaqueType>(
531 OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
532 ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
533
534 Methods.clear();
535 Methods.push_back(llvm::ConstantPointerNull::get(
536 llvm::PointerType::getUnqual(ObjCMethodListTy)));
537 Methods.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty,
538 MethodTypes.size()));
539 Methods.push_back(MethodArray);
540
541 // Create an instance of the structure
542 return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
543}
544
545/// Generates an IvarList. Used in construction of a objc_class.
546llvm::Constant *CGObjCGNU::GenerateIvarList(
547 const llvm::SmallVectorImpl<llvm::Constant *> &IvarNames,
548 const llvm::SmallVectorImpl<llvm::Constant *> &IvarTypes,
549 const llvm::SmallVectorImpl<llvm::Constant *> &IvarOffsets) {
550 // Get the method structure type.
551 llvm::StructType *ObjCIvarTy = llvm::StructType::get(
552 PtrToInt8Ty,
553 PtrToInt8Ty,
554 IntTy,
555 NULL);
556 std::vector<llvm::Constant*> Ivars;
557 std::vector<llvm::Constant*> Elements;
558 for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
559 Elements.clear();
560 Elements.push_back( llvm::ConstantExpr::getGetElementPtr(IvarNames[i],
561 Zeros, 2));
562 Elements.push_back( llvm::ConstantExpr::getGetElementPtr(IvarTypes[i],
563 Zeros, 2));
564 Elements.push_back(IvarOffsets[i]);
565 Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
566 }
567
568 // Array of method structures
569 llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
570 IvarNames.size());
571
572
573 Elements.clear();
Chris Lattner431dc382009-01-27 05:06:01 +0000574 Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000575 Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
576 // Structure containing array and array count
577 llvm::StructType *ObjCIvarListTy = llvm::StructType::get(IntTy,
578 ObjCIvarArrayTy,
579 NULL);
580
581 // Create an instance of the structure
582 return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
583}
584
585/// Generate a class structure
586llvm::Constant *CGObjCGNU::GenerateClassStructure(
587 llvm::Constant *MetaClass,
588 llvm::Constant *SuperClass,
589 unsigned info,
Chris Lattnerad9c3f32008-06-26 04:47:04 +0000590 const char *Name,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000591 llvm::Constant *Version,
592 llvm::Constant *InstanceSize,
593 llvm::Constant *IVars,
594 llvm::Constant *Methods,
595 llvm::Constant *Protocols) {
596 // Set up the class structure
597 // Note: Several of these are char*s when they should be ids. This is
598 // because the runtime performs this translation on load.
599 llvm::StructType *ClassTy = llvm::StructType::get(
600 PtrToInt8Ty, // class_pointer
601 PtrToInt8Ty, // super_class
602 PtrToInt8Ty, // name
603 LongTy, // version
604 LongTy, // info
605 LongTy, // instance_size
606 IVars->getType(), // ivars
607 Methods->getType(), // methods
608 // These are all filled in by the runtime, so we pretend
609 PtrTy, // dtable
610 PtrTy, // subclass_list
611 PtrTy, // sibling_class
612 PtrTy, // protocols
613 PtrTy, // gc_object_type
614 NULL);
615 llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
616 llvm::Constant *NullP =
Chris Lattner431dc382009-01-27 05:06:01 +0000617 llvm::ConstantPointerNull::get(PtrTy);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000618 // Fill in the structure
619 std::vector<llvm::Constant*> Elements;
620 Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
621 Elements.push_back(SuperClass);
Chris Lattnerad9c3f32008-06-26 04:47:04 +0000622 Elements.push_back(MakeConstantString(Name, ".class_name"));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000623 Elements.push_back(Zero);
624 Elements.push_back(llvm::ConstantInt::get(LongTy, info));
625 Elements.push_back(InstanceSize);
626 Elements.push_back(IVars);
627 Elements.push_back(Methods);
628 Elements.push_back(NullP);
629 Elements.push_back(NullP);
630 Elements.push_back(NullP);
631 Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
632 Elements.push_back(NullP);
633 // Create an instance of the structure
Chris Lattner4d018542008-07-21 06:31:05 +0000634 return MakeGlobal(ClassTy, Elements, SymbolNameForClass(Name));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000635}
636
637llvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
638 const llvm::SmallVectorImpl<llvm::Constant *> &MethodNames,
639 const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes) {
640 // Get the method structure type.
641 llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(
642 PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
643 PtrToInt8Ty,
644 NULL);
645 std::vector<llvm::Constant*> Methods;
646 std::vector<llvm::Constant*> Elements;
647 for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
648 Elements.clear();
649 Elements.push_back( llvm::ConstantExpr::getGetElementPtr(MethodNames[i],
650 Zeros, 2));
651 Elements.push_back(
652 llvm::ConstantExpr::getGetElementPtr(MethodTypes[i], Zeros, 2));
653 Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
654 }
655 llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
656 MethodNames.size());
657 llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy, Methods);
658 llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(
659 IntTy, ObjCMethodArrayTy, NULL);
660 Methods.clear();
661 Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
662 Methods.push_back(Array);
663 return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
664}
665// Create the protocol list structure used in classes, categories and so on
666llvm::Constant *CGObjCGNU::GenerateProtocolList(
667 const llvm::SmallVectorImpl<std::string> &Protocols) {
668 llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
669 Protocols.size());
670 llvm::StructType *ProtocolListTy = llvm::StructType::get(
671 PtrTy, //Should be a recurisve pointer, but it's always NULL here.
672 LongTy,//FIXME: Should be size_t
673 ProtocolArrayTy,
674 NULL);
675 std::vector<llvm::Constant*> Elements;
676 for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
677 iter != endIter ; iter++) {
Fariborz Jahanian926c3712009-03-31 18:27:22 +0000678 llvm::Constant *protocol = ExistingProtocols[*iter];
679 if (!protocol)
680 protocol = GenerateEmptyProtocol(*iter);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000681 llvm::Constant *Ptr =
Fariborz Jahanian926c3712009-03-31 18:27:22 +0000682 llvm::ConstantExpr::getBitCast(protocol, PtrToInt8Ty);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000683 Elements.push_back(Ptr);
684 }
685 llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
686 Elements);
687 Elements.clear();
688 Elements.push_back(NULLPtr);
Chris Lattner431dc382009-01-27 05:06:01 +0000689 Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000690 Elements.push_back(ProtocolArray);
691 return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
692}
693
Daniel Dunbard916e6e2008-11-01 01:53:16 +0000694llvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000695 const ObjCProtocolDecl *PD) {
Fariborz Jahanian926c3712009-03-31 18:27:22 +0000696 llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
697 const llvm::Type *T =
698 CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
699 return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
700}
701
702llvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
703 const std::string &ProtocolName) {
704 llvm::SmallVector<std::string, 0> EmptyStringVector;
705 llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
706
707 llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
708 llvm::Constant *InstanceMethodList =
709 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
710 llvm::Constant *ClassMethodList =
711 GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
712 // Protocols are objects containing lists of the methods implemented and
713 // protocols adopted.
714 llvm::StructType *ProtocolTy = llvm::StructType::get(IdTy,
715 PtrToInt8Ty,
716 ProtocolList->getType(),
717 InstanceMethodList->getType(),
718 ClassMethodList->getType(),
719 NULL);
720 std::vector<llvm::Constant*> Elements;
721 // The isa pointer must be set to a magic number so the runtime knows it's
722 // the correct layout.
723 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
724 llvm::ConstantInt::get(llvm::Type::Int32Ty, ProtocolVersion), IdTy));
725 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
726 Elements.push_back(ProtocolList);
727 Elements.push_back(InstanceMethodList);
728 Elements.push_back(ClassMethodList);
729 return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000730}
731
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000732void CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
733 ASTContext &Context = CGM.getContext();
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000734 std::string ProtocolName = PD->getNameAsString();
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000735 llvm::SmallVector<std::string, 16> Protocols;
736 for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
737 E = PD->protocol_end(); PI != E; ++PI)
Chris Lattner271d4c22008-11-24 05:29:24 +0000738 Protocols.push_back((*PI)->getNameAsString());
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000739 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
740 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregorc55b0b02009-04-09 21:40:53 +0000741 for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(Context),
742 E = PD->instmeth_end(Context); iter != E; iter++) {
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000743 std::string TypeStr;
744 Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
745 InstanceMethodNames.push_back(
Chris Lattner3a8f2942008-11-24 03:33:13 +0000746 CGM.GetAddrOfConstantCString((*iter)->getSelector().getAsString()));
Daniel Dunbar31fe9c32008-08-13 23:20:05 +0000747 InstanceMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000748 }
749 // Collect information about class methods:
750 llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
751 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregorc55b0b02009-04-09 21:40:53 +0000752 for (ObjCProtocolDecl::classmeth_iterator
753 iter = PD->classmeth_begin(Context),
754 endIter = PD->classmeth_end(Context) ; iter != endIter ; iter++) {
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000755 std::string TypeStr;
756 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
757 ClassMethodNames.push_back(
Chris Lattner3a8f2942008-11-24 03:33:13 +0000758 CGM.GetAddrOfConstantCString((*iter)->getSelector().getAsString()));
Daniel Dunbar31fe9c32008-08-13 23:20:05 +0000759 ClassMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
Daniel Dunbar84bb85f2008-08-13 00:59:25 +0000760 }
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000761
762 llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
763 llvm::Constant *InstanceMethodList =
764 GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
765 llvm::Constant *ClassMethodList =
766 GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
767 // Protocols are objects containing lists of the methods implemented and
768 // protocols adopted.
769 llvm::StructType *ProtocolTy = llvm::StructType::get(IdTy,
770 PtrToInt8Ty,
771 ProtocolList->getType(),
772 InstanceMethodList->getType(),
773 ClassMethodList->getType(),
774 NULL);
775 std::vector<llvm::Constant*> Elements;
776 // The isa pointer must be set to a magic number so the runtime knows it's
777 // the correct layout.
778 Elements.push_back(llvm::ConstantExpr::getIntToPtr(
779 llvm::ConstantInt::get(llvm::Type::Int32Ty, ProtocolVersion), IdTy));
780 Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
781 Elements.push_back(ProtocolList);
782 Elements.push_back(InstanceMethodList);
783 Elements.push_back(ClassMethodList);
784 ExistingProtocols[ProtocolName] =
785 llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
786 ".objc_protocol"), IdTy);
787}
788
Daniel Dunbarac93e472008-08-15 22:20:32 +0000789void CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000790 std::string ClassName = OCD->getClassInterface()->getNameAsString();
791 std::string CategoryName = OCD->getNameAsString();
Daniel Dunbarac93e472008-08-15 22:20:32 +0000792 // Collect information about instance methods
793 llvm::SmallVector<Selector, 16> InstanceMethodSels;
794 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregorcd19b572009-04-23 01:02:12 +0000795 for (ObjCCategoryImplDecl::instmeth_iterator
796 iter = OCD->instmeth_begin(CGM.getContext()),
797 endIter = OCD->instmeth_end(CGM.getContext());
798 iter != endIter ; iter++) {
Daniel Dunbarac93e472008-08-15 22:20:32 +0000799 InstanceMethodSels.push_back((*iter)->getSelector());
800 std::string TypeStr;
801 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
802 InstanceMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
803 }
804
805 // Collect information about class methods
806 llvm::SmallVector<Selector, 16> ClassMethodSels;
807 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregorcd19b572009-04-23 01:02:12 +0000808 for (ObjCCategoryImplDecl::classmeth_iterator
809 iter = OCD->classmeth_begin(CGM.getContext()),
810 endIter = OCD->classmeth_end(CGM.getContext());
811 iter != endIter ; iter++) {
Daniel Dunbarac93e472008-08-15 22:20:32 +0000812 ClassMethodSels.push_back((*iter)->getSelector());
813 std::string TypeStr;
814 CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
815 ClassMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
816 }
817
818 // Collect the names of referenced protocols
819 llvm::SmallVector<std::string, 16> Protocols;
820 const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
821 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
822 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
823 E = Protos.end(); I != E; ++I)
Chris Lattner271d4c22008-11-24 05:29:24 +0000824 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbarac93e472008-08-15 22:20:32 +0000825
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000826 std::vector<llvm::Constant*> Elements;
827 Elements.push_back(MakeConstantString(CategoryName));
828 Elements.push_back(MakeConstantString(ClassName));
829 // Instance method list
830 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattner578279d2008-06-26 05:08:00 +0000831 ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000832 false), PtrTy));
833 // Class method list
834 Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
Chris Lattner578279d2008-06-26 05:08:00 +0000835 ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000836 PtrTy));
837 // Protocol list
838 Elements.push_back(llvm::ConstantExpr::getBitCast(
839 GenerateProtocolList(Protocols), PtrTy));
840 Categories.push_back(llvm::ConstantExpr::getBitCast(
841 MakeGlobal(llvm::StructType::get(PtrToInt8Ty, PtrToInt8Ty, PtrTy,
842 PtrTy, PtrTy, NULL), Elements), PtrTy));
843}
Daniel Dunbarac93e472008-08-15 22:20:32 +0000844
845void CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
846 ASTContext &Context = CGM.getContext();
847
848 // Get the superclass name.
849 const ObjCInterfaceDecl * SuperClassDecl =
850 OID->getClassInterface()->getSuperClass();
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000851 std::string SuperClassName;
Chris Lattner9c445ae2009-06-15 01:09:11 +0000852 if (SuperClassDecl) {
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000853 SuperClassName = SuperClassDecl->getNameAsString();
Chris Lattner9c445ae2009-06-15 01:09:11 +0000854 EmitClassRef(SuperClassName);
855 }
Daniel Dunbarac93e472008-08-15 22:20:32 +0000856
857 // Get the class name
Chris Lattnerb63b5502009-04-01 02:00:48 +0000858 ObjCInterfaceDecl *ClassDecl =
859 const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000860 std::string ClassName = ClassDecl->getNameAsString();
Chris Lattner9c445ae2009-06-15 01:09:11 +0000861 // Emit the symbol that is used to generate linker errors if this class is
862 // referenced in other modules but not declared.
863 new llvm::GlobalVariable(LongTy, false, llvm::GlobalValue::ExternalLinkage,
864 llvm::ConstantInt::get(LongTy, 0), "__objc_class_name_" + ClassName,
865 &TheModule);
866
Daniel Dunbar07ddfa92009-05-03 10:46:44 +0000867 // Get the size of instances.
868 int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
Daniel Dunbarac93e472008-08-15 22:20:32 +0000869
870 // Collect information about instance variables.
871 llvm::SmallVector<llvm::Constant*, 16> IvarNames;
872 llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
873 llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
Fariborz Jahanianf25adf12009-05-20 18:41:51 +0000874
875 int superInstanceSize = !SuperClassDecl ? 0 :
876 Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
877 // For non-fragile ivars, set the instance size to 0 - {the size of just this
878 // class}. The runtime will then set this to the correct value on load.
879 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
880 instanceSize = 0 - (instanceSize - superInstanceSize);
881 }
Daniel Dunbarac93e472008-08-15 22:20:32 +0000882 for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
883 endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
884 // Store the name
Chris Lattner271d4c22008-11-24 05:29:24 +0000885 IvarNames.push_back(CGM.GetAddrOfConstantCString((*iter)
886 ->getNameAsString()));
Daniel Dunbarac93e472008-08-15 22:20:32 +0000887 // Get the type encoding for this ivar
888 std::string TypeStr;
Daniel Dunbarc9197cd2008-10-17 20:21:44 +0000889 Context.getObjCEncodingForType((*iter)->getType(), TypeStr);
Daniel Dunbarac93e472008-08-15 22:20:32 +0000890 IvarTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
891 // Get the offset
Fariborz Jahanianf25adf12009-05-20 18:41:51 +0000892 uint64_t Offset;
893 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
894 Offset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter) -
895 superInstanceSize;
896 ObjCIvarOffsetVariable(ClassDecl, *iter);
897 } else {
898 Offset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter);
899 }
Daniel Dunbarac93e472008-08-15 22:20:32 +0000900 IvarOffsets.push_back(
Daniel Dunbarbdeefcc2009-04-22 08:20:31 +0000901 llvm::ConstantInt::get(llvm::Type::Int32Ty, Offset));
Daniel Dunbarac93e472008-08-15 22:20:32 +0000902 }
903
904 // Collect information about instance methods
905 llvm::SmallVector<Selector, 16> InstanceMethodSels;
906 llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
Douglas Gregorcd19b572009-04-23 01:02:12 +0000907 for (ObjCImplementationDecl::instmeth_iterator
908 iter = OID->instmeth_begin(CGM.getContext()),
909 endIter = OID->instmeth_end(CGM.getContext());
910 iter != endIter ; iter++) {
Daniel Dunbarac93e472008-08-15 22:20:32 +0000911 InstanceMethodSels.push_back((*iter)->getSelector());
912 std::string TypeStr;
913 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
914 InstanceMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
915 }
Fariborz Jahanianf3ab5642009-05-17 16:49:27 +0000916 for (ObjCImplDecl::propimpl_iterator
917 iter = OID->propimpl_begin(CGM.getContext()),
918 endIter = OID->propimpl_end(CGM.getContext());
919 iter != endIter ; iter++) {
920 ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
921 if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
922 InstanceMethodSels.push_back(getter->getSelector());
923 std::string TypeStr;
924 Context.getObjCEncodingForMethodDecl(getter,TypeStr);
925 InstanceMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
926 }
927 if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
928 InstanceMethodSels.push_back(setter->getSelector());
929 std::string TypeStr;
930 Context.getObjCEncodingForMethodDecl(setter,TypeStr);
931 InstanceMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
932 }
933 }
Daniel Dunbarac93e472008-08-15 22:20:32 +0000934
935 // Collect information about class methods
936 llvm::SmallVector<Selector, 16> ClassMethodSels;
937 llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
Douglas Gregorcd19b572009-04-23 01:02:12 +0000938 for (ObjCImplementationDecl::classmeth_iterator
939 iter = OID->classmeth_begin(CGM.getContext()),
940 endIter = OID->classmeth_end(CGM.getContext());
941 iter != endIter ; iter++) {
Daniel Dunbarac93e472008-08-15 22:20:32 +0000942 ClassMethodSels.push_back((*iter)->getSelector());
943 std::string TypeStr;
944 Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
945 ClassMethodTypes.push_back(CGM.GetAddrOfConstantCString(TypeStr));
946 }
947 // Collect the names of referenced protocols
948 llvm::SmallVector<std::string, 16> Protocols;
949 const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
950 for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
951 E = Protos.end(); I != E; ++I)
Chris Lattner271d4c22008-11-24 05:29:24 +0000952 Protocols.push_back((*I)->getNameAsString());
Daniel Dunbarac93e472008-08-15 22:20:32 +0000953
954
955
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000956 // Get the superclass pointer.
957 llvm::Constant *SuperClass;
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000958 if (!SuperClassName.empty()) {
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000959 SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
960 } else {
Chris Lattner431dc382009-01-27 05:06:01 +0000961 SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000962 }
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000963 // Empty vector used to construct empty method lists
964 llvm::SmallVector<llvm::Constant*, 1> empty;
965 // Generate the method and instance variable lists
966 llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
Chris Lattner578279d2008-06-26 05:08:00 +0000967 InstanceMethodSels, InstanceMethodTypes, false);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000968 llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
Chris Lattner578279d2008-06-26 05:08:00 +0000969 ClassMethodSels, ClassMethodTypes, true);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000970 llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
971 IvarOffsets);
972 //Generate metaclass for class methods
973 llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
Chris Lattner4d018542008-07-21 06:31:05 +0000974 NULLPtr, 0x2L, /*name*/"", 0, Zeros[0], GenerateIvarList(
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000975 empty, empty, empty), ClassMethodList, NULLPtr);
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000976
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000977 // Generate the class structure
Chris Lattnerd120b9e2008-11-24 03:54:41 +0000978 llvm::Constant *ClassStruct =
979 GenerateClassStructure(MetaClassStruct, SuperClass, 0x1L,
980 ClassName.c_str(), 0,
Sebastian Redle9e21b32008-12-04 00:10:55 +0000981 llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000982 MethodList, GenerateProtocolList(Protocols));
Daniel Dunbarc4532f52009-05-04 15:31:17 +0000983
984 // Resolve the class aliases, if they exist.
985 if (ClassPtrAlias) {
986 ClassPtrAlias->setAliasee(
987 llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
988 ClassPtrAlias = 0;
989 }
990 if (MetaClassPtrAlias) {
991 MetaClassPtrAlias->setAliasee(
992 llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
993 MetaClassPtrAlias = 0;
994 }
995
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +0000996 // Add class structure to list to be added to the symtab later
997 ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
998 Classes.push_back(ClassStruct);
999}
1000
1001llvm::Function *CGObjCGNU::ModuleInitFunction() {
1002 // Only emit an ObjC load function if no Objective-C stuff has been called
1003 if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
1004 ExistingProtocols.empty() && TypedSelectors.empty() &&
Anton Korobeynikovb41e4972008-06-01 15:14:46 +00001005 UntypedSelectors.empty())
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001006 return NULL;
Eli Friedman7719bc82008-06-01 16:00:02 +00001007
Chris Lattner431dc382009-01-27 05:06:01 +00001008 const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1009 SelectorTy->getElementType());
1010 const llvm::Type *SelStructPtrTy = SelectorTy;
1011 bool isSelOpaque = false;
1012 if (SelStructTy == 0) {
1013 SelStructTy = llvm::StructType::get(PtrToInt8Ty, PtrToInt8Ty, NULL);
1014 SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
1015 isSelOpaque = true;
1016 }
1017
Eli Friedman7719bc82008-06-01 16:00:02 +00001018 // Name the ObjC types to make the IR a bit easier to read
Chris Lattner431dc382009-01-27 05:06:01 +00001019 TheModule.addTypeName(".objc_selector", SelStructPtrTy);
Eli Friedman7719bc82008-06-01 16:00:02 +00001020 TheModule.addTypeName(".objc_id", IdTy);
1021 TheModule.addTypeName(".objc_imp", IMPTy);
1022
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001023 std::vector<llvm::Constant*> Elements;
Chris Lattner7eaa9f02009-04-25 23:19:45 +00001024 llvm::Constant *Statics = NULLPtr;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001025 // Generate statics list:
Chris Lattner7eaa9f02009-04-25 23:19:45 +00001026 if (ConstantStrings.size()) {
1027 llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
1028 ConstantStrings.size() + 1);
1029 ConstantStrings.push_back(NULLPtr);
1030 Elements.push_back(MakeConstantString("NSConstantString",
1031 ".objc_static_class_name"));
1032 Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
1033 ConstantStrings));
1034 llvm::StructType *StaticsListTy =
1035 llvm::StructType::get(PtrToInt8Ty, StaticsArrayTy, NULL);
1036 llvm::Type *StaticsListPtrTy = llvm::PointerType::getUnqual(StaticsListTy);
1037 Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
1038 llvm::ArrayType *StaticsListArrayTy =
1039 llvm::ArrayType::get(StaticsListPtrTy, 2);
1040 Elements.clear();
1041 Elements.push_back(Statics);
1042 Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
1043 Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
1044 Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
1045 }
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001046 // Array of classes, categories, and constant objects
1047 llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
1048 Classes.size() + Categories.size() + 2);
Chris Lattner431dc382009-01-27 05:06:01 +00001049 llvm::StructType *SymTabTy = llvm::StructType::get(LongTy, SelStructPtrTy,
Chris Lattneref843042008-06-26 04:10:42 +00001050 llvm::Type::Int16Ty,
1051 llvm::Type::Int16Ty,
1052 ClassListTy, NULL);
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001053
1054 Elements.clear();
1055 // Pointer to an array of selectors used in this module.
1056 std::vector<llvm::Constant*> Selectors;
1057 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1058 iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
1059 iter != iterEnd ; ++iter) {
Chris Lattneref843042008-06-26 04:10:42 +00001060 Elements.push_back(MakeConstantString(iter->first.first, ".objc_sel_name"));
1061 Elements.push_back(MakeConstantString(iter->first.second,
1062 ".objc_sel_types"));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001063 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
1064 Elements.clear();
1065 }
1066 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1067 iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
Chris Lattneref843042008-06-26 04:10:42 +00001068 iter != iterEnd; ++iter) {
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001069 Elements.push_back(
Chris Lattneref843042008-06-26 04:10:42 +00001070 MakeConstantString(iter->getKeyData(), ".objc_sel_name"));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001071 Elements.push_back(NULLPtr);
1072 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
1073 Elements.clear();
1074 }
1075 Elements.push_back(NULLPtr);
1076 Elements.push_back(NULLPtr);
1077 Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
1078 Elements.clear();
1079 // Number of static selectors
1080 Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
1081 llvm::Constant *SelectorList = MakeGlobal(
1082 llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
1083 ".objc_selector_list");
Chris Lattner431dc382009-01-27 05:06:01 +00001084 Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
1085 SelStructPtrTy));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001086
1087 // Now that all of the static selectors exist, create pointers to them.
1088 int index = 0;
1089 for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
1090 iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
1091 iter != iterEnd; ++iter) {
1092 llvm::Constant *Idxs[] = {Zeros[0],
1093 llvm::ConstantInt::get(llvm::Type::Int32Ty, index++), Zeros[0]};
Chris Lattner431dc382009-01-27 05:06:01 +00001094 llvm::Constant *SelPtr = new llvm::GlobalVariable(SelStructPtrTy,
1095 true, llvm::GlobalValue::InternalLinkage,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001096 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1097 ".objc_sel_ptr", &TheModule);
Chris Lattner431dc382009-01-27 05:06:01 +00001098 // If selectors are defined as an opaque type, cast the pointer to this
1099 // type.
1100 if (isSelOpaque) {
1101 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1102 llvm::PointerType::getUnqual(SelectorTy));
1103 }
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001104 (*iter).second->setAliasee(SelPtr);
1105 }
1106 for (llvm::StringMap<llvm::GlobalAlias*>::iterator
1107 iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1108 iter != iterEnd; iter++) {
1109 llvm::Constant *Idxs[] = {Zeros[0],
1110 llvm::ConstantInt::get(llvm::Type::Int32Ty, index++), Zeros[0]};
Chris Lattner431dc382009-01-27 05:06:01 +00001111 llvm::Constant *SelPtr = new llvm::GlobalVariable(SelStructPtrTy, true,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001112 llvm::GlobalValue::InternalLinkage,
1113 llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
1114 ".objc_sel_ptr", &TheModule);
Chris Lattner431dc382009-01-27 05:06:01 +00001115 // If selectors are defined as an opaque type, cast the pointer to this
1116 // type.
1117 if (isSelOpaque) {
1118 SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
1119 llvm::PointerType::getUnqual(SelectorTy));
1120 }
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001121 (*iter).second->setAliasee(SelPtr);
1122 }
1123 // Number of classes defined.
1124 Elements.push_back(llvm::ConstantInt::get(llvm::Type::Int16Ty,
1125 Classes.size()));
1126 // Number of categories defined
1127 Elements.push_back(llvm::ConstantInt::get(llvm::Type::Int16Ty,
1128 Categories.size()));
1129 // Create an array of classes, then categories, then static object instances
1130 Classes.insert(Classes.end(), Categories.begin(), Categories.end());
1131 // NULL-terminated list of static object instances (mainly constant strings)
1132 Classes.push_back(Statics);
1133 Classes.push_back(NULLPtr);
1134 llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
1135 Elements.push_back(ClassList);
1136 // Construct the symbol table
1137 llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
1138
1139 // The symbol table is contained in a module which has some version-checking
1140 // constants
1141 llvm::StructType * ModuleTy = llvm::StructType::get(LongTy, LongTy,
1142 PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
1143 Elements.clear();
1144 // Runtime version used for compatibility checking.
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001145 if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
1146 Elements.push_back(llvm::ConstantInt::get(LongTy,
1147 NonFragileRuntimeVersion));
1148 } else {
1149 Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
1150 }
Fariborz Jahaniandefa06a2009-04-01 19:49:42 +00001151 // sizeof(ModuleTy)
1152 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
1153 Elements.push_back(llvm::ConstantInt::get(LongTy, td.getTypeSizeInBits(ModuleTy)/8));
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001154 //FIXME: Should be the path to the file where this module was declared
1155 Elements.push_back(NULLPtr);
1156 Elements.push_back(SymTab);
1157 llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
1158
1159 // Create the load function calling the runtime entry point with the module
1160 // structure
1161 std::vector<const llvm::Type*> VoidArgs;
1162 llvm::Function * LoadFunction = llvm::Function::Create(
1163 llvm::FunctionType::get(llvm::Type::VoidTy, VoidArgs, false),
1164 llvm::GlobalValue::InternalLinkage, ".objc_load_function",
1165 &TheModule);
1166 llvm::BasicBlock *EntryBB = llvm::BasicBlock::Create("entry", LoadFunction);
Daniel Dunbard916e6e2008-11-01 01:53:16 +00001167 CGBuilderTy Builder;
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001168 Builder.SetInsertPoint(EntryBB);
Fariborz Jahanian6158e692009-03-30 18:02:14 +00001169
1170 std::vector<const llvm::Type*> Params(1,
1171 llvm::PointerType::getUnqual(ModuleTy));
1172 llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
1173 llvm::Type::VoidTy, Params, true), "__objc_exec_class");
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001174 Builder.CreateCall(Register, Module);
1175 Builder.CreateRetVoid();
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001176
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001177 return LoadFunction;
1178}
Daniel Dunbarac93e472008-08-15 22:20:32 +00001179
Fariborz Jahanian0adaa8a2009-01-10 21:06:09 +00001180llvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
1181 const ObjCContainerDecl *CD) {
Daniel Dunbarac93e472008-08-15 22:20:32 +00001182 const ObjCCategoryImplDecl *OCD =
Steve Naroff438be772009-01-08 19:41:02 +00001183 dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
Chris Lattner3a8f2942008-11-24 03:33:13 +00001184 std::string CategoryName = OCD ? OCD->getNameAsString() : "";
1185 std::string ClassName = OMD->getClassInterface()->getNameAsString();
1186 std::string MethodName = OMD->getSelector().getAsString();
Douglas Gregor5d764842009-01-09 17:18:27 +00001187 bool isClassMethod = !OMD->isInstanceMethod();
Daniel Dunbarac93e472008-08-15 22:20:32 +00001188
Daniel Dunbar34bda882009-02-02 23:23:47 +00001189 CodeGenTypes &Types = CGM.getTypes();
Daniel Dunbar3ad1f072008-09-10 04:01:49 +00001190 const llvm::FunctionType *MethodTy =
Daniel Dunbar34bda882009-02-02 23:23:47 +00001191 Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001192 std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
1193 MethodName, isClassMethod);
1194
Gabor Greif815e2c12008-04-06 20:42:52 +00001195 llvm::Function *Method = llvm::Function::Create(MethodTy,
Chris Lattnerb326b172008-03-30 23:03:07 +00001196 llvm::GlobalValue::InternalLinkage,
Anton Korobeynikovcd5d08d2008-06-01 14:13:53 +00001197 FunctionName,
Chris Lattnerb326b172008-03-30 23:03:07 +00001198 &TheModule);
Chris Lattnerb326b172008-03-30 23:03:07 +00001199 return Method;
1200}
1201
Daniel Dunbarf7103722008-09-24 03:38:44 +00001202llvm::Function *CGObjCGNU::GetPropertyGetFunction() {
Fariborz Jahaniande629b62009-05-19 00:28:43 +00001203 std::vector<const llvm::Type*> Params;
1204 const llvm::Type *BoolTy =
1205 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1206 Params.push_back(IdTy);
1207 Params.push_back(SelectorTy);
1208 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1209 Params.push_back(LongTy);
1210 Params.push_back(BoolTy);
1211 // void objc_getProperty (id, SEL, ptrdiff_t, bool)
1212 const llvm::FunctionType *FTy =
1213 llvm::FunctionType::get(IdTy, Params, false);
1214 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1215 "objc_getProperty"));
Daniel Dunbarf7103722008-09-24 03:38:44 +00001216}
1217
1218llvm::Function *CGObjCGNU::GetPropertySetFunction() {
Fariborz Jahaniande629b62009-05-19 00:28:43 +00001219 std::vector<const llvm::Type*> Params;
1220 const llvm::Type *BoolTy =
1221 CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
1222 Params.push_back(IdTy);
1223 Params.push_back(SelectorTy);
1224 // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
1225 Params.push_back(LongTy);
1226 Params.push_back(IdTy);
1227 Params.push_back(BoolTy);
1228 Params.push_back(BoolTy);
1229 // void objc_setProperty (id, SEL, ptrdiff_t, id, bool, bool)
1230 const llvm::FunctionType *FTy =
1231 llvm::FunctionType::get(llvm::Type::VoidTy, Params, false);
1232 return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
1233 "objc_setProperty"));
Daniel Dunbarf7103722008-09-24 03:38:44 +00001234}
1235
1236llvm::Function *CGObjCGNU::EnumerationMutationFunction() {
Fariborz Jahanian6158e692009-03-30 18:02:14 +00001237 std::vector<const llvm::Type*> Params(1, IdTy);
1238 return cast<llvm::Function>(CGM.CreateRuntimeFunction(
1239 llvm::FunctionType::get(llvm::Type::VoidTy, Params, true),
1240 "objc_enumerationMutation"));
Anders Carlsson58d16242008-08-31 04:05:03 +00001241}
1242
Fariborz Jahanianfbeda7b2008-11-21 00:49:24 +00001243void CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1244 const Stmt &S) {
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001245 // Pointer to the personality function
1246 llvm::Constant *Personality =
1247 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::Int32Ty,
1248 std::vector<const llvm::Type*>(), true),
1249 "__gnu_objc_personality_v0");
1250 Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
1251 std::vector<const llvm::Type*> Params;
1252 Params.push_back(PtrTy);
1253 llvm::Value *RethrowFn =
1254 CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::VoidTy,
1255 Params, false), "_Unwind_Resume_or_Rethrow");
1256
1257 bool isTry = isa<ObjCAtTryStmt>(S);
1258 llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
1259 llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
1260 llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
Chris Lattner5b56c192009-05-11 18:16:28 +00001261 llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001262 llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
1263 llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
1264 llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
1265
1266 // GNU runtime does not currently support @synchronized()
1267 if (!isTry) {
Chris Lattner5b56c192009-05-11 18:16:28 +00001268 std::vector<const llvm::Type*> Args(1, IdTy);
1269 llvm::FunctionType *FTy =
1270 llvm::FunctionType::get(llvm::Type::VoidTy, Args, false);
1271 llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
1272 llvm::Value *SyncArg =
1273 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1274 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1275 CGF.Builder.CreateCall(SyncEnter, SyncArg);
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001276 }
1277
Chris Lattner5b56c192009-05-11 18:16:28 +00001278
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001279 // Push an EH context entry, used for handling rethrows and jumps
1280 // through finally.
1281 CGF.PushCleanupBlock(FinallyBlock);
1282
1283 // Emit the statements in the @try {} block
1284 CGF.setInvokeDest(TryHandler);
1285
1286 CGF.EmitBlock(TryBlock);
1287 CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
1288 : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
1289
1290 // Jump to @finally if there is no exception
1291 CGF.EmitBranchThroughCleanup(FinallyEnd);
1292
1293 // Emit the handlers
1294 CGF.EmitBlock(TryHandler);
1295
Chris Lattner13395b22009-05-08 17:36:08 +00001296 // Get the correct versions of the exception handling intrinsics
1297 llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
1298 int PointerWidth = td.getTypeSizeInBits(PtrTy);
1299 assert((PointerWidth == 32 || PointerWidth == 64) &&
1300 "Can't yet handle exceptions if pointers are not 32 or 64 bits");
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001301 llvm::Value *llvm_eh_exception =
1302 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
Chris Lattner13395b22009-05-08 17:36:08 +00001303 llvm::Value *llvm_eh_selector = PointerWidth == 32 ?
1304 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector_i32) :
1305 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector_i64);
1306 llvm::Value *llvm_eh_typeid_for = PointerWidth == 32 ?
1307 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for_i32) :
1308 CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for_i64);
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001309
1310 // Exception object
1311 llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1312 llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
1313
1314 llvm::SmallVector<llvm::Value*, 8> ESelArgs;
1315 llvm::SmallVector<std::pair<const ParmVarDecl*, const Stmt*>, 8> Handlers;
1316
1317 ESelArgs.push_back(Exc);
1318 ESelArgs.push_back(Personality);
1319
1320 bool HasCatchAll = false;
1321 // Only @try blocks are allowed @catch blocks, but both can have @finally
1322 if (isTry) {
1323 if (const ObjCAtCatchStmt* CatchStmt =
1324 cast<ObjCAtTryStmt>(S).getCatchStmts()) {
Chris Lattner5b56c192009-05-11 18:16:28 +00001325 CGF.setInvokeDest(CatchInCatch);
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001326
1327 for (; CatchStmt; CatchStmt = CatchStmt->getNextCatchStmt()) {
1328 const ParmVarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
1329 Handlers.push_back(std::make_pair(CatchDecl, CatchStmt->getCatchBody()));
1330
1331 // @catch() and @catch(id) both catch any ObjC exception
1332 if (!CatchDecl || CGF.getContext().isObjCIdType(CatchDecl->getType())
1333 || CatchDecl->getType()->isObjCQualifiedIdType()) {
1334 // Use i8* null here to signal this is a catch all, not a cleanup.
1335 ESelArgs.push_back(NULLPtr);
1336 HasCatchAll = true;
1337 // No further catches after this one will ever by reached
1338 break;
1339 }
1340
1341 // All other types should be Objective-C interface pointer types.
1342 const PointerType *PT = CatchDecl->getType()->getAsPointerType();
1343 assert(PT && "Invalid @catch type.");
1344 const ObjCInterfaceType *IT =
1345 PT->getPointeeType()->getAsObjCInterfaceType();
1346 assert(IT && "Invalid @catch type.");
Chris Lattner13395b22009-05-08 17:36:08 +00001347 llvm::Value *EHType =
1348 MakeConstantString(IT->getDecl()->getNameAsString());
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001349 ESelArgs.push_back(EHType);
1350 }
1351 }
1352 }
1353
1354 // We use a cleanup unless there was already a catch all.
1355 if (!HasCatchAll) {
1356 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, 0));
1357 Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
1358 }
1359
1360 // Find which handler was matched.
Chris Lattner13395b22009-05-08 17:36:08 +00001361 llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001362 ESelArgs.begin(), ESelArgs.end(), "selector");
1363
1364 for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
1365 const ParmVarDecl *CatchParam = Handlers[i].first;
1366 const Stmt *CatchBody = Handlers[i].second;
1367
1368 llvm::BasicBlock *Next = 0;
1369
1370 // The last handler always matches.
1371 if (i + 1 != e) {
1372 assert(CatchParam && "Only last handler can be a catch all.");
1373
1374 // Test whether this block matches the type for the selector and branch
1375 // to Match if it does, or to the next BB if it doesn't.
1376 llvm::BasicBlock *Match = CGF.createBasicBlock("match");
1377 Next = CGF.createBasicBlock("catch.next");
Chris Lattner13395b22009-05-08 17:36:08 +00001378 llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001379 CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
1380 CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
1381 Next);
1382
1383 CGF.EmitBlock(Match);
1384 }
1385
1386 if (CatchBody) {
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001387 llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
1388 CGF.ConvertType(CatchParam->getType()));
1389
1390 // Bind the catch parameter if it exists.
1391 if (CatchParam) {
1392 // CatchParam is a ParmVarDecl because of the grammar
1393 // construction used to handle this, but for codegen purposes
1394 // we treat this as a local decl.
1395 CGF.EmitLocalBlockVarDecl(*CatchParam);
1396 CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
1397 }
1398
1399 CGF.ObjCEHValueStack.push_back(ExcObject);
1400 CGF.EmitStmt(CatchBody);
1401 CGF.ObjCEHValueStack.pop_back();
1402
1403 CGF.EmitBranchThroughCleanup(FinallyEnd);
1404
1405 if (Next)
1406 CGF.EmitBlock(Next);
1407 } else {
1408 assert(!Next && "catchup should be last handler.");
1409
1410 CGF.Builder.CreateStore(Exc, RethrowPtr);
1411 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1412 }
1413 }
Chris Lattner5b56c192009-05-11 18:16:28 +00001414 // The @finally block is a secondary landing pad for any exceptions thrown in
1415 // @catch() blocks
1416 CGF.EmitBlock(CatchInCatch);
1417 Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1418 ESelArgs.clear();
1419 ESelArgs.push_back(Exc);
1420 ESelArgs.push_back(Personality);
1421 ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::Int32Ty, 0));
1422 CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
1423 "selector");
1424 CGF.Builder.CreateCall(llvm_eh_typeid_for,
1425 CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
1426 CGF.Builder.CreateStore(Exc, RethrowPtr);
1427 CGF.EmitBranchThroughCleanup(FinallyRethrow);
1428
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001429 CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
1430
1431 CGF.setInvokeDest(PrevLandingPad);
1432
1433 CGF.EmitBlock(FinallyBlock);
1434
Chris Lattner5b56c192009-05-11 18:16:28 +00001435
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001436 if (isTry) {
1437 if (const ObjCAtFinallyStmt* FinallyStmt =
1438 cast<ObjCAtTryStmt>(S).getFinallyStmt())
1439 CGF.EmitStmt(FinallyStmt->getFinallyBody());
1440 } else {
Chris Lattner5b56c192009-05-11 18:16:28 +00001441 // Emit 'objc_sync_exit(expr)' as finally's sole statement for
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001442 // @synchronized.
Chris Lattner5b56c192009-05-11 18:16:28 +00001443 std::vector<const llvm::Type*> Args(1, IdTy);
1444 llvm::FunctionType *FTy =
1445 llvm::FunctionType::get(llvm::Type::VoidTy, Args, false);
1446 llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
1447 llvm::Value *SyncArg =
1448 CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1449 SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1450 CGF.Builder.CreateCall(SyncExit, SyncArg);
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001451 }
1452
1453 if (Info.SwitchBlock)
1454 CGF.EmitBlock(Info.SwitchBlock);
1455 if (Info.EndBlock)
1456 CGF.EmitBlock(Info.EndBlock);
1457
1458 // Branch around the rethrow code.
1459 CGF.EmitBranch(FinallyEnd);
1460
1461 CGF.EmitBlock(FinallyRethrow);
1462 CGF.Builder.CreateCall(RethrowFn, CGF.Builder.CreateLoad(RethrowPtr));
1463 CGF.Builder.CreateUnreachable();
1464
1465 CGF.EmitBlock(FinallyEnd);
1466
Anders Carlssonb01a2112008-09-09 10:04:29 +00001467}
1468
1469void CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
Daniel Dunbarf7103722008-09-24 03:38:44 +00001470 const ObjCAtThrowStmt &S) {
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001471 llvm::Value *ExceptionAsObject;
1472
1473 std::vector<const llvm::Type*> Args(1, IdTy);
1474 llvm::FunctionType *FTy =
1475 llvm::FunctionType::get(llvm::Type::VoidTy, Args, false);
1476 llvm::Value *ThrowFn =
1477 CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
1478
1479 if (const Expr *ThrowExpr = S.getThrowExpr()) {
1480 llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
Fariborz Jahanianf3ab5642009-05-17 16:49:27 +00001481 ExceptionAsObject = Exception;
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001482 } else {
1483 assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
1484 "Unexpected rethrow outside @catch block.");
1485 ExceptionAsObject = CGF.ObjCEHValueStack.back();
1486 }
Fariborz Jahanianf3ab5642009-05-17 16:49:27 +00001487 ExceptionAsObject =
1488 CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
Chris Lattnerfee3eb02009-05-08 00:11:50 +00001489
1490 // Note: This may have to be an invoke, if we want to support constructs like:
1491 // @try {
1492 // @throw(obj);
1493 // }
1494 // @catch(id) ...
1495 //
1496 // This is effectively turning @throw into an incredibly-expensive goto, but
1497 // it may happen as a result of inlining followed by missed optimizations, or
1498 // as a result of stupidity.
1499 llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
1500 if (!UnwindBB) {
1501 CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
1502 CGF.Builder.CreateUnreachable();
1503 } else {
1504 CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
1505 &ExceptionAsObject+1);
1506 }
1507 // Clear the insertion point to indicate we are in unreachable code.
1508 CGF.Builder.ClearInsertionPoint();
Anders Carlssonb01a2112008-09-09 10:04:29 +00001509}
1510
Fariborz Jahanian252d87f2008-11-18 22:37:34 +00001511llvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
Fariborz Jahanian3305ad32008-11-18 21:45:40 +00001512 llvm::Value *AddrWeakObj)
1513{
1514 return 0;
1515}
1516
Fariborz Jahanian252d87f2008-11-18 22:37:34 +00001517void CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
1518 llvm::Value *src, llvm::Value *dst)
1519{
1520 return;
1521}
1522
Fariborz Jahanian17958902008-11-19 00:59:10 +00001523void CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
1524 llvm::Value *src, llvm::Value *dst)
1525{
1526 return;
1527}
1528
Fariborz Jahanianf310b592008-11-20 19:23:36 +00001529void CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
1530 llvm::Value *src, llvm::Value *dst)
1531{
1532 return;
1533}
1534
Fariborz Jahanian17958902008-11-19 00:59:10 +00001535void CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
1536 llvm::Value *src, llvm::Value *dst)
1537{
1538 return;
1539}
1540
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001541llvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
1542 const ObjCInterfaceDecl *ID,
1543 const ObjCIvarDecl *Ivar) {
1544 const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
1545 + '.' + Ivar->getNameAsString();
1546 // Emit the variable and initialize it with what we think the correct value
1547 // is. This allows code compiled with non-fragile ivars to work correctly
1548 // when linked against code which isn't (most of the time).
1549 llvm::GlobalVariable *IvarOffsetGV = CGM.getModule().getGlobalVariable(Name);
1550 if (!IvarOffsetGV) {
1551 uint64_t Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
1552 llvm::ConstantInt *OffsetGuess =
1553 llvm::ConstantInt::get(LongTy, Offset, "ivar");
1554 IvarOffsetGV = new llvm::GlobalVariable(LongTy, false,
1555 llvm::GlobalValue::CommonLinkage, OffsetGuess, Name, &TheModule);
1556 }
1557 return IvarOffsetGV;
1558}
1559
Fariborz Jahanianc912eb72009-02-03 19:03:09 +00001560LValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
1561 QualType ObjectTy,
1562 llvm::Value *BaseValue,
1563 const ObjCIvarDecl *Ivar,
Fariborz Jahanianc912eb72009-02-03 19:03:09 +00001564 unsigned CVRQualifiers) {
Daniel Dunbarf5254bd2009-04-21 01:19:28 +00001565 const ObjCInterfaceDecl *ID = ObjectTy->getAsObjCInterfaceType()->getDecl();
Daniel Dunbar85d37542009-04-22 07:32:20 +00001566 return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
1567 EmitIvarOffset(CGF, ID, Ivar));
Fariborz Jahanian4337afe2009-02-02 20:02:29 +00001568}
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001569static const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
1570 const ObjCInterfaceDecl *OID,
1571 const ObjCIvarDecl *OIVD) {
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001572 llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
Fariborz Jahanianb290be02009-06-04 01:19:09 +00001573 Context.ShallowCollectObjCIvars(OID, Ivars);
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001574 for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
1575 if (OIVD == Ivars[k])
1576 return OID;
1577 }
1578
1579 // Otherwise check in the super class.
1580 if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
1581 return FindIvarInterface(Context, Super, OIVD);
1582
1583 return 0;
1584}
Fariborz Jahanian4337afe2009-02-02 20:02:29 +00001585
Fariborz Jahanian27cc6662009-02-10 19:02:04 +00001586llvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
Daniel Dunbar61e14a62009-04-22 05:08:15 +00001587 const ObjCInterfaceDecl *Interface,
Fariborz Jahanian27cc6662009-02-10 19:02:04 +00001588 const ObjCIvarDecl *Ivar) {
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001589 if (CGF.getContext().getLangOptions().ObjCNonFragileABI)
1590 {
1591 Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
1592 return CGF.Builder.CreateLoad(ObjCIvarOffsetVariable(Interface, Ivar),
1593 false, "ivar");
1594 }
Daniel Dunbar85d37542009-04-22 07:32:20 +00001595 uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
Fariborz Jahanianf25adf12009-05-20 18:41:51 +00001596 return llvm::ConstantInt::get(LongTy, Offset, "ivar");
Fariborz Jahanian27cc6662009-02-10 19:02:04 +00001597}
1598
Daniel Dunbar8c85fac2008-08-11 02:45:11 +00001599CodeGen::CGObjCRuntime *CodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM){
Chris Lattner547907c2008-06-26 04:19:03 +00001600 return new CGObjCGNU(CGM);
Chris Lattnera0fd5ee2008-03-01 08:50:34 +00001601}