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