RewriteModernObjC.cpp revision fd4ce2c9d64032b42c037b2af3bf4ebeac5bbac8
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; 7664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned RewriteFailedDiag; 7764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ObjC string constant support. 7864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned NumObjCStringLiterals; 7964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *ConstantStringClassReference; 8064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *NSStringRecord; 8164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 8264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ObjC foreach break/continue generation support. 8364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int BcLabelCount; 8464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 8564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned TryFinallyContainsReturnDiag; 8664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Needed for super. 8764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *CurMethodDef; 8864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *SuperStructDecl; 8964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *ConstantStringDecl; 9064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 9164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendFunctionDecl; 9264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendSuperFunctionDecl; 9364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendStretFunctionDecl; 9464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendSuperStretFunctionDecl; 9564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendFpretFunctionDecl; 9664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *GetClassFunctionDecl; 9764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *GetMetaClassFunctionDecl; 9864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *GetSuperClassFunctionDecl; 9964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *SelGetUidFunctionDecl; 10064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *CFStringFunctionDecl; 10164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *SuperContructorFunctionDecl; 10264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *CurFunctionDef; 10364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *CurFunctionDeclToDeclareForBlock; 10464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 10564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* Misc. containers needed for meta-data rewrite. */ 10664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCImplementationDecl *, 8> ClassImplementation; 10764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCCategoryImplDecl *, 8> CategoryImplementation; 10864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCInterfaceDecl*, 8> ObjCSynthesizedStructs; 10964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCProtocolDecl*, 8> ObjCSynthesizedProtocols; 110cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian llvm::SmallPtrSet<ObjCInterfaceDecl*, 8> ObjCWrittenInterfaces; 11115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian llvm::SmallPtrSet<TagDecl*, 8> TagsDefinedInIvarDecls; 112cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian SmallVector<ObjCInterfaceDecl*, 32> ObjCInterfacesSeen; 11388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian /// DefinedNonLazyClasses - List of defined "non-lazy" classes. 11488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian SmallVector<ObjCInterfaceDecl*, 8> DefinedNonLazyClasses; 11588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 11688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian /// DefinedNonLazyCategories - List of defined "non-lazy" categories. 11788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian llvm::SmallVector<ObjCCategoryDecl*, 8> DefinedNonLazyCategories; 11888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 11964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Stmt *, 32> Stmts; 12064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<int, 8> ObjCBcLabelNo; 12164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remember all the @protocol(<expr>) expressions. 12264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCProtocolDecl *, 32> ProtocolExprDecls; 12364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 12464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseSet<uint64_t> CopyDestroyCache; 12564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 12664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block expressions. 12764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockExpr *, 32> Blocks; 12864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<int, 32> InnerDeclRefsCount; 129f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 32> InnerDeclRefs; 13064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 131f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 32> BlockDeclRefs; 13264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 13364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block related declarations. 13464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ValueDecl *, 8> BlockByCopyDecls; 13564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> BlockByCopyDeclsPtrSet; 13664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ValueDecl *, 8> BlockByRefDecls; 13764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> BlockByRefDeclsPtrSet; 13864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<ValueDecl *, unsigned> BlockByRefDeclNo; 13964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> ImportedBlockDecls; 14064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<VarDecl *, 8> ImportedLocalExternalDecls; 14164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 14264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<BlockExpr *, std::string> RewrittenBlockExprs; 14372c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian llvm::DenseMap<ObjCInterfaceDecl *, 14472c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian llvm::SmallPtrSet<ObjCIvarDecl *, 8> > ReferencedIvars; 14572c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 14664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This maps an original source AST to it's rewritten form. This allows 14764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // us to avoid rewriting the same node twice (which is very uncommon). 14864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is needed to support some of the exotic property rewriting. 14964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<Stmt *, Stmt *> ReplacedNodes; 15064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 15164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Needed for header files being rewritten 15264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsHeader; 15364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SilenceRewriteMacroWarning; 15464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool objc_impl_method; 15564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 15664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool DisableReplaceStmt; 15764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian class DisableReplaceStmtScope { 15864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteModernObjC &R; 15964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SavedValue; 16064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 16164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian public: 16264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope(RewriteModernObjC &R) 16364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : R(R), SavedValue(R.DisableReplaceStmt) { 16464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian R.DisableReplaceStmt = true; 16564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 16664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ~DisableReplaceStmtScope() { 16764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian R.DisableReplaceStmt = SavedValue; 16864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 16964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 17064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InitializeCommon(ASTContext &context); 17164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 17264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian public: 17390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian llvm::DenseMap<ObjCMethodDecl*, std::string> MethodInternalNames; 17464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Top Level Driver code. 17564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual bool HandleTopLevelDecl(DeclGroupRef D) { 17664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclGroupRef::iterator I = D.begin(), E = D.end(); I != E; ++I) { 17764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*I)) { 17864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Class->isThisDeclarationADefinition()) { 17964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassDecl(D); 18064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 181cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian } else { 182cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian // Keep track of all interface declarations seen. 183f329527bbeae86c8eac6eea6efaa20e807c16679Fariborz Jahanian ObjCInterfacesSeen.push_back(Class); 184cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian break; 18564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 18864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*I)) { 18964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Proto->isThisDeclarationADefinition()) { 19064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardProtocolDecl(D); 19164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 19264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 19364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 19464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 19564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HandleTopLevelSingleDecl(*I); 19664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 19764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 19864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 19964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void HandleTopLevelSingleDecl(Decl *D); 20064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void HandleDeclInMainFile(Decl *D); 20164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteModernObjC(std::string inFile, raw_ostream *OS, 20264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &D, const LangOptions &LOpts, 20364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool silenceMacroWarn); 20464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 20564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ~RewriteModernObjC() {} 20664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 20764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void HandleTranslationUnit(ASTContext &C); 20864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 20964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceStmt(Stmt *Old, Stmt *New) { 21064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = ReplacedNodes[Old]; 21164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ReplacingStmt) 21364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // We can't rewrite the same node twice. 21464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DisableReplaceStmt) 21664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 21764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If replacement succeeded or warning disabled return with no warning. 21964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceStmt(Old, New)) { 22064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacedNodes[Old] = New; 22164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 22264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 22364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SilenceRewriteMacroWarning) 22464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 22564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 22664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 22764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 22864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 22964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceStmtWithRange(Stmt *Old, Stmt *New, SourceRange SrcRange) { 23064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DisableReplaceStmt) 23164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 23264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 23364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Measure the old text. 23464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int Size = Rewrite.getRangeSize(SrcRange); 23564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Size == -1) { 23664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 23764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 23864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 23964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 24064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the new text. 24164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SStr; 24264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream S(SStr); 24364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian New->printPretty(S, *Context, 0, PrintingPolicy(LangOpts)); 24464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Str = S.str(); 24564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 24664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If replacement succeeded or warning disabled return with no warning. 24764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceText(SrcRange.getBegin(), Size, Str)) { 24864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacedNodes[Old] = New; 24964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 25064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 25164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SilenceRewriteMacroWarning) 25264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 25364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 25464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 25564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 25664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 25764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertText(SourceLocation Loc, StringRef Str, 25864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool InsertAfter = true) { 25964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If insertion succeeded or warning disabled return with no warning. 26064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.InsertText(Loc, Str, InsertAfter) || 26164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning) 26264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 26364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 26464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag); 26564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 26664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 26764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceText(SourceLocation Start, unsigned OrigLength, 26864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef Str) { 26964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If removal succeeded or warning disabled return with no warning. 27064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceText(Start, OrigLength, Str) || 27164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning) 27264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 27364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 27464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Start), RewriteFailedDiag); 27564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 27664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 27764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Syntactic Rewriting. 27864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteRecordBody(RecordDecl *RD); 27964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteInclude(); 28064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassDecl(DeclGroupRef D); 28164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassDecl(const llvm::SmallVector<Decl*, 8> &DG); 28264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassEpilogue(ObjCInterfaceDecl *ClassDecl, 28364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &typedefString); 28464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteImplementations(); 28564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewritePropertyImplDecl(ObjCPropertyImplDecl *PID, 28664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD, 28764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID); 28864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteInterfaceDecl(ObjCInterfaceDecl *Dcl); 28964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteImplementationDecl(Decl *Dcl); 29064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCMethodDecl(const ObjCInterfaceDecl *IDecl, 29164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MDecl, std::string &ResultStr); 29264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteTypeIntoString(QualType T, std::string &ResultStr, 29364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *&FPRetType); 29464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteByRefString(std::string &ResultStr, const std::string &Name, 29564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD, bool def=false); 29664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteCategoryDecl(ObjCCategoryDecl *Dcl); 29764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteProtocolDecl(ObjCProtocolDecl *Dcl); 29864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardProtocolDecl(DeclGroupRef D); 29964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardProtocolDecl(const llvm::SmallVector<Decl*, 8> &DG); 30064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteMethodDeclaration(ObjCMethodDecl *Method); 30164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteProperty(ObjCPropertyDecl *prop); 30264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteFunctionDecl(FunctionDecl *FD); 30364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerType(std::string& Str, QualType Type); 30464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerTypeVariable(std::string& Str, ValueDecl *VD); 30564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockLiteralFunctionDecl(FunctionDecl *FD); 30664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCQualifiedInterfaceTypes(Decl *Dcl); 30764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteTypeOfDecl(VarDecl *VD); 30864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCQualifiedInterfaceTypes(Expr *E); 30964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 31064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Expression Rewriting. 31164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteFunctionBodyOrGlobalInitializer(Stmt *S); 31264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteAtEncode(ObjCEncodeExpr *Exp); 31364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewritePropertyOrImplicitGetter(PseudoObjectExpr *Pseudo); 31464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewritePropertyOrImplicitSetter(PseudoObjectExpr *Pseudo); 31564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteAtSelector(ObjCSelectorExpr *Exp); 31664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteMessageExpr(ObjCMessageExpr *Exp); 31764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCStringLiteral(ObjCStringLiteral *Exp); 31864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp); 31964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCTryStmt(ObjCAtTryStmt *S); 32064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S); 32164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCThrowStmt(ObjCAtThrowStmt *S); 32264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S, 32364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OrigEnd); 32464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteBreakStmt(BreakStmt *S); 32564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteContinueStmt(ContinueStmt *S); 32664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteCastExpr(CStyleCastExpr *CE); 32764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 32864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block rewriting. 32964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlocksInFunctionProtoType(QualType funcType, NamedDecl *D); 33064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 33164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block specific rewrite rules. 33264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerDecl(NamedDecl *VD); 33364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteByRefVar(VarDecl *VD); 334f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Stmt *RewriteBlockDeclRefExpr(DeclRefExpr *VD); 33564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteLocalVariableExternalStorage(DeclRefExpr *DRE); 33664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerFunctionArgs(FunctionDecl *FD); 33764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 33864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCInternalStruct(ObjCInterfaceDecl *CDecl, 33964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 34064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 34115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian void RewriteObjCFieldDecl(FieldDecl *fieldDecl, std::string &Result); 34215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 34397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian bool RewriteObjCFieldDeclType(QualType &Type, std::string &Result); 34497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 34572c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian void RewriteIvarOffsetSymbols(ObjCInterfaceDecl *CDecl, 34672c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian std::string &Result); 34772c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 34864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void Initialize(ASTContext &context); 34964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 35064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Misc. AST transformation routines. Somtimes they end up calling 35164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // rewriting routines on the new ASTs. 35264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SynthesizeCallToFunctionDecl(FunctionDecl *FD, 35364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr **args, unsigned nargs, 35464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc=SourceLocation(), 35564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc=SourceLocation()); 35664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 35764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthMessageExpr(ObjCMessageExpr *Exp, 35864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc=SourceLocation(), 35964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc=SourceLocation()); 36064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 36164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthCountByEnumWithState(std::string &buf); 36264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendFunctionDecl(); 36364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendSuperFunctionDecl(); 36464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendStretFunctionDecl(); 36564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendFpretFunctionDecl(); 36664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendSuperStretFunctionDecl(); 36764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetClassFunctionDecl(); 36864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetMetaClassFunctionDecl(); 36964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetSuperClassFunctionDecl(); 37064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthSelGetUidFunctionDecl(); 37164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthSuperContructorFunctionDecl(); 37264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 37364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewriting metadata 37464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian template<typename MethodIterator> 37564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCMethodsMetaData(MethodIterator MethodBegin, 37664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodIterator MethodEnd, 37764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsInstanceMethod, 37864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, 37964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef ClassName, 38064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 381da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian void RewriteObjCProtocolMetaData(ObjCProtocolDecl *Protocol, 382da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::string &Result); 38364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCProtocolListMetaData( 38464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Prots, 38564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, StringRef ClassName, std::string &Result); 38664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl, 38764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 38864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteMetaDataIntoBuffer(std::string &Result); 38910cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian virtual void WriteImageInfo(std::string &Result); 39064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *CDecl, 39164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 39264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 39364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewriting ivar 39464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteIvarOffsetComputation(ObjCIvarDecl *ivar, 39564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 39664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual Stmt *RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV); 39764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 39864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 39964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeByrefCopyDestroyHelper(VarDecl *VD, int flag); 40064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockHelperFuncs(BlockExpr *CE, int i, 40164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, std::string Tag); 40264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockFunc(BlockExpr *CE, int i, 40364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, std::string Tag); 40464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockImpl(BlockExpr *CE, 40564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag, std::string Desc); 40664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockDescriptor(std::string DescTag, 40764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag, 40864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int i, StringRef funcName, 40964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned hasCopy); 41064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthesizeBlockCall(CallExpr *Exp, const Expr* BlockExp); 41164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthesizeBlockLiterals(SourceLocation FunLocStart, 41264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName); 41364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *SynthBlockInitFunctionDecl(StringRef name); 41464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthBlockInitExpr(BlockExpr *Exp, 415f4b88a45902af1802a1cb42ba48b1c474474f228John McCall const SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs); 41664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 41764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Misc. helper routines. 41864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getProtocolType(); 41964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void WarnAboutReturnGotoStmts(Stmt *S); 42064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void CheckFunctionPointerDecl(QualType dType, NamedDecl *ND); 42164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertBlockLiteralsWithinFunction(FunctionDecl *FD); 42264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertBlockLiteralsWithinMethod(ObjCMethodDecl *MD); 42364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 42464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsDeclStmtInForeachHeader(DeclStmt *DS); 42564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void CollectBlockDeclRefInfo(BlockExpr *Exp); 42664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetBlockDeclRefExprs(Stmt *S); 42764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetInnerBlockDeclRefExprs(Stmt *S, 428f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs, 42964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> &InnerContexts); 43064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 43164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We avoid calling Type::isBlockPointerType(), since it operates on the 43264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // canonical type. We only care if the top-level type is a closure pointer. 43364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isTopLevelBlockPointerType(QualType T) { 43464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return isa<BlockPointerType>(T); 43564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 43664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 43764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// convertBlockPointerToFunctionPointer - Converts a block-pointer type 43864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// to a function pointer type and upon success, returns true; false 43964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// otherwise. 44064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool convertBlockPointerToFunctionPointer(QualType &T) { 44164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(T)) { 44264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = T->getAs<BlockPointerType>(); 44364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getPointerType(BPT->getPointeeType()); 44464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 44564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 44664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 44764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 44864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 449164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian bool convertObjCTypeToCStyleType(QualType &T); 450164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian 45164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool needToScanForQualifiers(QualType T); 45264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getSuperStructType(); 45364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getConstantStringStructType(); 45464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType convertFunctionTypeOfBlocks(const FunctionType *FT); 45564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool BufferContainsPPDirectives(const char *startBuf, const char *endBuf); 45664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 45764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void convertToUnqualifiedObjCType(QualType &T) { 45864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 45964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getObjCIdType(); 46064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isObjCQualifiedClassType()) 46164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getObjCClassType(); 46264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isObjCObjectPointerType() && 46364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->getPointeeType()->isObjCQualifiedInterfaceType()) { 46464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const ObjCObjectPointerType * OBJPT = 46564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->getAsObjCInterfacePointerType()) { 46664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceType *IFaceT = OBJPT->getInterfaceType(); 46764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = QualType(IFaceT, 0); 46864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getPointerType(T); 46964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 47064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 47164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 47264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 47364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: This predicate seems like it would be useful to add to ASTContext. 47464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isObjCType(QualType T) { 47564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!LangOpts.ObjC1 && !LangOpts.ObjC2) 47664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 47764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 47864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType OCT = Context->getCanonicalType(T).getUnqualifiedType(); 47964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 48064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OCT == Context->getCanonicalType(Context->getObjCIdType()) || 48164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OCT == Context->getCanonicalType(Context->getObjCClassType())) 48264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 48364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 48464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType *PT = OCT->getAs<PointerType>()) { 48564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCInterfaceType>(PT->getPointeeType()) || 48664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PT->getPointeeType()->isObjCQualifiedIdType()) 48764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 48864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 48964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 49064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 49164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool PointerTypeTakesAnyBlockArguments(QualType QT); 49264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool PointerTypeTakesAnyObjCQualifiedType(QualType QT); 49364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetExtentOfArgList(const char *Name, const char *&LParen, 49464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&RParen); 49564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 49664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void QuoteDoublequotes(std::string &From, std::string &To) { 49764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < From.length(); i++) { 49864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (From[i] == '"') 49964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian To += "\\\""; 50064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 50164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian To += From[i]; 50264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 50364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 50464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 50564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getSimpleFunctionType(QualType result, 50664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const QualType *args, 50764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned numArgs, 50864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool variadic = false) { 50964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (result == Context->getObjCInstanceType()) 51064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian result = Context->getObjCIdType(); 51164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionProtoType::ExtProtoInfo fpi; 51264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian fpi.Variadic = variadic; 51364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getFunctionType(result, args, numArgs, fpi); 51464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 51564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 51664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Helper function: create a CStyleCastExpr with trivial type source info. 51764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CStyleCastExpr* NoTypeInfoCStyleCastExpr(ASTContext *Ctx, QualType Ty, 51864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind Kind, Expr *E) { 51964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *TInfo = Ctx->getTrivialTypeSourceInfo(Ty, SourceLocation()); 52064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CStyleCastExpr::Create(*Ctx, Ty, VK_RValue, Kind, E, 0, TInfo, 52164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation()); 52264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 52388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 52488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian bool ImplementationIsNonLazy(const ObjCImplDecl *OD) const { 52588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian IdentifierInfo* II = &Context->Idents.get("load"); 52688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Selector LoadSel = Context->Selectors.getSelector(0, &II); 52788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian return OD->getClassMethod(LoadSel) != 0; 52888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 52964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 53064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 53164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 53264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 53364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlocksInFunctionProtoType(QualType funcType, 53464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NamedDecl *D) { 53564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *fproto 53664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = dyn_cast<FunctionProtoType>(funcType.IgnoreParens())) { 53764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = fproto->arg_type_begin(), 53864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = fproto->arg_type_end(); I && (I != E); ++I) 53964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(*I)) { 54064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // All the args are checked/rewritten. Don't call twice! 54164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(D); 54264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 54364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 54464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 54564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 54664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 54764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::CheckFunctionPointerDecl(QualType funcType, NamedDecl *ND) { 54864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = funcType->getAs<PointerType>(); 54964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT && PointerTypeTakesAnyBlockArguments(funcType)) 55064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlocksInFunctionProtoType(PT->getPointeeType(), ND); 55164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 55264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 55364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool IsHeaderFile(const std::string &Filename) { 55464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string::size_type DotPos = Filename.rfind('.'); 55564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 55664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DotPos == std::string::npos) { 55764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // no file extension 55864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 55964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 56064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 56164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Ext = std::string(Filename.begin()+DotPos+1, Filename.end()); 56264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // C header: .h 56364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // C++ header: .hh or .H; 56464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Ext == "h" || Ext == "hh" || Ext == "H"; 56564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 56664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 56764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianRewriteModernObjC::RewriteModernObjC(std::string inFile, raw_ostream* OS, 56864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &D, const LangOptions &LOpts, 56964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool silenceMacroWarn) 57064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : Diags(D), LangOpts(LOpts), InFileName(inFile), OutFile(OS), 57164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning(silenceMacroWarn) { 57264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IsHeader = IsHeaderFile(inFile); 57364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFailedDiag = Diags.getCustomDiagID(DiagnosticsEngine::Warning, 57464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "rewriting sub-expression within a macro (may not be correct)"); 57564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TryFinallyContainsReturnDiag = Diags.getCustomDiagID( 57664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine::Warning, 57764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "rewriter doesn't support user-specified control flow semantics " 57864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "for @try/@finally (code may not execute properly)"); 57964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 58064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 58164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianASTConsumer *clang::CreateModernObjCRewriter(const std::string& InFile, 58264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian raw_ostream* OS, 58364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &Diags, 58464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const LangOptions &LOpts, 58564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SilenceRewriteMacroWarning) { 58664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return new RewriteModernObjC(InFile, OS, Diags, LOpts, SilenceRewriteMacroWarning); 58764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 58864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 58964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InitializeCommon(ASTContext &context) { 59064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context = &context; 59164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SM = &Context->getSourceManager(); 59264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TUDecl = Context->getTranslationUnitDecl(); 59364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFunctionDecl = 0; 59464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperFunctionDecl = 0; 59564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFunctionDecl = 0; 59664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperStretFunctionDecl = 0; 59764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFpretFunctionDecl = 0; 59864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetClassFunctionDecl = 0; 59964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetMetaClassFunctionDecl = 0; 60064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetSuperClassFunctionDecl = 0; 60164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = 0; 60264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CFStringFunctionDecl = 0; 60364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringClassReference = 0; 60464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NSStringRecord = 0; 60564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = 0; 60664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = 0; 60764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = 0; 60864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = 0; 60964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl = 0; 61064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolTypeDecl = 0; 61164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl = 0; 61264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BcLabelCount = 0; 61364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperContructorFunctionDecl = 0; 61464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NumObjCStringLiterals = 0; 61564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 61664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 61764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = false; 61864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_impl_method = false; 61964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 62064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the ID and start/end of the main file. 62164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileID = SM->getMainFileID(); 62264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const llvm::MemoryBuffer *MainBuf = SM->getBuffer(MainFileID); 62364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileStart = MainBuf->getBufferStart(); 62464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileEnd = MainBuf->getBufferEnd(); 62564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6264e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie Rewrite.setSourceMgr(Context->getSourceManager(), Context->getLangOpts()); 62764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 62864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 62964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 63064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Top Level Driver Code 63164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 63264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleTopLevelSingleDecl(Decl *D) { 63464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Diags.hasErrorOccurred()) 63564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 63664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Two cases: either the decl could be in the main file, or it could be in a 63864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // #included file. If the former, rewrite it now. If the later, check to see 63964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // if we rewrote the #include/#import. 64064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc = D->getLocation(); 64164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = SM->getExpansionLoc(Loc); 64264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 64364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If this is for a builtin, ignore it. 64464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Loc.isInvalid()) return; 64564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 64664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Look for built-in declarations that we need to refer during the rewrite. 64764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { 64864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionDecl(FD); 64964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (VarDecl *FVD = dyn_cast<VarDecl>(D)) { 65064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declared in <Foundation/NSString.h> 65164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FVD->getName() == "_NSConstantStringClassReference") { 65264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringClassReference = FVD; 65364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 65464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 65564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(D)) { 65664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCategoryDecl(CD); 65764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCProtocolDecl *PD = dyn_cast<ObjCProtocolDecl>(D)) { 65864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isThisDeclarationADefinition()) 65964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProtocolDecl(PD); 66064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (LinkageSpecDecl *LSD = dyn_cast<LinkageSpecDecl>(D)) { 66164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Recurse into linkage specifications 66264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclContext::decl_iterator DI = LSD->decls_begin(), 66364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DIEnd = LSD->decls_end(); 66464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DI != DIEnd; ) { 66564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>((*DI))) { 66664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!IFace->isThisDeclarationADefinition()) { 66764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Decl *, 8> DG; 66864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc = IFace->getLocStart(); 66964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian do { 67064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCInterfaceDecl>(*DI) && 67164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !cast<ObjCInterfaceDecl>(*DI)->isThisDeclarationADefinition() && 67264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc == (*DI)->getLocStart()) 67364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DG.push_back(*DI); 67464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 67564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 67664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 67764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 67864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } while (DI != DIEnd); 67964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassDecl(DG); 68064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 68164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 68264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 68364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 68464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>((*DI))) { 68564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Proto->isThisDeclarationADefinition()) { 68664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Decl *, 8> DG; 68764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc = Proto->getLocStart(); 68864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian do { 68964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCProtocolDecl>(*DI) && 69064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !cast<ObjCProtocolDecl>(*DI)->isThisDeclarationADefinition() && 69164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc == (*DI)->getLocStart()) 69264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DG.push_back(*DI); 69364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 69464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 69564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 69664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 69764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } while (DI != DIEnd); 69864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardProtocolDecl(DG); 69964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 70064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 70164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 70264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 70364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HandleTopLevelSingleDecl(*DI); 70464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 70564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 70664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 70764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we have a decl in the main file, see if we should rewrite it. 70864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SM->isFromMainFile(Loc)) 70964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return HandleDeclInMainFile(D); 71064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 71164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 71264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 71364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Syntactic (non-AST) Rewriting Code 71464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 71564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 71664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteInclude() { 71764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = SM->getLocForStartOfFile(MainFileID); 71864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef MainBuf = SM->getBufferData(MainFileID); 71964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *MainBufStart = MainBuf.begin(); 72064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *MainBufEnd = MainBuf.end(); 72164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian size_t ImportLen = strlen("import"); 72264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 72364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Loop over the whole file, looking for includes. 72464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (const char *BufPtr = MainBufStart; BufPtr < MainBufEnd; ++BufPtr) { 72564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*BufPtr == '#') { 72664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (++BufPtr == MainBufEnd) 72764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 72864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*BufPtr == ' ' || *BufPtr == '\t') 72964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (++BufPtr == MainBufEnd) 73064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 73164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strncmp(BufPtr, "import", ImportLen)) { 73264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace import with include 73364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation ImportLoc = 73464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LocStart.getLocWithOffset(BufPtr-MainBufStart); 73564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(ImportLoc, ImportLen, "include"); 73664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BufPtr += ImportLen; 73764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 73864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 73964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 74064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 74164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 74264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic std::string getIvarAccessString(ObjCIvarDecl *OID) { 74364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceDecl *ClassDecl = OID->getContainingInterface(); 74464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S; 74564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S = "((struct "; 74664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ClassDecl->getIdentifier()->getName(); 74764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_IMPL *)self)->"; 74864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += OID->getName(); 74964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 75064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 75164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 75264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewritePropertyImplDecl(ObjCPropertyImplDecl *PID, 75364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD, 75464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID) { 75564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objcGetPropertyDefined = false; 75664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objcSetPropertyDefined = false; 75764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = PID->getLocStart(); 75864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(startLoc, "// "); 75964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 76064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @synthesize location"); 76164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 76264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "@synthesize: can't find ';'"); 76364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation onePastSemiLoc = 76464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc.getLocWithOffset(semiBuf-startBuf+1); 76564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 76664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 76764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // FIXME: is this correct? 76864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 76964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the 'getter' function. 77064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl *PD = PID->getPropertyDecl(); 77164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCIvarDecl *OID = PID->getPropertyIvarDecl(); 77264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 77364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!OID) 77464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 77564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned Attributes = PD->getPropertyAttributes(); 77664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!PD->getGetterMethodDecl()->isDefined()) { 77764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool GenGetProperty = !(Attributes & ObjCPropertyDecl::OBJC_PR_nonatomic) && 77864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (Attributes & (ObjCPropertyDecl::OBJC_PR_retain | 77964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl::OBJC_PR_copy)); 78064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Getr; 78164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenGetProperty && !objcGetPropertyDefined) { 78264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objcGetPropertyDefined = true; 78364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Is this attribute correct in all cases? 78464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr = "\nextern \"C\" __declspec(dllimport) " 78564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "id objc_getProperty(id, SEL, long, bool);\n"; 78664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 78764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OID->getContainingInterface(), 78864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PD->getGetterMethodDecl(), Getr); 78964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "{ "; 79064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to gain access to the ivar. 79164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See objc-act.c:objc_synthesize_new_getter() for details. 79264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenGetProperty) { 79364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // return objc_getProperty(self, _cmd, offsetof(ClassDecl, OID), 1) 79464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "typedef "; 79564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FPRetType = 0; 79664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeIntoString(PD->getGetterMethodDecl()->getResultType(), Getr, 79764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FPRetType); 79864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += " _TYPE"; 79964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FPRetType) { 80064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ")"; // close the precedence "scope" for "*". 80164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 80264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, emit the argument types (if any). 80364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FPRetType)){ 80464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "("; 80564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) { 80664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i) Getr += ", "; 80764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr = FT->getArgType(i).getAsString( 80864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 80964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ParamStr; 81064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 81164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 81264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->getNumArgs()) Getr += ", "; 81364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "..."; 81464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 81564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ")"; 81664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 81764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "()"; 81864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 81964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ";\n"; 82064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "return (_TYPE)"; 82164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "objc_getProperty(self, _cmd, "; 82264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation(OID, Getr); 82364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ", 1)"; 82464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 82564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 82664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "return " + getIvarAccessString(OID); 82764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "; }"; 82864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, Getr); 82964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 83064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 83164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isReadOnly() || PD->getSetterMethodDecl()->isDefined()) 83264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 83364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 83464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the 'setter' function. 83564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Setr; 83664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool GenSetProperty = Attributes & (ObjCPropertyDecl::OBJC_PR_retain | 83764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl::OBJC_PR_copy); 83864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenSetProperty && !objcSetPropertyDefined) { 83964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objcSetPropertyDefined = true; 84064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Is this attribute correct in all cases? 84164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr = "\nextern \"C\" __declspec(dllimport) " 84264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "void objc_setProperty (id, SEL, long, id, bool, bool);\n"; 84364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 84464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 84564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OID->getContainingInterface(), 84664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PD->getSetterMethodDecl(), Setr); 84764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "{ "; 84864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to initialize the ivar. 84964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See objc-act.c:objc_synthesize_new_setter() for details. 85064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenSetProperty) { 85164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "objc_setProperty (self, _cmd, "; 85264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation(OID, Setr); 85364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += ", (id)"; 85464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += PD->getName(); 85564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += ", "; 85664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Attributes & ObjCPropertyDecl::OBJC_PR_nonatomic) 85764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "0, "; 85864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 85964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "1, "; 86064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Attributes & ObjCPropertyDecl::OBJC_PR_copy) 86164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "1)"; 86264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 86364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "0)"; 86464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 86564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 86664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += getIvarAccessString(OID) + " = "; 86764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += PD->getName(); 86864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 86964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "; }"; 87064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, Setr); 87164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 87264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 87364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void RewriteOneForwardClassDecl(ObjCInterfaceDecl *ForwardDecl, 87464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &typedefString) { 87564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "#ifndef _REWRITER_typedef_"; 87664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 87764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "\n"; 87864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "#define _REWRITER_typedef_"; 87964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 88064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "\n"; 88164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "typedef struct objc_object "; 88264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 883c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian // typedef struct { } _objc_exc_Classname; 884c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian typedefString += ";\ntypedef struct {} _objc_exc_"; 885c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 88664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n#endif\n"; 88764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 88864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 88964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassEpilogue(ObjCInterfaceDecl *ClassDecl, 89064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &typedefString) { 89164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = ClassDecl->getLocStart(); 89264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 89364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiPtr = strchr(startBuf, ';'); 89464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the @class with typedefs corresponding to the classes. 89564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, semiPtr-startBuf+1, typedefString); 89664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 89764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 89864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassDecl(DeclGroupRef D) { 89964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string typedefString; 90064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclGroupRef::iterator I = D.begin(), E = D.end(); I != E; ++I) { 90164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ForwardDecl = cast<ObjCInterfaceDecl>(*I); 90264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (I == D.begin()) { 90364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Translate to typedef's that forward reference structs with the same name 90464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as the class. As a convenience, we include the original declaration 90564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as a comment. 90664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "// @class "; 90764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 90864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n"; 90964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 91064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteOneForwardClassDecl(ForwardDecl, typedefString); 91164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 91264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclGroupRef::iterator I = D.begin(); 91364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassEpilogue(cast<ObjCInterfaceDecl>(*I), typedefString); 91464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 91564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 91664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassDecl( 91764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const llvm::SmallVector<Decl*, 8> &D) { 91864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string typedefString; 91964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < D.size(); i++) { 92064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ForwardDecl = cast<ObjCInterfaceDecl>(D[i]); 92164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i == 0) { 92264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "// @class "; 92364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 92464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n"; 92564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 92664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteOneForwardClassDecl(ForwardDecl, typedefString); 92764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 92864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassEpilogue(cast<ObjCInterfaceDecl>(D[0]), typedefString); 92964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 93064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 93164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteMethodDeclaration(ObjCMethodDecl *Method) { 93264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // When method is a synthesized one, such as a getter/setter there is 93364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // nothing to rewrite. 93464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Method->isImplicit()) 93564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 93664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = Method->getLocStart(); 93764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = Method->getLocEnd(); 93864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 93964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SM->getExpansionLineNumber(LocEnd) > 94064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SM->getExpansionLineNumber(LocStart)) { 94164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LocStart, "#if 0\n"); 94264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocEnd, 1, ";\n#endif\n"); 94364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 94464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LocStart, "// "); 94564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 94664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 94764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 94864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteProperty(ObjCPropertyDecl *prop) { 94964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc = prop->getAtLoc(); 95064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 95164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(Loc, 0, "// "); 95264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle properties that are declared across multiple lines. 95364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 95464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 95564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteCategoryDecl(ObjCCategoryDecl *CatDecl) { 95664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CatDecl->getLocStart(); 95764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 95864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle category headers that are declared across multiple lines. 95964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 960af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (CatDecl->getIvarLBraceLoc().isValid()) 961af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(CatDecl->getIvarLBraceLoc(), "// "); 962d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian for (ObjCCategoryDecl::ivar_iterator 963d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian I = CatDecl->ivar_begin(), E = CatDecl->ivar_end(); I != E; ++I) { 964d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian ObjCIvarDecl *Ivar = (*I); 965d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian SourceLocation LocStart = Ivar->getLocStart(); 966d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian ReplaceText(LocStart, 0, "// "); 967d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian } 968af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (CatDecl->getIvarRBraceLoc().isValid()) 969af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(CatDecl->getIvarRBraceLoc(), "// "); 970af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian 97164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::prop_iterator I = CatDecl->prop_begin(), 97264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = CatDecl->prop_end(); I != E; ++I) 97364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProperty(*I); 97464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 97564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::instmeth_iterator 97664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = CatDecl->instmeth_begin(), E = CatDecl->instmeth_end(); 97764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 97864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 97964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::classmeth_iterator 98064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = CatDecl->classmeth_begin(), E = CatDecl->classmeth_end(); 98164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 98264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 98364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 98464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Lastly, comment out the @end. 98564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(CatDecl->getAtEndRange().getBegin(), 98664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian strlen("@end"), "/* @end */"); 98764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 98864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 98964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) { 99064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = PDecl->getLocStart(); 99164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(PDecl->isThisDeclarationADefinition()); 99264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 99364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle protocol headers that are declared across multiple lines. 99464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 99564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 99664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCProtocolDecl::instmeth_iterator 99764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = PDecl->instmeth_begin(), E = PDecl->instmeth_end(); 99864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 99964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 100064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCProtocolDecl::classmeth_iterator 100164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = PDecl->classmeth_begin(), E = PDecl->classmeth_end(); 100264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 100364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 100464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 100564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCInterfaceDecl::prop_iterator I = PDecl->prop_begin(), 100664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = PDecl->prop_end(); I != E; ++I) 100764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProperty(*I); 100864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 100964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Lastly, comment out the @end. 101064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = PDecl->getAtEndRange().getBegin(); 101164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocEnd, strlen("@end"), "/* @end */"); 101264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 101364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Must comment out @optional/@required 101464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 101564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 101664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (const char *p = startBuf; p < endBuf; p++) { 101764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*p == '@' && !strncmp(p+1, "optional", strlen("optional"))) { 101864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OptionalLoc = LocStart.getLocWithOffset(p-startBuf); 101964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(OptionalLoc, strlen("@optional"), "/* @optional */"); 102064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 102164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 102264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*p == '@' && !strncmp(p+1, "required", strlen("required"))) { 102364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OptionalLoc = LocStart.getLocWithOffset(p-startBuf); 102464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(OptionalLoc, strlen("@required"), "/* @required */"); 102564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 102664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 102764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 102864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 102964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 103064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardProtocolDecl(DeclGroupRef D) { 103164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = (*D.begin())->getLocStart(); 103264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 103364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("Invalid SourceLocation"); 103464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle forward protocol that are declared across multiple lines. 103564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 103664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 103764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 103864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid 103964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianRewriteModernObjC::RewriteForwardProtocolDecl(const llvm::SmallVector<Decl*, 8> &DG) { 104064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = DG[0]->getLocStart(); 104164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 104264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("Invalid SourceLocation"); 104364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle forward protocol that are declared across multiple lines. 104464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 104564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 104664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 104764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteTypeIntoString(QualType T, std::string &ResultStr, 104864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *&FPRetType) { 104964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 105064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "id"; 105164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isFunctionPointerType() || 105264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->isBlockPointerType()) { 105364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // needs special handling, since pointer-to-functions have special 105464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // syntax (where a decaration models use). 105564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType retType = T; 105664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PointeeTy; 105764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType* PT = retType->getAs<PointerType>()) 105864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointeeTy = PT->getPointeeType(); 105964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const BlockPointerType *BPT = retType->getAs<BlockPointerType>()) 106064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointeeTy = BPT->getPointeeType(); 106164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((FPRetType = PointeeTy->getAs<FunctionType>())) { 106264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += FPRetType->getResultType().getAsString( 106364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 106464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "(*"; 106564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 106664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 106764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += T.getAsString(Context->getPrintingPolicy()); 106864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 106964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 107064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCMethodDecl(const ObjCInterfaceDecl *IDecl, 107164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD, 107264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &ResultStr) { 107364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //fprintf(stderr,"In RewriteObjCMethodDecl\n"); 107464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FPRetType = 0; 107564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "\nstatic "; 107664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeIntoString(OMD->getResultType(), ResultStr, FPRetType); 107764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " "; 107864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 107964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique method name 108064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string NameStr; 108164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 108264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isInstanceMethod()) 108364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_I_"; 108464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 108564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_C_"; 108664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 108764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += IDecl->getNameAsString(); 108864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_"; 108964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 109064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCCategoryImplDecl *CID = 109164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext())) { 109264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += CID->getNameAsString(); 109364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_"; 109464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 109564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Append selector names, replacing ':' with '_' 109664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 109764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string selString = OMD->getSelector().getAsString(); 109864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int len = selString.size(); 109964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < len; i++) 110064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (selString[i] == ':') 110164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian selString[i] = '_'; 110264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += selString; 110364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 110464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remember this name for metadata emission 110564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodInternalNames[OMD] = NameStr; 110664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += NameStr; 110764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 110864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite arguments 110964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "("; 111064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 111164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // invisible arguments 111264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isInstanceMethod()) { 111364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType selfTy = Context->getObjCInterfaceType(IDecl); 111464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian selfTy = Context->getPointerType(selfTy); 111564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!LangOpts.MicrosoftExt) { 111664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSynthesizedStructs.count(const_cast<ObjCInterfaceDecl*>(IDecl))) 111764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "struct "; 111864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 111964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // When rewriting for Microsoft, explicitly omit the structure name. 112064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += IDecl->getNameAsString(); 112164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " *"; 112264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 112364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 112464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Context->getObjCClassType().getAsString( 112564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 112664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 112764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " self, "; 112864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Context->getObjCSelType().getAsString(Context->getPrintingPolicy()); 112964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " _cmd"; 113064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 113164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Method arguments. 113264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(), 113364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = OMD->param_end(); PI != E; ++PI) { 113464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParmVarDecl *PDecl = *PI; 113564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ", "; 113664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PDecl->getType()->isObjCQualifiedIdType()) { 113764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "id "; 113864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += PDecl->getNameAsString(); 113964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 114064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = PDecl->getNameAsString(); 114164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = PDecl->getType(); 114264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 114364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (convertBlockPointerToFunctionPointer(QT)) 114464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(Name, Context->getPrintingPolicy()); 114564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 114664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PDecl->getType().getAsStringInternal(Name, Context->getPrintingPolicy()); 114764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Name; 114864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 114964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 115064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isVariadic()) 115164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ", ..."; 115264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ") "; 115364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 115464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FPRetType) { 115564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ")"; // close the precedence "scope" for "*". 115664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 115764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, emit the argument types (if any). 115864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FPRetType)) { 115964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "("; 116064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) { 116164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i) ResultStr += ", "; 116264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr = FT->getArgType(i).getAsString( 116364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 116464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ParamStr; 116564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 116664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 116764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->getNumArgs()) ResultStr += ", "; 116864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "..."; 116964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 117064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ")"; 117164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 117264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "()"; 117364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 117464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 117564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 117664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteImplementationDecl(Decl *OID) { 117764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD = dyn_cast<ObjCImplementationDecl>(OID); 117864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID = dyn_cast<ObjCCategoryImplDecl>(OID); 117964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1180d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian if (IMD) { 1181d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian InsertText(IMD->getLocStart(), "// "); 1182af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (IMD->getIvarLBraceLoc().isValid()) 1183af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(IMD->getIvarLBraceLoc(), "// "); 1184af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian for (ObjCImplementationDecl::ivar_iterator 1185af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian I = IMD->ivar_begin(), E = IMD->ivar_end(); I != E; ++I) { 1186af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian ObjCIvarDecl *Ivar = (*I); 1187af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian SourceLocation LocStart = Ivar->getLocStart(); 1188af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 1189d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian } 1190af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian if (IMD->getIvarRBraceLoc().isValid()) 1191af300298ab86752fa64e339ba34195888a830756Fariborz Jahanian InsertText(IMD->getIvarRBraceLoc(), "// "); 1192d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian } 1193d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian else 1194d2aea120011da5d411e486a9674e9fb05d22f51bFariborz Jahanian InsertText(CID->getLocStart(), "// "); 119564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 119664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::instmeth_iterator 119764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->instmeth_begin() : CID->instmeth_begin(), 119864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->instmeth_end() : CID->instmeth_end(); 119964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 120064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 120164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD = *I; 120264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OMD->getClassInterface(), OMD, ResultStr); 120364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = OMD->getLocStart(); 120464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = OMD->getCompoundBody()->getLocStart(); 120564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 120664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 120764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 120864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, ResultStr); 120964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 121064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 121164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::classmeth_iterator 121264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->classmeth_begin() : CID->classmeth_begin(), 121364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->classmeth_end() : CID->classmeth_end(); 121464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 121564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 121664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD = *I; 121764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OMD->getClassInterface(), OMD, ResultStr); 121864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = OMD->getLocStart(); 121964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = OMD->getCompoundBody()->getLocStart(); 122064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 122164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 122264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 122364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, ResultStr); 122464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 122564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::propimpl_iterator 122664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->propimpl_begin() : CID->propimpl_begin(), 122764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->propimpl_end() : CID->propimpl_end(); 122864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 122964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewritePropertyImplDecl(*I, IMD, CID); 123064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 123164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 123264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(IMD ? IMD->getLocEnd() : CID->getLocEnd(), "// "); 123364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 123464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 123564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteInterfaceDecl(ObjCInterfaceDecl *ClassDecl) { 12368f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // Do not synthesize more than once. 12378f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian if (ObjCSynthesizedStructs.count(ClassDecl)) 12388f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian return; 12398f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // Make sure super class's are written before current class is written. 12408f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian ObjCInterfaceDecl *SuperClass = ClassDecl->getSuperClass(); 12418f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian while (SuperClass) { 12428f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian RewriteInterfaceDecl(SuperClass); 12438f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian SuperClass = SuperClass->getSuperClass(); 12448f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian } 124564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 1246cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian if (!ObjCWrittenInterfaces.count(ClassDecl->getCanonicalDecl())) { 124764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we haven't seen a forward decl - generate a typedef. 1248c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian RewriteOneForwardClassDecl(ClassDecl, ResultStr); 124972c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian RewriteIvarOffsetSymbols(ClassDecl, ResultStr); 125072c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 12514339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteObjCInternalStruct(ClassDecl, ResultStr); 1252cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian // Mark this typedef as having been written into its c++ equivalent. 1253cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian ObjCWrittenInterfaces.insert(ClassDecl->getCanonicalDecl()); 12544339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian 12554339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian for (ObjCInterfaceDecl::prop_iterator I = ClassDecl->prop_begin(), 125664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ClassDecl->prop_end(); I != E; ++I) 12574339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteProperty(*I); 12584339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian for (ObjCInterfaceDecl::instmeth_iterator 125964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = ClassDecl->instmeth_begin(), E = ClassDecl->instmeth_end(); 12604339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian I != E; ++I) 12614339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteMethodDeclaration(*I); 12624339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian for (ObjCInterfaceDecl::classmeth_iterator 126364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = ClassDecl->classmeth_begin(), E = ClassDecl->classmeth_end(); 12644339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian I != E; ++I) 12654339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian RewriteMethodDeclaration(*I); 126664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 12674339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian // Lastly, comment out the @end. 12684339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian ReplaceText(ClassDecl->getAtEndRange().getBegin(), strlen("@end"), 12694339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian "/* @end */"); 12704339bb3a71b6463415708553fda16aa7e44d07ebFariborz Jahanian } 127164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 127264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 127364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewritePropertyOrImplicitSetter(PseudoObjectExpr *PseudoOp) { 127464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = PseudoOp->getSourceRange(); 127564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 127664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We just magically know some things about the structure of this 127764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression. 127864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *OldMsg = 127964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast<ObjCMessageExpr>(PseudoOp->getSemanticExpr( 128064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PseudoOp->getNumSemanticExprs() - 1)); 128164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 128264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Because the rewriter doesn't allow us to rewrite rewritten code, 128364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need to suppress rewriting the sub-statements. 128464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Base, *RHS; 128564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 128664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 128764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 128864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the base expression if we have one. 128964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = 0; 129064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OldMsg->getReceiverKind() == ObjCMessageExpr::Instance) { 129164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = OldMsg->getInstanceReceiver(); 129264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<OpaqueValueExpr>(Base)->getSourceExpr(); 129364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Base)); 129464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 129564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 129664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the RHS. 129764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS = cast<BinaryOperator>(PseudoOp->getSyntacticForm())->getRHS(); 129864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS = cast<OpaqueValueExpr>(RHS)->getSourceExpr(); 129964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(RHS)); 130064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 130164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 130264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // TODO: avoid this copy. 130364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<SourceLocation, 1> SelLocs; 130464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelectorLocs(SelLocs); 130564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 130664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *NewMsg = 0; 130764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (OldMsg->getReceiverKind()) { 130864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: 130964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 131064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 131164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 131264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getClassReceiverTypeInfo(), 131364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 131464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 131564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 131664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS, 131764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 131864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 131964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 132064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 132164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: 132264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 132364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 132464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 132564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base, 132664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 132764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 132864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 132964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS, 133064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 133164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 133264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 133364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 133464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: 133564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance: 133664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 133764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 133864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 133964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperLoc(), 134064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getReceiverKind() == ObjCMessageExpr::SuperInstance, 134164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperType(), 134264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 134364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 134464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 134564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS, 134664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 134764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 134864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 134964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 135064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 135164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *Replacement = SynthMessageExpr(NewMsg); 135264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmtWithRange(PseudoOp, Replacement, OldRange); 135364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 135464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 135564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 135664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewritePropertyOrImplicitGetter(PseudoObjectExpr *PseudoOp) { 135764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = PseudoOp->getSourceRange(); 135864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 135964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We just magically know some things about the structure of this 136064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression. 136164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *OldMsg = 136264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast<ObjCMessageExpr>(PseudoOp->getResultExpr()->IgnoreImplicit()); 136364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 136464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Because the rewriter doesn't allow us to rewrite rewritten code, 136564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need to suppress rewriting the sub-statements. 136664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Base = 0; 136764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 136864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 136964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 137064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the base expression if we have one. 137164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OldMsg->getReceiverKind() == ObjCMessageExpr::Instance) { 137264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = OldMsg->getInstanceReceiver(); 137364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<OpaqueValueExpr>(Base)->getSourceExpr(); 137464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Base)); 137564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 137664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 137764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 137864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Intentionally empty. 137964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<SourceLocation, 1> SelLocs; 138064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 1> Args; 138164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 138264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *NewMsg = 0; 138364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (OldMsg->getReceiverKind()) { 138464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: 138564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 138664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 138764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 138864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getClassReceiverTypeInfo(), 138964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 139064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 139164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 139264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 139364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 139464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 139564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 139664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 139764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: 139864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 139964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 140064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 140164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base, 140264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 140364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 140464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 140564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 140664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 140764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 140864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 140964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 141064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: 141164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance: 141264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 141364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 141464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 141564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperLoc(), 141664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getReceiverKind() == ObjCMessageExpr::SuperInstance, 141764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperType(), 141864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 141964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 142064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 142164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 142264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 142364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 142464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 142564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 142664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 142764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *Replacement = SynthMessageExpr(NewMsg); 142864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmtWithRange(PseudoOp, Replacement, OldRange); 142964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 143064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 143164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 143264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SynthCountByEnumWithState - To print: 143364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ((unsigned int (*) 143464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int)) 143564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (void *)objc_msgSend)((id)l_collection, 143664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// sel_registerName( 143764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// "countByEnumeratingWithState:objects:count:"), 143864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// &enumState, 143964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (id *)__rw_items, (unsigned int)16) 144064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 144164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthCountByEnumWithState(std::string &buf) { 144264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "((unsigned int (*) (id, SEL, struct __objcFastEnumerationState *, " 144364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "id *, unsigned int))(void *)objc_msgSend)"; 144464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 144564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "((id)l_collection,\n\t\t"; 144664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "sel_registerName(\"countByEnumeratingWithState:objects:count:\"),"; 144764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 144864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "&enumState, " 144964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "(id *)__rw_items, (unsigned int)16)"; 145064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 145164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 145264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteBreakStmt - Rewrite for a break-stmt inside an ObjC2's foreach 145364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// statement to exit to its outer synthesized loop. 145464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 145564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteBreakStmt(BreakStmt *S) { 145664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back())) 145764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 145864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace break with goto __break_label 145964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 146064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 146164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 146264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "goto __break_label_"; 146364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 146464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, strlen("break"), buf); 146564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 146664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 146764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 146864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 146964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteContinueStmt - Rewrite for a continue-stmt inside an ObjC2's foreach 147064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// statement to continue with its inner synthesized loop. 147164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 147264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteContinueStmt(ContinueStmt *S) { 147364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back())) 147464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 147564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace continue with goto __continue_label 147664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 147764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 147864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 147964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "goto __continue_label_"; 148064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 148164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, strlen("continue"), buf); 148264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 148364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 148464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 148564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 148664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCForCollectionStmt - Rewriter for ObjC2's foreach statement. 148764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// It rewrites: 148864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// for ( type elem in collection) { stmts; } 148964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 149064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// Into: 149164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// { 149264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// type elem; 149364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __objcFastEnumerationState enumState = { 0 }; 149464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id __rw_items[16]; 149564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id l_collection = (id)collection; 149664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long limit = [l_collection countByEnumeratingWithState:&enumState 149764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objects:__rw_items count:16]; 149864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// if (limit) { 149964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long startMutations = *enumState.mutationsPtr; 150064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// do { 150164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long counter = 0; 150264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// do { 150364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// if (startMutations != *enumState.mutationsPtr) 150464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objc_enumerationMutation(l_collection); 150564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = (type)enumState.itemsPtr[counter++]; 150664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// stmts; 150764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __continue_label: ; 150864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } while (counter < limit); 150964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } while (limit = [l_collection countByEnumeratingWithState:&enumState 151064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objects:__rw_items count:16]); 151164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = nil; 151264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __break_label: ; 151364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 151464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// else 151564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = nil; 151664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 151764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 151864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S, 151964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OrigEnd) { 152064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!Stmts.empty() && "ObjCForCollectionStmt - Statement stack empty"); 152164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(isa<ObjCForCollectionStmt>(Stmts.back()) && 152264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "ObjCForCollectionStmt Statement stack mismatch"); 152364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!ObjCBcLabelNo.empty() && 152464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "ObjCForCollectionStmt - Label No stack empty"); 152564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 152664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 152764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 152864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef elementName; 152964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string elementTypeAsString; 153064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 153164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "\n{\n\t"; 153264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclStmt *DS = dyn_cast<DeclStmt>(S->getElement())) { 153364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // type elem; 153464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NamedDecl* D = cast<NamedDecl>(DS->getSingleDecl()); 153564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ElementType = cast<ValueDecl>(D)->getType(); 153664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ElementType->isObjCQualifiedIdType() || 153764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ElementType->isObjCQualifiedInterfaceType()) 153864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simply use 'id' for all qualified types. 153964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = "id"; 154064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 154164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = ElementType.getAsString(Context->getPrintingPolicy()); 154264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 154364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " "; 154464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementName = D->getName(); 154564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 154664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ";\n\t"; 154764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 154864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 154964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DR = cast<DeclRefExpr>(S->getElement()); 155064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementName = DR->getDecl()->getName(); 155164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = cast<ValueDecl>(DR->getDecl()); 155264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getType()->isObjCQualifiedIdType() || 155364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD->getType()->isObjCQualifiedInterfaceType()) 155464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simply use 'id' for all qualified types. 155564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = "id"; 155664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 155764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = VD->getType().getAsString(Context->getPrintingPolicy()); 155864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 155964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 156064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct __objcFastEnumerationState enumState = { 0 }; 156164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "struct __objcFastEnumerationState enumState = { 0 };\n\t"; 156264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // id __rw_items[16]; 156364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id __rw_items[16];\n\t"; 156464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // id l_collection = (id) 156564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id l_collection = (id)"; 156664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find start location of 'collection' the hard way! 156764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startCollectionBuf = startBuf; 156864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf += 3; // skip 'for' 156964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf = strchr(startCollectionBuf, '('); 157064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf++; // skip '(' 157164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // find 'in' and skip it. 157264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startCollectionBuf != ' ' || 157364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *(startCollectionBuf+1) != 'i' || *(startCollectionBuf+2) != 'n' || 157464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*(startCollectionBuf+3) != ' ' && 157564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *(startCollectionBuf+3) != '[' && *(startCollectionBuf+3) != '(')) 157664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf++; 157764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf += 3; 157864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 157964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace: "for (type element in" with string constructed thus far. 158064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, startCollectionBuf - startBuf, buf); 158164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace ')' in for '(' type elem in collection ')' with ';' 158264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation rightParenLoc = S->getRParenLoc(); 158364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *rparenBuf = SM->getCharacterData(rightParenLoc); 158464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation lparenLoc = startLoc.getLocWithOffset(rparenBuf-startBuf); 158564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = ";\n\t"; 158664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 158764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // unsigned long limit = [l_collection countByEnumeratingWithState:&enumState 158864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // objects:__rw_items count:16]; 158964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // which is synthesized into: 159064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // unsigned int limit = 159164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ((unsigned int (*) 159264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int)) 159364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (void *)objc_msgSend)((id)l_collection, 159464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // sel_registerName( 159564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // "countByEnumeratingWithState:objects:count:"), 159664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct __objcFastEnumerationState *)&state, 159764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id *)__rw_items, (unsigned int)16); 159864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long limit =\n\t\t"; 159964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthCountByEnumWithState(buf); 160064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ";\n\t"; 160164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// if (limit) { 160264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// unsigned long startMutations = *enumState.mutationsPtr; 160364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// do { 160464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// unsigned long counter = 0; 160564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// do { 160664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// if (startMutations != *enumState.mutationsPtr) 160764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// objc_enumerationMutation(l_collection); 160864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = (type)enumState.itemsPtr[counter++]; 160964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (limit) {\n\t"; 161064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long startMutations = *enumState.mutationsPtr;\n\t"; 161164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "do {\n\t\t"; 161264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long counter = 0;\n\t\t"; 161364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "do {\n\t\t\t"; 161464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (startMutations != *enumState.mutationsPtr)\n\t\t\t\t"; 161564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "objc_enumerationMutation(l_collection);\n\t\t\t"; 161664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 161764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = ("; 161864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 161964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")enumState.itemsPtr[counter++];"; 162064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace ')' in for '(' type elem in collection ')' with all of these. 162164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(lparenLoc, 1, buf); 162264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 162364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// __continue_label: ; 162464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } while (counter < limit); 162564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } while (limit = [l_collection countByEnumeratingWithState:&enumState 162664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// objects:__rw_items count:16]); 162764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = nil; 162864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// __break_label: ; 162964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } 163064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// else 163164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = nil; 163264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } 163364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// 163464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = ";\n\t"; 163564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "__continue_label_"; 163664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 163764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ": ;"; 163864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 163964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "} while (counter < limit);\n\t"; 164064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "} while (limit = "; 164164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthCountByEnumWithState(buf); 164264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ");\n\t"; 164364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 164464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = (("; 164564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 164664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")0);\n\t"; 164764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "__break_label_"; 164864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 164964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ": ;\n\t"; 165064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n\t"; 165164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "else\n\t\t"; 165264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 165364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = (("; 165464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 165564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")0);\n\t"; 165664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 165764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 165864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert all these *after* the statement body. 165964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: If this should support Obj-C++, support CXXTryStmt 166064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<CompoundStmt>(S->getBody())) { 166164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endBodyLoc = OrigEnd.getLocWithOffset(1); 166264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(endBodyLoc, buf); 166364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 166464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* Need to treat single statements specially. For example: 166564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * 166664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * for (A *a in b) if (stuff()) break; 166764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * for (A *a in b) xxxyy; 166864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * 166964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * The following code simply scans ahead to the semi to find the actual end. 167064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 167164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *stmtBuf = SM->getCharacterData(OrigEnd); 167264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(stmtBuf, ';'); 167364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(semiBuf && "Can't find ';'"); 167464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endBodyLoc = OrigEnd.getLocWithOffset(semiBuf-stmtBuf+1); 167564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(endBodyLoc, buf); 167664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 167764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.pop_back(); 167864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCBcLabelNo.pop_back(); 167964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 168064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 168164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1682542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanianstatic void Write_RethrowObject(std::string &buf) { 1683542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "{ struct _FIN { _FIN(id reth) : rethrow(reth) {}\n"; 1684542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "\t~_FIN() { if (rethrow) objc_exception_throw(rethrow); }\n"; 1685542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "\tid rethrow;\n"; 1686542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "\t} _fin_force_rethow(_rethrow);"; 1687542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian} 1688542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 168964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCSynchronizedStmt - 169064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// This routine rewrites @synchronized(expr) stmt; 169164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// into: 169264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objc_sync_enter(expr); 169364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// @try stmt @finally { objc_sync_exit(expr); } 169464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 169564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S) { 169664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 169764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 169864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 169964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 170064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @synchronized location"); 170164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 170264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 170322e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf = "{ id _rethrow = 0; id _sync_obj = "; 1704542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 170564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *lparenBuf = startBuf; 170664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*lparenBuf != '(') lparenBuf++; 170764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, lparenBuf-startBuf+1, buf); 170822e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian 170922e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf = "; objc_sync_enter(_sync_obj);\n"; 171022e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "try {\n\tstruct _SYNC_EXIT { _SYNC_EXIT(id arg) : sync_exit(arg) {}"; 171122e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "\n\t~_SYNC_EXIT() {objc_sync_exit(sync_exit);}"; 171222e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "\n\tid sync_exit;"; 171322e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian buf += "\n\t} _sync_exit(_sync_obj);\n"; 171422e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian 171522e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // We can't use S->getSynchExpr()->getLocEnd() to find the end location, since 171622e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // the sync expression is typically a message expression that's already 171722e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // been rewritten! (which implies the SourceLocation's are invalid). 171822e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian SourceLocation RParenExprLoc = S->getSynchBody()->getLocStart(); 171922e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian const char *RParenExprLocBuf = SM->getCharacterData(RParenExprLoc); 172022e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian while (*RParenExprLocBuf != ')') RParenExprLocBuf--; 172122e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian RParenExprLoc = startLoc.getLocWithOffset(RParenExprLocBuf-startBuf); 172222e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian 172322e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian SourceLocation LBranceLoc = S->getSynchBody()->getLocStart(); 172422e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian const char *LBraceLocBuf = SM->getCharacterData(LBranceLoc); 172522e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian assert (*LBraceLocBuf == '{'); 172622e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian ReplaceText(RParenExprLoc, (LBraceLocBuf - SM->getCharacterData(RParenExprLoc) + 1), buf); 1727542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 1728b655bf06d99bbdb58de5d012642b87cabddabd83Fariborz Jahanian SourceLocation startRBraceLoc = S->getSynchBody()->getLocEnd(); 17299ab511cde246a3a4e47b5b1c4aa2b861fcff08d5Matt Beaumont-Gay assert((*SM->getCharacterData(startRBraceLoc) == '}') && 17309ab511cde246a3a4e47b5b1c4aa2b861fcff08d5Matt Beaumont-Gay "bogus @synchronized block"); 1731542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian 1732542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf = "} catch (id e) {_rethrow = e;}\n"; 1733542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian Write_RethrowObject(buf); 173464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 1735542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian buf += "}\n"; 173664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 1737b655bf06d99bbdb58de5d012642b87cabddabd83Fariborz Jahanian ReplaceText(startRBraceLoc, 1, buf); 173864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 173964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 174064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 174164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 174264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::WarnAboutReturnGotoStmts(Stmt *S) 174364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian{ 174464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up traversal of all children. 174564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 174664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) 174764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian WarnAboutReturnGotoStmts(*CI); 174864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 174964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ReturnStmt>(S) || isa<GotoStmt>(S)) { 175064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(S->getLocStart()), 175164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TryFinallyContainsReturnDiag); 175264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 175364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 175464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 175564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 175664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCTryStmt(ObjCAtTryStmt *S) { 1757b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian ObjCAtFinallyStmt *finalStmt = S->getFinallyStmt(); 1758220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian bool noCatch = S->getNumCatchStmts() == 0; 1759b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian std::string buf; 1760b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian 1761b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian if (finalStmt) { 1762220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian if (noCatch) 1763220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "{ id volatile _rethrow = 0;\n"; 1764220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian else { 1765220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "{ id volatile _rethrow = 0;\ntry {\n"; 1766220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian } 1767b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian } 176864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 176964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 177064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 177164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 177264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @try location"); 1773b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian if (finalStmt) 1774b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian ReplaceText(startLoc, 1, buf); 1775b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian else 1776b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian // @try -> try 1777b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian ReplaceText(startLoc, 1, ""); 1778b122818a33348a9d1694b10e44eed9746aa492b0Fariborz Jahanian 177964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I) { 178064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCAtCatchStmt *Catch = S->getCatchStmt(I); 17814c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian VarDecl *catchDecl = Catch->getCatchParamDecl(); 1782c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian 178364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = Catch->getLocStart(); 17844c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian bool AtRemoved = false; 17854c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (catchDecl) { 17864c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian QualType t = catchDecl->getType(); 17874c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (const ObjCObjectPointerType *Ptr = t->getAs<ObjCObjectPointerType>()) { 17884c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // Should be a pointer to a class. 17894c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ObjCInterfaceDecl *IDecl = Ptr->getObjectType()->getInterface(); 17904c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (IDecl) { 17914c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian std::string Result; 17924c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian startBuf = SM->getCharacterData(startLoc); 17934c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian assert((*startBuf == '@') && "bogus @catch location"); 17944c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian SourceLocation rParenLoc = Catch->getRParenLoc(); 17954c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian const char *rParenBuf = SM->getCharacterData(rParenLoc); 17964c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian 17974c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // _objc_exc_Foo *_e as argument to catch. 17984c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result = "catch (_objc_exc_"; Result += IDecl->getNameAsString(); 17994c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += " *_"; Result += catchDecl->getNameAsString(); 18004c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += ")"; 18014c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ReplaceText(startLoc, rParenBuf-startBuf+1, Result); 18024c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // Foo *e = (Foo *)_e; 18034c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result.clear(); 18044c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result = "{ "; 18054c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += IDecl->getNameAsString(); 18064c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += " *"; Result += catchDecl->getNameAsString(); 18074c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += " = ("; Result += IDecl->getNameAsString(); Result += "*)"; 18084c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += "_"; Result += catchDecl->getNameAsString(); 18094c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian 18104c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian Result += "; "; 18114c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian SourceLocation lBraceLoc = Catch->getCatchBody()->getLocStart(); 18124c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ReplaceText(lBraceLoc, 1, Result); 18134c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian AtRemoved = true; 18144c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian } 18154c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian } 18164c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian } 18174c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian if (!AtRemoved) 18184c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian // @catch -> catch 18194c14881e86350c8aa12d5182d35ce07860de2325Fariborz Jahanian ReplaceText(startLoc, 1, ""); 1820c38503b4c1cda42154700016d60c6467af438a15Fariborz Jahanian 182164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 1822220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian if (finalStmt) { 1823220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf.clear(); 1824220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian if (noCatch) 1825220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "catch (id e) {_rethrow = e;}\n"; 1826220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian else 1827220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf = "}\ncatch (id e) {_rethrow = e;}\n"; 1828220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian 1829220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian SourceLocation startFinalLoc = finalStmt->getLocStart(); 1830220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian ReplaceText(startFinalLoc, 8, buf); 1831220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian Stmt *body = finalStmt->getFinallyBody(); 1832220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian SourceLocation startFinalBodyLoc = body->getLocStart(); 1833220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian buf.clear(); 1834542125f84d63864b6f7abbed61576a5dffecd097Fariborz Jahanian Write_RethrowObject(buf); 1835220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian ReplaceText(startFinalBodyLoc, 1, buf); 1836220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian 1837220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian SourceLocation endFinalBodyLoc = body->getLocEnd(); 1838220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian ReplaceText(endFinalBodyLoc, 1, "}\n}"); 183922e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian // Now check for any return/continue/go statements within the @try. 184022e2f854cb7fd53e048fcbdbee1e78108cb8d146Fariborz Jahanian WarnAboutReturnGotoStmts(S->getTryBody()); 1841220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian } 1842220419acf26033d794edef802cbb916a4d2896b4Fariborz Jahanian 184364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 184464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 184564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 184664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// This can't be done with ReplaceStmt(S, ThrowExpr), since 184764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// the throw expression is typically a message expression that's already 184864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// been rewritten! (which implies the SourceLocation's are invalid). 184964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCThrowStmt(ObjCAtThrowStmt *S) { 185064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 185164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 185264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 185364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 185464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @throw location"); 185564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 185664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 185764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* void objc_exception_throw(id) __attribute__((noreturn)); */ 185864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (S->getThrowExpr()) 185964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "objc_exception_throw("; 186040539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian else 186140539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian buf = "throw"; 186264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 186364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // handle "@ throw" correctly. 186464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *wBuf = strchr(startBuf, 'w'); 186564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*wBuf == 'w') && "@throw: can't find 'w'"); 186664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, wBuf-startBuf+1, buf); 186764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 186864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 186964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "@throw: can't find ';'"); 187064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation semiLoc = startLoc.getLocWithOffset(semiBuf-startBuf); 187140539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian if (S->getThrowExpr()) 187240539467ec19e6bc53d47ba650f42aa3f414a53fFariborz Jahanian ReplaceText(semiLoc, 1, ");"); 187364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 187464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 187564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 187664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteAtEncode(ObjCEncodeExpr *Exp) { 187764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a new string expression. 187864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType StrType = Context->getPointerType(Context->CharTy); 187964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StrEncoding; 188064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForType(Exp->getEncodedType(), StrEncoding); 188164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Replacement = StringLiteral::Create(*Context, StrEncoding, 188264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 188364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StrType, SourceLocation()); 188464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, Replacement); 188564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 188664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace this subexpr in the parent. 188764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 188864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 188964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 189064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 189164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteAtSelector(ObjCSelectorExpr *Exp) { 189264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SelGetUidFunctionDecl) 189364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSelGetUidFunctionDecl(); 189464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(SelGetUidFunctionDecl && "Can't find sel_registerName() decl"); 189564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a call to sel_registerName("selName"). 189664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> SelExprs; 189764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 189864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 189964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getSelector().getAsString(), 190064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 190164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 190264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 190364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &SelExprs[0], SelExprs.size()); 190464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, SelExp); 190564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 190664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return SelExp; 190764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 190864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 190964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianCallExpr *RewriteModernObjC::SynthesizeCallToFunctionDecl( 191064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD, Expr **args, unsigned nargs, SourceLocation StartLoc, 191164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc) { 191264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 191364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = FD->getType(); 191464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 191564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 191664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = 1917f4b88a45902af1802a1cb42ba48b1c474474f228John McCall new (Context) DeclRefExpr(FD, false, msgSendType, VK_LValue, SourceLocation()); 191864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 191964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, we cast the reference to a pointer to the objc_msgSend type. 192064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType pToFunc = Context->getPointerType(msgSendType); 192164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImplicitCastExpr *ICE = 192264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImplicitCastExpr::Create(*Context, pToFunc, CK_FunctionToPointerDecay, 192364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DRE, 0, VK_RValue); 192464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 192564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 192664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 192764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Exp = 192864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) CallExpr(*Context, ICE, args, nargs, 192964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getCallResultType(*Context), 193064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, EndLoc); 193164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Exp; 193264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 193364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 193464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool scanForProtocolRefs(const char *startBuf, const char *endBuf, 193564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&startRef, const char *&endRef) { 193664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (startBuf < endBuf) { 193764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '<') 193864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startRef = startBuf; // mark the start. 193964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '>') { 194064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (startRef && *startRef == '<') { 194164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endRef = startBuf; // mark the end. 194264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 194364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 194464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 194564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 194664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 194764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 194864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 194964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 195064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 195164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void scanToNextArgument(const char *&argRef) { 195264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int angle = 0; 195364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argRef != ')' && (*argRef != ',' || angle > 0)) { 195464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*argRef == '<') 195564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian angle++; 195664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*argRef == '>') 195764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian angle--; 195864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argRef++; 195964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 196064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(angle == 0 && "scanToNextArgument - bad protocol type syntax"); 196164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 196264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 196364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::needToScanForQualifiers(QualType T) { 196464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 196564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 196664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType *PT = T->getAs<PointerType>()) { 196764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT->getPointeeType()->isObjCQualifiedIdType()) 196864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 196964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 197064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCObjectPointerType()) { 197164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = T->getPointeeType(); 197264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return T->isObjCQualifiedInterfaceType(); 197364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 197464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isArrayType()) { 197564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ElemTy = Context->getBaseElementType(T); 197664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return needToScanForQualifiers(ElemTy); 197764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 197864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 197964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 198064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 198164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCQualifiedInterfaceTypes(Expr *E) { 198264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = E->getType(); 198364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(Type)) { 198464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc, EndLoc; 198564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 198664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) { 198764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = ECE->getLParenLoc(); 198864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc = ECE->getRParenLoc(); 198964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 199064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = E->getLocStart(); 199164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc = E->getLocEnd(); 199264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 199364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This will defend against trying to rewrite synthesized expressions. 199464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Loc.isInvalid() || EndLoc.isInvalid()) 199564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 199664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 199764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(Loc); 199864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(EndLoc); 199964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 200064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 200164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 200264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = Loc.getLocWithOffset(startRef-startBuf); 200364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = Loc.getLocWithOffset(endRef-startBuf+1); 200464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 200564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 200664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 200764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 200864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 200964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 201064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 201164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCQualifiedInterfaceTypes(Decl *Dcl) { 201264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc; 201364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type; 201464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *proto = 0; 201564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(Dcl)) { 201664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = VD->getLocation(); 201764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = VD->getType(); 201864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 201964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FunctionDecl *FD = dyn_cast<FunctionDecl>(Dcl)) { 202064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = FD->getLocation(); 202164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Check for ObjC 'id' and class types that have been adorned with protocol 202264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // information (id<p>, C<p>*). The protocol references need to be rewritten! 202364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *funcType = FD->getType()->getAs<FunctionType>(); 202464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(funcType && "missing function type"); 202564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian proto = dyn_cast<FunctionProtoType>(funcType); 202664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 202764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 202864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = proto->getResultType(); 202964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 203064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FieldDecl *FD = dyn_cast<FieldDecl>(Dcl)) { 203164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = FD->getLocation(); 203264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = FD->getType(); 203364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 203464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 203564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 203664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 203764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(Type)) { 203864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since types are unique, we need to scan the buffer. 203964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 204064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(Loc); 204164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = endBuf; 204264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != ';' && *startBuf != '<' && startBuf != MainFileStart) 204364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf--; // scan backward (from the decl location) for return type. 204464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 204564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 204664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 204764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = Loc.getLocWithOffset(startRef-endBuf); 204864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = Loc.getLocWithOffset(endRef-endBuf+1); 204964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 205064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 205164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 205264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 205364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 205464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 205564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // most likely, was a variable 205664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now check arguments. 205764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(Loc); 205864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startFuncBuf = startBuf; 205964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < proto->getNumArgs(); i++) { 206064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(proto->getArgType(i))) { 206164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since types are unique, we need to scan the buffer. 206264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 206364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = startBuf; 206464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scan forward (from the decl location) for argument types. 206564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian scanToNextArgument(endBuf); 206664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 206764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 206864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 206964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = 207064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc.getLocWithOffset(startRef-startFuncBuf); 207164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = 207264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc.getLocWithOffset(endRef-startFuncBuf+1); 207364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 207464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 207564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 207664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 207764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = ++endBuf; 207864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 207964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 208064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If the function name is derived from a macro expansion, then the 208164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // argument buffer will not follow the name. Need to speak with Chris. 208264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf && *startBuf != ')' && *startBuf != ',') 208364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; // scan forward (from the decl location) for argument types. 208464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 208564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 208664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 208764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 208864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 208964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteTypeOfDecl(VarDecl *ND) { 209064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = ND->getType(); 209164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Type* TypePtr = QT->getAs<Type>(); 209264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isa<TypeOfExprType>(TypePtr)) 209364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 209464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (isa<TypeOfExprType>(TypePtr)) { 209564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const TypeOfExprType *TypeOfExprTypePtr = cast<TypeOfExprType>(TypePtr); 209664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = TypeOfExprTypePtr->getUnderlyingExpr()->getType(); 209764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypePtr = QT->getAs<Type>(); 209864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 209964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. This will not work for multiple declarators; as in: 210064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // __typeof__(a) b,c,d; 210164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeAsString(QT.getAsString(Context->getPrintingPolicy())); 210264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getTypeSpecStartLoc(); 210364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 210464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ND->getInit()) { 210564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 210664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeAsString += " " + Name + " = "; 210764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *E = ND->getInit(); 210864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc; 210964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) 211064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = ECE->getLParenLoc(); 211164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 211264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = E->getLocStart(); 211364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = SM->getExpansionLoc(startLoc); 211464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(startLoc); 211564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf-1, TypeAsString); 211664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 211764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 211864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation X = ND->getLocEnd(); 211964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian X = SM->getExpansionLoc(X); 212064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(X); 212164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf-1, TypeAsString); 212264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 212364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 212464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 212564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthSelGetUidFunctionDecl - SEL sel_registerName(const char *str); 212664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthSelGetUidFunctionDecl() { 212764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *SelGetUidIdent = &Context->Idents.get("sel_registerName"); 212864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 212964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 213064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getFuncType = 213164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSimpleFunctionType(Context->getObjCSelType(), &ArgTys[0], ArgTys.size()); 213264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 213364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 213464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 213564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidIdent, getFuncType, 0, 213664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 213764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 213864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 213964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 214064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteFunctionDecl(FunctionDecl *FD) { 214164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declared in <objc/objc.h> 214264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->getIdentifier() && 214364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getName() == "sel_registerName") { 214464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = FD; 214564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 214664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 214764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(FD); 214864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 214964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 215064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerType(std::string& Str, QualType Type) { 215164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString(Type.getAsString(Context->getPrintingPolicy())); 215264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = TypeString.c_str(); 215364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strchr(argPtr, '^')) { 215464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += TypeString; 215564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 215664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 215764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr) { 215864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += (*argPtr == '^' ? '*' : *argPtr); 215964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; 216064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 216164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 216264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 216364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// FIXME. Consolidate this routine with RewriteBlockPointerType. 216464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerTypeVariable(std::string& Str, 216564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD) { 216664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = VD->getType(); 216764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString(Type.getAsString(Context->getPrintingPolicy())); 216864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = TypeString.c_str(); 216964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int paren = 0; 217064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr) { 217164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 217264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': 217364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 217464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian paren++; 217564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 217664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': 217764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 217864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian paren--; 217964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 218064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 218164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += '*'; 218264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (paren == 1) 218364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += VD->getNameAsString(); 218464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 218564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: 218664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 218764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 218864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 218964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; 219064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 219164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 219264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 219364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 219464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockLiteralFunctionDecl(FunctionDecl *FD) { 219564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart = FD->getTypeSpecStartLoc(); 219664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *funcType = FD->getType()->getAs<FunctionType>(); 219764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *proto = dyn_cast<FunctionProtoType>(funcType); 219864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 219964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 220064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = proto->getResultType(); 220164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FdStr = Type.getAsString(Context->getPrintingPolicy()); 220264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += " "; 220364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += FD->getName(); 220464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += "("; 220564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned numArgs = proto->getNumArgs(); 220664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < numArgs; i++) { 220764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ArgType = proto->getArgType(i); 220864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(FdStr, ArgType); 220964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i+1 < numArgs) 221064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += ", "; 221164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 221264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += ");\n"; 221364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, FdStr); 221464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = 0; 221564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 221664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 221764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthSuperContructorFunctionDecl - id objc_super(id obj, id super); 221864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthSuperContructorFunctionDecl() { 221964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SuperContructorFunctionDecl) 222064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 222164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("__rw_objc_super"); 222264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 222364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 222464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 222564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 222664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 222764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 222864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 222964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperContructorFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 223064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 223164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 223264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 223364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 223464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 223564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 223664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 223764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendFunctionDecl - id objc_msgSend(id self, SEL op, ...); 223864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendFunctionDecl() { 223964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend"); 224064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 224164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 224264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 224364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 224464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 224564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 224664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 224764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 224864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 224964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 225064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 225164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 225264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 225364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 225464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 225564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 225664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 225764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 225864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendSuperFunctionDecl - id objc_msgSendSuper(struct objc_super *, SEL op, ...); 225964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendSuperFunctionDecl() { 226064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSendSuper"); 226164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 226264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 226364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 226464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_super")); 226564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getPointerType(Context->getTagDeclType(RD)); 226664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't build 'struct objc_super *' type"); 226764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 226864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 226964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 227064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 227164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 227264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 227364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 227464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 227564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 227664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 227764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 227864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 227964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 228064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 228164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 228264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendStretFunctionDecl - id objc_msgSend_stret(id self, SEL op, ...); 228364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendStretFunctionDecl() { 228464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_stret"); 228564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 228664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 228764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 228864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 228964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 229064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 229164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 229264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 229364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 229464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 229564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 229664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 229764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 229864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 229964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 230064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 230164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 230264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 230364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendSuperStretFunctionDecl - 230464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// id objc_msgSendSuper_stret(struct objc_super *, SEL op, ...); 230564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendSuperStretFunctionDecl() { 230664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = 230764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_msgSendSuper_stret"); 230864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 230964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 231064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 231164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_super")); 231264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getPointerType(Context->getTagDeclType(RD)); 231364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't build 'struct objc_super *' type"); 231464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 231564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 231664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 231764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 231864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 231964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 232064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 232164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 232264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 232364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 232464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 232564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 232664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 232764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 232864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 232964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendFpretFunctionDecl - double objc_msgSend_fpret(id self, SEL op, ...); 233064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendFpretFunctionDecl() { 233164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_fpret"); 233264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 233364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 233464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 233564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 233664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 233764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 233864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 233964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->DoubleTy, 234064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 234164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 234264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFpretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 234364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 234464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 234564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 234664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 234764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 234864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 234964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 235064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetClassFunctionDecl - id objc_getClass(const char *name); 235164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetClassFunctionDecl() { 235264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getClass"); 235364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 235464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 235564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCIdType(), 235664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 235764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 235864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 235964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 236064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassIdent, getClassType, 0, 236164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 236264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 236364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 236464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 236564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetSuperClassFunctionDecl - Class class_getSuperclass(Class cls); 236664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetSuperClassFunctionDecl() { 236764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getSuperClassIdent = 236864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("class_getSuperclass"); 236964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 237064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getObjCClassType()); 237164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCClassType(), 237264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 237364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetSuperClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 237464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 237564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 237664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSuperClassIdent, 237764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassType, 0, 237864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 237964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, 238064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false); 238164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 238264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 238364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetMetaClassFunctionDecl - id objc_getMetaClass(const char *name); 238464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetMetaClassFunctionDecl() { 238564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getMetaClass"); 238664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 238764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 238864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCIdType(), 238964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 239064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetMetaClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 239164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 239264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 239364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassIdent, getClassType, 0, 239464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 239564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 239664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 239764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 239864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCStringLiteral(ObjCStringLiteral *Exp) { 239964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType strType = getConstantStringStructType(); 240064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 240164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "__NSConstantStringImpl_"; 240264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 240364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string tmpName = InFileName; 240464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned i; 240564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (i=0; i < tmpName.length(); i++) { 240664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char c = tmpName.at(i); 240764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace any non alphanumeric characters with '_'. 240864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isalpha(c) && (c < '0' || c > '9')) 240964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian tmpName[i] = '_'; 241064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 241164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += tmpName; 241264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_"; 241364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(NumObjCStringLiterals++); 241464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 241564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "static __NSConstantStringImpl " + S; 241664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " __attribute__ ((section (\"__DATA, __cfstring\"))) = {__CFConstantStringClassReference,"; 241764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "0x000007c8,"; // utf8_str 241864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The pretty printer for StringLiteral handles escape characters properly. 241964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string prettyBufS; 242064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream prettyBuf(prettyBufS); 242164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getString()->printPretty(prettyBuf, *Context, 0, 242264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PrintingPolicy(LangOpts)); 242364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += prettyBuf.str(); 242464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += ","; 242564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += utostr(Exp->getString()->getByteLength()) + "};\n"; 242664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 242764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 242864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), &Context->Idents.get(S), 242964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian strType, 0, SC_Static, SC_None); 2430f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(NewVD, false, strType, VK_LValue, 243164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 243264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Unop = new (Context) UnaryOperator(DRE, UO_AddrOf, 243364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(DRE->getType()), 243464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 243564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 243664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // cast to NSConstantString * 243764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *cast = NoTypeInfoCStyleCastExpr(Context, Exp->getType(), 243864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_CPointerToObjCPointerCast, Unop); 243964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, cast); 244064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 244164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return cast; 244264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 244364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 244464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// struct objc_super { struct objc_object *receiver; struct objc_class *super; }; 244564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getSuperStructType() { 244664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SuperStructDecl) { 244764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 244864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 244964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_super")); 245064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FieldTypes[2]; 245164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 245264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_object *receiver; 245364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[0] = Context->getObjCIdType(); 245464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_class *super; 245564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[1] = Context->getObjCClassType(); 245664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 245764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create fields 245864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < 2; ++i) { 245964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl->addDecl(FieldDecl::Create(*Context, SuperStructDecl, 246064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 246164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 0, 246264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[i], 0, 246364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, 246464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*Mutable=*/false, 246564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false)); 246664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 246764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 246864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl->completeDefinition(); 246964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 247064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTagDeclType(SuperStructDecl); 247164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 247264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 247364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getConstantStringStructType() { 247464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ConstantStringDecl) { 247564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 247664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 247764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__NSConstantStringImpl")); 247864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FieldTypes[4]; 247964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 248064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_object *receiver; 248164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[0] = Context->getObjCIdType(); 248264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // int flags; 248364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[1] = Context->IntTy; 248464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // char *str; 248564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[2] = Context->getPointerType(Context->CharTy); 248664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // long length; 248764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[3] = Context->LongTy; 248864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 248964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create fields 249064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < 4; ++i) { 249164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl->addDecl(FieldDecl::Create(*Context, 249264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl, 249364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 249464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 0, 249564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[i], 0, 249664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, 249764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*Mutable=*/true, 249864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false)); 249964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 250064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 250164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl->completeDefinition(); 250264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 250364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTagDeclType(ConstantStringDecl); 250464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 250564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 250664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthMessageExpr(ObjCMessageExpr *Exp, 250764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc, 250864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc) { 250964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SelGetUidFunctionDecl) 251064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSelGetUidFunctionDecl(); 251164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendFunctionDecl) 251264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendFunctionDecl(); 251364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendSuperFunctionDecl) 251464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendSuperFunctionDecl(); 251564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendStretFunctionDecl) 251664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendStretFunctionDecl(); 251764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendSuperStretFunctionDecl) 251864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendSuperStretFunctionDecl(); 251964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendFpretFunctionDecl) 252064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendFpretFunctionDecl(); 252164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetClassFunctionDecl) 252264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetClassFunctionDecl(); 252364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetSuperClassFunctionDecl) 252464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetSuperClassFunctionDecl(); 252564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetMetaClassFunctionDecl) 252664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetMetaClassFunctionDecl(); 252764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 252864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // default to objc_msgSend(). 252964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendFlavor = MsgSendFunctionDecl; 253064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // May need to use objc_msgSend_stret() as well. 253164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendStretFlavor = 0; 253264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *mDecl = Exp->getMethodDecl()) { 253364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType resultType = mDecl->getResultType(); 253464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (resultType->isRecordType()) 253564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendStretFunctionDecl; 253664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (resultType->isRealFloatingType()) 253764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendFpretFunctionDecl; 253864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 253964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 254064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize a call to objc_msgSend(). 254164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> MsgExprs; 254264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (Exp->getReceiverKind()) { 254364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: { 254464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendSuperFunctionDecl; 254564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) 254664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendSuperStretFunctionDecl; 254764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(MsgSendFlavor && "MsgSendFlavor is NULL!"); 254864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 254964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = CurMethodDef->getClassInterface(); 255064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 255164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 255264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 255364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // set the receiver to self, the first argument to all methods. 255464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 255564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 255664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 255764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) DeclRefExpr(CurMethodDef->getSelfDecl(), 2558f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, 255964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 256064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, 256164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation())) 256264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ); // set the 'receiver'. 256364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 256464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 256564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 256664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 256764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 256864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassDecl->getIdentifier()->getName(), 256964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 257064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 257164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetMetaClassFunctionDecl, 257264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 257364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 257464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, 257564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 257664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (Class)objc_getClass("CurrentClass") 257764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *ArgExpr = NoTypeInfoCStyleCastExpr(Context, 257864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCClassType(), 257964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls); 258064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.clear(); 258164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(ArgExpr); 258264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Cls = SynthesizeCallToFunctionDecl(GetSuperClassFunctionDecl, 258364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], ClsExprs.size(), 258464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 258564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 258664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 258764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // To turn off a warning, type-cast to 'id' 258864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( // set 'super class', using class_getSuperclass(). 258964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, 259064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 259164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls)); 259264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_super 259364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType superType = getSuperStructType(); 259464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *SuperRep; 259564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 259664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 259764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSuperContructorFunctionDecl(); 259864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 259964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(SuperContructorFunctionDecl, 2600f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, superType, VK_LValue, 260164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 260264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], 260364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.size(), 260464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 260564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 260664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The code for super is a little tricky to prevent collision with 260764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the structure definition in the header. The rewriter has it's own 260864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // internal definition (__rw_objc_super) that is uses. This is why 260964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need the cast below. For example: 261064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super *)&__rw_objc_super((id)self, (id)objc_getClass("SUPER")) 261164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 261264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 261364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 261464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 261564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 261664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = NoTypeInfoCStyleCastExpr(Context, 261764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(superType), 261864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, SuperRep); 261964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 262064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super) { <exprs from above> } 262164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitListExpr *ILE = 262264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) InitListExpr(*Context, SourceLocation(), 262364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &InitExprs[0], InitExprs.size(), 262464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 262564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *superTInfo 262664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(superType); 262764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CompoundLiteralExpr(SourceLocation(), superTInfo, 262864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 262964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ILE, false); 263064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_super * 263164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 263264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 263364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 263464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 263564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 263664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SuperRep); 263764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 263864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 263964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 264064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: { 264164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 264264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 264364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *Class 264464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Exp->getClassReceiver()->getAs<ObjCObjectType>()->getInterface(); 264564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *clsName = Class->getIdentifier(); 264664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 264764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian clsName->getName(), 264864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 264964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 265064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 265164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 265264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 265364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 265464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(Cls); 265564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 265664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 265764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 265864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance:{ 265964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendSuperFunctionDecl; 266064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) 266164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendSuperStretFunctionDecl; 266264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(MsgSendFlavor && "MsgSendFlavor is NULL!"); 266364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = CurMethodDef->getClassInterface(); 266464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 266564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 266664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 266764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 266864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 266964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) DeclRefExpr(CurMethodDef->getSelfDecl(), 2670f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, 267164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 267264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, SourceLocation())) 267364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ); // set the 'receiver'. 267464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 267564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 267664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 267764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 267864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 267964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassDecl->getIdentifier()->getName(), 268064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, argType, 268164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation())); 268264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 268364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 268464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 268564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 268664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (Class)objc_getClass("CurrentClass") 268764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *ArgExpr = NoTypeInfoCStyleCastExpr(Context, 268864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCClassType(), 268964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls); 269064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.clear(); 269164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(ArgExpr); 269264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Cls = SynthesizeCallToFunctionDecl(GetSuperClassFunctionDecl, 269364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], ClsExprs.size(), 269464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 269564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 269664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 269764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // To turn off a warning, type-cast to 'id' 269864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 269964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // set 'super class', using class_getSuperclass(). 270064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 270164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls)); 270264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_super 270364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType superType = getSuperStructType(); 270464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *SuperRep; 270564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 270664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 270764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSuperContructorFunctionDecl(); 270864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 270964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(SuperContructorFunctionDecl, 2710f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, superType, VK_LValue, 271164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 271264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], 271364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.size(), 271464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, SourceLocation()); 271564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The code for super is a little tricky to prevent collision with 271664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the structure definition in the header. The rewriter has it's own 271764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // internal definition (__rw_objc_super) that is uses. This is why 271864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need the cast below. For example: 271964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super *)&__rw_objc_super((id)self, (id)objc_getClass("SUPER")) 272064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 272164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 272264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 272364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 272464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 272564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = NoTypeInfoCStyleCastExpr(Context, 272664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(superType), 272764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, SuperRep); 272864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 272964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super) { <exprs from above> } 273064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitListExpr *ILE = 273164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) InitListExpr(*Context, SourceLocation(), 273264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &InitExprs[0], InitExprs.size(), 273364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 273464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *superTInfo 273564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(superType); 273664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CompoundLiteralExpr(SourceLocation(), superTInfo, 273764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_RValue, ILE, 273864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false); 273964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 274064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SuperRep); 274164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 274264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 274364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 274464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: { 274564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remove all type-casts because it may contain objc-style types; e.g. 274664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Foo<Proto> *. 274764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *recExpr = Exp->getInstanceReceiver(); 274864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(recExpr)) 274964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian recExpr = CE->getSubExpr(); 275064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK = recExpr->getType()->isObjCObjectPointerType() 275164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BitCast : recExpr->getType()->isBlockPointerType() 275264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BlockPointerToObjCPointerCast 275364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : CK_CPointerToObjCPointerCast; 275464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 275564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian recExpr = NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 275664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK, recExpr); 275764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(recExpr); 275864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 275964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 276064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 276164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 276264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a call to sel_registerName("selName"), it will be the 2nd argument. 276364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> SelExprs; 276464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 276564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 276664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getSelector().getAsString(), 276764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 276864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 276964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 277064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &SelExprs[0], SelExprs.size(), 277164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, 277264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 277364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SelExp); 277464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 277564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now push any user supplied arguments. 277664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < Exp->getNumArgs(); i++) { 277764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *userExpr = Exp->getArg(i); 277864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make all implicit casts explicit...ICE comes in handy:-) 277964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(userExpr)) { 278064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Reuse the ICE type, it is exactly what the doctor ordered. 278164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType type = ICE->getType(); 278264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(type)) 278364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian type = Context->getObjCIdType(); 278464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "type (^)(...)" to "type (*)(...)". 278564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(type); 278664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Expr *SubExpr = ICE->IgnoreParenImpCasts(); 278764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK; 278864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SubExpr->getType()->isIntegralType(*Context) && 278964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian type->isBooleanType()) { 279064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_IntegralToBoolean; 279164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (type->isObjCObjectPointerType()) { 279264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SubExpr->getType()->isBlockPointerType()) { 279364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BlockPointerToObjCPointerCast; 279464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (SubExpr->getType()->isPointerType()) { 279564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_CPointerToObjCPointerCast; 279664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 279764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 279864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 279964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 280064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 280164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 280264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 280364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = NoTypeInfoCStyleCastExpr(Context, type, CK, userExpr); 280464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 280564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make id<P...> cast into an 'id' cast. 280664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(userExpr)) { 280764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CE->getType()->isObjCQualifiedIdType()) { 280864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while ((CE = dyn_cast<CStyleCastExpr>(userExpr))) 280964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = CE->getSubExpr(); 281064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK; 281164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (userExpr->getType()->isIntegralType(*Context)) { 281264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_IntegralToPointer; 281364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (userExpr->getType()->isBlockPointerType()) { 281464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BlockPointerToObjCPointerCast; 281564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (userExpr->getType()->isPointerType()) { 281664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_CPointerToObjCPointerCast; 281764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 281864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 281964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 282064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 282164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK, userExpr); 282264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 282364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 282464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(userExpr); 282564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We've transferred the ownership to MsgExprs. For now, we *don't* null 282664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // out the argument in the original expression (since we aren't deleting 282764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the ObjCMessageExpr). See RewritePropertyOrImplicitSetter() usage for more info. 282864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //Exp->setArg(i, 0); 282964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 283064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the funky cast. 283164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *cast; 283264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 283364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType returnType; 283464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 283564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push 'id' and 'SEL', the 2 implicit arguments. 283664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendFlavor == MsgSendSuperFunctionDecl) 283764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getPointerType(getSuperStructType())); 283864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 283964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getObjCIdType()); 284064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getObjCSelType()); 284164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *OMD = Exp->getMethodDecl()) { 284264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push any user argument types. 284364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(), 284464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = OMD->param_end(); PI != E; ++PI) { 284564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = (*PI)->getType()->isObjCQualifiedIdType() 284664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? Context->getObjCIdType() 284764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : (*PI)->getType(); 284864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 284964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(t); 285064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 285164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 285264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType = Exp->getType(); 285364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian convertToUnqualifiedObjCType(returnType); 285464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(returnType); 285564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 285664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType = Context->getObjCIdType(); 285764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 285864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 285964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = MsgSendFlavor->getType(); 286064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 286164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 2862f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(MsgSendFlavor, false, msgSendType, 286364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, SourceLocation()); 286464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 286564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to cast objc_msgSend to "void *" (to workaround a GCC bandaid). 286664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we don't do this cast, we get the following bizarre warning/note: 286764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // xx.m:13: warning: function called through a non-compatible type 286864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // xx.m:13: note: if this code is reached, the program will abort 286964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, 287064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidTy), 287164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, DRE); 287264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 287364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the "normal" pointer to function cast. 287464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castType = 287564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 287664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we don't have a method decl, force a variadic cast. 287764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : true); 287864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = Context->getPointerType(castType); 287964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 288064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast); 288164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 288264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 288364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(StartLoc, EndLoc, cast); 288464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 288564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 288664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 288764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.size(), 288864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getResultType(), VK_RValue, 288964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 289064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = CE; 289164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) { 289264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We have the method which returns a struct/union. Must also generate 289364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // call to objc_msgSend_stret and hang both varieties on a conditional 289464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression which dictate which one to envoke depending on size of 289564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // method's return type. 289664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 289764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend_stret() declaration. 2898f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *STDRE = new (Context) DeclRefExpr(MsgSendStretFlavor, 2899f4b88a45902af1802a1cb42ba48b1c474474f228John McCall false, msgSendType, 290064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, SourceLocation()); 290164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to cast objc_msgSend_stret to "void *" (see above comment). 290264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, 290364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidTy), 290464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, STDRE); 290564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the "normal" pointer to function cast. 290664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 290764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : false); 290864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = Context->getPointerType(castType); 290964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 291064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast); 291164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 291264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 291364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), cast); 291464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 291564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT = msgSendType->getAs<FunctionType>(); 291664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *STCE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 291764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.size(), 291864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getResultType(), VK_RValue, 291964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 292064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 292164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build sizeof(returnType) 292264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UnaryExprOrTypeTraitExpr *sizeofExpr = 292364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) UnaryExprOrTypeTraitExpr(UETT_SizeOf, 292464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getTrivialTypeSourceInfo(returnType), 292564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getSizeType(), SourceLocation(), 292664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 292764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...)) 292864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Value of 8 is base on ppc32/x86 ABI for the most common cases. 292964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For X86 it is more complicated and some kind of target specific routine 293064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // is needed to decide what to do. 293164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 293264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 293364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IntegerLiteral *limit = IntegerLiteral::Create(*Context, 293464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::APInt(IntSize, 8), 293564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->IntTy, 293664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 293764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BinaryOperator *lessThanExpr = 293864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) BinaryOperator(sizeofExpr, limit, BO_LE, Context->IntTy, 293964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 294064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...)) 294164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConditionalOperator *CondExpr = 294264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) ConditionalOperator(lessThanExpr, 294364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), CE, 294464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), STCE, 294564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType, VK_RValue, OK_Ordinary); 294664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacingStmt = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 294764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CondExpr); 294864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 294964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 295064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return ReplacingStmt; 295164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 295264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 295364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteMessageExpr(ObjCMessageExpr *Exp) { 295464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = SynthMessageExpr(Exp, Exp->getLocStart(), 295564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getLocEnd()); 295664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 295764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the actual rewrite. 295864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, ReplacingStmt); 295964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 296064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 296164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return ReplacingStmt; 296264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 296364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 296464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// typedef struct objc_object Protocol; 296564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getProtocolType() { 296664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ProtocolTypeDecl) { 296764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *TInfo 296864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(Context->getObjCIdType()); 296964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolTypeDecl = TypedefDecl::Create(*Context, TUDecl, 297064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 297164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("Protocol"), 297264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TInfo); 297364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 297464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTypeDeclType(ProtocolTypeDecl); 297564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 297664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 297764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCProtocolExpr - Rewrite a protocol expression into 297864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// a synthesized/forward data reference (to the protocol's metadata). 297964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// The forward references (and metadata) are generated in 298064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteModernObjC::HandleTranslationUnit(). 298164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp) { 298230650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian std::string Name = "_OBJC_PROTOCOL_REFERENCE_$_" + 298330650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Exp->getProtocol()->getNameAsString(); 298464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *ID = &Context->Idents.get(Name); 298564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *VD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 298664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), ID, getProtocolType(), 0, 298764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, SC_None); 2988f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(VD, false, getProtocolType(), 2989f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 299064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *DerefExpr = new (Context) UnaryOperator(DRE, UO_AddrOf, 299164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(DRE->getType()), 299264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 299364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, DerefExpr->getType(), 299464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 299564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DerefExpr); 299664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, castExpr); 299764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolExprDecls.insert(Exp->getProtocol()->getCanonicalDecl()); 299864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 299964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return castExpr; 300064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 300164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 300264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 300364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::BufferContainsPPDirectives(const char *startBuf, 300464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf) { 300564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (startBuf < endBuf) { 300664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '#') { 300764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Skip whitespace. 300864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (++startBuf; startBuf[0] == ' ' || startBuf[0] == '\t'; ++startBuf) 300964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ; 301064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strncmp(startBuf, "if", strlen("if")) || 301164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "ifdef", strlen("ifdef")) || 301264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "ifndef", strlen("ifndef")) || 301364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "define", strlen("define")) || 301464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "undef", strlen("undef")) || 301564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "else", strlen("else")) || 301664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "elif", strlen("elif")) || 301764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "endif", strlen("endif")) || 301864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "pragma", strlen("pragma")) || 301964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "include", strlen("include")) || 302064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "import", strlen("import")) || 302164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "include_next", strlen("include_next"))) 302264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 302364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 302464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 302564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 302664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 302764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 302864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 302997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian/// RewriteObjCFieldDeclType - This routine rewrites a type into the buffer. 303015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian/// It handles elaborated types, as well as enum types in the process. 303197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanianbool RewriteModernObjC::RewriteObjCFieldDeclType(QualType &Type, 303297c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian std::string &Result) { 303397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian if (Type->isArrayType()) { 303497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian QualType ElemTy = Context->getBaseElementType(Type); 303597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return RewriteObjCFieldDeclType(ElemTy, Result); 303697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 303797c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian else if (Type->isRecordType()) { 303815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian RecordDecl *RD = Type->getAs<RecordType>()->getDecl(); 303915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (RD->isCompleteDefinition()) { 304015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (RD->isStruct()) 304115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\n\tstruct "; 304215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian else if (RD->isUnion()) 304315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\n\tunion "; 304415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian else 304515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian assert(false && "class not allowed as an ivar type"); 304697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 304715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += RD->getName(); 304815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (TagsDefinedInIvarDecls.count(RD)) { 304915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian // This struct is already defined. Do not write its definition again. 305097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += " "; 305197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 305215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 305315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian TagsDefinedInIvarDecls.insert(RD); 305415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += " {\n"; 305515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian for (RecordDecl::field_iterator i = RD->field_begin(), 305697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian e = RD->field_end(); i != e; ++i) { 305715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian FieldDecl *FD = *i; 305815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian RewriteObjCFieldDecl(FD, Result); 305915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 306015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t} "; 306197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 306215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 306315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 306415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian else if (Type->isEnumeralType()) { 306515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian EnumDecl *ED = Type->getAs<EnumType>()->getDecl(); 306615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (ED->isCompleteDefinition()) { 306715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\n\tenum "; 306815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += ED->getName(); 306915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (TagsDefinedInIvarDecls.count(ED)) { 307015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian // This enum is already defined. Do not write its definition again. 307197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += " "; 307297c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 307315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 307415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian TagsDefinedInIvarDecls.insert(ED); 307515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 307615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += " {\n"; 307715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian for (EnumDecl::enumerator_iterator EC = ED->enumerator_begin(), 307815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian ECEnd = ED->enumerator_end(); EC != ECEnd; ++EC) { 307915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t"; Result += EC->getName(); Result += " = "; 308015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian llvm::APSInt Val = EC->getInitVal(); 308115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += Val.toString(10); 308215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += ",\n"; 308315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 308415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t} "; 308597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return true; 308615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 308715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 308815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 308915f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += "\t"; 309015f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian convertObjCTypeToCStyleType(Type); 309197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian return false; 309297c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian} 309397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 309497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 309597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian/// RewriteObjCFieldDecl - This routine rewrites a field into the buffer. 309697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian/// It handles elaborated types, as well as enum types in the process. 309797c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCFieldDecl(FieldDecl *fieldDecl, 309897c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian std::string &Result) { 309997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian QualType Type = fieldDecl->getType(); 310097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian std::string Name = fieldDecl->getNameAsString(); 310115f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 310297c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian bool EleboratedType = RewriteObjCFieldDeclType(Type, Result); 310397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian if (!EleboratedType) 310497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Type.getAsStringInternal(Name, Context->getPrintingPolicy()); 310515f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += Name; 310615f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian if (fieldDecl->isBitField()) { 310715f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += " : "; Result += utostr(fieldDecl->getBitWidthValue(*Context)); 310815f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian } 310997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian else if (EleboratedType && Type->isArrayType()) { 311097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian CanQualType CType = Context->getCanonicalType(Type); 311197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian while (isa<ArrayType>(CType)) { 311297c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian if (const ConstantArrayType *CAT = Context->getAsConstantArrayType(CType)) { 311397c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += "["; 311497c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian llvm::APInt Dim = CAT->getSize(); 311597c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += utostr(Dim.getZExtValue()); 311697c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian Result += "]"; 311797c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 311897c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian CType = CType->getAs<ArrayType>()->getElementType(); 311997c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 312097c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian } 312197c1fd651c24638a4989b4e1f8eb2e629c2073d1Fariborz Jahanian 312215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian Result += ";\n"; 312315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian} 312415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian 312564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCInternalStruct - Rewrite one internal struct corresponding to 312664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// an objective-c class with ivars. 312764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCInternalStruct(ObjCInterfaceDecl *CDecl, 312864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 312964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CDecl && "Class missing in SynthesizeObjCInternalStruct"); 313064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CDecl->getName() != "" && 313164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "Name missing in SynthesizeObjCInternalStruct"); 313264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *RCDecl = CDecl->getSuperClass(); 3133a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian SmallVector<ObjCIvarDecl *, 8> IVars; 3134a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian for (ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 31359a2105bc790b3d0d4d5edb77c617aeccc4187c0eFariborz Jahanian IVD; IVD = IVD->getNextIvar()) 3136a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian IVars.push_back(IVD); 31379a2105bc790b3d0d4d5edb77c617aeccc4187c0eFariborz Jahanian 313864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CDecl->getLocStart(); 313964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = CDecl->getEndOfDefinitionLoc(); 3140a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 314164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 314264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 3143a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 314464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If no ivars and no root or if its root, directly or indirectly, 314564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // have no ivars (thus not synthesized) then no need to synthesize this class. 3146a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian if ((!CDecl->isThisDeclarationADefinition() || IVars.size() == 0) && 314764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (!RCDecl || !ObjCSynthesizedStructs.count(RCDecl))) { 314864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endBuf += Lexer::MeasureTokenLength(LocEnd, *SM, LangOpts); 314964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, Result); 315064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 315164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 3152a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 315364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct "; 315464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 3155a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IMPL {\n"; 3156a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 3157f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (RCDecl && ObjCSynthesizedStructs.count(RCDecl)) { 3158a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "\tstruct "; Result += RCDecl->getNameAsString(); 3159a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IMPL "; Result += RCDecl->getNameAsString(); 3160a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IVARS;\n"; 3161a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian } 316215f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian TagsDefinedInIvarDecls.clear(); 316315f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian for (unsigned i = 0, e = IVars.size(); i < e; i++) 316415f87771cae0c31da45c05b34841d0a2ca70bc87Fariborz Jahanian RewriteObjCFieldDecl(IVars[i], Result); 31650b17b9a1b6ec0522dda5ddd8244844eb9ab1cc1cFariborz Jahanian 3166a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "};\n"; 3167a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian endBuf += Lexer::MeasureTokenLength(LocEnd, *SM, LangOpts); 3168a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, Result); 3169f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // Mark this struct as having been generated. 3170f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (!ObjCSynthesizedStructs.insert(CDecl)) 3171f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian llvm_unreachable("struct already synthesize- RewriteObjCInternalStruct"); 317264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 317364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 31747cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanianstatic void WriteInternalIvarName(ObjCInterfaceDecl *IDecl, 31757cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian ObjCIvarDecl *IvarDecl, std::string &Result) { 31767cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += "OBJC_IVAR_$_"; 31777cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += IDecl->getName(); 31787cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += "$"; 31797cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += IvarDecl->getName(); 31807cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian} 31817cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian 318272c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian/// RewriteIvarOffsetSymbols - Rewrite ivar offset symbols of those ivars which 318372c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian/// have been referenced in an ivar access expression. 318472c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanianvoid RewriteModernObjC::RewriteIvarOffsetSymbols(ObjCInterfaceDecl *CDecl, 318572c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian std::string &Result) { 318672c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian // write out ivar offset symbols which have been referenced in an ivar 318772c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian // access expression. 318872c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian llvm::SmallPtrSet<ObjCIvarDecl *, 8> Ivars = ReferencedIvars[CDecl]; 318972c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian if (Ivars.empty()) 319072c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian return; 319172c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian for (llvm::SmallPtrSet<ObjCIvarDecl *, 8>::iterator i = Ivars.begin(), 319272c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian e = Ivars.end(); i != e; i++) { 319372c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian ObjCIvarDecl *IvarDecl = (*i); 319440a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Result += "\n"; 319540a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (LangOpts.MicrosoftExt) 319640a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Result += "__declspec(allocate(\".objc_ivar$B\")) "; 319740a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (LangOpts.MicrosoftExt && 319840a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian IvarDecl->getAccessControl() != ObjCIvarDecl::Private && 319940a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian IvarDecl->getAccessControl() != ObjCIvarDecl::Package) { 320040a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian const ObjCInterfaceDecl *CDecl = IvarDecl->getContainingInterface(); 320140a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (CDecl->getImplementation()) 320240a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Result += "__declspec(dllexport) "; 320340a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian } 32047cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += "extern unsigned long "; 32057cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(CDecl, IvarDecl, Result); 32067cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += ";"; 320772c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian } 320872c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian} 320972c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 321064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 321164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Meta Data Emission 321264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 321364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 321464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 321564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteImplementations - This routine rewrites all method implementations 321664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// and emits meta-data. 321764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 321864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteImplementations() { 321964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int ClsDefCount = ClassImplementation.size(); 322064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int CatDefCount = CategoryImplementation.size(); 322164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 322264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite implemented methods 32238f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) { 32248f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian ObjCImplementationDecl *OIMP = ClassImplementation[i]; 32258f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian ObjCInterfaceDecl *CDecl = OIMP->getClassInterface(); 32268f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian if (CDecl->isImplicitInterfaceDecl()) 3227cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian assert(false && 3228cf4c60feae11b2f6a135d7cd8ecf97a5040951c3Fariborz Jahanian "Legacy implicit interface rewriting not supported in moder abi"); 32298f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian RewriteImplementationDecl(OIMP); 32308f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian } 323164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 32328c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian for (int i = 0; i < CatDefCount; i++) { 32338c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian ObjCCategoryImplDecl *CIMP = CategoryImplementation[i]; 32348c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian ObjCInterfaceDecl *CDecl = CIMP->getClassInterface(); 32358c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian if (CDecl->isImplicitInterfaceDecl()) 32368c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian assert(false && 32378c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian "Legacy implicit interface rewriting not supported in moder abi"); 32388c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian RewriteImplementationDecl(CIMP); 32398c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian } 324064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 324164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 324264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteByRefString(std::string &ResultStr, 324364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Name, 324464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD, bool def) { 324564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BlockByRefDeclNo.count(VD) && 324664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "RewriteByRefString: ByRef decl missing"); 324764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (def) 324864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "struct "; 324964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "__Block_byref_" + Name + 325064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "_" + utostr(BlockByRefDeclNo[VD]) ; 325164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 325264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 325364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool HasLocalVariableExternalStorage(ValueDecl *VD) { 325464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *Var = dyn_cast<VarDecl>(VD)) 325564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return (Var->isFunctionOrMethodVarDecl() && !Var->hasLocalStorage()); 325664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 325764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 325864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 325964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockFunc(BlockExpr *CE, int i, 326064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, 326164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag) { 326264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *AFT = CE->getFunctionType(); 326364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType RT = AFT->getResultType(); 326464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StructRef = "struct " + Tag; 326564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "static " + RT.getAsString(Context->getPrintingPolicy()) + " __" + 326664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian funcName.str() + "_" + "block_func_" + utostr(i); 326764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 326864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDecl *BD = CE->getBlockDecl(); 326964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 327064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<FunctionNoProtoType>(AFT)) { 327164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // No user-supplied arguments. Still need to pass in a pointer to the 327264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block (to reference imported block decl refs). 327364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef + " *__cself)"; 327464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (BD->param_empty()) { 327564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef + " *__cself)"; 327664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 327764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FT = cast<FunctionProtoType>(AFT); 327864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(FT && "SynthesizeBlockFunc: No function proto"); 327964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += '('; 328064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // first add the implicit argument. 328164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += StructRef + " *__cself, "; 328264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr; 328364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (BlockDecl::param_iterator AI = BD->param_begin(), 328464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BD->param_end(); AI != E; ++AI) { 328564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (AI != BD->param_begin()) S += ", "; 328664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParamStr = (*AI)->getNameAsString(); 328764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*AI)->getType(); 328864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (convertBlockPointerToFunctionPointer(QT)) 328964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(ParamStr, Context->getPrintingPolicy()); 329064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 329164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(ParamStr, Context->getPrintingPolicy()); 329264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ParamStr; 329364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 329464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 329564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BD->param_empty()) S += ", "; 329664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "..."; 329764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 329864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ')'; 329964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 330064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n"; 330164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 330264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create local declarations to avoid rewriting all closure decl ref exprs. 330364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // First, emit a declaration for all "by ref" decls. 330464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 330564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 330664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 330764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 330864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString; 330964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(TypeString, Name, (*I)); 331064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeString += " *"; 331164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = TypeString + Name; 331264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Name + " = __cself->" + (*I)->getNameAsString() + "; // bound by ref\n"; 331364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 331464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Next, emit a declaration for all "by copy" declarations. 331564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 331664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 331764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 331864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle nested closure invocation. For example: 331964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 332064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^myImportedClosure)(void); 332164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedClosure = ^(void) { setGlobalInt(x + y); }; 332264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 332364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^anotherClosure)(void); 332464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // anotherClosure = ^(void) { 332564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedClosure(); // import and invoke the closure 332664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // }; 332764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 332864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) { 332964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerTypeVariable(S, (*I)); 333064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " = ("; 333164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(S, (*I)->getType()); 333264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ")"; 333364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "__cself->" + (*I)->getNameAsString() + "; // bound by copy\n"; 333464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 333564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 333664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 333764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 333864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) 333964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 334064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(Name, Context->getPrintingPolicy()); 334164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Name + " = __cself->" + 334264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*I)->getNameAsString() + "; // bound by copy\n"; 334364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 334464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 334564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RewrittenStr = RewrittenBlockExprs[CE]; 334664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *cstr = RewrittenStr.c_str(); 334764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*cstr++ != '{') ; 334864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += cstr; 334964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "\n"; 335064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 335164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 335264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 335364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockHelperFuncs(BlockExpr *CE, int i, 335464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, 335564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag) { 335664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StructRef = "struct " + Tag; 335764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "static void __"; 335864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 335964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += funcName; 336064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_block_copy_" + utostr(i); 336164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef; 336264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*dst, " + StructRef; 336364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*src) {"; 336464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ValueDecl*,8>::iterator I = ImportedBlockDecls.begin(), 336564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ImportedBlockDecls.end(); I != E; ++I) { 336664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = (*I); 336764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_Block_object_assign((void*)&dst->"; 336864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 336964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", (void*)src->"; 337064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 337164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockByRefDeclsPtrSet.count((*I))) 337264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BYREF) + "/*BLOCK_FIELD_IS_BYREF*/);"; 337364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isBlockPointerType()) 337464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BLOCK) + "/*BLOCK_FIELD_IS_BLOCK*/);"; 337564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 337664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_OBJECT) + "/*BLOCK_FIELD_IS_OBJECT*/);"; 337764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 337864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "}\n"; 337964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 338064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "\nstatic void __"; 338164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += funcName; 338264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_block_dispose_" + utostr(i); 338364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef; 338464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*src) {"; 338564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ValueDecl*,8>::iterator I = ImportedBlockDecls.begin(), 338664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ImportedBlockDecls.end(); I != E; ++I) { 338764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = (*I); 338864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_Block_object_dispose((void*)src->"; 338964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 339064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockByRefDeclsPtrSet.count((*I))) 339164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BYREF) + "/*BLOCK_FIELD_IS_BYREF*/);"; 339264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isBlockPointerType()) 339364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BLOCK) + "/*BLOCK_FIELD_IS_BLOCK*/);"; 339464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 339564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_OBJECT) + "/*BLOCK_FIELD_IS_OBJECT*/);"; 339664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 339764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "}\n"; 339864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 339964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 340064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 340164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockImpl(BlockExpr *CE, std::string Tag, 340264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Desc) { 340364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "\nstruct " + Tag; 340464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Constructor = " " + Tag; 340564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 340664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n struct __block_impl impl;\n"; 340764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " struct " + Desc; 340864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "* Desc;\n"; 340964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 341064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "(void *fp, "; // Invoke function pointer. 341164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "struct " + Desc; // Descriptor pointer. 341264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " *desc"; 341364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 341464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 341564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by copy" declarations. 341664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 341764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 341864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 341964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FieldName = (*I)->getNameAsString(); 342064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ArgName = "_" + FieldName; 342164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle nested closure invocation. For example: 342264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 342364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^myImportedBlock)(void); 342464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedBlock = ^(void) { setGlobalInt(x + y); }; 342564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 342664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^anotherBlock)(void); 342764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // anotherBlock = ^(void) { 342864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedBlock(); // import and invoke the closure 342964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // }; 343064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 343164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) { 343264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "struct __block_impl *"; 343364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", void *" + ArgName; 343464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 343564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 343664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) 343764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 343864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(FieldName, Context->getPrintingPolicy()); 343964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(ArgName, Context->getPrintingPolicy()); 344064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", " + ArgName; 344164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 344264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += FieldName + ";\n"; 344364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 344464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by ref" declarations. 344564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 344664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 344764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 344864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FieldName = (*I)->getNameAsString(); 344964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ArgName = "_" + FieldName; 345064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 345164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString; 345264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(TypeString, FieldName, (*I)); 345364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeString += " *"; 345464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldName = TypeString + FieldName; 345564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgName = TypeString + ArgName; 345664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", " + ArgName; 345764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 345864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += FieldName + "; // by ref\n"; 345964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 346064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Finish writing the constructor. 346164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", int flags=0)"; 346264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize all "by copy" arguments. 346364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool firsTime = true; 346464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 346564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 346664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 346764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (firsTime) { 346864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " : "; 346964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian firsTime = false; 347064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 347164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 347264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", "; 347364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) 347464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "((struct __block_impl *)_" + Name + ")"; 347564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 347664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "(_" + Name + ")"; 347764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 347864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize all "by ref" arguments. 347964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 348064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 348164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 348264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (firsTime) { 348364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " : "; 348464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian firsTime = false; 348564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 348664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 348764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", "; 348864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "(_" + Name + "->__forwarding)"; 348964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 349064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 349164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " {\n"; 349264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl) 349364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteGlobalBlock;\n"; 349464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 349564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteStackBlock;\n"; 349664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.Flags = flags;\n impl.FuncPtr = fp;\n"; 349764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 349864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " Desc = desc;\n"; 349964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 350064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Finish writing the constructor. 350164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", int flags=0) {\n"; 350264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl) 350364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteGlobalBlock;\n"; 350464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 350564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteStackBlock;\n"; 350664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.Flags = flags;\n impl.FuncPtr = fp;\n"; 350764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " Desc = desc;\n"; 350864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 350964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " "; 351064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "}\n"; 351164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Constructor; 351264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "};\n"; 351364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 351464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 351564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 351664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockDescriptor(std::string DescTag, 351764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag, int i, 351864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName, 351964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned hasCopy) { 352064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "\nstatic struct " + DescTag; 352164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 352264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n unsigned long reserved;\n"; 352364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " unsigned long Block_size;\n"; 352464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasCopy) { 352564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " void (*copy)(struct "; 352664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*, struct "; 352764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*);\n"; 352864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 352964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " void (*dispose)(struct "; 353064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*);\n"; 353164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 353264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "} "; 353364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 353464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += DescTag + "_DATA = { 0, sizeof(struct "; 353564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag + ")"; 353664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasCopy) { 353764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", __" + FunName.str() + "_block_copy_" + utostr(i); 353864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", __" + FunName.str() + "_block_dispose_" + utostr(i); 353964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 354064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "};\n"; 354164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 354264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 354364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 354464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthesizeBlockLiterals(SourceLocation FunLocStart, 354564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName) { 354664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert declaration for the function in which block literal is used. 354764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDeclToDeclareForBlock && !Blocks.empty()) 354864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockLiteralFunctionDecl(CurFunctionDeclToDeclareForBlock); 354964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool RewriteSC = (GlobalVarDecl && 355064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !Blocks.empty() && 355164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl->getStorageClass() == SC_Static && 355264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl->getType().getCVRQualifiers()); 355364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RewriteSC) { 355464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SC(" void __"); 355564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += GlobalVarDecl->getNameAsString(); 355664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "() {}"; 355764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, SC); 355864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 355964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 356064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert closures that were part of the function. 356164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, count=0; i < Blocks.size(); i++) { 356264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CollectBlockDeclRefInfo(Blocks[i]); 356364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to copy-in the inner copied-in variables not actually used in this 356464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block. 356564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int j = 0; j < InnerDeclRefsCount[i]; j++) { 3566f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *Exp = InnerDeclRefs[count++]; 356764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = Exp->getDecl(); 356864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 3569f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!VD->hasAttr<BlocksAttr>()) { 3570f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!BlockByCopyDeclsPtrSet.count(VD)) { 3571f4b88a45902af1802a1cb42ba48b1c474474f228John McCall BlockByCopyDeclsPtrSet.insert(VD); 3572f4b88a45902af1802a1cb42ba48b1c474474f228John McCall BlockByCopyDecls.push_back(VD); 3573f4b88a45902af1802a1cb42ba48b1c474474f228John McCall } 3574f4b88a45902af1802a1cb42ba48b1c474474f228John McCall continue; 357564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 3576f4b88a45902af1802a1cb42ba48b1c474474f228John McCall 3577f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!BlockByRefDeclsPtrSet.count(VD)) { 357864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(VD); 357964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(VD); 358064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 3581f4b88a45902af1802a1cb42ba48b1c474474f228John McCall 358264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // imported objects in the inner blocks not used in the outer 358364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // blocks must be copied/disposed in the outer block as well. 3584f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (VD->getType()->isObjCObjectPointerType() || 358564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD->getType()->isBlockPointerType()) 358664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(VD); 358764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 358864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 358964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag = "__" + FunName.str() + "_block_impl_" + utostr(i); 359064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string DescTag = "__" + FunName.str() + "_block_desc_" + utostr(i); 359164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 359264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string CI = SynthesizeBlockImpl(Blocks[i], ImplTag, DescTag); 359364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 359464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, CI); 359564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 359664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string CF = SynthesizeBlockFunc(Blocks[i], i, FunName, ImplTag); 359764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 359864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, CF); 359964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 360064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImportedBlockDecls.size()) { 360164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string HF = SynthesizeBlockHelperFuncs(Blocks[i], i, FunName, ImplTag); 360264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, HF); 360364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 360464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string BD = SynthesizeBlockDescriptor(DescTag, ImplTag, i, FunName, 360564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.size() > 0); 360664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, BD); 360764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 360864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.clear(); 360964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.clear(); 361064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.clear(); 361164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.clear(); 361264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.clear(); 361364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.clear(); 361464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 361564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RewriteSC) { 361664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Must insert any 'const/volatile/static here. Since it has been 361764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // removed as result of rewriting of block literals. 361864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SC; 361964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getStorageClass() == SC_Static) 362064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC = "static "; 362164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isConstQualified()) 362264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "const "; 362364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isVolatileQualified()) 362464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "volatile "; 362564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isRestrictQualified()) 362664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "restrict "; 362764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, SC); 362864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 362964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 363064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Blocks.clear(); 363164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefsCount.clear(); 363264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.clear(); 363364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewrittenBlockExprs.clear(); 363464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 363564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 363664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InsertBlockLiteralsWithinFunction(FunctionDecl *FD) { 363764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart = FD->getTypeSpecStartLoc(); 363864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FuncName = FD->getName(); 363964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 364064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(FunLocStart, FuncName); 364164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 364264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 364364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void BuildUniqueMethodName(std::string &Name, 364464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MD) { 364564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *IFace = MD->getClassInterface(); 364664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = IFace->getName(); 364764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name += "__" + MD->getSelector().getAsString(); 364864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Convert colons to underscores. 364964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string::size_type loc = 0; 365064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while ((loc = Name.find(":", loc)) != std::string::npos) 365164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name.replace(loc, 1, "_"); 365264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 365364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 365464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InsertBlockLiteralsWithinMethod(ObjCMethodDecl *MD) { 365564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //fprintf(stderr,"In InsertBlockLiteralsWitinMethod\n"); 365664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //SourceLocation FunLocStart = MD->getLocStart(); 365764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart = MD->getLocStart(); 365864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FuncName; 365964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BuildUniqueMethodName(FuncName, MD); 366064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(FunLocStart, FuncName); 366164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 366264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 366364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetBlockDeclRefExprs(Stmt *S) { 366464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 366564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 366664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *CBE = dyn_cast<BlockExpr>(*CI)) 366764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(CBE->getBody()); 366864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 366964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(*CI); 367064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 367164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 3672f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) 3673f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (DRE->refersToEnclosingLocal() && 3674f4b88a45902af1802a1cb42ba48b1c474474f228John McCall HasLocalVariableExternalStorage(DRE->getDecl())) { 3675f4b88a45902af1802a1cb42ba48b1c474474f228John McCall BlockDeclRefs.push_back(DRE); 367664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 367764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 367864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 367964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 368064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 368164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetInnerBlockDeclRefExprs(Stmt *S, 3682f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs, 368364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> &InnerContexts) { 368464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 368564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 368664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *CBE = dyn_cast<BlockExpr>(*CI)) { 368764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts.insert(cast<DeclContext>(CBE->getBlockDecl())); 368864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(CBE->getBody(), 368964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, 369064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts); 369164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 369264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 369364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(*CI, 369464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, 369564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts); 369664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 369764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 369864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 3699f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) { 3700f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (DRE->refersToEnclosingLocal()) { 3701f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!isa<FunctionDecl>(DRE->getDecl()) && 3702f4b88a45902af1802a1cb42ba48b1c474474f228John McCall !InnerContexts.count(DRE->getDecl()->getDeclContext())) 3703f4b88a45902af1802a1cb42ba48b1c474474f228John McCall InnerBlockDeclRefs.push_back(DRE); 3704f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (VarDecl *Var = dyn_cast<VarDecl>(DRE->getDecl())) 3705f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (Var->isFunctionOrMethodVarDecl()) 3706f4b88a45902af1802a1cb42ba48b1c474474f228John McCall ImportedLocalExternalDecls.insert(Var); 3707f4b88a45902af1802a1cb42ba48b1c474474f228John McCall } 370864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 370964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 371064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 371164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 371264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 3713164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian/// convertObjCTypeToCStyleType - This routine converts such objc types 3714164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian/// as qualified objects, and blocks to their closest c/c++ types that 3715164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian/// it can. It returns true if input type was modified. 3716164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanianbool RewriteModernObjC::convertObjCTypeToCStyleType(QualType &T) { 3717164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian QualType oldT = T; 3718164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian convertBlockPointerToFunctionPointer(T); 3719164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (T->isFunctionPointerType()) { 3720164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian QualType PointeeTy; 3721164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (const PointerType* PT = T->getAs<PointerType>()) { 3722164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian PointeeTy = PT->getPointeeType(); 3723164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (const FunctionType *FT = PointeeTy->getAs<FunctionType>()) { 3724164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian T = convertFunctionTypeOfBlocks(FT); 3725164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian T = Context->getPointerType(T); 3726164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian } 3727164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian } 3728164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian } 3729164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian 3730164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian convertToUnqualifiedObjCType(T); 3731164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian return T != oldT; 3732164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian} 3733164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian 373464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// convertFunctionTypeOfBlocks - This routine converts a function type 373564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// whose result type may be a block pointer or whose argument type(s) 373664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// might be block pointers to an equivalent function type replacing 373764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// all block pointers to function pointers. 373864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::convertFunctionTypeOfBlocks(const FunctionType *FT) { 373964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP = dyn_cast<FunctionProtoType>(FT); 374064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FTP will be null for closures that don't take arguments. 374164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate a funky cast. 374264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 374364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Res = FT->getResultType(); 3744164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian bool modified = convertObjCTypeToCStyleType(Res); 374564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 374664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 374764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 374864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I && (I != E); ++I) { 374964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = *I; 375064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 3751164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (convertObjCTypeToCStyleType(t)) 3752164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian modified = true; 375364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 375464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 375564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 375664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FuncType; 3757164d6f8c7e040cb766b071ce92320a18dafbece1Fariborz Jahanian if (modified) 375864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncType = getSimpleFunctionType(Res, &ArgTypes[0], ArgTypes.size()); 375964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else FuncType = QualType(FT, 0); 376064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return FuncType; 376164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 376264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 376364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthesizeBlockCall(CallExpr *Exp, const Expr *BlockExp) { 376464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Navigate to relevant type information. 376564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *CPT = 0; 376664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 376764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(BlockExp)) { 376864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = DRE->getType()->getAs<BlockPointerType>(); 376964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const MemberExpr *MExpr = dyn_cast<MemberExpr>(BlockExp)) { 377064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = MExpr->getType()->getAs<BlockPointerType>(); 377164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 377264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ParenExpr *PRE = dyn_cast<ParenExpr>(BlockExp)) { 377364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return SynthesizeBlockCall(Exp, PRE->getSubExpr()); 377464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 377564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ImplicitCastExpr *IEXPR = dyn_cast<ImplicitCastExpr>(BlockExp)) 377664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = IEXPR->getType()->getAs<BlockPointerType>(); 377764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ConditionalOperator *CEXPR = 377864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ConditionalOperator>(BlockExp)) { 377964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *LHSExp = CEXPR->getLHS(); 378064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *LHSStmt = SynthesizeBlockCall(Exp, LHSExp); 378164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *RHSExp = CEXPR->getRHS(); 378264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RHSStmt = SynthesizeBlockCall(Exp, RHSExp); 378364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *CONDExp = CEXPR->getCond(); 378464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConditionalOperator *CondExpr = 378564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) ConditionalOperator(CONDExp, 378664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), cast<Expr>(LHSStmt), 378764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), cast<Expr>(RHSStmt), 378864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getType(), VK_RValue, OK_Ordinary); 378964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CondExpr; 379064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const ObjCIvarRefExpr *IRE = dyn_cast<ObjCIvarRefExpr>(BlockExp)) { 379164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = IRE->getType()->getAs<BlockPointerType>(); 379264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const PseudoObjectExpr *POE 379364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = dyn_cast<PseudoObjectExpr>(BlockExp)) { 379464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = POE->getType()->castAs<BlockPointerType>(); 379564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 379664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(1 && "RewriteBlockClass: Bad type"); 379764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 379864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CPT && "RewriteBlockClass: Bad type"); 379964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = CPT->getPointeeType()->getAs<FunctionType>(); 380064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(FT && "RewriteBlockClass: Bad type"); 380164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP = dyn_cast<FunctionProtoType>(FT); 380264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FTP will be null for closures that don't take arguments. 380364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 380464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 380564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 380664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__block_impl")); 380764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PtrBlock = Context->getPointerType(Context->getTagDeclType(RD)); 380864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 380964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate a funky cast. 381064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 381164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 381264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push the block argument type. 381364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(PtrBlock); 381464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 381564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 381664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I && (I != E); ++I) { 381764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = *I; 381864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 381964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!convertBlockPointerToFunctionPointer(t)) 382064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian convertToUnqualifiedObjCType(t); 382164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 382264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 382364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 382464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the pointer to function cast. 382564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PtrToFuncCastType 382664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = getSimpleFunctionType(Exp->getType(), &ArgTypes[0], ArgTypes.size()); 382764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 382864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PtrToFuncCastType = Context->getPointerType(PtrToFuncCastType); 382964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 383064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *BlkCast = NoTypeInfoCStyleCastExpr(Context, PtrBlock, 383164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 383264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const_cast<Expr*>(BlockExp)); 383364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 383464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 383564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkCast); 383664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //PE->dump(); 383764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 383864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = FieldDecl::Create(*Context, 0, SourceLocation(), 383964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 384064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("FuncPtr"), 384164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 384264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 384364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 384464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MemberExpr *ME = new (Context) MemberExpr(PE, true, FD, SourceLocation(), 384564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType(), VK_LValue, 384664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary); 384764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 384864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 384964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *FunkCast = NoTypeInfoCStyleCastExpr(Context, PtrToFuncCastType, 385064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, ME); 385164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), FunkCast); 385264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 385364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> BlkExprs; 385464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add the implicit argument. 385564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.push_back(BlkCast); 385664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add the user arguments. 385764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (CallExpr::arg_iterator I = Exp->arg_begin(), 385864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = Exp->arg_end(); I != E; ++I) { 385964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.push_back(*I); 386064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 386164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &BlkExprs[0], 386264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.size(), 386364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getType(), VK_RValue, 386464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 386564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CE; 386664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 386764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 386864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// We need to return the rewritten expression to handle cases where the 3869f4b88a45902af1802a1cb42ba48b1c474474f228John McCall// DeclRefExpr is embedded in another expression being rewritten. 387064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// For example: 387164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 387264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// int main() { 387364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// __block Foo *f; 387464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// __block int i; 387564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 387664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// void (^myblock)() = ^() { 3877f4b88a45902af1802a1cb42ba48b1c474474f228John McCall// [f test]; // f is a DeclRefExpr embedded in a message (which is being rewritten). 387864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// i = 77; 387964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// }; 388064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//} 3881f4b88a45902af1802a1cb42ba48b1c474474f228John McCallStmt *RewriteModernObjC::RewriteBlockDeclRefExpr(DeclRefExpr *DeclRefExp) { 388264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the byref variable into BYREFVAR->__forwarding->BYREFVAR 388364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for each DeclRefExp where BYREFVAR is name of the variable. 3884f4b88a45902af1802a1cb42ba48b1c474474f228John McCall ValueDecl *VD = DeclRefExp->getDecl(); 3885f4b88a45902af1802a1cb42ba48b1c474474f228John McCall bool isArrow = DeclRefExp->refersToEnclosingLocal(); 388664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 388764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = FieldDecl::Create(*Context, 0, SourceLocation(), 388864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 388964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__forwarding"), 389064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 389164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 389264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 389364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MemberExpr *ME = new (Context) MemberExpr(DeclRefExp, isArrow, 389464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD, SourceLocation(), 389564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType(), VK_LValue, 389664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary); 389764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 389864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef Name = VD->getName(); 389964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = FieldDecl::Create(*Context, 0, SourceLocation(), SourceLocation(), 390064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get(Name), 390164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 390264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 390364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 390464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ME = new (Context) MemberExpr(ME, true, FD, SourceLocation(), 390564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExp->getType(), VK_LValue, OK_Ordinary); 390664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 390764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 390864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 390964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need parens to enforce precedence. 391064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(DeclRefExp->getExprLoc(), 391164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExp->getExprLoc(), 391264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ME); 391364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(DeclRefExp, PE); 391464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return PE; 391564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 391664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 391764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Rewrites the imported local variable V with external storage 391864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// (static, extern, etc.) as *V 391964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 392064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteLocalVariableExternalStorage(DeclRefExpr *DRE) { 392164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = DRE->getDecl(); 392264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *Var = dyn_cast<VarDecl>(VD)) 392364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ImportedLocalExternalDecls.count(Var)) 392464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return DRE; 392564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Exp = new (Context) UnaryOperator(DRE, UO_Deref, DRE->getType(), 392664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, OK_Ordinary, 392764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DRE->getLocation()); 392864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need parens to enforce precedence. 392964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 393064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp); 393164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(DRE, PE); 393264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return PE; 393364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 393464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 393564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteCastExpr(CStyleCastExpr *CE) { 393664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CE->getLParenLoc(); 393764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = CE->getRParenLoc(); 393864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 393964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to avoid trying to rewrite synthesized casts. 394064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 394164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 394264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to avoid trying to rewrite casts contained in macros. 394364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewriter::isRewritable(LocStart) || !Rewriter::isRewritable(LocEnd)) 394464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 394564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 394664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 394764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 394864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = CE->getType(); 394964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Type* TypePtr = QT->getAs<Type>(); 395064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<TypeOfExprType>(TypePtr)) { 395164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const TypeOfExprType *TypeOfExprTypePtr = cast<TypeOfExprType>(TypePtr); 395264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = TypeOfExprTypePtr->getUnderlyingExpr()->getType(); 395364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeAsString = "("; 395464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(TypeAsString, QT); 395564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeAsString += ")"; 395664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf+1, TypeAsString); 395764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 395864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 395964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // advance the location to startArgList. 396064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = startBuf; 396164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 396264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr++ && (argPtr < endBuf)) { 396364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 396464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 396564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*'. 396664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LocStart = LocStart.getLocWithOffset(argPtr-startBuf); 396764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 1, "*"); 396864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 396964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 397064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 397164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 397264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 397364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 397464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerFunctionArgs(FunctionDecl *FD) { 397564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = FD->getLocation(); 397664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned parenCount = 0; 397764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 397864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We have 1 or more arguments that have closure pointers. 397964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 398064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startArgList = strchr(startBuf, '('); 398164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 398264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startArgList == '(') && "Rewriter fuzzy parser confused"); 398364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 398464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount++; 398564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // advance the location to startArgList. 398664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = DeclLoc.getLocWithOffset(startArgList-startBuf); 398764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((DeclLoc.isValid()) && "Invalid DeclLoc"); 398864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 398964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = startArgList; 399064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 399164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr++ && parenCount) { 399264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 399364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 399464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*'. 399564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = DeclLoc.getLocWithOffset(argPtr-startArgList); 399664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, 1, "*"); 399764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 399864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': 399964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount++; 400064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 400164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': 400264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount--; 400364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 400464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 400564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 400664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 400764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 400864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 400964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::PointerTypeTakesAnyBlockArguments(QualType QT) { 401064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP; 401164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = QT->getAs<PointerType>(); 401264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT) { 401364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = PT->getPointeeType()->getAs<FunctionProtoType>(); 401464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 401564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = QT->getAs<BlockPointerType>(); 401664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BPT && "BlockPointerTypeTakeAnyBlockArguments(): not a block pointer type"); 401764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = BPT->getPointeeType()->getAs<FunctionProtoType>(); 401864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 401964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 402064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 402164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I != E; ++I) 402264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(*I)) 402364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 402464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 402564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 402664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 402764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 402864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::PointerTypeTakesAnyObjCQualifiedType(QualType QT) { 402964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP; 403064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = QT->getAs<PointerType>(); 403164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT) { 403264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = PT->getPointeeType()->getAs<FunctionProtoType>(); 403364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 403464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = QT->getAs<BlockPointerType>(); 403564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BPT && "BlockPointerTypeTakeAnyBlockArguments(): not a block pointer type"); 403664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = BPT->getPointeeType()->getAs<FunctionProtoType>(); 403764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 403864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 403964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 404064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I != E; ++I) { 404164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*I)->isObjCQualifiedIdType()) 404264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 404364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*I)->isObjCObjectPointerType() && 404464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*I)->getPointeeType()->isObjCQualifiedInterfaceType()) 404564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 404664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 404764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 404864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 404964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 405064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 405164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 405264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetExtentOfArgList(const char *Name, const char *&LParen, 405364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&RParen) { 405464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = strchr(Name, '('); 405564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*argPtr == '(') && "Rewriter fuzzy parser confused"); 405664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 405764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LParen = argPtr; // output the start. 405864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; // skip past the left paren. 405964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned parenCount = 1; 406064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 406164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr && parenCount) { 406264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 406364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': parenCount++; break; 406464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': parenCount--; break; 406564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: break; 406664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 406764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (parenCount) argPtr++; 406864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 406964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*argPtr == ')') && "Rewriter fuzzy parser confused"); 407064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RParen = argPtr; // output the end 407164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 407264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 407364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerDecl(NamedDecl *ND) { 407464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) { 407564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerFunctionArgs(FD); 407664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 407764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 407864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle Variables and Typedefs. 407964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getLocation(); 408064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType DeclT; 408164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(ND)) 408264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = VD->getType(); 408364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TypedefNameDecl *TDD = dyn_cast<TypedefNameDecl>(ND)) 408464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = TDD->getUnderlyingType(); 408564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FieldDecl *FD = dyn_cast<FieldDecl>(ND)) 408664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = FD->getType(); 408764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 408864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("RewriteBlockPointerDecl(): Decl type not yet handled"); 408964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 409064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 409164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = startBuf; 409264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scan backward (from the decl location) for the end of the previous decl. 409364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != '^' && *startBuf != ';' && startBuf != MainFileStart) 409464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf--; 409564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Start = DeclLoc.getLocWithOffset(startBuf-endBuf); 409664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 409764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned OrigLength=0; 409864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // *startBuf != '^' if we are dealing with a pointer to function that 409964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // may take block argument types (which will be handled below). 410064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '^') { 410164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*', computing a negative offset. 410264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = '*'; 410364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 410464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 410564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 410664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != ')') { 410764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *startBuf; 410864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 410964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 411064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 411164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ')'; 411264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 411364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 411464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PointerTypeTakesAnyBlockArguments(DeclT) || 411564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointerTypeTakesAnyObjCQualifiedType(DeclT)) { 411664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*' for arguments. 411764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace id<P> with id/*<>*/ 411864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = ND->getLocation(); 411964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = SM->getCharacterData(DeclLoc); 412064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argListBegin, *argListEnd; 412164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetExtentOfArgList(startBuf, argListBegin, argListEnd); 412264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (argListBegin < argListEnd) { 412364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*argListBegin == '^') 412464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += '*'; 412564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*argListBegin == '<') { 412664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "/*"; 412764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin++; 412864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++;; 412964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argListBegin != '>') { 413064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin++; 413164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 413264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 413364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin; 413464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "*/"; 413564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 413664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 413764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin; 413864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argListBegin++; 413964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 414064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 414164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ')'; 414264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 414364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 414464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(Start, OrigLength, buf); 414564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 414664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 414764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 414864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 414964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 415064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SynthesizeByrefCopyDestroyHelper - This routine synthesizes: 415164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void __Block_byref_id_object_copy(struct Block_byref_id_object *dst, 415264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct Block_byref_id_object *src) { 415364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_assign (&_dest->object, _src->object, 415464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_OBJECT 415564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // object 415664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_assign(&_dest->object, _src->object, 415764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_BLOCK 415864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // block 415964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 416064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// And: 416164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void __Block_byref_id_object_dispose(struct Block_byref_id_object *_src) { 416264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_dispose(_src->object, 416364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_OBJECT 416464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // object 416564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_dispose(_src->object, 416664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_BLOCK 416764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // block 416864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 416964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 417064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeByrefCopyDestroyHelper(VarDecl *VD, 417164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag) { 417264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S; 417364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CopyDestroyCache.count(flag)) 417464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 417564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CopyDestroyCache.insert(flag); 417664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S = "static void __Block_byref_id_object_copy_"; 417764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 417864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(void *dst, void *src) {\n"; 417964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 418064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // offset into the object pointer is computed as: 418164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void * + void* + int + int + void* + void * 418264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 418364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 418464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned VoidPtrSize = 418564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->VoidPtrTy)); 418664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 418764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned offset = (VoidPtrSize*4 + IntSize + IntSize)/Context->getCharWidth(); 418864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " _Block_object_assign((char*)dst + "; 418964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 419064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", *(void * *) ((char*)src + "; 419164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 419264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "), "; 419364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 419464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ");\n}\n"; 419564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 419664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "static void __Block_byref_id_object_dispose_"; 419764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 419864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(void *src) {\n"; 419964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " _Block_object_dispose(*(void * *) ((char*)src + "; 420064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 420164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "), "; 420264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 420364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ");\n}\n"; 420464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 420564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 420664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 420764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteByRefVar - For each __block typex ND variable this routine transforms 420864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// the declaration into: 420964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND { 421064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__isa; // NULL for everything except __weak pointers 421164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND *__forwarding; 421264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// int32_t __flags; 421364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// int32_t __size; 421464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__Block_byref_id_object_copy; // If variable is __block ObjC object 421564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__Block_byref_id_object_dispose; // If variable is __block ObjC object 421664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// typex ND; 421764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// }; 421864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 421964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// It then replaces declaration of ND variable with: 422064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND ND = {__isa=0B, __forwarding=&ND, __flags=some_flag, 422164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __size=sizeof(struct __Block_byref_ND), 422264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ND=initializer-if-any}; 422364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 422464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 422564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteByRefVar(VarDecl *ND) { 422664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert declaration for the function in which block literal is 422764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // used. 422864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDeclToDeclareForBlock) 422964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockLiteralFunctionDecl(CurFunctionDeclToDeclareForBlock); 423064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag = 0; 423164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int isa = 0; 423264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getTypeSpecStartLoc(); 423364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclLoc.isInvalid()) 423464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If type location is missing, it is because of missing type (a warning). 423564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Use variable's location which is good for this case. 423664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = ND->getLocation(); 423764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 423864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation X = ND->getLocEnd(); 423964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian X = SM->getExpansionLoc(X); 424064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(X); 424164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 424264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ByrefType; 424364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND, true); 424464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " {\n"; 424564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void *__isa;\n"; 424664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 424764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " *__forwarding;\n"; 424864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " int __flags;\n"; 424964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " int __size;\n"; 425064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add void *__Block_byref_id_object_copy; 425164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void *__Block_byref_id_object_dispose; if needed. 425264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Ty = ND->getType(); 425364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool HasCopyAndDispose = Context->BlockRequiresCopying(Ty); 425464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 425564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void (*__Block_byref_id_object_copy)(void*, void*);\n"; 425664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void (*__Block_byref_id_object_dispose)(void*);\n"; 425764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 425864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 425964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType T = Ty; 426064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(T); 426164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T.getAsStringInternal(Name, Context->getPrintingPolicy()); 426264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 426364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name + ";\n"; 426464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "};\n"; 426564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert this type in global scope. It is needed by helper function. 426664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart; 426764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDef) 426864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunLocStart = CurFunctionDef->getTypeSpecStartLoc(); 426964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 427064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CurMethodDef && "RewriteByRefVar - CurMethodDef is null"); 427164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunLocStart = CurMethodDef->getLocStart(); 427264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 427364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, ByrefType); 427464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty.isObjCGCWeak()) { 427564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_WEAK; 427664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa = 1; 427764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 427864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 427964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 428064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag = BLOCK_BYREF_CALLER; 428164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Ty = ND->getType(); 428264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Handle __weak variable (BLOCK_FIELD_IS_WEAK) as well. 428364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty->isBlockPointerType()) 428464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_BLOCK; 428564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 428664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_OBJECT; 428764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string HF = SynthesizeByrefCopyDestroyHelper(ND, flag); 428864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!HF.empty()) 428964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, HF); 429064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 429164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 429264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct __Block_byref_ND ND = 429364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // {0, &ND, some_flag, __size=sizeof(struct __Block_byref_ND), 429464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // initializer-if-any}; 429564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool hasInit = (ND->getInit() != 0); 429664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned flags = 0; 429764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) 429864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flags |= BLOCK_HAS_COPY_DISPOSE; 429964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = ND->getNameAsString(); 430064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType.clear(); 430164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 430264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ForwardingCastType("("); 430364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ForwardingCastType += ByrefType + " *)"; 430464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!hasInit) { 430564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name + " = {(void*)"; 430664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(isa); 430764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "," + ForwardingCastType + "&" + Name + ", "; 430864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flags); 430964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 431064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "sizeof("; 431164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 431264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ")"; 431364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 431464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_copy_"; 431564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 431664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_dispose_"; 431764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 431864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 431964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "};\n"; 432064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned nameSize = Name.size(); 432164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for block or function pointer declaration. Name is aleady 432264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // part of the declaration. 432364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty->isBlockPointerType() || Ty->isFunctionPointerType()) 432464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian nameSize = 1; 432564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf+nameSize, ByrefType); 432664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 432764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 432864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc; 432964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *E = ND->getInit(); 433064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) 433164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = ECE->getLParenLoc(); 433264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 433364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = E->getLocStart(); 433464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = SM->getExpansionLoc(startLoc); 433564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endBuf = SM->getCharacterData(startLoc); 433664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name; 433764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " = {(void*)"; 433864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(isa); 433964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "," + ForwardingCastType + "&" + Name + ", "; 434064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flags); 434164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 434264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "sizeof("; 434364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 434464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "), "; 434564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 434664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "__Block_byref_id_object_copy_"; 434764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 434864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_dispose_"; 434964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 435064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 435164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 435264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf, ByrefType); 435364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 435464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Complete the newly synthesized compound expression by inserting a right 435564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // curly brace before the end of the declaration. 435664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: This approach avoids rewriting the initializer expression. It 435764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // also assumes there is only one declarator. For example, the following 435864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // isn't currently supported by this routine (in general): 435964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 436064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // double __block BYREFVAR = 1.34, BYREFVAR2 = 1.37; 436164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 436264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startInitializerBuf = SM->getCharacterData(startLoc); 436364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startInitializerBuf, ';'); 436464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "RewriteByRefVar: can't find ';'"); 436564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation semiLoc = 436664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc.getLocWithOffset(semiBuf-startInitializerBuf); 436764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 436864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(semiLoc, "}"); 436964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 437064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 437164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 437264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 437364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::CollectBlockDeclRefInfo(BlockExpr *Exp) { 437464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add initializers for any closure decl refs. 437564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(Exp->getBody()); 437664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 437764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique all "by copy" declarations. 437864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 4379f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!BlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>()) { 438064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BlockByCopyDeclsPtrSet.count(BlockDeclRefs[i]->getDecl())) { 438164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.insert(BlockDeclRefs[i]->getDecl()); 438264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.push_back(BlockDeclRefs[i]->getDecl()); 438364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 438464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 438564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique all "by ref" declarations. 438664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 4387f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (BlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>()) { 438864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BlockByRefDeclsPtrSet.count(BlockDeclRefs[i]->getDecl())) { 438964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(BlockDeclRefs[i]->getDecl()); 439064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(BlockDeclRefs[i]->getDecl()); 439164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 439264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 439364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find any imported blocks...they will need special attention. 439464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 4395f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (BlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>() || 439664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs[i]->getType()->isObjCObjectPointerType() || 439764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs[i]->getType()->isBlockPointerType()) 439864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(BlockDeclRefs[i]->getDecl()); 439964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 440064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 440164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 440264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianFunctionDecl *RewriteModernObjC::SynthBlockInitFunctionDecl(StringRef name) { 440364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *ID = &Context->Idents.get(name); 440464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FType = Context->getFunctionNoProtoType(Context->VoidPtrTy); 440564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return FunctionDecl::Create(*Context, TUDecl, SourceLocation(), 440664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), ID, FType, 0, SC_Extern, 440764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false, false); 440864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 440964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthBlockInitExpr(BlockExpr *Exp, 4411f4b88a45902af1802a1cb42ba48b1c474474f228John McCall const SmallVector<DeclRefExpr *, 8> &InnerBlockDeclRefs) { 441264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockDecl *block = Exp->getBlockDecl(); 441364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Blocks.push_back(Exp); 441464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CollectBlockDeclRefInfo(Exp); 441664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add inner imported variables now used in current block. 441864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int countOfInnerDecls = 0; 441964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!InnerBlockDeclRefs.empty()) { 442064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < InnerBlockDeclRefs.size(); i++) { 4421f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *Exp = InnerBlockDeclRefs[i]; 442264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = Exp->getDecl(); 4423f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (!VD->hasAttr<BlocksAttr>() && !BlockByCopyDeclsPtrSet.count(VD)) { 442464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We need to save the copied-in variables in nested 442564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // blocks because it is needed at the end for some of the API generations. 442664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See SynthesizeBlockLiterals routine. 442764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.push_back(Exp); countOfInnerDecls++; 442864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 442964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.insert(VD); 443064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.push_back(VD); 443164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 4432f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (VD->hasAttr<BlocksAttr>() && !BlockByRefDeclsPtrSet.count(VD)) { 443364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.push_back(Exp); countOfInnerDecls++; 443464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 443564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(VD); 443664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(VD); 443764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 443864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 443964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find any imported blocks...they will need special attention. 444064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < InnerBlockDeclRefs.size(); i++) 4441f4b88a45902af1802a1cb42ba48b1c474474f228John McCall if (InnerBlockDeclRefs[i]->getDecl()->hasAttr<BlocksAttr>() || 444264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs[i]->getType()->isObjCObjectPointerType() || 444364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs[i]->getType()->isBlockPointerType()) 444464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(InnerBlockDeclRefs[i]->getDecl()); 444564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 444664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefsCount.push_back(countOfInnerDecls); 444764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 444864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FuncName; 444964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 445064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDef) 445164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncName = CurFunctionDef->getNameAsString(); 445264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (CurMethodDef) 445364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BuildUniqueMethodName(FuncName, CurMethodDef); 445464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (GlobalVarDecl) 445564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncName = std::string(GlobalVarDecl->getNameAsString()); 445664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 445764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string BlockNumber = utostr(Blocks.size()-1); 445864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 445964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag = "__" + FuncName + "_block_impl_" + BlockNumber; 446064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Func = "__" + FuncName + "_block_func_" + BlockNumber; 446164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 446264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get a pointer to the function type so we can cast appropriately. 446364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType BFT = convertFunctionTypeOfBlocks(Exp->getFunctionType()); 446464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FType = Context->getPointerType(BFT); 446564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 446664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD; 446764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *NewRep; 446864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 446964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 447064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl(Tag); 4471f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(FD, false, FType, VK_RValue, 447264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 447364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 447464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 447564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 447664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize the block function. 447764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl(Func); 4478f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *Arg = new (Context) DeclRefExpr(FD, false, FD->getType(), 4479f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 448064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, Context->VoidPtrTy, 448164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Arg); 448264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(castExpr); 448364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 448464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize the block descriptor. 448564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string DescData = "__" + FuncName + "_block_desc_" + BlockNumber + "_DATA"; 448664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 448764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, 448864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 448964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get(DescData.c_str()), 449064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 449164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Static, SC_None); 449264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UnaryOperator *DescRefExpr = 4493f4b88a45902af1802a1cb42ba48b1c474474f228John McCall new (Context) UnaryOperator(new (Context) DeclRefExpr(NewVD, false, 449464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 449564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, 449664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()), 449764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UO_AddrOf, 449864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidPtrTy), 449964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 450064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 450164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(DescRefExpr); 450264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 450364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add initializers for any closure decl refs. 450464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 450564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Exp; 450664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by copy" declarations. 450764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 450864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 450964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isObjCType((*I)->getType())) { 451064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Conform to ABI ([[obj retain] autorelease]). 451164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 4512f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Exp = new (Context) DeclRefExpr(FD, false, FD->getType(), 4513f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 451464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) { 451564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 451664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 451764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, QT, VK_RValue, 451864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary, SourceLocation()); 451964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 452064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (isTopLevelBlockPointerType((*I)->getType())) { 452164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 4522f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Arg = new (Context) DeclRefExpr(FD, false, FD->getType(), 4523f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 452464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = NoTypeInfoCStyleCastExpr(Context, Context->VoidPtrTy, 452564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Arg); 452664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 452764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 4528f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Exp = new (Context) DeclRefExpr(FD, false, FD->getType(), 4529f4b88a45902af1802a1cb42ba48b1c474474f228John McCall VK_LValue, SourceLocation()); 453064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) { 453164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 453264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 453364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, QT, VK_RValue, 453464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary, SourceLocation()); 453564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 453664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 453764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 453864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(Exp); 453964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 454064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by ref" declarations. 454164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 454264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 454364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *ND = (*I); 454464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 454564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RecName; 454664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(RecName, Name, ND, true); 454764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *II = &Context->Idents.get(RecName.c_str() 454864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian + sizeof("struct")); 454964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 455064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 455164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian II); 455264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(RD && "SynthBlockInitExpr(): Can't find RecordDecl"); 455364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castT = Context->getPointerType(Context->getTagDeclType(RD)); 455464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 455564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 4556f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Exp = new (Context) DeclRefExpr(FD, false, FD->getType(), VK_LValue, 455764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 455864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isNestedCapturedVar = false; 455964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (block) 456064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (BlockDecl::capture_const_iterator ci = block->capture_begin(), 456164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ce = block->capture_end(); ci != ce; ++ci) { 456264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const VarDecl *variable = ci->getVariable(); 456364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (variable == ND && ci->isNested()) { 456464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert (ci->isByRef() && 456564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "SynthBlockInitExpr - captured block variable is not byref"); 456664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isNestedCapturedVar = true; 456764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 456864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 456964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 457064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // captured nested byref variable has its address passed. Do not take 457164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // its address again. 457264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isNestedCapturedVar) 457364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, 457464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Exp->getType()), 457564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 457664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = NoTypeInfoCStyleCastExpr(Context, castT, CK_BitCast, Exp); 457764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(Exp); 457864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 457964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 458064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImportedBlockDecls.size()) { 458164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // generate BLOCK_HAS_COPY_DISPOSE(have helper funcs) | BLOCK_HAS_DESCRIPTOR 458264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag = (BLOCK_HAS_COPY_DISPOSE | BLOCK_HAS_DESCRIPTOR); 458364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 458464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 458564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *FlagExp = IntegerLiteral::Create(*Context, llvm::APInt(IntSize, flag), 458664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->IntTy, SourceLocation()); 458764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(FlagExp); 458864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 458964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], InitExprs.size(), 459064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FType, VK_LValue, SourceLocation()); 459164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = new (Context) UnaryOperator(NewRep, UO_AddrOf, 459264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(NewRep->getType()), 459364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 459464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = NoTypeInfoCStyleCastExpr(Context, FType, CK_BitCast, 459564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep); 459664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.clear(); 459764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.clear(); 459864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.clear(); 459964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.clear(); 460064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.clear(); 460164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.clear(); 460264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return NewRep; 460364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 460464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 460564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::IsDeclStmtInForeachHeader(DeclStmt *DS) { 460664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const ObjCForCollectionStmt * CS = 460764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCForCollectionStmt>(Stmts.back())) 460864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CS->getElement() == DS; 460964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 461064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 461164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 461264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 461364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Function Body / Expression rewriting 461464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 461564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 461664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteFunctionBodyOrGlobalInitializer(Stmt *S) { 461764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) || 461864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(S) || isa<ForStmt>(S)) 461964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.push_back(S); 462064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (isa<ObjCForCollectionStmt>(S)) { 462164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.push_back(S); 462264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCBcLabelNo.push_back(++BcLabelCount); 462364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 462464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 462564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Pseudo-object operations and ivar references need special 462664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // treatment because we're going to recursively rewrite them. 462764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PseudoObjectExpr *PseudoOp = dyn_cast<PseudoObjectExpr>(S)) { 462864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<BinaryOperator>(PseudoOp->getSyntacticForm())) { 462964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewritePropertyOrImplicitSetter(PseudoOp); 463064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 463164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewritePropertyOrImplicitGetter(PseudoOp); 463264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 463364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCIvarRefExpr *IvarRefExpr = dyn_cast<ObjCIvarRefExpr>(S)) { 463464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCIvarRefExpr(IvarRefExpr); 463564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 463664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 463764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OrigStmtRange = S->getSourceRange(); 463864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 463964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up rewrite of all children. 464064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 464164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 464264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *childStmt = (*CI); 464364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *newStmt = RewriteFunctionBodyOrGlobalInitializer(childStmt); 464464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (newStmt) { 464564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *CI = newStmt; 464664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 464764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 464864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 464964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *BE = dyn_cast<BlockExpr>(S)) { 4650f4b88a45902af1802a1cb42ba48b1c474474f228John McCall SmallVector<DeclRefExpr *, 8> InnerBlockDeclRefs; 465164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> InnerContexts; 465264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts.insert(BE->getBlockDecl()); 465364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedLocalExternalDecls.clear(); 465464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(BE->getBody(), 465564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, InnerContexts); 465664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the block body in place. 465764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SaveCurrentBody = CurrentBody; 465864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = BE->getBody(); 465964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 466064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block literal on rhs of a property-dot-sytax assignment 466164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // must be replaced by its synthesize ast so getRewrittenText 466264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // works as expected. In this case, what actually ends up on RHS 466364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // is the blockTranscribed which is the helper function for the 466464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block literal; as in: self.c = ^() {[ace ARR];}; 466564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool saveDisableReplaceStmt = DisableReplaceStmt; 466664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = false; 466764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionBodyOrGlobalInitializer(BE->getBody()); 466864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = saveDisableReplaceStmt; 466964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = SaveCurrentBody; 467064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 467164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedLocalExternalDecls.clear(); 467264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now we snarf the rewritten text and stash it away for later use. 467364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Str = Rewrite.getRewrittenText(BE->getSourceRange()); 467464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewrittenBlockExprs[BE] = Str; 467564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 467664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *blockTranscribed = SynthBlockInitExpr(BE, InnerBlockDeclRefs); 467764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 467864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //blockTranscribed->dump(); 467964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(S, blockTranscribed); 468064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return blockTranscribed; 468164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 468264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 468364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCEncodeExpr *AtEncode = dyn_cast<ObjCEncodeExpr>(S)) 468464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteAtEncode(AtEncode); 468564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 468664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSelectorExpr *AtSelector = dyn_cast<ObjCSelectorExpr>(S)) 468764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteAtSelector(AtSelector); 468864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 468964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCStringLiteral *AtString = dyn_cast<ObjCStringLiteral>(S)) 469064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCStringLiteral(AtString); 469164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 469264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMessageExpr *MessExpr = dyn_cast<ObjCMessageExpr>(S)) { 469364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#if 0 469464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Before we rewrite it, put the original message expression in a comment. 469564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = MessExpr->getLocStart(); 469664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endLoc = MessExpr->getLocEnd(); 469764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 469864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 469964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(endLoc); 470064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 470164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string messString; 470264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString += "// "; 470364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString.append(startBuf, endBuf-startBuf+1); 470464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString += "\n"; 470564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 470664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Missing definition of 470764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // InsertText(clang::SourceLocation, char const*, unsigned int). 470864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // InsertText(startLoc, messString.c_str(), messString.size()); 470964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Tried this, but it didn't work either... 471064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ReplaceText(startLoc, 0, messString.c_str(), messString.size()); 471164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#endif 471264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteMessageExpr(MessExpr); 471364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 471464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 471564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtTryStmt *StmtTry = dyn_cast<ObjCAtTryStmt>(S)) 471664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCTryStmt(StmtTry); 471764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 471864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtSynchronizedStmt *StmtTry = dyn_cast<ObjCAtSynchronizedStmt>(S)) 471964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCSynchronizedStmt(StmtTry); 472064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 472164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtThrowStmt *StmtThrow = dyn_cast<ObjCAtThrowStmt>(S)) 472264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCThrowStmt(StmtThrow); 472364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 472464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolExpr *ProtocolExp = dyn_cast<ObjCProtocolExpr>(S)) 472564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCProtocolExpr(ProtocolExp); 472664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 472764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCForCollectionStmt *StmtForCollection = 472864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCForCollectionStmt>(S)) 472964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCForCollectionStmt(StmtForCollection, 473064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigStmtRange.getEnd()); 473164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BreakStmt *StmtBreakStmt = 473264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<BreakStmt>(S)) 473364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteBreakStmt(StmtBreakStmt); 473464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ContinueStmt *StmtContinueStmt = 473564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ContinueStmt>(S)) 473664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteContinueStmt(StmtContinueStmt); 473764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 473864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to check for protocol refs (id <P>, Foo <P> *) in variable decls 473964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and cast exprs. 474064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclStmt *DS = dyn_cast<DeclStmt>(S)) { 474164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: What we're doing here is modifying the type-specifier that 474264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // precedes the first Decl. In the future the DeclGroup should have 474364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // a separate type-specifier that we can rewrite. 474464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NOTE: We need to avoid rewriting the DeclStmt if it is within 474564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the context of an ObjCForCollectionStmt. For example: 474664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NSArray *someArray; 474764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for (id <FooProtocol> index in someArray) ; 474864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is because RewriteObjCForCollectionStmt() does textual rewriting 474964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and it depends on the original text locations/positions. 475064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !IsDeclStmtInForeachHeader(DS)) 475164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(*DS->decl_begin()); 475264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 475364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Blocks rewrite rules. 475464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclStmt::decl_iterator DI = DS->decl_begin(), DE = DS->decl_end(); 475564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DI != DE; ++DI) { 475664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Decl *SD = *DI; 475764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ValueDecl *ND = dyn_cast<ValueDecl>(SD)) { 475864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(ND->getType())) 475964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(ND); 476064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (ND->getType()->isFunctionPointerType()) 476164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(ND->getType(), ND); 476264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(SD)) { 476364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->hasAttr<BlocksAttr>()) { 476464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static unsigned uniqueByrefDeclCount = 0; 476564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!BlockByRefDeclNo.count(ND) && 476664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "RewriteFunctionBodyOrGlobalInitializer: Duplicate byref decl"); 476764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclNo[ND] = uniqueByrefDeclCount++; 476864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefVar(VD); 476964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 477064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 477164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeOfDecl(VD); 477264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 477364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 477464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (TypedefNameDecl *TD = dyn_cast<TypedefNameDecl>(SD)) { 477564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(TD->getUnderlyingType())) 477664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(TD); 477764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TD->getUnderlyingType()->isFunctionPointerType()) 477864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(TD->getUnderlyingType(), TD); 477964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 478064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 478164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 478264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 478364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(S)) 478464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(CE); 478564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 478664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) || 478764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(S) || isa<ForStmt>(S)) { 478864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!Stmts.empty() && "Statement stack is empty"); 478964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert ((isa<SwitchStmt>(Stmts.back()) || isa<WhileStmt>(Stmts.back()) || 479064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(Stmts.back()) || isa<ForStmt>(Stmts.back())) 479164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian && "Statement stack mismatch"); 479264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.pop_back(); 479364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 479464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle blocks rewriting. 479564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) { 479664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = DRE->getDecl(); 479764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->hasAttr<BlocksAttr>()) 479864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteBlockDeclRefExpr(DRE); 479964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(VD)) 480064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteLocalVariableExternalStorage(DRE); 480164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 480264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 480364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CallExpr *CE = dyn_cast<CallExpr>(S)) { 480464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CE->getCallee()->getType()->isBlockPointerType()) { 480564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *BlockCall = SynthesizeBlockCall(CE, CE->getCallee()); 480664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(S, BlockCall); 480764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return BlockCall; 480864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 480964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 481064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(S)) { 481164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 481264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 481364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#if 0 481464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(S)) { 481564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *Replacement = new (Context) CastExpr(ICE->getType(), 481664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ICE->getSubExpr(), 481764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 481864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the new text. 481964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SStr; 482064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream Buf(SStr); 482164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Replacement->printPretty(Buf, *Context); 482264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Str = Buf.str(); 482364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 482464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian printf("CAST = %s\n", &Str[0]); 482564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(ICE->getSubExpr()->getLocStart(), &Str[0], Str.size()); 482664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete S; 482764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 482864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 482964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#endif 483064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Return this stmt unmodified. 483164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 483264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 483364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 483464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteRecordBody(RecordDecl *RD) { 483564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (RecordDecl::field_iterator i = RD->field_begin(), 483664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian e = RD->field_end(); i != e; ++i) { 483764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = *i; 483864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(FD->getType())) 483964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(FD); 484064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->getType()->isObjCQualifiedIdType() || 484164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType()->isObjCQualifiedInterfaceType()) 484264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(FD); 484364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 484464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 484564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 484664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// HandleDeclInMainFile - This is called for each top-level decl defined in the 484764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// main file of the input. 484864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleDeclInMainFile(Decl *D) { 484964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (D->getKind()) { 485064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Function: { 485164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD = cast<FunctionDecl>(D); 485264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->isOverloadedOperator()) 485364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 485464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 485564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since function prototypes don't have ParmDecl's, we check the function 485664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // prototype. This enables us to rewrite function declarations and 485764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // definitions using the same code. 485864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlocksInFunctionProtoType(FD->getType(), FD); 485964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 48609335df324951c022bc696e64b5fe9f66a8ad5222Argyrios Kyrtzidis if (!FD->isThisDeclarationADefinition()) 48619335df324951c022bc696e64b5fe9f66a8ad5222Argyrios Kyrtzidis break; 48629335df324951c022bc696e64b5fe9f66a8ad5222Argyrios Kyrtzidis 486364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: If this should support Obj-C++, support CXXTryStmt 486464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CompoundStmt *Body = dyn_cast_or_null<CompoundStmt>(FD->getBody())) { 486564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = FD; 486664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = FD; 486764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = Body; 486864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Body = 486964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast_or_null<CompoundStmt>(RewriteFunctionBodyOrGlobalInitializer(Body)); 487064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->setBody(Body); 487164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 487264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 487364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 487464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 487564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 487664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This synthesizes and inserts the block "impl" struct, invoke function, 487764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and any copy/dispose helper functions. 487864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertBlockLiteralsWithinFunction(FD); 487964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = 0; 488064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = 0; 488164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 488264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 488364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 488464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCMethod: { 488564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MD = cast<ObjCMethodDecl>(D); 488664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CompoundStmt *Body = MD->getCompoundBody()) { 488764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = MD; 488864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = Body; 488964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Body = 489064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast_or_null<CompoundStmt>(RewriteFunctionBodyOrGlobalInitializer(Body)); 489164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MD->setBody(Body); 489264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 489364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 489464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 489564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 489664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 489764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertBlockLiteralsWithinMethod(MD); 489864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = 0; 489964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 490164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCImplementation: { 490364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *CI = cast<ObjCImplementationDecl>(D); 490464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassImplementation.push_back(CI); 490564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 490664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCCategoryImpl: { 490864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CI = cast<ObjCCategoryImplDecl>(D); 490964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CategoryImplementation.push_back(CI); 491064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 491164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 491264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Var: { 491364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *VD = cast<VarDecl>(D); 491464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(VD); 491564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(VD->getType())) 491664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(VD); 491764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isFunctionPointerType()) { 491864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(VD->getType(), VD); 491964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getInit()) { 492064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(VD->getInit())) { 492164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 492264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 492364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 492464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (VD->getType()->isRecordType()) { 492564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = VD->getType()->getAs<RecordType>()->getDecl(); 492664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RD->isCompleteDefinition()) 492764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteRecordBody(RD); 492864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 492964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getInit()) { 493064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = VD; 493164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = VD->getInit(); 493264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionBodyOrGlobalInitializer(VD->getInit()); 493364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 493464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 493564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 493664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 493764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 493864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(VD->getTypeSpecStartLoc(), VD->getName()); 493964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = 0; 494064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 494164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is needed for blocks. 494264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(VD->getInit())) { 494364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 494464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 494564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 494664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 494764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 494864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::TypeAlias: 494964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Typedef: { 495064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (TypedefNameDecl *TD = dyn_cast<TypedefNameDecl>(D)) { 495164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(TD->getUnderlyingType())) 495264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(TD); 495364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TD->getUnderlyingType()->isFunctionPointerType()) 495464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(TD->getUnderlyingType(), TD); 495564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 495664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 495764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 495864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::CXXRecord: 495964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Record: { 496064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = cast<RecordDecl>(D); 496164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RD->isCompleteDefinition()) 496264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteRecordBody(RD); 496364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 496464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 496564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: 496664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 496764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 496864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Nothing yet. 496964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 497064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 497130650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian/// Write_ProtocolExprReferencedMetadata - This routine writer out the 497230650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian/// protocol reference symbols in the for of: 497330650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian/// struct _protocol_t *PROTOCOL_REF = &PROTOCOL_METADATA. 497430650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanianstatic void Write_ProtocolExprReferencedMetadata(ASTContext *Context, 497530650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian ObjCProtocolDecl *PDecl, 497630650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian std::string &Result) { 497730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian // Also output .objc_protorefs$B section and its meta-data. 497830650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian if (Context->getLangOpts().MicrosoftExt) 497930650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "__declspec(allocate(\".objc_protorefs$B\")) "; 498030650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "struct _protocol_t *"; 498130650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "_OBJC_PROTOCOL_REFERENCE_$_"; 498230650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += PDecl->getNameAsString(); 498330650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += " = &"; 498430650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "_OBJC_PROTOCOL_"; Result += PDecl->getNameAsString(); 498530650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += ";\n"; 498630650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian} 498730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian 498864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleTranslationUnit(ASTContext &C) { 498964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Diags.hasErrorOccurred()) 499064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 499164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 499264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteInclude(); 499364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 499464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Here's a great place to add any extra declarations that may be needed. 499564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Write out meta data for each @protocol(<expr>). 499664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ObjCProtocolDecl *,8>::iterator I = ProtocolExprDecls.begin(), 499730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian E = ProtocolExprDecls.end(); I != E; ++I) { 4998da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Preamble); 499930650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Write_ProtocolExprReferencedMetadata(Context, (*I), Preamble); 500030650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian } 500164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 500264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(SM->getLocForStartOfFile(MainFileID), Preamble, false); 500357317788289549d61ada868d07792b637a32d52eFariborz Jahanian for (unsigned i = 0, e = ObjCInterfacesSeen.size(); i < e; i++) { 500457317788289549d61ada868d07792b637a32d52eFariborz Jahanian ObjCInterfaceDecl *CDecl = ObjCInterfacesSeen[i]; 500557317788289549d61ada868d07792b637a32d52eFariborz Jahanian // Write struct declaration for the class matching its ivar declarations. 500657317788289549d61ada868d07792b637a32d52eFariborz Jahanian // Note that for modern abi, this is postponed until the end of TU 500757317788289549d61ada868d07792b637a32d52eFariborz Jahanian // because class extensions and the implementation might declare their own 500857317788289549d61ada868d07792b637a32d52eFariborz Jahanian // private ivars. 500957317788289549d61ada868d07792b637a32d52eFariborz Jahanian RewriteInterfaceDecl(CDecl); 501057317788289549d61ada868d07792b637a32d52eFariborz Jahanian } 501172c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 501264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ClassImplementation.size() || CategoryImplementation.size()) 501364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteImplementations(); 501464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 501564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the buffer corresponding to MainFileID. If we haven't changed it, then 501664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we are done. 501764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const RewriteBuffer *RewriteBuf = 501864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Rewrite.getRewriteBufferFor(MainFileID)) { 501964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //printf("Changed:\n"); 502064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *OutFile << std::string(RewriteBuf->begin(), RewriteBuf->end()); 502164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 502264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::errs() << "No changes\n"; 502364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 502464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 502564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ClassImplementation.size() || CategoryImplementation.size() || 502664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolExprDecls.size()) { 502764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite Objective-c meta data* 502864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 502964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMetaDataIntoBuffer(ResultStr); 503064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Emit metadata. 503164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *OutFile << ResultStr; 503264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 503310cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // Emit ImageInfo; 503410cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian { 503510cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian std::string ResultStr; 503610cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian WriteImageInfo(ResultStr); 503710cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian *OutFile << ResultStr; 503810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian } 503964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OutFile->flush(); 504064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 504164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 504264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::Initialize(ASTContext &context) { 504364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitializeCommon(context); 504464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 50456991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian Preamble += "#ifndef __OBJC2__\n"; 50466991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian Preamble += "#define __OBJC2__\n"; 50476991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian Preamble += "#endif\n"; 50486991bc5a45b6d334a19e1ba995c8aea16553973bFariborz Jahanian 504964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declaring objc_selector outside the parameter list removes a silly 505064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scope related warning... 505164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IsHeader) 505264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble = "#pragma once\n"; 505364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct objc_selector; struct objc_class;\n"; 505464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __rw_objc_super { struct objc_object *object; "; 505564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct objc_object *superClass; "; 505664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 50571ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian // Define all sections using syntax that makes sense. 505810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // These are currently generated. 505910cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "\n#pragma section(\".objc_classlist$B\", long, read, write)\n"; 50601ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Preamble += "#pragma section(\".objc_catlist$B\", long, read, write)\n"; 506110cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".objc_protolist$B\", long, read, write)\n"; 506210cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".objc_imageinfo$B\", long, read, write)\n"; 506388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Preamble += "#pragma section(\".objc_nlclslist$B\", long, read, write)\n"; 506488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Preamble += "#pragma section(\".objc_nlcatlist$B\", long, read, write)\n"; 506510cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".objc_protorefs$B\", long, read, write)\n"; 506610cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // These are generated but not necessary for functionality. 506710cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".datacoal_nt$B\", long, read, write)\n"; 506810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Preamble += "#pragma section(\".cat_cls_meth$B\", long, read, write)\n"; 50691ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Preamble += "#pragma section(\".inst_meth$B\", long, read, write)\n"; 50701ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Preamble += "#pragma section(\".cls_meth$B\", long, read, write)\n"; 507140a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Preamble += "#pragma section(\".objc_ivar$B\", long, read, write)\n"; 5072de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian 507330650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian // These need be generated for performance. Currently they are not, 507430650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian // using API calls instead. 507530650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Preamble += "#pragma section(\".objc_selrefs$B\", long, read, write)\n"; 507630650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Preamble += "#pragma section(\".objc_classrefs$B\", long, read, write)\n"; 507730650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Preamble += "#pragma section(\".objc_superrefs$B\", long, read, write)\n"; 507830650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian 507964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add a constructor for creating temporary objects. 508064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__rw_objc_super(struct objc_object *o, struct objc_object *s) " 508164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ": "; 508264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "object(o), superClass(s) {} "; 508364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 508464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 508564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef _REWRITER_typedef_Protocol\n"; 508664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "typedef struct objc_object Protocol;\n"; 508764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define _REWRITER_typedef_Protocol\n"; 508864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 508964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 509064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_DLLIMPORT extern \"C\" __declspec(dllimport)\n"; 509164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_STATICIMPORT extern \"C\"\n"; 509264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 509364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_DLLIMPORT extern\n"; 509464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_msgSend"; 509564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_object *, struct objc_selector *, ...);\n"; 509664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_msgSendSuper"; 509764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_super *, struct objc_selector *, ...);\n"; 509864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object* objc_msgSend_stret"; 509964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_object *, struct objc_selector *, ...);\n"; 510064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object* objc_msgSendSuper_stret"; 510164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_super *, struct objc_selector *, ...);\n"; 510264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT double objc_msgSend_fpret"; 510364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_object *, struct objc_selector *, ...);\n"; 510464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_getClass"; 510564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(const char *);\n"; 510664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_class *class_getSuperclass"; 510764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_class *);\n"; 510864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_getMetaClass"; 510964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(const char *);\n"; 511055261afb725b8af330addddd2e452f9f28ddd96bFariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_exception_throw( struct objc_object *);\n"; 511164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // @synchronized hooks. 511255261afb725b8af330addddd2e452f9f28ddd96bFariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_sync_enter( struct objc_object *);\n"; 511355261afb725b8af330addddd2e452f9f28ddd96bFariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_sync_exit( struct objc_object *);\n"; 511464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT Protocol *objc_getProtocol(const char *);\n"; 511564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef __FASTENUMERATIONSTATE\n"; 511664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __objcFastEnumerationState {\n\t"; 511764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long state;\n\t"; 511864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "void **itemsPtr;\n\t"; 511964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long *mutationsPtr;\n\t"; 512064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long extra[5];\n};\n"; 512164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_enumerationMutation(struct objc_object *);\n"; 512264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __FASTENUMERATIONSTATE\n"; 512364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 512464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef __NSCONSTANTSTRINGIMPL\n"; 512564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __NSConstantStringImpl {\n"; 512664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int *isa;\n"; 512764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int flags;\n"; 512864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " char *str;\n"; 512964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " long length;\n"; 513064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 513164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifdef CF_EXPORT_CONSTANT_STRING\n"; 513264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) int __CFConstantStringClassReference[];\n"; 513364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#else\n"; 513464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT int __CFConstantStringClassReference[];\n"; 513564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 513664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __NSCONSTANTSTRINGIMPL\n"; 513764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 513864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Blocks preamble. 513964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef BLOCK_IMPL\n"; 514064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define BLOCK_IMPL\n"; 514164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __block_impl {\n"; 514264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " void *isa;\n"; 514364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int Flags;\n"; 514464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int Reserved;\n"; 514564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " void *FuncPtr;\n"; 514664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 514764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "// Runtime copy/destroy helper functions (from Block_private.h)\n"; 514864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifdef __OBJC_EXPORT_BLOCKS\n"; 514964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) " 515064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "void _Block_object_assign(void *, const void *, const int);\n"; 515164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void _Block_object_dispose(const void *, const int);\n"; 515264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void *_NSConcreteGlobalBlock[32];\n"; 515364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void *_NSConcreteStackBlock[32];\n"; 515464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#else\n"; 515564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void _Block_object_assign(void *, const void *, const int);\n"; 515664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void _Block_object_dispose(const void *, const int);\n"; 515764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void *_NSConcreteGlobalBlock[32];\n"; 515864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void *_NSConcreteStackBlock[32];\n"; 515964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 516064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 516164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 516264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#undef __OBJC_RW_DLLIMPORT\n"; 516364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#undef __OBJC_RW_STATICIMPORT\n"; 516464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef KEEP_ATTRIBUTES\n"; // We use this for clang tests. 516564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __attribute__(X)\n"; 516664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 516764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __weak\n"; 516864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 516964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 517064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __block\n"; 517164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __weak\n"; 517264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 517364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NOTE! Windows uses LLP64 for 64bit mode. So, cast pointer to long long 517464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as this avoids warning in any 64bit/32bit compilation model. 517564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "\n#define __OFFSETOFIVAR__(TYPE, MEMBER) ((long long) &((TYPE *)0)->MEMBER)\n"; 517664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 517764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 517864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteIvarOffsetComputation - This rutine synthesizes computation of 517964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ivar offset. 518064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteIvarOffsetComputation(ObjCIvarDecl *ivar, 518164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 518264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ivar->isBitField()) { 518364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: The hack below doesn't work for bitfields. For now, we simply 518464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // place all bitfields at offset 0. 518564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "0"; 518664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 518764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "__OFFSETOFIVAR__(struct "; 518864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ivar->getContainingInterface()->getNameAsString(); 518964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) 519064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_IMPL"; 519164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", "; 519264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ivar->getNameAsString(); 519364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ")"; 519464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 519564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 519664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 519764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// WriteModernMetadataDeclarations - Writes out metadata declarations for modern ABI. 519864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_t { 5199da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian/// const char *name; 520064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// char *attributes; 520164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 520264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 520364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_list_t { 520464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t entsize; // sizeof(struct _prop_t) 520564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t count_of_properties; 520664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_t prop_list[count_of_properties]; 520764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 520864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 520964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t; 521064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 521164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_list_t { 521264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// long protocol_count; // Note, this is 32/64 bit 521364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t * protocol_list[protocol_count]; 521464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 521564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 521664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _objc_method { 521764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SEL _cmd; 521877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian/// const char *method_type; 521964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// char *_imp; 522064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 522164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 522264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _method_list_t { 522364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t entsize; // sizeof(struct _objc_method) 522464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t method_count; 522564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _objc_method method_list[method_count]; 522664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 522764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 522864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t { 522964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id isa; // NULL 523064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const char * const protocol_name; 523164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct _protocol_list_t * protocol_list; // super protocols 523264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t * const instance_methods; 523364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t * const class_methods; 523464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t *optionalInstanceMethods; 523564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t *optionalClassMethods; 523664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct _prop_list_t * properties; 523764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const uint32_t size; // sizeof(struct _protocol_t) 523864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const uint32_t flags; // = 0 523964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const char ** extendedMethodTypes; 524064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 524164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 524242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _ivar_t { 524342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// unsigned long int *offset; // pointer to ivar offset location 5244ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// const char *name; 5245ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// const char *type; 524642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t alignment; 524742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t size; 524842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 524942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 525042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _ivar_list_t { 525142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32 entsize; // sizeof(struct _ivar_t) 525242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32 count; 5253ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// struct _ivar_t list[count]; 525442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 525542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 525642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_ro_t { 525742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const flags; 525842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const instanceStart; 525942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const instanceSize; 526042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const reserved; // only when building for 64bit targets 526142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const uint8_t * const ivarLayout; 526242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const char *const name; 526342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _method_list_t * const baseMethods; 52640a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian/// const struct _protocol_list_t *const baseProtocols; 526542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _ivar_list_t *const ivars; 526642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const uint8_t * const weakIvarLayout; 526742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _prop_list_t * const properties; 526842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 526942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 527042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t { 527142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t *isa; 5272fd4ce2c9d64032b42c037b2af3bf4ebeac5bbac8Fariborz Jahanian/// struct _class_t *superclass; 527342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// void *cache; 527442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// IMP *vtable; 52753f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian/// struct _class_ro_t *ro; 527642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 527742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 527842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _category_t { 527942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const char * const name; 528042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t *const cls; 528142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _method_list_t * const instance_methods; 528242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _method_list_t * const class_methods; 528342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _protocol_list_t * const protocols; 528442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _prop_list_t * const properties; 528542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 528642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 528742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// MessageRefTy - LLVM for: 528842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _message_ref_t { 528942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// IMP messenger; 529042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SEL name; 529142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// }; 529242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 529342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SuperMessageRefTy - LLVM for: 529442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _super_message_ref_t { 529542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SUPER_IMP messenger; 529642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SEL name; 529742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// }; 529842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 5299de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanianstatic void WriteModernMetadataDeclarations(ASTContext *Context, std::string &Result) { 530064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool meta_data_declared = false; 530164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (meta_data_declared) 530264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 530364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 530464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _prop_t {\n"; 5305da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tconst char *name;\n"; 5306da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tconst char *attributes;\n"; 530764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 530864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 530964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _protocol_t;\n"; 531064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 531164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_method {\n"; 531264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct objc_selector * _cmd;\n"; 531377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tconst char *method_type;\n"; 531490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "\tvoid *_imp;\n"; 531564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 531664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 531764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _protocol_t {\n"; 531864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tvoid * isa; // NULL\n"; 531964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char * const protocol_name;\n"; 532064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct _protocol_list_t * protocol_list; // super protocols\n"; 532164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t * const instance_methods;\n"; 532264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t * const class_methods;\n"; 532364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t *optionalInstanceMethods;\n"; 532464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t *optionalClassMethods;\n"; 532564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct _prop_list_t * properties;\n"; 532664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst unsigned int size; // sizeof(struct _protocol_t)\n"; 532764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst unsigned int flags; // = 0\n"; 532864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char ** extendedMethodTypes;\n"; 532964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 533064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 533142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _ivar_t {\n"; 533242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned long int *offset; // pointer to ivar offset location\n"; 5333ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tconst char *name;\n"; 5334ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tconst char *type;\n"; 533542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int alignment;\n"; 533642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int size;\n"; 533742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 533842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 533942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _class_ro_t {\n"; 534042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int const flags;\n"; 534142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int instanceStart;\n"; 534242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int const instanceSize;\n"; 5343de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian const llvm::Triple &Triple(Context->getTargetInfo().getTriple()); 5344de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian if (Triple.getArch() == llvm::Triple::x86_64) 5345de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian Result += "\tunsigned int const reserved;\n"; 534642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst unsigned char * const ivarLayout;\n"; 534742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst char *const name;\n"; 534842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _method_list_t * const baseMethods;\n"; 534942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _objc_protocol_list *const baseProtocols;\n"; 535042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _ivar_list_t *const ivars;\n"; 535142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst unsigned char *const weakIvarLayout;\n"; 535242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _prop_list_t *const properties;\n"; 535342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 535442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 535542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _class_t {\n"; 535642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tstruct _class_t *isa;\n"; 5357fd4ce2c9d64032b42c037b2af3bf4ebeac5bbac8Fariborz Jahanian Result += "\tstruct _class_t *superclass;\n"; 535842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tvoid *cache;\n"; 535942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tvoid *vtable;\n"; 53603f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "\tstruct _class_ro_t *ro;\n"; 536142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 536242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 536342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _category_t {\n"; 536442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst char * const name;\n"; 536542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tstruct _class_t *const cls;\n"; 536642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _method_list_t *const instance_methods;\n"; 536742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _method_list_t *const class_methods;\n"; 536842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _protocol_list_t *const protocols;\n"; 536942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _prop_list_t *const properties;\n"; 537042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 537142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 5372fd4ce2c9d64032b42c037b2af3bf4ebeac5bbac8Fariborz Jahanian Result += "__declspec(dllimport) extern struct objc_cache _objc_empty_cache;\n"; 53733f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "extern void *_objc_empty_vtable;\n"; 53743f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 537564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian meta_data_declared = true; 537664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 537764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 5378da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianstatic void Write_protocol_list_t_TypeDecl(std::string &Result, 5379da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian long super_protocol_count) { 5380da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "struct /*_protocol_list_t*/"; Result += " {\n"; 5381da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\tlong protocol_count; // Note, this is 32/64 bit\n"; 5382da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\tstruct _protocol_t *super_protocols["; 5383da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += utostr(super_protocol_count); Result += "];\n"; 5384da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "}"; 5385da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian} 5386da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 538777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanianstatic void Write_method_list_t_TypeDecl(std::string &Result, 538877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian unsigned int method_count) { 538977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "struct /*_method_list_t*/"; Result += " {\n"; 539077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _objc_method)\n"; 539177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tunsigned int method_count;\n"; 539277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tstruct _objc_method method_list["; 539377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += utostr(method_count); Result += "];\n"; 539477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "}"; 539577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian} 539677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 5397da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanianstatic void Write__prop_list_t_TypeDecl(std::string &Result, 5398da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian unsigned int property_count) { 5399da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "struct /*_prop_list_t*/"; Result += " {\n"; 5400da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _prop_t)\n"; 5401da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tunsigned int count_of_properties;\n"; 5402da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tstruct _prop_t prop_list["; 5403da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += utostr(property_count); Result += "];\n"; 5404da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "}"; 5405da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian} 5406da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 5407ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanianstatic void Write__ivar_list_t_TypeDecl(std::string &Result, 5408ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian unsigned int ivar_count) { 5409ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "struct /*_ivar_list_t*/"; Result += " {\n"; 5410ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _prop_t)\n"; 5411ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tunsigned int count;\n"; 5412ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tstruct _ivar_t ivar_list["; 5413ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += utostr(ivar_count); Result += "];\n"; 5414ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "}"; 5415ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian} 5416ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5417da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianstatic void Write_protocol_list_initializer(ASTContext *Context, std::string &Result, 5418da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian ArrayRef<ObjCProtocolDecl *> SuperProtocols, 5419da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian StringRef VarName, 5420da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian StringRef ProtocolName) { 5421da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (SuperProtocols.size() > 0) { 5422da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\nstatic "; 5423da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Write_protocol_list_t_TypeDecl(Result, SuperProtocols.size()); 5424da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += " "; Result += VarName; 5425da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += ProtocolName; 5426da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 5427da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t"; Result += utostr(SuperProtocols.size()); Result += ",\n"; 5428da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (unsigned i = 0, e = SuperProtocols.size(); i < e; i++) { 5429da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian ObjCProtocolDecl *SuperPD = SuperProtocols[i]; 5430da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t&"; Result += "_OBJC_PROTOCOL_"; 5431da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += SuperPD->getNameAsString(); 5432da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (i == e-1) 5433da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\n};\n"; 5434da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian else 5435da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += ",\n"; 5436da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 5437da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 5438da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian} 5439da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 544090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanianstatic void Write_method_list_t_initializer(RewriteModernObjC &RewriteObj, 544190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian ASTContext *Context, std::string &Result, 544277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ArrayRef<ObjCMethodDecl *> Methods, 544377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian StringRef VarName, 544490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian StringRef TopLevelDeclName, 544590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian bool MethodImpl) { 544677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (Methods.size() > 0) { 544777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\nstatic "; 544877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Write_method_list_t_TypeDecl(Result, Methods.size()); 544977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += " "; Result += VarName; 545090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += TopLevelDeclName; 545177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 545277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t"; Result += "sizeof(_objc_method)"; Result += ",\n"; 545377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t"; Result += utostr(Methods.size()); Result += ",\n"; 545477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (unsigned i = 0, e = Methods.size(); i < e; i++) { 545577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = Methods[i]; 545677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (i == 0) 545777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t{{(struct objc_selector *)\""; 545877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian else 545977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t{(struct objc_selector *)\""; 546077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += (MD)->getSelector().getAsString(); Result += "\""; 546177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += ", "; 546277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::string MethodTypeString; 546377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Context->getObjCEncodingForMethodDecl(MD, MethodTypeString); 546477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\""; Result += MethodTypeString; Result += "\""; 546577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += ", "; 546690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!MethodImpl) 546790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "0"; 546890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian else { 546990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "(void *)"; 547090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += RewriteObj.MethodInternalNames[MD]; 547190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian } 547277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (i == e-1) 547390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "}}\n"; 547477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian else 547590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Result += "},\n"; 547677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 547777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "};\n"; 547877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 547977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian} 548077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 5481f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanianstatic void Write_prop_list_t_initializer(RewriteModernObjC &RewriteObj, 5482da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ASTContext *Context, std::string &Result, 5483da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ArrayRef<ObjCPropertyDecl *> Properties, 5484da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian const Decl *Container, 5485da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian StringRef VarName, 5486da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian StringRef ProtocolName) { 5487da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (Properties.size() > 0) { 5488da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\nstatic "; 5489da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Write__prop_list_t_TypeDecl(Result, Properties.size()); 5490da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += " "; Result += VarName; 5491da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += ProtocolName; 5492da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 5493da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t"; Result += "sizeof(_prop_t)"; Result += ",\n"; 5494da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t"; Result += utostr(Properties.size()); Result += ",\n"; 5495da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian for (unsigned i = 0, e = Properties.size(); i < e; i++) { 5496da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ObjCPropertyDecl *PropDecl = Properties[i]; 5497da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (i == 0) 5498da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t{{\""; 5499da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian else 5500da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t{\""; 5501da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += PropDecl->getName(); Result += "\","; 5502da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian std::string PropertyTypeString, QuotePropertyTypeString; 5503da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Context->getObjCEncodingForPropertyDecl(PropDecl, Container, PropertyTypeString); 5504da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian RewriteObj.QuoteDoublequotes(PropertyTypeString, QuotePropertyTypeString); 5505da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\""; Result += QuotePropertyTypeString; Result += "\""; 5506da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (i == e-1) 5507da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "}}\n"; 5508da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian else 5509da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "},\n"; 5510da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian } 5511da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "};\n"; 5512da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian } 5513da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian} 5514da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 55156ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian// Metadata flags 55166ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanianenum MetaDataDlags { 55176ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS = 0x0, 55186ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_META = 0x1, 55196ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_ROOT = 0x2, 55206ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian OBJC2_CLS_HIDDEN = 0x10, 55216ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_EXCEPTION = 0x20, 55226ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 55236ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian /// (Obsolete) ARC-specific: this class has a .release_ivars method 55246ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_HAS_IVAR_RELEASER = 0x40, 55256ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian /// class was compiled with -fobjc-arr 55266ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CLS_COMPILED_BY_ARC = 0x80 // (1<<7) 55276ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian}; 55286ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 5529f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanianstatic void Write__class_ro_t_initializer(ASTContext *Context, std::string &Result, 5530f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian unsigned int flags, 5531f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian const std::string &InstanceStart, 5532f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian const std::string &InstanceSize, 5533f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCMethodDecl *>baseMethods, 5534f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCProtocolDecl *>baseProtocols, 5535f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCIvarDecl *>ivars, 5536f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ArrayRef<ObjCPropertyDecl *>Properties, 5537f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian StringRef VarName, 5538f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian StringRef ClassName) { 5539f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\nstatic struct _class_ro_t "; 5540f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += VarName; Result += ClassName; 5541f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 5542f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\t"; 5543f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += llvm::utostr(flags); Result += ", "; 5544f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += InstanceStart; Result += ", "; 5545f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += InstanceSize; Result += ", \n"; 5546f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\t"; 5547de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian const llvm::Triple &Triple(Context->getTargetInfo().getTriple()); 5548de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian if (Triple.getArch() == llvm::Triple::x86_64) 5549de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian // uint32_t const reserved; // only when building for 64bit targets 5550de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian Result += "(unsigned int)0, \n\t"; 5551f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // const uint8_t * const ivarLayout; 5552f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 5553f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "\""; Result += ClassName; Result += "\",\n\t"; 55546ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian bool metaclass = ((flags & CLS_META) != 0); 5555f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (baseMethods.size() > 0) { 5556f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _method_list_t *)&"; 55576ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (metaclass) 55586ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Result += "_OBJC_$_CLASS_METHODS_"; 55596ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian else 55606ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Result += "_OBJC_$_INSTANCE_METHODS_"; 55616ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Result += ClassName; 5562f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n\t"; 5563f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 5564f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 5565f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 5566f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 55676ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!metaclass && baseProtocols.size() > 0) { 5568f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _objc_protocol_list *)&"; 5569f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "_OBJC_CLASS_PROTOCOLS_$_"; Result += ClassName; 5570f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n\t"; 5571f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 5572f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 5573f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 5574f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 55756ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!metaclass && ivars.size() > 0) { 5576f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _ivar_list_t *)&"; 5577f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "_OBJC_$_INSTANCE_VARIABLES_"; Result += ClassName; 5578f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n\t"; 5579f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 5580f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 5581f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 5582f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 5583f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // weakIvarLayout 5584f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n\t"; 55856ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!metaclass && Properties.size() > 0) { 5586f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "(const struct _prop_list_t *)&"; 5587eeabf381862e53519cd0f9f65c1cc2a983c72159Fariborz Jahanian Result += "_OBJC_$_PROP_LIST_"; Result += ClassName; 5588f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += ",\n"; 5589f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 5590f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 5591f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "0, \n"; 5592f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 5593f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Result += "};\n"; 5594f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian} 5595f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 55963f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanianstatic void Write_class_t(ASTContext *Context, std::string &Result, 55973f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian StringRef VarName, 55983f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian const ObjCInterfaceDecl *CDecl, bool metadata) { 55993f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 56003f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian if (metadata && !CDecl->getSuperClass()) { 56013f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian // Need to handle a case of use of forward declaration. 5602ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "\n"; 5603ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian if (CDecl->getImplementation()) 5604ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "__declspec(dllexport) "; 5605ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "extern struct _class_t OBJC_CLASS_$_"; 56063f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 56073f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ";\n"; 56083f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 56093f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian // Also, for possibility of 'super' metadata class not having been defined yet. 56103f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian if (CDecl->getSuperClass()) { 5611ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "\n"; 5612ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian if (CDecl->getSuperClass()->getImplementation()) 5613ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "__declspec(dllexport) "; 5614ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += "extern struct _class_t "; 5615ce0d89744bdad195c991362bd6fc7b334e5c2d52Fariborz Jahanian Result += VarName; 56163f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getSuperClass()->getNameAsString(); 56173f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ";\n"; 56183f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 56193f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 5620e57303c8ce83ce058dac1fd4b963e68762666b9aFariborz Jahanian Result += "\n__declspec(dllexport) struct _class_t "; Result += VarName; Result += CDecl->getNameAsString(); 56213f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_data\"))) = {\n"; 56223f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "\t"; 56233f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian if (metadata) { 56243f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian if (CDecl->getSuperClass()) { 56253f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&"; Result += VarName; 56263f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getSuperClass()->getNameAsString(); 56273f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 56283f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&"; Result += VarName; 56293f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getSuperClass()->getNameAsString(); 56303f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 56313f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 56323f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else { 56333f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&"; Result += VarName; 56343f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 56353f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 56363f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&OBJC_CLASS_$_"; Result += CDecl->getNameAsString(); 56373f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 56383f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 56393f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 56403f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else { 56413f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&OBJC_METACLASS_$_"; 56423f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 56433f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 56443f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian if (CDecl->getSuperClass()) { 56453f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&"; Result += VarName; 56463f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getSuperClass()->getNameAsString(); 56473f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n\t"; 56483f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 56493f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else 56503f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "0,\n\t"; 56513f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian } 56523f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "(void *)&_objc_empty_cache,\n\t"; 56533f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "(void *)&_objc_empty_vtable,\n\t"; 56543f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian if (metadata) 56553f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&_OBJC_METACLASS_RO_$_"; 56563f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian else 56573f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += "&_OBJC_CLASS_RO_$_"; 56583f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += CDecl->getNameAsString(); 56593f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Result += ",\n};\n"; 56603f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian} 56613f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 56626118612837887cbae365676d53cc3e1c258249aaFariborz Jahanianstatic void Write_category_t(RewriteModernObjC &RewriteObj, ASTContext *Context, 56636118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian std::string &Result, 56646118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian StringRef CatName, 566588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian ObjCInterfaceDecl *ClassDecl, 56666118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCMethodDecl *> InstanceMethods, 56676118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCMethodDecl *> ClassMethods, 56686118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCProtocolDecl *> RefedProtocols, 56696118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ArrayRef<ObjCPropertyDecl *> ClassProperties) { 567088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 567188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian StringRef ClassName = ClassDecl->getNameAsString(); 56728c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian // must declare an extern class object in case this class is not implemented 56738c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian // in this TU. 567488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "\n"; 567588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (ClassDecl->getImplementation()) 567688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "__declspec(dllexport) "; 567788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 567888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "extern struct _class_t "; 56798c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian Result += "OBJC_CLASS_$_"; Result += ClassName; 56808c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian Result += ";\n"; 56818c00a1b66a1110d8354c3c48f0cbd037d5b04c38Fariborz Jahanian 56826118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\nstatic struct _category_t "; 56836118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_$_CATEGORY_"; 56846118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 56856118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = \n"; 56866118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "{\n"; 56876118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t\""; Result += ClassName; Result += "\",\n"; 56886118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t&"; Result += "OBJC_CLASS_$_"; Result += ClassName; 56896118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 56906118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (InstanceMethods.size() > 0) { 56916118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _method_list_t *)&"; 56926118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_$_CATEGORY_INSTANCE_METHODS_"; 56936118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 56946118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 56956118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 56966118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 56976118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 56986118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 56996118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (ClassMethods.size() > 0) { 57006118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _method_list_t *)&"; 57016118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_$_CATEGORY_CLASS_METHODS_"; 57026118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 57036118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 57046118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 57056118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 57066118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 57076118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 57086118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (RefedProtocols.size() > 0) { 57096118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _protocol_list_t *)&"; 57106118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_OBJC_CATEGORY_PROTOCOLS_$_"; 57116118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 57126118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 57136118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 57146118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 57156118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 57166118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 57176118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (ClassProperties.size() > 0) { 57186118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t(const struct _prop_list_t *)&"; Result += "_OBJC_$_PROP_LIST_"; 57196118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ClassName; Result += "_$_"; Result += CatName; 57206118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 57216118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 57226118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian else 57236118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t0,\n"; 57246118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 57256118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "};\n"; 57266118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian} 57276118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 5728e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanianstatic void Write__extendedMethodTypes_initializer(RewriteModernObjC &RewriteObj, 5729e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ASTContext *Context, std::string &Result, 5730e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ArrayRef<ObjCMethodDecl *> Methods, 5731e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian StringRef VarName, 5732e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian StringRef ProtocolName) { 5733e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (Methods.size() == 0) 5734e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian return; 5735e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 5736e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\nstatic const char *"; 5737e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += VarName; Result += ProtocolName; 5738e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += " [] __attribute__ ((used, section (\"__DATA,__objc_const\"))) = \n"; 5739e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "{\n"; 5740e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = Methods.size(); i < e; i++) { 5741e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ObjCMethodDecl *MD = Methods[i]; 5742e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian std::string MethodTypeString, QuoteMethodTypeString; 5743e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Context->getObjCEncodingForMethodDecl(MD, MethodTypeString, true); 5744e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian RewriteObj.QuoteDoublequotes(MethodTypeString, QuoteMethodTypeString); 5745e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t\""; Result += QuoteMethodTypeString; Result += "\""; 5746e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (i == e-1) 5747e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\n};\n"; 5748e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian else { 5749e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += ",\n"; 5750e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 5751e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 5752e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian} 5753e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 575440a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanianstatic void Write_IvarOffsetVar(ASTContext *Context, 575540a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian std::string &Result, 5756db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian ArrayRef<ObjCIvarDecl *> Ivars, 57577cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian ObjCInterfaceDecl *CDecl) { 5758db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian // FIXME. visibilty of offset symbols may have to be set; for Darwin 5759db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian // this is what happens: 5760db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian /** 5761db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian if (Ivar->getAccessControl() == ObjCIvarDecl::Private || 5762db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Ivar->getAccessControl() == ObjCIvarDecl::Package || 5763db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Class->getVisibility() == HiddenVisibility) 5764db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Visibility shoud be: HiddenVisibility; 5765db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian else 5766db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Visibility shoud be: DefaultVisibility; 5767db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian */ 5768db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian 576907e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian Result += "\n"; 577007e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian for (unsigned i =0, e = Ivars.size(); i < e; i++) { 577107e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian ObjCIvarDecl *IvarDecl = Ivars[i]; 577240a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (Context->getLangOpts().MicrosoftExt) 577340a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian Result += "__declspec(allocate(\".objc_ivar$B\")) "; 577440a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian 577540a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian if (!Context->getLangOpts().MicrosoftExt || 577640a777a157c00cc7e9c9db7197017d83311c76c6Fariborz Jahanian IvarDecl->getAccessControl() == ObjCIvarDecl::Private || 5777117591f6247b186a4161459ad8603563cfd1c54cFariborz Jahanian IvarDecl->getAccessControl() == ObjCIvarDecl::Package) 5778d1c84d3c0206abdf8787dc5c00aaf3b016ed3226Fariborz Jahanian Result += "unsigned long int "; 5779d1c84d3c0206abdf8787dc5c00aaf3b016ed3226Fariborz Jahanian else 5780d1c84d3c0206abdf8787dc5c00aaf3b016ed3226Fariborz Jahanian Result += "__declspec(dllexport) unsigned long int "; 57817cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(CDecl, IvarDecl, Result); 578207e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_ivar\")))"; 578307e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian Result += " = "; 578407e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian if (IvarDecl->isBitField()) { 578507e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian // FIXME: The hack below doesn't work for bitfields. For now, we simply 578607e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian // place all bitfields at offset 0. 578707e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian Result += "0;\n"; 578807e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian } 578907e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian else { 5790db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Result += "__OFFSETOFIVAR__(struct "; 57917cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += CDecl->getNameAsString(); 5792db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Result += "_IMPL, "; 5793db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Result += IvarDecl->getName(); Result += ");\n"; 5794db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian } 5795db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian } 5796db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian} 5797db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian 5798ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanianstatic void Write__ivar_list_t_initializer(RewriteModernObjC &RewriteObj, 5799ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ASTContext *Context, std::string &Result, 5800ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ArrayRef<ObjCIvarDecl *> Ivars, 5801ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian StringRef VarName, 58027cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian ObjCInterfaceDecl *CDecl) { 5803ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (Ivars.size() > 0) { 58047cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Write_IvarOffsetVar(Context, Result, Ivars, CDecl); 580507e5288c6532c8a61278fab4e0ae058ec81a0f5cFariborz Jahanian 5806ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\nstatic "; 5807ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Write__ivar_list_t_TypeDecl(Result, Ivars.size()); 5808ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += " "; Result += VarName; 58097cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += CDecl->getNameAsString(); 5810ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 5811ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t"; Result += "sizeof(_ivar_t)"; Result += ",\n"; 5812ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t"; Result += utostr(Ivars.size()); Result += ",\n"; 5813ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian for (unsigned i =0, e = Ivars.size(); i < e; i++) { 5814ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ObjCIvarDecl *IvarDecl = Ivars[i]; 5815ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (i == 0) 5816ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t{{"; 5817ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian else 5818ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t {"; 58197cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian Result += "(unsigned long int *)&"; 58207cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(CDecl, IvarDecl, Result); 5821db64923b255daeb58c2168eb890e480c93a6c41cFariborz Jahanian Result += ", "; 5822ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5823ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\""; Result += IvarDecl->getName(); Result += "\", "; 5824ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian std::string IvarTypeString, QuoteIvarTypeString; 5825ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Context->getObjCEncodingForType(IvarDecl->getType(), IvarTypeString, 5826ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IvarDecl); 5827ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian RewriteObj.QuoteDoublequotes(IvarTypeString, QuoteIvarTypeString); 5828ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\""; Result += QuoteIvarTypeString; Result += "\", "; 5829ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 58308f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // FIXME. this alignment represents the host alignment and need be changed to 58318f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian // represent the target alignment. 58328f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian unsigned Align = Context->getTypeAlign(IvarDecl->getType())/8; 58338f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian Align = llvm::Log2_32(Align); 5834ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += llvm::utostr(Align); Result += ", "; 5835a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian CharUnits Size = Context->getTypeSizeInChars(IvarDecl->getType()); 5836a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += llvm::utostr(Size.getQuantity()); 5837ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (i == e-1) 5838ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "}}\n"; 5839ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian else 5840ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "},\n"; 5841ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian } 5842ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "};\n"; 5843ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian } 5844ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian} 5845ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 584664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCProtocolMetaData - Rewrite protocols meta-data. 5847da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCProtocolMetaData(ObjCProtocolDecl *PDecl, 5848da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::string &Result) { 584964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 585064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Do not synthesize the protocol more than once. 585164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSynthesizedProtocols.count(PDecl->getCanonicalDecl())) 585264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 5853de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian WriteModernMetadataDeclarations(Context, Result); 585464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 585564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Def = PDecl->getDefinition()) 585664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PDecl = Def; 5857da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // Must write out all protocol definitions in current qualifier list, 5858da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // and in their nested qualifiers before writing out current definition. 5859da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (ObjCProtocolDecl::protocol_iterator I = PDecl->protocol_begin(), 5860da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian E = PDecl->protocol_end(); I != E; ++I) 5861da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Result); 586264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 586377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian // Construct method lists. 586477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::vector<ObjCMethodDecl *> InstanceMethods, ClassMethods; 586577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::vector<ObjCMethodDecl *> OptInstanceMethods, OptClassMethods; 586677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (ObjCProtocolDecl::instmeth_iterator 586777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I = PDecl->instmeth_begin(), E = PDecl->instmeth_end(); 586877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I != E; ++I) { 586977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = *I; 587077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (MD->getImplementationControl() == ObjCMethodDecl::Optional) { 587177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian OptInstanceMethods.push_back(MD); 587277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } else { 587377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian InstanceMethods.push_back(MD); 587477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 587577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 587677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 587777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (ObjCProtocolDecl::classmeth_iterator 587877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I = PDecl->classmeth_begin(), E = PDecl->classmeth_end(); 587977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I != E; ++I) { 588077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = *I; 588177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (MD->getImplementationControl() == ObjCMethodDecl::Optional) { 588277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian OptClassMethods.push_back(MD); 588377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } else { 588477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ClassMethods.push_back(MD); 588577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 588677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 5887e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian std::vector<ObjCMethodDecl *> AllMethods; 5888e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = InstanceMethods.size(); i < e; i++) 5889e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(InstanceMethods[i]); 5890e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = ClassMethods.size(); i < e; i++) 5891e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(ClassMethods[i]); 5892e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = OptInstanceMethods.size(); i < e; i++) 5893e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(OptInstanceMethods[i]); 5894e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = OptClassMethods.size(); i < e; i++) 5895e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(OptClassMethods[i]); 5896e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 5897e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Write__extendedMethodTypes_initializer(*this, Context, Result, 5898e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods, 5899e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian "_OBJC_PROTOCOL_METHOD_TYPES_", 5900e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian PDecl->getNameAsString()); 5901da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // Protocol's super protocol list 5902da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::vector<ObjCProtocolDecl *> SuperProtocols; 5903da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (ObjCProtocolDecl::protocol_iterator I = PDecl->protocol_begin(), 5904da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian E = PDecl->protocol_end(); I != E; ++I) 5905da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian SuperProtocols.push_back(*I); 5906da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 5907da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Write_protocol_list_initializer(Context, Result, SuperProtocols, 5908da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian "_OBJC_PROTOCOL_REFS_", 5909da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian PDecl->getNameAsString()); 5910da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 591190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, InstanceMethods, 591277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_INSTANCE_METHODS_", 591390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 591477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 591590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, ClassMethods, 591677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_CLASS_METHODS_", 591790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 591877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 591990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, OptInstanceMethods, 592077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_OPT_INSTANCE_METHODS_", 592190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 592277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 592390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, OptClassMethods, 592477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_OPT_CLASS_METHODS_", 592590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PDecl->getNameAsString(), false); 592677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 5927da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian // Protocol's property metadata. 5928da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian std::vector<ObjCPropertyDecl *> ProtocolProperties; 5929da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian for (ObjCContainerDecl::prop_iterator I = PDecl->prop_begin(), 5930da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian E = PDecl->prop_end(); I != E; ++I) 5931da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ProtocolProperties.push_back(*I); 5932da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 5933f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Write_prop_list_t_initializer(*this, Context, Result, ProtocolProperties, 5934da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian /* Container */0, 5935da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian "_OBJC_PROTOCOL_PROPERTIES_", 5936da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian PDecl->getNameAsString()); 5937da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 593882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian // Writer out root metadata for current protocol: struct _protocol_t 59391ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "\n"; 59401ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 59411ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".datacoal_nt$B\")) "; 594230650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "struct _protocol_t _OBJC_PROTOCOL_"; 594382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); 594482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__datacoal_nt,coalesced\"))) = {\n"; 594582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; // id is; is null 594682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t\""; Result += PDecl->getNameAsString(); Result += "\",\n"; 5947da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (SuperProtocols.size() > 0) { 5948da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t(const struct _protocol_list_t *)&"; Result += "_OBJC_PROTOCOL_REFS_"; 5949da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 5950da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 5951da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian else 5952da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t0,\n"; 595382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (InstanceMethods.size() > 0) { 595482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_INSTANCE_METHODS_"; 595582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 595664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 595782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 595882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 595982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 596082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (ClassMethods.size() > 0) { 596182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_CLASS_METHODS_"; 596282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 596364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 596482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 596582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 596664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 596782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (OptInstanceMethods.size() > 0) { 596882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_OPT_INSTANCE_METHODS_"; 596982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 597064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 597182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 597282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 597364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 597482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (OptClassMethods.size() > 0) { 597582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_OPT_CLASS_METHODS_"; 597682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 597764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 597864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 597982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 598082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 598182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (ProtocolProperties.size() > 0) { 598282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct _prop_list_t *)&_OBJC_PROTOCOL_PROPERTIES_"; 598382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 598464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 598564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 598682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 598764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 598882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t"; Result += "sizeof(_protocol_t)"; Result += ",\n"; 598982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 599082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 5991e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (AllMethods.size() > 0) { 5992e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t(const char **)&"; Result += "_OBJC_PROTOCOL_METHOD_TYPES_"; 5993e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += PDecl->getNameAsString(); 5994e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\n};\n"; 5995e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 5996e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian else 5997e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t0\n};\n"; 599810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian 599910cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // Use this protocol meta-data to build protocol list table in section 600010cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // .objc_protolist$B 600110cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // Unspecified visibility means 'private extern'. 600210cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian if (LangOpts.MicrosoftExt) 600310cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "__declspec(allocate(\".objc_protolist$B\")) "; 600410cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "struct _protocol_t *"; 600510cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "_OBJC_LABEL_PROTOCOL_$_"; Result += PDecl->getNameAsString(); 600610cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += " = &_OBJC_PROTOCOL_"; Result += PDecl->getNameAsString(); 600710cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += ";\n"; 600882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 600964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Mark this protocol as having been generated. 601064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ObjCSynthesizedProtocols.insert(PDecl->getCanonicalDecl())) 601164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("protocol already synthesized"); 601264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 601364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 601464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 601564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCProtocolListMetaData( 601664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Protocols, 601764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, StringRef ClassName, 601864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 601964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Protocols.empty()) return; 602064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 602164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i != Protocols.size(); i++) 6022da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(Protocols[i], Result); 602364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 602464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output the top lovel protocol meta-data for the class. 602564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_protocol_list { 602664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_protocol_list *next; 602764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int protocol_count; 602864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_protocol *class_protocols[]; 602964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 603064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 60311ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "\n"; 60321ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 60331ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".cat_cls_meth$B\")) "; 60341ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "static struct {\n"; 603564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol_list *next;\n"; 603664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tint protocol_count;\n"; 603764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol *class_protocols["; 603864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(Protocols.size()); 603964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "];\n} _OBJC_"; 604064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += prefix; 604164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_PROTOCOLS_"; 604264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassName; 604364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __cat_cls_meth\")))= " 604464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "{\n\t0, "; 604564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(Protocols.size()); 604664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 604764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 604864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,{&_OBJC_PROTOCOL_"; 604964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += Protocols[0]->getNameAsString(); 605064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " \n"; 605164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 605264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 1; i != Protocols.size(); i++) { 605364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t ,&_OBJC_PROTOCOL_"; 605464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += Protocols[i]->getNameAsString(); 605564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 605664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 605764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t }\n};\n"; 605864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 605964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 60606ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian/// hasObjCExceptionAttribute - Return true if this class or any super 60616ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian/// class has the __objc_exception__ attribute. 60626ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian/// FIXME. Move this to ASTContext.cpp as it is also used for IRGen. 60636ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanianstatic bool hasObjCExceptionAttribute(ASTContext &Context, 60646ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian const ObjCInterfaceDecl *OID) { 60656ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (OID->hasAttr<ObjCExceptionAttr>()) 60666ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian return true; 60676ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (const ObjCInterfaceDecl *Super = OID->getSuperClass()) 60686ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian return hasObjCExceptionAttribute(Context, Super); 60696ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian return false; 60706ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian} 6071f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 607264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl, 607364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 607464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *CDecl = IDecl->getClassInterface(); 607564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 607664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Explicitly declared @interface's are already synthesized. 6077ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (CDecl->isImplicitInterfaceDecl()) 6078ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian assert(false && 6079ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian "Legacy implicit interface rewriting not supported in moder abi"); 60808f1fed036eb78b747725560c10ad29d370d6ef62Fariborz Jahanian 6081de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian WriteModernMetadataDeclarations(Context, Result); 6082ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian SmallVector<ObjCIvarDecl *, 8> IVars; 6083ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 6084ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian for (ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 6085ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IVD; IVD = IVD->getNextIvar()) { 6086ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian // Ignore unnamed bit-fields. 6087ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (!IVD->getDeclName()) 6088ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian continue; 6089ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IVars.push_back(IVD); 609064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 609164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6092ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Write__ivar_list_t_initializer(*this, Context, Result, IVars, 6093f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian "_OBJC_$_INSTANCE_VARIABLES_", 60947cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian CDecl); 609590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 609690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian // Build _objc_method_list for class's instance methods if needed 609790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 609890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian InstanceMethods(IDecl->instmeth_begin(), IDecl->instmeth_end()); 609990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 610090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian // If any of our property implementations have associated getters or 610190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian // setters, produce metadata for them as well. 610290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian for (ObjCImplDecl::propimpl_iterator Prop = IDecl->propimpl_begin(), 610390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian PropEnd = IDecl->propimpl_end(); 610490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Prop != PropEnd; ++Prop) { 610590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if ((*Prop)->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 610690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 610790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!(*Prop)->getPropertyIvarDecl()) 610890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 610990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian ObjCPropertyDecl *PD = (*Prop)->getPropertyDecl(); 611090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!PD) 611190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 611290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (ObjCMethodDecl *Getter = PD->getGetterMethodDecl()) 611390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!Getter->isDefined()) 611490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian InstanceMethods.push_back(Getter); 611590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (PD->isReadOnly()) 611690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian continue; 611790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (ObjCMethodDecl *Setter = PD->getSetterMethodDecl()) 611890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian if (!Setter->isDefined()) 611990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian InstanceMethods.push_back(Setter); 612090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian } 612190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 612290af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, InstanceMethods, 612390af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian "_OBJC_$_INSTANCE_METHODS_", 612490af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian IDecl->getNameAsString(), true); 612590af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 612690af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 612790af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian ClassMethods(IDecl->classmeth_begin(), IDecl->classmeth_end()); 612890af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 612990af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, ClassMethods, 613090af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian "_OBJC_$_CLASS_METHODS_", 613190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian IDecl->getNameAsString(), true); 61320a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian 61330a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // Protocols referenced in class declaration? 61340a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // Protocol's super protocol list 61350a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian std::vector<ObjCProtocolDecl *> RefedProtocols; 61360a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Protocols = CDecl->getReferencedProtocols(); 61370a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian for (ObjCList<ObjCProtocolDecl>::iterator I = Protocols.begin(), 61380a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian E = Protocols.end(); 61390a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian I != E; ++I) { 61400a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian RefedProtocols.push_back(*I); 61410a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // Must write out all protocol definitions in current qualifier list, 61420a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian // and in their nested qualifiers before writing out current definition. 61430a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Result); 61440a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian } 61450a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian 61460a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian Write_protocol_list_initializer(Context, Result, 61470a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian RefedProtocols, 61480a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian "_OBJC_CLASS_PROTOCOLS_$_", 61490a525340ada570897ab34918c98de1cf94ad7770Fariborz Jahanian IDecl->getNameAsString()); 6150f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6151f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // Protocol's property metadata. 6152f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian std::vector<ObjCPropertyDecl *> ClassProperties; 6153f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian for (ObjCContainerDecl::prop_iterator I = CDecl->prop_begin(), 6154f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian E = CDecl->prop_end(); I != E; ++I) 6155f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ClassProperties.push_back(*I); 6156f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6157f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Write_prop_list_t_initializer(*this, Context, Result, ClassProperties, 6158f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian /* Container */0, 6159eeabf381862e53519cd0f9f65c1cc2a983c72159Fariborz Jahanian "_OBJC_$_PROP_LIST_", 6160f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian CDecl->getNameAsString()); 616190af4e22a8fb83f159b512bfd48fb77699337b3dFariborz Jahanian 6162f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 61636ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian // Data for initializing _class_ro_t metaclass meta-data 61646ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian uint32_t flags = CLS_META; 61656ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian std::string InstanceSize; 61666ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian std::string InstanceStart; 61676ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 61686ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 61696ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian bool classIsHidden = CDecl->getVisibility() == HiddenVisibility; 61706ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (classIsHidden) 61716ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= OBJC2_CLS_HIDDEN; 61726ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 61736ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (!CDecl->getSuperClass()) 61746ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian // class is root 61756ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= CLS_ROOT; 61766ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceSize = "sizeof(struct _class_t)"; 61776ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceStart = InstanceSize; 61786ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian Write__class_ro_t_initializer(Context, Result, flags, 61796ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceStart, InstanceSize, 61806ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian ClassMethods, 61816ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 0, 61826ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 0, 61836ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 0, 61846ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian "_OBJC_METACLASS_RO_$_", 61856ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian CDecl->getNameAsString()); 61866ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 61876ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 6188f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // Data for initializing _class_ro_t meta-data 61896ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags = CLS; 61906ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (classIsHidden) 61916ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= OBJC2_CLS_HIDDEN; 61926ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 61936ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian if (hasObjCExceptionAttribute(*Context, CDecl)) 61946ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian flags |= CLS_EXCEPTION; 61956ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian 6196f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (!CDecl->getSuperClass()) 6197f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian // class is root 6198f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian flags |= CLS_ROOT; 6199f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 62006ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceSize.clear(); 62016ade343d180d029fc61b17377553deab99527ff2Fariborz Jahanian InstanceStart.clear(); 6202f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (!ObjCSynthesizedStructs.count(CDecl)) { 6203f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize = "0"; 6204f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart = "0"; 6205f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6206f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else { 6207f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize = "sizeof(struct "; 6208f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize += CDecl->getNameAsString(); 6209f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceSize += "_IMPL)"; 6210f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian 6211f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 6212f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian if (IVD) { 6213f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart += "__OFFSETOFIVAR__(struct "; 6214f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart += CDecl->getNameAsString(); 6215f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart += "_IMPL, "; 6216f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart += IVD->getNameAsString(); 6217f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart += ")"; 6218f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6219f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian else 6220f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart = InstanceSize; 6221f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian } 6222f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian Write__class_ro_t_initializer(Context, Result, flags, 6223f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceStart, InstanceSize, 6224f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian InstanceMethods, 6225f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian RefedProtocols, 6226f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian IVars, 6227f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian ClassProperties, 6228f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian "_OBJC_CLASS_RO_$_", 6229f1c1d9a99dd02b2cd9c4052ec2ec759783175e5eFariborz Jahanian CDecl->getNameAsString()); 62303f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 62313f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Write_class_t(Context, Result, 62323f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian "OBJC_METACLASS_$_", 62333f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian CDecl, /*metaclass*/true); 62343f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 62353f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian Write_class_t(Context, Result, 62363f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian "OBJC_CLASS_$_", 62373f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian CDecl, /*metaclass*/false); 623888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 623988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (ImplementationIsNonLazy(IDecl)) 624088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian DefinedNonLazyClasses.push_back(CDecl); 62413f77c7b56d515a6756dba63a3dfa0a332711a779Fariborz Jahanian 624264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 624364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 624464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteMetaDataIntoBuffer(std::string &Result) { 624564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int ClsDefCount = ClassImplementation.size(); 624664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int CatDefCount = CategoryImplementation.size(); 624764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 624864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For each implemented class, write out all its meta data. 624964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) 625064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCClassMetaData(ClassImplementation[i], Result); 625164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 625264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For each implemented category, write out all its meta data. 625364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < CatDefCount; i++) 625464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCCategoryImplDecl(CategoryImplementation[i], Result); 625564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6256df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian if (ClsDefCount > 0) { 62571ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 62581ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".objc_classlist$B\")) "; 6259df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += "static struct _class_t *L_OBJC_LABEL_CLASS_$ ["; 6260df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += llvm::utostr(ClsDefCount); Result += "]"; 6261df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += 6262df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian " __attribute__((used, section (\"__DATA, __objc_classlist," 6263df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian "regular,no_dead_strip\")))= {\n"; 6264df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) { 6265df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += "\t&OBJC_CLASS_$_"; 6266df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += ClassImplementation[i]->getNameAsString(); 6267df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += ",\n"; 626864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 6269df79567796fabb406156d881ebd3319aa55b6f83Fariborz Jahanian Result += "};\n"; 627088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 627188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (!DefinedNonLazyClasses.empty()) { 627288f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (LangOpts.MicrosoftExt) 627388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "__declspec(allocate(\".objc_nlclslist$B\")) \n"; 627488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "static struct _class_t *_OBJC_LABEL_NONLAZY_CLASS_$[] = {\n\t"; 627588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian for (unsigned i = 0, e = DefinedNonLazyClasses.size(); i < e; i++) { 627688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "\t&OBJC_CLASS_$_"; Result += DefinedNonLazyClasses[i]->getNameAsString(); 627788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += ",\n"; 627888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 627988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "};\n"; 628088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 628164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 62826118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 62836118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian if (CatDefCount > 0) { 62841ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) 62851ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".objc_catlist$B\")) "; 62866118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "static struct _category_t *L_OBJC_LABEL_CATEGORY_$ ["; 62876118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += llvm::utostr(CatDefCount); Result += "]"; 62886118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += 62896118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian " __attribute__((used, section (\"__DATA, __objc_catlist," 62906118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "regular,no_dead_strip\")))= {\n"; 62916118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian for (int i = 0; i < CatDefCount; i++) { 62926118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "\t&_OBJC_$_CATEGORY_"; 62936118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += 62946118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian CategoryImplementation[i]->getClassInterface()->getNameAsString(); 62956118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "_$_"; 62966118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += CategoryImplementation[i]->getNameAsString(); 62976118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += ",\n"; 62986118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 62996118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Result += "};\n"; 63006118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian } 630188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 630288f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (!DefinedNonLazyCategories.empty()) { 630388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (LangOpts.MicrosoftExt) 630488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "__declspec(allocate(\".objc_nlcatlist$B\")) \n"; 630588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "static struct _category_t *_OBJC_LABEL_NONLAZY_CATEGORY_$[] = {\n\t"; 630688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian for (unsigned i = 0, e = DefinedNonLazyCategories.size(); i < e; i++) { 630788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "\t&_OBJC_$_CATEGORY_"; 630888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += 630988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian DefinedNonLazyCategories[i]->getClassInterface()->getNameAsString(); 631088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "_$_"; 631188f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += DefinedNonLazyCategories[i]->getNameAsString(); 631288f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += ",\n"; 631388f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 631488f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian Result += "};\n"; 631588f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian } 631664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 631764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 631810cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanianvoid RewriteModernObjC::WriteImageInfo(std::string &Result) { 631910cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian if (LangOpts.MicrosoftExt) 632010cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "__declspec(allocate(\".objc_imageinfo$B\")) \n"; 632110cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian 632210cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian Result += "static struct IMAGE_INFO { unsigned version; unsigned flag; } "; 632310cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian // version 0, ObjCABI is 2 632430650ebe652854d8bdce832cbdae1cf3c43e54aaFariborz Jahanian Result += "_OBJC_IMAGE_INFO = { 0, 2 };\n"; 632510cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian} 632610cde2fc0c1804427dd3e0c050c149ac6d415f51Fariborz Jahanian 632764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCCategoryImplDecl - Rewrite metadata for each category 632864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// implementation. 632964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *IDecl, 633064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 6331de5d946288bb3d68576c95974beb8cfd21d7043aFariborz Jahanian WriteModernMetadataDeclarations(Context, Result); 633264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = IDecl->getClassInterface(); 633364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find category declaration for this implementation. 63346118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ObjCCategoryDecl *CDecl=0; 633564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (CDecl = ClassDecl->getCategoryList(); CDecl; 633664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CDecl = CDecl->getNextClassCategory()) 633764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CDecl->getIdentifier() == IDecl->getIdentifier()) 633864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 633964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 634064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FullCategoryName = ClassDecl->getNameAsString(); 63416118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName += "_$_"; 63426118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName += CDecl->getNameAsString(); 634364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 634464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's instance methods if needed 634564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 634664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods(IDecl->instmeth_begin(), IDecl->instmeth_end()); 634764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 634864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If any of our property implementations have associated getters or 634964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // setters, produce metadata for them as well. 635064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCImplDecl::propimpl_iterator Prop = IDecl->propimpl_begin(), 635164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropEnd = IDecl->propimpl_end(); 635264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Prop != PropEnd; ++Prop) { 635364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*Prop)->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 635464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 635564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!(*Prop)->getPropertyIvarDecl()) 635664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 635764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl *PD = (*Prop)->getPropertyDecl(); 635864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!PD) 635964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 636064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Getter = PD->getGetterMethodDecl()) 636164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Getter); 636264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isReadOnly()) 636364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 636464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Setter = PD->getSetterMethodDecl()) 636564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Setter); 636664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 636764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63686118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, InstanceMethods, 63696118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_$_CATEGORY_INSTANCE_METHODS_", 63706118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName, true); 637164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63726118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 63736118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassMethods(IDecl->classmeth_begin(), IDecl->classmeth_end()); 637464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63756118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_method_list_t_initializer(*this, Context, Result, ClassMethods, 63766118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_$_CATEGORY_CLASS_METHODS_", 63776118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName, true); 637864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63796118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Protocols referenced in class declaration? 63806118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Protocol's super protocol list 63816118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian std::vector<ObjCProtocolDecl *> RefedProtocols; 6382a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis for (ObjCInterfaceDecl::protocol_iterator I = CDecl->protocol_begin(), 6383a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis E = CDecl->protocol_end(); 6384a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis 6385a5f4441de7890953460d95f4e88b9fa432b48dc2Argyrios Kyrtzidis I != E; ++I) { 63866118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RefedProtocols.push_back(*I); 63876118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Must write out all protocol definitions in current qualifier list, 63886118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // and in their nested qualifiers before writing out current definition. 63896118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RewriteObjCProtocolMetaData(*I, Result); 639064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 639164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 63926118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_protocol_list_initializer(Context, Result, 63936118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RefedProtocols, 63946118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_CATEGORY_PROTOCOLS_$_", 63956118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName); 63966118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 63976118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian // Protocol's property metadata. 63986118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian std::vector<ObjCPropertyDecl *> ClassProperties; 63996118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian for (ObjCContainerDecl::prop_iterator I = CDecl->prop_begin(), 64006118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian E = CDecl->prop_end(); I != E; ++I) 64016118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassProperties.push_back(*I); 64026118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 64036118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_prop_list_t_initializer(*this, Context, Result, ClassProperties, 64046118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian /* Container */0, 64056118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian "_OBJC_$_PROP_LIST_", 64066118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian FullCategoryName); 64076118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 64086118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian Write_category_t(*this, Context, Result, 64096118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian CDecl->getNameAsString(), 641088f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian ClassDecl, 64116118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian InstanceMethods, 64126118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassMethods, 64136118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian RefedProtocols, 64146118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian ClassProperties); 64156118612837887cbae365676d53cc3e1c258249aaFariborz Jahanian 641688f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian // Determine if this category is also "non-lazy". 641788f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian if (ImplementationIsNonLazy(IDecl)) 641888f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian DefinedNonLazyCategories.push_back(CDecl); 641988f7f75e35425f3f228473c4c84a2445793a39c1Fariborz Jahanian 642064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 642164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 642264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// RewriteObjCMethodsMetaData - Rewrite methods metadata for instance or 642364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// class methods. 642464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahaniantemplate<typename MethodIterator> 642564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCMethodsMetaData(MethodIterator MethodBegin, 642664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodIterator MethodEnd, 642764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsInstanceMethod, 642864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, 642964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef ClassName, 643064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 643164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MethodBegin == MethodEnd) return; 643264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 643364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!objc_impl_method) { 643464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_method { 643564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SEL _cmd; 643664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char *method_types; 643764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void *_imp; 643864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 643964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 644064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_method {\n"; 644164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tSEL _cmd;\n"; 644264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tchar *method_types;\n"; 644364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tvoid *_imp;\n"; 644464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 644564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 644664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_impl_method = true; 644764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 644864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 644964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's methods if needed 645064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 645164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct { 645264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method_list *next_method; 645364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int method_count; 645464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method method_list[]; 645564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 645664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 645764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned NumMethods = std::distance(MethodBegin, MethodEnd); 64581ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "\n"; 64591ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (LangOpts.MicrosoftExt) { 64601ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian if (IsInstanceMethod) 64611ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".inst_meth$B\")) "; 64621ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian else 64631ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "__declspec(allocate(\".cls_meth$B\")) "; 64641ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian } 64651ca052c09868169d4de4da5d1d45f07f2fe8511aFariborz Jahanian Result += "static struct {\n"; 646664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method_list *next_method;\n"; 646764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tint method_count;\n"; 646864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method method_list["; 646964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(NumMethods); 647064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "];\n} _OBJC_"; 647164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += prefix; 647264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IsInstanceMethod ? "INSTANCE" : "CLASS"; 647364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_METHODS_"; 647464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassName; 647564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __"; 647664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IsInstanceMethod ? "inst" : "cls"; 647764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_meth\")))= "; 647864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "{\n\t0, " + utostr(NumMethods) + "\n"; 647964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 648064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,{{(SEL)\""; 648164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*MethodBegin)->getSelector().getAsString().c_str(); 648264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string MethodTypeString; 648364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString); 648464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 648564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodTypeString; 648664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", (void *)"; 648764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodInternalNames[*MethodBegin]; 648864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 648964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (++MethodBegin; MethodBegin != MethodEnd; ++MethodBegin) { 649064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t ,{(SEL)\""; 649164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*MethodBegin)->getSelector().getAsString().c_str(); 649264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string MethodTypeString; 649364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString); 649464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 649564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodTypeString; 649664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", (void *)"; 649764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodInternalNames[*MethodBegin]; 649864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 649964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 650064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t }\n};\n"; 650164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 650264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 650364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV) { 650464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = IV->getSourceRange(); 650564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *BaseExpr = IV->getBase(); 650664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 650764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the base, but without actually doing replaces. 650864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 650964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 651064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BaseExpr = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(BaseExpr)); 651164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->setBase(BaseExpr); 651264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 651364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 651464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCIvarDecl *D = IV->getDecl(); 651564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 651664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Replacement = IV; 6517e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian 651864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BaseExpr->getType()->isObjCObjectPointerType()) { 651964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceType *iFaceDecl = 652064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCInterfaceType>(BaseExpr->getType()->getPointeeType()); 652164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(iFaceDecl && "RewriteObjCIvarRefExpr - iFaceDecl is null"); 652264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // lookup which class implements the instance variable. 652364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *clsDeclared = 0; 652464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian iFaceDecl->getDecl()->lookupInstanceVariable(D->getIdentifier(), 652564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian clsDeclared); 652664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(clsDeclared && "RewriteObjCIvarRefExpr(): Can't find class"); 652764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6528e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian // Build name of symbol holding ivar offset. 65297cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian std::string IvarOffsetName; 65307cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian WriteInternalIvarName(clsDeclared, D, IvarOffsetName); 65317cb2a1b4b6ef60aa57a5f2e5e820a57adef3ea77Fariborz Jahanian 653272c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian ReferencedIvars[clsDeclared].insert(D); 653372c88f13f79e2012b8a5f84cf48c6af779dfd1baFariborz Jahanian 6534e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian // cast offset to "char *". 6535e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, 6536e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Context->getPointerType(Context->CharTy), 653764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 6538e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian BaseExpr); 6539e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 6540e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation(), &Context->Idents.get(IvarOffsetName), 6541e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Context->UnsignedLongTy, 0, SC_Extern, SC_None); 6542f4b88a45902af1802a1cb42ba48b1c474474f228John McCall DeclRefExpr *DRE = new (Context) DeclRefExpr(NewVD, false, 6543f4b88a45902af1802a1cb42ba48b1c474474f228John McCall Context->UnsignedLongTy, VK_LValue, 6544e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation()); 6545e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian BinaryOperator *addExpr = 6546e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian new (Context) BinaryOperator(castExpr, DRE, BO_Add, 6547e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Context->getPointerType(Context->CharTy), 6548e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 654964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 6550e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), 6551e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation(), 6552e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian addExpr); 65530d6e22a38314fd6a9bdb0d83fcc5c5215641d5e9Fariborz Jahanian QualType IvarT = D->getType(); 65548e0913d4921e36ef82ecba565fd7b2c66165dd87Fariborz Jahanian convertObjCTypeToCStyleType(IvarT); 65550d6e22a38314fd6a9bdb0d83fcc5c5215641d5e9Fariborz Jahanian QualType castT = Context->getPointerType(IvarT); 655664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6557e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian castExpr = NoTypeInfoCStyleCastExpr(Context, 6558e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian castT, 6559e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian CK_BitCast, 6560e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian PE); 65618e0913d4921e36ef82ecba565fd7b2c66165dd87Fariborz Jahanian Expr *Exp = new (Context) UnaryOperator(castExpr, UO_Deref, IvarT, 6562e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian VK_LValue, OK_Ordinary, 6563e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian SourceLocation()); 6564e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian PE = new (Context) ParenExpr(OldRange.getBegin(), 6565e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian OldRange.getEnd(), 6566e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Exp); 6567e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian 6568e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian Replacement = PE; 656964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 657064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6571e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian ReplaceStmtWithRange(IV, Replacement, OldRange); 6572e7b3fa73596305bcc443369569d123b0f1b513cbFariborz Jahanian return Replacement; 657364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 6574