//===--- ObjCMT.cpp - ObjC Migrate Tool -----------------------------------===//

//

// 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 "clang/Analysis/RetainSummaryManager.h"

#include "clang/ARCMigrate/ARCMT.h"

#include "clang/ARCMigrate/ARCMTActions.h"

#include "clang/AST/ASTConsumer.h"

#include "clang/AST/ASTContext.h"

#include "clang/AST/Attr.h"

#include "clang/AST/NSAPI.h"

#include "clang/AST/ParentMap.h"

#include "clang/AST/RecursiveASTVisitor.h"

#include "clang/Analysis/DomainSpecific/CocoaConventions.h"

#include "clang/Basic/FileManager.h"

#include "clang/Edit/Commit.h"

#include "clang/Edit/EditedSource.h"

#include "clang/Edit/EditsReceiver.h"

#include "clang/Edit/Rewriters.h"

#include "clang/Frontend/CompilerInstance.h"

#include "clang/Frontend/MultiplexConsumer.h"

#include "clang/Lex/PPConditionalDirectiveRecord.h"

#include "clang/Lex/Preprocessor.h"

#include "clang/Rewrite/Core/Rewriter.h"

#include "llvm/ADT/SmallString.h"

#include "llvm/ADT/StringSet.h"

#include "llvm/Support/Path.h"

#include "llvm/Support/SourceMgr.h"

#include "llvm/Support/YAMLParser.h"

using namespace clang;

using namespace arcmt;

using namespace ento;

namespace {

class ObjCMigrateASTConsumer : public ASTConsumer {

  enum CF_BRIDGING_KIND {

    CF_BRIDGING_NONE,

    CF_BRIDGING_ENABLE,

    CF_BRIDGING_MAY_INCLUDE

  };

  void migrateDecl(Decl *D);

  void migrateObjCContainerDecl(ASTContext &Ctx, ObjCContainerDecl *D);

  void migrateProtocolConformance(ASTContext &Ctx,

                                  const ObjCImplementationDecl *ImpDecl);

  void CacheObjCNSIntegerTypedefed(const TypedefDecl *TypedefDcl);

  bool migrateNSEnumDecl(ASTContext &Ctx, const EnumDecl *EnumDcl,

                     const TypedefDecl *TypedefDcl);

  void migrateAllMethodInstaceType(ASTContext &Ctx, ObjCContainerDecl *CDecl);

  void migrateMethodInstanceType(ASTContext &Ctx, ObjCContainerDecl *CDecl,

                                 ObjCMethodDecl *OM);

  bool migrateProperty(ASTContext &Ctx, ObjCContainerDecl *D, ObjCMethodDecl *OM);

  void migrateNsReturnsInnerPointer(ASTContext &Ctx, ObjCMethodDecl *OM);

  void migratePropertyNsReturnsInnerPointer(ASTContext &Ctx, ObjCPropertyDecl *P);

  void migrateFactoryMethod(ASTContext &Ctx, ObjCContainerDecl *CDecl,

                            ObjCMethodDecl *OM,

                            ObjCInstanceTypeFamily OIT_Family = OIT_None);

  void migrateCFAnnotation(ASTContext &Ctx, const Decl *Decl);

  void AddCFAnnotations(ASTContext &Ctx,

                        const RetainSummary *RS,

                        const FunctionDecl *FuncDecl, bool ResultAnnotated);

  void AddCFAnnotations(ASTContext &Ctx,

                        const RetainSummary *RS,

                        const ObjCMethodDecl *MethodDecl, bool ResultAnnotated);

  void AnnotateImplicitBridging(ASTContext &Ctx);

  CF_BRIDGING_KIND migrateAddFunctionAnnotation(ASTContext &Ctx,

                                                const FunctionDecl *FuncDecl);

  void migrateARCSafeAnnotation(ASTContext &Ctx, ObjCContainerDecl *CDecl);

  void migrateAddMethodAnnotation(ASTContext &Ctx,

                                  const ObjCMethodDecl *MethodDecl);

  void inferDesignatedInitializers(ASTContext &Ctx,

                                   const ObjCImplementationDecl *ImplD);

  bool InsertFoundation(ASTContext &Ctx, SourceLocation Loc);

  std::unique_ptr<RetainSummaryManager> Summaries;

public:

  std::string MigrateDir;

  unsigned ASTMigrateActions;

  FileID FileId;

  const TypedefDecl *NSIntegerTypedefed;

  const TypedefDecl *NSUIntegerTypedefed;

  std::unique_ptr<NSAPI> NSAPIObj;

  std::unique_ptr<edit::EditedSource> Editor;

  FileRemapper &Remapper;

  FileManager &FileMgr;

  const PPConditionalDirectiveRecord *PPRec;

  Preprocessor &PP;

  bool IsOutputFile;

  bool FoundationIncluded;

  llvm::SmallPtrSet<ObjCProtocolDecl *, 32> ObjCProtocolDecls;

  llvm::SmallVector<const Decl *, 8> CFFunctionIBCandidates;

  llvm::StringSet<> AllowListFilenames;

  RetainSummaryManager &getSummaryManager(ASTContext &Ctx) {

    if (!Summaries)

      Summaries.reset(new RetainSummaryManager(Ctx,

                                               /*TrackNSCFObjects=*/true,

                                               /*trackOSObjects=*/false));

    return *Summaries;

  }

  ObjCMigrateASTConsumer(StringRef migrateDir, unsigned astMigrateActions,

                         FileRemapper &remapper, FileManager &fileMgr,

                         const PPConditionalDirectiveRecord *PPRec,

                         Preprocessor &PP, bool isOutputFile,

                         ArrayRef<std::string> AllowList)

      : MigrateDir(migrateDir), ASTMigrateActions(astMigrateActions),

