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 //===----------------------------------------------------------------------===//
5921 // Top Level Sema Entry Points
5922 //===----------------------------------------------------------------------===//
5923 
5924 /// ProcessDeclAttribute - Apply the specific attribute to the specified decl if
5925 /// the attribute applies to decls.  If the attribute is a type attribute, just
5926 /// silently ignore it if a GNU attribute.
5927 static void ProcessDeclAttribute(Sema &S, Scope *scope, Decl *D,
5928                                  const ParsedAttr &AL,
5929                                  bool IncludeCXX11Attributes) {
5930   if (AL.isInvalid() || AL.getKind() == ParsedAttr::IgnoredAttribute)
5931     return;
5932 
5933   // Ignore C++11 attributes on declarator chunks: they appertain to the type
5934   // instead.
5935   if (AL.isCXX11Attribute() && !IncludeCXX11Attributes)
5936     return;
5937 
5938   // Unknown attributes are automatically warned on. Target-specific attributes
5939   // which do not apply to the current target architecture are treated as
5940   // though they were unknown attributes.
5941   if (AL.getKind() == ParsedAttr::UnknownAttribute ||
5942       !AL.existsInTarget(S.Context.getTargetInfo())) {
5943     S.Diag(AL.getLoc(), AL.isDeclspecAttribute()
5944                             ? diag::warn_unhandled_ms_attribute_ignored
5945                             : diag::warn_unknown_attribute_ignored)
5946         << AL;
5947     return;
5948   }
5949 
5950   if (handleCommonAttributeFeatures(S, D, AL))
5951     return;
5952 
5953   switch (AL.getKind()) {
5954   default:
5955     if (!AL.isStmtAttr()) {
5956       // Type attributes are handled elsewhere; silently move on.
5957       assert(AL.isTypeAttr() && "Non-type attribute not handled");
5958       break;
5959     }
5960     S.Diag(AL.getLoc(), diag::err_stmt_attribute_invalid_on_decl)
5961         << AL << D->getLocation();
5962     break;
5963   case ParsedAttr::AT_Interrupt:
5964     handleInterruptAttr(S, D, AL);
5965     break;
5966   case ParsedAttr::AT_X86ForceAlignArgPointer:
5967     handleX86ForceAlignArgPointerAttr(S, D, AL);
5968     break;
5969   case ParsedAttr::AT_DLLExport:
5970   case ParsedAttr::AT_DLLImport:
5971     handleDLLAttr(S, D, AL);
5972     break;
5973   case ParsedAttr::AT_Mips16:
5974     handleSimpleAttributeWithExclusions<Mips16Attr, MicroMipsAttr,
5975                                         MipsInterruptAttr>(S, D, AL);
5976     break;
5977   case ParsedAttr::AT_NoMips16:
5978     handleSimpleAttribute<NoMips16Attr>(S, D, AL);
5979     break;
5980   case ParsedAttr::AT_MicroMips:
5981     handleSimpleAttributeWithExclusions<MicroMipsAttr, Mips16Attr>(S, D, AL);
5982     break;
5983   case ParsedAttr::AT_NoMicroMips:
5984     handleSimpleAttribute<NoMicroMipsAttr>(S, D, AL);
5985     break;
5986   case ParsedAttr::AT_MipsLongCall:
5987     handleSimpleAttributeWithExclusions<MipsLongCallAttr, MipsShortCallAttr>(
5988         S, D, AL);
5989     break;
5990   case ParsedAttr::AT_MipsShortCall:
5991     handleSimpleAttributeWithExclusions<MipsShortCallAttr, MipsLongCallAttr>(
5992         S, D, AL);
5993     break;
5994   case ParsedAttr::AT_AMDGPUFlatWorkGroupSize:
5995     handleAMDGPUFlatWorkGroupSizeAttr(S, D, AL);
5996     break;
5997   case ParsedAttr::AT_AMDGPUWavesPerEU:
5998     handleAMDGPUWavesPerEUAttr(S, D, AL);
5999     break;
6000   case ParsedAttr::AT_AMDGPUNumSGPR:
6001     handleAMDGPUNumSGPRAttr(S, D, AL);
6002     break;
6003   case ParsedAttr::AT_AMDGPUNumVGPR:
6004     handleAMDGPUNumVGPRAttr(S, D, AL);
6005     break;
6006   case ParsedAttr::AT_AVRSignal:
6007     handleAVRSignalAttr(S, D, AL);
6008     break;
6009   case ParsedAttr::AT_IBAction:
6010     handleSimpleAttribute<IBActionAttr>(S, D, AL);
6011     break;
6012   case ParsedAttr::AT_IBOutlet:
6013     handleIBOutlet(S, D, AL);
6014     break;
6015   case ParsedAttr::AT_IBOutletCollection:
6016     handleIBOutletCollection(S, D, AL);
6017     break;
6018   case ParsedAttr::AT_IFunc:
6019     handleIFuncAttr(S, D, AL);
6020     break;
6021   case ParsedAttr::AT_Alias:
6022     handleAliasAttr(S, D, AL);
6023     break;
6024   case ParsedAttr::AT_Aligned:
6025     handleAlignedAttr(S, D, AL);
6026     break;
6027   case ParsedAttr::AT_AlignValue:
6028     handleAlignValueAttr(S, D, AL);
6029     break;
6030   case ParsedAttr::AT_AllocSize:
6031     handleAllocSizeAttr(S, D, AL);
6032     break;
6033   case ParsedAttr::AT_AlwaysInline:
6034     handleAlwaysInlineAttr(S, D, AL);
6035     break;
6036   case ParsedAttr::AT_Artificial:
6037     handleSimpleAttribute<ArtificialAttr>(S, D, AL);
6038     break;
6039   case ParsedAttr::AT_AnalyzerNoReturn:
6040     handleAnalyzerNoReturnAttr(S, D, AL);
6041     break;
6042   case ParsedAttr::AT_TLSModel:
6043     handleTLSModelAttr(S, D, AL);
6044     break;
6045   case ParsedAttr::AT_Annotate:
6046     handleAnnotateAttr(S, D, AL);
6047     break;
6048   case ParsedAttr::AT_Availability:
6049     handleAvailabilityAttr(S, D, AL);
6050     break;
6051   case ParsedAttr::AT_CarriesDependency:
6052     handleDependencyAttr(S, scope, D, AL);
6053     break;
6054   case ParsedAttr::AT_CPUDispatch:
6055   case ParsedAttr::AT_CPUSpecific:
6056     handleCPUSpecificAttr(S, D, AL);
6057     break;
6058   case ParsedAttr::AT_Common:
6059     handleCommonAttr(S, D, AL);
6060     break;
6061   case ParsedAttr::AT_CUDAConstant:
6062     handleConstantAttr(S, D, AL);
6063     break;
6064   case ParsedAttr::AT_PassObjectSize:
6065     handlePassObjectSizeAttr(S, D, AL);
6066     break;
6067   case ParsedAttr::AT_Constructor:
6068     handleConstructorAttr(S, D, AL);
6069     break;
6070   case ParsedAttr::AT_CXX11NoReturn:
6071     handleSimpleAttribute<CXX11NoReturnAttr>(S, D, AL);
6072     break;
6073   case ParsedAttr::AT_Deprecated:
6074     handleDeprecatedAttr(S, D, AL);
6075     break;
6076   case ParsedAttr::AT_Destructor:
6077     handleDestructorAttr(S, D, AL);
6078     break;
6079   case ParsedAttr::AT_EnableIf:
6080     handleEnableIfAttr(S, D, AL);
6081     break;
6082   case ParsedAttr::AT_DiagnoseIf:
6083     handleDiagnoseIfAttr(S, D, AL);
6084     break;
6085   case ParsedAttr::AT_ExtVectorType:
6086     handleExtVectorTypeAttr(S, D, AL);
6087     break;
6088   case ParsedAttr::AT_ExternalSourceSymbol:
6089     handleExternalSourceSymbolAttr(S, D, AL);
6090     break;
6091   case ParsedAttr::AT_MinSize:
6092     handleMinSizeAttr(S, D, AL);
6093     break;
6094   case ParsedAttr::AT_OptimizeNone:
6095     handleOptimizeNoneAttr(S, D, AL);
6096     break;
6097   case ParsedAttr::AT_FlagEnum:
6098     handleSimpleAttribute<FlagEnumAttr>(S, D, AL);
6099     break;
6100   case ParsedAttr::AT_EnumExtensibility:
6101     handleEnumExtensibilityAttr(S, D, AL);
6102     break;
6103   case ParsedAttr::AT_Flatten:
6104     handleSimpleAttribute<FlattenAttr>(S, D, AL);
6105     break;
6106   case ParsedAttr::AT_Format:
6107     handleFormatAttr(S, D, AL);
6108     break;
6109   case ParsedAttr::AT_FormatArg:
6110     handleFormatArgAttr(S, D, AL);
6111     break;
6112   case ParsedAttr::AT_CUDAGlobal:
6113     handleGlobalAttr(S, D, AL);
6114     break;
6115   case ParsedAttr::AT_CUDADevice:
6116     handleSimpleAttributeWithExclusions<CUDADeviceAttr, CUDAGlobalAttr>(S, D,
6117                                                                         AL);
6118     break;
6119   case ParsedAttr::AT_CUDAHost:
6120     handleSimpleAttributeWithExclusions<CUDAHostAttr, CUDAGlobalAttr>(S, D, AL);
6121     break;
6122   case ParsedAttr::AT_GNUInline:
6123     handleGNUInlineAttr(S, D, AL);
6124     break;
6125   case ParsedAttr::AT_CUDALaunchBounds:
6126     handleLaunchBoundsAttr(S, D, AL);
6127     break;
6128   case ParsedAttr::AT_Restrict:
6129     handleRestrictAttr(S, D, AL);
6130     break;
6131   case ParsedAttr::AT_LifetimeBound:
6132     handleSimpleAttribute<LifetimeBoundAttr>(S, D, AL);
6133     break;
6134   case ParsedAttr::AT_MayAlias:
6135     handleSimpleAttribute<MayAliasAttr>(S, D, AL);
6136     break;
6137   case ParsedAttr::AT_Mode:
6138     handleModeAttr(S, D, AL);
6139     break;
6140   case ParsedAttr::AT_NoAlias:
6141     handleSimpleAttribute<NoAliasAttr>(S, D, AL);
6142     break;
6143   case ParsedAttr::AT_NoCommon:
6144     handleSimpleAttribute<NoCommonAttr>(S, D, AL);
6145     break;
6146   case ParsedAttr::AT_NoSplitStack:
6147     handleSimpleAttribute<NoSplitStackAttr>(S, D, AL);
6148     break;
6149   case ParsedAttr::AT_NonNull:
6150     if (auto *PVD = dyn_cast<ParmVarDecl>(D))
6151       handleNonNullAttrParameter(S, PVD, AL);
6152     else
6153       handleNonNullAttr(S, D, AL);
6154     break;
6155   case ParsedAttr::AT_ReturnsNonNull:
6156     handleReturnsNonNullAttr(S, D, AL);
6157     break;
6158   case ParsedAttr::AT_NoEscape:
6159     handleNoEscapeAttr(S, D, AL);
6160     break;
6161   case ParsedAttr::AT_AssumeAligned:
6162     handleAssumeAlignedAttr(S, D, AL);
6163     break;
6164   case ParsedAttr::AT_AllocAlign:
6165     handleAllocAlignAttr(S, D, AL);
6166     break;
6167   case ParsedAttr::AT_Overloadable:
6168     handleSimpleAttribute<OverloadableAttr>(S, D, AL);
6169     break;
6170   case ParsedAttr::AT_Ownership:
6171     handleOwnershipAttr(S, D, AL);
6172     break;
6173   case ParsedAttr::AT_Cold:
6174     handleSimpleAttributeWithExclusions<ColdAttr, HotAttr>(S, D, AL);
6175     break;
6176   case ParsedAttr::AT_Hot:
6177     handleSimpleAttributeWithExclusions<HotAttr, ColdAttr>(S, D, AL);
6178     break;
6179   case ParsedAttr::AT_Naked:
6180     handleNakedAttr(S, D, AL);
6181     break;
6182   case ParsedAttr::AT_NoReturn:
6183     handleNoReturnAttr(S, D, AL);
6184     break;
6185   case ParsedAttr::AT_AnyX86NoCfCheck:
6186     handleNoCfCheckAttr(S, D, AL);
6187     break;
6188   case ParsedAttr::AT_NoThrow:
6189     handleSimpleAttribute<NoThrowAttr>(S, D, AL);
6190     break;
6191   case ParsedAttr::AT_CUDAShared:
6192     handleSharedAttr(S, D, AL);
6193     break;
6194   case ParsedAttr::AT_VecReturn:
6195     handleVecReturnAttr(S, D, AL);
6196     break;
6197   case ParsedAttr::AT_ObjCOwnership:
6198     handleObjCOwnershipAttr(S, D, AL);
6199     break;
6200   case ParsedAttr::AT_ObjCPreciseLifetime:
6201     handleObjCPreciseLifetimeAttr(S, D, AL);
6202     break;
6203   case ParsedAttr::AT_ObjCReturnsInnerPointer:
6204     handleObjCReturnsInnerPointerAttr(S, D, AL);
6205     break;
6206   case ParsedAttr::AT_ObjCRequiresSuper:
6207     handleObjCRequiresSuperAttr(S, D, AL);
6208     break;
6209   case ParsedAttr::AT_ObjCBridge:
6210     handleObjCBridgeAttr(S, D, AL);
6211     break;
6212   case ParsedAttr::AT_ObjCBridgeMutable:
6213     handleObjCBridgeMutableAttr(S, D, AL);
6214     break;
6215   case ParsedAttr::AT_ObjCBridgeRelated:
6216     handleObjCBridgeRelatedAttr(S, D, AL);
6217     break;
6218   case ParsedAttr::AT_ObjCDesignatedInitializer:
6219     handleObjCDesignatedInitializer(S, D, AL);
6220     break;
6221   case ParsedAttr::AT_ObjCRuntimeName:
6222     handleObjCRuntimeName(S, D, AL);
6223     break;
6224   case ParsedAttr::AT_ObjCRuntimeVisible:
6225     handleSimpleAttribute<ObjCRuntimeVisibleAttr>(S, D, AL);
6226     break;
6227   case ParsedAttr::AT_ObjCBoxable:
6228     handleObjCBoxable(S, D, AL);
6229     break;
6230   case ParsedAttr::AT_CFAuditedTransfer:
6231     handleSimpleAttributeWithExclusions<CFAuditedTransferAttr,
6232                                         CFUnknownTransferAttr>(S, D, AL);
6233     break;
6234   case ParsedAttr::AT_CFUnknownTransfer:
6235     handleSimpleAttributeWithExclusions<CFUnknownTransferAttr,
6236                                         CFAuditedTransferAttr>(S, D, AL);
6237     break;
6238   case ParsedAttr::AT_CFConsumed:
6239   case ParsedAttr::AT_NSConsumed:
6240     handleNSConsumedAttr(S, D, AL);
6241     break;
6242   case ParsedAttr::AT_NSConsumesSelf:
6243     handleSimpleAttribute<NSConsumesSelfAttr>(S, D, AL);
6244     break;
6245   case ParsedAttr::AT_NSReturnsAutoreleased:
6246   case ParsedAttr::AT_NSReturnsNotRetained:
6247   case ParsedAttr::AT_CFReturnsNotRetained:
6248   case ParsedAttr::AT_NSReturnsRetained:
6249   case ParsedAttr::AT_CFReturnsRetained:
6250     handleNSReturnsRetainedAttr(S, D, AL);
6251     break;
6252   case ParsedAttr::AT_WorkGroupSizeHint:
6253     handleWorkGroupSize<WorkGroupSizeHintAttr>(S, D, AL);
6254     break;
6255   case ParsedAttr::AT_ReqdWorkGroupSize:
6256     handleWorkGroupSize<ReqdWorkGroupSizeAttr>(S, D, AL);
6257     break;
6258   case ParsedAttr::AT_OpenCLIntelReqdSubGroupSize:
6259     handleSubGroupSize(S, D, AL);
6260     break;
6261   case ParsedAttr::AT_VecTypeHint:
6262     handleVecTypeHint(S, D, AL);
6263     break;
6264   case ParsedAttr::AT_RequireConstantInit:
6265     handleSimpleAttribute<RequireConstantInitAttr>(S, D, AL);
6266     break;
6267   case ParsedAttr::AT_InitPriority:
6268     handleInitPriorityAttr(S, D, AL);
6269     break;
6270   case ParsedAttr::AT_Packed:
6271     handlePackedAttr(S, D, AL);
6272     break;
6273   case ParsedAttr::AT_Section:
6274     handleSectionAttr(S, D, AL);
6275     break;
6276   case ParsedAttr::AT_CodeSeg:
6277     handleCodeSegAttr(S, D, AL);
6278     break;
6279   case ParsedAttr::AT_Target:
6280     handleTargetAttr(S, D, AL);
6281     break;
6282   case ParsedAttr::AT_MinVectorWidth:
6283     handleMinVectorWidthAttr(S, D, AL);
6284     break;
6285   case ParsedAttr::AT_Unavailable:
6286     handleAttrWithMessage<UnavailableAttr>(S, D, AL);
6287     break;
6288   case ParsedAttr::AT_ArcWeakrefUnavailable:
6289     handleSimpleAttribute<ArcWeakrefUnavailableAttr>(S, D, AL);
6290     break;
6291   case ParsedAttr::AT_ObjCRootClass:
6292     handleSimpleAttribute<ObjCRootClassAttr>(S, D, AL);
6293     break;
6294   case ParsedAttr::AT_ObjCSubclassingRestricted:
6295     handleSimpleAttribute<ObjCSubclassingRestrictedAttr>(S, D, AL);
6296     break;
6297   case ParsedAttr::AT_ObjCExplicitProtocolImpl:
6298     handleObjCSuppresProtocolAttr(S, D, AL);
6299     break;
6300   case ParsedAttr::AT_ObjCRequiresPropertyDefs:
6301     handleSimpleAttribute<ObjCRequiresPropertyDefsAttr>(S, D, AL);
6302     break;
6303   case ParsedAttr::AT_Unused:
6304     handleUnusedAttr(S, D, AL);
6305     break;
6306   case ParsedAttr::AT_ReturnsTwice:
6307     handleSimpleAttribute<ReturnsTwiceAttr>(S, D, AL);
6308     break;
6309   case ParsedAttr::AT_NotTailCalled:
6310     handleSimpleAttributeWithExclusions<NotTailCalledAttr, AlwaysInlineAttr>(
6311         S, D, AL);
6312     break;
6313   case ParsedAttr::AT_DisableTailCalls:
6314     handleSimpleAttributeWithExclusions<DisableTailCallsAttr, NakedAttr>(S, D,
6315                                                                          AL);
6316     break;
6317   case ParsedAttr::AT_Used:
6318     handleSimpleAttribute<UsedAttr>(S, D, AL);
6319     break;
6320   case ParsedAttr::AT_Visibility:
6321     handleVisibilityAttr(S, D, AL, false);
6322     break;
6323   case ParsedAttr::AT_TypeVisibility:
6324     handleVisibilityAttr(S, D, AL, true);
6325     break;
6326   case ParsedAttr::AT_WarnUnused:
6327     handleSimpleAttribute<WarnUnusedAttr>(S, D, AL);
6328     break;
6329   case ParsedAttr::AT_WarnUnusedResult:
6330     handleWarnUnusedResult(S, D, AL);
6331     break;
6332   case ParsedAttr::AT_Weak:
6333     handleSimpleAttribute<WeakAttr>(S, D, AL);
6334     break;
6335   case ParsedAttr::AT_WeakRef:
6336     handleWeakRefAttr(S, D, AL);
6337     break;
6338   case ParsedAttr::AT_WeakImport:
6339     handleWeakImportAttr(S, D, AL);
6340     break;
6341   case ParsedAttr::AT_TransparentUnion:
6342     handleTransparentUnionAttr(S, D, AL);
6343     break;
6344   case ParsedAttr::AT_ObjCException:
6345     handleSimpleAttribute<ObjCExceptionAttr>(S, D, AL);
6346     break;
6347   case ParsedAttr::AT_ObjCMethodFamily:
6348     handleObjCMethodFamilyAttr(S, D, AL);
6349     break;
6350   case ParsedAttr::AT_ObjCNSObject:
6351     handleObjCNSObject(S, D, AL);
6352     break;
6353   case ParsedAttr::AT_ObjCIndependentClass:
6354     handleObjCIndependentClass(S, D, AL);
6355     break;
6356   case ParsedAttr::AT_Blocks:
6357     handleBlocksAttr(S, D, AL);
6358     break;
6359   case ParsedAttr::AT_Sentinel:
6360     handleSentinelAttr(S, D, AL);
6361     break;
6362   case ParsedAttr::AT_Const:
6363     handleSimpleAttribute<ConstAttr>(S, D, AL);
6364     break;
6365   case ParsedAttr::AT_Pure:
6366     handleSimpleAttribute<PureAttr>(S, D, AL);
6367     break;
6368   case ParsedAttr::AT_Cleanup:
6369     handleCleanupAttr(S, D, AL);
6370     break;
6371   case ParsedAttr::AT_NoDebug:
6372     handleNoDebugAttr(S, D, AL);
6373     break;
6374   case ParsedAttr::AT_NoDuplicate:
6375     handleSimpleAttribute<NoDuplicateAttr>(S, D, AL);
6376     break;
6377   case ParsedAttr::AT_Convergent:
6378     handleSimpleAttribute<ConvergentAttr>(S, D, AL);
6379     break;
6380   case ParsedAttr::AT_NoInline:
6381     handleSimpleAttribute<NoInlineAttr>(S, D, AL);
6382     break;
6383   case ParsedAttr::AT_NoInstrumentFunction: // Interacts with -pg.
6384     handleSimpleAttribute<NoInstrumentFunctionAttr>(S, D, AL);
6385     break;
6386   case ParsedAttr::AT_NoStackProtector:
6387     // Interacts with -fstack-protector options.
6388     handleSimpleAttribute<NoStackProtectorAttr>(S, D, AL);
6389     break;
6390   case ParsedAttr::AT_StdCall:
6391   case ParsedAttr::AT_CDecl:
6392   case ParsedAttr::AT_FastCall:
6393   case ParsedAttr::AT_ThisCall:
6394   case ParsedAttr::AT_Pascal:
6395   case ParsedAttr::AT_RegCall:
6396   case ParsedAttr::AT_SwiftCall:
6397   case ParsedAttr::AT_VectorCall:
6398   case ParsedAttr::AT_MSABI:
6399   case ParsedAttr::AT_SysVABI:
6400   case ParsedAttr::AT_Pcs:
6401   case ParsedAttr::AT_IntelOclBicc:
6402   case ParsedAttr::AT_PreserveMost:
6403   case ParsedAttr::AT_PreserveAll:
6404     handleCallConvAttr(S, D, AL);
6405     break;
6406   case ParsedAttr::AT_Suppress:
6407     handleSuppressAttr(S, D, AL);
6408     break;
6409   case ParsedAttr::AT_OpenCLKernel:
6410     handleSimpleAttribute<OpenCLKernelAttr>(S, D, AL);
6411     break;
6412   case ParsedAttr::AT_OpenCLAccess:
6413     handleOpenCLAccessAttr(S, D, AL);
6414     break;
6415   case ParsedAttr::AT_OpenCLNoSVM:
6416     handleOpenCLNoSVMAttr(S, D, AL);
6417     break;
6418   case ParsedAttr::AT_SwiftContext:
6419     handleParameterABIAttr(S, D, AL, ParameterABI::SwiftContext);
6420     break;
6421   case ParsedAttr::AT_SwiftErrorResult:
6422     handleParameterABIAttr(S, D, AL, ParameterABI::SwiftErrorResult);
6423     break;
6424   case ParsedAttr::AT_SwiftIndirectResult:
6425     handleParameterABIAttr(S, D, AL, ParameterABI::SwiftIndirectResult);
6426     break;
6427   case ParsedAttr::AT_InternalLinkage:
6428     handleInternalLinkageAttr(S, D, AL);
6429     break;
6430   case ParsedAttr::AT_LTOVisibilityPublic:
6431     handleSimpleAttribute<LTOVisibilityPublicAttr>(S, D, AL);
6432     break;
6433 
6434   // Microsoft attributes:
6435   case ParsedAttr::AT_EmptyBases:
6436     handleSimpleAttribute<EmptyBasesAttr>(S, D, AL);
6437     break;
6438   case ParsedAttr::AT_LayoutVersion:
6439     handleLayoutVersion(S, D, AL);
6440     break;
6441   case ParsedAttr::AT_TrivialABI:
6442     handleSimpleAttribute<TrivialABIAttr>(S, D, AL);
6443     break;
6444   case ParsedAttr::AT_MSNoVTable:
6445     handleSimpleAttribute<MSNoVTableAttr>(S, D, AL);
6446     break;
6447   case ParsedAttr::AT_MSStruct:
6448     handleSimpleAttribute<MSStructAttr>(S, D, AL);
6449     break;
6450   case ParsedAttr::AT_Uuid:
6451     handleUuidAttr(S, D, AL);
6452     break;
6453   case ParsedAttr::AT_MSInheritance:
6454     handleMSInheritanceAttr(S, D, AL);
6455     break;
6456   case ParsedAttr::AT_SelectAny:
6457     handleSimpleAttribute<SelectAnyAttr>(S, D, AL);
6458     break;
6459   case ParsedAttr::AT_Thread:
6460     handleDeclspecThreadAttr(S, D, AL);
6461     break;
6462 
6463   case ParsedAttr::AT_AbiTag:
6464     handleAbiTagAttr(S, D, AL);
6465     break;
6466 
6467   // Thread safety attributes:
6468   case ParsedAttr::AT_AssertExclusiveLock:
6469     handleAssertExclusiveLockAttr(S, D, AL);
6470     break;
6471   case ParsedAttr::AT_AssertSharedLock:
6472     handleAssertSharedLockAttr(S, D, AL);
6473     break;
6474   case ParsedAttr::AT_GuardedVar:
6475     handleSimpleAttribute<GuardedVarAttr>(S, D, AL);
6476     break;
6477   case ParsedAttr::AT_PtGuardedVar:
6478     handlePtGuardedVarAttr(S, D, AL);
6479     break;
6480   case ParsedAttr::AT_ScopedLockable:
6481     handleSimpleAttribute<ScopedLockableAttr>(S, D, AL);
6482     break;
6483   case ParsedAttr::AT_NoSanitize:
6484     handleNoSanitizeAttr(S, D, AL);
6485     break;
6486   case ParsedAttr::AT_NoSanitizeSpecific:
6487     handleNoSanitizeSpecificAttr(S, D, AL);
6488     break;
6489   case ParsedAttr::AT_NoThreadSafetyAnalysis:
6490     handleSimpleAttribute<NoThreadSafetyAnalysisAttr>(S, D, AL);
6491     break;
6492   case ParsedAttr::AT_GuardedBy:
6493     handleGuardedByAttr(S, D, AL);
6494     break;
6495   case ParsedAttr::AT_PtGuardedBy:
6496     handlePtGuardedByAttr(S, D, AL);
6497     break;
6498   case ParsedAttr::AT_ExclusiveTrylockFunction:
6499     handleExclusiveTrylockFunctionAttr(S, D, AL);
6500     break;
6501   case ParsedAttr::AT_LockReturned:
6502     handleLockReturnedAttr(S, D, AL);
6503     break;
6504   case ParsedAttr::AT_LocksExcluded:
6505     handleLocksExcludedAttr(S, D, AL);
6506     break;
6507   case ParsedAttr::AT_SharedTrylockFunction:
6508     handleSharedTrylockFunctionAttr(S, D, AL);
6509     break;
6510   case ParsedAttr::AT_AcquiredBefore:
6511     handleAcquiredBeforeAttr(S, D, AL);
6512     break;
6513   case ParsedAttr::AT_AcquiredAfter:
6514     handleAcquiredAfterAttr(S, D, AL);
6515     break;
6516 
6517   // Capability analysis attributes.
6518   case ParsedAttr::AT_Capability:
6519   case ParsedAttr::AT_Lockable:
6520     handleCapabilityAttr(S, D, AL);
6521     break;
6522   case ParsedAttr::AT_RequiresCapability:
6523     handleRequiresCapabilityAttr(S, D, AL);
6524     break;
6525 
6526   case ParsedAttr::AT_AssertCapability:
6527     handleAssertCapabilityAttr(S, D, AL);
6528     break;
6529   case ParsedAttr::AT_AcquireCapability:
6530     handleAcquireCapabilityAttr(S, D, AL);
6531     break;
6532   case ParsedAttr::AT_ReleaseCapability:
6533     handleReleaseCapabilityAttr(S, D, AL);
6534     break;
6535   case ParsedAttr::AT_TryAcquireCapability:
6536     handleTryAcquireCapabilityAttr(S, D, AL);
6537     break;
6538 
6539   // Consumed analysis attributes.
6540   case ParsedAttr::AT_Consumable:
6541     handleConsumableAttr(S, D, AL);
6542     break;
6543   case ParsedAttr::AT_ConsumableAutoCast:
6544     handleSimpleAttribute<ConsumableAutoCastAttr>(S, D, AL);
6545     break;
6546   case ParsedAttr::AT_ConsumableSetOnRead:
6547     handleSimpleAttribute<ConsumableSetOnReadAttr>(S, D, AL);
6548     break;
6549   case ParsedAttr::AT_CallableWhen:
6550     handleCallableWhenAttr(S, D, AL);
6551     break;
6552   case ParsedAttr::AT_ParamTypestate:
6553     handleParamTypestateAttr(S, D, AL);
6554     break;
6555   case ParsedAttr::AT_ReturnTypestate:
6556     handleReturnTypestateAttr(S, D, AL);
6557     break;
6558   case ParsedAttr::AT_SetTypestate:
6559     handleSetTypestateAttr(S, D, AL);
6560     break;
6561   case ParsedAttr::AT_TestTypestate:
6562     handleTestTypestateAttr(S, D, AL);
6563     break;
6564 
6565   // Type safety attributes.
6566   case ParsedAttr::AT_ArgumentWithTypeTag:
6567     handleArgumentWithTypeTagAttr(S, D, AL);
6568     break;
6569   case ParsedAttr::AT_TypeTagForDatatype:
6570     handleTypeTagForDatatypeAttr(S, D, AL);
6571     break;
6572   case ParsedAttr::AT_AnyX86NoCallerSavedRegisters:
6573     handleSimpleAttribute<AnyX86NoCallerSavedRegistersAttr>(S, D, AL);
6574     break;
6575   case ParsedAttr::AT_RenderScriptKernel:
6576     handleSimpleAttribute<RenderScriptKernelAttr>(S, D, AL);
6577     break;
6578   // XRay attributes.
6579   case ParsedAttr::AT_XRayInstrument:
6580     handleSimpleAttribute<XRayInstrumentAttr>(S, D, AL);
6581     break;
6582   case ParsedAttr::AT_XRayLogArgs:
6583     handleXRayLogArgsAttr(S, D, AL);
6584     break;
6585 
6586   // Move semantics attribute.
6587   case ParsedAttr::AT_Reinitializes:
6588     handleSimpleAttribute<ReinitializesAttr>(S, D, AL);
6589     break;
6590   }
6591 }
6592 
6593 /// ProcessDeclAttributeList - Apply all the decl attributes in the specified
6594 /// attribute list to the specified decl, ignoring any type attributes.
6595 void Sema::ProcessDeclAttributeList(Scope *S, Decl *D,
6596                                     const ParsedAttributesView &AttrList,
6597                                     bool IncludeCXX11Attributes) {
6598   if (AttrList.empty())
6599     return;
6600 
6601   for (const ParsedAttr &AL : AttrList)
6602     ProcessDeclAttribute(*this, S, D, AL, IncludeCXX11Attributes);
6603 
6604   // FIXME: We should be able to handle these cases in TableGen.
6605   // GCC accepts
6606   // static int a9 __attribute__((weakref));
6607   // but that looks really pointless. We reject it.
6608   if (D->hasAttr<WeakRefAttr>() && !D->hasAttr<AliasAttr>()) {
6609     Diag(AttrList.begin()->getLoc(), diag::err_attribute_weakref_without_alias)
6610         << cast<NamedDecl>(D);
6611     D->dropAttr<WeakRefAttr>();
6612     return;
6613   }
6614 
6615   // FIXME: We should be able to handle this in TableGen as well. It would be
6616   // good to have a way to specify "these attributes must appear as a group",
6617   // for these. Additionally, it would be good to have a way to specify "these
6618   // attribute must never appear as a group" for attributes like cold and hot.
6619   if (!D->hasAttr<OpenCLKernelAttr>()) {
6620     // These attributes cannot be applied to a non-kernel function.
6621     if (const auto *A = D->getAttr<ReqdWorkGroupSizeAttr>()) {
6622       // FIXME: This emits a different error message than
6623       // diag::err_attribute_wrong_decl_type + ExpectedKernelFunction.
6624       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6625       D->setInvalidDecl();
6626     } else if (const auto *A = D->getAttr<WorkGroupSizeHintAttr>()) {
6627       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6628       D->setInvalidDecl();
6629     } else if (const auto *A = D->getAttr<VecTypeHintAttr>()) {
6630       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6631       D->setInvalidDecl();
6632     } else if (const auto *A = D->getAttr<OpenCLIntelReqdSubGroupSizeAttr>()) {
6633       Diag(D->getLocation(), diag::err_opencl_kernel_attr) << A;
6634       D->setInvalidDecl();
6635     } else if (!D->hasAttr<CUDAGlobalAttr>()) {
6636       if (const auto *A = D->getAttr<AMDGPUFlatWorkGroupSizeAttr>()) {
6637         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6638             << A << ExpectedKernelFunction;
6639         D->setInvalidDecl();
6640       } else if (const auto *A = D->getAttr<AMDGPUWavesPerEUAttr>()) {
6641         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6642             << A << ExpectedKernelFunction;
6643         D->setInvalidDecl();
6644       } else if (const auto *A = D->getAttr<AMDGPUNumSGPRAttr>()) {
6645         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6646             << A << ExpectedKernelFunction;
6647         D->setInvalidDecl();
6648       } else if (const auto *A = D->getAttr<AMDGPUNumVGPRAttr>()) {
6649         Diag(D->getLocation(), diag::err_attribute_wrong_decl_type)
6650             << A << ExpectedKernelFunction;
6651         D->setInvalidDecl();
6652       }
6653     }
6654   }
6655 }
6656 
6657 // Helper for delayed processing TransparentUnion attribute.
6658 void Sema::ProcessDeclAttributeDelayed(Decl *D,
6659                                        const ParsedAttributesView &AttrList) {
6660   for (const ParsedAttr &AL : AttrList)
6661     if (AL.getKind() == ParsedAttr::AT_TransparentUnion) {
6662       handleTransparentUnionAttr(*this, D, AL);
6663       break;
6664     }
6665 }
6666 
6667 // Annotation attributes are the only attributes allowed after an access
6668 // specifier.
6669 bool Sema::ProcessAccessDeclAttributeList(
6670     AccessSpecDecl *ASDecl, const ParsedAttributesView &AttrList) {
6671   for (const ParsedAttr &AL : AttrList) {
6672     if (AL.getKind() == ParsedAttr::AT_Annotate) {
6673       ProcessDeclAttribute(*this, nullptr, ASDecl, AL, AL.isCXX11Attribute());
6674     } else {
6675       Diag(AL.getLoc(), diag::err_only_annotate_after_access_spec);
6676       return true;
6677     }
6678   }
6679   return false;
6680 }
6681 
6682 /// checkUnusedDeclAttributes - Check a list of attributes to see if it
6683 /// contains any decl attributes that we should warn about.
6684 static void checkUnusedDeclAttributes(Sema &S, const ParsedAttributesView &A) {
6685   for (const ParsedAttr &AL : A) {
6686     // Only warn if the attribute is an unignored, non-type attribute.
6687     if (AL.isUsedAsTypeAttr() || AL.isInvalid())
6688       continue;
6689     if (AL.getKind() == ParsedAttr::IgnoredAttribute)
6690       continue;
6691 
6692     if (AL.getKind() == ParsedAttr::UnknownAttribute) {
6693       S.Diag(AL.getLoc(), diag::warn_unknown_attribute_ignored)
6694           << AL << AL.getRange();
6695     } else {
6696       S.Diag(AL.getLoc(), diag::warn_attribute_not_on_decl) << AL
6697                                                             << AL.getRange();
6698     }
6699   }
6700 }
6701 
6702 /// checkUnusedDeclAttributes - Given a declarator which is not being
6703 /// used to build a declaration, complain about any decl attributes
6704 /// which might be lying around on it.
6705 void Sema::checkUnusedDeclAttributes(Declarator &D) {
6706   ::checkUnusedDeclAttributes(*this, D.getDeclSpec().getAttributes());
6707   ::checkUnusedDeclAttributes(*this, D.getAttributes());
6708   for (unsigned i = 0, e = D.getNumTypeObjects(); i != e; ++i)
6709     ::checkUnusedDeclAttributes(*this, D.getTypeObject(i).getAttrs());
6710 }
6711 
6712 /// DeclClonePragmaWeak - clone existing decl (maybe definition),
6713 /// \#pragma weak needs a non-definition decl and source may not have one.
6714 NamedDecl * Sema::DeclClonePragmaWeak(NamedDecl *ND, IdentifierInfo *II,
6715                                       SourceLocation Loc) {
6716   assert(isa<FunctionDecl>(ND) || isa<VarDecl>(ND));
6717   NamedDecl *NewD = nullptr;
6718   if (auto *FD = dyn_cast<FunctionDecl>(ND)) {
6719     FunctionDecl *NewFD;
6720     // FIXME: Missing call to CheckFunctionDeclaration().
6721     // FIXME: Mangling?
6722     // FIXME: Is the qualifier info correct?
6723     // FIXME: Is the DeclContext correct?
6724     NewFD = FunctionDecl::Create(FD->getASTContext(), FD->getDeclContext(),
6725                                  Loc, Loc, DeclarationName(II),
6726                                  FD->getType(), FD->getTypeSourceInfo(),
6727                                  SC_None, false/*isInlineSpecified*/,
6728                                  FD->hasPrototype(),
6729                                  false/*isConstexprSpecified*/);
6730     NewD = NewFD;
6731 
6732     if (FD->getQualifier())
6733       NewFD->setQualifierInfo(FD->getQualifierLoc());
6734 
6735     // Fake up parameter variables; they are declared as if this were
6736     // a typedef.
6737     QualType FDTy = FD->getType();
6738     if (const auto *FT = FDTy->getAs<FunctionProtoType>()) {
6739       SmallVector<ParmVarDecl*, 16> Params;
6740       for (const auto &AI : FT->param_types()) {
6741         ParmVarDecl *Param = BuildParmVarDeclForTypedef(NewFD, Loc, AI);
6742         Param->setScopeInfo(0, Params.size());
6743         Params.push_back(Param);
6744       }
6745       NewFD->setParams(Params);
6746     }
6747   } else if (auto *VD = dyn_cast<VarDecl>(ND)) {
6748     NewD = VarDecl::Create(VD->getASTContext(), VD->getDeclContext(),
6749                            VD->getInnerLocStart(), VD->getLocation(), II,
6750                            VD->getType(), VD->getTypeSourceInfo(),
6751                            VD->getStorageClass());
6752     if (VD->getQualifier())
6753       cast<VarDecl>(NewD)->setQualifierInfo(VD->getQualifierLoc());
6754   }
6755   return NewD;
6756 }
6757 
6758 /// DeclApplyPragmaWeak - A declaration (maybe definition) needs \#pragma weak
6759 /// applied to it, possibly with an alias.
6760 void Sema::DeclApplyPragmaWeak(Scope *S, NamedDecl *ND, WeakInfo &W) {
6761   if (W.getUsed()) return; // only do this once
6762   W.setUsed(true);
6763   if (W.getAlias()) { // clone decl, impersonate __attribute(weak,alias(...))
6764     IdentifierInfo *NDId = ND->getIdentifier();
6765     NamedDecl *NewD = DeclClonePragmaWeak(ND, W.getAlias(), W.getLocation());
6766     NewD->addAttr(AliasAttr::CreateImplicit(Context, NDId->getName(),
6767                                             W.getLocation()));
6768     NewD->addAttr(WeakAttr::CreateImplicit(Context, W.getLocation()));
6769     WeakTopLevelDecl.push_back(NewD);
6770     // FIXME: "hideous" code from Sema::LazilyCreateBuiltin
6771     // to insert Decl at TU scope, sorry.
6772     DeclContext *SavedContext = CurContext;
6773     CurContext = Context.getTranslationUnitDecl();
6774     NewD->setDeclContext(CurContext);
6775     NewD->setLexicalDeclContext(CurContext);
6776     PushOnScopeChains(NewD, S);
6777     CurContext = SavedContext;
6778   } else { // just add weak to existing
6779     ND->addAttr(WeakAttr::CreateImplicit(Context, W.getLocation()));
6780   }
6781 }
6782 
6783 void Sema::ProcessPragmaWeak(Scope *S, Decl *D) {
6784   // It's valid to "forward-declare" #pragma weak, in which case we
6785   // have to do this.
6786   LoadExternalWeakUndeclaredIdentifiers();
6787   if (!WeakUndeclaredIdentifiers.empty()) {
6788     NamedDecl *ND = nullptr;
6789     if (auto *VD = dyn_cast<VarDecl>(D))
6790       if (VD->isExternC())
6791         ND = VD;
6792     if (auto *FD = dyn_cast<FunctionDecl>(D))
6793       if (FD->isExternC())
6794         ND = FD;
6795     if (ND) {
6796       if (IdentifierInfo *Id = ND->getIdentifier()) {
6797         auto I = WeakUndeclaredIdentifiers.find(Id);
6798         if (I != WeakUndeclaredIdentifiers.end()) {
6799           WeakInfo W = I->second;
6800           DeclApplyPragmaWeak(S, ND, W);
6801           WeakUndeclaredIdentifiers[Id] = W;
6802         }
6803       }
6804     }
6805   }
6806 }
6807 
6808 /// ProcessDeclAttributes - Given a declarator (PD) with attributes indicated in
6809 /// it, apply them to D.  This is a bit tricky because PD can have attributes
6810 /// specified in many different places, and we need to find and apply them all.
6811 void Sema::ProcessDeclAttributes(Scope *S, Decl *D, const Declarator &PD) {
6812   // Apply decl attributes from the DeclSpec if present.
6813   if (!PD.getDeclSpec().getAttributes().empty())
6814     ProcessDeclAttributeList(S, D, PD.getDeclSpec().getAttributes());
6815 
6816   // Walk the declarator structure, applying decl attributes that were in a type
6817   // position to the decl itself.  This handles cases like:
6818   //   int *__attr__(x)** D;
6819   // when X is a decl attribute.
6820   for (unsigned i = 0, e = PD.getNumTypeObjects(); i != e; ++i)
6821     ProcessDeclAttributeList(S, D, PD.getTypeObject(i).getAttrs(),
6822                              /*IncludeCXX11Attributes=*/false);
6823 
6824   // Finally, apply any attributes on the decl itself.
6825   ProcessDeclAttributeList(S, D, PD.getAttributes());
6826 
6827   // Apply additional attributes specified by '#pragma clang attribute'.
6828   AddPragmaAttributes(S, D);
6829 }
6830 
6831 /// Is the given declaration allowed to use a forbidden type?
6832 /// If so, it'll still be annotated with an attribute that makes it
6833 /// illegal to actually use.
6834 static bool isForbiddenTypeAllowed(Sema &S, Decl *D,
6835                                    const DelayedDiagnostic &diag,
6836                                    UnavailableAttr::ImplicitReason &reason) {
6837   // Private ivars are always okay.  Unfortunately, people don't
6838   // always properly make their ivars private, even in system headers.
6839   // Plus we need to make fields okay, too.
6840   if (!isa<FieldDecl>(D) && !isa<ObjCPropertyDecl>(D) &&
6841       !isa<FunctionDecl>(D))
6842     return false;
6843 
6844   // Silently accept unsupported uses of __weak in both user and system
6845   // declarations when it's been disabled, for ease of integration with
6846   // -fno-objc-arc files.  We do have to take some care against attempts
6847   // to define such things;  for now, we've only done that for ivars
6848   // and properties.
6849   if ((isa<ObjCIvarDecl>(D) || isa<ObjCPropertyDecl>(D))) {
6850     if (diag.getForbiddenTypeDiagnostic() == diag::err_arc_weak_disabled ||
6851         diag.getForbiddenTypeDiagnostic() == diag::err_arc_weak_no_runtime) {
6852       reason = UnavailableAttr::IR_ForbiddenWeak;
6853       return true;
6854     }
6855   }
6856 
6857   // Allow all sorts of things in system headers.
6858   if (S.Context.getSourceManager().isInSystemHeader(D->getLocation())) {
6859     // Currently, all the failures dealt with this way are due to ARC
6860     // restrictions.
6861     reason = UnavailableAttr::IR_ARCForbiddenType;
6862     return true;
6863   }
6864 
6865   return false;
6866 }
6867 
6868 /// Handle a delayed forbidden-type diagnostic.
6869 static void handleDelayedForbiddenType(Sema &S, DelayedDiagnostic &DD,
6870                                        Decl *D) {
6871   auto Reason = UnavailableAttr::IR_None;
6872   if (D && isForbiddenTypeAllowed(S, D, DD, Reason)) {
6873     assert(Reason && "didn't set reason?");
6874     D->addAttr(UnavailableAttr::CreateImplicit(S.Context, "", Reason, DD.Loc));
6875     return;
6876   }
6877   if (S.getLangOpts().ObjCAutoRefCount)
6878     if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
6879       // FIXME: we may want to suppress diagnostics for all
6880       // kind of forbidden type messages on unavailable functions.
6881       if (FD->hasAttr<UnavailableAttr>() &&
6882           DD.getForbiddenTypeDiagnostic() ==
6883               diag::err_arc_array_param_no_ownership) {
6884         DD.Triggered = true;
6885         return;
6886       }
6887     }
6888 
6889   S.Diag(DD.Loc, DD.getForbiddenTypeDiagnostic())
6890       << DD.getForbiddenTypeOperand() << DD.getForbiddenTypeArgument();
6891   DD.Triggered = true;
6892 }
6893 
6894 static const AvailabilityAttr *getAttrForPlatform(ASTContext &Context,
6895                                                   const Decl *D) {
6896   // Check each AvailabilityAttr to find the one for this platform.
6897   for (const auto *A : D->attrs()) {
6898     if (const auto *Avail = dyn_cast<AvailabilityAttr>(A)) {
6899       // FIXME: this is copied from CheckAvailability. We should try to
6900       // de-duplicate.
6901 
6902       // Check if this is an App Extension "platform", and if so chop off
6903       // the suffix for matching with the actual platform.
6904       StringRef ActualPlatform = Avail->getPlatform()->getName();
6905       StringRef RealizedPlatform = ActualPlatform;
6906       if (Context.getLangOpts().AppExt) {
6907         size_t suffix = RealizedPlatform.rfind("_app_extension");
6908         if (suffix != StringRef::npos)
6909           RealizedPlatform = RealizedPlatform.slice(0, suffix);
6910       }
6911 
6912       StringRef TargetPlatform = Context.getTargetInfo().getPlatformName();
6913 
6914       // Match the platform name.
6915       if (RealizedPlatform == TargetPlatform)
6916         return Avail;
6917     }
6918   }
6919   return nullptr;
6920 }
6921 
6922 /// The diagnostic we should emit for \c D, and the declaration that
6923 /// originated it, or \c AR_Available.
6924 ///
6925 /// \param D The declaration to check.
6926 /// \param Message If non-null, this will be populated with the message from
6927 /// the availability attribute that is selected.
6928 static std::pair<AvailabilityResult, const NamedDecl *>
6929 ShouldDiagnoseAvailabilityOfDecl(const NamedDecl *D, std::string *Message) {
6930   AvailabilityResult Result = D->getAvailability(Message);
6931 
6932   // For typedefs, if the typedef declaration appears available look
6933   // to the underlying type to see if it is more restrictive.
6934   while (const auto *TD = dyn_cast<TypedefNameDecl>(D)) {
6935     if (Result == AR_Available) {
6936       if (const auto *TT = TD->getUnderlyingType()->getAs<TagType>()) {
6937         D = TT->getDecl();
6938         Result = D->getAvailability(Message);
6939         continue;
6940       }
6941     }
6942     break;
6943   }
6944 
6945   // Forward class declarations get their attributes from their definition.
6946   if (const auto *IDecl = dyn_cast<ObjCInterfaceDecl>(D)) {
6947     if (IDecl->getDefinition()) {
6948       D = IDecl->getDefinition();
6949       Result = D->getAvailability(Message);
6950     }
6951   }
6952 
6953   if (const auto *ECD = dyn_cast<EnumConstantDecl>(D))
6954     if (Result == AR_Available) {
6955       const DeclContext *DC = ECD->getDeclContext();
6956       if (const auto *TheEnumDecl = dyn_cast<EnumDecl>(DC)) {
6957         Result = TheEnumDecl->getAvailability(Message);
6958         D = TheEnumDecl;
6959       }
6960     }
6961 
6962   return {Result, D};
6963 }
6964 
6965 
6966 /// whether we should emit a diagnostic for \c K and \c DeclVersion in
6967 /// the context of \c Ctx. For example, we should emit an unavailable diagnostic
6968 /// in a deprecated context, but not the other way around.
6969 static bool ShouldDiagnoseAvailabilityInContext(Sema &S, AvailabilityResult K,
6970                                                 VersionTuple DeclVersion,
6971                                                 Decl *Ctx) {
6972   assert(K != AR_Available && "Expected an unavailable declaration here!");
6973 
6974   // Checks if we should emit the availability diagnostic in the context of C.
6975   auto CheckContext = [&](const Decl *C) {
6976     if (K == AR_NotYetIntroduced) {
6977       if (const AvailabilityAttr *AA = getAttrForPlatform(S.Context, C))
6978         if (AA->getIntroduced() >= DeclVersion)
6979           return true;
6980     } else if (K == AR_Deprecated)
6981       if (C->isDeprecated())
6982         return true;
6983 
6984     if (C->isUnavailable())
6985       return true;
6986     return false;
6987   };
6988 
6989   do {
6990     if (CheckContext(Ctx))
6991       return false;
6992 
6993     // An implementation implicitly has the availability of the interface.
6994     // Unless it is "+load" method.
6995     if (const auto *MethodD = dyn_cast<ObjCMethodDecl>(Ctx))
6996       if (MethodD->isClassMethod() &&
6997           MethodD->getSelector().getAsString() == "load")
6998         return true;
6999 
7000     if (const auto *CatOrImpl = dyn_cast<ObjCImplDecl>(Ctx)) {
7001       if (const ObjCInterfaceDecl *Interface = CatOrImpl->getClassInterface())
7002         if (CheckContext(Interface))
7003           return false;
7004     }
7005     // A category implicitly has the availability of the interface.
7006     else if (const auto *CatD = dyn_cast<ObjCCategoryDecl>(Ctx))
7007       if (const ObjCInterfaceDecl *Interface = CatD->getClassInterface())
7008         if (CheckContext(Interface))
7009           return false;
7010   } while ((Ctx = cast_or_null<Decl>(Ctx->getDeclContext())));
7011 
7012   return true;
7013 }
7014 
7015 static bool
7016 shouldDiagnoseAvailabilityByDefault(const ASTContext &Context,
7017                                     const VersionTuple &DeploymentVersion,
7018                                     const VersionTuple &DeclVersion) {
7019   const auto &Triple = Context.getTargetInfo().getTriple();
7020   VersionTuple ForceAvailabilityFromVersion;
7021   switch (Triple.getOS()) {
7022   case llvm::Triple::IOS:
7023   case llvm::Triple::TvOS:
7024     ForceAvailabilityFromVersion = VersionTuple(/*Major=*/11);
7025     break;
7026   case llvm::Triple::WatchOS:
7027     ForceAvailabilityFromVersion = VersionTuple(/*Major=*/4);
7028     break;
7029   case llvm::Triple::Darwin:
7030   case llvm::Triple::MacOSX:
7031     ForceAvailabilityFromVersion = VersionTuple(/*Major=*/10, /*Minor=*/13);
7032     break;
7033   default:
7034     // New targets should always warn about availability.
7035     return Triple.getVendor() == llvm::Triple::Apple;
7036   }
7037   return DeploymentVersion >= ForceAvailabilityFromVersion ||
7038          DeclVersion >= ForceAvailabilityFromVersion;
7039 }
7040 
7041 static NamedDecl *findEnclosingDeclToAnnotate(Decl *OrigCtx) {
7042   for (Decl *Ctx = OrigCtx; Ctx;
7043        Ctx = cast_or_null<Decl>(Ctx->getDeclContext())) {
7044     if (isa<TagDecl>(Ctx) || isa<FunctionDecl>(Ctx) || isa<ObjCMethodDecl>(Ctx))
7045       return cast<NamedDecl>(Ctx);
7046     if (auto *CD = dyn_cast<ObjCContainerDecl>(Ctx)) {
7047       if (auto *Imp = dyn_cast<ObjCImplDecl>(Ctx))
7048         return Imp->getClassInterface();
7049       return CD;
7050     }
7051   }
7052 
7053   return dyn_cast<NamedDecl>(OrigCtx);
7054 }
7055 
7056 namespace {
7057 
7058 struct AttributeInsertion {
7059   StringRef Prefix;
7060   SourceLocation Loc;
7061   StringRef Suffix;
7062 
7063   static AttributeInsertion createInsertionAfter(const NamedDecl *D) {
7064     return {" ", D->getEndLoc(), ""};
7065   }
7066   static AttributeInsertion createInsertionAfter(SourceLocation Loc) {
7067     return {" ", Loc, ""};
7068   }
7069   static AttributeInsertion createInsertionBefore(const NamedDecl *D) {
7070     return {"", D->getBeginLoc(), "\n"};
7071   }
7072 };
7073 
7074 } // end anonymous namespace
7075 
7076 /// Tries to parse a string as ObjC method name.
7077 ///
7078 /// \param Name The string to parse. Expected to originate from availability
7079 /// attribute argument.
7080 /// \param SlotNames The vector that will be populated with slot names. In case
7081 /// of unsuccessful parsing can contain invalid data.
7082 /// \returns A number of method parameters if parsing was successful, None
7083 /// otherwise.
7084 static Optional<unsigned>
7085 tryParseObjCMethodName(StringRef Name, SmallVectorImpl<StringRef> &SlotNames,
7086                        const LangOptions &LangOpts) {
7087   // Accept replacements starting with - or + as valid ObjC method names.
7088   if (!Name.empty() && (Name.front() == '-' || Name.front() == '+'))
7089     Name = Name.drop_front(1);
7090   if (Name.empty())
7091     return None;
7092   Name.split(SlotNames, ':');
7093   unsigned NumParams;
7094   if (Name.back() == ':') {
7095     // Remove an empty string at the end that doesn't represent any slot.
7096     SlotNames.pop_back();
7097     NumParams = SlotNames.size();
7098   } else {
7099     if (SlotNames.size() != 1)
7100       // Not a valid method name, just a colon-separated string.
7101       return None;
7102     NumParams = 0;
7103   }
7104   // Verify all slot names are valid.
7105   bool AllowDollar = LangOpts.DollarIdents;
7106   for (StringRef S : SlotNames) {
7107     if (S.empty())
7108       continue;
7109     if (!isValidIdentifier(S, AllowDollar))
7110       return None;
7111   }
7112   return NumParams;
7113 }
7114 
7115 /// Returns a source location in which it's appropriate to insert a new
7116 /// attribute for the given declaration \D.
7117 static Optional<AttributeInsertion>
7118 createAttributeInsertion(const NamedDecl *D, const SourceManager &SM,
7119                          const LangOptions &LangOpts) {
7120   if (isa<ObjCPropertyDecl>(D))
7121     return AttributeInsertion::createInsertionAfter(D);
7122   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D)) {
7123     if (MD->hasBody())
7124       return None;
7125     return AttributeInsertion::createInsertionAfter(D);
7126   }
7127   if (const auto *TD = dyn_cast<TagDecl>(D)) {
7128     SourceLocation Loc =
7129         Lexer::getLocForEndOfToken(TD->getInnerLocStart(), 0, SM, LangOpts);
7130     if (Loc.isInvalid())
7131       return None;
7132     // Insert after the 'struct'/whatever keyword.
7133     return AttributeInsertion::createInsertionAfter(Loc);
7134   }
7135   return AttributeInsertion::createInsertionBefore(D);
7136 }
7137 
7138 /// Actually emit an availability diagnostic for a reference to an unavailable
7139 /// decl.
7140 ///
7141 /// \param Ctx The context that the reference occurred in
7142 /// \param ReferringDecl The exact declaration that was referenced.
7143 /// \param OffendingDecl A related decl to \c ReferringDecl that has an
7144 /// availability attribute corresponding to \c K attached to it. Note that this
7145 /// may not be the same as ReferringDecl, i.e. if an EnumDecl is annotated and
7146 /// we refer to a member EnumConstantDecl, ReferringDecl is the EnumConstantDecl
7147 /// and OffendingDecl is the EnumDecl.
7148 static void DoEmitAvailabilityWarning(Sema &S, AvailabilityResult K,
7149                                       Decl *Ctx, const NamedDecl *ReferringDecl,
7150                                       const NamedDecl *OffendingDecl,
7151                                       StringRef Message,
7152                                       ArrayRef<SourceLocation> Locs,
7153                                       const ObjCInterfaceDecl *UnknownObjCClass,
7154                                       const ObjCPropertyDecl *ObjCProperty,
7155                                       bool ObjCPropertyAccess) {
7156   // Diagnostics for deprecated or unavailable.
7157   unsigned diag, diag_message, diag_fwdclass_message;
7158   unsigned diag_available_here = diag::note_availability_specified_here;
7159   SourceLocation NoteLocation = OffendingDecl->getLocation();
7160 
7161   // Matches 'diag::note_property_attribute' options.
7162   unsigned property_note_select;
7163 
7164   // Matches diag::note_availability_specified_here.
7165   unsigned available_here_select_kind;
7166 
7167   VersionTuple DeclVersion;
7168   if (const AvailabilityAttr *AA = getAttrForPlatform(S.Context, OffendingDecl))
7169     DeclVersion = AA->getIntroduced();
7170 
7171   if (!ShouldDiagnoseAvailabilityInContext(S, K, DeclVersion, Ctx))
7172     return;
7173 
7174   SourceLocation Loc = Locs.front();
7175 
7176   // The declaration can have multiple availability attributes, we are looking
7177   // at one of them.
7178   const AvailabilityAttr *A = getAttrForPlatform(S.Context, OffendingDecl);
7179   if (A && A->isInherited()) {
7180     for (const Decl *Redecl = OffendingDecl->getMostRecentDecl(); Redecl;
7181          Redecl = Redecl->getPreviousDecl()) {
7182       const AvailabilityAttr *AForRedecl =
7183           getAttrForPlatform(S.Context, Redecl);
7184       if (AForRedecl && !AForRedecl->isInherited()) {
7185         // If D is a declaration with inherited attributes, the note should
7186         // point to the declaration with actual attributes.
7187         NoteLocation = Redecl->getLocation();
7188         break;
7189       }
7190     }
7191   }
7192 
7193   switch (K) {
7194   case AR_NotYetIntroduced: {
7195     // We would like to emit the diagnostic even if -Wunguarded-availability is
7196     // not specified for deployment targets >= to iOS 11 or equivalent or
7197     // for declarations that were introduced in iOS 11 (macOS 10.13, ...) or
7198     // later.
7199     const AvailabilityAttr *AA =
7200         getAttrForPlatform(S.getASTContext(), OffendingDecl);
7201     VersionTuple Introduced = AA->getIntroduced();
7202 
7203     bool UseNewWarning = shouldDiagnoseAvailabilityByDefault(
7204         S.Context, S.Context.getTargetInfo().getPlatformMinVersion(),
7205         Introduced);
7206     unsigned Warning = UseNewWarning ? diag::warn_unguarded_availability_new
7207                                      : diag::warn_unguarded_availability;
7208 
7209     S.Diag(Loc, Warning)
7210         << OffendingDecl
7211         << AvailabilityAttr::getPrettyPlatformName(
7212                S.getASTContext().getTargetInfo().getPlatformName())
7213         << Introduced.getAsString();
7214 
7215     S.Diag(OffendingDecl->getLocation(), diag::note_availability_specified_here)
7216         << OffendingDecl << /* partial */ 3;
7217 
7218     if (const auto *Enclosing = findEnclosingDeclToAnnotate(Ctx)) {
7219       if (const auto *TD = dyn_cast<TagDecl>(Enclosing))
7220         if (TD->getDeclName().isEmpty()) {
7221           S.Diag(TD->getLocation(),
7222                  diag::note_decl_unguarded_availability_silence)
7223               << /*Anonymous*/ 1 << TD->getKindName();
7224           return;
7225         }
7226       auto FixitNoteDiag =
7227           S.Diag(Enclosing->getLocation(),
7228                  diag::note_decl_unguarded_availability_silence)
7229           << /*Named*/ 0 << Enclosing;
7230       // Don't offer a fixit for declarations with availability attributes.
7231       if (Enclosing->hasAttr<AvailabilityAttr>())
7232         return;
7233       if (!S.getPreprocessor().isMacroDefined("API_AVAILABLE"))
7234         return;
7235       Optional<AttributeInsertion> Insertion = createAttributeInsertion(
7236           Enclosing, S.getSourceManager(), S.getLangOpts());
7237       if (!Insertion)
7238         return;
7239       std::string PlatformName =
7240           AvailabilityAttr::getPlatformNameSourceSpelling(
7241               S.getASTContext().getTargetInfo().getPlatformName())
7242               .lower();
7243       std::string Introduced =
7244           OffendingDecl->getVersionIntroduced().getAsString();
7245       FixitNoteDiag << FixItHint::CreateInsertion(
7246           Insertion->Loc,
7247           (llvm::Twine(Insertion->Prefix) + "API_AVAILABLE(" + PlatformName +
7248            "(" + Introduced + "))" + Insertion->Suffix)
7249               .str());
7250     }
7251     return;
7252   }
7253   case AR_Deprecated:
7254     diag = !ObjCPropertyAccess ? diag::warn_deprecated
7255                                : diag::warn_property_method_deprecated;
7256     diag_message = diag::warn_deprecated_message;
7257     diag_fwdclass_message = diag::warn_deprecated_fwdclass_message;
7258     property_note_select = /* deprecated */ 0;
7259     available_here_select_kind = /* deprecated */ 2;
7260     if (const auto *AL = OffendingDecl->getAttr<DeprecatedAttr>())
7261       NoteLocation = AL->getLocation();
7262     break;
7263 
7264   case AR_Unavailable:
7265     diag = !ObjCPropertyAccess ? diag::err_unavailable
7266                                : diag::err_property_method_unavailable;
7267     diag_message = diag::err_unavailable_message;
7268     diag_fwdclass_message = diag::warn_unavailable_fwdclass_message;
7269     property_note_select = /* unavailable */ 1;
7270     available_here_select_kind = /* unavailable */ 0;
7271 
7272     if (auto AL = OffendingDecl->getAttr<UnavailableAttr>()) {
7273       if (AL->isImplicit() && AL->getImplicitReason()) {
7274         // Most of these failures are due to extra restrictions in ARC;
7275         // reflect that in the primary diagnostic when applicable.
7276         auto flagARCError = [&] {
7277           if (S.getLangOpts().ObjCAutoRefCount &&
7278               S.getSourceManager().isInSystemHeader(
7279                   OffendingDecl->getLocation()))
7280             diag = diag::err_unavailable_in_arc;
7281         };
7282 
7283         switch (AL->getImplicitReason()) {
7284         case UnavailableAttr::IR_None: break;
7285 
7286         case UnavailableAttr::IR_ARCForbiddenType:
7287           flagARCError();
7288           diag_available_here = diag::note_arc_forbidden_type;
7289           break;
7290 
7291         case UnavailableAttr::IR_ForbiddenWeak:
7292           if (S.getLangOpts().ObjCWeakRuntime)
7293             diag_available_here = diag::note_arc_weak_disabled;
7294           else
7295             diag_available_here = diag::note_arc_weak_no_runtime;
7296           break;
7297 
7298         case UnavailableAttr::IR_ARCForbiddenConversion:
7299           flagARCError();
7300           diag_available_here = diag::note_performs_forbidden_arc_conversion;
7301           break;
7302 
7303         case UnavailableAttr::IR_ARCInitReturnsUnrelated:
7304           flagARCError();
7305           diag_available_here = diag::note_arc_init_returns_unrelated;
7306           break;
7307 
7308         case UnavailableAttr::IR_ARCFieldWithOwnership:
7309           flagARCError();
7310           diag_available_here = diag::note_arc_field_with_ownership;
7311           break;
7312         }
7313       }
7314     }
7315     break;
7316 
7317   case AR_Available:
7318     llvm_unreachable("Warning for availability of available declaration?");
7319   }
7320 
7321   SmallVector<FixItHint, 12> FixIts;
7322   if (K == AR_Deprecated) {
7323     StringRef Replacement;
7324     if (auto AL = OffendingDecl->getAttr<DeprecatedAttr>())
7325       Replacement = AL->getReplacement();
7326     if (auto AL = getAttrForPlatform(S.Context, OffendingDecl))
7327       Replacement = AL->getReplacement();
7328 
7329     CharSourceRange UseRange;
7330     if (!Replacement.empty())
7331       UseRange =
7332           CharSourceRange::getCharRange(Loc, S.getLocForEndOfToken(Loc));
7333     if (UseRange.isValid()) {
7334       if (const auto *MethodDecl = dyn_cast<ObjCMethodDecl>(ReferringDecl)) {
7335         Selector Sel = MethodDecl->getSelector();
7336         SmallVector<StringRef, 12> SelectorSlotNames;
7337         Optional<unsigned> NumParams = tryParseObjCMethodName(
7338             Replacement, SelectorSlotNames, S.getLangOpts());
7339         if (NumParams && NumParams.getValue() == Sel.getNumArgs()) {
7340           assert(SelectorSlotNames.size() == Locs.size());
7341           for (unsigned I = 0; I < Locs.size(); ++I) {
7342             if (!Sel.getNameForSlot(I).empty()) {
7343               CharSourceRange NameRange = CharSourceRange::getCharRange(
7344                   Locs[I], S.getLocForEndOfToken(Locs[I]));
7345               FixIts.push_back(FixItHint::CreateReplacement(
7346                   NameRange, SelectorSlotNames[I]));
7347             } else
7348               FixIts.push_back(
7349                   FixItHint::CreateInsertion(Locs[I], SelectorSlotNames[I]));
7350           }
7351         } else
7352           FixIts.push_back(FixItHint::CreateReplacement(UseRange, Replacement));
7353       } else
7354         FixIts.push_back(FixItHint::CreateReplacement(UseRange, Replacement));
7355     }
7356   }
7357 
7358   if (!Message.empty()) {
7359     S.Diag(Loc, diag_message) << ReferringDecl << Message << FixIts;
7360     if (ObjCProperty)
7361       S.Diag(ObjCProperty->getLocation(), diag::note_property_attribute)
7362           << ObjCProperty->getDeclName() << property_note_select;
7363   } else if (!UnknownObjCClass) {
7364     S.Diag(Loc, diag) << ReferringDecl << FixIts;
7365     if (ObjCProperty)
7366       S.Diag(ObjCProperty->getLocation(), diag::note_property_attribute)
7367           << ObjCProperty->getDeclName() << property_note_select;
7368   } else {
7369     S.Diag(Loc, diag_fwdclass_message) << ReferringDecl << FixIts;
7370     S.Diag(UnknownObjCClass->getLocation(), diag::note_forward_class);
7371   }
7372 
7373   S.Diag(NoteLocation, diag_available_here)
7374     << OffendingDecl << available_here_select_kind;
7375 }
7376 
7377 static void handleDelayedAvailabilityCheck(Sema &S, DelayedDiagnostic &DD,
7378                                            Decl *Ctx) {
7379   assert(DD.Kind == DelayedDiagnostic::Availability &&
7380          "Expected an availability diagnostic here");
7381 
7382   DD.Triggered = true;
7383   DoEmitAvailabilityWarning(
7384       S, DD.getAvailabilityResult(), Ctx, DD.getAvailabilityReferringDecl(),
7385       DD.getAvailabilityOffendingDecl(), DD.getAvailabilityMessage(),
7386       DD.getAvailabilitySelectorLocs(), DD.getUnknownObjCClass(),
7387       DD.getObjCProperty(), false);
7388 }
7389 
7390 void Sema::PopParsingDeclaration(ParsingDeclState state, Decl *decl) {
7391   assert(DelayedDiagnostics.getCurrentPool());
7392   DelayedDiagnosticPool &poppedPool = *DelayedDiagnostics.getCurrentPool();
7393   DelayedDiagnostics.popWithoutEmitting(state);
7394 
7395   // When delaying diagnostics to run in the context of a parsed
7396   // declaration, we only want to actually emit anything if parsing
7397   // succeeds.
7398   if (!decl) return;
7399 
7400   // We emit all the active diagnostics in this pool or any of its
7401   // parents.  In general, we'll get one pool for the decl spec
7402   // and a child pool for each declarator; in a decl group like:
7403   //   deprecated_typedef foo, *bar, baz();
7404   // only the declarator pops will be passed decls.  This is correct;
7405   // we really do need to consider delayed diagnostics from the decl spec
7406   // for each of the different declarations.
7407   const DelayedDiagnosticPool *pool = &poppedPool;
7408   do {
7409     for (DelayedDiagnosticPool::pool_iterator
7410            i = pool->pool_begin(), e = pool->pool_end(); i != e; ++i) {
7411       // This const_cast is a bit lame.  Really, Triggered should be mutable.
7412       DelayedDiagnostic &diag = const_cast<DelayedDiagnostic&>(*i);
7413       if (diag.Triggered)
7414         continue;
7415 
7416       switch (diag.Kind) {
7417       case DelayedDiagnostic::Availability:
7418         // Don't bother giving deprecation/unavailable diagnostics if
7419         // the decl is invalid.
7420         if (!decl->isInvalidDecl())
7421           handleDelayedAvailabilityCheck(*this, diag, decl);
7422         break;
7423 
7424       case DelayedDiagnostic::Access:
7425         HandleDelayedAccessCheck(diag, decl);
7426         break;
7427 
7428       case DelayedDiagnostic::ForbiddenType:
7429         handleDelayedForbiddenType(*this, diag, decl);
7430         break;
7431       }
7432     }
7433   } while ((pool = pool->getParent()));
7434 }
7435 
7436 /// Given a set of delayed diagnostics, re-emit them as if they had
7437 /// been delayed in the current context instead of in the given pool.
7438 /// Essentially, this just moves them to the current pool.
7439 void Sema::redelayDiagnostics(DelayedDiagnosticPool &pool) {
7440   DelayedDiagnosticPool *curPool = DelayedDiagnostics.getCurrentPool();
7441   assert(curPool && "re-emitting in undelayed context not supported");
7442   curPool->steal(pool);
7443 }
7444 
7445 static void EmitAvailabilityWarning(Sema &S, AvailabilityResult AR,
7446                                     const NamedDecl *ReferringDecl,
7447                                     const NamedDecl *OffendingDecl,
7448                                     StringRef Message,
7449                                     ArrayRef<SourceLocation> Locs,
7450                                     const ObjCInterfaceDecl *UnknownObjCClass,
7451                                     const ObjCPropertyDecl *ObjCProperty,
7452                                     bool ObjCPropertyAccess) {
7453   // Delay if we're currently parsing a declaration.
7454   if (S.DelayedDiagnostics.shouldDelayDiagnostics()) {
7455     S.DelayedDiagnostics.add(
7456         DelayedDiagnostic::makeAvailability(
7457             AR, Locs, ReferringDecl, OffendingDecl, UnknownObjCClass,
7458             ObjCProperty, Message, ObjCPropertyAccess));
7459     return;
7460   }
7461 
7462   Decl *Ctx = cast<Decl>(S.getCurLexicalContext());
7463   DoEmitAvailabilityWarning(S, AR, Ctx, ReferringDecl, OffendingDecl,
7464                             Message, Locs, UnknownObjCClass, ObjCProperty,
7465                             ObjCPropertyAccess);
7466 }
7467 
7468 namespace {
7469 
7470 /// Returns true if the given statement can be a body-like child of \p Parent.
7471 bool isBodyLikeChildStmt(const Stmt *S, const Stmt *Parent) {
7472   switch (Parent->getStmtClass()) {
7473   case Stmt::IfStmtClass:
7474     return cast<IfStmt>(Parent)->getThen() == S ||
7475            cast<IfStmt>(Parent)->getElse() == S;
7476   case Stmt::WhileStmtClass:
7477     return cast<WhileStmt>(Parent)->getBody() == S;
7478   case Stmt::DoStmtClass:
7479     return cast<DoStmt>(Parent)->getBody() == S;
7480   case Stmt::ForStmtClass:
7481     return cast<ForStmt>(Parent)->getBody() == S;
7482   case Stmt::CXXForRangeStmtClass:
7483     return cast<CXXForRangeStmt>(Parent)->getBody() == S;
7484   case Stmt::ObjCForCollectionStmtClass:
7485     return cast<ObjCForCollectionStmt>(Parent)->getBody() == S;
7486   case Stmt::CaseStmtClass:
7487   case Stmt::DefaultStmtClass:
7488     return cast<SwitchCase>(Parent)->getSubStmt() == S;
7489   default:
7490     return false;
7491   }
7492 }
7493 
7494 class StmtUSEFinder : public RecursiveASTVisitor<StmtUSEFinder> {
7495   const Stmt *Target;
7496 
7497 public:
7498   bool VisitStmt(Stmt *S) { return S != Target; }
7499 
7500   /// Returns true if the given statement is present in the given declaration.
7501   static bool isContained(const Stmt *Target, const Decl *D) {
7502     StmtUSEFinder Visitor;
7503     Visitor.Target = Target;
7504     return !Visitor.TraverseDecl(const_cast<Decl *>(D));
7505   }
7506 };
7507 
7508 /// Traverses the AST and finds the last statement that used a given
7509 /// declaration.
7510 class LastDeclUSEFinder : public RecursiveASTVisitor<LastDeclUSEFinder> {
7511   const Decl *D;
7512 
7513 public:
7514   bool VisitDeclRefExpr(DeclRefExpr *DRE) {
7515     if (DRE->getDecl() == D)
7516       return false;
7517     return true;
7518   }
7519 
7520   static const Stmt *findLastStmtThatUsesDecl(const Decl *D,
7521                                               const CompoundStmt *Scope) {
7522     LastDeclUSEFinder Visitor;
7523     Visitor.D = D;
7524     for (auto I = Scope->body_rbegin(), E = Scope->body_rend(); I != E; ++I) {
7525       const Stmt *S = *I;
7526       if (!Visitor.TraverseStmt(const_cast<Stmt *>(S)))
7527         return S;
7528     }
7529     return nullptr;
7530   }
7531 };
7532 
7533 /// This class implements -Wunguarded-availability.
7534 ///
7535 /// This is done with a traversal of the AST of a function that makes reference
7536 /// to a partially available declaration. Whenever we encounter an \c if of the
7537 /// form: \c if(@available(...)), we use the version from the condition to visit
7538 /// the then statement.
7539 class DiagnoseUnguardedAvailability
7540     : public RecursiveASTVisitor<DiagnoseUnguardedAvailability> {
7541   typedef RecursiveASTVisitor<DiagnoseUnguardedAvailability> Base;
7542 
7543   Sema &SemaRef;
7544   Decl *Ctx;
7545 
7546   /// Stack of potentially nested 'if (@available(...))'s.
7547   SmallVector<VersionTuple, 8> AvailabilityStack;
7548   SmallVector<const Stmt *, 16> StmtStack;
7549 
7550   void DiagnoseDeclAvailability(NamedDecl *D, SourceRange Range);
7551 
7552 public:
7553   DiagnoseUnguardedAvailability(Sema &SemaRef, Decl *Ctx)
7554       : SemaRef(SemaRef), Ctx(Ctx) {
7555     AvailabilityStack.push_back(
7556         SemaRef.Context.getTargetInfo().getPlatformMinVersion());
7557   }
7558 
7559   bool TraverseDecl(Decl *D) {
7560     // Avoid visiting nested functions to prevent duplicate warnings.
7561     if (!D || isa<FunctionDecl>(D))
7562       return true;
7563     return Base::TraverseDecl(D);
7564   }
7565 
7566   bool TraverseStmt(Stmt *S) {
7567     if (!S)
7568       return true;
7569     StmtStack.push_back(S);
7570     bool Result = Base::TraverseStmt(S);
7571     StmtStack.pop_back();
7572     return Result;
7573   }
7574 
7575   void IssueDiagnostics(Stmt *S) { TraverseStmt(S); }
7576 
7577   bool TraverseIfStmt(IfStmt *If);
7578 
7579   bool TraverseLambdaExpr(LambdaExpr *E) { return true; }
7580 
7581   // for 'case X:' statements, don't bother looking at the 'X'; it can't lead
7582   // to any useful diagnostics.
7583   bool TraverseCaseStmt(CaseStmt *CS) { return TraverseStmt(CS->getSubStmt()); }
7584 
7585   bool VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *PRE) {
7586     if (PRE->isClassReceiver())
7587       DiagnoseDeclAvailability(PRE->getClassReceiver(), PRE->getReceiverLocation());
7588     return true;
7589   }
7590 
7591   bool VisitObjCMessageExpr(ObjCMessageExpr *Msg) {
7592     if (ObjCMethodDecl *D = Msg->getMethodDecl())
7593       DiagnoseDeclAvailability(
7594           D, SourceRange(Msg->getSelectorStartLoc(), Msg->getEndLoc()));
7595     return true;
7596   }
7597 
7598   bool VisitDeclRefExpr(DeclRefExpr *DRE) {
7599     DiagnoseDeclAvailability(DRE->getDecl(),
7600                              SourceRange(DRE->getBeginLoc(), DRE->getEndLoc()));
7601     return true;
7602   }
7603 
7604   bool VisitMemberExpr(MemberExpr *ME) {
7605     DiagnoseDeclAvailability(ME->getMemberDecl(),
7606                              SourceRange(ME->getBeginLoc(), ME->getEndLoc()));
7607     return true;
7608   }
7609 
7610   bool VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr *E) {
7611     SemaRef.Diag(E->getBeginLoc(), diag::warn_at_available_unchecked_use)
7612         << (!SemaRef.getLangOpts().ObjC1);
7613     return true;
7614   }
7615 
7616   bool VisitTypeLoc(TypeLoc Ty);
7617 };
7618 
7619 void DiagnoseUnguardedAvailability::DiagnoseDeclAvailability(
7620     NamedDecl *D, SourceRange Range) {
7621   AvailabilityResult Result;
7622   const NamedDecl *OffendingDecl;
7623   std::tie(Result, OffendingDecl) =
7624     ShouldDiagnoseAvailabilityOfDecl(D, nullptr);
7625   if (Result != AR_Available) {
7626     // All other diagnostic kinds have already been handled in
7627     // DiagnoseAvailabilityOfDecl.
7628     if (Result != AR_NotYetIntroduced)
7629       return;
7630 
7631     const AvailabilityAttr *AA =
7632       getAttrForPlatform(SemaRef.getASTContext(), OffendingDecl);
7633     VersionTuple Introduced = AA->getIntroduced();
7634 
7635     if (AvailabilityStack.back() >= Introduced)
7636       return;
7637 
7638     // If the context of this function is less available than D, we should not
7639     // emit a diagnostic.
7640     if (!ShouldDiagnoseAvailabilityInContext(SemaRef, Result, Introduced, Ctx))
7641       return;
7642 
7643     // We would like to emit the diagnostic even if -Wunguarded-availability is
7644     // not specified for deployment targets >= to iOS 11 or equivalent or
7645     // for declarations that were introduced in iOS 11 (macOS 10.13, ...) or
7646     // later.
7647     unsigned DiagKind =
7648         shouldDiagnoseAvailabilityByDefault(
7649             SemaRef.Context,
7650             SemaRef.Context.getTargetInfo().getPlatformMinVersion(), Introduced)
7651             ? diag::warn_unguarded_availability_new
7652             : diag::warn_unguarded_availability;
7653 
7654     SemaRef.Diag(Range.getBegin(), DiagKind)
7655         << Range << D
7656         << AvailabilityAttr::getPrettyPlatformName(
7657                SemaRef.getASTContext().getTargetInfo().getPlatformName())
7658         << Introduced.getAsString();
7659 
7660     SemaRef.Diag(OffendingDecl->getLocation(),
7661                  diag::note_availability_specified_here)
7662         << OffendingDecl << /* partial */ 3;
7663 
7664     auto FixitDiag =
7665         SemaRef.Diag(Range.getBegin(), diag::note_unguarded_available_silence)
7666         << Range << D
7667         << (SemaRef.getLangOpts().ObjC1 ? /*@available*/ 0
7668                                         : /*__builtin_available*/ 1);
7669 
7670     // Find the statement which should be enclosed in the if @available check.
7671     if (StmtStack.empty())
7672       return;
7673     const Stmt *StmtOfUse = StmtStack.back();
7674     const CompoundStmt *Scope = nullptr;
7675     for (const Stmt *S : llvm::reverse(StmtStack)) {
7676       if (const auto *CS = dyn_cast<CompoundStmt>(S)) {
7677         Scope = CS;
7678         break;
7679       }
7680       if (isBodyLikeChildStmt(StmtOfUse, S)) {
7681         // The declaration won't be seen outside of the statement, so we don't
7682         // have to wrap the uses of any declared variables in if (@available).
7683         // Therefore we can avoid setting Scope here.
7684         break;
7685       }
7686       StmtOfUse = S;
7687     }
7688     const Stmt *LastStmtOfUse = nullptr;
7689     if (isa<DeclStmt>(StmtOfUse) && Scope) {
7690       for (const Decl *D : cast<DeclStmt>(StmtOfUse)->decls()) {
7691         if (StmtUSEFinder::isContained(StmtStack.back(), D)) {
7692           LastStmtOfUse = LastDeclUSEFinder::findLastStmtThatUsesDecl(D, Scope);
7693           break;
7694         }
7695       }
7696     }
7697 
7698     const SourceManager &SM = SemaRef.getSourceManager();
7699     SourceLocation IfInsertionLoc =
7700         SM.getExpansionLoc(StmtOfUse->getBeginLoc());
7701     SourceLocation StmtEndLoc =
7702         SM.getExpansionRange(
7703               (LastStmtOfUse ? LastStmtOfUse : StmtOfUse)->getEndLoc())
7704             .getEnd();
7705     if (SM.getFileID(IfInsertionLoc) != SM.getFileID(StmtEndLoc))
7706       return;
7707 
7708     StringRef Indentation = Lexer::getIndentationForLine(IfInsertionLoc, SM);
7709     const char *ExtraIndentation = "    ";
7710     std::string FixItString;
7711     llvm::raw_string_ostream FixItOS(FixItString);
7712     FixItOS << "if (" << (SemaRef.getLangOpts().ObjC1 ? "@available"
7713                                                       : "__builtin_available")
7714             << "("
7715             << AvailabilityAttr::getPlatformNameSourceSpelling(
7716                    SemaRef.getASTContext().getTargetInfo().getPlatformName())
7717             << " " << Introduced.getAsString() << ", *)) {\n"
7718             << Indentation << ExtraIndentation;
7719     FixitDiag << FixItHint::CreateInsertion(IfInsertionLoc, FixItOS.str());
7720     SourceLocation ElseInsertionLoc = Lexer::findLocationAfterToken(
7721         StmtEndLoc, tok::semi, SM, SemaRef.getLangOpts(),
7722         /*SkipTrailingWhitespaceAndNewLine=*/false);
7723     if (ElseInsertionLoc.isInvalid())
7724       ElseInsertionLoc =
7725           Lexer::getLocForEndOfToken(StmtEndLoc, 0, SM, SemaRef.getLangOpts());
7726     FixItOS.str().clear();
7727     FixItOS << "\n"
7728             << Indentation << "} else {\n"
7729             << Indentation << ExtraIndentation
7730             << "// Fallback on earlier versions\n"
7731             << Indentation << "}";
7732     FixitDiag << FixItHint::CreateInsertion(ElseInsertionLoc, FixItOS.str());
7733   }
7734 }
7735 
7736 bool DiagnoseUnguardedAvailability::VisitTypeLoc(TypeLoc Ty) {
7737   const Type *TyPtr = Ty.getTypePtr();
7738   SourceRange Range{Ty.getBeginLoc(), Ty.getEndLoc()};
7739 
7740   if (Range.isInvalid())
7741     return true;
7742 
7743   if (const auto *TT = dyn_cast<TagType>(TyPtr)) {
7744     TagDecl *TD = TT->getDecl();
7745     DiagnoseDeclAvailability(TD, Range);
7746 
7747   } else if (const auto *TD = dyn_cast<TypedefType>(TyPtr)) {
7748     TypedefNameDecl *D = TD->getDecl();
7749     DiagnoseDeclAvailability(D, Range);
7750 
7751   } else if (const auto *ObjCO = dyn_cast<ObjCObjectType>(TyPtr)) {
7752     if (NamedDecl *D = ObjCO->getInterface())
7753       DiagnoseDeclAvailability(D, Range);
7754   }
7755 
7756   return true;
7757 }
7758 
7759 bool DiagnoseUnguardedAvailability::TraverseIfStmt(IfStmt *If) {
7760   VersionTuple CondVersion;
7761   if (auto *E = dyn_cast<ObjCAvailabilityCheckExpr>(If->getCond())) {
7762     CondVersion = E->getVersion();
7763 
7764     // If we're using the '*' case here or if this check is redundant, then we
7765     // use the enclosing version to check both branches.
7766     if (CondVersion.empty() || CondVersion <= AvailabilityStack.back())
7767       return TraverseStmt(If->getThen()) && TraverseStmt(If->getElse());
7768   } else {
7769     // This isn't an availability checking 'if', we can just continue.
7770     return Base::TraverseIfStmt(If);
7771   }
7772 
7773   AvailabilityStack.push_back(CondVersion);
7774   bool ShouldContinue = TraverseStmt(If->getThen());
7775   AvailabilityStack.pop_back();
7776 
7777   return ShouldContinue && TraverseStmt(If->getElse());
7778 }
7779 
7780 } // end anonymous namespace
7781 
7782 void Sema::DiagnoseUnguardedAvailabilityViolations(Decl *D) {
7783   Stmt *Body = nullptr;
7784 
7785   if (auto *FD = D->getAsFunction()) {
7786     // FIXME: We only examine the pattern decl for availability violations now,
7787     // but we should also examine instantiated templates.
7788     if (FD->isTemplateInstantiation())
7789       return;
7790 
7791     Body = FD->getBody();
7792   } else if (auto *MD = dyn_cast<ObjCMethodDecl>(D))
7793     Body = MD->getBody();
7794   else if (auto *BD = dyn_cast<BlockDecl>(D))
7795     Body = BD->getBody();
7796 
7797   assert(Body && "Need a body here!");
7798 
7799   DiagnoseUnguardedAvailability(*this, D).IssueDiagnostics(Body);
7800 }
7801 
7802 void Sema::DiagnoseAvailabilityOfDecl(NamedDecl *D,
7803                                       ArrayRef<SourceLocation> Locs,
7804                                       const ObjCInterfaceDecl *UnknownObjCClass,
7805                                       bool ObjCPropertyAccess,
7806                                       bool AvoidPartialAvailabilityChecks) {
7807   std::string Message;
7808   AvailabilityResult Result;
7809   const NamedDecl* OffendingDecl;
7810   // See if this declaration is unavailable, deprecated, or partial.
7811   std::tie(Result, OffendingDecl) = ShouldDiagnoseAvailabilityOfDecl(D, &Message);
7812   if (Result == AR_Available)
7813     return;
7814 
7815   if (Result == AR_NotYetIntroduced) {
7816     if (AvoidPartialAvailabilityChecks)
7817       return;
7818 
7819     // We need to know the @available context in the current function to
7820     // diagnose this use, let DiagnoseUnguardedAvailabilityViolations do that
7821     // when we're done parsing the current function.
7822     if (getCurFunctionOrMethodDecl()) {
7823       getEnclosingFunction()->HasPotentialAvailabilityViolations = true;
7824       return;
7825     } else if (getCurBlock() || getCurLambda()) {
7826       getCurFunction()->HasPotentialAvailabilityViolations = true;
7827       return;
7828     }
7829   }
7830 
7831   const ObjCPropertyDecl *ObjCPDecl = nullptr;
7832   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D)) {
7833     if (const ObjCPropertyDecl *PD = MD->findPropertyDecl()) {
7834       AvailabilityResult PDeclResult = PD->getAvailability(nullptr);
7835       if (PDeclResult == Result)
7836         ObjCPDecl = PD;
7837     }
7838   }
7839 
7840   EmitAvailabilityWarning(*this, Result, D, OffendingDecl, Message, Locs,
7841                           UnknownObjCClass, ObjCPDecl, ObjCPropertyAccess);
7842 }
7843