xref: /freebsd/contrib/llvm-project/clang/lib/Sema/SemaStmt.cpp (revision a7dea1671b87c07d2d266f836bfa8b58efc7c134)
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/Sema/Ownership.h"
140b57cec5SDimitry Andric #include "clang/Sema/SemaInternal.h"
150b57cec5SDimitry Andric #include "clang/AST/ASTContext.h"
160b57cec5SDimitry Andric #include "clang/AST/ASTDiagnostic.h"
170b57cec5SDimitry Andric #include "clang/AST/ASTLambda.h"
180b57cec5SDimitry Andric #include "clang/AST/CharUnits.h"
190b57cec5SDimitry Andric #include "clang/AST/CXXInheritance.h"
200b57cec5SDimitry Andric #include "clang/AST/DeclObjC.h"
210b57cec5SDimitry Andric #include "clang/AST/EvaluatedExprVisitor.h"
220b57cec5SDimitry Andric #include "clang/AST/ExprCXX.h"
230b57cec5SDimitry Andric #include "clang/AST/ExprObjC.h"
240b57cec5SDimitry Andric #include "clang/AST/RecursiveASTVisitor.h"
250b57cec5SDimitry Andric #include "clang/AST/StmtCXX.h"
260b57cec5SDimitry Andric #include "clang/AST/StmtObjC.h"
270b57cec5SDimitry Andric #include "clang/AST/TypeLoc.h"
280b57cec5SDimitry Andric #include "clang/AST/TypeOrdering.h"
290b57cec5SDimitry Andric #include "clang/Basic/TargetInfo.h"
300b57cec5SDimitry Andric #include "clang/Lex/Preprocessor.h"
310b57cec5SDimitry Andric #include "clang/Sema/Initialization.h"
320b57cec5SDimitry Andric #include "clang/Sema/Lookup.h"
330b57cec5SDimitry Andric #include "clang/Sema/Scope.h"
340b57cec5SDimitry Andric #include "clang/Sema/ScopeInfo.h"
350b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h"
360b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h"
370b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h"
380b57cec5SDimitry Andric #include "llvm/ADT/SmallPtrSet.h"
390b57cec5SDimitry Andric #include "llvm/ADT/SmallString.h"
400b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h"
410b57cec5SDimitry Andric 
420b57cec5SDimitry Andric using namespace clang;
430b57cec5SDimitry Andric using namespace sema;
440b57cec5SDimitry Andric 
450b57cec5SDimitry Andric StmtResult Sema::ActOnExprStmt(ExprResult FE, bool DiscardedValue) {
460b57cec5SDimitry Andric   if (FE.isInvalid())
470b57cec5SDimitry Andric     return StmtError();
480b57cec5SDimitry Andric 
490b57cec5SDimitry Andric   FE = ActOnFinishFullExpr(FE.get(), FE.get()->getExprLoc(), DiscardedValue);
500b57cec5SDimitry Andric   if (FE.isInvalid())
510b57cec5SDimitry Andric     return StmtError();
520b57cec5SDimitry Andric 
530b57cec5SDimitry Andric   // C99 6.8.3p2: The expression in an expression statement is evaluated as a
540b57cec5SDimitry Andric   // void expression for its side effects.  Conversion to void allows any
550b57cec5SDimitry Andric   // operand, even incomplete types.
560b57cec5SDimitry Andric 
570b57cec5SDimitry Andric   // Same thing in for stmt first clause (when expr) and third clause.
580b57cec5SDimitry Andric   return StmtResult(FE.getAs<Stmt>());
590b57cec5SDimitry Andric }
600b57cec5SDimitry Andric 
610b57cec5SDimitry Andric 
620b57cec5SDimitry Andric StmtResult Sema::ActOnExprStmtError() {
630b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
640b57cec5SDimitry Andric   return StmtError();
650b57cec5SDimitry Andric }
660b57cec5SDimitry Andric 
670b57cec5SDimitry Andric StmtResult Sema::ActOnNullStmt(SourceLocation SemiLoc,
680b57cec5SDimitry Andric                                bool HasLeadingEmptyMacro) {
690b57cec5SDimitry Andric   return new (Context) NullStmt(SemiLoc, HasLeadingEmptyMacro);
700b57cec5SDimitry Andric }
710b57cec5SDimitry Andric 
720b57cec5SDimitry Andric StmtResult Sema::ActOnDeclStmt(DeclGroupPtrTy dg, SourceLocation StartLoc,
730b57cec5SDimitry Andric                                SourceLocation EndLoc) {
740b57cec5SDimitry Andric   DeclGroupRef DG = dg.get();
750b57cec5SDimitry Andric 
760b57cec5SDimitry Andric   // If we have an invalid decl, just return an error.
770b57cec5SDimitry Andric   if (DG.isNull()) return StmtError();
780b57cec5SDimitry Andric 
790b57cec5SDimitry Andric   return new (Context) DeclStmt(DG, StartLoc, EndLoc);
800b57cec5SDimitry Andric }
810b57cec5SDimitry Andric 
820b57cec5SDimitry Andric void Sema::ActOnForEachDeclStmt(DeclGroupPtrTy dg) {
830b57cec5SDimitry Andric   DeclGroupRef DG = dg.get();
840b57cec5SDimitry Andric 
850b57cec5SDimitry Andric   // If we don't have a declaration, or we have an invalid declaration,
860b57cec5SDimitry Andric   // just return.
870b57cec5SDimitry Andric   if (DG.isNull() || !DG.isSingleDecl())
880b57cec5SDimitry Andric     return;
890b57cec5SDimitry Andric 
900b57cec5SDimitry Andric   Decl *decl = DG.getSingleDecl();
910b57cec5SDimitry Andric   if (!decl || decl->isInvalidDecl())
920b57cec5SDimitry Andric     return;
930b57cec5SDimitry Andric 
940b57cec5SDimitry Andric   // Only variable declarations are permitted.
950b57cec5SDimitry Andric   VarDecl *var = dyn_cast<VarDecl>(decl);
960b57cec5SDimitry Andric   if (!var) {
970b57cec5SDimitry Andric     Diag(decl->getLocation(), diag::err_non_variable_decl_in_for);
980b57cec5SDimitry Andric     decl->setInvalidDecl();
990b57cec5SDimitry Andric     return;
1000b57cec5SDimitry Andric   }
1010b57cec5SDimitry Andric 
1020b57cec5SDimitry Andric   // foreach variables are never actually initialized in the way that
1030b57cec5SDimitry Andric   // the parser came up with.
1040b57cec5SDimitry Andric   var->setInit(nullptr);
1050b57cec5SDimitry Andric 
1060b57cec5SDimitry Andric   // In ARC, we don't need to retain the iteration variable of a fast
1070b57cec5SDimitry Andric   // enumeration loop.  Rather than actually trying to catch that
1080b57cec5SDimitry Andric   // during declaration processing, we remove the consequences here.
1090b57cec5SDimitry Andric   if (getLangOpts().ObjCAutoRefCount) {
1100b57cec5SDimitry Andric     QualType type = var->getType();
1110b57cec5SDimitry Andric 
1120b57cec5SDimitry Andric     // Only do this if we inferred the lifetime.  Inferred lifetime
1130b57cec5SDimitry Andric     // will show up as a local qualifier because explicit lifetime
1140b57cec5SDimitry Andric     // should have shown up as an AttributedType instead.
1150b57cec5SDimitry Andric     if (type.getLocalQualifiers().getObjCLifetime() == Qualifiers::OCL_Strong) {
1160b57cec5SDimitry Andric       // Add 'const' and mark the variable as pseudo-strong.
1170b57cec5SDimitry Andric       var->setType(type.withConst());
1180b57cec5SDimitry Andric       var->setARCPseudoStrong(true);
1190b57cec5SDimitry Andric     }
1200b57cec5SDimitry Andric   }
1210b57cec5SDimitry Andric }
1220b57cec5SDimitry Andric 
1230b57cec5SDimitry Andric /// Diagnose unused comparisons, both builtin and overloaded operators.
1240b57cec5SDimitry Andric /// For '==' and '!=', suggest fixits for '=' or '|='.
1250b57cec5SDimitry Andric ///
1260b57cec5SDimitry Andric /// Adding a cast to void (or other expression wrappers) will prevent the
1270b57cec5SDimitry Andric /// warning from firing.
1280b57cec5SDimitry Andric static bool DiagnoseUnusedComparison(Sema &S, const Expr *E) {
1290b57cec5SDimitry Andric   SourceLocation Loc;
1300b57cec5SDimitry Andric   bool CanAssign;
1310b57cec5SDimitry Andric   enum { Equality, Inequality, Relational, ThreeWay } Kind;
1320b57cec5SDimitry Andric 
1330b57cec5SDimitry Andric   if (const BinaryOperator *Op = dyn_cast<BinaryOperator>(E)) {
1340b57cec5SDimitry Andric     if (!Op->isComparisonOp())
1350b57cec5SDimitry Andric       return false;
1360b57cec5SDimitry Andric 
1370b57cec5SDimitry Andric     if (Op->getOpcode() == BO_EQ)
1380b57cec5SDimitry Andric       Kind = Equality;
1390b57cec5SDimitry Andric     else if (Op->getOpcode() == BO_NE)
1400b57cec5SDimitry Andric       Kind = Inequality;
1410b57cec5SDimitry Andric     else if (Op->getOpcode() == BO_Cmp)
1420b57cec5SDimitry Andric       Kind = ThreeWay;
1430b57cec5SDimitry Andric     else {
1440b57cec5SDimitry Andric       assert(Op->isRelationalOp());
1450b57cec5SDimitry Andric       Kind = Relational;
1460b57cec5SDimitry Andric     }
1470b57cec5SDimitry Andric     Loc = Op->getOperatorLoc();
1480b57cec5SDimitry Andric     CanAssign = Op->getLHS()->IgnoreParenImpCasts()->isLValue();
1490b57cec5SDimitry Andric   } else if (const CXXOperatorCallExpr *Op = dyn_cast<CXXOperatorCallExpr>(E)) {
1500b57cec5SDimitry Andric     switch (Op->getOperator()) {
1510b57cec5SDimitry Andric     case OO_EqualEqual:
1520b57cec5SDimitry Andric       Kind = Equality;
1530b57cec5SDimitry Andric       break;
1540b57cec5SDimitry Andric     case OO_ExclaimEqual:
1550b57cec5SDimitry Andric       Kind = Inequality;
1560b57cec5SDimitry Andric       break;
1570b57cec5SDimitry Andric     case OO_Less:
1580b57cec5SDimitry Andric     case OO_Greater:
1590b57cec5SDimitry Andric     case OO_GreaterEqual:
1600b57cec5SDimitry Andric     case OO_LessEqual:
1610b57cec5SDimitry Andric       Kind = Relational;
1620b57cec5SDimitry Andric       break;
1630b57cec5SDimitry Andric     case OO_Spaceship:
1640b57cec5SDimitry Andric       Kind = ThreeWay;
1650b57cec5SDimitry Andric       break;
1660b57cec5SDimitry Andric     default:
1670b57cec5SDimitry Andric       return false;
1680b57cec5SDimitry Andric     }
1690b57cec5SDimitry Andric 
1700b57cec5SDimitry Andric     Loc = Op->getOperatorLoc();
1710b57cec5SDimitry Andric     CanAssign = Op->getArg(0)->IgnoreParenImpCasts()->isLValue();
1720b57cec5SDimitry Andric   } else {
1730b57cec5SDimitry Andric     // Not a typo-prone comparison.
1740b57cec5SDimitry Andric     return false;
1750b57cec5SDimitry Andric   }
1760b57cec5SDimitry Andric 
1770b57cec5SDimitry Andric   // Suppress warnings when the operator, suspicious as it may be, comes from
1780b57cec5SDimitry Andric   // a macro expansion.
1790b57cec5SDimitry Andric   if (S.SourceMgr.isMacroBodyExpansion(Loc))
1800b57cec5SDimitry Andric     return false;
1810b57cec5SDimitry Andric 
1820b57cec5SDimitry Andric   S.Diag(Loc, diag::warn_unused_comparison)
1830b57cec5SDimitry Andric     << (unsigned)Kind << E->getSourceRange();
1840b57cec5SDimitry Andric 
1850b57cec5SDimitry Andric   // If the LHS is a plausible entity to assign to, provide a fixit hint to
1860b57cec5SDimitry Andric   // correct common typos.
1870b57cec5SDimitry Andric   if (CanAssign) {
1880b57cec5SDimitry Andric     if (Kind == Inequality)
1890b57cec5SDimitry Andric       S.Diag(Loc, diag::note_inequality_comparison_to_or_assign)
1900b57cec5SDimitry Andric         << FixItHint::CreateReplacement(Loc, "|=");
1910b57cec5SDimitry Andric     else if (Kind == Equality)
1920b57cec5SDimitry Andric       S.Diag(Loc, diag::note_equality_comparison_to_assign)
1930b57cec5SDimitry Andric         << FixItHint::CreateReplacement(Loc, "=");
1940b57cec5SDimitry Andric   }
1950b57cec5SDimitry Andric 
1960b57cec5SDimitry Andric   return true;
1970b57cec5SDimitry Andric }
1980b57cec5SDimitry Andric 
199*a7dea167SDimitry Andric static bool DiagnoseNoDiscard(Sema &S, const WarnUnusedResultAttr *A,
200*a7dea167SDimitry Andric                               SourceLocation Loc, SourceRange R1,
201*a7dea167SDimitry Andric                               SourceRange R2, bool IsCtor) {
202*a7dea167SDimitry Andric   if (!A)
203*a7dea167SDimitry Andric     return false;
204*a7dea167SDimitry Andric   StringRef Msg = A->getMessage();
205*a7dea167SDimitry Andric 
206*a7dea167SDimitry Andric   if (Msg.empty()) {
207*a7dea167SDimitry Andric     if (IsCtor)
208*a7dea167SDimitry Andric       return S.Diag(Loc, diag::warn_unused_constructor) << A << R1 << R2;
209*a7dea167SDimitry Andric     return S.Diag(Loc, diag::warn_unused_result) << A << R1 << R2;
210*a7dea167SDimitry Andric   }
211*a7dea167SDimitry Andric 
212*a7dea167SDimitry Andric   if (IsCtor)
213*a7dea167SDimitry Andric     return S.Diag(Loc, diag::warn_unused_constructor_msg) << A << Msg << R1
214*a7dea167SDimitry Andric                                                           << R2;
215*a7dea167SDimitry Andric   return S.Diag(Loc, diag::warn_unused_result_msg) << A << Msg << R1 << R2;
216*a7dea167SDimitry Andric }
217*a7dea167SDimitry Andric 
2180b57cec5SDimitry Andric void Sema::DiagnoseUnusedExprResult(const Stmt *S) {
2190b57cec5SDimitry Andric   if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
2200b57cec5SDimitry Andric     return DiagnoseUnusedExprResult(Label->getSubStmt());
2210b57cec5SDimitry Andric 
2220b57cec5SDimitry Andric   const Expr *E = dyn_cast_or_null<Expr>(S);
2230b57cec5SDimitry Andric   if (!E)
2240b57cec5SDimitry Andric     return;
2250b57cec5SDimitry Andric 
2260b57cec5SDimitry Andric   // If we are in an unevaluated expression context, then there can be no unused
2270b57cec5SDimitry Andric   // results because the results aren't expected to be used in the first place.
2280b57cec5SDimitry Andric   if (isUnevaluatedContext())
2290b57cec5SDimitry Andric     return;
2300b57cec5SDimitry Andric 
2310b57cec5SDimitry Andric   SourceLocation ExprLoc = E->IgnoreParenImpCasts()->getExprLoc();
2320b57cec5SDimitry Andric   // In most cases, we don't want to warn if the expression is written in a
2330b57cec5SDimitry Andric   // macro body, or if the macro comes from a system header. If the offending
2340b57cec5SDimitry Andric   // expression is a call to a function with the warn_unused_result attribute,
2350b57cec5SDimitry Andric   // we warn no matter the location. Because of the order in which the various
2360b57cec5SDimitry Andric   // checks need to happen, we factor out the macro-related test here.
2370b57cec5SDimitry Andric   bool ShouldSuppress =
2380b57cec5SDimitry Andric       SourceMgr.isMacroBodyExpansion(ExprLoc) ||
2390b57cec5SDimitry Andric       SourceMgr.isInSystemMacro(ExprLoc);
2400b57cec5SDimitry Andric 
2410b57cec5SDimitry Andric   const Expr *WarnExpr;
2420b57cec5SDimitry Andric   SourceLocation Loc;
2430b57cec5SDimitry Andric   SourceRange R1, R2;
2440b57cec5SDimitry Andric   if (!E->isUnusedResultAWarning(WarnExpr, Loc, R1, R2, Context))
2450b57cec5SDimitry Andric     return;
2460b57cec5SDimitry Andric 
2470b57cec5SDimitry Andric   // If this is a GNU statement expression expanded from a macro, it is probably
2480b57cec5SDimitry Andric   // unused because it is a function-like macro that can be used as either an
2490b57cec5SDimitry Andric   // expression or statement.  Don't warn, because it is almost certainly a
2500b57cec5SDimitry Andric   // false positive.
2510b57cec5SDimitry Andric   if (isa<StmtExpr>(E) && Loc.isMacroID())
2520b57cec5SDimitry Andric     return;
2530b57cec5SDimitry Andric 
2540b57cec5SDimitry Andric   // Check if this is the UNREFERENCED_PARAMETER from the Microsoft headers.
2550b57cec5SDimitry Andric   // That macro is frequently used to suppress "unused parameter" warnings,
2560b57cec5SDimitry Andric   // but its implementation makes clang's -Wunused-value fire.  Prevent this.
2570b57cec5SDimitry Andric   if (isa<ParenExpr>(E->IgnoreImpCasts()) && Loc.isMacroID()) {
2580b57cec5SDimitry Andric     SourceLocation SpellLoc = Loc;
2590b57cec5SDimitry Andric     if (findMacroSpelling(SpellLoc, "UNREFERENCED_PARAMETER"))
2600b57cec5SDimitry Andric       return;
2610b57cec5SDimitry Andric   }
2620b57cec5SDimitry Andric 
2630b57cec5SDimitry Andric   // Okay, we have an unused result.  Depending on what the base expression is,
2640b57cec5SDimitry Andric   // we might want to make a more specific diagnostic.  Check for one of these
2650b57cec5SDimitry Andric   // cases now.
2660b57cec5SDimitry Andric   unsigned DiagID = diag::warn_unused_expr;
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;
276*a7dea167SDimitry Andric   if (const auto *Cast = dyn_cast<CastExpr>(E))
277*a7dea167SDimitry Andric     if (Cast->getCastKind() == CK_NoOp ||
278*a7dea167SDimitry Andric         Cast->getCastKind() == CK_ConstructorConversion)
279*a7dea167SDimitry Andric       E = Cast->getSubExpr()->IgnoreImpCasts();
280*a7dea167SDimitry Andric 
2810b57cec5SDimitry Andric   if (const CallExpr *CE = dyn_cast<CallExpr>(E)) {
2820b57cec5SDimitry Andric     if (E->getType()->isVoidType())
2830b57cec5SDimitry Andric       return;
2840b57cec5SDimitry Andric 
285*a7dea167SDimitry Andric     if (DiagnoseNoDiscard(*this, cast_or_null<WarnUnusedResultAttr>(
286*a7dea167SDimitry Andric                                      CE->getUnusedResultAttr(Context)),
287*a7dea167SDimitry 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     }
306*a7dea167SDimitry Andric   } else if (const auto *CE = dyn_cast<CXXConstructExpr>(E)) {
307*a7dea167SDimitry Andric     if (const CXXConstructorDecl *Ctor = CE->getConstructor()) {
308*a7dea167SDimitry Andric       const auto *A = Ctor->getAttr<WarnUnusedResultAttr>();
309*a7dea167SDimitry Andric       A = A ? A : Ctor->getParent()->getAttr<WarnUnusedResultAttr>();
310*a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, A, Loc, R1, R2, /*isCtor=*/true))
311*a7dea167SDimitry Andric         return;
312*a7dea167SDimitry Andric     }
313*a7dea167SDimitry Andric   } else if (const auto *ILE = dyn_cast<InitListExpr>(E)) {
314*a7dea167SDimitry Andric     if (const TagDecl *TD = ILE->getType()->getAsTagDecl()) {
315*a7dea167SDimitry Andric 
316*a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, TD->getAttr<WarnUnusedResultAttr>(), Loc, R1,
317*a7dea167SDimitry Andric                             R2, /*isCtor=*/false))
318*a7dea167SDimitry Andric         return;
319*a7dea167SDimitry Andric     }
3200b57cec5SDimitry Andric   } else if (ShouldSuppress)
3210b57cec5SDimitry Andric     return;
3220b57cec5SDimitry Andric 
323*a7dea167SDimitry 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) {
331*a7dea167SDimitry Andric       if (DiagnoseNoDiscard(*this, MD->getAttr<WarnUnusedResultAttr>(), Loc, R1,
332*a7dea167SDimitry 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();
3370b57cec5SDimitry Andric     if (isa<ObjCSubscriptRefExpr>(Source))
3380b57cec5SDimitry Andric       DiagID = diag::warn_unused_container_subscript_expr;
3390b57cec5SDimitry Andric     else
3400b57cec5SDimitry Andric       DiagID = diag::warn_unused_property_expr;
3410b57cec5SDimitry Andric   } else if (const CXXFunctionalCastExpr *FC
3420b57cec5SDimitry Andric                                        = dyn_cast<CXXFunctionalCastExpr>(E)) {
3430b57cec5SDimitry Andric     const Expr *E = FC->getSubExpr();
3440b57cec5SDimitry Andric     if (const CXXBindTemporaryExpr *TE = dyn_cast<CXXBindTemporaryExpr>(E))
3450b57cec5SDimitry Andric       E = TE->getSubExpr();
3460b57cec5SDimitry Andric     if (isa<CXXTemporaryObjectExpr>(E))
3470b57cec5SDimitry Andric       return;
3480b57cec5SDimitry Andric     if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
3490b57cec5SDimitry Andric       if (const CXXRecordDecl *RD = CE->getType()->getAsCXXRecordDecl())
3500b57cec5SDimitry Andric         if (!RD->getAttr<WarnUnusedAttr>())
3510b57cec5SDimitry Andric           return;
3520b57cec5SDimitry Andric   }
3530b57cec5SDimitry Andric   // Diagnose "(void*) blah" as a typo for "(void) blah".
3540b57cec5SDimitry Andric   else if (const CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(E)) {
3550b57cec5SDimitry Andric     TypeSourceInfo *TI = CE->getTypeInfoAsWritten();
3560b57cec5SDimitry Andric     QualType T = TI->getType();
3570b57cec5SDimitry Andric 
3580b57cec5SDimitry Andric     // We really do want to use the non-canonical type here.
3590b57cec5SDimitry Andric     if (T == Context.VoidPtrTy) {
3600b57cec5SDimitry Andric       PointerTypeLoc TL = TI->getTypeLoc().castAs<PointerTypeLoc>();
3610b57cec5SDimitry Andric 
3620b57cec5SDimitry Andric       Diag(Loc, diag::warn_unused_voidptr)
3630b57cec5SDimitry Andric         << FixItHint::CreateRemoval(TL.getStarLoc());
3640b57cec5SDimitry Andric       return;
3650b57cec5SDimitry Andric     }
3660b57cec5SDimitry Andric   }
3670b57cec5SDimitry Andric 
3680b57cec5SDimitry Andric   if (E->isGLValue() && E->getType().isVolatileQualified()) {
3690b57cec5SDimitry Andric     Diag(Loc, diag::warn_unused_volatile) << R1 << R2;
3700b57cec5SDimitry Andric     return;
3710b57cec5SDimitry Andric   }
3720b57cec5SDimitry Andric 
3730b57cec5SDimitry Andric   DiagRuntimeBehavior(Loc, nullptr, PDiag(DiagID) << R1 << R2);
3740b57cec5SDimitry Andric }
3750b57cec5SDimitry Andric 
3760b57cec5SDimitry Andric void Sema::ActOnStartOfCompoundStmt(bool IsStmtExpr) {
3770b57cec5SDimitry Andric   PushCompoundScope(IsStmtExpr);
3780b57cec5SDimitry Andric }
3790b57cec5SDimitry Andric 
3800b57cec5SDimitry Andric void Sema::ActOnFinishOfCompoundStmt() {
3810b57cec5SDimitry Andric   PopCompoundScope();
3820b57cec5SDimitry Andric }
3830b57cec5SDimitry Andric 
3840b57cec5SDimitry Andric sema::CompoundScopeInfo &Sema::getCurCompoundScope() const {
3850b57cec5SDimitry Andric   return getCurFunction()->CompoundScopes.back();
3860b57cec5SDimitry Andric }
3870b57cec5SDimitry Andric 
3880b57cec5SDimitry Andric StmtResult Sema::ActOnCompoundStmt(SourceLocation L, SourceLocation R,
3890b57cec5SDimitry Andric                                    ArrayRef<Stmt *> Elts, bool isStmtExpr) {
3900b57cec5SDimitry Andric   const unsigned NumElts = Elts.size();
3910b57cec5SDimitry Andric 
3920b57cec5SDimitry Andric   // If we're in C89 mode, check that we don't have any decls after stmts.  If
3930b57cec5SDimitry Andric   // so, emit an extension diagnostic.
3940b57cec5SDimitry Andric   if (!getLangOpts().C99 && !getLangOpts().CPlusPlus) {
3950b57cec5SDimitry Andric     // Note that __extension__ can be around a decl.
3960b57cec5SDimitry Andric     unsigned i = 0;
3970b57cec5SDimitry Andric     // Skip over all declarations.
3980b57cec5SDimitry Andric     for (; i != NumElts && isa<DeclStmt>(Elts[i]); ++i)
3990b57cec5SDimitry Andric       /*empty*/;
4000b57cec5SDimitry Andric 
4010b57cec5SDimitry Andric     // We found the end of the list or a statement.  Scan for another declstmt.
4020b57cec5SDimitry Andric     for (; i != NumElts && !isa<DeclStmt>(Elts[i]); ++i)
4030b57cec5SDimitry Andric       /*empty*/;
4040b57cec5SDimitry Andric 
4050b57cec5SDimitry Andric     if (i != NumElts) {
4060b57cec5SDimitry Andric       Decl *D = *cast<DeclStmt>(Elts[i])->decl_begin();
4070b57cec5SDimitry Andric       Diag(D->getLocation(), diag::ext_mixed_decls_code);
4080b57cec5SDimitry Andric     }
4090b57cec5SDimitry Andric   }
4100b57cec5SDimitry Andric 
4110b57cec5SDimitry Andric   // Check for suspicious empty body (null statement) in `for' and `while'
4120b57cec5SDimitry Andric   // statements.  Don't do anything for template instantiations, this just adds
4130b57cec5SDimitry Andric   // noise.
4140b57cec5SDimitry Andric   if (NumElts != 0 && !CurrentInstantiationScope &&
4150b57cec5SDimitry Andric       getCurCompoundScope().HasEmptyLoopBodies) {
4160b57cec5SDimitry Andric     for (unsigned i = 0; i != NumElts - 1; ++i)
4170b57cec5SDimitry Andric       DiagnoseEmptyLoopBody(Elts[i], Elts[i + 1]);
4180b57cec5SDimitry Andric   }
4190b57cec5SDimitry Andric 
4200b57cec5SDimitry Andric   return CompoundStmt::Create(Context, Elts, L, R);
4210b57cec5SDimitry Andric }
4220b57cec5SDimitry Andric 
4230b57cec5SDimitry Andric ExprResult
4240b57cec5SDimitry Andric Sema::ActOnCaseExpr(SourceLocation CaseLoc, ExprResult Val) {
4250b57cec5SDimitry Andric   if (!Val.get())
4260b57cec5SDimitry Andric     return Val;
4270b57cec5SDimitry Andric 
4280b57cec5SDimitry Andric   if (DiagnoseUnexpandedParameterPack(Val.get()))
4290b57cec5SDimitry Andric     return ExprError();
4300b57cec5SDimitry Andric 
4310b57cec5SDimitry Andric   // If we're not inside a switch, let the 'case' statement handling diagnose
4320b57cec5SDimitry Andric   // this. Just clean up after the expression as best we can.
433*a7dea167SDimitry Andric   if (getCurFunction()->SwitchStack.empty())
434*a7dea167SDimitry Andric     return ActOnFinishFullExpr(Val.get(), Val.get()->getExprLoc(), false,
435*a7dea167SDimitry Andric                                getLangOpts().CPlusPlus11);
436*a7dea167SDimitry Andric 
4370b57cec5SDimitry Andric   Expr *CondExpr =
4380b57cec5SDimitry Andric       getCurFunction()->SwitchStack.back().getPointer()->getCond();
4390b57cec5SDimitry Andric   if (!CondExpr)
4400b57cec5SDimitry Andric     return ExprError();
4410b57cec5SDimitry Andric   QualType CondType = CondExpr->getType();
4420b57cec5SDimitry Andric 
4430b57cec5SDimitry Andric   auto CheckAndFinish = [&](Expr *E) {
4440b57cec5SDimitry Andric     if (CondType->isDependentType() || E->isTypeDependent())
4450b57cec5SDimitry Andric       return ExprResult(E);
4460b57cec5SDimitry Andric 
4470b57cec5SDimitry Andric     if (getLangOpts().CPlusPlus11) {
4480b57cec5SDimitry Andric       // C++11 [stmt.switch]p2: the constant-expression shall be a converted
4490b57cec5SDimitry Andric       // constant expression of the promoted type of the switch condition.
4500b57cec5SDimitry Andric       llvm::APSInt TempVal;
4510b57cec5SDimitry Andric       return CheckConvertedConstantExpression(E, CondType, TempVal,
4520b57cec5SDimitry Andric                                               CCEK_CaseValue);
4530b57cec5SDimitry Andric     }
4540b57cec5SDimitry Andric 
4550b57cec5SDimitry Andric     ExprResult ER = E;
4560b57cec5SDimitry Andric     if (!E->isValueDependent())
4570b57cec5SDimitry Andric       ER = VerifyIntegerConstantExpression(E);
4580b57cec5SDimitry Andric     if (!ER.isInvalid())
4590b57cec5SDimitry Andric       ER = DefaultLvalueConversion(ER.get());
4600b57cec5SDimitry Andric     if (!ER.isInvalid())
4610b57cec5SDimitry Andric       ER = ImpCastExprToType(ER.get(), CondType, CK_IntegralCast);
462*a7dea167SDimitry Andric     if (!ER.isInvalid())
463*a7dea167SDimitry Andric       ER = ActOnFinishFullExpr(ER.get(), ER.get()->getExprLoc(), false);
4640b57cec5SDimitry Andric     return ER;
4650b57cec5SDimitry Andric   };
4660b57cec5SDimitry Andric 
4670b57cec5SDimitry Andric   ExprResult Converted = CorrectDelayedTyposInExpr(Val, CheckAndFinish);
4680b57cec5SDimitry Andric   if (Converted.get() == Val.get())
4690b57cec5SDimitry Andric     Converted = CheckAndFinish(Val.get());
470*a7dea167SDimitry Andric   return Converted;
4710b57cec5SDimitry Andric }
4720b57cec5SDimitry Andric 
4730b57cec5SDimitry Andric StmtResult
4740b57cec5SDimitry Andric Sema::ActOnCaseStmt(SourceLocation CaseLoc, ExprResult LHSVal,
4750b57cec5SDimitry Andric                     SourceLocation DotDotDotLoc, ExprResult RHSVal,
4760b57cec5SDimitry Andric                     SourceLocation ColonLoc) {
4770b57cec5SDimitry Andric   assert((LHSVal.isInvalid() || LHSVal.get()) && "missing LHS value");
4780b57cec5SDimitry Andric   assert((DotDotDotLoc.isInvalid() ? RHSVal.isUnset()
4790b57cec5SDimitry Andric                                    : RHSVal.isInvalid() || RHSVal.get()) &&
4800b57cec5SDimitry Andric          "missing RHS value");
4810b57cec5SDimitry Andric 
4820b57cec5SDimitry Andric   if (getCurFunction()->SwitchStack.empty()) {
4830b57cec5SDimitry Andric     Diag(CaseLoc, diag::err_case_not_in_switch);
4840b57cec5SDimitry Andric     return StmtError();
4850b57cec5SDimitry Andric   }
4860b57cec5SDimitry Andric 
4870b57cec5SDimitry Andric   if (LHSVal.isInvalid() || RHSVal.isInvalid()) {
4880b57cec5SDimitry Andric     getCurFunction()->SwitchStack.back().setInt(true);
4890b57cec5SDimitry Andric     return StmtError();
4900b57cec5SDimitry Andric   }
4910b57cec5SDimitry Andric 
4920b57cec5SDimitry Andric   auto *CS = CaseStmt::Create(Context, LHSVal.get(), RHSVal.get(),
4930b57cec5SDimitry Andric                               CaseLoc, DotDotDotLoc, ColonLoc);
4940b57cec5SDimitry Andric   getCurFunction()->SwitchStack.back().getPointer()->addSwitchCase(CS);
4950b57cec5SDimitry Andric   return CS;
4960b57cec5SDimitry Andric }
4970b57cec5SDimitry Andric 
4980b57cec5SDimitry Andric /// ActOnCaseStmtBody - This installs a statement as the body of a case.
4990b57cec5SDimitry Andric void Sema::ActOnCaseStmtBody(Stmt *S, Stmt *SubStmt) {
5000b57cec5SDimitry Andric   cast<CaseStmt>(S)->setSubStmt(SubStmt);
5010b57cec5SDimitry Andric }
5020b57cec5SDimitry Andric 
5030b57cec5SDimitry Andric StmtResult
5040b57cec5SDimitry Andric Sema::ActOnDefaultStmt(SourceLocation DefaultLoc, SourceLocation ColonLoc,
5050b57cec5SDimitry Andric                        Stmt *SubStmt, Scope *CurScope) {
5060b57cec5SDimitry Andric   if (getCurFunction()->SwitchStack.empty()) {
5070b57cec5SDimitry Andric     Diag(DefaultLoc, diag::err_default_not_in_switch);
5080b57cec5SDimitry Andric     return SubStmt;
5090b57cec5SDimitry Andric   }
5100b57cec5SDimitry Andric 
5110b57cec5SDimitry Andric   DefaultStmt *DS = new (Context) DefaultStmt(DefaultLoc, ColonLoc, SubStmt);
5120b57cec5SDimitry Andric   getCurFunction()->SwitchStack.back().getPointer()->addSwitchCase(DS);
5130b57cec5SDimitry Andric   return DS;
5140b57cec5SDimitry Andric }
5150b57cec5SDimitry Andric 
5160b57cec5SDimitry Andric StmtResult
5170b57cec5SDimitry Andric Sema::ActOnLabelStmt(SourceLocation IdentLoc, LabelDecl *TheDecl,
5180b57cec5SDimitry Andric                      SourceLocation ColonLoc, Stmt *SubStmt) {
5190b57cec5SDimitry Andric   // If the label was multiply defined, reject it now.
5200b57cec5SDimitry Andric   if (TheDecl->getStmt()) {
5210b57cec5SDimitry Andric     Diag(IdentLoc, diag::err_redefinition_of_label) << TheDecl->getDeclName();
5220b57cec5SDimitry Andric     Diag(TheDecl->getLocation(), diag::note_previous_definition);
5230b57cec5SDimitry Andric     return SubStmt;
5240b57cec5SDimitry Andric   }
5250b57cec5SDimitry Andric 
5260b57cec5SDimitry Andric   // Otherwise, things are good.  Fill in the declaration and return it.
5270b57cec5SDimitry Andric   LabelStmt *LS = new (Context) LabelStmt(IdentLoc, TheDecl, SubStmt);
5280b57cec5SDimitry Andric   TheDecl->setStmt(LS);
5290b57cec5SDimitry Andric   if (!TheDecl->isGnuLocal()) {
5300b57cec5SDimitry Andric     TheDecl->setLocStart(IdentLoc);
5310b57cec5SDimitry Andric     if (!TheDecl->isMSAsmLabel()) {
5320b57cec5SDimitry Andric       // Don't update the location of MS ASM labels.  These will result in
5330b57cec5SDimitry Andric       // a diagnostic, and changing the location here will mess that up.
5340b57cec5SDimitry Andric       TheDecl->setLocation(IdentLoc);
5350b57cec5SDimitry Andric     }
5360b57cec5SDimitry Andric   }
5370b57cec5SDimitry Andric   return LS;
5380b57cec5SDimitry Andric }
5390b57cec5SDimitry Andric 
5400b57cec5SDimitry Andric StmtResult Sema::ActOnAttributedStmt(SourceLocation AttrLoc,
5410b57cec5SDimitry Andric                                      ArrayRef<const Attr*> Attrs,
5420b57cec5SDimitry Andric                                      Stmt *SubStmt) {
5430b57cec5SDimitry Andric   // Fill in the declaration and return it.
5440b57cec5SDimitry Andric   AttributedStmt *LS = AttributedStmt::Create(Context, AttrLoc, Attrs, SubStmt);
5450b57cec5SDimitry Andric   return LS;
5460b57cec5SDimitry Andric }
5470b57cec5SDimitry Andric 
5480b57cec5SDimitry Andric namespace {
5490b57cec5SDimitry Andric class CommaVisitor : public EvaluatedExprVisitor<CommaVisitor> {
5500b57cec5SDimitry Andric   typedef EvaluatedExprVisitor<CommaVisitor> Inherited;
5510b57cec5SDimitry Andric   Sema &SemaRef;
5520b57cec5SDimitry Andric public:
5530b57cec5SDimitry Andric   CommaVisitor(Sema &SemaRef) : Inherited(SemaRef.Context), SemaRef(SemaRef) {}
5540b57cec5SDimitry Andric   void VisitBinaryOperator(BinaryOperator *E) {
5550b57cec5SDimitry Andric     if (E->getOpcode() == BO_Comma)
5560b57cec5SDimitry Andric       SemaRef.DiagnoseCommaOperator(E->getLHS(), E->getExprLoc());
5570b57cec5SDimitry Andric     EvaluatedExprVisitor<CommaVisitor>::VisitBinaryOperator(E);
5580b57cec5SDimitry Andric   }
5590b57cec5SDimitry Andric };
5600b57cec5SDimitry Andric }
5610b57cec5SDimitry Andric 
5620b57cec5SDimitry Andric StmtResult
5630b57cec5SDimitry Andric Sema::ActOnIfStmt(SourceLocation IfLoc, bool IsConstexpr, Stmt *InitStmt,
5640b57cec5SDimitry Andric                   ConditionResult Cond,
5650b57cec5SDimitry Andric                   Stmt *thenStmt, SourceLocation ElseLoc,
5660b57cec5SDimitry Andric                   Stmt *elseStmt) {
5670b57cec5SDimitry Andric   if (Cond.isInvalid())
5680b57cec5SDimitry Andric     Cond = ConditionResult(
5690b57cec5SDimitry Andric         *this, nullptr,
5700b57cec5SDimitry Andric         MakeFullExpr(new (Context) OpaqueValueExpr(SourceLocation(),
5710b57cec5SDimitry Andric                                                    Context.BoolTy, VK_RValue),
5720b57cec5SDimitry Andric                      IfLoc),
5730b57cec5SDimitry Andric         false);
5740b57cec5SDimitry Andric 
5750b57cec5SDimitry Andric   Expr *CondExpr = Cond.get().second;
5760b57cec5SDimitry Andric   // Only call the CommaVisitor when not C89 due to differences in scope flags.
5770b57cec5SDimitry Andric   if ((getLangOpts().C99 || getLangOpts().CPlusPlus) &&
5780b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, CondExpr->getExprLoc()))
5790b57cec5SDimitry Andric     CommaVisitor(*this).Visit(CondExpr);
5800b57cec5SDimitry Andric 
5810b57cec5SDimitry Andric   if (!elseStmt)
5820b57cec5SDimitry Andric     DiagnoseEmptyStmtBody(CondExpr->getEndLoc(), thenStmt,
5830b57cec5SDimitry Andric                           diag::warn_empty_if_body);
5840b57cec5SDimitry Andric 
5850b57cec5SDimitry Andric   return BuildIfStmt(IfLoc, IsConstexpr, InitStmt, Cond, thenStmt, ElseLoc,
5860b57cec5SDimitry Andric                      elseStmt);
5870b57cec5SDimitry Andric }
5880b57cec5SDimitry Andric 
5890b57cec5SDimitry Andric StmtResult Sema::BuildIfStmt(SourceLocation IfLoc, bool IsConstexpr,
5900b57cec5SDimitry Andric                              Stmt *InitStmt, ConditionResult Cond,
5910b57cec5SDimitry Andric                              Stmt *thenStmt, SourceLocation ElseLoc,
5920b57cec5SDimitry Andric                              Stmt *elseStmt) {
5930b57cec5SDimitry Andric   if (Cond.isInvalid())
5940b57cec5SDimitry Andric     return StmtError();
5950b57cec5SDimitry Andric 
5960b57cec5SDimitry Andric   if (IsConstexpr || isa<ObjCAvailabilityCheckExpr>(Cond.get().second))
5970b57cec5SDimitry Andric     setFunctionHasBranchProtectedScope();
5980b57cec5SDimitry Andric 
5990b57cec5SDimitry Andric   return IfStmt::Create(Context, IfLoc, IsConstexpr, InitStmt, Cond.get().first,
6000b57cec5SDimitry Andric                         Cond.get().second, thenStmt, ElseLoc, elseStmt);
6010b57cec5SDimitry Andric }
6020b57cec5SDimitry Andric 
6030b57cec5SDimitry Andric namespace {
6040b57cec5SDimitry Andric   struct CaseCompareFunctor {
6050b57cec5SDimitry Andric     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
6060b57cec5SDimitry Andric                     const llvm::APSInt &RHS) {
6070b57cec5SDimitry Andric       return LHS.first < RHS;
6080b57cec5SDimitry Andric     }
6090b57cec5SDimitry Andric     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
6100b57cec5SDimitry Andric                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
6110b57cec5SDimitry Andric       return LHS.first < RHS.first;
6120b57cec5SDimitry Andric     }
6130b57cec5SDimitry Andric     bool operator()(const llvm::APSInt &LHS,
6140b57cec5SDimitry Andric                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
6150b57cec5SDimitry Andric       return LHS < RHS.first;
6160b57cec5SDimitry Andric     }
6170b57cec5SDimitry Andric   };
6180b57cec5SDimitry Andric }
6190b57cec5SDimitry Andric 
6200b57cec5SDimitry Andric /// CmpCaseVals - Comparison predicate for sorting case values.
6210b57cec5SDimitry Andric ///
6220b57cec5SDimitry Andric static bool CmpCaseVals(const std::pair<llvm::APSInt, CaseStmt*>& lhs,
6230b57cec5SDimitry Andric                         const std::pair<llvm::APSInt, CaseStmt*>& rhs) {
6240b57cec5SDimitry Andric   if (lhs.first < rhs.first)
6250b57cec5SDimitry Andric     return true;
6260b57cec5SDimitry Andric 
6270b57cec5SDimitry Andric   if (lhs.first == rhs.first &&
6280b57cec5SDimitry Andric       lhs.second->getCaseLoc().getRawEncoding()
6290b57cec5SDimitry Andric        < rhs.second->getCaseLoc().getRawEncoding())
6300b57cec5SDimitry Andric     return true;
6310b57cec5SDimitry Andric   return false;
6320b57cec5SDimitry Andric }
6330b57cec5SDimitry Andric 
6340b57cec5SDimitry Andric /// CmpEnumVals - Comparison predicate for sorting enumeration values.
6350b57cec5SDimitry Andric ///
6360b57cec5SDimitry Andric static bool CmpEnumVals(const std::pair<llvm::APSInt, EnumConstantDecl*>& lhs,
6370b57cec5SDimitry Andric                         const std::pair<llvm::APSInt, EnumConstantDecl*>& rhs)
6380b57cec5SDimitry Andric {
6390b57cec5SDimitry Andric   return lhs.first < rhs.first;
6400b57cec5SDimitry Andric }
6410b57cec5SDimitry Andric 
6420b57cec5SDimitry Andric /// EqEnumVals - Comparison preficate for uniqing enumeration values.
6430b57cec5SDimitry Andric ///
6440b57cec5SDimitry Andric static bool EqEnumVals(const std::pair<llvm::APSInt, EnumConstantDecl*>& lhs,
6450b57cec5SDimitry Andric                        const std::pair<llvm::APSInt, EnumConstantDecl*>& rhs)
6460b57cec5SDimitry Andric {
6470b57cec5SDimitry Andric   return lhs.first == rhs.first;
6480b57cec5SDimitry Andric }
6490b57cec5SDimitry Andric 
6500b57cec5SDimitry Andric /// GetTypeBeforeIntegralPromotion - Returns the pre-promotion type of
6510b57cec5SDimitry Andric /// potentially integral-promoted expression @p expr.
6520b57cec5SDimitry Andric static QualType GetTypeBeforeIntegralPromotion(const Expr *&E) {
6530b57cec5SDimitry Andric   if (const auto *FE = dyn_cast<FullExpr>(E))
6540b57cec5SDimitry Andric     E = FE->getSubExpr();
6550b57cec5SDimitry Andric   while (const auto *ImpCast = dyn_cast<ImplicitCastExpr>(E)) {
6560b57cec5SDimitry Andric     if (ImpCast->getCastKind() != CK_IntegralCast) break;
6570b57cec5SDimitry Andric     E = ImpCast->getSubExpr();
6580b57cec5SDimitry Andric   }
6590b57cec5SDimitry Andric   return E->getType();
6600b57cec5SDimitry Andric }
6610b57cec5SDimitry Andric 
6620b57cec5SDimitry Andric ExprResult Sema::CheckSwitchCondition(SourceLocation SwitchLoc, Expr *Cond) {
6630b57cec5SDimitry Andric   class SwitchConvertDiagnoser : public ICEConvertDiagnoser {
6640b57cec5SDimitry Andric     Expr *Cond;
6650b57cec5SDimitry Andric 
6660b57cec5SDimitry Andric   public:
6670b57cec5SDimitry Andric     SwitchConvertDiagnoser(Expr *Cond)
6680b57cec5SDimitry Andric         : ICEConvertDiagnoser(/*AllowScopedEnumerations*/true, false, true),
6690b57cec5SDimitry Andric           Cond(Cond) {}
6700b57cec5SDimitry Andric 
6710b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseNotInt(Sema &S, SourceLocation Loc,
6720b57cec5SDimitry Andric                                          QualType T) override {
6730b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_typecheck_statement_requires_integer) << T;
6740b57cec5SDimitry Andric     }
6750b57cec5SDimitry Andric 
6760b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseIncomplete(
6770b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T) override {
6780b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_incomplete_class_type)
6790b57cec5SDimitry Andric                << T << Cond->getSourceRange();
6800b57cec5SDimitry Andric     }
6810b57cec5SDimitry Andric 
6820b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseExplicitConv(
6830b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
6840b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_explicit_conversion) << T << ConvTy;
6850b57cec5SDimitry Andric     }
6860b57cec5SDimitry Andric 
6870b57cec5SDimitry Andric     SemaDiagnosticBuilder noteExplicitConv(
6880b57cec5SDimitry Andric         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
6890b57cec5SDimitry Andric       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
6900b57cec5SDimitry Andric         << ConvTy->isEnumeralType() << ConvTy;
6910b57cec5SDimitry Andric     }
6920b57cec5SDimitry Andric 
6930b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseAmbiguous(Sema &S, SourceLocation Loc,
6940b57cec5SDimitry Andric                                             QualType T) override {
6950b57cec5SDimitry Andric       return S.Diag(Loc, diag::err_switch_multiple_conversions) << T;
6960b57cec5SDimitry Andric     }
6970b57cec5SDimitry Andric 
6980b57cec5SDimitry Andric     SemaDiagnosticBuilder noteAmbiguous(
6990b57cec5SDimitry Andric         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
7000b57cec5SDimitry Andric       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
7010b57cec5SDimitry Andric       << ConvTy->isEnumeralType() << ConvTy;
7020b57cec5SDimitry Andric     }
7030b57cec5SDimitry Andric 
7040b57cec5SDimitry Andric     SemaDiagnosticBuilder diagnoseConversion(
7050b57cec5SDimitry Andric         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
7060b57cec5SDimitry Andric       llvm_unreachable("conversion functions are permitted");
7070b57cec5SDimitry Andric     }
7080b57cec5SDimitry Andric   } SwitchDiagnoser(Cond);
7090b57cec5SDimitry Andric 
7100b57cec5SDimitry Andric   ExprResult CondResult =
7110b57cec5SDimitry Andric       PerformContextualImplicitConversion(SwitchLoc, Cond, SwitchDiagnoser);
7120b57cec5SDimitry Andric   if (CondResult.isInvalid())
7130b57cec5SDimitry Andric     return ExprError();
7140b57cec5SDimitry Andric 
7150b57cec5SDimitry Andric   // FIXME: PerformContextualImplicitConversion doesn't always tell us if it
7160b57cec5SDimitry Andric   // failed and produced a diagnostic.
7170b57cec5SDimitry Andric   Cond = CondResult.get();
7180b57cec5SDimitry Andric   if (!Cond->isTypeDependent() &&
7190b57cec5SDimitry Andric       !Cond->getType()->isIntegralOrEnumerationType())
7200b57cec5SDimitry Andric     return ExprError();
7210b57cec5SDimitry Andric 
7220b57cec5SDimitry Andric   // C99 6.8.4.2p5 - Integer promotions are performed on the controlling expr.
7230b57cec5SDimitry Andric   return UsualUnaryConversions(Cond);
7240b57cec5SDimitry Andric }
7250b57cec5SDimitry Andric 
7260b57cec5SDimitry Andric StmtResult Sema::ActOnStartOfSwitchStmt(SourceLocation SwitchLoc,
7270b57cec5SDimitry Andric                                         Stmt *InitStmt, ConditionResult Cond) {
7280b57cec5SDimitry Andric   Expr *CondExpr = Cond.get().second;
7290b57cec5SDimitry Andric   assert((Cond.isInvalid() || CondExpr) && "switch with no condition");
7300b57cec5SDimitry Andric 
7310b57cec5SDimitry Andric   if (CondExpr && !CondExpr->isTypeDependent()) {
7320b57cec5SDimitry Andric     // We have already converted the expression to an integral or enumeration
7330b57cec5SDimitry Andric     // type, when we parsed the switch condition. If we don't have an
7340b57cec5SDimitry Andric     // appropriate type now, enter the switch scope but remember that it's
7350b57cec5SDimitry Andric     // invalid.
7360b57cec5SDimitry Andric     assert(CondExpr->getType()->isIntegralOrEnumerationType() &&
7370b57cec5SDimitry Andric            "invalid condition type");
7380b57cec5SDimitry Andric     if (CondExpr->isKnownToHaveBooleanValue()) {
7390b57cec5SDimitry Andric       // switch(bool_expr) {...} is often a programmer error, e.g.
7400b57cec5SDimitry Andric       //   switch(n && mask) { ... }  // Doh - should be "n & mask".
7410b57cec5SDimitry Andric       // One can always use an if statement instead of switch(bool_expr).
7420b57cec5SDimitry Andric       Diag(SwitchLoc, diag::warn_bool_switch_condition)
7430b57cec5SDimitry Andric           << CondExpr->getSourceRange();
7440b57cec5SDimitry Andric     }
7450b57cec5SDimitry Andric   }
7460b57cec5SDimitry Andric 
7470b57cec5SDimitry Andric   setFunctionHasBranchIntoScope();
7480b57cec5SDimitry Andric 
7490b57cec5SDimitry Andric   auto *SS = SwitchStmt::Create(Context, InitStmt, Cond.get().first, CondExpr);
7500b57cec5SDimitry Andric   getCurFunction()->SwitchStack.push_back(
7510b57cec5SDimitry Andric       FunctionScopeInfo::SwitchInfo(SS, false));
7520b57cec5SDimitry Andric   return SS;
7530b57cec5SDimitry Andric }
7540b57cec5SDimitry Andric 
7550b57cec5SDimitry Andric static void AdjustAPSInt(llvm::APSInt &Val, unsigned BitWidth, bool IsSigned) {
7560b57cec5SDimitry Andric   Val = Val.extOrTrunc(BitWidth);
7570b57cec5SDimitry Andric   Val.setIsSigned(IsSigned);
7580b57cec5SDimitry Andric }
7590b57cec5SDimitry Andric 
7600b57cec5SDimitry Andric /// Check the specified case value is in range for the given unpromoted switch
7610b57cec5SDimitry Andric /// type.
7620b57cec5SDimitry Andric static void checkCaseValue(Sema &S, SourceLocation Loc, const llvm::APSInt &Val,
7630b57cec5SDimitry Andric                            unsigned UnpromotedWidth, bool UnpromotedSign) {
7640b57cec5SDimitry Andric   // In C++11 onwards, this is checked by the language rules.
7650b57cec5SDimitry Andric   if (S.getLangOpts().CPlusPlus11)
7660b57cec5SDimitry Andric     return;
7670b57cec5SDimitry Andric 
7680b57cec5SDimitry Andric   // If the case value was signed and negative and the switch expression is
7690b57cec5SDimitry Andric   // unsigned, don't bother to warn: this is implementation-defined behavior.
7700b57cec5SDimitry Andric   // FIXME: Introduce a second, default-ignored warning for this case?
7710b57cec5SDimitry Andric   if (UnpromotedWidth < Val.getBitWidth()) {
7720b57cec5SDimitry Andric     llvm::APSInt ConvVal(Val);
7730b57cec5SDimitry Andric     AdjustAPSInt(ConvVal, UnpromotedWidth, UnpromotedSign);
7740b57cec5SDimitry Andric     AdjustAPSInt(ConvVal, Val.getBitWidth(), Val.isSigned());
7750b57cec5SDimitry Andric     // FIXME: Use different diagnostics for overflow  in conversion to promoted
7760b57cec5SDimitry Andric     // type versus "switch expression cannot have this value". Use proper
7770b57cec5SDimitry Andric     // IntRange checking rather than just looking at the unpromoted type here.
7780b57cec5SDimitry Andric     if (ConvVal != Val)
7790b57cec5SDimitry Andric       S.Diag(Loc, diag::warn_case_value_overflow) << Val.toString(10)
7800b57cec5SDimitry Andric                                                   << ConvVal.toString(10);
7810b57cec5SDimitry Andric   }
7820b57cec5SDimitry Andric }
7830b57cec5SDimitry Andric 
7840b57cec5SDimitry Andric typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl*>, 64> EnumValsTy;
7850b57cec5SDimitry Andric 
7860b57cec5SDimitry Andric /// Returns true if we should emit a diagnostic about this case expression not
7870b57cec5SDimitry Andric /// being a part of the enum used in the switch controlling expression.
7880b57cec5SDimitry Andric static bool ShouldDiagnoseSwitchCaseNotInEnum(const Sema &S,
7890b57cec5SDimitry Andric                                               const EnumDecl *ED,
7900b57cec5SDimitry Andric                                               const Expr *CaseExpr,
7910b57cec5SDimitry Andric                                               EnumValsTy::iterator &EI,
7920b57cec5SDimitry Andric                                               EnumValsTy::iterator &EIEnd,
7930b57cec5SDimitry Andric                                               const llvm::APSInt &Val) {
7940b57cec5SDimitry Andric   if (!ED->isClosed())
7950b57cec5SDimitry Andric     return false;
7960b57cec5SDimitry Andric 
7970b57cec5SDimitry Andric   if (const DeclRefExpr *DRE =
7980b57cec5SDimitry Andric           dyn_cast<DeclRefExpr>(CaseExpr->IgnoreParenImpCasts())) {
7990b57cec5SDimitry Andric     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
8000b57cec5SDimitry Andric       QualType VarType = VD->getType();
8010b57cec5SDimitry Andric       QualType EnumType = S.Context.getTypeDeclType(ED);
8020b57cec5SDimitry Andric       if (VD->hasGlobalStorage() && VarType.isConstQualified() &&
8030b57cec5SDimitry Andric           S.Context.hasSameUnqualifiedType(EnumType, VarType))
8040b57cec5SDimitry Andric         return false;
8050b57cec5SDimitry Andric     }
8060b57cec5SDimitry Andric   }
8070b57cec5SDimitry Andric 
8080b57cec5SDimitry Andric   if (ED->hasAttr<FlagEnumAttr>())
8090b57cec5SDimitry Andric     return !S.IsValueInFlagEnum(ED, Val, false);
8100b57cec5SDimitry Andric 
8110b57cec5SDimitry Andric   while (EI != EIEnd && EI->first < Val)
8120b57cec5SDimitry Andric     EI++;
8130b57cec5SDimitry Andric 
8140b57cec5SDimitry Andric   if (EI != EIEnd && EI->first == Val)
8150b57cec5SDimitry Andric     return false;
8160b57cec5SDimitry Andric 
8170b57cec5SDimitry Andric   return true;
8180b57cec5SDimitry Andric }
8190b57cec5SDimitry Andric 
8200b57cec5SDimitry Andric static void checkEnumTypesInSwitchStmt(Sema &S, const Expr *Cond,
8210b57cec5SDimitry Andric                                        const Expr *Case) {
8220b57cec5SDimitry Andric   QualType CondType = Cond->getType();
8230b57cec5SDimitry Andric   QualType CaseType = Case->getType();
8240b57cec5SDimitry Andric 
8250b57cec5SDimitry Andric   const EnumType *CondEnumType = CondType->getAs<EnumType>();
8260b57cec5SDimitry Andric   const EnumType *CaseEnumType = CaseType->getAs<EnumType>();
8270b57cec5SDimitry Andric   if (!CondEnumType || !CaseEnumType)
8280b57cec5SDimitry Andric     return;
8290b57cec5SDimitry Andric 
8300b57cec5SDimitry Andric   // Ignore anonymous enums.
8310b57cec5SDimitry Andric   if (!CondEnumType->getDecl()->getIdentifier() &&
8320b57cec5SDimitry Andric       !CondEnumType->getDecl()->getTypedefNameForAnonDecl())
8330b57cec5SDimitry Andric     return;
8340b57cec5SDimitry Andric   if (!CaseEnumType->getDecl()->getIdentifier() &&
8350b57cec5SDimitry Andric       !CaseEnumType->getDecl()->getTypedefNameForAnonDecl())
8360b57cec5SDimitry Andric     return;
8370b57cec5SDimitry Andric 
8380b57cec5SDimitry Andric   if (S.Context.hasSameUnqualifiedType(CondType, CaseType))
8390b57cec5SDimitry Andric     return;
8400b57cec5SDimitry Andric 
8410b57cec5SDimitry Andric   S.Diag(Case->getExprLoc(), diag::warn_comparison_of_mixed_enum_types_switch)
8420b57cec5SDimitry Andric       << CondType << CaseType << Cond->getSourceRange()
8430b57cec5SDimitry Andric       << Case->getSourceRange();
8440b57cec5SDimitry Andric }
8450b57cec5SDimitry Andric 
8460b57cec5SDimitry Andric StmtResult
8470b57cec5SDimitry Andric Sema::ActOnFinishSwitchStmt(SourceLocation SwitchLoc, Stmt *Switch,
8480b57cec5SDimitry Andric                             Stmt *BodyStmt) {
8490b57cec5SDimitry Andric   SwitchStmt *SS = cast<SwitchStmt>(Switch);
8500b57cec5SDimitry Andric   bool CaseListIsIncomplete = getCurFunction()->SwitchStack.back().getInt();
8510b57cec5SDimitry Andric   assert(SS == getCurFunction()->SwitchStack.back().getPointer() &&
8520b57cec5SDimitry Andric          "switch stack missing push/pop!");
8530b57cec5SDimitry Andric 
8540b57cec5SDimitry Andric   getCurFunction()->SwitchStack.pop_back();
8550b57cec5SDimitry Andric 
8560b57cec5SDimitry Andric   if (!BodyStmt) return StmtError();
8570b57cec5SDimitry Andric   SS->setBody(BodyStmt, SwitchLoc);
8580b57cec5SDimitry Andric 
8590b57cec5SDimitry Andric   Expr *CondExpr = SS->getCond();
8600b57cec5SDimitry Andric   if (!CondExpr) return StmtError();
8610b57cec5SDimitry Andric 
8620b57cec5SDimitry Andric   QualType CondType = CondExpr->getType();
8630b57cec5SDimitry Andric 
8640b57cec5SDimitry Andric   // C++ 6.4.2.p2:
8650b57cec5SDimitry Andric   // Integral promotions are performed (on the switch condition).
8660b57cec5SDimitry Andric   //
8670b57cec5SDimitry Andric   // A case value unrepresentable by the original switch condition
8680b57cec5SDimitry Andric   // type (before the promotion) doesn't make sense, even when it can
8690b57cec5SDimitry Andric   // be represented by the promoted type.  Therefore we need to find
8700b57cec5SDimitry Andric   // the pre-promotion type of the switch condition.
8710b57cec5SDimitry Andric   const Expr *CondExprBeforePromotion = CondExpr;
8720b57cec5SDimitry Andric   QualType CondTypeBeforePromotion =
8730b57cec5SDimitry Andric       GetTypeBeforeIntegralPromotion(CondExprBeforePromotion);
8740b57cec5SDimitry Andric 
8750b57cec5SDimitry Andric   // Get the bitwidth of the switched-on value after promotions. We must
8760b57cec5SDimitry Andric   // convert the integer case values to this width before comparison.
8770b57cec5SDimitry Andric   bool HasDependentValue
8780b57cec5SDimitry Andric     = CondExpr->isTypeDependent() || CondExpr->isValueDependent();
8790b57cec5SDimitry Andric   unsigned CondWidth = HasDependentValue ? 0 : Context.getIntWidth(CondType);
8800b57cec5SDimitry Andric   bool CondIsSigned = CondType->isSignedIntegerOrEnumerationType();
8810b57cec5SDimitry Andric 
8820b57cec5SDimitry Andric   // Get the width and signedness that the condition might actually have, for
8830b57cec5SDimitry Andric   // warning purposes.
8840b57cec5SDimitry Andric   // FIXME: Grab an IntRange for the condition rather than using the unpromoted
8850b57cec5SDimitry Andric   // type.
8860b57cec5SDimitry Andric   unsigned CondWidthBeforePromotion
8870b57cec5SDimitry Andric     = HasDependentValue ? 0 : Context.getIntWidth(CondTypeBeforePromotion);
8880b57cec5SDimitry Andric   bool CondIsSignedBeforePromotion
8890b57cec5SDimitry Andric     = CondTypeBeforePromotion->isSignedIntegerOrEnumerationType();
8900b57cec5SDimitry Andric 
8910b57cec5SDimitry Andric   // Accumulate all of the case values in a vector so that we can sort them
8920b57cec5SDimitry Andric   // and detect duplicates.  This vector contains the APInt for the case after
8930b57cec5SDimitry Andric   // it has been converted to the condition type.
8940b57cec5SDimitry Andric   typedef SmallVector<std::pair<llvm::APSInt, CaseStmt*>, 64> CaseValsTy;
8950b57cec5SDimitry Andric   CaseValsTy CaseVals;
8960b57cec5SDimitry Andric 
8970b57cec5SDimitry Andric   // Keep track of any GNU case ranges we see.  The APSInt is the low value.
8980b57cec5SDimitry Andric   typedef std::vector<std::pair<llvm::APSInt, CaseStmt*> > CaseRangesTy;
8990b57cec5SDimitry Andric   CaseRangesTy CaseRanges;
9000b57cec5SDimitry Andric 
9010b57cec5SDimitry Andric   DefaultStmt *TheDefaultStmt = nullptr;
9020b57cec5SDimitry Andric 
9030b57cec5SDimitry Andric   bool CaseListIsErroneous = false;
9040b57cec5SDimitry Andric 
9050b57cec5SDimitry Andric   for (SwitchCase *SC = SS->getSwitchCaseList(); SC && !HasDependentValue;
9060b57cec5SDimitry Andric        SC = SC->getNextSwitchCase()) {
9070b57cec5SDimitry Andric 
9080b57cec5SDimitry Andric     if (DefaultStmt *DS = dyn_cast<DefaultStmt>(SC)) {
9090b57cec5SDimitry Andric       if (TheDefaultStmt) {
9100b57cec5SDimitry Andric         Diag(DS->getDefaultLoc(), diag::err_multiple_default_labels_defined);
9110b57cec5SDimitry Andric         Diag(TheDefaultStmt->getDefaultLoc(), diag::note_duplicate_case_prev);
9120b57cec5SDimitry Andric 
9130b57cec5SDimitry Andric         // FIXME: Remove the default statement from the switch block so that
9140b57cec5SDimitry Andric         // we'll return a valid AST.  This requires recursing down the AST and
9150b57cec5SDimitry Andric         // finding it, not something we are set up to do right now.  For now,
9160b57cec5SDimitry Andric         // just lop the entire switch stmt out of the AST.
9170b57cec5SDimitry Andric         CaseListIsErroneous = true;
9180b57cec5SDimitry Andric       }
9190b57cec5SDimitry Andric       TheDefaultStmt = DS;
9200b57cec5SDimitry Andric 
9210b57cec5SDimitry Andric     } else {
9220b57cec5SDimitry Andric       CaseStmt *CS = cast<CaseStmt>(SC);
9230b57cec5SDimitry Andric 
9240b57cec5SDimitry Andric       Expr *Lo = CS->getLHS();
9250b57cec5SDimitry Andric 
9260b57cec5SDimitry Andric       if (Lo->isValueDependent()) {
9270b57cec5SDimitry Andric         HasDependentValue = true;
9280b57cec5SDimitry Andric         break;
9290b57cec5SDimitry Andric       }
9300b57cec5SDimitry Andric 
9310b57cec5SDimitry Andric       // We already verified that the expression has a constant value;
9320b57cec5SDimitry Andric       // get that value (prior to conversions).
9330b57cec5SDimitry Andric       const Expr *LoBeforePromotion = Lo;
9340b57cec5SDimitry Andric       GetTypeBeforeIntegralPromotion(LoBeforePromotion);
9350b57cec5SDimitry Andric       llvm::APSInt LoVal = LoBeforePromotion->EvaluateKnownConstInt(Context);
9360b57cec5SDimitry Andric 
9370b57cec5SDimitry Andric       // Check the unconverted value is within the range of possible values of
9380b57cec5SDimitry Andric       // the switch expression.
9390b57cec5SDimitry Andric       checkCaseValue(*this, Lo->getBeginLoc(), LoVal, CondWidthBeforePromotion,
9400b57cec5SDimitry Andric                      CondIsSignedBeforePromotion);
9410b57cec5SDimitry Andric 
9420b57cec5SDimitry Andric       // FIXME: This duplicates the check performed for warn_not_in_enum below.
9430b57cec5SDimitry Andric       checkEnumTypesInSwitchStmt(*this, CondExprBeforePromotion,
9440b57cec5SDimitry Andric                                  LoBeforePromotion);
9450b57cec5SDimitry Andric 
9460b57cec5SDimitry Andric       // Convert the value to the same width/sign as the condition.
9470b57cec5SDimitry Andric       AdjustAPSInt(LoVal, CondWidth, CondIsSigned);
9480b57cec5SDimitry Andric 
9490b57cec5SDimitry Andric       // If this is a case range, remember it in CaseRanges, otherwise CaseVals.
9500b57cec5SDimitry Andric       if (CS->getRHS()) {
9510b57cec5SDimitry Andric         if (CS->getRHS()->isValueDependent()) {
9520b57cec5SDimitry Andric           HasDependentValue = true;
9530b57cec5SDimitry Andric           break;
9540b57cec5SDimitry Andric         }
9550b57cec5SDimitry Andric         CaseRanges.push_back(std::make_pair(LoVal, CS));
9560b57cec5SDimitry Andric       } else
9570b57cec5SDimitry Andric         CaseVals.push_back(std::make_pair(LoVal, CS));
9580b57cec5SDimitry Andric     }
9590b57cec5SDimitry Andric   }
9600b57cec5SDimitry Andric 
9610b57cec5SDimitry Andric   if (!HasDependentValue) {
9620b57cec5SDimitry Andric     // If we don't have a default statement, check whether the
9630b57cec5SDimitry Andric     // condition is constant.
9640b57cec5SDimitry Andric     llvm::APSInt ConstantCondValue;
9650b57cec5SDimitry Andric     bool HasConstantCond = false;
966*a7dea167SDimitry Andric     if (!TheDefaultStmt) {
9670b57cec5SDimitry Andric       Expr::EvalResult Result;
9680b57cec5SDimitry Andric       HasConstantCond = CondExpr->EvaluateAsInt(Result, Context,
9690b57cec5SDimitry Andric                                                 Expr::SE_AllowSideEffects);
9700b57cec5SDimitry Andric       if (Result.Val.isInt())
9710b57cec5SDimitry Andric         ConstantCondValue = Result.Val.getInt();
9720b57cec5SDimitry Andric       assert(!HasConstantCond ||
9730b57cec5SDimitry Andric              (ConstantCondValue.getBitWidth() == CondWidth &&
9740b57cec5SDimitry Andric               ConstantCondValue.isSigned() == CondIsSigned));
9750b57cec5SDimitry Andric     }
9760b57cec5SDimitry Andric     bool ShouldCheckConstantCond = HasConstantCond;
9770b57cec5SDimitry Andric 
9780b57cec5SDimitry Andric     // Sort all the scalar case values so we can easily detect duplicates.
9790b57cec5SDimitry Andric     llvm::stable_sort(CaseVals, CmpCaseVals);
9800b57cec5SDimitry Andric 
9810b57cec5SDimitry Andric     if (!CaseVals.empty()) {
9820b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseVals.size(); i != e; ++i) {
9830b57cec5SDimitry Andric         if (ShouldCheckConstantCond &&
9840b57cec5SDimitry Andric             CaseVals[i].first == ConstantCondValue)
9850b57cec5SDimitry Andric           ShouldCheckConstantCond = false;
9860b57cec5SDimitry Andric 
9870b57cec5SDimitry Andric         if (i != 0 && CaseVals[i].first == CaseVals[i-1].first) {
9880b57cec5SDimitry Andric           // If we have a duplicate, report it.
9890b57cec5SDimitry Andric           // First, determine if either case value has a name
9900b57cec5SDimitry Andric           StringRef PrevString, CurrString;
9910b57cec5SDimitry Andric           Expr *PrevCase = CaseVals[i-1].second->getLHS()->IgnoreParenCasts();
9920b57cec5SDimitry Andric           Expr *CurrCase = CaseVals[i].second->getLHS()->IgnoreParenCasts();
9930b57cec5SDimitry Andric           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(PrevCase)) {
9940b57cec5SDimitry Andric             PrevString = DeclRef->getDecl()->getName();
9950b57cec5SDimitry Andric           }
9960b57cec5SDimitry Andric           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(CurrCase)) {
9970b57cec5SDimitry Andric             CurrString = DeclRef->getDecl()->getName();
9980b57cec5SDimitry Andric           }
9990b57cec5SDimitry Andric           SmallString<16> CaseValStr;
10000b57cec5SDimitry Andric           CaseVals[i-1].first.toString(CaseValStr);
10010b57cec5SDimitry Andric 
10020b57cec5SDimitry Andric           if (PrevString == CurrString)
10030b57cec5SDimitry Andric             Diag(CaseVals[i].second->getLHS()->getBeginLoc(),
10040b57cec5SDimitry Andric                  diag::err_duplicate_case)
10050b57cec5SDimitry Andric                 << (PrevString.empty() ? StringRef(CaseValStr) : PrevString);
10060b57cec5SDimitry Andric           else
10070b57cec5SDimitry Andric             Diag(CaseVals[i].second->getLHS()->getBeginLoc(),
10080b57cec5SDimitry Andric                  diag::err_duplicate_case_differing_expr)
10090b57cec5SDimitry Andric                 << (PrevString.empty() ? StringRef(CaseValStr) : PrevString)
10100b57cec5SDimitry Andric                 << (CurrString.empty() ? StringRef(CaseValStr) : CurrString)
10110b57cec5SDimitry Andric                 << CaseValStr;
10120b57cec5SDimitry Andric 
10130b57cec5SDimitry Andric           Diag(CaseVals[i - 1].second->getLHS()->getBeginLoc(),
10140b57cec5SDimitry Andric                diag::note_duplicate_case_prev);
10150b57cec5SDimitry Andric           // FIXME: We really want to remove the bogus case stmt from the
10160b57cec5SDimitry Andric           // substmt, but we have no way to do this right now.
10170b57cec5SDimitry Andric           CaseListIsErroneous = true;
10180b57cec5SDimitry Andric         }
10190b57cec5SDimitry Andric       }
10200b57cec5SDimitry Andric     }
10210b57cec5SDimitry Andric 
10220b57cec5SDimitry Andric     // Detect duplicate case ranges, which usually don't exist at all in
10230b57cec5SDimitry Andric     // the first place.
10240b57cec5SDimitry Andric     if (!CaseRanges.empty()) {
10250b57cec5SDimitry Andric       // Sort all the case ranges by their low value so we can easily detect
10260b57cec5SDimitry Andric       // overlaps between ranges.
10270b57cec5SDimitry Andric       llvm::stable_sort(CaseRanges);
10280b57cec5SDimitry Andric 
10290b57cec5SDimitry Andric       // Scan the ranges, computing the high values and removing empty ranges.
10300b57cec5SDimitry Andric       std::vector<llvm::APSInt> HiVals;
10310b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
10320b57cec5SDimitry Andric         llvm::APSInt &LoVal = CaseRanges[i].first;
10330b57cec5SDimitry Andric         CaseStmt *CR = CaseRanges[i].second;
10340b57cec5SDimitry Andric         Expr *Hi = CR->getRHS();
10350b57cec5SDimitry Andric 
10360b57cec5SDimitry Andric         const Expr *HiBeforePromotion = Hi;
10370b57cec5SDimitry Andric         GetTypeBeforeIntegralPromotion(HiBeforePromotion);
10380b57cec5SDimitry Andric         llvm::APSInt HiVal = HiBeforePromotion->EvaluateKnownConstInt(Context);
10390b57cec5SDimitry Andric 
10400b57cec5SDimitry Andric         // Check the unconverted value is within the range of possible values of
10410b57cec5SDimitry Andric         // the switch expression.
10420b57cec5SDimitry Andric         checkCaseValue(*this, Hi->getBeginLoc(), HiVal,
10430b57cec5SDimitry Andric                        CondWidthBeforePromotion, CondIsSignedBeforePromotion);
10440b57cec5SDimitry Andric 
10450b57cec5SDimitry Andric         // Convert the value to the same width/sign as the condition.
10460b57cec5SDimitry Andric         AdjustAPSInt(HiVal, CondWidth, CondIsSigned);
10470b57cec5SDimitry Andric 
10480b57cec5SDimitry Andric         // If the low value is bigger than the high value, the case is empty.
10490b57cec5SDimitry Andric         if (LoVal > HiVal) {
10500b57cec5SDimitry Andric           Diag(CR->getLHS()->getBeginLoc(), diag::warn_case_empty_range)
10510b57cec5SDimitry Andric               << SourceRange(CR->getLHS()->getBeginLoc(), Hi->getEndLoc());
10520b57cec5SDimitry Andric           CaseRanges.erase(CaseRanges.begin()+i);
10530b57cec5SDimitry Andric           --i;
10540b57cec5SDimitry Andric           --e;
10550b57cec5SDimitry Andric           continue;
10560b57cec5SDimitry Andric         }
10570b57cec5SDimitry Andric 
10580b57cec5SDimitry Andric         if (ShouldCheckConstantCond &&
10590b57cec5SDimitry Andric             LoVal <= ConstantCondValue &&
10600b57cec5SDimitry Andric             ConstantCondValue <= HiVal)
10610b57cec5SDimitry Andric           ShouldCheckConstantCond = false;
10620b57cec5SDimitry Andric 
10630b57cec5SDimitry Andric         HiVals.push_back(HiVal);
10640b57cec5SDimitry Andric       }
10650b57cec5SDimitry Andric 
10660b57cec5SDimitry Andric       // Rescan the ranges, looking for overlap with singleton values and other
10670b57cec5SDimitry Andric       // ranges.  Since the range list is sorted, we only need to compare case
10680b57cec5SDimitry Andric       // ranges with their neighbors.
10690b57cec5SDimitry Andric       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
10700b57cec5SDimitry Andric         llvm::APSInt &CRLo = CaseRanges[i].first;
10710b57cec5SDimitry Andric         llvm::APSInt &CRHi = HiVals[i];
10720b57cec5SDimitry Andric         CaseStmt *CR = CaseRanges[i].second;
10730b57cec5SDimitry Andric 
10740b57cec5SDimitry Andric         // Check to see whether the case range overlaps with any
10750b57cec5SDimitry Andric         // singleton cases.
10760b57cec5SDimitry Andric         CaseStmt *OverlapStmt = nullptr;
10770b57cec5SDimitry Andric         llvm::APSInt OverlapVal(32);
10780b57cec5SDimitry Andric 
10790b57cec5SDimitry Andric         // Find the smallest value >= the lower bound.  If I is in the
10800b57cec5SDimitry Andric         // case range, then we have overlap.
10810b57cec5SDimitry Andric         CaseValsTy::iterator I =
10820b57cec5SDimitry Andric             llvm::lower_bound(CaseVals, CRLo, CaseCompareFunctor());
10830b57cec5SDimitry Andric         if (I != CaseVals.end() && I->first < CRHi) {
10840b57cec5SDimitry Andric           OverlapVal  = I->first;   // Found overlap with scalar.
10850b57cec5SDimitry Andric           OverlapStmt = I->second;
10860b57cec5SDimitry Andric         }
10870b57cec5SDimitry Andric 
10880b57cec5SDimitry Andric         // Find the smallest value bigger than the upper bound.
10890b57cec5SDimitry Andric         I = std::upper_bound(I, CaseVals.end(), CRHi, CaseCompareFunctor());
10900b57cec5SDimitry Andric         if (I != CaseVals.begin() && (I-1)->first >= CRLo) {
10910b57cec5SDimitry Andric           OverlapVal  = (I-1)->first;      // Found overlap with scalar.
10920b57cec5SDimitry Andric           OverlapStmt = (I-1)->second;
10930b57cec5SDimitry Andric         }
10940b57cec5SDimitry Andric 
10950b57cec5SDimitry Andric         // Check to see if this case stmt overlaps with the subsequent
10960b57cec5SDimitry Andric         // case range.
10970b57cec5SDimitry Andric         if (i && CRLo <= HiVals[i-1]) {
10980b57cec5SDimitry Andric           OverlapVal  = HiVals[i-1];       // Found overlap with range.
10990b57cec5SDimitry Andric           OverlapStmt = CaseRanges[i-1].second;
11000b57cec5SDimitry Andric         }
11010b57cec5SDimitry Andric 
11020b57cec5SDimitry Andric         if (OverlapStmt) {
11030b57cec5SDimitry Andric           // If we have a duplicate, report it.
11040b57cec5SDimitry Andric           Diag(CR->getLHS()->getBeginLoc(), diag::err_duplicate_case)
11050b57cec5SDimitry Andric               << OverlapVal.toString(10);
11060b57cec5SDimitry Andric           Diag(OverlapStmt->getLHS()->getBeginLoc(),
11070b57cec5SDimitry Andric                diag::note_duplicate_case_prev);
11080b57cec5SDimitry Andric           // FIXME: We really want to remove the bogus case stmt from the
11090b57cec5SDimitry Andric           // substmt, but we have no way to do this right now.
11100b57cec5SDimitry Andric           CaseListIsErroneous = true;
11110b57cec5SDimitry Andric         }
11120b57cec5SDimitry Andric       }
11130b57cec5SDimitry Andric     }
11140b57cec5SDimitry Andric 
11150b57cec5SDimitry Andric     // Complain if we have a constant condition and we didn't find a match.
11160b57cec5SDimitry Andric     if (!CaseListIsErroneous && !CaseListIsIncomplete &&
11170b57cec5SDimitry Andric         ShouldCheckConstantCond) {
11180b57cec5SDimitry Andric       // TODO: it would be nice if we printed enums as enums, chars as
11190b57cec5SDimitry Andric       // chars, etc.
11200b57cec5SDimitry Andric       Diag(CondExpr->getExprLoc(), diag::warn_missing_case_for_condition)
11210b57cec5SDimitry Andric         << ConstantCondValue.toString(10)
11220b57cec5SDimitry Andric         << CondExpr->getSourceRange();
11230b57cec5SDimitry Andric     }
11240b57cec5SDimitry Andric 
11250b57cec5SDimitry Andric     // Check to see if switch is over an Enum and handles all of its
11260b57cec5SDimitry Andric     // values.  We only issue a warning if there is not 'default:', but
11270b57cec5SDimitry Andric     // we still do the analysis to preserve this information in the AST
11280b57cec5SDimitry Andric     // (which can be used by flow-based analyes).
11290b57cec5SDimitry Andric     //
11300b57cec5SDimitry Andric     const EnumType *ET = CondTypeBeforePromotion->getAs<EnumType>();
11310b57cec5SDimitry Andric 
11320b57cec5SDimitry Andric     // If switch has default case, then ignore it.
11330b57cec5SDimitry Andric     if (!CaseListIsErroneous && !CaseListIsIncomplete && !HasConstantCond &&
11340b57cec5SDimitry Andric         ET && ET->getDecl()->isCompleteDefinition()) {
11350b57cec5SDimitry Andric       const EnumDecl *ED = ET->getDecl();
11360b57cec5SDimitry Andric       EnumValsTy EnumVals;
11370b57cec5SDimitry Andric 
11380b57cec5SDimitry Andric       // Gather all enum values, set their type and sort them,
11390b57cec5SDimitry Andric       // allowing easier comparison with CaseVals.
11400b57cec5SDimitry Andric       for (auto *EDI : ED->enumerators()) {
11410b57cec5SDimitry Andric         llvm::APSInt Val = EDI->getInitVal();
11420b57cec5SDimitry Andric         AdjustAPSInt(Val, CondWidth, CondIsSigned);
11430b57cec5SDimitry Andric         EnumVals.push_back(std::make_pair(Val, EDI));
11440b57cec5SDimitry Andric       }
11450b57cec5SDimitry Andric       llvm::stable_sort(EnumVals, CmpEnumVals);
11460b57cec5SDimitry Andric       auto EI = EnumVals.begin(), EIEnd =
11470b57cec5SDimitry Andric         std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
11480b57cec5SDimitry Andric 
11490b57cec5SDimitry Andric       // See which case values aren't in enum.
11500b57cec5SDimitry Andric       for (CaseValsTy::const_iterator CI = CaseVals.begin();
11510b57cec5SDimitry Andric           CI != CaseVals.end(); CI++) {
11520b57cec5SDimitry Andric         Expr *CaseExpr = CI->second->getLHS();
11530b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
11540b57cec5SDimitry Andric                                               CI->first))
11550b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
11560b57cec5SDimitry Andric             << CondTypeBeforePromotion;
11570b57cec5SDimitry Andric       }
11580b57cec5SDimitry Andric 
11590b57cec5SDimitry Andric       // See which of case ranges aren't in enum
11600b57cec5SDimitry Andric       EI = EnumVals.begin();
11610b57cec5SDimitry Andric       for (CaseRangesTy::const_iterator RI = CaseRanges.begin();
11620b57cec5SDimitry Andric           RI != CaseRanges.end(); RI++) {
11630b57cec5SDimitry Andric         Expr *CaseExpr = RI->second->getLHS();
11640b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
11650b57cec5SDimitry Andric                                               RI->first))
11660b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
11670b57cec5SDimitry Andric             << CondTypeBeforePromotion;
11680b57cec5SDimitry Andric 
11690b57cec5SDimitry Andric         llvm::APSInt Hi =
11700b57cec5SDimitry Andric           RI->second->getRHS()->EvaluateKnownConstInt(Context);
11710b57cec5SDimitry Andric         AdjustAPSInt(Hi, CondWidth, CondIsSigned);
11720b57cec5SDimitry Andric 
11730b57cec5SDimitry Andric         CaseExpr = RI->second->getRHS();
11740b57cec5SDimitry Andric         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
11750b57cec5SDimitry Andric                                               Hi))
11760b57cec5SDimitry Andric           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
11770b57cec5SDimitry Andric             << CondTypeBeforePromotion;
11780b57cec5SDimitry Andric       }
11790b57cec5SDimitry Andric 
11800b57cec5SDimitry Andric       // Check which enum vals aren't in switch
11810b57cec5SDimitry Andric       auto CI = CaseVals.begin();
11820b57cec5SDimitry Andric       auto RI = CaseRanges.begin();
11830b57cec5SDimitry Andric       bool hasCasesNotInSwitch = false;
11840b57cec5SDimitry Andric 
11850b57cec5SDimitry Andric       SmallVector<DeclarationName,8> UnhandledNames;
11860b57cec5SDimitry Andric 
11870b57cec5SDimitry Andric       for (EI = EnumVals.begin(); EI != EIEnd; EI++) {
11880b57cec5SDimitry Andric         // Don't warn about omitted unavailable EnumConstantDecls.
11890b57cec5SDimitry Andric         switch (EI->second->getAvailability()) {
11900b57cec5SDimitry Andric         case AR_Deprecated:
11910b57cec5SDimitry Andric           // Omitting a deprecated constant is ok; it should never materialize.
11920b57cec5SDimitry Andric         case AR_Unavailable:
11930b57cec5SDimitry Andric           continue;
11940b57cec5SDimitry Andric 
11950b57cec5SDimitry Andric         case AR_NotYetIntroduced:
11960b57cec5SDimitry Andric           // Partially available enum constants should be present. Note that we
11970b57cec5SDimitry Andric           // suppress -Wunguarded-availability diagnostics for such uses.
11980b57cec5SDimitry Andric         case AR_Available:
11990b57cec5SDimitry Andric           break;
12000b57cec5SDimitry Andric         }
12010b57cec5SDimitry Andric 
12020b57cec5SDimitry Andric         if (EI->second->hasAttr<UnusedAttr>())
12030b57cec5SDimitry Andric           continue;
12040b57cec5SDimitry Andric 
12050b57cec5SDimitry Andric         // Drop unneeded case values
12060b57cec5SDimitry Andric         while (CI != CaseVals.end() && CI->first < EI->first)
12070b57cec5SDimitry Andric           CI++;
12080b57cec5SDimitry Andric 
12090b57cec5SDimitry Andric         if (CI != CaseVals.end() && CI->first == EI->first)
12100b57cec5SDimitry Andric           continue;
12110b57cec5SDimitry Andric 
12120b57cec5SDimitry Andric         // Drop unneeded case ranges
12130b57cec5SDimitry Andric         for (; RI != CaseRanges.end(); RI++) {
12140b57cec5SDimitry Andric           llvm::APSInt Hi =
12150b57cec5SDimitry Andric             RI->second->getRHS()->EvaluateKnownConstInt(Context);
12160b57cec5SDimitry Andric           AdjustAPSInt(Hi, CondWidth, CondIsSigned);
12170b57cec5SDimitry Andric           if (EI->first <= Hi)
12180b57cec5SDimitry Andric             break;
12190b57cec5SDimitry Andric         }
12200b57cec5SDimitry Andric 
12210b57cec5SDimitry Andric         if (RI == CaseRanges.end() || EI->first < RI->first) {
12220b57cec5SDimitry Andric           hasCasesNotInSwitch = true;
12230b57cec5SDimitry Andric           UnhandledNames.push_back(EI->second->getDeclName());
12240b57cec5SDimitry Andric         }
12250b57cec5SDimitry Andric       }
12260b57cec5SDimitry Andric 
12270b57cec5SDimitry Andric       if (TheDefaultStmt && UnhandledNames.empty() && ED->isClosedNonFlag())
12280b57cec5SDimitry Andric         Diag(TheDefaultStmt->getDefaultLoc(), diag::warn_unreachable_default);
12290b57cec5SDimitry Andric 
12300b57cec5SDimitry Andric       // Produce a nice diagnostic if multiple values aren't handled.
12310b57cec5SDimitry Andric       if (!UnhandledNames.empty()) {
12320b57cec5SDimitry Andric         DiagnosticBuilder DB = Diag(CondExpr->getExprLoc(),
12330b57cec5SDimitry Andric                                     TheDefaultStmt ? diag::warn_def_missing_case
12340b57cec5SDimitry Andric                                                    : diag::warn_missing_case)
12350b57cec5SDimitry Andric                                << (int)UnhandledNames.size();
12360b57cec5SDimitry Andric 
12370b57cec5SDimitry Andric         for (size_t I = 0, E = std::min(UnhandledNames.size(), (size_t)3);
12380b57cec5SDimitry Andric              I != E; ++I)
12390b57cec5SDimitry Andric           DB << UnhandledNames[I];
12400b57cec5SDimitry Andric       }
12410b57cec5SDimitry Andric 
12420b57cec5SDimitry Andric       if (!hasCasesNotInSwitch)
12430b57cec5SDimitry Andric         SS->setAllEnumCasesCovered();
12440b57cec5SDimitry Andric     }
12450b57cec5SDimitry Andric   }
12460b57cec5SDimitry Andric 
12470b57cec5SDimitry Andric   if (BodyStmt)
12480b57cec5SDimitry Andric     DiagnoseEmptyStmtBody(CondExpr->getEndLoc(), BodyStmt,
12490b57cec5SDimitry Andric                           diag::warn_empty_switch_body);
12500b57cec5SDimitry Andric 
12510b57cec5SDimitry Andric   // FIXME: If the case list was broken is some way, we don't have a good system
12520b57cec5SDimitry Andric   // to patch it up.  Instead, just return the whole substmt as broken.
12530b57cec5SDimitry Andric   if (CaseListIsErroneous)
12540b57cec5SDimitry Andric     return StmtError();
12550b57cec5SDimitry Andric 
12560b57cec5SDimitry Andric   return SS;
12570b57cec5SDimitry Andric }
12580b57cec5SDimitry Andric 
12590b57cec5SDimitry Andric void
12600b57cec5SDimitry Andric Sema::DiagnoseAssignmentEnum(QualType DstType, QualType SrcType,
12610b57cec5SDimitry Andric                              Expr *SrcExpr) {
12620b57cec5SDimitry Andric   if (Diags.isIgnored(diag::warn_not_in_enum_assignment, SrcExpr->getExprLoc()))
12630b57cec5SDimitry Andric     return;
12640b57cec5SDimitry Andric 
12650b57cec5SDimitry Andric   if (const EnumType *ET = DstType->getAs<EnumType>())
12660b57cec5SDimitry Andric     if (!Context.hasSameUnqualifiedType(SrcType, DstType) &&
12670b57cec5SDimitry Andric         SrcType->isIntegerType()) {
12680b57cec5SDimitry Andric       if (!SrcExpr->isTypeDependent() && !SrcExpr->isValueDependent() &&
12690b57cec5SDimitry Andric           SrcExpr->isIntegerConstantExpr(Context)) {
12700b57cec5SDimitry Andric         // Get the bitwidth of the enum value before promotions.
12710b57cec5SDimitry Andric         unsigned DstWidth = Context.getIntWidth(DstType);
12720b57cec5SDimitry Andric         bool DstIsSigned = DstType->isSignedIntegerOrEnumerationType();
12730b57cec5SDimitry Andric 
12740b57cec5SDimitry Andric         llvm::APSInt RhsVal = SrcExpr->EvaluateKnownConstInt(Context);
12750b57cec5SDimitry Andric         AdjustAPSInt(RhsVal, DstWidth, DstIsSigned);
12760b57cec5SDimitry Andric         const EnumDecl *ED = ET->getDecl();
12770b57cec5SDimitry Andric 
12780b57cec5SDimitry Andric         if (!ED->isClosed())
12790b57cec5SDimitry Andric           return;
12800b57cec5SDimitry Andric 
12810b57cec5SDimitry Andric         if (ED->hasAttr<FlagEnumAttr>()) {
12820b57cec5SDimitry Andric           if (!IsValueInFlagEnum(ED, RhsVal, true))
12830b57cec5SDimitry Andric             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
12840b57cec5SDimitry Andric               << DstType.getUnqualifiedType();
12850b57cec5SDimitry Andric         } else {
12860b57cec5SDimitry Andric           typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl *>, 64>
12870b57cec5SDimitry Andric               EnumValsTy;
12880b57cec5SDimitry Andric           EnumValsTy EnumVals;
12890b57cec5SDimitry Andric 
12900b57cec5SDimitry Andric           // Gather all enum values, set their type and sort them,
12910b57cec5SDimitry Andric           // allowing easier comparison with rhs constant.
12920b57cec5SDimitry Andric           for (auto *EDI : ED->enumerators()) {
12930b57cec5SDimitry Andric             llvm::APSInt Val = EDI->getInitVal();
12940b57cec5SDimitry Andric             AdjustAPSInt(Val, DstWidth, DstIsSigned);
12950b57cec5SDimitry Andric             EnumVals.push_back(std::make_pair(Val, EDI));
12960b57cec5SDimitry Andric           }
12970b57cec5SDimitry Andric           if (EnumVals.empty())
12980b57cec5SDimitry Andric             return;
12990b57cec5SDimitry Andric           llvm::stable_sort(EnumVals, CmpEnumVals);
13000b57cec5SDimitry Andric           EnumValsTy::iterator EIend =
13010b57cec5SDimitry Andric               std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
13020b57cec5SDimitry Andric 
13030b57cec5SDimitry Andric           // See which values aren't in the enum.
13040b57cec5SDimitry Andric           EnumValsTy::const_iterator EI = EnumVals.begin();
13050b57cec5SDimitry Andric           while (EI != EIend && EI->first < RhsVal)
13060b57cec5SDimitry Andric             EI++;
13070b57cec5SDimitry Andric           if (EI == EIend || EI->first != RhsVal) {
13080b57cec5SDimitry Andric             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
13090b57cec5SDimitry Andric                 << DstType.getUnqualifiedType();
13100b57cec5SDimitry Andric           }
13110b57cec5SDimitry Andric         }
13120b57cec5SDimitry Andric       }
13130b57cec5SDimitry Andric     }
13140b57cec5SDimitry Andric }
13150b57cec5SDimitry Andric 
13160b57cec5SDimitry Andric StmtResult Sema::ActOnWhileStmt(SourceLocation WhileLoc, ConditionResult Cond,
13170b57cec5SDimitry Andric                                 Stmt *Body) {
13180b57cec5SDimitry Andric   if (Cond.isInvalid())
13190b57cec5SDimitry Andric     return StmtError();
13200b57cec5SDimitry Andric 
13210b57cec5SDimitry Andric   auto CondVal = Cond.get();
13220b57cec5SDimitry Andric   CheckBreakContinueBinding(CondVal.second);
13230b57cec5SDimitry Andric 
13240b57cec5SDimitry Andric   if (CondVal.second &&
13250b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, CondVal.second->getExprLoc()))
13260b57cec5SDimitry Andric     CommaVisitor(*this).Visit(CondVal.second);
13270b57cec5SDimitry Andric 
13280b57cec5SDimitry Andric   if (isa<NullStmt>(Body))
13290b57cec5SDimitry Andric     getCurCompoundScope().setHasEmptyLoopBodies();
13300b57cec5SDimitry Andric 
13310b57cec5SDimitry Andric   return WhileStmt::Create(Context, CondVal.first, CondVal.second, Body,
13320b57cec5SDimitry Andric                            WhileLoc);
13330b57cec5SDimitry Andric }
13340b57cec5SDimitry Andric 
13350b57cec5SDimitry Andric StmtResult
13360b57cec5SDimitry Andric Sema::ActOnDoStmt(SourceLocation DoLoc, Stmt *Body,
13370b57cec5SDimitry Andric                   SourceLocation WhileLoc, SourceLocation CondLParen,
13380b57cec5SDimitry Andric                   Expr *Cond, SourceLocation CondRParen) {
13390b57cec5SDimitry Andric   assert(Cond && "ActOnDoStmt(): missing expression");
13400b57cec5SDimitry Andric 
13410b57cec5SDimitry Andric   CheckBreakContinueBinding(Cond);
13420b57cec5SDimitry Andric   ExprResult CondResult = CheckBooleanCondition(DoLoc, Cond);
13430b57cec5SDimitry Andric   if (CondResult.isInvalid())
13440b57cec5SDimitry Andric     return StmtError();
13450b57cec5SDimitry Andric   Cond = CondResult.get();
13460b57cec5SDimitry Andric 
13470b57cec5SDimitry Andric   CondResult = ActOnFinishFullExpr(Cond, DoLoc, /*DiscardedValue*/ false);
13480b57cec5SDimitry Andric   if (CondResult.isInvalid())
13490b57cec5SDimitry Andric     return StmtError();
13500b57cec5SDimitry Andric   Cond = CondResult.get();
13510b57cec5SDimitry Andric 
13520b57cec5SDimitry Andric   // Only call the CommaVisitor for C89 due to differences in scope flags.
13530b57cec5SDimitry Andric   if (Cond && !getLangOpts().C99 && !getLangOpts().CPlusPlus &&
13540b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator, Cond->getExprLoc()))
13550b57cec5SDimitry Andric     CommaVisitor(*this).Visit(Cond);
13560b57cec5SDimitry Andric 
13570b57cec5SDimitry Andric   return new (Context) DoStmt(Body, Cond, DoLoc, WhileLoc, CondRParen);
13580b57cec5SDimitry Andric }
13590b57cec5SDimitry Andric 
13600b57cec5SDimitry Andric namespace {
13610b57cec5SDimitry Andric   // Use SetVector since the diagnostic cares about the ordering of the Decl's.
13620b57cec5SDimitry Andric   using DeclSetVector =
13630b57cec5SDimitry Andric       llvm::SetVector<VarDecl *, llvm::SmallVector<VarDecl *, 8>,
13640b57cec5SDimitry Andric                       llvm::SmallPtrSet<VarDecl *, 8>>;
13650b57cec5SDimitry Andric 
13660b57cec5SDimitry Andric   // This visitor will traverse a conditional statement and store all
13670b57cec5SDimitry Andric   // the evaluated decls into a vector.  Simple is set to true if none
13680b57cec5SDimitry Andric   // of the excluded constructs are used.
13690b57cec5SDimitry Andric   class DeclExtractor : public EvaluatedExprVisitor<DeclExtractor> {
13700b57cec5SDimitry Andric     DeclSetVector &Decls;
13710b57cec5SDimitry Andric     SmallVectorImpl<SourceRange> &Ranges;
13720b57cec5SDimitry Andric     bool Simple;
13730b57cec5SDimitry Andric   public:
13740b57cec5SDimitry Andric     typedef EvaluatedExprVisitor<DeclExtractor> Inherited;
13750b57cec5SDimitry Andric 
13760b57cec5SDimitry Andric     DeclExtractor(Sema &S, DeclSetVector &Decls,
13770b57cec5SDimitry Andric                   SmallVectorImpl<SourceRange> &Ranges) :
13780b57cec5SDimitry Andric         Inherited(S.Context),
13790b57cec5SDimitry Andric         Decls(Decls),
13800b57cec5SDimitry Andric         Ranges(Ranges),
13810b57cec5SDimitry Andric         Simple(true) {}
13820b57cec5SDimitry Andric 
13830b57cec5SDimitry Andric     bool isSimple() { return Simple; }
13840b57cec5SDimitry Andric 
13850b57cec5SDimitry Andric     // Replaces the method in EvaluatedExprVisitor.
13860b57cec5SDimitry Andric     void VisitMemberExpr(MemberExpr* E) {
13870b57cec5SDimitry Andric       Simple = false;
13880b57cec5SDimitry Andric     }
13890b57cec5SDimitry Andric 
13900b57cec5SDimitry Andric     // Any Stmt not whitelisted will cause the condition to be marked complex.
13910b57cec5SDimitry Andric     void VisitStmt(Stmt *S) {
13920b57cec5SDimitry Andric       Simple = false;
13930b57cec5SDimitry Andric     }
13940b57cec5SDimitry Andric 
13950b57cec5SDimitry Andric     void VisitBinaryOperator(BinaryOperator *E) {
13960b57cec5SDimitry Andric       Visit(E->getLHS());
13970b57cec5SDimitry Andric       Visit(E->getRHS());
13980b57cec5SDimitry Andric     }
13990b57cec5SDimitry Andric 
14000b57cec5SDimitry Andric     void VisitCastExpr(CastExpr *E) {
14010b57cec5SDimitry Andric       Visit(E->getSubExpr());
14020b57cec5SDimitry Andric     }
14030b57cec5SDimitry Andric 
14040b57cec5SDimitry Andric     void VisitUnaryOperator(UnaryOperator *E) {
14050b57cec5SDimitry Andric       // Skip checking conditionals with derefernces.
14060b57cec5SDimitry Andric       if (E->getOpcode() == UO_Deref)
14070b57cec5SDimitry Andric         Simple = false;
14080b57cec5SDimitry Andric       else
14090b57cec5SDimitry Andric         Visit(E->getSubExpr());
14100b57cec5SDimitry Andric     }
14110b57cec5SDimitry Andric 
14120b57cec5SDimitry Andric     void VisitConditionalOperator(ConditionalOperator *E) {
14130b57cec5SDimitry Andric       Visit(E->getCond());
14140b57cec5SDimitry Andric       Visit(E->getTrueExpr());
14150b57cec5SDimitry Andric       Visit(E->getFalseExpr());
14160b57cec5SDimitry Andric     }
14170b57cec5SDimitry Andric 
14180b57cec5SDimitry Andric     void VisitParenExpr(ParenExpr *E) {
14190b57cec5SDimitry Andric       Visit(E->getSubExpr());
14200b57cec5SDimitry Andric     }
14210b57cec5SDimitry Andric 
14220b57cec5SDimitry Andric     void VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
14230b57cec5SDimitry Andric       Visit(E->getOpaqueValue()->getSourceExpr());
14240b57cec5SDimitry Andric       Visit(E->getFalseExpr());
14250b57cec5SDimitry Andric     }
14260b57cec5SDimitry Andric 
14270b57cec5SDimitry Andric     void VisitIntegerLiteral(IntegerLiteral *E) { }
14280b57cec5SDimitry Andric     void VisitFloatingLiteral(FloatingLiteral *E) { }
14290b57cec5SDimitry Andric     void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) { }
14300b57cec5SDimitry Andric     void VisitCharacterLiteral(CharacterLiteral *E) { }
14310b57cec5SDimitry Andric     void VisitGNUNullExpr(GNUNullExpr *E) { }
14320b57cec5SDimitry Andric     void VisitImaginaryLiteral(ImaginaryLiteral *E) { }
14330b57cec5SDimitry Andric 
14340b57cec5SDimitry Andric     void VisitDeclRefExpr(DeclRefExpr *E) {
14350b57cec5SDimitry Andric       VarDecl *VD = dyn_cast<VarDecl>(E->getDecl());
14360b57cec5SDimitry Andric       if (!VD) {
14370b57cec5SDimitry Andric         // Don't allow unhandled Decl types.
14380b57cec5SDimitry Andric         Simple = false;
14390b57cec5SDimitry Andric         return;
14400b57cec5SDimitry Andric       }
14410b57cec5SDimitry Andric 
14420b57cec5SDimitry Andric       Ranges.push_back(E->getSourceRange());
14430b57cec5SDimitry Andric 
14440b57cec5SDimitry Andric       Decls.insert(VD);
14450b57cec5SDimitry Andric     }
14460b57cec5SDimitry Andric 
14470b57cec5SDimitry Andric   }; // end class DeclExtractor
14480b57cec5SDimitry Andric 
14490b57cec5SDimitry Andric   // DeclMatcher checks to see if the decls are used in a non-evaluated
14500b57cec5SDimitry Andric   // context.
14510b57cec5SDimitry Andric   class DeclMatcher : public EvaluatedExprVisitor<DeclMatcher> {
14520b57cec5SDimitry Andric     DeclSetVector &Decls;
14530b57cec5SDimitry Andric     bool FoundDecl;
14540b57cec5SDimitry Andric 
14550b57cec5SDimitry Andric   public:
14560b57cec5SDimitry Andric     typedef EvaluatedExprVisitor<DeclMatcher> Inherited;
14570b57cec5SDimitry Andric 
14580b57cec5SDimitry Andric     DeclMatcher(Sema &S, DeclSetVector &Decls, Stmt *Statement) :
14590b57cec5SDimitry Andric         Inherited(S.Context), Decls(Decls), FoundDecl(false) {
14600b57cec5SDimitry Andric       if (!Statement) return;
14610b57cec5SDimitry Andric 
14620b57cec5SDimitry Andric       Visit(Statement);
14630b57cec5SDimitry Andric     }
14640b57cec5SDimitry Andric 
14650b57cec5SDimitry Andric     void VisitReturnStmt(ReturnStmt *S) {
14660b57cec5SDimitry Andric       FoundDecl = true;
14670b57cec5SDimitry Andric     }
14680b57cec5SDimitry Andric 
14690b57cec5SDimitry Andric     void VisitBreakStmt(BreakStmt *S) {
14700b57cec5SDimitry Andric       FoundDecl = true;
14710b57cec5SDimitry Andric     }
14720b57cec5SDimitry Andric 
14730b57cec5SDimitry Andric     void VisitGotoStmt(GotoStmt *S) {
14740b57cec5SDimitry Andric       FoundDecl = true;
14750b57cec5SDimitry Andric     }
14760b57cec5SDimitry Andric 
14770b57cec5SDimitry Andric     void VisitCastExpr(CastExpr *E) {
14780b57cec5SDimitry Andric       if (E->getCastKind() == CK_LValueToRValue)
14790b57cec5SDimitry Andric         CheckLValueToRValueCast(E->getSubExpr());
14800b57cec5SDimitry Andric       else
14810b57cec5SDimitry Andric         Visit(E->getSubExpr());
14820b57cec5SDimitry Andric     }
14830b57cec5SDimitry Andric 
14840b57cec5SDimitry Andric     void CheckLValueToRValueCast(Expr *E) {
14850b57cec5SDimitry Andric       E = E->IgnoreParenImpCasts();
14860b57cec5SDimitry Andric 
14870b57cec5SDimitry Andric       if (isa<DeclRefExpr>(E)) {
14880b57cec5SDimitry Andric         return;
14890b57cec5SDimitry Andric       }
14900b57cec5SDimitry Andric 
14910b57cec5SDimitry Andric       if (ConditionalOperator *CO = dyn_cast<ConditionalOperator>(E)) {
14920b57cec5SDimitry Andric         Visit(CO->getCond());
14930b57cec5SDimitry Andric         CheckLValueToRValueCast(CO->getTrueExpr());
14940b57cec5SDimitry Andric         CheckLValueToRValueCast(CO->getFalseExpr());
14950b57cec5SDimitry Andric         return;
14960b57cec5SDimitry Andric       }
14970b57cec5SDimitry Andric 
14980b57cec5SDimitry Andric       if (BinaryConditionalOperator *BCO =
14990b57cec5SDimitry Andric               dyn_cast<BinaryConditionalOperator>(E)) {
15000b57cec5SDimitry Andric         CheckLValueToRValueCast(BCO->getOpaqueValue()->getSourceExpr());
15010b57cec5SDimitry Andric         CheckLValueToRValueCast(BCO->getFalseExpr());
15020b57cec5SDimitry Andric         return;
15030b57cec5SDimitry Andric       }
15040b57cec5SDimitry Andric 
15050b57cec5SDimitry Andric       Visit(E);
15060b57cec5SDimitry Andric     }
15070b57cec5SDimitry Andric 
15080b57cec5SDimitry Andric     void VisitDeclRefExpr(DeclRefExpr *E) {
15090b57cec5SDimitry Andric       if (VarDecl *VD = dyn_cast<VarDecl>(E->getDecl()))
15100b57cec5SDimitry Andric         if (Decls.count(VD))
15110b57cec5SDimitry Andric           FoundDecl = true;
15120b57cec5SDimitry Andric     }
15130b57cec5SDimitry Andric 
15140b57cec5SDimitry Andric     void VisitPseudoObjectExpr(PseudoObjectExpr *POE) {
15150b57cec5SDimitry Andric       // Only need to visit the semantics for POE.
15160b57cec5SDimitry Andric       // SyntaticForm doesn't really use the Decal.
15170b57cec5SDimitry Andric       for (auto *S : POE->semantics()) {
15180b57cec5SDimitry Andric         if (auto *OVE = dyn_cast<OpaqueValueExpr>(S))
15190b57cec5SDimitry Andric           // Look past the OVE into the expression it binds.
15200b57cec5SDimitry Andric           Visit(OVE->getSourceExpr());
15210b57cec5SDimitry Andric         else
15220b57cec5SDimitry Andric           Visit(S);
15230b57cec5SDimitry Andric       }
15240b57cec5SDimitry Andric     }
15250b57cec5SDimitry Andric 
15260b57cec5SDimitry Andric     bool FoundDeclInUse() { return FoundDecl; }
15270b57cec5SDimitry Andric 
15280b57cec5SDimitry Andric   };  // end class DeclMatcher
15290b57cec5SDimitry Andric 
15300b57cec5SDimitry Andric   void CheckForLoopConditionalStatement(Sema &S, Expr *Second,
15310b57cec5SDimitry Andric                                         Expr *Third, Stmt *Body) {
15320b57cec5SDimitry Andric     // Condition is empty
15330b57cec5SDimitry Andric     if (!Second) return;
15340b57cec5SDimitry Andric 
15350b57cec5SDimitry Andric     if (S.Diags.isIgnored(diag::warn_variables_not_in_loop_body,
15360b57cec5SDimitry Andric                           Second->getBeginLoc()))
15370b57cec5SDimitry Andric       return;
15380b57cec5SDimitry Andric 
15390b57cec5SDimitry Andric     PartialDiagnostic PDiag = S.PDiag(diag::warn_variables_not_in_loop_body);
15400b57cec5SDimitry Andric     DeclSetVector Decls;
15410b57cec5SDimitry Andric     SmallVector<SourceRange, 10> Ranges;
15420b57cec5SDimitry Andric     DeclExtractor DE(S, Decls, Ranges);
15430b57cec5SDimitry Andric     DE.Visit(Second);
15440b57cec5SDimitry Andric 
15450b57cec5SDimitry Andric     // Don't analyze complex conditionals.
15460b57cec5SDimitry Andric     if (!DE.isSimple()) return;
15470b57cec5SDimitry Andric 
15480b57cec5SDimitry Andric     // No decls found.
15490b57cec5SDimitry Andric     if (Decls.size() == 0) return;
15500b57cec5SDimitry Andric 
15510b57cec5SDimitry Andric     // Don't warn on volatile, static, or global variables.
15520b57cec5SDimitry Andric     for (auto *VD : Decls)
15530b57cec5SDimitry Andric       if (VD->getType().isVolatileQualified() || VD->hasGlobalStorage())
15540b57cec5SDimitry Andric         return;
15550b57cec5SDimitry Andric 
15560b57cec5SDimitry Andric     if (DeclMatcher(S, Decls, Second).FoundDeclInUse() ||
15570b57cec5SDimitry Andric         DeclMatcher(S, Decls, Third).FoundDeclInUse() ||
15580b57cec5SDimitry Andric         DeclMatcher(S, Decls, Body).FoundDeclInUse())
15590b57cec5SDimitry Andric       return;
15600b57cec5SDimitry Andric 
15610b57cec5SDimitry Andric     // Load decl names into diagnostic.
15620b57cec5SDimitry Andric     if (Decls.size() > 4) {
15630b57cec5SDimitry Andric       PDiag << 0;
15640b57cec5SDimitry Andric     } else {
15650b57cec5SDimitry Andric       PDiag << (unsigned)Decls.size();
15660b57cec5SDimitry Andric       for (auto *VD : Decls)
15670b57cec5SDimitry Andric         PDiag << VD->getDeclName();
15680b57cec5SDimitry Andric     }
15690b57cec5SDimitry Andric 
15700b57cec5SDimitry Andric     for (auto Range : Ranges)
15710b57cec5SDimitry Andric       PDiag << Range;
15720b57cec5SDimitry Andric 
15730b57cec5SDimitry Andric     S.Diag(Ranges.begin()->getBegin(), PDiag);
15740b57cec5SDimitry Andric   }
15750b57cec5SDimitry Andric 
15760b57cec5SDimitry Andric   // If Statement is an incemement or decrement, return true and sets the
15770b57cec5SDimitry Andric   // variables Increment and DRE.
15780b57cec5SDimitry Andric   bool ProcessIterationStmt(Sema &S, Stmt* Statement, bool &Increment,
15790b57cec5SDimitry Andric                             DeclRefExpr *&DRE) {
15800b57cec5SDimitry Andric     if (auto Cleanups = dyn_cast<ExprWithCleanups>(Statement))
15810b57cec5SDimitry Andric       if (!Cleanups->cleanupsHaveSideEffects())
15820b57cec5SDimitry Andric         Statement = Cleanups->getSubExpr();
15830b57cec5SDimitry Andric 
15840b57cec5SDimitry Andric     if (UnaryOperator *UO = dyn_cast<UnaryOperator>(Statement)) {
15850b57cec5SDimitry Andric       switch (UO->getOpcode()) {
15860b57cec5SDimitry Andric         default: return false;
15870b57cec5SDimitry Andric         case UO_PostInc:
15880b57cec5SDimitry Andric         case UO_PreInc:
15890b57cec5SDimitry Andric           Increment = true;
15900b57cec5SDimitry Andric           break;
15910b57cec5SDimitry Andric         case UO_PostDec:
15920b57cec5SDimitry Andric         case UO_PreDec:
15930b57cec5SDimitry Andric           Increment = false;
15940b57cec5SDimitry Andric           break;
15950b57cec5SDimitry Andric       }
15960b57cec5SDimitry Andric       DRE = dyn_cast<DeclRefExpr>(UO->getSubExpr());
15970b57cec5SDimitry Andric       return DRE;
15980b57cec5SDimitry Andric     }
15990b57cec5SDimitry Andric 
16000b57cec5SDimitry Andric     if (CXXOperatorCallExpr *Call = dyn_cast<CXXOperatorCallExpr>(Statement)) {
16010b57cec5SDimitry Andric       FunctionDecl *FD = Call->getDirectCallee();
16020b57cec5SDimitry Andric       if (!FD || !FD->isOverloadedOperator()) return false;
16030b57cec5SDimitry Andric       switch (FD->getOverloadedOperator()) {
16040b57cec5SDimitry Andric         default: return false;
16050b57cec5SDimitry Andric         case OO_PlusPlus:
16060b57cec5SDimitry Andric           Increment = true;
16070b57cec5SDimitry Andric           break;
16080b57cec5SDimitry Andric         case OO_MinusMinus:
16090b57cec5SDimitry Andric           Increment = false;
16100b57cec5SDimitry Andric           break;
16110b57cec5SDimitry Andric       }
16120b57cec5SDimitry Andric       DRE = dyn_cast<DeclRefExpr>(Call->getArg(0));
16130b57cec5SDimitry Andric       return DRE;
16140b57cec5SDimitry Andric     }
16150b57cec5SDimitry Andric 
16160b57cec5SDimitry Andric     return false;
16170b57cec5SDimitry Andric   }
16180b57cec5SDimitry Andric 
16190b57cec5SDimitry Andric   // A visitor to determine if a continue or break statement is a
16200b57cec5SDimitry Andric   // subexpression.
16210b57cec5SDimitry Andric   class BreakContinueFinder : public ConstEvaluatedExprVisitor<BreakContinueFinder> {
16220b57cec5SDimitry Andric     SourceLocation BreakLoc;
16230b57cec5SDimitry Andric     SourceLocation ContinueLoc;
16240b57cec5SDimitry Andric     bool InSwitch = false;
16250b57cec5SDimitry Andric 
16260b57cec5SDimitry Andric   public:
16270b57cec5SDimitry Andric     BreakContinueFinder(Sema &S, const Stmt* Body) :
16280b57cec5SDimitry Andric         Inherited(S.Context) {
16290b57cec5SDimitry Andric       Visit(Body);
16300b57cec5SDimitry Andric     }
16310b57cec5SDimitry Andric 
16320b57cec5SDimitry Andric     typedef ConstEvaluatedExprVisitor<BreakContinueFinder> Inherited;
16330b57cec5SDimitry Andric 
16340b57cec5SDimitry Andric     void VisitContinueStmt(const ContinueStmt* E) {
16350b57cec5SDimitry Andric       ContinueLoc = E->getContinueLoc();
16360b57cec5SDimitry Andric     }
16370b57cec5SDimitry Andric 
16380b57cec5SDimitry Andric     void VisitBreakStmt(const BreakStmt* E) {
16390b57cec5SDimitry Andric       if (!InSwitch)
16400b57cec5SDimitry Andric         BreakLoc = E->getBreakLoc();
16410b57cec5SDimitry Andric     }
16420b57cec5SDimitry Andric 
16430b57cec5SDimitry Andric     void VisitSwitchStmt(const SwitchStmt* S) {
16440b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
16450b57cec5SDimitry Andric         Visit(Init);
16460b57cec5SDimitry Andric       if (const Stmt *CondVar = S->getConditionVariableDeclStmt())
16470b57cec5SDimitry Andric         Visit(CondVar);
16480b57cec5SDimitry Andric       if (const Stmt *Cond = S->getCond())
16490b57cec5SDimitry Andric         Visit(Cond);
16500b57cec5SDimitry Andric 
16510b57cec5SDimitry Andric       // Don't return break statements from the body of a switch.
16520b57cec5SDimitry Andric       InSwitch = true;
16530b57cec5SDimitry Andric       if (const Stmt *Body = S->getBody())
16540b57cec5SDimitry Andric         Visit(Body);
16550b57cec5SDimitry Andric       InSwitch = false;
16560b57cec5SDimitry Andric     }
16570b57cec5SDimitry Andric 
16580b57cec5SDimitry Andric     void VisitForStmt(const ForStmt *S) {
16590b57cec5SDimitry Andric       // Only visit the init statement of a for loop; the body
16600b57cec5SDimitry Andric       // has a different break/continue scope.
16610b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
16620b57cec5SDimitry Andric         Visit(Init);
16630b57cec5SDimitry Andric     }
16640b57cec5SDimitry Andric 
16650b57cec5SDimitry Andric     void VisitWhileStmt(const WhileStmt *) {
16660b57cec5SDimitry Andric       // Do nothing; the children of a while loop have a different
16670b57cec5SDimitry Andric       // break/continue scope.
16680b57cec5SDimitry Andric     }
16690b57cec5SDimitry Andric 
16700b57cec5SDimitry Andric     void VisitDoStmt(const DoStmt *) {
16710b57cec5SDimitry Andric       // Do nothing; the children of a while loop have a different
16720b57cec5SDimitry Andric       // break/continue scope.
16730b57cec5SDimitry Andric     }
16740b57cec5SDimitry Andric 
16750b57cec5SDimitry Andric     void VisitCXXForRangeStmt(const CXXForRangeStmt *S) {
16760b57cec5SDimitry Andric       // Only visit the initialization of a for loop; the body
16770b57cec5SDimitry Andric       // has a different break/continue scope.
16780b57cec5SDimitry Andric       if (const Stmt *Init = S->getInit())
16790b57cec5SDimitry Andric         Visit(Init);
16800b57cec5SDimitry Andric       if (const Stmt *Range = S->getRangeStmt())
16810b57cec5SDimitry Andric         Visit(Range);
16820b57cec5SDimitry Andric       if (const Stmt *Begin = S->getBeginStmt())
16830b57cec5SDimitry Andric         Visit(Begin);
16840b57cec5SDimitry Andric       if (const Stmt *End = S->getEndStmt())
16850b57cec5SDimitry Andric         Visit(End);
16860b57cec5SDimitry Andric     }
16870b57cec5SDimitry Andric 
16880b57cec5SDimitry Andric     void VisitObjCForCollectionStmt(const ObjCForCollectionStmt *S) {
16890b57cec5SDimitry Andric       // Only visit the initialization of a for loop; the body
16900b57cec5SDimitry Andric       // has a different break/continue scope.
16910b57cec5SDimitry Andric       if (const Stmt *Element = S->getElement())
16920b57cec5SDimitry Andric         Visit(Element);
16930b57cec5SDimitry Andric       if (const Stmt *Collection = S->getCollection())
16940b57cec5SDimitry Andric         Visit(Collection);
16950b57cec5SDimitry Andric     }
16960b57cec5SDimitry Andric 
16970b57cec5SDimitry Andric     bool ContinueFound() { return ContinueLoc.isValid(); }
16980b57cec5SDimitry Andric     bool BreakFound() { return BreakLoc.isValid(); }
16990b57cec5SDimitry Andric     SourceLocation GetContinueLoc() { return ContinueLoc; }
17000b57cec5SDimitry Andric     SourceLocation GetBreakLoc() { return BreakLoc; }
17010b57cec5SDimitry Andric 
17020b57cec5SDimitry Andric   };  // end class BreakContinueFinder
17030b57cec5SDimitry Andric 
17040b57cec5SDimitry Andric   // Emit a warning when a loop increment/decrement appears twice per loop
17050b57cec5SDimitry Andric   // iteration.  The conditions which trigger this warning are:
17060b57cec5SDimitry Andric   // 1) The last statement in the loop body and the third expression in the
17070b57cec5SDimitry Andric   //    for loop are both increment or both decrement of the same variable
17080b57cec5SDimitry Andric   // 2) No continue statements in the loop body.
17090b57cec5SDimitry Andric   void CheckForRedundantIteration(Sema &S, Expr *Third, Stmt *Body) {
17100b57cec5SDimitry Andric     // Return when there is nothing to check.
17110b57cec5SDimitry Andric     if (!Body || !Third) return;
17120b57cec5SDimitry Andric 
17130b57cec5SDimitry Andric     if (S.Diags.isIgnored(diag::warn_redundant_loop_iteration,
17140b57cec5SDimitry Andric                           Third->getBeginLoc()))
17150b57cec5SDimitry Andric       return;
17160b57cec5SDimitry Andric 
17170b57cec5SDimitry Andric     // Get the last statement from the loop body.
17180b57cec5SDimitry Andric     CompoundStmt *CS = dyn_cast<CompoundStmt>(Body);
17190b57cec5SDimitry Andric     if (!CS || CS->body_empty()) return;
17200b57cec5SDimitry Andric     Stmt *LastStmt = CS->body_back();
17210b57cec5SDimitry Andric     if (!LastStmt) return;
17220b57cec5SDimitry Andric 
17230b57cec5SDimitry Andric     bool LoopIncrement, LastIncrement;
17240b57cec5SDimitry Andric     DeclRefExpr *LoopDRE, *LastDRE;
17250b57cec5SDimitry Andric 
17260b57cec5SDimitry Andric     if (!ProcessIterationStmt(S, Third, LoopIncrement, LoopDRE)) return;
17270b57cec5SDimitry Andric     if (!ProcessIterationStmt(S, LastStmt, LastIncrement, LastDRE)) return;
17280b57cec5SDimitry Andric 
17290b57cec5SDimitry Andric     // Check that the two statements are both increments or both decrements
17300b57cec5SDimitry Andric     // on the same variable.
17310b57cec5SDimitry Andric     if (LoopIncrement != LastIncrement ||
17320b57cec5SDimitry Andric         LoopDRE->getDecl() != LastDRE->getDecl()) return;
17330b57cec5SDimitry Andric 
17340b57cec5SDimitry Andric     if (BreakContinueFinder(S, Body).ContinueFound()) return;
17350b57cec5SDimitry Andric 
17360b57cec5SDimitry Andric     S.Diag(LastDRE->getLocation(), diag::warn_redundant_loop_iteration)
17370b57cec5SDimitry Andric          << LastDRE->getDecl() << LastIncrement;
17380b57cec5SDimitry Andric     S.Diag(LoopDRE->getLocation(), diag::note_loop_iteration_here)
17390b57cec5SDimitry Andric          << LoopIncrement;
17400b57cec5SDimitry Andric   }
17410b57cec5SDimitry Andric 
17420b57cec5SDimitry Andric } // end namespace
17430b57cec5SDimitry Andric 
17440b57cec5SDimitry Andric 
17450b57cec5SDimitry Andric void Sema::CheckBreakContinueBinding(Expr *E) {
17460b57cec5SDimitry Andric   if (!E || getLangOpts().CPlusPlus)
17470b57cec5SDimitry Andric     return;
17480b57cec5SDimitry Andric   BreakContinueFinder BCFinder(*this, E);
17490b57cec5SDimitry Andric   Scope *BreakParent = CurScope->getBreakParent();
17500b57cec5SDimitry Andric   if (BCFinder.BreakFound() && BreakParent) {
17510b57cec5SDimitry Andric     if (BreakParent->getFlags() & Scope::SwitchScope) {
17520b57cec5SDimitry Andric       Diag(BCFinder.GetBreakLoc(), diag::warn_break_binds_to_switch);
17530b57cec5SDimitry Andric     } else {
17540b57cec5SDimitry Andric       Diag(BCFinder.GetBreakLoc(), diag::warn_loop_ctrl_binds_to_inner)
17550b57cec5SDimitry Andric           << "break";
17560b57cec5SDimitry Andric     }
17570b57cec5SDimitry Andric   } else if (BCFinder.ContinueFound() && CurScope->getContinueParent()) {
17580b57cec5SDimitry Andric     Diag(BCFinder.GetContinueLoc(), diag::warn_loop_ctrl_binds_to_inner)
17590b57cec5SDimitry Andric         << "continue";
17600b57cec5SDimitry Andric   }
17610b57cec5SDimitry Andric }
17620b57cec5SDimitry Andric 
17630b57cec5SDimitry Andric StmtResult Sema::ActOnForStmt(SourceLocation ForLoc, SourceLocation LParenLoc,
17640b57cec5SDimitry Andric                               Stmt *First, ConditionResult Second,
17650b57cec5SDimitry Andric                               FullExprArg third, SourceLocation RParenLoc,
17660b57cec5SDimitry Andric                               Stmt *Body) {
17670b57cec5SDimitry Andric   if (Second.isInvalid())
17680b57cec5SDimitry Andric     return StmtError();
17690b57cec5SDimitry Andric 
17700b57cec5SDimitry Andric   if (!getLangOpts().CPlusPlus) {
17710b57cec5SDimitry Andric     if (DeclStmt *DS = dyn_cast_or_null<DeclStmt>(First)) {
17720b57cec5SDimitry Andric       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
17730b57cec5SDimitry Andric       // declare identifiers for objects having storage class 'auto' or
17740b57cec5SDimitry Andric       // 'register'.
17750b57cec5SDimitry Andric       for (auto *DI : DS->decls()) {
17760b57cec5SDimitry Andric         VarDecl *VD = dyn_cast<VarDecl>(DI);
17770b57cec5SDimitry Andric         if (VD && VD->isLocalVarDecl() && !VD->hasLocalStorage())
17780b57cec5SDimitry Andric           VD = nullptr;
17790b57cec5SDimitry Andric         if (!VD) {
17800b57cec5SDimitry Andric           Diag(DI->getLocation(), diag::err_non_local_variable_decl_in_for);
17810b57cec5SDimitry Andric           DI->setInvalidDecl();
17820b57cec5SDimitry Andric         }
17830b57cec5SDimitry Andric       }
17840b57cec5SDimitry Andric     }
17850b57cec5SDimitry Andric   }
17860b57cec5SDimitry Andric 
17870b57cec5SDimitry Andric   CheckBreakContinueBinding(Second.get().second);
17880b57cec5SDimitry Andric   CheckBreakContinueBinding(third.get());
17890b57cec5SDimitry Andric 
17900b57cec5SDimitry Andric   if (!Second.get().first)
17910b57cec5SDimitry Andric     CheckForLoopConditionalStatement(*this, Second.get().second, third.get(),
17920b57cec5SDimitry Andric                                      Body);
17930b57cec5SDimitry Andric   CheckForRedundantIteration(*this, third.get(), Body);
17940b57cec5SDimitry Andric 
17950b57cec5SDimitry Andric   if (Second.get().second &&
17960b57cec5SDimitry Andric       !Diags.isIgnored(diag::warn_comma_operator,
17970b57cec5SDimitry Andric                        Second.get().second->getExprLoc()))
17980b57cec5SDimitry Andric     CommaVisitor(*this).Visit(Second.get().second);
17990b57cec5SDimitry Andric 
18000b57cec5SDimitry Andric   Expr *Third  = third.release().getAs<Expr>();
18010b57cec5SDimitry Andric   if (isa<NullStmt>(Body))
18020b57cec5SDimitry Andric     getCurCompoundScope().setHasEmptyLoopBodies();
18030b57cec5SDimitry Andric 
18040b57cec5SDimitry Andric   return new (Context)
18050b57cec5SDimitry Andric       ForStmt(Context, First, Second.get().second, Second.get().first, Third,
18060b57cec5SDimitry Andric               Body, ForLoc, LParenLoc, RParenLoc);
18070b57cec5SDimitry Andric }
18080b57cec5SDimitry Andric 
18090b57cec5SDimitry Andric /// In an Objective C collection iteration statement:
18100b57cec5SDimitry Andric ///   for (x in y)
18110b57cec5SDimitry Andric /// x can be an arbitrary l-value expression.  Bind it up as a
18120b57cec5SDimitry Andric /// full-expression.
18130b57cec5SDimitry Andric StmtResult Sema::ActOnForEachLValueExpr(Expr *E) {
18140b57cec5SDimitry Andric   // Reduce placeholder expressions here.  Note that this rejects the
18150b57cec5SDimitry Andric   // use of pseudo-object l-values in this position.
18160b57cec5SDimitry Andric   ExprResult result = CheckPlaceholderExpr(E);
18170b57cec5SDimitry Andric   if (result.isInvalid()) return StmtError();
18180b57cec5SDimitry Andric   E = result.get();
18190b57cec5SDimitry Andric 
18200b57cec5SDimitry Andric   ExprResult FullExpr = ActOnFinishFullExpr(E, /*DiscardedValue*/ false);
18210b57cec5SDimitry Andric   if (FullExpr.isInvalid())
18220b57cec5SDimitry Andric     return StmtError();
18230b57cec5SDimitry Andric   return StmtResult(static_cast<Stmt*>(FullExpr.get()));
18240b57cec5SDimitry Andric }
18250b57cec5SDimitry Andric 
18260b57cec5SDimitry Andric ExprResult
18270b57cec5SDimitry Andric Sema::CheckObjCForCollectionOperand(SourceLocation forLoc, Expr *collection) {
18280b57cec5SDimitry Andric   if (!collection)
18290b57cec5SDimitry Andric     return ExprError();
18300b57cec5SDimitry Andric 
18310b57cec5SDimitry Andric   ExprResult result = CorrectDelayedTyposInExpr(collection);
18320b57cec5SDimitry Andric   if (!result.isUsable())
18330b57cec5SDimitry Andric     return ExprError();
18340b57cec5SDimitry Andric   collection = result.get();
18350b57cec5SDimitry Andric 
18360b57cec5SDimitry Andric   // Bail out early if we've got a type-dependent expression.
18370b57cec5SDimitry Andric   if (collection->isTypeDependent()) return collection;
18380b57cec5SDimitry Andric 
18390b57cec5SDimitry Andric   // Perform normal l-value conversion.
18400b57cec5SDimitry Andric   result = DefaultFunctionArrayLvalueConversion(collection);
18410b57cec5SDimitry Andric   if (result.isInvalid())
18420b57cec5SDimitry Andric     return ExprError();
18430b57cec5SDimitry Andric   collection = result.get();
18440b57cec5SDimitry Andric 
18450b57cec5SDimitry Andric   // The operand needs to have object-pointer type.
18460b57cec5SDimitry Andric   // TODO: should we do a contextual conversion?
18470b57cec5SDimitry Andric   const ObjCObjectPointerType *pointerType =
18480b57cec5SDimitry Andric     collection->getType()->getAs<ObjCObjectPointerType>();
18490b57cec5SDimitry Andric   if (!pointerType)
18500b57cec5SDimitry Andric     return Diag(forLoc, diag::err_collection_expr_type)
18510b57cec5SDimitry Andric              << collection->getType() << collection->getSourceRange();
18520b57cec5SDimitry Andric 
18530b57cec5SDimitry Andric   // Check that the operand provides
18540b57cec5SDimitry Andric   //   - countByEnumeratingWithState:objects:count:
18550b57cec5SDimitry Andric   const ObjCObjectType *objectType = pointerType->getObjectType();
18560b57cec5SDimitry Andric   ObjCInterfaceDecl *iface = objectType->getInterface();
18570b57cec5SDimitry Andric 
18580b57cec5SDimitry Andric   // If we have a forward-declared type, we can't do this check.
18590b57cec5SDimitry Andric   // Under ARC, it is an error not to have a forward-declared class.
18600b57cec5SDimitry Andric   if (iface &&
18610b57cec5SDimitry Andric       (getLangOpts().ObjCAutoRefCount
18620b57cec5SDimitry Andric            ? RequireCompleteType(forLoc, QualType(objectType, 0),
18630b57cec5SDimitry Andric                                  diag::err_arc_collection_forward, collection)
18640b57cec5SDimitry Andric            : !isCompleteType(forLoc, QualType(objectType, 0)))) {
18650b57cec5SDimitry Andric     // Otherwise, if we have any useful type information, check that
18660b57cec5SDimitry Andric     // the type declares the appropriate method.
18670b57cec5SDimitry Andric   } else if (iface || !objectType->qual_empty()) {
18680b57cec5SDimitry Andric     IdentifierInfo *selectorIdents[] = {
18690b57cec5SDimitry Andric       &Context.Idents.get("countByEnumeratingWithState"),
18700b57cec5SDimitry Andric       &Context.Idents.get("objects"),
18710b57cec5SDimitry Andric       &Context.Idents.get("count")
18720b57cec5SDimitry Andric     };
18730b57cec5SDimitry Andric     Selector selector = Context.Selectors.getSelector(3, &selectorIdents[0]);
18740b57cec5SDimitry Andric 
18750b57cec5SDimitry Andric     ObjCMethodDecl *method = nullptr;
18760b57cec5SDimitry Andric 
18770b57cec5SDimitry Andric     // If there's an interface, look in both the public and private APIs.
18780b57cec5SDimitry Andric     if (iface) {
18790b57cec5SDimitry Andric       method = iface->lookupInstanceMethod(selector);
18800b57cec5SDimitry Andric       if (!method) method = iface->lookupPrivateMethod(selector);
18810b57cec5SDimitry Andric     }
18820b57cec5SDimitry Andric 
18830b57cec5SDimitry Andric     // Also check protocol qualifiers.
18840b57cec5SDimitry Andric     if (!method)
18850b57cec5SDimitry Andric       method = LookupMethodInQualifiedType(selector, pointerType,
18860b57cec5SDimitry Andric                                            /*instance*/ true);
18870b57cec5SDimitry Andric 
18880b57cec5SDimitry Andric     // If we didn't find it anywhere, give up.
18890b57cec5SDimitry Andric     if (!method) {
18900b57cec5SDimitry Andric       Diag(forLoc, diag::warn_collection_expr_type)
18910b57cec5SDimitry Andric         << collection->getType() << selector << collection->getSourceRange();
18920b57cec5SDimitry Andric     }
18930b57cec5SDimitry Andric 
18940b57cec5SDimitry Andric     // TODO: check for an incompatible signature?
18950b57cec5SDimitry Andric   }
18960b57cec5SDimitry Andric 
18970b57cec5SDimitry Andric   // Wrap up any cleanups in the expression.
18980b57cec5SDimitry Andric   return collection;
18990b57cec5SDimitry Andric }
19000b57cec5SDimitry Andric 
19010b57cec5SDimitry Andric StmtResult
19020b57cec5SDimitry Andric Sema::ActOnObjCForCollectionStmt(SourceLocation ForLoc,
19030b57cec5SDimitry Andric                                  Stmt *First, Expr *collection,
19040b57cec5SDimitry Andric                                  SourceLocation RParenLoc) {
19050b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
19060b57cec5SDimitry Andric 
19070b57cec5SDimitry Andric   ExprResult CollectionExprResult =
19080b57cec5SDimitry Andric     CheckObjCForCollectionOperand(ForLoc, collection);
19090b57cec5SDimitry Andric 
19100b57cec5SDimitry Andric   if (First) {
19110b57cec5SDimitry Andric     QualType FirstType;
19120b57cec5SDimitry Andric     if (DeclStmt *DS = dyn_cast<DeclStmt>(First)) {
19130b57cec5SDimitry Andric       if (!DS->isSingleDecl())
19140b57cec5SDimitry Andric         return StmtError(Diag((*DS->decl_begin())->getLocation(),
19150b57cec5SDimitry Andric                          diag::err_toomany_element_decls));
19160b57cec5SDimitry Andric 
19170b57cec5SDimitry Andric       VarDecl *D = dyn_cast<VarDecl>(DS->getSingleDecl());
19180b57cec5SDimitry Andric       if (!D || D->isInvalidDecl())
19190b57cec5SDimitry Andric         return StmtError();
19200b57cec5SDimitry Andric 
19210b57cec5SDimitry Andric       FirstType = D->getType();
19220b57cec5SDimitry Andric       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
19230b57cec5SDimitry Andric       // declare identifiers for objects having storage class 'auto' or
19240b57cec5SDimitry Andric       // 'register'.
19250b57cec5SDimitry Andric       if (!D->hasLocalStorage())
19260b57cec5SDimitry Andric         return StmtError(Diag(D->getLocation(),
19270b57cec5SDimitry Andric                               diag::err_non_local_variable_decl_in_for));
19280b57cec5SDimitry Andric 
19290b57cec5SDimitry Andric       // If the type contained 'auto', deduce the 'auto' to 'id'.
19300b57cec5SDimitry Andric       if (FirstType->getContainedAutoType()) {
19310b57cec5SDimitry Andric         OpaqueValueExpr OpaqueId(D->getLocation(), Context.getObjCIdType(),
19320b57cec5SDimitry Andric                                  VK_RValue);
19330b57cec5SDimitry Andric         Expr *DeducedInit = &OpaqueId;
19340b57cec5SDimitry Andric         if (DeduceAutoType(D->getTypeSourceInfo(), DeducedInit, FirstType) ==
19350b57cec5SDimitry Andric                 DAR_Failed)
19360b57cec5SDimitry Andric           DiagnoseAutoDeductionFailure(D, DeducedInit);
19370b57cec5SDimitry Andric         if (FirstType.isNull()) {
19380b57cec5SDimitry Andric           D->setInvalidDecl();
19390b57cec5SDimitry Andric           return StmtError();
19400b57cec5SDimitry Andric         }
19410b57cec5SDimitry Andric 
19420b57cec5SDimitry Andric         D->setType(FirstType);
19430b57cec5SDimitry Andric 
19440b57cec5SDimitry Andric         if (!inTemplateInstantiation()) {
19450b57cec5SDimitry Andric           SourceLocation Loc =
19460b57cec5SDimitry Andric               D->getTypeSourceInfo()->getTypeLoc().getBeginLoc();
19470b57cec5SDimitry Andric           Diag(Loc, diag::warn_auto_var_is_id)
19480b57cec5SDimitry Andric             << D->getDeclName();
19490b57cec5SDimitry Andric         }
19500b57cec5SDimitry Andric       }
19510b57cec5SDimitry Andric 
19520b57cec5SDimitry Andric     } else {
19530b57cec5SDimitry Andric       Expr *FirstE = cast<Expr>(First);
19540b57cec5SDimitry Andric       if (!FirstE->isTypeDependent() && !FirstE->isLValue())
19550b57cec5SDimitry Andric         return StmtError(
19560b57cec5SDimitry Andric             Diag(First->getBeginLoc(), diag::err_selector_element_not_lvalue)
19570b57cec5SDimitry Andric             << First->getSourceRange());
19580b57cec5SDimitry Andric 
19590b57cec5SDimitry Andric       FirstType = static_cast<Expr*>(First)->getType();
19600b57cec5SDimitry Andric       if (FirstType.isConstQualified())
19610b57cec5SDimitry Andric         Diag(ForLoc, diag::err_selector_element_const_type)
19620b57cec5SDimitry Andric           << FirstType << First->getSourceRange();
19630b57cec5SDimitry Andric     }
19640b57cec5SDimitry Andric     if (!FirstType->isDependentType() &&
19650b57cec5SDimitry Andric         !FirstType->isObjCObjectPointerType() &&
19660b57cec5SDimitry Andric         !FirstType->isBlockPointerType())
19670b57cec5SDimitry Andric         return StmtError(Diag(ForLoc, diag::err_selector_element_type)
19680b57cec5SDimitry Andric                            << FirstType << First->getSourceRange());
19690b57cec5SDimitry Andric   }
19700b57cec5SDimitry Andric 
19710b57cec5SDimitry Andric   if (CollectionExprResult.isInvalid())
19720b57cec5SDimitry Andric     return StmtError();
19730b57cec5SDimitry Andric 
19740b57cec5SDimitry Andric   CollectionExprResult =
19750b57cec5SDimitry Andric       ActOnFinishFullExpr(CollectionExprResult.get(), /*DiscardedValue*/ false);
19760b57cec5SDimitry Andric   if (CollectionExprResult.isInvalid())
19770b57cec5SDimitry Andric     return StmtError();
19780b57cec5SDimitry Andric 
19790b57cec5SDimitry Andric   return new (Context) ObjCForCollectionStmt(First, CollectionExprResult.get(),
19800b57cec5SDimitry Andric                                              nullptr, ForLoc, RParenLoc);
19810b57cec5SDimitry Andric }
19820b57cec5SDimitry Andric 
19830b57cec5SDimitry Andric /// Finish building a variable declaration for a for-range statement.
19840b57cec5SDimitry Andric /// \return true if an error occurs.
19850b57cec5SDimitry Andric static bool FinishForRangeVarDecl(Sema &SemaRef, VarDecl *Decl, Expr *Init,
19860b57cec5SDimitry Andric                                   SourceLocation Loc, int DiagID) {
19870b57cec5SDimitry Andric   if (Decl->getType()->isUndeducedType()) {
19880b57cec5SDimitry Andric     ExprResult Res = SemaRef.CorrectDelayedTyposInExpr(Init);
19890b57cec5SDimitry Andric     if (!Res.isUsable()) {
19900b57cec5SDimitry Andric       Decl->setInvalidDecl();
19910b57cec5SDimitry Andric       return true;
19920b57cec5SDimitry Andric     }
19930b57cec5SDimitry Andric     Init = Res.get();
19940b57cec5SDimitry Andric   }
19950b57cec5SDimitry Andric 
19960b57cec5SDimitry Andric   // Deduce the type for the iterator variable now rather than leaving it to
19970b57cec5SDimitry Andric   // AddInitializerToDecl, so we can produce a more suitable diagnostic.
19980b57cec5SDimitry Andric   QualType InitType;
19990b57cec5SDimitry Andric   if ((!isa<InitListExpr>(Init) && Init->getType()->isVoidType()) ||
20000b57cec5SDimitry Andric       SemaRef.DeduceAutoType(Decl->getTypeSourceInfo(), Init, InitType) ==
20010b57cec5SDimitry Andric           Sema::DAR_Failed)
20020b57cec5SDimitry Andric     SemaRef.Diag(Loc, DiagID) << Init->getType();
20030b57cec5SDimitry Andric   if (InitType.isNull()) {
20040b57cec5SDimitry Andric     Decl->setInvalidDecl();
20050b57cec5SDimitry Andric     return true;
20060b57cec5SDimitry Andric   }
20070b57cec5SDimitry Andric   Decl->setType(InitType);
20080b57cec5SDimitry Andric 
20090b57cec5SDimitry Andric   // In ARC, infer lifetime.
20100b57cec5SDimitry Andric   // FIXME: ARC may want to turn this into 'const __unsafe_unretained' if
20110b57cec5SDimitry Andric   // we're doing the equivalent of fast iteration.
20120b57cec5SDimitry Andric   if (SemaRef.getLangOpts().ObjCAutoRefCount &&
20130b57cec5SDimitry Andric       SemaRef.inferObjCARCLifetime(Decl))
20140b57cec5SDimitry Andric     Decl->setInvalidDecl();
20150b57cec5SDimitry Andric 
20160b57cec5SDimitry Andric   SemaRef.AddInitializerToDecl(Decl, Init, /*DirectInit=*/false);
20170b57cec5SDimitry Andric   SemaRef.FinalizeDeclaration(Decl);
20180b57cec5SDimitry Andric   SemaRef.CurContext->addHiddenDecl(Decl);
20190b57cec5SDimitry Andric   return false;
20200b57cec5SDimitry Andric }
20210b57cec5SDimitry Andric 
20220b57cec5SDimitry Andric namespace {
20230b57cec5SDimitry Andric // An enum to represent whether something is dealing with a call to begin()
20240b57cec5SDimitry Andric // or a call to end() in a range-based for loop.
20250b57cec5SDimitry Andric enum BeginEndFunction {
20260b57cec5SDimitry Andric   BEF_begin,
20270b57cec5SDimitry Andric   BEF_end
20280b57cec5SDimitry Andric };
20290b57cec5SDimitry Andric 
20300b57cec5SDimitry Andric /// Produce a note indicating which begin/end function was implicitly called
20310b57cec5SDimitry Andric /// by a C++11 for-range statement. This is often not obvious from the code,
20320b57cec5SDimitry Andric /// nor from the diagnostics produced when analysing the implicit expressions
20330b57cec5SDimitry Andric /// required in a for-range statement.
20340b57cec5SDimitry Andric void NoteForRangeBeginEndFunction(Sema &SemaRef, Expr *E,
20350b57cec5SDimitry Andric                                   BeginEndFunction BEF) {
20360b57cec5SDimitry Andric   CallExpr *CE = dyn_cast<CallExpr>(E);
20370b57cec5SDimitry Andric   if (!CE)
20380b57cec5SDimitry Andric     return;
20390b57cec5SDimitry Andric   FunctionDecl *D = dyn_cast<FunctionDecl>(CE->getCalleeDecl());
20400b57cec5SDimitry Andric   if (!D)
20410b57cec5SDimitry Andric     return;
20420b57cec5SDimitry Andric   SourceLocation Loc = D->getLocation();
20430b57cec5SDimitry Andric 
20440b57cec5SDimitry Andric   std::string Description;
20450b57cec5SDimitry Andric   bool IsTemplate = false;
20460b57cec5SDimitry Andric   if (FunctionTemplateDecl *FunTmpl = D->getPrimaryTemplate()) {
20470b57cec5SDimitry Andric     Description = SemaRef.getTemplateArgumentBindingsText(
20480b57cec5SDimitry Andric       FunTmpl->getTemplateParameters(), *D->getTemplateSpecializationArgs());
20490b57cec5SDimitry Andric     IsTemplate = true;
20500b57cec5SDimitry Andric   }
20510b57cec5SDimitry Andric 
20520b57cec5SDimitry Andric   SemaRef.Diag(Loc, diag::note_for_range_begin_end)
20530b57cec5SDimitry Andric     << BEF << IsTemplate << Description << E->getType();
20540b57cec5SDimitry Andric }
20550b57cec5SDimitry Andric 
20560b57cec5SDimitry Andric /// Build a variable declaration for a for-range statement.
20570b57cec5SDimitry Andric VarDecl *BuildForRangeVarDecl(Sema &SemaRef, SourceLocation Loc,
20580b57cec5SDimitry Andric                               QualType Type, StringRef Name) {
20590b57cec5SDimitry Andric   DeclContext *DC = SemaRef.CurContext;
20600b57cec5SDimitry Andric   IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name);
20610b57cec5SDimitry Andric   TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc);
20620b57cec5SDimitry Andric   VarDecl *Decl = VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type,
20630b57cec5SDimitry Andric                                   TInfo, SC_None);
20640b57cec5SDimitry Andric   Decl->setImplicit();
20650b57cec5SDimitry Andric   return Decl;
20660b57cec5SDimitry Andric }
20670b57cec5SDimitry Andric 
20680b57cec5SDimitry Andric }
20690b57cec5SDimitry Andric 
20700b57cec5SDimitry Andric static bool ObjCEnumerationCollection(Expr *Collection) {
20710b57cec5SDimitry Andric   return !Collection->isTypeDependent()
20720b57cec5SDimitry Andric           && Collection->getType()->getAs<ObjCObjectPointerType>() != nullptr;
20730b57cec5SDimitry Andric }
20740b57cec5SDimitry Andric 
20750b57cec5SDimitry Andric /// ActOnCXXForRangeStmt - Check and build a C++11 for-range statement.
20760b57cec5SDimitry Andric ///
20770b57cec5SDimitry Andric /// C++11 [stmt.ranged]:
20780b57cec5SDimitry Andric ///   A range-based for statement is equivalent to
20790b57cec5SDimitry Andric ///
20800b57cec5SDimitry Andric ///   {
20810b57cec5SDimitry Andric ///     auto && __range = range-init;
20820b57cec5SDimitry Andric ///     for ( auto __begin = begin-expr,
20830b57cec5SDimitry Andric ///           __end = end-expr;
20840b57cec5SDimitry Andric ///           __begin != __end;
20850b57cec5SDimitry Andric ///           ++__begin ) {
20860b57cec5SDimitry Andric ///       for-range-declaration = *__begin;
20870b57cec5SDimitry Andric ///       statement
20880b57cec5SDimitry Andric ///     }
20890b57cec5SDimitry Andric ///   }
20900b57cec5SDimitry Andric ///
20910b57cec5SDimitry Andric /// The body of the loop is not available yet, since it cannot be analysed until
20920b57cec5SDimitry Andric /// we have determined the type of the for-range-declaration.
20930b57cec5SDimitry Andric StmtResult Sema::ActOnCXXForRangeStmt(Scope *S, SourceLocation ForLoc,
20940b57cec5SDimitry Andric                                       SourceLocation CoawaitLoc, Stmt *InitStmt,
20950b57cec5SDimitry Andric                                       Stmt *First, SourceLocation ColonLoc,
20960b57cec5SDimitry Andric                                       Expr *Range, SourceLocation RParenLoc,
20970b57cec5SDimitry Andric                                       BuildForRangeKind Kind) {
20980b57cec5SDimitry Andric   if (!First)
20990b57cec5SDimitry Andric     return StmtError();
21000b57cec5SDimitry Andric 
21010b57cec5SDimitry Andric   if (Range && ObjCEnumerationCollection(Range)) {
21020b57cec5SDimitry Andric     // FIXME: Support init-statements in Objective-C++20 ranged for statement.
21030b57cec5SDimitry Andric     if (InitStmt)
21040b57cec5SDimitry Andric       return Diag(InitStmt->getBeginLoc(), diag::err_objc_for_range_init_stmt)
21050b57cec5SDimitry Andric                  << InitStmt->getSourceRange();
21060b57cec5SDimitry Andric     return ActOnObjCForCollectionStmt(ForLoc, First, Range, RParenLoc);
21070b57cec5SDimitry Andric   }
21080b57cec5SDimitry Andric 
21090b57cec5SDimitry Andric   DeclStmt *DS = dyn_cast<DeclStmt>(First);
21100b57cec5SDimitry Andric   assert(DS && "first part of for range not a decl stmt");
21110b57cec5SDimitry Andric 
21120b57cec5SDimitry Andric   if (!DS->isSingleDecl()) {
21130b57cec5SDimitry Andric     Diag(DS->getBeginLoc(), diag::err_type_defined_in_for_range);
21140b57cec5SDimitry Andric     return StmtError();
21150b57cec5SDimitry Andric   }
21160b57cec5SDimitry Andric 
21170b57cec5SDimitry Andric   Decl *LoopVar = DS->getSingleDecl();
21180b57cec5SDimitry Andric   if (LoopVar->isInvalidDecl() || !Range ||
21190b57cec5SDimitry Andric       DiagnoseUnexpandedParameterPack(Range, UPPC_Expression)) {
21200b57cec5SDimitry Andric     LoopVar->setInvalidDecl();
21210b57cec5SDimitry Andric     return StmtError();
21220b57cec5SDimitry Andric   }
21230b57cec5SDimitry Andric 
21240b57cec5SDimitry Andric   // Build the coroutine state immediately and not later during template
21250b57cec5SDimitry Andric   // instantiation
21260b57cec5SDimitry Andric   if (!CoawaitLoc.isInvalid()) {
21270b57cec5SDimitry Andric     if (!ActOnCoroutineBodyStart(S, CoawaitLoc, "co_await"))
21280b57cec5SDimitry Andric       return StmtError();
21290b57cec5SDimitry Andric   }
21300b57cec5SDimitry Andric 
21310b57cec5SDimitry Andric   // Build  auto && __range = range-init
21320b57cec5SDimitry Andric   // Divide by 2, since the variables are in the inner scope (loop body).
21330b57cec5SDimitry Andric   const auto DepthStr = std::to_string(S->getDepth() / 2);
21340b57cec5SDimitry Andric   SourceLocation RangeLoc = Range->getBeginLoc();
21350b57cec5SDimitry Andric   VarDecl *RangeVar = BuildForRangeVarDecl(*this, RangeLoc,
21360b57cec5SDimitry Andric                                            Context.getAutoRRefDeductType(),
21370b57cec5SDimitry Andric                                            std::string("__range") + DepthStr);
21380b57cec5SDimitry Andric   if (FinishForRangeVarDecl(*this, RangeVar, Range, RangeLoc,
21390b57cec5SDimitry Andric                             diag::err_for_range_deduction_failure)) {
21400b57cec5SDimitry Andric     LoopVar->setInvalidDecl();
21410b57cec5SDimitry Andric     return StmtError();
21420b57cec5SDimitry Andric   }
21430b57cec5SDimitry Andric 
21440b57cec5SDimitry Andric   // Claim the type doesn't contain auto: we've already done the checking.
21450b57cec5SDimitry Andric   DeclGroupPtrTy RangeGroup =
21460b57cec5SDimitry Andric       BuildDeclaratorGroup(MutableArrayRef<Decl *>((Decl **)&RangeVar, 1));
21470b57cec5SDimitry Andric   StmtResult RangeDecl = ActOnDeclStmt(RangeGroup, RangeLoc, RangeLoc);
21480b57cec5SDimitry Andric   if (RangeDecl.isInvalid()) {
21490b57cec5SDimitry Andric     LoopVar->setInvalidDecl();
21500b57cec5SDimitry Andric     return StmtError();
21510b57cec5SDimitry Andric   }
21520b57cec5SDimitry Andric 
21530b57cec5SDimitry Andric   return BuildCXXForRangeStmt(
21540b57cec5SDimitry Andric       ForLoc, CoawaitLoc, InitStmt, ColonLoc, RangeDecl.get(),
21550b57cec5SDimitry Andric       /*BeginStmt=*/nullptr, /*EndStmt=*/nullptr,
21560b57cec5SDimitry Andric       /*Cond=*/nullptr, /*Inc=*/nullptr, DS, RParenLoc, Kind);
21570b57cec5SDimitry Andric }
21580b57cec5SDimitry Andric 
21590b57cec5SDimitry Andric /// Create the initialization, compare, and increment steps for
21600b57cec5SDimitry Andric /// the range-based for loop expression.
21610b57cec5SDimitry Andric /// This function does not handle array-based for loops,
21620b57cec5SDimitry Andric /// which are created in Sema::BuildCXXForRangeStmt.
21630b57cec5SDimitry Andric ///
21640b57cec5SDimitry Andric /// \returns a ForRangeStatus indicating success or what kind of error occurred.
21650b57cec5SDimitry Andric /// BeginExpr and EndExpr are set and FRS_Success is returned on success;
21660b57cec5SDimitry Andric /// CandidateSet and BEF are set and some non-success value is returned on
21670b57cec5SDimitry Andric /// failure.
21680b57cec5SDimitry Andric static Sema::ForRangeStatus
21690b57cec5SDimitry Andric BuildNonArrayForRange(Sema &SemaRef, Expr *BeginRange, Expr *EndRange,
21700b57cec5SDimitry Andric                       QualType RangeType, VarDecl *BeginVar, VarDecl *EndVar,
21710b57cec5SDimitry Andric                       SourceLocation ColonLoc, SourceLocation CoawaitLoc,
21720b57cec5SDimitry Andric                       OverloadCandidateSet *CandidateSet, ExprResult *BeginExpr,
21730b57cec5SDimitry Andric                       ExprResult *EndExpr, BeginEndFunction *BEF) {
21740b57cec5SDimitry Andric   DeclarationNameInfo BeginNameInfo(
21750b57cec5SDimitry Andric       &SemaRef.PP.getIdentifierTable().get("begin"), ColonLoc);
21760b57cec5SDimitry Andric   DeclarationNameInfo EndNameInfo(&SemaRef.PP.getIdentifierTable().get("end"),
21770b57cec5SDimitry Andric                                   ColonLoc);
21780b57cec5SDimitry Andric 
21790b57cec5SDimitry Andric   LookupResult BeginMemberLookup(SemaRef, BeginNameInfo,
21800b57cec5SDimitry Andric                                  Sema::LookupMemberName);
21810b57cec5SDimitry Andric   LookupResult EndMemberLookup(SemaRef, EndNameInfo, Sema::LookupMemberName);
21820b57cec5SDimitry Andric 
21830b57cec5SDimitry Andric   auto BuildBegin = [&] {
21840b57cec5SDimitry Andric     *BEF = BEF_begin;
21850b57cec5SDimitry Andric     Sema::ForRangeStatus RangeStatus =
21860b57cec5SDimitry Andric         SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, BeginNameInfo,
21870b57cec5SDimitry Andric                                           BeginMemberLookup, CandidateSet,
21880b57cec5SDimitry Andric                                           BeginRange, BeginExpr);
21890b57cec5SDimitry Andric 
21900b57cec5SDimitry Andric     if (RangeStatus != Sema::FRS_Success) {
21910b57cec5SDimitry Andric       if (RangeStatus == Sema::FRS_DiagnosticIssued)
21920b57cec5SDimitry Andric         SemaRef.Diag(BeginRange->getBeginLoc(), diag::note_in_for_range)
21930b57cec5SDimitry Andric             << ColonLoc << BEF_begin << BeginRange->getType();
21940b57cec5SDimitry Andric       return RangeStatus;
21950b57cec5SDimitry Andric     }
21960b57cec5SDimitry Andric     if (!CoawaitLoc.isInvalid()) {
21970b57cec5SDimitry Andric       // FIXME: getCurScope() should not be used during template instantiation.
21980b57cec5SDimitry Andric       // We should pick up the set of unqualified lookup results for operator
21990b57cec5SDimitry Andric       // co_await during the initial parse.
22000b57cec5SDimitry Andric       *BeginExpr = SemaRef.ActOnCoawaitExpr(SemaRef.getCurScope(), ColonLoc,
22010b57cec5SDimitry Andric                                             BeginExpr->get());
22020b57cec5SDimitry Andric       if (BeginExpr->isInvalid())
22030b57cec5SDimitry Andric         return Sema::FRS_DiagnosticIssued;
22040b57cec5SDimitry Andric     }
22050b57cec5SDimitry Andric     if (FinishForRangeVarDecl(SemaRef, BeginVar, BeginExpr->get(), ColonLoc,
22060b57cec5SDimitry Andric                               diag::err_for_range_iter_deduction_failure)) {
22070b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(SemaRef, BeginExpr->get(), *BEF);
22080b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
22090b57cec5SDimitry Andric     }
22100b57cec5SDimitry Andric     return Sema::FRS_Success;
22110b57cec5SDimitry Andric   };
22120b57cec5SDimitry Andric 
22130b57cec5SDimitry Andric   auto BuildEnd = [&] {
22140b57cec5SDimitry Andric     *BEF = BEF_end;
22150b57cec5SDimitry Andric     Sema::ForRangeStatus RangeStatus =
22160b57cec5SDimitry Andric         SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, EndNameInfo,
22170b57cec5SDimitry Andric                                           EndMemberLookup, CandidateSet,
22180b57cec5SDimitry Andric                                           EndRange, EndExpr);
22190b57cec5SDimitry Andric     if (RangeStatus != Sema::FRS_Success) {
22200b57cec5SDimitry Andric       if (RangeStatus == Sema::FRS_DiagnosticIssued)
22210b57cec5SDimitry Andric         SemaRef.Diag(EndRange->getBeginLoc(), diag::note_in_for_range)
22220b57cec5SDimitry Andric             << ColonLoc << BEF_end << EndRange->getType();
22230b57cec5SDimitry Andric       return RangeStatus;
22240b57cec5SDimitry Andric     }
22250b57cec5SDimitry Andric     if (FinishForRangeVarDecl(SemaRef, EndVar, EndExpr->get(), ColonLoc,
22260b57cec5SDimitry Andric                               diag::err_for_range_iter_deduction_failure)) {
22270b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(SemaRef, EndExpr->get(), *BEF);
22280b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
22290b57cec5SDimitry Andric     }
22300b57cec5SDimitry Andric     return Sema::FRS_Success;
22310b57cec5SDimitry Andric   };
22320b57cec5SDimitry Andric 
22330b57cec5SDimitry Andric   if (CXXRecordDecl *D = RangeType->getAsCXXRecordDecl()) {
22340b57cec5SDimitry Andric     // - if _RangeT is a class type, the unqualified-ids begin and end are
22350b57cec5SDimitry Andric     //   looked up in the scope of class _RangeT as if by class member access
22360b57cec5SDimitry Andric     //   lookup (3.4.5), and if either (or both) finds at least one
22370b57cec5SDimitry Andric     //   declaration, begin-expr and end-expr are __range.begin() and
22380b57cec5SDimitry Andric     //   __range.end(), respectively;
22390b57cec5SDimitry Andric     SemaRef.LookupQualifiedName(BeginMemberLookup, D);
22400b57cec5SDimitry Andric     if (BeginMemberLookup.isAmbiguous())
22410b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
22420b57cec5SDimitry Andric 
22430b57cec5SDimitry Andric     SemaRef.LookupQualifiedName(EndMemberLookup, D);
22440b57cec5SDimitry Andric     if (EndMemberLookup.isAmbiguous())
22450b57cec5SDimitry Andric       return Sema::FRS_DiagnosticIssued;
22460b57cec5SDimitry Andric 
22470b57cec5SDimitry Andric     if (BeginMemberLookup.empty() != EndMemberLookup.empty()) {
22480b57cec5SDimitry Andric       // Look up the non-member form of the member we didn't find, first.
22490b57cec5SDimitry Andric       // This way we prefer a "no viable 'end'" diagnostic over a "i found
22500b57cec5SDimitry Andric       // a 'begin' but ignored it because there was no member 'end'"
22510b57cec5SDimitry Andric       // diagnostic.
22520b57cec5SDimitry Andric       auto BuildNonmember = [&](
22530b57cec5SDimitry Andric           BeginEndFunction BEFFound, LookupResult &Found,
22540b57cec5SDimitry Andric           llvm::function_ref<Sema::ForRangeStatus()> BuildFound,
22550b57cec5SDimitry Andric           llvm::function_ref<Sema::ForRangeStatus()> BuildNotFound) {
22560b57cec5SDimitry Andric         LookupResult OldFound = std::move(Found);
22570b57cec5SDimitry Andric         Found.clear();
22580b57cec5SDimitry Andric 
22590b57cec5SDimitry Andric         if (Sema::ForRangeStatus Result = BuildNotFound())
22600b57cec5SDimitry Andric           return Result;
22610b57cec5SDimitry Andric 
22620b57cec5SDimitry Andric         switch (BuildFound()) {
22630b57cec5SDimitry Andric         case Sema::FRS_Success:
22640b57cec5SDimitry Andric           return Sema::FRS_Success;
22650b57cec5SDimitry Andric 
22660b57cec5SDimitry Andric         case Sema::FRS_NoViableFunction:
22670b57cec5SDimitry Andric           CandidateSet->NoteCandidates(
22680b57cec5SDimitry Andric               PartialDiagnosticAt(BeginRange->getBeginLoc(),
22690b57cec5SDimitry Andric                                   SemaRef.PDiag(diag::err_for_range_invalid)
22700b57cec5SDimitry Andric                                       << BeginRange->getType() << BEFFound),
22710b57cec5SDimitry Andric               SemaRef, OCD_AllCandidates, BeginRange);
22720b57cec5SDimitry Andric           LLVM_FALLTHROUGH;
22730b57cec5SDimitry Andric 
22740b57cec5SDimitry Andric         case Sema::FRS_DiagnosticIssued:
22750b57cec5SDimitry Andric           for (NamedDecl *D : OldFound) {
22760b57cec5SDimitry Andric             SemaRef.Diag(D->getLocation(),
22770b57cec5SDimitry Andric                          diag::note_for_range_member_begin_end_ignored)
22780b57cec5SDimitry Andric                 << BeginRange->getType() << BEFFound;
22790b57cec5SDimitry Andric           }
22800b57cec5SDimitry Andric           return Sema::FRS_DiagnosticIssued;
22810b57cec5SDimitry Andric         }
22820b57cec5SDimitry Andric         llvm_unreachable("unexpected ForRangeStatus");
22830b57cec5SDimitry Andric       };
22840b57cec5SDimitry Andric       if (BeginMemberLookup.empty())
22850b57cec5SDimitry Andric         return BuildNonmember(BEF_end, EndMemberLookup, BuildEnd, BuildBegin);
22860b57cec5SDimitry Andric       return BuildNonmember(BEF_begin, BeginMemberLookup, BuildBegin, BuildEnd);
22870b57cec5SDimitry Andric     }
22880b57cec5SDimitry Andric   } else {
22890b57cec5SDimitry Andric     // - otherwise, begin-expr and end-expr are begin(__range) and
22900b57cec5SDimitry Andric     //   end(__range), respectively, where begin and end are looked up with
22910b57cec5SDimitry Andric     //   argument-dependent lookup (3.4.2). For the purposes of this name
22920b57cec5SDimitry Andric     //   lookup, namespace std is an associated namespace.
22930b57cec5SDimitry Andric   }
22940b57cec5SDimitry Andric 
22950b57cec5SDimitry Andric   if (Sema::ForRangeStatus Result = BuildBegin())
22960b57cec5SDimitry Andric     return Result;
22970b57cec5SDimitry Andric   return BuildEnd();
22980b57cec5SDimitry Andric }
22990b57cec5SDimitry Andric 
23000b57cec5SDimitry Andric /// Speculatively attempt to dereference an invalid range expression.
23010b57cec5SDimitry Andric /// If the attempt fails, this function will return a valid, null StmtResult
23020b57cec5SDimitry Andric /// and emit no diagnostics.
23030b57cec5SDimitry Andric static StmtResult RebuildForRangeWithDereference(Sema &SemaRef, Scope *S,
23040b57cec5SDimitry Andric                                                  SourceLocation ForLoc,
23050b57cec5SDimitry Andric                                                  SourceLocation CoawaitLoc,
23060b57cec5SDimitry Andric                                                  Stmt *InitStmt,
23070b57cec5SDimitry Andric                                                  Stmt *LoopVarDecl,
23080b57cec5SDimitry Andric                                                  SourceLocation ColonLoc,
23090b57cec5SDimitry Andric                                                  Expr *Range,
23100b57cec5SDimitry Andric                                                  SourceLocation RangeLoc,
23110b57cec5SDimitry Andric                                                  SourceLocation RParenLoc) {
23120b57cec5SDimitry Andric   // Determine whether we can rebuild the for-range statement with a
23130b57cec5SDimitry Andric   // dereferenced range expression.
23140b57cec5SDimitry Andric   ExprResult AdjustedRange;
23150b57cec5SDimitry Andric   {
23160b57cec5SDimitry Andric     Sema::SFINAETrap Trap(SemaRef);
23170b57cec5SDimitry Andric 
23180b57cec5SDimitry Andric     AdjustedRange = SemaRef.BuildUnaryOp(S, RangeLoc, UO_Deref, Range);
23190b57cec5SDimitry Andric     if (AdjustedRange.isInvalid())
23200b57cec5SDimitry Andric       return StmtResult();
23210b57cec5SDimitry Andric 
23220b57cec5SDimitry Andric     StmtResult SR = SemaRef.ActOnCXXForRangeStmt(
23230b57cec5SDimitry Andric         S, ForLoc, CoawaitLoc, InitStmt, LoopVarDecl, ColonLoc,
23240b57cec5SDimitry Andric         AdjustedRange.get(), RParenLoc, Sema::BFRK_Check);
23250b57cec5SDimitry Andric     if (SR.isInvalid())
23260b57cec5SDimitry Andric       return StmtResult();
23270b57cec5SDimitry Andric   }
23280b57cec5SDimitry Andric 
23290b57cec5SDimitry Andric   // The attempt to dereference worked well enough that it could produce a valid
23300b57cec5SDimitry Andric   // loop. Produce a fixit, and rebuild the loop with diagnostics enabled, in
23310b57cec5SDimitry Andric   // case there are any other (non-fatal) problems with it.
23320b57cec5SDimitry Andric   SemaRef.Diag(RangeLoc, diag::err_for_range_dereference)
23330b57cec5SDimitry Andric     << Range->getType() << FixItHint::CreateInsertion(RangeLoc, "*");
23340b57cec5SDimitry Andric   return SemaRef.ActOnCXXForRangeStmt(
23350b57cec5SDimitry Andric       S, ForLoc, CoawaitLoc, InitStmt, LoopVarDecl, ColonLoc,
23360b57cec5SDimitry Andric       AdjustedRange.get(), RParenLoc, Sema::BFRK_Rebuild);
23370b57cec5SDimitry Andric }
23380b57cec5SDimitry Andric 
23390b57cec5SDimitry Andric namespace {
23400b57cec5SDimitry Andric /// RAII object to automatically invalidate a declaration if an error occurs.
23410b57cec5SDimitry Andric struct InvalidateOnErrorScope {
23420b57cec5SDimitry Andric   InvalidateOnErrorScope(Sema &SemaRef, Decl *D, bool Enabled)
23430b57cec5SDimitry Andric       : Trap(SemaRef.Diags), D(D), Enabled(Enabled) {}
23440b57cec5SDimitry Andric   ~InvalidateOnErrorScope() {
23450b57cec5SDimitry Andric     if (Enabled && Trap.hasErrorOccurred())
23460b57cec5SDimitry Andric       D->setInvalidDecl();
23470b57cec5SDimitry Andric   }
23480b57cec5SDimitry Andric 
23490b57cec5SDimitry Andric   DiagnosticErrorTrap Trap;
23500b57cec5SDimitry Andric   Decl *D;
23510b57cec5SDimitry Andric   bool Enabled;
23520b57cec5SDimitry Andric };
23530b57cec5SDimitry Andric }
23540b57cec5SDimitry Andric 
23550b57cec5SDimitry Andric /// BuildCXXForRangeStmt - Build or instantiate a C++11 for-range statement.
23560b57cec5SDimitry Andric StmtResult Sema::BuildCXXForRangeStmt(SourceLocation ForLoc,
23570b57cec5SDimitry Andric                                       SourceLocation CoawaitLoc, Stmt *InitStmt,
23580b57cec5SDimitry Andric                                       SourceLocation ColonLoc, Stmt *RangeDecl,
23590b57cec5SDimitry Andric                                       Stmt *Begin, Stmt *End, Expr *Cond,
23600b57cec5SDimitry Andric                                       Expr *Inc, Stmt *LoopVarDecl,
23610b57cec5SDimitry Andric                                       SourceLocation RParenLoc,
23620b57cec5SDimitry Andric                                       BuildForRangeKind Kind) {
23630b57cec5SDimitry Andric   // FIXME: This should not be used during template instantiation. We should
23640b57cec5SDimitry Andric   // pick up the set of unqualified lookup results for the != and + operators
23650b57cec5SDimitry Andric   // in the initial parse.
23660b57cec5SDimitry Andric   //
23670b57cec5SDimitry Andric   // Testcase (accepts-invalid):
23680b57cec5SDimitry Andric   //   template<typename T> void f() { for (auto x : T()) {} }
23690b57cec5SDimitry Andric   //   namespace N { struct X { X begin(); X end(); int operator*(); }; }
23700b57cec5SDimitry Andric   //   bool operator!=(N::X, N::X); void operator++(N::X);
23710b57cec5SDimitry Andric   //   void g() { f<N::X>(); }
23720b57cec5SDimitry Andric   Scope *S = getCurScope();
23730b57cec5SDimitry Andric 
23740b57cec5SDimitry Andric   DeclStmt *RangeDS = cast<DeclStmt>(RangeDecl);
23750b57cec5SDimitry Andric   VarDecl *RangeVar = cast<VarDecl>(RangeDS->getSingleDecl());
23760b57cec5SDimitry Andric   QualType RangeVarType = RangeVar->getType();
23770b57cec5SDimitry Andric 
23780b57cec5SDimitry Andric   DeclStmt *LoopVarDS = cast<DeclStmt>(LoopVarDecl);
23790b57cec5SDimitry Andric   VarDecl *LoopVar = cast<VarDecl>(LoopVarDS->getSingleDecl());
23800b57cec5SDimitry Andric 
23810b57cec5SDimitry Andric   // If we hit any errors, mark the loop variable as invalid if its type
23820b57cec5SDimitry Andric   // contains 'auto'.
23830b57cec5SDimitry Andric   InvalidateOnErrorScope Invalidate(*this, LoopVar,
23840b57cec5SDimitry Andric                                     LoopVar->getType()->isUndeducedType());
23850b57cec5SDimitry Andric 
23860b57cec5SDimitry Andric   StmtResult BeginDeclStmt = Begin;
23870b57cec5SDimitry Andric   StmtResult EndDeclStmt = End;
23880b57cec5SDimitry Andric   ExprResult NotEqExpr = Cond, IncrExpr = Inc;
23890b57cec5SDimitry Andric 
23900b57cec5SDimitry Andric   if (RangeVarType->isDependentType()) {
23910b57cec5SDimitry Andric     // The range is implicitly used as a placeholder when it is dependent.
23920b57cec5SDimitry Andric     RangeVar->markUsed(Context);
23930b57cec5SDimitry Andric 
23940b57cec5SDimitry Andric     // Deduce any 'auto's in the loop variable as 'DependentTy'. We'll fill
23950b57cec5SDimitry Andric     // them in properly when we instantiate the loop.
23960b57cec5SDimitry Andric     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
23970b57cec5SDimitry Andric       if (auto *DD = dyn_cast<DecompositionDecl>(LoopVar))
23980b57cec5SDimitry Andric         for (auto *Binding : DD->bindings())
23990b57cec5SDimitry Andric           Binding->setType(Context.DependentTy);
24000b57cec5SDimitry Andric       LoopVar->setType(SubstAutoType(LoopVar->getType(), Context.DependentTy));
24010b57cec5SDimitry Andric     }
24020b57cec5SDimitry Andric   } else if (!BeginDeclStmt.get()) {
24030b57cec5SDimitry Andric     SourceLocation RangeLoc = RangeVar->getLocation();
24040b57cec5SDimitry Andric 
24050b57cec5SDimitry Andric     const QualType RangeVarNonRefType = RangeVarType.getNonReferenceType();
24060b57cec5SDimitry Andric 
24070b57cec5SDimitry Andric     ExprResult BeginRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
24080b57cec5SDimitry Andric                                                 VK_LValue, ColonLoc);
24090b57cec5SDimitry Andric     if (BeginRangeRef.isInvalid())
24100b57cec5SDimitry Andric       return StmtError();
24110b57cec5SDimitry Andric 
24120b57cec5SDimitry Andric     ExprResult EndRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
24130b57cec5SDimitry Andric                                               VK_LValue, ColonLoc);
24140b57cec5SDimitry Andric     if (EndRangeRef.isInvalid())
24150b57cec5SDimitry Andric       return StmtError();
24160b57cec5SDimitry Andric 
24170b57cec5SDimitry Andric     QualType AutoType = Context.getAutoDeductType();
24180b57cec5SDimitry Andric     Expr *Range = RangeVar->getInit();
24190b57cec5SDimitry Andric     if (!Range)
24200b57cec5SDimitry Andric       return StmtError();
24210b57cec5SDimitry Andric     QualType RangeType = Range->getType();
24220b57cec5SDimitry Andric 
24230b57cec5SDimitry Andric     if (RequireCompleteType(RangeLoc, RangeType,
24240b57cec5SDimitry Andric                             diag::err_for_range_incomplete_type))
24250b57cec5SDimitry Andric       return StmtError();
24260b57cec5SDimitry Andric 
24270b57cec5SDimitry Andric     // Build auto __begin = begin-expr, __end = end-expr.
24280b57cec5SDimitry Andric     // Divide by 2, since the variables are in the inner scope (loop body).
24290b57cec5SDimitry Andric     const auto DepthStr = std::to_string(S->getDepth() / 2);
24300b57cec5SDimitry Andric     VarDecl *BeginVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
24310b57cec5SDimitry Andric                                              std::string("__begin") + DepthStr);
24320b57cec5SDimitry Andric     VarDecl *EndVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
24330b57cec5SDimitry Andric                                            std::string("__end") + DepthStr);
24340b57cec5SDimitry Andric 
24350b57cec5SDimitry Andric     // Build begin-expr and end-expr and attach to __begin and __end variables.
24360b57cec5SDimitry Andric     ExprResult BeginExpr, EndExpr;
24370b57cec5SDimitry Andric     if (const ArrayType *UnqAT = RangeType->getAsArrayTypeUnsafe()) {
24380b57cec5SDimitry Andric       // - if _RangeT is an array type, begin-expr and end-expr are __range and
24390b57cec5SDimitry Andric       //   __range + __bound, respectively, where __bound is the array bound. If
24400b57cec5SDimitry Andric       //   _RangeT is an array of unknown size or an array of incomplete type,
24410b57cec5SDimitry Andric       //   the program is ill-formed;
24420b57cec5SDimitry Andric 
24430b57cec5SDimitry Andric       // begin-expr is __range.
24440b57cec5SDimitry Andric       BeginExpr = BeginRangeRef;
24450b57cec5SDimitry Andric       if (!CoawaitLoc.isInvalid()) {
24460b57cec5SDimitry Andric         BeginExpr = ActOnCoawaitExpr(S, ColonLoc, BeginExpr.get());
24470b57cec5SDimitry Andric         if (BeginExpr.isInvalid())
24480b57cec5SDimitry Andric           return StmtError();
24490b57cec5SDimitry Andric       }
24500b57cec5SDimitry Andric       if (FinishForRangeVarDecl(*this, BeginVar, BeginRangeRef.get(), ColonLoc,
24510b57cec5SDimitry Andric                                 diag::err_for_range_iter_deduction_failure)) {
24520b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
24530b57cec5SDimitry Andric         return StmtError();
24540b57cec5SDimitry Andric       }
24550b57cec5SDimitry Andric 
24560b57cec5SDimitry Andric       // Find the array bound.
24570b57cec5SDimitry Andric       ExprResult BoundExpr;
24580b57cec5SDimitry Andric       if (const ConstantArrayType *CAT = dyn_cast<ConstantArrayType>(UnqAT))
24590b57cec5SDimitry Andric         BoundExpr = IntegerLiteral::Create(
24600b57cec5SDimitry Andric             Context, CAT->getSize(), Context.getPointerDiffType(), RangeLoc);
24610b57cec5SDimitry Andric       else if (const VariableArrayType *VAT =
24620b57cec5SDimitry Andric                dyn_cast<VariableArrayType>(UnqAT)) {
24630b57cec5SDimitry Andric         // For a variably modified type we can't just use the expression within
24640b57cec5SDimitry Andric         // the array bounds, since we don't want that to be re-evaluated here.
24650b57cec5SDimitry Andric         // Rather, we need to determine what it was when the array was first
24660b57cec5SDimitry Andric         // created - so we resort to using sizeof(vla)/sizeof(element).
24670b57cec5SDimitry Andric         // For e.g.
24680b57cec5SDimitry Andric         //  void f(int b) {
24690b57cec5SDimitry Andric         //    int vla[b];
24700b57cec5SDimitry Andric         //    b = -1;   <-- This should not affect the num of iterations below
24710b57cec5SDimitry Andric         //    for (int &c : vla) { .. }
24720b57cec5SDimitry Andric         //  }
24730b57cec5SDimitry Andric 
24740b57cec5SDimitry Andric         // FIXME: This results in codegen generating IR that recalculates the
24750b57cec5SDimitry Andric         // run-time number of elements (as opposed to just using the IR Value
24760b57cec5SDimitry Andric         // that corresponds to the run-time value of each bound that was
24770b57cec5SDimitry Andric         // generated when the array was created.) If this proves too embarrassing
24780b57cec5SDimitry Andric         // even for unoptimized IR, consider passing a magic-value/cookie to
24790b57cec5SDimitry Andric         // codegen that then knows to simply use that initial llvm::Value (that
24800b57cec5SDimitry Andric         // corresponds to the bound at time of array creation) within
24810b57cec5SDimitry Andric         // getelementptr.  But be prepared to pay the price of increasing a
24820b57cec5SDimitry Andric         // customized form of coupling between the two components - which  could
24830b57cec5SDimitry Andric         // be hard to maintain as the codebase evolves.
24840b57cec5SDimitry Andric 
24850b57cec5SDimitry Andric         ExprResult SizeOfVLAExprR = ActOnUnaryExprOrTypeTraitExpr(
24860b57cec5SDimitry Andric             EndVar->getLocation(), UETT_SizeOf,
24870b57cec5SDimitry Andric             /*IsType=*/true,
24880b57cec5SDimitry Andric             CreateParsedType(VAT->desugar(), Context.getTrivialTypeSourceInfo(
24890b57cec5SDimitry Andric                                                  VAT->desugar(), RangeLoc))
24900b57cec5SDimitry Andric                 .getAsOpaquePtr(),
24910b57cec5SDimitry Andric             EndVar->getSourceRange());
24920b57cec5SDimitry Andric         if (SizeOfVLAExprR.isInvalid())
24930b57cec5SDimitry Andric           return StmtError();
24940b57cec5SDimitry Andric 
24950b57cec5SDimitry Andric         ExprResult SizeOfEachElementExprR = ActOnUnaryExprOrTypeTraitExpr(
24960b57cec5SDimitry Andric             EndVar->getLocation(), UETT_SizeOf,
24970b57cec5SDimitry Andric             /*IsType=*/true,
24980b57cec5SDimitry Andric             CreateParsedType(VAT->desugar(),
24990b57cec5SDimitry Andric                              Context.getTrivialTypeSourceInfo(
25000b57cec5SDimitry Andric                                  VAT->getElementType(), RangeLoc))
25010b57cec5SDimitry Andric                 .getAsOpaquePtr(),
25020b57cec5SDimitry Andric             EndVar->getSourceRange());
25030b57cec5SDimitry Andric         if (SizeOfEachElementExprR.isInvalid())
25040b57cec5SDimitry Andric           return StmtError();
25050b57cec5SDimitry Andric 
25060b57cec5SDimitry Andric         BoundExpr =
25070b57cec5SDimitry Andric             ActOnBinOp(S, EndVar->getLocation(), tok::slash,
25080b57cec5SDimitry Andric                        SizeOfVLAExprR.get(), SizeOfEachElementExprR.get());
25090b57cec5SDimitry Andric         if (BoundExpr.isInvalid())
25100b57cec5SDimitry Andric           return StmtError();
25110b57cec5SDimitry Andric 
25120b57cec5SDimitry Andric       } else {
25130b57cec5SDimitry Andric         // Can't be a DependentSizedArrayType or an IncompleteArrayType since
25140b57cec5SDimitry Andric         // UnqAT is not incomplete and Range is not type-dependent.
25150b57cec5SDimitry Andric         llvm_unreachable("Unexpected array type in for-range");
25160b57cec5SDimitry Andric       }
25170b57cec5SDimitry Andric 
25180b57cec5SDimitry Andric       // end-expr is __range + __bound.
25190b57cec5SDimitry Andric       EndExpr = ActOnBinOp(S, ColonLoc, tok::plus, EndRangeRef.get(),
25200b57cec5SDimitry Andric                            BoundExpr.get());
25210b57cec5SDimitry Andric       if (EndExpr.isInvalid())
25220b57cec5SDimitry Andric         return StmtError();
25230b57cec5SDimitry Andric       if (FinishForRangeVarDecl(*this, EndVar, EndExpr.get(), ColonLoc,
25240b57cec5SDimitry Andric                                 diag::err_for_range_iter_deduction_failure)) {
25250b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
25260b57cec5SDimitry Andric         return StmtError();
25270b57cec5SDimitry Andric       }
25280b57cec5SDimitry Andric     } else {
25290b57cec5SDimitry Andric       OverloadCandidateSet CandidateSet(RangeLoc,
25300b57cec5SDimitry Andric                                         OverloadCandidateSet::CSK_Normal);
25310b57cec5SDimitry Andric       BeginEndFunction BEFFailure;
25320b57cec5SDimitry Andric       ForRangeStatus RangeStatus = BuildNonArrayForRange(
25330b57cec5SDimitry Andric           *this, BeginRangeRef.get(), EndRangeRef.get(), RangeType, BeginVar,
25340b57cec5SDimitry Andric           EndVar, ColonLoc, CoawaitLoc, &CandidateSet, &BeginExpr, &EndExpr,
25350b57cec5SDimitry Andric           &BEFFailure);
25360b57cec5SDimitry Andric 
25370b57cec5SDimitry Andric       if (Kind == BFRK_Build && RangeStatus == FRS_NoViableFunction &&
25380b57cec5SDimitry Andric           BEFFailure == BEF_begin) {
25390b57cec5SDimitry Andric         // If the range is being built from an array parameter, emit a
25400b57cec5SDimitry Andric         // a diagnostic that it is being treated as a pointer.
25410b57cec5SDimitry Andric         if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Range)) {
25420b57cec5SDimitry Andric           if (ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
25430b57cec5SDimitry Andric             QualType ArrayTy = PVD->getOriginalType();
25440b57cec5SDimitry Andric             QualType PointerTy = PVD->getType();
25450b57cec5SDimitry Andric             if (PointerTy->isPointerType() && ArrayTy->isArrayType()) {
25460b57cec5SDimitry Andric               Diag(Range->getBeginLoc(), diag::err_range_on_array_parameter)
25470b57cec5SDimitry Andric                   << RangeLoc << PVD << ArrayTy << PointerTy;
25480b57cec5SDimitry Andric               Diag(PVD->getLocation(), diag::note_declared_at);
25490b57cec5SDimitry Andric               return StmtError();
25500b57cec5SDimitry Andric             }
25510b57cec5SDimitry Andric           }
25520b57cec5SDimitry Andric         }
25530b57cec5SDimitry Andric 
25540b57cec5SDimitry Andric         // If building the range failed, try dereferencing the range expression
25550b57cec5SDimitry Andric         // unless a diagnostic was issued or the end function is problematic.
25560b57cec5SDimitry Andric         StmtResult SR = RebuildForRangeWithDereference(*this, S, ForLoc,
25570b57cec5SDimitry Andric                                                        CoawaitLoc, InitStmt,
25580b57cec5SDimitry Andric                                                        LoopVarDecl, ColonLoc,
25590b57cec5SDimitry Andric                                                        Range, RangeLoc,
25600b57cec5SDimitry Andric                                                        RParenLoc);
25610b57cec5SDimitry Andric         if (SR.isInvalid() || SR.isUsable())
25620b57cec5SDimitry Andric           return SR;
25630b57cec5SDimitry Andric       }
25640b57cec5SDimitry Andric 
25650b57cec5SDimitry Andric       // Otherwise, emit diagnostics if we haven't already.
25660b57cec5SDimitry Andric       if (RangeStatus == FRS_NoViableFunction) {
25670b57cec5SDimitry Andric         Expr *Range = BEFFailure ? EndRangeRef.get() : BeginRangeRef.get();
25680b57cec5SDimitry Andric         CandidateSet.NoteCandidates(
25690b57cec5SDimitry Andric             PartialDiagnosticAt(Range->getBeginLoc(),
25700b57cec5SDimitry Andric                                 PDiag(diag::err_for_range_invalid)
25710b57cec5SDimitry Andric                                     << RangeLoc << Range->getType()
25720b57cec5SDimitry Andric                                     << BEFFailure),
25730b57cec5SDimitry Andric             *this, OCD_AllCandidates, Range);
25740b57cec5SDimitry Andric       }
25750b57cec5SDimitry Andric       // Return an error if no fix was discovered.
25760b57cec5SDimitry Andric       if (RangeStatus != FRS_Success)
25770b57cec5SDimitry Andric         return StmtError();
25780b57cec5SDimitry Andric     }
25790b57cec5SDimitry Andric 
25800b57cec5SDimitry Andric     assert(!BeginExpr.isInvalid() && !EndExpr.isInvalid() &&
25810b57cec5SDimitry Andric            "invalid range expression in for loop");
25820b57cec5SDimitry Andric 
25830b57cec5SDimitry Andric     // C++11 [dcl.spec.auto]p7: BeginType and EndType must be the same.
25840b57cec5SDimitry Andric     // C++1z removes this restriction.
25850b57cec5SDimitry Andric     QualType BeginType = BeginVar->getType(), EndType = EndVar->getType();
25860b57cec5SDimitry Andric     if (!Context.hasSameType(BeginType, EndType)) {
25870b57cec5SDimitry Andric       Diag(RangeLoc, getLangOpts().CPlusPlus17
25880b57cec5SDimitry Andric                          ? diag::warn_for_range_begin_end_types_differ
25890b57cec5SDimitry Andric                          : diag::ext_for_range_begin_end_types_differ)
25900b57cec5SDimitry Andric           << BeginType << EndType;
25910b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
25920b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
25930b57cec5SDimitry Andric     }
25940b57cec5SDimitry Andric 
25950b57cec5SDimitry Andric     BeginDeclStmt =
25960b57cec5SDimitry Andric         ActOnDeclStmt(ConvertDeclToDeclGroup(BeginVar), ColonLoc, ColonLoc);
25970b57cec5SDimitry Andric     EndDeclStmt =
25980b57cec5SDimitry Andric         ActOnDeclStmt(ConvertDeclToDeclGroup(EndVar), ColonLoc, ColonLoc);
25990b57cec5SDimitry Andric 
26000b57cec5SDimitry Andric     const QualType BeginRefNonRefType = BeginType.getNonReferenceType();
26010b57cec5SDimitry Andric     ExprResult BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
26020b57cec5SDimitry Andric                                            VK_LValue, ColonLoc);
26030b57cec5SDimitry Andric     if (BeginRef.isInvalid())
26040b57cec5SDimitry Andric       return StmtError();
26050b57cec5SDimitry Andric 
26060b57cec5SDimitry Andric     ExprResult EndRef = BuildDeclRefExpr(EndVar, EndType.getNonReferenceType(),
26070b57cec5SDimitry Andric                                          VK_LValue, ColonLoc);
26080b57cec5SDimitry Andric     if (EndRef.isInvalid())
26090b57cec5SDimitry Andric       return StmtError();
26100b57cec5SDimitry Andric 
26110b57cec5SDimitry Andric     // Build and check __begin != __end expression.
26120b57cec5SDimitry Andric     NotEqExpr = ActOnBinOp(S, ColonLoc, tok::exclaimequal,
26130b57cec5SDimitry Andric                            BeginRef.get(), EndRef.get());
26140b57cec5SDimitry Andric     if (!NotEqExpr.isInvalid())
26150b57cec5SDimitry Andric       NotEqExpr = CheckBooleanCondition(ColonLoc, NotEqExpr.get());
26160b57cec5SDimitry Andric     if (!NotEqExpr.isInvalid())
26170b57cec5SDimitry Andric       NotEqExpr =
26180b57cec5SDimitry Andric           ActOnFinishFullExpr(NotEqExpr.get(), /*DiscardedValue*/ false);
26190b57cec5SDimitry Andric     if (NotEqExpr.isInvalid()) {
26200b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
26210b57cec5SDimitry Andric         << RangeLoc << 0 << BeginRangeRef.get()->getType();
26220b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
26230b57cec5SDimitry Andric       if (!Context.hasSameType(BeginType, EndType))
26240b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
26250b57cec5SDimitry Andric       return StmtError();
26260b57cec5SDimitry Andric     }
26270b57cec5SDimitry Andric 
26280b57cec5SDimitry Andric     // Build and check ++__begin expression.
26290b57cec5SDimitry Andric     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
26300b57cec5SDimitry Andric                                 VK_LValue, ColonLoc);
26310b57cec5SDimitry Andric     if (BeginRef.isInvalid())
26320b57cec5SDimitry Andric       return StmtError();
26330b57cec5SDimitry Andric 
26340b57cec5SDimitry Andric     IncrExpr = ActOnUnaryOp(S, ColonLoc, tok::plusplus, BeginRef.get());
26350b57cec5SDimitry Andric     if (!IncrExpr.isInvalid() && CoawaitLoc.isValid())
26360b57cec5SDimitry Andric       // FIXME: getCurScope() should not be used during template instantiation.
26370b57cec5SDimitry Andric       // We should pick up the set of unqualified lookup results for operator
26380b57cec5SDimitry Andric       // co_await during the initial parse.
26390b57cec5SDimitry Andric       IncrExpr = ActOnCoawaitExpr(S, CoawaitLoc, IncrExpr.get());
26400b57cec5SDimitry Andric     if (!IncrExpr.isInvalid())
26410b57cec5SDimitry Andric       IncrExpr = ActOnFinishFullExpr(IncrExpr.get(), /*DiscardedValue*/ false);
26420b57cec5SDimitry Andric     if (IncrExpr.isInvalid()) {
26430b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
26440b57cec5SDimitry Andric         << RangeLoc << 2 << BeginRangeRef.get()->getType() ;
26450b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
26460b57cec5SDimitry Andric       return StmtError();
26470b57cec5SDimitry Andric     }
26480b57cec5SDimitry Andric 
26490b57cec5SDimitry Andric     // Build and check *__begin  expression.
26500b57cec5SDimitry Andric     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
26510b57cec5SDimitry Andric                                 VK_LValue, ColonLoc);
26520b57cec5SDimitry Andric     if (BeginRef.isInvalid())
26530b57cec5SDimitry Andric       return StmtError();
26540b57cec5SDimitry Andric 
26550b57cec5SDimitry Andric     ExprResult DerefExpr = ActOnUnaryOp(S, ColonLoc, tok::star, BeginRef.get());
26560b57cec5SDimitry Andric     if (DerefExpr.isInvalid()) {
26570b57cec5SDimitry Andric       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
26580b57cec5SDimitry Andric         << RangeLoc << 1 << BeginRangeRef.get()->getType();
26590b57cec5SDimitry Andric       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
26600b57cec5SDimitry Andric       return StmtError();
26610b57cec5SDimitry Andric     }
26620b57cec5SDimitry Andric 
26630b57cec5SDimitry Andric     // Attach  *__begin  as initializer for VD. Don't touch it if we're just
26640b57cec5SDimitry Andric     // trying to determine whether this would be a valid range.
26650b57cec5SDimitry Andric     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
26660b57cec5SDimitry Andric       AddInitializerToDecl(LoopVar, DerefExpr.get(), /*DirectInit=*/false);
26670b57cec5SDimitry Andric       if (LoopVar->isInvalidDecl())
26680b57cec5SDimitry Andric         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
26690b57cec5SDimitry Andric     }
26700b57cec5SDimitry Andric   }
26710b57cec5SDimitry Andric 
26720b57cec5SDimitry Andric   // Don't bother to actually allocate the result if we're just trying to
26730b57cec5SDimitry Andric   // determine whether it would be valid.
26740b57cec5SDimitry Andric   if (Kind == BFRK_Check)
26750b57cec5SDimitry Andric     return StmtResult();
26760b57cec5SDimitry Andric 
2677*a7dea167SDimitry Andric   // In OpenMP loop region loop control variable must be private. Perform
2678*a7dea167SDimitry Andric   // analysis of first part (if any).
2679*a7dea167SDimitry Andric   if (getLangOpts().OpenMP >= 50 && BeginDeclStmt.isUsable())
2680*a7dea167SDimitry Andric     ActOnOpenMPLoopInitialization(ForLoc, BeginDeclStmt.get());
2681*a7dea167SDimitry Andric 
26820b57cec5SDimitry Andric   return new (Context) CXXForRangeStmt(
26830b57cec5SDimitry Andric       InitStmt, RangeDS, cast_or_null<DeclStmt>(BeginDeclStmt.get()),
26840b57cec5SDimitry Andric       cast_or_null<DeclStmt>(EndDeclStmt.get()), NotEqExpr.get(),
26850b57cec5SDimitry Andric       IncrExpr.get(), LoopVarDS, /*Body=*/nullptr, ForLoc, CoawaitLoc,
26860b57cec5SDimitry Andric       ColonLoc, RParenLoc);
26870b57cec5SDimitry Andric }
26880b57cec5SDimitry Andric 
26890b57cec5SDimitry Andric /// FinishObjCForCollectionStmt - Attach the body to a objective-C foreach
26900b57cec5SDimitry Andric /// statement.
26910b57cec5SDimitry Andric StmtResult Sema::FinishObjCForCollectionStmt(Stmt *S, Stmt *B) {
26920b57cec5SDimitry Andric   if (!S || !B)
26930b57cec5SDimitry Andric     return StmtError();
26940b57cec5SDimitry Andric   ObjCForCollectionStmt * ForStmt = cast<ObjCForCollectionStmt>(S);
26950b57cec5SDimitry Andric 
26960b57cec5SDimitry Andric   ForStmt->setBody(B);
26970b57cec5SDimitry Andric   return S;
26980b57cec5SDimitry Andric }
26990b57cec5SDimitry Andric 
27000b57cec5SDimitry Andric // Warn when the loop variable is a const reference that creates a copy.
27010b57cec5SDimitry Andric // Suggest using the non-reference type for copies.  If a copy can be prevented
27020b57cec5SDimitry Andric // suggest the const reference type that would do so.
27030b57cec5SDimitry Andric // For instance, given "for (const &Foo : Range)", suggest
27040b57cec5SDimitry Andric // "for (const Foo : Range)" to denote a copy is made for the loop.  If
27050b57cec5SDimitry Andric // possible, also suggest "for (const &Bar : Range)" if this type prevents
27060b57cec5SDimitry Andric // the copy altogether.
27070b57cec5SDimitry Andric static void DiagnoseForRangeReferenceVariableCopies(Sema &SemaRef,
27080b57cec5SDimitry Andric                                                     const VarDecl *VD,
27090b57cec5SDimitry Andric                                                     QualType RangeInitType) {
27100b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
27110b57cec5SDimitry Andric   if (!InitExpr)
27120b57cec5SDimitry Andric     return;
27130b57cec5SDimitry Andric 
27140b57cec5SDimitry Andric   QualType VariableType = VD->getType();
27150b57cec5SDimitry Andric 
27160b57cec5SDimitry Andric   if (auto Cleanups = dyn_cast<ExprWithCleanups>(InitExpr))
27170b57cec5SDimitry Andric     if (!Cleanups->cleanupsHaveSideEffects())
27180b57cec5SDimitry Andric       InitExpr = Cleanups->getSubExpr();
27190b57cec5SDimitry Andric 
27200b57cec5SDimitry Andric   const MaterializeTemporaryExpr *MTE =
27210b57cec5SDimitry Andric       dyn_cast<MaterializeTemporaryExpr>(InitExpr);
27220b57cec5SDimitry Andric 
27230b57cec5SDimitry Andric   // No copy made.
27240b57cec5SDimitry Andric   if (!MTE)
27250b57cec5SDimitry Andric     return;
27260b57cec5SDimitry Andric 
27270b57cec5SDimitry Andric   const Expr *E = MTE->GetTemporaryExpr()->IgnoreImpCasts();
27280b57cec5SDimitry Andric 
27290b57cec5SDimitry Andric   // Searching for either UnaryOperator for dereference of a pointer or
27300b57cec5SDimitry Andric   // CXXOperatorCallExpr for handling iterators.
27310b57cec5SDimitry Andric   while (!isa<CXXOperatorCallExpr>(E) && !isa<UnaryOperator>(E)) {
27320b57cec5SDimitry Andric     if (const CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(E)) {
27330b57cec5SDimitry Andric       E = CCE->getArg(0);
27340b57cec5SDimitry Andric     } else if (const CXXMemberCallExpr *Call = dyn_cast<CXXMemberCallExpr>(E)) {
27350b57cec5SDimitry Andric       const MemberExpr *ME = cast<MemberExpr>(Call->getCallee());
27360b57cec5SDimitry Andric       E = ME->getBase();
27370b57cec5SDimitry Andric     } else {
27380b57cec5SDimitry Andric       const MaterializeTemporaryExpr *MTE = cast<MaterializeTemporaryExpr>(E);
27390b57cec5SDimitry Andric       E = MTE->GetTemporaryExpr();
27400b57cec5SDimitry Andric     }
27410b57cec5SDimitry Andric     E = E->IgnoreImpCasts();
27420b57cec5SDimitry Andric   }
27430b57cec5SDimitry Andric 
27440b57cec5SDimitry Andric   bool ReturnsReference = false;
27450b57cec5SDimitry Andric   if (isa<UnaryOperator>(E)) {
27460b57cec5SDimitry Andric     ReturnsReference = true;
27470b57cec5SDimitry Andric   } else {
27480b57cec5SDimitry Andric     const CXXOperatorCallExpr *Call = cast<CXXOperatorCallExpr>(E);
27490b57cec5SDimitry Andric     const FunctionDecl *FD = Call->getDirectCallee();
27500b57cec5SDimitry Andric     QualType ReturnType = FD->getReturnType();
27510b57cec5SDimitry Andric     ReturnsReference = ReturnType->isReferenceType();
27520b57cec5SDimitry Andric   }
27530b57cec5SDimitry Andric 
27540b57cec5SDimitry Andric   if (ReturnsReference) {
27550b57cec5SDimitry Andric     // Loop variable creates a temporary.  Suggest either to go with
27560b57cec5SDimitry Andric     // non-reference loop variable to indicate a copy is made, or
27570b57cec5SDimitry Andric     // the correct time to bind a const reference.
27580b57cec5SDimitry Andric     SemaRef.Diag(VD->getLocation(), diag::warn_for_range_const_reference_copy)
27590b57cec5SDimitry Andric         << VD << VariableType << E->getType();
27600b57cec5SDimitry Andric     QualType NonReferenceType = VariableType.getNonReferenceType();
27610b57cec5SDimitry Andric     NonReferenceType.removeLocalConst();
27620b57cec5SDimitry Andric     QualType NewReferenceType =
27630b57cec5SDimitry Andric         SemaRef.Context.getLValueReferenceType(E->getType().withConst());
27640b57cec5SDimitry Andric     SemaRef.Diag(VD->getBeginLoc(), diag::note_use_type_or_non_reference)
27650b57cec5SDimitry Andric         << NonReferenceType << NewReferenceType << VD->getSourceRange();
27660b57cec5SDimitry Andric   } else {
27670b57cec5SDimitry Andric     // The range always returns a copy, so a temporary is always created.
27680b57cec5SDimitry Andric     // Suggest removing the reference from the loop variable.
27690b57cec5SDimitry Andric     SemaRef.Diag(VD->getLocation(), diag::warn_for_range_variable_always_copy)
27700b57cec5SDimitry Andric         << VD << RangeInitType;
27710b57cec5SDimitry Andric     QualType NonReferenceType = VariableType.getNonReferenceType();
27720b57cec5SDimitry Andric     NonReferenceType.removeLocalConst();
27730b57cec5SDimitry Andric     SemaRef.Diag(VD->getBeginLoc(), diag::note_use_non_reference_type)
27740b57cec5SDimitry Andric         << NonReferenceType << VD->getSourceRange();
27750b57cec5SDimitry Andric   }
27760b57cec5SDimitry Andric }
27770b57cec5SDimitry Andric 
27780b57cec5SDimitry Andric // Warns when the loop variable can be changed to a reference type to
27790b57cec5SDimitry Andric // prevent a copy.  For instance, if given "for (const Foo x : Range)" suggest
27800b57cec5SDimitry Andric // "for (const Foo &x : Range)" if this form does not make a copy.
27810b57cec5SDimitry Andric static void DiagnoseForRangeConstVariableCopies(Sema &SemaRef,
27820b57cec5SDimitry Andric                                                 const VarDecl *VD) {
27830b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
27840b57cec5SDimitry Andric   if (!InitExpr)
27850b57cec5SDimitry Andric     return;
27860b57cec5SDimitry Andric 
27870b57cec5SDimitry Andric   QualType VariableType = VD->getType();
27880b57cec5SDimitry Andric 
27890b57cec5SDimitry Andric   if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(InitExpr)) {
27900b57cec5SDimitry Andric     if (!CE->getConstructor()->isCopyConstructor())
27910b57cec5SDimitry Andric       return;
27920b57cec5SDimitry Andric   } else if (const CastExpr *CE = dyn_cast<CastExpr>(InitExpr)) {
27930b57cec5SDimitry Andric     if (CE->getCastKind() != CK_LValueToRValue)
27940b57cec5SDimitry Andric       return;
27950b57cec5SDimitry Andric   } else {
27960b57cec5SDimitry Andric     return;
27970b57cec5SDimitry Andric   }
27980b57cec5SDimitry Andric 
27990b57cec5SDimitry Andric   // TODO: Determine a maximum size that a POD type can be before a diagnostic
28000b57cec5SDimitry Andric   // should be emitted.  Also, only ignore POD types with trivial copy
28010b57cec5SDimitry Andric   // constructors.
28020b57cec5SDimitry Andric   if (VariableType.isPODType(SemaRef.Context))
28030b57cec5SDimitry Andric     return;
28040b57cec5SDimitry Andric 
28050b57cec5SDimitry Andric   // Suggest changing from a const variable to a const reference variable
28060b57cec5SDimitry Andric   // if doing so will prevent a copy.
28070b57cec5SDimitry Andric   SemaRef.Diag(VD->getLocation(), diag::warn_for_range_copy)
28080b57cec5SDimitry Andric       << VD << VariableType << InitExpr->getType();
28090b57cec5SDimitry Andric   SemaRef.Diag(VD->getBeginLoc(), diag::note_use_reference_type)
28100b57cec5SDimitry Andric       << SemaRef.Context.getLValueReferenceType(VariableType)
28110b57cec5SDimitry Andric       << VD->getSourceRange();
28120b57cec5SDimitry Andric }
28130b57cec5SDimitry Andric 
28140b57cec5SDimitry Andric /// DiagnoseForRangeVariableCopies - Diagnose three cases and fixes for them.
28150b57cec5SDimitry Andric /// 1) for (const foo &x : foos) where foos only returns a copy.  Suggest
28160b57cec5SDimitry Andric ///    using "const foo x" to show that a copy is made
28170b57cec5SDimitry Andric /// 2) for (const bar &x : foos) where bar is a temporary initialized by bar.
28180b57cec5SDimitry Andric ///    Suggest either "const bar x" to keep the copying or "const foo& x" to
28190b57cec5SDimitry Andric ///    prevent the copy.
28200b57cec5SDimitry Andric /// 3) for (const foo x : foos) where x is constructed from a reference foo.
28210b57cec5SDimitry Andric ///    Suggest "const foo &x" to prevent the copy.
28220b57cec5SDimitry Andric static void DiagnoseForRangeVariableCopies(Sema &SemaRef,
28230b57cec5SDimitry Andric                                            const CXXForRangeStmt *ForStmt) {
28240b57cec5SDimitry Andric   if (SemaRef.Diags.isIgnored(diag::warn_for_range_const_reference_copy,
28250b57cec5SDimitry Andric                               ForStmt->getBeginLoc()) &&
28260b57cec5SDimitry Andric       SemaRef.Diags.isIgnored(diag::warn_for_range_variable_always_copy,
28270b57cec5SDimitry Andric                               ForStmt->getBeginLoc()) &&
28280b57cec5SDimitry Andric       SemaRef.Diags.isIgnored(diag::warn_for_range_copy,
28290b57cec5SDimitry Andric                               ForStmt->getBeginLoc())) {
28300b57cec5SDimitry Andric     return;
28310b57cec5SDimitry Andric   }
28320b57cec5SDimitry Andric 
28330b57cec5SDimitry Andric   const VarDecl *VD = ForStmt->getLoopVariable();
28340b57cec5SDimitry Andric   if (!VD)
28350b57cec5SDimitry Andric     return;
28360b57cec5SDimitry Andric 
28370b57cec5SDimitry Andric   QualType VariableType = VD->getType();
28380b57cec5SDimitry Andric 
28390b57cec5SDimitry Andric   if (VariableType->isIncompleteType())
28400b57cec5SDimitry Andric     return;
28410b57cec5SDimitry Andric 
28420b57cec5SDimitry Andric   const Expr *InitExpr = VD->getInit();
28430b57cec5SDimitry Andric   if (!InitExpr)
28440b57cec5SDimitry Andric     return;
28450b57cec5SDimitry Andric 
28460b57cec5SDimitry Andric   if (VariableType->isReferenceType()) {
28470b57cec5SDimitry Andric     DiagnoseForRangeReferenceVariableCopies(SemaRef, VD,
28480b57cec5SDimitry Andric                                             ForStmt->getRangeInit()->getType());
28490b57cec5SDimitry Andric   } else if (VariableType.isConstQualified()) {
28500b57cec5SDimitry Andric     DiagnoseForRangeConstVariableCopies(SemaRef, VD);
28510b57cec5SDimitry Andric   }
28520b57cec5SDimitry Andric }
28530b57cec5SDimitry Andric 
28540b57cec5SDimitry Andric /// FinishCXXForRangeStmt - Attach the body to a C++0x for-range statement.
28550b57cec5SDimitry Andric /// This is a separate step from ActOnCXXForRangeStmt because analysis of the
28560b57cec5SDimitry Andric /// body cannot be performed until after the type of the range variable is
28570b57cec5SDimitry Andric /// determined.
28580b57cec5SDimitry Andric StmtResult Sema::FinishCXXForRangeStmt(Stmt *S, Stmt *B) {
28590b57cec5SDimitry Andric   if (!S || !B)
28600b57cec5SDimitry Andric     return StmtError();
28610b57cec5SDimitry Andric 
28620b57cec5SDimitry Andric   if (isa<ObjCForCollectionStmt>(S))
28630b57cec5SDimitry Andric     return FinishObjCForCollectionStmt(S, B);
28640b57cec5SDimitry Andric 
28650b57cec5SDimitry Andric   CXXForRangeStmt *ForStmt = cast<CXXForRangeStmt>(S);
28660b57cec5SDimitry Andric   ForStmt->setBody(B);
28670b57cec5SDimitry Andric 
28680b57cec5SDimitry Andric   DiagnoseEmptyStmtBody(ForStmt->getRParenLoc(), B,
28690b57cec5SDimitry Andric                         diag::warn_empty_range_based_for_body);
28700b57cec5SDimitry Andric 
28710b57cec5SDimitry Andric   DiagnoseForRangeVariableCopies(*this, ForStmt);
28720b57cec5SDimitry Andric 
28730b57cec5SDimitry Andric   return S;
28740b57cec5SDimitry Andric }
28750b57cec5SDimitry Andric 
28760b57cec5SDimitry Andric StmtResult Sema::ActOnGotoStmt(SourceLocation GotoLoc,
28770b57cec5SDimitry Andric                                SourceLocation LabelLoc,
28780b57cec5SDimitry Andric                                LabelDecl *TheDecl) {
28790b57cec5SDimitry Andric   setFunctionHasBranchIntoScope();
28800b57cec5SDimitry Andric   TheDecl->markUsed(Context);
28810b57cec5SDimitry Andric   return new (Context) GotoStmt(TheDecl, GotoLoc, LabelLoc);
28820b57cec5SDimitry Andric }
28830b57cec5SDimitry Andric 
28840b57cec5SDimitry Andric StmtResult
28850b57cec5SDimitry Andric Sema::ActOnIndirectGotoStmt(SourceLocation GotoLoc, SourceLocation StarLoc,
28860b57cec5SDimitry Andric                             Expr *E) {
28870b57cec5SDimitry Andric   // Convert operand to void*
28880b57cec5SDimitry Andric   if (!E->isTypeDependent()) {
28890b57cec5SDimitry Andric     QualType ETy = E->getType();
28900b57cec5SDimitry Andric     QualType DestTy = Context.getPointerType(Context.VoidTy.withConst());
28910b57cec5SDimitry Andric     ExprResult ExprRes = E;
28920b57cec5SDimitry Andric     AssignConvertType ConvTy =
28930b57cec5SDimitry Andric       CheckSingleAssignmentConstraints(DestTy, ExprRes);
28940b57cec5SDimitry Andric     if (ExprRes.isInvalid())
28950b57cec5SDimitry Andric       return StmtError();
28960b57cec5SDimitry Andric     E = ExprRes.get();
28970b57cec5SDimitry Andric     if (DiagnoseAssignmentResult(ConvTy, StarLoc, DestTy, ETy, E, AA_Passing))
28980b57cec5SDimitry Andric       return StmtError();
28990b57cec5SDimitry Andric   }
29000b57cec5SDimitry Andric 
29010b57cec5SDimitry Andric   ExprResult ExprRes = ActOnFinishFullExpr(E, /*DiscardedValue*/ false);
29020b57cec5SDimitry Andric   if (ExprRes.isInvalid())
29030b57cec5SDimitry Andric     return StmtError();
29040b57cec5SDimitry Andric   E = ExprRes.get();
29050b57cec5SDimitry Andric 
29060b57cec5SDimitry Andric   setFunctionHasIndirectGoto();
29070b57cec5SDimitry Andric 
29080b57cec5SDimitry Andric   return new (Context) IndirectGotoStmt(GotoLoc, StarLoc, E);
29090b57cec5SDimitry Andric }
29100b57cec5SDimitry Andric 
29110b57cec5SDimitry Andric static void CheckJumpOutOfSEHFinally(Sema &S, SourceLocation Loc,
29120b57cec5SDimitry Andric                                      const Scope &DestScope) {
29130b57cec5SDimitry Andric   if (!S.CurrentSEHFinally.empty() &&
29140b57cec5SDimitry Andric       DestScope.Contains(*S.CurrentSEHFinally.back())) {
29150b57cec5SDimitry Andric     S.Diag(Loc, diag::warn_jump_out_of_seh_finally);
29160b57cec5SDimitry Andric   }
29170b57cec5SDimitry Andric }
29180b57cec5SDimitry Andric 
29190b57cec5SDimitry Andric StmtResult
29200b57cec5SDimitry Andric Sema::ActOnContinueStmt(SourceLocation ContinueLoc, Scope *CurScope) {
29210b57cec5SDimitry Andric   Scope *S = CurScope->getContinueParent();
29220b57cec5SDimitry Andric   if (!S) {
29230b57cec5SDimitry Andric     // C99 6.8.6.2p1: A break shall appear only in or as a loop body.
29240b57cec5SDimitry Andric     return StmtError(Diag(ContinueLoc, diag::err_continue_not_in_loop));
29250b57cec5SDimitry Andric   }
29260b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, ContinueLoc, *S);
29270b57cec5SDimitry Andric 
29280b57cec5SDimitry Andric   return new (Context) ContinueStmt(ContinueLoc);
29290b57cec5SDimitry Andric }
29300b57cec5SDimitry Andric 
29310b57cec5SDimitry Andric StmtResult
29320b57cec5SDimitry Andric Sema::ActOnBreakStmt(SourceLocation BreakLoc, Scope *CurScope) {
29330b57cec5SDimitry Andric   Scope *S = CurScope->getBreakParent();
29340b57cec5SDimitry Andric   if (!S) {
29350b57cec5SDimitry Andric     // C99 6.8.6.3p1: A break shall appear only in or as a switch/loop body.
29360b57cec5SDimitry Andric     return StmtError(Diag(BreakLoc, diag::err_break_not_in_loop_or_switch));
29370b57cec5SDimitry Andric   }
29380b57cec5SDimitry Andric   if (S->isOpenMPLoopScope())
29390b57cec5SDimitry Andric     return StmtError(Diag(BreakLoc, diag::err_omp_loop_cannot_use_stmt)
29400b57cec5SDimitry Andric                      << "break");
29410b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, BreakLoc, *S);
29420b57cec5SDimitry Andric 
29430b57cec5SDimitry Andric   return new (Context) BreakStmt(BreakLoc);
29440b57cec5SDimitry Andric }
29450b57cec5SDimitry Andric 
29460b57cec5SDimitry Andric /// Determine whether the given expression is a candidate for
29470b57cec5SDimitry Andric /// copy elision in either a return statement or a throw expression.
29480b57cec5SDimitry Andric ///
29490b57cec5SDimitry Andric /// \param ReturnType If we're determining the copy elision candidate for
29500b57cec5SDimitry Andric /// a return statement, this is the return type of the function. If we're
29510b57cec5SDimitry Andric /// determining the copy elision candidate for a throw expression, this will
29520b57cec5SDimitry Andric /// be a NULL type.
29530b57cec5SDimitry Andric ///
29540b57cec5SDimitry Andric /// \param E The expression being returned from the function or block, or
29550b57cec5SDimitry Andric /// being thrown.
29560b57cec5SDimitry Andric ///
29570b57cec5SDimitry Andric /// \param CESK Whether we allow function parameters or
29580b57cec5SDimitry Andric /// id-expressions that could be moved out of the function to be considered NRVO
29590b57cec5SDimitry Andric /// candidates. C++ prohibits these for NRVO itself, but we re-use this logic to
29600b57cec5SDimitry Andric /// determine whether we should try to move as part of a return or throw (which
29610b57cec5SDimitry Andric /// does allow function parameters).
29620b57cec5SDimitry Andric ///
29630b57cec5SDimitry Andric /// \returns The NRVO candidate variable, if the return statement may use the
29640b57cec5SDimitry Andric /// NRVO, or NULL if there is no such candidate.
29650b57cec5SDimitry Andric VarDecl *Sema::getCopyElisionCandidate(QualType ReturnType, Expr *E,
29660b57cec5SDimitry Andric                                        CopyElisionSemanticsKind CESK) {
29670b57cec5SDimitry Andric   // - in a return statement in a function [where] ...
29680b57cec5SDimitry Andric   // ... the expression is the name of a non-volatile automatic object ...
29690b57cec5SDimitry Andric   DeclRefExpr *DR = dyn_cast<DeclRefExpr>(E->IgnoreParens());
29700b57cec5SDimitry Andric   if (!DR || DR->refersToEnclosingVariableOrCapture())
29710b57cec5SDimitry Andric     return nullptr;
29720b57cec5SDimitry Andric   VarDecl *VD = dyn_cast<VarDecl>(DR->getDecl());
29730b57cec5SDimitry Andric   if (!VD)
29740b57cec5SDimitry Andric     return nullptr;
29750b57cec5SDimitry Andric 
29760b57cec5SDimitry Andric   if (isCopyElisionCandidate(ReturnType, VD, CESK))
29770b57cec5SDimitry Andric     return VD;
29780b57cec5SDimitry Andric   return nullptr;
29790b57cec5SDimitry Andric }
29800b57cec5SDimitry Andric 
29810b57cec5SDimitry Andric bool Sema::isCopyElisionCandidate(QualType ReturnType, const VarDecl *VD,
29820b57cec5SDimitry Andric                                   CopyElisionSemanticsKind CESK) {
29830b57cec5SDimitry Andric   QualType VDType = VD->getType();
29840b57cec5SDimitry Andric   // - in a return statement in a function with ...
29850b57cec5SDimitry Andric   // ... a class return type ...
29860b57cec5SDimitry Andric   if (!ReturnType.isNull() && !ReturnType->isDependentType()) {
29870b57cec5SDimitry Andric     if (!ReturnType->isRecordType())
29880b57cec5SDimitry Andric       return false;
29890b57cec5SDimitry Andric     // ... the same cv-unqualified type as the function return type ...
29900b57cec5SDimitry Andric     // When considering moving this expression out, allow dissimilar types.
29910b57cec5SDimitry Andric     if (!(CESK & CES_AllowDifferentTypes) && !VDType->isDependentType() &&
29920b57cec5SDimitry Andric         !Context.hasSameUnqualifiedType(ReturnType, VDType))
29930b57cec5SDimitry Andric       return false;
29940b57cec5SDimitry Andric   }
29950b57cec5SDimitry Andric 
29960b57cec5SDimitry Andric   // ...object (other than a function or catch-clause parameter)...
29970b57cec5SDimitry Andric   if (VD->getKind() != Decl::Var &&
29980b57cec5SDimitry Andric       !((CESK & CES_AllowParameters) && VD->getKind() == Decl::ParmVar))
29990b57cec5SDimitry Andric     return false;
30000b57cec5SDimitry Andric   if (!(CESK & CES_AllowExceptionVariables) && VD->isExceptionVariable())
30010b57cec5SDimitry Andric     return false;
30020b57cec5SDimitry Andric 
30030b57cec5SDimitry Andric   // ...automatic...
30040b57cec5SDimitry Andric   if (!VD->hasLocalStorage()) return false;
30050b57cec5SDimitry Andric 
30060b57cec5SDimitry Andric   // Return false if VD is a __block variable. We don't want to implicitly move
30070b57cec5SDimitry Andric   // out of a __block variable during a return because we cannot assume the
30080b57cec5SDimitry Andric   // variable will no longer be used.
30090b57cec5SDimitry Andric   if (VD->hasAttr<BlocksAttr>()) return false;
30100b57cec5SDimitry Andric 
30110b57cec5SDimitry Andric   if (CESK & CES_AllowDifferentTypes)
30120b57cec5SDimitry Andric     return true;
30130b57cec5SDimitry Andric 
30140b57cec5SDimitry Andric   // ...non-volatile...
30150b57cec5SDimitry Andric   if (VD->getType().isVolatileQualified()) return false;
30160b57cec5SDimitry Andric 
30170b57cec5SDimitry Andric   // Variables with higher required alignment than their type's ABI
30180b57cec5SDimitry Andric   // alignment cannot use NRVO.
30190b57cec5SDimitry Andric   if (!VD->getType()->isDependentType() && VD->hasAttr<AlignedAttr>() &&
30200b57cec5SDimitry Andric       Context.getDeclAlign(VD) > Context.getTypeAlignInChars(VD->getType()))
30210b57cec5SDimitry Andric     return false;
30220b57cec5SDimitry Andric 
30230b57cec5SDimitry Andric   return true;
30240b57cec5SDimitry Andric }
30250b57cec5SDimitry Andric 
30260b57cec5SDimitry Andric /// Try to perform the initialization of a potentially-movable value,
30270b57cec5SDimitry Andric /// which is the operand to a return or throw statement.
30280b57cec5SDimitry Andric ///
30290b57cec5SDimitry Andric /// This routine implements C++14 [class.copy]p32, which attempts to treat
30300b57cec5SDimitry Andric /// returned lvalues as rvalues in certain cases (to prefer move construction),
30310b57cec5SDimitry Andric /// then falls back to treating them as lvalues if that failed.
30320b57cec5SDimitry Andric ///
30330b57cec5SDimitry Andric /// \param ConvertingConstructorsOnly If true, follow [class.copy]p32 and reject
30340b57cec5SDimitry Andric /// resolutions that find non-constructors, such as derived-to-base conversions
30350b57cec5SDimitry Andric /// or `operator T()&&` member functions. If false, do consider such
30360b57cec5SDimitry Andric /// conversion sequences.
30370b57cec5SDimitry Andric ///
30380b57cec5SDimitry Andric /// \param Res We will fill this in if move-initialization was possible.
30390b57cec5SDimitry Andric /// If move-initialization is not possible, such that we must fall back to
30400b57cec5SDimitry Andric /// treating the operand as an lvalue, we will leave Res in its original
30410b57cec5SDimitry Andric /// invalid state.
30420b57cec5SDimitry Andric static void TryMoveInitialization(Sema& S,
30430b57cec5SDimitry Andric                                   const InitializedEntity &Entity,
30440b57cec5SDimitry Andric                                   const VarDecl *NRVOCandidate,
30450b57cec5SDimitry Andric                                   QualType ResultType,
30460b57cec5SDimitry Andric                                   Expr *&Value,
30470b57cec5SDimitry Andric                                   bool ConvertingConstructorsOnly,
30480b57cec5SDimitry Andric                                   ExprResult &Res) {
30490b57cec5SDimitry Andric   ImplicitCastExpr AsRvalue(ImplicitCastExpr::OnStack, Value->getType(),
30500b57cec5SDimitry Andric                             CK_NoOp, Value, VK_XValue);
30510b57cec5SDimitry Andric 
30520b57cec5SDimitry Andric   Expr *InitExpr = &AsRvalue;
30530b57cec5SDimitry Andric 
30540b57cec5SDimitry Andric   InitializationKind Kind = InitializationKind::CreateCopy(
30550b57cec5SDimitry Andric       Value->getBeginLoc(), Value->getBeginLoc());
30560b57cec5SDimitry Andric 
30570b57cec5SDimitry Andric   InitializationSequence Seq(S, Entity, Kind, InitExpr);
30580b57cec5SDimitry Andric 
30590b57cec5SDimitry Andric   if (!Seq)
30600b57cec5SDimitry Andric     return;
30610b57cec5SDimitry Andric 
30620b57cec5SDimitry Andric   for (const InitializationSequence::Step &Step : Seq.steps()) {
30630b57cec5SDimitry Andric     if (Step.Kind != InitializationSequence::SK_ConstructorInitialization &&
30640b57cec5SDimitry Andric         Step.Kind != InitializationSequence::SK_UserConversion)
30650b57cec5SDimitry Andric       continue;
30660b57cec5SDimitry Andric 
30670b57cec5SDimitry Andric     FunctionDecl *FD = Step.Function.Function;
30680b57cec5SDimitry Andric     if (ConvertingConstructorsOnly) {
30690b57cec5SDimitry Andric       if (isa<CXXConstructorDecl>(FD)) {
30700b57cec5SDimitry Andric         // C++14 [class.copy]p32:
30710b57cec5SDimitry Andric         // [...] If the first overload resolution fails or was not performed,
30720b57cec5SDimitry Andric         // or if the type of the first parameter of the selected constructor
30730b57cec5SDimitry Andric         // is not an rvalue reference to the object's type (possibly
30740b57cec5SDimitry Andric         // cv-qualified), overload resolution is performed again, considering
30750b57cec5SDimitry Andric         // the object as an lvalue.
30760b57cec5SDimitry Andric         const RValueReferenceType *RRefType =
30770b57cec5SDimitry Andric             FD->getParamDecl(0)->getType()->getAs<RValueReferenceType>();
30780b57cec5SDimitry Andric         if (!RRefType)
30790b57cec5SDimitry Andric           break;
30800b57cec5SDimitry Andric         if (!S.Context.hasSameUnqualifiedType(RRefType->getPointeeType(),
30810b57cec5SDimitry Andric                                               NRVOCandidate->getType()))
30820b57cec5SDimitry Andric           break;
30830b57cec5SDimitry Andric       } else {
30840b57cec5SDimitry Andric         continue;
30850b57cec5SDimitry Andric       }
30860b57cec5SDimitry Andric     } else {
30870b57cec5SDimitry Andric       if (isa<CXXConstructorDecl>(FD)) {
30880b57cec5SDimitry Andric         // Check that overload resolution selected a constructor taking an
30890b57cec5SDimitry Andric         // rvalue reference. If it selected an lvalue reference, then we
30900b57cec5SDimitry Andric         // didn't need to cast this thing to an rvalue in the first place.
30910b57cec5SDimitry Andric         if (!isa<RValueReferenceType>(FD->getParamDecl(0)->getType()))
30920b57cec5SDimitry Andric           break;
30930b57cec5SDimitry Andric       } else if (isa<CXXMethodDecl>(FD)) {
30940b57cec5SDimitry Andric         // Check that overload resolution selected a conversion operator
30950b57cec5SDimitry Andric         // taking an rvalue reference.
30960b57cec5SDimitry Andric         if (cast<CXXMethodDecl>(FD)->getRefQualifier() != RQ_RValue)
30970b57cec5SDimitry Andric           break;
30980b57cec5SDimitry Andric       } else {
30990b57cec5SDimitry Andric         continue;
31000b57cec5SDimitry Andric       }
31010b57cec5SDimitry Andric     }
31020b57cec5SDimitry Andric 
31030b57cec5SDimitry Andric     // Promote "AsRvalue" to the heap, since we now need this
31040b57cec5SDimitry Andric     // expression node to persist.
31050b57cec5SDimitry Andric     Value = ImplicitCastExpr::Create(S.Context, Value->getType(), CK_NoOp,
31060b57cec5SDimitry Andric                                      Value, nullptr, VK_XValue);
31070b57cec5SDimitry Andric 
31080b57cec5SDimitry Andric     // Complete type-checking the initialization of the return type
31090b57cec5SDimitry Andric     // using the constructor we found.
31100b57cec5SDimitry Andric     Res = Seq.Perform(S, Entity, Kind, Value);
31110b57cec5SDimitry Andric   }
31120b57cec5SDimitry Andric }
31130b57cec5SDimitry Andric 
31140b57cec5SDimitry Andric /// Perform the initialization of a potentially-movable value, which
31150b57cec5SDimitry Andric /// is the result of return value.
31160b57cec5SDimitry Andric ///
31170b57cec5SDimitry Andric /// This routine implements C++14 [class.copy]p32, which attempts to treat
31180b57cec5SDimitry Andric /// returned lvalues as rvalues in certain cases (to prefer move construction),
31190b57cec5SDimitry Andric /// then falls back to treating them as lvalues if that failed.
31200b57cec5SDimitry Andric ExprResult
31210b57cec5SDimitry Andric Sema::PerformMoveOrCopyInitialization(const InitializedEntity &Entity,
31220b57cec5SDimitry Andric                                       const VarDecl *NRVOCandidate,
31230b57cec5SDimitry Andric                                       QualType ResultType,
31240b57cec5SDimitry Andric                                       Expr *Value,
31250b57cec5SDimitry Andric                                       bool AllowNRVO) {
31260b57cec5SDimitry Andric   // C++14 [class.copy]p32:
31270b57cec5SDimitry Andric   // When the criteria for elision of a copy/move operation are met, but not for
31280b57cec5SDimitry Andric   // an exception-declaration, and the object to be copied is designated by an
31290b57cec5SDimitry Andric   // lvalue, or when the expression in a return statement is a (possibly
31300b57cec5SDimitry Andric   // parenthesized) id-expression that names an object with automatic storage
31310b57cec5SDimitry Andric   // duration declared in the body or parameter-declaration-clause of the
31320b57cec5SDimitry Andric   // innermost enclosing function or lambda-expression, overload resolution to
31330b57cec5SDimitry Andric   // select the constructor for the copy is first performed as if the object
31340b57cec5SDimitry Andric   // were designated by an rvalue.
31350b57cec5SDimitry Andric   ExprResult Res = ExprError();
31360b57cec5SDimitry Andric 
31370b57cec5SDimitry Andric   if (AllowNRVO) {
31380b57cec5SDimitry Andric     bool AffectedByCWG1579 = false;
31390b57cec5SDimitry Andric 
31400b57cec5SDimitry Andric     if (!NRVOCandidate) {
31410b57cec5SDimitry Andric       NRVOCandidate = getCopyElisionCandidate(ResultType, Value, CES_Default);
31420b57cec5SDimitry Andric       if (NRVOCandidate &&
31430b57cec5SDimitry Andric           !getDiagnostics().isIgnored(diag::warn_return_std_move_in_cxx11,
31440b57cec5SDimitry Andric                                       Value->getExprLoc())) {
31450b57cec5SDimitry Andric         const VarDecl *NRVOCandidateInCXX11 =
31460b57cec5SDimitry Andric             getCopyElisionCandidate(ResultType, Value, CES_FormerDefault);
31470b57cec5SDimitry Andric         AffectedByCWG1579 = (!NRVOCandidateInCXX11);
31480b57cec5SDimitry Andric       }
31490b57cec5SDimitry Andric     }
31500b57cec5SDimitry Andric 
31510b57cec5SDimitry Andric     if (NRVOCandidate) {
31520b57cec5SDimitry Andric       TryMoveInitialization(*this, Entity, NRVOCandidate, ResultType, Value,
31530b57cec5SDimitry Andric                             true, Res);
31540b57cec5SDimitry Andric     }
31550b57cec5SDimitry Andric 
31560b57cec5SDimitry Andric     if (!Res.isInvalid() && AffectedByCWG1579) {
31570b57cec5SDimitry Andric       QualType QT = NRVOCandidate->getType();
31580b57cec5SDimitry Andric       if (QT.getNonReferenceType()
31590b57cec5SDimitry Andric                      .getUnqualifiedType()
31600b57cec5SDimitry Andric                      .isTriviallyCopyableType(Context)) {
31610b57cec5SDimitry Andric         // Adding 'std::move' around a trivially copyable variable is probably
31620b57cec5SDimitry Andric         // pointless. Don't suggest it.
31630b57cec5SDimitry Andric       } else {
31640b57cec5SDimitry Andric         // Common cases for this are returning unique_ptr<Derived> from a
31650b57cec5SDimitry Andric         // function of return type unique_ptr<Base>, or returning T from a
31660b57cec5SDimitry Andric         // function of return type Expected<T>. This is totally fine in a
31670b57cec5SDimitry Andric         // post-CWG1579 world, but was not fine before.
31680b57cec5SDimitry Andric         assert(!ResultType.isNull());
31690b57cec5SDimitry Andric         SmallString<32> Str;
31700b57cec5SDimitry Andric         Str += "std::move(";
31710b57cec5SDimitry Andric         Str += NRVOCandidate->getDeclName().getAsString();
31720b57cec5SDimitry Andric         Str += ")";
31730b57cec5SDimitry Andric         Diag(Value->getExprLoc(), diag::warn_return_std_move_in_cxx11)
31740b57cec5SDimitry Andric             << Value->getSourceRange()
31750b57cec5SDimitry Andric             << NRVOCandidate->getDeclName() << ResultType << QT;
31760b57cec5SDimitry Andric         Diag(Value->getExprLoc(), diag::note_add_std_move_in_cxx11)
31770b57cec5SDimitry Andric             << FixItHint::CreateReplacement(Value->getSourceRange(), Str);
31780b57cec5SDimitry Andric       }
31790b57cec5SDimitry Andric     } else if (Res.isInvalid() &&
31800b57cec5SDimitry Andric                !getDiagnostics().isIgnored(diag::warn_return_std_move,
31810b57cec5SDimitry Andric                                            Value->getExprLoc())) {
31820b57cec5SDimitry Andric       const VarDecl *FakeNRVOCandidate =
31830b57cec5SDimitry Andric           getCopyElisionCandidate(QualType(), Value, CES_AsIfByStdMove);
31840b57cec5SDimitry Andric       if (FakeNRVOCandidate) {
31850b57cec5SDimitry Andric         QualType QT = FakeNRVOCandidate->getType();
31860b57cec5SDimitry Andric         if (QT->isLValueReferenceType()) {
31870b57cec5SDimitry Andric           // Adding 'std::move' around an lvalue reference variable's name is
31880b57cec5SDimitry Andric           // dangerous. Don't suggest it.
31890b57cec5SDimitry Andric         } else if (QT.getNonReferenceType()
31900b57cec5SDimitry Andric                        .getUnqualifiedType()
31910b57cec5SDimitry Andric                        .isTriviallyCopyableType(Context)) {
31920b57cec5SDimitry Andric           // Adding 'std::move' around a trivially copyable variable is probably
31930b57cec5SDimitry Andric           // pointless. Don't suggest it.
31940b57cec5SDimitry Andric         } else {
31950b57cec5SDimitry Andric           ExprResult FakeRes = ExprError();
31960b57cec5SDimitry Andric           Expr *FakeValue = Value;
31970b57cec5SDimitry Andric           TryMoveInitialization(*this, Entity, FakeNRVOCandidate, ResultType,
31980b57cec5SDimitry Andric                                 FakeValue, false, FakeRes);
31990b57cec5SDimitry Andric           if (!FakeRes.isInvalid()) {
32000b57cec5SDimitry Andric             bool IsThrow =
32010b57cec5SDimitry Andric                 (Entity.getKind() == InitializedEntity::EK_Exception);
32020b57cec5SDimitry Andric             SmallString<32> Str;
32030b57cec5SDimitry Andric             Str += "std::move(";
32040b57cec5SDimitry Andric             Str += FakeNRVOCandidate->getDeclName().getAsString();
32050b57cec5SDimitry Andric             Str += ")";
32060b57cec5SDimitry Andric             Diag(Value->getExprLoc(), diag::warn_return_std_move)
32070b57cec5SDimitry Andric                 << Value->getSourceRange()
32080b57cec5SDimitry Andric                 << FakeNRVOCandidate->getDeclName() << IsThrow;
32090b57cec5SDimitry Andric             Diag(Value->getExprLoc(), diag::note_add_std_move)
32100b57cec5SDimitry Andric                 << FixItHint::CreateReplacement(Value->getSourceRange(), Str);
32110b57cec5SDimitry Andric           }
32120b57cec5SDimitry Andric         }
32130b57cec5SDimitry Andric       }
32140b57cec5SDimitry Andric     }
32150b57cec5SDimitry Andric   }
32160b57cec5SDimitry Andric 
32170b57cec5SDimitry Andric   // Either we didn't meet the criteria for treating an lvalue as an rvalue,
32180b57cec5SDimitry Andric   // above, or overload resolution failed. Either way, we need to try
32190b57cec5SDimitry Andric   // (again) now with the return value expression as written.
32200b57cec5SDimitry Andric   if (Res.isInvalid())
32210b57cec5SDimitry Andric     Res = PerformCopyInitialization(Entity, SourceLocation(), Value);
32220b57cec5SDimitry Andric 
32230b57cec5SDimitry Andric   return Res;
32240b57cec5SDimitry Andric }
32250b57cec5SDimitry Andric 
32260b57cec5SDimitry Andric /// Determine whether the declared return type of the specified function
32270b57cec5SDimitry Andric /// contains 'auto'.
32280b57cec5SDimitry Andric static bool hasDeducedReturnType(FunctionDecl *FD) {
32290b57cec5SDimitry Andric   const FunctionProtoType *FPT =
32300b57cec5SDimitry Andric       FD->getTypeSourceInfo()->getType()->castAs<FunctionProtoType>();
32310b57cec5SDimitry Andric   return FPT->getReturnType()->isUndeducedType();
32320b57cec5SDimitry Andric }
32330b57cec5SDimitry Andric 
32340b57cec5SDimitry Andric /// ActOnCapScopeReturnStmt - Utility routine to type-check return statements
32350b57cec5SDimitry Andric /// for capturing scopes.
32360b57cec5SDimitry Andric ///
32370b57cec5SDimitry Andric StmtResult
32380b57cec5SDimitry Andric Sema::ActOnCapScopeReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
32390b57cec5SDimitry Andric   // If this is the first return we've seen, infer the return type.
32400b57cec5SDimitry Andric   // [expr.prim.lambda]p4 in C++11; block literals follow the same rules.
32410b57cec5SDimitry Andric   CapturingScopeInfo *CurCap = cast<CapturingScopeInfo>(getCurFunction());
32420b57cec5SDimitry Andric   QualType FnRetType = CurCap->ReturnType;
32430b57cec5SDimitry Andric   LambdaScopeInfo *CurLambda = dyn_cast<LambdaScopeInfo>(CurCap);
32440b57cec5SDimitry Andric   bool HasDeducedReturnType =
32450b57cec5SDimitry Andric       CurLambda && hasDeducedReturnType(CurLambda->CallOperator);
32460b57cec5SDimitry Andric 
32470b57cec5SDimitry Andric   if (ExprEvalContexts.back().Context ==
32480b57cec5SDimitry Andric           ExpressionEvaluationContext::DiscardedStatement &&
32490b57cec5SDimitry Andric       (HasDeducedReturnType || CurCap->HasImplicitReturnType)) {
32500b57cec5SDimitry Andric     if (RetValExp) {
32510b57cec5SDimitry Andric       ExprResult ER =
32520b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
32530b57cec5SDimitry Andric       if (ER.isInvalid())
32540b57cec5SDimitry Andric         return StmtError();
32550b57cec5SDimitry Andric       RetValExp = ER.get();
32560b57cec5SDimitry Andric     }
32570b57cec5SDimitry Andric     return ReturnStmt::Create(Context, ReturnLoc, RetValExp,
32580b57cec5SDimitry Andric                               /* NRVOCandidate=*/nullptr);
32590b57cec5SDimitry Andric   }
32600b57cec5SDimitry Andric 
32610b57cec5SDimitry Andric   if (HasDeducedReturnType) {
32620b57cec5SDimitry Andric     // In C++1y, the return type may involve 'auto'.
32630b57cec5SDimitry Andric     // FIXME: Blocks might have a return type of 'auto' explicitly specified.
32640b57cec5SDimitry Andric     FunctionDecl *FD = CurLambda->CallOperator;
32650b57cec5SDimitry Andric     if (CurCap->ReturnType.isNull())
32660b57cec5SDimitry Andric       CurCap->ReturnType = FD->getReturnType();
32670b57cec5SDimitry Andric 
32680b57cec5SDimitry Andric     AutoType *AT = CurCap->ReturnType->getContainedAutoType();
32690b57cec5SDimitry Andric     assert(AT && "lost auto type from lambda return type");
32700b57cec5SDimitry Andric     if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
32710b57cec5SDimitry Andric       FD->setInvalidDecl();
32720b57cec5SDimitry Andric       return StmtError();
32730b57cec5SDimitry Andric     }
32740b57cec5SDimitry Andric     CurCap->ReturnType = FnRetType = FD->getReturnType();
32750b57cec5SDimitry Andric   } else if (CurCap->HasImplicitReturnType) {
32760b57cec5SDimitry Andric     // For blocks/lambdas with implicit return types, we check each return
32770b57cec5SDimitry Andric     // statement individually, and deduce the common return type when the block
32780b57cec5SDimitry Andric     // or lambda is completed.
32790b57cec5SDimitry Andric     // FIXME: Fold this into the 'auto' codepath above.
32800b57cec5SDimitry Andric     if (RetValExp && !isa<InitListExpr>(RetValExp)) {
32810b57cec5SDimitry Andric       ExprResult Result = DefaultFunctionArrayLvalueConversion(RetValExp);
32820b57cec5SDimitry Andric       if (Result.isInvalid())
32830b57cec5SDimitry Andric         return StmtError();
32840b57cec5SDimitry Andric       RetValExp = Result.get();
32850b57cec5SDimitry Andric 
32860b57cec5SDimitry Andric       // DR1048: even prior to C++14, we should use the 'auto' deduction rules
32870b57cec5SDimitry Andric       // when deducing a return type for a lambda-expression (or by extension
32880b57cec5SDimitry Andric       // for a block). These rules differ from the stated C++11 rules only in
32890b57cec5SDimitry Andric       // that they remove top-level cv-qualifiers.
32900b57cec5SDimitry Andric       if (!CurContext->isDependentContext())
32910b57cec5SDimitry Andric         FnRetType = RetValExp->getType().getUnqualifiedType();
32920b57cec5SDimitry Andric       else
32930b57cec5SDimitry Andric         FnRetType = CurCap->ReturnType = Context.DependentTy;
32940b57cec5SDimitry Andric     } else {
32950b57cec5SDimitry Andric       if (RetValExp) {
32960b57cec5SDimitry Andric         // C++11 [expr.lambda.prim]p4 bans inferring the result from an
32970b57cec5SDimitry Andric         // initializer list, because it is not an expression (even
32980b57cec5SDimitry Andric         // though we represent it as one). We still deduce 'void'.
32990b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_lambda_return_init_list)
33000b57cec5SDimitry Andric           << RetValExp->getSourceRange();
33010b57cec5SDimitry Andric       }
33020b57cec5SDimitry Andric 
33030b57cec5SDimitry Andric       FnRetType = Context.VoidTy;
33040b57cec5SDimitry Andric     }
33050b57cec5SDimitry Andric 
33060b57cec5SDimitry Andric     // Although we'll properly infer the type of the block once it's completed,
33070b57cec5SDimitry Andric     // make sure we provide a return type now for better error recovery.
33080b57cec5SDimitry Andric     if (CurCap->ReturnType.isNull())
33090b57cec5SDimitry Andric       CurCap->ReturnType = FnRetType;
33100b57cec5SDimitry Andric   }
33110b57cec5SDimitry Andric   assert(!FnRetType.isNull());
33120b57cec5SDimitry Andric 
3313*a7dea167SDimitry Andric   if (auto *CurBlock = dyn_cast<BlockScopeInfo>(CurCap)) {
3314*a7dea167SDimitry Andric     if (CurBlock->FunctionType->castAs<FunctionType>()->getNoReturnAttr()) {
33150b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_noreturn_block_has_return_expr);
33160b57cec5SDimitry Andric       return StmtError();
33170b57cec5SDimitry Andric     }
3318*a7dea167SDimitry Andric   } else if (auto *CurRegion = dyn_cast<CapturedRegionScopeInfo>(CurCap)) {
33190b57cec5SDimitry Andric     Diag(ReturnLoc, diag::err_return_in_captured_stmt) << CurRegion->getRegionName();
33200b57cec5SDimitry Andric     return StmtError();
33210b57cec5SDimitry Andric   } else {
33220b57cec5SDimitry Andric     assert(CurLambda && "unknown kind of captured scope");
3323*a7dea167SDimitry Andric     if (CurLambda->CallOperator->getType()
3324*a7dea167SDimitry Andric             ->castAs<FunctionType>()
33250b57cec5SDimitry Andric             ->getNoReturnAttr()) {
33260b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_noreturn_lambda_has_return_expr);
33270b57cec5SDimitry Andric       return StmtError();
33280b57cec5SDimitry Andric     }
33290b57cec5SDimitry Andric   }
33300b57cec5SDimitry Andric 
33310b57cec5SDimitry Andric   // Otherwise, verify that this result type matches the previous one.  We are
33320b57cec5SDimitry Andric   // pickier with blocks than for normal functions because we don't have GCC
33330b57cec5SDimitry Andric   // compatibility to worry about here.
33340b57cec5SDimitry Andric   const VarDecl *NRVOCandidate = nullptr;
33350b57cec5SDimitry Andric   if (FnRetType->isDependentType()) {
33360b57cec5SDimitry Andric     // Delay processing for now.  TODO: there are lots of dependent
33370b57cec5SDimitry Andric     // types we can conclusively prove aren't void.
33380b57cec5SDimitry Andric   } else if (FnRetType->isVoidType()) {
33390b57cec5SDimitry Andric     if (RetValExp && !isa<InitListExpr>(RetValExp) &&
33400b57cec5SDimitry Andric         !(getLangOpts().CPlusPlus &&
33410b57cec5SDimitry Andric           (RetValExp->isTypeDependent() ||
33420b57cec5SDimitry Andric            RetValExp->getType()->isVoidType()))) {
33430b57cec5SDimitry Andric       if (!getLangOpts().CPlusPlus &&
33440b57cec5SDimitry Andric           RetValExp->getType()->isVoidType())
33450b57cec5SDimitry Andric         Diag(ReturnLoc, diag::ext_return_has_void_expr) << "literal" << 2;
33460b57cec5SDimitry Andric       else {
33470b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_return_block_has_expr);
33480b57cec5SDimitry Andric         RetValExp = nullptr;
33490b57cec5SDimitry Andric       }
33500b57cec5SDimitry Andric     }
33510b57cec5SDimitry Andric   } else if (!RetValExp) {
33520b57cec5SDimitry Andric     return StmtError(Diag(ReturnLoc, diag::err_block_return_missing_expr));
33530b57cec5SDimitry Andric   } else if (!RetValExp->isTypeDependent()) {
33540b57cec5SDimitry Andric     // we have a non-void block with an expression, continue checking
33550b57cec5SDimitry Andric 
33560b57cec5SDimitry Andric     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
33570b57cec5SDimitry Andric     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
33580b57cec5SDimitry Andric     // function return.
33590b57cec5SDimitry Andric 
33600b57cec5SDimitry Andric     // In C++ the return statement is handled via a copy initialization.
33610b57cec5SDimitry Andric     // the C version of which boils down to CheckSingleAssignmentConstraints.
33620b57cec5SDimitry Andric     NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, CES_Strict);
33630b57cec5SDimitry Andric     InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
33640b57cec5SDimitry Andric                                                                    FnRetType,
33650b57cec5SDimitry Andric                                                       NRVOCandidate != nullptr);
33660b57cec5SDimitry Andric     ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
33670b57cec5SDimitry Andric                                                      FnRetType, RetValExp);
33680b57cec5SDimitry Andric     if (Res.isInvalid()) {
33690b57cec5SDimitry Andric       // FIXME: Cleanup temporaries here, anyway?
33700b57cec5SDimitry Andric       return StmtError();
33710b57cec5SDimitry Andric     }
33720b57cec5SDimitry Andric     RetValExp = Res.get();
33730b57cec5SDimitry Andric     CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc);
33740b57cec5SDimitry Andric   } else {
33750b57cec5SDimitry Andric     NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, CES_Strict);
33760b57cec5SDimitry Andric   }
33770b57cec5SDimitry Andric 
33780b57cec5SDimitry Andric   if (RetValExp) {
33790b57cec5SDimitry Andric     ExprResult ER =
33800b57cec5SDimitry Andric         ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
33810b57cec5SDimitry Andric     if (ER.isInvalid())
33820b57cec5SDimitry Andric       return StmtError();
33830b57cec5SDimitry Andric     RetValExp = ER.get();
33840b57cec5SDimitry Andric   }
33850b57cec5SDimitry Andric   auto *Result =
33860b57cec5SDimitry Andric       ReturnStmt::Create(Context, ReturnLoc, RetValExp, NRVOCandidate);
33870b57cec5SDimitry Andric 
33880b57cec5SDimitry Andric   // If we need to check for the named return value optimization,
33890b57cec5SDimitry Andric   // or if we need to infer the return type,
33900b57cec5SDimitry Andric   // save the return statement in our scope for later processing.
33910b57cec5SDimitry Andric   if (CurCap->HasImplicitReturnType || NRVOCandidate)
33920b57cec5SDimitry Andric     FunctionScopes.back()->Returns.push_back(Result);
33930b57cec5SDimitry Andric 
33940b57cec5SDimitry Andric   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
33950b57cec5SDimitry Andric     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
33960b57cec5SDimitry Andric 
33970b57cec5SDimitry Andric   return Result;
33980b57cec5SDimitry Andric }
33990b57cec5SDimitry Andric 
34000b57cec5SDimitry Andric namespace {
34010b57cec5SDimitry Andric /// Marks all typedefs in all local classes in a type referenced.
34020b57cec5SDimitry Andric ///
34030b57cec5SDimitry Andric /// In a function like
34040b57cec5SDimitry Andric /// auto f() {
34050b57cec5SDimitry Andric ///   struct S { typedef int a; };
34060b57cec5SDimitry Andric ///   return S();
34070b57cec5SDimitry Andric /// }
34080b57cec5SDimitry Andric ///
34090b57cec5SDimitry Andric /// the local type escapes and could be referenced in some TUs but not in
34100b57cec5SDimitry Andric /// others. Pretend that all local typedefs are always referenced, to not warn
34110b57cec5SDimitry Andric /// on this. This isn't necessary if f has internal linkage, or the typedef
34120b57cec5SDimitry Andric /// is private.
34130b57cec5SDimitry Andric class LocalTypedefNameReferencer
34140b57cec5SDimitry Andric     : public RecursiveASTVisitor<LocalTypedefNameReferencer> {
34150b57cec5SDimitry Andric public:
34160b57cec5SDimitry Andric   LocalTypedefNameReferencer(Sema &S) : S(S) {}
34170b57cec5SDimitry Andric   bool VisitRecordType(const RecordType *RT);
34180b57cec5SDimitry Andric private:
34190b57cec5SDimitry Andric   Sema &S;
34200b57cec5SDimitry Andric };
34210b57cec5SDimitry Andric bool LocalTypedefNameReferencer::VisitRecordType(const RecordType *RT) {
34220b57cec5SDimitry Andric   auto *R = dyn_cast<CXXRecordDecl>(RT->getDecl());
34230b57cec5SDimitry Andric   if (!R || !R->isLocalClass() || !R->isLocalClass()->isExternallyVisible() ||
34240b57cec5SDimitry Andric       R->isDependentType())
34250b57cec5SDimitry Andric     return true;
34260b57cec5SDimitry Andric   for (auto *TmpD : R->decls())
34270b57cec5SDimitry Andric     if (auto *T = dyn_cast<TypedefNameDecl>(TmpD))
34280b57cec5SDimitry Andric       if (T->getAccess() != AS_private || R->hasFriends())
34290b57cec5SDimitry Andric         S.MarkAnyDeclReferenced(T->getLocation(), T, /*OdrUse=*/false);
34300b57cec5SDimitry Andric   return true;
34310b57cec5SDimitry Andric }
34320b57cec5SDimitry Andric }
34330b57cec5SDimitry Andric 
34340b57cec5SDimitry Andric TypeLoc Sema::getReturnTypeLoc(FunctionDecl *FD) const {
34350b57cec5SDimitry Andric   return FD->getTypeSourceInfo()
34360b57cec5SDimitry Andric       ->getTypeLoc()
34370b57cec5SDimitry Andric       .getAsAdjusted<FunctionProtoTypeLoc>()
34380b57cec5SDimitry Andric       .getReturnLoc();
34390b57cec5SDimitry Andric }
34400b57cec5SDimitry Andric 
34410b57cec5SDimitry Andric /// Deduce the return type for a function from a returned expression, per
34420b57cec5SDimitry Andric /// C++1y [dcl.spec.auto]p6.
34430b57cec5SDimitry Andric bool Sema::DeduceFunctionTypeFromReturnExpr(FunctionDecl *FD,
34440b57cec5SDimitry Andric                                             SourceLocation ReturnLoc,
34450b57cec5SDimitry Andric                                             Expr *&RetExpr,
34460b57cec5SDimitry Andric                                             AutoType *AT) {
34470b57cec5SDimitry Andric   // If this is the conversion function for a lambda, we choose to deduce it
34480b57cec5SDimitry Andric   // type from the corresponding call operator, not from the synthesized return
34490b57cec5SDimitry Andric   // statement within it. See Sema::DeduceReturnType.
34500b57cec5SDimitry Andric   if (isLambdaConversionOperator(FD))
34510b57cec5SDimitry Andric     return false;
34520b57cec5SDimitry Andric 
34530b57cec5SDimitry Andric   TypeLoc OrigResultType = getReturnTypeLoc(FD);
34540b57cec5SDimitry Andric   QualType Deduced;
34550b57cec5SDimitry Andric 
34560b57cec5SDimitry Andric   if (RetExpr && isa<InitListExpr>(RetExpr)) {
34570b57cec5SDimitry Andric     //  If the deduction is for a return statement and the initializer is
34580b57cec5SDimitry Andric     //  a braced-init-list, the program is ill-formed.
34590b57cec5SDimitry Andric     Diag(RetExpr->getExprLoc(),
34600b57cec5SDimitry Andric          getCurLambda() ? diag::err_lambda_return_init_list
34610b57cec5SDimitry Andric                         : diag::err_auto_fn_return_init_list)
34620b57cec5SDimitry Andric         << RetExpr->getSourceRange();
34630b57cec5SDimitry Andric     return true;
34640b57cec5SDimitry Andric   }
34650b57cec5SDimitry Andric 
34660b57cec5SDimitry Andric   if (FD->isDependentContext()) {
34670b57cec5SDimitry Andric     // C++1y [dcl.spec.auto]p12:
34680b57cec5SDimitry Andric     //   Return type deduction [...] occurs when the definition is
34690b57cec5SDimitry Andric     //   instantiated even if the function body contains a return
34700b57cec5SDimitry Andric     //   statement with a non-type-dependent operand.
34710b57cec5SDimitry Andric     assert(AT->isDeduced() && "should have deduced to dependent type");
34720b57cec5SDimitry Andric     return false;
34730b57cec5SDimitry Andric   }
34740b57cec5SDimitry Andric 
34750b57cec5SDimitry Andric   if (RetExpr) {
34760b57cec5SDimitry Andric     //  Otherwise, [...] deduce a value for U using the rules of template
34770b57cec5SDimitry Andric     //  argument deduction.
34780b57cec5SDimitry Andric     DeduceAutoResult DAR = DeduceAutoType(OrigResultType, RetExpr, Deduced);
34790b57cec5SDimitry Andric 
34800b57cec5SDimitry Andric     if (DAR == DAR_Failed && !FD->isInvalidDecl())
34810b57cec5SDimitry Andric       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_deduction_failure)
34820b57cec5SDimitry Andric         << OrigResultType.getType() << RetExpr->getType();
34830b57cec5SDimitry Andric 
34840b57cec5SDimitry Andric     if (DAR != DAR_Succeeded)
34850b57cec5SDimitry Andric       return true;
34860b57cec5SDimitry Andric 
34870b57cec5SDimitry Andric     // If a local type is part of the returned type, mark its fields as
34880b57cec5SDimitry Andric     // referenced.
34890b57cec5SDimitry Andric     LocalTypedefNameReferencer Referencer(*this);
34900b57cec5SDimitry Andric     Referencer.TraverseType(RetExpr->getType());
34910b57cec5SDimitry Andric   } else {
34920b57cec5SDimitry Andric     //  In the case of a return with no operand, the initializer is considered
34930b57cec5SDimitry Andric     //  to be void().
34940b57cec5SDimitry Andric     //
34950b57cec5SDimitry Andric     // Deduction here can only succeed if the return type is exactly 'cv auto'
34960b57cec5SDimitry Andric     // or 'decltype(auto)', so just check for that case directly.
34970b57cec5SDimitry Andric     if (!OrigResultType.getType()->getAs<AutoType>()) {
34980b57cec5SDimitry Andric       Diag(ReturnLoc, diag::err_auto_fn_return_void_but_not_auto)
34990b57cec5SDimitry Andric         << OrigResultType.getType();
35000b57cec5SDimitry Andric       return true;
35010b57cec5SDimitry Andric     }
35020b57cec5SDimitry Andric     // We always deduce U = void in this case.
35030b57cec5SDimitry Andric     Deduced = SubstAutoType(OrigResultType.getType(), Context.VoidTy);
35040b57cec5SDimitry Andric     if (Deduced.isNull())
35050b57cec5SDimitry Andric       return true;
35060b57cec5SDimitry Andric   }
35070b57cec5SDimitry Andric 
3508*a7dea167SDimitry Andric   // CUDA: Kernel function must have 'void' return type.
3509*a7dea167SDimitry Andric   if (getLangOpts().CUDA)
3510*a7dea167SDimitry Andric     if (FD->hasAttr<CUDAGlobalAttr>() && !Deduced->isVoidType()) {
3511*a7dea167SDimitry Andric       Diag(FD->getLocation(), diag::err_kern_type_not_void_return)
3512*a7dea167SDimitry Andric           << FD->getType() << FD->getSourceRange();
3513*a7dea167SDimitry Andric       return true;
3514*a7dea167SDimitry Andric     }
3515*a7dea167SDimitry Andric 
35160b57cec5SDimitry Andric   //  If a function with a declared return type that contains a placeholder type
35170b57cec5SDimitry Andric   //  has multiple return statements, the return type is deduced for each return
35180b57cec5SDimitry Andric   //  statement. [...] if the type deduced is not the same in each deduction,
35190b57cec5SDimitry Andric   //  the program is ill-formed.
35200b57cec5SDimitry Andric   QualType DeducedT = AT->getDeducedType();
35210b57cec5SDimitry Andric   if (!DeducedT.isNull() && !FD->isInvalidDecl()) {
35220b57cec5SDimitry Andric     AutoType *NewAT = Deduced->getContainedAutoType();
35230b57cec5SDimitry Andric     // It is possible that NewAT->getDeducedType() is null. When that happens,
35240b57cec5SDimitry Andric     // we should not crash, instead we ignore this deduction.
35250b57cec5SDimitry Andric     if (NewAT->getDeducedType().isNull())
35260b57cec5SDimitry Andric       return false;
35270b57cec5SDimitry Andric 
35280b57cec5SDimitry Andric     CanQualType OldDeducedType = Context.getCanonicalFunctionResultType(
35290b57cec5SDimitry Andric                                    DeducedT);
35300b57cec5SDimitry Andric     CanQualType NewDeducedType = Context.getCanonicalFunctionResultType(
35310b57cec5SDimitry Andric                                    NewAT->getDeducedType());
35320b57cec5SDimitry Andric     if (!FD->isDependentContext() && OldDeducedType != NewDeducedType) {
35330b57cec5SDimitry Andric       const LambdaScopeInfo *LambdaSI = getCurLambda();
35340b57cec5SDimitry Andric       if (LambdaSI && LambdaSI->HasImplicitReturnType) {
35350b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_typecheck_missing_return_type_incompatible)
35360b57cec5SDimitry Andric           << NewAT->getDeducedType() << DeducedT
35370b57cec5SDimitry Andric           << true /*IsLambda*/;
35380b57cec5SDimitry Andric       } else {
35390b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_auto_fn_different_deductions)
35400b57cec5SDimitry Andric           << (AT->isDecltypeAuto() ? 1 : 0)
35410b57cec5SDimitry Andric           << NewAT->getDeducedType() << DeducedT;
35420b57cec5SDimitry Andric       }
35430b57cec5SDimitry Andric       return true;
35440b57cec5SDimitry Andric     }
35450b57cec5SDimitry Andric   } else if (!FD->isInvalidDecl()) {
35460b57cec5SDimitry Andric     // Update all declarations of the function to have the deduced return type.
35470b57cec5SDimitry Andric     Context.adjustDeducedFunctionResultType(FD, Deduced);
35480b57cec5SDimitry Andric   }
35490b57cec5SDimitry Andric 
35500b57cec5SDimitry Andric   return false;
35510b57cec5SDimitry Andric }
35520b57cec5SDimitry Andric 
35530b57cec5SDimitry Andric StmtResult
35540b57cec5SDimitry Andric Sema::ActOnReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp,
35550b57cec5SDimitry Andric                       Scope *CurScope) {
35560b57cec5SDimitry Andric   // Correct typos, in case the containing function returns 'auto' and
35570b57cec5SDimitry Andric   // RetValExp should determine the deduced type.
35580b57cec5SDimitry Andric   ExprResult RetVal = CorrectDelayedTyposInExpr(RetValExp);
35590b57cec5SDimitry Andric   if (RetVal.isInvalid())
35600b57cec5SDimitry Andric     return StmtError();
35610b57cec5SDimitry Andric   StmtResult R = BuildReturnStmt(ReturnLoc, RetVal.get());
35620b57cec5SDimitry Andric   if (R.isInvalid() || ExprEvalContexts.back().Context ==
35630b57cec5SDimitry Andric                            ExpressionEvaluationContext::DiscardedStatement)
35640b57cec5SDimitry Andric     return R;
35650b57cec5SDimitry Andric 
35660b57cec5SDimitry Andric   if (VarDecl *VD =
35670b57cec5SDimitry Andric       const_cast<VarDecl*>(cast<ReturnStmt>(R.get())->getNRVOCandidate())) {
35680b57cec5SDimitry Andric     CurScope->addNRVOCandidate(VD);
35690b57cec5SDimitry Andric   } else {
35700b57cec5SDimitry Andric     CurScope->setNoNRVO();
35710b57cec5SDimitry Andric   }
35720b57cec5SDimitry Andric 
35730b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, ReturnLoc, *CurScope->getFnParent());
35740b57cec5SDimitry Andric 
35750b57cec5SDimitry Andric   return R;
35760b57cec5SDimitry Andric }
35770b57cec5SDimitry Andric 
35780b57cec5SDimitry Andric StmtResult Sema::BuildReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
35790b57cec5SDimitry Andric   // Check for unexpanded parameter packs.
35800b57cec5SDimitry Andric   if (RetValExp && DiagnoseUnexpandedParameterPack(RetValExp))
35810b57cec5SDimitry Andric     return StmtError();
35820b57cec5SDimitry Andric 
35830b57cec5SDimitry Andric   if (isa<CapturingScopeInfo>(getCurFunction()))
35840b57cec5SDimitry Andric     return ActOnCapScopeReturnStmt(ReturnLoc, RetValExp);
35850b57cec5SDimitry Andric 
35860b57cec5SDimitry Andric   QualType FnRetType;
35870b57cec5SDimitry Andric   QualType RelatedRetType;
35880b57cec5SDimitry Andric   const AttrVec *Attrs = nullptr;
35890b57cec5SDimitry Andric   bool isObjCMethod = false;
35900b57cec5SDimitry Andric 
35910b57cec5SDimitry Andric   if (const FunctionDecl *FD = getCurFunctionDecl()) {
35920b57cec5SDimitry Andric     FnRetType = FD->getReturnType();
35930b57cec5SDimitry Andric     if (FD->hasAttrs())
35940b57cec5SDimitry Andric       Attrs = &FD->getAttrs();
35950b57cec5SDimitry Andric     if (FD->isNoReturn())
35960b57cec5SDimitry Andric       Diag(ReturnLoc, diag::warn_noreturn_function_has_return_expr)
35970b57cec5SDimitry Andric         << FD->getDeclName();
35980b57cec5SDimitry Andric     if (FD->isMain() && RetValExp)
35990b57cec5SDimitry Andric       if (isa<CXXBoolLiteralExpr>(RetValExp))
36000b57cec5SDimitry Andric         Diag(ReturnLoc, diag::warn_main_returns_bool_literal)
36010b57cec5SDimitry Andric           << RetValExp->getSourceRange();
36020b57cec5SDimitry Andric   } else if (ObjCMethodDecl *MD = getCurMethodDecl()) {
36030b57cec5SDimitry Andric     FnRetType = MD->getReturnType();
36040b57cec5SDimitry Andric     isObjCMethod = true;
36050b57cec5SDimitry Andric     if (MD->hasAttrs())
36060b57cec5SDimitry Andric       Attrs = &MD->getAttrs();
36070b57cec5SDimitry Andric     if (MD->hasRelatedResultType() && MD->getClassInterface()) {
36080b57cec5SDimitry Andric       // In the implementation of a method with a related return type, the
36090b57cec5SDimitry Andric       // type used to type-check the validity of return statements within the
36100b57cec5SDimitry Andric       // method body is a pointer to the type of the class being implemented.
36110b57cec5SDimitry Andric       RelatedRetType = Context.getObjCInterfaceType(MD->getClassInterface());
36120b57cec5SDimitry Andric       RelatedRetType = Context.getObjCObjectPointerType(RelatedRetType);
36130b57cec5SDimitry Andric     }
36140b57cec5SDimitry Andric   } else // If we don't have a function/method context, bail.
36150b57cec5SDimitry Andric     return StmtError();
36160b57cec5SDimitry Andric 
36170b57cec5SDimitry Andric   // C++1z: discarded return statements are not considered when deducing a
36180b57cec5SDimitry Andric   // return type.
36190b57cec5SDimitry Andric   if (ExprEvalContexts.back().Context ==
36200b57cec5SDimitry Andric           ExpressionEvaluationContext::DiscardedStatement &&
36210b57cec5SDimitry Andric       FnRetType->getContainedAutoType()) {
36220b57cec5SDimitry Andric     if (RetValExp) {
36230b57cec5SDimitry Andric       ExprResult ER =
36240b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
36250b57cec5SDimitry Andric       if (ER.isInvalid())
36260b57cec5SDimitry Andric         return StmtError();
36270b57cec5SDimitry Andric       RetValExp = ER.get();
36280b57cec5SDimitry Andric     }
36290b57cec5SDimitry Andric     return ReturnStmt::Create(Context, ReturnLoc, RetValExp,
36300b57cec5SDimitry Andric                               /* NRVOCandidate=*/nullptr);
36310b57cec5SDimitry Andric   }
36320b57cec5SDimitry Andric 
36330b57cec5SDimitry Andric   // FIXME: Add a flag to the ScopeInfo to indicate whether we're performing
36340b57cec5SDimitry Andric   // deduction.
36350b57cec5SDimitry Andric   if (getLangOpts().CPlusPlus14) {
36360b57cec5SDimitry Andric     if (AutoType *AT = FnRetType->getContainedAutoType()) {
36370b57cec5SDimitry Andric       FunctionDecl *FD = cast<FunctionDecl>(CurContext);
36380b57cec5SDimitry Andric       if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
36390b57cec5SDimitry Andric         FD->setInvalidDecl();
36400b57cec5SDimitry Andric         return StmtError();
36410b57cec5SDimitry Andric       } else {
36420b57cec5SDimitry Andric         FnRetType = FD->getReturnType();
36430b57cec5SDimitry Andric       }
36440b57cec5SDimitry Andric     }
36450b57cec5SDimitry Andric   }
36460b57cec5SDimitry Andric 
36470b57cec5SDimitry Andric   bool HasDependentReturnType = FnRetType->isDependentType();
36480b57cec5SDimitry Andric 
36490b57cec5SDimitry Andric   ReturnStmt *Result = nullptr;
36500b57cec5SDimitry Andric   if (FnRetType->isVoidType()) {
36510b57cec5SDimitry Andric     if (RetValExp) {
36520b57cec5SDimitry Andric       if (isa<InitListExpr>(RetValExp)) {
36530b57cec5SDimitry Andric         // We simply never allow init lists as the return value of void
36540b57cec5SDimitry Andric         // functions. This is compatible because this was never allowed before,
36550b57cec5SDimitry Andric         // so there's no legacy code to deal with.
36560b57cec5SDimitry Andric         NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
36570b57cec5SDimitry Andric         int FunctionKind = 0;
36580b57cec5SDimitry Andric         if (isa<ObjCMethodDecl>(CurDecl))
36590b57cec5SDimitry Andric           FunctionKind = 1;
36600b57cec5SDimitry Andric         else if (isa<CXXConstructorDecl>(CurDecl))
36610b57cec5SDimitry Andric           FunctionKind = 2;
36620b57cec5SDimitry Andric         else if (isa<CXXDestructorDecl>(CurDecl))
36630b57cec5SDimitry Andric           FunctionKind = 3;
36640b57cec5SDimitry Andric 
36650b57cec5SDimitry Andric         Diag(ReturnLoc, diag::err_return_init_list)
36660b57cec5SDimitry Andric           << CurDecl->getDeclName() << FunctionKind
36670b57cec5SDimitry Andric           << RetValExp->getSourceRange();
36680b57cec5SDimitry Andric 
36690b57cec5SDimitry Andric         // Drop the expression.
36700b57cec5SDimitry Andric         RetValExp = nullptr;
36710b57cec5SDimitry Andric       } else if (!RetValExp->isTypeDependent()) {
36720b57cec5SDimitry Andric         // C99 6.8.6.4p1 (ext_ since GCC warns)
36730b57cec5SDimitry Andric         unsigned D = diag::ext_return_has_expr;
36740b57cec5SDimitry Andric         if (RetValExp->getType()->isVoidType()) {
36750b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
36760b57cec5SDimitry Andric           if (isa<CXXConstructorDecl>(CurDecl) ||
36770b57cec5SDimitry Andric               isa<CXXDestructorDecl>(CurDecl))
36780b57cec5SDimitry Andric             D = diag::err_ctor_dtor_returns_void;
36790b57cec5SDimitry Andric           else
36800b57cec5SDimitry Andric             D = diag::ext_return_has_void_expr;
36810b57cec5SDimitry Andric         }
36820b57cec5SDimitry Andric         else {
36830b57cec5SDimitry Andric           ExprResult Result = RetValExp;
36840b57cec5SDimitry Andric           Result = IgnoredValueConversions(Result.get());
36850b57cec5SDimitry Andric           if (Result.isInvalid())
36860b57cec5SDimitry Andric             return StmtError();
36870b57cec5SDimitry Andric           RetValExp = Result.get();
36880b57cec5SDimitry Andric           RetValExp = ImpCastExprToType(RetValExp,
36890b57cec5SDimitry Andric                                         Context.VoidTy, CK_ToVoid).get();
36900b57cec5SDimitry Andric         }
36910b57cec5SDimitry Andric         // return of void in constructor/destructor is illegal in C++.
36920b57cec5SDimitry Andric         if (D == diag::err_ctor_dtor_returns_void) {
36930b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
36940b57cec5SDimitry Andric           Diag(ReturnLoc, D)
36950b57cec5SDimitry Andric             << CurDecl->getDeclName() << isa<CXXDestructorDecl>(CurDecl)
36960b57cec5SDimitry Andric             << RetValExp->getSourceRange();
36970b57cec5SDimitry Andric         }
36980b57cec5SDimitry Andric         // return (some void expression); is legal in C++.
36990b57cec5SDimitry Andric         else if (D != diag::ext_return_has_void_expr ||
37000b57cec5SDimitry Andric                  !getLangOpts().CPlusPlus) {
37010b57cec5SDimitry Andric           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
37020b57cec5SDimitry Andric 
37030b57cec5SDimitry Andric           int FunctionKind = 0;
37040b57cec5SDimitry Andric           if (isa<ObjCMethodDecl>(CurDecl))
37050b57cec5SDimitry Andric             FunctionKind = 1;
37060b57cec5SDimitry Andric           else if (isa<CXXConstructorDecl>(CurDecl))
37070b57cec5SDimitry Andric             FunctionKind = 2;
37080b57cec5SDimitry Andric           else if (isa<CXXDestructorDecl>(CurDecl))
37090b57cec5SDimitry Andric             FunctionKind = 3;
37100b57cec5SDimitry Andric 
37110b57cec5SDimitry Andric           Diag(ReturnLoc, D)
37120b57cec5SDimitry Andric             << CurDecl->getDeclName() << FunctionKind
37130b57cec5SDimitry Andric             << RetValExp->getSourceRange();
37140b57cec5SDimitry Andric         }
37150b57cec5SDimitry Andric       }
37160b57cec5SDimitry Andric 
37170b57cec5SDimitry Andric       if (RetValExp) {
37180b57cec5SDimitry Andric         ExprResult ER =
37190b57cec5SDimitry Andric             ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
37200b57cec5SDimitry Andric         if (ER.isInvalid())
37210b57cec5SDimitry Andric           return StmtError();
37220b57cec5SDimitry Andric         RetValExp = ER.get();
37230b57cec5SDimitry Andric       }
37240b57cec5SDimitry Andric     }
37250b57cec5SDimitry Andric 
37260b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, RetValExp,
37270b57cec5SDimitry Andric                                 /* NRVOCandidate=*/nullptr);
37280b57cec5SDimitry Andric   } else if (!RetValExp && !HasDependentReturnType) {
37290b57cec5SDimitry Andric     FunctionDecl *FD = getCurFunctionDecl();
37300b57cec5SDimitry Andric 
37310b57cec5SDimitry Andric     unsigned DiagID;
37320b57cec5SDimitry Andric     if (getLangOpts().CPlusPlus11 && FD && FD->isConstexpr()) {
37330b57cec5SDimitry Andric       // C++11 [stmt.return]p2
37340b57cec5SDimitry Andric       DiagID = diag::err_constexpr_return_missing_expr;
37350b57cec5SDimitry Andric       FD->setInvalidDecl();
37360b57cec5SDimitry Andric     } else if (getLangOpts().C99) {
37370b57cec5SDimitry Andric       // C99 6.8.6.4p1 (ext_ since GCC warns)
37380b57cec5SDimitry Andric       DiagID = diag::ext_return_missing_expr;
37390b57cec5SDimitry Andric     } else {
37400b57cec5SDimitry Andric       // C90 6.6.6.4p4
37410b57cec5SDimitry Andric       DiagID = diag::warn_return_missing_expr;
37420b57cec5SDimitry Andric     }
37430b57cec5SDimitry Andric 
37440b57cec5SDimitry Andric     if (FD)
37450b57cec5SDimitry Andric       Diag(ReturnLoc, DiagID)
37460b57cec5SDimitry Andric           << FD->getIdentifier() << 0 /*fn*/ << FD->isConsteval();
37470b57cec5SDimitry Andric     else
37480b57cec5SDimitry Andric       Diag(ReturnLoc, DiagID) << getCurMethodDecl()->getDeclName() << 1/*meth*/;
37490b57cec5SDimitry Andric 
37500b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, /* RetExpr=*/nullptr,
37510b57cec5SDimitry Andric                                 /* NRVOCandidate=*/nullptr);
37520b57cec5SDimitry Andric   } else {
37530b57cec5SDimitry Andric     assert(RetValExp || HasDependentReturnType);
37540b57cec5SDimitry Andric     const VarDecl *NRVOCandidate = nullptr;
37550b57cec5SDimitry Andric 
37560b57cec5SDimitry Andric     QualType RetType = RelatedRetType.isNull() ? FnRetType : RelatedRetType;
37570b57cec5SDimitry Andric 
37580b57cec5SDimitry Andric     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
37590b57cec5SDimitry Andric     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
37600b57cec5SDimitry Andric     // function return.
37610b57cec5SDimitry Andric 
37620b57cec5SDimitry Andric     // In C++ the return statement is handled via a copy initialization,
37630b57cec5SDimitry Andric     // the C version of which boils down to CheckSingleAssignmentConstraints.
37640b57cec5SDimitry Andric     if (RetValExp)
37650b57cec5SDimitry Andric       NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, CES_Strict);
37660b57cec5SDimitry Andric     if (!HasDependentReturnType && !RetValExp->isTypeDependent()) {
37670b57cec5SDimitry Andric       // we have a non-void function with an expression, continue checking
37680b57cec5SDimitry Andric       InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
37690b57cec5SDimitry Andric                                                                      RetType,
37700b57cec5SDimitry Andric                                                       NRVOCandidate != nullptr);
37710b57cec5SDimitry Andric       ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
37720b57cec5SDimitry Andric                                                        RetType, RetValExp);
37730b57cec5SDimitry Andric       if (Res.isInvalid()) {
37740b57cec5SDimitry Andric         // FIXME: Clean up temporaries here anyway?
37750b57cec5SDimitry Andric         return StmtError();
37760b57cec5SDimitry Andric       }
37770b57cec5SDimitry Andric       RetValExp = Res.getAs<Expr>();
37780b57cec5SDimitry Andric 
37790b57cec5SDimitry Andric       // If we have a related result type, we need to implicitly
37800b57cec5SDimitry Andric       // convert back to the formal result type.  We can't pretend to
37810b57cec5SDimitry Andric       // initialize the result again --- we might end double-retaining
37820b57cec5SDimitry Andric       // --- so instead we initialize a notional temporary.
37830b57cec5SDimitry Andric       if (!RelatedRetType.isNull()) {
37840b57cec5SDimitry Andric         Entity = InitializedEntity::InitializeRelatedResult(getCurMethodDecl(),
37850b57cec5SDimitry Andric                                                             FnRetType);
37860b57cec5SDimitry Andric         Res = PerformCopyInitialization(Entity, ReturnLoc, RetValExp);
37870b57cec5SDimitry Andric         if (Res.isInvalid()) {
37880b57cec5SDimitry Andric           // FIXME: Clean up temporaries here anyway?
37890b57cec5SDimitry Andric           return StmtError();
37900b57cec5SDimitry Andric         }
37910b57cec5SDimitry Andric         RetValExp = Res.getAs<Expr>();
37920b57cec5SDimitry Andric       }
37930b57cec5SDimitry Andric 
37940b57cec5SDimitry Andric       CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc, isObjCMethod, Attrs,
37950b57cec5SDimitry Andric                          getCurFunctionDecl());
37960b57cec5SDimitry Andric     }
37970b57cec5SDimitry Andric 
37980b57cec5SDimitry Andric     if (RetValExp) {
37990b57cec5SDimitry Andric       ExprResult ER =
38000b57cec5SDimitry Andric           ActOnFinishFullExpr(RetValExp, ReturnLoc, /*DiscardedValue*/ false);
38010b57cec5SDimitry Andric       if (ER.isInvalid())
38020b57cec5SDimitry Andric         return StmtError();
38030b57cec5SDimitry Andric       RetValExp = ER.get();
38040b57cec5SDimitry Andric     }
38050b57cec5SDimitry Andric     Result = ReturnStmt::Create(Context, ReturnLoc, RetValExp, NRVOCandidate);
38060b57cec5SDimitry Andric   }
38070b57cec5SDimitry Andric 
38080b57cec5SDimitry Andric   // If we need to check for the named return value optimization, save the
38090b57cec5SDimitry Andric   // return statement in our scope for later processing.
38100b57cec5SDimitry Andric   if (Result->getNRVOCandidate())
38110b57cec5SDimitry Andric     FunctionScopes.back()->Returns.push_back(Result);
38120b57cec5SDimitry Andric 
38130b57cec5SDimitry Andric   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
38140b57cec5SDimitry Andric     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
38150b57cec5SDimitry Andric 
38160b57cec5SDimitry Andric   return Result;
38170b57cec5SDimitry Andric }
38180b57cec5SDimitry Andric 
38190b57cec5SDimitry Andric StmtResult
38200b57cec5SDimitry Andric Sema::ActOnObjCAtCatchStmt(SourceLocation AtLoc,
38210b57cec5SDimitry Andric                            SourceLocation RParen, Decl *Parm,
38220b57cec5SDimitry Andric                            Stmt *Body) {
38230b57cec5SDimitry Andric   VarDecl *Var = cast_or_null<VarDecl>(Parm);
38240b57cec5SDimitry Andric   if (Var && Var->isInvalidDecl())
38250b57cec5SDimitry Andric     return StmtError();
38260b57cec5SDimitry Andric 
38270b57cec5SDimitry Andric   return new (Context) ObjCAtCatchStmt(AtLoc, RParen, Var, Body);
38280b57cec5SDimitry Andric }
38290b57cec5SDimitry Andric 
38300b57cec5SDimitry Andric StmtResult
38310b57cec5SDimitry Andric Sema::ActOnObjCAtFinallyStmt(SourceLocation AtLoc, Stmt *Body) {
38320b57cec5SDimitry Andric   return new (Context) ObjCAtFinallyStmt(AtLoc, Body);
38330b57cec5SDimitry Andric }
38340b57cec5SDimitry Andric 
38350b57cec5SDimitry Andric StmtResult
38360b57cec5SDimitry Andric Sema::ActOnObjCAtTryStmt(SourceLocation AtLoc, Stmt *Try,
38370b57cec5SDimitry Andric                          MultiStmtArg CatchStmts, Stmt *Finally) {
38380b57cec5SDimitry Andric   if (!getLangOpts().ObjCExceptions)
38390b57cec5SDimitry Andric     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@try";
38400b57cec5SDimitry Andric 
38410b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
38420b57cec5SDimitry Andric   unsigned NumCatchStmts = CatchStmts.size();
38430b57cec5SDimitry Andric   return ObjCAtTryStmt::Create(Context, AtLoc, Try, CatchStmts.data(),
38440b57cec5SDimitry Andric                                NumCatchStmts, Finally);
38450b57cec5SDimitry Andric }
38460b57cec5SDimitry Andric 
38470b57cec5SDimitry Andric StmtResult Sema::BuildObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw) {
38480b57cec5SDimitry Andric   if (Throw) {
38490b57cec5SDimitry Andric     ExprResult Result = DefaultLvalueConversion(Throw);
38500b57cec5SDimitry Andric     if (Result.isInvalid())
38510b57cec5SDimitry Andric       return StmtError();
38520b57cec5SDimitry Andric 
38530b57cec5SDimitry Andric     Result = ActOnFinishFullExpr(Result.get(), /*DiscardedValue*/ false);
38540b57cec5SDimitry Andric     if (Result.isInvalid())
38550b57cec5SDimitry Andric       return StmtError();
38560b57cec5SDimitry Andric     Throw = Result.get();
38570b57cec5SDimitry Andric 
38580b57cec5SDimitry Andric     QualType ThrowType = Throw->getType();
38590b57cec5SDimitry Andric     // Make sure the expression type is an ObjC pointer or "void *".
38600b57cec5SDimitry Andric     if (!ThrowType->isDependentType() &&
38610b57cec5SDimitry Andric         !ThrowType->isObjCObjectPointerType()) {
38620b57cec5SDimitry Andric       const PointerType *PT = ThrowType->getAs<PointerType>();
38630b57cec5SDimitry Andric       if (!PT || !PT->getPointeeType()->isVoidType())
38640b57cec5SDimitry Andric         return StmtError(Diag(AtLoc, diag::err_objc_throw_expects_object)
38650b57cec5SDimitry Andric                          << Throw->getType() << Throw->getSourceRange());
38660b57cec5SDimitry Andric     }
38670b57cec5SDimitry Andric   }
38680b57cec5SDimitry Andric 
38690b57cec5SDimitry Andric   return new (Context) ObjCAtThrowStmt(AtLoc, Throw);
38700b57cec5SDimitry Andric }
38710b57cec5SDimitry Andric 
38720b57cec5SDimitry Andric StmtResult
38730b57cec5SDimitry Andric Sema::ActOnObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw,
38740b57cec5SDimitry Andric                            Scope *CurScope) {
38750b57cec5SDimitry Andric   if (!getLangOpts().ObjCExceptions)
38760b57cec5SDimitry Andric     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@throw";
38770b57cec5SDimitry Andric 
38780b57cec5SDimitry Andric   if (!Throw) {
38790b57cec5SDimitry Andric     // @throw without an expression designates a rethrow (which must occur
38800b57cec5SDimitry Andric     // in the context of an @catch clause).
38810b57cec5SDimitry Andric     Scope *AtCatchParent = CurScope;
38820b57cec5SDimitry Andric     while (AtCatchParent && !AtCatchParent->isAtCatchScope())
38830b57cec5SDimitry Andric       AtCatchParent = AtCatchParent->getParent();
38840b57cec5SDimitry Andric     if (!AtCatchParent)
38850b57cec5SDimitry Andric       return StmtError(Diag(AtLoc, diag::err_rethrow_used_outside_catch));
38860b57cec5SDimitry Andric   }
38870b57cec5SDimitry Andric   return BuildObjCAtThrowStmt(AtLoc, Throw);
38880b57cec5SDimitry Andric }
38890b57cec5SDimitry Andric 
38900b57cec5SDimitry Andric ExprResult
38910b57cec5SDimitry Andric Sema::ActOnObjCAtSynchronizedOperand(SourceLocation atLoc, Expr *operand) {
38920b57cec5SDimitry Andric   ExprResult result = DefaultLvalueConversion(operand);
38930b57cec5SDimitry Andric   if (result.isInvalid())
38940b57cec5SDimitry Andric     return ExprError();
38950b57cec5SDimitry Andric   operand = result.get();
38960b57cec5SDimitry Andric 
38970b57cec5SDimitry Andric   // Make sure the expression type is an ObjC pointer or "void *".
38980b57cec5SDimitry Andric   QualType type = operand->getType();
38990b57cec5SDimitry Andric   if (!type->isDependentType() &&
39000b57cec5SDimitry Andric       !type->isObjCObjectPointerType()) {
39010b57cec5SDimitry Andric     const PointerType *pointerType = type->getAs<PointerType>();
39020b57cec5SDimitry Andric     if (!pointerType || !pointerType->getPointeeType()->isVoidType()) {
39030b57cec5SDimitry Andric       if (getLangOpts().CPlusPlus) {
39040b57cec5SDimitry Andric         if (RequireCompleteType(atLoc, type,
39050b57cec5SDimitry Andric                                 diag::err_incomplete_receiver_type))
39060b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
39070b57cec5SDimitry Andric                    << type << operand->getSourceRange();
39080b57cec5SDimitry Andric 
39090b57cec5SDimitry Andric         ExprResult result = PerformContextuallyConvertToObjCPointer(operand);
39100b57cec5SDimitry Andric         if (result.isInvalid())
39110b57cec5SDimitry Andric           return ExprError();
39120b57cec5SDimitry Andric         if (!result.isUsable())
39130b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
39140b57cec5SDimitry Andric                    << type << operand->getSourceRange();
39150b57cec5SDimitry Andric 
39160b57cec5SDimitry Andric         operand = result.get();
39170b57cec5SDimitry Andric       } else {
39180b57cec5SDimitry Andric           return Diag(atLoc, diag::err_objc_synchronized_expects_object)
39190b57cec5SDimitry Andric                    << type << operand->getSourceRange();
39200b57cec5SDimitry Andric       }
39210b57cec5SDimitry Andric     }
39220b57cec5SDimitry Andric   }
39230b57cec5SDimitry Andric 
39240b57cec5SDimitry Andric   // The operand to @synchronized is a full-expression.
39250b57cec5SDimitry Andric   return ActOnFinishFullExpr(operand, /*DiscardedValue*/ false);
39260b57cec5SDimitry Andric }
39270b57cec5SDimitry Andric 
39280b57cec5SDimitry Andric StmtResult
39290b57cec5SDimitry Andric Sema::ActOnObjCAtSynchronizedStmt(SourceLocation AtLoc, Expr *SyncExpr,
39300b57cec5SDimitry Andric                                   Stmt *SyncBody) {
39310b57cec5SDimitry Andric   // We can't jump into or indirect-jump out of a @synchronized block.
39320b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
39330b57cec5SDimitry Andric   return new (Context) ObjCAtSynchronizedStmt(AtLoc, SyncExpr, SyncBody);
39340b57cec5SDimitry Andric }
39350b57cec5SDimitry Andric 
39360b57cec5SDimitry Andric /// ActOnCXXCatchBlock - Takes an exception declaration and a handler block
39370b57cec5SDimitry Andric /// and creates a proper catch handler from them.
39380b57cec5SDimitry Andric StmtResult
39390b57cec5SDimitry Andric Sema::ActOnCXXCatchBlock(SourceLocation CatchLoc, Decl *ExDecl,
39400b57cec5SDimitry Andric                          Stmt *HandlerBlock) {
39410b57cec5SDimitry Andric   // There's nothing to test that ActOnExceptionDecl didn't already test.
39420b57cec5SDimitry Andric   return new (Context)
39430b57cec5SDimitry Andric       CXXCatchStmt(CatchLoc, cast_or_null<VarDecl>(ExDecl), HandlerBlock);
39440b57cec5SDimitry Andric }
39450b57cec5SDimitry Andric 
39460b57cec5SDimitry Andric StmtResult
39470b57cec5SDimitry Andric Sema::ActOnObjCAutoreleasePoolStmt(SourceLocation AtLoc, Stmt *Body) {
39480b57cec5SDimitry Andric   setFunctionHasBranchProtectedScope();
39490b57cec5SDimitry Andric   return new (Context) ObjCAutoreleasePoolStmt(AtLoc, Body);
39500b57cec5SDimitry Andric }
39510b57cec5SDimitry Andric 
39520b57cec5SDimitry Andric namespace {
39530b57cec5SDimitry Andric class CatchHandlerType {
39540b57cec5SDimitry Andric   QualType QT;
39550b57cec5SDimitry Andric   unsigned IsPointer : 1;
39560b57cec5SDimitry Andric 
39570b57cec5SDimitry Andric   // This is a special constructor to be used only with DenseMapInfo's
39580b57cec5SDimitry Andric   // getEmptyKey() and getTombstoneKey() functions.
39590b57cec5SDimitry Andric   friend struct llvm::DenseMapInfo<CatchHandlerType>;
39600b57cec5SDimitry Andric   enum Unique { ForDenseMap };
39610b57cec5SDimitry Andric   CatchHandlerType(QualType QT, Unique) : QT(QT), IsPointer(false) {}
39620b57cec5SDimitry Andric 
39630b57cec5SDimitry Andric public:
39640b57cec5SDimitry Andric   /// Used when creating a CatchHandlerType from a handler type; will determine
39650b57cec5SDimitry Andric   /// whether the type is a pointer or reference and will strip off the top
39660b57cec5SDimitry Andric   /// level pointer and cv-qualifiers.
39670b57cec5SDimitry Andric   CatchHandlerType(QualType Q) : QT(Q), IsPointer(false) {
39680b57cec5SDimitry Andric     if (QT->isPointerType())
39690b57cec5SDimitry Andric       IsPointer = true;
39700b57cec5SDimitry Andric 
39710b57cec5SDimitry Andric     if (IsPointer || QT->isReferenceType())
39720b57cec5SDimitry Andric       QT = QT->getPointeeType();
39730b57cec5SDimitry Andric     QT = QT.getUnqualifiedType();
39740b57cec5SDimitry Andric   }
39750b57cec5SDimitry Andric 
39760b57cec5SDimitry Andric   /// Used when creating a CatchHandlerType from a base class type; pretends the
39770b57cec5SDimitry Andric   /// type passed in had the pointer qualifier, does not need to get an
39780b57cec5SDimitry Andric   /// unqualified type.
39790b57cec5SDimitry Andric   CatchHandlerType(QualType QT, bool IsPointer)
39800b57cec5SDimitry Andric       : QT(QT), IsPointer(IsPointer) {}
39810b57cec5SDimitry Andric 
39820b57cec5SDimitry Andric   QualType underlying() const { return QT; }
39830b57cec5SDimitry Andric   bool isPointer() const { return IsPointer; }
39840b57cec5SDimitry Andric 
39850b57cec5SDimitry Andric   friend bool operator==(const CatchHandlerType &LHS,
39860b57cec5SDimitry Andric                          const CatchHandlerType &RHS) {
39870b57cec5SDimitry Andric     // If the pointer qualification does not match, we can return early.
39880b57cec5SDimitry Andric     if (LHS.IsPointer != RHS.IsPointer)
39890b57cec5SDimitry Andric       return false;
39900b57cec5SDimitry Andric     // Otherwise, check the underlying type without cv-qualifiers.
39910b57cec5SDimitry Andric     return LHS.QT == RHS.QT;
39920b57cec5SDimitry Andric   }
39930b57cec5SDimitry Andric };
39940b57cec5SDimitry Andric } // namespace
39950b57cec5SDimitry Andric 
39960b57cec5SDimitry Andric namespace llvm {
39970b57cec5SDimitry Andric template <> struct DenseMapInfo<CatchHandlerType> {
39980b57cec5SDimitry Andric   static CatchHandlerType getEmptyKey() {
39990b57cec5SDimitry Andric     return CatchHandlerType(DenseMapInfo<QualType>::getEmptyKey(),
40000b57cec5SDimitry Andric                        CatchHandlerType::ForDenseMap);
40010b57cec5SDimitry Andric   }
40020b57cec5SDimitry Andric 
40030b57cec5SDimitry Andric   static CatchHandlerType getTombstoneKey() {
40040b57cec5SDimitry Andric     return CatchHandlerType(DenseMapInfo<QualType>::getTombstoneKey(),
40050b57cec5SDimitry Andric                        CatchHandlerType::ForDenseMap);
40060b57cec5SDimitry Andric   }
40070b57cec5SDimitry Andric 
40080b57cec5SDimitry Andric   static unsigned getHashValue(const CatchHandlerType &Base) {
40090b57cec5SDimitry Andric     return DenseMapInfo<QualType>::getHashValue(Base.underlying());
40100b57cec5SDimitry Andric   }
40110b57cec5SDimitry Andric 
40120b57cec5SDimitry Andric   static bool isEqual(const CatchHandlerType &LHS,
40130b57cec5SDimitry Andric                       const CatchHandlerType &RHS) {
40140b57cec5SDimitry Andric     return LHS == RHS;
40150b57cec5SDimitry Andric   }
40160b57cec5SDimitry Andric };
40170b57cec5SDimitry Andric }
40180b57cec5SDimitry Andric 
40190b57cec5SDimitry Andric namespace {
40200b57cec5SDimitry Andric class CatchTypePublicBases {
40210b57cec5SDimitry Andric   ASTContext &Ctx;
40220b57cec5SDimitry Andric   const llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> &TypesToCheck;
40230b57cec5SDimitry Andric   const bool CheckAgainstPointer;
40240b57cec5SDimitry Andric 
40250b57cec5SDimitry Andric   CXXCatchStmt *FoundHandler;
40260b57cec5SDimitry Andric   CanQualType FoundHandlerType;
40270b57cec5SDimitry Andric 
40280b57cec5SDimitry Andric public:
40290b57cec5SDimitry Andric   CatchTypePublicBases(
40300b57cec5SDimitry Andric       ASTContext &Ctx,
40310b57cec5SDimitry Andric       const llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> &T, bool C)
40320b57cec5SDimitry Andric       : Ctx(Ctx), TypesToCheck(T), CheckAgainstPointer(C),
40330b57cec5SDimitry Andric         FoundHandler(nullptr) {}
40340b57cec5SDimitry Andric 
40350b57cec5SDimitry Andric   CXXCatchStmt *getFoundHandler() const { return FoundHandler; }
40360b57cec5SDimitry Andric   CanQualType getFoundHandlerType() const { return FoundHandlerType; }
40370b57cec5SDimitry Andric 
40380b57cec5SDimitry Andric   bool operator()(const CXXBaseSpecifier *S, CXXBasePath &) {
40390b57cec5SDimitry Andric     if (S->getAccessSpecifier() == AccessSpecifier::AS_public) {
40400b57cec5SDimitry Andric       CatchHandlerType Check(S->getType(), CheckAgainstPointer);
40410b57cec5SDimitry Andric       const auto &M = TypesToCheck;
40420b57cec5SDimitry Andric       auto I = M.find(Check);
40430b57cec5SDimitry Andric       if (I != M.end()) {
40440b57cec5SDimitry Andric         FoundHandler = I->second;
40450b57cec5SDimitry Andric         FoundHandlerType = Ctx.getCanonicalType(S->getType());
40460b57cec5SDimitry Andric         return true;
40470b57cec5SDimitry Andric       }
40480b57cec5SDimitry Andric     }
40490b57cec5SDimitry Andric     return false;
40500b57cec5SDimitry Andric   }
40510b57cec5SDimitry Andric };
40520b57cec5SDimitry Andric }
40530b57cec5SDimitry Andric 
40540b57cec5SDimitry Andric /// ActOnCXXTryBlock - Takes a try compound-statement and a number of
40550b57cec5SDimitry Andric /// handlers and creates a try statement from them.
40560b57cec5SDimitry Andric StmtResult Sema::ActOnCXXTryBlock(SourceLocation TryLoc, Stmt *TryBlock,
40570b57cec5SDimitry Andric                                   ArrayRef<Stmt *> Handlers) {
40580b57cec5SDimitry Andric   // Don't report an error if 'try' is used in system headers.
40590b57cec5SDimitry Andric   if (!getLangOpts().CXXExceptions &&
40600b57cec5SDimitry Andric       !getSourceManager().isInSystemHeader(TryLoc) && !getLangOpts().CUDA) {
40610b57cec5SDimitry Andric     // Delay error emission for the OpenMP device code.
40620b57cec5SDimitry Andric     targetDiag(TryLoc, diag::err_exceptions_disabled) << "try";
40630b57cec5SDimitry Andric   }
40640b57cec5SDimitry Andric 
40650b57cec5SDimitry Andric   // Exceptions aren't allowed in CUDA device code.
40660b57cec5SDimitry Andric   if (getLangOpts().CUDA)
40670b57cec5SDimitry Andric     CUDADiagIfDeviceCode(TryLoc, diag::err_cuda_device_exceptions)
40680b57cec5SDimitry Andric         << "try" << CurrentCUDATarget();
40690b57cec5SDimitry Andric 
40700b57cec5SDimitry Andric   if (getCurScope() && getCurScope()->isOpenMPSimdDirectiveScope())
40710b57cec5SDimitry Andric     Diag(TryLoc, diag::err_omp_simd_region_cannot_use_stmt) << "try";
40720b57cec5SDimitry Andric 
40730b57cec5SDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
40740b57cec5SDimitry Andric 
40750b57cec5SDimitry Andric   // C++ try is incompatible with SEH __try.
40760b57cec5SDimitry Andric   if (!getLangOpts().Borland && FSI->FirstSEHTryLoc.isValid()) {
40770b57cec5SDimitry Andric     Diag(TryLoc, diag::err_mixing_cxx_try_seh_try);
40780b57cec5SDimitry Andric     Diag(FSI->FirstSEHTryLoc, diag::note_conflicting_try_here) << "'__try'";
40790b57cec5SDimitry Andric   }
40800b57cec5SDimitry Andric 
40810b57cec5SDimitry Andric   const unsigned NumHandlers = Handlers.size();
40820b57cec5SDimitry Andric   assert(!Handlers.empty() &&
40830b57cec5SDimitry Andric          "The parser shouldn't call this if there are no handlers.");
40840b57cec5SDimitry Andric 
40850b57cec5SDimitry Andric   llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> HandledTypes;
40860b57cec5SDimitry Andric   for (unsigned i = 0; i < NumHandlers; ++i) {
40870b57cec5SDimitry Andric     CXXCatchStmt *H = cast<CXXCatchStmt>(Handlers[i]);
40880b57cec5SDimitry Andric 
40890b57cec5SDimitry Andric     // Diagnose when the handler is a catch-all handler, but it isn't the last
40900b57cec5SDimitry Andric     // handler for the try block. [except.handle]p5. Also, skip exception
40910b57cec5SDimitry Andric     // declarations that are invalid, since we can't usefully report on them.
40920b57cec5SDimitry Andric     if (!H->getExceptionDecl()) {
40930b57cec5SDimitry Andric       if (i < NumHandlers - 1)
40940b57cec5SDimitry Andric         return StmtError(Diag(H->getBeginLoc(), diag::err_early_catch_all));
40950b57cec5SDimitry Andric       continue;
40960b57cec5SDimitry Andric     } else if (H->getExceptionDecl()->isInvalidDecl())
40970b57cec5SDimitry Andric       continue;
40980b57cec5SDimitry Andric 
40990b57cec5SDimitry Andric     // Walk the type hierarchy to diagnose when this type has already been
41000b57cec5SDimitry Andric     // handled (duplication), or cannot be handled (derivation inversion). We
41010b57cec5SDimitry Andric     // ignore top-level cv-qualifiers, per [except.handle]p3
41020b57cec5SDimitry Andric     CatchHandlerType HandlerCHT =
41030b57cec5SDimitry Andric         (QualType)Context.getCanonicalType(H->getCaughtType());
41040b57cec5SDimitry Andric 
41050b57cec5SDimitry Andric     // We can ignore whether the type is a reference or a pointer; we need the
41060b57cec5SDimitry Andric     // underlying declaration type in order to get at the underlying record
41070b57cec5SDimitry Andric     // decl, if there is one.
41080b57cec5SDimitry Andric     QualType Underlying = HandlerCHT.underlying();
41090b57cec5SDimitry Andric     if (auto *RD = Underlying->getAsCXXRecordDecl()) {
41100b57cec5SDimitry Andric       if (!RD->hasDefinition())
41110b57cec5SDimitry Andric         continue;
41120b57cec5SDimitry Andric       // Check that none of the public, unambiguous base classes are in the
41130b57cec5SDimitry Andric       // map ([except.handle]p1). Give the base classes the same pointer
41140b57cec5SDimitry Andric       // qualification as the original type we are basing off of. This allows
41150b57cec5SDimitry Andric       // comparison against the handler type using the same top-level pointer
41160b57cec5SDimitry Andric       // as the original type.
41170b57cec5SDimitry Andric       CXXBasePaths Paths;
41180b57cec5SDimitry Andric       Paths.setOrigin(RD);
41190b57cec5SDimitry Andric       CatchTypePublicBases CTPB(Context, HandledTypes, HandlerCHT.isPointer());
41200b57cec5SDimitry Andric       if (RD->lookupInBases(CTPB, Paths)) {
41210b57cec5SDimitry Andric         const CXXCatchStmt *Problem = CTPB.getFoundHandler();
41220b57cec5SDimitry Andric         if (!Paths.isAmbiguous(CTPB.getFoundHandlerType())) {
41230b57cec5SDimitry Andric           Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
41240b57cec5SDimitry Andric                diag::warn_exception_caught_by_earlier_handler)
41250b57cec5SDimitry Andric               << H->getCaughtType();
41260b57cec5SDimitry Andric           Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
41270b57cec5SDimitry Andric                 diag::note_previous_exception_handler)
41280b57cec5SDimitry Andric               << Problem->getCaughtType();
41290b57cec5SDimitry Andric         }
41300b57cec5SDimitry Andric       }
41310b57cec5SDimitry Andric     }
41320b57cec5SDimitry Andric 
41330b57cec5SDimitry Andric     // Add the type the list of ones we have handled; diagnose if we've already
41340b57cec5SDimitry Andric     // handled it.
41350b57cec5SDimitry Andric     auto R = HandledTypes.insert(std::make_pair(H->getCaughtType(), H));
41360b57cec5SDimitry Andric     if (!R.second) {
41370b57cec5SDimitry Andric       const CXXCatchStmt *Problem = R.first->second;
41380b57cec5SDimitry Andric       Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
41390b57cec5SDimitry Andric            diag::warn_exception_caught_by_earlier_handler)
41400b57cec5SDimitry Andric           << H->getCaughtType();
41410b57cec5SDimitry Andric       Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
41420b57cec5SDimitry Andric            diag::note_previous_exception_handler)
41430b57cec5SDimitry Andric           << Problem->getCaughtType();
41440b57cec5SDimitry Andric     }
41450b57cec5SDimitry Andric   }
41460b57cec5SDimitry Andric 
41470b57cec5SDimitry Andric   FSI->setHasCXXTry(TryLoc);
41480b57cec5SDimitry Andric 
41490b57cec5SDimitry Andric   return CXXTryStmt::Create(Context, TryLoc, TryBlock, Handlers);
41500b57cec5SDimitry Andric }
41510b57cec5SDimitry Andric 
41520b57cec5SDimitry Andric StmtResult Sema::ActOnSEHTryBlock(bool IsCXXTry, SourceLocation TryLoc,
41530b57cec5SDimitry Andric                                   Stmt *TryBlock, Stmt *Handler) {
41540b57cec5SDimitry Andric   assert(TryBlock && Handler);
41550b57cec5SDimitry Andric 
41560b57cec5SDimitry Andric   sema::FunctionScopeInfo *FSI = getCurFunction();
41570b57cec5SDimitry Andric 
41580b57cec5SDimitry Andric   // SEH __try is incompatible with C++ try. Borland appears to support this,
41590b57cec5SDimitry Andric   // however.
41600b57cec5SDimitry Andric   if (!getLangOpts().Borland) {
41610b57cec5SDimitry Andric     if (FSI->FirstCXXTryLoc.isValid()) {
41620b57cec5SDimitry Andric       Diag(TryLoc, diag::err_mixing_cxx_try_seh_try);
41630b57cec5SDimitry Andric       Diag(FSI->FirstCXXTryLoc, diag::note_conflicting_try_here) << "'try'";
41640b57cec5SDimitry Andric     }
41650b57cec5SDimitry Andric   }
41660b57cec5SDimitry Andric 
41670b57cec5SDimitry Andric   FSI->setHasSEHTry(TryLoc);
41680b57cec5SDimitry Andric 
41690b57cec5SDimitry Andric   // Reject __try in Obj-C methods, blocks, and captured decls, since we don't
41700b57cec5SDimitry Andric   // track if they use SEH.
41710b57cec5SDimitry Andric   DeclContext *DC = CurContext;
41720b57cec5SDimitry Andric   while (DC && !DC->isFunctionOrMethod())
41730b57cec5SDimitry Andric     DC = DC->getParent();
41740b57cec5SDimitry Andric   FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(DC);
41750b57cec5SDimitry Andric   if (FD)
41760b57cec5SDimitry Andric     FD->setUsesSEHTry(true);
41770b57cec5SDimitry Andric   else
41780b57cec5SDimitry Andric     Diag(TryLoc, diag::err_seh_try_outside_functions);
41790b57cec5SDimitry Andric 
41800b57cec5SDimitry Andric   // Reject __try on unsupported targets.
41810b57cec5SDimitry Andric   if (!Context.getTargetInfo().isSEHTrySupported())
41820b57cec5SDimitry Andric     Diag(TryLoc, diag::err_seh_try_unsupported);
41830b57cec5SDimitry Andric 
41840b57cec5SDimitry Andric   return SEHTryStmt::Create(Context, IsCXXTry, TryLoc, TryBlock, Handler);
41850b57cec5SDimitry Andric }
41860b57cec5SDimitry Andric 
41870b57cec5SDimitry Andric StmtResult
41880b57cec5SDimitry Andric Sema::ActOnSEHExceptBlock(SourceLocation Loc,
41890b57cec5SDimitry Andric                           Expr *FilterExpr,
41900b57cec5SDimitry Andric                           Stmt *Block) {
41910b57cec5SDimitry Andric   assert(FilterExpr && Block);
41920b57cec5SDimitry Andric 
41930b57cec5SDimitry Andric   if(!FilterExpr->getType()->isIntegerType()) {
41940b57cec5SDimitry Andric     return StmtError(Diag(FilterExpr->getExprLoc(),
41950b57cec5SDimitry Andric                      diag::err_filter_expression_integral)
41960b57cec5SDimitry Andric                      << FilterExpr->getType());
41970b57cec5SDimitry Andric   }
41980b57cec5SDimitry Andric 
41990b57cec5SDimitry Andric   return SEHExceptStmt::Create(Context,Loc,FilterExpr,Block);
42000b57cec5SDimitry Andric }
42010b57cec5SDimitry Andric 
42020b57cec5SDimitry Andric void Sema::ActOnStartSEHFinallyBlock() {
42030b57cec5SDimitry Andric   CurrentSEHFinally.push_back(CurScope);
42040b57cec5SDimitry Andric }
42050b57cec5SDimitry Andric 
42060b57cec5SDimitry Andric void Sema::ActOnAbortSEHFinallyBlock() {
42070b57cec5SDimitry Andric   CurrentSEHFinally.pop_back();
42080b57cec5SDimitry Andric }
42090b57cec5SDimitry Andric 
42100b57cec5SDimitry Andric StmtResult Sema::ActOnFinishSEHFinallyBlock(SourceLocation Loc, Stmt *Block) {
42110b57cec5SDimitry Andric   assert(Block);
42120b57cec5SDimitry Andric   CurrentSEHFinally.pop_back();
42130b57cec5SDimitry Andric   return SEHFinallyStmt::Create(Context, Loc, Block);
42140b57cec5SDimitry Andric }
42150b57cec5SDimitry Andric 
42160b57cec5SDimitry Andric StmtResult
42170b57cec5SDimitry Andric Sema::ActOnSEHLeaveStmt(SourceLocation Loc, Scope *CurScope) {
42180b57cec5SDimitry Andric   Scope *SEHTryParent = CurScope;
42190b57cec5SDimitry Andric   while (SEHTryParent && !SEHTryParent->isSEHTryScope())
42200b57cec5SDimitry Andric     SEHTryParent = SEHTryParent->getParent();
42210b57cec5SDimitry Andric   if (!SEHTryParent)
42220b57cec5SDimitry Andric     return StmtError(Diag(Loc, diag::err_ms___leave_not_in___try));
42230b57cec5SDimitry Andric   CheckJumpOutOfSEHFinally(*this, Loc, *SEHTryParent);
42240b57cec5SDimitry Andric 
42250b57cec5SDimitry Andric   return new (Context) SEHLeaveStmt(Loc);
42260b57cec5SDimitry Andric }
42270b57cec5SDimitry Andric 
42280b57cec5SDimitry Andric StmtResult Sema::BuildMSDependentExistsStmt(SourceLocation KeywordLoc,
42290b57cec5SDimitry Andric                                             bool IsIfExists,
42300b57cec5SDimitry Andric                                             NestedNameSpecifierLoc QualifierLoc,
42310b57cec5SDimitry Andric                                             DeclarationNameInfo NameInfo,
42320b57cec5SDimitry Andric                                             Stmt *Nested)
42330b57cec5SDimitry Andric {
42340b57cec5SDimitry Andric   return new (Context) MSDependentExistsStmt(KeywordLoc, IsIfExists,
42350b57cec5SDimitry Andric                                              QualifierLoc, NameInfo,
42360b57cec5SDimitry Andric                                              cast<CompoundStmt>(Nested));
42370b57cec5SDimitry Andric }
42380b57cec5SDimitry Andric 
42390b57cec5SDimitry Andric 
42400b57cec5SDimitry Andric StmtResult Sema::ActOnMSDependentExistsStmt(SourceLocation KeywordLoc,
42410b57cec5SDimitry Andric                                             bool IsIfExists,
42420b57cec5SDimitry Andric                                             CXXScopeSpec &SS,
42430b57cec5SDimitry Andric                                             UnqualifiedId &Name,
42440b57cec5SDimitry Andric                                             Stmt *Nested) {
42450b57cec5SDimitry Andric   return BuildMSDependentExistsStmt(KeywordLoc, IsIfExists,
42460b57cec5SDimitry Andric                                     SS.getWithLocInContext(Context),
42470b57cec5SDimitry Andric                                     GetNameFromUnqualifiedId(Name),
42480b57cec5SDimitry Andric                                     Nested);
42490b57cec5SDimitry Andric }
42500b57cec5SDimitry Andric 
42510b57cec5SDimitry Andric RecordDecl*
42520b57cec5SDimitry Andric Sema::CreateCapturedStmtRecordDecl(CapturedDecl *&CD, SourceLocation Loc,
42530b57cec5SDimitry Andric                                    unsigned NumParams) {
42540b57cec5SDimitry Andric   DeclContext *DC = CurContext;
42550b57cec5SDimitry Andric   while (!(DC->isFunctionOrMethod() || DC->isRecord() || DC->isFileContext()))
42560b57cec5SDimitry Andric     DC = DC->getParent();
42570b57cec5SDimitry Andric 
42580b57cec5SDimitry Andric   RecordDecl *RD = nullptr;
42590b57cec5SDimitry Andric   if (getLangOpts().CPlusPlus)
42600b57cec5SDimitry Andric     RD = CXXRecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc,
42610b57cec5SDimitry Andric                                /*Id=*/nullptr);
42620b57cec5SDimitry Andric   else
42630b57cec5SDimitry Andric     RD = RecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc, /*Id=*/nullptr);
42640b57cec5SDimitry Andric 
42650b57cec5SDimitry Andric   RD->setCapturedRecord();
42660b57cec5SDimitry Andric   DC->addDecl(RD);
42670b57cec5SDimitry Andric   RD->setImplicit();
42680b57cec5SDimitry Andric   RD->startDefinition();
42690b57cec5SDimitry Andric 
42700b57cec5SDimitry Andric   assert(NumParams > 0 && "CapturedStmt requires context parameter");
42710b57cec5SDimitry Andric   CD = CapturedDecl::Create(Context, CurContext, NumParams);
42720b57cec5SDimitry Andric   DC->addDecl(CD);
42730b57cec5SDimitry Andric   return RD;
42740b57cec5SDimitry Andric }
42750b57cec5SDimitry Andric 
42760b57cec5SDimitry Andric static bool
42770b57cec5SDimitry Andric buildCapturedStmtCaptureList(Sema &S, CapturedRegionScopeInfo *RSI,
42780b57cec5SDimitry Andric                              SmallVectorImpl<CapturedStmt::Capture> &Captures,
42790b57cec5SDimitry Andric                              SmallVectorImpl<Expr *> &CaptureInits) {
42800b57cec5SDimitry Andric   for (const sema::Capture &Cap : RSI->Captures) {
42810b57cec5SDimitry Andric     if (Cap.isInvalid())
42820b57cec5SDimitry Andric       continue;
42830b57cec5SDimitry Andric 
42840b57cec5SDimitry Andric     // Form the initializer for the capture.
42850b57cec5SDimitry Andric     ExprResult Init = S.BuildCaptureInit(Cap, Cap.getLocation(),
42860b57cec5SDimitry Andric                                          RSI->CapRegionKind == CR_OpenMP);
42870b57cec5SDimitry Andric 
42880b57cec5SDimitry Andric     // FIXME: Bail out now if the capture is not used and the initializer has
42890b57cec5SDimitry Andric     // no side-effects.
42900b57cec5SDimitry Andric 
42910b57cec5SDimitry Andric     // Create a field for this capture.
42920b57cec5SDimitry Andric     FieldDecl *Field = S.BuildCaptureField(RSI->TheRecordDecl, Cap);
42930b57cec5SDimitry Andric 
42940b57cec5SDimitry Andric     // Add the capture to our list of captures.
42950b57cec5SDimitry Andric     if (Cap.isThisCapture()) {
42960b57cec5SDimitry Andric       Captures.push_back(CapturedStmt::Capture(Cap.getLocation(),
42970b57cec5SDimitry Andric                                                CapturedStmt::VCK_This));
42980b57cec5SDimitry Andric     } else if (Cap.isVLATypeCapture()) {
42990b57cec5SDimitry Andric       Captures.push_back(
43000b57cec5SDimitry Andric           CapturedStmt::Capture(Cap.getLocation(), CapturedStmt::VCK_VLAType));
43010b57cec5SDimitry Andric     } else {
43020b57cec5SDimitry Andric       assert(Cap.isVariableCapture() && "unknown kind of capture");
43030b57cec5SDimitry Andric 
43040b57cec5SDimitry Andric       if (S.getLangOpts().OpenMP && RSI->CapRegionKind == CR_OpenMP)
43050b57cec5SDimitry Andric         S.setOpenMPCaptureKind(Field, Cap.getVariable(), RSI->OpenMPLevel);
43060b57cec5SDimitry Andric 
43070b57cec5SDimitry Andric       Captures.push_back(CapturedStmt::Capture(Cap.getLocation(),
43080b57cec5SDimitry Andric                                                Cap.isReferenceCapture()
43090b57cec5SDimitry Andric                                                    ? CapturedStmt::VCK_ByRef
43100b57cec5SDimitry Andric                                                    : CapturedStmt::VCK_ByCopy,
43110b57cec5SDimitry Andric                                                Cap.getVariable()));
43120b57cec5SDimitry Andric     }
43130b57cec5SDimitry Andric     CaptureInits.push_back(Init.get());
43140b57cec5SDimitry Andric   }
43150b57cec5SDimitry Andric   return false;
43160b57cec5SDimitry Andric }
43170b57cec5SDimitry Andric 
43180b57cec5SDimitry Andric void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
43190b57cec5SDimitry Andric                                     CapturedRegionKind Kind,
43200b57cec5SDimitry Andric                                     unsigned NumParams) {
43210b57cec5SDimitry Andric   CapturedDecl *CD = nullptr;
43220b57cec5SDimitry Andric   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, NumParams);
43230b57cec5SDimitry Andric 
43240b57cec5SDimitry Andric   // Build the context parameter
43250b57cec5SDimitry Andric   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
43260b57cec5SDimitry Andric   IdentifierInfo *ParamName = &Context.Idents.get("__context");
43270b57cec5SDimitry Andric   QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
43280b57cec5SDimitry Andric   auto *Param =
43290b57cec5SDimitry Andric       ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
43300b57cec5SDimitry Andric                                 ImplicitParamDecl::CapturedContext);
43310b57cec5SDimitry Andric   DC->addDecl(Param);
43320b57cec5SDimitry Andric 
43330b57cec5SDimitry Andric   CD->setContextParam(0, Param);
43340b57cec5SDimitry Andric 
43350b57cec5SDimitry Andric   // Enter the capturing scope for this captured region.
43360b57cec5SDimitry Andric   PushCapturedRegionScope(CurScope, CD, RD, Kind);
43370b57cec5SDimitry Andric 
43380b57cec5SDimitry Andric   if (CurScope)
43390b57cec5SDimitry Andric     PushDeclContext(CurScope, CD);
43400b57cec5SDimitry Andric   else
43410b57cec5SDimitry Andric     CurContext = CD;
43420b57cec5SDimitry Andric 
43430b57cec5SDimitry Andric   PushExpressionEvaluationContext(
43440b57cec5SDimitry Andric       ExpressionEvaluationContext::PotentiallyEvaluated);
43450b57cec5SDimitry Andric }
43460b57cec5SDimitry Andric 
43470b57cec5SDimitry Andric void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
43480b57cec5SDimitry Andric                                     CapturedRegionKind Kind,
4349*a7dea167SDimitry Andric                                     ArrayRef<CapturedParamNameType> Params,
4350*a7dea167SDimitry Andric                                     unsigned OpenMPCaptureLevel) {
43510b57cec5SDimitry Andric   CapturedDecl *CD = nullptr;
43520b57cec5SDimitry Andric   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, Params.size());
43530b57cec5SDimitry Andric 
43540b57cec5SDimitry Andric   // Build the context parameter
43550b57cec5SDimitry Andric   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
43560b57cec5SDimitry Andric   bool ContextIsFound = false;
43570b57cec5SDimitry Andric   unsigned ParamNum = 0;
43580b57cec5SDimitry Andric   for (ArrayRef<CapturedParamNameType>::iterator I = Params.begin(),
43590b57cec5SDimitry Andric                                                  E = Params.end();
43600b57cec5SDimitry Andric        I != E; ++I, ++ParamNum) {
43610b57cec5SDimitry Andric     if (I->second.isNull()) {
43620b57cec5SDimitry Andric       assert(!ContextIsFound &&
43630b57cec5SDimitry Andric              "null type has been found already for '__context' parameter");
43640b57cec5SDimitry Andric       IdentifierInfo *ParamName = &Context.Idents.get("__context");
43650b57cec5SDimitry Andric       QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD))
43660b57cec5SDimitry Andric                                .withConst()
43670b57cec5SDimitry Andric                                .withRestrict();
43680b57cec5SDimitry Andric       auto *Param =
43690b57cec5SDimitry Andric           ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
43700b57cec5SDimitry Andric                                     ImplicitParamDecl::CapturedContext);
43710b57cec5SDimitry Andric       DC->addDecl(Param);
43720b57cec5SDimitry Andric       CD->setContextParam(ParamNum, Param);
43730b57cec5SDimitry Andric       ContextIsFound = true;
43740b57cec5SDimitry Andric     } else {
43750b57cec5SDimitry Andric       IdentifierInfo *ParamName = &Context.Idents.get(I->first);
43760b57cec5SDimitry Andric       auto *Param =
43770b57cec5SDimitry Andric           ImplicitParamDecl::Create(Context, DC, Loc, ParamName, I->second,
43780b57cec5SDimitry Andric                                     ImplicitParamDecl::CapturedContext);
43790b57cec5SDimitry Andric       DC->addDecl(Param);
43800b57cec5SDimitry Andric       CD->setParam(ParamNum, Param);
43810b57cec5SDimitry Andric     }
43820b57cec5SDimitry Andric   }
43830b57cec5SDimitry Andric   assert(ContextIsFound && "no null type for '__context' parameter");
43840b57cec5SDimitry Andric   if (!ContextIsFound) {
43850b57cec5SDimitry Andric     // Add __context implicitly if it is not specified.
43860b57cec5SDimitry Andric     IdentifierInfo *ParamName = &Context.Idents.get("__context");
43870b57cec5SDimitry Andric     QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
43880b57cec5SDimitry Andric     auto *Param =
43890b57cec5SDimitry Andric         ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType,
43900b57cec5SDimitry Andric                                   ImplicitParamDecl::CapturedContext);
43910b57cec5SDimitry Andric     DC->addDecl(Param);
43920b57cec5SDimitry Andric     CD->setContextParam(ParamNum, Param);
43930b57cec5SDimitry Andric   }
43940b57cec5SDimitry Andric   // Enter the capturing scope for this captured region.
4395*a7dea167SDimitry Andric   PushCapturedRegionScope(CurScope, CD, RD, Kind, OpenMPCaptureLevel);
43960b57cec5SDimitry Andric 
43970b57cec5SDimitry Andric   if (CurScope)
43980b57cec5SDimitry Andric     PushDeclContext(CurScope, CD);
43990b57cec5SDimitry Andric   else
44000b57cec5SDimitry Andric     CurContext = CD;
44010b57cec5SDimitry Andric 
44020b57cec5SDimitry Andric   PushExpressionEvaluationContext(
44030b57cec5SDimitry Andric       ExpressionEvaluationContext::PotentiallyEvaluated);
44040b57cec5SDimitry Andric }
44050b57cec5SDimitry Andric 
44060b57cec5SDimitry Andric void Sema::ActOnCapturedRegionError() {
44070b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
44080b57cec5SDimitry Andric   PopExpressionEvaluationContext();
44090b57cec5SDimitry Andric   PopDeclContext();
44100b57cec5SDimitry Andric   PoppedFunctionScopePtr ScopeRAII = PopFunctionScopeInfo();
44110b57cec5SDimitry Andric   CapturedRegionScopeInfo *RSI = cast<CapturedRegionScopeInfo>(ScopeRAII.get());
44120b57cec5SDimitry Andric 
44130b57cec5SDimitry Andric   RecordDecl *Record = RSI->TheRecordDecl;
44140b57cec5SDimitry Andric   Record->setInvalidDecl();
44150b57cec5SDimitry Andric 
44160b57cec5SDimitry Andric   SmallVector<Decl*, 4> Fields(Record->fields());
44170b57cec5SDimitry Andric   ActOnFields(/*Scope=*/nullptr, Record->getLocation(), Record, Fields,
44180b57cec5SDimitry Andric               SourceLocation(), SourceLocation(), ParsedAttributesView());
44190b57cec5SDimitry Andric }
44200b57cec5SDimitry Andric 
44210b57cec5SDimitry Andric StmtResult Sema::ActOnCapturedRegionEnd(Stmt *S) {
44220b57cec5SDimitry Andric   // Leave the captured scope before we start creating captures in the
44230b57cec5SDimitry Andric   // enclosing scope.
44240b57cec5SDimitry Andric   DiscardCleanupsInEvaluationContext();
44250b57cec5SDimitry Andric   PopExpressionEvaluationContext();
44260b57cec5SDimitry Andric   PopDeclContext();
44270b57cec5SDimitry Andric   PoppedFunctionScopePtr ScopeRAII = PopFunctionScopeInfo();
44280b57cec5SDimitry Andric   CapturedRegionScopeInfo *RSI = cast<CapturedRegionScopeInfo>(ScopeRAII.get());
44290b57cec5SDimitry Andric 
44300b57cec5SDimitry Andric   SmallVector<CapturedStmt::Capture, 4> Captures;
44310b57cec5SDimitry Andric   SmallVector<Expr *, 4> CaptureInits;
44320b57cec5SDimitry Andric   if (buildCapturedStmtCaptureList(*this, RSI, Captures, CaptureInits))
44330b57cec5SDimitry Andric     return StmtError();
44340b57cec5SDimitry Andric 
44350b57cec5SDimitry Andric   CapturedDecl *CD = RSI->TheCapturedDecl;
44360b57cec5SDimitry Andric   RecordDecl *RD = RSI->TheRecordDecl;
44370b57cec5SDimitry Andric 
44380b57cec5SDimitry Andric   CapturedStmt *Res = CapturedStmt::Create(
44390b57cec5SDimitry Andric       getASTContext(), S, static_cast<CapturedRegionKind>(RSI->CapRegionKind),
44400b57cec5SDimitry Andric       Captures, CaptureInits, CD, RD);
44410b57cec5SDimitry Andric 
44420b57cec5SDimitry Andric   CD->setBody(Res->getCapturedStmt());
44430b57cec5SDimitry Andric   RD->completeDefinition();
44440b57cec5SDimitry Andric 
44450b57cec5SDimitry Andric   return Res;
44460b57cec5SDimitry Andric }
4447