        NSIntegerTypedefed(nullptr), NSUIntegerTypedefed(nullptr),

        Remapper(remapper), FileMgr(fileMgr), PPRec(PPRec), PP(PP),

        IsOutputFile(isOutputFile), FoundationIncluded(false) {

    // FIXME: StringSet should have insert(iter, iter) to use here.

    for (const std::string &Val : AllowList)

      AllowListFilenames.insert(Val);

  }

protected:

  void Initialize(ASTContext &Context) override {

    NSAPIObj.reset(new NSAPI(Context));

    Editor.reset(new edit::EditedSource(Context.getSourceManager(),

                                        Context.getLangOpts(),

                                        PPRec));

  }

  bool HandleTopLevelDecl(DeclGroupRef DG) override {

    for (DeclGroupRef::iterator I = DG.begin(), E = DG.end(); I != E; 
++I1.02k
)

      migrateDecl(*I);

    return true;

  }

  void HandleInterestingDecl(DeclGroupRef DG) override {

    // Ignore decls from the PCH.

  }

  void HandleTopLevelDeclInObjCContainer(DeclGroupRef DG) override {

    ObjCMigrateASTConsumer::HandleTopLevelDecl(DG);

  }

  void HandleTranslationUnit(ASTContext &Ctx) override;

  bool canModifyFile(StringRef Path) {

    if (AllowListFilenames.empty())

      return true;

    return AllowListFilenames.find(llvm::sys::path::filename(Path)) !=

           AllowListFilenames.end();

  }

  bool canModifyFile(Optional<FileEntryRef> FE) {

    if (!FE)

      return false;

    return canModifyFile(FE->getName());

  }

  bool canModifyFile(FileID FID) {

    if (FID.isInvalid())

      return false;

    return canModifyFile(PP.getSourceManager().getFileEntryRefForID(FID));

  }

  bool canModify(const Decl *D) {

    if (!D)

      return false;

    if (const ObjCCategoryImplDecl *CatImpl = dyn_cast<ObjCCategoryImplDecl>(D))

      return canModify(CatImpl->getCategoryDecl());

    if (const ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D))

      return canModify(Impl->getClassInterface());

    if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))

      return canModify(cast<Decl>(MD->getDeclContext()));

    FileID FID = PP.getSourceManager().getFileID(D->getLocation());

    return canModifyFile(FID);

  }

};

} // end anonymous namespace

ObjCMigrateAction::ObjCMigrateAction(

    std::unique_ptr<FrontendAction> WrappedAction, StringRef migrateDir,

    unsigned migrateAction)

    : WrapperFrontendAction(std::move(WrappedAction)), MigrateDir(migrateDir),

      ObjCMigAction(migrateAction), CompInst(nullptr) {

  if (MigrateDir.empty())

    MigrateDir = "."; // user current directory if none is given.

}

std::unique_ptr<ASTConsumer>

ObjCMigrateAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {

  PPConditionalDirectiveRecord *

    PPRec = new PPConditionalDirectiveRecord(CompInst->getSourceManager());

  CI.getPreprocessor().addPPCallbacks(std::unique_ptr<PPCallbacks>(PPRec));

  std::vector<std::unique_ptr<ASTConsumer>> Consumers;

  Consumers.push_back(WrapperFrontendAction::CreateASTConsumer(CI, InFile));

  Consumers.push_back(std::make_unique<ObjCMigrateASTConsumer>(

      MigrateDir, ObjCMigAction, Remapper, CompInst->getFileManager(), PPRec,

      CompInst->getPreprocessor(), false, None));

  return std::make_unique<MultiplexConsumer>(std::move(Consumers));

}

bool ObjCMigrateAction::BeginInvocation(CompilerInstance &CI) {

  Remapper.initFromDisk(MigrateDir, CI.getDiagnostics(),

                        /*ignoreIfFilesChanged=*/true);

  CompInst = &CI;

  CI.getDiagnostics().setIgnoreAllWarnings(true);

  return true;

}

namespace {

  // FIXME. This duplicates one in RewriteObjCFoundationAPI.cpp

  bool subscriptOperatorNeedsParens(const Expr *FullExpr) {

    const Expr* Expr = FullExpr->IgnoreImpCasts();

    return !(isa<ArraySubscriptExpr>(Expr) || isa<CallExpr>(Expr) ||

             
isa<DeclRefExpr>(Expr)31
 || 
isa<CXXNamedCastExpr>(Expr)22
 ||

             
isa<CXXConstructExpr>(Expr)22
 || 
isa<CXXThisExpr>(Expr)22
 ||

             
isa<CXXTypeidExpr>(Expr)22
 ||

             
isa<CXXUnresolvedConstructExpr>(Expr)22
 ||

             
isa<ObjCMessageExpr>(Expr)22
 || 
isa<ObjCPropertyRefExpr>(Expr)18
 ||

             
isa<ObjCProtocolExpr>(Expr)18
 || 
isa<MemberExpr>(Expr)18
 ||

             
isa<ObjCIvarRefExpr>(Expr)18
 || 
isa<ParenExpr>(FullExpr)3
 ||

             
isa<ParenListExpr>(Expr)3
 || 
isa<SizeOfPackExpr>(Expr)3
);

  }

  /// - Rewrite message expression for Objective-C setter and getters into

  /// property-dot syntax.

