xref: /freebsd/contrib/llvm-project/clang/lib/Sema/SemaStmt.cpp (revision 972a253a57b6f144b0e4a3e2080a2a0076ec55a0)
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"
420b57cec5SDimitry Andric 
430b57cec5SDimitry Andric using namespace clang;
440b57cec5SDimitry Andric using namespace sema;
450b57cec5SDimitry Andric 
460b57cec5SDimitry Andric StmtResult Sema::ActOnExprStmt(ExprResult FE, bool DiscardedValue) {
470b57cec5SDimitry Andric   if (FE.isInvalid())
480b57cec5SDimitry Andric     return StmtError();
490b57cec5SDimitry Andric 
500b57cec5SDimitry Andric   FE = ActOnFinishFullExpr(FE.get(), FE.get()->getExprLoc(), DiscardedValue);
510b57cec5SDimitry Andric   if (FE.isInvalid())
520b57cec5SDimitry Andric     return StmtError();
530b57cec5SDimitry Andric 
540b57cec5SDimitry Andric   // C99 6.8.3p2: The expression in an expression statement is evaluated as a
550b57cec5SDimitry Andric   // void expression for its side effects.  Conversion to void allows any
560b57cec5SDimitry Andric   // operand, even incomplete types.
570b57cec5SDimitry Andric 
580b57cec5SDimitry Andric   // Same thing in for stmt first clause (when expr) and third clause.
590b57cec5SDimitry Andric   return StmtResult(FE.getAs<Stmt>());
600b57cec5SDimitry Andric }
610b57cec5SDimitry Andric 
620b57cec5SDimitry Andric 
630b57cec5SDimitry Andric StmtResult Sema::ActOnExprStmtError() {
640b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
650b57cec5SDimitry Andric   return StmtError();
660b57cec5SDimitry Andric }
670b57cec5SDimitry Andric 
680b57cec5SDimitry Andric StmtResult Sema::ActOnNullStmt(SourceLocation SemiLoc,
690b57cec5SDimitry Andric                                bool HasLeadingEmptyMacro) {
700b57cec5SDimitry Andric   return new (Context) NullStmt(SemiLoc, HasLeadingEmptyMacro);
710b57cec5SDimitry Andric }
720b57cec5SDimitry Andric 
730b57cec5SDimitry Andric StmtResult Sema::ActOnDeclStmt(DeclGroupPtrTy dg, SourceLocation StartLoc,
740b57cec5SDimitry Andric                                SourceLocation EndLoc) {
750b57cec5SDimitry Andric   DeclGroupRef DG = dg.get();
760b57cec5SDimitry Andric 
770b57cec5SDimitry Andric   // If we have an invalid decl, just return an error.
780b57cec5SDimitry Andric   if (DG.isNull()) return StmtError();
790b57cec5SDimitry Andric 
800b57cec5SDimitry Andric   return new (Context) DeclStmt(DG, StartLoc, EndLoc);
810b57cec5SDimitry Andric }
820b57cec5SDimitry Andric 
830b57cec5SDimitry Andric void Sema::ActOnForEachDeclStmt(DeclGroupPtrTy dg) {
840b57cec5SDimitry Andric   DeclGroupRef DG = dg.get();
850b57cec5SDimitry Andric 
860b57cec5SDimitry Andric   // If we don't have a declaration, or we have an invalid declaration,
870b57cec5SDimitry Andric   // just return.
880b57cec5SDimitry Andric   if (DG.isNull() || !DG.isSingleDecl())
890b57cec5SDimitry Andric     return;
900b57cec5SDimitry Andric 
910b57cec5SDimitry Andric   Decl *decl = DG.getSingleDecl();
920b57cec5SDimitry Andric   if (!decl || decl->isInvalidDecl())
930b57cec5SDimitry Andric     return;
940b57cec5SDimitry Andric 
950b57cec5SDimitry Andric   // Only variable declarations are permitted.
960b57cec5SDimitry Andric   VarDecl *var = dyn_cast<VarDecl>(decl);
970b57cec5SDimitry Andric   if (!var) {
980b57cec5SDimitry Andric     Diag(decl->getLocation(), diag::err_non_variable_decl_in_for);
990b57cec5SDimitry Andric     decl->setInvalidDecl();
1000b57cec5SDimitry Andric     return;
1010b57cec5SDimitry Andric   }
1020b57cec5SDimitry Andric 
1030b57cec5SDimitry Andric   // foreach variables are never actually initialized in the way that
1040b57cec5SDimitry Andric   // the parser came up with.
1050b57cec5SDimitry Andric   var->setInit(nullptr);
1060b57cec5SDimitry Andric 
1070b57cec5SDimitry Andric   // In ARC, we don't need to retain the iteration variable of a fast
1080b57cec5SDimitry Andric   // enumeration loop.  Rather than actually trying to catch that
1090b57cec5SDimitry Andric   // during declaration processing, we remove the consequences here.
1100b57cec5SDimitry Andric   if (getLangOpts().ObjCAutoRefCount) {
1110b57cec5SDimitry Andric     QualType type = var->getType();
1120b57cec5SDimitry Andric 
1130b57cec5SDimitry Andric     // Only do this if we inferred the lifetime.  Inferred lifetime
1140b57cec5SDimitry Andric     // will show up as a local qualifier because explicit lifetime
1150b57cec5SDimitry Andric     // should have shown up as an AttributedType instead.
1160b57cec5SDimitry Andric     if (type.getLocalQualifiers().getObjCLifetime() == Qualifiers::OCL_Strong) {
1170b57cec5SDimitry Andric       // Add 'const' and mark the variable as pseudo-strong.
1180b57cec5SDimitry Andric       var->setType(type.withConst());
1190b57cec5SDimitry Andric       var->setARCPseudoStrong(true);
1200b57cec5SDimitry Andric     }
1210b57cec5SDimitry Andric   }
1220b57cec5SDimitry Andric }
1230b57cec5SDimitry Andric 
1240b57cec5SDimitry Andric /// Diagnose unused comparisons, both builtin and overloaded operators.
1250b57cec5SDimitry Andric /// For '==' and '!=', suggest fixits for '=' or '|='.
1260b57cec5SDimitry Andric ///
1270b57cec5SDimitry Andric /// Adding a cast to void (or other expression wrappers) will prevent the
1280b57cec5SDimitry Andric /// warning from firing.
1290b57cec5SDimitry Andric static bool DiagnoseUnusedComparison(Sema &S, const Expr *E) {
1300b57cec5SDimitry Andric   SourceLocation Loc;
1310b57cec5SDimitry Andric   bool CanAssign;
1320b57cec5SDimitry Andric   enum { Equality, Inequality, Relational, ThreeWay } Kind;
1330b57cec5SDimitry Andric 
1340b57cec5SDimitry Andric   if (const BinaryOperator *Op = dyn_cast<BinaryOperator>(E)) {
1350b57cec5SDimitry Andric     if (!Op->isComparisonOp())
1360b57cec5SDimitry Andric       return false;
1370b57cec5SDimitry Andric 
1380b57cec5SDimitry Andric     if (Op->getOpcode() == BO_EQ)
1390b57cec5SDimitry Andric       Kind = Equality;
1400b57cec5SDimitry Andric     else if (Op->getOpcode() == BO_NE)
1410b57cec5SDimitry Andric       Kind = Inequality;
1420b57cec5SDimitry Andric     else if (Op->getOpcode() == BO_Cmp)
1430b57cec5SDimitry Andric       Kind = ThreeWay;
1440b57cec5SDimitry Andric     else {
1450b57cec5SDimitry Andric       assert(Op->isRelationalOp());
1460b57cec5SDimitry Andric       Kind = Relational;
1470b57cec5SDimitry Andric     }
1480b57cec5SDimitry Andric     Loc = Op->getOperatorLoc();
1490b57cec5SDimitry Andric     CanAssign = Op->getLHS()->IgnoreParenImpCasts()->isLValue();
1500b57cec5SDimitry Andric   } else if (const CXXOperatorCallExpr *Op = dyn_cast<CXXOperatorCallExpr>(E)) {
1510b57cec5SDimitry Andric     switch (Op->getOperator()) {
1520b57cec5SDimitry Andric     case OO_EqualEqual:
1530b57cec5SDimitry Andric       Kind = Equality;
1540b57cec5SDimitry Andric       break;
1550b57cec5SDimitry Andric     case OO_ExclaimEqual:
1560b57cec5SDimitry Andric       Kind = Inequality;
1570b57cec5SDimitry Andric       break;
1580b57cec5SDimitry Andric     case OO_Less:
1590b57cec5SDimitry Andric     case OO_Greater:
1600b57cec5SDimitry Andric     case OO_GreaterEqual:
1610b57cec5SDimitry Andric     case OO_LessEqual:
1620b57cec5SDimitry Andric       Kind = Relational;
1630b57cec5SDimitry Andric       break;
1640b57cec5SDimitry Andric     case OO_Spaceship:
1650b57cec5SDimitry Andric       Kind = ThreeWay;
1660b57cec5SDimitry Andric       break;
1670b57cec5SDimitry Andric     default:
1680b57cec5SDimitry Andric       return false;
1690b57cec5SDimitry Andric     }
1700b57cec5SDimitry Andric 
1710b57cec5SDimitry Andric     Loc = Op->getOperatorLoc();
1720b57cec5SDimitry Andric     CanAssign = Op->getArg(0)->IgnoreParenImpCasts()->isLValue();
1730b57cec5SDimitry Andric   } else {
1740b57cec5SDimitry Andric     // Not a typo-prone comparison.
1750b57cec5SDimitry Andric     return false;
1760b57cec5SDimitry Andric   }
1770b57cec5SDimitry Andric 
1780b57cec5SDimitry Andric   // Suppress warnings when the operator, suspicious as it may be, comes from
1790b57cec5SDimitry Andric   // a macro expansion.
1800b57cec5SDimitry Andric   if (S.SourceMgr.isMacroBodyExpansion(Loc))
1810b57cec5SDimitry Andric     return false;
1820b57cec5SDimitry Andric 
1830b57cec5SDimitry Andric   S.Diag(Loc, diag::warn_unused_comparison)
1840b57cec5SDimitry Andric     << (unsigned)Kind << E->getSourceRange();
1850b57cec5SDimitry Andric 
1860b57cec5SDimitry Andric   // If the LHS is a plausible entity to assign to, provide a fixit hint to
1870b57cec5SDimitry Andric   // correct common typos.
1880b57cec5SDimitry Andric   if (CanAssign) {
1890b57cec5SDimitry Andric     if (Kind == Inequality)
1900b57cec5SDimitry Andric       S.Diag(Loc, diag::note_inequality_comparison_to_or_assign)
1910b57cec5SDimitry Andric         << FixItHint::CreateReplacement(Loc, "|=");
1920b57cec5SDimitry Andric     else if (Kind == Equality)
1930b57cec5SDimitry Andric       S.Diag(Loc, diag::note_equality_comparison_to_assign)
1940b57cec5SDimitry Andric         << FixItHint::CreateReplacement(Loc, "=");
1950b57cec5SDimitry Andric   }
1960b57cec5SDimitry Andric 
1970b57cec5SDimitry Andric   return true;
1980b57cec5SDimitry Andric }
1990b57cec5SDimitry Andric 
200a7dea167SDimitry Andric static bool DiagnoseNoDiscard(Sema &S, const WarnUnusedResultAttr *A,
201a7dea167SDimitry Andric                               SourceLocation Loc, SourceRange R1,
202a7dea167SDimitry Andric                               SourceRange R2, bool IsCtor) {
203a7dea167SDimitry Andric   if (!A)
204a7dea167SDimitry Andric     return false;
205a7dea167SDimitry Andric   StringRef Msg = A->getMessage();
206a7dea167SDimitry Andric 
207a7dea167SDimitry Andric   if (Msg.empty()) {
208a7dea167SDimitry Andric     if (IsCtor)
209a7dea167SDimitry Andric       return S.Diag(Loc, diag::warn_unused_constructor) << A << R1 << R2;
210a7dea167SDimitry Andric     return S.Diag(Loc, diag::warn_unused_result) << A << R1 << R2;
211a7dea167SDimitry Andric   }
212a7dea167SDimitry Andric 
213a7dea167SDimitry Andric   if (IsCtor)
214a7dea167SDimitry Andric     return S.Diag(Loc, diag::warn_unused_constructor_msg) << A << Msg << R1
215a7dea167SDimitry Andric                                                           << R2;
216a7dea167SDimitry Andric   return S.Diag(Loc, diag::warn_unused_result_msg) << A << Msg << R1 << R2;
217a7dea167SDimitry Andric }
218a7dea167SDimitry Andric 
219349cc55cSDimitry Andric void Sema::DiagnoseUnusedExprResult(const Stmt *S, unsigned DiagID) {
2200b57cec5SDimitry Andric   if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
221349cc55cSDimitry Andric     return DiagnoseUnusedExprResult(Label->getSubStmt(), DiagID);
2220b57cec5SDimitry Andric 
2230b57cec5SDimitry Andric   const Expr *E = dyn_cast_or_null<Expr>(S);
2240b57cec5SDimitry Andric   if (!E)
2250b57cec5SDimitry Andric     return;
2260b57cec5SDimitry Andric 
2270b57cec5SDimitry Andric   // If we are in an unevaluated expression context, then there can be no unused
2280b57cec5SDimitry Andric   // results because the results aren't expected to be used in the first place.
2290b57cec5SDimitry Andric   if (isUnevaluatedContext())
2300b57cec5SDimitry Andric     return;
2310b57cec5SDimitry Andric 
2320b57cec5SDimitry Andric   SourceLocation ExprLoc = E->IgnoreParenImpCasts()->getExprLoc();
2330b57cec5SDimitry Andric   // In most cases, we don't want to warn if the expression is written in a
2340b57cec5SDimitry Andric   // macro body, or if the macro comes from a system header. If the offending
2350b57cec5SDimitry Andric   // expression is a call to a function with the warn_unused_result attribute,
2360b57cec5SDimitry Andric   // we warn no matter the location. Because of the order in which the various
2370b57cec5SDimitry Andric   // checks need to happen, we factor out the macro-related test here.
2380b57cec5SDimitry Andric   bool ShouldSuppress =
2390b57cec5SDimitry Andric       SourceMgr.isMacroBodyExpansion(ExprLoc) ||
2400b57cec5SDimitry Andric       SourceMgr.isInSystemMacro(ExprLoc);
2410b57cec5SDimitry Andric 
2420b57cec5SDimitry Andric   const Expr *WarnExpr;
2430b57cec5SDimitry Andric   SourceLocation Loc;
2440b57cec5SDimitry Andric   SourceRange R1, R2;
2450b57cec5SDimitry Andric   if (!E->isUnusedResultAWarning(WarnExpr, Loc, R1, R2, Context))
2460b57cec5SDimitry Andric     return;
2470b57cec5SDimitry Andric 
2480b57cec5SDimitry Andric   // If this is a GNU statement expression expanded from a macro, it is probably
2490b57cec5SDimitry Andric   // unused because it is a function-like macro that can be used as either an
2500b57cec5SDimitry Andric   // expression or statement.  Don't warn, because it is almost certainly a
2510b57cec5SDimitry Andric   // false positive.
2520b57cec5SDimitry Andric   if (isa<StmtExpr>(E) && Loc.isMacroID())
2530b57cec5SDimitry Andric     return;
2540b57cec5SDimitry Andric 
2550b57cec5SDimitry Andric   // Check if this is the UNREFERENCED_PARAMETER from the Microsoft headers.
2560b57cec5SDimitry Andric   // That macro is frequently used to suppress "unused parameter" warnings,
2570b57cec5SDimitry Andric   // but its implementation makes clang's -Wunused-value fire.  Prevent this.
2580b57cec5SDimitry Andric   if (isa<ParenExpr>(E->IgnoreImpCasts()) && Loc.isMacroID()) {
2590b57cec5SDimitry Andric     SourceLocation SpellLoc = Loc;
2600b57cec5SDimitry Andric     if (findMacroSpelling(SpellLoc, "UNREFERENCED_PARAMETER"))
2610b57cec5SDimitry Andric       return;
2620b57cec5SDimitry Andric   }
2630b57cec5SDimitry Andric 
2640b57cec5SDimitry Andric   // Okay, we have an unused result.  Depending on what the base expression is,
2650b57cec5SDimitry Andric   // we might want to make a more specific diagnostic.  Check for one of these
2660b57cec5SDimitry Andric   // cases now.
2670b57cec5SDimitry Andric   if (const FullExpr *Temps = dyn_cast<FullExpr>(E))
2680b57cec5SDimitry Andric     E = Temps->getSubExpr();
2690b57cec5SDimitry Andric   if (const CXXBindTemporaryExpr *TempExpr = dyn_cast<CXXBindTemporaryExpr>(E))
2700b57cec5SDimitry Andric     E = TempExpr->getSubExpr();
2710b57cec5SDimitry Andric 
2720b57cec5SDimitry Andric   if (DiagnoseUnusedComparison(*this, E))
2730b57cec5SDimitry Andric     return;
2740b57cec5SDimitry Andric 
2750b57cec5SDimitry Andric   E = WarnExpr;
276a7dea167SDimitry Andric   if (const auto *Cast = dyn_cast<CastExpr>(E))
277a7dea167SDimitry Andric     if (Cast->getCastKind() == CK_NoOp ||
278a7dea167SDimitry Andric         Cast->getCastKind() == CK_ConstructorConversion)
279a7dea167SDimitry Andric       E = Cast->getSubExpr()->IgnoreImpCasts();
280a7dea167SDimitry Andric 
2810b57cec5SDimitry Andric   if (const CallExpr *CE = dyn_cast<CallExpr>(E)) {
2820b57cec5SDimitry Andric     if (E->getType()->isVoidType())
2830b57cec5SDimitry Andric       return;
2840b57cec5SDimitry Andric 
285a7dea167SDimitry Andric     if (DiagnoseNoDiscard(*this, cast_or_null<WarnUnusedResultAttr>(
286a7dea167SDimitry Andric                                      CE->getUnusedResultAttr(Context)),
287a7dea167SDimitry Andric                           Loc, R1, R2, /*isCtor=*/false))
2880b57cec5SDimitry Andric       return;
2890b57cec5SDimitry Andric 
2900b57cec5SDimitry Andric     // If the callee has attribute pure, const, or warn_unused_result, warn with
2910b57cec5SDimitry Andric     // a more specific message to make it clear what is happening. If the call
2920b57cec5SDimitry Andric     // is written in a macro body, only warn if it has the warn_unused_result
2930b57cec5SDimitry Andric     // attribute.
2940b57cec5SDimitry Andric     if (const Decl *FD = CE->getCalleeDecl()) {
2950b57cec5SDimitry Andric       if (ShouldSuppress)
2960b57cec5SDimitry Andric         return;
2970b57cec5SDimitry Andric       if (FD->hasAttr<PureAttr>()) {
2980b57cec5SDimitry Andric         Diag(Loc, diag::warn_unused_call) << R1 << R2 << "pure";
2990b57cec5SDimitry Andric         return;
3000b57cec5SDimitry Andric       }
3010b57cec5SDimitry Andric       if (FD->hasAttr<ConstAttr>()) {
3020b57cec5SDimitry Andric         Diag(Loc, diag::warn_unused_call) << R1 << R2 << "const";
3030b57cec5SDimitry Andric         return;
3040b57cec5SDimitry Andric       }
3050b57cec5SDimitry Andric     }
306a7dea167SDimitry Andric   } else if (const auto *CE = dyn_cast<CXXConstructExpr>(E)) {
307a7dea167SDimitry Andric     if (const CXXConstructorDecl *Ctor = CE->getConstructor()) {
308a7dea167SDimitry Andric       const auto *A = Ctor->getAttr<WarnUnusedResultAttr>();
309a7dea167SDimitry Andric       A = A ? A : Ctor->getParent()->getAttr<WarnUnusedResultAttr>();
310a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, A, Loc, R1, R2, /*isCtor=*/true))
311a7dea167SDimitry Andric         return;
312a7dea167SDimitry Andric     }
313a7dea167SDimitry Andric   } else if (const auto *ILE = dyn_cast<InitListExpr>(E)) {
314a7dea167SDimitry Andric     if (const TagDecl *TD = ILE->getType()->getAsTagDecl()) {
315a7dea167SDimitry Andric 
316a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, TD->getAttr<WarnUnusedResultAttr>(), Loc, R1,
317a7dea167SDimitry Andric                             R2, /*isCtor=*/false))
318a7dea167SDimitry Andric         return;
319a7dea167SDimitry Andric     }
3200b57cec5SDimitry Andric   } else if (ShouldSuppress)
3210b57cec5SDimitry Andric     return;
3220b57cec5SDimitry Andric 
323a7dea167SDimitry Andric   E = WarnExpr;
3240b57cec5SDimitry Andric   if (const ObjCMessageExpr *ME = dyn_cast<ObjCMessageExpr>(E)) {
3250b57cec5SDimitry Andric     if (getLangOpts().ObjCAutoRefCount && ME->isDelegateInitCall()) {
3260b57cec5SDimitry Andric       Diag(Loc, diag::err_arc_unused_init_message) << R1;
3270b57cec5SDimitry Andric       return;
3280b57cec5SDimitry Andric     }
3290b57cec5SDimitry Andric     const ObjCMethodDecl *MD = ME->getMethodDecl();
3300b57cec5SDimitry Andric     if (MD) {
331a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, MD->getAttr<WarnUnusedResultAttr>(), Loc, R1,
332a7dea167SDimitry Andric                             R2, /*isCtor=*/false))
3330b57cec5SDimitry Andric         return;
3340b57cec5SDimitry Andric     }
3350b57cec5SDimitry Andric   } else if (const PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E)) {
3360b57cec5SDimitry Andric     const Expr *Source = POE->getSyntacticForm();
3375ffd83dbSDimitry Andric     // Handle the actually selected call of an OpenMP specialized call.
3385ffd83dbSDimitry Andric     if (LangOpts.OpenMP && isa<CallExpr>(Source) &&
3395ffd83dbSDimitry Andric         POE->getNumSemanticExprs() == 1 &&
3405ffd83dbSDimitry Andric         isa<CallExpr>(POE->getSemanticExpr(0)))
341349cc55cSDimitry Andric       return DiagnoseUnusedExprResult(POE->getSemanticExpr(0), DiagID);
3420b57cec5SDimitry Andric     if (isa<ObjCSubscriptRefExpr>(Source))
3430b57cec5SDimitry Andric       DiagID = diag::warn_unused_container_subscript_expr;
34481ad6265SDimitry Andric     else if (isa<ObjCPropertyRefExpr>(Source))
3450b57cec5SDimitry Andric       DiagID = diag::warn_unused_property_expr;
3460b57cec5SDimitry Andric   } else if (const CXXFunctionalCastExpr *FC
3470b57cec5SDimitry Andric                                        = dyn_cast<CXXFunctionalCastExpr>(E)) {
3480b57cec5SDimitry Andric     const Expr *E = FC->getSubExpr();
3490b57cec5SDimitry Andric     if (const CXXBindTemporaryExpr *TE = dyn_cast<CXXBindTemporaryExpr>(E))
3500b57cec5SDimitry Andric       E = TE->getSubExpr();
3510b57cec5SDimitry Andric     if (isa<CXXTemporaryObjectExpr>(E))
3520b57cec5SDimitry Andric       return;
3530b57cec5SDimitry Andric     if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
3540b57cec5SDimitry Andric       if (const CXXRecordDecl *RD = CE->getType()->getAsCXXRecordDecl())
3550b57cec5SDimitry Andric         if (!RD->getAttr<WarnUnusedAttr>())
3560b57cec5SDimitry Andric           return;
3570b57cec5SDimitry Andric   }
3580b57cec5SDimitry Andric   // Diagnose "(void*) blah" as a typo for "(void) blah".
3590b57cec5SDimitry Andric   else if (const CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(E)) {
3600b57cec5SDimitry Andric     TypeSourceInfo *TI = CE->getTypeInfoAsWritten();
3610b57cec5SDimitry Andric     QualType T = TI->getType();
3620b57cec5SDimitry Andric 
3630b57cec5SDimitry Andric     // We really do want to use the non-canonical type here.
3640b57cec5SDimitry Andric     if (T == Context.VoidPtrTy) {
3650b57cec5SDimitry Andric       PointerTypeLoc TL = TI->getTypeLoc().castAs<PointerTypeLoc>();
3660b57cec5SDimitry Andric 
3670b57cec5SDimitry Andric       Diag(Loc, diag::warn_unused_voidptr)
3680b57cec5SDimitry Andric         << FixItHint::CreateRemoval(TL.getStarLoc());
3690b57cec5SDimitry Andric       return;
3700b57cec5SDimitry Andric     }
3710b57cec5SDimitry Andric   }
3720b57cec5SDimitry Andric 
3735ffd83dbSDimitry Andric   // Tell the user to assign it into a variable to force a volatile load if this
3745ffd83dbSDimitry Andric   // isn't an array.
3755ffd83dbSDimitry Andric   if (E->isGLValue() && E->getType().isVolatileQualified() &&
3765ffd83dbSDimitry Andric       !E->getType()->isArrayType()) {
3770b57cec5SDimitry Andric     Diag(Loc, diag::warn_unused_volatile) << R1 << R2;
3780b57cec5SDimitry Andric     return;
3790b57cec5SDimitry Andric   }
3800b57cec5SDimitry Andric 
381349cc55cSDimitry Andric   // Do not diagnose use of a comma operator in a SFINAE context because the
382349cc55cSDimitry Andric   // type of the left operand could be used for SFINAE, so technically it is
383349cc55cSDimitry Andric   // *used*.
384349cc55cSDimitry Andric   if (DiagID != diag::warn_unused_comma_left_operand || !isSFINAEContext())
385349cc55cSDimitry Andric     DiagIfReachable(Loc, S ? llvm::makeArrayRef(S) : llvm::None,
386349cc55cSDimitry Andric                     PDiag(DiagID) << R1 << R2);
3870b57cec5SDimitry Andric }
3880b57cec5SDimitry Andric 
3890b57cec5SDimitry Andric void Sema::ActOnStartOfCompoundStmt(bool IsStmtExpr) {
3900b57cec5SDimitry Andric   PushCompoundScope(IsStmtExpr);
3910b57cec5SDimitry Andric }
3920b57cec5SDimitry Andric 
393e8d8bef9SDimitry Andric void Sema::ActOnAfterCompoundStatementLeadingPragmas() {
394e8d8bef9SDimitry Andric   if (getCurFPFeatures().isFPConstrained()) {
395e8d8bef9SDimitry Andric     FunctionScopeInfo *FSI = getCurFunction();
396e8d8bef9SDimitry Andric     assert(FSI);
397e8d8bef9SDimitry Andric     FSI->setUsesFPIntrin();
398e8d8bef9SDimitry Andric   }
399e8d8bef9SDimitry Andric }
400e8d8bef9SDimitry Andric 
4010b57cec5SDimitry Andric void Sema::ActOnFinishOfCompoundStmt() {
4020b57cec5SDimitry Andric   PopCompoundScope();
4030b57cec5SDimitry Andric }
4040b57cec5SDimitry Andric 
4050b57cec5SDimitry Andric sema::CompoundScopeInfo &Sema::getCurCompoundScope() const {
4060b57cec5SDimitry Andric   return getCurFunction()->CompoundScopes.back();
4070b57cec5SDimitry Andric }
4080b57cec5SDimitry Andric 
4090b57cec5SDimitry Andric StmtResult Sema::ActOnCompoundStmt(SourceLocation L, SourceLocation R,
4100b57cec5SDimitry Andric                                    ArrayRef<Stmt *> Elts, bool isStmtExpr) {
4110b57cec5SDimitry Andric   const unsigned NumElts = Elts.size();
4120b57cec5SDimitry Andric 
41304eeddc0SDimitry Andric   // If we're in C mode, check that we don't have any decls after stmts.  If
41404eeddc0SDimitry Andric   // so, emit an extension diagnostic in C89 and potentially a warning in later
41504eeddc0SDimitry Andric   // versions.
41604eeddc0SDimitry Andric   const unsigned MixedDeclsCodeID = getLangOpts().C99
41704eeddc0SDimitry Andric                                         ? diag::warn_mixed_decls_code
41804eeddc0SDimitry Andric                                         : diag::ext_mixed_decls_code;
41904eeddc0SDimitry Andric   if (!getLangOpts().CPlusPlus && !Diags.isIgnored(MixedDeclsCodeID, L)) {
4200b57cec5SDimitry Andric     // Note that __extension__ can be around a decl.
4210b57cec5SDimitry Andric     unsigned i = 0;
4220b57cec5SDimitry Andric     // Skip over all declarations.
4230b57cec5SDimitry Andric     for (; i != NumElts && isa<DeclStmt>(Elts[i]); ++i)
4240b57cec5SDimitry Andric       /*empty*/;
4250b57cec5SDimitry Andric 
4260b57cec5SDimitry Andric     // We found the end of the list or a statement.  Scan for another declstmt.
4270b57cec5SDimitry Andric     for (; i != NumElts && !isa<DeclStmt>(Elts[i]); ++i)
4280b57cec5SDimitry Andric       /*empty*/;
4290b57cec5SDimitry Andric 
4300b57cec5SDimitry Andric     if (i != NumElts) {
4310b57cec5SDimitry Andric       Decl *D = *cast<DeclStmt>(Elts[i])->decl_begin();
43204eeddc0SDimitry Andric       Diag(D->getLocation(), MixedDeclsCodeID);
4330b57cec5SDimitry Andric     }
4340b57cec5SDimitry Andric   }
4350b57cec5SDimitry Andric 
4360b57cec5SDimitry Andric   // Check for suspicious empty body (null statement) in `for' and `while'
4370b57cec5SDimitry Andric   // statements.  Don't do anything for template instantiations, this just adds
4380b57cec5SDimitry Andric   // noise.
4390b57cec5SDimitry Andric   if (NumElts != 0 && !CurrentInstantiationScope &&
4400b57cec5SDimitry Andric       getCurCompoundScope().HasEmptyLoopBodies) {
4410b57cec5SDimitry Andric     for (unsigned i = 0; i != NumElts - 1; ++i)
4420b57cec5SDimitry Andric       DiagnoseEmptyLoopBody(Elts[i], Elts[i + 1]);
4430b57cec5SDimitry Andric   }
4440b57cec5SDimitry Andric 
44581ad6265SDimitry Andric   // Calculate difference between FP options in this compound statement and in
44681ad6265SDimitry Andric   // the enclosing one. If this is a function body, take the difference against
44781ad6265SDimitry Andric   // default options. In this case the difference will indicate options that are
44881ad6265SDimitry Andric   // changed upon entry to the statement.
44981ad6265SDimitry Andric   FPOptions FPO = (getCurFunction()->CompoundScopes.size() == 1)
45081ad6265SDimitry Andric                       ? FPOptions(getLangOpts())
45181ad6265SDimitry Andric                       : getCurCompoundScope().InitialFPFeatures;
45281ad6265SDimitry Andric   FPOptionsOverride FPDiff = getCurFPFeatures().getChangesFrom(FPO);
45381ad6265SDimitry Andric 
45481ad6265SDimitry Andric   return CompoundStmt::Create(Context, Elts, FPDiff, L, R);
4550b57cec5SDimitry Andric }
4560b57cec5SDimitry Andric 
4570b57cec5SDimitry Andric ExprResult
4580b57cec5SDimitry Andric Sema::ActOnCaseExpr(SourceLocation CaseLoc, ExprResult Val) {
4590b57cec5SDimitry Andric   if (!Val.get())
4600b57cec5SDimitry Andric     return Val;
4610b57cec5SDimitry Andric 
4620b57cec5SDimitry Andric   if (DiagnoseUnexpandedParameterPack(Val.get()))
4630b57cec5SDimitry Andric     return ExprError();
4640b57cec5SDimitry Andric 
4650b57cec5SDimitry Andric   // If we're not inside a switch, let the 'case' statement handling diagnose
4660b57cec5SDimitry Andric   // this. Just clean up after the expression as best we can.
467a7dea167SDimitry Andric   if (getCurFunction()->SwitchStack.empty())
468a7dea167SDimitry Andric     return ActOnFinishFullExpr(Val.get(), Val.get()->getExprLoc(), false,
469a7dea167SDimitry Andric                                getLangOpts().CPlusPlus11);
470a7dea167SDimitry Andric 
4710b57cec5SDimitry Andric   Expr *CondExpr =
4720b57cec5SDimitry Andric       getCurFunction()->SwitchStack.back().getPointer()->getCond();
4730b57cec5SDimitry Andric   if (!CondExpr)
4740b57cec5SDimitry Andric     return ExprError();
4750b57cec5SDimitry Andric   QualType CondType = CondExpr->getType();
4760b57cec5SDimitry Andric 
4770b57cec5SDimitry Andric   auto CheckAndFinish = [&](Expr *E) {
4780b57cec5SDimitry Andric     if (CondType->isDependentType() || E->isTypeDependent())
4790b57cec5SDimitry Andric       return ExprResult(E);
4800b57cec5SDimitry Andric 
4810b57cec5SDimitry Andric     if (getLangOpts().CPlusPlus11) {
4820b57cec5SDimitry Andric       // C++11 [stmt.switch]p2: the constant-expression shall be a converted
4830b57cec5SDimitry Andric       // constant expression of the promoted type of the switch condition.
4840b57cec5SDimitry Andric       llvm::APSInt TempVal;
4850b57cec5SDimitry Andric       return CheckConvertedConstantExpression(E, CondType, TempVal,
4860b57cec5SDimitry Andric                                               CCEK_CaseValue);
4870b57cec5SDimitry Andric     }
4880b57cec5SDimitry Andric 
4890b57cec5SDimitry Andric     ExprResult ER = E;
4900b57cec5SDimitry Andric     if (!E->isValueDependent())
491e8d8bef9SDimitry Andric       ER = VerifyIntegerConstantExpression(E, AllowFold);
4920b57cec5SDimitry Andric     if (!ER.isInvalid())
4930b57cec5SDimitry Andric       ER = DefaultLvalueConversion(ER.get());
4940b57cec5SDimitry Andric     if (!ER.isInvalid())
4950b57cec5SDimitry Andric       ER = ImpCastExprToType(ER.get(), CondType, CK_IntegralCast);
496a7dea167SDimitry Andric     if (!ER.isInvalid())
497a7dea167SDimitry Andric       ER = ActOnFinishFullExpr(ER.get(), ER.get()->getExprLoc(), false);
4980b57cec5SDimitry Andric     return ER;
4990b57cec5SDimitry Andric   };
5000b57cec5SDimitry Andric 
5015ffd83dbSDimitry Andric   ExprResult Converted = CorrectDelayedTyposInExpr(
5025ffd83dbSDimitry Andric       Val, /*InitDecl=*/nullptr, /*RecoverUncorrectedTypos=*/false,
5035ffd83dbSDimitry Andric       CheckAndFinish);
5040b57cec5SDimitry Andric   if (Converted.get() == Val.get())
5050b57cec5SDimitry Andric     Converted = CheckAndFinish(Val.get());
506a7dea167SDimitry Andric   return Converted;
5070b57cec5SDimitry Andric }
5080b57cec5SDimitry Andric 
5090b57cec5SDimitry Andric StmtResult
5100b57cec5SDimitry Andric Sema::ActOnCaseStmt(SourceLocation CaseLoc, ExprResult LHSVal,
5110b57cec5SDimitry Andric                     SourceLocation DotDotDotLoc, ExprResult RHSVal,
5120b57cec5SDimitry Andric                     SourceLocation ColonLoc) {
5130b57cec5SDimitry Andric   assert((LHSVal.isInvalid() || LHSVal.get()) && "missing LHS value");
5140b57cec5SDimitry Andric   assert((DotDotDotLoc.isInvalid() ? RHSVal.isUnset()
5150b57cec5SDimitry Andric                                    : RHSVal.isInvalid() || RHSVal.get()) &&
5160b57cec5SDimitry Andric          "missing RHS value");
5170b57cec5SDimitry Andric 
5180b57cec5SDimitry Andric   if (getCurFunction()->SwitchStack.empty()) {
5190b57cec5SDimitry Andric     Diag(CaseLoc, diag::err_case_not_in_switch);
5200b57cec5SDimitry Andric     return StmtError();
5210b57cec5SDimitry Andric   }
5220b57cec5SDimitry Andric 
5230b57cec5SDimitry Andric   if (LHSVal.isInvalid() || RHSVal.isInvalid()) {
5240b57cec5SDimitry Andric     getCurFunction()->SwitchStack.back().setInt(true);
5250b57cec5SDimitry Andric     return StmtError();
5260b57cec5SDimitry Andric   }
5270b57cec5SDimitry Andric 
5280b57cec5SDimitry Andric   auto *CS = CaseStmt::Create(Context, LHSVal.get(), RHSVal.get(),
5290b57cec5SDimitry Andric                               CaseLoc, DotDotDotLoc, ColonLoc);
5300b57cec5SDimitry Andric   getCurFunction()->SwitchStack.back().getPointer()->addSwitchCase(CS);
5310b57cec5SDimitry Andric   return CS;
5320b57cec5SDimitry Andric }
5330b57cec5SDimitry Andric 
5340b57cec5SDimitry Andric /// ActOnCaseStmtBody - This installs a statement as the body of a case.
5350b57cec5SDimitry Andric void Sema::ActOnCaseStmtBody(Stmt *S, Stmt *SubStmt) {
5360b57cec5SDimitry Andric   cast<CaseStmt>(S)->setSubStmt(SubStmt);
5370b57cec5SDimitry Andric }
5380b57cec5SDimitry Andric 
5390b57cec5SDimitry Andric StmtResult
5400b57cec5SDimitry Andric Sema::ActOnDefaultStmt(SourceLocation DefaultLoc, SourceLocation ColonLoc,
5410b57cec5SDimitry Andric                        Stmt *SubStmt, Scope *CurScope) {
5420b57cec5SDimitry Andric   if (getCurFunction()->SwitchStack.empty()) {
5430b57cec5SDimitry Andric     Diag(DefaultLoc, diag::err_default_not_in_switch);
5440b57cec5SDimitry Andric     return SubStmt;
5450b57cec5SDimitry Andric   }
5460b57cec5SDimitry Andric 
5470b57cec5SDimitry Andric   DefaultStmt *DS = new (Context) DefaultStmt(DefaultLoc, ColonLoc, SubStmt);
5480b57cec5SDimitry Andric   getCurFunction()->SwitchStack.back().getPointer()->addSwitchCase(DS);
5490b57cec5SDimitry Andric   return DS;
5500b57cec5SDimitry Andric }
5510b57cec5SDimitry Andric 
5520b57cec5SDimitry Andric StmtResult
5530b57cec5SDimitry Andric Sema::ActOnLabelStmt(SourceLocation IdentLoc, LabelDecl *TheDecl,
5540b57cec5SDimitry Andric                      SourceLocation ColonLoc, Stmt *SubStmt) {
5550b57cec5SDimitry Andric   // If the label was multiply defined, reject it now.
5560b57cec5SDimitry Andric   if (TheDecl->getStmt()) {
5570b57cec5SDimitry Andric     Diag(IdentLoc, diag::err_redefinition_of_label) << TheDecl->getDeclName();
5580b57cec5SDimitry Andric     Diag(TheDecl->getLocation(), diag::note_previous_definition);
5590b57cec5SDimitry Andric     return SubStmt;
5600b57cec5SDimitry Andric   }
5610b57cec5SDimitry Andric 
562fe6060f1SDimitry Andric   ReservedIdentifierStatus Status = TheDecl->isReserved(getLangOpts());
563349cc55cSDimitry Andric   if (isReservedInAllContexts(Status) &&
564fe6060f1SDimitry Andric       !Context.getSourceManager().isInSystemHeader(IdentLoc))
565fe6060f1SDimitry Andric     Diag(IdentLoc, diag::warn_reserved_extern_symbol)
566fe6060f1SDimitry Andric         << TheDecl << static_cast<int>(Status);
567fe6060f1SDimitry Andric 
5680b57cec5SDimitry Andric   // Otherwise, things are good.  Fill in the declaration and return it.
5690b57cec5SDimitry Andric   LabelStmt *LS = new (Context) LabelStmt(IdentLoc, TheDecl, SubStmt);
5700b57cec5SDimitry Andric   TheDecl->setStmt(LS);
5710b57cec5SDimitry Andric   if (!TheDecl->isGnuLocal()) {
5720b57cec5SDimitry Andric     TheDecl->setLocStart(IdentLoc);
5730b57cec5SDimitry Andric     if (!TheDecl->isMSAsmLabel()) {
5740b57cec5SDimitry Andric       // Don't update the location of MS ASM labels.  These will result in
5750b57cec5SDimitry Andric       // a diagnostic, and changing the location here will mess that up.
5760b57cec5SDimitry Andric       TheDecl->setLocation(IdentLoc);
5770b57cec5SDimitry Andric     }
5780b57cec5SDimitry Andric   }
5790b57cec5SDimitry Andric   return LS;
5800b57cec5SDimitry Andric }
5810b57cec5SDimitry Andric 
582fe6060f1SDimitry Andric StmtResult Sema::BuildAttributedStmt(SourceLocation AttrsLoc,
5830b57cec5SDimitry Andric                                      ArrayRef<const Attr *> Attrs,
5840b57cec5SDimitry Andric                                      Stmt *SubStmt) {
585fe6060f1SDimitry Andric   // FIXME: this code should move when a planned refactoring around statement
586fe6060f1SDimitry Andric   // attributes lands.
587fe6060f1SDimitry Andric   for (const auto *A : Attrs) {
588fe6060f1SDimitry Andric     if (A->getKind() == attr::MustTail) {
589fe6060f1SDimitry Andric       if (!checkAndRewriteMustTailAttr(SubStmt, *A)) {
590fe6060f1SDimitry Andric         return SubStmt;
591fe6060f1SDimitry Andric       }
592fe6060f1SDimitry Andric       setFunctionHasMustTail();
593fe6060f1SDimitry Andric     }
594fe6060f1SDimitry Andric   }
595fe6060f1SDimitry Andric 
596fe6060f1SDimitry Andric   return AttributedStmt::Create(Context, AttrsLoc, Attrs, SubStmt);
597fe6060f1SDimitry Andric }
598fe6060f1SDimitry Andric 
59981ad6265SDimitry Andric StmtResult Sema::ActOnAttributedStmt(const ParsedAttributes &Attrs,
600fe6060f1SDimitry Andric                                      Stmt *SubStmt) {
601fe6060f1SDimitry Andric   SmallVector<const Attr *, 1> SemanticAttrs;
602fe6060f1SDimitry Andric   ProcessStmtAttributes(SubStmt, Attrs, SemanticAttrs);
603fe6060f1SDimitry Andric   if (!SemanticAttrs.empty())
604fe6060f1SDimitry Andric     return BuildAttributedStmt(Attrs.Range.getBegin(), SemanticAttrs, SubStmt);
605fe6060f1SDimitry Andric   // If none of the attributes applied, that's fine, we can recover by
606fe6060f1SDimitry Andric   // returning the substatement directly instead of making an AttributedStmt
607fe6060f1SDimitry Andric   // with no attributes on it.
608fe6060f1SDimitry Andric   return SubStmt;
609fe6060f1SDimitry Andric }
610fe6060f1SDimitry Andric 
611fe6060f1SDimitry Andric bool Sema::checkAndRewriteMustTailAttr(Stmt *St, const Attr &MTA) {
612fe6060f1SDimitry Andric   ReturnStmt *R = cast<ReturnStmt>(St);
613fe6060f1SDimitry Andric   Expr *E = R->getRetValue();
614fe6060f1SDimitry Andric 
615fe6060f1SDimitry Andric   if (CurContext->isDependentContext() || (E && E->isInstantiationDependent()))
616fe6060f1SDimitry Andric     // We have to suspend our check until template instantiation time.
617fe6060f1SDimitry Andric     return true;
618fe6060f1SDimitry Andric 
619fe6060f1SDimitry Andric   if (!checkMustTailAttr(St, MTA))
620fe6060f1SDimitry Andric     return false;
621fe6060f1SDimitry Andric 
622fe6060f1SDimitry Andric   // FIXME: Replace Expr::IgnoreImplicitAsWritten() with this function.
623fe6060f1SDimitry Andric   // Currently it does not skip implicit constructors in an initialization
624fe6060f1SDimitry Andric   // context.
625fe6060f1SDimitry Andric   auto IgnoreImplicitAsWritten = [](Expr *E) -> Expr * {
626fe6060f1SDimitry Andric     return IgnoreExprNodes(E, IgnoreImplicitAsWrittenSingleStep,
627fe6060f1SDimitry Andric                            IgnoreElidableImplicitConstructorSingleStep);
628fe6060f1SDimitry Andric   };
629fe6060f1SDimitry Andric 
630fe6060f1SDimitry Andric   // Now that we have verified that 'musttail' is valid here, rewrite the
631fe6060f1SDimitry Andric   // return value to remove all implicit nodes, but retain parentheses.
632fe6060f1SDimitry Andric   R->setRetValue(IgnoreImplicitAsWritten(E));
633fe6060f1SDimitry Andric   return true;
634fe6060f1SDimitry Andric }
635fe6060f1SDimitry Andric 
636fe6060f1SDimitry Andric bool Sema::checkMustTailAttr(const Stmt *St, const Attr &MTA) {
637fe6060f1SDimitry Andric   assert(!CurContext->isDependentContext() &&
638fe6060f1SDimitry Andric          "musttail cannot be checked from a dependent context");
639fe6060f1SDimitry Andric 
640fe6060f1SDimitry Andric   // FIXME: Add Expr::IgnoreParenImplicitAsWritten() with this definition.
641fe6060f1SDimitry Andric   auto IgnoreParenImplicitAsWritten = [](const Expr *E) -> const Expr * {
642fe6060f1SDimitry Andric     return IgnoreExprNodes(const_cast<Expr *>(E), IgnoreParensSingleStep,
643fe6060f1SDimitry Andric                            IgnoreImplicitAsWrittenSingleStep,
644fe6060f1SDimitry Andric                            IgnoreElidableImplicitConstructorSingleStep);
645fe6060f1SDimitry Andric   };
646fe6060f1SDimitry Andric 
647fe6060f1SDimitry Andric   const Expr *E = cast<ReturnStmt>(St)->getRetValue();
648fe6060f1SDimitry Andric   const auto *CE = dyn_cast_or_null<CallExpr>(IgnoreParenImplicitAsWritten(E));
649fe6060f1SDimitry Andric 
650fe6060f1SDimitry Andric   if (!CE) {
651fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_needs_call) << &MTA;
652fe6060f1SDimitry Andric     return false;
653fe6060f1SDimitry Andric   }
654fe6060f1SDimitry Andric 
655fe6060f1SDimitry Andric   if (const auto *EWC = dyn_cast<ExprWithCleanups>(E)) {
656fe6060f1SDimitry Andric     if (EWC->cleanupsHaveSideEffects()) {
657fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_needs_trivial_args) << &MTA;
658fe6060f1SDimitry Andric       return false;
659fe6060f1SDimitry Andric     }
660fe6060f1SDimitry Andric   }
661fe6060f1SDimitry Andric 
662fe6060f1SDimitry Andric   // We need to determine the full function type (including "this" type, if any)
663fe6060f1SDimitry Andric   // for both caller and callee.
664fe6060f1SDimitry Andric   struct FuncType {
665fe6060f1SDimitry Andric     enum {
666fe6060f1SDimitry Andric       ft_non_member,
667fe6060f1SDimitry Andric       ft_static_member,
668fe6060f1SDimitry Andric       ft_non_static_member,
669fe6060f1SDimitry Andric       ft_pointer_to_member,
670fe6060f1SDimitry Andric     } MemberType = ft_non_member;
671fe6060f1SDimitry Andric 
672fe6060f1SDimitry Andric     QualType This;
673fe6060f1SDimitry Andric     const FunctionProtoType *Func;
674fe6060f1SDimitry Andric     const CXXMethodDecl *Method = nullptr;
675fe6060f1SDimitry Andric   } CallerType, CalleeType;
676fe6060f1SDimitry Andric 
677fe6060f1SDimitry Andric   auto GetMethodType = [this, St, MTA](const CXXMethodDecl *CMD, FuncType &Type,
678fe6060f1SDimitry Andric                                        bool IsCallee) -> bool {
679fe6060f1SDimitry Andric     if (isa<CXXConstructorDecl, CXXDestructorDecl>(CMD)) {
680fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_structors_forbidden)
681fe6060f1SDimitry Andric           << IsCallee << isa<CXXDestructorDecl>(CMD);
682fe6060f1SDimitry Andric       if (IsCallee)
683fe6060f1SDimitry Andric         Diag(CMD->getBeginLoc(), diag::note_musttail_structors_forbidden)
684fe6060f1SDimitry Andric             << isa<CXXDestructorDecl>(CMD);
685fe6060f1SDimitry Andric       Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
686fe6060f1SDimitry Andric       return false;
687fe6060f1SDimitry Andric     }
688fe6060f1SDimitry Andric     if (CMD->isStatic())
689fe6060f1SDimitry Andric       Type.MemberType = FuncType::ft_static_member;
690fe6060f1SDimitry Andric     else {
691fe6060f1SDimitry Andric       Type.This = CMD->getThisType()->getPointeeType();
692fe6060f1SDimitry Andric       Type.MemberType = FuncType::ft_non_static_member;
693fe6060f1SDimitry Andric     }
694fe6060f1SDimitry Andric     Type.Func = CMD->getType()->castAs<FunctionProtoType>();
695fe6060f1SDimitry Andric     return true;
696fe6060f1SDimitry Andric   };
697fe6060f1SDimitry Andric 
698fe6060f1SDimitry Andric   const auto *CallerDecl = dyn_cast<FunctionDecl>(CurContext);
699fe6060f1SDimitry Andric 
700fe6060f1SDimitry Andric   // Find caller function signature.
701fe6060f1SDimitry Andric   if (!CallerDecl) {
702fe6060f1SDimitry Andric     int ContextType;
703fe6060f1SDimitry Andric     if (isa<BlockDecl>(CurContext))
704fe6060f1SDimitry Andric       ContextType = 0;
705fe6060f1SDimitry Andric     else if (isa<ObjCMethodDecl>(CurContext))
706fe6060f1SDimitry Andric       ContextType = 1;
707fe6060f1SDimitry Andric     else
708fe6060f1SDimitry Andric       ContextType = 2;
709fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_forbidden_from_this_context)
710fe6060f1SDimitry Andric         << &MTA << ContextType;
711fe6060f1SDimitry Andric     return false;
712fe6060f1SDimitry Andric   } else if (const auto *CMD = dyn_cast<CXXMethodDecl>(CurContext)) {
713fe6060f1SDimitry Andric     // Caller is a class/struct method.
714fe6060f1SDimitry Andric     if (!GetMethodType(CMD, CallerType, false))
715fe6060f1SDimitry Andric       return false;
716fe6060f1SDimitry Andric   } else {
717fe6060f1SDimitry Andric     // Caller is a non-method function.
718fe6060f1SDimitry Andric     CallerType.Func = CallerDecl->getType()->getAs<FunctionProtoType>();
719fe6060f1SDimitry Andric   }
720fe6060f1SDimitry Andric 
721fe6060f1SDimitry Andric   const Expr *CalleeExpr = CE->getCallee()->IgnoreParens();
722fe6060f1SDimitry Andric   const auto *CalleeBinOp = dyn_cast<BinaryOperator>(CalleeExpr);
723fe6060f1SDimitry Andric   SourceLocation CalleeLoc = CE->getCalleeDecl()
724fe6060f1SDimitry Andric                                  ? CE->getCalleeDecl()->getBeginLoc()
725fe6060f1SDimitry Andric                                  : St->getBeginLoc();
726fe6060f1SDimitry Andric 
727fe6060f1SDimitry Andric   // Find callee function signature.
728fe6060f1SDimitry Andric   if (const CXXMethodDecl *CMD =
729fe6060f1SDimitry Andric           dyn_cast_or_null<CXXMethodDecl>(CE->getCalleeDecl())) {
730fe6060f1SDimitry Andric     // Call is: obj.method(), obj->method(), functor(), etc.
731fe6060f1SDimitry Andric     if (!GetMethodType(CMD, CalleeType, true))
732fe6060f1SDimitry Andric       return false;
733fe6060f1SDimitry Andric   } else if (CalleeBinOp && CalleeBinOp->isPtrMemOp()) {
734fe6060f1SDimitry Andric     // Call is: obj->*method_ptr or obj.*method_ptr
735fe6060f1SDimitry Andric     const auto *MPT =
736fe6060f1SDimitry Andric         CalleeBinOp->getRHS()->getType()->castAs<MemberPointerType>();
737fe6060f1SDimitry Andric     CalleeType.This = QualType(MPT->getClass(), 0);
738fe6060f1SDimitry Andric     CalleeType.Func = MPT->getPointeeType()->castAs<FunctionProtoType>();
739fe6060f1SDimitry Andric     CalleeType.MemberType = FuncType::ft_pointer_to_member;
740fe6060f1SDimitry Andric   } else if (isa<CXXPseudoDestructorExpr>(CalleeExpr)) {
741fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_structors_forbidden)
742fe6060f1SDimitry Andric         << /* IsCallee = */ 1 << /* IsDestructor = */ 1;
743fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
744fe6060f1SDimitry Andric     return false;
745fe6060f1SDimitry Andric   } else {
746fe6060f1SDimitry Andric     // Non-method function.
747fe6060f1SDimitry Andric     CalleeType.Func =
748fe6060f1SDimitry Andric         CalleeExpr->getType()->getPointeeType()->getAs<FunctionProtoType>();
749fe6060f1SDimitry Andric   }
750fe6060f1SDimitry Andric 
751fe6060f1SDimitry Andric   // Both caller and callee must have a prototype (no K&R declarations).
752fe6060f1SDimitry Andric   if (!CalleeType.Func || !CallerType.Func) {
753fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_needs_prototype) << &MTA;
754fe6060f1SDimitry Andric     if (!CalleeType.Func && CE->getDirectCallee()) {
755fe6060f1SDimitry Andric       Diag(CE->getDirectCallee()->getBeginLoc(),
756fe6060f1SDimitry Andric            diag::note_musttail_fix_non_prototype);
757fe6060f1SDimitry Andric     }
758fe6060f1SDimitry Andric     if (!CallerType.Func)
759fe6060f1SDimitry Andric       Diag(CallerDecl->getBeginLoc(), diag::note_musttail_fix_non_prototype);
760fe6060f1SDimitry Andric     return false;
761fe6060f1SDimitry Andric   }
762fe6060f1SDimitry Andric 
763fe6060f1SDimitry Andric   // Caller and callee must have matching calling conventions.
764fe6060f1SDimitry Andric   //
765fe6060f1SDimitry Andric   // Some calling conventions are physically capable of supporting tail calls
766fe6060f1SDimitry Andric   // even if the function types don't perfectly match. LLVM is currently too
767fe6060f1SDimitry Andric   // strict to allow this, but if LLVM added support for this in the future, we
768fe6060f1SDimitry Andric   // could exit early here and skip the remaining checks if the functions are
769fe6060f1SDimitry Andric   // using such a calling convention.
770fe6060f1SDimitry Andric   if (CallerType.Func->getCallConv() != CalleeType.Func->getCallConv()) {
771fe6060f1SDimitry Andric     if (const auto *ND = dyn_cast_or_null<NamedDecl>(CE->getCalleeDecl()))
772fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_callconv_mismatch)
773fe6060f1SDimitry Andric           << true << ND->getDeclName();
774fe6060f1SDimitry Andric     else
775fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_callconv_mismatch) << false;
776fe6060f1SDimitry Andric     Diag(CalleeLoc, diag::note_musttail_callconv_mismatch)
777fe6060f1SDimitry Andric         << FunctionType::getNameForCallConv(CallerType.Func->getCallConv())
778fe6060f1SDimitry Andric         << FunctionType::getNameForCallConv(CalleeType.Func->getCallConv());
779fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
780fe6060f1SDimitry Andric     return false;
781fe6060f1SDimitry Andric   }
782fe6060f1SDimitry Andric 
783fe6060f1SDimitry Andric   if (CalleeType.Func->isVariadic() || CallerType.Func->isVariadic()) {
784fe6060f1SDimitry Andric     Diag(St->getBeginLoc(), diag::err_musttail_no_variadic) << &MTA;
785fe6060f1SDimitry Andric     return false;
786fe6060f1SDimitry Andric   }
787fe6060f1SDimitry Andric 
788fe6060f1SDimitry Andric   // Caller and callee must match in whether they have a "this" parameter.
789fe6060f1SDimitry Andric   if (CallerType.This.isNull() != CalleeType.This.isNull()) {
790fe6060f1SDimitry Andric     if (const auto *ND = dyn_cast_or_null<NamedDecl>(CE->getCalleeDecl())) {
791fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_member_mismatch)
792fe6060f1SDimitry Andric           << CallerType.MemberType << CalleeType.MemberType << true
793fe6060f1SDimitry Andric           << ND->getDeclName();
794fe6060f1SDimitry Andric       Diag(CalleeLoc, diag::note_musttail_callee_defined_here)
795fe6060f1SDimitry Andric           << ND->getDeclName();
796fe6060f1SDimitry Andric     } else
797fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_member_mismatch)
798fe6060f1SDimitry Andric           << CallerType.MemberType << CalleeType.MemberType << false;
799fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
800fe6060f1SDimitry Andric     return false;
801fe6060f1SDimitry Andric   }
802fe6060f1SDimitry Andric 
803fe6060f1SDimitry Andric   auto CheckTypesMatch = [this](FuncType CallerType, FuncType CalleeType,
804fe6060f1SDimitry Andric                                 PartialDiagnostic &PD) -> bool {
805fe6060f1SDimitry Andric     enum {
806fe6060f1SDimitry Andric       ft_different_class,
807fe6060f1SDimitry Andric       ft_parameter_arity,
808fe6060f1SDimitry Andric       ft_parameter_mismatch,
809fe6060f1SDimitry Andric       ft_return_type,
810fe6060f1SDimitry Andric     };
811fe6060f1SDimitry Andric 
812fe6060f1SDimitry Andric     auto DoTypesMatch = [this, &PD](QualType A, QualType B,
813fe6060f1SDimitry Andric                                     unsigned Select) -> bool {
814fe6060f1SDimitry Andric       if (!Context.hasSimilarType(A, B)) {
815fe6060f1SDimitry Andric         PD << Select << A.getUnqualifiedType() << B.getUnqualifiedType();
816fe6060f1SDimitry Andric         return false;
817fe6060f1SDimitry Andric       }
818fe6060f1SDimitry Andric       return true;
819fe6060f1SDimitry Andric     };
820fe6060f1SDimitry Andric 
821fe6060f1SDimitry Andric     if (!CallerType.This.isNull() &&
822fe6060f1SDimitry Andric         !DoTypesMatch(CallerType.This, CalleeType.This, ft_different_class))
823fe6060f1SDimitry Andric       return false;
824fe6060f1SDimitry Andric 
825fe6060f1SDimitry Andric     if (!DoTypesMatch(CallerType.Func->getReturnType(),
826fe6060f1SDimitry Andric                       CalleeType.Func->getReturnType(), ft_return_type))
827fe6060f1SDimitry Andric       return false;
828fe6060f1SDimitry Andric 
829fe6060f1SDimitry Andric     if (CallerType.Func->getNumParams() != CalleeType.Func->getNumParams()) {
830fe6060f1SDimitry Andric       PD << ft_parameter_arity << CallerType.Func->getNumParams()
831fe6060f1SDimitry Andric          << CalleeType.Func->getNumParams();
832fe6060f1SDimitry Andric       return false;
833fe6060f1SDimitry Andric     }
834fe6060f1SDimitry Andric 
835fe6060f1SDimitry Andric     ArrayRef<QualType> CalleeParams = CalleeType.Func->getParamTypes();
836fe6060f1SDimitry Andric     ArrayRef<QualType> CallerParams = CallerType.Func->getParamTypes();
837fe6060f1SDimitry Andric     size_t N = CallerType.Func->getNumParams();
838fe6060f1SDimitry Andric     for (size_t I = 0; I < N; I++) {
839fe6060f1SDimitry Andric       if (!DoTypesMatch(CalleeParams[I], CallerParams[I],
840fe6060f1SDimitry Andric                         ft_parameter_mismatch)) {
841fe6060f1SDimitry Andric         PD << static_cast<int>(I) + 1;
842fe6060f1SDimitry Andric         return false;
843fe6060f1SDimitry Andric       }
844fe6060f1SDimitry Andric     }
845fe6060f1SDimitry Andric 
846fe6060f1SDimitry Andric     return true;
847fe6060f1SDimitry Andric   };
848fe6060f1SDimitry Andric 
849fe6060f1SDimitry Andric   PartialDiagnostic PD = PDiag(diag::note_musttail_mismatch);
850fe6060f1SDimitry Andric   if (!CheckTypesMatch(CallerType, CalleeType, PD)) {
851fe6060f1SDimitry Andric     if (const auto *ND = dyn_cast_or_null<NamedDecl>(CE->getCalleeDecl()))
852fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_mismatch)
853fe6060f1SDimitry Andric           << true << ND->getDeclName();
854fe6060f1SDimitry Andric     else
855fe6060f1SDimitry Andric       Diag(St->getBeginLoc(), diag::err_musttail_mismatch) << false;
856fe6060f1SDimitry Andric     Diag(CalleeLoc, PD);
857fe6060f1SDimitry Andric     Diag(MTA.getLocation(), diag::note_tail_call_required) << &MTA;
858fe6060f1SDimitry Andric     return false;
859fe6060f1SDimitry Andric   }
860fe6060f1SDimitry Andric 
861fe6060f1SDimitry Andric   return true;
8620b57cec5SDimitry Andric }
8630b57cec5SDimitry Andric 
8640b57cec5SDimitry Andric namespace {
8650b57cec5SDimitry Andric class CommaVisitor : public EvaluatedExprVisitor<CommaVisitor> {
8660b57cec5SDimitry Andric   typedef EvaluatedExprVisitor<CommaVisitor> Inherited;
8670b57cec5SDimitry Andric   Sema &SemaRef;
8680b57cec5SDimitry Andric public:
8690b57cec5SDimitry Andric   CommaVisitor(Sema &SemaRef) : Inherited(SemaRef.Context), SemaRef(SemaRef) {}
8700b57cec5SDimitry Andric   void VisitBinaryOperator(BinaryOperator *E) {
8710b57cec5SDimitry Andric     if (E->getOpcode() == BO_Comma)
8720b57cec5SDimitry Andric       SemaRef.DiagnoseCommaOperator(E->getLHS(), E->getExprLoc());
8730b57cec5SDimitry Andric     EvaluatedExprVisitor<CommaVisitor>::VisitBinaryOperator(E);
8740b57cec5SDimitry Andric   }
8750b57cec5SDimitry Andric };
8760b57cec5SDimitry Andric }
8770b57cec5SDimitry Andric 
878349cc55cSDimitry Andric StmtResult Sema::ActOnIfStmt(SourceLocation IfLoc,
879349cc55cSDimitry Andric                              IfStatementKind StatementKind,
880e8d8bef9SDimitry Andric                              SourceLocation LParenLoc, Stmt *InitStmt,
881e8d8bef9SDimitry Andric                              ConditionResult Cond, SourceLocation RParenLoc,
8820b57cec5SDimitry Andric                              Stmt *thenStmt, SourceLocation ElseLoc,
8830b57cec5SDimitry Andric                              Stmt *elseStmt) {
8840b57cec5SDimitry Andric   if (Cond.isInvalid())
88504eeddc0SDimitry Andric     return StmtError();
8860b57cec5SDimitry Andric 
887349cc55cSDimitry Andric   bool ConstevalOrNegatedConsteval =
888349cc55cSDimitry Andric       StatementKind == IfStatementKind::ConstevalNonNegated ||
889349cc55cSDimitry Andric       StatementKind == IfStatementKind::ConstevalNegated;
890349cc55cSDimitry Andric 
8910b57cec5SDimitry Andric   Expr *CondExpr = Cond.get().second;
892349cc55cSDimitry Andric   assert((CondExpr || ConstevalOrNegatedConsteval) &&
893349cc55cSDimitry Andric          "If statement: missing condition");
8940b57cec5SDimitry Andric   // Only call the CommaVisitor when not C89 due to differences in scope flags.
895349cc55cSDimitry Andric   if (CondExpr && (getLangOpts().C99 || getLangOpts().CPlusPlus) &&
8960b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, CondExpr->getExprLoc()))
8970b57cec5SDimitry Andric     CommaVisitor(*this).Visit(CondExpr);
8980b57cec5SDimitry Andric 
899349cc55cSDimitry Andric   if (!ConstevalOrNegatedConsteval && !elseStmt)
90081ad6265SDimitry Andric     DiagnoseEmptyStmtBody(RParenLoc, thenStmt, diag::warn_empty_if_body);
9010b57cec5SDimitry Andric 
902349cc55cSDimitry Andric   if (ConstevalOrNegatedConsteval ||
903349cc55cSDimitry Andric       StatementKind == IfStatementKind::Constexpr) {
904e8d8bef9SDimitry Andric     auto DiagnoseLikelihood = [&](const Stmt *S) {
905e8d8bef9SDimitry Andric       if (const Attr *A = Stmt::getLikelihoodAttr(S)) {
906e8d8bef9SDimitry Andric         Diags.Report(A->getLocation(),
907349cc55cSDimitry Andric                      diag::warn_attribute_has_no_effect_on_compile_time_if)
908349cc55cSDimitry Andric             << A << ConstevalOrNegatedConsteval << A->getRange();
909e8d8bef9SDimitry Andric         Diags.Report(IfLoc,
910349cc55cSDimitry Andric                      diag::note_attribute_has_no_effect_on_compile_time_if_here)
911349cc55cSDimitry Andric             << ConstevalOrNegatedConsteval
912349cc55cSDimitry Andric             << SourceRange(IfLoc, (ConstevalOrNegatedConsteval
913349cc55cSDimitry Andric                                        ? thenStmt->getBeginLoc()
914349cc55cSDimitry Andric                                        : LParenLoc)
915349cc55cSDimitry Andric                                       .getLocWithOffset(-1));
916e8d8bef9SDimitry Andric       }
917e8d8bef9SDimitry Andric     };
918e8d8bef9SDimitry Andric     DiagnoseLikelihood(thenStmt);
919e8d8bef9SDimitry Andric     DiagnoseLikelihood(elseStmt);
920e8d8bef9SDimitry Andric   } else {
921e8d8bef9SDimitry Andric     std::tuple<bool, const Attr *, const Attr *> LHC =
922e8d8bef9SDimitry Andric         Stmt::determineLikelihoodConflict(thenStmt, elseStmt);
923e8d8bef9SDimitry Andric     if (std::get<0>(LHC)) {
924e8d8bef9SDimitry Andric       const Attr *ThenAttr = std::get<1>(LHC);
925e8d8bef9SDimitry Andric       const Attr *ElseAttr = std::get<2>(LHC);
926e8d8bef9SDimitry Andric       Diags.Report(ThenAttr->getLocation(),
927e8d8bef9SDimitry Andric                    diag::warn_attributes_likelihood_ifstmt_conflict)
928e8d8bef9SDimitry Andric           << ThenAttr << ThenAttr->getRange();
929e8d8bef9SDimitry Andric       Diags.Report(ElseAttr->getLocation(), diag::note_conflicting_attribute)
930e8d8bef9SDimitry Andric           << ElseAttr << ElseAttr->getRange();
931e8d8bef9SDimitry Andric     }
932e8d8bef9SDimitry Andric   }
933e8d8bef9SDimitry Andric 
934349cc55cSDimitry Andric   if (ConstevalOrNegatedConsteval) {
935349cc55cSDimitry Andric     bool Immediate = isImmediateFunctionContext();
936349cc55cSDimitry Andric     if (CurContext->isFunctionOrMethod()) {
937349cc55cSDimitry Andric       const auto *FD =
938349cc55cSDimitry Andric           dyn_cast<FunctionDecl>(Decl::castFromDeclContext(CurContext));
939349cc55cSDimitry Andric       if (FD && FD->isConsteval())
940349cc55cSDimitry Andric         Immediate = true;
941349cc55cSDimitry Andric     }
942349cc55cSDimitry Andric     if (isUnevaluatedContext() || Immediate)
943349cc55cSDimitry Andric       Diags.Report(IfLoc, diag::warn_consteval_if_always_true) << Immediate;
944349cc55cSDimitry Andric   }
945349cc55cSDimitry Andric 
946349cc55cSDimitry Andric   return BuildIfStmt(IfLoc, StatementKind, LParenLoc, InitStmt, Cond, RParenLoc,
947e8d8bef9SDimitry Andric                      thenStmt, ElseLoc, elseStmt);
9480b57cec5SDimitry Andric }
9490b57cec5SDimitry Andric 
950349cc55cSDimitry Andric StmtResult Sema::BuildIfStmt(SourceLocation IfLoc,
951349cc55cSDimitry Andric                              IfStatementKind StatementKind,
952e8d8bef9SDimitry Andric                              SourceLocation LParenLoc, Stmt *InitStmt,
953e8d8bef9SDimitry Andric                              ConditionResult Cond, SourceLocation RParenLoc,
9540b57cec5SDimitry Andric                              Stmt *thenStmt, SourceLocation ElseLoc,
9550b57cec5SDimitry Andric                              Stmt *elseStmt) {
9560b57cec5SDimitry Andric   if (Cond.isInvalid())
9570b57cec5SDimitry Andric     return StmtError();
9580b57cec5SDimitry Andric 
959349cc55cSDimitry Andric   if (StatementKind != IfStatementKind::Ordinary ||
960349cc55cSDimitry Andric       isa<ObjCAvailabilityCheckExpr>(Cond.get().second))
9610b57cec5SDimitry Andric     setFunctionHasBranchProtectedScope();
9620b57cec5SDimitry Andric 
963349cc55cSDimitry Andric   return IfStmt::Create(Context, IfLoc, StatementKind, InitStmt,
964349cc55cSDimitry Andric                         Cond.get().first, Cond.get().second, LParenLoc,
965349cc55cSDimitry Andric                         RParenLoc, thenStmt, ElseLoc, elseStmt);
9660b57cec5SDimitry Andric }
9670b57cec5SDimitry Andric 
9680b57cec5SDimitry Andric namespace {
9690b57cec5SDimitry Andric   struct CaseCompareFunctor {
9700b57cec5SDimitry Andric     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
9710b57cec5SDimitry Andric                     const llvm::APSInt &RHS) {
9720b57cec5SDimitry Andric       return LHS.first < RHS;
9730b57cec5SDimitry Andric     }
9740b57cec5SDimitry Andric     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
9750b57cec5SDimitry Andric                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
9760b57cec5SDimitry Andric       return LHS.first < RHS.first;
9770b57cec5SDimitry Andric     }
9780b57cec5SDimitry Andric     bool operator()(const llvm::APSInt &LHS,
9790b57cec5SDimitry Andric                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
9800b57cec5SDimitry Andric       return LHS < RHS.first;
9810b57cec5SDimitry Andric     }
9820b57cec5SDimitry Andric   };
9830b57cec5SDimitry Andric }
9840b57cec5SDimitry Andric 
9850b57cec5SDimitry Andric /// CmpCaseVals - Comparison predicate for sorting case values.
9860b57cec5SDimitry Andric ///
9870b57cec5SDimitry Andric static bool CmpCaseVals(const std::pair<llvm::APSInt, CaseStmt*>& lhs,
9880b57cec5SDimitry Andric                         const std::pair<llvm::APSInt, CaseStmt*>& rhs) {
9890b57cec5SDimitry Andric   if (lhs.first < rhs.first)
9900b57cec5SDimitry Andric     return true;
9910b57cec5SDimitry Andric 
9920b57cec5SDimitry Andric   if (lhs.first == rhs.first &&
993e8d8bef9SDimitry Andric       lhs.second->getCaseLoc() < rhs.second->getCaseLoc())
9940b57cec5SDimitry Andric     return true;
9950b57cec5SDimitry Andric   return false;
9960b57cec5SDimitry Andric }
9970b57cec5SDimitry Andric 
9980b57cec5SDimitry Andric /// CmpEnumVals - Comparison predicate for sorting enumeration values.
9990b57cec5SDimitry Andric ///
10000b57cec5SDimitry Andric static bool CmpEnumVals(const std::pair<llvm::APSInt, EnumConstantDecl*>& lhs,
10010b57cec5SDimitry Andric                         const std::pair<llvm::APSInt, EnumConstantDecl*>& rhs)
10020b57cec5SDimitry Andric {
10030b57cec5SDimitry Andric   return lhs.first < rhs.first;
10040b57cec5SDimitry Andric }
10050b57cec5SDimitry Andric 
10060b57cec5SDimitry Andric /// EqEnumVals - Comparison preficate for uniqing enumeration values.
10070b57cec5SDimitry Andric ///
10080b57cec5SDimitry Andric static bool EqEnumVals(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 /// GetTypeBeforeIntegralPromotion - Returns the pre-promotion type of
10150b57cec5SDimitry Andric /// potentially integral-promoted expression @p expr.
10160b57cec5SDimitry Andric static QualType GetTypeBeforeIntegralPromotion(const Expr *&E) {
10170b57cec5SDimitry Andric   if (const auto *FE = dyn_cast<FullExpr>(E))
10180b57cec5SDimitry Andric     E = FE->getSubExpr();
10190b57cec5SDimitry Andric   while (const auto *ImpCast = dyn_cast<ImplicitCastExpr>(E)) {
10200b57cec5SDimitry Andric     if (ImpCast->getCastKind() != CK_IntegralCast) break;
10210b57cec5SDimitry Andric     E = ImpCast->getSubExpr();
10220b57cec5SDimitry Andric   }
10230b57cec5SDimitry Andric   return E->getType();
10240b57cec5SDimitry Andric }
10250b57cec5SDimitry Andric 
10260b57cec5SDimitry Andric ExprResult Sema::CheckSwitchCondition(SourceLocation SwitchLoc, Expr *Cond) {
10270b57cec5SDimitry Andric   class SwitchConvertDiagnoser : public ICEConvertDiagnoser {
10280b57cec5SDimitry Andric     Expr *Cond;
10290b57cec5SDimitry Andric 
10300b57cec5SDimitry Andric   public:
10310b57cec5SDimitry Andric     SwitchConvertDiagnoser(Expr *Cond)
10320b57cec5SDimitry Andric         : ICEConvertDiagnoser(/*AllowScopedEnumerations*/true, false, true),
10330b57cec5SDimitry Andric           Cond(Cond) {}
10340b57cec5SDimitry Andric 
10350b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseNotInt(Sema &S, SourceLocation Loc,
10360b57cec5SDimitry Andric                                          QualType T) override {
10370b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_typecheck_statement_requires_integer) << T;
10380b57cec5SDimitry Andric     }
10390b57cec5SDimitry Andric 
10400b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseIncomplete(
10410b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T) override {
10420b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_incomplete_class_type)
10430b57cec5SDimitry Andric                << T << Cond->getSourceRange();
10440b57cec5SDimitry Andric     }
10450b57cec5SDimitry Andric 
10460b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseExplicitConv(
10470b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
10480b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_explicit_conversion) << T << ConvTy;
10490b57cec5SDimitry Andric     }
10500b57cec5SDimitry Andric 
10510b57cec5SDimitry Andric     SemaDiagnosticBuilder noteExplicitConv(
10520b57cec5SDimitry Andric         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
10530b57cec5SDimitry Andric       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
10540b57cec5SDimitry Andric         << ConvTy->isEnumeralType() << ConvTy;
10550b57cec5SDimitry Andric     }
10560b57cec5SDimitry Andric 
10570b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseAmbiguous(Sema &S, SourceLocation Loc,
10580b57cec5SDimitry Andric                                             QualType T) override {
10590b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_multiple_conversions) << T;
10600b57cec5SDimitry Andric     }
10610b57cec5SDimitry Andric 
10620b57cec5SDimitry Andric     SemaDiagnosticBuilder noteAmbiguous(
10630b57cec5SDimitry Andric         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
10640b57cec5SDimitry Andric       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
10650b57cec5SDimitry Andric       << ConvTy->isEnumeralType() << ConvTy;
10660b57cec5SDimitry Andric     }
10670b57cec5SDimitry Andric 
10680b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseConversion(
10690b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
10700b57cec5SDimitry Andric       llvm_unreachable("conversion functions are permitted");
10710b57cec5SDimitry Andric     }
10720b57cec5SDimitry Andric   } SwitchDiagnoser(Cond);
10730b57cec5SDimitry Andric 
10740b57cec5SDimitry Andric   ExprResult CondResult =
10750b57cec5SDimitry Andric       PerformContextualImplicitConversion(SwitchLoc, Cond, SwitchDiagnoser);
10760b57cec5SDimitry Andric   if (CondResult.isInvalid())
10770b57cec5SDimitry Andric     return ExprError();
10780b57cec5SDimitry Andric 
10790b57cec5SDimitry Andric   // FIXME: PerformContextualImplicitConversion doesn't always tell us if it
10800b57cec5SDimitry Andric   // failed and produced a diagnostic.
10810b57cec5SDimitry Andric   Cond = CondResult.get();
10820b57cec5SDimitry Andric   if (!Cond->isTypeDependent() &&
10830b57cec5SDimitry Andric       !Cond->getType()->isIntegralOrEnumerationType())
10840b57cec5SDimitry Andric     return ExprError();
10850b57cec5SDimitry Andric 
10860b57cec5SDimitry Andric   // C99 6.8.4.2p5 - Integer promotions are performed on the controlling expr.
10870b57cec5SDimitry Andric   return UsualUnaryConversions(Cond);
10880b57cec5SDimitry Andric }
10890b57cec5SDimitry Andric 
10900b57cec5SDimitry Andric StmtResult Sema::ActOnStartOfSwitchStmt(SourceLocation SwitchLoc,
1091e8d8bef9SDimitry Andric                                         SourceLocation LParenLoc,
1092e8d8bef9SDimitry Andric                                         Stmt *InitStmt, ConditionResult Cond,
1093e8d8bef9SDimitry Andric                                         SourceLocation RParenLoc) {
10940b57cec5SDimitry Andric   Expr *CondExpr = Cond.get().second;
10950b57cec5SDimitry Andric   assert((Cond.isInvalid() || CondExpr) && "switch with no condition");
10960b57cec5SDimitry Andric 
10970b57cec5SDimitry Andric   if (CondExpr && !CondExpr->isTypeDependent()) {
10980b57cec5SDimitry Andric     // We have already converted the expression to an integral or enumeration
10995ffd83dbSDimitry Andric     // type, when we parsed the switch condition. There are cases where we don't
11005ffd83dbSDimitry Andric     // have an appropriate type, e.g. a typo-expr Cond was corrected to an
11015ffd83dbSDimitry Andric     // inappropriate-type expr, we just return an error.
11025ffd83dbSDimitry Andric     if (!CondExpr->getType()->isIntegralOrEnumerationType())
11035ffd83dbSDimitry Andric       return StmtError();
11040b57cec5SDimitry Andric     if (CondExpr->isKnownToHaveBooleanValue()) {
11050b57cec5SDimitry Andric       // switch(bool_expr) {...} is often a programmer error, e.g.
11060b57cec5SDimitry Andric       //   switch(n && mask) { ... }  // Doh - should be "n & mask".
11070b57cec5SDimitry Andric       // One can always use an if statement instead of switch(bool_expr).
11080b57cec5SDimitry Andric       Diag(SwitchLoc, diag::warn_bool_switch_condition)
11090b57cec5SDimitry Andric           << CondExpr->getSourceRange();
11100b57cec5SDimitry Andric     }
11110b57cec5SDimitry Andric   }
11120b57cec5SDimitry Andric 
11130b57cec5SDimitry Andric   setFunctionHasBranchIntoScope();
11140b57cec5SDimitry Andric 
1115e8d8bef9SDimitry Andric   auto *SS = SwitchStmt::Create(Context, InitStmt, Cond.get().first, CondExpr,
1116e8d8bef9SDimitry Andric                                 LParenLoc, RParenLoc);
11170b57cec5SDimitry Andric   getCurFunction()->SwitchStack.push_back(
11180b57cec5SDimitry Andric       FunctionScopeInfo::SwitchInfo(SS, false));
11190b57cec5SDimitry Andric   return SS;
11200b57cec5SDimitry Andric }
11210b57cec5SDimitry Andric 
11220b57cec5SDimitry Andric static void AdjustAPSInt(llvm::APSInt &Val, unsigned BitWidth, bool IsSigned) {
11230b57cec5SDimitry Andric   Val = Val.extOrTrunc(BitWidth);
11240b57cec5SDimitry Andric   Val.setIsSigned(IsSigned);
11250b57cec5SDimitry Andric }
11260b57cec5SDimitry Andric 
11270b57cec5SDimitry Andric /// Check the specified case value is in range for the given unpromoted switch
11280b57cec5SDimitry Andric /// type.
11290b57cec5SDimitry Andric static void checkCaseValue(Sema &S, SourceLocation Loc, const llvm::APSInt &Val,
11300b57cec5SDimitry Andric                            unsigned UnpromotedWidth, bool UnpromotedSign) {
11310b57cec5SDimitry Andric   // In C++11 onwards, this is checked by the language rules.
11320b57cec5SDimitry Andric   if (S.getLangOpts().CPlusPlus11)
11330b57cec5SDimitry Andric     return;
11340b57cec5SDimitry Andric 
11350b57cec5SDimitry Andric   // If the case value was signed and negative and the switch expression is
11360b57cec5SDimitry Andric   // unsigned, don't bother to warn: this is implementation-defined behavior.
11370b57cec5SDimitry Andric   // FIXME: Introduce a second, default-ignored warning for this case?
11380b57cec5SDimitry Andric   if (UnpromotedWidth < Val.getBitWidth()) {
11390b57cec5SDimitry Andric     llvm::APSInt ConvVal(Val);
11400b57cec5SDimitry Andric     AdjustAPSInt(ConvVal, UnpromotedWidth, UnpromotedSign);
11410b57cec5SDimitry Andric     AdjustAPSInt(ConvVal, Val.getBitWidth(), Val.isSigned());
11420b57cec5SDimitry Andric     // FIXME: Use different diagnostics for overflow  in conversion to promoted
11430b57cec5SDimitry Andric     // type versus "switch expression cannot have this value". Use proper
11440b57cec5SDimitry Andric     // IntRange checking rather than just looking at the unpromoted type here.
11450b57cec5SDimitry Andric     if (ConvVal != Val)
1146fe6060f1SDimitry Andric       S.Diag(Loc, diag::warn_case_value_overflow) << toString(Val, 10)
1147fe6060f1SDimitry Andric                                                   << toString(ConvVal, 10);
11480b57cec5SDimitry Andric   }
11490b57cec5SDimitry Andric }
11500b57cec5SDimitry Andric 
11510b57cec5SDimitry Andric typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl*>, 64> EnumValsTy;
11520b57cec5SDimitry Andric 
11530b57cec5SDimitry Andric /// Returns true if we should emit a diagnostic about this case expression not
11540b57cec5SDimitry Andric /// being a part of the enum used in the switch controlling expression.
11550b57cec5SDimitry Andric static bool ShouldDiagnoseSwitchCaseNotInEnum(const Sema &S,
11560b57cec5SDimitry Andric                                               const EnumDecl *ED,
11570b57cec5SDimitry Andric                                               const Expr *CaseExpr,
11580b57cec5SDimitry Andric                                               EnumValsTy::iterator &EI,
11590b57cec5SDimitry Andric                                               EnumValsTy::iterator &EIEnd,
11600b57cec5SDimitry Andric                                               const llvm::APSInt &Val) {
11610b57cec5SDimitry Andric   if (!ED->isClosed())
11620b57cec5SDimitry Andric     return false;
11630b57cec5SDimitry Andric 
11640b57cec5SDimitry Andric   if (const DeclRefExpr *DRE =
11650b57cec5SDimitry Andric           dyn_cast<DeclRefExpr>(CaseExpr->IgnoreParenImpCasts())) {
11660b57cec5SDimitry Andric     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
11670b57cec5SDimitry Andric       QualType VarType = VD->getType();
11680b57cec5SDimitry Andric       QualType EnumType = S.Context.getTypeDeclType(ED);
11690b57cec5SDimitry Andric       if (VD->hasGlobalStorage() && VarType.isConstQualified() &&
11700b57cec5SDimitry Andric           S.Context.hasSameUnqualifiedType(EnumType, VarType))
11710b57cec5SDimitry Andric         return false;
11720b57cec5SDimitry Andric     }
11730b57cec5SDimitry Andric   }
11740b57cec5SDimitry Andric 
11750b57cec5SDimitry Andric   if (ED->hasAttr<FlagEnumAttr>())
11760b57cec5SDimitry Andric     return !S.IsValueInFlagEnum(ED, Val, false);
11770b57cec5SDimitry Andric 
11780b57cec5SDimitry Andric   while (EI != EIEnd && EI->first < Val)
11790b57cec5SDimitry Andric     EI++;
11800b57cec5SDimitry Andric 
11810b57cec5SDimitry Andric   if (EI != EIEnd && EI->first == Val)
11820b57cec5SDimitry Andric     return false;
11830b57cec5SDimitry Andric 
11840b57cec5SDimitry Andric   return true;
11850b57cec5SDimitry Andric }
11860b57cec5SDimitry Andric 
11870b57cec5SDimitry Andric static void checkEnumTypesInSwitchStmt(Sema &S, const Expr *Cond,
11880b57cec5SDimitry Andric                                        const Expr *Case) {
11890b57cec5SDimitry Andric   QualType CondType = Cond->getType();
11900b57cec5SDimitry Andric   QualType CaseType = Case->getType();
11910b57cec5SDimitry Andric 
11920b57cec5SDimitry Andric   const EnumType *CondEnumType = CondType->getAs<EnumType>();
11930b57cec5SDimitry Andric   const EnumType *CaseEnumType = CaseType->getAs<EnumType>();
11940b57cec5SDimitry Andric   if (!CondEnumType || !CaseEnumType)
11950b57cec5SDimitry Andric     return;
11960b57cec5SDimitry Andric 
11970b57cec5SDimitry Andric   // Ignore anonymous enums.
11980b57cec5SDimitry Andric   if (!CondEnumType->getDecl()->getIdentifier() &&
11990b57cec5SDimitry Andric       !CondEnumType->getDecl()->getTypedefNameForAnonDecl())
12000b57cec5SDimitry Andric     return;
12010b57cec5SDimitry Andric   if (!CaseEnumType->getDecl()->getIdentifier() &&
12020b57cec5SDimitry Andric       !CaseEnumType->getDecl()->getTypedefNameForAnonDecl())
12030b57cec5SDimitry Andric     return;
12040b57cec5SDimitry Andric 
12050b57cec5SDimitry Andric   if (S.Context.hasSameUnqualifiedType(CondType, CaseType))
12060b57cec5SDimitry Andric     return;
12070b57cec5SDimitry Andric 
12080b57cec5SDimitry Andric   S.Diag(Case->getExprLoc(), diag::warn_comparison_of_mixed_enum_types_switch)
12090b57cec5SDimitry Andric       << CondType << CaseType << Cond->getSourceRange()
12100b57cec5SDimitry Andric       << Case->getSourceRange();
12110b57cec5SDimitry Andric }
12120b57cec5SDimitry Andric 
12130b57cec5SDimitry Andric StmtResult
12140b57cec5SDimitry Andric Sema::ActOnFinishSwitchStmt(SourceLocation SwitchLoc, Stmt *Switch,
12150b57cec5SDimitry Andric                             Stmt *BodyStmt) {
12160b57cec5SDimitry Andric   SwitchStmt *SS = cast<SwitchStmt>(Switch);
12170b57cec5SDimitry Andric   bool CaseListIsIncomplete = getCurFunction()->SwitchStack.back().getInt();
12180b57cec5SDimitry Andric   assert(SS == getCurFunction()->SwitchStack.back().getPointer() &&
12190b57cec5SDimitry Andric          "switch stack missing push/pop!");
12200b57cec5SDimitry Andric 
12210b57cec5SDimitry Andric   getCurFunction()->SwitchStack.pop_back();
12220b57cec5SDimitry Andric 
12230b57cec5SDimitry Andric   if (!BodyStmt) return StmtError();
12240b57cec5SDimitry Andric   SS->setBody(BodyStmt, SwitchLoc);
12250b57cec5SDimitry Andric 
12260b57cec5SDimitry Andric   Expr *CondExpr = SS->getCond();
12270b57cec5SDimitry Andric   if (!CondExpr) return StmtError();
12280b57cec5SDimitry Andric 
12290b57cec5SDimitry Andric   QualType CondType = CondExpr->getType();
12300b57cec5SDimitry Andric 
12310b57cec5SDimitry Andric   // C++ 6.4.2.p2:
12320b57cec5SDimitry Andric   // Integral promotions are performed (on the switch condition).
12330b57cec5SDimitry Andric   //
12340b57cec5SDimitry Andric   // A case value unrepresentable by the original switch condition
12350b57cec5SDimitry Andric   // type (before the promotion) doesn't make sense, even when it can
12360b57cec5SDimitry Andric   // be represented by the promoted type.  Therefore we need to find
12370b57cec5SDimitry Andric   // the pre-promotion type of the switch condition.
12380b57cec5SDimitry Andric   const Expr *CondExprBeforePromotion = CondExpr;
12390b57cec5SDimitry Andric   QualType CondTypeBeforePromotion =
12400b57cec5SDimitry Andric       GetTypeBeforeIntegralPromotion(CondExprBeforePromotion);
12410b57cec5SDimitry Andric 
12420b57cec5SDimitry Andric   // Get the bitwidth of the switched-on value after promotions. We must
12430b57cec5SDimitry Andric   // convert the integer case values to this width before comparison.
12440b57cec5SDimitry Andric   bool HasDependentValue
12450b57cec5SDimitry Andric     = CondExpr->isTypeDependent() || CondExpr->isValueDependent();
12460b57cec5SDimitry Andric   unsigned CondWidth = HasDependentValue ? 0 : Context.getIntWidth(CondType);
12470b57cec5SDimitry Andric   bool CondIsSigned = CondType->isSignedIntegerOrEnumerationType();
12480b57cec5SDimitry Andric 
12490b57cec5SDimitry Andric   // Get the width and signedness that the condition might actually have, for
12500b57cec5SDimitry Andric   // warning purposes.
12510b57cec5SDimitry Andric   // FIXME: Grab an IntRange for the condition rather than using the unpromoted
12520b57cec5SDimitry Andric   // type.
12530b57cec5SDimitry Andric   unsigned CondWidthBeforePromotion
12540b57cec5SDimitry Andric     = HasDependentValue ? 0 : Context.getIntWidth(CondTypeBeforePromotion);
12550b57cec5SDimitry Andric   bool CondIsSignedBeforePromotion
12560b57cec5SDimitry Andric     = CondTypeBeforePromotion->isSignedIntegerOrEnumerationType();
12570b57cec5SDimitry Andric 
12580b57cec5SDimitry Andric   // Accumulate all of the case values in a vector so that we can sort them
12590b57cec5SDimitry Andric   // and detect duplicates.  This vector contains the APInt for the case after
12600b57cec5SDimitry Andric   // it has been converted to the condition type.
12610b57cec5SDimitry Andric   typedef SmallVector<std::pair<llvm::APSInt, CaseStmt*>, 64> CaseValsTy;
12620b57cec5SDimitry Andric   CaseValsTy CaseVals;
12630b57cec5SDimitry Andric 
12640b57cec5SDimitry Andric   // Keep track of any GNU case ranges we see.  The APSInt is the low value.
12650b57cec5SDimitry Andric   typedef std::vector<std::pair<llvm::APSInt, CaseStmt*> > CaseRangesTy;
12660b57cec5SDimitry Andric   CaseRangesTy CaseRanges;
12670b57cec5SDimitry Andric 
12680b57cec5SDimitry Andric   DefaultStmt *TheDefaultStmt = nullptr;
12690b57cec5SDimitry Andric 
12700b57cec5SDimitry Andric   bool CaseListIsErroneous = false;
12710b57cec5SDimitry Andric 
12720b57cec5SDimitry Andric   for (SwitchCase *SC = SS->getSwitchCaseList(); SC && !HasDependentValue;
12730b57cec5SDimitry Andric        SC = SC->getNextSwitchCase()) {
12740b57cec5SDimitry Andric 
12750b57cec5SDimitry Andric     if (DefaultStmt *DS = dyn_cast<DefaultStmt>(SC)) {
12760b57cec5SDimitry Andric       if (TheDefaultStmt) {
12770b57cec5SDimitry Andric         Diag(DS->getDefaultLoc(), diag::err_multiple_default_labels_defined);
12780b57cec5SDimitry Andric         Diag(TheDefaultStmt->getDefaultLoc(), diag::note_duplicate_case_prev);
12790b57cec5SDimitry Andric 
12800b57cec5SDimitry Andric         // FIXME: Remove the default statement from the switch block so that
12810b57cec5SDimitry Andric         // we'll return a valid AST.  This requires recursing down the AST and
12820b57cec5SDimitry Andric         // finding it, not something we are set up to do right now.  For now,
12830b57cec5SDimitry Andric         // just lop the entire switch stmt out of the AST.
12840b57cec5SDimitry Andric         CaseListIsErroneous = true;
12850b57cec5SDimitry Andric       }
12860b57cec5SDimitry Andric       TheDefaultStmt = DS;
12870b57cec5SDimitry Andric 
12880b57cec5SDimitry Andric     } else {
12890b57cec5SDimitry Andric       CaseStmt *CS = cast<CaseStmt>(SC);
12900b57cec5SDimitry Andric 
12910b57cec5SDimitry Andric       Expr *Lo = CS->getLHS();
12920b57cec5SDimitry Andric 
12930b57cec5SDimitry Andric       if (Lo->isValueDependent()) {
12940b57cec5SDimitry Andric         HasDependentValue = true;
12950b57cec5SDimitry Andric         break;
12960b57cec5SDimitry Andric       }
12970b57cec5SDimitry Andric 
12980b57cec5SDimitry Andric       // We already verified that the expression has a constant value;
12990b57cec5SDimitry Andric       // get that value (prior to conversions).
13000b57cec5SDimitry Andric       const Expr *LoBeforePromotion = Lo;
13010b57cec5SDimitry Andric       GetTypeBeforeIntegralPromotion(LoBeforePromotion);
13020b57cec5SDimitry Andric       llvm::APSInt LoVal = LoBeforePromotion->EvaluateKnownConstInt(Context);
13030b57cec5SDimitry Andric 
13040b57cec5SDimitry Andric       // Check the unconverted value is within the range of possible values of
13050b57cec5SDimitry Andric       // the switch expression.
13060b57cec5SDimitry Andric       checkCaseValue(*this, Lo->getBeginLoc(), LoVal, CondWidthBeforePromotion,
13070b57cec5SDimitry Andric                      CondIsSignedBeforePromotion);
13080b57cec5SDimitry Andric 
13090b57cec5SDimitry Andric       // FIXME: This duplicates the check performed for warn_not_in_enum below.
13100b57cec5SDimitry Andric       checkEnumTypesInSwitchStmt(*this, CondExprBeforePromotion,
13110b57cec5SDimitry Andric                                  LoBeforePromotion);
13120b57cec5SDimitry Andric 
13130b57cec5SDimitry Andric       // Convert the value to the same width/sign as the condition.
13140b57cec5SDimitry Andric       AdjustAPSInt(LoVal, CondWidth, CondIsSigned);
13150b57cec5SDimitry Andric 
13160b57cec5SDimitry Andric       // If this is a case range, remember it in CaseRanges, otherwise CaseVals.
13170b57cec5SDimitry Andric       if (CS->getRHS()) {
13180b57cec5SDimitry Andric         if (CS->getRHS()->isValueDependent()) {
13190b57cec5SDimitry Andric           HasDependentValue = true;
13200b57cec5SDimitry Andric           break;
13210b57cec5SDimitry Andric         }
13220b57cec5SDimitry Andric         CaseRanges.push_back(std::make_pair(LoVal, CS));
13230b57cec5SDimitry Andric       } else
13240b57cec5SDimitry Andric         CaseVals.push_back(std::make_pair(LoVal, CS));
13250b57cec5SDimitry Andric     }
13260b57cec5SDimitry Andric   }
13270b57cec5SDimitry Andric 
13280b57cec5SDimitry Andric   if (!HasDependentValue) {
13290b57cec5SDimitry Andric     // If we don't have a default statement, check whether the
13300b57cec5SDimitry Andric     // condition is constant.
13310b57cec5SDimitry Andric     llvm::APSInt ConstantCondValue;
13320b57cec5SDimitry Andric     bool HasConstantCond = false;
1333a7dea167SDimitry Andric     if (!TheDefaultStmt) {
13340b57cec5SDimitry Andric       Expr::EvalResult Result;
13350b57cec5SDimitry Andric       HasConstantCond = CondExpr->EvaluateAsInt(Result, Context,
13360b57cec5SDimitry Andric                                                 Expr::SE_AllowSideEffects);
13370b57cec5SDimitry Andric       if (Result.Val.isInt())
13380b57cec5SDimitry Andric         ConstantCondValue = Result.Val.getInt();
13390b57cec5SDimitry Andric       assert(!HasConstantCond ||
13400b57cec5SDimitry Andric              (ConstantCondValue.getBitWidth() == CondWidth &&
13410b57cec5SDimitry Andric               ConstantCondValue.isSigned() == CondIsSigned));
13420b57cec5SDimitry Andric     }
13430b57cec5SDimitry Andric     bool ShouldCheckConstantCond = HasConstantCond;
13440b57cec5SDimitry Andric 
13450b57cec5SDimitry Andric     // Sort all the scalar case values so we can easily detect duplicates.
13460b57cec5SDimitry Andric     llvm::stable_sort(CaseVals, CmpCaseVals);
13470b57cec5SDimitry Andric 
13480b57cec5SDimitry Andric     if (!CaseVals.empty()) {
13490b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseVals.size(); i != e; ++i) {
13500b57cec5SDimitry Andric         if (ShouldCheckConstantCond &&
13510b57cec5SDimitry Andric             CaseVals[i].first == ConstantCondValue)
13520b57cec5SDimitry Andric           ShouldCheckConstantCond = false;
13530b57cec5SDimitry Andric 
13540b57cec5SDimitry Andric         if (i != 0 && CaseVals[i].first == CaseVals[i-1].first) {
13550b57cec5SDimitry Andric           // If we have a duplicate, report it.
13560b57cec5SDimitry Andric           // First, determine if either case value has a name
13570b57cec5SDimitry Andric           StringRef PrevString, CurrString;
13580b57cec5SDimitry Andric           Expr *PrevCase = CaseVals[i-1].second->getLHS()->IgnoreParenCasts();
13590b57cec5SDimitry Andric           Expr *CurrCase = CaseVals[i].second->getLHS()->IgnoreParenCasts();
13600b57cec5SDimitry Andric           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(PrevCase)) {
13610b57cec5SDimitry Andric             PrevString = DeclRef->getDecl()->getName();
13620b57cec5SDimitry Andric           }
13630b57cec5SDimitry Andric           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(CurrCase)) {
13640b57cec5SDimitry Andric             CurrString = DeclRef->getDecl()->getName();
13650b57cec5SDimitry Andric           }
13660b57cec5SDimitry Andric           SmallString<16> CaseValStr;
13670b57cec5SDimitry Andric           CaseVals[i-1].first.toString(CaseValStr);
13680b57cec5SDimitry Andric 
13690b57cec5SDimitry Andric           if (PrevString == CurrString)
13700b57cec5SDimitry Andric             Diag(CaseVals[i].second->getLHS()->getBeginLoc(),
13710b57cec5SDimitry Andric                  diag::err_duplicate_case)
1372fe6060f1SDimitry Andric                 << (PrevString.empty() ? CaseValStr.str() : PrevString);
13730b57cec5SDimitry Andric           else
13740b57cec5SDimitry Andric             Diag(CaseVals[i].second->getLHS()->getBeginLoc(),
13750b57cec5SDimitry Andric                  diag::err_duplicate_case_differing_expr)
1376fe6060f1SDimitry Andric                 << (PrevString.empty() ? CaseValStr.str() : PrevString)
1377fe6060f1SDimitry Andric                 << (CurrString.empty() ? CaseValStr.str() : CurrString)
13780b57cec5SDimitry Andric                 << CaseValStr;
13790b57cec5SDimitry Andric 
13800b57cec5SDimitry Andric           Diag(CaseVals[i - 1].second->getLHS()->getBeginLoc(),
13810b57cec5SDimitry Andric                diag::note_duplicate_case_prev);
13820b57cec5SDimitry Andric           // FIXME: We really want to remove the bogus case stmt from the
13830b57cec5SDimitry Andric           // substmt, but we have no way to do this right now.
13840b57cec5SDimitry Andric           CaseListIsErroneous = true;
13850b57cec5SDimitry Andric         }
13860b57cec5SDimitry Andric       }
13870b57cec5SDimitry Andric     }
13880b57cec5SDimitry Andric 
13890b57cec5SDimitry Andric     // Detect duplicate case ranges, which usually don't exist at all in
13900b57cec5SDimitry Andric     // the first place.
13910b57cec5SDimitry Andric     if (!CaseRanges.empty()) {
13920b57cec5SDimitry Andric       // Sort all the case ranges by their low value so we can easily detect
13930b57cec5SDimitry Andric       // overlaps between ranges.
13940b57cec5SDimitry Andric       llvm::stable_sort(CaseRanges);
13950b57cec5SDimitry Andric 
13960b57cec5SDimitry Andric       // Scan the ranges, computing the high values and removing empty ranges.
13970b57cec5SDimitry Andric       std::vector<llvm::APSInt> HiVals;
13980b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
13990b57cec5SDimitry Andric         llvm::APSInt &LoVal = CaseRanges[i].first;
14000b57cec5SDimitry Andric         CaseStmt *CR = CaseRanges[i].second;
14010b57cec5SDimitry Andric         Expr *Hi = CR->getRHS();
14020b57cec5SDimitry Andric 
14030b57cec5SDimitry Andric         const Expr *HiBeforePromotion = Hi;
14040b57cec5SDimitry Andric         GetTypeBeforeIntegralPromotion(HiBeforePromotion);
14050b57cec5SDimitry Andric         llvm::APSInt HiVal = HiBeforePromotion->EvaluateKnownConstInt(Context);
14060b57cec5SDimitry Andric 
14070b57cec5SDimitry Andric         // Check the unconverted value is within the range of possible values of
14080b57cec5SDimitry Andric         // the switch expression.
14090b57cec5SDimitry Andric         checkCaseValue(*this, Hi->getBeginLoc(), HiVal,
14100b57cec5SDimitry Andric                        CondWidthBeforePromotion, CondIsSignedBeforePromotion);
14110b57cec5SDimitry Andric 
14120b57cec5SDimitry Andric         // Convert the value to the same width/sign as the condition.
14130b57cec5SDimitry Andric         AdjustAPSInt(HiVal, CondWidth, CondIsSigned);
14140b57cec5SDimitry Andric 
14150b57cec5SDimitry Andric         // If the low value is bigger than the high value, the case is empty.
14160b57cec5SDimitry Andric         if (LoVal > HiVal) {
14170b57cec5SDimitry Andric           Diag(CR->getLHS()->getBeginLoc(), diag::warn_case_empty_range)
14180b57cec5SDimitry Andric               << SourceRange(CR->getLHS()->getBeginLoc(), Hi->getEndLoc());
14190b57cec5SDimitry Andric           CaseRanges.erase(CaseRanges.begin()+i);
14200b57cec5SDimitry Andric           --i;
14210b57cec5SDimitry Andric           --e;
14220b57cec5SDimitry Andric           continue;
14230b57cec5SDimitry Andric         }
14240b57cec5SDimitry Andric 
14250b57cec5SDimitry Andric         if (ShouldCheckConstantCond &&
14260b57cec5SDimitry Andric             LoVal <= ConstantCondValue &&
14270b57cec5SDimitry Andric             ConstantCondValue <= HiVal)
14280b57cec5SDimitry Andric           ShouldCheckConstantCond = false;
14290b57cec5SDimitry Andric 
14300b57cec5SDimitry Andric         HiVals.push_back(HiVal);
14310b57cec5SDimitry Andric       }
14320b57cec5SDimitry Andric 
14330b57cec5SDimitry Andric       // Rescan the ranges, looking for overlap with singleton values and other
14340b57cec5SDimitry Andric       // ranges.  Since the range list is sorted, we only need to compare case
14350b57cec5SDimitry Andric       // ranges with their neighbors.
14360b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
14370b57cec5SDimitry Andric         llvm::APSInt &CRLo = CaseRanges[i].first;
14380b57cec5SDimitry Andric         llvm::APSInt &CRHi = HiVals[i];
14390b57cec5SDimitry Andric         CaseStmt *CR = CaseRanges[i].second;
14400b57cec5SDimitry Andric 
14410b57cec5SDimitry Andric         // Check to see whether the case range overlaps with any
14420b57cec5SDimitry Andric         // singleton cases.
14430b57cec5SDimitry Andric         CaseStmt *OverlapStmt = nullptr;
14440b57cec5SDimitry Andric         llvm::APSInt OverlapVal(32);
14450b57cec5SDimitry Andric 
14460b57cec5SDimitry Andric         // Find the smallest value >= the lower bound.  If I is in the
14470b57cec5SDimitry Andric         // case range, then we have overlap.
14480b57cec5SDimitry Andric         CaseValsTy::iterator I =
14490b57cec5SDimitry Andric             llvm::lower_bound(CaseVals, CRLo, CaseCompareFunctor());
14500b57cec5SDimitry Andric         if (I != CaseVals.end() && I->first < CRHi) {
14510b57cec5SDimitry Andric           OverlapVal  = I->first;   // Found overlap with scalar.
14520b57cec5SDimitry Andric           OverlapStmt = I->second;
14530b57cec5SDimitry Andric         }
14540b57cec5SDimitry Andric 
14550b57cec5SDimitry Andric         // Find the smallest value bigger than the upper bound.
14560b57cec5SDimitry Andric         I = std::upper_bound(I, CaseVals.end(), CRHi, CaseCompareFunctor());
14570b57cec5SDimitry Andric         if (I != CaseVals.begin() && (I-1)->first >= CRLo) {
14580b57cec5SDimitry Andric           OverlapVal  = (I-1)->first;      // Found overlap with scalar.
14590b57cec5SDimitry Andric           OverlapStmt = (I-1)->second;
14600b57cec5SDimitry Andric         }
14610b57cec5SDimitry Andric 
14620b57cec5SDimitry Andric         // Check to see if this case stmt overlaps with the subsequent
14630b57cec5SDimitry Andric         // case range.
14640b57cec5SDimitry Andric         if (i && CRLo <= HiVals[i-1]) {
14650b57cec5SDimitry Andric           OverlapVal  = HiVals[i-1];       // Found overlap with range.
14660b57cec5SDimitry Andric           OverlapStmt = CaseRanges[i-1].second;
14670b57cec5SDimitry Andric         }
14680b57cec5SDimitry Andric 
14690b57cec5SDimitry Andric         if (OverlapStmt) {
14700b57cec5SDimitry Andric           // If we have a duplicate, report it.
14710b57cec5SDimitry Andric           Diag(CR->getLHS()->getBeginLoc(), diag::err_duplicate_case)
1472fe6060f1SDimitry Andric               << toString(OverlapVal, 10);
14730b57cec5SDimitry Andric           Diag(OverlapStmt->getLHS()->getBeginLoc(),
14740b57cec5SDimitry Andric                diag::note_duplicate_case_prev);
14750b57cec5SDimitry Andric           // FIXME: We really want to remove the bogus case stmt from the
14760b57cec5SDimitry Andric           // substmt, but we have no way to do this right now.
14770b57cec5SDimitry Andric           CaseListIsErroneous = true;
14780b57cec5SDimitry Andric         }
14790b57cec5SDimitry Andric       }
14800b57cec5SDimitry Andric     }
14810b57cec5SDimitry Andric 
14820b57cec5SDimitry Andric     // Complain if we have a constant condition and we didn't find a match.
14830b57cec5SDimitry Andric     if (!CaseListIsErroneous && !CaseListIsIncomplete &&
14840b57cec5SDimitry Andric         ShouldCheckConstantCond) {
14850b57cec5SDimitry Andric       // TODO: it would be nice if we printed enums as enums, chars as
14860b57cec5SDimitry Andric       // chars, etc.
14870b57cec5SDimitry Andric       Diag(CondExpr->getExprLoc(), diag::warn_missing_case_for_condition)
1488fe6060f1SDimitry Andric         << toString(ConstantCondValue, 10)
14890b57cec5SDimitry Andric         << CondExpr->getSourceRange();
14900b57cec5SDimitry Andric     }
14910b57cec5SDimitry Andric 
14920b57cec5SDimitry Andric     // Check to see if switch is over an Enum and handles all of its
14930b57cec5SDimitry Andric     // values.  We only issue a warning if there is not 'default:', but
14940b57cec5SDimitry Andric     // we still do the analysis to preserve this information in the AST
14950b57cec5SDimitry Andric     // (which can be used by flow-based analyes).
14960b57cec5SDimitry Andric     //
14970b57cec5SDimitry Andric     const EnumType *ET = CondTypeBeforePromotion->getAs<EnumType>();
14980b57cec5SDimitry Andric 
14990b57cec5SDimitry Andric     // If switch has default case, then ignore it.
15000b57cec5SDimitry Andric     if (!CaseListIsErroneous && !CaseListIsIncomplete && !HasConstantCond &&
1501fe6060f1SDimitry Andric         ET && ET->getDecl()->isCompleteDefinition() &&
1502fe6060f1SDimitry Andric         !empty(ET->getDecl()->enumerators())) {
15030b57cec5SDimitry Andric       const EnumDecl *ED = ET->getDecl();
15040b57cec5SDimitry Andric       EnumValsTy EnumVals;
15050b57cec5SDimitry Andric 
15060b57cec5SDimitry Andric       // Gather all enum values, set their type and sort them,
15070b57cec5SDimitry Andric       // allowing easier comparison with CaseVals.
15080b57cec5SDimitry Andric       for (auto *EDI : ED->enumerators()) {
15090b57cec5SDimitry Andric         llvm::APSInt Val = EDI->getInitVal();
15100b57cec5SDimitry Andric         AdjustAPSInt(Val, CondWidth, CondIsSigned);
15110b57cec5SDimitry Andric         EnumVals.push_back(std::make_pair(Val, EDI));
15120b57cec5SDimitry Andric       }
15130b57cec5SDimitry Andric       llvm::stable_sort(EnumVals, CmpEnumVals);
15140b57cec5SDimitry Andric       auto EI = EnumVals.begin(), EIEnd =
15150b57cec5SDimitry Andric         std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
15160b57cec5SDimitry Andric 
15170b57cec5SDimitry Andric       // See which case values aren't in enum.
15180b57cec5SDimitry Andric       for (CaseValsTy::const_iterator CI = CaseVals.begin();
15190b57cec5SDimitry Andric           CI != CaseVals.end(); CI++) {
15200b57cec5SDimitry Andric         Expr *CaseExpr = CI->second->getLHS();
15210b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
15220b57cec5SDimitry Andric                                               CI->first))
15230b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
15240b57cec5SDimitry Andric             << CondTypeBeforePromotion;
15250b57cec5SDimitry Andric       }
15260b57cec5SDimitry Andric 
15270b57cec5SDimitry Andric       // See which of case ranges aren't in enum
15280b57cec5SDimitry Andric       EI = EnumVals.begin();
15290b57cec5SDimitry Andric       for (CaseRangesTy::const_iterator RI = CaseRanges.begin();
15300b57cec5SDimitry Andric           RI != CaseRanges.end(); RI++) {
15310b57cec5SDimitry Andric         Expr *CaseExpr = RI->second->getLHS();
15320b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
15330b57cec5SDimitry Andric                                               RI->first))
15340b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
15350b57cec5SDimitry Andric             << CondTypeBeforePromotion;
15360b57cec5SDimitry Andric 
15370b57cec5SDimitry Andric         llvm::APSInt Hi =
15380b57cec5SDimitry Andric           RI->second->getRHS()->EvaluateKnownConstInt(Context);
15390b57cec5SDimitry Andric         AdjustAPSInt(Hi, CondWidth, CondIsSigned);
15400b57cec5SDimitry Andric 
15410b57cec5SDimitry Andric         CaseExpr = RI->second->getRHS();
15420b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
15430b57cec5SDimitry Andric                                               Hi))
15440b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
15450b57cec5SDimitry Andric             << CondTypeBeforePromotion;
15460b57cec5SDimitry Andric       }
15470b57cec5SDimitry Andric 
15480b57cec5SDimitry Andric       // Check which enum vals aren't in switch
15490b57cec5SDimitry Andric       auto CI = CaseVals.begin();
15500b57cec5SDimitry Andric       auto RI = CaseRanges.begin();
15510b57cec5SDimitry Andric       bool hasCasesNotInSwitch = false;
15520b57cec5SDimitry Andric 
15530b57cec5SDimitry Andric       SmallVector<DeclarationName,8> UnhandledNames;
15540b57cec5SDimitry Andric 
15550b57cec5SDimitry Andric       for (EI = EnumVals.begin(); EI != EIEnd; EI++) {
15560b57cec5SDimitry Andric         // Don't warn about omitted unavailable EnumConstantDecls.
15570b57cec5SDimitry Andric         switch (EI->second->getAvailability()) {
15580b57cec5SDimitry Andric         case AR_Deprecated:
15590b57cec5SDimitry Andric           // Omitting a deprecated constant is ok; it should never materialize.
15600b57cec5SDimitry Andric         case AR_Unavailable:
15610b57cec5SDimitry Andric           continue;
15620b57cec5SDimitry Andric 
15630b57cec5SDimitry Andric         case AR_NotYetIntroduced:
15640b57cec5SDimitry Andric           // Partially available enum constants should be present. Note that we
15650b57cec5SDimitry Andric           // suppress -Wunguarded-availability diagnostics for such uses.
15660b57cec5SDimitry Andric         case AR_Available:
15670b57cec5SDimitry Andric           break;
15680b57cec5SDimitry Andric         }
15690b57cec5SDimitry Andric 
15700b57cec5SDimitry Andric         if (EI->second->hasAttr<UnusedAttr>())
15710b57cec5SDimitry Andric           continue;
15720b57cec5SDimitry Andric 
15730b57cec5SDimitry Andric         // Drop unneeded case values
15740b57cec5SDimitry Andric         while (CI != CaseVals.end() && CI->first < EI->first)
15750b57cec5SDimitry Andric           CI++;
15760b57cec5SDimitry Andric 
15770b57cec5SDimitry Andric         if (CI != CaseVals.end() && CI->first == EI->first)
15780b57cec5SDimitry Andric           continue;
15790b57cec5SDimitry Andric 
15800b57cec5SDimitry Andric         // Drop unneeded case ranges
15810b57cec5SDimitry Andric         for (; RI != CaseRanges.end(); RI++) {
15820b57cec5SDimitry Andric           llvm::APSInt Hi =
15830b57cec5SDimitry Andric             RI->second->getRHS()->EvaluateKnownConstInt(Context);
15840b57cec5SDimitry Andric           AdjustAPSInt(Hi, CondWidth, CondIsSigned);
15850b57cec5SDimitry Andric           if (EI->first <= Hi)
15860b57cec5SDimitry Andric             break;
15870b57cec5SDimitry Andric         }
15880b57cec5SDimitry Andric 
15890b57cec5SDimitry Andric         if (RI == CaseRanges.end() || EI->first < RI->first) {
15900b57cec5SDimitry Andric           hasCasesNotInSwitch = true;
15910b57cec5SDimitry Andric           UnhandledNames.push_back(EI->second->getDeclName());
15920b57cec5SDimitry Andric         }
15930b57cec5SDimitry Andric       }
15940b57cec5SDimitry Andric 
15950b57cec5SDimitry Andric       if (TheDefaultStmt && UnhandledNames.empty() && ED->isClosedNonFlag())
15960b57cec5SDimitry Andric         Diag(TheDefaultStmt->getDefaultLoc(), diag::warn_unreachable_default);
15970b57cec5SDimitry Andric 
15980b57cec5SDimitry Andric       // Produce a nice diagnostic if multiple values aren't handled.
15990b57cec5SDimitry Andric       if (!UnhandledNames.empty()) {
1600e8d8bef9SDimitry Andric         auto DB = Diag(CondExpr->getExprLoc(), TheDefaultStmt
1601e8d8bef9SDimitry Andric                                                    ? diag::warn_def_missing_case
16020b57cec5SDimitry Andric                                                    : diag::warn_missing_case)
1603349cc55cSDimitry Andric                   << CondExpr->getSourceRange() << (int)UnhandledNames.size();
16040b57cec5SDimitry Andric 
16050b57cec5SDimitry Andric         for (size_t I = 0, E = std::min(UnhandledNames.size(), (size_t)3);
16060b57cec5SDimitry Andric              I != E; ++I)
16070b57cec5SDimitry Andric           DB << UnhandledNames[I];
16080b57cec5SDimitry Andric       }
16090b57cec5SDimitry Andric 
16100b57cec5SDimitry Andric       if (!hasCasesNotInSwitch)
16110b57cec5SDimitry Andric         SS->setAllEnumCasesCovered();
16120b57cec5SDimitry Andric     }
16130b57cec5SDimitry Andric   }
16140b57cec5SDimitry Andric 
16150b57cec5SDimitry Andric   if (BodyStmt)
16160b57cec5SDimitry Andric     DiagnoseEmptyStmtBody(CondExpr->getEndLoc(), BodyStmt,
16170b57cec5SDimitry Andric                           diag::warn_empty_switch_body);
16180b57cec5SDimitry Andric 
16190b57cec5SDimitry Andric   // FIXME: If the case list was broken is some way, we don't have a good system
16200b57cec5SDimitry Andric   // to patch it up.  Instead, just return the whole substmt as broken.
16210b57cec5SDimitry Andric   if (CaseListIsErroneous)
16220b57cec5SDimitry Andric     return StmtError();
16230b57cec5SDimitry Andric 
16240b57cec5SDimitry Andric   return SS;
16250b57cec5SDimitry Andric }
16260b57cec5SDimitry Andric 
16270b57cec5SDimitry Andric void
16280b57cec5SDimitry Andric Sema::DiagnoseAssignmentEnum(QualType DstType, QualType SrcType,
16290b57cec5SDimitry Andric                              Expr *SrcExpr) {
16300b57cec5SDimitry Andric   if (Diags.isIgnored(diag::warn_not_in_enum_assignment, SrcExpr->getExprLoc()))
16310b57cec5SDimitry Andric     return;
16320b57cec5SDimitry Andric 
16330b57cec5SDimitry Andric   if (const EnumType *ET = DstType->getAs<EnumType>())
16340b57cec5SDimitry Andric     if (!Context.hasSameUnqualifiedType(SrcType, DstType) &&
16350b57cec5SDimitry Andric         SrcType->isIntegerType()) {
16360b57cec5SDimitry Andric       if (!SrcExpr->isTypeDependent() && !SrcExpr->isValueDependent() &&
16370b57cec5SDimitry Andric           SrcExpr->isIntegerConstantExpr(Context)) {
16380b57cec5SDimitry Andric         // Get the bitwidth of the enum value before promotions.
16390b57cec5SDimitry Andric         unsigned DstWidth = Context.getIntWidth(DstType);
16400b57cec5SDimitry Andric         bool DstIsSigned = DstType->isSignedIntegerOrEnumerationType();
16410b57cec5SDimitry Andric 
16420b57cec5SDimitry Andric         llvm::APSInt RhsVal = SrcExpr->EvaluateKnownConstInt(Context);
16430b57cec5SDimitry Andric         AdjustAPSInt(RhsVal, DstWidth, DstIsSigned);
16440b57cec5SDimitry Andric         const EnumDecl *ED = ET->getDecl();
16450b57cec5SDimitry Andric 
16460b57cec5SDimitry Andric         if (!ED->isClosed())
16470b57cec5SDimitry Andric           return;
16480b57cec5SDimitry Andric 
16490b57cec5SDimitry Andric         if (ED->hasAttr<FlagEnumAttr>()) {
16500b57cec5SDimitry Andric           if (!IsValueInFlagEnum(ED, RhsVal, true))
16510b57cec5SDimitry Andric             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
16520b57cec5SDimitry Andric               << DstType.getUnqualifiedType();
16530b57cec5SDimitry Andric         } else {
16540b57cec5SDimitry Andric           typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl *>, 64>
16550b57cec5SDimitry Andric               EnumValsTy;
16560b57cec5SDimitry Andric           EnumValsTy EnumVals;
16570b57cec5SDimitry Andric 
16580b57cec5SDimitry Andric           // Gather all enum values, set their type and sort them,
16590b57cec5SDimitry Andric           // allowing easier comparison with rhs constant.
16600b57cec5SDimitry Andric           for (auto *EDI : ED->enumerators()) {
16610b57cec5SDimitry Andric             llvm::APSInt Val = EDI->getInitVal();
16620b57cec5SDimitry Andric             AdjustAPSInt(Val, DstWidth, DstIsSigned);
16630b57cec5SDimitry Andric             EnumVals.push_back(std::make_pair(Val, EDI));
16640b57cec5SDimitry Andric           }
16650b57cec5SDimitry Andric           if (EnumVals.empty())
16660b57cec5SDimitry Andric             return;
16670b57cec5SDimitry Andric           llvm::stable_sort(EnumVals, CmpEnumVals);
16680b57cec5SDimitry Andric           EnumValsTy::iterator EIend =
16690b57cec5SDimitry Andric               std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
16700b57cec5SDimitry Andric 
16710b57cec5SDimitry Andric           // See which values aren't in the enum.
16720b57cec5SDimitry Andric           EnumValsTy::const_iterator EI = EnumVals.begin();
16730b57cec5SDimitry Andric           while (EI != EIend && EI->first < RhsVal)
16740b57cec5SDimitry Andric             EI++;
16750b57cec5SDimitry Andric           if (EI == EIend || EI->first != RhsVal) {
16760b57cec5SDimitry Andric             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
16770b57cec5SDimitry Andric                 << DstType.getUnqualifiedType();
16780b57cec5SDimitry Andric           }
16790b57cec5SDimitry Andric         }
16800b57cec5SDimitry Andric       }
16810b57cec5SDimitry Andric     }
16820b57cec5SDimitry Andric }
16830b57cec5SDimitry Andric 
16845ffd83dbSDimitry Andric StmtResult Sema::ActOnWhileStmt(SourceLocation WhileLoc,
16855ffd83dbSDimitry Andric                                 SourceLocation LParenLoc, ConditionResult Cond,
16865ffd83dbSDimitry Andric                                 SourceLocation RParenLoc, Stmt *Body) {
16870b57cec5SDimitry Andric   if (Cond.isInvalid())
16880b57cec5SDimitry Andric     return StmtError();
16890b57cec5SDimitry Andric 
16900b57cec5SDimitry Andric   auto CondVal = Cond.get();
16910b57cec5SDimitry Andric   CheckBreakContinueBinding(CondVal.second);
16920b57cec5SDimitry Andric 
16930b57cec5SDimitry Andric   if (CondVal.second &&
16940b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, CondVal.second->getExprLoc()))
16950b57cec5SDimitry Andric     CommaVisitor(*this).Visit(CondVal.second);
16960b57cec5SDimitry Andric 
16970b57cec5SDimitry Andric   if (isa<NullStmt>(Body))
16980b57cec5SDimitry Andric     getCurCompoundScope().setHasEmptyLoopBodies();
16990b57cec5SDimitry Andric 
17000b57cec5SDimitry Andric   return WhileStmt::Create(Context, CondVal.first, CondVal.second, Body,
17015ffd83dbSDimitry Andric                            WhileLoc, LParenLoc, RParenLoc);
17020b57cec5SDimitry Andric }
17030b57cec5SDimitry Andric 
17040b57cec5SDimitry Andric StmtResult
17050b57cec5SDimitry Andric Sema::ActOnDoStmt(SourceLocation DoLoc, Stmt *Body,
17060b57cec5SDimitry Andric                   SourceLocation WhileLoc, SourceLocation CondLParen,
17070b57cec5SDimitry Andric                   Expr *Cond, SourceLocation CondRParen) {
17080b57cec5SDimitry Andric   assert(Cond && "ActOnDoStmt(): missing expression");
17090b57cec5SDimitry Andric 
17100b57cec5SDimitry Andric   CheckBreakContinueBinding(Cond);
17110b57cec5SDimitry Andric   ExprResult CondResult = CheckBooleanCondition(DoLoc, Cond);
17120b57cec5SDimitry Andric   if (CondResult.isInvalid())
17130b57cec5SDimitry Andric     return StmtError();
17140b57cec5SDimitry Andric   Cond = CondResult.get();
17150b57cec5SDimitry Andric 
17160b57cec5SDimitry Andric   CondResult = ActOnFinishFullExpr(Cond, DoLoc, /*DiscardedValue*/ false);
17170b57cec5SDimitry Andric   if (CondResult.isInvalid())
17180b57cec5SDimitry Andric     return StmtError();
17190b57cec5SDimitry Andric   Cond = CondResult.get();
17200b57cec5SDimitry Andric 
17210b57cec5SDimitry Andric   // Only call the CommaVisitor for C89 due to differences in scope flags.
17220b57cec5SDimitry Andric   if (Cond && !getLangOpts().C99 && !getLangOpts().CPlusPlus &&
17230b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, Cond->getExprLoc()))
17240b57cec5SDimitry Andric     CommaVisitor(*this).Visit(Cond);
17250b57cec5SDimitry Andric 
17260b57cec5SDimitry Andric   return new (Context) DoStmt(Body, Cond, DoLoc, WhileLoc, CondRParen);
17270b57cec5SDimitry Andric }
17280b57cec5SDimitry Andric 
17290b57cec5SDimitry Andric namespace {
17300b57cec5SDimitry Andric   // Use SetVector since the diagnostic cares about the ordering of the Decl's.
17310b57cec5SDimitry Andric   using DeclSetVector =
17320b57cec5SDimitry Andric       llvm::SetVector<VarDecl *, llvm::SmallVector<VarDecl *, 8>,
17330b57cec5SDimitry Andric                       llvm::SmallPtrSet<VarDecl *, 8>>;
17340b57cec5SDimitry Andric 
17350b57cec5SDimitry Andric   // This visitor will traverse a conditional statement and store all
17360b57cec5SDimitry Andric   // the evaluated decls into a vector.  Simple is set to true if none
17370b57cec5SDimitry Andric   // of the excluded constructs are used.
17380b57cec5SDimitry Andric   class DeclExtractor : public EvaluatedExprVisitor<DeclExtractor> {
17390b57cec5SDimitry Andric     DeclSetVector &Decls;
17400b57cec5SDimitry Andric     SmallVectorImpl<SourceRange> &Ranges;
17410b57cec5SDimitry Andric     bool Simple;
17420b57cec5SDimitry Andric   public:
17430b57cec5SDimitry Andric     typedef EvaluatedExprVisitor<DeclExtractor> Inherited;
17440b57cec5SDimitry Andric 
17450b57cec5SDimitry Andric     DeclExtractor(Sema &S, DeclSetVector &Decls,
17460b57cec5SDimitry Andric                   SmallVectorImpl<SourceRange> &Ranges) :
17470b57cec5SDimitry Andric         Inherited(S.Context),
17480b57cec5SDimitry Andric         Decls(Decls),
17490b57cec5SDimitry Andric         Ranges(Ranges),
17500b57cec5SDimitry Andric         Simple(true) {}
17510b57cec5SDimitry Andric 
17520b57cec5SDimitry Andric     bool isSimple() { return Simple; }
17530b57cec5SDimitry Andric 
17540b57cec5SDimitry Andric     // Replaces the method in EvaluatedExprVisitor.
17550b57cec5SDimitry Andric     void VisitMemberExpr(MemberExpr* E) {
17560b57cec5SDimitry Andric       Simple = false;
17570b57cec5SDimitry Andric     }
17580b57cec5SDimitry Andric 
17595ffd83dbSDimitry Andric     // Any Stmt not explicitly listed will cause the condition to be marked
17605ffd83dbSDimitry Andric     // complex.
17615ffd83dbSDimitry Andric     void VisitStmt(Stmt *S) { Simple = false; }
17620b57cec5SDimitry Andric 
17630b57cec5SDimitry Andric     void VisitBinaryOperator(BinaryOperator *E) {
17640b57cec5SDimitry Andric       Visit(E->getLHS());
17650b57cec5SDimitry Andric       Visit(E->getRHS());
17660b57cec5SDimitry Andric     }
17670b57cec5SDimitry Andric 
17680b57cec5SDimitry Andric     void VisitCastExpr(CastExpr *E) {
17690b57cec5SDimitry Andric       Visit(E->getSubExpr());
17700b57cec5SDimitry Andric     }
17710b57cec5SDimitry Andric 
17720b57cec5SDimitry Andric     void VisitUnaryOperator(UnaryOperator *E) {
17730b57cec5SDimitry Andric       // Skip checking conditionals with derefernces.
17740b57cec5SDimitry Andric       if (E->getOpcode() == UO_Deref)
17750b57cec5SDimitry Andric         Simple = false;
17760b57cec5SDimitry Andric       else
17770b57cec5SDimitry Andric         Visit(E->getSubExpr());
17780b57cec5SDimitry Andric     }
17790b57cec5SDimitry Andric 
17800b57cec5SDimitry Andric     void VisitConditionalOperator(ConditionalOperator *E) {
17810b57cec5SDimitry Andric       Visit(E->getCond());
17820b57cec5SDimitry Andric       Visit(E->getTrueExpr());
17830b57cec5SDimitry Andric       Visit(E->getFalseExpr());
17840b57cec5SDimitry Andric     }
17850b57cec5SDimitry Andric 
17860b57cec5SDimitry Andric     void VisitParenExpr(ParenExpr *E) {
17870b57cec5SDimitry Andric       Visit(E->getSubExpr());
17880b57cec5SDimitry Andric     }
17890b57cec5SDimitry Andric 
17900b57cec5SDimitry Andric     void VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
17910b57cec5SDimitry Andric       Visit(E->getOpaqueValue()->getSourceExpr());
17920b57cec5SDimitry Andric       Visit(E->getFalseExpr());
17930b57cec5SDimitry Andric     }
17940b57cec5SDimitry Andric 
17950b57cec5SDimitry Andric     void VisitIntegerLiteral(IntegerLiteral *E) { }
17960b57cec5SDimitry Andric     void VisitFloatingLiteral(FloatingLiteral *E) { }
17970b57cec5SDimitry Andric     void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) { }
17980b57cec5SDimitry Andric     void VisitCharacterLiteral(CharacterLiteral *E) { }
17990b57cec5SDimitry Andric     void VisitGNUNullExpr(GNUNullExpr *E) { }
18000b57cec5SDimitry Andric     void VisitImaginaryLiteral(ImaginaryLiteral *E) { }
18010b57cec5SDimitry Andric 
18020b57cec5SDimitry Andric     void VisitDeclRefExpr(DeclRefExpr *E) {
18030b57cec5SDimitry Andric       VarDecl *VD = dyn_cast<VarDecl>(E->getDecl());
18040b57cec5SDimitry Andric       if (!VD) {
18050b57cec5SDimitry Andric         // Don't allow unhandled Decl types.
18060b57cec5SDimitry Andric         Simple = false;
18070b57cec5SDimitry Andric         return;
18080b57cec5SDimitry Andric       }
18090b57cec5SDimitry Andric 
18100b57cec5SDimitry Andric       Ranges.push_back(E->getSourceRange());
18110b57cec5SDimitry Andric 
18120b57cec5SDimitry Andric       Decls.insert(VD);
18130b57cec5SDimitry Andric     }
18140b57cec5SDimitry Andric 
18150b57cec5SDimitry Andric   }; // end class DeclExtractor
18160b57cec5SDimitry Andric 
18170b57cec5SDimitry Andric   // DeclMatcher checks to see if the decls are used in a non-evaluated
18180b57cec5SDimitry Andric   // context.
18190b57cec5SDimitry Andric   class DeclMatcher : public EvaluatedExprVisitor<DeclMatcher> {
18200b57cec5SDimitry Andric     DeclSetVector &Decls;
18210b57cec5SDimitry Andric     bool FoundDecl;
18220b57cec5SDimitry Andric 
18230b57cec5SDimitry Andric   public:
18240b57cec5SDimitry Andric     typedef EvaluatedExprVisitor<DeclMatcher> Inherited;
18250b57cec5SDimitry Andric 
18260b57cec5SDimitry Andric     DeclMatcher(Sema &S, DeclSetVector &Decls, Stmt *Statement) :
18270b57cec5SDimitry Andric         Inherited(S.Context), Decls(Decls), FoundDecl(false) {
18280b57cec5SDimitry Andric       if (!Statement) return;
18290b57cec5SDimitry Andric 
18300b57cec5SDimitry Andric       Visit(Statement);
18310b57cec5SDimitry Andric     }
18320b57cec5SDimitry Andric 
18330b57cec5SDimitry Andric     void VisitReturnStmt(ReturnStmt *S) {
18340b57cec5SDimitry Andric       FoundDecl = true;
18350b57cec5SDimitry Andric     }
18360b57cec5SDimitry Andric 
18370b57cec5SDimitry Andric     void VisitBreakStmt(BreakStmt *S) {
18380b57cec5SDimitry Andric       FoundDecl = true;
18390b57cec5SDimitry Andric     }
18400b57cec5SDimitry Andric 
18410b57cec5SDimitry Andric     void VisitGotoStmt(GotoStmt *S) {
18420b57cec5SDimitry Andric       FoundDecl = true;
18430b57cec5SDimitry Andric     }
18440b57cec5SDimitry Andric 
18450b57cec5SDimitry Andric     void VisitCastExpr(CastExpr *E) {
18460b57cec5SDimitry Andric       if (E->getCastKind() == CK_LValueToRValue)
18470b57cec5SDimitry Andric         CheckLValueToRValueCast(E->getSubExpr());
18480b57cec5SDimitry Andric       else
18490b57cec5SDimitry Andric         Visit(E->getSubExpr());
18500b57cec5SDimitry Andric     }
18510b57cec5SDimitry Andric 
18520b57cec5SDimitry Andric     void CheckLValueToRValueCast(Expr *E) {
18530b57cec5SDimitry Andric       E = E->IgnoreParenImpCasts();
18540b57cec5SDimitry Andric 
18550b57cec5SDimitry Andric       if (isa<DeclRefExpr>(E)) {
18560b57cec5SDimitry Andric         return;
18570b57cec5SDimitry Andric       }
18580b57cec5SDimitry Andric 
18590b57cec5SDimitry Andric       if (ConditionalOperator *CO = dyn_cast<ConditionalOperator>(E)) {
18600b57cec5SDimitry Andric         Visit(CO->getCond());
18610b57cec5SDimitry Andric         CheckLValueToRValueCast(CO->getTrueExpr());
18620b57cec5SDimitry Andric         CheckLValueToRValueCast(CO->getFalseExpr());
18630b57cec5SDimitry Andric         return;
18640b57cec5SDimitry Andric       }
18650b57cec5SDimitry Andric 
18660b57cec5SDimitry Andric       if (BinaryConditionalOperator *BCO =
18670b57cec5SDimitry Andric               dyn_cast<BinaryConditionalOperator>(E)) {
18680b57cec5SDimitry Andric         CheckLValueToRValueCast(BCO->getOpaqueValue()->getSourceExpr());
18690b57cec5SDimitry Andric         CheckLValueToRValueCast(BCO->getFalseExpr());
18700b57cec5SDimitry Andric         return;
18710b57cec5SDimitry Andric       }
18720b57cec5SDimitry Andric 
18730b57cec5SDimitry Andric       Visit(E);
18740b57cec5SDimitry Andric     }
18750b57cec5SDimitry Andric 
18760b57cec5SDimitry Andric     void VisitDeclRefExpr(DeclRefExpr *E) {
18770b57cec5SDimitry Andric       if (VarDecl *VD = dyn_cast<VarDecl>(E->getDecl()))
18780b57cec5SDimitry Andric         if (Decls.count(VD))
18790b57cec5SDimitry Andric           FoundDecl = true;
18800b57cec5SDimitry Andric     }
18810b57cec5SDimitry Andric 
18820b57cec5SDimitry Andric     void VisitPseudoObjectExpr(PseudoObjectExpr *POE) {
18830b57cec5SDimitry Andric       // Only need to visit the semantics for POE.
18840b57cec5SDimitry Andric       // SyntaticForm doesn't really use the Decal.
18850b57cec5SDimitry Andric       for (auto *S : POE->semantics()) {
18860b57cec5SDimitry Andric         if (auto *OVE = dyn_cast<OpaqueValueExpr>(S))
18870b57cec5SDimitry Andric           // Look past the OVE into the expression it binds.
18880b57cec5SDimitry Andric           Visit(OVE->getSourceExpr());
18890b57cec5SDimitry Andric         else
18900b57cec5SDimitry Andric           Visit(S);
18910b57cec5SDimitry Andric       }
18920b57cec5SDimitry Andric     }
18930b57cec5SDimitry Andric 
18940b57cec5SDimitry Andric     bool FoundDeclInUse() { return FoundDecl; }
18950b57cec5SDimitry Andric 
18960b57cec5SDimitry Andric   };  // end class DeclMatcher
18970b57cec5SDimitry Andric 
18980b57cec5SDimitry Andric   void CheckForLoopConditionalStatement(Sema &S, Expr *Second,
18990b57cec5SDimitry Andric                                         Expr *Third, Stmt *Body) {
19000b57cec5SDimitry Andric     // Condition is empty
19010b57cec5SDimitry Andric     if (!Second) return;
19020b57cec5SDimitry Andric 
19030b57cec5SDimitry Andric     if (S.Diags.isIgnored(diag::warn_variables_not_in_loop_body,
19040b57cec5SDimitry Andric                           Second->getBeginLoc()))
19050b57cec5SDimitry Andric       return;
19060b57cec5SDimitry Andric 
19070b57cec5SDimitry Andric     PartialDiagnostic PDiag = S.PDiag(diag::warn_variables_not_in_loop_body);
19080b57cec5SDimitry Andric     DeclSetVector Decls;
19090b57cec5SDimitry Andric     SmallVector<SourceRange, 10> Ranges;
19100b57cec5SDimitry Andric     DeclExtractor DE(S, Decls, Ranges);
19110b57cec5SDimitry Andric     DE.Visit(Second);
19120b57cec5SDimitry Andric 
19130b57cec5SDimitry Andric     // Don't analyze complex conditionals.
19140b57cec5SDimitry Andric     if (!DE.isSimple()) return;
19150b57cec5SDimitry Andric 
19160b57cec5SDimitry Andric     // No decls found.
19170b57cec5SDimitry Andric     if (Decls.size() == 0) return;
19180b57cec5SDimitry Andric 
19190b57cec5SDimitry Andric     // Don't warn on volatile, static, or global variables.
19200b57cec5SDimitry Andric     for (auto *VD : Decls)
19210b57cec5SDimitry Andric       if (VD->getType().isVolatileQualified() || VD->hasGlobalStorage())
19220b57cec5SDimitry Andric         return;
19230b57cec5SDimitry Andric 
19240b57cec5SDimitry Andric     if (DeclMatcher(S, Decls, Second).FoundDeclInUse() ||
19250b57cec5SDimitry Andric         DeclMatcher(S, Decls, Third).FoundDeclInUse() ||
19260b57cec5SDimitry Andric         DeclMatcher(S, Decls, Body).FoundDeclInUse())
19270b57cec5SDimitry Andric       return;
19280b57cec5SDimitry Andric 
19290b57cec5SDimitry Andric     // Load decl names into diagnostic.
19300b57cec5SDimitry Andric     if (Decls.size() > 4) {
19310b57cec5SDimitry Andric       PDiag << 0;
19320b57cec5SDimitry Andric     } else {
19330b57cec5SDimitry Andric       PDiag << (unsigned)Decls.size();
19340b57cec5SDimitry Andric       for (auto *VD : Decls)
19350b57cec5SDimitry Andric         PDiag << VD->getDeclName();
19360b57cec5SDimitry Andric     }
19370b57cec5SDimitry Andric 
19380b57cec5SDimitry Andric     for (auto Range : Ranges)
19390b57cec5SDimitry Andric       PDiag << Range;
19400b57cec5SDimitry Andric 
19410b57cec5SDimitry Andric     S.Diag(Ranges.begin()->getBegin(), PDiag);
19420b57cec5SDimitry Andric   }
19430b57cec5SDimitry Andric 
19440b57cec5SDimitry Andric   // If Statement is an incemement or decrement, return true and sets the
19450b57cec5SDimitry Andric   // variables Increment and DRE.
19460b57cec5SDimitry Andric   bool ProcessIterationStmt(Sema &S, Stmt* Statement, bool &Increment,
19470b57cec5SDimitry Andric                             DeclRefExpr *&DRE) {
19480b57cec5SDimitry Andric     if (auto Cleanups = dyn_cast<ExprWithCleanups>(Statement))
19490b57cec5SDimitry Andric       if (!Cleanups->cleanupsHaveSideEffects())
19500b57cec5SDimitry Andric         Statement = Cleanups->getSubExpr();
19510b57cec5SDimitry Andric 
19520b57cec5SDimitry Andric     if (UnaryOperator *UO = dyn_cast<UnaryOperator>(Statement)) {
19530b57cec5SDimitry Andric       switch (UO->getOpcode()) {
19540b57cec5SDimitry Andric         default: return false;
19550b57cec5SDimitry Andric         case UO_PostInc:
19560b57cec5SDimitry Andric         case UO_PreInc:
19570b57cec5SDimitry Andric           Increment = true;
19580b57cec5SDimitry Andric           break;
19590b57cec5SDimitry Andric         case UO_PostDec:
19600b57cec5SDimitry Andric         case UO_PreDec:
19610b57cec5SDimitry Andric           Increment = false;
19620b57cec5SDimitry Andric           break;
19630b57cec5SDimitry Andric       }
19640b57cec5SDimitry Andric       DRE = dyn_cast<DeclRefExpr>(UO->getSubExpr());
19650b57cec5SDimitry Andric       return DRE;
19660b57cec5SDimitry Andric     }
19670b57cec5SDimitry Andric 
19680b57cec5SDimitry Andric     if (CXXOperatorCallExpr *Call = dyn_cast<CXXOperatorCallExpr>(Statement)) {
19690b57cec5SDimitry Andric       FunctionDecl *FD = Call->getDirectCallee();
19700b57cec5SDimitry Andric       if (!FD || !FD->isOverloadedOperator()) return false;
19710b57cec5SDimitry Andric       switch (FD->getOverloadedOperator()) {
19720b57cec5SDimitry Andric         default: return false;
19730b57cec5SDimitry Andric         case OO_PlusPlus:
19740b57cec5SDimitry Andric           Increment = true;
19750b57cec5SDimitry Andric           break;
19760b57cec5SDimitry Andric         case OO_MinusMinus:
19770b57cec5SDimitry Andric           Increment = false;
19780b57cec5SDimitry Andric           break;
19790b57cec5SDimitry Andric       }
19800b57cec5SDimitry Andric       DRE = dyn_cast<DeclRefExpr>(Call->getArg(0));
19810b57cec5SDimitry Andric       return DRE;
19820b57cec5SDimitry Andric     }
19830b57cec5SDimitry Andric 
19840b57cec5SDimitry Andric     return false;
19850b57cec5SDimitry Andric   }
19860b57cec5SDimitry Andric 
19870b57cec5SDimitry Andric   // A visitor to determine if a continue or break statement is a
19880b57cec5SDimitry Andric   // subexpression.
19890b57cec5SDimitry Andric   class BreakContinueFinder : public ConstEvaluatedExprVisitor<BreakContinueFinder> {
19900b57cec5SDimitry Andric     SourceLocation BreakLoc;
19910b57cec5SDimitry Andric     SourceLocation ContinueLoc;
19920b57cec5SDimitry Andric     bool InSwitch = false;
19930b57cec5SDimitry Andric 
19940b57cec5SDimitry Andric   public:
19950b57cec5SDimitry Andric     BreakContinueFinder(Sema &S, const Stmt* Body) :
19960b57cec5SDimitry Andric         Inherited(S.Context) {
19970b57cec5SDimitry Andric       Visit(Body);
19980b57cec5SDimitry Andric     }
19990b57cec5SDimitry Andric 
20000b57cec5SDimitry Andric     typedef ConstEvaluatedExprVisitor<BreakContinueFinder> Inherited;
20010b57cec5SDimitry Andric 
20020b57cec5SDimitry Andric     void VisitContinueStmt(const ContinueStmt* E) {
20030b57cec5SDimitry Andric       ContinueLoc = E->getContinueLoc();
20040b57cec5SDimitry Andric     }
20050b57cec5SDimitry Andric 
20060b57cec5SDimitry Andric     void VisitBreakStmt(const BreakStmt* E) {
20070b57cec5SDimitry Andric       if (!InSwitch)
20080b57cec5SDimitry Andric         BreakLoc = E->getBreakLoc();
20090b57cec5SDimitry Andric     }
20100b57cec5SDimitry Andric 
20110b57cec5SDimitry Andric     void VisitSwitchStmt(const SwitchStmt* S) {
20120b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
20130b57cec5SDimitry Andric         Visit(Init);
20140b57cec5SDimitry Andric       if (const Stmt *CondVar = S->getConditionVariableDeclStmt())
20150b57cec5SDimitry Andric         Visit(CondVar);
20160b57cec5SDimitry Andric       if (const Stmt *Cond = S->getCond())
20170b57cec5SDimitry Andric         Visit(Cond);
20180b57cec5SDimitry Andric 
20190b57cec5SDimitry Andric       // Don't return break statements from the body of a switch.
20200b57cec5SDimitry Andric       InSwitch = true;
20210b57cec5SDimitry Andric       if (const Stmt *Body = S->getBody())
20220b57cec5SDimitry Andric         Visit(Body);
20230b57cec5SDimitry Andric       InSwitch = false;
20240b57cec5SDimitry Andric     }
20250b57cec5SDimitry Andric 
20260b57cec5SDimitry Andric     void VisitForStmt(const ForStmt *S) {
20270b57cec5SDimitry Andric       // Only visit the init statement of a for loop; the body
20280b57cec5SDimitry Andric       // has a different break/continue scope.
20290b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
20300b57cec5SDimitry Andric         Visit(Init);
20310b57cec5SDimitry Andric     }
20320b57cec5SDimitry Andric 
20330b57cec5SDimitry Andric     void VisitWhileStmt(const WhileStmt *) {
20340b57cec5SDimitry Andric       // Do nothing; the children of a while loop have a different
20350b57cec5SDimitry Andric       // break/continue scope.
20360b57cec5SDimitry Andric     }
20370b57cec5SDimitry Andric 
20380b57cec5SDimitry Andric     void VisitDoStmt(const DoStmt *) {
20390b57cec5SDimitry Andric       // Do nothing; the children of a while loop have a different
20400b57cec5SDimitry Andric       // break/continue scope.
20410b57cec5SDimitry Andric     }
20420b57cec5SDimitry Andric 
20430b57cec5SDimitry Andric     void VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
20440b57cec5SDimitry Andric       // Only visit the initialization of a for loop; the body
20450b57cec5SDimitry Andric       // has a different break/continue scope.
20460b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
20470b57cec5SDimitry Andric         Visit(Init);
20480b57cec5SDimitry Andric       if (const Stmt *Range = S->getRangeStmt())
20490b57cec5SDimitry Andric         Visit(Range);
20500b57cec5SDimitry Andric       if (const Stmt *Begin = S->getBeginStmt())
20510b57cec5SDimitry Andric         Visit(Begin);
20520b57cec5SDimitry Andric       if (const Stmt *End = S->getEndStmt())
20530b57cec5SDimitry Andric         Visit(End);
20540b57cec5SDimitry Andric     }
20550b57cec5SDimitry Andric 
20560b57cec5SDimitry Andric     void VisitObjCForCollectionStmt(const ObjCForCollectionStmt *S) {
20570b57cec5SDimitry Andric       // Only visit the initialization of a for loop; the body
20580b57cec5SDimitry Andric       // has a different break/continue scope.
20590b57cec5SDimitry Andric       if (const Stmt *Element = S->getElement())
20600b57cec5SDimitry Andric         Visit(Element);
20610b57cec5SDimitry Andric       if (const Stmt *Collection = S->getCollection())
20620b57cec5SDimitry Andric         Visit(Collection);
20630b57cec5SDimitry Andric     }
20640b57cec5SDimitry Andric 
20650b57cec5SDimitry Andric     bool ContinueFound() { return ContinueLoc.isValid(); }
20660b57cec5SDimitry Andric     bool BreakFound() { return BreakLoc.isValid(); }
20670b57cec5SDimitry Andric     SourceLocation GetContinueLoc() { return ContinueLoc; }
20680b57cec5SDimitry Andric     SourceLocation GetBreakLoc() { return BreakLoc; }
20690b57cec5SDimitry Andric 
20700b57cec5SDimitry Andric   };  // end class BreakContinueFinder
20710b57cec5SDimitry Andric 
20720b57cec5SDimitry Andric   // Emit a warning when a loop increment/decrement appears twice per loop
20730b57cec5SDimitry Andric   // iteration.  The conditions which trigger this warning are:
20740b57cec5SDimitry Andric   // 1) The last statement in the loop body and the third expression in the
20750b57cec5SDimitry Andric   //    for loop are both increment or both decrement of the same variable
20760b57cec5SDimitry Andric   // 2) No continue statements in the loop body.
20770b57cec5SDimitry Andric   void CheckForRedundantIteration(Sema &S, Expr *Third, Stmt *Body) {
20780b57cec5SDimitry Andric     // Return when there is nothing to check.
20790b57cec5SDimitry Andric     if (!Body || !Third) return;
20800b57cec5SDimitry Andric 
20810b57cec5SDimitry Andric     if (S.Diags.isIgnored(diag::warn_redundant_loop_iteration,
20820b57cec5SDimitry Andric                           Third->getBeginLoc()))
20830b57cec5SDimitry Andric       return;
20840b57cec5SDimitry Andric 
20850b57cec5SDimitry Andric     // Get the last statement from the loop body.
20860b57cec5SDimitry Andric     CompoundStmt *CS = dyn_cast<CompoundStmt>(Body);
20870b57cec5SDimitry Andric     if (!CS || CS->body_empty()) return;
20880b57cec5SDimitry Andric     Stmt *LastStmt = CS->body_back();
20890b57cec5SDimitry Andric     if (!LastStmt) return;
20900b57cec5SDimitry Andric 
20910b57cec5SDimitry Andric     bool LoopIncrement, LastIncrement;
20920b57cec5SDimitry Andric     DeclRefExpr *LoopDRE, *LastDRE;
20930b57cec5SDimitry Andric 
20940b57cec5SDimitry Andric     if (!ProcessIterationStmt(S, Third, LoopIncrement, LoopDRE)) return;
20950b57cec5SDimitry Andric     if (!ProcessIterationStmt(S, LastStmt, LastIncrement, LastDRE)) return;
20960b57cec5SDimitry Andric 
20970b57cec5SDimitry Andric     // Check that the two statements are both increments or both decrements
20980b57cec5SDimitry Andric     // on the same variable.
20990b57cec5SDimitry Andric     if (LoopIncrement != LastIncrement ||
21000b57cec5SDimitry Andric         LoopDRE->getDecl() != LastDRE->getDecl()) return;
21010b57cec5SDimitry Andric 
21020b57cec5SDimitry Andric     if (BreakContinueFinder(S, Body).ContinueFound()) return;
21030b57cec5SDimitry Andric 
21040b57cec5SDimitry Andric     S.Diag(LastDRE->getLocation(), diag::warn_redundant_loop_iteration)
21050b57cec5SDimitry Andric          << LastDRE->getDecl() << LastIncrement;
21060b57cec5SDimitry Andric     S.Diag(LoopDRE->getLocation(), diag::note_loop_iteration_here)
21070b57cec5SDimitry Andric          << LoopIncrement;
21080b57cec5SDimitry Andric   }
21090b57cec5SDimitry Andric 
21100b57cec5SDimitry Andric } // end namespace
21110b57cec5SDimitry Andric 
21120b57cec5SDimitry Andric 
21130b57cec5SDimitry Andric void Sema::CheckBreakContinueBinding(Expr *E) {
21140b57cec5SDimitry Andric   if (!E || getLangOpts().CPlusPlus)
21150b57cec5SDimitry Andric     return;
21160b57cec5SDimitry Andric   BreakContinueFinder BCFinder(*this, E);
21170b57cec5SDimitry Andric   Scope *BreakParent = CurScope->getBreakParent();
21180b57cec5SDimitry Andric   if (BCFinder.BreakFound() && BreakParent) {
21190b57cec5SDimitry Andric     if (BreakParent->getFlags() & Scope::SwitchScope) {
21200b57cec5SDimitry Andric       Diag(BCFinder.GetBreakLoc(), diag::warn_break_binds_to_switch);
21210b57cec5SDimitry Andric     } else {
21220b57cec5SDimitry Andric       Diag(BCFinder.GetBreakLoc(), diag::warn_loop_ctrl_binds_to_inner)
21230b57cec5SDimitry Andric           << "break";
21240b57cec5SDimitry Andric     }
21250b57cec5SDimitry Andric   } else if (BCFinder.ContinueFound() && CurScope->getContinueParent()) {
21260b57cec5SDimitry Andric     Diag(BCFinder.GetContinueLoc(), diag::warn_loop_ctrl_binds_to_inner)
21270b57cec5SDimitry Andric         << "continue";
21280b57cec5SDimitry Andric   }
21290b57cec5SDimitry Andric }
21300b57cec5SDimitry Andric 
21310b57cec5SDimitry Andric StmtResult Sema::ActOnForStmt(SourceLocation ForLoc, SourceLocation LParenLoc,
21320b57cec5SDimitry Andric                               Stmt *First, ConditionResult Second,
21330b57cec5SDimitry Andric                               FullExprArg third, SourceLocation RParenLoc,
21340b57cec5SDimitry Andric                               Stmt *Body) {
21350b57cec5SDimitry Andric   if (Second.isInvalid())
21360b57cec5SDimitry Andric     return StmtError();
21370b57cec5SDimitry Andric 
21380b57cec5SDimitry Andric   if (!getLangOpts().CPlusPlus) {
21390b57cec5SDimitry Andric     if (DeclStmt *DS = dyn_cast_or_null<DeclStmt>(First)) {
21400b57cec5SDimitry Andric       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
21410b57cec5SDimitry Andric       // declare identifiers for objects having storage class 'auto' or
21420b57cec5SDimitry Andric       // 'register'.
2143e8d8bef9SDimitry Andric       const Decl *NonVarSeen = nullptr;
2144e8d8bef9SDimitry Andric       bool VarDeclSeen = false;
21450b57cec5SDimitry Andric       for (auto *DI : DS->decls()) {
2146e8d8bef9SDimitry Andric         if (VarDecl *VD = dyn_cast<VarDecl>(DI)) {
2147e8d8bef9SDimitry Andric           VarDeclSeen = true;
2148e8d8bef9SDimitry Andric           if (VD->isLocalVarDecl() && !VD->hasLocalStorage()) {
21490b57cec5SDimitry Andric             Diag(DI->getLocation(), diag::err_non_local_variable_decl_in_for);
21500b57cec5SDimitry Andric             DI->setInvalidDecl();
21510b57cec5SDimitry Andric           }
2152e8d8bef9SDimitry Andric         } else if (!NonVarSeen) {
2153e8d8bef9SDimitry Andric           // Keep track of the first non-variable declaration we saw so that
2154e8d8bef9SDimitry Andric           // we can diagnose if we don't see any variable declarations. This
2155e8d8bef9SDimitry Andric           // covers a case like declaring a typedef, function, or structure
2156e8d8bef9SDimitry Andric           // type rather than a variable.
2157e8d8bef9SDimitry Andric           NonVarSeen = DI;
21580b57cec5SDimitry Andric         }
21590b57cec5SDimitry Andric       }
2160e8d8bef9SDimitry Andric       // Diagnose if we saw a non-variable declaration but no variable
2161e8d8bef9SDimitry Andric       // declarations.
2162e8d8bef9SDimitry Andric       if (NonVarSeen && !VarDeclSeen)
2163e8d8bef9SDimitry Andric         Diag(NonVarSeen->getLocation(), diag::err_non_variable_decl_in_for);
2164e8d8bef9SDimitry Andric     }
21650b57cec5SDimitry Andric   }
21660b57cec5SDimitry Andric 
21670b57cec5SDimitry Andric   CheckBreakContinueBinding(Second.get().second);
21680b57cec5SDimitry Andric   CheckBreakContinueBinding(third.get());
21690b57cec5SDimitry Andric 
21700b57cec5SDimitry Andric   if (!Second.get().first)
21710b57cec5SDimitry Andric     CheckForLoopConditionalStatement(*this, Second.get().second, third.get(),
21720b57cec5SDimitry Andric                                      Body);
21730b57cec5SDimitry Andric   CheckForRedundantIteration(*this, third.get(), Body);
21740b57cec5SDimitry Andric 
21750b57cec5SDimitry Andric   if (Second.get().second &&
21760b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator,
21770b57cec5SDimitry Andric                        Second.get().second->getExprLoc()))
21780b57cec5SDimitry Andric     CommaVisitor(*this).Visit(Second.get().second);
21790b57cec5SDimitry Andric 
21800b57cec5SDimitry Andric   Expr *Third  = third.release().getAs<Expr>();
21810b57cec5SDimitry Andric   if (isa<NullStmt>(Body))
21820b57cec5SDimitry Andric     getCurCompoundScope().setHasEmptyLoopBodies();
21830b57cec5SDimitry Andric 
21840b57cec5SDimitry Andric   return new (Context)
21850b57cec5SDimitry Andric       ForStmt(Context, First, Second.get().second, Second.get().first, Third,
21860b57cec5SDimitry Andric               Body, ForLoc, LParenLoc, RParenLoc);
21870b57cec5SDimitry Andric }
21880b57cec5SDimitry Andric 
21890b57cec5SDimitry Andric /// In an Objective C collection iteration statement:
21900b57cec5SDimitry Andric ///   for (x in y)
21910b57cec5SDimitry Andric /// x can be an arbitrary l-value expression.  Bind it up as a
21920b57cec5SDimitry Andric /// full-expression.
21930b57cec5SDimitry Andric StmtResult Sema::ActOnForEachLValueExpr(Expr *E) {
21940b57cec5SDimitry Andric   // Reduce placeholder expressions here.  Note that this rejects the
21950b57cec5SDimitry Andric   // use of pseudo-object l-values in this position.
21960b57cec5SDimitry Andric   ExprResult result = CheckPlaceholderExpr(E);
21970b57cec5SDimitry Andric   if (result.isInvalid()) return StmtError();
21980b57cec5SDimitry Andric   E = result.get();
21990b57cec5SDimitry Andric 
22000b57cec5SDimitry Andric   ExprResult FullExpr = ActOnFinishFullExpr(E, /*DiscardedValue*/ false);
22010b57cec5SDimitry Andric   if (FullExpr.isInvalid())
22020b57cec5SDimitry Andric     return StmtError();
22030b57cec5SDimitry Andric   return StmtResult(static_cast<Stmt*>(FullExpr.get()));
22040b57cec5SDimitry Andric }
22050b57cec5SDimitry Andric 
22060b57cec5SDimitry Andric ExprResult
22070b57cec5SDimitry Andric Sema::CheckObjCForCollectionOperand(SourceLocation forLoc, Expr *collection) {
22080b57cec5SDimitry Andric   if (!collection)
22090b57cec5SDimitry Andric     return ExprError();
22100b57cec5SDimitry Andric 
22110b57cec5SDimitry Andric   ExprResult result = CorrectDelayedTyposInExpr(collection);
22120b57cec5SDimitry Andric   if (!result.isUsable())
22130b57cec5SDimitry Andric     return ExprError();
22140b57cec5SDimitry Andric   collection = result.get();
22150b57cec5SDimitry Andric 
22160b57cec5SDimitry Andric   // Bail out early if we've got a type-dependent expression.
22170b57cec5SDimitry Andric   if (collection->isTypeDependent()) return collection;
22180b57cec5SDimitry Andric 
22190b57cec5SDimitry Andric   // Perform normal l-value conversion.
22200b57cec5SDimitry Andric   result = DefaultFunctionArrayLvalueConversion(collection);
22210b57cec5SDimitry Andric   if (result.isInvalid())
22220b57cec5SDimitry Andric     return ExprError();
22230b57cec5SDimitry Andric   collection = result.get();
22240b57cec5SDimitry Andric 
22250b57cec5SDimitry Andric   // The operand needs to have object-pointer type.
22260b57cec5SDimitry Andric   // TODO: should we do a contextual conversion?
22270b57cec5SDimitry Andric   const ObjCObjectPointerType *pointerType =
22280b57cec5SDimitry Andric     collection->getType()->getAs<ObjCObjectPointerType>();
22290b57cec5SDimitry Andric   if (!pointerType)
22300b57cec5SDimitry Andric     return Diag(forLoc, diag::err_collection_expr_type)
22310b57cec5SDimitry Andric              << collection->getType() << collection->getSourceRange();
22320b57cec5SDimitry Andric 
22330b57cec5SDimitry Andric   // Check that the operand provides
22340b57cec5SDimitry Andric   //   - countByEnumeratingWithState:objects:count:
22350b57cec5SDimitry Andric   const ObjCObjectType *objectType = pointerType->getObjectType();
22360b57cec5SDimitry Andric   ObjCInterfaceDecl *iface = objectType->getInterface();
22370b57cec5SDimitry Andric 
22380b57cec5SDimitry Andric   // If we have a forward-declared type, we can't do this check.
22390b57cec5SDimitry Andric   // Under ARC, it is an error not to have a forward-declared class.
22400b57cec5SDimitry Andric   if (iface &&
22410b57cec5SDimitry Andric       (getLangOpts().ObjCAutoRefCount
22420b57cec5SDimitry Andric            ? RequireCompleteType(forLoc, QualType(objectType, 0),
22430b57cec5SDimitry Andric                                  diag::err_arc_collection_forward, collection)
22440b57cec5SDimitry Andric            : !isCompleteType(forLoc, QualType(objectType, 0)))) {
22450b57cec5SDimitry Andric     // Otherwise, if we have any useful type information, check that
22460b57cec5SDimitry Andric     // the type declares the appropriate method.
22470b57cec5SDimitry Andric   } else if (iface || !objectType->qual_empty()) {
22480b57cec5SDimitry Andric     IdentifierInfo *selectorIdents[] = {
22490b57cec5SDimitry Andric       &Context.Idents.get("countByEnumeratingWithState"),
22500b57cec5SDimitry Andric       &Context.Idents.get("objects"),
22510b57cec5SDimitry Andric       &Context.Idents.get("count")
22520b57cec5SDimitry Andric     };
22530b57cec5SDimitry Andric     Selector selector = Context.Selectors.getSelector(3, &selectorIdents[0]);
22540b57cec5SDimitry Andric 
22550b57cec5SDimitry Andric     ObjCMethodDecl *method = nullptr;
22560b57cec5SDimitry Andric 
22570b57cec5SDimitry Andric     // If there's an interface, look in both the public and private APIs.
22580b57cec5SDimitry Andric     if (iface) {
22590b57cec5SDimitry Andric       method = iface->lookupInstanceMethod(selector);
22600b57cec5SDimitry Andric       if (!method) method = iface->lookupPrivateMethod(selector);
22610b57cec5SDimitry Andric     }
22620b57cec5SDimitry Andric 
22630b57cec5SDimitry Andric     // Also check protocol qualifiers.
22640b57cec5SDimitry Andric     if (!method)
22650b57cec5SDimitry Andric       method = LookupMethodInQualifiedType(selector, pointerType,
22660b57cec5SDimitry Andric                                            /*instance*/ true);
22670b57cec5SDimitry Andric 
22680b57cec5SDimitry Andric     // If we didn't find it anywhere, give up.
22690b57cec5SDimitry Andric     if (!method) {
22700b57cec5SDimitry Andric       Diag(forLoc, diag::warn_collection_expr_type)
22710b57cec5SDimitry Andric         << collection->getType() << selector << collection->getSourceRange();
22720b57cec5SDimitry Andric     }
22730b57cec5SDimitry Andric 
22740b57cec5SDimitry Andric     // TODO: check for an incompatible signature?
22750b57cec5SDimitry Andric   }
22760b57cec5SDimitry Andric 
22770b57cec5SDimitry Andric   // Wrap up any cleanups in the expression.
22780b57cec5SDimitry Andric   return collection;
22790b57cec5SDimitry Andric }
22800b57cec5SDimitry Andric 
22810b57cec5SDimitry Andric StmtResult
22820b57cec5SDimitry Andric Sema::ActOnObjCForCollectionStmt(SourceLocation ForLoc,
22830b57cec5SDimitry Andric                                  Stmt *First, Expr *collection,
22840b57cec5SDimitry Andric                                  SourceLocation RParenLoc) {
22850b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
22860b57cec5SDimitry Andric 
22870b57cec5SDimitry Andric   ExprResult CollectionExprResult =
22880b57cec5SDimitry Andric     CheckObjCForCollectionOperand(ForLoc, collection);
22890b57cec5SDimitry Andric 
22900b57cec5SDimitry Andric   if (First) {
22910b57cec5SDimitry Andric     QualType FirstType;
22920b57cec5SDimitry Andric     if (DeclStmt *DS = dyn_cast<DeclStmt>(First)) {
22930b57cec5SDimitry Andric       if (!DS->isSingleDecl())
22940b57cec5SDimitry Andric         return StmtError(Diag((*DS->decl_begin())->getLocation(),
22950b57cec5SDimitry Andric                          diag::err_toomany_element_decls));
22960b57cec5SDimitry Andric 
22970b57cec5SDimitry Andric       VarDecl *D = dyn_cast<VarDecl>(DS->getSingleDecl());
22980b57cec5SDimitry Andric       if (!D || D->isInvalidDecl())
22990b57cec5SDimitry Andric         return StmtError();
23000b57cec5SDimitry Andric 
23010b57cec5SDimitry Andric       FirstType = D->getType();
23020b57cec5SDimitry Andric       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
23030b57cec5SDimitry Andric       // declare identifiers for objects having storage class 'auto' or
23040b57cec5SDimitry Andric       // 'register'.
23050b57cec5SDimitry Andric       if (!D->hasLocalStorage())
23060b57cec5SDimitry Andric         return StmtError(Diag(D->getLocation(),
23070b57cec5SDimitry Andric                               diag::err_non_local_variable_decl_in_for));
23080b57cec5SDimitry Andric 
23090b57cec5SDimitry Andric       // If the type contained 'auto', deduce the 'auto' to 'id'.
23100b57cec5SDimitry Andric       if (FirstType->getContainedAutoType()) {
23110b57cec5SDimitry Andric         OpaqueValueExpr OpaqueId(D->getLocation(), Context.getObjCIdType(),
2312fe6060f1SDimitry Andric                                  VK_PRValue);
23130b57cec5SDimitry Andric         Expr *DeducedInit = &OpaqueId;
23140b57cec5SDimitry Andric         if (DeduceAutoType(D->getTypeSourceInfo(), DeducedInit, FirstType) ==
23150b57cec5SDimitry Andric                 DAR_Failed)
23160b57cec5SDimitry Andric           DiagnoseAutoDeductionFailure(D, DeducedInit);
23170b57cec5SDimitry Andric         if (FirstType.isNull()) {
23180b57cec5SDimitry Andric           D->setInvalidDecl();
23190b57cec5SDimitry Andric           return StmtError();
23200b57cec5SDimitry Andric         }
23210b57cec5SDimitry Andric 
23220b57cec5SDimitry Andric         D->setType(FirstType);
23230b57cec5SDimitry Andric 
23240b57cec5SDimitry Andric         if (!inTemplateInstantiation()) {
23250b57cec5SDimitry Andric           SourceLocation Loc =
23260b57cec5SDimitry Andric               D->getTypeSourceInfo()->getTypeLoc().getBeginLoc();
23270b57cec5SDimitry Andric           Diag(Loc, diag::warn_auto_var_is_id)
23280b57cec5SDimitry Andric             << D->getDeclName();
23290b57cec5SDimitry Andric         }
23300b57cec5SDimitry Andric       }
23310b57cec5SDimitry Andric 
23320b57cec5SDimitry Andric     } else {
23330b57cec5SDimitry Andric       Expr *FirstE = cast<Expr>(First);
23340b57cec5SDimitry Andric       if (!FirstE->isTypeDependent() && !FirstE->isLValue())
23350b57cec5SDimitry Andric         return StmtError(
23360b57cec5SDimitry Andric             Diag(First->getBeginLoc(), diag::err_selector_element_not_lvalue)
23370b57cec5SDimitry Andric             << First->getSourceRange());
23380b57cec5SDimitry Andric 
23390b57cec5SDimitry Andric       FirstType = static_cast<Expr*>(First)->getType();
23400b57cec5SDimitry Andric       if (FirstType.isConstQualified())
23410b57cec5SDimitry Andric         Diag(ForLoc, diag::err_selector_element_const_type)
23420b57cec5SDimitry Andric           << FirstType << First->getSourceRange();
23430b57cec5SDimitry Andric     }
23440b57cec5SDimitry Andric     if (!FirstType->isDependentType() &&
23450b57cec5SDimitry Andric         !FirstType->isObjCObjectPointerType() &&
23460b57cec5SDimitry Andric         !FirstType->isBlockPointerType())
23470b57cec5SDimitry Andric         return StmtError(Diag(ForLoc, diag::err_selector_element_type)
23480b57cec5SDimitry Andric                            << FirstType << First->getSourceRange());
23490b57cec5SDimitry Andric   }
23500b57cec5SDimitry Andric 
23510b57cec5SDimitry Andric   if (CollectionExprResult.isInvalid())
23520b57cec5SDimitry Andric     return StmtError();
23530b57cec5SDimitry Andric 
23540b57cec5SDimitry Andric   CollectionExprResult =
23550b57cec5SDimitry Andric       ActOnFinishFullExpr(CollectionExprResult.get(), /*DiscardedValue*/ false);
23560b57cec5SDimitry Andric   if (CollectionExprResult.isInvalid())
23570b57cec5SDimitry Andric     return StmtError();
23580b57cec5SDimitry Andric 
23590b57cec5SDimitry Andric   return new (Context) ObjCForCollectionStmt(First, CollectionExprResult.get(),
23600b57cec5SDimitry Andric                                              nullptr, ForLoc, RParenLoc);
23610b57cec5SDimitry Andric }
23620b57cec5SDimitry Andric 
23630b57cec5SDimitry Andric /// Finish building a variable declaration for a for-range statement.
23640b57cec5SDimitry Andric /// \return true if an error occurs.
23650b57cec5SDimitry Andric static bool FinishForRangeVarDecl(Sema &SemaRef, VarDecl *Decl, Expr *Init,
23660b57cec5SDimitry Andric                                   SourceLocation Loc, int DiagID) {
23670b57cec5SDimitry Andric   if (Decl->getType()->isUndeducedType()) {
23680b57cec5SDimitry Andric     ExprResult Res = SemaRef.CorrectDelayedTyposInExpr(Init);
23690b57cec5SDimitry Andric     if (!Res.isUsable()) {
23700b57cec5SDimitry Andric       Decl->setInvalidDecl();
23710b57cec5SDimitry Andric       return true;
23720b57cec5SDimitry Andric     }
23730b57cec5SDimitry Andric     Init = Res.get();
23740b57cec5SDimitry Andric   }
23750b57cec5SDimitry Andric 
23760b57cec5SDimitry Andric   // Deduce the type for the iterator variable now rather than leaving it to
23770b57cec5SDimitry Andric   // AddInitializerToDecl, so we can produce a more suitable diagnostic.
23780b57cec5SDimitry Andric   QualType InitType;
23790b57cec5SDimitry Andric   if ((!isa<InitListExpr>(Init) && Init->getType()->isVoidType()) ||
23800b57cec5SDimitry Andric       SemaRef.DeduceAutoType(Decl->getTypeSourceInfo(), Init, InitType) ==
23810b57cec5SDimitry Andric           Sema::DAR_Failed)
23820b57cec5SDimitry Andric     SemaRef.Diag(Loc, DiagID) << Init->getType();
23830b57cec5SDimitry Andric   if (InitType.isNull()) {
23840b57cec5SDimitry Andric     Decl->setInvalidDecl();
23850b57cec5SDimitry Andric     return true;
23860b57cec5SDimitry Andric   }
23870b57cec5SDimitry Andric   Decl->setType(InitType);
23880b57cec5SDimitry Andric 
23890b57cec5SDimitry Andric   // In ARC, infer lifetime.
23900b57cec5SDimitry Andric   // FIXME: ARC may want to turn this into 'const __unsafe_unretained' if
23910b57cec5SDimitry Andric   // we're doing the equivalent of fast iteration.
23920b57cec5SDimitry Andric   if (SemaRef.getLangOpts().ObjCAutoRefCount &&
23930b57cec5SDimitry Andric       SemaRef.inferObjCARCLifetime(Decl))
23940b57cec5SDimitry Andric     Decl->setInvalidDecl();
23950b57cec5SDimitry Andric 
23960b57cec5SDimitry Andric   SemaRef.AddInitializerToDecl(Decl, Init, /*DirectInit=*/false);
23970b57cec5SDimitry Andric   SemaRef.FinalizeDeclaration(Decl);
23980b57cec5SDimitry Andric   SemaRef.CurContext->addHiddenDecl(Decl);
23990b57cec5SDimitry Andric   return false;
24000b57cec5SDimitry Andric }
24010b57cec5SDimitry Andric 
24020b57cec5SDimitry Andric namespace {
24030b57cec5SDimitry Andric // An enum to represent whether something is dealing with a call to begin()
24040b57cec5SDimitry Andric // or a call to end() in a range-based for loop.
24050b57cec5SDimitry Andric enum BeginEndFunction {
24060b57cec5SDimitry Andric   BEF_begin,
24070b57cec5SDimitry Andric   BEF_end
24080b57cec5SDimitry Andric };
24090b57cec5SDimitry Andric 
24100b57cec5SDimitry Andric /// Produce a note indicating which begin/end function was implicitly called
24110b57cec5SDimitry Andric /// by a C++11 for-range statement. This is often not obvious from the code,
24120b57cec5SDimitry Andric /// nor from the diagnostics produced when analysing the implicit expressions
24130b57cec5SDimitry Andric /// required in a for-range statement.
24140b57cec5SDimitry Andric void NoteForRangeBeginEndFunction(Sema &SemaRef, Expr *E,
24150b57cec5SDimitry Andric                                   BeginEndFunction BEF) {
24160b57cec5SDimitry Andric   CallExpr *CE = dyn_cast<CallExpr>(E);
24170b57cec5SDimitry Andric   if (!CE)
24180b57cec5SDimitry Andric     return;
24190b57cec5SDimitry Andric   FunctionDecl *D = dyn_cast<FunctionDecl>(CE->getCalleeDecl());
24200b57cec5SDimitry Andric   if (!D)
24210b57cec5SDimitry Andric     return;
24220b57cec5SDimitry Andric   SourceLocation Loc = D->getLocation();
24230b57cec5SDimitry Andric 
24240b57cec5SDimitry Andric   std::string Description;
24250b57cec5SDimitry Andric   bool IsTemplate = false;
24260b57cec5SDimitry Andric   if (FunctionTemplateDecl *FunTmpl = D->getPrimaryTemplate()) {
24270b57cec5SDimitry Andric     Description = SemaRef.getTemplateArgumentBindingsText(
24280b57cec5SDimitry Andric       FunTmpl->getTemplateParameters(), *D->getTemplateSpecializationArgs());
24290b57cec5SDimitry Andric     IsTemplate = true;
24300b57cec5SDimitry Andric   }
24310b57cec5SDimitry Andric 
24320b57cec5SDimitry Andric   SemaRef.Diag(Loc, diag::note_for_range_begin_end)
24330b57cec5SDimitry Andric     << BEF << IsTemplate << Description << E->getType();
24340b57cec5SDimitry Andric }
24350b57cec5SDimitry Andric 
24360b57cec5SDimitry Andric /// Build a variable declaration for a for-range statement.
24370b57cec5SDimitry Andric VarDecl *BuildForRangeVarDecl(Sema &SemaRef, SourceLocation Loc,
24380b57cec5SDimitry Andric                               QualType Type, StringRef Name) {
24390b57cec5SDimitry Andric   DeclContext *DC = SemaRef.CurContext;
24400b57cec5SDimitry Andric   IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name);
24410b57cec5SDimitry Andric   TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc);
24420b57cec5SDimitry Andric   VarDecl *Decl = VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type,
24430b57cec5SDimitry Andric                                   TInfo, SC_None);
24440b57cec5SDimitry Andric   Decl->setImplicit();
24450b57cec5SDimitry Andric   return Decl;
24460b57cec5SDimitry Andric }
24470b57cec5SDimitry Andric 
24480b57cec5SDimitry Andric }
24490b57cec5SDimitry Andric 
24500b57cec5SDimitry Andric static bool ObjCEnumerationCollection(Expr *Collection) {
24510b57cec5SDimitry Andric   return !Collection->isTypeDependent()
24520b57cec5SDimitry Andric           && Collection->getType()->getAs<ObjCObjectPointerType>() != nullptr;
24530b57cec5SDimitry Andric }
24540b57cec5SDimitry Andric 
24550b57cec5SDimitry Andric /// ActOnCXXForRangeStmt - Check and build a C++11 for-range statement.
24560b57cec5SDimitry Andric ///
24570b57cec5SDimitry Andric /// C++11 [stmt.ranged]:
24580b57cec5SDimitry Andric ///   A range-based for statement is equivalent to
24590b57cec5SDimitry Andric ///
24600b57cec5SDimitry Andric ///   {
24610b57cec5SDimitry Andric ///     auto && __range = range-init;
24620b57cec5SDimitry Andric ///     for ( auto __begin = begin-expr,
24630b57cec5SDimitry Andric ///           __end = end-expr;
24640b57cec5SDimitry Andric ///           __begin != __end;
24650b57cec5SDimitry Andric ///           ++__begin ) {
24660b57cec5SDimitry Andric ///       for-range-declaration = *__begin;
24670b57cec5SDimitry Andric ///       statement
24680b57cec5SDimitry Andric ///     }
24690b57cec5SDimitry Andric ///   }
24700b57cec5SDimitry Andric ///
24710b57cec5SDimitry Andric /// The body of the loop is not available yet, since it cannot be analysed until
24720b57cec5SDimitry Andric /// we have determined the type of the for-range-declaration.
24730b57cec5SDimitry Andric StmtResult Sema::ActOnCXXForRangeStmt(Scope *S, SourceLocation ForLoc,
24740b57cec5SDimitry Andric                                       SourceLocation CoawaitLoc, Stmt *InitStmt,
24750b57cec5SDimitry Andric                                       Stmt *First, SourceLocation ColonLoc,
24760b57cec5SDimitry Andric                                       Expr *Range, SourceLocation RParenLoc,
24770b57cec5SDimitry Andric                                       BuildForRangeKind Kind) {
247804eeddc0SDimitry Andric   // FIXME: recover in order to allow the body to be parsed.
24790b57cec5SDimitry Andric   if (!First)
24800b57cec5SDimitry Andric     return StmtError();
24810b57cec5SDimitry Andric 
24820b57cec5SDimitry Andric   if (Range && ObjCEnumerationCollection(Range)) {
24830b57cec5SDimitry Andric     // FIXME: Support init-statements in Objective-C++20 ranged for statement.
24840b57cec5SDimitry Andric     if (InitStmt)
24850b57cec5SDimitry Andric       return Diag(InitStmt->getBeginLoc(), diag::err_objc_for_range_init_stmt)
24860b57cec5SDimitry Andric                  << InitStmt->getSourceRange();
24870b57cec5SDimitry Andric     return ActOnObjCForCollectionStmt(ForLoc, First, Range, RParenLoc);
24880b57cec5SDimitry Andric   }
24890b57cec5SDimitry Andric 
24900b57cec5SDimitry Andric   DeclStmt *DS = dyn_cast<DeclStmt>(First);
24910b57cec5SDimitry Andric   assert(DS && "first part of for range not a decl stmt");
24920b57cec5SDimitry Andric 
24930b57cec5SDimitry Andric   if (!DS->isSingleDecl()) {
24940b57cec5SDimitry Andric     Diag(DS->getBeginLoc(), diag::err_type_defined_in_for_range);
24950b57cec5SDimitry Andric     return StmtError();
24960b57cec5SDimitry Andric   }
24970b57cec5SDimitry Andric 
24985ffd83dbSDimitry Andric   // This function is responsible for attaching an initializer to LoopVar. We
24995ffd83dbSDimitry Andric   // must call ActOnInitializerError if we fail to do so.
25000b57cec5SDimitry Andric   Decl *LoopVar = DS->getSingleDecl();
25010b57cec5SDimitry Andric   if (LoopVar->isInvalidDecl() || !Range ||
25020b57cec5SDimitry Andric       DiagnoseUnexpandedParameterPack(Range, UPPC_Expression)) {
25035ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25040b57cec5SDimitry Andric     return StmtError();
25050b57cec5SDimitry Andric   }
25060b57cec5SDimitry Andric 
25070b57cec5SDimitry Andric   // Build the coroutine state immediately and not later during template
25080b57cec5SDimitry Andric   // instantiation
25090b57cec5SDimitry Andric   if (!CoawaitLoc.isInvalid()) {
25105ffd83dbSDimitry Andric     if (!ActOnCoroutineBodyStart(S, CoawaitLoc, "co_await")) {
25115ffd83dbSDimitry Andric       ActOnInitializerError(LoopVar);
25120b57cec5SDimitry Andric       return StmtError();
25130b57cec5SDimitry Andric     }
25145ffd83dbSDimitry Andric   }
25150b57cec5SDimitry Andric 
25160b57cec5SDimitry Andric   // Build  auto && __range = range-init
25170b57cec5SDimitry Andric   // Divide by 2, since the variables are in the inner scope (loop body).
25180b57cec5SDimitry Andric   const auto DepthStr = std::to_string(S->getDepth() / 2);
25190b57cec5SDimitry Andric   SourceLocation RangeLoc = Range->getBeginLoc();
25200b57cec5SDimitry Andric   VarDecl *RangeVar = BuildForRangeVarDecl(*this, RangeLoc,
25210b57cec5SDimitry Andric                                            Context.getAutoRRefDeductType(),
25220b57cec5SDimitry Andric                                            std::string("__range") + DepthStr);
25230b57cec5SDimitry Andric   if (FinishForRangeVarDecl(*this, RangeVar, Range, RangeLoc,
25240b57cec5SDimitry Andric                             diag::err_for_range_deduction_failure)) {
25255ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25260b57cec5SDimitry Andric     return StmtError();
25270b57cec5SDimitry Andric   }
25280b57cec5SDimitry Andric 
25290b57cec5SDimitry Andric   // Claim the type doesn't contain auto: we've already done the checking.
25300b57cec5SDimitry Andric   DeclGroupPtrTy RangeGroup =
25310b57cec5SDimitry Andric       BuildDeclaratorGroup(MutableArrayRef<Decl *>((Decl **)&RangeVar, 1));
25320b57cec5SDimitry Andric   StmtResult RangeDecl = ActOnDeclStmt(RangeGroup, RangeLoc, RangeLoc);
25330b57cec5SDimitry Andric   if (RangeDecl.isInvalid()) {
25345ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25350b57cec5SDimitry Andric     return StmtError();
25360b57cec5SDimitry Andric   }
25370b57cec5SDimitry Andric 
25385ffd83dbSDimitry Andric   StmtResult R = BuildCXXForRangeStmt(
25390b57cec5SDimitry Andric       ForLoc, CoawaitLoc, InitStmt, ColonLoc, RangeDecl.get(),
25400b57cec5SDimitry Andric       /*BeginStmt=*/nullptr, /*EndStmt=*/nullptr,
25410b57cec5SDimitry Andric       /*Cond=*/nullptr, /*Inc=*/nullptr, DS, RParenLoc, Kind);
25425ffd83dbSDimitry Andric   if (R.isInvalid()) {
25435ffd83dbSDimitry Andric     ActOnInitializerError(LoopVar);
25445ffd83dbSDimitry Andric     return StmtError();
25455ffd83dbSDimitry Andric   }
25465ffd83dbSDimitry Andric 
25475ffd83dbSDimitry Andric   return R;
25480b57cec5SDimitry Andric }
25490b57cec5SDimitry Andric 
25500b57cec5SDimitry Andric /// Create the initialization, compare, and increment steps for
25510b57cec5SDimitry Andric /// the range-based for loop expression.
25520b57cec5SDimitry Andric /// This function does not handle array-based for loops,
25530b57cec5SDimitry Andric /// which are created in Sema::BuildCXXForRangeStmt.
25540b57cec5SDimitry Andric ///
25550b57cec5SDimitry Andric /// \returns a ForRangeStatus indicating success or what kind of error occurred.
25560b57cec5SDimitry Andric /// BeginExpr and EndExpr are set and FRS_Success is returned on success;
25570b57cec5SDimitry Andric /// CandidateSet and BEF are set and some non-success value is returned on
25580b57cec5SDimitry Andric /// failure.
25590b57cec5SDimitry Andric static Sema::ForRangeStatus
25600b57cec5SDimitry Andric BuildNonArrayForRange(Sema &SemaRef, Expr *BeginRange, Expr *EndRange,
25610b57cec5SDimitry Andric                       QualType RangeType, VarDecl *BeginVar, VarDecl *EndVar,
25620b57cec5SDimitry Andric                       SourceLocation ColonLoc, SourceLocation CoawaitLoc,
25630b57cec5SDimitry Andric                       OverloadCandidateSet *CandidateSet, ExprResult *BeginExpr,
25640b57cec5SDimitry Andric                       ExprResult *EndExpr, BeginEndFunction *BEF) {
25650b57cec5SDimitry Andric   DeclarationNameInfo BeginNameInfo(
25660b57cec5SDimitry Andric       &SemaRef.PP.getIdentifierTable().get("begin"), ColonLoc);
25670b57cec5SDimitry Andric   DeclarationNameInfo EndNameInfo(&SemaRef.PP.getIdentifierTable().get("end"),
25680b57cec5SDimitry Andric                                   ColonLoc);
25690b57cec5SDimitry Andric 
25700b57cec5SDimitry Andric   LookupResult BeginMemberLookup(SemaRef, BeginNameInfo,
25710b57cec5SDimitry Andric                                  Sema::LookupMemberName);
25720b57cec5SDimitry Andric   LookupResult EndMemberLookup(SemaRef, EndNameInfo, Sema::LookupMemberName);
25730b57cec5SDimitry Andric 
25740b57cec5SDimitry Andric   auto BuildBegin = [&] {
25750b57cec5SDimitry Andric     *BEF = BEF_begin;
25760b57cec5SDimitry Andric     Sema::ForRangeStatus RangeStatus =
25770b57cec5SDimitry Andric         SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, BeginNameInfo,
25780b57cec5SDimitry Andric                                           BeginMemberLookup, CandidateSet,
25790b57cec5SDimitry Andric                                           BeginRange, BeginExpr);
25800b57cec5SDimitry Andric 
25810b57cec5SDimitry Andric     if (RangeStatus != Sema::FRS_Success) {
25820b57cec5SDimitry Andric       if (RangeStatus == Sema::FRS_DiagnosticIssued)
25830b57cec5SDimitry Andric         SemaRef.Diag(BeginRange->getBeginLoc(), diag::note_in_for_range)
25840b57cec5SDimitry Andric             << ColonLoc << BEF_begin << BeginRange->getType();
25850b57cec5SDimitry Andric       return RangeStatus;
25860b57cec5SDimitry Andric     }
25870b57cec5SDimitry Andric     if (!CoawaitLoc.isInvalid()) {
25880b57cec5SDimitry Andric       // FIXME: getCurScope() should not be used during template instantiation.
25890b57cec5SDimitry Andric       // We should pick up the set of unqualified lookup results for operator
25900b57cec5SDimitry Andric       // co_await during the initial parse.
25910b57cec5SDimitry Andric       *BeginExpr = SemaRef.ActOnCoawaitExpr(SemaRef.getCurScope(), ColonLoc,
25920b57cec5SDimitry Andric                                             BeginExpr->get());
25930b57cec5SDimitry Andric       if (BeginExpr->isInvalid())
25940b57cec5SDimitry Andric         return Sema::FRS_DiagnosticIssued;
25950b57cec5SDimitry Andric     }
25960b57cec5SDimitry Andric     if (FinishForRangeVarDecl(SemaRef, BeginVar, BeginExpr->get(), ColonLoc,
25970b57cec5SDimitry Andric                               diag::err_for_range_iter_deduction_failure)) {
25980b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(SemaRef, BeginExpr->get(), *BEF);
25990b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26000b57cec5SDimitry Andric     }
26010b57cec5SDimitry Andric     return Sema::FRS_Success;
26020b57cec5SDimitry Andric   };
26030b57cec5SDimitry Andric 
26040b57cec5SDimitry Andric   auto BuildEnd = [&] {
26050b57cec5SDimitry Andric     *BEF = BEF_end;
26060b57cec5SDimitry Andric     Sema::ForRangeStatus RangeStatus =
26070b57cec5SDimitry Andric         SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, EndNameInfo,
26080b57cec5SDimitry Andric                                           EndMemberLookup, CandidateSet,
26090b57cec5SDimitry Andric                                           EndRange, EndExpr);
26100b57cec5SDimitry Andric     if (RangeStatus != Sema::FRS_Success) {
26110b57cec5SDimitry Andric       if (RangeStatus == Sema::FRS_DiagnosticIssued)
26120b57cec5SDimitry Andric         SemaRef.Diag(EndRange->getBeginLoc(), diag::note_in_for_range)
26130b57cec5SDimitry Andric             << ColonLoc << BEF_end << EndRange->getType();
26140b57cec5SDimitry Andric       return RangeStatus;
26150b57cec5SDimitry Andric     }
26160b57cec5SDimitry Andric     if (FinishForRangeVarDecl(SemaRef, EndVar, EndExpr->get(), ColonLoc,
26170b57cec5SDimitry Andric                               diag::err_for_range_iter_deduction_failure)) {
26180b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(SemaRef, EndExpr->get(), *BEF);
26190b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26200b57cec5SDimitry Andric     }
26210b57cec5SDimitry Andric     return Sema::FRS_Success;
26220b57cec5SDimitry Andric   };
26230b57cec5SDimitry Andric 
26240b57cec5SDimitry Andric   if (CXXRecordDecl *D = RangeType->getAsCXXRecordDecl()) {
26250b57cec5SDimitry Andric     // - if _RangeT is a class type, the unqualified-ids begin and end are
26260b57cec5SDimitry Andric     //   looked up in the scope of class _RangeT as if by class member access
26270b57cec5SDimitry Andric     //   lookup (3.4.5), and if either (or both) finds at least one
26280b57cec5SDimitry Andric     //   declaration, begin-expr and end-expr are __range.begin() and
26290b57cec5SDimitry Andric     //   __range.end(), respectively;
26300b57cec5SDimitry Andric     SemaRef.LookupQualifiedName(BeginMemberLookup, D);
26310b57cec5SDimitry Andric     if (BeginMemberLookup.isAmbiguous())
26320b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26330b57cec5SDimitry Andric 
26340b57cec5SDimitry Andric     SemaRef.LookupQualifiedName(EndMemberLookup, D);
26350b57cec5SDimitry Andric     if (EndMemberLookup.isAmbiguous())
26360b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
26370b57cec5SDimitry Andric 
26380b57cec5SDimitry Andric     if (BeginMemberLookup.empty() != EndMemberLookup.empty()) {
26390b57cec5SDimitry Andric       // Look up the non-member form of the member we didn't find, first.
26400b57cec5SDimitry Andric       // This way we prefer a "no viable 'end'" diagnostic over a "i found
26410b57cec5SDimitry Andric       // a 'begin' but ignored it because there was no member 'end'"
26420b57cec5SDimitry Andric       // diagnostic.
26430b57cec5SDimitry Andric       auto BuildNonmember = [&](
26440b57cec5SDimitry Andric           BeginEndFunction BEFFound, LookupResult &Found,
26450b57cec5SDimitry Andric           llvm::function_ref<Sema::ForRangeStatus()> BuildFound,
26460b57cec5SDimitry Andric           llvm::function_ref<Sema::ForRangeStatus()> BuildNotFound) {
26470b57cec5SDimitry Andric         LookupResult OldFound = std::move(Found);
26480b57cec5SDimitry Andric         Found.clear();
26490b57cec5SDimitry Andric 
26500b57cec5SDimitry Andric         if (Sema::ForRangeStatus Result = BuildNotFound())
26510b57cec5SDimitry Andric           return Result;
26520b57cec5SDimitry Andric 
26530b57cec5SDimitry Andric         switch (BuildFound()) {
26540b57cec5SDimitry Andric         case Sema::FRS_Success:
26550b57cec5SDimitry Andric           return Sema::FRS_Success;
26560b57cec5SDimitry Andric 
26570b57cec5SDimitry Andric         case Sema::FRS_NoViableFunction:
26580b57cec5SDimitry Andric           CandidateSet->NoteCandidates(
26590b57cec5SDimitry Andric               PartialDiagnosticAt(BeginRange->getBeginLoc(),
26600b57cec5SDimitry Andric                                   SemaRef.PDiag(diag::err_for_range_invalid)
26610b57cec5SDimitry Andric                                       << BeginRange->getType() << BEFFound),
26620b57cec5SDimitry Andric               SemaRef, OCD_AllCandidates, BeginRange);
26630b57cec5SDimitry Andric           LLVM_FALLTHROUGH;
26640b57cec5SDimitry Andric 
26650b57cec5SDimitry Andric         case Sema::FRS_DiagnosticIssued:
26660b57cec5SDimitry Andric           for (NamedDecl *D : OldFound) {
26670b57cec5SDimitry Andric             SemaRef.Diag(D->getLocation(),
26680b57cec5SDimitry Andric                          diag::note_for_range_member_begin_end_ignored)
26690b57cec5SDimitry Andric                 << BeginRange->getType() << BEFFound;
26700b57cec5SDimitry Andric           }
26710b57cec5SDimitry Andric           return Sema::FRS_DiagnosticIssued;
26720b57cec5SDimitry Andric         }
26730b57cec5SDimitry Andric         llvm_unreachable("unexpected ForRangeStatus");
26740b57cec5SDimitry Andric       };
26750b57cec5SDimitry Andric       if (BeginMemberLookup.empty())
26760b57cec5SDimitry Andric         return BuildNonmember(BEF_end, EndMemberLookup, BuildEnd, BuildBegin);
26770b57cec5SDimitry Andric       return BuildNonmember(BEF_begin, BeginMemberLookup, BuildBegin, BuildEnd);
26780b57cec5SDimitry Andric     }
26790b57cec5SDimitry Andric   } else {
26800b57cec5SDimitry Andric     // - otherwise, begin-expr and end-expr are begin(__range) and
26810b57cec5SDimitry Andric     //   end(__range), respectively, where begin and end are looked up with
26820b57cec5SDimitry Andric     //   argument-dependent lookup (3.4.2). For the purposes of this name
26830b57cec5SDimitry Andric     //   lookup, namespace std is an associated namespace.
26840b57cec5SDimitry Andric   }
26850b57cec5SDimitry Andric 
26860b57cec5SDimitry Andric   if (Sema::ForRangeStatus Result = BuildBegin())
26870b57cec5SDimitry Andric     return Result;
26880b57cec5SDimitry Andric   return BuildEnd();
26890b57cec5SDimitry Andric }
26900b57cec5SDimitry Andric 
26910b57cec5SDimitry Andric /// Speculatively attempt to dereference an invalid range expression.
26920b57cec5SDimitry Andric /// If the attempt fails, this function will return a valid, null StmtResult
26930b57cec5SDimitry Andric /// and emit no diagnostics.
26940b57cec5SDimitry Andric static StmtResult RebuildForRangeWithDereference(Sema &SemaRef, Scope *S,
26950b57cec5SDimitry Andric                                                  SourceLocation ForLoc,
26960b57cec5SDimitry Andric                                                  SourceLocation CoawaitLoc,
26970b57cec5SDimitry Andric                                                  Stmt *InitStmt,
26980b57cec5SDimitry Andric                                                  Stmt *LoopVarDecl,
26990b57cec5SDimitry Andric                                                  SourceLocation ColonLoc,
27000b57cec5SDimitry Andric                                                  Expr *Range,
27010b57cec5SDimitry Andric                                                  SourceLocation RangeLoc,
27020b57cec5SDimitry Andric                                                  SourceLocation RParenLoc) {
27030b57cec5SDimitry Andric   // Determine whether we can rebuild the for-range statement with a
27040b57cec5SDimitry Andric   // dereferenced range expression.
27050b57cec5SDimitry Andric   ExprResult AdjustedRange;
27060b57cec5SDimitry Andric   {
27070b57cec5SDimitry Andric     Sema::SFINAETrap Trap(SemaRef);
27080b57cec5SDimitry Andric 
27090b57cec5SDimitry Andric     AdjustedRange = SemaRef.BuildUnaryOp(S, RangeLoc, UO_Deref, Range);
27100b57cec5SDimitry Andric     if (AdjustedRange.isInvalid())
27110b57cec5SDimitry Andric       return StmtResult();
27120b57cec5SDimitry Andric 
27130b57cec5SDimitry Andric     StmtResult SR = SemaRef.ActOnCXXForRangeStmt(
27140b57cec5SDimitry Andric         S, ForLoc, CoawaitLoc, InitStmt, LoopVarDecl, ColonLoc,
27150b57cec5SDimitry Andric         AdjustedRange.get(), RParenLoc, Sema::BFRK_Check);
27160b57cec5SDimitry Andric     if (SR.isInvalid())
27170b57cec5SDimitry Andric       return StmtResult();
27180b57cec5SDimitry Andric   }
27190b57cec5SDimitry Andric 
27200b57cec5SDimitry Andric   // The attempt to dereference worked well enough that it could produce a valid
27210b57cec5SDimitry Andric   // loop. Produce a fixit, and rebuild the loop with diagnostics enabled, in
27220b57cec5SDimitry Andric   // case there are any other (non-fatal) problems with it.
27230b57cec5SDimitry Andric   SemaRef.Diag(RangeLoc, diag::err_for_range_dereference)
27240b57cec5SDimitry Andric     << Range->getType() << FixItHint::CreateInsertion(RangeLoc, "*");
27250b57cec5SDimitry Andric   return SemaRef.ActOnCXXForRangeStmt(
27260b57cec5SDimitry Andric       S, ForLoc, CoawaitLoc, InitStmt, LoopVarDecl, ColonLoc,
27270b57cec5SDimitry Andric       AdjustedRange.get(), RParenLoc, Sema::BFRK_Rebuild);
27280b57cec5SDimitry Andric }
27290b57cec5SDimitry Andric 
27300b57cec5SDimitry Andric /// BuildCXXForRangeStmt - Build or instantiate a C++11 for-range statement.
27310b57cec5SDimitry Andric StmtResult Sema::BuildCXXForRangeStmt(SourceLocation ForLoc,
27320b57cec5SDimitry Andric                                       SourceLocation CoawaitLoc, Stmt *InitStmt,
27330b57cec5SDimitry Andric                                       SourceLocation ColonLoc, Stmt *RangeDecl,
27340b57cec5SDimitry Andric                                       Stmt *Begin, Stmt *End, Expr *Cond,
27350b57cec5SDimitry Andric                                       Expr *Inc, Stmt *LoopVarDecl,
27360b57cec5SDimitry Andric                                       SourceLocation RParenLoc,
27370b57cec5SDimitry Andric                                       BuildForRangeKind Kind) {
27380b57cec5SDimitry Andric   // FIXME: This should not be used during template instantiation. We should
27390b57cec5SDimitry Andric   // pick up the set of unqualified lookup results for the != and + operators
27400b57cec5SDimitry Andric   // in the initial parse.
27410b57cec5SDimitry Andric   //
27420b57cec5SDimitry Andric   // Testcase (accepts-invalid):
27430b57cec5SDimitry Andric   //   template<typename T> void f() { for (auto x : T()) {} }
27440b57cec5SDimitry Andric   //   namespace N { struct X { X begin(); X end(); int operator*(); }; }
27450b57cec5SDimitry Andric   //   bool operator!=(N::X, N::X); void operator++(N::X);
27460b57cec5SDimitry Andric   //   void g() { f<N::X>(); }
27470b57cec5SDimitry Andric   Scope *S = getCurScope();
27480b57cec5SDimitry Andric 
27490b57cec5SDimitry Andric   DeclStmt *RangeDS = cast<DeclStmt>(RangeDecl);
27500b57cec5SDimitry Andric   VarDecl *RangeVar = cast<VarDecl>(RangeDS->getSingleDecl());
27510b57cec5SDimitry Andric   QualType RangeVarType = RangeVar->getType();
27520b57cec5SDimitry Andric 
27530b57cec5SDimitry Andric   DeclStmt *LoopVarDS = cast<DeclStmt>(LoopVarDecl);
27540b57cec5SDimitry Andric   VarDecl *LoopVar = cast<VarDecl>(LoopVarDS->getSingleDecl());
27550b57cec5SDimitry Andric 
27560b57cec5SDimitry Andric   StmtResult BeginDeclStmt = Begin;
27570b57cec5SDimitry Andric   StmtResult EndDeclStmt = End;
27580b57cec5SDimitry Andric   ExprResult NotEqExpr = Cond, IncrExpr = Inc;
27590b57cec5SDimitry Andric 
27600b57cec5SDimitry Andric   if (RangeVarType->isDependentType()) {
27610b57cec5SDimitry Andric     // The range is implicitly used as a placeholder when it is dependent.
27620b57cec5SDimitry Andric     RangeVar->markUsed(Context);
27630b57cec5SDimitry Andric 
27640b57cec5SDimitry Andric     // Deduce any 'auto's in the loop variable as 'DependentTy'. We'll fill
27650b57cec5SDimitry Andric     // them in properly when we instantiate the loop.
27660b57cec5SDimitry Andric     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
27670b57cec5SDimitry Andric       if (auto *DD = dyn_cast<DecompositionDecl>(LoopVar))
27680b57cec5SDimitry Andric         for (auto *Binding : DD->bindings())
27690b57cec5SDimitry Andric           Binding->setType(Context.DependentTy);
2770349cc55cSDimitry Andric       LoopVar->setType(SubstAutoTypeDependent(LoopVar->getType()));
27710b57cec5SDimitry Andric     }
27720b57cec5SDimitry Andric   } else if (!BeginDeclStmt.get()) {
27730b57cec5SDimitry Andric     SourceLocation RangeLoc = RangeVar->getLocation();
27740b57cec5SDimitry Andric 
27750b57cec5SDimitry Andric     const QualType RangeVarNonRefType = RangeVarType.getNonReferenceType();
27760b57cec5SDimitry Andric 
27770b57cec5SDimitry Andric     ExprResult BeginRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
27780b57cec5SDimitry Andric                                                 VK_LValue, ColonLoc);
27790b57cec5SDimitry Andric     if (BeginRangeRef.isInvalid())
27800b57cec5SDimitry Andric       return StmtError();
27810b57cec5SDimitry Andric 
27820b57cec5SDimitry Andric     ExprResult EndRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
27830b57cec5SDimitry Andric                                               VK_LValue, ColonLoc);
27840b57cec5SDimitry Andric     if (EndRangeRef.isInvalid())
27850b57cec5SDimitry Andric       return StmtError();
27860b57cec5SDimitry Andric 
27870b57cec5SDimitry Andric     QualType AutoType = Context.getAutoDeductType();
27880b57cec5SDimitry Andric     Expr *Range = RangeVar->getInit();
27890b57cec5SDimitry Andric     if (!Range)
27900b57cec5SDimitry Andric       return StmtError();
27910b57cec5SDimitry Andric     QualType RangeType = Range->getType();
27920b57cec5SDimitry Andric 
27930b57cec5SDimitry Andric     if (RequireCompleteType(RangeLoc, RangeType,
27940b57cec5SDimitry Andric                             diag::err_for_range_incomplete_type))
27950b57cec5SDimitry Andric       return StmtError();
27960b57cec5SDimitry Andric 
27970b57cec5SDimitry Andric     // Build auto __begin = begin-expr, __end = end-expr.
27980b57cec5SDimitry Andric     // Divide by 2, since the variables are in the inner scope (loop body).
27990b57cec5SDimitry Andric     const auto DepthStr = std::to_string(S->getDepth() / 2);
28000b57cec5SDimitry Andric     VarDecl *BeginVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
28010b57cec5SDimitry Andric                                              std::string("__begin") + DepthStr);
28020b57cec5SDimitry Andric     VarDecl *EndVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
28030b57cec5SDimitry Andric                                            std::string("__end") + DepthStr);
28040b57cec5SDimitry Andric 
28050b57cec5SDimitry Andric     // Build begin-expr and end-expr and attach to __begin and __end variables.
28060b57cec5SDimitry Andric     ExprResult BeginExpr, EndExpr;
28070b57cec5SDimitry Andric     if (const ArrayType *UnqAT = RangeType->getAsArrayTypeUnsafe()) {
28080b57cec5SDimitry Andric       // - if _RangeT is an array type, begin-expr and end-expr are __range and
28090b57cec5SDimitry Andric       //   __range + __bound, respectively, where __bound is the array bound. If
28100b57cec5SDimitry Andric       //   _RangeT is an array of unknown size or an array of incomplete type,
28110b57cec5SDimitry Andric       //   the program is ill-formed;
28120b57cec5SDimitry Andric 
28130b57cec5SDimitry Andric       // begin-expr is __range.
28140b57cec5SDimitry Andric       BeginExpr = BeginRangeRef;
28150b57cec5SDimitry Andric       if (!CoawaitLoc.isInvalid()) {
28160b57cec5SDimitry Andric         BeginExpr = ActOnCoawaitExpr(S, ColonLoc, BeginExpr.get());
28170b57cec5SDimitry Andric         if (BeginExpr.isInvalid())
28180b57cec5SDimitry Andric           return StmtError();
28190b57cec5SDimitry Andric       }
28200b57cec5SDimitry Andric       if (FinishForRangeVarDecl(*this, BeginVar, BeginRangeRef.get(), ColonLoc,
28210b57cec5SDimitry Andric                                 diag::err_for_range_iter_deduction_failure)) {
28220b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
28230b57cec5SDimitry Andric         return StmtError();
28240b57cec5SDimitry Andric       }
28250b57cec5SDimitry Andric 
28260b57cec5SDimitry Andric       // Find the array bound.
28270b57cec5SDimitry Andric       ExprResult BoundExpr;
28280b57cec5SDimitry Andric       if (const ConstantArrayType *CAT = dyn_cast<ConstantArrayType>(UnqAT))
28290b57cec5SDimitry Andric         BoundExpr = IntegerLiteral::Create(
28300b57cec5SDimitry Andric             Context, CAT->getSize(), Context.getPointerDiffType(), RangeLoc);
28310b57cec5SDimitry Andric       else if (const VariableArrayType *VAT =
28320b57cec5SDimitry Andric                dyn_cast<VariableArrayType>(UnqAT)) {
28330b57cec5SDimitry Andric         // For a variably modified type we can't just use the expression within
28340b57cec5SDimitry Andric         // the array bounds, since we don't want that to be re-evaluated here.
28350b57cec5SDimitry Andric         // Rather, we need to determine what it was when the array was first
28360b57cec5SDimitry Andric         // created - so we resort to using sizeof(vla)/sizeof(element).
28370b57cec5SDimitry Andric         // For e.g.
28380b57cec5SDimitry Andric         //  void f(int b) {
28390b57cec5SDimitry Andric         //    int vla[b];
28400b57cec5SDimitry Andric         //    b = -1;   <-- This should not affect the num of iterations below
28410b57cec5SDimitry Andric         //    for (int &c : vla) { .. }
28420b57cec5SDimitry Andric         //  }
28430b57cec5SDimitry Andric 
28440b57cec5SDimitry Andric         // FIXME: This results in codegen generating IR that recalculates the
28450b57cec5SDimitry Andric         // run-time number of elements (as opposed to just using the IR Value
28460b57cec5SDimitry Andric         // that corresponds to the run-time value of each bound that was
28470b57cec5SDimitry Andric         // generated when the array was created.) If this proves too embarrassing
28480b57cec5SDimitry Andric         // even for unoptimized IR, consider passing a magic-value/cookie to
28490b57cec5SDimitry Andric         // codegen that then knows to simply use that initial llvm::Value (that
28500b57cec5SDimitry Andric         // corresponds to the bound at time of array creation) within
28510b57cec5SDimitry Andric         // getelementptr.  But be prepared to pay the price of increasing a
28520b57cec5SDimitry Andric         // customized form of coupling between the two components - which  could
28530b57cec5SDimitry Andric         // be hard to maintain as the codebase evolves.
28540b57cec5SDimitry Andric 
28550b57cec5SDimitry Andric         ExprResult SizeOfVLAExprR = ActOnUnaryExprOrTypeTraitExpr(
28560b57cec5SDimitry Andric             EndVar->getLocation(), UETT_SizeOf,
28570b57cec5SDimitry Andric             /*IsType=*/true,
28580b57cec5SDimitry Andric             CreateParsedType(VAT->desugar(), Context.getTrivialTypeSourceInfo(
28590b57cec5SDimitry Andric                                                  VAT->desugar(), RangeLoc))
28600b57cec5SDimitry Andric                 .getAsOpaquePtr(),
28610b57cec5SDimitry Andric             EndVar->getSourceRange());
28620b57cec5SDimitry Andric         if (SizeOfVLAExprR.isInvalid())
28630b57cec5SDimitry Andric           return StmtError();
28640b57cec5SDimitry Andric 
28650b57cec5SDimitry Andric         ExprResult SizeOfEachElementExprR = ActOnUnaryExprOrTypeTraitExpr(
28660b57cec5SDimitry Andric             EndVar->getLocation(), UETT_SizeOf,
28670b57cec5SDimitry Andric             /*IsType=*/true,
28680b57cec5SDimitry Andric             CreateParsedType(VAT->desugar(),
28690b57cec5SDimitry Andric                              Context.getTrivialTypeSourceInfo(
28700b57cec5SDimitry Andric                                  VAT->getElementType(), RangeLoc))
28710b57cec5SDimitry Andric                 .getAsOpaquePtr(),
28720b57cec5SDimitry Andric             EndVar->getSourceRange());
28730b57cec5SDimitry Andric         if (SizeOfEachElementExprR.isInvalid())
28740b57cec5SDimitry Andric           return StmtError();
28750b57cec5SDimitry Andric 
28760b57cec5SDimitry Andric         BoundExpr =
28770b57cec5SDimitry Andric             ActOnBinOp(S, EndVar->getLocation(), tok::slash,
28780b57cec5SDimitry Andric                        SizeOfVLAExprR.get(), SizeOfEachElementExprR.get());
28790b57cec5SDimitry Andric         if (BoundExpr.isInvalid())
28800b57cec5SDimitry Andric           return StmtError();
28810b57cec5SDimitry Andric 
28820b57cec5SDimitry Andric       } else {
28830b57cec5SDimitry Andric         // Can't be a DependentSizedArrayType or an IncompleteArrayType since
28840b57cec5SDimitry Andric         // UnqAT is not incomplete and Range is not type-dependent.
28850b57cec5SDimitry Andric         llvm_unreachable("Unexpected array type in for-range");
28860b57cec5SDimitry Andric       }
28870b57cec5SDimitry Andric 
28880b57cec5SDimitry Andric       // end-expr is __range + __bound.
28890b57cec5SDimitry Andric       EndExpr = ActOnBinOp(S, ColonLoc, tok::plus, EndRangeRef.get(),
28900b57cec5SDimitry Andric                            BoundExpr.get());
28910b57cec5SDimitry Andric       if (EndExpr.isInvalid())
28920b57cec5SDimitry Andric         return StmtError();
28930b57cec5SDimitry Andric       if (FinishForRangeVarDecl(*this, EndVar, EndExpr.get(), ColonLoc,
28940b57cec5SDimitry Andric                                 diag::err_for_range_iter_deduction_failure)) {
28950b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
28960b57cec5SDimitry Andric         return StmtError();
28970b57cec5SDimitry Andric       }
28980b57cec5SDimitry Andric     } else {
28990b57cec5SDimitry Andric       OverloadCandidateSet CandidateSet(RangeLoc,
29000b57cec5SDimitry Andric                                         OverloadCandidateSet::CSK_Normal);
29010b57cec5SDimitry Andric       BeginEndFunction BEFFailure;
29020b57cec5SDimitry Andric       ForRangeStatus RangeStatus = BuildNonArrayForRange(
29030b57cec5SDimitry Andric           *this, BeginRangeRef.get(), EndRangeRef.get(), RangeType, BeginVar,
29040b57cec5SDimitry Andric           EndVar, ColonLoc, CoawaitLoc, &CandidateSet, &BeginExpr, &EndExpr,
29050b57cec5SDimitry Andric           &BEFFailure);
29060b57cec5SDimitry Andric 
29070b57cec5SDimitry Andric       if (Kind == BFRK_Build && RangeStatus == FRS_NoViableFunction &&
29080b57cec5SDimitry Andric           BEFFailure == BEF_begin) {
29090b57cec5SDimitry Andric         // If the range is being built from an array parameter, emit a
29100b57cec5SDimitry Andric         // a diagnostic that it is being treated as a pointer.
29110b57cec5SDimitry Andric         if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Range)) {
29120b57cec5SDimitry Andric           if (ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
29130b57cec5SDimitry Andric             QualType ArrayTy = PVD->getOriginalType();
29140b57cec5SDimitry Andric             QualType PointerTy = PVD->getType();
29150b57cec5SDimitry Andric             if (PointerTy->isPointerType() && ArrayTy->isArrayType()) {
29160b57cec5SDimitry Andric               Diag(Range->getBeginLoc(), diag::err_range_on_array_parameter)
29170b57cec5SDimitry Andric                   << RangeLoc << PVD << ArrayTy << PointerTy;
29180b57cec5SDimitry Andric               Diag(PVD->getLocation(), diag::note_declared_at);
29190b57cec5SDimitry Andric               return StmtError();
29200b57cec5SDimitry Andric             }
29210b57cec5SDimitry Andric           }
29220b57cec5SDimitry Andric         }
29230b57cec5SDimitry Andric 
29240b57cec5SDimitry Andric         // If building the range failed, try dereferencing the range expression
29250b57cec5SDimitry Andric         // unless a diagnostic was issued or the end function is problematic.
29260b57cec5SDimitry Andric         StmtResult SR = RebuildForRangeWithDereference(*this, S, ForLoc,
29270b57cec5SDimitry Andric                                                        CoawaitLoc, InitStmt,
29280b57cec5SDimitry Andric                                                        LoopVarDecl, ColonLoc,
29290b57cec5SDimitry Andric                                                        Range, RangeLoc,
29300b57cec5SDimitry Andric                                                        RParenLoc);
29310b57cec5SDimitry Andric         if (SR.isInvalid() || SR.isUsable())
29320b57cec5SDimitry Andric           return SR;
29330b57cec5SDimitry Andric       }
29340b57cec5SDimitry Andric 
29350b57cec5SDimitry Andric       // Otherwise, emit diagnostics if we haven't already.
29360b57cec5SDimitry Andric       if (RangeStatus == FRS_NoViableFunction) {
29370b57cec5SDimitry Andric         Expr *Range = BEFFailure ? EndRangeRef.get() : BeginRangeRef.get();
29380b57cec5SDimitry Andric         CandidateSet.NoteCandidates(
29390b57cec5SDimitry Andric             PartialDiagnosticAt(Range->getBeginLoc(),
29400b57cec5SDimitry Andric                                 PDiag(diag::err_for_range_invalid)
29410b57cec5SDimitry Andric                                     << RangeLoc << Range->getType()
29420b57cec5SDimitry Andric                                     << BEFFailure),
29430b57cec5SDimitry Andric             *this, OCD_AllCandidates, Range);
29440b57cec5SDimitry Andric       }
29450b57cec5SDimitry Andric       // Return an error if no fix was discovered.
29460b57cec5SDimitry Andric       if (RangeStatus != FRS_Success)
29470b57cec5SDimitry Andric         return StmtError();
29480b57cec5SDimitry Andric     }
29490b57cec5SDimitry Andric 
29500b57cec5SDimitry Andric     assert(!BeginExpr.isInvalid() && !EndExpr.isInvalid() &&
29510b57cec5SDimitry Andric            "invalid range expression in for loop");
29520b57cec5SDimitry Andric 
29530b57cec5SDimitry Andric     // C++11 [dcl.spec.auto]p7: BeginType and EndType must be the same.
29540b57cec5SDimitry Andric     // C++1z removes this restriction.
29550b57cec5SDimitry Andric     QualType BeginType = BeginVar->getType(), EndType = EndVar->getType();
29560b57cec5SDimitry Andric     if (!Context.hasSameType(BeginType, EndType)) {
29570b57cec5SDimitry Andric       Diag(RangeLoc, getLangOpts().CPlusPlus17
29580b57cec5SDimitry Andric                          ? diag::warn_for_range_begin_end_types_differ
29590b57cec5SDimitry Andric                          : diag::ext_for_range_begin_end_types_differ)
29600b57cec5SDimitry Andric           << BeginType << EndType;
29610b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
29620b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
29630b57cec5SDimitry Andric     }
29640b57cec5SDimitry Andric 
29650b57cec5SDimitry Andric     BeginDeclStmt =
29660b57cec5SDimitry Andric         ActOnDeclStmt(ConvertDeclToDeclGroup(BeginVar), ColonLoc, ColonLoc);
29670b57cec5SDimitry Andric     EndDeclStmt =
29680b57cec5SDimitry Andric         ActOnDeclStmt(ConvertDeclToDeclGroup(EndVar), ColonLoc, ColonLoc);
29690b57cec5SDimitry Andric 
29700b57cec5SDimitry Andric     const QualType BeginRefNonRefType = BeginType.getNonReferenceType();
29710b57cec5SDimitry Andric     ExprResult BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
29720b57cec5SDimitry Andric                                            VK_LValue, ColonLoc);
29730b57cec5SDimitry Andric     if (BeginRef.isInvalid())
29740b57cec5SDimitry Andric       return StmtError();
29750b57cec5SDimitry Andric 
29760b57cec5SDimitry Andric     ExprResult EndRef = BuildDeclRefExpr(EndVar, EndType.getNonReferenceType(),
29770b57cec5SDimitry Andric                                          VK_LValue, ColonLoc);
29780b57cec5SDimitry Andric     if (EndRef.isInvalid())
29790b57cec5SDimitry Andric       return StmtError();
29800b57cec5SDimitry Andric 
29810b57cec5SDimitry Andric     // Build and check __begin != __end expression.
29820b57cec5SDimitry Andric     NotEqExpr = ActOnBinOp(S, ColonLoc, tok::exclaimequal,
29830b57cec5SDimitry Andric                            BeginRef.get(), EndRef.get());
29840b57cec5SDimitry Andric     if (!NotEqExpr.isInvalid())
29850b57cec5SDimitry Andric       NotEqExpr = CheckBooleanCondition(ColonLoc, NotEqExpr.get());
29860b57cec5SDimitry Andric     if (!NotEqExpr.isInvalid())
29870b57cec5SDimitry Andric       NotEqExpr =
29880b57cec5SDimitry Andric           ActOnFinishFullExpr(NotEqExpr.get(), /*DiscardedValue*/ false);
29890b57cec5SDimitry Andric     if (NotEqExpr.isInvalid()) {
29900b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
29910b57cec5SDimitry Andric         << RangeLoc << 0 << BeginRangeRef.get()->getType();
29920b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
29930b57cec5SDimitry Andric       if (!Context.hasSameType(BeginType, EndType))
29940b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
29950b57cec5SDimitry Andric       return StmtError();
29960b57cec5SDimitry Andric     }
29970b57cec5SDimitry Andric 
29980b57cec5SDimitry Andric     // Build and check ++__begin expression.
29990b57cec5SDimitry Andric     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
30000b57cec5SDimitry Andric                                 VK_LValue, ColonLoc);
30010b57cec5SDimitry Andric     if (BeginRef.isInvalid())
30020b57cec5SDimitry Andric       return StmtError();
30030b57cec5SDimitry Andric 
30040b57cec5SDimitry Andric     IncrExpr = ActOnUnaryOp(S, ColonLoc, tok::plusplus, BeginRef.get());
30050b57cec5SDimitry Andric     if (!IncrExpr.isInvalid() && CoawaitLoc.isValid())
30060b57cec5SDimitry Andric       // FIXME: getCurScope() should not be used during template instantiation.
30070b57cec5SDimitry Andric       // We should pick up the set of unqualified lookup results for operator
30080b57cec5SDimitry Andric       // co_await during the initial parse.
30090b57cec5SDimitry Andric       IncrExpr = ActOnCoawaitExpr(S, CoawaitLoc, IncrExpr.get());
30100b57cec5SDimitry Andric     if (!IncrExpr.isInvalid())
30110b57cec5SDimitry Andric       IncrExpr = ActOnFinishFullExpr(IncrExpr.get(), /*DiscardedValue*/ false);
30120b57cec5SDimitry Andric     if (IncrExpr.isInvalid()) {
30130b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
30140b57cec5SDimitry Andric         << RangeLoc << 2 << BeginRangeRef.get()->getType() ;
30150b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
30160b57cec5SDimitry Andric       return StmtError();
30170b57cec5SDimitry Andric     }
30180b57cec5SDimitry Andric 
30190b57cec5SDimitry Andric     // Build and check *__begin  expression.
30200b57cec5SDimitry Andric     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
30210b57cec5SDimitry Andric                                 VK_LValue, ColonLoc);
30220b57cec5SDimitry Andric     if (BeginRef.isInvalid())
30230b57cec5SDimitry Andric       return StmtError();
30240b57cec5SDimitry Andric 
30250b57cec5SDimitry Andric     ExprResult DerefExpr = ActOnUnaryOp(S, ColonLoc, tok::star, BeginRef.get());
30260b57cec5SDimitry Andric     if (DerefExpr.isInvalid()) {
30270b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
30280b57cec5SDimitry Andric         << RangeLoc << 1 << BeginRangeRef.get()->getType();
30290b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
30300b57cec5SDimitry Andric       return StmtError();
30310b57cec5SDimitry Andric     }
30320b57cec5SDimitry Andric 
30330b57cec5SDimitry Andric     // Attach  *__begin  as initializer for VD. Don't touch it if we're just
30340b57cec5SDimitry Andric     // trying to determine whether this would be a valid range.
30350b57cec5SDimitry Andric     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
30360b57cec5SDimitry Andric       AddInitializerToDecl(LoopVar, DerefExpr.get(), /*DirectInit=*/false);
30375ffd83dbSDimitry Andric       if (LoopVar->isInvalidDecl() ||
30385ffd83dbSDimitry Andric           (LoopVar->getInit() && LoopVar->getInit()->containsErrors()))
30390b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
30400b57cec5SDimitry Andric     }
30410b57cec5SDimitry Andric   }
30420b57cec5SDimitry Andric 
30430b57cec5SDimitry Andric   // Don't bother to actually allocate the result if we're just trying to
30440b57cec5SDimitry Andric   // determine whether it would be valid.
30450b57cec5SDimitry Andric   if (Kind == BFRK_Check)
30460b57cec5SDimitry Andric     return StmtResult();
30470b57cec5SDimitry Andric 
3048a7dea167SDimitry Andric   // In OpenMP loop region loop control variable must be private. Perform
3049a7dea167SDimitry Andric   // analysis of first part (if any).
3050a7dea167SDimitry Andric   if (getLangOpts().OpenMP >= 50 && BeginDeclStmt.isUsable())
3051a7dea167SDimitry Andric     ActOnOpenMPLoopInitialization(ForLoc, BeginDeclStmt.get());
3052a7dea167SDimitry Andric 
30530b57cec5SDimitry Andric   return new (Context) CXXForRangeStmt(
30540b57cec5SDimitry Andric       InitStmt, RangeDS, cast_or_null<DeclStmt>(BeginDeclStmt.get()),
30550b57cec5SDimitry Andric       cast_or_null<DeclStmt>(EndDeclStmt.get()), NotEqExpr.get(),
30560b57cec5SDimitry Andric       IncrExpr.get(), LoopVarDS, /*Body=*/nullptr, ForLoc, CoawaitLoc,
30570b57cec5SDimitry Andric       ColonLoc, RParenLoc);
30580b57cec5SDimitry Andric }
30590b57cec5SDimitry Andric 
30600b57cec5SDimitry Andric /// FinishObjCForCollectionStmt - Attach the body to a objective-C foreach
30610b57cec5SDimitry Andric /// statement.
30620b57cec5SDimitry Andric StmtResult Sema::FinishObjCForCollectionStmt(Stmt *S, Stmt *B) {
30630b57cec5SDimitry Andric   if (!S || !B)
30640b57cec5SDimitry Andric     return StmtError();
30650b57cec5SDimitry Andric   ObjCForCollectionStmt * ForStmt = cast<ObjCForCollectionStmt>(S);
30660b57cec5SDimitry Andric 
30670b57cec5SDimitry Andric   ForStmt->setBody(B);
30680b57cec5SDimitry Andric   return S;
30690b57cec5SDimitry Andric }
30700b57cec5SDimitry Andric 
30710b57cec5SDimitry Andric // Warn when the loop variable is a const reference that creates a copy.
30720b57cec5SDimitry Andric // Suggest using the non-reference type for copies.  If a copy can be prevented
30730b57cec5SDimitry Andric // suggest the const reference type that would do so.
30740b57cec5SDimitry Andric // For instance, given "for (const &Foo : Range)", suggest
30750b57cec5SDimitry Andric // "for (const Foo : Range)" to denote a copy is made for the loop.  If
30760b57cec5SDimitry Andric // possible, also suggest "for (const &Bar : Range)" if this type prevents
30770b57cec5SDimitry Andric // the copy altogether.
30780b57cec5SDimitry Andric static void DiagnoseForRangeReferenceVariableCopies(Sema &SemaRef,
30790b57cec5SDimitry Andric                                                     const VarDecl *VD,
30800b57cec5SDimitry Andric                                                     QualType RangeInitType) {
30810b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
30820b57cec5SDimitry Andric   if (!InitExpr)
30830b57cec5SDimitry Andric     return;
30840b57cec5SDimitry Andric 
30850b57cec5SDimitry Andric   QualType VariableType = VD->getType();
30860b57cec5SDimitry Andric 
30870b57cec5SDimitry Andric   if (auto Cleanups = dyn_cast<ExprWithCleanups>(InitExpr))
30880b57cec5SDimitry Andric     if (!Cleanups->cleanupsHaveSideEffects())
30890b57cec5SDimitry Andric       InitExpr = Cleanups->getSubExpr();
30900b57cec5SDimitry Andric 
30910b57cec5SDimitry Andric   const MaterializeTemporaryExpr *MTE =
30920b57cec5SDimitry Andric       dyn_cast<MaterializeTemporaryExpr>(InitExpr);
30930b57cec5SDimitry Andric 
30940b57cec5SDimitry Andric   // No copy made.
30950b57cec5SDimitry Andric   if (!MTE)
30960b57cec5SDimitry Andric     return;
30970b57cec5SDimitry Andric 
3098480093f4SDimitry Andric   const Expr *E = MTE->getSubExpr()->IgnoreImpCasts();
30990b57cec5SDimitry Andric 
31000b57cec5SDimitry Andric   // Searching for either UnaryOperator for dereference of a pointer or
31010b57cec5SDimitry Andric   // CXXOperatorCallExpr for handling iterators.
31020b57cec5SDimitry Andric   while (!isa<CXXOperatorCallExpr>(E) && !isa<UnaryOperator>(E)) {
31030b57cec5SDimitry Andric     if (const CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(E)) {
31040b57cec5SDimitry Andric       E = CCE->getArg(0);
31050b57cec5SDimitry Andric     } else if (const CXXMemberCallExpr *Call = dyn_cast<CXXMemberCallExpr>(E)) {
31060b57cec5SDimitry Andric       const MemberExpr *ME = cast<MemberExpr>(Call->getCallee());
31070b57cec5SDimitry Andric       E = ME->getBase();
31080b57cec5SDimitry Andric     } else {
31090b57cec5SDimitry Andric       const MaterializeTemporaryExpr *MTE = cast<MaterializeTemporaryExpr>(E);
3110480093f4SDimitry Andric       E = MTE->getSubExpr();
31110b57cec5SDimitry Andric     }
31120b57cec5SDimitry Andric     E = E->IgnoreImpCasts();
31130b57cec5SDimitry Andric   }
31140b57cec5SDimitry Andric 
31155ffd83dbSDimitry Andric   QualType ReferenceReturnType;
31160b57cec5SDimitry Andric   if (isa<UnaryOperator>(E)) {
31175ffd83dbSDimitry Andric     ReferenceReturnType = SemaRef.Context.getLValueReferenceType(E->getType());
31180b57cec5SDimitry Andric   } else {
31190b57cec5SDimitry Andric     const CXXOperatorCallExpr *Call = cast<CXXOperatorCallExpr>(E);
31200b57cec5SDimitry Andric     const FunctionDecl *FD = Call->getDirectCallee();
31210b57cec5SDimitry Andric     QualType ReturnType = FD->getReturnType();
31225ffd83dbSDimitry Andric     if (ReturnType->isReferenceType())
31235ffd83dbSDimitry Andric       ReferenceReturnType = ReturnType;
31240b57cec5SDimitry Andric   }
31250b57cec5SDimitry Andric 
31265ffd83dbSDimitry Andric   if (!ReferenceReturnType.isNull()) {
31270b57cec5SDimitry Andric     // Loop variable creates a temporary.  Suggest either to go with
31280b57cec5SDimitry Andric     // non-reference loop variable to indicate a copy is made, or
31295ffd83dbSDimitry Andric     // the correct type to bind a const reference.
31305ffd83dbSDimitry Andric     SemaRef.Diag(VD->getLocation(),
31315ffd83dbSDimitry Andric                  diag::warn_for_range_const_ref_binds_temp_built_from_ref)
31325ffd83dbSDimitry Andric         << VD << VariableType << ReferenceReturnType;
31330b57cec5SDimitry Andric     QualType NonReferenceType = VariableType.getNonReferenceType();
31340b57cec5SDimitry Andric     NonReferenceType.removeLocalConst();
31350b57cec5SDimitry Andric     QualType NewReferenceType =
31360b57cec5SDimitry Andric         SemaRef.Context.getLValueReferenceType(E->getType().withConst());
31370b57cec5SDimitry Andric     SemaRef.Diag(VD->getBeginLoc(), diag::note_use_type_or_non_reference)
3138480093f4SDimitry Andric         << NonReferenceType << NewReferenceType << VD->getSourceRange()
3139480093f4SDimitry Andric         << FixItHint::CreateRemoval(VD->getTypeSpecEndLoc());
3140480093f4SDimitry Andric   } else if (!VariableType->isRValueReferenceType()) {
31410b57cec5SDimitry Andric     // The range always returns a copy, so a temporary is always created.
31420b57cec5SDimitry Andric     // Suggest removing the reference from the loop variable.
3143480093f4SDimitry Andric     // If the type is a rvalue reference do not warn since that changes the
3144480093f4SDimitry Andric     // semantic of the code.
31455ffd83dbSDimitry Andric     SemaRef.Diag(VD->getLocation(), diag::warn_for_range_ref_binds_ret_temp)
31460b57cec5SDimitry Andric         << VD << RangeInitType;
31470b57cec5SDimitry Andric     QualType NonReferenceType = VariableType.getNonReferenceType();
31480b57cec5SDimitry Andric     NonReferenceType.removeLocalConst();
31490b57cec5SDimitry Andric     SemaRef.Diag(VD->getBeginLoc(), diag::note_use_non_reference_type)
3150480093f4SDimitry Andric         << NonReferenceType << VD->getSourceRange()
3151480093f4SDimitry Andric         << FixItHint::CreateRemoval(VD->getTypeSpecEndLoc());
31520b57cec5SDimitry Andric   }
31530b57cec5SDimitry Andric }
31540b57cec5SDimitry Andric 
3155480093f4SDimitry Andric /// Determines whether the @p VariableType's declaration is a record with the
3156480093f4SDimitry Andric /// clang::trivial_abi attribute.
3157480093f4SDimitry Andric static bool hasTrivialABIAttr(QualType VariableType) {
3158480093f4SDimitry Andric   if (CXXRecordDecl *RD = VariableType->getAsCXXRecordDecl())
3159480093f4SDimitry Andric     return RD->hasAttr<TrivialABIAttr>();
3160480093f4SDimitry Andric 
3161480093f4SDimitry Andric   return false;
3162480093f4SDimitry Andric }
3163480093f4SDimitry Andric 
31640b57cec5SDimitry Andric // Warns when the loop variable can be changed to a reference type to
31650b57cec5SDimitry Andric // prevent a copy.  For instance, if given "for (const Foo x : Range)" suggest
31660b57cec5SDimitry Andric // "for (const Foo &x : Range)" if this form does not make a copy.
31670b57cec5SDimitry Andric static void DiagnoseForRangeConstVariableCopies(Sema &SemaRef,
31680b57cec5SDimitry Andric                                                 const VarDecl *VD) {
31690b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
31700b57cec5SDimitry Andric   if (!InitExpr)
31710b57cec5SDimitry Andric     return;
31720b57cec5SDimitry Andric 
31730b57cec5SDimitry Andric   QualType VariableType = VD->getType();
31740b57cec5SDimitry Andric 
31750b57cec5SDimitry Andric   if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(InitExpr)) {
31760b57cec5SDimitry Andric     if (!CE->getConstructor()->isCopyConstructor())
31770b57cec5SDimitry Andric       return;
31780b57cec5SDimitry Andric   } else if (const CastExpr *CE = dyn_cast<CastExpr>(InitExpr)) {
31790b57cec5SDimitry Andric     if (CE->getCastKind() != CK_LValueToRValue)
31800b57cec5SDimitry Andric       return;
31810b57cec5SDimitry Andric   } else {
31820b57cec5SDimitry Andric     return;
31830b57cec5SDimitry Andric   }
31840b57cec5SDimitry Andric 
3185480093f4SDimitry Andric   // Small trivially copyable types are cheap to copy. Do not emit the
3186480093f4SDimitry Andric   // diagnostic for these instances. 64 bytes is a common size of a cache line.
3187480093f4SDimitry Andric   // (The function `getTypeSize` returns the size in bits.)
3188480093f4SDimitry Andric   ASTContext &Ctx = SemaRef.Context;
3189480093f4SDimitry Andric   if (Ctx.getTypeSize(VariableType) <= 64 * 8 &&
3190480093f4SDimitry Andric       (VariableType.isTriviallyCopyableType(Ctx) ||
3191480093f4SDimitry Andric        hasTrivialABIAttr(VariableType)))
31920b57cec5SDimitry Andric     return;
31930b57cec5SDimitry Andric 
31940b57cec5SDimitry Andric   // Suggest changing from a const variable to a const reference variable
31950b57cec5SDimitry Andric   // if doing so will prevent a copy.
31960b57cec5SDimitry Andric   SemaRef.Diag(VD->getLocation(), diag::warn_for_range_copy)
31975ffd83dbSDimitry Andric       << VD << VariableType;
31980b57cec5SDimitry Andric   SemaRef.Diag(VD->getBeginLoc(), diag::note_use_reference_type)
31990b57cec5SDimitry Andric       << SemaRef.Context.getLValueReferenceType(VariableType)
3200480093f4SDimitry Andric       << VD->getSourceRange()
3201480093f4SDimitry Andric       << FixItHint::CreateInsertion(VD->getLocation(), "&");
32020b57cec5SDimitry Andric }
32030b57cec5SDimitry Andric 
32040b57cec5SDimitry Andric /// DiagnoseForRangeVariableCopies - Diagnose three cases and fixes for them.
32050b57cec5SDimitry Andric /// 1) for (const foo &x : foos) where foos only returns a copy.  Suggest
32060b57cec5SDimitry Andric ///    using "const foo x" to show that a copy is made
32070b57cec5SDimitry Andric /// 2) for (const bar &x : foos) where bar is a temporary initialized by bar.
32080b57cec5SDimitry Andric ///    Suggest either "const bar x" to keep the copying or "const foo& x" to
32090b57cec5SDimitry Andric ///    prevent the copy.
32100b57cec5SDimitry Andric /// 3) for (const foo x : foos) where x is constructed from a reference foo.
32110b57cec5SDimitry Andric ///    Suggest "const foo &x" to prevent the copy.
32120b57cec5SDimitry Andric static void DiagnoseForRangeVariableCopies(Sema &SemaRef,
32130b57cec5SDimitry Andric                                            const CXXForRangeStmt *ForStmt) {
321455e4f9d5SDimitry Andric   if (SemaRef.inTemplateInstantiation())
321555e4f9d5SDimitry Andric     return;
321655e4f9d5SDimitry Andric 
32175ffd83dbSDimitry Andric   if (SemaRef.Diags.isIgnored(
32185ffd83dbSDimitry Andric           diag::warn_for_range_const_ref_binds_temp_built_from_ref,
32190b57cec5SDimitry Andric           ForStmt->getBeginLoc()) &&
32205ffd83dbSDimitry Andric       SemaRef.Diags.isIgnored(diag::warn_for_range_ref_binds_ret_temp,
32210b57cec5SDimitry Andric                               ForStmt->getBeginLoc()) &&
32220b57cec5SDimitry Andric       SemaRef.Diags.isIgnored(diag::warn_for_range_copy,
32230b57cec5SDimitry Andric                               ForStmt->getBeginLoc())) {
32240b57cec5SDimitry Andric     return;
32250b57cec5SDimitry Andric   }
32260b57cec5SDimitry Andric 
32270b57cec5SDimitry Andric   const VarDecl *VD = ForStmt->getLoopVariable();
32280b57cec5SDimitry Andric   if (!VD)
32290b57cec5SDimitry Andric     return;
32300b57cec5SDimitry Andric 
32310b57cec5SDimitry Andric   QualType VariableType = VD->getType();
32320b57cec5SDimitry Andric 
32330b57cec5SDimitry Andric   if (VariableType->isIncompleteType())
32340b57cec5SDimitry Andric     return;
32350b57cec5SDimitry Andric 
32360b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
32370b57cec5SDimitry Andric   if (!InitExpr)
32380b57cec5SDimitry Andric     return;
32390b57cec5SDimitry Andric 
324055e4f9d5SDimitry Andric   if (InitExpr->getExprLoc().isMacroID())
324155e4f9d5SDimitry Andric     return;
324255e4f9d5SDimitry Andric 
32430b57cec5SDimitry Andric   if (VariableType->isReferenceType()) {
32440b57cec5SDimitry Andric     DiagnoseForRangeReferenceVariableCopies(SemaRef, VD,
32450b57cec5SDimitry Andric                                             ForStmt->getRangeInit()->getType());
32460b57cec5SDimitry Andric   } else if (VariableType.isConstQualified()) {
32470b57cec5SDimitry Andric     DiagnoseForRangeConstVariableCopies(SemaRef, VD);
32480b57cec5SDimitry Andric   }
32490b57cec5SDimitry Andric }
32500b57cec5SDimitry Andric 
32510b57cec5SDimitry Andric /// FinishCXXForRangeStmt - Attach the body to a C++0x for-range statement.
32520b57cec5SDimitry Andric /// This is a separate step from ActOnCXXForRangeStmt because analysis of the
32530b57cec5SDimitry Andric /// body cannot be performed until after the type of the range variable is
32540b57cec5SDimitry Andric /// determined.
32550b57cec5SDimitry Andric StmtResult Sema::FinishCXXForRangeStmt(Stmt *S, Stmt *B) {
32560b57cec5SDimitry Andric   if (!S || !B)
32570b57cec5SDimitry Andric     return StmtError();
32580b57cec5SDimitry Andric 
32590b57cec5SDimitry Andric   if (isa<ObjCForCollectionStmt>(S))
32600b57cec5SDimitry Andric     return FinishObjCForCollectionStmt(S, B);
32610b57cec5SDimitry Andric 
32620b57cec5SDimitry Andric   CXXForRangeStmt *ForStmt = cast<CXXForRangeStmt>(S);
32630b57cec5SDimitry Andric   ForStmt->setBody(B);
32640b57cec5SDimitry Andric 
32650b57cec5SDimitry Andric   DiagnoseEmptyStmtBody(ForStmt->getRParenLoc(), B,
32660b57cec5SDimitry Andric                         diag::warn_empty_range_based_for_body);
32670b57cec5SDimitry Andric 
32680b57cec5SDimitry Andric   DiagnoseForRangeVariableCopies(*this, ForStmt);
32690b57cec5SDimitry Andric 
32700b57cec5SDimitry Andric   return S;
32710b57cec5SDimitry Andric }
32720b57cec5SDimitry Andric 
32730b57cec5SDimitry Andric StmtResult Sema::ActOnGotoStmt(SourceLocation GotoLoc,
32740b57cec5SDimitry Andric                                SourceLocation LabelLoc,
32750b57cec5SDimitry Andric                                LabelDecl *TheDecl) {
32760b57cec5SDimitry Andric   setFunctionHasBranchIntoScope();
32770b57cec5SDimitry Andric   TheDecl->markUsed(Context);
32780b57cec5SDimitry Andric   return new (Context) GotoStmt(TheDecl, GotoLoc, LabelLoc);
32790b57cec5SDimitry Andric }
32800b57cec5SDimitry Andric 
32810b57cec5SDimitry Andric StmtResult
32820b57cec5SDimitry Andric Sema::ActOnIndirectGotoStmt(SourceLocation GotoLoc, SourceLocation StarLoc,
32830b57cec5SDimitry Andric                             Expr *E) {
32840b57cec5SDimitry Andric   // Convert operand to void*
32850b57cec5SDimitry Andric   if (!E->isTypeDependent()) {
32860b57cec5SDimitry Andric     QualType ETy = E->getType();
32870b57cec5SDimitry Andric     QualType DestTy = Context.getPointerType(Context.VoidTy.withConst());
32880b57cec5SDimitry Andric     ExprResult ExprRes = E;
32890b57cec5SDimitry Andric     AssignConvertType ConvTy =
32900b57cec5SDimitry Andric       CheckSingleAssignmentConstraints(DestTy, ExprRes);
32910b57cec5SDimitry Andric     if (ExprRes.isInvalid())
32920b57cec5SDimitry Andric       return StmtError();
32930b57cec5SDimitry Andric     E = ExprRes.get();
32940b57cec5SDimitry Andric     if (DiagnoseAssignmentResult(ConvTy, StarLoc, DestTy, ETy, E, AA_Passing))
32950b57cec5SDimitry Andric       return StmtError();
32960b57cec5SDimitry Andric   }
32970b57cec5SDimitry Andric 
32980b57cec5SDimitry Andric   ExprResult ExprRes = ActOnFinishFullExpr(E, /*DiscardedValue*/ false);
32990b57cec5SDimitry Andric   if (ExprRes.isInvalid())
33000b57cec5SDimitry Andric     return StmtError();
33010b57cec5SDimitry Andric   E = ExprRes.get();
33020b57cec5SDimitry Andric 
33030b57cec5SDimitry Andric   setFunctionHasIndirectGoto();
33040b57cec5SDimitry Andric 
33050b57cec5SDimitry Andric   return new (Context) IndirectGotoStmt(GotoLoc, StarLoc, E);
33060b57cec5SDimitry Andric }
33070b57cec5SDimitry Andric 
33080b57cec5SDimitry Andric static void CheckJumpOutOfSEHFinally(Sema &S, SourceLocation Loc,
33090b57cec5SDimitry Andric                                      const Scope &DestScope) {
33100b57cec5SDimitry Andric   if (!S.CurrentSEHFinally.empty() &&
33110b57cec5SDimitry Andric       DestScope.Contains(*S.CurrentSEHFinally.back())) {
33120b57cec5SDimitry Andric     S.Diag(Loc, diag::warn_jump_out_of_seh_finally);
33130b57cec5SDimitry Andric   }
33140b57cec5SDimitry Andric }
33150b57cec5SDimitry Andric 
33160b57cec5SDimitry Andric StmtResult
33170b57cec5SDimitry Andric Sema::ActOnContinueStmt(SourceLocation ContinueLoc, Scope *CurScope) {
33180b57cec5SDimitry Andric   Scope *S = CurScope->getContinueParent();
33190b57cec5SDimitry Andric   if (!S) {
33200b57cec5SDimitry Andric     // C99 6.8.6.2p1: A break shall appear only in or as a loop body.
33210b57cec5SDimitry Andric     return StmtError(Diag(ContinueLoc, diag::err_continue_not_in_loop));
33220b57cec5SDimitry Andric   }
332381ad6265SDimitry Andric   if (S->isConditionVarScope()) {
3324fe6060f1SDimitry Andric     // We cannot 'continue;' from within a statement expression in the
3325fe6060f1SDimitry Andric     // initializer of a condition variable because we would jump past the
3326fe6060f1SDimitry Andric     // initialization of that variable.
3327fe6060f1SDimitry Andric     return StmtError(Diag(ContinueLoc, diag::err_continue_from_cond_var_init));
3328fe6060f1SDimitry Andric   }
33290b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, ContinueLoc, *S);
33300b57cec5SDimitry Andric 
33310b57cec5SDimitry Andric   return new (Context) ContinueStmt(ContinueLoc);
33320b57cec5SDimitry Andric }
33330b57cec5SDimitry Andric 
33340b57cec5SDimitry Andric StmtResult
33350b57cec5SDimitry Andric Sema::ActOnBreakStmt(SourceLocation BreakLoc, Scope *CurScope) {
33360b57cec5SDimitry Andric   Scope *S = CurScope->getBreakParent();
33370b57cec5SDimitry Andric   if (!S) {
33380b57cec5SDimitry Andric     // C99 6.8.6.3p1: A break shall appear only in or as a switch/loop body.
33390b57cec5SDimitry Andric     return StmtError(Diag(BreakLoc, diag::err_break_not_in_loop_or_switch));
33400b57cec5SDimitry Andric   }
33410b57cec5SDimitry Andric   if (S->isOpenMPLoopScope())
33420b57cec5SDimitry Andric     return StmtError(Diag(BreakLoc, diag::err_omp_loop_cannot_use_stmt)
33430b57cec5SDimitry Andric                      << "break");
33440b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, BreakLoc, *S);
33450b57cec5SDimitry Andric 
33460b57cec5SDimitry Andric   return new (Context) BreakStmt(BreakLoc);
33470b57cec5SDimitry Andric }
33480b57cec5SDimitry Andric 
3349fe6060f1SDimitry Andric /// Determine whether the given expression might be move-eligible or
3350fe6060f1SDimitry Andric /// copy-elidable in either a (co_)return statement or throw expression,
3351fe6060f1SDimitry Andric /// without considering function return type, if applicable.
33520b57cec5SDimitry Andric ///
3353fe6060f1SDimitry Andric /// \param E The expression being returned from the function or block,
3354fe6060f1SDimitry Andric /// being thrown, or being co_returned from a coroutine. This expression
3355fe6060f1SDimitry Andric /// might be modified by the implementation.
33560b57cec5SDimitry Andric ///
3357349cc55cSDimitry Andric /// \param Mode Overrides detection of current language mode
3358fe6060f1SDimitry Andric /// and uses the rules for C++2b.
33590b57cec5SDimitry Andric ///
3360fe6060f1SDimitry Andric /// \returns An aggregate which contains the Candidate and isMoveEligible
3361fe6060f1SDimitry Andric /// and isCopyElidable methods. If Candidate is non-null, it means
3362fe6060f1SDimitry Andric /// isMoveEligible() would be true under the most permissive language standard.
3363fe6060f1SDimitry Andric Sema::NamedReturnInfo Sema::getNamedReturnInfo(Expr *&E,
3364fe6060f1SDimitry Andric                                                SimplerImplicitMoveMode Mode) {
3365fe6060f1SDimitry Andric   if (!E)
3366fe6060f1SDimitry Andric     return NamedReturnInfo();
33670b57cec5SDimitry Andric   // - in a return statement in a function [where] ...
33680b57cec5SDimitry Andric   // ... the expression is the name of a non-volatile automatic object ...
3369fe6060f1SDimitry Andric   const auto *DR = dyn_cast<DeclRefExpr>(E->IgnoreParens());
33700b57cec5SDimitry Andric   if (!DR || DR->refersToEnclosingVariableOrCapture())
3371fe6060f1SDimitry Andric     return NamedReturnInfo();
3372fe6060f1SDimitry Andric   const auto *VD = dyn_cast<VarDecl>(DR->getDecl());
33730b57cec5SDimitry Andric   if (!VD)
3374fe6060f1SDimitry Andric     return NamedReturnInfo();
3375fe6060f1SDimitry Andric   NamedReturnInfo Res = getNamedReturnInfo(VD);
3376fe6060f1SDimitry Andric   if (Res.Candidate && !E->isXValue() &&
3377fe6060f1SDimitry Andric       (Mode == SimplerImplicitMoveMode::ForceOn ||
3378fe6060f1SDimitry Andric        (Mode != SimplerImplicitMoveMode::ForceOff &&
3379fe6060f1SDimitry Andric         getLangOpts().CPlusPlus2b))) {
3380fe6060f1SDimitry Andric     E = ImplicitCastExpr::Create(Context, VD->getType().getNonReferenceType(),
3381fe6060f1SDimitry Andric                                  CK_NoOp, E, nullptr, VK_XValue,
3382fe6060f1SDimitry Andric                                  FPOptionsOverride());
3383fe6060f1SDimitry Andric   }
3384fe6060f1SDimitry Andric   return Res;
33850b57cec5SDimitry Andric }
33860b57cec5SDimitry Andric 
3387fe6060f1SDimitry Andric /// Determine whether the given NRVO candidate variable is move-eligible or
3388fe6060f1SDimitry Andric /// copy-elidable, without considering function return type.
3389fe6060f1SDimitry Andric ///
3390fe6060f1SDimitry Andric /// \param VD The NRVO candidate variable.
3391fe6060f1SDimitry Andric ///
3392fe6060f1SDimitry Andric /// \returns An aggregate which contains the Candidate and isMoveEligible
3393fe6060f1SDimitry Andric /// and isCopyElidable methods. If Candidate is non-null, it means
3394fe6060f1SDimitry Andric /// isMoveEligible() would be true under the most permissive language standard.
3395fe6060f1SDimitry Andric Sema::NamedReturnInfo Sema::getNamedReturnInfo(const VarDecl *VD) {
3396fe6060f1SDimitry Andric   NamedReturnInfo Info{VD, NamedReturnInfo::MoveEligibleAndCopyElidable};
3397fe6060f1SDimitry Andric 
3398fe6060f1SDimitry Andric   // C++20 [class.copy.elision]p3:
33990b57cec5SDimitry Andric   // - in a return statement in a function with ...
3400fe6060f1SDimitry Andric   // (other than a function ... parameter)
3401fe6060f1SDimitry Andric   if (VD->getKind() == Decl::ParmVar)
3402fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
3403fe6060f1SDimitry Andric   else if (VD->getKind() != Decl::Var)
3404fe6060f1SDimitry Andric     return NamedReturnInfo();
34050b57cec5SDimitry Andric 
3406fe6060f1SDimitry Andric   // (other than ... a catch-clause parameter)
3407fe6060f1SDimitry Andric   if (VD->isExceptionVariable())
3408fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
34090b57cec5SDimitry Andric 
34100b57cec5SDimitry Andric   // ...automatic...
3411fe6060f1SDimitry Andric   if (!VD->hasLocalStorage())
3412fe6060f1SDimitry Andric     return NamedReturnInfo();
34130b57cec5SDimitry Andric 
3414fe6060f1SDimitry Andric   // We don't want to implicitly move out of a __block variable during a return
3415fe6060f1SDimitry Andric   // because we cannot assume the variable will no longer be used.
3416fe6060f1SDimitry Andric   if (VD->hasAttr<BlocksAttr>())
3417fe6060f1SDimitry Andric     return NamedReturnInfo();
34180b57cec5SDimitry Andric 
3419fe6060f1SDimitry Andric   QualType VDType = VD->getType();
3420fe6060f1SDimitry Andric   if (VDType->isObjectType()) {
3421fe6060f1SDimitry Andric     // C++17 [class.copy.elision]p3:
3422fe6060f1SDimitry Andric     // ...non-volatile automatic object...
3423fe6060f1SDimitry Andric     if (VDType.isVolatileQualified())
3424fe6060f1SDimitry Andric       return NamedReturnInfo();
3425fe6060f1SDimitry Andric   } else if (VDType->isRValueReferenceType()) {
3426fe6060f1SDimitry Andric     // C++20 [class.copy.elision]p3:
3427fe6060f1SDimitry Andric     // ...either a non-volatile object or an rvalue reference to a non-volatile
3428fe6060f1SDimitry Andric     // object type...
3429fe6060f1SDimitry Andric     QualType VDReferencedType = VDType.getNonReferenceType();
3430fe6060f1SDimitry Andric     if (VDReferencedType.isVolatileQualified() ||
3431fe6060f1SDimitry Andric         !VDReferencedType->isObjectType())
3432fe6060f1SDimitry Andric       return NamedReturnInfo();
3433fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
3434fe6060f1SDimitry Andric   } else {
3435fe6060f1SDimitry Andric     return NamedReturnInfo();
3436fe6060f1SDimitry Andric   }
34370b57cec5SDimitry Andric 
34380b57cec5SDimitry Andric   // Variables with higher required alignment than their type's ABI
34390b57cec5SDimitry Andric   // alignment cannot use NRVO.
3440fe6060f1SDimitry Andric   if (!VD->hasDependentAlignment() &&
3441fe6060f1SDimitry Andric       Context.getDeclAlign(VD) > Context.getTypeAlignInChars(VDType))
3442fe6060f1SDimitry Andric     Info.S = NamedReturnInfo::MoveEligible;
3443fe6060f1SDimitry Andric 
3444fe6060f1SDimitry Andric   return Info;
3445fe6060f1SDimitry Andric }
3446fe6060f1SDimitry Andric 
3447fe6060f1SDimitry Andric /// Updates given NamedReturnInfo's move-eligible and
3448fe6060f1SDimitry Andric /// copy-elidable statuses, considering the function
3449fe6060f1SDimitry Andric /// return type criteria as applicable to return statements.
3450fe6060f1SDimitry Andric ///
3451fe6060f1SDimitry Andric /// \param Info The NamedReturnInfo object to update.
3452fe6060f1SDimitry Andric ///
3453fe6060f1SDimitry Andric /// \param ReturnType This is the return type of the function.
3454fe6060f1SDimitry Andric /// \returns The copy elision candidate, in case the initial return expression
3455fe6060f1SDimitry Andric /// was copy elidable, or nullptr otherwise.
3456fe6060f1SDimitry Andric const VarDecl *Sema::getCopyElisionCandidate(NamedReturnInfo &Info,
3457fe6060f1SDimitry Andric                                              QualType ReturnType) {
3458fe6060f1SDimitry Andric   if (!Info.Candidate)
3459fe6060f1SDimitry Andric     return nullptr;
3460fe6060f1SDimitry Andric 
3461fe6060f1SDimitry Andric   auto invalidNRVO = [&] {
3462fe6060f1SDimitry Andric     Info = NamedReturnInfo();
3463fe6060f1SDimitry Andric     return nullptr;
3464fe6060f1SDimitry Andric   };
3465fe6060f1SDimitry Andric 
3466fe6060f1SDimitry Andric   // If we got a non-deduced auto ReturnType, we are in a dependent context and
3467fe6060f1SDimitry Andric   // there is no point in allowing copy elision since we won't have it deduced
3468fe6060f1SDimitry Andric   // by the point the VardDecl is instantiated, which is the last chance we have
3469fe6060f1SDimitry Andric   // of deciding if the candidate is really copy elidable.
3470fe6060f1SDimitry Andric   if ((ReturnType->getTypeClass() == Type::TypeClass::Auto &&
3471fe6060f1SDimitry Andric        ReturnType->isCanonicalUnqualified()) ||
3472fe6060f1SDimitry Andric       ReturnType->isSpecificBuiltinType(BuiltinType::Dependent))
3473fe6060f1SDimitry Andric     return invalidNRVO();
3474fe6060f1SDimitry Andric 
3475fe6060f1SDimitry Andric   if (!ReturnType->isDependentType()) {
3476fe6060f1SDimitry Andric     // - in a return statement in a function with ...
3477fe6060f1SDimitry Andric     // ... a class return type ...
3478fe6060f1SDimitry Andric     if (!ReturnType->isRecordType())
3479fe6060f1SDimitry Andric       return invalidNRVO();
3480fe6060f1SDimitry Andric 
3481fe6060f1SDimitry Andric     QualType VDType = Info.Candidate->getType();
3482fe6060f1SDimitry Andric     // ... the same cv-unqualified type as the function return type ...
3483fe6060f1SDimitry Andric     // When considering moving this expression out, allow dissimilar types.
3484fe6060f1SDimitry Andric     if (!VDType->isDependentType() &&
3485fe6060f1SDimitry Andric         !Context.hasSameUnqualifiedType(ReturnType, VDType))
3486fe6060f1SDimitry Andric       Info.S = NamedReturnInfo::MoveEligible;
3487fe6060f1SDimitry Andric   }
3488fe6060f1SDimitry Andric   return Info.isCopyElidable() ? Info.Candidate : nullptr;
3489fe6060f1SDimitry Andric }
3490fe6060f1SDimitry Andric 
3491fe6060f1SDimitry Andric /// Verify that the initialization sequence that was picked for the
3492fe6060f1SDimitry Andric /// first overload resolution is permissible under C++98.
3493fe6060f1SDimitry Andric ///
3494349cc55cSDimitry Andric /// Reject (possibly converting) constructors not taking an rvalue reference,
3495fe6060f1SDimitry Andric /// or user conversion operators which are not ref-qualified.
3496fe6060f1SDimitry Andric static bool
3497fe6060f1SDimitry Andric VerifyInitializationSequenceCXX98(const Sema &S,
3498fe6060f1SDimitry Andric                                   const InitializationSequence &Seq) {
3499fe6060f1SDimitry Andric   const auto *Step = llvm::find_if(Seq.steps(), [](const auto &Step) {
3500fe6060f1SDimitry Andric     return Step.Kind == InitializationSequence::SK_ConstructorInitialization ||
3501fe6060f1SDimitry Andric            Step.Kind == InitializationSequence::SK_UserConversion;
3502fe6060f1SDimitry Andric   });
3503fe6060f1SDimitry Andric   if (Step != Seq.step_end()) {
3504fe6060f1SDimitry Andric     const auto *FD = Step->Function.Function;
3505fe6060f1SDimitry Andric     if (isa<CXXConstructorDecl>(FD)
3506fe6060f1SDimitry Andric             ? !FD->getParamDecl(0)->getType()->isRValueReferenceType()
3507fe6060f1SDimitry Andric             : cast<CXXMethodDecl>(FD)->getRefQualifier() == RQ_None)
35080b57cec5SDimitry Andric       return false;
3509fe6060f1SDimitry Andric   }
35100b57cec5SDimitry Andric   return true;
35110b57cec5SDimitry Andric }
35120b57cec5SDimitry Andric 
35130b57cec5SDimitry Andric /// Perform the initialization of a potentially-movable value, which
35140b57cec5SDimitry Andric /// is the result of return value.
35150b57cec5SDimitry Andric ///
3516fe6060f1SDimitry Andric /// This routine implements C++20 [class.copy.elision]p3, which attempts to
3517fe6060f1SDimitry Andric /// treat returned lvalues as rvalues in certain cases (to prefer move
3518fe6060f1SDimitry Andric /// construction), then falls back to treating them as lvalues if that failed.
3519fe6060f1SDimitry Andric ExprResult Sema::PerformMoveOrCopyInitialization(
3520fe6060f1SDimitry Andric     const InitializedEntity &Entity, const NamedReturnInfo &NRInfo, Expr *Value,
3521fe6060f1SDimitry Andric     bool SupressSimplerImplicitMoves) {
35228c6f6c0cSDimitry Andric   if (getLangOpts().CPlusPlus &&
35238c6f6c0cSDimitry Andric       (!getLangOpts().CPlusPlus2b || SupressSimplerImplicitMoves) &&
3524fe6060f1SDimitry Andric       NRInfo.isMoveEligible()) {
3525fe6060f1SDimitry Andric     ImplicitCastExpr AsRvalue(ImplicitCastExpr::OnStack, Value->getType(),
3526fe6060f1SDimitry Andric                               CK_NoOp, Value, VK_XValue, FPOptionsOverride());
3527fe6060f1SDimitry Andric     Expr *InitExpr = &AsRvalue;
3528fe6060f1SDimitry Andric     auto Kind = InitializationKind::CreateCopy(Value->getBeginLoc(),
3529fe6060f1SDimitry Andric                                                Value->getBeginLoc());
3530fe6060f1SDimitry Andric     InitializationSequence Seq(*this, Entity, Kind, InitExpr);
3531fe6060f1SDimitry Andric     auto Res = Seq.getFailedOverloadResult();
3532fe6060f1SDimitry Andric     if ((Res == OR_Success || Res == OR_Deleted) &&
3533fe6060f1SDimitry Andric         (getLangOpts().CPlusPlus11 ||
3534fe6060f1SDimitry Andric          VerifyInitializationSequenceCXX98(*this, Seq))) {
3535fe6060f1SDimitry Andric       // Promote "AsRvalue" to the heap, since we now need this
3536fe6060f1SDimitry Andric       // expression node to persist.
3537fe6060f1SDimitry Andric       Value =
3538fe6060f1SDimitry Andric           ImplicitCastExpr::Create(Context, Value->getType(), CK_NoOp, Value,
3539fe6060f1SDimitry Andric                                    nullptr, VK_XValue, FPOptionsOverride());
3540fe6060f1SDimitry Andric       // Complete type-checking the initialization of the return type
3541fe6060f1SDimitry Andric       // using the constructor we found.
3542fe6060f1SDimitry Andric       return Seq.Perform(*this, Entity, Kind, Value);
35430b57cec5SDimitry Andric     }
35440b57cec5SDimitry Andric   }
35450b57cec5SDimitry Andric   // Either we didn't meet the criteria for treating an lvalue as an rvalue,
35460b57cec5SDimitry Andric   // above, or overload resolution failed. Either way, we need to try
35470b57cec5SDimitry Andric   // (again) now with the return value expression as written.
3548fe6060f1SDimitry Andric   return PerformCopyInitialization(Entity, SourceLocation(), Value);
35490b57cec5SDimitry Andric }
35500b57cec5SDimitry Andric 
35510b57cec5SDimitry Andric /// Determine whether the declared return type of the specified function
35520b57cec5SDimitry Andric /// contains 'auto'.
35530b57cec5SDimitry Andric static bool hasDeducedReturnType(FunctionDecl *FD) {
35540b57cec5SDimitry Andric   const FunctionProtoType *FPT =
35550b57cec5SDimitry Andric       FD->getTypeSourceInfo()->getType()->castAs<FunctionProtoType>();
35560b57cec5SDimitry Andric   return FPT->getReturnType()->isUndeducedType();
35570b57cec5SDimitry Andric }
35580b57cec5SDimitry Andric 
35590b57cec5SDimitry Andric /// ActOnCapScopeReturnStmt - Utility routine to type-check return statements
35600b57cec5SDimitry Andric /// for capturing scopes.
35610b57cec5SDimitry Andric ///
3562fe6060f1SDimitry Andric StmtResult Sema::ActOnCapScopeReturnStmt(SourceLocation ReturnLoc,
3563fe6060f1SDimitry Andric                                          Expr *RetValExp,
3564fe6060f1SDimitry Andric                                          NamedReturnInfo &NRInfo,
3565fe6060f1SDimitry Andric                                          bool SupressSimplerImplicitMoves) {
35660b57cec5SDimitry Andric   // If this is the first return we've seen, infer the return type.
35670b57cec5SDimitry Andric   // [expr.prim.lambda]p4 in C++11; block literals follow the same rules.
35680b57cec5SDimitry Andric   CapturingScopeInfo *CurCap = cast<CapturingScopeInfo>(getCurFunction());
35690b57cec5SDimitry Andric   QualType FnRetType = CurCap->ReturnType;
35700b57cec5SDimitry Andric   LambdaScopeInfo *CurLambda = dyn_cast<LambdaScopeInfo>(CurCap);
35710b57cec5SDimitry Andric   bool HasDeducedReturnType =
35720b57cec5SDimitry Andric       CurLambda && hasDeducedReturnType(CurLambda->CallOperator);
35730b57cec5SDimitry Andric 
35744824e7fdSDimitry Andric   if (ExprEvalContexts.back().isDiscardedStatementContext() &&
35750b57cec5SDimitry Andric       (HasDeducedReturnType || CurCap->HasImplicitReturnType)) {
35760b57cec5SDimitry Andric     if (RetValExp) {
35770b57cec5SDimitry Andric       ExprResult ER =
35780b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
35790b57cec5SDimitry Andric       if (ER.isInvalid())
35800b57cec5SDimitry Andric         return StmtError();
35810b57cec5SDimitry Andric       RetValExp = ER.get();
35820b57cec5SDimitry Andric     }
35830b57cec5SDimitry Andric     return ReturnStmt::Create(Context, ReturnLoc, RetValExp,
35840b57cec5SDimitry Andric                               /* NRVOCandidate=*/nullptr);
35850b57cec5SDimitry Andric   }
35860b57cec5SDimitry Andric 
35870b57cec5SDimitry Andric   if (HasDeducedReturnType) {
3588e8d8bef9SDimitry Andric     FunctionDecl *FD = CurLambda->CallOperator;
3589e8d8bef9SDimitry Andric     // If we've already decided this lambda is invalid, e.g. because
3590e8d8bef9SDimitry Andric     // we saw a `return` whose expression had an error, don't keep
3591e8d8bef9SDimitry Andric     // trying to deduce its return type.
3592e8d8bef9SDimitry Andric     if (FD->isInvalidDecl())
3593e8d8bef9SDimitry Andric       return StmtError();
35940b57cec5SDimitry Andric     // In C++1y, the return type may involve 'auto'.
35950b57cec5SDimitry Andric     // FIXME: Blocks might have a return type of 'auto' explicitly specified.
35960b57cec5SDimitry Andric     if (CurCap->ReturnType.isNull())
35970b57cec5SDimitry Andric       CurCap->ReturnType = FD->getReturnType();
35980b57cec5SDimitry Andric 
35990b57cec5SDimitry Andric     AutoType *AT = CurCap->ReturnType->getContainedAutoType();
36000b57cec5SDimitry Andric     assert(AT && "lost auto type from lambda return type");
36010b57cec5SDimitry Andric     if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
36020b57cec5SDimitry Andric       FD->setInvalidDecl();
36035ffd83dbSDimitry Andric       // FIXME: preserve the ill-formed return expression.
36040b57cec5SDimitry Andric       return StmtError();
36050b57cec5SDimitry Andric     }
36060b57cec5SDimitry Andric     CurCap->ReturnType = FnRetType = FD->getReturnType();
36070b57cec5SDimitry Andric   } else if (CurCap->HasImplicitReturnType) {
36080b57cec5SDimitry Andric     // For blocks/lambdas with implicit return types, we check each return
36090b57cec5SDimitry Andric     // statement individually, and deduce the common return type when the block
36100b57cec5SDimitry Andric     // or lambda is completed.
36110b57cec5SDimitry Andric     // FIXME: Fold this into the 'auto' codepath above.
36120b57cec5SDimitry Andric     if (RetValExp && !isa<InitListExpr>(RetValExp)) {
36130b57cec5SDimitry Andric       ExprResult Result = DefaultFunctionArrayLvalueConversion(RetValExp);
36140b57cec5SDimitry Andric       if (Result.isInvalid())
36150b57cec5SDimitry Andric         return StmtError();
36160b57cec5SDimitry Andric       RetValExp = Result.get();
36170b57cec5SDimitry Andric 
36180b57cec5SDimitry Andric       // DR1048: even prior to C++14, we should use the 'auto' deduction rules
36190b57cec5SDimitry Andric       // when deducing a return type for a lambda-expression (or by extension
36200b57cec5SDimitry Andric       // for a block). These rules differ from the stated C++11 rules only in
36210b57cec5SDimitry Andric       // that they remove top-level cv-qualifiers.
36220b57cec5SDimitry Andric       if (!CurContext->isDependentContext())
36230b57cec5SDimitry Andric         FnRetType = RetValExp->getType().getUnqualifiedType();
36240b57cec5SDimitry Andric       else
36250b57cec5SDimitry Andric         FnRetType = CurCap->ReturnType = Context.DependentTy;
36260b57cec5SDimitry Andric     } else {
36270b57cec5SDimitry Andric       if (RetValExp) {
36280b57cec5SDimitry Andric         // C++11 [expr.lambda.prim]p4 bans inferring the result from an
36290b57cec5SDimitry Andric         // initializer list, because it is not an expression (even
36300b57cec5SDimitry Andric         // though we represent it as one). We still deduce 'void'.
36310b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_lambda_return_init_list)
36320b57cec5SDimitry Andric           << RetValExp->getSourceRange();
36330b57cec5SDimitry Andric       }
36340b57cec5SDimitry Andric 
36350b57cec5SDimitry Andric       FnRetType = Context.VoidTy;
36360b57cec5SDimitry Andric     }
36370b57cec5SDimitry Andric 
36380b57cec5SDimitry Andric     // Although we'll properly infer the type of the block once it's completed,
36390b57cec5SDimitry Andric     // make sure we provide a return type now for better error recovery.
36400b57cec5SDimitry Andric     if (CurCap->ReturnType.isNull())
36410b57cec5SDimitry Andric       CurCap->ReturnType = FnRetType;
36420b57cec5SDimitry Andric   }
3643fe6060f1SDimitry Andric   const VarDecl *NRVOCandidate = getCopyElisionCandidate(NRInfo, FnRetType);
36440b57cec5SDimitry Andric 
3645a7dea167SDimitry Andric   if (auto *CurBlock = dyn_cast<BlockScopeInfo>(CurCap)) {
3646a7dea167SDimitry Andric     if (CurBlock->FunctionType->castAs<FunctionType>()->getNoReturnAttr()) {
36470b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_noreturn_block_has_return_expr);
36480b57cec5SDimitry Andric       return StmtError();
36490b57cec5SDimitry Andric     }
3650a7dea167SDimitry Andric   } else if (auto *CurRegion = dyn_cast<CapturedRegionScopeInfo>(CurCap)) {
36510b57cec5SDimitry Andric     Diag(ReturnLoc, diag::err_return_in_captured_stmt) << CurRegion->getRegionName();
36520b57cec5SDimitry Andric     return StmtError();
36530b57cec5SDimitry Andric   } else {
36540b57cec5SDimitry Andric     assert(CurLambda && "unknown kind of captured scope");
3655a7dea167SDimitry Andric     if (CurLambda->CallOperator->getType()
3656a7dea167SDimitry Andric             ->castAs<FunctionType>()
36570b57cec5SDimitry Andric             ->getNoReturnAttr()) {
36580b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_noreturn_lambda_has_return_expr);
36590b57cec5SDimitry Andric       return StmtError();
36600b57cec5SDimitry Andric     }
36610b57cec5SDimitry Andric   }
36620b57cec5SDimitry Andric 
36630b57cec5SDimitry Andric   // Otherwise, verify that this result type matches the previous one.  We are
36640b57cec5SDimitry Andric   // pickier with blocks than for normal functions because we don't have GCC
36650b57cec5SDimitry Andric   // compatibility to worry about here.
36660b57cec5SDimitry Andric   if (FnRetType->isDependentType()) {
36670b57cec5SDimitry Andric     // Delay processing for now.  TODO: there are lots of dependent
36680b57cec5SDimitry Andric     // types we can conclusively prove aren't void.
36690b57cec5SDimitry Andric   } else if (FnRetType->isVoidType()) {
36700b57cec5SDimitry Andric     if (RetValExp && !isa<InitListExpr>(RetValExp) &&
36710b57cec5SDimitry Andric         !(getLangOpts().CPlusPlus &&
36720b57cec5SDimitry Andric           (RetValExp->isTypeDependent() ||
36730b57cec5SDimitry Andric            RetValExp->getType()->isVoidType()))) {
36740b57cec5SDimitry Andric       if (!getLangOpts().CPlusPlus &&
36750b57cec5SDimitry Andric           RetValExp->getType()->isVoidType())
36760b57cec5SDimitry Andric         Diag(ReturnLoc, diag::ext_return_has_void_expr) << "literal" << 2;
36770b57cec5SDimitry Andric       else {
36780b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_return_block_has_expr);
36790b57cec5SDimitry Andric         RetValExp = nullptr;
36800b57cec5SDimitry Andric       }
36810b57cec5SDimitry Andric     }
36820b57cec5SDimitry Andric   } else if (!RetValExp) {
36830b57cec5SDimitry Andric     return StmtError(Diag(ReturnLoc, diag::err_block_return_missing_expr));
36840b57cec5SDimitry Andric   } else if (!RetValExp->isTypeDependent()) {
36850b57cec5SDimitry Andric     // we have a non-void block with an expression, continue checking
36860b57cec5SDimitry Andric 
36870b57cec5SDimitry Andric     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
36880b57cec5SDimitry Andric     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
36890b57cec5SDimitry Andric     // function return.
36900b57cec5SDimitry Andric 
36910b57cec5SDimitry Andric     // In C++ the return statement is handled via a copy initialization.
36920b57cec5SDimitry Andric     // the C version of which boils down to CheckSingleAssignmentConstraints.
369328a41182SDimitry Andric     InitializedEntity Entity =
369428a41182SDimitry Andric         InitializedEntity::InitializeResult(ReturnLoc, FnRetType);
3695fe6060f1SDimitry Andric     ExprResult Res = PerformMoveOrCopyInitialization(
3696fe6060f1SDimitry Andric         Entity, NRInfo, RetValExp, SupressSimplerImplicitMoves);
36970b57cec5SDimitry Andric     if (Res.isInvalid()) {
36980b57cec5SDimitry Andric       // FIXME: Cleanup temporaries here, anyway?
36990b57cec5SDimitry Andric       return StmtError();
37000b57cec5SDimitry Andric     }
37010b57cec5SDimitry Andric     RetValExp = Res.get();
37020b57cec5SDimitry Andric     CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc);
37030b57cec5SDimitry Andric   }
37040b57cec5SDimitry Andric 
37050b57cec5SDimitry Andric   if (RetValExp) {
37060b57cec5SDimitry Andric     ExprResult ER =
37070b57cec5SDimitry Andric         ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
37080b57cec5SDimitry Andric     if (ER.isInvalid())
37090b57cec5SDimitry Andric       return StmtError();
37100b57cec5SDimitry Andric     RetValExp = ER.get();
37110b57cec5SDimitry Andric   }
37120b57cec5SDimitry Andric   auto *Result =
37130b57cec5SDimitry Andric       ReturnStmt::Create(Context, ReturnLoc, RetValExp, NRVOCandidate);
37140b57cec5SDimitry Andric 
37150b57cec5SDimitry Andric   // If we need to check for the named return value optimization,
37160b57cec5SDimitry Andric   // or if we need to infer the return type,
37170b57cec5SDimitry Andric   // save the return statement in our scope for later processing.
37180b57cec5SDimitry Andric   if (CurCap->HasImplicitReturnType || NRVOCandidate)
37190b57cec5SDimitry Andric     FunctionScopes.back()->Returns.push_back(Result);
37200b57cec5SDimitry Andric 
37210b57cec5SDimitry Andric   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
37220b57cec5SDimitry Andric     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
37230b57cec5SDimitry Andric 
37240b57cec5SDimitry Andric   return Result;
37250b57cec5SDimitry Andric }
37260b57cec5SDimitry Andric 
37270b57cec5SDimitry Andric namespace {
37280b57cec5SDimitry Andric /// Marks all typedefs in all local classes in a type referenced.
37290b57cec5SDimitry Andric ///
37300b57cec5SDimitry Andric /// In a function like
37310b57cec5SDimitry Andric /// auto f() {
37320b57cec5SDimitry Andric ///   struct S { typedef int a; };
37330b57cec5SDimitry Andric ///   return S();
37340b57cec5SDimitry Andric /// }
37350b57cec5SDimitry Andric ///
37360b57cec5SDimitry Andric /// the local type escapes and could be referenced in some TUs but not in
37370b57cec5SDimitry Andric /// others. Pretend that all local typedefs are always referenced, to not warn
37380b57cec5SDimitry Andric /// on this. This isn't necessary if f has internal linkage, or the typedef
37390b57cec5SDimitry Andric /// is private.
37400b57cec5SDimitry Andric class LocalTypedefNameReferencer
37410b57cec5SDimitry Andric     : public RecursiveASTVisitor<LocalTypedefNameReferencer> {
37420b57cec5SDimitry Andric public:
37430b57cec5SDimitry Andric   LocalTypedefNameReferencer(Sema &S) : S(S) {}
37440b57cec5SDimitry Andric   bool VisitRecordType(const RecordType *RT);
37450b57cec5SDimitry Andric private:
37460b57cec5SDimitry Andric   Sema &S;
37470b57cec5SDimitry Andric };
37480b57cec5SDimitry Andric bool LocalTypedefNameReferencer::VisitRecordType(const RecordType *RT) {
37490b57cec5SDimitry Andric   auto *R = dyn_cast<CXXRecordDecl>(RT->getDecl());
37500b57cec5SDimitry Andric   if (!R || !R->isLocalClass() || !R->isLocalClass()->isExternallyVisible() ||
37510b57cec5SDimitry Andric       R->isDependentType())
37520b57cec5SDimitry Andric     return true;
37530b57cec5SDimitry Andric   for (auto *TmpD : R->decls())
37540b57cec5SDimitry Andric     if (auto *T = dyn_cast<TypedefNameDecl>(TmpD))
37550b57cec5SDimitry Andric       if (T->getAccess() != AS_private || R->hasFriends())
37560b57cec5SDimitry Andric         S.MarkAnyDeclReferenced(T->getLocation(), T, /*OdrUse=*/false);
37570b57cec5SDimitry Andric   return true;
37580b57cec5SDimitry Andric }
37590b57cec5SDimitry Andric }
37600b57cec5SDimitry Andric 
37610b57cec5SDimitry Andric TypeLoc Sema::getReturnTypeLoc(FunctionDecl *FD) const {
37620b57cec5SDimitry Andric   return FD->getTypeSourceInfo()
37630b57cec5SDimitry Andric       ->getTypeLoc()
37640b57cec5SDimitry Andric       .getAsAdjusted<FunctionProtoTypeLoc>()
37650b57cec5SDimitry Andric       .getReturnLoc();
37660b57cec5SDimitry Andric }
37670b57cec5SDimitry Andric 
37680b57cec5SDimitry Andric /// Deduce the return type for a function from a returned expression, per
37690b57cec5SDimitry Andric /// C++1y [dcl.spec.auto]p6.
37700b57cec5SDimitry Andric bool Sema::DeduceFunctionTypeFromReturnExpr(FunctionDecl *FD,
37710b57cec5SDimitry Andric                                             SourceLocation ReturnLoc,
37720b57cec5SDimitry Andric                                             Expr *&RetExpr,
377381ad6265SDimitry Andric                                             const AutoType *AT) {
377481ad6265SDimitry Andric   // If this is the conversion function for a lambda, we choose to deduce its
37750b57cec5SDimitry Andric   // type from the corresponding call operator, not from the synthesized return
37760b57cec5SDimitry Andric   // statement within it. See Sema::DeduceReturnType.
37770b57cec5SDimitry Andric   if (isLambdaConversionOperator(FD))
37780b57cec5SDimitry Andric     return false;
37790b57cec5SDimitry Andric 
37800b57cec5SDimitry Andric   TypeLoc OrigResultType = getReturnTypeLoc(FD);
37810b57cec5SDimitry Andric   QualType Deduced;
37820b57cec5SDimitry Andric 
37830b57cec5SDimitry Andric   if (RetExpr && isa<InitListExpr>(RetExpr)) {
37840b57cec5SDimitry Andric     //  If the deduction is for a return statement and the initializer is
37850b57cec5SDimitry Andric     //  a braced-init-list, the program is ill-formed.
37860b57cec5SDimitry Andric     Diag(RetExpr->getExprLoc(),
37870b57cec5SDimitry Andric          getCurLambda() ? diag::err_lambda_return_init_list
37880b57cec5SDimitry Andric                         : diag::err_auto_fn_return_init_list)
37890b57cec5SDimitry Andric         << RetExpr->getSourceRange();
37900b57cec5SDimitry Andric     return true;
37910b57cec5SDimitry Andric   }
37920b57cec5SDimitry Andric 
37930b57cec5SDimitry Andric   if (FD->isDependentContext()) {
37940b57cec5SDimitry Andric     // C++1y [dcl.spec.auto]p12:
37950b57cec5SDimitry Andric     //   Return type deduction [...] occurs when the definition is
37960b57cec5SDimitry Andric     //   instantiated even if the function body contains a return
37970b57cec5SDimitry Andric     //   statement with a non-type-dependent operand.
37980b57cec5SDimitry Andric     assert(AT->isDeduced() && "should have deduced to dependent type");
37990b57cec5SDimitry Andric     return false;
38000b57cec5SDimitry Andric   }
38010b57cec5SDimitry Andric 
38020b57cec5SDimitry Andric   if (RetExpr) {
38030b57cec5SDimitry Andric     //  Otherwise, [...] deduce a value for U using the rules of template
38040b57cec5SDimitry Andric     //  argument deduction.
38050b57cec5SDimitry Andric     DeduceAutoResult DAR = DeduceAutoType(OrigResultType, RetExpr, Deduced);
38060b57cec5SDimitry Andric 
38070b57cec5SDimitry Andric     if (DAR == DAR_Failed && !FD->isInvalidDecl())
38080b57cec5SDimitry Andric       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_deduction_failure)
38090b57cec5SDimitry Andric         << OrigResultType.getType() << RetExpr->getType();
38100b57cec5SDimitry Andric 
38110b57cec5SDimitry Andric     if (DAR != DAR_Succeeded)
38120b57cec5SDimitry Andric       return true;
38130b57cec5SDimitry Andric 
38140b57cec5SDimitry Andric     // If a local type is part of the returned type, mark its fields as
38150b57cec5SDimitry Andric     // referenced.
38160b57cec5SDimitry Andric     LocalTypedefNameReferencer Referencer(*this);
38170b57cec5SDimitry Andric     Referencer.TraverseType(RetExpr->getType());
38180b57cec5SDimitry Andric   } else {
381981ad6265SDimitry Andric     // For a function with a deduced result type to return void,
382081ad6265SDimitry Andric     // the result type as written must be 'auto' or 'decltype(auto)',
382181ad6265SDimitry Andric     // possibly cv-qualified or constrained, but not ref-qualified.
38220b57cec5SDimitry Andric     if (!OrigResultType.getType()->getAs<AutoType>()) {
38230b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_auto_fn_return_void_but_not_auto)
38240b57cec5SDimitry Andric         << OrigResultType.getType();
38250b57cec5SDimitry Andric       return true;
38260b57cec5SDimitry Andric     }
382781ad6265SDimitry Andric     // In the case of a return with no operand, the initializer is considered
382881ad6265SDimitry Andric     // to be 'void()'.
382981ad6265SDimitry Andric     Expr *Dummy = new (Context) CXXScalarValueInitExpr(
383081ad6265SDimitry Andric         Context.VoidTy,
383181ad6265SDimitry Andric         Context.getTrivialTypeSourceInfo(Context.VoidTy, ReturnLoc), ReturnLoc);
383281ad6265SDimitry Andric     DeduceAutoResult DAR = DeduceAutoType(OrigResultType, Dummy, Deduced);
383381ad6265SDimitry Andric 
383481ad6265SDimitry Andric     if (DAR == DAR_Failed && !FD->isInvalidDecl())
383581ad6265SDimitry Andric       Diag(ReturnLoc, diag::err_auto_fn_deduction_failure)
383681ad6265SDimitry Andric           << OrigResultType.getType() << Dummy->getType();
383781ad6265SDimitry Andric 
383881ad6265SDimitry Andric     if (DAR != DAR_Succeeded)
38390b57cec5SDimitry Andric       return true;
38400b57cec5SDimitry Andric   }
38410b57cec5SDimitry Andric 
3842a7dea167SDimitry Andric   // CUDA: Kernel function must have 'void' return type.
3843a7dea167SDimitry Andric   if (getLangOpts().CUDA)
3844a7dea167SDimitry Andric     if (FD->hasAttr<CUDAGlobalAttr>() && !Deduced->isVoidType()) {
3845a7dea167SDimitry Andric       Diag(FD->getLocation(), diag::err_kern_type_not_void_return)
3846a7dea167SDimitry Andric           << FD->getType() << FD->getSourceRange();
3847a7dea167SDimitry Andric       return true;
3848a7dea167SDimitry Andric     }
3849a7dea167SDimitry Andric 
38500b57cec5SDimitry Andric   //  If a function with a declared return type that contains a placeholder type
38510b57cec5SDimitry Andric   //  has multiple return statements, the return type is deduced for each return
38520b57cec5SDimitry Andric   //  statement. [...] if the type deduced is not the same in each deduction,
38530b57cec5SDimitry Andric   //  the program is ill-formed.
38540b57cec5SDimitry Andric   QualType DeducedT = AT->getDeducedType();
38550b57cec5SDimitry Andric   if (!DeducedT.isNull() && !FD->isInvalidDecl()) {
38560b57cec5SDimitry Andric     AutoType *NewAT = Deduced->getContainedAutoType();
38570b57cec5SDimitry Andric     // It is possible that NewAT->getDeducedType() is null. When that happens,
38580b57cec5SDimitry Andric     // we should not crash, instead we ignore this deduction.
38590b57cec5SDimitry Andric     if (NewAT->getDeducedType().isNull())
38600b57cec5SDimitry Andric       return false;
38610b57cec5SDimitry Andric 
38620b57cec5SDimitry Andric     CanQualType OldDeducedType = Context.getCanonicalFunctionResultType(
38630b57cec5SDimitry Andric                                    DeducedT);
38640b57cec5SDimitry Andric     CanQualType NewDeducedType = Context.getCanonicalFunctionResultType(
38650b57cec5SDimitry Andric                                    NewAT->getDeducedType());
38660b57cec5SDimitry Andric     if (!FD->isDependentContext() && OldDeducedType != NewDeducedType) {
38670b57cec5SDimitry Andric       const LambdaScopeInfo *LambdaSI = getCurLambda();
38680b57cec5SDimitry Andric       if (LambdaSI && LambdaSI->HasImplicitReturnType) {
38690b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_typecheck_missing_return_type_incompatible)
38700b57cec5SDimitry Andric           << NewAT->getDeducedType() << DeducedT
38710b57cec5SDimitry Andric           << true /*IsLambda*/;
38720b57cec5SDimitry Andric       } else {
38730b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_auto_fn_different_deductions)
38740b57cec5SDimitry Andric           << (AT->isDecltypeAuto() ? 1 : 0)
38750b57cec5SDimitry Andric           << NewAT->getDeducedType() << DeducedT;
38760b57cec5SDimitry Andric       }
38770b57cec5SDimitry Andric       return true;
38780b57cec5SDimitry Andric     }
38790b57cec5SDimitry Andric   } else if (!FD->isInvalidDecl()) {
38800b57cec5SDimitry Andric     // Update all declarations of the function to have the deduced return type.
38810b57cec5SDimitry Andric     Context.adjustDeducedFunctionResultType(FD, Deduced);
38820b57cec5SDimitry Andric   }
38830b57cec5SDimitry Andric 
38840b57cec5SDimitry Andric   return false;
38850b57cec5SDimitry Andric }
38860b57cec5SDimitry Andric 
38870b57cec5SDimitry Andric StmtResult
38880b57cec5SDimitry Andric Sema::ActOnReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp,
38890b57cec5SDimitry Andric                       Scope *CurScope) {
38900b57cec5SDimitry Andric   // Correct typos, in case the containing function returns 'auto' and
38910b57cec5SDimitry Andric   // RetValExp should determine the deduced type.
3892e8d8bef9SDimitry Andric   ExprResult RetVal = CorrectDelayedTyposInExpr(
3893e8d8bef9SDimitry Andric       RetValExp, nullptr, /*RecoverUncorrectedTypos=*/true);
38940b57cec5SDimitry Andric   if (RetVal.isInvalid())
38950b57cec5SDimitry Andric     return StmtError();
389604eeddc0SDimitry Andric   StmtResult R =
389704eeddc0SDimitry Andric       BuildReturnStmt(ReturnLoc, RetVal.get(), /*AllowRecovery=*/true);
38984824e7fdSDimitry Andric   if (R.isInvalid() || ExprEvalContexts.back().isDiscardedStatementContext())
38990b57cec5SDimitry Andric     return R;
39000b57cec5SDimitry Andric 
3901*972a253aSDimitry Andric   VarDecl *VD =
3902*972a253aSDimitry Andric       const_cast<VarDecl *>(cast<ReturnStmt>(R.get())->getNRVOCandidate());
3903*972a253aSDimitry Andric 
3904*972a253aSDimitry Andric   CurScope->updateNRVOCandidate(VD);
39050b57cec5SDimitry Andric 
39060b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, ReturnLoc, *CurScope->getFnParent());
39070b57cec5SDimitry Andric 
39080b57cec5SDimitry Andric   return R;
39090b57cec5SDimitry Andric }
39100b57cec5SDimitry Andric 
3911fe6060f1SDimitry Andric static bool CheckSimplerImplicitMovesMSVCWorkaround(const Sema &S,
3912fe6060f1SDimitry Andric                                                     const Expr *E) {
3913fe6060f1SDimitry Andric   if (!E || !S.getLangOpts().CPlusPlus2b || !S.getLangOpts().MSVCCompat)
3914fe6060f1SDimitry Andric     return false;
3915fe6060f1SDimitry Andric   const Decl *D = E->getReferencedDeclOfCallee();
3916fe6060f1SDimitry Andric   if (!D || !S.SourceMgr.isInSystemHeader(D->getLocation()))
3917fe6060f1SDimitry Andric     return false;
3918fe6060f1SDimitry Andric   for (const DeclContext *DC = D->getDeclContext(); DC; DC = DC->getParent()) {
3919fe6060f1SDimitry Andric     if (DC->isStdNamespace())
3920fe6060f1SDimitry Andric       return true;
3921fe6060f1SDimitry Andric   }
3922fe6060f1SDimitry Andric   return false;
3923fe6060f1SDimitry Andric }
3924fe6060f1SDimitry Andric 
392504eeddc0SDimitry Andric StmtResult Sema::BuildReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp,
392604eeddc0SDimitry Andric                                  bool AllowRecovery) {
39270b57cec5SDimitry Andric   // Check for unexpanded parameter packs.
39280b57cec5SDimitry Andric   if (RetValExp && DiagnoseUnexpandedParameterPack(RetValExp))
39290b57cec5SDimitry Andric     return StmtError();
39300b57cec5SDimitry Andric 
3931349cc55cSDimitry Andric   // HACK: We suppress simpler implicit move here in msvc compatibility mode
3932fe6060f1SDimitry Andric   // just as a temporary work around, as the MSVC STL has issues with
3933fe6060f1SDimitry Andric   // this change.
3934fe6060f1SDimitry Andric   bool SupressSimplerImplicitMoves =
3935fe6060f1SDimitry Andric       CheckSimplerImplicitMovesMSVCWorkaround(*this, RetValExp);
3936fe6060f1SDimitry Andric   NamedReturnInfo NRInfo = getNamedReturnInfo(
3937fe6060f1SDimitry Andric       RetValExp, SupressSimplerImplicitMoves ? SimplerImplicitMoveMode::ForceOff
3938fe6060f1SDimitry Andric                                              : SimplerImplicitMoveMode::Normal);
3939fe6060f1SDimitry Andric 
39400b57cec5SDimitry Andric   if (isa<CapturingScopeInfo>(getCurFunction()))
3941fe6060f1SDimitry Andric     return ActOnCapScopeReturnStmt(ReturnLoc, RetValExp, NRInfo,
3942fe6060f1SDimitry Andric                                    SupressSimplerImplicitMoves);
39430b57cec5SDimitry Andric 
39440b57cec5SDimitry Andric   QualType FnRetType;
39450b57cec5SDimitry Andric   QualType RelatedRetType;
39460b57cec5SDimitry Andric   const AttrVec *Attrs = nullptr;
39470b57cec5SDimitry Andric   bool isObjCMethod = false;
39480b57cec5SDimitry Andric 
39490b57cec5SDimitry Andric   if (const FunctionDecl *FD = getCurFunctionDecl()) {
39500b57cec5SDimitry Andric     FnRetType = FD->getReturnType();
39510b57cec5SDimitry Andric     if (FD->hasAttrs())
39520b57cec5SDimitry Andric       Attrs = &FD->getAttrs();
39530b57cec5SDimitry Andric     if (FD->isNoReturn())
3954e8d8bef9SDimitry Andric       Diag(ReturnLoc, diag::warn_noreturn_function_has_return_expr) << FD;
39550b57cec5SDimitry Andric     if (FD->isMain() && RetValExp)
39560b57cec5SDimitry Andric       if (isa<CXXBoolLiteralExpr>(RetValExp))
39570b57cec5SDimitry Andric         Diag(ReturnLoc, diag::warn_main_returns_bool_literal)
39580b57cec5SDimitry Andric             << RetValExp->getSourceRange();
39595ffd83dbSDimitry Andric     if (FD->hasAttr<CmseNSEntryAttr>() && RetValExp) {
39605ffd83dbSDimitry Andric       if (const auto *RT = dyn_cast<RecordType>(FnRetType.getCanonicalType())) {
39615ffd83dbSDimitry Andric         if (RT->getDecl()->isOrContainsUnion())
39625ffd83dbSDimitry Andric           Diag(RetValExp->getBeginLoc(), diag::warn_cmse_nonsecure_union) << 1;
39635ffd83dbSDimitry Andric       }
39645ffd83dbSDimitry Andric     }
39650b57cec5SDimitry Andric   } else if (ObjCMethodDecl *MD = getCurMethodDecl()) {
39660b57cec5SDimitry Andric     FnRetType = MD->getReturnType();
39670b57cec5SDimitry Andric     isObjCMethod = true;
39680b57cec5SDimitry Andric     if (MD->hasAttrs())
39690b57cec5SDimitry Andric       Attrs = &MD->getAttrs();
39700b57cec5SDimitry Andric     if (MD->hasRelatedResultType() && MD->getClassInterface()) {
39710b57cec5SDimitry Andric       // In the implementation of a method with a related return type, the
39720b57cec5SDimitry Andric       // type used to type-check the validity of return statements within the
39730b57cec5SDimitry Andric       // method body is a pointer to the type of the class being implemented.
39740b57cec5SDimitry Andric       RelatedRetType = Context.getObjCInterfaceType(MD->getClassInterface());
39750b57cec5SDimitry Andric       RelatedRetType = Context.getObjCObjectPointerType(RelatedRetType);
39760b57cec5SDimitry Andric     }
39770b57cec5SDimitry Andric   } else // If we don't have a function/method context, bail.
39780b57cec5SDimitry Andric     return StmtError();
39790b57cec5SDimitry Andric 
39800b57cec5SDimitry Andric   // C++1z: discarded return statements are not considered when deducing a
39810b57cec5SDimitry Andric   // return type.
39824824e7fdSDimitry Andric   if (ExprEvalContexts.back().isDiscardedStatementContext() &&
39830b57cec5SDimitry Andric       FnRetType->getContainedAutoType()) {
39840b57cec5SDimitry Andric     if (RetValExp) {
39850b57cec5SDimitry Andric       ExprResult ER =
39860b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
39870b57cec5SDimitry Andric       if (ER.isInvalid())
39880b57cec5SDimitry Andric         return StmtError();
39890b57cec5SDimitry Andric       RetValExp = ER.get();
39900b57cec5SDimitry Andric     }
39910b57cec5SDimitry Andric     return ReturnStmt::Create(Context, ReturnLoc, RetValExp,
39920b57cec5SDimitry Andric                               /* NRVOCandidate=*/nullptr);
39930b57cec5SDimitry Andric   }
39940b57cec5SDimitry Andric 
39950b57cec5SDimitry Andric   // FIXME: Add a flag to the ScopeInfo to indicate whether we're performing
39960b57cec5SDimitry Andric   // deduction.
39970b57cec5SDimitry Andric   if (getLangOpts().CPlusPlus14) {
39980b57cec5SDimitry Andric     if (AutoType *AT = FnRetType->getContainedAutoType()) {
39990b57cec5SDimitry Andric       FunctionDecl *FD = cast<FunctionDecl>(CurContext);
4000e8d8bef9SDimitry Andric       // If we've already decided this function is invalid, e.g. because
4001e8d8bef9SDimitry Andric       // we saw a `return` whose expression had an error, don't keep
4002e8d8bef9SDimitry Andric       // trying to deduce its return type.
400304eeddc0SDimitry Andric       // (Some return values may be needlessly wrapped in RecoveryExpr).
400404eeddc0SDimitry Andric       if (FD->isInvalidDecl() ||
400504eeddc0SDimitry Andric           DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
40060b57cec5SDimitry Andric         FD->setInvalidDecl();
400704eeddc0SDimitry Andric         if (!AllowRecovery)
40080b57cec5SDimitry Andric           return StmtError();
400904eeddc0SDimitry Andric         // The deduction failure is diagnosed and marked, try to recover.
401004eeddc0SDimitry Andric         if (RetValExp) {
401104eeddc0SDimitry Andric           // Wrap return value with a recovery expression of the previous type.
401204eeddc0SDimitry Andric           // If no deduction yet, use DependentTy.
401304eeddc0SDimitry Andric           auto Recovery = CreateRecoveryExpr(
401404eeddc0SDimitry Andric               RetValExp->getBeginLoc(), RetValExp->getEndLoc(), RetValExp,
401504eeddc0SDimitry Andric               AT->isDeduced() ? FnRetType : QualType());
401604eeddc0SDimitry Andric           if (Recovery.isInvalid())
401704eeddc0SDimitry Andric             return StmtError();
401804eeddc0SDimitry Andric           RetValExp = Recovery.get();
401904eeddc0SDimitry Andric         } else {
402004eeddc0SDimitry Andric           // Nothing to do: a ReturnStmt with no value is fine recovery.
402104eeddc0SDimitry Andric         }
40220b57cec5SDimitry Andric       } else {
40230b57cec5SDimitry Andric         FnRetType = FD->getReturnType();
40240b57cec5SDimitry Andric       }
40250b57cec5SDimitry Andric     }
40260b57cec5SDimitry Andric   }
4027fe6060f1SDimitry Andric   const VarDecl *NRVOCandidate = getCopyElisionCandidate(NRInfo, FnRetType);
40280b57cec5SDimitry Andric 
40290b57cec5SDimitry Andric   bool HasDependentReturnType = FnRetType->isDependentType();
40300b57cec5SDimitry Andric 
40310b57cec5SDimitry Andric   ReturnStmt *Result = nullptr;
40320b57cec5SDimitry Andric   if (FnRetType->isVoidType()) {
40330b57cec5SDimitry Andric     if (RetValExp) {
403404eeddc0SDimitry Andric       if (auto *ILE = dyn_cast<InitListExpr>(RetValExp)) {
40350b57cec5SDimitry Andric         // We simply never allow init lists as the return value of void
40360b57cec5SDimitry Andric         // functions. This is compatible because this was never allowed before,
40370b57cec5SDimitry Andric         // so there's no legacy code to deal with.
40380b57cec5SDimitry Andric         NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
40390b57cec5SDimitry Andric         int FunctionKind = 0;
40400b57cec5SDimitry Andric         if (isa<ObjCMethodDecl>(CurDecl))
40410b57cec5SDimitry Andric           FunctionKind = 1;
40420b57cec5SDimitry Andric         else if (isa<CXXConstructorDecl>(CurDecl))
40430b57cec5SDimitry Andric           FunctionKind = 2;
40440b57cec5SDimitry Andric         else if (isa<CXXDestructorDecl>(CurDecl))
40450b57cec5SDimitry Andric           FunctionKind = 3;
40460b57cec5SDimitry Andric 
40470b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_return_init_list)
4048e8d8bef9SDimitry Andric             << CurDecl << FunctionKind << RetValExp->getSourceRange();
40490b57cec5SDimitry Andric 
405004eeddc0SDimitry Andric         // Preserve the initializers in the AST.
405104eeddc0SDimitry Andric         RetValExp = AllowRecovery
405204eeddc0SDimitry Andric                         ? CreateRecoveryExpr(ILE->getLBraceLoc(),
405304eeddc0SDimitry Andric                                              ILE->getRBraceLoc(), ILE->inits())
405404eeddc0SDimitry Andric                               .get()
405504eeddc0SDimitry Andric                         : nullptr;
40560b57cec5SDimitry Andric       } else if (!RetValExp->isTypeDependent()) {
40570b57cec5SDimitry Andric         // C99 6.8.6.4p1 (ext_ since GCC warns)
40580b57cec5SDimitry Andric         unsigned D = diag::ext_return_has_expr;
40590b57cec5SDimitry Andric         if (RetValExp->getType()->isVoidType()) {
40600b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
40610b57cec5SDimitry Andric           if (isa<CXXConstructorDecl>(CurDecl) ||
40620b57cec5SDimitry Andric               isa<CXXDestructorDecl>(CurDecl))
40630b57cec5SDimitry Andric             D = diag::err_ctor_dtor_returns_void;
40640b57cec5SDimitry Andric           else
40650b57cec5SDimitry Andric             D = diag::ext_return_has_void_expr;
40660b57cec5SDimitry Andric         }
40670b57cec5SDimitry Andric         else {
40680b57cec5SDimitry Andric           ExprResult Result = RetValExp;
40690b57cec5SDimitry Andric           Result = IgnoredValueConversions(Result.get());
40700b57cec5SDimitry Andric           if (Result.isInvalid())
40710b57cec5SDimitry Andric             return StmtError();
40720b57cec5SDimitry Andric           RetValExp = Result.get();
40730b57cec5SDimitry Andric           RetValExp = ImpCastExprToType(RetValExp,
40740b57cec5SDimitry Andric                                         Context.VoidTy, CK_ToVoid).get();
40750b57cec5SDimitry Andric         }
40760b57cec5SDimitry Andric         // return of void in constructor/destructor is illegal in C++.
40770b57cec5SDimitry Andric         if (D == diag::err_ctor_dtor_returns_void) {
40780b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
4079e8d8bef9SDimitry Andric           Diag(ReturnLoc, D) << CurDecl << isa<CXXDestructorDecl>(CurDecl)
40800b57cec5SDimitry Andric                              << RetValExp->getSourceRange();
40810b57cec5SDimitry Andric         }
40820b57cec5SDimitry Andric         // return (some void expression); is legal in C++.
40830b57cec5SDimitry Andric         else if (D != diag::ext_return_has_void_expr ||
40840b57cec5SDimitry Andric                  !getLangOpts().CPlusPlus) {
40850b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
40860b57cec5SDimitry Andric 
40870b57cec5SDimitry Andric           int FunctionKind = 0;
40880b57cec5SDimitry Andric           if (isa<ObjCMethodDecl>(CurDecl))
40890b57cec5SDimitry Andric             FunctionKind = 1;
40900b57cec5SDimitry Andric           else if (isa<CXXConstructorDecl>(CurDecl))
40910b57cec5SDimitry Andric             FunctionKind = 2;
40920b57cec5SDimitry Andric           else if (isa<CXXDestructorDecl>(CurDecl))
40930b57cec5SDimitry Andric             FunctionKind = 3;
40940b57cec5SDimitry Andric 
40950b57cec5SDimitry Andric           Diag(ReturnLoc, D)
4096e8d8bef9SDimitry Andric               << CurDecl << FunctionKind << RetValExp->getSourceRange();
40970b57cec5SDimitry Andric         }
40980b57cec5SDimitry Andric       }
40990b57cec5SDimitry Andric 
41000b57cec5SDimitry Andric       if (RetValExp) {
41010b57cec5SDimitry Andric         ExprResult ER =
41020b57cec5SDimitry Andric             ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
41030b57cec5SDimitry Andric         if (ER.isInvalid())
41040b57cec5SDimitry Andric           return StmtError();
41050b57cec5SDimitry Andric         RetValExp = ER.get();
41060b57cec5SDimitry Andric       }
41070b57cec5SDimitry Andric     }
41080b57cec5SDimitry Andric 
41090b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, RetValExp,
41100b57cec5SDimitry Andric                                 /* NRVOCandidate=*/nullptr);
41110b57cec5SDimitry Andric   } else if (!RetValExp && !HasDependentReturnType) {
41120b57cec5SDimitry Andric     FunctionDecl *FD = getCurFunctionDecl();
41130b57cec5SDimitry Andric 
411481ad6265SDimitry Andric     if ((FD && FD->isInvalidDecl()) || FnRetType->containsErrors()) {
411581ad6265SDimitry Andric       // The intended return type might have been "void", so don't warn.
411681ad6265SDimitry Andric     } else if (getLangOpts().CPlusPlus11 && FD && FD->isConstexpr()) {
41170b57cec5SDimitry Andric       // C++11 [stmt.return]p2
4118e8d8bef9SDimitry Andric       Diag(ReturnLoc, diag::err_constexpr_return_missing_expr)
4119e8d8bef9SDimitry Andric           << FD << FD->isConsteval();
41200b57cec5SDimitry Andric       FD->setInvalidDecl();
41210b57cec5SDimitry Andric     } else {
4122e8d8bef9SDimitry Andric       // C99 6.8.6.4p1 (ext_ since GCC warns)
41230b57cec5SDimitry Andric       // C90 6.6.6.4p4
4124e8d8bef9SDimitry Andric       unsigned DiagID = getLangOpts().C99 ? diag::ext_return_missing_expr
4125e8d8bef9SDimitry Andric                                           : diag::warn_return_missing_expr;
4126e8d8bef9SDimitry Andric       // Note that at this point one of getCurFunctionDecl() or
4127e8d8bef9SDimitry Andric       // getCurMethodDecl() must be non-null (see above).
4128e8d8bef9SDimitry Andric       assert((getCurFunctionDecl() || getCurMethodDecl()) &&
4129e8d8bef9SDimitry Andric              "Not in a FunctionDecl or ObjCMethodDecl?");
4130e8d8bef9SDimitry Andric       bool IsMethod = FD == nullptr;
4131e8d8bef9SDimitry Andric       const NamedDecl *ND =
4132e8d8bef9SDimitry Andric           IsMethod ? cast<NamedDecl>(getCurMethodDecl()) : cast<NamedDecl>(FD);
4133e8d8bef9SDimitry Andric       Diag(ReturnLoc, DiagID) << ND << IsMethod;
41340b57cec5SDimitry Andric     }
41350b57cec5SDimitry Andric 
41360b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, /* RetExpr=*/nullptr,
41370b57cec5SDimitry Andric                                 /* NRVOCandidate=*/nullptr);
41380b57cec5SDimitry Andric   } else {
41390b57cec5SDimitry Andric     assert(RetValExp || HasDependentReturnType);
41400b57cec5SDimitry Andric     QualType RetType = RelatedRetType.isNull() ? FnRetType : RelatedRetType;
41410b57cec5SDimitry Andric 
41420b57cec5SDimitry Andric     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
41430b57cec5SDimitry Andric     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
41440b57cec5SDimitry Andric     // function return.
41450b57cec5SDimitry Andric 
41460b57cec5SDimitry Andric     // In C++ the return statement is handled via a copy initialization,
41470b57cec5SDimitry Andric     // the C version of which boils down to CheckSingleAssignmentConstraints.
41480b57cec5SDimitry Andric     if (!HasDependentReturnType && !RetValExp->isTypeDependent()) {
41490b57cec5SDimitry Andric       // we have a non-void function with an expression, continue checking
415028a41182SDimitry Andric       InitializedEntity Entity =
415128a41182SDimitry Andric           InitializedEntity::InitializeResult(ReturnLoc, RetType);
4152fe6060f1SDimitry Andric       ExprResult Res = PerformMoveOrCopyInitialization(
4153fe6060f1SDimitry Andric           Entity, NRInfo, RetValExp, SupressSimplerImplicitMoves);
415404eeddc0SDimitry Andric       if (Res.isInvalid() && AllowRecovery)
415504eeddc0SDimitry Andric         Res = CreateRecoveryExpr(RetValExp->getBeginLoc(),
415604eeddc0SDimitry Andric                                  RetValExp->getEndLoc(), RetValExp, RetType);
41570b57cec5SDimitry Andric       if (Res.isInvalid()) {
41580b57cec5SDimitry Andric         // FIXME: Clean up temporaries here anyway?
41590b57cec5SDimitry Andric         return StmtError();
41600b57cec5SDimitry Andric       }
41610b57cec5SDimitry Andric       RetValExp = Res.getAs<Expr>();
41620b57cec5SDimitry Andric 
41630b57cec5SDimitry Andric       // If we have a related result type, we need to implicitly
41640b57cec5SDimitry Andric       // convert back to the formal result type.  We can't pretend to
41650b57cec5SDimitry Andric       // initialize the result again --- we might end double-retaining
41660b57cec5SDimitry Andric       // --- so instead we initialize a notional temporary.
41670b57cec5SDimitry Andric       if (!RelatedRetType.isNull()) {
41680b57cec5SDimitry Andric         Entity = InitializedEntity::InitializeRelatedResult(getCurMethodDecl(),
41690b57cec5SDimitry Andric                                                             FnRetType);
41700b57cec5SDimitry Andric         Res = PerformCopyInitialization(Entity, ReturnLoc, RetValExp);
41710b57cec5SDimitry Andric         if (Res.isInvalid()) {
41720b57cec5SDimitry Andric           // FIXME: Clean up temporaries here anyway?
41730b57cec5SDimitry Andric           return StmtError();
41740b57cec5SDimitry Andric         }
41750b57cec5SDimitry Andric         RetValExp = Res.getAs<Expr>();
41760b57cec5SDimitry Andric       }
41770b57cec5SDimitry Andric 
41780b57cec5SDimitry Andric       CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc, isObjCMethod, Attrs,
41790b57cec5SDimitry Andric                          getCurFunctionDecl());
41800b57cec5SDimitry Andric     }
41810b57cec5SDimitry Andric 
41820b57cec5SDimitry Andric     if (RetValExp) {
41830b57cec5SDimitry Andric       ExprResult ER =
41840b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
41850b57cec5SDimitry Andric       if (ER.isInvalid())
41860b57cec5SDimitry Andric         return StmtError();
41870b57cec5SDimitry Andric       RetValExp = ER.get();
41880b57cec5SDimitry Andric     }
41890b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, RetValExp, NRVOCandidate);
41900b57cec5SDimitry Andric   }
41910b57cec5SDimitry Andric 
41920b57cec5SDimitry Andric   // If we need to check for the named return value optimization, save the
41930b57cec5SDimitry Andric   // return statement in our scope for later processing.
41940b57cec5SDimitry Andric   if (Result->getNRVOCandidate())
41950b57cec5SDimitry Andric     FunctionScopes.back()->Returns.push_back(Result);
41960b57cec5SDimitry Andric 
41970b57cec5SDimitry Andric   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
41980b57cec5SDimitry Andric     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
41990b57cec5SDimitry Andric 
42000b57cec5SDimitry Andric   return Result;
42010b57cec5SDimitry Andric }
42020b57cec5SDimitry Andric 
42030b57cec5SDimitry Andric StmtResult
42040b57cec5SDimitry Andric Sema::ActOnObjCAtCatchStmt(SourceLocation AtLoc,
42050b57cec5SDimitry Andric                            SourceLocation RParen, Decl *Parm,
42060b57cec5SDimitry Andric                            Stmt *Body) {
42070b57cec5SDimitry Andric   VarDecl *Var = cast_or_null<VarDecl>(Parm);
42080b57cec5SDimitry Andric   if (Var && Var->isInvalidDecl())
42090b57cec5SDimitry Andric     return StmtError();
42100b57cec5SDimitry Andric 
42110b57cec5SDimitry Andric   return new (Context) ObjCAtCatchStmt(AtLoc, RParen, Var, Body);
42120b57cec5SDimitry Andric }
42130b57cec5SDimitry Andric 
42140b57cec5SDimitry Andric StmtResult
42150b57cec5SDimitry Andric Sema::ActOnObjCAtFinallyStmt(SourceLocation AtLoc, Stmt *Body) {
42160b57cec5SDimitry Andric   return new (Context) ObjCAtFinallyStmt(AtLoc, Body);
42170b57cec5SDimitry Andric }
42180b57cec5SDimitry Andric 
42190b57cec5SDimitry Andric StmtResult
42200b57cec5SDimitry Andric Sema::ActOnObjCAtTryStmt(SourceLocation AtLoc, Stmt *Try,
42210b57cec5SDimitry Andric                          MultiStmtArg CatchStmts, Stmt *Finally) {
42220b57cec5SDimitry Andric   if (!getLangOpts().ObjCExceptions)
42230b57cec5SDimitry Andric     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@try";
42240b57cec5SDimitry Andric 
4225349cc55cSDimitry Andric   // Objective-C try is incompatible with SEH __try.
4226349cc55cSDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
4227349cc55cSDimitry Andric   if (FSI->FirstSEHTryLoc.isValid()) {
4228349cc55cSDimitry Andric     Diag(AtLoc, diag::err_mixing_cxx_try_seh_try) << 1;
4229349cc55cSDimitry Andric     Diag(FSI->FirstSEHTryLoc, diag::note_conflicting_try_here) << "'__try'";
4230349cc55cSDimitry Andric   }
4231349cc55cSDimitry Andric 
4232349cc55cSDimitry Andric   FSI->setHasObjCTry(AtLoc);
42330b57cec5SDimitry Andric   unsigned NumCatchStmts = CatchStmts.size();
42340b57cec5SDimitry Andric   return ObjCAtTryStmt::Create(Context, AtLoc, Try, CatchStmts.data(),
42350b57cec5SDimitry Andric                                NumCatchStmts, Finally);
42360b57cec5SDimitry Andric }
42370b57cec5SDimitry Andric 
42380b57cec5SDimitry Andric StmtResult Sema::BuildObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw) {
42390b57cec5SDimitry Andric   if (Throw) {
42400b57cec5SDimitry Andric     ExprResult Result = DefaultLvalueConversion(Throw);
42410b57cec5SDimitry Andric     if (Result.isInvalid())
42420b57cec5SDimitry Andric       return StmtError();
42430b57cec5SDimitry Andric 
42440b57cec5SDimitry Andric     Result = ActOnFinishFullExpr(Result.get(), /*DiscardedValue*/ false);
42450b57cec5SDimitry Andric     if (Result.isInvalid())
42460b57cec5SDimitry Andric       return StmtError();
42470b57cec5SDimitry Andric     Throw = Result.get();
42480b57cec5SDimitry Andric 
42490b57cec5SDimitry Andric     QualType ThrowType = Throw->getType();
42500b57cec5SDimitry Andric     // Make sure the expression type is an ObjC pointer or "void *".
42510b57cec5SDimitry Andric     if (!ThrowType->isDependentType() &&
42520b57cec5SDimitry Andric         !ThrowType->isObjCObjectPointerType()) {
42530b57cec5SDimitry Andric       const PointerType *PT = ThrowType->getAs<PointerType>();
42540b57cec5SDimitry Andric       if (!PT || !PT->getPointeeType()->isVoidType())
42550b57cec5SDimitry Andric         return StmtError(Diag(AtLoc, diag::err_objc_throw_expects_object)
42560b57cec5SDimitry Andric                          << Throw->getType() << Throw->getSourceRange());
42570b57cec5SDimitry Andric     }
42580b57cec5SDimitry Andric   }
42590b57cec5SDimitry Andric 
42600b57cec5SDimitry Andric   return new (Context) ObjCAtThrowStmt(AtLoc, Throw);
42610b57cec5SDimitry Andric }
42620b57cec5SDimitry Andric 
42630b57cec5SDimitry Andric StmtResult
42640b57cec5SDimitry Andric Sema::ActOnObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw,
42650b57cec5SDimitry Andric                            Scope *CurScope) {
42660b57cec5SDimitry Andric   if (!getLangOpts().ObjCExceptions)
42670b57cec5SDimitry Andric     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@throw";
42680b57cec5SDimitry Andric 
42690b57cec5SDimitry Andric   if (!Throw) {
42700b57cec5SDimitry Andric     // @throw without an expression designates a rethrow (which must occur
42710b57cec5SDimitry Andric     // in the context of an @catch clause).
42720b57cec5SDimitry Andric     Scope *AtCatchParent = CurScope;
42730b57cec5SDimitry Andric     while (AtCatchParent && !AtCatchParent->isAtCatchScope())
42740b57cec5SDimitry Andric       AtCatchParent = AtCatchParent->getParent();
42750b57cec5SDimitry Andric     if (!AtCatchParent)
42760b57cec5SDimitry Andric       return StmtError(Diag(AtLoc, diag::err_rethrow_used_outside_catch));
42770b57cec5SDimitry Andric   }
42780b57cec5SDimitry Andric   return BuildObjCAtThrowStmt(AtLoc, Throw);
42790b57cec5SDimitry Andric }
42800b57cec5SDimitry Andric 
42810b57cec5SDimitry Andric ExprResult
42820b57cec5SDimitry Andric Sema::ActOnObjCAtSynchronizedOperand(SourceLocation atLoc, Expr *operand) {
42830b57cec5SDimitry Andric   ExprResult result = DefaultLvalueConversion(operand);
42840b57cec5SDimitry Andric   if (result.isInvalid())
42850b57cec5SDimitry Andric     return ExprError();
42860b57cec5SDimitry Andric   operand = result.get();
42870b57cec5SDimitry Andric 
42880b57cec5SDimitry Andric   // Make sure the expression type is an ObjC pointer or "void *".
42890b57cec5SDimitry Andric   QualType type = operand->getType();
42900b57cec5SDimitry Andric   if (!type->isDependentType() &&
42910b57cec5SDimitry Andric       !type->isObjCObjectPointerType()) {
42920b57cec5SDimitry Andric     const PointerType *pointerType = type->getAs<PointerType>();
42930b57cec5SDimitry Andric     if (!pointerType || !pointerType->getPointeeType()->isVoidType()) {
42940b57cec5SDimitry Andric       if (getLangOpts().CPlusPlus) {
42950b57cec5SDimitry Andric         if (RequireCompleteType(atLoc, type,
42960b57cec5SDimitry Andric                                 diag::err_incomplete_receiver_type))
42970b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
42980b57cec5SDimitry Andric                    << type << operand->getSourceRange();
42990b57cec5SDimitry Andric 
43000b57cec5SDimitry Andric         ExprResult result = PerformContextuallyConvertToObjCPointer(operand);
43010b57cec5SDimitry Andric         if (result.isInvalid())
43020b57cec5SDimitry Andric           return ExprError();
43030b57cec5SDimitry Andric         if (!result.isUsable())
43040b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
43050b57cec5SDimitry Andric                    << type << operand->getSourceRange();
43060b57cec5SDimitry Andric 
43070b57cec5SDimitry Andric         operand = result.get();
43080b57cec5SDimitry Andric       } else {
43090b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
43100b57cec5SDimitry Andric                    << type << operand->getSourceRange();
43110b57cec5SDimitry Andric       }
43120b57cec5SDimitry Andric     }
43130b57cec5SDimitry Andric   }
43140b57cec5SDimitry Andric 
43150b57cec5SDimitry Andric   // The operand to @synchronized is a full-expression.
43160b57cec5SDimitry Andric   return ActOnFinishFullExpr(operand, /*DiscardedValue*/ false);
43170b57cec5SDimitry Andric }
43180b57cec5SDimitry Andric 
43190b57cec5SDimitry Andric StmtResult
43200b57cec5SDimitry Andric Sema::ActOnObjCAtSynchronizedStmt(SourceLocation AtLoc, Expr *SyncExpr,
43210b57cec5SDimitry Andric                                   Stmt *SyncBody) {
43220b57cec5SDimitry Andric   // We can't jump into or indirect-jump out of a @synchronized block.
43230b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
43240b57cec5SDimitry Andric   return new (Context) ObjCAtSynchronizedStmt(AtLoc, SyncExpr, SyncBody);
43250b57cec5SDimitry Andric }
43260b57cec5SDimitry Andric 
43270b57cec5SDimitry Andric /// ActOnCXXCatchBlock - Takes an exception declaration and a handler block
43280b57cec5SDimitry Andric /// and creates a proper catch handler from them.
43290b57cec5SDimitry Andric StmtResult
43300b57cec5SDimitry Andric Sema::ActOnCXXCatchBlock(SourceLocation CatchLoc, Decl *ExDecl,
43310b57cec5SDimitry Andric                          Stmt *HandlerBlock) {
43320b57cec5SDimitry Andric   // There's nothing to test that ActOnExceptionDecl didn't already test.
43330b57cec5SDimitry Andric   return new (Context)
43340b57cec5SDimitry Andric       CXXCatchStmt(CatchLoc, cast_or_null<VarDecl>(ExDecl), HandlerBlock);
43350b57cec5SDimitry Andric }
43360b57cec5SDimitry Andric 
43370b57cec5SDimitry Andric StmtResult
43380b57cec5SDimitry Andric Sema::ActOnObjCAutoreleasePoolStmt(SourceLocation AtLoc, Stmt *Body) {
43390b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
43400b57cec5SDimitry Andric   return new (Context) ObjCAutoreleasePoolStmt(AtLoc, Body);
43410b57cec5SDimitry Andric }
43420b57cec5SDimitry Andric 
43430b57cec5SDimitry Andric namespace {
43440b57cec5SDimitry Andric class CatchHandlerType {
43450b57cec5SDimitry Andric   QualType QT;
43460b57cec5SDimitry Andric   unsigned IsPointer : 1;
43470b57cec5SDimitry Andric 
43480b57cec5SDimitry Andric   // This is a special constructor to be used only with DenseMapInfo's
43490b57cec5SDimitry Andric   // getEmptyKey() and getTombstoneKey() functions.
43500b57cec5SDimitry Andric   friend struct llvm::DenseMapInfo<CatchHandlerType>;
43510b57cec5SDimitry Andric   enum Unique { ForDenseMap };
43520b57cec5SDimitry Andric   CatchHandlerType(QualType QT, Unique) : QT(QT), IsPointer(false) {}
43530b57cec5SDimitry Andric 
43540b57cec5SDimitry Andric public:
43550b57cec5SDimitry Andric   /// Used when creating a CatchHandlerType from a handler type; will determine
43560b57cec5SDimitry Andric   /// whether the type is a pointer or reference and will strip off the top
43570b57cec5SDimitry Andric   /// level pointer and cv-qualifiers.
43580b57cec5SDimitry Andric   CatchHandlerType(QualType Q) : QT(Q), IsPointer(false) {
43590b57cec5SDimitry Andric     if (QT->isPointerType())
43600b57cec5SDimitry Andric       IsPointer = true;
43610b57cec5SDimitry Andric 
43620b57cec5SDimitry Andric     if (IsPointer || QT->isReferenceType())
43630b57cec5SDimitry Andric       QT = QT->getPointeeType();
43640b57cec5SDimitry Andric     QT = QT.getUnqualifiedType();
43650b57cec5SDimitry Andric   }
43660b57cec5SDimitry Andric 
43670b57cec5SDimitry Andric   /// Used when creating a CatchHandlerType from a base class type; pretends the
43680b57cec5SDimitry Andric   /// type passed in had the pointer qualifier, does not need to get an
43690b57cec5SDimitry Andric   /// unqualified type.
43700b57cec5SDimitry Andric   CatchHandlerType(QualType QT, bool IsPointer)
43710b57cec5SDimitry Andric       : QT(QT), IsPointer(IsPointer) {}
43720b57cec5SDimitry Andric 
43730b57cec5SDimitry Andric   QualType underlying() const { return QT; }
43740b57cec5SDimitry Andric   bool isPointer() const { return IsPointer; }
43750b57cec5SDimitry Andric 
43760b57cec5SDimitry Andric   friend bool operator==(const CatchHandlerType &LHS,
43770b57cec5SDimitry Andric                          const CatchHandlerType &RHS) {
43780b57cec5SDimitry Andric     // If the pointer qualification does not match, we can return early.
43790b57cec5SDimitry Andric     if (LHS.IsPointer != RHS.IsPointer)
43800b57cec5SDimitry Andric       return false;
43810b57cec5SDimitry Andric     // Otherwise, check the underlying type without cv-qualifiers.
43820b57cec5SDimitry Andric     return LHS.QT == RHS.QT;
43830b57cec5SDimitry Andric   }
43840b57cec5SDimitry Andric };
43850b57cec5SDimitry Andric } // namespace
43860b57cec5SDimitry Andric 
43870b57cec5SDimitry Andric namespace llvm {
43880b57cec5SDimitry Andric template <> struct DenseMapInfo<CatchHandlerType> {
43890b57cec5SDimitry Andric   static CatchHandlerType getEmptyKey() {
43900b57cec5SDimitry Andric     return CatchHandlerType(DenseMapInfo<QualType>::getEmptyKey(),
43910b57cec5SDimitry Andric                        CatchHandlerType::ForDenseMap);
43920b57cec5SDimitry Andric   }
43930b57cec5SDimitry Andric 
43940b57cec5SDimitry Andric   static CatchHandlerType getTombstoneKey() {
43950b57cec5SDimitry Andric     return CatchHandlerType(DenseMapInfo<QualType>::getTombstoneKey(),
43960b57cec5SDimitry Andric                        CatchHandlerType::ForDenseMap);
43970b57cec5SDimitry Andric   }
43980b57cec5SDimitry Andric 
43990b57cec5SDimitry Andric   static unsigned getHashValue(const CatchHandlerType &Base) {
44000b57cec5SDimitry Andric     return DenseMapInfo<QualType>::getHashValue(Base.underlying());
44010b57cec5SDimitry Andric   }
44020b57cec5SDimitry Andric 
44030b57cec5SDimitry Andric   static bool isEqual(const CatchHandlerType &LHS,
44040b57cec5SDimitry Andric                       const CatchHandlerType &RHS) {
44050b57cec5SDimitry Andric     return LHS == RHS;
44060b57cec5SDimitry Andric   }
44070b57cec5SDimitry Andric };
44080b57cec5SDimitry Andric }
44090b57cec5SDimitry Andric 
44100b57cec5SDimitry Andric namespace {
44110b57cec5SDimitry Andric class CatchTypePublicBases {
44120b57cec5SDimitry Andric   ASTContext &Ctx;
44130b57cec5SDimitry Andric   const llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> &TypesToCheck;
44140b57cec5SDimitry Andric   const bool CheckAgainstPointer;
44150b57cec5SDimitry Andric 
44160b57cec5SDimitry Andric   CXXCatchStmt *FoundHandler;
44170b57cec5SDimitry Andric   CanQualType FoundHandlerType;
44180b57cec5SDimitry Andric 
44190b57cec5SDimitry Andric public:
44200b57cec5SDimitry Andric   CatchTypePublicBases(
44210b57cec5SDimitry Andric       ASTContext &Ctx,
44220b57cec5SDimitry Andric       const llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> &T, bool C)
44230b57cec5SDimitry Andric       : Ctx(Ctx), TypesToCheck(T), CheckAgainstPointer(C),
44240b57cec5SDimitry Andric         FoundHandler(nullptr) {}
44250b57cec5SDimitry Andric 
44260b57cec5SDimitry Andric   CXXCatchStmt *getFoundHandler() const { return FoundHandler; }
44270b57cec5SDimitry Andric   CanQualType getFoundHandlerType() const { return FoundHandlerType; }
44280b57cec5SDimitry Andric 
44290b57cec5SDimitry Andric   bool operator()(const CXXBaseSpecifier *S, CXXBasePath &) {
44300b57cec5SDimitry Andric     if (S->getAccessSpecifier() == AccessSpecifier::AS_public) {
44310b57cec5SDimitry Andric       CatchHandlerType Check(S->getType(), CheckAgainstPointer);
44320b57cec5SDimitry Andric       const auto &M = TypesToCheck;
44330b57cec5SDimitry Andric       auto I = M.find(Check);
44340b57cec5SDimitry Andric       if (I != M.end()) {
44350b57cec5SDimitry Andric         FoundHandler = I->second;
44360b57cec5SDimitry Andric         FoundHandlerType = Ctx.getCanonicalType(S->getType());
44370b57cec5SDimitry Andric         return true;
44380b57cec5SDimitry Andric       }
44390b57cec5SDimitry Andric     }
44400b57cec5SDimitry Andric     return false;
44410b57cec5SDimitry Andric   }
44420b57cec5SDimitry Andric };
44430b57cec5SDimitry Andric }
44440b57cec5SDimitry Andric 
44450b57cec5SDimitry Andric /// ActOnCXXTryBlock - Takes a try compound-statement and a number of
44460b57cec5SDimitry Andric /// handlers and creates a try statement from them.
44470b57cec5SDimitry Andric StmtResult Sema::ActOnCXXTryBlock(SourceLocation TryLoc, Stmt *TryBlock,
44480b57cec5SDimitry Andric                                   ArrayRef<Stmt *> Handlers) {
44490b57cec5SDimitry Andric   // Don't report an error if 'try' is used in system headers.
44500b57cec5SDimitry Andric   if (!getLangOpts().CXXExceptions &&
44510b57cec5SDimitry Andric       !getSourceManager().isInSystemHeader(TryLoc) && !getLangOpts().CUDA) {
44520b57cec5SDimitry Andric     // Delay error emission for the OpenMP device code.
44530b57cec5SDimitry Andric     targetDiag(TryLoc, diag::err_exceptions_disabled) << "try";
44540b57cec5SDimitry Andric   }
44550b57cec5SDimitry Andric 
44560b57cec5SDimitry Andric   // Exceptions aren't allowed in CUDA device code.
44570b57cec5SDimitry Andric   if (getLangOpts().CUDA)
44580b57cec5SDimitry Andric     CUDADiagIfDeviceCode(TryLoc, diag::err_cuda_device_exceptions)
44590b57cec5SDimitry Andric         << "try" << CurrentCUDATarget();
44600b57cec5SDimitry Andric 
44610b57cec5SDimitry Andric   if (getCurScope() && getCurScope()->isOpenMPSimdDirectiveScope())
44620b57cec5SDimitry Andric     Diag(TryLoc, diag::err_omp_simd_region_cannot_use_stmt) << "try";
44630b57cec5SDimitry Andric 
44640b57cec5SDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
44650b57cec5SDimitry Andric 
44660b57cec5SDimitry Andric   // C++ try is incompatible with SEH __try.
44670b57cec5SDimitry Andric   if (!getLangOpts().Borland && FSI->FirstSEHTryLoc.isValid()) {
4468349cc55cSDimitry Andric     Diag(TryLoc, diag::err_mixing_cxx_try_seh_try) << 0;
44690b57cec5SDimitry Andric     Diag(FSI->FirstSEHTryLoc, diag::note_conflicting_try_here) << "'__try'";
44700b57cec5SDimitry Andric   }
44710b57cec5SDimitry Andric 
44720b57cec5SDimitry Andric   const unsigned NumHandlers = Handlers.size();
44730b57cec5SDimitry Andric   assert(!Handlers.empty() &&
44740b57cec5SDimitry Andric          "The parser shouldn't call this if there are no handlers.");
44750b57cec5SDimitry Andric 
44760b57cec5SDimitry Andric   llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> HandledTypes;
44770b57cec5SDimitry Andric   for (unsigned i = 0; i < NumHandlers; ++i) {
44780b57cec5SDimitry Andric     CXXCatchStmt *H = cast<CXXCatchStmt>(Handlers[i]);
44790b57cec5SDimitry Andric 
44800b57cec5SDimitry Andric     // Diagnose when the handler is a catch-all handler, but it isn't the last
44810b57cec5SDimitry Andric     // handler for the try block. [except.handle]p5. Also, skip exception
44820b57cec5SDimitry Andric     // declarations that are invalid, since we can't usefully report on them.
44830b57cec5SDimitry Andric     if (!H->getExceptionDecl()) {
44840b57cec5SDimitry Andric       if (i < NumHandlers - 1)
44850b57cec5SDimitry Andric         return StmtError(Diag(H->getBeginLoc(), diag::err_early_catch_all));
44860b57cec5SDimitry Andric       continue;
44870b57cec5SDimitry Andric     } else if (H->getExceptionDecl()->isInvalidDecl())
44880b57cec5SDimitry Andric       continue;
44890b57cec5SDimitry Andric 
44900b57cec5SDimitry Andric     // Walk the type hierarchy to diagnose when this type has already been
44910b57cec5SDimitry Andric     // handled (duplication), or cannot be handled (derivation inversion). We
44920b57cec5SDimitry Andric     // ignore top-level cv-qualifiers, per [except.handle]p3
44930b57cec5SDimitry Andric     CatchHandlerType HandlerCHT =
44940b57cec5SDimitry Andric         (QualType)Context.getCanonicalType(H->getCaughtType());
44950b57cec5SDimitry Andric 
44960b57cec5SDimitry Andric     // We can ignore whether the type is a reference or a pointer; we need the
44970b57cec5SDimitry Andric     // underlying declaration type in order to get at the underlying record
44980b57cec5SDimitry Andric     // decl, if there is one.
44990b57cec5SDimitry Andric     QualType Underlying = HandlerCHT.underlying();
45000b57cec5SDimitry Andric     if (auto *RD = Underlying->getAsCXXRecordDecl()) {
45010b57cec5SDimitry Andric       if (!RD->hasDefinition())
45020b57cec5SDimitry Andric         continue;
45030b57cec5SDimitry Andric       // Check that none of the public, unambiguous base classes are in the
45040b57cec5SDimitry Andric       // map ([except.handle]p1). Give the base classes the same pointer
45050b57cec5SDimitry Andric       // qualification as the original type we are basing off of. This allows
45060b57cec5SDimitry Andric       // comparison against the handler type using the same top-level pointer
45070b57cec5SDimitry Andric       // as the original type.
45080b57cec5SDimitry Andric       CXXBasePaths Paths;
45090b57cec5SDimitry Andric       Paths.setOrigin(RD);
45100b57cec5SDimitry Andric       CatchTypePublicBases CTPB(Context, HandledTypes, HandlerCHT.isPointer());
45110b57cec5SDimitry Andric       if (RD->lookupInBases(CTPB, Paths)) {
45120b57cec5SDimitry Andric         const CXXCatchStmt *Problem = CTPB.getFoundHandler();
45130b57cec5SDimitry Andric         if (!Paths.isAmbiguous(CTPB.getFoundHandlerType())) {
45140b57cec5SDimitry Andric           Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
45150b57cec5SDimitry Andric                diag::warn_exception_caught_by_earlier_handler)
45160b57cec5SDimitry Andric               << H->getCaughtType();
45170b57cec5SDimitry Andric           Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
45180b57cec5SDimitry Andric                 diag::note_previous_exception_handler)
45190b57cec5SDimitry Andric               << Problem->getCaughtType();
45200b57cec5SDimitry Andric         }
45210b57cec5SDimitry Andric       }
45220b57cec5SDimitry Andric     }
45230b57cec5SDimitry Andric 
45240b57cec5SDimitry Andric     // Add the type the list of ones we have handled; diagnose if we've already
45250b57cec5SDimitry Andric     // handled it.
45260b57cec5SDimitry Andric     auto R = HandledTypes.insert(std::make_pair(H->getCaughtType(), H));
45270b57cec5SDimitry Andric     if (!R.second) {
45280b57cec5SDimitry Andric       const CXXCatchStmt *Problem = R.first->second;
45290b57cec5SDimitry Andric       Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
45300b57cec5SDimitry Andric            diag::warn_exception_caught_by_earlier_handler)
45310b57cec5SDimitry Andric           << H->getCaughtType();
45320b57cec5SDimitry Andric       Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
45330b57cec5SDimitry Andric            diag::note_previous_exception_handler)
45340b57cec5SDimitry Andric           << Problem->getCaughtType();
45350b57cec5SDimitry Andric     }
45360b57cec5SDimitry Andric   }
45370b57cec5SDimitry Andric 
45380b57cec5SDimitry Andric   FSI->setHasCXXTry(TryLoc);
45390b57cec5SDimitry Andric 
45400b57cec5SDimitry Andric   return CXXTryStmt::Create(Context, TryLoc, TryBlock, Handlers);
45410b57cec5SDimitry Andric }
45420b57cec5SDimitry Andric 
45430b57cec5SDimitry Andric StmtResult Sema::ActOnSEHTryBlock(bool IsCXXTry, SourceLocation TryLoc,
45440b57cec5SDimitry Andric                                   Stmt *TryBlock, Stmt *Handler) {
45450b57cec5SDimitry Andric   assert(TryBlock && Handler);
45460b57cec5SDimitry Andric 
45470b57cec5SDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
45480b57cec5SDimitry Andric 
45490b57cec5SDimitry Andric   // SEH __try is incompatible with C++ try. Borland appears to support this,
45500b57cec5SDimitry Andric   // however.
45510b57cec5SDimitry Andric   if (!getLangOpts().Borland) {
4552349cc55cSDimitry Andric     if (FSI->FirstCXXOrObjCTryLoc.isValid()) {
4553349cc55cSDimitry Andric       Diag(TryLoc, diag::err_mixing_cxx_try_seh_try) << FSI->FirstTryType;
4554349cc55cSDimitry Andric       Diag(FSI->FirstCXXOrObjCTryLoc, diag::note_conflicting_try_here)
4555349cc55cSDimitry Andric           << (FSI->FirstTryType == sema::FunctionScopeInfo::TryLocIsCXX
4556349cc55cSDimitry Andric                   ? "'try'"
4557349cc55cSDimitry Andric                   : "'@try'");
45580b57cec5SDimitry Andric     }
45590b57cec5SDimitry Andric   }
45600b57cec5SDimitry Andric 
45610b57cec5SDimitry Andric   FSI->setHasSEHTry(TryLoc);
45620b57cec5SDimitry Andric 
45630b57cec5SDimitry Andric   // Reject __try in Obj-C methods, blocks, and captured decls, since we don't
45640b57cec5SDimitry Andric   // track if they use SEH.
45650b57cec5SDimitry Andric   DeclContext *DC = CurContext;
45660b57cec5SDimitry Andric   while (DC && !DC->isFunctionOrMethod())
45670b57cec5SDimitry Andric     DC = DC->getParent();
45680b57cec5SDimitry Andric   FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(DC);
45690b57cec5SDimitry Andric   if (FD)
45700b57cec5SDimitry Andric     FD->setUsesSEHTry(true);
45710b57cec5SDimitry Andric   else
45720b57cec5SDimitry Andric     Diag(TryLoc, diag::err_seh_try_outside_functions);
45730b57cec5SDimitry Andric 
45740b57cec5SDimitry Andric   // Reject __try on unsupported targets.
45750b57cec5SDimitry Andric   if (!Context.getTargetInfo().isSEHTrySupported())
45760b57cec5SDimitry Andric     Diag(TryLoc, diag::err_seh_try_unsupported);
45770b57cec5SDimitry Andric 
45780b57cec5SDimitry Andric   return SEHTryStmt::Create(Context, IsCXXTry, TryLoc, TryBlock, Handler);
45790b57cec5SDimitry Andric }
45800b57cec5SDimitry Andric 
4581480093f4SDimitry Andric StmtResult Sema::ActOnSEHExceptBlock(SourceLocation Loc, Expr *FilterExpr,
45820b57cec5SDimitry Andric                                      Stmt *Block) {
45830b57cec5SDimitry Andric   assert(FilterExpr && Block);
4584480093f4SDimitry Andric   QualType FTy = FilterExpr->getType();
4585480093f4SDimitry Andric   if (!FTy->isIntegerType() && !FTy->isDependentType()) {
4586480093f4SDimitry Andric     return StmtError(
4587480093f4SDimitry Andric         Diag(FilterExpr->getExprLoc(), diag::err_filter_expression_integral)
4588480093f4SDimitry Andric         << FTy);
45890b57cec5SDimitry Andric   }
45900b57cec5SDimitry Andric   return SEHExceptStmt::Create(Context, Loc, FilterExpr, Block);
45910b57cec5SDimitry Andric }
45920b57cec5SDimitry Andric 
45930b57cec5SDimitry Andric void Sema::ActOnStartSEHFinallyBlock() {
45940b57cec5SDimitry Andric   CurrentSEHFinally.push_back(CurScope);
45950b57cec5SDimitry Andric }
45960b57cec5SDimitry Andric 
45970b57cec5SDimitry Andric void Sema::ActOnAbortSEHFinallyBlock() {
45980b57cec5SDimitry Andric   CurrentSEHFinally.pop_back();
45990b57cec5SDimitry Andric }
46000b57cec5SDimitry Andric 
46010b57cec5SDimitry Andric StmtResult Sema::ActOnFinishSEHFinallyBlock(SourceLocation Loc, Stmt *Block) {
46020b57cec5SDimitry Andric   assert(Block);
46030b57cec5SDimitry Andric   CurrentSEHFinally.pop_back();
46040b57cec5SDimitry Andric   return SEHFinallyStmt::Create(Context, Loc, Block);
46050b57cec5SDimitry Andric }
46060b57cec5SDimitry Andric 
46070b57cec5SDimitry Andric StmtResult
46080b57cec5SDimitry Andric Sema::ActOnSEHLeaveStmt(SourceLocation Loc, Scope *CurScope) {
46090b57cec5SDimitry Andric   Scope *SEHTryParent = CurScope;
46100b57cec5SDimitry Andric   while (SEHTryParent && !SEHTryParent->isSEHTryScope())
46110b57cec5SDimitry Andric     SEHTryParent = SEHTryParent->getParent();
46120b57cec5SDimitry Andric   if (!SEHTryParent)
46130b57cec5SDimitry Andric     return StmtError(Diag(Loc, diag::err_ms___leave_not_in___try));
46140b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, Loc, *SEHTryParent);
46150b57cec5SDimitry Andric 
46160b57cec5SDimitry Andric   return new (Context) SEHLeaveStmt(Loc);
46170b57cec5SDimitry Andric }
46180b57cec5SDimitry Andric 
46190b57cec5SDimitry Andric StmtResult Sema::BuildMSDependentExistsStmt(SourceLocation KeywordLoc,
46200b57cec5SDimitry Andric                                             bool IsIfExists,
46210b57cec5SDimitry Andric                                             NestedNameSpecifierLoc QualifierLoc,
46220b57cec5SDimitry Andric                                             DeclarationNameInfo NameInfo,
46230b57cec5SDimitry Andric                                             Stmt *Nested)
46240b57cec5SDimitry Andric {
46250b57cec5SDimitry Andric   return new (Context) MSDependentExistsStmt(KeywordLoc, IsIfExists,
46260b57cec5SDimitry Andric                                              QualifierLoc, NameInfo,
46270b57cec5SDimitry Andric                                              cast<CompoundStmt>(Nested));
46280b57cec5SDimitry Andric }
46290b57cec5SDimitry Andric 
46300b57cec5SDimitry Andric 
46310b57cec5SDimitry Andric StmtResult Sema::ActOnMSDependentExistsStmt(SourceLocation KeywordLoc,
46320b57cec5SDimitry Andric                                             bool IsIfExists,
46330b57cec5SDimitry Andric                                             CXXScopeSpec &SS,
46340b57cec5SDimitry Andric                                             UnqualifiedId &Name,
46350b57cec5SDimitry Andric                                             Stmt *Nested) {
46360b57cec5SDimitry Andric   return BuildMSDependentExistsStmt(KeywordLoc, IsIfExists,
46370b57cec5SDimitry Andric                                     SS.getWithLocInContext(Context),
46380b57cec5SDimitry Andric                                     GetNameFromUnqualifiedId(Name),
46390b57cec5SDimitry Andric                                     Nested);
46400b57cec5SDimitry Andric }
46410b57cec5SDimitry Andric 
46420b57cec5SDimitry Andric RecordDecl*
46430b57cec5SDimitry Andric Sema::CreateCapturedStmtRecordDecl(CapturedDecl *&CD, SourceLocation Loc,
46440b57cec5SDimitry Andric                                    unsigned NumParams) {
46450b57cec5SDimitry Andric   DeclContext *DC = CurContext;
46460b57cec5SDimitry Andric   while (!(DC->isFunctionOrMethod() || DC->isRecord() || DC->isFileContext()))
46470b57cec5SDimitry Andric     DC = DC->getParent();
46480b57cec5SDimitry Andric 
46490b57cec5SDimitry Andric   RecordDecl *RD = nullptr;
46500b57cec5SDimitry Andric   if (getLangOpts().CPlusPlus)
46510b57cec5SDimitry Andric     RD = CXXRecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc,
46520b57cec5SDimitry Andric                                /*Id=*/nullptr);
46530b57cec5SDimitry Andric   else
46540b57cec5SDimitry Andric     RD = RecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc, /*Id=*/nullptr);
46550b57cec5SDimitry Andric 
46560b57cec5SDimitry Andric   RD->setCapturedRecord();
46570b57cec5SDimitry Andric   DC->addDecl(RD);
46580b57cec5SDimitry Andric   RD->setImplicit();
46590b57cec5SDimitry Andric   RD->startDefinition();
46600b57cec5SDimitry Andric 
46610b57cec5SDimitry Andric   assert(NumParams > 0 && "CapturedStmt requires context parameter");
46620b57cec5SDimitry Andric   CD = CapturedDecl::Create(Context, CurContext, NumParams);
46630b57cec5SDimitry Andric   DC->addDecl(CD);
46640b57cec5SDimitry Andric   return RD;
46650b57cec5SDimitry Andric }
46660b57cec5SDimitry Andric 
46670b57cec5SDimitry Andric static bool
46680b57cec5SDimitry Andric buildCapturedStmtCaptureList(Sema &S, CapturedRegionScopeInfo *RSI,
46690b57cec5SDimitry Andric                              SmallVectorImpl<CapturedStmt::Capture> &Captures,
46700b57cec5SDimitry Andric                              SmallVectorImpl<Expr *> &CaptureInits) {
46710b57cec5SDimitry Andric   for (const sema::Capture &Cap : RSI->Captures) {
46720b57cec5SDimitry Andric     if (Cap.isInvalid())
46730b57cec5SDimitry Andric       continue;
46740b57cec5SDimitry Andric 
46750b57cec5SDimitry Andric     // Form the initializer for the capture.
46760b57cec5SDimitry Andric     ExprResult Init = S.BuildCaptureInit(Cap, Cap.getLocation(),
46770b57cec5SDimitry Andric                                          RSI->CapRegionKind == CR_OpenMP);
46780b57cec5SDimitry Andric 
46790b57cec5SDimitry Andric     // FIXME: Bail out now if the capture is not used and the initializer has
46800b57cec5SDimitry Andric     // no side-effects.
46810b57cec5SDimitry Andric 
46820b57cec5SDimitry Andric     // Create a field for this capture.
46830b57cec5SDimitry Andric     FieldDecl *Field = S.BuildCaptureField(RSI->TheRecordDecl, Cap);
46840b57cec5SDimitry Andric 
46850b57cec5SDimitry Andric     // Add the capture to our list of captures.
46860b57cec5SDimitry Andric     if (Cap.isThisCapture()) {
46870b57cec5SDimitry Andric       Captures.push_back(CapturedStmt::Capture(Cap.getLocation(),
46880b57cec5SDimitry Andric                                                CapturedStmt::VCK_This));
46890b57cec5SDimitry Andric     } else if (Cap.isVLATypeCapture()) {
46900b57cec5SDimitry Andric       Captures.push_back(
46910b57cec5SDimitry Andric           CapturedStmt::Capture(Cap.getLocation(), CapturedStmt::VCK_VLAType));
46920b57cec5SDimitry Andric     } else {
46930b57cec5SDimitry Andric       assert(Cap.isVariableCapture() && "unknown kind of capture");
46940b57cec5SDimitry Andric 
46950b57cec5SDimitry Andric       if (S.getLangOpts().OpenMP && RSI->CapRegionKind == CR_OpenMP)
46960b57cec5SDimitry Andric         S.setOpenMPCaptureKind(Field, Cap.getVariable(), RSI->OpenMPLevel);
46970b57cec5SDimitry Andric 
46980b57cec5SDimitry Andric       Captures.push_back(CapturedStmt::Capture(Cap.getLocation(),
46990b57cec5SDimitry Andric                                                Cap.isReferenceCapture()
47000b57cec5SDimitry Andric                                                    ? CapturedStmt::VCK_ByRef
47010b57cec5SDimitry Andric                                                    : CapturedStmt::VCK_ByCopy,
47020b57cec5SDimitry Andric                                                Cap.getVariable()));
47030b57cec5SDimitry Andric     }
47040b57cec5SDimitry Andric     CaptureInits.push_back(Init.get());
47050b57cec5SDimitry Andric   }
47060b57cec5SDimitry Andric   return false;
47070b57cec5SDimitry Andric }
47080b57cec5SDimitry Andric 
47090b57cec5SDimitry Andric void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
47100b57cec5SDimitry Andric                                     CapturedRegionKind Kind,
47110b57cec5SDimitry Andric                                     unsigned NumParams) {
47120b57cec5SDimitry Andric   CapturedDecl *CD = nullptr;
47130b57cec5SDimitry Andric   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, NumParams);
47140b57cec5SDimitry Andric 
47150b57cec5SDimitry Andric   // Build the context parameter
47160b57cec5SDimitry Andric   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
47170b57cec5SDimitry Andric   IdentifierInfo *ParamName = &Context.Idents.get("__context");
47180b57cec5SDimitry Andric   QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
47190b57cec5SDimitry Andric   auto *Param =
47200b57cec5SDimitry Andric       ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
47210b57cec5SDimitry Andric                                 ImplicitParamDecl::CapturedContext);
47220b57cec5SDimitry Andric   DC->addDecl(Param);
47230b57cec5SDimitry Andric 
47240b57cec5SDimitry Andric   CD->setContextParam(0, Param);
47250b57cec5SDimitry Andric 
47260b57cec5SDimitry Andric   // Enter the capturing scope for this captured region.
47270b57cec5SDimitry Andric   PushCapturedRegionScope(CurScope, CD, RD, Kind);
47280b57cec5SDimitry Andric 
47290b57cec5SDimitry Andric   if (CurScope)
47300b57cec5SDimitry Andric     PushDeclContext(CurScope, CD);
47310b57cec5SDimitry Andric   else
47320b57cec5SDimitry Andric     CurContext = CD;
47330b57cec5SDimitry Andric 
47340b57cec5SDimitry Andric   PushExpressionEvaluationContext(
47350b57cec5SDimitry Andric       ExpressionEvaluationContext::PotentiallyEvaluated);
47360b57cec5SDimitry Andric }
47370b57cec5SDimitry Andric 
47380b57cec5SDimitry Andric void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
47390b57cec5SDimitry Andric                                     CapturedRegionKind Kind,
4740a7dea167SDimitry Andric                                     ArrayRef<CapturedParamNameType> Params,
4741a7dea167SDimitry Andric                                     unsigned OpenMPCaptureLevel) {
47420b57cec5SDimitry Andric   CapturedDecl *CD = nullptr;
47430b57cec5SDimitry Andric   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, Params.size());
47440b57cec5SDimitry Andric 
47450b57cec5SDimitry Andric   // Build the context parameter
47460b57cec5SDimitry Andric   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
47470b57cec5SDimitry Andric   bool ContextIsFound = false;
47480b57cec5SDimitry Andric   unsigned ParamNum = 0;
47490b57cec5SDimitry Andric   for (ArrayRef<CapturedParamNameType>::iterator I = Params.begin(),
47500b57cec5SDimitry Andric                                                  E = Params.end();
47510b57cec5SDimitry Andric        I != E; ++I, ++ParamNum) {
47520b57cec5SDimitry Andric     if (I->second.isNull()) {
47530b57cec5SDimitry Andric       assert(!ContextIsFound &&
47540b57cec5SDimitry Andric              "null type has been found already for '__context' parameter");
47550b57cec5SDimitry Andric       IdentifierInfo *ParamName = &Context.Idents.get("__context");
47560b57cec5SDimitry Andric       QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD))
47570b57cec5SDimitry Andric                                .withConst()
47580b57cec5SDimitry Andric                                .withRestrict();
47590b57cec5SDimitry Andric       auto *Param =
47600b57cec5SDimitry Andric           ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
47610b57cec5SDimitry Andric                                     ImplicitParamDecl::CapturedContext);
47620b57cec5SDimitry Andric       DC->addDecl(Param);
47630b57cec5SDimitry Andric       CD->setContextParam(ParamNum, Param);
47640b57cec5SDimitry Andric       ContextIsFound = true;
47650b57cec5SDimitry Andric     } else {
47660b57cec5SDimitry Andric       IdentifierInfo *ParamName = &Context.Idents.get(I->first);
47670b57cec5SDimitry Andric       auto *Param =
47680b57cec5SDimitry Andric           ImplicitParamDecl::Create(Context, DC, Loc, ParamName, I->second,
47690b57cec5SDimitry Andric                                     ImplicitParamDecl::CapturedContext);
47700b57cec5SDimitry Andric       DC->addDecl(Param);
47710b57cec5SDimitry Andric       CD->setParam(ParamNum, Param);
47720b57cec5SDimitry Andric     }
47730b57cec5SDimitry Andric   }
47740b57cec5SDimitry Andric   assert(ContextIsFound && "no null type for '__context' parameter");
47750b57cec5SDimitry Andric   if (!ContextIsFound) {
47760b57cec5SDimitry Andric     // Add __context implicitly if it is not specified.
47770b57cec5SDimitry Andric     IdentifierInfo *ParamName = &Context.Idents.get("__context");
47780b57cec5SDimitry Andric     QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
47790b57cec5SDimitry Andric     auto *Param =
47800b57cec5SDimitry Andric         ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
47810b57cec5SDimitry Andric                                   ImplicitParamDecl::CapturedContext);
47820b57cec5SDimitry Andric     DC->addDecl(Param);
47830b57cec5SDimitry Andric     CD->setContextParam(ParamNum, Param);
47840b57cec5SDimitry Andric   }
47850b57cec5SDimitry Andric   // Enter the capturing scope for this captured region.
4786a7dea167SDimitry Andric   PushCapturedRegionScope(CurScope, CD, RD, Kind, OpenMPCaptureLevel);
47870b57cec5SDimitry Andric 
47880b57cec5SDimitry Andric   if (CurScope)
47890b57cec5SDimitry Andric     PushDeclContext(CurScope, CD);
47900b57cec5SDimitry Andric   else
47910b57cec5SDimitry Andric     CurContext = CD;
47920b57cec5SDimitry Andric 
47930b57cec5SDimitry Andric   PushExpressionEvaluationContext(
47940b57cec5SDimitry Andric       ExpressionEvaluationContext::PotentiallyEvaluated);
47950b57cec5SDimitry Andric }
47960b57cec5SDimitry Andric 
47970b57cec5SDimitry Andric void Sema::ActOnCapturedRegionError() {
47980b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
47990b57cec5SDimitry Andric   PopExpressionEvaluationContext();
48000b57cec5SDimitry Andric   PopDeclContext();
48010b57cec5SDimitry Andric   PoppedFunctionScopePtr ScopeRAII = PopFunctionScopeInfo();
48020b57cec5SDimitry Andric   CapturedRegionScopeInfo *RSI = cast<CapturedRegionScopeInfo>(ScopeRAII.get());
48030b57cec5SDimitry Andric 
48040b57cec5SDimitry Andric   RecordDecl *Record = RSI->TheRecordDecl;
48050b57cec5SDimitry Andric   Record->setInvalidDecl();
48060b57cec5SDimitry Andric 
48070b57cec5SDimitry Andric   SmallVector<Decl*, 4> Fields(Record->fields());
48080b57cec5SDimitry Andric   ActOnFields(/*Scope=*/nullptr, Record->getLocation(), Record, Fields,
48090b57cec5SDimitry Andric               SourceLocation(), SourceLocation(), ParsedAttributesView());
48100b57cec5SDimitry Andric }
48110b57cec5SDimitry Andric 
48120b57cec5SDimitry Andric StmtResult Sema::ActOnCapturedRegionEnd(Stmt *S) {
48130b57cec5SDimitry Andric   // Leave the captured scope before we start creating captures in the
48140b57cec5SDimitry Andric   // enclosing scope.
48150b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
48160b57cec5SDimitry Andric   PopExpressionEvaluationContext();
48170b57cec5SDimitry Andric   PopDeclContext();
48180b57cec5SDimitry Andric   PoppedFunctionScopePtr ScopeRAII = PopFunctionScopeInfo();
48190b57cec5SDimitry Andric   CapturedRegionScopeInfo *RSI = cast<CapturedRegionScopeInfo>(ScopeRAII.get());
48200b57cec5SDimitry Andric 
48210b57cec5SDimitry Andric   SmallVector<CapturedStmt::Capture, 4> Captures;
48220b57cec5SDimitry Andric   SmallVector<Expr *, 4> CaptureInits;
48230b57cec5SDimitry Andric   if (buildCapturedStmtCaptureList(*this, RSI, Captures, CaptureInits))
48240b57cec5SDimitry Andric     return StmtError();
48250b57cec5SDimitry Andric 
48260b57cec5SDimitry Andric   CapturedDecl *CD = RSI->TheCapturedDecl;
48270b57cec5SDimitry Andric   RecordDecl *RD = RSI->TheRecordDecl;
48280b57cec5SDimitry Andric 
48290b57cec5SDimitry Andric   CapturedStmt *Res = CapturedStmt::Create(
48300b57cec5SDimitry Andric       getASTContext(), S, static_cast<CapturedRegionKind>(RSI->CapRegionKind),
48310b57cec5SDimitry Andric       Captures, CaptureInits, CD, RD);
48320b57cec5SDimitry Andric 
48330b57cec5SDimitry Andric   CD->setBody(Res->getCapturedStmt());
48340b57cec5SDimitry Andric   RD->completeDefinition();
48350b57cec5SDimitry Andric 
48360b57cec5SDimitry Andric   return Res;
48370b57cec5SDimitry Andric }
4838