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
15using namespace clang;
16using namespace arcmt;
17using namespace trans;
18
19namespace {
20
21class GCCollectableCallsChecker :
22                         public RecursiveASTVisitor<GCCollectableCallsChecker> {
23  MigrationContext &MigrateCtx;
24  IdentifierInfo *NSMakeCollectableII;
25  IdentifierInfo *CFMakeCollectableII;
26
27public:
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
35  bool shouldWalkTypesOfTypeLocs() const { return false; }
36
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
74void GCCollectableCallsTraverser::traverseBody(BodyContext &BodyCtx) {
75  GCCollectableCallsChecker(BodyCtx.getMigrationContext())
76                                            .TraverseStmt(BodyCtx.getTopStmt());
77}
78