  bool rewriteToPropertyDotSyntax(const ObjCMessageExpr *Msg,

                                  Preprocessor &PP,

                                  const NSAPI &NS, edit::Commit &commit,

                                  const ParentMap *PMap) {

    if (!Msg || Msg->isImplicit() ||

        (Msg->getReceiverKind() != ObjCMessageExpr::Instance &&

         
Msg->getReceiverKind() != ObjCMessageExpr::SuperInstance7
))

      return false;

    if (const Expr *Receiver = Msg->getInstanceReceiver())

      if (Receiver->getType()->isObjCBuiltinType())

        return false;

    const ObjCMethodDecl *Method = Msg->getMethodDecl();

    if (!Method)

      return false;

    if (!Method->isPropertyAccessor())

      return false;

    const ObjCPropertyDecl *Prop = Method->findPropertyDecl();

    if (!Prop)

      return false;

    SourceRange MsgRange = Msg->getSourceRange();

    bool ReceiverIsSuper =

      (Msg->getReceiverKind() == ObjCMessageExpr::SuperInstance);

    // for 'super' receiver is nullptr.

    const Expr *receiver = Msg->getInstanceReceiver();

    bool NeedsParen =

      ReceiverIsSuper ? 
false7
 : 
subscriptOperatorNeedsParens(receiver)33
;

    bool IsGetter = (Msg->getNumArgs() == 0);

    if (IsGetter) {

      // Find space location range between receiver expression and getter method.

      SourceLocation BegLoc =

          ReceiverIsSuper ? 
Msg->getSuperLoc()4
 : 
receiver->getEndLoc()19
;

      BegLoc = PP.getLocForEndOfToken(BegLoc);

      SourceLocation EndLoc = Msg->getSelectorLoc(0);

      SourceRange SpaceRange(BegLoc, EndLoc);

      std::string PropertyDotString;

      // rewrite getter method expression into: receiver.property or

      // (receiver).property

      if (NeedsParen) {

        commit.insertBefore(receiver->getBeginLoc(), "(");

        PropertyDotString = ").";

      }

      else

        PropertyDotString = ".";

      PropertyDotString += Prop->getName();

      commit.replace(SpaceRange, PropertyDotString);

      // remove '[' ']'

      commit.replace(SourceRange(MsgRange.getBegin(), MsgRange.getBegin()), "");

      commit.replace(SourceRange(MsgRange.getEnd(), MsgRange.getEnd()), "");

    } else {

      if (NeedsParen)

        commit.insertWrap("(", receiver->getSourceRange(), ")");

      std::string PropertyDotString = ".";

      PropertyDotString += Prop->getName();

      PropertyDotString += " =";

      const Expr*const* Args = Msg->getArgs();

      const Expr *RHS = Args[0];

      if (!RHS)

        return false;

      SourceLocation BegLoc =

          ReceiverIsSuper ? 
Msg->getSuperLoc()3
 : 
receiver->getEndLoc()14
;

      BegLoc = PP.getLocForEndOfToken(BegLoc);

      SourceLocation EndLoc = RHS->getBeginLoc();

      EndLoc = EndLoc.getLocWithOffset(-1);

      const char *colon = PP.getSourceManager().getCharacterData(EndLoc);

      // Add a space after '=' if there is no space between RHS and '='

      if (colon && colon[0] == ':')

        PropertyDotString += " ";

      SourceRange Range(BegLoc, EndLoc);

      commit.replace(Range, PropertyDotString);

      // remove '[' ']'

      commit.replace(SourceRange(MsgRange.getBegin(), MsgRange.getBegin()), "");

      commit.replace(SourceRange(MsgRange.getEnd(), MsgRange.getEnd()), "");

    }

    return true;

  }

class ObjCMigrator : public RecursiveASTVisitor<ObjCMigrator> {

  ObjCMigrateASTConsumer &Consumer;

  ParentMap &PMap;

public:

  ObjCMigrator(ObjCMigrateASTConsumer &consumer, ParentMap &PMap)

    : Consumer(consumer), PMap(PMap) { }

  bool shouldVisitTemplateInstantiations() const { return false; }

  bool shouldWalkTypesOfTypeLocs() const { return false; }

  bool VisitObjCMessageExpr(ObjCMessageExpr *E) {

    if (Consumer.ASTMigrateActions & FrontendOptions::ObjCMT_Literals) {

      edit::Commit commit(*Consumer.Editor);

      edit::rewriteToObjCLiteralSyntax(E, *Consumer.NSAPIObj, commit, &PMap);

      Consumer.Editor->commit(commit);

    }

    if (Consumer.ASTMigrateActions & FrontendOptions::ObjCMT_Subscripting) {

      edit::Commit commit(*Consumer.Editor);

      edit::rewriteToObjCSubscriptSyntax(E, *Consumer.NSAPIObj, commit);

      Consumer.Editor->commit(commit);

    }

    if (Consumer.ASTMigrateActions & FrontendOptions::ObjCMT_PropertyDotSyntax) {

      edit::Commit commit(*Consumer.Editor);

      rewriteToPropertyDotSyntax(E, Consumer.PP, *Consumer.NSAPIObj,

                                 commit, &PMap);

      Consumer.Editor->commit(commit);

    }

    return true;

  }

  bool TraverseObjCMessageExpr(ObjCMessageExpr *E) {

    // Do depth first; we want to rewrite the subexpressions first so that if

    // we have to move expressions we will move them already rewritten.

    for (Stmt *SubStmt : E->children())

      if (!TraverseStmt(SubStmt))

        return false;

    return WalkUpFromObjCMessageExpr(E);

  }

};

class BodyMigrator : public RecursiveASTVisitor<BodyMigrator> {

  ObjCMigrateASTConsumer &Consumer;

  std::unique_ptr<ParentMap> PMap;

public:

  BodyMigrator(ObjCMigrateASTConsumer &consumer) : Consumer(consumer) { }

  bool shouldVisitTemplateInstantiations() const { return false; }

  bool shouldWalkTypesOfTypeLocs() const { return false; }

