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