CGObjCGNU.cpp revision 6d5a1c28593443f3973ef38f8fa042d59182412d
120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov//===------- CGObjCGNU.cpp - Emit LLVM Code from ASTs for a Module --------===//
20f984268b05edab2cc555a427c441baa9c252658Chris Lattner//
30f984268b05edab2cc555a427c441baa9c252658Chris Lattner//                     The LLVM Compiler Infrastructure
40f984268b05edab2cc555a427c441baa9c252658Chris Lattner//
50f984268b05edab2cc555a427c441baa9c252658Chris Lattner// This file is distributed under the University of Illinois Open Source
60f984268b05edab2cc555a427c441baa9c252658Chris Lattner// License. See LICENSE.TXT for details.
70f984268b05edab2cc555a427c441baa9c252658Chris Lattner//
80f984268b05edab2cc555a427c441baa9c252658Chris Lattner//===----------------------------------------------------------------------===//
90f984268b05edab2cc555a427c441baa9c252658Chris Lattner//
1020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// This provides Objective-C code generation targetting the GNU runtime.  The
1120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// class in this file generates structures used by the GNU Objective-C runtime
1220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// library.  These structures are defined in objc/objc.h and objc/objc-api.h in
1320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// the GNU runtime distribution.
140f984268b05edab2cc555a427c441baa9c252658Chris Lattner//
150f984268b05edab2cc555a427c441baa9c252658Chris Lattner//===----------------------------------------------------------------------===//
160f984268b05edab2cc555a427c441baa9c252658Chris Lattner
170f984268b05edab2cc555a427c441baa9c252658Chris Lattner#include "CGObjCRuntime.h"
18dce1406f1c1f572cfd61c494546572d63461c741Chris Lattner#include "CodeGenModule.h"
198f2926b73ed635afecd020da787af6a837601a2bDaniel Dunbar#include "CodeGenFunction.h"
205dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
21dce1406f1c1f572cfd61c494546572d63461c741Chris Lattner#include "clang/AST/ASTContext.h"
22e91593ef084479340582b2ba177b44be50a717b7Daniel Dunbar#include "clang/AST/Decl.h"
23af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar#include "clang/AST/DeclObjC.h"
2419cc4abea06a9b49e0e16a50d335c064cd723572Anders Carlsson#include "clang/AST/RecordLayout.h"
2516f0049415ec596504891259e2a83e19871c0d52Chris Lattner#include "clang/AST/StmtObjC.h"
265dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
275dc0867af17b3bd6f567897433853b2b4767446cChris Lattner#include "llvm/Intrinsics.h"
280f984268b05edab2cc555a427c441baa9c252658Chris Lattner#include "llvm/Module.h"
290f984268b05edab2cc555a427c441baa9c252658Chris Lattner#include "llvm/ADT/SmallVector.h"
3020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov#include "llvm/ADT/StringMap.h"
317ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar#include "llvm/Support/Compiler.h"
327ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar#include "llvm/Target/TargetData.h"
335dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
3420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov#include <map>
35e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner
36e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner
37dce1406f1c1f572cfd61c494546572d63461c741Chris Lattnerusing namespace clang;
3846f45b9bec4a265ad8400a538e5ec3a5683617f1Daniel Dunbarusing namespace CodeGen;
3920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovusing llvm::dyn_cast;
4020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
4120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// The version of the runtime that this class targets.  Must match the version
4220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// in the runtime.
4320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovstatic const int RuntimeVersion = 8;
449cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanianstatic const int NonFragileRuntimeVersion = 9;
4520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovstatic const int ProtocolVersion = 2;
46d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanianstatic const int NonFragileProtocolVersion = 3;
470f984268b05edab2cc555a427c441baa9c252658Chris Lattner
480f984268b05edab2cc555a427c441baa9c252658Chris Lattnernamespace {
49dce1406f1c1f572cfd61c494546572d63461c741Chris Lattnerclass CGObjCGNU : public CodeGen::CGObjCRuntime {
500f984268b05edab2cc555a427c441baa9c252658Chris Lattnerprivate:
51dce1406f1c1f572cfd61c494546572d63461c741Chris Lattner  CodeGen::CodeGenModule &CGM;
520f984268b05edab2cc555a427c441baa9c252658Chris Lattner  llvm::Module &TheModule;
53e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::PointerType *SelectorTy;
54d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  const llvm::IntegerType *Int8Ty;
55e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::PointerType *PtrToInt8Ty;
56b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  const llvm::FunctionType *IMPTy;
57e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::PointerType *IdTy;
58ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  const llvm::PointerType *PtrToIdTy;
590f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall  QualType ASTIdTy;
60e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::IntegerType *IntTy;
61e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::PointerType *PtrTy;
62e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::IntegerType *LongTy;
63e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::PointerType *PtrToIntTy;
645efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  llvm::GlobalAlias *ClassPtrAlias;
655efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  llvm::GlobalAlias *MetaClassPtrAlias;
6620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Classes;
6720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Categories;
6820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> ConstantStrings;
6948272a078e54d501cb05edd797c54b8e82ceb520David Chisnall  llvm::StringMap<llvm::Constant*> ObjCStrings;
7020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Function *LoadFunction;
7120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::StringMap<llvm::Constant*> ExistingProtocols;
7220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  typedef std::pair<std::string, std::string> TypedSelector;
7320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::map<TypedSelector, llvm::GlobalAlias*> TypedSelectors;
7420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::StringMap<llvm::GlobalAlias*> UntypedSelectors;
75ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  // Selectors that we don't emit in GC mode
76ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  Selector RetainSel, ReleaseSel, AutoreleaseSel;
77ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  // Functions used for GC.
78ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  llvm::Constant *IvarAssignFn, *StrongCastAssignFn, *MemMoveFn, *WeakReadFn,
79ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    *WeakAssignFn, *GlobalAssignFn;
8020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Some zeros used for GEPs in lots of places.
8120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *Zeros[2];
8220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *NULLPtr;
83a1cf15f4680e5cf39e72e28c5ea854fcba792e84Owen Anderson  llvm::LLVMContext &VMContext;
8420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovprivate:
8520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *GenerateIvarList(
8620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      const llvm::SmallVectorImpl<llvm::Constant *>  &IvarNames,
8720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      const llvm::SmallVectorImpl<llvm::Constant *>  &IvarTypes,
8820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      const llvm::SmallVectorImpl<llvm::Constant *>  &IvarOffsets);
8920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *GenerateMethodList(const std::string &ClassName,
9020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      const std::string &CategoryName,
911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      const llvm::SmallVectorImpl<Selector>  &MethodSels,
921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      const llvm::SmallVectorImpl<llvm::Constant *>  &MethodTypes,
9320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      bool isClassMethodList);
94f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  llvm::Constant *GenerateEmptyProtocol(const std::string &ProtocolName);
95d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *GeneratePropertyList(const ObjCImplementationDecl *OID,
96d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
97d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes);
9820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *GenerateProtocolList(
9920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      const llvm::SmallVectorImpl<std::string> &Protocols);
100d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // To ensure that all protocols are seen by the runtime, we add a category on
101d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // a class defined in the runtime, declaring no methods, but adopting the
102d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // protocols.
103d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  void GenerateProtocolHolderCategory(void);
10420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *GenerateClassStructure(
10520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      llvm::Constant *MetaClass,
10620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      llvm::Constant *SuperClass,
10720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      unsigned info,
108d002cc6fc5b72bf00e3b7b4571ccf0f23c789b4bChris Lattner      const char *Name,
10920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      llvm::Constant *Version,
11020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      llvm::Constant *InstanceSize,
11120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      llvm::Constant *IVars,
11220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      llvm::Constant *Methods,
113d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::Constant *Protocols,
114d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::Constant *IvarOffsets,
115d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::Constant *Properties);
11620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *GenerateProtocolMethodList(
11720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      const llvm::SmallVectorImpl<llvm::Constant *>  &MethodNames,
11820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      const llvm::SmallVectorImpl<llvm::Constant *>  &MethodTypes);
11920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *MakeConstantString(const std::string &Str, const std::string
12020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      &Name="");
121d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *ExportUniqueString(const std::string &Str, const std::string
122d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          prefix);
12320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *MakeGlobal(const llvm::StructType *Ty,
12441d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    std::vector<llvm::Constant*> &V, const std::string &Name="",
12541d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
12620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *MakeGlobal(const llvm::ArrayType *Ty,
12741d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    std::vector<llvm::Constant*> &V, const std::string &Name="",
12841d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    llvm::GlobalValue::LinkageTypes linkage=llvm::GlobalValue::InternalLinkage);
1299cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  llvm::GlobalVariable *ObjCIvarOffsetVariable(const ObjCInterfaceDecl *ID,
1309cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian      const ObjCIvarDecl *Ivar);
1312a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  void EmitClassRef(const std::string &className);
132ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  llvm::Value* EnforceType(CGBuilderTy B, llvm::Value *V, const llvm::Type *Ty){
133ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    if (V->getType() == Ty) return V;
134ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    return B.CreateBitCast(V, Ty);
135ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  }
1360f984268b05edab2cc555a427c441baa9c252658Chris Lattnerpublic:
137dce1406f1c1f572cfd61c494546572d63461c741Chris Lattner  CGObjCGNU(CodeGen::CodeGenModule &cgm);
1380d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  virtual llvm::Constant *GenerateConstantString(const StringLiteral *);
1391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual CodeGen::RValue
1408f2926b73ed635afecd020da787af6a837601a2bDaniel Dunbar  GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
1417f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                      QualType ResultType,
1427f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                      Selector Sel,
143f56f1913e91ad32bed52dd3f6afc26735d336584Daniel Dunbar                      llvm::Value *Receiver,
14419cd87eb5fb3c197e631ce08fd52c446c4d4e8f1Daniel Dunbar                      bool IsClassMessage,
145df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian                      const CallArgList &CallArgs,
146df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian                      const ObjCMethodDecl *Method);
1471eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual CodeGen::RValue
1488f2926b73ed635afecd020da787af6a837601a2bDaniel Dunbar  GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
1497f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                           QualType ResultType,
1507f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                           Selector Sel,
151f56f1913e91ad32bed52dd3f6afc26735d336584Daniel Dunbar                           const ObjCInterfaceDecl *Class,
1527ce77920a35060f1c8dd72e541e42ce296ccd168Fariborz Jahanian                           bool isCategoryImpl,
153f56f1913e91ad32bed52dd3f6afc26735d336584Daniel Dunbar                           llvm::Value *Receiver,
15419cd87eb5fb3c197e631ce08fd52c446c4d4e8f1Daniel Dunbar                           bool IsClassMessage,
155d6c93d703541c992e06eb9a59a2d826a30da65b2Daniel Dunbar                           const CallArgList &CallArgs,
156d6c93d703541c992e06eb9a59a2d826a30da65b2Daniel Dunbar                           const ObjCMethodDecl *Method);
15745d196b8387dcefc4df26cda114fa34c6528e928Daniel Dunbar  virtual llvm::Value *GetClass(CGBuilderTy &Builder,
158ddb2a3d55a24a1dbdf9152621642d9a4b4fc2f61Daniel Dunbar                                const ObjCInterfaceDecl *OID);
1596d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar  virtual llvm::Value *GetSelector(CGBuilderTy &Builder, Selector Sel);
1606d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar  virtual llvm::Value *GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
1616d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar      *Method);
1621eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  virtual llvm::Function *GenerateMethod(const ObjCMethodDecl *OMD,
164679a502d462ef819e6175b58e255ca3f3391e7cfFariborz Jahanian                                         const ObjCContainerDecl *CD);
1657ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  virtual void GenerateCategory(const ObjCCategoryImplDecl *CMD);
1667ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  virtual void GenerateClass(const ObjCImplementationDecl *ClassDecl);
16745d196b8387dcefc4df26cda114fa34c6528e928Daniel Dunbar  virtual llvm::Value *GenerateProtocolRef(CGBuilderTy &Builder,
168af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar                                           const ObjCProtocolDecl *PD);
169af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  virtual void GenerateProtocol(const ObjCProtocolDecl *PD);
17020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  virtual llvm::Function *ModuleInitFunction();
17149f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbar  virtual llvm::Function *GetPropertyGetFunction();
17249f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbar  virtual llvm::Function *GetPropertySetFunction();
173309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar  virtual llvm::Constant *EnumerationMutationFunction();
1741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
175bd71be4683c195260d5245118b1e13e6b2e20504Fariborz Jahanian  virtual void EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
176bd71be4683c195260d5245118b1e13e6b2e20504Fariborz Jahanian                                         const Stmt &S);
17764d5d6c5903157c521af496479d06dc26032d718Anders Carlsson  virtual void EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
17864d5d6c5903157c521af496479d06dc26032d718Anders Carlsson                             const ObjCAtThrowStmt &S);
1793e283e344595e0bd499b13b30a92b7d9c10a2140Fariborz Jahanian  virtual llvm::Value * EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
1806dc2317b59cb1180a59f6c283d96b7a5dfeb5307Fariborz Jahanian                                         llvm::Value *AddrWeakObj);
1813e283e344595e0bd499b13b30a92b7d9c10a2140Fariborz Jahanian  virtual void EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
1823e283e344595e0bd499b13b30a92b7d9c10a2140Fariborz Jahanian                                  llvm::Value *src, llvm::Value *dst);
18358626500527695865683d1d65053743de8770b60Fariborz Jahanian  virtual void EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
18458626500527695865683d1d65053743de8770b60Fariborz Jahanian                                    llvm::Value *src, llvm::Value *dest);
1857eda8367cf63caee8acf907356b1d199ccaa6e89Fariborz Jahanian  virtual void EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
1866c7a1f364796ce1acb988714e9e42076d1ce332eFariborz Jahanian                                    llvm::Value *src, llvm::Value *dest,
1876c7a1f364796ce1acb988714e9e42076d1ce332eFariborz Jahanian                                    llvm::Value *ivarOffset);
18858626500527695865683d1d65053743de8770b60Fariborz Jahanian  virtual void EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
18958626500527695865683d1d65053743de8770b60Fariborz Jahanian                                        llvm::Value *src, llvm::Value *dest);
190082b02e8403d3ee9d2ded969fbe0e5d472f04cd8Fariborz Jahanian  virtual void EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
1911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                        llvm::Value *DestPtr,
192082b02e8403d3ee9d2ded969fbe0e5d472f04cd8Fariborz Jahanian                                        llvm::Value *SrcPtr,
19308c321380fff07d476a19daab6d29522c046cd49Fariborz Jahanian                                        QualType Ty);
194598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian  virtual LValue EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
195598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                      QualType ObjectTy,
196598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                      llvm::Value *BaseValue,
197598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                      const ObjCIvarDecl *Ivar,
198598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                      unsigned CVRQualifiers);
199f63aa3fd429cdb9145d78f0b656bc78754efedb9Fariborz Jahanian  virtual llvm::Value *EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
2002a03192a02dbf4fdff438d1e658356bde871aba4Daniel Dunbar                                      const ObjCInterfaceDecl *Interface,
201f63aa3fd429cdb9145d78f0b656bc78754efedb9Fariborz Jahanian                                      const ObjCIvarDecl *Ivar);
2020f984268b05edab2cc555a427c441baa9c252658Chris Lattner};
2030f984268b05edab2cc555a427c441baa9c252658Chris Lattner} // end anonymous namespace
2040f984268b05edab2cc555a427c441baa9c252658Chris Lattner
20520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
2062a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner/// Emits a reference to a dummy variable which is emitted with each class.
2072a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner/// This ensures that a linker error will be generated when trying to link
2082a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner/// together modules where a referenced class is not defined.
209bb1c8600218b3244340331165fc7cba7bf227655Mike Stumpvoid CGObjCGNU::EmitClassRef(const std::string &className) {
2102a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  std::string symbolRef = "__objc_class_ref_" + className;
2112a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  // Don't emit two copies of the same symbol
212bb1c8600218b3244340331165fc7cba7bf227655Mike Stump  if (TheModule.getGlobalVariable(symbolRef))
213bb1c8600218b3244340331165fc7cba7bf227655Mike Stump    return;
2142a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  std::string symbolName = "__objc_class_name_" + className;
2152a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  llvm::GlobalVariable *ClassSymbol = TheModule.getGlobalVariable(symbolName);
2162a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  if (!ClassSymbol) {
2171c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson    ClassSymbol = new llvm::GlobalVariable(TheModule, LongTy, false,
2181c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson        llvm::GlobalValue::ExternalLinkage, 0, symbolName);
2192a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  }
2201c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson  new llvm::GlobalVariable(TheModule, ClassSymbol->getType(), true,
221f35271b8677a5f31498b89587b1da42a06dd46fdChris Lattner    llvm::GlobalValue::WeakAnyLinkage, ClassSymbol, symbolRef);
2222a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner}
22320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
22420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovstatic std::string SymbolNameForClass(const std::string &ClassName) {
2252a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  return "_OBJC_CLASS_" + ClassName;
22620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
22720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
228d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanianstatic std::string SymbolNameForMethod(const std::string &ClassName, const
229d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::string &CategoryName, const std::string &MethodName, bool isClassMethod)
230d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian{
231d346736dfe5ef584a2d0c9d27d217408ee394b9bDavid Chisnall  std::string MethodNameColonStripped = MethodName;
232d346736dfe5ef584a2d0c9d27d217408ee394b9bDavid Chisnall  std::replace(MethodNameColonStripped.begin(), MethodNameColonStripped.end(),
233d346736dfe5ef584a2d0c9d27d217408ee394b9bDavid Chisnall      ':', '_');
234d346736dfe5ef584a2d0c9d27d217408ee394b9bDavid Chisnall  return std::string(isClassMethod ? "_c_" : "_i_") + ClassName + "_" +
235d346736dfe5ef584a2d0c9d27d217408ee394b9bDavid Chisnall    CategoryName + "_" + MethodNameColonStripped;
23620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
23720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
238dce1406f1c1f572cfd61c494546572d63461c741Chris LattnerCGObjCGNU::CGObjCGNU(CodeGen::CodeGenModule &cgm)
2395efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  : CGM(cgm), TheModule(CGM.getModule()), ClassPtrAlias(0),
240a1cf15f4680e5cf39e72e28c5ea854fcba792e84Owen Anderson    MetaClassPtrAlias(0), VMContext(cgm.getLLVMContext()) {
241e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  IntTy = cast<llvm::IntegerType>(
242e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner      CGM.getTypes().ConvertType(CGM.getContext().IntTy));
243e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  LongTy = cast<llvm::IntegerType>(
244e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner      CGM.getTypes().ConvertType(CGM.getContext().LongTy));
2451eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
246d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Int8Ty = llvm::Type::getInt8Ty(VMContext);
247d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // C string type.  Used in lots of places.
248d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  PtrToInt8Ty = llvm::PointerType::getUnqual(Int8Ty);
249d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
2504a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Zeros[0] = llvm::ConstantInt::get(LongTy, 0);
25120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Zeros[1] = Zeros[0];
252d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  NULLPtr = llvm::ConstantPointerNull::get(PtrToInt8Ty);
253391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  // Get the selector Type.
2540d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  QualType selTy = CGM.getContext().getObjCSelType();
2550d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  if (QualType() == selTy) {
2560d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall    SelectorTy = PtrToInt8Ty;
2570d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  } else {
2580d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall    SelectorTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(selTy));
2590d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  }
260e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner
26196e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  PtrToIntTy = llvm::PointerType::getUnqual(IntTy);
262391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  PtrTy = PtrToInt8Ty;
2631eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
264391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  // Object type
2650f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall  ASTIdTy = CGM.getContext().getObjCIdType();
2660d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  if (QualType() == ASTIdTy) {
2670d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall    IdTy = PtrToInt8Ty;
2680d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  } else {
2690d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall    IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
2700d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  }
271ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  PtrToIdTy = llvm::PointerType::getUnqual(IdTy);
2721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
273391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  // IMP type
274391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  std::vector<const llvm::Type*> IMPArgs;
275391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  IMPArgs.push_back(IdTy);
276391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  IMPArgs.push_back(SelectorTy);
27796e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  IMPTy = llvm::FunctionType::get(IdTy, IMPArgs, true);
278ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall
279ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
280ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // Get selectors needed in GC mode
281ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    RetainSel = GetNullarySelector("retain", CGM.getContext());
282ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    ReleaseSel = GetNullarySelector("release", CGM.getContext());
283ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    AutoreleaseSel = GetNullarySelector("autorelease", CGM.getContext());
284ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall
285ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // Get functions needed in GC mode
286ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall
287ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // id objc_assign_ivar(id, id, ptrdiff_t);
288ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    std::vector<const llvm::Type*> Args(1, IdTy);
289ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.push_back(PtrToIdTy);
290ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // FIXME: ptrdiff_t
291ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.push_back(LongTy);
292ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    llvm::FunctionType *FTy = llvm::FunctionType::get(IdTy, Args, false);
293ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    IvarAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_ivar");
294ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // id objc_assign_strongCast (id, id*)
295ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.pop_back();
296ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    FTy = llvm::FunctionType::get(IdTy, Args, false);
297ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    StrongCastAssignFn =
298ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall        CGM.CreateRuntimeFunction(FTy, "objc_assign_strongCast");
299ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // id objc_assign_global(id, id*);
300ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    FTy = llvm::FunctionType::get(IdTy, Args, false);
301ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    GlobalAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_global");
302ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // id objc_assign_weak(id, id*);
303ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    FTy = llvm::FunctionType::get(IdTy, Args, false);
304ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    WeakAssignFn = CGM.CreateRuntimeFunction(FTy, "objc_assign_weak");
305ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // id objc_read_weak(id*);
306ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.clear();
307ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.push_back(PtrToIdTy);
308ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    FTy = llvm::FunctionType::get(IdTy, Args, false);
309ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    WeakReadFn = CGM.CreateRuntimeFunction(FTy, "objc_read_weak");
310ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // void *objc_memmove_collectable(void*, void *, size_t);
311ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.clear();
312ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.push_back(PtrToInt8Ty);
313ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.push_back(PtrToInt8Ty);
314ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    // FIXME: size_t
315ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    Args.push_back(LongTy);
316ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    FTy = llvm::FunctionType::get(IdTy, Args, false);
317ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    MemMoveFn = CGM.CreateRuntimeFunction(FTy, "objc_memmove_collectable");
318ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  }
31920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
320bb1c8600218b3244340331165fc7cba7bf227655Mike Stump
32120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// This has to perform the lookup every time, since posing and related
32220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// techniques can modify the name -> class mapping.
32345d196b8387dcefc4df26cda114fa34c6528e928Daniel Dunbarllvm::Value *CGObjCGNU::GetClass(CGBuilderTy &Builder,
324ddb2a3d55a24a1dbdf9152621642d9a4b4fc2f61Daniel Dunbar                                 const ObjCInterfaceDecl *OID) {
325d9d22dd9c94618490dbffb0e2caf222530ca39d3Chris Lattner  llvm::Value *ClassName = CGM.GetAddrOfConstantCString(OID->getNameAsString());
32641d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall  // With the incompatible ABI, this will need to be replaced with a direct
32741d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall  // reference to the class symbol.  For the compatible nonfragile ABI we are
32841d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall  // still performing this lookup at run time but emitting the symbol for the
32941d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall  // class externally so that we can make the switch later.
3302a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  EmitClassRef(OID->getNameAsString());
331ddb2a3d55a24a1dbdf9152621642d9a4b4fc2f61Daniel Dunbar  ClassName = Builder.CreateStructGEP(ClassName, 0);
332ddb2a3d55a24a1dbdf9152621642d9a4b4fc2f61Daniel Dunbar
33326c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian  std::vector<const llvm::Type*> Params(1, PtrToInt8Ty);
33420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *ClassLookupFn =
33596e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson    CGM.CreateRuntimeFunction(llvm::FunctionType::get(IdTy,
33626c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian                                                      Params,
33726c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian                                                      true),
33826c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian                              "objc_lookup_class");
33920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  return Builder.CreateCall(ClassLookupFn, ClassName);
340391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner}
341391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner
3426d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbarllvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, Selector Sel) {
343077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner  llvm::GlobalAlias *&US = UntypedSelectors[Sel.getAsString()];
3448e67b63530b4f39a48bc12d97376f373a6901279Chris Lattner  if (US == 0)
3456d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar    US = new llvm::GlobalAlias(llvm::PointerType::getUnqual(SelectorTy),
3460f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall                               llvm::GlobalValue::PrivateLinkage,
3470f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall                               ".objc_untyped_selector_alias"+Sel.getAsString(),
3488e67b63530b4f39a48bc12d97376f373a6901279Chris Lattner                               NULL, &TheModule);
3491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
3506d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar  return Builder.CreateLoad(US);
351df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian}
352df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian
3536d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbarllvm::Value *CGObjCGNU::GetSelector(CGBuilderTy &Builder, const ObjCMethodDecl
354df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian    *Method) {
355df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian
356df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  std::string SelName = Method->getSelector().getAsString();
357df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  std::string SelTypes;
358df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  CGM.getContext().getObjCEncodingForMethodDecl(Method, SelTypes);
359df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  // Typed selectors
360df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  TypedSelector Selector = TypedSelector(SelName,
361df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian          SelTypes);
362df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian
363df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  // If it's already cached, return it.
364bb1c8600218b3244340331165fc7cba7bf227655Mike Stump  if (TypedSelectors[Selector]) {
3656d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar    return Builder.CreateLoad(TypedSelectors[Selector]);
366df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  }
367df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian
368df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  // If it isn't, cache it.
369df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  llvm::GlobalAlias *Sel = new llvm::GlobalAlias(
3706d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar          llvm::PointerType::getUnqual(SelectorTy),
3710f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall          llvm::GlobalValue::PrivateLinkage, ".objc_selector_alias" + SelName,
372df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian          NULL, &TheModule);
373df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  TypedSelectors[Selector] = Sel;
374df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian
3756d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar  return Builder.CreateLoad(Sel);
3768e67b63530b4f39a48bc12d97376f373a6901279Chris Lattner}
3778e67b63530b4f39a48bc12d97376f373a6901279Chris Lattner
3785e7dcc62e4637591959f9c372f6c5258d0d263e0Chris Lattnerllvm::Constant *CGObjCGNU::MakeConstantString(const std::string &Str,
3795e7dcc62e4637591959f9c372f6c5258d0d263e0Chris Lattner                                              const std::string &Name) {
3808a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  llvm::Constant *ConstStr = CGM.GetAddrOfConstantCString(Str, Name.c_str());
3813c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
38220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
383d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanianllvm::Constant *CGObjCGNU::ExportUniqueString(const std::string &Str,
384d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        const std::string prefix) {
385d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::string name = prefix + Str;
386d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *ConstStr = TheModule.getGlobalVariable(name);
387d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  if (!ConstStr) {
388d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Constant *value = llvm::ConstantArray::get(VMContext, Str, true);
389d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    ConstStr = new llvm::GlobalVariable(TheModule, value->getType(), true,
390d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian            llvm::GlobalValue::LinkOnceODRLinkage, value, prefix + Str);
391d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  }
392d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  return llvm::ConstantExpr::getGetElementPtr(ConstStr, Zeros, 2);
393d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian}
394bb1c8600218b3244340331165fc7cba7bf227655Mike Stump
39520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovllvm::Constant *CGObjCGNU::MakeGlobal(const llvm::StructType *Ty,
39641d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    std::vector<llvm::Constant*> &V, const std::string &Name,
39741d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    llvm::GlobalValue::LinkageTypes linkage) {
39808e252425ca2cbdc44ba65d9a657ed5398014e36Owen Anderson  llvm::Constant *C = llvm::ConstantStruct::get(Ty, V);
3991c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson  return new llvm::GlobalVariable(TheModule, Ty, false,
4001c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson      llvm::GlobalValue::InternalLinkage, C, Name);
40120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
402bb1c8600218b3244340331165fc7cba7bf227655Mike Stump
40320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovllvm::Constant *CGObjCGNU::MakeGlobal(const llvm::ArrayType *Ty,
40441d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    std::vector<llvm::Constant*> &V, const std::string &Name,
40541d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall    llvm::GlobalValue::LinkageTypes linkage) {
4067db6d838aad4083fe86d7bf703a75fe6e8a17856Owen Anderson  llvm::Constant *C = llvm::ConstantArray::get(Ty, V);
4071c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson  return new llvm::GlobalVariable(TheModule, Ty, false,
408bb1c8600218b3244340331165fc7cba7bf227655Mike Stump                                  llvm::GlobalValue::InternalLinkage, C, Name);
40920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
41020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
41120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov/// Generate an NSConstantString object.
4120d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnallllvm::Constant *CGObjCGNU::GenerateConstantString(const StringLiteral *SL) {
41348272a078e54d501cb05edd797c54b8e82ceb520David Chisnall
4140d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall  std::string Str(SL->getStrData(), SL->getByteLength());
4150d13f6fdbdd6f06e2449b8834dda53334abd399aDavid Chisnall
41648272a078e54d501cb05edd797c54b8e82ceb520David Chisnall  // Look for an existing one
41748272a078e54d501cb05edd797c54b8e82ceb520David Chisnall  llvm::StringMap<llvm::Constant*>::iterator old = ObjCStrings.find(Str);
41848272a078e54d501cb05edd797c54b8e82ceb520David Chisnall  if (old != ObjCStrings.end())
41948272a078e54d501cb05edd797c54b8e82ceb520David Chisnall    return old->getValue();
42048272a078e54d501cb05edd797c54b8e82ceb520David Chisnall
42120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Ivars;
42220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Ivars.push_back(NULLPtr);
42313fd7e5111032f54b538dd66d035b0ccc1f82467Chris Lattner  Ivars.push_back(MakeConstantString(Str));
4244a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Ivars.push_back(llvm::ConstantInt::get(IntTy, Str.size()));
42520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *ObjCStr = MakeGlobal(
42647a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson    llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty, IntTy, NULL),
42720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Ivars, ".objc_str");
42848272a078e54d501cb05edd797c54b8e82ceb520David Chisnall  ObjCStr = llvm::ConstantExpr::getBitCast(ObjCStr, PtrToInt8Ty);
42948272a078e54d501cb05edd797c54b8e82ceb520David Chisnall  ObjCStrings[Str] = ObjCStr;
43048272a078e54d501cb05edd797c54b8e82ceb520David Chisnall  ConstantStrings.push_back(ObjCStr);
43120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  return ObjCStr;
43220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
43320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
43420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov///Generates a message send where the super is the receiver.  This is a message
43520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov///send to self with special delivery semantics indicating which class's method
43620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov///should be called.
4378f2926b73ed635afecd020da787af6a837601a2bDaniel DunbarCodeGen::RValue
4388f2926b73ed635afecd020da787af6a837601a2bDaniel DunbarCGObjCGNU::GenerateMessageSendSuper(CodeGen::CodeGenFunction &CGF,
4397f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                                    QualType ResultType,
4407f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                                    Selector Sel,
441f56f1913e91ad32bed52dd3f6afc26735d336584Daniel Dunbar                                    const ObjCInterfaceDecl *Class,
4427ce77920a35060f1c8dd72e541e42ce296ccd168Fariborz Jahanian                                    bool isCategoryImpl,
443f56f1913e91ad32bed52dd3f6afc26735d336584Daniel Dunbar                                    llvm::Value *Receiver,
44419cd87eb5fb3c197e631ce08fd52c446c4d4e8f1Daniel Dunbar                                    bool IsClassMessage,
445d6c93d703541c992e06eb9a59a2d826a30da65b2Daniel Dunbar                                    const CallArgList &CallArgs,
446d6c93d703541c992e06eb9a59a2d826a30da65b2Daniel Dunbar                                    const ObjCMethodDecl *Method) {
447ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
448ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    if (Sel == RetainSel || Sel == AutoreleaseSel) {
449ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall      return RValue::get(Receiver);
450ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    }
451ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    if (Sel == ReleaseSel) {
452ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall      return RValue::get(0);
453ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    }
454ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  }
455b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  llvm::Value *cmd = GetSelector(CGF.Builder, Sel);
456b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian
457b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  CallArgList ActualArgs;
458b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian
459b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  ActualArgs.push_back(
4600f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall      std::make_pair(RValue::get(CGF.Builder.CreateBitCast(Receiver, IdTy)),
4610f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall      ASTIdTy));
462b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  ActualArgs.push_back(std::make_pair(RValue::get(cmd),
463b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian                                      CGF.getContext().getObjCSelType()));
464b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
465b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian
466b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  CodeGenTypes &Types = CGM.getTypes();
467b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
4686793966aa221764f579cc8436bd641e1ec339c6dDaniel Dunbar  const llvm::FunctionType *impType =
4696793966aa221764f579cc8436bd641e1ec339c6dDaniel Dunbar    Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
470b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian
4715efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  llvm::Value *ReceiverClass = 0;
47248e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner  if (isCategoryImpl) {
47348e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    llvm::Constant *classLookupFunction = 0;
47448e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    std::vector<const llvm::Type*> Params;
47548e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    Params.push_back(PtrTy);
47648e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    if (IsClassMessage)  {
47796e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
47848e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner            IdTy, Params, true), "objc_get_meta_class");
47948e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    } else {
48096e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      classLookupFunction = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
48148e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner            IdTy, Params, true), "objc_get_class");
4825efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar    }
48348e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    ReceiverClass = CGF.Builder.CreateCall(classLookupFunction,
48448e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner        MakeConstantString(Class->getNameAsString()));
4855efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  } else {
48648e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    // Set up global aliases for the metaclass or class pointer if they do not
48748e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    // already exist.  These will are forward-references which will be set to
488bb1c8600218b3244340331165fc7cba7bf227655Mike Stump    // pointers to the class and metaclass structure created for the runtime
489bb1c8600218b3244340331165fc7cba7bf227655Mike Stump    // load function.  To send a message to super, we look up the value of the
49048e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    // super_class pointer from either the class or metaclass structure.
49148e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    if (IsClassMessage)  {
49248e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner      if (!MetaClassPtrAlias) {
49348e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner        MetaClassPtrAlias = new llvm::GlobalAlias(IdTy,
49448e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner            llvm::GlobalValue::InternalLinkage, ".objc_metaclass_ref" +
49548e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner            Class->getNameAsString(), NULL, &TheModule);
49648e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner      }
49748e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner      ReceiverClass = MetaClassPtrAlias;
49848e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner    } else {
49948e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner      if (!ClassPtrAlias) {
50048e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner        ClassPtrAlias = new llvm::GlobalAlias(IdTy,
50148e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner            llvm::GlobalValue::InternalLinkage, ".objc_class_ref" +
50248e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner            Class->getNameAsString(), NULL, &TheModule);
50348e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner      }
50448e6e7e8106f2404bfdd2d8abe7a840aeeaf047bChris Lattner      ReceiverClass = ClassPtrAlias;
5055efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar    }
50671238f67c5df13eef19450d66b7a7ab28377192aChris Lattner  }
5075efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  // Cast the pointer to a simplified version of the class structure
5081eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  ReceiverClass = CGF.Builder.CreateBitCast(ReceiverClass,
50996e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      llvm::PointerType::getUnqual(
51047a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson        llvm::StructType::get(VMContext, IdTy, IdTy, NULL)));
5115efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  // Get the superclass pointer
5125efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  ReceiverClass = CGF.Builder.CreateStructGEP(ReceiverClass, 1);
5135efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  // Load the superclass pointer
5145efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  ReceiverClass = CGF.Builder.CreateLoad(ReceiverClass);
51520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Construct the structure used to look up the IMP
51647a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ObjCSuperTy = llvm::StructType::get(VMContext,
51747a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson      Receiver->getType(), IdTy, NULL);
5188f2926b73ed635afecd020da787af6a837601a2bDaniel Dunbar  llvm::Value *ObjCSuper = CGF.Builder.CreateAlloca(ObjCSuperTy);
519b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian
5208f2926b73ed635afecd020da787af6a837601a2bDaniel Dunbar  CGF.Builder.CreateStore(Receiver, CGF.Builder.CreateStructGEP(ObjCSuper, 0));
5215efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  CGF.Builder.CreateStore(ReceiverClass,
5225efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar      CGF.Builder.CreateStructGEP(ObjCSuper, 1));
52320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
52420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Get the IMP
52526c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian  std::vector<const llvm::Type*> Params;
52696e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  Params.push_back(llvm::PointerType::getUnqual(ObjCSuperTy));
52726c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian  Params.push_back(SelectorTy);
5281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  llvm::Constant *lookupFunction =
52996e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson    CGM.CreateRuntimeFunction(llvm::FunctionType::get(
53096e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson          llvm::PointerType::getUnqual(impType), Params, true),
53126c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian        "objc_msg_lookup_super");
53226c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian
53320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Value *lookupArgs[] = {ObjCSuper, cmd};
5348f2926b73ed635afecd020da787af6a837601a2bDaniel Dunbar  llvm::Value *imp = CGF.Builder.CreateCall(lookupFunction, lookupArgs,
53520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      lookupArgs+2);
53620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
537f3c47c9525153aea2de0ec4bd615b9cf2d81c103Anders Carlsson  return CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
53820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
53920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
5401eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// Generate code for a message send expression.
5418f2926b73ed635afecd020da787af6a837601a2bDaniel DunbarCodeGen::RValue
5428f2926b73ed635afecd020da787af6a837601a2bDaniel DunbarCGObjCGNU::GenerateMessageSend(CodeGen::CodeGenFunction &CGF,
5437f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                               QualType ResultType,
5447f8ea5c5b3a6a4332a841eefdd86b0726722ea7bDaniel Dunbar                               Selector Sel,
545f56f1913e91ad32bed52dd3f6afc26735d336584Daniel Dunbar                               llvm::Value *Receiver,
54619cd87eb5fb3c197e631ce08fd52c446c4d4e8f1Daniel Dunbar                               bool IsClassMessage,
547df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian                               const CallArgList &CallArgs,
548df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian                               const ObjCMethodDecl *Method) {
549ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  if (CGM.getLangOptions().getGCMode() != LangOptions::NonGC) {
550ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    if (Sel == RetainSel || Sel == AutoreleaseSel) {
551ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall      return RValue::get(Receiver);
552ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    }
553ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    if (Sel == ReleaseSel) {
554ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall      return RValue::get(0);
555ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall    }
556ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  }
557d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  CGBuilderTy &Builder = CGF.Builder;
5580f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall  IdTy = cast<llvm::PointerType>(CGM.getTypes().ConvertType(ASTIdTy));
559df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  llvm::Value *cmd;
560df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  if (Method)
561d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    cmd = GetSelector(Builder, Method);
562df9ccc6381314ccca6407abb209155e9273a631dFariborz Jahanian  else
563d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    cmd = GetSelector(Builder, Sel);
564b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  CallArgList ActualArgs;
565b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian
566d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Receiver = Builder.CreateBitCast(Receiver, IdTy);
567b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  ActualArgs.push_back(
568d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    std::make_pair(RValue::get(Receiver), ASTIdTy));
569b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  ActualArgs.push_back(std::make_pair(RValue::get(cmd),
570b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian                                      CGF.getContext().getObjCSelType()));
571b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  ActualArgs.insert(ActualArgs.end(), CallArgs.begin(), CallArgs.end());
572b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian
573b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  CodeGenTypes &Types = CGM.getTypes();
574b3716efb0966620f205c939b671b579ad331ca0dFariborz Jahanian  const CGFunctionInfo &FnInfo = Types.getFunctionInfo(ResultType, ActualArgs);
5756793966aa221764f579cc8436bd641e1ec339c6dDaniel Dunbar  const llvm::FunctionType *impType =
5766793966aa221764f579cc8436bd641e1ec339c6dDaniel Dunbar    Types.GetFunctionType(FnInfo, Method ? Method->isVariadic() : false);
5770f984268b05edab2cc555a427c441baa9c252658Chris Lattner
57834e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian  llvm::Value *imp;
57934e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian  // For sender-aware dispatch, we pass the sender as the third argument to a
58034e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian  // lookup function.  When sending messages from C code, the sender is nil.
581d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // objc_msg_lookup_sender(id *receiver, SEL selector, id sender);
582d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
583d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
584d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    std::vector<const llvm::Type*> Params;
585d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Value *ReceiverPtr = CGF.CreateTempAlloca(Receiver->getType());
586d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Builder.CreateStore(Receiver, ReceiverPtr);
587d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Params.push_back(ReceiverPtr->getType());
588d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Params.push_back(SelectorTy);
58934e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian    llvm::Value *self;
59034e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian
59134e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian    if (isa<ObjCMethodDecl>(CGF.CurFuncDecl)) {
59234e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian      self = CGF.LoadObjCSelf();
59334e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian    } else {
59403e205031b08669f05c41eed5b896fc94c4a12bbOwen Anderson      self = llvm::ConstantPointerNull::get(IdTy);
59534e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian    }
596d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
59734e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian    Params.push_back(self->getType());
598d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
599d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    // The lookup function returns a slot, which can be safely cached.
600d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Type *SlotTy = llvm::StructType::get(VMContext, PtrTy, PtrTy, PtrTy,
601d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian            IntTy, llvm::PointerType::getUnqual(impType), NULL);
6021eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::Constant *lookupFunction =
60396e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      CGM.CreateRuntimeFunction(llvm::FunctionType::get(
604d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          llvm::PointerType::getUnqual(SlotTy), Params, true),
60534e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian        "objc_msg_lookup_sender");
60626c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian
607d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    // The lookup function is guaranteed not to capture the receiver pointer.
608d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if (llvm::Function *LookupFn = dyn_cast<llvm::Function>(lookupFunction)) {
609d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      LookupFn->setDoesNotCapture(1);
610d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
611d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
612d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Value *slot =
613d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        Builder.CreateCall3(lookupFunction, ReceiverPtr, cmd, self);
614d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    imp = Builder.CreateLoad(Builder.CreateStructGEP(slot, 4));
615d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    // The lookup function may have changed the receiver, so make sure we use
616d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    // the new one.
617d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    ActualArgs[0] =
618d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        std::make_pair(RValue::get(Builder.CreateLoad(ReceiverPtr)), ASTIdTy);
61934e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian  } else {
620d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    std::vector<const llvm::Type*> Params;
621d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Params.push_back(Receiver->getType());
622d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Params.push_back(SelectorTy);
6231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::Constant *lookupFunction =
62496e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson    CGM.CreateRuntimeFunction(llvm::FunctionType::get(
62596e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson        llvm::PointerType::getUnqual(impType), Params, true),
62634e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian      "objc_msg_lookup");
62734e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian
628d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    imp = Builder.CreateCall2(lookupFunction, Receiver, cmd);
62934e65770ad6142b033d6cec4f476ed20fb248bf3Fariborz Jahanian  }
6303eae03e7d165f20a863a9a4d7122ba2a691ab16dChris Lattner
631f3c47c9525153aea2de0ec4bd615b9cf2d81c103Anders Carlsson  return CGF.EmitCall(FnInfo, imp, ReturnValueSlot(), ActualArgs);
63220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
63320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
6341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump/// Generates a MethodList.  Used in construction of a objc_class and
63520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov/// objc_category structures.
63620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovllvm::Constant *CGObjCGNU::GenerateMethodList(const std::string &ClassName,
6371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                              const std::string &CategoryName,
6381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    const llvm::SmallVectorImpl<Selector> &MethodSels,
6391eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    const llvm::SmallVectorImpl<llvm::Constant *> &MethodTypes,
64020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    bool isClassMethodList) {
6410f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall  if (MethodSels.empty())
6420f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall    return NULLPtr;
6431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Get the method structure type.
64447a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ObjCMethodTy = llvm::StructType::get(VMContext,
64520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    PtrToInt8Ty, // Really a selector, but the runtime creates it us.
64620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    PtrToInt8Ty, // Method types
64796e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson    llvm::PointerType::getUnqual(IMPTy), //Method pointer
64820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    NULL);
64920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Methods;
65020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Elements;
65120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (unsigned int i = 0, e = MethodTypes.size(); i < e; ++i) {
65220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.clear();
6531e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian    if (llvm::Constant *Method =
65420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      TheModule.getFunction(SymbolNameForMethod(ClassName, CategoryName,
655077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner                                                MethodSels[i].getAsString(),
6561e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian                                                isClassMethodList))) {
6578a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      llvm::Constant *C = MakeConstantString(MethodSels[i].getAsString());
6588a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      Elements.push_back(C);
6598a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      Elements.push_back(MethodTypes[i]);
6603c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson      Method = llvm::ConstantExpr::getBitCast(Method,
66196e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson          llvm::PointerType::getUnqual(IMPTy));
6621e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian      Elements.push_back(Method);
66308e252425ca2cbdc44ba65d9a657ed5398014e36Owen Anderson      Methods.push_back(llvm::ConstantStruct::get(ObjCMethodTy, Elements));
6641e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian    }
66520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
66620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
66720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Array of method structures
66896e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodTy,
6691e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian                                                            Methods.size());
6707db6d838aad4083fe86d7bf703a75fe6e8a17856Owen Anderson  llvm::Constant *MethodArray = llvm::ConstantArray::get(ObjCMethodArrayTy,
671fba676396703e87a5034e5f308aa30a633468f66Chris Lattner                                                         Methods);
67220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
67320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Structure containing list pointer, array and array count
67420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::SmallVector<const llvm::Type*, 16> ObjCMethodListFields;
6758c8f69ec1477424e11faef6b235505584d61e522Owen Anderson  llvm::PATypeHolder OpaqueNextTy = llvm::OpaqueType::get(VMContext);
67696e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  llvm::Type *NextPtrTy = llvm::PointerType::getUnqual(OpaqueNextTy);
67747a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ObjCMethodListTy = llvm::StructType::get(VMContext,
6781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      NextPtrTy,
6791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      IntTy,
68020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      ObjCMethodArrayTy,
68120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      NULL);
68220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Refine next pointer type to concrete type
68320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::cast<llvm::OpaqueType>(
68420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      OpaqueNextTy.get())->refineAbstractTypeTo(ObjCMethodListTy);
68520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  ObjCMethodListTy = llvm::cast<llvm::StructType>(OpaqueNextTy.get());
68620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
68720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Methods.clear();
68803e205031b08669f05c41eed5b896fc94c4a12bbOwen Anderson  Methods.push_back(llvm::ConstantPointerNull::get(
68996e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson        llvm::PointerType::getUnqual(ObjCMethodListTy)));
6900032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson  Methods.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext),
69120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov        MethodTypes.size()));
69220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Methods.push_back(MethodArray);
6931eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
69420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Create an instance of the structure
69520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  return MakeGlobal(ObjCMethodListTy, Methods, ".objc_method_list");
69620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
69720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
69820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov/// Generates an IvarList.  Used in construction of a objc_class.
69920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovllvm::Constant *CGObjCGNU::GenerateIvarList(
70020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    const llvm::SmallVectorImpl<llvm::Constant *>  &IvarNames,
70120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    const llvm::SmallVectorImpl<llvm::Constant *>  &IvarTypes,
70220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    const llvm::SmallVectorImpl<llvm::Constant *>  &IvarOffsets) {
7031804463ee790ac3cc73f72597ccb38b983f7d263David Chisnall  if (IvarNames.size() == 0)
7041804463ee790ac3cc73f72597ccb38b983f7d263David Chisnall    return NULLPtr;
7051eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Get the method structure type.
70647a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ObjCIvarTy = llvm::StructType::get(VMContext,
70720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    PtrToInt8Ty,
70820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    PtrToInt8Ty,
70920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    IntTy,
71020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    NULL);
71120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Ivars;
71220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Elements;
71320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (unsigned int i = 0, e = IvarNames.size() ; i < e ; i++) {
71420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.clear();
7158a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    Elements.push_back(IvarNames[i]);
7168a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    Elements.push_back(IvarTypes[i]);
71720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.push_back(IvarOffsets[i]);
71808e252425ca2cbdc44ba65d9a657ed5398014e36Owen Anderson    Ivars.push_back(llvm::ConstantStruct::get(ObjCIvarTy, Elements));
71920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
72020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
72120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Array of method structures
72296e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  llvm::ArrayType *ObjCIvarArrayTy = llvm::ArrayType::get(ObjCIvarTy,
72320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      IvarNames.size());
72420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
7251eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
72620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.clear();
7274a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Elements.push_back(llvm::ConstantInt::get(IntTy, (int)IvarNames.size()));
7287db6d838aad4083fe86d7bf703a75fe6e8a17856Owen Anderson  Elements.push_back(llvm::ConstantArray::get(ObjCIvarArrayTy, Ivars));
72920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Structure containing array and array count
73047a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ObjCIvarListTy = llvm::StructType::get(VMContext, IntTy,
73120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    ObjCIvarArrayTy,
73220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    NULL);
73320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
73420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Create an instance of the structure
73520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  return MakeGlobal(ObjCIvarListTy, Elements, ".objc_ivar_list");
7360f984268b05edab2cc555a427c441baa9c252658Chris Lattner}
7370f984268b05edab2cc555a427c441baa9c252658Chris Lattner
73820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov/// Generate a class structure
73920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovllvm::Constant *CGObjCGNU::GenerateClassStructure(
74020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *MetaClass,
74120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *SuperClass,
74220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    unsigned info,
743d002cc6fc5b72bf00e3b7b4571ccf0f23c789b4bChris Lattner    const char *Name,
74420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *Version,
74520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *InstanceSize,
74620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *IVars,
74720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *Methods,
748d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Constant *Protocols,
749d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Constant *IvarOffsets,
750d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Constant *Properties) {
75120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Set up the class structure
75220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Note:  Several of these are char*s when they should be ids.  This is
75320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // because the runtime performs this translation on load.
754d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  //
755d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Fields marked New ABI are part of the GNUstep runtime.  We emit them
756d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // anyway; the classes will still work with the GNU runtime, they will just
757d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // be ignored.
75847a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ClassTy = llvm::StructType::get(VMContext,
75920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrToInt8Ty,        // class_pointer
76020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrToInt8Ty,        // super_class
76120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrToInt8Ty,        // name
76220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      LongTy,             // version
76320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      LongTy,             // info
76420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      LongTy,             // instance_size
76520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      IVars->getType(),   // ivars
76620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      Methods->getType(), // methods
7671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      // These are all filled in by the runtime, so we pretend
76820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrTy,              // dtable
76920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrTy,              // subclass_list
77020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrTy,              // sibling_class
77120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrTy,              // protocols
77220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrTy,              // gc_object_type
773d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      // New ABI:
774d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      LongTy,                 // abi_version
775d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      IvarOffsets->getType(), // ivar_offsets
776d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Properties->getType(),  // properties
77720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      NULL);
7784a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  llvm::Constant *Zero = llvm::ConstantInt::get(LongTy, 0);
77920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Fill in the structure
78020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Elements;
7813c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Elements.push_back(llvm::ConstantExpr::getBitCast(MetaClass, PtrToInt8Ty));
78220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(SuperClass);
783d002cc6fc5b72bf00e3b7b4571ccf0f23c789b4bChris Lattner  Elements.push_back(MakeConstantString(Name, ".class_name"));
78420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(Zero);
7854a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Elements.push_back(llvm::ConstantInt::get(LongTy, info));
78620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(InstanceSize);
78720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(IVars);
78820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(Methods);
789d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(NULLPtr);
790d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(NULLPtr);
791d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(NULLPtr);
7923c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Elements.push_back(llvm::ConstantExpr::getBitCast(Protocols, PtrTy));
793d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(NULLPtr);
794d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(Zero);
795d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(IvarOffsets);
796d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(Properties);
79720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Create an instance of the structure
79841d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall  // This is now an externally visible symbol, so that we can speed up class
79941d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall  // messages in the next ABI.
80041d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall  return MakeGlobal(ClassTy, Elements, SymbolNameForClass(Name),
80141d63ed274d1bf7ba7459f218c8f6a385fc5c6d6David Chisnall         llvm::GlobalValue::ExternalLinkage);
80220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
80320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
80420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovllvm::Constant *CGObjCGNU::GenerateProtocolMethodList(
80520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    const llvm::SmallVectorImpl<llvm::Constant *>  &MethodNames,
80620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    const llvm::SmallVectorImpl<llvm::Constant *>  &MethodTypes) {
8071eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Get the method structure type.
80847a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ObjCMethodDescTy = llvm::StructType::get(VMContext,
80920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    PtrToInt8Ty, // Really a selector, but the runtime does the casting for us.
81020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    PtrToInt8Ty,
81120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    NULL);
81220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Methods;
81320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Elements;
81420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (unsigned int i = 0, e = MethodTypes.size() ; i < e ; i++) {
81520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.clear();
8161eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    Elements.push_back(MethodNames[i]);
8178a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    Elements.push_back(MethodTypes[i]);
81808e252425ca2cbdc44ba65d9a657ed5398014e36Owen Anderson    Methods.push_back(llvm::ConstantStruct::get(ObjCMethodDescTy, Elements));
81920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
82096e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  llvm::ArrayType *ObjCMethodArrayTy = llvm::ArrayType::get(ObjCMethodDescTy,
82120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      MethodNames.size());
8227db6d838aad4083fe86d7bf703a75fe6e8a17856Owen Anderson  llvm::Constant *Array = llvm::ConstantArray::get(ObjCMethodArrayTy,
823bb1c8600218b3244340331165fc7cba7bf227655Mike Stump                                                   Methods);
82447a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ObjCMethodDescListTy = llvm::StructType::get(VMContext,
82520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      IntTy, ObjCMethodArrayTy, NULL);
82620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Methods.clear();
8274a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Methods.push_back(llvm::ConstantInt::get(IntTy, MethodNames.size()));
82820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Methods.push_back(Array);
82920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  return MakeGlobal(ObjCMethodDescListTy, Methods, ".objc_method_list");
83020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
831bb1c8600218b3244340331165fc7cba7bf227655Mike Stump
83220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov// Create the protocol list structure used in classes, categories and so on
83320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikovllvm::Constant *CGObjCGNU::GenerateProtocolList(
83420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    const llvm::SmallVectorImpl<std::string> &Protocols) {
83596e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
83620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      Protocols.size());
83747a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
83820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrTy, //Should be a recurisve pointer, but it's always NULL here.
83920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      LongTy,//FIXME: Should be size_t
84020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      ProtocolArrayTy,
84120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      NULL);
8421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  std::vector<llvm::Constant*> Elements;
84320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (const std::string *iter = Protocols.begin(), *endIter = Protocols.end();
84420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      iter != endIter ; iter++) {
845ff80fabfa42d4aadbbaab4d61fa518d80b5a981aDavid Chisnall    llvm::Constant *protocol = 0;
846ff80fabfa42d4aadbbaab4d61fa518d80b5a981aDavid Chisnall    llvm::StringMap<llvm::Constant*>::iterator value =
847ff80fabfa42d4aadbbaab4d61fa518d80b5a981aDavid Chisnall      ExistingProtocols.find(*iter);
848ff80fabfa42d4aadbbaab4d61fa518d80b5a981aDavid Chisnall    if (value == ExistingProtocols.end()) {
849f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian      protocol = GenerateEmptyProtocol(*iter);
850ff80fabfa42d4aadbbaab4d61fa518d80b5a981aDavid Chisnall    } else {
851ff80fabfa42d4aadbbaab4d61fa518d80b5a981aDavid Chisnall      protocol = value->getValue();
852ff80fabfa42d4aadbbaab4d61fa518d80b5a981aDavid Chisnall    }
8533c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson    llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(protocol,
854a1cf15f4680e5cf39e72e28c5ea854fcba792e84Owen Anderson                                                           PtrToInt8Ty);
85520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.push_back(Ptr);
85620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
8577db6d838aad4083fe86d7bf703a75fe6e8a17856Owen Anderson  llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
85820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      Elements);
85920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.clear();
86020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(NULLPtr);
8614a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Elements.push_back(llvm::ConstantInt::get(LongTy, Protocols.size()));
86220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(ProtocolArray);
86320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  return MakeGlobal(ProtocolListTy, Elements, ".objc_protocol_list");
86420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
86520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
8661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpllvm::Value *CGObjCGNU::GenerateProtocolRef(CGBuilderTy &Builder,
867af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar                                            const ObjCProtocolDecl *PD) {
868f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  llvm::Value *protocol = ExistingProtocols[PD->getNameAsString()];
8691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const llvm::Type *T =
870f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian    CGM.getTypes().ConvertType(CGM.getContext().getObjCProtoType());
87196e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  return Builder.CreateBitCast(protocol, llvm::PointerType::getUnqual(T));
872f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian}
873f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian
874f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanianllvm::Constant *CGObjCGNU::GenerateEmptyProtocol(
875f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  const std::string &ProtocolName) {
876f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  llvm::SmallVector<std::string, 0> EmptyStringVector;
877f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  llvm::SmallVector<llvm::Constant*, 0> EmptyConstantVector;
878f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian
879f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  llvm::Constant *ProtocolList = GenerateProtocolList(EmptyStringVector);
880d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *MethodList =
881f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian    GenerateProtocolMethodList(EmptyConstantVector, EmptyConstantVector);
882f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  // Protocols are objects containing lists of the methods implemented and
883f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  // protocols adopted.
88447a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
885f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian      PtrToInt8Ty,
886f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian      ProtocolList->getType(),
887d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      MethodList->getType(),
888d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      MethodList->getType(),
889d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      MethodList->getType(),
890d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      MethodList->getType(),
891f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian      NULL);
8921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  std::vector<llvm::Constant*> Elements;
893f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  // The isa pointer must be set to a magic number so the runtime knows it's
894f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  // the correct layout.
895d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
896d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      NonFragileProtocolVersion : ProtocolVersion;
8973c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Elements.push_back(llvm::ConstantExpr::getIntToPtr(
898d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
899f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
900f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  Elements.push_back(ProtocolList);
901d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(MethodList);
902d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(MethodList);
903d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(MethodList);
904d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(MethodList);
905f8c4f5469d7db591eeacfc3381d91f773fb7af43Fariborz Jahanian  return MakeGlobal(ProtocolTy, Elements, ".objc_protocol");
90620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
90720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
908af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbarvoid CGObjCGNU::GenerateProtocol(const ObjCProtocolDecl *PD) {
909af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  ASTContext &Context = CGM.getContext();
9108ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner  std::string ProtocolName = PD->getNameAsString();
911af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  llvm::SmallVector<std::string, 16> Protocols;
912af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  for (ObjCProtocolDecl::protocol_iterator PI = PD->protocol_begin(),
913af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar       E = PD->protocol_end(); PI != E; ++PI)
914d9d22dd9c94618490dbffb0e2caf222530ca39d3Chris Lattner    Protocols.push_back((*PI)->getNameAsString());
915af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> InstanceMethodNames;
916af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
917d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodNames;
918d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::SmallVector<llvm::Constant*, 16> OptionalInstanceMethodTypes;
91917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis  for (ObjCProtocolDecl::instmeth_iterator iter = PD->instmeth_begin(),
92017945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis       E = PD->instmeth_end(); iter != E; iter++) {
921af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar    std::string TypeStr;
922af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar    Context.getObjCEncodingForMethodDecl(*iter, TypeStr);
923d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
924d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodNames.push_back(
925d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          MakeConstantString((*iter)->getSelector().getAsString()));
926d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
927d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    } else {
928d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalInstanceMethodNames.push_back(
929d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          MakeConstantString((*iter)->getSelector().getAsString()));
930d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalInstanceMethodTypes.push_back(MakeConstantString(TypeStr));
931d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
932af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  }
933af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  // Collect information about class methods:
934af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> ClassMethodNames;
935af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
936d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodNames;
937d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::SmallVector<llvm::Constant*, 16> OptionalClassMethodTypes;
9381eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  for (ObjCProtocolDecl::classmeth_iterator
93917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis         iter = PD->classmeth_begin(), endIter = PD->classmeth_end();
94017945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis       iter != endIter ; iter++) {
941af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar    std::string TypeStr;
942af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar    Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
943d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if ((*iter)->getImplementationControl() == ObjCMethodDecl::Optional) {
944d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      ClassMethodNames.push_back(
945d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          MakeConstantString((*iter)->getSelector().getAsString()));
946d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      ClassMethodTypes.push_back(MakeConstantString(TypeStr));
947d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    } else {
948d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalClassMethodNames.push_back(
949d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          MakeConstantString((*iter)->getSelector().getAsString()));
950d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalClassMethodTypes.push_back(MakeConstantString(TypeStr));
951d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
952af2f62ce32e462f256855cd24b06dec4755d2827Daniel Dunbar  }
95320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
95420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *ProtocolList = GenerateProtocolList(Protocols);
95520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *InstanceMethodList =
95620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    GenerateProtocolMethodList(InstanceMethodNames, InstanceMethodTypes);
95720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *ClassMethodList =
95820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    GenerateProtocolMethodList(ClassMethodNames, ClassMethodTypes);
959d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *OptionalInstanceMethodList =
960d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    GenerateProtocolMethodList(OptionalInstanceMethodNames,
961d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian            OptionalInstanceMethodTypes);
962d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *OptionalClassMethodList =
963d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    GenerateProtocolMethodList(OptionalClassMethodNames,
964d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian            OptionalClassMethodTypes);
965d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
966d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Property metadata: name, attributes, isSynthesized, setter name, setter
967d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // types, getter name, getter types.
968d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // The isSynthesized value is always set to 0 in a protocol.  It exists to
969d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // simplify the runtime library by allowing it to use the same data
970d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // structures for protocol metadata everywhere.
971d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
972d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
973d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          PtrToInt8Ty, NULL);
974d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::vector<llvm::Constant*> Properties;
975d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::vector<llvm::Constant*> OptionalProperties;
976d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
977d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Add all of the property methods need adding to the method list and to the
978d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // property metadata list.
979d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  for (ObjCContainerDecl::prop_iterator
980d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian         iter = PD->prop_begin(), endIter = PD->prop_end();
981d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian       iter != endIter ; iter++) {
982d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    std::vector<llvm::Constant*> Fields;
983d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    ObjCPropertyDecl *property = (*iter);
984d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
985d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Fields.push_back(MakeConstantString(property->getNameAsString()));
986d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Fields.push_back(llvm::ConstantInt::get(Int8Ty,
987d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian                property->getPropertyAttributes()));
988d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Fields.push_back(llvm::ConstantInt::get(Int8Ty, 0));
989d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
990d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      std::string TypeStr;
991d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Context.getObjCEncodingForMethodDecl(getter,TypeStr);
992d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
993d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodTypes.push_back(TypeEncoding);
994d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
995d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(TypeEncoding);
996d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    } else {
997d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
998d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
999d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
1000d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1001d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      std::string TypeStr;
1002d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1003d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1004d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodTypes.push_back(TypeEncoding);
1005d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1006d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(TypeEncoding);
1007d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    } else {
1008d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
1009d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
1010d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
1011d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if (property->getPropertyImplementation() == ObjCPropertyDecl::Optional) {
1012d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalProperties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1013d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    } else {
1014d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1015d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
1016d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  }
1017d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *PropertyArray = llvm::ConstantArray::get(
1018d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::ArrayType::get(PropertyMetadataTy, Properties.size()), Properties);
1019d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant* PropertyListInitFields[] =
1020d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1021d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1022d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *PropertyListInit =
10230d36dd28bb40b08a93d7f64196794882eee08ee8Nick Lewycky      llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
1024d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *PropertyList = new llvm::GlobalVariable(TheModule,
1025d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      PropertyListInit->getType(), false, llvm::GlobalValue::InternalLinkage,
1026d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      PropertyListInit, ".objc_property_list");
1027d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1028d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *OptionalPropertyArray =
1029d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::ConstantArray::get(llvm::ArrayType::get(PropertyMetadataTy,
1030d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          OptionalProperties.size()) , OptionalProperties);
1031d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant* OptionalPropertyListInitFields[] = {
1032d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::ConstantInt::get(IntTy, OptionalProperties.size()), NULLPtr,
1033d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalPropertyArray };
1034d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1035d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *OptionalPropertyListInit =
10360d36dd28bb40b08a93d7f64196794882eee08ee8Nick Lewycky      llvm::ConstantStruct::get(VMContext, OptionalPropertyListInitFields, 3, false);
1037d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *OptionalPropertyList = new llvm::GlobalVariable(TheModule,
1038d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          OptionalPropertyListInit->getType(), false,
1039d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          llvm::GlobalValue::InternalLinkage, OptionalPropertyListInit,
1040d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          ".objc_property_list");
1041d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
104220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Protocols are objects containing lists of the methods implemented and
104320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // protocols adopted.
104447a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType *ProtocolTy = llvm::StructType::get(VMContext, IdTy,
104520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      PtrToInt8Ty,
104620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      ProtocolList->getType(),
104720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      InstanceMethodList->getType(),
104820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      ClassMethodList->getType(),
1049d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalInstanceMethodList->getType(),
1050d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalClassMethodList->getType(),
1051d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      PropertyList->getType(),
1052d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      OptionalPropertyList->getType(),
105320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      NULL);
10541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  std::vector<llvm::Constant*> Elements;
105520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // The isa pointer must be set to a magic number so the runtime knows it's
105620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // the correct layout.
1057d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  int Version = CGM.getContext().getLangOptions().ObjCNonFragileABI ?
1058d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      NonFragileProtocolVersion : ProtocolVersion;
10593c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Elements.push_back(llvm::ConstantExpr::getIntToPtr(
1060d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Version), IdTy));
106120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(MakeConstantString(ProtocolName, ".objc_protocol_name"));
106220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(ProtocolList);
106320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(InstanceMethodList);
106420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(ClassMethodList);
1065d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(OptionalInstanceMethodList);
1066d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(OptionalClassMethodList);
1067d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(PropertyList);
1068d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(OptionalPropertyList);
10691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  ExistingProtocols[ProtocolName] =
10703c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson    llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolTy, Elements,
107120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov          ".objc_protocol"), IdTy);
107220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
1073d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanianvoid CGObjCGNU::GenerateProtocolHolderCategory(void) {
1074d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Collect information about instance methods
1075d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::SmallVector<Selector, 1> MethodSels;
1076d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::SmallVector<llvm::Constant*, 1> MethodTypes;
1077d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1078d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::vector<llvm::Constant*> Elements;
1079d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  const std::string ClassName = "__ObjC_Protocol_Holder_Ugly_Hack";
1080d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  const std::string CategoryName = "AnotherHack";
1081d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(MakeConstantString(CategoryName));
1082d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(MakeConstantString(ClassName));
1083d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Instance method list
1084d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1085d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          ClassName, CategoryName, MethodSels, MethodTypes, false), PtrTy));
1086d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Class method list
1087d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1088d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          ClassName, CategoryName, MethodSels, MethodTypes, true), PtrTy));
1089d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Protocol list
1090d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::ArrayType *ProtocolArrayTy = llvm::ArrayType::get(PtrTy,
1091d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      ExistingProtocols.size());
1092d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::StructType *ProtocolListTy = llvm::StructType::get(VMContext,
1093d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      PtrTy, //Should be a recurisve pointer, but it's always NULL here.
1094d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      LongTy,//FIXME: Should be size_t
1095d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      ProtocolArrayTy,
1096d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      NULL);
1097d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::vector<llvm::Constant*> ProtocolElements;
1098d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  for (llvm::StringMapIterator<llvm::Constant*> iter =
1099d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian       ExistingProtocols.begin(), endIter = ExistingProtocols.end();
1100d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian       iter != endIter ; iter++) {
1101d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    llvm::Constant *Ptr = llvm::ConstantExpr::getBitCast(iter->getValue(),
1102d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian            PtrTy);
1103d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    ProtocolElements.push_back(Ptr);
1104d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  }
1105d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant * ProtocolArray = llvm::ConstantArray::get(ProtocolArrayTy,
1106d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      ProtocolElements);
1107d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  ProtocolElements.clear();
1108d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  ProtocolElements.push_back(NULLPtr);
1109d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  ProtocolElements.push_back(llvm::ConstantInt::get(LongTy,
1110d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian              ExistingProtocols.size()));
1111d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  ProtocolElements.push_back(ProtocolArray);
1112d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Elements.push_back(llvm::ConstantExpr::getBitCast(MakeGlobal(ProtocolListTy,
1113d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian                  ProtocolElements, ".objc_protocol_list"), PtrTy));
1114d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  Categories.push_back(llvm::ConstantExpr::getBitCast(
1115d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
1116d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian            PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
1117d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian}
111820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
11197ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbarvoid CGObjCGNU::GenerateCategory(const ObjCCategoryImplDecl *OCD) {
11208ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner  std::string ClassName = OCD->getClassInterface()->getNameAsString();
11218ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner  std::string CategoryName = OCD->getNameAsString();
11227ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Collect information about instance methods
11237ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<Selector, 16> InstanceMethodSels;
11247ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
1125653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor  for (ObjCCategoryImplDecl::instmeth_iterator
112617945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis         iter = OCD->instmeth_begin(), endIter = OCD->instmeth_end();
1127653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor       iter != endIter ; iter++) {
11287ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    InstanceMethodSels.push_back((*iter)->getSelector());
11297ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    std::string TypeStr;
11307ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
11318a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
11327ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  }
11337ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
11347ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Collect information about class methods
11357ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<Selector, 16> ClassMethodSels;
11367ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
11371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  for (ObjCCategoryImplDecl::classmeth_iterator
113817945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis         iter = OCD->classmeth_begin(), endIter = OCD->classmeth_end();
1139653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor       iter != endIter ; iter++) {
11407ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    ClassMethodSels.push_back((*iter)->getSelector());
11417ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    std::string TypeStr;
11427ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    CGM.getContext().getObjCEncodingForMethodDecl(*iter,TypeStr);
11438a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    ClassMethodTypes.push_back(MakeConstantString(TypeStr));
11447ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  }
11457ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
11467ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Collect the names of referenced protocols
11477ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<std::string, 16> Protocols;
11487ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  const ObjCInterfaceDecl *ClassDecl = OCD->getClassInterface();
11497ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
11507ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
11517ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar       E = Protos.end(); I != E; ++I)
1152d9d22dd9c94618490dbffb0e2caf222530ca39d3Chris Lattner    Protocols.push_back((*I)->getNameAsString());
11537ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
115420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Elements;
115520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(MakeConstantString(CategoryName));
115620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(MakeConstantString(ClassName));
11571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Instance method list
11583c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1159a4210076fc1d7ac0a20b8b4a79e18a8ae33b9c69Chris Lattner          ClassName, CategoryName, InstanceMethodSels, InstanceMethodTypes,
116020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov          false), PtrTy));
116120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Class method list
11623c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Elements.push_back(llvm::ConstantExpr::getBitCast(GenerateMethodList(
1163a4210076fc1d7ac0a20b8b4a79e18a8ae33b9c69Chris Lattner          ClassName, CategoryName, ClassMethodSels, ClassMethodTypes, true),
116420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov        PtrTy));
116520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Protocol list
11663c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Elements.push_back(llvm::ConstantExpr::getBitCast(
116720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov        GenerateProtocolList(Protocols), PtrTy));
11683c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Categories.push_back(llvm::ConstantExpr::getBitCast(
11691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        MakeGlobal(llvm::StructType::get(VMContext, PtrToInt8Ty, PtrToInt8Ty,
117047a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson            PtrTy, PtrTy, PtrTy, NULL), Elements), PtrTy));
117120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
11727ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
1173d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanianllvm::Constant *CGObjCGNU::GeneratePropertyList(const ObjCImplementationDecl *OID,
1174d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        llvm::SmallVectorImpl<Selector> &InstanceMethodSels,
1175d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        llvm::SmallVectorImpl<llvm::Constant*> &InstanceMethodTypes) {
1176d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  ASTContext &Context = CGM.getContext();
1177d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  //
1178d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Property metadata: name, attributes, isSynthesized, setter name, setter
1179d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // types, getter name, getter types.
1180d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::StructType *PropertyMetadataTy = llvm::StructType::get(VMContext,
1181d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          PtrToInt8Ty, Int8Ty, Int8Ty, PtrToInt8Ty, PtrToInt8Ty, PtrToInt8Ty,
1182d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          PtrToInt8Ty, NULL);
1183d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::vector<llvm::Constant*> Properties;
1184d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1185d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1186d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Add all of the property methods need adding to the method list and to the
1187d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // property metadata list.
1188d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  for (ObjCImplDecl::propimpl_iterator
1189d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian         iter = OID->propimpl_begin(), endIter = OID->propimpl_end();
1190d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian       iter != endIter ; iter++) {
1191d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    std::vector<llvm::Constant*> Fields;
1192d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    ObjCPropertyDecl *property = (*iter)->getPropertyDecl();
1193d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1194d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Fields.push_back(MakeConstantString(property->getNameAsString()));
1195d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1196d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian                property->getPropertyAttributes()));
1197d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Fields.push_back(llvm::ConstantInt::get(Int8Ty,
1198d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian                (*iter)->getPropertyImplementation() ==
1199d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian                ObjCPropertyImplDecl::Synthesize));
1200d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if (ObjCMethodDecl *getter = property->getGetterMethodDecl()) {
1201d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodSels.push_back(getter->getSelector());
1202d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      std::string TypeStr;
1203d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Context.getObjCEncodingForMethodDecl(getter,TypeStr);
1204d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1205d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodTypes.push_back(TypeEncoding);
1206d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(MakeConstantString(getter->getSelector().getAsString()));
1207d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(TypeEncoding);
1208d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    } else {
1209d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
1210d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
1211d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
1212d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    if (ObjCMethodDecl *setter = property->getSetterMethodDecl()) {
1213d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodSels.push_back(setter->getSelector());
1214d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      std::string TypeStr;
1215d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Context.getObjCEncodingForMethodDecl(setter,TypeStr);
1216d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::Constant *TypeEncoding = MakeConstantString(TypeStr);
1217d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      InstanceMethodTypes.push_back(TypeEncoding);
1218d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(MakeConstantString(setter->getSelector().getAsString()));
1219d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(TypeEncoding);
1220d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    } else {
1221d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
1222d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Fields.push_back(NULLPtr);
1223d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    }
1224d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Properties.push_back(llvm::ConstantStruct::get(PropertyMetadataTy, Fields));
1225d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  }
1226d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::ArrayType *PropertyArrayTy =
1227d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::ArrayType::get(PropertyMetadataTy, Properties.size());
1228d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *PropertyArray = llvm::ConstantArray::get(PropertyArrayTy,
1229d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          Properties);
1230d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant* PropertyListInitFields[] =
1231d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    {llvm::ConstantInt::get(IntTy, Properties.size()), NULLPtr, PropertyArray};
1232d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1233d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *PropertyListInit =
12340d36dd28bb40b08a93d7f64196794882eee08ee8Nick Lewycky      llvm::ConstantStruct::get(VMContext, PropertyListInitFields, 3, false);
1235d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  return new llvm::GlobalVariable(TheModule, PropertyListInit->getType(), false,
1236d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          llvm::GlobalValue::InternalLinkage, PropertyListInit,
1237d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          ".objc_property_list");
1238d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian}
1239d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
12407ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbarvoid CGObjCGNU::GenerateClass(const ObjCImplementationDecl *OID) {
12417ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  ASTContext &Context = CGM.getContext();
12427ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
12437ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Get the superclass name.
12441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const ObjCInterfaceDecl * SuperClassDecl =
12457ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    OID->getClassInterface()->getSuperClass();
12468ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner  std::string SuperClassName;
12472a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  if (SuperClassDecl) {
12488ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner    SuperClassName = SuperClassDecl->getNameAsString();
12492a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner    EmitClassRef(SuperClassName);
12502a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  }
12517ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
12527ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Get the class name
125309dc6660487fc2f4a4bb2032e30123d3e0da7230Chris Lattner  ObjCInterfaceDecl *ClassDecl =
125409dc6660487fc2f4a4bb2032e30123d3e0da7230Chris Lattner    const_cast<ObjCInterfaceDecl *>(OID->getClassInterface());
12558ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner  std::string ClassName = ClassDecl->getNameAsString();
12562a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  // Emit the symbol that is used to generate linker errors if this class is
12572a8e4e1d8548dc3e30d7c9ba92127c7884a11448Chris Lattner  // referenced in other modules but not declared.
1258c51db23f58bd887b16cf91554aa86c430a2496e7Fariborz Jahanian  std::string classSymbolName = "__objc_class_name_" + ClassName;
12591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  if (llvm::GlobalVariable *symbol =
1260c51db23f58bd887b16cf91554aa86c430a2496e7Fariborz Jahanian      TheModule.getGlobalVariable(classSymbolName)) {
12614a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson    symbol->setInitializer(llvm::ConstantInt::get(LongTy, 0));
1262c51db23f58bd887b16cf91554aa86c430a2496e7Fariborz Jahanian  } else {
12631c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson    new llvm::GlobalVariable(TheModule, LongTy, false,
12644a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson    llvm::GlobalValue::ExternalLinkage, llvm::ConstantInt::get(LongTy, 0),
12651c431b323d776362490bbf7cc796b74fedaf19f2Owen Anderson    classSymbolName);
1266c51db23f58bd887b16cf91554aa86c430a2496e7Fariborz Jahanian  }
12671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
12682bebbf0acee55404de4b8846713b64429e744e8fDaniel Dunbar  // Get the size of instances.
12692bebbf0acee55404de4b8846713b64429e744e8fDaniel Dunbar  int instanceSize = Context.getASTObjCImplementationLayout(OID).getSize() / 8;
12707ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
12717ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Collect information about instance variables.
12727ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> IvarNames;
12737ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> IvarTypes;
12747ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> IvarOffsets;
12751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
1276d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  std::vector<llvm::Constant*> IvarOffsetValues;
1277d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
12781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  int superInstanceSize = !SuperClassDecl ? 0 :
12799cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    Context.getASTObjCInterfaceLayout(SuperClassDecl).getSize() / 8;
12809cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  // For non-fragile ivars, set the instance size to 0 - {the size of just this
12819cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  // class}.  The runtime will then set this to the correct value on load.
12829cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
12839cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    instanceSize = 0 - (instanceSize - superInstanceSize);
12849cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  }
12857ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
12867ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar      endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
12877ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar      // Store the name
12888a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      IvarNames.push_back(MakeConstantString((*iter)->getNameAsString()));
12897ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar      // Get the type encoding for this ivar
12907ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar      std::string TypeStr;
12910d504c1da852e58ff802545c823ecff3b6c654b8Daniel Dunbar      Context.getObjCEncodingForType((*iter)->getType(), TypeStr);
12928a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      IvarTypes.push_back(MakeConstantString(TypeStr));
12937ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar      // Get the offset
1294d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      uint64_t BaseOffset = ComputeIvarBaseOffset(CGM, ClassDecl, *iter);
1295aecbf24a6bad049437f8fec97e557d414003a3dbDavid Chisnall      uint64_t Offset = BaseOffset;
12969cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian      if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
1297d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        Offset = BaseOffset - superInstanceSize;
12989cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian      }
12997ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar      IvarOffsets.push_back(
13000032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson          llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset));
1301d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      IvarOffsetValues.push_back(new llvm::GlobalVariable(TheModule, IntTy,
1302d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          false, llvm::GlobalValue::ExternalLinkage,
1303d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          llvm::ConstantInt::get(IntTy, BaseOffset),
1304d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          "__objc_ivar_offset_value_" + ClassName +"." +
1305d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          (*iter)->getNameAsString()));
13067ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  }
1307d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *IvarOffsetArrayInit =
1308d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      llvm::ConstantArray::get(llvm::ArrayType::get(PtrToIntTy,
1309d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian                  IvarOffsetValues.size()), IvarOffsetValues);
1310d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::GlobalVariable *IvarOffsetArray = new llvm::GlobalVariable(TheModule,
1311d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          IvarOffsetArrayInit->getType(), false,
1312d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          llvm::GlobalValue::InternalLinkage, IvarOffsetArrayInit,
1313d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          ".ivar.offsets");
13147ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
13157ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Collect information about instance methods
13167ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<Selector, 16> InstanceMethodSels;
13177ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> InstanceMethodTypes;
13181eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  for (ObjCImplementationDecl::instmeth_iterator
131917945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis         iter = OID->instmeth_begin(), endIter = OID->instmeth_end();
1320653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor       iter != endIter ; iter++) {
13217ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    InstanceMethodSels.push_back((*iter)->getSelector());
13227ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    std::string TypeStr;
13237ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
13248a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    InstanceMethodTypes.push_back(MakeConstantString(TypeStr));
13257ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  }
1326d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1327d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  llvm::Constant *Properties = GeneratePropertyList(OID, InstanceMethodSels,
1328d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian          InstanceMethodTypes);
1329d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
13307ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
13317ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Collect information about class methods
13327ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<Selector, 16> ClassMethodSels;
13337ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<llvm::Constant*, 16> ClassMethodTypes;
1334653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor  for (ObjCImplementationDecl::classmeth_iterator
133517945a0f64fe03ff6ec0c2146005a87636e3ac12Argyrios Kyrtzidis         iter = OID->classmeth_begin(), endIter = OID->classmeth_end();
1336653f1b1bf293a9bd96fd4dd6372e779cc7af1597Douglas Gregor       iter != endIter ; iter++) {
13377ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    ClassMethodSels.push_back((*iter)->getSelector());
13387ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    std::string TypeStr;
13397ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar    Context.getObjCEncodingForMethodDecl((*iter),TypeStr);
13408a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    ClassMethodTypes.push_back(MakeConstantString(TypeStr));
13417ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  }
13427ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  // Collect the names of referenced protocols
13437ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  llvm::SmallVector<std::string, 16> Protocols;
13447ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  const ObjCList<ObjCProtocolDecl> &Protos =ClassDecl->getReferencedProtocols();
13457ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar  for (ObjCList<ObjCProtocolDecl>::iterator I = Protos.begin(),
13467ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar       E = Protos.end(); I != E; ++I)
1347d9d22dd9c94618490dbffb0e2caf222530ca39d3Chris Lattner    Protocols.push_back((*I)->getNameAsString());
13487ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
13497ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
13507ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
135120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Get the superclass pointer.
135220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *SuperClass;
13538ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner  if (!SuperClassName.empty()) {
135420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    SuperClass = MakeConstantString(SuperClassName, ".super_class_name");
135520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  } else {
135603e205031b08669f05c41eed5b896fc94c4a12bbOwen Anderson    SuperClass = llvm::ConstantPointerNull::get(PtrToInt8Ty);
135720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
135820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Empty vector used to construct empty method lists
135920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::SmallVector<llvm::Constant*, 1>  empty;
136020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Generate the method and instance variable lists
136120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *MethodList = GenerateMethodList(ClassName, "",
1362a4210076fc1d7ac0a20b8b4a79e18a8ae33b9c69Chris Lattner      InstanceMethodSels, InstanceMethodTypes, false);
136320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *ClassMethodList = GenerateMethodList(ClassName, "",
1364a4210076fc1d7ac0a20b8b4a79e18a8ae33b9c69Chris Lattner      ClassMethodSels, ClassMethodTypes, true);
136520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *IvarList = GenerateIvarList(IvarNames, IvarTypes,
136620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      IvarOffsets);
13671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Irrespective of whether we are compiling for a fragile or non-fragile ABI,
13688a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // we emit a symbol containing the offset for each ivar in the class.  This
13698a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // allows code compiled for the non-Fragile ABI to inherit from code compiled
13708a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // for the legacy ABI, without causing problems.  The converse is also
13718a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // possible, but causes all ivar accesses to be fragile.
13728a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  int i = 0;
13738a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // Offset pointer for getting at the correct field in the ivar list when
13748a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // setting up the alias.  These are: The base address for the global, the
13758a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // ivar array (second field), the ivar in this list (set for each ivar), and
13768a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  // the offset (third field in ivar structure)
13778a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  const llvm::Type *IndexTy = llvm::Type::getInt32Ty(VMContext);
13788a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  llvm::Constant *offsetPointerIndexes[] = {Zeros[0],
13791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump      llvm::ConstantInt::get(IndexTy, 1), 0,
13808a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      llvm::ConstantInt::get(IndexTy, 2) };
13818a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall
13828a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  for (ObjCInterfaceDecl::ivar_iterator iter = ClassDecl->ivar_begin(),
13838a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      endIter = ClassDecl->ivar_end() ; iter != endIter ; iter++) {
13848a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      const std::string Name = "__objc_ivar_offset_" + ClassName + '.'
13858a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          +(*iter)->getNameAsString();
13868a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      offsetPointerIndexes[2] = llvm::ConstantInt::get(IndexTy, i++);
13878a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      // Get the correct ivar field
13888a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      llvm::Constant *offsetValue = llvm::ConstantExpr::getGetElementPtr(
13898a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall              IvarList, offsetPointerIndexes, 4);
13908a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      // Get the existing alias, if one exists.
13918a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      llvm::GlobalVariable *offset = TheModule.getNamedGlobal(Name);
13928a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      if (offset) {
13938a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          offset->setInitializer(offsetValue);
13948a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          // If this is the real definition, change its linkage type so that
13958a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          // different modules will use this one, rather than their private
13968a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          // copy.
13978a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          offset->setLinkage(llvm::GlobalValue::ExternalLinkage);
13988a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      } else {
13998a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          // Add a new alias if there isn't one already.
14008a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall          offset = new llvm::GlobalVariable(TheModule, offsetValue->getType(),
14018a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall                  false, llvm::GlobalValue::ExternalLinkage, offsetValue, Name);
14028a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      }
14038a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  }
140420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  //Generate metaclass for class methods
140520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *MetaClassStruct = GenerateClassStructure(NULLPtr,
14061804463ee790ac3cc73f72597ccb38b983f7d263David Chisnall      NULLPtr, 0x12L, ClassName.c_str(), 0, Zeros[0], GenerateIvarList(
1407d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        empty, empty, empty), ClassMethodList, NULLPtr, NULLPtr, NULLPtr);
14085efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar
140920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Generate the class structure
14108ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner  llvm::Constant *ClassStruct =
1411d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    GenerateClassStructure(MetaClassStruct, SuperClass, 0x11L,
14128ec03f58c33c33a917f54bb7f2cd61b6d7ffe0caChris Lattner                           ClassName.c_str(), 0,
14134a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson      llvm::ConstantInt::get(LongTy, instanceSize), IvarList,
1414d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      MethodList, GenerateProtocolList(Protocols), IvarOffsetArray,
1415d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian      Properties);
14165efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar
14175efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  // Resolve the class aliases, if they exist.
14185efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  if (ClassPtrAlias) {
14195efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar    ClassPtrAlias->setAliasee(
14203c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson        llvm::ConstantExpr::getBitCast(ClassStruct, IdTy));
14215efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar    ClassPtrAlias = 0;
14225efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  }
14235efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  if (MetaClassPtrAlias) {
14245efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar    MetaClassPtrAlias->setAliasee(
14253c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson        llvm::ConstantExpr::getBitCast(MetaClassStruct, IdTy));
14265efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar    MetaClassPtrAlias = 0;
14275efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar  }
14285efccb1ab81a029ddeb3ea18e239d0ddc124ec2bDaniel Dunbar
142920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Add class structure to list to be added to the symtab later
14303c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  ClassStruct = llvm::ConstantExpr::getBitCast(ClassStruct, PtrToInt8Ty);
143120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Classes.push_back(ClassStruct);
143220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
143320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
1434c38e9affd4519ea199af22419c8c794973cc4b23Fariborz Jahanian
14351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stumpllvm::Function *CGObjCGNU::ModuleInitFunction() {
143620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Only emit an ObjC load function if no Objective-C stuff has been called
143720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  if (Classes.empty() && Categories.empty() && ConstantStrings.empty() &&
143820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      ExistingProtocols.empty() && TypedSelectors.empty() &&
14395f58b911304f7d114f929d1726a09cd1c828a5ddAnton Korobeynikov      UntypedSelectors.empty())
144020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    return NULL;
14411b8956e1ce4c5770982d6e59edb16979b500677aEli Friedman
1442d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  // Add all referenced protocols to a category.
1443d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian  GenerateProtocolHolderCategory();
1444d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian
1445e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::StructType *SelStructTy = dyn_cast<llvm::StructType>(
1446e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner          SelectorTy->getElementType());
1447e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  const llvm::Type *SelStructPtrTy = SelectorTy;
1448e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  bool isSelOpaque = false;
1449e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  if (SelStructTy == 0) {
145047a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson    SelStructTy = llvm::StructType::get(VMContext, PtrToInt8Ty,
145147a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson                                        PtrToInt8Ty, NULL);
145296e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson    SelStructPtrTy = llvm::PointerType::getUnqual(SelStructTy);
1453e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    isSelOpaque = true;
1454e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  }
1455e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner
14561b8956e1ce4c5770982d6e59edb16979b500677aEli Friedman  // Name the ObjC types to make the IR a bit easier to read
1457e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner  TheModule.addTypeName(".objc_selector", SelStructPtrTy);
14581b8956e1ce4c5770982d6e59edb16979b500677aEli Friedman  TheModule.addTypeName(".objc_id", IdTy);
14591b8956e1ce4c5770982d6e59edb16979b500677aEli Friedman  TheModule.addTypeName(".objc_imp", IMPTy);
14601b8956e1ce4c5770982d6e59edb16979b500677aEli Friedman
146120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Elements;
146271238f67c5df13eef19450d66b7a7ab28377192aChris Lattner  llvm::Constant *Statics = NULLPtr;
146320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Generate statics list:
146471238f67c5df13eef19450d66b7a7ab28377192aChris Lattner  if (ConstantStrings.size()) {
146596e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson    llvm::ArrayType *StaticsArrayTy = llvm::ArrayType::get(PtrToInt8Ty,
146671238f67c5df13eef19450d66b7a7ab28377192aChris Lattner        ConstantStrings.size() + 1);
146771238f67c5df13eef19450d66b7a7ab28377192aChris Lattner    ConstantStrings.push_back(NULLPtr);
14688a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall
14691b0969590e712d7d52fc9c0d43d3ab85c36d07a6Daniel Dunbar    llvm::StringRef StringClass = CGM.getLangOptions().ObjCConstantStringClass;
14701b0969590e712d7d52fc9c0d43d3ab85c36d07a6Daniel Dunbar    if (StringClass.empty()) StringClass = "NXConstantString";
14718a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    Elements.push_back(MakeConstantString(StringClass,
14728a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall                ".objc_static_class_name"));
14737db6d838aad4083fe86d7bf703a75fe6e8a17856Owen Anderson    Elements.push_back(llvm::ConstantArray::get(StaticsArrayTy,
147471238f67c5df13eef19450d66b7a7ab28377192aChris Lattner       ConstantStrings));
14751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::StructType *StaticsListTy =
147647a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson      llvm::StructType::get(VMContext, PtrToInt8Ty, StaticsArrayTy, NULL);
1477a1cf15f4680e5cf39e72e28c5ea854fcba792e84Owen Anderson    llvm::Type *StaticsListPtrTy =
147896e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      llvm::PointerType::getUnqual(StaticsListTy);
147971238f67c5df13eef19450d66b7a7ab28377192aChris Lattner    Statics = MakeGlobal(StaticsListTy, Elements, ".objc_statics");
14801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::ArrayType *StaticsListArrayTy =
148196e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      llvm::ArrayType::get(StaticsListPtrTy, 2);
148271238f67c5df13eef19450d66b7a7ab28377192aChris Lattner    Elements.clear();
148371238f67c5df13eef19450d66b7a7ab28377192aChris Lattner    Elements.push_back(Statics);
1484c9c88b4159791c48e486ca94e3743b5979e2b7a6Owen Anderson    Elements.push_back(llvm::Constant::getNullValue(StaticsListPtrTy));
148571238f67c5df13eef19450d66b7a7ab28377192aChris Lattner    Statics = MakeGlobal(StaticsListArrayTy, Elements, ".objc_statics_ptr");
14863c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson    Statics = llvm::ConstantExpr::getBitCast(Statics, PtrTy);
148771238f67c5df13eef19450d66b7a7ab28377192aChris Lattner  }
148820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Array of classes, categories, and constant objects
148996e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  llvm::ArrayType *ClassListTy = llvm::ArrayType::get(PtrToInt8Ty,
149020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      Classes.size() + Categories.size()  + 2);
14911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  llvm::StructType *SymTabTy = llvm::StructType::get(VMContext,
149247a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson                                                     LongTy, SelStructPtrTy,
14930032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson                                                     llvm::Type::getInt16Ty(VMContext),
14940032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson                                                     llvm::Type::getInt16Ty(VMContext),
1495630404b16e886238ed68a1abbfe2e548f43aa664Chris Lattner                                                     ClassListTy, NULL);
149620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
149720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.clear();
149820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Pointer to an array of selectors used in this module.
149920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::vector<llvm::Constant*> Selectors;
150020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
150120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov     iter = TypedSelectors.begin(), iterEnd = TypedSelectors.end();
150220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov     iter != iterEnd ; ++iter) {
1503d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian    Elements.push_back(ExportUniqueString(iter->first.first, ".objc_sel_name"));
1504a7c5b0832cc78efa899463b8faa573a1473d53d2David Chisnall    Elements.push_back(MakeConstantString(iter->first.second,
1505630404b16e886238ed68a1abbfe2e548f43aa664Chris Lattner                                          ".objc_sel_types"));
150608e252425ca2cbdc44ba65d9a657ed5398014e36Owen Anderson    Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
150720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.clear();
150820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
150920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (llvm::StringMap<llvm::GlobalAlias*>::iterator
151020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      iter = UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
1511630404b16e886238ed68a1abbfe2e548f43aa664Chris Lattner      iter != iterEnd; ++iter) {
151220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.push_back(
1513d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        ExportUniqueString(iter->getKeyData(), ".objc_sel_name"));
151420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.push_back(NULLPtr);
151508e252425ca2cbdc44ba65d9a657ed5398014e36Owen Anderson    Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
151620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    Elements.clear();
151720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
151820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(NULLPtr);
151920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(NULLPtr);
152008e252425ca2cbdc44ba65d9a657ed5398014e36Owen Anderson  Selectors.push_back(llvm::ConstantStruct::get(SelStructTy, Elements));
152120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.clear();
152220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Number of static selectors
15234a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Elements.push_back(llvm::ConstantInt::get(LongTy, Selectors.size() ));
152420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *SelectorList = MakeGlobal(
152596e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson          llvm::ArrayType::get(SelStructTy, Selectors.size()), Selectors,
152620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov          ".objc_selector_list");
15271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Elements.push_back(llvm::ConstantExpr::getBitCast(SelectorList,
1528e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    SelStructPtrTy));
152920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
153020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Now that all of the static selectors exist, create pointers to them.
153120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  int index = 0;
153220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (std::map<TypedSelector, llvm::GlobalAlias*>::iterator
153320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov     iter=TypedSelectors.begin(), iterEnd =TypedSelectors.end();
153420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov     iter != iterEnd; ++iter) {
153520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *Idxs[] = {Zeros[0],
15360032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson      llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
15376d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar    llvm::Constant *SelPtr = new llvm::GlobalVariable(TheModule, SelStructPtrTy,
15386d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar        true, llvm::GlobalValue::InternalLinkage,
15396d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar        llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
15406d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar        ".objc_sel_ptr");
1541e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    // If selectors are defined as an opaque type, cast the pointer to this
1542e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    // type.
1543e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    if (isSelOpaque) {
15446d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar      SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
15456d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar        llvm::PointerType::getUnqual(SelectorTy));
1546e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    }
15476d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar    (*iter).second->setAliasee(SelPtr);
154820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
154920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  for (llvm::StringMap<llvm::GlobalAlias*>::iterator
155020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      iter=UntypedSelectors.begin(), iterEnd = UntypedSelectors.end();
155120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      iter != iterEnd; iter++) {
155220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov    llvm::Constant *Idxs[] = {Zeros[0],
15530032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson      llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), index++), Zeros[0]};
15546d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar    llvm::Constant *SelPtr = new llvm::GlobalVariable
15556d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar      (TheModule, SelStructPtrTy,
15566d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar       true, llvm::GlobalValue::InternalLinkage,
15576d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar       llvm::ConstantExpr::getGetElementPtr(SelectorList, Idxs, 2),
15586d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar       ".objc_sel_ptr");
1559e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    // If selectors are defined as an opaque type, cast the pointer to this
1560e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    // type.
1561e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    if (isSelOpaque) {
15626d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar      SelPtr = llvm::ConstantExpr::getBitCast(SelPtr,
15636d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar        llvm::PointerType::getUnqual(SelectorTy));
1564e160c9b09d2f5098dfdcca99b9c6485487e0c252Chris Lattner    }
15656d5a1c28593443f3973ef38f8fa042d59182412dDaniel Dunbar    (*iter).second->setAliasee(SelPtr);
156620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  }
156720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Number of classes defined.
15681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
156920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov        Classes.size()));
157020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Number of categories defined
15711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Elements.push_back(llvm::ConstantInt::get(llvm::Type::getInt16Ty(VMContext),
157220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov        Categories.size()));
157320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Create an array of classes, then categories, then static object instances
157420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Classes.insert(Classes.end(), Categories.begin(), Categories.end());
157520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  //  NULL-terminated list of static object instances (mainly constant strings)
157620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Classes.push_back(Statics);
157720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Classes.push_back(NULLPtr);
15787db6d838aad4083fe86d7bf703a75fe6e8a17856Owen Anderson  llvm::Constant *ClassList = llvm::ConstantArray::get(ClassListTy, Classes);
157920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(ClassList);
15801eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // Construct the symbol table
158120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Constant *SymTab= MakeGlobal(SymTabTy, Elements);
158220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
158320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // The symbol table is contained in a module which has some version-checking
158420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // constants
158547a434ff3d49e7906eda88e8e8242e4297725b32Owen Anderson  llvm::StructType * ModuleTy = llvm::StructType::get(VMContext, LongTy, LongTy,
158696e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      PtrToInt8Ty, llvm::PointerType::getUnqual(SymTabTy), NULL);
158720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.clear();
158820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Runtime version used for compatibility checking.
15899cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  if (CGM.getContext().getLangOptions().ObjCNonFragileABI) {
15901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    Elements.push_back(llvm::ConstantInt::get(LongTy,
1591d9a1db3a4ded1c8f0daa43c0d6167576e2766453Fariborz Jahanian        NonFragileRuntimeVersion));
15929cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  } else {
15934a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson    Elements.push_back(llvm::ConstantInt::get(LongTy, RuntimeVersion));
15949cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  }
159591a0b51438e5259a79d68d7450c453827beed9e2Fariborz Jahanian  // sizeof(ModuleTy)
159691a0b51438e5259a79d68d7450c453827beed9e2Fariborz Jahanian  llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
15974a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  Elements.push_back(llvm::ConstantInt::get(LongTy,
1598a1cf15f4680e5cf39e72e28c5ea854fcba792e84Owen Anderson                     td.getTypeSizeInBits(ModuleTy)/8));
159920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  //FIXME: Should be the path to the file where this module was declared
160020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(NULLPtr);
160120ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Elements.push_back(SymTab);
160220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Value *Module = MakeGlobal(ModuleTy, Elements);
160320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
160420ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // Create the load function calling the runtime entry point with the module
160520ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  // structure
160620ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  llvm::Function * LoadFunction = llvm::Function::Create(
16070032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson      llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), false),
160820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      llvm::GlobalValue::InternalLinkage, ".objc_load_function",
160920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      &TheModule);
16100032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson  llvm::BasicBlock *EntryBB =
16110032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson      llvm::BasicBlock::Create(VMContext, "entry", LoadFunction);
1612a1cf15f4680e5cf39e72e28c5ea854fcba792e84Owen Anderson  CGBuilderTy Builder(VMContext);
161320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Builder.SetInsertPoint(EntryBB);
161426c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian
161526c8294dacc2efc83d8d560593ff863e7a32a48fFariborz Jahanian  std::vector<const llvm::Type*> Params(1,
161696e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson      llvm::PointerType::getUnqual(ModuleTy));
161796e0fc726c6fe7538522c60743705d5e696b40afOwen Anderson  llvm::Value *Register = CGM.CreateRuntimeFunction(llvm::FunctionType::get(
16180032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson        llvm::Type::getVoidTy(VMContext), Params, true), "__objc_exec_class");
161920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Builder.CreateCall(Register, Module);
162020ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  Builder.CreateRetVoid();
16219cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian
162220ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  return LoadFunction;
162320ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov}
16247ded7f4983dc4a20561db7a8d02c6b2435030961Daniel Dunbar
1625679a502d462ef819e6175b58e255ca3f3391e7cfFariborz Jahanianllvm::Function *CGObjCGNU::GenerateMethod(const ObjCMethodDecl *OMD,
16261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                          const ObjCContainerDecl *CD) {
16271eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const ObjCCategoryImplDecl *OCD =
16283e0a540b6d846178857289ec0eb8470a278d11a3Steve Naroff    dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext());
1629077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner  std::string CategoryName = OCD ? OCD->getNameAsString() : "";
1630077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner  std::string ClassName = OMD->getClassInterface()->getNameAsString();
1631077bf5e2f48acfa9e7d69429b6e4ba86ea14896dChris Lattner  std::string MethodName = OMD->getSelector().getAsString();
1632f8d49f64ef6ab7e632717a31631fc289aab69428Douglas Gregor  bool isClassMethod = !OMD->isInstanceMethod();
1633391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner
1634541b63b1a9db77e4a8670e9823711c2c12e58afbDaniel Dunbar  CodeGenTypes &Types = CGM.getTypes();
16351eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const llvm::FunctionType *MethodTy =
1636541b63b1a9db77e4a8670e9823711c2c12e58afbDaniel Dunbar    Types.GetFunctionType(Types.getFunctionInfo(OMD), OMD->isVariadic());
163720ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov  std::string FunctionName = SymbolNameForMethod(ClassName, CategoryName,
163820ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov      MethodName, isClassMethod);
163920ff3108fcd2c3bd734dc79efc22ebaa090abd41Anton Korobeynikov
1640d6c93d703541c992e06eb9a59a2d826a30da65b2Daniel Dunbar  llvm::Function *Method
16411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    = llvm::Function::Create(MethodTy,
16421eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                             llvm::GlobalValue::InternalLinkage,
16431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                             FunctionName,
16441eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                             &TheModule);
1645391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner  return Method;
1646391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner}
1647391d77a26382dddf25da73e29fc1fa5aaaea4c6fChris Lattner
164849f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbarllvm::Function *CGObjCGNU::GetPropertyGetFunction() {
16491eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  std::vector<const llvm::Type*> Params;
16501eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const llvm::Type *BoolTy =
16511eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
16521eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(IdTy);
16531eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(SelectorTy);
16541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
16551eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(LongTy);
16561eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(BoolTy);
16571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // void objc_getProperty (id, SEL, ptrdiff_t, bool)
16581eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const llvm::FunctionType *FTy =
16591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::FunctionType::get(IdTy, Params, false);
16601eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
16611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                                        "objc_getProperty"));
166249f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbar}
166349f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbar
166449f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbarllvm::Function *CGObjCGNU::GetPropertySetFunction() {
16651eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  std::vector<const llvm::Type*> Params;
16661eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const llvm::Type *BoolTy =
16671eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    CGM.getTypes().ConvertType(CGM.getContext().BoolTy);
16681eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(IdTy);
16691eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(SelectorTy);
16701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // FIXME: Using LongTy for ptrdiff_t is probably broken on Win64
16711eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(LongTy);
16721eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(IdTy);
16731eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(BoolTy);
16741eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  Params.push_back(BoolTy);
16751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  // void objc_setProperty (id, SEL, ptrdiff_t, id, bool, bool)
16761eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  const llvm::FunctionType *FTy =
16771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Params, false);
16781eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  return cast<llvm::Function>(CGM.CreateRuntimeFunction(FTy,
16791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                                        "objc_setProperty"));
168049f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbar}
168149f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbar
1682309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbarllvm::Constant *CGObjCGNU::EnumerationMutationFunction() {
1683309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar  CodeGen::CodeGenTypes &Types = CGM.getTypes();
1684309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar  ASTContext &Ctx = CGM.getContext();
1685309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar  // void objc_enumerationMutation (id)
1686309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar  llvm::SmallVector<QualType,16> Params;
16870f436560640a1cff5b6d96f80f540770f139453fDavid Chisnall  Params.push_back(ASTIdTy);
1688309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar  const llvm::FunctionType *FTy =
1689309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar    Types.GetFunctionType(Types.getFunctionInfo(Ctx.VoidTy, Params), false);
1690309a4368cd299cff30ab22709e2b772cf8059e45Daniel Dunbar  return CGM.CreateRuntimeFunction(FTy, "objc_enumerationMutation");
16912abd89c039e835e84519a4cd8a7495899a70153dAnders Carlsson}
16922abd89c039e835e84519a4cd8a7495899a70153dAnders Carlsson
1693bd71be4683c195260d5245118b1e13e6b2e20504Fariborz Jahanianvoid CGObjCGNU::EmitTryOrSynchronizedStmt(CodeGen::CodeGenFunction &CGF,
1694bd71be4683c195260d5245118b1e13e6b2e20504Fariborz Jahanian                                          const Stmt &S) {
16955dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Pointer to the personality function
16965dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::Constant *Personality =
16970032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson    CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getInt32Ty(VMContext),
1698b59761b2530bf54926bf67de11956f3820ac38d9Chris Lattner          true),
16995dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        "__gnu_objc_personality_v0");
17003c4972def972f8ca44dcd0561779a12aaa6fec97Owen Anderson  Personality = llvm::ConstantExpr::getBitCast(Personality, PtrTy);
17015dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  std::vector<const llvm::Type*> Params;
17025dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  Params.push_back(PtrTy);
17035dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::Value *RethrowFn =
17040032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson    CGM.CreateRuntimeFunction(llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext),
170590665bd87f4ef7c230742e038cd0bad68e40fc78David Chisnall          Params, false), "_Unwind_Resume");
17065dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17075dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  bool isTry = isa<ObjCAtTryStmt>(S);
17085dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::BasicBlock *TryBlock = CGF.createBasicBlock("try");
17095dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::BasicBlock *PrevLandingPad = CGF.getInvokeDest();
17105dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::BasicBlock *TryHandler = CGF.createBasicBlock("try.handler");
1711d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  llvm::BasicBlock *CatchInCatch = CGF.createBasicBlock("catch.rethrow");
17125dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::BasicBlock *FinallyBlock = CGF.createBasicBlock("finally");
17135dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::BasicBlock *FinallyRethrow = CGF.createBasicBlock("finally.throw");
17145dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::BasicBlock *FinallyEnd = CGF.createBasicBlock("finally.end");
17155dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17160faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  // @synchronized()
17175dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (!isTry) {
1718d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    std::vector<const llvm::Type*> Args(1, IdTy);
1719d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    llvm::FunctionType *FTy =
17200032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson      llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
1721d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    llvm::Value *SyncEnter = CGM.CreateRuntimeFunction(FTy, "objc_sync_enter");
17221eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::Value *SyncArg =
1723d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner      CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1724d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1725d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    CGF.Builder.CreateCall(SyncEnter, SyncArg);
17265dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  }
17275dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
1728d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner
17295dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Push an EH context entry, used for handling rethrows and jumps
17305dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // through finally.
17315dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.PushCleanupBlock(FinallyBlock);
17325dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17335dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Emit the statements in the @try {} block
17345dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.setInvokeDest(TryHandler);
17355dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17365dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.EmitBlock(TryBlock);
17371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  CGF.EmitStmt(isTry ? cast<ObjCAtTryStmt>(S).getTryBody()
17385dc0867af17b3bd6f567897433853b2b4767446cChris Lattner                     : cast<ObjCAtSynchronizedStmt>(S).getSynchBody());
17395dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17405dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Jump to @finally if there is no exception
17415dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.EmitBranchThroughCleanup(FinallyEnd);
17425dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17435dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Emit the handlers
17445dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.EmitBlock(TryHandler);
17455dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
1746bb422ad4f449cd195a141309a886ff467baac5b2Chris Lattner  // Get the correct versions of the exception handling intrinsics
1747bb422ad4f449cd195a141309a886ff467baac5b2Chris Lattner  llvm::TargetData td = llvm::TargetData::TargetData(&TheModule);
17481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  llvm::Value *llvm_eh_exception =
17495dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_exception);
1750790b20ed2cec72b4a79d1829ffc2c0339acbdec3Duncan Sands  llvm::Value *llvm_eh_selector =
1751790b20ed2cec72b4a79d1829ffc2c0339acbdec3Duncan Sands    CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_selector);
1752790b20ed2cec72b4a79d1829ffc2c0339acbdec3Duncan Sands  llvm::Value *llvm_eh_typeid_for =
1753790b20ed2cec72b4a79d1829ffc2c0339acbdec3Duncan Sands    CGF.CGM.getIntrinsic(llvm::Intrinsic::eh_typeid_for);
17545dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17555dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Exception object
17565dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::Value *Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
17575dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::Value *RethrowPtr = CGF.CreateTempAlloca(Exc->getType(), "_rethrow");
17585dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17595dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::SmallVector<llvm::Value*, 8> ESelArgs;
17605dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::SmallVector<std::pair<const ParmVarDecl*, const Stmt*>, 8> Handlers;
17615dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17625dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  ESelArgs.push_back(Exc);
17635dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  ESelArgs.push_back(Personality);
17645dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17655dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  bool HasCatchAll = false;
17665dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Only @try blocks are allowed @catch blocks, but both can have @finally
17675dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (isTry) {
17685dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    if (const ObjCAtCatchStmt* CatchStmt =
17695dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      cast<ObjCAtTryStmt>(S).getCatchStmts())  {
1770d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner      CGF.setInvokeDest(CatchInCatch);
17715dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17725dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      for (; CatchStmt; CatchStmt = CatchStmt->getNextCatchStmt()) {
17735dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        const ParmVarDecl *CatchDecl = CatchStmt->getCatchParamDecl();
1774bb1c8600218b3244340331165fc7cba7bf227655Mike Stump        Handlers.push_back(std::make_pair(CatchDecl,
1775bb1c8600218b3244340331165fc7cba7bf227655Mike Stump                                          CatchStmt->getCatchBody()));
17765dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17775dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        // @catch() and @catch(id) both catch any ObjC exception
177814108da7f7fc059772711e4ffee1322a27b152a7Steve Naroff        if (!CatchDecl || CatchDecl->getType()->isObjCIdType()
17795dc0867af17b3bd6f567897433853b2b4767446cChris Lattner            || CatchDecl->getType()->isObjCQualifiedIdType()) {
17805dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          // Use i8* null here to signal this is a catch all, not a cleanup.
17815dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          ESelArgs.push_back(NULLPtr);
17825dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          HasCatchAll = true;
17835dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          // No further catches after this one will ever by reached
17845dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          break;
17851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        }
17865dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
17875dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        // All other types should be Objective-C interface pointer types.
17881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        const ObjCObjectPointerType *OPT =
1789183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall          CatchDecl->getType()->getAs<ObjCObjectPointerType>();
179014108da7f7fc059772711e4ffee1322a27b152a7Steve Naroff        assert(OPT && "Invalid @catch type.");
17911eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump        const ObjCInterfaceType *IT =
1792183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall          OPT->getPointeeType()->getAs<ObjCInterfaceType>();
17935dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        assert(IT && "Invalid @catch type.");
1794bb422ad4f449cd195a141309a886ff467baac5b2Chris Lattner        llvm::Value *EHType =
1795bb422ad4f449cd195a141309a886ff467baac5b2Chris Lattner          MakeConstantString(IT->getDecl()->getNameAsString());
17965dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        ESelArgs.push_back(EHType);
17975dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      }
17985dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    }
17995dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  }
18005dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18015dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // We use a cleanup unless there was already a catch all.
18025dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (!HasCatchAll) {
18030032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson    ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
18045dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    Handlers.push_back(std::make_pair((const ParmVarDecl*) 0, (const Stmt*) 0));
18055dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  }
18065dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18075dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Find which handler was matched.
1808bb422ad4f449cd195a141309a886ff467baac5b2Chris Lattner  llvm::Value *ESelector = CGF.Builder.CreateCall(llvm_eh_selector,
18095dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      ESelArgs.begin(), ESelArgs.end(), "selector");
18105dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18115dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  for (unsigned i = 0, e = Handlers.size(); i != e; ++i) {
18125dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    const ParmVarDecl *CatchParam = Handlers[i].first;
18135dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    const Stmt *CatchBody = Handlers[i].second;
18145dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18155dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    llvm::BasicBlock *Next = 0;
18165dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18175dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    // The last handler always matches.
18185dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    if (i + 1 != e) {
18195dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      assert(CatchParam && "Only last handler can be a catch all.");
18205dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18215dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      // Test whether this block matches the type for the selector and branch
18225dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      // to Match if it does, or to the next BB if it doesn't.
18235dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      llvm::BasicBlock *Match = CGF.createBasicBlock("match");
18245dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      Next = CGF.createBasicBlock("catch.next");
1825bb422ad4f449cd195a141309a886ff467baac5b2Chris Lattner      llvm::Value *Id = CGF.Builder.CreateCall(llvm_eh_typeid_for,
18265dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          CGF.Builder.CreateBitCast(ESelArgs[i+2], PtrTy));
18275dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.Builder.CreateCondBr(CGF.Builder.CreateICmpEQ(ESelector, Id), Match,
18285dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          Next);
18295dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18305dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.EmitBlock(Match);
18315dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    }
18321eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
18335dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    if (CatchBody) {
18345dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      llvm::Value *ExcObject = CGF.Builder.CreateBitCast(Exc,
18355dc0867af17b3bd6f567897433853b2b4767446cChris Lattner          CGF.ConvertType(CatchParam->getType()));
18361eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
18375dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      // Bind the catch parameter if it exists.
18385dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      if (CatchParam) {
18395dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        // CatchParam is a ParmVarDecl because of the grammar
18405dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        // construction used to handle this, but for codegen purposes
18415dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        // we treat this as a local decl.
18425dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        CGF.EmitLocalBlockVarDecl(*CatchParam);
18435dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        CGF.Builder.CreateStore(ExcObject, CGF.GetAddrOfLocalVar(CatchParam));
18445dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      }
18455dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18465dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.ObjCEHValueStack.push_back(ExcObject);
18475dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.EmitStmt(CatchBody);
18485dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.ObjCEHValueStack.pop_back();
18495dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18505dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.EmitBranchThroughCleanup(FinallyEnd);
18515dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18525dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      if (Next)
18535dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        CGF.EmitBlock(Next);
18545dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    } else {
18555dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      assert(!Next && "catchup should be last handler.");
18565dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18575dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.Builder.CreateStore(Exc, RethrowPtr);
18585dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.EmitBranchThroughCleanup(FinallyRethrow);
18595dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    }
18605dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  }
1861d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  // The @finally block is a secondary landing pad for any exceptions thrown in
1862d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  // @catch() blocks
1863d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  CGF.EmitBlock(CatchInCatch);
1864d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  Exc = CGF.Builder.CreateCall(llvm_eh_exception, "exc");
1865d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  ESelArgs.clear();
1866d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  ESelArgs.push_back(Exc);
1867d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  ESelArgs.push_back(Personality);
18680faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  // If there is a @catch or @finally clause in outside of this one then we
18690faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  // need to make sure that we catch and rethrow it.
18700faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  if (PrevLandingPad) {
18710faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    ESelArgs.push_back(NULLPtr);
18720faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  } else {
18730faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    ESelArgs.push_back(llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), 0));
18740faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  }
1875d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  CGF.Builder.CreateCall(llvm_eh_selector, ESelArgs.begin(), ESelArgs.end(),
1876d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner      "selector");
1877d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  CGF.Builder.CreateCall(llvm_eh_typeid_for,
1878d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner      CGF.Builder.CreateIntToPtr(ESelArgs[2], PtrTy));
1879d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  CGF.Builder.CreateStore(Exc, RethrowPtr);
1880d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner  CGF.EmitBranchThroughCleanup(FinallyRethrow);
1881d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner
18825dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CodeGenFunction::CleanupBlockInfo Info = CGF.PopCleanupBlock();
18835dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18845dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.setInvokeDest(PrevLandingPad);
18855dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
18865dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.EmitBlock(FinallyBlock);
18875dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
1888d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner
18895dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (isTry) {
18901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    if (const ObjCAtFinallyStmt* FinallyStmt =
18915dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        cast<ObjCAtTryStmt>(S).getFinallyStmt())
18925dc0867af17b3bd6f567897433853b2b4767446cChris Lattner      CGF.EmitStmt(FinallyStmt->getFinallyBody());
18935dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  } else {
1894d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    // Emit 'objc_sync_exit(expr)' as finally's sole statement for
18955dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    // @synchronized.
1896d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    std::vector<const llvm::Type*> Args(1, IdTy);
1897d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    llvm::FunctionType *FTy =
18980032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson      llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
1899d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    llvm::Value *SyncExit = CGM.CreateRuntimeFunction(FTy, "objc_sync_exit");
19001eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    llvm::Value *SyncArg =
1901d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner      CGF.EmitScalarExpr(cast<ObjCAtSynchronizedStmt>(S).getSynchExpr());
1902d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    SyncArg = CGF.Builder.CreateBitCast(SyncArg, IdTy);
1903d6a9907ceff2c52738a72e38b6a04ab44ce1173fChris Lattner    CGF.Builder.CreateCall(SyncExit, SyncArg);
19045dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  }
19055dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
19065dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (Info.SwitchBlock)
19075dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    CGF.EmitBlock(Info.SwitchBlock);
19085dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (Info.EndBlock)
19095dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    CGF.EmitBlock(Info.EndBlock);
19105dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
19115dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Branch around the rethrow code.
19125dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.EmitBranch(FinallyEnd);
19135dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
19145dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.EmitBlock(FinallyRethrow);
19151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19160faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  llvm::Value *ExceptionObject = CGF.Builder.CreateLoad(RethrowPtr);
19170faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
19180faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  if (!UnwindBB) {
19190faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    CGF.Builder.CreateCall(RethrowFn, ExceptionObject);
19200faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    // Exception always thrown, next instruction is never reached.
19210faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    CGF.Builder.CreateUnreachable();
19220faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  } else {
19230faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    // If there is a @catch block outside this scope, we invoke instead of
19240faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    // calling because we may return to this function.  This is very slow, but
19250faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    // some people still do it.  It would be nice to add an optimised path for
19260faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    // this.
19270faa51632584b8d56311d5792a2a275729bc67e3David Chisnall    CGF.Builder.CreateInvoke(RethrowFn, UnwindBB, UnwindBB, &ExceptionObject,
19280faa51632584b8d56311d5792a2a275729bc67e3David Chisnall        &ExceptionObject+1);
19290faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  }
19305dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
19310faa51632584b8d56311d5792a2a275729bc67e3David Chisnall  CGF.EmitBlock(FinallyEnd);
193264d5d6c5903157c521af496479d06dc26032d718Anders Carlsson}
193364d5d6c5903157c521af496479d06dc26032d718Anders Carlsson
193464d5d6c5903157c521af496479d06dc26032d718Anders Carlssonvoid CGObjCGNU::EmitThrowStmt(CodeGen::CodeGenFunction &CGF,
193549f6602707887eea1a558a1dffe0213102f887f2Daniel Dunbar                              const ObjCAtThrowStmt &S) {
19365dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::Value *ExceptionAsObject;
19375dc0867af17b3bd6f567897433853b2b4767446cChris Lattner
19385dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  std::vector<const llvm::Type*> Args(1, IdTy);
19395dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::FunctionType *FTy =
19400032b2781b4deb131f8c9b7968f2030bf2489cddOwen Anderson    llvm::FunctionType::get(llvm::Type::getVoidTy(VMContext), Args, false);
19411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump  llvm::Value *ThrowFn =
19425dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    CGM.CreateRuntimeFunction(FTy, "objc_exception_throw");
19431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19445dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (const Expr *ThrowExpr = S.getThrowExpr()) {
19455dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    llvm::Value *Exception = CGF.EmitScalarExpr(ThrowExpr);
19461e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian    ExceptionAsObject = Exception;
19475dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  } else {
19481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    assert((!CGF.ObjCEHValueStack.empty() && CGF.ObjCEHValueStack.back()) &&
19495dc0867af17b3bd6f567897433853b2b4767446cChris Lattner           "Unexpected rethrow outside @catch block.");
19505dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    ExceptionAsObject = CGF.ObjCEHValueStack.back();
19515dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  }
19521e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian  ExceptionAsObject =
19531e64a9531bf6e9d99ad3271b3a735ef702689b06Fariborz Jahanian      CGF.Builder.CreateBitCast(ExceptionAsObject, IdTy, "tmp");
19541eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
19555dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Note: This may have to be an invoke, if we want to support constructs like:
19565dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // @try {
19575dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  //  @throw(obj);
19585dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // }
19595dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // @catch(id) ...
19605dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  //
19615dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // This is effectively turning @throw into an incredibly-expensive goto, but
19625dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // it may happen as a result of inlining followed by missed optimizations, or
19635dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // as a result of stupidity.
19645dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  llvm::BasicBlock *UnwindBB = CGF.getInvokeDest();
19655dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  if (!UnwindBB) {
19665dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    CGF.Builder.CreateCall(ThrowFn, ExceptionAsObject);
19675dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    CGF.Builder.CreateUnreachable();
19685dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  } else {
19695dc0867af17b3bd6f567897433853b2b4767446cChris Lattner    CGF.Builder.CreateInvoke(ThrowFn, UnwindBB, UnwindBB, &ExceptionAsObject,
19705dc0867af17b3bd6f567897433853b2b4767446cChris Lattner        &ExceptionAsObject+1);
19715dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  }
19725dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  // Clear the insertion point to indicate we are in unreachable code.
19735dc0867af17b3bd6f567897433853b2b4767446cChris Lattner  CGF.Builder.ClearInsertionPoint();
197464d5d6c5903157c521af496479d06dc26032d718Anders Carlsson}
197564d5d6c5903157c521af496479d06dc26032d718Anders Carlsson
19763e283e344595e0bd499b13b30a92b7d9c10a2140Fariborz Jahanianllvm::Value * CGObjCGNU::EmitObjCWeakRead(CodeGen::CodeGenFunction &CGF,
19771eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                          llvm::Value *AddrWeakObj) {
1978ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  CGBuilderTy B = CGF.Builder;
1979ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  AddrWeakObj = EnforceType(B, AddrWeakObj, IdTy);
1980ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  return B.CreateCall(WeakReadFn, AddrWeakObj);
19816dc2317b59cb1180a59f6c283d96b7a5dfeb5307Fariborz Jahanian}
19826dc2317b59cb1180a59f6c283d96b7a5dfeb5307Fariborz Jahanian
19833e283e344595e0bd499b13b30a92b7d9c10a2140Fariborz Jahanianvoid CGObjCGNU::EmitObjCWeakAssign(CodeGen::CodeGenFunction &CGF,
19841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                   llvm::Value *src, llvm::Value *dst) {
1985ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  CGBuilderTy B = CGF.Builder;
1986ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  src = EnforceType(B, src, IdTy);
1987ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  dst = EnforceType(B, dst, PtrToIdTy);
1988ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  B.CreateCall2(WeakAssignFn, src, dst);
19893e283e344595e0bd499b13b30a92b7d9c10a2140Fariborz Jahanian}
19903e283e344595e0bd499b13b30a92b7d9c10a2140Fariborz Jahanian
199158626500527695865683d1d65053743de8770b60Fariborz Jahanianvoid CGObjCGNU::EmitObjCGlobalAssign(CodeGen::CodeGenFunction &CGF,
19921eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                     llvm::Value *src, llvm::Value *dst) {
1993ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  CGBuilderTy B = CGF.Builder;
1994ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  src = EnforceType(B, src, IdTy);
1995ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  dst = EnforceType(B, dst, PtrToIdTy);
1996ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  B.CreateCall2(GlobalAssignFn, src, dst);
199758626500527695865683d1d65053743de8770b60Fariborz Jahanian}
199858626500527695865683d1d65053743de8770b60Fariborz Jahanian
19997eda8367cf63caee8acf907356b1d199ccaa6e89Fariborz Jahanianvoid CGObjCGNU::EmitObjCIvarAssign(CodeGen::CodeGenFunction &CGF,
20006c7a1f364796ce1acb988714e9e42076d1ce332eFariborz Jahanian                                   llvm::Value *src, llvm::Value *dst,
20016c7a1f364796ce1acb988714e9e42076d1ce332eFariborz Jahanian                                   llvm::Value *ivarOffset) {
2002ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  CGBuilderTy B = CGF.Builder;
2003ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  src = EnforceType(B, src, IdTy);
2004ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  dst = EnforceType(B, dst, PtrToIdTy);
2005ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  B.CreateCall3(IvarAssignFn, src, dst, ivarOffset);
20067eda8367cf63caee8acf907356b1d199ccaa6e89Fariborz Jahanian}
20077eda8367cf63caee8acf907356b1d199ccaa6e89Fariborz Jahanian
200858626500527695865683d1d65053743de8770b60Fariborz Jahanianvoid CGObjCGNU::EmitObjCStrongCastAssign(CodeGen::CodeGenFunction &CGF,
20091eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                         llvm::Value *src, llvm::Value *dst) {
2010ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  CGBuilderTy B = CGF.Builder;
2011ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  src = EnforceType(B, src, IdTy);
2012ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  dst = EnforceType(B, dst, PtrToIdTy);
2013ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  B.CreateCall2(StrongCastAssignFn, src, dst);
201458626500527695865683d1d65053743de8770b60Fariborz Jahanian}
201558626500527695865683d1d65053743de8770b60Fariborz Jahanian
2016082b02e8403d3ee9d2ded969fbe0e5d472f04cd8Fariborz Jahanianvoid CGObjCGNU::EmitGCMemmoveCollectable(CodeGen::CodeGenFunction &CGF,
20171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump                                         llvm::Value *DestPtr,
2018082b02e8403d3ee9d2ded969fbe0e5d472f04cd8Fariborz Jahanian                                         llvm::Value *SrcPtr,
201908c321380fff07d476a19daab6d29522c046cd49Fariborz Jahanian                                         QualType Ty) {
2020ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  CGBuilderTy B = CGF.Builder;
2021ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  DestPtr = EnforceType(B, DestPtr, IdTy);
2022ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  SrcPtr = EnforceType(B, SrcPtr, PtrToIdTy);
2023ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall
2024ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  std::pair<uint64_t, unsigned> TypeInfo = CGM.getContext().getTypeInfo(Ty);
2025ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  unsigned long size = TypeInfo.first/8;
2026ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  // FIXME: size_t
2027ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  llvm::Value *N = llvm::ConstantInt::get(LongTy, size);
2028ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall
2029ef6e0f3218b64d8ea3b8b57ee657fec8da228a43David Chisnall  B.CreateCall3(MemMoveFn, DestPtr, SrcPtr, N);
2030082b02e8403d3ee9d2ded969fbe0e5d472f04cd8Fariborz Jahanian}
2031082b02e8403d3ee9d2ded969fbe0e5d472f04cd8Fariborz Jahanian
20329cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanianllvm::GlobalVariable *CGObjCGNU::ObjCIvarOffsetVariable(
20339cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian                              const ObjCInterfaceDecl *ID,
20349cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian                              const ObjCIvarDecl *Ivar) {
20359cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  const std::string Name = "__objc_ivar_offset_" + ID->getNameAsString()
20369cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    + '.' + Ivar->getNameAsString();
20379cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  // Emit the variable and initialize it with what we think the correct value
20389cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  // is.  This allows code compiled with non-fragile ivars to work correctly
20399cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  // when linked against code which isn't (most of the time).
20408a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  llvm::GlobalVariable *IvarOffsetPointer = TheModule.getNamedGlobal(Name);
20418a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  if (!IvarOffsetPointer) {
20429cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    uint64_t Offset = ComputeIvarBaseOffset(CGM, ID, Ivar);
20439cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    llvm::ConstantInt *OffsetGuess =
2044f950837b8250c7f08d0b7b15ace0b35740721c9dDavid Chisnall      llvm::ConstantInt::get(llvm::Type::getInt32Ty(VMContext), Offset, "ivar");
20458a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    // Don't emit the guess in non-PIC code because the linker will not be able
20468a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    // to replace it with the real version for a library.  In non-PIC code you
20478a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    // must compile with the fragile ABI if you want to use ivars from a
20481eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump    // GCC-compiled class.
20498a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    if (CGM.getLangOptions().PICLevel) {
20508a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      llvm::GlobalVariable *IvarOffsetGV = new llvm::GlobalVariable(TheModule,
20518a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall            llvm::Type::getInt32Ty(VMContext), false,
20528a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall            llvm::GlobalValue::PrivateLinkage, OffsetGuess, Name+".guess");
20538a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
20548a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall            IvarOffsetGV->getType(), false, llvm::GlobalValue::LinkOnceAnyLinkage,
20558a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall            IvarOffsetGV, Name);
20568a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    } else {
20578a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall      IvarOffsetPointer = new llvm::GlobalVariable(TheModule,
20583c0ef8cc0dc246bd3083e8cdd63005e8873d36d2Benjamin Kramer              llvm::Type::getInt32PtrTy(VMContext), false,
20593c0ef8cc0dc246bd3083e8cdd63005e8873d36d2Benjamin Kramer              llvm::GlobalValue::ExternalLinkage, 0, Name);
20608a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    }
20619cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  }
20628a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  return IvarOffsetPointer;
20639cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian}
20649cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian
2065598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz JahanianLValue CGObjCGNU::EmitObjCValueForIvar(CodeGen::CodeGenFunction &CGF,
2066598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                       QualType ObjectTy,
2067598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                       llvm::Value *BaseValue,
2068598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                       const ObjCIvarDecl *Ivar,
2069598d3f61b6ca854e9d3c2f3359e24468502a61aaFariborz Jahanian                                       unsigned CVRQualifiers) {
2070183700f494ec9b6701b6efe82bcb25f4c79ba561John McCall  const ObjCInterfaceDecl *ID = ObjectTy->getAs<ObjCInterfaceType>()->getDecl();
20719777687562c338601c2f17906e65e1c1a0aad96fDaniel Dunbar  return EmitValueForIvarAtOffset(CGF, ID, BaseValue, Ivar, CVRQualifiers,
20729777687562c338601c2f17906e65e1c1a0aad96fDaniel Dunbar                                  EmitIvarOffset(CGF, ID, Ivar));
20730bb20361a321593887f067515dd04cf109f4c74aFariborz Jahanian}
2074bb1c8600218b3244340331165fc7cba7bf227655Mike Stump
20759cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanianstatic const ObjCInterfaceDecl *FindIvarInterface(ASTContext &Context,
20769cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian                                                  const ObjCInterfaceDecl *OID,
20779cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian                                                  const ObjCIvarDecl *OIVD) {
20789cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  llvm::SmallVector<ObjCIvarDecl*, 16> Ivars;
20798e6ac1d80055fa37b9b84029c7e751624ba7f84cFariborz Jahanian  Context.ShallowCollectObjCIvars(OID, Ivars);
20809cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  for (unsigned k = 0, e = Ivars.size(); k != e; ++k) {
20819cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    if (OIVD == Ivars[k])
20829cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian      return OID;
20839cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  }
20841eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20859cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  // Otherwise check in the super class.
20869cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  if (const ObjCInterfaceDecl *Super = OID->getSuperClass())
20879cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    return FindIvarInterface(Context, Super, OIVD);
20881eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump
20899cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  return 0;
20909cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian}
20910bb20361a321593887f067515dd04cf109f4c74aFariborz Jahanian
2092f63aa3fd429cdb9145d78f0b656bc78754efedb9Fariborz Jahanianllvm::Value *CGObjCGNU::EmitIvarOffset(CodeGen::CodeGenFunction &CGF,
20932a03192a02dbf4fdff438d1e658356bde871aba4Daniel Dunbar                         const ObjCInterfaceDecl *Interface,
2094f63aa3fd429cdb9145d78f0b656bc78754efedb9Fariborz Jahanian                         const ObjCIvarDecl *Ivar) {
20958a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall  if (CGM.getLangOptions().ObjCNonFragileABI) {
20969cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian    Interface = FindIvarInterface(CGM.getContext(), Interface, Ivar);
20978a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall    return CGF.Builder.CreateLoad(CGF.Builder.CreateLoad(
20988a5a9aaddb627c0884c2ed8db55cc29fdb601195David Chisnall                ObjCIvarOffsetVariable(Interface, Ivar), false, "ivar"));
20999cd96ff7dfcb51fdba03df8803fff6cc36e9633fFariborz Jahanian  }
21009777687562c338601c2f17906e65e1c1a0aad96fDaniel Dunbar  uint64_t Offset = ComputeIvarBaseOffset(CGF.CGM, Interface, Ivar);
21014a28d5deeba33722aa009eab488591fb9055cc7eOwen Anderson  return llvm::ConstantInt::get(LongTy, Offset, "ivar");
2102f63aa3fd429cdb9145d78f0b656bc78754efedb9Fariborz Jahanian}
2103f63aa3fd429cdb9145d78f0b656bc78754efedb9Fariborz Jahanian
2104bb1c8600218b3244340331165fc7cba7bf227655Mike StumpCodeGen::CGObjCRuntime *
2105bb1c8600218b3244340331165fc7cba7bf227655Mike StumpCodeGen::CreateGNUObjCRuntime(CodeGen::CodeGenModule &CGM) {
2106dce1406f1c1f572cfd61c494546572d63461c741Chris Lattner  return new CGObjCGNU(CGM);
21070f984268b05edab2cc555a427c441baa9c252658Chris Lattner}
2108