  bool TraverseStmt(Stmt *S) {

    PMap.reset(new ParentMap(S));

    ObjCMigrator(Consumer, *PMap).TraverseStmt(S);

    return true;

  }

};

} // end anonymous namespace

void ObjCMigrateASTConsumer::migrateDecl(Decl *D) {

  if (!D)

    return;

  if (isa<ObjCMethodDecl>(D))

    return; // Wait for the ObjC container declaration.

  BodyMigrator(*this).TraverseDecl(D);

}

static void append_attr(std::string &PropertyString, const char *attr,

                        bool &LParenAdded) {

  if (!LParenAdded) {

    PropertyString += "(";

    LParenAdded = true;

  }

  else

    PropertyString += ", ";

  PropertyString += attr;

}

static

void MigrateBlockOrFunctionPointerTypeVariable(std::string & PropertyString,

                                               const std::string& TypeString,

                                               const char *name) {

  const char *argPtr = TypeString.c_str();

  int paren = 0;

  while (*argPtr) {

    switch (*argPtr) {

      case '(':

        PropertyString += *argPtr;

        paren++;

        break;

      case ')':

        PropertyString += *argPtr;

        paren--;

        break;

      case '^':

      case '*':

        PropertyString += (*argPtr);

        if (paren == 1) {

          PropertyString += name;

          name = "";

        }

        break;

      default:

        PropertyString += *argPtr;

        break;

    }

    argPtr++;

  }

}

static const char *PropertyMemoryAttribute(ASTContext &Context, QualType ArgType) {

  Qualifiers::ObjCLifetime propertyLifetime = ArgType.getObjCLifetime();

  bool RetainableObject = ArgType->isObjCRetainableType();

  if (RetainableObject &&

      
(85
propertyLifetime == Qualifiers::OCL_Strong85

       || 
propertyLifetime == Qualifiers::OCL_None19
)) {

    if (const ObjCObjectPointerType *ObjPtrTy =

        ArgType->getAs<ObjCObjectPointerType>()) {

      ObjCInterfaceDecl *IDecl = ObjPtrTy->getObjectType()->getInterface();

      if (IDecl &&

          
IDecl->lookupNestedProtocol(&Context.Idents.get("NSCopying"))50
)

        return "copy";

      else

        return "strong";

    }

    else if (ArgType->isBlockPointerType())

      return "copy";

  } else if (propertyLifetime == Qualifiers::OCL_Weak)

    // TODO. More precise determination of 'weak' attribute requires

    // looking into setter's implementation for backing weak ivar.

    return "weak";

  else if (RetainableObject)

    return ArgType->isBlockPointerType() ? "copy" : "strong";

  return nullptr;

}

static void rewriteToObjCProperty(const ObjCMethodDecl *Getter,

                                  const ObjCMethodDecl *Setter,

                                  const NSAPI &NS, edit::Commit &commit,

                                  unsigned LengthOfPrefix,

                                  bool Atomic, bool UseNsIosOnlyMacro,

                                  bool AvailabilityArgsMatch) {

  ASTContext &Context = NS.getASTContext();

  bool LParenAdded = false;

  std::string PropertyString = "@property ";

  if (UseNsIosOnlyMacro && 
NS.isMacroDefined("NS_NONATOMIC_IOSONLY")68
) {

    PropertyString += "(NS_NONATOMIC_IOSONLY";

    LParenAdded = true;

  } else if (!Atomic) {

    PropertyString += "(nonatomic";

    LParenAdded = true;

  }

  std::string PropertyNameString = Getter->getNameAsString();

  StringRef PropertyName(PropertyNameString);

  if (LengthOfPrefix > 0) {

    if (!LParenAdded) {

      PropertyString += "(getter=";

      LParenAdded = true;

    }

    else

      PropertyString += ", getter=";

    PropertyString += PropertyNameString;

  }

  // Property with no setter may be suggested as a 'readonly' property.

  if (!Setter)

    append_attr(PropertyString, "readonly", LParenAdded);

  // Short circuit 'delegate' properties that contain the name "delegate" or

  // "dataSource", or have exact name "target" to have 'assign' attribute.

  if (PropertyName.equals("target") || 
PropertyName.contains("delegate")189
 ||

      
PropertyName.contains("dataSource")182
) {

    QualType QT = Getter->getReturnType();

    if (!QT->isRealType())

      append_attr(PropertyString, "assign", LParenAdded);

  } else if (!Setter) {

    QualType ResType = Context.getCanonicalType(Getter->getReturnType());

    if (const char *MemoryManagementAttr = PropertyMemoryAttribute(Context, ResType))

      append_attr(PropertyString, MemoryManagementAttr, LParenAdded);

  } else {

    const ParmVarDecl *argDecl = *Setter->param_begin();

    QualType ArgType = Context.getCanonicalType(argDecl->getType());

    if (const char *MemoryManagementAttr = PropertyMemoryAttribute(Context, ArgType))

      append_attr(PropertyString, MemoryManagementAttr, LParenAdded);

  }

  if (LParenAdded)

    PropertyString += ')';

  QualType RT = Getter->getReturnType();

  if (!isa<TypedefType>(RT)) {

    // strip off any ARC lifetime qualifier.

    QualType CanResultTy = Context.getCanonicalType(RT);

    if (CanResultTy.getQualifiers().hasObjCLifetime()) {

      Qualifiers Qs = CanResultTy.getQualifiers();

      Qs.removeObjCLifetime();

      RT = Context.getQualifiedType(CanResultTy.getUnqualifiedType(), Qs);

    }

  }

  PropertyString += " ";

  PrintingPolicy SubPolicy(Context.getPrintingPolicy());

  SubPolicy.SuppressStrongLifetime = true;

  SubPolicy.SuppressLifetimeQualifiers = true;

  std::string TypeString = RT.getAsString(SubPolicy);

  if (LengthOfPrefix > 0) {

    // property name must strip off "is" and lower case the first character

    // after that; e.g. isContinuous will become continuous.

    StringRef PropertyNameStringRef(PropertyNameString);

    PropertyNameStringRef = PropertyNameStringRef.drop_front(LengthOfPrefix);

    PropertyNameString = std::string(PropertyNameStringRef);

    bool NoLowering = (isUppercase(PropertyNameString[0]) &&

                       
PropertyNameString.size() > 148
 &&

                       
isUppercase(PropertyNameString[1])42
);

    if (!NoLowering)

      PropertyNameString[0] = toLowercase(PropertyNameString[0]);

  }

  if (RT->isBlockPointerType() || 
RT->isFunctionPointerType()190
)

    MigrateBlockOrFunctionPointerTypeVariable(PropertyString,

                                              TypeString,

                                              PropertyNameString.c_str());

  else {

    char LastChar = TypeString[TypeString.size()-1];

    PropertyString += TypeString;

    if (LastChar != '*')

      PropertyString += ' ';

    PropertyString += PropertyNameString;

  }

  SourceLocation StartGetterSelectorLoc = Getter->getSelectorStartLoc();

  Selector GetterSelector = Getter->getSelector();

  SourceLocation EndGetterSelectorLoc =

    StartGetterSelectorLoc.getLocWithOffset(GetterSelector.getNameForSlot(0).size());

  commit.replace(CharSourceRange::getCharRange(Getter->getBeginLoc(),

                                               EndGetterSelectorLoc),

                 PropertyString);

  if (Setter && 
AvailabilityArgsMatch111
) {

    SourceLocation EndLoc = Setter->getDeclaratorEndLoc();

    // Get location past ';'

    EndLoc = EndLoc.getLocWithOffset(1);

    SourceLocation BeginOfSetterDclLoc = Setter->getBeginLoc();

    // FIXME. This assumes that setter decl; is immediately preceded by eoln.

    // It is trying to remove the setter method decl. line entirely.

    BeginOfSetterDclLoc = BeginOfSetterDclLoc.getLocWithOffset(-1);

    commit.remove(SourceRange(BeginOfSetterDclLoc, EndLoc));

  }

}

