blob: 9a4657e522b297fa13f415867e7793ee4a1867ce [file] [log] [blame]
Steve Naroffb29b4272008-04-14 22:03:09 +00001//===--- RewriteObjC.cpp - Playground for the code rewriter ---------------===//
Chris Lattner77cd2a02007-10-11 00:43:27 +00002//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattner77cd2a02007-10-11 00:43:27 +00007//
8//===----------------------------------------------------------------------===//
9//
10// Hacks and fun related to the code rewriter.
11//
12//===----------------------------------------------------------------------===//
13
14#include "ASTConsumers.h"
Chris Lattner8a12c272007-10-11 18:38:32 +000015#include "clang/Rewrite/Rewriter.h"
Chris Lattner77cd2a02007-10-11 00:43:27 +000016#include "clang/AST/AST.h"
17#include "clang/AST/ASTConsumer.h"
Ted Kremeneke3a61982008-05-31 20:11:04 +000018#include "clang/AST/TranslationUnit.h"
Chris Lattner8a12c272007-10-11 18:38:32 +000019#include "clang/Basic/SourceManager.h"
Steve Naroffebf2b562007-10-23 23:50:29 +000020#include "clang/Basic/IdentifierTable.h"
Chris Lattner07506182007-11-30 22:53:43 +000021#include "clang/Basic/Diagnostic.h"
Chris Lattner26de4652007-12-02 01:13:47 +000022#include "clang/Lex/Lexer.h"
Chris Lattner158ecb92007-10-25 17:07:24 +000023#include "llvm/ADT/StringExtras.h"
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +000024#include "llvm/ADT/SmallPtrSet.h"
Chris Lattner26de4652007-12-02 01:13:47 +000025#include "llvm/Support/MemoryBuffer.h"
Steve Naroff0113c9d2008-01-28 21:34:52 +000026#include "llvm/Support/CommandLine.h"
Dan Gohman63e2dcc2008-05-21 20:19:16 +000027#include "llvm/Support/Streams.h"
Chris Lattnerc68ab772008-03-22 00:08:40 +000028#include "llvm/System/Path.h"
Steve Naroff0d4e9632008-04-14 22:53:48 +000029#include <sstream>
Chris Lattnerc68ab772008-03-22 00:08:40 +000030#include <fstream>
Chris Lattner77cd2a02007-10-11 00:43:27 +000031using namespace clang;
Chris Lattner158ecb92007-10-25 17:07:24 +000032using llvm::utostr;
Chris Lattner77cd2a02007-10-11 00:43:27 +000033
Steve Naroff0113c9d2008-01-28 21:34:52 +000034static llvm::cl::opt<bool>
35SilenceRewriteMacroWarning("Wno-rewrite-macros", llvm::cl::init(false),
36 llvm::cl::desc("Silence ObjC rewriting warnings"));
37
Chris Lattner77cd2a02007-10-11 00:43:27 +000038namespace {
Steve Naroffb29b4272008-04-14 22:03:09 +000039 class RewriteObjC : public ASTConsumer {
Chris Lattner2c64b7b2007-10-16 21:07:07 +000040 Rewriter Rewrite;
Chris Lattnere365c502007-11-30 22:25:36 +000041 Diagnostic &Diags;
Steve Naroff4f943c22008-03-10 20:43:59 +000042 const LangOptions &LangOpts;
Steve Narofff69cc5d2008-01-30 19:17:43 +000043 unsigned RewriteFailedDiag;
44
Chris Lattner01c57482007-10-17 22:35:30 +000045 ASTContext *Context;
Chris Lattner77cd2a02007-10-11 00:43:27 +000046 SourceManager *SM;
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +000047 TranslationUnitDecl *TUDecl;
Chris Lattner8a12c272007-10-11 18:38:32 +000048 unsigned MainFileID;
Chris Lattner26de4652007-12-02 01:13:47 +000049 const char *MainFileStart, *MainFileEnd;
Chris Lattner2c64b7b2007-10-16 21:07:07 +000050 SourceLocation LastIncLoc;
Steve Naroffba92b2e2008-03-27 22:29:16 +000051
Ted Kremeneka526c5c2008-01-07 19:49:32 +000052 llvm::SmallVector<ObjCImplementationDecl *, 8> ClassImplementation;
53 llvm::SmallVector<ObjCCategoryImplDecl *, 8> CategoryImplementation;
54 llvm::SmallPtrSet<ObjCInterfaceDecl*, 8> ObjCSynthesizedStructs;
Steve Narofffbfe8252008-05-06 18:26:51 +000055 llvm::SmallPtrSet<ObjCProtocolDecl*, 8> ObjCSynthesizedProtocols;
Ted Kremeneka526c5c2008-01-07 19:49:32 +000056 llvm::SmallPtrSet<ObjCInterfaceDecl*, 8> ObjCForwardDecls;
57 llvm::DenseMap<ObjCMethodDecl*, std::string> MethodInternalNames;
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +000058 llvm::SmallVector<Stmt *, 32> Stmts;
59 llvm::SmallVector<int, 8> ObjCBcLabelNo;
Fariborz Jahanian7d6b46d2008-01-22 22:44:46 +000060 llvm::SmallVector<const RecordType *, 8> EncodingRecordTypes;
Steve Naroffebf2b562007-10-23 23:50:29 +000061
Steve Naroffd82a9ab2008-03-15 00:55:56 +000062 unsigned NumObjCStringLiterals;
63
Steve Naroffebf2b562007-10-23 23:50:29 +000064 FunctionDecl *MsgSendFunctionDecl;
Steve Naroff874e2322007-11-15 10:28:18 +000065 FunctionDecl *MsgSendSuperFunctionDecl;
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +000066 FunctionDecl *MsgSendStretFunctionDecl;
67 FunctionDecl *MsgSendSuperStretFunctionDecl;
Fariborz Jahanianacb49772007-12-03 21:26:48 +000068 FunctionDecl *MsgSendFpretFunctionDecl;
Steve Naroffebf2b562007-10-23 23:50:29 +000069 FunctionDecl *GetClassFunctionDecl;
Steve Naroff9bcb5fc2007-12-07 03:50:46 +000070 FunctionDecl *GetMetaClassFunctionDecl;
Steve Naroff934f2762007-10-24 22:48:43 +000071 FunctionDecl *SelGetUidFunctionDecl;
Steve Naroff96984642007-11-08 14:30:50 +000072 FunctionDecl *CFStringFunctionDecl;
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +000073 FunctionDecl *GetProtocolFunctionDecl;
Steve Naroffc0a123c2008-03-11 17:37:02 +000074 FunctionDecl *SuperContructorFunctionDecl;
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +000075
Steve Naroffbeaf2992007-11-03 11:27:19 +000076 // ObjC string constant support.
Steve Naroff248a7532008-04-15 22:42:06 +000077 VarDecl *ConstantStringClassReference;
Steve Naroffbeaf2992007-11-03 11:27:19 +000078 RecordDecl *NSStringRecord;
Steve Naroffab972d32007-11-04 22:37:50 +000079
Fariborz Jahanianb586cce2008-01-16 00:09:11 +000080 // ObjC foreach break/continue generation support.
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +000081 int BcLabelCount;
82
Steve Naroff874e2322007-11-15 10:28:18 +000083 // Needed for super.
Ted Kremeneka526c5c2008-01-07 19:49:32 +000084 ObjCMethodDecl *CurMethodDecl;
Steve Naroff874e2322007-11-15 10:28:18 +000085 RecordDecl *SuperStructDecl;
Steve Naroffd82a9ab2008-03-15 00:55:56 +000086 RecordDecl *ConstantStringDecl;
Steve Naroff874e2322007-11-15 10:28:18 +000087
Fariborz Jahanianb4b2f0c2008-01-18 01:15:54 +000088 // Needed for header files being rewritten
89 bool IsHeader;
90
Steve Naroffa7b402d2008-03-28 22:26:09 +000091 std::string InFileName;
92 std::string OutFileName;
93
Steve Naroffba92b2e2008-03-27 22:29:16 +000094 std::string Preamble;
95
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +000096 static const int OBJC_ABI_VERSION =7 ;
Chris Lattner77cd2a02007-10-11 00:43:27 +000097 public:
Ted Kremeneke3a61982008-05-31 20:11:04 +000098 virtual void Initialize(ASTContext &context);
99
100 virtual void InitializeTU(TranslationUnit &TU) {
101 TU.SetOwnsDecls(false);
102 Initialize(TU.getContext());
103 }
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000104
Chris Lattner8a12c272007-10-11 18:38:32 +0000105
Chris Lattnerf04da132007-10-24 17:06:59 +0000106 // Top Level Driver code.
107 virtual void HandleTopLevelDecl(Decl *D);
Chris Lattner2c64b7b2007-10-16 21:07:07 +0000108 void HandleDeclInMainFile(Decl *D);
Steve Naroffb29b4272008-04-14 22:03:09 +0000109 RewriteObjC(std::string inFile, std::string outFile,
Steve Naroffa7b402d2008-03-28 22:26:09 +0000110 Diagnostic &D, const LangOptions &LOpts);
Steve Naroffb29b4272008-04-14 22:03:09 +0000111 ~RewriteObjC();
Chris Lattnerdcbc5b02008-01-31 19:37:57 +0000112
113 void ReplaceStmt(Stmt *Old, Stmt *New) {
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000114 // If replacement succeeded or warning disabled return with no warning.
115 if (!Rewrite.ReplaceStmt(Old, New) || SilenceRewriteMacroWarning)
Chris Lattnerdcbc5b02008-01-31 19:37:57 +0000116 return;
117
Chris Lattnerdcbc5b02008-01-31 19:37:57 +0000118 SourceRange Range = Old->getSourceRange();
119 Diags.Report(Context->getFullLoc(Old->getLocStart()), RewriteFailedDiag,
120 0, 0, &Range, 1);
121 }
122
Steve Naroffba92b2e2008-03-27 22:29:16 +0000123 void InsertText(SourceLocation Loc, const char *StrData, unsigned StrLen,
124 bool InsertAfter = true) {
Chris Lattneraadaf782008-01-31 19:51:04 +0000125 // If insertion succeeded or warning disabled return with no warning.
Steve Naroffba92b2e2008-03-27 22:29:16 +0000126 if (!Rewrite.InsertText(Loc, StrData, StrLen, InsertAfter) ||
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000127 SilenceRewriteMacroWarning)
128 return;
129
130 Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag);
131 }
Chris Lattnerdcbc5b02008-01-31 19:37:57 +0000132
Chris Lattneraadaf782008-01-31 19:51:04 +0000133 void RemoveText(SourceLocation Loc, unsigned StrLen) {
134 // If removal succeeded or warning disabled return with no warning.
135 if (!Rewrite.RemoveText(Loc, StrLen) || SilenceRewriteMacroWarning)
136 return;
137
138 Diags.Report(Context->getFullLoc(Loc), RewriteFailedDiag);
139 }
Chris Lattnerf04da132007-10-24 17:06:59 +0000140
Chris Lattneraadaf782008-01-31 19:51:04 +0000141 void ReplaceText(SourceLocation Start, unsigned OrigLength,
142 const char *NewStr, unsigned NewLength) {
143 // If removal succeeded or warning disabled return with no warning.
144 if (!Rewrite.ReplaceText(Start, OrigLength, NewStr, NewLength) ||
145 SilenceRewriteMacroWarning)
146 return;
147
148 Diags.Report(Context->getFullLoc(Start), RewriteFailedDiag);
149 }
150
Chris Lattnerf04da132007-10-24 17:06:59 +0000151 // Syntactic Rewriting.
Steve Naroffab972d32007-11-04 22:37:50 +0000152 void RewritePrologue(SourceLocation Loc);
Fariborz Jahanian452b8992008-01-19 00:30:35 +0000153 void RewriteInclude();
Chris Lattnerf04da132007-10-24 17:06:59 +0000154 void RewriteTabs();
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000155 void RewriteForwardClassDecl(ObjCClassDecl *Dcl);
156 void RewriteInterfaceDecl(ObjCInterfaceDecl *Dcl);
Fariborz Jahanian66d6b292007-11-13 20:04:28 +0000157 void RewriteImplementationDecl(NamedDecl *Dcl);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000158 void RewriteObjCMethodDecl(ObjCMethodDecl *MDecl, std::string &ResultStr);
159 void RewriteCategoryDecl(ObjCCategoryDecl *Dcl);
160 void RewriteProtocolDecl(ObjCProtocolDecl *Dcl);
161 void RewriteForwardProtocolDecl(ObjCForwardProtocolDecl *Dcl);
162 void RewriteMethodDeclaration(ObjCMethodDecl *Method);
Chris Lattner55d13b42008-03-16 21:23:50 +0000163 void RewriteProperties(unsigned nProperties, ObjCPropertyDecl **Properties);
Steve Naroff09b266e2007-10-30 23:14:51 +0000164 void RewriteFunctionDecl(FunctionDecl *FD);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000165 void RewriteObjCQualifiedInterfaceTypes(Decl *Dcl);
Steve Naroffd5255f52007-11-01 13:24:47 +0000166 bool needToScanForQualifiers(QualType T);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000167 ObjCInterfaceDecl *isSuperReceiver(Expr *recExpr);
Steve Naroff874e2322007-11-15 10:28:18 +0000168 QualType getSuperStructType();
Steve Naroffd82a9ab2008-03-15 00:55:56 +0000169 QualType getConstantStringStructType();
Steve Naroffbaf58c32008-05-31 14:15:04 +0000170 bool BufferContainsPPDirectives(const char *startBuf, const char *endBuf);
Chris Lattner311ff022007-10-16 22:36:42 +0000171
Chris Lattnerf04da132007-10-24 17:06:59 +0000172 // Expression Rewriting.
Steve Narofff3473a72007-11-09 15:20:18 +0000173 Stmt *RewriteFunctionBodyOrGlobalInitializer(Stmt *S);
Chris Lattnere64b7772007-10-24 16:57:36 +0000174 Stmt *RewriteAtEncode(ObjCEncodeExpr *Exp);
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000175 Stmt *RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV, SourceLocation OrigStart);
Steve Naroffb42f8412007-11-05 14:50:49 +0000176 Stmt *RewriteAtSelector(ObjCSelectorExpr *Exp);
Chris Lattnere64b7772007-10-24 16:57:36 +0000177 Stmt *RewriteMessageExpr(ObjCMessageExpr *Exp);
Steve Naroffbeaf2992007-11-03 11:27:19 +0000178 Stmt *RewriteObjCStringLiteral(ObjCStringLiteral *Exp);
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +0000179 Stmt *RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000180 Stmt *RewriteObjCTryStmt(ObjCAtTryStmt *S);
Fariborz Jahaniana0f55792008-01-29 22:59:37 +0000181 Stmt *RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000182 Stmt *RewriteObjCCatchStmt(ObjCAtCatchStmt *S);
183 Stmt *RewriteObjCFinallyStmt(ObjCAtFinallyStmt *S);
184 Stmt *RewriteObjCThrowStmt(ObjCAtThrowStmt *S);
Chris Lattner338d1e22008-01-31 05:10:40 +0000185 Stmt *RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S,
186 SourceLocation OrigEnd);
Steve Naroff934f2762007-10-24 22:48:43 +0000187 CallExpr *SynthesizeCallToFunctionDecl(FunctionDecl *FD,
188 Expr **args, unsigned nargs);
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +0000189 Stmt *SynthMessageExpr(ObjCMessageExpr *Exp);
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +0000190 Stmt *RewriteBreakStmt(BreakStmt *S);
191 Stmt *RewriteContinueStmt(ContinueStmt *S);
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +0000192 void SynthCountByEnumWithState(std::string &buf);
193
Steve Naroff09b266e2007-10-30 23:14:51 +0000194 void SynthMsgSendFunctionDecl();
Steve Naroff874e2322007-11-15 10:28:18 +0000195 void SynthMsgSendSuperFunctionDecl();
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +0000196 void SynthMsgSendStretFunctionDecl();
Fariborz Jahanianacb49772007-12-03 21:26:48 +0000197 void SynthMsgSendFpretFunctionDecl();
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +0000198 void SynthMsgSendSuperStretFunctionDecl();
Steve Naroff09b266e2007-10-30 23:14:51 +0000199 void SynthGetClassFunctionDecl();
Steve Naroff9bcb5fc2007-12-07 03:50:46 +0000200 void SynthGetMetaClassFunctionDecl();
Fariborz Jahaniana70711b2007-12-04 21:47:40 +0000201 void SynthSelGetUidFunctionDecl();
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +0000202 void SynthGetProtocolFunctionDecl();
Steve Naroffc0a123c2008-03-11 17:37:02 +0000203 void SynthSuperContructorFunctionDecl();
Steve Naroff96984642007-11-08 14:30:50 +0000204
Chris Lattnerf04da132007-10-24 17:06:59 +0000205 // Metadata emission.
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000206 void RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl,
Fariborz Jahanianccd87b02007-10-25 20:55:25 +0000207 std::string &Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +0000208
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000209 void RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *CDecl,
Fariborz Jahanianccd87b02007-10-25 20:55:25 +0000210 std::string &Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +0000211
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000212 typedef ObjCCategoryImplDecl::instmeth_iterator instmeth_iterator;
213 void RewriteObjCMethodsMetaData(instmeth_iterator MethodBegin,
Chris Lattnerab4c4d52007-12-12 07:46:12 +0000214 instmeth_iterator MethodEnd,
Fariborz Jahanian8e991ba2007-10-25 00:14:44 +0000215 bool IsInstanceMethod,
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +0000216 const char *prefix,
Chris Lattner158ecb92007-10-25 17:07:24 +0000217 const char *ClassName,
218 std::string &Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +0000219
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000220 void RewriteObjCProtocolsMetaData(ObjCProtocolDecl **Protocols,
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +0000221 int NumProtocols,
222 const char *prefix,
Fariborz Jahanianccd87b02007-10-25 20:55:25 +0000223 const char *ClassName,
224 std::string &Result);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000225 void SynthesizeObjCInternalStruct(ObjCInterfaceDecl *CDecl,
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +0000226 std::string &Result);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000227 void SynthesizeIvarOffsetComputation(ObjCImplementationDecl *IDecl,
228 ObjCIvarDecl *ivar,
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +0000229 std::string &Result);
Fariborz Jahanian7a3279d2007-11-13 19:21:13 +0000230 void RewriteImplementations(std::string &Result);
Chris Lattner77cd2a02007-10-11 00:43:27 +0000231 };
232}
233
Fariborz Jahanianb4b2f0c2008-01-18 01:15:54 +0000234static bool IsHeaderFile(const std::string &Filename) {
235 std::string::size_type DotPos = Filename.rfind('.');
236
237 if (DotPos == std::string::npos) {
238 // no file extension
239 return false;
240 }
241
242 std::string Ext = std::string(Filename.begin()+DotPos+1, Filename.end());
243 // C header: .h
244 // C++ header: .hh or .H;
245 return Ext == "h" || Ext == "hh" || Ext == "H";
246}
247
Steve Naroffb29b4272008-04-14 22:03:09 +0000248RewriteObjC::RewriteObjC(std::string inFile, std::string outFile,
Steve Naroffa7b402d2008-03-28 22:26:09 +0000249 Diagnostic &D, const LangOptions &LOpts)
250 : Diags(D), LangOpts(LOpts) {
251 IsHeader = IsHeaderFile(inFile);
252 InFileName = inFile;
253 OutFileName = outFile;
254 RewriteFailedDiag = Diags.getCustomDiagID(Diagnostic::Warning,
255 "rewriting sub-expression within a macro (may not be correct)");
256}
257
Fariborz Jahanianb4b2f0c2008-01-18 01:15:54 +0000258ASTConsumer *clang::CreateCodeRewriterTest(const std::string& InFile,
Chris Lattnerc68ab772008-03-22 00:08:40 +0000259 const std::string& OutFile,
Steve Naroff4f943c22008-03-10 20:43:59 +0000260 Diagnostic &Diags,
261 const LangOptions &LOpts) {
Steve Naroffb29b4272008-04-14 22:03:09 +0000262 return new RewriteObjC(InFile, OutFile, Diags, LOpts);
Chris Lattnere365c502007-11-30 22:25:36 +0000263}
Chris Lattner77cd2a02007-10-11 00:43:27 +0000264
Steve Naroffb29b4272008-04-14 22:03:09 +0000265void RewriteObjC::Initialize(ASTContext &context) {
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000266 Context = &context;
267 SM = &Context->getSourceManager();
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +0000268 TUDecl = Context->getTranslationUnitDecl();
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000269 MsgSendFunctionDecl = 0;
270 MsgSendSuperFunctionDecl = 0;
271 MsgSendStretFunctionDecl = 0;
272 MsgSendSuperStretFunctionDecl = 0;
273 MsgSendFpretFunctionDecl = 0;
274 GetClassFunctionDecl = 0;
275 GetMetaClassFunctionDecl = 0;
276 SelGetUidFunctionDecl = 0;
277 CFStringFunctionDecl = 0;
278 GetProtocolFunctionDecl = 0;
279 ConstantStringClassReference = 0;
280 NSStringRecord = 0;
281 CurMethodDecl = 0;
282 SuperStructDecl = 0;
Steve Naroff9630ec52008-03-27 22:59:54 +0000283 ConstantStringDecl = 0;
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000284 BcLabelCount = 0;
Steve Naroffc0a123c2008-03-11 17:37:02 +0000285 SuperContructorFunctionDecl = 0;
Steve Naroffd82a9ab2008-03-15 00:55:56 +0000286 NumObjCStringLiterals = 0;
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000287
288 // Get the ID and start/end of the main file.
289 MainFileID = SM->getMainFileID();
290 const llvm::MemoryBuffer *MainBuf = SM->getBuffer(MainFileID);
291 MainFileStart = MainBuf->getBufferStart();
292 MainFileEnd = MainBuf->getBufferEnd();
Steve Naroffba92b2e2008-03-27 22:29:16 +0000293
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000294 Rewrite.setSourceMgr(Context->getSourceManager());
Steve Naroff61ed9ca2008-03-10 23:16:54 +0000295
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000296 // declaring objc_selector outside the parameter list removes a silly
297 // scope related warning...
Steve Naroffba92b2e2008-03-27 22:29:16 +0000298 if (IsHeader)
299 Preamble = "#pragma once\n";
300 Preamble += "struct objc_selector; struct objc_class;\n";
301 Preamble += "#ifndef OBJC_SUPER\n";
302 Preamble += "struct objc_super { struct objc_object *object; ";
303 Preamble += "struct objc_object *superClass; ";
Steve Naroffc0a123c2008-03-11 17:37:02 +0000304 if (LangOpts.Microsoft) {
305 // Add a constructor for creating temporary objects.
Steve Naroffba92b2e2008-03-27 22:29:16 +0000306 Preamble += "objc_super(struct objc_object *o, struct objc_object *s) : ";
307 Preamble += "object(o), superClass(s) {} ";
Steve Naroffc0a123c2008-03-11 17:37:02 +0000308 }
Steve Naroffba92b2e2008-03-27 22:29:16 +0000309 Preamble += "};\n";
310 Preamble += "#define OBJC_SUPER\n";
311 Preamble += "#endif\n";
312 Preamble += "#ifndef _REWRITER_typedef_Protocol\n";
313 Preamble += "typedef struct objc_object Protocol;\n";
314 Preamble += "#define _REWRITER_typedef_Protocol\n";
315 Preamble += "#endif\n";
Steve Naroff3c64d9e2008-03-12 13:19:12 +0000316 if (LangOpts.Microsoft)
Steve Naroff69c827f2008-05-06 22:45:19 +0000317 Preamble += "#define __OBJC_RW_EXTERN extern \"C\" __declspec(dllimport)\n";
318 else
319 Preamble += "#define __OBJC_RW_EXTERN extern\n";
320 Preamble += "__OBJC_RW_EXTERN struct objc_object *objc_msgSend";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000321 Preamble += "(struct objc_object *, struct objc_selector *, ...);\n";
Steve Naroff69c827f2008-05-06 22:45:19 +0000322 Preamble += "__OBJC_RW_EXTERN struct objc_object *objc_msgSendSuper";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000323 Preamble += "(struct objc_super *, struct objc_selector *, ...);\n";
Steve Naroff69c827f2008-05-06 22:45:19 +0000324 Preamble += "__OBJC_RW_EXTERN struct objc_object *objc_msgSend_stret";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000325 Preamble += "(struct objc_object *, struct objc_selector *, ...);\n";
Steve Naroff69c827f2008-05-06 22:45:19 +0000326 Preamble += "__OBJC_RW_EXTERN struct objc_object *objc_msgSendSuper_stret";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000327 Preamble += "(struct objc_super *, struct objc_selector *, ...);\n";
Steve Naroff1284db82008-05-08 22:02:18 +0000328 Preamble += "__OBJC_RW_EXTERN double objc_msgSend_fpret";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000329 Preamble += "(struct objc_object *, struct objc_selector *, ...);\n";
Steve Narofff0757612008-05-08 17:52:16 +0000330 Preamble += "__OBJC_RW_EXTERN struct objc_object *objc_getClass";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000331 Preamble += "(const char *);\n";
Steve Naroff69c827f2008-05-06 22:45:19 +0000332 Preamble += "__OBJC_RW_EXTERN struct objc_object *objc_getMetaClass";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000333 Preamble += "(const char *);\n";
Steve Naroff69c827f2008-05-06 22:45:19 +0000334 Preamble += "__OBJC_RW_EXTERN void objc_exception_throw(struct objc_object *);\n";
335 Preamble += "__OBJC_RW_EXTERN void objc_exception_try_enter(void *);\n";
336 Preamble += "__OBJC_RW_EXTERN void objc_exception_try_exit(void *);\n";
337 Preamble += "__OBJC_RW_EXTERN struct objc_object *objc_exception_extract(void *);\n";
338 Preamble += "__OBJC_RW_EXTERN int objc_exception_match";
Steve Naroff580ca782008-05-09 21:17:56 +0000339 Preamble += "(struct objc_class *, struct objc_object *);\n";
Steve Naroff69c827f2008-05-06 22:45:19 +0000340 Preamble += "__OBJC_RW_EXTERN Protocol *objc_getProtocol(const char *);\n";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000341 Preamble += "#ifndef __FASTENUMERATIONSTATE\n";
342 Preamble += "struct __objcFastEnumerationState {\n\t";
343 Preamble += "unsigned long state;\n\t";
Steve Naroffb10f2732008-04-04 22:58:22 +0000344 Preamble += "void **itemsPtr;\n\t";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000345 Preamble += "unsigned long *mutationsPtr;\n\t";
346 Preamble += "unsigned long extra[5];\n};\n";
Steve Naroff73ebd6d2008-06-02 20:23:21 +0000347 Preamble += "__OBJC_RW_EXTERN void objc_enumerationMutation(struct objc_object *);\n";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000348 Preamble += "#define __FASTENUMERATIONSTATE\n";
349 Preamble += "#endif\n";
350 Preamble += "#ifndef __NSCONSTANTSTRINGIMPL\n";
351 Preamble += "struct __NSConstantStringImpl {\n";
352 Preamble += " int *isa;\n";
353 Preamble += " int flags;\n";
354 Preamble += " char *str;\n";
355 Preamble += " long length;\n";
356 Preamble += "};\n";
Steve Naroff73b17cd2008-05-15 21:12:10 +0000357 Preamble += "__OBJC_RW_EXTERN int __CFConstantStringClassReference[];\n";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000358 Preamble += "#define __NSCONSTANTSTRINGIMPL\n";
359 Preamble += "#endif\n";
Steve Naroff73b17cd2008-05-15 21:12:10 +0000360 if (LangOpts.Microsoft) {
361 Preamble += "#undef __OBJC_RW_EXTERN\n";
Steve Naroffba92b2e2008-03-27 22:29:16 +0000362 Preamble += "#define __attribute__(X)\n";
Steve Naroff73b17cd2008-05-15 21:12:10 +0000363 }
Chris Lattner9e13c2e2008-01-31 19:38:44 +0000364}
365
366
Chris Lattnerf04da132007-10-24 17:06:59 +0000367//===----------------------------------------------------------------------===//
368// Top Level Driver Code
369//===----------------------------------------------------------------------===//
370
Steve Naroffb29b4272008-04-14 22:03:09 +0000371void RewriteObjC::HandleTopLevelDecl(Decl *D) {
Chris Lattner2c64b7b2007-10-16 21:07:07 +0000372 // Two cases: either the decl could be in the main file, or it could be in a
373 // #included file. If the former, rewrite it now. If the later, check to see
374 // if we rewrote the #include/#import.
375 SourceLocation Loc = D->getLocation();
376 Loc = SM->getLogicalLoc(Loc);
377
378 // If this is for a builtin, ignore it.
379 if (Loc.isInvalid()) return;
380
Steve Naroffebf2b562007-10-23 23:50:29 +0000381 // Look for built-in declarations that we need to refer during the rewrite.
382 if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
Steve Naroff09b266e2007-10-30 23:14:51 +0000383 RewriteFunctionDecl(FD);
Steve Naroff248a7532008-04-15 22:42:06 +0000384 } else if (VarDecl *FVD = dyn_cast<VarDecl>(D)) {
Steve Naroffbeaf2992007-11-03 11:27:19 +0000385 // declared in <Foundation/NSString.h>
386 if (strcmp(FVD->getName(), "_NSConstantStringClassReference") == 0) {
387 ConstantStringClassReference = FVD;
388 return;
389 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000390 } else if (ObjCInterfaceDecl *MD = dyn_cast<ObjCInterfaceDecl>(D)) {
Steve Naroffbef11852007-10-26 20:53:56 +0000391 RewriteInterfaceDecl(MD);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000392 } else if (ObjCCategoryDecl *CD = dyn_cast<ObjCCategoryDecl>(D)) {
Steve Naroff423cb562007-10-30 13:30:57 +0000393 RewriteCategoryDecl(CD);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000394 } else if (ObjCProtocolDecl *PD = dyn_cast<ObjCProtocolDecl>(D)) {
Steve Naroff752d6ef2007-10-30 16:42:30 +0000395 RewriteProtocolDecl(PD);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000396 } else if (ObjCForwardProtocolDecl *FP =
397 dyn_cast<ObjCForwardProtocolDecl>(D)){
Fariborz Jahaniand175ddf2007-11-14 00:42:16 +0000398 RewriteForwardProtocolDecl(FP);
Steve Naroffebf2b562007-10-23 23:50:29 +0000399 }
Chris Lattnerf04da132007-10-24 17:06:59 +0000400 // If we have a decl in the main file, see if we should rewrite it.
Ted Kremenekcf7e9582008-04-14 21:24:13 +0000401 if (SM->isFromMainFile(Loc))
Chris Lattner2c64b7b2007-10-16 21:07:07 +0000402 return HandleDeclInMainFile(D);
Chris Lattner2c64b7b2007-10-16 21:07:07 +0000403}
404
Chris Lattnerf04da132007-10-24 17:06:59 +0000405/// HandleDeclInMainFile - This is called for each top-level decl defined in the
406/// main file of the input.
Steve Naroffb29b4272008-04-14 22:03:09 +0000407void RewriteObjC::HandleDeclInMainFile(Decl *D) {
Chris Lattnerf04da132007-10-24 17:06:59 +0000408 if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
409 if (Stmt *Body = FD->getBody())
Steve Narofff3473a72007-11-09 15:20:18 +0000410 FD->setBody(RewriteFunctionBodyOrGlobalInitializer(Body));
Steve Narofff69cc5d2008-01-30 19:17:43 +0000411
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000412 if (ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D)) {
Steve Naroff874e2322007-11-15 10:28:18 +0000413 if (Stmt *Body = MD->getBody()) {
414 //Body->dump();
415 CurMethodDecl = MD;
Steve Naroff71c0a952007-11-13 23:01:27 +0000416 MD->setBody(RewriteFunctionBodyOrGlobalInitializer(Body));
Steve Naroff874e2322007-11-15 10:28:18 +0000417 CurMethodDecl = 0;
418 }
Steve Naroff71c0a952007-11-13 23:01:27 +0000419 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000420 if (ObjCImplementationDecl *CI = dyn_cast<ObjCImplementationDecl>(D))
Chris Lattnerf04da132007-10-24 17:06:59 +0000421 ClassImplementation.push_back(CI);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000422 else if (ObjCCategoryImplDecl *CI = dyn_cast<ObjCCategoryImplDecl>(D))
Chris Lattnerf04da132007-10-24 17:06:59 +0000423 CategoryImplementation.push_back(CI);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000424 else if (ObjCClassDecl *CD = dyn_cast<ObjCClassDecl>(D))
Chris Lattnerf04da132007-10-24 17:06:59 +0000425 RewriteForwardClassDecl(CD);
Steve Narofff3473a72007-11-09 15:20:18 +0000426 else if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000427 RewriteObjCQualifiedInterfaceTypes(VD);
Steve Narofff3473a72007-11-09 15:20:18 +0000428 if (VD->getInit())
429 RewriteFunctionBodyOrGlobalInitializer(VD->getInit());
430 }
Chris Lattnerf04da132007-10-24 17:06:59 +0000431 // Nothing yet.
432}
433
Steve Naroffb29b4272008-04-14 22:03:09 +0000434RewriteObjC::~RewriteObjC() {
Chris Lattnerf04da132007-10-24 17:06:59 +0000435 // Get the top-level buffer that this corresponds to.
Chris Lattner74a0c772007-11-08 04:27:23 +0000436
437 // Rewrite tabs if we care.
438 //RewriteTabs();
Chris Lattnerf04da132007-10-24 17:06:59 +0000439
Steve Naroffa7b402d2008-03-28 22:26:09 +0000440 if (Diags.hasErrorOccurred())
441 return;
442
443 // Create the output file.
444
445 std::ostream *OutFile;
446 if (OutFileName == "-") {
447 OutFile = llvm::cout.stream();
448 } else if (!OutFileName.empty()) {
449 OutFile = new std::ofstream(OutFileName.c_str(),
450 std::ios_base::binary|std::ios_base::out);
451 } else if (InFileName == "-") {
452 OutFile = llvm::cout.stream();
453 } else {
454 llvm::sys::Path Path(InFileName);
455 Path.eraseSuffix();
456 Path.appendSuffix("cpp");
457 OutFile = new std::ofstream(Path.toString().c_str(),
458 std::ios_base::binary|std::ios_base::out);
459 }
460
Fariborz Jahanian452b8992008-01-19 00:30:35 +0000461 RewriteInclude();
462
Steve Naroffba92b2e2008-03-27 22:29:16 +0000463 InsertText(SourceLocation::getFileLoc(MainFileID, 0),
464 Preamble.c_str(), Preamble.size(), false);
465
Fariborz Jahanian909f02a2007-11-05 17:47:33 +0000466 // Rewrite Objective-c meta data*
467 std::string ResultStr;
Fariborz Jahanian7a3279d2007-11-13 19:21:13 +0000468 RewriteImplementations(ResultStr);
Fariborz Jahanian909f02a2007-11-05 17:47:33 +0000469
Chris Lattnerf04da132007-10-24 17:06:59 +0000470 // Get the buffer corresponding to MainFileID. If we haven't changed it, then
471 // we are done.
472 if (const RewriteBuffer *RewriteBuf =
473 Rewrite.getRewriteBufferFor(MainFileID)) {
Steve Naroffbeaf2992007-11-03 11:27:19 +0000474 //printf("Changed:\n");
Steve Naroffa7b402d2008-03-28 22:26:09 +0000475 *OutFile << std::string(RewriteBuf->begin(), RewriteBuf->end());
Chris Lattnerf04da132007-10-24 17:06:59 +0000476 } else {
Chris Lattnerc68ab772008-03-22 00:08:40 +0000477 fprintf(stderr, "No changes\n");
Chris Lattnerf04da132007-10-24 17:06:59 +0000478 }
Fariborz Jahanian4402d812007-11-07 18:40:28 +0000479 // Emit metadata.
Steve Naroffa7b402d2008-03-28 22:26:09 +0000480 *OutFile << ResultStr;
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +0000481}
482
Chris Lattnerf04da132007-10-24 17:06:59 +0000483//===----------------------------------------------------------------------===//
484// Syntactic (non-AST) Rewriting Code
485//===----------------------------------------------------------------------===//
486
Steve Naroffb29b4272008-04-14 22:03:09 +0000487void RewriteObjC::RewriteInclude() {
Fariborz Jahanian452b8992008-01-19 00:30:35 +0000488 SourceLocation LocStart = SourceLocation::getFileLoc(MainFileID, 0);
489 std::pair<const char*, const char*> MainBuf = SM->getBufferData(MainFileID);
490 const char *MainBufStart = MainBuf.first;
491 const char *MainBufEnd = MainBuf.second;
492 size_t ImportLen = strlen("import");
493 size_t IncludeLen = strlen("include");
494
Fariborz Jahanianaf57b462008-01-19 01:03:17 +0000495 // Loop over the whole file, looking for includes.
Fariborz Jahanian452b8992008-01-19 00:30:35 +0000496 for (const char *BufPtr = MainBufStart; BufPtr < MainBufEnd; ++BufPtr) {
497 if (*BufPtr == '#') {
498 if (++BufPtr == MainBufEnd)
499 return;
500 while (*BufPtr == ' ' || *BufPtr == '\t')
501 if (++BufPtr == MainBufEnd)
502 return;
503 if (!strncmp(BufPtr, "import", ImportLen)) {
504 // replace import with include
505 SourceLocation ImportLoc =
506 LocStart.getFileLocWithOffset(BufPtr-MainBufStart);
Chris Lattneraadaf782008-01-31 19:51:04 +0000507 ReplaceText(ImportLoc, ImportLen, "include", IncludeLen);
Fariborz Jahanian452b8992008-01-19 00:30:35 +0000508 BufPtr += ImportLen;
509 }
510 }
511 }
Chris Lattner2c64b7b2007-10-16 21:07:07 +0000512}
513
Steve Naroffb29b4272008-04-14 22:03:09 +0000514void RewriteObjC::RewriteTabs() {
Chris Lattnerf04da132007-10-24 17:06:59 +0000515 std::pair<const char*, const char*> MainBuf = SM->getBufferData(MainFileID);
516 const char *MainBufStart = MainBuf.first;
517 const char *MainBufEnd = MainBuf.second;
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +0000518
Chris Lattnerf04da132007-10-24 17:06:59 +0000519 // Loop over the whole file, looking for tabs.
520 for (const char *BufPtr = MainBufStart; BufPtr != MainBufEnd; ++BufPtr) {
521 if (*BufPtr != '\t')
522 continue;
523
524 // Okay, we found a tab. This tab will turn into at least one character,
525 // but it depends on which 'virtual column' it is in. Compute that now.
526 unsigned VCol = 0;
527 while (BufPtr-VCol != MainBufStart && BufPtr[-VCol-1] != '\t' &&
528 BufPtr[-VCol-1] != '\n' && BufPtr[-VCol-1] != '\r')
529 ++VCol;
530
531 // Okay, now that we know the virtual column, we know how many spaces to
532 // insert. We assume 8-character tab-stops.
533 unsigned Spaces = 8-(VCol & 7);
534
535 // Get the location of the tab.
536 SourceLocation TabLoc =
537 SourceLocation::getFileLoc(MainFileID, BufPtr-MainBufStart);
538
539 // Rewrite the single tab character into a sequence of spaces.
Chris Lattneraadaf782008-01-31 19:51:04 +0000540 ReplaceText(TabLoc, 1, " ", Spaces);
Chris Lattnerf04da132007-10-24 17:06:59 +0000541 }
Chris Lattner8a12c272007-10-11 18:38:32 +0000542}
543
544
Steve Naroffb29b4272008-04-14 22:03:09 +0000545void RewriteObjC::RewriteForwardClassDecl(ObjCClassDecl *ClassDecl) {
Chris Lattnerf04da132007-10-24 17:06:59 +0000546 int numDecls = ClassDecl->getNumForwardDecls();
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000547 ObjCInterfaceDecl **ForwardDecls = ClassDecl->getForwardDecls();
Chris Lattnerf04da132007-10-24 17:06:59 +0000548
549 // Get the start location and compute the semi location.
550 SourceLocation startLoc = ClassDecl->getLocation();
551 const char *startBuf = SM->getCharacterData(startLoc);
552 const char *semiPtr = strchr(startBuf, ';');
553
554 // Translate to typedef's that forward reference structs with the same name
555 // as the class. As a convenience, we include the original declaration
556 // as a comment.
557 std::string typedefString;
558 typedefString += "// ";
Steve Naroff934f2762007-10-24 22:48:43 +0000559 typedefString.append(startBuf, semiPtr-startBuf+1);
560 typedefString += "\n";
561 for (int i = 0; i < numDecls; i++) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000562 ObjCInterfaceDecl *ForwardDecl = ForwardDecls[i];
Steve Naroff32174822007-11-09 12:50:28 +0000563 typedefString += "#ifndef _REWRITER_typedef_";
564 typedefString += ForwardDecl->getName();
565 typedefString += "\n";
566 typedefString += "#define _REWRITER_typedef_";
567 typedefString += ForwardDecl->getName();
568 typedefString += "\n";
Steve Naroff352336b2007-11-05 14:36:37 +0000569 typedefString += "typedef struct objc_object ";
Steve Naroff934f2762007-10-24 22:48:43 +0000570 typedefString += ForwardDecl->getName();
Steve Naroff32174822007-11-09 12:50:28 +0000571 typedefString += ";\n#endif\n";
Steve Naroff934f2762007-10-24 22:48:43 +0000572 }
573
574 // Replace the @class with typedefs corresponding to the classes.
Chris Lattneraadaf782008-01-31 19:51:04 +0000575 ReplaceText(startLoc, semiPtr-startBuf+1,
576 typedefString.c_str(), typedefString.size());
Chris Lattnerf04da132007-10-24 17:06:59 +0000577}
578
Steve Naroffb29b4272008-04-14 22:03:09 +0000579void RewriteObjC::RewriteMethodDeclaration(ObjCMethodDecl *Method) {
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000580 SourceLocation LocStart = Method->getLocStart();
581 SourceLocation LocEnd = Method->getLocEnd();
Steve Naroff423cb562007-10-30 13:30:57 +0000582
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000583 if (SM->getLineNumber(LocEnd) > SM->getLineNumber(LocStart)) {
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000584 InsertText(LocStart, "/* ", 3);
Chris Lattneraadaf782008-01-31 19:51:04 +0000585 ReplaceText(LocEnd, 1, ";*/ ", 4);
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000586 } else {
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000587 InsertText(LocStart, "// ", 3);
Steve Naroff423cb562007-10-30 13:30:57 +0000588 }
589}
590
Steve Naroffb29b4272008-04-14 22:03:09 +0000591void RewriteObjC::RewriteProperties(unsigned nProperties, ObjCPropertyDecl **Properties)
Fariborz Jahanian957cf652007-11-07 00:09:37 +0000592{
Chris Lattner55d13b42008-03-16 21:23:50 +0000593 for (unsigned i = 0; i < nProperties; i++) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000594 ObjCPropertyDecl *Property = Properties[i];
Fariborz Jahanian957cf652007-11-07 00:09:37 +0000595 SourceLocation Loc = Property->getLocation();
596
Chris Lattneraadaf782008-01-31 19:51:04 +0000597 ReplaceText(Loc, 0, "// ", 3);
Fariborz Jahanian957cf652007-11-07 00:09:37 +0000598
599 // FIXME: handle properties that are declared across multiple lines.
600 }
601}
602
Steve Naroffb29b4272008-04-14 22:03:09 +0000603void RewriteObjC::RewriteCategoryDecl(ObjCCategoryDecl *CatDecl) {
Steve Naroff423cb562007-10-30 13:30:57 +0000604 SourceLocation LocStart = CatDecl->getLocStart();
605
606 // FIXME: handle category headers that are declared across multiple lines.
Chris Lattneraadaf782008-01-31 19:51:04 +0000607 ReplaceText(LocStart, 0, "// ", 3);
Steve Naroff423cb562007-10-30 13:30:57 +0000608
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000609 for (ObjCCategoryDecl::instmeth_iterator I = CatDecl->instmeth_begin(),
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000610 E = CatDecl->instmeth_end(); I != E; ++I)
611 RewriteMethodDeclaration(*I);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000612 for (ObjCCategoryDecl::classmeth_iterator I = CatDecl->classmeth_begin(),
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000613 E = CatDecl->classmeth_end(); I != E; ++I)
614 RewriteMethodDeclaration(*I);
615
Steve Naroff423cb562007-10-30 13:30:57 +0000616 // Lastly, comment out the @end.
Chris Lattneraadaf782008-01-31 19:51:04 +0000617 ReplaceText(CatDecl->getAtEndLoc(), 0, "// ", 3);
Steve Naroff423cb562007-10-30 13:30:57 +0000618}
619
Steve Naroffb29b4272008-04-14 22:03:09 +0000620void RewriteObjC::RewriteProtocolDecl(ObjCProtocolDecl *PDecl) {
Fariborz Jahanianb82b3ea2007-11-14 01:37:46 +0000621 std::pair<const char*, const char*> MainBuf = SM->getBufferData(MainFileID);
Fariborz Jahanianb82b3ea2007-11-14 01:37:46 +0000622
Steve Naroff752d6ef2007-10-30 16:42:30 +0000623 SourceLocation LocStart = PDecl->getLocStart();
624
625 // FIXME: handle protocol headers that are declared across multiple lines.
Chris Lattneraadaf782008-01-31 19:51:04 +0000626 ReplaceText(LocStart, 0, "// ", 3);
Steve Naroff752d6ef2007-10-30 16:42:30 +0000627
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000628 for (ObjCProtocolDecl::instmeth_iterator I = PDecl->instmeth_begin(),
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000629 E = PDecl->instmeth_end(); I != E; ++I)
630 RewriteMethodDeclaration(*I);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000631 for (ObjCProtocolDecl::classmeth_iterator I = PDecl->classmeth_begin(),
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000632 E = PDecl->classmeth_end(); I != E; ++I)
633 RewriteMethodDeclaration(*I);
634
Steve Naroff752d6ef2007-10-30 16:42:30 +0000635 // Lastly, comment out the @end.
Fariborz Jahanianb82b3ea2007-11-14 01:37:46 +0000636 SourceLocation LocEnd = PDecl->getAtEndLoc();
Chris Lattneraadaf782008-01-31 19:51:04 +0000637 ReplaceText(LocEnd, 0, "// ", 3);
Steve Naroff8cc764c2007-11-14 15:03:57 +0000638
Fariborz Jahanianb82b3ea2007-11-14 01:37:46 +0000639 // Must comment out @optional/@required
640 const char *startBuf = SM->getCharacterData(LocStart);
641 const char *endBuf = SM->getCharacterData(LocEnd);
642 for (const char *p = startBuf; p < endBuf; p++) {
643 if (*p == '@' && !strncmp(p+1, "optional", strlen("optional"))) {
644 std::string CommentedOptional = "/* @optional */";
Steve Naroff8cc764c2007-11-14 15:03:57 +0000645 SourceLocation OptionalLoc = LocStart.getFileLocWithOffset(p-startBuf);
Chris Lattneraadaf782008-01-31 19:51:04 +0000646 ReplaceText(OptionalLoc, strlen("@optional"),
647 CommentedOptional.c_str(), CommentedOptional.size());
Fariborz Jahanianb82b3ea2007-11-14 01:37:46 +0000648
649 }
650 else if (*p == '@' && !strncmp(p+1, "required", strlen("required"))) {
651 std::string CommentedRequired = "/* @required */";
Steve Naroff8cc764c2007-11-14 15:03:57 +0000652 SourceLocation OptionalLoc = LocStart.getFileLocWithOffset(p-startBuf);
Chris Lattneraadaf782008-01-31 19:51:04 +0000653 ReplaceText(OptionalLoc, strlen("@required"),
654 CommentedRequired.c_str(), CommentedRequired.size());
Fariborz Jahanianb82b3ea2007-11-14 01:37:46 +0000655
656 }
657 }
Steve Naroff752d6ef2007-10-30 16:42:30 +0000658}
659
Steve Naroffb29b4272008-04-14 22:03:09 +0000660void RewriteObjC::RewriteForwardProtocolDecl(ObjCForwardProtocolDecl *PDecl) {
Fariborz Jahaniand175ddf2007-11-14 00:42:16 +0000661 SourceLocation LocStart = PDecl->getLocation();
Steve Naroffb7fa9922007-11-14 03:37:28 +0000662 if (LocStart.isInvalid())
663 assert(false && "Invalid SourceLocation");
Fariborz Jahaniand175ddf2007-11-14 00:42:16 +0000664 // FIXME: handle forward protocol that are declared across multiple lines.
Chris Lattneraadaf782008-01-31 19:51:04 +0000665 ReplaceText(LocStart, 0, "// ", 3);
Fariborz Jahaniand175ddf2007-11-14 00:42:16 +0000666}
667
Steve Naroffb29b4272008-04-14 22:03:09 +0000668void RewriteObjC::RewriteObjCMethodDecl(ObjCMethodDecl *OMD,
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000669 std::string &ResultStr) {
670 ResultStr += "\nstatic ";
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000671 if (OMD->getResultType()->isObjCQualifiedIdType())
Fariborz Jahanianc5692492007-12-17 21:03:50 +0000672 ResultStr += "id";
673 else
674 ResultStr += OMD->getResultType().getAsString();
Fariborz Jahanian531a1ea2008-01-10 01:39:52 +0000675 ResultStr += " ";
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000676
677 // Unique method name
Fariborz Jahanianb7908b52007-11-13 21:02:00 +0000678 std::string NameStr;
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000679
Fariborz Jahanianb7908b52007-11-13 21:02:00 +0000680 if (OMD->isInstance())
681 NameStr += "_I_";
682 else
683 NameStr += "_C_";
684
685 NameStr += OMD->getClassInterface()->getName();
686 NameStr += "_";
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000687
688 NamedDecl *MethodContext = OMD->getMethodContext();
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000689 if (ObjCCategoryImplDecl *CID =
690 dyn_cast<ObjCCategoryImplDecl>(MethodContext)) {
Fariborz Jahanianb7908b52007-11-13 21:02:00 +0000691 NameStr += CID->getName();
692 NameStr += "_";
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000693 }
Steve Naroff563b8282008-04-04 22:23:44 +0000694 // Append selector names, replacing ':' with '_'
695 if (OMD->getSelector().getName().find(':') == std::string::npos)
Fariborz Jahanianb7908b52007-11-13 21:02:00 +0000696 NameStr += OMD->getSelector().getName();
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000697 else {
698 std::string selString = OMD->getSelector().getName();
699 int len = selString.size();
700 for (int i = 0; i < len; i++)
701 if (selString[i] == ':')
702 selString[i] = '_';
Fariborz Jahanianb7908b52007-11-13 21:02:00 +0000703 NameStr += selString;
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000704 }
Fariborz Jahanianb7908b52007-11-13 21:02:00 +0000705 // Remember this name for metadata emission
706 MethodInternalNames[OMD] = NameStr;
707 ResultStr += NameStr;
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000708
709 // Rewrite arguments
710 ResultStr += "(";
711
712 // invisible arguments
713 if (OMD->isInstance()) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000714 QualType selfTy = Context->getObjCInterfaceType(OMD->getClassInterface());
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000715 selfTy = Context->getPointerType(selfTy);
Steve Naroff05b8c782008-03-12 00:25:36 +0000716 if (!LangOpts.Microsoft) {
717 if (ObjCSynthesizedStructs.count(OMD->getClassInterface()))
718 ResultStr += "struct ";
719 }
720 // When rewriting for Microsoft, explicitly omit the structure name.
Steve Naroff61ed9ca2008-03-10 23:16:54 +0000721 ResultStr += OMD->getClassInterface()->getName();
Steve Naroff61ed9ca2008-03-10 23:16:54 +0000722 ResultStr += " *";
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000723 }
724 else
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000725 ResultStr += Context->getObjCIdType().getAsString();
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000726
727 ResultStr += " self, ";
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000728 ResultStr += Context->getObjCSelType().getAsString();
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000729 ResultStr += " _cmd";
730
731 // Method arguments.
Chris Lattner58cce3b2008-03-16 01:07:14 +0000732 for (unsigned i = 0; i < OMD->getNumParams(); i++) {
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000733 ParmVarDecl *PDecl = OMD->getParamDecl(i);
734 ResultStr += ", ";
Steve Naroff543409e2008-04-18 21:13:19 +0000735 if (PDecl->getType()->isObjCQualifiedIdType()) {
736 ResultStr += "id ";
737 ResultStr += PDecl->getName();
738 } else {
739 std::string Name = PDecl->getName();
740 PDecl->getType().getAsStringInternal(Name);
741 ResultStr += Name;
742 }
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000743 }
Fariborz Jahanian7c39ff72008-01-21 20:14:23 +0000744 if (OMD->isVariadic())
745 ResultStr += ", ...";
Fariborz Jahanian531a1ea2008-01-10 01:39:52 +0000746 ResultStr += ") ";
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000747
748}
Steve Naroffb29b4272008-04-14 22:03:09 +0000749void RewriteObjC::RewriteImplementationDecl(NamedDecl *OID) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000750 ObjCImplementationDecl *IMD = dyn_cast<ObjCImplementationDecl>(OID);
751 ObjCCategoryImplDecl *CID = dyn_cast<ObjCCategoryImplDecl>(OID);
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000752
Fariborz Jahanian66d6b292007-11-13 20:04:28 +0000753 if (IMD)
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000754 InsertText(IMD->getLocStart(), "// ", 3);
Fariborz Jahanian66d6b292007-11-13 20:04:28 +0000755 else
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000756 InsertText(CID->getLocStart(), "// ", 3);
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000757
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000758 for (ObjCCategoryImplDecl::instmeth_iterator
Chris Lattnerab4c4d52007-12-12 07:46:12 +0000759 I = IMD ? IMD->instmeth_begin() : CID->instmeth_begin(),
760 E = IMD ? IMD->instmeth_end() : CID->instmeth_end(); I != E; ++I) {
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000761 std::string ResultStr;
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000762 ObjCMethodDecl *OMD = *I;
763 RewriteObjCMethodDecl(OMD, ResultStr);
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000764 SourceLocation LocStart = OMD->getLocStart();
765 SourceLocation LocEnd = OMD->getBody()->getLocStart();
766
767 const char *startBuf = SM->getCharacterData(LocStart);
768 const char *endBuf = SM->getCharacterData(LocEnd);
Chris Lattneraadaf782008-01-31 19:51:04 +0000769 ReplaceText(LocStart, endBuf-startBuf,
770 ResultStr.c_str(), ResultStr.size());
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000771 }
772
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000773 for (ObjCCategoryImplDecl::classmeth_iterator
Chris Lattnerab4c4d52007-12-12 07:46:12 +0000774 I = IMD ? IMD->classmeth_begin() : CID->classmeth_begin(),
775 E = IMD ? IMD->classmeth_end() : CID->classmeth_end(); I != E; ++I) {
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000776 std::string ResultStr;
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000777 ObjCMethodDecl *OMD = *I;
778 RewriteObjCMethodDecl(OMD, ResultStr);
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000779 SourceLocation LocStart = OMD->getLocStart();
780 SourceLocation LocEnd = OMD->getBody()->getLocStart();
781
782 const char *startBuf = SM->getCharacterData(LocStart);
783 const char *endBuf = SM->getCharacterData(LocEnd);
Chris Lattneraadaf782008-01-31 19:51:04 +0000784 ReplaceText(LocStart, endBuf-startBuf,
785 ResultStr.c_str(), ResultStr.size());
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000786 }
Fariborz Jahanian66d6b292007-11-13 20:04:28 +0000787 if (IMD)
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000788 InsertText(IMD->getLocEnd(), "// ", 3);
Fariborz Jahanian66d6b292007-11-13 20:04:28 +0000789 else
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000790 InsertText(CID->getLocEnd(), "// ", 3);
Fariborz Jahanian48a0b6a2007-11-13 18:44:14 +0000791}
792
Steve Naroffb29b4272008-04-14 22:03:09 +0000793void RewriteObjC::RewriteInterfaceDecl(ObjCInterfaceDecl *ClassDecl) {
Steve Narofff908a872007-10-30 02:23:23 +0000794 std::string ResultStr;
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000795 if (!ObjCForwardDecls.count(ClassDecl)) {
Steve Naroff6c6a2db2007-11-01 03:35:41 +0000796 // we haven't seen a forward decl - generate a typedef.
Steve Naroff5086a8d2007-11-14 23:02:56 +0000797 ResultStr = "#ifndef _REWRITER_typedef_";
Steve Naroff32174822007-11-09 12:50:28 +0000798 ResultStr += ClassDecl->getName();
799 ResultStr += "\n";
800 ResultStr += "#define _REWRITER_typedef_";
801 ResultStr += ClassDecl->getName();
802 ResultStr += "\n";
Steve Naroff61ed9ca2008-03-10 23:16:54 +0000803 ResultStr += "typedef struct objc_object ";
Steve Naroff6c6a2db2007-11-01 03:35:41 +0000804 ResultStr += ClassDecl->getName();
Steve Naroff32174822007-11-09 12:50:28 +0000805 ResultStr += ";\n#endif\n";
Steve Naroff6c6a2db2007-11-01 03:35:41 +0000806 // Mark this typedef as having been generated.
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000807 ObjCForwardDecls.insert(ClassDecl);
Steve Naroff6c6a2db2007-11-01 03:35:41 +0000808 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000809 SynthesizeObjCInternalStruct(ClassDecl, ResultStr);
Steve Narofff908a872007-10-30 02:23:23 +0000810
Fariborz Jahanian957cf652007-11-07 00:09:37 +0000811 RewriteProperties(ClassDecl->getNumPropertyDecl(),
812 ClassDecl->getPropertyDecl());
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000813 for (ObjCInterfaceDecl::instmeth_iterator I = ClassDecl->instmeth_begin(),
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000814 E = ClassDecl->instmeth_end(); I != E; ++I)
815 RewriteMethodDeclaration(*I);
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000816 for (ObjCInterfaceDecl::classmeth_iterator I = ClassDecl->classmeth_begin(),
Steve Naroff58dbdeb2007-12-14 23:37:57 +0000817 E = ClassDecl->classmeth_end(); I != E; ++I)
818 RewriteMethodDeclaration(*I);
819
Steve Naroff2feac5e2007-10-30 03:43:13 +0000820 // Lastly, comment out the @end.
Chris Lattneraadaf782008-01-31 19:51:04 +0000821 ReplaceText(ClassDecl->getAtEndLoc(), 0, "// ", 3);
Steve Naroffbef11852007-10-26 20:53:56 +0000822}
823
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000824Stmt *RewriteObjC::RewriteObjCIvarRefExpr(ObjCIvarRefExpr *IV,
825 SourceLocation OrigStart) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000826 ObjCIvarDecl *D = IV->getDecl();
Steve Naroff05b8c782008-03-12 00:25:36 +0000827 if (CurMethodDecl) {
828 if (const PointerType *pType = IV->getBase()->getType()->getAsPointerType()) {
Steve Narofff0757612008-05-08 17:52:16 +0000829 ObjCInterfaceType *iFaceDecl = dyn_cast<ObjCInterfaceType>(pType->getPointeeType());
830 // lookup which class implements the instance variable.
831 ObjCInterfaceDecl *clsDeclared = 0;
832 iFaceDecl->getDecl()->lookupInstanceVariable(D->getIdentifier(), clsDeclared);
833 assert(clsDeclared && "RewriteObjCIvarRefExpr(): Can't find class");
834
835 // Synthesize an explicit cast to gain access to the ivar.
836 std::string RecName = clsDeclared->getIdentifier()->getName();
837 RecName += "_IMPL";
838 IdentifierInfo *II = &Context->Idents.get(RecName.c_str());
Argyrios Kyrtzidis39ba4ae2008-06-09 23:19:58 +0000839 RecordDecl *RD = RecordDecl::Create(*Context, TagDecl::TK_struct, TUDecl,
Steve Narofff0757612008-05-08 17:52:16 +0000840 SourceLocation(), II, 0);
841 assert(RD && "RewriteObjCIvarRefExpr(): Can't find RecordDecl");
842 QualType castT = Context->getPointerType(Context->getTagDeclType(RD));
843 CastExpr *castExpr = new CastExpr(castT, IV->getBase(), SourceLocation());
844 // Don't forget the parens to enforce the proper binding.
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000845 ParenExpr *PE = new ParenExpr(IV->getBase()->getLocStart(),
846 IV->getBase()->getLocEnd(),
847 castExpr);
Steve Narofff0757612008-05-08 17:52:16 +0000848 if (IV->isFreeIvar() &&
849 CurMethodDecl->getClassInterface() == iFaceDecl->getDecl()) {
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000850 MemberExpr *ME = new MemberExpr(PE, true, D, IV->getLocation(),
851 D->getType());
Steve Narofff0757612008-05-08 17:52:16 +0000852 ReplaceStmt(IV, ME);
853 delete IV;
854 return ME;
Steve Naroffc2a689b2007-11-15 11:33:00 +0000855 }
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000856
857 ReplaceStmt(IV->getBase(), PE);
858 // Cannot delete IV->getBase(), since PE points to it.
859 // Replace the old base with the cast. This is important when doing
860 // embedded rewrites. For example, [newInv->_container addObject:0].
861 IV->setBase(PE);
862 return IV;
Steve Naroffc2a689b2007-11-15 11:33:00 +0000863 }
Steve Naroff84472a82008-04-18 21:55:08 +0000864 } else { // we are outside a method.
Steve Naroff9f525972008-05-06 23:20:07 +0000865 assert(!IV->isFreeIvar() && "Cannot have a free standing ivar outside a method");
866
867 // Explicit ivar refs need to have a cast inserted.
868 // FIXME: consider sharing some of this code with the code above.
869 if (const PointerType *pType = IV->getBase()->getType()->getAsPointerType()) {
Steve Narofff0757612008-05-08 17:52:16 +0000870 ObjCInterfaceType *iFaceDecl = dyn_cast<ObjCInterfaceType>(pType->getPointeeType());
Steve Naroff9f525972008-05-06 23:20:07 +0000871 // lookup which class implements the instance variable.
872 ObjCInterfaceDecl *clsDeclared = 0;
Steve Narofff0757612008-05-08 17:52:16 +0000873 iFaceDecl->getDecl()->lookupInstanceVariable(D->getIdentifier(), clsDeclared);
Steve Naroff9f525972008-05-06 23:20:07 +0000874 assert(clsDeclared && "RewriteObjCIvarRefExpr(): Can't find class");
875
876 // Synthesize an explicit cast to gain access to the ivar.
877 std::string RecName = clsDeclared->getIdentifier()->getName();
878 RecName += "_IMPL";
879 IdentifierInfo *II = &Context->Idents.get(RecName.c_str());
Argyrios Kyrtzidis39ba4ae2008-06-09 23:19:58 +0000880 RecordDecl *RD = RecordDecl::Create(*Context, TagDecl::TK_struct, TUDecl,
Steve Naroff9f525972008-05-06 23:20:07 +0000881 SourceLocation(), II, 0);
882 assert(RD && "RewriteObjCIvarRefExpr(): Can't find RecordDecl");
883 QualType castT = Context->getPointerType(Context->getTagDeclType(RD));
884 CastExpr *castExpr = new CastExpr(castT, IV->getBase(), SourceLocation());
885 // Don't forget the parens to enforce the proper binding.
Chris Lattner8d366162008-05-28 16:38:23 +0000886 ParenExpr *PE = new ParenExpr(IV->getBase()->getLocStart(),
887 IV->getBase()->getLocEnd(), castExpr);
Steve Naroff9f525972008-05-06 23:20:07 +0000888 ReplaceStmt(IV->getBase(), PE);
889 // Cannot delete IV->getBase(), since PE points to it.
890 // Replace the old base with the cast. This is important when doing
891 // embedded rewrites. For example, [newInv->_container addObject:0].
892 IV->setBase(PE);
893 return IV;
894 }
Steve Naroffc2a689b2007-11-15 11:33:00 +0000895 }
Steve Naroff84472a82008-04-18 21:55:08 +0000896 return IV;
Steve Naroff7e3411b2007-11-15 02:58:25 +0000897}
898
Chris Lattnerf04da132007-10-24 17:06:59 +0000899//===----------------------------------------------------------------------===//
900// Function Body / Expression rewriting
901//===----------------------------------------------------------------------===//
902
Steve Naroffb29b4272008-04-14 22:03:09 +0000903Stmt *RewriteObjC::RewriteFunctionBodyOrGlobalInitializer(Stmt *S) {
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +0000904 if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) ||
905 isa<DoStmt>(S) || isa<ForStmt>(S))
906 Stmts.push_back(S);
907 else if (isa<ObjCForCollectionStmt>(S)) {
908 Stmts.push_back(S);
909 ObjCBcLabelNo.push_back(++BcLabelCount);
910 }
911
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000912 SourceRange OrigStmtRange = S->getSourceRange();
Chris Lattner338d1e22008-01-31 05:10:40 +0000913
914 // Start by rewriting all children.
Chris Lattner311ff022007-10-16 22:36:42 +0000915 for (Stmt::child_iterator CI = S->child_begin(), E = S->child_end();
916 CI != E; ++CI)
Steve Naroff75730982007-11-07 04:08:17 +0000917 if (*CI) {
Steve Narofff3473a72007-11-09 15:20:18 +0000918 Stmt *newStmt = RewriteFunctionBodyOrGlobalInitializer(*CI);
Steve Naroff75730982007-11-07 04:08:17 +0000919 if (newStmt)
920 *CI = newStmt;
921 }
Steve Naroffebf2b562007-10-23 23:50:29 +0000922
923 // Handle specific things.
924 if (ObjCEncodeExpr *AtEncode = dyn_cast<ObjCEncodeExpr>(S))
925 return RewriteAtEncode(AtEncode);
Steve Naroff7e3411b2007-11-15 02:58:25 +0000926
927 if (ObjCIvarRefExpr *IvarRefExpr = dyn_cast<ObjCIvarRefExpr>(S))
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000928 return RewriteObjCIvarRefExpr(IvarRefExpr, OrigStmtRange.getBegin());
Steve Naroffb42f8412007-11-05 14:50:49 +0000929
930 if (ObjCSelectorExpr *AtSelector = dyn_cast<ObjCSelectorExpr>(S))
931 return RewriteAtSelector(AtSelector);
Steve Narofff69cc5d2008-01-30 19:17:43 +0000932
Steve Naroffbeaf2992007-11-03 11:27:19 +0000933 if (ObjCStringLiteral *AtString = dyn_cast<ObjCStringLiteral>(S))
934 return RewriteObjCStringLiteral(AtString);
Steve Naroffebf2b562007-10-23 23:50:29 +0000935
Steve Naroff934f2762007-10-24 22:48:43 +0000936 if (ObjCMessageExpr *MessExpr = dyn_cast<ObjCMessageExpr>(S)) {
937 // Before we rewrite it, put the original message expression in a comment.
938 SourceLocation startLoc = MessExpr->getLocStart();
939 SourceLocation endLoc = MessExpr->getLocEnd();
940
941 const char *startBuf = SM->getCharacterData(startLoc);
942 const char *endBuf = SM->getCharacterData(endLoc);
943
944 std::string messString;
945 messString += "// ";
946 messString.append(startBuf, endBuf-startBuf+1);
947 messString += "\n";
Steve Naroffbef11852007-10-26 20:53:56 +0000948
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000949 // FIXME: Missing definition of
950 // InsertText(clang::SourceLocation, char const*, unsigned int).
951 // InsertText(startLoc, messString.c_str(), messString.size());
Steve Naroff934f2762007-10-24 22:48:43 +0000952 // Tried this, but it didn't work either...
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000953 // ReplaceText(startLoc, 0, messString.c_str(), messString.size());
Steve Naroffebf2b562007-10-23 23:50:29 +0000954 return RewriteMessageExpr(MessExpr);
Steve Naroff934f2762007-10-24 22:48:43 +0000955 }
Fariborz Jahanian909f02a2007-11-05 17:47:33 +0000956
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000957 if (ObjCAtTryStmt *StmtTry = dyn_cast<ObjCAtTryStmt>(S))
958 return RewriteObjCTryStmt(StmtTry);
Steve Naroff2bd03922007-11-07 15:32:26 +0000959
Fariborz Jahaniana0f55792008-01-29 22:59:37 +0000960 if (ObjCAtSynchronizedStmt *StmtTry = dyn_cast<ObjCAtSynchronizedStmt>(S))
961 return RewriteObjCSynchronizedStmt(StmtTry);
962
Ted Kremeneka526c5c2008-01-07 19:49:32 +0000963 if (ObjCAtThrowStmt *StmtThrow = dyn_cast<ObjCAtThrowStmt>(S))
964 return RewriteObjCThrowStmt(StmtThrow);
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +0000965
966 if (ObjCProtocolExpr *ProtocolExp = dyn_cast<ObjCProtocolExpr>(S))
967 return RewriteObjCProtocolExpr(ProtocolExp);
Fariborz Jahanian10d24f02008-01-07 21:40:22 +0000968
969 if (ObjCForCollectionStmt *StmtForCollection =
970 dyn_cast<ObjCForCollectionStmt>(S))
Chris Lattner3b2c58c2008-05-23 20:40:52 +0000971 return RewriteObjCForCollectionStmt(StmtForCollection,
972 OrigStmtRange.getEnd());
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +0000973 if (BreakStmt *StmtBreakStmt =
974 dyn_cast<BreakStmt>(S))
975 return RewriteBreakStmt(StmtBreakStmt);
976 if (ContinueStmt *StmtContinueStmt =
977 dyn_cast<ContinueStmt>(S))
978 return RewriteContinueStmt(StmtContinueStmt);
Fariborz Jahanian10d24f02008-01-07 21:40:22 +0000979
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +0000980 if (isa<SwitchStmt>(S) || isa<WhileStmt>(S) ||
981 isa<DoStmt>(S) || isa<ForStmt>(S)) {
982 assert(!Stmts.empty() && "Statement stack is empty");
983 assert ((isa<SwitchStmt>(Stmts.back()) || isa<WhileStmt>(Stmts.back()) ||
984 isa<DoStmt>(Stmts.back()) || isa<ForStmt>(Stmts.back()))
985 && "Statement stack mismatch");
986 Stmts.pop_back();
987 }
Steve Naroff874e2322007-11-15 10:28:18 +0000988#if 0
989 if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(S)) {
990 CastExpr *Replacement = new CastExpr(ICE->getType(), ICE->getSubExpr(), SourceLocation());
991 // Get the new text.
992 std::ostringstream Buf;
993 Replacement->printPretty(Buf);
994 const std::string &Str = Buf.str();
995
996 printf("CAST = %s\n", &Str[0]);
Chris Lattnerf3dd57e2008-01-31 19:42:41 +0000997 InsertText(ICE->getSubExpr()->getLocStart(), &Str[0], Str.size());
Steve Naroff874e2322007-11-15 10:28:18 +0000998 delete S;
999 return Replacement;
1000 }
1001#endif
Chris Lattnere64b7772007-10-24 16:57:36 +00001002 // Return this stmt unmodified.
1003 return S;
Chris Lattner311ff022007-10-16 22:36:42 +00001004}
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001005
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001006/// SynthCountByEnumWithState - To print:
1007/// ((unsigned int (*)
1008/// (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int))
1009/// (void *)objc_msgSend)((id)l_collection,
1010/// sel_registerName(
1011/// "countByEnumeratingWithState:objects:count:"),
1012/// &enumState,
1013/// (id *)items, (unsigned int)16)
1014///
Steve Naroffb29b4272008-04-14 22:03:09 +00001015void RewriteObjC::SynthCountByEnumWithState(std::string &buf) {
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001016 buf += "((unsigned int (*) (id, SEL, struct __objcFastEnumerationState *, "
1017 "id *, unsigned int))(void *)objc_msgSend)";
1018 buf += "\n\t\t";
1019 buf += "((id)l_collection,\n\t\t";
1020 buf += "sel_registerName(\"countByEnumeratingWithState:objects:count:\"),";
1021 buf += "\n\t\t";
1022 buf += "&enumState, "
1023 "(id *)items, (unsigned int)16)";
1024}
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001025
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001026/// RewriteBreakStmt - Rewrite for a break-stmt inside an ObjC2's foreach
1027/// statement to exit to its outer synthesized loop.
1028///
Steve Naroffb29b4272008-04-14 22:03:09 +00001029Stmt *RewriteObjC::RewriteBreakStmt(BreakStmt *S) {
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001030 if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back()))
1031 return S;
1032 // replace break with goto __break_label
1033 std::string buf;
1034
1035 SourceLocation startLoc = S->getLocStart();
1036 buf = "goto __break_label_";
1037 buf += utostr(ObjCBcLabelNo.back());
Chris Lattneraadaf782008-01-31 19:51:04 +00001038 ReplaceText(startLoc, strlen("break"), buf.c_str(), buf.size());
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001039
1040 return 0;
1041}
1042
1043/// RewriteContinueStmt - Rewrite for a continue-stmt inside an ObjC2's foreach
1044/// statement to continue with its inner synthesized loop.
1045///
Steve Naroffb29b4272008-04-14 22:03:09 +00001046Stmt *RewriteObjC::RewriteContinueStmt(ContinueStmt *S) {
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001047 if (Stmts.empty() || !isa<ObjCForCollectionStmt>(Stmts.back()))
1048 return S;
1049 // replace continue with goto __continue_label
1050 std::string buf;
1051
1052 SourceLocation startLoc = S->getLocStart();
1053 buf = "goto __continue_label_";
1054 buf += utostr(ObjCBcLabelNo.back());
Chris Lattneraadaf782008-01-31 19:51:04 +00001055 ReplaceText(startLoc, strlen("continue"), buf.c_str(), buf.size());
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001056
1057 return 0;
1058}
1059
Fariborz Jahaniana0f55792008-01-29 22:59:37 +00001060/// RewriteObjCForCollectionStmt - Rewriter for ObjC2's foreach statement.
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001061/// It rewrites:
1062/// for ( type elem in collection) { stmts; }
Fariborz Jahanianf4d331d2007-10-18 22:09:03 +00001063
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001064/// Into:
1065/// {
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001066/// type elem;
1067/// struct __objcFastEnumerationState enumState = { 0 };
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001068/// id items[16];
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001069/// id l_collection = (id)collection;
1070/// unsigned long limit = [l_collection countByEnumeratingWithState:&enumState
1071/// objects:items count:16];
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001072/// if (limit) {
1073/// unsigned long startMutations = *enumState.mutationsPtr;
1074/// do {
1075/// unsigned long counter = 0;
1076/// do {
1077/// if (startMutations != *enumState.mutationsPtr)
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001078/// objc_enumerationMutation(l_collection);
Fariborz Jahanian88f50f32008-01-09 18:15:42 +00001079/// elem = (type)enumState.itemsPtr[counter++];
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001080/// stmts;
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001081/// __continue_label: ;
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001082/// } while (counter < limit);
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001083/// } while (limit = [l_collection countByEnumeratingWithState:&enumState
1084/// objects:items count:16]);
1085/// elem = nil;
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001086/// __break_label: ;
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001087/// }
1088/// else
1089/// elem = nil;
1090/// }
1091///
Steve Naroffb29b4272008-04-14 22:03:09 +00001092Stmt *RewriteObjC::RewriteObjCForCollectionStmt(ObjCForCollectionStmt *S,
Chris Lattner338d1e22008-01-31 05:10:40 +00001093 SourceLocation OrigEnd) {
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001094 assert(!Stmts.empty() && "ObjCForCollectionStmt - Statement stack empty");
1095 assert(isa<ObjCForCollectionStmt>(Stmts.back()) &&
1096 "ObjCForCollectionStmt Statement stack mismatch");
1097 assert(!ObjCBcLabelNo.empty() &&
1098 "ObjCForCollectionStmt - Label No stack empty");
1099
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001100 SourceLocation startLoc = S->getLocStart();
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001101 const char *startBuf = SM->getCharacterData(startLoc);
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001102 const char *elementName;
Fariborz Jahanian88f50f32008-01-09 18:15:42 +00001103 std::string elementTypeAsString;
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001104 std::string buf;
1105 buf = "\n{\n\t";
1106 if (DeclStmt *DS = dyn_cast<DeclStmt>(S->getElement())) {
1107 // type elem;
1108 QualType ElementType = cast<ValueDecl>(DS->getDecl())->getType();
Fariborz Jahanian88f50f32008-01-09 18:15:42 +00001109 elementTypeAsString = ElementType.getAsString();
1110 buf += elementTypeAsString;
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001111 buf += " ";
1112 elementName = DS->getDecl()->getName();
1113 buf += elementName;
1114 buf += ";\n\t";
1115 }
Chris Lattner06767512008-04-08 05:52:18 +00001116 else {
1117 DeclRefExpr *DR = cast<DeclRefExpr>(S->getElement());
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001118 elementName = DR->getDecl()->getName();
Fariborz Jahanian88f50f32008-01-09 18:15:42 +00001119 elementTypeAsString = DR->getDecl()->getType().getAsString();
1120 }
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001121
1122 // struct __objcFastEnumerationState enumState = { 0 };
1123 buf += "struct __objcFastEnumerationState enumState = { 0 };\n\t";
1124 // id items[16];
1125 buf += "id items[16];\n\t";
1126 // id l_collection = (id)
1127 buf += "id l_collection = (id)";
Fariborz Jahanian75712282008-01-10 00:24:29 +00001128 // Find start location of 'collection' the hard way!
1129 const char *startCollectionBuf = startBuf;
1130 startCollectionBuf += 3; // skip 'for'
1131 startCollectionBuf = strchr(startCollectionBuf, '(');
1132 startCollectionBuf++; // skip '('
1133 // find 'in' and skip it.
1134 while (*startCollectionBuf != ' ' ||
1135 *(startCollectionBuf+1) != 'i' || *(startCollectionBuf+2) != 'n' ||
1136 (*(startCollectionBuf+3) != ' ' &&
1137 *(startCollectionBuf+3) != '[' && *(startCollectionBuf+3) != '('))
1138 startCollectionBuf++;
1139 startCollectionBuf += 3;
1140
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001141 // Replace: "for (type element in" with string constructed thus far.
Chris Lattneraadaf782008-01-31 19:51:04 +00001142 ReplaceText(startLoc, startCollectionBuf - startBuf,
1143 buf.c_str(), buf.size());
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001144 // Replace ')' in for '(' type elem in collection ')' with ';'
Fariborz Jahanian75712282008-01-10 00:24:29 +00001145 SourceLocation rightParenLoc = S->getRParenLoc();
1146 const char *rparenBuf = SM->getCharacterData(rightParenLoc);
1147 SourceLocation lparenLoc = startLoc.getFileLocWithOffset(rparenBuf-startBuf);
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001148 buf = ";\n\t";
1149
1150 // unsigned long limit = [l_collection countByEnumeratingWithState:&enumState
1151 // objects:items count:16];
1152 // which is synthesized into:
1153 // unsigned int limit =
1154 // ((unsigned int (*)
1155 // (id, SEL, struct __objcFastEnumerationState *, id *, unsigned int))
1156 // (void *)objc_msgSend)((id)l_collection,
1157 // sel_registerName(
1158 // "countByEnumeratingWithState:objects:count:"),
1159 // (struct __objcFastEnumerationState *)&state,
1160 // (id *)items, (unsigned int)16);
1161 buf += "unsigned long limit =\n\t\t";
1162 SynthCountByEnumWithState(buf);
1163 buf += ";\n\t";
1164 /// if (limit) {
1165 /// unsigned long startMutations = *enumState.mutationsPtr;
1166 /// do {
1167 /// unsigned long counter = 0;
1168 /// do {
1169 /// if (startMutations != *enumState.mutationsPtr)
1170 /// objc_enumerationMutation(l_collection);
Fariborz Jahanian88f50f32008-01-09 18:15:42 +00001171 /// elem = (type)enumState.itemsPtr[counter++];
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001172 buf += "if (limit) {\n\t";
1173 buf += "unsigned long startMutations = *enumState.mutationsPtr;\n\t";
1174 buf += "do {\n\t\t";
1175 buf += "unsigned long counter = 0;\n\t\t";
1176 buf += "do {\n\t\t\t";
1177 buf += "if (startMutations != *enumState.mutationsPtr)\n\t\t\t\t";
1178 buf += "objc_enumerationMutation(l_collection);\n\t\t\t";
1179 buf += elementName;
Fariborz Jahanian88f50f32008-01-09 18:15:42 +00001180 buf += " = (";
1181 buf += elementTypeAsString;
1182 buf += ")enumState.itemsPtr[counter++];";
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001183 // Replace ')' in for '(' type elem in collection ')' with all of these.
Chris Lattneraadaf782008-01-31 19:51:04 +00001184 ReplaceText(lparenLoc, 1, buf.c_str(), buf.size());
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001185
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001186 /// __continue_label: ;
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001187 /// } while (counter < limit);
1188 /// } while (limit = [l_collection countByEnumeratingWithState:&enumState
1189 /// objects:items count:16]);
1190 /// elem = nil;
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001191 /// __break_label: ;
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001192 /// }
1193 /// else
1194 /// elem = nil;
1195 /// }
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001196 ///
1197 buf = ";\n\t";
1198 buf += "__continue_label_";
1199 buf += utostr(ObjCBcLabelNo.back());
1200 buf += ": ;";
1201 buf += "\n\t\t";
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001202 buf += "} while (counter < limit);\n\t";
1203 buf += "} while (limit = ";
1204 SynthCountByEnumWithState(buf);
1205 buf += ");\n\t";
1206 buf += elementName;
1207 buf += " = nil;\n\t";
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001208 buf += "__break_label_";
1209 buf += utostr(ObjCBcLabelNo.back());
1210 buf += ": ;\n\t";
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001211 buf += "}\n\t";
1212 buf += "else\n\t\t";
1213 buf += elementName;
1214 buf += " = nil;\n";
1215 buf += "}\n";
1216 // Insert all these *after* the statement body.
Chris Lattner338d1e22008-01-31 05:10:40 +00001217 SourceLocation endBodyLoc = OrigEnd.getFileLocWithOffset(1);
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001218 InsertText(endBodyLoc, buf.c_str(), buf.size());
Fariborz Jahaniane8d1c052008-01-15 23:58:23 +00001219 Stmts.pop_back();
1220 ObjCBcLabelNo.pop_back();
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00001221 return 0;
Fariborz Jahanian10d24f02008-01-07 21:40:22 +00001222}
1223
Fariborz Jahaniana0f55792008-01-29 22:59:37 +00001224/// RewriteObjCSynchronizedStmt -
1225/// This routine rewrites @synchronized(expr) stmt;
1226/// into:
1227/// objc_sync_enter(expr);
1228/// @try stmt @finally { objc_sync_exit(expr); }
1229///
Steve Naroffb29b4272008-04-14 22:03:09 +00001230Stmt *RewriteObjC::RewriteObjCSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
Fariborz Jahaniana0f55792008-01-29 22:59:37 +00001231 // Get the start location and compute the semi location.
1232 SourceLocation startLoc = S->getLocStart();
1233 const char *startBuf = SM->getCharacterData(startLoc);
1234
1235 assert((*startBuf == '@') && "bogus @synchronized location");
1236
1237 std::string buf;
1238 buf = "objc_sync_enter";
Chris Lattneraadaf782008-01-31 19:51:04 +00001239 ReplaceText(startLoc, 13, buf.c_str(), buf.size());
Fariborz Jahaniana0f55792008-01-29 22:59:37 +00001240 SourceLocation endLoc = S->getSynchExpr()->getLocEnd();
1241 const char *endBuf = SM->getCharacterData(endLoc);
1242 endBuf++;
1243 const char *rparenBuf = strchr(endBuf, ')');
1244 SourceLocation rparenLoc = startLoc.getFileLocWithOffset(rparenBuf-startBuf);
1245 buf = ");\n";
1246 // declare a new scope with two variables, _stack and _rethrow.
1247 buf += "/* @try scope begin */ \n{ struct _objc_exception_data {\n";
1248 buf += "int buf[18/*32-bit i386*/];\n";
1249 buf += "char *pointers[4];} _stack;\n";
1250 buf += "id volatile _rethrow = 0;\n";
1251 buf += "objc_exception_try_enter(&_stack);\n";
1252 buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n";
Chris Lattneraadaf782008-01-31 19:51:04 +00001253 ReplaceText(rparenLoc, 1, buf.c_str(), buf.size());
Fariborz Jahaniana0f55792008-01-29 22:59:37 +00001254 startLoc = S->getSynchBody()->getLocEnd();
1255 startBuf = SM->getCharacterData(startLoc);
1256
1257 assert((*startBuf == '}') && "bogus @try block");
1258 SourceLocation lastCurlyLoc = startLoc;
1259 buf = "}\nelse {\n";
1260 buf += " _rethrow = objc_exception_extract(&_stack);\n";
1261 buf += " if (!_rethrow) objc_exception_try_exit(&_stack);\n";
1262 // FIXME: This must be objc_sync_exit(syncExpr);
1263 buf += " objc_sync_exit();\n";
1264 buf += " if (_rethrow) objc_exception_throw(_rethrow);\n";
1265 buf += "}\n";
1266 buf += "}";
1267
Chris Lattneraadaf782008-01-31 19:51:04 +00001268 ReplaceText(lastCurlyLoc, 1, buf.c_str(), buf.size());
Fariborz Jahaniana0f55792008-01-29 22:59:37 +00001269 return 0;
1270}
1271
Steve Naroffb29b4272008-04-14 22:03:09 +00001272Stmt *RewriteObjC::RewriteObjCTryStmt(ObjCAtTryStmt *S) {
Steve Naroff75730982007-11-07 04:08:17 +00001273 // Get the start location and compute the semi location.
1274 SourceLocation startLoc = S->getLocStart();
1275 const char *startBuf = SM->getCharacterData(startLoc);
1276
1277 assert((*startBuf == '@') && "bogus @try location");
1278
1279 std::string buf;
1280 // declare a new scope with two variables, _stack and _rethrow.
1281 buf = "/* @try scope begin */ { struct _objc_exception_data {\n";
1282 buf += "int buf[18/*32-bit i386*/];\n";
1283 buf += "char *pointers[4];} _stack;\n";
1284 buf += "id volatile _rethrow = 0;\n";
1285 buf += "objc_exception_try_enter(&_stack);\n";
Steve Naroff21867b12007-11-07 18:43:40 +00001286 buf += "if (!_setjmp(_stack.buf)) /* @try block continue */\n";
Steve Naroff75730982007-11-07 04:08:17 +00001287
Chris Lattneraadaf782008-01-31 19:51:04 +00001288 ReplaceText(startLoc, 4, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001289
1290 startLoc = S->getTryBody()->getLocEnd();
1291 startBuf = SM->getCharacterData(startLoc);
1292
1293 assert((*startBuf == '}') && "bogus @try block");
1294
1295 SourceLocation lastCurlyLoc = startLoc;
1296
1297 startLoc = startLoc.getFileLocWithOffset(1);
1298 buf = " /* @catch begin */ else {\n";
1299 buf += " id _caught = objc_exception_extract(&_stack);\n";
1300 buf += " objc_exception_try_enter (&_stack);\n";
Steve Naroff21867b12007-11-07 18:43:40 +00001301 buf += " if (_setjmp(_stack.buf))\n";
Steve Naroff75730982007-11-07 04:08:17 +00001302 buf += " _rethrow = objc_exception_extract(&_stack);\n";
1303 buf += " else { /* @catch continue */";
1304
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001305 InsertText(startLoc, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001306
1307 bool sawIdTypedCatch = false;
1308 Stmt *lastCatchBody = 0;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001309 ObjCAtCatchStmt *catchList = S->getCatchStmts();
Steve Naroff75730982007-11-07 04:08:17 +00001310 while (catchList) {
1311 Stmt *catchStmt = catchList->getCatchParamStmt();
1312
1313 if (catchList == S->getCatchStmts())
1314 buf = "if ("; // we are generating code for the first catch clause
1315 else
1316 buf = "else if (";
1317 startLoc = catchList->getLocStart();
1318 startBuf = SM->getCharacterData(startLoc);
1319
1320 assert((*startBuf == '@') && "bogus @catch location");
1321
1322 const char *lParenLoc = strchr(startBuf, '(');
1323
Steve Naroffbe4b3332008-02-01 22:08:12 +00001324 if (catchList->hasEllipsis()) {
Steve Naroffe12e6922008-02-01 20:02:07 +00001325 // Now rewrite the body...
1326 lastCatchBody = catchList->getCatchBody();
Steve Naroffe12e6922008-02-01 20:02:07 +00001327 SourceLocation bodyLoc = lastCatchBody->getLocStart();
1328 const char *bodyBuf = SM->getCharacterData(bodyLoc);
Chris Lattner06767512008-04-08 05:52:18 +00001329 assert(*SM->getCharacterData(catchList->getRParenLoc()) == ')' &&
1330 "bogus @catch paren location");
Steve Naroffe12e6922008-02-01 20:02:07 +00001331 assert((*bodyBuf == '{') && "bogus @catch body location");
1332
1333 buf += "1) { id _tmp = _caught;";
1334 Rewrite.ReplaceText(startLoc, bodyBuf-startBuf+1,
1335 buf.c_str(), buf.size());
1336 } else if (DeclStmt *declStmt = dyn_cast<DeclStmt>(catchStmt)) {
Steve Naroff75730982007-11-07 04:08:17 +00001337 QualType t = dyn_cast<ValueDecl>(declStmt->getDecl())->getType();
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001338 if (t == Context->getObjCIdType()) {
Steve Naroff75730982007-11-07 04:08:17 +00001339 buf += "1) { ";
Chris Lattneraadaf782008-01-31 19:51:04 +00001340 ReplaceText(startLoc, lParenLoc-startBuf+1, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001341 sawIdTypedCatch = true;
1342 } else if (const PointerType *pType = t->getAsPointerType()) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001343 ObjCInterfaceType *cls; // Should be a pointer to a class.
Steve Naroff75730982007-11-07 04:08:17 +00001344
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001345 cls = dyn_cast<ObjCInterfaceType>(pType->getPointeeType().getTypePtr());
Steve Naroff75730982007-11-07 04:08:17 +00001346 if (cls) {
Steve Naroff21867b12007-11-07 18:43:40 +00001347 buf += "objc_exception_match((struct objc_class *)objc_getClass(\"";
Steve Naroff75730982007-11-07 04:08:17 +00001348 buf += cls->getDecl()->getName();
Steve Naroff21867b12007-11-07 18:43:40 +00001349 buf += "\"), (struct objc_object *)_caught)) { ";
Chris Lattneraadaf782008-01-31 19:51:04 +00001350 ReplaceText(startLoc, lParenLoc-startBuf+1, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001351 }
1352 }
1353 // Now rewrite the body...
1354 lastCatchBody = catchList->getCatchBody();
1355 SourceLocation rParenLoc = catchList->getRParenLoc();
1356 SourceLocation bodyLoc = lastCatchBody->getLocStart();
1357 const char *bodyBuf = SM->getCharacterData(bodyLoc);
1358 const char *rParenBuf = SM->getCharacterData(rParenLoc);
1359 assert((*rParenBuf == ')') && "bogus @catch paren location");
1360 assert((*bodyBuf == '{') && "bogus @catch body location");
1361
1362 buf = " = _caught;";
1363 // Here we replace ") {" with "= _caught;" (which initializes and
1364 // declares the @catch parameter).
Chris Lattneraadaf782008-01-31 19:51:04 +00001365 ReplaceText(rParenLoc, bodyBuf-rParenBuf+1, buf.c_str(), buf.size());
Steve Naroff2bd03922007-11-07 15:32:26 +00001366 } else if (!isa<NullStmt>(catchStmt)) {
Steve Naroff75730982007-11-07 04:08:17 +00001367 assert(false && "@catch rewrite bug");
Steve Naroff2bd03922007-11-07 15:32:26 +00001368 }
Steve Naroffe12e6922008-02-01 20:02:07 +00001369 // make sure all the catch bodies get rewritten!
Steve Naroff75730982007-11-07 04:08:17 +00001370 catchList = catchList->getNextCatchStmt();
1371 }
1372 // Complete the catch list...
1373 if (lastCatchBody) {
1374 SourceLocation bodyLoc = lastCatchBody->getLocEnd();
Chris Lattner06767512008-04-08 05:52:18 +00001375 assert(*SM->getCharacterData(bodyLoc) == '}' &&
1376 "bogus @catch body location");
Steve Naroff75730982007-11-07 04:08:17 +00001377 bodyLoc = bodyLoc.getFileLocWithOffset(1);
1378 buf = " } } /* @catch end */\n";
1379
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001380 InsertText(bodyLoc, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001381
1382 // Set lastCurlyLoc
1383 lastCurlyLoc = lastCatchBody->getLocEnd();
1384 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001385 if (ObjCAtFinallyStmt *finalStmt = S->getFinallyStmt()) {
Steve Naroff75730982007-11-07 04:08:17 +00001386 startLoc = finalStmt->getLocStart();
1387 startBuf = SM->getCharacterData(startLoc);
1388 assert((*startBuf == '@') && "bogus @finally start");
1389
1390 buf = "/* @finally */";
Chris Lattneraadaf782008-01-31 19:51:04 +00001391 ReplaceText(startLoc, 8, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001392
1393 Stmt *body = finalStmt->getFinallyBody();
1394 SourceLocation startLoc = body->getLocStart();
1395 SourceLocation endLoc = body->getLocEnd();
Chris Lattner06767512008-04-08 05:52:18 +00001396 assert(*SM->getCharacterData(startLoc) == '{' &&
1397 "bogus @finally body location");
1398 assert(*SM->getCharacterData(endLoc) == '}' &&
1399 "bogus @finally body location");
Steve Naroff75730982007-11-07 04:08:17 +00001400
1401 startLoc = startLoc.getFileLocWithOffset(1);
1402 buf = " if (!_rethrow) objc_exception_try_exit(&_stack);\n";
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001403 InsertText(startLoc, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001404 endLoc = endLoc.getFileLocWithOffset(-1);
1405 buf = " if (_rethrow) objc_exception_throw(_rethrow);\n";
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001406 InsertText(endLoc, buf.c_str(), buf.size());
Steve Naroff75730982007-11-07 04:08:17 +00001407
1408 // Set lastCurlyLoc
1409 lastCurlyLoc = body->getLocEnd();
1410 }
1411 // Now emit the final closing curly brace...
1412 lastCurlyLoc = lastCurlyLoc.getFileLocWithOffset(1);
1413 buf = " } /* @try scope end */\n";
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001414 InsertText(lastCurlyLoc, buf.c_str(), buf.size());
Fariborz Jahanian909f02a2007-11-05 17:47:33 +00001415 return 0;
1416}
1417
Steve Naroffb29b4272008-04-14 22:03:09 +00001418Stmt *RewriteObjC::RewriteObjCCatchStmt(ObjCAtCatchStmt *S) {
Fariborz Jahanian909f02a2007-11-05 17:47:33 +00001419 return 0;
1420}
1421
Steve Naroffb29b4272008-04-14 22:03:09 +00001422Stmt *RewriteObjC::RewriteObjCFinallyStmt(ObjCAtFinallyStmt *S) {
Fariborz Jahanian909f02a2007-11-05 17:47:33 +00001423 return 0;
1424}
1425
Chris Lattnerdcbc5b02008-01-31 19:37:57 +00001426// This can't be done with ReplaceStmt(S, ThrowExpr), since
Steve Naroff2bd03922007-11-07 15:32:26 +00001427// the throw expression is typically a message expression that's already
1428// been rewritten! (which implies the SourceLocation's are invalid).
Steve Naroffb29b4272008-04-14 22:03:09 +00001429Stmt *RewriteObjC::RewriteObjCThrowStmt(ObjCAtThrowStmt *S) {
Steve Naroff2bd03922007-11-07 15:32:26 +00001430 // Get the start location and compute the semi location.
1431 SourceLocation startLoc = S->getLocStart();
1432 const char *startBuf = SM->getCharacterData(startLoc);
1433
1434 assert((*startBuf == '@') && "bogus @throw location");
1435
1436 std::string buf;
1437 /* void objc_exception_throw(id) __attribute__((noreturn)); */
Steve Naroff20ebf8f2008-01-19 00:42:38 +00001438 if (S->getThrowExpr())
1439 buf = "objc_exception_throw(";
1440 else // add an implicit argument
1441 buf = "objc_exception_throw(_caught";
Chris Lattneraadaf782008-01-31 19:51:04 +00001442 ReplaceText(startLoc, 6, buf.c_str(), buf.size());
Steve Naroff2bd03922007-11-07 15:32:26 +00001443 const char *semiBuf = strchr(startBuf, ';');
1444 assert((*semiBuf == ';') && "@throw: can't find ';'");
1445 SourceLocation semiLoc = startLoc.getFileLocWithOffset(semiBuf-startBuf);
1446 buf = ");";
Chris Lattneraadaf782008-01-31 19:51:04 +00001447 ReplaceText(semiLoc, 1, buf.c_str(), buf.size());
Steve Naroff2bd03922007-11-07 15:32:26 +00001448 return 0;
1449}
Fariborz Jahanian909f02a2007-11-05 17:47:33 +00001450
Steve Naroffb29b4272008-04-14 22:03:09 +00001451Stmt *RewriteObjC::RewriteAtEncode(ObjCEncodeExpr *Exp) {
Chris Lattner01c57482007-10-17 22:35:30 +00001452 // Create a new string expression.
1453 QualType StrType = Context->getPointerType(Context->CharTy);
Anders Carlsson85f9bce2007-10-29 05:01:08 +00001454 std::string StrEncoding;
Fariborz Jahanian7d6b46d2008-01-22 22:44:46 +00001455 Context->getObjCEncodingForType(Exp->getEncodedType(), StrEncoding,
1456 EncodingRecordTypes);
Anders Carlsson85f9bce2007-10-29 05:01:08 +00001457 Expr *Replacement = new StringLiteral(StrEncoding.c_str(),
1458 StrEncoding.length(), false, StrType,
Chris Lattner01c57482007-10-17 22:35:30 +00001459 SourceLocation(), SourceLocation());
Chris Lattnerdcbc5b02008-01-31 19:37:57 +00001460 ReplaceStmt(Exp, Replacement);
Chris Lattnere365c502007-11-30 22:25:36 +00001461
Chris Lattner07506182007-11-30 22:53:43 +00001462 // Replace this subexpr in the parent.
Chris Lattnere64b7772007-10-24 16:57:36 +00001463 delete Exp;
1464 return Replacement;
Chris Lattner311ff022007-10-16 22:36:42 +00001465}
1466
Steve Naroffb29b4272008-04-14 22:03:09 +00001467Stmt *RewriteObjC::RewriteAtSelector(ObjCSelectorExpr *Exp) {
Steve Naroffb42f8412007-11-05 14:50:49 +00001468 assert(SelGetUidFunctionDecl && "Can't find sel_registerName() decl");
1469 // Create a call to sel_registerName("selName").
1470 llvm::SmallVector<Expr*, 8> SelExprs;
1471 QualType argType = Context->getPointerType(Context->CharTy);
1472 SelExprs.push_back(new StringLiteral(Exp->getSelector().getName().c_str(),
1473 Exp->getSelector().getName().size(),
1474 false, argType, SourceLocation(),
1475 SourceLocation()));
1476 CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl,
1477 &SelExprs[0], SelExprs.size());
Chris Lattnerdcbc5b02008-01-31 19:37:57 +00001478 ReplaceStmt(Exp, SelExp);
Steve Naroffb42f8412007-11-05 14:50:49 +00001479 delete Exp;
1480 return SelExp;
1481}
1482
Steve Naroffb29b4272008-04-14 22:03:09 +00001483CallExpr *RewriteObjC::SynthesizeCallToFunctionDecl(
Steve Naroff934f2762007-10-24 22:48:43 +00001484 FunctionDecl *FD, Expr **args, unsigned nargs) {
Steve Naroffebf2b562007-10-23 23:50:29 +00001485 // Get the type, we will need to reference it in a couple spots.
Steve Naroff934f2762007-10-24 22:48:43 +00001486 QualType msgSendType = FD->getType();
Steve Naroffebf2b562007-10-23 23:50:29 +00001487
1488 // Create a reference to the objc_msgSend() declaration.
Steve Naroff934f2762007-10-24 22:48:43 +00001489 DeclRefExpr *DRE = new DeclRefExpr(FD, msgSendType, SourceLocation());
Steve Naroffebf2b562007-10-23 23:50:29 +00001490
1491 // Now, we cast the reference to a pointer to the objc_msgSend type.
Chris Lattnerf04da132007-10-24 17:06:59 +00001492 QualType pToFunc = Context->getPointerType(msgSendType);
Steve Naroffebf2b562007-10-23 23:50:29 +00001493 ImplicitCastExpr *ICE = new ImplicitCastExpr(pToFunc, DRE);
1494
1495 const FunctionType *FT = msgSendType->getAsFunctionType();
Chris Lattnere64b7772007-10-24 16:57:36 +00001496
Steve Naroff934f2762007-10-24 22:48:43 +00001497 return new CallExpr(ICE, args, nargs, FT->getResultType(), SourceLocation());
1498}
1499
Steve Naroffd5255f52007-11-01 13:24:47 +00001500static bool scanForProtocolRefs(const char *startBuf, const char *endBuf,
1501 const char *&startRef, const char *&endRef) {
1502 while (startBuf < endBuf) {
1503 if (*startBuf == '<')
1504 startRef = startBuf; // mark the start.
1505 if (*startBuf == '>') {
Steve Naroff32174822007-11-09 12:50:28 +00001506 if (startRef && *startRef == '<') {
1507 endRef = startBuf; // mark the end.
1508 return true;
1509 }
1510 return false;
Steve Naroffd5255f52007-11-01 13:24:47 +00001511 }
1512 startBuf++;
1513 }
1514 return false;
1515}
1516
Fariborz Jahanian61477f72007-12-11 22:50:14 +00001517static void scanToNextArgument(const char *&argRef) {
1518 int angle = 0;
1519 while (*argRef != ')' && (*argRef != ',' || angle > 0)) {
1520 if (*argRef == '<')
1521 angle++;
1522 else if (*argRef == '>')
1523 angle--;
1524 argRef++;
1525 }
1526 assert(angle == 0 && "scanToNextArgument - bad protocol type syntax");
1527}
Fariborz Jahanian291e04b2007-12-11 23:04:08 +00001528
Steve Naroffb29b4272008-04-14 22:03:09 +00001529bool RewriteObjC::needToScanForQualifiers(QualType T) {
Fariborz Jahanianc5692492007-12-17 21:03:50 +00001530
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001531 if (T->isObjCQualifiedIdType())
Fariborz Jahanianc5692492007-12-17 21:03:50 +00001532 return true;
1533
Steve Naroffd5255f52007-11-01 13:24:47 +00001534 if (const PointerType *pType = T->getAsPointerType()) {
Steve Naroff9165ad32007-10-31 04:38:33 +00001535 Type *pointeeType = pType->getPointeeType().getTypePtr();
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001536 if (isa<ObjCQualifiedInterfaceType>(pointeeType))
Steve Naroff9165ad32007-10-31 04:38:33 +00001537 return true; // we have "Class <Protocol> *".
1538 }
Steve Naroffd5255f52007-11-01 13:24:47 +00001539 return false;
1540}
1541
Steve Naroffb29b4272008-04-14 22:03:09 +00001542void RewriteObjC::RewriteObjCQualifiedInterfaceTypes(Decl *Dcl) {
Fariborz Jahanian61477f72007-12-11 22:50:14 +00001543 SourceLocation Loc;
1544 QualType Type;
1545 const FunctionTypeProto *proto = 0;
1546 if (VarDecl *VD = dyn_cast<VarDecl>(Dcl)) {
1547 Loc = VD->getLocation();
1548 Type = VD->getType();
1549 }
1550 else if (FunctionDecl *FD = dyn_cast<FunctionDecl>(Dcl)) {
1551 Loc = FD->getLocation();
1552 // Check for ObjC 'id' and class types that have been adorned with protocol
1553 // information (id<p>, C<p>*). The protocol references need to be rewritten!
1554 const FunctionType *funcType = FD->getType()->getAsFunctionType();
1555 assert(funcType && "missing function type");
1556 proto = dyn_cast<FunctionTypeProto>(funcType);
1557 if (!proto)
1558 return;
1559 Type = proto->getResultType();
1560 }
1561 else
1562 return;
Steve Naroffd5255f52007-11-01 13:24:47 +00001563
Fariborz Jahanian61477f72007-12-11 22:50:14 +00001564 if (needToScanForQualifiers(Type)) {
Steve Naroffd5255f52007-11-01 13:24:47 +00001565 // Since types are unique, we need to scan the buffer.
Steve Naroffd5255f52007-11-01 13:24:47 +00001566
1567 const char *endBuf = SM->getCharacterData(Loc);
1568 const char *startBuf = endBuf;
Steve Naroff6cafbf22008-05-31 05:02:17 +00001569 while (*startBuf != ';' && *startBuf != '<' && startBuf != MainFileStart)
Steve Naroffd5255f52007-11-01 13:24:47 +00001570 startBuf--; // scan backward (from the decl location) for return type.
1571 const char *startRef = 0, *endRef = 0;
1572 if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) {
1573 // Get the locations of the startRef, endRef.
1574 SourceLocation LessLoc = Loc.getFileLocWithOffset(startRef-endBuf);
1575 SourceLocation GreaterLoc = Loc.getFileLocWithOffset(endRef-endBuf+1);
1576 // Comment out the protocol references.
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001577 InsertText(LessLoc, "/*", 2);
1578 InsertText(GreaterLoc, "*/", 2);
Steve Naroff9165ad32007-10-31 04:38:33 +00001579 }
1580 }
Fariborz Jahanian61477f72007-12-11 22:50:14 +00001581 if (!proto)
1582 return; // most likely, was a variable
Steve Naroffd5255f52007-11-01 13:24:47 +00001583 // Now check arguments.
Fariborz Jahanian61477f72007-12-11 22:50:14 +00001584 const char *startBuf = SM->getCharacterData(Loc);
1585 const char *startFuncBuf = startBuf;
Steve Naroffd5255f52007-11-01 13:24:47 +00001586 for (unsigned i = 0; i < proto->getNumArgs(); i++) {
1587 if (needToScanForQualifiers(proto->getArgType(i))) {
1588 // Since types are unique, we need to scan the buffer.
Steve Naroffd5255f52007-11-01 13:24:47 +00001589
Steve Naroffd5255f52007-11-01 13:24:47 +00001590 const char *endBuf = startBuf;
Fariborz Jahanian61477f72007-12-11 22:50:14 +00001591 // scan forward (from the decl location) for argument types.
1592 scanToNextArgument(endBuf);
Steve Naroffd5255f52007-11-01 13:24:47 +00001593 const char *startRef = 0, *endRef = 0;
1594 if (scanForProtocolRefs(startBuf, endBuf, startRef, endRef)) {
1595 // Get the locations of the startRef, endRef.
Fariborz Jahanian291e04b2007-12-11 23:04:08 +00001596 SourceLocation LessLoc =
1597 Loc.getFileLocWithOffset(startRef-startFuncBuf);
1598 SourceLocation GreaterLoc =
1599 Loc.getFileLocWithOffset(endRef-startFuncBuf+1);
Steve Naroffd5255f52007-11-01 13:24:47 +00001600 // Comment out the protocol references.
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00001601 InsertText(LessLoc, "/*", 2);
1602 InsertText(GreaterLoc, "*/", 2);
Steve Naroffd5255f52007-11-01 13:24:47 +00001603 }
Fariborz Jahanian61477f72007-12-11 22:50:14 +00001604 startBuf = ++endBuf;
1605 }
1606 else {
1607 while (*startBuf != ')' && *startBuf != ',')
1608 startBuf++; // scan forward (from the decl location) for argument types.
1609 startBuf++;
1610 }
Steve Naroffd5255f52007-11-01 13:24:47 +00001611 }
Steve Naroff9165ad32007-10-31 04:38:33 +00001612}
1613
Fariborz Jahaniana70711b2007-12-04 21:47:40 +00001614// SynthSelGetUidFunctionDecl - SEL sel_registerName(const char *str);
Steve Naroffb29b4272008-04-14 22:03:09 +00001615void RewriteObjC::SynthSelGetUidFunctionDecl() {
Fariborz Jahaniana70711b2007-12-04 21:47:40 +00001616 IdentifierInfo *SelGetUidIdent = &Context->Idents.get("sel_registerName");
1617 llvm::SmallVector<QualType, 16> ArgTys;
1618 ArgTys.push_back(Context->getPointerType(
1619 Context->CharTy.getQualifiedType(QualType::Const)));
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001620 QualType getFuncType = Context->getFunctionType(Context->getObjCSelType(),
Fariborz Jahaniana70711b2007-12-04 21:47:40 +00001621 &ArgTys[0], ArgTys.size(),
1622 false /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001623 SelGetUidFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001624 SourceLocation(),
Fariborz Jahaniana70711b2007-12-04 21:47:40 +00001625 SelGetUidIdent, getFuncType,
1626 FunctionDecl::Extern, false, 0);
1627}
1628
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +00001629// SynthGetProtocolFunctionDecl - Protocol objc_getProtocol(const char *proto);
Steve Naroffb29b4272008-04-14 22:03:09 +00001630void RewriteObjC::SynthGetProtocolFunctionDecl() {
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +00001631 IdentifierInfo *SelGetProtoIdent = &Context->Idents.get("objc_getProtocol");
1632 llvm::SmallVector<QualType, 16> ArgTys;
1633 ArgTys.push_back(Context->getPointerType(
1634 Context->CharTy.getQualifiedType(QualType::Const)));
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001635 QualType getFuncType = Context->getFunctionType(Context->getObjCProtoType(),
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +00001636 &ArgTys[0], ArgTys.size(),
1637 false /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001638 GetProtocolFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001639 SourceLocation(),
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +00001640 SelGetProtoIdent, getFuncType,
1641 FunctionDecl::Extern, false, 0);
1642}
1643
Steve Naroffb29b4272008-04-14 22:03:09 +00001644void RewriteObjC::RewriteFunctionDecl(FunctionDecl *FD) {
Steve Naroff09b266e2007-10-30 23:14:51 +00001645 // declared in <objc/objc.h>
Steve Naroffbeaf2992007-11-03 11:27:19 +00001646 if (strcmp(FD->getName(), "sel_registerName") == 0) {
Steve Naroff09b266e2007-10-30 23:14:51 +00001647 SelGetUidFunctionDecl = FD;
Steve Naroff9165ad32007-10-31 04:38:33 +00001648 return;
1649 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001650 RewriteObjCQualifiedInterfaceTypes(FD);
Steve Naroff09b266e2007-10-30 23:14:51 +00001651}
1652
Steve Naroffc0a123c2008-03-11 17:37:02 +00001653// SynthSuperContructorFunctionDecl - id objc_super(id obj, id super);
Steve Naroffb29b4272008-04-14 22:03:09 +00001654void RewriteObjC::SynthSuperContructorFunctionDecl() {
Steve Naroffc0a123c2008-03-11 17:37:02 +00001655 if (SuperContructorFunctionDecl)
1656 return;
1657 IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_super");
1658 llvm::SmallVector<QualType, 16> ArgTys;
1659 QualType argT = Context->getObjCIdType();
1660 assert(!argT.isNull() && "Can't find 'id' type");
1661 ArgTys.push_back(argT);
1662 ArgTys.push_back(argT);
1663 QualType msgSendType = Context->getFunctionType(Context->getObjCIdType(),
1664 &ArgTys[0], ArgTys.size(),
1665 false);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001666 SuperContructorFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001667 SourceLocation(),
Steve Naroffc0a123c2008-03-11 17:37:02 +00001668 msgSendIdent, msgSendType,
1669 FunctionDecl::Extern, false, 0);
1670}
1671
Steve Naroff09b266e2007-10-30 23:14:51 +00001672// SynthMsgSendFunctionDecl - id objc_msgSend(id self, SEL op, ...);
Steve Naroffb29b4272008-04-14 22:03:09 +00001673void RewriteObjC::SynthMsgSendFunctionDecl() {
Steve Naroff09b266e2007-10-30 23:14:51 +00001674 IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend");
1675 llvm::SmallVector<QualType, 16> ArgTys;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001676 QualType argT = Context->getObjCIdType();
Steve Naroff09b266e2007-10-30 23:14:51 +00001677 assert(!argT.isNull() && "Can't find 'id' type");
1678 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001679 argT = Context->getObjCSelType();
Steve Naroff09b266e2007-10-30 23:14:51 +00001680 assert(!argT.isNull() && "Can't find 'SEL' type");
1681 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001682 QualType msgSendType = Context->getFunctionType(Context->getObjCIdType(),
Steve Naroff09b266e2007-10-30 23:14:51 +00001683 &ArgTys[0], ArgTys.size(),
1684 true /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001685 MsgSendFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001686 SourceLocation(),
Steve Naroff09b266e2007-10-30 23:14:51 +00001687 msgSendIdent, msgSendType,
1688 FunctionDecl::Extern, false, 0);
1689}
1690
Steve Naroff874e2322007-11-15 10:28:18 +00001691// SynthMsgSendSuperFunctionDecl - id objc_msgSendSuper(struct objc_super *, SEL op, ...);
Steve Naroffb29b4272008-04-14 22:03:09 +00001692void RewriteObjC::SynthMsgSendSuperFunctionDecl() {
Steve Naroff874e2322007-11-15 10:28:18 +00001693 IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSendSuper");
1694 llvm::SmallVector<QualType, 16> ArgTys;
Argyrios Kyrtzidis39ba4ae2008-06-09 23:19:58 +00001695 RecordDecl *RD = RecordDecl::Create(*Context, TagDecl::TK_struct, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001696 SourceLocation(),
Chris Lattnerc63e6602008-03-15 21:32:50 +00001697 &Context->Idents.get("objc_super"), 0);
Steve Naroff874e2322007-11-15 10:28:18 +00001698 QualType argT = Context->getPointerType(Context->getTagDeclType(RD));
1699 assert(!argT.isNull() && "Can't build 'struct objc_super *' type");
1700 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001701 argT = Context->getObjCSelType();
Steve Naroff874e2322007-11-15 10:28:18 +00001702 assert(!argT.isNull() && "Can't find 'SEL' type");
1703 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001704 QualType msgSendType = Context->getFunctionType(Context->getObjCIdType(),
Steve Naroff874e2322007-11-15 10:28:18 +00001705 &ArgTys[0], ArgTys.size(),
1706 true /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001707 MsgSendSuperFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001708 SourceLocation(),
Steve Naroff874e2322007-11-15 10:28:18 +00001709 msgSendIdent, msgSendType,
1710 FunctionDecl::Extern, false, 0);
1711}
1712
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001713// SynthMsgSendStretFunctionDecl - id objc_msgSend_stret(id self, SEL op, ...);
Steve Naroffb29b4272008-04-14 22:03:09 +00001714void RewriteObjC::SynthMsgSendStretFunctionDecl() {
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001715 IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_stret");
1716 llvm::SmallVector<QualType, 16> ArgTys;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001717 QualType argT = Context->getObjCIdType();
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001718 assert(!argT.isNull() && "Can't find 'id' type");
1719 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001720 argT = Context->getObjCSelType();
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001721 assert(!argT.isNull() && "Can't find 'SEL' type");
1722 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001723 QualType msgSendType = Context->getFunctionType(Context->getObjCIdType(),
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001724 &ArgTys[0], ArgTys.size(),
1725 true /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001726 MsgSendStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001727 SourceLocation(),
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001728 msgSendIdent, msgSendType,
1729 FunctionDecl::Extern, false, 0);
1730}
1731
1732// SynthMsgSendSuperStretFunctionDecl -
1733// id objc_msgSendSuper_stret(struct objc_super *, SEL op, ...);
Steve Naroffb29b4272008-04-14 22:03:09 +00001734void RewriteObjC::SynthMsgSendSuperStretFunctionDecl() {
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001735 IdentifierInfo *msgSendIdent =
1736 &Context->Idents.get("objc_msgSendSuper_stret");
1737 llvm::SmallVector<QualType, 16> ArgTys;
Argyrios Kyrtzidis39ba4ae2008-06-09 23:19:58 +00001738 RecordDecl *RD = RecordDecl::Create(*Context, TagDecl::TK_struct, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001739 SourceLocation(),
Chris Lattnerc63e6602008-03-15 21:32:50 +00001740 &Context->Idents.get("objc_super"), 0);
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001741 QualType argT = Context->getPointerType(Context->getTagDeclType(RD));
1742 assert(!argT.isNull() && "Can't build 'struct objc_super *' type");
1743 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001744 argT = Context->getObjCSelType();
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001745 assert(!argT.isNull() && "Can't find 'SEL' type");
1746 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001747 QualType msgSendType = Context->getFunctionType(Context->getObjCIdType(),
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001748 &ArgTys[0], ArgTys.size(),
1749 true /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001750 MsgSendSuperStretFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattnera98e58d2008-03-15 21:24:04 +00001751 SourceLocation(),
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001752 msgSendIdent, msgSendType,
1753 FunctionDecl::Extern, false, 0);
1754}
1755
Steve Naroff1284db82008-05-08 22:02:18 +00001756// SynthMsgSendFpretFunctionDecl - double objc_msgSend_fpret(id self, SEL op, ...);
Steve Naroffb29b4272008-04-14 22:03:09 +00001757void RewriteObjC::SynthMsgSendFpretFunctionDecl() {
Fariborz Jahanianacb49772007-12-03 21:26:48 +00001758 IdentifierInfo *msgSendIdent = &Context->Idents.get("objc_msgSend_fpret");
1759 llvm::SmallVector<QualType, 16> ArgTys;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001760 QualType argT = Context->getObjCIdType();
Fariborz Jahanianacb49772007-12-03 21:26:48 +00001761 assert(!argT.isNull() && "Can't find 'id' type");
1762 ArgTys.push_back(argT);
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001763 argT = Context->getObjCSelType();
Fariborz Jahanianacb49772007-12-03 21:26:48 +00001764 assert(!argT.isNull() && "Can't find 'SEL' type");
1765 ArgTys.push_back(argT);
Steve Naroff1284db82008-05-08 22:02:18 +00001766 QualType msgSendType = Context->getFunctionType(Context->DoubleTy,
Fariborz Jahanianacb49772007-12-03 21:26:48 +00001767 &ArgTys[0], ArgTys.size(),
1768 true /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001769 MsgSendFpretFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001770 SourceLocation(),
Fariborz Jahanianacb49772007-12-03 21:26:48 +00001771 msgSendIdent, msgSendType,
1772 FunctionDecl::Extern, false, 0);
1773}
1774
Steve Naroff09b266e2007-10-30 23:14:51 +00001775// SynthGetClassFunctionDecl - id objc_getClass(const char *name);
Steve Naroffb29b4272008-04-14 22:03:09 +00001776void RewriteObjC::SynthGetClassFunctionDecl() {
Steve Naroff09b266e2007-10-30 23:14:51 +00001777 IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getClass");
1778 llvm::SmallVector<QualType, 16> ArgTys;
1779 ArgTys.push_back(Context->getPointerType(
1780 Context->CharTy.getQualifiedType(QualType::Const)));
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001781 QualType getClassType = Context->getFunctionType(Context->getObjCIdType(),
Steve Naroff09b266e2007-10-30 23:14:51 +00001782 &ArgTys[0], ArgTys.size(),
1783 false /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001784 GetClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001785 SourceLocation(),
Steve Naroff09b266e2007-10-30 23:14:51 +00001786 getClassIdent, getClassType,
1787 FunctionDecl::Extern, false, 0);
1788}
1789
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001790// SynthGetMetaClassFunctionDecl - id objc_getClass(const char *name);
Steve Naroffb29b4272008-04-14 22:03:09 +00001791void RewriteObjC::SynthGetMetaClassFunctionDecl() {
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001792 IdentifierInfo *getClassIdent = &Context->Idents.get("objc_getMetaClass");
1793 llvm::SmallVector<QualType, 16> ArgTys;
1794 ArgTys.push_back(Context->getPointerType(
1795 Context->CharTy.getQualifiedType(QualType::Const)));
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001796 QualType getClassType = Context->getFunctionType(Context->getObjCIdType(),
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001797 &ArgTys[0], ArgTys.size(),
1798 false /*isVariadic*/);
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001799 GetMetaClassFunctionDecl = FunctionDecl::Create(*Context, TUDecl,
Chris Lattner0ed844b2008-04-04 06:12:32 +00001800 SourceLocation(),
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001801 getClassIdent, getClassType,
1802 FunctionDecl::Extern, false, 0);
1803}
1804
Steve Naroffb29b4272008-04-14 22:03:09 +00001805Stmt *RewriteObjC::RewriteObjCStringLiteral(ObjCStringLiteral *Exp) {
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001806 QualType strType = getConstantStringStructType();
1807
1808 std::string S = "__NSConstantStringImpl_";
Steve Naroff7691d9b2008-05-31 03:35:42 +00001809
1810 std::string tmpName = InFileName;
1811 unsigned i;
1812 for (i=0; i < tmpName.length(); i++) {
1813 char c = tmpName.at(i);
1814 // replace any non alphanumeric characters with '_'.
1815 if (!isalpha(c) && (c < '0' || c > '9'))
1816 tmpName[i] = '_';
1817 }
1818 S += tmpName;
1819 S += "_";
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001820 S += utostr(NumObjCStringLiterals++);
1821
Steve Naroffba92b2e2008-03-27 22:29:16 +00001822 Preamble += "static __NSConstantStringImpl " + S;
1823 Preamble += " __attribute__ ((section (\"__DATA, __cfstring\"))) = {__CFConstantStringClassReference,";
1824 Preamble += "0x000007c8,"; // utf8_str
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001825 // The pretty printer for StringLiteral handles escape characters properly.
1826 std::ostringstream prettyBuf;
1827 Exp->getString()->printPretty(prettyBuf);
Steve Naroffba92b2e2008-03-27 22:29:16 +00001828 Preamble += prettyBuf.str();
1829 Preamble += ",";
Steve Naroff3652c2d2008-03-15 01:36:04 +00001830 // The minus 2 removes the begin/end double quotes.
Steve Naroffba92b2e2008-03-27 22:29:16 +00001831 Preamble += utostr(prettyBuf.str().size()-2) + "};\n";
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001832
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00001833 VarDecl *NewVD = VarDecl::Create(*Context, TUDecl, SourceLocation(),
Steve Naroff248a7532008-04-15 22:42:06 +00001834 &Context->Idents.get(S.c_str()), strType,
1835 VarDecl::Static, NULL);
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001836 DeclRefExpr *DRE = new DeclRefExpr(NewVD, strType, SourceLocation());
1837 Expr *Unop = new UnaryOperator(DRE, UnaryOperator::AddrOf,
1838 Context->getPointerType(DRE->getType()),
1839 SourceLocation());
Steve Naroff96984642007-11-08 14:30:50 +00001840 // cast to NSConstantString *
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001841 CastExpr *cast = new CastExpr(Exp->getType(), Unop, SourceLocation());
Chris Lattnerdcbc5b02008-01-31 19:37:57 +00001842 ReplaceStmt(Exp, cast);
Steve Naroff96984642007-11-08 14:30:50 +00001843 delete Exp;
1844 return cast;
Steve Naroffbeaf2992007-11-03 11:27:19 +00001845}
1846
Steve Naroffb29b4272008-04-14 22:03:09 +00001847ObjCInterfaceDecl *RewriteObjC::isSuperReceiver(Expr *recExpr) {
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001848 // check if we are sending a message to 'super'
Chris Lattner6c2b6eb2008-03-15 06:12:44 +00001849 if (!CurMethodDecl || !CurMethodDecl->isInstance()) return 0;
1850
1851 CastExpr *CE = dyn_cast<CastExpr>(recExpr);
1852 if (!CE) return 0;
1853
1854 DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CE->getSubExpr());
1855 if (!DRE) return 0;
1856
1857 ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl());
1858 if (!PVD) return 0;
1859
1860 if (strcmp(PVD->getName(), "self") != 0)
1861 return 0;
1862
1863 // is this id<P1..> type?
1864 if (CE->getType()->isObjCQualifiedIdType())
1865 return 0;
1866 const PointerType *PT = CE->getType()->getAsPointerType();
1867 if (!PT) return 0;
1868
1869 ObjCInterfaceType *IT = dyn_cast<ObjCInterfaceType>(PT->getPointeeType());
1870 if (!IT) return 0;
1871
1872 if (IT->getDecl() != CurMethodDecl->getClassInterface()->getSuperClass())
1873 return 0;
1874
1875 return IT->getDecl();
Steve Naroff874e2322007-11-15 10:28:18 +00001876}
1877
1878// struct objc_super { struct objc_object *receiver; struct objc_class *super; };
Steve Naroffb29b4272008-04-14 22:03:09 +00001879QualType RewriteObjC::getSuperStructType() {
Steve Naroff874e2322007-11-15 10:28:18 +00001880 if (!SuperStructDecl) {
Argyrios Kyrtzidis39ba4ae2008-06-09 23:19:58 +00001881 SuperStructDecl = RecordDecl::Create(*Context, TagDecl::TK_struct, TUDecl,
Chris Lattnerc63e6602008-03-15 21:32:50 +00001882 SourceLocation(),
1883 &Context->Idents.get("objc_super"), 0);
Steve Naroff874e2322007-11-15 10:28:18 +00001884 QualType FieldTypes[2];
1885
1886 // struct objc_object *receiver;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001887 FieldTypes[0] = Context->getObjCIdType();
Steve Naroff874e2322007-11-15 10:28:18 +00001888 // struct objc_class *super;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001889 FieldTypes[1] = Context->getObjCClassType();
Steve Naroff874e2322007-11-15 10:28:18 +00001890 // Create fields
1891 FieldDecl *FieldDecls[2];
1892
1893 for (unsigned i = 0; i < 2; ++i)
Chris Lattnerb048c982008-04-06 04:47:34 +00001894 FieldDecls[i] = FieldDecl::Create(*Context, SourceLocation(), 0,
Chris Lattner8e25d862008-03-16 00:16:02 +00001895 FieldTypes[i]);
Steve Naroff874e2322007-11-15 10:28:18 +00001896
1897 SuperStructDecl->defineBody(FieldDecls, 4);
1898 }
1899 return Context->getTagDeclType(SuperStructDecl);
1900}
1901
Steve Naroffb29b4272008-04-14 22:03:09 +00001902QualType RewriteObjC::getConstantStringStructType() {
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001903 if (!ConstantStringDecl) {
Argyrios Kyrtzidis39ba4ae2008-06-09 23:19:58 +00001904 ConstantStringDecl = RecordDecl::Create(*Context, TagDecl::TK_struct, TUDecl,
Chris Lattnerc63e6602008-03-15 21:32:50 +00001905 SourceLocation(),
1906 &Context->Idents.get("__NSConstantStringImpl"), 0);
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001907 QualType FieldTypes[4];
1908
1909 // struct objc_object *receiver;
1910 FieldTypes[0] = Context->getObjCIdType();
1911 // int flags;
1912 FieldTypes[1] = Context->IntTy;
1913 // char *str;
1914 FieldTypes[2] = Context->getPointerType(Context->CharTy);
1915 // long length;
1916 FieldTypes[3] = Context->LongTy;
1917 // Create fields
Steve Naroff9630ec52008-03-27 22:59:54 +00001918 FieldDecl *FieldDecls[4];
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001919
1920 for (unsigned i = 0; i < 4; ++i)
Chris Lattnerb048c982008-04-06 04:47:34 +00001921 FieldDecls[i] = FieldDecl::Create(*Context, SourceLocation(), 0,
Chris Lattner8e25d862008-03-16 00:16:02 +00001922 FieldTypes[i]);
Steve Naroffd82a9ab2008-03-15 00:55:56 +00001923
1924 ConstantStringDecl->defineBody(FieldDecls, 4);
1925 }
1926 return Context->getTagDeclType(ConstantStringDecl);
1927}
1928
Steve Naroffb29b4272008-04-14 22:03:09 +00001929Stmt *RewriteObjC::SynthMessageExpr(ObjCMessageExpr *Exp) {
Fariborz Jahaniana70711b2007-12-04 21:47:40 +00001930 if (!SelGetUidFunctionDecl)
1931 SynthSelGetUidFunctionDecl();
Steve Naroff09b266e2007-10-30 23:14:51 +00001932 if (!MsgSendFunctionDecl)
1933 SynthMsgSendFunctionDecl();
Steve Naroff874e2322007-11-15 10:28:18 +00001934 if (!MsgSendSuperFunctionDecl)
1935 SynthMsgSendSuperFunctionDecl();
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001936 if (!MsgSendStretFunctionDecl)
1937 SynthMsgSendStretFunctionDecl();
1938 if (!MsgSendSuperStretFunctionDecl)
1939 SynthMsgSendSuperStretFunctionDecl();
Fariborz Jahanianacb49772007-12-03 21:26:48 +00001940 if (!MsgSendFpretFunctionDecl)
1941 SynthMsgSendFpretFunctionDecl();
Steve Naroff09b266e2007-10-30 23:14:51 +00001942 if (!GetClassFunctionDecl)
1943 SynthGetClassFunctionDecl();
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001944 if (!GetMetaClassFunctionDecl)
1945 SynthGetMetaClassFunctionDecl();
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001946
Steve Naroff874e2322007-11-15 10:28:18 +00001947 // default to objc_msgSend().
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001948 FunctionDecl *MsgSendFlavor = MsgSendFunctionDecl;
1949 // May need to use objc_msgSend_stret() as well.
1950 FunctionDecl *MsgSendStretFlavor = 0;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001951 if (ObjCMethodDecl *mDecl = Exp->getMethodDecl()) {
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001952 QualType resultType = mDecl->getResultType();
1953 if (resultType.getCanonicalType()->isStructureType()
1954 || resultType.getCanonicalType()->isUnionType())
1955 MsgSendStretFlavor = MsgSendStretFunctionDecl;
Fariborz Jahanianacb49772007-12-03 21:26:48 +00001956 else if (resultType.getCanonicalType()->isRealFloatingType())
1957 MsgSendFlavor = MsgSendFpretFunctionDecl;
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00001958 }
Steve Naroff874e2322007-11-15 10:28:18 +00001959
Steve Naroff934f2762007-10-24 22:48:43 +00001960 // Synthesize a call to objc_msgSend().
1961 llvm::SmallVector<Expr*, 8> MsgExprs;
1962 IdentifierInfo *clsName = Exp->getClassName();
1963
1964 // Derive/push the receiver/selector, 2 implicit arguments to objc_msgSend().
1965 if (clsName) { // class message.
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001966 if (!strcmp(clsName->getName(), "super")) {
1967 MsgSendFlavor = MsgSendSuperFunctionDecl;
1968 if (MsgSendStretFlavor)
1969 MsgSendStretFlavor = MsgSendSuperStretFunctionDecl;
1970 assert(MsgSendFlavor && "MsgSendFlavor is NULL!");
1971
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001972 ObjCInterfaceDecl *SuperDecl =
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001973 CurMethodDecl->getClassInterface()->getSuperClass();
1974
1975 llvm::SmallVector<Expr*, 4> InitExprs;
1976
1977 // set the receiver to self, the first argument to all methods.
Chris Lattner41110242008-06-17 18:05:57 +00001978 InitExprs.push_back(new DeclRefExpr(
1979 CurMethodDecl->getSelfDecl(),
1980 Context->getObjCIdType(),
1981 SourceLocation()));
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001982 llvm::SmallVector<Expr*, 8> ClsExprs;
1983 QualType argType = Context->getPointerType(Context->CharTy);
1984 ClsExprs.push_back(new StringLiteral(SuperDecl->getIdentifier()->getName(),
1985 SuperDecl->getIdentifier()->getLength(),
1986 false, argType, SourceLocation(),
1987 SourceLocation()));
1988 CallExpr *Cls = SynthesizeCallToFunctionDecl(GetMetaClassFunctionDecl,
1989 &ClsExprs[0],
1990 ClsExprs.size());
1991 // To turn off a warning, type-cast to 'id'
1992 InitExprs.push_back(
Ted Kremeneka526c5c2008-01-07 19:49:32 +00001993 new CastExpr(Context->getObjCIdType(),
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00001994 Cls, SourceLocation())); // set 'super class', using objc_getClass().
1995 // struct objc_super
1996 QualType superType = getSuperStructType();
Steve Naroff23f41272008-03-11 18:14:26 +00001997 Expr *SuperRep;
Steve Naroffc0a123c2008-03-11 17:37:02 +00001998
Steve Naroff23f41272008-03-11 18:14:26 +00001999 if (LangOpts.Microsoft) {
2000 SynthSuperContructorFunctionDecl();
2001 // Simulate a contructor call...
2002 DeclRefExpr *DRE = new DeclRefExpr(SuperContructorFunctionDecl,
2003 superType, SourceLocation());
2004 SuperRep = new CallExpr(DRE, &InitExprs[0], InitExprs.size(),
2005 superType, SourceLocation());
2006 } else {
2007 // (struct objc_super) { <exprs from above> }
2008 InitListExpr *ILE = new InitListExpr(SourceLocation(),
2009 &InitExprs[0], InitExprs.size(),
2010 SourceLocation());
2011 SuperRep = new CompoundLiteralExpr(SourceLocation(), superType, ILE, false);
2012 }
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00002013 // struct objc_super *
2014 Expr *Unop = new UnaryOperator(SuperRep, UnaryOperator::AddrOf,
2015 Context->getPointerType(SuperRep->getType()),
2016 SourceLocation());
2017 MsgExprs.push_back(Unop);
2018 } else {
2019 llvm::SmallVector<Expr*, 8> ClsExprs;
2020 QualType argType = Context->getPointerType(Context->CharTy);
2021 ClsExprs.push_back(new StringLiteral(clsName->getName(),
2022 clsName->getLength(),
2023 false, argType, SourceLocation(),
2024 SourceLocation()));
2025 CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl,
2026 &ClsExprs[0],
2027 ClsExprs.size());
2028 MsgExprs.push_back(Cls);
2029 }
Steve Naroff6568d4d2007-11-14 23:54:14 +00002030 } else { // instance message.
2031 Expr *recExpr = Exp->getReceiver();
Steve Naroff874e2322007-11-15 10:28:18 +00002032
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002033 if (ObjCInterfaceDecl *SuperDecl = isSuperReceiver(recExpr)) {
Steve Naroff874e2322007-11-15 10:28:18 +00002034 MsgSendFlavor = MsgSendSuperFunctionDecl;
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00002035 if (MsgSendStretFlavor)
2036 MsgSendStretFlavor = MsgSendSuperStretFunctionDecl;
Steve Naroff874e2322007-11-15 10:28:18 +00002037 assert(MsgSendFlavor && "MsgSendFlavor is NULL!");
2038
2039 llvm::SmallVector<Expr*, 4> InitExprs;
2040
Fariborz Jahanianceee3e82007-12-04 22:32:58 +00002041 InitExprs.push_back(
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002042 new CastExpr(Context->getObjCIdType(),
Fariborz Jahanianceee3e82007-12-04 22:32:58 +00002043 recExpr, SourceLocation())); // set the 'receiver'.
Steve Naroff874e2322007-11-15 10:28:18 +00002044
2045 llvm::SmallVector<Expr*, 8> ClsExprs;
2046 QualType argType = Context->getPointerType(Context->CharTy);
Steve Naroff9bcb5fc2007-12-07 03:50:46 +00002047 ClsExprs.push_back(new StringLiteral(SuperDecl->getIdentifier()->getName(),
2048 SuperDecl->getIdentifier()->getLength(),
Steve Naroff874e2322007-11-15 10:28:18 +00002049 false, argType, SourceLocation(),
2050 SourceLocation()));
2051 CallExpr *Cls = SynthesizeCallToFunctionDecl(GetClassFunctionDecl,
Fariborz Jahanianceee3e82007-12-04 22:32:58 +00002052 &ClsExprs[0],
2053 ClsExprs.size());
Fariborz Jahanian71274312007-12-05 17:29:46 +00002054 // To turn off a warning, type-cast to 'id'
Fariborz Jahanianceee3e82007-12-04 22:32:58 +00002055 InitExprs.push_back(
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002056 new CastExpr(Context->getObjCIdType(),
Fariborz Jahanianceee3e82007-12-04 22:32:58 +00002057 Cls, SourceLocation())); // set 'super class', using objc_getClass().
Steve Naroff874e2322007-11-15 10:28:18 +00002058 // struct objc_super
2059 QualType superType = getSuperStructType();
Steve Naroffc0a123c2008-03-11 17:37:02 +00002060 Expr *SuperRep;
2061
2062 if (LangOpts.Microsoft) {
2063 SynthSuperContructorFunctionDecl();
2064 // Simulate a contructor call...
2065 DeclRefExpr *DRE = new DeclRefExpr(SuperContructorFunctionDecl,
2066 superType, SourceLocation());
2067 SuperRep = new CallExpr(DRE, &InitExprs[0], InitExprs.size(),
2068 superType, SourceLocation());
2069 } else {
2070 // (struct objc_super) { <exprs from above> }
2071 InitListExpr *ILE = new InitListExpr(SourceLocation(),
2072 &InitExprs[0], InitExprs.size(),
2073 SourceLocation());
2074 SuperRep = new CompoundLiteralExpr(SourceLocation(), superType, ILE, false);
2075 }
Steve Naroff874e2322007-11-15 10:28:18 +00002076 // struct objc_super *
2077 Expr *Unop = new UnaryOperator(SuperRep, UnaryOperator::AddrOf,
2078 Context->getPointerType(SuperRep->getType()),
2079 SourceLocation());
2080 MsgExprs.push_back(Unop);
2081 } else {
Fariborz Jahanian7dd82832007-12-07 21:21:21 +00002082 // Remove all type-casts because it may contain objc-style types; e.g.
2083 // Foo<Proto> *.
2084 while (CastExpr *CE = dyn_cast<CastExpr>(recExpr))
2085 recExpr = CE->getSubExpr();
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002086 recExpr = new CastExpr(Context->getObjCIdType(), recExpr, SourceLocation());
Steve Naroff874e2322007-11-15 10:28:18 +00002087 MsgExprs.push_back(recExpr);
2088 }
Steve Naroff6568d4d2007-11-14 23:54:14 +00002089 }
Steve Naroffbeaf2992007-11-03 11:27:19 +00002090 // Create a call to sel_registerName("selName"), it will be the 2nd argument.
Steve Naroff934f2762007-10-24 22:48:43 +00002091 llvm::SmallVector<Expr*, 8> SelExprs;
2092 QualType argType = Context->getPointerType(Context->CharTy);
2093 SelExprs.push_back(new StringLiteral(Exp->getSelector().getName().c_str(),
2094 Exp->getSelector().getName().size(),
2095 false, argType, SourceLocation(),
2096 SourceLocation()));
2097 CallExpr *SelExp = SynthesizeCallToFunctionDecl(SelGetUidFunctionDecl,
2098 &SelExprs[0], SelExprs.size());
2099 MsgExprs.push_back(SelExp);
2100
2101 // Now push any user supplied arguments.
2102 for (unsigned i = 0; i < Exp->getNumArgs(); i++) {
Steve Naroff6568d4d2007-11-14 23:54:14 +00002103 Expr *userExpr = Exp->getArg(i);
Steve Naroff7e3411b2007-11-15 02:58:25 +00002104 // Make all implicit casts explicit...ICE comes in handy:-)
2105 if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(userExpr)) {
2106 // Reuse the ICE type, it is exactly what the doctor ordered.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002107 userExpr = new CastExpr(ICE->getType()->isObjCQualifiedIdType()
2108 ? Context->getObjCIdType()
Fariborz Jahanianc5692492007-12-17 21:03:50 +00002109 : ICE->getType(), userExpr, SourceLocation());
Fariborz Jahaniand58fabf2007-12-18 21:33:44 +00002110 }
2111 // Make id<P...> cast into an 'id' cast.
2112 else if (CastExpr *CE = dyn_cast<CastExpr>(userExpr)) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002113 if (CE->getType()->isObjCQualifiedIdType()) {
Fariborz Jahaniand58fabf2007-12-18 21:33:44 +00002114 while ((CE = dyn_cast<CastExpr>(userExpr)))
2115 userExpr = CE->getSubExpr();
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002116 userExpr = new CastExpr(Context->getObjCIdType(),
Fariborz Jahaniand58fabf2007-12-18 21:33:44 +00002117 userExpr, SourceLocation());
2118 }
Steve Naroff7e3411b2007-11-15 02:58:25 +00002119 }
Steve Naroff6568d4d2007-11-14 23:54:14 +00002120 MsgExprs.push_back(userExpr);
Steve Naroff934f2762007-10-24 22:48:43 +00002121 // We've transferred the ownership to MsgExprs. Null out the argument in
2122 // the original expression, since we will delete it below.
2123 Exp->setArg(i, 0);
2124 }
Steve Naroffab972d32007-11-04 22:37:50 +00002125 // Generate the funky cast.
2126 CastExpr *cast;
2127 llvm::SmallVector<QualType, 8> ArgTypes;
2128 QualType returnType;
2129
2130 // Push 'id' and 'SEL', the 2 implicit arguments.
Steve Naroffc3a438c2007-11-15 10:43:57 +00002131 if (MsgSendFlavor == MsgSendSuperFunctionDecl)
2132 ArgTypes.push_back(Context->getPointerType(getSuperStructType()));
2133 else
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002134 ArgTypes.push_back(Context->getObjCIdType());
2135 ArgTypes.push_back(Context->getObjCSelType());
2136 if (ObjCMethodDecl *mDecl = Exp->getMethodDecl()) {
Steve Naroffab972d32007-11-04 22:37:50 +00002137 // Push any user argument types.
Chris Lattner58cce3b2008-03-16 01:07:14 +00002138 for (unsigned i = 0; i < mDecl->getNumParams(); i++) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002139 QualType t = mDecl->getParamDecl(i)->getType()->isObjCQualifiedIdType()
2140 ? Context->getObjCIdType()
Fariborz Jahanianc5692492007-12-17 21:03:50 +00002141 : mDecl->getParamDecl(i)->getType();
Steve Naroff352336b2007-11-05 14:36:37 +00002142 ArgTypes.push_back(t);
2143 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002144 returnType = mDecl->getResultType()->isObjCQualifiedIdType()
2145 ? Context->getObjCIdType() : mDecl->getResultType();
Steve Naroffab972d32007-11-04 22:37:50 +00002146 } else {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002147 returnType = Context->getObjCIdType();
Steve Naroffab972d32007-11-04 22:37:50 +00002148 }
2149 // Get the type, we will need to reference it in a couple spots.
Steve Naroff874e2322007-11-15 10:28:18 +00002150 QualType msgSendType = MsgSendFlavor->getType();
Steve Naroffab972d32007-11-04 22:37:50 +00002151
2152 // Create a reference to the objc_msgSend() declaration.
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00002153 DeclRefExpr *DRE = new DeclRefExpr(MsgSendFlavor, msgSendType,
2154 SourceLocation());
Steve Naroffab972d32007-11-04 22:37:50 +00002155
2156 // Need to cast objc_msgSend to "void *" (to workaround a GCC bandaid).
2157 // If we don't do this cast, we get the following bizarre warning/note:
2158 // xx.m:13: warning: function called through a non-compatible type
2159 // xx.m:13: note: if this code is reached, the program will abort
2160 cast = new CastExpr(Context->getPointerType(Context->VoidTy), DRE,
2161 SourceLocation());
Steve Naroff335eafa2007-11-15 12:35:21 +00002162
Steve Naroffab972d32007-11-04 22:37:50 +00002163 // Now do the "normal" pointer to function cast.
2164 QualType castType = Context->getFunctionType(returnType,
Fariborz Jahaniand0ee6f92007-12-06 19:49:56 +00002165 &ArgTypes[0], ArgTypes.size(),
Steve Naroff2679e482008-03-18 02:02:04 +00002166 // If we don't have a method decl, force a variadic cast.
2167 Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : true);
Steve Naroffab972d32007-11-04 22:37:50 +00002168 castType = Context->getPointerType(castType);
2169 cast = new CastExpr(castType, cast, SourceLocation());
2170
2171 // Don't forget the parens to enforce the proper binding.
2172 ParenExpr *PE = new ParenExpr(SourceLocation(), SourceLocation(), cast);
2173
2174 const FunctionType *FT = msgSendType->getAsFunctionType();
2175 CallExpr *CE = new CallExpr(PE, &MsgExprs[0], MsgExprs.size(),
2176 FT->getResultType(), SourceLocation());
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00002177 Stmt *ReplacingStmt = CE;
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00002178 if (MsgSendStretFlavor) {
2179 // We have the method which returns a struct/union. Must also generate
2180 // call to objc_msgSend_stret and hang both varieties on a conditional
2181 // expression which dictate which one to envoke depending on size of
2182 // method's return type.
2183
2184 // Create a reference to the objc_msgSend_stret() declaration.
2185 DeclRefExpr *STDRE = new DeclRefExpr(MsgSendStretFlavor, msgSendType,
2186 SourceLocation());
2187 // Need to cast objc_msgSend_stret to "void *" (see above comment).
2188 cast = new CastExpr(Context->getPointerType(Context->VoidTy), STDRE,
2189 SourceLocation());
2190 // Now do the "normal" pointer to function cast.
2191 castType = Context->getFunctionType(returnType,
Fariborz Jahaniand0ee6f92007-12-06 19:49:56 +00002192 &ArgTypes[0], ArgTypes.size(),
2193 Exp->getMethodDecl() ? Exp->getMethodDecl()->isVariadic() : false);
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00002194 castType = Context->getPointerType(castType);
2195 cast = new CastExpr(castType, cast, SourceLocation());
2196
2197 // Don't forget the parens to enforce the proper binding.
2198 PE = new ParenExpr(SourceLocation(), SourceLocation(), cast);
2199
2200 FT = msgSendType->getAsFunctionType();
2201 CallExpr *STCE = new CallExpr(PE, &MsgExprs[0], MsgExprs.size(),
2202 FT->getResultType(), SourceLocation());
2203
2204 // Build sizeof(returnType)
2205 SizeOfAlignOfTypeExpr *sizeofExpr = new SizeOfAlignOfTypeExpr(true,
2206 returnType, Context->getSizeType(),
2207 SourceLocation(), SourceLocation());
2208 // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...))
2209 // FIXME: Value of 8 is base on ppc32/x86 ABI for the most common cases.
2210 // For X86 it is more complicated and some kind of target specific routine
2211 // is needed to decide what to do.
Chris Lattner98be4942008-03-05 18:54:05 +00002212 unsigned IntSize =
2213 static_cast<unsigned>(Context->getTypeSize(Context->IntTy));
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00002214 IntegerLiteral *limit = new IntegerLiteral(llvm::APInt(IntSize, 8),
2215 Context->IntTy,
2216 SourceLocation());
2217 BinaryOperator *lessThanExpr = new BinaryOperator(sizeofExpr, limit,
2218 BinaryOperator::LE,
2219 Context->IntTy,
2220 SourceLocation());
2221 // (sizeof(returnType) <= 8 ? objc_msgSend(...) : objc_msgSend_stret(...))
2222 ConditionalOperator *CondExpr =
2223 new ConditionalOperator(lessThanExpr, CE, STCE, returnType);
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00002224 ReplacingStmt = new ParenExpr(SourceLocation(), SourceLocation(), CondExpr);
Fariborz Jahanian80a6a5a2007-12-03 19:17:29 +00002225 }
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00002226 return ReplacingStmt;
2227}
2228
Steve Naroffb29b4272008-04-14 22:03:09 +00002229Stmt *RewriteObjC::RewriteMessageExpr(ObjCMessageExpr *Exp) {
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00002230 Stmt *ReplacingStmt = SynthMessageExpr(Exp);
Steve Naroff934f2762007-10-24 22:48:43 +00002231 // Now do the actual rewrite.
Chris Lattnerdcbc5b02008-01-31 19:37:57 +00002232 ReplaceStmt(Exp, ReplacingStmt);
Steve Naroff934f2762007-10-24 22:48:43 +00002233
Chris Lattnere64b7772007-10-24 16:57:36 +00002234 delete Exp;
Fariborz Jahanian33b9c4e2008-01-08 22:06:28 +00002235 return ReplacingStmt;
Steve Naroffebf2b562007-10-23 23:50:29 +00002236}
2237
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +00002238/// RewriteObjCProtocolExpr - Rewrite a protocol expression into
2239/// call to objc_getProtocol("proto-name").
Steve Naroffb29b4272008-04-14 22:03:09 +00002240Stmt *RewriteObjC::RewriteObjCProtocolExpr(ObjCProtocolExpr *Exp) {
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +00002241 if (!GetProtocolFunctionDecl)
2242 SynthGetProtocolFunctionDecl();
2243 // Create a call to objc_getProtocol("ProtocolName").
2244 llvm::SmallVector<Expr*, 8> ProtoExprs;
2245 QualType argType = Context->getPointerType(Context->CharTy);
2246 ProtoExprs.push_back(new StringLiteral(Exp->getProtocol()->getName(),
2247 strlen(Exp->getProtocol()->getName()),
2248 false, argType, SourceLocation(),
2249 SourceLocation()));
2250 CallExpr *ProtoExp = SynthesizeCallToFunctionDecl(GetProtocolFunctionDecl,
2251 &ProtoExprs[0],
2252 ProtoExprs.size());
Chris Lattnerdcbc5b02008-01-31 19:37:57 +00002253 ReplaceStmt(Exp, ProtoExp);
Fariborz Jahanian36ee2cb2007-12-07 18:47:10 +00002254 delete Exp;
2255 return ProtoExp;
2256
2257}
2258
Steve Naroffbaf58c32008-05-31 14:15:04 +00002259bool RewriteObjC::BufferContainsPPDirectives(const char *startBuf,
2260 const char *endBuf) {
2261 while (startBuf < endBuf) {
2262 if (*startBuf == '#') {
2263 // Skip whitespace.
2264 for (++startBuf; startBuf[0] == ' ' || startBuf[0] == '\t'; ++startBuf)
2265 ;
2266 if (!strncmp(startBuf, "if", strlen("if")) ||
2267 !strncmp(startBuf, "ifdef", strlen("ifdef")) ||
2268 !strncmp(startBuf, "ifndef", strlen("ifndef")) ||
2269 !strncmp(startBuf, "define", strlen("define")) ||
2270 !strncmp(startBuf, "undef", strlen("undef")) ||
2271 !strncmp(startBuf, "else", strlen("else")) ||
2272 !strncmp(startBuf, "elif", strlen("elif")) ||
2273 !strncmp(startBuf, "endif", strlen("endif")) ||
2274 !strncmp(startBuf, "pragma", strlen("pragma")) ||
2275 !strncmp(startBuf, "include", strlen("include")) ||
2276 !strncmp(startBuf, "import", strlen("import")) ||
2277 !strncmp(startBuf, "include_next", strlen("include_next")))
2278 return true;
2279 }
2280 startBuf++;
2281 }
2282 return false;
2283}
2284
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002285/// SynthesizeObjCInternalStruct - Rewrite one internal struct corresponding to
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002286/// an objective-c class with ivars.
Steve Naroffb29b4272008-04-14 22:03:09 +00002287void RewriteObjC::SynthesizeObjCInternalStruct(ObjCInterfaceDecl *CDecl,
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002288 std::string &Result) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002289 assert(CDecl && "Class missing in SynthesizeObjCInternalStruct");
2290 assert(CDecl->getName() && "Name missing in SynthesizeObjCInternalStruct");
Fariborz Jahanian212b7682007-10-31 23:08:24 +00002291 // Do not synthesize more than once.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002292 if (ObjCSynthesizedStructs.count(CDecl))
Fariborz Jahanian212b7682007-10-31 23:08:24 +00002293 return;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002294 ObjCInterfaceDecl *RCDecl = CDecl->getSuperClass();
Chris Lattnerf3a7af92008-03-16 21:08:55 +00002295 int NumIvars = CDecl->ivar_size();
Steve Narofffea763e82007-11-14 19:25:57 +00002296 SourceLocation LocStart = CDecl->getLocStart();
2297 SourceLocation LocEnd = CDecl->getLocEnd();
2298
2299 const char *startBuf = SM->getCharacterData(LocStart);
2300 const char *endBuf = SM->getCharacterData(LocEnd);
Steve Naroffbaf58c32008-05-31 14:15:04 +00002301
Fariborz Jahanian2c7038b2007-11-26 19:52:57 +00002302 // If no ivars and no root or if its root, directly or indirectly,
2303 // have no ivars (thus not synthesized) then no need to synthesize this class.
Chris Lattnerf3a7af92008-03-16 21:08:55 +00002304 if ((CDecl->isForwardDecl() || NumIvars == 0) &&
2305 (!RCDecl || !ObjCSynthesizedStructs.count(RCDecl))) {
Fariborz Jahanian2c7038b2007-11-26 19:52:57 +00002306 endBuf += Lexer::MeasureTokenLength(LocEnd, *SM);
Chris Lattneraadaf782008-01-31 19:51:04 +00002307 ReplaceText(LocStart, endBuf-startBuf, Result.c_str(), Result.size());
Fariborz Jahanian2c7038b2007-11-26 19:52:57 +00002308 return;
2309 }
2310
2311 // FIXME: This has potential of causing problem. If
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002312 // SynthesizeObjCInternalStruct is ever called recursively.
Fariborz Jahanian2c7038b2007-11-26 19:52:57 +00002313 Result += "\nstruct ";
2314 Result += CDecl->getName();
Steve Naroff61ed9ca2008-03-10 23:16:54 +00002315 if (LangOpts.Microsoft)
2316 Result += "_IMPL";
Steve Naroff05b8c782008-03-12 00:25:36 +00002317
Fariborz Jahanianfdc08a02007-10-31 17:29:28 +00002318 if (NumIvars > 0) {
Steve Narofffea763e82007-11-14 19:25:57 +00002319 const char *cursor = strchr(startBuf, '{');
2320 assert((cursor && endBuf)
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002321 && "SynthesizeObjCInternalStruct - malformed @interface");
Steve Naroffbaf58c32008-05-31 14:15:04 +00002322 // If the buffer contains preprocessor directives, we do more fine-grained
2323 // rewrites. This is intended to fix code that looks like (which occurs in
2324 // NSURL.h, for example):
2325 //
2326 // #ifdef XYZ
2327 // @interface Foo : NSObject
2328 // #else
2329 // @interface FooBar : NSObject
2330 // #endif
2331 // {
2332 // int i;
2333 // }
2334 // @end
2335 //
2336 // This clause is segregated to avoid breaking the common case.
2337 if (BufferContainsPPDirectives(startBuf, cursor)) {
2338 SourceLocation L = RCDecl ? CDecl->getSuperClassLoc() :
2339 CDecl->getClassLoc();
2340 const char *endHeader = SM->getCharacterData(L);
2341 endHeader += Lexer::MeasureTokenLength(L, *SM);
2342
2343 if (CDecl->getNumIntfRefProtocols()) {
2344 // advance to the end of the referenced protocols.
2345 while (endHeader < cursor && *endHeader != '>') endHeader++;
2346 endHeader++;
2347 }
2348 // rewrite the original header
2349 ReplaceText(LocStart, endHeader-startBuf, Result.c_str(), Result.size());
2350 } else {
2351 // rewrite the original header *without* disturbing the '{'
2352 ReplaceText(LocStart, cursor-startBuf-1, Result.c_str(), Result.size());
2353 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002354 if (RCDecl && ObjCSynthesizedStructs.count(RCDecl)) {
Steve Narofffea763e82007-11-14 19:25:57 +00002355 Result = "\n struct ";
2356 Result += RCDecl->getName();
Steve Naroff39bbd9f2008-03-12 21:09:20 +00002357 Result += "_IMPL ";
2358 Result += RCDecl->getName();
Steve Naroff819173c2008-03-12 21:22:52 +00002359 Result += "_IVARS;\n";
Steve Narofffea763e82007-11-14 19:25:57 +00002360
2361 // insert the super class structure definition.
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00002362 SourceLocation OnePastCurly =
2363 LocStart.getFileLocWithOffset(cursor-startBuf+1);
2364 InsertText(OnePastCurly, Result.c_str(), Result.size());
Steve Narofffea763e82007-11-14 19:25:57 +00002365 }
2366 cursor++; // past '{'
2367
2368 // Now comment out any visibility specifiers.
2369 while (cursor < endBuf) {
2370 if (*cursor == '@') {
2371 SourceLocation atLoc = LocStart.getFileLocWithOffset(cursor-startBuf);
Chris Lattnerdf6a51b2007-11-14 22:57:51 +00002372 // Skip whitespace.
2373 for (++cursor; cursor[0] == ' ' || cursor[0] == '\t'; ++cursor)
2374 /*scan*/;
2375
Fariborz Jahanianfdc08a02007-10-31 17:29:28 +00002376 // FIXME: presence of @public, etc. inside comment results in
2377 // this transformation as well, which is still correct c-code.
Steve Narofffea763e82007-11-14 19:25:57 +00002378 if (!strncmp(cursor, "public", strlen("public")) ||
2379 !strncmp(cursor, "private", strlen("private")) ||
Steve Naroffc5e32772008-04-04 22:34:24 +00002380 !strncmp(cursor, "package", strlen("package")) ||
Fariborz Jahanian95673922007-11-14 22:26:25 +00002381 !strncmp(cursor, "protected", strlen("protected")))
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00002382 InsertText(atLoc, "// ", 3);
Fariborz Jahanianfdc08a02007-10-31 17:29:28 +00002383 }
Fariborz Jahanian95673922007-11-14 22:26:25 +00002384 // FIXME: If there are cases where '<' is used in ivar declaration part
2385 // of user code, then scan the ivar list and use needToScanForQualifiers
2386 // for type checking.
2387 else if (*cursor == '<') {
2388 SourceLocation atLoc = LocStart.getFileLocWithOffset(cursor-startBuf);
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00002389 InsertText(atLoc, "/* ", 3);
Fariborz Jahanian95673922007-11-14 22:26:25 +00002390 cursor = strchr(cursor, '>');
2391 cursor++;
2392 atLoc = LocStart.getFileLocWithOffset(cursor-startBuf);
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00002393 InsertText(atLoc, " */", 3);
Fariborz Jahanian95673922007-11-14 22:26:25 +00002394 }
Steve Narofffea763e82007-11-14 19:25:57 +00002395 cursor++;
Fariborz Jahanianfdc08a02007-10-31 17:29:28 +00002396 }
Steve Narofffea763e82007-11-14 19:25:57 +00002397 // Don't forget to add a ';'!!
Chris Lattnerf3dd57e2008-01-31 19:42:41 +00002398 InsertText(LocEnd.getFileLocWithOffset(1), ";", 1);
Steve Narofffea763e82007-11-14 19:25:57 +00002399 } else { // we don't have any instance variables - insert super struct.
2400 endBuf += Lexer::MeasureTokenLength(LocEnd, *SM);
2401 Result += " {\n struct ";
2402 Result += RCDecl->getName();
Steve Naroff39bbd9f2008-03-12 21:09:20 +00002403 Result += "_IMPL ";
2404 Result += RCDecl->getName();
Steve Naroff819173c2008-03-12 21:22:52 +00002405 Result += "_IVARS;\n};\n";
Chris Lattneraadaf782008-01-31 19:51:04 +00002406 ReplaceText(LocStart, endBuf-startBuf, Result.c_str(), Result.size());
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002407 }
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002408 // Mark this struct as having been generated.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002409 if (!ObjCSynthesizedStructs.insert(CDecl))
Steve Narofffbfe8252008-05-06 18:26:51 +00002410 assert(false && "struct already synthesize- SynthesizeObjCInternalStruct");
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002411}
2412
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002413// RewriteObjCMethodsMetaData - Rewrite methods metadata for instance or
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002414/// class methods.
Steve Naroffb29b4272008-04-14 22:03:09 +00002415void RewriteObjC::RewriteObjCMethodsMetaData(instmeth_iterator MethodBegin,
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002416 instmeth_iterator MethodEnd,
Fariborz Jahanian8e991ba2007-10-25 00:14:44 +00002417 bool IsInstanceMethod,
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002418 const char *prefix,
Chris Lattner158ecb92007-10-25 17:07:24 +00002419 const char *ClassName,
2420 std::string &Result) {
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002421 if (MethodBegin == MethodEnd) return;
2422
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002423 static bool objc_impl_method = false;
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002424 if (!objc_impl_method) {
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002425 /* struct _objc_method {
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002426 SEL _cmd;
2427 char *method_types;
2428 void *_imp;
2429 }
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002430 */
Chris Lattner158ecb92007-10-25 17:07:24 +00002431 Result += "\nstruct _objc_method {\n";
2432 Result += "\tSEL _cmd;\n";
2433 Result += "\tchar *method_types;\n";
2434 Result += "\tvoid *_imp;\n";
2435 Result += "};\n";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002436
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002437 objc_impl_method = true;
Fariborz Jahanian776d6ff2007-10-19 00:36:46 +00002438 }
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002439
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002440 // Build _objc_method_list for class's methods if needed
Steve Naroff946a6932008-03-11 00:12:29 +00002441
2442 /* struct {
2443 struct _objc_method_list *next_method;
2444 int method_count;
2445 struct _objc_method method_list[];
2446 }
2447 */
2448 Result += "\nstatic struct {\n";
2449 Result += "\tstruct _objc_method_list *next_method;\n";
2450 Result += "\tint method_count;\n";
2451 Result += "\tstruct _objc_method method_list[";
2452 Result += utostr(MethodEnd-MethodBegin);
2453 Result += "];\n} _OBJC_";
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002454 Result += prefix;
2455 Result += IsInstanceMethod ? "INSTANCE" : "CLASS";
2456 Result += "_METHODS_";
2457 Result += ClassName;
Steve Naroffdbb65432008-03-12 17:18:30 +00002458 Result += " __attribute__ ((used, section (\"__OBJC, __";
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002459 Result += IsInstanceMethod ? "inst" : "cls";
2460 Result += "_meth\")))= ";
2461 Result += "{\n\t0, " + utostr(MethodEnd-MethodBegin) + "\n";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002462
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002463 Result += "\t,{{(SEL)\"";
2464 Result += (*MethodBegin)->getSelector().getName().c_str();
2465 std::string MethodTypeString;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002466 Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString);
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002467 Result += "\", \"";
2468 Result += MethodTypeString;
Steve Naroff946a6932008-03-11 00:12:29 +00002469 Result += "\", (void *)";
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002470 Result += MethodInternalNames[*MethodBegin];
2471 Result += "}\n";
2472 for (++MethodBegin; MethodBegin != MethodEnd; ++MethodBegin) {
2473 Result += "\t ,{(SEL)\"";
2474 Result += (*MethodBegin)->getSelector().getName().c_str();
Fariborz Jahanian33e1d642007-10-29 22:57:28 +00002475 std::string MethodTypeString;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002476 Context->getObjCEncodingForMethodDecl(*MethodBegin, MethodTypeString);
Fariborz Jahanian33e1d642007-10-29 22:57:28 +00002477 Result += "\", \"";
2478 Result += MethodTypeString;
Steve Naroff946a6932008-03-11 00:12:29 +00002479 Result += "\", (void *)";
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002480 Result += MethodInternalNames[*MethodBegin];
Fariborz Jahanianb7908b52007-11-13 21:02:00 +00002481 Result += "}\n";
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002482 }
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002483 Result += "\t }\n};\n";
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002484}
2485
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002486/// RewriteObjCProtocolsMetaData - Rewrite protocols meta-data.
Steve Naroffb29b4272008-04-14 22:03:09 +00002487void RewriteObjC::RewriteObjCProtocolsMetaData(ObjCProtocolDecl **Protocols,
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002488 int NumProtocols,
2489 const char *prefix,
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002490 const char *ClassName,
2491 std::string &Result) {
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002492 static bool objc_protocol_methods = false;
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002493 if (NumProtocols > 0) {
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002494 for (int i = 0; i < NumProtocols; i++) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002495 ObjCProtocolDecl *PDecl = Protocols[i];
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002496 // Output struct protocol_methods holder of method selector and type.
Chris Lattnerc8581052008-03-16 20:19:15 +00002497 if (!objc_protocol_methods && !PDecl->isForwardDecl()) {
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002498 /* struct protocol_methods {
2499 SEL _cmd;
2500 char *method_types;
2501 }
2502 */
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002503 Result += "\nstruct protocol_methods {\n";
2504 Result += "\tSEL _cmd;\n";
2505 Result += "\tchar *method_types;\n";
2506 Result += "};\n";
2507
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002508 objc_protocol_methods = true;
2509 }
Steve Narofffbfe8252008-05-06 18:26:51 +00002510 // Do not synthesize the protocol more than once.
2511 if (ObjCSynthesizedProtocols.count(PDecl))
2512 continue;
2513
Chris Lattnerc8581052008-03-16 20:19:15 +00002514 if (PDecl->instmeth_begin() != PDecl->instmeth_end()) {
2515 unsigned NumMethods = PDecl->getNumInstanceMethods();
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002516 /* struct _objc_protocol_method_list {
2517 int protocol_method_count;
2518 struct protocol_methods protocols[];
2519 }
2520 */
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002521 Result += "\nstatic struct {\n";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002522 Result += "\tint protocol_method_count;\n";
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002523 Result += "\tstruct protocol_methods protocols[";
2524 Result += utostr(NumMethods);
2525 Result += "];\n} _OBJC_PROTOCOL_INSTANCE_METHODS_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002526 Result += PDecl->getName();
Steve Naroffdbb65432008-03-12 17:18:30 +00002527 Result += " __attribute__ ((used, section (\"__OBJC, __cat_inst_meth\")))= "
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002528 "{\n\t" + utostr(NumMethods) + "\n";
2529
Fariborz Jahanianfc591ac2007-12-17 17:56:10 +00002530 // Output instance methods declared in this protocol.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002531 for (ObjCProtocolDecl::instmeth_iterator I = PDecl->instmeth_begin(),
Fariborz Jahanianfc591ac2007-12-17 17:56:10 +00002532 E = PDecl->instmeth_end(); I != E; ++I) {
2533 if (I == PDecl->instmeth_begin())
2534 Result += "\t ,{{(SEL)\"";
2535 else
2536 Result += "\t ,{(SEL)\"";
2537 Result += (*I)->getSelector().getName().c_str();
2538 std::string MethodTypeString;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002539 Context->getObjCEncodingForMethodDecl((*I), MethodTypeString);
Fariborz Jahanianfc591ac2007-12-17 17:56:10 +00002540 Result += "\", \"";
2541 Result += MethodTypeString;
2542 Result += "\"}\n";
2543 }
2544 Result += "\t }\n};\n";
2545 }
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002546
2547 // Output class methods declared in this protocol.
Chris Lattnerc8581052008-03-16 20:19:15 +00002548 int NumMethods = PDecl->getNumClassMethods();
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002549 if (NumMethods > 0) {
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002550 /* struct _objc_protocol_method_list {
2551 int protocol_method_count;
2552 struct protocol_methods protocols[];
2553 }
2554 */
2555 Result += "\nstatic struct {\n";
2556 Result += "\tint protocol_method_count;\n";
2557 Result += "\tstruct protocol_methods protocols[";
2558 Result += utostr(NumMethods);
2559 Result += "];\n} _OBJC_PROTOCOL_CLASS_METHODS_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002560 Result += PDecl->getName();
Steve Naroffdbb65432008-03-12 17:18:30 +00002561 Result += " __attribute__ ((used, section (\"__OBJC, __cat_cls_meth\")))= "
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002562 "{\n\t";
2563 Result += utostr(NumMethods);
2564 Result += "\n";
2565
Fariborz Jahanianc6e2c2a2007-12-17 18:07:01 +00002566 // Output instance methods declared in this protocol.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002567 for (ObjCProtocolDecl::classmeth_iterator I = PDecl->classmeth_begin(),
Fariborz Jahanianc6e2c2a2007-12-17 18:07:01 +00002568 E = PDecl->classmeth_end(); I != E; ++I) {
2569 if (I == PDecl->classmeth_begin())
2570 Result += "\t ,{{(SEL)\"";
2571 else
2572 Result += "\t ,{(SEL)\"";
Steve Naroff58dbdeb2007-12-14 23:37:57 +00002573 Result += (*I)->getSelector().getName().c_str();
Fariborz Jahanian33e1d642007-10-29 22:57:28 +00002574 std::string MethodTypeString;
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002575 Context->getObjCEncodingForMethodDecl((*I), MethodTypeString);
Fariborz Jahanian33e1d642007-10-29 22:57:28 +00002576 Result += "\", \"";
2577 Result += MethodTypeString;
2578 Result += "\"}\n";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002579 }
2580 Result += "\t }\n};\n";
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002581 }
Steve Narofffbfe8252008-05-06 18:26:51 +00002582
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002583 // Output:
2584 /* struct _objc_protocol {
2585 // Objective-C 1.0 extensions
2586 struct _objc_protocol_extension *isa;
2587 char *protocol_name;
2588 struct _objc_protocol **protocol_list;
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002589 struct _objc_protocol_method_list *instance_methods;
2590 struct _objc_protocol_method_list *class_methods;
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002591 };
2592 */
2593 static bool objc_protocol = false;
2594 if (!objc_protocol) {
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002595 Result += "\nstruct _objc_protocol {\n";
2596 Result += "\tstruct _objc_protocol_extension *isa;\n";
2597 Result += "\tchar *protocol_name;\n";
2598 Result += "\tstruct _objc_protocol **protocol_list;\n";
2599 Result += "\tstruct _objc_protocol_method_list *instance_methods;\n";
2600 Result += "\tstruct _objc_protocol_method_list *class_methods;\n";
2601 Result += "};\n";
2602
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002603 objc_protocol = true;
2604 }
2605
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002606 Result += "\nstatic struct _objc_protocol _OBJC_PROTOCOL_";
2607 Result += PDecl->getName();
Steve Naroffdbb65432008-03-12 17:18:30 +00002608 Result += " __attribute__ ((used, section (\"__OBJC, __protocol\")))= "
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002609 "{\n\t0, \"";
2610 Result += PDecl->getName();
2611 Result += "\", 0, ";
Chris Lattnerc8581052008-03-16 20:19:15 +00002612 if (PDecl->instmeth_begin() != PDecl->instmeth_end()) {
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002613 Result += "(struct _objc_protocol_method_list *)&_OBJC_PROTOCOL_INSTANCE_METHODS_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002614 Result += PDecl->getName();
2615 Result += ", ";
2616 }
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002617 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002618 Result += "0, ";
Steve Naroff58dbdeb2007-12-14 23:37:57 +00002619 if (PDecl->getNumClassMethods() > 0) {
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002620 Result += "(struct _objc_protocol_method_list *)&_OBJC_PROTOCOL_CLASS_METHODS_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002621 Result += PDecl->getName();
2622 Result += "\n";
2623 }
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002624 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002625 Result += "0\n";
2626 Result += "};\n";
Steve Narofffbfe8252008-05-06 18:26:51 +00002627
2628 // Mark this protocol as having been generated.
2629 if (!ObjCSynthesizedProtocols.insert(PDecl))
2630 assert(false && "protocol already synthesized");
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002631 }
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002632 // Output the top lovel protocol meta-data for the class.
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002633 /* struct _objc_protocol_list {
2634 struct _objc_protocol_list *next;
2635 int protocol_count;
2636 struct _objc_protocol *class_protocols[];
2637 }
2638 */
2639 Result += "\nstatic struct {\n";
2640 Result += "\tstruct _objc_protocol_list *next;\n";
2641 Result += "\tint protocol_count;\n";
2642 Result += "\tstruct _objc_protocol *class_protocols[";
2643 Result += utostr(NumProtocols);
2644 Result += "];\n} _OBJC_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002645 Result += prefix;
2646 Result += "_PROTOCOLS_";
2647 Result += ClassName;
Steve Naroffdbb65432008-03-12 17:18:30 +00002648 Result += " __attribute__ ((used, section (\"__OBJC, __cat_cls_meth\")))= "
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002649 "{\n\t0, ";
2650 Result += utostr(NumProtocols);
2651 Result += "\n";
2652
2653 Result += "\t,{&_OBJC_PROTOCOL_";
2654 Result += Protocols[0]->getName();
2655 Result += " \n";
2656
2657 for (int i = 1; i < NumProtocols; i++) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002658 ObjCProtocolDecl *PDecl = Protocols[i];
Steve Narofffa8ec622008-04-18 22:15:15 +00002659 Result += "\t ,&_OBJC_PROTOCOL_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002660 Result += PDecl->getName();
2661 Result += "\n";
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002662 }
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002663 Result += "\t }\n};\n";
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002664 }
2665}
2666
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002667/// RewriteObjCCategoryImplDecl - Rewrite metadata for each category
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002668/// implementation.
Steve Naroffb29b4272008-04-14 22:03:09 +00002669void RewriteObjC::RewriteObjCCategoryImplDecl(ObjCCategoryImplDecl *IDecl,
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002670 std::string &Result) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002671 ObjCInterfaceDecl *ClassDecl = IDecl->getClassInterface();
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002672 // Find category declaration for this implementation.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002673 ObjCCategoryDecl *CDecl;
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002674 for (CDecl = ClassDecl->getCategoryList(); CDecl;
2675 CDecl = CDecl->getNextClassCategory())
2676 if (CDecl->getIdentifier() == IDecl->getIdentifier())
2677 break;
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002678
Chris Lattnereb44eee2007-12-23 01:40:15 +00002679 std::string FullCategoryName = ClassDecl->getName();
2680 FullCategoryName += '_';
2681 FullCategoryName += IDecl->getName();
2682
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002683 // Build _objc_method_list for class's instance methods if needed
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002684 RewriteObjCMethodsMetaData(IDecl->instmeth_begin(), IDecl->instmeth_end(),
Chris Lattnereb44eee2007-12-23 01:40:15 +00002685 true, "CATEGORY_", FullCategoryName.c_str(),
2686 Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002687
2688 // Build _objc_method_list for class's class methods if needed
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002689 RewriteObjCMethodsMetaData(IDecl->classmeth_begin(), IDecl->classmeth_end(),
Chris Lattnereb44eee2007-12-23 01:40:15 +00002690 false, "CATEGORY_", FullCategoryName.c_str(),
2691 Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002692
2693 // Protocols referenced in class declaration?
Fariborz Jahanianbac97d42007-11-13 22:09:49 +00002694 // Null CDecl is case of a category implementation with no category interface
2695 if (CDecl)
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002696 RewriteObjCProtocolsMetaData(CDecl->getReferencedProtocols(),
Fariborz Jahanianbac97d42007-11-13 22:09:49 +00002697 CDecl->getNumReferencedProtocols(),
2698 "CATEGORY",
Chris Lattnereb44eee2007-12-23 01:40:15 +00002699 FullCategoryName.c_str(), Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002700
2701 /* struct _objc_category {
2702 char *category_name;
2703 char *class_name;
2704 struct _objc_method_list *instance_methods;
2705 struct _objc_method_list *class_methods;
2706 struct _objc_protocol_list *protocols;
2707 // Objective-C 1.0 extensions
2708 uint32_t size; // sizeof (struct _objc_category)
2709 struct _objc_property_list *instance_properties; // category's own
2710 // @property decl.
2711 };
2712 */
2713
2714 static bool objc_category = false;
2715 if (!objc_category) {
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002716 Result += "\nstruct _objc_category {\n";
2717 Result += "\tchar *category_name;\n";
2718 Result += "\tchar *class_name;\n";
2719 Result += "\tstruct _objc_method_list *instance_methods;\n";
2720 Result += "\tstruct _objc_method_list *class_methods;\n";
2721 Result += "\tstruct _objc_protocol_list *protocols;\n";
2722 Result += "\tunsigned int size;\n";
2723 Result += "\tstruct _objc_property_list *instance_properties;\n";
2724 Result += "};\n";
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002725 objc_category = true;
Fariborz Jahaniane887c092007-10-22 21:41:37 +00002726 }
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002727 Result += "\nstatic struct _objc_category _OBJC_CATEGORY_";
2728 Result += FullCategoryName;
Steve Naroffdbb65432008-03-12 17:18:30 +00002729 Result += " __attribute__ ((used, section (\"__OBJC, __category\")))= {\n\t\"";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002730 Result += IDecl->getName();
2731 Result += "\"\n\t, \"";
2732 Result += ClassDecl->getName();
2733 Result += "\"\n";
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002734
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002735 if (IDecl->getNumInstanceMethods() > 0) {
2736 Result += "\t, (struct _objc_method_list *)"
2737 "&_OBJC_CATEGORY_INSTANCE_METHODS_";
2738 Result += FullCategoryName;
2739 Result += "\n";
2740 }
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002741 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002742 Result += "\t, 0\n";
2743 if (IDecl->getNumClassMethods() > 0) {
2744 Result += "\t, (struct _objc_method_list *)"
2745 "&_OBJC_CATEGORY_CLASS_METHODS_";
2746 Result += FullCategoryName;
2747 Result += "\n";
2748 }
2749 else
2750 Result += "\t, 0\n";
2751
Fariborz Jahanianbac97d42007-11-13 22:09:49 +00002752 if (CDecl && CDecl->getNumReferencedProtocols() > 0) {
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002753 Result += "\t, (struct _objc_protocol_list *)&_OBJC_CATEGORY_PROTOCOLS_";
2754 Result += FullCategoryName;
2755 Result += "\n";
2756 }
2757 else
2758 Result += "\t, 0\n";
2759 Result += "\t, sizeof(struct _objc_category), 0\n};\n";
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002760}
2761
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002762/// SynthesizeIvarOffsetComputation - This rutine synthesizes computation of
2763/// ivar offset.
Steve Naroffb29b4272008-04-14 22:03:09 +00002764void RewriteObjC::SynthesizeIvarOffsetComputation(ObjCImplementationDecl *IDecl,
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002765 ObjCIvarDecl *ivar,
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002766 std::string &Result) {
Steve Naroff5df5b762008-05-07 21:23:49 +00002767 Result += "__OFFSETOFIVAR__(struct ";
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002768 Result += IDecl->getName();
Steve Naroff61ed9ca2008-03-10 23:16:54 +00002769 if (LangOpts.Microsoft)
2770 Result += "_IMPL";
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002771 Result += ", ";
2772 Result += ivar->getName();
2773 Result += ")";
2774}
2775
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002776//===----------------------------------------------------------------------===//
2777// Meta Data Emission
2778//===----------------------------------------------------------------------===//
2779
Steve Naroffb29b4272008-04-14 22:03:09 +00002780void RewriteObjC::RewriteObjCClassMetaData(ObjCImplementationDecl *IDecl,
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002781 std::string &Result) {
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002782 ObjCInterfaceDecl *CDecl = IDecl->getClassInterface();
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002783
Fariborz Jahanianebe668f2007-11-26 20:59:57 +00002784 // Explictly declared @interface's are already synthesized.
2785 if (CDecl->ImplicitInterfaceDecl()) {
2786 // FIXME: Implementation of a class with no @interface (legacy) doese not
2787 // produce correct synthesis as yet.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002788 SynthesizeObjCInternalStruct(CDecl, Result);
Fariborz Jahanianebe668f2007-11-26 20:59:57 +00002789 }
Fariborz Jahanian4d733d32007-10-26 23:09:28 +00002790
Chris Lattnerbe6df082007-12-12 07:56:42 +00002791 // Build _objc_ivar_list metadata for classes ivars if needed
Chris Lattnerf3a7af92008-03-16 21:08:55 +00002792 unsigned NumIvars = !IDecl->ivar_empty()
2793 ? IDecl->ivar_size()
2794 : (CDecl ? CDecl->ivar_size() : 0);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002795 if (NumIvars > 0) {
2796 static bool objc_ivar = false;
2797 if (!objc_ivar) {
2798 /* struct _objc_ivar {
2799 char *ivar_name;
2800 char *ivar_type;
2801 int ivar_offset;
2802 };
2803 */
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002804 Result += "\nstruct _objc_ivar {\n";
2805 Result += "\tchar *ivar_name;\n";
2806 Result += "\tchar *ivar_type;\n";
2807 Result += "\tint ivar_offset;\n";
2808 Result += "};\n";
2809
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002810 objc_ivar = true;
2811 }
2812
Steve Naroff946a6932008-03-11 00:12:29 +00002813 /* struct {
2814 int ivar_count;
2815 struct _objc_ivar ivar_list[nIvars];
2816 };
2817 */
2818 Result += "\nstatic struct {\n";
2819 Result += "\tint ivar_count;\n";
2820 Result += "\tstruct _objc_ivar ivar_list[";
2821 Result += utostr(NumIvars);
2822 Result += "];\n} _OBJC_INSTANCE_VARIABLES_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002823 Result += IDecl->getName();
Steve Naroffdbb65432008-03-12 17:18:30 +00002824 Result += " __attribute__ ((used, section (\"__OBJC, __instance_vars\")))= "
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002825 "{\n\t";
2826 Result += utostr(NumIvars);
2827 Result += "\n";
Chris Lattnerbe6df082007-12-12 07:56:42 +00002828
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002829 ObjCInterfaceDecl::ivar_iterator IVI, IVE;
Chris Lattnerf3a7af92008-03-16 21:08:55 +00002830 if (!IDecl->ivar_empty()) {
Chris Lattnerbe6df082007-12-12 07:56:42 +00002831 IVI = IDecl->ivar_begin();
2832 IVE = IDecl->ivar_end();
2833 } else {
2834 IVI = CDecl->ivar_begin();
2835 IVE = CDecl->ivar_end();
2836 }
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002837 Result += "\t,{{\"";
Chris Lattnerbe6df082007-12-12 07:56:42 +00002838 Result += (*IVI)->getName();
Fariborz Jahanian160eb652007-10-29 17:16:25 +00002839 Result += "\", \"";
2840 std::string StrEncoding;
Fariborz Jahanian7d6b46d2008-01-22 22:44:46 +00002841 Context->getObjCEncodingForType((*IVI)->getType(), StrEncoding,
2842 EncodingRecordTypes);
Fariborz Jahanian160eb652007-10-29 17:16:25 +00002843 Result += StrEncoding;
2844 Result += "\", ";
Chris Lattnerbe6df082007-12-12 07:56:42 +00002845 SynthesizeIvarOffsetComputation(IDecl, *IVI, Result);
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002846 Result += "}\n";
Chris Lattnerbe6df082007-12-12 07:56:42 +00002847 for (++IVI; IVI != IVE; ++IVI) {
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002848 Result += "\t ,{\"";
Chris Lattnerbe6df082007-12-12 07:56:42 +00002849 Result += (*IVI)->getName();
Fariborz Jahanian160eb652007-10-29 17:16:25 +00002850 Result += "\", \"";
2851 std::string StrEncoding;
Fariborz Jahanian7d6b46d2008-01-22 22:44:46 +00002852 Context->getObjCEncodingForType((*IVI)->getType(), StrEncoding,
2853 EncodingRecordTypes);
Fariborz Jahanian160eb652007-10-29 17:16:25 +00002854 Result += StrEncoding;
2855 Result += "\", ";
Chris Lattnerbe6df082007-12-12 07:56:42 +00002856 SynthesizeIvarOffsetComputation(IDecl, (*IVI), Result);
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00002857 Result += "}\n";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002858 }
2859
2860 Result += "\t }\n};\n";
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002861 }
2862
2863 // Build _objc_method_list for class's instance methods if needed
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002864 RewriteObjCMethodsMetaData(IDecl->instmeth_begin(), IDecl->instmeth_end(),
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002865 true, "", IDecl->getName(), Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002866
2867 // Build _objc_method_list for class's class methods if needed
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002868 RewriteObjCMethodsMetaData(IDecl->classmeth_begin(), IDecl->classmeth_end(),
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002869 false, "", IDecl->getName(), Result);
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002870
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002871 // Protocols referenced in class declaration?
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002872 RewriteObjCProtocolsMetaData(CDecl->getReferencedProtocols(),
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002873 CDecl->getNumIntfRefProtocols(),
Chris Lattnerab4c4d52007-12-12 07:46:12 +00002874 "CLASS", CDecl->getName(), Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00002875
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00002876
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00002877 // Declaration of class/meta-class metadata
2878 /* struct _objc_class {
2879 struct _objc_class *isa; // or const char *root_class_name when metadata
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00002880 const char *super_class_name;
2881 char *name;
2882 long version;
2883 long info;
2884 long instance_size;
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00002885 struct _objc_ivar_list *ivars;
2886 struct _objc_method_list *methods;
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00002887 struct objc_cache *cache;
2888 struct objc_protocol_list *protocols;
2889 const char *ivar_layout;
2890 struct _objc_class_ext *ext;
2891 };
2892 */
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00002893 static bool objc_class = false;
2894 if (!objc_class) {
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002895 Result += "\nstruct _objc_class {\n";
2896 Result += "\tstruct _objc_class *isa;\n";
2897 Result += "\tconst char *super_class_name;\n";
2898 Result += "\tchar *name;\n";
2899 Result += "\tlong version;\n";
2900 Result += "\tlong info;\n";
2901 Result += "\tlong instance_size;\n";
2902 Result += "\tstruct _objc_ivar_list *ivars;\n";
2903 Result += "\tstruct _objc_method_list *methods;\n";
2904 Result += "\tstruct objc_cache *cache;\n";
2905 Result += "\tstruct _objc_protocol_list *protocols;\n";
2906 Result += "\tconst char *ivar_layout;\n";
2907 Result += "\tstruct _objc_class_ext *ext;\n";
2908 Result += "};\n";
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00002909 objc_class = true;
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00002910 }
2911
2912 // Meta-class metadata generation.
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002913 ObjCInterfaceDecl *RootClass = 0;
2914 ObjCInterfaceDecl *SuperClass = CDecl->getSuperClass();
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00002915 while (SuperClass) {
2916 RootClass = SuperClass;
2917 SuperClass = SuperClass->getSuperClass();
2918 }
2919 SuperClass = CDecl->getSuperClass();
2920
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002921 Result += "\nstatic struct _objc_class _OBJC_METACLASS_";
2922 Result += CDecl->getName();
Steve Naroffdbb65432008-03-12 17:18:30 +00002923 Result += " __attribute__ ((used, section (\"__OBJC, __meta_class\")))= "
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002924 "{\n\t(struct _objc_class *)\"";
2925 Result += (RootClass ? RootClass->getName() : CDecl->getName());
2926 Result += "\"";
2927
2928 if (SuperClass) {
2929 Result += ", \"";
2930 Result += SuperClass->getName();
2931 Result += "\", \"";
2932 Result += CDecl->getName();
2933 Result += "\"";
2934 }
2935 else {
2936 Result += ", 0, \"";
2937 Result += CDecl->getName();
2938 Result += "\"";
2939 }
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002940 // Set 'ivars' field for root class to 0. ObjC1 runtime does not use it.
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00002941 // 'info' field is initialized to CLS_META(2) for metaclass
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002942 Result += ", 0,2, sizeof(struct _objc_class), 0";
Steve Naroffb26d7132007-12-05 21:49:40 +00002943 if (IDecl->getNumClassMethods() > 0) {
Steve Naroff23f41272008-03-11 18:14:26 +00002944 Result += "\n\t, (struct _objc_method_list *)&_OBJC_CLASS_METHODS_";
Steve Naroffb26d7132007-12-05 21:49:40 +00002945 Result += IDecl->getName();
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002946 Result += "\n";
2947 }
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00002948 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002949 Result += ", 0\n";
2950 if (CDecl->getNumIntfRefProtocols() > 0) {
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00002951 Result += "\t,0, (struct _objc_protocol_list *)&_OBJC_CLASS_PROTOCOLS_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002952 Result += CDecl->getName();
2953 Result += ",0,0\n";
2954 }
Fariborz Jahanian454cb012007-10-24 20:54:23 +00002955 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002956 Result += "\t,0,0,0,0\n";
2957 Result += "};\n";
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00002958
2959 // class metadata generation.
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002960 Result += "\nstatic struct _objc_class _OBJC_CLASS_";
2961 Result += CDecl->getName();
Steve Naroffdbb65432008-03-12 17:18:30 +00002962 Result += " __attribute__ ((used, section (\"__OBJC, __class\")))= "
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002963 "{\n\t&_OBJC_METACLASS_";
2964 Result += CDecl->getName();
2965 if (SuperClass) {
2966 Result += ", \"";
2967 Result += SuperClass->getName();
2968 Result += "\", \"";
2969 Result += CDecl->getName();
2970 Result += "\"";
2971 }
2972 else {
2973 Result += ", 0, \"";
2974 Result += CDecl->getName();
2975 Result += "\"";
2976 }
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00002977 // 'info' field is initialized to CLS_CLASS(1) for class
Fariborz Jahanian4d733d32007-10-26 23:09:28 +00002978 Result += ", 0,1";
Ted Kremeneka526c5c2008-01-07 19:49:32 +00002979 if (!ObjCSynthesizedStructs.count(CDecl))
Fariborz Jahanian4d733d32007-10-26 23:09:28 +00002980 Result += ",0";
2981 else {
2982 // class has size. Must synthesize its size.
Fariborz Jahanian909f02a2007-11-05 17:47:33 +00002983 Result += ",sizeof(struct ";
Fariborz Jahanian4d733d32007-10-26 23:09:28 +00002984 Result += CDecl->getName();
Steve Naroffba9ac4e2008-03-10 23:33:22 +00002985 if (LangOpts.Microsoft)
2986 Result += "_IMPL";
Fariborz Jahanian4d733d32007-10-26 23:09:28 +00002987 Result += ")";
2988 }
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002989 if (NumIvars > 0) {
Steve Naroffc0a123c2008-03-11 17:37:02 +00002990 Result += ", (struct _objc_ivar_list *)&_OBJC_INSTANCE_VARIABLES_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002991 Result += CDecl->getName();
2992 Result += "\n\t";
2993 }
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00002994 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002995 Result += ",0";
2996 if (IDecl->getNumInstanceMethods() > 0) {
Steve Naroff946a6932008-03-11 00:12:29 +00002997 Result += ", (struct _objc_method_list *)&_OBJC_INSTANCE_METHODS_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00002998 Result += CDecl->getName();
2999 Result += ", 0\n\t";
3000 }
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00003001 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003002 Result += ",0,0";
3003 if (CDecl->getNumIntfRefProtocols() > 0) {
Steve Naroff8eb4a5e2008-03-12 01:06:30 +00003004 Result += ", (struct _objc_protocol_list*)&_OBJC_CLASS_PROTOCOLS_";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003005 Result += CDecl->getName();
3006 Result += ", 0,0\n";
3007 }
Fariborz Jahaniandeef5182007-10-23 18:53:48 +00003008 else
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003009 Result += ",0,0,0\n";
3010 Result += "};\n";
Fariborz Jahanian9f0a1cb2007-10-23 00:02:02 +00003011}
Fariborz Jahanianf4d331d2007-10-18 22:09:03 +00003012
Fariborz Jahanian7a3279d2007-11-13 19:21:13 +00003013/// RewriteImplementations - This routine rewrites all method implementations
3014/// and emits meta-data.
3015
Steve Naroffb29b4272008-04-14 22:03:09 +00003016void RewriteObjC::RewriteImplementations(std::string &Result) {
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003017 int ClsDefCount = ClassImplementation.size();
3018 int CatDefCount = CategoryImplementation.size();
Fariborz Jahanian7a3279d2007-11-13 19:21:13 +00003019
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003020 if (ClsDefCount == 0 && CatDefCount == 0)
3021 return;
Fariborz Jahanian7a3279d2007-11-13 19:21:13 +00003022 // Rewrite implemented methods
3023 for (int i = 0; i < ClsDefCount; i++)
3024 RewriteImplementationDecl(ClassImplementation[i]);
Fariborz Jahanianf4d331d2007-10-18 22:09:03 +00003025
Fariborz Jahanian66d6b292007-11-13 20:04:28 +00003026 for (int i = 0; i < CatDefCount; i++)
3027 RewriteImplementationDecl(CategoryImplementation[i]);
3028
Steve Naroff5df5b762008-05-07 21:23:49 +00003029 // This is needed for determining instance variable offsets.
3030 Result += "#define __OFFSETOFIVAR__(TYPE, MEMBER) ((int) &((TYPE *)0)->MEMBER)\n";
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00003031 // For each implemented class, write out all its meta data.
Fariborz Jahanianf4d331d2007-10-18 22:09:03 +00003032 for (int i = 0; i < ClsDefCount; i++)
Ted Kremeneka526c5c2008-01-07 19:49:32 +00003033 RewriteObjCClassMetaData(ClassImplementation[i], Result);
Fariborz Jahanian2e6d9352007-10-24 19:23:36 +00003034
3035 // For each implemented category, write out all its meta data.
3036 for (int i = 0; i < CatDefCount; i++)
Ted Kremeneka526c5c2008-01-07 19:49:32 +00003037 RewriteObjCCategoryImplDecl(CategoryImplementation[i], Result);
Fariborz Jahanianf4d331d2007-10-18 22:09:03 +00003038
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003039 // Write objc_symtab metadata
3040 /*
3041 struct _objc_symtab
3042 {
3043 long sel_ref_cnt;
3044 SEL *refs;
3045 short cls_def_cnt;
3046 short cat_def_cnt;
3047 void *defs[cls_def_cnt + cat_def_cnt];
3048 };
3049 */
3050
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003051 Result += "\nstruct _objc_symtab {\n";
3052 Result += "\tlong sel_ref_cnt;\n";
3053 Result += "\tSEL *refs;\n";
3054 Result += "\tshort cls_def_cnt;\n";
3055 Result += "\tshort cat_def_cnt;\n";
3056 Result += "\tvoid *defs[" + utostr(ClsDefCount + CatDefCount)+ "];\n";
3057 Result += "};\n\n";
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003058
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003059 Result += "static struct _objc_symtab "
Steve Naroffdbb65432008-03-12 17:18:30 +00003060 "_OBJC_SYMBOLS __attribute__((used, section (\"__OBJC, __symbols\")))= {\n";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003061 Result += "\t0, 0, " + utostr(ClsDefCount)
3062 + ", " + utostr(CatDefCount) + "\n";
3063 for (int i = 0; i < ClsDefCount; i++) {
3064 Result += "\t,&_OBJC_CLASS_";
3065 Result += ClassImplementation[i]->getName();
3066 Result += "\n";
3067 }
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003068
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003069 for (int i = 0; i < CatDefCount; i++) {
3070 Result += "\t,&_OBJC_CATEGORY_";
3071 Result += CategoryImplementation[i]->getClassInterface()->getName();
3072 Result += "_";
3073 Result += CategoryImplementation[i]->getName();
3074 Result += "\n";
3075 }
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003076
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003077 Result += "};\n\n";
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003078
3079 // Write objc_module metadata
3080
3081 /*
3082 struct _objc_module {
3083 long version;
3084 long size;
3085 const char *name;
3086 struct _objc_symtab *symtab;
3087 }
3088 */
3089
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003090 Result += "\nstruct _objc_module {\n";
3091 Result += "\tlong version;\n";
3092 Result += "\tlong size;\n";
3093 Result += "\tconst char *name;\n";
3094 Result += "\tstruct _objc_symtab *symtab;\n";
3095 Result += "};\n\n";
3096 Result += "static struct _objc_module "
Steve Naroffdbb65432008-03-12 17:18:30 +00003097 "_OBJC_MODULES __attribute__ ((used, section (\"__OBJC, __module_info\")))= {\n";
Fariborz Jahanian26e4cd32007-10-26 19:46:17 +00003098 Result += "\t" + utostr(OBJC_ABI_VERSION) +
3099 ", sizeof(struct _objc_module), \"\", &_OBJC_SYMBOLS\n";
Fariborz Jahanianccd87b02007-10-25 20:55:25 +00003100 Result += "};\n\n";
Steve Naroff4f943c22008-03-10 20:43:59 +00003101
3102 if (LangOpts.Microsoft) {
3103 Result += "#pragma section(\".objc_module_info$B\",long,read,write)\n";
Steve Naroff19190322008-05-07 00:06:16 +00003104 Result += "#pragma data_seg(push, \".objc_module_info$B\")\n";
Steve Naroff4f943c22008-03-10 20:43:59 +00003105 Result += "static struct _objc_module *_POINTER_OBJC_MODULES = ";
3106 Result += "&_OBJC_MODULES;\n";
3107 Result += "#pragma data_seg(pop)\n\n";
3108 }
Fariborz Jahanian545b9ae2007-10-18 19:23:00 +00003109}
Chris Lattner311ff022007-10-16 22:36:42 +00003110
Argyrios Kyrtzidisef177822008-04-17 14:40:12 +00003111
Argyrios Kyrtzidis39ba4ae2008-06-09 23:19:58 +00003112
3113