llvm-project

1
//===--- TransGCCalls.cpp - Transformations to ARC mode -------------------===//
2
//
3
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4
// See https://llvm.org/LICENSE.txt for license information.
5
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6
//
7
//===----------------------------------------------------------------------===//
8

9
#include "Transforms.h"
10
#include "Internals.h"
11
#include "clang/AST/ASTContext.h"
12
#include "clang/Sema/SemaDiagnostic.h"
13

14
using namespace clang;
15
using namespace arcmt;
16
using namespace trans;
17

18
namespace {
19

20
class GCCollectableCallsChecker :
21
                         public RecursiveASTVisitor<GCCollectableCallsChecker> {
22
  MigrationContext &MigrateCtx;
23
  IdentifierInfo *NSMakeCollectableII;
24
  IdentifierInfo *CFMakeCollectableII;
25

26
public:
27
  GCCollectableCallsChecker(MigrationContext &ctx)
28
    : MigrateCtx(ctx) {
29
    IdentifierTable &Ids = MigrateCtx.Pass.Ctx.Idents;
30
    NSMakeCollectableII = &Ids.get("NSMakeCollectable");
31
    CFMakeCollectableII = &Ids.get("CFMakeCollectable");
32
  }
33

34
  bool shouldWalkTypesOfTypeLocs() const { return false; }
35

36
  bool VisitCallExpr(CallExpr *E) {
37
    TransformActions &TA = MigrateCtx.Pass.TA;
38

39
    if (MigrateCtx.isGCOwnedNonObjC(E->getType())) {
40
      TA.report(E->getBeginLoc(), diag::warn_arcmt_nsalloc_realloc,
41
                E->getSourceRange());
42
      return true;
43
    }
44

45
    Expr *CEE = E->getCallee()->IgnoreParenImpCasts();
46
    if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CEE)) {
47
      if (FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(DRE->getDecl())) {
48
        if (!FD->getDeclContext()->getRedeclContext()->isFileContext())
49
          return true;
50

51
        if (FD->getIdentifier() == NSMakeCollectableII) {
52
          Transaction Trans(TA);
53
          TA.clearDiagnostic(diag::err_unavailable,
54
                             diag::err_unavailable_message,
55
                             diag::err_ovl_deleted_call, // ObjC++
56
                             DRE->getSourceRange());
57
          TA.replace(DRE->getSourceRange(), "CFBridgingRelease");
58

59
        } else if (FD->getIdentifier() == CFMakeCollectableII) {
60
          TA.reportError("CFMakeCollectable will leak the object that it "
61
                         "receives in ARC", DRE->getLocation(),
62
                         DRE->getSourceRange());
63
        }
64
      }
65
    }
66

67
    return true;
68
  }
69
};
70

71
} // anonymous namespace
72

73
void GCCollectableCallsTraverser::traverseBody(BodyContext &BodyCtx) {
74
  GCCollectableCallsChecker(BodyCtx.getMigrationContext())
75
                                            .TraverseStmt(BodyCtx.getTopStmt());
76
}
77