static bool IsCategoryNameWithDeprecatedSuffix(ObjCContainerDecl *D) {

  if (ObjCCategoryDecl *CatDecl = dyn_cast<ObjCCategoryDecl>(D)) {

    StringRef Name = CatDecl->getName();

    return Name.endswith("Deprecated");

  }

  return false;

}

void ObjCMigrateASTConsumer::migrateObjCContainerDecl(ASTContext &Ctx,

                                                      ObjCContainerDecl *D) {

  if (D->isDeprecated() || 
IsCategoryNameWithDeprecatedSuffix(D)248
)

    return;

  
for (auto *Method : D->methods())247
 {

    if (Method->isDeprecated())

      continue;

    bool PropertyInferred = migrateProperty(Ctx, D, Method);

    // If a property is inferred, do not attempt to attach NS_RETURNS_INNER_POINTER to

    // the getter method as it ends up on the property itself which we don't want

    // to do unless -objcmt-returns-innerpointer-property  option is on.

    if (!PropertyInferred ||

        
(ASTMigrateActions & FrontendOptions::ObjCMT_ReturnsInnerPointerProperty)196
)

      if (ASTMigrateActions & FrontendOptions::ObjCMT_Annotation)

        migrateNsReturnsInnerPointer(Ctx, Method);

  }

  if (!(ASTMigrateActions & FrontendOptions::ObjCMT_ReturnsInnerPointerProperty))

    return;

  for (auto *Prop : D->instance_properties()) {

    if ((ASTMigrateActions & FrontendOptions::ObjCMT_Annotation) &&

        !Prop->isDeprecated())

      migratePropertyNsReturnsInnerPointer(Ctx, Prop);

  }

}

static bool

ClassImplementsAllMethodsAndProperties(ASTContext &Ctx,

                                      const ObjCImplementationDecl *ImpDecl,

                                       const ObjCInterfaceDecl *IDecl,

                                      ObjCProtocolDecl *Protocol) {

  // In auto-synthesis, protocol properties are not synthesized. So,

  // a conforming protocol must have its required properties declared

  // in class interface.

  bool HasAtleastOneRequiredProperty = false;

  if (const ObjCProtocolDecl *PDecl = Protocol->getDefinition())

    for (const auto *Property : PDecl->instance_properties()) {

      if (Property->getPropertyImplementation() == ObjCPropertyDecl::Optional)

        continue;

      HasAtleastOneRequiredProperty = true;

      DeclContext::lookup_result R = IDecl->lookup(Property->getDeclName());

      if (R.empty()) {

        // Relax the rule and look into class's implementation for a synthesize

        // or dynamic declaration. Class is implementing a property coming from

        // another protocol. This still makes the target protocol as conforming.

        if (!ImpDecl->FindPropertyImplDecl(

                                  Property->getDeclName().getAsIdentifierInfo(),

                                  Property->getQueryKind()))

          return false;

      } else 
if (auto *4
ClassProperty4
 = R.find_first<ObjCPropertyDecl>()) {

        if ((ClassProperty->getPropertyAttributes() !=

             Property->getPropertyAttributes()) ||

            !Ctx.hasSameType(ClassProperty->getType(), Property->getType()))

          return false;

      } else

        return false;

    }

  // At this point, all required properties in this protocol conform to those

  // declared in the class.

  // Check that class implements the required methods of the protocol too.

  bool HasAtleastOneRequiredMethod = false;

  if (const ObjCProtocolDecl *PDecl = Protocol->getDefinition()) {

    if (PDecl->meth_begin() == PDecl->meth_end())

      return HasAtleastOneRequiredProperty;

    
for (const auto *MD : PDecl->methods())19
 {

      if (MD->isImplicit())

        continue;

      if (MD->getImplementationControl() == ObjCMethodDecl::Optional)

        continue;

      DeclContext::lookup_result R = ImpDecl->lookup(MD->getDeclName());

      if (R.empty())

        return false;

      bool match = false;

      HasAtleastOneRequiredMethod = true;

      for (NamedDecl *ND : R)

        if (ObjCMethodDecl *ImpMD = dyn_cast<ObjCMethodDecl>(ND))

          if (Ctx.ObjCMethodsAreEqual(MD, ImpMD)) {

            match = true;

            break;

          }

      if (!match)

        return false;

    }

  }

  return HasAtleastOneRequiredProperty || 
HasAtleastOneRequiredMethod10
;

}

