1 //===--- TransGCCalls.cpp - Transformations to ARC mode -------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "Transforms.h" 11 #include "Internals.h" 12 #include "clang/AST/ASTContext.h" 13 #include "clang/Sema/SemaDiagnostic.h" 14 15 using namespace clang; 16 using namespace arcmt; 17 using namespace trans; 18 19 namespace { 20 21 class GCCollectableCallsChecker : 22 public RecursiveASTVisitor<GCCollectableCallsChecker> { 23 MigrationContext &MigrateCtx; 24 IdentifierInfo *NSMakeCollectableII; 25 IdentifierInfo *CFMakeCollectableII; 26 27 public: GCCollectableCallsChecker(MigrationContext & ctx)28 GCCollectableCallsChecker(MigrationContext &ctx) 29 : MigrateCtx(ctx) { 30 IdentifierTable &Ids = MigrateCtx.Pass.Ctx.Idents; 31 NSMakeCollectableII = &Ids.get("NSMakeCollectable"); 32 CFMakeCollectableII = &Ids.get("CFMakeCollectable"); 33 } 34 shouldWalkTypesOfTypeLocs() const35 bool shouldWalkTypesOfTypeLocs() const { return false; } 36 VisitCallExpr(CallExpr * E)37 bool VisitCallExpr(CallExpr *E) { 38 TransformActions &TA = MigrateCtx.Pass.TA; 39 40 if (MigrateCtx.isGCOwnedNonObjC(E->getType())) { 41 TA.report(E->getLocStart(), diag::warn_arcmt_nsalloc_realloc, 42 E->getSourceRange()); 43 return true; 44 } 45 46 Expr *CEE = E->getCallee()->IgnoreParenImpCasts(); 47 if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CEE)) { 48 if (FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(DRE->getDecl())) { 49 if (!FD->getDeclContext()->getRedeclContext()->isFileContext()) 50 return true; 51 52 if (FD->getIdentifier() == NSMakeCollectableII) { 53 Transaction Trans(TA); 54 TA.clearDiagnostic(diag::err_unavailable, 55 diag::err_unavailable_message, 56 diag::err_ovl_deleted_call, // ObjC++ 57 DRE->getSourceRange()); 58 TA.replace(DRE->getSourceRange(), "CFBridgingRelease"); 59 60 } else if (FD->getIdentifier() == CFMakeCollectableII) { 61 TA.reportError("CFMakeCollectable will leak the object that it " 62 "receives in ARC", DRE->getLocation(), 63 DRE->getSourceRange()); 64 } 65 } 66 } 67 68 return true; 69 } 70 }; 71 72 } // anonymous namespace 73 traverseBody(BodyContext & BodyCtx)74void GCCollectableCallsTraverser::traverseBody(BodyContext &BodyCtx) { 75 GCCollectableCallsChecker(BodyCtx.getMigrationContext()) 76 .TraverseStmt(BodyCtx.getTopStmt()); 77 } 78