RewriteModernObjC.cpp revision a63b422eda86b9ff5a08395dcfc288a31419038e
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; 11064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCInterfaceDecl*, 8> ObjCForwardDecls; 11164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<ObjCMethodDecl*, std::string> MethodInternalNames; 11264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Stmt *, 32> Stmts; 11364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<int, 8> ObjCBcLabelNo; 11464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remember all the @protocol(<expr>) expressions. 11564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ObjCProtocolDecl *, 32> ProtocolExprDecls; 11664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 11764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseSet<uint64_t> CopyDestroyCache; 11864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 11964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block expressions. 12064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockExpr *, 32> Blocks; 12164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<int, 32> InnerDeclRefsCount; 12264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockDeclRefExpr *, 32> InnerDeclRefs; 12364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 12464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockDeclRefExpr *, 32> BlockDeclRefs; 12564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 12664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block related declarations. 12764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ValueDecl *, 8> BlockByCopyDecls; 12864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> BlockByCopyDeclsPtrSet; 12964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ValueDecl *, 8> BlockByRefDecls; 13064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> BlockByRefDeclsPtrSet; 13164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<ValueDecl *, unsigned> BlockByRefDeclNo; 13264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<ValueDecl *, 8> ImportedBlockDecls; 13364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<VarDecl *, 8> ImportedLocalExternalDecls; 13464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 13564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<BlockExpr *, std::string> RewrittenBlockExprs; 13664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 13764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This maps an original source AST to it's rewritten form. This allows 13864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // us to avoid rewriting the same node twice (which is very uncommon). 13964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is needed to support some of the exotic property rewriting. 14064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::DenseMap<Stmt *, Stmt *> ReplacedNodes; 14164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 14264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Needed for header files being rewritten 14364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsHeader; 14464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SilenceRewriteMacroWarning; 14564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool objc_impl_method; 14664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 14764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool DisableReplaceStmt; 14864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian class DisableReplaceStmtScope { 14964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteModernObjC &R; 15064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SavedValue; 15164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 15264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian public: 15364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope(RewriteModernObjC &R) 15464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : R(R), SavedValue(R.DisableReplaceStmt) { 15564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian R.DisableReplaceStmt = true; 15664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 15764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ~DisableReplaceStmtScope() { 15864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian R.DisableReplaceStmt = SavedValue; 15964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 16064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 16164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InitializeCommon(ASTContext &context); 16264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 16364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian public: 16464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 16564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Top Level Driver code. 16664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual bool HandleTopLevelDecl(DeclGroupRef D) { 16764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclGroupRef::iterator I = D.begin(), E = D.end(); I != E; ++I) { 16864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(*I)) { 16964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Class->isThisDeclarationADefinition()) { 17064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassDecl(D); 17164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 17264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 17364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 17464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 17564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>(*I)) { 17664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Proto->isThisDeclarationADefinition()) { 17764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardProtocolDecl(D); 17864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 17964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 18264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HandleTopLevelSingleDecl(*I); 18364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 18564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 18664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void HandleTopLevelSingleDecl(Decl *D); 18764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void HandleDeclInMainFile(Decl *D); 18864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteModernObjC(std::string inFile, raw_ostream *OS, 18964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &D, const LangOptions &LOpts, 19064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool silenceMacroWarn); 19164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 19264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ~RewriteModernObjC() {} 19364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 19464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void HandleTranslationUnit(ASTContext &C); 19564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 19664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceStmt(Stmt *Old, Stmt *New) { 19764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = ReplacedNodes[Old]; 19864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 19964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ReplacingStmt) 20064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // We can't rewrite the same node twice. 20164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 20264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DisableReplaceStmt) 20364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 20464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 20564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If replacement succeeded or warning disabled return with no warning. 20664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceStmt(Old, New)) { 20764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacedNodes[Old] = New; 20864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 20964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 21064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SilenceRewriteMacroWarning) 21164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 21264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 21364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 21464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 21564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 21664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceStmtWithRange(Stmt *Old, Stmt *New, SourceRange SrcRange) { 21764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DisableReplaceStmt) 21864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 21964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 22064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Measure the old text. 22164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int Size = Rewrite.getRangeSize(SrcRange); 22264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Size == -1) { 22364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 22464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 22564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 22664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 22764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the new text. 22864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SStr; 22964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream S(SStr); 23064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian New->printPretty(S, *Context, 0, PrintingPolicy(LangOpts)); 23164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Str = S.str(); 23264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 23364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If replacement succeeded or warning disabled return with no warning. 23464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceText(SrcRange.getBegin(), Size, Str)) { 23564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacedNodes[Old] = New; 23664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 23764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 23864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SilenceRewriteMacroWarning) 23964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 24064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag) 24164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian << Old->getSourceRange(); 24264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 24364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 24464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertText(SourceLocation Loc, StringRef Str, 24564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool InsertAfter = true) { 24664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If insertion succeeded or warning disabled return with no warning. 24764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.InsertText(Loc, Str, InsertAfter) || 24864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning) 24964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 25064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 25164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag); 25264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 25364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 25464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void ReplaceText(SourceLocation Start, unsigned OrigLength, 25564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef Str) { 25664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If removal succeeded or warning disabled return with no warning. 25764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewrite.ReplaceText(Start, OrigLength, Str) || 25864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning) 25964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 26064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 26164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(Start), RewriteFailedDiag); 26264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 26364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 26464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Syntactic Rewriting. 26564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteRecordBody(RecordDecl *RD); 26664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteInclude(); 26764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassDecl(DeclGroupRef D); 26864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassDecl(const llvm::SmallVector<Decl*, 8> &DG); 26964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardClassEpilogue(ObjCInterfaceDecl *ClassDecl, 27064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &typedefString); 27164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteImplementations(); 27264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewritePropertyImplDecl(ObjCPropertyImplDecl *PID, 27364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD, 27464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID); 27564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteInterfaceDecl(ObjCInterfaceDecl *Dcl); 27664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteImplementationDecl(Decl *Dcl); 27764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCMethodDecl(const ObjCInterfaceDecl *IDecl, 27864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MDecl, std::string &ResultStr); 27964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteTypeIntoString(QualType T, std::string &ResultStr, 28064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *&FPRetType); 28164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteByRefString(std::string &ResultStr, const std::string &Name, 28264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD, bool def=false); 28364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteCategoryDecl(ObjCCategoryDecl *Dcl); 28464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteProtocolDecl(ObjCProtocolDecl *Dcl); 28564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardProtocolDecl(DeclGroupRef D); 28664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteForwardProtocolDecl(const llvm::SmallVector<Decl*, 8> &DG); 28764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteMethodDeclaration(ObjCMethodDecl *Method); 28864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteProperty(ObjCPropertyDecl *prop); 28964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteFunctionDecl(FunctionDecl *FD); 29064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerType(std::string& Str, QualType Type); 29164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerTypeVariable(std::string& Str, ValueDecl *VD); 29264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockLiteralFunctionDecl(FunctionDecl *FD); 29364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCQualifiedInterfaceTypes(Decl *Dcl); 29464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteTypeOfDecl(VarDecl *VD); 29564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCQualifiedInterfaceTypes(Expr *E); 29664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 29764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Expression Rewriting. 29864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteFunctionBodyOrGlobalInitializer(Stmt *S); 29964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteAtEncode(ObjCEncodeExpr *Exp); 30064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewritePropertyOrImplicitGetter(PseudoObjectExpr *Pseudo); 30164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewritePropertyOrImplicitSetter(PseudoObjectExpr *Pseudo); 30264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteAtSelector(ObjCSelectorExpr *Exp); 30364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteMessageExpr(ObjCMessageExpr *Exp); 30464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCStringLiteral(ObjCStringLiteral *Exp); 30564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp); 30664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteTryReturnStmts(Stmt *S); 30764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteSyncReturnStmts(Stmt *S, std::string buf); 30864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCTryStmt(ObjCAtTryStmt *S); 30964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S); 31064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCThrowStmt(ObjCAtThrowStmt *S); 31164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S, 31264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OrigEnd); 31364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteBreakStmt(BreakStmt *S); 31464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteContinueStmt(ContinueStmt *S); 31564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteCastExpr(CStyleCastExpr *CE); 31664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 31764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block rewriting. 31864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlocksInFunctionProtoType(QualType funcType, NamedDecl *D); 31964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 32064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Block specific rewrite rules. 32164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerDecl(NamedDecl *VD); 32264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteByRefVar(VarDecl *VD); 32364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteBlockDeclRefExpr(Expr *VD); 32464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RewriteLocalVariableExternalStorage(DeclRefExpr *DRE); 32564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteBlockPointerFunctionArgs(FunctionDecl *FD); 32664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 32764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCInternalStruct(ObjCInterfaceDecl *CDecl, 32864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 32964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 33064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void Initialize(ASTContext &context); 33164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 33264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Misc. AST transformation routines. Somtimes they end up calling 33364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // rewriting routines on the new ASTs. 33464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SynthesizeCallToFunctionDecl(FunctionDecl *FD, 33564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr **args, unsigned nargs, 33664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc=SourceLocation(), 33764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc=SourceLocation()); 33864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 33964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthMessageExpr(ObjCMessageExpr *Exp, 34064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc=SourceLocation(), 34164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc=SourceLocation()); 34264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 34364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthCountByEnumWithState(std::string &buf); 34464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendFunctionDecl(); 34564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendSuperFunctionDecl(); 34664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendStretFunctionDecl(); 34764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendFpretFunctionDecl(); 34864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthMsgSendSuperStretFunctionDecl(); 34964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetClassFunctionDecl(); 35064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetMetaClassFunctionDecl(); 35164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthGetSuperClassFunctionDecl(); 35264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthSelGetUidFunctionDecl(); 35364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthSuperContructorFunctionDecl(); 35464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 35564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewriting metadata 35664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian template<typename MethodIterator> 35764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void RewriteObjCMethodsMetaData(MethodIterator MethodBegin, 35864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodIterator MethodEnd, 35964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsInstanceMethod, 36064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, 36164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef ClassName, 36264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 363da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian void RewriteObjCProtocolMetaData(ObjCProtocolDecl *Protocol, 364da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::string &Result); 36564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCProtocolListMetaData( 36664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Prots, 36764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, StringRef ClassName, std::string &Result); 36864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl, 36964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 37064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteMetaDataIntoBuffer(std::string &Result); 37164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *CDecl, 37264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 37364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 37464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewriting ivar 37564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual void RewriteIvarOffsetComputation(ObjCIvarDecl *ivar, 37664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result); 37764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian virtual Stmt *RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV); 37864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 37964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 38064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeByrefCopyDestroyHelper(VarDecl *VD, int flag); 38164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockHelperFuncs(BlockExpr *CE, int i, 38264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, std::string Tag); 38364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockFunc(BlockExpr *CE, int i, 38464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, std::string Tag); 38564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockImpl(BlockExpr *CE, 38664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag, std::string Desc); 38764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SynthesizeBlockDescriptor(std::string DescTag, 38864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag, 38964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int i, StringRef funcName, 39064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned hasCopy); 39164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthesizeBlockCall(CallExpr *Exp, const Expr* BlockExp); 39264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void SynthesizeBlockLiterals(SourceLocation FunLocStart, 39364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName); 39464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *SynthBlockInitFunctionDecl(StringRef name); 39564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SynthBlockInitExpr(BlockExpr *Exp, 39664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const SmallVector<BlockDeclRefExpr *, 8> &InnerBlockDeclRefs); 39764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 39864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Misc. helper routines. 39964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getProtocolType(); 40064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void WarnAboutReturnGotoStmts(Stmt *S); 40164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void HasReturnStmts(Stmt *S, bool &hasReturns); 40264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void CheckFunctionPointerDecl(QualType dType, NamedDecl *ND); 40364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertBlockLiteralsWithinFunction(FunctionDecl *FD); 40464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void InsertBlockLiteralsWithinMethod(ObjCMethodDecl *MD); 40564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 40664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsDeclStmtInForeachHeader(DeclStmt *DS); 40764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void CollectBlockDeclRefInfo(BlockExpr *Exp); 40864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetBlockDeclRefExprs(Stmt *S); 40964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetInnerBlockDeclRefExprs(Stmt *S, 41064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockDeclRefExpr *, 8> &InnerBlockDeclRefs, 41164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> &InnerContexts); 41264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 41364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We avoid calling Type::isBlockPointerType(), since it operates on the 41464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // canonical type. We only care if the top-level type is a closure pointer. 41564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isTopLevelBlockPointerType(QualType T) { 41664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return isa<BlockPointerType>(T); 41764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 41864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 41964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// convertBlockPointerToFunctionPointer - Converts a block-pointer type 42064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// to a function pointer type and upon success, returns true; false 42164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// otherwise. 42264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool convertBlockPointerToFunctionPointer(QualType &T) { 42364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(T)) { 42464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = T->getAs<BlockPointerType>(); 42564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getPointerType(BPT->getPointeeType()); 42664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 42764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 42864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 42964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 43064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 43164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool needToScanForQualifiers(QualType T); 43264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getSuperStructType(); 43364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getConstantStringStructType(); 43464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType convertFunctionTypeOfBlocks(const FunctionType *FT); 43564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool BufferContainsPPDirectives(const char *startBuf, const char *endBuf); 43664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 43764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void convertToUnqualifiedObjCType(QualType &T) { 43864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 43964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getObjCIdType(); 44064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isObjCQualifiedClassType()) 44164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getObjCClassType(); 44264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isObjCObjectPointerType() && 44364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->getPointeeType()->isObjCQualifiedInterfaceType()) { 44464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const ObjCObjectPointerType * OBJPT = 44564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->getAsObjCInterfacePointerType()) { 44664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceType *IFaceT = OBJPT->getInterfaceType(); 44764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = QualType(IFaceT, 0); 44864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = Context->getPointerType(T); 44964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 45064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 45164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 45264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 45364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: This predicate seems like it would be useful to add to ASTContext. 45464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isObjCType(QualType T) { 45564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!LangOpts.ObjC1 && !LangOpts.ObjC2) 45664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 45764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 45864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType OCT = Context->getCanonicalType(T).getUnqualifiedType(); 45964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 46064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OCT == Context->getCanonicalType(Context->getObjCIdType()) || 46164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OCT == Context->getCanonicalType(Context->getObjCClassType())) 46264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 46364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 46464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType *PT = OCT->getAs<PointerType>()) { 46564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCInterfaceType>(PT->getPointeeType()) || 46664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PT->getPointeeType()->isObjCQualifiedIdType()) 46764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 46864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 46964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 47064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 47164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool PointerTypeTakesAnyBlockArguments(QualType QT); 47264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool PointerTypeTakesAnyObjCQualifiedType(QualType QT); 47364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void GetExtentOfArgList(const char *Name, const char *&LParen, 47464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&RParen); 47564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 47664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void QuoteDoublequotes(std::string &From, std::string &To) { 47764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < From.length(); i++) { 47864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (From[i] == '"') 47964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian To += "\\\""; 48064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 48164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian To += From[i]; 48264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 48364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 48464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 48564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getSimpleFunctionType(QualType result, 48664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const QualType *args, 48764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned numArgs, 48864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool variadic = false) { 48964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (result == Context->getObjCInstanceType()) 49064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian result = Context->getObjCIdType(); 49164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionProtoType::ExtProtoInfo fpi; 49264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian fpi.Variadic = variadic; 49364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getFunctionType(result, args, numArgs, fpi); 49464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 49564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 49664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Helper function: create a CStyleCastExpr with trivial type source info. 49764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CStyleCastExpr* NoTypeInfoCStyleCastExpr(ASTContext *Ctx, QualType Ty, 49864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind Kind, Expr *E) { 49964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *TInfo = Ctx->getTrivialTypeSourceInfo(Ty, SourceLocation()); 50064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CStyleCastExpr::Create(*Ctx, Ty, VK_RValue, Kind, E, 0, TInfo, 50164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation()); 50264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 50364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 50464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 50564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 50664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 50764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlocksInFunctionProtoType(QualType funcType, 50864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NamedDecl *D) { 50964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *fproto 51064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = dyn_cast<FunctionProtoType>(funcType.IgnoreParens())) { 51164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = fproto->arg_type_begin(), 51264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = fproto->arg_type_end(); I && (I != E); ++I) 51364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(*I)) { 51464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // All the args are checked/rewritten. Don't call twice! 51564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(D); 51664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 51764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 51864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 51964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 52064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 52164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::CheckFunctionPointerDecl(QualType funcType, NamedDecl *ND) { 52264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = funcType->getAs<PointerType>(); 52364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT && PointerTypeTakesAnyBlockArguments(funcType)) 52464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlocksInFunctionProtoType(PT->getPointeeType(), ND); 52564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 52664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 52764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool IsHeaderFile(const std::string &Filename) { 52864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string::size_type DotPos = Filename.rfind('.'); 52964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 53064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DotPos == std::string::npos) { 53164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // no file extension 53264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 53364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 53464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 53564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Ext = std::string(Filename.begin()+DotPos+1, Filename.end()); 53664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // C header: .h 53764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // C++ header: .hh or .H; 53864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Ext == "h" || Ext == "hh" || Ext == "H"; 53964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 54064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 54164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianRewriteModernObjC::RewriteModernObjC(std::string inFile, raw_ostream* OS, 54264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &D, const LangOptions &LOpts, 54364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool silenceMacroWarn) 54464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : Diags(D), LangOpts(LOpts), InFileName(inFile), OutFile(OS), 54564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SilenceRewriteMacroWarning(silenceMacroWarn) { 54664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IsHeader = IsHeaderFile(inFile); 54764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFailedDiag = Diags.getCustomDiagID(DiagnosticsEngine::Warning, 54864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "rewriting sub-expression within a macro (may not be correct)"); 54964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TryFinallyContainsReturnDiag = Diags.getCustomDiagID( 55064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine::Warning, 55164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "rewriter doesn't support user-specified control flow semantics " 55264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "for @try/@finally (code may not execute properly)"); 55364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 55464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 55564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianASTConsumer *clang::CreateModernObjCRewriter(const std::string& InFile, 55664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian raw_ostream* OS, 55764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DiagnosticsEngine &Diags, 55864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const LangOptions &LOpts, 55964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool SilenceRewriteMacroWarning) { 56064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return new RewriteModernObjC(InFile, OS, Diags, LOpts, SilenceRewriteMacroWarning); 56164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 56264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 56364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InitializeCommon(ASTContext &context) { 56464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context = &context; 56564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SM = &Context->getSourceManager(); 56664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TUDecl = Context->getTranslationUnitDecl(); 56764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFunctionDecl = 0; 56864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperFunctionDecl = 0; 56964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFunctionDecl = 0; 57064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperStretFunctionDecl = 0; 57164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFpretFunctionDecl = 0; 57264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetClassFunctionDecl = 0; 57364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetMetaClassFunctionDecl = 0; 57464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetSuperClassFunctionDecl = 0; 57564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = 0; 57664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CFStringFunctionDecl = 0; 57764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringClassReference = 0; 57864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NSStringRecord = 0; 57964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = 0; 58064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = 0; 58164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = 0; 58264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = 0; 58364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl = 0; 58464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolTypeDecl = 0; 58564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl = 0; 58664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BcLabelCount = 0; 58764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperContructorFunctionDecl = 0; 58864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NumObjCStringLiterals = 0; 58964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 59064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 59164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = false; 59264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_impl_method = false; 59364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 59464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the ID and start/end of the main file. 59564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileID = SM->getMainFileID(); 59664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const llvm::MemoryBuffer *MainBuf = SM->getBuffer(MainFileID); 59764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileStart = MainBuf->getBufferStart(); 59864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MainFileEnd = MainBuf->getBufferEnd(); 59964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 60064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Rewrite.setSourceMgr(Context->getSourceManager(), Context->getLangOptions()); 60164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 60264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 60364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 60464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Top Level Driver Code 60564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 60664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 60764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleTopLevelSingleDecl(Decl *D) { 60864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Diags.hasErrorOccurred()) 60964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 61064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 61164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Two cases: either the decl could be in the main file, or it could be in a 61264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // #included file. If the former, rewrite it now. If the later, check to see 61364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // if we rewrote the #include/#import. 61464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc = D->getLocation(); 61564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = SM->getExpansionLoc(Loc); 61664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 61764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If this is for a builtin, ignore it. 61864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Loc.isInvalid()) return; 61964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 62064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Look for built-in declarations that we need to refer during the rewrite. 62164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { 62264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionDecl(FD); 62364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (VarDecl *FVD = dyn_cast<VarDecl>(D)) { 62464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declared in <Foundation/NSString.h> 62564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FVD->getName() == "_NSConstantStringClassReference") { 62664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringClassReference = FVD; 62764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 62864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 62964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(D)) { 63064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ID->isThisDeclarationADefinition()) 63164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteInterfaceDecl(ID); 63264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(D)) { 63364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCategoryDecl(CD); 63464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCProtocolDecl *PD = dyn_cast<ObjCProtocolDecl>(D)) { 63564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isThisDeclarationADefinition()) 63664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProtocolDecl(PD); 63764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (LinkageSpecDecl *LSD = dyn_cast<LinkageSpecDecl>(D)) { 63864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Recurse into linkage specifications 63964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclContext::decl_iterator DI = LSD->decls_begin(), 64064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DIEnd = LSD->decls_end(); 64164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DI != DIEnd; ) { 64264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>((*DI))) { 64364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!IFace->isThisDeclarationADefinition()) { 64464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Decl *, 8> DG; 64564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc = IFace->getLocStart(); 64664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian do { 64764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCInterfaceDecl>(*DI) && 64864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !cast<ObjCInterfaceDecl>(*DI)->isThisDeclarationADefinition() && 64964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc == (*DI)->getLocStart()) 65064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DG.push_back(*DI); 65164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 65264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 65364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 65464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 65564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } while (DI != DIEnd); 65664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassDecl(DG); 65764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 65864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 65964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 66064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 66164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Proto = dyn_cast<ObjCProtocolDecl>((*DI))) { 66264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Proto->isThisDeclarationADefinition()) { 66364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Decl *, 8> DG; 66464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc = Proto->getLocStart(); 66564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian do { 66664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ObjCProtocolDecl>(*DI) && 66764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !cast<ObjCProtocolDecl>(*DI)->isThisDeclarationADefinition() && 66864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc == (*DI)->getLocStart()) 66964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DG.push_back(*DI); 67064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 67164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 67264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 67364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 67464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } while (DI != DIEnd); 67564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardProtocolDecl(DG); 67664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 67764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 67864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 67964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 68064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HandleTopLevelSingleDecl(*DI); 68164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ++DI; 68264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 68364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 68464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we have a decl in the main file, see if we should rewrite it. 68564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SM->isFromMainFile(Loc)) 68664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return HandleDeclInMainFile(D); 68764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 68864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 68964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 69064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Syntactic (non-AST) Rewriting Code 69164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 69264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 69364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteInclude() { 69464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = SM->getLocForStartOfFile(MainFileID); 69564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef MainBuf = SM->getBufferData(MainFileID); 69664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *MainBufStart = MainBuf.begin(); 69764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *MainBufEnd = MainBuf.end(); 69864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian size_t ImportLen = strlen("import"); 69964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 70064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Loop over the whole file, looking for includes. 70164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (const char *BufPtr = MainBufStart; BufPtr < MainBufEnd; ++BufPtr) { 70264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*BufPtr == '#') { 70364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (++BufPtr == MainBufEnd) 70464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 70564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*BufPtr == ' ' || *BufPtr == '\t') 70664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (++BufPtr == MainBufEnd) 70764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 70864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strncmp(BufPtr, "import", ImportLen)) { 70964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace import with include 71064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation ImportLoc = 71164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LocStart.getLocWithOffset(BufPtr-MainBufStart); 71264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(ImportLoc, ImportLen, "include"); 71364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BufPtr += ImportLen; 71464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 71564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 71664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 71764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 71864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 71964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic std::string getIvarAccessString(ObjCIvarDecl *OID) { 72064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceDecl *ClassDecl = OID->getContainingInterface(); 72164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S; 72264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S = "((struct "; 72364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ClassDecl->getIdentifier()->getName(); 72464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_IMPL *)self)->"; 72564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += OID->getName(); 72664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 72764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 72864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 72964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewritePropertyImplDecl(ObjCPropertyImplDecl *PID, 73064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD, 73164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID) { 73264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objcGetPropertyDefined = false; 73364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objcSetPropertyDefined = false; 73464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = PID->getLocStart(); 73564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(startLoc, "// "); 73664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 73764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @synthesize location"); 73864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 73964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "@synthesize: can't find ';'"); 74064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation onePastSemiLoc = 74164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc.getLocWithOffset(semiBuf-startBuf+1); 74264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 74364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PID->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 74464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // FIXME: is this correct? 74564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 74664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the 'getter' function. 74764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl *PD = PID->getPropertyDecl(); 74864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCIvarDecl *OID = PID->getPropertyIvarDecl(); 74964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 75064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!OID) 75164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 75264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned Attributes = PD->getPropertyAttributes(); 75364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!PD->getGetterMethodDecl()->isDefined()) { 75464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool GenGetProperty = !(Attributes & ObjCPropertyDecl::OBJC_PR_nonatomic) && 75564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (Attributes & (ObjCPropertyDecl::OBJC_PR_retain | 75664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl::OBJC_PR_copy)); 75764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Getr; 75864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenGetProperty && !objcGetPropertyDefined) { 75964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objcGetPropertyDefined = true; 76064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Is this attribute correct in all cases? 76164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr = "\nextern \"C\" __declspec(dllimport) " 76264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "id objc_getProperty(id, SEL, long, bool);\n"; 76364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 76464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OID->getContainingInterface(), 76564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PD->getGetterMethodDecl(), Getr); 76664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "{ "; 76764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to gain access to the ivar. 76864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See objc-act.c:objc_synthesize_new_getter() for details. 76964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenGetProperty) { 77064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // return objc_getProperty(self, _cmd, offsetof(ClassDecl, OID), 1) 77164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "typedef "; 77264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FPRetType = 0; 77364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeIntoString(PD->getGetterMethodDecl()->getResultType(), Getr, 77464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FPRetType); 77564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += " _TYPE"; 77664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FPRetType) { 77764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ")"; // close the precedence "scope" for "*". 77864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 77964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, emit the argument types (if any). 78064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FPRetType)){ 78164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "("; 78264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) { 78364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i) Getr += ", "; 78464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr = FT->getArgType(i).getAsString( 78564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 78664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ParamStr; 78764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 78864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 78964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->getNumArgs()) Getr += ", "; 79064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "..."; 79164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 79264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ")"; 79364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 79464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "()"; 79564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 79664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ";\n"; 79764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "return (_TYPE)"; 79864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "objc_getProperty(self, _cmd, "; 79964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation(OID, Getr); 80064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += ", 1)"; 80164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 80264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 80364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "return " + getIvarAccessString(OID); 80464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Getr += "; }"; 80564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, Getr); 80664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 80764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 80864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isReadOnly() || PD->getSetterMethodDecl()->isDefined()) 80964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 81064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 81164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the 'setter' function. 81264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Setr; 81364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool GenSetProperty = Attributes & (ObjCPropertyDecl::OBJC_PR_retain | 81464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl::OBJC_PR_copy); 81564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenSetProperty && !objcSetPropertyDefined) { 81664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objcSetPropertyDefined = true; 81764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Is this attribute correct in all cases? 81864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr = "\nextern \"C\" __declspec(dllimport) " 81964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "void objc_setProperty (id, SEL, long, id, bool, bool);\n"; 82064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 82164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 82264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OID->getContainingInterface(), 82364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PD->getSetterMethodDecl(), Setr); 82464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "{ "; 82564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to initialize the ivar. 82664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See objc-act.c:objc_synthesize_new_setter() for details. 82764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GenSetProperty) { 82864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "objc_setProperty (self, _cmd, "; 82964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation(OID, Setr); 83064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += ", (id)"; 83164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += PD->getName(); 83264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += ", "; 83364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Attributes & ObjCPropertyDecl::OBJC_PR_nonatomic) 83464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "0, "; 83564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 83664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "1, "; 83764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Attributes & ObjCPropertyDecl::OBJC_PR_copy) 83864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "1)"; 83964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 84064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "0)"; 84164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 84264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 84364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += getIvarAccessString(OID) + " = "; 84464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += PD->getName(); 84564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 84664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Setr += "; }"; 84764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, Setr); 84864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 84964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 85064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void RewriteOneForwardClassDecl(ObjCInterfaceDecl *ForwardDecl, 85164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &typedefString) { 85264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "#ifndef _REWRITER_typedef_"; 85364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 85464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "\n"; 85564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "#define _REWRITER_typedef_"; 85664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 85764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "\n"; 85864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "typedef struct objc_object "; 85964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 86064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n#endif\n"; 86164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 86264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 86364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassEpilogue(ObjCInterfaceDecl *ClassDecl, 86464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &typedefString) { 86564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = ClassDecl->getLocStart(); 86664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 86764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiPtr = strchr(startBuf, ';'); 86864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the @class with typedefs corresponding to the classes. 86964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, semiPtr-startBuf+1, typedefString); 87064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 87164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 87264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassDecl(DeclGroupRef D) { 87364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string typedefString; 87464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclGroupRef::iterator I = D.begin(), E = D.end(); I != E; ++I) { 87564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ForwardDecl = cast<ObjCInterfaceDecl>(*I); 87664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (I == D.begin()) { 87764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Translate to typedef's that forward reference structs with the same name 87864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as the class. As a convenience, we include the original declaration 87964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as a comment. 88064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "// @class "; 88164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 88264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n"; 88364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 88464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteOneForwardClassDecl(ForwardDecl, typedefString); 88564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 88664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclGroupRef::iterator I = D.begin(); 88764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassEpilogue(cast<ObjCInterfaceDecl>(*I), typedefString); 88864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 88964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 89064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardClassDecl( 89164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const llvm::SmallVector<Decl*, 8> &D) { 89264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string typedefString; 89364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < D.size(); i++) { 89464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ForwardDecl = cast<ObjCInterfaceDecl>(D[i]); 89564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i == 0) { 89664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += "// @class "; 89764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ForwardDecl->getNameAsString(); 89864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian typedefString += ";\n"; 89964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 90064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteOneForwardClassDecl(ForwardDecl, typedefString); 90164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 90264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteForwardClassEpilogue(cast<ObjCInterfaceDecl>(D[0]), typedefString); 90364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 90464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 90564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteMethodDeclaration(ObjCMethodDecl *Method) { 90664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // When method is a synthesized one, such as a getter/setter there is 90764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // nothing to rewrite. 90864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Method->isImplicit()) 90964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 91064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = Method->getLocStart(); 91164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = Method->getLocEnd(); 91264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 91364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SM->getExpansionLineNumber(LocEnd) > 91464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SM->getExpansionLineNumber(LocStart)) { 91564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LocStart, "#if 0\n"); 91664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocEnd, 1, ";\n#endif\n"); 91764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 91864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LocStart, "// "); 91964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 92064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 92164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 92264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteProperty(ObjCPropertyDecl *prop) { 92364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc = prop->getAtLoc(); 92464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 92564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(Loc, 0, "// "); 92664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle properties that are declared across multiple lines. 92764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 92864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 92964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteCategoryDecl(ObjCCategoryDecl *CatDecl) { 93064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CatDecl->getLocStart(); 93164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 93264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle category headers that are declared across multiple lines. 93364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 93464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 93564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::prop_iterator I = CatDecl->prop_begin(), 93664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = CatDecl->prop_end(); I != E; ++I) 93764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProperty(*I); 93864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 93964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::instmeth_iterator 94064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = CatDecl->instmeth_begin(), E = CatDecl->instmeth_end(); 94164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 94264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 94364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryDecl::classmeth_iterator 94464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = CatDecl->classmeth_begin(), E = CatDecl->classmeth_end(); 94564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 94664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 94764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 94864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Lastly, comment out the @end. 94964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(CatDecl->getAtEndRange().getBegin(), 95064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian strlen("@end"), "/* @end */"); 95164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 95264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 95364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) { 95464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = PDecl->getLocStart(); 95564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(PDecl->isThisDeclarationADefinition()); 95664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 95764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle protocol headers that are declared across multiple lines. 95864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 95964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 96064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCProtocolDecl::instmeth_iterator 96164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = PDecl->instmeth_begin(), E = PDecl->instmeth_end(); 96264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 96364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 96464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCProtocolDecl::classmeth_iterator 96564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = PDecl->classmeth_begin(), E = PDecl->classmeth_end(); 96664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 96764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 96864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 96964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCInterfaceDecl::prop_iterator I = PDecl->prop_begin(), 97064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = PDecl->prop_end(); I != E; ++I) 97164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProperty(*I); 97264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 97364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Lastly, comment out the @end. 97464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = PDecl->getAtEndRange().getBegin(); 97564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocEnd, strlen("@end"), "/* @end */"); 97664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 97764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Must comment out @optional/@required 97864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 97964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 98064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (const char *p = startBuf; p < endBuf; p++) { 98164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*p == '@' && !strncmp(p+1, "optional", strlen("optional"))) { 98264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OptionalLoc = LocStart.getLocWithOffset(p-startBuf); 98364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(OptionalLoc, strlen("@optional"), "/* @optional */"); 98464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 98564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 98664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*p == '@' && !strncmp(p+1, "required", strlen("required"))) { 98764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OptionalLoc = LocStart.getLocWithOffset(p-startBuf); 98864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(OptionalLoc, strlen("@required"), "/* @required */"); 98964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 99064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 99164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 99264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 99364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 99464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteForwardProtocolDecl(DeclGroupRef D) { 99564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = (*D.begin())->getLocStart(); 99664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 99764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("Invalid SourceLocation"); 99864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle forward protocol that are declared across multiple lines. 99964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 100064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 100164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 100264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid 100364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianRewriteModernObjC::RewriteForwardProtocolDecl(const llvm::SmallVector<Decl*, 8> &DG) { 100464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = DG[0]->getLocStart(); 100564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 100664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("Invalid SourceLocation"); 100764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: handle forward protocol that are declared across multiple lines. 100864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 0, "// "); 100964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 101064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 101164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteTypeIntoString(QualType T, std::string &ResultStr, 101264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *&FPRetType) { 101364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 101464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "id"; 101564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (T->isFunctionPointerType() || 101664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T->isBlockPointerType()) { 101764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // needs special handling, since pointer-to-functions have special 101864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // syntax (where a decaration models use). 101964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType retType = T; 102064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PointeeTy; 102164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType* PT = retType->getAs<PointerType>()) 102264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointeeTy = PT->getPointeeType(); 102364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const BlockPointerType *BPT = retType->getAs<BlockPointerType>()) 102464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointeeTy = BPT->getPointeeType(); 102564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((FPRetType = PointeeTy->getAs<FunctionType>())) { 102664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += FPRetType->getResultType().getAsString( 102764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 102864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "(*"; 102964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 103064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 103164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += T.getAsString(Context->getPrintingPolicy()); 103264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 103364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 103464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCMethodDecl(const ObjCInterfaceDecl *IDecl, 103564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD, 103664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &ResultStr) { 103764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //fprintf(stderr,"In RewriteObjCMethodDecl\n"); 103864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FPRetType = 0; 103964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "\nstatic "; 104064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeIntoString(OMD->getResultType(), ResultStr, FPRetType); 104164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " "; 104264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 104364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique method name 104464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string NameStr; 104564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 104664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isInstanceMethod()) 104764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_I_"; 104864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 104964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_C_"; 105064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 105164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += IDecl->getNameAsString(); 105264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_"; 105364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 105464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCCategoryImplDecl *CID = 105564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCCategoryImplDecl>(OMD->getDeclContext())) { 105664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += CID->getNameAsString(); 105764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += "_"; 105864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 105964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Append selector names, replacing ':' with '_' 106064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 106164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string selString = OMD->getSelector().getAsString(); 106264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int len = selString.size(); 106364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < len; i++) 106464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (selString[i] == ':') 106564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian selString[i] = '_'; 106664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NameStr += selString; 106764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 106864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remember this name for metadata emission 106964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodInternalNames[OMD] = NameStr; 107064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += NameStr; 107164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 107264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite arguments 107364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "("; 107464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 107564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // invisible arguments 107664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isInstanceMethod()) { 107764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType selfTy = Context->getObjCInterfaceType(IDecl); 107864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian selfTy = Context->getPointerType(selfTy); 107964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!LangOpts.MicrosoftExt) { 108064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSynthesizedStructs.count(const_cast<ObjCInterfaceDecl*>(IDecl))) 108164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "struct "; 108264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 108364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // When rewriting for Microsoft, explicitly omit the structure name. 108464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += IDecl->getNameAsString(); 108564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " *"; 108664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 108764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 108864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Context->getObjCClassType().getAsString( 108964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 109064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 109164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " self, "; 109264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Context->getObjCSelType().getAsString(Context->getPrintingPolicy()); 109364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += " _cmd"; 109464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 109564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Method arguments. 109664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(), 109764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = OMD->param_end(); PI != E; ++PI) { 109864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParmVarDecl *PDecl = *PI; 109964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ", "; 110064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PDecl->getType()->isObjCQualifiedIdType()) { 110164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "id "; 110264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += PDecl->getNameAsString(); 110364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 110464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = PDecl->getNameAsString(); 110564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = PDecl->getType(); 110664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 110764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (convertBlockPointerToFunctionPointer(QT)) 110864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(Name, Context->getPrintingPolicy()); 110964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 111064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PDecl->getType().getAsStringInternal(Name, Context->getPrintingPolicy()); 111164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += Name; 111264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 111364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 111464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OMD->isVariadic()) 111564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ", ..."; 111664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ") "; 111764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 111864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FPRetType) { 111964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ")"; // close the precedence "scope" for "*". 112064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 112164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, emit the argument types (if any). 112264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const FunctionProtoType *FT = dyn_cast<FunctionProtoType>(FPRetType)) { 112364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "("; 112464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, e = FT->getNumArgs(); i != e; ++i) { 112564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i) ResultStr += ", "; 112664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr = FT->getArgType(i).getAsString( 112764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPrintingPolicy()); 112864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ParamStr; 112964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 113064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 113164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->getNumArgs()) ResultStr += ", "; 113264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "..."; 113364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 113464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ")"; 113564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 113664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "()"; 113764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 113864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 113964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 114064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteImplementationDecl(Decl *OID) { 114164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *IMD = dyn_cast<ObjCImplementationDecl>(OID); 114264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CID = dyn_cast<ObjCCategoryImplDecl>(OID); 114364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 114464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(IMD ? IMD->getLocStart() : CID->getLocStart(), "// "); 114564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 114664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::instmeth_iterator 114764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->instmeth_begin() : CID->instmeth_begin(), 114864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->instmeth_end() : CID->instmeth_end(); 114964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 115064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 115164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD = *I; 115264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OMD->getClassInterface(), OMD, ResultStr); 115364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = OMD->getLocStart(); 115464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = OMD->getCompoundBody()->getLocStart(); 115564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 115664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 115764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 115864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, ResultStr); 115964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 116064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 116164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::classmeth_iterator 116264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->classmeth_begin() : CID->classmeth_begin(), 116364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->classmeth_end() : CID->classmeth_end(); 116464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 116564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 116664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *OMD = *I; 116764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodDecl(OMD->getClassInterface(), OMD, ResultStr); 116864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = OMD->getLocStart(); 116964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = OMD->getCompoundBody()->getLocStart(); 117064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 117164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 117264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 117364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, ResultStr); 117464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 117564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCCategoryImplDecl::propimpl_iterator 117664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = IMD ? IMD->propimpl_begin() : CID->propimpl_begin(), 117764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = IMD ? IMD->propimpl_end() : CID->propimpl_end(); 117864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) { 117964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewritePropertyImplDecl(*I, IMD, CID); 118064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 118164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 118264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(IMD ? IMD->getLocEnd() : CID->getLocEnd(), "// "); 118364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 118464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 118564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteInterfaceDecl(ObjCInterfaceDecl *ClassDecl) { 118664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 118764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ObjCForwardDecls.count(ClassDecl->getCanonicalDecl())) { 118864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we haven't seen a forward decl - generate a typedef. 118964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr = "#ifndef _REWRITER_typedef_"; 119064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ClassDecl->getNameAsString(); 119164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "\n"; 119264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "#define _REWRITER_typedef_"; 119364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ClassDecl->getNameAsString(); 119464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "\n"; 119564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "typedef struct objc_object "; 119664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ClassDecl->getNameAsString(); 119764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += ";\n#endif\n"; 119864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Mark this typedef as having been generated. 119964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCForwardDecls.insert(ClassDecl->getCanonicalDecl()); 120064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 120164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCInternalStruct(ClassDecl, ResultStr); 120264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 120364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCInterfaceDecl::prop_iterator I = ClassDecl->prop_begin(), 120464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ClassDecl->prop_end(); I != E; ++I) 120564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteProperty(*I); 120664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCInterfaceDecl::instmeth_iterator 120764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = ClassDecl->instmeth_begin(), E = ClassDecl->instmeth_end(); 120864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 120964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 121064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCInterfaceDecl::classmeth_iterator 121164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I = ClassDecl->classmeth_begin(), E = ClassDecl->classmeth_end(); 121264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian I != E; ++I) 121364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMethodDeclaration(*I); 121464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 121564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Lastly, comment out the @end. 121664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(ClassDecl->getAtEndRange().getBegin(), strlen("@end"), 121764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "/* @end */"); 121864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 121964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 122064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewritePropertyOrImplicitSetter(PseudoObjectExpr *PseudoOp) { 122164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = PseudoOp->getSourceRange(); 122264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 122364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We just magically know some things about the structure of this 122464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression. 122564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *OldMsg = 122664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast<ObjCMessageExpr>(PseudoOp->getSemanticExpr( 122764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PseudoOp->getNumSemanticExprs() - 1)); 122864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 122964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Because the rewriter doesn't allow us to rewrite rewritten code, 123064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need to suppress rewriting the sub-statements. 123164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Base, *RHS; 123264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 123364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 123464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 123564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the base expression if we have one. 123664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = 0; 123764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OldMsg->getReceiverKind() == ObjCMessageExpr::Instance) { 123864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = OldMsg->getInstanceReceiver(); 123964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<OpaqueValueExpr>(Base)->getSourceExpr(); 124064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Base)); 124164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 124264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 124364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the RHS. 124464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS = cast<BinaryOperator>(PseudoOp->getSyntacticForm())->getRHS(); 124564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS = cast<OpaqueValueExpr>(RHS)->getSourceExpr(); 124664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(RHS)); 124764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 124864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 124964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // TODO: avoid this copy. 125064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<SourceLocation, 1> SelLocs; 125164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelectorLocs(SelLocs); 125264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 125364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *NewMsg = 0; 125464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (OldMsg->getReceiverKind()) { 125564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: 125664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 125764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 125864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 125964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getClassReceiverTypeInfo(), 126064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 126164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 126264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 126364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS, 126464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 126564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 126664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 126764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 126864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: 126964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 127064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 127164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 127264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base, 127364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 127464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 127564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 127664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS, 127764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 127864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 127964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 128064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 128164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: 128264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance: 128364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 128464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 128564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 128664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperLoc(), 128764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getReceiverKind() == ObjCMessageExpr::SuperInstance, 128864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperType(), 128964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 129064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 129164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 129264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RHS, 129364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 129464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 129564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 129664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 129764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 129864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *Replacement = SynthMessageExpr(NewMsg); 129964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmtWithRange(PseudoOp, Replacement, OldRange); 130064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 130164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 130264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 130364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewritePropertyOrImplicitGetter(PseudoObjectExpr *PseudoOp) { 130464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = PseudoOp->getSourceRange(); 130564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 130664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We just magically know some things about the structure of this 130764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression. 130864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *OldMsg = 130964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast<ObjCMessageExpr>(PseudoOp->getResultExpr()->IgnoreImplicit()); 131064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 131164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Because the rewriter doesn't allow us to rewrite rewritten code, 131264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need to suppress rewriting the sub-statements. 131364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Base = 0; 131464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 131564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 131664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 131764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rebuild the base expression if we have one. 131864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (OldMsg->getReceiverKind() == ObjCMessageExpr::Instance) { 131964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = OldMsg->getInstanceReceiver(); 132064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<OpaqueValueExpr>(Base)->getSourceExpr(); 132164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(Base)); 132264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 132364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 132464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 132564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Intentionally empty. 132664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<SourceLocation, 1> SelLocs; 132764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 1> Args; 132864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 132964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMessageExpr *NewMsg = 0; 133064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (OldMsg->getReceiverKind()) { 133164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: 133264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 133364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 133464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 133564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getClassReceiverTypeInfo(), 133664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 133764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 133864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 133964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 134064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 134164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 134264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 134364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 134464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: 134564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 134664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 134764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 134864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Base, 134964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 135064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 135164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 135264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 135364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 135464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 135564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 135664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 135764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: 135864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance: 135964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewMsg = ObjCMessageExpr::Create(*Context, OldMsg->getType(), 136064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getValueKind(), 136164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getLeftLoc(), 136264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperLoc(), 136364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getReceiverKind() == ObjCMessageExpr::SuperInstance, 136464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSuperType(), 136564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getSelector(), 136664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelLocs, 136764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getMethodDecl(), 136864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Args, 136964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->getRightLoc(), 137064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldMsg->isImplicit()); 137164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 137264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 137364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 137464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *Replacement = SynthMessageExpr(NewMsg); 137564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmtWithRange(PseudoOp, Replacement, OldRange); 137664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 137764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 137864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 137964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SynthCountByEnumWithState - To print: 138064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ((unsigned int (*) 138164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int)) 138264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (void *)objc_msgSend)((id)l_collection, 138364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// sel_registerName( 138464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// "countByEnumeratingWithState:objects:count:"), 138564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// &enumState, 138664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// (id *)__rw_items, (unsigned int)16) 138764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 138864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthCountByEnumWithState(std::string &buf) { 138964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "((unsigned int (*) (id, SEL, struct __objcFastEnumerationState *, " 139064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "id *, unsigned int))(void *)objc_msgSend)"; 139164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 139264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "((id)l_collection,\n\t\t"; 139364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "sel_registerName(\"countByEnumeratingWithState:objects:count:\"),"; 139464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 139564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "&enumState, " 139664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "(id *)__rw_items, (unsigned int)16)"; 139764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 139864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 139964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteBreakStmt - Rewrite for a break-stmt inside an ObjC2's foreach 140064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// statement to exit to its outer synthesized loop. 140164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 140264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteBreakStmt(BreakStmt *S) { 140364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back())) 140464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 140564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace break with goto __break_label 140664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 140764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 140864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 140964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "goto __break_label_"; 141064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 141164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, strlen("break"), buf); 141264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 141364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 141464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 141564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 141664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteContinueStmt - Rewrite for a continue-stmt inside an ObjC2's foreach 141764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// statement to continue with its inner synthesized loop. 141864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 141964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteContinueStmt(ContinueStmt *S) { 142064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back())) 142164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 142264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace continue with goto __continue_label 142364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 142464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 142564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 142664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "goto __continue_label_"; 142764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 142864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, strlen("continue"), buf); 142964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 143064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 143164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 143264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 143364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCForCollectionStmt - Rewriter for ObjC2's foreach statement. 143464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// It rewrites: 143564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// for ( type elem in collection) { stmts; } 143664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 143764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// Into: 143864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// { 143964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// type elem; 144064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __objcFastEnumerationState enumState = { 0 }; 144164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id __rw_items[16]; 144264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id l_collection = (id)collection; 144364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long limit = [l_collection countByEnumeratingWithState:&enumState 144464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objects:__rw_items count:16]; 144564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// if (limit) { 144664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long startMutations = *enumState.mutationsPtr; 144764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// do { 144864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// unsigned long counter = 0; 144964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// do { 145064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// if (startMutations != *enumState.mutationsPtr) 145164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objc_enumerationMutation(l_collection); 145264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = (type)enumState.itemsPtr[counter++]; 145364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// stmts; 145464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __continue_label: ; 145564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } while (counter < limit); 145664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } while (limit = [l_collection countByEnumeratingWithState:&enumState 145764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objects:__rw_items count:16]); 145864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = nil; 145964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __break_label: ; 146064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 146164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// else 146264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// elem = nil; 146364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 146464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 146564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S, 146664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation OrigEnd) { 146764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!Stmts.empty() && "ObjCForCollectionStmt - Statement stack empty"); 146864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(isa<ObjCForCollectionStmt>(Stmts.back()) && 146964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "ObjCForCollectionStmt Statement stack mismatch"); 147064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!ObjCBcLabelNo.empty() && 147164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "ObjCForCollectionStmt - Label No stack empty"); 147264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 147364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 147464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 147564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef elementName; 147664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string elementTypeAsString; 147764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 147864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "\n{\n\t"; 147964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclStmt *DS = dyn_cast<DeclStmt>(S->getElement())) { 148064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // type elem; 148164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NamedDecl* D = cast<NamedDecl>(DS->getSingleDecl()); 148264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ElementType = cast<ValueDecl>(D)->getType(); 148364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ElementType->isObjCQualifiedIdType() || 148464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ElementType->isObjCQualifiedInterfaceType()) 148564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simply use 'id' for all qualified types. 148664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = "id"; 148764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 148864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = ElementType.getAsString(Context->getPrintingPolicy()); 148964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 149064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " "; 149164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementName = D->getName(); 149264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 149364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ";\n\t"; 149464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 149564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 149664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DR = cast<DeclRefExpr>(S->getElement()); 149764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementName = DR->getDecl()->getName(); 149864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = cast<ValueDecl>(DR->getDecl()); 149964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getType()->isObjCQualifiedIdType() || 150064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD->getType()->isObjCQualifiedInterfaceType()) 150164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simply use 'id' for all qualified types. 150264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = "id"; 150364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 150464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian elementTypeAsString = VD->getType().getAsString(Context->getPrintingPolicy()); 150564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 150664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 150764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct __objcFastEnumerationState enumState = { 0 }; 150864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "struct __objcFastEnumerationState enumState = { 0 };\n\t"; 150964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // id __rw_items[16]; 151064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id __rw_items[16];\n\t"; 151164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // id l_collection = (id) 151264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id l_collection = (id)"; 151364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find start location of 'collection' the hard way! 151464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startCollectionBuf = startBuf; 151564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf += 3; // skip 'for' 151664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf = strchr(startCollectionBuf, '('); 151764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf++; // skip '(' 151864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // find 'in' and skip it. 151964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startCollectionBuf != ' ' || 152064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *(startCollectionBuf+1) != 'i' || *(startCollectionBuf+2) != 'n' || 152164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*(startCollectionBuf+3) != ' ' && 152264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *(startCollectionBuf+3) != '[' && *(startCollectionBuf+3) != '(')) 152364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf++; 152464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startCollectionBuf += 3; 152564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 152664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace: "for (type element in" with string constructed thus far. 152764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, startCollectionBuf - startBuf, buf); 152864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace ')' in for '(' type elem in collection ')' with ';' 152964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation rightParenLoc = S->getRParenLoc(); 153064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *rparenBuf = SM->getCharacterData(rightParenLoc); 153164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation lparenLoc = startLoc.getLocWithOffset(rparenBuf-startBuf); 153264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = ";\n\t"; 153364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 153464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // unsigned long limit = [l_collection countByEnumeratingWithState:&enumState 153564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // objects:__rw_items count:16]; 153664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // which is synthesized into: 153764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // unsigned int limit = 153864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ((unsigned int (*) 153964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int)) 154064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (void *)objc_msgSend)((id)l_collection, 154164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // sel_registerName( 154264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // "countByEnumeratingWithState:objects:count:"), 154364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct __objcFastEnumerationState *)&state, 154464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id *)__rw_items, (unsigned int)16); 154564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long limit =\n\t\t"; 154664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthCountByEnumWithState(buf); 154764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ";\n\t"; 154864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// if (limit) { 154964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// unsigned long startMutations = *enumState.mutationsPtr; 155064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// do { 155164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// unsigned long counter = 0; 155264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// do { 155364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// if (startMutations != *enumState.mutationsPtr) 155464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// objc_enumerationMutation(l_collection); 155564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = (type)enumState.itemsPtr[counter++]; 155664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (limit) {\n\t"; 155764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long startMutations = *enumState.mutationsPtr;\n\t"; 155864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "do {\n\t\t"; 155964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "unsigned long counter = 0;\n\t\t"; 156064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "do {\n\t\t\t"; 156164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (startMutations != *enumState.mutationsPtr)\n\t\t\t\t"; 156264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "objc_enumerationMutation(l_collection);\n\t\t\t"; 156364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 156464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = ("; 156564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 156664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")enumState.itemsPtr[counter++];"; 156764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace ')' in for '(' type elem in collection ')' with all of these. 156864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(lparenLoc, 1, buf); 156964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 157064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// __continue_label: ; 157164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } while (counter < limit); 157264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } while (limit = [l_collection countByEnumeratingWithState:&enumState 157364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// objects:__rw_items count:16]); 157464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = nil; 157564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// __break_label: ; 157664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } 157764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// else 157864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// elem = nil; 157964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// } 158064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /// 158164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = ";\n\t"; 158264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "__continue_label_"; 158364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 158464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ": ;"; 158564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n\t\t"; 158664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "} while (counter < limit);\n\t"; 158764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "} while (limit = "; 158864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthCountByEnumWithState(buf); 158964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ");\n\t"; 159064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 159164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = (("; 159264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 159364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")0);\n\t"; 159464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "__break_label_"; 159564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += utostr(ObjCBcLabelNo.back()); 159664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ": ;\n\t"; 159764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n\t"; 159864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "else\n\t\t"; 159964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementName; 160064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " = (("; 160164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += elementTypeAsString; 160264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ")0);\n\t"; 160364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 160464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 160564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert all these *after* the statement body. 160664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: If this should support Obj-C++, support CXXTryStmt 160764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<CompoundStmt>(S->getBody())) { 160864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endBodyLoc = OrigEnd.getLocWithOffset(1); 160964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(endBodyLoc, buf); 161064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 161164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* Need to treat single statements specially. For example: 161264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * 161364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * for (A *a in b) if (stuff()) break; 161464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * for (A *a in b) xxxyy; 161564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * 161664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian * The following code simply scans ahead to the semi to find the actual end. 161764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 161864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *stmtBuf = SM->getCharacterData(OrigEnd); 161964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(stmtBuf, ';'); 162064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(semiBuf && "Can't find ';'"); 162164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endBodyLoc = OrigEnd.getLocWithOffset(semiBuf-stmtBuf+1); 162264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(endBodyLoc, buf); 162364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 162464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.pop_back(); 162564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCBcLabelNo.pop_back(); 162664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 162764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 162864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 162964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCSynchronizedStmt - 163064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// This routine rewrites @synchronized(expr) stmt; 163164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// into: 163264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// objc_sync_enter(expr); 163364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// @try stmt @finally { objc_sync_exit(expr); } 163464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 163564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S) { 163664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 163764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 163864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 163964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 164064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @synchronized location"); 164164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 164264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 164364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "objc_sync_enter((id)"; 164464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *lparenBuf = startBuf; 164564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*lparenBuf != '(') lparenBuf++; 164664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, lparenBuf-startBuf+1, buf); 164764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We can't use S->getSynchExpr()->getLocEnd() to find the end location, since 164864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the sync expression is typically a message expression that's already 164964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // been rewritten! (which implies the SourceLocation's are invalid). 165064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endLoc = S->getSynchBody()->getLocStart(); 165164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(endLoc); 165264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*endBuf != ')') endBuf--; 165364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation rparenLoc = startLoc.getLocWithOffset(endBuf-startBuf); 165464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = ");\n"; 165564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declare a new scope with two variables, _stack and _rethrow. 165664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "/* @try scope begin */ \n{ struct _objc_exception_data {\n"; 165764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "int buf[18/*32-bit i386*/];\n"; 165864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "char *pointers[4];} _stack;\n"; 165964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id volatile _rethrow = 0;\n"; 166064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "objc_exception_try_enter(&_stack);\n"; 166164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n"; 166264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(rparenLoc, 1, buf); 166364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = S->getSynchBody()->getLocEnd(); 166464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = SM->getCharacterData(startLoc); 166564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 166664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '}') && "bogus @synchronized block"); 166764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation lastCurlyLoc = startLoc; 166864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "}\nelse {\n"; 166964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " _rethrow = objc_exception_extract(&_stack);\n"; 167064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 167164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "{ /* implicit finally clause */\n"; 167264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " if (!_rethrow) objc_exception_try_exit(&_stack);\n"; 167364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 167464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string syncBuf; 167564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian syncBuf += " objc_sync_exit("; 167664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 167764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *syncExpr = S->getSynchExpr(); 167864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK = syncExpr->getType()->isObjCObjectPointerType() 167964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BitCast : 168064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian syncExpr->getType()->isBlockPointerType() 168164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BlockPointerToObjCPointerCast 168264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : CK_CPointerToObjCPointerCast; 168364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian syncExpr = NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 168464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK, syncExpr); 168564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string syncExprBufS; 168664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream syncExprBuf(syncExprBufS); 168764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian syncExpr->printPretty(syncExprBuf, *Context, 0, 168864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PrintingPolicy(LangOpts)); 168964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian syncBuf += syncExprBuf.str(); 169064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian syncBuf += ");"; 169164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 169264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += syncBuf; 169364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\n if (_rethrow) objc_exception_throw(_rethrow);\n"; 169464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 169564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}"; 169664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 169764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(lastCurlyLoc, 1, buf); 169864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 169964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool hasReturns = false; 170064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HasReturnStmts(S->getSynchBody(), hasReturns); 170164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasReturns) 170264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteSyncReturnStmts(S->getSynchBody(), syncBuf); 170364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 170464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 170564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 170664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 170764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::WarnAboutReturnGotoStmts(Stmt *S) 170864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian{ 170964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up traversal of all children. 171064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 171164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) 171264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian WarnAboutReturnGotoStmts(*CI); 171364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 171464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ReturnStmt>(S) || isa<GotoStmt>(S)) { 171564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Diags.Report(Context->getFullLoc(S->getLocStart()), 171664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TryFinallyContainsReturnDiag); 171764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 171864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 171964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 172064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 172164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HasReturnStmts(Stmt *S, bool &hasReturns) 172264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian{ 172364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up traversal of all children. 172464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 172564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) 172664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HasReturnStmts(*CI, hasReturns); 172764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 172864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ReturnStmt>(S)) 172964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian hasReturns = true; 173064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 173164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 173264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 173364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteTryReturnStmts(Stmt *S) { 173464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up traversal of all children. 173564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 173664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 173764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTryReturnStmts(*CI); 173864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 173964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ReturnStmt>(S)) { 174064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 174164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 174264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 174364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 174464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "RewriteTryReturnStmts: can't find ';'"); 174564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation onePastSemiLoc = startLoc.getLocWithOffset(semiBuf-startBuf+1); 174664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 174764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 174864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "{ objc_exception_try_exit(&_stack); return"; 174964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 175064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, 6, buf); 175164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, "}"); 175264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 175364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 175464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 175564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 175664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteSyncReturnStmts(Stmt *S, std::string syncExitBuf) { 175764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up traversal of all children. 175864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 175964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 176064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteSyncReturnStmts(*CI, syncExitBuf); 176164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 176264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<ReturnStmt>(S)) { 176364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 176464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 176564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 176664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 176764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "RewriteSyncReturnStmts: can't find ';'"); 176864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation onePastSemiLoc = startLoc.getLocWithOffset(semiBuf-startBuf+1); 176964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 177064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 177164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "{ objc_exception_try_exit(&_stack);"; 177264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += syncExitBuf; 177364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " return"; 177464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 177564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, 6, buf); 177664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(onePastSemiLoc, "}"); 177764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 177864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 177964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 178064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 178164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCTryStmt(ObjCAtTryStmt *S) { 178264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 178364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 178464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 178564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 178664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @try location"); 178764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 178864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 178964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declare a new scope with two variables, _stack and _rethrow. 179064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "/* @try scope begin */ { struct _objc_exception_data {\n"; 179164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "int buf[18/*32-bit i386*/];\n"; 179264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "char *pointers[4];} _stack;\n"; 179364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "id volatile _rethrow = 0;\n"; 179464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "objc_exception_try_enter(&_stack);\n"; 179564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n"; 179664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 179764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, 4, buf); 179864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 179964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = S->getTryBody()->getLocEnd(); 180064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = SM->getCharacterData(startLoc); 180164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 180264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '}') && "bogus @try block"); 180364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 180464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation lastCurlyLoc = startLoc; 180564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (S->getNumCatchStmts()) { 180664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = startLoc.getLocWithOffset(1); 180764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = " /* @catch begin */ else {\n"; 180864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " id _caught = objc_exception_extract(&_stack);\n"; 180964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " objc_exception_try_enter (&_stack);\n"; 181064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " if (_setjmp(_stack.buf))\n"; 181164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " _rethrow = objc_exception_extract(&_stack);\n"; 181264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " else { /* @catch continue */"; 181364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 181464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(startLoc, buf); 181564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { /* no catch list */ 181664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "}\nelse {\n"; 181764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " _rethrow = objc_exception_extract(&_stack);\n"; 181864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}"; 181964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(lastCurlyLoc, 1, buf); 182064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 182164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *lastCatchBody = 0; 182264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I) { 182364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCAtCatchStmt *Catch = S->getCatchStmt(I); 182464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *catchDecl = Catch->getCatchParamDecl(); 182564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 182664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (I == 0) 182764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "if ("; // we are generating code for the first catch clause 182864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 182964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "else if ("; 183064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = Catch->getLocStart(); 183164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = SM->getCharacterData(startLoc); 183264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 183364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @catch location"); 183464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 183564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *lParenLoc = strchr(startBuf, '('); 183664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 183764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Catch->hasEllipsis()) { 183864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now rewrite the body... 183964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian lastCatchBody = Catch->getCatchBody(); 184064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation bodyLoc = lastCatchBody->getLocStart(); 184164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *bodyBuf = SM->getCharacterData(bodyLoc); 184264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(*SM->getCharacterData(Catch->getRParenLoc()) == ')' && 184364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "bogus @catch paren location"); 184464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*bodyBuf == '{') && "bogus @catch body location"); 184564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 184664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "1) { id _tmp = _caught;"; 184764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Rewrite.ReplaceText(startLoc, bodyBuf-startBuf+1, buf); 184864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (catchDecl) { 184964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = catchDecl->getType(); 185064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (t == Context->getObjCIdType()) { 185164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "1) { "; 185264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, lParenLoc-startBuf+1, buf); 185364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const ObjCObjectPointerType *Ptr = 185464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian t->getAs<ObjCObjectPointerType>()) { 185564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Should be a pointer to a class. 185664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *IDecl = Ptr->getObjectType()->getInterface(); 185764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IDecl) { 185864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "objc_exception_match((struct objc_class *)objc_getClass(\""; 185964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += IDecl->getNameAsString(); 186064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "\"), (struct objc_object *)_caught)) { "; 186164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, lParenLoc-startBuf+1, buf); 186264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 186364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 186464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now rewrite the body... 186564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian lastCatchBody = Catch->getCatchBody(); 186664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation rParenLoc = Catch->getRParenLoc(); 186764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation bodyLoc = lastCatchBody->getLocStart(); 186864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *bodyBuf = SM->getCharacterData(bodyLoc); 186964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *rParenBuf = SM->getCharacterData(rParenLoc); 187064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*rParenBuf == ')') && "bogus @catch paren location"); 187164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*bodyBuf == '{') && "bogus @catch body location"); 187264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 187364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Here we replace ") {" with "= _caught;" (which initializes and 187464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declares the @catch parameter). 187564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(rParenLoc, bodyBuf-rParenBuf+1, " = _caught;"); 187664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 187764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("@catch rewrite bug"); 187864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 187964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 188064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Complete the catch list... 188164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (lastCatchBody) { 188264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation bodyLoc = lastCatchBody->getLocEnd(); 188364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(*SM->getCharacterData(bodyLoc) == '}' && 188464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "bogus @catch body location"); 188564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 188664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert the last (implicit) else clause *before* the right curly brace. 188764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bodyLoc = bodyLoc.getLocWithOffset(-1); 188864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "} /* last catch end */\n"; 188964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "else {\n"; 189064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " _rethrow = _caught;\n"; 189164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " objc_exception_try_exit(&_stack);\n"; 189264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "} } /* @catch end */\n"; 189364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!S->getFinallyStmt()) 189464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}\n"; 189564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(bodyLoc, buf); 189664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 189764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Set lastCurlyLoc 189864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian lastCurlyLoc = lastCatchBody->getLocEnd(); 189964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 190064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtFinallyStmt *finalStmt = S->getFinallyStmt()) { 190164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = finalStmt->getLocStart(); 190264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = SM->getCharacterData(startLoc); 190364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @finally start"); 190464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 190564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, 8, "/* @finally */"); 190664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 190764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *body = finalStmt->getFinallyBody(); 190864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = body->getLocStart(); 190964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endLoc = body->getLocEnd(); 191064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(*SM->getCharacterData(startLoc) == '{' && 191164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "bogus @finally body location"); 191264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(*SM->getCharacterData(endLoc) == '}' && 191364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "bogus @finally body location"); 191464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 191564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = startLoc.getLocWithOffset(1); 191664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(startLoc, " if (!_rethrow) objc_exception_try_exit(&_stack);\n"); 191764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endLoc = endLoc.getLocWithOffset(-1); 191864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(endLoc, " if (_rethrow) objc_exception_throw(_rethrow);\n"); 191964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 192064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Set lastCurlyLoc 192164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian lastCurlyLoc = body->getLocEnd(); 192264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 192364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now check for any return/continue/go statements within the @try. 192464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian WarnAboutReturnGotoStmts(S->getTryBody()); 192564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { /* no finally clause - make sure we synthesize an implicit one */ 192664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "{ /* implicit finally clause */\n"; 192764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " if (!_rethrow) objc_exception_try_exit(&_stack);\n"; 192864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += " if (_rethrow) objc_exception_throw(_rethrow);\n"; 192964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "}"; 193064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(lastCurlyLoc, 1, buf); 193164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 193264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now check for any return/continue/go statements within the @try. 193364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The implicit finally clause won't called if the @try contains any 193464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // jump statements. 193564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool hasReturns = false; 193664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HasReturnStmts(S->getTryBody(), hasReturns); 193764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasReturns) 193864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTryReturnStmts(S->getTryBody()); 193964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 194064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now emit the final closing curly brace... 194164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian lastCurlyLoc = lastCurlyLoc.getLocWithOffset(1); 194264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(lastCurlyLoc, " } /* @try scope end */\n"); 194364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 194464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 194564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 194664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// This can't be done with ReplaceStmt(S, ThrowExpr), since 194764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// the throw expression is typically a message expression that's already 194864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// been rewritten! (which implies the SourceLocation's are invalid). 194964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCThrowStmt(ObjCAtThrowStmt *S) { 195064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the start location and compute the semi location. 195164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = S->getLocStart(); 195264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 195364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 195464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startBuf == '@') && "bogus @throw location"); 195564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 195664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 195764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* void objc_exception_throw(id) __attribute__((noreturn)); */ 195864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (S->getThrowExpr()) 195964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "objc_exception_throw("; 196064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else // add an implicit argument 196164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = "objc_exception_throw(_caught"; 196264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 196364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // handle "@ throw" correctly. 196464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *wBuf = strchr(startBuf, 'w'); 196564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*wBuf == 'w') && "@throw: can't find 'w'"); 196664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(startLoc, wBuf-startBuf+1, buf); 196764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 196864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startBuf, ';'); 196964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "@throw: can't find ';'"); 197064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation semiLoc = startLoc.getLocWithOffset(semiBuf-startBuf); 197164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(semiLoc, 1, ");"); 197264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return 0; 197364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 197464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 197564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteAtEncode(ObjCEncodeExpr *Exp) { 197664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a new string expression. 197764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType StrType = Context->getPointerType(Context->CharTy); 197864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StrEncoding; 197964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForType(Exp->getEncodedType(), StrEncoding); 198064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Replacement = StringLiteral::Create(*Context, StrEncoding, 198164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 198264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StrType, SourceLocation()); 198364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, Replacement); 198464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 198564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace this subexpr in the parent. 198664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 198764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 198864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 198964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 199064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteAtSelector(ObjCSelectorExpr *Exp) { 199164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SelGetUidFunctionDecl) 199264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSelGetUidFunctionDecl(); 199364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(SelGetUidFunctionDecl && "Can't find sel_registerName() decl"); 199464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a call to sel_registerName("selName"). 199564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> SelExprs; 199664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 199764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 199864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getSelector().getAsString(), 199964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 200064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 200164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 200264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &SelExprs[0], SelExprs.size()); 200364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, SelExp); 200464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 200564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return SelExp; 200664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 200764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 200864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianCallExpr *RewriteModernObjC::SynthesizeCallToFunctionDecl( 200964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD, Expr **args, unsigned nargs, SourceLocation StartLoc, 201064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc) { 201164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 201264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = FD->getType(); 201364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 201464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 201564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = 201664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) DeclRefExpr(FD, msgSendType, VK_LValue, SourceLocation()); 201764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 201864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now, we cast the reference to a pointer to the objc_msgSend type. 201964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType pToFunc = Context->getPointerType(msgSendType); 202064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImplicitCastExpr *ICE = 202164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImplicitCastExpr::Create(*Context, pToFunc, CK_FunctionToPointerDecay, 202264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DRE, 0, VK_RValue); 202364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 202464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 202564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 202664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Exp = 202764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) CallExpr(*Context, ICE, args, nargs, 202864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getCallResultType(*Context), 202964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, EndLoc); 203064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Exp; 203164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 203264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 203364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool scanForProtocolRefs(const char *startBuf, const char *endBuf, 203464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&startRef, const char *&endRef) { 203564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (startBuf < endBuf) { 203664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '<') 203764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startRef = startBuf; // mark the start. 203864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '>') { 203964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (startRef && *startRef == '<') { 204064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endRef = startBuf; // mark the end. 204164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 204264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 204364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 204464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 204564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 204664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 204764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 204864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 204964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 205064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void scanToNextArgument(const char *&argRef) { 205164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int angle = 0; 205264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argRef != ')' && (*argRef != ',' || angle > 0)) { 205364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*argRef == '<') 205464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian angle++; 205564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*argRef == '>') 205664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian angle--; 205764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argRef++; 205864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 205964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(angle == 0 && "scanToNextArgument - bad protocol type syntax"); 206064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 206164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 206264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::needToScanForQualifiers(QualType T) { 206364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCQualifiedIdType()) 206464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 206564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const PointerType *PT = T->getAs<PointerType>()) { 206664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT->getPointeeType()->isObjCQualifiedIdType()) 206764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 206864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 206964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isObjCObjectPointerType()) { 207064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T = T->getPointeeType(); 207164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return T->isObjCQualifiedInterfaceType(); 207264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 207364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (T->isArrayType()) { 207464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ElemTy = Context->getBaseElementType(T); 207564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return needToScanForQualifiers(ElemTy); 207664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 207764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 207864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 207964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 208064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCQualifiedInterfaceTypes(Expr *E) { 208164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = E->getType(); 208264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(Type)) { 208364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc, EndLoc; 208464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 208564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) { 208664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = ECE->getLParenLoc(); 208764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc = ECE->getRParenLoc(); 208864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 208964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = E->getLocStart(); 209064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc = E->getLocEnd(); 209164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 209264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This will defend against trying to rewrite synthesized expressions. 209364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Loc.isInvalid() || EndLoc.isInvalid()) 209464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 209564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 209664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(Loc); 209764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(EndLoc); 209864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 209964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 210064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 210164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = Loc.getLocWithOffset(startRef-startBuf); 210264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = Loc.getLocWithOffset(endRef-startBuf+1); 210364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 210464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 210564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 210664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 210764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 210864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 210964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 211064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCQualifiedInterfaceTypes(Decl *Dcl) { 211164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Loc; 211264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type; 211364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *proto = 0; 211464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(Dcl)) { 211564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = VD->getLocation(); 211664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = VD->getType(); 211764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 211864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FunctionDecl *FD = dyn_cast<FunctionDecl>(Dcl)) { 211964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = FD->getLocation(); 212064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Check for ObjC 'id' and class types that have been adorned with protocol 212164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // information (id<p>, C<p>*). The protocol references need to be rewritten! 212264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *funcType = FD->getType()->getAs<FunctionType>(); 212364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(funcType && "missing function type"); 212464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian proto = dyn_cast<FunctionProtoType>(funcType); 212564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 212664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 212764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = proto->getResultType(); 212864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 212964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FieldDecl *FD = dyn_cast<FieldDecl>(Dcl)) { 213064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc = FD->getLocation(); 213164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Type = FD->getType(); 213264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 213364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 213464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 213564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 213664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(Type)) { 213764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since types are unique, we need to scan the buffer. 213864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 213964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(Loc); 214064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = endBuf; 214164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != ';' && *startBuf != '<' && startBuf != MainFileStart) 214264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf--; // scan backward (from the decl location) for return type. 214364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 214464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 214564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 214664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = Loc.getLocWithOffset(startRef-endBuf); 214764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = Loc.getLocWithOffset(endRef-endBuf+1); 214864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 214964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 215064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 215164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 215264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 215364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 215464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; // most likely, was a variable 215564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now check arguments. 215664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(Loc); 215764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startFuncBuf = startBuf; 215864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < proto->getNumArgs(); i++) { 215964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(proto->getArgType(i))) { 216064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since types are unique, we need to scan the buffer. 216164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 216264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = startBuf; 216364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scan forward (from the decl location) for argument types. 216464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian scanToNextArgument(endBuf); 216564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startRef = 0, *endRef = 0; 216664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) { 216764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the locations of the startRef, endRef. 216864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LessLoc = 216964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc.getLocWithOffset(startRef-startFuncBuf); 217064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation GreaterLoc = 217164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Loc.getLocWithOffset(endRef-startFuncBuf+1); 217264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Comment out the protocol references. 217364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(LessLoc, "/*"); 217464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(GreaterLoc, "*/"); 217564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 217664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = ++endBuf; 217764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 217864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 217964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If the function name is derived from a macro expansion, then the 218064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // argument buffer will not follow the name. Need to speak with Chris. 218164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf && *startBuf != ')' && *startBuf != ',') 218264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; // scan forward (from the decl location) for argument types. 218364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 218464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 218564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 218664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 218764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 218864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteTypeOfDecl(VarDecl *ND) { 218964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = ND->getType(); 219064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Type* TypePtr = QT->getAs<Type>(); 219164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isa<TypeOfExprType>(TypePtr)) 219264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 219364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (isa<TypeOfExprType>(TypePtr)) { 219464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const TypeOfExprType *TypeOfExprTypePtr = cast<TypeOfExprType>(TypePtr); 219564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = TypeOfExprTypePtr->getUnderlyingExpr()->getType(); 219664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypePtr = QT->getAs<Type>(); 219764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 219864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. This will not work for multiple declarators; as in: 219964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // __typeof__(a) b,c,d; 220064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeAsString(QT.getAsString(Context->getPrintingPolicy())); 220164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getTypeSpecStartLoc(); 220264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 220364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ND->getInit()) { 220464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 220564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeAsString += " " + Name + " = "; 220664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *E = ND->getInit(); 220764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc; 220864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) 220964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = ECE->getLParenLoc(); 221064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 221164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = E->getLocStart(); 221264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = SM->getExpansionLoc(startLoc); 221364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(startLoc); 221464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf-1, TypeAsString); 221564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 221664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 221764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation X = ND->getLocEnd(); 221864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian X = SM->getExpansionLoc(X); 221964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(X); 222064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf-1, TypeAsString); 222164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 222264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 222364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 222464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthSelGetUidFunctionDecl - SEL sel_registerName(const char *str); 222564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthSelGetUidFunctionDecl() { 222664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *SelGetUidIdent = &Context->Idents.get("sel_registerName"); 222764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 222864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 222964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getFuncType = 223064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSimpleFunctionType(Context->getObjCSelType(), &ArgTys[0], ArgTys.size()); 223164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 223264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 223364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 223464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidIdent, getFuncType, 0, 223564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 223664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 223764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 223864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 223964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteFunctionDecl(FunctionDecl *FD) { 224064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declared in <objc/objc.h> 224164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->getIdentifier() && 224264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getName() == "sel_registerName") { 224364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelGetUidFunctionDecl = FD; 224464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 224564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 224664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(FD); 224764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 224864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 224964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerType(std::string& Str, QualType Type) { 225064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString(Type.getAsString(Context->getPrintingPolicy())); 225164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = TypeString.c_str(); 225264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strchr(argPtr, '^')) { 225364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += TypeString; 225464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 225564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 225664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr) { 225764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += (*argPtr == '^' ? '*' : *argPtr); 225864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; 225964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 226064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 226164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 226264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// FIXME. Consolidate this routine with RewriteBlockPointerType. 226364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerTypeVariable(std::string& Str, 226464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD) { 226564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = VD->getType(); 226664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString(Type.getAsString(Context->getPrintingPolicy())); 226764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = TypeString.c_str(); 226864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int paren = 0; 226964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr) { 227064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 227164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': 227264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 227364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian paren++; 227464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 227564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': 227664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 227764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian paren--; 227864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 227964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 228064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += '*'; 228164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (paren == 1) 228264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += VD->getNameAsString(); 228364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 228464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: 228564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Str += *argPtr; 228664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 228764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 228864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; 228964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 229064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 229164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 229264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 229364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockLiteralFunctionDecl(FunctionDecl *FD) { 229464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart = FD->getTypeSpecStartLoc(); 229564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *funcType = FD->getType()->getAs<FunctionType>(); 229664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *proto = dyn_cast<FunctionProtoType>(funcType); 229764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!proto) 229864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 229964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Type = proto->getResultType(); 230064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FdStr = Type.getAsString(Context->getPrintingPolicy()); 230164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += " "; 230264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += FD->getName(); 230364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += "("; 230464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned numArgs = proto->getNumArgs(); 230564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < numArgs; i++) { 230664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType ArgType = proto->getArgType(i); 230764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(FdStr, ArgType); 230864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (i+1 < numArgs) 230964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += ", "; 231064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 231164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FdStr += ");\n"; 231264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, FdStr); 231364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = 0; 231464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 231564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 231664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthSuperContructorFunctionDecl - id objc_super(id obj, id super); 231764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthSuperContructorFunctionDecl() { 231864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SuperContructorFunctionDecl) 231964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 232064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("__rw_objc_super"); 232164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 232264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 232364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 232464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 232564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 232664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 232764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 232864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperContructorFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 232964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 233064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 233164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 233264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 233364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 233464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 233564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 233664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendFunctionDecl - id objc_msgSend(id self, SEL op, ...); 233764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendFunctionDecl() { 233864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend"); 233964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 234064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 234164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 234264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 234364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 234464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 234564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 234664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 234764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 234864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 234964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 235064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 235164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 235264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 235364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 235464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 235564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 235664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 235764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendSuperFunctionDecl - id objc_msgSendSuper(struct objc_super *, SEL op, ...); 235864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendSuperFunctionDecl() { 235964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSendSuper"); 236064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 236164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 236264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 236364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_super")); 236464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getPointerType(Context->getTagDeclType(RD)); 236564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't build 'struct objc_super *' type"); 236664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 236764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 236864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 236964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 237064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 237164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 237264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 237364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 237464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 237564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 237664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 237764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 237864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 237964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 238064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 238164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendStretFunctionDecl - id objc_msgSend_stret(id self, SEL op, ...); 238264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendStretFunctionDecl() { 238364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_stret"); 238464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 238564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 238664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 238764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 238864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 238964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 239064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 239164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 239264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 239364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 239464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 239564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 239664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 239764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 239864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 239964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 240064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 240164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 240264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendSuperStretFunctionDecl - 240364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// id objc_msgSendSuper_stret(struct objc_super *, SEL op, ...); 240464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendSuperStretFunctionDecl() { 240564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = 240664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_msgSendSuper_stret"); 240764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 240864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 240964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 241064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_super")); 241164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getPointerType(Context->getTagDeclType(RD)); 241264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't build 'struct objc_super *' type"); 241364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 241464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 241564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 241664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 241764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->getObjCIdType(), 241864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 241964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 242064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendSuperStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 242164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 242264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 242364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 242464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 242564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 242664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 242764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 242864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthMsgSendFpretFunctionDecl - double objc_msgSend_fpret(id self, SEL op, ...); 242964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthMsgSendFpretFunctionDecl() { 243064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_fpret"); 243164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 243264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argT = Context->getObjCIdType(); 243364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'id' type"); 243464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 243564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argT = Context->getObjCSelType(); 243664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!argT.isNull() && "Can't find 'SEL' type"); 243764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(argT); 243864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = getSimpleFunctionType(Context->DoubleTy, 243964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size(), 244064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true /*isVariadic*/); 244164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFpretFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 244264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 244364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 244464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian msgSendIdent, msgSendType, 0, 244564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 244664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 244764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 244864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 244964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetClassFunctionDecl - id objc_getClass(const char *name); 245064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetClassFunctionDecl() { 245164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getClass"); 245264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 245364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 245464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCIdType(), 245564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 245664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 245764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 245864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 245964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassIdent, getClassType, 0, 246064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 246164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 246264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 246364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 246464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetSuperClassFunctionDecl - Class class_getSuperclass(Class cls); 246564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetSuperClassFunctionDecl() { 246664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getSuperClassIdent = 246764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("class_getSuperclass"); 246864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 246964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getObjCClassType()); 247064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCClassType(), 247164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 247264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetSuperClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 247364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 247464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 247564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSuperClassIdent, 247664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassType, 0, 247764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 247864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, 247964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false); 248064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 248164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 248264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// SynthGetMetaClassFunctionDecl - id objc_getMetaClass(const char *name); 248364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthGetMetaClassFunctionDecl() { 248464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getMetaClass"); 248564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 16> ArgTys; 248664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTys.push_back(Context->getPointerType(Context->CharTy.withConst())); 248764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType getClassType = getSimpleFunctionType(Context->getObjCIdType(), 248864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ArgTys[0], ArgTys.size()); 248964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetMetaClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl, 249064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 249164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 249264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getClassIdent, getClassType, 0, 249364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, 249464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false); 249564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 249664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 249764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCStringLiteral(ObjCStringLiteral *Exp) { 249864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType strType = getConstantStringStructType(); 249964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 250064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "__NSConstantStringImpl_"; 250164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 250264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string tmpName = InFileName; 250364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned i; 250464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (i=0; i < tmpName.length(); i++) { 250564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char c = tmpName.at(i); 250664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // replace any non alphanumeric characters with '_'. 250764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isalpha(c) && (c < '0' || c > '9')) 250864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian tmpName[i] = '_'; 250964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 251064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += tmpName; 251164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_"; 251264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(NumObjCStringLiterals++); 251364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 251464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "static __NSConstantStringImpl " + S; 251564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " __attribute__ ((section (\"__DATA, __cfstring\"))) = {__CFConstantStringClassReference,"; 251664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "0x000007c8,"; // utf8_str 251764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The pretty printer for StringLiteral handles escape characters properly. 251864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string prettyBufS; 251964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream prettyBuf(prettyBufS); 252064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getString()->printPretty(prettyBuf, *Context, 0, 252164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PrintingPolicy(LangOpts)); 252264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += prettyBuf.str(); 252364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += ","; 252464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += utostr(Exp->getString()->getByteLength()) + "};\n"; 252564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 252664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 252764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), &Context->Idents.get(S), 252864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian strType, 0, SC_Static, SC_None); 252964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(NewVD, strType, VK_LValue, 253064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 253164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Unop = new (Context) UnaryOperator(DRE, UO_AddrOf, 253264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(DRE->getType()), 253364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 253464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 253564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // cast to NSConstantString * 253664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *cast = NoTypeInfoCStyleCastExpr(Context, Exp->getType(), 253764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_CPointerToObjCPointerCast, Unop); 253864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, cast); 253964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 254064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return cast; 254164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 254264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 254364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// struct objc_super { struct objc_object *receiver; struct objc_class *super; }; 254464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getSuperStructType() { 254564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SuperStructDecl) { 254664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 254764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 254864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("objc_super")); 254964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FieldTypes[2]; 255064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 255164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_object *receiver; 255264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[0] = Context->getObjCIdType(); 255364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_class *super; 255464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[1] = Context->getObjCClassType(); 255564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 255664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create fields 255764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < 2; ++i) { 255864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl->addDecl(FieldDecl::Create(*Context, SuperStructDecl, 255964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 256064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 0, 256164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[i], 0, 256264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, 256364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*Mutable=*/false, 256464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false)); 256564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 256664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 256764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperStructDecl->completeDefinition(); 256864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 256964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTagDeclType(SuperStructDecl); 257064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 257164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 257264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getConstantStringStructType() { 257364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ConstantStringDecl) { 257464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 257564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 257664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__NSConstantStringImpl")); 257764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FieldTypes[4]; 257864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 257964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_object *receiver; 258064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[0] = Context->getObjCIdType(); 258164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // int flags; 258264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[1] = Context->IntTy; 258364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // char *str; 258464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[2] = Context->getPointerType(Context->CharTy); 258564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // long length; 258664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[3] = Context->LongTy; 258764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 258864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create fields 258964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < 4; ++i) { 259064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl->addDecl(FieldDecl::Create(*Context, 259164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl, 259264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 259364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 0, 259464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldTypes[i], 0, 259564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, 259664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*Mutable=*/true, 259764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false)); 259864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 259964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 260064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConstantStringDecl->completeDefinition(); 260164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 260264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTagDeclType(ConstantStringDecl); 260364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 260464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 260564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthMessageExpr(ObjCMessageExpr *Exp, 260664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation StartLoc, 260764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation EndLoc) { 260864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!SelGetUidFunctionDecl) 260964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSelGetUidFunctionDecl(); 261064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendFunctionDecl) 261164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendFunctionDecl(); 261264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendSuperFunctionDecl) 261364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendSuperFunctionDecl(); 261464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendStretFunctionDecl) 261564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendStretFunctionDecl(); 261664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendSuperStretFunctionDecl) 261764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendSuperStretFunctionDecl(); 261864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!MsgSendFpretFunctionDecl) 261964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthMsgSendFpretFunctionDecl(); 262064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetClassFunctionDecl) 262164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetClassFunctionDecl(); 262264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetSuperClassFunctionDecl) 262364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetSuperClassFunctionDecl(); 262464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!GetMetaClassFunctionDecl) 262564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthGetMetaClassFunctionDecl(); 262664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 262764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // default to objc_msgSend(). 262864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendFlavor = MsgSendFunctionDecl; 262964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // May need to use objc_msgSend_stret() as well. 263064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *MsgSendStretFlavor = 0; 263164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *mDecl = Exp->getMethodDecl()) { 263264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType resultType = mDecl->getResultType(); 263364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (resultType->isRecordType()) 263464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendStretFunctionDecl; 263564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (resultType->isRealFloatingType()) 263664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendFpretFunctionDecl; 263764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 263864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 263964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize a call to objc_msgSend(). 264064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> MsgExprs; 264164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (Exp->getReceiverKind()) { 264264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperClass: { 264364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendSuperFunctionDecl; 264464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) 264564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendSuperStretFunctionDecl; 264664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(MsgSendFlavor && "MsgSendFlavor is NULL!"); 264764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 264864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = CurMethodDef->getClassInterface(); 264964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 265064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 265164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 265264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // set the receiver to self, the first argument to all methods. 265364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 265464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 265564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 265664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) DeclRefExpr(CurMethodDef->getSelfDecl(), 265764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 265864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, 265964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation())) 266064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ); // set the 'receiver'. 266164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 266264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 266364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 266464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 266564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 266664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassDecl->getIdentifier()->getName(), 266764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 266864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 266964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetMetaClassFunctionDecl, 267064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 267164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 267264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, 267364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 267464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (Class)objc_getClass("CurrentClass") 267564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *ArgExpr = NoTypeInfoCStyleCastExpr(Context, 267664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCClassType(), 267764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls); 267864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.clear(); 267964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(ArgExpr); 268064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Cls = SynthesizeCallToFunctionDecl(GetSuperClassFunctionDecl, 268164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], ClsExprs.size(), 268264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 268364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 268464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 268564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // To turn off a warning, type-cast to 'id' 268664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( // set 'super class', using class_getSuperclass(). 268764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, 268864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 268964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls)); 269064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_super 269164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType superType = getSuperStructType(); 269264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *SuperRep; 269364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 269464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 269564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSuperContructorFunctionDecl(); 269664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 269764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(SuperContructorFunctionDecl, 269864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 269964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 270064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], 270164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.size(), 270264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 270364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 270464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The code for super is a little tricky to prevent collision with 270564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the structure definition in the header. The rewriter has it's own 270664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // internal definition (__rw_objc_super) that is uses. This is why 270764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need the cast below. For example: 270864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super *)&__rw_objc_super((id)self, (id)objc_getClass("SUPER")) 270964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 271064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 271164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 271264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 271364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 271464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = NoTypeInfoCStyleCastExpr(Context, 271564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(superType), 271664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, SuperRep); 271764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 271864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super) { <exprs from above> } 271964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitListExpr *ILE = 272064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) InitListExpr(*Context, SourceLocation(), 272164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &InitExprs[0], InitExprs.size(), 272264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 272364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *superTInfo 272464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(superType); 272564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CompoundLiteralExpr(SourceLocation(), superTInfo, 272664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 272764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ILE, false); 272864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_super * 272964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 273064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 273164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 273264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 273364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 273464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SuperRep); 273564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 273664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 273764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 273864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Class: { 273964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 274064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 274164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *Class 274264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Exp->getClassReceiver()->getAs<ObjCObjectType>()->getInterface(); 274364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *clsName = Class->getIdentifier(); 274464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 274564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian clsName->getName(), 274664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 274764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 274864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 274964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 275064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 275164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 275264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(Cls); 275364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 275464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 275564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 275664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::SuperInstance:{ 275764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendFlavor = MsgSendSuperFunctionDecl; 275864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) 275964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgSendStretFlavor = MsgSendSuperStretFunctionDecl; 276064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(MsgSendFlavor && "MsgSendFlavor is NULL!"); 276164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = CurMethodDef->getClassInterface(); 276264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 276364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 276464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 276564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 276664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 276764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) DeclRefExpr(CurMethodDef->getSelfDecl(), 276864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCIdType(), 276964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, SourceLocation())) 277064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ); // set the 'receiver'. 277164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 277264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 277364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> ClsExprs; 277464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 277564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(StringLiteral::Create(*Context, 277664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassDecl->getIdentifier()->getName(), 277764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, argType, 277864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation())); 277964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl, 278064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], 278164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.size(), 278264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 278364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (Class)objc_getClass("CurrentClass") 278464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *ArgExpr = NoTypeInfoCStyleCastExpr(Context, 278564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCClassType(), 278664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls); 278764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.clear(); 278864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClsExprs.push_back(ArgExpr); 278964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Cls = SynthesizeCallToFunctionDecl(GetSuperClassFunctionDecl, 279064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &ClsExprs[0], ClsExprs.size(), 279164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, EndLoc); 279264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 279364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (id)class_getSuperclass((Class)objc_getClass("CurrentClass")) 279464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // To turn off a warning, type-cast to 'id' 279564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back( 279664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // set 'super class', using class_getSuperclass(). 279764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 279864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Cls)); 279964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct objc_super 280064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType superType = getSuperStructType(); 280164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *SuperRep; 280264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 280364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 280464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthSuperContructorFunctionDecl(); 280564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 280664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(SuperContructorFunctionDecl, 280764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, 280864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 280964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], 281064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.size(), 281164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_LValue, SourceLocation()); 281264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // The code for super is a little tricky to prevent collision with 281364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the structure definition in the header. The rewriter has it's own 281464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // internal definition (__rw_objc_super) that is uses. This is why 281564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we need the cast below. For example: 281664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super *)&__rw_objc_super((id)self, (id)objc_getClass("SUPER")) 281764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 281864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) UnaryOperator(SuperRep, UO_AddrOf, 281964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(SuperRep->getType()), 282064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 282164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 282264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = NoTypeInfoCStyleCastExpr(Context, 282364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(superType), 282464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, SuperRep); 282564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 282664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (struct objc_super) { <exprs from above> } 282764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitListExpr *ILE = 282864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) InitListExpr(*Context, SourceLocation(), 282964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &InitExprs[0], InitExprs.size(), 283064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 283164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *superTInfo 283264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(superType); 283364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperRep = new (Context) CompoundLiteralExpr(SourceLocation(), superTInfo, 283464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian superType, VK_RValue, ILE, 283564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false); 283664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 283764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SuperRep); 283864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 283964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 284064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 284164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ObjCMessageExpr::Instance: { 284264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Remove all type-casts because it may contain objc-style types; e.g. 284364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Foo<Proto> *. 284464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *recExpr = Exp->getInstanceReceiver(); 284564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(recExpr)) 284664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian recExpr = CE->getSubExpr(); 284764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK = recExpr->getType()->isObjCObjectPointerType() 284864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BitCast : recExpr->getType()->isBlockPointerType() 284964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? CK_BlockPointerToObjCPointerCast 285064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : CK_CPointerToObjCPointerCast; 285164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 285264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian recExpr = NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 285364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK, recExpr); 285464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(recExpr); 285564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 285664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 285764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 285864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 285964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a call to sel_registerName("selName"), it will be the 2nd argument. 286064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> SelExprs; 286164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType argType = Context->getPointerType(Context->CharTy); 286264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SelExprs.push_back(StringLiteral::Create(*Context, 286364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getSelector().getAsString(), 286464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringLiteral::Ascii, false, 286564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argType, SourceLocation())); 286664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl, 286764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &SelExprs[0], SelExprs.size(), 286864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StartLoc, 286964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 287064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(SelExp); 287164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 287264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now push any user supplied arguments. 287364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < Exp->getNumArgs(); i++) { 287464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *userExpr = Exp->getArg(i); 287564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make all implicit casts explicit...ICE comes in handy:-) 287664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(userExpr)) { 287764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Reuse the ICE type, it is exactly what the doctor ordered. 287864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType type = ICE->getType(); 287964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (needToScanForQualifiers(type)) 288064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian type = Context->getObjCIdType(); 288164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "type (^)(...)" to "type (*)(...)". 288264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(type); 288364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Expr *SubExpr = ICE->IgnoreParenImpCasts(); 288464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK; 288564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SubExpr->getType()->isIntegralType(*Context) && 288664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian type->isBooleanType()) { 288764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_IntegralToBoolean; 288864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (type->isObjCObjectPointerType()) { 288964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SubExpr->getType()->isBlockPointerType()) { 289064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BlockPointerToObjCPointerCast; 289164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (SubExpr->getType()->isPointerType()) { 289264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_CPointerToObjCPointerCast; 289364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 289464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 289564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 289664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 289764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 289864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 289964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 290064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = NoTypeInfoCStyleCastExpr(Context, type, CK, userExpr); 290164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 290264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make id<P...> cast into an 'id' cast. 290364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(userExpr)) { 290464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CE->getType()->isObjCQualifiedIdType()) { 290564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while ((CE = dyn_cast<CStyleCastExpr>(userExpr))) 290664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = CE->getSubExpr(); 290764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastKind CK; 290864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (userExpr->getType()->isIntegralType(*Context)) { 290964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_IntegralToPointer; 291064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (userExpr->getType()->isBlockPointerType()) { 291164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BlockPointerToObjCPointerCast; 291264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (userExpr->getType()->isPointerType()) { 291364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_CPointerToObjCPointerCast; 291464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 291564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK = CK_BitCast; 291664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 291764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian userExpr = NoTypeInfoCStyleCastExpr(Context, Context->getObjCIdType(), 291864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK, userExpr); 291964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 292064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 292164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.push_back(userExpr); 292264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We've transferred the ownership to MsgExprs. For now, we *don't* null 292364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // out the argument in the original expression (since we aren't deleting 292464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the ObjCMessageExpr). See RewritePropertyOrImplicitSetter() usage for more info. 292564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //Exp->setArg(i, 0); 292664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 292764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate the funky cast. 292864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *cast; 292964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 293064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType returnType; 293164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 293264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push 'id' and 'SEL', the 2 implicit arguments. 293364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendFlavor == MsgSendSuperFunctionDecl) 293464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getPointerType(getSuperStructType())); 293564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 293664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getObjCIdType()); 293764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(Context->getObjCSelType()); 293864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *OMD = Exp->getMethodDecl()) { 293964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push any user argument types. 294064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCMethodDecl::param_iterator PI = OMD->param_begin(), 294164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = OMD->param_end(); PI != E; ++PI) { 294264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = (*PI)->getType()->isObjCQualifiedIdType() 294364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ? Context->getObjCIdType() 294464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian : (*PI)->getType(); 294564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 294664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(t); 294764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 294864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 294964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType = Exp->getType(); 295064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian convertToUnqualifiedObjCType(returnType); 295164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(returnType); 295264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 295364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType = Context->getObjCIdType(); 295464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 295564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the type, we will need to reference it in a couple spots. 295664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType msgSendType = MsgSendFlavor->getType(); 295764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 295864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend() declaration. 295964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(MsgSendFlavor, msgSendType, 296064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, SourceLocation()); 296164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 296264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to cast objc_msgSend to "void *" (to workaround a GCC bandaid). 296364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we don't do this cast, we get the following bizarre warning/note: 296464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // xx.m:13: warning: function called through a non-compatible type 296564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // xx.m:13: note: if this code is reached, the program will abort 296664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, 296764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidTy), 296864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, DRE); 296964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 297064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the "normal" pointer to function cast. 297164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castType = 297264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 297364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If we don't have a method decl, force a variadic cast. 297464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : true); 297564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = Context->getPointerType(castType); 297664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 297764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast); 297864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 297964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 298064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(StartLoc, EndLoc, cast); 298164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 298264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = msgSendType->getAs<FunctionType>(); 298364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 298464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.size(), 298564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getResultType(), VK_RValue, 298664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian EndLoc); 298764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = CE; 298864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MsgSendStretFlavor) { 298964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We have the method which returns a struct/union. Must also generate 299064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // call to objc_msgSend_stret and hang both varieties on a conditional 299164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // expression which dictate which one to envoke depending on size of 299264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // method's return type. 299364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 299464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create a reference to the objc_msgSend_stret() declaration. 299564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *STDRE = new (Context) DeclRefExpr(MsgSendStretFlavor, msgSendType, 299664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, SourceLocation()); 299764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to cast objc_msgSend_stret to "void *" (see above comment). 299864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, 299964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidTy), 300064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, STDRE); 300164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the "normal" pointer to function cast. 300264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = getSimpleFunctionType(returnType, &ArgTypes[0], ArgTypes.size(), 300364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : false); 300464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castType = Context->getPointerType(castType); 300564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast = NoTypeInfoCStyleCastExpr(Context, castType, CK_BitCast, 300664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast); 300764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 300864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 300964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), cast); 301064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 301164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT = msgSendType->getAs<FunctionType>(); 301264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *STCE = new (Context) CallExpr(*Context, PE, &MsgExprs[0], 301364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MsgExprs.size(), 301464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FT->getResultType(), VK_RValue, 301564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 301664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 301764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build sizeof(returnType) 301864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UnaryExprOrTypeTraitExpr *sizeofExpr = 301964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) UnaryExprOrTypeTraitExpr(UETT_SizeOf, 302064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getTrivialTypeSourceInfo(returnType), 302164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getSizeType(), SourceLocation(), 302264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 302364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...)) 302464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Value of 8 is base on ppc32/x86 ABI for the most common cases. 302564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For X86 it is more complicated and some kind of target specific routine 302664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // is needed to decide what to do. 302764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 302864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 302964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IntegerLiteral *limit = IntegerLiteral::Create(*Context, 303064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::APInt(IntSize, 8), 303164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->IntTy, 303264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 303364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BinaryOperator *lessThanExpr = 303464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) BinaryOperator(sizeofExpr, limit, BO_LE, Context->IntTy, 303564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 303664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...)) 303764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConditionalOperator *CondExpr = 303864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) ConditionalOperator(lessThanExpr, 303964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), CE, 304064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), STCE, 304164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian returnType, VK_RValue, OK_Ordinary); 304264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplacingStmt = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 304364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CondExpr); 304464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 304564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 304664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return ReplacingStmt; 304764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 304864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 304964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteMessageExpr(ObjCMessageExpr *Exp) { 305064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *ReplacingStmt = SynthMessageExpr(Exp, Exp->getLocStart(), 305164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getLocEnd()); 305264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 305364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the actual rewrite. 305464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, ReplacingStmt); 305564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 305664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 305764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return ReplacingStmt; 305864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 305964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 306064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// typedef struct objc_object Protocol; 306164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::getProtocolType() { 306264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ProtocolTypeDecl) { 306364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeSourceInfo *TInfo 306464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = Context->getTrivialTypeSourceInfo(Context->getObjCIdType()); 306564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolTypeDecl = TypedefDecl::Create(*Context, TUDecl, 306664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 306764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("Protocol"), 306864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TInfo); 306964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 307064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Context->getTypeDeclType(ProtocolTypeDecl); 307164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 307264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 307364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCProtocolExpr - Rewrite a protocol expression into 307464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// a synthesized/forward data reference (to the protocol's metadata). 307564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// The forward references (and metadata) are generated in 307664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteModernObjC::HandleTranslationUnit(). 307764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp) { 307864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = "_OBJC_PROTOCOL_" + Exp->getProtocol()->getNameAsString(); 307964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *ID = &Context->Idents.get(Name); 308064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *VD = VarDecl::Create(*Context, TUDecl, SourceLocation(), 308164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), ID, getProtocolType(), 0, 308264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Extern, SC_None); 308364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(VD, getProtocolType(), VK_LValue, 308464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 308564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *DerefExpr = new (Context) UnaryOperator(DRE, UO_AddrOf, 308664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(DRE->getType()), 308764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 308864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, DerefExpr->getType(), 308964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 309064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DerefExpr); 309164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(Exp, castExpr); 309264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolExprDecls.insert(Exp->getProtocol()->getCanonicalDecl()); 309364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // delete Exp; leak for now, see RewritePropertyOrImplicitSetter() usage for more info. 309464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return castExpr; 309564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 309664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 309764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 309864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::BufferContainsPPDirectives(const char *startBuf, 309964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf) { 310064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (startBuf < endBuf) { 310164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '#') { 310264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Skip whitespace. 310364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (++startBuf; startBuf[0] == ' ' || startBuf[0] == '\t'; ++startBuf) 310464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ; 310564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!strncmp(startBuf, "if", strlen("if")) || 310664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "ifdef", strlen("ifdef")) || 310764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "ifndef", strlen("ifndef")) || 310864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "define", strlen("define")) || 310964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "undef", strlen("undef")) || 311064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "else", strlen("else")) || 311164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "elif", strlen("elif")) || 311264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "endif", strlen("endif")) || 311364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "pragma", strlen("pragma")) || 311464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "include", strlen("include")) || 311564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "import", strlen("import")) || 311664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !strncmp(startBuf, "include_next", strlen("include_next"))) 311764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 311864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 311964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 312064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 312164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 312264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 312364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 312464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCInternalStruct - Rewrite one internal struct corresponding to 312564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// an objective-c class with ivars. 312664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCInternalStruct(ObjCInterfaceDecl *CDecl, 312764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 312864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CDecl && "Class missing in SynthesizeObjCInternalStruct"); 312964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CDecl->getName() != "" && 313064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "Name missing in SynthesizeObjCInternalStruct"); 313164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Do not synthesize more than once. 313264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSynthesizedStructs.count(CDecl)) 313364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 313464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *RCDecl = CDecl->getSuperClass(); 3135a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian SmallVector<ObjCIvarDecl *, 8> IVars; 3136a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian for (ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 3137a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian IVD; IVD = IVD->getNextIvar()) { 3138a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian // Ignore unnamed bit-fields. 3139a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian if (!IVD->getDeclName()) 3140a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian continue; 3141a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian IVars.push_back(IVD); 3142a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian } 314364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CDecl->getLocStart(); 314464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = CDecl->getEndOfDefinitionLoc(); 3145a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 314664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 314764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 3148a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 314964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If no ivars and no root or if its root, directly or indirectly, 315064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // have no ivars (thus not synthesized) then no need to synthesize this class. 3151a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian if ((!CDecl->isThisDeclarationADefinition() || IVars.size() == 0) && 315264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (!RCDecl || !ObjCSynthesizedStructs.count(RCDecl))) { 315364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endBuf += Lexer::MeasureTokenLength(LocEnd, *SM, LangOpts); 315464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, Result); 315564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 315664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 3157a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 315864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct "; 315964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 3160a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IMPL {\n"; 3161a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 3162a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian if (RCDecl) { 3163a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "\tstruct "; Result += RCDecl->getNameAsString(); 3164a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IMPL "; Result += RCDecl->getNameAsString(); 3165a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "_IVARS;\n"; 3166a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian } 3167a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 3168a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian for (unsigned i = 0, e = IVars.size(); i < e; i++) { 3169a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian ObjCIvarDecl *IvarDecl = IVars[i]; 3170a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian QualType Type = IvarDecl->getType(); 3171a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian std::string TypeString(Type.getAsString(Context->getPrintingPolicy())); 3172a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "\t"; 3173a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += TypeString; Result += " "; Result += IvarDecl->getNameAsString(); 3174a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += ";\n"; 317564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 3176a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += "};\n"; 3177a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian endBuf += Lexer::MeasureTokenLength(LocEnd, *SM, LangOpts); 3178a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian ReplaceText(LocStart, endBuf-startBuf, Result); 3179a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian 318064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Mark this struct as having been generated. 318164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ObjCSynthesizedStructs.insert(CDecl)) 318264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("struct already synthesize- SynthesizeObjCInternalStruct"); 318364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 318464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 318564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 318664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Meta Data Emission 318764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 318864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 318964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 319064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteImplementations - This routine rewrites all method implementations 319164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// and emits meta-data. 319264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 319364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteImplementations() { 319464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int ClsDefCount = ClassImplementation.size(); 319564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int CatDefCount = CategoryImplementation.size(); 319664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 319764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite implemented methods 319864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) 319964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteImplementationDecl(ClassImplementation[i]); 320064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 320164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < CatDefCount; i++) 320264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteImplementationDecl(CategoryImplementation[i]); 320364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 320464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 320564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteByRefString(std::string &ResultStr, 320664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Name, 320764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD, bool def) { 320864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BlockByRefDeclNo.count(VD) && 320964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "RewriteByRefString: ByRef decl missing"); 321064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (def) 321164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "struct "; 321264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ResultStr += "__Block_byref_" + Name + 321364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "_" + utostr(BlockByRefDeclNo[VD]) ; 321464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 321564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 321664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic bool HasLocalVariableExternalStorage(ValueDecl *VD) { 321764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *Var = dyn_cast<VarDecl>(VD)) 321864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return (Var->isFunctionOrMethodVarDecl() && !Var->hasLocalStorage()); 321964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 322064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 322164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 322264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockFunc(BlockExpr *CE, int i, 322364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, 322464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag) { 322564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *AFT = CE->getFunctionType(); 322664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType RT = AFT->getResultType(); 322764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StructRef = "struct " + Tag; 322864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "static " + RT.getAsString(Context->getPrintingPolicy()) + " __" + 322964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian funcName.str() + "_" + "block_func_" + utostr(i); 323064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 323164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDecl *BD = CE->getBlockDecl(); 323264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 323364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<FunctionNoProtoType>(AFT)) { 323464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // No user-supplied arguments. Still need to pass in a pointer to the 323564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block (to reference imported block decl refs). 323664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef + " *__cself)"; 323764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (BD->param_empty()) { 323864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef + " *__cself)"; 323964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 324064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FT = cast<FunctionProtoType>(AFT); 324164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(FT && "SynthesizeBlockFunc: No function proto"); 324264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += '('; 324364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // first add the implicit argument. 324464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += StructRef + " *__cself, "; 324564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ParamStr; 324664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (BlockDecl::param_iterator AI = BD->param_begin(), 324764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BD->param_end(); AI != E; ++AI) { 324864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (AI != BD->param_begin()) S += ", "; 324964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParamStr = (*AI)->getNameAsString(); 325064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*AI)->getType(); 325164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (convertBlockPointerToFunctionPointer(QT)) 325264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(ParamStr, Context->getPrintingPolicy()); 325364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 325464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(ParamStr, Context->getPrintingPolicy()); 325564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ParamStr; 325664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 325764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FT->isVariadic()) { 325864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BD->param_empty()) S += ", "; 325964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "..."; 326064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 326164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ')'; 326264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 326364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n"; 326464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 326564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Create local declarations to avoid rewriting all closure decl ref exprs. 326664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // First, emit a declaration for all "by ref" decls. 326764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 326864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 326964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 327064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 327164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString; 327264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(TypeString, Name, (*I)); 327364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeString += " *"; 327464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = TypeString + Name; 327564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Name + " = __cself->" + (*I)->getNameAsString() + "; // bound by ref\n"; 327664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 327764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Next, emit a declaration for all "by copy" declarations. 327864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 327964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 328064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 328164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle nested closure invocation. For example: 328264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 328364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^myImportedClosure)(void); 328464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedClosure = ^(void) { setGlobalInt(x + y); }; 328564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 328664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^anotherClosure)(void); 328764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // anotherClosure = ^(void) { 328864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedClosure(); // import and invoke the closure 328964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // }; 329064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 329164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) { 329264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerTypeVariable(S, (*I)); 329364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " = ("; 329464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(S, (*I)->getType()); 329564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ")"; 329664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "__cself->" + (*I)->getNameAsString() + "; // bound by copy\n"; 329764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 329864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 329964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 330064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 330164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) 330264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 330364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(Name, Context->getPrintingPolicy()); 330464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Name + " = __cself->" + 330564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*I)->getNameAsString() + "; // bound by copy\n"; 330664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 330764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 330864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RewrittenStr = RewrittenBlockExprs[CE]; 330964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *cstr = RewrittenStr.c_str(); 331064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*cstr++ != '{') ; 331164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += cstr; 331264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "\n"; 331364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 331464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 331564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 331664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockHelperFuncs(BlockExpr *CE, int i, 331764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef funcName, 331864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag) { 331964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string StructRef = "struct " + Tag; 332064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "static void __"; 332164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 332264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += funcName; 332364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_block_copy_" + utostr(i); 332464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef; 332564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*dst, " + StructRef; 332664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*src) {"; 332764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ValueDecl*,8>::iterator I = ImportedBlockDecls.begin(), 332864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ImportedBlockDecls.end(); I != E; ++I) { 332964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = (*I); 333064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_Block_object_assign((void*)&dst->"; 333164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 333264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", (void*)src->"; 333364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 333464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockByRefDeclsPtrSet.count((*I))) 333564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BYREF) + "/*BLOCK_FIELD_IS_BYREF*/);"; 333664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isBlockPointerType()) 333764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BLOCK) + "/*BLOCK_FIELD_IS_BLOCK*/);"; 333864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 333964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_OBJECT) + "/*BLOCK_FIELD_IS_OBJECT*/);"; 334064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 334164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "}\n"; 334264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 334364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "\nstatic void __"; 334464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += funcName; 334564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_block_dispose_" + utostr(i); 334664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(" + StructRef; 334764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "*src) {"; 334864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ValueDecl*,8>::iterator I = ImportedBlockDecls.begin(), 334964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ImportedBlockDecls.end(); I != E; ++I) { 335064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = (*I); 335164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "_Block_object_dispose((void*)src->"; 335264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += (*I)->getNameAsString(); 335364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockByRefDeclsPtrSet.count((*I))) 335464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BYREF) + "/*BLOCK_FIELD_IS_BYREF*/);"; 335564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isBlockPointerType()) 335664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_BLOCK) + "/*BLOCK_FIELD_IS_BLOCK*/);"; 335764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 335864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", " + utostr(BLOCK_FIELD_IS_OBJECT) + "/*BLOCK_FIELD_IS_OBJECT*/);"; 335964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 336064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "}\n"; 336164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 336264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 336364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 336464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockImpl(BlockExpr *CE, std::string Tag, 336564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Desc) { 336664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "\nstruct " + Tag; 336764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Constructor = " " + Tag; 336864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 336964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n struct __block_impl impl;\n"; 337064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " struct " + Desc; 337164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "* Desc;\n"; 337264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 337364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "(void *fp, "; // Invoke function pointer. 337464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "struct " + Desc; // Descriptor pointer. 337564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " *desc"; 337664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 337764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 337864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by copy" declarations. 337964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 338064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 338164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 338264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FieldName = (*I)->getNameAsString(); 338364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ArgName = "_" + FieldName; 338464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle nested closure invocation. For example: 338564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 338664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^myImportedBlock)(void); 338764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedBlock = ^(void) { setGlobalInt(x + y); }; 338864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 338964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void (^anotherBlock)(void); 339064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // anotherBlock = ^(void) { 339164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // myImportedBlock(); // import and invoke the closure 339264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // }; 339364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 339464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) { 339564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "struct __block_impl *"; 339664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", void *" + ArgName; 339764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 339864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 339964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) 340064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 340164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(FieldName, Context->getPrintingPolicy()); 340264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT.getAsStringInternal(ArgName, Context->getPrintingPolicy()); 340364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", " + ArgName; 340464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 340564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += FieldName + ";\n"; 340664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 340764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by ref" declarations. 340864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 340964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 341064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " "; 341164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FieldName = (*I)->getNameAsString(); 341264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ArgName = "_" + FieldName; 341364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 341464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeString; 341564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(TypeString, FieldName, (*I)); 341664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeString += " *"; 341764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldName = TypeString + FieldName; 341864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgName = TypeString + ArgName; 341964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", " + ArgName; 342064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 342164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += FieldName + "; // by ref\n"; 342264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 342364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Finish writing the constructor. 342464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", int flags=0)"; 342564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize all "by copy" arguments. 342664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool firsTime = true; 342764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 342864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 342964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 343064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (firsTime) { 343164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " : "; 343264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian firsTime = false; 343364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 343464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 343564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", "; 343664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType((*I)->getType())) 343764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "((struct __block_impl *)_" + Name + ")"; 343864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 343964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "(_" + Name + ")"; 344064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 344164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize all "by ref" arguments. 344264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 344364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 344464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name = (*I)->getNameAsString(); 344564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (firsTime) { 344664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " : "; 344764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian firsTime = false; 344864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 344964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 345064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", "; 345164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += Name + "(_" + Name + "->__forwarding)"; 345264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 345364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 345464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " {\n"; 345564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl) 345664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteGlobalBlock;\n"; 345764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 345864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteStackBlock;\n"; 345964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.Flags = flags;\n impl.FuncPtr = fp;\n"; 346064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 346164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " Desc = desc;\n"; 346264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 346364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Finish writing the constructor. 346464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += ", int flags=0) {\n"; 346564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl) 346664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteGlobalBlock;\n"; 346764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 346864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.isa = &_NSConcreteStackBlock;\n"; 346964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " impl.Flags = flags;\n impl.FuncPtr = fp;\n"; 347064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " Desc = desc;\n"; 347164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 347264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += " "; 347364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Constructor += "}\n"; 347464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += Constructor; 347564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "};\n"; 347664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 347764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 347864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 347964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeBlockDescriptor(std::string DescTag, 348064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag, int i, 348164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName, 348264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned hasCopy) { 348364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S = "\nstatic struct " + DescTag; 348464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 348564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " {\n unsigned long reserved;\n"; 348664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " unsigned long Block_size;\n"; 348764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasCopy) { 348864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " void (*copy)(struct "; 348964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*, struct "; 349064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*);\n"; 349164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 349264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " void (*dispose)(struct "; 349364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag; S += "*);\n"; 349464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 349564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "} "; 349664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 349764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += DescTag + "_DATA = { 0, sizeof(struct "; 349864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ImplTag + ")"; 349964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (hasCopy) { 350064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", __" + FunName.str() + "_block_copy_" + utostr(i); 350164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", __" + FunName.str() + "_block_dispose_" + utostr(i); 350264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 350364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "};\n"; 350464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 350564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 350664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 350764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::SynthesizeBlockLiterals(SourceLocation FunLocStart, 350864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FunName) { 350964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert declaration for the function in which block literal is used. 351064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDeclToDeclareForBlock && !Blocks.empty()) 351164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockLiteralFunctionDecl(CurFunctionDeclToDeclareForBlock); 351264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool RewriteSC = (GlobalVarDecl && 351364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !Blocks.empty() && 351464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl->getStorageClass() == SC_Static && 351564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl->getType().getCVRQualifiers()); 351664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RewriteSC) { 351764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SC(" void __"); 351864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += GlobalVarDecl->getNameAsString(); 351964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "() {}"; 352064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, SC); 352164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 352264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 352364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert closures that were part of the function. 352464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0, count=0; i < Blocks.size(); i++) { 352564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CollectBlockDeclRefInfo(Blocks[i]); 352664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to copy-in the inner copied-in variables not actually used in this 352764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block. 352864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int j = 0; j < InnerDeclRefsCount[i]; j++) { 352964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefExpr *Exp = InnerDeclRefs[count++]; 353064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = Exp->getDecl(); 353164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 353264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Exp->isByRef() && !BlockByCopyDeclsPtrSet.count(VD)) { 353364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.insert(VD); 353464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.push_back(VD); 353564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 353664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Exp->isByRef() && !BlockByRefDeclsPtrSet.count(VD)) { 353764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(VD); 353864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(VD); 353964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 354064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // imported objects in the inner blocks not used in the outer 354164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // blocks must be copied/disposed in the outer block as well. 354264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Exp->isByRef() || 354364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD->getType()->isObjCObjectPointerType() || 354464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD->getType()->isBlockPointerType()) 354564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(VD); 354664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 354764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 354864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ImplTag = "__" + FunName.str() + "_block_impl_" + utostr(i); 354964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string DescTag = "__" + FunName.str() + "_block_desc_" + utostr(i); 355064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 355164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string CI = SynthesizeBlockImpl(Blocks[i], ImplTag, DescTag); 355264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 355364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, CI); 355464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 355564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string CF = SynthesizeBlockFunc(Blocks[i], i, FunName, ImplTag); 355664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 355764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, CF); 355864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 355964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImportedBlockDecls.size()) { 356064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string HF = SynthesizeBlockHelperFuncs(Blocks[i], i, FunName, ImplTag); 356164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, HF); 356264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 356364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string BD = SynthesizeBlockDescriptor(DescTag, ImplTag, i, FunName, 356464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.size() > 0); 356564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, BD); 356664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 356764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.clear(); 356864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.clear(); 356964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.clear(); 357064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.clear(); 357164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.clear(); 357264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.clear(); 357364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 357464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RewriteSC) { 357564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Must insert any 'const/volatile/static here. Since it has been 357664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // removed as result of rewriting of block literals. 357764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SC; 357864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getStorageClass() == SC_Static) 357964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC = "static "; 358064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isConstQualified()) 358164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "const "; 358264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isVolatileQualified()) 358364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "volatile "; 358464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (GlobalVarDecl->getType().isRestrictQualified()) 358564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC += "restrict "; 358664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, SC); 358764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 358864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 358964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Blocks.clear(); 359064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefsCount.clear(); 359164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.clear(); 359264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewrittenBlockExprs.clear(); 359364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 359464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 359564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InsertBlockLiteralsWithinFunction(FunctionDecl *FD) { 359664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart = FD->getTypeSpecStartLoc(); 359764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef FuncName = FD->getName(); 359864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 359964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(FunLocStart, FuncName); 360064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 360164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 360264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void BuildUniqueMethodName(std::string &Name, 360364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MD) { 360464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *IFace = MD->getClassInterface(); 360564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = IFace->getName(); 360664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name += "__" + MD->getSelector().getAsString(); 360764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Convert colons to underscores. 360864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string::size_type loc = 0; 360964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while ((loc = Name.find(":", loc)) != std::string::npos) 361064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name.replace(loc, 1, "_"); 361164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 361264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 361364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::InsertBlockLiteralsWithinMethod(ObjCMethodDecl *MD) { 361464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //fprintf(stderr,"In InsertBlockLiteralsWitinMethod\n"); 361564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //SourceLocation FunLocStart = MD->getLocStart(); 361664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart = MD->getLocStart(); 361764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FuncName; 361864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BuildUniqueMethodName(FuncName, MD); 361964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(FunLocStart, FuncName); 362064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 362164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 362264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetBlockDeclRefExprs(Stmt *S) { 362364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 362464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 362564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *CBE = dyn_cast<BlockExpr>(*CI)) 362664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(CBE->getBody()); 362764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 362864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(*CI); 362964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 363064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 363164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefExpr *CDRE = dyn_cast<BlockDeclRefExpr>(S)) { 363264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Handle enums. 363364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isa<FunctionDecl>(CDRE->getDecl())) 363464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(CDRE); 363564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 363664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) 363764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(DRE->getDecl())) { 363864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefExpr *BDRE = 363964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context)BlockDeclRefExpr(cast<VarDecl>(DRE->getDecl()), 364064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DRE->getType(), 364164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, DRE->getLocation(), false); 364264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(BDRE); 364364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 364464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 364564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 364664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 364764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 364864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetInnerBlockDeclRefExprs(Stmt *S, 364964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockDeclRefExpr *, 8> &InnerBlockDeclRefs, 365064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> &InnerContexts) { 365164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 365264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 365364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *CBE = dyn_cast<BlockExpr>(*CI)) { 365464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts.insert(cast<DeclContext>(CBE->getBlockDecl())); 365564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(CBE->getBody(), 365664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, 365764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts); 365864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 365964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 366064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(*CI, 366164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, 366264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts); 366364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 366464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 366564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 366664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefExpr *CDRE = dyn_cast<BlockDeclRefExpr>(S)) { 366764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isa<FunctionDecl>(CDRE->getDecl()) && 366864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian !InnerContexts.count(CDRE->getDecl()->getDeclContext())) 366964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs.push_back(CDRE); 367064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 367164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) { 367264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *Var = dyn_cast<VarDecl>(DRE->getDecl())) 367364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Var->isFunctionOrMethodVarDecl()) 367464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedLocalExternalDecls.insert(Var); 367564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 367664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 367764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 367864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 367964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 368064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// convertFunctionTypeOfBlocks - This routine converts a function type 368164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// whose result type may be a block pointer or whose argument type(s) 368264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// might be block pointers to an equivalent function type replacing 368364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// all block pointers to function pointers. 368464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianQualType RewriteModernObjC::convertFunctionTypeOfBlocks(const FunctionType *FT) { 368564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP = dyn_cast<FunctionProtoType>(FT); 368664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FTP will be null for closures that don't take arguments. 368764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate a funky cast. 368864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 368964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Res = FT->getResultType(); 369064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool HasBlockType = convertBlockPointerToFunctionPointer(Res); 369164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 369264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 369364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 369464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I && (I != E); ++I) { 369564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = *I; 369664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 369764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (convertBlockPointerToFunctionPointer(t)) 369864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian HasBlockType = true; 369964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 370064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 370164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 370264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FuncType; 370364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Does this work if block takes no argument but has a return type 370464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // which is of block type? 370564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasBlockType) 370664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncType = getSimpleFunctionType(Res, &ArgTypes[0], ArgTypes.size()); 370764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else FuncType = QualType(FT, 0); 370864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return FuncType; 370964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 371064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 371164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthesizeBlockCall(CallExpr *Exp, const Expr *BlockExp) { 371264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Navigate to relevant type information. 371364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *CPT = 0; 371464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 371564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(BlockExp)) { 371664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = DRE->getType()->getAs<BlockPointerType>(); 371764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const BlockDeclRefExpr *CDRE = 371864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<BlockDeclRefExpr>(BlockExp)) { 371964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = CDRE->getType()->getAs<BlockPointerType>(); 372064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const MemberExpr *MExpr = dyn_cast<MemberExpr>(BlockExp)) { 372164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = MExpr->getType()->getAs<BlockPointerType>(); 372264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 372364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ParenExpr *PRE = dyn_cast<ParenExpr>(BlockExp)) { 372464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return SynthesizeBlockCall(Exp, PRE->getSubExpr()); 372564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 372664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ImplicitCastExpr *IEXPR = dyn_cast<ImplicitCastExpr>(BlockExp)) 372764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = IEXPR->getType()->getAs<BlockPointerType>(); 372864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (const ConditionalOperator *CEXPR = 372964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ConditionalOperator>(BlockExp)) { 373064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *LHSExp = CEXPR->getLHS(); 373164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *LHSStmt = SynthesizeBlockCall(Exp, LHSExp); 373264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *RHSExp = CEXPR->getRHS(); 373364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *RHSStmt = SynthesizeBlockCall(Exp, RHSExp); 373464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *CONDExp = CEXPR->getCond(); 373564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ConditionalOperator *CondExpr = 373664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) ConditionalOperator(CONDExp, 373764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), cast<Expr>(LHSStmt), 373864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), cast<Expr>(RHSStmt), 373964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getType(), VK_RValue, OK_Ordinary); 374064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CondExpr; 374164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const ObjCIvarRefExpr *IRE = dyn_cast<ObjCIvarRefExpr>(BlockExp)) { 374264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = IRE->getType()->getAs<BlockPointerType>(); 374364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (const PseudoObjectExpr *POE 374464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = dyn_cast<PseudoObjectExpr>(BlockExp)) { 374564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CPT = POE->getType()->castAs<BlockPointerType>(); 374664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 374764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(1 && "RewriteBlockClass: Bad type"); 374864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 374964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CPT && "RewriteBlockClass: Bad type"); 375064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionType *FT = CPT->getPointeeType()->getAs<FunctionType>(); 375164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(FT && "RewriteBlockClass: Bad type"); 375264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP = dyn_cast<FunctionProtoType>(FT); 375364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FTP will be null for closures that don't take arguments. 375464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 375564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 375664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 375764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__block_impl")); 375864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PtrBlock = Context->getPointerType(Context->getTagDeclType(RD)); 375964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 376064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Generate a funky cast. 376164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<QualType, 8> ArgTypes; 376264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 376364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Push the block argument type. 376464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(PtrBlock); 376564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 376664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 376764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I && (I != E); ++I) { 376864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType t = *I; 376964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Make sure we convert "t (^)(...)" to "t (*)(...)". 377064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!convertBlockPointerToFunctionPointer(t)) 377164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian convertToUnqualifiedObjCType(t); 377264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ArgTypes.push_back(t); 377364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 377464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 377564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now do the pointer to function cast. 377664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType PtrToFuncCastType 377764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian = getSimpleFunctionType(Exp->getType(), &ArgTypes[0], ArgTypes.size()); 377864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 377964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PtrToFuncCastType = Context->getPointerType(PtrToFuncCastType); 378064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 378164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *BlkCast = NoTypeInfoCStyleCastExpr(Context, PtrBlock, 378264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 378364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const_cast<Expr*>(BlockExp)); 378464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 378564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 378664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkCast); 378764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //PE->dump(); 378864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 378964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = FieldDecl::Create(*Context, 0, SourceLocation(), 379064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 379164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("FuncPtr"), 379264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 379364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 379464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 379564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MemberExpr *ME = new (Context) MemberExpr(PE, true, FD, SourceLocation(), 379664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType(), VK_LValue, 379764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary); 379864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 379964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 380064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *FunkCast = NoTypeInfoCStyleCastExpr(Context, PtrToFuncCastType, 380164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, ME); 380264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), FunkCast); 380364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 380464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 8> BlkExprs; 380564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add the implicit argument. 380664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.push_back(BlkCast); 380764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add the user arguments. 380864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (CallExpr::arg_iterator I = Exp->arg_begin(), 380964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = Exp->arg_end(); I != E; ++I) { 381064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.push_back(*I); 381164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 381264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CallExpr *CE = new (Context) CallExpr(*Context, PE, &BlkExprs[0], 381364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlkExprs.size(), 381464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp->getType(), VK_RValue, 381564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 381664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CE; 381764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 381864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 381964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// We need to return the rewritten expression to handle cases where the 382064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// BlockDeclRefExpr is embedded in another expression being rewritten. 382164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// For example: 382264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 382364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// int main() { 382464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// __block Foo *f; 382564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// __block int i; 382664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 382764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// void (^myblock)() = ^() { 382864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// [f test]; // f is a BlockDeclRefExpr embedded in a message (which is being rewritten). 382964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// i = 77; 383064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// }; 383164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//} 383264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteBlockDeclRefExpr(Expr *DeclRefExp) { 383364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the byref variable into BYREFVAR->__forwarding->BYREFVAR 383464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for each DeclRefExp where BYREFVAR is name of the variable. 383564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD; 383664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isArrow = true; 383764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefExpr *BDRE = dyn_cast<BlockDeclRefExpr>(DeclRefExp)) 383864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD = BDRE->getDecl(); 383964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 384064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VD = cast<DeclRefExpr>(DeclRefExp)->getDecl(); 384164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isArrow = false; 384264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 384364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 384464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = FieldDecl::Create(*Context, 0, SourceLocation(), 384564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), 384664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get("__forwarding"), 384764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 384864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 384964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 385064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MemberExpr *ME = new (Context) MemberExpr(DeclRefExp, isArrow, 385164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD, SourceLocation(), 385264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType(), VK_LValue, 385364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary); 385464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 385564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef Name = VD->getName(); 385664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = FieldDecl::Create(*Context, 0, SourceLocation(), SourceLocation(), 385764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get(Name), 385864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 385964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*BitWidth=*/0, /*Mutable=*/true, 386064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /*HasInit=*/false); 386164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ME = new (Context) MemberExpr(ME, true, FD, SourceLocation(), 386264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExp->getType(), VK_LValue, OK_Ordinary); 386364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 386464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 386564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 386664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need parens to enforce precedence. 386764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(DeclRefExp->getExprLoc(), 386864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExp->getExprLoc(), 386964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ME); 387064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(DeclRefExp, PE); 387164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return PE; 387264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 387364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 387464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Rewrites the imported local variable V with external storage 387564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// (static, extern, etc.) as *V 387664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// 387764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteLocalVariableExternalStorage(DeclRefExpr *DRE) { 387864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = DRE->getDecl(); 387964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *Var = dyn_cast<VarDecl>(VD)) 388064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ImportedLocalExternalDecls.count(Var)) 388164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return DRE; 388264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Exp = new (Context) UnaryOperator(DRE, UO_Deref, DRE->getType(), 388364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, OK_Ordinary, 388464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DRE->getLocation()); 388564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need parens to enforce precedence. 388664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(SourceLocation(), SourceLocation(), 388764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp); 388864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(DRE, PE); 388964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return PE; 389064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 389164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 389264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteCastExpr(CStyleCastExpr *CE) { 389364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocStart = CE->getLParenLoc(); 389464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation LocEnd = CE->getRParenLoc(); 389564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 389664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to avoid trying to rewrite synthesized casts. 389764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LocStart.isInvalid()) 389864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 389964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to avoid trying to rewrite casts contained in macros. 390064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Rewriter::isRewritable(LocStart) || !Rewriter::isRewritable(LocEnd)) 390164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 390264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 390364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(LocStart); 390464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(LocEnd); 390564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = CE->getType(); 390664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const Type* TypePtr = QT->getAs<Type>(); 390764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<TypeOfExprType>(TypePtr)) { 390864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const TypeOfExprType *TypeOfExprTypePtr = cast<TypeOfExprType>(TypePtr); 390964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = TypeOfExprTypePtr->getUnderlyingExpr()->getType(); 391064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TypeAsString = "("; 391164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerType(TypeAsString, QT); 391264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian TypeAsString += ")"; 391364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, endBuf-startBuf+1, TypeAsString); 391464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 391564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 391664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // advance the location to startArgList. 391764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = startBuf; 391864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 391964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr++ && (argPtr < endBuf)) { 392064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 392164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 392264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*'. 392364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LocStart = LocStart.getLocWithOffset(argPtr-startBuf); 392464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(LocStart, 1, "*"); 392564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 392664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 392764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 392864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 392964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 393064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 393164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerFunctionArgs(FunctionDecl *FD) { 393264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = FD->getLocation(); 393364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned parenCount = 0; 393464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 393564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We have 1 or more arguments that have closure pointers. 393664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 393764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startArgList = strchr(startBuf, '('); 393864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 393964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*startArgList == '(') && "Rewriter fuzzy parser confused"); 394064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 394164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount++; 394264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // advance the location to startArgList. 394364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = DeclLoc.getLocWithOffset(startArgList-startBuf); 394464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((DeclLoc.isValid()) && "Invalid DeclLoc"); 394564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 394664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = startArgList; 394764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 394864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr++ && parenCount) { 394964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 395064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '^': 395164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*'. 395264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = DeclLoc.getLocWithOffset(argPtr-startArgList); 395364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, 1, "*"); 395464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 395564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': 395664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount++; 395764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 395864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': 395964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian parenCount--; 396064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 396164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 396264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 396364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 396464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 396564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 396664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::PointerTypeTakesAnyBlockArguments(QualType QT) { 396764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP; 396864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = QT->getAs<PointerType>(); 396964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT) { 397064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = PT->getPointeeType()->getAs<FunctionProtoType>(); 397164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 397264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = QT->getAs<BlockPointerType>(); 397364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BPT && "BlockPointerTypeTakeAnyBlockArguments(): not a block pointer type"); 397464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = BPT->getPointeeType()->getAs<FunctionProtoType>(); 397564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 397664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 397764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 397864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I != E; ++I) 397964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(*I)) 398064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 398164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 398264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 398364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 398464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 398564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::PointerTypeTakesAnyObjCQualifiedType(QualType QT) { 398664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const FunctionProtoType *FTP; 398764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const PointerType *PT = QT->getAs<PointerType>(); 398864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PT) { 398964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = PT->getPointeeType()->getAs<FunctionProtoType>(); 399064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 399164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockPointerType *BPT = QT->getAs<BlockPointerType>(); 399264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(BPT && "BlockPointerTypeTakeAnyBlockArguments(): not a block pointer type"); 399364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FTP = BPT->getPointeeType()->getAs<FunctionProtoType>(); 399464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 399564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FTP) { 399664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (FunctionProtoType::arg_type_iterator I = FTP->arg_type_begin(), 399764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = FTP->arg_type_end(); I != E; ++I) { 399864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*I)->isObjCQualifiedIdType()) 399964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 400064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*I)->isObjCObjectPointerType() && 400164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (*I)->getPointeeType()->isObjCQualifiedInterfaceType()) 400264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return true; 400364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 400464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 400564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 400664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 400764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 400864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 400964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::GetExtentOfArgList(const char *Name, const char *&LParen, 401064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *&RParen) { 401164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argPtr = strchr(Name, '('); 401264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*argPtr == '(') && "Rewriter fuzzy parser confused"); 401364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 401464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian LParen = argPtr; // output the start. 401564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argPtr++; // skip past the left paren. 401664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned parenCount = 1; 401764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 401864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argPtr && parenCount) { 401964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (*argPtr) { 402064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case '(': parenCount++; break; 402164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case ')': parenCount--; break; 402264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: break; 402364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 402464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (parenCount) argPtr++; 402564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 402664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*argPtr == ')') && "Rewriter fuzzy parser confused"); 402764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RParen = argPtr; // output the end 402864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 402964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 403064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteBlockPointerDecl(NamedDecl *ND) { 403164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) { 403264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerFunctionArgs(FD); 403364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 403464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 403564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle Variables and Typedefs. 403664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getLocation(); 403764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType DeclT; 403864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(ND)) 403964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = VD->getType(); 404064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TypedefNameDecl *TDD = dyn_cast<TypedefNameDecl>(ND)) 404164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = TDD->getUnderlyingType(); 404264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (FieldDecl *FD = dyn_cast<FieldDecl>(ND)) 404364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclT = FD->getType(); 404464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 404564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("RewriteBlockPointerDecl(): Decl type not yet handled"); 404664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 404764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 404864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = startBuf; 404964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scan backward (from the decl location) for the end of the previous decl. 405064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != '^' && *startBuf != ';' && startBuf != MainFileStart) 405164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf--; 405264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation Start = DeclLoc.getLocWithOffset(startBuf-endBuf); 405364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string buf; 405464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned OrigLength=0; 405564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // *startBuf != '^' if we are dealing with a pointer to function that 405664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // may take block argument types (which will be handled below). 405764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*startBuf == '^') { 405864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*', computing a negative offset. 405964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf = '*'; 406064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 406164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 406264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 406364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*startBuf != ')') { 406464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *startBuf; 406564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf++; 406664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 406764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 406864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ')'; 406964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 407064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 407164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PointerTypeTakesAnyBlockArguments(DeclT) || 407264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PointerTypeTakesAnyObjCQualifiedType(DeclT)) { 407364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the '^' with '*' for arguments. 407464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace id<P> with id/*<>*/ 407564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = ND->getLocation(); 407664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startBuf = SM->getCharacterData(DeclLoc); 407764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *argListBegin, *argListEnd; 407864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetExtentOfArgList(startBuf, argListBegin, argListEnd); 407964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (argListBegin < argListEnd) { 408064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*argListBegin == '^') 408164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += '*'; 408264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (*argListBegin == '<') { 408364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "/*"; 408464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin++; 408564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++;; 408664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (*argListBegin != '>') { 408764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin++; 408864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 408964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 409064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin; 409164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += "*/"; 409264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 409364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 409464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += *argListBegin; 409564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian argListBegin++; 409664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 409764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 409864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian buf += ')'; 409964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigLength++; 410064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 410164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(Start, OrigLength, buf); 410264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 410364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 410464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 410564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 410664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 410764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SynthesizeByrefCopyDestroyHelper - This routine synthesizes: 410864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void __Block_byref_id_object_copy(struct Block_byref_id_object *dst, 410964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct Block_byref_id_object *src) { 411064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_assign (&_dest->object, _src->object, 411164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_OBJECT 411264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // object 411364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_assign(&_dest->object, _src->object, 411464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_BLOCK 411564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // block 411664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 411764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// And: 411864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void __Block_byref_id_object_dispose(struct Block_byref_id_object *_src) { 411964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_dispose(_src->object, 412064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_OBJECT 412164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // object 412264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// _Block_object_dispose(_src->object, 412364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// BLOCK_BYREF_CALLER | BLOCK_FIELD_IS_BLOCK 412464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// [|BLOCK_FIELD_IS_WEAK]) // block 412564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 412664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 412764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstd::string RewriteModernObjC::SynthesizeByrefCopyDestroyHelper(VarDecl *VD, 412864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag) { 412964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string S; 413064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CopyDestroyCache.count(flag)) 413164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 413264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CopyDestroyCache.insert(flag); 413364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S = "static void __Block_byref_id_object_copy_"; 413464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 413564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(void *dst, void *src) {\n"; 413664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 413764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // offset into the object pointer is computed as: 413864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void * + void* + int + int + void* + void * 413964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 414064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 414164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned VoidPtrSize = 414264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->VoidPtrTy)); 414364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 414464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned offset = (VoidPtrSize*4 + IntSize + IntSize)/Context->getCharWidth(); 414564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " _Block_object_assign((char*)dst + "; 414664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 414764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ", *(void * *) ((char*)src + "; 414864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 414964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "), "; 415064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 415164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ");\n}\n"; 415264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 415364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "static void __Block_byref_id_object_dispose_"; 415464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 415564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "(void *src) {\n"; 415664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += " _Block_object_dispose(*(void * *) ((char*)src + "; 415764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(offset); 415864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += "), "; 415964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += utostr(flag); 416064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian S += ");\n}\n"; 416164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 416264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 416364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 416464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteByRefVar - For each __block typex ND variable this routine transforms 416564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// the declaration into: 416664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND { 416764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__isa; // NULL for everything except __weak pointers 416864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND *__forwarding; 416964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// int32_t __flags; 417064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// int32_t __size; 417164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__Block_byref_id_object_copy; // If variable is __block ObjC object 417264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// void *__Block_byref_id_object_dispose; // If variable is __block ObjC object 417364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// typex ND; 417464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// }; 417564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 417664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// It then replaces declaration of ND variable with: 417764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct __Block_byref_ND ND = {__isa=0B, __forwarding=&ND, __flags=some_flag, 417864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// __size=sizeof(struct __Block_byref_ND), 417964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ND=initializer-if-any}; 418064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 418164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// 418264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteByRefVar(VarDecl *ND) { 418364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert declaration for the function in which block literal is 418464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // used. 418564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDeclToDeclareForBlock) 418664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockLiteralFunctionDecl(CurFunctionDeclToDeclareForBlock); 418764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag = 0; 418864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int isa = 0; 418964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation DeclLoc = ND->getTypeSpecStartLoc(); 419064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclLoc.isInvalid()) 419164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If type location is missing, it is because of missing type (a warning). 419264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Use variable's location which is good for this case. 419364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclLoc = ND->getLocation(); 419464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(DeclLoc); 419564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation X = ND->getLocEnd(); 419664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian X = SM->getExpansionLoc(X); 419764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(X); 419864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 419964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ByrefType; 420064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND, true); 420164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " {\n"; 420264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void *__isa;\n"; 420364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 420464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " *__forwarding;\n"; 420564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " int __flags;\n"; 420664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " int __size;\n"; 420764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add void *__Block_byref_id_object_copy; 420864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // void *__Block_byref_id_object_dispose; if needed. 420964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Ty = ND->getType(); 421064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool HasCopyAndDispose = Context->BlockRequiresCopying(Ty); 421164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 421264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void (*__Block_byref_id_object_copy)(void*, void*);\n"; 421364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " void (*__Block_byref_id_object_dispose)(void*);\n"; 421464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 421564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 421664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType T = Ty; 421764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian (void)convertBlockPointerToFunctionPointer(T); 421864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian T.getAsStringInternal(Name, Context->getPrintingPolicy()); 421964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 422064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name + ";\n"; 422164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "};\n"; 422264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Insert this type in global scope. It is needed by helper function. 422364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation FunLocStart; 422464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDef) 422564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunLocStart = CurFunctionDef->getTypeSpecStartLoc(); 422664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 422764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(CurMethodDef && "RewriteByRefVar - CurMethodDef is null"); 422864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunLocStart = CurMethodDef->getLocStart(); 422964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 423064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, ByrefType); 423164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty.isObjCGCWeak()) { 423264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_WEAK; 423364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa = 1; 423464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 423564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 423664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 423764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag = BLOCK_BYREF_CALLER; 423864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType Ty = ND->getType(); 423964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME. Handle __weak variable (BLOCK_FIELD_IS_WEAK) as well. 424064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty->isBlockPointerType()) 424164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_BLOCK; 424264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 424364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flag |= BLOCK_FIELD_IS_OBJECT; 424464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string HF = SynthesizeByrefCopyDestroyHelper(ND, flag); 424564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!HF.empty()) 424664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(FunLocStart, HF); 424764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 424864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 424964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // struct __Block_byref_ND ND = 425064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // {0, &ND, some_flag, __size=sizeof(struct __Block_byref_ND), 425164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // initializer-if-any}; 425264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool hasInit = (ND->getInit() != 0); 425364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned flags = 0; 425464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) 425564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian flags |= BLOCK_HAS_COPY_DISPOSE; 425664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Name = ND->getNameAsString(); 425764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType.clear(); 425864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 425964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ForwardingCastType("("); 426064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ForwardingCastType += ByrefType + " *)"; 426164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!hasInit) { 426264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name + " = {(void*)"; 426364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(isa); 426464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "," + ForwardingCastType + "&" + Name + ", "; 426564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flags); 426664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 426764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "sizeof("; 426864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 426964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ")"; 427064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 427164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_copy_"; 427264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 427364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_dispose_"; 427464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 427564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 427664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "};\n"; 427764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned nameSize = Name.size(); 427864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for block or function pointer declaration. Name is aleady 427964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // part of the declaration. 428064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Ty->isBlockPointerType() || Ty->isFunctionPointerType()) 428164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian nameSize = 1; 428264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf+nameSize, ByrefType); 428364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 428464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 428564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc; 428664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *E = ND->getInit(); 428764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const CStyleCastExpr *ECE = dyn_cast<CStyleCastExpr>(E)) 428864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = ECE->getLParenLoc(); 428964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 429064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = E->getLocStart(); 429164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc = SM->getExpansionLoc(startLoc); 429264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian endBuf = SM->getCharacterData(startLoc); 429364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " " + Name; 429464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += " = {(void*)"; 429564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(isa); 429664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "," + ForwardingCastType + "&" + Name + ", "; 429764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flags); 429864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 429964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "sizeof("; 430064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(ByrefType, Name, ND); 430164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "), "; 430264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasCopyAndDispose) { 430364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += "__Block_byref_id_object_copy_"; 430464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 430564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", __Block_byref_id_object_dispose_"; 430664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += utostr(flag); 430764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ByrefType += ", "; 430864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 430964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceText(DeclLoc, endBuf-startBuf, ByrefType); 431064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 431164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Complete the newly synthesized compound expression by inserting a right 431264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // curly brace before the end of the declaration. 431364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: This approach avoids rewriting the initializer expression. It 431464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // also assumes there is only one declarator. For example, the following 431564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // isn't currently supported by this routine (in general): 431664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 431764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // double __block BYREFVAR = 1.34, BYREFVAR2 = 1.37; 431864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 431964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startInitializerBuf = SM->getCharacterData(startLoc); 432064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *semiBuf = strchr(startInitializerBuf, ';'); 432164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert((*semiBuf == ';') && "RewriteByRefVar: can't find ';'"); 432264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation semiLoc = 432364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian startLoc.getLocWithOffset(semiBuf-startInitializerBuf); 432464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 432564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(semiLoc, "}"); 432664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 432764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 432864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 432964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 433064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::CollectBlockDeclRefInfo(BlockExpr *Exp) { 433164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add initializers for any closure decl refs. 433264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetBlockDeclRefExprs(Exp->getBody()); 433364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 433464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique all "by copy" declarations. 433564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 433664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BlockDeclRefs[i]->isByRef()) { 433764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BlockByCopyDeclsPtrSet.count(BlockDeclRefs[i]->getDecl())) { 433864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.insert(BlockDeclRefs[i]->getDecl()); 433964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.push_back(BlockDeclRefs[i]->getDecl()); 434064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 434164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 434264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Unique all "by ref" declarations. 434364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 434464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs[i]->isByRef()) { 434564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!BlockByRefDeclsPtrSet.count(BlockDeclRefs[i]->getDecl())) { 434664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(BlockDeclRefs[i]->getDecl()); 434764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(BlockDeclRefs[i]->getDecl()); 434864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 434964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 435064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find any imported blocks...they will need special attention. 435164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < BlockDeclRefs.size(); i++) 435264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs[i]->isByRef() || 435364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs[i]->getType()->isObjCObjectPointerType() || 435464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs[i]->getType()->isBlockPointerType()) 435564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(BlockDeclRefs[i]->getDecl()); 435664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 435764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 435864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 435964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianFunctionDecl *RewriteModernObjC::SynthBlockInitFunctionDecl(StringRef name) { 436064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *ID = &Context->Idents.get(name); 436164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FType = Context->getFunctionNoProtoType(Context->VoidPtrTy); 436264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return FunctionDecl::Create(*Context, TUDecl, SourceLocation(), 436364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), ID, FType, 0, SC_Extern, 436464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_None, false, false); 436564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 436664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 436764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::SynthBlockInitExpr(BlockExpr *Exp, 436864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const SmallVector<BlockDeclRefExpr *, 8> &InnerBlockDeclRefs) { 436964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const BlockDecl *block = Exp->getBlockDecl(); 437064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Blocks.push_back(Exp); 437164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 437264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CollectBlockDeclRefInfo(Exp); 437364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 437464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add inner imported variables now used in current block. 437564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int countOfInnerDecls = 0; 437664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!InnerBlockDeclRefs.empty()) { 437764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < InnerBlockDeclRefs.size(); i++) { 437864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefExpr *Exp = InnerBlockDeclRefs[i]; 437964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = Exp->getDecl(); 438064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Exp->isByRef() && !BlockByCopyDeclsPtrSet.count(VD)) { 438164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // We need to save the copied-in variables in nested 438264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // blocks because it is needed at the end for some of the API generations. 438364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // See SynthesizeBlockLiterals routine. 438464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.push_back(Exp); countOfInnerDecls++; 438564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 438664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.insert(VD); 438764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.push_back(VD); 438864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 438964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Exp->isByRef() && !BlockByRefDeclsPtrSet.count(VD)) { 439064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefs.push_back(Exp); countOfInnerDecls++; 439164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.push_back(Exp); 439264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.insert(VD); 439364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.push_back(VD); 439464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 439564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 439664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find any imported blocks...they will need special attention. 439764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i < InnerBlockDeclRefs.size(); i++) 439864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (InnerBlockDeclRefs[i]->isByRef() || 439964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs[i]->getType()->isObjCObjectPointerType() || 440064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs[i]->getType()->isBlockPointerType()) 440164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.insert(InnerBlockDeclRefs[i]->getDecl()); 440264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 440364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerDeclRefsCount.push_back(countOfInnerDecls); 440464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 440564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FuncName; 440664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 440764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurFunctionDef) 440864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncName = CurFunctionDef->getNameAsString(); 440964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (CurMethodDef) 441064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BuildUniqueMethodName(FuncName, CurMethodDef); 441164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (GlobalVarDecl) 441264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FuncName = std::string(GlobalVarDecl->getNameAsString()); 441364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string BlockNumber = utostr(Blocks.size()-1); 441564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Tag = "__" + FuncName + "_block_impl_" + BlockNumber; 441764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Func = "__" + FuncName + "_block_func_" + BlockNumber; 441864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 441964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get a pointer to the function type so we can cast appropriately. 442064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType BFT = convertFunctionTypeOfBlocks(Exp->getFunctionType()); 442164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType FType = Context->getPointerType(BFT); 442264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 442364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD; 442464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *NewRep; 442564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 442664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Simulate a contructor call... 442764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl(Tag); 442864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *DRE = new (Context) DeclRefExpr(FD, FType, VK_RValue, 442964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 443064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 443164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<Expr*, 4> InitExprs; 443264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 443364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize the block function. 443464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl(Func); 443564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DeclRefExpr *Arg = new (Context) DeclRefExpr(FD, FD->getType(), VK_LValue, 443664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 443764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, Context->VoidPtrTy, 443864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Arg); 443964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(castExpr); 444064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 444164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Initialize the block descriptor. 444264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string DescData = "__" + FuncName + "_block_desc_" + BlockNumber + "_DATA"; 444364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 444464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, 444564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 444664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian &Context->Idents.get(DescData.c_str()), 444764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 0, 444864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SC_Static, SC_None); 444964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UnaryOperator *DescRefExpr = 445064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian new (Context) UnaryOperator(new (Context) DeclRefExpr(NewVD, 445164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->VoidPtrTy, 445264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, 445364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()), 445464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian UO_AddrOf, 445564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Context->VoidPtrTy), 445664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, 445764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 445864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(DescRefExpr); 445964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 446064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add initializers for any closure decl refs. 446164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefs.size()) { 446264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Exp; 446364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by copy" declarations. 446464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByCopyDecls.begin(), 446564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByCopyDecls.end(); I != E; ++I) { 446664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isObjCType((*I)->getType())) { 446764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Conform to ABI ([[obj retain] autorelease]). 446864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 446964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) DeclRefExpr(FD, FD->getType(), VK_LValue, 447064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 447164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) { 447264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 447364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 447464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, QT, VK_RValue, 447564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary, SourceLocation()); 447664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 447764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (isTopLevelBlockPointerType((*I)->getType())) { 447864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 447964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Arg = new (Context) DeclRefExpr(FD, FD->getType(), VK_LValue, 448064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 448164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = NoTypeInfoCStyleCastExpr(Context, Context->VoidPtrTy, 448264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, Arg); 448364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 448464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 448564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) DeclRefExpr(FD, FD->getType(), VK_LValue, 448664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 448764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(*I)) { 448864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType QT = (*I)->getType(); 448964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QT = Context->getPointerType(QT); 449064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, QT, VK_RValue, 449164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OK_Ordinary, SourceLocation()); 449264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 449364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 449464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 449564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(Exp); 449664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 449764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output all "by ref" declarations. 449864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (SmallVector<ValueDecl*,8>::iterator I = BlockByRefDecls.begin(), 449964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = BlockByRefDecls.end(); I != E; ++I) { 450064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *ND = (*I); 450164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Name(ND->getNameAsString()); 450264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RecName; 450364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefString(RecName, Name, ND, true); 450464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *II = &Context->Idents.get(RecName.c_str() 450564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian + sizeof("struct")); 450664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 450764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 450864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian II); 450964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(RD && "SynthBlockInitExpr(): Can't find RecordDecl"); 451064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castT = Context->getPointerType(Context->getTagDeclType(RD)); 451164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 451264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD = SynthBlockInitFunctionDecl((*I)->getName()); 451364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) DeclRefExpr(FD, FD->getType(), VK_LValue, 451464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 451564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool isNestedCapturedVar = false; 451664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (block) 451764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (BlockDecl::capture_const_iterator ci = block->capture_begin(), 451864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ce = block->capture_end(); ci != ce; ++ci) { 451964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const VarDecl *variable = ci->getVariable(); 452064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (variable == ND && ci->isNested()) { 452164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert (ci->isByRef() && 452264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "SynthBlockInitExpr - captured block variable is not byref"); 452364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isNestedCapturedVar = true; 452464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 452564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 452664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 452764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // captured nested byref variable has its address passed. Do not take 452864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // its address again. 452964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!isNestedCapturedVar) 453064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = new (Context) UnaryOperator(Exp, UO_AddrOf, 453164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(Exp->getType()), 453264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 453364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Exp = NoTypeInfoCStyleCastExpr(Context, castT, CK_BitCast, Exp); 453464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(Exp); 453564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 453664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 453764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImportedBlockDecls.size()) { 453864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // generate BLOCK_HAS_COPY_DISPOSE(have helper funcs) | BLOCK_HAS_DESCRIPTOR 453964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int flag = (BLOCK_HAS_COPY_DISPOSE | BLOCK_HAS_DESCRIPTOR); 454064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned IntSize = 454164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static_cast<unsigned>(Context->getTypeSize(Context->IntTy)); 454264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *FlagExp = IntegerLiteral::Create(*Context, llvm::APInt(IntSize, flag), 454364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->IntTy, SourceLocation()); 454464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitExprs.push_back(FlagExp); 454564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 454664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = new (Context) CallExpr(*Context, DRE, &InitExprs[0], InitExprs.size(), 454764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FType, VK_LValue, SourceLocation()); 454864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = new (Context) UnaryOperator(NewRep, UO_AddrOf, 454964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getPointerType(NewRep->getType()), 455064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_RValue, OK_Ordinary, SourceLocation()); 455164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep = NoTypeInfoCStyleCastExpr(Context, FType, CK_BitCast, 455264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian NewRep); 455364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockDeclRefs.clear(); 455464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDecls.clear(); 455564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclsPtrSet.clear(); 455664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDecls.clear(); 455764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByCopyDeclsPtrSet.clear(); 455864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedBlockDecls.clear(); 455964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return NewRep; 456064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 456164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 456264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianbool RewriteModernObjC::IsDeclStmtInForeachHeader(DeclStmt *DS) { 456364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const ObjCForCollectionStmt * CS = 456464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCForCollectionStmt>(Stmts.back())) 456564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return CS->getElement() == DS; 456664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return false; 456764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 456864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 456964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 457064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// Function Body / Expression rewriting 457164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian//===----------------------------------------------------------------------===// 457264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 457364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteFunctionBodyOrGlobalInitializer(Stmt *S) { 457464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) || 457564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(S) || isa<ForStmt>(S)) 457664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.push_back(S); 457764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (isa<ObjCForCollectionStmt>(S)) { 457864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.push_back(S); 457964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCBcLabelNo.push_back(++BcLabelCount); 458064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 458164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 458264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Pseudo-object operations and ivar references need special 458364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // treatment because we're going to recursively rewrite them. 458464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PseudoObjectExpr *PseudoOp = dyn_cast<PseudoObjectExpr>(S)) { 458564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<BinaryOperator>(PseudoOp->getSyntacticForm())) { 458664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewritePropertyOrImplicitSetter(PseudoOp); 458764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 458864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewritePropertyOrImplicitGetter(PseudoOp); 458964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 459064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (ObjCIvarRefExpr *IvarRefExpr = dyn_cast<ObjCIvarRefExpr>(S)) { 459164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCIvarRefExpr(IvarRefExpr); 459264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 459364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 459464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OrigStmtRange = S->getSourceRange(); 459564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 459664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Perform a bottom up rewrite of all children. 459764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (Stmt::child_range CI = S->children(); CI; ++CI) 459864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (*CI) { 459964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *childStmt = (*CI); 460064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *newStmt = RewriteFunctionBodyOrGlobalInitializer(childStmt); 460164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (newStmt) { 460264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *CI = newStmt; 460364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 460464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 460564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 460664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockExpr *BE = dyn_cast<BlockExpr>(S)) { 460764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<BlockDeclRefExpr *, 8> InnerBlockDeclRefs; 460864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::SmallPtrSet<const DeclContext *, 8> InnerContexts; 460964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerContexts.insert(BE->getBlockDecl()); 461064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedLocalExternalDecls.clear(); 461164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GetInnerBlockDeclRefExprs(BE->getBody(), 461264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InnerBlockDeclRefs, InnerContexts); 461364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the block body in place. 461464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *SaveCurrentBody = CurrentBody; 461564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = BE->getBody(); 461664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 461764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block literal on rhs of a property-dot-sytax assignment 461864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // must be replaced by its synthesize ast so getRewrittenText 461964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // works as expected. In this case, what actually ends up on RHS 462064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // is the blockTranscribed which is the helper function for the 462164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // block literal; as in: self.c = ^() {[ace ARR];}; 462264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool saveDisableReplaceStmt = DisableReplaceStmt; 462364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = false; 462464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionBodyOrGlobalInitializer(BE->getBody()); 462564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmt = saveDisableReplaceStmt; 462664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = SaveCurrentBody; 462764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 462864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ImportedLocalExternalDecls.clear(); 462964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Now we snarf the rewritten text and stash it away for later use. 463064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string Str = Rewrite.getRewrittenText(BE->getSourceRange()); 463164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewrittenBlockExprs[BE] = Str; 463264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 463364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *blockTranscribed = SynthBlockInitExpr(BE, InnerBlockDeclRefs); 463464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 463564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //blockTranscribed->dump(); 463664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(S, blockTranscribed); 463764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return blockTranscribed; 463864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 463964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle specific things. 464064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCEncodeExpr *AtEncode = dyn_cast<ObjCEncodeExpr>(S)) 464164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteAtEncode(AtEncode); 464264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 464364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSelectorExpr *AtSelector = dyn_cast<ObjCSelectorExpr>(S)) 464464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteAtSelector(AtSelector); 464564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 464664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCStringLiteral *AtString = dyn_cast<ObjCStringLiteral>(S)) 464764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCStringLiteral(AtString); 464864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 464964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMessageExpr *MessExpr = dyn_cast<ObjCMessageExpr>(S)) { 465064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#if 0 465164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Before we rewrite it, put the original message expression in a comment. 465264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation startLoc = MessExpr->getLocStart(); 465364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation endLoc = MessExpr->getLocEnd(); 465464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 465564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *startBuf = SM->getCharacterData(startLoc); 465664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *endBuf = SM->getCharacterData(endLoc); 465764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 465864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string messString; 465964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString += "// "; 466064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString.append(startBuf, endBuf-startBuf+1); 466164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian messString += "\n"; 466264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 466364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: Missing definition of 466464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // InsertText(clang::SourceLocation, char const*, unsigned int). 466564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // InsertText(startLoc, messString.c_str(), messString.size()); 466664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Tried this, but it didn't work either... 466764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // ReplaceText(startLoc, 0, messString.c_str(), messString.size()); 466864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#endif 466964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteMessageExpr(MessExpr); 467064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 467164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 467264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtTryStmt *StmtTry = dyn_cast<ObjCAtTryStmt>(S)) 467364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCTryStmt(StmtTry); 467464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 467564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtSynchronizedStmt *StmtTry = dyn_cast<ObjCAtSynchronizedStmt>(S)) 467664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCSynchronizedStmt(StmtTry); 467764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 467864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCAtThrowStmt *StmtThrow = dyn_cast<ObjCAtThrowStmt>(S)) 467964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCThrowStmt(StmtThrow); 468064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 468164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolExpr *ProtocolExp = dyn_cast<ObjCProtocolExpr>(S)) 468264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCProtocolExpr(ProtocolExp); 468364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 468464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCForCollectionStmt *StmtForCollection = 468564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCForCollectionStmt>(S)) 468664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteObjCForCollectionStmt(StmtForCollection, 468764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OrigStmtRange.getEnd()); 468864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BreakStmt *StmtBreakStmt = 468964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<BreakStmt>(S)) 469064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteBreakStmt(StmtBreakStmt); 469164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ContinueStmt *StmtContinueStmt = 469264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ContinueStmt>(S)) 469364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteContinueStmt(StmtContinueStmt); 469464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 469564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Need to check for protocol refs (id <P>, Foo <P> *) in variable decls 469664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and cast exprs. 469764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclStmt *DS = dyn_cast<DeclStmt>(S)) { 469864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: What we're doing here is modifying the type-specifier that 469964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // precedes the first Decl. In the future the DeclGroup should have 470064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // a separate type-specifier that we can rewrite. 470164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NOTE: We need to avoid rewriting the DeclStmt if it is within 470264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // the context of an ObjCForCollectionStmt. For example: 470364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NSArray *someArray; 470464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // for (id <FooProtocol> index in someArray) ; 470564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is because RewriteObjCForCollectionStmt() does textual rewriting 470664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and it depends on the original text locations/positions. 470764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Stmts.empty() || !IsDeclStmtInForeachHeader(DS)) 470864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(*DS->decl_begin()); 470964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 471064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Blocks rewrite rules. 471164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (DeclStmt::decl_iterator DI = DS->decl_begin(), DE = DS->decl_end(); 471264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DI != DE; ++DI) { 471364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Decl *SD = *DI; 471464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ValueDecl *ND = dyn_cast<ValueDecl>(SD)) { 471564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(ND->getType())) 471664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(ND); 471764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (ND->getType()->isFunctionPointerType()) 471864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(ND->getType(), ND); 471964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VarDecl *VD = dyn_cast<VarDecl>(SD)) { 472064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->hasAttr<BlocksAttr>()) { 472164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static unsigned uniqueByrefDeclCount = 0; 472264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!BlockByRefDeclNo.count(ND) && 472364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "RewriteFunctionBodyOrGlobalInitializer: Duplicate byref decl"); 472464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BlockByRefDeclNo[ND] = uniqueByrefDeclCount++; 472564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteByRefVar(VD); 472664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 472764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 472864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteTypeOfDecl(VD); 472964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 473064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 473164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (TypedefNameDecl *TD = dyn_cast<TypedefNameDecl>(SD)) { 473264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(TD->getUnderlyingType())) 473364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(TD); 473464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TD->getUnderlyingType()->isFunctionPointerType()) 473564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(TD->getUnderlyingType(), TD); 473664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 473764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 473864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 473964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 474064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(S)) 474164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(CE); 474264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 474364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) || 474464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(S) || isa<ForStmt>(S)) { 474564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!Stmts.empty() && "Statement stack is empty"); 474664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert ((isa<SwitchStmt>(Stmts.back()) || isa<WhileStmt>(Stmts.back()) || 474764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian isa<DoStmt>(Stmts.back()) || isa<ForStmt>(Stmts.back())) 474864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian && "Statement stack mismatch"); 474964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmts.pop_back(); 475064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 475164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Handle blocks rewriting. 475264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BlockDeclRefExpr *BDRE = dyn_cast<BlockDeclRefExpr>(S)) { 475364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BDRE->isByRef()) 475464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteBlockDeclRefExpr(BDRE); 475564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 475664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(S)) { 475764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ValueDecl *VD = DRE->getDecl(); 475864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->hasAttr<BlocksAttr>()) 475964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteBlockDeclRefExpr(DRE); 476064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (HasLocalVariableExternalStorage(VD)) 476164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return RewriteLocalVariableExternalStorage(DRE); 476264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 476364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 476464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CallExpr *CE = dyn_cast<CallExpr>(S)) { 476564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CE->getCallee()->getType()->isBlockPointerType()) { 476664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Stmt *BlockCall = SynthesizeBlockCall(CE, CE->getCallee()); 476764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmt(S, BlockCall); 476864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return BlockCall; 476964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 477064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 477164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(S)) { 477264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 477364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 477464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#if 0 477564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(S)) { 477664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *Replacement = new (Context) CastExpr(ICE->getType(), 477764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ICE->getSubExpr(), 477864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation()); 477964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the new text. 478064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string SStr; 478164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::raw_string_ostream Buf(SStr); 478264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Replacement->printPretty(Buf, *Context); 478364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const std::string &Str = Buf.str(); 478464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 478564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian printf("CAST = %s\n", &Str[0]); 478664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(ICE->getSubExpr()->getLocStart(), &Str[0], Str.size()); 478764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete S; 478864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 478964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 479064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian#endif 479164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Return this stmt unmodified. 479264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return S; 479364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 479464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 479564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteRecordBody(RecordDecl *RD) { 479664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (RecordDecl::field_iterator i = RD->field_begin(), 479764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian e = RD->field_end(); i != e; ++i) { 479864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FieldDecl *FD = *i; 479964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(FD->getType())) 480064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(FD); 480164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->getType()->isObjCQualifiedIdType() || 480264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->getType()->isObjCQualifiedInterfaceType()) 480364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(FD); 480464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 480564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 480664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 480764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// HandleDeclInMainFile - This is called for each top-level decl defined in the 480864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// main file of the input. 480964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleDeclInMainFile(Decl *D) { 481064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian switch (D->getKind()) { 481164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Function: { 481264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FunctionDecl *FD = cast<FunctionDecl>(D); 481364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (FD->isOverloadedOperator()) 481464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 481564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 481664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Since function prototypes don't have ParmDecl's, we check the function 481764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // prototype. This enables us to rewrite function declarations and 481864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // definitions using the same code. 481964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlocksInFunctionProtoType(FD->getType(), FD); 482064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 482164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: If this should support Obj-C++, support CXXTryStmt 482264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CompoundStmt *Body = dyn_cast_or_null<CompoundStmt>(FD->getBody())) { 482364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = FD; 482464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = FD; 482564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = Body; 482664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Body = 482764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast_or_null<CompoundStmt>(RewriteFunctionBodyOrGlobalInitializer(Body)); 482864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FD->setBody(Body); 482964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 483064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 483164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 483264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 483364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 483464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This synthesizes and inserts the block "impl" struct, invoke function, 483564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // and any copy/dispose helper functions. 483664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertBlockLiteralsWithinFunction(FD); 483764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDef = 0; 483864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurFunctionDeclToDeclareForBlock = 0; 483964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 484064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 484164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 484264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCMethod: { 484364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCMethodDecl *MD = cast<ObjCMethodDecl>(D); 484464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CompoundStmt *Body = MD->getCompoundBody()) { 484564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = MD; 484664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = Body; 484764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Body = 484864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian cast_or_null<CompoundStmt>(RewriteFunctionBodyOrGlobalInitializer(Body)); 484964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MD->setBody(Body); 485064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 485164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 485264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 485364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 485464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 485564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertBlockLiteralsWithinMethod(MD); 485664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurMethodDef = 0; 485764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 485864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 485964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 486064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCImplementation: { 486164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCImplementationDecl *CI = cast<ObjCImplementationDecl>(D); 486264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ClassImplementation.push_back(CI); 486364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 486464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 486564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::ObjCCategoryImpl: { 486664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryImplDecl *CI = cast<ObjCCategoryImplDecl>(D); 486764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CategoryImplementation.push_back(CI); 486864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 486964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 487064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Var: { 487164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VarDecl *VD = cast<VarDecl>(D); 487264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCQualifiedInterfaceTypes(VD); 487364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(VD->getType())) 487464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(VD); 487564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (VD->getType()->isFunctionPointerType()) { 487664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(VD->getType(), VD); 487764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getInit()) { 487864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(VD->getInit())) { 487964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 488064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 488164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 488264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else if (VD->getType()->isRecordType()) { 488364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = VD->getType()->getAs<RecordType>()->getDecl(); 488464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RD->isCompleteDefinition()) 488564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteRecordBody(RD); 488664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 488764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (VD->getInit()) { 488864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = VD; 488964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = VD->getInit(); 489064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteFunctionBodyOrGlobalInitializer(VD->getInit()); 489164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CurrentBody = 0; 489264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PropParentMap) { 489364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian delete PropParentMap; 489464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropParentMap = 0; 489564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 489664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SynthesizeBlockLiterals(VD->getTypeSpecStartLoc(), VD->getName()); 489764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian GlobalVarDecl = 0; 489864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 489964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // This is needed for blocks. 490064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(VD->getInit())) { 490164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteCastExpr(CE); 490264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 490564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 490664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::TypeAlias: 490764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Typedef: { 490864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (TypedefNameDecl *TD = dyn_cast<TypedefNameDecl>(D)) { 490964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (isTopLevelBlockPointerType(TD->getUnderlyingType())) 491064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteBlockPointerDecl(TD); 491164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else if (TD->getUnderlyingType()->isFunctionPointerType()) 491264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CheckFunctionPointerDecl(TD->getUnderlyingType(), TD); 491364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 491464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 491564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 491664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::CXXRecord: 491764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian case Decl::Record: { 491864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = cast<RecordDecl>(D); 491964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (RD->isCompleteDefinition()) 492064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteRecordBody(RD); 492164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 492264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 492364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian default: 492464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 492564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 492664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Nothing yet. 492764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 492864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 492964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::HandleTranslationUnit(ASTContext &C) { 493064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Diags.hasErrorOccurred()) 493164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 493264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 493364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteInclude(); 493464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 493564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Here's a great place to add any extra declarations that may be needed. 493664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Write out meta data for each @protocol(<expr>). 493764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ObjCProtocolDecl *,8>::iterator I = ProtocolExprDecls.begin(), 493864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ProtocolExprDecls.end(); I != E; ++I) 4939da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Preamble); 494064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 494164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InsertText(SM->getLocForStartOfFile(MainFileID), Preamble, false); 494264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ClassImplementation.size() || CategoryImplementation.size()) 494364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteImplementations(); 494464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 494564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Get the buffer corresponding to MainFileID. If we haven't changed it, then 494664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // we are done. 494764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (const RewriteBuffer *RewriteBuf = 494864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Rewrite.getRewriteBufferFor(MainFileID)) { 494964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian //printf("Changed:\n"); 495064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *OutFile << std::string(RewriteBuf->begin(), RewriteBuf->end()); 495164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 495264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm::errs() << "No changes\n"; 495364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 495464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 495564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ClassImplementation.size() || CategoryImplementation.size() || 495664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ProtocolExprDecls.size()) { 495764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite Objective-c meta data* 495864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string ResultStr; 495964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteMetaDataIntoBuffer(ResultStr); 496064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Emit metadata. 496164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian *OutFile << ResultStr; 496264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 496364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OutFile->flush(); 496464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 496564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 496664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::Initialize(ASTContext &context) { 496764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InitializeCommon(context); 496864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 496964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // declaring objc_selector outside the parameter list removes a silly 497064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // scope related warning... 497164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IsHeader) 497264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble = "#pragma once\n"; 497364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct objc_selector; struct objc_class;\n"; 497464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __rw_objc_super { struct objc_object *object; "; 497564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct objc_object *superClass; "; 497664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 497764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Add a constructor for creating temporary objects. 497864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__rw_objc_super(struct objc_object *o, struct objc_object *s) " 497964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ": "; 498064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "object(o), superClass(s) {} "; 498164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 498264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 498364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef _REWRITER_typedef_Protocol\n"; 498464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "typedef struct objc_object Protocol;\n"; 498564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define _REWRITER_typedef_Protocol\n"; 498664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 498764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 498864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_DLLIMPORT extern \"C\" __declspec(dllimport)\n"; 498964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_STATICIMPORT extern \"C\"\n"; 499064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else 499164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __OBJC_RW_DLLIMPORT extern\n"; 499264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_msgSend"; 499364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_object *, struct objc_selector *, ...);\n"; 499464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_msgSendSuper"; 499564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_super *, struct objc_selector *, ...);\n"; 499664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object* objc_msgSend_stret"; 499764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_object *, struct objc_selector *, ...);\n"; 499864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object* objc_msgSendSuper_stret"; 499964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_super *, struct objc_selector *, ...);\n"; 500064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT double objc_msgSend_fpret"; 500164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_object *, struct objc_selector *, ...);\n"; 500264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_getClass"; 500364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(const char *);\n"; 500464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_class *class_getSuperclass"; 500564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_class *);\n"; 500664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_getMetaClass"; 500764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(const char *);\n"; 500864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_exception_throw(struct objc_object *);\n"; 500964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_exception_try_enter(void *);\n"; 501064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_exception_try_exit(void *);\n"; 501164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT struct objc_object *objc_exception_extract(void *);\n"; 501264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT int objc_exception_match"; 501364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "(struct objc_class *, struct objc_object *);\n"; 501464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // @synchronized hooks. 501564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_sync_enter(struct objc_object *);\n"; 501664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_sync_exit(struct objc_object *);\n"; 501764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT Protocol *objc_getProtocol(const char *);\n"; 501864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef __FASTENUMERATIONSTATE\n"; 501964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __objcFastEnumerationState {\n\t"; 502064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long state;\n\t"; 502164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "void **itemsPtr;\n\t"; 502264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long *mutationsPtr;\n\t"; 502364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "unsigned long extra[5];\n};\n"; 502464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void objc_enumerationMutation(struct objc_object *);\n"; 502564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __FASTENUMERATIONSTATE\n"; 502664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 502764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef __NSCONSTANTSTRINGIMPL\n"; 502864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __NSConstantStringImpl {\n"; 502964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int *isa;\n"; 503064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int flags;\n"; 503164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " char *str;\n"; 503264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " long length;\n"; 503364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 503464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifdef CF_EXPORT_CONSTANT_STRING\n"; 503564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) int __CFConstantStringClassReference[];\n"; 503664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#else\n"; 503764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT int __CFConstantStringClassReference[];\n"; 503864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 503964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __NSCONSTANTSTRINGIMPL\n"; 504064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 504164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Blocks preamble. 504264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef BLOCK_IMPL\n"; 504364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define BLOCK_IMPL\n"; 504464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "struct __block_impl {\n"; 504564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " void *isa;\n"; 504664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int Flags;\n"; 504764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " int Reserved;\n"; 504864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += " void *FuncPtr;\n"; 504964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "};\n"; 505064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "// Runtime copy/destroy helper functions (from Block_private.h)\n"; 505164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifdef __OBJC_EXPORT_BLOCKS\n"; 505264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) " 505364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "void _Block_object_assign(void *, const void *, const int);\n"; 505464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void _Block_object_dispose(const void *, const int);\n"; 505564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void *_NSConcreteGlobalBlock[32];\n"; 505664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "extern \"C\" __declspec(dllexport) void *_NSConcreteStackBlock[32];\n"; 505764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#else\n"; 505864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void _Block_object_assign(void *, const void *, const int);\n"; 505964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void _Block_object_dispose(const void *, const int);\n"; 506064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void *_NSConcreteGlobalBlock[32];\n"; 506164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "__OBJC_RW_DLLIMPORT void *_NSConcreteStackBlock[32];\n"; 506264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 506364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 506464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 506564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#undef __OBJC_RW_DLLIMPORT\n"; 506664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#undef __OBJC_RW_STATICIMPORT\n"; 506764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#ifndef KEEP_ATTRIBUTES\n"; // We use this for clang tests. 506864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __attribute__(X)\n"; 506964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#endif\n"; 507064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __weak\n"; 507164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 507264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 507364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __block\n"; 507464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "#define __weak\n"; 507564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 507664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // NOTE! Windows uses LLP64 for 64bit mode. So, cast pointer to long long 507764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // as this avoids warning in any 64bit/32bit compilation model. 507864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Preamble += "\n#define __OFFSETOFIVAR__(TYPE, MEMBER) ((long long) &((TYPE *)0)->MEMBER)\n"; 507964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 508064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 508164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteIvarOffsetComputation - This rutine synthesizes computation of 508264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// ivar offset. 508364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteIvarOffsetComputation(ObjCIvarDecl *ivar, 508464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 508564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ivar->isBitField()) { 508664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: The hack below doesn't work for bitfields. For now, we simply 508764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // place all bitfields at offset 0. 508864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "0"; 508964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 509064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "__OFFSETOFIVAR__(struct "; 509164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ivar->getContainingInterface()->getNameAsString(); 509264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) 509364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_IMPL"; 509464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", "; 509564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ivar->getNameAsString(); 509664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ")"; 509764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 509864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 509964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 510064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// WriteModernMetadataDeclarations - Writes out metadata declarations for modern ABI. 510164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_t { 5102da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian/// const char *name; 510364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// char *attributes; 510464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 510564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 510664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_list_t { 510764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t entsize; // sizeof(struct _prop_t) 510864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t count_of_properties; 510964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _prop_t prop_list[count_of_properties]; 511064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 511164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 511264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t; 511364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 511464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_list_t { 511564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// long protocol_count; // Note, this is 32/64 bit 511664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t * protocol_list[protocol_count]; 511764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 511864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 511964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _objc_method { 512064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// SEL _cmd; 512177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian/// const char *method_type; 512264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// char *_imp; 512364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 512464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 512564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _method_list_t { 512664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t entsize; // sizeof(struct _objc_method) 512764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// uint32_t method_count; 512864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _objc_method method_list[method_count]; 512964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 513064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 513164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// struct _protocol_t { 513264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// id isa; // NULL 513364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const char * const protocol_name; 513464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct _protocol_list_t * protocol_list; // super protocols 513564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t * const instance_methods; 513664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t * const class_methods; 513764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t *optionalInstanceMethods; 513864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct method_list_t *optionalClassMethods; 513964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const struct _prop_list_t * properties; 514064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const uint32_t size; // sizeof(struct _protocol_t) 514164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const uint32_t flags; // = 0 514264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// const char ** extendedMethodTypes; 514364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// } 514464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 514542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _ivar_t { 514642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// unsigned long int *offset; // pointer to ivar offset location 5147ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// const char *name; 5148ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// const char *type; 514942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t alignment; 515042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t size; 515142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 515242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 515342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _ivar_list_t { 515442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32 entsize; // sizeof(struct _ivar_t) 515542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32 count; 5156ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian/// struct _ivar_t list[count]; 515742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 515842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 515942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_ro_t { 516042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const flags; 516142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const instanceStart; 516242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const instanceSize; 516342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// uint32_t const reserved; // only when building for 64bit targets 516442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const uint8_t * const ivarLayout; 516542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const char *const name; 516642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _method_list_t * const baseMethods; 516742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _objc_protocol_list *const baseProtocols; 516842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _ivar_list_t *const ivars; 516942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const uint8_t * const weakIvarLayout; 517042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _prop_list_t * const properties; 517142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 517242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 517342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t { 517442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t *isa; 517542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t * const superclass; 517642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// void *cache; 517742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// IMP *vtable; 517842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct class_ro_t *ro; 517942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 518042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 518142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _category_t { 518242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const char * const name; 518342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _class_t *const cls; 518442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _method_list_t * const instance_methods; 518542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _method_list_t * const class_methods; 518642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _protocol_list_t * const protocols; 518742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// const struct _prop_list_t * const properties; 518842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// } 518942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 519042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// MessageRefTy - LLVM for: 519142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _message_ref_t { 519242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// IMP messenger; 519342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SEL name; 519442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// }; 519542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 519642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SuperMessageRefTy - LLVM for: 519742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// struct _super_message_ref_t { 519842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SUPER_IMP messenger; 519942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// SEL name; 520042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian/// }; 520142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 520264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianstatic void WriteModernMetadataDeclarations(std::string &Result) { 520364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool meta_data_declared = false; 520464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (meta_data_declared) 520564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 520664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 520764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _prop_t {\n"; 5208da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tconst char *name;\n"; 5209da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tconst char *attributes;\n"; 521064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 521164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 521264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _protocol_t;\n"; 521364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 521464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_method {\n"; 521564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct objc_selector * _cmd;\n"; 521677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tconst char *method_type;\n"; 521764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tchar *_imp;\n"; 521864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 521964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 522064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _protocol_t {\n"; 522164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tvoid * isa; // NULL\n"; 522264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char * const protocol_name;\n"; 522364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct _protocol_list_t * protocol_list; // super protocols\n"; 522464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t * const instance_methods;\n"; 522564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t * const class_methods;\n"; 522664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t *optionalInstanceMethods;\n"; 522764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct method_list_t *optionalClassMethods;\n"; 522864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst struct _prop_list_t * properties;\n"; 522964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst unsigned int size; // sizeof(struct _protocol_t)\n"; 523064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst unsigned int flags; // = 0\n"; 523164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char ** extendedMethodTypes;\n"; 523264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 523364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 523442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _ivar_t {\n"; 523542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned long int *offset; // pointer to ivar offset location\n"; 5236ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tconst char *name;\n"; 5237ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tconst char *type;\n"; 523842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int alignment;\n"; 523942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int size;\n"; 524042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 524142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 524242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _class_ro_t {\n"; 524342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int const flags;\n"; 524442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int instanceStart;\n"; 524542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int const instanceSize;\n"; 524642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tunsigned int const reserved; // only when building for 64bit targets\n"; 524742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst unsigned char * const ivarLayout;\n"; 524842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst char *const name;\n"; 524942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _method_list_t * const baseMethods;\n"; 525042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _objc_protocol_list *const baseProtocols;\n"; 525142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _ivar_list_t *const ivars;\n"; 525242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst unsigned char *const weakIvarLayout;\n"; 525342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _prop_list_t *const properties;\n"; 525442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 525542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 525642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _class_t {\n"; 525742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tstruct _class_t *isa;\n"; 525842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tstruct _class_t *const superclass;\n"; 525942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tvoid *cache;\n"; 526042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tvoid *vtable;\n"; 526142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tstruct class_ro_t *ro;\n"; 526242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 526342e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 526442e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\nstruct _category_t {\n"; 526542e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst char * const name;\n"; 526642e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tstruct _class_t *const cls;\n"; 526742e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _method_list_t *const instance_methods;\n"; 526842e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _method_list_t *const class_methods;\n"; 526942e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _protocol_list_t *const protocols;\n"; 527042e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "\tconst struct _prop_list_t *const properties;\n"; 527142e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian Result += "};\n"; 527242e9a35ce0676f1ad2adf83ee1f1f4f503596353Fariborz Jahanian 527364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian meta_data_declared = true; 527464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 527564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 5276da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianstatic void Write_protocol_list_t_TypeDecl(std::string &Result, 5277da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian long super_protocol_count) { 5278da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "struct /*_protocol_list_t*/"; Result += " {\n"; 5279da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\tlong protocol_count; // Note, this is 32/64 bit\n"; 5280da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\tstruct _protocol_t *super_protocols["; 5281da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += utostr(super_protocol_count); Result += "];\n"; 5282da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "}"; 5283da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian} 5284da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 528577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanianstatic void Write_method_list_t_TypeDecl(std::string &Result, 528677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian unsigned int method_count) { 528777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "struct /*_method_list_t*/"; Result += " {\n"; 528877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _objc_method)\n"; 528977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tunsigned int method_count;\n"; 529077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\tstruct _objc_method method_list["; 529177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += utostr(method_count); Result += "];\n"; 529277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "}"; 529377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian} 529477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 5295da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanianstatic void Write__prop_list_t_TypeDecl(std::string &Result, 5296da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian unsigned int property_count) { 5297da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "struct /*_prop_list_t*/"; Result += " {\n"; 5298da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _prop_t)\n"; 5299da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tunsigned int count_of_properties;\n"; 5300da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\tstruct _prop_t prop_list["; 5301da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += utostr(property_count); Result += "];\n"; 5302da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "}"; 5303da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian} 5304da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 5305ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanianstatic void Write__ivar_list_t_TypeDecl(std::string &Result, 5306ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian unsigned int ivar_count) { 5307ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "struct /*_ivar_list_t*/"; Result += " {\n"; 5308ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tunsigned int entsize; // sizeof(struct _prop_t)\n"; 5309ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tunsigned int count;\n"; 5310ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\tstruct _ivar_t ivar_list["; 5311ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += utostr(ivar_count); Result += "];\n"; 5312ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "}"; 5313ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian} 5314ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5315da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianstatic void Write_protocol_list_initializer(ASTContext *Context, std::string &Result, 5316da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian ArrayRef<ObjCProtocolDecl *> SuperProtocols, 5317da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian StringRef VarName, 5318da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian StringRef ProtocolName) { 5319da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (SuperProtocols.size() > 0) { 5320da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\nstatic "; 5321da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Write_protocol_list_t_TypeDecl(Result, SuperProtocols.size()); 5322da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += " "; Result += VarName; 5323da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += ProtocolName; 5324da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 5325da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t"; Result += utostr(SuperProtocols.size()); Result += ",\n"; 5326da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (unsigned i = 0, e = SuperProtocols.size(); i < e; i++) { 5327da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian ObjCProtocolDecl *SuperPD = SuperProtocols[i]; 5328da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t&"; Result += "_OBJC_PROTOCOL_"; 5329da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += SuperPD->getNameAsString(); 5330da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (i == e-1) 5331da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\n};\n"; 5332da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian else 5333da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += ",\n"; 5334da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 5335da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 5336da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian} 5337da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 533877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanianstatic void Write_method_list_t_initializer(ASTContext *Context, std::string &Result, 533977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ArrayRef<ObjCMethodDecl *> Methods, 534077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian StringRef VarName, 534177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian StringRef ProtocolName) { 534277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (Methods.size() > 0) { 534377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\nstatic "; 534477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Write_method_list_t_TypeDecl(Result, Methods.size()); 534577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += " "; Result += VarName; 534677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += ProtocolName; 534777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 534877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t"; Result += "sizeof(_objc_method)"; Result += ",\n"; 534977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t"; Result += utostr(Methods.size()); Result += ",\n"; 535077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (unsigned i = 0, e = Methods.size(); i < e; i++) { 535177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = Methods[i]; 535277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (i == 0) 535377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t{{(struct objc_selector *)\""; 535477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian else 535577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\t{(struct objc_selector *)\""; 535677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += (MD)->getSelector().getAsString(); Result += "\""; 535777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += ", "; 535877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::string MethodTypeString; 535977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Context->getObjCEncodingForMethodDecl(MD, MethodTypeString); 536077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "\""; Result += MethodTypeString; Result += "\""; 536177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += ", "; 536277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian // FIXME is _imp always null here? 536377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (i == e-1) 536477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "0}}\n"; 536577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian else 536677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "0},\n"; 536777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 536877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Result += "};\n"; 536977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 537077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian} 537177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 5372da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanianstatic void Write__prop_list_t_initializer(RewriteModernObjC &RewriteObj, 5373da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ASTContext *Context, std::string &Result, 5374da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ArrayRef<ObjCPropertyDecl *> Properties, 5375da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian const Decl *Container, 5376da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian StringRef VarName, 5377da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian StringRef ProtocolName) { 5378da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (Properties.size() > 0) { 5379da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\nstatic "; 5380da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Write__prop_list_t_TypeDecl(Result, Properties.size()); 5381da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += " "; Result += VarName; 5382da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += ProtocolName; 5383da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 5384da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t"; Result += "sizeof(_prop_t)"; Result += ",\n"; 5385da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t"; Result += utostr(Properties.size()); Result += ",\n"; 5386da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian for (unsigned i = 0, e = Properties.size(); i < e; i++) { 5387da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ObjCPropertyDecl *PropDecl = Properties[i]; 5388da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (i == 0) 5389da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t{{\""; 5390da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian else 5391da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\t{\""; 5392da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += PropDecl->getName(); Result += "\","; 5393da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian std::string PropertyTypeString, QuotePropertyTypeString; 5394da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Context->getObjCEncodingForPropertyDecl(PropDecl, Container, PropertyTypeString); 5395da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian RewriteObj.QuoteDoublequotes(PropertyTypeString, QuotePropertyTypeString); 5396da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "\""; Result += QuotePropertyTypeString; Result += "\""; 5397da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian if (i == e-1) 5398da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "}}\n"; 5399da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian else 5400da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "},\n"; 5401da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian } 5402da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Result += "};\n"; 5403da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian } 5404da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian} 5405da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 5406e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanianstatic void Write__extendedMethodTypes_initializer(RewriteModernObjC &RewriteObj, 5407e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ASTContext *Context, std::string &Result, 5408e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ArrayRef<ObjCMethodDecl *> Methods, 5409e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian StringRef VarName, 5410e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian StringRef ProtocolName) { 5411e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (Methods.size() == 0) 5412e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian return; 5413e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 5414e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\nstatic const char *"; 5415e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += VarName; Result += ProtocolName; 5416e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += " [] __attribute__ ((used, section (\"__DATA,__objc_const\"))) = \n"; 5417e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "{\n"; 5418e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = Methods.size(); i < e; i++) { 5419e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian ObjCMethodDecl *MD = Methods[i]; 5420e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian std::string MethodTypeString, QuoteMethodTypeString; 5421e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Context->getObjCEncodingForMethodDecl(MD, MethodTypeString, true); 5422e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian RewriteObj.QuoteDoublequotes(MethodTypeString, QuoteMethodTypeString); 5423e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t\""; Result += QuoteMethodTypeString; Result += "\""; 5424e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (i == e-1) 5425e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\n};\n"; 5426e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian else { 5427e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += ",\n"; 5428e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 5429e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 5430e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian} 5431e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 5432ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanianstatic void Write__ivar_list_t_initializer(RewriteModernObjC &RewriteObj, 5433ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ASTContext *Context, std::string &Result, 5434ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ArrayRef<ObjCIvarDecl *> Ivars, 5435ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian StringRef VarName, 5436ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian StringRef ClassName) { 5437ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (Ivars.size() > 0) { 5438ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\nstatic "; 5439ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Write__ivar_list_t_TypeDecl(Result, Ivars.size()); 5440ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += " "; Result += VarName; 5441ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += ClassName; 5442ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__objc_const\"))) = {\n"; 5443ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t"; Result += "sizeof(_ivar_t)"; Result += ",\n"; 5444ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t"; Result += utostr(Ivars.size()); Result += ",\n"; 5445ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian for (unsigned i =0, e = Ivars.size(); i < e; i++) { 5446ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian ObjCIvarDecl *IvarDecl = Ivars[i]; 5447ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (i == 0) 5448ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t{{"; 5449ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian else 5450ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\t {"; 5451ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian // FIXME: // pointer to ivar offset location 5452ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "(unsigned long int *)0, "; 5453ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5454ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\""; Result += IvarDecl->getName(); Result += "\", "; 5455ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian std::string IvarTypeString, QuoteIvarTypeString; 5456ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Context->getObjCEncodingForType(IvarDecl->getType(), IvarTypeString, 5457ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IvarDecl); 5458ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian RewriteObj.QuoteDoublequotes(IvarTypeString, QuoteIvarTypeString); 5459ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "\""; Result += QuoteIvarTypeString; Result += "\", "; 5460ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5461ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian // FIXME: what should the alignment be? 5462ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian unsigned Align = 0; 5463ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += llvm::utostr(Align); Result += ", "; 5464a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian CharUnits Size = Context->getTypeSizeInChars(IvarDecl->getType()); 5465a63b422eda86b9ff5a08395dcfc288a31419038eFariborz Jahanian Result += llvm::utostr(Size.getQuantity()); 5466ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (i == e-1) 5467ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "}}\n"; 5468ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian else 5469ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "},\n"; 5470ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian } 5471ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Result += "};\n"; 5472ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian } 5473ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian} 5474ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 547564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCProtocolMetaData - Rewrite protocols meta-data. 5476da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCProtocolMetaData(ObjCProtocolDecl *PDecl, 5477da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::string &Result) { 547864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 547964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Do not synthesize the protocol more than once. 548064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCSynthesizedProtocols.count(PDecl->getCanonicalDecl())) 548164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return; 548282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian WriteModernMetadataDeclarations(Result); 548364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 548464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCProtocolDecl *Def = PDecl->getDefinition()) 548564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PDecl = Def; 5486da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // Must write out all protocol definitions in current qualifier list, 5487da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // and in their nested qualifiers before writing out current definition. 5488da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (ObjCProtocolDecl::protocol_iterator I = PDecl->protocol_begin(), 5489da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian E = PDecl->protocol_end(); I != E; ++I) 5490da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(*I, Result); 549164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 549277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian // Construct method lists. 549377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::vector<ObjCMethodDecl *> InstanceMethods, ClassMethods; 549477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian std::vector<ObjCMethodDecl *> OptInstanceMethods, OptClassMethods; 549577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (ObjCProtocolDecl::instmeth_iterator 549677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I = PDecl->instmeth_begin(), E = PDecl->instmeth_end(); 549777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I != E; ++I) { 549877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = *I; 549977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (MD->getImplementationControl() == ObjCMethodDecl::Optional) { 550077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian OptInstanceMethods.push_back(MD); 550177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } else { 550277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian InstanceMethods.push_back(MD); 550377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 550477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 550577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 550677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian for (ObjCProtocolDecl::classmeth_iterator 550777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I = PDecl->classmeth_begin(), E = PDecl->classmeth_end(); 550877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian I != E; ++I) { 550977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ObjCMethodDecl *MD = *I; 551077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian if (MD->getImplementationControl() == ObjCMethodDecl::Optional) { 551177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian OptClassMethods.push_back(MD); 551277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } else { 551377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian ClassMethods.push_back(MD); 551477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 551577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian } 5516e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian std::vector<ObjCMethodDecl *> AllMethods; 5517e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = InstanceMethods.size(); i < e; i++) 5518e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(InstanceMethods[i]); 5519e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = ClassMethods.size(); i < e; i++) 5520e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(ClassMethods[i]); 5521e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = OptInstanceMethods.size(); i < e; i++) 5522e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(OptInstanceMethods[i]); 5523e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian for (unsigned i = 0, e = OptClassMethods.size(); i < e; i++) 5524e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods.push_back(OptClassMethods[i]); 5525e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian 5526e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Write__extendedMethodTypes_initializer(*this, Context, Result, 5527e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian AllMethods, 5528e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian "_OBJC_PROTOCOL_METHOD_TYPES_", 5529e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian PDecl->getNameAsString()); 5530da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian // Protocol's super protocol list 5531da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian std::vector<ObjCProtocolDecl *> SuperProtocols; 5532da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian for (ObjCProtocolDecl::protocol_iterator I = PDecl->protocol_begin(), 5533da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian E = PDecl->protocol_end(); I != E; ++I) 5534da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian SuperProtocols.push_back(*I); 5535da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 5536da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Write_protocol_list_initializer(Context, Result, SuperProtocols, 5537da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian "_OBJC_PROTOCOL_REFS_", 5538da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian PDecl->getNameAsString()); 5539da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian 554077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Write_method_list_t_initializer(Context, Result, InstanceMethods, 554177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_INSTANCE_METHODS_", 554277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian PDecl->getNameAsString()); 554377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 554477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Write_method_list_t_initializer(Context, Result, ClassMethods, 554577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_CLASS_METHODS_", 554677e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian PDecl->getNameAsString()); 554777e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 554877e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Write_method_list_t_initializer(Context, Result, OptInstanceMethods, 554977e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_OPT_INSTANCE_METHODS_", 555077e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian PDecl->getNameAsString()); 555177e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 555277e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian Write_method_list_t_initializer(Context, Result, OptClassMethods, 555377e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian "_OBJC_PROTOCOL_OPT_CLASS_METHODS_", 555477e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian PDecl->getNameAsString()); 555577e4bcacd5c80d8bffe19a4489b5f059b618d4bcFariborz Jahanian 5556da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian // Protocol's property metadata. 5557da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian std::vector<ObjCPropertyDecl *> ProtocolProperties; 5558da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian for (ObjCContainerDecl::prop_iterator I = PDecl->prop_begin(), 5559da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian E = PDecl->prop_end(); I != E; ++I) 5560da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian ProtocolProperties.push_back(*I); 5561da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 5562da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian Write__prop_list_t_initializer(*this, Context, Result, ProtocolProperties, 5563da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian /* Container */0, 5564da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian "_OBJC_PROTOCOL_PROPERTIES_", 5565da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian PDecl->getNameAsString()); 5566da35eacc6e05a99ab30f7290f1d42df32d06b22eFariborz Jahanian 556782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian // Writer out root metadata for current protocol: struct _protocol_t 556882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\nstatic struct _protocol_t _OBJC_PROTOCOL_"; 556982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); 557082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += " __attribute__ ((used, section (\"__DATA,__datacoal_nt,coalesced\"))) = {\n"; 557182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; // id is; is null 557282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t\""; Result += PDecl->getNameAsString(); Result += "\",\n"; 5573da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian if (SuperProtocols.size() > 0) { 5574da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t(const struct _protocol_list_t *)&"; Result += "_OBJC_PROTOCOL_REFS_"; 5575da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 5576da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian } 5577da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian else 5578da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian Result += "\t0,\n"; 557982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (InstanceMethods.size() > 0) { 558082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_INSTANCE_METHODS_"; 558182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 558264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 558382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 558482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 558582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 558682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (ClassMethods.size() > 0) { 558782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_CLASS_METHODS_"; 558882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 558964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 559082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 559182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 559264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 559382848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (OptInstanceMethods.size() > 0) { 559482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_OPT_INSTANCE_METHODS_"; 559582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 559664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 559782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian else 559882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 559964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 560082848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (OptClassMethods.size() > 0) { 560182848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct method_list_t *)&_OBJC_PROTOCOL_OPT_CLASS_METHODS_"; 560282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 560364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 560464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 560582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 560682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 560782848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian if (ProtocolProperties.size() > 0) { 560882848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t(const struct _prop_list_t *)&_OBJC_PROTOCOL_PROPERTIES_"; 560982848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += PDecl->getNameAsString(); Result += ",\n"; 561064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 561164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 561282848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 561364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 561482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t"; Result += "sizeof(_protocol_t)"; Result += ",\n"; 561582848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian Result += "\t0,\n"; 561682848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 5617e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian if (AllMethods.size() > 0) { 5618e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t(const char **)&"; Result += "_OBJC_PROTOCOL_METHOD_TYPES_"; 5619e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += PDecl->getNameAsString(); 5620e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\n};\n"; 5621e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian } 5622e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian else 5623e0adbd8386d9ff7312b74e4d2d1f27337f94732cFariborz Jahanian Result += "\t0\n};\n"; 562482848c26c896d46c0b98e1776a7781e754a8ee25Fariborz Jahanian 562564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Mark this protocol as having been generated. 562664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ObjCSynthesizedProtocols.insert(PDecl->getCanonicalDecl())) 562764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian llvm_unreachable("protocol already synthesized"); 562864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 562964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 563064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 563164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCProtocolListMetaData( 563264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCList<ObjCProtocolDecl> &Protocols, 563364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, StringRef ClassName, 563464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 563564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (Protocols.empty()) return; 563664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 563764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 0; i != Protocols.size(); i++) 5638da9624a77352bc61b9713c31edd671c02a3ff260Fariborz Jahanian RewriteObjCProtocolMetaData(Protocols[i], Result); 563964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 564064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Output the top lovel protocol meta-data for the class. 564164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_protocol_list { 564264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_protocol_list *next; 564364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int protocol_count; 564464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_protocol *class_protocols[]; 564564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 564664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 564764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstatic struct {\n"; 564864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol_list *next;\n"; 564964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tint protocol_count;\n"; 565064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol *class_protocols["; 565164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(Protocols.size()); 565264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "];\n} _OBJC_"; 565364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += prefix; 565464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_PROTOCOLS_"; 565564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassName; 565664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __cat_cls_meth\")))= " 565764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "{\n\t0, "; 565864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(Protocols.size()); 565964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 566064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 566164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,{&_OBJC_PROTOCOL_"; 566264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += Protocols[0]->getNameAsString(); 566364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " \n"; 566464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 566564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (unsigned i = 1; i != Protocols.size(); i++) { 566664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t ,&_OBJC_PROTOCOL_"; 566764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += Protocols[i]->getNameAsString(); 566864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 566964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 567064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t }\n};\n"; 567164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 567264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 567364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl, 567464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 567564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *CDecl = IDecl->getClassInterface(); 567664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 567764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Explicitly declared @interface's are already synthesized. 5678ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (CDecl->isImplicitInterfaceDecl()) 5679ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian assert(false && 5680ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian "Legacy implicit interface rewriting not supported in moder abi"); 5681ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian WriteModernMetadataDeclarations(Result); 5682ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian SmallVector<ObjCIvarDecl *, 8> IVars; 5683ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5684ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian for (ObjCIvarDecl *IVD = CDecl->all_declared_ivar_begin(); 5685ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IVD; IVD = IVD->getNextIvar()) { 5686ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian // Ignore unnamed bit-fields. 5687ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian if (!IVD->getDeclName()) 5688ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian continue; 5689ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian IVars.push_back(IVD); 569064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 569164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 5692ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian Write__ivar_list_t_initializer(*this, Context, Result, IVars, 5693ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian "_OBJC_INSTANCE_VARIABLES_", 5694ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian CDecl->getNameAsString()); 5695ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 5696ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian // FIXME. Handle modern abi's private ivars declared in @implementation. 569764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_ivar_list metadata for classes ivars if needed 5698ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian unsigned NumIvars = CDecl->ivar_size(); 5699ae9329570ba16e48ffdfe3d55349b53fd893a711Fariborz Jahanian 570064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (NumIvars > 0) { 570164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 570264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct { 570364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int ivar_count; 570464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_ivar ivar_list[nIvars]; 570564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 570664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 570764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstatic struct {\n"; 570864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tint ivar_count;\n"; 570964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_ivar ivar_list["; 571064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(NumIvars); 571164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "];\n} _OBJC_INSTANCE_VARIABLES_"; 571264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IDecl->getNameAsString(); 571364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __instance_vars\")))= " 571464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "{\n\t"; 571564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(NumIvars); 571664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 571764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 571864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl::ivar_iterator IVI, IVE; 571964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCIvarDecl *, 8> IVars; 572064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!IDecl->ivar_empty()) { 572164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCInterfaceDecl::ivar_iterator 572264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV = IDecl->ivar_begin(), IVEnd = IDecl->ivar_end(); 572364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV != IVEnd; ++IV) 572464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IVars.push_back(*IV); 572564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IVI = IDecl->ivar_begin(); 572664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IVE = IDecl->ivar_end(); 572764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 572864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IVI = CDecl->ivar_begin(); 572964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IVE = CDecl->ivar_end(); 573064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 573164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,{{\""; 573264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*IVI)->getNameAsString(); 573364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 573464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TmpString, StrEncoding; 573564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForType((*IVI)->getType(), TmpString, *IVI); 573664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QuoteDoublequotes(TmpString, StrEncoding); 573764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += StrEncoding; 573864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", "; 573964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation(*IVI, Result); 574064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 574164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (++IVI; IVI != IVE; ++IVI) { 574264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t ,{\""; 574364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*IVI)->getNameAsString(); 574464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 574564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string TmpString, StrEncoding; 574664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForType((*IVI)->getType(), TmpString, *IVI); 574764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QuoteDoublequotes(TmpString, StrEncoding); 574864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += StrEncoding; 574964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", "; 575064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteIvarOffsetComputation((*IVI), Result); 575164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 575264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 575364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 575464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t }\n};\n"; 575564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 575664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 575764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's instance methods if needed 575864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 575964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods(IDecl->instmeth_begin(), IDecl->instmeth_end()); 576064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 576164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If any of our property implementations have associated getters or 576264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // setters, produce metadata for them as well. 576364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCImplDecl::propimpl_iterator Prop = IDecl->propimpl_begin(), 576464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropEnd = IDecl->propimpl_end(); 576564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Prop != PropEnd; ++Prop) { 576664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*Prop)->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 576764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 576864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!(*Prop)->getPropertyIvarDecl()) 576964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 577064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl *PD = (*Prop)->getPropertyDecl(); 577164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!PD) 577264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 577364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Getter = PD->getGetterMethodDecl()) 577464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Getter->isDefined()) 577564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Getter); 577664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isReadOnly()) 577764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 577864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Setter = PD->getSetterMethodDecl()) 577964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!Setter->isDefined()) 578064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Setter); 578164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 578264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodsMetaData(InstanceMethods.begin(), InstanceMethods.end(), 578364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true, "", IDecl->getName(), Result); 578464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 578564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's class methods if needed 578664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodsMetaData(IDecl->classmeth_begin(), IDecl->classmeth_end(), 578764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false, "", IDecl->getName(), Result); 578864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 578964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Protocols referenced in class declaration? 579064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCProtocolListMetaData(CDecl->getReferencedProtocols(), 579164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "CLASS", CDecl->getName(), Result); 579264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 579364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Declaration of class/meta-class metadata 579464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_class { 579564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_class *isa; // or const char *root_class_name when metadata 579664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *super_class_name; 579764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char *name; 579864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian long version; 579964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian long info; 580064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian long instance_size; 580164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_ivar_list *ivars; 580264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method_list *methods; 580364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct objc_cache *cache; 580464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct objc_protocol_list *protocols; 580564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *ivar_layout; 580664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_class_ext *ext; 580764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 580864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 580964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objc_class = false; 581064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!objc_class) { 581164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_class {\n"; 581264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_class *isa;\n"; 581364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char *super_class_name;\n"; 581464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tchar *name;\n"; 581564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tlong version;\n"; 581664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tlong info;\n"; 581764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tlong instance_size;\n"; 581864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_ivar_list *ivars;\n"; 581964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method_list *methods;\n"; 582064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct objc_cache *cache;\n"; 582164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol_list *protocols;\n"; 582264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char *ivar_layout;\n"; 582364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_class_ext *ext;\n"; 582464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 582564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_class = true; 582664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 582764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 582864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Meta-class metadata generation. 582964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *RootClass = 0; 583064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *SuperClass = CDecl->getSuperClass(); 583164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian while (SuperClass) { 583264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RootClass = SuperClass; 583364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperClass = SuperClass->getSuperClass(); 583464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 583564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SuperClass = CDecl->getSuperClass(); 583664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 583764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstatic struct _objc_class _OBJC_METACLASS_"; 583864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 583964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __meta_class\")))= " 584064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "{\n\t(struct _objc_class *)\""; 584164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (RootClass ? RootClass->getNameAsString() : CDecl->getNameAsString()); 584264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\""; 584364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 584464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SuperClass) { 584564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", \""; 584664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += SuperClass->getNameAsString(); 584764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 584864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 584964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\""; 585064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 585164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 585264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", 0, \""; 585364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 585464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\""; 585564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 585664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Set 'ivars' field for root class to 0. ObjC1 runtime does not use it. 585764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 'info' field is initialized to CLS_META(2) for metaclass 585864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", 0,2, sizeof(struct _objc_class), 0"; 585964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IDecl->classmeth_begin() != IDecl->classmeth_end()) { 586064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n\t, (struct _objc_method_list *)&_OBJC_CLASS_METHODS_"; 586164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IDecl->getNameAsString(); 586264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 586364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 586464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 586564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", 0\n"; 586664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CDecl->protocol_begin() != CDecl->protocol_end()) { 586764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,0, (struct _objc_protocol_list *)&_OBJC_CLASS_PROTOCOLS_"; 586864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 586964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ",0,0\n"; 587064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 587164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 587264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,0,0,0,0\n"; 587364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 587464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 587564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // class metadata generation. 587664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstatic struct _objc_class _OBJC_CLASS_"; 587764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 587864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __class\")))= " 587964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "{\n\t&_OBJC_METACLASS_"; 588064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 588164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (SuperClass) { 588264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", \""; 588364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += SuperClass->getNameAsString(); 588464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 588564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 588664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\""; 588764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 588864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 588964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", 0, \""; 589064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 589164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\""; 589264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 589364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // 'info' field is initialized to CLS_CLASS(1) for class 589464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", 0,1"; 589564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!ObjCSynthesizedStructs.count(CDecl)) 589664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ",0"; 589764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else { 589864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // class has size. Must synthesize its size. 589964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ",sizeof(struct "; 590064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 590164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) 590264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_IMPL"; 590364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ")"; 590464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 590564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (NumIvars > 0) { 590664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", (struct _objc_ivar_list *)&_OBJC_INSTANCE_VARIABLES_"; 590764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 590864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n\t"; 590964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 591064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 591164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ",0"; 591264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IDecl->instmeth_begin() != IDecl->instmeth_end()) { 591364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", (struct _objc_method_list *)&_OBJC_INSTANCE_METHODS_"; 591464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 591564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", 0\n\t"; 591664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 591764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 591864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ",0,0"; 591964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CDecl->protocol_begin() != CDecl->protocol_end()) { 592064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", (struct _objc_protocol_list*)&_OBJC_CLASS_PROTOCOLS_"; 592164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CDecl->getNameAsString(); 592264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ", 0,0\n"; 592364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 592464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 592564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ",0,0,0\n"; 592664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 592764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 592864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 592964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteMetaDataIntoBuffer(std::string &Result) { 593064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int ClsDefCount = ClassImplementation.size(); 593164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int CatDefCount = CategoryImplementation.size(); 593264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 593364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For each implemented class, write out all its meta data. 593464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) 593564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCClassMetaData(ClassImplementation[i], Result); 593664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 593764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // For each implemented category, write out all its meta data. 593864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < CatDefCount; i++) 593964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCCategoryImplDecl(CategoryImplementation[i], Result); 594064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 594164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Write objc_symtab metadata 594264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* 594364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_symtab 594464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 594564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian long sel_ref_cnt; 594664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SEL *refs; 594764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian short cls_def_cnt; 594864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian short cat_def_cnt; 594964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void *defs[cls_def_cnt + cat_def_cnt]; 595064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 595164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 595264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 595364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_symtab {\n"; 595464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tlong sel_ref_cnt;\n"; 595564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tSEL *refs;\n"; 595664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tshort cls_def_cnt;\n"; 595764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tshort cat_def_cnt;\n"; 595864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tvoid *defs[" + utostr(ClsDefCount + CatDefCount)+ "];\n"; 595964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n\n"; 596064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 596164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "static struct _objc_symtab " 596264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "_OBJC_SYMBOLS __attribute__((used, section (\"__OBJC, __symbols\")))= {\n"; 596364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t0, 0, " + utostr(ClsDefCount) 596464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian + ", " + utostr(CatDefCount) + "\n"; 596564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < ClsDefCount; i++) { 596664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,&_OBJC_CLASS_"; 596764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassImplementation[i]->getNameAsString(); 596864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 596964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 597064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 597164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (int i = 0; i < CatDefCount; i++) { 597264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,&_OBJC_CATEGORY_"; 597364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CategoryImplementation[i]->getClassInterface()->getNameAsString(); 597464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_"; 597564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += CategoryImplementation[i]->getNameAsString(); 597664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 597764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 597864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 597964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n\n"; 598064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 598164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Write objc_module metadata 598264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 598364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* 598464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_module { 598564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian long version; 598664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian long size; 598764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const char *name; 598864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_symtab *symtab; 598964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 599064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 599164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 599264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_module {\n"; 599364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tlong version;\n"; 599464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tlong size;\n"; 599564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tconst char *name;\n"; 599664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_symtab *symtab;\n"; 599764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n\n"; 599864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "static struct _objc_module " 599964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "_OBJC_MODULES __attribute__ ((used, section (\"__OBJC, __module_info\")))= {\n"; 600064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t" + utostr(OBJC_ABI_VERSION) + 600164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ", sizeof(struct _objc_module), \"\", &_OBJC_SYMBOLS\n"; 600264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n\n"; 600364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 600464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (LangOpts.MicrosoftExt) { 600564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ProtocolExprDecls.size()) { 600664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "#pragma section(\".objc_protocol$B\",long,read,write)\n"; 600764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "#pragma data_seg(push, \".objc_protocol$B\")\n"; 600864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (llvm::SmallPtrSet<ObjCProtocolDecl *,8>::iterator I = ProtocolExprDecls.begin(), 600964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian E = ProtocolExprDecls.end(); I != E; ++I) { 601064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "static struct _objc_protocol *_POINTER_OBJC_PROTOCOL_"; 601164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*I)->getNameAsString(); 601264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " = &_OBJC_PROTOCOL_"; 601364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*I)->getNameAsString(); 601464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ";\n"; 601564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 601664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "#pragma data_seg(pop)\n\n"; 601764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 601864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "#pragma section(\".objc_module_info$B\",long,read,write)\n"; 601964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "#pragma data_seg(push, \".objc_module_info$B\")\n"; 602064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "static struct _objc_module *_POINTER_OBJC_MODULES = "; 602164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "&_OBJC_MODULES;\n"; 602264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "#pragma data_seg(pop)\n\n"; 602364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 602464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 602564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 602664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// RewriteObjCCategoryImplDecl - Rewrite metadata for each category 602764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// implementation. 602864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *IDecl, 602964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 603064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *ClassDecl = IDecl->getClassInterface(); 603164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Find category declaration for this implementation. 603264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCCategoryDecl *CDecl; 603364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (CDecl = ClassDecl->getCategoryList(); CDecl; 603464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CDecl = CDecl->getNextClassCategory()) 603564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CDecl->getIdentifier() == IDecl->getIdentifier()) 603664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian break; 603764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 603864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string FullCategoryName = ClassDecl->getNameAsString(); 603964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FullCategoryName += '_'; 604064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FullCategoryName += IDecl->getNameAsString(); 604164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 604264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's instance methods if needed 604364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SmallVector<ObjCMethodDecl *, 32> 604464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods(IDecl->instmeth_begin(), IDecl->instmeth_end()); 604564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 604664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // If any of our property implementations have associated getters or 604764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // setters, produce metadata for them as well. 604864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (ObjCImplDecl::propimpl_iterator Prop = IDecl->propimpl_begin(), 604964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian PropEnd = IDecl->propimpl_end(); 605064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Prop != PropEnd; ++Prop) { 605164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if ((*Prop)->getPropertyImplementation() == ObjCPropertyImplDecl::Dynamic) 605264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 605364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!(*Prop)->getPropertyIvarDecl()) 605464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 605564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCPropertyDecl *PD = (*Prop)->getPropertyDecl(); 605664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!PD) 605764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 605864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Getter = PD->getGetterMethodDecl()) 605964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Getter); 606064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (PD->isReadOnly()) 606164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian continue; 606264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (ObjCMethodDecl *Setter = PD->getSetterMethodDecl()) 606364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian InstanceMethods.push_back(Setter); 606464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 606564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodsMetaData(InstanceMethods.begin(), InstanceMethods.end(), 606664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian true, "CATEGORY_", FullCategoryName.c_str(), 606764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result); 606864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 606964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's class methods if needed 607064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCMethodsMetaData(IDecl->classmeth_begin(), IDecl->classmeth_end(), 607164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian false, "CATEGORY_", FullCategoryName.c_str(), 607264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result); 607364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 607464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Protocols referenced in class declaration? 607564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Null CDecl is case of a category implementation with no category interface 607664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CDecl) 607764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RewriteObjCProtocolListMetaData(CDecl->getReferencedProtocols(), "CATEGORY", 607864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian FullCategoryName, Result); 607964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_category { 608064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char *category_name; 608164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char *class_name; 608264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method_list *instance_methods; 608364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method_list *class_methods; 608464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_protocol_list *protocols; 608564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Objective-C 1.0 extensions 608664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian uint32_t size; // sizeof (struct _objc_category) 608764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_property_list *instance_properties; // category's own 608864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // @property decl. 608964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian }; 609064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 609164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 609264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian static bool objc_category = false; 609364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!objc_category) { 609464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_category {\n"; 609564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tchar *category_name;\n"; 609664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tchar *class_name;\n"; 609764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method_list *instance_methods;\n"; 609864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method_list *class_methods;\n"; 609964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_protocol_list *protocols;\n"; 610064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tunsigned int size;\n"; 610164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_property_list *instance_properties;\n"; 610264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 610364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_category = true; 610464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 610564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstatic struct _objc_category _OBJC_CATEGORY_"; 610664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += FullCategoryName; 610764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __category\")))= {\n\t\""; 610864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IDecl->getNameAsString(); 610964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\"\n\t, \""; 611064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassDecl->getNameAsString(); 611164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\"\n"; 611264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 611364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IDecl->instmeth_begin() != IDecl->instmeth_end()) { 611464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t, (struct _objc_method_list *)" 611564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "&_OBJC_CATEGORY_INSTANCE_METHODS_"; 611664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += FullCategoryName; 611764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 611864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 611964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 612064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t, 0\n"; 612164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IDecl->classmeth_begin() != IDecl->classmeth_end()) { 612264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t, (struct _objc_method_list *)" 612364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian "&_OBJC_CATEGORY_CLASS_METHODS_"; 612464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += FullCategoryName; 612564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 612664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 612764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 612864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t, 0\n"; 612964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 613064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CDecl && CDecl->protocol_begin() != CDecl->protocol_end()) { 613164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t, (struct _objc_protocol_list *)&_OBJC_CATEGORY_PROTOCOLS_"; 613264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += FullCategoryName; 613364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\n"; 613464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 613564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian else 613664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t, 0\n"; 613764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t, sizeof(struct _objc_category), 0\n};\n"; 613864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 613964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 614064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian// RewriteObjCMethodsMetaData - Rewrite methods metadata for instance or 614164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian/// class methods. 614264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahaniantemplate<typename MethodIterator> 614364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanianvoid RewriteModernObjC::RewriteObjCMethodsMetaData(MethodIterator MethodBegin, 614464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MethodIterator MethodEnd, 614564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian bool IsInstanceMethod, 614664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef prefix, 614764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian StringRef ClassName, 614864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string &Result) { 614964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (MethodBegin == MethodEnd) return; 615064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 615164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (!objc_impl_method) { 615264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct _objc_method { 615364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SEL _cmd; 615464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian char *method_types; 615564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian void *_imp; 615664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 615764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 615864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstruct _objc_method {\n"; 615964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tSEL _cmd;\n"; 616064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tchar *method_types;\n"; 616164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tvoid *_imp;\n"; 616264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "};\n"; 616364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 616464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian objc_impl_method = true; 616564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 616664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 616764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Build _objc_method_list for class's methods if needed 616864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 616964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian /* struct { 617064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method_list *next_method; 617164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian int method_count; 617264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian struct _objc_method method_list[]; 617364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 617464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian */ 617564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian unsigned NumMethods = std::distance(MethodBegin, MethodEnd); 617664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\nstatic struct {\n"; 617764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method_list *next_method;\n"; 617864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tint method_count;\n"; 617964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\tstruct _objc_method method_list["; 618064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += utostr(NumMethods); 618164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "];\n} _OBJC_"; 618264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += prefix; 618364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IsInstanceMethod ? "INSTANCE" : "CLASS"; 618464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_METHODS_"; 618564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += ClassName; 618664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += " __attribute__ ((used, section (\"__OBJC, __"; 618764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += IsInstanceMethod ? "inst" : "cls"; 618864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "_meth\")))= "; 618964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "{\n\t0, " + utostr(NumMethods) + "\n"; 619064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 619164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t,{{(SEL)\""; 619264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*MethodBegin)->getSelector().getAsString().c_str(); 619364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string MethodTypeString; 619464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString); 619564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 619664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodTypeString; 619764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", (void *)"; 619864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodInternalNames[*MethodBegin]; 619964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 620064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian for (++MethodBegin; MethodBegin != MethodEnd; ++MethodBegin) { 620164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t ,{(SEL)\""; 620264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += (*MethodBegin)->getSelector().getAsString().c_str(); 620364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string MethodTypeString; 620464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString); 620564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", \""; 620664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodTypeString; 620764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\", (void *)"; 620864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += MethodInternalNames[*MethodBegin]; 620964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "}\n"; 621064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 621164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Result += "\t }\n};\n"; 621264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 621364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 621464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz JahanianStmt *RewriteModernObjC::RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV) { 621564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceRange OldRange = IV->getSourceRange(); 621664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *BaseExpr = IV->getBase(); 621764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 621864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Rewrite the base, but without actually doing replaces. 621964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian { 622064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian DisableReplaceStmtScope S(*this); 622164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian BaseExpr = cast<Expr>(RewriteFunctionBodyOrGlobalInitializer(BaseExpr)); 622264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->setBase(BaseExpr); 622364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 622464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 622564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCIvarDecl *D = IV->getDecl(); 622664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 622764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Expr *Replacement = IV; 622864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (CurMethodDef) { 622964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BaseExpr->getType()->isObjCObjectPointerType()) { 623064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceType *iFaceDecl = 623164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCInterfaceType>(BaseExpr->getType()->getPointeeType()); 623264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(iFaceDecl && "RewriteObjCIvarRefExpr - iFaceDecl is null"); 623364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // lookup which class implements the instance variable. 623464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *clsDeclared = 0; 623564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian iFaceDecl->getDecl()->lookupInstanceVariable(D->getIdentifier(), 623664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian clsDeclared); 623764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(clsDeclared && "RewriteObjCIvarRefExpr(): Can't find class"); 623864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 623964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to gain access to the ivar. 624064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RecName = clsDeclared->getIdentifier()->getName(); 624164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecName += "_IMPL"; 624264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *II = &Context->Idents.get(RecName); 624364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 624464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 624564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian II); 624664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(RD && "RewriteObjCIvarRefExpr(): Can't find RecordDecl"); 624764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castT = Context->getPointerType(Context->getTagDeclType(RD)); 624864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, castT, 624964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 625064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->getBase()); 625164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 625264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(OldRange.getBegin(), 625364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian OldRange.getEnd(), 625464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian castExpr); 625564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (IV->isFreeIvar() && 625664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian declaresSameEntity(CurMethodDef->getClassInterface(), iFaceDecl->getDecl())) { 625764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian MemberExpr *ME = new (Context) MemberExpr(PE, true, D, 625864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->getLocation(), 625964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian D->getType(), 626064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian VK_LValue, OK_Ordinary); 626164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian Replacement = ME; 626264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { 626364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->setBase(PE); 626464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 626564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 626664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } else { // we are outside a method. 626764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(!IV->isFreeIvar() && "Cannot have a free standing ivar outside a method"); 626864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 626964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Explicit ivar refs need to have a cast inserted. 627064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // FIXME: consider sharing some of this code with the code above. 627164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian if (BaseExpr->getType()->isObjCObjectPointerType()) { 627264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian const ObjCInterfaceType *iFaceDecl = 627364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian dyn_cast<ObjCInterfaceType>(BaseExpr->getType()->getPointeeType()); 627464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // lookup which class implements the instance variable. 627564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ObjCInterfaceDecl *clsDeclared = 0; 627664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian iFaceDecl->getDecl()->lookupInstanceVariable(D->getIdentifier(), 627764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian clsDeclared); 627864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(clsDeclared && "RewriteObjCIvarRefExpr(): Can't find class"); 627964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 628064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Synthesize an explicit cast to gain access to the ivar. 628164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian std::string RecName = clsDeclared->getIdentifier()->getName(); 628264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecName += "_IMPL"; 628364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IdentifierInfo *II = &Context->Idents.get(RecName); 628464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian RecordDecl *RD = RecordDecl::Create(*Context, TTK_Struct, TUDecl, 628564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian SourceLocation(), SourceLocation(), 628664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian II); 628764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian assert(RD && "RewriteObjCIvarRefExpr(): Can't find RecordDecl"); 628864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian QualType castT = Context->getPointerType(Context->getTagDeclType(RD)); 628964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CastExpr *castExpr = NoTypeInfoCStyleCastExpr(Context, castT, 629064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian CK_BitCast, 629164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->getBase()); 629264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Don't forget the parens to enforce the proper binding. 629364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ParenExpr *PE = new (Context) ParenExpr(IV->getBase()->getLocStart(), 629464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->getBase()->getLocEnd(), castExpr); 629564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Cannot delete IV->getBase(), since PE points to it. 629664cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // Replace the old base with the cast. This is important when doing 629764cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian // embedded rewrites. For example, [newInv->_container addObject:0]. 629864cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian IV->setBase(PE); 629964cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 630064cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian } 630164cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 630264cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian ReplaceStmtWithRange(IV, Replacement, OldRange); 630364cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian return Replacement; 630464cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian} 630564cb63a0cfa76b278e0791141c7b5a1b424ab434Fariborz Jahanian 6306