1 //===--- SemaStmt.cpp - Semantic Analysis for Statements ------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 //  This file implements semantic analysis for statements.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/Sema/SemaInternal.h"
15 #include "clang/AST/ASTContext.h"
16 #include "clang/AST/ASTDiagnostic.h"
17 #include "clang/AST/CharUnits.h"
18 #include "clang/AST/CXXInheritance.h"
19 #include "clang/AST/DeclObjC.h"
20 #include "clang/AST/EvaluatedExprVisitor.h"
21 #include "clang/AST/ExprCXX.h"
22 #include "clang/AST/ExprObjC.h"
23 #include "clang/AST/RecursiveASTVisitor.h"
24 #include "clang/AST/StmtCXX.h"
25 #include "clang/AST/StmtObjC.h"
26 #include "clang/AST/TypeLoc.h"
27 #include "clang/AST/TypeOrdering.h"
28 #include "clang/Basic/TargetInfo.h"
29 #include "clang/Lex/Preprocessor.h"
30 #include "clang/Sema/Initialization.h"
31 #include "clang/Sema/Lookup.h"
32 #include "clang/Sema/Scope.h"
33 #include "clang/Sema/ScopeInfo.h"
34 #include "llvm/ADT/ArrayRef.h"
35 #include "llvm/ADT/DenseMap.h"
36 #include "llvm/ADT/STLExtras.h"
37 #include "llvm/ADT/SmallPtrSet.h"
38 #include "llvm/ADT/SmallString.h"
39 #include "llvm/ADT/SmallVector.h"
40 
41 using namespace clang;
42 using namespace sema;
43 
44 StmtResult Sema::ActOnExprStmt(ExprResult FE) {
45   if (FE.isInvalid())
46     return StmtError();
47 
48   FE = ActOnFinishFullExpr(FE.get(), FE.get()->getExprLoc(),
49                            /*DiscardedValue*/ true);
50   if (FE.isInvalid())
51     return StmtError();
52 
53   // C99 6.8.3p2: The expression in an expression statement is evaluated as a
54   // void expression for its side effects.  Conversion to void allows any
55   // operand, even incomplete types.
56 
57   // Same thing in for stmt first clause (when expr) and third clause.
58   return StmtResult(FE.getAs<Stmt>());
59 }
60 
61 
62 StmtResult Sema::ActOnExprStmtError() {
63   DiscardCleanupsInEvaluationContext();
64   return StmtError();
65 }
66 
67 StmtResult Sema::ActOnNullStmt(SourceLocation SemiLoc,
68                                bool HasLeadingEmptyMacro) {
69   return new (Context) NullStmt(SemiLoc, HasLeadingEmptyMacro);
70 }
71 
72 StmtResult Sema::ActOnDeclStmt(DeclGroupPtrTy dg, SourceLocation StartLoc,
73                                SourceLocation EndLoc) {
74   DeclGroupRef DG = dg.get();
75 
76   // If we have an invalid decl, just return an error.
77   if (DG.isNull()) return StmtError();
78 
79   return new (Context) DeclStmt(DG, StartLoc, EndLoc);
80 }
81 
82 void Sema::ActOnForEachDeclStmt(DeclGroupPtrTy dg) {
83   DeclGroupRef DG = dg.get();
84 
85   // If we don't have a declaration, or we have an invalid declaration,
86   // just return.
87   if (DG.isNull() || !DG.isSingleDecl())
88     return;
89 
90   Decl *decl = DG.getSingleDecl();
91   if (!decl || decl->isInvalidDecl())
92     return;
93 
94   // Only variable declarations are permitted.
95   VarDecl *var = dyn_cast<VarDecl>(decl);
96   if (!var) {
97     Diag(decl->getLocation(), diag::err_non_variable_decl_in_for);
98     decl->setInvalidDecl();
99     return;
100   }
101 
102   // foreach variables are never actually initialized in the way that
103   // the parser came up with.
104   var->setInit(nullptr);
105 
106   // In ARC, we don't need to retain the iteration variable of a fast
107   // enumeration loop.  Rather than actually trying to catch that
108   // during declaration processing, we remove the consequences here.
109   if (getLangOpts().ObjCAutoRefCount) {
110     QualType type = var->getType();
111 
112     // Only do this if we inferred the lifetime.  Inferred lifetime
113     // will show up as a local qualifier because explicit lifetime
114     // should have shown up as an AttributedType instead.
115     if (type.getLocalQualifiers().getObjCLifetime() == Qualifiers::OCL_Strong) {
116       // Add 'const' and mark the variable as pseudo-strong.
117       var->setType(type.withConst());
118       var->setARCPseudoStrong(true);
119     }
120   }
121 }
122 
123 /// \brief Diagnose unused comparisons, both builtin and overloaded operators.
124 /// For '==' and '!=', suggest fixits for '=' or '|='.
125 ///
126 /// Adding a cast to void (or other expression wrappers) will prevent the
127 /// warning from firing.
128 static bool DiagnoseUnusedComparison(Sema &S, const Expr *E) {
129   SourceLocation Loc;
130   bool IsNotEqual, CanAssign, IsRelational;
131 
132   if (const BinaryOperator *Op = dyn_cast<BinaryOperator>(E)) {
133     if (!Op->isComparisonOp())
134       return false;
135 
136     IsRelational = Op->isRelationalOp();
137     Loc = Op->getOperatorLoc();
138     IsNotEqual = Op->getOpcode() == BO_NE;
139     CanAssign = Op->getLHS()->IgnoreParenImpCasts()->isLValue();
140   } else if (const CXXOperatorCallExpr *Op = dyn_cast<CXXOperatorCallExpr>(E)) {
141     switch (Op->getOperator()) {
142     default:
143       return false;
144     case OO_EqualEqual:
145     case OO_ExclaimEqual:
146       IsRelational = false;
147       break;
148     case OO_Less:
149     case OO_Greater:
150     case OO_GreaterEqual:
151     case OO_LessEqual:
152       IsRelational = true;
153       break;
154     }
155 
156     Loc = Op->getOperatorLoc();
157     IsNotEqual = Op->getOperator() == OO_ExclaimEqual;
158     CanAssign = Op->getArg(0)->IgnoreParenImpCasts()->isLValue();
159   } else {
160     // Not a typo-prone comparison.
161     return false;
162   }
163 
164   // Suppress warnings when the operator, suspicious as it may be, comes from
165   // a macro expansion.
166   if (S.SourceMgr.isMacroBodyExpansion(Loc))
167     return false;
168 
169   S.Diag(Loc, diag::warn_unused_comparison)
170     << (unsigned)IsRelational << (unsigned)IsNotEqual << E->getSourceRange();
171 
172   // If the LHS is a plausible entity to assign to, provide a fixit hint to
173   // correct common typos.
174   if (!IsRelational && CanAssign) {
175     if (IsNotEqual)
176       S.Diag(Loc, diag::note_inequality_comparison_to_or_assign)
177         << FixItHint::CreateReplacement(Loc, "|=");
178     else
179       S.Diag(Loc, diag::note_equality_comparison_to_assign)
180         << FixItHint::CreateReplacement(Loc, "=");
181   }
182 
183   return true;
184 }
185 
186 void Sema::DiagnoseUnusedExprResult(const Stmt *S) {
187   if (const LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
188     return DiagnoseUnusedExprResult(Label->getSubStmt());
189 
190   const Expr *E = dyn_cast_or_null<Expr>(S);
191   if (!E)
192     return;
193 
194   // If we are in an unevaluated expression context, then there can be no unused
195   // results because the results aren't expected to be used in the first place.
196   if (isUnevaluatedContext())
197     return;
198 
199   SourceLocation ExprLoc = E->IgnoreParenImpCasts()->getExprLoc();
200   // In most cases, we don't want to warn if the expression is written in a
201   // macro body, or if the macro comes from a system header. If the offending
202   // expression is a call to a function with the warn_unused_result attribute,
203   // we warn no matter the location. Because of the order in which the various
204   // checks need to happen, we factor out the macro-related test here.
205   bool ShouldSuppress =
206       SourceMgr.isMacroBodyExpansion(ExprLoc) ||
207       SourceMgr.isInSystemMacro(ExprLoc);
208 
209   const Expr *WarnExpr;
210   SourceLocation Loc;
211   SourceRange R1, R2;
212   if (!E->isUnusedResultAWarning(WarnExpr, Loc, R1, R2, Context))
213     return;
214 
215   // If this is a GNU statement expression expanded from a macro, it is probably
216   // unused because it is a function-like macro that can be used as either an
217   // expression or statement.  Don't warn, because it is almost certainly a
218   // false positive.
219   if (isa<StmtExpr>(E) && Loc.isMacroID())
220     return;
221 
222   // Check if this is the UNREFERENCED_PARAMETER from the Microsoft headers.
223   // That macro is frequently used to suppress "unused parameter" warnings,
224   // but its implementation makes clang's -Wunused-value fire.  Prevent this.
225   if (isa<ParenExpr>(E->IgnoreImpCasts()) && Loc.isMacroID()) {
226     SourceLocation SpellLoc = Loc;
227     if (findMacroSpelling(SpellLoc, "UNREFERENCED_PARAMETER"))
228       return;
229   }
230 
231   // Okay, we have an unused result.  Depending on what the base expression is,
232   // we might want to make a more specific diagnostic.  Check for one of these
233   // cases now.
234   unsigned DiagID = diag::warn_unused_expr;
235   if (const ExprWithCleanups *Temps = dyn_cast<ExprWithCleanups>(E))
236     E = Temps->getSubExpr();
237   if (const CXXBindTemporaryExpr *TempExpr = dyn_cast<CXXBindTemporaryExpr>(E))
238     E = TempExpr->getSubExpr();
239 
240   if (DiagnoseUnusedComparison(*this, E))
241     return;
242 
243   E = WarnExpr;
244   if (const CallExpr *CE = dyn_cast<CallExpr>(E)) {
245     if (E->getType()->isVoidType())
246       return;
247 
248     // If the callee has attribute pure, const, or warn_unused_result, warn with
249     // a more specific message to make it clear what is happening. If the call
250     // is written in a macro body, only warn if it has the warn_unused_result
251     // attribute.
252     if (const Decl *FD = CE->getCalleeDecl()) {
253       if (const Attr *A = isa<FunctionDecl>(FD)
254                               ? cast<FunctionDecl>(FD)->getUnusedResultAttr()
255                               : FD->getAttr<WarnUnusedResultAttr>()) {
256         Diag(Loc, diag::warn_unused_result) << A << R1 << R2;
257         return;
258       }
259       if (ShouldSuppress)
260         return;
261       if (FD->hasAttr<PureAttr>()) {
262         Diag(Loc, diag::warn_unused_call) << R1 << R2 << "pure";
263         return;
264       }
265       if (FD->hasAttr<ConstAttr>()) {
266         Diag(Loc, diag::warn_unused_call) << R1 << R2 << "const";
267         return;
268       }
269     }
270   } else if (ShouldSuppress)
271     return;
272 
273   if (const ObjCMessageExpr *ME = dyn_cast<ObjCMessageExpr>(E)) {
274     if (getLangOpts().ObjCAutoRefCount && ME->isDelegateInitCall()) {
275       Diag(Loc, diag::err_arc_unused_init_message) << R1;
276       return;
277     }
278     const ObjCMethodDecl *MD = ME->getMethodDecl();
279     if (MD) {
280       if (const auto *A = MD->getAttr<WarnUnusedResultAttr>()) {
281         Diag(Loc, diag::warn_unused_result) << A << R1 << R2;
282         return;
283       }
284     }
285   } else if (const PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E)) {
286     const Expr *Source = POE->getSyntacticForm();
287     if (isa<ObjCSubscriptRefExpr>(Source))
288       DiagID = diag::warn_unused_container_subscript_expr;
289     else
290       DiagID = diag::warn_unused_property_expr;
291   } else if (const CXXFunctionalCastExpr *FC
292                                        = dyn_cast<CXXFunctionalCastExpr>(E)) {
293     if (isa<CXXConstructExpr>(FC->getSubExpr()) ||
294         isa<CXXTemporaryObjectExpr>(FC->getSubExpr()))
295       return;
296   }
297   // Diagnose "(void*) blah" as a typo for "(void) blah".
298   else if (const CStyleCastExpr *CE = dyn_cast<CStyleCastExpr>(E)) {
299     TypeSourceInfo *TI = CE->getTypeInfoAsWritten();
300     QualType T = TI->getType();
301 
302     // We really do want to use the non-canonical type here.
303     if (T == Context.VoidPtrTy) {
304       PointerTypeLoc TL = TI->getTypeLoc().castAs<PointerTypeLoc>();
305 
306       Diag(Loc, diag::warn_unused_voidptr)
307         << FixItHint::CreateRemoval(TL.getStarLoc());
308       return;
309     }
310   }
311 
312   if (E->isGLValue() && E->getType().isVolatileQualified()) {
313     Diag(Loc, diag::warn_unused_volatile) << R1 << R2;
314     return;
315   }
316 
317   DiagRuntimeBehavior(Loc, nullptr, PDiag(DiagID) << R1 << R2);
318 }
319 
320 void Sema::ActOnStartOfCompoundStmt() {
321   PushCompoundScope();
322 }
323 
324 void Sema::ActOnFinishOfCompoundStmt() {
325   PopCompoundScope();
326 }
327 
328 sema::CompoundScopeInfo &Sema::getCurCompoundScope() const {
329   return getCurFunction()->CompoundScopes.back();
330 }
331 
332 StmtResult Sema::ActOnCompoundStmt(SourceLocation L, SourceLocation R,
333                                    ArrayRef<Stmt *> Elts, bool isStmtExpr) {
334   const unsigned NumElts = Elts.size();
335 
336   // If we're in C89 mode, check that we don't have any decls after stmts.  If
337   // so, emit an extension diagnostic.
338   if (!getLangOpts().C99 && !getLangOpts().CPlusPlus) {
339     // Note that __extension__ can be around a decl.
340     unsigned i = 0;
341     // Skip over all declarations.
342     for (; i != NumElts && isa<DeclStmt>(Elts[i]); ++i)
343       /*empty*/;
344 
345     // We found the end of the list or a statement.  Scan for another declstmt.
346     for (; i != NumElts && !isa<DeclStmt>(Elts[i]); ++i)
347       /*empty*/;
348 
349     if (i != NumElts) {
350       Decl *D = *cast<DeclStmt>(Elts[i])->decl_begin();
351       Diag(D->getLocation(), diag::ext_mixed_decls_code);
352     }
353   }
354   // Warn about unused expressions in statements.
355   for (unsigned i = 0; i != NumElts; ++i) {
356     // Ignore statements that are last in a statement expression.
357     if (isStmtExpr && i == NumElts - 1)
358       continue;
359 
360     DiagnoseUnusedExprResult(Elts[i]);
361   }
362 
363   // Check for suspicious empty body (null statement) in `for' and `while'
364   // statements.  Don't do anything for template instantiations, this just adds
365   // noise.
366   if (NumElts != 0 && !CurrentInstantiationScope &&
367       getCurCompoundScope().HasEmptyLoopBodies) {
368     for (unsigned i = 0; i != NumElts - 1; ++i)
369       DiagnoseEmptyLoopBody(Elts[i], Elts[i + 1]);
370   }
371 
372   return new (Context) CompoundStmt(Context, Elts, L, R);
373 }
374 
375 StmtResult
376 Sema::ActOnCaseStmt(SourceLocation CaseLoc, Expr *LHSVal,
377                     SourceLocation DotDotDotLoc, Expr *RHSVal,
378                     SourceLocation ColonLoc) {
379   assert(LHSVal && "missing expression in case statement");
380 
381   if (getCurFunction()->SwitchStack.empty()) {
382     Diag(CaseLoc, diag::err_case_not_in_switch);
383     return StmtError();
384   }
385 
386   ExprResult LHS =
387       CorrectDelayedTyposInExpr(LHSVal, [this](class Expr *E) {
388         if (!getLangOpts().CPlusPlus11)
389           return VerifyIntegerConstantExpression(E);
390         if (Expr *CondExpr =
391                 getCurFunction()->SwitchStack.back()->getCond()) {
392           QualType CondType = CondExpr->getType();
393           llvm::APSInt TempVal;
394           return CheckConvertedConstantExpression(E, CondType, TempVal,
395                                                         CCEK_CaseValue);
396         }
397         return ExprError();
398       });
399   if (LHS.isInvalid())
400     return StmtError();
401   LHSVal = LHS.get();
402 
403   if (!getLangOpts().CPlusPlus11) {
404     // C99 6.8.4.2p3: The expression shall be an integer constant.
405     // However, GCC allows any evaluatable integer expression.
406     if (!LHSVal->isTypeDependent() && !LHSVal->isValueDependent()) {
407       LHSVal = VerifyIntegerConstantExpression(LHSVal).get();
408       if (!LHSVal)
409         return StmtError();
410     }
411 
412     // GCC extension: The expression shall be an integer constant.
413 
414     if (RHSVal && !RHSVal->isTypeDependent() && !RHSVal->isValueDependent()) {
415       RHSVal = VerifyIntegerConstantExpression(RHSVal).get();
416       // Recover from an error by just forgetting about it.
417     }
418   }
419 
420   LHS = ActOnFinishFullExpr(LHSVal, LHSVal->getExprLoc(), false,
421                                  getLangOpts().CPlusPlus11);
422   if (LHS.isInvalid())
423     return StmtError();
424 
425   auto RHS = RHSVal ? ActOnFinishFullExpr(RHSVal, RHSVal->getExprLoc(), false,
426                                           getLangOpts().CPlusPlus11)
427                     : ExprResult();
428   if (RHS.isInvalid())
429     return StmtError();
430 
431   CaseStmt *CS = new (Context)
432       CaseStmt(LHS.get(), RHS.get(), CaseLoc, DotDotDotLoc, ColonLoc);
433   getCurFunction()->SwitchStack.back()->addSwitchCase(CS);
434   return CS;
435 }
436 
437 /// ActOnCaseStmtBody - This installs a statement as the body of a case.
438 void Sema::ActOnCaseStmtBody(Stmt *caseStmt, Stmt *SubStmt) {
439   DiagnoseUnusedExprResult(SubStmt);
440 
441   CaseStmt *CS = static_cast<CaseStmt*>(caseStmt);
442   CS->setSubStmt(SubStmt);
443 }
444 
445 StmtResult
446 Sema::ActOnDefaultStmt(SourceLocation DefaultLoc, SourceLocation ColonLoc,
447                        Stmt *SubStmt, Scope *CurScope) {
448   DiagnoseUnusedExprResult(SubStmt);
449 
450   if (getCurFunction()->SwitchStack.empty()) {
451     Diag(DefaultLoc, diag::err_default_not_in_switch);
452     return SubStmt;
453   }
454 
455   DefaultStmt *DS = new (Context) DefaultStmt(DefaultLoc, ColonLoc, SubStmt);
456   getCurFunction()->SwitchStack.back()->addSwitchCase(DS);
457   return DS;
458 }
459 
460 StmtResult
461 Sema::ActOnLabelStmt(SourceLocation IdentLoc, LabelDecl *TheDecl,
462                      SourceLocation ColonLoc, Stmt *SubStmt) {
463   // If the label was multiply defined, reject it now.
464   if (TheDecl->getStmt()) {
465     Diag(IdentLoc, diag::err_redefinition_of_label) << TheDecl->getDeclName();
466     Diag(TheDecl->getLocation(), diag::note_previous_definition);
467     return SubStmt;
468   }
469 
470   // Otherwise, things are good.  Fill in the declaration and return it.
471   LabelStmt *LS = new (Context) LabelStmt(IdentLoc, TheDecl, SubStmt);
472   TheDecl->setStmt(LS);
473   if (!TheDecl->isGnuLocal()) {
474     TheDecl->setLocStart(IdentLoc);
475     if (!TheDecl->isMSAsmLabel()) {
476       // Don't update the location of MS ASM labels.  These will result in
477       // a diagnostic, and changing the location here will mess that up.
478       TheDecl->setLocation(IdentLoc);
479     }
480   }
481   return LS;
482 }
483 
484 StmtResult Sema::ActOnAttributedStmt(SourceLocation AttrLoc,
485                                      ArrayRef<const Attr*> Attrs,
486                                      Stmt *SubStmt) {
487   // Fill in the declaration and return it.
488   AttributedStmt *LS = AttributedStmt::Create(Context, AttrLoc, Attrs, SubStmt);
489   return LS;
490 }
491 
492 namespace {
493 class CommaVisitor : public EvaluatedExprVisitor<CommaVisitor> {
494   typedef EvaluatedExprVisitor<CommaVisitor> Inherited;
495   Sema &SemaRef;
496 public:
497   CommaVisitor(Sema &SemaRef) : Inherited(SemaRef.Context), SemaRef(SemaRef) {}
498   void VisitBinaryOperator(BinaryOperator *E) {
499     if (E->getOpcode() == BO_Comma)
500       SemaRef.DiagnoseCommaOperator(E->getLHS(), E->getExprLoc());
501     EvaluatedExprVisitor<CommaVisitor>::VisitBinaryOperator(E);
502   }
503 };
504 }
505 
506 StmtResult
507 Sema::ActOnIfStmt(SourceLocation IfLoc, FullExprArg CondVal, Decl *CondVar,
508                   Stmt *thenStmt, SourceLocation ElseLoc,
509                   Stmt *elseStmt) {
510   ExprResult CondResult(CondVal.release());
511 
512   VarDecl *ConditionVar = nullptr;
513   if (CondVar) {
514     ConditionVar = cast<VarDecl>(CondVar);
515     CondResult = CheckConditionVariable(ConditionVar, IfLoc, true);
516     CondResult = ActOnFinishFullExpr(CondResult.get(), IfLoc);
517   }
518   Expr *ConditionExpr = CondResult.getAs<Expr>();
519   if (ConditionExpr) {
520 
521     if (!Diags.isIgnored(diag::warn_comma_operator,
522                          ConditionExpr->getExprLoc()))
523       CommaVisitor(*this).Visit(ConditionExpr);
524 
525     DiagnoseUnusedExprResult(thenStmt);
526 
527     if (!elseStmt) {
528       DiagnoseEmptyStmtBody(ConditionExpr->getLocEnd(), thenStmt,
529                             diag::warn_empty_if_body);
530     }
531 
532     DiagnoseUnusedExprResult(elseStmt);
533   } else {
534     // Create a dummy Expr for the condition for error recovery
535     ConditionExpr = new (Context) OpaqueValueExpr(SourceLocation(),
536                                                   Context.BoolTy, VK_RValue);
537   }
538 
539   return new (Context) IfStmt(Context, IfLoc, ConditionVar, ConditionExpr,
540                               thenStmt, ElseLoc, elseStmt);
541 }
542 
543 namespace {
544   struct CaseCompareFunctor {
545     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
546                     const llvm::APSInt &RHS) {
547       return LHS.first < RHS;
548     }
549     bool operator()(const std::pair<llvm::APSInt, CaseStmt*> &LHS,
550                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
551       return LHS.first < RHS.first;
552     }
553     bool operator()(const llvm::APSInt &LHS,
554                     const std::pair<llvm::APSInt, CaseStmt*> &RHS) {
555       return LHS < RHS.first;
556     }
557   };
558 }
559 
560 /// CmpCaseVals - Comparison predicate for sorting case values.
561 ///
562 static bool CmpCaseVals(const std::pair<llvm::APSInt, CaseStmt*>& lhs,
563                         const std::pair<llvm::APSInt, CaseStmt*>& rhs) {
564   if (lhs.first < rhs.first)
565     return true;
566 
567   if (lhs.first == rhs.first &&
568       lhs.second->getCaseLoc().getRawEncoding()
569        < rhs.second->getCaseLoc().getRawEncoding())
570     return true;
571   return false;
572 }
573 
574 /// CmpEnumVals - Comparison predicate for sorting enumeration values.
575 ///
576 static bool CmpEnumVals(const std::pair<llvm::APSInt, EnumConstantDecl*>& lhs,
577                         const std::pair<llvm::APSInt, EnumConstantDecl*>& rhs)
578 {
579   return lhs.first < rhs.first;
580 }
581 
582 /// EqEnumVals - Comparison preficate for uniqing enumeration values.
583 ///
584 static bool EqEnumVals(const std::pair<llvm::APSInt, EnumConstantDecl*>& lhs,
585                        const std::pair<llvm::APSInt, EnumConstantDecl*>& rhs)
586 {
587   return lhs.first == rhs.first;
588 }
589 
590 /// GetTypeBeforeIntegralPromotion - Returns the pre-promotion type of
591 /// potentially integral-promoted expression @p expr.
592 static QualType GetTypeBeforeIntegralPromotion(Expr *&expr) {
593   if (ExprWithCleanups *cleanups = dyn_cast<ExprWithCleanups>(expr))
594     expr = cleanups->getSubExpr();
595   while (ImplicitCastExpr *impcast = dyn_cast<ImplicitCastExpr>(expr)) {
596     if (impcast->getCastKind() != CK_IntegralCast) break;
597     expr = impcast->getSubExpr();
598   }
599   return expr->getType();
600 }
601 
602 StmtResult
603 Sema::ActOnStartOfSwitchStmt(SourceLocation SwitchLoc, Expr *Cond,
604                              Decl *CondVar) {
605   ExprResult CondResult;
606 
607   VarDecl *ConditionVar = nullptr;
608   if (CondVar) {
609     ConditionVar = cast<VarDecl>(CondVar);
610     CondResult = CheckConditionVariable(ConditionVar, SourceLocation(), false);
611     if (CondResult.isInvalid())
612       return StmtError();
613 
614     Cond = CondResult.get();
615   }
616 
617   if (!Cond)
618     return StmtError();
619 
620   class SwitchConvertDiagnoser : public ICEConvertDiagnoser {
621     Expr *Cond;
622 
623   public:
624     SwitchConvertDiagnoser(Expr *Cond)
625         : ICEConvertDiagnoser(/*AllowScopedEnumerations*/true, false, true),
626           Cond(Cond) {}
627 
628     SemaDiagnosticBuilder diagnoseNotInt(Sema &S, SourceLocation Loc,
629                                          QualType T) override {
630       return S.Diag(Loc, diag::err_typecheck_statement_requires_integer) << T;
631     }
632 
633     SemaDiagnosticBuilder diagnoseIncomplete(
634         Sema &S, SourceLocation Loc, QualType T) override {
635       return S.Diag(Loc, diag::err_switch_incomplete_class_type)
636                << T << Cond->getSourceRange();
637     }
638 
639     SemaDiagnosticBuilder diagnoseExplicitConv(
640         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
641       return S.Diag(Loc, diag::err_switch_explicit_conversion) << T << ConvTy;
642     }
643 
644     SemaDiagnosticBuilder noteExplicitConv(
645         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
646       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
647         << ConvTy->isEnumeralType() << ConvTy;
648     }
649 
650     SemaDiagnosticBuilder diagnoseAmbiguous(Sema &S, SourceLocation Loc,
651                                             QualType T) override {
652       return S.Diag(Loc, diag::err_switch_multiple_conversions) << T;
653     }
654 
655     SemaDiagnosticBuilder noteAmbiguous(
656         Sema &S, CXXConversionDecl *Conv, QualType ConvTy) override {
657       return S.Diag(Conv->getLocation(), diag::note_switch_conversion)
658       << ConvTy->isEnumeralType() << ConvTy;
659     }
660 
661     SemaDiagnosticBuilder diagnoseConversion(
662         Sema &S, SourceLocation Loc, QualType T, QualType ConvTy) override {
663       llvm_unreachable("conversion functions are permitted");
664     }
665   } SwitchDiagnoser(Cond);
666 
667   CondResult =
668       PerformContextualImplicitConversion(SwitchLoc, Cond, SwitchDiagnoser);
669   if (CondResult.isInvalid()) return StmtError();
670   Cond = CondResult.get();
671 
672   // C99 6.8.4.2p5 - Integer promotions are performed on the controlling expr.
673   CondResult = UsualUnaryConversions(Cond);
674   if (CondResult.isInvalid()) return StmtError();
675   Cond = CondResult.get();
676 
677   CondResult = ActOnFinishFullExpr(Cond, SwitchLoc);
678   if (CondResult.isInvalid())
679     return StmtError();
680   Cond = CondResult.get();
681 
682   getCurFunction()->setHasBranchIntoScope();
683 
684   SwitchStmt *SS = new (Context) SwitchStmt(Context, ConditionVar, Cond);
685   getCurFunction()->SwitchStack.push_back(SS);
686   return SS;
687 }
688 
689 static void AdjustAPSInt(llvm::APSInt &Val, unsigned BitWidth, bool IsSigned) {
690   Val = Val.extOrTrunc(BitWidth);
691   Val.setIsSigned(IsSigned);
692 }
693 
694 /// Check the specified case value is in range for the given unpromoted switch
695 /// type.
696 static void checkCaseValue(Sema &S, SourceLocation Loc, const llvm::APSInt &Val,
697                            unsigned UnpromotedWidth, bool UnpromotedSign) {
698   // If the case value was signed and negative and the switch expression is
699   // unsigned, don't bother to warn: this is implementation-defined behavior.
700   // FIXME: Introduce a second, default-ignored warning for this case?
701   if (UnpromotedWidth < Val.getBitWidth()) {
702     llvm::APSInt ConvVal(Val);
703     AdjustAPSInt(ConvVal, UnpromotedWidth, UnpromotedSign);
704     AdjustAPSInt(ConvVal, Val.getBitWidth(), Val.isSigned());
705     // FIXME: Use different diagnostics for overflow  in conversion to promoted
706     // type versus "switch expression cannot have this value". Use proper
707     // IntRange checking rather than just looking at the unpromoted type here.
708     if (ConvVal != Val)
709       S.Diag(Loc, diag::warn_case_value_overflow) << Val.toString(10)
710                                                   << ConvVal.toString(10);
711   }
712 }
713 
714 typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl*>, 64> EnumValsTy;
715 
716 /// Returns true if we should emit a diagnostic about this case expression not
717 /// being a part of the enum used in the switch controlling expression.
718 static bool ShouldDiagnoseSwitchCaseNotInEnum(const Sema &S,
719                                               const EnumDecl *ED,
720                                               const Expr *CaseExpr,
721                                               EnumValsTy::iterator &EI,
722                                               EnumValsTy::iterator &EIEnd,
723                                               const llvm::APSInt &Val) {
724   if (const DeclRefExpr *DRE =
725           dyn_cast<DeclRefExpr>(CaseExpr->IgnoreParenImpCasts())) {
726     if (const VarDecl *VD = dyn_cast<VarDecl>(DRE->getDecl())) {
727       QualType VarType = VD->getType();
728       QualType EnumType = S.Context.getTypeDeclType(ED);
729       if (VD->hasGlobalStorage() && VarType.isConstQualified() &&
730           S.Context.hasSameUnqualifiedType(EnumType, VarType))
731         return false;
732     }
733   }
734 
735   if (ED->hasAttr<FlagEnumAttr>()) {
736     return !S.IsValueInFlagEnum(ED, Val, false);
737   } else {
738     while (EI != EIEnd && EI->first < Val)
739       EI++;
740 
741     if (EI != EIEnd && EI->first == Val)
742       return false;
743   }
744 
745   return true;
746 }
747 
748 StmtResult
749 Sema::ActOnFinishSwitchStmt(SourceLocation SwitchLoc, Stmt *Switch,
750                             Stmt *BodyStmt) {
751   SwitchStmt *SS = cast<SwitchStmt>(Switch);
752   assert(SS == getCurFunction()->SwitchStack.back() &&
753          "switch stack missing push/pop!");
754 
755   getCurFunction()->SwitchStack.pop_back();
756 
757   if (!BodyStmt) return StmtError();
758   SS->setBody(BodyStmt, SwitchLoc);
759 
760   Expr *CondExpr = SS->getCond();
761   if (!CondExpr) return StmtError();
762 
763   QualType CondType = CondExpr->getType();
764 
765   Expr *CondExprBeforePromotion = CondExpr;
766   QualType CondTypeBeforePromotion =
767       GetTypeBeforeIntegralPromotion(CondExprBeforePromotion);
768 
769   // C++ 6.4.2.p2:
770   // Integral promotions are performed (on the switch condition).
771   //
772   // A case value unrepresentable by the original switch condition
773   // type (before the promotion) doesn't make sense, even when it can
774   // be represented by the promoted type.  Therefore we need to find
775   // the pre-promotion type of the switch condition.
776   if (!CondExpr->isTypeDependent()) {
777     // We have already converted the expression to an integral or enumeration
778     // type, when we started the switch statement. If we don't have an
779     // appropriate type now, just return an error.
780     if (!CondType->isIntegralOrEnumerationType())
781       return StmtError();
782 
783     if (CondExpr->isKnownToHaveBooleanValue()) {
784       // switch(bool_expr) {...} is often a programmer error, e.g.
785       //   switch(n && mask) { ... }  // Doh - should be "n & mask".
786       // One can always use an if statement instead of switch(bool_expr).
787       Diag(SwitchLoc, diag::warn_bool_switch_condition)
788           << CondExpr->getSourceRange();
789     }
790   }
791 
792   // Get the bitwidth of the switched-on value after promotions. We must
793   // convert the integer case values to this width before comparison.
794   bool HasDependentValue
795     = CondExpr->isTypeDependent() || CondExpr->isValueDependent();
796   unsigned CondWidth = HasDependentValue ? 0 : Context.getIntWidth(CondType);
797   bool CondIsSigned = CondType->isSignedIntegerOrEnumerationType();
798 
799   // Get the width and signedness that the condition might actually have, for
800   // warning purposes.
801   // FIXME: Grab an IntRange for the condition rather than using the unpromoted
802   // type.
803   unsigned CondWidthBeforePromotion
804     = HasDependentValue ? 0 : Context.getIntWidth(CondTypeBeforePromotion);
805   bool CondIsSignedBeforePromotion
806     = CondTypeBeforePromotion->isSignedIntegerOrEnumerationType();
807 
808   // Accumulate all of the case values in a vector so that we can sort them
809   // and detect duplicates.  This vector contains the APInt for the case after
810   // it has been converted to the condition type.
811   typedef SmallVector<std::pair<llvm::APSInt, CaseStmt*>, 64> CaseValsTy;
812   CaseValsTy CaseVals;
813 
814   // Keep track of any GNU case ranges we see.  The APSInt is the low value.
815   typedef std::vector<std::pair<llvm::APSInt, CaseStmt*> > CaseRangesTy;
816   CaseRangesTy CaseRanges;
817 
818   DefaultStmt *TheDefaultStmt = nullptr;
819 
820   bool CaseListIsErroneous = false;
821 
822   for (SwitchCase *SC = SS->getSwitchCaseList(); SC && !HasDependentValue;
823        SC = SC->getNextSwitchCase()) {
824 
825     if (DefaultStmt *DS = dyn_cast<DefaultStmt>(SC)) {
826       if (TheDefaultStmt) {
827         Diag(DS->getDefaultLoc(), diag::err_multiple_default_labels_defined);
828         Diag(TheDefaultStmt->getDefaultLoc(), diag::note_duplicate_case_prev);
829 
830         // FIXME: Remove the default statement from the switch block so that
831         // we'll return a valid AST.  This requires recursing down the AST and
832         // finding it, not something we are set up to do right now.  For now,
833         // just lop the entire switch stmt out of the AST.
834         CaseListIsErroneous = true;
835       }
836       TheDefaultStmt = DS;
837 
838     } else {
839       CaseStmt *CS = cast<CaseStmt>(SC);
840 
841       Expr *Lo = CS->getLHS();
842 
843       if (Lo->isTypeDependent() || Lo->isValueDependent()) {
844         HasDependentValue = true;
845         break;
846       }
847 
848       llvm::APSInt LoVal;
849 
850       if (getLangOpts().CPlusPlus11) {
851         // C++11 [stmt.switch]p2: the constant-expression shall be a converted
852         // constant expression of the promoted type of the switch condition.
853         ExprResult ConvLo =
854           CheckConvertedConstantExpression(Lo, CondType, LoVal, CCEK_CaseValue);
855         if (ConvLo.isInvalid()) {
856           CaseListIsErroneous = true;
857           continue;
858         }
859         Lo = ConvLo.get();
860       } else {
861         // We already verified that the expression has a i-c-e value (C99
862         // 6.8.4.2p3) - get that value now.
863         LoVal = Lo->EvaluateKnownConstInt(Context);
864 
865         // If the LHS is not the same type as the condition, insert an implicit
866         // cast.
867         Lo = DefaultLvalueConversion(Lo).get();
868         Lo = ImpCastExprToType(Lo, CondType, CK_IntegralCast).get();
869       }
870 
871       // Check the unconverted value is within the range of possible values of
872       // the switch expression.
873       checkCaseValue(*this, Lo->getLocStart(), LoVal,
874                      CondWidthBeforePromotion, CondIsSignedBeforePromotion);
875 
876       // Convert the value to the same width/sign as the condition.
877       AdjustAPSInt(LoVal, CondWidth, CondIsSigned);
878 
879       CS->setLHS(Lo);
880 
881       // If this is a case range, remember it in CaseRanges, otherwise CaseVals.
882       if (CS->getRHS()) {
883         if (CS->getRHS()->isTypeDependent() ||
884             CS->getRHS()->isValueDependent()) {
885           HasDependentValue = true;
886           break;
887         }
888         CaseRanges.push_back(std::make_pair(LoVal, CS));
889       } else
890         CaseVals.push_back(std::make_pair(LoVal, CS));
891     }
892   }
893 
894   if (!HasDependentValue) {
895     // If we don't have a default statement, check whether the
896     // condition is constant.
897     llvm::APSInt ConstantCondValue;
898     bool HasConstantCond = false;
899     if (!HasDependentValue && !TheDefaultStmt) {
900       HasConstantCond = CondExpr->EvaluateAsInt(ConstantCondValue, Context,
901                                                 Expr::SE_AllowSideEffects);
902       assert(!HasConstantCond ||
903              (ConstantCondValue.getBitWidth() == CondWidth &&
904               ConstantCondValue.isSigned() == CondIsSigned));
905     }
906     bool ShouldCheckConstantCond = HasConstantCond;
907 
908     // Sort all the scalar case values so we can easily detect duplicates.
909     std::stable_sort(CaseVals.begin(), CaseVals.end(), CmpCaseVals);
910 
911     if (!CaseVals.empty()) {
912       for (unsigned i = 0, e = CaseVals.size(); i != e; ++i) {
913         if (ShouldCheckConstantCond &&
914             CaseVals[i].first == ConstantCondValue)
915           ShouldCheckConstantCond = false;
916 
917         if (i != 0 && CaseVals[i].first == CaseVals[i-1].first) {
918           // If we have a duplicate, report it.
919           // First, determine if either case value has a name
920           StringRef PrevString, CurrString;
921           Expr *PrevCase = CaseVals[i-1].second->getLHS()->IgnoreParenCasts();
922           Expr *CurrCase = CaseVals[i].second->getLHS()->IgnoreParenCasts();
923           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(PrevCase)) {
924             PrevString = DeclRef->getDecl()->getName();
925           }
926           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(CurrCase)) {
927             CurrString = DeclRef->getDecl()->getName();
928           }
929           SmallString<16> CaseValStr;
930           CaseVals[i-1].first.toString(CaseValStr);
931 
932           if (PrevString == CurrString)
933             Diag(CaseVals[i].second->getLHS()->getLocStart(),
934                  diag::err_duplicate_case) <<
935                  (PrevString.empty() ? StringRef(CaseValStr) : PrevString);
936           else
937             Diag(CaseVals[i].second->getLHS()->getLocStart(),
938                  diag::err_duplicate_case_differing_expr) <<
939                  (PrevString.empty() ? StringRef(CaseValStr) : PrevString) <<
940                  (CurrString.empty() ? StringRef(CaseValStr) : CurrString) <<
941                  CaseValStr;
942 
943           Diag(CaseVals[i-1].second->getLHS()->getLocStart(),
944                diag::note_duplicate_case_prev);
945           // FIXME: We really want to remove the bogus case stmt from the
946           // substmt, but we have no way to do this right now.
947           CaseListIsErroneous = true;
948         }
949       }
950     }
951 
952     // Detect duplicate case ranges, which usually don't exist at all in
953     // the first place.
954     if (!CaseRanges.empty()) {
955       // Sort all the case ranges by their low value so we can easily detect
956       // overlaps between ranges.
957       std::stable_sort(CaseRanges.begin(), CaseRanges.end());
958 
959       // Scan the ranges, computing the high values and removing empty ranges.
960       std::vector<llvm::APSInt> HiVals;
961       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
962         llvm::APSInt &LoVal = CaseRanges[i].first;
963         CaseStmt *CR = CaseRanges[i].second;
964         Expr *Hi = CR->getRHS();
965         llvm::APSInt HiVal;
966 
967         if (getLangOpts().CPlusPlus11) {
968           // C++11 [stmt.switch]p2: the constant-expression shall be a converted
969           // constant expression of the promoted type of the switch condition.
970           ExprResult ConvHi =
971             CheckConvertedConstantExpression(Hi, CondType, HiVal,
972                                              CCEK_CaseValue);
973           if (ConvHi.isInvalid()) {
974             CaseListIsErroneous = true;
975             continue;
976           }
977           Hi = ConvHi.get();
978         } else {
979           HiVal = Hi->EvaluateKnownConstInt(Context);
980 
981           // If the RHS is not the same type as the condition, insert an
982           // implicit cast.
983           Hi = DefaultLvalueConversion(Hi).get();
984           Hi = ImpCastExprToType(Hi, CondType, CK_IntegralCast).get();
985         }
986 
987         // Check the unconverted value is within the range of possible values of
988         // the switch expression.
989         checkCaseValue(*this, Hi->getLocStart(), HiVal,
990                        CondWidthBeforePromotion, CondIsSignedBeforePromotion);
991 
992         // Convert the value to the same width/sign as the condition.
993         AdjustAPSInt(HiVal, CondWidth, CondIsSigned);
994 
995         CR->setRHS(Hi);
996 
997         // If the low value is bigger than the high value, the case is empty.
998         if (LoVal > HiVal) {
999           Diag(CR->getLHS()->getLocStart(), diag::warn_case_empty_range)
1000             << SourceRange(CR->getLHS()->getLocStart(),
1001                            Hi->getLocEnd());
1002           CaseRanges.erase(CaseRanges.begin()+i);
1003           --i;
1004           --e;
1005           continue;
1006         }
1007 
1008         if (ShouldCheckConstantCond &&
1009             LoVal <= ConstantCondValue &&
1010             ConstantCondValue <= HiVal)
1011           ShouldCheckConstantCond = false;
1012 
1013         HiVals.push_back(HiVal);
1014       }
1015 
1016       // Rescan the ranges, looking for overlap with singleton values and other
1017       // ranges.  Since the range list is sorted, we only need to compare case
1018       // ranges with their neighbors.
1019       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
1020         llvm::APSInt &CRLo = CaseRanges[i].first;
1021         llvm::APSInt &CRHi = HiVals[i];
1022         CaseStmt *CR = CaseRanges[i].second;
1023 
1024         // Check to see whether the case range overlaps with any
1025         // singleton cases.
1026         CaseStmt *OverlapStmt = nullptr;
1027         llvm::APSInt OverlapVal(32);
1028 
1029         // Find the smallest value >= the lower bound.  If I is in the
1030         // case range, then we have overlap.
1031         CaseValsTy::iterator I = std::lower_bound(CaseVals.begin(),
1032                                                   CaseVals.end(), CRLo,
1033                                                   CaseCompareFunctor());
1034         if (I != CaseVals.end() && I->first < CRHi) {
1035           OverlapVal  = I->first;   // Found overlap with scalar.
1036           OverlapStmt = I->second;
1037         }
1038 
1039         // Find the smallest value bigger than the upper bound.
1040         I = std::upper_bound(I, CaseVals.end(), CRHi, CaseCompareFunctor());
1041         if (I != CaseVals.begin() && (I-1)->first >= CRLo) {
1042           OverlapVal  = (I-1)->first;      // Found overlap with scalar.
1043           OverlapStmt = (I-1)->second;
1044         }
1045 
1046         // Check to see if this case stmt overlaps with the subsequent
1047         // case range.
1048         if (i && CRLo <= HiVals[i-1]) {
1049           OverlapVal  = HiVals[i-1];       // Found overlap with range.
1050           OverlapStmt = CaseRanges[i-1].second;
1051         }
1052 
1053         if (OverlapStmt) {
1054           // If we have a duplicate, report it.
1055           Diag(CR->getLHS()->getLocStart(), diag::err_duplicate_case)
1056             << OverlapVal.toString(10);
1057           Diag(OverlapStmt->getLHS()->getLocStart(),
1058                diag::note_duplicate_case_prev);
1059           // FIXME: We really want to remove the bogus case stmt from the
1060           // substmt, but we have no way to do this right now.
1061           CaseListIsErroneous = true;
1062         }
1063       }
1064     }
1065 
1066     // Complain if we have a constant condition and we didn't find a match.
1067     if (!CaseListIsErroneous && ShouldCheckConstantCond) {
1068       // TODO: it would be nice if we printed enums as enums, chars as
1069       // chars, etc.
1070       Diag(CondExpr->getExprLoc(), diag::warn_missing_case_for_condition)
1071         << ConstantCondValue.toString(10)
1072         << CondExpr->getSourceRange();
1073     }
1074 
1075     // Check to see if switch is over an Enum and handles all of its
1076     // values.  We only issue a warning if there is not 'default:', but
1077     // we still do the analysis to preserve this information in the AST
1078     // (which can be used by flow-based analyes).
1079     //
1080     const EnumType *ET = CondTypeBeforePromotion->getAs<EnumType>();
1081 
1082     // If switch has default case, then ignore it.
1083     if (!CaseListIsErroneous  && !HasConstantCond && ET) {
1084       const EnumDecl *ED = ET->getDecl();
1085       EnumValsTy EnumVals;
1086 
1087       // Gather all enum values, set their type and sort them,
1088       // allowing easier comparison with CaseVals.
1089       for (auto *EDI : ED->enumerators()) {
1090         llvm::APSInt Val = EDI->getInitVal();
1091         AdjustAPSInt(Val, CondWidth, CondIsSigned);
1092         EnumVals.push_back(std::make_pair(Val, EDI));
1093       }
1094       std::stable_sort(EnumVals.begin(), EnumVals.end(), CmpEnumVals);
1095       auto EI = EnumVals.begin(), EIEnd =
1096         std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
1097 
1098       // See which case values aren't in enum.
1099       for (CaseValsTy::const_iterator CI = CaseVals.begin();
1100           CI != CaseVals.end(); CI++) {
1101         Expr *CaseExpr = CI->second->getLHS();
1102         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
1103                                               CI->first))
1104           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
1105             << CondTypeBeforePromotion;
1106       }
1107 
1108       // See which of case ranges aren't in enum
1109       EI = EnumVals.begin();
1110       for (CaseRangesTy::const_iterator RI = CaseRanges.begin();
1111           RI != CaseRanges.end(); RI++) {
1112         Expr *CaseExpr = RI->second->getLHS();
1113         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
1114                                               RI->first))
1115           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
1116             << CondTypeBeforePromotion;
1117 
1118         llvm::APSInt Hi =
1119           RI->second->getRHS()->EvaluateKnownConstInt(Context);
1120         AdjustAPSInt(Hi, CondWidth, CondIsSigned);
1121 
1122         CaseExpr = RI->second->getRHS();
1123         if (ShouldDiagnoseSwitchCaseNotInEnum(*this, ED, CaseExpr, EI, EIEnd,
1124                                               Hi))
1125           Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
1126             << CondTypeBeforePromotion;
1127       }
1128 
1129       // Check which enum vals aren't in switch
1130       auto CI = CaseVals.begin();
1131       auto RI = CaseRanges.begin();
1132       bool hasCasesNotInSwitch = false;
1133 
1134       SmallVector<DeclarationName,8> UnhandledNames;
1135 
1136       for (EI = EnumVals.begin(); EI != EIEnd; EI++){
1137         // Drop unneeded case values
1138         while (CI != CaseVals.end() && CI->first < EI->first)
1139           CI++;
1140 
1141         if (CI != CaseVals.end() && CI->first == EI->first)
1142           continue;
1143 
1144         // Drop unneeded case ranges
1145         for (; RI != CaseRanges.end(); RI++) {
1146           llvm::APSInt Hi =
1147             RI->second->getRHS()->EvaluateKnownConstInt(Context);
1148           AdjustAPSInt(Hi, CondWidth, CondIsSigned);
1149           if (EI->first <= Hi)
1150             break;
1151         }
1152 
1153         if (RI == CaseRanges.end() || EI->first < RI->first) {
1154           hasCasesNotInSwitch = true;
1155           UnhandledNames.push_back(EI->second->getDeclName());
1156         }
1157       }
1158 
1159       if (TheDefaultStmt && UnhandledNames.empty())
1160         Diag(TheDefaultStmt->getDefaultLoc(), diag::warn_unreachable_default);
1161 
1162       // Produce a nice diagnostic if multiple values aren't handled.
1163       if (!UnhandledNames.empty()) {
1164         DiagnosticBuilder DB = Diag(CondExpr->getExprLoc(),
1165                                     TheDefaultStmt ? diag::warn_def_missing_case
1166                                                    : diag::warn_missing_case)
1167                                << (int)UnhandledNames.size();
1168 
1169         for (size_t I = 0, E = std::min(UnhandledNames.size(), (size_t)3);
1170              I != E; ++I)
1171           DB << UnhandledNames[I];
1172       }
1173 
1174       if (!hasCasesNotInSwitch)
1175         SS->setAllEnumCasesCovered();
1176     }
1177   }
1178 
1179   if (BodyStmt)
1180     DiagnoseEmptyStmtBody(CondExpr->getLocEnd(), BodyStmt,
1181                           diag::warn_empty_switch_body);
1182 
1183   // FIXME: If the case list was broken is some way, we don't have a good system
1184   // to patch it up.  Instead, just return the whole substmt as broken.
1185   if (CaseListIsErroneous)
1186     return StmtError();
1187 
1188   return SS;
1189 }
1190 
1191 void
1192 Sema::DiagnoseAssignmentEnum(QualType DstType, QualType SrcType,
1193                              Expr *SrcExpr) {
1194   if (Diags.isIgnored(diag::warn_not_in_enum_assignment, SrcExpr->getExprLoc()))
1195     return;
1196 
1197   if (const EnumType *ET = DstType->getAs<EnumType>())
1198     if (!Context.hasSameUnqualifiedType(SrcType, DstType) &&
1199         SrcType->isIntegerType()) {
1200       if (!SrcExpr->isTypeDependent() && !SrcExpr->isValueDependent() &&
1201           SrcExpr->isIntegerConstantExpr(Context)) {
1202         // Get the bitwidth of the enum value before promotions.
1203         unsigned DstWidth = Context.getIntWidth(DstType);
1204         bool DstIsSigned = DstType->isSignedIntegerOrEnumerationType();
1205 
1206         llvm::APSInt RhsVal = SrcExpr->EvaluateKnownConstInt(Context);
1207         AdjustAPSInt(RhsVal, DstWidth, DstIsSigned);
1208         const EnumDecl *ED = ET->getDecl();
1209 
1210         if (ED->hasAttr<FlagEnumAttr>()) {
1211           if (!IsValueInFlagEnum(ED, RhsVal, true))
1212             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
1213               << DstType.getUnqualifiedType();
1214         } else {
1215           typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl *>, 64>
1216               EnumValsTy;
1217           EnumValsTy EnumVals;
1218 
1219           // Gather all enum values, set their type and sort them,
1220           // allowing easier comparison with rhs constant.
1221           for (auto *EDI : ED->enumerators()) {
1222             llvm::APSInt Val = EDI->getInitVal();
1223             AdjustAPSInt(Val, DstWidth, DstIsSigned);
1224             EnumVals.push_back(std::make_pair(Val, EDI));
1225           }
1226           if (EnumVals.empty())
1227             return;
1228           std::stable_sort(EnumVals.begin(), EnumVals.end(), CmpEnumVals);
1229           EnumValsTy::iterator EIend =
1230               std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
1231 
1232           // See which values aren't in the enum.
1233           EnumValsTy::const_iterator EI = EnumVals.begin();
1234           while (EI != EIend && EI->first < RhsVal)
1235             EI++;
1236           if (EI == EIend || EI->first != RhsVal) {
1237             Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
1238                 << DstType.getUnqualifiedType();
1239           }
1240         }
1241       }
1242     }
1243 }
1244 
1245 StmtResult
1246 Sema::ActOnWhileStmt(SourceLocation WhileLoc, FullExprArg Cond,
1247                      Decl *CondVar, Stmt *Body) {
1248   ExprResult CondResult(Cond.release());
1249 
1250   VarDecl *ConditionVar = nullptr;
1251   if (CondVar) {
1252     ConditionVar = cast<VarDecl>(CondVar);
1253     CondResult = CheckConditionVariable(ConditionVar, WhileLoc, true);
1254     CondResult = ActOnFinishFullExpr(CondResult.get(), WhileLoc);
1255     if (CondResult.isInvalid())
1256       return StmtError();
1257   }
1258   Expr *ConditionExpr = CondResult.get();
1259   if (!ConditionExpr)
1260     return StmtError();
1261   CheckBreakContinueBinding(ConditionExpr);
1262 
1263   if (ConditionExpr &&
1264       !Diags.isIgnored(diag::warn_comma_operator, ConditionExpr->getExprLoc()))
1265     CommaVisitor(*this).Visit(ConditionExpr);
1266 
1267   DiagnoseUnusedExprResult(Body);
1268 
1269   if (isa<NullStmt>(Body))
1270     getCurCompoundScope().setHasEmptyLoopBodies();
1271 
1272   return new (Context)
1273       WhileStmt(Context, ConditionVar, ConditionExpr, Body, WhileLoc);
1274 }
1275 
1276 StmtResult
1277 Sema::ActOnDoStmt(SourceLocation DoLoc, Stmt *Body,
1278                   SourceLocation WhileLoc, SourceLocation CondLParen,
1279                   Expr *Cond, SourceLocation CondRParen) {
1280   assert(Cond && "ActOnDoStmt(): missing expression");
1281 
1282   CheckBreakContinueBinding(Cond);
1283   ExprResult CondResult = CheckBooleanCondition(Cond, DoLoc);
1284   if (CondResult.isInvalid())
1285     return StmtError();
1286   Cond = CondResult.get();
1287 
1288   CondResult = ActOnFinishFullExpr(Cond, DoLoc);
1289   if (CondResult.isInvalid())
1290     return StmtError();
1291   Cond = CondResult.get();
1292 
1293   DiagnoseUnusedExprResult(Body);
1294 
1295   return new (Context) DoStmt(Body, Cond, DoLoc, WhileLoc, CondRParen);
1296 }
1297 
1298 namespace {
1299   // This visitor will traverse a conditional statement and store all
1300   // the evaluated decls into a vector.  Simple is set to true if none
1301   // of the excluded constructs are used.
1302   class DeclExtractor : public EvaluatedExprVisitor<DeclExtractor> {
1303     llvm::SmallPtrSetImpl<VarDecl*> &Decls;
1304     SmallVectorImpl<SourceRange> &Ranges;
1305     bool Simple;
1306   public:
1307     typedef EvaluatedExprVisitor<DeclExtractor> Inherited;
1308 
1309     DeclExtractor(Sema &S, llvm::SmallPtrSetImpl<VarDecl*> &Decls,
1310                   SmallVectorImpl<SourceRange> &Ranges) :
1311         Inherited(S.Context),
1312         Decls(Decls),
1313         Ranges(Ranges),
1314         Simple(true) {}
1315 
1316     bool isSimple() { return Simple; }
1317 
1318     // Replaces the method in EvaluatedExprVisitor.
1319     void VisitMemberExpr(MemberExpr* E) {
1320       Simple = false;
1321     }
1322 
1323     // Any Stmt not whitelisted will cause the condition to be marked complex.
1324     void VisitStmt(Stmt *S) {
1325       Simple = false;
1326     }
1327 
1328     void VisitBinaryOperator(BinaryOperator *E) {
1329       Visit(E->getLHS());
1330       Visit(E->getRHS());
1331     }
1332 
1333     void VisitCastExpr(CastExpr *E) {
1334       Visit(E->getSubExpr());
1335     }
1336 
1337     void VisitUnaryOperator(UnaryOperator *E) {
1338       // Skip checking conditionals with derefernces.
1339       if (E->getOpcode() == UO_Deref)
1340         Simple = false;
1341       else
1342         Visit(E->getSubExpr());
1343     }
1344 
1345     void VisitConditionalOperator(ConditionalOperator *E) {
1346       Visit(E->getCond());
1347       Visit(E->getTrueExpr());
1348       Visit(E->getFalseExpr());
1349     }
1350 
1351     void VisitParenExpr(ParenExpr *E) {
1352       Visit(E->getSubExpr());
1353     }
1354 
1355     void VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
1356       Visit(E->getOpaqueValue()->getSourceExpr());
1357       Visit(E->getFalseExpr());
1358     }
1359 
1360     void VisitIntegerLiteral(IntegerLiteral *E) { }
1361     void VisitFloatingLiteral(FloatingLiteral *E) { }
1362     void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) { }
1363     void VisitCharacterLiteral(CharacterLiteral *E) { }
1364     void VisitGNUNullExpr(GNUNullExpr *E) { }
1365     void VisitImaginaryLiteral(ImaginaryLiteral *E) { }
1366 
1367     void VisitDeclRefExpr(DeclRefExpr *E) {
1368       VarDecl *VD = dyn_cast<VarDecl>(E->getDecl());
1369       if (!VD) return;
1370 
1371       Ranges.push_back(E->getSourceRange());
1372 
1373       Decls.insert(VD);
1374     }
1375 
1376   }; // end class DeclExtractor
1377 
1378   // DeclMatcher checks to see if the decls are used in a non-evaluated
1379   // context.
1380   class DeclMatcher : public EvaluatedExprVisitor<DeclMatcher> {
1381     llvm::SmallPtrSetImpl<VarDecl*> &Decls;
1382     bool FoundDecl;
1383 
1384   public:
1385     typedef EvaluatedExprVisitor<DeclMatcher> Inherited;
1386 
1387     DeclMatcher(Sema &S, llvm::SmallPtrSetImpl<VarDecl*> &Decls,
1388                 Stmt *Statement) :
1389         Inherited(S.Context), Decls(Decls), FoundDecl(false) {
1390       if (!Statement) return;
1391 
1392       Visit(Statement);
1393     }
1394 
1395     void VisitReturnStmt(ReturnStmt *S) {
1396       FoundDecl = true;
1397     }
1398 
1399     void VisitBreakStmt(BreakStmt *S) {
1400       FoundDecl = true;
1401     }
1402 
1403     void VisitGotoStmt(GotoStmt *S) {
1404       FoundDecl = true;
1405     }
1406 
1407     void VisitCastExpr(CastExpr *E) {
1408       if (E->getCastKind() == CK_LValueToRValue)
1409         CheckLValueToRValueCast(E->getSubExpr());
1410       else
1411         Visit(E->getSubExpr());
1412     }
1413 
1414     void CheckLValueToRValueCast(Expr *E) {
1415       E = E->IgnoreParenImpCasts();
1416 
1417       if (isa<DeclRefExpr>(E)) {
1418         return;
1419       }
1420 
1421       if (ConditionalOperator *CO = dyn_cast<ConditionalOperator>(E)) {
1422         Visit(CO->getCond());
1423         CheckLValueToRValueCast(CO->getTrueExpr());
1424         CheckLValueToRValueCast(CO->getFalseExpr());
1425         return;
1426       }
1427 
1428       if (BinaryConditionalOperator *BCO =
1429               dyn_cast<BinaryConditionalOperator>(E)) {
1430         CheckLValueToRValueCast(BCO->getOpaqueValue()->getSourceExpr());
1431         CheckLValueToRValueCast(BCO->getFalseExpr());
1432         return;
1433       }
1434 
1435       Visit(E);
1436     }
1437 
1438     void VisitDeclRefExpr(DeclRefExpr *E) {
1439       if (VarDecl *VD = dyn_cast<VarDecl>(E->getDecl()))
1440         if (Decls.count(VD))
1441           FoundDecl = true;
1442     }
1443 
1444     void VisitPseudoObjectExpr(PseudoObjectExpr *POE) {
1445       // Only need to visit the semantics for POE.
1446       // SyntaticForm doesn't really use the Decal.
1447       for (auto *S : POE->semantics()) {
1448         if (auto *OVE = dyn_cast<OpaqueValueExpr>(S))
1449           // Look past the OVE into the expression it binds.
1450           Visit(OVE->getSourceExpr());
1451         else
1452           Visit(S);
1453       }
1454     }
1455 
1456     bool FoundDeclInUse() { return FoundDecl; }
1457 
1458   };  // end class DeclMatcher
1459 
1460   void CheckForLoopConditionalStatement(Sema &S, Expr *Second,
1461                                         Expr *Third, Stmt *Body) {
1462     // Condition is empty
1463     if (!Second) return;
1464 
1465     if (S.Diags.isIgnored(diag::warn_variables_not_in_loop_body,
1466                           Second->getLocStart()))
1467       return;
1468 
1469     PartialDiagnostic PDiag = S.PDiag(diag::warn_variables_not_in_loop_body);
1470     llvm::SmallPtrSet<VarDecl*, 8> Decls;
1471     SmallVector<SourceRange, 10> Ranges;
1472     DeclExtractor DE(S, Decls, Ranges);
1473     DE.Visit(Second);
1474 
1475     // Don't analyze complex conditionals.
1476     if (!DE.isSimple()) return;
1477 
1478     // No decls found.
1479     if (Decls.size() == 0) return;
1480 
1481     // Don't warn on volatile, static, or global variables.
1482     for (llvm::SmallPtrSetImpl<VarDecl*>::iterator I = Decls.begin(),
1483                                                    E = Decls.end();
1484          I != E; ++I)
1485       if ((*I)->getType().isVolatileQualified() ||
1486           (*I)->hasGlobalStorage()) return;
1487 
1488     if (DeclMatcher(S, Decls, Second).FoundDeclInUse() ||
1489         DeclMatcher(S, Decls, Third).FoundDeclInUse() ||
1490         DeclMatcher(S, Decls, Body).FoundDeclInUse())
1491       return;
1492 
1493     // Load decl names into diagnostic.
1494     if (Decls.size() > 4)
1495       PDiag << 0;
1496     else {
1497       PDiag << Decls.size();
1498       for (llvm::SmallPtrSetImpl<VarDecl*>::iterator I = Decls.begin(),
1499                                                      E = Decls.end();
1500            I != E; ++I)
1501         PDiag << (*I)->getDeclName();
1502     }
1503 
1504     // Load SourceRanges into diagnostic if there is room.
1505     // Otherwise, load the SourceRange of the conditional expression.
1506     if (Ranges.size() <= PartialDiagnostic::MaxArguments)
1507       for (SmallVectorImpl<SourceRange>::iterator I = Ranges.begin(),
1508                                                   E = Ranges.end();
1509            I != E; ++I)
1510         PDiag << *I;
1511     else
1512       PDiag << Second->getSourceRange();
1513 
1514     S.Diag(Ranges.begin()->getBegin(), PDiag);
1515   }
1516 
1517   // If Statement is an incemement or decrement, return true and sets the
1518   // variables Increment and DRE.
1519   bool ProcessIterationStmt(Sema &S, Stmt* Statement, bool &Increment,
1520                             DeclRefExpr *&DRE) {
1521     if (UnaryOperator *UO = dyn_cast<UnaryOperator>(Statement)) {
1522       switch (UO->getOpcode()) {
1523         default: return false;
1524         case UO_PostInc:
1525         case UO_PreInc:
1526           Increment = true;
1527           break;
1528         case UO_PostDec:
1529         case UO_PreDec:
1530           Increment = false;
1531           break;
1532       }
1533       DRE = dyn_cast<DeclRefExpr>(UO->getSubExpr());
1534       return DRE;
1535     }
1536 
1537     if (CXXOperatorCallExpr *Call = dyn_cast<CXXOperatorCallExpr>(Statement)) {
1538       FunctionDecl *FD = Call->getDirectCallee();
1539       if (!FD || !FD->isOverloadedOperator()) return false;
1540       switch (FD->getOverloadedOperator()) {
1541         default: return false;
1542         case OO_PlusPlus:
1543           Increment = true;
1544           break;
1545         case OO_MinusMinus:
1546           Increment = false;
1547           break;
1548       }
1549       DRE = dyn_cast<DeclRefExpr>(Call->getArg(0));
1550       return DRE;
1551     }
1552 
1553     return false;
1554   }
1555 
1556   // A visitor to determine if a continue or break statement is a
1557   // subexpression.
1558   class BreakContinueFinder : public EvaluatedExprVisitor<BreakContinueFinder> {
1559     SourceLocation BreakLoc;
1560     SourceLocation ContinueLoc;
1561   public:
1562     BreakContinueFinder(Sema &S, Stmt* Body) :
1563         Inherited(S.Context) {
1564       Visit(Body);
1565     }
1566 
1567     typedef EvaluatedExprVisitor<BreakContinueFinder> Inherited;
1568 
1569     void VisitContinueStmt(ContinueStmt* E) {
1570       ContinueLoc = E->getContinueLoc();
1571     }
1572 
1573     void VisitBreakStmt(BreakStmt* E) {
1574       BreakLoc = E->getBreakLoc();
1575     }
1576 
1577     bool ContinueFound() { return ContinueLoc.isValid(); }
1578     bool BreakFound() { return BreakLoc.isValid(); }
1579     SourceLocation GetContinueLoc() { return ContinueLoc; }
1580     SourceLocation GetBreakLoc() { return BreakLoc; }
1581 
1582   };  // end class BreakContinueFinder
1583 
1584   // Emit a warning when a loop increment/decrement appears twice per loop
1585   // iteration.  The conditions which trigger this warning are:
1586   // 1) The last statement in the loop body and the third expression in the
1587   //    for loop are both increment or both decrement of the same variable
1588   // 2) No continue statements in the loop body.
1589   void CheckForRedundantIteration(Sema &S, Expr *Third, Stmt *Body) {
1590     // Return when there is nothing to check.
1591     if (!Body || !Third) return;
1592 
1593     if (S.Diags.isIgnored(diag::warn_redundant_loop_iteration,
1594                           Third->getLocStart()))
1595       return;
1596 
1597     // Get the last statement from the loop body.
1598     CompoundStmt *CS = dyn_cast<CompoundStmt>(Body);
1599     if (!CS || CS->body_empty()) return;
1600     Stmt *LastStmt = CS->body_back();
1601     if (!LastStmt) return;
1602 
1603     bool LoopIncrement, LastIncrement;
1604     DeclRefExpr *LoopDRE, *LastDRE;
1605 
1606     if (!ProcessIterationStmt(S, Third, LoopIncrement, LoopDRE)) return;
1607     if (!ProcessIterationStmt(S, LastStmt, LastIncrement, LastDRE)) return;
1608 
1609     // Check that the two statements are both increments or both decrements
1610     // on the same variable.
1611     if (LoopIncrement != LastIncrement ||
1612         LoopDRE->getDecl() != LastDRE->getDecl()) return;
1613 
1614     if (BreakContinueFinder(S, Body).ContinueFound()) return;
1615 
1616     S.Diag(LastDRE->getLocation(), diag::warn_redundant_loop_iteration)
1617          << LastDRE->getDecl() << LastIncrement;
1618     S.Diag(LoopDRE->getLocation(), diag::note_loop_iteration_here)
1619          << LoopIncrement;
1620   }
1621 
1622 } // end namespace
1623 
1624 
1625 void Sema::CheckBreakContinueBinding(Expr *E) {
1626   if (!E || getLangOpts().CPlusPlus)
1627     return;
1628   BreakContinueFinder BCFinder(*this, E);
1629   Scope *BreakParent = CurScope->getBreakParent();
1630   if (BCFinder.BreakFound() && BreakParent) {
1631     if (BreakParent->getFlags() & Scope::SwitchScope) {
1632       Diag(BCFinder.GetBreakLoc(), diag::warn_break_binds_to_switch);
1633     } else {
1634       Diag(BCFinder.GetBreakLoc(), diag::warn_loop_ctrl_binds_to_inner)
1635           << "break";
1636     }
1637   } else if (BCFinder.ContinueFound() && CurScope->getContinueParent()) {
1638     Diag(BCFinder.GetContinueLoc(), diag::warn_loop_ctrl_binds_to_inner)
1639         << "continue";
1640   }
1641 }
1642 
1643 StmtResult
1644 Sema::ActOnForStmt(SourceLocation ForLoc, SourceLocation LParenLoc,
1645                    Stmt *First, FullExprArg second, Decl *secondVar,
1646                    FullExprArg third,
1647                    SourceLocation RParenLoc, Stmt *Body) {
1648   if (!getLangOpts().CPlusPlus) {
1649     if (DeclStmt *DS = dyn_cast_or_null<DeclStmt>(First)) {
1650       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
1651       // declare identifiers for objects having storage class 'auto' or
1652       // 'register'.
1653       for (auto *DI : DS->decls()) {
1654         VarDecl *VD = dyn_cast<VarDecl>(DI);
1655         if (VD && VD->isLocalVarDecl() && !VD->hasLocalStorage())
1656           VD = nullptr;
1657         if (!VD) {
1658           Diag(DI->getLocation(), diag::err_non_local_variable_decl_in_for);
1659           DI->setInvalidDecl();
1660         }
1661       }
1662     }
1663   }
1664 
1665   CheckBreakContinueBinding(second.get());
1666   CheckBreakContinueBinding(third.get());
1667 
1668   CheckForLoopConditionalStatement(*this, second.get(), third.get(), Body);
1669   CheckForRedundantIteration(*this, third.get(), Body);
1670 
1671   ExprResult SecondResult(second.release());
1672   VarDecl *ConditionVar = nullptr;
1673   if (secondVar) {
1674     ConditionVar = cast<VarDecl>(secondVar);
1675     SecondResult = CheckConditionVariable(ConditionVar, ForLoc, true);
1676     SecondResult = ActOnFinishFullExpr(SecondResult.get(), ForLoc);
1677     if (SecondResult.isInvalid())
1678       return StmtError();
1679   }
1680 
1681   if (SecondResult.get() &&
1682       !Diags.isIgnored(diag::warn_comma_operator,
1683                        SecondResult.get()->getExprLoc()))
1684     CommaVisitor(*this).Visit(SecondResult.get());
1685 
1686   Expr *Third  = third.release().getAs<Expr>();
1687 
1688   DiagnoseUnusedExprResult(First);
1689   DiagnoseUnusedExprResult(Third);
1690   DiagnoseUnusedExprResult(Body);
1691 
1692   if (isa<NullStmt>(Body))
1693     getCurCompoundScope().setHasEmptyLoopBodies();
1694 
1695   return new (Context) ForStmt(Context, First, SecondResult.get(), ConditionVar,
1696                                Third, Body, ForLoc, LParenLoc, RParenLoc);
1697 }
1698 
1699 /// In an Objective C collection iteration statement:
1700 ///   for (x in y)
1701 /// x can be an arbitrary l-value expression.  Bind it up as a
1702 /// full-expression.
1703 StmtResult Sema::ActOnForEachLValueExpr(Expr *E) {
1704   // Reduce placeholder expressions here.  Note that this rejects the
1705   // use of pseudo-object l-values in this position.
1706   ExprResult result = CheckPlaceholderExpr(E);
1707   if (result.isInvalid()) return StmtError();
1708   E = result.get();
1709 
1710   ExprResult FullExpr = ActOnFinishFullExpr(E);
1711   if (FullExpr.isInvalid())
1712     return StmtError();
1713   return StmtResult(static_cast<Stmt*>(FullExpr.get()));
1714 }
1715 
1716 ExprResult
1717 Sema::CheckObjCForCollectionOperand(SourceLocation forLoc, Expr *collection) {
1718   if (!collection)
1719     return ExprError();
1720 
1721   ExprResult result = CorrectDelayedTyposInExpr(collection);
1722   if (!result.isUsable())
1723     return ExprError();
1724   collection = result.get();
1725 
1726   // Bail out early if we've got a type-dependent expression.
1727   if (collection->isTypeDependent()) return collection;
1728 
1729   // Perform normal l-value conversion.
1730   result = DefaultFunctionArrayLvalueConversion(collection);
1731   if (result.isInvalid())
1732     return ExprError();
1733   collection = result.get();
1734 
1735   // The operand needs to have object-pointer type.
1736   // TODO: should we do a contextual conversion?
1737   const ObjCObjectPointerType *pointerType =
1738     collection->getType()->getAs<ObjCObjectPointerType>();
1739   if (!pointerType)
1740     return Diag(forLoc, diag::err_collection_expr_type)
1741              << collection->getType() << collection->getSourceRange();
1742 
1743   // Check that the operand provides
1744   //   - countByEnumeratingWithState:objects:count:
1745   const ObjCObjectType *objectType = pointerType->getObjectType();
1746   ObjCInterfaceDecl *iface = objectType->getInterface();
1747 
1748   // If we have a forward-declared type, we can't do this check.
1749   // Under ARC, it is an error not to have a forward-declared class.
1750   if (iface &&
1751       (getLangOpts().ObjCAutoRefCount
1752            ? RequireCompleteType(forLoc, QualType(objectType, 0),
1753                                  diag::err_arc_collection_forward, collection)
1754            : !isCompleteType(forLoc, QualType(objectType, 0)))) {
1755     // Otherwise, if we have any useful type information, check that
1756     // the type declares the appropriate method.
1757   } else if (iface || !objectType->qual_empty()) {
1758     IdentifierInfo *selectorIdents[] = {
1759       &Context.Idents.get("countByEnumeratingWithState"),
1760       &Context.Idents.get("objects"),
1761       &Context.Idents.get("count")
1762     };
1763     Selector selector = Context.Selectors.getSelector(3, &selectorIdents[0]);
1764 
1765     ObjCMethodDecl *method = nullptr;
1766 
1767     // If there's an interface, look in both the public and private APIs.
1768     if (iface) {
1769       method = iface->lookupInstanceMethod(selector);
1770       if (!method) method = iface->lookupPrivateMethod(selector);
1771     }
1772 
1773     // Also check protocol qualifiers.
1774     if (!method)
1775       method = LookupMethodInQualifiedType(selector, pointerType,
1776                                            /*instance*/ true);
1777 
1778     // If we didn't find it anywhere, give up.
1779     if (!method) {
1780       Diag(forLoc, diag::warn_collection_expr_type)
1781         << collection->getType() << selector << collection->getSourceRange();
1782     }
1783 
1784     // TODO: check for an incompatible signature?
1785   }
1786 
1787   // Wrap up any cleanups in the expression.
1788   return collection;
1789 }
1790 
1791 StmtResult
1792 Sema::ActOnObjCForCollectionStmt(SourceLocation ForLoc,
1793                                  Stmt *First, Expr *collection,
1794                                  SourceLocation RParenLoc) {
1795 
1796   ExprResult CollectionExprResult =
1797     CheckObjCForCollectionOperand(ForLoc, collection);
1798 
1799   if (First) {
1800     QualType FirstType;
1801     if (DeclStmt *DS = dyn_cast<DeclStmt>(First)) {
1802       if (!DS->isSingleDecl())
1803         return StmtError(Diag((*DS->decl_begin())->getLocation(),
1804                          diag::err_toomany_element_decls));
1805 
1806       VarDecl *D = dyn_cast<VarDecl>(DS->getSingleDecl());
1807       if (!D || D->isInvalidDecl())
1808         return StmtError();
1809 
1810       FirstType = D->getType();
1811       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
1812       // declare identifiers for objects having storage class 'auto' or
1813       // 'register'.
1814       if (!D->hasLocalStorage())
1815         return StmtError(Diag(D->getLocation(),
1816                               diag::err_non_local_variable_decl_in_for));
1817 
1818       // If the type contained 'auto', deduce the 'auto' to 'id'.
1819       if (FirstType->getContainedAutoType()) {
1820         OpaqueValueExpr OpaqueId(D->getLocation(), Context.getObjCIdType(),
1821                                  VK_RValue);
1822         Expr *DeducedInit = &OpaqueId;
1823         if (DeduceAutoType(D->getTypeSourceInfo(), DeducedInit, FirstType) ==
1824                 DAR_Failed)
1825           DiagnoseAutoDeductionFailure(D, DeducedInit);
1826         if (FirstType.isNull()) {
1827           D->setInvalidDecl();
1828           return StmtError();
1829         }
1830 
1831         D->setType(FirstType);
1832 
1833         if (ActiveTemplateInstantiations.empty()) {
1834           SourceLocation Loc =
1835               D->getTypeSourceInfo()->getTypeLoc().getBeginLoc();
1836           Diag(Loc, diag::warn_auto_var_is_id)
1837             << D->getDeclName();
1838         }
1839       }
1840 
1841     } else {
1842       Expr *FirstE = cast<Expr>(First);
1843       if (!FirstE->isTypeDependent() && !FirstE->isLValue())
1844         return StmtError(Diag(First->getLocStart(),
1845                    diag::err_selector_element_not_lvalue)
1846           << First->getSourceRange());
1847 
1848       FirstType = static_cast<Expr*>(First)->getType();
1849       if (FirstType.isConstQualified())
1850         Diag(ForLoc, diag::err_selector_element_const_type)
1851           << FirstType << First->getSourceRange();
1852     }
1853     if (!FirstType->isDependentType() &&
1854         !FirstType->isObjCObjectPointerType() &&
1855         !FirstType->isBlockPointerType())
1856         return StmtError(Diag(ForLoc, diag::err_selector_element_type)
1857                            << FirstType << First->getSourceRange());
1858   }
1859 
1860   if (CollectionExprResult.isInvalid())
1861     return StmtError();
1862 
1863   CollectionExprResult = ActOnFinishFullExpr(CollectionExprResult.get());
1864   if (CollectionExprResult.isInvalid())
1865     return StmtError();
1866 
1867   return new (Context) ObjCForCollectionStmt(First, CollectionExprResult.get(),
1868                                              nullptr, ForLoc, RParenLoc);
1869 }
1870 
1871 /// Finish building a variable declaration for a for-range statement.
1872 /// \return true if an error occurs.
1873 static bool FinishForRangeVarDecl(Sema &SemaRef, VarDecl *Decl, Expr *Init,
1874                                   SourceLocation Loc, int DiagID) {
1875   if (Decl->getType()->isUndeducedType()) {
1876     ExprResult Res = SemaRef.CorrectDelayedTyposInExpr(Init);
1877     if (!Res.isUsable()) {
1878       Decl->setInvalidDecl();
1879       return true;
1880     }
1881     Init = Res.get();
1882   }
1883 
1884   // Deduce the type for the iterator variable now rather than leaving it to
1885   // AddInitializerToDecl, so we can produce a more suitable diagnostic.
1886   QualType InitType;
1887   if ((!isa<InitListExpr>(Init) && Init->getType()->isVoidType()) ||
1888       SemaRef.DeduceAutoType(Decl->getTypeSourceInfo(), Init, InitType) ==
1889           Sema::DAR_Failed)
1890     SemaRef.Diag(Loc, DiagID) << Init->getType();
1891   if (InitType.isNull()) {
1892     Decl->setInvalidDecl();
1893     return true;
1894   }
1895   Decl->setType(InitType);
1896 
1897   // In ARC, infer lifetime.
1898   // FIXME: ARC may want to turn this into 'const __unsafe_unretained' if
1899   // we're doing the equivalent of fast iteration.
1900   if (SemaRef.getLangOpts().ObjCAutoRefCount &&
1901       SemaRef.inferObjCARCLifetime(Decl))
1902     Decl->setInvalidDecl();
1903 
1904   SemaRef.AddInitializerToDecl(Decl, Init, /*DirectInit=*/false,
1905                                /*TypeMayContainAuto=*/false);
1906   SemaRef.FinalizeDeclaration(Decl);
1907   SemaRef.CurContext->addHiddenDecl(Decl);
1908   return false;
1909 }
1910 
1911 namespace {
1912 // An enum to represent whether something is dealing with a call to begin()
1913 // or a call to end() in a range-based for loop.
1914 enum BeginEndFunction {
1915   BEF_begin,
1916   BEF_end
1917 };
1918 
1919 /// Produce a note indicating which begin/end function was implicitly called
1920 /// by a C++11 for-range statement. This is often not obvious from the code,
1921 /// nor from the diagnostics produced when analysing the implicit expressions
1922 /// required in a for-range statement.
1923 void NoteForRangeBeginEndFunction(Sema &SemaRef, Expr *E,
1924                                   BeginEndFunction BEF) {
1925   CallExpr *CE = dyn_cast<CallExpr>(E);
1926   if (!CE)
1927     return;
1928   FunctionDecl *D = dyn_cast<FunctionDecl>(CE->getCalleeDecl());
1929   if (!D)
1930     return;
1931   SourceLocation Loc = D->getLocation();
1932 
1933   std::string Description;
1934   bool IsTemplate = false;
1935   if (FunctionTemplateDecl *FunTmpl = D->getPrimaryTemplate()) {
1936     Description = SemaRef.getTemplateArgumentBindingsText(
1937       FunTmpl->getTemplateParameters(), *D->getTemplateSpecializationArgs());
1938     IsTemplate = true;
1939   }
1940 
1941   SemaRef.Diag(Loc, diag::note_for_range_begin_end)
1942     << BEF << IsTemplate << Description << E->getType();
1943 }
1944 
1945 /// Build a variable declaration for a for-range statement.
1946 VarDecl *BuildForRangeVarDecl(Sema &SemaRef, SourceLocation Loc,
1947                               QualType Type, const char *Name) {
1948   DeclContext *DC = SemaRef.CurContext;
1949   IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name);
1950   TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc);
1951   VarDecl *Decl = VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type,
1952                                   TInfo, SC_None);
1953   Decl->setImplicit();
1954   return Decl;
1955 }
1956 
1957 }
1958 
1959 static bool ObjCEnumerationCollection(Expr *Collection) {
1960   return !Collection->isTypeDependent()
1961           && Collection->getType()->getAs<ObjCObjectPointerType>() != nullptr;
1962 }
1963 
1964 /// ActOnCXXForRangeStmt - Check and build a C++11 for-range statement.
1965 ///
1966 /// C++11 [stmt.ranged]:
1967 ///   A range-based for statement is equivalent to
1968 ///
1969 ///   {
1970 ///     auto && __range = range-init;
1971 ///     for ( auto __begin = begin-expr,
1972 ///           __end = end-expr;
1973 ///           __begin != __end;
1974 ///           ++__begin ) {
1975 ///       for-range-declaration = *__begin;
1976 ///       statement
1977 ///     }
1978 ///   }
1979 ///
1980 /// The body of the loop is not available yet, since it cannot be analysed until
1981 /// we have determined the type of the for-range-declaration.
1982 StmtResult Sema::ActOnCXXForRangeStmt(Scope *S, SourceLocation ForLoc,
1983                                       SourceLocation CoawaitLoc, Stmt *First,
1984                                       SourceLocation ColonLoc, Expr *Range,
1985                                       SourceLocation RParenLoc,
1986                                       BuildForRangeKind Kind) {
1987   if (!First)
1988     return StmtError();
1989 
1990   if (Range && ObjCEnumerationCollection(Range))
1991     return ActOnObjCForCollectionStmt(ForLoc, First, Range, RParenLoc);
1992 
1993   DeclStmt *DS = dyn_cast<DeclStmt>(First);
1994   assert(DS && "first part of for range not a decl stmt");
1995 
1996   if (!DS->isSingleDecl()) {
1997     Diag(DS->getStartLoc(), diag::err_type_defined_in_for_range);
1998     return StmtError();
1999   }
2000 
2001   Decl *LoopVar = DS->getSingleDecl();
2002   if (LoopVar->isInvalidDecl() || !Range ||
2003       DiagnoseUnexpandedParameterPack(Range, UPPC_Expression)) {
2004     LoopVar->setInvalidDecl();
2005     return StmtError();
2006   }
2007 
2008   // Coroutines: 'for co_await' implicitly co_awaits its range.
2009   if (CoawaitLoc.isValid()) {
2010     ExprResult Coawait = ActOnCoawaitExpr(S, CoawaitLoc, Range);
2011     if (Coawait.isInvalid()) return StmtError();
2012     Range = Coawait.get();
2013   }
2014 
2015   // Build  auto && __range = range-init
2016   SourceLocation RangeLoc = Range->getLocStart();
2017   VarDecl *RangeVar = BuildForRangeVarDecl(*this, RangeLoc,
2018                                            Context.getAutoRRefDeductType(),
2019                                            "__range");
2020   if (FinishForRangeVarDecl(*this, RangeVar, Range, RangeLoc,
2021                             diag::err_for_range_deduction_failure)) {
2022     LoopVar->setInvalidDecl();
2023     return StmtError();
2024   }
2025 
2026   // Claim the type doesn't contain auto: we've already done the checking.
2027   DeclGroupPtrTy RangeGroup =
2028       BuildDeclaratorGroup(MutableArrayRef<Decl *>((Decl **)&RangeVar, 1),
2029                            /*TypeMayContainAuto=*/ false);
2030   StmtResult RangeDecl = ActOnDeclStmt(RangeGroup, RangeLoc, RangeLoc);
2031   if (RangeDecl.isInvalid()) {
2032     LoopVar->setInvalidDecl();
2033     return StmtError();
2034   }
2035 
2036   return BuildCXXForRangeStmt(ForLoc, CoawaitLoc, ColonLoc, RangeDecl.get(),
2037                               /*BeginStmt=*/nullptr, /*EndStmt=*/nullptr,
2038                               /*Cond=*/nullptr, /*Inc=*/nullptr,
2039                               DS, RParenLoc, Kind);
2040 }
2041 
2042 /// \brief Create the initialization, compare, and increment steps for
2043 /// the range-based for loop expression.
2044 /// This function does not handle array-based for loops,
2045 /// which are created in Sema::BuildCXXForRangeStmt.
2046 ///
2047 /// \returns a ForRangeStatus indicating success or what kind of error occurred.
2048 /// BeginExpr and EndExpr are set and FRS_Success is returned on success;
2049 /// CandidateSet and BEF are set and some non-success value is returned on
2050 /// failure.
2051 static Sema::ForRangeStatus BuildNonArrayForRange(Sema &SemaRef,
2052                                             Expr *BeginRange, Expr *EndRange,
2053                                             QualType RangeType,
2054                                             VarDecl *BeginVar,
2055                                             VarDecl *EndVar,
2056                                             SourceLocation ColonLoc,
2057                                             OverloadCandidateSet *CandidateSet,
2058                                             ExprResult *BeginExpr,
2059                                             ExprResult *EndExpr,
2060                                             BeginEndFunction *BEF) {
2061   DeclarationNameInfo BeginNameInfo(
2062       &SemaRef.PP.getIdentifierTable().get("begin"), ColonLoc);
2063   DeclarationNameInfo EndNameInfo(&SemaRef.PP.getIdentifierTable().get("end"),
2064                                   ColonLoc);
2065 
2066   LookupResult BeginMemberLookup(SemaRef, BeginNameInfo,
2067                                  Sema::LookupMemberName);
2068   LookupResult EndMemberLookup(SemaRef, EndNameInfo, Sema::LookupMemberName);
2069 
2070   if (CXXRecordDecl *D = RangeType->getAsCXXRecordDecl()) {
2071     // - if _RangeT is a class type, the unqualified-ids begin and end are
2072     //   looked up in the scope of class _RangeT as if by class member access
2073     //   lookup (3.4.5), and if either (or both) finds at least one
2074     //   declaration, begin-expr and end-expr are __range.begin() and
2075     //   __range.end(), respectively;
2076     SemaRef.LookupQualifiedName(BeginMemberLookup, D);
2077     SemaRef.LookupQualifiedName(EndMemberLookup, D);
2078 
2079     if (BeginMemberLookup.empty() != EndMemberLookup.empty()) {
2080       SourceLocation RangeLoc = BeginVar->getLocation();
2081       *BEF = BeginMemberLookup.empty() ? BEF_end : BEF_begin;
2082 
2083       SemaRef.Diag(RangeLoc, diag::err_for_range_member_begin_end_mismatch)
2084           << RangeLoc << BeginRange->getType() << *BEF;
2085       return Sema::FRS_DiagnosticIssued;
2086     }
2087   } else {
2088     // - otherwise, begin-expr and end-expr are begin(__range) and
2089     //   end(__range), respectively, where begin and end are looked up with
2090     //   argument-dependent lookup (3.4.2). For the purposes of this name
2091     //   lookup, namespace std is an associated namespace.
2092 
2093   }
2094 
2095   *BEF = BEF_begin;
2096   Sema::ForRangeStatus RangeStatus =
2097       SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, BeginNameInfo,
2098                                         BeginMemberLookup, CandidateSet,
2099                                         BeginRange, BeginExpr);
2100 
2101   if (RangeStatus != Sema::FRS_Success) {
2102     if (RangeStatus == Sema::FRS_DiagnosticIssued)
2103       SemaRef.Diag(BeginRange->getLocStart(), diag::note_in_for_range)
2104           << ColonLoc << BEF_begin << BeginRange->getType();
2105     return RangeStatus;
2106   }
2107   if (FinishForRangeVarDecl(SemaRef, BeginVar, BeginExpr->get(), ColonLoc,
2108                             diag::err_for_range_iter_deduction_failure)) {
2109     NoteForRangeBeginEndFunction(SemaRef, BeginExpr->get(), *BEF);
2110     return Sema::FRS_DiagnosticIssued;
2111   }
2112 
2113   *BEF = BEF_end;
2114   RangeStatus =
2115       SemaRef.BuildForRangeBeginEndCall(ColonLoc, ColonLoc, EndNameInfo,
2116                                         EndMemberLookup, CandidateSet,
2117                                         EndRange, EndExpr);
2118   if (RangeStatus != Sema::FRS_Success) {
2119     if (RangeStatus == Sema::FRS_DiagnosticIssued)
2120       SemaRef.Diag(EndRange->getLocStart(), diag::note_in_for_range)
2121           << ColonLoc << BEF_end << EndRange->getType();
2122     return RangeStatus;
2123   }
2124   if (FinishForRangeVarDecl(SemaRef, EndVar, EndExpr->get(), ColonLoc,
2125                             diag::err_for_range_iter_deduction_failure)) {
2126     NoteForRangeBeginEndFunction(SemaRef, EndExpr->get(), *BEF);
2127     return Sema::FRS_DiagnosticIssued;
2128   }
2129   return Sema::FRS_Success;
2130 }
2131 
2132 /// Speculatively attempt to dereference an invalid range expression.
2133 /// If the attempt fails, this function will return a valid, null StmtResult
2134 /// and emit no diagnostics.
2135 static StmtResult RebuildForRangeWithDereference(Sema &SemaRef, Scope *S,
2136                                                  SourceLocation ForLoc,
2137                                                  SourceLocation CoawaitLoc,
2138                                                  Stmt *LoopVarDecl,
2139                                                  SourceLocation ColonLoc,
2140                                                  Expr *Range,
2141                                                  SourceLocation RangeLoc,
2142                                                  SourceLocation RParenLoc) {
2143   // Determine whether we can rebuild the for-range statement with a
2144   // dereferenced range expression.
2145   ExprResult AdjustedRange;
2146   {
2147     Sema::SFINAETrap Trap(SemaRef);
2148 
2149     AdjustedRange = SemaRef.BuildUnaryOp(S, RangeLoc, UO_Deref, Range);
2150     if (AdjustedRange.isInvalid())
2151       return StmtResult();
2152 
2153     StmtResult SR = SemaRef.ActOnCXXForRangeStmt(
2154         S, ForLoc, CoawaitLoc, LoopVarDecl, ColonLoc, AdjustedRange.get(),
2155         RParenLoc, Sema::BFRK_Check);
2156     if (SR.isInvalid())
2157       return StmtResult();
2158   }
2159 
2160   // The attempt to dereference worked well enough that it could produce a valid
2161   // loop. Produce a fixit, and rebuild the loop with diagnostics enabled, in
2162   // case there are any other (non-fatal) problems with it.
2163   SemaRef.Diag(RangeLoc, diag::err_for_range_dereference)
2164     << Range->getType() << FixItHint::CreateInsertion(RangeLoc, "*");
2165   return SemaRef.ActOnCXXForRangeStmt(S, ForLoc, CoawaitLoc, LoopVarDecl,
2166                                       ColonLoc, AdjustedRange.get(), RParenLoc,
2167                                       Sema::BFRK_Rebuild);
2168 }
2169 
2170 namespace {
2171 /// RAII object to automatically invalidate a declaration if an error occurs.
2172 struct InvalidateOnErrorScope {
2173   InvalidateOnErrorScope(Sema &SemaRef, Decl *D, bool Enabled)
2174       : Trap(SemaRef.Diags), D(D), Enabled(Enabled) {}
2175   ~InvalidateOnErrorScope() {
2176     if (Enabled && Trap.hasErrorOccurred())
2177       D->setInvalidDecl();
2178   }
2179 
2180   DiagnosticErrorTrap Trap;
2181   Decl *D;
2182   bool Enabled;
2183 };
2184 }
2185 
2186 /// BuildCXXForRangeStmt - Build or instantiate a C++11 for-range statement.
2187 StmtResult
2188 Sema::BuildCXXForRangeStmt(SourceLocation ForLoc, SourceLocation CoawaitLoc,
2189                            SourceLocation ColonLoc, Stmt *RangeDecl,
2190                            Stmt *Begin, Stmt *End, Expr *Cond,
2191                            Expr *Inc, Stmt *LoopVarDecl,
2192                            SourceLocation RParenLoc, BuildForRangeKind Kind) {
2193   // FIXME: This should not be used during template instantiation. We should
2194   // pick up the set of unqualified lookup results for the != and + operators
2195   // in the initial parse.
2196   //
2197   // Testcase (accepts-invalid):
2198   //   template<typename T> void f() { for (auto x : T()) {} }
2199   //   namespace N { struct X { X begin(); X end(); int operator*(); }; }
2200   //   bool operator!=(N::X, N::X); void operator++(N::X);
2201   //   void g() { f<N::X>(); }
2202   Scope *S = getCurScope();
2203 
2204   DeclStmt *RangeDS = cast<DeclStmt>(RangeDecl);
2205   VarDecl *RangeVar = cast<VarDecl>(RangeDS->getSingleDecl());
2206   QualType RangeVarType = RangeVar->getType();
2207 
2208   DeclStmt *LoopVarDS = cast<DeclStmt>(LoopVarDecl);
2209   VarDecl *LoopVar = cast<VarDecl>(LoopVarDS->getSingleDecl());
2210 
2211   // If we hit any errors, mark the loop variable as invalid if its type
2212   // contains 'auto'.
2213   InvalidateOnErrorScope Invalidate(*this, LoopVar,
2214                                     LoopVar->getType()->isUndeducedType());
2215 
2216   StmtResult BeginDeclStmt = Begin;
2217   StmtResult EndDeclStmt = End;
2218   ExprResult NotEqExpr = Cond, IncrExpr = Inc;
2219 
2220   if (RangeVarType->isDependentType()) {
2221     // The range is implicitly used as a placeholder when it is dependent.
2222     RangeVar->markUsed(Context);
2223 
2224     // Deduce any 'auto's in the loop variable as 'DependentTy'. We'll fill
2225     // them in properly when we instantiate the loop.
2226     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check)
2227       LoopVar->setType(SubstAutoType(LoopVar->getType(), Context.DependentTy));
2228   } else if (!BeginDeclStmt.get()) {
2229     SourceLocation RangeLoc = RangeVar->getLocation();
2230 
2231     const QualType RangeVarNonRefType = RangeVarType.getNonReferenceType();
2232 
2233     ExprResult BeginRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
2234                                                 VK_LValue, ColonLoc);
2235     if (BeginRangeRef.isInvalid())
2236       return StmtError();
2237 
2238     ExprResult EndRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
2239                                               VK_LValue, ColonLoc);
2240     if (EndRangeRef.isInvalid())
2241       return StmtError();
2242 
2243     QualType AutoType = Context.getAutoDeductType();
2244     Expr *Range = RangeVar->getInit();
2245     if (!Range)
2246       return StmtError();
2247     QualType RangeType = Range->getType();
2248 
2249     if (RequireCompleteType(RangeLoc, RangeType,
2250                             diag::err_for_range_incomplete_type))
2251       return StmtError();
2252 
2253     // Build auto __begin = begin-expr, __end = end-expr.
2254     VarDecl *BeginVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
2255                                              "__begin");
2256     VarDecl *EndVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
2257                                            "__end");
2258 
2259     // Build begin-expr and end-expr and attach to __begin and __end variables.
2260     ExprResult BeginExpr, EndExpr;
2261     if (const ArrayType *UnqAT = RangeType->getAsArrayTypeUnsafe()) {
2262       // - if _RangeT is an array type, begin-expr and end-expr are __range and
2263       //   __range + __bound, respectively, where __bound is the array bound. If
2264       //   _RangeT is an array of unknown size or an array of incomplete type,
2265       //   the program is ill-formed;
2266 
2267       // begin-expr is __range.
2268       BeginExpr = BeginRangeRef;
2269       if (FinishForRangeVarDecl(*this, BeginVar, BeginRangeRef.get(), ColonLoc,
2270                                 diag::err_for_range_iter_deduction_failure)) {
2271         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2272         return StmtError();
2273       }
2274 
2275       // Find the array bound.
2276       ExprResult BoundExpr;
2277       if (const ConstantArrayType *CAT = dyn_cast<ConstantArrayType>(UnqAT))
2278         BoundExpr = IntegerLiteral::Create(
2279             Context, CAT->getSize(), Context.getPointerDiffType(), RangeLoc);
2280       else if (const VariableArrayType *VAT =
2281                dyn_cast<VariableArrayType>(UnqAT))
2282         BoundExpr = VAT->getSizeExpr();
2283       else {
2284         // Can't be a DependentSizedArrayType or an IncompleteArrayType since
2285         // UnqAT is not incomplete and Range is not type-dependent.
2286         llvm_unreachable("Unexpected array type in for-range");
2287       }
2288 
2289       // end-expr is __range + __bound.
2290       EndExpr = ActOnBinOp(S, ColonLoc, tok::plus, EndRangeRef.get(),
2291                            BoundExpr.get());
2292       if (EndExpr.isInvalid())
2293         return StmtError();
2294       if (FinishForRangeVarDecl(*this, EndVar, EndExpr.get(), ColonLoc,
2295                                 diag::err_for_range_iter_deduction_failure)) {
2296         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2297         return StmtError();
2298       }
2299     } else {
2300       OverloadCandidateSet CandidateSet(RangeLoc,
2301                                         OverloadCandidateSet::CSK_Normal);
2302       BeginEndFunction BEFFailure;
2303       ForRangeStatus RangeStatus =
2304           BuildNonArrayForRange(*this, BeginRangeRef.get(),
2305                                 EndRangeRef.get(), RangeType,
2306                                 BeginVar, EndVar, ColonLoc, &CandidateSet,
2307                                 &BeginExpr, &EndExpr, &BEFFailure);
2308 
2309       if (Kind == BFRK_Build && RangeStatus == FRS_NoViableFunction &&
2310           BEFFailure == BEF_begin) {
2311         // If the range is being built from an array parameter, emit a
2312         // a diagnostic that it is being treated as a pointer.
2313         if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Range)) {
2314           if (ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
2315             QualType ArrayTy = PVD->getOriginalType();
2316             QualType PointerTy = PVD->getType();
2317             if (PointerTy->isPointerType() && ArrayTy->isArrayType()) {
2318               Diag(Range->getLocStart(), diag::err_range_on_array_parameter)
2319                 << RangeLoc << PVD << ArrayTy << PointerTy;
2320               Diag(PVD->getLocation(), diag::note_declared_at);
2321               return StmtError();
2322             }
2323           }
2324         }
2325 
2326         // If building the range failed, try dereferencing the range expression
2327         // unless a diagnostic was issued or the end function is problematic.
2328         StmtResult SR = RebuildForRangeWithDereference(*this, S, ForLoc,
2329                                                        CoawaitLoc,
2330                                                        LoopVarDecl, ColonLoc,
2331                                                        Range, RangeLoc,
2332                                                        RParenLoc);
2333         if (SR.isInvalid() || SR.isUsable())
2334           return SR;
2335       }
2336 
2337       // Otherwise, emit diagnostics if we haven't already.
2338       if (RangeStatus == FRS_NoViableFunction) {
2339         Expr *Range = BEFFailure ? EndRangeRef.get() : BeginRangeRef.get();
2340         Diag(Range->getLocStart(), diag::err_for_range_invalid)
2341             << RangeLoc << Range->getType() << BEFFailure;
2342         CandidateSet.NoteCandidates(*this, OCD_AllCandidates, Range);
2343       }
2344       // Return an error if no fix was discovered.
2345       if (RangeStatus != FRS_Success)
2346         return StmtError();
2347     }
2348 
2349     assert(!BeginExpr.isInvalid() && !EndExpr.isInvalid() &&
2350            "invalid range expression in for loop");
2351 
2352     // C++11 [dcl.spec.auto]p7: BeginType and EndType must be the same.
2353     // C++1z removes this restriction.
2354     QualType BeginType = BeginVar->getType(), EndType = EndVar->getType();
2355     if (!Context.hasSameType(BeginType, EndType)) {
2356       Diag(RangeLoc, getLangOpts().CPlusPlus1z
2357                          ? diag::warn_for_range_begin_end_types_differ
2358                          : diag::ext_for_range_begin_end_types_differ)
2359           << BeginType << EndType;
2360       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2361       NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2362     }
2363 
2364     BeginDeclStmt =
2365         ActOnDeclStmt(ConvertDeclToDeclGroup(BeginVar), ColonLoc, ColonLoc);
2366     EndDeclStmt =
2367         ActOnDeclStmt(ConvertDeclToDeclGroup(EndVar), ColonLoc, ColonLoc);
2368 
2369     const QualType BeginRefNonRefType = BeginType.getNonReferenceType();
2370     ExprResult BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2371                                            VK_LValue, ColonLoc);
2372     if (BeginRef.isInvalid())
2373       return StmtError();
2374 
2375     ExprResult EndRef = BuildDeclRefExpr(EndVar, EndType.getNonReferenceType(),
2376                                          VK_LValue, ColonLoc);
2377     if (EndRef.isInvalid())
2378       return StmtError();
2379 
2380     // Build and check __begin != __end expression.
2381     NotEqExpr = ActOnBinOp(S, ColonLoc, tok::exclaimequal,
2382                            BeginRef.get(), EndRef.get());
2383     NotEqExpr = ActOnBooleanCondition(S, ColonLoc, NotEqExpr.get());
2384     NotEqExpr = ActOnFinishFullExpr(NotEqExpr.get());
2385     if (NotEqExpr.isInvalid()) {
2386       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2387         << RangeLoc << 0 << BeginRangeRef.get()->getType();
2388       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2389       if (!Context.hasSameType(BeginType, EndType))
2390         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2391       return StmtError();
2392     }
2393 
2394     // Build and check ++__begin expression.
2395     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2396                                 VK_LValue, ColonLoc);
2397     if (BeginRef.isInvalid())
2398       return StmtError();
2399 
2400     IncrExpr = ActOnUnaryOp(S, ColonLoc, tok::plusplus, BeginRef.get());
2401     if (!IncrExpr.isInvalid() && CoawaitLoc.isValid())
2402       IncrExpr = ActOnCoawaitExpr(S, CoawaitLoc, IncrExpr.get());
2403     if (!IncrExpr.isInvalid())
2404       IncrExpr = ActOnFinishFullExpr(IncrExpr.get());
2405     if (IncrExpr.isInvalid()) {
2406       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2407         << RangeLoc << 2 << BeginRangeRef.get()->getType() ;
2408       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2409       return StmtError();
2410     }
2411 
2412     // Build and check *__begin  expression.
2413     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2414                                 VK_LValue, ColonLoc);
2415     if (BeginRef.isInvalid())
2416       return StmtError();
2417 
2418     ExprResult DerefExpr = ActOnUnaryOp(S, ColonLoc, tok::star, BeginRef.get());
2419     if (DerefExpr.isInvalid()) {
2420       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2421         << RangeLoc << 1 << BeginRangeRef.get()->getType();
2422       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2423       return StmtError();
2424     }
2425 
2426     // Attach  *__begin  as initializer for VD. Don't touch it if we're just
2427     // trying to determine whether this would be a valid range.
2428     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
2429       AddInitializerToDecl(LoopVar, DerefExpr.get(), /*DirectInit=*/false,
2430                            /*TypeMayContainAuto=*/true);
2431       if (LoopVar->isInvalidDecl())
2432         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2433     }
2434   }
2435 
2436   // Don't bother to actually allocate the result if we're just trying to
2437   // determine whether it would be valid.
2438   if (Kind == BFRK_Check)
2439     return StmtResult();
2440 
2441   return new (Context) CXXForRangeStmt(
2442       RangeDS, cast_or_null<DeclStmt>(BeginDeclStmt.get()),
2443       cast_or_null<DeclStmt>(EndDeclStmt.get()), NotEqExpr.get(),
2444       IncrExpr.get(), LoopVarDS, /*Body=*/nullptr, ForLoc, CoawaitLoc,
2445       ColonLoc, RParenLoc);
2446 }
2447 
2448 /// FinishObjCForCollectionStmt - Attach the body to a objective-C foreach
2449 /// statement.
2450 StmtResult Sema::FinishObjCForCollectionStmt(Stmt *S, Stmt *B) {
2451   if (!S || !B)
2452     return StmtError();
2453   ObjCForCollectionStmt * ForStmt = cast<ObjCForCollectionStmt>(S);
2454 
2455   ForStmt->setBody(B);
2456   return S;
2457 }
2458 
2459 // Warn when the loop variable is a const reference that creates a copy.
2460 // Suggest using the non-reference type for copies.  If a copy can be prevented
2461 // suggest the const reference type that would do so.
2462 // For instance, given "for (const &Foo : Range)", suggest
2463 // "for (const Foo : Range)" to denote a copy is made for the loop.  If
2464 // possible, also suggest "for (const &Bar : Range)" if this type prevents
2465 // the copy altogether.
2466 static void DiagnoseForRangeReferenceVariableCopies(Sema &SemaRef,
2467                                                     const VarDecl *VD,
2468                                                     QualType RangeInitType) {
2469   const Expr *InitExpr = VD->getInit();
2470   if (!InitExpr)
2471     return;
2472 
2473   QualType VariableType = VD->getType();
2474 
2475   const MaterializeTemporaryExpr *MTE =
2476       dyn_cast<MaterializeTemporaryExpr>(InitExpr);
2477 
2478   // No copy made.
2479   if (!MTE)
2480     return;
2481 
2482   const Expr *E = MTE->GetTemporaryExpr()->IgnoreImpCasts();
2483 
2484   // Searching for either UnaryOperator for dereference of a pointer or
2485   // CXXOperatorCallExpr for handling iterators.
2486   while (!isa<CXXOperatorCallExpr>(E) && !isa<UnaryOperator>(E)) {
2487     if (const CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(E)) {
2488       E = CCE->getArg(0);
2489     } else if (const CXXMemberCallExpr *Call = dyn_cast<CXXMemberCallExpr>(E)) {
2490       const MemberExpr *ME = cast<MemberExpr>(Call->getCallee());
2491       E = ME->getBase();
2492     } else {
2493       const MaterializeTemporaryExpr *MTE = cast<MaterializeTemporaryExpr>(E);
2494       E = MTE->GetTemporaryExpr();
2495     }
2496     E = E->IgnoreImpCasts();
2497   }
2498 
2499   bool ReturnsReference = false;
2500   if (isa<UnaryOperator>(E)) {
2501     ReturnsReference = true;
2502   } else {
2503     const CXXOperatorCallExpr *Call = cast<CXXOperatorCallExpr>(E);
2504     const FunctionDecl *FD = Call->getDirectCallee();
2505     QualType ReturnType = FD->getReturnType();
2506     ReturnsReference = ReturnType->isReferenceType();
2507   }
2508 
2509   if (ReturnsReference) {
2510     // Loop variable creates a temporary.  Suggest either to go with
2511     // non-reference loop variable to indiciate a copy is made, or
2512     // the correct time to bind a const reference.
2513     SemaRef.Diag(VD->getLocation(), diag::warn_for_range_const_reference_copy)
2514         << VD << VariableType << E->getType();
2515     QualType NonReferenceType = VariableType.getNonReferenceType();
2516     NonReferenceType.removeLocalConst();
2517     QualType NewReferenceType =
2518         SemaRef.Context.getLValueReferenceType(E->getType().withConst());
2519     SemaRef.Diag(VD->getLocStart(), diag::note_use_type_or_non_reference)
2520         << NonReferenceType << NewReferenceType << VD->getSourceRange();
2521   } else {
2522     // The range always returns a copy, so a temporary is always created.
2523     // Suggest removing the reference from the loop variable.
2524     SemaRef.Diag(VD->getLocation(), diag::warn_for_range_variable_always_copy)
2525         << VD << RangeInitType;
2526     QualType NonReferenceType = VariableType.getNonReferenceType();
2527     NonReferenceType.removeLocalConst();
2528     SemaRef.Diag(VD->getLocStart(), diag::note_use_non_reference_type)
2529         << NonReferenceType << VD->getSourceRange();
2530   }
2531 }
2532 
2533 // Warns when the loop variable can be changed to a reference type to
2534 // prevent a copy.  For instance, if given "for (const Foo x : Range)" suggest
2535 // "for (const Foo &x : Range)" if this form does not make a copy.
2536 static void DiagnoseForRangeConstVariableCopies(Sema &SemaRef,
2537                                                 const VarDecl *VD) {
2538   const Expr *InitExpr = VD->getInit();
2539   if (!InitExpr)
2540     return;
2541 
2542   QualType VariableType = VD->getType();
2543 
2544   if (const CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(InitExpr)) {
2545     if (!CE->getConstructor()->isCopyConstructor())
2546       return;
2547   } else if (const CastExpr *CE = dyn_cast<CastExpr>(InitExpr)) {
2548     if (CE->getCastKind() != CK_LValueToRValue)
2549       return;
2550   } else {
2551     return;
2552   }
2553 
2554   // TODO: Determine a maximum size that a POD type can be before a diagnostic
2555   // should be emitted.  Also, only ignore POD types with trivial copy
2556   // constructors.
2557   if (VariableType.isPODType(SemaRef.Context))
2558     return;
2559 
2560   // Suggest changing from a const variable to a const reference variable
2561   // if doing so will prevent a copy.
2562   SemaRef.Diag(VD->getLocation(), diag::warn_for_range_copy)
2563       << VD << VariableType << InitExpr->getType();
2564   SemaRef.Diag(VD->getLocStart(), diag::note_use_reference_type)
2565       << SemaRef.Context.getLValueReferenceType(VariableType)
2566       << VD->getSourceRange();
2567 }
2568 
2569 /// DiagnoseForRangeVariableCopies - Diagnose three cases and fixes for them.
2570 /// 1) for (const foo &x : foos) where foos only returns a copy.  Suggest
2571 ///    using "const foo x" to show that a copy is made
2572 /// 2) for (const bar &x : foos) where bar is a temporary intialized by bar.
2573 ///    Suggest either "const bar x" to keep the copying or "const foo& x" to
2574 ///    prevent the copy.
2575 /// 3) for (const foo x : foos) where x is constructed from a reference foo.
2576 ///    Suggest "const foo &x" to prevent the copy.
2577 static void DiagnoseForRangeVariableCopies(Sema &SemaRef,
2578                                            const CXXForRangeStmt *ForStmt) {
2579   if (SemaRef.Diags.isIgnored(diag::warn_for_range_const_reference_copy,
2580                               ForStmt->getLocStart()) &&
2581       SemaRef.Diags.isIgnored(diag::warn_for_range_variable_always_copy,
2582                               ForStmt->getLocStart()) &&
2583       SemaRef.Diags.isIgnored(diag::warn_for_range_copy,
2584                               ForStmt->getLocStart())) {
2585     return;
2586   }
2587 
2588   const VarDecl *VD = ForStmt->getLoopVariable();
2589   if (!VD)
2590     return;
2591 
2592   QualType VariableType = VD->getType();
2593 
2594   if (VariableType->isIncompleteType())
2595     return;
2596 
2597   const Expr *InitExpr = VD->getInit();
2598   if (!InitExpr)
2599     return;
2600 
2601   if (VariableType->isReferenceType()) {
2602     DiagnoseForRangeReferenceVariableCopies(SemaRef, VD,
2603                                             ForStmt->getRangeInit()->getType());
2604   } else if (VariableType.isConstQualified()) {
2605     DiagnoseForRangeConstVariableCopies(SemaRef, VD);
2606   }
2607 }
2608 
2609 /// FinishCXXForRangeStmt - Attach the body to a C++0x for-range statement.
2610 /// This is a separate step from ActOnCXXForRangeStmt because analysis of the
2611 /// body cannot be performed until after the type of the range variable is
2612 /// determined.
2613 StmtResult Sema::FinishCXXForRangeStmt(Stmt *S, Stmt *B) {
2614   if (!S || !B)
2615     return StmtError();
2616 
2617   if (isa<ObjCForCollectionStmt>(S))
2618     return FinishObjCForCollectionStmt(S, B);
2619 
2620   CXXForRangeStmt *ForStmt = cast<CXXForRangeStmt>(S);
2621   ForStmt->setBody(B);
2622 
2623   DiagnoseEmptyStmtBody(ForStmt->getRParenLoc(), B,
2624                         diag::warn_empty_range_based_for_body);
2625 
2626   DiagnoseForRangeVariableCopies(*this, ForStmt);
2627 
2628   return S;
2629 }
2630 
2631 StmtResult Sema::ActOnGotoStmt(SourceLocation GotoLoc,
2632                                SourceLocation LabelLoc,
2633                                LabelDecl *TheDecl) {
2634   getCurFunction()->setHasBranchIntoScope();
2635   TheDecl->markUsed(Context);
2636   return new (Context) GotoStmt(TheDecl, GotoLoc, LabelLoc);
2637 }
2638 
2639 StmtResult
2640 Sema::ActOnIndirectGotoStmt(SourceLocation GotoLoc, SourceLocation StarLoc,
2641                             Expr *E) {
2642   // Convert operand to void*
2643   if (!E->isTypeDependent()) {
2644     QualType ETy = E->getType();
2645     QualType DestTy = Context.getPointerType(Context.VoidTy.withConst());
2646     ExprResult ExprRes = E;
2647     AssignConvertType ConvTy =
2648       CheckSingleAssignmentConstraints(DestTy, ExprRes);
2649     if (ExprRes.isInvalid())
2650       return StmtError();
2651     E = ExprRes.get();
2652     if (DiagnoseAssignmentResult(ConvTy, StarLoc, DestTy, ETy, E, AA_Passing))
2653       return StmtError();
2654   }
2655 
2656   ExprResult ExprRes = ActOnFinishFullExpr(E);
2657   if (ExprRes.isInvalid())
2658     return StmtError();
2659   E = ExprRes.get();
2660 
2661   getCurFunction()->setHasIndirectGoto();
2662 
2663   return new (Context) IndirectGotoStmt(GotoLoc, StarLoc, E);
2664 }
2665 
2666 static void CheckJumpOutOfSEHFinally(Sema &S, SourceLocation Loc,
2667                                      const Scope &DestScope) {
2668   if (!S.CurrentSEHFinally.empty() &&
2669       DestScope.Contains(*S.CurrentSEHFinally.back())) {
2670     S.Diag(Loc, diag::warn_jump_out_of_seh_finally);
2671   }
2672 }
2673 
2674 StmtResult
2675 Sema::ActOnContinueStmt(SourceLocation ContinueLoc, Scope *CurScope) {
2676   Scope *S = CurScope->getContinueParent();
2677   if (!S) {
2678     // C99 6.8.6.2p1: A break shall appear only in or as a loop body.
2679     return StmtError(Diag(ContinueLoc, diag::err_continue_not_in_loop));
2680   }
2681   CheckJumpOutOfSEHFinally(*this, ContinueLoc, *S);
2682 
2683   return new (Context) ContinueStmt(ContinueLoc);
2684 }
2685 
2686 StmtResult
2687 Sema::ActOnBreakStmt(SourceLocation BreakLoc, Scope *CurScope) {
2688   Scope *S = CurScope->getBreakParent();
2689   if (!S) {
2690     // C99 6.8.6.3p1: A break shall appear only in or as a switch/loop body.
2691     return StmtError(Diag(BreakLoc, diag::err_break_not_in_loop_or_switch));
2692   }
2693   if (S->isOpenMPLoopScope())
2694     return StmtError(Diag(BreakLoc, diag::err_omp_loop_cannot_use_stmt)
2695                      << "break");
2696   CheckJumpOutOfSEHFinally(*this, BreakLoc, *S);
2697 
2698   return new (Context) BreakStmt(BreakLoc);
2699 }
2700 
2701 /// \brief Determine whether the given expression is a candidate for
2702 /// copy elision in either a return statement or a throw expression.
2703 ///
2704 /// \param ReturnType If we're determining the copy elision candidate for
2705 /// a return statement, this is the return type of the function. If we're
2706 /// determining the copy elision candidate for a throw expression, this will
2707 /// be a NULL type.
2708 ///
2709 /// \param E The expression being returned from the function or block, or
2710 /// being thrown.
2711 ///
2712 /// \param AllowFunctionParameter Whether we allow function parameters to
2713 /// be considered NRVO candidates. C++ prohibits this for NRVO itself, but
2714 /// we re-use this logic to determine whether we should try to move as part of
2715 /// a return or throw (which does allow function parameters).
2716 ///
2717 /// \returns The NRVO candidate variable, if the return statement may use the
2718 /// NRVO, or NULL if there is no such candidate.
2719 VarDecl *Sema::getCopyElisionCandidate(QualType ReturnType,
2720                                        Expr *E,
2721                                        bool AllowFunctionParameter) {
2722   if (!getLangOpts().CPlusPlus)
2723     return nullptr;
2724 
2725   // - in a return statement in a function [where] ...
2726   // ... the expression is the name of a non-volatile automatic object ...
2727   DeclRefExpr *DR = dyn_cast<DeclRefExpr>(E->IgnoreParens());
2728   if (!DR || DR->refersToEnclosingVariableOrCapture())
2729     return nullptr;
2730   VarDecl *VD = dyn_cast<VarDecl>(DR->getDecl());
2731   if (!VD)
2732     return nullptr;
2733 
2734   if (isCopyElisionCandidate(ReturnType, VD, AllowFunctionParameter))
2735     return VD;
2736   return nullptr;
2737 }
2738 
2739 bool Sema::isCopyElisionCandidate(QualType ReturnType, const VarDecl *VD,
2740                                   bool AllowFunctionParameter) {
2741   QualType VDType = VD->getType();
2742   // - in a return statement in a function with ...
2743   // ... a class return type ...
2744   if (!ReturnType.isNull() && !ReturnType->isDependentType()) {
2745     if (!ReturnType->isRecordType())
2746       return false;
2747     // ... the same cv-unqualified type as the function return type ...
2748     if (!VDType->isDependentType() &&
2749         !Context.hasSameUnqualifiedType(ReturnType, VDType))
2750       return false;
2751   }
2752 
2753   // ...object (other than a function or catch-clause parameter)...
2754   if (VD->getKind() != Decl::Var &&
2755       !(AllowFunctionParameter && VD->getKind() == Decl::ParmVar))
2756     return false;
2757   if (VD->isExceptionVariable()) return false;
2758 
2759   // ...automatic...
2760   if (!VD->hasLocalStorage()) return false;
2761 
2762   // ...non-volatile...
2763   if (VD->getType().isVolatileQualified()) return false;
2764 
2765   // __block variables can't be allocated in a way that permits NRVO.
2766   if (VD->hasAttr<BlocksAttr>()) return false;
2767 
2768   // Variables with higher required alignment than their type's ABI
2769   // alignment cannot use NRVO.
2770   if (!VD->getType()->isDependentType() && VD->hasAttr<AlignedAttr>() &&
2771       Context.getDeclAlign(VD) > Context.getTypeAlignInChars(VD->getType()))
2772     return false;
2773 
2774   return true;
2775 }
2776 
2777 /// \brief Perform the initialization of a potentially-movable value, which
2778 /// is the result of return value.
2779 ///
2780 /// This routine implements C++0x [class.copy]p33, which attempts to treat
2781 /// returned lvalues as rvalues in certain cases (to prefer move construction),
2782 /// then falls back to treating them as lvalues if that failed.
2783 ExprResult
2784 Sema::PerformMoveOrCopyInitialization(const InitializedEntity &Entity,
2785                                       const VarDecl *NRVOCandidate,
2786                                       QualType ResultType,
2787                                       Expr *Value,
2788                                       bool AllowNRVO) {
2789   // C++0x [class.copy]p33:
2790   //   When the criteria for elision of a copy operation are met or would
2791   //   be met save for the fact that the source object is a function
2792   //   parameter, and the object to be copied is designated by an lvalue,
2793   //   overload resolution to select the constructor for the copy is first
2794   //   performed as if the object were designated by an rvalue.
2795   ExprResult Res = ExprError();
2796   if (AllowNRVO &&
2797       (NRVOCandidate || getCopyElisionCandidate(ResultType, Value, true))) {
2798     ImplicitCastExpr AsRvalue(ImplicitCastExpr::OnStack,
2799                               Value->getType(), CK_NoOp, Value, VK_XValue);
2800 
2801     Expr *InitExpr = &AsRvalue;
2802     InitializationKind Kind
2803       = InitializationKind::CreateCopy(Value->getLocStart(),
2804                                        Value->getLocStart());
2805     InitializationSequence Seq(*this, Entity, Kind, InitExpr);
2806 
2807     //   [...] If overload resolution fails, or if the type of the first
2808     //   parameter of the selected constructor is not an rvalue reference
2809     //   to the object's type (possibly cv-qualified), overload resolution
2810     //   is performed again, considering the object as an lvalue.
2811     if (Seq) {
2812       for (InitializationSequence::step_iterator Step = Seq.step_begin(),
2813            StepEnd = Seq.step_end();
2814            Step != StepEnd; ++Step) {
2815         if (Step->Kind != InitializationSequence::SK_ConstructorInitialization)
2816           continue;
2817 
2818         CXXConstructorDecl *Constructor
2819         = cast<CXXConstructorDecl>(Step->Function.Function);
2820 
2821         const RValueReferenceType *RRefType
2822           = Constructor->getParamDecl(0)->getType()
2823                                                  ->getAs<RValueReferenceType>();
2824 
2825         // If we don't meet the criteria, break out now.
2826         if (!RRefType ||
2827             !Context.hasSameUnqualifiedType(RRefType->getPointeeType(),
2828                             Context.getTypeDeclType(Constructor->getParent())))
2829           break;
2830 
2831         // Promote "AsRvalue" to the heap, since we now need this
2832         // expression node to persist.
2833         Value = ImplicitCastExpr::Create(Context, Value->getType(),
2834                                          CK_NoOp, Value, nullptr, VK_XValue);
2835 
2836         // Complete type-checking the initialization of the return type
2837         // using the constructor we found.
2838         Res = Seq.Perform(*this, Entity, Kind, Value);
2839       }
2840     }
2841   }
2842 
2843   // Either we didn't meet the criteria for treating an lvalue as an rvalue,
2844   // above, or overload resolution failed. Either way, we need to try
2845   // (again) now with the return value expression as written.
2846   if (Res.isInvalid())
2847     Res = PerformCopyInitialization(Entity, SourceLocation(), Value);
2848 
2849   return Res;
2850 }
2851 
2852 /// \brief Determine whether the declared return type of the specified function
2853 /// contains 'auto'.
2854 static bool hasDeducedReturnType(FunctionDecl *FD) {
2855   const FunctionProtoType *FPT =
2856       FD->getTypeSourceInfo()->getType()->castAs<FunctionProtoType>();
2857   return FPT->getReturnType()->isUndeducedType();
2858 }
2859 
2860 /// ActOnCapScopeReturnStmt - Utility routine to type-check return statements
2861 /// for capturing scopes.
2862 ///
2863 StmtResult
2864 Sema::ActOnCapScopeReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
2865   // If this is the first return we've seen, infer the return type.
2866   // [expr.prim.lambda]p4 in C++11; block literals follow the same rules.
2867   CapturingScopeInfo *CurCap = cast<CapturingScopeInfo>(getCurFunction());
2868   QualType FnRetType = CurCap->ReturnType;
2869   LambdaScopeInfo *CurLambda = dyn_cast<LambdaScopeInfo>(CurCap);
2870 
2871   if (CurLambda && hasDeducedReturnType(CurLambda->CallOperator)) {
2872     // In C++1y, the return type may involve 'auto'.
2873     // FIXME: Blocks might have a return type of 'auto' explicitly specified.
2874     FunctionDecl *FD = CurLambda->CallOperator;
2875     if (CurCap->ReturnType.isNull())
2876       CurCap->ReturnType = FD->getReturnType();
2877 
2878     AutoType *AT = CurCap->ReturnType->getContainedAutoType();
2879     assert(AT && "lost auto type from lambda return type");
2880     if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
2881       FD->setInvalidDecl();
2882       return StmtError();
2883     }
2884     CurCap->ReturnType = FnRetType = FD->getReturnType();
2885   } else if (CurCap->HasImplicitReturnType) {
2886     // For blocks/lambdas with implicit return types, we check each return
2887     // statement individually, and deduce the common return type when the block
2888     // or lambda is completed.
2889     // FIXME: Fold this into the 'auto' codepath above.
2890     if (RetValExp && !isa<InitListExpr>(RetValExp)) {
2891       ExprResult Result = DefaultFunctionArrayLvalueConversion(RetValExp);
2892       if (Result.isInvalid())
2893         return StmtError();
2894       RetValExp = Result.get();
2895 
2896       // DR1048: even prior to C++14, we should use the 'auto' deduction rules
2897       // when deducing a return type for a lambda-expression (or by extension
2898       // for a block). These rules differ from the stated C++11 rules only in
2899       // that they remove top-level cv-qualifiers.
2900       if (!CurContext->isDependentContext())
2901         FnRetType = RetValExp->getType().getUnqualifiedType();
2902       else
2903         FnRetType = CurCap->ReturnType = Context.DependentTy;
2904     } else {
2905       if (RetValExp) {
2906         // C++11 [expr.lambda.prim]p4 bans inferring the result from an
2907         // initializer list, because it is not an expression (even
2908         // though we represent it as one). We still deduce 'void'.
2909         Diag(ReturnLoc, diag::err_lambda_return_init_list)
2910           << RetValExp->getSourceRange();
2911       }
2912 
2913       FnRetType = Context.VoidTy;
2914     }
2915 
2916     // Although we'll properly infer the type of the block once it's completed,
2917     // make sure we provide a return type now for better error recovery.
2918     if (CurCap->ReturnType.isNull())
2919       CurCap->ReturnType = FnRetType;
2920   }
2921   assert(!FnRetType.isNull());
2922 
2923   if (BlockScopeInfo *CurBlock = dyn_cast<BlockScopeInfo>(CurCap)) {
2924     if (CurBlock->FunctionType->getAs<FunctionType>()->getNoReturnAttr()) {
2925       Diag(ReturnLoc, diag::err_noreturn_block_has_return_expr);
2926       return StmtError();
2927     }
2928   } else if (CapturedRegionScopeInfo *CurRegion =
2929                  dyn_cast<CapturedRegionScopeInfo>(CurCap)) {
2930     Diag(ReturnLoc, diag::err_return_in_captured_stmt) << CurRegion->getRegionName();
2931     return StmtError();
2932   } else {
2933     assert(CurLambda && "unknown kind of captured scope");
2934     if (CurLambda->CallOperator->getType()->getAs<FunctionType>()
2935             ->getNoReturnAttr()) {
2936       Diag(ReturnLoc, diag::err_noreturn_lambda_has_return_expr);
2937       return StmtError();
2938     }
2939   }
2940 
2941   // Otherwise, verify that this result type matches the previous one.  We are
2942   // pickier with blocks than for normal functions because we don't have GCC
2943   // compatibility to worry about here.
2944   const VarDecl *NRVOCandidate = nullptr;
2945   if (FnRetType->isDependentType()) {
2946     // Delay processing for now.  TODO: there are lots of dependent
2947     // types we can conclusively prove aren't void.
2948   } else if (FnRetType->isVoidType()) {
2949     if (RetValExp && !isa<InitListExpr>(RetValExp) &&
2950         !(getLangOpts().CPlusPlus &&
2951           (RetValExp->isTypeDependent() ||
2952            RetValExp->getType()->isVoidType()))) {
2953       if (!getLangOpts().CPlusPlus &&
2954           RetValExp->getType()->isVoidType())
2955         Diag(ReturnLoc, diag::ext_return_has_void_expr) << "literal" << 2;
2956       else {
2957         Diag(ReturnLoc, diag::err_return_block_has_expr);
2958         RetValExp = nullptr;
2959       }
2960     }
2961   } else if (!RetValExp) {
2962     return StmtError(Diag(ReturnLoc, diag::err_block_return_missing_expr));
2963   } else if (!RetValExp->isTypeDependent()) {
2964     // we have a non-void block with an expression, continue checking
2965 
2966     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
2967     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
2968     // function return.
2969 
2970     // In C++ the return statement is handled via a copy initialization.
2971     // the C version of which boils down to CheckSingleAssignmentConstraints.
2972     NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, false);
2973     InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
2974                                                                    FnRetType,
2975                                                       NRVOCandidate != nullptr);
2976     ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
2977                                                      FnRetType, RetValExp);
2978     if (Res.isInvalid()) {
2979       // FIXME: Cleanup temporaries here, anyway?
2980       return StmtError();
2981     }
2982     RetValExp = Res.get();
2983     CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc);
2984   } else {
2985     NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, false);
2986   }
2987 
2988   if (RetValExp) {
2989     ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
2990     if (ER.isInvalid())
2991       return StmtError();
2992     RetValExp = ER.get();
2993   }
2994   ReturnStmt *Result = new (Context) ReturnStmt(ReturnLoc, RetValExp,
2995                                                 NRVOCandidate);
2996 
2997   // If we need to check for the named return value optimization,
2998   // or if we need to infer the return type,
2999   // save the return statement in our scope for later processing.
3000   if (CurCap->HasImplicitReturnType || NRVOCandidate)
3001     FunctionScopes.back()->Returns.push_back(Result);
3002 
3003   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
3004     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
3005 
3006   return Result;
3007 }
3008 
3009 namespace {
3010 /// \brief Marks all typedefs in all local classes in a type referenced.
3011 ///
3012 /// In a function like
3013 /// auto f() {
3014 ///   struct S { typedef int a; };
3015 ///   return S();
3016 /// }
3017 ///
3018 /// the local type escapes and could be referenced in some TUs but not in
3019 /// others. Pretend that all local typedefs are always referenced, to not warn
3020 /// on this. This isn't necessary if f has internal linkage, or the typedef
3021 /// is private.
3022 class LocalTypedefNameReferencer
3023     : public RecursiveASTVisitor<LocalTypedefNameReferencer> {
3024 public:
3025   LocalTypedefNameReferencer(Sema &S) : S(S) {}
3026   bool VisitRecordType(const RecordType *RT);
3027 private:
3028   Sema &S;
3029 };
3030 bool LocalTypedefNameReferencer::VisitRecordType(const RecordType *RT) {
3031   auto *R = dyn_cast<CXXRecordDecl>(RT->getDecl());
3032   if (!R || !R->isLocalClass() || !R->isLocalClass()->isExternallyVisible() ||
3033       R->isDependentType())
3034     return true;
3035   for (auto *TmpD : R->decls())
3036     if (auto *T = dyn_cast<TypedefNameDecl>(TmpD))
3037       if (T->getAccess() != AS_private || R->hasFriends())
3038         S.MarkAnyDeclReferenced(T->getLocation(), T, /*OdrUse=*/false);
3039   return true;
3040 }
3041 }
3042 
3043 TypeLoc Sema::getReturnTypeLoc(FunctionDecl *FD) const {
3044   TypeLoc TL = FD->getTypeSourceInfo()->getTypeLoc().IgnoreParens();
3045   while (auto ATL = TL.getAs<AttributedTypeLoc>())
3046     TL = ATL.getModifiedLoc().IgnoreParens();
3047   return TL.castAs<FunctionProtoTypeLoc>().getReturnLoc();
3048 }
3049 
3050 /// Deduce the return type for a function from a returned expression, per
3051 /// C++1y [dcl.spec.auto]p6.
3052 bool Sema::DeduceFunctionTypeFromReturnExpr(FunctionDecl *FD,
3053                                             SourceLocation ReturnLoc,
3054                                             Expr *&RetExpr,
3055                                             AutoType *AT) {
3056   TypeLoc OrigResultType = getReturnTypeLoc(FD);
3057   QualType Deduced;
3058 
3059   if (RetExpr && isa<InitListExpr>(RetExpr)) {
3060     //  If the deduction is for a return statement and the initializer is
3061     //  a braced-init-list, the program is ill-formed.
3062     Diag(RetExpr->getExprLoc(),
3063          getCurLambda() ? diag::err_lambda_return_init_list
3064                         : diag::err_auto_fn_return_init_list)
3065         << RetExpr->getSourceRange();
3066     return true;
3067   }
3068 
3069   if (FD->isDependentContext()) {
3070     // C++1y [dcl.spec.auto]p12:
3071     //   Return type deduction [...] occurs when the definition is
3072     //   instantiated even if the function body contains a return
3073     //   statement with a non-type-dependent operand.
3074     assert(AT->isDeduced() && "should have deduced to dependent type");
3075     return false;
3076   }
3077 
3078   if (RetExpr) {
3079     //  Otherwise, [...] deduce a value for U using the rules of template
3080     //  argument deduction.
3081     DeduceAutoResult DAR = DeduceAutoType(OrigResultType, RetExpr, Deduced);
3082 
3083     if (DAR == DAR_Failed && !FD->isInvalidDecl())
3084       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_deduction_failure)
3085         << OrigResultType.getType() << RetExpr->getType();
3086 
3087     if (DAR != DAR_Succeeded)
3088       return true;
3089 
3090     // If a local type is part of the returned type, mark its fields as
3091     // referenced.
3092     LocalTypedefNameReferencer Referencer(*this);
3093     Referencer.TraverseType(RetExpr->getType());
3094   } else {
3095     //  In the case of a return with no operand, the initializer is considered
3096     //  to be void().
3097     //
3098     // Deduction here can only succeed if the return type is exactly 'cv auto'
3099     // or 'decltype(auto)', so just check for that case directly.
3100     if (!OrigResultType.getType()->getAs<AutoType>()) {
3101       Diag(ReturnLoc, diag::err_auto_fn_return_void_but_not_auto)
3102         << OrigResultType.getType();
3103       return true;
3104     }
3105     // We always deduce U = void in this case.
3106     Deduced = SubstAutoType(OrigResultType.getType(), Context.VoidTy);
3107     if (Deduced.isNull())
3108       return true;
3109   }
3110 
3111   //  If a function with a declared return type that contains a placeholder type
3112   //  has multiple return statements, the return type is deduced for each return
3113   //  statement. [...] if the type deduced is not the same in each deduction,
3114   //  the program is ill-formed.
3115   QualType DeducedT = AT->getDeducedType();
3116   if (!DeducedT.isNull() && !FD->isInvalidDecl()) {
3117     AutoType *NewAT = Deduced->getContainedAutoType();
3118     // It is possible that NewAT->getDeducedType() is null. When that happens,
3119     // we should not crash, instead we ignore this deduction.
3120     if (NewAT->getDeducedType().isNull())
3121       return false;
3122 
3123     CanQualType OldDeducedType = Context.getCanonicalFunctionResultType(
3124                                    DeducedT);
3125     CanQualType NewDeducedType = Context.getCanonicalFunctionResultType(
3126                                    NewAT->getDeducedType());
3127     if (!FD->isDependentContext() && OldDeducedType != NewDeducedType) {
3128       const LambdaScopeInfo *LambdaSI = getCurLambda();
3129       if (LambdaSI && LambdaSI->HasImplicitReturnType) {
3130         Diag(ReturnLoc, diag::err_typecheck_missing_return_type_incompatible)
3131           << NewAT->getDeducedType() << DeducedT
3132           << true /*IsLambda*/;
3133       } else {
3134         Diag(ReturnLoc, diag::err_auto_fn_different_deductions)
3135           << (AT->isDecltypeAuto() ? 1 : 0)
3136           << NewAT->getDeducedType() << DeducedT;
3137       }
3138       return true;
3139     }
3140   } else if (!FD->isInvalidDecl()) {
3141     // Update all declarations of the function to have the deduced return type.
3142     Context.adjustDeducedFunctionResultType(FD, Deduced);
3143   }
3144 
3145   return false;
3146 }
3147 
3148 StmtResult
3149 Sema::ActOnReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp,
3150                       Scope *CurScope) {
3151   StmtResult R = BuildReturnStmt(ReturnLoc, RetValExp);
3152   if (R.isInvalid()) {
3153     return R;
3154   }
3155 
3156   if (VarDecl *VD =
3157       const_cast<VarDecl*>(cast<ReturnStmt>(R.get())->getNRVOCandidate())) {
3158     CurScope->addNRVOCandidate(VD);
3159   } else {
3160     CurScope->setNoNRVO();
3161   }
3162 
3163   CheckJumpOutOfSEHFinally(*this, ReturnLoc, *CurScope->getFnParent());
3164 
3165   return R;
3166 }
3167 
3168 StmtResult Sema::BuildReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
3169   // Check for unexpanded parameter packs.
3170   if (RetValExp && DiagnoseUnexpandedParameterPack(RetValExp))
3171     return StmtError();
3172 
3173   if (isa<CapturingScopeInfo>(getCurFunction()))
3174     return ActOnCapScopeReturnStmt(ReturnLoc, RetValExp);
3175 
3176   QualType FnRetType;
3177   QualType RelatedRetType;
3178   const AttrVec *Attrs = nullptr;
3179   bool isObjCMethod = false;
3180 
3181   if (const FunctionDecl *FD = getCurFunctionDecl()) {
3182     FnRetType = FD->getReturnType();
3183     if (FD->hasAttrs())
3184       Attrs = &FD->getAttrs();
3185     if (FD->isNoReturn())
3186       Diag(ReturnLoc, diag::warn_noreturn_function_has_return_expr)
3187         << FD->getDeclName();
3188   } else if (ObjCMethodDecl *MD = getCurMethodDecl()) {
3189     FnRetType = MD->getReturnType();
3190     isObjCMethod = true;
3191     if (MD->hasAttrs())
3192       Attrs = &MD->getAttrs();
3193     if (MD->hasRelatedResultType() && MD->getClassInterface()) {
3194       // In the implementation of a method with a related return type, the
3195       // type used to type-check the validity of return statements within the
3196       // method body is a pointer to the type of the class being implemented.
3197       RelatedRetType = Context.getObjCInterfaceType(MD->getClassInterface());
3198       RelatedRetType = Context.getObjCObjectPointerType(RelatedRetType);
3199     }
3200   } else // If we don't have a function/method context, bail.
3201     return StmtError();
3202 
3203   // FIXME: Add a flag to the ScopeInfo to indicate whether we're performing
3204   // deduction.
3205   if (getLangOpts().CPlusPlus14) {
3206     if (AutoType *AT = FnRetType->getContainedAutoType()) {
3207       FunctionDecl *FD = cast<FunctionDecl>(CurContext);
3208       if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
3209         FD->setInvalidDecl();
3210         return StmtError();
3211       } else {
3212         FnRetType = FD->getReturnType();
3213       }
3214     }
3215   }
3216 
3217   bool HasDependentReturnType = FnRetType->isDependentType();
3218 
3219   ReturnStmt *Result = nullptr;
3220   if (FnRetType->isVoidType()) {
3221     if (RetValExp) {
3222       if (isa<InitListExpr>(RetValExp)) {
3223         // We simply never allow init lists as the return value of void
3224         // functions. This is compatible because this was never allowed before,
3225         // so there's no legacy code to deal with.
3226         NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
3227         int FunctionKind = 0;
3228         if (isa<ObjCMethodDecl>(CurDecl))
3229           FunctionKind = 1;
3230         else if (isa<CXXConstructorDecl>(CurDecl))
3231           FunctionKind = 2;
3232         else if (isa<CXXDestructorDecl>(CurDecl))
3233           FunctionKind = 3;
3234 
3235         Diag(ReturnLoc, diag::err_return_init_list)
3236           << CurDecl->getDeclName() << FunctionKind
3237           << RetValExp->getSourceRange();
3238 
3239         // Drop the expression.
3240         RetValExp = nullptr;
3241       } else if (!RetValExp->isTypeDependent()) {
3242         // C99 6.8.6.4p1 (ext_ since GCC warns)
3243         unsigned D = diag::ext_return_has_expr;
3244         if (RetValExp->getType()->isVoidType()) {
3245           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
3246           if (isa<CXXConstructorDecl>(CurDecl) ||
3247               isa<CXXDestructorDecl>(CurDecl))
3248             D = diag::err_ctor_dtor_returns_void;
3249           else
3250             D = diag::ext_return_has_void_expr;
3251         }
3252         else {
3253           ExprResult Result = RetValExp;
3254           Result = IgnoredValueConversions(Result.get());
3255           if (Result.isInvalid())
3256             return StmtError();
3257           RetValExp = Result.get();
3258           RetValExp = ImpCastExprToType(RetValExp,
3259                                         Context.VoidTy, CK_ToVoid).get();
3260         }
3261         // return of void in constructor/destructor is illegal in C++.
3262         if (D == diag::err_ctor_dtor_returns_void) {
3263           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
3264           Diag(ReturnLoc, D)
3265             << CurDecl->getDeclName() << isa<CXXDestructorDecl>(CurDecl)
3266             << RetValExp->getSourceRange();
3267         }
3268         // return (some void expression); is legal in C++.
3269         else if (D != diag::ext_return_has_void_expr ||
3270                  !getLangOpts().CPlusPlus) {
3271           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
3272 
3273           int FunctionKind = 0;
3274           if (isa<ObjCMethodDecl>(CurDecl))
3275             FunctionKind = 1;
3276           else if (isa<CXXConstructorDecl>(CurDecl))
3277             FunctionKind = 2;
3278           else if (isa<CXXDestructorDecl>(CurDecl))
3279             FunctionKind = 3;
3280 
3281           Diag(ReturnLoc, D)
3282             << CurDecl->getDeclName() << FunctionKind
3283             << RetValExp->getSourceRange();
3284         }
3285       }
3286 
3287       if (RetValExp) {
3288         ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
3289         if (ER.isInvalid())
3290           return StmtError();
3291         RetValExp = ER.get();
3292       }
3293     }
3294 
3295     Result = new (Context) ReturnStmt(ReturnLoc, RetValExp, nullptr);
3296   } else if (!RetValExp && !HasDependentReturnType) {
3297     FunctionDecl *FD = getCurFunctionDecl();
3298 
3299     unsigned DiagID;
3300     if (getLangOpts().CPlusPlus11 && FD && FD->isConstexpr()) {
3301       // C++11 [stmt.return]p2
3302       DiagID = diag::err_constexpr_return_missing_expr;
3303       FD->setInvalidDecl();
3304     } else if (getLangOpts().C99) {
3305       // C99 6.8.6.4p1 (ext_ since GCC warns)
3306       DiagID = diag::ext_return_missing_expr;
3307     } else {
3308       // C90 6.6.6.4p4
3309       DiagID = diag::warn_return_missing_expr;
3310     }
3311 
3312     if (FD)
3313       Diag(ReturnLoc, DiagID) << FD->getIdentifier() << 0/*fn*/;
3314     else
3315       Diag(ReturnLoc, DiagID) << getCurMethodDecl()->getDeclName() << 1/*meth*/;
3316 
3317     Result = new (Context) ReturnStmt(ReturnLoc);
3318   } else {
3319     assert(RetValExp || HasDependentReturnType);
3320     const VarDecl *NRVOCandidate = nullptr;
3321 
3322     QualType RetType = RelatedRetType.isNull() ? FnRetType : RelatedRetType;
3323 
3324     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
3325     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
3326     // function return.
3327 
3328     // In C++ the return statement is handled via a copy initialization,
3329     // the C version of which boils down to CheckSingleAssignmentConstraints.
3330     if (RetValExp)
3331       NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, false);
3332     if (!HasDependentReturnType && !RetValExp->isTypeDependent()) {
3333       // we have a non-void function with an expression, continue checking
3334       InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
3335                                                                      RetType,
3336                                                       NRVOCandidate != nullptr);
3337       ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
3338                                                        RetType, RetValExp);
3339       if (Res.isInvalid()) {
3340         // FIXME: Clean up temporaries here anyway?
3341         return StmtError();
3342       }
3343       RetValExp = Res.getAs<Expr>();
3344 
3345       // If we have a related result type, we need to implicitly
3346       // convert back to the formal result type.  We can't pretend to
3347       // initialize the result again --- we might end double-retaining
3348       // --- so instead we initialize a notional temporary.
3349       if (!RelatedRetType.isNull()) {
3350         Entity = InitializedEntity::InitializeRelatedResult(getCurMethodDecl(),
3351                                                             FnRetType);
3352         Res = PerformCopyInitialization(Entity, ReturnLoc, RetValExp);
3353         if (Res.isInvalid()) {
3354           // FIXME: Clean up temporaries here anyway?
3355           return StmtError();
3356         }
3357         RetValExp = Res.getAs<Expr>();
3358       }
3359 
3360       CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc, isObjCMethod, Attrs,
3361                          getCurFunctionDecl());
3362     }
3363 
3364     if (RetValExp) {
3365       ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
3366       if (ER.isInvalid())
3367         return StmtError();
3368       RetValExp = ER.get();
3369     }
3370     Result = new (Context) ReturnStmt(ReturnLoc, RetValExp, NRVOCandidate);
3371   }
3372 
3373   // If we need to check for the named return value optimization, save the
3374   // return statement in our scope for later processing.
3375   if (Result->getNRVOCandidate())
3376     FunctionScopes.back()->Returns.push_back(Result);
3377 
3378   if (FunctionScopes.back()->FirstReturnLoc.isInvalid())
3379     FunctionScopes.back()->FirstReturnLoc = ReturnLoc;
3380 
3381   return Result;
3382 }
3383 
3384 StmtResult
3385 Sema::ActOnObjCAtCatchStmt(SourceLocation AtLoc,
3386                            SourceLocation RParen, Decl *Parm,
3387                            Stmt *Body) {
3388   VarDecl *Var = cast_or_null<VarDecl>(Parm);
3389   if (Var && Var->isInvalidDecl())
3390     return StmtError();
3391 
3392   return new (Context) ObjCAtCatchStmt(AtLoc, RParen, Var, Body);
3393 }
3394 
3395 StmtResult
3396 Sema::ActOnObjCAtFinallyStmt(SourceLocation AtLoc, Stmt *Body) {
3397   return new (Context) ObjCAtFinallyStmt(AtLoc, Body);
3398 }
3399 
3400 StmtResult
3401 Sema::ActOnObjCAtTryStmt(SourceLocation AtLoc, Stmt *Try,
3402                          MultiStmtArg CatchStmts, Stmt *Finally) {
3403   if (!getLangOpts().ObjCExceptions)
3404     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@try";
3405 
3406   getCurFunction()->setHasBranchProtectedScope();
3407   unsigned NumCatchStmts = CatchStmts.size();
3408   return ObjCAtTryStmt::Create(Context, AtLoc, Try, CatchStmts.data(),
3409                                NumCatchStmts, Finally);
3410 }
3411 
3412 StmtResult Sema::BuildObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw) {
3413   if (Throw) {
3414     ExprResult Result = DefaultLvalueConversion(Throw);
3415     if (Result.isInvalid())
3416       return StmtError();
3417 
3418     Result = ActOnFinishFullExpr(Result.get());
3419     if (Result.isInvalid())
3420       return StmtError();
3421     Throw = Result.get();
3422 
3423     QualType ThrowType = Throw->getType();
3424     // Make sure the expression type is an ObjC pointer or "void *".
3425     if (!ThrowType->isDependentType() &&
3426         !ThrowType->isObjCObjectPointerType()) {
3427       const PointerType *PT = ThrowType->getAs<PointerType>();
3428       if (!PT || !PT->getPointeeType()->isVoidType())
3429         return StmtError(Diag(AtLoc, diag::error_objc_throw_expects_object)
3430                          << Throw->getType() << Throw->getSourceRange());
3431     }
3432   }
3433 
3434   return new (Context) ObjCAtThrowStmt(AtLoc, Throw);
3435 }
3436 
3437 StmtResult
3438 Sema::ActOnObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw,
3439                            Scope *CurScope) {
3440   if (!getLangOpts().ObjCExceptions)
3441     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@throw";
3442 
3443   if (!Throw) {
3444     // @throw without an expression designates a rethrow (which must occur
3445     // in the context of an @catch clause).
3446     Scope *AtCatchParent = CurScope;
3447     while (AtCatchParent && !AtCatchParent->isAtCatchScope())
3448       AtCatchParent = AtCatchParent->getParent();
3449     if (!AtCatchParent)
3450       return StmtError(Diag(AtLoc, diag::error_rethrow_used_outside_catch));
3451   }
3452   return BuildObjCAtThrowStmt(AtLoc, Throw);
3453 }
3454 
3455 ExprResult
3456 Sema::ActOnObjCAtSynchronizedOperand(SourceLocation atLoc, Expr *operand) {
3457   ExprResult result = DefaultLvalueConversion(operand);
3458   if (result.isInvalid())
3459     return ExprError();
3460   operand = result.get();
3461 
3462   // Make sure the expression type is an ObjC pointer or "void *".
3463   QualType type = operand->getType();
3464   if (!type->isDependentType() &&
3465       !type->isObjCObjectPointerType()) {
3466     const PointerType *pointerType = type->getAs<PointerType>();
3467     if (!pointerType || !pointerType->getPointeeType()->isVoidType()) {
3468       if (getLangOpts().CPlusPlus) {
3469         if (RequireCompleteType(atLoc, type,
3470                                 diag::err_incomplete_receiver_type))
3471           return Diag(atLoc, diag::error_objc_synchronized_expects_object)
3472                    << type << operand->getSourceRange();
3473 
3474         ExprResult result = PerformContextuallyConvertToObjCPointer(operand);
3475         if (!result.isUsable())
3476           return Diag(atLoc, diag::error_objc_synchronized_expects_object)
3477                    << type << operand->getSourceRange();
3478 
3479         operand = result.get();
3480       } else {
3481           return Diag(atLoc, diag::error_objc_synchronized_expects_object)
3482                    << type << operand->getSourceRange();
3483       }
3484     }
3485   }
3486 
3487   // The operand to @synchronized is a full-expression.
3488   return ActOnFinishFullExpr(operand);
3489 }
3490 
3491 StmtResult
3492 Sema::ActOnObjCAtSynchronizedStmt(SourceLocation AtLoc, Expr *SyncExpr,
3493                                   Stmt *SyncBody) {
3494   // We can't jump into or indirect-jump out of a @synchronized block.
3495   getCurFunction()->setHasBranchProtectedScope();
3496   return new (Context) ObjCAtSynchronizedStmt(AtLoc, SyncExpr, SyncBody);
3497 }
3498 
3499 /// ActOnCXXCatchBlock - Takes an exception declaration and a handler block
3500 /// and creates a proper catch handler from them.
3501 StmtResult
3502 Sema::ActOnCXXCatchBlock(SourceLocation CatchLoc, Decl *ExDecl,
3503                          Stmt *HandlerBlock) {
3504   // There's nothing to test that ActOnExceptionDecl didn't already test.
3505   return new (Context)
3506       CXXCatchStmt(CatchLoc, cast_or_null<VarDecl>(ExDecl), HandlerBlock);
3507 }
3508 
3509 StmtResult
3510 Sema::ActOnObjCAutoreleasePoolStmt(SourceLocation AtLoc, Stmt *Body) {
3511   getCurFunction()->setHasBranchProtectedScope();
3512   return new (Context) ObjCAutoreleasePoolStmt(AtLoc, Body);
3513 }
3514 
3515 namespace {
3516 class CatchHandlerType {
3517   QualType QT;
3518   unsigned IsPointer : 1;
3519 
3520   // This is a special constructor to be used only with DenseMapInfo's
3521   // getEmptyKey() and getTombstoneKey() functions.
3522   friend struct llvm::DenseMapInfo<CatchHandlerType>;
3523   enum Unique { ForDenseMap };
3524   CatchHandlerType(QualType QT, Unique) : QT(QT), IsPointer(false) {}
3525 
3526 public:
3527   /// Used when creating a CatchHandlerType from a handler type; will determine
3528   /// whether the type is a pointer or reference and will strip off the top
3529   /// level pointer and cv-qualifiers.
3530   CatchHandlerType(QualType Q) : QT(Q), IsPointer(false) {
3531     if (QT->isPointerType())
3532       IsPointer = true;
3533 
3534     if (IsPointer || QT->isReferenceType())
3535       QT = QT->getPointeeType();
3536     QT = QT.getUnqualifiedType();
3537   }
3538 
3539   /// Used when creating a CatchHandlerType from a base class type; pretends the
3540   /// type passed in had the pointer qualifier, does not need to get an
3541   /// unqualified type.
3542   CatchHandlerType(QualType QT, bool IsPointer)
3543       : QT(QT), IsPointer(IsPointer) {}
3544 
3545   QualType underlying() const { return QT; }
3546   bool isPointer() const { return IsPointer; }
3547 
3548   friend bool operator==(const CatchHandlerType &LHS,
3549                          const CatchHandlerType &RHS) {
3550     // If the pointer qualification does not match, we can return early.
3551     if (LHS.IsPointer != RHS.IsPointer)
3552       return false;
3553     // Otherwise, check the underlying type without cv-qualifiers.
3554     return LHS.QT == RHS.QT;
3555   }
3556 };
3557 } // namespace
3558 
3559 namespace llvm {
3560 template <> struct DenseMapInfo<CatchHandlerType> {
3561   static CatchHandlerType getEmptyKey() {
3562     return CatchHandlerType(DenseMapInfo<QualType>::getEmptyKey(),
3563                        CatchHandlerType::ForDenseMap);
3564   }
3565 
3566   static CatchHandlerType getTombstoneKey() {
3567     return CatchHandlerType(DenseMapInfo<QualType>::getTombstoneKey(),
3568                        CatchHandlerType::ForDenseMap);
3569   }
3570 
3571   static unsigned getHashValue(const CatchHandlerType &Base) {
3572     return DenseMapInfo<QualType>::getHashValue(Base.underlying());
3573   }
3574 
3575   static bool isEqual(const CatchHandlerType &LHS,
3576                       const CatchHandlerType &RHS) {
3577     return LHS == RHS;
3578   }
3579 };
3580 }
3581 
3582 namespace {
3583 class CatchTypePublicBases {
3584   ASTContext &Ctx;
3585   const llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> &TypesToCheck;
3586   const bool CheckAgainstPointer;
3587 
3588   CXXCatchStmt *FoundHandler;
3589   CanQualType FoundHandlerType;
3590 
3591 public:
3592   CatchTypePublicBases(
3593       ASTContext &Ctx,
3594       const llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> &T, bool C)
3595       : Ctx(Ctx), TypesToCheck(T), CheckAgainstPointer(C),
3596         FoundHandler(nullptr) {}
3597 
3598   CXXCatchStmt *getFoundHandler() const { return FoundHandler; }
3599   CanQualType getFoundHandlerType() const { return FoundHandlerType; }
3600 
3601   bool operator()(const CXXBaseSpecifier *S, CXXBasePath &) {
3602     if (S->getAccessSpecifier() == AccessSpecifier::AS_public) {
3603       CatchHandlerType Check(S->getType(), CheckAgainstPointer);
3604       const auto &M = TypesToCheck;
3605       auto I = M.find(Check);
3606       if (I != M.end()) {
3607         FoundHandler = I->second;
3608         FoundHandlerType = Ctx.getCanonicalType(S->getType());
3609         return true;
3610       }
3611     }
3612     return false;
3613   }
3614 };
3615 }
3616 
3617 /// ActOnCXXTryBlock - Takes a try compound-statement and a number of
3618 /// handlers and creates a try statement from them.
3619 StmtResult Sema::ActOnCXXTryBlock(SourceLocation TryLoc, Stmt *TryBlock,
3620                                   ArrayRef<Stmt *> Handlers) {
3621   // Don't report an error if 'try' is used in system headers.
3622   if (!getLangOpts().CXXExceptions &&
3623       !getSourceManager().isInSystemHeader(TryLoc))
3624     Diag(TryLoc, diag::err_exceptions_disabled) << "try";
3625 
3626   if (getCurScope() && getCurScope()->isOpenMPSimdDirectiveScope())
3627     Diag(TryLoc, diag::err_omp_simd_region_cannot_use_stmt) << "try";
3628 
3629   sema::FunctionScopeInfo *FSI = getCurFunction();
3630 
3631   // C++ try is incompatible with SEH __try.
3632   if (!getLangOpts().Borland && FSI->FirstSEHTryLoc.isValid()) {
3633     Diag(TryLoc, diag::err_mixing_cxx_try_seh_try);
3634     Diag(FSI->FirstSEHTryLoc, diag::note_conflicting_try_here) << "'__try'";
3635   }
3636 
3637   const unsigned NumHandlers = Handlers.size();
3638   assert(!Handlers.empty() &&
3639          "The parser shouldn't call this if there are no handlers.");
3640 
3641   llvm::DenseMap<CatchHandlerType, CXXCatchStmt *> HandledTypes;
3642   for (unsigned i = 0; i < NumHandlers; ++i) {
3643     CXXCatchStmt *H = cast<CXXCatchStmt>(Handlers[i]);
3644 
3645     // Diagnose when the handler is a catch-all handler, but it isn't the last
3646     // handler for the try block. [except.handle]p5. Also, skip exception
3647     // declarations that are invalid, since we can't usefully report on them.
3648     if (!H->getExceptionDecl()) {
3649       if (i < NumHandlers - 1)
3650         return StmtError(Diag(H->getLocStart(), diag::err_early_catch_all));
3651       continue;
3652     } else if (H->getExceptionDecl()->isInvalidDecl())
3653       continue;
3654 
3655     // Walk the type hierarchy to diagnose when this type has already been
3656     // handled (duplication), or cannot be handled (derivation inversion). We
3657     // ignore top-level cv-qualifiers, per [except.handle]p3
3658     CatchHandlerType HandlerCHT =
3659         (QualType)Context.getCanonicalType(H->getCaughtType());
3660 
3661     // We can ignore whether the type is a reference or a pointer; we need the
3662     // underlying declaration type in order to get at the underlying record
3663     // decl, if there is one.
3664     QualType Underlying = HandlerCHT.underlying();
3665     if (auto *RD = Underlying->getAsCXXRecordDecl()) {
3666       if (!RD->hasDefinition())
3667         continue;
3668       // Check that none of the public, unambiguous base classes are in the
3669       // map ([except.handle]p1). Give the base classes the same pointer
3670       // qualification as the original type we are basing off of. This allows
3671       // comparison against the handler type using the same top-level pointer
3672       // as the original type.
3673       CXXBasePaths Paths;
3674       Paths.setOrigin(RD);
3675       CatchTypePublicBases CTPB(Context, HandledTypes, HandlerCHT.isPointer());
3676       if (RD->lookupInBases(CTPB, Paths)) {
3677         const CXXCatchStmt *Problem = CTPB.getFoundHandler();
3678         if (!Paths.isAmbiguous(CTPB.getFoundHandlerType())) {
3679           Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
3680                diag::warn_exception_caught_by_earlier_handler)
3681               << H->getCaughtType();
3682           Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
3683                 diag::note_previous_exception_handler)
3684               << Problem->getCaughtType();
3685         }
3686       }
3687     }
3688 
3689     // Add the type the list of ones we have handled; diagnose if we've already
3690     // handled it.
3691     auto R = HandledTypes.insert(std::make_pair(H->getCaughtType(), H));
3692     if (!R.second) {
3693       const CXXCatchStmt *Problem = R.first->second;
3694       Diag(H->getExceptionDecl()->getTypeSpecStartLoc(),
3695            diag::warn_exception_caught_by_earlier_handler)
3696           << H->getCaughtType();
3697       Diag(Problem->getExceptionDecl()->getTypeSpecStartLoc(),
3698            diag::note_previous_exception_handler)
3699           << Problem->getCaughtType();
3700     }
3701   }
3702 
3703   FSI->setHasCXXTry(TryLoc);
3704 
3705   return CXXTryStmt::Create(Context, TryLoc, TryBlock, Handlers);
3706 }
3707 
3708 StmtResult Sema::ActOnSEHTryBlock(bool IsCXXTry, SourceLocation TryLoc,
3709                                   Stmt *TryBlock, Stmt *Handler) {
3710   assert(TryBlock && Handler);
3711 
3712   sema::FunctionScopeInfo *FSI = getCurFunction();
3713 
3714   // SEH __try is incompatible with C++ try. Borland appears to support this,
3715   // however.
3716   if (!getLangOpts().Borland) {
3717     if (FSI->FirstCXXTryLoc.isValid()) {
3718       Diag(TryLoc, diag::err_mixing_cxx_try_seh_try);
3719       Diag(FSI->FirstCXXTryLoc, diag::note_conflicting_try_here) << "'try'";
3720     }
3721   }
3722 
3723   FSI->setHasSEHTry(TryLoc);
3724 
3725   // Reject __try in Obj-C methods, blocks, and captured decls, since we don't
3726   // track if they use SEH.
3727   DeclContext *DC = CurContext;
3728   while (DC && !DC->isFunctionOrMethod())
3729     DC = DC->getParent();
3730   FunctionDecl *FD = dyn_cast_or_null<FunctionDecl>(DC);
3731   if (FD)
3732     FD->setUsesSEHTry(true);
3733   else
3734     Diag(TryLoc, diag::err_seh_try_outside_functions);
3735 
3736   // Reject __try on unsupported targets.
3737   if (!Context.getTargetInfo().isSEHTrySupported())
3738     Diag(TryLoc, diag::err_seh_try_unsupported);
3739 
3740   return SEHTryStmt::Create(Context, IsCXXTry, TryLoc, TryBlock, Handler);
3741 }
3742 
3743 StmtResult
3744 Sema::ActOnSEHExceptBlock(SourceLocation Loc,
3745                           Expr *FilterExpr,
3746                           Stmt *Block) {
3747   assert(FilterExpr && Block);
3748 
3749   if(!FilterExpr->getType()->isIntegerType()) {
3750     return StmtError(Diag(FilterExpr->getExprLoc(),
3751                      diag::err_filter_expression_integral)
3752                      << FilterExpr->getType());
3753   }
3754 
3755   return SEHExceptStmt::Create(Context,Loc,FilterExpr,Block);
3756 }
3757 
3758 void Sema::ActOnStartSEHFinallyBlock() {
3759   CurrentSEHFinally.push_back(CurScope);
3760 }
3761 
3762 void Sema::ActOnAbortSEHFinallyBlock() {
3763   CurrentSEHFinally.pop_back();
3764 }
3765 
3766 StmtResult Sema::ActOnFinishSEHFinallyBlock(SourceLocation Loc, Stmt *Block) {
3767   assert(Block);
3768   CurrentSEHFinally.pop_back();
3769   return SEHFinallyStmt::Create(Context, Loc, Block);
3770 }
3771 
3772 StmtResult
3773 Sema::ActOnSEHLeaveStmt(SourceLocation Loc, Scope *CurScope) {
3774   Scope *SEHTryParent = CurScope;
3775   while (SEHTryParent && !SEHTryParent->isSEHTryScope())
3776     SEHTryParent = SEHTryParent->getParent();
3777   if (!SEHTryParent)
3778     return StmtError(Diag(Loc, diag::err_ms___leave_not_in___try));
3779   CheckJumpOutOfSEHFinally(*this, Loc, *SEHTryParent);
3780 
3781   return new (Context) SEHLeaveStmt(Loc);
3782 }
3783 
3784 StmtResult Sema::BuildMSDependentExistsStmt(SourceLocation KeywordLoc,
3785                                             bool IsIfExists,
3786                                             NestedNameSpecifierLoc QualifierLoc,
3787                                             DeclarationNameInfo NameInfo,
3788                                             Stmt *Nested)
3789 {
3790   return new (Context) MSDependentExistsStmt(KeywordLoc, IsIfExists,
3791                                              QualifierLoc, NameInfo,
3792                                              cast<CompoundStmt>(Nested));
3793 }
3794 
3795 
3796 StmtResult Sema::ActOnMSDependentExistsStmt(SourceLocation KeywordLoc,
3797                                             bool IsIfExists,
3798                                             CXXScopeSpec &SS,
3799                                             UnqualifiedId &Name,
3800                                             Stmt *Nested) {
3801   return BuildMSDependentExistsStmt(KeywordLoc, IsIfExists,
3802                                     SS.getWithLocInContext(Context),
3803                                     GetNameFromUnqualifiedId(Name),
3804                                     Nested);
3805 }
3806 
3807 RecordDecl*
3808 Sema::CreateCapturedStmtRecordDecl(CapturedDecl *&CD, SourceLocation Loc,
3809                                    unsigned NumParams) {
3810   DeclContext *DC = CurContext;
3811   while (!(DC->isFunctionOrMethod() || DC->isRecord() || DC->isFileContext()))
3812     DC = DC->getParent();
3813 
3814   RecordDecl *RD = nullptr;
3815   if (getLangOpts().CPlusPlus)
3816     RD = CXXRecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc,
3817                                /*Id=*/nullptr);
3818   else
3819     RD = RecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc, /*Id=*/nullptr);
3820 
3821   RD->setCapturedRecord();
3822   DC->addDecl(RD);
3823   RD->setImplicit();
3824   RD->startDefinition();
3825 
3826   assert(NumParams > 0 && "CapturedStmt requires context parameter");
3827   CD = CapturedDecl::Create(Context, CurContext, NumParams);
3828   DC->addDecl(CD);
3829   return RD;
3830 }
3831 
3832 static void buildCapturedStmtCaptureList(
3833     SmallVectorImpl<CapturedStmt::Capture> &Captures,
3834     SmallVectorImpl<Expr *> &CaptureInits,
3835     ArrayRef<CapturingScopeInfo::Capture> Candidates) {
3836 
3837   typedef ArrayRef<CapturingScopeInfo::Capture>::const_iterator CaptureIter;
3838   for (CaptureIter Cap = Candidates.begin(); Cap != Candidates.end(); ++Cap) {
3839 
3840     if (Cap->isThisCapture()) {
3841       Captures.push_back(CapturedStmt::Capture(Cap->getLocation(),
3842                                                CapturedStmt::VCK_This));
3843       CaptureInits.push_back(Cap->getInitExpr());
3844       continue;
3845     } else if (Cap->isVLATypeCapture()) {
3846       Captures.push_back(
3847           CapturedStmt::Capture(Cap->getLocation(), CapturedStmt::VCK_VLAType));
3848       CaptureInits.push_back(nullptr);
3849       continue;
3850     }
3851 
3852     Captures.push_back(CapturedStmt::Capture(Cap->getLocation(),
3853                                              Cap->isReferenceCapture()
3854                                                  ? CapturedStmt::VCK_ByRef
3855                                                  : CapturedStmt::VCK_ByCopy,
3856                                              Cap->getVariable()));
3857     CaptureInits.push_back(Cap->getInitExpr());
3858   }
3859 }
3860 
3861 void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
3862                                     CapturedRegionKind Kind,
3863                                     unsigned NumParams) {
3864   CapturedDecl *CD = nullptr;
3865   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, NumParams);
3866 
3867   // Build the context parameter
3868   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
3869   IdentifierInfo *ParamName = &Context.Idents.get("__context");
3870   QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
3871   ImplicitParamDecl *Param
3872     = ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType);
3873   DC->addDecl(Param);
3874 
3875   CD->setContextParam(0, Param);
3876 
3877   // Enter the capturing scope for this captured region.
3878   PushCapturedRegionScope(CurScope, CD, RD, Kind);
3879 
3880   if (CurScope)
3881     PushDeclContext(CurScope, CD);
3882   else
3883     CurContext = CD;
3884 
3885   PushExpressionEvaluationContext(PotentiallyEvaluated);
3886 }
3887 
3888 void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
3889                                     CapturedRegionKind Kind,
3890                                     ArrayRef<CapturedParamNameType> Params) {
3891   CapturedDecl *CD = nullptr;
3892   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, Params.size());
3893 
3894   // Build the context parameter
3895   DeclContext *DC = CapturedDecl::castToDeclContext(CD);
3896   bool ContextIsFound = false;
3897   unsigned ParamNum = 0;
3898   for (ArrayRef<CapturedParamNameType>::iterator I = Params.begin(),
3899                                                  E = Params.end();
3900        I != E; ++I, ++ParamNum) {
3901     if (I->second.isNull()) {
3902       assert(!ContextIsFound &&
3903              "null type has been found already for '__context' parameter");
3904       IdentifierInfo *ParamName = &Context.Idents.get("__context");
3905       QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
3906       ImplicitParamDecl *Param
3907         = ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType);
3908       DC->addDecl(Param);
3909       CD->setContextParam(ParamNum, Param);
3910       ContextIsFound = true;
3911     } else {
3912       IdentifierInfo *ParamName = &Context.Idents.get(I->first);
3913       ImplicitParamDecl *Param
3914         = ImplicitParamDecl::Create(Context, DC, Loc, ParamName, I->second);
3915       DC->addDecl(Param);
3916       CD->setParam(ParamNum, Param);
3917     }
3918   }
3919   assert(ContextIsFound && "no null type for '__context' parameter");
3920   if (!ContextIsFound) {
3921     // Add __context implicitly if it is not specified.
3922     IdentifierInfo *ParamName = &Context.Idents.get("__context");
3923     QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
3924     ImplicitParamDecl *Param =
3925         ImplicitParamDecl::Create(Context, DC, Loc, ParamName, ParamType);
3926     DC->addDecl(Param);
3927     CD->setContextParam(ParamNum, Param);
3928   }
3929   // Enter the capturing scope for this captured region.
3930   PushCapturedRegionScope(CurScope, CD, RD, Kind);
3931 
3932   if (CurScope)
3933     PushDeclContext(CurScope, CD);
3934   else
3935     CurContext = CD;
3936 
3937   PushExpressionEvaluationContext(PotentiallyEvaluated);
3938 }
3939 
3940 void Sema::ActOnCapturedRegionError() {
3941   DiscardCleanupsInEvaluationContext();
3942   PopExpressionEvaluationContext();
3943 
3944   CapturedRegionScopeInfo *RSI = getCurCapturedRegion();
3945   RecordDecl *Record = RSI->TheRecordDecl;
3946   Record->setInvalidDecl();
3947 
3948   SmallVector<Decl*, 4> Fields(Record->fields());
3949   ActOnFields(/*Scope=*/nullptr, Record->getLocation(), Record, Fields,
3950               SourceLocation(), SourceLocation(), /*AttributeList=*/nullptr);
3951 
3952   PopDeclContext();
3953   PopFunctionScopeInfo();
3954 }
3955 
3956 StmtResult Sema::ActOnCapturedRegionEnd(Stmt *S) {
3957   CapturedRegionScopeInfo *RSI = getCurCapturedRegion();
3958 
3959   SmallVector<CapturedStmt::Capture, 4> Captures;
3960   SmallVector<Expr *, 4> CaptureInits;
3961   buildCapturedStmtCaptureList(Captures, CaptureInits, RSI->Captures);
3962 
3963   CapturedDecl *CD = RSI->TheCapturedDecl;
3964   RecordDecl *RD = RSI->TheRecordDecl;
3965 
3966   CapturedStmt *Res = CapturedStmt::Create(getASTContext(), S,
3967                                            RSI->CapRegionKind, Captures,
3968                                            CaptureInits, CD, RD);
3969 
3970   CD->setBody(Res->getCapturedStmt());
3971   RD->completeDefinition();
3972 
3973   DiscardCleanupsInEvaluationContext();
3974   PopExpressionEvaluationContext();
3975 
3976   PopDeclContext();
3977   PopFunctionScopeInfo();
3978 
3979   return Res;
3980 }
3981