77 lines
2.5 KiB
C++
77 lines
2.5 KiB
C++
|
//===--- TransGCCalls.cpp - Transformations to ARC mode -------------------===//
|
||
|
|
//
|
||
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
||
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
||
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#include "Transforms.h"
|
||
|
|
#include "Internals.h"
|
||
|
|
#include "clang/AST/ASTContext.h"
|
||
|
|
#include "clang/Sema/SemaDiagnostic.h"
|
||
|
|
|
||
|
|
using namespace clang;
|
||
|
|
using namespace arcmt;
|
||
|
|
using namespace trans;
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
class GCCollectableCallsChecker :
|
||
|
|
public RecursiveASTVisitor<GCCollectableCallsChecker> {
|
||
|
|
MigrationContext &MigrateCtx;
|
||
|
|
IdentifierInfo *NSMakeCollectableII;
|
||
|
|
IdentifierInfo *CFMakeCollectableII;
|
||
|
|
|
||
|
|
public:
|
||
|
|
GCCollectableCallsChecker(MigrationContext &ctx)
|
||
|
|
: MigrateCtx(ctx) {
|
||
|
|
IdentifierTable &Ids = MigrateCtx.Pass.Ctx.Idents;
|
||
|
|
NSMakeCollectableII = &Ids.get("NSMakeCollectable");
|
||
|
|
CFMakeCollectableII = &Ids.get("CFMakeCollectable");
|
||
|
|
}
|
||
|
|
|
||
|
|
bool shouldWalkTypesOfTypeLocs() const { return false; }
|
||
|
|
|
||
|
|
bool VisitCallExpr(CallExpr *E) {
|
||
|
|
TransformActions &TA = MigrateCtx.Pass.TA;
|
||
|
|
|
||
|
|
if (MigrateCtx.isGCOwnedNonObjC(E->getType())) {
|
||
|
|
TA.report(E->getBeginLoc(), diag::warn_arcmt_nsalloc_realloc,
|
||
|
|
E->getSourceRange());
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
Expr *CEE = E->getCallee()->IgnoreParenImpCasts();
|
||
|
|
if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(CEE)) {
|
||
|
|
if (FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(DRE->getDecl())) {
|
||
|
|
if (!FD->getDeclContext()->getRedeclContext()->isFileContext())
|
||
|
|
return true;
|
||
|
|
|
||
|
|
if (FD->getIdentifier() == NSMakeCollectableII) {
|
||
|
|
Transaction Trans(TA);
|
||
|
|
TA.clearDiagnostic(diag::err_unavailable,
|
||
|
|
diag::err_unavailable_message,
|
||
|
|
diag::err_ovl_deleted_call, // ObjC++
|
||
|
|
DRE->getSourceRange());
|
||
|
|
TA.replace(DRE->getSourceRange(), "CFBridgingRelease");
|
||
|
|
|
||
|
|
} else if (FD->getIdentifier() == CFMakeCollectableII) {
|
||
|
|
TA.reportError("CFMakeCollectable will leak the object that it "
|
||
|
|
"receives in ARC", DRE->getLocation(),
|
||
|
|
DRE->getSourceRange());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // anonymous namespace
|
||
|
|
|
||
|
|
void GCCollectableCallsTraverser::traverseBody(BodyContext &BodyCtx) {
|
||
|
|
GCCollectableCallsChecker(BodyCtx.getMigrationContext())
|
||
|
|
.TraverseStmt(BodyCtx.getTopStmt());
|
||
|
|
}
|