1accaf19bc1129c0273ec50dba52318e60bc29103Benjamin Kramer//===--- Transforms.cpp - Transformations to ARC mode ---------------------===// 28f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall// 38f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall// The LLVM Compiler Infrastructure 48f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall// 58f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall// This file is distributed under the University of Illinois Open Source 68f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall// License. See LICENSE.TXT for details. 78f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall// 88f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall//===----------------------------------------------------------------------===// 98f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 107196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis#include "Transforms.h" 118f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall#include "Internals.h" 12471c8b49982d1132f30b0b0da27fef94fd6e4f67Benjamin Kramer#include "clang/AST/ASTContext.h" 138f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall#include "clang/AST/RecursiveASTVisitor.h" 148f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall#include "clang/AST/StmtVisitor.h" 1555fc873017f10f6f566b182b70f6fc22aefa3464Chandler Carruth#include "clang/Analysis/DomainSpecific/CocoaConventions.h" 168f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall#include "clang/Basic/SourceManager.h" 179852f58f50b4fc20914fbce5b4454135a42343f4Benjamin Kramer#include "clang/Basic/TargetInfo.h" 18471c8b49982d1132f30b0b0da27fef94fd6e4f67Benjamin Kramer#include "clang/Lex/Lexer.h" 19471c8b49982d1132f30b0b0da27fef94fd6e4f67Benjamin Kramer#include "clang/Sema/Sema.h" 20471c8b49982d1132f30b0b0da27fef94fd6e4f67Benjamin Kramer#include "clang/Sema/SemaDiagnostic.h" 218f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall#include "llvm/ADT/DenseSet.h" 229852f58f50b4fc20914fbce5b4454135a42343f4Benjamin Kramer#include "llvm/ADT/StringSwitch.h" 238f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall#include <map> 248f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 258f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallusing namespace clang; 268f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallusing namespace arcmt; 277196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidisusing namespace trans; 288f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 29e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios KyrtzidisASTTraverser::~ASTTraverser() { } 30e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 31684190b8dbe5258f4708ffbd816b8c5ee5b3502dArgyrios Kyrtzidisbool MigrationPass::CFBridgingFunctionsDefined() { 32684190b8dbe5258f4708ffbd816b8c5ee5b3502dArgyrios Kyrtzidis if (!EnableCFBridgeFns.hasValue()) 33684190b8dbe5258f4708ffbd816b8c5ee5b3502dArgyrios Kyrtzidis EnableCFBridgeFns = SemaRef.isKnownName("CFBridgingRetain") && 34684190b8dbe5258f4708ffbd816b8c5ee5b3502dArgyrios Kyrtzidis SemaRef.isKnownName("CFBridgingRelease"); 35684190b8dbe5258f4708ffbd816b8c5ee5b3502dArgyrios Kyrtzidis return *EnableCFBridgeFns; 36684190b8dbe5258f4708ffbd816b8c5ee5b3502dArgyrios Kyrtzidis} 37684190b8dbe5258f4708ffbd816b8c5ee5b3502dArgyrios Kyrtzidis 388f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall//===----------------------------------------------------------------------===// 397196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis// Helpers. 408f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall//===----------------------------------------------------------------------===// 418f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 4212192cf50a96cb59a3039af044b7fa97f043101cArgyrios Kyrtzidisbool trans::canApplyWeak(ASTContext &Ctx, QualType type, 4312192cf50a96cb59a3039af044b7fa97f043101cArgyrios Kyrtzidis bool AllowOnUnknownClass) { 440a7dd788dbef975f35f273c7ab913f480f7edd60John McCall if (!Ctx.getLangOpts().ObjCARCWeak) 4586625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis return false; 4686625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis 4786625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis QualType T = type; 48b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (T.isNull()) 49b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return false; 50b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 512af504bf157cce51bdb4a9bab0e6d3b34f898f6eArgyrios Kyrtzidis // iOS is always safe to use 'weak'. 5257b1da1588a3f5785ad8bd5d9f2d795d685e1058Cameron Esfahani if (Ctx.getTargetInfo().getTriple().isiOS()) 532af504bf157cce51bdb4a9bab0e6d3b34f898f6eArgyrios Kyrtzidis AllowOnUnknownClass = true; 542af504bf157cce51bdb4a9bab0e6d3b34f898f6eArgyrios Kyrtzidis 5586625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis while (const PointerType *ptr = T->getAs<PointerType>()) 5686625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis T = ptr->getPointeeType(); 5786625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis if (const ObjCObjectPointerType *ObjT = T->getAs<ObjCObjectPointerType>()) { 5886625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis ObjCInterfaceDecl *Class = ObjT->getInterfaceDecl(); 5912192cf50a96cb59a3039af044b7fa97f043101cArgyrios Kyrtzidis if (!AllowOnUnknownClass && (!Class || Class->getName() == "NSObject")) 6086625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis return false; // id/NSObject is not safe for weak. 617723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor if (!AllowOnUnknownClass && !Class->hasDefinition()) 625363e8df5d274cb32c0c47fd2df45aa02cf68dfeArgyrios Kyrtzidis return false; // forward classes are not verifiable, therefore not safe. 63a8eaf008e92759142982f7b40720b2b2674bd663Richard Smith if (Class && Class->isArcWeakrefUnavailable()) 6486625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis return false; 6586625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis } 6686625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis 6786625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis return true; 6886625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis} 6986625b5650cdddc38c0b4cc1eb7fb460478c9d11Argyrios Kyrtzidis 701b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidisbool trans::isPlusOneAssign(const BinaryOperator *E) { 711b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (E->getOpcode() != BO_Assign) 721b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis return false; 731b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis 7420bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis return isPlusOne(E->getRHS()); 7520bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis} 7620bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis 7720bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidisbool trans::isPlusOne(const Expr *E) { 7820bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis if (!E) 7920bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis return false; 8020bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis if (const ExprWithCleanups *EWC = dyn_cast<ExprWithCleanups>(E)) 8120bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis E = EWC->getSubExpr(); 8220bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis 831b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (const ObjCMessageExpr * 8420bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis ME = dyn_cast<ObjCMessageExpr>(E->IgnoreParenCasts())) 851b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (ME->getMethodFamily() == OMF_retain) 861b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis return true; 871b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis 881b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (const CallExpr * 8920bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis callE = dyn_cast<CallExpr>(E->IgnoreParenCasts())) { 901b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (const FunctionDecl *FD = callE->getDirectCallee()) { 91651f13cea278ec967336033dd032faef0e9fc2ecStephen Hines if (FD->hasAttr<CFReturnsRetainedAttr>()) 921b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis return true; 931b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis 941b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (FD->isGlobal() && 951b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis FD->getIdentifier() && 961b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis FD->getParent()->isTranslationUnit() && 97181e3ecc0907ae0103586a9f4db52241995a8267Rafael Espindola FD->isExternallyVisible() && 981b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis ento::cocoa::isRefType(callE->getType(), "CF", 991b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis FD->getIdentifier()->getName())) { 1001b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis StringRef fname = FD->getIdentifier()->getName(); 1011b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (fname.endswith("Retain") || 1021b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis fname.find("Create") != StringRef::npos || 1031b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis fname.find("Copy") != StringRef::npos) { 1041b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis return true; 1051b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis } 1061b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis } 1071b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis } 1081b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis } 1091b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis 11020bcd4ead46e3764a5d1279845f8ed351f4d3984Argyrios Kyrtzidis const ImplicitCastExpr *implCE = dyn_cast<ImplicitCastExpr>(E); 1111b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis while (implCE && implCE->getCastKind() == CK_BitCast) 1121b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis implCE = dyn_cast<ImplicitCastExpr>(implCE->getSubExpr()); 1131b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis 1141b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis if (implCE && implCE->getCastKind() == CK_ARCConsumeObject) 1151b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis return true; 1161b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis 1171b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis return false; 1181b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis} 1191b8fbd3601e009803565e74d2ec54abecb5cbf73Argyrios Kyrtzidis 1207196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis/// \brief 'Loc' is the end of a statement range. This returns the location 1217196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis/// immediately after the semicolon following the statement. 1227196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis/// If no semicolon is found or the location is inside a macro, the returned 1237196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis/// source location will be invalid. 1247196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios KyrtzidisSourceLocation trans::findLocationAfterSemi(SourceLocation loc, 1251f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian ASTContext &Ctx, bool IsDecl) { 1261f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian SourceLocation SemiLoc = findSemiAfterLocation(loc, Ctx, IsDecl); 127aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis if (SemiLoc.isInvalid()) 128aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis return SourceLocation(); 129a64ccefdf0ea4e03ec88805d71b0af74950c7472Argyrios Kyrtzidis return SemiLoc.getLocWithOffset(1); 130aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis} 131aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis 132aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis/// \brief \arg Loc is the end of a statement range. This returns the location 133aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis/// of the semicolon following the statement. 134aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis/// If no semicolon is found or the location is inside a macro, the returned 135aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis/// source location will be invalid. 136aec230d29835285777ecc467e268c83b33a2addeArgyrios KyrtzidisSourceLocation trans::findSemiAfterLocation(SourceLocation loc, 1371f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian ASTContext &Ctx, 1381f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian bool IsDecl) { 1397196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis SourceManager &SM = Ctx.getSourceManager(); 1407196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis if (loc.isMacroID()) { 1414e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie if (!Lexer::isAtEndOfMacroExpansion(loc, SM, Ctx.getLangOpts(), &loc)) 1427196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis return SourceLocation(); 1437196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis } 1444e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie loc = Lexer::getLocForEndOfToken(loc, /*Offset=*/0, SM, Ctx.getLangOpts()); 1458f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 1467196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis // Break down the source location. 1477196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis std::pair<FileID, unsigned> locInfo = SM.getDecomposedLoc(loc); 1488f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 1497196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis // Try to load the file buffer. 1507196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool invalidTemp = false; 1515f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner StringRef file = SM.getBufferData(locInfo.first, &invalidTemp); 1527196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis if (invalidTemp) 1537196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis return SourceLocation(); 1547196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 1557196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis const char *tokenBegin = file.data() + locInfo.second; 1567196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 1577196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis // Lex from the start of the given location. 1587196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis Lexer lexer(SM.getLocForStartOfFile(locInfo.first), 1594e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie Ctx.getLangOpts(), 1607196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis file.begin(), tokenBegin, file.end()); 1617196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis Token tok; 1627196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis lexer.LexFromRawLexer(tok); 1631f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian if (tok.isNot(tok::semi)) { 1641f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian if (!IsDecl) 1651f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian return SourceLocation(); 1661f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian // Declaration may be followed with other tokens; such as an __attribute, 1671f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian // before ending with a semicolon. 1681f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian return findSemiAfterLocation(tok.getLocation(), Ctx, /*IsDecl*/true); 1691f9a09d4e50c3e8e622292350823eef776b93e3dFariborz Jahanian } 1708f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 171aec230d29835285777ecc467e268c83b33a2addeArgyrios Kyrtzidis return tok.getLocation(); 1727196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis} 1738f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 1747196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidisbool trans::hasSideEffects(Expr *E, ASTContext &Ctx) { 1758f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall if (!E || !E->HasSideEffects(Ctx)) 1768f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return false; 1778f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 1788f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall E = E->IgnoreParenCasts(); 1798f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall ObjCMessageExpr *ME = dyn_cast<ObjCMessageExpr>(E); 1808f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall if (!ME) 1818f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 1828f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall switch (ME->getMethodFamily()) { 1838f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall case OMF_autorelease: 1848f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall case OMF_dealloc: 1858f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall case OMF_release: 1868f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall case OMF_retain: 1878f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall switch (ME->getReceiverKind()) { 1888f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall case ObjCMessageExpr::SuperInstance: 1898f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return false; 1908f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall case ObjCMessageExpr::Instance: 1917196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis return hasSideEffects(ME->getInstanceReceiver(), Ctx); 1928f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall default: 1938f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall break; 1948f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 1958f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall break; 1968f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall default: 1978f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall break; 1988f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 1998f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2008f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2018f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall} 2028f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2032c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidisbool trans::isGlobalVar(Expr *E) { 2042c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis E = E->IgnoreParenCasts(); 2052c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) 20618fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios Kyrtzidis return DRE->getDecl()->getDeclContext()->isFileContext() && 207181e3ecc0907ae0103586a9f4db52241995a8267Rafael Espindola DRE->getDecl()->isExternallyVisible(); 2082c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis if (ConditionalOperator *condOp = dyn_cast<ConditionalOperator>(E)) 2092c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis return isGlobalVar(condOp->getTrueExpr()) && 2102c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis isGlobalVar(condOp->getFalseExpr()); 2112c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis 2122c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis return false; 2132c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis} 2142c18ca0575b60082f2a9f4563b4071071960d37cArgyrios Kyrtzidis 21518fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios KyrtzidisStringRef trans::getNilString(ASTContext &Ctx) { 21618fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios Kyrtzidis if (Ctx.Idents.get("nil").hasMacroDefinition()) 21718fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios Kyrtzidis return "nil"; 21818fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios Kyrtzidis else 21918fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios Kyrtzidis return "0"; 22018fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios Kyrtzidis} 22118fd0c6915b45c4daafe18e3cd324c13306f913fArgyrios Kyrtzidis 2228f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallnamespace { 2238f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2248f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallclass ReferenceClear : public RecursiveASTVisitor<ReferenceClear> { 2257196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis ExprSet &Refs; 2268f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallpublic: 2277196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis ReferenceClear(ExprSet &refs) : Refs(refs) { } 2288f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall bool VisitDeclRefExpr(DeclRefExpr *E) { Refs.erase(E); return true; } 2298f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall}; 2308f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2318f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallclass ReferenceCollector : public RecursiveASTVisitor<ReferenceCollector> { 2328f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall ValueDecl *Dcl; 2337196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis ExprSet &Refs; 2348f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2358f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallpublic: 2367196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis ReferenceCollector(ValueDecl *D, ExprSet &refs) 2377196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis : Dcl(D), Refs(refs) { } 2388f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2398f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall bool VisitDeclRefExpr(DeclRefExpr *E) { 2408f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall if (E->getDecl() == Dcl) 2418f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall Refs.insert(E); 2428f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2438f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2448f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall}; 2458f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2467196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidisclass RemovablesCollector : public RecursiveASTVisitor<RemovablesCollector> { 2477196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis ExprSet &Removables; 2488f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 2498f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallpublic: 2507196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis RemovablesCollector(ExprSet &removables) 2517196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis : Removables(removables) { } 2527196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2537196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool shouldWalkTypesOfTypeLocs() const { return false; } 2547196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2557196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool TraverseStmtExpr(StmtExpr *E) { 2567196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis CompoundStmt *S = E->getSubStmt(); 2577196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis for (CompoundStmt::body_iterator 2587196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis I = S->body_begin(), E = S->body_end(); I != E; ++I) { 2597196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis if (I != E - 1) 2607196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(*I); 2617196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis TraverseStmt(*I); 2628f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2638f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2648f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2657196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2667196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool VisitCompoundStmt(CompoundStmt *S) { 267651f13cea278ec967336033dd032faef0e9fc2ecStephen Hines for (auto *I : S->body()) 268651f13cea278ec967336033dd032faef0e9fc2ecStephen Hines mark(I); 2698f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2708f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2717196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2727196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool VisitIfStmt(IfStmt *S) { 2737196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(S->getThen()); 2747196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(S->getElse()); 2758f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2768f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2777196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2787196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool VisitWhileStmt(WhileStmt *S) { 2797196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(S->getBody()); 2808f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2818f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2827196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2837196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool VisitDoStmt(DoStmt *S) { 2847196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(S->getBody()); 2858f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2868f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2877196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2887196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis bool VisitForStmt(ForStmt *S) { 2897196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(S->getInit()); 2907196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(S->getInc()); 2917196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis mark(S->getBody()); 2928f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return true; 2938f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall } 2947196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2958f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallprivate: 2967196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis void mark(Stmt *S) { 2977196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis if (!S) return; 2987196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis 2997e5e5f4cc36fe50f46ad76dca7a266434c94f475John McCall while (LabelStmt *Label = dyn_cast<LabelStmt>(S)) 3007e5e5f4cc36fe50f46ad76dca7a266434c94f475John McCall S = Label->getSubStmt(); 3017e5e5f4cc36fe50f46ad76dca7a266434c94f475John McCall S = S->IgnoreImplicit(); 3027196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis if (Expr *E = dyn_cast<Expr>(S)) 3037196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis Removables.insert(E); 3048e2ce7fdf9e8033b77788662a9c3f61334eb5dafArgyrios Kyrtzidis } 3057196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis}; 3068e2ce7fdf9e8033b77788662a9c3f61334eb5dafArgyrios Kyrtzidis 3077196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis} // end anonymous namespace 3088f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 3097196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidisvoid trans::clearRefsIn(Stmt *S, ExprSet &refs) { 3107196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis ReferenceClear(refs).TraverseStmt(S); 3117196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis} 3128f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 3137196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidisvoid trans::collectRefs(ValueDecl *D, Stmt *S, ExprSet &refs) { 3147196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis ReferenceCollector(D, refs).TraverseStmt(S); 3158f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall} 3168f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 3177196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidisvoid trans::collectRemovables(Stmt *S, ExprSet &exprs) { 3187196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis RemovablesCollector(exprs).TraverseStmt(S); 3198f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall} 3208f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 3218f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall//===----------------------------------------------------------------------===// 322e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis// MigrationContext 323e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis//===----------------------------------------------------------------------===// 324e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 325e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidisnamespace { 326e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 327e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidisclass ASTTransform : public RecursiveASTVisitor<ASTTransform> { 328e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis MigrationContext &MigrateCtx; 329a33849b368b0973e64ee9c49af4c080a59f99eb4Argyrios Kyrtzidis typedef RecursiveASTVisitor<ASTTransform> base; 330e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 331e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidispublic: 332e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis ASTTransform(MigrationContext &MigrateCtx) : MigrateCtx(MigrateCtx) { } 333e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 334f38fa73e605587abfa32ab6a39887cdebc0a320aArgyrios Kyrtzidis bool shouldWalkTypesOfTypeLocs() const { return false; } 335f38fa73e605587abfa32ab6a39887cdebc0a320aArgyrios Kyrtzidis 336b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis bool TraverseObjCImplementationDecl(ObjCImplementationDecl *D) { 337b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis ObjCImplementationContext ImplCtx(MigrateCtx, D); 338b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis for (MigrationContext::traverser_iterator 339b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis I = MigrateCtx.traversers_begin(), 340b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis E = MigrateCtx.traversers_end(); I != E; ++I) 341b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis (*I)->traverseObjCImplementation(ImplCtx); 342b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis 343a33849b368b0973e64ee9c49af4c080a59f99eb4Argyrios Kyrtzidis return base::TraverseObjCImplementationDecl(D); 344b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis } 345b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis 346e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis bool TraverseStmt(Stmt *rootS) { 347e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis if (!rootS) 348e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis return true; 349e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 350e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis BodyContext BodyCtx(MigrateCtx, rootS); 351e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis for (MigrationContext::traverser_iterator 352e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis I = MigrateCtx.traversers_begin(), 353e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis E = MigrateCtx.traversers_end(); I != E; ++I) 354e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis (*I)->traverseBody(BodyCtx); 355e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 356e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis return true; 357e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis } 358e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis}; 359e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 360e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis} 361e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 362e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios KyrtzidisMigrationContext::~MigrationContext() { 363e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis for (traverser_iterator 364e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis I = traversers_begin(), E = traversers_end(); I != E; ++I) 365e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis delete *I; 366e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis} 367e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 3681fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidisbool MigrationContext::isGCOwnedNonObjC(QualType T) { 3691fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis while (!T.isNull()) { 3701fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis if (const AttributedType *AttrT = T->getAs<AttributedType>()) { 3711fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis if (AttrT->getAttrKind() == AttributedType::attr_objc_ownership) 3721fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis return !AttrT->getModifiedType()->isObjCRetainableType(); 3731fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis } 3741fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis 3751fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis if (T->isArrayType()) 3761fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis T = Pass.Ctx.getBaseElementType(T); 3771fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis else if (const PointerType *PT = T->getAs<PointerType>()) 3781fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis T = PT->getPointeeType(); 3791fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis else if (const ReferenceType *RT = T->getAs<ReferenceType>()) 3801fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis T = RT->getPointeeType(); 3811fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis else 3821fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis break; 3831fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis } 3841fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis 3851fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis return false; 3861fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis} 3871fe4203ca05d0a3283efc8a2e8c01ecdf78fbf2eArgyrios Kyrtzidis 388b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidisbool MigrationContext::rewritePropertyAttribute(StringRef fromAttr, 389b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis StringRef toAttr, 390b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis SourceLocation atLoc) { 391b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (atLoc.isMacroID()) 392b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return false; 393b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 394b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis SourceManager &SM = Pass.Ctx.getSourceManager(); 395b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 396b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis // Break down the source location. 397b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis std::pair<FileID, unsigned> locInfo = SM.getDecomposedLoc(atLoc); 398b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 399b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis // Try to load the file buffer. 400b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis bool invalidTemp = false; 401b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis StringRef file = SM.getBufferData(locInfo.first, &invalidTemp); 402b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (invalidTemp) 403b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return false; 404b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 405b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis const char *tokenBegin = file.data() + locInfo.second; 406b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 407b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis // Lex from the start of the given location. 408b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Lexer lexer(SM.getLocForStartOfFile(locInfo.first), 4094e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie Pass.Ctx.getLangOpts(), 410b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis file.begin(), tokenBegin, file.end()); 411b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Token tok; 412b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis lexer.LexFromRawLexer(tok); 413b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (tok.isNot(tok::at)) return false; 414b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis lexer.LexFromRawLexer(tok); 415b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (tok.isNot(tok::raw_identifier)) return false; 4166bcf27bb9a4b5c3f79cb44c0e4654a6d7619ad89Stephen Hines if (tok.getRawIdentifier() != "property") 417b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return false; 418b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis lexer.LexFromRawLexer(tok); 419b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (tok.isNot(tok::l_paren)) return false; 420b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 421b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Token BeforeTok = tok; 422b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Token AfterTok; 423b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis AfterTok.startToken(); 424b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis SourceLocation AttrLoc; 425b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 426b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis lexer.LexFromRawLexer(tok); 427b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (tok.is(tok::r_paren)) 428b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return false; 429b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 430b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis while (1) { 431b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (tok.isNot(tok::raw_identifier)) return false; 4326bcf27bb9a4b5c3f79cb44c0e4654a6d7619ad89Stephen Hines if (tok.getRawIdentifier() == fromAttr) { 433b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (!toAttr.empty()) { 434b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Pass.TA.replaceText(tok.getLocation(), fromAttr, toAttr); 435b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return true; 436b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } 437b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis // We want to remove the attribute. 438b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis AttrLoc = tok.getLocation(); 439b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } 440b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 441b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis do { 442b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis lexer.LexFromRawLexer(tok); 443b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (AttrLoc.isValid() && AfterTok.is(tok::unknown)) 444b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis AfterTok = tok; 445b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } while (tok.isNot(tok::comma) && tok.isNot(tok::r_paren)); 446b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (tok.is(tok::r_paren)) 447b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis break; 448b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (AttrLoc.isInvalid()) 449b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis BeforeTok = tok; 450b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis lexer.LexFromRawLexer(tok); 451b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } 452b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 453b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (toAttr.empty() && AttrLoc.isValid() && AfterTok.isNot(tok::unknown)) { 454b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis // We want to remove the attribute. 455b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis if (BeforeTok.is(tok::l_paren) && AfterTok.is(tok::r_paren)) { 456b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Pass.TA.remove(SourceRange(BeforeTok.getLocation(), 457b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis AfterTok.getLocation())); 458b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } else if (BeforeTok.is(tok::l_paren) && AfterTok.is(tok::comma)) { 459b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Pass.TA.remove(SourceRange(AttrLoc, AfterTok.getLocation())); 460b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } else { 461b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis Pass.TA.remove(SourceRange(BeforeTok.getLocation(), AttrLoc)); 462b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } 463b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 464b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return true; 465b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis } 466b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 467b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis return false; 468b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis} 469b0e1e121b32a9a04b39f1b77b3068bce2f3be05aArgyrios Kyrtzidis 4706da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidisbool MigrationContext::addPropertyAttribute(StringRef attr, 4716da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis SourceLocation atLoc) { 4726da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis if (atLoc.isMacroID()) 4736da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis return false; 4746da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 4756da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis SourceManager &SM = Pass.Ctx.getSourceManager(); 4766da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 4776da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis // Break down the source location. 4786da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis std::pair<FileID, unsigned> locInfo = SM.getDecomposedLoc(atLoc); 4796da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 4806da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis // Try to load the file buffer. 4816da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis bool invalidTemp = false; 4826da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis StringRef file = SM.getBufferData(locInfo.first, &invalidTemp); 4836da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis if (invalidTemp) 4846da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis return false; 4856da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 4866da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis const char *tokenBegin = file.data() + locInfo.second; 4876da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 4886da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis // Lex from the start of the given location. 4896da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis Lexer lexer(SM.getLocForStartOfFile(locInfo.first), 4904e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie Pass.Ctx.getLangOpts(), 4916da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis file.begin(), tokenBegin, file.end()); 4926da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis Token tok; 4936da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis lexer.LexFromRawLexer(tok); 4946da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis if (tok.isNot(tok::at)) return false; 4956da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis lexer.LexFromRawLexer(tok); 4966da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis if (tok.isNot(tok::raw_identifier)) return false; 4976bcf27bb9a4b5c3f79cb44c0e4654a6d7619ad89Stephen Hines if (tok.getRawIdentifier() != "property") 4986da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis return false; 4996da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis lexer.LexFromRawLexer(tok); 5006da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 5016da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis if (tok.isNot(tok::l_paren)) { 5026da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis Pass.TA.insert(tok.getLocation(), std::string("(") + attr.str() + ") "); 5036da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis return true; 5046da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis } 5056da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 5066da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis lexer.LexFromRawLexer(tok); 5076da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis if (tok.is(tok::r_paren)) { 5086da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis Pass.TA.insert(tok.getLocation(), attr); 5096da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis return true; 5106da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis } 5116da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 5126da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis if (tok.isNot(tok::raw_identifier)) return false; 5136da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 5146da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis Pass.TA.insert(tok.getLocation(), std::string(attr) + ", "); 5156da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis return true; 5166da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis} 5176da4274724f6bd0f24f0fbfd1af076cd14b14710Argyrios Kyrtzidis 518e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidisvoid MigrationContext::traverse(TranslationUnitDecl *TU) { 519f38fa73e605587abfa32ab6a39887cdebc0a320aArgyrios Kyrtzidis for (traverser_iterator 520f38fa73e605587abfa32ab6a39887cdebc0a320aArgyrios Kyrtzidis I = traversers_begin(), E = traversers_end(); I != E; ++I) 521f38fa73e605587abfa32ab6a39887cdebc0a320aArgyrios Kyrtzidis (*I)->traverseTU(*this); 522f38fa73e605587abfa32ab6a39887cdebc0a320aArgyrios Kyrtzidis 523e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis ASTTransform(*this).TraverseDecl(TU); 524e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis} 525e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 526bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanianstatic void GCRewriteFinalize(MigrationPass &pass) { 527bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian ASTContext &Ctx = pass.Ctx; 528bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian TransformActions &TA = pass.TA; 529bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian DeclContext *DC = Ctx.getTranslationUnitDecl(); 530bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian Selector FinalizeSel = 531bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian Ctx.Selectors.getNullarySelector(&pass.Ctx.Idents.get("finalize")); 532bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian 533bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian typedef DeclContext::specific_decl_iterator<ObjCImplementationDecl> 534bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian impl_iterator; 535bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian for (impl_iterator I = impl_iterator(DC->decls_begin()), 536bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian E = impl_iterator(DC->decls_end()); I != E; ++I) { 537651f13cea278ec967336033dd032faef0e9fc2ecStephen Hines for (const auto *MD : I->instance_methods()) { 538bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian if (!MD->hasBody()) 539bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian continue; 540bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian 541bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian if (MD->isInstanceMethod() && MD->getSelector() == FinalizeSel) { 542651f13cea278ec967336033dd032faef0e9fc2ecStephen Hines const ObjCMethodDecl *FinalizeM = MD; 543bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian Transaction Trans(TA); 544bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian TA.insert(FinalizeM->getSourceRange().getBegin(), 545bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian "#if !__has_feature(objc_arc)\n"); 546bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian CharSourceRange::getTokenRange(FinalizeM->getSourceRange()); 547bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian const SourceManager &SM = pass.Ctx.getSourceManager(); 5484e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie const LangOptions &LangOpts = pass.Ctx.getLangOpts(); 549bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian bool Invalid; 550bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian std::string str = "\n#endif\n"; 551bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian str += Lexer::getSourceText( 552bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian CharSourceRange::getTokenRange(FinalizeM->getSourceRange()), 553bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian SM, LangOpts, &Invalid); 554bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian TA.insertAfterToken(FinalizeM->getSourceRange().getEnd(), str); 555bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian 556bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian break; 557bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian } 558bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian } 559bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian } 560bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian} 561bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian 562e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis//===----------------------------------------------------------------------===// 5638f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall// getAllTransformations. 5648f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall//===----------------------------------------------------------------------===// 5658f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 566e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidisstatic void traverseAST(MigrationPass &pass) { 567e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis MigrationContext MigrateCtx(pass); 568e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 569e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis if (pass.isGCMigration()) { 570e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis MigrateCtx.addTraverser(new GCCollectableCallsTraverser); 571f38fa73e605587abfa32ab6a39887cdebc0a320aArgyrios Kyrtzidis MigrateCtx.addTraverser(new GCAttrsTraverser()); 572e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis } 573b0d5db1b7cac1a0680877c17fe92c6d64cf1c730Argyrios Kyrtzidis MigrateCtx.addTraverser(new PropertyRewriteTraverser()); 5742a2781805a6b55573d369e34c5dcfba307ce83e9Argyrios Kyrtzidis MigrateCtx.addTraverser(new BlockObjCVariableTraverser()); 575ea2224d7078c4d31ad32adbaba4bdc2d85a3d609Argyrios Kyrtzidis MigrateCtx.addTraverser(new ProtectedScopeTraverser()); 576e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 577e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis MigrateCtx.traverse(pass.Ctx.getTranslationUnitDecl()); 578e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis} 579e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis 5808f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCallstatic void independentTransforms(MigrationPass &pass) { 5818f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall rewriteAutoreleasePool(pass); 582e7ef8556f4ee3012a0479308c993af0fbee448dfArgyrios Kyrtzidis removeRetainReleaseDeallocFinalize(pass); 5837196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis rewriteUnusedInitDelegate(pass); 584e7ef8556f4ee3012a0479308c993af0fbee448dfArgyrios Kyrtzidis removeZeroOutPropsInDeallocFinalize(pass); 5858f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall makeAssignARCSafe(pass); 5867196d06c2fb020a91a26e727be1871110b4a0dc9Argyrios Kyrtzidis rewriteUnbridgedCasts(pass); 587fd10398c10ffdcbdeb1e3e299c74d70e689f503cArgyrios Kyrtzidis checkAPIUses(pass); 588e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidis traverseAST(pass); 5898f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall} 5908f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 591e0ac7454bae910ab3d67a92f6e2e5046d3bb8c1aArgyrios Kyrtzidisstd::vector<TransformFn> arcmt::getAllTransformations( 592bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian LangOptions::GCMode OrigGCMode, 593bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian bool NoFinalizeRemoval) { 5948f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall std::vector<TransformFn> transforms; 5958f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 596bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian if (OrigGCMode == LangOptions::GCOnly && NoFinalizeRemoval) 597bbdfad581fa300fa0d162d968ec14de3c95fc760Fariborz Jahanian transforms.push_back(GCRewriteFinalize); 5988f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall transforms.push_back(independentTransforms); 599fd3455a9ebecf67b9a1c46f0675697bf518e6b80Argyrios Kyrtzidis // This depends on previous transformations removing various expressions. 600e7ef8556f4ee3012a0479308c993af0fbee448dfArgyrios Kyrtzidis transforms.push_back(removeEmptyStatementsAndDeallocFinalize); 6018f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall 6028f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall return transforms; 6038f0e8d22960d56f8390f4971e2c0f2f0a0884602John McCall} 604