static bool rewriteToObjCInterfaceDecl(const ObjCInterfaceDecl *IDecl,

                    llvm::SmallVectorImpl<ObjCProtocolDecl*> &ConformingProtocols,

                    const NSAPI &NS, edit::Commit &commit) {

  const ObjCList<ObjCProtocolDecl> &Protocols = IDecl->getReferencedProtocols();

  std::string ClassString;

  SourceLocation EndLoc =

  IDecl->getSuperClass() ? 
IDecl->getSuperClassLoc()4
 : 
IDecl->getLocation()2
;

  if (Protocols.empty()) {

    ClassString = '<';

    for (unsigned i = 0, e = ConformingProtocols.size(); i != e; 
i++3
) {

      ClassString += ConformingProtocols[i]->getNameAsString();

      if (i != (e-1))

        ClassString += ", ";

    }

    ClassString += "> ";

  }

  else {

    ClassString = ", ";

    for (unsigned i = 0, e = ConformingProtocols.size(); i != e; 
i++3
) {

      ClassString += ConformingProtocols[i]->getNameAsString();

      if (i != (e-1))

        ClassString += ", ";

    }

    ObjCInterfaceDecl::protocol_loc_iterator PL = IDecl->protocol_loc_end() - 1;

    EndLoc = *PL;

  }

  commit.insertAfterToken(EndLoc, ClassString);

  return true;

}

static StringRef GetUnsignedName(StringRef NSIntegerName) {

  StringRef UnsignedName = llvm::StringSwitch<StringRef>(NSIntegerName)

    .Case("int8_t", "uint8_t")

    .Case("int16_t", "uint16_t")

    .Case("int32_t", "uint32_t")

    .Case("NSInteger", "NSUInteger")

    .Case("int64_t", "uint64_t")

    .Default(NSIntegerName);

  return UnsignedName;

}

static bool rewriteToNSEnumDecl(const EnumDecl *EnumDcl,

                                const TypedefDecl *TypedefDcl,

                                const NSAPI &NS, edit::Commit &commit,

                                StringRef NSIntegerName,

                                bool NSOptions) {

  std::string ClassString;

  if (NSOptions) {

    ClassString = "typedef NS_OPTIONS(";

    ClassString += GetUnsignedName(NSIntegerName);

  }

  else {

    ClassString = "typedef NS_ENUM(";

    ClassString += NSIntegerName;

  }

  ClassString += ", ";

  ClassString += TypedefDcl->getIdentifier()->getName();

  ClassString += ')';

  SourceRange R(EnumDcl->getBeginLoc(), EnumDcl->getBeginLoc());

  commit.replace(R, ClassString);

  SourceLocation EndOfEnumDclLoc = EnumDcl->getEndLoc();

  EndOfEnumDclLoc = trans::findSemiAfterLocation(EndOfEnumDclLoc,

                                                 NS.getASTContext(), /*IsDecl*/true);

  if (EndOfEnumDclLoc.isValid()) {

    SourceRange EnumDclRange(EnumDcl->getBeginLoc(), EndOfEnumDclLoc);

    commit.insertFromRange(TypedefDcl->getBeginLoc(), EnumDclRange);

  }

  else

    return false;

  SourceLocation EndTypedefDclLoc = TypedefDcl->getEndLoc();

  EndTypedefDclLoc = trans::findSemiAfterLocation(EndTypedefDclLoc,

                                                 NS.getASTContext(), /*IsDecl*/true);

  if (EndTypedefDclLoc.isValid()) {

    SourceRange TDRange(TypedefDcl->getBeginLoc(), EndTypedefDclLoc);

    commit.remove(TDRange);

  }

  else

    return false;

  EndOfEnumDclLoc =

      trans::findLocationAfterSemi(EnumDcl->getEndLoc(), NS.getASTContext(),

                                   /*IsDecl*/ true);

  if (EndOfEnumDclLoc.isValid()) {

    SourceLocation BeginOfEnumDclLoc = EnumDcl->getBeginLoc();

    // FIXME. This assumes that enum decl; is immediately preceded by eoln.

    // It is trying to remove the enum decl. lines entirely.

    BeginOfEnumDclLoc = BeginOfEnumDclLoc.getLocWithOffset(-1);

    commit.remove(SourceRange(BeginOfEnumDclLoc, EndOfEnumDclLoc));

    return true;

  }

  return false;

}

