//===--- 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());

}