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