static void rewriteToNSMacroDecl(ASTContext &Ctx,

                                 const EnumDecl *EnumDcl,

                                const TypedefDecl *TypedefDcl,

                                const NSAPI &NS, edit::Commit &commit,

                                 bool IsNSIntegerType) {

  QualType DesignatedEnumType = EnumDcl->getIntegerType();

  assert(!DesignatedEnumType.isNull()

         && "rewriteToNSMacroDecl - underlying enum type is null");

  PrintingPolicy Policy(Ctx.getPrintingPolicy());

  std::string TypeString = DesignatedEnumType.getAsString(Policy);

  std::string ClassString = IsNSIntegerType ? 
"NS_ENUM("23
 : 
"NS_OPTIONS("5
;

  ClassString += TypeString;

  ClassString += ", ";

  ClassString += TypedefDcl->getIdentifier()->getName();

  ClassString += ") ";

  SourceLocation EndLoc = EnumDcl->getBraceRange().getBegin();

  if (EndLoc.isInvalid())

    return;

  CharSourceRange R =

      CharSourceRange::getCharRange(EnumDcl->getBeginLoc(), EndLoc);

  commit.replace(R, ClassString);

  // This is to remove spaces between '}' and typedef name.

  SourceLocation StartTypedefLoc = EnumDcl->getEndLoc();

  StartTypedefLoc = StartTypedefLoc.getLocWithOffset(+1);

  SourceLocation EndTypedefLoc = TypedefDcl->getEndLoc();

  commit.remove(SourceRange(StartTypedefLoc, EndTypedefLoc));

}

static bool UseNSOptionsMacro(Preprocessor &PP, ASTContext &Ctx,

                              const EnumDecl *EnumDcl) {

  bool PowerOfTwo = true;

  bool AllHexdecimalEnumerator = true;

  uint64_t MaxPowerOfTwoVal = 0;

  for (auto Enumerator : EnumDcl->enumerators()) {

    const Expr *InitExpr = Enumerator->getInitExpr();

    if (!InitExpr) {

      PowerOfTwo = false;

      AllHexdecimalEnumerator = false;

      continue;

    }

    InitExpr = InitExpr->IgnoreParenCasts();

    if (const BinaryOperator *BO = dyn_cast<BinaryOperator>(InitExpr))

      if (BO->isShiftOp() || 
BO->isBitwiseOp()1
)

        return true;

    uint64_t EnumVal = Enumerator->getInitVal().getZExtValue();

    if (PowerOfTwo && 
EnumVal85
) {

      if (!llvm::isPowerOf2_64(EnumVal))

        PowerOfTwo = false;

      else if (EnumVal > MaxPowerOfTwoVal)

        MaxPowerOfTwoVal = EnumVal;

    }

    if (AllHexdecimalEnumerator && 
EnumVal78
) {

      bool FoundHexdecimalEnumerator = false;

      SourceLocation EndLoc = Enumerator->getEndLoc();

      Token Tok;

      if (!PP.getRawToken(EndLoc, Tok, /*IgnoreWhiteSpace=*/true))

        if (Tok.isLiteral() && 
Tok.getLength() > 258
) {

          if (const char *StringLit = Tok.getLiteralData())

            FoundHexdecimalEnumerator =

              (StringLit[0] == '0' && 
(toLowercase(StringLit[1]) == 'x')41
);

        }

      if (!FoundHexdecimalEnumerator)

        AllHexdecimalEnumerator = false;

    }

  }

  return AllHexdecimalEnumerator || 
(35
PowerOfTwo35
 && 
(MaxPowerOfTwoVal > 2)11
);

}

void ObjCMigrateASTConsumer::migrateProtocolConformance(ASTContext &Ctx,

                                            const ObjCImplementationDecl *ImpDecl) {

  const ObjCInterfaceDecl *IDecl = ImpDecl->getClassInterface();

  if (!IDecl || ObjCProtocolDecls.empty() || IDecl->isDeprecated())

    return;

  // Find all implicit conforming protocols for this class

  // and make them explicit.

  llvm::SmallPtrSet<ObjCProtocolDecl *, 8> ExplicitProtocols;

  Ctx.CollectInheritedProtocols(IDecl, ExplicitProtocols);

  llvm::SmallVector<ObjCProtocolDecl *, 8> PotentialImplicitProtocols;

  for (ObjCProtocolDecl *ProtDecl : ObjCProtocolDecls)

    if (!ExplicitProtocols.count(ProtDecl))

      PotentialImplicitProtocols.push_back(ProtDecl);

  if (PotentialImplicitProtocols.empty())

    return;

  // go through list of non-optional methods and properties in each protocol

  // in the PotentialImplicitProtocols list. If class implements every one of the

  // methods and properties, then this class conforms to this protocol.

  llvm::SmallVector<ObjCProtocolDecl*, 8> ConformingProtocols;

  for (unsigned i = 0, e = PotentialImplicitProtocols.size(); i != e; 
i++40
)

    if (ClassImplementsAllMethodsAndProperties(Ctx, ImpDecl, IDecl,

                                              PotentialImplicitProtocols[i]))

      ConformingProtocols.push_back(PotentialImplicitProtocols[i]);

  if (ConformingProtocols.empty())

    return;

  // Further reduce number of conforming protocols. If protocol P1 is in the list

  // protocol P2 (P2<P1>), No need to include P1.

  llvm::SmallVector<ObjCProtocolDecl*, 8> MinimalConformingProtocols;

  for (unsigned i = 0, e = ConformingProtocols.size(); i != e; 
i++9
) {

    bool DropIt = false;

    ObjCProtocolDecl *TargetPDecl = ConformingProtocols[i];

    for (unsigned i1 = 0, e1 = ConformingProtocols.size(); i1 != e1; 
i1++16
) {

      ObjCProtocolDecl *PDecl = ConformingProtocols[i1];

      if (PDecl == TargetPDecl)

        continue;

      if (PDecl->lookupProtocolNamed(

            TargetPDecl->getDeclName().getAsIdentifierInfo())) {

        DropIt = true;

        break;

      }

    }

    if (!DropIt)

      MinimalConformingProtocols.push_back(TargetPDecl);

  }

  if (MinimalConformingProtocols.empty())

    return;

  edit::Commit commit(*Editor);

  rewriteToObjCInterfaceDecl(IDecl, MinimalConformingProtocols,

                             *NSAPIObj, commit);

  Editor->commit(commit);

}

