clang  3.8.0
ObjCMT.cpp
Go to the documentation of this file.
1 //===--- ObjCMT.cpp - ObjC Migrate Tool -----------------------------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "Transforms.h"
11 #include "clang/ARCMigrate/ARCMT.h"
13 #include "clang/AST/ASTConsumer.h"
14 #include "clang/AST/ASTContext.h"
15 #include "clang/AST/Attr.h"
16 #include "clang/AST/NSAPI.h"
17 #include "clang/AST/ParentMap.h"
21 #include "clang/Edit/Commit.h"
24 #include "clang/Edit/Rewriters.h"
28 #include "clang/Lex/Preprocessor.h"
31 #include "llvm/ADT/SmallString.h"
32 #include "llvm/ADT/StringSet.h"
33 #include "llvm/Support/Path.h"
34 #include "llvm/Support/SourceMgr.h"
35 #include "llvm/Support/YAMLParser.h"
36 
37 using namespace clang;
38 using namespace arcmt;
39 using namespace ento::objc_retain;
40 
41 namespace {
42 
43 class ObjCMigrateASTConsumer : public ASTConsumer {
44  enum CF_BRIDGING_KIND {
45  CF_BRIDGING_NONE,
46  CF_BRIDGING_ENABLE,
47  CF_BRIDGING_MAY_INCLUDE
48  };
49 
50  void migrateDecl(Decl *D);
51  void migrateObjCContainerDecl(ASTContext &Ctx, ObjCContainerDecl *D);
52  void migrateProtocolConformance(ASTContext &Ctx,
53  const ObjCImplementationDecl *ImpDecl);
54  void CacheObjCNSIntegerTypedefed(const TypedefDecl *TypedefDcl);
55  bool migrateNSEnumDecl(ASTContext &Ctx, const EnumDecl *EnumDcl,
56  const TypedefDecl *TypedefDcl);
57  void migrateAllMethodInstaceType(ASTContext &Ctx, ObjCContainerDecl *CDecl);
58  void migrateMethodInstanceType(ASTContext &Ctx, ObjCContainerDecl *CDecl,
59  ObjCMethodDecl *OM);
60  bool migrateProperty(ASTContext &Ctx, ObjCContainerDecl *D, ObjCMethodDecl *OM);
61  void migrateNsReturnsInnerPointer(ASTContext &Ctx, ObjCMethodDecl *OM);
62  void migratePropertyNsReturnsInnerPointer(ASTContext &Ctx, ObjCPropertyDecl *P);
63  void migrateFactoryMethod(ASTContext &Ctx, ObjCContainerDecl *CDecl,
64  ObjCMethodDecl *OM,
65  ObjCInstanceTypeFamily OIT_Family = OIT_None);
66 
67  void migrateCFAnnotation(ASTContext &Ctx, const Decl *Decl);
68  void AddCFAnnotations(ASTContext &Ctx, const CallEffects &CE,
69  const FunctionDecl *FuncDecl, bool ResultAnnotated);
70  void AddCFAnnotations(ASTContext &Ctx, const CallEffects &CE,
71  const ObjCMethodDecl *MethodDecl, bool ResultAnnotated);
72 
73  void AnnotateImplicitBridging(ASTContext &Ctx);
74 
75  CF_BRIDGING_KIND migrateAddFunctionAnnotation(ASTContext &Ctx,
76  const FunctionDecl *FuncDecl);
77 
78  void migrateARCSafeAnnotation(ASTContext &Ctx, ObjCContainerDecl *CDecl);
79 
80  void migrateAddMethodAnnotation(ASTContext &Ctx,
81  const ObjCMethodDecl *MethodDecl);
82 
83  void inferDesignatedInitializers(ASTContext &Ctx,
84  const ObjCImplementationDecl *ImplD);
85 
86  bool InsertFoundation(ASTContext &Ctx, SourceLocation Loc);
87 
88 public:
89  std::string MigrateDir;
90  unsigned ASTMigrateActions;
91  FileID FileId;
92  const TypedefDecl *NSIntegerTypedefed;
93  const TypedefDecl *NSUIntegerTypedefed;
94  std::unique_ptr<NSAPI> NSAPIObj;
95  std::unique_ptr<edit::EditedSource> Editor;
96  FileRemapper &Remapper;
97  FileManager &FileMgr;
98  const PPConditionalDirectiveRecord *PPRec;
99  Preprocessor &PP;
100  bool IsOutputFile;
101  bool FoundationIncluded;
102  llvm::SmallPtrSet<ObjCProtocolDecl *, 32> ObjCProtocolDecls;
103  llvm::SmallVector<const Decl *, 8> CFFunctionIBCandidates;
104  llvm::StringSet<> WhiteListFilenames;
105 
106  ObjCMigrateASTConsumer(StringRef migrateDir,
107  unsigned astMigrateActions,
108  FileRemapper &remapper,
109  FileManager &fileMgr,
110  const PPConditionalDirectiveRecord *PPRec,
111  Preprocessor &PP,
112  bool isOutputFile,
113  ArrayRef<std::string> WhiteList)
114  : MigrateDir(migrateDir),
115  ASTMigrateActions(astMigrateActions),
116  NSIntegerTypedefed(nullptr), NSUIntegerTypedefed(nullptr),
117  Remapper(remapper), FileMgr(fileMgr), PPRec(PPRec), PP(PP),
118  IsOutputFile(isOutputFile),
119  FoundationIncluded(false){
120 
121  // FIXME: StringSet should have insert(iter, iter) to use here.
122  for (const std::string &Val : WhiteList)
123  WhiteListFilenames.insert(Val);
124  }
125 
126 protected:
127  void Initialize(ASTContext &Context) override {
128  NSAPIObj.reset(new NSAPI(Context));
129  Editor.reset(new edit::EditedSource(Context.getSourceManager(),
130  Context.getLangOpts(),
131  PPRec));
132  }
133 
134  bool HandleTopLevelDecl(DeclGroupRef DG) override {
135  for (DeclGroupRef::iterator I = DG.begin(), E = DG.end(); I != E; ++I)
136  migrateDecl(*I);
137  return true;
138  }
139  void HandleInterestingDecl(DeclGroupRef DG) override {
140  // Ignore decls from the PCH.
141  }
142  void HandleTopLevelDeclInObjCContainer(DeclGroupRef DG) override {
143  ObjCMigrateASTConsumer::HandleTopLevelDecl(DG);
144  }
145 
146  void HandleTranslationUnit(ASTContext &Ctx) override;
147 
148  bool canModifyFile(StringRef Path) {
149  if (WhiteListFilenames.empty())
150  return true;
151  return WhiteListFilenames.find(llvm::sys::path::filename(Path))
152  != WhiteListFilenames.end();
153  }
154  bool canModifyFile(const FileEntry *FE) {
155  if (!FE)
156  return false;
157  return canModifyFile(FE->getName());
158  }
159  bool canModifyFile(FileID FID) {
160  if (FID.isInvalid())
161  return false;
162  return canModifyFile(PP.getSourceManager().getFileEntryForID(FID));
163  }
164 
165  bool canModify(const Decl *D) {
166  if (!D)
167  return false;
168  if (const ObjCCategoryImplDecl *CatImpl = dyn_cast<ObjCCategoryImplDecl>(D))
169  return canModify(CatImpl->getCategoryDecl());
170  if (const ObjCImplementationDecl *Impl = dyn_cast<ObjCImplementationDecl>(D))
171  return canModify(Impl->getClassInterface());
172  if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
173  return canModify(cast<Decl>(MD->getDeclContext()));
174 
175  FileID FID = PP.getSourceManager().getFileID(D->getLocation());
176  return canModifyFile(FID);
177  }
178 };
179 
180 }
181 
183  StringRef migrateDir,
184  unsigned migrateAction)
185  : WrapperFrontendAction(WrappedAction), MigrateDir(migrateDir),
186  ObjCMigAction(migrateAction),
187  CompInst(nullptr) {
188  if (MigrateDir.empty())
189  MigrateDir = "."; // user current directory if none is given.
190 }
191 
192 std::unique_ptr<ASTConsumer>
195  PPRec = new PPConditionalDirectiveRecord(CompInst->getSourceManager());
196  CI.getPreprocessor().addPPCallbacks(std::unique_ptr<PPCallbacks>(PPRec));
197  std::vector<std::unique_ptr<ASTConsumer>> Consumers;
198  Consumers.push_back(WrapperFrontendAction::CreateASTConsumer(CI, InFile));
199  Consumers.push_back(llvm::make_unique<ObjCMigrateASTConsumer>(
200  MigrateDir, ObjCMigAction, Remapper, CompInst->getFileManager(), PPRec,
201  CompInst->getPreprocessor(), false, None));
202  return llvm::make_unique<MultiplexConsumer>(std::move(Consumers));
203 }
204 
206  Remapper.initFromDisk(MigrateDir, CI.getDiagnostics(),
207  /*ignoreIfFilesChanges=*/true);
208  CompInst = &CI;
210  return true;
211 }
212 
213 namespace {
214  // FIXME. This duplicates one in RewriteObjCFoundationAPI.cpp
215  bool subscriptOperatorNeedsParens(const Expr *FullExpr) {
216  const Expr* Expr = FullExpr->IgnoreImpCasts();
217  return !(isa<ArraySubscriptExpr>(Expr) || isa<CallExpr>(Expr) ||
218  isa<DeclRefExpr>(Expr) || isa<CXXNamedCastExpr>(Expr) ||
219  isa<CXXConstructExpr>(Expr) || isa<CXXThisExpr>(Expr) ||
220  isa<CXXTypeidExpr>(Expr) ||
221  isa<CXXUnresolvedConstructExpr>(Expr) ||
222  isa<ObjCMessageExpr>(Expr) || isa<ObjCPropertyRefExpr>(Expr) ||
223  isa<ObjCProtocolExpr>(Expr) || isa<MemberExpr>(Expr) ||
224  isa<ObjCIvarRefExpr>(Expr) || isa<ParenExpr>(FullExpr) ||
225  isa<ParenListExpr>(Expr) || isa<SizeOfPackExpr>(Expr));
226  }
227 
228  /// \brief - Rewrite message expression for Objective-C setter and getters into
229  /// property-dot syntax.
230  bool rewriteToPropertyDotSyntax(const ObjCMessageExpr *Msg,
231  Preprocessor &PP,
232  const NSAPI &NS, edit::Commit &commit,
233  const ParentMap *PMap) {
234  if (!Msg || Msg->isImplicit() ||
237  return false;
238  if (const Expr *Receiver = Msg->getInstanceReceiver())
239  if (Receiver->getType()->isObjCBuiltinType())
240  return false;
241 
242  const ObjCMethodDecl *Method = Msg->getMethodDecl();
243  if (!Method)
244  return false;
245  if (!Method->isPropertyAccessor())
246  return false;
247 
248  const ObjCPropertyDecl *Prop = Method->findPropertyDecl();
249  if (!Prop)
250  return false;
251 
252  SourceRange MsgRange = Msg->getSourceRange();
253  bool ReceiverIsSuper =
255  // for 'super' receiver is nullptr.
256  const Expr *receiver = Msg->getInstanceReceiver();
257  bool NeedsParen =
258  ReceiverIsSuper ? false : subscriptOperatorNeedsParens(receiver);
259  bool IsGetter = (Msg->getNumArgs() == 0);
260  if (IsGetter) {
261  // Find space location range between receiver expression and getter method.
262  SourceLocation BegLoc =
263  ReceiverIsSuper ? Msg->getSuperLoc() : receiver->getLocEnd();
264  BegLoc = PP.getLocForEndOfToken(BegLoc);
265  SourceLocation EndLoc = Msg->getSelectorLoc(0);
266  SourceRange SpaceRange(BegLoc, EndLoc);
267  std::string PropertyDotString;
268  // rewrite getter method expression into: receiver.property or
269  // (receiver).property
270  if (NeedsParen) {
271  commit.insertBefore(receiver->getLocStart(), "(");
272  PropertyDotString = ").";
273  }
274  else
275  PropertyDotString = ".";
276  PropertyDotString += Prop->getName();
277  commit.replace(SpaceRange, PropertyDotString);
278 
279  // remove '[' ']'
280  commit.replace(SourceRange(MsgRange.getBegin(), MsgRange.getBegin()), "");
281  commit.replace(SourceRange(MsgRange.getEnd(), MsgRange.getEnd()), "");
282  } else {
283  if (NeedsParen)
284  commit.insertWrap("(", receiver->getSourceRange(), ")");
285  std::string PropertyDotString = ".";
286  PropertyDotString += Prop->getName();
287  PropertyDotString += " =";
288  const Expr*const* Args = Msg->getArgs();
289  const Expr *RHS = Args[0];
290  if (!RHS)
291  return false;
292  SourceLocation BegLoc =
293  ReceiverIsSuper ? Msg->getSuperLoc() : receiver->getLocEnd();
294  BegLoc = PP.getLocForEndOfToken(BegLoc);
295  SourceLocation EndLoc = RHS->getLocStart();
296  EndLoc = EndLoc.getLocWithOffset(-1);
297  const char *colon = PP.getSourceManager().getCharacterData(EndLoc);
298  // Add a space after '=' if there is no space between RHS and '='
299  if (colon && colon[0] == ':')
300  PropertyDotString += " ";
301  SourceRange Range(BegLoc, EndLoc);
302  commit.replace(Range, PropertyDotString);
303  // remove '[' ']'
304  commit.replace(SourceRange(MsgRange.getBegin(), MsgRange.getBegin()), "");
305  commit.replace(SourceRange(MsgRange.getEnd(), MsgRange.getEnd()), "");
306  }
307  return true;
308  }
309 
310 
311 class ObjCMigrator : public RecursiveASTVisitor<ObjCMigrator> {
312  ObjCMigrateASTConsumer &Consumer;
313  ParentMap &PMap;
314 
315 public:
316  ObjCMigrator(ObjCMigrateASTConsumer &consumer, ParentMap &PMap)
317  : Consumer(consumer), PMap(PMap) { }
318 
319  bool shouldVisitTemplateInstantiations() const { return false; }
320  bool shouldWalkTypesOfTypeLocs() const { return false; }
321 
322  bool VisitObjCMessageExpr(ObjCMessageExpr *E) {
323  if (Consumer.ASTMigrateActions & FrontendOptions::ObjCMT_Literals) {
324  edit::Commit commit(*Consumer.Editor);
325  edit::rewriteToObjCLiteralSyntax(E, *Consumer.NSAPIObj, commit, &PMap);
326  Consumer.Editor->commit(commit);
327  }
328 
329  if (Consumer.ASTMigrateActions & FrontendOptions::ObjCMT_Subscripting) {
330  edit::Commit commit(*Consumer.Editor);
331  edit::rewriteToObjCSubscriptSyntax(E, *Consumer.NSAPIObj, commit);
332  Consumer.Editor->commit(commit);
333  }
334 
335  if (Consumer.ASTMigrateActions & FrontendOptions::ObjCMT_PropertyDotSyntax) {
336  edit::Commit commit(*Consumer.Editor);
337  rewriteToPropertyDotSyntax(E, Consumer.PP, *Consumer.NSAPIObj,
338  commit, &PMap);
339  Consumer.Editor->commit(commit);
340  }
341 
342  return true;
343  }
344 
345  bool TraverseObjCMessageExpr(ObjCMessageExpr *E) {
346  // Do depth first; we want to rewrite the subexpressions first so that if
347  // we have to move expressions we will move them already rewritten.
348  for (Stmt *SubStmt : E->children())
349  if (!TraverseStmt(SubStmt))
350  return false;
351 
352  return WalkUpFromObjCMessageExpr(E);
353  }
354 };
355 
356 class BodyMigrator : public RecursiveASTVisitor<BodyMigrator> {
357  ObjCMigrateASTConsumer &Consumer;
358  std::unique_ptr<ParentMap> PMap;
359 
360 public:
361  BodyMigrator(ObjCMigrateASTConsumer &consumer) : Consumer(consumer) { }
362 
363  bool shouldVisitTemplateInstantiations() const { return false; }
364  bool shouldWalkTypesOfTypeLocs() const { return false; }
365 
366  bool TraverseStmt(Stmt *S) {
367  PMap.reset(new ParentMap(S));
368  ObjCMigrator(Consumer, *PMap).TraverseStmt(S);
369  return true;
370  }
371 };
372 }
373 
374 void ObjCMigrateASTConsumer::migrateDecl(Decl *D) {
375  if (!D)
376  return;
377  if (isa<ObjCMethodDecl>(D))
378  return; // Wait for the ObjC container declaration.
379 
380  BodyMigrator(*this).TraverseDecl(D);
381 }
382 
383 static void append_attr(std::string &PropertyString, const char *attr,
384  bool &LParenAdded) {
385  if (!LParenAdded) {
386  PropertyString += "(";
387  LParenAdded = true;
388  }
389  else
390  PropertyString += ", ";
391  PropertyString += attr;
392 }
393 
394 static
395 void MigrateBlockOrFunctionPointerTypeVariable(std::string & PropertyString,
396  const std::string& TypeString,
397  const char *name) {
398  const char *argPtr = TypeString.c_str();
399  int paren = 0;
400  while (*argPtr) {
401  switch (*argPtr) {
402  case '(':
403  PropertyString += *argPtr;
404  paren++;
405  break;
406  case ')':
407  PropertyString += *argPtr;
408  paren--;
409  break;
410  case '^':
411  case '*':
412  PropertyString += (*argPtr);
413  if (paren == 1) {
414  PropertyString += name;
415  name = "";
416  }
417  break;
418  default:
419  PropertyString += *argPtr;
420  break;
421  }
422  argPtr++;
423  }
424 }
425 
426 static const char *PropertyMemoryAttribute(ASTContext &Context, QualType ArgType) {
427  Qualifiers::ObjCLifetime propertyLifetime = ArgType.getObjCLifetime();
428  bool RetainableObject = ArgType->isObjCRetainableType();
429  if (RetainableObject &&
430  (propertyLifetime == Qualifiers::OCL_Strong
431  || propertyLifetime == Qualifiers::OCL_None)) {
432  if (const ObjCObjectPointerType *ObjPtrTy =
433  ArgType->getAs<ObjCObjectPointerType>()) {
434  ObjCInterfaceDecl *IDecl = ObjPtrTy->getObjectType()->getInterface();
435  if (IDecl &&
436  IDecl->lookupNestedProtocol(&Context.Idents.get("NSCopying")))
437  return "copy";
438  else
439  return "strong";
440  }
441  else if (ArgType->isBlockPointerType())
442  return "copy";
443  } else if (propertyLifetime == Qualifiers::OCL_Weak)
444  // TODO. More precise determination of 'weak' attribute requires
445  // looking into setter's implementation for backing weak ivar.
446  return "weak";
447  else if (RetainableObject)
448  return ArgType->isBlockPointerType() ? "copy" : "strong";
449  return nullptr;
450 }
451 
452 static void rewriteToObjCProperty(const ObjCMethodDecl *Getter,
453  const ObjCMethodDecl *Setter,
454  const NSAPI &NS, edit::Commit &commit,
455  unsigned LengthOfPrefix,
456  bool Atomic, bool UseNsIosOnlyMacro,
457  bool AvailabilityArgsMatch) {
458  ASTContext &Context = NS.getASTContext();
459  bool LParenAdded = false;
460  std::string PropertyString = "@property ";
461  if (UseNsIosOnlyMacro && NS.isMacroDefined("NS_NONATOMIC_IOSONLY")) {
462  PropertyString += "(NS_NONATOMIC_IOSONLY";
463  LParenAdded = true;
464  } else if (!Atomic) {
465  PropertyString += "(nonatomic";
466  LParenAdded = true;
467  }
468 
469  std::string PropertyNameString = Getter->getNameAsString();
470  StringRef PropertyName(PropertyNameString);
471  if (LengthOfPrefix > 0) {
472  if (!LParenAdded) {
473  PropertyString += "(getter=";
474  LParenAdded = true;
475  }
476  else
477  PropertyString += ", getter=";
478  PropertyString += PropertyNameString;
479  }
480  // Property with no setter may be suggested as a 'readonly' property.
481  if (!Setter)
482  append_attr(PropertyString, "readonly", LParenAdded);
483 
484 
485  // Short circuit 'delegate' properties that contain the name "delegate" or
486  // "dataSource", or have exact name "target" to have 'assign' attribute.
487  if (PropertyName.equals("target") ||
488  (PropertyName.find("delegate") != StringRef::npos) ||
489  (PropertyName.find("dataSource") != StringRef::npos)) {
490  QualType QT = Getter->getReturnType();
491  if (!QT->isRealType())
492  append_attr(PropertyString, "assign", LParenAdded);
493  } else if (!Setter) {
494  QualType ResType = Context.getCanonicalType(Getter->getReturnType());
495  if (const char *MemoryManagementAttr = PropertyMemoryAttribute(Context, ResType))
496  append_attr(PropertyString, MemoryManagementAttr, LParenAdded);
497  } else {
498  const ParmVarDecl *argDecl = *Setter->param_begin();
499  QualType ArgType = Context.getCanonicalType(argDecl->getType());
500  if (const char *MemoryManagementAttr = PropertyMemoryAttribute(Context, ArgType))
501  append_attr(PropertyString, MemoryManagementAttr, LParenAdded);
502  }
503  if (LParenAdded)
504  PropertyString += ')';
505  QualType RT = Getter->getReturnType();
506  if (!isa<TypedefType>(RT)) {
507  // strip off any ARC lifetime qualifier.
508  QualType CanResultTy = Context.getCanonicalType(RT);
509  if (CanResultTy.getQualifiers().hasObjCLifetime()) {
510  Qualifiers Qs = CanResultTy.getQualifiers();
511  Qs.removeObjCLifetime();
512  RT = Context.getQualifiedType(CanResultTy.getUnqualifiedType(), Qs);
513  }
514  }
515  PropertyString += " ";
516  PrintingPolicy SubPolicy(Context.getPrintingPolicy());
517  SubPolicy.SuppressStrongLifetime = true;
518  SubPolicy.SuppressLifetimeQualifiers = true;
519  std::string TypeString = RT.getAsString(SubPolicy);
520  if (LengthOfPrefix > 0) {
521  // property name must strip off "is" and lower case the first character
522  // after that; e.g. isContinuous will become continuous.
523  StringRef PropertyNameStringRef(PropertyNameString);
524  PropertyNameStringRef = PropertyNameStringRef.drop_front(LengthOfPrefix);
525  PropertyNameString = PropertyNameStringRef;
526  bool NoLowering = (isUppercase(PropertyNameString[0]) &&
527  PropertyNameString.size() > 1 &&
528  isUppercase(PropertyNameString[1]));
529  if (!NoLowering)
530  PropertyNameString[0] = toLowercase(PropertyNameString[0]);
531  }
532  if (RT->isBlockPointerType() || RT->isFunctionPointerType())
534  TypeString,
535  PropertyNameString.c_str());
536  else {
537  char LastChar = TypeString[TypeString.size()-1];
538  PropertyString += TypeString;
539  if (LastChar != '*')
540  PropertyString += ' ';
541  PropertyString += PropertyNameString;
542  }
543  SourceLocation StartGetterSelectorLoc = Getter->getSelectorStartLoc();
544  Selector GetterSelector = Getter->getSelector();
545 
546  SourceLocation EndGetterSelectorLoc =
547  StartGetterSelectorLoc.getLocWithOffset(GetterSelector.getNameForSlot(0).size());
549  EndGetterSelectorLoc),
550  PropertyString);
551  if (Setter && AvailabilityArgsMatch) {
552  SourceLocation EndLoc = Setter->getDeclaratorEndLoc();
553  // Get location past ';'
554  EndLoc = EndLoc.getLocWithOffset(1);
555  SourceLocation BeginOfSetterDclLoc = Setter->getLocStart();
556  // FIXME. This assumes that setter decl; is immediately preceded by eoln.
557  // It is trying to remove the setter method decl. line entirely.
558  BeginOfSetterDclLoc = BeginOfSetterDclLoc.getLocWithOffset(-1);
559  commit.remove(SourceRange(BeginOfSetterDclLoc, EndLoc));
560  }
561 }
562 
564  if (ObjCCategoryDecl *CatDecl = dyn_cast<ObjCCategoryDecl>(D)) {
565  StringRef Name = CatDecl->getName();
566  return Name.endswith("Deprecated");
567  }
568  return false;
569 }
570 
571 void ObjCMigrateASTConsumer::migrateObjCContainerDecl(ASTContext &Ctx,
572  ObjCContainerDecl *D) {
574  return;
575 
576  for (auto *Method : D->methods()) {
577  if (Method->isDeprecated())
578  continue;
579  bool PropertyInferred = migrateProperty(Ctx, D, Method);
580  // If a property is inferred, do not attempt to attach NS_RETURNS_INNER_POINTER to
581  // the getter method as it ends up on the property itself which we don't want
582  // to do unless -objcmt-returns-innerpointer-property option is on.
583  if (!PropertyInferred ||
585  if (ASTMigrateActions & FrontendOptions::ObjCMT_Annotation)
586  migrateNsReturnsInnerPointer(Ctx, Method);
587  }
588  if (!(ASTMigrateActions & FrontendOptions::ObjCMT_ReturnsInnerPointerProperty))
589  return;
590 
591  for (auto *Prop : D->properties()) {
592  if ((ASTMigrateActions & FrontendOptions::ObjCMT_Annotation) &&
593  !Prop->isDeprecated())
594  migratePropertyNsReturnsInnerPointer(Ctx, Prop);
595  }
596 }
597 
598 static bool
600  const ObjCImplementationDecl *ImpDecl,
601  const ObjCInterfaceDecl *IDecl,
602  ObjCProtocolDecl *Protocol) {
603  // In auto-synthesis, protocol properties are not synthesized. So,
604  // a conforming protocol must have its required properties declared
605  // in class interface.
606  bool HasAtleastOneRequiredProperty = false;
607  if (const ObjCProtocolDecl *PDecl = Protocol->getDefinition())
608  for (const auto *Property : PDecl->properties()) {
609  if (Property->getPropertyImplementation() == ObjCPropertyDecl::Optional)
610  continue;
611  HasAtleastOneRequiredProperty = true;
612  DeclContext::lookup_result R = IDecl->lookup(Property->getDeclName());
613  if (R.size() == 0) {
614  // Relax the rule and look into class's implementation for a synthesize
615  // or dynamic declaration. Class is implementing a property coming from
616  // another protocol. This still makes the target protocol as conforming.
617  if (!ImpDecl->FindPropertyImplDecl(
618  Property->getDeclName().getAsIdentifierInfo()))
619  return false;
620  }
621  else if (ObjCPropertyDecl *ClassProperty = dyn_cast<ObjCPropertyDecl>(R[0])) {
622  if ((ClassProperty->getPropertyAttributes()
623  != Property->getPropertyAttributes()) ||
624  !Ctx.hasSameType(ClassProperty->getType(), Property->getType()))
625  return false;
626  }
627  else
628  return false;
629  }
630 
631  // At this point, all required properties in this protocol conform to those
632  // declared in the class.
633  // Check that class implements the required methods of the protocol too.
634  bool HasAtleastOneRequiredMethod = false;
635  if (const ObjCProtocolDecl *PDecl = Protocol->getDefinition()) {
636  if (PDecl->meth_begin() == PDecl->meth_end())
637  return HasAtleastOneRequiredProperty;
638  for (const auto *MD : PDecl->methods()) {
639  if (MD->isImplicit())
640  continue;
641  if (MD->getImplementationControl() == ObjCMethodDecl::Optional)
642  continue;
643  DeclContext::lookup_result R = ImpDecl->lookup(MD->getDeclName());
644  if (R.size() == 0)
645  return false;
646  bool match = false;
647  HasAtleastOneRequiredMethod = true;
648  for (unsigned I = 0, N = R.size(); I != N; ++I)
649  if (ObjCMethodDecl *ImpMD = dyn_cast<ObjCMethodDecl>(R[0]))
650  if (Ctx.ObjCMethodsAreEqual(MD, ImpMD)) {
651  match = true;
652  break;
653  }
654  if (!match)
655  return false;
656  }
657  }
658  return HasAtleastOneRequiredProperty || HasAtleastOneRequiredMethod;
659 }
660 
662  llvm::SmallVectorImpl<ObjCProtocolDecl*> &ConformingProtocols,
663  const NSAPI &NS, edit::Commit &commit) {
664  const ObjCList<ObjCProtocolDecl> &Protocols = IDecl->getReferencedProtocols();
665  std::string ClassString;
666  SourceLocation EndLoc =
667  IDecl->getSuperClass() ? IDecl->getSuperClassLoc() : IDecl->getLocation();
668 
669  if (Protocols.empty()) {
670  ClassString = '<';
671  for (unsigned i = 0, e = ConformingProtocols.size(); i != e; i++) {
672  ClassString += ConformingProtocols[i]->getNameAsString();
673  if (i != (e-1))
674  ClassString += ", ";
675  }
676  ClassString += "> ";
677  }
678  else {
679  ClassString = ", ";
680  for (unsigned i = 0, e = ConformingProtocols.size(); i != e; i++) {
681  ClassString += ConformingProtocols[i]->getNameAsString();
682  if (i != (e-1))
683  ClassString += ", ";
684  }
686  EndLoc = *PL;
687  }
688 
689  commit.insertAfterToken(EndLoc, ClassString);
690  return true;
691 }
692 
693 static StringRef GetUnsignedName(StringRef NSIntegerName) {
694  StringRef UnsignedName = llvm::StringSwitch<StringRef>(NSIntegerName)
695  .Case("int8_t", "uint8_t")
696  .Case("int16_t", "uint16_t")
697  .Case("int32_t", "uint32_t")
698  .Case("NSInteger", "NSUInteger")
699  .Case("int64_t", "uint64_t")
700  .Default(NSIntegerName);
701  return UnsignedName;
702 }
703 
704 static bool rewriteToNSEnumDecl(const EnumDecl *EnumDcl,
705  const TypedefDecl *TypedefDcl,
706  const NSAPI &NS, edit::Commit &commit,
707  StringRef NSIntegerName,
708  bool NSOptions) {
709  std::string ClassString;
710  if (NSOptions) {
711  ClassString = "typedef NS_OPTIONS(";
712  ClassString += GetUnsignedName(NSIntegerName);
713  }
714  else {
715  ClassString = "typedef NS_ENUM(";
716  ClassString += NSIntegerName;
717  }
718  ClassString += ", ";
719 
720  ClassString += TypedefDcl->getIdentifier()->getName();
721  ClassString += ')';
722  SourceRange R(EnumDcl->getLocStart(), EnumDcl->getLocStart());
723  commit.replace(R, ClassString);
724  SourceLocation EndOfEnumDclLoc = EnumDcl->getLocEnd();
725  EndOfEnumDclLoc = trans::findSemiAfterLocation(EndOfEnumDclLoc,
726  NS.getASTContext(), /*IsDecl*/true);
727  if (EndOfEnumDclLoc.isValid()) {
728  SourceRange EnumDclRange(EnumDcl->getLocStart(), EndOfEnumDclLoc);
729  commit.insertFromRange(TypedefDcl->getLocStart(), EnumDclRange);
730  }
731  else
732  return false;
733 
734  SourceLocation EndTypedefDclLoc = TypedefDcl->getLocEnd();
735  EndTypedefDclLoc = trans::findSemiAfterLocation(EndTypedefDclLoc,
736  NS.getASTContext(), /*IsDecl*/true);
737  if (EndTypedefDclLoc.isValid()) {
738  SourceRange TDRange(TypedefDcl->getLocStart(), EndTypedefDclLoc);
739  commit.remove(TDRange);
740  }
741  else
742  return false;
743 
744  EndOfEnumDclLoc = trans::findLocationAfterSemi(EnumDcl->getLocEnd(), NS.getASTContext(),
745  /*IsDecl*/true);
746  if (EndOfEnumDclLoc.isValid()) {
747  SourceLocation BeginOfEnumDclLoc = EnumDcl->getLocStart();
748  // FIXME. This assumes that enum decl; is immediately preceded by eoln.
749  // It is trying to remove the enum decl. lines entirely.
750  BeginOfEnumDclLoc = BeginOfEnumDclLoc.getLocWithOffset(-1);
751  commit.remove(SourceRange(BeginOfEnumDclLoc, EndOfEnumDclLoc));
752  return true;
753  }
754  return false;
755 }
756 
758  const EnumDecl *EnumDcl,
759  const TypedefDecl *TypedefDcl,
760  const NSAPI &NS, edit::Commit &commit,
761  bool IsNSIntegerType) {
762  QualType DesignatedEnumType = EnumDcl->getIntegerType();
763  assert(!DesignatedEnumType.isNull()
764  && "rewriteToNSMacroDecl - underlying enum type is null");
765 
766  PrintingPolicy Policy(Ctx.getPrintingPolicy());
767  std::string TypeString = DesignatedEnumType.getAsString(Policy);
768  std::string ClassString = IsNSIntegerType ? "NS_ENUM(" : "NS_OPTIONS(";
769  ClassString += TypeString;
770  ClassString += ", ";
771 
772  ClassString += TypedefDcl->getIdentifier()->getName();
773  ClassString += ')';
774  SourceLocation EndLoc;
775  if (EnumDcl->getIntegerTypeSourceInfo()) {
776  TypeSourceInfo *TSourceInfo = EnumDcl->getIntegerTypeSourceInfo();
777  TypeLoc TLoc = TSourceInfo->getTypeLoc();
778  EndLoc = TLoc.getLocEnd();
779  const char *lbrace = Ctx.getSourceManager().getCharacterData(EndLoc);
780  unsigned count = 0;
781  if (lbrace)
782  while (lbrace[count] != '{')
783  ++count;
784  if (count > 0)
785  EndLoc = EndLoc.getLocWithOffset(count-1);
786  }
787  else
788  EndLoc = EnumDcl->getLocStart();
789  SourceRange R(EnumDcl->getLocStart(), EndLoc);
790  commit.replace(R, ClassString);
791  // This is to remove spaces between '}' and typedef name.
792  SourceLocation StartTypedefLoc = EnumDcl->getLocEnd();
793  StartTypedefLoc = StartTypedefLoc.getLocWithOffset(+1);
794  SourceLocation EndTypedefLoc = TypedefDcl->getLocEnd();
795 
796  commit.remove(SourceRange(StartTypedefLoc, EndTypedefLoc));
797 }
798 
800  const EnumDecl *EnumDcl) {
801  bool PowerOfTwo = true;
802  bool AllHexdecimalEnumerator = true;
803  uint64_t MaxPowerOfTwoVal = 0;
804  for (auto Enumerator : EnumDcl->enumerators()) {
805  const Expr *InitExpr = Enumerator->getInitExpr();
806  if (!InitExpr) {
807  PowerOfTwo = false;
808  AllHexdecimalEnumerator = false;
809  continue;
810  }
811  InitExpr = InitExpr->IgnoreParenCasts();
812  if (const BinaryOperator *BO = dyn_cast<BinaryOperator>(InitExpr))
813  if (BO->isShiftOp() || BO->isBitwiseOp())
814  return true;
815 
816  uint64_t EnumVal = Enumerator->getInitVal().getZExtValue();
817  if (PowerOfTwo && EnumVal) {
818  if (!llvm::isPowerOf2_64(EnumVal))
819  PowerOfTwo = false;
820  else if (EnumVal > MaxPowerOfTwoVal)
821  MaxPowerOfTwoVal = EnumVal;
822  }
823  if (AllHexdecimalEnumerator && EnumVal) {
824  bool FoundHexdecimalEnumerator = false;
825  SourceLocation EndLoc = Enumerator->getLocEnd();
826  Token Tok;
827  if (!PP.getRawToken(EndLoc, Tok, /*IgnoreWhiteSpace=*/true))
828  if (Tok.isLiteral() && Tok.getLength() > 2) {
829  if (const char *StringLit = Tok.getLiteralData())
830  FoundHexdecimalEnumerator =
831  (StringLit[0] == '0' && (toLowercase(StringLit[1]) == 'x'));
832  }
833  if (!FoundHexdecimalEnumerator)
834  AllHexdecimalEnumerator = false;
835  }
836  }
837  return AllHexdecimalEnumerator || (PowerOfTwo && (MaxPowerOfTwoVal > 2));
838 }
839 
840 void ObjCMigrateASTConsumer::migrateProtocolConformance(ASTContext &Ctx,
841  const ObjCImplementationDecl *ImpDecl) {
842  const ObjCInterfaceDecl *IDecl = ImpDecl->getClassInterface();
843  if (!IDecl || ObjCProtocolDecls.empty() || IDecl->isDeprecated())
844  return;
845  // Find all implicit conforming protocols for this class
846  // and make them explicit.
847  llvm::SmallPtrSet<ObjCProtocolDecl *, 8> ExplicitProtocols;
848  Ctx.CollectInheritedProtocols(IDecl, ExplicitProtocols);
849  llvm::SmallVector<ObjCProtocolDecl *, 8> PotentialImplicitProtocols;
850 
851  for (ObjCProtocolDecl *ProtDecl : ObjCProtocolDecls)
852  if (!ExplicitProtocols.count(ProtDecl))
853  PotentialImplicitProtocols.push_back(ProtDecl);
854 
855  if (PotentialImplicitProtocols.empty())
856  return;
857 
858  // go through list of non-optional methods and properties in each protocol
859  // in the PotentialImplicitProtocols list. If class implements every one of the
860  // methods and properties, then this class conforms to this protocol.
861  llvm::SmallVector<ObjCProtocolDecl*, 8> ConformingProtocols;
862  for (unsigned i = 0, e = PotentialImplicitProtocols.size(); i != e; i++)
863  if (ClassImplementsAllMethodsAndProperties(Ctx, ImpDecl, IDecl,
864  PotentialImplicitProtocols[i]))
865  ConformingProtocols.push_back(PotentialImplicitProtocols[i]);
866 
867  if (ConformingProtocols.empty())
868  return;
869 
870  // Further reduce number of conforming protocols. If protocol P1 is in the list
871  // protocol P2 (P2<P1>), No need to include P1.
872  llvm::SmallVector<ObjCProtocolDecl*, 8> MinimalConformingProtocols;
873  for (unsigned i = 0, e = ConformingProtocols.size(); i != e; i++) {
874  bool DropIt = false;
875  ObjCProtocolDecl *TargetPDecl = ConformingProtocols[i];
876  for (unsigned i1 = 0, e1 = ConformingProtocols.size(); i1 != e1; i1++) {
877  ObjCProtocolDecl *PDecl = ConformingProtocols[i1];
878  if (PDecl == TargetPDecl)
879  continue;
880  if (PDecl->lookupProtocolNamed(
881  TargetPDecl->getDeclName().getAsIdentifierInfo())) {
882  DropIt = true;
883  break;
884  }
885  }
886  if (!DropIt)
887  MinimalConformingProtocols.push_back(TargetPDecl);
888  }
889  if (MinimalConformingProtocols.empty())
890  return;
891  edit::Commit commit(*Editor);
892  rewriteToObjCInterfaceDecl(IDecl, MinimalConformingProtocols,
893  *NSAPIObj, commit);
894  Editor->commit(commit);
895 }
896 
897 void ObjCMigrateASTConsumer::CacheObjCNSIntegerTypedefed(
898  const TypedefDecl *TypedefDcl) {
899 
900  QualType qt = TypedefDcl->getTypeSourceInfo()->getType();
901  if (NSAPIObj->isObjCNSIntegerType(qt))
902  NSIntegerTypedefed = TypedefDcl;
903  else if (NSAPIObj->isObjCNSUIntegerType(qt))
904  NSUIntegerTypedefed = TypedefDcl;
905 }
906 
907 bool ObjCMigrateASTConsumer::migrateNSEnumDecl(ASTContext &Ctx,
908  const EnumDecl *EnumDcl,
909  const TypedefDecl *TypedefDcl) {
910  if (!EnumDcl->isCompleteDefinition() || EnumDcl->getIdentifier() ||
911  EnumDcl->isDeprecated())
912  return false;
913  if (!TypedefDcl) {
914  if (NSIntegerTypedefed) {
915  TypedefDcl = NSIntegerTypedefed;
916  NSIntegerTypedefed = nullptr;
917  }
918  else if (NSUIntegerTypedefed) {
919  TypedefDcl = NSUIntegerTypedefed;
920  NSUIntegerTypedefed = nullptr;
921  }
922  else
923  return false;
924  FileID FileIdOfTypedefDcl =
925  PP.getSourceManager().getFileID(TypedefDcl->getLocation());
926  FileID FileIdOfEnumDcl =
927  PP.getSourceManager().getFileID(EnumDcl->getLocation());
928  if (FileIdOfTypedefDcl != FileIdOfEnumDcl)
929  return false;
930  }
931  if (TypedefDcl->isDeprecated())
932  return false;
933 
934  QualType qt = TypedefDcl->getTypeSourceInfo()->getType();
935  StringRef NSIntegerName = NSAPIObj->GetNSIntegralKind(qt);
936 
937  if (NSIntegerName.empty()) {
938  // Also check for typedef enum {...} TD;
939  if (const EnumType *EnumTy = qt->getAs<EnumType>()) {
940  if (EnumTy->getDecl() == EnumDcl) {
941  bool NSOptions = UseNSOptionsMacro(PP, Ctx, EnumDcl);
942  if (!InsertFoundation(Ctx, TypedefDcl->getLocStart()))
943  return false;
944  edit::Commit commit(*Editor);
945  rewriteToNSMacroDecl(Ctx, EnumDcl, TypedefDcl, *NSAPIObj, commit, !NSOptions);
946  Editor->commit(commit);
947  return true;
948  }
949  }
950  return false;
951  }
952 
953  // We may still use NS_OPTIONS based on what we find in the enumertor list.
954  bool NSOptions = UseNSOptionsMacro(PP, Ctx, EnumDcl);
955  if (!InsertFoundation(Ctx, TypedefDcl->getLocStart()))
956  return false;
957  edit::Commit commit(*Editor);
958  bool Res = rewriteToNSEnumDecl(EnumDcl, TypedefDcl, *NSAPIObj,
959  commit, NSIntegerName, NSOptions);
960  Editor->commit(commit);
961  return Res;
962 }
963 
965  const ObjCMigrateASTConsumer &ASTC,
966  ObjCMethodDecl *OM) {
967  if (OM->getReturnType() == Ctx.getObjCInstanceType())
968  return; // already has instancetype.
969 
970  SourceRange R;
971  std::string ClassString;
972  if (TypeSourceInfo *TSInfo = OM->getReturnTypeSourceInfo()) {
973  TypeLoc TL = TSInfo->getTypeLoc();
974  R = SourceRange(TL.getBeginLoc(), TL.getEndLoc());
975  ClassString = "instancetype";
976  }
977  else {
978  R = SourceRange(OM->getLocStart(), OM->getLocStart());
979  ClassString = OM->isInstanceMethod() ? '-' : '+';
980  ClassString += " (instancetype)";
981  }
982  edit::Commit commit(*ASTC.Editor);
983  commit.replace(R, ClassString);
984  ASTC.Editor->commit(commit);
985 }
986 
987 static void ReplaceWithClasstype(const ObjCMigrateASTConsumer &ASTC,
988  ObjCMethodDecl *OM) {
989  ObjCInterfaceDecl *IDecl = OM->getClassInterface();
990  SourceRange R;
991  std::string ClassString;
992  if (TypeSourceInfo *TSInfo = OM->getReturnTypeSourceInfo()) {
993  TypeLoc TL = TSInfo->getTypeLoc();
994  R = SourceRange(TL.getBeginLoc(), TL.getEndLoc()); {
995  ClassString = IDecl->getName();
996  ClassString += "*";
997  }
998  }
999  else {
1000  R = SourceRange(OM->getLocStart(), OM->getLocStart());
1001  ClassString = "+ (";
1002  ClassString += IDecl->getName(); ClassString += "*)";
1003  }
1004  edit::Commit commit(*ASTC.Editor);
1005  commit.replace(R, ClassString);
1006  ASTC.Editor->commit(commit);
1007 }
1008 
1009 void ObjCMigrateASTConsumer::migrateMethodInstanceType(ASTContext &Ctx,
1010  ObjCContainerDecl *CDecl,
1011  ObjCMethodDecl *OM) {
1012  ObjCInstanceTypeFamily OIT_Family =
1014 
1015  std::string ClassName;
1016  switch (OIT_Family) {
1017  case OIT_None:
1018  migrateFactoryMethod(Ctx, CDecl, OM);
1019  return;
1020  case OIT_Array:
1021  ClassName = "NSArray";
1022  break;
1023  case OIT_Dictionary:
1024  ClassName = "NSDictionary";
1025  break;
1026  case OIT_Singleton:
1027  migrateFactoryMethod(Ctx, CDecl, OM, OIT_Singleton);
1028  return;
1029  case OIT_Init:
1030  if (OM->getReturnType()->isObjCIdType())
1031  ReplaceWithInstancetype(Ctx, *this, OM);
1032  return;
1033  case OIT_ReturnsSelf:
1034  migrateFactoryMethod(Ctx, CDecl, OM, OIT_ReturnsSelf);
1035  return;
1036  }
1037  if (!OM->getReturnType()->isObjCIdType())
1038  return;
1039 
1040  ObjCInterfaceDecl *IDecl = dyn_cast<ObjCInterfaceDecl>(CDecl);
1041  if (!IDecl) {
1042  if (ObjCCategoryDecl *CatDecl = dyn_cast<ObjCCategoryDecl>(CDecl))
1043  IDecl = CatDecl->getClassInterface();
1044  else if (ObjCImplDecl *ImpDecl = dyn_cast<ObjCImplDecl>(CDecl))
1045  IDecl = ImpDecl->getClassInterface();
1046  }
1047  if (!IDecl ||
1048  !IDecl->lookupInheritedClass(&Ctx.Idents.get(ClassName))) {
1049  migrateFactoryMethod(Ctx, CDecl, OM);
1050  return;
1051  }
1052  ReplaceWithInstancetype(Ctx, *this, OM);
1053 }
1054 
1056  if (!T->isAnyPointerType())
1057  return false;
1058  if (T->isObjCObjectPointerType() || T->isObjCBuiltinType() ||
1061  return false;
1062  // Also, typedef-of-pointer-to-incomplete-struct is something that we assume
1063  // is not an innter pointer type.
1064  QualType OrigT = T;
1065  while (const TypedefType *TD = dyn_cast<TypedefType>(T.getTypePtr()))
1066  T = TD->getDecl()->getUnderlyingType();
1067  if (OrigT == T || !T->isPointerType())
1068  return true;
1069  const PointerType* PT = T->getAs<PointerType>();
1070  QualType UPointeeT = PT->getPointeeType().getUnqualifiedType();
1071  if (UPointeeT->isRecordType()) {
1072  const RecordType *RecordTy = UPointeeT->getAs<RecordType>();
1073  if (!RecordTy->getDecl()->isCompleteDefinition())
1074  return false;
1075  }
1076  return true;
1077 }
1078 
1079 /// \brief Check whether the two versions match.
1080 static bool versionsMatch(const VersionTuple &X, const VersionTuple &Y) {
1081  return (X == Y);
1082 }
1083 
1084 /// AvailabilityAttrsMatch - This routine checks that if comparing two
1085 /// availability attributes, all their components match. It returns
1086 /// true, if not dealing with availability or when all components of
1087 /// availability attributes match. This routine is only called when
1088 /// the attributes are of the same kind.
1089 static bool AvailabilityAttrsMatch(Attr *At1, Attr *At2) {
1090  const AvailabilityAttr *AA1 = dyn_cast<AvailabilityAttr>(At1);
1091  if (!AA1)
1092  return true;
1093  const AvailabilityAttr *AA2 = dyn_cast<AvailabilityAttr>(At2);
1094 
1095  VersionTuple Introduced1 = AA1->getIntroduced();
1096  VersionTuple Deprecated1 = AA1->getDeprecated();
1097  VersionTuple Obsoleted1 = AA1->getObsoleted();
1098  bool IsUnavailable1 = AA1->getUnavailable();
1099  VersionTuple Introduced2 = AA2->getIntroduced();
1100  VersionTuple Deprecated2 = AA2->getDeprecated();
1101  VersionTuple Obsoleted2 = AA2->getObsoleted();
1102  bool IsUnavailable2 = AA2->getUnavailable();
1103  return (versionsMatch(Introduced1, Introduced2) &&
1104  versionsMatch(Deprecated1, Deprecated2) &&
1105  versionsMatch(Obsoleted1, Obsoleted2) &&
1106  IsUnavailable1 == IsUnavailable2);
1107 
1108 }
1109 
1110 static bool MatchTwoAttributeLists(const AttrVec &Attrs1, const AttrVec &Attrs2,
1111  bool &AvailabilityArgsMatch) {
1112  // This list is very small, so this need not be optimized.
1113  for (unsigned i = 0, e = Attrs1.size(); i != e; i++) {
1114  bool match = false;
1115  for (unsigned j = 0, f = Attrs2.size(); j != f; j++) {
1116  // Matching attribute kind only. Except for Availabilty attributes,
1117  // we are not getting into details of the attributes. For all practical purposes
1118  // this is sufficient.
1119  if (Attrs1[i]->getKind() == Attrs2[j]->getKind()) {
1120  if (AvailabilityArgsMatch)
1121  AvailabilityArgsMatch = AvailabilityAttrsMatch(Attrs1[i], Attrs2[j]);
1122  match = true;
1123  break;
1124  }
1125  }
1126  if (!match)
1127  return false;
1128  }
1129  return true;
1130 }
1131 
1132 /// AttributesMatch - This routine checks list of attributes for two
1133 /// decls. It returns false, if there is a mismatch in kind of
1134 /// attributes seen in the decls. It returns true if the two decls
1135 /// have list of same kind of attributes. Furthermore, when there
1136 /// are availability attributes in the two decls, it sets the
1137 /// AvailabilityArgsMatch to false if availability attributes have
1138 /// different versions, etc.
1139 static bool AttributesMatch(const Decl *Decl1, const Decl *Decl2,
1140  bool &AvailabilityArgsMatch) {
1141  if (!Decl1->hasAttrs() || !Decl2->hasAttrs()) {
1142  AvailabilityArgsMatch = (Decl1->hasAttrs() == Decl2->hasAttrs());
1143  return true;
1144  }
1145  AvailabilityArgsMatch = true;
1146  const AttrVec &Attrs1 = Decl1->getAttrs();
1147  const AttrVec &Attrs2 = Decl2->getAttrs();
1148  bool match = MatchTwoAttributeLists(Attrs1, Attrs2, AvailabilityArgsMatch);
1149  if (match && (Attrs2.size() > Attrs1.size()))
1150  return MatchTwoAttributeLists(Attrs2, Attrs1, AvailabilityArgsMatch);
1151  return match;
1152 }
1153 
1155  const char *Name) {
1156  if (!isIdentifierHead(Name[0]))
1157  return false;
1158  std::string NameString = Name;
1159  NameString[0] = toLowercase(NameString[0]);
1160  IdentifierInfo *II = &Ctx.Idents.get(NameString);
1161  return II->getTokenID() == tok::identifier;
1162 }
1163 
1164 bool ObjCMigrateASTConsumer::migrateProperty(ASTContext &Ctx,
1165  ObjCContainerDecl *D,
1166  ObjCMethodDecl *Method) {
1167  if (Method->isPropertyAccessor() || !Method->isInstanceMethod() ||
1168  Method->param_size() != 0)
1169  return false;
1170  // Is this method candidate to be a getter?
1171  QualType GRT = Method->getReturnType();
1172  if (GRT->isVoidType())
1173  return false;
1174 
1175  Selector GetterSelector = Method->getSelector();
1176  ObjCInstanceTypeFamily OIT_Family =
1177  Selector::getInstTypeMethodFamily(GetterSelector);
1178 
1179  if (OIT_Family != OIT_None)
1180  return false;
1181 
1182  IdentifierInfo *getterName = GetterSelector.getIdentifierInfoForSlot(0);
1183  Selector SetterSelector =
1185  PP.getSelectorTable(),
1186  getterName);
1187  ObjCMethodDecl *SetterMethod = D->getInstanceMethod(SetterSelector);
1188  unsigned LengthOfPrefix = 0;
1189  if (!SetterMethod) {
1190  // try a different naming convention for getter: isXxxxx
1191  StringRef getterNameString = getterName->getName();
1192  bool IsPrefix = getterNameString.startswith("is");
1193  // Note that we don't want to change an isXXX method of retainable object
1194  // type to property (readonly or otherwise).
1195  if (IsPrefix && GRT->isObjCRetainableType())
1196  return false;
1197  if (IsPrefix || getterNameString.startswith("get")) {
1198  LengthOfPrefix = (IsPrefix ? 2 : 3);
1199  const char *CGetterName = getterNameString.data() + LengthOfPrefix;
1200  // Make sure that first character after "is" or "get" prefix can
1201  // start an identifier.
1202  if (!IsValidIdentifier(Ctx, CGetterName))
1203  return false;
1204  if (CGetterName[0] && isUppercase(CGetterName[0])) {
1205  getterName = &Ctx.Idents.get(CGetterName);
1206  SetterSelector =
1208  PP.getSelectorTable(),
1209  getterName);
1210  SetterMethod = D->getInstanceMethod(SetterSelector);
1211  }
1212  }
1213  }
1214 
1215  if (SetterMethod) {
1216  if ((ASTMigrateActions & FrontendOptions::ObjCMT_ReadwriteProperty) == 0)
1217  return false;
1218  bool AvailabilityArgsMatch;
1219  if (SetterMethod->isDeprecated() ||
1220  !AttributesMatch(Method, SetterMethod, AvailabilityArgsMatch))
1221  return false;
1222 
1223  // Is this a valid setter, matching the target getter?
1224  QualType SRT = SetterMethod->getReturnType();
1225  if (!SRT->isVoidType())
1226  return false;
1227  const ParmVarDecl *argDecl = *SetterMethod->param_begin();
1228  QualType ArgType = argDecl->getType();
1229  if (!Ctx.hasSameUnqualifiedType(ArgType, GRT))
1230  return false;
1231  edit::Commit commit(*Editor);
1232  rewriteToObjCProperty(Method, SetterMethod, *NSAPIObj, commit,
1233  LengthOfPrefix,
1234  (ASTMigrateActions &
1236  (ASTMigrateActions &
1238  AvailabilityArgsMatch);
1239  Editor->commit(commit);
1240  return true;
1241  }
1242  else if (ASTMigrateActions & FrontendOptions::ObjCMT_ReadonlyProperty) {
1243  // Try a non-void method with no argument (and no setter or property of same name
1244  // as a 'readonly' property.
1245  edit::Commit commit(*Editor);
1246  rewriteToObjCProperty(Method, nullptr /*SetterMethod*/, *NSAPIObj, commit,
1247  LengthOfPrefix,
1248  (ASTMigrateActions &
1249  FrontendOptions::ObjCMT_AtomicProperty) != 0,
1250  (ASTMigrateActions &
1252  /*AvailabilityArgsMatch*/false);
1253  Editor->commit(commit);
1254  return true;
1255  }
1256  return false;
1257 }
1258 
1259 void ObjCMigrateASTConsumer::migrateNsReturnsInnerPointer(ASTContext &Ctx,
1260  ObjCMethodDecl *OM) {
1261  if (OM->isImplicit() ||
1262  !OM->isInstanceMethod() ||
1263  OM->hasAttr<ObjCReturnsInnerPointerAttr>())
1264  return;
1265 
1266  QualType RT = OM->getReturnType();
1267  if (!TypeIsInnerPointer(RT) ||
1268  !NSAPIObj->isMacroDefined("NS_RETURNS_INNER_POINTER"))
1269  return;
1270 
1271  edit::Commit commit(*Editor);
1272  commit.insertBefore(OM->getLocEnd(), " NS_RETURNS_INNER_POINTER");
1273  Editor->commit(commit);
1274 }
1275 
1276 void ObjCMigrateASTConsumer::migratePropertyNsReturnsInnerPointer(ASTContext &Ctx,
1277  ObjCPropertyDecl *P) {
1278  QualType T = P->getType();
1279 
1280  if (!TypeIsInnerPointer(T) ||
1281  !NSAPIObj->isMacroDefined("NS_RETURNS_INNER_POINTER"))
1282  return;
1283  edit::Commit commit(*Editor);
1284  commit.insertBefore(P->getLocEnd(), " NS_RETURNS_INNER_POINTER ");
1285  Editor->commit(commit);
1286 }
1287 
1288 void ObjCMigrateASTConsumer::migrateAllMethodInstaceType(ASTContext &Ctx,
1289  ObjCContainerDecl *CDecl) {
1290  if (CDecl->isDeprecated() || IsCategoryNameWithDeprecatedSuffix(CDecl))
1291  return;
1292 
1293  // migrate methods which can have instancetype as their result type.
1294  for (auto *Method : CDecl->methods()) {
1295  if (Method->isDeprecated())
1296  continue;
1297  migrateMethodInstanceType(Ctx, CDecl, Method);
1298  }
1299 }
1300 
1301 void ObjCMigrateASTConsumer::migrateFactoryMethod(ASTContext &Ctx,
1302  ObjCContainerDecl *CDecl,
1303  ObjCMethodDecl *OM,
1304  ObjCInstanceTypeFamily OIT_Family) {
1305  if (OM->isInstanceMethod() ||
1306  OM->getReturnType() == Ctx.getObjCInstanceType() ||
1307  !OM->getReturnType()->isObjCIdType())
1308  return;
1309 
1310  // Candidate factory methods are + (id) NaMeXXX : ... which belong to a class
1311  // NSYYYNamE with matching names be at least 3 characters long.
1312  ObjCInterfaceDecl *IDecl = dyn_cast<ObjCInterfaceDecl>(CDecl);
1313  if (!IDecl) {
1314  if (ObjCCategoryDecl *CatDecl = dyn_cast<ObjCCategoryDecl>(CDecl))
1315  IDecl = CatDecl->getClassInterface();
1316  else if (ObjCImplDecl *ImpDecl = dyn_cast<ObjCImplDecl>(CDecl))
1317  IDecl = ImpDecl->getClassInterface();
1318  }
1319  if (!IDecl)
1320  return;
1321 
1322  std::string StringClassName = IDecl->getName();
1323  StringRef LoweredClassName(StringClassName);
1324  std::string StringLoweredClassName = LoweredClassName.lower();
1325  LoweredClassName = StringLoweredClassName;
1326 
1327  IdentifierInfo *MethodIdName = OM->getSelector().getIdentifierInfoForSlot(0);
1328  // Handle method with no name at its first selector slot; e.g. + (id):(int)x.
1329  if (!MethodIdName)
1330  return;
1331 
1332  std::string MethodName = MethodIdName->getName();
1333  if (OIT_Family == OIT_Singleton || OIT_Family == OIT_ReturnsSelf) {
1334  StringRef STRefMethodName(MethodName);
1335  size_t len = 0;
1336  if (STRefMethodName.startswith("standard"))
1337  len = strlen("standard");
1338  else if (STRefMethodName.startswith("shared"))
1339  len = strlen("shared");
1340  else if (STRefMethodName.startswith("default"))
1341  len = strlen("default");
1342  else
1343  return;
1344  MethodName = STRefMethodName.substr(len);
1345  }
1346  std::string MethodNameSubStr = MethodName.substr(0, 3);
1347  StringRef MethodNamePrefix(MethodNameSubStr);
1348  std::string StringLoweredMethodNamePrefix = MethodNamePrefix.lower();
1349  MethodNamePrefix = StringLoweredMethodNamePrefix;
1350  size_t Ix = LoweredClassName.rfind(MethodNamePrefix);
1351  if (Ix == StringRef::npos)
1352  return;
1353  std::string ClassNamePostfix = LoweredClassName.substr(Ix);
1354  StringRef LoweredMethodName(MethodName);
1355  std::string StringLoweredMethodName = LoweredMethodName.lower();
1356  LoweredMethodName = StringLoweredMethodName;
1357  if (!LoweredMethodName.startswith(ClassNamePostfix))
1358  return;
1359  if (OIT_Family == OIT_ReturnsSelf)
1360  ReplaceWithClasstype(*this, OM);
1361  else
1362  ReplaceWithInstancetype(Ctx, *this, OM);
1363 }
1364 
1365 static bool IsVoidStarType(QualType Ty) {
1366  if (!Ty->isPointerType())
1367  return false;
1368 
1369  while (const TypedefType *TD = dyn_cast<TypedefType>(Ty.getTypePtr()))
1370  Ty = TD->getDecl()->getUnderlyingType();
1371 
1372  // Is the type void*?
1373  const PointerType* PT = Ty->getAs<PointerType>();
1375  return true;
1376  return IsVoidStarType(PT->getPointeeType());
1377 }
1378 
1379 /// AuditedType - This routine audits the type AT and returns false if it is one of known
1380 /// CF object types or of the "void *" variety. It returns true if we don't care about the type
1381 /// such as a non-pointer or pointers which have no ownership issues (such as "int *").
1382 static bool AuditedType (QualType AT) {
1383  if (!AT->isAnyPointerType() && !AT->isBlockPointerType())
1384  return true;
1385  // FIXME. There isn't much we can say about CF pointer type; or is there?
1387  IsVoidStarType(AT) ||
1388  // If an ObjC object is type, assuming that it is not a CF function and
1389  // that it is an un-audited function.
1391  return false;
1392  // All other pointers are assumed audited as harmless.
1393  return true;
1394 }
1395 
1396 void ObjCMigrateASTConsumer::AnnotateImplicitBridging(ASTContext &Ctx) {
1397  if (CFFunctionIBCandidates.empty())
1398  return;
1399  if (!NSAPIObj->isMacroDefined("CF_IMPLICIT_BRIDGING_ENABLED")) {
1400  CFFunctionIBCandidates.clear();
1401  FileId = FileID();
1402  return;
1403  }
1404  // Insert CF_IMPLICIT_BRIDGING_ENABLE/CF_IMPLICIT_BRIDGING_DISABLED
1405  const Decl *FirstFD = CFFunctionIBCandidates[0];
1406  const Decl *LastFD =
1407  CFFunctionIBCandidates[CFFunctionIBCandidates.size()-1];
1408  const char *PragmaString = "\nCF_IMPLICIT_BRIDGING_ENABLED\n\n";
1409  edit::Commit commit(*Editor);
1410  commit.insertBefore(FirstFD->getLocStart(), PragmaString);
1411  PragmaString = "\n\nCF_IMPLICIT_BRIDGING_DISABLED\n";
1412  SourceLocation EndLoc = LastFD->getLocEnd();
1413  // get location just past end of function location.
1414  EndLoc = PP.getLocForEndOfToken(EndLoc);
1415  if (isa<FunctionDecl>(LastFD)) {
1416  // For Methods, EndLoc points to the ending semcolon. So,
1417  // not of these extra work is needed.
1418  Token Tok;
1419  // get locaiton of token that comes after end of function.
1420  bool Failed = PP.getRawToken(EndLoc, Tok, /*IgnoreWhiteSpace=*/true);
1421  if (!Failed)
1422  EndLoc = Tok.getLocation();
1423  }
1424  commit.insertAfterToken(EndLoc, PragmaString);
1425  Editor->commit(commit);
1426  FileId = FileID();
1427  CFFunctionIBCandidates.clear();
1428 }
1429 
1430 void ObjCMigrateASTConsumer::migrateCFAnnotation(ASTContext &Ctx, const Decl *Decl) {
1431  if (Decl->isDeprecated())
1432  return;
1433 
1434  if (Decl->hasAttr<CFAuditedTransferAttr>()) {
1435  assert(CFFunctionIBCandidates.empty() &&
1436  "Cannot have audited functions/methods inside user "
1437  "provided CF_IMPLICIT_BRIDGING_ENABLE");
1438  return;
1439  }
1440 
1441  // Finction must be annotated first.
1442  if (const FunctionDecl *FuncDecl = dyn_cast<FunctionDecl>(Decl)) {
1443  CF_BRIDGING_KIND AuditKind = migrateAddFunctionAnnotation(Ctx, FuncDecl);
1444  if (AuditKind == CF_BRIDGING_ENABLE) {
1445  CFFunctionIBCandidates.push_back(Decl);
1446  if (FileId.isInvalid())
1447  FileId = PP.getSourceManager().getFileID(Decl->getLocation());
1448  }
1449  else if (AuditKind == CF_BRIDGING_MAY_INCLUDE) {
1450  if (!CFFunctionIBCandidates.empty()) {
1451  CFFunctionIBCandidates.push_back(Decl);
1452  if (FileId.isInvalid())
1453  FileId = PP.getSourceManager().getFileID(Decl->getLocation());
1454  }
1455  }
1456  else
1457  AnnotateImplicitBridging(Ctx);
1458  }
1459  else {
1460  migrateAddMethodAnnotation(Ctx, cast<ObjCMethodDecl>(Decl));
1461  AnnotateImplicitBridging(Ctx);
1462  }
1463 }
1464 
1465 void ObjCMigrateASTConsumer::AddCFAnnotations(ASTContext &Ctx,
1466  const CallEffects &CE,
1467  const FunctionDecl *FuncDecl,
1468  bool ResultAnnotated) {
1469  // Annotate function.
1470  if (!ResultAnnotated) {
1471  RetEffect Ret = CE.getReturnValue();
1472  const char *AnnotationString = nullptr;
1473  if (Ret.getObjKind() == RetEffect::CF) {
1474  if (Ret.isOwned() && NSAPIObj->isMacroDefined("CF_RETURNS_RETAINED"))
1475  AnnotationString = " CF_RETURNS_RETAINED";
1476  else if (Ret.notOwned() &&
1477  NSAPIObj->isMacroDefined("CF_RETURNS_NOT_RETAINED"))
1478  AnnotationString = " CF_RETURNS_NOT_RETAINED";
1479  }
1480  else if (Ret.getObjKind() == RetEffect::ObjC) {
1481  if (Ret.isOwned() && NSAPIObj->isMacroDefined("NS_RETURNS_RETAINED"))
1482  AnnotationString = " NS_RETURNS_RETAINED";
1483  }
1484 
1485  if (AnnotationString) {
1486  edit::Commit commit(*Editor);
1487  commit.insertAfterToken(FuncDecl->getLocEnd(), AnnotationString);
1488  Editor->commit(commit);
1489  }
1490  }
1491  ArrayRef<ArgEffect> AEArgs = CE.getArgs();
1492  unsigned i = 0;
1493  for (FunctionDecl::param_const_iterator pi = FuncDecl->param_begin(),
1494  pe = FuncDecl->param_end(); pi != pe; ++pi, ++i) {
1495  const ParmVarDecl *pd = *pi;
1496  ArgEffect AE = AEArgs[i];
1497  if (AE == DecRef && !pd->hasAttr<CFConsumedAttr>() &&
1498  NSAPIObj->isMacroDefined("CF_CONSUMED")) {
1499  edit::Commit commit(*Editor);
1500  commit.insertBefore(pd->getLocation(), "CF_CONSUMED ");
1501  Editor->commit(commit);
1502  }
1503  else if (AE == DecRefMsg && !pd->hasAttr<NSConsumedAttr>() &&
1504  NSAPIObj->isMacroDefined("NS_CONSUMED")) {
1505  edit::Commit commit(*Editor);
1506  commit.insertBefore(pd->getLocation(), "NS_CONSUMED ");
1507  Editor->commit(commit);
1508  }
1509  }
1510 }
1511 
1512 
1513 ObjCMigrateASTConsumer::CF_BRIDGING_KIND
1514  ObjCMigrateASTConsumer::migrateAddFunctionAnnotation(
1515  ASTContext &Ctx,
1516  const FunctionDecl *FuncDecl) {
1517  if (FuncDecl->hasBody())
1518  return CF_BRIDGING_NONE;
1519 
1520  CallEffects CE = CallEffects::getEffect(FuncDecl);
1521  bool FuncIsReturnAnnotated = (FuncDecl->hasAttr<CFReturnsRetainedAttr>() ||
1522  FuncDecl->hasAttr<CFReturnsNotRetainedAttr>() ||
1523  FuncDecl->hasAttr<NSReturnsRetainedAttr>() ||
1524  FuncDecl->hasAttr<NSReturnsNotRetainedAttr>() ||
1525  FuncDecl->hasAttr<NSReturnsAutoreleasedAttr>());
1526 
1527  // Trivial case of when function is annotated and has no argument.
1528  if (FuncIsReturnAnnotated && FuncDecl->getNumParams() == 0)
1529  return CF_BRIDGING_NONE;
1530 
1531  bool ReturnCFAudited = false;
1532  if (!FuncIsReturnAnnotated) {
1533  RetEffect Ret = CE.getReturnValue();
1534  if (Ret.getObjKind() == RetEffect::CF &&
1535  (Ret.isOwned() || Ret.notOwned()))
1536  ReturnCFAudited = true;
1537  else if (!AuditedType(FuncDecl->getReturnType()))
1538  return CF_BRIDGING_NONE;
1539  }
1540 
1541  // At this point result type is audited for potential inclusion.
1542  // Now, how about argument types.
1543  ArrayRef<ArgEffect> AEArgs = CE.getArgs();
1544  unsigned i = 0;
1545  bool ArgCFAudited = false;
1546  for (FunctionDecl::param_const_iterator pi = FuncDecl->param_begin(),
1547  pe = FuncDecl->param_end(); pi != pe; ++pi, ++i) {
1548  const ParmVarDecl *pd = *pi;
1549  ArgEffect AE = AEArgs[i];
1550  if (AE == DecRef /*CFConsumed annotated*/ || AE == IncRef) {
1551  if (AE == DecRef && !pd->hasAttr<CFConsumedAttr>())
1552  ArgCFAudited = true;
1553  else if (AE == IncRef)
1554  ArgCFAudited = true;
1555  }
1556  else {
1557  QualType AT = pd->getType();
1558  if (!AuditedType(AT)) {
1559  AddCFAnnotations(Ctx, CE, FuncDecl, FuncIsReturnAnnotated);
1560  return CF_BRIDGING_NONE;
1561  }
1562  }
1563  }
1564  if (ReturnCFAudited || ArgCFAudited)
1565  return CF_BRIDGING_ENABLE;
1566 
1567  return CF_BRIDGING_MAY_INCLUDE;
1568 }
1569 
1570 void ObjCMigrateASTConsumer::migrateARCSafeAnnotation(ASTContext &Ctx,
1571  ObjCContainerDecl *CDecl) {
1572  if (!isa<ObjCInterfaceDecl>(CDecl) || CDecl->isDeprecated())
1573  return;
1574 
1575  // migrate methods which can have instancetype as their result type.
1576  for (const auto *Method : CDecl->methods())
1577  migrateCFAnnotation(Ctx, Method);
1578 }
1579 
1580 void ObjCMigrateASTConsumer::AddCFAnnotations(ASTContext &Ctx,
1581  const CallEffects &CE,
1582  const ObjCMethodDecl *MethodDecl,
1583  bool ResultAnnotated) {
1584  // Annotate function.
1585  if (!ResultAnnotated) {
1586  RetEffect Ret = CE.getReturnValue();
1587  const char *AnnotationString = nullptr;
1588  if (Ret.getObjKind() == RetEffect::CF) {
1589  if (Ret.isOwned() && NSAPIObj->isMacroDefined("CF_RETURNS_RETAINED"))
1590  AnnotationString = " CF_RETURNS_RETAINED";
1591  else if (Ret.notOwned() &&
1592  NSAPIObj->isMacroDefined("CF_RETURNS_NOT_RETAINED"))
1593  AnnotationString = " CF_RETURNS_NOT_RETAINED";
1594  }
1595  else if (Ret.getObjKind() == RetEffect::ObjC) {
1596  ObjCMethodFamily OMF = MethodDecl->getMethodFamily();
1597  switch (OMF) {
1598  case clang::OMF_alloc:
1599  case clang::OMF_new:
1600  case clang::OMF_copy:
1601  case clang::OMF_init:
1603  break;
1604 
1605  default:
1606  if (Ret.isOwned() && NSAPIObj->isMacroDefined("NS_RETURNS_RETAINED"))
1607  AnnotationString = " NS_RETURNS_RETAINED";
1608  break;
1609  }
1610  }
1611 
1612  if (AnnotationString) {
1613  edit::Commit commit(*Editor);
1614  commit.insertBefore(MethodDecl->getLocEnd(), AnnotationString);
1615  Editor->commit(commit);
1616  }
1617  }
1618  ArrayRef<ArgEffect> AEArgs = CE.getArgs();
1619  unsigned i = 0;
1620  for (ObjCMethodDecl::param_const_iterator pi = MethodDecl->param_begin(),
1621  pe = MethodDecl->param_end(); pi != pe; ++pi, ++i) {
1622  const ParmVarDecl *pd = *pi;
1623  ArgEffect AE = AEArgs[i];
1624  if (AE == DecRef && !pd->hasAttr<CFConsumedAttr>() &&
1625  NSAPIObj->isMacroDefined("CF_CONSUMED")) {
1626  edit::Commit commit(*Editor);
1627  commit.insertBefore(pd->getLocation(), "CF_CONSUMED ");
1628  Editor->commit(commit);
1629  }
1630  }
1631 }
1632 
1633 void ObjCMigrateASTConsumer::migrateAddMethodAnnotation(
1634  ASTContext &Ctx,
1635  const ObjCMethodDecl *MethodDecl) {
1636  if (MethodDecl->hasBody() || MethodDecl->isImplicit())
1637  return;
1638 
1639  CallEffects CE = CallEffects::getEffect(MethodDecl);
1640  bool MethodIsReturnAnnotated = (MethodDecl->hasAttr<CFReturnsRetainedAttr>() ||
1641  MethodDecl->hasAttr<CFReturnsNotRetainedAttr>() ||
1642  MethodDecl->hasAttr<NSReturnsRetainedAttr>() ||
1643  MethodDecl->hasAttr<NSReturnsNotRetainedAttr>() ||
1644  MethodDecl->hasAttr<NSReturnsAutoreleasedAttr>());
1645 
1646  if (CE.getReceiver() == DecRefMsg &&
1647  !MethodDecl->hasAttr<NSConsumesSelfAttr>() &&
1648  MethodDecl->getMethodFamily() != OMF_init &&
1649  MethodDecl->getMethodFamily() != OMF_release &&
1650  NSAPIObj->isMacroDefined("NS_CONSUMES_SELF")) {
1651  edit::Commit commit(*Editor);
1652  commit.insertBefore(MethodDecl->getLocEnd(), " NS_CONSUMES_SELF");
1653  Editor->commit(commit);
1654  }
1655 
1656  // Trivial case of when function is annotated and has no argument.
1657  if (MethodIsReturnAnnotated &&
1658  (MethodDecl->param_begin() == MethodDecl->param_end()))
1659  return;
1660 
1661  if (!MethodIsReturnAnnotated) {
1662  RetEffect Ret = CE.getReturnValue();
1663  if ((Ret.getObjKind() == RetEffect::CF ||
1664  Ret.getObjKind() == RetEffect::ObjC) &&
1665  (Ret.isOwned() || Ret.notOwned())) {
1666  AddCFAnnotations(Ctx, CE, MethodDecl, false);
1667  return;
1668  } else if (!AuditedType(MethodDecl->getReturnType()))
1669  return;
1670  }
1671 
1672  // At this point result type is either annotated or audited.
1673  // Now, how about argument types.
1674  ArrayRef<ArgEffect> AEArgs = CE.getArgs();
1675  unsigned i = 0;
1676  for (ObjCMethodDecl::param_const_iterator pi = MethodDecl->param_begin(),
1677  pe = MethodDecl->param_end(); pi != pe; ++pi, ++i) {
1678  const ParmVarDecl *pd = *pi;
1679  ArgEffect AE = AEArgs[i];
1680  if ((AE == DecRef && !pd->hasAttr<CFConsumedAttr>()) || AE == IncRef ||
1681  !AuditedType(pd->getType())) {
1682  AddCFAnnotations(Ctx, CE, MethodDecl, MethodIsReturnAnnotated);
1683  return;
1684  }
1685  }
1686  return;
1687 }
1688 
1689 namespace {
1690 class SuperInitChecker : public RecursiveASTVisitor<SuperInitChecker> {
1691 public:
1692  bool shouldVisitTemplateInstantiations() const { return false; }
1693  bool shouldWalkTypesOfTypeLocs() const { return false; }
1694 
1695  bool VisitObjCMessageExpr(ObjCMessageExpr *E) {
1697  if (E->getMethodFamily() == OMF_init)
1698  return false;
1699  }
1700  return true;
1701  }
1702 };
1703 } // anonymous namespace
1704 
1705 static bool hasSuperInitCall(const ObjCMethodDecl *MD) {
1706  return !SuperInitChecker().TraverseStmt(MD->getBody());
1707 }
1708 
1709 void ObjCMigrateASTConsumer::inferDesignatedInitializers(
1710  ASTContext &Ctx,
1711  const ObjCImplementationDecl *ImplD) {
1712 
1713  const ObjCInterfaceDecl *IFace = ImplD->getClassInterface();
1714  if (!IFace || IFace->hasDesignatedInitializers())
1715  return;
1716  if (!NSAPIObj->isMacroDefined("NS_DESIGNATED_INITIALIZER"))
1717  return;
1718 
1719  for (const auto *MD : ImplD->instance_methods()) {
1720  if (MD->isDeprecated() ||
1721  MD->getMethodFamily() != OMF_init ||
1722  MD->isDesignatedInitializerForTheInterface())
1723  continue;
1724  const ObjCMethodDecl *IFaceM = IFace->getMethod(MD->getSelector(),
1725  /*isInstance=*/true);
1726  if (!IFaceM)
1727  continue;
1728  if (hasSuperInitCall(MD)) {
1729  edit::Commit commit(*Editor);
1730  commit.insert(IFaceM->getLocEnd(), " NS_DESIGNATED_INITIALIZER");
1731  Editor->commit(commit);
1732  }
1733  }
1734 }
1735 
1736 bool ObjCMigrateASTConsumer::InsertFoundation(ASTContext &Ctx,
1737  SourceLocation Loc) {
1738  if (FoundationIncluded)
1739  return true;
1740  if (Loc.isInvalid())
1741  return false;
1742  edit::Commit commit(*Editor);
1743  if (Ctx.getLangOpts().Modules)
1744  commit.insert(Loc, "#ifndef NS_ENUM\n@import Foundation;\n#endif\n");
1745  else
1746  commit.insert(Loc, "#ifndef NS_ENUM\n#import <Foundation/Foundation.h>\n#endif\n");
1747  Editor->commit(commit);
1748  FoundationIncluded = true;
1749  return true;
1750 }
1751 
1752 namespace {
1753 
1754 class RewritesReceiver : public edit::EditsReceiver {
1755  Rewriter &Rewrite;
1756 
1757 public:
1758  RewritesReceiver(Rewriter &Rewrite) : Rewrite(Rewrite) { }
1759 
1760  void insert(SourceLocation loc, StringRef text) override {
1761  Rewrite.InsertText(loc, text);
1762  }
1763  void replace(CharSourceRange range, StringRef text) override {
1764  Rewrite.ReplaceText(range.getBegin(), Rewrite.getRangeSize(range), text);
1765  }
1766 };
1767 
1768 class JSONEditWriter : public edit::EditsReceiver {
1770  llvm::raw_ostream &OS;
1771 
1772 public:
1773  JSONEditWriter(SourceManager &SM, llvm::raw_ostream &OS)
1774  : SourceMgr(SM), OS(OS) {
1775  OS << "[\n";
1776  }
1777  ~JSONEditWriter() override { OS << "]\n"; }
1778 
1779 private:
1780  struct EntryWriter {
1782  llvm::raw_ostream &OS;
1783 
1784  EntryWriter(SourceManager &SM, llvm::raw_ostream &OS)
1785  : SourceMgr(SM), OS(OS) {
1786  OS << " {\n";
1787  }
1788  ~EntryWriter() {
1789  OS << " },\n";
1790  }
1791 
1792  void writeLoc(SourceLocation Loc) {
1793  FileID FID;
1794  unsigned Offset;
1795  std::tie(FID, Offset) = SourceMgr.getDecomposedLoc(Loc);
1796  assert(FID.isValid());
1797  SmallString<200> Path =
1798  StringRef(SourceMgr.getFileEntryForID(FID)->getName());
1799  llvm::sys::fs::make_absolute(Path);
1800  OS << " \"file\": \"";
1801  OS.write_escaped(Path.str()) << "\",\n";
1802  OS << " \"offset\": " << Offset << ",\n";
1803  }
1804 
1805  void writeRemove(CharSourceRange Range) {
1806  assert(Range.isCharRange());
1807  std::pair<FileID, unsigned> Begin =
1809  std::pair<FileID, unsigned> End =
1811  assert(Begin.first == End.first);
1812  assert(Begin.second <= End.second);
1813  unsigned Length = End.second - Begin.second;
1814 
1815  OS << " \"remove\": " << Length << ",\n";
1816  }
1817 
1818  void writeText(StringRef Text) {
1819  OS << " \"text\": \"";
1820  OS.write_escaped(Text) << "\",\n";
1821  }
1822  };
1823 
1824  void insert(SourceLocation Loc, StringRef Text) override {
1825  EntryWriter Writer(SourceMgr, OS);
1826  Writer.writeLoc(Loc);
1827  Writer.writeText(Text);
1828  }
1829 
1830  void replace(CharSourceRange Range, StringRef Text) override {
1831  EntryWriter Writer(SourceMgr, OS);
1832  Writer.writeLoc(Range.getBegin());
1833  Writer.writeRemove(Range);
1834  Writer.writeText(Text);
1835  }
1836 
1837  void remove(CharSourceRange Range) override {
1838  EntryWriter Writer(SourceMgr, OS);
1839  Writer.writeLoc(Range.getBegin());
1840  Writer.writeRemove(Range);
1841  }
1842 };
1843 
1844 }
1845 
1846 void ObjCMigrateASTConsumer::HandleTranslationUnit(ASTContext &Ctx) {
1847 
1849  if (ASTMigrateActions & FrontendOptions::ObjCMT_MigrateDecls) {
1850  for (DeclContext::decl_iterator D = TU->decls_begin(), DEnd = TU->decls_end();
1851  D != DEnd; ++D) {
1852  FileID FID = PP.getSourceManager().getFileID((*D)->getLocation());
1853  if (FID.isValid())
1854  if (FileId.isValid() && FileId != FID) {
1855  if (ASTMigrateActions & FrontendOptions::ObjCMT_Annotation)
1856  AnnotateImplicitBridging(Ctx);
1857  }
1858 
1859  if (ObjCInterfaceDecl *CDecl = dyn_cast<ObjCInterfaceDecl>(*D))
1860  if (canModify(CDecl))
1861  migrateObjCContainerDecl(Ctx, CDecl);
1862  if (ObjCCategoryDecl *CatDecl = dyn_cast<ObjCCategoryDecl>(*D)) {
1863  if (canModify(CatDecl))
1864  migrateObjCContainerDecl(Ctx, CatDecl);
1865  }
1866  else if (ObjCProtocolDecl *PDecl = dyn_cast<ObjCProtocolDecl>(*D)) {
1867  ObjCProtocolDecls.insert(PDecl->getCanonicalDecl());
1868  if (canModify(PDecl))
1869  migrateObjCContainerDecl(Ctx, PDecl);
1870  }
1871  else if (const ObjCImplementationDecl *ImpDecl =
1872  dyn_cast<ObjCImplementationDecl>(*D)) {
1873  if ((ASTMigrateActions & FrontendOptions::ObjCMT_ProtocolConformance) &&
1874  canModify(ImpDecl))
1875  migrateProtocolConformance(Ctx, ImpDecl);
1876  }
1877  else if (const EnumDecl *ED = dyn_cast<EnumDecl>(*D)) {
1878  if (!(ASTMigrateActions & FrontendOptions::ObjCMT_NsMacros))
1879  continue;
1880  if (!canModify(ED))
1881  continue;
1883  if (++N != DEnd) {
1884  const TypedefDecl *TD = dyn_cast<TypedefDecl>(*N);
1885  if (migrateNSEnumDecl(Ctx, ED, TD) && TD)
1886  D++;
1887  }
1888  else
1889  migrateNSEnumDecl(Ctx, ED, /*TypedefDecl */nullptr);
1890  }
1891  else if (const TypedefDecl *TD = dyn_cast<TypedefDecl>(*D)) {
1892  if (!(ASTMigrateActions & FrontendOptions::ObjCMT_NsMacros))
1893  continue;
1894  if (!canModify(TD))
1895  continue;
1897  if (++N == DEnd)
1898  continue;
1899  if (const EnumDecl *ED = dyn_cast<EnumDecl>(*N)) {
1900  if (++N != DEnd)
1901  if (const TypedefDecl *TDF = dyn_cast<TypedefDecl>(*N)) {
1902  // prefer typedef-follows-enum to enum-follows-typedef pattern.
1903  if (migrateNSEnumDecl(Ctx, ED, TDF)) {
1904  ++D; ++D;
1905  CacheObjCNSIntegerTypedefed(TD);
1906  continue;
1907  }
1908  }
1909  if (migrateNSEnumDecl(Ctx, ED, TD)) {
1910  ++D;
1911  continue;
1912  }
1913  }
1914  CacheObjCNSIntegerTypedefed(TD);
1915  }
1916  else if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(*D)) {
1917  if ((ASTMigrateActions & FrontendOptions::ObjCMT_Annotation) &&
1918  canModify(FD))
1919  migrateCFAnnotation(Ctx, FD);
1920  }
1921 
1922  if (ObjCContainerDecl *CDecl = dyn_cast<ObjCContainerDecl>(*D)) {
1923  bool CanModify = canModify(CDecl);
1924  // migrate methods which can have instancetype as their result type.
1925  if ((ASTMigrateActions & FrontendOptions::ObjCMT_Instancetype) &&
1926  CanModify)
1927  migrateAllMethodInstaceType(Ctx, CDecl);
1928  // annotate methods with CF annotations.
1929  if ((ASTMigrateActions & FrontendOptions::ObjCMT_Annotation) &&
1930  CanModify)
1931  migrateARCSafeAnnotation(Ctx, CDecl);
1932  }
1933 
1934  if (const ObjCImplementationDecl *
1935  ImplD = dyn_cast<ObjCImplementationDecl>(*D)) {
1936  if ((ASTMigrateActions & FrontendOptions::ObjCMT_DesignatedInitializer) &&
1937  canModify(ImplD))
1938  inferDesignatedInitializers(Ctx, ImplD);
1939  }
1940  }
1941  if (ASTMigrateActions & FrontendOptions::ObjCMT_Annotation)
1942  AnnotateImplicitBridging(Ctx);
1943  }
1944 
1945  if (IsOutputFile) {
1946  std::error_code EC;
1947  llvm::raw_fd_ostream OS(MigrateDir, EC, llvm::sys::fs::F_None);
1948  if (EC) {
1949  DiagnosticsEngine &Diags = Ctx.getDiagnostics();
1950  Diags.Report(Diags.getCustomDiagID(DiagnosticsEngine::Error, "%0"))
1951  << EC.message();
1952  return;
1953  }
1954 
1955  JSONEditWriter Writer(Ctx.getSourceManager(), OS);
1956  Editor->applyRewrites(Writer);
1957  return;
1958  }
1959 
1960  Rewriter rewriter(Ctx.getSourceManager(), Ctx.getLangOpts());
1961  RewritesReceiver Rec(rewriter);
1962  Editor->applyRewrites(Rec);
1963 
1965  I = rewriter.buffer_begin(), E = rewriter.buffer_end(); I != E; ++I) {
1966  FileID FID = I->first;
1967  RewriteBuffer &buf = I->second;
1968  const FileEntry *file = Ctx.getSourceManager().getFileEntryForID(FID);
1969  assert(file);
1970  SmallString<512> newText;
1971  llvm::raw_svector_ostream vecOS(newText);
1972  buf.write(vecOS);
1973  std::unique_ptr<llvm::MemoryBuffer> memBuf(
1974  llvm::MemoryBuffer::getMemBufferCopy(
1975  StringRef(newText.data(), newText.size()), file->getName()));
1976  SmallString<64> filePath(file->getName());
1977  FileMgr.FixupRelativePath(filePath);
1978  Remapper.remap(filePath.str(), std::move(memBuf));
1979  }
1980 
1981  if (IsOutputFile) {
1982  Remapper.flushToFile(MigrateDir, Ctx.getDiagnostics());
1983  } else {
1984  Remapper.flushToDisk(MigrateDir, Ctx.getDiagnostics());
1985  }
1986 }
1987 
1990  return true;
1991 }
1992 
1993 static std::vector<std::string> getWhiteListFilenames(StringRef DirPath) {
1994  using namespace llvm::sys::fs;
1995  using namespace llvm::sys::path;
1996 
1997  std::vector<std::string> Filenames;
1998  if (DirPath.empty() || !is_directory(DirPath))
1999  return Filenames;
2000 
2001  std::error_code EC;
2002  directory_iterator DI = directory_iterator(DirPath, EC);
2003  directory_iterator DE;
2004  for (; !EC && DI != DE; DI = DI.increment(EC)) {
2005  if (is_regular_file(DI->path()))
2006  Filenames.push_back(filename(DI->path()));
2007  }
2008 
2009  return Filenames;
2010 }
2011 
2012 std::unique_ptr<ASTConsumer>
2016  unsigned ObjCMTAction = CI.getFrontendOpts().ObjCMTAction;
2017  unsigned ObjCMTOpts = ObjCMTAction;
2018  // These are companion flags, they do not enable transformations.
2019  ObjCMTOpts &= ~(FrontendOptions::ObjCMT_AtomicProperty |
2021  if (ObjCMTOpts == FrontendOptions::ObjCMT_None) {
2022  // If no specific option was given, enable literals+subscripting transforms
2023  // by default.
2024  ObjCMTAction |= FrontendOptions::ObjCMT_Literals |
2026  }
2027  CI.getPreprocessor().addPPCallbacks(std::unique_ptr<PPCallbacks>(PPRec));
2028  std::vector<std::string> WhiteList =
2030  return llvm::make_unique<ObjCMigrateASTConsumer>(
2031  CI.getFrontendOpts().OutputFile, ObjCMTAction, Remapper,
2032  CI.getFileManager(), PPRec, CI.getPreprocessor(),
2033  /*isOutputFile=*/true, WhiteList);
2034 }
2035 
2036 namespace {
2037 struct EditEntry {
2038  const FileEntry *File;
2039  unsigned Offset;
2040  unsigned RemoveLen;
2041  std::string Text;
2042 
2043  EditEntry() : File(), Offset(), RemoveLen() {}
2044 };
2045 }
2046 
2047 namespace llvm {
2048 template<> struct DenseMapInfo<EditEntry> {
2049  static inline EditEntry getEmptyKey() {
2050  EditEntry Entry;
2051  Entry.Offset = unsigned(-1);
2052  return Entry;
2053  }
2054  static inline EditEntry getTombstoneKey() {
2055  EditEntry Entry;
2056  Entry.Offset = unsigned(-2);
2057  return Entry;
2058  }
2059  static unsigned getHashValue(const EditEntry& Val) {
2060  llvm::FoldingSetNodeID ID;
2061  ID.AddPointer(Val.File);
2062  ID.AddInteger(Val.Offset);
2063  ID.AddInteger(Val.RemoveLen);
2064  ID.AddString(Val.Text);
2065  return ID.ComputeHash();
2066  }
2067  static bool isEqual(const EditEntry &LHS, const EditEntry &RHS) {
2068  return LHS.File == RHS.File &&
2069  LHS.Offset == RHS.Offset &&
2070  LHS.RemoveLen == RHS.RemoveLen &&
2071  LHS.Text == RHS.Text;
2072  }
2073 };
2074 }
2075 
2076 namespace {
2077 class RemapFileParser {
2078  FileManager &FileMgr;
2079 
2080 public:
2081  RemapFileParser(FileManager &FileMgr) : FileMgr(FileMgr) { }
2082 
2083  bool parse(StringRef File, SmallVectorImpl<EditEntry> &Entries) {
2084  using namespace llvm::yaml;
2085 
2086  llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> FileBufOrErr =
2087  llvm::MemoryBuffer::getFile(File);
2088  if (!FileBufOrErr)
2089  return true;
2090 
2092  Stream YAMLStream(FileBufOrErr.get()->getMemBufferRef(), SM);
2093  document_iterator I = YAMLStream.begin();
2094  if (I == YAMLStream.end())
2095  return true;
2096  Node *Root = I->getRoot();
2097  if (!Root)
2098  return true;
2099 
2100  SequenceNode *SeqNode = dyn_cast<SequenceNode>(Root);
2101  if (!SeqNode)
2102  return true;
2103 
2105  AI = SeqNode->begin(), AE = SeqNode->end(); AI != AE; ++AI) {
2106  MappingNode *MapNode = dyn_cast<MappingNode>(&*AI);
2107  if (!MapNode)
2108  continue;
2109  parseEdit(MapNode, Entries);
2110  }
2111 
2112  return false;
2113  }
2114 
2115 private:
2116  void parseEdit(llvm::yaml::MappingNode *Node,
2117  SmallVectorImpl<EditEntry> &Entries) {
2118  using namespace llvm::yaml;
2119  EditEntry Entry;
2120  bool Ignore = false;
2121 
2123  KVI = Node->begin(), KVE = Node->end(); KVI != KVE; ++KVI) {
2124  ScalarNode *KeyString = dyn_cast<ScalarNode>((*KVI).getKey());
2125  if (!KeyString)
2126  continue;
2127  SmallString<10> KeyStorage;
2128  StringRef Key = KeyString->getValue(KeyStorage);
2129 
2130  ScalarNode *ValueString = dyn_cast<ScalarNode>((*KVI).getValue());
2131  if (!ValueString)
2132  continue;
2133  SmallString<64> ValueStorage;
2134  StringRef Val = ValueString->getValue(ValueStorage);
2135 
2136  if (Key == "file") {
2137  const FileEntry *FE = FileMgr.getFile(Val);
2138  if (!FE)
2139  Ignore = true;
2140  Entry.File = FE;
2141  } else if (Key == "offset") {
2142  if (Val.getAsInteger(10, Entry.Offset))
2143  Ignore = true;
2144  } else if (Key == "remove") {
2145  if (Val.getAsInteger(10, Entry.RemoveLen))
2146  Ignore = true;
2147  } else if (Key == "text") {
2148  Entry.Text = Val;
2149  }
2150  }
2151 
2152  if (!Ignore)
2153  Entries.push_back(Entry);
2154  }
2155 };
2156 }
2157 
2158 static bool reportDiag(const Twine &Err, DiagnosticsEngine &Diag) {
2160  << Err.str();
2161  return true;
2162 }
2163 
2164 static std::string applyEditsToTemp(const FileEntry *FE,
2165  ArrayRef<EditEntry> Edits,
2166  FileManager &FileMgr,
2168  using namespace llvm::sys;
2169 
2170  SourceManager SM(Diag, FileMgr);
2172  LangOptions LangOpts;
2173  edit::EditedSource Editor(SM, LangOpts);
2175  I = Edits.begin(), E = Edits.end(); I != E; ++I) {
2176  const EditEntry &Entry = *I;
2177  assert(Entry.File == FE);
2178  SourceLocation Loc =
2179  SM.getLocForStartOfFile(FID).getLocWithOffset(Entry.Offset);
2180  CharSourceRange Range;
2181  if (Entry.RemoveLen != 0) {
2182  Range = CharSourceRange::getCharRange(Loc,
2183  Loc.getLocWithOffset(Entry.RemoveLen));
2184  }
2185 
2186  edit::Commit commit(Editor);
2187  if (Range.isInvalid()) {
2188  commit.insert(Loc, Entry.Text);
2189  } else if (Entry.Text.empty()) {
2190  commit.remove(Range);
2191  } else {
2192  commit.replace(Range, Entry.Text);
2193  }
2194  Editor.commit(commit);
2195  }
2196 
2197  Rewriter rewriter(SM, LangOpts);
2198  RewritesReceiver Rec(rewriter);
2199  Editor.applyRewrites(Rec);
2200 
2201  const RewriteBuffer *Buf = rewriter.getRewriteBufferFor(FID);
2202  SmallString<512> NewText;
2203  llvm::raw_svector_ostream OS(NewText);
2204  Buf->write(OS);
2205 
2206  SmallString<64> TempPath;
2207  int FD;
2208  if (fs::createTemporaryFile(path::filename(FE->getName()),
2209  path::extension(FE->getName()).drop_front(), FD,
2210  TempPath)) {
2211  reportDiag("Could not create file: " + TempPath.str(), Diag);
2212  return std::string();
2213  }
2214 
2215  llvm::raw_fd_ostream TmpOut(FD, /*shouldClose=*/true);
2216  TmpOut.write(NewText.data(), NewText.size());
2217  TmpOut.close();
2218 
2219  return TempPath.str();
2220 }
2221 
2223  std::vector<std::pair<std::string,std::string> > &remap,
2224  ArrayRef<StringRef> remapFiles,
2225  DiagnosticConsumer *DiagClient) {
2226  bool hasErrorOccurred = false;
2227 
2228  FileSystemOptions FSOpts;
2229  FileManager FileMgr(FSOpts);
2230  RemapFileParser Parser(FileMgr);
2231 
2234  new DiagnosticsEngine(DiagID, new DiagnosticOptions,
2235  DiagClient, /*ShouldOwnClient=*/false));
2236 
2237  typedef llvm::DenseMap<const FileEntry *, std::vector<EditEntry> >
2238  FileEditEntriesTy;
2239  FileEditEntriesTy FileEditEntries;
2240 
2241  llvm::DenseSet<EditEntry> EntriesSet;
2242 
2244  I = remapFiles.begin(), E = remapFiles.end(); I != E; ++I) {
2246  if (Parser.parse(*I, Entries))
2247  continue;
2248 
2250  EI = Entries.begin(), EE = Entries.end(); EI != EE; ++EI) {
2251  EditEntry &Entry = *EI;
2252  if (!Entry.File)
2253  continue;
2255  Insert = EntriesSet.insert(Entry);
2256  if (!Insert.second)
2257  continue;
2258 
2259  FileEditEntries[Entry.File].push_back(Entry);
2260  }
2261  }
2262 
2264  I = FileEditEntries.begin(), E = FileEditEntries.end(); I != E; ++I) {
2265  std::string TempFile = applyEditsToTemp(I->first, I->second,
2266  FileMgr, *Diags);
2267  if (TempFile.empty()) {
2268  hasErrorOccurred = true;
2269  continue;
2270  }
2271 
2272  remap.emplace_back(I->first->getName(), TempFile);
2273  }
2274 
2275  return hasErrorOccurred;
2276 }
std::string OutputFile
The output file, if any.
The receiver is the instance of the superclass object.
Definition: ExprObjC.h:1009
SourceManager & getSourceManager() const
Definition: Preprocessor.h:687
param_const_iterator param_begin() const
Definition: DeclObjC.h:359
bool BeginInvocation(CompilerInstance &CI) override
Callback before starting processing a single input, giving the opportunity to modify the CompilerInvo...
Definition: ObjCMT.cpp:1988
Defines the clang::ASTContext interface.
SourceLocation getEnd() const
bool remove(CharSourceRange range)
Definition: Commit.cpp:86
std::unique_ptr< ASTConsumer > CreateASTConsumer(CompilerInstance &CI, StringRef InFile) override
Create the AST consumer object for this action, if supported.
Definition: ObjCMT.cpp:193
static bool AttributesMatch(const Decl *Decl1, const Decl *Decl2, bool &AvailabilityArgsMatch)
AttributesMatch - This routine checks list of attributes for two decls.
Definition: ObjCMT.cpp:1139
FunctionDecl - An instance of this class is created to represent a function declaration or definition...
Definition: Decl.h:1483
static bool MatchTwoAttributeLists(const AttrVec &Attrs1, const AttrVec &Attrs2, bool &AvailabilityArgsMatch)
Definition: ObjCMT.cpp:1110
static DiagnosticBuilder Diag(DiagnosticsEngine *Diags, const LangOptions &Features, FullSourceLoc TokLoc, const char *TokBegin, const char *TokRangeBegin, const char *TokRangeEnd, unsigned DiagID)
Produce a diagnostic highlighting some portion of a literal.
The receiver is an object instance.
Definition: ExprObjC.h:1005
ObjCMigrateAction(FrontendAction *WrappedAction, StringRef migrateDir, unsigned migrateAction)
Definition: ObjCMT.cpp:182
StringRef getName() const
getName - Get the name of identifier for this declaration as a StringRef.
Definition: Decl.h:169
std::string ObjCMTWhiteListPath
ParmVarDecl *const * param_const_iterator
Definition: Decl.h:1902
Smart pointer class that efficiently represents Objective-C method names.
SelectorTable & getSelectorTable()
Definition: Preprocessor.h:692
PointerType - C99 6.7.5.1 - Pointer Declarators.
Definition: Type.h:2147
Enable migration to add conforming protocols.
unsigned Length
A (possibly-)qualified type.
Definition: Type.h:575
SourceLocation getBegin() const
Represents a version number in the form major[.minor[.subminor[.build]]].
Definition: VersionTuple.h:26
bool rewriteToObjCSubscriptSyntax(const ObjCMessageExpr *Msg, const NSAPI &NS, Commit &commit)
Implements support for file system lookup, file system caching, and directory search management...
Definition: FileManager.h:115
ObjCMethodFamily getMethodFamily() const
Definition: ExprObjC.h:1270
ObjCInterfaceDecl * getClassInterface()
Definition: DeclObjC.cpp:1043
ASTConsumer - This is an abstract interface that should be implemented by clients that read ASTs...
Definition: ASTConsumer.h:36
Defines the clang::FileManager interface and associated types.
bool rewriteToObjCLiteralSyntax(const ObjCMessageExpr *Msg, const NSAPI &NS, Commit &commit, const ParentMap *PMap)
static bool hasSuperInitCall(const ObjCMethodDecl *MD)
Definition: ObjCMT.cpp:1705
Enable inferring NS_DESIGNATED_INITIALIZER for ObjC methods.
IdentifierInfo * getIdentifier() const
getIdentifier - Get the identifier that names this declaration, if there is one.
Definition: Decl.h:164
protocol_loc_iterator protocol_loc_end() const
Definition: DeclObjC.h:1081
QualType getQualifiedType(SplitQualType split) const
Un-split a SplitQualType.
Definition: ASTContext.h:1673
bool insertWrap(StringRef before, CharSourceRange range, StringRef after)
Definition: Commit.cpp:98
static LLVM_READONLY bool isUppercase(unsigned char c)
Return true if this character is an uppercase ASCII letter: [A-Z].
Definition: CharInfo.h:106
TypedefDecl - Represents the declaration of a typedef-name via the 'typedef' type specifier...
Definition: Decl.h:2598
iterator end()
Definition: DeclGroup.h:108
Abstract base class for actions which can be performed by the frontend.
Decl - This represents one declaration (or definition), e.g.
Definition: DeclBase.h:77
const RewriteBuffer * getRewriteBufferFor(FileID FID) const
getRewriteBufferFor - Return the rewrite buffer for the specified FileID.
Definition: Rewriter.h:170
const char * getCharacterData(SourceLocation SL, bool *Invalid=nullptr) const
Return a pointer to the start of the specified location in the appropriate spelling MemoryBuffer...
The argument has its reference count decreased by 1.
std::string getAsString() const
Definition: Type.h:901
static bool TypeIsInnerPointer(QualType T)
Definition: ObjCMT.cpp:1055
IdentifierInfo * getAsIdentifierInfo() const
getAsIdentifierInfo - Retrieve the IdentifierInfo * stored in this declaration name, or NULL if this declaration name isn't a simple identifier.
ObjCProtocolDecl * lookupNestedProtocol(IdentifierInfo *Name)
Definition: DeclObjC.cpp:613
SourceLocation getDeclaratorEndLoc() const
Returns the location where the declarator ends.
Definition: DeclObjC.h:288
DiagnosticBuilder Report(SourceLocation Loc, unsigned DiagID)
Issue the message to the client.
Definition: Diagnostic.h:1117
bool insertAfterToken(SourceLocation loc, StringRef text, bool beforePreviousInsertions=false)
Definition: Commit.h:69
A container of type source information.
Definition: Decl.h:61
Parser - This implements a parser for the C family of languages.
Definition: Parse/Parser.h:56
bool insertFromRange(SourceLocation loc, CharSourceRange range, bool afterToken=false, bool beforePreviousInsertions=false)
Definition: Commit.cpp:59
TypeSourceInfo * getIntegerTypeSourceInfo() const
Return the type source info for the underlying integer type, if no type source info exists...
Definition: Decl.h:3070
bool isBlockPointerType() const
Definition: Type.h:5311
SourceLocation getLocEnd() const LLVM_READONLY
Definition: DeclBase.h:380
const ObjCPropertyDecl * findPropertyDecl(bool CheckOverrides=true) const
Returns the property associated with this method's selector.
Definition: DeclObjC.cpp:1203
ObjCMethodDecl * getMethod(Selector Sel, bool isInstance, bool AllowHidden=false) const
Definition: DeclObjC.cpp:68
void removeObjCLifetime()
Definition: Type.h:296
SourceLocation findSemiAfterLocation(SourceLocation loc, ASTContext &Ctx, bool IsDecl=false)
'Loc' is the end of a statement range.
Definition: Transforms.cpp:135
SourceManager & getSourceManager() const
Return the current source manager.
static std::vector< std::string > getWhiteListFilenames(StringRef DirPath)
Definition: ObjCMT.cpp:1993
bool getRawToken(SourceLocation Loc, Token &Result, bool IgnoreWhiteSpace=false)
Relex the token at the specified location.
SourceLocation getLocStart() const LLVM_READONLY
Definition: DeclObjC.h:291
void applyRewrites(EditsReceiver &receiver)
Abstract interface, implemented by clients of the front-end, which formats and prints fully processed...
Definition: Diagnostic.h:1307
Enable migration to NS_ENUM/NS_OPTIONS macros.
ObjCMethodDecl - Represents an instance or class method declaration.
Definition: DeclObjC.h:113
RewriteBuffer - As code is rewritten, SourceBuffer's from the original input with modifications get a...
Definition: RewriteBuffer.h:27
bool isRealType() const
Definition: Type.cpp:1799
Describes how types, statements, expressions, and declarations should be printed. ...
Definition: PrettyPrinter.h:35
decl_iterator decls_end() const
Definition: DeclBase.h:1441
unsigned param_size() const
Definition: DeclObjC.h:348
ParmVarDecl - Represents a parameter to a function.
Definition: Decl.h:1299
bool isObjCRetainableType() const
Definition: Type.cpp:3704
bool isVoidType() const
Definition: Type.h:5546
QualType getType() const
Definition: DeclObjC.h:2497
The collection of all-type qualifiers we support.
Definition: Type.h:116
static EditEntry getTombstoneKey()
Definition: ObjCMT.cpp:2054
Enable annotation of ObjCMethods of all kinds.
Base wrapper for a particular "section" of type source info.
Definition: TypeLoc.h:40
Expr * IgnoreImpCasts() LLVM_READONLY
IgnoreImpCasts - Skip past any implicit casts which might surround this expression.
Definition: Expr.h:2755
bool insert(SourceLocation loc, StringRef text, bool afterToken=false, bool beforePreviousInsertions=false)
Definition: Commit.cpp:43
SourceLocation getLocForEndOfToken(SourceLocation Loc, unsigned Offset=0)
Computes the source location just past the end of the token at this source location.
bool hasBody(const FunctionDecl *&Definition) const
Returns true if the function has a body (definition).
Definition: Decl.cpp:2460
ObjCProtocolDecl * getCanonicalDecl() override
Retrieves the canonical declaration of this Objective-C protocol.
Definition: DeclObjC.h:1897
One of these records is kept for each identifier that is lexed.
SourceLocation getSelectorStartLoc() const
Definition: DeclObjC.h:297
static ObjCInstanceTypeFamily getInstTypeMethodFamily(Selector sel)
class LLVM_ALIGNAS(8) DependentTemplateSpecializationType const IdentifierInfo * Name
Represents a template specialization type whose template cannot be resolved, e.g. ...
Definition: Type.h:4381
bool hasAttr() const
Definition: DeclBase.h:498
static std::string applyEditsToTemp(const FileEntry *FE, ArrayRef< EditEntry > Edits, FileManager &FileMgr, DiagnosticsEngine &Diag)
Definition: ObjCMT.cpp:2164
IdentifierInfo * getIdentifierInfoForSlot(unsigned argIndex) const
Retrieve the identifier at a given position in the selector.
Holds long-lived AST nodes (such as types and decls) that can be referred to throughout the semantic ...
Definition: ASTContext.h:91
The results of name lookup within a DeclContext.
Definition: DeclBase.h:1054
ObjCMethodFamily
A family of Objective-C methods.
QualType getReturnType() const
Definition: Decl.h:1956
bool isCompleteDefinition() const
isCompleteDefinition - Return true if this decl has its body fully specified.
Definition: Decl.h:2788
bool isAnyPointerType() const
Definition: Type.h:5308
std::unique_ptr< ASTConsumer > CreateASTConsumer(CompilerInstance &CI, StringRef InFile) override
Create the AST consumer object for this action, if supported.
bool hasSameType(QualType T1, QualType T2) const
Determine whether the given types T1 and T2 are equivalent.
Definition: ASTContext.h:1962
Token - This structure provides full information about a lexed token.
Definition: Token.h:37
method_range methods() const
Definition: DeclObjC.h:729
Enable migration to modern ObjC readwrite property.
SourceLocation findLocationAfterSemi(SourceLocation loc, ASTContext &Ctx, bool IsDecl=false)
'Loc' is the end of a statement range.
Definition: Transforms.cpp:123
Keeps track of the various options that can be enabled, which controls the dialect of C or C++ that i...
Definition: LangOptions.h:48
bool hasSameUnqualifiedType(QualType T1, QualType T2) const
Determine whether the given types are equivalent after cvr-qualifiers have been removed.
Definition: ASTContext.h:1987
IdentifierTable & Idents
Definition: ASTContext.h:451
unsigned SuppressStrongLifetime
When true, suppress printing of the __strong lifetime qualifier in ARC.
SourceLocation getSuperLoc() const
Retrieve the location of the 'super' keyword for a class or instance message to 'super', otherwise an invalid source location.
Definition: ExprObjC.h:1196
static void ReplaceWithInstancetype(ASTContext &Ctx, const ObjCMigrateASTConsumer &ASTC, ObjCMethodDecl *OM)
Definition: ObjCMT.cpp:964
SourceLocation getBeginLoc() const
Get the begin source location.
Definition: TypeLoc.cpp:170
static Selector constructSetterSelector(IdentifierTable &Idents, SelectorTable &SelTable, const IdentifierInfo *Name)
Return the default setter selector for the given identifier.
bool replace(CharSourceRange range, StringRef text)
Definition: Commit.cpp:111
static const char * PropertyMemoryAttribute(ASTContext &Context, QualType ArgType)
Definition: ObjCMT.cpp:426
SourceLocation getLocWithOffset(int Offset) const
Return a source location with the specified offset from this SourceLocation.
ObjCContainerDecl - Represents a container for method declarations.
Definition: DeclObjC.h:696
const LangOptions & getLangOpts() const
Definition: ASTContext.h:596
bool isImplicit() const
isImplicit - Indicates whether the declaration was implicitly generated by the implementation.
Definition: DeclBase.h:514
FrontendOptions & getFrontendOpts()
SourceLocation getSuperClassLoc() const
Retrieve the starting location of the superclass.
Definition: DeclObjC.cpp:309
uint32_t Offset
Definition: CacheTokens.cpp:44
static bool UseNSOptionsMacro(Preprocessor &PP, ASTContext &Ctx, const EnumDecl *EnumDcl)
Definition: ObjCMT.cpp:799
The argument has its reference count increased by 1.
ObjCMethodFamily getMethodFamily() const
Determines the family of this method.
Definition: DeclObjC.cpp:885
static bool rewriteToNSEnumDecl(const EnumDecl *EnumDcl, const TypedefDecl *TypedefDcl, const NSAPI &NS, edit::Commit &commit, StringRef NSIntegerName, bool NSOptions)
Definition: ObjCMT.cpp:704
tok::TokenKind getTokenID() const
If this is a source-language token (e.g.
Concrete class used by the front-end to report problems and issues.
Definition: Diagnostic.h:135
bool BeginInvocation(CompilerInstance &CI) override
Callback before starting processing a single input, giving the opportunity to modify the CompilerInvo...
Definition: ObjCMT.cpp:205
bool hasDesignatedInitializers() const
Returns true if this interface decl contains at least one initializer marked with the 'objc_designate...
Definition: DeclObjC.cpp:1420
A builtin binary operation expression such as "x + y" or "x <= y".
Definition: Expr.h:2875
RecordDecl * getDecl() const
Definition: Type.h:3553
iterator begin()
Definition: DeclGroup.h:102
std::string getNameAsString() const
getNameAsString - Get a human-readable name for the declaration, even if it is one of the special kin...
Definition: Decl.h:184
Expr * IgnoreParenCasts() LLVM_READONLY
IgnoreParenCasts - Ignore parentheses and casts.
Definition: Expr.cpp:2464
ObjCProtocolDecl * getDefinition()
Retrieve the definition of this protocol, if any.
Definition: DeclObjC.h:1858
static bool reportDiag(const Twine &Err, DiagnosticsEngine &Diag)
Definition: ObjCMT.cpp:2158
child_range children()
Definition: ExprObjC.cpp:339
bool empty() const
Definition: DeclObjC.h:46
Represents an Objective-C protocol declaration.
Definition: DeclObjC.h:1728
SmallVector< BoundNodes, 1 > match(MatcherT Matcher, const NodeT &Node, ASTContext &Context)
Returns the results of matching Matcher on Node.
A class that does preorder depth-first traversal on the entire Clang AST and visits each node...
Represents an ObjC class declaration.
Definition: DeclObjC.h:853
SourceLocation getLocEnd() const LLVM_READONLY
Definition: DeclObjC.cpp:879
static EditEntry getEmptyKey()
Definition: ObjCMT.cpp:2049
decl_iterator decls_begin() const
Definition: DeclBase.cpp:1167
detail::InMemoryDirectory::const_iterator I
Preprocessor & getPreprocessor() const
Return the current preprocessor.
QualType getType() const
Definition: Decl.h:530
bool isInvalid() const
param_iterator param_begin()
Definition: Decl.h:1906
DiagnosticsEngine & getDiagnostics() const
AnnotatingParser & P
ArgEffect
An ArgEffect summarizes the retain count behavior on an argument or receiver to a function or method...
annotate property with NS_RETURNS_INNER_POINTER
const ParmVarDecl *const * param_const_iterator
Definition: DeclObjC.h:349
const FileEntry * getFileEntryForID(FileID FID) const
Returns the FileEntry record for the provided FileID.
SourceLocation getLocStart() const LLVM_READONLY
Definition: Decl.h:2510
ObjCProtocolDecl * lookupProtocolNamed(IdentifierInfo *PName)
Definition: DeclObjC.cpp:1771
FileID getFileID(SourceLocation SpellingLoc) const
Return the FileID for a SourceLocation.
Qualifiers::ObjCLifetime getObjCLifetime() const
Returns lifetime attribute of this type.
Definition: Type.h:980
static bool AuditedType(QualType AT)
AuditedType - This routine audits the type AT and returns false if it is one of known CF object types...
Definition: ObjCMT.cpp:1382
ASTContext * Context
const SmallVectorImpl< AnnotatedLine * >::const_iterator End
Enable migration to modern ObjC literals.
ID
Defines the set of possible language-specific address spaces.
Definition: AddressSpaces.h:27
bool isFunctionPointerType() const
Definition: Type.h:5323
SourceManager & SM
bool ObjCMethodsAreEqual(const ObjCMethodDecl *MethodDecl, const ObjCMethodDecl *MethodImp)
const ObjCMethodDecl * getMethodDecl() const
Definition: ExprObjC.h:1251
Expr - This represents one expression.
Definition: Expr.h:104
StringRef getName() const
Return the actual identifier string.
Represents a character-granular source range.
SourceLocation getEnd() const
FileID createFileID(const FileEntry *SourceFile, SourceLocation IncludePos, SrcMgr::CharacteristicKind FileCharacter, int LoadedID=0, unsigned LoadedOffset=0)
Create a new FileID that represents the specified file being #included from the specified IncludePosi...
SourceManager & SourceMgr
Definition: Format.cpp:1352
TranslationUnitDecl * getTranslationUnitDecl() const
Definition: ASTContext.h:875
Defines the clang::Preprocessor interface.
static bool IsCategoryNameWithDeprecatedSuffix(ObjCContainerDecl *D)
Definition: ObjCMT.cpp:563
Expr ** getArgs()
Retrieve the arguments to this message, not including the receiver.
Definition: ExprObjC.h:1281
ObjCInstanceTypeFamily
A family of Objective-C methods.
bool isObjCIdType() const
Definition: Type.h:5401
SourceLocation getLocation() const
Return a source location identifier for the specified offset in the current file. ...
Definition: Token.h:124
bool isInstanceMethod() const
Definition: DeclObjC.h:419
static bool IsValidIdentifier(ASTContext &Ctx, const char *Name)
Definition: ObjCMT.cpp:1154
An expression that sends a message to the given Objective-C object or class.
Definition: ExprObjC.h:860
static bool subscriptOperatorNeedsParens(const Expr *FullExpr)
DeclarationName getDeclName() const
getDeclName - Get the actual, stored name of the declaration, which may be a special name...
Definition: Decl.h:190
DiagnosticsEngine & getDiagnostics() const
Get the current diagnostics engine.
void CollectInheritedProtocols(const Decl *CDecl, llvm::SmallPtrSet< ObjCProtocolDecl *, 8 > &Protocols)
CollectInheritedProtocols - Collect all protocols in current class and those inherited by it...
SourceLocation getLocEnd() const LLVM_READONLY
Definition: TypeLoc.h:131
const clang::PrintingPolicy & getPrintingPolicy() const
Definition: ASTContext.h:545
TypeSourceInfo * getReturnTypeSourceInfo() const
Definition: DeclObjC.h:344
bool isMacroDefined(StringRef Id) const
Returns true if Id is currently defined as a macro.
Definition: NSAPI.cpp:520
AttrVec & getAttrs()
Definition: DeclBase.h:443
bool hasObjCLifetime() const
Definition: Type.h:289
static CharSourceRange getCharRange(SourceRange R)
param_const_iterator param_end() const
Definition: DeclObjC.h:362
SourceLocation getEndLoc() const
Get the end source location.
Definition: TypeLoc.cpp:207
const char * getLiteralData() const
getLiteralData - For a literal token (numeric constant, string, etc), this returns a pointer to the s...
Definition: Token.h:215
std::unique_ptr< ASTConsumer > CreateASTConsumer(CompilerInstance &CI, StringRef InFile) override
Create the AST consumer object for this action, if supported.
Definition: ObjCMT.cpp:2013
static void ReplaceWithClasstype(const ObjCMigrateASTConsumer &ASTC, ObjCMethodDecl *OM)
Definition: ObjCMT.cpp:987
SourceLocation getLocStart() const LLVM_READONLY
Definition: DeclBase.h:377
There is no lifetime qualification on this type.
Definition: Type.h:133
TypeLoc getTypeLoc() const
Return the TypeLoc wrapper for the type source info.
Definition: TypeLoc.h:215
ObjCPropertyImplDecl * FindPropertyImplDecl(IdentifierInfo *propertyId) const
FindPropertyImplDecl - This method looks up a previous ObjCPropertyImplDecl added to the list of thos...
Definition: DeclObjC.cpp:2014
#define false
Definition: stdbool.h:33
Assigning into this object requires the old value to be released and the new value to be retained...
Definition: Type.h:144
static void append_attr(std::string &PropertyString, const char *attr, bool &LParenAdded)
Definition: ObjCMT.cpp:383
const char * getName() const
Definition: FileManager.h:84
CompilerInstance - Helper class for managing a single instance of the Clang compiler.
Encodes a location in the source.
enumerator_range enumerators() const
Definition: Decl.h:3026
IdentifierInfo & get(StringRef Name)
Return the identifier token info for the specified named identifier.
unsigned getNumParams() const
getNumParams - Return the number of parameters this function must have based on its FunctionType...
Definition: Decl.cpp:2743
const Type * getTypePtr() const
Retrieves a pointer to the underlying (unqualified) type.
Definition: Type.h:5089
A helper class that allows the use of isa/cast/dyncast to detect TagType objects of enums...
Definition: Type.h:3570
const TemplateArgument * iterator
Definition: Type.h:4070
static unsigned getHashValue(const EditEntry &Val)
Definition: ObjCMT.cpp:2059
bool isValid() const
Return true if this is a valid SourceLocation object.
Options for controlling the compiler diagnostics engine.
static void MigrateBlockOrFunctionPointerTypeVariable(std::string &PropertyString, const std::string &TypeString, const char *name)
Definition: ObjCMT.cpp:395
bool isCharRange() const
StringRef getNameForSlot(unsigned argIndex) const
Retrieve the name at a given position in the selector.
Records preprocessor conditional directive regions and allows querying in which region source locatio...
IdentifierTable & getIdentifierTable()
Definition: Preprocessor.h:690
static bool ClassImplementsAllMethodsAndProperties(ASTContext &Ctx, const ObjCImplementationDecl *ImpDecl, const ObjCInterfaceDecl *IDecl, ObjCProtocolDecl *Protocol)
Definition: ObjCMT.cpp:599
Cached information about one file (either on disk or in the virtual file system). ...
Definition: FileManager.h:53
Stmt * getBody() const override
Retrieve the body of this method, if it has one.
Definition: DeclObjC.cpp:756
bool InsertText(SourceLocation Loc, StringRef Str, bool InsertAfter=true, bool indentNewLines=false)
InsertText - Insert the specified string at the specified location in the original buffer...
Definition: Rewriter.cpp:238
bool isObjCBuiltinType() const
Definition: Type.h:5416
ObjCInterfaceDecl * lookupInheritedClass(const IdentifierInfo *ICName)
lookupInheritedClass - This method returns ObjCInterfaceDecl * of the super class whose name is passe...
Definition: DeclObjC.cpp:594
ObjCCategoryDecl - Represents a category declaration.
Definition: DeclObjC.h:1931
static bool AvailabilityAttrsMatch(Attr *At1, Attr *At2)
AvailabilityAttrsMatch - This routine checks that if comparing two availability attributes, all their components match.
Definition: ObjCMT.cpp:1089
const ObjCInterfaceDecl * getClassInterface() const
Definition: DeclObjC.h:2085
bool isPropertyAccessor() const
Definition: DeclObjC.h:426
Represents one property declaration in an Objective-C interface.
Definition: DeclObjC.h:2416
Enable converting setter/getter expressions to property-dot syntx.
QualType getReturnType() const
Definition: DeclObjC.h:330
SourceLocation getBegin() const
lookup_result lookup(DeclarationName Name) const
lookup - Find the declarations (if any) with the given Name in this context.
Definition: DeclBase.cpp:1368
unsigned getCustomDiagID(Level L, const char(&FormatString)[N])
Return an ID for a diagnostic with the specified format string and level.
Definition: Diagnostic.h:602
static StringRef GetUnsignedName(StringRef NSIntegerName)
Definition: ObjCMT.cpp:693
QualType getType() const
Return the type wrapped by this type source info.
Definition: Decl.h:69
QualType getObjCInstanceType()
Retrieve the Objective-C "instancetype" type, if already known; otherwise, returns a NULL type;...
Definition: ASTContext.h:1469
QualType getPointeeType() const
Definition: Type.h:2161
void setIgnoreAllWarnings(bool Val)
When set to true, any unmapped warnings are ignored.
Definition: Diagnostic.h:440
decl_iterator - Iterates through the declarations stored within this context.
Definition: DeclBase.h:1398
Expr * getInstanceReceiver()
Returns the object expression (receiver) for an instance message, or null for a message that is not a...
Definition: ExprObjC.h:1155
ast_type_traits::DynTypedNode Node
An opaque identifier used by SourceManager which refers to a source file (MemoryBuffer) along with it...
ASTContext & getASTContext() const
Definition: NSAPI.h:28
instmeth_range instance_methods() const
Definition: DeclObjC.h:744
Enable migration to modern ObjC readonly property.
SourceLocation getSelectorLoc(unsigned Index) const
Definition: ExprObjC.h:1318
prop_range properties() const
Definition: DeclObjC.h:716
ObjCMethodDecl * getInstanceMethod(Selector Sel, bool AllowHidden=false) const
Definition: DeclObjC.h:772
static void rewriteToNSMacroDecl(ASTContext &Ctx, const EnumDecl *EnumDcl, const TypedefDecl *TypedefDcl, const NSAPI &NS, edit::Commit &commit, bool IsNSIntegerType)
Definition: ObjCMT.cpp:757
FileManager & getFileManager() const
Return the current file manager to the caller.
Used for handling and querying diagnostic IDs.
Selector getSelector() const
Definition: DeclObjC.h:328
EnumDecl - Represents an enum.
Definition: Decl.h:2930
bool hasAttrs() const
Definition: DeclBase.h:439
detail::InMemoryDirectory::const_iterator E
CanQualType getCanonicalType(QualType T) const
Return the canonical (structural) type corresponding to the specified potentially non-canonical type ...
Definition: ASTContext.h:1946
param_iterator param_end()
Definition: Decl.h:1907
A frontend action which simply wraps some other runtime-specified frontend action.
bool isLiteral() const
Return true if this is a "literal", like a numeric constant, string, etc.
Definition: Token.h:113
Represents a pointer to an Objective C object.
Definition: Type.h:4821
static bool IsVoidStarType(QualType Ty)
Definition: ObjCMT.cpp:1365
ObjCImplementationDecl - Represents a class definition - this is where method definitions are specifi...
Definition: DeclObjC.h:2220
A helper class that allows the use of isa/cast/dyncast to detect TagType objects of structs/unions/cl...
Definition: Type.h:3544
raw_ostream & write(raw_ostream &Stream) const
Write to Stream the result of applying all changes to the original buffer.
Definition: Rewriter.cpp:25
const T * getAs() const
Member-template getAs<specific type>'.
Definition: Type.h:5675
use NS_NONATOMIC_IOSONLY for property 'atomic' attribute
bool isInvalid() const
QualType getIntegerType() const
getIntegerType - Return the integer type this enum decl corresponds to.
Definition: Decl.h:3054
std::map< FileID, RewriteBuffer >::iterator buffer_iterator
Definition: Rewriter.h:53
bool hasBody() const override
Determine whether this method has a body.
Definition: DeclObjC.h:486
Enable migration to modern ObjC subscripting.
bool commit(const Commit &commit)
static bool isEqual(const EditEntry &LHS, const EditEntry &RHS)
Definition: ObjCMT.cpp:2067
const ObjCProtocolList & getReferencedProtocols() const
Definition: DeclObjC.h:1016
unsigned getNumArgs() const
Return the number of actual arguments in this message, not counting the receiver. ...
Definition: ExprObjC.h:1277
bool getFileRemappingsFromFileList(std::vector< std::pair< std::string, std::string > > &remap, ArrayRef< StringRef > remapFiles, DiagnosticConsumer *DiagClient)
Get the set of file remappings from a list of files with remapping info.
Definition: ObjCMT.cpp:2222
SourceManager & getSourceManager()
Definition: ASTContext.h:553
Keeps track of options that affect how file operations are performed.
X
Add a minimal nested name specifier fixit hint to allow lookup of a tag name from an outer enclosing ...
Definition: SemaDecl.cpp:11761
Reading or writing from this object requires a barrier call.
Definition: Type.h:147
bool initFromDisk(StringRef outputDir, DiagnosticsEngine &Diag, bool ignoreIfFilesChanged)
Rewriter - This is the main interface to the rewrite buffers.
Definition: Rewriter.h:31
QualType getUnqualifiedType() const
Retrieve the unqualified variant of the given type, removing as little sugar as possible.
Definition: Type.h:5169
bool isObjCObjectPointerType() const
Definition: Type.h:5377
prefer 'atomic' property over 'nonatomic'.
static LLVM_READONLY char toLowercase(char c)
Converts the given ASCII character to its lowercase equivalent.
Definition: CharInfo.h:165
bool isImplicit() const
Indicates whether the message send was implicitly generated by the implementation.
Definition: ExprObjC.h:1132
static bool versionsMatch(const VersionTuple &X, const VersionTuple &Y)
Check whether the two versions match.
Definition: ObjCMT.cpp:1080
static void rewriteToObjCProperty(const ObjCMethodDecl *Getter, const ObjCMethodDecl *Setter, const NSAPI &NS, edit::Commit &commit, unsigned LengthOfPrefix, bool Atomic, bool UseNsIosOnlyMacro, bool AvailabilityArgsMatch)
Definition: ObjCMT.cpp:452
Enable migration of ObjC methods to 'instancetype'.
ObjCInterfaceDecl * getSuperClass() const
Definition: DeclObjC.cpp:289
bool insertBefore(SourceLocation loc, StringRef text)
Definition: Commit.h:73
static Decl::Kind getKind(const Decl *D)
Definition: DeclBase.cpp:772
TranslationUnitDecl - The top declaration context.
Definition: Decl.h:79
unsigned getLength() const
Definition: Token.h:127
SourceLocation getLocForStartOfFile(FileID FID) const
Return the source location corresponding to the first byte of the specified file. ...
static LLVM_READONLY bool isIdentifierHead(unsigned char c, bool AllowDollar=false)
Returns true if this is a valid first character of a C identifier, which is [a-zA-Z_].
Definition: CharInfo.h:49
The argument has its reference count decreased by 1.
StringRef Text
Definition: Format.cpp:1724
A trivial tuple used to represent a source range.
SourceLocation getLocation() const
Definition: DeclBase.h:384
bool isValid() const
std::pair< FileID, unsigned > getDecomposedLoc(SourceLocation Loc) const
Decompose the specified location into a raw FileID + Offset pair.
bool isNull() const
Return true if this QualType doesn't point to a type yet.
Definition: Type.h:642
TypeSourceInfo * getTypeSourceInfo() const
Definition: Decl.h:2561
ReceiverKind getReceiverKind() const
Determine the kind of receiver that this message is being sent to.
Definition: ExprObjC.h:1136
ObjCCategoryImplDecl - An object of this class encapsulates a category @implementation declaration...
Definition: DeclObjC.h:2139
void addPPCallbacks(std::unique_ptr< PPCallbacks > C)
Definition: Preprocessor.h:778
This class handles loading and caching of source files into memory.
Attr - This represents one attribute.
Definition: Attr.h:44
static bool rewriteToObjCInterfaceDecl(const ObjCInterfaceDecl *IDecl, llvm::SmallVectorImpl< ObjCProtocolDecl * > &ConformingProtocols, const NSAPI &NS, edit::Commit &commit)
Definition: ObjCMT.cpp:661
bool isDeprecated(std::string *Message=nullptr) const
Determine whether this declaration is marked 'deprecated'.
Definition: DeclBase.h:604
Engages in a tight little dance with the lexer to efficiently preprocess tokens.
Definition: Preprocessor.h:96
Qualifiers getQualifiers() const
Retrieve the set of qualifiers applied to this type.
Definition: Type.h:5116
bool isPointerType() const
Definition: Type.h:5305