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