void ObjCMigrateASTConsumer::CacheObjCNSIntegerTypedefed(

                                          const TypedefDecl *TypedefDcl) {

  QualType qt = TypedefDcl->getTypeSourceInfo()->getType();

  if (NSAPIObj->isObjCNSIntegerType(qt))

    NSIntegerTypedefed = TypedefDcl;

  else if (NSAPIObj->isObjCNSUIntegerType(qt))

    NSUIntegerTypedefed = TypedefDcl;

}

bool ObjCMigrateASTConsumer::migrateNSEnumDecl(ASTContext &Ctx,

                                           const EnumDecl *EnumDcl,

                                           const TypedefDecl *TypedefDcl) {

  if (!EnumDcl->isCompleteDefinition() || 
EnumDcl->getIdentifier()59
 ||

      
EnumDcl->isDeprecated()56
)

    return false;

  if (!TypedefDcl) {

    if (NSIntegerTypedefed) {

      TypedefDcl = NSIntegerTypedefed;

      NSIntegerTypedefed = nullptr;

    }

    else if (NSUIntegerTypedefed) {

      TypedefDcl = NSUIntegerTypedefed;

      NSUIntegerTypedefed = nullptr;

    }

    else

      return false;

    FileID FileIdOfTypedefDcl =

      PP.getSourceManager().getFileID(TypedefDcl->getLocation());

    FileID FileIdOfEnumDcl =

      PP.getSourceManager().getFileID(EnumDcl->getLocation());

    if (FileIdOfTypedefDcl != FileIdOfEnumDcl)

      return false;

  }

  if (TypedefDcl->isDeprecated())

    return false;

  QualType qt = TypedefDcl->getTypeSourceInfo()->getType();

  StringRef NSIntegerName = NSAPIObj->GetNSIntegralKind(qt);

  if (NSIntegerName.empty()) {

    // Also check for typedef enum {...} TD;

    if (const EnumType *EnumTy = qt->getAs<EnumType>()) {

      if (EnumTy->getDecl() == EnumDcl) {

        bool NSOptions = UseNSOptionsMacro(PP, Ctx, EnumDcl);

        if (!InsertFoundation(Ctx, TypedefDcl->getBeginLoc()))

          return false;

        edit::Commit commit(*Editor);

        rewriteToNSMacroDecl(Ctx, EnumDcl, TypedefDcl, *NSAPIObj, commit, !NSOptions);

        Editor->commit(commit);

        return true;

      }

    }

    return false;

  }

  // We may still use NS_OPTIONS based on what we find in the enumertor list.

  bool NSOptions = UseNSOptionsMacro(PP, Ctx, EnumDcl);

  if (!InsertFoundation(Ctx, TypedefDcl->getBeginLoc()))

    return false;

  edit::Commit commit(*Editor);

  bool Res = rewriteToNSEnumDecl(EnumDcl, TypedefDcl, *NSAPIObj,

                                 commit, NSIntegerName, NSOptions);

  Editor->commit(commit);

  return Res;

}

static void ReplaceWithInstancetype(ASTContext &Ctx,

                                    const ObjCMigrateASTConsumer &ASTC,

                                    ObjCMethodDecl *OM) {

  if (OM->getReturnType() == Ctx.getObjCInstanceType())

    return; // already has instancetype.

  SourceRange R;

  std::string ClassString;

  if (TypeSourceInfo *TSInfo = OM->getReturnTypeSourceInfo()) {

    TypeLoc TL = TSInfo->getTypeLoc();

    R = SourceRange(TL.getBeginLoc(), TL.getEndLoc());

    ClassString = "instancetype";

  }

  else {

    R = SourceRange(OM->getBeginLoc(), OM->getBeginLoc());

    ClassString = OM->isInstanceMethod() ? 
'-'0
 : '+';

    ClassString += " (instancetype)";

  }

  edit::Commit commit(*ASTC.Editor);

  commit.replace(R, ClassString);

  ASTC.Editor->commit(commit);

}

static void ReplaceWithClasstype(const ObjCMigrateASTConsumer &ASTC,

                                    ObjCMethodDecl *OM) {

  ObjCInterfaceDecl *IDecl = OM->getClassInterface();

  SourceRange R;

  std::string ClassString;

  if (TypeSourceInfo *TSInfo = OM->getReturnTypeSourceInfo()) {

    TypeLoc TL = TSInfo->getTypeLoc();

    R = SourceRange(TL.getBeginLoc(), TL.getEndLoc()); {

      ClassString = std::string(IDecl->getName());

      ClassString += "*";

    }

  }

  else {

    R = SourceRange(OM->getBeginLoc(), OM->getBeginLoc());

    ClassString = "+ (";

    ClassString += IDecl->getName(); ClassString += "*)";

  }

  edit::Commit commit(*ASTC.Editor);

  commit.replace(R, ClassString);

  ASTC.Editor->commit(commit);

}

void ObjCMigrateASTConsumer::migrateMethodInstanceType(ASTContext &Ctx,

                                                       ObjCContainerDecl *CDecl,

                                                       ObjCMethodDecl *OM) {

  ObjCInstanceTypeFamily OIT_Family =

    Selector::getInstTypeMethodFamily(OM->getSelector());

  std::string ClassName;

  switch (OIT_Family) {

    case OIT_None:

      migrateFactoryMethod(Ctx, CDecl, OM);

      return;

    case OIT_Array:

      ClassName = "NSArray";

      break;

    case OIT_Dictionary:

      ClassName = "NSDictionary";