164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===--- RewriteObjC.cpp - Playground for the code rewriter ---------------===// 264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// The LLVM Compiler Infrastructure 464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// This file is distributed under the University of Illinois Open Source 664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// License. See LICENSE.TXT for details. 764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 1064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Hacks and fun related to the code rewriter. 1164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 1264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 1364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/Rewrite/ASTConsumers.h" 1564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/Rewrite/Rewriter.h" 1664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/AST/AST.h" 1764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/AST/ASTConsumer.h" 1864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/AST/ParentMap.h" 1964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/Basic/SourceManager.h" 2064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/Basic/IdentifierTable.h" 2164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/Basic/Diagnostic.h" 2264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "clang/Lex/Lexer.h" 2364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "llvm/Support/MemoryBuffer.h" 2464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "llvm/Support/raw_ostream.h" 2564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "llvm/ADT/StringExtras.h" 2664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "llvm/ADT/SmallPtrSet.h" 2764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "llvm/ADT/OwningPtr.h" 2864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#include "llvm/ADT/DenseSet.h" 2964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 3064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianusing namespace clang; 3164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianusing llvm::utostr; 3264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 3364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahaniannamespace { 3464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian class RewriteModernObjC : public ASTConsumer { 3564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian protected: 3664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 3764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian enum { 3864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_FIELD_IS_OBJECT = 3, /* id, NSObject, __attribute__((NSObject)), 3964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian block, ... */ 4064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_FIELD_IS_BLOCK = 7, /* a block variable */ 4164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_FIELD_IS_BYREF = 8, /* the on stack structure holding the 4264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian __block variable */ 4364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_FIELD_IS_WEAK = 16, /* declared __weak, only used in byref copy 4464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian helpers */ 4564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_BYREF_CALLER = 128, /* called from __block (byref) copy/dispose 4664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian support routines */ 4764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_BYREF_CURRENT_MAX = 256 4864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 4964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 5064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian enum { 5164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_NEEDS_FREE = (1 << 24), 5264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_HAS_COPY_DISPOSE = (1 << 25), 5364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_HAS_CXX_OBJ = (1 << 26), 5464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_IS_GC = (1 << 27), 5564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_IS_GLOBAL = (1 << 28), 5664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BLOCK_HAS_DESCRIPTOR = (1 << 29) 5764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 5864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static const int OBJC_ABI_VERSION = 7; 5964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Rewriter Rewrite; 6164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &Diags; 6264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const LangOptions &LangOpts; 6364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ASTContext *Context; 6464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceManager *SM; 6564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TranslationUnitDecl *TUDecl; 6664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FileID MainFileID; 6764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *MainFileStart, *MainFileEnd; 6864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *CurrentBody; 6964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParentMap *PropParentMap; // created lazily. 7064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string InFileName; 7164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian raw_ostream* OutFile; 7264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Preamble; 7364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 7464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeDecl *ProtocolTypeDecl; 7564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *GlobalVarDecl; 76df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Expr *GlobalConstructionExp; 7764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned RewriteFailedDiag; 78d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian unsigned GlobalBlockRewriteFailedDiag; 7964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ObjC string constant support. 8064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned NumObjCStringLiterals; 8164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *ConstantStringClassReference; 8264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *NSStringRecord; 8364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 8464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ObjC foreach break/continue generation support. 8564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int BcLabelCount; 8664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 8764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned TryFinallyContainsReturnDiag; 8864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Needed for super. 8964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *CurMethodDef; 9064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *SuperStructDecl; 9164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *ConstantStringDecl; 9264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 9364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendFunctionDecl; 9464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendSuperFunctionDecl; 9564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendStretFunctionDecl; 9664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendSuperStretFunctionDecl; 9764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendFpretFunctionDecl; 9864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *GetClassFunctionDecl; 9964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *GetMetaClassFunctionDecl; 10064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *GetSuperClassFunctionDecl; 10164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *SelGetUidFunctionDecl; 10264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *CFStringFunctionDecl; 10364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *SuperContructorFunctionDecl; 10464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *CurFunctionDef; 10564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 10664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* Misc. containers needed for meta-data rewrite. */ 10764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCImplementationDecl *, 8> ClassImplementation; 10864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCCategoryImplDecl *, 8> CategoryImplementation; 10964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCInterfaceDecl*, 8> ObjCSynthesizedStructs; 11064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCProtocolDecl*, 8> ObjCSynthesizedProtocols; 111cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian llvm::SmallPtrSet<ObjCInterfaceDecl*, 8> ObjCWrittenInterfaces; 11215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian llvm::SmallPtrSet<TagDecl*, 8> TagsDefinedInIvarDecls; 113cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian SmallVector<ObjCInterfaceDecl*, 32> ObjCInterfacesSeen; 11488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian /// DefinedNonLazyClasses - List of defined "non-lazy" classes. 11588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian SmallVector<ObjCInterfaceDecl*, 8> DefinedNonLazyClasses; 11688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 11788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian /// DefinedNonLazyCategories - List of defined "non-lazy" categories. 11888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian llvm::SmallVector<ObjCCategoryDecl*, 8> DefinedNonLazyCategories; 11988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 12064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Stmt *, 32> Stmts; 12164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<int, 8> ObjCBcLabelNo; 12264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remember all the @protocol(<expr>) expressions. 12364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCProtocolDecl *, 32> ProtocolExprDecls; 12464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 12564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseSet<uint64_t> CopyDestroyCache; 12664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 12764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block expressions. 12864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockExpr *, 32> Blocks; 12964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<int, 32> InnerDeclRefsCount; 130f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 32> InnerDeclRefs; 13164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 132f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 32> BlockDeclRefs; 13364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 13464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block related declarations. 13564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ValueDecl *, 8> BlockByCopyDecls; 13664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> BlockByCopyDeclsPtrSet; 13764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ValueDecl *, 8> BlockByRefDecls; 13864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> BlockByRefDeclsPtrSet; 13964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<ValueDecl *, unsigned> BlockByRefDeclNo; 14064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> ImportedBlockDecls; 14164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<VarDecl *, 8> ImportedLocalExternalDecls; 14264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 14364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<BlockExpr *, std::string> RewrittenBlockExprs; 14472c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian llvm::DenseMap<ObjCInterfaceDecl *, 14572c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian llvm::SmallPtrSet<ObjCIvarDecl *, 8> > ReferencedIvars; 14672c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 14764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This maps an original source AST to it's rewritten form. This allows 14864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // us to avoid rewriting the same node twice (which is very uncommon). 14964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is needed to support some of the exotic property rewriting. 15064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<Stmt *, Stmt *> ReplacedNodes; 15164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 15264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Needed for header files being rewritten 15364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsHeader; 15464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SilenceRewriteMacroWarning; 15564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool objc_impl_method; 15664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 15764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool DisableReplaceStmt; 15864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian class DisableReplaceStmtScope { 15964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteModernObjC &R; 16064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SavedValue; 16164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 16264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian public: 16364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope(RewriteModernObjC &R) 16464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : R(R), SavedValue(R.DisableReplaceStmt) { 16564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian R.DisableReplaceStmt = true; 16664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 16764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ~DisableReplaceStmtScope() { 16864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian R.DisableReplaceStmt = SavedValue; 16964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 17064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 17164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InitializeCommon(ASTContext &context); 17264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 17364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian public: 17490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian llvm::DenseMap<ObjCMethodDecl*, std::string> MethodInternalNames; 17564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Top Level Driver code. 17664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual bool HandleTopLevelDecl(DeclGroupRef D) { 17764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclGroupRef::iterator I = D.begin(), E = D.end(); I != E; ++I) { 17864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*I)) { 17964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Class->isThisDeclarationADefinition()) { 18064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassDecl(D); 18164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 182cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian } else { 183cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian // Keep track of all interface declarations seen. 184f329527bbeae86c8eac6eea6efaa20e807c16679Fariborz Jahanian ObjCInterfacesSeen.push_back(Class); 185cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian break; 18664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 18964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*I)) { 19064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Proto->isThisDeclarationADefinition()) { 19164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardProtocolDecl(D); 19264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 19364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 19464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 19564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 19664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HandleTopLevelSingleDecl(*I); 19764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 19864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 19964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 20064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void HandleTopLevelSingleDecl(Decl *D); 20164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void HandleDeclInMainFile(Decl *D); 20264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteModernObjC(std::string inFile, raw_ostream *OS, 20364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &D, const LangOptions &LOpts, 20464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool silenceMacroWarn); 20564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 20664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ~RewriteModernObjC() {} 20764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 20864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void HandleTranslationUnit(ASTContext &C); 20964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceStmt(Stmt *Old, Stmt *New) { 21164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = ReplacedNodes[Old]; 21264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ReplacingStmt) 21464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // We can't rewrite the same node twice. 21564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DisableReplaceStmt) 21764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 21864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If replacement succeeded or warning disabled return with no warning. 22064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceStmt(Old, New)) { 22164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacedNodes[Old] = New; 22264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 22364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 22464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SilenceRewriteMacroWarning) 22564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 22664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 22764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 22864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 22964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 23064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceStmtWithRange(Stmt *Old, Stmt *New, SourceRange SrcRange) { 23164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DisableReplaceStmt) 23264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 23364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 23464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Measure the old text. 23564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int Size = Rewrite.getRangeSize(SrcRange); 23664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Size == -1) { 23764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 23864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 23964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 24064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 24164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the new text. 24264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SStr; 24364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream S(SStr); 24464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian New->printPretty(S, *Context, 0, PrintingPolicy(LangOpts)); 24564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Str = S.str(); 24664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 24764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If replacement succeeded or warning disabled return with no warning. 24864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceText(SrcRange.getBegin(), Size, Str)) { 24964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacedNodes[Old] = New; 25064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 25164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 25264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SilenceRewriteMacroWarning) 25364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 25464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 25564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 25664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 25764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 25864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertText(SourceLocation Loc, StringRef Str, 25964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool InsertAfter = true) { 26064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If insertion succeeded or warning disabled return with no warning. 26164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.InsertText(Loc, Str, InsertAfter) || 26264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning) 26364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 26464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 26564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag); 26664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 26764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 26864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceText(SourceLocation Start, unsigned OrigLength, 26964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef Str) { 27064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If removal succeeded or warning disabled return with no warning. 27164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceText(Start, OrigLength, Str) || 27264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning) 27364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 27464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 27564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Start), RewriteFailedDiag); 27664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 27764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 27864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Syntactic Rewriting. 27964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteRecordBody(RecordDecl *RD); 28064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteInclude(); 28164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassDecl(DeclGroupRef D); 28264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassDecl(const llvm::SmallVector<Decl*, 8> &DG); 28364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassEpilogue(ObjCInterfaceDecl *ClassDecl, 28464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &typedefString); 28564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteImplementations(); 28664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewritePropertyImplDecl(ObjCPropertyImplDecl *PID, 28764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD, 28864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID); 28964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteInterfaceDecl(ObjCInterfaceDecl *Dcl); 29064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteImplementationDecl(Decl *Dcl); 29164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCMethodDecl(const ObjCInterfaceDecl *IDecl, 29264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MDecl, std::string &ResultStr); 29364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteTypeIntoString(QualType T, std::string &ResultStr, 29464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *&FPRetType); 29564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteByRefString(std::string &ResultStr, const std::string &Name, 29664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD, bool def=false); 29764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteCategoryDecl(ObjCCategoryDecl *Dcl); 29864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteProtocolDecl(ObjCProtocolDecl *Dcl); 29964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardProtocolDecl(DeclGroupRef D); 30064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardProtocolDecl(const llvm::SmallVector<Decl*, 8> &DG); 30164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteMethodDeclaration(ObjCMethodDecl *Method); 30264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteProperty(ObjCPropertyDecl *prop); 30364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteFunctionDecl(FunctionDecl *FD); 30464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerType(std::string& Str, QualType Type); 30564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerTypeVariable(std::string& Str, ValueDecl *VD); 306b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian void RewriteBlockLiteralFunctionDecl(FunctionDecl *FD); 30764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCQualifiedInterfaceTypes(Decl *Dcl); 30864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteTypeOfDecl(VarDecl *VD); 30964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCQualifiedInterfaceTypes(Expr *E); 31064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 31164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Expression Rewriting. 31264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteFunctionBodyOrGlobalInitializer(Stmt *S); 31364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteAtEncode(ObjCEncodeExpr *Exp); 31464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewritePropertyOrImplicitGetter(PseudoObjectExpr *Pseudo); 31564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewritePropertyOrImplicitSetter(PseudoObjectExpr *Pseudo); 31664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteAtSelector(ObjCSelectorExpr *Exp); 31764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteMessageExpr(ObjCMessageExpr *Exp); 31864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCStringLiteral(ObjCStringLiteral *Exp); 31955947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian Stmt *RewriteObjCBoolLiteralExpr(ObjCBoolLiteralExpr *Exp); 320eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard Stmt *RewriteObjCBoxedExpr(ObjCBoxedExpr *Exp); 32186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian Stmt *RewriteObjCArrayLiteralExpr(ObjCArrayLiteral *Exp); 322e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Stmt *RewriteObjCDictionaryLiteralExpr(ObjCDictionaryLiteral *Exp); 32364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp); 32464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCTryStmt(ObjCAtTryStmt *S); 32564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S); 32664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCThrowStmt(ObjCAtThrowStmt *S); 32764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S, 32864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OrigEnd); 32964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteBreakStmt(BreakStmt *S); 33064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteContinueStmt(ContinueStmt *S); 33164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteCastExpr(CStyleCastExpr *CE); 332f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian void RewriteImplicitCastObjCExpr(CastExpr *IE); 333b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian void RewriteLinkageSpec(LinkageSpecDecl *LSD); 33464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 33564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block rewriting. 33664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlocksInFunctionProtoType(QualType funcType, NamedDecl *D); 33764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 33864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block specific rewrite rules. 33964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerDecl(NamedDecl *VD); 34064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteByRefVar(VarDecl *VD); 341f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Stmt *RewriteBlockDeclRefExpr(DeclRefExpr *VD); 34264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteLocalVariableExternalStorage(DeclRefExpr *DRE); 34364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerFunctionArgs(FunctionDecl *FD); 34464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 34564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCInternalStruct(ObjCInterfaceDecl *CDecl, 34664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 34764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 34815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian void RewriteObjCFieldDecl(FieldDecl *fieldDecl, std::string &Result); 34915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 35097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian bool RewriteObjCFieldDeclType(QualType &Type, std::string &Result); 35197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 35272c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian void RewriteIvarOffsetSymbols(ObjCInterfaceDecl *CDecl, 35372c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian std::string &Result); 35472c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 35564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void Initialize(ASTContext &context); 35664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 35764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Misc. AST transformation routines. Somtimes they end up calling 35864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // rewriting routines on the new ASTs. 35964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SynthesizeCallToFunctionDecl(FunctionDecl *FD, 36064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr **args, unsigned nargs, 36164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc=SourceLocation(), 36264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc=SourceLocation()); 36364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 36464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthMessageExpr(ObjCMessageExpr *Exp, 36564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc=SourceLocation(), 36664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc=SourceLocation()); 36764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 36864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthCountByEnumWithState(std::string &buf); 36964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendFunctionDecl(); 37064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendSuperFunctionDecl(); 37164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendStretFunctionDecl(); 37264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendFpretFunctionDecl(); 37364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendSuperStretFunctionDecl(); 37464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetClassFunctionDecl(); 37564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetMetaClassFunctionDecl(); 37664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetSuperClassFunctionDecl(); 37764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthSelGetUidFunctionDecl(); 37864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthSuperContructorFunctionDecl(); 37964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 38064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewriting metadata 38164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian template<typename MethodIterator> 38264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCMethodsMetaData(MethodIterator MethodBegin, 38364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodIterator MethodEnd, 38464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsInstanceMethod, 38564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, 38664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef ClassName, 38764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 388da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian void RewriteObjCProtocolMetaData(ObjCProtocolDecl *Protocol, 389da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::string &Result); 39064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCProtocolListMetaData( 39164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Prots, 39264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, StringRef ClassName, std::string &Result); 39364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl, 39464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 395e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian virtual void RewriteClassSetupInitHook(std::string &Result); 396e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian 39764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteMetaDataIntoBuffer(std::string &Result); 39810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian virtual void WriteImageInfo(std::string &Result); 39964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *CDecl, 40064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 401e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian virtual void RewriteCategorySetupInitHook(std::string &Result); 40264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 40364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewriting ivar 40464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteIvarOffsetComputation(ObjCIvarDecl *ivar, 40564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 40664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual Stmt *RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV); 40764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 40864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 40964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeByrefCopyDestroyHelper(VarDecl *VD, int flag); 41064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockHelperFuncs(BlockExpr *CE, int i, 41164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, std::string Tag); 41264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockFunc(BlockExpr *CE, int i, 41364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, std::string Tag); 41464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockImpl(BlockExpr *CE, 41564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag, std::string Desc); 41664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockDescriptor(std::string DescTag, 41764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag, 41864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int i, StringRef funcName, 41964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned hasCopy); 42064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthesizeBlockCall(CallExpr *Exp, const Expr* BlockExp); 42164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthesizeBlockLiterals(SourceLocation FunLocStart, 42264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName); 42364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *SynthBlockInitFunctionDecl(StringRef name); 42464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthBlockInitExpr(BlockExpr *Exp, 425f4b88a45902af1802a1cb42ba48b1c474474f228John McCall const SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs); 42664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 42764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Misc. helper routines. 42864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getProtocolType(); 42964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void WarnAboutReturnGotoStmts(Stmt *S); 43064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void CheckFunctionPointerDecl(QualType dType, NamedDecl *ND); 43164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertBlockLiteralsWithinFunction(FunctionDecl *FD); 43264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertBlockLiteralsWithinMethod(ObjCMethodDecl *MD); 43364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 43464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsDeclStmtInForeachHeader(DeclStmt *DS); 43564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void CollectBlockDeclRefInfo(BlockExpr *Exp); 43664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetBlockDeclRefExprs(Stmt *S); 43764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetInnerBlockDeclRefExprs(Stmt *S, 438f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs, 43964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> &InnerContexts); 44064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 44164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We avoid calling Type::isBlockPointerType(), since it operates on the 44264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // canonical type. We only care if the top-level type is a closure pointer. 44364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isTopLevelBlockPointerType(QualType T) { 44464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return isa<BlockPointerType>(T); 44564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 44664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 44764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// convertBlockPointerToFunctionPointer - Converts a block-pointer type 44864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// to a function pointer type and upon success, returns true; false 44964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// otherwise. 45064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool convertBlockPointerToFunctionPointer(QualType &T) { 45164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(T)) { 45264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = T->getAs<BlockPointerType>(); 45364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getPointerType(BPT->getPointeeType()); 45464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 45564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 45664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 45764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 45864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 459164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian bool convertObjCTypeToCStyleType(QualType &T); 460164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian 46164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool needToScanForQualifiers(QualType T); 46264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getSuperStructType(); 46364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getConstantStringStructType(); 46464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType convertFunctionTypeOfBlocks(const FunctionType *FT); 46564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool BufferContainsPPDirectives(const char *startBuf, const char *endBuf); 46664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 46764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void convertToUnqualifiedObjCType(QualType &T) { 468e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian if (T->isObjCQualifiedIdType()) { 469e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian bool isConst = T.isConstQualified(); 470e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian T = isConst ? Context->getObjCIdType().withConst() 471e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian : Context->getObjCIdType(); 472e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian } 47364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isObjCQualifiedClassType()) 47464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getObjCClassType(); 47564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isObjCObjectPointerType() && 47664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->getPointeeType()->isObjCQualifiedInterfaceType()) { 47764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const ObjCObjectPointerType * OBJPT = 47864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->getAsObjCInterfacePointerType()) { 47964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceType *IFaceT = OBJPT->getInterfaceType(); 48064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = QualType(IFaceT, 0); 48164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getPointerType(T); 48264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 48364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 48464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 48564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 48664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: This predicate seems like it would be useful to add to ASTContext. 48764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isObjCType(QualType T) { 48864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!LangOpts.ObjC1 && !LangOpts.ObjC2) 48964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 49064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 49164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType OCT = Context->getCanonicalType(T).getUnqualifiedType(); 49264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 49364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OCT == Context->getCanonicalType(Context->getObjCIdType()) || 49464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OCT == Context->getCanonicalType(Context->getObjCClassType())) 49564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 49664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 49764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType *PT = OCT->getAs<PointerType>()) { 49864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCInterfaceType>(PT->getPointeeType()) || 49964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PT->getPointeeType()->isObjCQualifiedIdType()) 50064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 50164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 50264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 50364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 50464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool PointerTypeTakesAnyBlockArguments(QualType QT); 50564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool PointerTypeTakesAnyObjCQualifiedType(QualType QT); 50664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetExtentOfArgList(const char *Name, const char *&LParen, 50764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&RParen); 50864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 50964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void QuoteDoublequotes(std::string &From, std::string &To) { 51064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < From.length(); i++) { 51164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (From[i] == '"') 51264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian To += "\\\""; 51364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 51464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian To += From[i]; 51564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 51664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 51764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 51864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getSimpleFunctionType(QualType result, 51964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const QualType *args, 52064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned numArgs, 52164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool variadic = false) { 52264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (result == Context->getObjCInstanceType()) 52364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian result = Context->getObjCIdType(); 52464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionProtoType::ExtProtoInfo fpi; 52564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian fpi.Variadic = variadic; 52664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getFunctionType(result, args, numArgs, fpi); 52764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 52864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 52964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Helper function: create a CStyleCastExpr with trivial type source info. 53064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CStyleCastExpr* NoTypeInfoCStyleCastExpr(ASTContext *Ctx, QualType Ty, 53164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind Kind, Expr *E) { 53264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *TInfo = Ctx->getTrivialTypeSourceInfo(Ty, SourceLocation()); 53364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CStyleCastExpr::Create(*Ctx, Ty, VK_RValue, Kind, E, 0, TInfo, 53464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation()); 53564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 53688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 53788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian bool ImplementationIsNonLazy(const ObjCImplDecl *OD) const { 53888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian IdentifierInfo* II = &Context->Idents.get("load"); 53988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Selector LoadSel = Context->Selectors.getSelector(0, &II); 54088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian return OD->getClassMethod(LoadSel) != 0; 54188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 54264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 54364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 54464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 54564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 54664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlocksInFunctionProtoType(QualType funcType, 54764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NamedDecl *D) { 54864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *fproto 54964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = dyn_cast<FunctionProtoType>(funcType.IgnoreParens())) { 55064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = fproto->arg_type_begin(), 55164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = fproto->arg_type_end(); I && (I != E); ++I) 55264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(*I)) { 55364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // All the args are checked/rewritten. Don't call twice! 55464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(D); 55564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 55664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 55764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 55864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 55964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 56064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::CheckFunctionPointerDecl(QualType funcType, NamedDecl *ND) { 56164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = funcType->getAs<PointerType>(); 56264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT && PointerTypeTakesAnyBlockArguments(funcType)) 56364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlocksInFunctionProtoType(PT->getPointeeType(), ND); 56464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 56564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 56664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool IsHeaderFile(const std::string &Filename) { 56764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string::size_type DotPos = Filename.rfind('.'); 56864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 56964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DotPos == std::string::npos) { 57064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // no file extension 57164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 57264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 57364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 57464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Ext = std::string(Filename.begin()+DotPos+1, Filename.end()); 57564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // C header: .h 57664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // C++ header: .hh or .H; 57764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Ext == "h" || Ext == "hh" || Ext == "H"; 57864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 57964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 58064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianRewriteModernObjC::RewriteModernObjC(std::string inFile, raw_ostream* OS, 58164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &D, const LangOptions &LOpts, 58264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool silenceMacroWarn) 58364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : Diags(D), LangOpts(LOpts), InFileName(inFile), OutFile(OS), 58464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning(silenceMacroWarn) { 58564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IsHeader = IsHeaderFile(inFile); 58664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFailedDiag = Diags.getCustomDiagID(DiagnosticsEngine::Warning, 58764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "rewriting sub-expression within a macro (may not be correct)"); 588d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian // FIXME. This should be an error. But if block is not called, it is OK. And it 589d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian // may break including some headers. 590d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian GlobalBlockRewriteFailedDiag = Diags.getCustomDiagID(DiagnosticsEngine::Warning, 591d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian "rewriting block literal declared in global scope is not implemented"); 592d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian 59364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TryFinallyContainsReturnDiag = Diags.getCustomDiagID( 59464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine::Warning, 59564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "rewriter doesn't support user-specified control flow semantics " 59664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "for @try/@finally (code may not execute properly)"); 59764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 59864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 59964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianASTConsumer *clang::CreateModernObjCRewriter(const std::string& InFile, 60064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian raw_ostream* OS, 60164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &Diags, 60264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const LangOptions &LOpts, 60364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SilenceRewriteMacroWarning) { 60464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return new RewriteModernObjC(InFile, OS, Diags, LOpts, SilenceRewriteMacroWarning); 60564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 60664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 60764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InitializeCommon(ASTContext &context) { 60864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context = &context; 60964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SM = &Context->getSourceManager(); 61064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TUDecl = Context->getTranslationUnitDecl(); 61164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFunctionDecl = 0; 61264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperFunctionDecl = 0; 61364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFunctionDecl = 0; 61464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperStretFunctionDecl = 0; 61564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFpretFunctionDecl = 0; 61664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetClassFunctionDecl = 0; 61764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetMetaClassFunctionDecl = 0; 61864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetSuperClassFunctionDecl = 0; 61964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = 0; 62064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CFStringFunctionDecl = 0; 62164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringClassReference = 0; 62264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NSStringRecord = 0; 62364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = 0; 62464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = 0; 62564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = 0; 626df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian GlobalConstructionExp = 0; 62764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl = 0; 62864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolTypeDecl = 0; 62964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl = 0; 63064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BcLabelCount = 0; 63164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperContructorFunctionDecl = 0; 63264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NumObjCStringLiterals = 0; 63364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 63464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 63564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = false; 63664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_impl_method = false; 63764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the ID and start/end of the main file. 63964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileID = SM->getMainFileID(); 64064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const llvm::MemoryBuffer *MainBuf = SM->getBuffer(MainFileID); 64164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileStart = MainBuf->getBufferStart(); 64264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileEnd = MainBuf->getBufferEnd(); 64364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6444e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie Rewrite.setSourceMgr(Context->getSourceManager(), Context->getLangOpts()); 64564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 64664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 64764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 64864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Top Level Driver Code 64964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 65064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 65164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleTopLevelSingleDecl(Decl *D) { 65264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Diags.hasErrorOccurred()) 65364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 65464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 65564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Two cases: either the decl could be in the main file, or it could be in a 65664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // #included file. If the former, rewrite it now. If the later, check to see 65764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // if we rewrote the #include/#import. 65864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc = D->getLocation(); 65964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = SM->getExpansionLoc(Loc); 66064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 66164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If this is for a builtin, ignore it. 66264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Loc.isInvalid()) return; 66364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 66464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Look for built-in declarations that we need to refer during the rewrite. 66564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { 66664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionDecl(FD); 66764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (VarDecl *FVD = dyn_cast<VarDecl>(D)) { 66864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declared in <Foundation/NSString.h> 66964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FVD->getName() == "_NSConstantStringClassReference") { 67064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringClassReference = FVD; 67164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 67264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 67364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(D)) { 67464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCategoryDecl(CD); 67564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCProtocolDecl *PD = dyn_cast<ObjCProtocolDecl>(D)) { 67664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isThisDeclarationADefinition()) 67764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProtocolDecl(PD); 67864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (LinkageSpecDecl *LSD = dyn_cast<LinkageSpecDecl>(D)) { 6798e86b2d03e6e58ef9a58d1d3ad70726ae7a3b4fdFariborz Jahanian // FIXME. This will not work in all situations and leaving it out 6808e86b2d03e6e58ef9a58d1d3ad70726ae7a3b4fdFariborz Jahanian // is harmless. 6818e86b2d03e6e58ef9a58d1d3ad70726ae7a3b4fdFariborz Jahanian // RewriteLinkageSpec(LSD); 6828e86b2d03e6e58ef9a58d1d3ad70726ae7a3b4fdFariborz Jahanian 68364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Recurse into linkage specifications 68464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclContext::decl_iterator DI = LSD->decls_begin(), 68564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DIEnd = LSD->decls_end(); 68664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DI != DIEnd; ) { 68764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>((*DI))) { 68864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!IFace->isThisDeclarationADefinition()) { 68964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Decl *, 8> DG; 69064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc = IFace->getLocStart(); 69164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian do { 69264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCInterfaceDecl>(*DI) && 69364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !cast<ObjCInterfaceDecl>(*DI)->isThisDeclarationADefinition() && 69464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc == (*DI)->getLocStart()) 69564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DG.push_back(*DI); 69664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 69764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 69864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 69964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 70064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } while (DI != DIEnd); 70164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassDecl(DG); 70264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 70364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 704b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian else { 705b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian // Keep track of all interface declarations seen. 706b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian ObjCInterfacesSeen.push_back(IFace); 707b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian ++DI; 708b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian continue; 709b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian } 71064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 71164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 71264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>((*DI))) { 71364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Proto->isThisDeclarationADefinition()) { 71464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Decl *, 8> DG; 71564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc = Proto->getLocStart(); 71664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian do { 71764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCProtocolDecl>(*DI) && 71864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !cast<ObjCProtocolDecl>(*DI)->isThisDeclarationADefinition() && 71964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc == (*DI)->getLocStart()) 72064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DG.push_back(*DI); 72164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 72264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 72364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 72464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 72564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } while (DI != DIEnd); 72664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardProtocolDecl(DG); 72764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 72864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 72964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 73064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 73164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HandleTopLevelSingleDecl(*DI); 73264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 73364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 73464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 73564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we have a decl in the main file, see if we should rewrite it. 73664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SM->isFromMainFile(Loc)) 73764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return HandleDeclInMainFile(D); 73864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 73964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 74064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 74164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Syntactic (non-AST) Rewriting Code 74264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 74364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 74464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteInclude() { 74564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = SM->getLocForStartOfFile(MainFileID); 74664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef MainBuf = SM->getBufferData(MainFileID); 74764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *MainBufStart = MainBuf.begin(); 74864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *MainBufEnd = MainBuf.end(); 74964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian size_t ImportLen = strlen("import"); 75064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 75164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Loop over the whole file, looking for includes. 75264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (const char *BufPtr = MainBufStart; BufPtr < MainBufEnd; ++BufPtr) { 75364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*BufPtr == '#') { 75464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (++BufPtr == MainBufEnd) 75564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 75664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*BufPtr == ' ' || *BufPtr == '\t') 75764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (++BufPtr == MainBufEnd) 75864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 75964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strncmp(BufPtr, "import", ImportLen)) { 76064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace import with include 76164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation ImportLoc = 76264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LocStart.getLocWithOffset(BufPtr-MainBufStart); 76364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(ImportLoc, ImportLen, "include"); 76464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BufPtr += ImportLen; 76564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 76664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 76764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 76864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 76964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 77064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic std::string getIvarAccessString(ObjCIvarDecl *OID) { 77164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceDecl *ClassDecl = OID->getContainingInterface(); 77264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S; 77364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S = "((struct "; 77464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ClassDecl->getIdentifier()->getName(); 77564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_IMPL *)self)->"; 77664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += OID->getName(); 77764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 77864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 77964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 78064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewritePropertyImplDecl(ObjCPropertyImplDecl *PID, 78164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD, 78264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID) { 78364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objcGetPropertyDefined = false; 78464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objcSetPropertyDefined = false; 78564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = PID->getLocStart(); 78664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(startLoc, "// "); 78764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 78864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @synthesize location"); 78964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 79064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "@synthesize: can't find ';'"); 79164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation onePastSemiLoc = 79264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc.getLocWithOffset(semiBuf-startBuf+1); 79364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 79464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 79564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // FIXME: is this correct? 79664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 79764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the 'getter' function. 79864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl *PD = PID->getPropertyDecl(); 79964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCIvarDecl *OID = PID->getPropertyIvarDecl(); 80064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 80164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!OID) 80264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 80364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned Attributes = PD->getPropertyAttributes(); 80464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!PD->getGetterMethodDecl()->isDefined()) { 80564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool GenGetProperty = !(Attributes & ObjCPropertyDecl::OBJC_PR_nonatomic) && 80664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (Attributes & (ObjCPropertyDecl::OBJC_PR_retain | 80764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl::OBJC_PR_copy)); 80864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Getr; 80964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenGetProperty && !objcGetPropertyDefined) { 81064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objcGetPropertyDefined = true; 81164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Is this attribute correct in all cases? 81264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr = "\nextern \"C\" __declspec(dllimport) " 81364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "id objc_getProperty(id, SEL, long, bool);\n"; 81464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 81564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OID->getContainingInterface(), 81664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PD->getGetterMethodDecl(), Getr); 81764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "{ "; 81864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to gain access to the ivar. 81964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See objc-act.c:objc_synthesize_new_getter() for details. 82064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenGetProperty) { 82164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // return objc_getProperty(self, _cmd, offsetof(ClassDecl, OID), 1) 82264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "typedef "; 82364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FPRetType = 0; 82464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeIntoString(PD->getGetterMethodDecl()->getResultType(), Getr, 82564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FPRetType); 82664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += " _TYPE"; 82764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FPRetType) { 82864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ")"; // close the precedence "scope" for "*". 82964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 83064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, emit the argument types (if any). 83164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FPRetType)){ 83264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "("; 83364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) { 83464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i) Getr += ", "; 83564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr = FT->getArgType(i).getAsString( 83664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 83764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ParamStr; 83864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 83964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 84064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->getNumArgs()) Getr += ", "; 84164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "..."; 84264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 84364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ")"; 84464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 84564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "()"; 84664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 84764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ";\n"; 84864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "return (_TYPE)"; 84964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "objc_getProperty(self, _cmd, "; 85064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation(OID, Getr); 85164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ", 1)"; 85264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 85364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 85464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "return " + getIvarAccessString(OID); 85564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "; }"; 85664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, Getr); 85764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 85864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 85964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isReadOnly() || PD->getSetterMethodDecl()->isDefined()) 86064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 86164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 86264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the 'setter' function. 86364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Setr; 86464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool GenSetProperty = Attributes & (ObjCPropertyDecl::OBJC_PR_retain | 86564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl::OBJC_PR_copy); 86664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenSetProperty && !objcSetPropertyDefined) { 86764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objcSetPropertyDefined = true; 86864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Is this attribute correct in all cases? 86964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr = "\nextern \"C\" __declspec(dllimport) " 87064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "void objc_setProperty (id, SEL, long, id, bool, bool);\n"; 87164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 87264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 87364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OID->getContainingInterface(), 87464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PD->getSetterMethodDecl(), Setr); 87564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "{ "; 87664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to initialize the ivar. 87764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See objc-act.c:objc_synthesize_new_setter() for details. 87864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenSetProperty) { 87964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "objc_setProperty (self, _cmd, "; 88064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation(OID, Setr); 88164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += ", (id)"; 88264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += PD->getName(); 88364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += ", "; 88464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Attributes & ObjCPropertyDecl::OBJC_PR_nonatomic) 88564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "0, "; 88664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 88764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "1, "; 88864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Attributes & ObjCPropertyDecl::OBJC_PR_copy) 88964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "1)"; 89064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 89164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "0)"; 89264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 89364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 89464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += getIvarAccessString(OID) + " = "; 89564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += PD->getName(); 89664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 89764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "; }"; 89864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, Setr); 89964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 90064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 90164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void RewriteOneForwardClassDecl(ObjCInterfaceDecl *ForwardDecl, 90264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &typedefString) { 90364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "#ifndef _REWRITER_typedef_"; 90464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 90564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "\n"; 90664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "#define _REWRITER_typedef_"; 90764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 90864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "\n"; 90964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "typedef struct objc_object "; 91064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 911c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian // typedef struct { } _objc_exc_Classname; 912c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian typedefString += ";\ntypedef struct {} _objc_exc_"; 913c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 91464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n#endif\n"; 91564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 91664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 91764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassEpilogue(ObjCInterfaceDecl *ClassDecl, 91864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &typedefString) { 91964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = ClassDecl->getLocStart(); 92064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 92164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiPtr = strchr(startBuf, ';'); 92264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the @class with typedefs corresponding to the classes. 92364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, semiPtr-startBuf+1, typedefString); 92464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 92564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 92664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassDecl(DeclGroupRef D) { 92764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string typedefString; 92864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclGroupRef::iterator I = D.begin(), E = D.end(); I != E; ++I) { 92964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ForwardDecl = cast<ObjCInterfaceDecl>(*I); 93064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (I == D.begin()) { 93164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Translate to typedef's that forward reference structs with the same name 93264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as the class. As a convenience, we include the original declaration 93364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as a comment. 93464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "// @class "; 93564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 93664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n"; 93764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 93864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteOneForwardClassDecl(ForwardDecl, typedefString); 93964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 94064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclGroupRef::iterator I = D.begin(); 94164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassEpilogue(cast<ObjCInterfaceDecl>(*I), typedefString); 94264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 94364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 94464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassDecl( 94564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const llvm::SmallVector<Decl*, 8> &D) { 94664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string typedefString; 94764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < D.size(); i++) { 94864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ForwardDecl = cast<ObjCInterfaceDecl>(D[i]); 94964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i == 0) { 95064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "// @class "; 95164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 95264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n"; 95364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 95464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteOneForwardClassDecl(ForwardDecl, typedefString); 95564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 95664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassEpilogue(cast<ObjCInterfaceDecl>(D[0]), typedefString); 95764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 95864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 95964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteMethodDeclaration(ObjCMethodDecl *Method) { 96064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // When method is a synthesized one, such as a getter/setter there is 96164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // nothing to rewrite. 96264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Method->isImplicit()) 96364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 96464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = Method->getLocStart(); 96564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = Method->getLocEnd(); 96664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 96764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SM->getExpansionLineNumber(LocEnd) > 96864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SM->getExpansionLineNumber(LocStart)) { 96964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LocStart, "#if 0\n"); 97064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocEnd, 1, ";\n#endif\n"); 97164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 97264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LocStart, "// "); 97364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 97464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 97564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 97664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteProperty(ObjCPropertyDecl *prop) { 97764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc = prop->getAtLoc(); 97864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 97964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(Loc, 0, "// "); 98064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle properties that are declared across multiple lines. 98164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 98264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 98364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteCategoryDecl(ObjCCategoryDecl *CatDecl) { 98464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CatDecl->getLocStart(); 98564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 98664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle category headers that are declared across multiple lines. 98764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 988af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (CatDecl->getIvarLBraceLoc().isValid()) 989af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(CatDecl->getIvarLBraceLoc(), "// "); 990d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian for (ObjCCategoryDecl::ivar_iterator 991d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian I = CatDecl->ivar_begin(), E = CatDecl->ivar_end(); I != E; ++I) { 992d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian ObjCIvarDecl *Ivar = (*I); 993d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian SourceLocation LocStart = Ivar->getLocStart(); 994d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian ReplaceText(LocStart, 0, "// "); 995d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian } 996af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (CatDecl->getIvarRBraceLoc().isValid()) 997af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(CatDecl->getIvarRBraceLoc(), "// "); 998af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian 99964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::prop_iterator I = CatDecl->prop_begin(), 100064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = CatDecl->prop_end(); I != E; ++I) 100164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProperty(*I); 100264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 100364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::instmeth_iterator 100464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = CatDecl->instmeth_begin(), E = CatDecl->instmeth_end(); 100564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 100664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 100764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::classmeth_iterator 100864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = CatDecl->classmeth_begin(), E = CatDecl->classmeth_end(); 100964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 101064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 101164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 101264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Lastly, comment out the @end. 101364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(CatDecl->getAtEndRange().getBegin(), 101464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian strlen("@end"), "/* @end */"); 101564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 101664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 101764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) { 101864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = PDecl->getLocStart(); 101964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(PDecl->isThisDeclarationADefinition()); 102064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 102164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle protocol headers that are declared across multiple lines. 102264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 102364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 102464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCProtocolDecl::instmeth_iterator 102564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = PDecl->instmeth_begin(), E = PDecl->instmeth_end(); 102664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 102764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 102864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCProtocolDecl::classmeth_iterator 102964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = PDecl->classmeth_begin(), E = PDecl->classmeth_end(); 103064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 103164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 103264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 103364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCInterfaceDecl::prop_iterator I = PDecl->prop_begin(), 103464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = PDecl->prop_end(); I != E; ++I) 103564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProperty(*I); 103664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 103764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Lastly, comment out the @end. 103864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = PDecl->getAtEndRange().getBegin(); 103964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocEnd, strlen("@end"), "/* @end */"); 104064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 104164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Must comment out @optional/@required 104264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 104364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 104464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (const char *p = startBuf; p < endBuf; p++) { 104564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*p == '@' && !strncmp(p+1, "optional", strlen("optional"))) { 104664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OptionalLoc = LocStart.getLocWithOffset(p-startBuf); 104764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(OptionalLoc, strlen("@optional"), "/* @optional */"); 104864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 104964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 105064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*p == '@' && !strncmp(p+1, "required", strlen("required"))) { 105164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OptionalLoc = LocStart.getLocWithOffset(p-startBuf); 105264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(OptionalLoc, strlen("@required"), "/* @required */"); 105364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 105464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 105564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 105664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 105764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 105864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardProtocolDecl(DeclGroupRef D) { 105964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = (*D.begin())->getLocStart(); 106064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 106164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("Invalid SourceLocation"); 106264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle forward protocol that are declared across multiple lines. 106364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 106464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 106564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 106664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid 106764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianRewriteModernObjC::RewriteForwardProtocolDecl(const llvm::SmallVector<Decl*, 8> &DG) { 106864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = DG[0]->getLocStart(); 106964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 107064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("Invalid SourceLocation"); 107164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle forward protocol that are declared across multiple lines. 107264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 107364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 107464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1075b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanianvoid 1076b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz JahanianRewriteModernObjC::RewriteLinkageSpec(LinkageSpecDecl *LSD) { 1077b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian SourceLocation LocStart = LSD->getExternLoc(); 1078b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian if (LocStart.isInvalid()) 1079b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian llvm_unreachable("Invalid extern SourceLocation"); 1080b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian 1081b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 1082b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian if (!LSD->hasBraces()) 1083b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian return; 1084b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian // FIXME. We don't rewrite well if '{' is not on same line as 'extern'. 1085b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian SourceLocation LocRBrace = LSD->getRBraceLoc(); 1086b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian if (LocRBrace.isInvalid()) 1087b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian llvm_unreachable("Invalid rbrace SourceLocation"); 1088b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian ReplaceText(LocRBrace, 0, "// "); 1089b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian} 1090b3f904f79bbe55f3e088d7174d64d20d186914a1Fariborz Jahanian 109164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteTypeIntoString(QualType T, std::string &ResultStr, 109264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *&FPRetType) { 109364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 109464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "id"; 109564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isFunctionPointerType() || 109664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->isBlockPointerType()) { 109764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // needs special handling, since pointer-to-functions have special 109864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // syntax (where a decaration models use). 109964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType retType = T; 110064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PointeeTy; 110164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType* PT = retType->getAs<PointerType>()) 110264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointeeTy = PT->getPointeeType(); 110364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const BlockPointerType *BPT = retType->getAs<BlockPointerType>()) 110464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointeeTy = BPT->getPointeeType(); 110564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((FPRetType = PointeeTy->getAs<FunctionType>())) { 110664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += FPRetType->getResultType().getAsString( 110764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 110864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "(*"; 110964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 111064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 111164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += T.getAsString(Context->getPrintingPolicy()); 111264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 111364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 111464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCMethodDecl(const ObjCInterfaceDecl *IDecl, 111564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD, 111664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &ResultStr) { 111764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //fprintf(stderr,"In RewriteObjCMethodDecl\n"); 111864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FPRetType = 0; 111964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "\nstatic "; 112064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeIntoString(OMD->getResultType(), ResultStr, FPRetType); 112164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " "; 112264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 112364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique method name 112464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string NameStr; 112564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 112664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isInstanceMethod()) 112764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_I_"; 112864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 112964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_C_"; 113064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 113164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += IDecl->getNameAsString(); 113264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_"; 113364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 113464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCCategoryImplDecl *CID = 113564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext())) { 113664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += CID->getNameAsString(); 113764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_"; 113864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 113964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Append selector names, replacing ':' with '_' 114064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 114164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string selString = OMD->getSelector().getAsString(); 114264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int len = selString.size(); 114364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < len; i++) 114464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (selString[i] == ':') 114564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian selString[i] = '_'; 114664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += selString; 114764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 114864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remember this name for metadata emission 114964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodInternalNames[OMD] = NameStr; 115064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += NameStr; 115164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 115264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite arguments 115364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "("; 115464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 115564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // invisible arguments 115664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isInstanceMethod()) { 115764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType selfTy = Context->getObjCInterfaceType(IDecl); 115864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian selfTy = Context->getPointerType(selfTy); 115964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!LangOpts.MicrosoftExt) { 116064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSynthesizedStructs.count(const_cast<ObjCInterfaceDecl*>(IDecl))) 116164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "struct "; 116264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 116364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // When rewriting for Microsoft, explicitly omit the structure name. 116464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += IDecl->getNameAsString(); 116564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " *"; 116664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 116764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 116864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Context->getObjCClassType().getAsString( 116964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 117064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 117164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " self, "; 117264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Context->getObjCSelType().getAsString(Context->getPrintingPolicy()); 117364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " _cmd"; 117464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 117564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Method arguments. 117664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(), 117764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = OMD->param_end(); PI != E; ++PI) { 117864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParmVarDecl *PDecl = *PI; 117964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ", "; 118064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PDecl->getType()->isObjCQualifiedIdType()) { 118164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "id "; 118264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += PDecl->getNameAsString(); 118364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 118464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = PDecl->getNameAsString(); 118564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = PDecl->getType(); 118664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 11872610f90135571be7bf983c80a1cf7bfa0844dc57Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(QT); 11882610f90135571be7bf983c80a1cf7bfa0844dc57Fariborz Jahanian QT.getAsStringInternal(Name, Context->getPrintingPolicy()); 118964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Name; 119064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 119164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 119264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isVariadic()) 119364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ", ..."; 119464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ") "; 119564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 119664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FPRetType) { 119764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ")"; // close the precedence "scope" for "*". 119864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 119964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, emit the argument types (if any). 120064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FPRetType)) { 120164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "("; 120264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) { 120364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i) ResultStr += ", "; 120464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr = FT->getArgType(i).getAsString( 120564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 120664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ParamStr; 120764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 120864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 120964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->getNumArgs()) ResultStr += ", "; 121064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "..."; 121164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 121264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ")"; 121364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 121464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "()"; 121564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 121664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 121764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 121864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteImplementationDecl(Decl *OID) { 121964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD = dyn_cast<ObjCImplementationDecl>(OID); 122064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID = dyn_cast<ObjCCategoryImplDecl>(OID); 122164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1222d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian if (IMD) { 1223d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian InsertText(IMD->getLocStart(), "// "); 1224af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (IMD->getIvarLBraceLoc().isValid()) 1225af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(IMD->getIvarLBraceLoc(), "// "); 1226af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian for (ObjCImplementationDecl::ivar_iterator 1227af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian I = IMD->ivar_begin(), E = IMD->ivar_end(); I != E; ++I) { 1228af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian ObjCIvarDecl *Ivar = (*I); 1229af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian SourceLocation LocStart = Ivar->getLocStart(); 1230af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 1231d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian } 1232af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (IMD->getIvarRBraceLoc().isValid()) 1233af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(IMD->getIvarRBraceLoc(), "// "); 1234d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian } 1235d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian else 1236d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian InsertText(CID->getLocStart(), "// "); 123764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 123864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::instmeth_iterator 123964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->instmeth_begin() : CID->instmeth_begin(), 124064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->instmeth_end() : CID->instmeth_end(); 124164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 124264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 124364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD = *I; 124464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OMD->getClassInterface(), OMD, ResultStr); 124564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = OMD->getLocStart(); 124664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = OMD->getCompoundBody()->getLocStart(); 124764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 124864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 124964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 125064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, ResultStr); 125164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 125264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 125364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::classmeth_iterator 125464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->classmeth_begin() : CID->classmeth_begin(), 125564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->classmeth_end() : CID->classmeth_end(); 125664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 125764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 125864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD = *I; 125964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OMD->getClassInterface(), OMD, ResultStr); 126064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = OMD->getLocStart(); 126164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = OMD->getCompoundBody()->getLocStart(); 126264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 126364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 126464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 126564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, ResultStr); 126664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 126764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::propimpl_iterator 126864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->propimpl_begin() : CID->propimpl_begin(), 126964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->propimpl_end() : CID->propimpl_end(); 127064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 127164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewritePropertyImplDecl(*I, IMD, CID); 127264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 127364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 127464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(IMD ? IMD->getLocEnd() : CID->getLocEnd(), "// "); 127564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 127664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 127764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteInterfaceDecl(ObjCInterfaceDecl *ClassDecl) { 12788f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // Do not synthesize more than once. 12798f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian if (ObjCSynthesizedStructs.count(ClassDecl)) 12808f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian return; 12818f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // Make sure super class's are written before current class is written. 12828f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian ObjCInterfaceDecl *SuperClass = ClassDecl->getSuperClass(); 12838f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian while (SuperClass) { 12848f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian RewriteInterfaceDecl(SuperClass); 12858f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian SuperClass = SuperClass->getSuperClass(); 12868f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian } 128764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 1288cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian if (!ObjCWrittenInterfaces.count(ClassDecl->getCanonicalDecl())) { 128964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we haven't seen a forward decl - generate a typedef. 1290c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian RewriteOneForwardClassDecl(ClassDecl, ResultStr); 129172c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian RewriteIvarOffsetSymbols(ClassDecl, ResultStr); 129272c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 12934339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteObjCInternalStruct(ClassDecl, ResultStr); 1294cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian // Mark this typedef as having been written into its c++ equivalent. 1295cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian ObjCWrittenInterfaces.insert(ClassDecl->getCanonicalDecl()); 12964339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian 12974339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian for (ObjCInterfaceDecl::prop_iterator I = ClassDecl->prop_begin(), 129864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ClassDecl->prop_end(); I != E; ++I) 12994339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteProperty(*I); 13004339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian for (ObjCInterfaceDecl::instmeth_iterator 130164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = ClassDecl->instmeth_begin(), E = ClassDecl->instmeth_end(); 13024339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian I != E; ++I) 13034339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteMethodDeclaration(*I); 13044339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian for (ObjCInterfaceDecl::classmeth_iterator 130564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = ClassDecl->classmeth_begin(), E = ClassDecl->classmeth_end(); 13064339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian I != E; ++I) 13074339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteMethodDeclaration(*I); 130864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 13094339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian // Lastly, comment out the @end. 13104339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian ReplaceText(ClassDecl->getAtEndRange().getBegin(), strlen("@end"), 13114339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian "/* @end */"); 13124339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian } 131364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 131464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 131564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewritePropertyOrImplicitSetter(PseudoObjectExpr *PseudoOp) { 131664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = PseudoOp->getSourceRange(); 131764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 131864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We just magically know some things about the structure of this 131964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression. 132064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *OldMsg = 132164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast<ObjCMessageExpr>(PseudoOp->getSemanticExpr( 132264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PseudoOp->getNumSemanticExprs() - 1)); 132364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 132464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Because the rewriter doesn't allow us to rewrite rewritten code, 132564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need to suppress rewriting the sub-statements. 132688ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Expr *Base; 132788ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian SmallVector<Expr*, 2> Args; 132864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 132964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 133064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 133164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the base expression if we have one. 133264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = 0; 133364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OldMsg->getReceiverKind() == ObjCMessageExpr::Instance) { 133464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = OldMsg->getInstanceReceiver(); 133564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<OpaqueValueExpr>(Base)->getSourceExpr(); 133664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Base)); 133764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 133888ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian 133988ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian unsigned numArgs = OldMsg->getNumArgs(); 134088ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian for (unsigned i = 0; i < numArgs; i++) { 134188ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Expr *Arg = OldMsg->getArg(i); 134288ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian if (isa<OpaqueValueExpr>(Arg)) 134388ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Arg = cast<OpaqueValueExpr>(Arg)->getSourceExpr(); 134488ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Arg = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Arg)); 134588ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Args.push_back(Arg); 134688ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian } 134764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 134864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 134964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // TODO: avoid this copy. 135064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<SourceLocation, 1> SelLocs; 135164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelectorLocs(SelLocs); 135264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 135364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *NewMsg = 0; 135464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (OldMsg->getReceiverKind()) { 135564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: 135664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 135764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 135864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 135964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getClassReceiverTypeInfo(), 136064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 136164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 136264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 136388ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Args, 136464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 136564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 136664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 136764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 136864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: 136964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 137064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 137164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 137264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base, 137364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 137464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 137564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 137688ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Args, 137764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 137864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 137964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 138064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 138164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: 138264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance: 138364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 138464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 138564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 138664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperLoc(), 138764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getReceiverKind() == ObjCMessageExpr::SuperInstance, 138864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperType(), 138964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 139064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 139164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 139288ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Args, 139364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 139464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 139564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 139664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 139764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 139864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *Replacement = SynthMessageExpr(NewMsg); 139964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmtWithRange(PseudoOp, Replacement, OldRange); 140064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 140164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 140264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 140364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewritePropertyOrImplicitGetter(PseudoObjectExpr *PseudoOp) { 140464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = PseudoOp->getSourceRange(); 140564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 140664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We just magically know some things about the structure of this 140764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression. 140864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *OldMsg = 140964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast<ObjCMessageExpr>(PseudoOp->getResultExpr()->IgnoreImplicit()); 141064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 141164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Because the rewriter doesn't allow us to rewrite rewritten code, 141264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need to suppress rewriting the sub-statements. 141364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Base = 0; 141488ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian SmallVector<Expr*, 1> Args; 141564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 141664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 141764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the base expression if we have one. 141864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OldMsg->getReceiverKind() == ObjCMessageExpr::Instance) { 141964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = OldMsg->getInstanceReceiver(); 142064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<OpaqueValueExpr>(Base)->getSourceExpr(); 142164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Base)); 142264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 142388ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian unsigned numArgs = OldMsg->getNumArgs(); 142488ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian for (unsigned i = 0; i < numArgs; i++) { 142588ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Expr *Arg = OldMsg->getArg(i); 142688ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian if (isa<OpaqueValueExpr>(Arg)) 142788ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Arg = cast<OpaqueValueExpr>(Arg)->getSourceExpr(); 142888ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Arg = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Arg)); 142988ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian Args.push_back(Arg); 143088ec6100812d8f551a824d7884e21aaa68138330Fariborz Jahanian } 143164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 143264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 143364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Intentionally empty. 143464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<SourceLocation, 1> SelLocs; 143564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 143664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *NewMsg = 0; 143764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (OldMsg->getReceiverKind()) { 143864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: 143964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 144064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 144164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 144264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getClassReceiverTypeInfo(), 144364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 144464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 144564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 144664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 144764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 144864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 144964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 145064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 145164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: 145264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 145364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 145464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 145564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base, 145664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 145764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 145864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 145964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 146064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 146164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 146264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 146364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 146464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: 146564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance: 146664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 146764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 146864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 146964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperLoc(), 147064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getReceiverKind() == ObjCMessageExpr::SuperInstance, 147164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperType(), 147264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 147364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 147464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 147564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 147664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 147764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 147864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 147964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 148064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 148164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *Replacement = SynthMessageExpr(NewMsg); 148264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmtWithRange(PseudoOp, Replacement, OldRange); 148364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 148464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 148564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 148664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SynthCountByEnumWithState - To print: 148764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ((unsigned int (*) 148864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int)) 148964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (void *)objc_msgSend)((id)l_collection, 149064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// sel_registerName( 149164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// "countByEnumeratingWithState:objects:count:"), 149264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// &enumState, 149364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (id *)__rw_items, (unsigned int)16) 149464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 149564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthCountByEnumWithState(std::string &buf) { 149664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "((unsigned int (*) (id, SEL, struct __objcFastEnumerationState *, " 149764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "id *, unsigned int))(void *)objc_msgSend)"; 149864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 149964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "((id)l_collection,\n\t\t"; 150064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "sel_registerName(\"countByEnumeratingWithState:objects:count:\"),"; 150164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 150264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "&enumState, " 150364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "(id *)__rw_items, (unsigned int)16)"; 150464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 150564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 150664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteBreakStmt - Rewrite for a break-stmt inside an ObjC2's foreach 150764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// statement to exit to its outer synthesized loop. 150864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 150964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteBreakStmt(BreakStmt *S) { 151064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back())) 151164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 151264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace break with goto __break_label 151364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 151464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 151564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 151664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "goto __break_label_"; 151764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 151864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, strlen("break"), buf); 151964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 152064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 152164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 152264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 152364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteContinueStmt - Rewrite for a continue-stmt inside an ObjC2's foreach 152464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// statement to continue with its inner synthesized loop. 152564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 152664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteContinueStmt(ContinueStmt *S) { 152764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back())) 152864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 152964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace continue with goto __continue_label 153064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 153164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 153264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 153364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "goto __continue_label_"; 153464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 153564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, strlen("continue"), buf); 153664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 153764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 153864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 153964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 154064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCForCollectionStmt - Rewriter for ObjC2's foreach statement. 154164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// It rewrites: 154264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// for ( type elem in collection) { stmts; } 154364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 154464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// Into: 154564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// { 154664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// type elem; 154764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __objcFastEnumerationState enumState = { 0 }; 154864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id __rw_items[16]; 154964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id l_collection = (id)collection; 155064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long limit = [l_collection countByEnumeratingWithState:&enumState 155164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objects:__rw_items count:16]; 155264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// if (limit) { 155364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long startMutations = *enumState.mutationsPtr; 155464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// do { 155564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long counter = 0; 155664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// do { 155764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// if (startMutations != *enumState.mutationsPtr) 155864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objc_enumerationMutation(l_collection); 155964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = (type)enumState.itemsPtr[counter++]; 156064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// stmts; 156164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __continue_label: ; 156264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } while (counter < limit); 156364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } while (limit = [l_collection countByEnumeratingWithState:&enumState 156464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objects:__rw_items count:16]); 156564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = nil; 156664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __break_label: ; 156764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 156864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// else 156964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = nil; 157064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 157164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 157264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S, 157364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OrigEnd) { 157464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!Stmts.empty() && "ObjCForCollectionStmt - Statement stack empty"); 157564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(isa<ObjCForCollectionStmt>(Stmts.back()) && 157664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "ObjCForCollectionStmt Statement stack mismatch"); 157764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!ObjCBcLabelNo.empty() && 157864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "ObjCForCollectionStmt - Label No stack empty"); 157964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 158064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 158164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 158264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef elementName; 158364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string elementTypeAsString; 158464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 158564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "\n{\n\t"; 158664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclStmt *DS = dyn_cast<DeclStmt>(S->getElement())) { 158764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // type elem; 158864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NamedDecl* D = cast<NamedDecl>(DS->getSingleDecl()); 158964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ElementType = cast<ValueDecl>(D)->getType(); 159064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ElementType->isObjCQualifiedIdType() || 159164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ElementType->isObjCQualifiedInterfaceType()) 159264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simply use 'id' for all qualified types. 159364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = "id"; 159464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 159564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = ElementType.getAsString(Context->getPrintingPolicy()); 159664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 159764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " "; 159864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementName = D->getName(); 159964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 160064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ";\n\t"; 160164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 160264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 160364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DR = cast<DeclRefExpr>(S->getElement()); 160464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementName = DR->getDecl()->getName(); 160564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = cast<ValueDecl>(DR->getDecl()); 160664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getType()->isObjCQualifiedIdType() || 160764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD->getType()->isObjCQualifiedInterfaceType()) 160864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simply use 'id' for all qualified types. 160964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = "id"; 161064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 161164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = VD->getType().getAsString(Context->getPrintingPolicy()); 161264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 161364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 161464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct __objcFastEnumerationState enumState = { 0 }; 161564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "struct __objcFastEnumerationState enumState = { 0 };\n\t"; 161664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // id __rw_items[16]; 161764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id __rw_items[16];\n\t"; 161864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // id l_collection = (id) 161964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id l_collection = (id)"; 162064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find start location of 'collection' the hard way! 162164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startCollectionBuf = startBuf; 162264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf += 3; // skip 'for' 162364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf = strchr(startCollectionBuf, '('); 162464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf++; // skip '(' 162564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // find 'in' and skip it. 162664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startCollectionBuf != ' ' || 162764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *(startCollectionBuf+1) != 'i' || *(startCollectionBuf+2) != 'n' || 162864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*(startCollectionBuf+3) != ' ' && 162964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *(startCollectionBuf+3) != '[' && *(startCollectionBuf+3) != '(')) 163064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf++; 163164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf += 3; 163264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 163364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace: "for (type element in" with string constructed thus far. 163464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, startCollectionBuf - startBuf, buf); 163564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace ')' in for '(' type elem in collection ')' with ';' 163664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation rightParenLoc = S->getRParenLoc(); 163764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *rparenBuf = SM->getCharacterData(rightParenLoc); 163864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation lparenLoc = startLoc.getLocWithOffset(rparenBuf-startBuf); 163964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = ";\n\t"; 164064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 164164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // unsigned long limit = [l_collection countByEnumeratingWithState:&enumState 164264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // objects:__rw_items count:16]; 164364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // which is synthesized into: 164464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // unsigned int limit = 164564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ((unsigned int (*) 164664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int)) 164764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (void *)objc_msgSend)((id)l_collection, 164864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // sel_registerName( 164964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // "countByEnumeratingWithState:objects:count:"), 165064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct __objcFastEnumerationState *)&state, 165164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id *)__rw_items, (unsigned int)16); 165264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long limit =\n\t\t"; 165364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthCountByEnumWithState(buf); 165464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ";\n\t"; 165564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// if (limit) { 165664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// unsigned long startMutations = *enumState.mutationsPtr; 165764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// do { 165864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// unsigned long counter = 0; 165964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// do { 166064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// if (startMutations != *enumState.mutationsPtr) 166164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// objc_enumerationMutation(l_collection); 166264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = (type)enumState.itemsPtr[counter++]; 166364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (limit) {\n\t"; 166464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long startMutations = *enumState.mutationsPtr;\n\t"; 166564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "do {\n\t\t"; 166664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long counter = 0;\n\t\t"; 166764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "do {\n\t\t\t"; 166864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (startMutations != *enumState.mutationsPtr)\n\t\t\t\t"; 166964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "objc_enumerationMutation(l_collection);\n\t\t\t"; 167064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 167164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = ("; 167264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 167364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")enumState.itemsPtr[counter++];"; 167464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace ')' in for '(' type elem in collection ')' with all of these. 167564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(lparenLoc, 1, buf); 167664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 167764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// __continue_label: ; 167864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } while (counter < limit); 167964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } while (limit = [l_collection countByEnumeratingWithState:&enumState 168064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// objects:__rw_items count:16]); 168164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = nil; 168264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// __break_label: ; 168364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } 168464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// else 168564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = nil; 168664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } 168764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// 168864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = ";\n\t"; 168964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "__continue_label_"; 169064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 169164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ": ;"; 169264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 169364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "} while (counter < limit);\n\t"; 169464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "} while (limit = "; 169564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthCountByEnumWithState(buf); 169664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ");\n\t"; 169764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 169864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = (("; 169964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 170064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")0);\n\t"; 170164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "__break_label_"; 170264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 170364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ": ;\n\t"; 170464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n\t"; 170564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "else\n\t\t"; 170664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 170764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = (("; 170864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 170964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")0);\n\t"; 171064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 171164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 171264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert all these *after* the statement body. 171364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: If this should support Obj-C++, support CXXTryStmt 171464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<CompoundStmt>(S->getBody())) { 171564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endBodyLoc = OrigEnd.getLocWithOffset(1); 171664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(endBodyLoc, buf); 171764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 171864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* Need to treat single statements specially. For example: 171964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * 172064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * for (A *a in b) if (stuff()) break; 172164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * for (A *a in b) xxxyy; 172264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * 172364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * The following code simply scans ahead to the semi to find the actual end. 172464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 172564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *stmtBuf = SM->getCharacterData(OrigEnd); 172664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(stmtBuf, ';'); 172764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(semiBuf && "Can't find ';'"); 172864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endBodyLoc = OrigEnd.getLocWithOffset(semiBuf-stmtBuf+1); 172964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(endBodyLoc, buf); 173064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 173164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.pop_back(); 173264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCBcLabelNo.pop_back(); 173364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 173464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 173564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1736542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanianstatic void Write_RethrowObject(std::string &buf) { 1737542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "{ struct _FIN { _FIN(id reth) : rethrow(reth) {}\n"; 1738542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "\t~_FIN() { if (rethrow) objc_exception_throw(rethrow); }\n"; 1739542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "\tid rethrow;\n"; 1740542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "\t} _fin_force_rethow(_rethrow);"; 1741542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian} 1742542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 174364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCSynchronizedStmt - 174464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// This routine rewrites @synchronized(expr) stmt; 174564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// into: 174664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objc_sync_enter(expr); 174764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// @try stmt @finally { objc_sync_exit(expr); } 174864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 174964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S) { 175064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 175164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 175264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 175364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 175464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @synchronized location"); 175564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 175664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 175722e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf = "{ id _rethrow = 0; id _sync_obj = "; 1758542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 175964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *lparenBuf = startBuf; 176064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*lparenBuf != '(') lparenBuf++; 176164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, lparenBuf-startBuf+1, buf); 176222e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian 176322e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf = "; objc_sync_enter(_sync_obj);\n"; 176422e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "try {\n\tstruct _SYNC_EXIT { _SYNC_EXIT(id arg) : sync_exit(arg) {}"; 176522e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "\n\t~_SYNC_EXIT() {objc_sync_exit(sync_exit);}"; 176622e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "\n\tid sync_exit;"; 176722e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "\n\t} _sync_exit(_sync_obj);\n"; 176822e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian 176922e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // We can't use S->getSynchExpr()->getLocEnd() to find the end location, since 177022e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // the sync expression is typically a message expression that's already 177122e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // been rewritten! (which implies the SourceLocation's are invalid). 177222e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian SourceLocation RParenExprLoc = S->getSynchBody()->getLocStart(); 177322e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian const char *RParenExprLocBuf = SM->getCharacterData(RParenExprLoc); 177422e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian while (*RParenExprLocBuf != ')') RParenExprLocBuf--; 177522e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian RParenExprLoc = startLoc.getLocWithOffset(RParenExprLocBuf-startBuf); 177622e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian 177722e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian SourceLocation LBranceLoc = S->getSynchBody()->getLocStart(); 177822e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian const char *LBraceLocBuf = SM->getCharacterData(LBranceLoc); 177922e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian assert (*LBraceLocBuf == '{'); 178022e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian ReplaceText(RParenExprLoc, (LBraceLocBuf - SM->getCharacterData(RParenExprLoc) + 1), buf); 1781542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 1782b655bf06d99bbdb58de5d012642b87cabddabd83Fariborz Jahanian SourceLocation startRBraceLoc = S->getSynchBody()->getLocEnd(); 17839ab511cde246a3a4e47b5b1c4aa2b861fcff08d5Matt Beaumont-Gay assert((*SM->getCharacterData(startRBraceLoc) == '}') && 17849ab511cde246a3a4e47b5b1c4aa2b861fcff08d5Matt Beaumont-Gay "bogus @synchronized block"); 1785542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 1786542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf = "} catch (id e) {_rethrow = e;}\n"; 1787542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian Write_RethrowObject(buf); 178864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 1789542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "}\n"; 179064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1791b655bf06d99bbdb58de5d012642b87cabddabd83Fariborz Jahanian ReplaceText(startRBraceLoc, 1, buf); 179264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 179364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 179464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 179564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 179664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::WarnAboutReturnGotoStmts(Stmt *S) 179764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian{ 179864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up traversal of all children. 179964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 180064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) 180164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian WarnAboutReturnGotoStmts(*CI); 180264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 180364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ReturnStmt>(S) || isa<GotoStmt>(S)) { 180464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(S->getLocStart()), 180564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TryFinallyContainsReturnDiag); 180664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 180764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 180864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 180964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 181064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCTryStmt(ObjCAtTryStmt *S) { 1811b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian ObjCAtFinallyStmt *finalStmt = S->getFinallyStmt(); 1812220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian bool noCatch = S->getNumCatchStmts() == 0; 1813b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian std::string buf; 1814b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian 1815b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian if (finalStmt) { 1816220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian if (noCatch) 1817220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "{ id volatile _rethrow = 0;\n"; 1818220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian else { 1819220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "{ id volatile _rethrow = 0;\ntry {\n"; 1820220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian } 1821b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian } 182264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 182364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 182464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 182564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 182664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @try location"); 1827b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian if (finalStmt) 1828b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian ReplaceText(startLoc, 1, buf); 1829b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian else 1830b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian // @try -> try 1831b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian ReplaceText(startLoc, 1, ""); 1832b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian 183364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I) { 183464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCAtCatchStmt *Catch = S->getCatchStmt(I); 18354c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian VarDecl *catchDecl = Catch->getCatchParamDecl(); 1836c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian 183764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = Catch->getLocStart(); 18384c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian bool AtRemoved = false; 18394c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (catchDecl) { 18404c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian QualType t = catchDecl->getType(); 18414c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (const ObjCObjectPointerType *Ptr = t->getAs<ObjCObjectPointerType>()) { 18424c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // Should be a pointer to a class. 18434c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ObjCInterfaceDecl *IDecl = Ptr->getObjectType()->getInterface(); 18444c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (IDecl) { 18454c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian std::string Result; 18464c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian startBuf = SM->getCharacterData(startLoc); 18474c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian assert((*startBuf == '@') && "bogus @catch location"); 18484c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian SourceLocation rParenLoc = Catch->getRParenLoc(); 18494c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian const char *rParenBuf = SM->getCharacterData(rParenLoc); 18504c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian 18514c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // _objc_exc_Foo *_e as argument to catch. 18524c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result = "catch (_objc_exc_"; Result += IDecl->getNameAsString(); 18534c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += " *_"; Result += catchDecl->getNameAsString(); 18544c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += ")"; 18554c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ReplaceText(startLoc, rParenBuf-startBuf+1, Result); 18564c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // Foo *e = (Foo *)_e; 18574c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result.clear(); 18584c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result = "{ "; 18594c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += IDecl->getNameAsString(); 18604c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += " *"; Result += catchDecl->getNameAsString(); 18614c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += " = ("; Result += IDecl->getNameAsString(); Result += "*)"; 18624c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += "_"; Result += catchDecl->getNameAsString(); 18634c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian 18644c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += "; "; 18654c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian SourceLocation lBraceLoc = Catch->getCatchBody()->getLocStart(); 18664c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ReplaceText(lBraceLoc, 1, Result); 18674c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian AtRemoved = true; 18684c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian } 18694c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian } 18704c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian } 18714c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (!AtRemoved) 18724c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // @catch -> catch 18734c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ReplaceText(startLoc, 1, ""); 1874c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian 187564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 1876220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian if (finalStmt) { 1877220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf.clear(); 1878220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian if (noCatch) 1879220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "catch (id e) {_rethrow = e;}\n"; 1880220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian else 1881220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "}\ncatch (id e) {_rethrow = e;}\n"; 1882220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian 1883220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian SourceLocation startFinalLoc = finalStmt->getLocStart(); 1884220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian ReplaceText(startFinalLoc, 8, buf); 1885220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian Stmt *body = finalStmt->getFinallyBody(); 1886220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian SourceLocation startFinalBodyLoc = body->getLocStart(); 1887220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf.clear(); 1888542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian Write_RethrowObject(buf); 1889220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian ReplaceText(startFinalBodyLoc, 1, buf); 1890220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian 1891220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian SourceLocation endFinalBodyLoc = body->getLocEnd(); 1892220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian ReplaceText(endFinalBodyLoc, 1, "}\n}"); 189322e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // Now check for any return/continue/go statements within the @try. 189422e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian WarnAboutReturnGotoStmts(S->getTryBody()); 1895220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian } 1896220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian 189764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 189864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 189964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 190064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// This can't be done with ReplaceStmt(S, ThrowExpr), since 190164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// the throw expression is typically a message expression that's already 190264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// been rewritten! (which implies the SourceLocation's are invalid). 190364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCThrowStmt(ObjCAtThrowStmt *S) { 190464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 190564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 190664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 190764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 190864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @throw location"); 190964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 191064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 191164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* void objc_exception_throw(id) __attribute__((noreturn)); */ 191264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (S->getThrowExpr()) 191364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "objc_exception_throw("; 191440539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian else 191540539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian buf = "throw"; 191664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 191764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // handle "@ throw" correctly. 191864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *wBuf = strchr(startBuf, 'w'); 191964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*wBuf == 'w') && "@throw: can't find 'w'"); 192064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, wBuf-startBuf+1, buf); 192164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 192264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 192364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "@throw: can't find ';'"); 192464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation semiLoc = startLoc.getLocWithOffset(semiBuf-startBuf); 192540539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian if (S->getThrowExpr()) 192640539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian ReplaceText(semiLoc, 1, ");"); 192764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 192864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 192964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 193064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteAtEncode(ObjCEncodeExpr *Exp) { 193164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a new string expression. 193264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType StrType = Context->getPointerType(Context->CharTy); 193364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StrEncoding; 193464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForType(Exp->getEncodedType(), StrEncoding); 193564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Replacement = StringLiteral::Create(*Context, StrEncoding, 193664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 193764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StrType, SourceLocation()); 193864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, Replacement); 193964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 194064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace this subexpr in the parent. 194164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 194264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 194364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 194464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 194564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteAtSelector(ObjCSelectorExpr *Exp) { 194664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SelGetUidFunctionDecl) 194764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSelGetUidFunctionDecl(); 194864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(SelGetUidFunctionDecl && "Can't find sel_registerName() decl"); 194964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a call to sel_registerName("selName"). 195064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> SelExprs; 195164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 195264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 195364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getSelector().getAsString(), 195464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 195564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 195664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 195764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &SelExprs[0], SelExprs.size()); 195864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, SelExp); 195964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 196064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return SelExp; 196164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 196264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 196364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianCallExpr *RewriteModernObjC::SynthesizeCallToFunctionDecl( 196464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD, Expr **args, unsigned nargs, SourceLocation StartLoc, 196564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc) { 196664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 196764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = FD->getType(); 196864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 196964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 197064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = 1971f4b88a45902af1802a1cb42ba48b1c474474f228John McCall new (Context) DeclRefExpr(FD, false, msgSendType, VK_LValue, SourceLocation()); 197264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 197364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, we cast the reference to a pointer to the objc_msgSend type. 197464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType pToFunc = Context->getPointerType(msgSendType); 197564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImplicitCastExpr *ICE = 197664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImplicitCastExpr::Create(*Context, pToFunc, CK_FunctionToPointerDecay, 197764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DRE, 0, VK_RValue); 197864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 197964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 198064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 198164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Exp = 198264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) CallExpr(*Context, ICE, args, nargs, 198364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getCallResultType(*Context), 198464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, EndLoc); 198564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Exp; 198664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 198764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 198864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool scanForProtocolRefs(const char *startBuf, const char *endBuf, 198964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&startRef, const char *&endRef) { 199064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (startBuf < endBuf) { 199164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '<') 199264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startRef = startBuf; // mark the start. 199364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '>') { 199464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (startRef && *startRef == '<') { 199564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endRef = startBuf; // mark the end. 199664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 199764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 199864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 199964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 200064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 200164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 200264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 200364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 200464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 200564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void scanToNextArgument(const char *&argRef) { 200664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int angle = 0; 200764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argRef != ')' && (*argRef != ',' || angle > 0)) { 200864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*argRef == '<') 200964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian angle++; 201064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*argRef == '>') 201164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian angle--; 201264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argRef++; 201364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 201464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(angle == 0 && "scanToNextArgument - bad protocol type syntax"); 201564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 201664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 201764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::needToScanForQualifiers(QualType T) { 201864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 201964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 202064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType *PT = T->getAs<PointerType>()) { 202164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT->getPointeeType()->isObjCQualifiedIdType()) 202264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 202364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 202464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCObjectPointerType()) { 202564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = T->getPointeeType(); 202664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return T->isObjCQualifiedInterfaceType(); 202764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 202864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isArrayType()) { 202964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ElemTy = Context->getBaseElementType(T); 203064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return needToScanForQualifiers(ElemTy); 203164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 203264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 203364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 203464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 203564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCQualifiedInterfaceTypes(Expr *E) { 203664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = E->getType(); 203764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(Type)) { 203864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc, EndLoc; 203964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 204064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) { 204164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = ECE->getLParenLoc(); 204264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc = ECE->getRParenLoc(); 204364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 204464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = E->getLocStart(); 204564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc = E->getLocEnd(); 204664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 204764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This will defend against trying to rewrite synthesized expressions. 204864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Loc.isInvalid() || EndLoc.isInvalid()) 204964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 205064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 205164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(Loc); 205264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(EndLoc); 205364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 205464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 205564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 205664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = Loc.getLocWithOffset(startRef-startBuf); 205764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = Loc.getLocWithOffset(endRef-startBuf+1); 205864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 205964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 206064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 206164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 206264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 206364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 206464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 206564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCQualifiedInterfaceTypes(Decl *Dcl) { 206664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc; 206764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type; 206864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *proto = 0; 206964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(Dcl)) { 207064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = VD->getLocation(); 207164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = VD->getType(); 207264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 207364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FunctionDecl *FD = dyn_cast<FunctionDecl>(Dcl)) { 207464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = FD->getLocation(); 207564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Check for ObjC 'id' and class types that have been adorned with protocol 207664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // information (id<p>, C<p>*). The protocol references need to be rewritten! 207764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *funcType = FD->getType()->getAs<FunctionType>(); 207864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(funcType && "missing function type"); 207964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian proto = dyn_cast<FunctionProtoType>(funcType); 208064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 208164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 208264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = proto->getResultType(); 208364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 208464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FieldDecl *FD = dyn_cast<FieldDecl>(Dcl)) { 208564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = FD->getLocation(); 208664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = FD->getType(); 208764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 208864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 208964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 209064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 209164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(Type)) { 209264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since types are unique, we need to scan the buffer. 209364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 209464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(Loc); 209564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = endBuf; 209664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != ';' && *startBuf != '<' && startBuf != MainFileStart) 209764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf--; // scan backward (from the decl location) for return type. 209864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 209964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 210064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 210164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = Loc.getLocWithOffset(startRef-endBuf); 210264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = Loc.getLocWithOffset(endRef-endBuf+1); 210364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 210464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 210564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 210664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 210764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 210864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 210964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // most likely, was a variable 211064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now check arguments. 211164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(Loc); 211264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startFuncBuf = startBuf; 211364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < proto->getNumArgs(); i++) { 211464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(proto->getArgType(i))) { 211564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since types are unique, we need to scan the buffer. 211664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 211764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = startBuf; 211864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scan forward (from the decl location) for argument types. 211964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian scanToNextArgument(endBuf); 212064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 212164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 212264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 212364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = 212464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc.getLocWithOffset(startRef-startFuncBuf); 212564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = 212664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc.getLocWithOffset(endRef-startFuncBuf+1); 212764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 212864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 212964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 213064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 213164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = ++endBuf; 213264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 213364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 213464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If the function name is derived from a macro expansion, then the 213564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // argument buffer will not follow the name. Need to speak with Chris. 213664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf && *startBuf != ')' && *startBuf != ',') 213764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; // scan forward (from the decl location) for argument types. 213864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 213964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 214064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 214164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 214264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 214364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteTypeOfDecl(VarDecl *ND) { 214464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = ND->getType(); 214564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Type* TypePtr = QT->getAs<Type>(); 214664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isa<TypeOfExprType>(TypePtr)) 214764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 214864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (isa<TypeOfExprType>(TypePtr)) { 214964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const TypeOfExprType *TypeOfExprTypePtr = cast<TypeOfExprType>(TypePtr); 215064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = TypeOfExprTypePtr->getUnderlyingExpr()->getType(); 215164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypePtr = QT->getAs<Type>(); 215264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 215364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. This will not work for multiple declarators; as in: 215464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // __typeof__(a) b,c,d; 215564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeAsString(QT.getAsString(Context->getPrintingPolicy())); 215664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getTypeSpecStartLoc(); 215764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 215864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ND->getInit()) { 215964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 216064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeAsString += " " + Name + " = "; 216164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *E = ND->getInit(); 216264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc; 216364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) 216464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = ECE->getLParenLoc(); 216564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 216664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = E->getLocStart(); 216764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = SM->getExpansionLoc(startLoc); 216864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(startLoc); 216964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf-1, TypeAsString); 217064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 217164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 217264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation X = ND->getLocEnd(); 217364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian X = SM->getExpansionLoc(X); 217464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(X); 217564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf-1, TypeAsString); 217664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 217764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 217864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 217964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthSelGetUidFunctionDecl - SEL sel_registerName(const char *str); 218064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthSelGetUidFunctionDecl() { 218164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *SelGetUidIdent = &Context->Idents.get("sel_registerName"); 218264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 218364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 218464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getFuncType = 218564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSimpleFunctionType(Context->getObjCSelType(), &ArgTys[0], ArgTys.size()); 218664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 218764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 218864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 218964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidIdent, getFuncType, 0, 219064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 219164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 219264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 219364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 219464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteFunctionDecl(FunctionDecl *FD) { 219564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declared in <objc/objc.h> 219664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->getIdentifier() && 219764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getName() == "sel_registerName") { 219864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = FD; 219964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 220064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 220164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(FD); 220264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 220364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 220464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerType(std::string& Str, QualType Type) { 220564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString(Type.getAsString(Context->getPrintingPolicy())); 220664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = TypeString.c_str(); 220764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strchr(argPtr, '^')) { 220864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += TypeString; 220964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 221064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 221164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr) { 221264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += (*argPtr == '^' ? '*' : *argPtr); 221364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; 221464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 221564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 221664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 221764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// FIXME. Consolidate this routine with RewriteBlockPointerType. 221864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerTypeVariable(std::string& Str, 221964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD) { 222064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = VD->getType(); 222164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString(Type.getAsString(Context->getPrintingPolicy())); 222264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = TypeString.c_str(); 222364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int paren = 0; 222464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr) { 222564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 222664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': 222764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 222864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian paren++; 222964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 223064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': 223164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 223264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian paren--; 223364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 223464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 223564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += '*'; 223664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (paren == 1) 223764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += VD->getNameAsString(); 223864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 223964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: 224064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 224164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 224264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 224364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; 224464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 224564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 224664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 2247b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanianvoid RewriteModernObjC::RewriteBlockLiteralFunctionDecl(FunctionDecl *FD) { 2248b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian SourceLocation FunLocStart = FD->getTypeSpecStartLoc(); 2249b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian const FunctionType *funcType = FD->getType()->getAs<FunctionType>(); 2250b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian const FunctionProtoType *proto = dyn_cast<FunctionProtoType>(funcType); 2251b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian if (!proto) 2252b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian return; 2253b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian QualType Type = proto->getResultType(); 2254b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian std::string FdStr = Type.getAsString(Context->getPrintingPolicy()); 2255b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian FdStr += " "; 2256b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian FdStr += FD->getName(); 2257b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian FdStr += "("; 2258b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian unsigned numArgs = proto->getNumArgs(); 2259b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian for (unsigned i = 0; i < numArgs; i++) { 2260b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian QualType ArgType = proto->getArgType(i); 2261b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian RewriteBlockPointerType(FdStr, ArgType); 2262b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian if (i+1 < numArgs) 2263b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian FdStr += ", "; 2264b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian } 2265b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian FdStr += ");\n"; 2266b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian InsertText(FunLocStart, FdStr); 2267b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian} 2268b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian 2269b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian// SynthSuperContructorFunctionDecl - id __rw_objc_super(id obj, id super); 227064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthSuperContructorFunctionDecl() { 227164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SuperContructorFunctionDecl) 227264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 227364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("__rw_objc_super"); 227464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 227564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 227664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 227764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 227864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 227964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 228064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 228164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperContructorFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 228264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 228364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 228464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 228564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 228664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 228764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 228864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 228964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendFunctionDecl - id objc_msgSend(id self, SEL op, ...); 229064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendFunctionDecl() { 229164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend"); 229264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 229364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 229464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 229564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 229664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 229764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 229864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 229964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 230064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 230164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 230264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 230364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 230464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 230564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 230664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 230764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 230864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 230964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 2310b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian// SynthMsgSendSuperFunctionDecl - id objc_msgSendSuper(void); 231164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendSuperFunctionDecl() { 231264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSendSuper"); 2313b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian SmallVector<QualType, 2> ArgTys; 2314b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian ArgTys.push_back(Context->VoidTy); 231564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 2316b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian &ArgTys[0], 1, 231764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 231864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 231964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 232064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 232164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 232264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 232364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 232464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 232564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 232664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendStretFunctionDecl - id objc_msgSend_stret(id self, SEL op, ...); 232764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendStretFunctionDecl() { 232864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_stret"); 232964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 233064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 233164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 233264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 233364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 233464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 233564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 233664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 233764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 233864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 233964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 234064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 234164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 234264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 234364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 234464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 234564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 234664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 234764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendSuperStretFunctionDecl - 2348b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian// id objc_msgSendSuper_stret(void); 234964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendSuperStretFunctionDecl() { 235064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = 235164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_msgSendSuper_stret"); 2352b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian SmallVector<QualType, 2> ArgTys; 2353b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian ArgTys.push_back(Context->VoidTy); 235464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 2355b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian &ArgTys[0], 1, 235664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 235764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 235864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 235964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 236064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 236164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 236264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 236364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 236464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 236564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendFpretFunctionDecl - double objc_msgSend_fpret(id self, SEL op, ...); 236664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendFpretFunctionDecl() { 236764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_fpret"); 236864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 236964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 237064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 237164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 237264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 237364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 237464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 237564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->DoubleTy, 237664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 237764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 237864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFpretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 237964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 238064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 238164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 238264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 238364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 238464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 238564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 238664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetClassFunctionDecl - id objc_getClass(const char *name); 238764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetClassFunctionDecl() { 238864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getClass"); 238964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 239064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 239164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCIdType(), 239264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 239364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 239464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 239564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 239664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassIdent, getClassType, 0, 239764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 239864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 239964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 240064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 240164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetSuperClassFunctionDecl - Class class_getSuperclass(Class cls); 240264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetSuperClassFunctionDecl() { 240364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getSuperClassIdent = 240464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("class_getSuperclass"); 240564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 240664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getObjCClassType()); 240764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCClassType(), 240864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 240964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetSuperClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 241064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 241164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 241264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSuperClassIdent, 241364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassType, 0, 241464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 241564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, 241664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false); 241764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 241864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 241964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetMetaClassFunctionDecl - id objc_getMetaClass(const char *name); 242064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetMetaClassFunctionDecl() { 242164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getMetaClass"); 242264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 242364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 242464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCIdType(), 242564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 242664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetMetaClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 242764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 242864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 242964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassIdent, getClassType, 0, 243064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 243164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 243264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 243364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 243464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCStringLiteral(ObjCStringLiteral *Exp) { 243564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType strType = getConstantStringStructType(); 243664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 243764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "__NSConstantStringImpl_"; 243864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 243964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string tmpName = InFileName; 244064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned i; 244164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (i=0; i < tmpName.length(); i++) { 244264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char c = tmpName.at(i); 244364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace any non alphanumeric characters with '_'. 244464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isalpha(c) && (c < '0' || c > '9')) 244564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian tmpName[i] = '_'; 244664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 244764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += tmpName; 244864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_"; 244964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(NumObjCStringLiterals++); 245064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 245164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "static __NSConstantStringImpl " + S; 245264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " __attribute__ ((section (\"__DATA, __cfstring\"))) = {__CFConstantStringClassReference,"; 245364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "0x000007c8,"; // utf8_str 245464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The pretty printer for StringLiteral handles escape characters properly. 245564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string prettyBufS; 245664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream prettyBuf(prettyBufS); 245764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getString()->printPretty(prettyBuf, *Context, 0, 245864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PrintingPolicy(LangOpts)); 245964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += prettyBuf.str(); 246064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += ","; 246164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += utostr(Exp->getString()->getByteLength()) + "};\n"; 246264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 246364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 246464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), &Context->Idents.get(S), 246564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian strType, 0, SC_Static, SC_None); 2466f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(NewVD, false, strType, VK_LValue, 246764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 246864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Unop = new (Context) UnaryOperator(DRE, UO_AddrOf, 246964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(DRE->getType()), 247064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 247164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 247264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // cast to NSConstantString * 247364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *cast = NoTypeInfoCStyleCastExpr(Context, Exp->getType(), 247464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_CPointerToObjCPointerCast, Unop); 247564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, cast); 247664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 247764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return cast; 247864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 247964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 248055947042d86773b7e16aec12070e1ffe62bff4a6Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCBoolLiteralExpr(ObjCBoolLiteralExpr *Exp) { 248155947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian unsigned IntSize = 248255947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 248355947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian 248455947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian Expr *FlagExp = IntegerLiteral::Create(*Context, 248555947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian llvm::APInt(IntSize, Exp->getValue()), 248655947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian Context->IntTy, Exp->getLocation()); 248755947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian CastExpr *cast = NoTypeInfoCStyleCastExpr(Context, Context->ObjCBuiltinBoolTy, 248855947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian CK_BitCast, FlagExp); 248955947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(Exp->getLocation(), Exp->getExprLoc(), 249055947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian cast); 249155947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian ReplaceStmt(Exp, PE); 249255947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian return PE; 249355947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian} 249455947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian 2495eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick BeardStmt *RewriteModernObjC::RewriteObjCBoxedExpr(ObjCBoxedExpr *Exp) { 24960f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // synthesize declaration of helper functions needed in this routine. 24970f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian if (!SelGetUidFunctionDecl) 24980f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SynthSelGetUidFunctionDecl(); 24990f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // use objc_msgSend() for all. 25000f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian if (!MsgSendFunctionDecl) 25010f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SynthMsgSendFunctionDecl(); 25020f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian if (!GetClassFunctionDecl) 25030f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SynthGetClassFunctionDecl(); 25040f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25050f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian FunctionDecl *MsgSendFlavor = MsgSendFunctionDecl; 25060f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SourceLocation StartLoc = Exp->getLocStart(); 25070f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SourceLocation EndLoc = Exp->getLocEnd(); 25080f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25090f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // Synthesize a call to objc_msgSend(). 25100f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SmallVector<Expr*, 4> MsgExprs; 25110f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SmallVector<Expr*, 4> ClsExprs; 25120f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 25130f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 2514eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard // Create a call to objc_getClass("<BoxingClass>"). It will be the 1st argument. 2515eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard ObjCMethodDecl *BoxingMethod = Exp->getBoxingMethod(); 2516eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard ObjCInterfaceDecl *BoxingClass = BoxingMethod->getClassInterface(); 25170f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 2518eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard IdentifierInfo *clsName = BoxingClass->getIdentifier(); 25190f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 25200f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian clsName->getName(), 25210f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian StringLiteral::Ascii, false, 25220f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian argType, SourceLocation())); 25230f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 25240f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian &ClsExprs[0], 25250f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian ClsExprs.size(), 25260f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian StartLoc, EndLoc); 25270f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian MsgExprs.push_back(Cls); 25280f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 2529eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard // Create a call to sel_registerName("<BoxingMethod>:"), etc. 25300f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // it will be the 2nd argument. 25310f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SmallVector<Expr*, 4> SelExprs; 25320f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 2533eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard BoxingMethod->getSelector().getAsString(), 25340f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian StringLiteral::Ascii, false, 25350f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian argType, SourceLocation())); 25360f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 25370f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian &SelExprs[0], SelExprs.size(), 25380f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian StartLoc, EndLoc); 25390f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian MsgExprs.push_back(SelExp); 25400f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 2541eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard // User provided sub-expression is the 3rd, and last, argument. 2542eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard Expr *subExpr = Exp->getSubExpr(); 2543eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(subExpr)) { 25440f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian QualType type = ICE->getType(); 25450f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian const Expr *SubExpr = ICE->IgnoreParenImpCasts(); 25460f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian CastKind CK = CK_BitCast; 25470f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian if (SubExpr->getType()->isIntegralType(*Context) && type->isBooleanType()) 25480f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian CK = CK_IntegralToBoolean; 2549eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard subExpr = NoTypeInfoCStyleCastExpr(Context, type, CK, subExpr); 25500f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian } 2551eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard MsgExprs.push_back(subExpr); 25520f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25530f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian SmallVector<QualType, 4> ArgTypes; 25540f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian ArgTypes.push_back(Context->getObjCIdType()); 25550f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian ArgTypes.push_back(Context->getObjCSelType()); 2556eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard for (ObjCMethodDecl::param_iterator PI = BoxingMethod->param_begin(), 2557eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard E = BoxingMethod->param_end(); PI != E; ++PI) 25580f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian ArgTypes.push_back((*PI)->getType()); 2559eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard 25600f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian QualType returnType = Exp->getType(); 25610f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 25620f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian QualType msgSendType = MsgSendFlavor->getType(); 25630f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25640f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 25650f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(MsgSendFlavor, false, msgSendType, 25660f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian VK_LValue, SourceLocation()); 25670f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25680f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian CastExpr *cast = NoTypeInfoCStyleCastExpr(Context, 2569eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard Context->getPointerType(Context->VoidTy), 2570eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard CK_BitCast, DRE); 25710f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25720f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // Now do the "normal" pointer to function cast. 25730f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian QualType castType = 2574eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 2575eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard BoxingMethod->isVariadic()); 25760f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian castType = Context->getPointerType(castType); 25770f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 25780f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian cast); 25790f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25800f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 25810f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(StartLoc, EndLoc, cast); 25820f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 25830f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 25840f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 25850f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian MsgExprs.size(), 25860f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian FT->getResultType(), VK_RValue, 25870f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian EndLoc); 25880f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian ReplaceStmt(Exp, CE); 25890f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian return CE; 25900f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian} 25910f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 259286cff6080ba1db907009885adb5c6c64007e8b48Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCArrayLiteralExpr(ObjCArrayLiteral *Exp) { 259386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // synthesize declaration of helper functions needed in this routine. 259486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian if (!SelGetUidFunctionDecl) 259586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SynthSelGetUidFunctionDecl(); 259686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // use objc_msgSend() for all. 259786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian if (!MsgSendFunctionDecl) 259886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SynthMsgSendFunctionDecl(); 259986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian if (!GetClassFunctionDecl) 260086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SynthGetClassFunctionDecl(); 260186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 260286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian FunctionDecl *MsgSendFlavor = MsgSendFunctionDecl; 260386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SourceLocation StartLoc = Exp->getLocStart(); 260486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SourceLocation EndLoc = Exp->getLocEnd(); 260586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 2606e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Build the expression: __NSContainer_literal(int, ...).arr 2607b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian QualType IntQT = Context->IntTy; 2608b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian QualType NSArrayFType = 2609b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian getSimpleFunctionType(Context->VoidTy, &IntQT, 1, true); 2610e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian std::string NSArrayFName("__NSContainer_literal"); 2611b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian FunctionDecl *NSArrayFD = SynthBlockInitFunctionDecl(NSArrayFName); 2612b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian DeclRefExpr *NSArrayDRE = 2613b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian new (Context) DeclRefExpr(NSArrayFD, false, NSArrayFType, VK_RValue, 2614b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian SourceLocation()); 2615b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian 2616b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian SmallVector<Expr*, 16> InitExprs; 2617b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian unsigned NumElements = Exp->getNumElements(); 2618b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian unsigned UnsignedIntSize = 2619b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->UnsignedIntTy)); 2620b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Expr *count = IntegerLiteral::Create(*Context, 2621b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian llvm::APInt(UnsignedIntSize, NumElements), 2622b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Context->UnsignedIntTy, SourceLocation()); 2623b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian InitExprs.push_back(count); 2624b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian for (unsigned i = 0; i < NumElements; i++) 2625b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian InitExprs.push_back(Exp->getElement(i)); 2626b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Expr *NSArrayCallExpr = 2627b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian new (Context) CallExpr(*Context, NSArrayDRE, &InitExprs[0], InitExprs.size(), 2628b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian NSArrayFType, VK_LValue, SourceLocation()); 2629b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian 2630b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian FieldDecl *ARRFD = FieldDecl::Create(*Context, 0, SourceLocation(), 2631b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian SourceLocation(), 2632b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian &Context->Idents.get("arr"), 2633b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Context->getPointerType(Context->VoidPtrTy), 0, 2634b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 2635b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian /*HasInit=*/false); 2636b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian MemberExpr *ArrayLiteralME = 2637b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian new (Context) MemberExpr(NSArrayCallExpr, false, ARRFD, 2638b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian SourceLocation(), 2639b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian ARRFD->getType(), VK_LValue, 2640b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian OK_Ordinary); 2641b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian QualType ConstIdT = Context->getObjCIdType().withConst(); 2642b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian CStyleCastExpr * ArrayLiteralObjects = 2643b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, 2644b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Context->getPointerType(ConstIdT), 2645b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian CK_BitCast, 2646b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian ArrayLiteralME); 2647b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian 264886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // Synthesize a call to objc_msgSend(). 264986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SmallVector<Expr*, 32> MsgExprs; 265086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SmallVector<Expr*, 4> ClsExprs; 265186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 265286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian QualType expType = Exp->getType(); 265386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 265486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // Create a call to objc_getClass("NSArray"). It will be th 1st argument. 265586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ObjCInterfaceDecl *Class = 265686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian expType->getPointeeType()->getAs<ObjCObjectType>()->getInterface(); 265786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 265886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian IdentifierInfo *clsName = Class->getIdentifier(); 265986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 266086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian clsName->getName(), 266186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian StringLiteral::Ascii, false, 266286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian argType, SourceLocation())); 266386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 266486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian &ClsExprs[0], 266586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ClsExprs.size(), 266686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian StartLoc, EndLoc); 266786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian MsgExprs.push_back(Cls); 266886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 266986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // Create a call to sel_registerName("arrayWithObjects:count:"). 267086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // it will be the 2nd argument. 267186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SmallVector<Expr*, 4> SelExprs; 267286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ObjCMethodDecl *ArrayMethod = Exp->getArrayWithObjectsMethod(); 267386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 267486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ArrayMethod->getSelector().getAsString(), 267586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian StringLiteral::Ascii, false, 267686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian argType, SourceLocation())); 267786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 267886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian &SelExprs[0], SelExprs.size(), 267986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian StartLoc, EndLoc); 268086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian MsgExprs.push_back(SelExp); 268186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 2682b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian // (const id [])objects 2683b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian MsgExprs.push_back(ArrayLiteralObjects); 268486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 2685b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian // (NSUInteger)cnt 2686b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Expr *cnt = IntegerLiteral::Create(*Context, 2687b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian llvm::APInt(UnsignedIntSize, NumElements), 2688b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Context->UnsignedIntTy, SourceLocation()); 2689b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian MsgExprs.push_back(cnt); 269086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 269186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 269286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian SmallVector<QualType, 4> ArgTypes; 269386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ArgTypes.push_back(Context->getObjCIdType()); 269486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ArgTypes.push_back(Context->getObjCSelType()); 269586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian for (ObjCMethodDecl::param_iterator PI = ArrayMethod->param_begin(), 269686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian E = ArrayMethod->param_end(); PI != E; ++PI) 269786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ArgTypes.push_back((*PI)->getType()); 269886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 269986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian QualType returnType = Exp->getType(); 270086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 270186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian QualType msgSendType = MsgSendFlavor->getType(); 270286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 270386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 270486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(MsgSendFlavor, false, msgSendType, 270586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian VK_LValue, SourceLocation()); 270686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 270786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian CastExpr *cast = NoTypeInfoCStyleCastExpr(Context, 270886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian Context->getPointerType(Context->VoidTy), 270986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian CK_BitCast, DRE); 271086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 271186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // Now do the "normal" pointer to function cast. 271286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian QualType castType = 271386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 271486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ArrayMethod->isVariadic()); 271586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian castType = Context->getPointerType(castType); 271686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 271786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian cast); 271886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 271986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 272086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(StartLoc, EndLoc, cast); 272186cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 272286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 272386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 272486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian MsgExprs.size(), 272586cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian FT->getResultType(), VK_RValue, 272686cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian EndLoc); 272786cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian ReplaceStmt(Exp, CE); 272886cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian return CE; 272986cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian} 273086cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 2731e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz JahanianStmt *RewriteModernObjC::RewriteObjCDictionaryLiteralExpr(ObjCDictionaryLiteral *Exp) { 2732e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // synthesize declaration of helper functions needed in this routine. 2733e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian if (!SelGetUidFunctionDecl) 2734e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SynthSelGetUidFunctionDecl(); 2735e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // use objc_msgSend() for all. 2736e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian if (!MsgSendFunctionDecl) 2737e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SynthMsgSendFunctionDecl(); 2738e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian if (!GetClassFunctionDecl) 2739e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SynthGetClassFunctionDecl(); 2740e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2741e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian FunctionDecl *MsgSendFlavor = MsgSendFunctionDecl; 2742e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SourceLocation StartLoc = Exp->getLocStart(); 2743e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SourceLocation EndLoc = Exp->getLocEnd(); 2744e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2745e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Build the expression: __NSContainer_literal(int, ...).arr 2746e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType IntQT = Context->IntTy; 2747e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType NSDictFType = 2748e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian getSimpleFunctionType(Context->VoidTy, &IntQT, 1, true); 2749e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian std::string NSDictFName("__NSContainer_literal"); 2750e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian FunctionDecl *NSDictFD = SynthBlockInitFunctionDecl(NSDictFName); 2751e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian DeclRefExpr *NSDictDRE = 2752e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian new (Context) DeclRefExpr(NSDictFD, false, NSDictFType, VK_RValue, 2753e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SourceLocation()); 2754e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2755e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SmallVector<Expr*, 16> KeyExprs; 2756e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SmallVector<Expr*, 16> ValueExprs; 2757e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2758e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian unsigned NumElements = Exp->getNumElements(); 2759e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian unsigned UnsignedIntSize = 2760e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->UnsignedIntTy)); 2761e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Expr *count = IntegerLiteral::Create(*Context, 2762e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian llvm::APInt(UnsignedIntSize, NumElements), 2763e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Context->UnsignedIntTy, SourceLocation()); 2764e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian KeyExprs.push_back(count); 2765e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ValueExprs.push_back(count); 2766e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian for (unsigned i = 0; i < NumElements; i++) { 2767e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ObjCDictionaryElement Element = Exp->getKeyValueElement(i); 2768e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian KeyExprs.push_back(Element.Key); 2769e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ValueExprs.push_back(Element.Value); 2770e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian } 2771e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2772e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // (const id [])objects 2773e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Expr *NSValueCallExpr = 2774e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian new (Context) CallExpr(*Context, NSDictDRE, &ValueExprs[0], ValueExprs.size(), 2775e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian NSDictFType, VK_LValue, SourceLocation()); 2776e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2777e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian FieldDecl *ARRFD = FieldDecl::Create(*Context, 0, SourceLocation(), 2778e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SourceLocation(), 2779e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian &Context->Idents.get("arr"), 2780e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Context->getPointerType(Context->VoidPtrTy), 0, 2781e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 2782e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian /*HasInit=*/false); 2783e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MemberExpr *DictLiteralValueME = 2784e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian new (Context) MemberExpr(NSValueCallExpr, false, ARRFD, 2785e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SourceLocation(), 2786e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ARRFD->getType(), VK_LValue, 2787e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian OK_Ordinary); 2788e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType ConstIdT = Context->getObjCIdType().withConst(); 2789e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CStyleCastExpr * DictValueObjects = 2790e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian NoTypeInfoCStyleCastExpr(Context, 2791e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Context->getPointerType(ConstIdT), 2792e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CK_BitCast, 2793e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian DictLiteralValueME); 2794e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // (const id <NSCopying> [])keys 2795e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Expr *NSKeyCallExpr = 2796e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian new (Context) CallExpr(*Context, NSDictDRE, &KeyExprs[0], KeyExprs.size(), 2797e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian NSDictFType, VK_LValue, SourceLocation()); 2798e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2799e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MemberExpr *DictLiteralKeyME = 2800e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian new (Context) MemberExpr(NSKeyCallExpr, false, ARRFD, 2801e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SourceLocation(), 2802e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ARRFD->getType(), VK_LValue, 2803e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian OK_Ordinary); 2804e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2805e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CStyleCastExpr * DictKeyObjects = 2806e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian NoTypeInfoCStyleCastExpr(Context, 2807e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Context->getPointerType(ConstIdT), 2808e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CK_BitCast, 2809e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian DictLiteralKeyME); 2810e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2811e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2812e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2813e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Synthesize a call to objc_msgSend(). 2814e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SmallVector<Expr*, 32> MsgExprs; 2815e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SmallVector<Expr*, 4> ClsExprs; 2816e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 2817e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType expType = Exp->getType(); 2818e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2819e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Create a call to objc_getClass("NSArray"). It will be th 1st argument. 2820e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ObjCInterfaceDecl *Class = 2821e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian expType->getPointeeType()->getAs<ObjCObjectType>()->getInterface(); 2822e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2823e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian IdentifierInfo *clsName = Class->getIdentifier(); 2824e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 2825e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian clsName->getName(), 2826e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian StringLiteral::Ascii, false, 2827e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian argType, SourceLocation())); 2828e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 2829e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian &ClsExprs[0], 2830e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ClsExprs.size(), 2831e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian StartLoc, EndLoc); 2832e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MsgExprs.push_back(Cls); 2833e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2834e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Create a call to sel_registerName("arrayWithObjects:count:"). 2835e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // it will be the 2nd argument. 2836e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SmallVector<Expr*, 4> SelExprs; 2837e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ObjCMethodDecl *DictMethod = Exp->getDictWithObjectsMethod(); 2838e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 2839e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian DictMethod->getSelector().getAsString(), 2840e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian StringLiteral::Ascii, false, 2841e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian argType, SourceLocation())); 2842e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 2843e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian &SelExprs[0], SelExprs.size(), 2844e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian StartLoc, EndLoc); 2845e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MsgExprs.push_back(SelExp); 2846e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2847e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // (const id [])objects 2848e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MsgExprs.push_back(DictValueObjects); 2849e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2850e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // (const id <NSCopying> [])keys 2851e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MsgExprs.push_back(DictKeyObjects); 2852e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2853e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // (NSUInteger)cnt 2854e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Expr *cnt = IntegerLiteral::Create(*Context, 2855e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian llvm::APInt(UnsignedIntSize, NumElements), 2856e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Context->UnsignedIntTy, SourceLocation()); 2857e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MsgExprs.push_back(cnt); 2858e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2859e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2860e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian SmallVector<QualType, 8> ArgTypes; 2861e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ArgTypes.push_back(Context->getObjCIdType()); 2862e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ArgTypes.push_back(Context->getObjCSelType()); 2863e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian for (ObjCMethodDecl::param_iterator PI = DictMethod->param_begin(), 2864e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian E = DictMethod->param_end(); PI != E; ++PI) { 2865e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType T = (*PI)->getType(); 2866e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian if (const PointerType* PT = T->getAs<PointerType>()) { 2867e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType PointeeTy = PT->getPointeeType(); 2868e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian convertToUnqualifiedObjCType(PointeeTy); 2869e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian T = Context->getPointerType(PointeeTy); 2870e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian } 2871e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ArgTypes.push_back(T); 2872e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian } 2873e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2874e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType returnType = Exp->getType(); 2875e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Get the type, we will need to reference it in a couple spots. 2876e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType msgSendType = MsgSendFlavor->getType(); 2877e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2878e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Create a reference to the objc_msgSend() declaration. 2879e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(MsgSendFlavor, false, msgSendType, 2880e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian VK_LValue, SourceLocation()); 2881e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2882e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CastExpr *cast = NoTypeInfoCStyleCastExpr(Context, 2883e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Context->getPointerType(Context->VoidTy), 2884e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CK_BitCast, DRE); 2885e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2886e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Now do the "normal" pointer to function cast. 2887e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian QualType castType = 2888e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 2889e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian DictMethod->isVariadic()); 2890e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian castType = Context->getPointerType(castType); 2891e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 2892e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian cast); 2893e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2894e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian // Don't forget the parens to enforce the proper binding. 2895e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(StartLoc, EndLoc, cast); 2896e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2897e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 2898e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 2899e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian MsgExprs.size(), 2900e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian FT->getResultType(), VK_RValue, 2901e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian EndLoc); 2902e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian ReplaceStmt(Exp, CE); 2903e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian return CE; 2904e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian} 2905e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 2906b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian// struct __rw_objc_super { 2907b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian// struct objc_object *object; struct objc_object *superClass; 2908b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian// }; 290964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getSuperStructType() { 291064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SuperStructDecl) { 291164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 291264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 2913b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian &Context->Idents.get("__rw_objc_super")); 291464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FieldTypes[2]; 291564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 2916b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // struct objc_object *object; 291764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[0] = Context->getObjCIdType(); 2918b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // struct objc_object *superClass; 2919b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian FieldTypes[1] = Context->getObjCIdType(); 292064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 292164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create fields 292264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < 2; ++i) { 292364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl->addDecl(FieldDecl::Create(*Context, SuperStructDecl, 292464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 292564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 0, 292664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[i], 0, 292764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, 292864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*Mutable=*/false, 292964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false)); 293064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 293164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 293264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl->completeDefinition(); 293364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 293464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTagDeclType(SuperStructDecl); 293564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 293664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 293764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getConstantStringStructType() { 293864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ConstantStringDecl) { 293964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 294064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 294164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__NSConstantStringImpl")); 294264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FieldTypes[4]; 294364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 294464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_object *receiver; 294564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[0] = Context->getObjCIdType(); 294664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // int flags; 294764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[1] = Context->IntTy; 294864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // char *str; 294964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[2] = Context->getPointerType(Context->CharTy); 295064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // long length; 295164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[3] = Context->LongTy; 295264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 295364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create fields 295464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < 4; ++i) { 295564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl->addDecl(FieldDecl::Create(*Context, 295664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl, 295764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 295864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 0, 295964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[i], 0, 296064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, 296164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*Mutable=*/true, 296264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false)); 296364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 296464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 296564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl->completeDefinition(); 296664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 296764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTagDeclType(ConstantStringDecl); 296864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 296964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 297064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthMessageExpr(ObjCMessageExpr *Exp, 297164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc, 297264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc) { 297364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SelGetUidFunctionDecl) 297464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSelGetUidFunctionDecl(); 297564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendFunctionDecl) 297664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendFunctionDecl(); 297764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendSuperFunctionDecl) 297864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendSuperFunctionDecl(); 297964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendStretFunctionDecl) 298064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendStretFunctionDecl(); 298164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendSuperStretFunctionDecl) 298264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendSuperStretFunctionDecl(); 298364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendFpretFunctionDecl) 298464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendFpretFunctionDecl(); 298564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetClassFunctionDecl) 298664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetClassFunctionDecl(); 298764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetSuperClassFunctionDecl) 298864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetSuperClassFunctionDecl(); 298964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetMetaClassFunctionDecl) 299064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetMetaClassFunctionDecl(); 299164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 299264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // default to objc_msgSend(). 299364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendFlavor = MsgSendFunctionDecl; 299464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // May need to use objc_msgSend_stret() as well. 299564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendStretFlavor = 0; 299664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *mDecl = Exp->getMethodDecl()) { 299764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType resultType = mDecl->getResultType(); 299864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (resultType->isRecordType()) 299964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendStretFunctionDecl; 300064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (resultType->isRealFloatingType()) 300164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendFpretFunctionDecl; 300264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 300364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 300464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize a call to objc_msgSend(). 300564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> MsgExprs; 300664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (Exp->getReceiverKind()) { 300764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: { 300864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendSuperFunctionDecl; 300964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) 301064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendSuperStretFunctionDecl; 301164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(MsgSendFlavor && "MsgSendFlavor is NULL!"); 301264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 301364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = CurMethodDef->getClassInterface(); 301464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 301564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 301664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 301764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // set the receiver to self, the first argument to all methods. 301864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 301964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 302064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 302164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) DeclRefExpr(CurMethodDef->getSelfDecl(), 3022f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, 302364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 302464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, 302564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation())) 302664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ); // set the 'receiver'. 302764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 302864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 302964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 303064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 303164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 303264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassDecl->getIdentifier()->getName(), 303364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 303464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 303564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetMetaClassFunctionDecl, 303664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 303764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 303864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, 303964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 304064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (Class)objc_getClass("CurrentClass") 304164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *ArgExpr = NoTypeInfoCStyleCastExpr(Context, 304264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCClassType(), 304364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls); 304464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.clear(); 304564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(ArgExpr); 304664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Cls = SynthesizeCallToFunctionDecl(GetSuperClassFunctionDecl, 304764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], ClsExprs.size(), 304864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 304964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 305064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 305164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // To turn off a warning, type-cast to 'id' 305264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( // set 'super class', using class_getSuperclass(). 305364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, 305464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 305564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls)); 3056b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // struct __rw_objc_super 305764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType superType = getSuperStructType(); 305864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *SuperRep; 305964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 306064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 306164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSuperContructorFunctionDecl(); 306264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 306364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(SuperContructorFunctionDecl, 3064f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, superType, VK_LValue, 306564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 306664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], 306764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.size(), 306864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 306964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 307064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The code for super is a little tricky to prevent collision with 307164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the structure definition in the header. The rewriter has it's own 307264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // internal definition (__rw_objc_super) that is uses. This is why 307364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need the cast below. For example: 3074b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // (struct __rw_objc_super *)&__rw_objc_super((id)self, (id)objc_getClass("SUPER")) 307564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 307664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 307764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 307864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 307964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 308064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = NoTypeInfoCStyleCastExpr(Context, 308164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(superType), 308264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, SuperRep); 308364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 3084b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // (struct __rw_objc_super) { <exprs from above> } 308564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitListExpr *ILE = 308664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) InitListExpr(*Context, SourceLocation(), 308764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &InitExprs[0], InitExprs.size(), 308864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 308964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *superTInfo 309064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(superType); 309164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CompoundLiteralExpr(SourceLocation(), superTInfo, 309264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 309364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ILE, false); 3094b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // struct __rw_objc_super * 309564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 309664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 309764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 309864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 309964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 310064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SuperRep); 310164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 310264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 310364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 310464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: { 310564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 310664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 310764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *Class 310864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Exp->getClassReceiver()->getAs<ObjCObjectType>()->getInterface(); 310964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *clsName = Class->getIdentifier(); 311064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 311164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian clsName->getName(), 311264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 311364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 311464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 311564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 311664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 311764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 311864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(Cls); 311964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 312064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 312164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 312264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance:{ 312364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendSuperFunctionDecl; 312464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) 312564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendSuperStretFunctionDecl; 312664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(MsgSendFlavor && "MsgSendFlavor is NULL!"); 312764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = CurMethodDef->getClassInterface(); 312864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 312964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 313064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 313164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 313264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 313364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) DeclRefExpr(CurMethodDef->getSelfDecl(), 3134f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, 313564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 313664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, SourceLocation())) 313764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ); // set the 'receiver'. 313864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 313964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 314064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 314164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 314264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 314364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassDecl->getIdentifier()->getName(), 314464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, argType, 314564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation())); 314664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 314764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 314864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 314964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 315064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (Class)objc_getClass("CurrentClass") 315164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *ArgExpr = NoTypeInfoCStyleCastExpr(Context, 315264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCClassType(), 315364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls); 315464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.clear(); 315564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(ArgExpr); 315664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Cls = SynthesizeCallToFunctionDecl(GetSuperClassFunctionDecl, 315764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], ClsExprs.size(), 315864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 315964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 316064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 316164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // To turn off a warning, type-cast to 'id' 316264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 316364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // set 'super class', using class_getSuperclass(). 316464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 316564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls)); 3166b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // struct __rw_objc_super 316764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType superType = getSuperStructType(); 316864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *SuperRep; 316964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 317064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 317164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSuperContructorFunctionDecl(); 317264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 317364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(SuperContructorFunctionDecl, 3174f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, superType, VK_LValue, 317564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 317664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], 317764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.size(), 317864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, SourceLocation()); 317964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The code for super is a little tricky to prevent collision with 318064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the structure definition in the header. The rewriter has it's own 318164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // internal definition (__rw_objc_super) that is uses. This is why 318264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need the cast below. For example: 3183b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // (struct __rw_objc_super *)&__rw_objc_super((id)self, (id)objc_getClass("SUPER")) 318464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 318564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 318664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 318764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 318864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 318964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = NoTypeInfoCStyleCastExpr(Context, 319064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(superType), 319164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, SuperRep); 319264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 3193b20c46ea980b5aed7b480761ea1daf2f26c23b2dFariborz Jahanian // (struct __rw_objc_super) { <exprs from above> } 319464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitListExpr *ILE = 319564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) InitListExpr(*Context, SourceLocation(), 319664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &InitExprs[0], InitExprs.size(), 319764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 319864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *superTInfo 319964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(superType); 320064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CompoundLiteralExpr(SourceLocation(), superTInfo, 320164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_RValue, ILE, 320264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false); 320364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 320464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SuperRep); 320564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 320664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 320764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 320864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: { 320964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remove all type-casts because it may contain objc-style types; e.g. 321064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Foo<Proto> *. 321164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *recExpr = Exp->getInstanceReceiver(); 321264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(recExpr)) 321364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian recExpr = CE->getSubExpr(); 321464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK = recExpr->getType()->isObjCObjectPointerType() 321564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BitCast : recExpr->getType()->isBlockPointerType() 321664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BlockPointerToObjCPointerCast 321764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : CK_CPointerToObjCPointerCast; 321864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 321964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian recExpr = NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 322064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK, recExpr); 322164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(recExpr); 322264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 322364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 322464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 322564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 322664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a call to sel_registerName("selName"), it will be the 2nd argument. 322764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> SelExprs; 322864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 322964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 323064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getSelector().getAsString(), 323164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 323264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 323364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 323464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &SelExprs[0], SelExprs.size(), 323564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, 323664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 323764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SelExp); 323864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 323964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now push any user supplied arguments. 324064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < Exp->getNumArgs(); i++) { 324164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *userExpr = Exp->getArg(i); 324264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make all implicit casts explicit...ICE comes in handy:-) 324364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(userExpr)) { 324464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Reuse the ICE type, it is exactly what the doctor ordered. 324564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType type = ICE->getType(); 324664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(type)) 324764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian type = Context->getObjCIdType(); 324864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "type (^)(...)" to "type (*)(...)". 324964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(type); 325064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Expr *SubExpr = ICE->IgnoreParenImpCasts(); 325164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK; 325264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SubExpr->getType()->isIntegralType(*Context) && 325364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian type->isBooleanType()) { 325464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_IntegralToBoolean; 325564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (type->isObjCObjectPointerType()) { 325664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SubExpr->getType()->isBlockPointerType()) { 325764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BlockPointerToObjCPointerCast; 325864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (SubExpr->getType()->isPointerType()) { 325964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_CPointerToObjCPointerCast; 326064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 326164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 326264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 326364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 326464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 326564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 326664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 326764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = NoTypeInfoCStyleCastExpr(Context, type, CK, userExpr); 326864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 326964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make id<P...> cast into an 'id' cast. 327064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(userExpr)) { 327164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CE->getType()->isObjCQualifiedIdType()) { 327264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while ((CE = dyn_cast<CStyleCastExpr>(userExpr))) 327364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = CE->getSubExpr(); 327464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK; 327564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (userExpr->getType()->isIntegralType(*Context)) { 327664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_IntegralToPointer; 327764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (userExpr->getType()->isBlockPointerType()) { 327864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BlockPointerToObjCPointerCast; 327964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (userExpr->getType()->isPointerType()) { 328064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_CPointerToObjCPointerCast; 328164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 328264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 328364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 328464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 328564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK, userExpr); 328664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 328764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 328864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(userExpr); 328964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We've transferred the ownership to MsgExprs. For now, we *don't* null 329064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // out the argument in the original expression (since we aren't deleting 329164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the ObjCMessageExpr). See RewritePropertyOrImplicitSetter() usage for more info. 329264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //Exp->setArg(i, 0); 329364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 329464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the funky cast. 329564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *cast; 329664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 329764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType returnType; 329864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 329964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push 'id' and 'SEL', the 2 implicit arguments. 330064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendFlavor == MsgSendSuperFunctionDecl) 330164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getPointerType(getSuperStructType())); 330264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 330364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getObjCIdType()); 330464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getObjCSelType()); 330564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *OMD = Exp->getMethodDecl()) { 330664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push any user argument types. 330764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(), 330864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = OMD->param_end(); PI != E; ++PI) { 330964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = (*PI)->getType()->isObjCQualifiedIdType() 331064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? Context->getObjCIdType() 331164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : (*PI)->getType(); 331264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 331364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(t); 331464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 331564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 331664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType = Exp->getType(); 331764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian convertToUnqualifiedObjCType(returnType); 331864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(returnType); 331964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 332064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType = Context->getObjCIdType(); 332164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 332264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 332364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = MsgSendFlavor->getType(); 332464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 332564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 3326f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(MsgSendFlavor, false, msgSendType, 332764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, SourceLocation()); 332864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 332964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to cast objc_msgSend to "void *" (to workaround a GCC bandaid). 333064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we don't do this cast, we get the following bizarre warning/note: 333164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // xx.m:13: warning: function called through a non-compatible type 333264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // xx.m:13: note: if this code is reached, the program will abort 333364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, 333464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidTy), 333564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, DRE); 333664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 333764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the "normal" pointer to function cast. 333864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castType = 333964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 334064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we don't have a method decl, force a variadic cast. 334164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : true); 334264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = Context->getPointerType(castType); 334364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 334464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast); 334564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 334664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 334764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(StartLoc, EndLoc, cast); 334864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 334964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 335064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 335164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.size(), 335264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getResultType(), VK_RValue, 335364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 335464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = CE; 335564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) { 335664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We have the method which returns a struct/union. Must also generate 335764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // call to objc_msgSend_stret and hang both varieties on a conditional 335864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression which dictate which one to envoke depending on size of 335964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // method's return type. 336064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 336164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend_stret() declaration. 3362f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *STDRE = new (Context) DeclRefExpr(MsgSendStretFlavor, 3363f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, msgSendType, 336464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, SourceLocation()); 336564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to cast objc_msgSend_stret to "void *" (see above comment). 336664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, 336764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidTy), 336864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, STDRE); 336964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the "normal" pointer to function cast. 337064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 337164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : false); 337264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = Context->getPointerType(castType); 337364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 337464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast); 337564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 337664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 337764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), cast); 337864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 337964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT = msgSendType->getAs<FunctionType>(); 338064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *STCE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 338164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.size(), 338264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getResultType(), VK_RValue, 338364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 338464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 338564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build sizeof(returnType) 338664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UnaryExprOrTypeTraitExpr *sizeofExpr = 338764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) UnaryExprOrTypeTraitExpr(UETT_SizeOf, 338864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getTrivialTypeSourceInfo(returnType), 338964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getSizeType(), SourceLocation(), 339064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 339164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...)) 339264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Value of 8 is base on ppc32/x86 ABI for the most common cases. 339364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For X86 it is more complicated and some kind of target specific routine 339464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // is needed to decide what to do. 339564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 339664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 339764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IntegerLiteral *limit = IntegerLiteral::Create(*Context, 339864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::APInt(IntSize, 8), 339964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->IntTy, 340064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 340164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BinaryOperator *lessThanExpr = 340264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) BinaryOperator(sizeofExpr, limit, BO_LE, Context->IntTy, 340364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 340464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...)) 340564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConditionalOperator *CondExpr = 340664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) ConditionalOperator(lessThanExpr, 340764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), CE, 340864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), STCE, 340964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType, VK_RValue, OK_Ordinary); 341064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacingStmt = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 341164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CondExpr); 341264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 341364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 341464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return ReplacingStmt; 341564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 341664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 341764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteMessageExpr(ObjCMessageExpr *Exp) { 341864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = SynthMessageExpr(Exp, Exp->getLocStart(), 341964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getLocEnd()); 342064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 342164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the actual rewrite. 342264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, ReplacingStmt); 342364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 342464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 342564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return ReplacingStmt; 342664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 342764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 342864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// typedef struct objc_object Protocol; 342964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getProtocolType() { 343064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ProtocolTypeDecl) { 343164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *TInfo 343264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(Context->getObjCIdType()); 343364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolTypeDecl = TypedefDecl::Create(*Context, TUDecl, 343464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 343564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("Protocol"), 343664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TInfo); 343764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 343864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTypeDeclType(ProtocolTypeDecl); 343964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 344064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 344164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCProtocolExpr - Rewrite a protocol expression into 344264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// a synthesized/forward data reference (to the protocol's metadata). 344364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// The forward references (and metadata) are generated in 344464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteModernObjC::HandleTranslationUnit(). 344564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp) { 344630650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian std::string Name = "_OBJC_PROTOCOL_REFERENCE_$_" + 344730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Exp->getProtocol()->getNameAsString(); 344864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *ID = &Context->Idents.get(Name); 344964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *VD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 345064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), ID, getProtocolType(), 0, 345164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, SC_None); 3452f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(VD, false, getProtocolType(), 3453f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 345464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *DerefExpr = new (Context) UnaryOperator(DRE, UO_AddrOf, 345564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(DRE->getType()), 345664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 345764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, DerefExpr->getType(), 345864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 345964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DerefExpr); 346064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, castExpr); 346164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolExprDecls.insert(Exp->getProtocol()->getCanonicalDecl()); 346264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 346364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return castExpr; 346464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 346564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 346664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 346764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::BufferContainsPPDirectives(const char *startBuf, 346864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf) { 346964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (startBuf < endBuf) { 347064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '#') { 347164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Skip whitespace. 347264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (++startBuf; startBuf[0] == ' ' || startBuf[0] == '\t'; ++startBuf) 347364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ; 347464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strncmp(startBuf, "if", strlen("if")) || 347564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "ifdef", strlen("ifdef")) || 347664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "ifndef", strlen("ifndef")) || 347764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "define", strlen("define")) || 347864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "undef", strlen("undef")) || 347964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "else", strlen("else")) || 348064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "elif", strlen("elif")) || 348164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "endif", strlen("endif")) || 348264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "pragma", strlen("pragma")) || 348364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "include", strlen("include")) || 348464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "import", strlen("import")) || 348564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "include_next", strlen("include_next"))) 348664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 348764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 348864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 348964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 349064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 349164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 349264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 349397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian/// RewriteObjCFieldDeclType - This routine rewrites a type into the buffer. 349415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian/// It handles elaborated types, as well as enum types in the process. 349597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanianbool RewriteModernObjC::RewriteObjCFieldDeclType(QualType &Type, 349697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian std::string &Result) { 349797c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian if (Type->isArrayType()) { 349897c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian QualType ElemTy = Context->getBaseElementType(Type); 349997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return RewriteObjCFieldDeclType(ElemTy, Result); 350097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 350197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian else if (Type->isRecordType()) { 350215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian RecordDecl *RD = Type->getAs<RecordType>()->getDecl(); 350315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (RD->isCompleteDefinition()) { 350415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (RD->isStruct()) 350515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\n\tstruct "; 350615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian else if (RD->isUnion()) 350715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\n\tunion "; 350815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian else 350915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian assert(false && "class not allowed as an ivar type"); 351097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 351115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += RD->getName(); 351215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (TagsDefinedInIvarDecls.count(RD)) { 351315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian // This struct is already defined. Do not write its definition again. 351497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += " "; 351597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 351615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 351715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian TagsDefinedInIvarDecls.insert(RD); 351815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += " {\n"; 351915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian for (RecordDecl::field_iterator i = RD->field_begin(), 352097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian e = RD->field_end(); i != e; ++i) { 352115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian FieldDecl *FD = *i; 352215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian RewriteObjCFieldDecl(FD, Result); 352315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 352415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t} "; 352597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 352615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 352715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 352815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian else if (Type->isEnumeralType()) { 352915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian EnumDecl *ED = Type->getAs<EnumType>()->getDecl(); 353015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (ED->isCompleteDefinition()) { 353115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\n\tenum "; 353215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += ED->getName(); 353315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (TagsDefinedInIvarDecls.count(ED)) { 353415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian // This enum is already defined. Do not write its definition again. 353597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += " "; 353697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 353715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 353815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian TagsDefinedInIvarDecls.insert(ED); 353915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 354015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += " {\n"; 354115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian for (EnumDecl::enumerator_iterator EC = ED->enumerator_begin(), 354215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian ECEnd = ED->enumerator_end(); EC != ECEnd; ++EC) { 354315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t"; Result += EC->getName(); Result += " = "; 354415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian llvm::APSInt Val = EC->getInitVal(); 354515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += Val.toString(10); 354615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += ",\n"; 354715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 354815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t} "; 354997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 355015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 355115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 355215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 355315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t"; 355415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian convertObjCTypeToCStyleType(Type); 355597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return false; 355697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian} 355797c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 355897c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 355997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian/// RewriteObjCFieldDecl - This routine rewrites a field into the buffer. 356097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian/// It handles elaborated types, as well as enum types in the process. 356197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCFieldDecl(FieldDecl *fieldDecl, 356297c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian std::string &Result) { 356397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian QualType Type = fieldDecl->getType(); 356497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian std::string Name = fieldDecl->getNameAsString(); 356515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 356697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian bool EleboratedType = RewriteObjCFieldDeclType(Type, Result); 356797c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian if (!EleboratedType) 356897c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Type.getAsStringInternal(Name, Context->getPrintingPolicy()); 356915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += Name; 357015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (fieldDecl->isBitField()) { 357115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += " : "; Result += utostr(fieldDecl->getBitWidthValue(*Context)); 357215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 357397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian else if (EleboratedType && Type->isArrayType()) { 357497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian CanQualType CType = Context->getCanonicalType(Type); 357597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian while (isa<ArrayType>(CType)) { 357697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian if (const ConstantArrayType *CAT = Context->getAsConstantArrayType(CType)) { 357797c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += "["; 357897c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian llvm::APInt Dim = CAT->getSize(); 357997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += utostr(Dim.getZExtValue()); 358097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += "]"; 358197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 358297c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian CType = CType->getAs<ArrayType>()->getElementType(); 358397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 358497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 358597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 358615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += ";\n"; 358715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian} 358815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 358964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCInternalStruct - Rewrite one internal struct corresponding to 359064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// an objective-c class with ivars. 359164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCInternalStruct(ObjCInterfaceDecl *CDecl, 359264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 359364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CDecl && "Class missing in SynthesizeObjCInternalStruct"); 359464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CDecl->getName() != "" && 359564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "Name missing in SynthesizeObjCInternalStruct"); 359664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *RCDecl = CDecl->getSuperClass(); 3597a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian SmallVector<ObjCIvarDecl *, 8> IVars; 3598a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian for (ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 35999a2105bc790b3d0d4d5edb77c617aeccc4187c0eFariborz Jahanian IVD; IVD = IVD->getNextIvar()) 3600a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian IVars.push_back(IVD); 36019a2105bc790b3d0d4d5edb77c617aeccc4187c0eFariborz Jahanian 360264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CDecl->getLocStart(); 360364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = CDecl->getEndOfDefinitionLoc(); 3604a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 360564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 360664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 3607a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 360864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If no ivars and no root or if its root, directly or indirectly, 360964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // have no ivars (thus not synthesized) then no need to synthesize this class. 3610a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian if ((!CDecl->isThisDeclarationADefinition() || IVars.size() == 0) && 361164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (!RCDecl || !ObjCSynthesizedStructs.count(RCDecl))) { 361264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endBuf += Lexer::MeasureTokenLength(LocEnd, *SM, LangOpts); 361364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, Result); 361464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 361564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 3616a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 361764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct "; 361864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 3619a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IMPL {\n"; 3620a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 3621f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (RCDecl && ObjCSynthesizedStructs.count(RCDecl)) { 3622a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "\tstruct "; Result += RCDecl->getNameAsString(); 3623a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IMPL "; Result += RCDecl->getNameAsString(); 3624a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IVARS;\n"; 3625a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian } 362615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian TagsDefinedInIvarDecls.clear(); 362715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian for (unsigned i = 0, e = IVars.size(); i < e; i++) 362815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian RewriteObjCFieldDecl(IVars[i], Result); 36290b17b9a1b6ec0522dda5ddd8244844eb9ab1cc1cFariborz Jahanian 3630a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "};\n"; 3631a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian endBuf += Lexer::MeasureTokenLength(LocEnd, *SM, LangOpts); 3632a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, Result); 3633f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // Mark this struct as having been generated. 3634f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (!ObjCSynthesizedStructs.insert(CDecl)) 3635f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian llvm_unreachable("struct already synthesize- RewriteObjCInternalStruct"); 363664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 363764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 36387cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanianstatic void WriteInternalIvarName(ObjCInterfaceDecl *IDecl, 36397cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian ObjCIvarDecl *IvarDecl, std::string &Result) { 36407cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += "OBJC_IVAR_$_"; 36417cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += IDecl->getName(); 36427cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += "$"; 36437cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += IvarDecl->getName(); 36447cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian} 36457cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian 364672c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian/// RewriteIvarOffsetSymbols - Rewrite ivar offset symbols of those ivars which 364772c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian/// have been referenced in an ivar access expression. 364872c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanianvoid RewriteModernObjC::RewriteIvarOffsetSymbols(ObjCInterfaceDecl *CDecl, 364972c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian std::string &Result) { 365072c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian // write out ivar offset symbols which have been referenced in an ivar 365172c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian // access expression. 365272c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian llvm::SmallPtrSet<ObjCIvarDecl *, 8> Ivars = ReferencedIvars[CDecl]; 365372c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian if (Ivars.empty()) 365472c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian return; 365572c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian for (llvm::SmallPtrSet<ObjCIvarDecl *, 8>::iterator i = Ivars.begin(), 365672c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian e = Ivars.end(); i != e; i++) { 365772c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian ObjCIvarDecl *IvarDecl = (*i); 365840a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Result += "\n"; 365940a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (LangOpts.MicrosoftExt) 366040a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Result += "__declspec(allocate(\".objc_ivar$B\")) "; 3661297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "extern \"C\" "; 366240a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (LangOpts.MicrosoftExt && 366340a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian IvarDecl->getAccessControl() != ObjCIvarDecl::Private && 3664297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian IvarDecl->getAccessControl() != ObjCIvarDecl::Package) 3665297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "__declspec(dllimport) "; 3666297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian 36673f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "unsigned long "; 36687cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(CDecl, IvarDecl, Result); 36697cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += ";"; 367072c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian } 367172c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian} 367272c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 367364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 367464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Meta Data Emission 367564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 367664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 367764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 367864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteImplementations - This routine rewrites all method implementations 367964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// and emits meta-data. 368064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 368164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteImplementations() { 368264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int ClsDefCount = ClassImplementation.size(); 368364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int CatDefCount = CategoryImplementation.size(); 368464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 368564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite implemented methods 36868f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) { 36878f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian ObjCImplementationDecl *OIMP = ClassImplementation[i]; 36888f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian ObjCInterfaceDecl *CDecl = OIMP->getClassInterface(); 36898f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian if (CDecl->isImplicitInterfaceDecl()) 3690cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian assert(false && 3691cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian "Legacy implicit interface rewriting not supported in moder abi"); 36928f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian RewriteImplementationDecl(OIMP); 36938f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian } 369464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 36958c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian for (int i = 0; i < CatDefCount; i++) { 36968c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian ObjCCategoryImplDecl *CIMP = CategoryImplementation[i]; 36978c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian ObjCInterfaceDecl *CDecl = CIMP->getClassInterface(); 36988c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian if (CDecl->isImplicitInterfaceDecl()) 36998c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian assert(false && 37008c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian "Legacy implicit interface rewriting not supported in moder abi"); 37018c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian RewriteImplementationDecl(CIMP); 37028c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian } 370364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 370464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 370564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteByRefString(std::string &ResultStr, 370664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Name, 370764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD, bool def) { 370864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BlockByRefDeclNo.count(VD) && 370964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "RewriteByRefString: ByRef decl missing"); 371064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (def) 371164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "struct "; 371264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "__Block_byref_" + Name + 371364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "_" + utostr(BlockByRefDeclNo[VD]) ; 371464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 371564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 371664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool HasLocalVariableExternalStorage(ValueDecl *VD) { 371764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *Var = dyn_cast<VarDecl>(VD)) 371864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return (Var->isFunctionOrMethodVarDecl() && !Var->hasLocalStorage()); 371964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 372064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 372164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 372264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockFunc(BlockExpr *CE, int i, 372364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, 372464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag) { 372564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *AFT = CE->getFunctionType(); 372664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType RT = AFT->getResultType(); 372764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StructRef = "struct " + Tag; 372864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "static " + RT.getAsString(Context->getPrintingPolicy()) + " __" + 3729df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian funcName.str() + "_block_func_" + utostr(i); 373064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 373164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDecl *BD = CE->getBlockDecl(); 373264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 373364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<FunctionNoProtoType>(AFT)) { 373464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // No user-supplied arguments. Still need to pass in a pointer to the 373564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block (to reference imported block decl refs). 373664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef + " *__cself)"; 373764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (BD->param_empty()) { 373864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef + " *__cself)"; 373964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 374064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FT = cast<FunctionProtoType>(AFT); 374164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(FT && "SynthesizeBlockFunc: No function proto"); 374264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += '('; 374364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // first add the implicit argument. 374464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += StructRef + " *__cself, "; 374564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr; 374664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (BlockDecl::param_iterator AI = BD->param_begin(), 374764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BD->param_end(); AI != E; ++AI) { 374864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (AI != BD->param_begin()) S += ", "; 374964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParamStr = (*AI)->getNameAsString(); 375064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*AI)->getType(); 37512610f90135571be7bf983c80a1cf7bfa0844dc57Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(QT); 37522610f90135571be7bf983c80a1cf7bfa0844dc57Fariborz Jahanian QT.getAsStringInternal(ParamStr, Context->getPrintingPolicy()); 375364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ParamStr; 375464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 375564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 375664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BD->param_empty()) S += ", "; 375764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "..."; 375864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 375964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ')'; 376064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 376164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n"; 376264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 376364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create local declarations to avoid rewriting all closure decl ref exprs. 376464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // First, emit a declaration for all "by ref" decls. 376564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 376664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 376764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 376864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 376964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString; 377064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(TypeString, Name, (*I)); 377164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeString += " *"; 377264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = TypeString + Name; 377364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Name + " = __cself->" + (*I)->getNameAsString() + "; // bound by ref\n"; 377464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 377564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Next, emit a declaration for all "by copy" declarations. 377664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 377764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 377864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 377964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle nested closure invocation. For example: 378064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 378164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^myImportedClosure)(void); 378264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedClosure = ^(void) { setGlobalInt(x + y); }; 378364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 378464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^anotherClosure)(void); 378564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // anotherClosure = ^(void) { 378664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedClosure(); // import and invoke the closure 378764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // }; 378864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 378964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) { 379064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerTypeVariable(S, (*I)); 379164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " = ("; 379264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(S, (*I)->getType()); 379364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ")"; 379464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "__cself->" + (*I)->getNameAsString() + "; // bound by copy\n"; 379564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 379664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 379764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 379864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 379964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) 380064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 380164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(Name, Context->getPrintingPolicy()); 380264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Name + " = __cself->" + 380364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*I)->getNameAsString() + "; // bound by copy\n"; 380464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 380564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 380664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RewrittenStr = RewrittenBlockExprs[CE]; 380764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *cstr = RewrittenStr.c_str(); 380864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*cstr++ != '{') ; 380964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += cstr; 381064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "\n"; 381164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 381264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 381364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 381464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockHelperFuncs(BlockExpr *CE, int i, 381564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, 381664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag) { 381764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StructRef = "struct " + Tag; 381864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "static void __"; 381964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 382064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += funcName; 382164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_block_copy_" + utostr(i); 382264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef; 382364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*dst, " + StructRef; 382464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*src) {"; 382564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ValueDecl*,8>::iterator I = ImportedBlockDecls.begin(), 382664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ImportedBlockDecls.end(); I != E; ++I) { 382764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = (*I); 382864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_Block_object_assign((void*)&dst->"; 382964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 383064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", (void*)src->"; 383164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 383264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockByRefDeclsPtrSet.count((*I))) 383364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BYREF) + "/*BLOCK_FIELD_IS_BYREF*/);"; 383464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isBlockPointerType()) 383564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BLOCK) + "/*BLOCK_FIELD_IS_BLOCK*/);"; 383664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 383764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_OBJECT) + "/*BLOCK_FIELD_IS_OBJECT*/);"; 383864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 383964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "}\n"; 384064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 384164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "\nstatic void __"; 384264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += funcName; 384364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_block_dispose_" + utostr(i); 384464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef; 384564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*src) {"; 384664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ValueDecl*,8>::iterator I = ImportedBlockDecls.begin(), 384764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ImportedBlockDecls.end(); I != E; ++I) { 384864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = (*I); 384964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_Block_object_dispose((void*)src->"; 385064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 385164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockByRefDeclsPtrSet.count((*I))) 385264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BYREF) + "/*BLOCK_FIELD_IS_BYREF*/);"; 385364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isBlockPointerType()) 385464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BLOCK) + "/*BLOCK_FIELD_IS_BLOCK*/);"; 385564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 385664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_OBJECT) + "/*BLOCK_FIELD_IS_OBJECT*/);"; 385764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 385864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "}\n"; 385964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 386064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 386164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 386264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockImpl(BlockExpr *CE, std::string Tag, 386364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Desc) { 386464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "\nstruct " + Tag; 386564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Constructor = " " + Tag; 386664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 386764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n struct __block_impl impl;\n"; 386864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " struct " + Desc; 386964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "* Desc;\n"; 387064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 387164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "(void *fp, "; // Invoke function pointer. 387264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "struct " + Desc; // Descriptor pointer. 387364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " *desc"; 387464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 387564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 387664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by copy" declarations. 387764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 387864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 387964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 388064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FieldName = (*I)->getNameAsString(); 388164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ArgName = "_" + FieldName; 388264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle nested closure invocation. For example: 388364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 388464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^myImportedBlock)(void); 388564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedBlock = ^(void) { setGlobalInt(x + y); }; 388664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 388764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^anotherBlock)(void); 388864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // anotherBlock = ^(void) { 388964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedBlock(); // import and invoke the closure 389064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // }; 389164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 389264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) { 389364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "struct __block_impl *"; 389464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", void *" + ArgName; 389564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 389664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 389764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) 389864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 389964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(FieldName, Context->getPrintingPolicy()); 390064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(ArgName, Context->getPrintingPolicy()); 390164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", " + ArgName; 390264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 390364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += FieldName + ";\n"; 390464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 390564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by ref" declarations. 390664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 390764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 390864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 390964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FieldName = (*I)->getNameAsString(); 391064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ArgName = "_" + FieldName; 391164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 391264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString; 391364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(TypeString, FieldName, (*I)); 391464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeString += " *"; 391564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldName = TypeString + FieldName; 391664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgName = TypeString + ArgName; 391764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", " + ArgName; 391864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 391964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += FieldName + "; // by ref\n"; 392064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 392164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Finish writing the constructor. 392264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", int flags=0)"; 392364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize all "by copy" arguments. 392464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool firsTime = true; 392564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 392664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 392764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 392864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (firsTime) { 392964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " : "; 393064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian firsTime = false; 393164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 393264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 393364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", "; 393464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) 393564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "((struct __block_impl *)_" + Name + ")"; 393664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 393764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "(_" + Name + ")"; 393864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 393964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize all "by ref" arguments. 394064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 394164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 394264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 394364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (firsTime) { 394464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " : "; 394564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian firsTime = false; 394664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 394764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 394864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", "; 394964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "(_" + Name + "->__forwarding)"; 395064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 395164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 395264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " {\n"; 395364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl) 395464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteGlobalBlock;\n"; 395564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 395664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteStackBlock;\n"; 395764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.Flags = flags;\n impl.FuncPtr = fp;\n"; 395864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 395964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " Desc = desc;\n"; 396064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 396164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Finish writing the constructor. 396264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", int flags=0) {\n"; 396364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl) 396464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteGlobalBlock;\n"; 396564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 396664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteStackBlock;\n"; 396764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.Flags = flags;\n impl.FuncPtr = fp;\n"; 396864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " Desc = desc;\n"; 396964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 397064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " "; 397164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "}\n"; 397264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Constructor; 397364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "};\n"; 397464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 397564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 397664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 397764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockDescriptor(std::string DescTag, 397864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag, int i, 397964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName, 398064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned hasCopy) { 398164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "\nstatic struct " + DescTag; 398264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 398364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n unsigned long reserved;\n"; 398464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " unsigned long Block_size;\n"; 398564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasCopy) { 398664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " void (*copy)(struct "; 398764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*, struct "; 398864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*);\n"; 398964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 399064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " void (*dispose)(struct "; 399164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*);\n"; 399264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 399364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "} "; 399464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 399564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += DescTag + "_DATA = { 0, sizeof(struct "; 399664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag + ")"; 399764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasCopy) { 399864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", __" + FunName.str() + "_block_copy_" + utostr(i); 399964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", __" + FunName.str() + "_block_dispose_" + utostr(i); 400064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 400164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "};\n"; 400264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 400364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 400464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 400576a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian/// getFunctionSourceLocation - returns start location of a function 400676a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian/// definition. Complication arises when function has declared as 400776a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian/// extern "C" or extern "C" {...} 4008b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanianstatic SourceLocation getFunctionSourceLocation (RewriteModernObjC &R, 4009b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian FunctionDecl *FD) { 4010b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian if (!FD->isExternC() || FD->isMain()) { 4011b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian if (FD->getStorageClassAsWritten() != SC_None) 4012b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian R.RewriteBlockLiteralFunctionDecl(FD); 401376a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian return FD->getTypeSpecStartLoc(); 4014b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian } 401576a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian const DeclContext *DC = FD->getDeclContext(); 401676a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian if (const LinkageSpecDecl *LSD = dyn_cast<LinkageSpecDecl>(DC)) { 401776a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian SourceLocation BodyRBrace = LSD->getRBraceLoc(); 401876a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian // if it is extern "C" {...}, return function decl's own location. 401976a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian if (BodyRBrace.isValid()) 402076a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian return FD->getTypeSpecStartLoc(); 402176a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian return LSD->getExternLoc(); 402276a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian } 402376a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian return FD->getTypeSpecStartLoc(); 402476a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian} 402576a98bef86becf111af6a77a009aab7790a3c2e8Fariborz Jahanian 402664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthesizeBlockLiterals(SourceLocation FunLocStart, 402764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName) { 402864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool RewriteSC = (GlobalVarDecl && 402964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !Blocks.empty() && 403064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl->getStorageClass() == SC_Static && 403164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl->getType().getCVRQualifiers()); 403264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RewriteSC) { 403364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SC(" void __"); 403464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += GlobalVarDecl->getNameAsString(); 403564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "() {}"; 403664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, SC); 403764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 403864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 403964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert closures that were part of the function. 404064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, count=0; i < Blocks.size(); i++) { 404164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CollectBlockDeclRefInfo(Blocks[i]); 404264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to copy-in the inner copied-in variables not actually used in this 404364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block. 404464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int j = 0; j < InnerDeclRefsCount[i]; j++) { 4045f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *Exp = InnerDeclRefs[count++]; 404664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = Exp->getDecl(); 404764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 4048f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!VD->hasAttr<BlocksAttr>()) { 4049f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!BlockByCopyDeclsPtrSet.count(VD)) { 4050f4b88a45902af1802a1cb42ba48b1c474474f228John McCall BlockByCopyDeclsPtrSet.insert(VD); 4051f4b88a45902af1802a1cb42ba48b1c474474f228John McCall BlockByCopyDecls.push_back(VD); 4052f4b88a45902af1802a1cb42ba48b1c474474f228John McCall } 4053f4b88a45902af1802a1cb42ba48b1c474474f228John McCall continue; 405464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 4055f4b88a45902af1802a1cb42ba48b1c474474f228John McCall 4056f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!BlockByRefDeclsPtrSet.count(VD)) { 405764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(VD); 405864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(VD); 405964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 4060f4b88a45902af1802a1cb42ba48b1c474474f228John McCall 406164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // imported objects in the inner blocks not used in the outer 406264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // blocks must be copied/disposed in the outer block as well. 4063f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (VD->getType()->isObjCObjectPointerType() || 406464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD->getType()->isBlockPointerType()) 406564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(VD); 406664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 406764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 406864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag = "__" + FunName.str() + "_block_impl_" + utostr(i); 406964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string DescTag = "__" + FunName.str() + "_block_desc_" + utostr(i); 407064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 407164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string CI = SynthesizeBlockImpl(Blocks[i], ImplTag, DescTag); 407264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 407364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, CI); 407464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 407564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string CF = SynthesizeBlockFunc(Blocks[i], i, FunName, ImplTag); 407664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 407764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, CF); 407864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 407964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImportedBlockDecls.size()) { 408064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string HF = SynthesizeBlockHelperFuncs(Blocks[i], i, FunName, ImplTag); 408164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, HF); 408264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 408364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string BD = SynthesizeBlockDescriptor(DescTag, ImplTag, i, FunName, 408464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.size() > 0); 408564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, BD); 408664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 408764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.clear(); 408864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.clear(); 408964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.clear(); 409064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.clear(); 409164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.clear(); 409264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.clear(); 409364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 409464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RewriteSC) { 409564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Must insert any 'const/volatile/static here. Since it has been 409664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // removed as result of rewriting of block literals. 409764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SC; 409864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getStorageClass() == SC_Static) 409964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC = "static "; 410064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isConstQualified()) 410164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "const "; 410264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isVolatileQualified()) 410364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "volatile "; 410464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isRestrictQualified()) 410564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "restrict "; 410664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, SC); 410764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 4108df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian if (GlobalConstructionExp) { 4109df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian // extra fancy dance for global literal expression. 4110df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 4111df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian // Always the latest block expression on the block stack. 4112df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian std::string Tag = "__"; 4113df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Tag += FunName; 4114df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Tag += "_block_impl_"; 4115df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Tag += utostr(Blocks.size()-1); 4116df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian std::string globalBuf = "static "; 4117df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian globalBuf += Tag; globalBuf += " "; 4118df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian std::string SStr; 411964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 4120df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian llvm::raw_string_ostream constructorExprBuf(SStr); 4121df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian GlobalConstructionExp->printPretty(constructorExprBuf, *Context, 0, 4122df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian PrintingPolicy(LangOpts)); 4123df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian globalBuf += constructorExprBuf.str(); 4124df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian globalBuf += ";\n"; 4125df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian InsertText(FunLocStart, globalBuf); 4126df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian GlobalConstructionExp = 0; 4127df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian } 4128df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 412964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Blocks.clear(); 413064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefsCount.clear(); 413164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.clear(); 413264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewrittenBlockExprs.clear(); 413364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 413464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 413564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InsertBlockLiteralsWithinFunction(FunctionDecl *FD) { 4136b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian SourceLocation FunLocStart = getFunctionSourceLocation(*this, FD); 413764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FuncName = FD->getName(); 413864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 413964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(FunLocStart, FuncName); 414064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 414164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 414264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void BuildUniqueMethodName(std::string &Name, 414364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MD) { 414464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *IFace = MD->getClassInterface(); 414564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = IFace->getName(); 414664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name += "__" + MD->getSelector().getAsString(); 414764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Convert colons to underscores. 414864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string::size_type loc = 0; 414964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while ((loc = Name.find(":", loc)) != std::string::npos) 415064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name.replace(loc, 1, "_"); 415164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 415264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 415364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InsertBlockLiteralsWithinMethod(ObjCMethodDecl *MD) { 415464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //fprintf(stderr,"In InsertBlockLiteralsWitinMethod\n"); 415564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //SourceLocation FunLocStart = MD->getLocStart(); 415664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart = MD->getLocStart(); 415764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FuncName; 415864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BuildUniqueMethodName(FuncName, MD); 415964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(FunLocStart, FuncName); 416064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 416164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 416264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetBlockDeclRefExprs(Stmt *S) { 416364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 416464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 416564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *CBE = dyn_cast<BlockExpr>(*CI)) 416664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(CBE->getBody()); 416764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 416864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(*CI); 416964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 417064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 41710e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) { 41720e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian if (DRE->refersToEnclosingLocal()) { 41730e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian // FIXME: Handle enums. 41740e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian if (!isa<FunctionDecl>(DRE->getDecl())) 41750e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian BlockDeclRefs.push_back(DRE); 41760e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian if (HasLocalVariableExternalStorage(DRE->getDecl())) 41770e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian BlockDeclRefs.push_back(DRE); 417864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 41790e976818ea1c7624b7dc250e7e2f9d251845874fFariborz Jahanian } 418064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 418164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 418264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 418364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 418464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetInnerBlockDeclRefExprs(Stmt *S, 4185f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs, 418664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> &InnerContexts) { 418764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 418864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 418964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *CBE = dyn_cast<BlockExpr>(*CI)) { 419064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts.insert(cast<DeclContext>(CBE->getBlockDecl())); 419164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(CBE->getBody(), 419264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, 419364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts); 419464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 419564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 419664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(*CI, 419764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, 419864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts); 419964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 420064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 420164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 4202f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) { 4203f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (DRE->refersToEnclosingLocal()) { 4204f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!isa<FunctionDecl>(DRE->getDecl()) && 4205f4b88a45902af1802a1cb42ba48b1c474474f228John McCall !InnerContexts.count(DRE->getDecl()->getDeclContext())) 4206f4b88a45902af1802a1cb42ba48b1c474474f228John McCall InnerBlockDeclRefs.push_back(DRE); 4207f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (VarDecl *Var = dyn_cast<VarDecl>(DRE->getDecl())) 4208f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (Var->isFunctionOrMethodVarDecl()) 4209f4b88a45902af1802a1cb42ba48b1c474474f228John McCall ImportedLocalExternalDecls.insert(Var); 4210f4b88a45902af1802a1cb42ba48b1c474474f228John McCall } 421164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 421264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 421364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 421464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 421564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 4216164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian/// convertObjCTypeToCStyleType - This routine converts such objc types 4217164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian/// as qualified objects, and blocks to their closest c/c++ types that 4218164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian/// it can. It returns true if input type was modified. 4219164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanianbool RewriteModernObjC::convertObjCTypeToCStyleType(QualType &T) { 4220164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian QualType oldT = T; 4221164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian convertBlockPointerToFunctionPointer(T); 4222164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (T->isFunctionPointerType()) { 4223164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian QualType PointeeTy; 4224164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (const PointerType* PT = T->getAs<PointerType>()) { 4225164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian PointeeTy = PT->getPointeeType(); 4226164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (const FunctionType *FT = PointeeTy->getAs<FunctionType>()) { 4227164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian T = convertFunctionTypeOfBlocks(FT); 4228164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian T = Context->getPointerType(T); 4229164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian } 4230164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian } 4231164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian } 4232164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian 4233164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian convertToUnqualifiedObjCType(T); 4234164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian return T != oldT; 4235164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian} 4236164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian 423764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// convertFunctionTypeOfBlocks - This routine converts a function type 423864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// whose result type may be a block pointer or whose argument type(s) 423964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// might be block pointers to an equivalent function type replacing 424064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// all block pointers to function pointers. 424164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::convertFunctionTypeOfBlocks(const FunctionType *FT) { 424264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP = dyn_cast<FunctionProtoType>(FT); 424364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FTP will be null for closures that don't take arguments. 424464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate a funky cast. 424564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 424664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Res = FT->getResultType(); 4247164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian bool modified = convertObjCTypeToCStyleType(Res); 424864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 424964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 425064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 425164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I && (I != E); ++I) { 425264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = *I; 425364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 4254164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (convertObjCTypeToCStyleType(t)) 4255164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian modified = true; 425664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 425764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 425864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 425964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FuncType; 4260164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (modified) 426164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncType = getSimpleFunctionType(Res, &ArgTypes[0], ArgTypes.size()); 426264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else FuncType = QualType(FT, 0); 426364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return FuncType; 426464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 426564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 426664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthesizeBlockCall(CallExpr *Exp, const Expr *BlockExp) { 426764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Navigate to relevant type information. 426864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *CPT = 0; 426964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 427064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(BlockExp)) { 427164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = DRE->getType()->getAs<BlockPointerType>(); 427264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const MemberExpr *MExpr = dyn_cast<MemberExpr>(BlockExp)) { 427364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = MExpr->getType()->getAs<BlockPointerType>(); 427464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 427564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ParenExpr *PRE = dyn_cast<ParenExpr>(BlockExp)) { 427664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return SynthesizeBlockCall(Exp, PRE->getSubExpr()); 427764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 427864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ImplicitCastExpr *IEXPR = dyn_cast<ImplicitCastExpr>(BlockExp)) 427964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = IEXPR->getType()->getAs<BlockPointerType>(); 428064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ConditionalOperator *CEXPR = 428164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ConditionalOperator>(BlockExp)) { 428264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *LHSExp = CEXPR->getLHS(); 428364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *LHSStmt = SynthesizeBlockCall(Exp, LHSExp); 428464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *RHSExp = CEXPR->getRHS(); 428564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RHSStmt = SynthesizeBlockCall(Exp, RHSExp); 428664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *CONDExp = CEXPR->getCond(); 428764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConditionalOperator *CondExpr = 428864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) ConditionalOperator(CONDExp, 428964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), cast<Expr>(LHSStmt), 429064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), cast<Expr>(RHSStmt), 429164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getType(), VK_RValue, OK_Ordinary); 429264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CondExpr; 429364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const ObjCIvarRefExpr *IRE = dyn_cast<ObjCIvarRefExpr>(BlockExp)) { 429464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = IRE->getType()->getAs<BlockPointerType>(); 429564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const PseudoObjectExpr *POE 429664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = dyn_cast<PseudoObjectExpr>(BlockExp)) { 429764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = POE->getType()->castAs<BlockPointerType>(); 429864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 429964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(1 && "RewriteBlockClass: Bad type"); 430064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 430164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CPT && "RewriteBlockClass: Bad type"); 430264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = CPT->getPointeeType()->getAs<FunctionType>(); 430364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(FT && "RewriteBlockClass: Bad type"); 430464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP = dyn_cast<FunctionProtoType>(FT); 430564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FTP will be null for closures that don't take arguments. 430664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 430764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 430864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 430964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__block_impl")); 431064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PtrBlock = Context->getPointerType(Context->getTagDeclType(RD)); 431164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 431264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate a funky cast. 431364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 431464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 431564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push the block argument type. 431664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(PtrBlock); 431764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 431864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 431964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I && (I != E); ++I) { 432064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = *I; 432164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 432264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!convertBlockPointerToFunctionPointer(t)) 432364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian convertToUnqualifiedObjCType(t); 432464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 432564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 432664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 432764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the pointer to function cast. 432864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PtrToFuncCastType 432964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = getSimpleFunctionType(Exp->getType(), &ArgTypes[0], ArgTypes.size()); 433064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 433164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PtrToFuncCastType = Context->getPointerType(PtrToFuncCastType); 433264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 433364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *BlkCast = NoTypeInfoCStyleCastExpr(Context, PtrBlock, 433464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 433564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const_cast<Expr*>(BlockExp)); 433664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 433764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 433864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkCast); 433964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //PE->dump(); 434064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 434164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = FieldDecl::Create(*Context, 0, SourceLocation(), 434264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 434364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("FuncPtr"), 434464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 434564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 434664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 434764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MemberExpr *ME = new (Context) MemberExpr(PE, true, FD, SourceLocation(), 434864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType(), VK_LValue, 434964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary); 435064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 435164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 435264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *FunkCast = NoTypeInfoCStyleCastExpr(Context, PtrToFuncCastType, 435364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, ME); 435464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), FunkCast); 435564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 435664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> BlkExprs; 435764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add the implicit argument. 435864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.push_back(BlkCast); 435964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add the user arguments. 436064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (CallExpr::arg_iterator I = Exp->arg_begin(), 436164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = Exp->arg_end(); I != E; ++I) { 436264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.push_back(*I); 436364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 436464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &BlkExprs[0], 436564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.size(), 436664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getType(), VK_RValue, 436764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 436864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CE; 436964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 437064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 437164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// We need to return the rewritten expression to handle cases where the 4372f4b88a45902af1802a1cb42ba48b1c474474f228John McCall// DeclRefExpr is embedded in another expression being rewritten. 437364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// For example: 437464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 437564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// int main() { 437664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// __block Foo *f; 437764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// __block int i; 437864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 437964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// void (^myblock)() = ^() { 4380f4b88a45902af1802a1cb42ba48b1c474474f228John McCall// [f test]; // f is a DeclRefExpr embedded in a message (which is being rewritten). 438164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// i = 77; 438264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// }; 438364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//} 4384f4b88a45902af1802a1cb42ba48b1c474474f228John McCallStmt *RewriteModernObjC::RewriteBlockDeclRefExpr(DeclRefExpr *DeclRefExp) { 438564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the byref variable into BYREFVAR->__forwarding->BYREFVAR 438664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for each DeclRefExp where BYREFVAR is name of the variable. 4387f4b88a45902af1802a1cb42ba48b1c474474f228John McCall ValueDecl *VD = DeclRefExp->getDecl(); 4388f4b88a45902af1802a1cb42ba48b1c474474f228John McCall bool isArrow = DeclRefExp->refersToEnclosingLocal(); 438964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 439064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = FieldDecl::Create(*Context, 0, SourceLocation(), 439164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 439264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__forwarding"), 439364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 439464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 439564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 439664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MemberExpr *ME = new (Context) MemberExpr(DeclRefExp, isArrow, 439764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD, SourceLocation(), 439864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType(), VK_LValue, 439964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary); 440064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 440164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef Name = VD->getName(); 440264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = FieldDecl::Create(*Context, 0, SourceLocation(), SourceLocation(), 440364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get(Name), 440464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 440564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 440664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 440764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ME = new (Context) MemberExpr(ME, true, FD, SourceLocation(), 440864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExp->getType(), VK_LValue, OK_Ordinary); 440964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need parens to enforce precedence. 441364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(DeclRefExp->getExprLoc(), 441464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExp->getExprLoc(), 441564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ME); 441664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(DeclRefExp, PE); 441764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return PE; 441864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 441964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 442064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Rewrites the imported local variable V with external storage 442164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// (static, extern, etc.) as *V 442264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 442364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteLocalVariableExternalStorage(DeclRefExpr *DRE) { 442464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = DRE->getDecl(); 442564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *Var = dyn_cast<VarDecl>(VD)) 442664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ImportedLocalExternalDecls.count(Var)) 442764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return DRE; 442864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Exp = new (Context) UnaryOperator(DRE, UO_Deref, DRE->getType(), 442964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, OK_Ordinary, 443064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DRE->getLocation()); 443164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need parens to enforce precedence. 443264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 443364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp); 443464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(DRE, PE); 443564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return PE; 443664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 443764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 443864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteCastExpr(CStyleCastExpr *CE) { 443964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CE->getLParenLoc(); 444064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = CE->getRParenLoc(); 444164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 444264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to avoid trying to rewrite synthesized casts. 444364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 444464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 444564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to avoid trying to rewrite casts contained in macros. 444664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewriter::isRewritable(LocStart) || !Rewriter::isRewritable(LocEnd)) 444764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 444864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 444964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 445064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 445164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = CE->getType(); 445264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Type* TypePtr = QT->getAs<Type>(); 445364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<TypeOfExprType>(TypePtr)) { 445464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const TypeOfExprType *TypeOfExprTypePtr = cast<TypeOfExprType>(TypePtr); 445564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = TypeOfExprTypePtr->getUnderlyingExpr()->getType(); 445664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeAsString = "("; 445764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(TypeAsString, QT); 445864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeAsString += ")"; 445964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf+1, TypeAsString); 446064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 446164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 446264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // advance the location to startArgList. 446364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = startBuf; 446464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 446564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr++ && (argPtr < endBuf)) { 446664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 446764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 446864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*'. 446964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LocStart = LocStart.getLocWithOffset(argPtr-startBuf); 447064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 1, "*"); 447164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 447264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 447364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 447464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 447564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 447664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 4477f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanianvoid RewriteModernObjC::RewriteImplicitCastObjCExpr(CastExpr *IC) { 4478f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian CastKind CastKind = IC->getCastKind(); 447943aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian if (CastKind != CK_BlockPointerToObjCPointerCast && 448043aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian CastKind != CK_AnyPointerToBlockPointerCast) 448143aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian return; 4482f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian 448343aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian QualType QT = IC->getType(); 448443aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian (void)convertBlockPointerToFunctionPointer(QT); 448543aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian std::string TypeString(QT.getAsString(Context->getPrintingPolicy())); 448643aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian std::string Str = "("; 448743aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian Str += TypeString; 448843aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian Str += ")"; 448943aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian InsertText(IC->getSubExpr()->getLocStart(), &Str[0], Str.size()); 449043aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian 4491f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian return; 4492f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian} 4493f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian 449464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerFunctionArgs(FunctionDecl *FD) { 449564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = FD->getLocation(); 449664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned parenCount = 0; 449764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 449864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We have 1 or more arguments that have closure pointers. 449964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 450064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startArgList = strchr(startBuf, '('); 450164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 450264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startArgList == '(') && "Rewriter fuzzy parser confused"); 450364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 450464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount++; 450564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // advance the location to startArgList. 450664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = DeclLoc.getLocWithOffset(startArgList-startBuf); 450764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((DeclLoc.isValid()) && "Invalid DeclLoc"); 450864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 450964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = startArgList; 451064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 451164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr++ && parenCount) { 451264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 451364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 451464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*'. 451564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = DeclLoc.getLocWithOffset(argPtr-startArgList); 451664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, 1, "*"); 451764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 451864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': 451964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount++; 452064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 452164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': 452264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount--; 452364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 452464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 452564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 452664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 452764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 452864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 452964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::PointerTypeTakesAnyBlockArguments(QualType QT) { 453064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP; 453164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = QT->getAs<PointerType>(); 453264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT) { 453364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = PT->getPointeeType()->getAs<FunctionProtoType>(); 453464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 453564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = QT->getAs<BlockPointerType>(); 453664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BPT && "BlockPointerTypeTakeAnyBlockArguments(): not a block pointer type"); 453764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = BPT->getPointeeType()->getAs<FunctionProtoType>(); 453864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 453964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 454064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 454164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I != E; ++I) 454264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(*I)) 454364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 454464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 454564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 454664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 454764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 454864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::PointerTypeTakesAnyObjCQualifiedType(QualType QT) { 454964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP; 455064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = QT->getAs<PointerType>(); 455164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT) { 455264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = PT->getPointeeType()->getAs<FunctionProtoType>(); 455364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 455464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = QT->getAs<BlockPointerType>(); 455564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BPT && "BlockPointerTypeTakeAnyBlockArguments(): not a block pointer type"); 455664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = BPT->getPointeeType()->getAs<FunctionProtoType>(); 455764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 455864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 455964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 456064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I != E; ++I) { 456164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*I)->isObjCQualifiedIdType()) 456264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 456364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*I)->isObjCObjectPointerType() && 456464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*I)->getPointeeType()->isObjCQualifiedInterfaceType()) 456564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 456664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 456764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 456864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 456964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 457064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 457164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 457264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetExtentOfArgList(const char *Name, const char *&LParen, 457364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&RParen) { 457464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = strchr(Name, '('); 457564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*argPtr == '(') && "Rewriter fuzzy parser confused"); 457664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 457764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LParen = argPtr; // output the start. 457864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; // skip past the left paren. 457964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned parenCount = 1; 458064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 458164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr && parenCount) { 458264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 458364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': parenCount++; break; 458464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': parenCount--; break; 458564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: break; 458664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 458764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (parenCount) argPtr++; 458864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 458964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*argPtr == ')') && "Rewriter fuzzy parser confused"); 459064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RParen = argPtr; // output the end 459164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 459264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 459364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerDecl(NamedDecl *ND) { 459464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) { 459564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerFunctionArgs(FD); 459664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 459764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 459864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle Variables and Typedefs. 459964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getLocation(); 460064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType DeclT; 460164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(ND)) 460264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = VD->getType(); 460364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TypedefNameDecl *TDD = dyn_cast<TypedefNameDecl>(ND)) 460464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = TDD->getUnderlyingType(); 460564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FieldDecl *FD = dyn_cast<FieldDecl>(ND)) 460664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = FD->getType(); 460764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 460864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("RewriteBlockPointerDecl(): Decl type not yet handled"); 460964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 461064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 461164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = startBuf; 461264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scan backward (from the decl location) for the end of the previous decl. 461364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != '^' && *startBuf != ';' && startBuf != MainFileStart) 461464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf--; 461564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Start = DeclLoc.getLocWithOffset(startBuf-endBuf); 461664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 461764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned OrigLength=0; 461864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // *startBuf != '^' if we are dealing with a pointer to function that 461964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // may take block argument types (which will be handled below). 462064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '^') { 462164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*', computing a negative offset. 462264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = '*'; 462364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 462464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 462564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 462664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != ')') { 462764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *startBuf; 462864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 462964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 463064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 463164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ')'; 463264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 463364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 463464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PointerTypeTakesAnyBlockArguments(DeclT) || 463564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointerTypeTakesAnyObjCQualifiedType(DeclT)) { 463664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*' for arguments. 463764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace id<P> with id/*<>*/ 463864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = ND->getLocation(); 463964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = SM->getCharacterData(DeclLoc); 464064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argListBegin, *argListEnd; 464164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetExtentOfArgList(startBuf, argListBegin, argListEnd); 464264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (argListBegin < argListEnd) { 464364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*argListBegin == '^') 464464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += '*'; 464564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*argListBegin == '<') { 464664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "/*"; 464764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin++; 464864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++;; 464964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argListBegin != '>') { 465064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin++; 465164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 465264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 465364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin; 465464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "*/"; 465564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 465664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 465764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin; 465864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argListBegin++; 465964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 466064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 466164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ')'; 466264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 466364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 466464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(Start, OrigLength, buf); 466564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 466664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 466764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 466864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 466964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 467064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SynthesizeByrefCopyDestroyHelper - This routine synthesizes: 467164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void __Block_byref_id_object_copy(struct Block_byref_id_object *dst, 467264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct Block_byref_id_object *src) { 467364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_assign (&_dest->object, _src->object, 467464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_OBJECT 467564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // object 467664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_assign(&_dest->object, _src->object, 467764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_BLOCK 467864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // block 467964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 468064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// And: 468164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void __Block_byref_id_object_dispose(struct Block_byref_id_object *_src) { 468264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_dispose(_src->object, 468364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_OBJECT 468464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // object 468564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_dispose(_src->object, 468664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_BLOCK 468764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // block 468864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 468964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 469064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeByrefCopyDestroyHelper(VarDecl *VD, 469164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag) { 469264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S; 469364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CopyDestroyCache.count(flag)) 469464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 469564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CopyDestroyCache.insert(flag); 469664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S = "static void __Block_byref_id_object_copy_"; 469764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 469864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(void *dst, void *src) {\n"; 469964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 470064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // offset into the object pointer is computed as: 470164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void * + void* + int + int + void* + void * 470264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 470364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 470464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned VoidPtrSize = 470564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->VoidPtrTy)); 470664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 470764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned offset = (VoidPtrSize*4 + IntSize + IntSize)/Context->getCharWidth(); 470864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " _Block_object_assign((char*)dst + "; 470964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 471064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", *(void * *) ((char*)src + "; 471164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 471264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "), "; 471364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 471464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ");\n}\n"; 471564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 471664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "static void __Block_byref_id_object_dispose_"; 471764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 471864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(void *src) {\n"; 471964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " _Block_object_dispose(*(void * *) ((char*)src + "; 472064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 472164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "), "; 472264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 472364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ");\n}\n"; 472464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 472564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 472664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 472764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteByRefVar - For each __block typex ND variable this routine transforms 472864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// the declaration into: 472964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND { 473064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__isa; // NULL for everything except __weak pointers 473164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND *__forwarding; 473264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// int32_t __flags; 473364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// int32_t __size; 473464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__Block_byref_id_object_copy; // If variable is __block ObjC object 473564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__Block_byref_id_object_dispose; // If variable is __block ObjC object 473664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// typex ND; 473764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// }; 473864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 473964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// It then replaces declaration of ND variable with: 474064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND ND = {__isa=0B, __forwarding=&ND, __flags=some_flag, 474164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __size=sizeof(struct __Block_byref_ND), 474264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ND=initializer-if-any}; 474364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 474464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 474564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteByRefVar(VarDecl *ND) { 474664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag = 0; 474764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int isa = 0; 474864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getTypeSpecStartLoc(); 474964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclLoc.isInvalid()) 475064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If type location is missing, it is because of missing type (a warning). 475164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Use variable's location which is good for this case. 475264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = ND->getLocation(); 475364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 475464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation X = ND->getLocEnd(); 475564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian X = SM->getExpansionLoc(X); 475664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(X); 475764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 475864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ByrefType; 475964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND, true); 476064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " {\n"; 476164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void *__isa;\n"; 476264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 476364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " *__forwarding;\n"; 476464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " int __flags;\n"; 476564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " int __size;\n"; 476664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add void *__Block_byref_id_object_copy; 476764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void *__Block_byref_id_object_dispose; if needed. 476864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Ty = ND->getType(); 476964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool HasCopyAndDispose = Context->BlockRequiresCopying(Ty); 477064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 477164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void (*__Block_byref_id_object_copy)(void*, void*);\n"; 477264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void (*__Block_byref_id_object_dispose)(void*);\n"; 477364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 477464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 477564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType T = Ty; 477664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(T); 477764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T.getAsStringInternal(Name, Context->getPrintingPolicy()); 477864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 477964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name + ";\n"; 478064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "};\n"; 478164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert this type in global scope. It is needed by helper function. 478264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart; 478364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDef) 4784b75f8de7ddc4f9365dad90ef21a463e94dcdcbdaFariborz Jahanian FunLocStart = getFunctionSourceLocation(*this, CurFunctionDef); 478564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 478664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CurMethodDef && "RewriteByRefVar - CurMethodDef is null"); 478764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunLocStart = CurMethodDef->getLocStart(); 478864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 478964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, ByrefType); 479064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty.isObjCGCWeak()) { 479164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_WEAK; 479264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa = 1; 479364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 479464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 479564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 479664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag = BLOCK_BYREF_CALLER; 479764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Ty = ND->getType(); 479864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Handle __weak variable (BLOCK_FIELD_IS_WEAK) as well. 479964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty->isBlockPointerType()) 480064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_BLOCK; 480164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 480264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_OBJECT; 480364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string HF = SynthesizeByrefCopyDestroyHelper(ND, flag); 480464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!HF.empty()) 480564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, HF); 480664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 480764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 480864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct __Block_byref_ND ND = 480964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // {0, &ND, some_flag, __size=sizeof(struct __Block_byref_ND), 481064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // initializer-if-any}; 481164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool hasInit = (ND->getInit() != 0); 4812104dbf982582f16888fb15f9342fce9748af48c5Fariborz Jahanian // FIXME. rewriter does not support __block c++ objects which 4813104dbf982582f16888fb15f9342fce9748af48c5Fariborz Jahanian // require construction. 4814104dbf982582f16888fb15f9342fce9748af48c5Fariborz Jahanian if (hasInit && dyn_cast<CXXConstructExpr>(ND->getInit())) 4815104dbf982582f16888fb15f9342fce9748af48c5Fariborz Jahanian hasInit = false; 481664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned flags = 0; 481764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) 481864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flags |= BLOCK_HAS_COPY_DISPOSE; 481964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = ND->getNameAsString(); 482064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType.clear(); 482164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 482264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ForwardingCastType("("); 482364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ForwardingCastType += ByrefType + " *)"; 482464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!hasInit) { 482564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name + " = {(void*)"; 482664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(isa); 482764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "," + ForwardingCastType + "&" + Name + ", "; 482864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flags); 482964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 483064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "sizeof("; 483164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 483264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ")"; 483364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 483464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_copy_"; 483564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 483664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_dispose_"; 483764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 483864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 483964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "};\n"; 484064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned nameSize = Name.size(); 484164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for block or function pointer declaration. Name is aleady 484264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // part of the declaration. 484364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty->isBlockPointerType() || Ty->isFunctionPointerType()) 484464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian nameSize = 1; 484564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf+nameSize, ByrefType); 484664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 484764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 484864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc; 484964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *E = ND->getInit(); 485064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) 485164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = ECE->getLParenLoc(); 485264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 485364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = E->getLocStart(); 485464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = SM->getExpansionLoc(startLoc); 485564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endBuf = SM->getCharacterData(startLoc); 485664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name; 485764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " = {(void*)"; 485864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(isa); 485964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "," + ForwardingCastType + "&" + Name + ", "; 486064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flags); 486164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 486264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "sizeof("; 486364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 486464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "), "; 486564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 486664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "__Block_byref_id_object_copy_"; 486764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 486864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_dispose_"; 486964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 487064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 487164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 487264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf, ByrefType); 487364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 487464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Complete the newly synthesized compound expression by inserting a right 487564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // curly brace before the end of the declaration. 487664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: This approach avoids rewriting the initializer expression. It 487764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // also assumes there is only one declarator. For example, the following 487864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // isn't currently supported by this routine (in general): 487964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 488064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // double __block BYREFVAR = 1.34, BYREFVAR2 = 1.37; 488164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 488264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startInitializerBuf = SM->getCharacterData(startLoc); 488364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startInitializerBuf, ';'); 488464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "RewriteByRefVar: can't find ';'"); 488564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation semiLoc = 488664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc.getLocWithOffset(semiBuf-startInitializerBuf); 488764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 488864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(semiLoc, "}"); 488964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 489064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 489164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 489264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 489364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::CollectBlockDeclRefInfo(BlockExpr *Exp) { 489464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add initializers for any closure decl refs. 489564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(Exp->getBody()); 489664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 489764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique all "by copy" declarations. 489864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 4899f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!BlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>()) { 490064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BlockByCopyDeclsPtrSet.count(BlockDeclRefs[i]->getDecl())) { 490164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.insert(BlockDeclRefs[i]->getDecl()); 490264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.push_back(BlockDeclRefs[i]->getDecl()); 490364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique all "by ref" declarations. 490664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 4907f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (BlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>()) { 490864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BlockByRefDeclsPtrSet.count(BlockDeclRefs[i]->getDecl())) { 490964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(BlockDeclRefs[i]->getDecl()); 491064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(BlockDeclRefs[i]->getDecl()); 491164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 491264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 491364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find any imported blocks...they will need special attention. 491464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 4915f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (BlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>() || 491664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs[i]->getType()->isObjCObjectPointerType() || 491764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs[i]->getType()->isBlockPointerType()) 491864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(BlockDeclRefs[i]->getDecl()); 491964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 492064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 492164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 492264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianFunctionDecl *RewriteModernObjC::SynthBlockInitFunctionDecl(StringRef name) { 492364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *ID = &Context->Idents.get(name); 492464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FType = Context->getFunctionNoProtoType(Context->VoidPtrTy); 492564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return FunctionDecl::Create(*Context, TUDecl, SourceLocation(), 492664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), ID, FType, 0, SC_Extern, 492764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false, false); 492864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 492964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 493064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthBlockInitExpr(BlockExpr *Exp, 4931f4b88a45902af1802a1cb42ba48b1c474474f228John McCall const SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs) { 4932d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian 493364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockDecl *block = Exp->getBlockDecl(); 4934d13c2c291f9fec2ed0e12a65f1638710ca979bb0Fariborz Jahanian 493564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Blocks.push_back(Exp); 493664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 493764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CollectBlockDeclRefInfo(Exp); 493864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 493964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add inner imported variables now used in current block. 494064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int countOfInnerDecls = 0; 494164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!InnerBlockDeclRefs.empty()) { 494264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < InnerBlockDeclRefs.size(); i++) { 4943f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *Exp = InnerBlockDeclRefs[i]; 494464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = Exp->getDecl(); 4945f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!VD->hasAttr<BlocksAttr>() && !BlockByCopyDeclsPtrSet.count(VD)) { 494664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We need to save the copied-in variables in nested 494764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // blocks because it is needed at the end for some of the API generations. 494864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See SynthesizeBlockLiterals routine. 494964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.push_back(Exp); countOfInnerDecls++; 495064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 495164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.insert(VD); 495264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.push_back(VD); 495364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 4954f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (VD->hasAttr<BlocksAttr>() && !BlockByRefDeclsPtrSet.count(VD)) { 495564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.push_back(Exp); countOfInnerDecls++; 495664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 495764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(VD); 495864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(VD); 495964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 496064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 496164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find any imported blocks...they will need special attention. 496264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < InnerBlockDeclRefs.size(); i++) 4963f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (InnerBlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>() || 496464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs[i]->getType()->isObjCObjectPointerType() || 496564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs[i]->getType()->isBlockPointerType()) 496664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(InnerBlockDeclRefs[i]->getDecl()); 496764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 496864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefsCount.push_back(countOfInnerDecls); 496964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 497064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FuncName; 497164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 497264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDef) 497364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncName = CurFunctionDef->getNameAsString(); 497464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (CurMethodDef) 497564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BuildUniqueMethodName(FuncName, CurMethodDef); 497664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (GlobalVarDecl) 497764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncName = std::string(GlobalVarDecl->getNameAsString()); 497864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 4979df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian bool GlobalBlockExpr = 4980df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian block->getDeclContext()->getRedeclContext()->isFileContext(); 4981df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 4982df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian if (GlobalBlockExpr && !GlobalVarDecl) { 4983df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Diags.Report(block->getLocation(), GlobalBlockRewriteFailedDiag); 4984df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian GlobalBlockExpr = false; 4985df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian } 4986df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 498764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string BlockNumber = utostr(Blocks.size()-1); 498864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 498964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Func = "__" + FuncName + "_block_func_" + BlockNumber; 499064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 499164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get a pointer to the function type so we can cast appropriately. 499264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType BFT = convertFunctionTypeOfBlocks(Exp->getFunctionType()); 499364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FType = Context->getPointerType(BFT); 499464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 499564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD; 499664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *NewRep; 499764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 499864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 4999df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian std::string Tag; 5000df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 5001df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian if (GlobalBlockExpr) 5002df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Tag = "__global_"; 5003df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian else 5004df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Tag = "__"; 5005df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian Tag += FuncName + "_block_impl_" + BlockNumber; 5006df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 500764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl(Tag); 5008f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(FD, false, FType, VK_RValue, 500964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 501064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 501164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 501264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 501364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize the block function. 501464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl(Func); 5015f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *Arg = new (Context) DeclRefExpr(FD, false, FD->getType(), 5016f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 501764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, Context->VoidPtrTy, 501864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Arg); 501964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(castExpr); 502064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 502164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize the block descriptor. 502264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string DescData = "__" + FuncName + "_block_desc_" + BlockNumber + "_DATA"; 502364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 502464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, 502564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 502664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get(DescData.c_str()), 502764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 502864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Static, SC_None); 502964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UnaryOperator *DescRefExpr = 5030f4b88a45902af1802a1cb42ba48b1c474474f228John McCall new (Context) UnaryOperator(new (Context) DeclRefExpr(NewVD, false, 503164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 503264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, 503364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()), 503464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UO_AddrOf, 503564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidPtrTy), 503664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 503764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 503864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(DescRefExpr); 503964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 504064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add initializers for any closure decl refs. 504164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 504264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Exp; 504364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by copy" declarations. 504464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 504564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 504664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isObjCType((*I)->getType())) { 504764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Conform to ABI ([[obj retain] autorelease]). 504864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 5049f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Exp = new (Context) DeclRefExpr(FD, false, FD->getType(), 5050f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 505164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) { 505264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 505364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 505464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, QT, VK_RValue, 505564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary, SourceLocation()); 505664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 505764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (isTopLevelBlockPointerType((*I)->getType())) { 505864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 5059f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Arg = new (Context) DeclRefExpr(FD, false, FD->getType(), 5060f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 506164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = NoTypeInfoCStyleCastExpr(Context, Context->VoidPtrTy, 506264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Arg); 506364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 506464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 5065f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Exp = new (Context) DeclRefExpr(FD, false, FD->getType(), 5066f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 506764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) { 506864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 506964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 507064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, QT, VK_RValue, 507164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary, SourceLocation()); 507264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 507364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 507464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 507564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(Exp); 507664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 507764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by ref" declarations. 507864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 507964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 508064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *ND = (*I); 508164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 508264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RecName; 508364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(RecName, Name, ND, true); 508464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *II = &Context->Idents.get(RecName.c_str() 508564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian + sizeof("struct")); 508664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 508764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 508864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian II); 508964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(RD && "SynthBlockInitExpr(): Can't find RecordDecl"); 509064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castT = Context->getPointerType(Context->getTagDeclType(RD)); 509164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 509264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 5093f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Exp = new (Context) DeclRefExpr(FD, false, FD->getType(), VK_LValue, 509464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 509564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isNestedCapturedVar = false; 509664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (block) 509764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (BlockDecl::capture_const_iterator ci = block->capture_begin(), 509864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ce = block->capture_end(); ci != ce; ++ci) { 509964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const VarDecl *variable = ci->getVariable(); 510064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (variable == ND && ci->isNested()) { 510164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert (ci->isByRef() && 510264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "SynthBlockInitExpr - captured block variable is not byref"); 510364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isNestedCapturedVar = true; 510464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 510564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 510664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 510764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // captured nested byref variable has its address passed. Do not take 510864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // its address again. 510964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isNestedCapturedVar) 511064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, 511164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Exp->getType()), 511264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 511364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = NoTypeInfoCStyleCastExpr(Context, castT, CK_BitCast, Exp); 511464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(Exp); 511564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 511664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 511764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImportedBlockDecls.size()) { 511864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // generate BLOCK_HAS_COPY_DISPOSE(have helper funcs) | BLOCK_HAS_DESCRIPTOR 511964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag = (BLOCK_HAS_COPY_DISPOSE | BLOCK_HAS_DESCRIPTOR); 512064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 512164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 512264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *FlagExp = IntegerLiteral::Create(*Context, llvm::APInt(IntSize, flag), 512364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->IntTy, SourceLocation()); 512464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(FlagExp); 512564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 512664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], InitExprs.size(), 512764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FType, VK_LValue, SourceLocation()); 5128df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 5129df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian if (GlobalBlockExpr) { 5130df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian assert (GlobalConstructionExp == 0 && 5131df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian "SynthBlockInitExpr - GlobalConstructionExp must be null"); 5132df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian GlobalConstructionExp = NewRep; 5133df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian NewRep = DRE; 5134df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian } 5135df474ec64a86bb88c7543875634d3fc628105bb5Fariborz Jahanian 513664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = new (Context) UnaryOperator(NewRep, UO_AddrOf, 513764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(NewRep->getType()), 513864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 513964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = NoTypeInfoCStyleCastExpr(Context, FType, CK_BitCast, 514064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep); 514164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.clear(); 514264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.clear(); 514364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.clear(); 514464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.clear(); 514564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.clear(); 514664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.clear(); 514764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return NewRep; 514864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 514964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 515064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::IsDeclStmtInForeachHeader(DeclStmt *DS) { 515164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const ObjCForCollectionStmt * CS = 515264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCForCollectionStmt>(Stmts.back())) 515364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CS->getElement() == DS; 515464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 515564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 515664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 515764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 515864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Function Body / Expression rewriting 515964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 516064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 516164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteFunctionBodyOrGlobalInitializer(Stmt *S) { 516264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) || 516364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(S) || isa<ForStmt>(S)) 516464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.push_back(S); 516564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (isa<ObjCForCollectionStmt>(S)) { 516664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.push_back(S); 516764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCBcLabelNo.push_back(++BcLabelCount); 516864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 516964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 517064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Pseudo-object operations and ivar references need special 517164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // treatment because we're going to recursively rewrite them. 517264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PseudoObjectExpr *PseudoOp = dyn_cast<PseudoObjectExpr>(S)) { 517364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<BinaryOperator>(PseudoOp->getSyntacticForm())) { 517464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewritePropertyOrImplicitSetter(PseudoOp); 517564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 517664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewritePropertyOrImplicitGetter(PseudoOp); 517764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 517864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCIvarRefExpr *IvarRefExpr = dyn_cast<ObjCIvarRefExpr>(S)) { 517964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCIvarRefExpr(IvarRefExpr); 518064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 518164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 518264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OrigStmtRange = S->getSourceRange(); 518364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 518464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up rewrite of all children. 518564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 518664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 518764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *childStmt = (*CI); 518864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *newStmt = RewriteFunctionBodyOrGlobalInitializer(childStmt); 518964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (newStmt) { 519064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *CI = newStmt; 519164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 519264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 519364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 519464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *BE = dyn_cast<BlockExpr>(S)) { 5195f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 8> InnerBlockDeclRefs; 519664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> InnerContexts; 519764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts.insert(BE->getBlockDecl()); 519864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedLocalExternalDecls.clear(); 519964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(BE->getBody(), 520064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, InnerContexts); 520164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the block body in place. 520264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SaveCurrentBody = CurrentBody; 520364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = BE->getBody(); 520464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 520564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block literal on rhs of a property-dot-sytax assignment 520664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // must be replaced by its synthesize ast so getRewrittenText 520764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // works as expected. In this case, what actually ends up on RHS 520864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // is the blockTranscribed which is the helper function for the 520964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block literal; as in: self.c = ^() {[ace ARR];}; 521064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool saveDisableReplaceStmt = DisableReplaceStmt; 521164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = false; 521264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionBodyOrGlobalInitializer(BE->getBody()); 521364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = saveDisableReplaceStmt; 521464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = SaveCurrentBody; 521564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 521664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedLocalExternalDecls.clear(); 521764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now we snarf the rewritten text and stash it away for later use. 521864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Str = Rewrite.getRewrittenText(BE->getSourceRange()); 521964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewrittenBlockExprs[BE] = Str; 522064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 522164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *blockTranscribed = SynthBlockInitExpr(BE, InnerBlockDeclRefs); 522264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 522364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //blockTranscribed->dump(); 522464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(S, blockTranscribed); 522564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return blockTranscribed; 522664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 522764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 522864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCEncodeExpr *AtEncode = dyn_cast<ObjCEncodeExpr>(S)) 522964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteAtEncode(AtEncode); 523064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 523164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSelectorExpr *AtSelector = dyn_cast<ObjCSelectorExpr>(S)) 523264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteAtSelector(AtSelector); 523364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 523464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCStringLiteral *AtString = dyn_cast<ObjCStringLiteral>(S)) 523564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCStringLiteral(AtString); 523655947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian 523755947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian if (ObjCBoolLiteralExpr *BoolLitExpr = dyn_cast<ObjCBoolLiteralExpr>(S)) 523855947042d86773b7e16aec12070e1ffe62bff4a6Fariborz Jahanian return RewriteObjCBoolLiteralExpr(BoolLitExpr); 52390f9b18ed048cd5c1ebe786d1dee05577e84d0c27Fariborz Jahanian 5240eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard if (ObjCBoxedExpr *BoxedExpr = dyn_cast<ObjCBoxedExpr>(S)) 5241eb382ec1507cf2c8c12d7443d0b67c076223aec6Patrick Beard return RewriteObjCBoxedExpr(BoxedExpr); 524286cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian 524386cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian if (ObjCArrayLiteral *ArrayLitExpr = dyn_cast<ObjCArrayLiteral>(S)) 524486cff6080ba1db907009885adb5c6c64007e8b48Fariborz Jahanian return RewriteObjCArrayLiteralExpr(ArrayLitExpr); 5245e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian 5246e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian if (ObjCDictionaryLiteral *DictionaryLitExpr = 5247e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian dyn_cast<ObjCDictionaryLiteral>(S)) 5248e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian return RewriteObjCDictionaryLiteralExpr(DictionaryLitExpr); 524964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 525064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMessageExpr *MessExpr = dyn_cast<ObjCMessageExpr>(S)) { 525164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#if 0 525264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Before we rewrite it, put the original message expression in a comment. 525364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = MessExpr->getLocStart(); 525464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endLoc = MessExpr->getLocEnd(); 525564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 525664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 525764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(endLoc); 525864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 525964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string messString; 526064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString += "// "; 526164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString.append(startBuf, endBuf-startBuf+1); 526264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString += "\n"; 526364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 526464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Missing definition of 526564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // InsertText(clang::SourceLocation, char const*, unsigned int). 526664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // InsertText(startLoc, messString.c_str(), messString.size()); 526764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Tried this, but it didn't work either... 526864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ReplaceText(startLoc, 0, messString.c_str(), messString.size()); 526964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#endif 527064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteMessageExpr(MessExpr); 527164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 527264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 527364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtTryStmt *StmtTry = dyn_cast<ObjCAtTryStmt>(S)) 527464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCTryStmt(StmtTry); 527564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 527664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtSynchronizedStmt *StmtTry = dyn_cast<ObjCAtSynchronizedStmt>(S)) 527764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCSynchronizedStmt(StmtTry); 527864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 527964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtThrowStmt *StmtThrow = dyn_cast<ObjCAtThrowStmt>(S)) 528064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCThrowStmt(StmtThrow); 528164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 528264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolExpr *ProtocolExp = dyn_cast<ObjCProtocolExpr>(S)) 528364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCProtocolExpr(ProtocolExp); 528464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 528564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCForCollectionStmt *StmtForCollection = 528664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCForCollectionStmt>(S)) 528764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCForCollectionStmt(StmtForCollection, 528864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigStmtRange.getEnd()); 528964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BreakStmt *StmtBreakStmt = 529064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<BreakStmt>(S)) 529164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteBreakStmt(StmtBreakStmt); 529264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ContinueStmt *StmtContinueStmt = 529364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ContinueStmt>(S)) 529464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteContinueStmt(StmtContinueStmt); 529564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 529664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to check for protocol refs (id <P>, Foo <P> *) in variable decls 529764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and cast exprs. 529864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclStmt *DS = dyn_cast<DeclStmt>(S)) { 529964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: What we're doing here is modifying the type-specifier that 530064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // precedes the first Decl. In the future the DeclGroup should have 530164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // a separate type-specifier that we can rewrite. 530264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NOTE: We need to avoid rewriting the DeclStmt if it is within 530364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the context of an ObjCForCollectionStmt. For example: 530464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NSArray *someArray; 530564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for (id <FooProtocol> index in someArray) ; 530664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is because RewriteObjCForCollectionStmt() does textual rewriting 530764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and it depends on the original text locations/positions. 530864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !IsDeclStmtInForeachHeader(DS)) 530964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(*DS->decl_begin()); 531064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 531164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Blocks rewrite rules. 531264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclStmt::decl_iterator DI = DS->decl_begin(), DE = DS->decl_end(); 531364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DI != DE; ++DI) { 531464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Decl *SD = *DI; 531564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ValueDecl *ND = dyn_cast<ValueDecl>(SD)) { 531664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(ND->getType())) 531764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(ND); 531864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (ND->getType()->isFunctionPointerType()) 531964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(ND->getType(), ND); 532064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(SD)) { 532164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->hasAttr<BlocksAttr>()) { 532264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static unsigned uniqueByrefDeclCount = 0; 532364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!BlockByRefDeclNo.count(ND) && 532464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "RewriteFunctionBodyOrGlobalInitializer: Duplicate byref decl"); 532564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclNo[ND] = uniqueByrefDeclCount++; 532664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefVar(VD); 532764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 532864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 532964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeOfDecl(VD); 533064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 533164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 533264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (TypedefNameDecl *TD = dyn_cast<TypedefNameDecl>(SD)) { 533364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(TD->getUnderlyingType())) 533464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(TD); 533564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TD->getUnderlyingType()->isFunctionPointerType()) 533664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(TD->getUnderlyingType(), TD); 533764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 533864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 533964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 534064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 534164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(S)) 534264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(CE); 534364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 534464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) || 534564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(S) || isa<ForStmt>(S)) { 534664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!Stmts.empty() && "Statement stack is empty"); 534764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert ((isa<SwitchStmt>(Stmts.back()) || isa<WhileStmt>(Stmts.back()) || 534864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(Stmts.back()) || isa<ForStmt>(Stmts.back())) 534964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian && "Statement stack mismatch"); 535064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.pop_back(); 535164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 535264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle blocks rewriting. 535364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) { 535464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = DRE->getDecl(); 535564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->hasAttr<BlocksAttr>()) 535664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteBlockDeclRefExpr(DRE); 535764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(VD)) 535864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteLocalVariableExternalStorage(DRE); 535964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 536064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 536164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CallExpr *CE = dyn_cast<CallExpr>(S)) { 536264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CE->getCallee()->getType()->isBlockPointerType()) { 536364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *BlockCall = SynthesizeBlockCall(CE, CE->getCallee()); 536464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(S, BlockCall); 536564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return BlockCall; 536664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 536764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 536864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(S)) { 536964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 537064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 5371f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(S)) { 5372f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian RewriteImplicitCastObjCExpr(ICE); 5373f1ee68710ae8ca93a35ab11995ed5df321ade9ecFariborz Jahanian } 537443aa1c302b31492ab0abfbf656b06ca446d93b3aFariborz Jahanian#if 0 5375653b7cff0efe0d32ab12535846196e3465445fefFariborz Jahanian 537664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(S)) { 537764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *Replacement = new (Context) CastExpr(ICE->getType(), 537864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ICE->getSubExpr(), 537964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 538064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the new text. 538164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SStr; 538264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream Buf(SStr); 538364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Replacement->printPretty(Buf, *Context); 538464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Str = Buf.str(); 538564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 538664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian printf("CAST = %s\n", &Str[0]); 538764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(ICE->getSubExpr()->getLocStart(), &Str[0], Str.size()); 538864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete S; 538964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 539064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 539164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#endif 539264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Return this stmt unmodified. 539364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 539464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 539564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 539664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteRecordBody(RecordDecl *RD) { 539764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (RecordDecl::field_iterator i = RD->field_begin(), 539864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian e = RD->field_end(); i != e; ++i) { 539964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = *i; 540064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(FD->getType())) 540164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(FD); 540264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->getType()->isObjCQualifiedIdType() || 540364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType()->isObjCQualifiedInterfaceType()) 540464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(FD); 540564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 540664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 540764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 540864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// HandleDeclInMainFile - This is called for each top-level decl defined in the 540964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// main file of the input. 541064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleDeclInMainFile(Decl *D) { 541164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (D->getKind()) { 541264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Function: { 541364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD = cast<FunctionDecl>(D); 541464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->isOverloadedOperator()) 541564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 541664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 541764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since function prototypes don't have ParmDecl's, we check the function 541864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // prototype. This enables us to rewrite function declarations and 541964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // definitions using the same code. 542064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlocksInFunctionProtoType(FD->getType(), FD); 542164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 54229335df324951c022bc696e64b5fe9f66a8ad5222Argyrios Kyrtzidis if (!FD->isThisDeclarationADefinition()) 54239335df324951c022bc696e64b5fe9f66a8ad5222Argyrios Kyrtzidis break; 54249335df324951c022bc696e64b5fe9f66a8ad5222Argyrios Kyrtzidis 542564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: If this should support Obj-C++, support CXXTryStmt 542664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CompoundStmt *Body = dyn_cast_or_null<CompoundStmt>(FD->getBody())) { 542764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = FD; 542864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = Body; 542964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Body = 543064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast_or_null<CompoundStmt>(RewriteFunctionBodyOrGlobalInitializer(Body)); 543164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->setBody(Body); 543264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 543364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 543464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 543564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 543664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 543764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This synthesizes and inserts the block "impl" struct, invoke function, 543864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and any copy/dispose helper functions. 543964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertBlockLiteralsWithinFunction(FD); 544064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = 0; 544164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 544264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 544364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 544464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCMethod: { 544564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MD = cast<ObjCMethodDecl>(D); 544664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CompoundStmt *Body = MD->getCompoundBody()) { 544764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = MD; 544864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = Body; 544964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Body = 545064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast_or_null<CompoundStmt>(RewriteFunctionBodyOrGlobalInitializer(Body)); 545164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MD->setBody(Body); 545264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 545364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 545464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 545564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 545664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 545764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertBlockLiteralsWithinMethod(MD); 545864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = 0; 545964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 546064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 546164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 546264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCImplementation: { 546364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *CI = cast<ObjCImplementationDecl>(D); 546464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassImplementation.push_back(CI); 546564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 546664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 546764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCCategoryImpl: { 546864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CI = cast<ObjCCategoryImplDecl>(D); 546964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CategoryImplementation.push_back(CI); 547064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 547164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 547264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Var: { 547364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *VD = cast<VarDecl>(D); 547464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(VD); 547564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(VD->getType())) 547664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(VD); 547764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isFunctionPointerType()) { 547864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(VD->getType(), VD); 547964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getInit()) { 548064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(VD->getInit())) { 548164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 548264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 548364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 548464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (VD->getType()->isRecordType()) { 548564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = VD->getType()->getAs<RecordType>()->getDecl(); 548664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RD->isCompleteDefinition()) 548764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteRecordBody(RD); 548864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 548964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getInit()) { 549064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = VD; 549164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = VD->getInit(); 549264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionBodyOrGlobalInitializer(VD->getInit()); 549364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 549464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 549564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 549664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 549764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 549864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(VD->getTypeSpecStartLoc(), VD->getName()); 549964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = 0; 550064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 550164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is needed for blocks. 550264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(VD->getInit())) { 550364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 550464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 550564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 550664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 550764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 550864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::TypeAlias: 550964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Typedef: { 551064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (TypedefNameDecl *TD = dyn_cast<TypedefNameDecl>(D)) { 551164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(TD->getUnderlyingType())) 551264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(TD); 551364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TD->getUnderlyingType()->isFunctionPointerType()) 551464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(TD->getUnderlyingType(), TD); 551564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 551664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 551764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 551864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::CXXRecord: 551964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Record: { 552064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = cast<RecordDecl>(D); 552164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RD->isCompleteDefinition()) 552264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteRecordBody(RD); 552364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 552464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 552564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: 552664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 552764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 552864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Nothing yet. 552964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 553064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 553130650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian/// Write_ProtocolExprReferencedMetadata - This routine writer out the 553230650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian/// protocol reference symbols in the for of: 553330650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian/// struct _protocol_t *PROTOCOL_REF = &PROTOCOL_METADATA. 553430650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanianstatic void Write_ProtocolExprReferencedMetadata(ASTContext *Context, 553530650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian ObjCProtocolDecl *PDecl, 553630650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian std::string &Result) { 553730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian // Also output .objc_protorefs$B section and its meta-data. 553830650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian if (Context->getLangOpts().MicrosoftExt) 553930650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "__declspec(allocate(\".objc_protorefs$B\")) "; 554030650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "struct _protocol_t *"; 554130650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "_OBJC_PROTOCOL_REFERENCE_$_"; 554230650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += PDecl->getNameAsString(); 554330650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += " = &"; 554430650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "_OBJC_PROTOCOL_"; Result += PDecl->getNameAsString(); 554530650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += ";\n"; 554630650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian} 554730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian 554864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleTranslationUnit(ASTContext &C) { 554964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Diags.hasErrorOccurred()) 555064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 555164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 555264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteInclude(); 555364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 555464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Here's a great place to add any extra declarations that may be needed. 555564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Write out meta data for each @protocol(<expr>). 555664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ObjCProtocolDecl *,8>::iterator I = ProtocolExprDecls.begin(), 555730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian E = ProtocolExprDecls.end(); I != E; ++I) { 5558da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Preamble); 555930650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Write_ProtocolExprReferencedMetadata(Context, (*I), Preamble); 556030650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian } 556164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 556264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(SM->getLocForStartOfFile(MainFileID), Preamble, false); 556357317788289549d61ada868d07792b637a32d52eFariborz Jahanian for (unsigned i = 0, e = ObjCInterfacesSeen.size(); i < e; i++) { 556457317788289549d61ada868d07792b637a32d52eFariborz Jahanian ObjCInterfaceDecl *CDecl = ObjCInterfacesSeen[i]; 556557317788289549d61ada868d07792b637a32d52eFariborz Jahanian // Write struct declaration for the class matching its ivar declarations. 556657317788289549d61ada868d07792b637a32d52eFariborz Jahanian // Note that for modern abi, this is postponed until the end of TU 556757317788289549d61ada868d07792b637a32d52eFariborz Jahanian // because class extensions and the implementation might declare their own 556857317788289549d61ada868d07792b637a32d52eFariborz Jahanian // private ivars. 556957317788289549d61ada868d07792b637a32d52eFariborz Jahanian RewriteInterfaceDecl(CDecl); 557057317788289549d61ada868d07792b637a32d52eFariborz Jahanian } 557172c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 557264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ClassImplementation.size() || CategoryImplementation.size()) 557364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteImplementations(); 557464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 557564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the buffer corresponding to MainFileID. If we haven't changed it, then 557664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we are done. 557764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const RewriteBuffer *RewriteBuf = 557864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Rewrite.getRewriteBufferFor(MainFileID)) { 557964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //printf("Changed:\n"); 558064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *OutFile << std::string(RewriteBuf->begin(), RewriteBuf->end()); 558164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 558264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::errs() << "No changes\n"; 558364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 558464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 558564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ClassImplementation.size() || CategoryImplementation.size() || 558664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolExprDecls.size()) { 558764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite Objective-c meta data* 558864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 558964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMetaDataIntoBuffer(ResultStr); 559064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Emit metadata. 559164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *OutFile << ResultStr; 559264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 559310cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // Emit ImageInfo; 559410cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian { 559510cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian std::string ResultStr; 559610cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian WriteImageInfo(ResultStr); 559710cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian *OutFile << ResultStr; 559810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian } 559964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OutFile->flush(); 560064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 560164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 560264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::Initialize(ASTContext &context) { 560364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitializeCommon(context); 560464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 56056991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian Preamble += "#ifndef __OBJC2__\n"; 56066991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian Preamble += "#define __OBJC2__\n"; 56076991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian Preamble += "#endif\n"; 56086991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian 560964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declaring objc_selector outside the parameter list removes a silly 561064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scope related warning... 561164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IsHeader) 561264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble = "#pragma once\n"; 561364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct objc_selector; struct objc_class;\n"; 5614e2d87bc0be8e5a5b2e45118675ff878d89daee44Fariborz Jahanian Preamble += "struct __rw_objc_super { \n\tstruct objc_object *object; "; 5615e2d87bc0be8e5a5b2e45118675ff878d89daee44Fariborz Jahanian Preamble += "\n\tstruct objc_object *superClass; "; 5616e2d87bc0be8e5a5b2e45118675ff878d89daee44Fariborz Jahanian // Add a constructor for creating temporary objects. 5617e2d87bc0be8e5a5b2e45118675ff878d89daee44Fariborz Jahanian Preamble += "\n\t__rw_objc_super(struct objc_object *o, struct objc_object *s) "; 5618e2d87bc0be8e5a5b2e45118675ff878d89daee44Fariborz Jahanian Preamble += ": object(o), superClass(s) {} "; 5619e2d87bc0be8e5a5b2e45118675ff878d89daee44Fariborz Jahanian Preamble += "\n};\n"; 5620e2d87bc0be8e5a5b2e45118675ff878d89daee44Fariborz Jahanian 562164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 56221ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian // Define all sections using syntax that makes sense. 562310cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // These are currently generated. 562410cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "\n#pragma section(\".objc_classlist$B\", long, read, write)\n"; 56251ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Preamble += "#pragma section(\".objc_catlist$B\", long, read, write)\n"; 562610cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".objc_imageinfo$B\", long, read, write)\n"; 562788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Preamble += "#pragma section(\".objc_nlclslist$B\", long, read, write)\n"; 562888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Preamble += "#pragma section(\".objc_nlcatlist$B\", long, read, write)\n"; 562910cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".objc_protorefs$B\", long, read, write)\n"; 563010cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // These are generated but not necessary for functionality. 563110cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".cat_cls_meth$B\", long, read, write)\n"; 56321ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Preamble += "#pragma section(\".inst_meth$B\", long, read, write)\n"; 56331ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Preamble += "#pragma section(\".cls_meth$B\", long, read, write)\n"; 563440a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Preamble += "#pragma section(\".objc_ivar$B\", long, read, write)\n"; 5635de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian 563630650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian // These need be generated for performance. Currently they are not, 563730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian // using API calls instead. 563830650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Preamble += "#pragma section(\".objc_selrefs$B\", long, read, write)\n"; 563930650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Preamble += "#pragma section(\".objc_classrefs$B\", long, read, write)\n"; 564030650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Preamble += "#pragma section(\".objc_superrefs$B\", long, read, write)\n"; 564130650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian 564264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 564364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef _REWRITER_typedef_Protocol\n"; 564464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "typedef struct objc_object Protocol;\n"; 564564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define _REWRITER_typedef_Protocol\n"; 564664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 564764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 564864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_DLLIMPORT extern \"C\" __declspec(dllimport)\n"; 564964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_STATICIMPORT extern \"C\"\n"; 56505cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian } 56515cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian else 565264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_DLLIMPORT extern\n"; 56535cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian 56545cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_msgSend(void);\n"; 56555cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_msgSendSuper(void);\n"; 56565cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_msgSend_stret(void);\n"; 56575cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_msgSendSuper_stret(void);\n"; 56585cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_msgSend_fpret(void);\n"; 56595cf6b6ca7c08d3bed3fd899e7a18fc5f58c736b5Fariborz Jahanian 566064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_getClass"; 566164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(const char *);\n"; 566264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_class *class_getSuperclass"; 566364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_class *);\n"; 566464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_getMetaClass"; 566564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(const char *);\n"; 566655261afb725b8af330addddd2e452f9f28ddd96bFariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_exception_throw( struct objc_object *);\n"; 566764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // @synchronized hooks. 566855261afb725b8af330addddd2e452f9f28ddd96bFariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_sync_enter( struct objc_object *);\n"; 566955261afb725b8af330addddd2e452f9f28ddd96bFariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_sync_exit( struct objc_object *);\n"; 567064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT Protocol *objc_getProtocol(const char *);\n"; 567164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef __FASTENUMERATIONSTATE\n"; 567264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __objcFastEnumerationState {\n\t"; 567364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long state;\n\t"; 567464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "void **itemsPtr;\n\t"; 567564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long *mutationsPtr;\n\t"; 567664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long extra[5];\n};\n"; 567764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_enumerationMutation(struct objc_object *);\n"; 567864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __FASTENUMERATIONSTATE\n"; 567964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 568064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef __NSCONSTANTSTRINGIMPL\n"; 568164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __NSConstantStringImpl {\n"; 568264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int *isa;\n"; 568364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int flags;\n"; 568464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " char *str;\n"; 568564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " long length;\n"; 568664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 568764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifdef CF_EXPORT_CONSTANT_STRING\n"; 568864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) int __CFConstantStringClassReference[];\n"; 568964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#else\n"; 569064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT int __CFConstantStringClassReference[];\n"; 569164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 569264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __NSCONSTANTSTRINGIMPL\n"; 569364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 569464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Blocks preamble. 569564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef BLOCK_IMPL\n"; 569664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define BLOCK_IMPL\n"; 569764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __block_impl {\n"; 569864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " void *isa;\n"; 569964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int Flags;\n"; 570064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int Reserved;\n"; 570164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " void *FuncPtr;\n"; 570264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 570364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "// Runtime copy/destroy helper functions (from Block_private.h)\n"; 570464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifdef __OBJC_EXPORT_BLOCKS\n"; 570564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) " 570664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "void _Block_object_assign(void *, const void *, const int);\n"; 570764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void _Block_object_dispose(const void *, const int);\n"; 570864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void *_NSConcreteGlobalBlock[32];\n"; 570964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void *_NSConcreteStackBlock[32];\n"; 571064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#else\n"; 571164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void _Block_object_assign(void *, const void *, const int);\n"; 571264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void _Block_object_dispose(const void *, const int);\n"; 571364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void *_NSConcreteGlobalBlock[32];\n"; 571464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void *_NSConcreteStackBlock[32];\n"; 571564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 571664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 571764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 571864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#undef __OBJC_RW_DLLIMPORT\n"; 571964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#undef __OBJC_RW_STATICIMPORT\n"; 572064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef KEEP_ATTRIBUTES\n"; // We use this for clang tests. 572164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __attribute__(X)\n"; 572264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 57235ce2827de3aa069a4796dea404c0679620581dbeFariborz Jahanian Preamble += "#ifndef __weak\n"; 572464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __weak\n"; 57255ce2827de3aa069a4796dea404c0679620581dbeFariborz Jahanian Preamble += "#endif\n"; 57265ce2827de3aa069a4796dea404c0679620581dbeFariborz Jahanian Preamble += "#ifndef __block\n"; 57275ce2827de3aa069a4796dea404c0679620581dbeFariborz Jahanian Preamble += "#define __block\n"; 57285ce2827de3aa069a4796dea404c0679620581dbeFariborz Jahanian Preamble += "#endif\n"; 572964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 573064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 573164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __block\n"; 573264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __weak\n"; 573364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 5734b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian 5735b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian // Declarations required for modern objective-c array and dictionary literals. 5736b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\n#include <stdarg.h>\n"; 5737e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Preamble += "struct __NSContainer_literal {\n"; 5738b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += " void * *arr;\n"; 5739e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Preamble += " __NSContainer_literal (unsigned int count, ...) {\n"; 5740b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\tva_list marker;\n"; 5741b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\tva_start(marker, count);\n"; 5742b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\tarr = new void *[count];\n"; 5743b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\tfor (unsigned i = 0; i < count; i++)\n"; 5744b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\t arr[i] = va_arg(marker, void *);\n"; 5745b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\tva_end( marker );\n"; 5746b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += " };\n"; 5747e35abe1fd3f867ae51d5c68d98578d537eb6becaFariborz Jahanian Preamble += " __NSContainer_literal() {\n"; 5748b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "\tdelete[] arr;\n"; 5749b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += " }\n"; 5750b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian Preamble += "};\n"; 5751b0f245ccce94a069c9334e51f3c5d502eb2d0215Fariborz Jahanian 575264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NOTE! Windows uses LLP64 for 64bit mode. So, cast pointer to long long 575364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as this avoids warning in any 64bit/32bit compilation model. 575464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "\n#define __OFFSETOFIVAR__(TYPE, MEMBER) ((long long) &((TYPE *)0)->MEMBER)\n"; 575564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 575664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 575764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteIvarOffsetComputation - This rutine synthesizes computation of 575864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ivar offset. 575964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteIvarOffsetComputation(ObjCIvarDecl *ivar, 576064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 576164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ivar->isBitField()) { 576264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: The hack below doesn't work for bitfields. For now, we simply 576364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // place all bitfields at offset 0. 576464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "0"; 576564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 576664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "__OFFSETOFIVAR__(struct "; 576764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ivar->getContainingInterface()->getNameAsString(); 576864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) 576964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_IMPL"; 577064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", "; 577164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ivar->getNameAsString(); 577264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ")"; 577364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 577464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 577564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 577664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// WriteModernMetadataDeclarations - Writes out metadata declarations for modern ABI. 577764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_t { 5778da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian/// const char *name; 577964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// char *attributes; 578064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 578164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 578264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_list_t { 578364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t entsize; // sizeof(struct _prop_t) 578464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t count_of_properties; 578564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_t prop_list[count_of_properties]; 578664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 578764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 578864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t; 578964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 579064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_list_t { 579164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// long protocol_count; // Note, this is 32/64 bit 579264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t * protocol_list[protocol_count]; 579364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 579464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 579564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _objc_method { 579664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SEL _cmd; 579777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian/// const char *method_type; 579864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// char *_imp; 579964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 580064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 580164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _method_list_t { 580264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t entsize; // sizeof(struct _objc_method) 580364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t method_count; 580464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _objc_method method_list[method_count]; 580564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 580664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 580764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t { 580864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id isa; // NULL 58094e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const char *protocol_name; 581064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct _protocol_list_t * protocol_list; // super protocols 58114e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct method_list_t *instance_methods; 58124e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct method_list_t *class_methods; 581364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t *optionalInstanceMethods; 581464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t *optionalClassMethods; 581564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct _prop_list_t * properties; 581664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const uint32_t size; // sizeof(struct _protocol_t) 581764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const uint32_t flags; // = 0 581864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const char ** extendedMethodTypes; 581964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 582064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 582142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _ivar_t { 582242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// unsigned long int *offset; // pointer to ivar offset location 5823ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// const char *name; 5824ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// const char *type; 582542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t alignment; 582642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t size; 582742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 582842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 582942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _ivar_list_t { 583042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32 entsize; // sizeof(struct _ivar_t) 583142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32 count; 5832ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// struct _ivar_t list[count]; 583342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 583442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 583542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_ro_t { 5836249cd102cbea7410ed5cd4ab93dabf7804d79c7bFariborz Jahanian/// uint32_t flags; 5837249cd102cbea7410ed5cd4ab93dabf7804d79c7bFariborz Jahanian/// uint32_t instanceStart; 5838249cd102cbea7410ed5cd4ab93dabf7804d79c7bFariborz Jahanian/// uint32_t instanceSize; 5839249cd102cbea7410ed5cd4ab93dabf7804d79c7bFariborz Jahanian/// uint32_t reserved; // only when building for 64bit targets 58404e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const uint8_t *ivarLayout; 58414e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const char *name; 58424e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _method_list_t *baseMethods; 58434e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _protocol_list_t *baseProtocols; 58444e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _ivar_list_t *ivars; 58454e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const uint8_t *weakIvarLayout; 58464e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _prop_list_t *properties; 584742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 584842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 584942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t { 585042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t *isa; 5851fd4ce2c9d64032b42c037b2af3bf4ebeac5bbac8Fariborz Jahanian/// struct _class_t *superclass; 585242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// void *cache; 585342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// IMP *vtable; 58543f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian/// struct _class_ro_t *ro; 585542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 585642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 585742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _category_t { 58584e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const char *name; 58594b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian/// struct _class_t *cls; 58604e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _method_list_t *instance_methods; 58614e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _method_list_t *class_methods; 58624e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _protocol_list_t *protocols; 58634e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian/// const struct _prop_list_t *properties; 586442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 586542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 586642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// MessageRefTy - LLVM for: 586742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _message_ref_t { 586842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// IMP messenger; 586942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SEL name; 587042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// }; 587142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 587242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SuperMessageRefTy - LLVM for: 587342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _super_message_ref_t { 587442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SUPER_IMP messenger; 587542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SEL name; 587642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// }; 587742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 5878de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanianstatic void WriteModernMetadataDeclarations(ASTContext *Context, std::string &Result) { 587964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool meta_data_declared = false; 588064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (meta_data_declared) 588164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 588264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 588364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _prop_t {\n"; 5884da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tconst char *name;\n"; 5885da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tconst char *attributes;\n"; 588664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 588764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 588864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _protocol_t;\n"; 588964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 589064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_method {\n"; 589164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct objc_selector * _cmd;\n"; 589277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tconst char *method_type;\n"; 589390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "\tvoid *_imp;\n"; 589464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 589564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 589664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _protocol_t {\n"; 589764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tvoid * isa; // NULL\n"; 58984e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst char *protocol_name;\n"; 589964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct _protocol_list_t * protocol_list; // super protocols\n"; 59004e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct method_list_t *instance_methods;\n"; 59014e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct method_list_t *class_methods;\n"; 590264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t *optionalInstanceMethods;\n"; 590364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t *optionalClassMethods;\n"; 590464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct _prop_list_t * properties;\n"; 590564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst unsigned int size; // sizeof(struct _protocol_t)\n"; 590664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst unsigned int flags; // = 0\n"; 590764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char ** extendedMethodTypes;\n"; 590864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 590964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 591042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _ivar_t {\n"; 591142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned long int *offset; // pointer to ivar offset location\n"; 5912ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tconst char *name;\n"; 5913ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tconst char *type;\n"; 591442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int alignment;\n"; 591542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int size;\n"; 591642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 591742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 591842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _class_ro_t {\n"; 5919249cd102cbea7410ed5cd4ab93dabf7804d79c7bFariborz Jahanian Result += "\tunsigned int flags;\n"; 592042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int instanceStart;\n"; 5921249cd102cbea7410ed5cd4ab93dabf7804d79c7bFariborz Jahanian Result += "\tunsigned int instanceSize;\n"; 5922de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian const llvm::Triple &Triple(Context->getTargetInfo().getTriple()); 5923de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian if (Triple.getArch() == llvm::Triple::x86_64) 5924249cd102cbea7410ed5cd4ab93dabf7804d79c7bFariborz Jahanian Result += "\tunsigned int reserved;\n"; 59254e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst unsigned char *ivarLayout;\n"; 59264e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst char *name;\n"; 59274e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _method_list_t *baseMethods;\n"; 59284e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _objc_protocol_list *baseProtocols;\n"; 59294e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _ivar_list_t *ivars;\n"; 59304e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst unsigned char *weakIvarLayout;\n"; 59314e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _prop_list_t *properties;\n"; 593242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 593342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 593442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _class_t {\n"; 593542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tstruct _class_t *isa;\n"; 5936fd4ce2c9d64032b42c037b2af3bf4ebeac5bbac8Fariborz Jahanian Result += "\tstruct _class_t *superclass;\n"; 593742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tvoid *cache;\n"; 593842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tvoid *vtable;\n"; 59393f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "\tstruct _class_ro_t *ro;\n"; 594042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 594142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 594242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _category_t {\n"; 59434e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst char *name;\n"; 59444b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += "\tstruct _class_t *cls;\n"; 59454e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _method_list_t *instance_methods;\n"; 59464e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _method_list_t *class_methods;\n"; 59474e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _protocol_list_t *protocols;\n"; 59484e825dfc1260163564ae0f072089c4411b39e2e5Fariborz Jahanian Result += "\tconst struct _prop_list_t *properties;\n"; 594942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 595042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 59513f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "extern \"C\" __declspec(dllimport) struct objc_cache _objc_empty_cache;\n"; 5952297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "#pragma warning(disable:4273)\n"; 595364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian meta_data_declared = true; 595464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 595564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 5956da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianstatic void Write_protocol_list_t_TypeDecl(std::string &Result, 5957da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian long super_protocol_count) { 5958da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "struct /*_protocol_list_t*/"; Result += " {\n"; 5959da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\tlong protocol_count; // Note, this is 32/64 bit\n"; 5960da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\tstruct _protocol_t *super_protocols["; 5961da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += utostr(super_protocol_count); Result += "];\n"; 5962da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "}"; 5963da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian} 5964da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 596577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanianstatic void Write_method_list_t_TypeDecl(std::string &Result, 596677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian unsigned int method_count) { 596777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "struct /*_method_list_t*/"; Result += " {\n"; 596877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _objc_method)\n"; 596977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tunsigned int method_count;\n"; 597077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tstruct _objc_method method_list["; 597177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += utostr(method_count); Result += "];\n"; 597277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "}"; 597377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian} 597477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 5975da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanianstatic void Write__prop_list_t_TypeDecl(std::string &Result, 5976da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian unsigned int property_count) { 5977da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "struct /*_prop_list_t*/"; Result += " {\n"; 5978da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _prop_t)\n"; 5979da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tunsigned int count_of_properties;\n"; 5980da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tstruct _prop_t prop_list["; 5981da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += utostr(property_count); Result += "];\n"; 5982da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "}"; 5983da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian} 5984da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 5985ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanianstatic void Write__ivar_list_t_TypeDecl(std::string &Result, 5986ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian unsigned int ivar_count) { 5987ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "struct /*_ivar_list_t*/"; Result += " {\n"; 5988ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _prop_t)\n"; 5989ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tunsigned int count;\n"; 5990ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tstruct _ivar_t ivar_list["; 5991ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += utostr(ivar_count); Result += "];\n"; 5992ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "}"; 5993ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian} 5994ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5995da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianstatic void Write_protocol_list_initializer(ASTContext *Context, std::string &Result, 5996da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian ArrayRef<ObjCProtocolDecl *> SuperProtocols, 5997da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian StringRef VarName, 5998da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian StringRef ProtocolName) { 5999da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (SuperProtocols.size() > 0) { 6000da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\nstatic "; 6001da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Write_protocol_list_t_TypeDecl(Result, SuperProtocols.size()); 6002da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += " "; Result += VarName; 6003da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += ProtocolName; 6004da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 6005da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t"; Result += utostr(SuperProtocols.size()); Result += ",\n"; 6006da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (unsigned i = 0, e = SuperProtocols.size(); i < e; i++) { 6007da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian ObjCProtocolDecl *SuperPD = SuperProtocols[i]; 6008da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t&"; Result += "_OBJC_PROTOCOL_"; 6009da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += SuperPD->getNameAsString(); 6010da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (i == e-1) 6011da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\n};\n"; 6012da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian else 6013da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += ",\n"; 6014da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 6015da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 6016da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian} 6017da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 601890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanianstatic void Write_method_list_t_initializer(RewriteModernObjC &RewriteObj, 601990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian ASTContext *Context, std::string &Result, 602077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ArrayRef<ObjCMethodDecl *> Methods, 602177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian StringRef VarName, 602290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian StringRef TopLevelDeclName, 602390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian bool MethodImpl) { 602477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (Methods.size() > 0) { 602577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\nstatic "; 602677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Write_method_list_t_TypeDecl(Result, Methods.size()); 602777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += " "; Result += VarName; 602890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += TopLevelDeclName; 602977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 603077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t"; Result += "sizeof(_objc_method)"; Result += ",\n"; 603177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t"; Result += utostr(Methods.size()); Result += ",\n"; 603277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (unsigned i = 0, e = Methods.size(); i < e; i++) { 603377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = Methods[i]; 603477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (i == 0) 603577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t{{(struct objc_selector *)\""; 603677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian else 603777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t{(struct objc_selector *)\""; 603877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += (MD)->getSelector().getAsString(); Result += "\""; 603977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += ", "; 604077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::string MethodTypeString; 604177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Context->getObjCEncodingForMethodDecl(MD, MethodTypeString); 604277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\""; Result += MethodTypeString; Result += "\""; 604377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += ", "; 604490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!MethodImpl) 604590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "0"; 604690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian else { 604790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "(void *)"; 604890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += RewriteObj.MethodInternalNames[MD]; 604990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian } 605077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (i == e-1) 605190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "}}\n"; 605277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian else 605390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "},\n"; 605477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 605577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "};\n"; 605677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 605777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian} 605877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 6059f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanianstatic void Write_prop_list_t_initializer(RewriteModernObjC &RewriteObj, 6060da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ASTContext *Context, std::string &Result, 6061da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ArrayRef<ObjCPropertyDecl *> Properties, 6062da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian const Decl *Container, 6063da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian StringRef VarName, 6064da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian StringRef ProtocolName) { 6065da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (Properties.size() > 0) { 6066da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\nstatic "; 6067da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Write__prop_list_t_TypeDecl(Result, Properties.size()); 6068da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += " "; Result += VarName; 6069da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += ProtocolName; 6070da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 6071da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t"; Result += "sizeof(_prop_t)"; Result += ",\n"; 6072da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t"; Result += utostr(Properties.size()); Result += ",\n"; 6073da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian for (unsigned i = 0, e = Properties.size(); i < e; i++) { 6074da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ObjCPropertyDecl *PropDecl = Properties[i]; 6075da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (i == 0) 6076da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t{{\""; 6077da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian else 6078da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t{\""; 6079da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += PropDecl->getName(); Result += "\","; 6080da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian std::string PropertyTypeString, QuotePropertyTypeString; 6081da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Context->getObjCEncodingForPropertyDecl(PropDecl, Container, PropertyTypeString); 6082da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian RewriteObj.QuoteDoublequotes(PropertyTypeString, QuotePropertyTypeString); 6083da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\""; Result += QuotePropertyTypeString; Result += "\""; 6084da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (i == e-1) 6085da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "}}\n"; 6086da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian else 6087da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "},\n"; 6088da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian } 6089da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "};\n"; 6090da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian } 6091da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian} 6092da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 60936ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian// Metadata flags 60946ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanianenum MetaDataDlags { 60956ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS = 0x0, 60966ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_META = 0x1, 60976ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_ROOT = 0x2, 60986ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian OBJC2_CLS_HIDDEN = 0x10, 60996ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_EXCEPTION = 0x20, 61006ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 61016ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian /// (Obsolete) ARC-specific: this class has a .release_ivars method 61026ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_HAS_IVAR_RELEASER = 0x40, 61036ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian /// class was compiled with -fobjc-arr 61046ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_COMPILED_BY_ARC = 0x80 // (1<<7) 61056ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian}; 61066ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 6107f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanianstatic void Write__class_ro_t_initializer(ASTContext *Context, std::string &Result, 6108f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian unsigned int flags, 6109f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian const std::string &InstanceStart, 6110f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian const std::string &InstanceSize, 6111f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCMethodDecl *>baseMethods, 6112f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCProtocolDecl *>baseProtocols, 6113f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCIvarDecl *>ivars, 6114f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCPropertyDecl *>Properties, 6115f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian StringRef VarName, 6116f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian StringRef ClassName) { 6117f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\nstatic struct _class_ro_t "; 6118f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += VarName; Result += ClassName; 6119f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 6120f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\t"; 6121f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += llvm::utostr(flags); Result += ", "; 6122f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += InstanceStart; Result += ", "; 6123f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += InstanceSize; Result += ", \n"; 6124f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\t"; 6125de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian const llvm::Triple &Triple(Context->getTargetInfo().getTriple()); 6126de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian if (Triple.getArch() == llvm::Triple::x86_64) 6127de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian // uint32_t const reserved; // only when building for 64bit targets 6128de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian Result += "(unsigned int)0, \n\t"; 6129f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // const uint8_t * const ivarLayout; 6130f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 6131f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\""; Result += ClassName; Result += "\",\n\t"; 61326ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian bool metaclass = ((flags & CLS_META) != 0); 6133f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (baseMethods.size() > 0) { 6134f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _method_list_t *)&"; 61356ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (metaclass) 61366ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Result += "_OBJC_$_CLASS_METHODS_"; 61376ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian else 61386ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Result += "_OBJC_$_INSTANCE_METHODS_"; 61396ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Result += ClassName; 6140f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n\t"; 6141f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6142f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 6143f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 6144f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 61456ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!metaclass && baseProtocols.size() > 0) { 6146f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _objc_protocol_list *)&"; 6147f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "_OBJC_CLASS_PROTOCOLS_$_"; Result += ClassName; 6148f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n\t"; 6149f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6150f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 6151f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 6152f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 61536ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!metaclass && ivars.size() > 0) { 6154f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _ivar_list_t *)&"; 6155f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "_OBJC_$_INSTANCE_VARIABLES_"; Result += ClassName; 6156f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n\t"; 6157f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6158f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 6159f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 6160f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6161f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // weakIvarLayout 6162f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 61636ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!metaclass && Properties.size() > 0) { 6164f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _prop_list_t *)&"; 6165eeabf381862e53519cd0f9f65c1cc2a983c72159Fariborz Jahanian Result += "_OBJC_$_PROP_LIST_"; Result += ClassName; 6166f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n"; 6167f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6168f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 6169f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n"; 6170f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6171f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "};\n"; 6172f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian} 6173f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 61743f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanianstatic void Write_class_t(ASTContext *Context, std::string &Result, 61753f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian StringRef VarName, 6176a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian const ObjCInterfaceDecl *CDecl, bool metaclass) { 6177a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian bool rootClass = (!CDecl->getSuperClass()); 6178a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian const ObjCInterfaceDecl *RootClass = CDecl; 61793f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 6180a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (!rootClass) { 6181a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian // Find the Root class 6182a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian RootClass = CDecl->getSuperClass(); 6183a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian while (RootClass->getSuperClass()) { 6184a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian RootClass = RootClass->getSuperClass(); 6185a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian } 6186a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian } 6187a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6188a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (metaclass && rootClass) { 61893f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian // Need to handle a case of use of forward declaration. 6190ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "\n"; 61913f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "extern \"C\" "; 6192ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian if (CDecl->getImplementation()) 6193ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "__declspec(dllexport) "; 6194297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian else 6195297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "__declspec(dllimport) "; 6196297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian 61973f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "struct _class_t OBJC_CLASS_$_"; 61983f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 61993f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ";\n"; 62003f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 62013f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian // Also, for possibility of 'super' metadata class not having been defined yet. 6202a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (!rootClass) { 6203868e985d4e2845cf19517dac932dfd013a8494b9Fariborz Jahanian ObjCInterfaceDecl *SuperClass = CDecl->getSuperClass(); 6204ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "\n"; 62053f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "extern \"C\" "; 6206868e985d4e2845cf19517dac932dfd013a8494b9Fariborz Jahanian if (SuperClass->getImplementation()) 6207ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "__declspec(dllexport) "; 6208297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian else 6209297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "__declspec(dllimport) "; 6210297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian 62113f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "struct _class_t "; 6212ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += VarName; 6213868e985d4e2845cf19517dac932dfd013a8494b9Fariborz Jahanian Result += SuperClass->getNameAsString(); 62143f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ";\n"; 6215452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian 6216868e985d4e2845cf19517dac932dfd013a8494b9Fariborz Jahanian if (metaclass && RootClass != SuperClass) { 62173f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "extern \"C\" "; 6218452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian if (RootClass->getImplementation()) 6219452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += "__declspec(dllexport) "; 6220297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian else 6221297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "__declspec(dllimport) "; 6222297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian 62233f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "struct _class_t "; 6224452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += VarName; 6225452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += RootClass->getNameAsString(); 6226452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += ";\n"; 6227452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian } 62283f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 62293f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 6230297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "\nextern \"C\" __declspec(dllexport) struct _class_t "; 6231297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += VarName; Result += CDecl->getNameAsString(); 62323f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_data\"))) = {\n"; 62333f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "\t"; 6234a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (metaclass) { 6235a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (!rootClass) { 6236a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "0, // &"; Result += VarName; 6237a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += RootClass->getNameAsString(); 62383f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 6239a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "0, // &"; Result += VarName; 62403f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getSuperClass()->getNameAsString(); 62413f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 62423f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 62433f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else { 6244452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += "0, // &"; Result += VarName; 62453f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 62463f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 6247a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "0, // &OBJC_CLASS_$_"; Result += CDecl->getNameAsString(); 62483f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 62493f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 62503f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 62513f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else { 6252a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "0, // &OBJC_METACLASS_$_"; 62533f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 62543f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 6255a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (!rootClass) { 6256a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "0, // &"; Result += VarName; 62573f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getSuperClass()->getNameAsString(); 62583f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 62593f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 62603f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else 62613f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "0,\n\t"; 62623f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 6263a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "0, // (void *)&_objc_empty_cache,\n\t"; 6264a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "0, // unused, was (void *)&_objc_empty_vtable,\n\t"; 6265a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (metaclass) 62663f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&_OBJC_METACLASS_RO_$_"; 62673f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else 62683f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&_OBJC_CLASS_RO_$_"; 62693f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 62703f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n};\n"; 6271a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6272a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian // Add static function to initialize some of the meta-data fields. 6273a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian // avoid doing it twice. 6274a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (metaclass) 6275a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian return; 6276a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6277a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian const ObjCInterfaceDecl *SuperClass = 6278a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian rootClass ? CDecl : CDecl->getSuperClass(); 6279a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6280a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "static void OBJC_CLASS_SETUP_$_"; 6281a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += CDecl->getNameAsString(); 6282a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "(void ) {\n"; 6283a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "\tOBJC_METACLASS_$_"; Result += CDecl->getNameAsString(); 6284a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += ".isa = "; Result += "&OBJC_METACLASS_$_"; 6285452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += RootClass->getNameAsString(); Result += ";\n"; 6286a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6287a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "\tOBJC_METACLASS_$_"; Result += CDecl->getNameAsString(); 6288452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += ".superclass = "; 6289452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian if (rootClass) 6290452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += "&OBJC_CLASS_$_"; 6291452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian else 6292452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian Result += "&OBJC_METACLASS_$_"; 6293452eac1fd71bc13c23eb034f5f66281ae27cf817Fariborz Jahanian 6294a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += SuperClass->getNameAsString(); Result += ";\n"; 6295a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6296a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "\tOBJC_METACLASS_$_"; Result += CDecl->getNameAsString(); 6297a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += ".cache = "; Result += "&_objc_empty_cache"; Result += ";\n"; 6298a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6299a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "\tOBJC_CLASS_$_"; Result += CDecl->getNameAsString(); 6300a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += ".isa = "; Result += "&OBJC_METACLASS_$_"; 6301a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += CDecl->getNameAsString(); Result += ";\n"; 6302a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6303a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian if (!rootClass) { 6304a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "\tOBJC_CLASS_$_"; Result += CDecl->getNameAsString(); 6305a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += ".superclass = "; Result += "&OBJC_CLASS_$_"; 6306a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += SuperClass->getNameAsString(); Result += ";\n"; 6307a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian } 6308a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian 6309a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "\tOBJC_CLASS_$_"; Result += CDecl->getNameAsString(); 6310a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += ".cache = "; Result += "&_objc_empty_cache"; Result += ";\n"; 6311a03e40ce9d729fca50e817351e3b1b48691dcd64Fariborz Jahanian Result += "}\n"; 63123f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian} 63133f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 63146118612837887cbae365676d53cc3e1c258249aaFariborz Jahanianstatic void Write_category_t(RewriteModernObjC &RewriteObj, ASTContext *Context, 63156118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian std::string &Result, 6316e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian ObjCCategoryDecl *CatDecl, 631788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian ObjCInterfaceDecl *ClassDecl, 63186118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCMethodDecl *> InstanceMethods, 63196118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCMethodDecl *> ClassMethods, 63206118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCProtocolDecl *> RefedProtocols, 63216118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCPropertyDecl *> ClassProperties) { 6322e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian StringRef CatName = CatDecl->getName(); 632320f89390c7f9eaf7c62bcc9145fbc5ba7725557bNAKAMURA Takumi StringRef ClassName = ClassDecl->getName(); 63248c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian // must declare an extern class object in case this class is not implemented 63258c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian // in this TU. 632688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "\n"; 63273f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "extern \"C\" "; 632888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (ClassDecl->getImplementation()) 632988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "__declspec(dllexport) "; 6330297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian else 6331297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "__declspec(dllimport) "; 633288f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 63333f162c398be7fd02167e7382b3a6728c7b2b672dFariborz Jahanian Result += "struct _class_t "; 63348c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian Result += "OBJC_CLASS_$_"; Result += ClassName; 63358c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian Result += ";\n"; 63368c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian 63376118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\nstatic struct _category_t "; 63386118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_$_CATEGORY_"; 63396118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 63406118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = \n"; 63416118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "{\n"; 63426118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t\""; Result += ClassName; Result += "\",\n"; 63434b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += "\t0, // &"; Result += "OBJC_CLASS_$_"; Result += ClassName; 63446118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 63456118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (InstanceMethods.size() > 0) { 63466118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _method_list_t *)&"; 63476118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_$_CATEGORY_INSTANCE_METHODS_"; 63486118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 63496118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 63506118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 63516118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 63526118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 63536118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 63546118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (ClassMethods.size() > 0) { 63556118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _method_list_t *)&"; 63566118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_$_CATEGORY_CLASS_METHODS_"; 63576118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 63586118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 63596118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 63606118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 63616118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 63626118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 63636118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (RefedProtocols.size() > 0) { 63646118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _protocol_list_t *)&"; 63656118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_CATEGORY_PROTOCOLS_$_"; 63666118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 63676118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 63686118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 63696118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 63706118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 63716118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 63726118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (ClassProperties.size() > 0) { 63736118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _prop_list_t *)&"; Result += "_OBJC_$_PROP_LIST_"; 63746118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 63756118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 63766118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 63776118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 63786118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 63796118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 63806118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "};\n"; 63814b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian 63824b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian // Add static function to initialize the class pointer in the category structure. 63834b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += "static void OBJC_CATEGORY_SETUP_$_"; 63844b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += ClassDecl->getNameAsString(); 63854b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += "_$_"; 63864b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += CatName; 63874b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += "(void ) {\n"; 63884b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += "\t_OBJC_$_CATEGORY_"; 63894b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += ClassDecl->getNameAsString(); 63904b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += "_$_"; 63914b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += CatName; 63924b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += ".cls = "; Result += "&OBJC_CLASS_$_"; Result += ClassName; 63934b2fe6e48fe72122fc749a79953755c6fa055900Fariborz Jahanian Result += ";\n}\n"; 63946118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian} 63956118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 6396e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanianstatic void Write__extendedMethodTypes_initializer(RewriteModernObjC &RewriteObj, 6397e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ASTContext *Context, std::string &Result, 6398e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ArrayRef<ObjCMethodDecl *> Methods, 6399e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian StringRef VarName, 6400e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian StringRef ProtocolName) { 6401e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (Methods.size() == 0) 6402e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian return; 6403e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 6404e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\nstatic const char *"; 6405e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += VarName; Result += ProtocolName; 6406e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += " [] __attribute__ ((used, section (\"__DATA,__objc_const\"))) = \n"; 6407e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "{\n"; 6408e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = Methods.size(); i < e; i++) { 6409e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ObjCMethodDecl *MD = Methods[i]; 6410e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian std::string MethodTypeString, QuoteMethodTypeString; 6411e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Context->getObjCEncodingForMethodDecl(MD, MethodTypeString, true); 6412e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian RewriteObj.QuoteDoublequotes(MethodTypeString, QuoteMethodTypeString); 6413e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t\""; Result += QuoteMethodTypeString; Result += "\""; 6414e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (i == e-1) 6415e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\n};\n"; 6416e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian else { 6417e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += ",\n"; 6418e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 6419e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 6420e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian} 6421e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 6422acee1c9dbe180c6493d1bcda6a0a7ee4018ed7b1Fariborz Jahanianstatic void Write_IvarOffsetVar(RewriteModernObjC &RewriteObj, 6423acee1c9dbe180c6493d1bcda6a0a7ee4018ed7b1Fariborz Jahanian ASTContext *Context, 642440a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian std::string &Result, 6425db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian ArrayRef<ObjCIvarDecl *> Ivars, 64267cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian ObjCInterfaceDecl *CDecl) { 6427db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian // FIXME. visibilty of offset symbols may have to be set; for Darwin 6428db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian // this is what happens: 6429db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian /** 6430db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian if (Ivar->getAccessControl() == ObjCIvarDecl::Private || 6431db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Ivar->getAccessControl() == ObjCIvarDecl::Package || 6432db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Class->getVisibility() == HiddenVisibility) 6433db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Visibility shoud be: HiddenVisibility; 6434db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian else 6435db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Visibility shoud be: DefaultVisibility; 6436db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian */ 6437db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian 643807e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian Result += "\n"; 643907e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian for (unsigned i =0, e = Ivars.size(); i < e; i++) { 644007e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian ObjCIvarDecl *IvarDecl = Ivars[i]; 644140a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (Context->getLangOpts().MicrosoftExt) 644240a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Result += "__declspec(allocate(\".objc_ivar$B\")) "; 644340a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian 644440a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (!Context->getLangOpts().MicrosoftExt || 644540a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian IvarDecl->getAccessControl() == ObjCIvarDecl::Private || 6446117591f6247b186a4161459ad8603563cfd1c54cFariborz Jahanian IvarDecl->getAccessControl() == ObjCIvarDecl::Package) 6447297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "extern \"C\" unsigned long int "; 6448d1c84d3c0206abdf8787dc5c00aaf3b016ed3226Fariborz Jahanian else 6449297976d3dc9a5c402b52c734dc06d69f2f987e1cFariborz Jahanian Result += "extern \"C\" __declspec(dllexport) unsigned long int "; 64507cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(CDecl, IvarDecl, Result); 645107e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_ivar\")))"; 645207e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian Result += " = "; 6453acee1c9dbe180c6493d1bcda6a0a7ee4018ed7b1Fariborz Jahanian RewriteObj.RewriteIvarOffsetComputation(IvarDecl, Result); 6454acee1c9dbe180c6493d1bcda6a0a7ee4018ed7b1Fariborz Jahanian Result += ";\n"; 6455db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian } 6456db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian} 6457db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian 6458ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanianstatic void Write__ivar_list_t_initializer(RewriteModernObjC &RewriteObj, 6459ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ASTContext *Context, std::string &Result, 6460ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ArrayRef<ObjCIvarDecl *> Ivars, 6461ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian StringRef VarName, 64627cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian ObjCInterfaceDecl *CDecl) { 6463ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (Ivars.size() > 0) { 6464acee1c9dbe180c6493d1bcda6a0a7ee4018ed7b1Fariborz Jahanian Write_IvarOffsetVar(RewriteObj, Context, Result, Ivars, CDecl); 646507e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian 6466ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\nstatic "; 6467ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Write__ivar_list_t_TypeDecl(Result, Ivars.size()); 6468ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += " "; Result += VarName; 64697cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += CDecl->getNameAsString(); 6470ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 6471ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t"; Result += "sizeof(_ivar_t)"; Result += ",\n"; 6472ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t"; Result += utostr(Ivars.size()); Result += ",\n"; 6473ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian for (unsigned i =0, e = Ivars.size(); i < e; i++) { 6474ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ObjCIvarDecl *IvarDecl = Ivars[i]; 6475ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (i == 0) 6476ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t{{"; 6477ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian else 6478ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t {"; 64797cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += "(unsigned long int *)&"; 64807cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(CDecl, IvarDecl, Result); 6481db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Result += ", "; 6482ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 6483ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\""; Result += IvarDecl->getName(); Result += "\", "; 6484ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian std::string IvarTypeString, QuoteIvarTypeString; 6485ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Context->getObjCEncodingForType(IvarDecl->getType(), IvarTypeString, 6486ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IvarDecl); 6487ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian RewriteObj.QuoteDoublequotes(IvarTypeString, QuoteIvarTypeString); 6488ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\""; Result += QuoteIvarTypeString; Result += "\", "; 6489ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 64908f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // FIXME. this alignment represents the host alignment and need be changed to 64918f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // represent the target alignment. 64928f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian unsigned Align = Context->getTypeAlign(IvarDecl->getType())/8; 64938f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian Align = llvm::Log2_32(Align); 6494ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += llvm::utostr(Align); Result += ", "; 6495a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian CharUnits Size = Context->getTypeSizeInChars(IvarDecl->getType()); 6496a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += llvm::utostr(Size.getQuantity()); 6497ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (i == e-1) 6498ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "}}\n"; 6499ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian else 6500ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "},\n"; 6501ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian } 6502ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "};\n"; 6503ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian } 6504ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian} 6505ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 650664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCProtocolMetaData - Rewrite protocols meta-data. 6507da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCProtocolMetaData(ObjCProtocolDecl *PDecl, 6508da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::string &Result) { 650964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 651064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Do not synthesize the protocol more than once. 651164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSynthesizedProtocols.count(PDecl->getCanonicalDecl())) 651264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 6513de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian WriteModernMetadataDeclarations(Context, Result); 651464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 651564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Def = PDecl->getDefinition()) 651664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PDecl = Def; 6517da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // Must write out all protocol definitions in current qualifier list, 6518da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // and in their nested qualifiers before writing out current definition. 6519da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (ObjCProtocolDecl::protocol_iterator I = PDecl->protocol_begin(), 6520da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian E = PDecl->protocol_end(); I != E; ++I) 6521da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Result); 652264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 652377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian // Construct method lists. 652477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::vector<ObjCMethodDecl *> InstanceMethods, ClassMethods; 652577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::vector<ObjCMethodDecl *> OptInstanceMethods, OptClassMethods; 652677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (ObjCProtocolDecl::instmeth_iterator 652777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I = PDecl->instmeth_begin(), E = PDecl->instmeth_end(); 652877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I != E; ++I) { 652977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = *I; 653077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (MD->getImplementationControl() == ObjCMethodDecl::Optional) { 653177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian OptInstanceMethods.push_back(MD); 653277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } else { 653377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian InstanceMethods.push_back(MD); 653477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 653577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 653677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 653777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (ObjCProtocolDecl::classmeth_iterator 653877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I = PDecl->classmeth_begin(), E = PDecl->classmeth_end(); 653977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I != E; ++I) { 654077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = *I; 654177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (MD->getImplementationControl() == ObjCMethodDecl::Optional) { 654277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian OptClassMethods.push_back(MD); 654377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } else { 654477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ClassMethods.push_back(MD); 654577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 654677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 6547e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian std::vector<ObjCMethodDecl *> AllMethods; 6548e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = InstanceMethods.size(); i < e; i++) 6549e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(InstanceMethods[i]); 6550e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = ClassMethods.size(); i < e; i++) 6551e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(ClassMethods[i]); 6552e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = OptInstanceMethods.size(); i < e; i++) 6553e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(OptInstanceMethods[i]); 6554e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = OptClassMethods.size(); i < e; i++) 6555e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(OptClassMethods[i]); 6556e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 6557e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Write__extendedMethodTypes_initializer(*this, Context, Result, 6558e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods, 6559e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian "_OBJC_PROTOCOL_METHOD_TYPES_", 6560e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian PDecl->getNameAsString()); 6561da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // Protocol's super protocol list 6562da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::vector<ObjCProtocolDecl *> SuperProtocols; 6563da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (ObjCProtocolDecl::protocol_iterator I = PDecl->protocol_begin(), 6564da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian E = PDecl->protocol_end(); I != E; ++I) 6565da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian SuperProtocols.push_back(*I); 6566da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 6567da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Write_protocol_list_initializer(Context, Result, SuperProtocols, 6568da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian "_OBJC_PROTOCOL_REFS_", 6569da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian PDecl->getNameAsString()); 6570da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 657190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, InstanceMethods, 657277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_INSTANCE_METHODS_", 657390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 657477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 657590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, ClassMethods, 657677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_CLASS_METHODS_", 657790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 657877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 657990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, OptInstanceMethods, 658077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_OPT_INSTANCE_METHODS_", 658190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 658277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 658390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, OptClassMethods, 658477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_OPT_CLASS_METHODS_", 658590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 658677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 6587da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian // Protocol's property metadata. 6588da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian std::vector<ObjCPropertyDecl *> ProtocolProperties; 6589da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian for (ObjCContainerDecl::prop_iterator I = PDecl->prop_begin(), 6590da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian E = PDecl->prop_end(); I != E; ++I) 6591da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ProtocolProperties.push_back(*I); 6592da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 6593f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Write_prop_list_t_initializer(*this, Context, Result, ProtocolProperties, 6594da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian /* Container */0, 6595da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian "_OBJC_PROTOCOL_PROPERTIES_", 6596da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian PDecl->getNameAsString()); 6597da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 659882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian // Writer out root metadata for current protocol: struct _protocol_t 65991ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "\n"; 66001ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 66018590d86ea35d0dd04711edd6b8dc57fa9a528305Fariborz Jahanian Result += "static "; 660230650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "struct _protocol_t _OBJC_PROTOCOL_"; 660382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); 660482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__datacoal_nt,coalesced\"))) = {\n"; 660582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; // id is; is null 660682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t\""; Result += PDecl->getNameAsString(); Result += "\",\n"; 6607da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (SuperProtocols.size() > 0) { 6608da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t(const struct _protocol_list_t *)&"; Result += "_OBJC_PROTOCOL_REFS_"; 6609da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 6610da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 6611da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian else 6612da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t0,\n"; 661382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (InstanceMethods.size() > 0) { 661482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_INSTANCE_METHODS_"; 661582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 661664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 661782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 661882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 661982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 662082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (ClassMethods.size() > 0) { 662182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_CLASS_METHODS_"; 662282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 662364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 662482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 662582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 662664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 662782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (OptInstanceMethods.size() > 0) { 662882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_OPT_INSTANCE_METHODS_"; 662982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 663064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 663182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 663282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 663364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 663482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (OptClassMethods.size() > 0) { 663582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_OPT_CLASS_METHODS_"; 663682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 663764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 663864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 663982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 664082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 664182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (ProtocolProperties.size() > 0) { 664282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct _prop_list_t *)&_OBJC_PROTOCOL_PROPERTIES_"; 664382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 664464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 664564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 664682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 664764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 664882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t"; Result += "sizeof(_protocol_t)"; Result += ",\n"; 664982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 665082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 6651e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (AllMethods.size() > 0) { 6652e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t(const char **)&"; Result += "_OBJC_PROTOCOL_METHOD_TYPES_"; 6653e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += PDecl->getNameAsString(); 6654e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\n};\n"; 6655e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 6656e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian else 6657e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t0\n};\n"; 665810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian 665910cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian if (LangOpts.MicrosoftExt) 66608590d86ea35d0dd04711edd6b8dc57fa9a528305Fariborz Jahanian Result += "static "; 666110cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "struct _protocol_t *"; 666210cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "_OBJC_LABEL_PROTOCOL_$_"; Result += PDecl->getNameAsString(); 666310cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += " = &_OBJC_PROTOCOL_"; Result += PDecl->getNameAsString(); 666410cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += ";\n"; 666582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 666664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Mark this protocol as having been generated. 666764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ObjCSynthesizedProtocols.insert(PDecl->getCanonicalDecl())) 666864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("protocol already synthesized"); 666964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 667064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 667164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 667264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCProtocolListMetaData( 667364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Protocols, 667464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, StringRef ClassName, 667564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 667664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Protocols.empty()) return; 667764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 667864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i != Protocols.size(); i++) 6679da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(Protocols[i], Result); 668064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 668164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output the top lovel protocol meta-data for the class. 668264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_protocol_list { 668364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_protocol_list *next; 668464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int protocol_count; 668564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_protocol *class_protocols[]; 668664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 668764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 66881ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "\n"; 66891ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 66901ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".cat_cls_meth$B\")) "; 66911ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "static struct {\n"; 669264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol_list *next;\n"; 669364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tint protocol_count;\n"; 669464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol *class_protocols["; 669564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(Protocols.size()); 669664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "];\n} _OBJC_"; 669764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += prefix; 669864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_PROTOCOLS_"; 669964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassName; 670064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __cat_cls_meth\")))= " 670164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "{\n\t0, "; 670264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(Protocols.size()); 670364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 670464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 670564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,{&_OBJC_PROTOCOL_"; 670664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += Protocols[0]->getNameAsString(); 670764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " \n"; 670864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 670964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 1; i != Protocols.size(); i++) { 671064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t ,&_OBJC_PROTOCOL_"; 671164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += Protocols[i]->getNameAsString(); 671264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 671364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 671464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t }\n};\n"; 671564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 671664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 67176ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian/// hasObjCExceptionAttribute - Return true if this class or any super 67186ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian/// class has the __objc_exception__ attribute. 67196ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian/// FIXME. Move this to ASTContext.cpp as it is also used for IRGen. 67206ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanianstatic bool hasObjCExceptionAttribute(ASTContext &Context, 67216ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian const ObjCInterfaceDecl *OID) { 67226ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (OID->hasAttr<ObjCExceptionAttr>()) 67236ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian return true; 67246ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (const ObjCInterfaceDecl *Super = OID->getSuperClass()) 67256ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian return hasObjCExceptionAttribute(Context, Super); 67266ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian return false; 67276ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian} 6728f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 672964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl, 673064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 673164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *CDecl = IDecl->getClassInterface(); 673264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 673364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Explicitly declared @interface's are already synthesized. 6734ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (CDecl->isImplicitInterfaceDecl()) 6735ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian assert(false && 6736ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian "Legacy implicit interface rewriting not supported in moder abi"); 67378f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian 6738de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian WriteModernMetadataDeclarations(Context, Result); 6739ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian SmallVector<ObjCIvarDecl *, 8> IVars; 6740ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 6741ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian for (ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 6742ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IVD; IVD = IVD->getNextIvar()) { 6743ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian // Ignore unnamed bit-fields. 6744ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (!IVD->getDeclName()) 6745ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian continue; 6746ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IVars.push_back(IVD); 674764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 674864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6749ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Write__ivar_list_t_initializer(*this, Context, Result, IVars, 6750f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian "_OBJC_$_INSTANCE_VARIABLES_", 67517cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian CDecl); 675290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 675390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian // Build _objc_method_list for class's instance methods if needed 675490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 675590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian InstanceMethods(IDecl->instmeth_begin(), IDecl->instmeth_end()); 675690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 675790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian // If any of our property implementations have associated getters or 675890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian // setters, produce metadata for them as well. 675990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian for (ObjCImplDecl::propimpl_iterator Prop = IDecl->propimpl_begin(), 676090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PropEnd = IDecl->propimpl_end(); 676190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Prop != PropEnd; ++Prop) { 676290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if ((*Prop)->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 676390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 676490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!(*Prop)->getPropertyIvarDecl()) 676590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 676690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian ObjCPropertyDecl *PD = (*Prop)->getPropertyDecl(); 676790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!PD) 676890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 676990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (ObjCMethodDecl *Getter = PD->getGetterMethodDecl()) 677090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!Getter->isDefined()) 677190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian InstanceMethods.push_back(Getter); 677290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (PD->isReadOnly()) 677390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 677490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (ObjCMethodDecl *Setter = PD->getSetterMethodDecl()) 677590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!Setter->isDefined()) 677690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian InstanceMethods.push_back(Setter); 677790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian } 677890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 677990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, InstanceMethods, 678090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian "_OBJC_$_INSTANCE_METHODS_", 678190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian IDecl->getNameAsString(), true); 678290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 678390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 678490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian ClassMethods(IDecl->classmeth_begin(), IDecl->classmeth_end()); 678590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 678690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, ClassMethods, 678790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian "_OBJC_$_CLASS_METHODS_", 678890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian IDecl->getNameAsString(), true); 67890a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian 67900a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // Protocols referenced in class declaration? 67910a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // Protocol's super protocol list 67920a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian std::vector<ObjCProtocolDecl *> RefedProtocols; 67930a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Protocols = CDecl->getReferencedProtocols(); 67940a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), 67950a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian E = Protocols.end(); 67960a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian I != E; ++I) { 67970a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian RefedProtocols.push_back(*I); 67980a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // Must write out all protocol definitions in current qualifier list, 67990a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // and in their nested qualifiers before writing out current definition. 68000a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Result); 68010a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian } 68020a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian 68030a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian Write_protocol_list_initializer(Context, Result, 68040a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian RefedProtocols, 68050a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian "_OBJC_CLASS_PROTOCOLS_$_", 68060a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian IDecl->getNameAsString()); 6807f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6808f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // Protocol's property metadata. 6809f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian std::vector<ObjCPropertyDecl *> ClassProperties; 6810f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian for (ObjCContainerDecl::prop_iterator I = CDecl->prop_begin(), 6811f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian E = CDecl->prop_end(); I != E; ++I) 6812f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ClassProperties.push_back(*I); 6813f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6814f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Write_prop_list_t_initializer(*this, Context, Result, ClassProperties, 68152df089d4d8d47c3d0d43325d5bb69ec49147f81fFariborz Jahanian /* Container */IDecl, 6816eeabf381862e53519cd0f9f65c1cc2a983c72159Fariborz Jahanian "_OBJC_$_PROP_LIST_", 6817f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian CDecl->getNameAsString()); 681890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 6819f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 68206ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian // Data for initializing _class_ro_t metaclass meta-data 68216ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian uint32_t flags = CLS_META; 68226ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian std::string InstanceSize; 68236ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian std::string InstanceStart; 68246ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 68256ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 68266ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian bool classIsHidden = CDecl->getVisibility() == HiddenVisibility; 68276ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (classIsHidden) 68286ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= OBJC2_CLS_HIDDEN; 68296ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 68306ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!CDecl->getSuperClass()) 68316ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian // class is root 68326ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= CLS_ROOT; 68336ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceSize = "sizeof(struct _class_t)"; 68346ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceStart = InstanceSize; 68356ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Write__class_ro_t_initializer(Context, Result, flags, 68366ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceStart, InstanceSize, 68376ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian ClassMethods, 68386ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 0, 68396ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 0, 68406ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 0, 68416ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian "_OBJC_METACLASS_RO_$_", 68426ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CDecl->getNameAsString()); 68436ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 68446ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 6845f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // Data for initializing _class_ro_t meta-data 68466ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags = CLS; 68476ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (classIsHidden) 68486ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= OBJC2_CLS_HIDDEN; 68496ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 68506ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (hasObjCExceptionAttribute(*Context, CDecl)) 68516ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= CLS_EXCEPTION; 68526ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 6853f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (!CDecl->getSuperClass()) 6854f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // class is root 6855f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian flags |= CLS_ROOT; 6856f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 68576ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceSize.clear(); 68586ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceStart.clear(); 6859f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (!ObjCSynthesizedStructs.count(CDecl)) { 6860f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize = "0"; 6861f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart = "0"; 6862f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6863f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else { 6864f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize = "sizeof(struct "; 6865f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize += CDecl->getNameAsString(); 6866f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize += "_IMPL)"; 6867f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6868f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 6869f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (IVD) { 6870acee1c9dbe180c6493d1bcda6a0a7ee4018ed7b1Fariborz Jahanian RewriteIvarOffsetComputation(IVD, InstanceStart); 6871f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6872f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 6873f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart = InstanceSize; 6874f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6875f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Write__class_ro_t_initializer(Context, Result, flags, 6876f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart, InstanceSize, 6877f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceMethods, 6878f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian RefedProtocols, 6879f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian IVars, 6880f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ClassProperties, 6881f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian "_OBJC_CLASS_RO_$_", 6882f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian CDecl->getNameAsString()); 68833f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 68843f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Write_class_t(Context, Result, 68853f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian "OBJC_METACLASS_$_", 68863f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian CDecl, /*metaclass*/true); 68873f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 68883f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Write_class_t(Context, Result, 68893f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian "OBJC_CLASS_$_", 68903f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian CDecl, /*metaclass*/false); 689188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 689288f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (ImplementationIsNonLazy(IDecl)) 689388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian DefinedNonLazyClasses.push_back(CDecl); 68943f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 689564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 689664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6897e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanianvoid RewriteModernObjC::RewriteClassSetupInitHook(std::string &Result) { 6898e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian int ClsDefCount = ClassImplementation.size(); 6899e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian if (!ClsDefCount) 6900e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian return; 6901e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "#pragma section(\".objc_inithooks$B\", long, read, write)\n"; 6902e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "__declspec(allocate(\".objc_inithooks$B\")) "; 6903e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "static void *OBJC_CLASS_SETUP[] = {\n"; 6904e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian for (int i = 0; i < ClsDefCount; i++) { 6905e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian ObjCImplementationDecl *IDecl = ClassImplementation[i]; 6906e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian ObjCInterfaceDecl *CDecl = IDecl->getClassInterface(); 6907e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "\t(void *)&OBJC_CLASS_SETUP_$_"; 6908e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += CDecl->getName(); Result += ",\n"; 6909e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian } 6910e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "};\n"; 6911e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian} 6912e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian 691364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteMetaDataIntoBuffer(std::string &Result) { 691464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int ClsDefCount = ClassImplementation.size(); 691564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int CatDefCount = CategoryImplementation.size(); 691664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 691764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For each implemented class, write out all its meta data. 691864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) 691964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCClassMetaData(ClassImplementation[i], Result); 692064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6921e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian RewriteClassSetupInitHook(Result); 6922e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian 692364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For each implemented category, write out all its meta data. 692464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < CatDefCount; i++) 692564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCCategoryImplDecl(CategoryImplementation[i], Result); 692664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6927e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian RewriteCategorySetupInitHook(Result); 6928e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian 6929df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian if (ClsDefCount > 0) { 69301ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 69311ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".objc_classlist$B\")) "; 6932df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += "static struct _class_t *L_OBJC_LABEL_CLASS_$ ["; 6933df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += llvm::utostr(ClsDefCount); Result += "]"; 6934df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += 6935df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian " __attribute__((used, section (\"__DATA, __objc_classlist," 6936df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian "regular,no_dead_strip\")))= {\n"; 6937df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) { 6938df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += "\t&OBJC_CLASS_$_"; 6939df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += ClassImplementation[i]->getNameAsString(); 6940df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += ",\n"; 694164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 6942df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += "};\n"; 694388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 694488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (!DefinedNonLazyClasses.empty()) { 694588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (LangOpts.MicrosoftExt) 694688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "__declspec(allocate(\".objc_nlclslist$B\")) \n"; 694788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "static struct _class_t *_OBJC_LABEL_NONLAZY_CLASS_$[] = {\n\t"; 694888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian for (unsigned i = 0, e = DefinedNonLazyClasses.size(); i < e; i++) { 694988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "\t&OBJC_CLASS_$_"; Result += DefinedNonLazyClasses[i]->getNameAsString(); 695088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += ",\n"; 695188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 695288f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "};\n"; 695388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 695464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 69556118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 69566118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (CatDefCount > 0) { 69571ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 69581ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".objc_catlist$B\")) "; 69596118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "static struct _category_t *L_OBJC_LABEL_CATEGORY_$ ["; 69606118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += llvm::utostr(CatDefCount); Result += "]"; 69616118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += 69626118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian " __attribute__((used, section (\"__DATA, __objc_catlist," 69636118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "regular,no_dead_strip\")))= {\n"; 69646118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian for (int i = 0; i < CatDefCount; i++) { 69656118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t&_OBJC_$_CATEGORY_"; 69666118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += 69676118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian CategoryImplementation[i]->getClassInterface()->getNameAsString(); 69686118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_$_"; 69696118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += CategoryImplementation[i]->getNameAsString(); 69706118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 69716118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 69726118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "};\n"; 69736118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 697488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 697588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (!DefinedNonLazyCategories.empty()) { 697688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (LangOpts.MicrosoftExt) 697788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "__declspec(allocate(\".objc_nlcatlist$B\")) \n"; 697888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "static struct _category_t *_OBJC_LABEL_NONLAZY_CATEGORY_$[] = {\n\t"; 697988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian for (unsigned i = 0, e = DefinedNonLazyCategories.size(); i < e; i++) { 698088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "\t&_OBJC_$_CATEGORY_"; 698188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += 698288f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian DefinedNonLazyCategories[i]->getClassInterface()->getNameAsString(); 698388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "_$_"; 698488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += DefinedNonLazyCategories[i]->getNameAsString(); 698588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += ",\n"; 698688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 698788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "};\n"; 698888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 698964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 699064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 699110cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanianvoid RewriteModernObjC::WriteImageInfo(std::string &Result) { 699210cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian if (LangOpts.MicrosoftExt) 699310cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "__declspec(allocate(\".objc_imageinfo$B\")) \n"; 699410cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian 699510cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "static struct IMAGE_INFO { unsigned version; unsigned flag; } "; 699610cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // version 0, ObjCABI is 2 699730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "_OBJC_IMAGE_INFO = { 0, 2 };\n"; 699810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian} 699910cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian 700064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCCategoryImplDecl - Rewrite metadata for each category 700164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// implementation. 700264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *IDecl, 700364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 7004de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian WriteModernMetadataDeclarations(Context, Result); 700564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = IDecl->getClassInterface(); 700664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find category declaration for this implementation. 70076118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ObjCCategoryDecl *CDecl=0; 700864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (CDecl = ClassDecl->getCategoryList(); CDecl; 700964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CDecl = CDecl->getNextClassCategory()) 701064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CDecl->getIdentifier() == IDecl->getIdentifier()) 701164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 701264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 701364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FullCategoryName = ClassDecl->getNameAsString(); 70146118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName += "_$_"; 70156118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName += CDecl->getNameAsString(); 701664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 701764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's instance methods if needed 701864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 701964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods(IDecl->instmeth_begin(), IDecl->instmeth_end()); 702064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 702164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If any of our property implementations have associated getters or 702264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // setters, produce metadata for them as well. 702364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCImplDecl::propimpl_iterator Prop = IDecl->propimpl_begin(), 702464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropEnd = IDecl->propimpl_end(); 702564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Prop != PropEnd; ++Prop) { 702664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*Prop)->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 702764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 702864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!(*Prop)->getPropertyIvarDecl()) 702964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 703064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl *PD = (*Prop)->getPropertyDecl(); 703164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!PD) 703264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 703364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Getter = PD->getGetterMethodDecl()) 703464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Getter); 703564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isReadOnly()) 703664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 703764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Setter = PD->getSetterMethodDecl()) 703864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Setter); 703964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 704064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 70416118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, InstanceMethods, 70426118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_$_CATEGORY_INSTANCE_METHODS_", 70436118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName, true); 704464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 70456118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 70466118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassMethods(IDecl->classmeth_begin(), IDecl->classmeth_end()); 704764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 70486118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, ClassMethods, 70496118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_$_CATEGORY_CLASS_METHODS_", 70506118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName, true); 705164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 70526118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Protocols referenced in class declaration? 70536118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Protocol's super protocol list 70546118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian std::vector<ObjCProtocolDecl *> RefedProtocols; 7055a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis for (ObjCInterfaceDecl::protocol_iterator I = CDecl->protocol_begin(), 7056a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis E = CDecl->protocol_end(); 7057a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis 7058a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis I != E; ++I) { 70596118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RefedProtocols.push_back(*I); 70606118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Must write out all protocol definitions in current qualifier list, 70616118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // and in their nested qualifiers before writing out current definition. 70626118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RewriteObjCProtocolMetaData(*I, Result); 706364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 706464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 70656118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_protocol_list_initializer(Context, Result, 70666118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RefedProtocols, 70676118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_CATEGORY_PROTOCOLS_$_", 70686118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName); 70696118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 70706118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Protocol's property metadata. 70716118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian std::vector<ObjCPropertyDecl *> ClassProperties; 70726118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian for (ObjCContainerDecl::prop_iterator I = CDecl->prop_begin(), 70736118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian E = CDecl->prop_end(); I != E; ++I) 70746118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassProperties.push_back(*I); 70756118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 70766118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_prop_list_t_initializer(*this, Context, Result, ClassProperties, 70776118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian /* Container */0, 70786118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_$_PROP_LIST_", 70796118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName); 70806118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 70816118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_category_t(*this, Context, Result, 7082e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian CDecl, 708388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian ClassDecl, 70846118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian InstanceMethods, 70856118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassMethods, 70866118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RefedProtocols, 70876118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassProperties); 70886118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 708988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian // Determine if this category is also "non-lazy". 709088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (ImplementationIsNonLazy(IDecl)) 709188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian DefinedNonLazyCategories.push_back(CDecl); 7092e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian 7093e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian} 7094e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian 7095e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanianvoid RewriteModernObjC::RewriteCategorySetupInitHook(std::string &Result) { 7096e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian int CatDefCount = CategoryImplementation.size(); 7097e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian if (!CatDefCount) 7098e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian return; 7099e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "#pragma section(\".objc_inithooks$B\", long, read, write)\n"; 7100e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "__declspec(allocate(\".objc_inithooks$B\")) "; 7101e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "static void *OBJC_CATEGORY_SETUP[] = {\n"; 7102e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian for (int i = 0; i < CatDefCount; i++) { 7103e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian ObjCCategoryImplDecl *IDecl = CategoryImplementation[i]; 7104e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian ObjCCategoryDecl *CatDecl= IDecl->getCategoryDecl(); 7105e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian ObjCInterfaceDecl *ClassDecl = IDecl->getClassInterface(); 7106e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "\t(void *)&OBJC_CATEGORY_SETUP_$_"; 7107e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += ClassDecl->getName(); 7108e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "_$_"; 7109e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += CatDecl->getName(); 7110e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += ",\n"; 7111e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian } 7112e0335786c17a6ca45966f66094146e9d8ba2d54aFariborz Jahanian Result += "};\n"; 711364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 711464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 711564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// RewriteObjCMethodsMetaData - Rewrite methods metadata for instance or 711664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// class methods. 711764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahaniantemplate<typename MethodIterator> 711864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCMethodsMetaData(MethodIterator MethodBegin, 711964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodIterator MethodEnd, 712064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsInstanceMethod, 712164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, 712264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef ClassName, 712364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 712464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MethodBegin == MethodEnd) return; 712564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 712664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!objc_impl_method) { 712764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_method { 712864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SEL _cmd; 712964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char *method_types; 713064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void *_imp; 713164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 713264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 713364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_method {\n"; 713464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tSEL _cmd;\n"; 713564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tchar *method_types;\n"; 713664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tvoid *_imp;\n"; 713764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 713864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 713964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_impl_method = true; 714064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 714164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 714264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's methods if needed 714364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 714464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct { 714564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method_list *next_method; 714664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int method_count; 714764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method method_list[]; 714864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 714964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 715064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned NumMethods = std::distance(MethodBegin, MethodEnd); 71511ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "\n"; 71521ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) { 71531ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (IsInstanceMethod) 71541ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".inst_meth$B\")) "; 71551ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian else 71561ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".cls_meth$B\")) "; 71571ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian } 71581ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "static struct {\n"; 715964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method_list *next_method;\n"; 716064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tint method_count;\n"; 716164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method method_list["; 716264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(NumMethods); 716364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "];\n} _OBJC_"; 716464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += prefix; 716564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IsInstanceMethod ? "INSTANCE" : "CLASS"; 716664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_METHODS_"; 716764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassName; 716864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __"; 716964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IsInstanceMethod ? "inst" : "cls"; 717064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_meth\")))= "; 717164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "{\n\t0, " + utostr(NumMethods) + "\n"; 717264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 717364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,{{(SEL)\""; 717464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*MethodBegin)->getSelector().getAsString().c_str(); 717564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string MethodTypeString; 717664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString); 717764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 717864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodTypeString; 717964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", (void *)"; 718064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodInternalNames[*MethodBegin]; 718164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 718264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (++MethodBegin; MethodBegin != MethodEnd; ++MethodBegin) { 718364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t ,{(SEL)\""; 718464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*MethodBegin)->getSelector().getAsString().c_str(); 718564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string MethodTypeString; 718664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString); 718764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 718864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodTypeString; 718964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", (void *)"; 719064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodInternalNames[*MethodBegin]; 719164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 719264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 719364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t }\n};\n"; 719464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 719564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 719664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV) { 719764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = IV->getSourceRange(); 719864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *BaseExpr = IV->getBase(); 719964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 720064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the base, but without actually doing replaces. 720164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 720264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 720364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BaseExpr = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(BaseExpr)); 720464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->setBase(BaseExpr); 720564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 720664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 720764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCIvarDecl *D = IV->getDecl(); 720864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 720964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Replacement = IV; 7210e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian 721164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BaseExpr->getType()->isObjCObjectPointerType()) { 721264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceType *iFaceDecl = 721364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCInterfaceType>(BaseExpr->getType()->getPointeeType()); 721464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(iFaceDecl && "RewriteObjCIvarRefExpr - iFaceDecl is null"); 721564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // lookup which class implements the instance variable. 721664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *clsDeclared = 0; 721764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian iFaceDecl->getDecl()->lookupInstanceVariable(D->getIdentifier(), 721864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian clsDeclared); 721964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(clsDeclared && "RewriteObjCIvarRefExpr(): Can't find class"); 722064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 7221e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian // Build name of symbol holding ivar offset. 72227cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian std::string IvarOffsetName; 72237cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(clsDeclared, D, IvarOffsetName); 72247cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian 722572c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian ReferencedIvars[clsDeclared].insert(D); 722672c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 7227e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian // cast offset to "char *". 7228e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, 7229e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Context->getPointerType(Context->CharTy), 723064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 7231e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian BaseExpr); 7232e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 7233e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation(), &Context->Idents.get(IvarOffsetName), 7234e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Context->UnsignedLongTy, 0, SC_Extern, SC_None); 7235f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(NewVD, false, 7236f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Context->UnsignedLongTy, VK_LValue, 7237e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation()); 7238e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian BinaryOperator *addExpr = 7239e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian new (Context) BinaryOperator(castExpr, DRE, BO_Add, 7240e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Context->getPointerType(Context->CharTy), 7241e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 724264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 7243e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), 7244e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation(), 7245e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian addExpr); 72460d6e22a38314fd6a9bdb0d83fcc5c5215641d5e9Fariborz Jahanian QualType IvarT = D->getType(); 72478e0913d4921e36ef82ecba565fd7b2c66165dd87Fariborz Jahanian convertObjCTypeToCStyleType(IvarT); 72480d6e22a38314fd6a9bdb0d83fcc5c5215641d5e9Fariborz Jahanian QualType castT = Context->getPointerType(IvarT); 724964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 7250e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian castExpr = NoTypeInfoCStyleCastExpr(Context, 7251e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian castT, 7252e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian CK_BitCast, 7253e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian PE); 72548e0913d4921e36ef82ecba565fd7b2c66165dd87Fariborz Jahanian Expr *Exp = new (Context) UnaryOperator(castExpr, UO_Deref, IvarT, 7255e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian VK_LValue, OK_Ordinary, 7256e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation()); 7257e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian PE = new (Context) ParenExpr(OldRange.getBegin(), 7258e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian OldRange.getEnd(), 7259e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Exp); 7260e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian 7261e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Replacement = PE; 726264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 726364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 7264e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian ReplaceStmtWithRange(IV, Replacement, OldRange); 7265e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian return Replacement; 726664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 7267