xref: /freebsd/contrib/llvm-project/clang/lib/Sema/SemaStmt.cpp (revision 7a6dacaca14b62ca4b74406814becb87a3fefac0)
10b57cec5SDimitry Andric //===--- SemaStmt.cpp - Semantic Analysis for Statements ------------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric //  This file implements semantic analysis for statements.
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
120b57cec5SDimitry Andric 
130b57cec5SDimitry Andric #include "clang/AST/ASTContext.h"
140b57cec5SDimitry Andric #include "clang/AST/ASTDiagnostic.h"
150b57cec5SDimitry Andric #include "clang/AST/ASTLambda.h"
160b57cec5SDimitry Andric #include "clang/AST/CXXInheritance.h"
17fe6060f1SDimitry Andric #include "clang/AST/CharUnits.h"
180b57cec5SDimitry Andric #include "clang/AST/DeclObjC.h"
190b57cec5SDimitry Andric #include "clang/AST/EvaluatedExprVisitor.h"
200b57cec5SDimitry Andric #include "clang/AST/ExprCXX.h"
210b57cec5SDimitry Andric #include "clang/AST/ExprObjC.h"
22fe6060f1SDimitry Andric #include "clang/AST/IgnoreExpr.h"
230b57cec5SDimitry Andric #include "clang/AST/RecursiveASTVisitor.h"
240b57cec5SDimitry Andric #include "clang/AST/StmtCXX.h"
250b57cec5SDimitry Andric #include "clang/AST/StmtObjC.h"
260b57cec5SDimitry Andric #include "clang/AST/TypeLoc.h"
270b57cec5SDimitry Andric #include "clang/AST/TypeOrdering.h"
280b57cec5SDimitry Andric #include "clang/Basic/TargetInfo.h"
290b57cec5SDimitry Andric #include "clang/Lex/Preprocessor.h"
300b57cec5SDimitry Andric #include "clang/Sema/Initialization.h"
310b57cec5SDimitry Andric #include "clang/Sema/Lookup.h"
32fe6060f1SDimitry Andric #include "clang/Sema/Ownership.h"
330b57cec5SDimitry Andric #include "clang/Sema/Scope.h"
340b57cec5SDimitry Andric #include "clang/Sema/ScopeInfo.h"
35fe6060f1SDimitry Andric #include "clang/Sema/SemaInternal.h"
360b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h"
370b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h"
380b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h"
390b57cec5SDimitry Andric #include "llvm/ADT/SmallPtrSet.h"
400b57cec5SDimitry Andric #include "llvm/ADT/SmallString.h"
410b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h"
4206c3fb27SDimitry Andric #include "llvm/ADT/StringExtras.h"
430b57cec5SDimitry Andric 
440b57cec5SDimitry Andric using namespace clang;
450b57cec5SDimitry Andric using namespace sema;
460b57cec5SDimitry Andric 
470b57cec5SDimitry Andric StmtResult Sema::ActOnExprStmt(ExprResult FE, bool DiscardedValue) {
480b57cec5SDimitry Andric   if (FE.isInvalid())
490b57cec5SDimitry Andric     return StmtError();
500b57cec5SDimitry Andric 
510b57cec5SDimitry Andric   FE = ActOnFinishFullExpr(FE.get(), FE.get()->getExprLoc(), DiscardedValue);
520b57cec5SDimitry Andric   if (FE.isInvalid())
530b57cec5SDimitry Andric     return StmtError();
540b57cec5SDimitry Andric 
550b57cec5SDimitry Andric   // C99 6.8.3p2: The expression in an expression statement is evaluated as a
560b57cec5SDimitry Andric   // void expression for its side effects.  Conversion to void allows any
570b57cec5SDimitry Andric   // operand, even incomplete types.
580b57cec5SDimitry Andric 
590b57cec5SDimitry Andric   // Same thing in for stmt first clause (when expr) and third clause.
600b57cec5SDimitry Andric   return StmtResult(FE.getAs<Stmt>());
610b57cec5SDimitry Andric }
620b57cec5SDimitry Andric 
630b57cec5SDimitry Andric 
640b57cec5SDimitry Andric StmtResult Sema::ActOnExprStmtError() {
650b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
660b57cec5SDimitry Andric   return StmtError();
670b57cec5SDimitry Andric }
680b57cec5SDimitry Andric 
690b57cec5SDimitry Andric StmtResult Sema::ActOnNullStmt(SourceLocation SemiLoc,
700b57cec5SDimitry Andric                                bool HasLeadingEmptyMacro) {
710b57cec5SDimitry Andric   return new (Context) NullStmt(SemiLoc, HasLeadingEmptyMacro);
720b57cec5SDimitry Andric }
730b57cec5SDimitry Andric 
740b57cec5SDimitry Andric StmtResult Sema::ActOnDeclStmt(DeclGroupPtrTy dg, SourceLocation StartLoc,
750b57cec5SDimitry Andric                                SourceLocation EndLoc) {
760b57cec5SDimitry Andric   DeclGroupRef DG = dg.get();
770b57cec5SDimitry Andric 
780b57cec5SDimitry Andric   // If we have an invalid decl, just return an error.
790b57cec5SDimitry Andric   if (DG.isNull()) return StmtError();
800b57cec5SDimitry Andric 
810b57cec5SDimitry Andric   return new (Context) DeclStmt(DG, StartLoc, EndLoc);
820b57cec5SDimitry Andric }
830b57cec5SDimitry Andric 
840b57cec5SDimitry Andric void Sema::ActOnForEachDeclStmt(DeclGroupPtrTy dg) {
850b57cec5SDimitry Andric   DeclGroupRef DG = dg.get();
860b57cec5SDimitry Andric 
870b57cec5SDimitry Andric   // If we don't have a declaration, or we have an invalid declaration,
880b57cec5SDimitry Andric   // just return.
890b57cec5SDimitry Andric   if (DG.isNull() || !DG.isSingleDecl())
900b57cec5SDimitry Andric     return;
910b57cec5SDimitry Andric 
920b57cec5SDimitry Andric   Decl *decl = DG.getSingleDecl();
930b57cec5SDimitry Andric   if (!decl || decl->isInvalidDecl())
940b57cec5SDimitry Andric     return;
950b57cec5SDimitry Andric 
960b57cec5SDimitry Andric   // Only variable declarations are permitted.
970b57cec5SDimitry Andric   VarDecl *var = dyn_cast<VarDecl>(decl);
980b57cec5SDimitry Andric   if (!var) {
990b57cec5SDimitry Andric     Diag(decl->getLocation(), diag::err_non_variable_decl_in_for);
1000b57cec5SDimitry Andric     decl->setInvalidDecl();
1010b57cec5SDimitry Andric     return;
1020b57cec5SDimitry Andric   }
1030b57cec5SDimitry Andric 
1040b57cec5SDimitry Andric   // foreach variables are never actually initialized in the way that
1050b57cec5SDimitry Andric   // the parser came up with.
1060b57cec5SDimitry Andric   var->setInit(nullptr);
1070b57cec5SDimitry Andric 
1080b57cec5SDimitry Andric   // In ARC, we don't need to retain the iteration variable of a fast
1090b57cec5SDimitry Andric   // enumeration loop.  Rather than actually trying to catch that
1100b57cec5SDimitry Andric   // during declaration processing, we remove the consequences here.
1110b57cec5SDimitry Andric   if (getLangOpts().ObjCAutoRefCount) {
1120b57cec5SDimitry Andric     QualType type = var->getType();
1130b57cec5SDimitry Andric 
1140b57cec5SDimitry Andric     // Only do this if we inferred the lifetime.  Inferred lifetime
1150b57cec5SDimitry Andric     // will show up as a local qualifier because explicit lifetime
1160b57cec5SDimitry Andric     // should have shown up as an AttributedType instead.
1170b57cec5SDimitry Andric     if (type.getLocalQualifiers().getObjCLifetime() == Qualifiers::OCL_Strong) {
1180b57cec5SDimitry Andric       // Add 'const' and mark the variable as pseudo-strong.
1190b57cec5SDimitry Andric       var->setType(type.withConst());
1200b57cec5SDimitry Andric       var->setARCPseudoStrong(true);
1210b57cec5SDimitry Andric     }
1220b57cec5SDimitry Andric   }
1230b57cec5SDimitry Andric }
1240b57cec5SDimitry Andric 
1250b57cec5SDimitry Andric /// Diagnose unused comparisons, both builtin and overloaded operators.
1260b57cec5SDimitry Andric /// For '==' and '!=', suggest fixits for '=' or '|='.
1270b57cec5SDimitry Andric ///
1280b57cec5SDimitry Andric /// Adding a cast to void (or other expression wrappers) will prevent the
1290b57cec5SDimitry Andric /// warning from firing.
1300b57cec5SDimitry Andric static bool DiagnoseUnusedComparison(Sema &S, const Expr *E) {
1310b57cec5SDimitry Andric   SourceLocation Loc;
1320b57cec5SDimitry Andric   bool CanAssign;
1330b57cec5SDimitry Andric   enum { Equality, Inequality, Relational, ThreeWay } Kind;
1340b57cec5SDimitry Andric 
1350b57cec5SDimitry Andric   if (const BinaryOperator *Op = dyn_cast<BinaryOperator>(E)) {
1360b57cec5SDimitry Andric     if (!Op->isComparisonOp())
1370b57cec5SDimitry Andric       return false;
1380b57cec5SDimitry Andric 
1390b57cec5SDimitry Andric     if (Op->getOpcode() == BO_EQ)
1400b57cec5SDimitry Andric       Kind = Equality;
1410b57cec5SDimitry Andric     else if (Op->getOpcode() == BO_NE)
1420b57cec5SDimitry Andric       Kind = Inequality;
1430b57cec5SDimitry Andric     else if (Op->getOpcode() == BO_Cmp)
1440b57cec5SDimitry Andric       Kind = ThreeWay;
1450b57cec5SDimitry Andric     else {
1460b57cec5SDimitry Andric       assert(Op->isRelationalOp());
1470b57cec5SDimitry Andric       Kind = Relational;
1480b57cec5SDimitry Andric     }
1490b57cec5SDimitry Andric     Loc = Op->getOperatorLoc();
1500b57cec5SDimitry Andric     CanAssign = Op->getLHS()->IgnoreParenImpCasts()->isLValue();
1510b57cec5SDimitry Andric   } else if (const CXXOperatorCallExpr *Op = dyn_cast<CXXOperatorCallExpr>(E)) {
1520b57cec5SDimitry Andric     switch (Op->getOperator()) {
1530b57cec5SDimitry Andric     case OO_EqualEqual:
1540b57cec5SDimitry Andric       Kind = Equality;
1550b57cec5SDimitry Andric       break;
1560b57cec5SDimitry Andric     case OO_ExclaimEqual:
1570b57cec5SDimitry Andric       Kind = Inequality;
1580b57cec5SDimitry Andric       break;
1590b57cec5SDimitry Andric     case OO_Less:
1600b57cec5SDimitry Andric     case OO_Greater:
1610b57cec5SDimitry Andric     case OO_GreaterEqual:
1620b57cec5SDimitry Andric     case OO_LessEqual:
1630b57cec5SDimitry Andric       Kind = Relational;
1640b57cec5SDimitry Andric       break;
1650b57cec5SDimitry Andric     case OO_Spaceship:
1660b57cec5SDimitry Andric       Kind = ThreeWay;
1670b57cec5SDimitry Andric       break;
1680b57cec5SDimitry Andric     default:
1690b57cec5SDimitry Andric       return false;
1700b57cec5SDimitry Andric     }
1710b57cec5SDimitry Andric 
1720b57cec5SDimitry Andric     Loc = Op->getOperatorLoc();
1730b57cec5SDimitry Andric     CanAssign = Op->getArg(0)->IgnoreParenImpCasts()->isLValue();
1740b57cec5SDimitry Andric   } else {
1750b57cec5SDimitry Andric     // Not a typo-prone comparison.
1760b57cec5SDimitry Andric     return false;
1770b57cec5SDimitry Andric   }
1780b57cec5SDimitry Andric 
1790b57cec5SDimitry Andric   // Suppress warnings when the operator, suspicious as it may be, comes from
1800b57cec5SDimitry Andric   // a macro expansion.
1810b57cec5SDimitry Andric   if (S.SourceMgr.isMacroBodyExpansion(Loc))
1820b57cec5SDimitry Andric     return false;
1830b57cec5SDimitry Andric 
1840b57cec5SDimitry Andric   S.Diag(Loc, diag::warn_unused_comparison)
1850b57cec5SDimitry Andric     << (unsigned)Kind << E->getSourceRange();
1860b57cec5SDimitry Andric 
1870b57cec5SDimitry Andric   // If the LHS is a plausible entity to assign to, provide a fixit hint to
1880b57cec5SDimitry Andric   // correct common typos.
1890b57cec5SDimitry Andric   if (CanAssign) {
1900b57cec5SDimitry Andric     if (Kind == Inequality)
1910b57cec5SDimitry Andric       S.Diag(Loc, diag::note_inequality_comparison_to_or_assign)
1920b57cec5SDimitry Andric         << FixItHint::CreateReplacement(Loc, "|=");
1930b57cec5SDimitry Andric     else if (Kind == Equality)
1940b57cec5SDimitry Andric       S.Diag(Loc, diag::note_equality_comparison_to_assign)
1950b57cec5SDimitry Andric         << FixItHint::CreateReplacement(Loc, "=");
1960b57cec5SDimitry Andric   }
1970b57cec5SDimitry Andric 
1980b57cec5SDimitry Andric   return true;
1990b57cec5SDimitry Andric }
2000b57cec5SDimitry Andric 
201a7dea167SDimitry Andric static bool DiagnoseNoDiscard(Sema &S, const WarnUnusedResultAttr *A,
202a7dea167SDimitry Andric                               SourceLocation Loc, SourceRange R1,
203a7dea167SDimitry Andric                               SourceRange R2, bool IsCtor) {
204a7dea167SDimitry Andric   if (!A)
205a7dea167SDimitry Andric     return false;
206a7dea167SDimitry Andric   StringRef Msg = A->getMessage();
207a7dea167SDimitry Andric 
208a7dea167SDimitry Andric   if (Msg.empty()) {
209a7dea167SDimitry Andric     if (IsCtor)
210a7dea167SDimitry Andric       return S.Diag(Loc, diag::warn_unused_constructor) << A << R1 << R2;
211a7dea167SDimitry Andric     return S.Diag(Loc, diag::warn_unused_result) << A << R1 << R2;
212a7dea167SDimitry Andric   }
213a7dea167SDimitry Andric 
214a7dea167SDimitry Andric   if (IsCtor)
215a7dea167SDimitry Andric     return S.Diag(Loc, diag::warn_unused_constructor_msg) << A << Msg << R1
216a7dea167SDimitry Andric                                                           << R2;
217a7dea167SDimitry Andric   return S.Diag(Loc, diag::warn_unused_result_msg) << A << Msg << R1 << R2;
218a7dea167SDimitry Andric }
219a7dea167SDimitry Andric 
220349cc55cSDimitry Andric void Sema::DiagnoseUnusedExprResult(const Stmt *S, unsigned DiagID) {
2210b57cec5SDimitry Andric   if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
222349cc55cSDimitry Andric     return DiagnoseUnusedExprResult(Label->getSubStmt(), DiagID);
2230b57cec5SDimitry Andric 
2240b57cec5SDimitry Andric   const Expr *E = dyn_cast_or_null<Expr>(S);
2250b57cec5SDimitry Andric   if (!E)
2260b57cec5SDimitry Andric     return;
2270b57cec5SDimitry Andric 
2280b57cec5SDimitry Andric   // If we are in an unevaluated expression context, then there can be no unused
2290b57cec5SDimitry Andric   // results because the results aren't expected to be used in the first place.
2300b57cec5SDimitry Andric   if (isUnevaluatedContext())
2310b57cec5SDimitry Andric     return;
2320b57cec5SDimitry Andric 
2330b57cec5SDimitry Andric   SourceLocation ExprLoc = E->IgnoreParenImpCasts()->getExprLoc();
2340b57cec5SDimitry Andric   // In most cases, we don't want to warn if the expression is written in a
2350b57cec5SDimitry Andric   // macro body, or if the macro comes from a system header. If the offending
2360b57cec5SDimitry Andric   // expression is a call to a function with the warn_unused_result attribute,
2370b57cec5SDimitry Andric   // we warn no matter the location. Because of the order in which the various
2380b57cec5SDimitry Andric   // checks need to happen, we factor out the macro-related test here.
2390b57cec5SDimitry Andric   bool ShouldSuppress =
2400b57cec5SDimitry Andric       SourceMgr.isMacroBodyExpansion(ExprLoc) ||
2410b57cec5SDimitry Andric       SourceMgr.isInSystemMacro(ExprLoc);
2420b57cec5SDimitry Andric 
2430b57cec5SDimitry Andric   const Expr *WarnExpr;
2440b57cec5SDimitry Andric   SourceLocation Loc;
2450b57cec5SDimitry Andric   SourceRange R1, R2;
2460b57cec5SDimitry Andric   if (!E->isUnusedResultAWarning(WarnExpr, Loc, R1, R2, Context))
2470b57cec5SDimitry Andric     return;
2480b57cec5SDimitry Andric 
2490b57cec5SDimitry Andric   // If this is a GNU statement expression expanded from a macro, it is probably
2500b57cec5SDimitry Andric   // unused because it is a function-like macro that can be used as either an
2510b57cec5SDimitry Andric   // expression or statement.  Don't warn, because it is almost certainly a
2520b57cec5SDimitry Andric   // false positive.
2530b57cec5SDimitry Andric   if (isa<StmtExpr>(E) && Loc.isMacroID())
2540b57cec5SDimitry Andric     return;
2550b57cec5SDimitry Andric 
2560b57cec5SDimitry Andric   // Check if this is the UNREFERENCED_PARAMETER from the Microsoft headers.
2570b57cec5SDimitry Andric   // That macro is frequently used to suppress "unused parameter" warnings,
2580b57cec5SDimitry Andric   // but its implementation makes clang's -Wunused-value fire.  Prevent this.
2590b57cec5SDimitry Andric   if (isa<ParenExpr>(E->IgnoreImpCasts()) && Loc.isMacroID()) {
2600b57cec5SDimitry Andric     SourceLocation SpellLoc = Loc;
2610b57cec5SDimitry Andric     if (findMacroSpelling(SpellLoc, "UNREFERENCED_PARAMETER"))
2620b57cec5SDimitry Andric       return;
2630b57cec5SDimitry Andric   }
2640b57cec5SDimitry Andric 
2650b57cec5SDimitry Andric   // Okay, we have an unused result.  Depending on what the base expression is,
2660b57cec5SDimitry Andric   // we might want to make a more specific diagnostic.  Check for one of these
2670b57cec5SDimitry Andric   // cases now.
2680b57cec5SDimitry Andric   if (const FullExpr *Temps = dyn_cast<FullExpr>(E))
2690b57cec5SDimitry Andric     E = Temps->getSubExpr();
2700b57cec5SDimitry Andric   if (const CXXBindTemporaryExpr *TempExpr = dyn_cast<CXXBindTemporaryExpr>(E))
2710b57cec5SDimitry Andric     E = TempExpr->getSubExpr();
2720b57cec5SDimitry Andric 
2730b57cec5SDimitry Andric   if (DiagnoseUnusedComparison(*this, E))
2740b57cec5SDimitry Andric     return;
2750b57cec5SDimitry Andric 
2760b57cec5SDimitry Andric   E = WarnExpr;
277a7dea167SDimitry Andric   if (const auto *Cast = dyn_cast<CastExpr>(E))
278a7dea167SDimitry Andric     if (Cast->getCastKind() == CK_NoOp ||
279a7dea167SDimitry Andric         Cast->getCastKind() == CK_ConstructorConversion)
280a7dea167SDimitry Andric       E = Cast->getSubExpr()->IgnoreImpCasts();
281a7dea167SDimitry Andric 
2820b57cec5SDimitry Andric   if (const CallExpr *CE = dyn_cast<CallExpr>(E)) {
2830b57cec5SDimitry Andric     if (E->getType()->isVoidType())
2840b57cec5SDimitry Andric       return;
2850b57cec5SDimitry Andric 
286a7dea167SDimitry Andric     if (DiagnoseNoDiscard(*this, cast_or_null<WarnUnusedResultAttr>(
287a7dea167SDimitry Andric                                      CE->getUnusedResultAttr(Context)),
288a7dea167SDimitry Andric                           Loc, R1, R2, /*isCtor=*/false))
2890b57cec5SDimitry Andric       return;
2900b57cec5SDimitry Andric 
2910b57cec5SDimitry Andric     // If the callee has attribute pure, const, or warn_unused_result, warn with
2920b57cec5SDimitry Andric     // a more specific message to make it clear what is happening. If the call
2930b57cec5SDimitry Andric     // is written in a macro body, only warn if it has the warn_unused_result
2940b57cec5SDimitry Andric     // attribute.
2950b57cec5SDimitry Andric     if (const Decl *FD = CE->getCalleeDecl()) {
2960b57cec5SDimitry Andric       if (ShouldSuppress)
2970b57cec5SDimitry Andric         return;
2980b57cec5SDimitry Andric       if (FD->hasAttr<PureAttr>()) {
2990b57cec5SDimitry Andric         Diag(Loc, diag::warn_unused_call) << R1 << R2 << "pure";
3000b57cec5SDimitry Andric         return;
3010b57cec5SDimitry Andric       }
3020b57cec5SDimitry Andric       if (FD->hasAttr<ConstAttr>()) {
3030b57cec5SDimitry Andric         Diag(Loc, diag::warn_unused_call) << R1 << R2 << "const";
3040b57cec5SDimitry Andric         return;
3050b57cec5SDimitry Andric       }
3060b57cec5SDimitry Andric     }
307a7dea167SDimitry Andric   } else if (const auto *CE = dyn_cast<CXXConstructExpr>(E)) {
308a7dea167SDimitry Andric     if (const CXXConstructorDecl *Ctor = CE->getConstructor()) {
309a7dea167SDimitry Andric       const auto *A = Ctor->getAttr<WarnUnusedResultAttr>();
310a7dea167SDimitry Andric       A = A ? A : Ctor->getParent()->getAttr<WarnUnusedResultAttr>();
311a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, A, Loc, R1, R2, /*isCtor=*/true))
312a7dea167SDimitry Andric         return;
313a7dea167SDimitry Andric     }
314a7dea167SDimitry Andric   } else if (const auto *ILE = dyn_cast<InitListExpr>(E)) {
315a7dea167SDimitry Andric     if (const TagDecl *TD = ILE->getType()->getAsTagDecl()) {
316a7dea167SDimitry Andric 
317a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, TD->getAttr<WarnUnusedResultAttr>(), Loc, R1,
318a7dea167SDimitry Andric                             R2, /*isCtor=*/false))
319a7dea167SDimitry Andric         return;
320a7dea167SDimitry Andric     }
3210b57cec5SDimitry Andric   } else if (ShouldSuppress)
3220b57cec5SDimitry Andric     return;
3230b57cec5SDimitry Andric 
324a7dea167SDimitry Andric   E = WarnExpr;
3250b57cec5SDimitry Andric   if (const ObjCMessageExpr *ME = dyn_cast<ObjCMessageExpr>(E)) {
3260b57cec5SDimitry Andric     if (getLangOpts().ObjCAutoRefCount && ME->isDelegateInitCall()) {
3270b57cec5SDimitry Andric       Diag(Loc, diag::err_arc_unused_init_message) << R1;
3280b57cec5SDimitry Andric       return;
3290b57cec5SDimitry Andric     }
3300b57cec5SDimitry Andric     const ObjCMethodDecl *MD = ME->getMethodDecl();
3310b57cec5SDimitry Andric     if (MD) {
332a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, MD->getAttr<WarnUnusedResultAttr>(), Loc, R1,
333a7dea167SDimitry Andric                             R2, /*isCtor=*/false))
3340b57cec5SDimitry Andric         return;
3350b57cec5SDimitry Andric     }
3360b57cec5SDimitry Andric   } else if (const PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E)) {
3370b57cec5SDimitry Andric     const Expr *Source = POE->getSyntacticForm();
3385ffd83dbSDimitry Andric     // Handle the actually selected call of an OpenMP specialized call.
3395ffd83dbSDimitry Andric     if (LangOpts.OpenMP && isa<CallExpr>(Source) &&
3405ffd83dbSDimitry Andric         POE->getNumSemanticExprs() == 1 &&
3415ffd83dbSDimitry Andric         isa<CallExpr>(POE->getSemanticExpr(0)))
342349cc55cSDimitry Andric       return DiagnoseUnusedExprResult(POE->getSemanticExpr(0), DiagID);
3430b57cec5SDimitry Andric     if (isa<ObjCSubscriptRefExpr>(Source))
3440b57cec5SDimitry Andric       DiagID = diag::warn_unused_container_subscript_expr;
34581ad6265SDimitry Andric     else if (isa<ObjCPropertyRefExpr>(Source))
3460b57cec5SDimitry Andric       DiagID = diag::warn_unused_property_expr;
3470b57cec5SDimitry Andric   } else if (const CXXFunctionalCastExpr *FC
3480b57cec5SDimitry Andric                                        = dyn_cast<CXXFunctionalCastExpr>(E)) {
3490b57cec5SDimitry Andric     const Expr *E = FC->getSubExpr();
3500b57cec5SDimitry Andric     if (const CXXBindTemporaryExpr *TE = dyn_cast<CXXBindTemporaryExpr>(E))
3510b57cec5SDimitry Andric       E = TE->getSubExpr();
3520b57cec5SDimitry Andric     if (isa<CXXTemporaryObjectExpr>(E))
3530b57cec5SDimitry Andric       return;
3540b57cec5SDimitry Andric     if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
3550b57cec5SDimitry Andric       if (const CXXRecordDecl *RD = CE->getType()->getAsCXXRecordDecl())
3560b57cec5SDimitry Andric         if (!RD->getAttr<WarnUnusedAttr>())
3570b57cec5SDimitry Andric           return;
3580b57cec5SDimitry Andric   }
3590b57cec5SDimitry Andric   // Diagnose "(void*) blah" as a typo for "(void) blah".
3600b57cec5SDimitry Andric   else if (const CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(E)) {
3610b57cec5SDimitry Andric     TypeSourceInfo *TI = CE->getTypeInfoAsWritten();
3620b57cec5SDimitry Andric     QualType T = TI->getType();
3630b57cec5SDimitry Andric 
3640b57cec5SDimitry Andric     // We really do want to use the non-canonical type here.
3650b57cec5SDimitry Andric     if (T == Context.VoidPtrTy) {
3660b57cec5SDimitry Andric       PointerTypeLoc TL = TI->getTypeLoc().castAs<PointerTypeLoc>();
3670b57cec5SDimitry Andric 
3680b57cec5SDimitry Andric       Diag(Loc, diag::warn_unused_voidptr)
3690b57cec5SDimitry Andric         << FixItHint::CreateRemoval(TL.getStarLoc());
3700b57cec5SDimitry Andric       return;
3710b57cec5SDimitry Andric     }
3720b57cec5SDimitry Andric   }
3730b57cec5SDimitry Andric 
3745ffd83dbSDimitry Andric   // Tell the user to assign it into a variable to force a volatile load if this
3755ffd83dbSDimitry Andric   // isn't an array.
3765ffd83dbSDimitry Andric   if (E->isGLValue() && E->getType().isVolatileQualified() &&
3775ffd83dbSDimitry Andric       !E->getType()->isArrayType()) {
3780b57cec5SDimitry Andric     Diag(Loc, diag::warn_unused_volatile) << R1 << R2;
3790b57cec5SDimitry Andric     return;
3800b57cec5SDimitry Andric   }
3810b57cec5SDimitry Andric 
382349cc55cSDimitry Andric   // Do not diagnose use of a comma operator in a SFINAE context because the
383349cc55cSDimitry Andric   // type of the left operand could be used for SFINAE, so technically it is
384349cc55cSDimitry Andric   // *used*.
385349cc55cSDimitry Andric   if (DiagID != diag::warn_unused_comma_left_operand || !isSFINAEContext())
386bdd1243dSDimitry Andric     DiagIfReachable(Loc, S ? llvm::ArrayRef(S) : std::nullopt,
387349cc55cSDimitry Andric                     PDiag(DiagID) << R1 << R2);
3880b57cec5SDimitry Andric }
3890b57cec5SDimitry Andric 
3900b57cec5SDimitry Andric void Sema::ActOnStartOfCompoundStmt(bool IsStmtExpr) {
3910b57cec5SDimitry Andric   PushCompoundScope(IsStmtExpr);
3920b57cec5SDimitry Andric }
3930b57cec5SDimitry Andric 
394e8d8bef9SDimitry Andric void Sema::ActOnAfterCompoundStatementLeadingPragmas() {
395e8d8bef9SDimitry Andric   if (getCurFPFeatures().isFPConstrained()) {
396e8d8bef9SDimitry Andric     FunctionScopeInfo *FSI = getCurFunction();
397e8d8bef9SDimitry Andric     assert(FSI);
398e8d8bef9SDimitry Andric     FSI->setUsesFPIntrin();
399e8d8bef9SDimitry Andric   }
400e8d8bef9SDimitry Andric }
401e8d8bef9SDimitry Andric 
4020b57cec5SDimitry Andric void Sema::ActOnFinishOfCompoundStmt() {
4030b57cec5SDimitry Andric   PopCompoundScope();
4040b57cec5SDimitry Andric }
4050b57cec5SDimitry Andric 
4060b57cec5SDimitry Andric sema::CompoundScopeInfo &Sema::getCurCompoundScope() const {
4070b57cec5SDimitry Andric   return getCurFunction()->CompoundScopes.back();
4080b57cec5SDimitry Andric }
4090b57cec5SDimitry Andric 
4100b57cec5SDimitry Andric StmtResult Sema::ActOnCompoundStmt(SourceLocation L, SourceLocation R,
4110b57cec5SDimitry Andric                                    ArrayRef<Stmt *> Elts, bool isStmtExpr) {
4120b57cec5SDimitry Andric   const unsigned NumElts = Elts.size();
4130b57cec5SDimitry Andric 
41404eeddc0SDimitry Andric   // If we're in C mode, check that we don't have any decls after stmts.  If
41504eeddc0SDimitry Andric   // so, emit an extension diagnostic in C89 and potentially a warning in later
41604eeddc0SDimitry Andric   // versions.
41704eeddc0SDimitry Andric   const unsigned MixedDeclsCodeID = getLangOpts().C99
41804eeddc0SDimitry Andric                                         ? diag::warn_mixed_decls_code
41904eeddc0SDimitry Andric                                         : diag::ext_mixed_decls_code;
42004eeddc0SDimitry Andric   if (!getLangOpts().CPlusPlus && !Diags.isIgnored(MixedDeclsCodeID, L)) {
4210b57cec5SDimitry Andric     // Note that __extension__ can be around a decl.
4220b57cec5SDimitry Andric     unsigned i = 0;
4230b57cec5SDimitry Andric     // Skip over all declarations.
4240b57cec5SDimitry Andric     for (; i != NumElts && isa<DeclStmt>(Elts[i]); ++i)
4250b57cec5SDimitry Andric       /*empty*/;
4260b57cec5SDimitry Andric 
4270b57cec5SDimitry Andric     // We found the end of the list or a statement.  Scan for another declstmt.
4280b57cec5SDimitry Andric     for (; i != NumElts && !isa<DeclStmt>(Elts[i]); ++i)
4290b57cec5SDimitry Andric       /*empty*/;
4300b57cec5SDimitry Andric 
4310b57cec5SDimitry Andric     if (i != NumElts) {
4320b57cec5SDimitry Andric       Decl *D = *cast<DeclStmt>(Elts[i])->decl_begin();
43304eeddc0SDimitry Andric       Diag(D->getLocation(), MixedDeclsCodeID);
4340b57cec5SDimitry Andric     }
4350b57cec5SDimitry Andric   }
4360b57cec5SDimitry Andric 
4370b57cec5SDimitry Andric   // Check for suspicious empty body (null statement) in `for' and `while'
4380b57cec5SDimitry Andric   // statements.  Don't do anything for template instantiations, this just adds
4390b57cec5SDimitry Andric   // noise.
4400b57cec5SDimitry Andric   if (NumElts != 0 && !CurrentInstantiationScope &&
4410b57cec5SDimitry Andric       getCurCompoundScope().HasEmptyLoopBodies) {
4420b57cec5SDimitry Andric     for (unsigned i = 0; i != NumElts - 1; ++i)
4430b57cec5SDimitry Andric       DiagnoseEmptyLoopBody(Elts[i], Elts[i + 1]);
4440b57cec5SDimitry Andric   }
4450b57cec5SDimitry Andric 
44681ad6265SDimitry Andric   // Calculate difference between FP options in this compound statement and in
44781ad6265SDimitry Andric   // the enclosing one. If this is a function body, take the difference against
44881ad6265SDimitry Andric   // default options. In this case the difference will indicate options that are
44981ad6265SDimitry Andric   // changed upon entry to the statement.
45081ad6265SDimitry Andric   FPOptions FPO = (getCurFunction()->CompoundScopes.size() == 1)
45181ad6265SDimitry Andric                       ? FPOptions(getLangOpts())
45281ad6265SDimitry Andric                       : getCurCompoundScope().InitialFPFeatures;
45381ad6265SDimitry Andric   FPOptionsOverride FPDiff = getCurFPFeatures().getChangesFrom(FPO);
45481ad6265SDimitry Andric 
45581ad6265SDimitry Andric   return CompoundStmt::Create(Context, Elts, FPDiff, L, R);
4560b57cec5SDimitry Andric }
4570b57cec5SDimitry Andric 
4580b57cec5SDimitry Andric ExprResult
4590b57cec5SDimitry Andric Sema::ActOnCaseExpr(SourceLocation CaseLoc, ExprResult Val) {
4600b57cec5SDimitry Andric   if (!Val.get())
4610b57cec5SDimitry Andric     return Val;
4620b57cec5SDimitry Andric 
4630b57cec5SDimitry Andric   if (DiagnoseUnexpandedParameterPack(Val.get()))
4640b57cec5SDimitry Andric     return ExprError();
4650b57cec5SDimitry Andric 
4660b57cec5SDimitry Andric   // If we're not inside a switch, let the 'case' statement handling diagnose
4670b57cec5SDimitry Andric   // this. Just clean up after the expression as best we can.
468a7dea167SDimitry Andric   if (getCurFunction()->SwitchStack.empty())
469a7dea167SDimitry Andric     return ActOnFinishFullExpr(Val.get(), Val.get()->getExprLoc(), false,
470a7dea167SDimitry Andric                                getLangOpts().CPlusPlus11);
471a7dea167SDimitry Andric 
4720b57cec5SDimitry Andric   Expr *CondExpr =
4730b57cec5SDimitry Andric       getCurFunction()->SwitchStack.back().getPointer()->getCond();
4740b57cec5SDimitry Andric   if (!CondExpr)
4750b57cec5SDimitry Andric     return ExprError();
4760b57cec5SDimitry Andric   QualType CondType = CondExpr->getType();
4770b57cec5SDimitry Andric 
4780b57cec5SDimitry Andric   auto CheckAndFinish = [&](Expr *E) {
4790b57cec5SDimitry Andric     if (CondType->isDependentType() || E->isTypeDependent())
4800b57cec5SDimitry Andric       return ExprResult(E);
4810b57cec5SDimitry Andric 
4820b57cec5SDimitry Andric     if (getLangOpts().CPlusPlus11) {
4830b57cec5SDimitry Andric       // C++11 [stmt.switch]p2: the constant-expression shall be a converted
4840b57cec5SDimitry Andric       // constant expression of the promoted type of the switch condition.
4850b57cec5SDimitry Andric       llvm::APSInt TempVal;
4860b57cec5SDimitry Andric       return CheckConvertedConstantExpression(E, CondType, TempVal,
4870b57cec5SDimitry Andric                                               CCEK_CaseValue);
4880b57cec5SDimitry Andric     }
4890b57cec5SDimitry Andric 
4900b57cec5SDimitry Andric     ExprResult ER = E;
4910b57cec5SDimitry Andric     if (!E->isValueDependent())
492e8d8bef9SDimitry Andric       ER = VerifyIntegerConstantExpression(E, AllowFold);
4930b57cec5SDimitry Andric     if (!ER.isInvalid())
4940b57cec5SDimitry Andric       ER = DefaultLvalueConversion(ER.get());
4950b57cec5SDimitry Andric     if (!ER.isInvalid())
4960b57cec5SDimitry Andric       ER = ImpCastExprToType(ER.get(), CondType, CK_IntegralCast);
497a7dea167SDimitry Andric     if (!ER.isInvalid())
498a7dea167SDimitry Andric       ER = ActOnFinishFullExpr(ER.get(), ER.get()->getExprLoc(), false);
4990b57cec5SDimitry Andric     return ER;
5000b57cec5SDimitry Andric   };
5010b57cec5SDimitry Andric 
5025ffd83dbSDimitry Andric   ExprResult Converted = CorrectDelayedTyposInExpr(
5035ffd83dbSDimitry Andric       Val, /*InitDecl=*/nullptr, /*RecoverUncorrectedTypos=*/false,
5045ffd83dbSDimitry Andric       CheckAndFinish);
5050b57cec5SDimitry Andric   if (Converted.get() == Val.get())
5060b57cec5SDimitry Andric     Converted = CheckAndFinish(Val.get());
507a7dea167SDimitry Andric   return Converted;
5080b57cec5SDimitry Andric }
5090b57cec5SDimitry Andric 
5100b57cec5SDimitry Andric StmtResult
5110b57cec5SDimitry Andric Sema::ActOnCaseStmt(SourceLocation CaseLoc, ExprResult LHSVal,
5120b57cec5SDimitry Andric                     SourceLocation DotDotDotLoc, ExprResult RHSVal,
5130b57cec5SDimitry Andric                     SourceLocation ColonLoc) {
5140b57cec5SDimitry Andric   assert((LHSVal.isInvalid() || LHSVal.get()) && "missing LHS value");
5150b57cec5SDimitry Andric   assert((DotDotDotLoc.isInvalid() ? RHSVal.isUnset()
5160b57cec5SDimitry Andric                                    : RHSVal.isInvalid() || RHSVal.get()) &&
5170b57cec5SDimitry Andric          "missing RHS value");
5180b57cec5SDimitry Andric 
5190b57cec5SDimitry Andric   if (getCurFunction()->SwitchStack.empty()) {
5200b57cec5SDimitry Andric     Diag(CaseLoc, diag::err_case_not_in_switch);
5210b57cec5SDimitry Andric     return StmtError();
5220b57cec5SDimitry Andric   }
5230b57cec5SDimitry Andric 
5240b57cec5SDimitry Andric   if (LHSVal.isInvalid() || RHSVal.isInvalid()) {
5250b57cec5SDimitry Andric     getCurFunction()->SwitchStack.back().setInt(true);
5260b57cec5SDimitry Andric     return StmtError();
5270b57cec5SDimitry Andric   }
5280b57cec5SDimitry Andric 
5290b57cec5SDimitry Andric   auto *CS = CaseStmt::Create(Context, LHSVal.get(), RHSVal.get(),
5300b57cec5SDimitry Andric                               CaseLoc, DotDotDotLoc, ColonLoc);
5310b57cec5SDimitry Andric   getCurFunction()->SwitchStack.back().getPointer()->addSwitchCase(CS);
5320b57cec5SDimitry Andric   return CS;
5330b57cec5SDimitry Andric }
5340b57cec5SDimitry Andric 
5350b57cec5SDimitry Andric /// ActOnCaseStmtBody - This installs a statement as the body of a case.
5360b57cec5SDimitry Andric void Sema::ActOnCaseStmtBody(Stmt *S, Stmt *SubStmt) {
5370b57cec5SDimitry Andric   cast<CaseStmt>(S)->setSubStmt(SubStmt);
5380b57cec5SDimitry Andric }
5390b57cec5SDimitry Andric 
5400b57cec5SDimitry Andric StmtResult
5410b57cec5SDimitry Andric Sema::ActOnDefaultStmt(SourceLocation DefaultLoc, SourceLocation ColonLoc,
5420b57cec5SDimitry Andric                        Stmt *SubStmt, Scope *CurScope) {
5430b57cec5SDimitry Andric   if (getCurFunction()->SwitchStack.empty()) {
5440b57cec5SDimitry Andric     Diag(DefaultLoc, diag::err_default_not_in_switch);
5450b57cec5SDimitry Andric     return SubStmt;
5460b57cec5SDimitry Andric   }
5470b57cec5SDimitry Andric 
5480b57cec5SDimitry Andric   DefaultStmt *DS = new (Context) DefaultStmt(DefaultLoc, ColonLoc, SubStmt);
5490b57cec5SDimitry Andric   getCurFunction()->SwitchStack.back().getPointer()->addSwitchCase(DS);
5500b57cec5SDimitry Andric   return DS;
5510b57cec5SDimitry Andric }
5520b57cec5SDimitry Andric 
5530b57cec5SDimitry Andric StmtResult
5540b57cec5SDimitry Andric Sema::ActOnLabelStmt(SourceLocation IdentLoc, LabelDecl *TheDecl,
5550b57cec5SDimitry Andric                      SourceLocation ColonLoc, Stmt *SubStmt) {
5560b57cec5SDimitry Andric   // If the label was multiply defined, reject it now.
5570b57cec5SDimitry Andric   if (TheDecl->getStmt()) {
5580b57cec5SDimitry Andric     Diag(IdentLoc, diag::err_redefinition_of_label) << TheDecl->getDeclName();
5590b57cec5SDimitry Andric     Diag(TheDecl->getLocation(), diag::note_previous_definition);
5600b57cec5SDimitry Andric     return SubStmt;
5610b57cec5SDimitry Andric   }
5620b57cec5SDimitry Andric 
563fe6060f1SDimitry Andric   ReservedIdentifierStatus Status = TheDecl->isReserved(getLangOpts());
564349cc55cSDimitry Andric   if (isReservedInAllContexts(Status) &&
565fe6060f1SDimitry Andric       !Context.getSourceManager().isInSystemHeader(IdentLoc))
566fe6060f1SDimitry Andric     Diag(IdentLoc, diag::warn_reserved_extern_symbol)
567fe6060f1SDimitry Andric         << TheDecl << static_cast<int>(Status);
568fe6060f1SDimitry Andric 
5690b57cec5SDimitry Andric   // Otherwise, things are good.  Fill in the declaration and return it.
5700b57cec5SDimitry Andric   LabelStmt *LS = new (Context) LabelStmt(IdentLoc, TheDecl, SubStmt);
5710b57cec5SDimitry Andric   TheDecl->setStmt(LS);
5720b57cec5SDimitry Andric   if (!TheDecl->isGnuLocal()) {
5730b57cec5SDimitry Andric     TheDecl->setLocStart(IdentLoc);
5740b57cec5SDimitry Andric     if (!TheDecl->isMSAsmLabel()) {
5750b57cec5SDimitry Andric       // Don't update the location of MS ASM labels.  These will result in
5760b57cec5SDimitry Andric       // a diagnostic, and changing the location here will mess that up.
5770b57cec5SDimitry Andric       TheDecl->setLocation(IdentLoc);
5780b57cec5SDimitry Andric     }
5790b57cec5SDimitry Andric   }
5800b57cec5SDimitry Andric   return LS;
5810b57cec5SDimitry Andric }
5820b57cec5SDimitry Andric 
583fe6060f1SDimitry Andric StmtResult Sema::BuildAttributedStmt(SourceLocation AttrsLoc,
5840b57cec5SDimitry Andric                                      ArrayRef<const Attr *> Attrs,
5850b57cec5SDimitry Andric                                      Stmt *SubStmt) {
586fe6060f1SDimitry Andric   // FIXME: this code should move when a planned refactoring around statement
587fe6060f1SDimitry Andric   // attributes lands.
588fe6060f1SDimitry Andric   for (const auto *A : Attrs) {
589fe6060f1SDimitry Andric     if (A->getKind() == attr::MustTail) {
590fe6060f1SDimitry Andric       if (!checkAndRewriteMustTailAttr(SubStmt, *A)) {
591fe6060f1SDimitry Andric         return SubStmt;
592fe6060f1SDimitry Andric       }
593fe6060f1SDimitry Andric       setFunctionHasMustTail();
594fe6060f1SDimitry Andric     }
595fe6060f1SDimitry Andric   }
596fe6060f1SDimitry Andric 
597fe6060f1SDimitry Andric   return AttributedStmt::Create(Context, AttrsLoc, Attrs, SubStmt);
598fe6060f1SDimitry Andric }
599fe6060f1SDimitry Andric 
60081ad6265SDimitry Andric StmtResult Sema::ActOnAttributedStmt(const ParsedAttributes &Attrs,
601fe6060f1SDimitry Andric                                      Stmt *SubStmt) {
602fe6060f1SDimitry Andric   SmallVector<const Attr *, 1> SemanticAttrs;
603fe6060f1SDimitry Andric   ProcessStmtAttributes(SubStmt, Attrs, SemanticAttrs);
604fe6060f1SDimitry Andric   if (!SemanticAttrs.empty())
605fe6060f1SDimitry Andric     return BuildAttributedStmt(Attrs.Range.getBegin(), SemanticAttrs, SubStmt);
606fe6060f1SDimitry Andric   // If none of the attributes applied, that's fine, we can recover by
607fe6060f1SDimitry Andric   // returning the substatement directly instead of making an AttributedStmt
608fe6060f1SDimitry Andric   // with no attributes on it.
609fe6060f1SDimitry Andric   return SubStmt;
610fe6060f1SDimitry Andric }
611fe6060f1SDimitry Andric 
612fe6060f1SDimitry Andric bool Sema::checkAndRewriteMustTailAttr(Stmt *St, const Attr &MTA) {
613fe6060f1SDimitry Andric   ReturnStmt *R = cast<ReturnStmt>(St);
614fe6060f1SDimitry Andric   Expr *E = R->getRetValue();
615fe6060f1SDimitry Andric 
616fe6060f1SDimitry Andric   if (CurContext->isDependentContext() || (E && E->isInstantiationDependent()))
617fe6060f1SDimitry Andric     // We have to suspend our check until template instantiation time.
618fe6060f1SDimitry Andric     return true;
619fe6060f1SDimitry Andric 
620fe6060f1SDimitry Andric   if (!checkMustTailAttr(St, MTA))
621fe6060f1SDimitry Andric     return false;
622fe6060f1SDimitry Andric 
623fe6060f1SDimitry Andric   // FIXME: Replace Expr::IgnoreImplicitAsWritten() with this function.
624fe6060f1SDimitry Andric   // Currently it does not skip implicit constructors in an initialization
625fe6060f1SDimitry Andric   // context.
626fe6060f1SDimitry Andric   auto IgnoreImplicitAsWritten = [](Expr *E) -> Expr * {
627fe6060f1SDimitry Andric     return IgnoreExprNodes(E, IgnoreImplicitAsWrittenSingleStep,
628fe6060f1SDimitry Andric                            IgnoreElidableImplicitConstructorSingleStep);
629fe6060f1SDimitry Andric   };
630fe6060f1SDimitry Andric 
631fe6060f1SDimitry Andric   // Now that we have verified that 'musttail' is valid here, rewrite the
632fe6060f1SDimitry Andric   // return value to remove all implicit nodes, but retain parentheses.
633fe6060f1SDimitry Andric   R->setRetValue(IgnoreImplicitAsWritten(E));
634fe6060f1SDimitry Andric   return true;
635fe6060f1SDimitry Andric }
636fe6060f1SDimitry Andric 
637fe6060f1SDimitry Andric bool Sema::checkMustTailAttr(const Stmt *St, const Attr &MTA) {
638fe6060f1SDimitry Andric   assert(!CurContext->isDependentContext() &&
639fe6060f1SDimitry Andric          "musttail cannot be checked from a dependent context");
640fe6060f1SDimitry Andric 
641fe6060f1SDimitry Andric   // FIXME: Add Expr::IgnoreParenImplicitAsWritten() with this definition.
642fe6060f1SDimitry Andric   auto IgnoreParenImplicitAsWritten = [](const Expr *E) -> const Expr * {
643fe6060f1SDimitry Andric     return IgnoreExprNodes(const_cast<Expr *>(E), IgnoreParensSingleStep,
644fe6060f1SDimitry Andric                            IgnoreImplicitAsWrittenSingleStep,
645fe6060f1SDimitry Andric                            IgnoreElidableImplicitConstructorSingleStep);
646fe6060f1SDimitry Andric   };
647fe6060f1SDimitry Andric 
648fe6060f1SDimitry Andric   const Expr *E = cast<ReturnStmt>(St)->getRetValue();
649fe6060f1SDimitry Andric   const auto *CE = dyn_cast_or_null<CallExpr>(IgnoreParenImplicitAsWritten(E));
650fe6060f1SDimitry Andric 
651fe6060f1SDimitry Andric   if (!CE) {
652fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_needs_call) << &MTA;
653fe6060f1SDimitry Andric     return false;
654fe6060f1SDimitry Andric   }
655fe6060f1SDimitry Andric 
656fe6060f1SDimitry Andric   if (const auto *EWC = dyn_cast<ExprWithCleanups>(E)) {
657fe6060f1SDimitry Andric     if (EWC->cleanupsHaveSideEffects()) {
658fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_needs_trivial_args) << &MTA;
659fe6060f1SDimitry Andric       return false;
660fe6060f1SDimitry Andric     }
661fe6060f1SDimitry Andric   }
662fe6060f1SDimitry Andric 
663fe6060f1SDimitry Andric   // We need to determine the full function type (including "this" type, if any)
664fe6060f1SDimitry Andric   // for both caller and callee.
665fe6060f1SDimitry Andric   struct FuncType {
666fe6060f1SDimitry Andric     enum {
667fe6060f1SDimitry Andric       ft_non_member,
668fe6060f1SDimitry Andric       ft_static_member,
669fe6060f1SDimitry Andric       ft_non_static_member,
670fe6060f1SDimitry Andric       ft_pointer_to_member,
671fe6060f1SDimitry Andric     } MemberType = ft_non_member;
672fe6060f1SDimitry Andric 
673fe6060f1SDimitry Andric     QualType This;
674fe6060f1SDimitry Andric     const FunctionProtoType *Func;
675fe6060f1SDimitry Andric     const CXXMethodDecl *Method = nullptr;
676fe6060f1SDimitry Andric   } CallerType, CalleeType;
677fe6060f1SDimitry Andric 
678fe6060f1SDimitry Andric   auto GetMethodType = [this, St, MTA](const CXXMethodDecl *CMD, FuncType &Type,
679fe6060f1SDimitry Andric                                        bool IsCallee) -> bool {
680fe6060f1SDimitry Andric     if (isa<CXXConstructorDecl, CXXDestructorDecl>(CMD)) {
681fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_structors_forbidden)
682fe6060f1SDimitry Andric           << IsCallee << isa<CXXDestructorDecl>(CMD);
683fe6060f1SDimitry Andric       if (IsCallee)
684fe6060f1SDimitry Andric         Diag(CMD->getBeginLoc(), diag::note_musttail_structors_forbidden)
685fe6060f1SDimitry Andric             << isa<CXXDestructorDecl>(CMD);
686fe6060f1SDimitry Andric       Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
687fe6060f1SDimitry Andric       return false;
688fe6060f1SDimitry Andric     }
689fe6060f1SDimitry Andric     if (CMD->isStatic())
690fe6060f1SDimitry Andric       Type.MemberType = FuncType::ft_static_member;
691fe6060f1SDimitry Andric     else {
6925f757f3fSDimitry Andric       Type.This = CMD->getFunctionObjectParameterType();
693fe6060f1SDimitry Andric       Type.MemberType = FuncType::ft_non_static_member;
694fe6060f1SDimitry Andric     }
695fe6060f1SDimitry Andric     Type.Func = CMD->getType()->castAs<FunctionProtoType>();
696fe6060f1SDimitry Andric     return true;
697fe6060f1SDimitry Andric   };
698fe6060f1SDimitry Andric 
699fe6060f1SDimitry Andric   const auto *CallerDecl = dyn_cast<FunctionDecl>(CurContext);
700fe6060f1SDimitry Andric 
701fe6060f1SDimitry Andric   // Find caller function signature.
702fe6060f1SDimitry Andric   if (!CallerDecl) {
703fe6060f1SDimitry Andric     int ContextType;
704fe6060f1SDimitry Andric     if (isa<BlockDecl>(CurContext))
705fe6060f1SDimitry Andric       ContextType = 0;
706fe6060f1SDimitry Andric     else if (isa<ObjCMethodDecl>(CurContext))
707fe6060f1SDimitry Andric       ContextType = 1;
708fe6060f1SDimitry Andric     else
709fe6060f1SDimitry Andric       ContextType = 2;
710fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_forbidden_from_this_context)
711fe6060f1SDimitry Andric         << &MTA << ContextType;
712fe6060f1SDimitry Andric     return false;
713fe6060f1SDimitry Andric   } else if (const auto *CMD = dyn_cast<CXXMethodDecl>(CurContext)) {
714fe6060f1SDimitry Andric     // Caller is a class/struct method.
715fe6060f1SDimitry Andric     if (!GetMethodType(CMD, CallerType, false))
716fe6060f1SDimitry Andric       return false;
717fe6060f1SDimitry Andric   } else {
718fe6060f1SDimitry Andric     // Caller is a non-method function.
719fe6060f1SDimitry Andric     CallerType.Func = CallerDecl->getType()->getAs<FunctionProtoType>();
720fe6060f1SDimitry Andric   }
721fe6060f1SDimitry Andric 
722fe6060f1SDimitry Andric   const Expr *CalleeExpr = CE->getCallee()->IgnoreParens();
723fe6060f1SDimitry Andric   const auto *CalleeBinOp = dyn_cast<BinaryOperator>(CalleeExpr);
724fe6060f1SDimitry Andric   SourceLocation CalleeLoc = CE->getCalleeDecl()
725fe6060f1SDimitry Andric                                  ? CE->getCalleeDecl()->getBeginLoc()
726fe6060f1SDimitry Andric                                  : St->getBeginLoc();
727fe6060f1SDimitry Andric 
728fe6060f1SDimitry Andric   // Find callee function signature.
729fe6060f1SDimitry Andric   if (const CXXMethodDecl *CMD =
730fe6060f1SDimitry Andric           dyn_cast_or_null<CXXMethodDecl>(CE->getCalleeDecl())) {
731fe6060f1SDimitry Andric     // Call is: obj.method(), obj->method(), functor(), etc.
732fe6060f1SDimitry Andric     if (!GetMethodType(CMD, CalleeType, true))
733fe6060f1SDimitry Andric       return false;
734fe6060f1SDimitry Andric   } else if (CalleeBinOp && CalleeBinOp->isPtrMemOp()) {
735fe6060f1SDimitry Andric     // Call is: obj->*method_ptr or obj.*method_ptr
736fe6060f1SDimitry Andric     const auto *MPT =
737fe6060f1SDimitry Andric         CalleeBinOp->getRHS()->getType()->castAs<MemberPointerType>();
738fe6060f1SDimitry Andric     CalleeType.This = QualType(MPT->getClass(), 0);
739fe6060f1SDimitry Andric     CalleeType.Func = MPT->getPointeeType()->castAs<FunctionProtoType>();
740fe6060f1SDimitry Andric     CalleeType.MemberType = FuncType::ft_pointer_to_member;
741fe6060f1SDimitry Andric   } else if (isa<CXXPseudoDestructorExpr>(CalleeExpr)) {
742fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_structors_forbidden)
743fe6060f1SDimitry Andric         << /* IsCallee = */ 1 << /* IsDestructor = */ 1;
744fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
745fe6060f1SDimitry Andric     return false;
746fe6060f1SDimitry Andric   } else {
747fe6060f1SDimitry Andric     // Non-method function.
748fe6060f1SDimitry Andric     CalleeType.Func =
749fe6060f1SDimitry Andric         CalleeExpr->getType()->getPointeeType()->getAs<FunctionProtoType>();
750fe6060f1SDimitry Andric   }
751fe6060f1SDimitry Andric 
752fe6060f1SDimitry Andric   // Both caller and callee must have a prototype (no K&R declarations).
753fe6060f1SDimitry Andric   if (!CalleeType.Func || !CallerType.Func) {
754fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_needs_prototype) << &MTA;
755fe6060f1SDimitry Andric     if (!CalleeType.Func && CE->getDirectCallee()) {
756fe6060f1SDimitry Andric       Diag(CE->getDirectCallee()->getBeginLoc(),
757fe6060f1SDimitry Andric            diag::note_musttail_fix_non_prototype);
758fe6060f1SDimitry Andric     }
759fe6060f1SDimitry Andric     if (!CallerType.Func)
760fe6060f1SDimitry Andric       Diag(CallerDecl->getBeginLoc(), diag::note_musttail_fix_non_prototype);
761fe6060f1SDimitry Andric     return false;
762fe6060f1SDimitry Andric   }
763fe6060f1SDimitry Andric 
764fe6060f1SDimitry Andric   // Caller and callee must have matching calling conventions.
765fe6060f1SDimitry Andric   //
766fe6060f1SDimitry Andric   // Some calling conventions are physically capable of supporting tail calls
767fe6060f1SDimitry Andric   // even if the function types don't perfectly match. LLVM is currently too
768fe6060f1SDimitry Andric   // strict to allow this, but if LLVM added support for this in the future, we
769fe6060f1SDimitry Andric   // could exit early here and skip the remaining checks if the functions are
770fe6060f1SDimitry Andric   // using such a calling convention.
771fe6060f1SDimitry Andric   if (CallerType.Func->getCallConv() != CalleeType.Func->getCallConv()) {
772fe6060f1SDimitry Andric     if (const auto *ND = dyn_cast_or_null<NamedDecl>(CE->getCalleeDecl()))
773fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_callconv_mismatch)
774fe6060f1SDimitry Andric           << true << ND->getDeclName();
775fe6060f1SDimitry Andric     else
776fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_callconv_mismatch) << false;
777fe6060f1SDimitry Andric     Diag(CalleeLoc, diag::note_musttail_callconv_mismatch)
778fe6060f1SDimitry Andric         << FunctionType::getNameForCallConv(CallerType.Func->getCallConv())
779fe6060f1SDimitry Andric         << FunctionType::getNameForCallConv(CalleeType.Func->getCallConv());
780fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
781fe6060f1SDimitry Andric     return false;
782fe6060f1SDimitry Andric   }
783fe6060f1SDimitry Andric 
784fe6060f1SDimitry Andric   if (CalleeType.Func->isVariadic() || CallerType.Func->isVariadic()) {
785fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_no_variadic) << &MTA;
786fe6060f1SDimitry Andric     return false;
787fe6060f1SDimitry Andric   }
788fe6060f1SDimitry Andric 
789*7a6dacacSDimitry Andric   const auto *CalleeDecl = CE->getCalleeDecl();
790*7a6dacacSDimitry Andric   if (CalleeDecl && CalleeDecl->hasAttr<CXX11NoReturnAttr>()) {
791*7a6dacacSDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_no_return) << &MTA;
792*7a6dacacSDimitry Andric     return false;
793*7a6dacacSDimitry Andric   }
794*7a6dacacSDimitry Andric 
795fe6060f1SDimitry Andric   // Caller and callee must match in whether they have a "this" parameter.
796fe6060f1SDimitry Andric   if (CallerType.This.isNull() != CalleeType.This.isNull()) {
797fe6060f1SDimitry Andric     if (const auto *ND = dyn_cast_or_null<NamedDecl>(CE->getCalleeDecl())) {
798fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_member_mismatch)
799fe6060f1SDimitry Andric           << CallerType.MemberType << CalleeType.MemberType << true
800fe6060f1SDimitry Andric           << ND->getDeclName();
801fe6060f1SDimitry Andric       Diag(CalleeLoc, diag::note_musttail_callee_defined_here)
802fe6060f1SDimitry Andric           << ND->getDeclName();
803fe6060f1SDimitry Andric     } else
804fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_member_mismatch)
805fe6060f1SDimitry Andric           << CallerType.MemberType << CalleeType.MemberType << false;
806fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
807fe6060f1SDimitry Andric     return false;
808fe6060f1SDimitry Andric   }
809fe6060f1SDimitry Andric 
810fe6060f1SDimitry Andric   auto CheckTypesMatch = [this](FuncType CallerType, FuncType CalleeType,
811fe6060f1SDimitry Andric                                 PartialDiagnostic &PD) -> bool {
812fe6060f1SDimitry Andric     enum {
813fe6060f1SDimitry Andric       ft_different_class,
814fe6060f1SDimitry Andric       ft_parameter_arity,
815fe6060f1SDimitry Andric       ft_parameter_mismatch,
816fe6060f1SDimitry Andric       ft_return_type,
817fe6060f1SDimitry Andric     };
818fe6060f1SDimitry Andric 
819fe6060f1SDimitry Andric     auto DoTypesMatch = [this, &PD](QualType A, QualType B,
820fe6060f1SDimitry Andric                                     unsigned Select) -> bool {
821fe6060f1SDimitry Andric       if (!Context.hasSimilarType(A, B)) {
822fe6060f1SDimitry Andric         PD << Select << A.getUnqualifiedType() << B.getUnqualifiedType();
823fe6060f1SDimitry Andric         return false;
824fe6060f1SDimitry Andric       }
825fe6060f1SDimitry Andric       return true;
826fe6060f1SDimitry Andric     };
827fe6060f1SDimitry Andric 
828fe6060f1SDimitry Andric     if (!CallerType.This.isNull() &&
829fe6060f1SDimitry Andric         !DoTypesMatch(CallerType.This, CalleeType.This, ft_different_class))
830fe6060f1SDimitry Andric       return false;
831fe6060f1SDimitry Andric 
832fe6060f1SDimitry Andric     if (!DoTypesMatch(CallerType.Func->getReturnType(),
833fe6060f1SDimitry Andric                       CalleeType.Func->getReturnType(), ft_return_type))
834fe6060f1SDimitry Andric       return false;
835fe6060f1SDimitry Andric 
836fe6060f1SDimitry Andric     if (CallerType.Func->getNumParams() != CalleeType.Func->getNumParams()) {
837fe6060f1SDimitry Andric       PD << ft_parameter_arity << CallerType.Func->getNumParams()
838fe6060f1SDimitry Andric          << CalleeType.Func->getNumParams();
839fe6060f1SDimitry Andric       return false;
840fe6060f1SDimitry Andric     }
841fe6060f1SDimitry Andric 
842fe6060f1SDimitry Andric     ArrayRef<QualType> CalleeParams = CalleeType.Func->getParamTypes();
843fe6060f1SDimitry Andric     ArrayRef<QualType> CallerParams = CallerType.Func->getParamTypes();
844fe6060f1SDimitry Andric     size_t N = CallerType.Func->getNumParams();
845fe6060f1SDimitry Andric     for (size_t I = 0; I < N; I++) {
846fe6060f1SDimitry Andric       if (!DoTypesMatch(CalleeParams[I], CallerParams[I],
847fe6060f1SDimitry Andric                         ft_parameter_mismatch)) {
848fe6060f1SDimitry Andric         PD << static_cast<int>(I) + 1;
849fe6060f1SDimitry Andric         return false;
850fe6060f1SDimitry Andric       }
851fe6060f1SDimitry Andric     }
852fe6060f1SDimitry Andric 
853fe6060f1SDimitry Andric     return true;
854fe6060f1SDimitry Andric   };
855fe6060f1SDimitry Andric 
856fe6060f1SDimitry Andric   PartialDiagnostic PD = PDiag(diag::note_musttail_mismatch);
857fe6060f1SDimitry Andric   if (!CheckTypesMatch(CallerType, CalleeType, PD)) {
858fe6060f1SDimitry Andric     if (const auto *ND = dyn_cast_or_null<NamedDecl>(CE->getCalleeDecl()))
859fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_mismatch)
860fe6060f1SDimitry Andric           << true << ND->getDeclName();
861fe6060f1SDimitry Andric     else
862fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_mismatch) << false;
863fe6060f1SDimitry Andric     Diag(CalleeLoc, PD);
864fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
865fe6060f1SDimitry Andric     return false;
866fe6060f1SDimitry Andric   }
867fe6060f1SDimitry Andric 
868fe6060f1SDimitry Andric   return true;
8690b57cec5SDimitry Andric }
8700b57cec5SDimitry Andric 
8710b57cec5SDimitry Andric namespace {
8720b57cec5SDimitry Andric class CommaVisitor : public EvaluatedExprVisitor<CommaVisitor> {
8730b57cec5SDimitry Andric   typedef EvaluatedExprVisitor<CommaVisitor> Inherited;
8740b57cec5SDimitry Andric   Sema &SemaRef;
8750b57cec5SDimitry Andric public:
8760b57cec5SDimitry Andric   CommaVisitor(Sema &SemaRef) : Inherited(SemaRef.Context), SemaRef(SemaRef) {}
8770b57cec5SDimitry Andric   void VisitBinaryOperator(BinaryOperator *E) {
8780b57cec5SDimitry Andric     if (E->getOpcode() == BO_Comma)
8790b57cec5SDimitry Andric       SemaRef.DiagnoseCommaOperator(E->getLHS(), E->getExprLoc());
8800b57cec5SDimitry Andric     EvaluatedExprVisitor<CommaVisitor>::VisitBinaryOperator(E);
8810b57cec5SDimitry Andric   }
8820b57cec5SDimitry Andric };
8830b57cec5SDimitry Andric }
8840b57cec5SDimitry Andric 
885349cc55cSDimitry Andric StmtResult Sema::ActOnIfStmt(SourceLocation IfLoc,
886349cc55cSDimitry Andric                              IfStatementKind StatementKind,
887e8d8bef9SDimitry Andric                              SourceLocation LParenLoc, Stmt *InitStmt,
888e8d8bef9SDimitry Andric                              ConditionResult Cond, SourceLocation RParenLoc,
8890b57cec5SDimitry Andric                              Stmt *thenStmt, SourceLocation ElseLoc,
8900b57cec5SDimitry Andric                              Stmt *elseStmt) {
8910b57cec5SDimitry Andric   if (Cond.isInvalid())
89204eeddc0SDimitry Andric     return StmtError();
8930b57cec5SDimitry Andric 
894349cc55cSDimitry Andric   bool ConstevalOrNegatedConsteval =
895349cc55cSDimitry Andric       StatementKind == IfStatementKind::ConstevalNonNegated ||
896349cc55cSDimitry Andric       StatementKind == IfStatementKind::ConstevalNegated;
897349cc55cSDimitry Andric 
8980b57cec5SDimitry Andric   Expr *CondExpr = Cond.get().second;
899349cc55cSDimitry Andric   assert((CondExpr || ConstevalOrNegatedConsteval) &&
900349cc55cSDimitry Andric          "If statement: missing condition");
9010b57cec5SDimitry Andric   // Only call the CommaVisitor when not C89 due to differences in scope flags.
902349cc55cSDimitry Andric   if (CondExpr && (getLangOpts().C99 || getLangOpts().CPlusPlus) &&
9030b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, CondExpr->getExprLoc()))
9040b57cec5SDimitry Andric     CommaVisitor(*this).Visit(CondExpr);
9050b57cec5SDimitry Andric 
906349cc55cSDimitry Andric   if (!ConstevalOrNegatedConsteval && !elseStmt)
90781ad6265SDimitry Andric     DiagnoseEmptyStmtBody(RParenLoc, thenStmt, diag::warn_empty_if_body);
9080b57cec5SDimitry Andric 
909349cc55cSDimitry Andric   if (ConstevalOrNegatedConsteval ||
910349cc55cSDimitry Andric       StatementKind == IfStatementKind::Constexpr) {
911e8d8bef9SDimitry Andric     auto DiagnoseLikelihood = [&](const Stmt *S) {
912e8d8bef9SDimitry Andric       if (const Attr *A = Stmt::getLikelihoodAttr(S)) {
913e8d8bef9SDimitry Andric         Diags.Report(A->getLocation(),
914349cc55cSDimitry Andric                      diag::warn_attribute_has_no_effect_on_compile_time_if)
915349cc55cSDimitry Andric             << A << ConstevalOrNegatedConsteval << A->getRange();
916e8d8bef9SDimitry Andric         Diags.Report(IfLoc,
917349cc55cSDimitry Andric                      diag::note_attribute_has_no_effect_on_compile_time_if_here)
918349cc55cSDimitry Andric             << ConstevalOrNegatedConsteval
919349cc55cSDimitry Andric             << SourceRange(IfLoc, (ConstevalOrNegatedConsteval
920349cc55cSDimitry Andric                                        ? thenStmt->getBeginLoc()
921349cc55cSDimitry Andric                                        : LParenLoc)
922349cc55cSDimitry Andric                                       .getLocWithOffset(-1));
923e8d8bef9SDimitry Andric       }
924e8d8bef9SDimitry Andric     };
925e8d8bef9SDimitry Andric     DiagnoseLikelihood(thenStmt);
926e8d8bef9SDimitry Andric     DiagnoseLikelihood(elseStmt);
927e8d8bef9SDimitry Andric   } else {
928e8d8bef9SDimitry Andric     std::tuple<bool, const Attr *, const Attr *> LHC =
929e8d8bef9SDimitry Andric         Stmt::determineLikelihoodConflict(thenStmt, elseStmt);
930e8d8bef9SDimitry Andric     if (std::get<0>(LHC)) {
931e8d8bef9SDimitry Andric       const Attr *ThenAttr = std::get<1>(LHC);
932e8d8bef9SDimitry Andric       const Attr *ElseAttr = std::get<2>(LHC);
933e8d8bef9SDimitry Andric       Diags.Report(ThenAttr->getLocation(),
934e8d8bef9SDimitry Andric                    diag::warn_attributes_likelihood_ifstmt_conflict)
935e8d8bef9SDimitry Andric           << ThenAttr << ThenAttr->getRange();
936e8d8bef9SDimitry Andric       Diags.Report(ElseAttr->getLocation(), diag::note_conflicting_attribute)
937e8d8bef9SDimitry Andric           << ElseAttr << ElseAttr->getRange();
938e8d8bef9SDimitry Andric     }
939e8d8bef9SDimitry Andric   }
940e8d8bef9SDimitry Andric 
941349cc55cSDimitry Andric   if (ConstevalOrNegatedConsteval) {
94206c3fb27SDimitry Andric     bool Immediate = ExprEvalContexts.back().Context ==
94306c3fb27SDimitry Andric                      ExpressionEvaluationContext::ImmediateFunctionContext;
944349cc55cSDimitry Andric     if (CurContext->isFunctionOrMethod()) {
945349cc55cSDimitry Andric       const auto *FD =
946349cc55cSDimitry Andric           dyn_cast<FunctionDecl>(Decl::castFromDeclContext(CurContext));
94706c3fb27SDimitry Andric       if (FD && FD->isImmediateFunction())
948349cc55cSDimitry Andric         Immediate = true;
949349cc55cSDimitry Andric     }
950349cc55cSDimitry Andric     if (isUnevaluatedContext() || Immediate)
951349cc55cSDimitry Andric       Diags.Report(IfLoc, diag::warn_consteval_if_always_true) << Immediate;
952349cc55cSDimitry Andric   }
953349cc55cSDimitry Andric 
954349cc55cSDimitry Andric   return BuildIfStmt(IfLoc, StatementKind, LParenLoc, InitStmt, Cond, RParenLoc,
955e8d8bef9SDimitry Andric                      thenStmt, ElseLoc, elseStmt);
9560b57cec5SDimitry Andric }
9570b57cec5SDimitry Andric 
958349cc55cSDimitry Andric StmtResult Sema::BuildIfStmt(SourceLocation IfLoc,
959349cc55cSDimitry Andric                              IfStatementKind StatementKind,
960e8d8bef9SDimitry Andric                              SourceLocation LParenLoc, Stmt *InitStmt,
961e8d8bef9SDimitry Andric                              ConditionResult Cond, SourceLocation RParenLoc,
9620b57cec5SDimitry Andric                              Stmt *thenStmt, SourceLocation ElseLoc,
9630b57cec5SDimitry Andric                              Stmt *elseStmt) {
9640b57cec5SDimitry Andric   if (Cond.isInvalid())
9650b57cec5SDimitry Andric     return StmtError();
9660b57cec5SDimitry Andric 
967349cc55cSDimitry Andric   if (StatementKind != IfStatementKind::Ordinary ||
968349cc55cSDimitry Andric       isa<ObjCAvailabilityCheckExpr>(Cond.get().second))
9690b57cec5SDimitry Andric     setFunctionHasBranchProtectedScope();
9700b57cec5SDimitry Andric 
971349cc55cSDimitry Andric   return IfStmt::Create(Context, IfLoc, StatementKind, InitStmt,
972349cc55cSDimitry Andric                         Cond.get().first, Cond.get().second, LParenLoc,
973349cc55cSDimitry Andric                         RParenLoc, thenStmt, ElseLoc, elseStmt);
9740b57cec5SDimitry Andric }
9750b57cec5SDimitry Andric 
9760b57cec5SDimitry Andric namespace {
9770b57cec5SDimitry Andric   struct CaseCompareFunctor {
9780b57cec5SDimitry Andric     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
9790b57cec5SDimitry Andric                     const llvm::APSInt &RHS) {
9800b57cec5SDimitry Andric       return LHS.first < RHS;
9810b57cec5SDimitry Andric     }
9820b57cec5SDimitry Andric     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
9830b57cec5SDimitry Andric                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
9840b57cec5SDimitry Andric       return LHS.first < RHS.first;
9850b57cec5SDimitry Andric     }
9860b57cec5SDimitry Andric     bool operator()(const llvm::APSInt &LHS,
9870b57cec5SDimitry Andric                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
9880b57cec5SDimitry Andric       return LHS < RHS.first;
9890b57cec5SDimitry Andric     }
9900b57cec5SDimitry Andric   };
9910b57cec5SDimitry Andric }
9920b57cec5SDimitry Andric 
9930b57cec5SDimitry Andric /// CmpCaseVals - Comparison predicate for sorting case values.
9940b57cec5SDimitry Andric ///
9950b57cec5SDimitry Andric static bool CmpCaseVals(const std::pair<llvm::APSInt, CaseStmt*>& lhs,
9960b57cec5SDimitry Andric                         const std::pair<llvm::APSInt, CaseStmt*>& rhs) {
9970b57cec5SDimitry Andric   if (lhs.first < rhs.first)
9980b57cec5SDimitry Andric     return true;
9990b57cec5SDimitry Andric 
10000b57cec5SDimitry Andric   if (lhs.first == rhs.first &&
1001e8d8bef9SDimitry Andric       lhs.second->getCaseLoc() < rhs.second->getCaseLoc())
10020b57cec5SDimitry Andric     return true;
10030b57cec5SDimitry Andric   return false;
10040b57cec5SDimitry Andric }
10050b57cec5SDimitry Andric 
10060b57cec5SDimitry Andric /// CmpEnumVals - Comparison predicate for sorting enumeration values.
10070b57cec5SDimitry Andric ///
10080b57cec5SDimitry Andric static bool CmpEnumVals(const std::pair<llvm::APSInt, EnumConstantDecl*>& lhs,
10090b57cec5SDimitry Andric                         const std::pair<llvm::APSInt, EnumConstantDecl*>& rhs)
10100b57cec5SDimitry Andric {
10110b57cec5SDimitry Andric   return lhs.first < rhs.first;
10120b57cec5SDimitry Andric }
10130b57cec5SDimitry Andric 
10140b57cec5SDimitry Andric /// EqEnumVals - Comparison preficate for uniqing enumeration values.
10150b57cec5SDimitry Andric ///
10160b57cec5SDimitry Andric static bool EqEnumVals(const std::pair<llvm::APSInt, EnumConstantDecl*>& lhs,
10170b57cec5SDimitry Andric                        const std::pair<llvm::APSInt, EnumConstantDecl*>& rhs)
10180b57cec5SDimitry Andric {
10190b57cec5SDimitry Andric   return lhs.first == rhs.first;
10200b57cec5SDimitry Andric }
10210b57cec5SDimitry Andric 
10220b57cec5SDimitry Andric /// GetTypeBeforeIntegralPromotion - Returns the pre-promotion type of
10230b57cec5SDimitry Andric /// potentially integral-promoted expression @p expr.
10240b57cec5SDimitry Andric static QualType GetTypeBeforeIntegralPromotion(const Expr *&E) {
10250b57cec5SDimitry Andric   if (const auto *FE = dyn_cast<FullExpr>(E))
10260b57cec5SDimitry Andric     E = FE->getSubExpr();
10270b57cec5SDimitry Andric   while (const auto *ImpCast = dyn_cast<ImplicitCastExpr>(E)) {
10280b57cec5SDimitry Andric     if (ImpCast->getCastKind() != CK_IntegralCast) break;
10290b57cec5SDimitry Andric     E = ImpCast->getSubExpr();
10300b57cec5SDimitry Andric   }
10310b57cec5SDimitry Andric   return E->getType();
10320b57cec5SDimitry Andric }
10330b57cec5SDimitry Andric 
10340b57cec5SDimitry Andric ExprResult Sema::CheckSwitchCondition(SourceLocation SwitchLoc, Expr *Cond) {
10350b57cec5SDimitry Andric   class SwitchConvertDiagnoser : public ICEConvertDiagnoser {
10360b57cec5SDimitry Andric     Expr *Cond;
10370b57cec5SDimitry Andric 
10380b57cec5SDimitry Andric   public:
10390b57cec5SDimitry Andric     SwitchConvertDiagnoser(Expr *Cond)
10400b57cec5SDimitry Andric         : ICEConvertDiagnoser(/*AllowScopedEnumerations*/true, false, true),
10410b57cec5SDimitry Andric           Cond(Cond) {}
10420b57cec5SDimitry Andric 
10430b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseNotInt(Sema &S, SourceLocation Loc,
10440b57cec5SDimitry Andric                                          QualType T) override {
10450b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_typecheck_statement_requires_integer) << T;
10460b57cec5SDimitry Andric     }
10470b57cec5SDimitry Andric 
10480b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseIncomplete(
10490b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T) override {
10500b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_incomplete_class_type)
10510b57cec5SDimitry Andric                << T << Cond->getSourceRange();
10520b57cec5SDimitry Andric     }
10530b57cec5SDimitry Andric 
10540b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseExplicitConv(
10550b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
10560b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_explicit_conversion) << T << ConvTy;
10570b57cec5SDimitry Andric     }
10580b57cec5SDimitry Andric 
10590b57cec5SDimitry Andric     SemaDiagnosticBuilder noteExplicitConv(
10600b57cec5SDimitry Andric         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
10610b57cec5SDimitry Andric       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
10620b57cec5SDimitry Andric         << ConvTy->isEnumeralType() << ConvTy;
10630b57cec5SDimitry Andric     }
10640b57cec5SDimitry Andric 
10650b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseAmbiguous(Sema &S, SourceLocation Loc,
10660b57cec5SDimitry Andric                                             QualType T) override {
10670b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_multiple_conversions) << T;
10680b57cec5SDimitry Andric     }
10690b57cec5SDimitry Andric 
10700b57cec5SDimitry Andric     SemaDiagnosticBuilder noteAmbiguous(
10710b57cec5SDimitry Andric         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
10720b57cec5SDimitry Andric       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
10730b57cec5SDimitry Andric       << ConvTy->isEnumeralType() << ConvTy;
10740b57cec5SDimitry Andric     }
10750b57cec5SDimitry Andric 
10760b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseConversion(
10770b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
10780b57cec5SDimitry Andric       llvm_unreachable("conversion functions are permitted");
10790b57cec5SDimitry Andric     }
10800b57cec5SDimitry Andric   } SwitchDiagnoser(Cond);
10810b57cec5SDimitry Andric 
10820b57cec5SDimitry Andric   ExprResult CondResult =
10830b57cec5SDimitry Andric       PerformContextualImplicitConversion(SwitchLoc, Cond, SwitchDiagnoser);
10840b57cec5SDimitry Andric   if (CondResult.isInvalid())
10850b57cec5SDimitry Andric     return ExprError();
10860b57cec5SDimitry Andric 
10870b57cec5SDimitry Andric   // FIXME: PerformContextualImplicitConversion doesn't always tell us if it
10880b57cec5SDimitry Andric   // failed and produced a diagnostic.
10890b57cec5SDimitry Andric   Cond = CondResult.get();
10900b57cec5SDimitry Andric   if (!Cond->isTypeDependent() &&
10910b57cec5SDimitry Andric       !Cond->getType()->isIntegralOrEnumerationType())
10920b57cec5SDimitry Andric     return ExprError();
10930b57cec5SDimitry Andric 
10940b57cec5SDimitry Andric   // C99 6.8.4.2p5 - Integer promotions are performed on the controlling expr.
10950b57cec5SDimitry Andric   return UsualUnaryConversions(Cond);
10960b57cec5SDimitry Andric }
10970b57cec5SDimitry Andric 
10980b57cec5SDimitry Andric StmtResult Sema::ActOnStartOfSwitchStmt(SourceLocation SwitchLoc,
1099e8d8bef9SDimitry Andric                                         SourceLocation LParenLoc,
1100e8d8bef9SDimitry Andric                                         Stmt *InitStmt, ConditionResult Cond,
1101e8d8bef9SDimitry Andric                                         SourceLocation RParenLoc) {
11020b57cec5SDimitry Andric   Expr *CondExpr = Cond.get().second;
11030b57cec5SDimitry Andric   assert((Cond.isInvalid() || CondExpr) && "switch with no condition");
11040b57cec5SDimitry Andric 
11050b57cec5SDimitry Andric   if (CondExpr && !CondExpr->isTypeDependent()) {
11060b57cec5SDimitry Andric     // We have already converted the expression to an integral or enumeration
11075ffd83dbSDimitry Andric     // type, when we parsed the switch condition. There are cases where we don't
11085ffd83dbSDimitry Andric     // have an appropriate type, e.g. a typo-expr Cond was corrected to an
11095ffd83dbSDimitry Andric     // inappropriate-type expr, we just return an error.
11105ffd83dbSDimitry Andric     if (!CondExpr->getType()->isIntegralOrEnumerationType())
11115ffd83dbSDimitry Andric       return StmtError();
11120b57cec5SDimitry Andric     if (CondExpr->isKnownToHaveBooleanValue()) {
11130b57cec5SDimitry Andric       // switch(bool_expr) {...} is often a programmer error, e.g.
11140b57cec5SDimitry Andric       //   switch(n && mask) { ... }  // Doh - should be "n & mask".
11150b57cec5SDimitry Andric       // One can always use an if statement instead of switch(bool_expr).
11160b57cec5SDimitry Andric       Diag(SwitchLoc, diag::warn_bool_switch_condition)
11170b57cec5SDimitry Andric           << CondExpr->getSourceRange();
11180b57cec5SDimitry Andric     }
11190b57cec5SDimitry Andric   }
11200b57cec5SDimitry Andric 
11210b57cec5SDimitry Andric   setFunctionHasBranchIntoScope();
11220b57cec5SDimitry Andric 
1123e8d8bef9SDimitry Andric   auto *SS = SwitchStmt::Create(Context, InitStmt, Cond.get().first, CondExpr,
1124e8d8bef9SDimitry Andric                                 LParenLoc, RParenLoc);
11250b57cec5SDimitry Andric   getCurFunction()->SwitchStack.push_back(
11260b57cec5SDimitry Andric       FunctionScopeInfo::SwitchInfo(SS, false));
11270b57cec5SDimitry Andric   return SS;
11280b57cec5SDimitry Andric }
11290b57cec5SDimitry Andric 
11300b57cec5SDimitry Andric static void AdjustAPSInt(llvm::APSInt &Val, unsigned BitWidth, bool IsSigned) {
11310b57cec5SDimitry Andric   Val = Val.extOrTrunc(BitWidth);
11320b57cec5SDimitry Andric   Val.setIsSigned(IsSigned);
11330b57cec5SDimitry Andric }
11340b57cec5SDimitry Andric 
11350b57cec5SDimitry Andric /// Check the specified case value is in range for the given unpromoted switch
11360b57cec5SDimitry Andric /// type.
11370b57cec5SDimitry Andric static void checkCaseValue(Sema &S, SourceLocation Loc, const llvm::APSInt &Val,
11380b57cec5SDimitry Andric                            unsigned UnpromotedWidth, bool UnpromotedSign) {
11390b57cec5SDimitry Andric   // In C++11 onwards, this is checked by the language rules.
11400b57cec5SDimitry Andric   if (S.getLangOpts().CPlusPlus11)
11410b57cec5SDimitry Andric     return;
11420b57cec5SDimitry Andric 
11430b57cec5SDimitry Andric   // If the case value was signed and negative and the switch expression is
11440b57cec5SDimitry Andric   // unsigned, don't bother to warn: this is implementation-defined behavior.
11450b57cec5SDimitry Andric   // FIXME: Introduce a second, default-ignored warning for this case?
11460b57cec5SDimitry Andric   if (UnpromotedWidth < Val.getBitWidth()) {
11470b57cec5SDimitry Andric     llvm::APSInt ConvVal(Val);
11480b57cec5SDimitry Andric     AdjustAPSInt(ConvVal, UnpromotedWidth, UnpromotedSign);
11490b57cec5SDimitry Andric     AdjustAPSInt(ConvVal, Val.getBitWidth(), Val.isSigned());
11500b57cec5SDimitry Andric     // FIXME: Use different diagnostics for overflow  in conversion to promoted
11510b57cec5SDimitry Andric     // type versus "switch expression cannot have this value". Use proper
11520b57cec5SDimitry Andric     // IntRange checking rather than just looking at the unpromoted type here.
11530b57cec5SDimitry Andric     if (ConvVal != Val)
1154fe6060f1SDimitry Andric       S.Diag(Loc, diag::warn_case_value_overflow) << toString(Val, 10)
1155fe6060f1SDimitry Andric                                                   << toString(ConvVal, 10);
11560b57cec5SDimitry Andric   }
11570b57cec5SDimitry Andric }
11580b57cec5SDimitry Andric 
11590b57cec5SDimitry Andric typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl*>, 64> EnumValsTy;
11600b57cec5SDimitry Andric 
11610b57cec5SDimitry Andric /// Returns true if we should emit a diagnostic about this case expression not
11620b57cec5SDimitry Andric /// being a part of the enum used in the switch controlling expression.
11630b57cec5SDimitry Andric static bool ShouldDiagnoseSwitchCaseNotInEnum(const Sema &S,
11640b57cec5SDimitry Andric                                               const EnumDecl *ED,
11650b57cec5SDimitry Andric                                               const Expr *CaseExpr,
11660b57cec5SDimitry Andric                                               EnumValsTy::iterator &EI,
11670b57cec5SDimitry Andric                                               EnumValsTy::iterator &EIEnd,
11680b57cec5SDimitry Andric                                               const llvm::APSInt &Val) {
11690b57cec5SDimitry Andric   if (!ED->isClosed())
11700b57cec5SDimitry Andric     return false;
11710b57cec5SDimitry Andric 
11720b57cec5SDimitry Andric   if (const DeclRefExpr *DRE =
11730b57cec5SDimitry Andric           dyn_cast<DeclRefExpr>(CaseExpr->IgnoreParenImpCasts())) {
11740b57cec5SDimitry Andric     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
11750b57cec5SDimitry Andric       QualType VarType = VD->getType();
11760b57cec5SDimitry Andric       QualType EnumType = S.Context.getTypeDeclType(ED);
11770b57cec5SDimitry Andric       if (VD->hasGlobalStorage() && VarType.isConstQualified() &&
11780b57cec5SDimitry Andric           S.Context.hasSameUnqualifiedType(EnumType, VarType))
11790b57cec5SDimitry Andric         return false;
11800b57cec5SDimitry Andric     }
11810b57cec5SDimitry Andric   }
11820b57cec5SDimitry Andric 
11830b57cec5SDimitry Andric   if (ED->hasAttr<FlagEnumAttr>())
11840b57cec5SDimitry Andric     return !S.IsValueInFlagEnum(ED, Val, false);
11850b57cec5SDimitry Andric 
11860b57cec5SDimitry Andric   while (EI != EIEnd && EI->first < Val)
11870b57cec5SDimitry Andric     EI++;
11880b57cec5SDimitry Andric 
11890b57cec5SDimitry Andric   if (EI != EIEnd && EI->first == Val)
11900b57cec5SDimitry Andric     return false;
11910b57cec5SDimitry Andric 
11920b57cec5SDimitry Andric   return true;
11930b57cec5SDimitry Andric }
11940b57cec5SDimitry Andric 
11950b57cec5SDimitry Andric static void checkEnumTypesInSwitchStmt(Sema &S, const Expr *Cond,
11960b57cec5SDimitry Andric                                        const Expr *Case) {
11970b57cec5SDimitry Andric   QualType CondType = Cond->getType();
11980b57cec5SDimitry Andric   QualType CaseType = Case->getType();
11990b57cec5SDimitry Andric 
12000b57cec5SDimitry Andric   const EnumType *CondEnumType = CondType->getAs<EnumType>();
12010b57cec5SDimitry Andric   const EnumType *CaseEnumType = CaseType->getAs<EnumType>();
12020b57cec5SDimitry Andric   if (!CondEnumType || !CaseEnumType)
12030b57cec5SDimitry Andric     return;
12040b57cec5SDimitry Andric 
12050b57cec5SDimitry Andric   // Ignore anonymous enums.
12060b57cec5SDimitry Andric   if (!CondEnumType->getDecl()->getIdentifier() &&
12070b57cec5SDimitry Andric       !CondEnumType->getDecl()->getTypedefNameForAnonDecl())
12080b57cec5SDimitry Andric     return;
12090b57cec5SDimitry Andric   if (!CaseEnumType->getDecl()->getIdentifier() &&
12100b57cec5SDimitry Andric       !CaseEnumType->getDecl()->getTypedefNameForAnonDecl())
12110b57cec5SDimitry Andric     return;
12120b57cec5SDimitry Andric 
12130b57cec5SDimitry Andric   if (S.Context.hasSameUnqualifiedType(CondType, CaseType))
12140b57cec5SDimitry Andric     return;
12150b57cec5SDimitry Andric 
12160b57cec5SDimitry Andric   S.Diag(Case->getExprLoc(), diag::warn_comparison_of_mixed_enum_types_switch)
12170b57cec5SDimitry Andric       << CondType << CaseType << Cond->getSourceRange()
12180b57cec5SDimitry Andric       << Case->getSourceRange();
12190b57cec5SDimitry Andric }
12200b57cec5SDimitry Andric 
12210b57cec5SDimitry Andric StmtResult
12220b57cec5SDimitry Andric Sema::ActOnFinishSwitchStmt(SourceLocation SwitchLoc, Stmt *Switch,
12230b57cec5SDimitry Andric                             Stmt *BodyStmt) {
12240b57cec5SDimitry Andric   SwitchStmt *SS = cast<SwitchStmt>(Switch);
12250b57cec5SDimitry Andric   bool CaseListIsIncomplete = getCurFunction()->SwitchStack.back().getInt();
12260b57cec5SDimitry Andric   assert(SS == getCurFunction()->SwitchStack.back().getPointer() &&
12270b57cec5SDimitry Andric          "switch stack missing push/pop!");
12280b57cec5SDimitry Andric 
12290b57cec5SDimitry Andric   getCurFunction()->SwitchStack.pop_back();
12300b57cec5SDimitry Andric 
12310b57cec5SDimitry Andric   if (!BodyStmt) return StmtError();
12320b57cec5SDimitry Andric   SS->setBody(BodyStmt, SwitchLoc);
12330b57cec5SDimitry Andric 
12340b57cec5SDimitry Andric   Expr *CondExpr = SS->getCond();
12350b57cec5SDimitry Andric   if (!CondExpr) return StmtError();
12360b57cec5SDimitry Andric 
12370b57cec5SDimitry Andric   QualType CondType = CondExpr->getType();
12380b57cec5SDimitry Andric 
12390b57cec5SDimitry Andric   // C++ 6.4.2.p2:
12400b57cec5SDimitry Andric   // Integral promotions are performed (on the switch condition).
12410b57cec5SDimitry Andric   //
12420b57cec5SDimitry Andric   // A case value unrepresentable by the original switch condition
12430b57cec5SDimitry Andric   // type (before the promotion) doesn't make sense, even when it can
12440b57cec5SDimitry Andric   // be represented by the promoted type.  Therefore we need to find
12450b57cec5SDimitry Andric   // the pre-promotion type of the switch condition.
12460b57cec5SDimitry Andric   const Expr *CondExprBeforePromotion = CondExpr;
12470b57cec5SDimitry Andric   QualType CondTypeBeforePromotion =
12480b57cec5SDimitry Andric       GetTypeBeforeIntegralPromotion(CondExprBeforePromotion);
12490b57cec5SDimitry Andric 
12500b57cec5SDimitry Andric   // Get the bitwidth of the switched-on value after promotions. We must
12510b57cec5SDimitry Andric   // convert the integer case values to this width before comparison.
12520b57cec5SDimitry Andric   bool HasDependentValue
12530b57cec5SDimitry Andric     = CondExpr->isTypeDependent() || CondExpr->isValueDependent();
12540b57cec5SDimitry Andric   unsigned CondWidth = HasDependentValue ? 0 : Context.getIntWidth(CondType);
12550b57cec5SDimitry Andric   bool CondIsSigned = CondType->isSignedIntegerOrEnumerationType();
12560b57cec5SDimitry Andric 
12570b57cec5SDimitry Andric   // Get the width and signedness that the condition might actually have, for
12580b57cec5SDimitry Andric   // warning purposes.
12590b57cec5SDimitry Andric   // FIXME: Grab an IntRange for the condition rather than using the unpromoted
12600b57cec5SDimitry Andric   // type.
12610b57cec5SDimitry Andric   unsigned CondWidthBeforePromotion
12620b57cec5SDimitry Andric     = HasDependentValue ? 0 : Context.getIntWidth(CondTypeBeforePromotion);
12630b57cec5SDimitry Andric   bool CondIsSignedBeforePromotion
12640b57cec5SDimitry Andric     = CondTypeBeforePromotion->isSignedIntegerOrEnumerationType();
12650b57cec5SDimitry Andric 
12660b57cec5SDimitry Andric   // Accumulate all of the case values in a vector so that we can sort them
12670b57cec5SDimitry Andric   // and detect duplicates.  This vector contains the APInt for the case after
12680b57cec5SDimitry Andric   // it has been converted to the condition type.
12690b57cec5SDimitry Andric   typedef SmallVector<std::pair<llvm::APSInt, CaseStmt*>, 64> CaseValsTy;
12700b57cec5SDimitry Andric   CaseValsTy CaseVals;
12710b57cec5SDimitry Andric 
12720b57cec5SDimitry Andric   // Keep track of any GNU case ranges we see.  The APSInt is the low value.
12730b57cec5SDimitry Andric   typedef std::vector<std::pair<llvm::APSInt, CaseStmt*> > CaseRangesTy;
12740b57cec5SDimitry Andric   CaseRangesTy CaseRanges;
12750b57cec5SDimitry Andric 
12760b57cec5SDimitry Andric   DefaultStmt *TheDefaultStmt = nullptr;
12770b57cec5SDimitry Andric 
12780b57cec5SDimitry Andric   bool CaseListIsErroneous = false;
12790b57cec5SDimitry Andric 
1280cb14a3feSDimitry Andric   // FIXME: We'd better diagnose missing or duplicate default labels even
1281cb14a3feSDimitry Andric   // in the dependent case. Because default labels themselves are never
1282cb14a3feSDimitry Andric   // dependent.
12830b57cec5SDimitry Andric   for (SwitchCase *SC = SS->getSwitchCaseList(); SC && !HasDependentValue;
12840b57cec5SDimitry Andric        SC = SC->getNextSwitchCase()) {
12850b57cec5SDimitry Andric 
12860b57cec5SDimitry Andric     if (DefaultStmt *DS = dyn_cast<DefaultStmt>(SC)) {
12870b57cec5SDimitry Andric       if (TheDefaultStmt) {
12880b57cec5SDimitry Andric         Diag(DS->getDefaultLoc(), diag::err_multiple_default_labels_defined);
12890b57cec5SDimitry Andric         Diag(TheDefaultStmt->getDefaultLoc(), diag::note_duplicate_case_prev);
12900b57cec5SDimitry Andric 
12910b57cec5SDimitry Andric         // FIXME: Remove the default statement from the switch block so that
12920b57cec5SDimitry Andric         // we'll return a valid AST.  This requires recursing down the AST and
12930b57cec5SDimitry Andric         // finding it, not something we are set up to do right now.  For now,
12940b57cec5SDimitry Andric         // just lop the entire switch stmt out of the AST.
12950b57cec5SDimitry Andric         CaseListIsErroneous = true;
12960b57cec5SDimitry Andric       }
12970b57cec5SDimitry Andric       TheDefaultStmt = DS;
12980b57cec5SDimitry Andric 
12990b57cec5SDimitry Andric     } else {
13000b57cec5SDimitry Andric       CaseStmt *CS = cast<CaseStmt>(SC);
13010b57cec5SDimitry Andric 
13020b57cec5SDimitry Andric       Expr *Lo = CS->getLHS();
13030b57cec5SDimitry Andric 
13040b57cec5SDimitry Andric       if (Lo->isValueDependent()) {
13050b57cec5SDimitry Andric         HasDependentValue = true;
13060b57cec5SDimitry Andric         break;
13070b57cec5SDimitry Andric       }
13080b57cec5SDimitry Andric 
13090b57cec5SDimitry Andric       // We already verified that the expression has a constant value;
13100b57cec5SDimitry Andric       // get that value (prior to conversions).
13110b57cec5SDimitry Andric       const Expr *LoBeforePromotion = Lo;
13120b57cec5SDimitry Andric       GetTypeBeforeIntegralPromotion(LoBeforePromotion);
13130b57cec5SDimitry Andric       llvm::APSInt LoVal = LoBeforePromotion->EvaluateKnownConstInt(Context);
13140b57cec5SDimitry Andric 
13150b57cec5SDimitry Andric       // Check the unconverted value is within the range of possible values of
13160b57cec5SDimitry Andric       // the switch expression.
13170b57cec5SDimitry Andric       checkCaseValue(*this, Lo->getBeginLoc(), LoVal, CondWidthBeforePromotion,
13180b57cec5SDimitry Andric                      CondIsSignedBeforePromotion);
13190b57cec5SDimitry Andric 
13200b57cec5SDimitry Andric       // FIXME: This duplicates the check performed for warn_not_in_enum below.
13210b57cec5SDimitry Andric       checkEnumTypesInSwitchStmt(*this, CondExprBeforePromotion,
13220b57cec5SDimitry Andric                                  LoBeforePromotion);
13230b57cec5SDimitry Andric 
13240b57cec5SDimitry Andric       // Convert the value to the same width/sign as the condition.
13250b57cec5SDimitry Andric       AdjustAPSInt(LoVal, CondWidth, CondIsSigned);
13260b57cec5SDimitry Andric 
13270b57cec5SDimitry Andric       // If this is a case range, remember it in CaseRanges, otherwise CaseVals.
13280b57cec5SDimitry Andric       if (CS->getRHS()) {
13290b57cec5SDimitry Andric         if (CS->getRHS()->isValueDependent()) {
13300b57cec5SDimitry Andric           HasDependentValue = true;
13310b57cec5SDimitry Andric           break;
13320b57cec5SDimitry Andric         }
13330b57cec5SDimitry Andric         CaseRanges.push_back(std::make_pair(LoVal, CS));
13340b57cec5SDimitry Andric       } else
13350b57cec5SDimitry Andric         CaseVals.push_back(std::make_pair(LoVal, CS));
13360b57cec5SDimitry Andric     }
13370b57cec5SDimitry Andric   }
13380b57cec5SDimitry Andric 
13390b57cec5SDimitry Andric   if (!HasDependentValue) {
13400b57cec5SDimitry Andric     // If we don't have a default statement, check whether the
13410b57cec5SDimitry Andric     // condition is constant.
13420b57cec5SDimitry Andric     llvm::APSInt ConstantCondValue;
13430b57cec5SDimitry Andric     bool HasConstantCond = false;
1344a7dea167SDimitry Andric     if (!TheDefaultStmt) {
13450b57cec5SDimitry Andric       Expr::EvalResult Result;
13460b57cec5SDimitry Andric       HasConstantCond = CondExpr->EvaluateAsInt(Result, Context,
13470b57cec5SDimitry Andric                                                 Expr::SE_AllowSideEffects);
13480b57cec5SDimitry Andric       if (Result.Val.isInt())
13490b57cec5SDimitry Andric         ConstantCondValue = Result.Val.getInt();
13500b57cec5SDimitry Andric       assert(!HasConstantCond ||
13510b57cec5SDimitry Andric              (ConstantCondValue.getBitWidth() == CondWidth &&
13520b57cec5SDimitry Andric               ConstantCondValue.isSigned() == CondIsSigned));
1353cb14a3feSDimitry Andric       Diag(SwitchLoc, diag::warn_switch_default);
13540b57cec5SDimitry Andric     }
13550b57cec5SDimitry Andric     bool ShouldCheckConstantCond = HasConstantCond;
13560b57cec5SDimitry Andric 
13570b57cec5SDimitry Andric     // Sort all the scalar case values so we can easily detect duplicates.
13580b57cec5SDimitry Andric     llvm::stable_sort(CaseVals, CmpCaseVals);
13590b57cec5SDimitry Andric 
13600b57cec5SDimitry Andric     if (!CaseVals.empty()) {
13610b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseVals.size(); i != e; ++i) {
13620b57cec5SDimitry Andric         if (ShouldCheckConstantCond &&
13630b57cec5SDimitry Andric             CaseVals[i].first == ConstantCondValue)
13640b57cec5SDimitry Andric           ShouldCheckConstantCond = false;
13650b57cec5SDimitry Andric 
13660b57cec5SDimitry Andric         if (i != 0 && CaseVals[i].first == CaseVals[i-1].first) {
13670b57cec5SDimitry Andric           // If we have a duplicate, report it.
13680b57cec5SDimitry Andric           // First, determine if either case value has a name
13690b57cec5SDimitry Andric           StringRef PrevString, CurrString;
13700b57cec5SDimitry Andric           Expr *PrevCase = CaseVals[i-1].second->getLHS()->IgnoreParenCasts();
13710b57cec5SDimitry Andric           Expr *CurrCase = CaseVals[i].second->getLHS()->IgnoreParenCasts();
13720b57cec5SDimitry Andric           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(PrevCase)) {
13730b57cec5SDimitry Andric             PrevString = DeclRef->getDecl()->getName();
13740b57cec5SDimitry Andric           }
13750b57cec5SDimitry Andric           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(CurrCase)) {
13760b57cec5SDimitry Andric             CurrString = DeclRef->getDecl()->getName();
13770b57cec5SDimitry Andric           }
13780b57cec5SDimitry Andric           SmallString<16> CaseValStr;
13790b57cec5SDimitry Andric           CaseVals[i-1].first.toString(CaseValStr);
13800b57cec5SDimitry Andric 
13810b57cec5SDimitry Andric           if (PrevString == CurrString)
13820b57cec5SDimitry Andric             Diag(CaseVals[i].second->getLHS()->getBeginLoc(),
13830b57cec5SDimitry Andric                  diag::err_duplicate_case)
1384fe6060f1SDimitry Andric                 << (PrevString.empty() ? CaseValStr.str() : PrevString);
13850b57cec5SDimitry Andric           else
13860b57cec5SDimitry Andric             Diag(CaseVals[i].second->getLHS()->getBeginLoc(),
13870b57cec5SDimitry Andric                  diag::err_duplicate_case_differing_expr)
1388fe6060f1SDimitry Andric                 << (PrevString.empty() ? CaseValStr.str() : PrevString)
1389fe6060f1SDimitry Andric                 << (CurrString.empty() ? CaseValStr.str() : CurrString)
13900b57cec5SDimitry Andric                 << CaseValStr;
13910b57cec5SDimitry Andric 
13920b57cec5SDimitry Andric           Diag(CaseVals[i - 1].second->getLHS()->getBeginLoc(),
13930b57cec5SDimitry Andric                diag::note_duplicate_case_prev);
13940b57cec5SDimitry Andric           // FIXME: We really want to remove the bogus case stmt from the
13950b57cec5SDimitry Andric           // substmt, but we have no way to do this right now.
13960b57cec5SDimitry Andric           CaseListIsErroneous = true;
13970b57cec5SDimitry Andric         }
13980b57cec5SDimitry Andric       }
13990b57cec5SDimitry Andric     }
14000b57cec5SDimitry Andric 
14010b57cec5SDimitry Andric     // Detect duplicate case ranges, which usually don't exist at all in
14020b57cec5SDimitry Andric     // the first place.
14030b57cec5SDimitry Andric     if (!CaseRanges.empty()) {
14040b57cec5SDimitry Andric       // Sort all the case ranges by their low value so we can easily detect
14050b57cec5SDimitry Andric       // overlaps between ranges.
14060b57cec5SDimitry Andric       llvm::stable_sort(CaseRanges);
14070b57cec5SDimitry Andric 
14080b57cec5SDimitry Andric       // Scan the ranges, computing the high values and removing empty ranges.
14090b57cec5SDimitry Andric       std::vector<llvm::APSInt> HiVals;
14100b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
14110b57cec5SDimitry Andric         llvm::APSInt &LoVal = CaseRanges[i].first;
14120b57cec5SDimitry Andric         CaseStmt *CR = CaseRanges[i].second;
14130b57cec5SDimitry Andric         Expr *Hi = CR->getRHS();
14140b57cec5SDimitry Andric 
14150b57cec5SDimitry Andric         const Expr *HiBeforePromotion = Hi;
14160b57cec5SDimitry Andric         GetTypeBeforeIntegralPromotion(HiBeforePromotion);
14170b57cec5SDimitry Andric         llvm::APSInt HiVal = HiBeforePromotion->EvaluateKnownConstInt(Context);
14180b57cec5SDimitry Andric 
14190b57cec5SDimitry Andric         // Check the unconverted value is within the range of possible values of
14200b57cec5SDimitry Andric         // the switch expression.
14210b57cec5SDimitry Andric         checkCaseValue(*this, Hi->getBeginLoc(), HiVal,
14220b57cec5SDimitry Andric                        CondWidthBeforePromotion, CondIsSignedBeforePromotion);
14230b57cec5SDimitry Andric 
14240b57cec5SDimitry Andric         // Convert the value to the same width/sign as the condition.
14250b57cec5SDimitry Andric         AdjustAPSInt(HiVal, CondWidth, CondIsSigned);
14260b57cec5SDimitry Andric 
14270b57cec5SDimitry Andric         // If the low value is bigger than the high value, the case is empty.
14280b57cec5SDimitry Andric         if (LoVal > HiVal) {
14290b57cec5SDimitry Andric           Diag(CR->getLHS()->getBeginLoc(), diag::warn_case_empty_range)
14300b57cec5SDimitry Andric               << SourceRange(CR->getLHS()->getBeginLoc(), Hi->getEndLoc());
14310b57cec5SDimitry Andric           CaseRanges.erase(CaseRanges.begin()+i);
14320b57cec5SDimitry Andric           --i;
14330b57cec5SDimitry Andric           --e;
14340b57cec5SDimitry Andric           continue;
14350b57cec5SDimitry Andric         }
14360b57cec5SDimitry Andric 
14370b57cec5SDimitry Andric         if (ShouldCheckConstantCond &&
14380b57cec5SDimitry Andric             LoVal <= ConstantCondValue &&
14390b57cec5SDimitry Andric             ConstantCondValue <= HiVal)
14400b57cec5SDimitry Andric           ShouldCheckConstantCond = false;
14410b57cec5SDimitry Andric 
14420b57cec5SDimitry Andric         HiVals.push_back(HiVal);
14430b57cec5SDimitry Andric       }
14440b57cec5SDimitry Andric 
14450b57cec5SDimitry Andric       // Rescan the ranges, looking for overlap with singleton values and other
14460b57cec5SDimitry Andric       // ranges.  Since the range list is sorted, we only need to compare case
14470b57cec5SDimitry Andric       // ranges with their neighbors.
14480b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
14490b57cec5SDimitry Andric         llvm::APSInt &CRLo = CaseRanges[i].first;
14500b57cec5SDimitry Andric         llvm::APSInt &CRHi = HiVals[i];
14510b57cec5SDimitry Andric         CaseStmt *CR = CaseRanges[i].second;
14520b57cec5SDimitry Andric 
14530b57cec5SDimitry Andric         // Check to see whether the case range overlaps with any
14540b57cec5SDimitry Andric         // singleton cases.
14550b57cec5SDimitry Andric         CaseStmt *OverlapStmt = nullptr;
14560b57cec5SDimitry Andric         llvm::APSInt OverlapVal(32);
14570b57cec5SDimitry Andric 
14580b57cec5SDimitry Andric         // Find the smallest value >= the lower bound.  If I is in the
14590b57cec5SDimitry Andric         // case range, then we have overlap.
14600b57cec5SDimitry Andric         CaseValsTy::iterator I =
14610b57cec5SDimitry Andric             llvm::lower_bound(CaseVals, CRLo, CaseCompareFunctor());
14620b57cec5SDimitry Andric         if (I != CaseVals.end() && I->first < CRHi) {
14630b57cec5SDimitry Andric           OverlapVal  = I->first;   // Found overlap with scalar.
14640b57cec5SDimitry Andric           OverlapStmt = I->second;
14650b57cec5SDimitry Andric         }
14660b57cec5SDimitry Andric 
14670b57cec5SDimitry Andric         // Find the smallest value bigger than the upper bound.
14680b57cec5SDimitry Andric         I = std::upper_bound(I, CaseVals.end(), CRHi, CaseCompareFunctor());
14690b57cec5SDimitry Andric         if (I != CaseVals.begin() && (I-1)->first >= CRLo) {
14700b57cec5SDimitry Andric           OverlapVal  = (I-1)->first;      // Found overlap with scalar.
14710b57cec5SDimitry Andric           OverlapStmt = (I-1)->second;
14720b57cec5SDimitry Andric         }
14730b57cec5SDimitry Andric 
14740b57cec5SDimitry Andric         // Check to see if this case stmt overlaps with the subsequent
14750b57cec5SDimitry Andric         // case range.
14760b57cec5SDimitry Andric         if (i && CRLo <= HiVals[i-1]) {
14770b57cec5SDimitry Andric           OverlapVal  = HiVals[i-1];       // Found overlap with range.
14780b57cec5SDimitry Andric           OverlapStmt = CaseRanges[i-1].second;
14790b57cec5SDimitry Andric         }
14800b57cec5SDimitry Andric 
14810b57cec5SDimitry Andric         if (OverlapStmt) {
14820b57cec5SDimitry Andric           // If we have a duplicate, report it.
14830b57cec5SDimitry Andric           Diag(CR->getLHS()->getBeginLoc(), diag::err_duplicate_case)
1484fe6060f1SDimitry Andric               << toString(OverlapVal, 10);
14850b57cec5SDimitry Andric           Diag(OverlapStmt->getLHS()->getBeginLoc(),
14860b57cec5SDimitry Andric                diag::note_duplicate_case_prev);
14870b57cec5SDimitry Andric           // FIXME: We really want to remove the bogus case stmt from the
14880b57cec5SDimitry Andric           // substmt, but we have no way to do this right now.
14890b57cec5SDimitry Andric           CaseListIsErroneous = true;
14900b57cec5SDimitry Andric         }
14910b57cec5SDimitry Andric       }
14920b57cec5SDimitry Andric     }
14930b57cec5SDimitry Andric 
14940b57cec5SDimitry Andric     // Complain if we have a constant condition and we didn't find a match.
14950b57cec5SDimitry Andric     if (!CaseListIsErroneous && !CaseListIsIncomplete &&
14960b57cec5SDimitry Andric         ShouldCheckConstantCond) {
14970b57cec5SDimitry Andric       // TODO: it would be nice if we printed enums as enums, chars as
14980b57cec5SDimitry Andric       // chars, etc.
14990b57cec5SDimitry Andric       Diag(CondExpr->getExprLoc(), diag::warn_missing_case_for_condition)
1500fe6060f1SDimitry Andric         << toString(ConstantCondValue, 10)
15010b57cec5SDimitry Andric         << CondExpr->getSourceRange();
15020b57cec5SDimitry Andric     }
15030b57cec5SDimitry Andric 
15040b57cec5SDimitry Andric     // Check to see if switch is over an Enum and handles all of its
15050b57cec5SDimitry Andric     // values.  We only issue a warning if there is not 'default:', but
15060b57cec5SDimitry Andric     // we still do the analysis to preserve this information in the AST
15070b57cec5SDimitry Andric     // (which can be used by flow-based analyes).
15080b57cec5SDimitry Andric     //
15090b57cec5SDimitry Andric     const EnumType *ET = CondTypeBeforePromotion->getAs<EnumType>();
15100b57cec5SDimitry Andric 
15110b57cec5SDimitry Andric     // If switch has default case, then ignore it.
15120b57cec5SDimitry Andric     if (!CaseListIsErroneous && !CaseListIsIncomplete && !HasConstantCond &&
1513fe6060f1SDimitry Andric         ET && ET->getDecl()->isCompleteDefinition() &&
1514bdd1243dSDimitry Andric         !ET->getDecl()->enumerators().empty()) {
15150b57cec5SDimitry Andric       const EnumDecl *ED = ET->getDecl();
15160b57cec5SDimitry Andric       EnumValsTy EnumVals;
15170b57cec5SDimitry Andric 
15180b57cec5SDimitry Andric       // Gather all enum values, set their type and sort them,
15190b57cec5SDimitry Andric       // allowing easier comparison with CaseVals.
15200b57cec5SDimitry Andric       for (auto *EDI : ED->enumerators()) {
15210b57cec5SDimitry Andric         llvm::APSInt Val = EDI->getInitVal();
15220b57cec5SDimitry Andric         AdjustAPSInt(Val, CondWidth, CondIsSigned);
15230b57cec5SDimitry Andric         EnumVals.push_back(std::make_pair(Val, EDI));
15240b57cec5SDimitry Andric       }
15250b57cec5SDimitry Andric       llvm::stable_sort(EnumVals, CmpEnumVals);
15260b57cec5SDimitry Andric       auto EI = EnumVals.begin(), EIEnd =
15270b57cec5SDimitry Andric         std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
15280b57cec5SDimitry Andric 
15290b57cec5SDimitry Andric       // See which case values aren't in enum.
15300b57cec5SDimitry Andric       for (CaseValsTy::const_iterator CI = CaseVals.begin();
15310b57cec5SDimitry Andric           CI != CaseVals.end(); CI++) {
15320b57cec5SDimitry Andric         Expr *CaseExpr = CI->second->getLHS();
15330b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
15340b57cec5SDimitry Andric                                               CI->first))
15350b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
15360b57cec5SDimitry Andric             << CondTypeBeforePromotion;
15370b57cec5SDimitry Andric       }
15380b57cec5SDimitry Andric 
15390b57cec5SDimitry Andric       // See which of case ranges aren't in enum
15400b57cec5SDimitry Andric       EI = EnumVals.begin();
15410b57cec5SDimitry Andric       for (CaseRangesTy::const_iterator RI = CaseRanges.begin();
15420b57cec5SDimitry Andric           RI != CaseRanges.end(); RI++) {
15430b57cec5SDimitry Andric         Expr *CaseExpr = RI->second->getLHS();
15440b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
15450b57cec5SDimitry Andric                                               RI->first))
15460b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
15470b57cec5SDimitry Andric             << CondTypeBeforePromotion;
15480b57cec5SDimitry Andric 
15490b57cec5SDimitry Andric         llvm::APSInt Hi =
15500b57cec5SDimitry Andric           RI->second->getRHS()->EvaluateKnownConstInt(Context);
15510b57cec5SDimitry Andric         AdjustAPSInt(Hi, CondWidth, CondIsSigned);
15520b57cec5SDimitry Andric 
15530b57cec5SDimitry Andric         CaseExpr = RI->second->getRHS();
15540b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
15550b57cec5SDimitry Andric                                               Hi))
15560b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
15570b57cec5SDimitry Andric             << CondTypeBeforePromotion;
15580b57cec5SDimitry Andric       }
15590b57cec5SDimitry Andric 
15600b57cec5SDimitry Andric       // Check which enum vals aren't in switch
15610b57cec5SDimitry Andric       auto CI = CaseVals.begin();
15620b57cec5SDimitry Andric       auto RI = CaseRanges.begin();
15630b57cec5SDimitry Andric       bool hasCasesNotInSwitch = false;
15640b57cec5SDimitry Andric 
15650b57cec5SDimitry Andric       SmallVector<DeclarationName,8> UnhandledNames;
15660b57cec5SDimitry Andric 
15670b57cec5SDimitry Andric       for (EI = EnumVals.begin(); EI != EIEnd; EI++) {
15680b57cec5SDimitry Andric         // Don't warn about omitted unavailable EnumConstantDecls.
15690b57cec5SDimitry Andric         switch (EI->second->getAvailability()) {
15700b57cec5SDimitry Andric         case AR_Deprecated:
15710b57cec5SDimitry Andric           // Omitting a deprecated constant is ok; it should never materialize.
15720b57cec5SDimitry Andric         case AR_Unavailable:
15730b57cec5SDimitry Andric           continue;
15740b57cec5SDimitry Andric 
15750b57cec5SDimitry Andric         case AR_NotYetIntroduced:
15760b57cec5SDimitry Andric           // Partially available enum constants should be present. Note that we
15770b57cec5SDimitry Andric           // suppress -Wunguarded-availability diagnostics for such uses.
15780b57cec5SDimitry Andric         case AR_Available:
15790b57cec5SDimitry Andric           break;
15800b57cec5SDimitry Andric         }
15810b57cec5SDimitry Andric 
15820b57cec5SDimitry Andric         if (EI->second->hasAttr<UnusedAttr>())
15830b57cec5SDimitry Andric           continue;
15840b57cec5SDimitry Andric 
15850b57cec5SDimitry Andric         // Drop unneeded case values
15860b57cec5SDimitry Andric         while (CI != CaseVals.end() && CI->first < EI->first)
15870b57cec5SDimitry Andric           CI++;
15880b57cec5SDimitry Andric 
15890b57cec5SDimitry Andric         if (CI != CaseVals.end() && CI->first == EI->first)
15900b57cec5SDimitry Andric           continue;
15910b57cec5SDimitry Andric 
15920b57cec5SDimitry Andric         // Drop unneeded case ranges
15930b57cec5SDimitry Andric         for (; RI != CaseRanges.end(); RI++) {
15940b57cec5SDimitry Andric           llvm::APSInt Hi =
15950b57cec5SDimitry Andric             RI->second->getRHS()->EvaluateKnownConstInt(Context);
15960b57cec5SDimitry Andric           AdjustAPSInt(Hi, CondWidth, CondIsSigned);
15970b57cec5SDimitry Andric           if (EI->first <= Hi)
15980b57cec5SDimitry Andric             break;
15990b57cec5SDimitry Andric         }
16000b57cec5SDimitry Andric 
16010b57cec5SDimitry Andric         if (RI == CaseRanges.end() || EI->first < RI->first) {
16020b57cec5SDimitry Andric           hasCasesNotInSwitch = true;
16030b57cec5SDimitry Andric           UnhandledNames.push_back(EI->second->getDeclName());
16040b57cec5SDimitry Andric         }
16050b57cec5SDimitry Andric       }
16060b57cec5SDimitry Andric 
16070b57cec5SDimitry Andric       if (TheDefaultStmt && UnhandledNames.empty() && ED->isClosedNonFlag())
16080b57cec5SDimitry Andric         Diag(TheDefaultStmt->getDefaultLoc(), diag::warn_unreachable_default);
16090b57cec5SDimitry Andric 
16100b57cec5SDimitry Andric       // Produce a nice diagnostic if multiple values aren't handled.
16110b57cec5SDimitry Andric       if (!UnhandledNames.empty()) {
1612e8d8bef9SDimitry Andric         auto DB = Diag(CondExpr->getExprLoc(), TheDefaultStmt
1613e8d8bef9SDimitry Andric                                                    ? diag::warn_def_missing_case
16140b57cec5SDimitry Andric                                                    : diag::warn_missing_case)
1615349cc55cSDimitry Andric                   << CondExpr->getSourceRange() << (int)UnhandledNames.size();
16160b57cec5SDimitry Andric 
16170b57cec5SDimitry Andric         for (size_t I = 0, E = std::min(UnhandledNames.size(), (size_t)3);
16180b57cec5SDimitry Andric              I != E; ++I)
16190b57cec5SDimitry Andric           DB << UnhandledNames[I];
16200b57cec5SDimitry Andric       }
16210b57cec5SDimitry Andric 
16220b57cec5SDimitry Andric       if (!hasCasesNotInSwitch)
16230b57cec5SDimitry Andric         SS->setAllEnumCasesCovered();
16240b57cec5SDimitry Andric     }
16250b57cec5SDimitry Andric   }
16260b57cec5SDimitry Andric 
16270b57cec5SDimitry Andric   if (BodyStmt)
16280b57cec5SDimitry Andric     DiagnoseEmptyStmtBody(CondExpr->getEndLoc(), BodyStmt,
16290b57cec5SDimitry Andric                           diag::warn_empty_switch_body);
16300b57cec5SDimitry Andric 
16310b57cec5SDimitry Andric   // FIXME: If the case list was broken is some way, we don't have a good system
16320b57cec5SDimitry Andric   // to patch it up.  Instead, just return the whole substmt as broken.
16330b57cec5SDimitry Andric   if (CaseListIsErroneous)
16340b57cec5SDimitry Andric     return StmtError();
16350b57cec5SDimitry Andric 
16360b57cec5SDimitry Andric   return SS;
16370b57cec5SDimitry Andric }
16380b57cec5SDimitry Andric 
16390b57cec5SDimitry Andric void
16400b57cec5SDimitry Andric Sema::DiagnoseAssignmentEnum(QualType DstType, QualType SrcType,
16410b57cec5SDimitry Andric                              Expr *SrcExpr) {
16420b57cec5SDimitry Andric   if (Diags.isIgnored(diag::warn_not_in_enum_assignment, SrcExpr->getExprLoc()))
16430b57cec5SDimitry Andric     return;
16440b57cec5SDimitry Andric 
16450b57cec5SDimitry Andric   if (const EnumType *ET = DstType->getAs<EnumType>())
16460b57cec5SDimitry Andric     if (!Context.hasSameUnqualifiedType(SrcType, DstType) &&
16470b57cec5SDimitry Andric         SrcType->isIntegerType()) {
16480b57cec5SDimitry Andric       if (!SrcExpr->isTypeDependent() && !SrcExpr->isValueDependent() &&
16490b57cec5SDimitry Andric           SrcExpr->isIntegerConstantExpr(Context)) {
16500b57cec5SDimitry Andric         // Get the bitwidth of the enum value before promotions.
16510b57cec5SDimitry Andric         unsigned DstWidth = Context.getIntWidth(DstType);
16520b57cec5SDimitry Andric         bool DstIsSigned = DstType->isSignedIntegerOrEnumerationType();
16530b57cec5SDimitry Andric 
16540b57cec5SDimitry Andric         llvm::APSInt RhsVal = SrcExpr->EvaluateKnownConstInt(Context);
16550b57cec5SDimitry Andric         AdjustAPSInt(RhsVal, DstWidth, DstIsSigned);
16560b57cec5SDimitry Andric         const EnumDecl *ED = ET->getDecl();
16570b57cec5SDimitry Andric 
16580b57cec5SDimitry Andric         if (!ED->isClosed())
16590b57cec5SDimitry Andric           return;
16600b57cec5SDimitry Andric 
16610b57cec5SDimitry Andric         if (ED->hasAttr<FlagEnumAttr>()) {
16620b57cec5SDimitry Andric           if (!IsValueInFlagEnum(ED, RhsVal, true))
16630b57cec5SDimitry Andric             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
16640b57cec5SDimitry Andric               << DstType.getUnqualifiedType();
16650b57cec5SDimitry Andric         } else {
16660b57cec5SDimitry Andric           typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl *>, 64>
16670b57cec5SDimitry Andric               EnumValsTy;
16680b57cec5SDimitry Andric           EnumValsTy EnumVals;
16690b57cec5SDimitry Andric 
16700b57cec5SDimitry Andric           // Gather all enum values, set their type and sort them,
16710b57cec5SDimitry Andric           // allowing easier comparison with rhs constant.
16720b57cec5SDimitry Andric           for (auto *EDI : ED->enumerators()) {
16730b57cec5SDimitry Andric             llvm::APSInt Val = EDI->getInitVal();
16740b57cec5SDimitry Andric             AdjustAPSInt(Val, DstWidth, DstIsSigned);
16750b57cec5SDimitry Andric             EnumVals.push_back(std::make_pair(Val, EDI));
16760b57cec5SDimitry Andric           }
16770b57cec5SDimitry Andric           if (EnumVals.empty())
16780b57cec5SDimitry Andric             return;
16790b57cec5SDimitry Andric           llvm::stable_sort(EnumVals, CmpEnumVals);
16800b57cec5SDimitry Andric           EnumValsTy::iterator EIend =
16810b57cec5SDimitry Andric               std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
16820b57cec5SDimitry Andric 
16830b57cec5SDimitry Andric           // See which values aren't in the enum.
16840b57cec5SDimitry Andric           EnumValsTy::const_iterator EI = EnumVals.begin();
16850b57cec5SDimitry Andric           while (EI != EIend && EI->first < RhsVal)
16860b57cec5SDimitry Andric             EI++;
16870b57cec5SDimitry Andric           if (EI == EIend || EI->first != RhsVal) {
16880b57cec5SDimitry Andric             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
16890b57cec5SDimitry Andric                 << DstType.getUnqualifiedType();
16900b57cec5SDimitry Andric           }
16910b57cec5SDimitry Andric         }
16920b57cec5SDimitry Andric       }
16930b57cec5SDimitry Andric     }
16940b57cec5SDimitry Andric }
16950b57cec5SDimitry Andric 
16965ffd83dbSDimitry Andric StmtResult Sema::ActOnWhileStmt(SourceLocation WhileLoc,
16975ffd83dbSDimitry Andric                                 SourceLocation LParenLoc, ConditionResult Cond,
16985ffd83dbSDimitry Andric                                 SourceLocation RParenLoc, Stmt *Body) {
16990b57cec5SDimitry Andric   if (Cond.isInvalid())
17000b57cec5SDimitry Andric     return StmtError();
17010b57cec5SDimitry Andric 
17020b57cec5SDimitry Andric   auto CondVal = Cond.get();
17030b57cec5SDimitry Andric   CheckBreakContinueBinding(CondVal.second);
17040b57cec5SDimitry Andric 
17050b57cec5SDimitry Andric   if (CondVal.second &&
17060b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, CondVal.second->getExprLoc()))
17070b57cec5SDimitry Andric     CommaVisitor(*this).Visit(CondVal.second);
17080b57cec5SDimitry Andric 
17090b57cec5SDimitry Andric   if (isa<NullStmt>(Body))
17100b57cec5SDimitry Andric     getCurCompoundScope().setHasEmptyLoopBodies();
17110b57cec5SDimitry Andric 
17120b57cec5SDimitry Andric   return WhileStmt::Create(Context, CondVal.first, CondVal.second, Body,
17135ffd83dbSDimitry Andric                            WhileLoc, LParenLoc, RParenLoc);
17140b57cec5SDimitry Andric }
17150b57cec5SDimitry Andric 
17160b57cec5SDimitry Andric StmtResult
17170b57cec5SDimitry Andric Sema::ActOnDoStmt(SourceLocation DoLoc, Stmt *Body,
17180b57cec5SDimitry Andric                   SourceLocation WhileLoc, SourceLocation CondLParen,
17190b57cec5SDimitry Andric                   Expr *Cond, SourceLocation CondRParen) {
17200b57cec5SDimitry Andric   assert(Cond && "ActOnDoStmt(): missing expression");
17210b57cec5SDimitry Andric 
17220b57cec5SDimitry Andric   CheckBreakContinueBinding(Cond);
17230b57cec5SDimitry Andric   ExprResult CondResult = CheckBooleanCondition(DoLoc, Cond);
17240b57cec5SDimitry Andric   if (CondResult.isInvalid())
17250b57cec5SDimitry Andric     return StmtError();
17260b57cec5SDimitry Andric   Cond = CondResult.get();
17270b57cec5SDimitry Andric 
17280b57cec5SDimitry Andric   CondResult = ActOnFinishFullExpr(Cond, DoLoc, /*DiscardedValue*/ false);
17290b57cec5SDimitry Andric   if (CondResult.isInvalid())
17300b57cec5SDimitry Andric     return StmtError();
17310b57cec5SDimitry Andric   Cond = CondResult.get();
17320b57cec5SDimitry Andric 
17330b57cec5SDimitry Andric   // Only call the CommaVisitor for C89 due to differences in scope flags.
17340b57cec5SDimitry Andric   if (Cond && !getLangOpts().C99 && !getLangOpts().CPlusPlus &&
17350b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, Cond->getExprLoc()))
17360b57cec5SDimitry Andric     CommaVisitor(*this).Visit(Cond);
17370b57cec5SDimitry Andric 
17380b57cec5SDimitry Andric   return new (Context) DoStmt(Body, Cond, DoLoc, WhileLoc, CondRParen);
17390b57cec5SDimitry Andric }
17400b57cec5SDimitry Andric 
17410b57cec5SDimitry Andric namespace {
17420b57cec5SDimitry Andric   // Use SetVector since the diagnostic cares about the ordering of the Decl's.
174306c3fb27SDimitry Andric   using DeclSetVector = llvm::SmallSetVector<VarDecl *, 8>;
17440b57cec5SDimitry Andric 
17450b57cec5SDimitry Andric   // This visitor will traverse a conditional statement and store all
17460b57cec5SDimitry Andric   // the evaluated decls into a vector.  Simple is set to true if none
17470b57cec5SDimitry Andric   // of the excluded constructs are used.
17480b57cec5SDimitry Andric   class DeclExtractor : public EvaluatedExprVisitor<DeclExtractor> {
17490b57cec5SDimitry Andric     DeclSetVector &Decls;
17500b57cec5SDimitry Andric     SmallVectorImpl<SourceRange> &Ranges;
17510b57cec5SDimitry Andric     bool Simple;
17520b57cec5SDimitry Andric   public:
17530b57cec5SDimitry Andric     typedef EvaluatedExprVisitor<DeclExtractor> Inherited;
17540b57cec5SDimitry Andric 
17550b57cec5SDimitry Andric     DeclExtractor(Sema &S, DeclSetVector &Decls,
17560b57cec5SDimitry Andric                   SmallVectorImpl<SourceRange> &Ranges) :
17570b57cec5SDimitry Andric         Inherited(S.Context),
17580b57cec5SDimitry Andric         Decls(Decls),
17590b57cec5SDimitry Andric         Ranges(Ranges),
17600b57cec5SDimitry Andric         Simple(true) {}
17610b57cec5SDimitry Andric 
17620b57cec5SDimitry Andric     bool isSimple() { return Simple; }
17630b57cec5SDimitry Andric 
17640b57cec5SDimitry Andric     // Replaces the method in EvaluatedExprVisitor.
17650b57cec5SDimitry Andric     void VisitMemberExpr(MemberExpr* E) {
17660b57cec5SDimitry Andric       Simple = false;
17670b57cec5SDimitry Andric     }
17680b57cec5SDimitry Andric 
17695ffd83dbSDimitry Andric     // Any Stmt not explicitly listed will cause the condition to be marked
17705ffd83dbSDimitry Andric     // complex.
17715ffd83dbSDimitry Andric     void VisitStmt(Stmt *S) { Simple = false; }
17720b57cec5SDimitry Andric 
17730b57cec5SDimitry Andric     void VisitBinaryOperator(BinaryOperator *E) {
17740b57cec5SDimitry Andric       Visit(E->getLHS());
17750b57cec5SDimitry Andric       Visit(E->getRHS());
17760b57cec5SDimitry Andric     }
17770b57cec5SDimitry Andric 
17780b57cec5SDimitry Andric     void VisitCastExpr(CastExpr *E) {
17790b57cec5SDimitry Andric       Visit(E->getSubExpr());
17800b57cec5SDimitry Andric     }
17810b57cec5SDimitry Andric 
17820b57cec5SDimitry Andric     void VisitUnaryOperator(UnaryOperator *E) {
17830b57cec5SDimitry Andric       // Skip checking conditionals with derefernces.
17840b57cec5SDimitry Andric       if (E->getOpcode() == UO_Deref)
17850b57cec5SDimitry Andric         Simple = false;
17860b57cec5SDimitry Andric       else
17870b57cec5SDimitry Andric         Visit(E->getSubExpr());
17880b57cec5SDimitry Andric     }
17890b57cec5SDimitry Andric 
17900b57cec5SDimitry Andric     void VisitConditionalOperator(ConditionalOperator *E) {
17910b57cec5SDimitry Andric       Visit(E->getCond());
17920b57cec5SDimitry Andric       Visit(E->getTrueExpr());
17930b57cec5SDimitry Andric       Visit(E->getFalseExpr());
17940b57cec5SDimitry Andric     }
17950b57cec5SDimitry Andric 
17960b57cec5SDimitry Andric     void VisitParenExpr(ParenExpr *E) {
17970b57cec5SDimitry Andric       Visit(E->getSubExpr());
17980b57cec5SDimitry Andric     }
17990b57cec5SDimitry Andric 
18000b57cec5SDimitry Andric     void VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
18010b57cec5SDimitry Andric       Visit(E->getOpaqueValue()->getSourceExpr());
18020b57cec5SDimitry Andric       Visit(E->getFalseExpr());
18030b57cec5SDimitry Andric     }
18040b57cec5SDimitry Andric 
18050b57cec5SDimitry Andric     void VisitIntegerLiteral(IntegerLiteral *E) { }
18060b57cec5SDimitry Andric     void VisitFloatingLiteral(FloatingLiteral *E) { }
18070b57cec5SDimitry Andric     void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) { }
18080b57cec5SDimitry Andric     void VisitCharacterLiteral(CharacterLiteral *E) { }
18090b57cec5SDimitry Andric     void VisitGNUNullExpr(GNUNullExpr *E) { }
18100b57cec5SDimitry Andric     void VisitImaginaryLiteral(ImaginaryLiteral *E) { }
18110b57cec5SDimitry Andric 
18120b57cec5SDimitry Andric     void VisitDeclRefExpr(DeclRefExpr *E) {
18130b57cec5SDimitry Andric       VarDecl *VD = dyn_cast<VarDecl>(E->getDecl());
18140b57cec5SDimitry Andric       if (!VD) {
18150b57cec5SDimitry Andric         // Don't allow unhandled Decl types.
18160b57cec5SDimitry Andric         Simple = false;
18170b57cec5SDimitry Andric         return;
18180b57cec5SDimitry Andric       }
18190b57cec5SDimitry Andric 
18200b57cec5SDimitry Andric       Ranges.push_back(E->getSourceRange());
18210b57cec5SDimitry Andric 
18220b57cec5SDimitry Andric       Decls.insert(VD);
18230b57cec5SDimitry Andric     }
18240b57cec5SDimitry Andric 
18250b57cec5SDimitry Andric   }; // end class DeclExtractor
18260b57cec5SDimitry Andric 
18270b57cec5SDimitry Andric   // DeclMatcher checks to see if the decls are used in a non-evaluated
18280b57cec5SDimitry Andric   // context.
18290b57cec5SDimitry Andric   class DeclMatcher : public EvaluatedExprVisitor<DeclMatcher> {
18300b57cec5SDimitry Andric     DeclSetVector &Decls;
18310b57cec5SDimitry Andric     bool FoundDecl;
18320b57cec5SDimitry Andric 
18330b57cec5SDimitry Andric   public:
18340b57cec5SDimitry Andric     typedef EvaluatedExprVisitor<DeclMatcher> Inherited;
18350b57cec5SDimitry Andric 
18360b57cec5SDimitry Andric     DeclMatcher(Sema &S, DeclSetVector &Decls, Stmt *Statement) :
18370b57cec5SDimitry Andric         Inherited(S.Context), Decls(Decls), FoundDecl(false) {
18380b57cec5SDimitry Andric       if (!Statement) return;
18390b57cec5SDimitry Andric 
18400b57cec5SDimitry Andric       Visit(Statement);
18410b57cec5SDimitry Andric     }
18420b57cec5SDimitry Andric 
18430b57cec5SDimitry Andric     void VisitReturnStmt(ReturnStmt *S) {
18440b57cec5SDimitry Andric       FoundDecl = true;
18450b57cec5SDimitry Andric     }
18460b57cec5SDimitry Andric 
18470b57cec5SDimitry Andric     void VisitBreakStmt(BreakStmt *S) {
18480b57cec5SDimitry Andric       FoundDecl = true;
18490b57cec5SDimitry Andric     }
18500b57cec5SDimitry Andric 
18510b57cec5SDimitry Andric     void VisitGotoStmt(GotoStmt *S) {
18520b57cec5SDimitry Andric       FoundDecl = true;
18530b57cec5SDimitry Andric     }
18540b57cec5SDimitry Andric 
18550b57cec5SDimitry Andric     void VisitCastExpr(CastExpr *E) {
18560b57cec5SDimitry Andric       if (E->getCastKind() == CK_LValueToRValue)
18570b57cec5SDimitry Andric         CheckLValueToRValueCast(E->getSubExpr());
18580b57cec5SDimitry Andric       else
18590b57cec5SDimitry Andric         Visit(E->getSubExpr());
18600b57cec5SDimitry Andric     }
18610b57cec5SDimitry Andric 
18620b57cec5SDimitry Andric     void CheckLValueToRValueCast(Expr *E) {
18630b57cec5SDimitry Andric       E = E->IgnoreParenImpCasts();
18640b57cec5SDimitry Andric 
18650b57cec5SDimitry Andric       if (isa<DeclRefExpr>(E)) {
18660b57cec5SDimitry Andric         return;
18670b57cec5SDimitry Andric       }
18680b57cec5SDimitry Andric 
18690b57cec5SDimitry Andric       if (ConditionalOperator *CO = dyn_cast<ConditionalOperator>(E)) {
18700b57cec5SDimitry Andric         Visit(CO->getCond());
18710b57cec5SDimitry Andric         CheckLValueToRValueCast(CO->getTrueExpr());
18720b57cec5SDimitry Andric         CheckLValueToRValueCast(CO->getFalseExpr());
18730b57cec5SDimitry Andric         return;
18740b57cec5SDimitry Andric       }
18750b57cec5SDimitry Andric 
18760b57cec5SDimitry Andric       if (BinaryConditionalOperator *BCO =
18770b57cec5SDimitry Andric               dyn_cast<BinaryConditionalOperator>(E)) {
18780b57cec5SDimitry Andric         CheckLValueToRValueCast(BCO->getOpaqueValue()->getSourceExpr());
18790b57cec5SDimitry Andric         CheckLValueToRValueCast(BCO->getFalseExpr());
18800b57cec5SDimitry Andric         return;
18810b57cec5SDimitry Andric       }
18820b57cec5SDimitry Andric 
18830b57cec5SDimitry Andric       Visit(E);
18840b57cec5SDimitry Andric     }
18850b57cec5SDimitry Andric 
18860b57cec5SDimitry Andric     void VisitDeclRefExpr(DeclRefExpr *E) {
18870b57cec5SDimitry Andric       if (VarDecl *VD = dyn_cast<VarDecl>(E->getDecl()))
18880b57cec5SDimitry Andric         if (Decls.count(VD))
18890b57cec5SDimitry Andric           FoundDecl = true;
18900b57cec5SDimitry Andric     }
18910b57cec5SDimitry Andric 
18920b57cec5SDimitry Andric     void VisitPseudoObjectExpr(PseudoObjectExpr *POE) {
18930b57cec5SDimitry Andric       // Only need to visit the semantics for POE.
18940b57cec5SDimitry Andric       // SyntaticForm doesn't really use the Decal.
18950b57cec5SDimitry Andric       for (auto *S : POE->semantics()) {
18960b57cec5SDimitry Andric         if (auto *OVE = dyn_cast<OpaqueValueExpr>(S))
18970b57cec5SDimitry Andric           // Look past the OVE into the expression it binds.
18980b57cec5SDimitry Andric           Visit(OVE->getSourceExpr());
18990b57cec5SDimitry Andric         else
19000b57cec5SDimitry Andric           Visit(S);
19010b57cec5SDimitry Andric       }
19020b57cec5SDimitry Andric     }
19030b57cec5SDimitry Andric 
19040b57cec5SDimitry Andric     bool FoundDeclInUse() { return FoundDecl; }
19050b57cec5SDimitry Andric 
19060b57cec5SDimitry Andric   };  // end class DeclMatcher
19070b57cec5SDimitry Andric 
19080b57cec5SDimitry Andric   void CheckForLoopConditionalStatement(Sema &S, Expr *Second,
19090b57cec5SDimitry Andric                                         Expr *Third, Stmt *Body) {
19100b57cec5SDimitry Andric     // Condition is empty
19110b57cec5SDimitry Andric     if (!Second) return;
19120b57cec5SDimitry Andric 
19130b57cec5SDimitry Andric     if (S.Diags.isIgnored(diag::warn_variables_not_in_loop_body,
19140b57cec5SDimitry Andric                           Second->getBeginLoc()))
19150b57cec5SDimitry Andric       return;
19160b57cec5SDimitry Andric 
19170b57cec5SDimitry Andric     PartialDiagnostic PDiag = S.PDiag(diag::warn_variables_not_in_loop_body);
19180b57cec5SDimitry Andric     DeclSetVector Decls;
19190b57cec5SDimitry Andric     SmallVector<SourceRange, 10> Ranges;
19200b57cec5SDimitry Andric     DeclExtractor DE(S, Decls, Ranges);
19210b57cec5SDimitry Andric     DE.Visit(Second);
19220b57cec5SDimitry Andric 
19230b57cec5SDimitry Andric     // Don't analyze complex conditionals.
19240b57cec5SDimitry Andric     if (!DE.isSimple()) return;
19250b57cec5SDimitry Andric 
19260b57cec5SDimitry Andric     // No decls found.
19270b57cec5SDimitry Andric     if (Decls.size() == 0) return;
19280b57cec5SDimitry Andric 
19290b57cec5SDimitry Andric     // Don't warn on volatile, static, or global variables.
19300b57cec5SDimitry Andric     for (auto *VD : Decls)
19310b57cec5SDimitry Andric       if (VD->getType().isVolatileQualified() || VD->hasGlobalStorage())
19320b57cec5SDimitry Andric         return;
19330b57cec5SDimitry Andric 
19340b57cec5SDimitry Andric     if (DeclMatcher(S, Decls, Second).FoundDeclInUse() ||
19350b57cec5SDimitry Andric         DeclMatcher(S, Decls, Third).FoundDeclInUse() ||
19360b57cec5SDimitry Andric         DeclMatcher(S, Decls, Body).FoundDeclInUse())
19370b57cec5SDimitry Andric       return;
19380b57cec5SDimitry Andric 
19390b57cec5SDimitry Andric     // Load decl names into diagnostic.
19400b57cec5SDimitry Andric     if (Decls.size() > 4) {
19410b57cec5SDimitry Andric       PDiag << 0;
19420b57cec5SDimitry Andric     } else {
19430b57cec5SDimitry Andric       PDiag << (unsigned)Decls.size();
19440b57cec5SDimitry Andric       for (auto *VD : Decls)
19450b57cec5SDimitry Andric         PDiag << VD->getDeclName();
19460b57cec5SDimitry Andric     }
19470b57cec5SDimitry Andric 
19480b57cec5SDimitry Andric     for (auto Range : Ranges)
19490b57cec5SDimitry Andric       PDiag << Range;
19500b57cec5SDimitry Andric 
19510b57cec5SDimitry Andric     S.Diag(Ranges.begin()->getBegin(), PDiag);
19520b57cec5SDimitry Andric   }
19530b57cec5SDimitry Andric 
19540b57cec5SDimitry Andric   // If Statement is an incemement or decrement, return true and sets the
19550b57cec5SDimitry Andric   // variables Increment and DRE.
19560b57cec5SDimitry Andric   bool ProcessIterationStmt(Sema &S, Stmt* Statement, bool &Increment,
19570b57cec5SDimitry Andric                             DeclRefExpr *&DRE) {
19580b57cec5SDimitry Andric     if (auto Cleanups = dyn_cast<ExprWithCleanups>(Statement))
19590b57cec5SDimitry Andric       if (!Cleanups->cleanupsHaveSideEffects())
19600b57cec5SDimitry Andric         Statement = Cleanups->getSubExpr();
19610b57cec5SDimitry Andric 
19620b57cec5SDimitry Andric     if (UnaryOperator *UO = dyn_cast<UnaryOperator>(Statement)) {
19630b57cec5SDimitry Andric       switch (UO->getOpcode()) {
19640b57cec5SDimitry Andric         default: return false;
19650b57cec5SDimitry Andric         case UO_PostInc:
19660b57cec5SDimitry Andric         case UO_PreInc:
19670b57cec5SDimitry Andric           Increment = true;
19680b57cec5SDimitry Andric           break;
19690b57cec5SDimitry Andric         case UO_PostDec:
19700b57cec5SDimitry Andric         case UO_PreDec:
19710b57cec5SDimitry Andric           Increment = false;
19720b57cec5SDimitry Andric           break;
19730b57cec5SDimitry Andric       }
19740b57cec5SDimitry Andric       DRE = dyn_cast<DeclRefExpr>(UO->getSubExpr());
19750b57cec5SDimitry Andric       return DRE;
19760b57cec5SDimitry Andric     }
19770b57cec5SDimitry Andric 
19780b57cec5SDimitry Andric     if (CXXOperatorCallExpr *Call = dyn_cast<CXXOperatorCallExpr>(Statement)) {
19790b57cec5SDimitry Andric       FunctionDecl *FD = Call->getDirectCallee();
19800b57cec5SDimitry Andric       if (!FD || !FD->isOverloadedOperator()) return false;
19810b57cec5SDimitry Andric       switch (FD->getOverloadedOperator()) {
19820b57cec5SDimitry Andric         default: return false;
19830b57cec5SDimitry Andric         case OO_PlusPlus:
19840b57cec5SDimitry Andric           Increment = true;
19850b57cec5SDimitry Andric           break;
19860b57cec5SDimitry Andric         case OO_MinusMinus:
19870b57cec5SDimitry Andric           Increment = false;
19880b57cec5SDimitry Andric           break;
19890b57cec5SDimitry Andric       }
19900b57cec5SDimitry Andric       DRE = dyn_cast<DeclRefExpr>(Call->getArg(0));
19910b57cec5SDimitry Andric       return DRE;
19920b57cec5SDimitry Andric     }
19930b57cec5SDimitry Andric 
19940b57cec5SDimitry Andric     return false;
19950b57cec5SDimitry Andric   }
19960b57cec5SDimitry Andric 
19970b57cec5SDimitry Andric   // A visitor to determine if a continue or break statement is a
19980b57cec5SDimitry Andric   // subexpression.
19990b57cec5SDimitry Andric   class BreakContinueFinder : public ConstEvaluatedExprVisitor<BreakContinueFinder> {
20000b57cec5SDimitry Andric     SourceLocation BreakLoc;
20010b57cec5SDimitry Andric     SourceLocation ContinueLoc;
20020b57cec5SDimitry Andric     bool InSwitch = false;
20030b57cec5SDimitry Andric 
20040b57cec5SDimitry Andric   public:
20050b57cec5SDimitry Andric     BreakContinueFinder(Sema &S, const Stmt* Body) :
20060b57cec5SDimitry Andric         Inherited(S.Context) {
20070b57cec5SDimitry Andric       Visit(Body);
20080b57cec5SDimitry Andric     }
20090b57cec5SDimitry Andric 
20100b57cec5SDimitry Andric     typedef ConstEvaluatedExprVisitor<BreakContinueFinder> Inherited;
20110b57cec5SDimitry Andric 
20120b57cec5SDimitry Andric     void VisitContinueStmt(const ContinueStmt* E) {
20130b57cec5SDimitry Andric       ContinueLoc = E->getContinueLoc();
20140b57cec5SDimitry Andric     }
20150b57cec5SDimitry Andric 
20160b57cec5SDimitry Andric     void VisitBreakStmt(const BreakStmt* E) {
20170b57cec5SDimitry Andric       if (!InSwitch)
20180b57cec5SDimitry Andric         BreakLoc = E->getBreakLoc();
20190b57cec5SDimitry Andric     }
20200b57cec5SDimitry Andric 
20210b57cec5SDimitry Andric     void VisitSwitchStmt(const SwitchStmt* S) {
20220b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
20230b57cec5SDimitry Andric         Visit(Init);
20240b57cec5SDimitry Andric       if (const Stmt *CondVar = S->getConditionVariableDeclStmt())
20250b57cec5SDimitry Andric         Visit(CondVar);
20260b57cec5SDimitry Andric       if (const Stmt *Cond = S->getCond())
20270b57cec5SDimitry Andric         Visit(Cond);
20280b57cec5SDimitry Andric 
20290b57cec5SDimitry Andric       // Don't return break statements from the body of a switch.
20300b57cec5SDimitry Andric       InSwitch = true;
20310b57cec5SDimitry Andric       if (const Stmt *Body = S->getBody())
20320b57cec5SDimitry Andric         Visit(Body);
20330b57cec5SDimitry Andric       InSwitch = false;
20340b57cec5SDimitry Andric     }
20350b57cec5SDimitry Andric 
20360b57cec5SDimitry Andric     void VisitForStmt(const ForStmt *S) {
20370b57cec5SDimitry Andric       // Only visit the init statement of a for loop; the body
20380b57cec5SDimitry Andric       // has a different break/continue scope.
20390b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
20400b57cec5SDimitry Andric         Visit(Init);
20410b57cec5SDimitry Andric     }
20420b57cec5SDimitry Andric 
20430b57cec5SDimitry Andric     void VisitWhileStmt(const WhileStmt *) {
20440b57cec5SDimitry Andric       // Do nothing; the children of a while loop have a different
20450b57cec5SDimitry Andric       // break/continue scope.
20460b57cec5SDimitry Andric     }
20470b57cec5SDimitry Andric 
20480b57cec5SDimitry Andric     void VisitDoStmt(const DoStmt *) {
20490b57cec5SDimitry Andric       // Do nothing; the children of a while loop have a different
20500b57cec5SDimitry Andric       // break/continue scope.
20510b57cec5SDimitry Andric     }
20520b57cec5SDimitry Andric 
20530b57cec5SDimitry Andric     void VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
20540b57cec5SDimitry Andric       // Only visit the initialization of a for loop; the body
20550b57cec5SDimitry Andric       // has a different break/continue scope.
20560b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
20570b57cec5SDimitry Andric         Visit(Init);
20580b57cec5SDimitry Andric       if (const Stmt *Range = S->getRangeStmt())
20590b57cec5SDimitry Andric         Visit(Range);
20600b57cec5SDimitry Andric       if (const Stmt *Begin = S->getBeginStmt())
20610b57cec5SDimitry Andric         Visit(Begin);
20620b57cec5SDimitry Andric       if (const Stmt *End = S->getEndStmt())
20630b57cec5SDimitry Andric         Visit(End);
20640b57cec5SDimitry Andric     }
20650b57cec5SDimitry Andric 
20660b57cec5SDimitry Andric     void VisitObjCForCollectionStmt(const ObjCForCollectionStmt *S) {
20670b57cec5SDimitry Andric       // Only visit the initialization of a for loop; the body
20680b57cec5SDimitry Andric       // has a different break/continue scope.
20690b57cec5SDimitry Andric       if (const Stmt *Element = S->getElement())
20700b57cec5SDimitry Andric         Visit(Element);
20710b57cec5SDimitry Andric       if (const Stmt *Collection = S->getCollection())
20720b57cec5SDimitry Andric         Visit(Collection);
20730b57cec5SDimitry Andric     }
20740b57cec5SDimitry Andric 
20750b57cec5SDimitry Andric     bool ContinueFound() { return ContinueLoc.isValid(); }
20760b57cec5SDimitry Andric     bool BreakFound() { return BreakLoc.isValid(); }
20770b57cec5SDimitry Andric     SourceLocation GetContinueLoc() { return ContinueLoc; }
20780b57cec5SDimitry Andric     SourceLocation GetBreakLoc() { return BreakLoc; }
20790b57cec5SDimitry Andric 
20800b57cec5SDimitry Andric   };  // end class BreakContinueFinder
20810b57cec5SDimitry Andric 
20820b57cec5SDimitry Andric   // Emit a warning when a loop increment/decrement appears twice per loop
20830b57cec5SDimitry Andric   // iteration.  The conditions which trigger this warning are:
20840b57cec5SDimitry Andric   // 1) The last statement in the loop body and the third expression in the
20850b57cec5SDimitry Andric   //    for loop are both increment or both decrement of the same variable
20860b57cec5SDimitry Andric   // 2) No continue statements in the loop body.
20870b57cec5SDimitry Andric   void CheckForRedundantIteration(Sema &S, Expr *Third, Stmt *Body) {
20880b57cec5SDimitry Andric     // Return when there is nothing to check.
20890b57cec5SDimitry Andric     if (!Body || !Third) return;
20900b57cec5SDimitry Andric 
20910b57cec5SDimitry Andric     if (S.Diags.isIgnored(diag::warn_redundant_loop_iteration,
20920b57cec5SDimitry Andric                           Third->getBeginLoc()))
20930b57cec5SDimitry Andric       return;
20940b57cec5SDimitry Andric 
20950b57cec5SDimitry Andric     // Get the last statement from the loop body.
20960b57cec5SDimitry Andric     CompoundStmt *CS = dyn_cast<CompoundStmt>(Body);
20970b57cec5SDimitry Andric     if (!CS || CS->body_empty()) return;
20980b57cec5SDimitry Andric     Stmt *LastStmt = CS->body_back();
20990b57cec5SDimitry Andric     if (!LastStmt) return;
21000b57cec5SDimitry Andric 
21010b57cec5SDimitry Andric     bool LoopIncrement, LastIncrement;
21020b57cec5SDimitry Andric     DeclRefExpr *LoopDRE, *LastDRE;
21030b57cec5SDimitry Andric 
21040b57cec5SDimitry Andric     if (!ProcessIterationStmt(S, Third, LoopIncrement, LoopDRE)) return;
21050b57cec5SDimitry Andric     if (!ProcessIterationStmt(S, LastStmt, LastIncrement, LastDRE)) return;
21060b57cec5SDimitry Andric 
21070b57cec5SDimitry Andric     // Check that the two statements are both increments or both decrements
21080b57cec5SDimitry Andric     // on the same variable.
21090b57cec5SDimitry Andric     if (LoopIncrement != LastIncrement ||
21100b57cec5SDimitry Andric         LoopDRE->getDecl() != LastDRE->getDecl()) return;
21110b57cec5SDimitry Andric 
21120b57cec5SDimitry Andric     if (BreakContinueFinder(S, Body).ContinueFound()) return;
21130b57cec5SDimitry Andric 
21140b57cec5SDimitry Andric     S.Diag(LastDRE->getLocation(), diag::warn_redundant_loop_iteration)
21150b57cec5SDimitry Andric          << LastDRE->getDecl() << LastIncrement;
21160b57cec5SDimitry Andric     S.Diag(LoopDRE->getLocation(), diag::note_loop_iteration_here)
21170b57cec5SDimitry Andric          << LoopIncrement;
21180b57cec5SDimitry Andric   }
21190b57cec5SDimitry Andric 
21200b57cec5SDimitry Andric } // end namespace
21210b57cec5SDimitry Andric 
21220b57cec5SDimitry Andric 
21230b57cec5SDimitry Andric void Sema::CheckBreakContinueBinding(Expr *E) {
21240b57cec5SDimitry Andric   if (!E || getLangOpts().CPlusPlus)
21250b57cec5SDimitry Andric     return;
21260b57cec5SDimitry Andric   BreakContinueFinder BCFinder(*this, E);
21270b57cec5SDimitry Andric   Scope *BreakParent = CurScope->getBreakParent();
21280b57cec5SDimitry Andric   if (BCFinder.BreakFound() && BreakParent) {
21290b57cec5SDimitry Andric     if (BreakParent->getFlags() & Scope::SwitchScope) {
21300b57cec5SDimitry Andric       Diag(BCFinder.GetBreakLoc(), diag::warn_break_binds_to_switch);
21310b57cec5SDimitry Andric     } else {
21320b57cec5SDimitry Andric       Diag(BCFinder.GetBreakLoc(), diag::warn_loop_ctrl_binds_to_inner)
21330b57cec5SDimitry Andric           << "break";
21340b57cec5SDimitry Andric     }
21350b57cec5SDimitry Andric   } else if (BCFinder.ContinueFound() && CurScope->getContinueParent()) {
21360b57cec5SDimitry Andric     Diag(BCFinder.GetContinueLoc(), diag::warn_loop_ctrl_binds_to_inner)
21370b57cec5SDimitry Andric         << "continue";
21380b57cec5SDimitry Andric   }
21390b57cec5SDimitry Andric }
21400b57cec5SDimitry Andric 
21410b57cec5SDimitry Andric StmtResult Sema::ActOnForStmt(SourceLocation ForLoc, SourceLocation LParenLoc,
21420b57cec5SDimitry Andric                               Stmt *First, ConditionResult Second,
21430b57cec5SDimitry Andric                               FullExprArg third, SourceLocation RParenLoc,
21440b57cec5SDimitry Andric                               Stmt *Body) {
21450b57cec5SDimitry Andric   if (Second.isInvalid())
21460b57cec5SDimitry Andric     return StmtError();
21470b57cec5SDimitry Andric 
21480b57cec5SDimitry Andric   if (!getLangOpts().CPlusPlus) {
21490b57cec5SDimitry Andric     if (DeclStmt *DS = dyn_cast_or_null<DeclStmt>(First)) {
21500b57cec5SDimitry Andric       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
21510b57cec5SDimitry Andric       // declare identifiers for objects having storage class 'auto' or
21520b57cec5SDimitry Andric       // 'register'.
2153e8d8bef9SDimitry Andric       const Decl *NonVarSeen = nullptr;
2154e8d8bef9SDimitry Andric       bool VarDeclSeen = false;
21550b57cec5SDimitry Andric       for (auto *DI : DS->decls()) {
2156e8d8bef9SDimitry Andric         if (VarDecl *VD = dyn_cast<VarDecl>(DI)) {
2157e8d8bef9SDimitry Andric           VarDeclSeen = true;
2158e8d8bef9SDimitry Andric           if (VD->isLocalVarDecl() && !VD->hasLocalStorage()) {
21590b57cec5SDimitry Andric             Diag(DI->getLocation(), diag::err_non_local_variable_decl_in_for);
21600b57cec5SDimitry Andric             DI->setInvalidDecl();
21610b57cec5SDimitry Andric           }
2162e8d8bef9SDimitry Andric         } else if (!NonVarSeen) {
2163e8d8bef9SDimitry Andric           // Keep track of the first non-variable declaration we saw so that
2164e8d8bef9SDimitry Andric           // we can diagnose if we don't see any variable declarations. This
2165e8d8bef9SDimitry Andric           // covers a case like declaring a typedef, function, or structure
2166e8d8bef9SDimitry Andric           // type rather than a variable.
2167e8d8bef9SDimitry Andric           NonVarSeen = DI;
21680b57cec5SDimitry Andric         }
21690b57cec5SDimitry Andric       }
2170e8d8bef9SDimitry Andric       // Diagnose if we saw a non-variable declaration but no variable
2171e8d8bef9SDimitry Andric       // declarations.
2172e8d8bef9SDimitry Andric       if (NonVarSeen && !VarDeclSeen)
2173e8d8bef9SDimitry Andric         Diag(NonVarSeen->getLocation(), diag::err_non_variable_decl_in_for);
2174e8d8bef9SDimitry Andric     }
21750b57cec5SDimitry Andric   }
21760b57cec5SDimitry Andric 
21770b57cec5SDimitry Andric   CheckBreakContinueBinding(Second.get().second);
21780b57cec5SDimitry Andric   CheckBreakContinueBinding(third.get());
21790b57cec5SDimitry Andric 
21800b57cec5SDimitry Andric   if (!Second.get().first)
21810b57cec5SDimitry Andric     CheckForLoopConditionalStatement(*this, Second.get().second, third.get(),
21820b57cec5SDimitry Andric                                      Body);
21830b57cec5SDimitry Andric   CheckForRedundantIteration(*this, third.get(), Body);
21840b57cec5SDimitry Andric 
21850b57cec5SDimitry Andric   if (Second.get().second &&
21860b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator,
21870b57cec5SDimitry Andric                        Second.get().second->getExprLoc()))
21880b57cec5SDimitry Andric     CommaVisitor(*this).Visit(Second.get().second);
21890b57cec5SDimitry Andric 
21900b57cec5SDimitry Andric   Expr *Third  = third.release().getAs<Expr>();
21910b57cec5SDimitry Andric   if (isa<NullStmt>(Body))
21920b57cec5SDimitry Andric     getCurCompoundScope().setHasEmptyLoopBodies();
21930b57cec5SDimitry Andric 
21940b57cec5SDimitry Andric   return new (Context)
21950b57cec5SDimitry Andric       ForStmt(Context, First, Second.get().second, Second.get().first, Third,
21960b57cec5SDimitry Andric               Body, ForLoc, LParenLoc, RParenLoc);
21970b57cec5SDimitry Andric }
21980b57cec5SDimitry Andric 
21990b57cec5SDimitry Andric /// In an Objective C collection iteration statement:
22000b57cec5SDimitry Andric ///   for (x in y)
22010b57cec5SDimitry Andric /// x can be an arbitrary l-value expression.  Bind it up as a
22020b57cec5SDimitry Andric /// full-expression.
22030b57cec5SDimitry Andric StmtResult Sema::ActOnForEachLValueExpr(Expr *E) {
22040b57cec5SDimitry Andric   // Reduce placeholder expressions here.  Note that this rejects the
22050b57cec5SDimitry Andric   // use of pseudo-object l-values in this position.
22060b57cec5SDimitry Andric   ExprResult result = CheckPlaceholderExpr(E);
22070b57cec5SDimitry Andric   if (result.isInvalid()) return StmtError();
22080b57cec5SDimitry Andric   E = result.get();
22090b57cec5SDimitry Andric 
22100b57cec5SDimitry Andric   ExprResult FullExpr = ActOnFinishFullExpr(E, /*DiscardedValue*/ false);
22110b57cec5SDimitry Andric   if (FullExpr.isInvalid())
22120b57cec5SDimitry Andric     return StmtError();
22130b57cec5SDimitry Andric   return StmtResult(static_cast<Stmt*>(FullExpr.get()));
22140b57cec5SDimitry Andric }
22150b57cec5SDimitry Andric 
22160b57cec5SDimitry Andric ExprResult
22170b57cec5SDimitry Andric Sema::CheckObjCForCollectionOperand(SourceLocation forLoc, Expr *collection) {
22180b57cec5SDimitry Andric   if (!collection)
22190b57cec5SDimitry Andric     return ExprError();
22200b57cec5SDimitry Andric 
22210b57cec5SDimitry Andric   ExprResult result = CorrectDelayedTyposInExpr(collection);
22220b57cec5SDimitry Andric   if (!result.isUsable())
22230b57cec5SDimitry Andric     return ExprError();
22240b57cec5SDimitry Andric   collection = result.get();
22250b57cec5SDimitry Andric 
22260b57cec5SDimitry Andric   // Bail out early if we've got a type-dependent expression.
22270b57cec5SDimitry Andric   if (collection->isTypeDependent()) return collection;
22280b57cec5SDimitry Andric 
22290b57cec5SDimitry Andric   // Perform normal l-value conversion.
22300b57cec5SDimitry Andric   result = DefaultFunctionArrayLvalueConversion(collection);
22310b57cec5SDimitry Andric   if (result.isInvalid())
22320b57cec5SDimitry Andric     return ExprError();
22330b57cec5SDimitry Andric   collection = result.get();
22340b57cec5SDimitry Andric 
22350b57cec5SDimitry Andric   // The operand needs to have object-pointer type.
22360b57cec5SDimitry Andric   // TODO: should we do a contextual conversion?
22370b57cec5SDimitry Andric   const ObjCObjectPointerType *pointerType =
22380b57cec5SDimitry Andric     collection->getType()->getAs<ObjCObjectPointerType>();
22390b57cec5SDimitry Andric   if (!pointerType)
22400b57cec5SDimitry Andric     return Diag(forLoc, diag::err_collection_expr_type)
22410b57cec5SDimitry Andric              << collection->getType() << collection->getSourceRange();
22420b57cec5SDimitry Andric 
22430b57cec5SDimitry Andric   // Check that the operand provides
22440b57cec5SDimitry Andric   //   - countByEnumeratingWithState:objects:count:
22450b57cec5SDimitry Andric   const ObjCObjectType *objectType = pointerType->getObjectType();
22460b57cec5SDimitry Andric   ObjCInterfaceDecl *iface = objectType->getInterface();
22470b57cec5SDimitry Andric 
22480b57cec5SDimitry Andric   // If we have a forward-declared type, we can't do this check.
22490b57cec5SDimitry Andric   // Under ARC, it is an error not to have a forward-declared class.
22500b57cec5SDimitry Andric   if (iface &&
22510b57cec5SDimitry Andric       (getLangOpts().ObjCAutoRefCount
22520b57cec5SDimitry Andric            ? RequireCompleteType(forLoc, QualType(objectType, 0),
22530b57cec5SDimitry Andric                                  diag::err_arc_collection_forward, collection)
22540b57cec5SDimitry Andric            : !isCompleteType(forLoc, QualType(objectType, 0)))) {
22550b57cec5SDimitry Andric     // Otherwise, if we have any useful type information, check that
22560b57cec5SDimitry Andric     // the type declares the appropriate method.
22570b57cec5SDimitry Andric   } else if (iface || !objectType->qual_empty()) {
22580b57cec5SDimitry Andric     IdentifierInfo *selectorIdents[] = {
22590b57cec5SDimitry Andric       &Context.Idents.get("countByEnumeratingWithState"),
22600b57cec5SDimitry Andric       &Context.Idents.get("objects"),
22610b57cec5SDimitry Andric       &Context.Idents.get("count")
22620b57cec5SDimitry Andric     };
22630b57cec5SDimitry Andric     Selector selector = Context.Selectors.getSelector(3, &selectorIdents[0]);
22640b57cec5SDimitry Andric 
22650b57cec5SDimitry Andric     ObjCMethodDecl *method = nullptr;
22660b57cec5SDimitry Andric 
22670b57cec5SDimitry Andric     // If there's an interface, look in both the public and private APIs.
22680b57cec5SDimitry Andric     if (iface) {
22690b57cec5SDimitry Andric       method = iface->lookupInstanceMethod(selector);
22700b57cec5SDimitry Andric       if (!method) method = iface->lookupPrivateMethod(selector);
22710b57cec5SDimitry Andric     }
22720b57cec5SDimitry Andric 
22730b57cec5SDimitry Andric     // Also check protocol qualifiers.
22740b57cec5SDimitry Andric     if (!method)
22750b57cec5SDimitry Andric       method = LookupMethodInQualifiedType(selector, pointerType,
22760b57cec5SDimitry Andric                                            /*instance*/ true);
22770b57cec5SDimitry Andric 
22780b57cec5SDimitry Andric     // If we didn't find it anywhere, give up.
22790b57cec5SDimitry Andric     if (!method) {
22800b57cec5SDimitry Andric       Diag(forLoc, diag::warn_collection_expr_type)
22810b57cec5SDimitry Andric         << collection->getType() << selector << collection->getSourceRange();
22820b57cec5SDimitry Andric     }
22830b57cec5SDimitry Andric 
22840b57cec5SDimitry Andric     // TODO: check for an incompatible signature?
22850b57cec5SDimitry Andric   }
22860b57cec5SDimitry Andric 
22870b57cec5SDimitry Andric   // Wrap up any cleanups in the expression.
22880b57cec5SDimitry Andric   return collection;
22890b57cec5SDimitry Andric }
22900b57cec5SDimitry Andric 
22910b57cec5SDimitry Andric StmtResult
22920b57cec5SDimitry Andric Sema::ActOnObjCForCollectionStmt(SourceLocation ForLoc,
22930b57cec5SDimitry Andric                                  Stmt *First, Expr *collection,
22940b57cec5SDimitry Andric                                  SourceLocation RParenLoc) {
22950b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
22960b57cec5SDimitry Andric 
22970b57cec5SDimitry Andric   ExprResult CollectionExprResult =
22980b57cec5SDimitry Andric     CheckObjCForCollectionOperand(ForLoc, collection);
22990b57cec5SDimitry Andric 
23000b57cec5SDimitry Andric   if (First) {
23010b57cec5SDimitry Andric     QualType FirstType;
23020b57cec5SDimitry Andric     if (DeclStmt *DS = dyn_cast<DeclStmt>(First)) {
23030b57cec5SDimitry Andric       if (!DS->isSingleDecl())
23040b57cec5SDimitry Andric         return StmtError(Diag((*DS->decl_begin())->getLocation(),
23050b57cec5SDimitry Andric                          diag::err_toomany_element_decls));
23060b57cec5SDimitry Andric 
23070b57cec5SDimitry Andric       VarDecl *D = dyn_cast<VarDecl>(DS->getSingleDecl());
23080b57cec5SDimitry Andric       if (!D || D->isInvalidDecl())
23090b57cec5SDimitry Andric         return StmtError();
23100b57cec5SDimitry Andric 
23110b57cec5SDimitry Andric       FirstType = D->getType();
23120b57cec5SDimitry Andric       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
23130b57cec5SDimitry Andric       // declare identifiers for objects having storage class 'auto' or
23140b57cec5SDimitry Andric       // 'register'.
23150b57cec5SDimitry Andric       if (!D->hasLocalStorage())
23160b57cec5SDimitry Andric         return StmtError(Diag(D->getLocation(),
23170b57cec5SDimitry Andric                               diag::err_non_local_variable_decl_in_for));
23180b57cec5SDimitry Andric 
23190b57cec5SDimitry Andric       // If the type contained 'auto', deduce the 'auto' to 'id'.
23200b57cec5SDimitry Andric       if (FirstType->getContainedAutoType()) {
2321bdd1243dSDimitry Andric         SourceLocation Loc = D->getLocation();
2322bdd1243dSDimitry Andric         OpaqueValueExpr OpaqueId(Loc, Context.getObjCIdType(), VK_PRValue);
23230b57cec5SDimitry Andric         Expr *DeducedInit = &OpaqueId;
2324bdd1243dSDimitry Andric         TemplateDeductionInfo Info(Loc);
2325bdd1243dSDimitry Andric         FirstType = QualType();
2326bdd1243dSDimitry Andric         TemplateDeductionResult Result = DeduceAutoType(
2327bdd1243dSDimitry Andric             D->getTypeSourceInfo()->getTypeLoc(), DeducedInit, FirstType, Info);
2328bdd1243dSDimitry Andric         if (Result != TDK_Success && Result != TDK_AlreadyDiagnosed)
23290b57cec5SDimitry Andric           DiagnoseAutoDeductionFailure(D, DeducedInit);
23300b57cec5SDimitry Andric         if (FirstType.isNull()) {
23310b57cec5SDimitry Andric           D->setInvalidDecl();
23320b57cec5SDimitry Andric           return StmtError();
23330b57cec5SDimitry Andric         }
23340b57cec5SDimitry Andric 
23350b57cec5SDimitry Andric         D->setType(FirstType);
23360b57cec5SDimitry Andric 
23370b57cec5SDimitry Andric         if (!inTemplateInstantiation()) {
23380b57cec5SDimitry Andric           SourceLocation Loc =
23390b57cec5SDimitry Andric               D->getTypeSourceInfo()->getTypeLoc().getBeginLoc();
23400b57cec5SDimitry Andric           Diag(Loc, diag::warn_auto_var_is_id)
23410b57cec5SDimitry Andric             << D->getDeclName();
23420b57cec5SDimitry Andric         }
23430b57cec5SDimitry Andric       }
23440b57cec5SDimitry Andric 
23450b57cec5SDimitry Andric     } else {
23460b57cec5SDimitry Andric       Expr *FirstE = cast<Expr>(First);
23470b57cec5SDimitry Andric       if (!FirstE->isTypeDependent() && !FirstE->isLValue())
23480b57cec5SDimitry Andric         return StmtError(
23490b57cec5SDimitry Andric             Diag(First->getBeginLoc(), diag::err_selector_element_not_lvalue)
23500b57cec5SDimitry Andric             << First->getSourceRange());
23510b57cec5SDimitry Andric 
23520b57cec5SDimitry Andric       FirstType = static_cast<Expr*>(First)->getType();
23530b57cec5SDimitry Andric       if (FirstType.isConstQualified())
23540b57cec5SDimitry Andric         Diag(ForLoc, diag::err_selector_element_const_type)
23550b57cec5SDimitry Andric           << FirstType << First->getSourceRange();
23560b57cec5SDimitry Andric     }
23570b57cec5SDimitry Andric     if (!FirstType->isDependentType() &&
23580b57cec5SDimitry Andric         !FirstType->isObjCObjectPointerType() &&
23590b57cec5SDimitry Andric         !FirstType->isBlockPointerType())
23600b57cec5SDimitry Andric         return StmtError(Diag(ForLoc, diag::err_selector_element_type)
23610b57cec5SDimitry Andric                            << FirstType << First->getSourceRange());
23620b57cec5SDimitry Andric   }
23630b57cec5SDimitry Andric 
23640b57cec5SDimitry Andric   if (CollectionExprResult.isInvalid())
23650b57cec5SDimitry Andric     return StmtError();
23660b57cec5SDimitry Andric 
23670b57cec5SDimitry Andric   CollectionExprResult =
23680b57cec5SDimitry Andric       ActOnFinishFullExpr(CollectionExprResult.get(), /*DiscardedValue*/ false);
23690b57cec5SDimitry Andric   if (CollectionExprResult.isInvalid())
23700b57cec5SDimitry Andric     return StmtError();
23710b57cec5SDimitry Andric 
23720b57cec5SDimitry Andric   return new (Context) ObjCForCollectionStmt(First, CollectionExprResult.get(),
23730b57cec5SDimitry Andric                                              nullptr, ForLoc, RParenLoc);
23740b57cec5SDimitry Andric }
23750b57cec5SDimitry Andric 
23760b57cec5SDimitry Andric /// Finish building a variable declaration for a for-range statement.
23770b57cec5SDimitry Andric /// \return true if an error occurs.
23780b57cec5SDimitry Andric static bool FinishForRangeVarDecl(Sema &SemaRef, VarDecl *Decl, Expr *Init,
23790b57cec5SDimitry Andric                                   SourceLocation Loc, int DiagID) {
23800b57cec5SDimitry Andric   if (Decl->getType()->isUndeducedType()) {
23810b57cec5SDimitry Andric     ExprResult Res = SemaRef.CorrectDelayedTyposInExpr(Init);
23820b57cec5SDimitry Andric     if (!Res.isUsable()) {
23830b57cec5SDimitry Andric       Decl->setInvalidDecl();
23840b57cec5SDimitry Andric       return true;
23850b57cec5SDimitry Andric     }
23860b57cec5SDimitry Andric     Init = Res.get();
23870b57cec5SDimitry Andric   }
23880b57cec5SDimitry Andric 
23890b57cec5SDimitry Andric   // Deduce the type for the iterator variable now rather than leaving it to
23900b57cec5SDimitry Andric   // AddInitializerToDecl, so we can produce a more suitable diagnostic.
23910b57cec5SDimitry Andric   QualType InitType;
2392bdd1243dSDimitry Andric   if (!isa<InitListExpr>(Init) && Init->getType()->isVoidType()) {
23930b57cec5SDimitry Andric     SemaRef.Diag(Loc, DiagID) << Init->getType();
2394bdd1243dSDimitry Andric   } else {
2395bdd1243dSDimitry Andric     TemplateDeductionInfo Info(Init->getExprLoc());
2396bdd1243dSDimitry Andric     Sema::TemplateDeductionResult Result = SemaRef.DeduceAutoType(
2397bdd1243dSDimitry Andric         Decl->getTypeSourceInfo()->getTypeLoc(), Init, InitType, Info);
2398bdd1243dSDimitry Andric     if (Result != Sema::TDK_Success && Result != Sema::TDK_AlreadyDiagnosed)
2399bdd1243dSDimitry Andric       SemaRef.Diag(Loc, DiagID) << Init->getType();
2400bdd1243dSDimitry Andric   }
2401bdd1243dSDimitry Andric 
24020b57cec5SDimitry Andric   if (InitType.isNull()) {
24030b57cec5SDimitry Andric     Decl->setInvalidDecl();
24040b57cec5SDimitry Andric     return true;
24050b57cec5SDimitry Andric   }
24060b57cec5SDimitry Andric   Decl->setType(InitType);
24070b57cec5SDimitry Andric 
24080b57cec5SDimitry Andric   // In ARC, infer lifetime.
24090b57cec5SDimitry Andric   // FIXME: ARC may want to turn this into 'const __unsafe_unretained' if
24100b57cec5SDimitry Andric   // we're doing the equivalent of fast iteration.
24110b57cec5SDimitry Andric   if (SemaRef.getLangOpts().ObjCAutoRefCount &&
24120b57cec5SDimitry Andric       SemaRef.inferObjCARCLifetime(Decl))
24130b57cec5SDimitry Andric     Decl->setInvalidDecl();
24140b57cec5SDimitry Andric 
24150b57cec5SDimitry Andric   SemaRef.AddInitializerToDecl(Decl, Init, /*DirectInit=*/false);
24160b57cec5SDimitry Andric   SemaRef.FinalizeDeclaration(Decl);
24170b57cec5SDimitry Andric   SemaRef.CurContext->addHiddenDecl(Decl);
24180b57cec5SDimitry Andric   return false;
24190b57cec5SDimitry Andric }
24200b57cec5SDimitry Andric 
24210b57cec5SDimitry Andric namespace {
24220b57cec5SDimitry Andric // An enum to represent whether something is dealing with a call to begin()
24230b57cec5SDimitry Andric // or a call to end() in a range-based for loop.
24240b57cec5SDimitry Andric enum BeginEndFunction {
24250b57cec5SDimitry Andric   BEF_begin,
24260b57cec5SDimitry Andric   BEF_end
24270b57cec5SDimitry Andric };
24280b57cec5SDimitry Andric 
24290b57cec5SDimitry Andric /// Produce a note indicating which begin/end function was implicitly called
24300b57cec5SDimitry Andric /// by a C++11 for-range statement. This is often not obvious from the code,
24310b57cec5SDimitry Andric /// nor from the diagnostics produced when analysing the implicit expressions
24320b57cec5SDimitry Andric /// required in a for-range statement.
24330b57cec5SDimitry Andric void NoteForRangeBeginEndFunction(Sema &SemaRef, Expr *E,
24340b57cec5SDimitry Andric                                   BeginEndFunction BEF) {
24350b57cec5SDimitry Andric   CallExpr *CE = dyn_cast<CallExpr>(E);
24360b57cec5SDimitry Andric   if (!CE)
24370b57cec5SDimitry Andric     return;
24380b57cec5SDimitry Andric   FunctionDecl *D = dyn_cast<FunctionDecl>(CE->getCalleeDecl());
24390b57cec5SDimitry Andric   if (!D)
24400b57cec5SDimitry Andric     return;
24410b57cec5SDimitry Andric   SourceLocation Loc = D->getLocation();
24420b57cec5SDimitry Andric 
24430b57cec5SDimitry Andric   std::string Description;
24440b57cec5SDimitry Andric   bool IsTemplate = false;
24450b57cec5SDimitry Andric   if (FunctionTemplateDecl *FunTmpl = D->getPrimaryTemplate()) {
24460b57cec5SDimitry Andric     Description = SemaRef.getTemplateArgumentBindingsText(
24470b57cec5SDimitry Andric       FunTmpl->getTemplateParameters(), *D->getTemplateSpecializationArgs());
24480b57cec5SDimitry Andric     IsTemplate = true;
24490b57cec5SDimitry Andric   }
24500b57cec5SDimitry Andric 
24510b57cec5SDimitry Andric   SemaRef.Diag(Loc, diag::note_for_range_begin_end)
24520b57cec5SDimitry Andric     << BEF << IsTemplate << Description << E->getType();
24530b57cec5SDimitry Andric }
24540b57cec5SDimitry Andric 
24550b57cec5SDimitry Andric /// Build a variable declaration for a for-range statement.
24560b57cec5SDimitry Andric VarDecl *BuildForRangeVarDecl(Sema &SemaRef, SourceLocation Loc,
24570b57cec5SDimitry Andric                               QualType Type, StringRef Name) {
24580b57cec5SDimitry Andric   DeclContext *DC = SemaRef.CurContext;
24590b57cec5SDimitry Andric   IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name);
24600b57cec5SDimitry Andric   TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc);
24610b57cec5SDimitry Andric   VarDecl *Decl = VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type,
24620b57cec5SDimitry Andric                                   TInfo, SC_None);
24630b57cec5SDimitry Andric   Decl->setImplicit();
24640b57cec5SDimitry Andric   return Decl;
24650b57cec5SDimitry Andric }
24660b57cec5SDimitry Andric 
24670b57cec5SDimitry Andric }
24680b57cec5SDimitry Andric 
24690b57cec5SDimitry Andric static bool ObjCEnumerationCollection(Expr *Collection) {
24700b57cec5SDimitry Andric   return !Collection->isTypeDependent()
24710b57cec5SDimitry Andric           && Collection->getType()->getAs<ObjCObjectPointerType>() != nullptr;
24720b57cec5SDimitry Andric }
24730b57cec5SDimitry Andric 
24740b57cec5SDimitry Andric /// ActOnCXXForRangeStmt - Check and build a C++11 for-range statement.
24750b57cec5SDimitry Andric ///
24760b57cec5SDimitry Andric /// C++11 [stmt.ranged]:
24770b57cec5SDimitry Andric ///   A range-based for statement is equivalent to
24780b57cec5SDimitry Andric ///
24790b57cec5SDimitry Andric ///   {
24800b57cec5SDimitry Andric ///     auto && __range = range-init;
24810b57cec5SDimitry Andric ///     for ( auto __begin = begin-expr,
24820b57cec5SDimitry Andric ///           __end = end-expr;
24830b57cec5SDimitry Andric ///           __begin != __end;
24840b57cec5SDimitry Andric ///           ++__begin ) {
24850b57cec5SDimitry Andric ///       for-range-declaration = *__begin;
24860b57cec5SDimitry Andric ///       statement
24870b57cec5SDimitry Andric ///     }
24880b57cec5SDimitry Andric ///   }
24890b57cec5SDimitry Andric ///
24900b57cec5SDimitry Andric /// The body of the loop is not available yet, since it cannot be analysed until
24910b57cec5SDimitry Andric /// we have determined the type of the for-range-declaration.
24920b57cec5SDimitry Andric StmtResult Sema::ActOnCXXForRangeStmt(Scope *S, SourceLocation ForLoc,
24930b57cec5SDimitry Andric                                       SourceLocation CoawaitLoc, Stmt *InitStmt,
24940b57cec5SDimitry Andric                                       Stmt *First, SourceLocation ColonLoc,
24950b57cec5SDimitry Andric                                       Expr *Range, SourceLocation RParenLoc,
24960b57cec5SDimitry Andric                                       BuildForRangeKind Kind) {
249704eeddc0SDimitry Andric   // FIXME: recover in order to allow the body to be parsed.
24980b57cec5SDimitry Andric   if (!First)
24990b57cec5SDimitry Andric     return StmtError();
25000b57cec5SDimitry Andric 
25010b57cec5SDimitry Andric   if (Range && ObjCEnumerationCollection(Range)) {
25020b57cec5SDimitry Andric     // FIXME: Support init-statements in Objective-C++20 ranged for statement.
25030b57cec5SDimitry Andric     if (InitStmt)
25040b57cec5SDimitry Andric       return Diag(InitStmt->getBeginLoc(), diag::err_objc_for_range_init_stmt)
25050b57cec5SDimitry Andric                  << InitStmt->getSourceRange();
25060b57cec5SDimitry Andric     return ActOnObjCForCollectionStmt(ForLoc, First, Range, RParenLoc);
25070b57cec5SDimitry Andric   }
25080b57cec5SDimitry Andric 
25090b57cec5SDimitry Andric   DeclStmt *DS = dyn_cast<DeclStmt>(First);
25100b57cec5SDimitry Andric   assert(DS && "first part of for range not a decl stmt");
25110b57cec5SDimitry Andric 
25120b57cec5SDimitry Andric   if (!DS->isSingleDecl()) {
25130b57cec5SDimitry Andric     Diag(DS->getBeginLoc(), diag::err_type_defined_in_for_range);
25140b57cec5SDimitry Andric     return StmtError();
25150b57cec5SDimitry Andric   }
25160b57cec5SDimitry Andric 
25175ffd83dbSDimitry Andric   // This function is responsible for attaching an initializer to LoopVar. We
25185ffd83dbSDimitry Andric   // must call ActOnInitializerError if we fail to do so.
25190b57cec5SDimitry Andric   Decl *LoopVar = DS->getSingleDecl();
25200b57cec5SDimitry Andric   if (LoopVar->isInvalidDecl() || !Range ||
25210b57cec5SDimitry Andric       DiagnoseUnexpandedParameterPack(Range, UPPC_Expression)) {
25225ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25230b57cec5SDimitry Andric     return StmtError();
25240b57cec5SDimitry Andric   }
25250b57cec5SDimitry Andric 
25260b57cec5SDimitry Andric   // Build the coroutine state immediately and not later during template
25270b57cec5SDimitry Andric   // instantiation
25280b57cec5SDimitry Andric   if (!CoawaitLoc.isInvalid()) {
25295ffd83dbSDimitry Andric     if (!ActOnCoroutineBodyStart(S, CoawaitLoc, "co_await")) {
25305ffd83dbSDimitry Andric       ActOnInitializerError(LoopVar);
25310b57cec5SDimitry Andric       return StmtError();
25320b57cec5SDimitry Andric     }
25335ffd83dbSDimitry Andric   }
25340b57cec5SDimitry Andric 
25350b57cec5SDimitry Andric   // Build  auto && __range = range-init
25360b57cec5SDimitry Andric   // Divide by 2, since the variables are in the inner scope (loop body).
25370b57cec5SDimitry Andric   const auto DepthStr = std::to_string(S->getDepth() / 2);
25380b57cec5SDimitry Andric   SourceLocation RangeLoc = Range->getBeginLoc();
25390b57cec5SDimitry Andric   VarDecl *RangeVar = BuildForRangeVarDecl(*this, RangeLoc,
25400b57cec5SDimitry Andric                                            Context.getAutoRRefDeductType(),
25410b57cec5SDimitry Andric                                            std::string("__range") + DepthStr);
25420b57cec5SDimitry Andric   if (FinishForRangeVarDecl(*this, RangeVar, Range, RangeLoc,
25430b57cec5SDimitry Andric                             diag::err_for_range_deduction_failure)) {
25445ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25450b57cec5SDimitry Andric     return StmtError();
25460b57cec5SDimitry Andric   }
25470b57cec5SDimitry Andric 
25480b57cec5SDimitry Andric   // Claim the type doesn't contain auto: we've already done the checking.
25490b57cec5SDimitry Andric   DeclGroupPtrTy RangeGroup =
25500b57cec5SDimitry Andric       BuildDeclaratorGroup(MutableArrayRef<Decl *>((Decl **)&RangeVar, 1));
25510b57cec5SDimitry Andric   StmtResult RangeDecl = ActOnDeclStmt(RangeGroup, RangeLoc, RangeLoc);
25520b57cec5SDimitry Andric   if (RangeDecl.isInvalid()) {
25535ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25540b57cec5SDimitry Andric     return StmtError();
25550b57cec5SDimitry Andric   }
25560b57cec5SDimitry Andric 
25575ffd83dbSDimitry Andric   StmtResult R = BuildCXXForRangeStmt(
25580b57cec5SDimitry Andric       ForLoc, CoawaitLoc, InitStmt, ColonLoc, RangeDecl.get(),
25590b57cec5SDimitry Andric       /*BeginStmt=*/nullptr, /*EndStmt=*/nullptr,
25600b57cec5SDimitry Andric       /*Cond=*/nullptr, /*Inc=*/nullptr, DS, RParenLoc, Kind);
25615ffd83dbSDimitry Andric   if (R.isInvalid()) {
25625ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25635ffd83dbSDimitry Andric     return StmtError();
25645ffd83dbSDimitry Andric   }
25655ffd83dbSDimitry Andric 
25665ffd83dbSDimitry Andric   return R;
25670b57cec5SDimitry Andric }
25680b57cec5SDimitry Andric 
25690b57cec5SDimitry Andric /// Create the initialization, compare, and increment steps for
25700b57cec5SDimitry Andric /// the range-based for loop expression.
25710b57cec5SDimitry Andric /// This function does not handle array-based for loops,
25720b57cec5SDimitry Andric /// which are created in Sema::BuildCXXForRangeStmt.
25730b57cec5SDimitry Andric ///
25740b57cec5SDimitry Andric /// \returns a ForRangeStatus indicating success or what kind of error occurred.
25750b57cec5SDimitry Andric /// BeginExpr and EndExpr are set and FRS_Success is returned on success;
25760b57cec5SDimitry Andric /// CandidateSet and BEF are set and some non-success value is returned on
25770b57cec5SDimitry Andric /// failure.
25780b57cec5SDimitry Andric static Sema::ForRangeStatus
25790b57cec5SDimitry Andric BuildNonArrayForRange(Sema &SemaRef, Expr *BeginRange, Expr *EndRange,
25800b57cec5SDimitry Andric                       QualType RangeType, VarDecl *BeginVar, VarDecl *EndVar,
25810b57cec5SDimitry Andric                       SourceLocation ColonLoc, SourceLocation CoawaitLoc,
25820b57cec5SDimitry Andric                       OverloadCandidateSet *CandidateSet, ExprResult *BeginExpr,
25830b57cec5SDimitry Andric                       ExprResult *EndExpr, BeginEndFunction *BEF) {
25840b57cec5SDimitry Andric   DeclarationNameInfo BeginNameInfo(
25850b57cec5SDimitry Andric       &SemaRef.PP.getIdentifierTable().get("begin"), ColonLoc);
25860b57cec5SDimitry Andric   DeclarationNameInfo EndNameInfo(&SemaRef.PP.getIdentifierTable().get("end"),
25870b57cec5SDimitry Andric                                   ColonLoc);
25880b57cec5SDimitry Andric 
25890b57cec5SDimitry Andric   LookupResult BeginMemberLookup(SemaRef, BeginNameInfo,
25900b57cec5SDimitry Andric                                  Sema::LookupMemberName);
25910b57cec5SDimitry Andric   LookupResult EndMemberLookup(SemaRef, EndNameInfo, Sema::LookupMemberName);
25920b57cec5SDimitry Andric 
25930b57cec5SDimitry Andric   auto BuildBegin = [&] {
25940b57cec5SDimitry Andric     *BEF = BEF_begin;
25950b57cec5SDimitry Andric     Sema::ForRangeStatus RangeStatus =
25960b57cec5SDimitry Andric         SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, BeginNameInfo,
25970b57cec5SDimitry Andric                                           BeginMemberLookup, CandidateSet,
25980b57cec5SDimitry Andric                                           BeginRange, BeginExpr);
25990b57cec5SDimitry Andric 
26000b57cec5SDimitry Andric     if (RangeStatus != Sema::FRS_Success) {
26010b57cec5SDimitry Andric       if (RangeStatus == Sema::FRS_DiagnosticIssued)
26020b57cec5SDimitry Andric         SemaRef.Diag(BeginRange->getBeginLoc(), diag::note_in_for_range)
26030b57cec5SDimitry Andric             << ColonLoc << BEF_begin << BeginRange->getType();
26040b57cec5SDimitry Andric       return RangeStatus;
26050b57cec5SDimitry Andric     }
26060b57cec5SDimitry Andric     if (!CoawaitLoc.isInvalid()) {
26070b57cec5SDimitry Andric       // FIXME: getCurScope() should not be used during template instantiation.
26080b57cec5SDimitry Andric       // We should pick up the set of unqualified lookup results for operator
26090b57cec5SDimitry Andric       // co_await during the initial parse.
26100b57cec5SDimitry Andric       *BeginExpr = SemaRef.ActOnCoawaitExpr(SemaRef.getCurScope(), ColonLoc,
26110b57cec5SDimitry Andric                                             BeginExpr->get());
26120b57cec5SDimitry Andric       if (BeginExpr->isInvalid())
26130b57cec5SDimitry Andric         return Sema::FRS_DiagnosticIssued;
26140b57cec5SDimitry Andric     }
26150b57cec5SDimitry Andric     if (FinishForRangeVarDecl(SemaRef, BeginVar, BeginExpr->get(), ColonLoc,
26160b57cec5SDimitry Andric                               diag::err_for_range_iter_deduction_failure)) {
26170b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(SemaRef, BeginExpr->get(), *BEF);
26180b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26190b57cec5SDimitry Andric     }
26200b57cec5SDimitry Andric     return Sema::FRS_Success;
26210b57cec5SDimitry Andric   };
26220b57cec5SDimitry Andric 
26230b57cec5SDimitry Andric   auto BuildEnd = [&] {
26240b57cec5SDimitry Andric     *BEF = BEF_end;
26250b57cec5SDimitry Andric     Sema::ForRangeStatus RangeStatus =
26260b57cec5SDimitry Andric         SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, EndNameInfo,
26270b57cec5SDimitry Andric                                           EndMemberLookup, CandidateSet,
26280b57cec5SDimitry Andric                                           EndRange, EndExpr);
26290b57cec5SDimitry Andric     if (RangeStatus != Sema::FRS_Success) {
26300b57cec5SDimitry Andric       if (RangeStatus == Sema::FRS_DiagnosticIssued)
26310b57cec5SDimitry Andric         SemaRef.Diag(EndRange->getBeginLoc(), diag::note_in_for_range)
26320b57cec5SDimitry Andric             << ColonLoc << BEF_end << EndRange->getType();
26330b57cec5SDimitry Andric       return RangeStatus;
26340b57cec5SDimitry Andric     }
26350b57cec5SDimitry Andric     if (FinishForRangeVarDecl(SemaRef, EndVar, EndExpr->get(), ColonLoc,
26360b57cec5SDimitry Andric                               diag::err_for_range_iter_deduction_failure)) {
26370b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(SemaRef, EndExpr->get(), *BEF);
26380b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26390b57cec5SDimitry Andric     }
26400b57cec5SDimitry Andric     return Sema::FRS_Success;
26410b57cec5SDimitry Andric   };
26420b57cec5SDimitry Andric 
26430b57cec5SDimitry Andric   if (CXXRecordDecl *D = RangeType->getAsCXXRecordDecl()) {
26440b57cec5SDimitry Andric     // - if _RangeT is a class type, the unqualified-ids begin and end are
26450b57cec5SDimitry Andric     //   looked up in the scope of class _RangeT as if by class member access
26460b57cec5SDimitry Andric     //   lookup (3.4.5), and if either (or both) finds at least one
26470b57cec5SDimitry Andric     //   declaration, begin-expr and end-expr are __range.begin() and
26480b57cec5SDimitry Andric     //   __range.end(), respectively;
26490b57cec5SDimitry Andric     SemaRef.LookupQualifiedName(BeginMemberLookup, D);
26500b57cec5SDimitry Andric     if (BeginMemberLookup.isAmbiguous())
26510b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26520b57cec5SDimitry Andric 
26530b57cec5SDimitry Andric     SemaRef.LookupQualifiedName(EndMemberLookup, D);
26540b57cec5SDimitry Andric     if (EndMemberLookup.isAmbiguous())
26550b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26560b57cec5SDimitry Andric 
26570b57cec5SDimitry Andric     if (BeginMemberLookup.empty() != EndMemberLookup.empty()) {
26580b57cec5SDimitry Andric       // Look up the non-member form of the member we didn't find, first.
26590b57cec5SDimitry Andric       // This way we prefer a "no viable 'end'" diagnostic over a "i found
26600b57cec5SDimitry Andric       // a 'begin' but ignored it because there was no member 'end'"
26610b57cec5SDimitry Andric       // diagnostic.
26620b57cec5SDimitry Andric       auto BuildNonmember = [&](
26630b57cec5SDimitry Andric           BeginEndFunction BEFFound, LookupResult &Found,
26640b57cec5SDimitry Andric           llvm::function_ref<Sema::ForRangeStatus()> BuildFound,
26650b57cec5SDimitry Andric           llvm::function_ref<Sema::ForRangeStatus()> BuildNotFound) {
26660b57cec5SDimitry Andric         LookupResult OldFound = std::move(Found);
26670b57cec5SDimitry Andric         Found.clear();
26680b57cec5SDimitry Andric 
26690b57cec5SDimitry Andric         if (Sema::ForRangeStatus Result = BuildNotFound())
26700b57cec5SDimitry Andric           return Result;
26710b57cec5SDimitry Andric 
26720b57cec5SDimitry Andric         switch (BuildFound()) {
26730b57cec5SDimitry Andric         case Sema::FRS_Success:
26740b57cec5SDimitry Andric           return Sema::FRS_Success;
26750b57cec5SDimitry Andric 
26760b57cec5SDimitry Andric         case Sema::FRS_NoViableFunction:
26770b57cec5SDimitry Andric           CandidateSet->NoteCandidates(
26780b57cec5SDimitry Andric               PartialDiagnosticAt(BeginRange->getBeginLoc(),
26790b57cec5SDimitry Andric                                   SemaRef.PDiag(diag::err_for_range_invalid)
26800b57cec5SDimitry Andric                                       << BeginRange->getType() << BEFFound),
26810b57cec5SDimitry Andric               SemaRef, OCD_AllCandidates, BeginRange);
2682bdd1243dSDimitry Andric           [[fallthrough]];
26830b57cec5SDimitry Andric 
26840b57cec5SDimitry Andric         case Sema::FRS_DiagnosticIssued:
26850b57cec5SDimitry Andric           for (NamedDecl *D : OldFound) {
26860b57cec5SDimitry Andric             SemaRef.Diag(D->getLocation(),
26870b57cec5SDimitry Andric                          diag::note_for_range_member_begin_end_ignored)
26880b57cec5SDimitry Andric                 << BeginRange->getType() << BEFFound;
26890b57cec5SDimitry Andric           }
26900b57cec5SDimitry Andric           return Sema::FRS_DiagnosticIssued;
26910b57cec5SDimitry Andric         }
26920b57cec5SDimitry Andric         llvm_unreachable("unexpected ForRangeStatus");
26930b57cec5SDimitry Andric       };
26940b57cec5SDimitry Andric       if (BeginMemberLookup.empty())
26950b57cec5SDimitry Andric         return BuildNonmember(BEF_end, EndMemberLookup, BuildEnd, BuildBegin);
26960b57cec5SDimitry Andric       return BuildNonmember(BEF_begin, BeginMemberLookup, BuildBegin, BuildEnd);
26970b57cec5SDimitry Andric     }
26980b57cec5SDimitry Andric   } else {
26990b57cec5SDimitry Andric     // - otherwise, begin-expr and end-expr are begin(__range) and
27000b57cec5SDimitry Andric     //   end(__range), respectively, where begin and end are looked up with
27010b57cec5SDimitry Andric     //   argument-dependent lookup (3.4.2). For the purposes of this name
27020b57cec5SDimitry Andric     //   lookup, namespace std is an associated namespace.
27030b57cec5SDimitry Andric   }
27040b57cec5SDimitry Andric 
27050b57cec5SDimitry Andric   if (Sema::ForRangeStatus Result = BuildBegin())
27060b57cec5SDimitry Andric     return Result;
27070b57cec5SDimitry Andric   return BuildEnd();
27080b57cec5SDimitry Andric }
27090b57cec5SDimitry Andric 
27100b57cec5SDimitry Andric /// Speculatively attempt to dereference an invalid range expression.
27110b57cec5SDimitry Andric /// If the attempt fails, this function will return a valid, null StmtResult
27120b57cec5SDimitry Andric /// and emit no diagnostics.
27130b57cec5SDimitry Andric static StmtResult RebuildForRangeWithDereference(Sema &SemaRef, Scope *S,
27140b57cec5SDimitry Andric                                                  SourceLocation ForLoc,
27150b57cec5SDimitry Andric                                                  SourceLocation CoawaitLoc,
27160b57cec5SDimitry Andric                                                  Stmt *InitStmt,
27170b57cec5SDimitry Andric                                                  Stmt *LoopVarDecl,
27180b57cec5SDimitry Andric                                                  SourceLocation ColonLoc,
27190b57cec5SDimitry Andric                                                  Expr *Range,
27200b57cec5SDimitry Andric                                                  SourceLocation RangeLoc,
27210b57cec5SDimitry Andric                                                  SourceLocation RParenLoc) {
27220b57cec5SDimitry Andric   // Determine whether we can rebuild the for-range statement with a
27230b57cec5SDimitry Andric   // dereferenced range expression.
27240b57cec5SDimitry Andric   ExprResult AdjustedRange;
27250b57cec5SDimitry Andric   {
27260b57cec5SDimitry Andric     Sema::SFINAETrap Trap(SemaRef);
27270b57cec5SDimitry Andric 
27280b57cec5SDimitry Andric     AdjustedRange = SemaRef.BuildUnaryOp(S, RangeLoc, UO_Deref, Range);
27290b57cec5SDimitry Andric     if (AdjustedRange.isInvalid())
27300b57cec5SDimitry Andric       return StmtResult();
27310b57cec5SDimitry Andric 
27320b57cec5SDimitry Andric     StmtResult SR = SemaRef.ActOnCXXForRangeStmt(
27330b57cec5SDimitry Andric         S, ForLoc, CoawaitLoc, InitStmt, LoopVarDecl, ColonLoc,
27340b57cec5SDimitry Andric         AdjustedRange.get(), RParenLoc, Sema::BFRK_Check);
27350b57cec5SDimitry Andric     if (SR.isInvalid())
27360b57cec5SDimitry Andric       return StmtResult();
27370b57cec5SDimitry Andric   }
27380b57cec5SDimitry Andric 
27390b57cec5SDimitry Andric   // The attempt to dereference worked well enough that it could produce a valid
27400b57cec5SDimitry Andric   // loop. Produce a fixit, and rebuild the loop with diagnostics enabled, in
27410b57cec5SDimitry Andric   // case there are any other (non-fatal) problems with it.
27420b57cec5SDimitry Andric   SemaRef.Diag(RangeLoc, diag::err_for_range_dereference)
27430b57cec5SDimitry Andric     << Range->getType() << FixItHint::CreateInsertion(RangeLoc, "*");
27440b57cec5SDimitry Andric   return SemaRef.ActOnCXXForRangeStmt(
27450b57cec5SDimitry Andric       S, ForLoc, CoawaitLoc, InitStmt, LoopVarDecl, ColonLoc,
27460b57cec5SDimitry Andric       AdjustedRange.get(), RParenLoc, Sema::BFRK_Rebuild);
27470b57cec5SDimitry Andric }
27480b57cec5SDimitry Andric 
27490b57cec5SDimitry Andric /// BuildCXXForRangeStmt - Build or instantiate a C++11 for-range statement.
27500b57cec5SDimitry Andric StmtResult Sema::BuildCXXForRangeStmt(SourceLocation ForLoc,
27510b57cec5SDimitry Andric                                       SourceLocation CoawaitLoc, Stmt *InitStmt,
27520b57cec5SDimitry Andric                                       SourceLocation ColonLoc, Stmt *RangeDecl,
27530b57cec5SDimitry Andric                                       Stmt *Begin, Stmt *End, Expr *Cond,
27540b57cec5SDimitry Andric                                       Expr *Inc, Stmt *LoopVarDecl,
27550b57cec5SDimitry Andric                                       SourceLocation RParenLoc,
27560b57cec5SDimitry Andric                                       BuildForRangeKind Kind) {
27570b57cec5SDimitry Andric   // FIXME: This should not be used during template instantiation. We should
27580b57cec5SDimitry Andric   // pick up the set of unqualified lookup results for the != and + operators
27590b57cec5SDimitry Andric   // in the initial parse.
27600b57cec5SDimitry Andric   //
27610b57cec5SDimitry Andric   // Testcase (accepts-invalid):
27620b57cec5SDimitry Andric   //   template<typename T> void f() { for (auto x : T()) {} }
27630b57cec5SDimitry Andric   //   namespace N { struct X { X begin(); X end(); int operator*(); }; }
27640b57cec5SDimitry Andric   //   bool operator!=(N::X, N::X); void operator++(N::X);
27650b57cec5SDimitry Andric   //   void g() { f<N::X>(); }
27660b57cec5SDimitry Andric   Scope *S = getCurScope();
27670b57cec5SDimitry Andric 
27680b57cec5SDimitry Andric   DeclStmt *RangeDS = cast<DeclStmt>(RangeDecl);
27690b57cec5SDimitry Andric   VarDecl *RangeVar = cast<VarDecl>(RangeDS->getSingleDecl());
27700b57cec5SDimitry Andric   QualType RangeVarType = RangeVar->getType();
27710b57cec5SDimitry Andric 
27720b57cec5SDimitry Andric   DeclStmt *LoopVarDS = cast<DeclStmt>(LoopVarDecl);
27730b57cec5SDimitry Andric   VarDecl *LoopVar = cast<VarDecl>(LoopVarDS->getSingleDecl());
27740b57cec5SDimitry Andric 
27750b57cec5SDimitry Andric   StmtResult BeginDeclStmt = Begin;
27760b57cec5SDimitry Andric   StmtResult EndDeclStmt = End;
27770b57cec5SDimitry Andric   ExprResult NotEqExpr = Cond, IncrExpr = Inc;
27780b57cec5SDimitry Andric 
27790b57cec5SDimitry Andric   if (RangeVarType->isDependentType()) {
27800b57cec5SDimitry Andric     // The range is implicitly used as a placeholder when it is dependent.
27810b57cec5SDimitry Andric     RangeVar->markUsed(Context);
27820b57cec5SDimitry Andric 
27830b57cec5SDimitry Andric     // Deduce any 'auto's in the loop variable as 'DependentTy'. We'll fill
27840b57cec5SDimitry Andric     // them in properly when we instantiate the loop.
27850b57cec5SDimitry Andric     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
27860b57cec5SDimitry Andric       if (auto *DD = dyn_cast<DecompositionDecl>(LoopVar))
27870b57cec5SDimitry Andric         for (auto *Binding : DD->bindings())
27880b57cec5SDimitry Andric           Binding->setType(Context.DependentTy);
2789349cc55cSDimitry Andric       LoopVar->setType(SubstAutoTypeDependent(LoopVar->getType()));
27900b57cec5SDimitry Andric     }
27910b57cec5SDimitry Andric   } else if (!BeginDeclStmt.get()) {
27920b57cec5SDimitry Andric     SourceLocation RangeLoc = RangeVar->getLocation();
27930b57cec5SDimitry Andric 
27940b57cec5SDimitry Andric     const QualType RangeVarNonRefType = RangeVarType.getNonReferenceType();
27950b57cec5SDimitry Andric 
27960b57cec5SDimitry Andric     ExprResult BeginRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
27970b57cec5SDimitry Andric                                                 VK_LValue, ColonLoc);
27980b57cec5SDimitry Andric     if (BeginRangeRef.isInvalid())
27990b57cec5SDimitry Andric       return StmtError();
28000b57cec5SDimitry Andric 
28010b57cec5SDimitry Andric     ExprResult EndRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
28020b57cec5SDimitry Andric                                               VK_LValue, ColonLoc);
28030b57cec5SDimitry Andric     if (EndRangeRef.isInvalid())
28040b57cec5SDimitry Andric       return StmtError();
28050b57cec5SDimitry Andric 
28060b57cec5SDimitry Andric     QualType AutoType = Context.getAutoDeductType();
28070b57cec5SDimitry Andric     Expr *Range = RangeVar->getInit();
28080b57cec5SDimitry Andric     if (!Range)
28090b57cec5SDimitry Andric       return StmtError();
28100b57cec5SDimitry Andric     QualType RangeType = Range->getType();
28110b57cec5SDimitry Andric 
28120b57cec5SDimitry Andric     if (RequireCompleteType(RangeLoc, RangeType,
28130b57cec5SDimitry Andric                             diag::err_for_range_incomplete_type))
28140b57cec5SDimitry Andric       return StmtError();
28150b57cec5SDimitry Andric 
28160b57cec5SDimitry Andric     // Build auto __begin = begin-expr, __end = end-expr.
28170b57cec5SDimitry Andric     // Divide by 2, since the variables are in the inner scope (loop body).
28180b57cec5SDimitry Andric     const auto DepthStr = std::to_string(S->getDepth() / 2);
28190b57cec5SDimitry Andric     VarDecl *BeginVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
28200b57cec5SDimitry Andric                                              std::string("__begin") + DepthStr);
28210b57cec5SDimitry Andric     VarDecl *EndVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
28220b57cec5SDimitry Andric                                            std::string("__end") + DepthStr);
28230b57cec5SDimitry Andric 
28240b57cec5SDimitry Andric     // Build begin-expr and end-expr and attach to __begin and __end variables.
28250b57cec5SDimitry Andric     ExprResult BeginExpr, EndExpr;
28260b57cec5SDimitry Andric     if (const ArrayType *UnqAT = RangeType->getAsArrayTypeUnsafe()) {
28270b57cec5SDimitry Andric       // - if _RangeT is an array type, begin-expr and end-expr are __range and
28280b57cec5SDimitry Andric       //   __range + __bound, respectively, where __bound is the array bound. If
28290b57cec5SDimitry Andric       //   _RangeT is an array of unknown size or an array of incomplete type,
28300b57cec5SDimitry Andric       //   the program is ill-formed;
28310b57cec5SDimitry Andric 
28320b57cec5SDimitry Andric       // begin-expr is __range.
28330b57cec5SDimitry Andric       BeginExpr = BeginRangeRef;
28340b57cec5SDimitry Andric       if (!CoawaitLoc.isInvalid()) {
28350b57cec5SDimitry Andric         BeginExpr = ActOnCoawaitExpr(S, ColonLoc, BeginExpr.get());
28360b57cec5SDimitry Andric         if (BeginExpr.isInvalid())
28370b57cec5SDimitry Andric           return StmtError();
28380b57cec5SDimitry Andric       }
28390b57cec5SDimitry Andric       if (FinishForRangeVarDecl(*this, BeginVar, BeginRangeRef.get(), ColonLoc,
28400b57cec5SDimitry Andric                                 diag::err_for_range_iter_deduction_failure)) {
28410b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
28420b57cec5SDimitry Andric         return StmtError();
28430b57cec5SDimitry Andric       }
28440b57cec5SDimitry Andric 
28450b57cec5SDimitry Andric       // Find the array bound.
28460b57cec5SDimitry Andric       ExprResult BoundExpr;
28470b57cec5SDimitry Andric       if (const ConstantArrayType *CAT = dyn_cast<ConstantArrayType>(UnqAT))
28480b57cec5SDimitry Andric         BoundExpr = IntegerLiteral::Create(
28490b57cec5SDimitry Andric             Context, CAT->getSize(), Context.getPointerDiffType(), RangeLoc);
28500b57cec5SDimitry Andric       else if (const VariableArrayType *VAT =
28510b57cec5SDimitry Andric                dyn_cast<VariableArrayType>(UnqAT)) {
28520b57cec5SDimitry Andric         // For a variably modified type we can't just use the expression within
28530b57cec5SDimitry Andric         // the array bounds, since we don't want that to be re-evaluated here.
28540b57cec5SDimitry Andric         // Rather, we need to determine what it was when the array was first
28550b57cec5SDimitry Andric         // created - so we resort to using sizeof(vla)/sizeof(element).
28560b57cec5SDimitry Andric         // For e.g.
28570b57cec5SDimitry Andric         //  void f(int b) {
28580b57cec5SDimitry Andric         //    int vla[b];
28590b57cec5SDimitry Andric         //    b = -1;   <-- This should not affect the num of iterations below
28600b57cec5SDimitry Andric         //    for (int &c : vla) { .. }
28610b57cec5SDimitry Andric         //  }
28620b57cec5SDimitry Andric 
28630b57cec5SDimitry Andric         // FIXME: This results in codegen generating IR that recalculates the
28640b57cec5SDimitry Andric         // run-time number of elements (as opposed to just using the IR Value
28650b57cec5SDimitry Andric         // that corresponds to the run-time value of each bound that was
28660b57cec5SDimitry Andric         // generated when the array was created.) If this proves too embarrassing
28670b57cec5SDimitry Andric         // even for unoptimized IR, consider passing a magic-value/cookie to
28680b57cec5SDimitry Andric         // codegen that then knows to simply use that initial llvm::Value (that
28690b57cec5SDimitry Andric         // corresponds to the bound at time of array creation) within
28700b57cec5SDimitry Andric         // getelementptr.  But be prepared to pay the price of increasing a
28710b57cec5SDimitry Andric         // customized form of coupling between the two components - which  could
28720b57cec5SDimitry Andric         // be hard to maintain as the codebase evolves.
28730b57cec5SDimitry Andric 
28740b57cec5SDimitry Andric         ExprResult SizeOfVLAExprR = ActOnUnaryExprOrTypeTraitExpr(
28750b57cec5SDimitry Andric             EndVar->getLocation(), UETT_SizeOf,
28760b57cec5SDimitry Andric             /*IsType=*/true,
28770b57cec5SDimitry Andric             CreateParsedType(VAT->desugar(), Context.getTrivialTypeSourceInfo(
28780b57cec5SDimitry Andric                                                  VAT->desugar(), RangeLoc))
28790b57cec5SDimitry Andric                 .getAsOpaquePtr(),
28800b57cec5SDimitry Andric             EndVar->getSourceRange());
28810b57cec5SDimitry Andric         if (SizeOfVLAExprR.isInvalid())
28820b57cec5SDimitry Andric           return StmtError();
28830b57cec5SDimitry Andric 
28840b57cec5SDimitry Andric         ExprResult SizeOfEachElementExprR = ActOnUnaryExprOrTypeTraitExpr(
28850b57cec5SDimitry Andric             EndVar->getLocation(), UETT_SizeOf,
28860b57cec5SDimitry Andric             /*IsType=*/true,
28870b57cec5SDimitry Andric             CreateParsedType(VAT->desugar(),
28880b57cec5SDimitry Andric                              Context.getTrivialTypeSourceInfo(
28890b57cec5SDimitry Andric                                  VAT->getElementType(), RangeLoc))
28900b57cec5SDimitry Andric                 .getAsOpaquePtr(),
28910b57cec5SDimitry Andric             EndVar->getSourceRange());
28920b57cec5SDimitry Andric         if (SizeOfEachElementExprR.isInvalid())
28930b57cec5SDimitry Andric           return StmtError();
28940b57cec5SDimitry Andric 
28950b57cec5SDimitry Andric         BoundExpr =
28960b57cec5SDimitry Andric             ActOnBinOp(S, EndVar->getLocation(), tok::slash,
28970b57cec5SDimitry Andric                        SizeOfVLAExprR.get(), SizeOfEachElementExprR.get());
28980b57cec5SDimitry Andric         if (BoundExpr.isInvalid())
28990b57cec5SDimitry Andric           return StmtError();
29000b57cec5SDimitry Andric 
29010b57cec5SDimitry Andric       } else {
29020b57cec5SDimitry Andric         // Can't be a DependentSizedArrayType or an IncompleteArrayType since
29030b57cec5SDimitry Andric         // UnqAT is not incomplete and Range is not type-dependent.
29040b57cec5SDimitry Andric         llvm_unreachable("Unexpected array type in for-range");
29050b57cec5SDimitry Andric       }
29060b57cec5SDimitry Andric 
29070b57cec5SDimitry Andric       // end-expr is __range + __bound.
29080b57cec5SDimitry Andric       EndExpr = ActOnBinOp(S, ColonLoc, tok::plus, EndRangeRef.get(),
29090b57cec5SDimitry Andric                            BoundExpr.get());
29100b57cec5SDimitry Andric       if (EndExpr.isInvalid())
29110b57cec5SDimitry Andric         return StmtError();
29120b57cec5SDimitry Andric       if (FinishForRangeVarDecl(*this, EndVar, EndExpr.get(), ColonLoc,
29130b57cec5SDimitry Andric                                 diag::err_for_range_iter_deduction_failure)) {
29140b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
29150b57cec5SDimitry Andric         return StmtError();
29160b57cec5SDimitry Andric       }
29170b57cec5SDimitry Andric     } else {
29180b57cec5SDimitry Andric       OverloadCandidateSet CandidateSet(RangeLoc,
29190b57cec5SDimitry Andric                                         OverloadCandidateSet::CSK_Normal);
29200b57cec5SDimitry Andric       BeginEndFunction BEFFailure;
29210b57cec5SDimitry Andric       ForRangeStatus RangeStatus = BuildNonArrayForRange(
29220b57cec5SDimitry Andric           *this, BeginRangeRef.get(), EndRangeRef.get(), RangeType, BeginVar,
29230b57cec5SDimitry Andric           EndVar, ColonLoc, CoawaitLoc, &CandidateSet, &BeginExpr, &EndExpr,
29240b57cec5SDimitry Andric           &BEFFailure);
29250b57cec5SDimitry Andric 
29260b57cec5SDimitry Andric       if (Kind == BFRK_Build && RangeStatus == FRS_NoViableFunction &&
29270b57cec5SDimitry Andric           BEFFailure == BEF_begin) {
29280b57cec5SDimitry Andric         // If the range is being built from an array parameter, emit a
29290b57cec5SDimitry Andric         // a diagnostic that it is being treated as a pointer.
29300b57cec5SDimitry Andric         if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Range)) {
29310b57cec5SDimitry Andric           if (ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
29320b57cec5SDimitry Andric             QualType ArrayTy = PVD->getOriginalType();
29330b57cec5SDimitry Andric             QualType PointerTy = PVD->getType();
29340b57cec5SDimitry Andric             if (PointerTy->isPointerType() && ArrayTy->isArrayType()) {
29350b57cec5SDimitry Andric               Diag(Range->getBeginLoc(), diag::err_range_on_array_parameter)
29360b57cec5SDimitry Andric                   << RangeLoc << PVD << ArrayTy << PointerTy;
29370b57cec5SDimitry Andric               Diag(PVD->getLocation(), diag::note_declared_at);
29380b57cec5SDimitry Andric               return StmtError();
29390b57cec5SDimitry Andric             }
29400b57cec5SDimitry Andric           }
29410b57cec5SDimitry Andric         }
29420b57cec5SDimitry Andric 
29430b57cec5SDimitry Andric         // If building the range failed, try dereferencing the range expression
29440b57cec5SDimitry Andric         // unless a diagnostic was issued or the end function is problematic.
29450b57cec5SDimitry Andric         StmtResult SR = RebuildForRangeWithDereference(*this, S, ForLoc,
29460b57cec5SDimitry Andric                                                        CoawaitLoc, InitStmt,
29470b57cec5SDimitry Andric                                                        LoopVarDecl, ColonLoc,
29480b57cec5SDimitry Andric                                                        Range, RangeLoc,
29490b57cec5SDimitry Andric                                                        RParenLoc);
29500b57cec5SDimitry Andric         if (SR.isInvalid() || SR.isUsable())
29510b57cec5SDimitry Andric           return SR;
29520b57cec5SDimitry Andric       }
29530b57cec5SDimitry Andric 
29540b57cec5SDimitry Andric       // Otherwise, emit diagnostics if we haven't already.
29550b57cec5SDimitry Andric       if (RangeStatus == FRS_NoViableFunction) {
29560b57cec5SDimitry Andric         Expr *Range = BEFFailure ? EndRangeRef.get() : BeginRangeRef.get();
29570b57cec5SDimitry Andric         CandidateSet.NoteCandidates(
29580b57cec5SDimitry Andric             PartialDiagnosticAt(Range->getBeginLoc(),
29590b57cec5SDimitry Andric                                 PDiag(diag::err_for_range_invalid)
29600b57cec5SDimitry Andric                                     << RangeLoc << Range->getType()
29610b57cec5SDimitry Andric                                     << BEFFailure),
29620b57cec5SDimitry Andric             *this, OCD_AllCandidates, Range);
29630b57cec5SDimitry Andric       }
29640b57cec5SDimitry Andric       // Return an error if no fix was discovered.
29650b57cec5SDimitry Andric       if (RangeStatus != FRS_Success)
29660b57cec5SDimitry Andric         return StmtError();
29670b57cec5SDimitry Andric     }
29680b57cec5SDimitry Andric 
29690b57cec5SDimitry Andric     assert(!BeginExpr.isInvalid() && !EndExpr.isInvalid() &&
29700b57cec5SDimitry Andric            "invalid range expression in for loop");
29710b57cec5SDimitry Andric 
29720b57cec5SDimitry Andric     // C++11 [dcl.spec.auto]p7: BeginType and EndType must be the same.
29730b57cec5SDimitry Andric     // C++1z removes this restriction.
29740b57cec5SDimitry Andric     QualType BeginType = BeginVar->getType(), EndType = EndVar->getType();
29750b57cec5SDimitry Andric     if (!Context.hasSameType(BeginType, EndType)) {
29760b57cec5SDimitry Andric       Diag(RangeLoc, getLangOpts().CPlusPlus17
29770b57cec5SDimitry Andric                          ? diag::warn_for_range_begin_end_types_differ
29780b57cec5SDimitry Andric                          : diag::ext_for_range_begin_end_types_differ)
29790b57cec5SDimitry Andric           << BeginType << EndType;
29800b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
29810b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
29820b57cec5SDimitry Andric     }
29830b57cec5SDimitry Andric 
29840b57cec5SDimitry Andric     BeginDeclStmt =
29850b57cec5SDimitry Andric         ActOnDeclStmt(ConvertDeclToDeclGroup(BeginVar), ColonLoc, ColonLoc);
29860b57cec5SDimitry Andric     EndDeclStmt =
29870b57cec5SDimitry Andric         ActOnDeclStmt(ConvertDeclToDeclGroup(EndVar), ColonLoc, ColonLoc);
29880b57cec5SDimitry Andric 
29890b57cec5SDimitry Andric     const QualType BeginRefNonRefType = BeginType.getNonReferenceType();
29900b57cec5SDimitry Andric     ExprResult BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
29910b57cec5SDimitry Andric                                            VK_LValue, ColonLoc);
29920b57cec5SDimitry Andric     if (BeginRef.isInvalid())
29930b57cec5SDimitry Andric       return StmtError();
29940b57cec5SDimitry Andric 
29950b57cec5SDimitry Andric     ExprResult EndRef = BuildDeclRefExpr(EndVar, EndType.getNonReferenceType(),
29960b57cec5SDimitry Andric                                          VK_LValue, ColonLoc);
29970b57cec5SDimitry Andric     if (EndRef.isInvalid())
29980b57cec5SDimitry Andric       return StmtError();
29990b57cec5SDimitry Andric 
30000b57cec5SDimitry Andric     // Build and check __begin != __end expression.
30010b57cec5SDimitry Andric     NotEqExpr = ActOnBinOp(S, ColonLoc, tok::exclaimequal,
30020b57cec5SDimitry Andric                            BeginRef.get(), EndRef.get());
30030b57cec5SDimitry Andric     if (!NotEqExpr.isInvalid())
30040b57cec5SDimitry Andric       NotEqExpr = CheckBooleanCondition(ColonLoc, NotEqExpr.get());
30050b57cec5SDimitry Andric     if (!NotEqExpr.isInvalid())
30060b57cec5SDimitry Andric       NotEqExpr =
30070b57cec5SDimitry Andric           ActOnFinishFullExpr(NotEqExpr.get(), /*DiscardedValue*/ false);
30080b57cec5SDimitry Andric     if (NotEqExpr.isInvalid()) {
30090b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
30100b57cec5SDimitry Andric         << RangeLoc << 0 << BeginRangeRef.get()->getType();
30110b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
30120b57cec5SDimitry Andric       if (!Context.hasSameType(BeginType, EndType))
30130b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
30140b57cec5SDimitry Andric       return StmtError();
30150b57cec5SDimitry Andric     }
30160b57cec5SDimitry Andric 
30170b57cec5SDimitry Andric     // Build and check ++__begin expression.
30180b57cec5SDimitry Andric     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
30190b57cec5SDimitry Andric                                 VK_LValue, ColonLoc);
30200b57cec5SDimitry Andric     if (BeginRef.isInvalid())
30210b57cec5SDimitry Andric       return StmtError();
30220b57cec5SDimitry Andric 
30230b57cec5SDimitry Andric     IncrExpr = ActOnUnaryOp(S, ColonLoc, tok::plusplus, BeginRef.get());
30240b57cec5SDimitry Andric     if (!IncrExpr.isInvalid() && CoawaitLoc.isValid())
30250b57cec5SDimitry Andric       // FIXME: getCurScope() should not be used during template instantiation.
30260b57cec5SDimitry Andric       // We should pick up the set of unqualified lookup results for operator
30270b57cec5SDimitry Andric       // co_await during the initial parse.
30280b57cec5SDimitry Andric       IncrExpr = ActOnCoawaitExpr(S, CoawaitLoc, IncrExpr.get());
30290b57cec5SDimitry Andric     if (!IncrExpr.isInvalid())
30300b57cec5SDimitry Andric       IncrExpr = ActOnFinishFullExpr(IncrExpr.get(), /*DiscardedValue*/ false);
30310b57cec5SDimitry Andric     if (IncrExpr.isInvalid()) {
30320b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
30330b57cec5SDimitry Andric         << RangeLoc << 2 << BeginRangeRef.get()->getType() ;
30340b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
30350b57cec5SDimitry Andric       return StmtError();
30360b57cec5SDimitry Andric     }
30370b57cec5SDimitry Andric 
30380b57cec5SDimitry Andric     // Build and check *__begin  expression.
30390b57cec5SDimitry Andric     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
30400b57cec5SDimitry Andric                                 VK_LValue, ColonLoc);
30410b57cec5SDimitry Andric     if (BeginRef.isInvalid())
30420b57cec5SDimitry Andric       return StmtError();
30430b57cec5SDimitry Andric 
30440b57cec5SDimitry Andric     ExprResult DerefExpr = ActOnUnaryOp(S, ColonLoc, tok::star, BeginRef.get());
30450b57cec5SDimitry Andric     if (DerefExpr.isInvalid()) {
30460b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
30470b57cec5SDimitry Andric         << RangeLoc << 1 << BeginRangeRef.get()->getType();
30480b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
30490b57cec5SDimitry Andric       return StmtError();
30500b57cec5SDimitry Andric     }
30510b57cec5SDimitry Andric 
30520b57cec5SDimitry Andric     // Attach  *__begin  as initializer for VD. Don't touch it if we're just
30530b57cec5SDimitry Andric     // trying to determine whether this would be a valid range.
30540b57cec5SDimitry Andric     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
30550b57cec5SDimitry Andric       AddInitializerToDecl(LoopVar, DerefExpr.get(), /*DirectInit=*/false);
30565ffd83dbSDimitry Andric       if (LoopVar->isInvalidDecl() ||
30575ffd83dbSDimitry Andric           (LoopVar->getInit() && LoopVar->getInit()->containsErrors()))
30580b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
30590b57cec5SDimitry Andric     }
30600b57cec5SDimitry Andric   }
30610b57cec5SDimitry Andric 
30620b57cec5SDimitry Andric   // Don't bother to actually allocate the result if we're just trying to
30630b57cec5SDimitry Andric   // determine whether it would be valid.
30640b57cec5SDimitry Andric   if (Kind == BFRK_Check)
30650b57cec5SDimitry Andric     return StmtResult();
30660b57cec5SDimitry Andric 
3067a7dea167SDimitry Andric   // In OpenMP loop region loop control variable must be private. Perform
3068a7dea167SDimitry Andric   // analysis of first part (if any).
3069a7dea167SDimitry Andric   if (getLangOpts().OpenMP >= 50 && BeginDeclStmt.isUsable())
3070a7dea167SDimitry Andric     ActOnOpenMPLoopInitialization(ForLoc, BeginDeclStmt.get());
3071a7dea167SDimitry Andric 
30720b57cec5SDimitry Andric   return new (Context) CXXForRangeStmt(
30730b57cec5SDimitry Andric       InitStmt, RangeDS, cast_or_null<DeclStmt>(BeginDeclStmt.get()),
30740b57cec5SDimitry Andric       cast_or_null<DeclStmt>(EndDeclStmt.get()), NotEqExpr.get(),
30750b57cec5SDimitry Andric       IncrExpr.get(), LoopVarDS, /*Body=*/nullptr, ForLoc, CoawaitLoc,
30760b57cec5SDimitry Andric       ColonLoc, RParenLoc);
30770b57cec5SDimitry Andric }
30780b57cec5SDimitry Andric 
30790b57cec5SDimitry Andric /// FinishObjCForCollectionStmt - Attach the body to a objective-C foreach
30800b57cec5SDimitry Andric /// statement.
30810b57cec5SDimitry Andric StmtResult Sema::FinishObjCForCollectionStmt(Stmt *S, Stmt *B) {
30820b57cec5SDimitry Andric   if (!S || !B)
30830b57cec5SDimitry Andric     return StmtError();
30840b57cec5SDimitry Andric   ObjCForCollectionStmt * ForStmt = cast<ObjCForCollectionStmt>(S);
30850b57cec5SDimitry Andric 
30860b57cec5SDimitry Andric   ForStmt->setBody(B);
30870b57cec5SDimitry Andric   return S;
30880b57cec5SDimitry Andric }
30890b57cec5SDimitry Andric 
30900b57cec5SDimitry Andric // Warn when the loop variable is a const reference that creates a copy.
30910b57cec5SDimitry Andric // Suggest using the non-reference type for copies.  If a copy can be prevented
30920b57cec5SDimitry Andric // suggest the const reference type that would do so.
30930b57cec5SDimitry Andric // For instance, given "for (const &Foo : Range)", suggest
30940b57cec5SDimitry Andric // "for (const Foo : Range)" to denote a copy is made for the loop.  If
30950b57cec5SDimitry Andric // possible, also suggest "for (const &Bar : Range)" if this type prevents
30960b57cec5SDimitry Andric // the copy altogether.
30970b57cec5SDimitry Andric static void DiagnoseForRangeReferenceVariableCopies(Sema &SemaRef,
30980b57cec5SDimitry Andric                                                     const VarDecl *VD,
30990b57cec5SDimitry Andric                                                     QualType RangeInitType) {
31000b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
31010b57cec5SDimitry Andric   if (!InitExpr)
31020b57cec5SDimitry Andric     return;
31030b57cec5SDimitry Andric 
31040b57cec5SDimitry Andric   QualType VariableType = VD->getType();
31050b57cec5SDimitry Andric 
31060b57cec5SDimitry Andric   if (auto Cleanups = dyn_cast<ExprWithCleanups>(InitExpr))
31070b57cec5SDimitry Andric     if (!Cleanups->cleanupsHaveSideEffects())
31080b57cec5SDimitry Andric       InitExpr = Cleanups->getSubExpr();
31090b57cec5SDimitry Andric 
31100b57cec5SDimitry Andric   const MaterializeTemporaryExpr *MTE =
31110b57cec5SDimitry Andric       dyn_cast<MaterializeTemporaryExpr>(InitExpr);
31120b57cec5SDimitry Andric 
31130b57cec5SDimitry Andric   // No copy made.
31140b57cec5SDimitry Andric   if (!MTE)
31150b57cec5SDimitry Andric     return;
31160b57cec5SDimitry Andric 
3117480093f4SDimitry Andric   const Expr *E = MTE->getSubExpr()->IgnoreImpCasts();
31180b57cec5SDimitry Andric 
31190b57cec5SDimitry Andric   // Searching for either UnaryOperator for dereference of a pointer or
31200b57cec5SDimitry Andric   // CXXOperatorCallExpr for handling iterators.
31210b57cec5SDimitry Andric   while (!isa<CXXOperatorCallExpr>(E) && !isa<UnaryOperator>(E)) {
31220b57cec5SDimitry Andric     if (const CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(E)) {
31230b57cec5SDimitry Andric       E = CCE->getArg(0);
31240b57cec5SDimitry Andric     } else if (const CXXMemberCallExpr *Call = dyn_cast<CXXMemberCallExpr>(E)) {
31250b57cec5SDimitry Andric       const MemberExpr *ME = cast<MemberExpr>(Call->getCallee());
31260b57cec5SDimitry Andric       E = ME->getBase();
31270b57cec5SDimitry Andric     } else {
31280b57cec5SDimitry Andric       const MaterializeTemporaryExpr *MTE = cast<MaterializeTemporaryExpr>(E);
3129480093f4SDimitry Andric       E = MTE->getSubExpr();
31300b57cec5SDimitry Andric     }
31310b57cec5SDimitry Andric     E = E->IgnoreImpCasts();
31320b57cec5SDimitry Andric   }
31330b57cec5SDimitry Andric 
31345ffd83dbSDimitry Andric   QualType ReferenceReturnType;
31350b57cec5SDimitry Andric   if (isa<UnaryOperator>(E)) {
31365ffd83dbSDimitry Andric     ReferenceReturnType = SemaRef.Context.getLValueReferenceType(E->getType());
31370b57cec5SDimitry Andric   } else {
31380b57cec5SDimitry Andric     const CXXOperatorCallExpr *Call = cast<CXXOperatorCallExpr>(E);
31390b57cec5SDimitry Andric     const FunctionDecl *FD = Call->getDirectCallee();
31400b57cec5SDimitry Andric     QualType ReturnType = FD->getReturnType();
31415ffd83dbSDimitry Andric     if (ReturnType->isReferenceType())
31425ffd83dbSDimitry Andric       ReferenceReturnType = ReturnType;
31430b57cec5SDimitry Andric   }
31440b57cec5SDimitry Andric 
31455ffd83dbSDimitry Andric   if (!ReferenceReturnType.isNull()) {
31460b57cec5SDimitry Andric     // Loop variable creates a temporary.  Suggest either to go with
31470b57cec5SDimitry Andric     // non-reference loop variable to indicate a copy is made, or
31485ffd83dbSDimitry Andric     // the correct type to bind a const reference.
31495ffd83dbSDimitry Andric     SemaRef.Diag(VD->getLocation(),
31505ffd83dbSDimitry Andric                  diag::warn_for_range_const_ref_binds_temp_built_from_ref)
31515ffd83dbSDimitry Andric         << VD << VariableType << ReferenceReturnType;
31520b57cec5SDimitry Andric     QualType NonReferenceType = VariableType.getNonReferenceType();
31530b57cec5SDimitry Andric     NonReferenceType.removeLocalConst();
31540b57cec5SDimitry Andric     QualType NewReferenceType =
31550b57cec5SDimitry Andric         SemaRef.Context.getLValueReferenceType(E->getType().withConst());
31560b57cec5SDimitry Andric     SemaRef.Diag(VD->getBeginLoc(), diag::note_use_type_or_non_reference)
3157480093f4SDimitry Andric         << NonReferenceType << NewReferenceType << VD->getSourceRange()
3158480093f4SDimitry Andric         << FixItHint::CreateRemoval(VD->getTypeSpecEndLoc());
3159480093f4SDimitry Andric   } else if (!VariableType->isRValueReferenceType()) {
31600b57cec5SDimitry Andric     // The range always returns a copy, so a temporary is always created.
31610b57cec5SDimitry Andric     // Suggest removing the reference from the loop variable.
3162480093f4SDimitry Andric     // If the type is a rvalue reference do not warn since that changes the
3163480093f4SDimitry Andric     // semantic of the code.
31645ffd83dbSDimitry Andric     SemaRef.Diag(VD->getLocation(), diag::warn_for_range_ref_binds_ret_temp)
31650b57cec5SDimitry Andric         << VD << RangeInitType;
31660b57cec5SDimitry Andric     QualType NonReferenceType = VariableType.getNonReferenceType();
31670b57cec5SDimitry Andric     NonReferenceType.removeLocalConst();
31680b57cec5SDimitry Andric     SemaRef.Diag(VD->getBeginLoc(), diag::note_use_non_reference_type)
3169480093f4SDimitry Andric         << NonReferenceType << VD->getSourceRange()
3170480093f4SDimitry Andric         << FixItHint::CreateRemoval(VD->getTypeSpecEndLoc());
31710b57cec5SDimitry Andric   }
31720b57cec5SDimitry Andric }
31730b57cec5SDimitry Andric 
3174480093f4SDimitry Andric /// Determines whether the @p VariableType's declaration is a record with the
3175480093f4SDimitry Andric /// clang::trivial_abi attribute.
3176480093f4SDimitry Andric static bool hasTrivialABIAttr(QualType VariableType) {
3177480093f4SDimitry Andric   if (CXXRecordDecl *RD = VariableType->getAsCXXRecordDecl())
3178480093f4SDimitry Andric     return RD->hasAttr<TrivialABIAttr>();
3179480093f4SDimitry Andric 
3180480093f4SDimitry Andric   return false;
3181480093f4SDimitry Andric }
3182480093f4SDimitry Andric 
31830b57cec5SDimitry Andric // Warns when the loop variable can be changed to a reference type to
31840b57cec5SDimitry Andric // prevent a copy.  For instance, if given "for (const Foo x : Range)" suggest
31850b57cec5SDimitry Andric // "for (const Foo &x : Range)" if this form does not make a copy.
31860b57cec5SDimitry Andric static void DiagnoseForRangeConstVariableCopies(Sema &SemaRef,
31870b57cec5SDimitry Andric                                                 const VarDecl *VD) {
31880b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
31890b57cec5SDimitry Andric   if (!InitExpr)
31900b57cec5SDimitry Andric     return;
31910b57cec5SDimitry Andric 
31920b57cec5SDimitry Andric   QualType VariableType = VD->getType();
31930b57cec5SDimitry Andric 
31940b57cec5SDimitry Andric   if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(InitExpr)) {
31950b57cec5SDimitry Andric     if (!CE->getConstructor()->isCopyConstructor())
31960b57cec5SDimitry Andric       return;
31970b57cec5SDimitry Andric   } else if (const CastExpr *CE = dyn_cast<CastExpr>(InitExpr)) {
31980b57cec5SDimitry Andric     if (CE->getCastKind() != CK_LValueToRValue)
31990b57cec5SDimitry Andric       return;
32000b57cec5SDimitry Andric   } else {
32010b57cec5SDimitry Andric     return;
32020b57cec5SDimitry Andric   }
32030b57cec5SDimitry Andric 
3204480093f4SDimitry Andric   // Small trivially copyable types are cheap to copy. Do not emit the
3205480093f4SDimitry Andric   // diagnostic for these instances. 64 bytes is a common size of a cache line.
3206480093f4SDimitry Andric   // (The function `getTypeSize` returns the size in bits.)
3207480093f4SDimitry Andric   ASTContext &Ctx = SemaRef.Context;
3208480093f4SDimitry Andric   if (Ctx.getTypeSize(VariableType) <= 64 * 8 &&
3209297eecfbSDimitry Andric       (VariableType.isTriviallyCopyConstructibleType(Ctx) ||
3210480093f4SDimitry Andric        hasTrivialABIAttr(VariableType)))
32110b57cec5SDimitry Andric     return;
32120b57cec5SDimitry Andric 
32130b57cec5SDimitry Andric   // Suggest changing from a const variable to a const reference variable
32140b57cec5SDimitry Andric   // if doing so will prevent a copy.
32150b57cec5SDimitry Andric   SemaRef.Diag(VD->getLocation(), diag::warn_for_range_copy)
32165ffd83dbSDimitry Andric       << VD << VariableType;
32170b57cec5SDimitry Andric   SemaRef.Diag(VD->getBeginLoc(), diag::note_use_reference_type)
32180b57cec5SDimitry Andric       << SemaRef.Context.getLValueReferenceType(VariableType)
3219480093f4SDimitry Andric       << VD->getSourceRange()
3220480093f4SDimitry Andric       << FixItHint::CreateInsertion(VD->getLocation(), "&");
32210b57cec5SDimitry Andric }
32220b57cec5SDimitry Andric 
32230b57cec5SDimitry Andric /// DiagnoseForRangeVariableCopies - Diagnose three cases and fixes for them.
32240b57cec5SDimitry Andric /// 1) for (const foo &x : foos) where foos only returns a copy.  Suggest
32250b57cec5SDimitry Andric ///    using "const foo x" to show that a copy is made
32260b57cec5SDimitry Andric /// 2) for (const bar &x : foos) where bar is a temporary initialized by bar.
32270b57cec5SDimitry Andric ///    Suggest either "const bar x" to keep the copying or "const foo& x" to
32280b57cec5SDimitry Andric ///    prevent the copy.
32290b57cec5SDimitry Andric /// 3) for (const foo x : foos) where x is constructed from a reference foo.
32300b57cec5SDimitry Andric ///    Suggest "const foo &x" to prevent the copy.
32310b57cec5SDimitry Andric static void DiagnoseForRangeVariableCopies(Sema &SemaRef,
32320b57cec5SDimitry Andric                                            const CXXForRangeStmt *ForStmt) {
323355e4f9d5SDimitry Andric   if (SemaRef.inTemplateInstantiation())
323455e4f9d5SDimitry Andric     return;
323555e4f9d5SDimitry Andric 
32365ffd83dbSDimitry Andric   if (SemaRef.Diags.isIgnored(
32375ffd83dbSDimitry Andric           diag::warn_for_range_const_ref_binds_temp_built_from_ref,
32380b57cec5SDimitry Andric           ForStmt->getBeginLoc()) &&
32395ffd83dbSDimitry Andric       SemaRef.Diags.isIgnored(diag::warn_for_range_ref_binds_ret_temp,
32400b57cec5SDimitry Andric                               ForStmt->getBeginLoc()) &&
32410b57cec5SDimitry Andric       SemaRef.Diags.isIgnored(diag::warn_for_range_copy,
32420b57cec5SDimitry Andric                               ForStmt->getBeginLoc())) {
32430b57cec5SDimitry Andric     return;
32440b57cec5SDimitry Andric   }
32450b57cec5SDimitry Andric 
32460b57cec5SDimitry Andric   const VarDecl *VD = ForStmt->getLoopVariable();
32470b57cec5SDimitry Andric   if (!VD)
32480b57cec5SDimitry Andric     return;
32490b57cec5SDimitry Andric 
32500b57cec5SDimitry Andric   QualType VariableType = VD->getType();
32510b57cec5SDimitry Andric 
32520b57cec5SDimitry Andric   if (VariableType->isIncompleteType())
32530b57cec5SDimitry Andric     return;
32540b57cec5SDimitry Andric 
32550b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
32560b57cec5SDimitry Andric   if (!InitExpr)
32570b57cec5SDimitry Andric     return;
32580b57cec5SDimitry Andric 
325955e4f9d5SDimitry Andric   if (InitExpr->getExprLoc().isMacroID())
326055e4f9d5SDimitry Andric     return;
326155e4f9d5SDimitry Andric 
32620b57cec5SDimitry Andric   if (VariableType->isReferenceType()) {
32630b57cec5SDimitry Andric     DiagnoseForRangeReferenceVariableCopies(SemaRef, VD,
32640b57cec5SDimitry Andric                                             ForStmt->getRangeInit()->getType());
32650b57cec5SDimitry Andric   } else if (VariableType.isConstQualified()) {
32660b57cec5SDimitry Andric     DiagnoseForRangeConstVariableCopies(SemaRef, VD);
32670b57cec5SDimitry Andric   }
32680b57cec5SDimitry Andric }
32690b57cec5SDimitry Andric 
32700b57cec5SDimitry Andric /// FinishCXXForRangeStmt - Attach the body to a C++0x for-range statement.
32710b57cec5SDimitry Andric /// This is a separate step from ActOnCXXForRangeStmt because analysis of the
32720b57cec5SDimitry Andric /// body cannot be performed until after the type of the range variable is
32730b57cec5SDimitry Andric /// determined.
32740b57cec5SDimitry Andric StmtResult Sema::FinishCXXForRangeStmt(Stmt *S, Stmt *B) {
32750b57cec5SDimitry Andric   if (!S || !B)
32760b57cec5SDimitry Andric     return StmtError();
32770b57cec5SDimitry Andric 
32780b57cec5SDimitry Andric   if (isa<ObjCForCollectionStmt>(S))
32790b57cec5SDimitry Andric     return FinishObjCForCollectionStmt(S, B);
32800b57cec5SDimitry Andric 
32810b57cec5SDimitry Andric   CXXForRangeStmt *ForStmt = cast<CXXForRangeStmt>(S);
32820b57cec5SDimitry Andric   ForStmt->setBody(B);
32830b57cec5SDimitry Andric 
32840b57cec5SDimitry Andric   DiagnoseEmptyStmtBody(ForStmt->getRParenLoc(), B,
32850b57cec5SDimitry Andric                         diag::warn_empty_range_based_for_body);
32860b57cec5SDimitry Andric 
32870b57cec5SDimitry Andric   DiagnoseForRangeVariableCopies(*this, ForStmt);
32880b57cec5SDimitry Andric 
32890b57cec5SDimitry Andric   return S;
32900b57cec5SDimitry Andric }
32910b57cec5SDimitry Andric 
32920b57cec5SDimitry Andric StmtResult Sema::ActOnGotoStmt(SourceLocation GotoLoc,
32930b57cec5SDimitry Andric                                SourceLocation LabelLoc,
32940b57cec5SDimitry Andric                                LabelDecl *TheDecl) {
32950b57cec5SDimitry Andric   setFunctionHasBranchIntoScope();
32960b57cec5SDimitry Andric   TheDecl->markUsed(Context);
32970b57cec5SDimitry Andric   return new (Context) GotoStmt(TheDecl, GotoLoc, LabelLoc);
32980b57cec5SDimitry Andric }
32990b57cec5SDimitry Andric 
33000b57cec5SDimitry Andric StmtResult
33010b57cec5SDimitry Andric Sema::ActOnIndirectGotoStmt(SourceLocation GotoLoc, SourceLocation StarLoc,
33020b57cec5SDimitry Andric                             Expr *E) {
33030b57cec5SDimitry Andric   // Convert operand to void*
33040b57cec5SDimitry Andric   if (!E->isTypeDependent()) {
33050b57cec5SDimitry Andric     QualType ETy = E->getType();
33060b57cec5SDimitry Andric     QualType DestTy = Context.getPointerType(Context.VoidTy.withConst());
33070b57cec5SDimitry Andric     ExprResult ExprRes = E;
33080b57cec5SDimitry Andric     AssignConvertType ConvTy =
33090b57cec5SDimitry Andric       CheckSingleAssignmentConstraints(DestTy, ExprRes);
33100b57cec5SDimitry Andric     if (ExprRes.isInvalid())
33110b57cec5SDimitry Andric       return StmtError();
33120b57cec5SDimitry Andric     E = ExprRes.get();
33130b57cec5SDimitry Andric     if (DiagnoseAssignmentResult(ConvTy, StarLoc, DestTy, ETy, E, AA_Passing))
33140b57cec5SDimitry Andric       return StmtError();
33150b57cec5SDimitry Andric   }
33160b57cec5SDimitry Andric 
33170b57cec5SDimitry Andric   ExprResult ExprRes = ActOnFinishFullExpr(E, /*DiscardedValue*/ false);
33180b57cec5SDimitry Andric   if (ExprRes.isInvalid())
33190b57cec5SDimitry Andric     return StmtError();
33200b57cec5SDimitry Andric   E = ExprRes.get();
33210b57cec5SDimitry Andric 
33220b57cec5SDimitry Andric   setFunctionHasIndirectGoto();
33230b57cec5SDimitry Andric 
33240b57cec5SDimitry Andric   return new (Context) IndirectGotoStmt(GotoLoc, StarLoc, E);
33250b57cec5SDimitry Andric }
33260b57cec5SDimitry Andric 
33270b57cec5SDimitry Andric static void CheckJumpOutOfSEHFinally(Sema &S, SourceLocation Loc,
33280b57cec5SDimitry Andric                                      const Scope &DestScope) {
33290b57cec5SDimitry Andric   if (!S.CurrentSEHFinally.empty() &&
33300b57cec5SDimitry Andric       DestScope.Contains(*S.CurrentSEHFinally.back())) {
33310b57cec5SDimitry Andric     S.Diag(Loc, diag::warn_jump_out_of_seh_finally);
33320b57cec5SDimitry Andric   }
33330b57cec5SDimitry Andric }
33340b57cec5SDimitry Andric 
33350b57cec5SDimitry Andric StmtResult
33360b57cec5SDimitry Andric Sema::ActOnContinueStmt(SourceLocation ContinueLoc, Scope *CurScope) {
33370b57cec5SDimitry Andric   Scope *S = CurScope->getContinueParent();
33380b57cec5SDimitry Andric   if (!S) {
33390b57cec5SDimitry Andric     // C99 6.8.6.2p1: A break shall appear only in or as a loop body.
33400b57cec5SDimitry Andric     return StmtError(Diag(ContinueLoc, diag::err_continue_not_in_loop));
33410b57cec5SDimitry Andric   }
334281ad6265SDimitry Andric   if (S->isConditionVarScope()) {
3343fe6060f1SDimitry Andric     // We cannot 'continue;' from within a statement expression in the
3344fe6060f1SDimitry Andric     // initializer of a condition variable because we would jump past the
3345fe6060f1SDimitry Andric     // initialization of that variable.
3346fe6060f1SDimitry Andric     return StmtError(Diag(ContinueLoc, diag::err_continue_from_cond_var_init));
3347fe6060f1SDimitry Andric   }
33480b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, ContinueLoc, *S);
33490b57cec5SDimitry Andric 
33500b57cec5SDimitry Andric   return new (Context) ContinueStmt(ContinueLoc);
33510b57cec5SDimitry Andric }
33520b57cec5SDimitry Andric 
33530b57cec5SDimitry Andric StmtResult
33540b57cec5SDimitry Andric Sema::ActOnBreakStmt(SourceLocation BreakLoc, Scope *CurScope) {
33550b57cec5SDimitry Andric   Scope *S = CurScope->getBreakParent();
33560b57cec5SDimitry Andric   if (!S) {
33570b57cec5SDimitry Andric     // C99 6.8.6.3p1: A break shall appear only in or as a switch/loop body.
33580b57cec5SDimitry Andric     return StmtError(Diag(BreakLoc, diag::err_break_not_in_loop_or_switch));
33590b57cec5SDimitry Andric   }
33600b57cec5SDimitry Andric   if (S->isOpenMPLoopScope())
33610b57cec5SDimitry Andric     return StmtError(Diag(BreakLoc, diag::err_omp_loop_cannot_use_stmt)
33620b57cec5SDimitry Andric                      << "break");
33630b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, BreakLoc, *S);
33640b57cec5SDimitry Andric 
33650b57cec5SDimitry Andric   return new (Context) BreakStmt(BreakLoc);
33660b57cec5SDimitry Andric }
33670b57cec5SDimitry Andric 
3368fe6060f1SDimitry Andric /// Determine whether the given expression might be move-eligible or
3369fe6060f1SDimitry Andric /// copy-elidable in either a (co_)return statement or throw expression,
3370fe6060f1SDimitry Andric /// without considering function return type, if applicable.
33710b57cec5SDimitry Andric ///
3372fe6060f1SDimitry Andric /// \param E The expression being returned from the function or block,
3373fe6060f1SDimitry Andric /// being thrown, or being co_returned from a coroutine. This expression
3374fe6060f1SDimitry Andric /// might be modified by the implementation.
33750b57cec5SDimitry Andric ///
3376349cc55cSDimitry Andric /// \param Mode Overrides detection of current language mode
337706c3fb27SDimitry Andric /// and uses the rules for C++23.
33780b57cec5SDimitry Andric ///
3379fe6060f1SDimitry Andric /// \returns An aggregate which contains the Candidate and isMoveEligible
3380fe6060f1SDimitry Andric /// and isCopyElidable methods. If Candidate is non-null, it means
3381fe6060f1SDimitry Andric /// isMoveEligible() would be true under the most permissive language standard.
3382fe6060f1SDimitry Andric Sema::NamedReturnInfo Sema::getNamedReturnInfo(Expr *&E,
3383fe6060f1SDimitry Andric                                                SimplerImplicitMoveMode Mode) {
3384fe6060f1SDimitry Andric   if (!E)
3385fe6060f1SDimitry Andric     return NamedReturnInfo();
33860b57cec5SDimitry Andric   // - in a return statement in a function [where] ...
33870b57cec5SDimitry Andric   // ... the expression is the name of a non-volatile automatic object ...
3388fe6060f1SDimitry Andric   const auto *DR = dyn_cast<DeclRefExpr>(E->IgnoreParens());
33890b57cec5SDimitry Andric   if (!DR || DR->refersToEnclosingVariableOrCapture())
3390fe6060f1SDimitry Andric     return NamedReturnInfo();
3391fe6060f1SDimitry Andric   const auto *VD = dyn_cast<VarDecl>(DR->getDecl());
33920b57cec5SDimitry Andric   if (!VD)
3393fe6060f1SDimitry Andric     return NamedReturnInfo();
3394fe6060f1SDimitry Andric   NamedReturnInfo Res = getNamedReturnInfo(VD);
3395fe6060f1SDimitry Andric   if (Res.Candidate && !E->isXValue() &&
3396fe6060f1SDimitry Andric       (Mode == SimplerImplicitMoveMode::ForceOn ||
3397fe6060f1SDimitry Andric        (Mode != SimplerImplicitMoveMode::ForceOff &&
339806c3fb27SDimitry Andric         getLangOpts().CPlusPlus23))) {
3399fe6060f1SDimitry Andric     E = ImplicitCastExpr::Create(Context, VD->getType().getNonReferenceType(),
3400fe6060f1SDimitry Andric                                  CK_NoOp, E, nullptr, VK_XValue,
3401fe6060f1SDimitry Andric                                  FPOptionsOverride());
3402fe6060f1SDimitry Andric   }
3403fe6060f1SDimitry Andric   return Res;
34040b57cec5SDimitry Andric }
34050b57cec5SDimitry Andric 
3406fe6060f1SDimitry Andric /// Determine whether the given NRVO candidate variable is move-eligible or
3407fe6060f1SDimitry Andric /// copy-elidable, without considering function return type.
3408fe6060f1SDimitry Andric ///
3409fe6060f1SDimitry Andric /// \param VD The NRVO candidate variable.
3410fe6060f1SDimitry Andric ///
3411fe6060f1SDimitry Andric /// \returns An aggregate which contains the Candidate and isMoveEligible
3412fe6060f1SDimitry Andric /// and isCopyElidable methods. If Candidate is non-null, it means
3413fe6060f1SDimitry Andric /// isMoveEligible() would be true under the most permissive language standard.
3414fe6060f1SDimitry Andric Sema::NamedReturnInfo Sema::getNamedReturnInfo(const VarDecl *VD) {
3415fe6060f1SDimitry Andric   NamedReturnInfo Info{VD, NamedReturnInfo::MoveEligibleAndCopyElidable};
3416fe6060f1SDimitry Andric 
3417fe6060f1SDimitry Andric   // C++20 [class.copy.elision]p3:
34180b57cec5SDimitry Andric   // - in a return statement in a function with ...
3419fe6060f1SDimitry Andric   // (other than a function ... parameter)
3420fe6060f1SDimitry Andric   if (VD->getKind() == Decl::ParmVar)
3421fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
3422fe6060f1SDimitry Andric   else if (VD->getKind() != Decl::Var)
3423fe6060f1SDimitry Andric     return NamedReturnInfo();
34240b57cec5SDimitry Andric 
3425fe6060f1SDimitry Andric   // (other than ... a catch-clause parameter)
3426fe6060f1SDimitry Andric   if (VD->isExceptionVariable())
3427fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
34280b57cec5SDimitry Andric 
34290b57cec5SDimitry Andric   // ...automatic...
3430fe6060f1SDimitry Andric   if (!VD->hasLocalStorage())
3431fe6060f1SDimitry Andric     return NamedReturnInfo();
34320b57cec5SDimitry Andric 
3433fe6060f1SDimitry Andric   // We don't want to implicitly move out of a __block variable during a return
3434fe6060f1SDimitry Andric   // because we cannot assume the variable will no longer be used.
3435fe6060f1SDimitry Andric   if (VD->hasAttr<BlocksAttr>())
3436fe6060f1SDimitry Andric     return NamedReturnInfo();
34370b57cec5SDimitry Andric 
3438fe6060f1SDimitry Andric   QualType VDType = VD->getType();
3439fe6060f1SDimitry Andric   if (VDType->isObjectType()) {
3440fe6060f1SDimitry Andric     // C++17 [class.copy.elision]p3:
3441fe6060f1SDimitry Andric     // ...non-volatile automatic object...
3442fe6060f1SDimitry Andric     if (VDType.isVolatileQualified())
3443fe6060f1SDimitry Andric       return NamedReturnInfo();
3444fe6060f1SDimitry Andric   } else if (VDType->isRValueReferenceType()) {
3445fe6060f1SDimitry Andric     // C++20 [class.copy.elision]p3:
3446fe6060f1SDimitry Andric     // ...either a non-volatile object or an rvalue reference to a non-volatile
3447fe6060f1SDimitry Andric     // object type...
3448fe6060f1SDimitry Andric     QualType VDReferencedType = VDType.getNonReferenceType();
3449fe6060f1SDimitry Andric     if (VDReferencedType.isVolatileQualified() ||
3450fe6060f1SDimitry Andric         !VDReferencedType->isObjectType())
3451fe6060f1SDimitry Andric       return NamedReturnInfo();
3452fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
3453fe6060f1SDimitry Andric   } else {
3454fe6060f1SDimitry Andric     return NamedReturnInfo();
3455fe6060f1SDimitry Andric   }
34560b57cec5SDimitry Andric 
34570b57cec5SDimitry Andric   // Variables with higher required alignment than their type's ABI
34580b57cec5SDimitry Andric   // alignment cannot use NRVO.
3459fe6060f1SDimitry Andric   if (!VD->hasDependentAlignment() &&
3460fe6060f1SDimitry Andric       Context.getDeclAlign(VD) > Context.getTypeAlignInChars(VDType))
3461fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
3462fe6060f1SDimitry Andric 
3463fe6060f1SDimitry Andric   return Info;
3464fe6060f1SDimitry Andric }
3465fe6060f1SDimitry Andric 
3466fe6060f1SDimitry Andric /// Updates given NamedReturnInfo's move-eligible and
3467fe6060f1SDimitry Andric /// copy-elidable statuses, considering the function
3468fe6060f1SDimitry Andric /// return type criteria as applicable to return statements.
3469fe6060f1SDimitry Andric ///
3470fe6060f1SDimitry Andric /// \param Info The NamedReturnInfo object to update.
3471fe6060f1SDimitry Andric ///
3472fe6060f1SDimitry Andric /// \param ReturnType This is the return type of the function.
3473fe6060f1SDimitry Andric /// \returns The copy elision candidate, in case the initial return expression
3474fe6060f1SDimitry Andric /// was copy elidable, or nullptr otherwise.
3475fe6060f1SDimitry Andric const VarDecl *Sema::getCopyElisionCandidate(NamedReturnInfo &Info,
3476fe6060f1SDimitry Andric                                              QualType ReturnType) {
3477fe6060f1SDimitry Andric   if (!Info.Candidate)
3478fe6060f1SDimitry Andric     return nullptr;
3479fe6060f1SDimitry Andric 
3480fe6060f1SDimitry Andric   auto invalidNRVO = [&] {
3481fe6060f1SDimitry Andric     Info = NamedReturnInfo();
3482fe6060f1SDimitry Andric     return nullptr;
3483fe6060f1SDimitry Andric   };
3484fe6060f1SDimitry Andric 
3485fe6060f1SDimitry Andric   // If we got a non-deduced auto ReturnType, we are in a dependent context and
3486fe6060f1SDimitry Andric   // there is no point in allowing copy elision since we won't have it deduced
3487fe6060f1SDimitry Andric   // by the point the VardDecl is instantiated, which is the last chance we have
3488fe6060f1SDimitry Andric   // of deciding if the candidate is really copy elidable.
3489fe6060f1SDimitry Andric   if ((ReturnType->getTypeClass() == Type::TypeClass::Auto &&
3490fe6060f1SDimitry Andric        ReturnType->isCanonicalUnqualified()) ||
3491fe6060f1SDimitry Andric       ReturnType->isSpecificBuiltinType(BuiltinType::Dependent))
3492fe6060f1SDimitry Andric     return invalidNRVO();
3493fe6060f1SDimitry Andric 
3494fe6060f1SDimitry Andric   if (!ReturnType->isDependentType()) {
3495fe6060f1SDimitry Andric     // - in a return statement in a function with ...
3496fe6060f1SDimitry Andric     // ... a class return type ...
3497fe6060f1SDimitry Andric     if (!ReturnType->isRecordType())
3498fe6060f1SDimitry Andric       return invalidNRVO();
3499fe6060f1SDimitry Andric 
3500fe6060f1SDimitry Andric     QualType VDType = Info.Candidate->getType();
3501fe6060f1SDimitry Andric     // ... the same cv-unqualified type as the function return type ...
3502fe6060f1SDimitry Andric     // When considering moving this expression out, allow dissimilar types.
3503fe6060f1SDimitry Andric     if (!VDType->isDependentType() &&
3504fe6060f1SDimitry Andric         !Context.hasSameUnqualifiedType(ReturnType, VDType))
3505fe6060f1SDimitry Andric       Info.S = NamedReturnInfo::MoveEligible;
3506fe6060f1SDimitry Andric   }
3507fe6060f1SDimitry Andric   return Info.isCopyElidable() ? Info.Candidate : nullptr;
3508fe6060f1SDimitry Andric }
3509fe6060f1SDimitry Andric 
3510fe6060f1SDimitry Andric /// Verify that the initialization sequence that was picked for the
3511fe6060f1SDimitry Andric /// first overload resolution is permissible under C++98.
3512fe6060f1SDimitry Andric ///
3513349cc55cSDimitry Andric /// Reject (possibly converting) constructors not taking an rvalue reference,
3514fe6060f1SDimitry Andric /// or user conversion operators which are not ref-qualified.
3515fe6060f1SDimitry Andric static bool
3516fe6060f1SDimitry Andric VerifyInitializationSequenceCXX98(const Sema &S,
3517fe6060f1SDimitry Andric                                   const InitializationSequence &Seq) {
3518fe6060f1SDimitry Andric   const auto *Step = llvm::find_if(Seq.steps(), [](const auto &Step) {
3519fe6060f1SDimitry Andric     return Step.Kind == InitializationSequence::SK_ConstructorInitialization ||
3520fe6060f1SDimitry Andric            Step.Kind == InitializationSequence::SK_UserConversion;
3521fe6060f1SDimitry Andric   });
3522fe6060f1SDimitry Andric   if (Step != Seq.step_end()) {
3523fe6060f1SDimitry Andric     const auto *FD = Step->Function.Function;
3524fe6060f1SDimitry Andric     if (isa<CXXConstructorDecl>(FD)
3525fe6060f1SDimitry Andric             ? !FD->getParamDecl(0)->getType()->isRValueReferenceType()
3526fe6060f1SDimitry Andric             : cast<CXXMethodDecl>(FD)->getRefQualifier() == RQ_None)
35270b57cec5SDimitry Andric       return false;
3528fe6060f1SDimitry Andric   }
35290b57cec5SDimitry Andric   return true;
35300b57cec5SDimitry Andric }
35310b57cec5SDimitry Andric 
35320b57cec5SDimitry Andric /// Perform the initialization of a potentially-movable value, which
35330b57cec5SDimitry Andric /// is the result of return value.
35340b57cec5SDimitry Andric ///
3535fe6060f1SDimitry Andric /// This routine implements C++20 [class.copy.elision]p3, which attempts to
3536fe6060f1SDimitry Andric /// treat returned lvalues as rvalues in certain cases (to prefer move
3537fe6060f1SDimitry Andric /// construction), then falls back to treating them as lvalues if that failed.
3538fe6060f1SDimitry Andric ExprResult Sema::PerformMoveOrCopyInitialization(
3539fe6060f1SDimitry Andric     const InitializedEntity &Entity, const NamedReturnInfo &NRInfo, Expr *Value,
3540fe6060f1SDimitry Andric     bool SupressSimplerImplicitMoves) {
35418c6f6c0cSDimitry Andric   if (getLangOpts().CPlusPlus &&
354206c3fb27SDimitry Andric       (!getLangOpts().CPlusPlus23 || SupressSimplerImplicitMoves) &&
3543fe6060f1SDimitry Andric       NRInfo.isMoveEligible()) {
3544fe6060f1SDimitry Andric     ImplicitCastExpr AsRvalue(ImplicitCastExpr::OnStack, Value->getType(),
3545fe6060f1SDimitry Andric                               CK_NoOp, Value, VK_XValue, FPOptionsOverride());
3546fe6060f1SDimitry Andric     Expr *InitExpr = &AsRvalue;
3547fe6060f1SDimitry Andric     auto Kind = InitializationKind::CreateCopy(Value->getBeginLoc(),
3548fe6060f1SDimitry Andric                                                Value->getBeginLoc());
3549fe6060f1SDimitry Andric     InitializationSequence Seq(*this, Entity, Kind, InitExpr);
3550fe6060f1SDimitry Andric     auto Res = Seq.getFailedOverloadResult();
3551fe6060f1SDimitry Andric     if ((Res == OR_Success || Res == OR_Deleted) &&
3552fe6060f1SDimitry Andric         (getLangOpts().CPlusPlus11 ||
3553fe6060f1SDimitry Andric          VerifyInitializationSequenceCXX98(*this, Seq))) {
3554fe6060f1SDimitry Andric       // Promote "AsRvalue" to the heap, since we now need this
3555fe6060f1SDimitry Andric       // expression node to persist.
3556fe6060f1SDimitry Andric       Value =
3557fe6060f1SDimitry Andric           ImplicitCastExpr::Create(Context, Value->getType(), CK_NoOp, Value,
3558fe6060f1SDimitry Andric                                    nullptr, VK_XValue, FPOptionsOverride());
3559fe6060f1SDimitry Andric       // Complete type-checking the initialization of the return type
3560fe6060f1SDimitry Andric       // using the constructor we found.
3561fe6060f1SDimitry Andric       return Seq.Perform(*this, Entity, Kind, Value);
35620b57cec5SDimitry Andric     }
35630b57cec5SDimitry Andric   }
35640b57cec5SDimitry Andric   // Either we didn't meet the criteria for treating an lvalue as an rvalue,
35650b57cec5SDimitry Andric   // above, or overload resolution failed. Either way, we need to try
35660b57cec5SDimitry Andric   // (again) now with the return value expression as written.
3567fe6060f1SDimitry Andric   return PerformCopyInitialization(Entity, SourceLocation(), Value);
35680b57cec5SDimitry Andric }
35690b57cec5SDimitry Andric 
35700b57cec5SDimitry Andric /// Determine whether the declared return type of the specified function
35710b57cec5SDimitry Andric /// contains 'auto'.
35720b57cec5SDimitry Andric static bool hasDeducedReturnType(FunctionDecl *FD) {
35730b57cec5SDimitry Andric   const FunctionProtoType *FPT =
35740b57cec5SDimitry Andric       FD->getTypeSourceInfo()->getType()->castAs<FunctionProtoType>();
35750b57cec5SDimitry Andric   return FPT->getReturnType()->isUndeducedType();
35760b57cec5SDimitry Andric }
35770b57cec5SDimitry Andric 
35780b57cec5SDimitry Andric /// ActOnCapScopeReturnStmt - Utility routine to type-check return statements
35790b57cec5SDimitry Andric /// for capturing scopes.
35800b57cec5SDimitry Andric ///
3581fe6060f1SDimitry Andric StmtResult Sema::ActOnCapScopeReturnStmt(SourceLocation ReturnLoc,
3582fe6060f1SDimitry Andric                                          Expr *RetValExp,
3583fe6060f1SDimitry Andric                                          NamedReturnInfo &NRInfo,
3584fe6060f1SDimitry Andric                                          bool SupressSimplerImplicitMoves) {
35850b57cec5SDimitry Andric   // If this is the first return we've seen, infer the return type.
35860b57cec5SDimitry Andric   // [expr.prim.lambda]p4 in C++11; block literals follow the same rules.
35870b57cec5SDimitry Andric   CapturingScopeInfo *CurCap = cast<CapturingScopeInfo>(getCurFunction());
35880b57cec5SDimitry Andric   QualType FnRetType = CurCap->ReturnType;
35890b57cec5SDimitry Andric   LambdaScopeInfo *CurLambda = dyn_cast<LambdaScopeInfo>(CurCap);
35905f757f3fSDimitry Andric   if (CurLambda && CurLambda->CallOperator->getType().isNull())
35915f757f3fSDimitry Andric     return StmtError();
35920b57cec5SDimitry Andric   bool HasDeducedReturnType =
35930b57cec5SDimitry Andric       CurLambda && hasDeducedReturnType(CurLambda->CallOperator);
35940b57cec5SDimitry Andric 
35954824e7fdSDimitry Andric   if (ExprEvalContexts.back().isDiscardedStatementContext() &&
35960b57cec5SDimitry Andric       (HasDeducedReturnType || CurCap->HasImplicitReturnType)) {
35970b57cec5SDimitry Andric     if (RetValExp) {
35980b57cec5SDimitry Andric       ExprResult ER =
35990b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
36000b57cec5SDimitry Andric       if (ER.isInvalid())
36010b57cec5SDimitry Andric         return StmtError();
36020b57cec5SDimitry Andric       RetValExp = ER.get();
36030b57cec5SDimitry Andric     }
36040b57cec5SDimitry Andric     return ReturnStmt::Create(Context, ReturnLoc, RetValExp,
36050b57cec5SDimitry Andric                               /* NRVOCandidate=*/nullptr);
36060b57cec5SDimitry Andric   }
36070b57cec5SDimitry Andric 
36080b57cec5SDimitry Andric   if (HasDeducedReturnType) {
3609e8d8bef9SDimitry Andric     FunctionDecl *FD = CurLambda->CallOperator;
3610e8d8bef9SDimitry Andric     // If we've already decided this lambda is invalid, e.g. because
3611e8d8bef9SDimitry Andric     // we saw a `return` whose expression had an error, don't keep
3612e8d8bef9SDimitry Andric     // trying to deduce its return type.
3613e8d8bef9SDimitry Andric     if (FD->isInvalidDecl())
3614e8d8bef9SDimitry Andric       return StmtError();
36150b57cec5SDimitry Andric     // In C++1y, the return type may involve 'auto'.
36160b57cec5SDimitry Andric     // FIXME: Blocks might have a return type of 'auto' explicitly specified.
36170b57cec5SDimitry Andric     if (CurCap->ReturnType.isNull())
36180b57cec5SDimitry Andric       CurCap->ReturnType = FD->getReturnType();
36190b57cec5SDimitry Andric 
36200b57cec5SDimitry Andric     AutoType *AT = CurCap->ReturnType->getContainedAutoType();
36210b57cec5SDimitry Andric     assert(AT && "lost auto type from lambda return type");
36220b57cec5SDimitry Andric     if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
36230b57cec5SDimitry Andric       FD->setInvalidDecl();
36245ffd83dbSDimitry Andric       // FIXME: preserve the ill-formed return expression.
36250b57cec5SDimitry Andric       return StmtError();
36260b57cec5SDimitry Andric     }
36270b57cec5SDimitry Andric     CurCap->ReturnType = FnRetType = FD->getReturnType();
36280b57cec5SDimitry Andric   } else if (CurCap->HasImplicitReturnType) {
36290b57cec5SDimitry Andric     // For blocks/lambdas with implicit return types, we check each return
36300b57cec5SDimitry Andric     // statement individually, and deduce the common return type when the block
36310b57cec5SDimitry Andric     // or lambda is completed.
36320b57cec5SDimitry Andric     // FIXME: Fold this into the 'auto' codepath above.
36330b57cec5SDimitry Andric     if (RetValExp && !isa<InitListExpr>(RetValExp)) {
36340b57cec5SDimitry Andric       ExprResult Result = DefaultFunctionArrayLvalueConversion(RetValExp);
36350b57cec5SDimitry Andric       if (Result.isInvalid())
36360b57cec5SDimitry Andric         return StmtError();
36370b57cec5SDimitry Andric       RetValExp = Result.get();
36380b57cec5SDimitry Andric 
36390b57cec5SDimitry Andric       // DR1048: even prior to C++14, we should use the 'auto' deduction rules
36400b57cec5SDimitry Andric       // when deducing a return type for a lambda-expression (or by extension
36410b57cec5SDimitry Andric       // for a block). These rules differ from the stated C++11 rules only in
36420b57cec5SDimitry Andric       // that they remove top-level cv-qualifiers.
36430b57cec5SDimitry Andric       if (!CurContext->isDependentContext())
36440b57cec5SDimitry Andric         FnRetType = RetValExp->getType().getUnqualifiedType();
36450b57cec5SDimitry Andric       else
36460b57cec5SDimitry Andric         FnRetType = CurCap->ReturnType = Context.DependentTy;
36470b57cec5SDimitry Andric     } else {
36480b57cec5SDimitry Andric       if (RetValExp) {
36490b57cec5SDimitry Andric         // C++11 [expr.lambda.prim]p4 bans inferring the result from an
36500b57cec5SDimitry Andric         // initializer list, because it is not an expression (even
36510b57cec5SDimitry Andric         // though we represent it as one). We still deduce 'void'.
36520b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_lambda_return_init_list)
36530b57cec5SDimitry Andric           << RetValExp->getSourceRange();
36540b57cec5SDimitry Andric       }
36550b57cec5SDimitry Andric 
36560b57cec5SDimitry Andric       FnRetType = Context.VoidTy;
36570b57cec5SDimitry Andric     }
36580b57cec5SDimitry Andric 
36590b57cec5SDimitry Andric     // Although we'll properly infer the type of the block once it's completed,
36600b57cec5SDimitry Andric     // make sure we provide a return type now for better error recovery.
36610b57cec5SDimitry Andric     if (CurCap->ReturnType.isNull())
36620b57cec5SDimitry Andric       CurCap->ReturnType = FnRetType;
36630b57cec5SDimitry Andric   }
3664fe6060f1SDimitry Andric   const VarDecl *NRVOCandidate = getCopyElisionCandidate(NRInfo, FnRetType);
36650b57cec5SDimitry Andric 
3666a7dea167SDimitry Andric   if (auto *CurBlock = dyn_cast<BlockScopeInfo>(CurCap)) {
3667a7dea167SDimitry Andric     if (CurBlock->FunctionType->castAs<FunctionType>()->getNoReturnAttr()) {
36680b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_noreturn_block_has_return_expr);
36690b57cec5SDimitry Andric       return StmtError();
36700b57cec5SDimitry Andric     }
3671a7dea167SDimitry Andric   } else if (auto *CurRegion = dyn_cast<CapturedRegionScopeInfo>(CurCap)) {
36720b57cec5SDimitry Andric     Diag(ReturnLoc, diag::err_return_in_captured_stmt) << CurRegion->getRegionName();
36730b57cec5SDimitry Andric     return StmtError();
36740b57cec5SDimitry Andric   } else {
36750b57cec5SDimitry Andric     assert(CurLambda && "unknown kind of captured scope");
3676a7dea167SDimitry Andric     if (CurLambda->CallOperator->getType()
3677a7dea167SDimitry Andric             ->castAs<FunctionType>()
36780b57cec5SDimitry Andric             ->getNoReturnAttr()) {
36790b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_noreturn_lambda_has_return_expr);
36800b57cec5SDimitry Andric       return StmtError();
36810b57cec5SDimitry Andric     }
36820b57cec5SDimitry Andric   }
36830b57cec5SDimitry Andric 
36840b57cec5SDimitry Andric   // Otherwise, verify that this result type matches the previous one.  We are
36850b57cec5SDimitry Andric   // pickier with blocks than for normal functions because we don't have GCC
36860b57cec5SDimitry Andric   // compatibility to worry about here.
36870b57cec5SDimitry Andric   if (FnRetType->isDependentType()) {
36880b57cec5SDimitry Andric     // Delay processing for now.  TODO: there are lots of dependent
36890b57cec5SDimitry Andric     // types we can conclusively prove aren't void.
36900b57cec5SDimitry Andric   } else if (FnRetType->isVoidType()) {
36910b57cec5SDimitry Andric     if (RetValExp && !isa<InitListExpr>(RetValExp) &&
36920b57cec5SDimitry Andric         !(getLangOpts().CPlusPlus &&
36930b57cec5SDimitry Andric           (RetValExp->isTypeDependent() ||
36940b57cec5SDimitry Andric            RetValExp->getType()->isVoidType()))) {
36950b57cec5SDimitry Andric       if (!getLangOpts().CPlusPlus &&
36960b57cec5SDimitry Andric           RetValExp->getType()->isVoidType())
36970b57cec5SDimitry Andric         Diag(ReturnLoc, diag::ext_return_has_void_expr) << "literal" << 2;
36980b57cec5SDimitry Andric       else {
36990b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_return_block_has_expr);
37000b57cec5SDimitry Andric         RetValExp = nullptr;
37010b57cec5SDimitry Andric       }
37020b57cec5SDimitry Andric     }
37030b57cec5SDimitry Andric   } else if (!RetValExp) {
37040b57cec5SDimitry Andric     return StmtError(Diag(ReturnLoc, diag::err_block_return_missing_expr));
37050b57cec5SDimitry Andric   } else if (!RetValExp->isTypeDependent()) {
37060b57cec5SDimitry Andric     // we have a non-void block with an expression, continue checking
37070b57cec5SDimitry Andric 
37080b57cec5SDimitry Andric     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
37090b57cec5SDimitry Andric     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
37100b57cec5SDimitry Andric     // function return.
37110b57cec5SDimitry Andric 
37120b57cec5SDimitry Andric     // In C++ the return statement is handled via a copy initialization.
37130b57cec5SDimitry Andric     // the C version of which boils down to CheckSingleAssignmentConstraints.
371428a41182SDimitry Andric     InitializedEntity Entity =
371528a41182SDimitry Andric         InitializedEntity::InitializeResult(ReturnLoc, FnRetType);
3716fe6060f1SDimitry Andric     ExprResult Res = PerformMoveOrCopyInitialization(
3717fe6060f1SDimitry Andric         Entity, NRInfo, RetValExp, SupressSimplerImplicitMoves);
37180b57cec5SDimitry Andric     if (Res.isInvalid()) {
37190b57cec5SDimitry Andric       // FIXME: Cleanup temporaries here, anyway?
37200b57cec5SDimitry Andric       return StmtError();
37210b57cec5SDimitry Andric     }
37220b57cec5SDimitry Andric     RetValExp = Res.get();
37230b57cec5SDimitry Andric     CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc);
37240b57cec5SDimitry Andric   }
37250b57cec5SDimitry Andric 
37260b57cec5SDimitry Andric   if (RetValExp) {
37270b57cec5SDimitry Andric     ExprResult ER =
37280b57cec5SDimitry Andric         ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
37290b57cec5SDimitry Andric     if (ER.isInvalid())
37300b57cec5SDimitry Andric       return StmtError();
37310b57cec5SDimitry Andric     RetValExp = ER.get();
37320b57cec5SDimitry Andric   }
37330b57cec5SDimitry Andric   auto *Result =
37340b57cec5SDimitry Andric       ReturnStmt::Create(Context, ReturnLoc, RetValExp, NRVOCandidate);
37350b57cec5SDimitry Andric 
37360b57cec5SDimitry Andric   // If we need to check for the named return value optimization,
37370b57cec5SDimitry Andric   // or if we need to infer the return type,
37380b57cec5SDimitry Andric   // save the return statement in our scope for later processing.
37390b57cec5SDimitry Andric   if (CurCap->HasImplicitReturnType || NRVOCandidate)
37400b57cec5SDimitry Andric     FunctionScopes.back()->Returns.push_back(Result);
37410b57cec5SDimitry Andric 
37420b57cec5SDimitry Andric   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
37430b57cec5SDimitry Andric     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
37440b57cec5SDimitry Andric 
374506c3fb27SDimitry Andric   if (auto *CurBlock = dyn_cast<BlockScopeInfo>(CurCap);
374606c3fb27SDimitry Andric       CurBlock && CurCap->HasImplicitReturnType && RetValExp &&
374706c3fb27SDimitry Andric       RetValExp->containsErrors())
374806c3fb27SDimitry Andric     CurBlock->TheDecl->setInvalidDecl();
374906c3fb27SDimitry Andric 
37500b57cec5SDimitry Andric   return Result;
37510b57cec5SDimitry Andric }
37520b57cec5SDimitry Andric 
37530b57cec5SDimitry Andric namespace {
37540b57cec5SDimitry Andric /// Marks all typedefs in all local classes in a type referenced.
37550b57cec5SDimitry Andric ///
37560b57cec5SDimitry Andric /// In a function like
37570b57cec5SDimitry Andric /// auto f() {
37580b57cec5SDimitry Andric ///   struct S { typedef int a; };
37590b57cec5SDimitry Andric ///   return S();
37600b57cec5SDimitry Andric /// }
37610b57cec5SDimitry Andric ///
37620b57cec5SDimitry Andric /// the local type escapes and could be referenced in some TUs but not in
37630b57cec5SDimitry Andric /// others. Pretend that all local typedefs are always referenced, to not warn
37640b57cec5SDimitry Andric /// on this. This isn't necessary if f has internal linkage, or the typedef
37650b57cec5SDimitry Andric /// is private.
37660b57cec5SDimitry Andric class LocalTypedefNameReferencer
37670b57cec5SDimitry Andric     : public RecursiveASTVisitor<LocalTypedefNameReferencer> {
37680b57cec5SDimitry Andric public:
37690b57cec5SDimitry Andric   LocalTypedefNameReferencer(Sema &S) : S(S) {}
37700b57cec5SDimitry Andric   bool VisitRecordType(const RecordType *RT);
37710b57cec5SDimitry Andric private:
37720b57cec5SDimitry Andric   Sema &S;
37730b57cec5SDimitry Andric };
37740b57cec5SDimitry Andric bool LocalTypedefNameReferencer::VisitRecordType(const RecordType *RT) {
37750b57cec5SDimitry Andric   auto *R = dyn_cast<CXXRecordDecl>(RT->getDecl());
37760b57cec5SDimitry Andric   if (!R || !R->isLocalClass() || !R->isLocalClass()->isExternallyVisible() ||
37770b57cec5SDimitry Andric       R->isDependentType())
37780b57cec5SDimitry Andric     return true;
37790b57cec5SDimitry Andric   for (auto *TmpD : R->decls())
37800b57cec5SDimitry Andric     if (auto *T = dyn_cast<TypedefNameDecl>(TmpD))
37810b57cec5SDimitry Andric       if (T->getAccess() != AS_private || R->hasFriends())
37820b57cec5SDimitry Andric         S.MarkAnyDeclReferenced(T->getLocation(), T, /*OdrUse=*/false);
37830b57cec5SDimitry Andric   return true;
37840b57cec5SDimitry Andric }
37850b57cec5SDimitry Andric }
37860b57cec5SDimitry Andric 
37870b57cec5SDimitry Andric TypeLoc Sema::getReturnTypeLoc(FunctionDecl *FD) const {
37880b57cec5SDimitry Andric   return FD->getTypeSourceInfo()
37890b57cec5SDimitry Andric       ->getTypeLoc()
37900b57cec5SDimitry Andric       .getAsAdjusted<FunctionProtoTypeLoc>()
37910b57cec5SDimitry Andric       .getReturnLoc();
37920b57cec5SDimitry Andric }
37930b57cec5SDimitry Andric 
37940b57cec5SDimitry Andric /// Deduce the return type for a function from a returned expression, per
37950b57cec5SDimitry Andric /// C++1y [dcl.spec.auto]p6.
37960b57cec5SDimitry Andric bool Sema::DeduceFunctionTypeFromReturnExpr(FunctionDecl *FD,
37970b57cec5SDimitry Andric                                             SourceLocation ReturnLoc,
3798bdd1243dSDimitry Andric                                             Expr *RetExpr, const AutoType *AT) {
379981ad6265SDimitry Andric   // If this is the conversion function for a lambda, we choose to deduce its
38000b57cec5SDimitry Andric   // type from the corresponding call operator, not from the synthesized return
38010b57cec5SDimitry Andric   // statement within it. See Sema::DeduceReturnType.
38020b57cec5SDimitry Andric   if (isLambdaConversionOperator(FD))
38030b57cec5SDimitry Andric     return false;
38040b57cec5SDimitry Andric 
38050b57cec5SDimitry Andric   if (RetExpr && isa<InitListExpr>(RetExpr)) {
38060b57cec5SDimitry Andric     //  If the deduction is for a return statement and the initializer is
38070b57cec5SDimitry Andric     //  a braced-init-list, the program is ill-formed.
38080b57cec5SDimitry Andric     Diag(RetExpr->getExprLoc(),
38090b57cec5SDimitry Andric          getCurLambda() ? diag::err_lambda_return_init_list
38100b57cec5SDimitry Andric                         : diag::err_auto_fn_return_init_list)
38110b57cec5SDimitry Andric         << RetExpr->getSourceRange();
38120b57cec5SDimitry Andric     return true;
38130b57cec5SDimitry Andric   }
38140b57cec5SDimitry Andric 
38150b57cec5SDimitry Andric   if (FD->isDependentContext()) {
38160b57cec5SDimitry Andric     // C++1y [dcl.spec.auto]p12:
38170b57cec5SDimitry Andric     //   Return type deduction [...] occurs when the definition is
38180b57cec5SDimitry Andric     //   instantiated even if the function body contains a return
38190b57cec5SDimitry Andric     //   statement with a non-type-dependent operand.
38200b57cec5SDimitry Andric     assert(AT->isDeduced() && "should have deduced to dependent type");
38210b57cec5SDimitry Andric     return false;
38220b57cec5SDimitry Andric   }
38230b57cec5SDimitry Andric 
3824bdd1243dSDimitry Andric   TypeLoc OrigResultType = getReturnTypeLoc(FD);
3825bdd1243dSDimitry Andric   //  In the case of a return with no operand, the initializer is considered
3826bdd1243dSDimitry Andric   //  to be void().
3827bdd1243dSDimitry Andric   CXXScalarValueInitExpr VoidVal(Context.VoidTy, nullptr, SourceLocation());
3828bdd1243dSDimitry Andric   if (!RetExpr) {
3829bdd1243dSDimitry Andric     // For a function with a deduced result type to return with omitted
3830bdd1243dSDimitry Andric     // expression, the result type as written must be 'auto' or
3831bdd1243dSDimitry Andric     // 'decltype(auto)', possibly cv-qualified or constrained, but not
3832bdd1243dSDimitry Andric     // ref-qualified.
38330b57cec5SDimitry Andric     if (!OrigResultType.getType()->getAs<AutoType>()) {
38340b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_auto_fn_return_void_but_not_auto)
38350b57cec5SDimitry Andric           << OrigResultType.getType();
38360b57cec5SDimitry Andric       return true;
38370b57cec5SDimitry Andric     }
3838bdd1243dSDimitry Andric     RetExpr = &VoidVal;
38390b57cec5SDimitry Andric   }
38400b57cec5SDimitry Andric 
3841bdd1243dSDimitry Andric   QualType Deduced = AT->getDeducedType();
3842bdd1243dSDimitry Andric   {
3843bdd1243dSDimitry Andric     //  Otherwise, [...] deduce a value for U using the rules of template
3844bdd1243dSDimitry Andric     //  argument deduction.
384506c3fb27SDimitry Andric     auto RetExprLoc = RetExpr->getExprLoc();
384606c3fb27SDimitry Andric     TemplateDeductionInfo Info(RetExprLoc);
384706c3fb27SDimitry Andric     SourceLocation TemplateSpecLoc;
384806c3fb27SDimitry Andric     if (RetExpr->getType() == Context.OverloadTy) {
384906c3fb27SDimitry Andric       auto FindResult = OverloadExpr::find(RetExpr);
385006c3fb27SDimitry Andric       if (FindResult.Expression)
385106c3fb27SDimitry Andric         TemplateSpecLoc = FindResult.Expression->getNameLoc();
385206c3fb27SDimitry Andric     }
385306c3fb27SDimitry Andric     TemplateSpecCandidateSet FailedTSC(TemplateSpecLoc);
385406c3fb27SDimitry Andric     TemplateDeductionResult Res = DeduceAutoType(
385506c3fb27SDimitry Andric         OrigResultType, RetExpr, Deduced, Info, /*DependentDeduction=*/false,
385606c3fb27SDimitry Andric         /*IgnoreConstraints=*/false, &FailedTSC);
3857bdd1243dSDimitry Andric     if (Res != TDK_Success && FD->isInvalidDecl())
3858bdd1243dSDimitry Andric       return true;
3859bdd1243dSDimitry Andric     switch (Res) {
3860bdd1243dSDimitry Andric     case TDK_Success:
3861bdd1243dSDimitry Andric       break;
3862bdd1243dSDimitry Andric     case TDK_AlreadyDiagnosed:
3863bdd1243dSDimitry Andric       return true;
3864bdd1243dSDimitry Andric     case TDK_Inconsistent: {
3865bdd1243dSDimitry Andric       //  If a function with a declared return type that contains a placeholder
3866bdd1243dSDimitry Andric       //  type has multiple return statements, the return type is deduced for
3867bdd1243dSDimitry Andric       //  each return statement. [...] if the type deduced is not the same in
3868bdd1243dSDimitry Andric       //  each deduction, the program is ill-formed.
3869bdd1243dSDimitry Andric       const LambdaScopeInfo *LambdaSI = getCurLambda();
3870bdd1243dSDimitry Andric       if (LambdaSI && LambdaSI->HasImplicitReturnType)
3871bdd1243dSDimitry Andric         Diag(ReturnLoc, diag::err_typecheck_missing_return_type_incompatible)
3872bdd1243dSDimitry Andric             << Info.SecondArg << Info.FirstArg << true /*IsLambda*/;
3873bdd1243dSDimitry Andric       else
3874bdd1243dSDimitry Andric         Diag(ReturnLoc, diag::err_auto_fn_different_deductions)
3875bdd1243dSDimitry Andric             << (AT->isDecltypeAuto() ? 1 : 0) << Info.SecondArg
3876bdd1243dSDimitry Andric             << Info.FirstArg;
3877bdd1243dSDimitry Andric       return true;
3878bdd1243dSDimitry Andric     }
3879bdd1243dSDimitry Andric     default:
3880bdd1243dSDimitry Andric       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_deduction_failure)
3881bdd1243dSDimitry Andric           << OrigResultType.getType() << RetExpr->getType();
388206c3fb27SDimitry Andric       FailedTSC.NoteCandidates(*this, RetExprLoc);
3883bdd1243dSDimitry Andric       return true;
3884bdd1243dSDimitry Andric     }
3885bdd1243dSDimitry Andric   }
3886bdd1243dSDimitry Andric 
3887bdd1243dSDimitry Andric   // If a local type is part of the returned type, mark its fields as
3888bdd1243dSDimitry Andric   // referenced.
3889bdd1243dSDimitry Andric   LocalTypedefNameReferencer(*this).TraverseType(RetExpr->getType());
3890bdd1243dSDimitry Andric 
3891a7dea167SDimitry Andric   // CUDA: Kernel function must have 'void' return type.
3892bdd1243dSDimitry Andric   if (getLangOpts().CUDA && FD->hasAttr<CUDAGlobalAttr>() &&
3893bdd1243dSDimitry Andric       !Deduced->isVoidType()) {
3894a7dea167SDimitry Andric     Diag(FD->getLocation(), diag::err_kern_type_not_void_return)
3895a7dea167SDimitry Andric         << FD->getType() << FD->getSourceRange();
3896a7dea167SDimitry Andric     return true;
3897a7dea167SDimitry Andric   }
3898a7dea167SDimitry Andric 
3899bdd1243dSDimitry Andric   if (!FD->isInvalidDecl() && AT->getDeducedType() != Deduced)
39000b57cec5SDimitry Andric     // Update all declarations of the function to have the deduced return type.
39010b57cec5SDimitry Andric     Context.adjustDeducedFunctionResultType(FD, Deduced);
39020b57cec5SDimitry Andric 
39030b57cec5SDimitry Andric   return false;
39040b57cec5SDimitry Andric }
39050b57cec5SDimitry Andric 
39060b57cec5SDimitry Andric StmtResult
39070b57cec5SDimitry Andric Sema::ActOnReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp,
39080b57cec5SDimitry Andric                       Scope *CurScope) {
39090b57cec5SDimitry Andric   // Correct typos, in case the containing function returns 'auto' and
39100b57cec5SDimitry Andric   // RetValExp should determine the deduced type.
3911e8d8bef9SDimitry Andric   ExprResult RetVal = CorrectDelayedTyposInExpr(
3912e8d8bef9SDimitry Andric       RetValExp, nullptr, /*RecoverUncorrectedTypos=*/true);
39130b57cec5SDimitry Andric   if (RetVal.isInvalid())
39140b57cec5SDimitry Andric     return StmtError();
391504eeddc0SDimitry Andric   StmtResult R =
391604eeddc0SDimitry Andric       BuildReturnStmt(ReturnLoc, RetVal.get(), /*AllowRecovery=*/true);
39174824e7fdSDimitry Andric   if (R.isInvalid() || ExprEvalContexts.back().isDiscardedStatementContext())
39180b57cec5SDimitry Andric     return R;
39190b57cec5SDimitry Andric 
3920972a253aSDimitry Andric   VarDecl *VD =
3921972a253aSDimitry Andric       const_cast<VarDecl *>(cast<ReturnStmt>(R.get())->getNRVOCandidate());
3922972a253aSDimitry Andric 
3923972a253aSDimitry Andric   CurScope->updateNRVOCandidate(VD);
39240b57cec5SDimitry Andric 
39250b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, ReturnLoc, *CurScope->getFnParent());
39260b57cec5SDimitry Andric 
39270b57cec5SDimitry Andric   return R;
39280b57cec5SDimitry Andric }
39290b57cec5SDimitry Andric 
3930fe6060f1SDimitry Andric static bool CheckSimplerImplicitMovesMSVCWorkaround(const Sema &S,
3931fe6060f1SDimitry Andric                                                     const Expr *E) {
393206c3fb27SDimitry Andric   if (!E || !S.getLangOpts().CPlusPlus23 || !S.getLangOpts().MSVCCompat)
3933fe6060f1SDimitry Andric     return false;
3934fe6060f1SDimitry Andric   const Decl *D = E->getReferencedDeclOfCallee();
3935fe6060f1SDimitry Andric   if (!D || !S.SourceMgr.isInSystemHeader(D->getLocation()))
3936fe6060f1SDimitry Andric     return false;
3937fe6060f1SDimitry Andric   for (const DeclContext *DC = D->getDeclContext(); DC; DC = DC->getParent()) {
3938fe6060f1SDimitry Andric     if (DC->isStdNamespace())
3939fe6060f1SDimitry Andric       return true;
3940fe6060f1SDimitry Andric   }
3941fe6060f1SDimitry Andric   return false;
3942fe6060f1SDimitry Andric }
3943fe6060f1SDimitry Andric 
394404eeddc0SDimitry Andric StmtResult Sema::BuildReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp,
394504eeddc0SDimitry Andric                                  bool AllowRecovery) {
39460b57cec5SDimitry Andric   // Check for unexpanded parameter packs.
39470b57cec5SDimitry Andric   if (RetValExp && DiagnoseUnexpandedParameterPack(RetValExp))
39480b57cec5SDimitry Andric     return StmtError();
39490b57cec5SDimitry Andric 
3950349cc55cSDimitry Andric   // HACK: We suppress simpler implicit move here in msvc compatibility mode
3951fe6060f1SDimitry Andric   // just as a temporary work around, as the MSVC STL has issues with
3952fe6060f1SDimitry Andric   // this change.
3953fe6060f1SDimitry Andric   bool SupressSimplerImplicitMoves =
3954fe6060f1SDimitry Andric       CheckSimplerImplicitMovesMSVCWorkaround(*this, RetValExp);
3955fe6060f1SDimitry Andric   NamedReturnInfo NRInfo = getNamedReturnInfo(
3956fe6060f1SDimitry Andric       RetValExp, SupressSimplerImplicitMoves ? SimplerImplicitMoveMode::ForceOff
3957fe6060f1SDimitry Andric                                              : SimplerImplicitMoveMode::Normal);
3958fe6060f1SDimitry Andric 
39590b57cec5SDimitry Andric   if (isa<CapturingScopeInfo>(getCurFunction()))
3960fe6060f1SDimitry Andric     return ActOnCapScopeReturnStmt(ReturnLoc, RetValExp, NRInfo,
3961fe6060f1SDimitry Andric                                    SupressSimplerImplicitMoves);
39620b57cec5SDimitry Andric 
39630b57cec5SDimitry Andric   QualType FnRetType;
39640b57cec5SDimitry Andric   QualType RelatedRetType;
39650b57cec5SDimitry Andric   const AttrVec *Attrs = nullptr;
39660b57cec5SDimitry Andric   bool isObjCMethod = false;
39670b57cec5SDimitry Andric 
39680b57cec5SDimitry Andric   if (const FunctionDecl *FD = getCurFunctionDecl()) {
39690b57cec5SDimitry Andric     FnRetType = FD->getReturnType();
39700b57cec5SDimitry Andric     if (FD->hasAttrs())
39710b57cec5SDimitry Andric       Attrs = &FD->getAttrs();
39720b57cec5SDimitry Andric     if (FD->isNoReturn())
3973e8d8bef9SDimitry Andric       Diag(ReturnLoc, diag::warn_noreturn_function_has_return_expr) << FD;
39740b57cec5SDimitry Andric     if (FD->isMain() && RetValExp)
39750b57cec5SDimitry Andric       if (isa<CXXBoolLiteralExpr>(RetValExp))
39760b57cec5SDimitry Andric         Diag(ReturnLoc, diag::warn_main_returns_bool_literal)
39770b57cec5SDimitry Andric             << RetValExp->getSourceRange();
39785ffd83dbSDimitry Andric     if (FD->hasAttr<CmseNSEntryAttr>() && RetValExp) {
39795ffd83dbSDimitry Andric       if (const auto *RT = dyn_cast<RecordType>(FnRetType.getCanonicalType())) {
39805ffd83dbSDimitry Andric         if (RT->getDecl()->isOrContainsUnion())
39815ffd83dbSDimitry Andric           Diag(RetValExp->getBeginLoc(), diag::warn_cmse_nonsecure_union) << 1;
39825ffd83dbSDimitry Andric       }
39835ffd83dbSDimitry Andric     }
39840b57cec5SDimitry Andric   } else if (ObjCMethodDecl *MD = getCurMethodDecl()) {
39850b57cec5SDimitry Andric     FnRetType = MD->getReturnType();
39860b57cec5SDimitry Andric     isObjCMethod = true;
39870b57cec5SDimitry Andric     if (MD->hasAttrs())
39880b57cec5SDimitry Andric       Attrs = &MD->getAttrs();
39890b57cec5SDimitry Andric     if (MD->hasRelatedResultType() && MD->getClassInterface()) {
39900b57cec5SDimitry Andric       // In the implementation of a method with a related return type, the
39910b57cec5SDimitry Andric       // type used to type-check the validity of return statements within the
39920b57cec5SDimitry Andric       // method body is a pointer to the type of the class being implemented.
39930b57cec5SDimitry Andric       RelatedRetType = Context.getObjCInterfaceType(MD->getClassInterface());
39940b57cec5SDimitry Andric       RelatedRetType = Context.getObjCObjectPointerType(RelatedRetType);
39950b57cec5SDimitry Andric     }
39960b57cec5SDimitry Andric   } else // If we don't have a function/method context, bail.
39970b57cec5SDimitry Andric     return StmtError();
39980b57cec5SDimitry Andric 
399906c3fb27SDimitry Andric   if (RetValExp) {
400006c3fb27SDimitry Andric     const auto *ATy = dyn_cast<ArrayType>(RetValExp->getType());
400106c3fb27SDimitry Andric     if (ATy && ATy->getElementType().isWebAssemblyReferenceType()) {
400206c3fb27SDimitry Andric       Diag(ReturnLoc, diag::err_wasm_table_art) << 1;
400306c3fb27SDimitry Andric       return StmtError();
400406c3fb27SDimitry Andric     }
400506c3fb27SDimitry Andric   }
400606c3fb27SDimitry Andric 
40070b57cec5SDimitry Andric   // C++1z: discarded return statements are not considered when deducing a
40080b57cec5SDimitry Andric   // return type.
40094824e7fdSDimitry Andric   if (ExprEvalContexts.back().isDiscardedStatementContext() &&
40100b57cec5SDimitry Andric       FnRetType->getContainedAutoType()) {
40110b57cec5SDimitry Andric     if (RetValExp) {
40120b57cec5SDimitry Andric       ExprResult ER =
40130b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
40140b57cec5SDimitry Andric       if (ER.isInvalid())
40150b57cec5SDimitry Andric         return StmtError();
40160b57cec5SDimitry Andric       RetValExp = ER.get();
40170b57cec5SDimitry Andric     }
40180b57cec5SDimitry Andric     return ReturnStmt::Create(Context, ReturnLoc, RetValExp,
40190b57cec5SDimitry Andric                               /* NRVOCandidate=*/nullptr);
40200b57cec5SDimitry Andric   }
40210b57cec5SDimitry Andric 
40220b57cec5SDimitry Andric   // FIXME: Add a flag to the ScopeInfo to indicate whether we're performing
40230b57cec5SDimitry Andric   // deduction.
40240b57cec5SDimitry Andric   if (getLangOpts().CPlusPlus14) {
40250b57cec5SDimitry Andric     if (AutoType *AT = FnRetType->getContainedAutoType()) {
40260b57cec5SDimitry Andric       FunctionDecl *FD = cast<FunctionDecl>(CurContext);
4027e8d8bef9SDimitry Andric       // If we've already decided this function is invalid, e.g. because
4028e8d8bef9SDimitry Andric       // we saw a `return` whose expression had an error, don't keep
4029e8d8bef9SDimitry Andric       // trying to deduce its return type.
403004eeddc0SDimitry Andric       // (Some return values may be needlessly wrapped in RecoveryExpr).
403104eeddc0SDimitry Andric       if (FD->isInvalidDecl() ||
403204eeddc0SDimitry Andric           DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
40330b57cec5SDimitry Andric         FD->setInvalidDecl();
403404eeddc0SDimitry Andric         if (!AllowRecovery)
40350b57cec5SDimitry Andric           return StmtError();
403604eeddc0SDimitry Andric         // The deduction failure is diagnosed and marked, try to recover.
403704eeddc0SDimitry Andric         if (RetValExp) {
403804eeddc0SDimitry Andric           // Wrap return value with a recovery expression of the previous type.
403904eeddc0SDimitry Andric           // If no deduction yet, use DependentTy.
404004eeddc0SDimitry Andric           auto Recovery = CreateRecoveryExpr(
404104eeddc0SDimitry Andric               RetValExp->getBeginLoc(), RetValExp->getEndLoc(), RetValExp,
404204eeddc0SDimitry Andric               AT->isDeduced() ? FnRetType : QualType());
404304eeddc0SDimitry Andric           if (Recovery.isInvalid())
404404eeddc0SDimitry Andric             return StmtError();
404504eeddc0SDimitry Andric           RetValExp = Recovery.get();
404604eeddc0SDimitry Andric         } else {
404704eeddc0SDimitry Andric           // Nothing to do: a ReturnStmt with no value is fine recovery.
404804eeddc0SDimitry Andric         }
40490b57cec5SDimitry Andric       } else {
40500b57cec5SDimitry Andric         FnRetType = FD->getReturnType();
40510b57cec5SDimitry Andric       }
40520b57cec5SDimitry Andric     }
40530b57cec5SDimitry Andric   }
4054fe6060f1SDimitry Andric   const VarDecl *NRVOCandidate = getCopyElisionCandidate(NRInfo, FnRetType);
40550b57cec5SDimitry Andric 
40560b57cec5SDimitry Andric   bool HasDependentReturnType = FnRetType->isDependentType();
40570b57cec5SDimitry Andric 
40580b57cec5SDimitry Andric   ReturnStmt *Result = nullptr;
40590b57cec5SDimitry Andric   if (FnRetType->isVoidType()) {
40600b57cec5SDimitry Andric     if (RetValExp) {
406104eeddc0SDimitry Andric       if (auto *ILE = dyn_cast<InitListExpr>(RetValExp)) {
40620b57cec5SDimitry Andric         // We simply never allow init lists as the return value of void
40630b57cec5SDimitry Andric         // functions. This is compatible because this was never allowed before,
40640b57cec5SDimitry Andric         // so there's no legacy code to deal with.
40650b57cec5SDimitry Andric         NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
40660b57cec5SDimitry Andric         int FunctionKind = 0;
40670b57cec5SDimitry Andric         if (isa<ObjCMethodDecl>(CurDecl))
40680b57cec5SDimitry Andric           FunctionKind = 1;
40690b57cec5SDimitry Andric         else if (isa<CXXConstructorDecl>(CurDecl))
40700b57cec5SDimitry Andric           FunctionKind = 2;
40710b57cec5SDimitry Andric         else if (isa<CXXDestructorDecl>(CurDecl))
40720b57cec5SDimitry Andric           FunctionKind = 3;
40730b57cec5SDimitry Andric 
40740b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_return_init_list)
4075e8d8bef9SDimitry Andric             << CurDecl << FunctionKind << RetValExp->getSourceRange();
40760b57cec5SDimitry Andric 
407704eeddc0SDimitry Andric         // Preserve the initializers in the AST.
407804eeddc0SDimitry Andric         RetValExp = AllowRecovery
407904eeddc0SDimitry Andric                         ? CreateRecoveryExpr(ILE->getLBraceLoc(),
408004eeddc0SDimitry Andric                                              ILE->getRBraceLoc(), ILE->inits())
408104eeddc0SDimitry Andric                               .get()
408204eeddc0SDimitry Andric                         : nullptr;
40830b57cec5SDimitry Andric       } else if (!RetValExp->isTypeDependent()) {
40840b57cec5SDimitry Andric         // C99 6.8.6.4p1 (ext_ since GCC warns)
40850b57cec5SDimitry Andric         unsigned D = diag::ext_return_has_expr;
40860b57cec5SDimitry Andric         if (RetValExp->getType()->isVoidType()) {
40870b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
40880b57cec5SDimitry Andric           if (isa<CXXConstructorDecl>(CurDecl) ||
40890b57cec5SDimitry Andric               isa<CXXDestructorDecl>(CurDecl))
40900b57cec5SDimitry Andric             D = diag::err_ctor_dtor_returns_void;
40910b57cec5SDimitry Andric           else
40920b57cec5SDimitry Andric             D = diag::ext_return_has_void_expr;
40930b57cec5SDimitry Andric         }
40940b57cec5SDimitry Andric         else {
40950b57cec5SDimitry Andric           ExprResult Result = RetValExp;
40960b57cec5SDimitry Andric           Result = IgnoredValueConversions(Result.get());
40970b57cec5SDimitry Andric           if (Result.isInvalid())
40980b57cec5SDimitry Andric             return StmtError();
40990b57cec5SDimitry Andric           RetValExp = Result.get();
41000b57cec5SDimitry Andric           RetValExp = ImpCastExprToType(RetValExp,
41010b57cec5SDimitry Andric                                         Context.VoidTy, CK_ToVoid).get();
41020b57cec5SDimitry Andric         }
41030b57cec5SDimitry Andric         // return of void in constructor/destructor is illegal in C++.
41040b57cec5SDimitry Andric         if (D == diag::err_ctor_dtor_returns_void) {
41050b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
4106e8d8bef9SDimitry Andric           Diag(ReturnLoc, D) << CurDecl << isa<CXXDestructorDecl>(CurDecl)
41070b57cec5SDimitry Andric                              << RetValExp->getSourceRange();
41080b57cec5SDimitry Andric         }
41090b57cec5SDimitry Andric         // return (some void expression); is legal in C++.
41100b57cec5SDimitry Andric         else if (D != diag::ext_return_has_void_expr ||
41110b57cec5SDimitry Andric                  !getLangOpts().CPlusPlus) {
41120b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
41130b57cec5SDimitry Andric 
41140b57cec5SDimitry Andric           int FunctionKind = 0;
41150b57cec5SDimitry Andric           if (isa<ObjCMethodDecl>(CurDecl))
41160b57cec5SDimitry Andric             FunctionKind = 1;
41170b57cec5SDimitry Andric           else if (isa<CXXConstructorDecl>(CurDecl))
41180b57cec5SDimitry Andric             FunctionKind = 2;
41190b57cec5SDimitry Andric           else if (isa<CXXDestructorDecl>(CurDecl))
41200b57cec5SDimitry Andric             FunctionKind = 3;
41210b57cec5SDimitry Andric 
41220b57cec5SDimitry Andric           Diag(ReturnLoc, D)
4123e8d8bef9SDimitry Andric               << CurDecl << FunctionKind << RetValExp->getSourceRange();
41240b57cec5SDimitry Andric         }
41250b57cec5SDimitry Andric       }
41260b57cec5SDimitry Andric 
41270b57cec5SDimitry Andric       if (RetValExp) {
41280b57cec5SDimitry Andric         ExprResult ER =
41290b57cec5SDimitry Andric             ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
41300b57cec5SDimitry Andric         if (ER.isInvalid())
41310b57cec5SDimitry Andric           return StmtError();
41320b57cec5SDimitry Andric         RetValExp = ER.get();
41330b57cec5SDimitry Andric       }
41340b57cec5SDimitry Andric     }
41350b57cec5SDimitry Andric 
41360b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, RetValExp,
41370b57cec5SDimitry Andric                                 /* NRVOCandidate=*/nullptr);
41380b57cec5SDimitry Andric   } else if (!RetValExp && !HasDependentReturnType) {
41390b57cec5SDimitry Andric     FunctionDecl *FD = getCurFunctionDecl();
41400b57cec5SDimitry Andric 
414181ad6265SDimitry Andric     if ((FD && FD->isInvalidDecl()) || FnRetType->containsErrors()) {
414281ad6265SDimitry Andric       // The intended return type might have been "void", so don't warn.
414381ad6265SDimitry Andric     } else if (getLangOpts().CPlusPlus11 && FD && FD->isConstexpr()) {
41440b57cec5SDimitry Andric       // C++11 [stmt.return]p2
4145e8d8bef9SDimitry Andric       Diag(ReturnLoc, diag::err_constexpr_return_missing_expr)
4146e8d8bef9SDimitry Andric           << FD << FD->isConsteval();
41470b57cec5SDimitry Andric       FD->setInvalidDecl();
41480b57cec5SDimitry Andric     } else {
4149e8d8bef9SDimitry Andric       // C99 6.8.6.4p1 (ext_ since GCC warns)
41500b57cec5SDimitry Andric       // C90 6.6.6.4p4
4151e8d8bef9SDimitry Andric       unsigned DiagID = getLangOpts().C99 ? diag::ext_return_missing_expr
4152e8d8bef9SDimitry Andric                                           : diag::warn_return_missing_expr;
4153e8d8bef9SDimitry Andric       // Note that at this point one of getCurFunctionDecl() or
4154e8d8bef9SDimitry Andric       // getCurMethodDecl() must be non-null (see above).
4155e8d8bef9SDimitry Andric       assert((getCurFunctionDecl() || getCurMethodDecl()) &&
4156e8d8bef9SDimitry Andric              "Not in a FunctionDecl or ObjCMethodDecl?");
4157e8d8bef9SDimitry Andric       bool IsMethod = FD == nullptr;
4158e8d8bef9SDimitry Andric       const NamedDecl *ND =
4159e8d8bef9SDimitry Andric           IsMethod ? cast<NamedDecl>(getCurMethodDecl()) : cast<NamedDecl>(FD);
4160e8d8bef9SDimitry Andric       Diag(ReturnLoc, DiagID) << ND << IsMethod;
41610b57cec5SDimitry Andric     }
41620b57cec5SDimitry Andric 
41630b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, /* RetExpr=*/nullptr,
41640b57cec5SDimitry Andric                                 /* NRVOCandidate=*/nullptr);
41650b57cec5SDimitry Andric   } else {
41660b57cec5SDimitry Andric     assert(RetValExp || HasDependentReturnType);
41670b57cec5SDimitry Andric     QualType RetType = RelatedRetType.isNull() ? FnRetType : RelatedRetType;
41680b57cec5SDimitry Andric 
41690b57cec5SDimitry Andric     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
41700b57cec5SDimitry Andric     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
41710b57cec5SDimitry Andric     // function return.
41720b57cec5SDimitry Andric 
41730b57cec5SDimitry Andric     // In C++ the return statement is handled via a copy initialization,
41740b57cec5SDimitry Andric     // the C version of which boils down to CheckSingleAssignmentConstraints.
41750b57cec5SDimitry Andric     if (!HasDependentReturnType && !RetValExp->isTypeDependent()) {
41760b57cec5SDimitry Andric       // we have a non-void function with an expression, continue checking
417728a41182SDimitry Andric       InitializedEntity Entity =
417828a41182SDimitry Andric           InitializedEntity::InitializeResult(ReturnLoc, RetType);
4179fe6060f1SDimitry Andric       ExprResult Res = PerformMoveOrCopyInitialization(
4180fe6060f1SDimitry Andric           Entity, NRInfo, RetValExp, SupressSimplerImplicitMoves);
418104eeddc0SDimitry Andric       if (Res.isInvalid() && AllowRecovery)
418204eeddc0SDimitry Andric         Res = CreateRecoveryExpr(RetValExp->getBeginLoc(),
418304eeddc0SDimitry Andric                                  RetValExp->getEndLoc(), RetValExp, RetType);
41840b57cec5SDimitry Andric       if (Res.isInvalid()) {
41850b57cec5SDimitry Andric         // FIXME: Clean up temporaries here anyway?
41860b57cec5SDimitry Andric         return StmtError();
41870b57cec5SDimitry Andric       }
41880b57cec5SDimitry Andric       RetValExp = Res.getAs<Expr>();
41890b57cec5SDimitry Andric 
41900b57cec5SDimitry Andric       // If we have a related result type, we need to implicitly
41910b57cec5SDimitry Andric       // convert back to the formal result type.  We can't pretend to
41920b57cec5SDimitry Andric       // initialize the result again --- we might end double-retaining
41930b57cec5SDimitry Andric       // --- so instead we initialize a notional temporary.
41940b57cec5SDimitry Andric       if (!RelatedRetType.isNull()) {
41950b57cec5SDimitry Andric         Entity = InitializedEntity::InitializeRelatedResult(getCurMethodDecl(),
41960b57cec5SDimitry Andric                                                             FnRetType);
41970b57cec5SDimitry Andric         Res = PerformCopyInitialization(Entity, ReturnLoc, RetValExp);
41980b57cec5SDimitry Andric         if (Res.isInvalid()) {
41990b57cec5SDimitry Andric           // FIXME: Clean up temporaries here anyway?
42000b57cec5SDimitry Andric           return StmtError();
42010b57cec5SDimitry Andric         }
42020b57cec5SDimitry Andric         RetValExp = Res.getAs<Expr>();
42030b57cec5SDimitry Andric       }
42040b57cec5SDimitry Andric 
42050b57cec5SDimitry Andric       CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc, isObjCMethod, Attrs,
42060b57cec5SDimitry Andric                          getCurFunctionDecl());
42070b57cec5SDimitry Andric     }
42080b57cec5SDimitry Andric 
42090b57cec5SDimitry Andric     if (RetValExp) {
42100b57cec5SDimitry Andric       ExprResult ER =
42110b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
42120b57cec5SDimitry Andric       if (ER.isInvalid())
42130b57cec5SDimitry Andric         return StmtError();
42140b57cec5SDimitry Andric       RetValExp = ER.get();
42150b57cec5SDimitry Andric     }
42160b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, RetValExp, NRVOCandidate);
42170b57cec5SDimitry Andric   }
42180b57cec5SDimitry Andric 
42190b57cec5SDimitry Andric   // If we need to check for the named return value optimization, save the
42200b57cec5SDimitry Andric   // return statement in our scope for later processing.
42210b57cec5SDimitry Andric   if (Result->getNRVOCandidate())
42220b57cec5SDimitry Andric     FunctionScopes.back()->Returns.push_back(Result);
42230b57cec5SDimitry Andric 
42240b57cec5SDimitry Andric   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
42250b57cec5SDimitry Andric     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
42260b57cec5SDimitry Andric 
42270b57cec5SDimitry Andric   return Result;
42280b57cec5SDimitry Andric }
42290b57cec5SDimitry Andric 
42300b57cec5SDimitry Andric StmtResult
42310b57cec5SDimitry Andric Sema::ActOnObjCAtCatchStmt(SourceLocation AtLoc,
42320b57cec5SDimitry Andric                            SourceLocation RParen, Decl *Parm,
42330b57cec5SDimitry Andric                            Stmt *Body) {
42340b57cec5SDimitry Andric   VarDecl *Var = cast_or_null<VarDecl>(Parm);
42350b57cec5SDimitry Andric   if (Var && Var->isInvalidDecl())
42360b57cec5SDimitry Andric     return StmtError();
42370b57cec5SDimitry Andric 
42380b57cec5SDimitry Andric   return new (Context) ObjCAtCatchStmt(AtLoc, RParen, Var, Body);
42390b57cec5SDimitry Andric }
42400b57cec5SDimitry Andric 
42410b57cec5SDimitry Andric StmtResult
42420b57cec5SDimitry Andric Sema::ActOnObjCAtFinallyStmt(SourceLocation AtLoc, Stmt *Body) {
42430b57cec5SDimitry Andric   return new (Context) ObjCAtFinallyStmt(AtLoc, Body);
42440b57cec5SDimitry Andric }
42450b57cec5SDimitry Andric 
42460b57cec5SDimitry Andric StmtResult
42470b57cec5SDimitry Andric Sema::ActOnObjCAtTryStmt(SourceLocation AtLoc, Stmt *Try,
42480b57cec5SDimitry Andric                          MultiStmtArg CatchStmts, Stmt *Finally) {
42490b57cec5SDimitry Andric   if (!getLangOpts().ObjCExceptions)
42500b57cec5SDimitry Andric     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@try";
42510b57cec5SDimitry Andric 
4252349cc55cSDimitry Andric   // Objective-C try is incompatible with SEH __try.
4253349cc55cSDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
4254349cc55cSDimitry Andric   if (FSI->FirstSEHTryLoc.isValid()) {
4255349cc55cSDimitry Andric     Diag(AtLoc, diag::err_mixing_cxx_try_seh_try) << 1;
4256349cc55cSDimitry Andric     Diag(FSI->FirstSEHTryLoc, diag::note_conflicting_try_here) << "'__try'";
4257349cc55cSDimitry Andric   }
4258349cc55cSDimitry Andric 
4259349cc55cSDimitry Andric   FSI->setHasObjCTry(AtLoc);
42600b57cec5SDimitry Andric   unsigned NumCatchStmts = CatchStmts.size();
42610b57cec5SDimitry Andric   return ObjCAtTryStmt::Create(Context, AtLoc, Try, CatchStmts.data(),
42620b57cec5SDimitry Andric                                NumCatchStmts, Finally);
42630b57cec5SDimitry Andric }
42640b57cec5SDimitry Andric 
42650b57cec5SDimitry Andric StmtResult Sema::BuildObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw) {
42660b57cec5SDimitry Andric   if (Throw) {
42670b57cec5SDimitry Andric     ExprResult Result = DefaultLvalueConversion(Throw);
42680b57cec5SDimitry Andric     if (Result.isInvalid())
42690b57cec5SDimitry Andric       return StmtError();
42700b57cec5SDimitry Andric 
42710b57cec5SDimitry Andric     Result = ActOnFinishFullExpr(Result.get(), /*DiscardedValue*/ false);
42720b57cec5SDimitry Andric     if (Result.isInvalid())
42730b57cec5SDimitry Andric       return StmtError();
42740b57cec5SDimitry Andric     Throw = Result.get();
42750b57cec5SDimitry Andric 
42760b57cec5SDimitry Andric     QualType ThrowType = Throw->getType();
42770b57cec5SDimitry Andric     // Make sure the expression type is an ObjC pointer or "void *".
42780b57cec5SDimitry Andric     if (!ThrowType->isDependentType() &&
42790b57cec5SDimitry Andric         !ThrowType->isObjCObjectPointerType()) {
42800b57cec5SDimitry Andric       const PointerType *PT = ThrowType->getAs<PointerType>();
42810b57cec5SDimitry Andric       if (!PT || !PT->getPointeeType()->isVoidType())
42820b57cec5SDimitry Andric         return StmtError(Diag(AtLoc, diag::err_objc_throw_expects_object)
42830b57cec5SDimitry Andric                          << Throw->getType() << Throw->getSourceRange());
42840b57cec5SDimitry Andric     }
42850b57cec5SDimitry Andric   }
42860b57cec5SDimitry Andric 
42870b57cec5SDimitry Andric   return new (Context) ObjCAtThrowStmt(AtLoc, Throw);
42880b57cec5SDimitry Andric }
42890b57cec5SDimitry Andric 
42900b57cec5SDimitry Andric StmtResult
42910b57cec5SDimitry Andric Sema::ActOnObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw,
42920b57cec5SDimitry Andric                            Scope *CurScope) {
42930b57cec5SDimitry Andric   if (!getLangOpts().ObjCExceptions)
42940b57cec5SDimitry Andric     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@throw";
42950b57cec5SDimitry Andric 
42960b57cec5SDimitry Andric   if (!Throw) {
42970b57cec5SDimitry Andric     // @throw without an expression designates a rethrow (which must occur
42980b57cec5SDimitry Andric     // in the context of an @catch clause).
42990b57cec5SDimitry Andric     Scope *AtCatchParent = CurScope;
43000b57cec5SDimitry Andric     while (AtCatchParent && !AtCatchParent->isAtCatchScope())
43010b57cec5SDimitry Andric       AtCatchParent = AtCatchParent->getParent();
43020b57cec5SDimitry Andric     if (!AtCatchParent)
43030b57cec5SDimitry Andric       return StmtError(Diag(AtLoc, diag::err_rethrow_used_outside_catch));
43040b57cec5SDimitry Andric   }
43050b57cec5SDimitry Andric   return BuildObjCAtThrowStmt(AtLoc, Throw);
43060b57cec5SDimitry Andric }
43070b57cec5SDimitry Andric 
43080b57cec5SDimitry Andric ExprResult
43090b57cec5SDimitry Andric Sema::ActOnObjCAtSynchronizedOperand(SourceLocation atLoc, Expr *operand) {
43100b57cec5SDimitry Andric   ExprResult result = DefaultLvalueConversion(operand);
43110b57cec5SDimitry Andric   if (result.isInvalid())
43120b57cec5SDimitry Andric     return ExprError();
43130b57cec5SDimitry Andric   operand = result.get();
43140b57cec5SDimitry Andric 
43150b57cec5SDimitry Andric   // Make sure the expression type is an ObjC pointer or "void *".
43160b57cec5SDimitry Andric   QualType type = operand->getType();
43170b57cec5SDimitry Andric   if (!type->isDependentType() &&
43180b57cec5SDimitry Andric       !type->isObjCObjectPointerType()) {
43190b57cec5SDimitry Andric     const PointerType *pointerType = type->getAs<PointerType>();
43200b57cec5SDimitry Andric     if (!pointerType || !pointerType->getPointeeType()->isVoidType()) {
43210b57cec5SDimitry Andric       if (getLangOpts().CPlusPlus) {
43220b57cec5SDimitry Andric         if (RequireCompleteType(atLoc, type,
43230b57cec5SDimitry Andric                                 diag::err_incomplete_receiver_type))
43240b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
43250b57cec5SDimitry Andric                    << type << operand->getSourceRange();
43260b57cec5SDimitry Andric 
43270b57cec5SDimitry Andric         ExprResult result = PerformContextuallyConvertToObjCPointer(operand);
43280b57cec5SDimitry Andric         if (result.isInvalid())
43290b57cec5SDimitry Andric           return ExprError();
43300b57cec5SDimitry Andric         if (!result.isUsable())
43310b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
43320b57cec5SDimitry Andric                    << type << operand->getSourceRange();
43330b57cec5SDimitry Andric 
43340b57cec5SDimitry Andric         operand = result.get();
43350b57cec5SDimitry Andric       } else {
43360b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
43370b57cec5SDimitry Andric                    << type << operand->getSourceRange();
43380b57cec5SDimitry Andric       }
43390b57cec5SDimitry Andric     }
43400b57cec5SDimitry Andric   }
43410b57cec5SDimitry Andric 
43420b57cec5SDimitry Andric   // The operand to @synchronized is a full-expression.
43430b57cec5SDimitry Andric   return ActOnFinishFullExpr(operand, /*DiscardedValue*/ false);
43440b57cec5SDimitry Andric }
43450b57cec5SDimitry Andric 
43460b57cec5SDimitry Andric StmtResult
43470b57cec5SDimitry Andric Sema::ActOnObjCAtSynchronizedStmt(SourceLocation AtLoc, Expr *SyncExpr,
43480b57cec5SDimitry Andric                                   Stmt *SyncBody) {
43490b57cec5SDimitry Andric   // We can't jump into or indirect-jump out of a @synchronized block.
43500b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
43510b57cec5SDimitry Andric   return new (Context) ObjCAtSynchronizedStmt(AtLoc, SyncExpr, SyncBody);
43520b57cec5SDimitry Andric }
43530b57cec5SDimitry Andric 
43540b57cec5SDimitry Andric /// ActOnCXXCatchBlock - Takes an exception declaration and a handler block
43550b57cec5SDimitry Andric /// and creates a proper catch handler from them.
43560b57cec5SDimitry Andric StmtResult
43570b57cec5SDimitry Andric Sema::ActOnCXXCatchBlock(SourceLocation CatchLoc, Decl *ExDecl,
43580b57cec5SDimitry Andric                          Stmt *HandlerBlock) {
43590b57cec5SDimitry Andric   // There's nothing to test that ActOnExceptionDecl didn't already test.
43600b57cec5SDimitry Andric   return new (Context)
43610b57cec5SDimitry Andric       CXXCatchStmt(CatchLoc, cast_or_null<VarDecl>(ExDecl), HandlerBlock);
43620b57cec5SDimitry Andric }
43630b57cec5SDimitry Andric 
43640b57cec5SDimitry Andric StmtResult
43650b57cec5SDimitry Andric Sema::ActOnObjCAutoreleasePoolStmt(SourceLocation AtLoc, Stmt *Body) {
43660b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
43670b57cec5SDimitry Andric   return new (Context) ObjCAutoreleasePoolStmt(AtLoc, Body);
43680b57cec5SDimitry Andric }
43690b57cec5SDimitry Andric 
43700b57cec5SDimitry Andric namespace {
43710b57cec5SDimitry Andric class CatchHandlerType {
43720b57cec5SDimitry Andric   QualType QT;
43730b57cec5SDimitry Andric   unsigned IsPointer : 1;
43740b57cec5SDimitry Andric 
43750b57cec5SDimitry Andric   // This is a special constructor to be used only with DenseMapInfo's
43760b57cec5SDimitry Andric   // getEmptyKey() and getTombstoneKey() functions.
43770b57cec5SDimitry Andric   friend struct llvm::DenseMapInfo<CatchHandlerType>;
43780b57cec5SDimitry Andric   enum Unique { ForDenseMap };
43790b57cec5SDimitry Andric   CatchHandlerType(QualType QT, Unique) : QT(QT), IsPointer(false) {}
43800b57cec5SDimitry Andric 
43810b57cec5SDimitry Andric public:
43820b57cec5SDimitry Andric   /// Used when creating a CatchHandlerType from a handler type; will determine
43830b57cec5SDimitry Andric   /// whether the type is a pointer or reference and will strip off the top
43840b57cec5SDimitry Andric   /// level pointer and cv-qualifiers.
43850b57cec5SDimitry Andric   CatchHandlerType(QualType Q) : QT(Q), IsPointer(false) {
43860b57cec5SDimitry Andric     if (QT->isPointerType())
43870b57cec5SDimitry Andric       IsPointer = true;
43880b57cec5SDimitry Andric 
438906c3fb27SDimitry Andric     QT = QT.getUnqualifiedType();
43900b57cec5SDimitry Andric     if (IsPointer || QT->isReferenceType())
43910b57cec5SDimitry Andric       QT = QT->getPointeeType();
43920b57cec5SDimitry Andric   }
43930b57cec5SDimitry Andric 
43940b57cec5SDimitry Andric   /// Used when creating a CatchHandlerType from a base class type; pretends the
43950b57cec5SDimitry Andric   /// type passed in had the pointer qualifier, does not need to get an
43960b57cec5SDimitry Andric   /// unqualified type.
43970b57cec5SDimitry Andric   CatchHandlerType(QualType QT, bool IsPointer)
43980b57cec5SDimitry Andric       : QT(QT), IsPointer(IsPointer) {}
43990b57cec5SDimitry Andric 
44000b57cec5SDimitry Andric   QualType underlying() const { return QT; }
44010b57cec5SDimitry Andric   bool isPointer() const { return IsPointer; }
44020b57cec5SDimitry Andric 
44030b57cec5SDimitry Andric   friend bool operator==(const CatchHandlerType &LHS,
44040b57cec5SDimitry Andric                          const CatchHandlerType &RHS) {
44050b57cec5SDimitry Andric     // If the pointer qualification does not match, we can return early.
44060b57cec5SDimitry Andric     if (LHS.IsPointer != RHS.IsPointer)
44070b57cec5SDimitry Andric       return false;
44080b57cec5SDimitry Andric     // Otherwise, check the underlying type without cv-qualifiers.
44090b57cec5SDimitry Andric     return LHS.QT == RHS.QT;
44100b57cec5SDimitry Andric   }
44110b57cec5SDimitry Andric };
44120b57cec5SDimitry Andric } // namespace
44130b57cec5SDimitry Andric 
44140b57cec5SDimitry Andric namespace llvm {
44150b57cec5SDimitry Andric template <> struct DenseMapInfo<CatchHandlerType> {
44160b57cec5SDimitry Andric   static CatchHandlerType getEmptyKey() {
44170b57cec5SDimitry Andric     return CatchHandlerType(DenseMapInfo<QualType>::getEmptyKey(),
44180b57cec5SDimitry Andric                        CatchHandlerType::ForDenseMap);
44190b57cec5SDimitry Andric   }
44200b57cec5SDimitry Andric 
44210b57cec5SDimitry Andric   static CatchHandlerType getTombstoneKey() {
44220b57cec5SDimitry Andric     return CatchHandlerType(DenseMapInfo<QualType>::getTombstoneKey(),
44230b57cec5SDimitry Andric                        CatchHandlerType::ForDenseMap);
44240b57cec5SDimitry Andric   }
44250b57cec5SDimitry Andric 
44260b57cec5SDimitry Andric   static unsigned getHashValue(const CatchHandlerType &Base) {
44270b57cec5SDimitry Andric     return DenseMapInfo<QualType>::getHashValue(Base.underlying());
44280b57cec5SDimitry Andric   }
44290b57cec5SDimitry Andric 
44300b57cec5SDimitry Andric   static bool isEqual(const CatchHandlerType &LHS,
44310b57cec5SDimitry Andric                       const CatchHandlerType &RHS) {
44320b57cec5SDimitry Andric     return LHS == RHS;
44330b57cec5SDimitry Andric   }
44340b57cec5SDimitry Andric };
44350b57cec5SDimitry Andric }
44360b57cec5SDimitry Andric 
44370b57cec5SDimitry Andric namespace {
44380b57cec5SDimitry Andric class CatchTypePublicBases {
443906c3fb27SDimitry Andric   const llvm::DenseMap<QualType, CXXCatchStmt *> &TypesToCheck;
44400b57cec5SDimitry Andric 
44410b57cec5SDimitry Andric   CXXCatchStmt *FoundHandler;
444206c3fb27SDimitry Andric   QualType FoundHandlerType;
444306c3fb27SDimitry Andric   QualType TestAgainstType;
44440b57cec5SDimitry Andric 
44450b57cec5SDimitry Andric public:
444606c3fb27SDimitry Andric   CatchTypePublicBases(const llvm::DenseMap<QualType, CXXCatchStmt *> &T,
444706c3fb27SDimitry Andric                        QualType QT)
444806c3fb27SDimitry Andric       : TypesToCheck(T), FoundHandler(nullptr), TestAgainstType(QT) {}
44490b57cec5SDimitry Andric 
44500b57cec5SDimitry Andric   CXXCatchStmt *getFoundHandler() const { return FoundHandler; }
445106c3fb27SDimitry Andric   QualType getFoundHandlerType() const { return FoundHandlerType; }
44520b57cec5SDimitry Andric 
44530b57cec5SDimitry Andric   bool operator()(const CXXBaseSpecifier *S, CXXBasePath &) {
44540b57cec5SDimitry Andric     if (S->getAccessSpecifier() == AccessSpecifier::AS_public) {
445506c3fb27SDimitry Andric       QualType Check = S->getType().getCanonicalType();
44560b57cec5SDimitry Andric       const auto &M = TypesToCheck;
44570b57cec5SDimitry Andric       auto I = M.find(Check);
44580b57cec5SDimitry Andric       if (I != M.end()) {
445906c3fb27SDimitry Andric         // We're pretty sure we found what we need to find. However, we still
446006c3fb27SDimitry Andric         // need to make sure that we properly compare for pointers and
446106c3fb27SDimitry Andric         // references, to handle cases like:
446206c3fb27SDimitry Andric         //
446306c3fb27SDimitry Andric         // } catch (Base *b) {
446406c3fb27SDimitry Andric         // } catch (Derived &d) {
446506c3fb27SDimitry Andric         // }
446606c3fb27SDimitry Andric         //
446706c3fb27SDimitry Andric         // where there is a qualification mismatch that disqualifies this
446806c3fb27SDimitry Andric         // handler as a potential problem.
446906c3fb27SDimitry Andric         if (I->second->getCaughtType()->isPointerType() ==
447006c3fb27SDimitry Andric                 TestAgainstType->isPointerType()) {
44710b57cec5SDimitry Andric           FoundHandler = I->second;
447206c3fb27SDimitry Andric           FoundHandlerType = Check;
44730b57cec5SDimitry Andric           return true;
44740b57cec5SDimitry Andric         }
44750b57cec5SDimitry Andric       }
447606c3fb27SDimitry Andric     }
44770b57cec5SDimitry Andric     return false;
44780b57cec5SDimitry Andric   }
44790b57cec5SDimitry Andric };
44800b57cec5SDimitry Andric }
44810b57cec5SDimitry Andric 
44820b57cec5SDimitry Andric /// ActOnCXXTryBlock - Takes a try compound-statement and a number of
44830b57cec5SDimitry Andric /// handlers and creates a try statement from them.
44840b57cec5SDimitry Andric StmtResult Sema::ActOnCXXTryBlock(SourceLocation TryLoc, Stmt *TryBlock,
44850b57cec5SDimitry Andric                                   ArrayRef<Stmt *> Handlers) {
44865f757f3fSDimitry Andric   const llvm::Triple &T = Context.getTargetInfo().getTriple();
44875f757f3fSDimitry Andric   const bool IsOpenMPGPUTarget =
44885f757f3fSDimitry Andric       getLangOpts().OpenMPIsTargetDevice && (T.isNVPTX() || T.isAMDGCN());
44895f757f3fSDimitry Andric   // Don't report an error if 'try' is used in system headers or in an OpenMP
44905f757f3fSDimitry Andric   // target region compiled for a GPU architecture.
44915f757f3fSDimitry Andric   if (!IsOpenMPGPUTarget && !getLangOpts().CXXExceptions &&
44920b57cec5SDimitry Andric       !getSourceManager().isInSystemHeader(TryLoc) && !getLangOpts().CUDA) {
44930b57cec5SDimitry Andric     // Delay error emission for the OpenMP device code.
44940b57cec5SDimitry Andric     targetDiag(TryLoc, diag::err_exceptions_disabled) << "try";
44950b57cec5SDimitry Andric   }
44960b57cec5SDimitry Andric 
44975f757f3fSDimitry Andric   // In OpenMP target regions, we assume that catch is never reached on GPU
44985f757f3fSDimitry Andric   // targets.
44995f757f3fSDimitry Andric   if (IsOpenMPGPUTarget)
45005f757f3fSDimitry Andric     targetDiag(TryLoc, diag::warn_try_not_valid_on_target) << T.str();
45015f757f3fSDimitry Andric 
45020b57cec5SDimitry Andric   // Exceptions aren't allowed in CUDA device code.
45030b57cec5SDimitry Andric   if (getLangOpts().CUDA)
45040b57cec5SDimitry Andric     CUDADiagIfDeviceCode(TryLoc, diag::err_cuda_device_exceptions)
45050b57cec5SDimitry Andric         << "try" << CurrentCUDATarget();
45060b57cec5SDimitry Andric 
45070b57cec5SDimitry Andric   if (getCurScope() && getCurScope()->isOpenMPSimdDirectiveScope())
45080b57cec5SDimitry Andric     Diag(TryLoc, diag::err_omp_simd_region_cannot_use_stmt) << "try";
45090b57cec5SDimitry Andric 
45100b57cec5SDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
45110b57cec5SDimitry Andric 
45120b57cec5SDimitry Andric   // C++ try is incompatible with SEH __try.
45130b57cec5SDimitry Andric   if (!getLangOpts().Borland && FSI->FirstSEHTryLoc.isValid()) {
4514349cc55cSDimitry Andric     Diag(TryLoc, diag::err_mixing_cxx_try_seh_try) << 0;
45150b57cec5SDimitry Andric     Diag(FSI->FirstSEHTryLoc, diag::note_conflicting_try_here) << "'__try'";
45160b57cec5SDimitry Andric   }
45170b57cec5SDimitry Andric 
45180b57cec5SDimitry Andric   const unsigned NumHandlers = Handlers.size();
45190b57cec5SDimitry Andric   assert(!Handlers.empty() &&
45200b57cec5SDimitry Andric          "The parser shouldn't call this if there are no handlers.");
45210b57cec5SDimitry Andric 
452206c3fb27SDimitry Andric   llvm::DenseMap<QualType, CXXCatchStmt *> HandledBaseTypes;
45230b57cec5SDimitry Andric   llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> HandledTypes;
45240b57cec5SDimitry Andric   for (unsigned i = 0; i < NumHandlers; ++i) {
45250b57cec5SDimitry Andric     CXXCatchStmt *H = cast<CXXCatchStmt>(Handlers[i]);
45260b57cec5SDimitry Andric 
45270b57cec5SDimitry Andric     // Diagnose when the handler is a catch-all handler, but it isn't the last
45280b57cec5SDimitry Andric     // handler for the try block. [except.handle]p5. Also, skip exception
45290b57cec5SDimitry Andric     // declarations that are invalid, since we can't usefully report on them.
45300b57cec5SDimitry Andric     if (!H->getExceptionDecl()) {
45310b57cec5SDimitry Andric       if (i < NumHandlers - 1)
45320b57cec5SDimitry Andric         return StmtError(Diag(H->getBeginLoc(), diag::err_early_catch_all));
45330b57cec5SDimitry Andric       continue;
45340b57cec5SDimitry Andric     } else if (H->getExceptionDecl()->isInvalidDecl())
45350b57cec5SDimitry Andric       continue;
45360b57cec5SDimitry Andric 
45370b57cec5SDimitry Andric     // Walk the type hierarchy to diagnose when this type has already been
45380b57cec5SDimitry Andric     // handled (duplication), or cannot be handled (derivation inversion). We
45390b57cec5SDimitry Andric     // ignore top-level cv-qualifiers, per [except.handle]p3
454006c3fb27SDimitry Andric     CatchHandlerType HandlerCHT = H->getCaughtType().getCanonicalType();
45410b57cec5SDimitry Andric 
45420b57cec5SDimitry Andric     // We can ignore whether the type is a reference or a pointer; we need the
45430b57cec5SDimitry Andric     // underlying declaration type in order to get at the underlying record
45440b57cec5SDimitry Andric     // decl, if there is one.
45450b57cec5SDimitry Andric     QualType Underlying = HandlerCHT.underlying();
45460b57cec5SDimitry Andric     if (auto *RD = Underlying->getAsCXXRecordDecl()) {
45470b57cec5SDimitry Andric       if (!RD->hasDefinition())
45480b57cec5SDimitry Andric         continue;
45490b57cec5SDimitry Andric       // Check that none of the public, unambiguous base classes are in the
45500b57cec5SDimitry Andric       // map ([except.handle]p1). Give the base classes the same pointer
45510b57cec5SDimitry Andric       // qualification as the original type we are basing off of. This allows
45520b57cec5SDimitry Andric       // comparison against the handler type using the same top-level pointer
45530b57cec5SDimitry Andric       // as the original type.
45540b57cec5SDimitry Andric       CXXBasePaths Paths;
45550b57cec5SDimitry Andric       Paths.setOrigin(RD);
455606c3fb27SDimitry Andric       CatchTypePublicBases CTPB(HandledBaseTypes,
455706c3fb27SDimitry Andric                                 H->getCaughtType().getCanonicalType());
45580b57cec5SDimitry Andric       if (RD->lookupInBases(CTPB, Paths)) {
45590b57cec5SDimitry Andric         const CXXCatchStmt *Problem = CTPB.getFoundHandler();
456006c3fb27SDimitry Andric         if (!Paths.isAmbiguous(
456106c3fb27SDimitry Andric                 CanQualType::CreateUnsafe(CTPB.getFoundHandlerType()))) {
45620b57cec5SDimitry Andric           Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
45630b57cec5SDimitry Andric                diag::warn_exception_caught_by_earlier_handler)
45640b57cec5SDimitry Andric               << H->getCaughtType();
45650b57cec5SDimitry Andric           Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
45660b57cec5SDimitry Andric                 diag::note_previous_exception_handler)
45670b57cec5SDimitry Andric               << Problem->getCaughtType();
45680b57cec5SDimitry Andric         }
45690b57cec5SDimitry Andric       }
457006c3fb27SDimitry Andric       // Strip the qualifiers here because we're going to be comparing this
457106c3fb27SDimitry Andric       // type to the base type specifiers of a class, which are ignored in a
457206c3fb27SDimitry Andric       // base specifier per [class.derived.general]p2.
457306c3fb27SDimitry Andric       HandledBaseTypes[Underlying.getUnqualifiedType()] = H;
45740b57cec5SDimitry Andric     }
45750b57cec5SDimitry Andric 
45760b57cec5SDimitry Andric     // Add the type the list of ones we have handled; diagnose if we've already
45770b57cec5SDimitry Andric     // handled it.
457806c3fb27SDimitry Andric     auto R = HandledTypes.insert(
457906c3fb27SDimitry Andric         std::make_pair(H->getCaughtType().getCanonicalType(), H));
45800b57cec5SDimitry Andric     if (!R.second) {
45810b57cec5SDimitry Andric       const CXXCatchStmt *Problem = R.first->second;
45820b57cec5SDimitry Andric       Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
45830b57cec5SDimitry Andric            diag::warn_exception_caught_by_earlier_handler)
45840b57cec5SDimitry Andric           << H->getCaughtType();
45850b57cec5SDimitry Andric       Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
45860b57cec5SDimitry Andric            diag::note_previous_exception_handler)
45870b57cec5SDimitry Andric           << Problem->getCaughtType();
45880b57cec5SDimitry Andric     }
45890b57cec5SDimitry Andric   }
45900b57cec5SDimitry Andric 
45910b57cec5SDimitry Andric   FSI->setHasCXXTry(TryLoc);
45920b57cec5SDimitry Andric 
459306c3fb27SDimitry Andric   return CXXTryStmt::Create(Context, TryLoc, cast<CompoundStmt>(TryBlock),
459406c3fb27SDimitry Andric                             Handlers);
45950b57cec5SDimitry Andric }
45960b57cec5SDimitry Andric 
45970b57cec5SDimitry Andric StmtResult Sema::ActOnSEHTryBlock(bool IsCXXTry, SourceLocation TryLoc,
45980b57cec5SDimitry Andric                                   Stmt *TryBlock, Stmt *Handler) {
45990b57cec5SDimitry Andric   assert(TryBlock && Handler);
46000b57cec5SDimitry Andric 
46010b57cec5SDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
46020b57cec5SDimitry Andric 
46030b57cec5SDimitry Andric   // SEH __try is incompatible with C++ try. Borland appears to support this,
46040b57cec5SDimitry Andric   // however.
46050b57cec5SDimitry Andric   if (!getLangOpts().Borland) {
4606349cc55cSDimitry Andric     if (FSI->FirstCXXOrObjCTryLoc.isValid()) {
4607349cc55cSDimitry Andric       Diag(TryLoc, diag::err_mixing_cxx_try_seh_try) << FSI->FirstTryType;
4608349cc55cSDimitry Andric       Diag(FSI->FirstCXXOrObjCTryLoc, diag::note_conflicting_try_here)
4609349cc55cSDimitry Andric           << (FSI->FirstTryType == sema::FunctionScopeInfo::TryLocIsCXX
4610349cc55cSDimitry Andric                   ? "'try'"
4611349cc55cSDimitry Andric                   : "'@try'");
46120b57cec5SDimitry Andric     }
46130b57cec5SDimitry Andric   }
46140b57cec5SDimitry Andric 
46150b57cec5SDimitry Andric   FSI->setHasSEHTry(TryLoc);
46160b57cec5SDimitry Andric 
46170b57cec5SDimitry Andric   // Reject __try in Obj-C methods, blocks, and captured decls, since we don't
46180b57cec5SDimitry Andric   // track if they use SEH.
46190b57cec5SDimitry Andric   DeclContext *DC = CurContext;
46200b57cec5SDimitry Andric   while (DC && !DC->isFunctionOrMethod())
46210b57cec5SDimitry Andric     DC = DC->getParent();
46220b57cec5SDimitry Andric   FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(DC);
46230b57cec5SDimitry Andric   if (FD)
46240b57cec5SDimitry Andric     FD->setUsesSEHTry(true);
46250b57cec5SDimitry Andric   else
46260b57cec5SDimitry Andric     Diag(TryLoc, diag::err_seh_try_outside_functions);
46270b57cec5SDimitry Andric 
46280b57cec5SDimitry Andric   // Reject __try on unsupported targets.
46290b57cec5SDimitry Andric   if (!Context.getTargetInfo().isSEHTrySupported())
46300b57cec5SDimitry Andric     Diag(TryLoc, diag::err_seh_try_unsupported);
46310b57cec5SDimitry Andric 
46320b57cec5SDimitry Andric   return SEHTryStmt::Create(Context, IsCXXTry, TryLoc, TryBlock, Handler);
46330b57cec5SDimitry Andric }
46340b57cec5SDimitry Andric 
4635480093f4SDimitry Andric StmtResult Sema::ActOnSEHExceptBlock(SourceLocation Loc, Expr *FilterExpr,
46360b57cec5SDimitry Andric                                      Stmt *Block) {
46370b57cec5SDimitry Andric   assert(FilterExpr && Block);
4638480093f4SDimitry Andric   QualType FTy = FilterExpr->getType();
4639480093f4SDimitry Andric   if (!FTy->isIntegerType() && !FTy->isDependentType()) {
4640480093f4SDimitry Andric     return StmtError(
4641480093f4SDimitry Andric         Diag(FilterExpr->getExprLoc(), diag::err_filter_expression_integral)
4642480093f4SDimitry Andric         << FTy);
46430b57cec5SDimitry Andric   }
46440b57cec5SDimitry Andric   return SEHExceptStmt::Create(Context, Loc, FilterExpr, Block);
46450b57cec5SDimitry Andric }
46460b57cec5SDimitry Andric 
46470b57cec5SDimitry Andric void Sema::ActOnStartSEHFinallyBlock() {
46480b57cec5SDimitry Andric   CurrentSEHFinally.push_back(CurScope);
46490b57cec5SDimitry Andric }
46500b57cec5SDimitry Andric 
46510b57cec5SDimitry Andric void Sema::ActOnAbortSEHFinallyBlock() {
46520b57cec5SDimitry Andric   CurrentSEHFinally.pop_back();
46530b57cec5SDimitry Andric }
46540b57cec5SDimitry Andric 
46550b57cec5SDimitry Andric StmtResult Sema::ActOnFinishSEHFinallyBlock(SourceLocation Loc, Stmt *Block) {
46560b57cec5SDimitry Andric   assert(Block);
46570b57cec5SDimitry Andric   CurrentSEHFinally.pop_back();
46580b57cec5SDimitry Andric   return SEHFinallyStmt::Create(Context, Loc, Block);
46590b57cec5SDimitry Andric }
46600b57cec5SDimitry Andric 
46610b57cec5SDimitry Andric StmtResult
46620b57cec5SDimitry Andric Sema::ActOnSEHLeaveStmt(SourceLocation Loc, Scope *CurScope) {
46630b57cec5SDimitry Andric   Scope *SEHTryParent = CurScope;
46640b57cec5SDimitry Andric   while (SEHTryParent && !SEHTryParent->isSEHTryScope())
46650b57cec5SDimitry Andric     SEHTryParent = SEHTryParent->getParent();
46660b57cec5SDimitry Andric   if (!SEHTryParent)
46670b57cec5SDimitry Andric     return StmtError(Diag(Loc, diag::err_ms___leave_not_in___try));
46680b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, Loc, *SEHTryParent);
46690b57cec5SDimitry Andric 
46700b57cec5SDimitry Andric   return new (Context) SEHLeaveStmt(Loc);
46710b57cec5SDimitry Andric }
46720b57cec5SDimitry Andric 
46730b57cec5SDimitry Andric StmtResult Sema::BuildMSDependentExistsStmt(SourceLocation KeywordLoc,
46740b57cec5SDimitry Andric                                             bool IsIfExists,
46750b57cec5SDimitry Andric                                             NestedNameSpecifierLoc QualifierLoc,
46760b57cec5SDimitry Andric                                             DeclarationNameInfo NameInfo,
46770b57cec5SDimitry Andric                                             Stmt *Nested)
46780b57cec5SDimitry Andric {
46790b57cec5SDimitry Andric   return new (Context) MSDependentExistsStmt(KeywordLoc, IsIfExists,
46800b57cec5SDimitry Andric                                              QualifierLoc, NameInfo,
46810b57cec5SDimitry Andric                                              cast<CompoundStmt>(Nested));
46820b57cec5SDimitry Andric }
46830b57cec5SDimitry Andric 
46840b57cec5SDimitry Andric 
46850b57cec5SDimitry Andric StmtResult Sema::ActOnMSDependentExistsStmt(SourceLocation KeywordLoc,
46860b57cec5SDimitry Andric                                             bool IsIfExists,
46870b57cec5SDimitry Andric                                             CXXScopeSpec &SS,
46880b57cec5SDimitry Andric                                             UnqualifiedId &Name,
46890b57cec5SDimitry Andric                                             Stmt *Nested) {
46900b57cec5SDimitry Andric   return BuildMSDependentExistsStmt(KeywordLoc, IsIfExists,
46910b57cec5SDimitry Andric                                     SS.getWithLocInContext(Context),
46920b57cec5SDimitry Andric                                     GetNameFromUnqualifiedId(Name),
46930b57cec5SDimitry Andric                                     Nested);
46940b57cec5SDimitry Andric }
46950b57cec5SDimitry Andric 
46960b57cec5SDimitry Andric RecordDecl*
46970b57cec5SDimitry Andric Sema::CreateCapturedStmtRecordDecl(CapturedDecl *&CD, SourceLocation Loc,
46980b57cec5SDimitry Andric                                    unsigned NumParams) {
46990b57cec5SDimitry Andric   DeclContext *DC = CurContext;
47000b57cec5SDimitry Andric   while (!(DC->isFunctionOrMethod() || DC->isRecord() || DC->isFileContext()))
47010b57cec5SDimitry Andric     DC = DC->getParent();
47020b57cec5SDimitry Andric 
47030b57cec5SDimitry Andric   RecordDecl *RD = nullptr;
47040b57cec5SDimitry Andric   if (getLangOpts().CPlusPlus)
47055f757f3fSDimitry Andric     RD = CXXRecordDecl::Create(Context, TagTypeKind::Struct, DC, Loc, Loc,
47060b57cec5SDimitry Andric                                /*Id=*/nullptr);
47070b57cec5SDimitry Andric   else
47085f757f3fSDimitry Andric     RD = RecordDecl::Create(Context, TagTypeKind::Struct, DC, Loc, Loc,
47095f757f3fSDimitry Andric                             /*Id=*/nullptr);
47100b57cec5SDimitry Andric 
47110b57cec5SDimitry Andric   RD->setCapturedRecord();
47120b57cec5SDimitry Andric   DC->addDecl(RD);
47130b57cec5SDimitry Andric   RD->setImplicit();
47140b57cec5SDimitry Andric   RD->startDefinition();
47150b57cec5SDimitry Andric 
47160b57cec5SDimitry Andric   assert(NumParams > 0 && "CapturedStmt requires context parameter");
47170b57cec5SDimitry Andric   CD = CapturedDecl::Create(Context, CurContext, NumParams);
47180b57cec5SDimitry Andric   DC->addDecl(CD);
47190b57cec5SDimitry Andric   return RD;
47200b57cec5SDimitry Andric }
47210b57cec5SDimitry Andric 
47220b57cec5SDimitry Andric static bool
47230b57cec5SDimitry Andric buildCapturedStmtCaptureList(Sema &S, CapturedRegionScopeInfo *RSI,
47240b57cec5SDimitry Andric                              SmallVectorImpl<CapturedStmt::Capture> &Captures,
47250b57cec5SDimitry Andric                              SmallVectorImpl<Expr *> &CaptureInits) {
47260b57cec5SDimitry Andric   for (const sema::Capture &Cap : RSI->Captures) {
47270b57cec5SDimitry Andric     if (Cap.isInvalid())
47280b57cec5SDimitry Andric       continue;
47290b57cec5SDimitry Andric 
47300b57cec5SDimitry Andric     // Form the initializer for the capture.
47310b57cec5SDimitry Andric     ExprResult Init = S.BuildCaptureInit(Cap, Cap.getLocation(),
47320b57cec5SDimitry Andric                                          RSI->CapRegionKind == CR_OpenMP);
47330b57cec5SDimitry Andric 
47340b57cec5SDimitry Andric     // FIXME: Bail out now if the capture is not used and the initializer has
47350b57cec5SDimitry Andric     // no side-effects.
47360b57cec5SDimitry Andric 
47370b57cec5SDimitry Andric     // Create a field for this capture.
47380b57cec5SDimitry Andric     FieldDecl *Field = S.BuildCaptureField(RSI->TheRecordDecl, Cap);
47390b57cec5SDimitry Andric 
47400b57cec5SDimitry Andric     // Add the capture to our list of captures.
47410b57cec5SDimitry Andric     if (Cap.isThisCapture()) {
47420b57cec5SDimitry Andric       Captures.push_back(CapturedStmt::Capture(Cap.getLocation(),
47430b57cec5SDimitry Andric                                                CapturedStmt::VCK_This));
47440b57cec5SDimitry Andric     } else if (Cap.isVLATypeCapture()) {
47450b57cec5SDimitry Andric       Captures.push_back(
47460b57cec5SDimitry Andric           CapturedStmt::Capture(Cap.getLocation(), CapturedStmt::VCK_VLAType));
47470b57cec5SDimitry Andric     } else {
47480b57cec5SDimitry Andric       assert(Cap.isVariableCapture() && "unknown kind of capture");
47490b57cec5SDimitry Andric 
47500b57cec5SDimitry Andric       if (S.getLangOpts().OpenMP && RSI->CapRegionKind == CR_OpenMP)
47510b57cec5SDimitry Andric         S.setOpenMPCaptureKind(Field, Cap.getVariable(), RSI->OpenMPLevel);
47520b57cec5SDimitry Andric 
4753bdd1243dSDimitry Andric       Captures.push_back(CapturedStmt::Capture(
4754bdd1243dSDimitry Andric           Cap.getLocation(),
4755bdd1243dSDimitry Andric           Cap.isReferenceCapture() ? CapturedStmt::VCK_ByRef
47560b57cec5SDimitry Andric                                    : CapturedStmt::VCK_ByCopy,
4757bdd1243dSDimitry Andric           cast<VarDecl>(Cap.getVariable())));
47580b57cec5SDimitry Andric     }
47590b57cec5SDimitry Andric     CaptureInits.push_back(Init.get());
47600b57cec5SDimitry Andric   }
47610b57cec5SDimitry Andric   return false;
47620b57cec5SDimitry Andric }
47630b57cec5SDimitry Andric 
47640b57cec5SDimitry Andric void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
47650b57cec5SDimitry Andric                                     CapturedRegionKind Kind,
47660b57cec5SDimitry Andric                                     unsigned NumParams) {
47670b57cec5SDimitry Andric   CapturedDecl *CD = nullptr;
47680b57cec5SDimitry Andric   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, NumParams);
47690b57cec5SDimitry Andric 
47700b57cec5SDimitry Andric   // Build the context parameter
47710b57cec5SDimitry Andric   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
47720b57cec5SDimitry Andric   IdentifierInfo *ParamName = &Context.Idents.get("__context");
47730b57cec5SDimitry Andric   QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
47740b57cec5SDimitry Andric   auto *Param =
47750b57cec5SDimitry Andric       ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
47765f757f3fSDimitry Andric                                 ImplicitParamKind::CapturedContext);
47770b57cec5SDimitry Andric   DC->addDecl(Param);
47780b57cec5SDimitry Andric 
47790b57cec5SDimitry Andric   CD->setContextParam(0, Param);
47800b57cec5SDimitry Andric 
47810b57cec5SDimitry Andric   // Enter the capturing scope for this captured region.
47820b57cec5SDimitry Andric   PushCapturedRegionScope(CurScope, CD, RD, Kind);
47830b57cec5SDimitry Andric 
47840b57cec5SDimitry Andric   if (CurScope)
47850b57cec5SDimitry Andric     PushDeclContext(CurScope, CD);
47860b57cec5SDimitry Andric   else
47870b57cec5SDimitry Andric     CurContext = CD;
47880b57cec5SDimitry Andric 
47890b57cec5SDimitry Andric   PushExpressionEvaluationContext(
47900b57cec5SDimitry Andric       ExpressionEvaluationContext::PotentiallyEvaluated);
479106c3fb27SDimitry Andric   ExprEvalContexts.back().InImmediateEscalatingFunctionContext = false;
47920b57cec5SDimitry Andric }
47930b57cec5SDimitry Andric 
47940b57cec5SDimitry Andric void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
47950b57cec5SDimitry Andric                                     CapturedRegionKind Kind,
4796a7dea167SDimitry Andric                                     ArrayRef<CapturedParamNameType> Params,
4797a7dea167SDimitry Andric                                     unsigned OpenMPCaptureLevel) {
47980b57cec5SDimitry Andric   CapturedDecl *CD = nullptr;
47990b57cec5SDimitry Andric   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, Params.size());
48000b57cec5SDimitry Andric 
48010b57cec5SDimitry Andric   // Build the context parameter
48020b57cec5SDimitry Andric   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
48030b57cec5SDimitry Andric   bool ContextIsFound = false;
48040b57cec5SDimitry Andric   unsigned ParamNum = 0;
48050b57cec5SDimitry Andric   for (ArrayRef<CapturedParamNameType>::iterator I = Params.begin(),
48060b57cec5SDimitry Andric                                                  E = Params.end();
48070b57cec5SDimitry Andric        I != E; ++I, ++ParamNum) {
48080b57cec5SDimitry Andric     if (I->second.isNull()) {
48090b57cec5SDimitry Andric       assert(!ContextIsFound &&
48100b57cec5SDimitry Andric              "null type has been found already for '__context' parameter");
48110b57cec5SDimitry Andric       IdentifierInfo *ParamName = &Context.Idents.get("__context");
48120b57cec5SDimitry Andric       QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD))
48130b57cec5SDimitry Andric                                .withConst()
48140b57cec5SDimitry Andric                                .withRestrict();
48150b57cec5SDimitry Andric       auto *Param =
48160b57cec5SDimitry Andric           ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
48175f757f3fSDimitry Andric                                     ImplicitParamKind::CapturedContext);
48180b57cec5SDimitry Andric       DC->addDecl(Param);
48190b57cec5SDimitry Andric       CD->setContextParam(ParamNum, Param);
48200b57cec5SDimitry Andric       ContextIsFound = true;
48210b57cec5SDimitry Andric     } else {
48220b57cec5SDimitry Andric       IdentifierInfo *ParamName = &Context.Idents.get(I->first);
48230b57cec5SDimitry Andric       auto *Param =
48240b57cec5SDimitry Andric           ImplicitParamDecl::Create(Context, DC, Loc, ParamName, I->second,
48255f757f3fSDimitry Andric                                     ImplicitParamKind::CapturedContext);
48260b57cec5SDimitry Andric       DC->addDecl(Param);
48270b57cec5SDimitry Andric       CD->setParam(ParamNum, Param);
48280b57cec5SDimitry Andric     }
48290b57cec5SDimitry Andric   }
48300b57cec5SDimitry Andric   assert(ContextIsFound && "no null type for '__context' parameter");
48310b57cec5SDimitry Andric   if (!ContextIsFound) {
48320b57cec5SDimitry Andric     // Add __context implicitly if it is not specified.
48330b57cec5SDimitry Andric     IdentifierInfo *ParamName = &Context.Idents.get("__context");
48340b57cec5SDimitry Andric     QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
48350b57cec5SDimitry Andric     auto *Param =
48360b57cec5SDimitry Andric         ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
48375f757f3fSDimitry Andric                                   ImplicitParamKind::CapturedContext);
48380b57cec5SDimitry Andric     DC->addDecl(Param);
48390b57cec5SDimitry Andric     CD->setContextParam(ParamNum, Param);
48400b57cec5SDimitry Andric   }
48410b57cec5SDimitry Andric   // Enter the capturing scope for this captured region.
4842a7dea167SDimitry Andric   PushCapturedRegionScope(CurScope, CD, RD, Kind, OpenMPCaptureLevel);
48430b57cec5SDimitry Andric 
48440b57cec5SDimitry Andric   if (CurScope)
48450b57cec5SDimitry Andric     PushDeclContext(CurScope, CD);
48460b57cec5SDimitry Andric   else
48470b57cec5SDimitry Andric     CurContext = CD;
48480b57cec5SDimitry Andric 
48490b57cec5SDimitry Andric   PushExpressionEvaluationContext(
48500b57cec5SDimitry Andric       ExpressionEvaluationContext::PotentiallyEvaluated);
48510b57cec5SDimitry Andric }
48520b57cec5SDimitry Andric 
48530b57cec5SDimitry Andric void Sema::ActOnCapturedRegionError() {
48540b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
48550b57cec5SDimitry Andric   PopExpressionEvaluationContext();
48560b57cec5SDimitry Andric   PopDeclContext();
48570b57cec5SDimitry Andric   PoppedFunctionScopePtr ScopeRAII = PopFunctionScopeInfo();
48580b57cec5SDimitry Andric   CapturedRegionScopeInfo *RSI = cast<CapturedRegionScopeInfo>(ScopeRAII.get());
48590b57cec5SDimitry Andric 
48600b57cec5SDimitry Andric   RecordDecl *Record = RSI->TheRecordDecl;
48610b57cec5SDimitry Andric   Record->setInvalidDecl();
48620b57cec5SDimitry Andric 
48630b57cec5SDimitry Andric   SmallVector<Decl*, 4> Fields(Record->fields());
48640b57cec5SDimitry Andric   ActOnFields(/*Scope=*/nullptr, Record->getLocation(), Record, Fields,
48650b57cec5SDimitry Andric               SourceLocation(), SourceLocation(), ParsedAttributesView());
48660b57cec5SDimitry Andric }
48670b57cec5SDimitry Andric 
48680b57cec5SDimitry Andric StmtResult Sema::ActOnCapturedRegionEnd(Stmt *S) {
48690b57cec5SDimitry Andric   // Leave the captured scope before we start creating captures in the
48700b57cec5SDimitry Andric   // enclosing scope.
48710b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
48720b57cec5SDimitry Andric   PopExpressionEvaluationContext();
48730b57cec5SDimitry Andric   PopDeclContext();
48740b57cec5SDimitry Andric   PoppedFunctionScopePtr ScopeRAII = PopFunctionScopeInfo();
48750b57cec5SDimitry Andric   CapturedRegionScopeInfo *RSI = cast<CapturedRegionScopeInfo>(ScopeRAII.get());
48760b57cec5SDimitry Andric 
48770b57cec5SDimitry Andric   SmallVector<CapturedStmt::Capture, 4> Captures;
48780b57cec5SDimitry Andric   SmallVector<Expr *, 4> CaptureInits;
48790b57cec5SDimitry Andric   if (buildCapturedStmtCaptureList(*this, RSI, Captures, CaptureInits))
48800b57cec5SDimitry Andric     return StmtError();
48810b57cec5SDimitry Andric 
48820b57cec5SDimitry Andric   CapturedDecl *CD = RSI->TheCapturedDecl;
48830b57cec5SDimitry Andric   RecordDecl *RD = RSI->TheRecordDecl;
48840b57cec5SDimitry Andric 
48850b57cec5SDimitry Andric   CapturedStmt *Res = CapturedStmt::Create(
48860b57cec5SDimitry Andric       getASTContext(), S, static_cast<CapturedRegionKind>(RSI->CapRegionKind),
48870b57cec5SDimitry Andric       Captures, CaptureInits, CD, RD);
48880b57cec5SDimitry Andric 
48890b57cec5SDimitry Andric   CD->setBody(Res->getCapturedStmt());
48900b57cec5SDimitry Andric   RD->completeDefinition();
48910b57cec5SDimitry Andric 
48920b57cec5SDimitry Andric   return Res;
48930b57cec5SDimitry Andric }
4894