1 //===--- SemaDeclAttr.cpp - Declaration Attribute Handling ----------------===//
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 //  This file implements decl-related attribute processing.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/AST/ASTConsumer.h"
15 #include "clang/AST/ASTContext.h"
16 #include "clang/AST/ASTMutationListener.h"
17 #include "clang/AST/CXXInheritance.h"
18 #include "clang/AST/DeclCXX.h"
19 #include "clang/AST/DeclObjC.h"
20 #include "clang/AST/DeclTemplate.h"
21 #include "clang/AST/Expr.h"
22 #include "clang/AST/ExprCXX.h"
23 #include "clang/AST/Mangle.h"
24 #include "clang/AST/RecursiveASTVisitor.h"
25 #include "clang/Basic/CharInfo.h"
26 #include "clang/Basic/SourceManager.h"
27 #include "clang/Basic/TargetInfo.h"
28 #include "clang/Lex/Preprocessor.h"
29 #include "clang/Sema/DeclSpec.h"
30 #include "clang/Sema/DelayedDiagnostic.h"
31 #include "clang/Sema/Initialization.h"
32 #include "clang/Sema/Lookup.h"
33 #include "clang/Sema/Scope.h"
34 #include "clang/Sema/ScopeInfo.h"
35 #include "clang/Sema/SemaInternal.h"
36 #include "llvm/ADT/STLExtras.h"
37 #include "llvm/ADT/StringExtras.h"
38 #include "llvm/Support/MathExtras.h"
39 
40 using namespace clang;
41 using namespace sema;
42 
43 namespace AttributeLangSupport {
44   enum LANG {
45     C,
46     Cpp,
47     ObjC
48   };
49 } // end namespace AttributeLangSupport
50 
51 //===----------------------------------------------------------------------===//
52 //  Helper functions
53 //===----------------------------------------------------------------------===//
54 
55 /// isFunctionOrMethod - Return true if the given decl has function
56 /// type (function or function-typed variable) or an Objective-C
57 /// method.
58 static bool isFunctionOrMethod(const Decl *D) {
59   return (D->getFunctionType() != nullptr) || isa<ObjCMethodDecl>(D);
60 }
61 
62 /// Return true if the given decl has function type (function or
63 /// function-typed variable) or an Objective-C method or a block.
64 static bool isFunctionOrMethodOrBlock(const Decl *D) {
65   return isFunctionOrMethod(D) || isa<BlockDecl>(D);
66 }
67 
68 /// Return true if the given decl has a declarator that should have
69 /// been processed by Sema::GetTypeForDeclarator.
70 static bool hasDeclarator(const Decl *D) {
71   // In some sense, TypedefDecl really *ought* to be a DeclaratorDecl.
72   return isa<DeclaratorDecl>(D) || isa<BlockDecl>(D) || isa<TypedefNameDecl>(D) ||
73          isa<ObjCPropertyDecl>(D);
74 }
75 
76 /// hasFunctionProto - Return true if the given decl has a argument
77 /// information. This decl should have already passed
78 /// isFunctionOrMethod or isFunctionOrMethodOrBlock.
79 static bool hasFunctionProto(const Decl *D) {
80   if (const FunctionType *FnTy = D->getFunctionType())
81     return isa<FunctionProtoType>(FnTy);
82   return isa<ObjCMethodDecl>(D) || isa<BlockDecl>(D);
83 }
84 
85 /// getFunctionOrMethodNumParams - Return number of function or method
86 /// parameters. It is an error to call this on a K&R function (use
87 /// hasFunctionProto first).
88 static unsigned getFunctionOrMethodNumParams(const Decl *D) {
89   if (const FunctionType *FnTy = D->getFunctionType())
90     return cast<FunctionProtoType>(FnTy)->getNumParams();
91   if (const auto *BD = dyn_cast<BlockDecl>(D))
92     return BD->getNumParams();
93   return cast<ObjCMethodDecl>(D)->param_size();
94 }
95 
96 static QualType getFunctionOrMethodParamType(const Decl *D, unsigned Idx) {
97   if (const FunctionType *FnTy = D->getFunctionType())
98     return cast<FunctionProtoType>(FnTy)->getParamType(Idx);
99   if (const auto *BD = dyn_cast<BlockDecl>(D))
100     return BD->getParamDecl(Idx)->getType();
101 
102   return cast<ObjCMethodDecl>(D)->parameters()[Idx]->getType();
103 }
104 
105 static SourceRange getFunctionOrMethodParamRange(const Decl *D, unsigned Idx) {
106   if (const auto *FD = dyn_cast<FunctionDecl>(D))
107     return FD->getParamDecl(Idx)->getSourceRange();
108   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D))
109     return MD->parameters()[Idx]->getSourceRange();
110   if (const auto *BD = dyn_cast<BlockDecl>(D))
111     return BD->getParamDecl(Idx)->getSourceRange();
112   return SourceRange();
113 }
114 
115 static QualType getFunctionOrMethodResultType(const Decl *D) {
116   if (const FunctionType *FnTy = D->getFunctionType())
117     return FnTy->getReturnType();
118   return cast<ObjCMethodDecl>(D)->getReturnType();
119 }
120 
121 static SourceRange getFunctionOrMethodResultSourceRange(const Decl *D) {
122   if (const auto *FD = dyn_cast<FunctionDecl>(D))
123     return FD->getReturnTypeSourceRange();
124   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D))
125     return MD->getReturnTypeSourceRange();
126   return SourceRange();
127 }
128 
129 static bool isFunctionOrMethodVariadic(const Decl *D) {
130   if (const FunctionType *FnTy = D->getFunctionType())
131     return cast<FunctionProtoType>(FnTy)->isVariadic();
132   if (const auto *BD = dyn_cast<BlockDecl>(D))
133     return BD->isVariadic();
134   return cast<ObjCMethodDecl>(D)->isVariadic();
135 }
136 
137 static bool isInstanceMethod(const Decl *D) {
138   if (const auto *MethodDecl = dyn_cast<CXXMethodDecl>(D))
139     return MethodDecl->isInstance();
140   return false;
141 }
142 
143 static inline bool isNSStringType(QualType T, ASTContext &Ctx) {
144   const auto *PT = T->getAs<ObjCObjectPointerType>();
145   if (!PT)
146     return false;
147 
148   ObjCInterfaceDecl *Cls = PT->getObjectType()->getInterface();
149   if (!Cls)
150     return false;
151 
152   IdentifierInfo* ClsName = Cls->getIdentifier();
153 
154   // FIXME: Should we walk the chain of classes?
155   return ClsName == &Ctx.Idents.get("NSString") ||
156          ClsName == &Ctx.Idents.get("NSMutableString");
157 }
158 
159 static inline bool isCFStringType(QualType T, ASTContext &Ctx) {
160   const auto *PT = T->getAs<PointerType>();
161   if (!PT)
162     return false;
163 
164   const auto *RT = PT->getPointeeType()->getAs<RecordType>();
165   if (!RT)
166     return false;
167 
168   const RecordDecl *RD = RT->getDecl();
169   if (RD->getTagKind() != TTK_Struct)
170     return false;
171 
172   return RD->getIdentifier() == &Ctx.Idents.get("__CFString");
173 }
174 
175 static unsigned getNumAttributeArgs(const ParsedAttr &AL) {
176   // FIXME: Include the type in the argument list.
177   return AL.getNumArgs() + AL.hasParsedType();
178 }
179 
180 template <typename Compare>
181 static bool checkAttributeNumArgsImpl(Sema &S, const ParsedAttr &AL,
182                                       unsigned Num, unsigned Diag,
183                                       Compare Comp) {
184   if (Comp(getNumAttributeArgs(AL), Num)) {
185     S.Diag(AL.getLoc(), Diag) << AL << Num;
186     return false;
187   }
188 
189   return true;
190 }
191 
192 /// Check if the attribute has exactly as many args as Num. May
193 /// output an error.
194 static bool checkAttributeNumArgs(Sema &S, const ParsedAttr &AL, unsigned Num) {
195   return checkAttributeNumArgsImpl(S, AL, Num,
196                                    diag::err_attribute_wrong_number_arguments,
197                                    std::not_equal_to<unsigned>());
198 }
199 
200 /// Check if the attribute has at least as many args as Num. May
201 /// output an error.
202 static bool checkAttributeAtLeastNumArgs(Sema &S, const ParsedAttr &AL,
203                                          unsigned Num) {
204   return checkAttributeNumArgsImpl(S, AL, Num,
205                                    diag::err_attribute_too_few_arguments,
206                                    std::less<unsigned>());
207 }
208 
209 /// Check if the attribute has at most as many args as Num. May
210 /// output an error.
211 static bool checkAttributeAtMostNumArgs(Sema &S, const ParsedAttr &AL,
212                                         unsigned Num) {
213   return checkAttributeNumArgsImpl(S, AL, Num,
214                                    diag::err_attribute_too_many_arguments,
215                                    std::greater<unsigned>());
216 }
217 
218 /// A helper function to provide Attribute Location for the Attr types
219 /// AND the ParsedAttr.
220 template <typename AttrInfo>
221 static typename std::enable_if<std::is_base_of<Attr, AttrInfo>::value,
222                                SourceLocation>::type
223 getAttrLoc(const AttrInfo &AL) {
224   return AL.getLocation();
225 }
226 static SourceLocation getAttrLoc(const ParsedAttr &AL) { return AL.getLoc(); }
227 
228 /// If Expr is a valid integer constant, get the value of the integer
229 /// expression and return success or failure. May output an error.
230 template <typename AttrInfo>
231 static bool checkUInt32Argument(Sema &S, const AttrInfo &AI, const Expr *Expr,
232                                 uint32_t &Val, unsigned Idx = UINT_MAX) {
233   llvm::APSInt I(32);
234   if (Expr->isTypeDependent() || Expr->isValueDependent() ||
235       !Expr->isIntegerConstantExpr(I, S.Context)) {
236     if (Idx != UINT_MAX)
237       S.Diag(getAttrLoc(AI), diag::err_attribute_argument_n_type)
238           << AI << Idx << AANT_ArgumentIntegerConstant
239           << Expr->getSourceRange();
240     else
241       S.Diag(getAttrLoc(AI), diag::err_attribute_argument_type)
242           << AI << AANT_ArgumentIntegerConstant << Expr->getSourceRange();
243     return false;
244   }
245 
246   if (!I.isIntN(32)) {
247     S.Diag(Expr->getExprLoc(), diag::err_ice_too_large)
248         << I.toString(10, false) << 32 << /* Unsigned */ 1;
249     return false;
250   }
251 
252   Val = (uint32_t)I.getZExtValue();
253   return true;
254 }
255 
256 /// Wrapper around checkUInt32Argument, with an extra check to be sure
257 /// that the result will fit into a regular (signed) int. All args have the same
258 /// purpose as they do in checkUInt32Argument.
259 template <typename AttrInfo>
260 static bool checkPositiveIntArgument(Sema &S, const AttrInfo &AI, const Expr *Expr,
261                                      int &Val, unsigned Idx = UINT_MAX) {
262   uint32_t UVal;
263   if (!checkUInt32Argument(S, AI, Expr, UVal, Idx))
264     return false;
265 
266   if (UVal > (uint32_t)std::numeric_limits<int>::max()) {
267     llvm::APSInt I(32); // for toString
268     I = UVal;
269     S.Diag(Expr->getExprLoc(), diag::err_ice_too_large)
270         << I.toString(10, false) << 32 << /* Unsigned */ 0;
271     return false;
272   }
273 
274   Val = UVal;
275   return true;
276 }
277 
278 /// Diagnose mutually exclusive attributes when present on a given
279 /// declaration. Returns true if diagnosed.
280 template <typename AttrTy>
281 static bool checkAttrMutualExclusion(Sema &S, Decl *D, const ParsedAttr &AL) {
282   if (const auto *A = D->getAttr<AttrTy>()) {
283     S.Diag(AL.getLoc(), diag::err_attributes_are_not_compatible) << AL << A;
284     S.Diag(A->getLocation(), diag::note_conflicting_attribute);
285     return true;
286   }
287   return false;
288 }
289 
290 template <typename AttrTy>
291 static bool checkAttrMutualExclusion(Sema &S, Decl *D, const Attr &AL) {
292   if (const auto *A = D->getAttr<AttrTy>()) {
293     S.Diag(AL.getLocation(), diag::err_attributes_are_not_compatible) << &AL
294                                                                       << A;
295     S.Diag(A->getLocation(), diag::note_conflicting_attribute);
296     return true;
297   }
298   return false;
299 }
300 
301 /// Check if IdxExpr is a valid parameter index for a function or
302 /// instance method D.  May output an error.
303 ///
304 /// \returns true if IdxExpr is a valid index.
305 template <typename AttrInfo>
306 static bool checkFunctionOrMethodParameterIndex(
307     Sema &S, const Decl *D, const AttrInfo &AI, unsigned AttrArgNum,
308     const Expr *IdxExpr, ParamIdx &Idx, bool CanIndexImplicitThis = false) {
309   assert(isFunctionOrMethodOrBlock(D));
310 
311   // In C++ the implicit 'this' function parameter also counts.
312   // Parameters are counted from one.
313   bool HP = hasFunctionProto(D);
314   bool HasImplicitThisParam = isInstanceMethod(D);
315   bool IV = HP && isFunctionOrMethodVariadic(D);
316   unsigned NumParams =
317       (HP ? getFunctionOrMethodNumParams(D) : 0) + HasImplicitThisParam;
318 
319   llvm::APSInt IdxInt;
320   if (IdxExpr->isTypeDependent() || IdxExpr->isValueDependent() ||
321       !IdxExpr->isIntegerConstantExpr(IdxInt, S.Context)) {
322     S.Diag(getAttrLoc(AI), diag::err_attribute_argument_n_type)
323         << &AI << AttrArgNum << AANT_ArgumentIntegerConstant
324         << IdxExpr->getSourceRange();
325     return false;
326   }
327 
328   unsigned IdxSource = IdxInt.getLimitedValue(UINT_MAX);
329   if (IdxSource < 1 || (!IV && IdxSource > NumParams)) {
330     S.Diag(getAttrLoc(AI), diag::err_attribute_argument_out_of_bounds)
331         << &AI << AttrArgNum << IdxExpr->getSourceRange();
332     return false;
333   }
334   if (HasImplicitThisParam && !CanIndexImplicitThis) {
335     if (IdxSource == 1) {
336       S.Diag(getAttrLoc(AI), diag::err_attribute_invalid_implicit_this_argument)
337           << &AI << IdxExpr->getSourceRange();
338       return false;
339     }
340   }
341 
342   Idx = ParamIdx(IdxSource, D);
343   return true;
344 }
345 
346 /// Check if the argument \p ArgNum of \p Attr is a ASCII string literal.
347 /// If not emit an error and return false. If the argument is an identifier it
348 /// will emit an error with a fixit hint and treat it as if it was a string
349 /// literal.
350 bool Sema::checkStringLiteralArgumentAttr(const ParsedAttr &AL, unsigned ArgNum,
351                                           StringRef &Str,
352                                           SourceLocation *ArgLocation) {
353   // Look for identifiers. If we have one emit a hint to fix it to a literal.
354   if (AL.isArgIdent(ArgNum)) {
355     IdentifierLoc *Loc = AL.getArgAsIdent(ArgNum);
356     Diag(Loc->Loc, diag::err_attribute_argument_type)
357         << AL << AANT_ArgumentString
358         << FixItHint::CreateInsertion(Loc->Loc, "\"")
359         << FixItHint::CreateInsertion(getLocForEndOfToken(Loc->Loc), "\"");
360     Str = Loc->Ident->getName();
361     if (ArgLocation)
362       *ArgLocation = Loc->Loc;
363     return true;
364   }
365 
366   // Now check for an actual string literal.
367   Expr *ArgExpr = AL.getArgAsExpr(ArgNum);
368   const auto *Literal = dyn_cast<StringLiteral>(ArgExpr->IgnoreParenCasts());
369   if (ArgLocation)
370     *ArgLocation = ArgExpr->getBeginLoc();
371 
372   if (!Literal || !Literal->isAscii()) {
373     Diag(ArgExpr->getBeginLoc(), diag::err_attribute_argument_type)
374         << AL << AANT_ArgumentString;
375     return false;
376   }
377 
378   Str = Literal->getString();
379   return true;
380 }
381 
382 /// Applies the given attribute to the Decl without performing any
383 /// additional semantic checking.
384 template <typename AttrType>
385 static void handleSimpleAttribute(Sema &S, Decl *D, const ParsedAttr &AL) {
386   D->addAttr(::new (S.Context) AttrType(AL.getRange(), S.Context,
387                                         AL.getAttributeSpellingListIndex()));
388 }
389 
390 template <typename AttrType>
391 static void handleSimpleAttributeWithExclusions(Sema &S, Decl *D,
392                                                 const ParsedAttr &AL) {
393   handleSimpleAttribute<AttrType>(S, D, AL);
394 }
395 
396 /// Applies the given attribute to the Decl so long as the Decl doesn't
397 /// already have one of the given incompatible attributes.
398 template <typename AttrType, typename IncompatibleAttrType,
399           typename... IncompatibleAttrTypes>
400 static void handleSimpleAttributeWithExclusions(Sema &S, Decl *D,
401                                                 const ParsedAttr &AL) {
402   if (checkAttrMutualExclusion<IncompatibleAttrType>(S, D, AL))
403     return;
404   handleSimpleAttributeWithExclusions<AttrType, IncompatibleAttrTypes...>(S, D,
405                                                                           AL);
406 }
407 
408 /// Check if the passed-in expression is of type int or bool.
409 static bool isIntOrBool(Expr *Exp) {
410   QualType QT = Exp->getType();
411   return QT->isBooleanType() || QT->isIntegerType();
412 }
413 
414 
415 // Check to see if the type is a smart pointer of some kind.  We assume
416 // it's a smart pointer if it defines both operator-> and operator*.
417 static bool threadSafetyCheckIsSmartPointer(Sema &S, const RecordType* RT) {
418   DeclContextLookupResult Res1 = RT->getDecl()->lookup(
419       S.Context.DeclarationNames.getCXXOperatorName(OO_Star));
420   if (Res1.empty())
421     return false;
422 
423   DeclContextLookupResult Res2 = RT->getDecl()->lookup(
424       S.Context.DeclarationNames.getCXXOperatorName(OO_Arrow));
425   if (Res2.empty())
426     return false;
427 
428   return true;
429 }
430 
431 /// Check if passed in Decl is a pointer type.
432 /// Note that this function may produce an error message.
433 /// \return true if the Decl is a pointer type; false otherwise
434 static bool threadSafetyCheckIsPointer(Sema &S, const Decl *D,
435                                        const ParsedAttr &AL) {
436   const auto *VD = cast<ValueDecl>(D);
437   QualType QT = VD->getType();
438   if (QT->isAnyPointerType())
439     return true;
440 
441   if (const auto *RT = QT->getAs<RecordType>()) {
442     // If it's an incomplete type, it could be a smart pointer; skip it.
443     // (We don't want to force template instantiation if we can avoid it,
444     // since that would alter the order in which templates are instantiated.)
445     if (RT->isIncompleteType())
446       return true;
447 
448     if (threadSafetyCheckIsSmartPointer(S, RT))
449       return true;
450   }
451 
452   S.Diag(AL.getLoc(), diag::warn_thread_attribute_decl_not_pointer) << AL << QT;
453   return false;
454 }
455 
456 /// Checks that the passed in QualType either is of RecordType or points
457 /// to RecordType. Returns the relevant RecordType, null if it does not exit.
458 static const RecordType *getRecordType(QualType QT) {
459   if (const auto *RT = QT->getAs<RecordType>())
460     return RT;
461 
462   // Now check if we point to record type.
463   if (const auto *PT = QT->getAs<PointerType>())
464     return PT->getPointeeType()->getAs<RecordType>();
465 
466   return nullptr;
467 }
468 
469 static bool checkRecordTypeForCapability(Sema &S, QualType Ty) {
470   const RecordType *RT = getRecordType(Ty);
471 
472   if (!RT)
473     return false;
474 
475   // Don't check for the capability if the class hasn't been defined yet.
476   if (RT->isIncompleteType())
477     return true;
478 
479   // Allow smart pointers to be used as capability objects.
480   // FIXME -- Check the type that the smart pointer points to.
481   if (threadSafetyCheckIsSmartPointer(S, RT))
482     return true;
483 
484   // Check if the record itself has a capability.
485   RecordDecl *RD = RT->getDecl();
486   if (RD->hasAttr<CapabilityAttr>())
487     return true;
488 
489   // Else check if any base classes have a capability.
490   if (const auto *CRD = dyn_cast<CXXRecordDecl>(RD)) {
491     CXXBasePaths BPaths(false, false);
492     if (CRD->lookupInBases([](const CXXBaseSpecifier *BS, CXXBasePath &) {
493           const auto *Type = BS->getType()->getAs<RecordType>();
494           return Type->getDecl()->hasAttr<CapabilityAttr>();
495         }, BPaths))
496       return true;
497   }
498   return false;
499 }
500 
501 static bool checkTypedefTypeForCapability(QualType Ty) {
502   const auto *TD = Ty->getAs<TypedefType>();
503   if (!TD)
504     return false;
505 
506   TypedefNameDecl *TN = TD->getDecl();
507   if (!TN)
508     return false;
509 
510   return TN->hasAttr<CapabilityAttr>();
511 }
512 
513 static bool typeHasCapability(Sema &S, QualType Ty) {
514   if (checkTypedefTypeForCapability(Ty))
515     return true;
516 
517   if (checkRecordTypeForCapability(S, Ty))
518     return true;
519 
520   return false;
521 }
522 
523 static bool isCapabilityExpr(Sema &S, const Expr *Ex) {
524   // Capability expressions are simple expressions involving the boolean logic
525   // operators &&, || or !, a simple DeclRefExpr, CastExpr or a ParenExpr. Once
526   // a DeclRefExpr is found, its type should be checked to determine whether it
527   // is a capability or not.
528 
529   if (const auto *E = dyn_cast<CastExpr>(Ex))
530     return isCapabilityExpr(S, E->getSubExpr());
531   else if (const auto *E = dyn_cast<ParenExpr>(Ex))
532     return isCapabilityExpr(S, E->getSubExpr());
533   else if (const auto *E = dyn_cast<UnaryOperator>(Ex)) {
534     if (E->getOpcode() == UO_LNot || E->getOpcode() == UO_AddrOf ||
535         E->getOpcode() == UO_Deref)
536       return isCapabilityExpr(S, E->getSubExpr());
537     return false;
538   } else if (const auto *E = dyn_cast<BinaryOperator>(Ex)) {
539     if (E->getOpcode() == BO_LAnd || E->getOpcode() == BO_LOr)
540       return isCapabilityExpr(S, E->getLHS()) &&
541              isCapabilityExpr(S, E->getRHS());
542     return false;
543   }
544 
545   return typeHasCapability(S, Ex->getType());
546 }
547 
548 /// Checks that all attribute arguments, starting from Sidx, resolve to
549 /// a capability object.
550 /// \param Sidx The attribute argument index to start checking with.
551 /// \param ParamIdxOk Whether an argument can be indexing into a function
552 /// parameter list.
553 static void checkAttrArgsAreCapabilityObjs(Sema &S, Decl *D,
554                                            const ParsedAttr &AL,
555                                            SmallVectorImpl<Expr *> &Args,
556                                            int Sidx = 0,
557                                            bool ParamIdxOk = false) {
558   for (unsigned Idx = Sidx; Idx < AL.getNumArgs(); ++Idx) {
559     Expr *ArgExp = AL.getArgAsExpr(Idx);
560 
561     if (ArgExp->isTypeDependent()) {
562       // FIXME -- need to check this again on template instantiation
563       Args.push_back(ArgExp);
564       continue;
565     }
566 
567     if (const auto *StrLit = dyn_cast<StringLiteral>(ArgExp)) {
568       if (StrLit->getLength() == 0 ||
569           (StrLit->isAscii() && StrLit->getString() == StringRef("*"))) {
570         // Pass empty strings to the analyzer without warnings.
571         // Treat "*" as the universal lock.
572         Args.push_back(ArgExp);
573         continue;
574       }
575 
576       // We allow constant strings to be used as a placeholder for expressions
577       // that are not valid C++ syntax, but warn that they are ignored.
578       S.Diag(AL.getLoc(), diag::warn_thread_attribute_ignored) << AL;
579       Args.push_back(ArgExp);
580       continue;
581     }
582 
583     QualType ArgTy = ArgExp->getType();
584 
585     // A pointer to member expression of the form  &MyClass::mu is treated
586     // specially -- we need to look at the type of the member.
587     if (const auto *UOp = dyn_cast<UnaryOperator>(ArgExp))
588       if (UOp->getOpcode() == UO_AddrOf)
589         if (const auto *DRE = dyn_cast<DeclRefExpr>(UOp->getSubExpr()))
590           if (DRE->getDecl()->isCXXInstanceMember())
591             ArgTy = DRE->getDecl()->getType();
592 
593     // First see if we can just cast to record type, or pointer to record type.
594     const RecordType *RT = getRecordType(ArgTy);
595 
596     // Now check if we index into a record type function param.
597     if(!RT && ParamIdxOk) {
598       const auto *FD = dyn_cast<FunctionDecl>(D);
599       const auto *IL = dyn_cast<IntegerLiteral>(ArgExp);
600       if(FD && IL) {
601         unsigned int NumParams = FD->getNumParams();
602         llvm::APInt ArgValue = IL->getValue();
603         uint64_t ParamIdxFromOne = ArgValue.getZExtValue();
604         uint64_t ParamIdxFromZero = ParamIdxFromOne - 1;
605         if (!ArgValue.isStrictlyPositive() || ParamIdxFromOne > NumParams) {
606           S.Diag(AL.getLoc(), diag::err_attribute_argument_out_of_range)
607               << AL << Idx + 1 << NumParams;
608           continue;
609         }
610         ArgTy = FD->getParamDecl(ParamIdxFromZero)->getType();
611       }
612     }
613 
614     // If the type does not have a capability, see if the components of the
615     // expression have capabilities. This allows for writing C code where the
616     // capability may be on the type, and the expression is a capability
617     // boolean logic expression. Eg) requires_capability(A || B && !C)
618     if (!typeHasCapability(S, ArgTy) && !isCapabilityExpr(S, ArgExp))
619       S.Diag(AL.getLoc(), diag::warn_thread_attribute_argument_not_lockable)
620           << AL << ArgTy;
621 
622     Args.push_back(ArgExp);
623   }
624 }
625 
626 //===----------------------------------------------------------------------===//
627 // Attribute Implementations
628 //===----------------------------------------------------------------------===//
629 
630 static void handlePtGuardedVarAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
631   if (!threadSafetyCheckIsPointer(S, D, AL))
632     return;
633 
634   D->addAttr(::new (S.Context)
635              PtGuardedVarAttr(AL.getRange(), S.Context,
636                               AL.getAttributeSpellingListIndex()));
637 }
638 
639 static bool checkGuardedByAttrCommon(Sema &S, Decl *D, const ParsedAttr &AL,
640                                      Expr *&Arg) {
641   SmallVector<Expr *, 1> Args;
642   // check that all arguments are lockable objects
643   checkAttrArgsAreCapabilityObjs(S, D, AL, Args);
644   unsigned Size = Args.size();
645   if (Size != 1)
646     return false;
647 
648   Arg = Args[0];
649 
650   return true;
651 }
652 
653 static void handleGuardedByAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
654   Expr *Arg = nullptr;
655   if (!checkGuardedByAttrCommon(S, D, AL, Arg))
656     return;
657 
658   D->addAttr(::new (S.Context) GuardedByAttr(
659       AL.getRange(), S.Context, Arg, AL.getAttributeSpellingListIndex()));
660 }
661 
662 static void handlePtGuardedByAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
663   Expr *Arg = nullptr;
664   if (!checkGuardedByAttrCommon(S, D, AL, Arg))
665     return;
666 
667   if (!threadSafetyCheckIsPointer(S, D, AL))
668     return;
669 
670   D->addAttr(::new (S.Context) PtGuardedByAttr(
671       AL.getRange(), S.Context, Arg, AL.getAttributeSpellingListIndex()));
672 }
673 
674 static bool checkAcquireOrderAttrCommon(Sema &S, Decl *D, const ParsedAttr &AL,
675                                         SmallVectorImpl<Expr *> &Args) {
676   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
677     return false;
678 
679   // Check that this attribute only applies to lockable types.
680   QualType QT = cast<ValueDecl>(D)->getType();
681   if (!QT->isDependentType() && !typeHasCapability(S, QT)) {
682     S.Diag(AL.getLoc(), diag::warn_thread_attribute_decl_not_lockable) << AL;
683     return false;
684   }
685 
686   // Check that all arguments are lockable objects.
687   checkAttrArgsAreCapabilityObjs(S, D, AL, Args);
688   if (Args.empty())
689     return false;
690 
691   return true;
692 }
693 
694 static void handleAcquiredAfterAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
695   SmallVector<Expr *, 1> Args;
696   if (!checkAcquireOrderAttrCommon(S, D, AL, Args))
697     return;
698 
699   Expr **StartArg = &Args[0];
700   D->addAttr(::new (S.Context) AcquiredAfterAttr(
701       AL.getRange(), S.Context, StartArg, Args.size(),
702       AL.getAttributeSpellingListIndex()));
703 }
704 
705 static void handleAcquiredBeforeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
706   SmallVector<Expr *, 1> Args;
707   if (!checkAcquireOrderAttrCommon(S, D, AL, Args))
708     return;
709 
710   Expr **StartArg = &Args[0];
711   D->addAttr(::new (S.Context) AcquiredBeforeAttr(
712       AL.getRange(), S.Context, StartArg, Args.size(),
713       AL.getAttributeSpellingListIndex()));
714 }
715 
716 static bool checkLockFunAttrCommon(Sema &S, Decl *D, const ParsedAttr &AL,
717                                    SmallVectorImpl<Expr *> &Args) {
718   // zero or more arguments ok
719   // check that all arguments are lockable objects
720   checkAttrArgsAreCapabilityObjs(S, D, AL, Args, 0, /*ParamIdxOk=*/true);
721 
722   return true;
723 }
724 
725 static void handleAssertSharedLockAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
726   SmallVector<Expr *, 1> Args;
727   if (!checkLockFunAttrCommon(S, D, AL, Args))
728     return;
729 
730   unsigned Size = Args.size();
731   Expr **StartArg = Size == 0 ? nullptr : &Args[0];
732   D->addAttr(::new (S.Context)
733                  AssertSharedLockAttr(AL.getRange(), S.Context, StartArg, Size,
734                                       AL.getAttributeSpellingListIndex()));
735 }
736 
737 static void handleAssertExclusiveLockAttr(Sema &S, Decl *D,
738                                           const ParsedAttr &AL) {
739   SmallVector<Expr *, 1> Args;
740   if (!checkLockFunAttrCommon(S, D, AL, Args))
741     return;
742 
743   unsigned Size = Args.size();
744   Expr **StartArg = Size == 0 ? nullptr : &Args[0];
745   D->addAttr(::new (S.Context) AssertExclusiveLockAttr(
746       AL.getRange(), S.Context, StartArg, Size,
747       AL.getAttributeSpellingListIndex()));
748 }
749 
750 /// Checks to be sure that the given parameter number is in bounds, and
751 /// is an integral type. Will emit appropriate diagnostics if this returns
752 /// false.
753 ///
754 /// AttrArgNo is used to actually retrieve the argument, so it's base-0.
755 template <typename AttrInfo>
756 static bool checkParamIsIntegerType(Sema &S, const FunctionDecl *FD,
757                                     const AttrInfo &AI, unsigned AttrArgNo) {
758   assert(AI.isArgExpr(AttrArgNo) && "Expected expression argument");
759   Expr *AttrArg = AI.getArgAsExpr(AttrArgNo);
760   ParamIdx Idx;
761   if (!checkFunctionOrMethodParameterIndex(S, FD, AI, AttrArgNo + 1, AttrArg,
762                                            Idx))
763     return false;
764 
765   const ParmVarDecl *Param = FD->getParamDecl(Idx.getASTIndex());
766   if (!Param->getType()->isIntegerType() && !Param->getType()->isCharType()) {
767     SourceLocation SrcLoc = AttrArg->getBeginLoc();
768     S.Diag(SrcLoc, diag::err_attribute_integers_only)
769         << AI << Param->getSourceRange();
770     return false;
771   }
772   return true;
773 }
774 
775 static void handleAllocSizeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
776   if (!checkAttributeAtLeastNumArgs(S, AL, 1) ||
777       !checkAttributeAtMostNumArgs(S, AL, 2))
778     return;
779 
780   const auto *FD = cast<FunctionDecl>(D);
781   if (!FD->getReturnType()->isPointerType()) {
782     S.Diag(AL.getLoc(), diag::warn_attribute_return_pointers_only) << AL;
783     return;
784   }
785 
786   const Expr *SizeExpr = AL.getArgAsExpr(0);
787   int SizeArgNoVal;
788   // Parameter indices are 1-indexed, hence Index=1
789   if (!checkPositiveIntArgument(S, AL, SizeExpr, SizeArgNoVal, /*Index=*/1))
790     return;
791   if (!checkParamIsIntegerType(S, FD, AL, /*AttrArgNo=*/0))
792     return;
793   ParamIdx SizeArgNo(SizeArgNoVal, D);
794 
795   ParamIdx NumberArgNo;
796   if (AL.getNumArgs() == 2) {
797     const Expr *NumberExpr = AL.getArgAsExpr(1);
798     int Val;
799     // Parameter indices are 1-based, hence Index=2
800     if (!checkPositiveIntArgument(S, AL, NumberExpr, Val, /*Index=*/2))
801       return;
802     if (!checkParamIsIntegerType(S, FD, AL, /*AttrArgNo=*/1))
803       return;
804     NumberArgNo = ParamIdx(Val, D);
805   }
806 
807   D->addAttr(::new (S.Context)
808                  AllocSizeAttr(AL.getRange(), S.Context, SizeArgNo, NumberArgNo,
809                                AL.getAttributeSpellingListIndex()));
810 }
811 
812 static bool checkTryLockFunAttrCommon(Sema &S, Decl *D, const ParsedAttr &AL,
813                                       SmallVectorImpl<Expr *> &Args) {
814   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
815     return false;
816 
817   if (!isIntOrBool(AL.getArgAsExpr(0))) {
818     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
819         << AL << 1 << AANT_ArgumentIntOrBool;
820     return false;
821   }
822 
823   // check that all arguments are lockable objects
824   checkAttrArgsAreCapabilityObjs(S, D, AL, Args, 1);
825 
826   return true;
827 }
828 
829 static void handleSharedTrylockFunctionAttr(Sema &S, Decl *D,
830                                             const ParsedAttr &AL) {
831   SmallVector<Expr*, 2> Args;
832   if (!checkTryLockFunAttrCommon(S, D, AL, Args))
833     return;
834 
835   D->addAttr(::new (S.Context) SharedTrylockFunctionAttr(
836       AL.getRange(), S.Context, AL.getArgAsExpr(0), Args.data(), Args.size(),
837       AL.getAttributeSpellingListIndex()));
838 }
839 
840 static void handleExclusiveTrylockFunctionAttr(Sema &S, Decl *D,
841                                                const ParsedAttr &AL) {
842   SmallVector<Expr*, 2> Args;
843   if (!checkTryLockFunAttrCommon(S, D, AL, Args))
844     return;
845 
846   D->addAttr(::new (S.Context) ExclusiveTrylockFunctionAttr(
847       AL.getRange(), S.Context, AL.getArgAsExpr(0), Args.data(),
848       Args.size(), AL.getAttributeSpellingListIndex()));
849 }
850 
851 static void handleLockReturnedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
852   // check that the argument is lockable object
853   SmallVector<Expr*, 1> Args;
854   checkAttrArgsAreCapabilityObjs(S, D, AL, Args);
855   unsigned Size = Args.size();
856   if (Size == 0)
857     return;
858 
859   D->addAttr(::new (S.Context)
860              LockReturnedAttr(AL.getRange(), S.Context, Args[0],
861                               AL.getAttributeSpellingListIndex()));
862 }
863 
864 static void handleLocksExcludedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
865   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
866     return;
867 
868   // check that all arguments are lockable objects
869   SmallVector<Expr*, 1> Args;
870   checkAttrArgsAreCapabilityObjs(S, D, AL, Args);
871   unsigned Size = Args.size();
872   if (Size == 0)
873     return;
874   Expr **StartArg = &Args[0];
875 
876   D->addAttr(::new (S.Context)
877              LocksExcludedAttr(AL.getRange(), S.Context, StartArg, Size,
878                                AL.getAttributeSpellingListIndex()));
879 }
880 
881 static bool checkFunctionConditionAttr(Sema &S, Decl *D, const ParsedAttr &AL,
882                                        Expr *&Cond, StringRef &Msg) {
883   Cond = AL.getArgAsExpr(0);
884   if (!Cond->isTypeDependent()) {
885     ExprResult Converted = S.PerformContextuallyConvertToBool(Cond);
886     if (Converted.isInvalid())
887       return false;
888     Cond = Converted.get();
889   }
890 
891   if (!S.checkStringLiteralArgumentAttr(AL, 1, Msg))
892     return false;
893 
894   if (Msg.empty())
895     Msg = "<no message provided>";
896 
897   SmallVector<PartialDiagnosticAt, 8> Diags;
898   if (isa<FunctionDecl>(D) && !Cond->isValueDependent() &&
899       !Expr::isPotentialConstantExprUnevaluated(Cond, cast<FunctionDecl>(D),
900                                                 Diags)) {
901     S.Diag(AL.getLoc(), diag::err_attr_cond_never_constant_expr) << AL;
902     for (const PartialDiagnosticAt &PDiag : Diags)
903       S.Diag(PDiag.first, PDiag.second);
904     return false;
905   }
906   return true;
907 }
908 
909 static void handleEnableIfAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
910   S.Diag(AL.getLoc(), diag::ext_clang_enable_if);
911 
912   Expr *Cond;
913   StringRef Msg;
914   if (checkFunctionConditionAttr(S, D, AL, Cond, Msg))
915     D->addAttr(::new (S.Context)
916                    EnableIfAttr(AL.getRange(), S.Context, Cond, Msg,
917                                 AL.getAttributeSpellingListIndex()));
918 }
919 
920 namespace {
921 /// Determines if a given Expr references any of the given function's
922 /// ParmVarDecls, or the function's implicit `this` parameter (if applicable).
923 class ArgumentDependenceChecker
924     : public RecursiveASTVisitor<ArgumentDependenceChecker> {
925 #ifndef NDEBUG
926   const CXXRecordDecl *ClassType;
927 #endif
928   llvm::SmallPtrSet<const ParmVarDecl *, 16> Parms;
929   bool Result;
930 
931 public:
932   ArgumentDependenceChecker(const FunctionDecl *FD) {
933 #ifndef NDEBUG
934     if (const auto *MD = dyn_cast<CXXMethodDecl>(FD))
935       ClassType = MD->getParent();
936     else
937       ClassType = nullptr;
938 #endif
939     Parms.insert(FD->param_begin(), FD->param_end());
940   }
941 
942   bool referencesArgs(Expr *E) {
943     Result = false;
944     TraverseStmt(E);
945     return Result;
946   }
947 
948   bool VisitCXXThisExpr(CXXThisExpr *E) {
949     assert(E->getType()->getPointeeCXXRecordDecl() == ClassType &&
950            "`this` doesn't refer to the enclosing class?");
951     Result = true;
952     return false;
953   }
954 
955   bool VisitDeclRefExpr(DeclRefExpr *DRE) {
956     if (const auto *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl()))
957       if (Parms.count(PVD)) {
958         Result = true;
959         return false;
960       }
961     return true;
962   }
963 };
964 }
965 
966 static void handleDiagnoseIfAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
967   S.Diag(AL.getLoc(), diag::ext_clang_diagnose_if);
968 
969   Expr *Cond;
970   StringRef Msg;
971   if (!checkFunctionConditionAttr(S, D, AL, Cond, Msg))
972     return;
973 
974   StringRef DiagTypeStr;
975   if (!S.checkStringLiteralArgumentAttr(AL, 2, DiagTypeStr))
976     return;
977 
978   DiagnoseIfAttr::DiagnosticType DiagType;
979   if (!DiagnoseIfAttr::ConvertStrToDiagnosticType(DiagTypeStr, DiagType)) {
980     S.Diag(AL.getArgAsExpr(2)->getBeginLoc(),
981            diag::err_diagnose_if_invalid_diagnostic_type);
982     return;
983   }
984 
985   bool ArgDependent = false;
986   if (const auto *FD = dyn_cast<FunctionDecl>(D))
987     ArgDependent = ArgumentDependenceChecker(FD).referencesArgs(Cond);
988   D->addAttr(::new (S.Context) DiagnoseIfAttr(
989       AL.getRange(), S.Context, Cond, Msg, DiagType, ArgDependent,
990       cast<NamedDecl>(D), AL.getAttributeSpellingListIndex()));
991 }
992 
993 static void handlePassObjectSizeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
994   if (D->hasAttr<PassObjectSizeAttr>()) {
995     S.Diag(D->getBeginLoc(), diag::err_attribute_only_once_per_parameter) << AL;
996     return;
997   }
998 
999   Expr *E = AL.getArgAsExpr(0);
1000   uint32_t Type;
1001   if (!checkUInt32Argument(S, AL, E, Type, /*Idx=*/1))
1002     return;
1003 
1004   // pass_object_size's argument is passed in as the second argument of
1005   // __builtin_object_size. So, it has the same constraints as that second
1006   // argument; namely, it must be in the range [0, 3].
1007   if (Type > 3) {
1008     S.Diag(E->getBeginLoc(), diag::err_attribute_argument_outof_range)
1009         << AL << 0 << 3 << E->getSourceRange();
1010     return;
1011   }
1012 
1013   // pass_object_size is only supported on constant pointer parameters; as a
1014   // kindness to users, we allow the parameter to be non-const for declarations.
1015   // At this point, we have no clue if `D` belongs to a function declaration or
1016   // definition, so we defer the constness check until later.
1017   if (!cast<ParmVarDecl>(D)->getType()->isPointerType()) {
1018     S.Diag(D->getBeginLoc(), diag::err_attribute_pointers_only) << AL << 1;
1019     return;
1020   }
1021 
1022   D->addAttr(::new (S.Context) PassObjectSizeAttr(
1023       AL.getRange(), S.Context, (int)Type, AL.getAttributeSpellingListIndex()));
1024 }
1025 
1026 static void handleConsumableAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1027   ConsumableAttr::ConsumedState DefaultState;
1028 
1029   if (AL.isArgIdent(0)) {
1030     IdentifierLoc *IL = AL.getArgAsIdent(0);
1031     if (!ConsumableAttr::ConvertStrToConsumedState(IL->Ident->getName(),
1032                                                    DefaultState)) {
1033       S.Diag(IL->Loc, diag::warn_attribute_type_not_supported) << AL
1034                                                                << IL->Ident;
1035       return;
1036     }
1037   } else {
1038     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
1039         << AL << AANT_ArgumentIdentifier;
1040     return;
1041   }
1042 
1043   D->addAttr(::new (S.Context)
1044              ConsumableAttr(AL.getRange(), S.Context, DefaultState,
1045                             AL.getAttributeSpellingListIndex()));
1046 }
1047 
1048 static bool checkForConsumableClass(Sema &S, const CXXMethodDecl *MD,
1049                                     const ParsedAttr &AL) {
1050   ASTContext &CurrContext = S.getASTContext();
1051   QualType ThisType = MD->getThisType(CurrContext)->getPointeeType();
1052 
1053   if (const CXXRecordDecl *RD = ThisType->getAsCXXRecordDecl()) {
1054     if (!RD->hasAttr<ConsumableAttr>()) {
1055       S.Diag(AL.getLoc(), diag::warn_attr_on_unconsumable_class) <<
1056         RD->getNameAsString();
1057 
1058       return false;
1059     }
1060   }
1061 
1062   return true;
1063 }
1064 
1065 static void handleCallableWhenAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1066   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
1067     return;
1068 
1069   if (!checkForConsumableClass(S, cast<CXXMethodDecl>(D), AL))
1070     return;
1071 
1072   SmallVector<CallableWhenAttr::ConsumedState, 3> States;
1073   for (unsigned ArgIndex = 0; ArgIndex < AL.getNumArgs(); ++ArgIndex) {
1074     CallableWhenAttr::ConsumedState CallableState;
1075 
1076     StringRef StateString;
1077     SourceLocation Loc;
1078     if (AL.isArgIdent(ArgIndex)) {
1079       IdentifierLoc *Ident = AL.getArgAsIdent(ArgIndex);
1080       StateString = Ident->Ident->getName();
1081       Loc = Ident->Loc;
1082     } else {
1083       if (!S.checkStringLiteralArgumentAttr(AL, ArgIndex, StateString, &Loc))
1084         return;
1085     }
1086 
1087     if (!CallableWhenAttr::ConvertStrToConsumedState(StateString,
1088                                                      CallableState)) {
1089       S.Diag(Loc, diag::warn_attribute_type_not_supported) << AL << StateString;
1090       return;
1091     }
1092 
1093     States.push_back(CallableState);
1094   }
1095 
1096   D->addAttr(::new (S.Context)
1097              CallableWhenAttr(AL.getRange(), S.Context, States.data(),
1098                States.size(), AL.getAttributeSpellingListIndex()));
1099 }
1100 
1101 static void handleParamTypestateAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1102   ParamTypestateAttr::ConsumedState ParamState;
1103 
1104   if (AL.isArgIdent(0)) {
1105     IdentifierLoc *Ident = AL.getArgAsIdent(0);
1106     StringRef StateString = Ident->Ident->getName();
1107 
1108     if (!ParamTypestateAttr::ConvertStrToConsumedState(StateString,
1109                                                        ParamState)) {
1110       S.Diag(Ident->Loc, diag::warn_attribute_type_not_supported)
1111           << AL << StateString;
1112       return;
1113     }
1114   } else {
1115     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
1116         << AL << AANT_ArgumentIdentifier;
1117     return;
1118   }
1119 
1120   // FIXME: This check is currently being done in the analysis.  It can be
1121   //        enabled here only after the parser propagates attributes at
1122   //        template specialization definition, not declaration.
1123   //QualType ReturnType = cast<ParmVarDecl>(D)->getType();
1124   //const CXXRecordDecl *RD = ReturnType->getAsCXXRecordDecl();
1125   //
1126   //if (!RD || !RD->hasAttr<ConsumableAttr>()) {
1127   //    S.Diag(AL.getLoc(), diag::warn_return_state_for_unconsumable_type) <<
1128   //      ReturnType.getAsString();
1129   //    return;
1130   //}
1131 
1132   D->addAttr(::new (S.Context)
1133              ParamTypestateAttr(AL.getRange(), S.Context, ParamState,
1134                                 AL.getAttributeSpellingListIndex()));
1135 }
1136 
1137 static void handleReturnTypestateAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1138   ReturnTypestateAttr::ConsumedState ReturnState;
1139 
1140   if (AL.isArgIdent(0)) {
1141     IdentifierLoc *IL = AL.getArgAsIdent(0);
1142     if (!ReturnTypestateAttr::ConvertStrToConsumedState(IL->Ident->getName(),
1143                                                         ReturnState)) {
1144       S.Diag(IL->Loc, diag::warn_attribute_type_not_supported) << AL
1145                                                                << IL->Ident;
1146       return;
1147     }
1148   } else {
1149     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
1150         << AL << AANT_ArgumentIdentifier;
1151     return;
1152   }
1153 
1154   // FIXME: This check is currently being done in the analysis.  It can be
1155   //        enabled here only after the parser propagates attributes at
1156   //        template specialization definition, not declaration.
1157   //QualType ReturnType;
1158   //
1159   //if (const ParmVarDecl *Param = dyn_cast<ParmVarDecl>(D)) {
1160   //  ReturnType = Param->getType();
1161   //
1162   //} else if (const CXXConstructorDecl *Constructor =
1163   //             dyn_cast<CXXConstructorDecl>(D)) {
1164   //  ReturnType = Constructor->getThisType(S.getASTContext())->getPointeeType();
1165   //
1166   //} else {
1167   //
1168   //  ReturnType = cast<FunctionDecl>(D)->getCallResultType();
1169   //}
1170   //
1171   //const CXXRecordDecl *RD = ReturnType->getAsCXXRecordDecl();
1172   //
1173   //if (!RD || !RD->hasAttr<ConsumableAttr>()) {
1174   //    S.Diag(Attr.getLoc(), diag::warn_return_state_for_unconsumable_type) <<
1175   //      ReturnType.getAsString();
1176   //    return;
1177   //}
1178 
1179   D->addAttr(::new (S.Context)
1180                  ReturnTypestateAttr(AL.getRange(), S.Context, ReturnState,
1181                                      AL.getAttributeSpellingListIndex()));
1182 }
1183 
1184 static void handleSetTypestateAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1185   if (!checkForConsumableClass(S, cast<CXXMethodDecl>(D), AL))
1186     return;
1187 
1188   SetTypestateAttr::ConsumedState NewState;
1189   if (AL.isArgIdent(0)) {
1190     IdentifierLoc *Ident = AL.getArgAsIdent(0);
1191     StringRef Param = Ident->Ident->getName();
1192     if (!SetTypestateAttr::ConvertStrToConsumedState(Param, NewState)) {
1193       S.Diag(Ident->Loc, diag::warn_attribute_type_not_supported) << AL
1194                                                                   << Param;
1195       return;
1196     }
1197   } else {
1198     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
1199         << AL << AANT_ArgumentIdentifier;
1200     return;
1201   }
1202 
1203   D->addAttr(::new (S.Context)
1204              SetTypestateAttr(AL.getRange(), S.Context, NewState,
1205                               AL.getAttributeSpellingListIndex()));
1206 }
1207 
1208 static void handleTestTypestateAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1209   if (!checkForConsumableClass(S, cast<CXXMethodDecl>(D), AL))
1210     return;
1211 
1212   TestTypestateAttr::ConsumedState TestState;
1213   if (AL.isArgIdent(0)) {
1214     IdentifierLoc *Ident = AL.getArgAsIdent(0);
1215     StringRef Param = Ident->Ident->getName();
1216     if (!TestTypestateAttr::ConvertStrToConsumedState(Param, TestState)) {
1217       S.Diag(Ident->Loc, diag::warn_attribute_type_not_supported) << AL
1218                                                                   << Param;
1219       return;
1220     }
1221   } else {
1222     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
1223         << AL << AANT_ArgumentIdentifier;
1224     return;
1225   }
1226 
1227   D->addAttr(::new (S.Context)
1228              TestTypestateAttr(AL.getRange(), S.Context, TestState,
1229                                 AL.getAttributeSpellingListIndex()));
1230 }
1231 
1232 static void handleExtVectorTypeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1233   // Remember this typedef decl, we will need it later for diagnostics.
1234   S.ExtVectorDecls.push_back(cast<TypedefNameDecl>(D));
1235 }
1236 
1237 static void handlePackedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1238   if (auto *TD = dyn_cast<TagDecl>(D))
1239     TD->addAttr(::new (S.Context) PackedAttr(AL.getRange(), S.Context,
1240                                         AL.getAttributeSpellingListIndex()));
1241   else if (auto *FD = dyn_cast<FieldDecl>(D)) {
1242     bool BitfieldByteAligned = (!FD->getType()->isDependentType() &&
1243                                 !FD->getType()->isIncompleteType() &&
1244                                 FD->isBitField() &&
1245                                 S.Context.getTypeAlign(FD->getType()) <= 8);
1246 
1247     if (S.getASTContext().getTargetInfo().getTriple().isPS4()) {
1248       if (BitfieldByteAligned)
1249         // The PS4 target needs to maintain ABI backwards compatibility.
1250         S.Diag(AL.getLoc(), diag::warn_attribute_ignored_for_field_of_type)
1251             << AL << FD->getType();
1252       else
1253         FD->addAttr(::new (S.Context) PackedAttr(
1254                     AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
1255     } else {
1256       // Report warning about changed offset in the newer compiler versions.
1257       if (BitfieldByteAligned)
1258         S.Diag(AL.getLoc(), diag::warn_attribute_packed_for_bitfield);
1259 
1260       FD->addAttr(::new (S.Context) PackedAttr(
1261                   AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
1262     }
1263 
1264   } else
1265     S.Diag(AL.getLoc(), diag::warn_attribute_ignored) << AL;
1266 }
1267 
1268 static bool checkIBOutletCommon(Sema &S, Decl *D, const ParsedAttr &AL) {
1269   // The IBOutlet/IBOutletCollection attributes only apply to instance
1270   // variables or properties of Objective-C classes.  The outlet must also
1271   // have an object reference type.
1272   if (const auto *VD = dyn_cast<ObjCIvarDecl>(D)) {
1273     if (!VD->getType()->getAs<ObjCObjectPointerType>()) {
1274       S.Diag(AL.getLoc(), diag::warn_iboutlet_object_type)
1275           << AL << VD->getType() << 0;
1276       return false;
1277     }
1278   }
1279   else if (const auto *PD = dyn_cast<ObjCPropertyDecl>(D)) {
1280     if (!PD->getType()->getAs<ObjCObjectPointerType>()) {
1281       S.Diag(AL.getLoc(), diag::warn_iboutlet_object_type)
1282           << AL << PD->getType() << 1;
1283       return false;
1284     }
1285   }
1286   else {
1287     S.Diag(AL.getLoc(), diag::warn_attribute_iboutlet) << AL;
1288     return false;
1289   }
1290 
1291   return true;
1292 }
1293 
1294 static void handleIBOutlet(Sema &S, Decl *D, const ParsedAttr &AL) {
1295   if (!checkIBOutletCommon(S, D, AL))
1296     return;
1297 
1298   D->addAttr(::new (S.Context)
1299              IBOutletAttr(AL.getRange(), S.Context,
1300                           AL.getAttributeSpellingListIndex()));
1301 }
1302 
1303 static void handleIBOutletCollection(Sema &S, Decl *D, const ParsedAttr &AL) {
1304 
1305   // The iboutletcollection attribute can have zero or one arguments.
1306   if (AL.getNumArgs() > 1) {
1307     S.Diag(AL.getLoc(), diag::err_attribute_wrong_number_arguments) << AL << 1;
1308     return;
1309   }
1310 
1311   if (!checkIBOutletCommon(S, D, AL))
1312     return;
1313 
1314   ParsedType PT;
1315 
1316   if (AL.hasParsedType())
1317     PT = AL.getTypeArg();
1318   else {
1319     PT = S.getTypeName(S.Context.Idents.get("NSObject"), AL.getLoc(),
1320                        S.getScopeForContext(D->getDeclContext()->getParent()));
1321     if (!PT) {
1322       S.Diag(AL.getLoc(), diag::err_iboutletcollection_type) << "NSObject";
1323       return;
1324     }
1325   }
1326 
1327   TypeSourceInfo *QTLoc = nullptr;
1328   QualType QT = S.GetTypeFromParser(PT, &QTLoc);
1329   if (!QTLoc)
1330     QTLoc = S.Context.getTrivialTypeSourceInfo(QT, AL.getLoc());
1331 
1332   // Diagnose use of non-object type in iboutletcollection attribute.
1333   // FIXME. Gnu attribute extension ignores use of builtin types in
1334   // attributes. So, __attribute__((iboutletcollection(char))) will be
1335   // treated as __attribute__((iboutletcollection())).
1336   if (!QT->isObjCIdType() && !QT->isObjCObjectType()) {
1337     S.Diag(AL.getLoc(),
1338            QT->isBuiltinType() ? diag::err_iboutletcollection_builtintype
1339                                : diag::err_iboutletcollection_type) << QT;
1340     return;
1341   }
1342 
1343   D->addAttr(::new (S.Context)
1344              IBOutletCollectionAttr(AL.getRange(), S.Context, QTLoc,
1345                                     AL.getAttributeSpellingListIndex()));
1346 }
1347 
1348 bool Sema::isValidPointerAttrType(QualType T, bool RefOkay) {
1349   if (RefOkay) {
1350     if (T->isReferenceType())
1351       return true;
1352   } else {
1353     T = T.getNonReferenceType();
1354   }
1355 
1356   // The nonnull attribute, and other similar attributes, can be applied to a
1357   // transparent union that contains a pointer type.
1358   if (const RecordType *UT = T->getAsUnionType()) {
1359     if (UT && UT->getDecl()->hasAttr<TransparentUnionAttr>()) {
1360       RecordDecl *UD = UT->getDecl();
1361       for (const auto *I : UD->fields()) {
1362         QualType QT = I->getType();
1363         if (QT->isAnyPointerType() || QT->isBlockPointerType())
1364           return true;
1365       }
1366     }
1367   }
1368 
1369   return T->isAnyPointerType() || T->isBlockPointerType();
1370 }
1371 
1372 static bool attrNonNullArgCheck(Sema &S, QualType T, const ParsedAttr &AL,
1373                                 SourceRange AttrParmRange,
1374                                 SourceRange TypeRange,
1375                                 bool isReturnValue = false) {
1376   if (!S.isValidPointerAttrType(T)) {
1377     if (isReturnValue)
1378       S.Diag(AL.getLoc(), diag::warn_attribute_return_pointers_only)
1379           << AL << AttrParmRange << TypeRange;
1380     else
1381       S.Diag(AL.getLoc(), diag::warn_attribute_pointers_only)
1382           << AL << AttrParmRange << TypeRange << 0;
1383     return false;
1384   }
1385   return true;
1386 }
1387 
1388 static void handleNonNullAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1389   SmallVector<ParamIdx, 8> NonNullArgs;
1390   for (unsigned I = 0; I < AL.getNumArgs(); ++I) {
1391     Expr *Ex = AL.getArgAsExpr(I);
1392     ParamIdx Idx;
1393     if (!checkFunctionOrMethodParameterIndex(S, D, AL, I + 1, Ex, Idx))
1394       return;
1395 
1396     // Is the function argument a pointer type?
1397     if (Idx.getASTIndex() < getFunctionOrMethodNumParams(D) &&
1398         !attrNonNullArgCheck(
1399             S, getFunctionOrMethodParamType(D, Idx.getASTIndex()), AL,
1400             Ex->getSourceRange(),
1401             getFunctionOrMethodParamRange(D, Idx.getASTIndex())))
1402       continue;
1403 
1404     NonNullArgs.push_back(Idx);
1405   }
1406 
1407   // If no arguments were specified to __attribute__((nonnull)) then all pointer
1408   // arguments have a nonnull attribute; warn if there aren't any. Skip this
1409   // check if the attribute came from a macro expansion or a template
1410   // instantiation.
1411   if (NonNullArgs.empty() && AL.getLoc().isFileID() &&
1412       !S.inTemplateInstantiation()) {
1413     bool AnyPointers = isFunctionOrMethodVariadic(D);
1414     for (unsigned I = 0, E = getFunctionOrMethodNumParams(D);
1415          I != E && !AnyPointers; ++I) {
1416       QualType T = getFunctionOrMethodParamType(D, I);
1417       if (T->isDependentType() || S.isValidPointerAttrType(T))
1418         AnyPointers = true;
1419     }
1420 
1421     if (!AnyPointers)
1422       S.Diag(AL.getLoc(), diag::warn_attribute_nonnull_no_pointers);
1423   }
1424 
1425   ParamIdx *Start = NonNullArgs.data();
1426   unsigned Size = NonNullArgs.size();
1427   llvm::array_pod_sort(Start, Start + Size);
1428   D->addAttr(::new (S.Context)
1429                  NonNullAttr(AL.getRange(), S.Context, Start, Size,
1430                              AL.getAttributeSpellingListIndex()));
1431 }
1432 
1433 static void handleNonNullAttrParameter(Sema &S, ParmVarDecl *D,
1434                                        const ParsedAttr &AL) {
1435   if (AL.getNumArgs() > 0) {
1436     if (D->getFunctionType()) {
1437       handleNonNullAttr(S, D, AL);
1438     } else {
1439       S.Diag(AL.getLoc(), diag::warn_attribute_nonnull_parm_no_args)
1440         << D->getSourceRange();
1441     }
1442     return;
1443   }
1444 
1445   // Is the argument a pointer type?
1446   if (!attrNonNullArgCheck(S, D->getType(), AL, SourceRange(),
1447                            D->getSourceRange()))
1448     return;
1449 
1450   D->addAttr(::new (S.Context)
1451                  NonNullAttr(AL.getRange(), S.Context, nullptr, 0,
1452                              AL.getAttributeSpellingListIndex()));
1453 }
1454 
1455 static void handleReturnsNonNullAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1456   QualType ResultType = getFunctionOrMethodResultType(D);
1457   SourceRange SR = getFunctionOrMethodResultSourceRange(D);
1458   if (!attrNonNullArgCheck(S, ResultType, AL, SourceRange(), SR,
1459                            /* isReturnValue */ true))
1460     return;
1461 
1462   D->addAttr(::new (S.Context)
1463             ReturnsNonNullAttr(AL.getRange(), S.Context,
1464                                AL.getAttributeSpellingListIndex()));
1465 }
1466 
1467 static void handleNoEscapeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1468   if (D->isInvalidDecl())
1469     return;
1470 
1471   // noescape only applies to pointer types.
1472   QualType T = cast<ParmVarDecl>(D)->getType();
1473   if (!S.isValidPointerAttrType(T, /* RefOkay */ true)) {
1474     S.Diag(AL.getLoc(), diag::warn_attribute_pointers_only)
1475         << AL << AL.getRange() << 0;
1476     return;
1477   }
1478 
1479   D->addAttr(::new (S.Context) NoEscapeAttr(
1480       AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
1481 }
1482 
1483 static void handleAssumeAlignedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1484   Expr *E = AL.getArgAsExpr(0),
1485        *OE = AL.getNumArgs() > 1 ? AL.getArgAsExpr(1) : nullptr;
1486   S.AddAssumeAlignedAttr(AL.getRange(), D, E, OE,
1487                          AL.getAttributeSpellingListIndex());
1488 }
1489 
1490 static void handleAllocAlignAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1491   S.AddAllocAlignAttr(AL.getRange(), D, AL.getArgAsExpr(0),
1492                       AL.getAttributeSpellingListIndex());
1493 }
1494 
1495 void Sema::AddAssumeAlignedAttr(SourceRange AttrRange, Decl *D, Expr *E,
1496                                 Expr *OE, unsigned SpellingListIndex) {
1497   QualType ResultType = getFunctionOrMethodResultType(D);
1498   SourceRange SR = getFunctionOrMethodResultSourceRange(D);
1499 
1500   AssumeAlignedAttr TmpAttr(AttrRange, Context, E, OE, SpellingListIndex);
1501   SourceLocation AttrLoc = AttrRange.getBegin();
1502 
1503   if (!isValidPointerAttrType(ResultType, /* RefOkay */ true)) {
1504     Diag(AttrLoc, diag::warn_attribute_return_pointers_refs_only)
1505       << &TmpAttr << AttrRange << SR;
1506     return;
1507   }
1508 
1509   if (!E->isValueDependent()) {
1510     llvm::APSInt I(64);
1511     if (!E->isIntegerConstantExpr(I, Context)) {
1512       if (OE)
1513         Diag(AttrLoc, diag::err_attribute_argument_n_type)
1514           << &TmpAttr << 1 << AANT_ArgumentIntegerConstant
1515           << E->getSourceRange();
1516       else
1517         Diag(AttrLoc, diag::err_attribute_argument_type)
1518           << &TmpAttr << AANT_ArgumentIntegerConstant
1519           << E->getSourceRange();
1520       return;
1521     }
1522 
1523     if (!I.isPowerOf2()) {
1524       Diag(AttrLoc, diag::err_alignment_not_power_of_two)
1525         << E->getSourceRange();
1526       return;
1527     }
1528   }
1529 
1530   if (OE) {
1531     if (!OE->isValueDependent()) {
1532       llvm::APSInt I(64);
1533       if (!OE->isIntegerConstantExpr(I, Context)) {
1534         Diag(AttrLoc, diag::err_attribute_argument_n_type)
1535           << &TmpAttr << 2 << AANT_ArgumentIntegerConstant
1536           << OE->getSourceRange();
1537         return;
1538       }
1539     }
1540   }
1541 
1542   D->addAttr(::new (Context)
1543             AssumeAlignedAttr(AttrRange, Context, E, OE, SpellingListIndex));
1544 }
1545 
1546 void Sema::AddAllocAlignAttr(SourceRange AttrRange, Decl *D, Expr *ParamExpr,
1547                              unsigned SpellingListIndex) {
1548   QualType ResultType = getFunctionOrMethodResultType(D);
1549 
1550   AllocAlignAttr TmpAttr(AttrRange, Context, ParamIdx(), SpellingListIndex);
1551   SourceLocation AttrLoc = AttrRange.getBegin();
1552 
1553   if (!ResultType->isDependentType() &&
1554       !isValidPointerAttrType(ResultType, /* RefOkay */ true)) {
1555     Diag(AttrLoc, diag::warn_attribute_return_pointers_refs_only)
1556         << &TmpAttr << AttrRange << getFunctionOrMethodResultSourceRange(D);
1557     return;
1558   }
1559 
1560   ParamIdx Idx;
1561   const auto *FuncDecl = cast<FunctionDecl>(D);
1562   if (!checkFunctionOrMethodParameterIndex(*this, FuncDecl, TmpAttr,
1563                                            /*AttrArgNo=*/1, ParamExpr, Idx))
1564     return;
1565 
1566   QualType Ty = getFunctionOrMethodParamType(D, Idx.getASTIndex());
1567   if (!Ty->isDependentType() && !Ty->isIntegralType(Context)) {
1568     Diag(ParamExpr->getBeginLoc(), diag::err_attribute_integers_only)
1569         << &TmpAttr
1570         << FuncDecl->getParamDecl(Idx.getASTIndex())->getSourceRange();
1571     return;
1572   }
1573 
1574   D->addAttr(::new (Context)
1575                  AllocAlignAttr(AttrRange, Context, Idx, SpellingListIndex));
1576 }
1577 
1578 /// Normalize the attribute, __foo__ becomes foo.
1579 /// Returns true if normalization was applied.
1580 static bool normalizeName(StringRef &AttrName) {
1581   if (AttrName.size() > 4 && AttrName.startswith("__") &&
1582       AttrName.endswith("__")) {
1583     AttrName = AttrName.drop_front(2).drop_back(2);
1584     return true;
1585   }
1586   return false;
1587 }
1588 
1589 static void handleOwnershipAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1590   // This attribute must be applied to a function declaration. The first
1591   // argument to the attribute must be an identifier, the name of the resource,
1592   // for example: malloc. The following arguments must be argument indexes, the
1593   // arguments must be of integer type for Returns, otherwise of pointer type.
1594   // The difference between Holds and Takes is that a pointer may still be used
1595   // after being held. free() should be __attribute((ownership_takes)), whereas
1596   // a list append function may well be __attribute((ownership_holds)).
1597 
1598   if (!AL.isArgIdent(0)) {
1599     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
1600         << AL << 1 << AANT_ArgumentIdentifier;
1601     return;
1602   }
1603 
1604   // Figure out our Kind.
1605   OwnershipAttr::OwnershipKind K =
1606       OwnershipAttr(AL.getLoc(), S.Context, nullptr, nullptr, 0,
1607                     AL.getAttributeSpellingListIndex()).getOwnKind();
1608 
1609   // Check arguments.
1610   switch (K) {
1611   case OwnershipAttr::Takes:
1612   case OwnershipAttr::Holds:
1613     if (AL.getNumArgs() < 2) {
1614       S.Diag(AL.getLoc(), diag::err_attribute_too_few_arguments) << AL << 2;
1615       return;
1616     }
1617     break;
1618   case OwnershipAttr::Returns:
1619     if (AL.getNumArgs() > 2) {
1620       S.Diag(AL.getLoc(), diag::err_attribute_too_many_arguments) << AL << 1;
1621       return;
1622     }
1623     break;
1624   }
1625 
1626   IdentifierInfo *Module = AL.getArgAsIdent(0)->Ident;
1627 
1628   StringRef ModuleName = Module->getName();
1629   if (normalizeName(ModuleName)) {
1630     Module = &S.PP.getIdentifierTable().get(ModuleName);
1631   }
1632 
1633   SmallVector<ParamIdx, 8> OwnershipArgs;
1634   for (unsigned i = 1; i < AL.getNumArgs(); ++i) {
1635     Expr *Ex = AL.getArgAsExpr(i);
1636     ParamIdx Idx;
1637     if (!checkFunctionOrMethodParameterIndex(S, D, AL, i, Ex, Idx))
1638       return;
1639 
1640     // Is the function argument a pointer type?
1641     QualType T = getFunctionOrMethodParamType(D, Idx.getASTIndex());
1642     int Err = -1;  // No error
1643     switch (K) {
1644       case OwnershipAttr::Takes:
1645       case OwnershipAttr::Holds:
1646         if (!T->isAnyPointerType() && !T->isBlockPointerType())
1647           Err = 0;
1648         break;
1649       case OwnershipAttr::Returns:
1650         if (!T->isIntegerType())
1651           Err = 1;
1652         break;
1653     }
1654     if (-1 != Err) {
1655       S.Diag(AL.getLoc(), diag::err_ownership_type) << AL << Err
1656                                                     << Ex->getSourceRange();
1657       return;
1658     }
1659 
1660     // Check we don't have a conflict with another ownership attribute.
1661     for (const auto *I : D->specific_attrs<OwnershipAttr>()) {
1662       // Cannot have two ownership attributes of different kinds for the same
1663       // index.
1664       if (I->getOwnKind() != K && I->args_end() !=
1665           std::find(I->args_begin(), I->args_end(), Idx)) {
1666         S.Diag(AL.getLoc(), diag::err_attributes_are_not_compatible) << AL << I;
1667         return;
1668       } else if (K == OwnershipAttr::Returns &&
1669                  I->getOwnKind() == OwnershipAttr::Returns) {
1670         // A returns attribute conflicts with any other returns attribute using
1671         // a different index.
1672         if (std::find(I->args_begin(), I->args_end(), Idx) == I->args_end()) {
1673           S.Diag(I->getLocation(), diag::err_ownership_returns_index_mismatch)
1674               << I->args_begin()->getSourceIndex();
1675           if (I->args_size())
1676             S.Diag(AL.getLoc(), diag::note_ownership_returns_index_mismatch)
1677                 << Idx.getSourceIndex() << Ex->getSourceRange();
1678           return;
1679         }
1680       }
1681     }
1682     OwnershipArgs.push_back(Idx);
1683   }
1684 
1685   ParamIdx *Start = OwnershipArgs.data();
1686   unsigned Size = OwnershipArgs.size();
1687   llvm::array_pod_sort(Start, Start + Size);
1688   D->addAttr(::new (S.Context)
1689                  OwnershipAttr(AL.getLoc(), S.Context, Module, Start, Size,
1690                                AL.getAttributeSpellingListIndex()));
1691 }
1692 
1693 static void handleWeakRefAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1694   // Check the attribute arguments.
1695   if (AL.getNumArgs() > 1) {
1696     S.Diag(AL.getLoc(), diag::err_attribute_wrong_number_arguments) << AL << 1;
1697     return;
1698   }
1699 
1700   // gcc rejects
1701   // class c {
1702   //   static int a __attribute__((weakref ("v2")));
1703   //   static int b() __attribute__((weakref ("f3")));
1704   // };
1705   // and ignores the attributes of
1706   // void f(void) {
1707   //   static int a __attribute__((weakref ("v2")));
1708   // }
1709   // we reject them
1710   const DeclContext *Ctx = D->getDeclContext()->getRedeclContext();
1711   if (!Ctx->isFileContext()) {
1712     S.Diag(AL.getLoc(), diag::err_attribute_weakref_not_global_context)
1713         << cast<NamedDecl>(D);
1714     return;
1715   }
1716 
1717   // The GCC manual says
1718   //
1719   // At present, a declaration to which `weakref' is attached can only
1720   // be `static'.
1721   //
1722   // It also says
1723   //
1724   // Without a TARGET,
1725   // given as an argument to `weakref' or to `alias', `weakref' is
1726   // equivalent to `weak'.
1727   //
1728   // gcc 4.4.1 will accept
1729   // int a7 __attribute__((weakref));
1730   // as
1731   // int a7 __attribute__((weak));
1732   // This looks like a bug in gcc. We reject that for now. We should revisit
1733   // it if this behaviour is actually used.
1734 
1735   // GCC rejects
1736   // static ((alias ("y"), weakref)).
1737   // Should we? How to check that weakref is before or after alias?
1738 
1739   // FIXME: it would be good for us to keep the WeakRefAttr as-written instead
1740   // of transforming it into an AliasAttr.  The WeakRefAttr never uses the
1741   // StringRef parameter it was given anyway.
1742   StringRef Str;
1743   if (AL.getNumArgs() && S.checkStringLiteralArgumentAttr(AL, 0, Str))
1744     // GCC will accept anything as the argument of weakref. Should we
1745     // check for an existing decl?
1746     D->addAttr(::new (S.Context) AliasAttr(AL.getRange(), S.Context, Str,
1747                                         AL.getAttributeSpellingListIndex()));
1748 
1749   D->addAttr(::new (S.Context)
1750              WeakRefAttr(AL.getRange(), S.Context,
1751                          AL.getAttributeSpellingListIndex()));
1752 }
1753 
1754 static void handleIFuncAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1755   StringRef Str;
1756   if (!S.checkStringLiteralArgumentAttr(AL, 0, Str))
1757     return;
1758 
1759   // Aliases should be on declarations, not definitions.
1760   const auto *FD = cast<FunctionDecl>(D);
1761   if (FD->isThisDeclarationADefinition()) {
1762     S.Diag(AL.getLoc(), diag::err_alias_is_definition) << FD << 1;
1763     return;
1764   }
1765 
1766   D->addAttr(::new (S.Context) IFuncAttr(AL.getRange(), S.Context, Str,
1767                                          AL.getAttributeSpellingListIndex()));
1768 }
1769 
1770 static void handleAliasAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1771   StringRef Str;
1772   if (!S.checkStringLiteralArgumentAttr(AL, 0, Str))
1773     return;
1774 
1775   if (S.Context.getTargetInfo().getTriple().isOSDarwin()) {
1776     S.Diag(AL.getLoc(), diag::err_alias_not_supported_on_darwin);
1777     return;
1778   }
1779   if (S.Context.getTargetInfo().getTriple().isNVPTX()) {
1780     S.Diag(AL.getLoc(), diag::err_alias_not_supported_on_nvptx);
1781   }
1782 
1783   // Aliases should be on declarations, not definitions.
1784   if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
1785     if (FD->isThisDeclarationADefinition()) {
1786       S.Diag(AL.getLoc(), diag::err_alias_is_definition) << FD << 0;
1787       return;
1788     }
1789   } else {
1790     const auto *VD = cast<VarDecl>(D);
1791     if (VD->isThisDeclarationADefinition() && VD->isExternallyVisible()) {
1792       S.Diag(AL.getLoc(), diag::err_alias_is_definition) << VD << 0;
1793       return;
1794     }
1795   }
1796 
1797   // FIXME: check if target symbol exists in current file
1798 
1799   D->addAttr(::new (S.Context) AliasAttr(AL.getRange(), S.Context, Str,
1800                                          AL.getAttributeSpellingListIndex()));
1801 }
1802 
1803 static void handleTLSModelAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1804   StringRef Model;
1805   SourceLocation LiteralLoc;
1806   // Check that it is a string.
1807   if (!S.checkStringLiteralArgumentAttr(AL, 0, Model, &LiteralLoc))
1808     return;
1809 
1810   // Check that the value.
1811   if (Model != "global-dynamic" && Model != "local-dynamic"
1812       && Model != "initial-exec" && Model != "local-exec") {
1813     S.Diag(LiteralLoc, diag::err_attr_tlsmodel_arg);
1814     return;
1815   }
1816 
1817   D->addAttr(::new (S.Context)
1818              TLSModelAttr(AL.getRange(), S.Context, Model,
1819                           AL.getAttributeSpellingListIndex()));
1820 }
1821 
1822 static void handleRestrictAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1823   QualType ResultType = getFunctionOrMethodResultType(D);
1824   if (ResultType->isAnyPointerType() || ResultType->isBlockPointerType()) {
1825     D->addAttr(::new (S.Context) RestrictAttr(
1826         AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
1827     return;
1828   }
1829 
1830   S.Diag(AL.getLoc(), diag::warn_attribute_return_pointers_only)
1831       << AL << getFunctionOrMethodResultSourceRange(D);
1832 }
1833 
1834 static void handleCPUSpecificAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1835   FunctionDecl *FD = cast<FunctionDecl>(D);
1836   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
1837     return;
1838 
1839   SmallVector<IdentifierInfo *, 8> CPUs;
1840   for (unsigned ArgNo = 0; ArgNo < getNumAttributeArgs(AL); ++ArgNo) {
1841     if (!AL.isArgIdent(ArgNo)) {
1842       S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
1843           << AL << AANT_ArgumentIdentifier;
1844       return;
1845     }
1846 
1847     IdentifierLoc *CPUArg = AL.getArgAsIdent(ArgNo);
1848     StringRef CPUName = CPUArg->Ident->getName().trim();
1849 
1850     if (!S.Context.getTargetInfo().validateCPUSpecificCPUDispatch(CPUName)) {
1851       S.Diag(CPUArg->Loc, diag::err_invalid_cpu_specific_dispatch_value)
1852           << CPUName << (AL.getKind() == ParsedAttr::AT_CPUDispatch);
1853       return;
1854     }
1855 
1856     const TargetInfo &Target = S.Context.getTargetInfo();
1857     if (llvm::any_of(CPUs, [CPUName, &Target](const IdentifierInfo *Cur) {
1858           return Target.CPUSpecificManglingCharacter(CPUName) ==
1859                  Target.CPUSpecificManglingCharacter(Cur->getName());
1860         })) {
1861       S.Diag(AL.getLoc(), diag::warn_multiversion_duplicate_entries);
1862       return;
1863     }
1864     CPUs.push_back(CPUArg->Ident);
1865   }
1866 
1867   FD->setIsMultiVersion(true);
1868   if (AL.getKind() == ParsedAttr::AT_CPUSpecific)
1869     D->addAttr(::new (S.Context) CPUSpecificAttr(
1870         AL.getRange(), S.Context, CPUs.data(), CPUs.size(),
1871         AL.getAttributeSpellingListIndex()));
1872   else
1873     D->addAttr(::new (S.Context) CPUDispatchAttr(
1874         AL.getRange(), S.Context, CPUs.data(), CPUs.size(),
1875         AL.getAttributeSpellingListIndex()));
1876 }
1877 
1878 static void handleCommonAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1879   if (S.LangOpts.CPlusPlus) {
1880     S.Diag(AL.getLoc(), diag::err_attribute_not_supported_in_lang)
1881         << AL << AttributeLangSupport::Cpp;
1882     return;
1883   }
1884 
1885   if (CommonAttr *CA = S.mergeCommonAttr(D, AL))
1886     D->addAttr(CA);
1887 }
1888 
1889 static void handleNakedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1890   if (checkAttrMutualExclusion<DisableTailCallsAttr>(S, D, AL))
1891     return;
1892 
1893   if (AL.isDeclspecAttribute()) {
1894     const auto &Triple = S.getASTContext().getTargetInfo().getTriple();
1895     const auto &Arch = Triple.getArch();
1896     if (Arch != llvm::Triple::x86 &&
1897         (Arch != llvm::Triple::arm && Arch != llvm::Triple::thumb)) {
1898       S.Diag(AL.getLoc(), diag::err_attribute_not_supported_on_arch)
1899           << AL << Triple.getArchName();
1900       return;
1901     }
1902   }
1903 
1904   D->addAttr(::new (S.Context) NakedAttr(AL.getRange(), S.Context,
1905                                          AL.getAttributeSpellingListIndex()));
1906 }
1907 
1908 static void handleNoReturnAttr(Sema &S, Decl *D, const ParsedAttr &Attrs) {
1909   if (hasDeclarator(D)) return;
1910 
1911   if (!isa<ObjCMethodDecl>(D)) {
1912     S.Diag(Attrs.getLoc(), diag::warn_attribute_wrong_decl_type)
1913         << Attrs << ExpectedFunctionOrMethod;
1914     return;
1915   }
1916 
1917   D->addAttr(::new (S.Context) NoReturnAttr(
1918       Attrs.getRange(), S.Context, Attrs.getAttributeSpellingListIndex()));
1919 }
1920 
1921 static void handleNoCfCheckAttr(Sema &S, Decl *D, const ParsedAttr &Attrs) {
1922   if (!S.getLangOpts().CFProtectionBranch)
1923     S.Diag(Attrs.getLoc(), diag::warn_nocf_check_attribute_ignored);
1924   else
1925     handleSimpleAttribute<AnyX86NoCfCheckAttr>(S, D, Attrs);
1926 }
1927 
1928 bool Sema::CheckAttrNoArgs(const ParsedAttr &Attrs) {
1929   if (!checkAttributeNumArgs(*this, Attrs, 0)) {
1930     Attrs.setInvalid();
1931     return true;
1932   }
1933 
1934   return false;
1935 }
1936 
1937 bool Sema::CheckAttrTarget(const ParsedAttr &AL) {
1938   // Check whether the attribute is valid on the current target.
1939   if (!AL.existsInTarget(Context.getTargetInfo())) {
1940     Diag(AL.getLoc(), diag::warn_unknown_attribute_ignored) << AL;
1941     AL.setInvalid();
1942     return true;
1943   }
1944 
1945   return false;
1946 }
1947 
1948 static void handleAnalyzerNoReturnAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1949 
1950   // The checking path for 'noreturn' and 'analyzer_noreturn' are different
1951   // because 'analyzer_noreturn' does not impact the type.
1952   if (!isFunctionOrMethodOrBlock(D)) {
1953     ValueDecl *VD = dyn_cast<ValueDecl>(D);
1954     if (!VD || (!VD->getType()->isBlockPointerType() &&
1955                 !VD->getType()->isFunctionPointerType())) {
1956       S.Diag(AL.getLoc(), AL.isCXX11Attribute()
1957                               ? diag::err_attribute_wrong_decl_type
1958                               : diag::warn_attribute_wrong_decl_type)
1959           << AL << ExpectedFunctionMethodOrBlock;
1960       return;
1961     }
1962   }
1963 
1964   D->addAttr(::new (S.Context)
1965              AnalyzerNoReturnAttr(AL.getRange(), S.Context,
1966                                   AL.getAttributeSpellingListIndex()));
1967 }
1968 
1969 // PS3 PPU-specific.
1970 static void handleVecReturnAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
1971   /*
1972     Returning a Vector Class in Registers
1973 
1974     According to the PPU ABI specifications, a class with a single member of
1975     vector type is returned in memory when used as the return value of a
1976     function.
1977     This results in inefficient code when implementing vector classes. To return
1978     the value in a single vector register, add the vecreturn attribute to the
1979     class definition. This attribute is also applicable to struct types.
1980 
1981     Example:
1982 
1983     struct Vector
1984     {
1985       __vector float xyzw;
1986     } __attribute__((vecreturn));
1987 
1988     Vector Add(Vector lhs, Vector rhs)
1989     {
1990       Vector result;
1991       result.xyzw = vec_add(lhs.xyzw, rhs.xyzw);
1992       return result; // This will be returned in a register
1993     }
1994   */
1995   if (VecReturnAttr *A = D->getAttr<VecReturnAttr>()) {
1996     S.Diag(AL.getLoc(), diag::err_repeat_attribute) << A;
1997     return;
1998   }
1999 
2000   const auto *R = cast<RecordDecl>(D);
2001   int count = 0;
2002 
2003   if (!isa<CXXRecordDecl>(R)) {
2004     S.Diag(AL.getLoc(), diag::err_attribute_vecreturn_only_vector_member);
2005     return;
2006   }
2007 
2008   if (!cast<CXXRecordDecl>(R)->isPOD()) {
2009     S.Diag(AL.getLoc(), diag::err_attribute_vecreturn_only_pod_record);
2010     return;
2011   }
2012 
2013   for (const auto *I : R->fields()) {
2014     if ((count == 1) || !I->getType()->isVectorType()) {
2015       S.Diag(AL.getLoc(), diag::err_attribute_vecreturn_only_vector_member);
2016       return;
2017     }
2018     count++;
2019   }
2020 
2021   D->addAttr(::new (S.Context) VecReturnAttr(
2022       AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
2023 }
2024 
2025 static void handleDependencyAttr(Sema &S, Scope *Scope, Decl *D,
2026                                  const ParsedAttr &AL) {
2027   if (isa<ParmVarDecl>(D)) {
2028     // [[carries_dependency]] can only be applied to a parameter if it is a
2029     // parameter of a function declaration or lambda.
2030     if (!(Scope->getFlags() & clang::Scope::FunctionDeclarationScope)) {
2031       S.Diag(AL.getLoc(),
2032              diag::err_carries_dependency_param_not_function_decl);
2033       return;
2034     }
2035   }
2036 
2037   D->addAttr(::new (S.Context) CarriesDependencyAttr(
2038                                    AL.getRange(), S.Context,
2039                                    AL.getAttributeSpellingListIndex()));
2040 }
2041 
2042 static void handleUnusedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2043   bool IsCXX17Attr = AL.isCXX11Attribute() && !AL.getScopeName();
2044 
2045   // If this is spelled as the standard C++17 attribute, but not in C++17, warn
2046   // about using it as an extension.
2047   if (!S.getLangOpts().CPlusPlus17 && IsCXX17Attr)
2048     S.Diag(AL.getLoc(), diag::ext_cxx17_attr) << AL;
2049 
2050   D->addAttr(::new (S.Context) UnusedAttr(
2051       AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
2052 }
2053 
2054 static void handleConstructorAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2055   uint32_t priority = ConstructorAttr::DefaultPriority;
2056   if (AL.getNumArgs() &&
2057       !checkUInt32Argument(S, AL, AL.getArgAsExpr(0), priority))
2058     return;
2059 
2060   D->addAttr(::new (S.Context)
2061              ConstructorAttr(AL.getRange(), S.Context, priority,
2062                              AL.getAttributeSpellingListIndex()));
2063 }
2064 
2065 static void handleDestructorAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2066   uint32_t priority = DestructorAttr::DefaultPriority;
2067   if (AL.getNumArgs() &&
2068       !checkUInt32Argument(S, AL, AL.getArgAsExpr(0), priority))
2069     return;
2070 
2071   D->addAttr(::new (S.Context)
2072              DestructorAttr(AL.getRange(), S.Context, priority,
2073                             AL.getAttributeSpellingListIndex()));
2074 }
2075 
2076 template <typename AttrTy>
2077 static void handleAttrWithMessage(Sema &S, Decl *D, const ParsedAttr &AL) {
2078   // Handle the case where the attribute has a text message.
2079   StringRef Str;
2080   if (AL.getNumArgs() == 1 && !S.checkStringLiteralArgumentAttr(AL, 0, Str))
2081     return;
2082 
2083   D->addAttr(::new (S.Context) AttrTy(AL.getRange(), S.Context, Str,
2084                                       AL.getAttributeSpellingListIndex()));
2085 }
2086 
2087 static void handleObjCSuppresProtocolAttr(Sema &S, Decl *D,
2088                                           const ParsedAttr &AL) {
2089   if (!cast<ObjCProtocolDecl>(D)->isThisDeclarationADefinition()) {
2090     S.Diag(AL.getLoc(), diag::err_objc_attr_protocol_requires_definition)
2091         << AL << AL.getRange();
2092     return;
2093   }
2094 
2095   D->addAttr(::new (S.Context)
2096           ObjCExplicitProtocolImplAttr(AL.getRange(), S.Context,
2097                                        AL.getAttributeSpellingListIndex()));
2098 }
2099 
2100 static bool checkAvailabilityAttr(Sema &S, SourceRange Range,
2101                                   IdentifierInfo *Platform,
2102                                   VersionTuple Introduced,
2103                                   VersionTuple Deprecated,
2104                                   VersionTuple Obsoleted) {
2105   StringRef PlatformName
2106     = AvailabilityAttr::getPrettyPlatformName(Platform->getName());
2107   if (PlatformName.empty())
2108     PlatformName = Platform->getName();
2109 
2110   // Ensure that Introduced <= Deprecated <= Obsoleted (although not all
2111   // of these steps are needed).
2112   if (!Introduced.empty() && !Deprecated.empty() &&
2113       !(Introduced <= Deprecated)) {
2114     S.Diag(Range.getBegin(), diag::warn_availability_version_ordering)
2115       << 1 << PlatformName << Deprecated.getAsString()
2116       << 0 << Introduced.getAsString();
2117     return true;
2118   }
2119 
2120   if (!Introduced.empty() && !Obsoleted.empty() &&
2121       !(Introduced <= Obsoleted)) {
2122     S.Diag(Range.getBegin(), diag::warn_availability_version_ordering)
2123       << 2 << PlatformName << Obsoleted.getAsString()
2124       << 0 << Introduced.getAsString();
2125     return true;
2126   }
2127 
2128   if (!Deprecated.empty() && !Obsoleted.empty() &&
2129       !(Deprecated <= Obsoleted)) {
2130     S.Diag(Range.getBegin(), diag::warn_availability_version_ordering)
2131       << 2 << PlatformName << Obsoleted.getAsString()
2132       << 1 << Deprecated.getAsString();
2133     return true;
2134   }
2135 
2136   return false;
2137 }
2138 
2139 /// Check whether the two versions match.
2140 ///
2141 /// If either version tuple is empty, then they are assumed to match. If
2142 /// \p BeforeIsOkay is true, then \p X can be less than or equal to \p Y.
2143 static bool versionsMatch(const VersionTuple &X, const VersionTuple &Y,
2144                           bool BeforeIsOkay) {
2145   if (X.empty() || Y.empty())
2146     return true;
2147 
2148   if (X == Y)
2149     return true;
2150 
2151   if (BeforeIsOkay && X < Y)
2152     return true;
2153 
2154   return false;
2155 }
2156 
2157 AvailabilityAttr *Sema::mergeAvailabilityAttr(NamedDecl *D, SourceRange Range,
2158                                               IdentifierInfo *Platform,
2159                                               bool Implicit,
2160                                               VersionTuple Introduced,
2161                                               VersionTuple Deprecated,
2162                                               VersionTuple Obsoleted,
2163                                               bool IsUnavailable,
2164                                               StringRef Message,
2165                                               bool IsStrict,
2166                                               StringRef Replacement,
2167                                               AvailabilityMergeKind AMK,
2168                                               unsigned AttrSpellingListIndex) {
2169   VersionTuple MergedIntroduced = Introduced;
2170   VersionTuple MergedDeprecated = Deprecated;
2171   VersionTuple MergedObsoleted = Obsoleted;
2172   bool FoundAny = false;
2173   bool OverrideOrImpl = false;
2174   switch (AMK) {
2175   case AMK_None:
2176   case AMK_Redeclaration:
2177     OverrideOrImpl = false;
2178     break;
2179 
2180   case AMK_Override:
2181   case AMK_ProtocolImplementation:
2182     OverrideOrImpl = true;
2183     break;
2184   }
2185 
2186   if (D->hasAttrs()) {
2187     AttrVec &Attrs = D->getAttrs();
2188     for (unsigned i = 0, e = Attrs.size(); i != e;) {
2189       const auto *OldAA = dyn_cast<AvailabilityAttr>(Attrs[i]);
2190       if (!OldAA) {
2191         ++i;
2192         continue;
2193       }
2194 
2195       IdentifierInfo *OldPlatform = OldAA->getPlatform();
2196       if (OldPlatform != Platform) {
2197         ++i;
2198         continue;
2199       }
2200 
2201       // If there is an existing availability attribute for this platform that
2202       // is explicit and the new one is implicit use the explicit one and
2203       // discard the new implicit attribute.
2204       if (!OldAA->isImplicit() && Implicit) {
2205         return nullptr;
2206       }
2207 
2208       // If there is an existing attribute for this platform that is implicit
2209       // and the new attribute is explicit then erase the old one and
2210       // continue processing the attributes.
2211       if (!Implicit && OldAA->isImplicit()) {
2212         Attrs.erase(Attrs.begin() + i);
2213         --e;
2214         continue;
2215       }
2216 
2217       FoundAny = true;
2218       VersionTuple OldIntroduced = OldAA->getIntroduced();
2219       VersionTuple OldDeprecated = OldAA->getDeprecated();
2220       VersionTuple OldObsoleted = OldAA->getObsoleted();
2221       bool OldIsUnavailable = OldAA->getUnavailable();
2222 
2223       if (!versionsMatch(OldIntroduced, Introduced, OverrideOrImpl) ||
2224           !versionsMatch(Deprecated, OldDeprecated, OverrideOrImpl) ||
2225           !versionsMatch(Obsoleted, OldObsoleted, OverrideOrImpl) ||
2226           !(OldIsUnavailable == IsUnavailable ||
2227             (OverrideOrImpl && !OldIsUnavailable && IsUnavailable))) {
2228         if (OverrideOrImpl) {
2229           int Which = -1;
2230           VersionTuple FirstVersion;
2231           VersionTuple SecondVersion;
2232           if (!versionsMatch(OldIntroduced, Introduced, OverrideOrImpl)) {
2233             Which = 0;
2234             FirstVersion = OldIntroduced;
2235             SecondVersion = Introduced;
2236           } else if (!versionsMatch(Deprecated, OldDeprecated, OverrideOrImpl)) {
2237             Which = 1;
2238             FirstVersion = Deprecated;
2239             SecondVersion = OldDeprecated;
2240           } else if (!versionsMatch(Obsoleted, OldObsoleted, OverrideOrImpl)) {
2241             Which = 2;
2242             FirstVersion = Obsoleted;
2243             SecondVersion = OldObsoleted;
2244           }
2245 
2246           if (Which == -1) {
2247             Diag(OldAA->getLocation(),
2248                  diag::warn_mismatched_availability_override_unavail)
2249               << AvailabilityAttr::getPrettyPlatformName(Platform->getName())
2250               << (AMK == AMK_Override);
2251           } else {
2252             Diag(OldAA->getLocation(),
2253                  diag::warn_mismatched_availability_override)
2254               << Which
2255               << AvailabilityAttr::getPrettyPlatformName(Platform->getName())
2256               << FirstVersion.getAsString() << SecondVersion.getAsString()
2257               << (AMK == AMK_Override);
2258           }
2259           if (AMK == AMK_Override)
2260             Diag(Range.getBegin(), diag::note_overridden_method);
2261           else
2262             Diag(Range.getBegin(), diag::note_protocol_method);
2263         } else {
2264           Diag(OldAA->getLocation(), diag::warn_mismatched_availability);
2265           Diag(Range.getBegin(), diag::note_previous_attribute);
2266         }
2267 
2268         Attrs.erase(Attrs.begin() + i);
2269         --e;
2270         continue;
2271       }
2272 
2273       VersionTuple MergedIntroduced2 = MergedIntroduced;
2274       VersionTuple MergedDeprecated2 = MergedDeprecated;
2275       VersionTuple MergedObsoleted2 = MergedObsoleted;
2276 
2277       if (MergedIntroduced2.empty())
2278         MergedIntroduced2 = OldIntroduced;
2279       if (MergedDeprecated2.empty())
2280         MergedDeprecated2 = OldDeprecated;
2281       if (MergedObsoleted2.empty())
2282         MergedObsoleted2 = OldObsoleted;
2283 
2284       if (checkAvailabilityAttr(*this, OldAA->getRange(), Platform,
2285                                 MergedIntroduced2, MergedDeprecated2,
2286                                 MergedObsoleted2)) {
2287         Attrs.erase(Attrs.begin() + i);
2288         --e;
2289         continue;
2290       }
2291 
2292       MergedIntroduced = MergedIntroduced2;
2293       MergedDeprecated = MergedDeprecated2;
2294       MergedObsoleted = MergedObsoleted2;
2295       ++i;
2296     }
2297   }
2298 
2299   if (FoundAny &&
2300       MergedIntroduced == Introduced &&
2301       MergedDeprecated == Deprecated &&
2302       MergedObsoleted == Obsoleted)
2303     return nullptr;
2304 
2305   // Only create a new attribute if !OverrideOrImpl, but we want to do
2306   // the checking.
2307   if (!checkAvailabilityAttr(*this, Range, Platform, MergedIntroduced,
2308                              MergedDeprecated, MergedObsoleted) &&
2309       !OverrideOrImpl) {
2310     auto *Avail =  ::new (Context) AvailabilityAttr(Range, Context, Platform,
2311                                             Introduced, Deprecated,
2312                                             Obsoleted, IsUnavailable, Message,
2313                                             IsStrict, Replacement,
2314                                             AttrSpellingListIndex);
2315     Avail->setImplicit(Implicit);
2316     return Avail;
2317   }
2318   return nullptr;
2319 }
2320 
2321 static void handleAvailabilityAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2322   if (!checkAttributeNumArgs(S, AL, 1))
2323     return;
2324   IdentifierLoc *Platform = AL.getArgAsIdent(0);
2325   unsigned Index = AL.getAttributeSpellingListIndex();
2326 
2327   IdentifierInfo *II = Platform->Ident;
2328   if (AvailabilityAttr::getPrettyPlatformName(II->getName()).empty())
2329     S.Diag(Platform->Loc, diag::warn_availability_unknown_platform)
2330       << Platform->Ident;
2331 
2332   auto *ND = dyn_cast<NamedDecl>(D);
2333   if (!ND) // We warned about this already, so just return.
2334     return;
2335 
2336   AvailabilityChange Introduced = AL.getAvailabilityIntroduced();
2337   AvailabilityChange Deprecated = AL.getAvailabilityDeprecated();
2338   AvailabilityChange Obsoleted = AL.getAvailabilityObsoleted();
2339   bool IsUnavailable = AL.getUnavailableLoc().isValid();
2340   bool IsStrict = AL.getStrictLoc().isValid();
2341   StringRef Str;
2342   if (const auto *SE = dyn_cast_or_null<StringLiteral>(AL.getMessageExpr()))
2343     Str = SE->getString();
2344   StringRef Replacement;
2345   if (const auto *SE = dyn_cast_or_null<StringLiteral>(AL.getReplacementExpr()))
2346     Replacement = SE->getString();
2347 
2348   AvailabilityAttr *NewAttr = S.mergeAvailabilityAttr(ND, AL.getRange(), II,
2349                                                       false/*Implicit*/,
2350                                                       Introduced.Version,
2351                                                       Deprecated.Version,
2352                                                       Obsoleted.Version,
2353                                                       IsUnavailable, Str,
2354                                                       IsStrict, Replacement,
2355                                                       Sema::AMK_None,
2356                                                       Index);
2357   if (NewAttr)
2358     D->addAttr(NewAttr);
2359 
2360   // Transcribe "ios" to "watchos" (and add a new attribute) if the versioning
2361   // matches before the start of the watchOS platform.
2362   if (S.Context.getTargetInfo().getTriple().isWatchOS()) {
2363     IdentifierInfo *NewII = nullptr;
2364     if (II->getName() == "ios")
2365       NewII = &S.Context.Idents.get("watchos");
2366     else if (II->getName() == "ios_app_extension")
2367       NewII = &S.Context.Idents.get("watchos_app_extension");
2368 
2369     if (NewII) {
2370         auto adjustWatchOSVersion = [](VersionTuple Version) -> VersionTuple {
2371           if (Version.empty())
2372             return Version;
2373           auto Major = Version.getMajor();
2374           auto NewMajor = Major >= 9 ? Major - 7 : 0;
2375           if (NewMajor >= 2) {
2376             if (Version.getMinor().hasValue()) {
2377               if (Version.getSubminor().hasValue())
2378                 return VersionTuple(NewMajor, Version.getMinor().getValue(),
2379                                     Version.getSubminor().getValue());
2380               else
2381                 return VersionTuple(NewMajor, Version.getMinor().getValue());
2382             }
2383           }
2384 
2385           return VersionTuple(2, 0);
2386         };
2387 
2388         auto NewIntroduced = adjustWatchOSVersion(Introduced.Version);
2389         auto NewDeprecated = adjustWatchOSVersion(Deprecated.Version);
2390         auto NewObsoleted = adjustWatchOSVersion(Obsoleted.Version);
2391 
2392         AvailabilityAttr *NewAttr = S.mergeAvailabilityAttr(ND,
2393                                                             AL.getRange(),
2394                                                             NewII,
2395                                                             true/*Implicit*/,
2396                                                             NewIntroduced,
2397                                                             NewDeprecated,
2398                                                             NewObsoleted,
2399                                                             IsUnavailable, Str,
2400                                                             IsStrict,
2401                                                             Replacement,
2402                                                             Sema::AMK_None,
2403                                                             Index);
2404         if (NewAttr)
2405           D->addAttr(NewAttr);
2406       }
2407   } else if (S.Context.getTargetInfo().getTriple().isTvOS()) {
2408     // Transcribe "ios" to "tvos" (and add a new attribute) if the versioning
2409     // matches before the start of the tvOS platform.
2410     IdentifierInfo *NewII = nullptr;
2411     if (II->getName() == "ios")
2412       NewII = &S.Context.Idents.get("tvos");
2413     else if (II->getName() == "ios_app_extension")
2414       NewII = &S.Context.Idents.get("tvos_app_extension");
2415 
2416     if (NewII) {
2417         AvailabilityAttr *NewAttr = S.mergeAvailabilityAttr(ND,
2418                                                             AL.getRange(),
2419                                                             NewII,
2420                                                             true/*Implicit*/,
2421                                                             Introduced.Version,
2422                                                             Deprecated.Version,
2423                                                             Obsoleted.Version,
2424                                                             IsUnavailable, Str,
2425                                                             IsStrict,
2426                                                             Replacement,
2427                                                             Sema::AMK_None,
2428                                                             Index);
2429         if (NewAttr)
2430           D->addAttr(NewAttr);
2431       }
2432   }
2433 }
2434 
2435 static void handleExternalSourceSymbolAttr(Sema &S, Decl *D,
2436                                            const ParsedAttr &AL) {
2437   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
2438     return;
2439   assert(checkAttributeAtMostNumArgs(S, AL, 3) &&
2440          "Invalid number of arguments in an external_source_symbol attribute");
2441 
2442   StringRef Language;
2443   if (const auto *SE = dyn_cast_or_null<StringLiteral>(AL.getArgAsExpr(0)))
2444     Language = SE->getString();
2445   StringRef DefinedIn;
2446   if (const auto *SE = dyn_cast_or_null<StringLiteral>(AL.getArgAsExpr(1)))
2447     DefinedIn = SE->getString();
2448   bool IsGeneratedDeclaration = AL.getArgAsIdent(2) != nullptr;
2449 
2450   D->addAttr(::new (S.Context) ExternalSourceSymbolAttr(
2451       AL.getRange(), S.Context, Language, DefinedIn, IsGeneratedDeclaration,
2452       AL.getAttributeSpellingListIndex()));
2453 }
2454 
2455 template <class T>
2456 static T *mergeVisibilityAttr(Sema &S, Decl *D, SourceRange range,
2457                               typename T::VisibilityType value,
2458                               unsigned attrSpellingListIndex) {
2459   T *existingAttr = D->getAttr<T>();
2460   if (existingAttr) {
2461     typename T::VisibilityType existingValue = existingAttr->getVisibility();
2462     if (existingValue == value)
2463       return nullptr;
2464     S.Diag(existingAttr->getLocation(), diag::err_mismatched_visibility);
2465     S.Diag(range.getBegin(), diag::note_previous_attribute);
2466     D->dropAttr<T>();
2467   }
2468   return ::new (S.Context) T(range, S.Context, value, attrSpellingListIndex);
2469 }
2470 
2471 VisibilityAttr *Sema::mergeVisibilityAttr(Decl *D, SourceRange Range,
2472                                           VisibilityAttr::VisibilityType Vis,
2473                                           unsigned AttrSpellingListIndex) {
2474   return ::mergeVisibilityAttr<VisibilityAttr>(*this, D, Range, Vis,
2475                                                AttrSpellingListIndex);
2476 }
2477 
2478 TypeVisibilityAttr *Sema::mergeTypeVisibilityAttr(Decl *D, SourceRange Range,
2479                                       TypeVisibilityAttr::VisibilityType Vis,
2480                                       unsigned AttrSpellingListIndex) {
2481   return ::mergeVisibilityAttr<TypeVisibilityAttr>(*this, D, Range, Vis,
2482                                                    AttrSpellingListIndex);
2483 }
2484 
2485 static void handleVisibilityAttr(Sema &S, Decl *D, const ParsedAttr &AL,
2486                                  bool isTypeVisibility) {
2487   // Visibility attributes don't mean anything on a typedef.
2488   if (isa<TypedefNameDecl>(D)) {
2489     S.Diag(AL.getRange().getBegin(), diag::warn_attribute_ignored) << AL;
2490     return;
2491   }
2492 
2493   // 'type_visibility' can only go on a type or namespace.
2494   if (isTypeVisibility &&
2495       !(isa<TagDecl>(D) ||
2496         isa<ObjCInterfaceDecl>(D) ||
2497         isa<NamespaceDecl>(D))) {
2498     S.Diag(AL.getRange().getBegin(), diag::err_attribute_wrong_decl_type)
2499         << AL << ExpectedTypeOrNamespace;
2500     return;
2501   }
2502 
2503   // Check that the argument is a string literal.
2504   StringRef TypeStr;
2505   SourceLocation LiteralLoc;
2506   if (!S.checkStringLiteralArgumentAttr(AL, 0, TypeStr, &LiteralLoc))
2507     return;
2508 
2509   VisibilityAttr::VisibilityType type;
2510   if (!VisibilityAttr::ConvertStrToVisibilityType(TypeStr, type)) {
2511     S.Diag(LiteralLoc, diag::warn_attribute_type_not_supported) << AL
2512                                                                 << TypeStr;
2513     return;
2514   }
2515 
2516   // Complain about attempts to use protected visibility on targets
2517   // (like Darwin) that don't support it.
2518   if (type == VisibilityAttr::Protected &&
2519       !S.Context.getTargetInfo().hasProtectedVisibility()) {
2520     S.Diag(AL.getLoc(), diag::warn_attribute_protected_visibility);
2521     type = VisibilityAttr::Default;
2522   }
2523 
2524   unsigned Index = AL.getAttributeSpellingListIndex();
2525   Attr *newAttr;
2526   if (isTypeVisibility) {
2527     newAttr = S.mergeTypeVisibilityAttr(D, AL.getRange(),
2528                                     (TypeVisibilityAttr::VisibilityType) type,
2529                                         Index);
2530   } else {
2531     newAttr = S.mergeVisibilityAttr(D, AL.getRange(), type, Index);
2532   }
2533   if (newAttr)
2534     D->addAttr(newAttr);
2535 }
2536 
2537 static void handleObjCMethodFamilyAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2538   const auto *M = cast<ObjCMethodDecl>(D);
2539   if (!AL.isArgIdent(0)) {
2540     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
2541         << AL << 1 << AANT_ArgumentIdentifier;
2542     return;
2543   }
2544 
2545   IdentifierLoc *IL = AL.getArgAsIdent(0);
2546   ObjCMethodFamilyAttr::FamilyKind F;
2547   if (!ObjCMethodFamilyAttr::ConvertStrToFamilyKind(IL->Ident->getName(), F)) {
2548     S.Diag(IL->Loc, diag::warn_attribute_type_not_supported) << AL << IL->Ident;
2549     return;
2550   }
2551 
2552   if (F == ObjCMethodFamilyAttr::OMF_init &&
2553       !M->getReturnType()->isObjCObjectPointerType()) {
2554     S.Diag(M->getLocation(), diag::err_init_method_bad_return_type)
2555         << M->getReturnType();
2556     // Ignore the attribute.
2557     return;
2558   }
2559 
2560   D->addAttr(new (S.Context) ObjCMethodFamilyAttr(
2561       AL.getRange(), S.Context, F, AL.getAttributeSpellingListIndex()));
2562 }
2563 
2564 static void handleObjCNSObject(Sema &S, Decl *D, const ParsedAttr &AL) {
2565   if (const auto *TD = dyn_cast<TypedefNameDecl>(D)) {
2566     QualType T = TD->getUnderlyingType();
2567     if (!T->isCARCBridgableType()) {
2568       S.Diag(TD->getLocation(), diag::err_nsobject_attribute);
2569       return;
2570     }
2571   }
2572   else if (const auto *PD = dyn_cast<ObjCPropertyDecl>(D)) {
2573     QualType T = PD->getType();
2574     if (!T->isCARCBridgableType()) {
2575       S.Diag(PD->getLocation(), diag::err_nsobject_attribute);
2576       return;
2577     }
2578   }
2579   else {
2580     // It is okay to include this attribute on properties, e.g.:
2581     //
2582     //  @property (retain, nonatomic) struct Bork *Q __attribute__((NSObject));
2583     //
2584     // In this case it follows tradition and suppresses an error in the above
2585     // case.
2586     S.Diag(D->getLocation(), diag::warn_nsobject_attribute);
2587   }
2588   D->addAttr(::new (S.Context)
2589              ObjCNSObjectAttr(AL.getRange(), S.Context,
2590                               AL.getAttributeSpellingListIndex()));
2591 }
2592 
2593 static void handleObjCIndependentClass(Sema &S, Decl *D, const ParsedAttr &AL) {
2594   if (const auto *TD = dyn_cast<TypedefNameDecl>(D)) {
2595     QualType T = TD->getUnderlyingType();
2596     if (!T->isObjCObjectPointerType()) {
2597       S.Diag(TD->getLocation(), diag::warn_ptr_independentclass_attribute);
2598       return;
2599     }
2600   } else {
2601     S.Diag(D->getLocation(), diag::warn_independentclass_attribute);
2602     return;
2603   }
2604   D->addAttr(::new (S.Context)
2605              ObjCIndependentClassAttr(AL.getRange(), S.Context,
2606                               AL.getAttributeSpellingListIndex()));
2607 }
2608 
2609 static void handleBlocksAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2610   if (!AL.isArgIdent(0)) {
2611     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
2612         << AL << 1 << AANT_ArgumentIdentifier;
2613     return;
2614   }
2615 
2616   IdentifierInfo *II = AL.getArgAsIdent(0)->Ident;
2617   BlocksAttr::BlockType type;
2618   if (!BlocksAttr::ConvertStrToBlockType(II->getName(), type)) {
2619     S.Diag(AL.getLoc(), diag::warn_attribute_type_not_supported) << AL << II;
2620     return;
2621   }
2622 
2623   D->addAttr(::new (S.Context)
2624              BlocksAttr(AL.getRange(), S.Context, type,
2625                         AL.getAttributeSpellingListIndex()));
2626 }
2627 
2628 static void handleSentinelAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2629   unsigned sentinel = (unsigned)SentinelAttr::DefaultSentinel;
2630   if (AL.getNumArgs() > 0) {
2631     Expr *E = AL.getArgAsExpr(0);
2632     llvm::APSInt Idx(32);
2633     if (E->isTypeDependent() || E->isValueDependent() ||
2634         !E->isIntegerConstantExpr(Idx, S.Context)) {
2635       S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
2636           << AL << 1 << AANT_ArgumentIntegerConstant << E->getSourceRange();
2637       return;
2638     }
2639 
2640     if (Idx.isSigned() && Idx.isNegative()) {
2641       S.Diag(AL.getLoc(), diag::err_attribute_sentinel_less_than_zero)
2642         << E->getSourceRange();
2643       return;
2644     }
2645 
2646     sentinel = Idx.getZExtValue();
2647   }
2648 
2649   unsigned nullPos = (unsigned)SentinelAttr::DefaultNullPos;
2650   if (AL.getNumArgs() > 1) {
2651     Expr *E = AL.getArgAsExpr(1);
2652     llvm::APSInt Idx(32);
2653     if (E->isTypeDependent() || E->isValueDependent() ||
2654         !E->isIntegerConstantExpr(Idx, S.Context)) {
2655       S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
2656           << AL << 2 << AANT_ArgumentIntegerConstant << E->getSourceRange();
2657       return;
2658     }
2659     nullPos = Idx.getZExtValue();
2660 
2661     if ((Idx.isSigned() && Idx.isNegative()) || nullPos > 1) {
2662       // FIXME: This error message could be improved, it would be nice
2663       // to say what the bounds actually are.
2664       S.Diag(AL.getLoc(), diag::err_attribute_sentinel_not_zero_or_one)
2665         << E->getSourceRange();
2666       return;
2667     }
2668   }
2669 
2670   if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
2671     const FunctionType *FT = FD->getType()->castAs<FunctionType>();
2672     if (isa<FunctionNoProtoType>(FT)) {
2673       S.Diag(AL.getLoc(), diag::warn_attribute_sentinel_named_arguments);
2674       return;
2675     }
2676 
2677     if (!cast<FunctionProtoType>(FT)->isVariadic()) {
2678       S.Diag(AL.getLoc(), diag::warn_attribute_sentinel_not_variadic) << 0;
2679       return;
2680     }
2681   } else if (const auto *MD = dyn_cast<ObjCMethodDecl>(D)) {
2682     if (!MD->isVariadic()) {
2683       S.Diag(AL.getLoc(), diag::warn_attribute_sentinel_not_variadic) << 0;
2684       return;
2685     }
2686   } else if (const auto *BD = dyn_cast<BlockDecl>(D)) {
2687     if (!BD->isVariadic()) {
2688       S.Diag(AL.getLoc(), diag::warn_attribute_sentinel_not_variadic) << 1;
2689       return;
2690     }
2691   } else if (const auto *V = dyn_cast<VarDecl>(D)) {
2692     QualType Ty = V->getType();
2693     if (Ty->isBlockPointerType() || Ty->isFunctionPointerType()) {
2694       const FunctionType *FT = Ty->isFunctionPointerType()
2695        ? D->getFunctionType()
2696        : Ty->getAs<BlockPointerType>()->getPointeeType()->getAs<FunctionType>();
2697       if (!cast<FunctionProtoType>(FT)->isVariadic()) {
2698         int m = Ty->isFunctionPointerType() ? 0 : 1;
2699         S.Diag(AL.getLoc(), diag::warn_attribute_sentinel_not_variadic) << m;
2700         return;
2701       }
2702     } else {
2703       S.Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type)
2704           << AL << ExpectedFunctionMethodOrBlock;
2705       return;
2706     }
2707   } else {
2708     S.Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type)
2709         << AL << ExpectedFunctionMethodOrBlock;
2710     return;
2711   }
2712   D->addAttr(::new (S.Context)
2713              SentinelAttr(AL.getRange(), S.Context, sentinel, nullPos,
2714                           AL.getAttributeSpellingListIndex()));
2715 }
2716 
2717 static void handleWarnUnusedResult(Sema &S, Decl *D, const ParsedAttr &AL) {
2718   if (D->getFunctionType() &&
2719       D->getFunctionType()->getReturnType()->isVoidType()) {
2720     S.Diag(AL.getLoc(), diag::warn_attribute_void_function_method) << AL << 0;
2721     return;
2722   }
2723   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D))
2724     if (MD->getReturnType()->isVoidType()) {
2725       S.Diag(AL.getLoc(), diag::warn_attribute_void_function_method) << AL << 1;
2726       return;
2727     }
2728 
2729   // If this is spelled as the standard C++17 attribute, but not in C++17, warn
2730   // about using it as an extension.
2731   if (!S.getLangOpts().CPlusPlus17 && AL.isCXX11Attribute() &&
2732       !AL.getScopeName())
2733     S.Diag(AL.getLoc(), diag::ext_cxx17_attr) << AL;
2734 
2735   D->addAttr(::new (S.Context)
2736              WarnUnusedResultAttr(AL.getRange(), S.Context,
2737                                   AL.getAttributeSpellingListIndex()));
2738 }
2739 
2740 static void handleWeakImportAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2741   // weak_import only applies to variable & function declarations.
2742   bool isDef = false;
2743   if (!D->canBeWeakImported(isDef)) {
2744     if (isDef)
2745       S.Diag(AL.getLoc(), diag::warn_attribute_invalid_on_definition)
2746         << "weak_import";
2747     else if (isa<ObjCPropertyDecl>(D) || isa<ObjCMethodDecl>(D) ||
2748              (S.Context.getTargetInfo().getTriple().isOSDarwin() &&
2749               (isa<ObjCInterfaceDecl>(D) || isa<EnumDecl>(D)))) {
2750       // Nothing to warn about here.
2751     } else
2752       S.Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type)
2753           << AL << ExpectedVariableOrFunction;
2754 
2755     return;
2756   }
2757 
2758   D->addAttr(::new (S.Context)
2759              WeakImportAttr(AL.getRange(), S.Context,
2760                             AL.getAttributeSpellingListIndex()));
2761 }
2762 
2763 // Handles reqd_work_group_size and work_group_size_hint.
2764 template <typename WorkGroupAttr>
2765 static void handleWorkGroupSize(Sema &S, Decl *D, const ParsedAttr &AL) {
2766   uint32_t WGSize[3];
2767   for (unsigned i = 0; i < 3; ++i) {
2768     const Expr *E = AL.getArgAsExpr(i);
2769     if (!checkUInt32Argument(S, AL, E, WGSize[i], i))
2770       return;
2771     if (WGSize[i] == 0) {
2772       S.Diag(AL.getLoc(), diag::err_attribute_argument_is_zero)
2773           << AL << E->getSourceRange();
2774       return;
2775     }
2776   }
2777 
2778   WorkGroupAttr *Existing = D->getAttr<WorkGroupAttr>();
2779   if (Existing && !(Existing->getXDim() == WGSize[0] &&
2780                     Existing->getYDim() == WGSize[1] &&
2781                     Existing->getZDim() == WGSize[2]))
2782     S.Diag(AL.getLoc(), diag::warn_duplicate_attribute) << AL;
2783 
2784   D->addAttr(::new (S.Context) WorkGroupAttr(AL.getRange(), S.Context,
2785                                              WGSize[0], WGSize[1], WGSize[2],
2786                                        AL.getAttributeSpellingListIndex()));
2787 }
2788 
2789 // Handles intel_reqd_sub_group_size.
2790 static void handleSubGroupSize(Sema &S, Decl *D, const ParsedAttr &AL) {
2791   uint32_t SGSize;
2792   const Expr *E = AL.getArgAsExpr(0);
2793   if (!checkUInt32Argument(S, AL, E, SGSize))
2794     return;
2795   if (SGSize == 0) {
2796     S.Diag(AL.getLoc(), diag::err_attribute_argument_is_zero)
2797         << AL << E->getSourceRange();
2798     return;
2799   }
2800 
2801   OpenCLIntelReqdSubGroupSizeAttr *Existing =
2802       D->getAttr<OpenCLIntelReqdSubGroupSizeAttr>();
2803   if (Existing && Existing->getSubGroupSize() != SGSize)
2804     S.Diag(AL.getLoc(), diag::warn_duplicate_attribute) << AL;
2805 
2806   D->addAttr(::new (S.Context) OpenCLIntelReqdSubGroupSizeAttr(
2807       AL.getRange(), S.Context, SGSize,
2808       AL.getAttributeSpellingListIndex()));
2809 }
2810 
2811 static void handleVecTypeHint(Sema &S, Decl *D, const ParsedAttr &AL) {
2812   if (!AL.hasParsedType()) {
2813     S.Diag(AL.getLoc(), diag::err_attribute_wrong_number_arguments) << AL << 1;
2814     return;
2815   }
2816 
2817   TypeSourceInfo *ParmTSI = nullptr;
2818   QualType ParmType = S.GetTypeFromParser(AL.getTypeArg(), &ParmTSI);
2819   assert(ParmTSI && "no type source info for attribute argument");
2820 
2821   if (!ParmType->isExtVectorType() && !ParmType->isFloatingType() &&
2822       (ParmType->isBooleanType() ||
2823        !ParmType->isIntegralType(S.getASTContext()))) {
2824     S.Diag(AL.getLoc(), diag::err_attribute_argument_vec_type_hint)
2825         << ParmType;
2826     return;
2827   }
2828 
2829   if (VecTypeHintAttr *A = D->getAttr<VecTypeHintAttr>()) {
2830     if (!S.Context.hasSameType(A->getTypeHint(), ParmType)) {
2831       S.Diag(AL.getLoc(), diag::warn_duplicate_attribute) << AL;
2832       return;
2833     }
2834   }
2835 
2836   D->addAttr(::new (S.Context) VecTypeHintAttr(AL.getLoc(), S.Context,
2837                                                ParmTSI,
2838                                         AL.getAttributeSpellingListIndex()));
2839 }
2840 
2841 SectionAttr *Sema::mergeSectionAttr(Decl *D, SourceRange Range,
2842                                     StringRef Name,
2843                                     unsigned AttrSpellingListIndex) {
2844   // Explicit or partial specializations do not inherit
2845   // the section attribute from the primary template.
2846   if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
2847     if (AttrSpellingListIndex == SectionAttr::Declspec_allocate &&
2848         FD->isFunctionTemplateSpecialization())
2849       return nullptr;
2850   }
2851   if (SectionAttr *ExistingAttr = D->getAttr<SectionAttr>()) {
2852     if (ExistingAttr->getName() == Name)
2853       return nullptr;
2854     Diag(ExistingAttr->getLocation(), diag::warn_mismatched_section)
2855          << 1 /*section*/;
2856     Diag(Range.getBegin(), diag::note_previous_attribute);
2857     return nullptr;
2858   }
2859   return ::new (Context) SectionAttr(Range, Context, Name,
2860                                      AttrSpellingListIndex);
2861 }
2862 
2863 bool Sema::checkSectionName(SourceLocation LiteralLoc, StringRef SecName) {
2864   std::string Error = Context.getTargetInfo().isValidSectionSpecifier(SecName);
2865   if (!Error.empty()) {
2866     Diag(LiteralLoc, diag::err_attribute_section_invalid_for_target) << Error
2867          << 1 /*'section'*/;
2868     return false;
2869   }
2870   return true;
2871 }
2872 
2873 static void handleSectionAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2874   // Make sure that there is a string literal as the sections's single
2875   // argument.
2876   StringRef Str;
2877   SourceLocation LiteralLoc;
2878   if (!S.checkStringLiteralArgumentAttr(AL, 0, Str, &LiteralLoc))
2879     return;
2880 
2881   if (!S.checkSectionName(LiteralLoc, Str))
2882     return;
2883 
2884   // If the target wants to validate the section specifier, make it happen.
2885   std::string Error = S.Context.getTargetInfo().isValidSectionSpecifier(Str);
2886   if (!Error.empty()) {
2887     S.Diag(LiteralLoc, diag::err_attribute_section_invalid_for_target)
2888     << Error;
2889     return;
2890   }
2891 
2892   unsigned Index = AL.getAttributeSpellingListIndex();
2893   SectionAttr *NewAttr = S.mergeSectionAttr(D, AL.getRange(), Str, Index);
2894   if (NewAttr)
2895     D->addAttr(NewAttr);
2896 }
2897 
2898 static bool checkCodeSegName(Sema&S, SourceLocation LiteralLoc, StringRef CodeSegName) {
2899   std::string Error = S.Context.getTargetInfo().isValidSectionSpecifier(CodeSegName);
2900   if (!Error.empty()) {
2901     S.Diag(LiteralLoc, diag::err_attribute_section_invalid_for_target) << Error
2902            << 0 /*'code-seg'*/;
2903     return false;
2904   }
2905   return true;
2906 }
2907 
2908 CodeSegAttr *Sema::mergeCodeSegAttr(Decl *D, SourceRange Range,
2909                                     StringRef Name,
2910                                     unsigned AttrSpellingListIndex) {
2911   // Explicit or partial specializations do not inherit
2912   // the code_seg attribute from the primary template.
2913   if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
2914     if (FD->isFunctionTemplateSpecialization())
2915       return nullptr;
2916   }
2917   if (const auto *ExistingAttr = D->getAttr<CodeSegAttr>()) {
2918     if (ExistingAttr->getName() == Name)
2919       return nullptr;
2920     Diag(ExistingAttr->getLocation(), diag::warn_mismatched_section)
2921          << 0 /*codeseg*/;
2922     Diag(Range.getBegin(), diag::note_previous_attribute);
2923     return nullptr;
2924   }
2925   return ::new (Context) CodeSegAttr(Range, Context, Name,
2926                                      AttrSpellingListIndex);
2927 }
2928 
2929 static void handleCodeSegAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2930   StringRef Str;
2931   SourceLocation LiteralLoc;
2932   if (!S.checkStringLiteralArgumentAttr(AL, 0, Str, &LiteralLoc))
2933     return;
2934   if (!checkCodeSegName(S, LiteralLoc, Str))
2935     return;
2936   if (const auto *ExistingAttr = D->getAttr<CodeSegAttr>()) {
2937     if (!ExistingAttr->isImplicit()) {
2938       S.Diag(AL.getLoc(),
2939              ExistingAttr->getName() == Str
2940              ? diag::warn_duplicate_codeseg_attribute
2941              : diag::err_conflicting_codeseg_attribute);
2942       return;
2943     }
2944     D->dropAttr<CodeSegAttr>();
2945   }
2946   if (CodeSegAttr *CSA = S.mergeCodeSegAttr(D, AL.getRange(), Str,
2947                                             AL.getAttributeSpellingListIndex()))
2948     D->addAttr(CSA);
2949 }
2950 
2951 // Check for things we'd like to warn about. Multiversioning issues are
2952 // handled later in the process, once we know how many exist.
2953 bool Sema::checkTargetAttr(SourceLocation LiteralLoc, StringRef AttrStr) {
2954   enum FirstParam { Unsupported, Duplicate };
2955   enum SecondParam { None, Architecture };
2956   for (auto Str : {"tune=", "fpmath="})
2957     if (AttrStr.find(Str) != StringRef::npos)
2958       return Diag(LiteralLoc, diag::warn_unsupported_target_attribute)
2959              << Unsupported << None << Str;
2960 
2961   TargetAttr::ParsedTargetAttr ParsedAttrs = TargetAttr::parse(AttrStr);
2962 
2963   if (!ParsedAttrs.Architecture.empty() &&
2964       !Context.getTargetInfo().isValidCPUName(ParsedAttrs.Architecture))
2965     return Diag(LiteralLoc, diag::warn_unsupported_target_attribute)
2966            << Unsupported << Architecture << ParsedAttrs.Architecture;
2967 
2968   if (ParsedAttrs.DuplicateArchitecture)
2969     return Diag(LiteralLoc, diag::warn_unsupported_target_attribute)
2970            << Duplicate << None << "arch=";
2971 
2972   for (const auto &Feature : ParsedAttrs.Features) {
2973     auto CurFeature = StringRef(Feature).drop_front(); // remove + or -.
2974     if (!Context.getTargetInfo().isValidFeatureName(CurFeature))
2975       return Diag(LiteralLoc, diag::warn_unsupported_target_attribute)
2976              << Unsupported << None << CurFeature;
2977   }
2978 
2979   return false;
2980 }
2981 
2982 static void handleTargetAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2983   StringRef Str;
2984   SourceLocation LiteralLoc;
2985   if (!S.checkStringLiteralArgumentAttr(AL, 0, Str, &LiteralLoc) ||
2986       S.checkTargetAttr(LiteralLoc, Str))
2987     return;
2988 
2989   unsigned Index = AL.getAttributeSpellingListIndex();
2990   TargetAttr *NewAttr =
2991       ::new (S.Context) TargetAttr(AL.getRange(), S.Context, Str, Index);
2992   D->addAttr(NewAttr);
2993 }
2994 
2995 static void handleMinVectorWidthAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
2996   Expr *E = AL.getArgAsExpr(0);
2997   uint32_t VecWidth;
2998   if (!checkUInt32Argument(S, AL, E, VecWidth)) {
2999     AL.setInvalid();
3000     return;
3001   }
3002 
3003   MinVectorWidthAttr *Existing = D->getAttr<MinVectorWidthAttr>();
3004   if (Existing && Existing->getVectorWidth() != VecWidth) {
3005     S.Diag(AL.getLoc(), diag::warn_duplicate_attribute) << AL;
3006     return;
3007   }
3008 
3009   D->addAttr(::new (S.Context)
3010              MinVectorWidthAttr(AL.getRange(), S.Context, VecWidth,
3011                                 AL.getAttributeSpellingListIndex()));
3012 }
3013 
3014 static void handleCleanupAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3015   Expr *E = AL.getArgAsExpr(0);
3016   SourceLocation Loc = E->getExprLoc();
3017   FunctionDecl *FD = nullptr;
3018   DeclarationNameInfo NI;
3019 
3020   // gcc only allows for simple identifiers. Since we support more than gcc, we
3021   // will warn the user.
3022   if (auto *DRE = dyn_cast<DeclRefExpr>(E)) {
3023     if (DRE->hasQualifier())
3024       S.Diag(Loc, diag::warn_cleanup_ext);
3025     FD = dyn_cast<FunctionDecl>(DRE->getDecl());
3026     NI = DRE->getNameInfo();
3027     if (!FD) {
3028       S.Diag(Loc, diag::err_attribute_cleanup_arg_not_function) << 1
3029         << NI.getName();
3030       return;
3031     }
3032   } else if (auto *ULE = dyn_cast<UnresolvedLookupExpr>(E)) {
3033     if (ULE->hasExplicitTemplateArgs())
3034       S.Diag(Loc, diag::warn_cleanup_ext);
3035     FD = S.ResolveSingleFunctionTemplateSpecialization(ULE, true);
3036     NI = ULE->getNameInfo();
3037     if (!FD) {
3038       S.Diag(Loc, diag::err_attribute_cleanup_arg_not_function) << 2
3039         << NI.getName();
3040       if (ULE->getType() == S.Context.OverloadTy)
3041         S.NoteAllOverloadCandidates(ULE);
3042       return;
3043     }
3044   } else {
3045     S.Diag(Loc, diag::err_attribute_cleanup_arg_not_function) << 0;
3046     return;
3047   }
3048 
3049   if (FD->getNumParams() != 1) {
3050     S.Diag(Loc, diag::err_attribute_cleanup_func_must_take_one_arg)
3051       << NI.getName();
3052     return;
3053   }
3054 
3055   // We're currently more strict than GCC about what function types we accept.
3056   // If this ever proves to be a problem it should be easy to fix.
3057   QualType Ty = S.Context.getPointerType(cast<VarDecl>(D)->getType());
3058   QualType ParamTy = FD->getParamDecl(0)->getType();
3059   if (S.CheckAssignmentConstraints(FD->getParamDecl(0)->getLocation(),
3060                                    ParamTy, Ty) != Sema::Compatible) {
3061     S.Diag(Loc, diag::err_attribute_cleanup_func_arg_incompatible_type)
3062       << NI.getName() << ParamTy << Ty;
3063     return;
3064   }
3065 
3066   D->addAttr(::new (S.Context)
3067              CleanupAttr(AL.getRange(), S.Context, FD,
3068                          AL.getAttributeSpellingListIndex()));
3069 }
3070 
3071 static void handleEnumExtensibilityAttr(Sema &S, Decl *D,
3072                                         const ParsedAttr &AL) {
3073   if (!AL.isArgIdent(0)) {
3074     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
3075         << AL << 0 << AANT_ArgumentIdentifier;
3076     return;
3077   }
3078 
3079   EnumExtensibilityAttr::Kind ExtensibilityKind;
3080   IdentifierInfo *II = AL.getArgAsIdent(0)->Ident;
3081   if (!EnumExtensibilityAttr::ConvertStrToKind(II->getName(),
3082                                                ExtensibilityKind)) {
3083     S.Diag(AL.getLoc(), diag::warn_attribute_type_not_supported) << AL << II;
3084     return;
3085   }
3086 
3087   D->addAttr(::new (S.Context) EnumExtensibilityAttr(
3088       AL.getRange(), S.Context, ExtensibilityKind,
3089       AL.getAttributeSpellingListIndex()));
3090 }
3091 
3092 /// Handle __attribute__((format_arg((idx)))) attribute based on
3093 /// http://gcc.gnu.org/onlinedocs/gcc/Function-Attributes.html
3094 static void handleFormatArgAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3095   Expr *IdxExpr = AL.getArgAsExpr(0);
3096   ParamIdx Idx;
3097   if (!checkFunctionOrMethodParameterIndex(S, D, AL, 1, IdxExpr, Idx))
3098     return;
3099 
3100   // Make sure the format string is really a string.
3101   QualType Ty = getFunctionOrMethodParamType(D, Idx.getASTIndex());
3102 
3103   bool NotNSStringTy = !isNSStringType(Ty, S.Context);
3104   if (NotNSStringTy &&
3105       !isCFStringType(Ty, S.Context) &&
3106       (!Ty->isPointerType() ||
3107        !Ty->getAs<PointerType>()->getPointeeType()->isCharType())) {
3108     S.Diag(AL.getLoc(), diag::err_format_attribute_not)
3109         << "a string type" << IdxExpr->getSourceRange()
3110         << getFunctionOrMethodParamRange(D, 0);
3111     return;
3112   }
3113   Ty = getFunctionOrMethodResultType(D);
3114   if (!isNSStringType(Ty, S.Context) &&
3115       !isCFStringType(Ty, S.Context) &&
3116       (!Ty->isPointerType() ||
3117        !Ty->getAs<PointerType>()->getPointeeType()->isCharType())) {
3118     S.Diag(AL.getLoc(), diag::err_format_attribute_result_not)
3119         << (NotNSStringTy ? "string type" : "NSString")
3120         << IdxExpr->getSourceRange() << getFunctionOrMethodParamRange(D, 0);
3121     return;
3122   }
3123 
3124   D->addAttr(::new (S.Context) FormatArgAttr(
3125       AL.getRange(), S.Context, Idx, AL.getAttributeSpellingListIndex()));
3126 }
3127 
3128 enum FormatAttrKind {
3129   CFStringFormat,
3130   NSStringFormat,
3131   StrftimeFormat,
3132   SupportedFormat,
3133   IgnoredFormat,
3134   InvalidFormat
3135 };
3136 
3137 /// getFormatAttrKind - Map from format attribute names to supported format
3138 /// types.
3139 static FormatAttrKind getFormatAttrKind(StringRef Format) {
3140   return llvm::StringSwitch<FormatAttrKind>(Format)
3141       // Check for formats that get handled specially.
3142       .Case("NSString", NSStringFormat)
3143       .Case("CFString", CFStringFormat)
3144       .Case("strftime", StrftimeFormat)
3145 
3146       // Otherwise, check for supported formats.
3147       .Cases("scanf", "printf", "printf0", "strfmon", SupportedFormat)
3148       .Cases("cmn_err", "vcmn_err", "zcmn_err", SupportedFormat)
3149       .Case("kprintf", SupportedFormat)         // OpenBSD.
3150       .Case("freebsd_kprintf", SupportedFormat) // FreeBSD.
3151       .Case("os_trace", SupportedFormat)
3152       .Case("os_log", SupportedFormat)
3153 
3154       .Cases("gcc_diag", "gcc_cdiag", "gcc_cxxdiag", "gcc_tdiag", IgnoredFormat)
3155       .Default(InvalidFormat);
3156 }
3157 
3158 /// Handle __attribute__((init_priority(priority))) attributes based on
3159 /// http://gcc.gnu.org/onlinedocs/gcc/C_002b_002b-Attributes.html
3160 static void handleInitPriorityAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3161   if (!S.getLangOpts().CPlusPlus) {
3162     S.Diag(AL.getLoc(), diag::warn_attribute_ignored) << AL;
3163     return;
3164   }
3165 
3166   if (S.getCurFunctionOrMethodDecl()) {
3167     S.Diag(AL.getLoc(), diag::err_init_priority_object_attr);
3168     AL.setInvalid();
3169     return;
3170   }
3171   QualType T = cast<VarDecl>(D)->getType();
3172   if (S.Context.getAsArrayType(T))
3173     T = S.Context.getBaseElementType(T);
3174   if (!T->getAs<RecordType>()) {
3175     S.Diag(AL.getLoc(), diag::err_init_priority_object_attr);
3176     AL.setInvalid();
3177     return;
3178   }
3179 
3180   Expr *E = AL.getArgAsExpr(0);
3181   uint32_t prioritynum;
3182   if (!checkUInt32Argument(S, AL, E, prioritynum)) {
3183     AL.setInvalid();
3184     return;
3185   }
3186 
3187   if (prioritynum < 101 || prioritynum > 65535) {
3188     S.Diag(AL.getLoc(), diag::err_attribute_argument_outof_range)
3189         << E->getSourceRange() << AL << 101 << 65535;
3190     AL.setInvalid();
3191     return;
3192   }
3193   D->addAttr(::new (S.Context)
3194              InitPriorityAttr(AL.getRange(), S.Context, prioritynum,
3195                               AL.getAttributeSpellingListIndex()));
3196 }
3197 
3198 FormatAttr *Sema::mergeFormatAttr(Decl *D, SourceRange Range,
3199                                   IdentifierInfo *Format, int FormatIdx,
3200                                   int FirstArg,
3201                                   unsigned AttrSpellingListIndex) {
3202   // Check whether we already have an equivalent format attribute.
3203   for (auto *F : D->specific_attrs<FormatAttr>()) {
3204     if (F->getType() == Format &&
3205         F->getFormatIdx() == FormatIdx &&
3206         F->getFirstArg() == FirstArg) {
3207       // If we don't have a valid location for this attribute, adopt the
3208       // location.
3209       if (F->getLocation().isInvalid())
3210         F->setRange(Range);
3211       return nullptr;
3212     }
3213   }
3214 
3215   return ::new (Context) FormatAttr(Range, Context, Format, FormatIdx,
3216                                     FirstArg, AttrSpellingListIndex);
3217 }
3218 
3219 /// Handle __attribute__((format(type,idx,firstarg))) attributes based on
3220 /// http://gcc.gnu.org/onlinedocs/gcc/Function-Attributes.html
3221 static void handleFormatAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3222   if (!AL.isArgIdent(0)) {
3223     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
3224         << AL << 1 << AANT_ArgumentIdentifier;
3225     return;
3226   }
3227 
3228   // In C++ the implicit 'this' function parameter also counts, and they are
3229   // counted from one.
3230   bool HasImplicitThisParam = isInstanceMethod(D);
3231   unsigned NumArgs = getFunctionOrMethodNumParams(D) + HasImplicitThisParam;
3232 
3233   IdentifierInfo *II = AL.getArgAsIdent(0)->Ident;
3234   StringRef Format = II->getName();
3235 
3236   if (normalizeName(Format)) {
3237     // If we've modified the string name, we need a new identifier for it.
3238     II = &S.Context.Idents.get(Format);
3239   }
3240 
3241   // Check for supported formats.
3242   FormatAttrKind Kind = getFormatAttrKind(Format);
3243 
3244   if (Kind == IgnoredFormat)
3245     return;
3246 
3247   if (Kind == InvalidFormat) {
3248     S.Diag(AL.getLoc(), diag::warn_attribute_type_not_supported)
3249         << AL << II->getName();
3250     return;
3251   }
3252 
3253   // checks for the 2nd argument
3254   Expr *IdxExpr = AL.getArgAsExpr(1);
3255   uint32_t Idx;
3256   if (!checkUInt32Argument(S, AL, IdxExpr, Idx, 2))
3257     return;
3258 
3259   if (Idx < 1 || Idx > NumArgs) {
3260     S.Diag(AL.getLoc(), diag::err_attribute_argument_out_of_bounds)
3261         << AL << 2 << IdxExpr->getSourceRange();
3262     return;
3263   }
3264 
3265   // FIXME: Do we need to bounds check?
3266   unsigned ArgIdx = Idx - 1;
3267 
3268   if (HasImplicitThisParam) {
3269     if (ArgIdx == 0) {
3270       S.Diag(AL.getLoc(),
3271              diag::err_format_attribute_implicit_this_format_string)
3272         << IdxExpr->getSourceRange();
3273       return;
3274     }
3275     ArgIdx--;
3276   }
3277 
3278   // make sure the format string is really a string
3279   QualType Ty = getFunctionOrMethodParamType(D, ArgIdx);
3280 
3281   if (Kind == CFStringFormat) {
3282     if (!isCFStringType(Ty, S.Context)) {
3283       S.Diag(AL.getLoc(), diag::err_format_attribute_not)
3284         << "a CFString" << IdxExpr->getSourceRange()
3285         << getFunctionOrMethodParamRange(D, ArgIdx);
3286       return;
3287     }
3288   } else if (Kind == NSStringFormat) {
3289     // FIXME: do we need to check if the type is NSString*?  What are the
3290     // semantics?
3291     if (!isNSStringType(Ty, S.Context)) {
3292       S.Diag(AL.getLoc(), diag::err_format_attribute_not)
3293         << "an NSString" << IdxExpr->getSourceRange()
3294         << getFunctionOrMethodParamRange(D, ArgIdx);
3295       return;
3296     }
3297   } else if (!Ty->isPointerType() ||
3298              !Ty->getAs<PointerType>()->getPointeeType()->isCharType()) {
3299     S.Diag(AL.getLoc(), diag::err_format_attribute_not)
3300       << "a string type" << IdxExpr->getSourceRange()
3301       << getFunctionOrMethodParamRange(D, ArgIdx);
3302     return;
3303   }
3304 
3305   // check the 3rd argument
3306   Expr *FirstArgExpr = AL.getArgAsExpr(2);
3307   uint32_t FirstArg;
3308   if (!checkUInt32Argument(S, AL, FirstArgExpr, FirstArg, 3))
3309     return;
3310 
3311   // check if the function is variadic if the 3rd argument non-zero
3312   if (FirstArg != 0) {
3313     if (isFunctionOrMethodVariadic(D)) {
3314       ++NumArgs; // +1 for ...
3315     } else {
3316       S.Diag(D->getLocation(), diag::err_format_attribute_requires_variadic);
3317       return;
3318     }
3319   }
3320 
3321   // strftime requires FirstArg to be 0 because it doesn't read from any
3322   // variable the input is just the current time + the format string.
3323   if (Kind == StrftimeFormat) {
3324     if (FirstArg != 0) {
3325       S.Diag(AL.getLoc(), diag::err_format_strftime_third_parameter)
3326         << FirstArgExpr->getSourceRange();
3327       return;
3328     }
3329   // if 0 it disables parameter checking (to use with e.g. va_list)
3330   } else if (FirstArg != 0 && FirstArg != NumArgs) {
3331     S.Diag(AL.getLoc(), diag::err_attribute_argument_out_of_bounds)
3332         << AL << 3 << FirstArgExpr->getSourceRange();
3333     return;
3334   }
3335 
3336   FormatAttr *NewAttr = S.mergeFormatAttr(D, AL.getRange(), II,
3337                                           Idx, FirstArg,
3338                                           AL.getAttributeSpellingListIndex());
3339   if (NewAttr)
3340     D->addAttr(NewAttr);
3341 }
3342 
3343 static void handleTransparentUnionAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3344   // Try to find the underlying union declaration.
3345   RecordDecl *RD = nullptr;
3346   const auto *TD = dyn_cast<TypedefNameDecl>(D);
3347   if (TD && TD->getUnderlyingType()->isUnionType())
3348     RD = TD->getUnderlyingType()->getAsUnionType()->getDecl();
3349   else
3350     RD = dyn_cast<RecordDecl>(D);
3351 
3352   if (!RD || !RD->isUnion()) {
3353     S.Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type) << AL
3354                                                               << ExpectedUnion;
3355     return;
3356   }
3357 
3358   if (!RD->isCompleteDefinition()) {
3359     if (!RD->isBeingDefined())
3360       S.Diag(AL.getLoc(),
3361              diag::warn_transparent_union_attribute_not_definition);
3362     return;
3363   }
3364 
3365   RecordDecl::field_iterator Field = RD->field_begin(),
3366                           FieldEnd = RD->field_end();
3367   if (Field == FieldEnd) {
3368     S.Diag(AL.getLoc(), diag::warn_transparent_union_attribute_zero_fields);
3369     return;
3370   }
3371 
3372   FieldDecl *FirstField = *Field;
3373   QualType FirstType = FirstField->getType();
3374   if (FirstType->hasFloatingRepresentation() || FirstType->isVectorType()) {
3375     S.Diag(FirstField->getLocation(),
3376            diag::warn_transparent_union_attribute_floating)
3377       << FirstType->isVectorType() << FirstType;
3378     return;
3379   }
3380 
3381   if (FirstType->isIncompleteType())
3382     return;
3383   uint64_t FirstSize = S.Context.getTypeSize(FirstType);
3384   uint64_t FirstAlign = S.Context.getTypeAlign(FirstType);
3385   for (; Field != FieldEnd; ++Field) {
3386     QualType FieldType = Field->getType();
3387     if (FieldType->isIncompleteType())
3388       return;
3389     // FIXME: this isn't fully correct; we also need to test whether the
3390     // members of the union would all have the same calling convention as the
3391     // first member of the union. Checking just the size and alignment isn't
3392     // sufficient (consider structs passed on the stack instead of in registers
3393     // as an example).
3394     if (S.Context.getTypeSize(FieldType) != FirstSize ||
3395         S.Context.getTypeAlign(FieldType) > FirstAlign) {
3396       // Warn if we drop the attribute.
3397       bool isSize = S.Context.getTypeSize(FieldType) != FirstSize;
3398       unsigned FieldBits = isSize? S.Context.getTypeSize(FieldType)
3399                                  : S.Context.getTypeAlign(FieldType);
3400       S.Diag(Field->getLocation(),
3401           diag::warn_transparent_union_attribute_field_size_align)
3402         << isSize << Field->getDeclName() << FieldBits;
3403       unsigned FirstBits = isSize? FirstSize : FirstAlign;
3404       S.Diag(FirstField->getLocation(),
3405              diag::note_transparent_union_first_field_size_align)
3406         << isSize << FirstBits;
3407       return;
3408     }
3409   }
3410 
3411   RD->addAttr(::new (S.Context)
3412               TransparentUnionAttr(AL.getRange(), S.Context,
3413                                    AL.getAttributeSpellingListIndex()));
3414 }
3415 
3416 static void handleAnnotateAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3417   // Make sure that there is a string literal as the annotation's single
3418   // argument.
3419   StringRef Str;
3420   if (!S.checkStringLiteralArgumentAttr(AL, 0, Str))
3421     return;
3422 
3423   // Don't duplicate annotations that are already set.
3424   for (const auto *I : D->specific_attrs<AnnotateAttr>()) {
3425     if (I->getAnnotation() == Str)
3426       return;
3427   }
3428 
3429   D->addAttr(::new (S.Context)
3430              AnnotateAttr(AL.getRange(), S.Context, Str,
3431                           AL.getAttributeSpellingListIndex()));
3432 }
3433 
3434 static void handleAlignValueAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3435   S.AddAlignValueAttr(AL.getRange(), D, AL.getArgAsExpr(0),
3436                       AL.getAttributeSpellingListIndex());
3437 }
3438 
3439 void Sema::AddAlignValueAttr(SourceRange AttrRange, Decl *D, Expr *E,
3440                              unsigned SpellingListIndex) {
3441   AlignValueAttr TmpAttr(AttrRange, Context, E, SpellingListIndex);
3442   SourceLocation AttrLoc = AttrRange.getBegin();
3443 
3444   QualType T;
3445   if (const auto *TD = dyn_cast<TypedefNameDecl>(D))
3446     T = TD->getUnderlyingType();
3447   else if (const auto *VD = dyn_cast<ValueDecl>(D))
3448     T = VD->getType();
3449   else
3450     llvm_unreachable("Unknown decl type for align_value");
3451 
3452   if (!T->isDependentType() && !T->isAnyPointerType() &&
3453       !T->isReferenceType() && !T->isMemberPointerType()) {
3454     Diag(AttrLoc, diag::warn_attribute_pointer_or_reference_only)
3455       << &TmpAttr /*TmpAttr.getName()*/ << T << D->getSourceRange();
3456     return;
3457   }
3458 
3459   if (!E->isValueDependent()) {
3460     llvm::APSInt Alignment;
3461     ExprResult ICE
3462       = VerifyIntegerConstantExpression(E, &Alignment,
3463           diag::err_align_value_attribute_argument_not_int,
3464             /*AllowFold*/ false);
3465     if (ICE.isInvalid())
3466       return;
3467 
3468     if (!Alignment.isPowerOf2()) {
3469       Diag(AttrLoc, diag::err_alignment_not_power_of_two)
3470         << E->getSourceRange();
3471       return;
3472     }
3473 
3474     D->addAttr(::new (Context)
3475                AlignValueAttr(AttrRange, Context, ICE.get(),
3476                SpellingListIndex));
3477     return;
3478   }
3479 
3480   // Save dependent expressions in the AST to be instantiated.
3481   D->addAttr(::new (Context) AlignValueAttr(TmpAttr));
3482 }
3483 
3484 static void handleAlignedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3485   // check the attribute arguments.
3486   if (AL.getNumArgs() > 1) {
3487     S.Diag(AL.getLoc(), diag::err_attribute_wrong_number_arguments) << AL << 1;
3488     return;
3489   }
3490 
3491   if (AL.getNumArgs() == 0) {
3492     D->addAttr(::new (S.Context) AlignedAttr(AL.getRange(), S.Context,
3493                true, nullptr, AL.getAttributeSpellingListIndex()));
3494     return;
3495   }
3496 
3497   Expr *E = AL.getArgAsExpr(0);
3498   if (AL.isPackExpansion() && !E->containsUnexpandedParameterPack()) {
3499     S.Diag(AL.getEllipsisLoc(),
3500            diag::err_pack_expansion_without_parameter_packs);
3501     return;
3502   }
3503 
3504   if (!AL.isPackExpansion() && S.DiagnoseUnexpandedParameterPack(E))
3505     return;
3506 
3507   S.AddAlignedAttr(AL.getRange(), D, E, AL.getAttributeSpellingListIndex(),
3508                    AL.isPackExpansion());
3509 }
3510 
3511 void Sema::AddAlignedAttr(SourceRange AttrRange, Decl *D, Expr *E,
3512                           unsigned SpellingListIndex, bool IsPackExpansion) {
3513   AlignedAttr TmpAttr(AttrRange, Context, true, E, SpellingListIndex);
3514   SourceLocation AttrLoc = AttrRange.getBegin();
3515 
3516   // C++11 alignas(...) and C11 _Alignas(...) have additional requirements.
3517   if (TmpAttr.isAlignas()) {
3518     // C++11 [dcl.align]p1:
3519     //   An alignment-specifier may be applied to a variable or to a class
3520     //   data member, but it shall not be applied to a bit-field, a function
3521     //   parameter, the formal parameter of a catch clause, or a variable
3522     //   declared with the register storage class specifier. An
3523     //   alignment-specifier may also be applied to the declaration of a class
3524     //   or enumeration type.
3525     // C11 6.7.5/2:
3526     //   An alignment attribute shall not be specified in a declaration of
3527     //   a typedef, or a bit-field, or a function, or a parameter, or an
3528     //   object declared with the register storage-class specifier.
3529     int DiagKind = -1;
3530     if (isa<ParmVarDecl>(D)) {
3531       DiagKind = 0;
3532     } else if (const auto *VD = dyn_cast<VarDecl>(D)) {
3533       if (VD->getStorageClass() == SC_Register)
3534         DiagKind = 1;
3535       if (VD->isExceptionVariable())
3536         DiagKind = 2;
3537     } else if (const auto *FD = dyn_cast<FieldDecl>(D)) {
3538       if (FD->isBitField())
3539         DiagKind = 3;
3540     } else if (!isa<TagDecl>(D)) {
3541       Diag(AttrLoc, diag::err_attribute_wrong_decl_type) << &TmpAttr
3542         << (TmpAttr.isC11() ? ExpectedVariableOrField
3543                             : ExpectedVariableFieldOrTag);
3544       return;
3545     }
3546     if (DiagKind != -1) {
3547       Diag(AttrLoc, diag::err_alignas_attribute_wrong_decl_type)
3548         << &TmpAttr << DiagKind;
3549       return;
3550     }
3551   }
3552 
3553   if (E->isValueDependent()) {
3554     // We can't support a dependent alignment on a non-dependent type,
3555     // because we have no way to model that a type is "alignment-dependent"
3556     // but not dependent in any other way.
3557     if (const auto *TND = dyn_cast<TypedefNameDecl>(D)) {
3558       if (!TND->getUnderlyingType()->isDependentType()) {
3559         Diag(AttrLoc, diag::err_alignment_dependent_typedef_name)
3560             << E->getSourceRange();
3561         return;
3562       }
3563     }
3564 
3565     // Save dependent expressions in the AST to be instantiated.
3566     AlignedAttr *AA = ::new (Context) AlignedAttr(TmpAttr);
3567     AA->setPackExpansion(IsPackExpansion);
3568     D->addAttr(AA);
3569     return;
3570   }
3571 
3572   // FIXME: Cache the number on the AL object?
3573   llvm::APSInt Alignment;
3574   ExprResult ICE
3575     = VerifyIntegerConstantExpression(E, &Alignment,
3576         diag::err_aligned_attribute_argument_not_int,
3577         /*AllowFold*/ false);
3578   if (ICE.isInvalid())
3579     return;
3580 
3581   uint64_t AlignVal = Alignment.getZExtValue();
3582 
3583   // C++11 [dcl.align]p2:
3584   //   -- if the constant expression evaluates to zero, the alignment
3585   //      specifier shall have no effect
3586   // C11 6.7.5p6:
3587   //   An alignment specification of zero has no effect.
3588   if (!(TmpAttr.isAlignas() && !Alignment)) {
3589     if (!llvm::isPowerOf2_64(AlignVal)) {
3590       Diag(AttrLoc, diag::err_alignment_not_power_of_two)
3591         << E->getSourceRange();
3592       return;
3593     }
3594   }
3595 
3596   // Alignment calculations can wrap around if it's greater than 2**28.
3597   unsigned MaxValidAlignment =
3598       Context.getTargetInfo().getTriple().isOSBinFormatCOFF() ? 8192
3599                                                               : 268435456;
3600   if (AlignVal > MaxValidAlignment) {
3601     Diag(AttrLoc, diag::err_attribute_aligned_too_great) << MaxValidAlignment
3602                                                          << E->getSourceRange();
3603     return;
3604   }
3605 
3606   if (Context.getTargetInfo().isTLSSupported()) {
3607     unsigned MaxTLSAlign =
3608         Context.toCharUnitsFromBits(Context.getTargetInfo().getMaxTLSAlign())
3609             .getQuantity();
3610     const auto *VD = dyn_cast<VarDecl>(D);
3611     if (MaxTLSAlign && AlignVal > MaxTLSAlign && VD &&
3612         VD->getTLSKind() != VarDecl::TLS_None) {
3613       Diag(VD->getLocation(), diag::err_tls_var_aligned_over_maximum)
3614           << (unsigned)AlignVal << VD << MaxTLSAlign;
3615       return;
3616     }
3617   }
3618 
3619   AlignedAttr *AA = ::new (Context) AlignedAttr(AttrRange, Context, true,
3620                                                 ICE.get(), SpellingListIndex);
3621   AA->setPackExpansion(IsPackExpansion);
3622   D->addAttr(AA);
3623 }
3624 
3625 void Sema::AddAlignedAttr(SourceRange AttrRange, Decl *D, TypeSourceInfo *TS,
3626                           unsigned SpellingListIndex, bool IsPackExpansion) {
3627   // FIXME: Cache the number on the AL object if non-dependent?
3628   // FIXME: Perform checking of type validity
3629   AlignedAttr *AA = ::new (Context) AlignedAttr(AttrRange, Context, false, TS,
3630                                                 SpellingListIndex);
3631   AA->setPackExpansion(IsPackExpansion);
3632   D->addAttr(AA);
3633 }
3634 
3635 void Sema::CheckAlignasUnderalignment(Decl *D) {
3636   assert(D->hasAttrs() && "no attributes on decl");
3637 
3638   QualType UnderlyingTy, DiagTy;
3639   if (const auto *VD = dyn_cast<ValueDecl>(D)) {
3640     UnderlyingTy = DiagTy = VD->getType();
3641   } else {
3642     UnderlyingTy = DiagTy = Context.getTagDeclType(cast<TagDecl>(D));
3643     if (const auto *ED = dyn_cast<EnumDecl>(D))
3644       UnderlyingTy = ED->getIntegerType();
3645   }
3646   if (DiagTy->isDependentType() || DiagTy->isIncompleteType())
3647     return;
3648 
3649   // C++11 [dcl.align]p5, C11 6.7.5/4:
3650   //   The combined effect of all alignment attributes in a declaration shall
3651   //   not specify an alignment that is less strict than the alignment that
3652   //   would otherwise be required for the entity being declared.
3653   AlignedAttr *AlignasAttr = nullptr;
3654   unsigned Align = 0;
3655   for (auto *I : D->specific_attrs<AlignedAttr>()) {
3656     if (I->isAlignmentDependent())
3657       return;
3658     if (I->isAlignas())
3659       AlignasAttr = I;
3660     Align = std::max(Align, I->getAlignment(Context));
3661   }
3662 
3663   if (AlignasAttr && Align) {
3664     CharUnits RequestedAlign = Context.toCharUnitsFromBits(Align);
3665     CharUnits NaturalAlign = Context.getTypeAlignInChars(UnderlyingTy);
3666     if (NaturalAlign > RequestedAlign)
3667       Diag(AlignasAttr->getLocation(), diag::err_alignas_underaligned)
3668         << DiagTy << (unsigned)NaturalAlign.getQuantity();
3669   }
3670 }
3671 
3672 bool Sema::checkMSInheritanceAttrOnDefinition(
3673     CXXRecordDecl *RD, SourceRange Range, bool BestCase,
3674     MSInheritanceAttr::Spelling SemanticSpelling) {
3675   assert(RD->hasDefinition() && "RD has no definition!");
3676 
3677   // We may not have seen base specifiers or any virtual methods yet.  We will
3678   // have to wait until the record is defined to catch any mismatches.
3679   if (!RD->getDefinition()->isCompleteDefinition())
3680     return false;
3681 
3682   // The unspecified model never matches what a definition could need.
3683   if (SemanticSpelling == MSInheritanceAttr::Keyword_unspecified_inheritance)
3684     return false;
3685 
3686   if (BestCase) {
3687     if (RD->calculateInheritanceModel() == SemanticSpelling)
3688       return false;
3689   } else {
3690     if (RD->calculateInheritanceModel() <= SemanticSpelling)
3691       return false;
3692   }
3693 
3694   Diag(Range.getBegin(), diag::err_mismatched_ms_inheritance)
3695       << 0 /*definition*/;
3696   Diag(RD->getDefinition()->getLocation(), diag::note_defined_here)
3697       << RD->getNameAsString();
3698   return true;
3699 }
3700 
3701 /// parseModeAttrArg - Parses attribute mode string and returns parsed type
3702 /// attribute.
3703 static void parseModeAttrArg(Sema &S, StringRef Str, unsigned &DestWidth,
3704                              bool &IntegerMode, bool &ComplexMode) {
3705   IntegerMode = true;
3706   ComplexMode = false;
3707   switch (Str.size()) {
3708   case 2:
3709     switch (Str[0]) {
3710     case 'Q':
3711       DestWidth = 8;
3712       break;
3713     case 'H':
3714       DestWidth = 16;
3715       break;
3716     case 'S':
3717       DestWidth = 32;
3718       break;
3719     case 'D':
3720       DestWidth = 64;
3721       break;
3722     case 'X':
3723       DestWidth = 96;
3724       break;
3725     case 'T':
3726       DestWidth = 128;
3727       break;
3728     }
3729     if (Str[1] == 'F') {
3730       IntegerMode = false;
3731     } else if (Str[1] == 'C') {
3732       IntegerMode = false;
3733       ComplexMode = true;
3734     } else if (Str[1] != 'I') {
3735       DestWidth = 0;
3736     }
3737     break;
3738   case 4:
3739     // FIXME: glibc uses 'word' to define register_t; this is narrower than a
3740     // pointer on PIC16 and other embedded platforms.
3741     if (Str == "word")
3742       DestWidth = S.Context.getTargetInfo().getRegisterWidth();
3743     else if (Str == "byte")
3744       DestWidth = S.Context.getTargetInfo().getCharWidth();
3745     break;
3746   case 7:
3747     if (Str == "pointer")
3748       DestWidth = S.Context.getTargetInfo().getPointerWidth(0);
3749     break;
3750   case 11:
3751     if (Str == "unwind_word")
3752       DestWidth = S.Context.getTargetInfo().getUnwindWordWidth();
3753     break;
3754   }
3755 }
3756 
3757 /// handleModeAttr - This attribute modifies the width of a decl with primitive
3758 /// type.
3759 ///
3760 /// Despite what would be logical, the mode attribute is a decl attribute, not a
3761 /// type attribute: 'int ** __attribute((mode(HI))) *G;' tries to make 'G' be
3762 /// HImode, not an intermediate pointer.
3763 static void handleModeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3764   // This attribute isn't documented, but glibc uses it.  It changes
3765   // the width of an int or unsigned int to the specified size.
3766   if (!AL.isArgIdent(0)) {
3767     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
3768         << AL << AANT_ArgumentIdentifier;
3769     return;
3770   }
3771 
3772   IdentifierInfo *Name = AL.getArgAsIdent(0)->Ident;
3773 
3774   S.AddModeAttr(AL.getRange(), D, Name, AL.getAttributeSpellingListIndex());
3775 }
3776 
3777 void Sema::AddModeAttr(SourceRange AttrRange, Decl *D, IdentifierInfo *Name,
3778                        unsigned SpellingListIndex, bool InInstantiation) {
3779   StringRef Str = Name->getName();
3780   normalizeName(Str);
3781   SourceLocation AttrLoc = AttrRange.getBegin();
3782 
3783   unsigned DestWidth = 0;
3784   bool IntegerMode = true;
3785   bool ComplexMode = false;
3786   llvm::APInt VectorSize(64, 0);
3787   if (Str.size() >= 4 && Str[0] == 'V') {
3788     // Minimal length of vector mode is 4: 'V' + NUMBER(>=1) + TYPE(>=2).
3789     size_t StrSize = Str.size();
3790     size_t VectorStringLength = 0;
3791     while ((VectorStringLength + 1) < StrSize &&
3792            isdigit(Str[VectorStringLength + 1]))
3793       ++VectorStringLength;
3794     if (VectorStringLength &&
3795         !Str.substr(1, VectorStringLength).getAsInteger(10, VectorSize) &&
3796         VectorSize.isPowerOf2()) {
3797       parseModeAttrArg(*this, Str.substr(VectorStringLength + 1), DestWidth,
3798                        IntegerMode, ComplexMode);
3799       // Avoid duplicate warning from template instantiation.
3800       if (!InInstantiation)
3801         Diag(AttrLoc, diag::warn_vector_mode_deprecated);
3802     } else {
3803       VectorSize = 0;
3804     }
3805   }
3806 
3807   if (!VectorSize)
3808     parseModeAttrArg(*this, Str, DestWidth, IntegerMode, ComplexMode);
3809 
3810   // FIXME: Sync this with InitializePredefinedMacros; we need to match int8_t
3811   // and friends, at least with glibc.
3812   // FIXME: Make sure floating-point mappings are accurate
3813   // FIXME: Support XF and TF types
3814   if (!DestWidth) {
3815     Diag(AttrLoc, diag::err_machine_mode) << 0 /*Unknown*/ << Name;
3816     return;
3817   }
3818 
3819   QualType OldTy;
3820   if (const auto *TD = dyn_cast<TypedefNameDecl>(D))
3821     OldTy = TD->getUnderlyingType();
3822   else if (const auto *ED = dyn_cast<EnumDecl>(D)) {
3823     // Something like 'typedef enum { X } __attribute__((mode(XX))) T;'.
3824     // Try to get type from enum declaration, default to int.
3825     OldTy = ED->getIntegerType();
3826     if (OldTy.isNull())
3827       OldTy = Context.IntTy;
3828   } else
3829     OldTy = cast<ValueDecl>(D)->getType();
3830 
3831   if (OldTy->isDependentType()) {
3832     D->addAttr(::new (Context)
3833                ModeAttr(AttrRange, Context, Name, SpellingListIndex));
3834     return;
3835   }
3836 
3837   // Base type can also be a vector type (see PR17453).
3838   // Distinguish between base type and base element type.
3839   QualType OldElemTy = OldTy;
3840   if (const auto *VT = OldTy->getAs<VectorType>())
3841     OldElemTy = VT->getElementType();
3842 
3843   // GCC allows 'mode' attribute on enumeration types (even incomplete), except
3844   // for vector modes. So, 'enum X __attribute__((mode(QI)));' forms a complete
3845   // type, 'enum { A } __attribute__((mode(V4SI)))' is rejected.
3846   if ((isa<EnumDecl>(D) || OldElemTy->getAs<EnumType>()) &&
3847       VectorSize.getBoolValue()) {
3848     Diag(AttrLoc, diag::err_enum_mode_vector_type) << Name << AttrRange;
3849     return;
3850   }
3851   bool IntegralOrAnyEnumType =
3852       OldElemTy->isIntegralOrEnumerationType() || OldElemTy->getAs<EnumType>();
3853 
3854   if (!OldElemTy->getAs<BuiltinType>() && !OldElemTy->isComplexType() &&
3855       !IntegralOrAnyEnumType)
3856     Diag(AttrLoc, diag::err_mode_not_primitive);
3857   else if (IntegerMode) {
3858     if (!IntegralOrAnyEnumType)
3859       Diag(AttrLoc, diag::err_mode_wrong_type);
3860   } else if (ComplexMode) {
3861     if (!OldElemTy->isComplexType())
3862       Diag(AttrLoc, diag::err_mode_wrong_type);
3863   } else {
3864     if (!OldElemTy->isFloatingType())
3865       Diag(AttrLoc, diag::err_mode_wrong_type);
3866   }
3867 
3868   QualType NewElemTy;
3869 
3870   if (IntegerMode)
3871     NewElemTy = Context.getIntTypeForBitwidth(DestWidth,
3872                                               OldElemTy->isSignedIntegerType());
3873   else
3874     NewElemTy = Context.getRealTypeForBitwidth(DestWidth);
3875 
3876   if (NewElemTy.isNull()) {
3877     Diag(AttrLoc, diag::err_machine_mode) << 1 /*Unsupported*/ << Name;
3878     return;
3879   }
3880 
3881   if (ComplexMode) {
3882     NewElemTy = Context.getComplexType(NewElemTy);
3883   }
3884 
3885   QualType NewTy = NewElemTy;
3886   if (VectorSize.getBoolValue()) {
3887     NewTy = Context.getVectorType(NewTy, VectorSize.getZExtValue(),
3888                                   VectorType::GenericVector);
3889   } else if (const auto *OldVT = OldTy->getAs<VectorType>()) {
3890     // Complex machine mode does not support base vector types.
3891     if (ComplexMode) {
3892       Diag(AttrLoc, diag::err_complex_mode_vector_type);
3893       return;
3894     }
3895     unsigned NumElements = Context.getTypeSize(OldElemTy) *
3896                            OldVT->getNumElements() /
3897                            Context.getTypeSize(NewElemTy);
3898     NewTy =
3899         Context.getVectorType(NewElemTy, NumElements, OldVT->getVectorKind());
3900   }
3901 
3902   if (NewTy.isNull()) {
3903     Diag(AttrLoc, diag::err_mode_wrong_type);
3904     return;
3905   }
3906 
3907   // Install the new type.
3908   if (auto *TD = dyn_cast<TypedefNameDecl>(D))
3909     TD->setModedTypeSourceInfo(TD->getTypeSourceInfo(), NewTy);
3910   else if (auto *ED = dyn_cast<EnumDecl>(D))
3911     ED->setIntegerType(NewTy);
3912   else
3913     cast<ValueDecl>(D)->setType(NewTy);
3914 
3915   D->addAttr(::new (Context)
3916              ModeAttr(AttrRange, Context, Name, SpellingListIndex));
3917 }
3918 
3919 static void handleNoDebugAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
3920   D->addAttr(::new (S.Context)
3921              NoDebugAttr(AL.getRange(), S.Context,
3922                          AL.getAttributeSpellingListIndex()));
3923 }
3924 
3925 AlwaysInlineAttr *Sema::mergeAlwaysInlineAttr(Decl *D, SourceRange Range,
3926                                               IdentifierInfo *Ident,
3927                                               unsigned AttrSpellingListIndex) {
3928   if (OptimizeNoneAttr *Optnone = D->getAttr<OptimizeNoneAttr>()) {
3929     Diag(Range.getBegin(), diag::warn_attribute_ignored) << Ident;
3930     Diag(Optnone->getLocation(), diag::note_conflicting_attribute);
3931     return nullptr;
3932   }
3933 
3934   if (D->hasAttr<AlwaysInlineAttr>())
3935     return nullptr;
3936 
3937   return ::new (Context) AlwaysInlineAttr(Range, Context,
3938                                           AttrSpellingListIndex);
3939 }
3940 
3941 CommonAttr *Sema::mergeCommonAttr(Decl *D, const ParsedAttr &AL) {
3942   if (checkAttrMutualExclusion<InternalLinkageAttr>(*this, D, AL))
3943     return nullptr;
3944 
3945   return ::new (Context)
3946       CommonAttr(AL.getRange(), Context, AL.getAttributeSpellingListIndex());
3947 }
3948 
3949 CommonAttr *Sema::mergeCommonAttr(Decl *D, const CommonAttr &AL) {
3950   if (checkAttrMutualExclusion<InternalLinkageAttr>(*this, D, AL))
3951     return nullptr;
3952 
3953   return ::new (Context)
3954       CommonAttr(AL.getRange(), Context, AL.getSpellingListIndex());
3955 }
3956 
3957 InternalLinkageAttr *Sema::mergeInternalLinkageAttr(Decl *D,
3958                                                     const ParsedAttr &AL) {
3959   if (const auto *VD = dyn_cast<VarDecl>(D)) {
3960     // Attribute applies to Var but not any subclass of it (like ParmVar,
3961     // ImplicitParm or VarTemplateSpecialization).
3962     if (VD->getKind() != Decl::Var) {
3963       Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type)
3964           << AL << (getLangOpts().CPlusPlus ? ExpectedFunctionVariableOrClass
3965                                             : ExpectedVariableOrFunction);
3966       return nullptr;
3967     }
3968     // Attribute does not apply to non-static local variables.
3969     if (VD->hasLocalStorage()) {
3970       Diag(VD->getLocation(), diag::warn_internal_linkage_local_storage);
3971       return nullptr;
3972     }
3973   }
3974 
3975   if (checkAttrMutualExclusion<CommonAttr>(*this, D, AL))
3976     return nullptr;
3977 
3978   return ::new (Context) InternalLinkageAttr(
3979       AL.getRange(), Context, AL.getAttributeSpellingListIndex());
3980 }
3981 InternalLinkageAttr *
3982 Sema::mergeInternalLinkageAttr(Decl *D, const InternalLinkageAttr &AL) {
3983   if (const auto *VD = dyn_cast<VarDecl>(D)) {
3984     // Attribute applies to Var but not any subclass of it (like ParmVar,
3985     // ImplicitParm or VarTemplateSpecialization).
3986     if (VD->getKind() != Decl::Var) {
3987       Diag(AL.getLocation(), diag::warn_attribute_wrong_decl_type)
3988           << &AL << (getLangOpts().CPlusPlus ? ExpectedFunctionVariableOrClass
3989                                              : ExpectedVariableOrFunction);
3990       return nullptr;
3991     }
3992     // Attribute does not apply to non-static local variables.
3993     if (VD->hasLocalStorage()) {
3994       Diag(VD->getLocation(), diag::warn_internal_linkage_local_storage);
3995       return nullptr;
3996     }
3997   }
3998 
3999   if (checkAttrMutualExclusion<CommonAttr>(*this, D, AL))
4000     return nullptr;
4001 
4002   return ::new (Context)
4003       InternalLinkageAttr(AL.getRange(), Context, AL.getSpellingListIndex());
4004 }
4005 
4006 MinSizeAttr *Sema::mergeMinSizeAttr(Decl *D, SourceRange Range,
4007                                     unsigned AttrSpellingListIndex) {
4008   if (OptimizeNoneAttr *Optnone = D->getAttr<OptimizeNoneAttr>()) {
4009     Diag(Range.getBegin(), diag::warn_attribute_ignored) << "'minsize'";
4010     Diag(Optnone->getLocation(), diag::note_conflicting_attribute);
4011     return nullptr;
4012   }
4013 
4014   if (D->hasAttr<MinSizeAttr>())
4015     return nullptr;
4016 
4017   return ::new (Context) MinSizeAttr(Range, Context, AttrSpellingListIndex);
4018 }
4019 
4020 OptimizeNoneAttr *Sema::mergeOptimizeNoneAttr(Decl *D, SourceRange Range,
4021                                               unsigned AttrSpellingListIndex) {
4022   if (AlwaysInlineAttr *Inline = D->getAttr<AlwaysInlineAttr>()) {
4023     Diag(Inline->getLocation(), diag::warn_attribute_ignored) << Inline;
4024     Diag(Range.getBegin(), diag::note_conflicting_attribute);
4025     D->dropAttr<AlwaysInlineAttr>();
4026   }
4027   if (MinSizeAttr *MinSize = D->getAttr<MinSizeAttr>()) {
4028     Diag(MinSize->getLocation(), diag::warn_attribute_ignored) << MinSize;
4029     Diag(Range.getBegin(), diag::note_conflicting_attribute);
4030     D->dropAttr<MinSizeAttr>();
4031   }
4032 
4033   if (D->hasAttr<OptimizeNoneAttr>())
4034     return nullptr;
4035 
4036   return ::new (Context) OptimizeNoneAttr(Range, Context,
4037                                           AttrSpellingListIndex);
4038 }
4039 
4040 static void handleAlwaysInlineAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4041   if (checkAttrMutualExclusion<NotTailCalledAttr>(S, D, AL))
4042     return;
4043 
4044   if (AlwaysInlineAttr *Inline = S.mergeAlwaysInlineAttr(
4045           D, AL.getRange(), AL.getName(),
4046           AL.getAttributeSpellingListIndex()))
4047     D->addAttr(Inline);
4048 }
4049 
4050 static void handleMinSizeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4051   if (MinSizeAttr *MinSize = S.mergeMinSizeAttr(
4052           D, AL.getRange(), AL.getAttributeSpellingListIndex()))
4053     D->addAttr(MinSize);
4054 }
4055 
4056 static void handleOptimizeNoneAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4057   if (OptimizeNoneAttr *Optnone = S.mergeOptimizeNoneAttr(
4058           D, AL.getRange(), AL.getAttributeSpellingListIndex()))
4059     D->addAttr(Optnone);
4060 }
4061 
4062 static void handleConstantAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4063   if (checkAttrMutualExclusion<CUDASharedAttr>(S, D, AL))
4064     return;
4065   const auto *VD = cast<VarDecl>(D);
4066   if (!VD->hasGlobalStorage()) {
4067     S.Diag(AL.getLoc(), diag::err_cuda_nonglobal_constant);
4068     return;
4069   }
4070   D->addAttr(::new (S.Context) CUDAConstantAttr(
4071       AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4072 }
4073 
4074 static void handleSharedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4075   if (checkAttrMutualExclusion<CUDAConstantAttr>(S, D, AL))
4076     return;
4077   const auto *VD = cast<VarDecl>(D);
4078   // extern __shared__ is only allowed on arrays with no length (e.g.
4079   // "int x[]").
4080   if (!S.getLangOpts().CUDARelocatableDeviceCode && VD->hasExternalStorage() &&
4081       !isa<IncompleteArrayType>(VD->getType())) {
4082     S.Diag(AL.getLoc(), diag::err_cuda_extern_shared) << VD;
4083     return;
4084   }
4085   if (S.getLangOpts().CUDA && VD->hasLocalStorage() &&
4086       S.CUDADiagIfHostCode(AL.getLoc(), diag::err_cuda_host_shared)
4087           << S.CurrentCUDATarget())
4088     return;
4089   D->addAttr(::new (S.Context) CUDASharedAttr(
4090       AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4091 }
4092 
4093 static void handleGlobalAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4094   if (checkAttrMutualExclusion<CUDADeviceAttr>(S, D, AL) ||
4095       checkAttrMutualExclusion<CUDAHostAttr>(S, D, AL)) {
4096     return;
4097   }
4098   const auto *FD = cast<FunctionDecl>(D);
4099   if (!FD->getReturnType()->isVoidType()) {
4100     SourceRange RTRange = FD->getReturnTypeSourceRange();
4101     S.Diag(FD->getTypeSpecStartLoc(), diag::err_kern_type_not_void_return)
4102         << FD->getType()
4103         << (RTRange.isValid() ? FixItHint::CreateReplacement(RTRange, "void")
4104                               : FixItHint());
4105     return;
4106   }
4107   if (const auto *Method = dyn_cast<CXXMethodDecl>(FD)) {
4108     if (Method->isInstance()) {
4109       S.Diag(Method->getBeginLoc(), diag::err_kern_is_nonstatic_method)
4110           << Method;
4111       return;
4112     }
4113     S.Diag(Method->getBeginLoc(), diag::warn_kern_is_method) << Method;
4114   }
4115   // Only warn for "inline" when compiling for host, to cut down on noise.
4116   if (FD->isInlineSpecified() && !S.getLangOpts().CUDAIsDevice)
4117     S.Diag(FD->getBeginLoc(), diag::warn_kern_is_inline) << FD;
4118 
4119   D->addAttr(::new (S.Context)
4120               CUDAGlobalAttr(AL.getRange(), S.Context,
4121                              AL.getAttributeSpellingListIndex()));
4122 }
4123 
4124 static void handleGNUInlineAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4125   const auto *Fn = cast<FunctionDecl>(D);
4126   if (!Fn->isInlineSpecified()) {
4127     S.Diag(AL.getLoc(), diag::warn_gnu_inline_attribute_requires_inline);
4128     return;
4129   }
4130 
4131   D->addAttr(::new (S.Context)
4132              GNUInlineAttr(AL.getRange(), S.Context,
4133                            AL.getAttributeSpellingListIndex()));
4134 }
4135 
4136 static void handleCallConvAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4137   if (hasDeclarator(D)) return;
4138 
4139   // Diagnostic is emitted elsewhere: here we store the (valid) AL
4140   // in the Decl node for syntactic reasoning, e.g., pretty-printing.
4141   CallingConv CC;
4142   if (S.CheckCallingConvAttr(AL, CC, /*FD*/nullptr))
4143     return;
4144 
4145   if (!isa<ObjCMethodDecl>(D)) {
4146     S.Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type)
4147         << AL << ExpectedFunctionOrMethod;
4148     return;
4149   }
4150 
4151   switch (AL.getKind()) {
4152   case ParsedAttr::AT_FastCall:
4153     D->addAttr(::new (S.Context)
4154                FastCallAttr(AL.getRange(), S.Context,
4155                             AL.getAttributeSpellingListIndex()));
4156     return;
4157   case ParsedAttr::AT_StdCall:
4158     D->addAttr(::new (S.Context)
4159                StdCallAttr(AL.getRange(), S.Context,
4160                            AL.getAttributeSpellingListIndex()));
4161     return;
4162   case ParsedAttr::AT_ThisCall:
4163     D->addAttr(::new (S.Context)
4164                ThisCallAttr(AL.getRange(), S.Context,
4165                             AL.getAttributeSpellingListIndex()));
4166     return;
4167   case ParsedAttr::AT_CDecl:
4168     D->addAttr(::new (S.Context)
4169                CDeclAttr(AL.getRange(), S.Context,
4170                          AL.getAttributeSpellingListIndex()));
4171     return;
4172   case ParsedAttr::AT_Pascal:
4173     D->addAttr(::new (S.Context)
4174                PascalAttr(AL.getRange(), S.Context,
4175                           AL.getAttributeSpellingListIndex()));
4176     return;
4177   case ParsedAttr::AT_SwiftCall:
4178     D->addAttr(::new (S.Context)
4179                SwiftCallAttr(AL.getRange(), S.Context,
4180                              AL.getAttributeSpellingListIndex()));
4181     return;
4182   case ParsedAttr::AT_VectorCall:
4183     D->addAttr(::new (S.Context)
4184                VectorCallAttr(AL.getRange(), S.Context,
4185                               AL.getAttributeSpellingListIndex()));
4186     return;
4187   case ParsedAttr::AT_MSABI:
4188     D->addAttr(::new (S.Context)
4189                MSABIAttr(AL.getRange(), S.Context,
4190                          AL.getAttributeSpellingListIndex()));
4191     return;
4192   case ParsedAttr::AT_SysVABI:
4193     D->addAttr(::new (S.Context)
4194                SysVABIAttr(AL.getRange(), S.Context,
4195                            AL.getAttributeSpellingListIndex()));
4196     return;
4197   case ParsedAttr::AT_RegCall:
4198     D->addAttr(::new (S.Context) RegCallAttr(
4199         AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4200     return;
4201   case ParsedAttr::AT_Pcs: {
4202     PcsAttr::PCSType PCS;
4203     switch (CC) {
4204     case CC_AAPCS:
4205       PCS = PcsAttr::AAPCS;
4206       break;
4207     case CC_AAPCS_VFP:
4208       PCS = PcsAttr::AAPCS_VFP;
4209       break;
4210     default:
4211       llvm_unreachable("unexpected calling convention in pcs attribute");
4212     }
4213 
4214     D->addAttr(::new (S.Context)
4215                PcsAttr(AL.getRange(), S.Context, PCS,
4216                        AL.getAttributeSpellingListIndex()));
4217     return;
4218   }
4219   case ParsedAttr::AT_IntelOclBicc:
4220     D->addAttr(::new (S.Context)
4221                IntelOclBiccAttr(AL.getRange(), S.Context,
4222                                 AL.getAttributeSpellingListIndex()));
4223     return;
4224   case ParsedAttr::AT_PreserveMost:
4225     D->addAttr(::new (S.Context) PreserveMostAttr(
4226         AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4227     return;
4228   case ParsedAttr::AT_PreserveAll:
4229     D->addAttr(::new (S.Context) PreserveAllAttr(
4230         AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4231     return;
4232   default:
4233     llvm_unreachable("unexpected attribute kind");
4234   }
4235 }
4236 
4237 static void handleSuppressAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4238   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
4239     return;
4240 
4241   std::vector<StringRef> DiagnosticIdentifiers;
4242   for (unsigned I = 0, E = AL.getNumArgs(); I != E; ++I) {
4243     StringRef RuleName;
4244 
4245     if (!S.checkStringLiteralArgumentAttr(AL, I, RuleName, nullptr))
4246       return;
4247 
4248     // FIXME: Warn if the rule name is unknown. This is tricky because only
4249     // clang-tidy knows about available rules.
4250     DiagnosticIdentifiers.push_back(RuleName);
4251   }
4252   D->addAttr(::new (S.Context) SuppressAttr(
4253       AL.getRange(), S.Context, DiagnosticIdentifiers.data(),
4254       DiagnosticIdentifiers.size(), AL.getAttributeSpellingListIndex()));
4255 }
4256 
4257 bool Sema::CheckCallingConvAttr(const ParsedAttr &Attrs, CallingConv &CC,
4258                                 const FunctionDecl *FD) {
4259   if (Attrs.isInvalid())
4260     return true;
4261 
4262   if (Attrs.hasProcessingCache()) {
4263     CC = (CallingConv) Attrs.getProcessingCache();
4264     return false;
4265   }
4266 
4267   unsigned ReqArgs = Attrs.getKind() == ParsedAttr::AT_Pcs ? 1 : 0;
4268   if (!checkAttributeNumArgs(*this, Attrs, ReqArgs)) {
4269     Attrs.setInvalid();
4270     return true;
4271   }
4272 
4273   // TODO: diagnose uses of these conventions on the wrong target.
4274   switch (Attrs.getKind()) {
4275   case ParsedAttr::AT_CDecl:
4276     CC = CC_C;
4277     break;
4278   case ParsedAttr::AT_FastCall:
4279     CC = CC_X86FastCall;
4280     break;
4281   case ParsedAttr::AT_StdCall:
4282     CC = CC_X86StdCall;
4283     break;
4284   case ParsedAttr::AT_ThisCall:
4285     CC = CC_X86ThisCall;
4286     break;
4287   case ParsedAttr::AT_Pascal:
4288     CC = CC_X86Pascal;
4289     break;
4290   case ParsedAttr::AT_SwiftCall:
4291     CC = CC_Swift;
4292     break;
4293   case ParsedAttr::AT_VectorCall:
4294     CC = CC_X86VectorCall;
4295     break;
4296   case ParsedAttr::AT_RegCall:
4297     CC = CC_X86RegCall;
4298     break;
4299   case ParsedAttr::AT_MSABI:
4300     CC = Context.getTargetInfo().getTriple().isOSWindows() ? CC_C :
4301                                                              CC_Win64;
4302     break;
4303   case ParsedAttr::AT_SysVABI:
4304     CC = Context.getTargetInfo().getTriple().isOSWindows() ? CC_X86_64SysV :
4305                                                              CC_C;
4306     break;
4307   case ParsedAttr::AT_Pcs: {
4308     StringRef StrRef;
4309     if (!checkStringLiteralArgumentAttr(Attrs, 0, StrRef)) {
4310       Attrs.setInvalid();
4311       return true;
4312     }
4313     if (StrRef == "aapcs") {
4314       CC = CC_AAPCS;
4315       break;
4316     } else if (StrRef == "aapcs-vfp") {
4317       CC = CC_AAPCS_VFP;
4318       break;
4319     }
4320 
4321     Attrs.setInvalid();
4322     Diag(Attrs.getLoc(), diag::err_invalid_pcs);
4323     return true;
4324   }
4325   case ParsedAttr::AT_IntelOclBicc:
4326     CC = CC_IntelOclBicc;
4327     break;
4328   case ParsedAttr::AT_PreserveMost:
4329     CC = CC_PreserveMost;
4330     break;
4331   case ParsedAttr::AT_PreserveAll:
4332     CC = CC_PreserveAll;
4333     break;
4334   default: llvm_unreachable("unexpected attribute kind");
4335   }
4336 
4337   const TargetInfo &TI = Context.getTargetInfo();
4338   TargetInfo::CallingConvCheckResult A = TI.checkCallingConvention(CC);
4339   if (A != TargetInfo::CCCR_OK) {
4340     if (A == TargetInfo::CCCR_Warning)
4341       Diag(Attrs.getLoc(), diag::warn_cconv_ignored) << Attrs;
4342 
4343     // This convention is not valid for the target. Use the default function or
4344     // method calling convention.
4345     bool IsCXXMethod = false, IsVariadic = false;
4346     if (FD) {
4347       IsCXXMethod = FD->isCXXInstanceMember();
4348       IsVariadic = FD->isVariadic();
4349     }
4350     CC = Context.getDefaultCallingConvention(IsVariadic, IsCXXMethod);
4351   }
4352 
4353   Attrs.setProcessingCache((unsigned) CC);
4354   return false;
4355 }
4356 
4357 /// Pointer-like types in the default address space.
4358 static bool isValidSwiftContextType(QualType Ty) {
4359   if (!Ty->hasPointerRepresentation())
4360     return Ty->isDependentType();
4361   return Ty->getPointeeType().getAddressSpace() == LangAS::Default;
4362 }
4363 
4364 /// Pointers and references in the default address space.
4365 static bool isValidSwiftIndirectResultType(QualType Ty) {
4366   if (const auto *PtrType = Ty->getAs<PointerType>()) {
4367     Ty = PtrType->getPointeeType();
4368   } else if (const auto *RefType = Ty->getAs<ReferenceType>()) {
4369     Ty = RefType->getPointeeType();
4370   } else {
4371     return Ty->isDependentType();
4372   }
4373   return Ty.getAddressSpace() == LangAS::Default;
4374 }
4375 
4376 /// Pointers and references to pointers in the default address space.
4377 static bool isValidSwiftErrorResultType(QualType Ty) {
4378   if (const auto *PtrType = Ty->getAs<PointerType>()) {
4379     Ty = PtrType->getPointeeType();
4380   } else if (const auto *RefType = Ty->getAs<ReferenceType>()) {
4381     Ty = RefType->getPointeeType();
4382   } else {
4383     return Ty->isDependentType();
4384   }
4385   if (!Ty.getQualifiers().empty())
4386     return false;
4387   return isValidSwiftContextType(Ty);
4388 }
4389 
4390 static void handleParameterABIAttr(Sema &S, Decl *D, const ParsedAttr &Attrs,
4391                                    ParameterABI Abi) {
4392   S.AddParameterABIAttr(Attrs.getRange(), D, Abi,
4393                         Attrs.getAttributeSpellingListIndex());
4394 }
4395 
4396 void Sema::AddParameterABIAttr(SourceRange range, Decl *D, ParameterABI abi,
4397                                unsigned spellingIndex) {
4398 
4399   QualType type = cast<ParmVarDecl>(D)->getType();
4400 
4401   if (auto existingAttr = D->getAttr<ParameterABIAttr>()) {
4402     if (existingAttr->getABI() != abi) {
4403       Diag(range.getBegin(), diag::err_attributes_are_not_compatible)
4404         << getParameterABISpelling(abi) << existingAttr;
4405       Diag(existingAttr->getLocation(), diag::note_conflicting_attribute);
4406       return;
4407     }
4408   }
4409 
4410   switch (abi) {
4411   case ParameterABI::Ordinary:
4412     llvm_unreachable("explicit attribute for ordinary parameter ABI?");
4413 
4414   case ParameterABI::SwiftContext:
4415     if (!isValidSwiftContextType(type)) {
4416       Diag(range.getBegin(), diag::err_swift_abi_parameter_wrong_type)
4417         << getParameterABISpelling(abi)
4418         << /*pointer to pointer */ 0 << type;
4419     }
4420     D->addAttr(::new (Context)
4421                SwiftContextAttr(range, Context, spellingIndex));
4422     return;
4423 
4424   case ParameterABI::SwiftErrorResult:
4425     if (!isValidSwiftErrorResultType(type)) {
4426       Diag(range.getBegin(), diag::err_swift_abi_parameter_wrong_type)
4427         << getParameterABISpelling(abi)
4428         << /*pointer to pointer */ 1 << type;
4429     }
4430     D->addAttr(::new (Context)
4431                SwiftErrorResultAttr(range, Context, spellingIndex));
4432     return;
4433 
4434   case ParameterABI::SwiftIndirectResult:
4435     if (!isValidSwiftIndirectResultType(type)) {
4436       Diag(range.getBegin(), diag::err_swift_abi_parameter_wrong_type)
4437         << getParameterABISpelling(abi)
4438         << /*pointer*/ 0 << type;
4439     }
4440     D->addAttr(::new (Context)
4441                SwiftIndirectResultAttr(range, Context, spellingIndex));
4442     return;
4443   }
4444   llvm_unreachable("bad parameter ABI attribute");
4445 }
4446 
4447 /// Checks a regparm attribute, returning true if it is ill-formed and
4448 /// otherwise setting numParams to the appropriate value.
4449 bool Sema::CheckRegparmAttr(const ParsedAttr &AL, unsigned &numParams) {
4450   if (AL.isInvalid())
4451     return true;
4452 
4453   if (!checkAttributeNumArgs(*this, AL, 1)) {
4454     AL.setInvalid();
4455     return true;
4456   }
4457 
4458   uint32_t NP;
4459   Expr *NumParamsExpr = AL.getArgAsExpr(0);
4460   if (!checkUInt32Argument(*this, AL, NumParamsExpr, NP)) {
4461     AL.setInvalid();
4462     return true;
4463   }
4464 
4465   if (Context.getTargetInfo().getRegParmMax() == 0) {
4466     Diag(AL.getLoc(), diag::err_attribute_regparm_wrong_platform)
4467       << NumParamsExpr->getSourceRange();
4468     AL.setInvalid();
4469     return true;
4470   }
4471 
4472   numParams = NP;
4473   if (numParams > Context.getTargetInfo().getRegParmMax()) {
4474     Diag(AL.getLoc(), diag::err_attribute_regparm_invalid_number)
4475       << Context.getTargetInfo().getRegParmMax() << NumParamsExpr->getSourceRange();
4476     AL.setInvalid();
4477     return true;
4478   }
4479 
4480   return false;
4481 }
4482 
4483 // Checks whether an argument of launch_bounds attribute is
4484 // acceptable, performs implicit conversion to Rvalue, and returns
4485 // non-nullptr Expr result on success. Otherwise, it returns nullptr
4486 // and may output an error.
4487 static Expr *makeLaunchBoundsArgExpr(Sema &S, Expr *E,
4488                                      const CUDALaunchBoundsAttr &AL,
4489                                      const unsigned Idx) {
4490   if (S.DiagnoseUnexpandedParameterPack(E))
4491     return nullptr;
4492 
4493   // Accept template arguments for now as they depend on something else.
4494   // We'll get to check them when they eventually get instantiated.
4495   if (E->isValueDependent())
4496     return E;
4497 
4498   llvm::APSInt I(64);
4499   if (!E->isIntegerConstantExpr(I, S.Context)) {
4500     S.Diag(E->getExprLoc(), diag::err_attribute_argument_n_type)
4501         << &AL << Idx << AANT_ArgumentIntegerConstant << E->getSourceRange();
4502     return nullptr;
4503   }
4504   // Make sure we can fit it in 32 bits.
4505   if (!I.isIntN(32)) {
4506     S.Diag(E->getExprLoc(), diag::err_ice_too_large) << I.toString(10, false)
4507                                                      << 32 << /* Unsigned */ 1;
4508     return nullptr;
4509   }
4510   if (I < 0)
4511     S.Diag(E->getExprLoc(), diag::warn_attribute_argument_n_negative)
4512         << &AL << Idx << E->getSourceRange();
4513 
4514   // We may need to perform implicit conversion of the argument.
4515   InitializedEntity Entity = InitializedEntity::InitializeParameter(
4516       S.Context, S.Context.getConstType(S.Context.IntTy), /*consume*/ false);
4517   ExprResult ValArg = S.PerformCopyInitialization(Entity, SourceLocation(), E);
4518   assert(!ValArg.isInvalid() &&
4519          "Unexpected PerformCopyInitialization() failure.");
4520 
4521   return ValArg.getAs<Expr>();
4522 }
4523 
4524 void Sema::AddLaunchBoundsAttr(SourceRange AttrRange, Decl *D, Expr *MaxThreads,
4525                                Expr *MinBlocks, unsigned SpellingListIndex) {
4526   CUDALaunchBoundsAttr TmpAttr(AttrRange, Context, MaxThreads, MinBlocks,
4527                                SpellingListIndex);
4528   MaxThreads = makeLaunchBoundsArgExpr(*this, MaxThreads, TmpAttr, 0);
4529   if (MaxThreads == nullptr)
4530     return;
4531 
4532   if (MinBlocks) {
4533     MinBlocks = makeLaunchBoundsArgExpr(*this, MinBlocks, TmpAttr, 1);
4534     if (MinBlocks == nullptr)
4535       return;
4536   }
4537 
4538   D->addAttr(::new (Context) CUDALaunchBoundsAttr(
4539       AttrRange, Context, MaxThreads, MinBlocks, SpellingListIndex));
4540 }
4541 
4542 static void handleLaunchBoundsAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4543   if (!checkAttributeAtLeastNumArgs(S, AL, 1) ||
4544       !checkAttributeAtMostNumArgs(S, AL, 2))
4545     return;
4546 
4547   S.AddLaunchBoundsAttr(AL.getRange(), D, AL.getArgAsExpr(0),
4548                         AL.getNumArgs() > 1 ? AL.getArgAsExpr(1) : nullptr,
4549                         AL.getAttributeSpellingListIndex());
4550 }
4551 
4552 static void handleArgumentWithTypeTagAttr(Sema &S, Decl *D,
4553                                           const ParsedAttr &AL) {
4554   if (!AL.isArgIdent(0)) {
4555     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
4556         << AL << /* arg num = */ 1 << AANT_ArgumentIdentifier;
4557     return;
4558   }
4559 
4560   ParamIdx ArgumentIdx;
4561   if (!checkFunctionOrMethodParameterIndex(S, D, AL, 2, AL.getArgAsExpr(1),
4562                                            ArgumentIdx))
4563     return;
4564 
4565   ParamIdx TypeTagIdx;
4566   if (!checkFunctionOrMethodParameterIndex(S, D, AL, 3, AL.getArgAsExpr(2),
4567                                            TypeTagIdx))
4568     return;
4569 
4570   bool IsPointer = AL.getName()->getName() == "pointer_with_type_tag";
4571   if (IsPointer) {
4572     // Ensure that buffer has a pointer type.
4573     unsigned ArgumentIdxAST = ArgumentIdx.getASTIndex();
4574     if (ArgumentIdxAST >= getFunctionOrMethodNumParams(D) ||
4575         !getFunctionOrMethodParamType(D, ArgumentIdxAST)->isPointerType())
4576       S.Diag(AL.getLoc(), diag::err_attribute_pointers_only) << AL << 0;
4577   }
4578 
4579   D->addAttr(::new (S.Context) ArgumentWithTypeTagAttr(
4580       AL.getRange(), S.Context, AL.getArgAsIdent(0)->Ident, ArgumentIdx,
4581       TypeTagIdx, IsPointer, AL.getAttributeSpellingListIndex()));
4582 }
4583 
4584 static void handleTypeTagForDatatypeAttr(Sema &S, Decl *D,
4585                                          const ParsedAttr &AL) {
4586   if (!AL.isArgIdent(0)) {
4587     S.Diag(AL.getLoc(), diag::err_attribute_argument_n_type)
4588         << AL << 1 << AANT_ArgumentIdentifier;
4589     return;
4590   }
4591 
4592   if (!checkAttributeNumArgs(S, AL, 1))
4593     return;
4594 
4595   if (!isa<VarDecl>(D)) {
4596     S.Diag(AL.getLoc(), diag::err_attribute_wrong_decl_type)
4597         << AL << ExpectedVariable;
4598     return;
4599   }
4600 
4601   IdentifierInfo *PointerKind = AL.getArgAsIdent(0)->Ident;
4602   TypeSourceInfo *MatchingCTypeLoc = nullptr;
4603   S.GetTypeFromParser(AL.getMatchingCType(), &MatchingCTypeLoc);
4604   assert(MatchingCTypeLoc && "no type source info for attribute argument");
4605 
4606   D->addAttr(::new (S.Context)
4607              TypeTagForDatatypeAttr(AL.getRange(), S.Context, PointerKind,
4608                                     MatchingCTypeLoc,
4609                                     AL.getLayoutCompatible(),
4610                                     AL.getMustBeNull(),
4611                                     AL.getAttributeSpellingListIndex()));
4612 }
4613 
4614 static void handleXRayLogArgsAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4615   ParamIdx ArgCount;
4616 
4617   if (!checkFunctionOrMethodParameterIndex(S, D, AL, 1, AL.getArgAsExpr(0),
4618                                            ArgCount,
4619                                            true /* CanIndexImplicitThis */))
4620     return;
4621 
4622   // ArgCount isn't a parameter index [0;n), it's a count [1;n]
4623   D->addAttr(::new (S.Context) XRayLogArgsAttr(
4624       AL.getRange(), S.Context, ArgCount.getSourceIndex(),
4625       AL.getAttributeSpellingListIndex()));
4626 }
4627 
4628 //===----------------------------------------------------------------------===//
4629 // Checker-specific attribute handlers.
4630 //===----------------------------------------------------------------------===//
4631 
4632 static bool isValidSubjectOfNSReturnsRetainedAttribute(QualType QT) {
4633   return QT->isDependentType() || QT->isObjCRetainableType();
4634 }
4635 
4636 static bool isValidSubjectOfNSAttribute(Sema &S, QualType QT) {
4637   return QT->isDependentType() || QT->isObjCObjectPointerType() ||
4638          S.Context.isObjCNSObjectType(QT);
4639 }
4640 
4641 static bool isValidSubjectOfCFAttribute(Sema &S, QualType QT) {
4642   return QT->isDependentType() || QT->isPointerType() ||
4643          isValidSubjectOfNSAttribute(S, QT);
4644 }
4645 
4646 static void handleNSConsumedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4647   S.AddNSConsumedAttr(AL.getRange(), D, AL.getAttributeSpellingListIndex(),
4648                       AL.getKind() == ParsedAttr::AT_NSConsumed,
4649                       /*template instantiation*/ false);
4650 }
4651 
4652 void Sema::AddNSConsumedAttr(SourceRange AttrRange, Decl *D,
4653                              unsigned SpellingIndex, bool IsNSConsumed,
4654                              bool IsTemplateInstantiation) {
4655   const auto *Param = cast<ParmVarDecl>(D);
4656   bool TypeOK;
4657 
4658   if (IsNSConsumed)
4659     TypeOK = isValidSubjectOfNSAttribute(*this, Param->getType());
4660   else
4661     TypeOK = isValidSubjectOfCFAttribute(*this, Param->getType());
4662 
4663   if (!TypeOK) {
4664     // These attributes are normally just advisory, but in ARC, ns_consumed
4665     // is significant.  Allow non-dependent code to contain inappropriate
4666     // attributes even in ARC, but require template instantiations to be
4667     // set up correctly.
4668     Diag(D->getBeginLoc(), (IsTemplateInstantiation && IsNSConsumed &&
4669                                     getLangOpts().ObjCAutoRefCount
4670                                 ? diag::err_ns_attribute_wrong_parameter_type
4671                                 : diag::warn_ns_attribute_wrong_parameter_type))
4672         << AttrRange << (IsNSConsumed ? "ns_consumed" : "cf_consumed")
4673         << (IsNSConsumed ? /*objc pointers*/ 0 : /*cf pointers*/ 1);
4674     return;
4675   }
4676 
4677   if (IsNSConsumed)
4678     D->addAttr(::new (Context)
4679                    NSConsumedAttr(AttrRange, Context, SpellingIndex));
4680   else
4681     D->addAttr(::new (Context)
4682                    CFConsumedAttr(AttrRange, Context, SpellingIndex));
4683 }
4684 
4685 bool Sema::checkNSReturnsRetainedReturnType(SourceLocation Loc, QualType QT) {
4686   if (isValidSubjectOfNSReturnsRetainedAttribute(QT))
4687     return false;
4688 
4689   Diag(Loc, diag::warn_ns_attribute_wrong_return_type)
4690       << "'ns_returns_retained'" << 0 << 0;
4691   return true;
4692 }
4693 
4694 static void handleNSReturnsRetainedAttr(Sema &S, Decl *D,
4695                                         const ParsedAttr &AL) {
4696   QualType ReturnType;
4697 
4698   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D))
4699     ReturnType = MD->getReturnType();
4700   else if (S.getLangOpts().ObjCAutoRefCount && hasDeclarator(D) &&
4701            (AL.getKind() == ParsedAttr::AT_NSReturnsRetained))
4702     return; // ignore: was handled as a type attribute
4703   else if (const auto *PD = dyn_cast<ObjCPropertyDecl>(D))
4704     ReturnType = PD->getType();
4705   else if (const auto *FD = dyn_cast<FunctionDecl>(D))
4706     ReturnType = FD->getReturnType();
4707   else if (const auto *Param = dyn_cast<ParmVarDecl>(D)) {
4708     ReturnType = Param->getType()->getPointeeType();
4709     if (ReturnType.isNull()) {
4710       S.Diag(D->getBeginLoc(), diag::warn_ns_attribute_wrong_parameter_type)
4711           << AL << /*pointer-to-CF*/ 2 << AL.getRange();
4712       return;
4713     }
4714   } else if (AL.isUsedAsTypeAttr()) {
4715     return;
4716   } else {
4717     AttributeDeclKind ExpectedDeclKind;
4718     switch (AL.getKind()) {
4719     default: llvm_unreachable("invalid ownership attribute");
4720     case ParsedAttr::AT_NSReturnsRetained:
4721     case ParsedAttr::AT_NSReturnsAutoreleased:
4722     case ParsedAttr::AT_NSReturnsNotRetained:
4723       ExpectedDeclKind = ExpectedFunctionOrMethod;
4724       break;
4725 
4726     case ParsedAttr::AT_CFReturnsRetained:
4727     case ParsedAttr::AT_CFReturnsNotRetained:
4728       ExpectedDeclKind = ExpectedFunctionMethodOrParameter;
4729       break;
4730     }
4731     S.Diag(D->getBeginLoc(), diag::warn_attribute_wrong_decl_type)
4732         << AL.getRange() << AL << ExpectedDeclKind;
4733     return;
4734   }
4735 
4736   bool TypeOK;
4737   bool Cf;
4738   switch (AL.getKind()) {
4739   default: llvm_unreachable("invalid ownership attribute");
4740   case ParsedAttr::AT_NSReturnsRetained:
4741     TypeOK = isValidSubjectOfNSReturnsRetainedAttribute(ReturnType);
4742     Cf = false;
4743     break;
4744 
4745   case ParsedAttr::AT_NSReturnsAutoreleased:
4746   case ParsedAttr::AT_NSReturnsNotRetained:
4747     TypeOK = isValidSubjectOfNSAttribute(S, ReturnType);
4748     Cf = false;
4749     break;
4750 
4751   case ParsedAttr::AT_CFReturnsRetained:
4752   case ParsedAttr::AT_CFReturnsNotRetained:
4753     TypeOK = isValidSubjectOfCFAttribute(S, ReturnType);
4754     Cf = true;
4755     break;
4756   }
4757 
4758   if (!TypeOK) {
4759     if (AL.isUsedAsTypeAttr())
4760       return;
4761 
4762     if (isa<ParmVarDecl>(D)) {
4763       S.Diag(D->getBeginLoc(), diag::warn_ns_attribute_wrong_parameter_type)
4764           << AL << /*pointer-to-CF*/ 2 << AL.getRange();
4765     } else {
4766       // Needs to be kept in sync with warn_ns_attribute_wrong_return_type.
4767       enum : unsigned {
4768         Function,
4769         Method,
4770         Property
4771       } SubjectKind = Function;
4772       if (isa<ObjCMethodDecl>(D))
4773         SubjectKind = Method;
4774       else if (isa<ObjCPropertyDecl>(D))
4775         SubjectKind = Property;
4776       S.Diag(D->getBeginLoc(), diag::warn_ns_attribute_wrong_return_type)
4777           << AL << SubjectKind << Cf << AL.getRange();
4778     }
4779     return;
4780   }
4781 
4782   switch (AL.getKind()) {
4783     default:
4784       llvm_unreachable("invalid ownership attribute");
4785     case ParsedAttr::AT_NSReturnsAutoreleased:
4786       D->addAttr(::new (S.Context) NSReturnsAutoreleasedAttr(
4787           AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4788       return;
4789     case ParsedAttr::AT_CFReturnsNotRetained:
4790       D->addAttr(::new (S.Context) CFReturnsNotRetainedAttr(
4791           AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4792       return;
4793     case ParsedAttr::AT_NSReturnsNotRetained:
4794       D->addAttr(::new (S.Context) NSReturnsNotRetainedAttr(
4795           AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4796       return;
4797     case ParsedAttr::AT_CFReturnsRetained:
4798       D->addAttr(::new (S.Context) CFReturnsRetainedAttr(
4799           AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4800       return;
4801     case ParsedAttr::AT_NSReturnsRetained:
4802       D->addAttr(::new (S.Context) NSReturnsRetainedAttr(
4803           AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
4804       return;
4805   };
4806 }
4807 
4808 static void handleObjCReturnsInnerPointerAttr(Sema &S, Decl *D,
4809                                               const ParsedAttr &Attrs) {
4810   const int EP_ObjCMethod = 1;
4811   const int EP_ObjCProperty = 2;
4812 
4813   SourceLocation loc = Attrs.getLoc();
4814   QualType resultType;
4815   if (isa<ObjCMethodDecl>(D))
4816     resultType = cast<ObjCMethodDecl>(D)->getReturnType();
4817   else
4818     resultType = cast<ObjCPropertyDecl>(D)->getType();
4819 
4820   if (!resultType->isReferenceType() &&
4821       (!resultType->isPointerType() || resultType->isObjCRetainableType())) {
4822     S.Diag(D->getBeginLoc(), diag::warn_ns_attribute_wrong_return_type)
4823         << SourceRange(loc) << Attrs
4824         << (isa<ObjCMethodDecl>(D) ? EP_ObjCMethod : EP_ObjCProperty)
4825         << /*non-retainable pointer*/ 2;
4826 
4827     // Drop the attribute.
4828     return;
4829   }
4830 
4831   D->addAttr(::new (S.Context) ObjCReturnsInnerPointerAttr(
4832       Attrs.getRange(), S.Context, Attrs.getAttributeSpellingListIndex()));
4833 }
4834 
4835 static void handleObjCRequiresSuperAttr(Sema &S, Decl *D,
4836                                         const ParsedAttr &Attrs) {
4837   const auto *Method = cast<ObjCMethodDecl>(D);
4838 
4839   const DeclContext *DC = Method->getDeclContext();
4840   if (const auto *PDecl = dyn_cast_or_null<ObjCProtocolDecl>(DC)) {
4841     S.Diag(D->getBeginLoc(), diag::warn_objc_requires_super_protocol) << Attrs
4842                                                                       << 0;
4843     S.Diag(PDecl->getLocation(), diag::note_protocol_decl);
4844     return;
4845   }
4846   if (Method->getMethodFamily() == OMF_dealloc) {
4847     S.Diag(D->getBeginLoc(), diag::warn_objc_requires_super_protocol) << Attrs
4848                                                                       << 1;
4849     return;
4850   }
4851 
4852   D->addAttr(::new (S.Context) ObjCRequiresSuperAttr(
4853       Attrs.getRange(), S.Context, Attrs.getAttributeSpellingListIndex()));
4854 }
4855 
4856 static void handleObjCBridgeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4857   IdentifierLoc *Parm = AL.isArgIdent(0) ? AL.getArgAsIdent(0) : nullptr;
4858 
4859   if (!Parm) {
4860     S.Diag(D->getBeginLoc(), diag::err_objc_attr_not_id) << AL << 0;
4861     return;
4862   }
4863 
4864   // Typedefs only allow objc_bridge(id) and have some additional checking.
4865   if (const auto *TD = dyn_cast<TypedefNameDecl>(D)) {
4866     if (!Parm->Ident->isStr("id")) {
4867       S.Diag(AL.getLoc(), diag::err_objc_attr_typedef_not_id) << AL;
4868       return;
4869     }
4870 
4871     // Only allow 'cv void *'.
4872     QualType T = TD->getUnderlyingType();
4873     if (!T->isVoidPointerType()) {
4874       S.Diag(AL.getLoc(), diag::err_objc_attr_typedef_not_void_pointer);
4875       return;
4876     }
4877   }
4878 
4879   D->addAttr(::new (S.Context)
4880              ObjCBridgeAttr(AL.getRange(), S.Context, Parm->Ident,
4881                            AL.getAttributeSpellingListIndex()));
4882 }
4883 
4884 static void handleObjCBridgeMutableAttr(Sema &S, Decl *D,
4885                                         const ParsedAttr &AL) {
4886   IdentifierLoc *Parm = AL.isArgIdent(0) ? AL.getArgAsIdent(0) : nullptr;
4887 
4888   if (!Parm) {
4889     S.Diag(D->getBeginLoc(), diag::err_objc_attr_not_id) << AL << 0;
4890     return;
4891   }
4892 
4893   D->addAttr(::new (S.Context)
4894              ObjCBridgeMutableAttr(AL.getRange(), S.Context, Parm->Ident,
4895                             AL.getAttributeSpellingListIndex()));
4896 }
4897 
4898 static void handleObjCBridgeRelatedAttr(Sema &S, Decl *D,
4899                                         const ParsedAttr &AL) {
4900   IdentifierInfo *RelatedClass =
4901       AL.isArgIdent(0) ? AL.getArgAsIdent(0)->Ident : nullptr;
4902   if (!RelatedClass) {
4903     S.Diag(D->getBeginLoc(), diag::err_objc_attr_not_id) << AL << 0;
4904     return;
4905   }
4906   IdentifierInfo *ClassMethod =
4907     AL.getArgAsIdent(1) ? AL.getArgAsIdent(1)->Ident : nullptr;
4908   IdentifierInfo *InstanceMethod =
4909     AL.getArgAsIdent(2) ? AL.getArgAsIdent(2)->Ident : nullptr;
4910   D->addAttr(::new (S.Context)
4911              ObjCBridgeRelatedAttr(AL.getRange(), S.Context, RelatedClass,
4912                                    ClassMethod, InstanceMethod,
4913                                    AL.getAttributeSpellingListIndex()));
4914 }
4915 
4916 static void handleObjCDesignatedInitializer(Sema &S, Decl *D,
4917                                             const ParsedAttr &AL) {
4918   ObjCInterfaceDecl *IFace;
4919   if (auto *CatDecl = dyn_cast<ObjCCategoryDecl>(D->getDeclContext()))
4920     IFace = CatDecl->getClassInterface();
4921   else
4922     IFace = cast<ObjCInterfaceDecl>(D->getDeclContext());
4923 
4924   if (!IFace)
4925     return;
4926 
4927   IFace->setHasDesignatedInitializers();
4928   D->addAttr(::new (S.Context)
4929                   ObjCDesignatedInitializerAttr(AL.getRange(), S.Context,
4930                                          AL.getAttributeSpellingListIndex()));
4931 }
4932 
4933 static void handleObjCRuntimeName(Sema &S, Decl *D, const ParsedAttr &AL) {
4934   StringRef MetaDataName;
4935   if (!S.checkStringLiteralArgumentAttr(AL, 0, MetaDataName))
4936     return;
4937   D->addAttr(::new (S.Context)
4938              ObjCRuntimeNameAttr(AL.getRange(), S.Context,
4939                                  MetaDataName,
4940                                  AL.getAttributeSpellingListIndex()));
4941 }
4942 
4943 // When a user wants to use objc_boxable with a union or struct
4944 // but they don't have access to the declaration (legacy/third-party code)
4945 // then they can 'enable' this feature with a typedef:
4946 // typedef struct __attribute((objc_boxable)) legacy_struct legacy_struct;
4947 static void handleObjCBoxable(Sema &S, Decl *D, const ParsedAttr &AL) {
4948   bool notify = false;
4949 
4950   auto *RD = dyn_cast<RecordDecl>(D);
4951   if (RD && RD->getDefinition()) {
4952     RD = RD->getDefinition();
4953     notify = true;
4954   }
4955 
4956   if (RD) {
4957     ObjCBoxableAttr *BoxableAttr = ::new (S.Context)
4958                           ObjCBoxableAttr(AL.getRange(), S.Context,
4959                                           AL.getAttributeSpellingListIndex());
4960     RD->addAttr(BoxableAttr);
4961     if (notify) {
4962       // we need to notify ASTReader/ASTWriter about
4963       // modification of existing declaration
4964       if (ASTMutationListener *L = S.getASTMutationListener())
4965         L->AddedAttributeToRecord(BoxableAttr, RD);
4966     }
4967   }
4968 }
4969 
4970 static void handleObjCOwnershipAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
4971   if (hasDeclarator(D)) return;
4972 
4973   S.Diag(D->getBeginLoc(), diag::err_attribute_wrong_decl_type)
4974       << AL.getRange() << AL << ExpectedVariable;
4975 }
4976 
4977 static void handleObjCPreciseLifetimeAttr(Sema &S, Decl *D,
4978                                           const ParsedAttr &AL) {
4979   const auto *VD = cast<ValueDecl>(D);
4980   QualType QT = VD->getType();
4981 
4982   if (!QT->isDependentType() &&
4983       !QT->isObjCLifetimeType()) {
4984     S.Diag(AL.getLoc(), diag::err_objc_precise_lifetime_bad_type)
4985       << QT;
4986     return;
4987   }
4988 
4989   Qualifiers::ObjCLifetime Lifetime = QT.getObjCLifetime();
4990 
4991   // If we have no lifetime yet, check the lifetime we're presumably
4992   // going to infer.
4993   if (Lifetime == Qualifiers::OCL_None && !QT->isDependentType())
4994     Lifetime = QT->getObjCARCImplicitLifetime();
4995 
4996   switch (Lifetime) {
4997   case Qualifiers::OCL_None:
4998     assert(QT->isDependentType() &&
4999            "didn't infer lifetime for non-dependent type?");
5000     break;
5001 
5002   case Qualifiers::OCL_Weak:   // meaningful
5003   case Qualifiers::OCL_Strong: // meaningful
5004     break;
5005 
5006   case Qualifiers::OCL_ExplicitNone:
5007   case Qualifiers::OCL_Autoreleasing:
5008     S.Diag(AL.getLoc(), diag::warn_objc_precise_lifetime_meaningless)
5009         << (Lifetime == Qualifiers::OCL_Autoreleasing);
5010     break;
5011   }
5012 
5013   D->addAttr(::new (S.Context)
5014              ObjCPreciseLifetimeAttr(AL.getRange(), S.Context,
5015                                      AL.getAttributeSpellingListIndex()));
5016 }
5017 
5018 //===----------------------------------------------------------------------===//
5019 // Microsoft specific attribute handlers.
5020 //===----------------------------------------------------------------------===//
5021 
5022 UuidAttr *Sema::mergeUuidAttr(Decl *D, SourceRange Range,
5023                               unsigned AttrSpellingListIndex, StringRef Uuid) {
5024   if (const auto *UA = D->getAttr<UuidAttr>()) {
5025     if (UA->getGuid().equals_lower(Uuid))
5026       return nullptr;
5027     Diag(UA->getLocation(), diag::err_mismatched_uuid);
5028     Diag(Range.getBegin(), diag::note_previous_uuid);
5029     D->dropAttr<UuidAttr>();
5030   }
5031 
5032   return ::new (Context) UuidAttr(Range, Context, Uuid, AttrSpellingListIndex);
5033 }
5034 
5035 static void handleUuidAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5036   if (!S.LangOpts.CPlusPlus) {
5037     S.Diag(AL.getLoc(), diag::err_attribute_not_supported_in_lang)
5038         << AL << AttributeLangSupport::C;
5039     return;
5040   }
5041 
5042   StringRef StrRef;
5043   SourceLocation LiteralLoc;
5044   if (!S.checkStringLiteralArgumentAttr(AL, 0, StrRef, &LiteralLoc))
5045     return;
5046 
5047   // GUID format is "XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX" or
5048   // "{XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}", normalize to the former.
5049   if (StrRef.size() == 38 && StrRef.front() == '{' && StrRef.back() == '}')
5050     StrRef = StrRef.drop_front().drop_back();
5051 
5052   // Validate GUID length.
5053   if (StrRef.size() != 36) {
5054     S.Diag(LiteralLoc, diag::err_attribute_uuid_malformed_guid);
5055     return;
5056   }
5057 
5058   for (unsigned i = 0; i < 36; ++i) {
5059     if (i == 8 || i == 13 || i == 18 || i == 23) {
5060       if (StrRef[i] != '-') {
5061         S.Diag(LiteralLoc, diag::err_attribute_uuid_malformed_guid);
5062         return;
5063       }
5064     } else if (!isHexDigit(StrRef[i])) {
5065       S.Diag(LiteralLoc, diag::err_attribute_uuid_malformed_guid);
5066       return;
5067     }
5068   }
5069 
5070   // FIXME: It'd be nice to also emit a fixit removing uuid(...) (and, if it's
5071   // the only thing in the [] list, the [] too), and add an insertion of
5072   // __declspec(uuid(...)).  But sadly, neither the SourceLocs of the commas
5073   // separating attributes nor of the [ and the ] are in the AST.
5074   // Cf "SourceLocations of attribute list delimiters - [[ ... , ... ]] etc"
5075   // on cfe-dev.
5076   if (AL.isMicrosoftAttribute()) // Check for [uuid(...)] spelling.
5077     S.Diag(AL.getLoc(), diag::warn_atl_uuid_deprecated);
5078 
5079   UuidAttr *UA = S.mergeUuidAttr(D, AL.getRange(),
5080                                  AL.getAttributeSpellingListIndex(), StrRef);
5081   if (UA)
5082     D->addAttr(UA);
5083 }
5084 
5085 static void handleMSInheritanceAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5086   if (!S.LangOpts.CPlusPlus) {
5087     S.Diag(AL.getLoc(), diag::err_attribute_not_supported_in_lang)
5088         << AL << AttributeLangSupport::C;
5089     return;
5090   }
5091   MSInheritanceAttr *IA = S.mergeMSInheritanceAttr(
5092       D, AL.getRange(), /*BestCase=*/true,
5093       AL.getAttributeSpellingListIndex(),
5094       (MSInheritanceAttr::Spelling)AL.getSemanticSpelling());
5095   if (IA) {
5096     D->addAttr(IA);
5097     S.Consumer.AssignInheritanceModel(cast<CXXRecordDecl>(D));
5098   }
5099 }
5100 
5101 static void handleDeclspecThreadAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5102   const auto *VD = cast<VarDecl>(D);
5103   if (!S.Context.getTargetInfo().isTLSSupported()) {
5104     S.Diag(AL.getLoc(), diag::err_thread_unsupported);
5105     return;
5106   }
5107   if (VD->getTSCSpec() != TSCS_unspecified) {
5108     S.Diag(AL.getLoc(), diag::err_declspec_thread_on_thread_variable);
5109     return;
5110   }
5111   if (VD->hasLocalStorage()) {
5112     S.Diag(AL.getLoc(), diag::err_thread_non_global) << "__declspec(thread)";
5113     return;
5114   }
5115   D->addAttr(::new (S.Context) ThreadAttr(AL.getRange(), S.Context,
5116                                           AL.getAttributeSpellingListIndex()));
5117 }
5118 
5119 static void handleAbiTagAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5120   SmallVector<StringRef, 4> Tags;
5121   for (unsigned I = 0, E = AL.getNumArgs(); I != E; ++I) {
5122     StringRef Tag;
5123     if (!S.checkStringLiteralArgumentAttr(AL, I, Tag))
5124       return;
5125     Tags.push_back(Tag);
5126   }
5127 
5128   if (const auto *NS = dyn_cast<NamespaceDecl>(D)) {
5129     if (!NS->isInline()) {
5130       S.Diag(AL.getLoc(), diag::warn_attr_abi_tag_namespace) << 0;
5131       return;
5132     }
5133     if (NS->isAnonymousNamespace()) {
5134       S.Diag(AL.getLoc(), diag::warn_attr_abi_tag_namespace) << 1;
5135       return;
5136     }
5137     if (AL.getNumArgs() == 0)
5138       Tags.push_back(NS->getName());
5139   } else if (!checkAttributeAtLeastNumArgs(S, AL, 1))
5140     return;
5141 
5142   // Store tags sorted and without duplicates.
5143   llvm::sort(Tags.begin(), Tags.end());
5144   Tags.erase(std::unique(Tags.begin(), Tags.end()), Tags.end());
5145 
5146   D->addAttr(::new (S.Context)
5147              AbiTagAttr(AL.getRange(), S.Context, Tags.data(), Tags.size(),
5148                         AL.getAttributeSpellingListIndex()));
5149 }
5150 
5151 static void handleARMInterruptAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5152   // Check the attribute arguments.
5153   if (AL.getNumArgs() > 1) {
5154     S.Diag(AL.getLoc(), diag::err_attribute_too_many_arguments) << AL << 1;
5155     return;
5156   }
5157 
5158   StringRef Str;
5159   SourceLocation ArgLoc;
5160 
5161   if (AL.getNumArgs() == 0)
5162     Str = "";
5163   else if (!S.checkStringLiteralArgumentAttr(AL, 0, Str, &ArgLoc))
5164     return;
5165 
5166   ARMInterruptAttr::InterruptType Kind;
5167   if (!ARMInterruptAttr::ConvertStrToInterruptType(Str, Kind)) {
5168     S.Diag(AL.getLoc(), diag::warn_attribute_type_not_supported) << AL << Str
5169                                                                  << ArgLoc;
5170     return;
5171   }
5172 
5173   unsigned Index = AL.getAttributeSpellingListIndex();
5174   D->addAttr(::new (S.Context)
5175              ARMInterruptAttr(AL.getLoc(), S.Context, Kind, Index));
5176 }
5177 
5178 static void handleMSP430InterruptAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5179   if (!checkAttributeNumArgs(S, AL, 1))
5180     return;
5181 
5182   if (!AL.isArgExpr(0)) {
5183     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
5184         << AL << AANT_ArgumentIntegerConstant;
5185     return;
5186   }
5187 
5188   // FIXME: Check for decl - it should be void ()(void).
5189 
5190   Expr *NumParamsExpr = static_cast<Expr *>(AL.getArgAsExpr(0));
5191   llvm::APSInt NumParams(32);
5192   if (!NumParamsExpr->isIntegerConstantExpr(NumParams, S.Context)) {
5193     S.Diag(AL.getLoc(), diag::err_attribute_argument_type)
5194         << AL << AANT_ArgumentIntegerConstant
5195         << NumParamsExpr->getSourceRange();
5196     return;
5197   }
5198 
5199   unsigned Num = NumParams.getLimitedValue(255);
5200   if ((Num & 1) || Num > 30) {
5201     S.Diag(AL.getLoc(), diag::err_attribute_argument_out_of_bounds)
5202         << AL << (int)NumParams.getSExtValue()
5203         << NumParamsExpr->getSourceRange();
5204     return;
5205   }
5206 
5207   D->addAttr(::new (S.Context)
5208               MSP430InterruptAttr(AL.getLoc(), S.Context, Num,
5209                                   AL.getAttributeSpellingListIndex()));
5210   D->addAttr(UsedAttr::CreateImplicit(S.Context));
5211 }
5212 
5213 static void handleMipsInterruptAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5214   // Only one optional argument permitted.
5215   if (AL.getNumArgs() > 1) {
5216     S.Diag(AL.getLoc(), diag::err_attribute_too_many_arguments) << AL << 1;
5217     return;
5218   }
5219 
5220   StringRef Str;
5221   SourceLocation ArgLoc;
5222 
5223   if (AL.getNumArgs() == 0)
5224     Str = "";
5225   else if (!S.checkStringLiteralArgumentAttr(AL, 0, Str, &ArgLoc))
5226     return;
5227 
5228   // Semantic checks for a function with the 'interrupt' attribute for MIPS:
5229   // a) Must be a function.
5230   // b) Must have no parameters.
5231   // c) Must have the 'void' return type.
5232   // d) Cannot have the 'mips16' attribute, as that instruction set
5233   //    lacks the 'eret' instruction.
5234   // e) The attribute itself must either have no argument or one of the
5235   //    valid interrupt types, see [MipsInterruptDocs].
5236 
5237   if (!isFunctionOrMethod(D)) {
5238     S.Diag(D->getLocation(), diag::warn_attribute_wrong_decl_type)
5239         << "'interrupt'" << ExpectedFunctionOrMethod;
5240     return;
5241   }
5242 
5243   if (hasFunctionProto(D) && getFunctionOrMethodNumParams(D) != 0) {
5244     S.Diag(D->getLocation(), diag::warn_mips_interrupt_attribute)
5245         << 0;
5246     return;
5247   }
5248 
5249   if (!getFunctionOrMethodResultType(D)->isVoidType()) {
5250     S.Diag(D->getLocation(), diag::warn_mips_interrupt_attribute)
5251         << 1;
5252     return;
5253   }
5254 
5255   if (checkAttrMutualExclusion<Mips16Attr>(S, D, AL))
5256     return;
5257 
5258   MipsInterruptAttr::InterruptType Kind;
5259   if (!MipsInterruptAttr::ConvertStrToInterruptType(Str, Kind)) {
5260     S.Diag(AL.getLoc(), diag::warn_attribute_type_not_supported)
5261         << AL << "'" + std::string(Str) + "'";
5262     return;
5263   }
5264 
5265   D->addAttr(::new (S.Context) MipsInterruptAttr(
5266       AL.getLoc(), S.Context, Kind, AL.getAttributeSpellingListIndex()));
5267 }
5268 
5269 static void handleAnyX86InterruptAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5270   // Semantic checks for a function with the 'interrupt' attribute.
5271   // a) Must be a function.
5272   // b) Must have the 'void' return type.
5273   // c) Must take 1 or 2 arguments.
5274   // d) The 1st argument must be a pointer.
5275   // e) The 2nd argument (if any) must be an unsigned integer.
5276   if (!isFunctionOrMethod(D) || !hasFunctionProto(D) || isInstanceMethod(D) ||
5277       CXXMethodDecl::isStaticOverloadedOperator(
5278           cast<NamedDecl>(D)->getDeclName().getCXXOverloadedOperator())) {
5279     S.Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type)
5280         << AL << ExpectedFunctionWithProtoType;
5281     return;
5282   }
5283   // Interrupt handler must have void return type.
5284   if (!getFunctionOrMethodResultType(D)->isVoidType()) {
5285     S.Diag(getFunctionOrMethodResultSourceRange(D).getBegin(),
5286            diag::err_anyx86_interrupt_attribute)
5287         << (S.Context.getTargetInfo().getTriple().getArch() == llvm::Triple::x86
5288                 ? 0
5289                 : 1)
5290         << 0;
5291     return;
5292   }
5293   // Interrupt handler must have 1 or 2 parameters.
5294   unsigned NumParams = getFunctionOrMethodNumParams(D);
5295   if (NumParams < 1 || NumParams > 2) {
5296     S.Diag(D->getBeginLoc(), diag::err_anyx86_interrupt_attribute)
5297         << (S.Context.getTargetInfo().getTriple().getArch() == llvm::Triple::x86
5298                 ? 0
5299                 : 1)
5300         << 1;
5301     return;
5302   }
5303   // The first argument must be a pointer.
5304   if (!getFunctionOrMethodParamType(D, 0)->isPointerType()) {
5305     S.Diag(getFunctionOrMethodParamRange(D, 0).getBegin(),
5306            diag::err_anyx86_interrupt_attribute)
5307         << (S.Context.getTargetInfo().getTriple().getArch() == llvm::Triple::x86
5308                 ? 0
5309                 : 1)
5310         << 2;
5311     return;
5312   }
5313   // The second argument, if present, must be an unsigned integer.
5314   unsigned TypeSize =
5315       S.Context.getTargetInfo().getTriple().getArch() == llvm::Triple::x86_64
5316           ? 64
5317           : 32;
5318   if (NumParams == 2 &&
5319       (!getFunctionOrMethodParamType(D, 1)->isUnsignedIntegerType() ||
5320        S.Context.getTypeSize(getFunctionOrMethodParamType(D, 1)) != TypeSize)) {
5321     S.Diag(getFunctionOrMethodParamRange(D, 1).getBegin(),
5322            diag::err_anyx86_interrupt_attribute)
5323         << (S.Context.getTargetInfo().getTriple().getArch() == llvm::Triple::x86
5324                 ? 0
5325                 : 1)
5326         << 3 << S.Context.getIntTypeForBitwidth(TypeSize, /*Signed=*/false);
5327     return;
5328   }
5329   D->addAttr(::new (S.Context) AnyX86InterruptAttr(
5330       AL.getLoc(), S.Context, AL.getAttributeSpellingListIndex()));
5331   D->addAttr(UsedAttr::CreateImplicit(S.Context));
5332 }
5333 
5334 static void handleAVRInterruptAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5335   if (!isFunctionOrMethod(D)) {
5336     S.Diag(D->getLocation(), diag::warn_attribute_wrong_decl_type)
5337         << "'interrupt'" << ExpectedFunction;
5338     return;
5339   }
5340 
5341   if (!checkAttributeNumArgs(S, AL, 0))
5342     return;
5343 
5344   handleSimpleAttribute<AVRInterruptAttr>(S, D, AL);
5345 }
5346 
5347 static void handleAVRSignalAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5348   if (!isFunctionOrMethod(D)) {
5349     S.Diag(D->getLocation(), diag::warn_attribute_wrong_decl_type)
5350         << "'signal'" << ExpectedFunction;
5351     return;
5352   }
5353 
5354   if (!checkAttributeNumArgs(S, AL, 0))
5355     return;
5356 
5357   handleSimpleAttribute<AVRSignalAttr>(S, D, AL);
5358 }
5359 
5360 
5361 static void handleRISCVInterruptAttr(Sema &S, Decl *D,
5362                                      const ParsedAttr &AL) {
5363   // Warn about repeated attributes.
5364   if (const auto *A = D->getAttr<RISCVInterruptAttr>()) {
5365     S.Diag(AL.getRange().getBegin(),
5366       diag::warn_riscv_repeated_interrupt_attribute);
5367     S.Diag(A->getLocation(), diag::note_riscv_repeated_interrupt_attribute);
5368     return;
5369   }
5370 
5371   // Check the attribute argument. Argument is optional.
5372   if (!checkAttributeAtMostNumArgs(S, AL, 1))
5373     return;
5374 
5375   StringRef Str;
5376   SourceLocation ArgLoc;
5377 
5378   // 'machine'is the default interrupt mode.
5379   if (AL.getNumArgs() == 0)
5380     Str = "machine";
5381   else if (!S.checkStringLiteralArgumentAttr(AL, 0, Str, &ArgLoc))
5382     return;
5383 
5384   // Semantic checks for a function with the 'interrupt' attribute:
5385   // - Must be a function.
5386   // - Must have no parameters.
5387   // - Must have the 'void' return type.
5388   // - The attribute itself must either have no argument or one of the
5389   //   valid interrupt types, see [RISCVInterruptDocs].
5390 
5391   if (D->getFunctionType() == nullptr) {
5392     S.Diag(D->getLocation(), diag::warn_attribute_wrong_decl_type)
5393       << "'interrupt'" << ExpectedFunction;
5394     return;
5395   }
5396 
5397   if (hasFunctionProto(D) && getFunctionOrMethodNumParams(D) != 0) {
5398     S.Diag(D->getLocation(), diag::warn_riscv_interrupt_attribute) << 0;
5399     return;
5400   }
5401 
5402   if (!getFunctionOrMethodResultType(D)->isVoidType()) {
5403     S.Diag(D->getLocation(), diag::warn_riscv_interrupt_attribute) << 1;
5404     return;
5405   }
5406 
5407   RISCVInterruptAttr::InterruptType Kind;
5408   if (!RISCVInterruptAttr::ConvertStrToInterruptType(Str, Kind)) {
5409     S.Diag(AL.getLoc(), diag::warn_attribute_type_not_supported) << AL << Str
5410                                                                  << ArgLoc;
5411     return;
5412   }
5413 
5414   D->addAttr(::new (S.Context) RISCVInterruptAttr(
5415     AL.getLoc(), S.Context, Kind, AL.getAttributeSpellingListIndex()));
5416 }
5417 
5418 static void handleInterruptAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5419   // Dispatch the interrupt attribute based on the current target.
5420   switch (S.Context.getTargetInfo().getTriple().getArch()) {
5421   case llvm::Triple::msp430:
5422     handleMSP430InterruptAttr(S, D, AL);
5423     break;
5424   case llvm::Triple::mipsel:
5425   case llvm::Triple::mips:
5426     handleMipsInterruptAttr(S, D, AL);
5427     break;
5428   case llvm::Triple::x86:
5429   case llvm::Triple::x86_64:
5430     handleAnyX86InterruptAttr(S, D, AL);
5431     break;
5432   case llvm::Triple::avr:
5433     handleAVRInterruptAttr(S, D, AL);
5434     break;
5435   case llvm::Triple::riscv32:
5436   case llvm::Triple::riscv64:
5437     handleRISCVInterruptAttr(S, D, AL);
5438     break;
5439   default:
5440     handleARMInterruptAttr(S, D, AL);
5441     break;
5442   }
5443 }
5444 
5445 static void handleAMDGPUFlatWorkGroupSizeAttr(Sema &S, Decl *D,
5446                                               const ParsedAttr &AL) {
5447   uint32_t Min = 0;
5448   Expr *MinExpr = AL.getArgAsExpr(0);
5449   if (!checkUInt32Argument(S, AL, MinExpr, Min))
5450     return;
5451 
5452   uint32_t Max = 0;
5453   Expr *MaxExpr = AL.getArgAsExpr(1);
5454   if (!checkUInt32Argument(S, AL, MaxExpr, Max))
5455     return;
5456 
5457   if (Min == 0 && Max != 0) {
5458     S.Diag(AL.getLoc(), diag::err_attribute_argument_invalid) << AL << 0;
5459     return;
5460   }
5461   if (Min > Max) {
5462     S.Diag(AL.getLoc(), diag::err_attribute_argument_invalid) << AL << 1;
5463     return;
5464   }
5465 
5466   D->addAttr(::new (S.Context)
5467              AMDGPUFlatWorkGroupSizeAttr(AL.getLoc(), S.Context, Min, Max,
5468                                          AL.getAttributeSpellingListIndex()));
5469 }
5470 
5471 static void handleAMDGPUWavesPerEUAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5472   uint32_t Min = 0;
5473   Expr *MinExpr = AL.getArgAsExpr(0);
5474   if (!checkUInt32Argument(S, AL, MinExpr, Min))
5475     return;
5476 
5477   uint32_t Max = 0;
5478   if (AL.getNumArgs() == 2) {
5479     Expr *MaxExpr = AL.getArgAsExpr(1);
5480     if (!checkUInt32Argument(S, AL, MaxExpr, Max))
5481       return;
5482   }
5483 
5484   if (Min == 0 && Max != 0) {
5485     S.Diag(AL.getLoc(), diag::err_attribute_argument_invalid) << AL << 0;
5486     return;
5487   }
5488   if (Max != 0 && Min > Max) {
5489     S.Diag(AL.getLoc(), diag::err_attribute_argument_invalid) << AL << 1;
5490     return;
5491   }
5492 
5493   D->addAttr(::new (S.Context)
5494              AMDGPUWavesPerEUAttr(AL.getLoc(), S.Context, Min, Max,
5495                                   AL.getAttributeSpellingListIndex()));
5496 }
5497 
5498 static void handleAMDGPUNumSGPRAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5499   uint32_t NumSGPR = 0;
5500   Expr *NumSGPRExpr = AL.getArgAsExpr(0);
5501   if (!checkUInt32Argument(S, AL, NumSGPRExpr, NumSGPR))
5502     return;
5503 
5504   D->addAttr(::new (S.Context)
5505              AMDGPUNumSGPRAttr(AL.getLoc(), S.Context, NumSGPR,
5506                                AL.getAttributeSpellingListIndex()));
5507 }
5508 
5509 static void handleAMDGPUNumVGPRAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5510   uint32_t NumVGPR = 0;
5511   Expr *NumVGPRExpr = AL.getArgAsExpr(0);
5512   if (!checkUInt32Argument(S, AL, NumVGPRExpr, NumVGPR))
5513     return;
5514 
5515   D->addAttr(::new (S.Context)
5516              AMDGPUNumVGPRAttr(AL.getLoc(), S.Context, NumVGPR,
5517                                AL.getAttributeSpellingListIndex()));
5518 }
5519 
5520 static void handleX86ForceAlignArgPointerAttr(Sema &S, Decl *D,
5521                                               const ParsedAttr &AL) {
5522   // If we try to apply it to a function pointer, don't warn, but don't
5523   // do anything, either. It doesn't matter anyway, because there's nothing
5524   // special about calling a force_align_arg_pointer function.
5525   const auto *VD = dyn_cast<ValueDecl>(D);
5526   if (VD && VD->getType()->isFunctionPointerType())
5527     return;
5528   // Also don't warn on function pointer typedefs.
5529   const auto *TD = dyn_cast<TypedefNameDecl>(D);
5530   if (TD && (TD->getUnderlyingType()->isFunctionPointerType() ||
5531     TD->getUnderlyingType()->isFunctionType()))
5532     return;
5533   // Attribute can only be applied to function types.
5534   if (!isa<FunctionDecl>(D)) {
5535     S.Diag(AL.getLoc(), diag::warn_attribute_wrong_decl_type)
5536         << AL << ExpectedFunction;
5537     return;
5538   }
5539 
5540   D->addAttr(::new (S.Context)
5541               X86ForceAlignArgPointerAttr(AL.getRange(), S.Context,
5542                                         AL.getAttributeSpellingListIndex()));
5543 }
5544 
5545 static void handleLayoutVersion(Sema &S, Decl *D, const ParsedAttr &AL) {
5546   uint32_t Version;
5547   Expr *VersionExpr = static_cast<Expr *>(AL.getArgAsExpr(0));
5548   if (!checkUInt32Argument(S, AL, AL.getArgAsExpr(0), Version))
5549     return;
5550 
5551   // TODO: Investigate what happens with the next major version of MSVC.
5552   if (Version != LangOptions::MSVC2015) {
5553     S.Diag(AL.getLoc(), diag::err_attribute_argument_out_of_bounds)
5554         << AL << Version << VersionExpr->getSourceRange();
5555     return;
5556   }
5557 
5558   D->addAttr(::new (S.Context)
5559                  LayoutVersionAttr(AL.getRange(), S.Context, Version,
5560                                    AL.getAttributeSpellingListIndex()));
5561 }
5562 
5563 DLLImportAttr *Sema::mergeDLLImportAttr(Decl *D, SourceRange Range,
5564                                         unsigned AttrSpellingListIndex) {
5565   if (D->hasAttr<DLLExportAttr>()) {
5566     Diag(Range.getBegin(), diag::warn_attribute_ignored) << "'dllimport'";
5567     return nullptr;
5568   }
5569 
5570   if (D->hasAttr<DLLImportAttr>())
5571     return nullptr;
5572 
5573   return ::new (Context) DLLImportAttr(Range, Context, AttrSpellingListIndex);
5574 }
5575 
5576 DLLExportAttr *Sema::mergeDLLExportAttr(Decl *D, SourceRange Range,
5577                                         unsigned AttrSpellingListIndex) {
5578   if (DLLImportAttr *Import = D->getAttr<DLLImportAttr>()) {
5579     Diag(Import->getLocation(), diag::warn_attribute_ignored) << Import;
5580     D->dropAttr<DLLImportAttr>();
5581   }
5582 
5583   if (D->hasAttr<DLLExportAttr>())
5584     return nullptr;
5585 
5586   return ::new (Context) DLLExportAttr(Range, Context, AttrSpellingListIndex);
5587 }
5588 
5589 static void handleDLLAttr(Sema &S, Decl *D, const ParsedAttr &A) {
5590   if (isa<ClassTemplatePartialSpecializationDecl>(D) &&
5591       S.Context.getTargetInfo().getCXXABI().isMicrosoft()) {
5592     S.Diag(A.getRange().getBegin(), diag::warn_attribute_ignored) << A;
5593     return;
5594   }
5595 
5596   if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
5597     if (FD->isInlined() && A.getKind() == ParsedAttr::AT_DLLImport &&
5598         !S.Context.getTargetInfo().getCXXABI().isMicrosoft()) {
5599       // MinGW doesn't allow dllimport on inline functions.
5600       S.Diag(A.getRange().getBegin(), diag::warn_attribute_ignored_on_inline)
5601           << A;
5602       return;
5603     }
5604   }
5605 
5606   if (const auto *MD = dyn_cast<CXXMethodDecl>(D)) {
5607     if (S.Context.getTargetInfo().getCXXABI().isMicrosoft() &&
5608         MD->getParent()->isLambda()) {
5609       S.Diag(A.getRange().getBegin(), diag::err_attribute_dll_lambda) << A;
5610       return;
5611     }
5612   }
5613 
5614   unsigned Index = A.getAttributeSpellingListIndex();
5615   Attr *NewAttr = A.getKind() == ParsedAttr::AT_DLLExport
5616                       ? (Attr *)S.mergeDLLExportAttr(D, A.getRange(), Index)
5617                       : (Attr *)S.mergeDLLImportAttr(D, A.getRange(), Index);
5618   if (NewAttr)
5619     D->addAttr(NewAttr);
5620 }
5621 
5622 MSInheritanceAttr *
5623 Sema::mergeMSInheritanceAttr(Decl *D, SourceRange Range, bool BestCase,
5624                              unsigned AttrSpellingListIndex,
5625                              MSInheritanceAttr::Spelling SemanticSpelling) {
5626   if (MSInheritanceAttr *IA = D->getAttr<MSInheritanceAttr>()) {
5627     if (IA->getSemanticSpelling() == SemanticSpelling)
5628       return nullptr;
5629     Diag(IA->getLocation(), diag::err_mismatched_ms_inheritance)
5630         << 1 /*previous declaration*/;
5631     Diag(Range.getBegin(), diag::note_previous_ms_inheritance);
5632     D->dropAttr<MSInheritanceAttr>();
5633   }
5634 
5635   auto *RD = cast<CXXRecordDecl>(D);
5636   if (RD->hasDefinition()) {
5637     if (checkMSInheritanceAttrOnDefinition(RD, Range, BestCase,
5638                                            SemanticSpelling)) {
5639       return nullptr;
5640     }
5641   } else {
5642     if (isa<ClassTemplatePartialSpecializationDecl>(RD)) {
5643       Diag(Range.getBegin(), diag::warn_ignored_ms_inheritance)
5644           << 1 /*partial specialization*/;
5645       return nullptr;
5646     }
5647     if (RD->getDescribedClassTemplate()) {
5648       Diag(Range.getBegin(), diag::warn_ignored_ms_inheritance)
5649           << 0 /*primary template*/;
5650       return nullptr;
5651     }
5652   }
5653 
5654   return ::new (Context)
5655       MSInheritanceAttr(Range, Context, BestCase, AttrSpellingListIndex);
5656 }
5657 
5658 static void handleCapabilityAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5659   // The capability attributes take a single string parameter for the name of
5660   // the capability they represent. The lockable attribute does not take any
5661   // parameters. However, semantically, both attributes represent the same
5662   // concept, and so they use the same semantic attribute. Eventually, the
5663   // lockable attribute will be removed.
5664   //
5665   // For backward compatibility, any capability which has no specified string
5666   // literal will be considered a "mutex."
5667   StringRef N("mutex");
5668   SourceLocation LiteralLoc;
5669   if (AL.getKind() == ParsedAttr::AT_Capability &&
5670       !S.checkStringLiteralArgumentAttr(AL, 0, N, &LiteralLoc))
5671     return;
5672 
5673   // Currently, there are only two names allowed for a capability: role and
5674   // mutex (case insensitive). Diagnose other capability names.
5675   if (!N.equals_lower("mutex") && !N.equals_lower("role"))
5676     S.Diag(LiteralLoc, diag::warn_invalid_capability_name) << N;
5677 
5678   D->addAttr(::new (S.Context) CapabilityAttr(AL.getRange(), S.Context, N,
5679                                         AL.getAttributeSpellingListIndex()));
5680 }
5681 
5682 static void handleAssertCapabilityAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5683   SmallVector<Expr*, 1> Args;
5684   if (!checkLockFunAttrCommon(S, D, AL, Args))
5685     return;
5686 
5687   D->addAttr(::new (S.Context) AssertCapabilityAttr(AL.getRange(), S.Context,
5688                                                     Args.data(), Args.size(),
5689                                         AL.getAttributeSpellingListIndex()));
5690 }
5691 
5692 static void handleAcquireCapabilityAttr(Sema &S, Decl *D,
5693                                         const ParsedAttr &AL) {
5694   SmallVector<Expr*, 1> Args;
5695   if (!checkLockFunAttrCommon(S, D, AL, Args))
5696     return;
5697 
5698   D->addAttr(::new (S.Context) AcquireCapabilityAttr(AL.getRange(),
5699                                                      S.Context,
5700                                                      Args.data(), Args.size(),
5701                                         AL.getAttributeSpellingListIndex()));
5702 }
5703 
5704 static void handleTryAcquireCapabilityAttr(Sema &S, Decl *D,
5705                                            const ParsedAttr &AL) {
5706   SmallVector<Expr*, 2> Args;
5707   if (!checkTryLockFunAttrCommon(S, D, AL, Args))
5708     return;
5709 
5710   D->addAttr(::new (S.Context) TryAcquireCapabilityAttr(AL.getRange(),
5711                                                         S.Context,
5712                                                         AL.getArgAsExpr(0),
5713                                                         Args.data(),
5714                                                         Args.size(),
5715                                         AL.getAttributeSpellingListIndex()));
5716 }
5717 
5718 static void handleReleaseCapabilityAttr(Sema &S, Decl *D,
5719                                         const ParsedAttr &AL) {
5720   // Check that all arguments are lockable objects.
5721   SmallVector<Expr *, 1> Args;
5722   checkAttrArgsAreCapabilityObjs(S, D, AL, Args, 0, true);
5723 
5724   D->addAttr(::new (S.Context) ReleaseCapabilityAttr(
5725       AL.getRange(), S.Context, Args.data(), Args.size(),
5726       AL.getAttributeSpellingListIndex()));
5727 }
5728 
5729 static void handleRequiresCapabilityAttr(Sema &S, Decl *D,
5730                                          const ParsedAttr &AL) {
5731   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
5732     return;
5733 
5734   // check that all arguments are lockable objects
5735   SmallVector<Expr*, 1> Args;
5736   checkAttrArgsAreCapabilityObjs(S, D, AL, Args);
5737   if (Args.empty())
5738     return;
5739 
5740   RequiresCapabilityAttr *RCA = ::new (S.Context)
5741     RequiresCapabilityAttr(AL.getRange(), S.Context, Args.data(),
5742                            Args.size(), AL.getAttributeSpellingListIndex());
5743 
5744   D->addAttr(RCA);
5745 }
5746 
5747 static void handleDeprecatedAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5748   if (const auto *NSD = dyn_cast<NamespaceDecl>(D)) {
5749     if (NSD->isAnonymousNamespace()) {
5750       S.Diag(AL.getLoc(), diag::warn_deprecated_anonymous_namespace);
5751       // Do not want to attach the attribute to the namespace because that will
5752       // cause confusing diagnostic reports for uses of declarations within the
5753       // namespace.
5754       return;
5755     }
5756   }
5757 
5758   // Handle the cases where the attribute has a text message.
5759   StringRef Str, Replacement;
5760   if (AL.isArgExpr(0) && AL.getArgAsExpr(0) &&
5761       !S.checkStringLiteralArgumentAttr(AL, 0, Str))
5762     return;
5763 
5764   // Only support a single optional message for Declspec and CXX11.
5765   if (AL.isDeclspecAttribute() || AL.isCXX11Attribute())
5766     checkAttributeAtMostNumArgs(S, AL, 1);
5767   else if (AL.isArgExpr(1) && AL.getArgAsExpr(1) &&
5768            !S.checkStringLiteralArgumentAttr(AL, 1, Replacement))
5769     return;
5770 
5771   if (!S.getLangOpts().CPlusPlus14)
5772     if (AL.isCXX11Attribute() &&
5773         !(AL.hasScope() && AL.getScopeName()->isStr("gnu")))
5774       S.Diag(AL.getLoc(), diag::ext_cxx14_attr) << AL;
5775 
5776   D->addAttr(::new (S.Context)
5777                  DeprecatedAttr(AL.getRange(), S.Context, Str, Replacement,
5778                                 AL.getAttributeSpellingListIndex()));
5779 }
5780 
5781 static bool isGlobalVar(const Decl *D) {
5782   if (const auto *S = dyn_cast<VarDecl>(D))
5783     return S->hasGlobalStorage();
5784   return false;
5785 }
5786 
5787 static void handleNoSanitizeAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5788   if (!checkAttributeAtLeastNumArgs(S, AL, 1))
5789     return;
5790 
5791   std::vector<StringRef> Sanitizers;
5792 
5793   for (unsigned I = 0, E = AL.getNumArgs(); I != E; ++I) {
5794     StringRef SanitizerName;
5795     SourceLocation LiteralLoc;
5796 
5797     if (!S.checkStringLiteralArgumentAttr(AL, I, SanitizerName, &LiteralLoc))
5798       return;
5799 
5800     if (parseSanitizerValue(SanitizerName, /*AllowGroups=*/true) == 0)
5801       S.Diag(LiteralLoc, diag::warn_unknown_sanitizer_ignored) << SanitizerName;
5802     else if (isGlobalVar(D) && SanitizerName != "address")
5803       S.Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
5804           << AL << ExpectedFunctionOrMethod;
5805     Sanitizers.push_back(SanitizerName);
5806   }
5807 
5808   D->addAttr(::new (S.Context) NoSanitizeAttr(
5809       AL.getRange(), S.Context, Sanitizers.data(), Sanitizers.size(),
5810       AL.getAttributeSpellingListIndex()));
5811 }
5812 
5813 static void handleNoSanitizeSpecificAttr(Sema &S, Decl *D,
5814                                          const ParsedAttr &AL) {
5815   StringRef AttrName = AL.getName()->getName();
5816   normalizeName(AttrName);
5817   StringRef SanitizerName = llvm::StringSwitch<StringRef>(AttrName)
5818                                 .Case("no_address_safety_analysis", "address")
5819                                 .Case("no_sanitize_address", "address")
5820                                 .Case("no_sanitize_thread", "thread")
5821                                 .Case("no_sanitize_memory", "memory");
5822   if (isGlobalVar(D) && SanitizerName != "address")
5823     S.Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
5824         << AL << ExpectedFunction;
5825   D->addAttr(::new (S.Context)
5826                  NoSanitizeAttr(AL.getRange(), S.Context, &SanitizerName, 1,
5827                                 AL.getAttributeSpellingListIndex()));
5828 }
5829 
5830 static void handleInternalLinkageAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5831   if (InternalLinkageAttr *Internal = S.mergeInternalLinkageAttr(D, AL))
5832     D->addAttr(Internal);
5833 }
5834 
5835 static void handleOpenCLNoSVMAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5836   if (S.LangOpts.OpenCLVersion != 200)
5837     S.Diag(AL.getLoc(), diag::err_attribute_requires_opencl_version)
5838         << AL << "2.0" << 0;
5839   else
5840     S.Diag(AL.getLoc(), diag::warn_opencl_attr_deprecated_ignored) << AL
5841                                                                    << "2.0";
5842 }
5843 
5844 /// Handles semantic checking for features that are common to all attributes,
5845 /// such as checking whether a parameter was properly specified, or the correct
5846 /// number of arguments were passed, etc.
5847 static bool handleCommonAttributeFeatures(Sema &S, Decl *D,
5848                                           const ParsedAttr &AL) {
5849   // Several attributes carry different semantics than the parsing requires, so
5850   // those are opted out of the common argument checks.
5851   //
5852   // We also bail on unknown and ignored attributes because those are handled
5853   // as part of the target-specific handling logic.
5854   if (AL.getKind() == ParsedAttr::UnknownAttribute)
5855     return false;
5856   // Check whether the attribute requires specific language extensions to be
5857   // enabled.
5858   if (!AL.diagnoseLangOpts(S))
5859     return true;
5860   // Check whether the attribute appertains to the given subject.
5861   if (!AL.diagnoseAppertainsTo(S, D))
5862     return true;
5863   if (AL.hasCustomParsing())
5864     return false;
5865 
5866   if (AL.getMinArgs() == AL.getMaxArgs()) {
5867     // If there are no optional arguments, then checking for the argument count
5868     // is trivial.
5869     if (!checkAttributeNumArgs(S, AL, AL.getMinArgs()))
5870       return true;
5871   } else {
5872     // There are optional arguments, so checking is slightly more involved.
5873     if (AL.getMinArgs() &&
5874         !checkAttributeAtLeastNumArgs(S, AL, AL.getMinArgs()))
5875       return true;
5876     else if (!AL.hasVariadicArg() && AL.getMaxArgs() &&
5877              !checkAttributeAtMostNumArgs(S, AL, AL.getMaxArgs()))
5878       return true;
5879   }
5880 
5881   if (S.CheckAttrTarget(AL))
5882     return true;
5883 
5884   return false;
5885 }
5886 
5887 static void handleOpenCLAccessAttr(Sema &S, Decl *D, const ParsedAttr &AL) {
5888   if (D->isInvalidDecl())
5889     return;
5890 
5891   // Check if there is only one access qualifier.
5892   if (D->hasAttr<OpenCLAccessAttr>()) {
5893     S.Diag(AL.getLoc(), diag::err_opencl_multiple_access_qualifiers)
5894         << D->getSourceRange();
5895     D->setInvalidDecl(true);
5896     return;
5897   }
5898 
5899   // OpenCL v2.0 s6.6 - read_write can be used for image types to specify that an
5900   // image object can be read and written.
5901   // OpenCL v2.0 s6.13.6 - A kernel cannot read from and write to the same pipe
5902   // object. Using the read_write (or __read_write) qualifier with the pipe
5903   // qualifier is a compilation error.
5904   if (const auto *PDecl = dyn_cast<ParmVarDecl>(D)) {
5905     const Type *DeclTy = PDecl->getType().getCanonicalType().getTypePtr();
5906     if (AL.getName()->getName().find("read_write") != StringRef::npos) {
5907       if (S.getLangOpts().OpenCLVersion < 200 || DeclTy->isPipeType()) {
5908         S.Diag(AL.getLoc(), diag::err_opencl_invalid_read_write)
5909             << AL << PDecl->getType() << DeclTy->isImageType();
5910         D->setInvalidDecl(true);
5911         return;
5912       }
5913     }
5914   }
5915 
5916   D->addAttr(::new (S.Context) OpenCLAccessAttr(
5917       AL.getRange(), S.Context, AL.getAttributeSpellingListIndex()));
5918 }
5919 
5920 static void handleDestroyAttr(Sema &S, Decl *D, const ParsedAttr &A) {
5921   if (!cast<VarDecl>(D)->hasGlobalStorage()) {
5922     S.Diag(D->getLocation(), diag::err_destroy_attr_on_non_static_var)
5923         << (A.getKind() == ParsedAttr::AT_AlwaysDestroy);
5924     return;
5925   }
5926 
5927   if (A.getKind() == ParsedAttr::AT_AlwaysDestroy)
5928     handleSimpleAttributeWithExclusions<AlwaysDestroyAttr, NoDestroyAttr>(S, D, A);
5929   else
5930     handleSimpleAttributeWithExclusions<NoDestroyAttr, AlwaysDestroyAttr>(S, D, A);
5931 }
5932 
5933 //===----------------------------------------------------------------------===//
5934 // Top Level Sema Entry Points
5935 //===----------------------------------------------------------------------===//
5936 
5937 /// ProcessDeclAttribute - Apply the specific attribute to the specified decl if
5938 /// the attribute applies to decls.  If the attribute is a type attribute, just
5939 /// silently ignore it if a GNU attribute.
5940 static void ProcessDeclAttribute(Sema &S, Scope *scope, Decl *D,
5941                                  const ParsedAttr &AL,
5942                                  bool IncludeCXX11Attributes) {
5943   if (AL.isInvalid() || AL.getKind() == ParsedAttr::IgnoredAttribute)
5944     return;
5945 
5946   // Ignore C++11 attributes on declarator chunks: they appertain to the type
5947   // instead.
5948   if (AL.isCXX11Attribute() && !IncludeCXX11Attributes)
5949     return;
5950 
5951   // Unknown attributes are automatically warned on. Target-specific attributes
5952   // which do not apply to the current target architecture are treated as
5953   // though they were unknown attributes.
5954   if (AL.getKind() == ParsedAttr::UnknownAttribute ||
5955       !AL.existsInTarget(S.Context.getTargetInfo())) {
5956     S.Diag(AL.getLoc(), AL.isDeclspecAttribute()
5957                             ? diag::warn_unhandled_ms_attribute_ignored
5958                             : diag::warn_unknown_attribute_ignored)
5959         << AL;
5960     return;
5961   }
5962 
5963   if (handleCommonAttributeFeatures(S, D, AL))
5964     return;
5965 
5966   switch (AL.getKind()) {
5967   default:
5968     if (!AL.isStmtAttr()) {
5969       // Type attributes are handled elsewhere; silently move on.
5970       assert(AL.isTypeAttr() && "Non-type attribute not handled");
5971       break;
5972     }
5973     S.Diag(AL.getLoc(), diag::err_stmt_attribute_invalid_on_decl)
5974         << AL << D->getLocation();
5975     break;
5976   case ParsedAttr::AT_Interrupt:
5977     handleInterruptAttr(S, D, AL);
5978     break;
5979   case ParsedAttr::AT_X86ForceAlignArgPointer:
5980     handleX86ForceAlignArgPointerAttr(S, D, AL);
5981     break;
5982   case ParsedAttr::AT_DLLExport:
5983   case ParsedAttr::AT_DLLImport:
5984     handleDLLAttr(S, D, AL);
5985     break;
5986   case ParsedAttr::AT_Mips16:
5987     handleSimpleAttributeWithExclusions<Mips16Attr, MicroMipsAttr,
5988                                         MipsInterruptAttr>(S, D, AL);
5989     break;
5990   case ParsedAttr::AT_NoMips16:
5991     handleSimpleAttribute<NoMips16Attr>(S, D, AL);
5992     break;
5993   case ParsedAttr::AT_MicroMips:
5994     handleSimpleAttributeWithExclusions<MicroMipsAttr, Mips16Attr>(S, D, AL);
5995     break;
5996   case ParsedAttr::AT_NoMicroMips:
5997     handleSimpleAttribute<NoMicroMipsAttr>(S, D, AL);
5998     break;
5999   case ParsedAttr::AT_MipsLongCall:
6000     handleSimpleAttributeWithExclusions<MipsLongCallAttr, MipsShortCallAttr>(
6001         S, D, AL);
6002     break;
6003   case ParsedAttr::AT_MipsShortCall:
6004     handleSimpleAttributeWithExclusions<MipsShortCallAttr, MipsLongCallAttr>(
6005         S, D, AL);
6006     break;
6007   case ParsedAttr::AT_AMDGPUFlatWorkGroupSize:
6008     handleAMDGPUFlatWorkGroupSizeAttr(S, D, AL);
6009     break;
6010   case ParsedAttr::AT_AMDGPUWavesPerEU:
6011     handleAMDGPUWavesPerEUAttr(S, D, AL);
6012     break;
6013   case ParsedAttr::AT_AMDGPUNumSGPR:
6014     handleAMDGPUNumSGPRAttr(S, D, AL);
6015     break;
6016   case ParsedAttr::AT_AMDGPUNumVGPR:
6017     handleAMDGPUNumVGPRAttr(S, D, AL);
6018     break;
6019   case ParsedAttr::AT_AVRSignal:
6020     handleAVRSignalAttr(S, D, AL);
6021     break;
6022   case ParsedAttr::AT_IBAction:
6023     handleSimpleAttribute<IBActionAttr>(S, D, AL);
6024     break;
6025   case ParsedAttr::AT_IBOutlet:
6026     handleIBOutlet(S, D, AL);
6027     break;
6028   case ParsedAttr::AT_IBOutletCollection:
6029     handleIBOutletCollection(S, D, AL);
6030     break;
6031   case ParsedAttr::AT_IFunc:
6032     handleIFuncAttr(S, D, AL);
6033     break;
6034   case ParsedAttr::AT_Alias:
6035     handleAliasAttr(S, D, AL);
6036     break;
6037   case ParsedAttr::AT_Aligned:
6038     handleAlignedAttr(S, D, AL);
6039     break;
6040   case ParsedAttr::AT_AlignValue:
6041     handleAlignValueAttr(S, D, AL);
6042     break;
6043   case ParsedAttr::AT_AllocSize:
6044     handleAllocSizeAttr(S, D, AL);
6045     break;
6046   case ParsedAttr::AT_AlwaysInline:
6047     handleAlwaysInlineAttr(S, D, AL);
6048     break;
6049   case ParsedAttr::AT_Artificial:
6050     handleSimpleAttribute<ArtificialAttr>(S, D, AL);
6051     break;
6052   case ParsedAttr::AT_AnalyzerNoReturn:
6053     handleAnalyzerNoReturnAttr(S, D, AL);
6054     break;
6055   case ParsedAttr::AT_TLSModel:
6056     handleTLSModelAttr(S, D, AL);
6057     break;
6058   case ParsedAttr::AT_Annotate:
6059     handleAnnotateAttr(S, D, AL);
6060     break;
6061   case ParsedAttr::AT_Availability:
6062     handleAvailabilityAttr(S, D, AL);
6063     break;
6064   case ParsedAttr::AT_CarriesDependency:
6065     handleDependencyAttr(S, scope, D, AL);
6066     break;
6067   case ParsedAttr::AT_CPUDispatch:
6068   case ParsedAttr::AT_CPUSpecific:
6069     handleCPUSpecificAttr(S, D, AL);
6070     break;
6071   case ParsedAttr::AT_Common:
6072     handleCommonAttr(S, D, AL);
6073     break;
6074   case ParsedAttr::AT_CUDAConstant:
6075     handleConstantAttr(S, D, AL);
6076     break;
6077   case ParsedAttr::AT_PassObjectSize:
6078     handlePassObjectSizeAttr(S, D, AL);
6079     break;
6080   case ParsedAttr::AT_Constructor:
6081     handleConstructorAttr(S, D, AL);
6082     break;
6083   case ParsedAttr::AT_CXX11NoReturn:
6084     handleSimpleAttribute<CXX11NoReturnAttr>(S, D, AL);
6085     break;
6086   case ParsedAttr::AT_Deprecated:
6087     handleDeprecatedAttr(S, D, AL);
6088     break;
6089   case ParsedAttr::AT_Destructor:
6090     handleDestructorAttr(S, D, AL);
6091     break;
6092   case ParsedAttr::AT_EnableIf:
6093     handleEnableIfAttr(S, D, AL);
6094     break;
6095   case ParsedAttr::AT_DiagnoseIf:
6096     handleDiagnoseIfAttr(S, D, AL);
6097     break;
6098   case ParsedAttr::AT_ExtVectorType:
6099     handleExtVectorTypeAttr(S, D, AL);
6100     break;
6101   case ParsedAttr::AT_ExternalSourceSymbol:
6102     handleExternalSourceSymbolAttr(S, D, AL);
6103     break;
6104   case ParsedAttr::AT_MinSize:
6105     handleMinSizeAttr(S, D, AL);
6106     break;
6107   case ParsedAttr::AT_OptimizeNone:
6108     handleOptimizeNoneAttr(S, D, AL);
6109     break;
6110   case ParsedAttr::AT_FlagEnum:
6111     handleSimpleAttribute<FlagEnumAttr>(S, D, AL);
6112     break;
6113   case ParsedAttr::AT_EnumExtensibility:
6114     handleEnumExtensibilityAttr(S, D, AL);
6115     break;
6116   case ParsedAttr::AT_Flatten:
6117     handleSimpleAttribute<FlattenAttr>(S, D, AL);
6118     break;
6119   case ParsedAttr::AT_Format:
6120     handleFormatAttr(S, D, AL);
6121     break;
6122   case ParsedAttr::AT_FormatArg:
6123     handleFormatArgAttr(S, D, AL);
6124     break;
6125   case ParsedAttr::AT_CUDAGlobal:
6126     handleGlobalAttr(S, D, AL);
6127     break;
6128   case ParsedAttr::AT_CUDADevice:
6129     handleSimpleAttributeWithExclusions<CUDADeviceAttr, CUDAGlobalAttr>(S, D,
6130                                                                         AL);
6131     break;
6132   case ParsedAttr::AT_CUDAHost:
6133     handleSimpleAttributeWithExclusions<CUDAHostAttr, CUDAGlobalAttr>(S, D, AL);
6134     break;
6135   case ParsedAttr::AT_GNUInline:
6136     handleGNUInlineAttr(S, D, AL);
6137     break;
6138   case ParsedAttr::AT_CUDALaunchBounds:
6139     handleLaunchBoundsAttr(S, D, AL);
6140     break;
6141   case ParsedAttr::AT_Restrict:
6142     handleRestrictAttr(S, D, AL);
6143     break;
6144   case ParsedAttr::AT_LifetimeBound:
6145     handleSimpleAttribute<LifetimeBoundAttr>(S, D, AL);
6146     break;
6147   case ParsedAttr::AT_MayAlias:
6148     handleSimpleAttribute<MayAliasAttr>(S, D, AL);
6149     break;
6150   case ParsedAttr::AT_Mode:
6151     handleModeAttr(S, D, AL);
6152     break;
6153   case ParsedAttr::AT_NoAlias:
6154     handleSimpleAttribute<NoAliasAttr>(S, D, AL);
6155     break;
6156   case ParsedAttr::AT_NoCommon:
6157     handleSimpleAttribute<NoCommonAttr>(S, D, AL);
6158     break;
6159   case ParsedAttr::AT_NoSplitStack:
6160     handleSimpleAttribute<NoSplitStackAttr>(S, D, AL);
6161     break;
6162   case ParsedAttr::AT_NonNull:
6163     if (auto *PVD = dyn_cast<ParmVarDecl>(D))
6164       handleNonNullAttrParameter(S, PVD, AL);
6165     else
6166       handleNonNullAttr(S, D, AL);
6167     break;
6168   case ParsedAttr::AT_ReturnsNonNull:
6169     handleReturnsNonNullAttr(S, D, AL);
6170     break;
6171   case ParsedAttr::AT_NoEscape:
6172     handleNoEscapeAttr(S, D, AL);
6173     break;
6174   case ParsedAttr::AT_AssumeAligned:
6175     handleAssumeAlignedAttr(S, D, AL);
6176     break;
6177   case ParsedAttr::AT_AllocAlign:
6178     handleAllocAlignAttr(S, D, AL);
6179     break;
6180   case ParsedAttr::AT_Overloadable:
6181     handleSimpleAttribute<OverloadableAttr>(S, D, AL);
6182     break;
6183   case ParsedAttr::AT_Ownership:
6184     handleOwnershipAttr(S, D, AL);
6185     break;
6186   case ParsedAttr::AT_Cold:
6187     handleSimpleAttributeWithExclusions<ColdAttr, HotAttr>(S, D, AL);
6188     break;
6189   case ParsedAttr::AT_Hot:
6190     handleSimpleAttributeWithExclusions<HotAttr, ColdAttr>(S, D, AL);
6191     break;
6192   case ParsedAttr::AT_Naked:
6193     handleNakedAttr(S, D, AL);
6194     break;
6195   case ParsedAttr::AT_NoReturn:
6196     handleNoReturnAttr(S, D, AL);
6197     break;
6198   case ParsedAttr::AT_AnyX86NoCfCheck:
6199     handleNoCfCheckAttr(S, D, AL);
6200     break;
6201   case ParsedAttr::AT_NoThrow:
6202     handleSimpleAttribute<NoThrowAttr>(S, D, AL);
6203     break;
6204   case ParsedAttr::AT_CUDAShared:
6205     handleSharedAttr(S, D, AL);
6206     break;
6207   case ParsedAttr::AT_VecReturn:
6208     handleVecReturnAttr(S, D, AL);
6209     break;
6210   case ParsedAttr::AT_ObjCOwnership:
6211     handleObjCOwnershipAttr(S, D, AL);
6212     break;
6213   case ParsedAttr::AT_ObjCPreciseLifetime:
6214     handleObjCPreciseLifetimeAttr(S, D, AL);
6215     break;
6216   case ParsedAttr::AT_ObjCReturnsInnerPointer:
6217     handleObjCReturnsInnerPointerAttr(S, D, AL);
6218     break;
6219   case ParsedAttr::AT_ObjCRequiresSuper:
6220     handleObjCRequiresSuperAttr(S, D, AL);
6221     break;
6222   case ParsedAttr::AT_ObjCBridge:
6223     handleObjCBridgeAttr(S, D, AL);
6224     break;
6225   case ParsedAttr::AT_ObjCBridgeMutable:
6226     handleObjCBridgeMutableAttr(S, D, AL);
6227     break;
6228   case ParsedAttr::AT_ObjCBridgeRelated:
6229     handleObjCBridgeRelatedAttr(S, D, AL);
6230     break;
6231   case ParsedAttr::AT_ObjCDesignatedInitializer:
6232     handleObjCDesignatedInitializer(S, D, AL);
6233     break;
6234   case ParsedAttr::AT_ObjCRuntimeName:
6235     handleObjCRuntimeName(S, D, AL);
6236     break;
6237   case ParsedAttr::AT_ObjCRuntimeVisible:
6238     handleSimpleAttribute<ObjCRuntimeVisibleAttr>(S, D, AL);
6239     break;
6240   case ParsedAttr::AT_ObjCBoxable:
6241     handleObjCBoxable(S, D, AL);
6242     break;
6243   case ParsedAttr::AT_CFAuditedTransfer:
6244     handleSimpleAttributeWithExclusions<CFAuditedTransferAttr,
6245                                         CFUnknownTransferAttr>(S, D, AL);
6246     break;
6247   case ParsedAttr::AT_CFUnknownTransfer:
6248     handleSimpleAttributeWithExclusions<CFUnknownTransferAttr,
6249                                         CFAuditedTransferAttr>(S, D, AL);
6250     break;
6251   case ParsedAttr::AT_CFConsumed:
6252   case ParsedAttr::AT_NSConsumed:
6253     handleNSConsumedAttr(S, D, AL);
6254     break;
6255   case ParsedAttr::AT_NSConsumesSelf:
6256     handleSimpleAttribute<NSConsumesSelfAttr>(S, D, AL);
6257     break;
6258   case ParsedAttr::AT_NSReturnsAutoreleased:
6259   case ParsedAttr::AT_NSReturnsNotRetained:
6260   case ParsedAttr::AT_CFReturnsNotRetained:
6261   case ParsedAttr::AT_NSReturnsRetained:
6262   case ParsedAttr::AT_CFReturnsRetained:
6263     handleNSReturnsRetainedAttr(S, D, AL);
6264     break;
6265   case ParsedAttr::AT_WorkGroupSizeHint:
6266     handleWorkGroupSize<WorkGroupSizeHintAttr>(S, D, AL);
6267     break;
6268   case ParsedAttr::AT_ReqdWorkGroupSize:
6269     handleWorkGroupSize<ReqdWorkGroupSizeAttr>(S, D, AL);
6270     break;
6271   case ParsedAttr::AT_OpenCLIntelReqdSubGroupSize:
6272     handleSubGroupSize(S, D, AL);
6273     break;
6274   case ParsedAttr::AT_VecTypeHint:
6275     handleVecTypeHint(S, D, AL);
6276     break;
6277   case ParsedAttr::AT_RequireConstantInit:
6278     handleSimpleAttribute<RequireConstantInitAttr>(S, D, AL);
6279     break;
6280   case ParsedAttr::AT_InitPriority:
6281     handleInitPriorityAttr(S, D, AL);
6282     break;
6283   case ParsedAttr::AT_Packed:
6284     handlePackedAttr(S, D, AL);
6285     break;
6286   case ParsedAttr::AT_Section:
6287     handleSectionAttr(S, D, AL);
6288     break;
6289   case ParsedAttr::AT_CodeSeg:
6290     handleCodeSegAttr(S, D, AL);
6291     break;
6292   case ParsedAttr::AT_Target:
6293     handleTargetAttr(S, D, AL);
6294     break;
6295   case ParsedAttr::AT_MinVectorWidth:
6296     handleMinVectorWidthAttr(S, D, AL);
6297     break;
6298   case ParsedAttr::AT_Unavailable:
6299     handleAttrWithMessage<UnavailableAttr>(S, D, AL);
6300     break;
6301   case ParsedAttr::AT_ArcWeakrefUnavailable:
6302     handleSimpleAttribute<ArcWeakrefUnavailableAttr>(S, D, AL);
6303     break;
6304   case ParsedAttr::AT_ObjCRootClass:
6305     handleSimpleAttribute<ObjCRootClassAttr>(S, D, AL);
6306     break;
6307   case ParsedAttr::AT_ObjCSubclassingRestricted:
6308     handleSimpleAttribute<ObjCSubclassingRestrictedAttr>(S, D, AL);
6309     break;
6310   case ParsedAttr::AT_ObjCExplicitProtocolImpl:
6311     handleObjCSuppresProtocolAttr(S, D, AL);
6312     break;
6313   case ParsedAttr::AT_ObjCRequiresPropertyDefs:
6314     handleSimpleAttribute<ObjCRequiresPropertyDefsAttr>(S, D, AL);
6315     break;
6316   case ParsedAttr::AT_Unused:
6317     handleUnusedAttr(S, D, AL);
6318     break;
6319   case ParsedAttr::AT_ReturnsTwice:
6320     handleSimpleAttribute<ReturnsTwiceAttr>(S, D, AL);
6321     break;
6322   case ParsedAttr::AT_NotTailCalled:
6323     handleSimpleAttributeWithExclusions<NotTailCalledAttr, AlwaysInlineAttr>(
6324         S, D, AL);
6325     break;
6326   case ParsedAttr::AT_DisableTailCalls:
6327     handleSimpleAttributeWithExclusions<DisableTailCallsAttr, NakedAttr>(S, D,
6328                                                                          AL);
6329     break;
6330   case ParsedAttr::AT_Used:
6331     handleSimpleAttribute<UsedAttr>(S, D, AL);
6332     break;
6333   case ParsedAttr::AT_Visibility:
6334     handleVisibilityAttr(S, D, AL, false);
6335     break;
6336   case ParsedAttr::AT_TypeVisibility:
6337     handleVisibilityAttr(S, D, AL, true);
6338     break;
6339   case ParsedAttr::AT_WarnUnused:
6340     handleSimpleAttribute<WarnUnusedAttr>(S, D, AL);
6341     break;
6342   case ParsedAttr::AT_WarnUnusedResult:
6343     handleWarnUnusedResult(S, D, AL);
6344     break;
6345   case ParsedAttr::AT_Weak:
6346     handleSimpleAttribute<WeakAttr>(S, D, AL);
6347     break;
6348   case ParsedAttr::AT_WeakRef:
6349     handleWeakRefAttr(S, D, AL);
6350     break;
6351   case ParsedAttr::AT_WeakImport:
6352     handleWeakImportAttr(S, D, AL);
6353     break;
6354   case ParsedAttr::AT_TransparentUnion:
6355     handleTransparentUnionAttr(S, D, AL);
6356     break;
6357   case ParsedAttr::AT_ObjCException:
6358     handleSimpleAttribute<ObjCExceptionAttr>(S, D, AL);
6359     break;
6360   case ParsedAttr::AT_ObjCMethodFamily:
6361     handleObjCMethodFamilyAttr(S, D, AL);
6362     break;
6363   case ParsedAttr::AT_ObjCNSObject:
6364     handleObjCNSObject(S, D, AL);
6365     break;
6366   case ParsedAttr::AT_ObjCIndependentClass:
6367     handleObjCIndependentClass(S, D, AL);
6368     break;
6369   case ParsedAttr::AT_Blocks:
6370     handleBlocksAttr(S, D, AL);
6371     break;
6372   case ParsedAttr::AT_Sentinel:
6373     handleSentinelAttr(S, D, AL);
6374     break;
6375   case ParsedAttr::AT_Const:
6376     handleSimpleAttribute<ConstAttr>(S, D, AL);
6377     break;
6378   case ParsedAttr::AT_Pure:
6379     handleSimpleAttribute<PureAttr>(S, D, AL);
6380     break;
6381   case ParsedAttr::AT_Cleanup:
6382     handleCleanupAttr(S, D, AL);
6383     break;
6384   case ParsedAttr::AT_NoDebug:
6385     handleNoDebugAttr(S, D, AL);
6386     break;
6387   case ParsedAttr::AT_NoDuplicate:
6388     handleSimpleAttribute<NoDuplicateAttr>(S, D, AL);
6389     break;
6390   case ParsedAttr::AT_Convergent:
6391     handleSimpleAttribute<ConvergentAttr>(S, D, AL);
6392     break;
6393   case ParsedAttr::AT_NoInline:
6394     handleSimpleAttribute<NoInlineAttr>(S, D, AL);
6395     break;
6396   case ParsedAttr::AT_NoInstrumentFunction: // Interacts with -pg.
6397     handleSimpleAttribute<NoInstrumentFunctionAttr>(S, D, AL);
6398     break;
6399   case ParsedAttr::AT_NoStackProtector:
6400     // Interacts with -fstack-protector options.
6401     handleSimpleAttribute<NoStackProtectorAttr>(S, D, AL);
6402     break;
6403   case ParsedAttr::AT_StdCall:
6404   case ParsedAttr::AT_CDecl:
6405   case ParsedAttr::AT_FastCall:
6406   case ParsedAttr::AT_ThisCall:
6407   case ParsedAttr::AT_Pascal:
6408   case ParsedAttr::AT_RegCall:
6409   case ParsedAttr::AT_SwiftCall:
6410   case ParsedAttr::AT_VectorCall:
6411   case ParsedAttr::AT_MSABI:
6412   case ParsedAttr::AT_SysVABI:
6413   case ParsedAttr::AT_Pcs:
6414   case ParsedAttr::AT_IntelOclBicc:
6415   case ParsedAttr::AT_PreserveMost:
6416   case ParsedAttr::AT_PreserveAll:
6417     handleCallConvAttr(S, D, AL);
6418     break;
6419   case ParsedAttr::AT_Suppress:
6420     handleSuppressAttr(S, D, AL);
6421     break;
6422   case ParsedAttr::AT_OpenCLKernel:
6423     handleSimpleAttribute<OpenCLKernelAttr>(S, D, AL);
6424     break;
6425   case ParsedAttr::AT_OpenCLAccess:
6426     handleOpenCLAccessAttr(S, D, AL);
6427     break;
6428   case ParsedAttr::AT_OpenCLNoSVM:
6429     handleOpenCLNoSVMAttr(S, D, AL);
6430     break;
6431   case ParsedAttr::AT_SwiftContext:
6432     handleParameterABIAttr(S, D, AL, ParameterABI::SwiftContext);
6433     break;
6434   case ParsedAttr::AT_SwiftErrorResult:
6435     handleParameterABIAttr(S, D, AL, ParameterABI::SwiftErrorResult);
6436     break;
6437   case ParsedAttr::AT_SwiftIndirectResult:
6438     handleParameterABIAttr(S, D, AL, ParameterABI::SwiftIndirectResult);
6439     break;
6440   case ParsedAttr::AT_InternalLinkage:
6441     handleInternalLinkageAttr(S, D, AL);
6442     break;
6443   case ParsedAttr::AT_LTOVisibilityPublic:
6444     handleSimpleAttribute<LTOVisibilityPublicAttr>(S, D, AL);
6445     break;
6446 
6447   // Microsoft attributes:
6448   case ParsedAttr::AT_EmptyBases:
6449     handleSimpleAttribute<EmptyBasesAttr>(S, D, AL);
6450     break;
6451   case ParsedAttr::AT_LayoutVersion:
6452     handleLayoutVersion(S, D, AL);
6453     break;
6454   case ParsedAttr::AT_TrivialABI:
6455     handleSimpleAttribute<TrivialABIAttr>(S, D, AL);
6456     break;
6457   case ParsedAttr::AT_MSNoVTable:
6458     handleSimpleAttribute<MSNoVTableAttr>(S, D, AL);
6459     break;
6460   case ParsedAttr::AT_MSStruct:
6461     handleSimpleAttribute<MSStructAttr>(S, D, AL);
6462     break;
6463   case ParsedAttr::AT_Uuid:
6464     handleUuidAttr(S, D, AL);
6465     break;
6466   case ParsedAttr::AT_MSInheritance:
6467     handleMSInheritanceAttr(S, D, AL);
6468     break;
6469   case ParsedAttr::AT_SelectAny:
6470     handleSimpleAttribute<SelectAnyAttr>(S, D, AL);
6471     break;
6472   case ParsedAttr::AT_Thread:
6473     handleDeclspecThreadAttr(S, D, AL);
6474     break;
6475 
6476   case ParsedAttr::AT_AbiTag:
6477     handleAbiTagAttr(S, D, AL);
6478     break;
6479 
6480   // Thread safety attributes:
6481   case ParsedAttr::AT_AssertExclusiveLock:
6482     handleAssertExclusiveLockAttr(S, D, AL);
6483     break;
6484   case ParsedAttr::AT_AssertSharedLock:
6485     handleAssertSharedLockAttr(S, D, AL);
6486     break;
6487   case ParsedAttr::AT_GuardedVar:
6488     handleSimpleAttribute<GuardedVarAttr>(S, D, AL);
6489     break;
6490   case ParsedAttr::AT_PtGuardedVar:
6491     handlePtGuardedVarAttr(S, D, AL);
6492     break;
6493   case ParsedAttr::AT_ScopedLockable:
6494     handleSimpleAttribute<ScopedLockableAttr>(S, D, AL);
6495     break;
6496   case ParsedAttr::AT_NoSanitize:
6497     handleNoSanitizeAttr(S, D, AL);
6498     break;
6499   case ParsedAttr::AT_NoSanitizeSpecific:
6500     handleNoSanitizeSpecificAttr(S, D, AL);
6501     break;
6502   case ParsedAttr::AT_NoThreadSafetyAnalysis:
6503     handleSimpleAttribute<NoThreadSafetyAnalysisAttr>(S, D, AL);
6504     break;
6505   case ParsedAttr::AT_GuardedBy:
6506     handleGuardedByAttr(S, D, AL);
6507     break;
6508   case ParsedAttr::AT_PtGuardedBy:
6509     handlePtGuardedByAttr(S, D, AL);
6510     break;
6511   case ParsedAttr::AT_ExclusiveTrylockFunction:
6512     handleExclusiveTrylockFunctionAttr(S, D, AL);
6513     break;
6514   case ParsedAttr::AT_LockReturned:
6515     handleLockReturnedAttr(S, D, AL);
6516     break;
6517   case ParsedAttr::AT_LocksExcluded:
6518     handleLocksExcludedAttr(S, D, AL);
6519     break;
6520   case ParsedAttr::AT_SharedTrylockFunction:
6521     handleSharedTrylockFunctionAttr(S, D, AL);
6522     break;
6523   case ParsedAttr::AT_AcquiredBefore:
6524     handleAcquiredBeforeAttr(S, D, AL);
6525     break;
6526   case ParsedAttr::AT_AcquiredAfter:
6527     handleAcquiredAfterAttr(S, D, AL);
6528     break;
6529 
6530   // Capability analysis attributes.
6531   case ParsedAttr::AT_Capability:
6532   case ParsedAttr::AT_Lockable:
6533     handleCapabilityAttr(S, D, AL);
6534     break;
6535   case ParsedAttr::AT_RequiresCapability:
6536     handleRequiresCapabilityAttr(S, D, AL);
6537     break;
6538 
6539   case ParsedAttr::AT_AssertCapability:
6540     handleAssertCapabilityAttr(S, D, AL);
6541     break;
6542   case ParsedAttr::AT_AcquireCapability:
6543     handleAcquireCapabilityAttr(S, D, AL);
6544     break;
6545   case ParsedAttr::AT_ReleaseCapability:
6546     handleReleaseCapabilityAttr(S, D, AL);
6547     break;
6548   case ParsedAttr::AT_TryAcquireCapability:
6549     handleTryAcquireCapabilityAttr(S, D, AL);
6550     break;
6551 
6552   // Consumed analysis attributes.
6553   case ParsedAttr::AT_Consumable:
6554     handleConsumableAttr(S, D, AL);
6555     break;
6556   case ParsedAttr::AT_ConsumableAutoCast:
6557     handleSimpleAttribute<ConsumableAutoCastAttr>(S, D, AL);
6558     break;
6559   case ParsedAttr::AT_ConsumableSetOnRead:
6560     handleSimpleAttribute<ConsumableSetOnReadAttr>(S, D, AL);
6561     break;
6562   case ParsedAttr::AT_CallableWhen:
6563     handleCallableWhenAttr(S, D, AL);
6564     break;
6565   case ParsedAttr::AT_ParamTypestate:
6566     handleParamTypestateAttr(S, D, AL);
6567     break;
6568   case ParsedAttr::AT_ReturnTypestate:
6569     handleReturnTypestateAttr(S, D, AL);
6570     break;
6571   case ParsedAttr::AT_SetTypestate:
6572     handleSetTypestateAttr(S, D, AL);
6573     break;
6574   case ParsedAttr::AT_TestTypestate:
6575     handleTestTypestateAttr(S, D, AL);
6576     break;
6577 
6578   // Type safety attributes.
6579   case ParsedAttr::AT_ArgumentWithTypeTag:
6580     handleArgumentWithTypeTagAttr(S, D, AL);
6581     break;
6582   case ParsedAttr::AT_TypeTagForDatatype:
6583     handleTypeTagForDatatypeAttr(S, D, AL);
6584     break;
6585   case ParsedAttr::AT_AnyX86NoCallerSavedRegisters:
6586     handleSimpleAttribute<AnyX86NoCallerSavedRegistersAttr>(S, D, AL);
6587     break;
6588   case ParsedAttr::AT_RenderScriptKernel:
6589     handleSimpleAttribute<RenderScriptKernelAttr>(S, D, AL);
6590     break;
6591   // XRay attributes.
6592   case ParsedAttr::AT_XRayInstrument:
6593     handleSimpleAttribute<XRayInstrumentAttr>(S, D, AL);
6594     break;
6595   case ParsedAttr::AT_XRayLogArgs:
6596     handleXRayLogArgsAttr(S, D, AL);
6597     break;
6598 
6599   // Move semantics attribute.
6600   case ParsedAttr::AT_Reinitializes:
6601     handleSimpleAttribute<ReinitializesAttr>(S, D, AL);
6602     break;
6603 
6604   case ParsedAttr::AT_AlwaysDestroy:
6605   case ParsedAttr::AT_NoDestroy:
6606     handleDestroyAttr(S, D, AL);
6607     break;
6608   }
6609 }
6610 
6611 /// ProcessDeclAttributeList - Apply all the decl attributes in the specified
6612 /// attribute list to the specified decl, ignoring any type attributes.
6613 void Sema::ProcessDeclAttributeList(Scope *S, Decl *D,
6614                                     const ParsedAttributesView &AttrList,
6615                                     bool IncludeCXX11Attributes) {
6616   if (AttrList.empty())
6617     return;
6618 
6619   for (const ParsedAttr &AL : AttrList)
6620     ProcessDeclAttribute(*this, S, D, AL, IncludeCXX11Attributes);
6621 
6622   // FIXME: We should be able to handle these cases in TableGen.
6623   // GCC accepts
6624   // static int a9 __attribute__((weakref));
6625   // but that looks really pointless. We reject it.
6626   if (D->hasAttr<WeakRefAttr>() && !D->hasAttr<AliasAttr>()) {
6627     Diag(AttrList.begin()->getLoc(), diag::err_attribute_weakref_without_alias)
6628         << cast<NamedDecl>(D);
6629     D->dropAttr<WeakRefAttr>();
6630     return;
6631   }
6632 
6633   // FIXME: We should be able to handle this in TableGen as well. It would be
6634   // good to have a way to specify "these attributes must appear as a group",
6635   // for these. Additionally, it would be good to have a way to specify "these
6636   // attribute must never appear as a group" for attributes like cold and hot.
6637   if (!D->hasAttr<OpenCLKernelAttr>()) {
6638     // These attributes cannot be applied to a non-kernel function.
6639     if (const auto *A = D->getAttr<ReqdWorkGroupSizeAttr>()) {
6640       // FIXME: This emits a different error message than
6641       // diag::err_attribute_wrong_decl_type + ExpectedKernelFunction.
6642       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6643       D->setInvalidDecl();
6644     } else if (const auto *A = D->getAttr<WorkGroupSizeHintAttr>()) {
6645       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6646       D->setInvalidDecl();
6647     } else if (const auto *A = D->getAttr<VecTypeHintAttr>()) {
6648       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6649       D->setInvalidDecl();
6650     } else if (const auto *A = D->getAttr<OpenCLIntelReqdSubGroupSizeAttr>()) {
6651       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6652       D->setInvalidDecl();
6653     } else if (!D->hasAttr<CUDAGlobalAttr>()) {
6654       if (const auto *A = D->getAttr<AMDGPUFlatWorkGroupSizeAttr>()) {
6655         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6656             << A << ExpectedKernelFunction;
6657         D->setInvalidDecl();
6658       } else if (const auto *A = D->getAttr<AMDGPUWavesPerEUAttr>()) {
6659         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6660             << A << ExpectedKernelFunction;
6661         D->setInvalidDecl();
6662       } else if (const auto *A = D->getAttr<AMDGPUNumSGPRAttr>()) {
6663         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6664             << A << ExpectedKernelFunction;
6665         D->setInvalidDecl();
6666       } else if (const auto *A = D->getAttr<AMDGPUNumVGPRAttr>()) {
6667         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6668             << A << ExpectedKernelFunction;
6669         D->setInvalidDecl();
6670       }
6671     }
6672   }
6673 }
6674 
6675 // Helper for delayed processing TransparentUnion attribute.
6676 void Sema::ProcessDeclAttributeDelayed(Decl *D,
6677                                        const ParsedAttributesView &AttrList) {
6678   for (const ParsedAttr &AL : AttrList)
6679     if (AL.getKind() == ParsedAttr::AT_TransparentUnion) {
6680       handleTransparentUnionAttr(*this, D, AL);
6681       break;
6682     }
6683 }
6684 
6685 // Annotation attributes are the only attributes allowed after an access
6686 // specifier.
6687 bool Sema::ProcessAccessDeclAttributeList(
6688     AccessSpecDecl *ASDecl, const ParsedAttributesView &AttrList) {
6689   for (const ParsedAttr &AL : AttrList) {
6690     if (AL.getKind() == ParsedAttr::AT_Annotate) {
6691       ProcessDeclAttribute(*this, nullptr, ASDecl, AL, AL.isCXX11Attribute());
6692     } else {
6693       Diag(AL.getLoc(), diag::err_only_annotate_after_access_spec);
6694       return true;
6695     }
6696   }
6697   return false;
6698 }
6699 
6700 /// checkUnusedDeclAttributes - Check a list of attributes to see if it
6701 /// contains any decl attributes that we should warn about.
6702 static void checkUnusedDeclAttributes(Sema &S, const ParsedAttributesView &A) {
6703   for (const ParsedAttr &AL : A) {
6704     // Only warn if the attribute is an unignored, non-type attribute.
6705     if (AL.isUsedAsTypeAttr() || AL.isInvalid())
6706       continue;
6707     if (AL.getKind() == ParsedAttr::IgnoredAttribute)
6708       continue;
6709 
6710     if (AL.getKind() == ParsedAttr::UnknownAttribute) {
6711       S.Diag(AL.getLoc(), diag::warn_unknown_attribute_ignored)
6712           << AL << AL.getRange();
6713     } else {
6714       S.Diag(AL.getLoc(), diag::warn_attribute_not_on_decl) << AL
6715                                                             << AL.getRange();
6716     }
6717   }
6718 }
6719 
6720 /// checkUnusedDeclAttributes - Given a declarator which is not being
6721 /// used to build a declaration, complain about any decl attributes
6722 /// which might be lying around on it.
6723 void Sema::checkUnusedDeclAttributes(Declarator &D) {
6724   ::checkUnusedDeclAttributes(*this, D.getDeclSpec().getAttributes());
6725   ::checkUnusedDeclAttributes(*this, D.getAttributes());
6726   for (unsigned i = 0, e = D.getNumTypeObjects(); i != e; ++i)
6727     ::checkUnusedDeclAttributes(*this, D.getTypeObject(i).getAttrs());
6728 }
6729 
6730 /// DeclClonePragmaWeak - clone existing decl (maybe definition),
6731 /// \#pragma weak needs a non-definition decl and source may not have one.
6732 NamedDecl * Sema::DeclClonePragmaWeak(NamedDecl *ND, IdentifierInfo *II,
6733                                       SourceLocation Loc) {
6734   assert(isa<FunctionDecl>(ND) || isa<VarDecl>(ND));
6735   NamedDecl *NewD = nullptr;
6736   if (auto *FD = dyn_cast<FunctionDecl>(ND)) {
6737     FunctionDecl *NewFD;
6738     // FIXME: Missing call to CheckFunctionDeclaration().
6739     // FIXME: Mangling?
6740     // FIXME: Is the qualifier info correct?
6741     // FIXME: Is the DeclContext correct?
6742     NewFD = FunctionDecl::Create(FD->getASTContext(), FD->getDeclContext(),
6743                                  Loc, Loc, DeclarationName(II),
6744                                  FD->getType(), FD->getTypeSourceInfo(),
6745                                  SC_None, false/*isInlineSpecified*/,
6746                                  FD->hasPrototype(),
6747                                  false/*isConstexprSpecified*/);
6748     NewD = NewFD;
6749 
6750     if (FD->getQualifier())
6751       NewFD->setQualifierInfo(FD->getQualifierLoc());
6752 
6753     // Fake up parameter variables; they are declared as if this were
6754     // a typedef.
6755     QualType FDTy = FD->getType();
6756     if (const auto *FT = FDTy->getAs<FunctionProtoType>()) {
6757       SmallVector<ParmVarDecl*, 16> Params;
6758       for (const auto &AI : FT->param_types()) {
6759         ParmVarDecl *Param = BuildParmVarDeclForTypedef(NewFD, Loc, AI);
6760         Param->setScopeInfo(0, Params.size());
6761         Params.push_back(Param);
6762       }
6763       NewFD->setParams(Params);
6764     }
6765   } else if (auto *VD = dyn_cast<VarDecl>(ND)) {
6766     NewD = VarDecl::Create(VD->getASTContext(), VD->getDeclContext(),
6767                            VD->getInnerLocStart(), VD->getLocation(), II,
6768                            VD->getType(), VD->getTypeSourceInfo(),
6769                            VD->getStorageClass());
6770     if (VD->getQualifier())
6771       cast<VarDecl>(NewD)->setQualifierInfo(VD->getQualifierLoc());
6772   }
6773   return NewD;
6774 }
6775 
6776 /// DeclApplyPragmaWeak - A declaration (maybe definition) needs \#pragma weak
6777 /// applied to it, possibly with an alias.
6778 void Sema::DeclApplyPragmaWeak(Scope *S, NamedDecl *ND, WeakInfo &W) {
6779   if (W.getUsed()) return; // only do this once
6780   W.setUsed(true);
6781   if (W.getAlias()) { // clone decl, impersonate __attribute(weak,alias(...))
6782     IdentifierInfo *NDId = ND->getIdentifier();
6783     NamedDecl *NewD = DeclClonePragmaWeak(ND, W.getAlias(), W.getLocation());
6784     NewD->addAttr(AliasAttr::CreateImplicit(Context, NDId->getName(),
6785                                             W.getLocation()));
6786     NewD->addAttr(WeakAttr::CreateImplicit(Context, W.getLocation()));
6787     WeakTopLevelDecl.push_back(NewD);
6788     // FIXME: "hideous" code from Sema::LazilyCreateBuiltin
6789     // to insert Decl at TU scope, sorry.
6790     DeclContext *SavedContext = CurContext;
6791     CurContext = Context.getTranslationUnitDecl();
6792     NewD->setDeclContext(CurContext);
6793     NewD->setLexicalDeclContext(CurContext);
6794     PushOnScopeChains(NewD, S);
6795     CurContext = SavedContext;
6796   } else { // just add weak to existing
6797     ND->addAttr(WeakAttr::CreateImplicit(Context, W.getLocation()));
6798   }
6799 }
6800 
6801 void Sema::ProcessPragmaWeak(Scope *S, Decl *D) {
6802   // It's valid to "forward-declare" #pragma weak, in which case we
6803   // have to do this.
6804   LoadExternalWeakUndeclaredIdentifiers();
6805   if (!WeakUndeclaredIdentifiers.empty()) {
6806     NamedDecl *ND = nullptr;
6807     if (auto *VD = dyn_cast<VarDecl>(D))
6808       if (VD->isExternC())
6809         ND = VD;
6810     if (auto *FD = dyn_cast<FunctionDecl>(D))
6811       if (FD->isExternC())
6812         ND = FD;
6813     if (ND) {
6814       if (IdentifierInfo *Id = ND->getIdentifier()) {
6815         auto I = WeakUndeclaredIdentifiers.find(Id);
6816         if (I != WeakUndeclaredIdentifiers.end()) {
6817           WeakInfo W = I->second;
6818           DeclApplyPragmaWeak(S, ND, W);
6819           WeakUndeclaredIdentifiers[Id] = W;
6820         }
6821       }
6822     }
6823   }
6824 }
6825 
6826 /// ProcessDeclAttributes - Given a declarator (PD) with attributes indicated in
6827 /// it, apply them to D.  This is a bit tricky because PD can have attributes
6828 /// specified in many different places, and we need to find and apply them all.
6829 void Sema::ProcessDeclAttributes(Scope *S, Decl *D, const Declarator &PD) {
6830   // Apply decl attributes from the DeclSpec if present.
6831   if (!PD.getDeclSpec().getAttributes().empty())
6832     ProcessDeclAttributeList(S, D, PD.getDeclSpec().getAttributes());
6833 
6834   // Walk the declarator structure, applying decl attributes that were in a type
6835   // position to the decl itself.  This handles cases like:
6836   //   int *__attr__(x)** D;
6837   // when X is a decl attribute.
6838   for (unsigned i = 0, e = PD.getNumTypeObjects(); i != e; ++i)
6839     ProcessDeclAttributeList(S, D, PD.getTypeObject(i).getAttrs(),
6840                              /*IncludeCXX11Attributes=*/false);
6841 
6842   // Finally, apply any attributes on the decl itself.
6843   ProcessDeclAttributeList(S, D, PD.getAttributes());
6844 
6845   // Apply additional attributes specified by '#pragma clang attribute'.
6846   AddPragmaAttributes(S, D);
6847 }
6848 
6849 /// Is the given declaration allowed to use a forbidden type?
6850 /// If so, it'll still be annotated with an attribute that makes it
6851 /// illegal to actually use.
6852 static bool isForbiddenTypeAllowed(Sema &S, Decl *D,
6853                                    const DelayedDiagnostic &diag,
6854                                    UnavailableAttr::ImplicitReason &reason) {
6855   // Private ivars are always okay.  Unfortunately, people don't
6856   // always properly make their ivars private, even in system headers.
6857   // Plus we need to make fields okay, too.
6858   if (!isa<FieldDecl>(D) && !isa<ObjCPropertyDecl>(D) &&
6859       !isa<FunctionDecl>(D))
6860     return false;
6861 
6862   // Silently accept unsupported uses of __weak in both user and system
6863   // declarations when it's been disabled, for ease of integration with
6864   // -fno-objc-arc files.  We do have to take some care against attempts
6865   // to define such things;  for now, we've only done that for ivars
6866   // and properties.
6867   if ((isa<ObjCIvarDecl>(D) || isa<ObjCPropertyDecl>(D))) {
6868     if (diag.getForbiddenTypeDiagnostic() == diag::err_arc_weak_disabled ||
6869         diag.getForbiddenTypeDiagnostic() == diag::err_arc_weak_no_runtime) {
6870       reason = UnavailableAttr::IR_ForbiddenWeak;
6871       return true;
6872     }
6873   }
6874 
6875   // Allow all sorts of things in system headers.
6876   if (S.Context.getSourceManager().isInSystemHeader(D->getLocation())) {
6877     // Currently, all the failures dealt with this way are due to ARC
6878     // restrictions.
6879     reason = UnavailableAttr::IR_ARCForbiddenType;
6880     return true;
6881   }
6882 
6883   return false;
6884 }
6885 
6886 /// Handle a delayed forbidden-type diagnostic.
6887 static void handleDelayedForbiddenType(Sema &S, DelayedDiagnostic &DD,
6888                                        Decl *D) {
6889   auto Reason = UnavailableAttr::IR_None;
6890   if (D && isForbiddenTypeAllowed(S, D, DD, Reason)) {
6891     assert(Reason && "didn't set reason?");
6892     D->addAttr(UnavailableAttr::CreateImplicit(S.Context, "", Reason, DD.Loc));
6893     return;
6894   }
6895   if (S.getLangOpts().ObjCAutoRefCount)
6896     if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
6897       // FIXME: we may want to suppress diagnostics for all
6898       // kind of forbidden type messages on unavailable functions.
6899       if (FD->hasAttr<UnavailableAttr>() &&
6900           DD.getForbiddenTypeDiagnostic() ==
6901               diag::err_arc_array_param_no_ownership) {
6902         DD.Triggered = true;
6903         return;
6904       }
6905     }
6906 
6907   S.Diag(DD.Loc, DD.getForbiddenTypeDiagnostic())
6908       << DD.getForbiddenTypeOperand() << DD.getForbiddenTypeArgument();
6909   DD.Triggered = true;
6910 }
6911 
6912 static const AvailabilityAttr *getAttrForPlatform(ASTContext &Context,
6913                                                   const Decl *D) {
6914   // Check each AvailabilityAttr to find the one for this platform.
6915   for (const auto *A : D->attrs()) {
6916     if (const auto *Avail = dyn_cast<AvailabilityAttr>(A)) {
6917       // FIXME: this is copied from CheckAvailability. We should try to
6918       // de-duplicate.
6919 
6920       // Check if this is an App Extension "platform", and if so chop off
6921       // the suffix for matching with the actual platform.
6922       StringRef ActualPlatform = Avail->getPlatform()->getName();
6923       StringRef RealizedPlatform = ActualPlatform;
6924       if (Context.getLangOpts().AppExt) {
6925         size_t suffix = RealizedPlatform.rfind("_app_extension");
6926         if (suffix != StringRef::npos)
6927           RealizedPlatform = RealizedPlatform.slice(0, suffix);
6928       }
6929 
6930       StringRef TargetPlatform = Context.getTargetInfo().getPlatformName();
6931 
6932       // Match the platform name.
6933       if (RealizedPlatform == TargetPlatform)
6934         return Avail;
6935     }
6936   }
6937   return nullptr;
6938 }
6939 
6940 /// The diagnostic we should emit for \c D, and the declaration that
6941 /// originated it, or \c AR_Available.
6942 ///
6943 /// \param D The declaration to check.
6944 /// \param Message If non-null, this will be populated with the message from
6945 /// the availability attribute that is selected.
6946 static std::pair<AvailabilityResult, const NamedDecl *>
6947 ShouldDiagnoseAvailabilityOfDecl(const NamedDecl *D, std::string *Message) {
6948   AvailabilityResult Result = D->getAvailability(Message);
6949 
6950   // For typedefs, if the typedef declaration appears available look
6951   // to the underlying type to see if it is more restrictive.
6952   while (const auto *TD = dyn_cast<TypedefNameDecl>(D)) {
6953     if (Result == AR_Available) {
6954       if (const auto *TT = TD->getUnderlyingType()->getAs<TagType>()) {
6955         D = TT->getDecl();
6956         Result = D->getAvailability(Message);
6957         continue;
6958       }
6959     }
6960     break;
6961   }
6962 
6963   // Forward class declarations get their attributes from their definition.
6964   if (const auto *IDecl = dyn_cast<ObjCInterfaceDecl>(D)) {
6965     if (IDecl->getDefinition()) {
6966       D = IDecl->getDefinition();
6967       Result = D->getAvailability(Message);
6968     }
6969   }
6970 
6971   if (const auto *ECD = dyn_cast<EnumConstantDecl>(D))
6972     if (Result == AR_Available) {
6973       const DeclContext *DC = ECD->getDeclContext();
6974       if (const auto *TheEnumDecl = dyn_cast<EnumDecl>(DC)) {
6975         Result = TheEnumDecl->getAvailability(Message);
6976         D = TheEnumDecl;
6977       }
6978     }
6979 
6980   return {Result, D};
6981 }
6982 
6983 
6984 /// whether we should emit a diagnostic for \c K and \c DeclVersion in
6985 /// the context of \c Ctx. For example, we should emit an unavailable diagnostic
6986 /// in a deprecated context, but not the other way around.
6987 static bool ShouldDiagnoseAvailabilityInContext(Sema &S, AvailabilityResult K,
6988                                                 VersionTuple DeclVersion,
6989                                                 Decl *Ctx) {
6990   assert(K != AR_Available && "Expected an unavailable declaration here!");
6991 
6992   // Checks if we should emit the availability diagnostic in the context of C.
6993   auto CheckContext = [&](const Decl *C) {
6994     if (K == AR_NotYetIntroduced) {
6995       if (const AvailabilityAttr *AA = getAttrForPlatform(S.Context, C))
6996         if (AA->getIntroduced() >= DeclVersion)
6997           return true;
6998     } else if (K == AR_Deprecated)
6999       if (C->isDeprecated())
7000         return true;
7001 
7002     if (C->isUnavailable())
7003       return true;
7004     return false;
7005   };
7006 
7007   do {
7008     if (CheckContext(Ctx))
7009       return false;
7010 
7011     // An implementation implicitly has the availability of the interface.
7012     // Unless it is "+load" method.
7013     if (const auto *MethodD = dyn_cast<ObjCMethodDecl>(Ctx))
7014       if (MethodD->isClassMethod() &&
7015           MethodD->getSelector().getAsString() == "load")
7016         return true;
7017 
7018     if (const auto *CatOrImpl = dyn_cast<ObjCImplDecl>(Ctx)) {
7019       if (const ObjCInterfaceDecl *Interface = CatOrImpl->getClassInterface())
7020         if (CheckContext(Interface))
7021           return false;
7022     }
7023     // A category implicitly has the availability of the interface.
7024     else if (const auto *CatD = dyn_cast<ObjCCategoryDecl>(Ctx))
7025       if (const ObjCInterfaceDecl *Interface = CatD->getClassInterface())
7026         if (CheckContext(Interface))
7027           return false;
7028   } while ((Ctx = cast_or_null<Decl>(Ctx->getDeclContext())));
7029 
7030   return true;
7031 }
7032 
7033 static bool
7034 shouldDiagnoseAvailabilityByDefault(const ASTContext &Context,
7035                                     const VersionTuple &DeploymentVersion,
7036                                     const VersionTuple &DeclVersion) {
7037   const auto &Triple = Context.getTargetInfo().getTriple();
7038   VersionTuple ForceAvailabilityFromVersion;
7039   switch (Triple.getOS()) {
7040   case llvm::Triple::IOS:
7041   case llvm::Triple::TvOS:
7042     ForceAvailabilityFromVersion = VersionTuple(/*Major=*/11);
7043     break;
7044   case llvm::Triple::WatchOS:
7045     ForceAvailabilityFromVersion = VersionTuple(/*Major=*/4);
7046     break;
7047   case llvm::Triple::Darwin:
7048   case llvm::Triple::MacOSX:
7049     ForceAvailabilityFromVersion = VersionTuple(/*Major=*/10, /*Minor=*/13);
7050     break;
7051   default:
7052     // New targets should always warn about availability.
7053     return Triple.getVendor() == llvm::Triple::Apple;
7054   }
7055   return DeploymentVersion >= ForceAvailabilityFromVersion ||
7056          DeclVersion >= ForceAvailabilityFromVersion;
7057 }
7058 
7059 static NamedDecl *findEnclosingDeclToAnnotate(Decl *OrigCtx) {
7060   for (Decl *Ctx = OrigCtx; Ctx;
7061        Ctx = cast_or_null<Decl>(Ctx->getDeclContext())) {
7062     if (isa<TagDecl>(Ctx) || isa<FunctionDecl>(Ctx) || isa<ObjCMethodDecl>(Ctx))
7063       return cast<NamedDecl>(Ctx);
7064     if (auto *CD = dyn_cast<ObjCContainerDecl>(Ctx)) {
7065       if (auto *Imp = dyn_cast<ObjCImplDecl>(Ctx))
7066         return Imp->getClassInterface();
7067       return CD;
7068     }
7069   }
7070 
7071   return dyn_cast<NamedDecl>(OrigCtx);
7072 }
7073 
7074 namespace {
7075 
7076 struct AttributeInsertion {
7077   StringRef Prefix;
7078   SourceLocation Loc;
7079   StringRef Suffix;
7080 
7081   static AttributeInsertion createInsertionAfter(const NamedDecl *D) {
7082     return {" ", D->getEndLoc(), ""};
7083   }
7084   static AttributeInsertion createInsertionAfter(SourceLocation Loc) {
7085     return {" ", Loc, ""};
7086   }
7087   static AttributeInsertion createInsertionBefore(const NamedDecl *D) {
7088     return {"", D->getBeginLoc(), "\n"};
7089   }
7090 };
7091 
7092 } // end anonymous namespace
7093 
7094 /// Tries to parse a string as ObjC method name.
7095 ///
7096 /// \param Name The string to parse. Expected to originate from availability
7097 /// attribute argument.
7098 /// \param SlotNames The vector that will be populated with slot names. In case
7099 /// of unsuccessful parsing can contain invalid data.
7100 /// \returns A number of method parameters if parsing was successful, None
7101 /// otherwise.
7102 static Optional<unsigned>
7103 tryParseObjCMethodName(StringRef Name, SmallVectorImpl<StringRef> &SlotNames,
7104                        const LangOptions &LangOpts) {
7105   // Accept replacements starting with - or + as valid ObjC method names.
7106   if (!Name.empty() && (Name.front() == '-' || Name.front() == '+'))
7107     Name = Name.drop_front(1);
7108   if (Name.empty())
7109     return None;
7110   Name.split(SlotNames, ':');
7111   unsigned NumParams;
7112   if (Name.back() == ':') {
7113     // Remove an empty string at the end that doesn't represent any slot.
7114     SlotNames.pop_back();
7115     NumParams = SlotNames.size();
7116   } else {
7117     if (SlotNames.size() != 1)
7118       // Not a valid method name, just a colon-separated string.
7119       return None;
7120     NumParams = 0;
7121   }
7122   // Verify all slot names are valid.
7123   bool AllowDollar = LangOpts.DollarIdents;
7124   for (StringRef S : SlotNames) {
7125     if (S.empty())
7126       continue;
7127     if (!isValidIdentifier(S, AllowDollar))
7128       return None;
7129   }
7130   return NumParams;
7131 }
7132 
7133 /// Returns a source location in which it's appropriate to insert a new
7134 /// attribute for the given declaration \D.
7135 static Optional<AttributeInsertion>
7136 createAttributeInsertion(const NamedDecl *D, const SourceManager &SM,
7137                          const LangOptions &LangOpts) {
7138   if (isa<ObjCPropertyDecl>(D))
7139     return AttributeInsertion::createInsertionAfter(D);
7140   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D)) {
7141     if (MD->hasBody())
7142       return None;
7143     return AttributeInsertion::createInsertionAfter(D);
7144   }
7145   if (const auto *TD = dyn_cast<TagDecl>(D)) {
7146     SourceLocation Loc =
7147         Lexer::getLocForEndOfToken(TD->getInnerLocStart(), 0, SM, LangOpts);
7148     if (Loc.isInvalid())
7149       return None;
7150     // Insert after the 'struct'/whatever keyword.
7151     return AttributeInsertion::createInsertionAfter(Loc);
7152   }
7153   return AttributeInsertion::createInsertionBefore(D);
7154 }
7155 
7156 /// Actually emit an availability diagnostic for a reference to an unavailable
7157 /// decl.
7158 ///
7159 /// \param Ctx The context that the reference occurred in
7160 /// \param ReferringDecl The exact declaration that was referenced.
7161 /// \param OffendingDecl A related decl to \c ReferringDecl that has an
7162 /// availability attribute corresponding to \c K attached to it. Note that this
7163 /// may not be the same as ReferringDecl, i.e. if an EnumDecl is annotated and
7164 /// we refer to a member EnumConstantDecl, ReferringDecl is the EnumConstantDecl
7165 /// and OffendingDecl is the EnumDecl.
7166 static void DoEmitAvailabilityWarning(Sema &S, AvailabilityResult K,
7167                                       Decl *Ctx, const NamedDecl *ReferringDecl,
7168                                       const NamedDecl *OffendingDecl,
7169                                       StringRef Message,
7170                                       ArrayRef<SourceLocation> Locs,
7171                                       const ObjCInterfaceDecl *UnknownObjCClass,
7172                                       const ObjCPropertyDecl *ObjCProperty,
7173                                       bool ObjCPropertyAccess) {
7174   // Diagnostics for deprecated or unavailable.
7175   unsigned diag, diag_message, diag_fwdclass_message;
7176   unsigned diag_available_here = diag::note_availability_specified_here;
7177   SourceLocation NoteLocation = OffendingDecl->getLocation();
7178 
7179   // Matches 'diag::note_property_attribute' options.
7180   unsigned property_note_select;
7181 
7182   // Matches diag::note_availability_specified_here.
7183   unsigned available_here_select_kind;
7184 
7185   VersionTuple DeclVersion;
7186   if (const AvailabilityAttr *AA = getAttrForPlatform(S.Context, OffendingDecl))
7187     DeclVersion = AA->getIntroduced();
7188 
7189   if (!ShouldDiagnoseAvailabilityInContext(S, K, DeclVersion, Ctx))
7190     return;
7191 
7192   SourceLocation Loc = Locs.front();
7193 
7194   // The declaration can have multiple availability attributes, we are looking
7195   // at one of them.
7196   const AvailabilityAttr *A = getAttrForPlatform(S.Context, OffendingDecl);
7197   if (A && A->isInherited()) {
7198     for (const Decl *Redecl = OffendingDecl->getMostRecentDecl(); Redecl;
7199          Redecl = Redecl->getPreviousDecl()) {
7200       const AvailabilityAttr *AForRedecl =
7201           getAttrForPlatform(S.Context, Redecl);
7202       if (AForRedecl && !AForRedecl->isInherited()) {
7203         // If D is a declaration with inherited attributes, the note should
7204         // point to the declaration with actual attributes.
7205         NoteLocation = Redecl->getLocation();
7206         break;
7207       }
7208     }
7209   }
7210 
7211   switch (K) {
7212   case AR_NotYetIntroduced: {
7213     // We would like to emit the diagnostic even if -Wunguarded-availability is
7214     // not specified for deployment targets >= to iOS 11 or equivalent or
7215     // for declarations that were introduced in iOS 11 (macOS 10.13, ...) or
7216     // later.
7217     const AvailabilityAttr *AA =
7218         getAttrForPlatform(S.getASTContext(), OffendingDecl);
7219     VersionTuple Introduced = AA->getIntroduced();
7220 
7221     bool UseNewWarning = shouldDiagnoseAvailabilityByDefault(
7222         S.Context, S.Context.getTargetInfo().getPlatformMinVersion(),
7223         Introduced);
7224     unsigned Warning = UseNewWarning ? diag::warn_unguarded_availability_new
7225                                      : diag::warn_unguarded_availability;
7226 
7227     S.Diag(Loc, Warning)
7228         << OffendingDecl
7229         << AvailabilityAttr::getPrettyPlatformName(
7230                S.getASTContext().getTargetInfo().getPlatformName())
7231         << Introduced.getAsString();
7232 
7233     S.Diag(OffendingDecl->getLocation(), diag::note_availability_specified_here)
7234         << OffendingDecl << /* partial */ 3;
7235 
7236     if (const auto *Enclosing = findEnclosingDeclToAnnotate(Ctx)) {
7237       if (const auto *TD = dyn_cast<TagDecl>(Enclosing))
7238         if (TD->getDeclName().isEmpty()) {
7239           S.Diag(TD->getLocation(),
7240                  diag::note_decl_unguarded_availability_silence)
7241               << /*Anonymous*/ 1 << TD->getKindName();
7242           return;
7243         }
7244       auto FixitNoteDiag =
7245           S.Diag(Enclosing->getLocation(),
7246                  diag::note_decl_unguarded_availability_silence)
7247           << /*Named*/ 0 << Enclosing;
7248       // Don't offer a fixit for declarations with availability attributes.
7249       if (Enclosing->hasAttr<AvailabilityAttr>())
7250         return;
7251       if (!S.getPreprocessor().isMacroDefined("API_AVAILABLE"))
7252         return;
7253       Optional<AttributeInsertion> Insertion = createAttributeInsertion(
7254           Enclosing, S.getSourceManager(), S.getLangOpts());
7255       if (!Insertion)
7256         return;
7257       std::string PlatformName =
7258           AvailabilityAttr::getPlatformNameSourceSpelling(
7259               S.getASTContext().getTargetInfo().getPlatformName())
7260               .lower();
7261       std::string Introduced =
7262           OffendingDecl->getVersionIntroduced().getAsString();
7263       FixitNoteDiag << FixItHint::CreateInsertion(
7264           Insertion->Loc,
7265           (llvm::Twine(Insertion->Prefix) + "API_AVAILABLE(" + PlatformName +
7266            "(" + Introduced + "))" + Insertion->Suffix)
7267               .str());
7268     }
7269     return;
7270   }
7271   case AR_Deprecated:
7272     diag = !ObjCPropertyAccess ? diag::warn_deprecated
7273                                : diag::warn_property_method_deprecated;
7274     diag_message = diag::warn_deprecated_message;
7275     diag_fwdclass_message = diag::warn_deprecated_fwdclass_message;
7276     property_note_select = /* deprecated */ 0;
7277     available_here_select_kind = /* deprecated */ 2;
7278     if (const auto *AL = OffendingDecl->getAttr<DeprecatedAttr>())
7279       NoteLocation = AL->getLocation();
7280     break;
7281 
7282   case AR_Unavailable:
7283     diag = !ObjCPropertyAccess ? diag::err_unavailable
7284                                : diag::err_property_method_unavailable;
7285     diag_message = diag::err_unavailable_message;
7286     diag_fwdclass_message = diag::warn_unavailable_fwdclass_message;
7287     property_note_select = /* unavailable */ 1;
7288     available_here_select_kind = /* unavailable */ 0;
7289 
7290     if (auto AL = OffendingDecl->getAttr<UnavailableAttr>()) {
7291       if (AL->isImplicit() && AL->getImplicitReason()) {
7292         // Most of these failures are due to extra restrictions in ARC;
7293         // reflect that in the primary diagnostic when applicable.
7294         auto flagARCError = [&] {
7295           if (S.getLangOpts().ObjCAutoRefCount &&
7296               S.getSourceManager().isInSystemHeader(
7297                   OffendingDecl->getLocation()))
7298             diag = diag::err_unavailable_in_arc;
7299         };
7300 
7301         switch (AL->getImplicitReason()) {
7302         case UnavailableAttr::IR_None: break;
7303 
7304         case UnavailableAttr::IR_ARCForbiddenType:
7305           flagARCError();
7306           diag_available_here = diag::note_arc_forbidden_type;
7307           break;
7308 
7309         case UnavailableAttr::IR_ForbiddenWeak:
7310           if (S.getLangOpts().ObjCWeakRuntime)
7311             diag_available_here = diag::note_arc_weak_disabled;
7312           else
7313             diag_available_here = diag::note_arc_weak_no_runtime;
7314           break;
7315 
7316         case UnavailableAttr::IR_ARCForbiddenConversion:
7317           flagARCError();
7318           diag_available_here = diag::note_performs_forbidden_arc_conversion;
7319           break;
7320 
7321         case UnavailableAttr::IR_ARCInitReturnsUnrelated:
7322           flagARCError();
7323           diag_available_here = diag::note_arc_init_returns_unrelated;
7324           break;
7325 
7326         case UnavailableAttr::IR_ARCFieldWithOwnership:
7327           flagARCError();
7328           diag_available_here = diag::note_arc_field_with_ownership;
7329           break;
7330         }
7331       }
7332     }
7333     break;
7334 
7335   case AR_Available:
7336     llvm_unreachable("Warning for availability of available declaration?");
7337   }
7338 
7339   SmallVector<FixItHint, 12> FixIts;
7340   if (K == AR_Deprecated) {
7341     StringRef Replacement;
7342     if (auto AL = OffendingDecl->getAttr<DeprecatedAttr>())
7343       Replacement = AL->getReplacement();
7344     if (auto AL = getAttrForPlatform(S.Context, OffendingDecl))
7345       Replacement = AL->getReplacement();
7346 
7347     CharSourceRange UseRange;
7348     if (!Replacement.empty())
7349       UseRange =
7350           CharSourceRange::getCharRange(Loc, S.getLocForEndOfToken(Loc));
7351     if (UseRange.isValid()) {
7352       if (const auto *MethodDecl = dyn_cast<ObjCMethodDecl>(ReferringDecl)) {
7353         Selector Sel = MethodDecl->getSelector();
7354         SmallVector<StringRef, 12> SelectorSlotNames;
7355         Optional<unsigned> NumParams = tryParseObjCMethodName(
7356             Replacement, SelectorSlotNames, S.getLangOpts());
7357         if (NumParams && NumParams.getValue() == Sel.getNumArgs()) {
7358           assert(SelectorSlotNames.size() == Locs.size());
7359           for (unsigned I = 0; I < Locs.size(); ++I) {
7360             if (!Sel.getNameForSlot(I).empty()) {
7361               CharSourceRange NameRange = CharSourceRange::getCharRange(
7362                   Locs[I], S.getLocForEndOfToken(Locs[I]));
7363               FixIts.push_back(FixItHint::CreateReplacement(
7364                   NameRange, SelectorSlotNames[I]));
7365             } else
7366               FixIts.push_back(
7367                   FixItHint::CreateInsertion(Locs[I], SelectorSlotNames[I]));
7368           }
7369         } else
7370           FixIts.push_back(FixItHint::CreateReplacement(UseRange, Replacement));
7371       } else
7372         FixIts.push_back(FixItHint::CreateReplacement(UseRange, Replacement));
7373     }
7374   }
7375 
7376   if (!Message.empty()) {
7377     S.Diag(Loc, diag_message) << ReferringDecl << Message << FixIts;
7378     if (ObjCProperty)
7379       S.Diag(ObjCProperty->getLocation(), diag::note_property_attribute)
7380           << ObjCProperty->getDeclName() << property_note_select;
7381   } else if (!UnknownObjCClass) {
7382     S.Diag(Loc, diag) << ReferringDecl << FixIts;
7383     if (ObjCProperty)
7384       S.Diag(ObjCProperty->getLocation(), diag::note_property_attribute)
7385           << ObjCProperty->getDeclName() << property_note_select;
7386   } else {
7387     S.Diag(Loc, diag_fwdclass_message) << ReferringDecl << FixIts;
7388     S.Diag(UnknownObjCClass->getLocation(), diag::note_forward_class);
7389   }
7390 
7391   S.Diag(NoteLocation, diag_available_here)
7392     << OffendingDecl << available_here_select_kind;
7393 }
7394 
7395 static void handleDelayedAvailabilityCheck(Sema &S, DelayedDiagnostic &DD,
7396                                            Decl *Ctx) {
7397   assert(DD.Kind == DelayedDiagnostic::Availability &&
7398          "Expected an availability diagnostic here");
7399 
7400   DD.Triggered = true;
7401   DoEmitAvailabilityWarning(
7402       S, DD.getAvailabilityResult(), Ctx, DD.getAvailabilityReferringDecl(),
7403       DD.getAvailabilityOffendingDecl(), DD.getAvailabilityMessage(),
7404       DD.getAvailabilitySelectorLocs(), DD.getUnknownObjCClass(),
7405       DD.getObjCProperty(), false);
7406 }
7407 
7408 void Sema::PopParsingDeclaration(ParsingDeclState state, Decl *decl) {
7409   assert(DelayedDiagnostics.getCurrentPool());
7410   DelayedDiagnosticPool &poppedPool = *DelayedDiagnostics.getCurrentPool();
7411   DelayedDiagnostics.popWithoutEmitting(state);
7412 
7413   // When delaying diagnostics to run in the context of a parsed
7414   // declaration, we only want to actually emit anything if parsing
7415   // succeeds.
7416   if (!decl) return;
7417 
7418   // We emit all the active diagnostics in this pool or any of its
7419   // parents.  In general, we'll get one pool for the decl spec
7420   // and a child pool for each declarator; in a decl group like:
7421   //   deprecated_typedef foo, *bar, baz();
7422   // only the declarator pops will be passed decls.  This is correct;
7423   // we really do need to consider delayed diagnostics from the decl spec
7424   // for each of the different declarations.
7425   const DelayedDiagnosticPool *pool = &poppedPool;
7426   do {
7427     for (DelayedDiagnosticPool::pool_iterator
7428            i = pool->pool_begin(), e = pool->pool_end(); i != e; ++i) {
7429       // This const_cast is a bit lame.  Really, Triggered should be mutable.
7430       DelayedDiagnostic &diag = const_cast<DelayedDiagnostic&>(*i);
7431       if (diag.Triggered)
7432         continue;
7433 
7434       switch (diag.Kind) {
7435       case DelayedDiagnostic::Availability:
7436         // Don't bother giving deprecation/unavailable diagnostics if
7437         // the decl is invalid.
7438         if (!decl->isInvalidDecl())
7439           handleDelayedAvailabilityCheck(*this, diag, decl);
7440         break;
7441 
7442       case DelayedDiagnostic::Access:
7443         HandleDelayedAccessCheck(diag, decl);
7444         break;
7445 
7446       case DelayedDiagnostic::ForbiddenType:
7447         handleDelayedForbiddenType(*this, diag, decl);
7448         break;
7449       }
7450     }
7451   } while ((pool = pool->getParent()));
7452 }
7453 
7454 /// Given a set of delayed diagnostics, re-emit them as if they had
7455 /// been delayed in the current context instead of in the given pool.
7456 /// Essentially, this just moves them to the current pool.
7457 void Sema::redelayDiagnostics(DelayedDiagnosticPool &pool) {
7458   DelayedDiagnosticPool *curPool = DelayedDiagnostics.getCurrentPool();
7459   assert(curPool && "re-emitting in undelayed context not supported");
7460   curPool->steal(pool);
7461 }
7462 
7463 static void EmitAvailabilityWarning(Sema &S, AvailabilityResult AR,
7464                                     const NamedDecl *ReferringDecl,
7465                                     const NamedDecl *OffendingDecl,
7466                                     StringRef Message,
7467                                     ArrayRef<SourceLocation> Locs,
7468                                     const ObjCInterfaceDecl *UnknownObjCClass,
7469                                     const ObjCPropertyDecl *ObjCProperty,
7470                                     bool ObjCPropertyAccess) {
7471   // Delay if we're currently parsing a declaration.
7472   if (S.DelayedDiagnostics.shouldDelayDiagnostics()) {
7473     S.DelayedDiagnostics.add(
7474         DelayedDiagnostic::makeAvailability(
7475             AR, Locs, ReferringDecl, OffendingDecl, UnknownObjCClass,
7476             ObjCProperty, Message, ObjCPropertyAccess));
7477     return;
7478   }
7479 
7480   Decl *Ctx = cast<Decl>(S.getCurLexicalContext());
7481   DoEmitAvailabilityWarning(S, AR, Ctx, ReferringDecl, OffendingDecl,
7482                             Message, Locs, UnknownObjCClass, ObjCProperty,
7483                             ObjCPropertyAccess);
7484 }
7485 
7486 namespace {
7487 
7488 /// Returns true if the given statement can be a body-like child of \p Parent.
7489 bool isBodyLikeChildStmt(const Stmt *S, const Stmt *Parent) {
7490   switch (Parent->getStmtClass()) {
7491   case Stmt::IfStmtClass:
7492     return cast<IfStmt>(Parent)->getThen() == S ||
7493            cast<IfStmt>(Parent)->getElse() == S;
7494   case Stmt::WhileStmtClass:
7495     return cast<WhileStmt>(Parent)->getBody() == S;
7496   case Stmt::DoStmtClass:
7497     return cast<DoStmt>(Parent)->getBody() == S;
7498   case Stmt::ForStmtClass:
7499     return cast<ForStmt>(Parent)->getBody() == S;
7500   case Stmt::CXXForRangeStmtClass:
7501     return cast<CXXForRangeStmt>(Parent)->getBody() == S;
7502   case Stmt::ObjCForCollectionStmtClass:
7503     return cast<ObjCForCollectionStmt>(Parent)->getBody() == S;
7504   case Stmt::CaseStmtClass:
7505   case Stmt::DefaultStmtClass:
7506     return cast<SwitchCase>(Parent)->getSubStmt() == S;
7507   default:
7508     return false;
7509   }
7510 }
7511 
7512 class StmtUSEFinder : public RecursiveASTVisitor<StmtUSEFinder> {
7513   const Stmt *Target;
7514 
7515 public:
7516   bool VisitStmt(Stmt *S) { return S != Target; }
7517 
7518   /// Returns true if the given statement is present in the given declaration.
7519   static bool isContained(const Stmt *Target, const Decl *D) {
7520     StmtUSEFinder Visitor;
7521     Visitor.Target = Target;
7522     return !Visitor.TraverseDecl(const_cast<Decl *>(D));
7523   }
7524 };
7525 
7526 /// Traverses the AST and finds the last statement that used a given
7527 /// declaration.
7528 class LastDeclUSEFinder : public RecursiveASTVisitor<LastDeclUSEFinder> {
7529   const Decl *D;
7530 
7531 public:
7532   bool VisitDeclRefExpr(DeclRefExpr *DRE) {
7533     if (DRE->getDecl() == D)
7534       return false;
7535     return true;
7536   }
7537 
7538   static const Stmt *findLastStmtThatUsesDecl(const Decl *D,
7539                                               const CompoundStmt *Scope) {
7540     LastDeclUSEFinder Visitor;
7541     Visitor.D = D;
7542     for (auto I = Scope->body_rbegin(), E = Scope->body_rend(); I != E; ++I) {
7543       const Stmt *S = *I;
7544       if (!Visitor.TraverseStmt(const_cast<Stmt *>(S)))
7545         return S;
7546     }
7547     return nullptr;
7548   }
7549 };
7550 
7551 /// This class implements -Wunguarded-availability.
7552 ///
7553 /// This is done with a traversal of the AST of a function that makes reference
7554 /// to a partially available declaration. Whenever we encounter an \c if of the
7555 /// form: \c if(@available(...)), we use the version from the condition to visit
7556 /// the then statement.
7557 class DiagnoseUnguardedAvailability
7558     : public RecursiveASTVisitor<DiagnoseUnguardedAvailability> {
7559   typedef RecursiveASTVisitor<DiagnoseUnguardedAvailability> Base;
7560 
7561   Sema &SemaRef;
7562   Decl *Ctx;
7563 
7564   /// Stack of potentially nested 'if (@available(...))'s.
7565   SmallVector<VersionTuple, 8> AvailabilityStack;
7566   SmallVector<const Stmt *, 16> StmtStack;
7567 
7568   void DiagnoseDeclAvailability(NamedDecl *D, SourceRange Range);
7569 
7570 public:
7571   DiagnoseUnguardedAvailability(Sema &SemaRef, Decl *Ctx)
7572       : SemaRef(SemaRef), Ctx(Ctx) {
7573     AvailabilityStack.push_back(
7574         SemaRef.Context.getTargetInfo().getPlatformMinVersion());
7575   }
7576 
7577   bool TraverseDecl(Decl *D) {
7578     // Avoid visiting nested functions to prevent duplicate warnings.
7579     if (!D || isa<FunctionDecl>(D))
7580       return true;
7581     return Base::TraverseDecl(D);
7582   }
7583 
7584   bool TraverseStmt(Stmt *S) {
7585     if (!S)
7586       return true;
7587     StmtStack.push_back(S);
7588     bool Result = Base::TraverseStmt(S);
7589     StmtStack.pop_back();
7590     return Result;
7591   }
7592 
7593   void IssueDiagnostics(Stmt *S) { TraverseStmt(S); }
7594 
7595   bool TraverseIfStmt(IfStmt *If);
7596 
7597   bool TraverseLambdaExpr(LambdaExpr *E) { return true; }
7598 
7599   // for 'case X:' statements, don't bother looking at the 'X'; it can't lead
7600   // to any useful diagnostics.
7601   bool TraverseCaseStmt(CaseStmt *CS) { return TraverseStmt(CS->getSubStmt()); }
7602 
7603   bool VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *PRE) {
7604     if (PRE->isClassReceiver())
7605       DiagnoseDeclAvailability(PRE->getClassReceiver(), PRE->getReceiverLocation());
7606     return true;
7607   }
7608 
7609   bool VisitObjCMessageExpr(ObjCMessageExpr *Msg) {
7610     if (ObjCMethodDecl *D = Msg->getMethodDecl())
7611       DiagnoseDeclAvailability(
7612           D, SourceRange(Msg->getSelectorStartLoc(), Msg->getEndLoc()));
7613     return true;
7614   }
7615 
7616   bool VisitDeclRefExpr(DeclRefExpr *DRE) {
7617     DiagnoseDeclAvailability(DRE->getDecl(),
7618                              SourceRange(DRE->getBeginLoc(), DRE->getEndLoc()));
7619     return true;
7620   }
7621 
7622   bool VisitMemberExpr(MemberExpr *ME) {
7623     DiagnoseDeclAvailability(ME->getMemberDecl(),
7624                              SourceRange(ME->getBeginLoc(), ME->getEndLoc()));
7625     return true;
7626   }
7627 
7628   bool VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr *E) {
7629     SemaRef.Diag(E->getBeginLoc(), diag::warn_at_available_unchecked_use)
7630         << (!SemaRef.getLangOpts().ObjC1);
7631     return true;
7632   }
7633 
7634   bool VisitTypeLoc(TypeLoc Ty);
7635 };
7636 
7637 void DiagnoseUnguardedAvailability::DiagnoseDeclAvailability(
7638     NamedDecl *D, SourceRange Range) {
7639   AvailabilityResult Result;
7640   const NamedDecl *OffendingDecl;
7641   std::tie(Result, OffendingDecl) =
7642     ShouldDiagnoseAvailabilityOfDecl(D, nullptr);
7643   if (Result != AR_Available) {
7644     // All other diagnostic kinds have already been handled in
7645     // DiagnoseAvailabilityOfDecl.
7646     if (Result != AR_NotYetIntroduced)
7647       return;
7648 
7649     const AvailabilityAttr *AA =
7650       getAttrForPlatform(SemaRef.getASTContext(), OffendingDecl);
7651     VersionTuple Introduced = AA->getIntroduced();
7652 
7653     if (AvailabilityStack.back() >= Introduced)
7654       return;
7655 
7656     // If the context of this function is less available than D, we should not
7657     // emit a diagnostic.
7658     if (!ShouldDiagnoseAvailabilityInContext(SemaRef, Result, Introduced, Ctx))
7659       return;
7660 
7661     // We would like to emit the diagnostic even if -Wunguarded-availability is
7662     // not specified for deployment targets >= to iOS 11 or equivalent or
7663     // for declarations that were introduced in iOS 11 (macOS 10.13, ...) or
7664     // later.
7665     unsigned DiagKind =
7666         shouldDiagnoseAvailabilityByDefault(
7667             SemaRef.Context,
7668             SemaRef.Context.getTargetInfo().getPlatformMinVersion(), Introduced)
7669             ? diag::warn_unguarded_availability_new
7670             : diag::warn_unguarded_availability;
7671 
7672     SemaRef.Diag(Range.getBegin(), DiagKind)
7673         << Range << D
7674         << AvailabilityAttr::getPrettyPlatformName(
7675                SemaRef.getASTContext().getTargetInfo().getPlatformName())
7676         << Introduced.getAsString();
7677 
7678     SemaRef.Diag(OffendingDecl->getLocation(),
7679                  diag::note_availability_specified_here)
7680         << OffendingDecl << /* partial */ 3;
7681 
7682     auto FixitDiag =
7683         SemaRef.Diag(Range.getBegin(), diag::note_unguarded_available_silence)
7684         << Range << D
7685         << (SemaRef.getLangOpts().ObjC1 ? /*@available*/ 0
7686                                         : /*__builtin_available*/ 1);
7687 
7688     // Find the statement which should be enclosed in the if @available check.
7689     if (StmtStack.empty())
7690       return;
7691     const Stmt *StmtOfUse = StmtStack.back();
7692     const CompoundStmt *Scope = nullptr;
7693     for (const Stmt *S : llvm::reverse(StmtStack)) {
7694       if (const auto *CS = dyn_cast<CompoundStmt>(S)) {
7695         Scope = CS;
7696         break;
7697       }
7698       if (isBodyLikeChildStmt(StmtOfUse, S)) {
7699         // The declaration won't be seen outside of the statement, so we don't
7700         // have to wrap the uses of any declared variables in if (@available).
7701         // Therefore we can avoid setting Scope here.
7702         break;
7703       }
7704       StmtOfUse = S;
7705     }
7706     const Stmt *LastStmtOfUse = nullptr;
7707     if (isa<DeclStmt>(StmtOfUse) && Scope) {
7708       for (const Decl *D : cast<DeclStmt>(StmtOfUse)->decls()) {
7709         if (StmtUSEFinder::isContained(StmtStack.back(), D)) {
7710           LastStmtOfUse = LastDeclUSEFinder::findLastStmtThatUsesDecl(D, Scope);
7711           break;
7712         }
7713       }
7714     }
7715 
7716     const SourceManager &SM = SemaRef.getSourceManager();
7717     SourceLocation IfInsertionLoc =
7718         SM.getExpansionLoc(StmtOfUse->getBeginLoc());
7719     SourceLocation StmtEndLoc =
7720         SM.getExpansionRange(
7721               (LastStmtOfUse ? LastStmtOfUse : StmtOfUse)->getEndLoc())
7722             .getEnd();
7723     if (SM.getFileID(IfInsertionLoc) != SM.getFileID(StmtEndLoc))
7724       return;
7725 
7726     StringRef Indentation = Lexer::getIndentationForLine(IfInsertionLoc, SM);
7727     const char *ExtraIndentation = "    ";
7728     std::string FixItString;
7729     llvm::raw_string_ostream FixItOS(FixItString);
7730     FixItOS << "if (" << (SemaRef.getLangOpts().ObjC1 ? "@available"
7731                                                       : "__builtin_available")
7732             << "("
7733             << AvailabilityAttr::getPlatformNameSourceSpelling(
7734                    SemaRef.getASTContext().getTargetInfo().getPlatformName())
7735             << " " << Introduced.getAsString() << ", *)) {\n"
7736             << Indentation << ExtraIndentation;
7737     FixitDiag << FixItHint::CreateInsertion(IfInsertionLoc, FixItOS.str());
7738     SourceLocation ElseInsertionLoc = Lexer::findLocationAfterToken(
7739         StmtEndLoc, tok::semi, SM, SemaRef.getLangOpts(),
7740         /*SkipTrailingWhitespaceAndNewLine=*/false);
7741     if (ElseInsertionLoc.isInvalid())
7742       ElseInsertionLoc =
7743           Lexer::getLocForEndOfToken(StmtEndLoc, 0, SM, SemaRef.getLangOpts());
7744     FixItOS.str().clear();
7745     FixItOS << "\n"
7746             << Indentation << "} else {\n"
7747             << Indentation << ExtraIndentation
7748             << "// Fallback on earlier versions\n"
7749             << Indentation << "}";
7750     FixitDiag << FixItHint::CreateInsertion(ElseInsertionLoc, FixItOS.str());
7751   }
7752 }
7753 
7754 bool DiagnoseUnguardedAvailability::VisitTypeLoc(TypeLoc Ty) {
7755   const Type *TyPtr = Ty.getTypePtr();
7756   SourceRange Range{Ty.getBeginLoc(), Ty.getEndLoc()};
7757 
7758   if (Range.isInvalid())
7759     return true;
7760 
7761   if (const auto *TT = dyn_cast<TagType>(TyPtr)) {
7762     TagDecl *TD = TT->getDecl();
7763     DiagnoseDeclAvailability(TD, Range);
7764 
7765   } else if (const auto *TD = dyn_cast<TypedefType>(TyPtr)) {
7766     TypedefNameDecl *D = TD->getDecl();
7767     DiagnoseDeclAvailability(D, Range);
7768 
7769   } else if (const auto *ObjCO = dyn_cast<ObjCObjectType>(TyPtr)) {
7770     if (NamedDecl *D = ObjCO->getInterface())
7771       DiagnoseDeclAvailability(D, Range);
7772   }
7773 
7774   return true;
7775 }
7776 
7777 bool DiagnoseUnguardedAvailability::TraverseIfStmt(IfStmt *If) {
7778   VersionTuple CondVersion;
7779   if (auto *E = dyn_cast<ObjCAvailabilityCheckExpr>(If->getCond())) {
7780     CondVersion = E->getVersion();
7781 
7782     // If we're using the '*' case here or if this check is redundant, then we
7783     // use the enclosing version to check both branches.
7784     if (CondVersion.empty() || CondVersion <= AvailabilityStack.back())
7785       return TraverseStmt(If->getThen()) && TraverseStmt(If->getElse());
7786   } else {
7787     // This isn't an availability checking 'if', we can just continue.
7788     return Base::TraverseIfStmt(If);
7789   }
7790 
7791   AvailabilityStack.push_back(CondVersion);
7792   bool ShouldContinue = TraverseStmt(If->getThen());
7793   AvailabilityStack.pop_back();
7794 
7795   return ShouldContinue && TraverseStmt(If->getElse());
7796 }
7797 
7798 } // end anonymous namespace
7799 
7800 void Sema::DiagnoseUnguardedAvailabilityViolations(Decl *D) {
7801   Stmt *Body = nullptr;
7802 
7803   if (auto *FD = D->getAsFunction()) {
7804     // FIXME: We only examine the pattern decl for availability violations now,
7805     // but we should also examine instantiated templates.
7806     if (FD->isTemplateInstantiation())
7807       return;
7808 
7809     Body = FD->getBody();
7810   } else if (auto *MD = dyn_cast<ObjCMethodDecl>(D))
7811     Body = MD->getBody();
7812   else if (auto *BD = dyn_cast<BlockDecl>(D))
7813     Body = BD->getBody();
7814 
7815   assert(Body && "Need a body here!");
7816 
7817   DiagnoseUnguardedAvailability(*this, D).IssueDiagnostics(Body);
7818 }
7819 
7820 void Sema::DiagnoseAvailabilityOfDecl(NamedDecl *D,
7821                                       ArrayRef<SourceLocation> Locs,
7822                                       const ObjCInterfaceDecl *UnknownObjCClass,
7823                                       bool ObjCPropertyAccess,
7824                                       bool AvoidPartialAvailabilityChecks) {
7825   std::string Message;
7826   AvailabilityResult Result;
7827   const NamedDecl* OffendingDecl;
7828   // See if this declaration is unavailable, deprecated, or partial.
7829   std::tie(Result, OffendingDecl) = ShouldDiagnoseAvailabilityOfDecl(D, &Message);
7830   if (Result == AR_Available)
7831     return;
7832 
7833   if (Result == AR_NotYetIntroduced) {
7834     if (AvoidPartialAvailabilityChecks)
7835       return;
7836 
7837     // We need to know the @available context in the current function to
7838     // diagnose this use, let DiagnoseUnguardedAvailabilityViolations do that
7839     // when we're done parsing the current function.
7840     if (getCurFunctionOrMethodDecl()) {
7841       getEnclosingFunction()->HasPotentialAvailabilityViolations = true;
7842       return;
7843     } else if (getCurBlock() || getCurLambda()) {
7844       getCurFunction()->HasPotentialAvailabilityViolations = true;
7845       return;
7846     }
7847   }
7848 
7849   const ObjCPropertyDecl *ObjCPDecl = nullptr;
7850   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D)) {
7851     if (const ObjCPropertyDecl *PD = MD->findPropertyDecl()) {
7852       AvailabilityResult PDeclResult = PD->getAvailability(nullptr);
7853       if (PDeclResult == Result)
7854         ObjCPDecl = PD;
7855     }
7856   }
7857 
7858   EmitAvailabilityWarning(*this, Result, D, OffendingDecl, Message, Locs,
7859                           UnknownObjCClass, ObjCPDecl, ObjCPropertyAccess);
7860 }
7861