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 Owned(static_cast<Stmt*>(FE.take()));
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 Owned(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 Owned(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(0);
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, 0, 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 Owned(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 != 0) && "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).take();
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).take();
374       // Recover from an error by just forgetting about it.
375     }
376   }
377 
378   LHSVal = ActOnFinishFullExpr(LHSVal, LHSVal->getExprLoc(), false,
379                                getLangOpts().CPlusPlus11).take();
380   if (RHSVal)
381     RHSVal = ActOnFinishFullExpr(RHSVal, RHSVal->getExprLoc(), false,
382                                  getLangOpts().CPlusPlus11).take();
383 
384   CaseStmt *CS = new (Context) CaseStmt(LHSVal, RHSVal, CaseLoc, DotDotDotLoc,
385                                         ColonLoc);
386   getCurFunction()->SwitchStack.back()->addSwitchCase(CS);
387   return Owned(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 Owned(SubStmt);
406   }
407 
408   DefaultStmt *DS = new (Context) DefaultStmt(DefaultLoc, ColonLoc, SubStmt);
409   getCurFunction()->SwitchStack.back()->addSwitchCase(DS);
410   return Owned(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 Owned(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 Owned(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 Owned(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 = 0;
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.takeAs<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 Owned(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 = 0;
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.release();
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.take();
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.take();
652 
653   if (!CondVar) {
654     CondResult = ActOnFinishFullExpr(Cond, SwitchLoc);
655     if (CondResult.isInvalid())
656       return StmtError();
657     Cond = CondResult.take();
658   }
659 
660   getCurFunction()->setHasBranchIntoScope();
661 
662   SwitchStmt *SS = new (Context) SwitchStmt(Context, ConditionVar, Cond);
663   getCurFunction()->SwitchStack.push_back(SS);
664   return Owned(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   SS->setBody(BodyStmt, SwitchLoc);
706   getCurFunction()->SwitchStack.pop_back();
707 
708   Expr *CondExpr = SS->getCond();
709   if (!CondExpr) return StmtError();
710 
711   QualType CondType = CondExpr->getType();
712 
713   Expr *CondExprBeforePromotion = CondExpr;
714   QualType CondTypeBeforePromotion =
715       GetTypeBeforeIntegralPromotion(CondExprBeforePromotion);
716 
717   // C++ 6.4.2.p2:
718   // Integral promotions are performed (on the switch condition).
719   //
720   // A case value unrepresentable by the original switch condition
721   // type (before the promotion) doesn't make sense, even when it can
722   // be represented by the promoted type.  Therefore we need to find
723   // the pre-promotion type of the switch condition.
724   if (!CondExpr->isTypeDependent()) {
725     // We have already converted the expression to an integral or enumeration
726     // type, when we started the switch statement. If we don't have an
727     // appropriate type now, just return an error.
728     if (!CondType->isIntegralOrEnumerationType())
729       return StmtError();
730 
731     if (CondExpr->isKnownToHaveBooleanValue()) {
732       // switch(bool_expr) {...} is often a programmer error, e.g.
733       //   switch(n && mask) { ... }  // Doh - should be "n & mask".
734       // One can always use an if statement instead of switch(bool_expr).
735       Diag(SwitchLoc, diag::warn_bool_switch_condition)
736           << CondExpr->getSourceRange();
737     }
738   }
739 
740   // Get the bitwidth of the switched-on value before promotions.  We must
741   // convert the integer case values to this width before comparison.
742   bool HasDependentValue
743     = CondExpr->isTypeDependent() || CondExpr->isValueDependent();
744   unsigned CondWidth
745     = HasDependentValue ? 0 : Context.getIntWidth(CondTypeBeforePromotion);
746   bool CondIsSigned
747     = CondTypeBeforePromotion->isSignedIntegerOrEnumerationType();
748 
749   // Accumulate all of the case values in a vector so that we can sort them
750   // and detect duplicates.  This vector contains the APInt for the case after
751   // it has been converted to the condition type.
752   typedef SmallVector<std::pair<llvm::APSInt, CaseStmt*>, 64> CaseValsTy;
753   CaseValsTy CaseVals;
754 
755   // Keep track of any GNU case ranges we see.  The APSInt is the low value.
756   typedef std::vector<std::pair<llvm::APSInt, CaseStmt*> > CaseRangesTy;
757   CaseRangesTy CaseRanges;
758 
759   DefaultStmt *TheDefaultStmt = 0;
760 
761   bool CaseListIsErroneous = false;
762 
763   for (SwitchCase *SC = SS->getSwitchCaseList(); SC && !HasDependentValue;
764        SC = SC->getNextSwitchCase()) {
765 
766     if (DefaultStmt *DS = dyn_cast<DefaultStmt>(SC)) {
767       if (TheDefaultStmt) {
768         Diag(DS->getDefaultLoc(), diag::err_multiple_default_labels_defined);
769         Diag(TheDefaultStmt->getDefaultLoc(), diag::note_duplicate_case_prev);
770 
771         // FIXME: Remove the default statement from the switch block so that
772         // we'll return a valid AST.  This requires recursing down the AST and
773         // finding it, not something we are set up to do right now.  For now,
774         // just lop the entire switch stmt out of the AST.
775         CaseListIsErroneous = true;
776       }
777       TheDefaultStmt = DS;
778 
779     } else {
780       CaseStmt *CS = cast<CaseStmt>(SC);
781 
782       Expr *Lo = CS->getLHS();
783 
784       if (Lo->isTypeDependent() || Lo->isValueDependent()) {
785         HasDependentValue = true;
786         break;
787       }
788 
789       llvm::APSInt LoVal;
790 
791       if (getLangOpts().CPlusPlus11) {
792         // C++11 [stmt.switch]p2: the constant-expression shall be a converted
793         // constant expression of the promoted type of the switch condition.
794         ExprResult ConvLo =
795           CheckConvertedConstantExpression(Lo, CondType, LoVal, CCEK_CaseValue);
796         if (ConvLo.isInvalid()) {
797           CaseListIsErroneous = true;
798           continue;
799         }
800         Lo = ConvLo.take();
801       } else {
802         // We already verified that the expression has a i-c-e value (C99
803         // 6.8.4.2p3) - get that value now.
804         LoVal = Lo->EvaluateKnownConstInt(Context);
805 
806         // If the LHS is not the same type as the condition, insert an implicit
807         // cast.
808         Lo = DefaultLvalueConversion(Lo).take();
809         Lo = ImpCastExprToType(Lo, CondType, CK_IntegralCast).take();
810       }
811 
812       // Convert the value to the same width/sign as the condition had prior to
813       // integral promotions.
814       //
815       // FIXME: This causes us to reject valid code:
816       //   switch ((char)c) { case 256: case 0: return 0; }
817       // Here we claim there is a duplicated condition value, but there is not.
818       ConvertIntegerToTypeWarnOnOverflow(LoVal, CondWidth, CondIsSigned,
819                                          Lo->getLocStart(),
820                                          diag::warn_case_value_overflow);
821 
822       CS->setLHS(Lo);
823 
824       // If this is a case range, remember it in CaseRanges, otherwise CaseVals.
825       if (CS->getRHS()) {
826         if (CS->getRHS()->isTypeDependent() ||
827             CS->getRHS()->isValueDependent()) {
828           HasDependentValue = true;
829           break;
830         }
831         CaseRanges.push_back(std::make_pair(LoVal, CS));
832       } else
833         CaseVals.push_back(std::make_pair(LoVal, CS));
834     }
835   }
836 
837   if (!HasDependentValue) {
838     // If we don't have a default statement, check whether the
839     // condition is constant.
840     llvm::APSInt ConstantCondValue;
841     bool HasConstantCond = false;
842     if (!HasDependentValue && !TheDefaultStmt) {
843       HasConstantCond
844         = CondExprBeforePromotion->EvaluateAsInt(ConstantCondValue, Context,
845                                                  Expr::SE_AllowSideEffects);
846       assert(!HasConstantCond ||
847              (ConstantCondValue.getBitWidth() == CondWidth &&
848               ConstantCondValue.isSigned() == CondIsSigned));
849     }
850     bool ShouldCheckConstantCond = HasConstantCond;
851 
852     // Sort all the scalar case values so we can easily detect duplicates.
853     std::stable_sort(CaseVals.begin(), CaseVals.end(), CmpCaseVals);
854 
855     if (!CaseVals.empty()) {
856       for (unsigned i = 0, e = CaseVals.size(); i != e; ++i) {
857         if (ShouldCheckConstantCond &&
858             CaseVals[i].first == ConstantCondValue)
859           ShouldCheckConstantCond = false;
860 
861         if (i != 0 && CaseVals[i].first == CaseVals[i-1].first) {
862           // If we have a duplicate, report it.
863           // First, determine if either case value has a name
864           StringRef PrevString, CurrString;
865           Expr *PrevCase = CaseVals[i-1].second->getLHS()->IgnoreParenCasts();
866           Expr *CurrCase = CaseVals[i].second->getLHS()->IgnoreParenCasts();
867           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(PrevCase)) {
868             PrevString = DeclRef->getDecl()->getName();
869           }
870           if (DeclRefExpr *DeclRef = dyn_cast<DeclRefExpr>(CurrCase)) {
871             CurrString = DeclRef->getDecl()->getName();
872           }
873           SmallString<16> CaseValStr;
874           CaseVals[i-1].first.toString(CaseValStr);
875 
876           if (PrevString == CurrString)
877             Diag(CaseVals[i].second->getLHS()->getLocStart(),
878                  diag::err_duplicate_case) <<
879                  (PrevString.empty() ? CaseValStr.str() : PrevString);
880           else
881             Diag(CaseVals[i].second->getLHS()->getLocStart(),
882                  diag::err_duplicate_case_differing_expr) <<
883                  (PrevString.empty() ? CaseValStr.str() : PrevString) <<
884                  (CurrString.empty() ? CaseValStr.str() : CurrString) <<
885                  CaseValStr;
886 
887           Diag(CaseVals[i-1].second->getLHS()->getLocStart(),
888                diag::note_duplicate_case_prev);
889           // FIXME: We really want to remove the bogus case stmt from the
890           // substmt, but we have no way to do this right now.
891           CaseListIsErroneous = true;
892         }
893       }
894     }
895 
896     // Detect duplicate case ranges, which usually don't exist at all in
897     // the first place.
898     if (!CaseRanges.empty()) {
899       // Sort all the case ranges by their low value so we can easily detect
900       // overlaps between ranges.
901       std::stable_sort(CaseRanges.begin(), CaseRanges.end());
902 
903       // Scan the ranges, computing the high values and removing empty ranges.
904       std::vector<llvm::APSInt> HiVals;
905       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
906         llvm::APSInt &LoVal = CaseRanges[i].first;
907         CaseStmt *CR = CaseRanges[i].second;
908         Expr *Hi = CR->getRHS();
909         llvm::APSInt HiVal;
910 
911         if (getLangOpts().CPlusPlus11) {
912           // C++11 [stmt.switch]p2: the constant-expression shall be a converted
913           // constant expression of the promoted type of the switch condition.
914           ExprResult ConvHi =
915             CheckConvertedConstantExpression(Hi, CondType, HiVal,
916                                              CCEK_CaseValue);
917           if (ConvHi.isInvalid()) {
918             CaseListIsErroneous = true;
919             continue;
920           }
921           Hi = ConvHi.take();
922         } else {
923           HiVal = Hi->EvaluateKnownConstInt(Context);
924 
925           // If the RHS is not the same type as the condition, insert an
926           // implicit cast.
927           Hi = DefaultLvalueConversion(Hi).take();
928           Hi = ImpCastExprToType(Hi, CondType, CK_IntegralCast).take();
929         }
930 
931         // Convert the value to the same width/sign as the condition.
932         ConvertIntegerToTypeWarnOnOverflow(HiVal, CondWidth, CondIsSigned,
933                                            Hi->getLocStart(),
934                                            diag::warn_case_value_overflow);
935 
936         CR->setRHS(Hi);
937 
938         // If the low value is bigger than the high value, the case is empty.
939         if (LoVal > HiVal) {
940           Diag(CR->getLHS()->getLocStart(), diag::warn_case_empty_range)
941             << SourceRange(CR->getLHS()->getLocStart(),
942                            Hi->getLocEnd());
943           CaseRanges.erase(CaseRanges.begin()+i);
944           --i, --e;
945           continue;
946         }
947 
948         if (ShouldCheckConstantCond &&
949             LoVal <= ConstantCondValue &&
950             ConstantCondValue <= HiVal)
951           ShouldCheckConstantCond = false;
952 
953         HiVals.push_back(HiVal);
954       }
955 
956       // Rescan the ranges, looking for overlap with singleton values and other
957       // ranges.  Since the range list is sorted, we only need to compare case
958       // ranges with their neighbors.
959       for (unsigned i = 0, e = CaseRanges.size(); i != e; ++i) {
960         llvm::APSInt &CRLo = CaseRanges[i].first;
961         llvm::APSInt &CRHi = HiVals[i];
962         CaseStmt *CR = CaseRanges[i].second;
963 
964         // Check to see whether the case range overlaps with any
965         // singleton cases.
966         CaseStmt *OverlapStmt = 0;
967         llvm::APSInt OverlapVal(32);
968 
969         // Find the smallest value >= the lower bound.  If I is in the
970         // case range, then we have overlap.
971         CaseValsTy::iterator I = std::lower_bound(CaseVals.begin(),
972                                                   CaseVals.end(), CRLo,
973                                                   CaseCompareFunctor());
974         if (I != CaseVals.end() && I->first < CRHi) {
975           OverlapVal  = I->first;   // Found overlap with scalar.
976           OverlapStmt = I->second;
977         }
978 
979         // Find the smallest value bigger than the upper bound.
980         I = std::upper_bound(I, CaseVals.end(), CRHi, CaseCompareFunctor());
981         if (I != CaseVals.begin() && (I-1)->first >= CRLo) {
982           OverlapVal  = (I-1)->first;      // Found overlap with scalar.
983           OverlapStmt = (I-1)->second;
984         }
985 
986         // Check to see if this case stmt overlaps with the subsequent
987         // case range.
988         if (i && CRLo <= HiVals[i-1]) {
989           OverlapVal  = HiVals[i-1];       // Found overlap with range.
990           OverlapStmt = CaseRanges[i-1].second;
991         }
992 
993         if (OverlapStmt) {
994           // If we have a duplicate, report it.
995           Diag(CR->getLHS()->getLocStart(), diag::err_duplicate_case)
996             << OverlapVal.toString(10);
997           Diag(OverlapStmt->getLHS()->getLocStart(),
998                diag::note_duplicate_case_prev);
999           // FIXME: We really want to remove the bogus case stmt from the
1000           // substmt, but we have no way to do this right now.
1001           CaseListIsErroneous = true;
1002         }
1003       }
1004     }
1005 
1006     // Complain if we have a constant condition and we didn't find a match.
1007     if (!CaseListIsErroneous && ShouldCheckConstantCond) {
1008       // TODO: it would be nice if we printed enums as enums, chars as
1009       // chars, etc.
1010       Diag(CondExpr->getExprLoc(), diag::warn_missing_case_for_condition)
1011         << ConstantCondValue.toString(10)
1012         << CondExpr->getSourceRange();
1013     }
1014 
1015     // Check to see if switch is over an Enum and handles all of its
1016     // values.  We only issue a warning if there is not 'default:', but
1017     // we still do the analysis to preserve this information in the AST
1018     // (which can be used by flow-based analyes).
1019     //
1020     const EnumType *ET = CondTypeBeforePromotion->getAs<EnumType>();
1021 
1022     // If switch has default case, then ignore it.
1023     if (!CaseListIsErroneous  && !HasConstantCond && ET) {
1024       const EnumDecl *ED = ET->getDecl();
1025       typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl*>, 64>
1026         EnumValsTy;
1027       EnumValsTy EnumVals;
1028 
1029       // Gather all enum values, set their type and sort them,
1030       // allowing easier comparison with CaseVals.
1031       for (auto *EDI : ED->enumerators()) {
1032         llvm::APSInt Val = EDI->getInitVal();
1033         AdjustAPSInt(Val, CondWidth, CondIsSigned);
1034         EnumVals.push_back(std::make_pair(Val, EDI));
1035       }
1036       std::stable_sort(EnumVals.begin(), EnumVals.end(), CmpEnumVals);
1037       EnumValsTy::iterator EIend =
1038         std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
1039 
1040       // See which case values aren't in enum.
1041       EnumValsTy::const_iterator EI = EnumVals.begin();
1042       for (CaseValsTy::const_iterator CI = CaseVals.begin();
1043            CI != CaseVals.end(); CI++) {
1044         while (EI != EIend && EI->first < CI->first)
1045           EI++;
1046         if (EI == EIend || EI->first > CI->first) {
1047           Expr *CaseExpr = CI->second->getLHS();
1048           if (ShouldDiagnoseSwitchCaseNotInEnum(Context, ED, CaseExpr))
1049             Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
1050               << CondTypeBeforePromotion;
1051         }
1052       }
1053       // See which of case ranges aren't in enum
1054       EI = EnumVals.begin();
1055       for (CaseRangesTy::const_iterator RI = CaseRanges.begin();
1056            RI != CaseRanges.end() && EI != EIend; RI++) {
1057         while (EI != EIend && EI->first < RI->first)
1058           EI++;
1059 
1060         if (EI == EIend || EI->first != RI->first) {
1061           Expr *CaseExpr = RI->second->getLHS();
1062           if (ShouldDiagnoseSwitchCaseNotInEnum(Context, ED, CaseExpr))
1063             Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
1064               << CondTypeBeforePromotion;
1065         }
1066 
1067         llvm::APSInt Hi =
1068           RI->second->getRHS()->EvaluateKnownConstInt(Context);
1069         AdjustAPSInt(Hi, CondWidth, CondIsSigned);
1070         while (EI != EIend && EI->first < Hi)
1071           EI++;
1072         if (EI == EIend || EI->first != Hi) {
1073           Expr *CaseExpr = RI->second->getRHS();
1074           if (ShouldDiagnoseSwitchCaseNotInEnum(Context, ED, CaseExpr))
1075             Diag(CaseExpr->getExprLoc(), diag::warn_not_in_enum)
1076               << CondTypeBeforePromotion;
1077         }
1078       }
1079 
1080       // Check which enum vals aren't in switch
1081       CaseValsTy::const_iterator CI = CaseVals.begin();
1082       CaseRangesTy::const_iterator RI = CaseRanges.begin();
1083       bool hasCasesNotInSwitch = false;
1084 
1085       SmallVector<DeclarationName,8> UnhandledNames;
1086 
1087       for (EI = EnumVals.begin(); EI != EIend; EI++){
1088         // Drop unneeded case values
1089         while (CI != CaseVals.end() && CI->first < EI->first)
1090           CI++;
1091 
1092         if (CI != CaseVals.end() && CI->first == EI->first)
1093           continue;
1094 
1095         // Drop unneeded case ranges
1096         for (; RI != CaseRanges.end(); RI++) {
1097           llvm::APSInt Hi =
1098             RI->second->getRHS()->EvaluateKnownConstInt(Context);
1099           AdjustAPSInt(Hi, CondWidth, CondIsSigned);
1100           if (EI->first <= Hi)
1101             break;
1102         }
1103 
1104         if (RI == CaseRanges.end() || EI->first < RI->first) {
1105           hasCasesNotInSwitch = true;
1106           UnhandledNames.push_back(EI->second->getDeclName());
1107         }
1108       }
1109 
1110       if (TheDefaultStmt && UnhandledNames.empty())
1111         Diag(TheDefaultStmt->getDefaultLoc(), diag::warn_unreachable_default);
1112 
1113       // Produce a nice diagnostic if multiple values aren't handled.
1114       switch (UnhandledNames.size()) {
1115       case 0: break;
1116       case 1:
1117         Diag(CondExpr->getExprLoc(), TheDefaultStmt
1118           ? diag::warn_def_missing_case1 : diag::warn_missing_case1)
1119           << UnhandledNames[0];
1120         break;
1121       case 2:
1122         Diag(CondExpr->getExprLoc(), TheDefaultStmt
1123           ? diag::warn_def_missing_case2 : diag::warn_missing_case2)
1124           << UnhandledNames[0] << UnhandledNames[1];
1125         break;
1126       case 3:
1127         Diag(CondExpr->getExprLoc(), TheDefaultStmt
1128           ? diag::warn_def_missing_case3 : diag::warn_missing_case3)
1129           << UnhandledNames[0] << UnhandledNames[1] << UnhandledNames[2];
1130         break;
1131       default:
1132         Diag(CondExpr->getExprLoc(), TheDefaultStmt
1133           ? diag::warn_def_missing_cases : diag::warn_missing_cases)
1134           << (unsigned)UnhandledNames.size()
1135           << UnhandledNames[0] << UnhandledNames[1] << UnhandledNames[2];
1136         break;
1137       }
1138 
1139       if (!hasCasesNotInSwitch)
1140         SS->setAllEnumCasesCovered();
1141     }
1142   }
1143 
1144   DiagnoseEmptyStmtBody(CondExpr->getLocEnd(), BodyStmt,
1145                         diag::warn_empty_switch_body);
1146 
1147   // FIXME: If the case list was broken is some way, we don't have a good system
1148   // to patch it up.  Instead, just return the whole substmt as broken.
1149   if (CaseListIsErroneous)
1150     return StmtError();
1151 
1152   return Owned(SS);
1153 }
1154 
1155 void
1156 Sema::DiagnoseAssignmentEnum(QualType DstType, QualType SrcType,
1157                              Expr *SrcExpr) {
1158   if (Diags.getDiagnosticLevel(diag::warn_not_in_enum_assignment,
1159                                SrcExpr->getExprLoc()) ==
1160       DiagnosticsEngine::Ignored)
1161     return;
1162 
1163   if (const EnumType *ET = DstType->getAs<EnumType>())
1164     if (!Context.hasSameUnqualifiedType(SrcType, DstType) &&
1165         SrcType->isIntegerType()) {
1166       if (!SrcExpr->isTypeDependent() && !SrcExpr->isValueDependent() &&
1167           SrcExpr->isIntegerConstantExpr(Context)) {
1168         // Get the bitwidth of the enum value before promotions.
1169         unsigned DstWidth = Context.getIntWidth(DstType);
1170         bool DstIsSigned = DstType->isSignedIntegerOrEnumerationType();
1171 
1172         llvm::APSInt RhsVal = SrcExpr->EvaluateKnownConstInt(Context);
1173         AdjustAPSInt(RhsVal, DstWidth, DstIsSigned);
1174         const EnumDecl *ED = ET->getDecl();
1175         typedef SmallVector<std::pair<llvm::APSInt, EnumConstantDecl *>, 64>
1176             EnumValsTy;
1177         EnumValsTy EnumVals;
1178 
1179         // Gather all enum values, set their type and sort them,
1180         // allowing easier comparison with rhs constant.
1181         for (auto *EDI : ED->enumerators()) {
1182           llvm::APSInt Val = EDI->getInitVal();
1183           AdjustAPSInt(Val, DstWidth, DstIsSigned);
1184           EnumVals.push_back(std::make_pair(Val, EDI));
1185         }
1186         if (EnumVals.empty())
1187           return;
1188         std::stable_sort(EnumVals.begin(), EnumVals.end(), CmpEnumVals);
1189         EnumValsTy::iterator EIend =
1190             std::unique(EnumVals.begin(), EnumVals.end(), EqEnumVals);
1191 
1192         // See which values aren't in the enum.
1193         EnumValsTy::const_iterator EI = EnumVals.begin();
1194         while (EI != EIend && EI->first < RhsVal)
1195           EI++;
1196         if (EI == EIend || EI->first != RhsVal) {
1197           Diag(SrcExpr->getExprLoc(), diag::warn_not_in_enum_assignment)
1198               << DstType.getUnqualifiedType();
1199         }
1200       }
1201     }
1202 }
1203 
1204 StmtResult
1205 Sema::ActOnWhileStmt(SourceLocation WhileLoc, FullExprArg Cond,
1206                      Decl *CondVar, Stmt *Body) {
1207   ExprResult CondResult(Cond.release());
1208 
1209   VarDecl *ConditionVar = 0;
1210   if (CondVar) {
1211     ConditionVar = cast<VarDecl>(CondVar);
1212     CondResult = CheckConditionVariable(ConditionVar, WhileLoc, true);
1213     if (CondResult.isInvalid())
1214       return StmtError();
1215   }
1216   Expr *ConditionExpr = CondResult.take();
1217   if (!ConditionExpr)
1218     return StmtError();
1219   CheckBreakContinueBinding(ConditionExpr);
1220 
1221   DiagnoseUnusedExprResult(Body);
1222 
1223   if (isa<NullStmt>(Body))
1224     getCurCompoundScope().setHasEmptyLoopBodies();
1225 
1226   return Owned(new (Context) WhileStmt(Context, ConditionVar, ConditionExpr,
1227                                        Body, WhileLoc));
1228 }
1229 
1230 StmtResult
1231 Sema::ActOnDoStmt(SourceLocation DoLoc, Stmt *Body,
1232                   SourceLocation WhileLoc, SourceLocation CondLParen,
1233                   Expr *Cond, SourceLocation CondRParen) {
1234   assert(Cond && "ActOnDoStmt(): missing expression");
1235 
1236   CheckBreakContinueBinding(Cond);
1237   ExprResult CondResult = CheckBooleanCondition(Cond, DoLoc);
1238   if (CondResult.isInvalid())
1239     return StmtError();
1240   Cond = CondResult.take();
1241 
1242   CondResult = ActOnFinishFullExpr(Cond, DoLoc);
1243   if (CondResult.isInvalid())
1244     return StmtError();
1245   Cond = CondResult.take();
1246 
1247   DiagnoseUnusedExprResult(Body);
1248 
1249   return Owned(new (Context) DoStmt(Body, Cond, DoLoc, WhileLoc, CondRParen));
1250 }
1251 
1252 namespace {
1253   // This visitor will traverse a conditional statement and store all
1254   // the evaluated decls into a vector.  Simple is set to true if none
1255   // of the excluded constructs are used.
1256   class DeclExtractor : public EvaluatedExprVisitor<DeclExtractor> {
1257     llvm::SmallPtrSet<VarDecl*, 8> &Decls;
1258     SmallVectorImpl<SourceRange> &Ranges;
1259     bool Simple;
1260   public:
1261     typedef EvaluatedExprVisitor<DeclExtractor> Inherited;
1262 
1263     DeclExtractor(Sema &S, llvm::SmallPtrSet<VarDecl*, 8> &Decls,
1264                   SmallVectorImpl<SourceRange> &Ranges) :
1265         Inherited(S.Context),
1266         Decls(Decls),
1267         Ranges(Ranges),
1268         Simple(true) {}
1269 
1270     bool isSimple() { return Simple; }
1271 
1272     // Replaces the method in EvaluatedExprVisitor.
1273     void VisitMemberExpr(MemberExpr* E) {
1274       Simple = false;
1275     }
1276 
1277     // Any Stmt not whitelisted will cause the condition to be marked complex.
1278     void VisitStmt(Stmt *S) {
1279       Simple = false;
1280     }
1281 
1282     void VisitBinaryOperator(BinaryOperator *E) {
1283       Visit(E->getLHS());
1284       Visit(E->getRHS());
1285     }
1286 
1287     void VisitCastExpr(CastExpr *E) {
1288       Visit(E->getSubExpr());
1289     }
1290 
1291     void VisitUnaryOperator(UnaryOperator *E) {
1292       // Skip checking conditionals with derefernces.
1293       if (E->getOpcode() == UO_Deref)
1294         Simple = false;
1295       else
1296         Visit(E->getSubExpr());
1297     }
1298 
1299     void VisitConditionalOperator(ConditionalOperator *E) {
1300       Visit(E->getCond());
1301       Visit(E->getTrueExpr());
1302       Visit(E->getFalseExpr());
1303     }
1304 
1305     void VisitParenExpr(ParenExpr *E) {
1306       Visit(E->getSubExpr());
1307     }
1308 
1309     void VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
1310       Visit(E->getOpaqueValue()->getSourceExpr());
1311       Visit(E->getFalseExpr());
1312     }
1313 
1314     void VisitIntegerLiteral(IntegerLiteral *E) { }
1315     void VisitFloatingLiteral(FloatingLiteral *E) { }
1316     void VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) { }
1317     void VisitCharacterLiteral(CharacterLiteral *E) { }
1318     void VisitGNUNullExpr(GNUNullExpr *E) { }
1319     void VisitImaginaryLiteral(ImaginaryLiteral *E) { }
1320 
1321     void VisitDeclRefExpr(DeclRefExpr *E) {
1322       VarDecl *VD = dyn_cast<VarDecl>(E->getDecl());
1323       if (!VD) return;
1324 
1325       Ranges.push_back(E->getSourceRange());
1326 
1327       Decls.insert(VD);
1328     }
1329 
1330   }; // end class DeclExtractor
1331 
1332   // DeclMatcher checks to see if the decls are used in a non-evauluated
1333   // context.
1334   class DeclMatcher : public EvaluatedExprVisitor<DeclMatcher> {
1335     llvm::SmallPtrSet<VarDecl*, 8> &Decls;
1336     bool FoundDecl;
1337 
1338   public:
1339     typedef EvaluatedExprVisitor<DeclMatcher> Inherited;
1340 
1341     DeclMatcher(Sema &S, llvm::SmallPtrSet<VarDecl*, 8> &Decls,
1342                 Stmt *Statement) :
1343         Inherited(S.Context), Decls(Decls), FoundDecl(false) {
1344       if (!Statement) return;
1345 
1346       Visit(Statement);
1347     }
1348 
1349     void VisitReturnStmt(ReturnStmt *S) {
1350       FoundDecl = true;
1351     }
1352 
1353     void VisitBreakStmt(BreakStmt *S) {
1354       FoundDecl = true;
1355     }
1356 
1357     void VisitGotoStmt(GotoStmt *S) {
1358       FoundDecl = true;
1359     }
1360 
1361     void VisitCastExpr(CastExpr *E) {
1362       if (E->getCastKind() == CK_LValueToRValue)
1363         CheckLValueToRValueCast(E->getSubExpr());
1364       else
1365         Visit(E->getSubExpr());
1366     }
1367 
1368     void CheckLValueToRValueCast(Expr *E) {
1369       E = E->IgnoreParenImpCasts();
1370 
1371       if (isa<DeclRefExpr>(E)) {
1372         return;
1373       }
1374 
1375       if (ConditionalOperator *CO = dyn_cast<ConditionalOperator>(E)) {
1376         Visit(CO->getCond());
1377         CheckLValueToRValueCast(CO->getTrueExpr());
1378         CheckLValueToRValueCast(CO->getFalseExpr());
1379         return;
1380       }
1381 
1382       if (BinaryConditionalOperator *BCO =
1383               dyn_cast<BinaryConditionalOperator>(E)) {
1384         CheckLValueToRValueCast(BCO->getOpaqueValue()->getSourceExpr());
1385         CheckLValueToRValueCast(BCO->getFalseExpr());
1386         return;
1387       }
1388 
1389       Visit(E);
1390     }
1391 
1392     void VisitDeclRefExpr(DeclRefExpr *E) {
1393       if (VarDecl *VD = dyn_cast<VarDecl>(E->getDecl()))
1394         if (Decls.count(VD))
1395           FoundDecl = true;
1396     }
1397 
1398     bool FoundDeclInUse() { return FoundDecl; }
1399 
1400   };  // end class DeclMatcher
1401 
1402   void CheckForLoopConditionalStatement(Sema &S, Expr *Second,
1403                                         Expr *Third, Stmt *Body) {
1404     // Condition is empty
1405     if (!Second) return;
1406 
1407     if (S.Diags.getDiagnosticLevel(diag::warn_variables_not_in_loop_body,
1408                                    Second->getLocStart())
1409         == DiagnosticsEngine::Ignored)
1410       return;
1411 
1412     PartialDiagnostic PDiag = S.PDiag(diag::warn_variables_not_in_loop_body);
1413     llvm::SmallPtrSet<VarDecl*, 8> Decls;
1414     SmallVector<SourceRange, 10> Ranges;
1415     DeclExtractor DE(S, Decls, Ranges);
1416     DE.Visit(Second);
1417 
1418     // Don't analyze complex conditionals.
1419     if (!DE.isSimple()) return;
1420 
1421     // No decls found.
1422     if (Decls.size() == 0) return;
1423 
1424     // Don't warn on volatile, static, or global variables.
1425     for (llvm::SmallPtrSet<VarDecl*, 8>::iterator I = Decls.begin(),
1426                                                   E = Decls.end();
1427          I != E; ++I)
1428       if ((*I)->getType().isVolatileQualified() ||
1429           (*I)->hasGlobalStorage()) return;
1430 
1431     if (DeclMatcher(S, Decls, Second).FoundDeclInUse() ||
1432         DeclMatcher(S, Decls, Third).FoundDeclInUse() ||
1433         DeclMatcher(S, Decls, Body).FoundDeclInUse())
1434       return;
1435 
1436     // Load decl names into diagnostic.
1437     if (Decls.size() > 4)
1438       PDiag << 0;
1439     else {
1440       PDiag << Decls.size();
1441       for (llvm::SmallPtrSet<VarDecl*, 8>::iterator I = Decls.begin(),
1442                                                     E = Decls.end();
1443            I != E; ++I)
1444         PDiag << (*I)->getDeclName();
1445     }
1446 
1447     // Load SourceRanges into diagnostic if there is room.
1448     // Otherwise, load the SourceRange of the conditional expression.
1449     if (Ranges.size() <= PartialDiagnostic::MaxArguments)
1450       for (SmallVectorImpl<SourceRange>::iterator I = Ranges.begin(),
1451                                                   E = Ranges.end();
1452            I != E; ++I)
1453         PDiag << *I;
1454     else
1455       PDiag << Second->getSourceRange();
1456 
1457     S.Diag(Ranges.begin()->getBegin(), PDiag);
1458   }
1459 
1460   // If Statement is an incemement or decrement, return true and sets the
1461   // variables Increment and DRE.
1462   bool ProcessIterationStmt(Sema &S, Stmt* Statement, bool &Increment,
1463                             DeclRefExpr *&DRE) {
1464     if (UnaryOperator *UO = dyn_cast<UnaryOperator>(Statement)) {
1465       switch (UO->getOpcode()) {
1466         default: return false;
1467         case UO_PostInc:
1468         case UO_PreInc:
1469           Increment = true;
1470           break;
1471         case UO_PostDec:
1472         case UO_PreDec:
1473           Increment = false;
1474           break;
1475       }
1476       DRE = dyn_cast<DeclRefExpr>(UO->getSubExpr());
1477       return DRE;
1478     }
1479 
1480     if (CXXOperatorCallExpr *Call = dyn_cast<CXXOperatorCallExpr>(Statement)) {
1481       FunctionDecl *FD = Call->getDirectCallee();
1482       if (!FD || !FD->isOverloadedOperator()) return false;
1483       switch (FD->getOverloadedOperator()) {
1484         default: return false;
1485         case OO_PlusPlus:
1486           Increment = true;
1487           break;
1488         case OO_MinusMinus:
1489           Increment = false;
1490           break;
1491       }
1492       DRE = dyn_cast<DeclRefExpr>(Call->getArg(0));
1493       return DRE;
1494     }
1495 
1496     return false;
1497   }
1498 
1499   // A visitor to determine if a continue or break statement is a
1500   // subexpression.
1501   class BreakContinueFinder : public EvaluatedExprVisitor<BreakContinueFinder> {
1502     SourceLocation BreakLoc;
1503     SourceLocation ContinueLoc;
1504   public:
1505     BreakContinueFinder(Sema &S, Stmt* Body) :
1506         Inherited(S.Context) {
1507       Visit(Body);
1508     }
1509 
1510     typedef EvaluatedExprVisitor<BreakContinueFinder> Inherited;
1511 
1512     void VisitContinueStmt(ContinueStmt* E) {
1513       ContinueLoc = E->getContinueLoc();
1514     }
1515 
1516     void VisitBreakStmt(BreakStmt* E) {
1517       BreakLoc = E->getBreakLoc();
1518     }
1519 
1520     bool ContinueFound() { return ContinueLoc.isValid(); }
1521     bool BreakFound() { return BreakLoc.isValid(); }
1522     SourceLocation GetContinueLoc() { return ContinueLoc; }
1523     SourceLocation GetBreakLoc() { return BreakLoc; }
1524 
1525   };  // end class BreakContinueFinder
1526 
1527   // Emit a warning when a loop increment/decrement appears twice per loop
1528   // iteration.  The conditions which trigger this warning are:
1529   // 1) The last statement in the loop body and the third expression in the
1530   //    for loop are both increment or both decrement of the same variable
1531   // 2) No continue statements in the loop body.
1532   void CheckForRedundantIteration(Sema &S, Expr *Third, Stmt *Body) {
1533     // Return when there is nothing to check.
1534     if (!Body || !Third) return;
1535 
1536     if (S.Diags.getDiagnosticLevel(diag::warn_redundant_loop_iteration,
1537                                    Third->getLocStart())
1538         == DiagnosticsEngine::Ignored)
1539       return;
1540 
1541     // Get the last statement from the loop body.
1542     CompoundStmt *CS = dyn_cast<CompoundStmt>(Body);
1543     if (!CS || CS->body_empty()) return;
1544     Stmt *LastStmt = CS->body_back();
1545     if (!LastStmt) return;
1546 
1547     bool LoopIncrement, LastIncrement;
1548     DeclRefExpr *LoopDRE, *LastDRE;
1549 
1550     if (!ProcessIterationStmt(S, Third, LoopIncrement, LoopDRE)) return;
1551     if (!ProcessIterationStmt(S, LastStmt, LastIncrement, LastDRE)) return;
1552 
1553     // Check that the two statements are both increments or both decrements
1554     // on the same variable.
1555     if (LoopIncrement != LastIncrement ||
1556         LoopDRE->getDecl() != LastDRE->getDecl()) return;
1557 
1558     if (BreakContinueFinder(S, Body).ContinueFound()) return;
1559 
1560     S.Diag(LastDRE->getLocation(), diag::warn_redundant_loop_iteration)
1561          << LastDRE->getDecl() << LastIncrement;
1562     S.Diag(LoopDRE->getLocation(), diag::note_loop_iteration_here)
1563          << LoopIncrement;
1564   }
1565 
1566 } // end namespace
1567 
1568 
1569 void Sema::CheckBreakContinueBinding(Expr *E) {
1570   if (!E || getLangOpts().CPlusPlus)
1571     return;
1572   BreakContinueFinder BCFinder(*this, E);
1573   Scope *BreakParent = CurScope->getBreakParent();
1574   if (BCFinder.BreakFound() && BreakParent) {
1575     if (BreakParent->getFlags() & Scope::SwitchScope) {
1576       Diag(BCFinder.GetBreakLoc(), diag::warn_break_binds_to_switch);
1577     } else {
1578       Diag(BCFinder.GetBreakLoc(), diag::warn_loop_ctrl_binds_to_inner)
1579           << "break";
1580     }
1581   } else if (BCFinder.ContinueFound() && CurScope->getContinueParent()) {
1582     Diag(BCFinder.GetContinueLoc(), diag::warn_loop_ctrl_binds_to_inner)
1583         << "continue";
1584   }
1585 }
1586 
1587 StmtResult
1588 Sema::ActOnForStmt(SourceLocation ForLoc, SourceLocation LParenLoc,
1589                    Stmt *First, FullExprArg second, Decl *secondVar,
1590                    FullExprArg third,
1591                    SourceLocation RParenLoc, Stmt *Body) {
1592   if (!getLangOpts().CPlusPlus) {
1593     if (DeclStmt *DS = dyn_cast_or_null<DeclStmt>(First)) {
1594       // C99 6.8.5p3: The declaration part of a 'for' statement shall only
1595       // declare identifiers for objects having storage class 'auto' or
1596       // 'register'.
1597       for (auto *DI : DS->decls()) {
1598         VarDecl *VD = dyn_cast<VarDecl>(DI);
1599         if (VD && VD->isLocalVarDecl() && !VD->hasLocalStorage())
1600           VD = 0;
1601         if (VD == 0) {
1602           Diag(DI->getLocation(), diag::err_non_local_variable_decl_in_for);
1603           DI->setInvalidDecl();
1604         }
1605       }
1606     }
1607   }
1608 
1609   CheckBreakContinueBinding(second.get());
1610   CheckBreakContinueBinding(third.get());
1611 
1612   CheckForLoopConditionalStatement(*this, second.get(), third.get(), Body);
1613   CheckForRedundantIteration(*this, third.get(), Body);
1614 
1615   ExprResult SecondResult(second.release());
1616   VarDecl *ConditionVar = 0;
1617   if (secondVar) {
1618     ConditionVar = cast<VarDecl>(secondVar);
1619     SecondResult = CheckConditionVariable(ConditionVar, ForLoc, true);
1620     if (SecondResult.isInvalid())
1621       return StmtError();
1622   }
1623 
1624   Expr *Third  = third.release().takeAs<Expr>();
1625 
1626   DiagnoseUnusedExprResult(First);
1627   DiagnoseUnusedExprResult(Third);
1628   DiagnoseUnusedExprResult(Body);
1629 
1630   if (isa<NullStmt>(Body))
1631     getCurCompoundScope().setHasEmptyLoopBodies();
1632 
1633   return Owned(new (Context) ForStmt(Context, First,
1634                                      SecondResult.take(), ConditionVar,
1635                                      Third, Body, ForLoc, LParenLoc,
1636                                      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.take();
1649 
1650   ExprResult FullExpr = ActOnFinishFullExpr(E);
1651   if (FullExpr.isInvalid())
1652     return StmtError();
1653   return StmtResult(static_cast<Stmt*>(FullExpr.take()));
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 Owned(collection);
1663 
1664   // Perform normal l-value conversion.
1665   ExprResult result = DefaultFunctionArrayLvalueConversion(collection);
1666   if (result.isInvalid())
1667     return ExprError();
1668   collection = result.take();
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 = 0;
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 Owned(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.take());
1800   if (CollectionExprResult.isInvalid())
1801     return StmtError();
1802 
1803   return Owned(new (Context) ObjCForCollectionStmt(First,
1804                                                    CollectionExprResult.take(), 0,
1805                                                    ForLoc, RParenLoc));
1806 }
1807 
1808 /// Finish building a variable declaration for a for-range statement.
1809 /// \return true if an error occurs.
1810 static bool FinishForRangeVarDecl(Sema &SemaRef, VarDecl *Decl, Expr *Init,
1811                                   SourceLocation Loc, int DiagID) {
1812   // Deduce the type for the iterator variable now rather than leaving it to
1813   // AddInitializerToDecl, so we can produce a more suitable diagnostic.
1814   QualType InitType;
1815   if ((!isa<InitListExpr>(Init) && Init->getType()->isVoidType()) ||
1816       SemaRef.DeduceAutoType(Decl->getTypeSourceInfo(), Init, InitType) ==
1817           Sema::DAR_Failed)
1818     SemaRef.Diag(Loc, DiagID) << Init->getType();
1819   if (InitType.isNull()) {
1820     Decl->setInvalidDecl();
1821     return true;
1822   }
1823   Decl->setType(InitType);
1824 
1825   // In ARC, infer lifetime.
1826   // FIXME: ARC may want to turn this into 'const __unsafe_unretained' if
1827   // we're doing the equivalent of fast iteration.
1828   if (SemaRef.getLangOpts().ObjCAutoRefCount &&
1829       SemaRef.inferObjCARCLifetime(Decl))
1830     Decl->setInvalidDecl();
1831 
1832   SemaRef.AddInitializerToDecl(Decl, Init, /*DirectInit=*/false,
1833                                /*TypeMayContainAuto=*/false);
1834   SemaRef.FinalizeDeclaration(Decl);
1835   SemaRef.CurContext->addHiddenDecl(Decl);
1836   return false;
1837 }
1838 
1839 namespace {
1840 
1841 /// Produce a note indicating which begin/end function was implicitly called
1842 /// by a C++11 for-range statement. This is often not obvious from the code,
1843 /// nor from the diagnostics produced when analysing the implicit expressions
1844 /// required in a for-range statement.
1845 void NoteForRangeBeginEndFunction(Sema &SemaRef, Expr *E,
1846                                   Sema::BeginEndFunction BEF) {
1847   CallExpr *CE = dyn_cast<CallExpr>(E);
1848   if (!CE)
1849     return;
1850   FunctionDecl *D = dyn_cast<FunctionDecl>(CE->getCalleeDecl());
1851   if (!D)
1852     return;
1853   SourceLocation Loc = D->getLocation();
1854 
1855   std::string Description;
1856   bool IsTemplate = false;
1857   if (FunctionTemplateDecl *FunTmpl = D->getPrimaryTemplate()) {
1858     Description = SemaRef.getTemplateArgumentBindingsText(
1859       FunTmpl->getTemplateParameters(), *D->getTemplateSpecializationArgs());
1860     IsTemplate = true;
1861   }
1862 
1863   SemaRef.Diag(Loc, diag::note_for_range_begin_end)
1864     << BEF << IsTemplate << Description << E->getType();
1865 }
1866 
1867 /// Build a variable declaration for a for-range statement.
1868 VarDecl *BuildForRangeVarDecl(Sema &SemaRef, SourceLocation Loc,
1869                               QualType Type, const char *Name) {
1870   DeclContext *DC = SemaRef.CurContext;
1871   IdentifierInfo *II = &SemaRef.PP.getIdentifierTable().get(Name);
1872   TypeSourceInfo *TInfo = SemaRef.Context.getTrivialTypeSourceInfo(Type, Loc);
1873   VarDecl *Decl = VarDecl::Create(SemaRef.Context, DC, Loc, Loc, II, Type,
1874                                   TInfo, SC_None);
1875   Decl->setImplicit();
1876   return Decl;
1877 }
1878 
1879 }
1880 
1881 static bool ObjCEnumerationCollection(Expr *Collection) {
1882   return !Collection->isTypeDependent()
1883           && Collection->getType()->getAs<ObjCObjectPointerType>() != 0;
1884 }
1885 
1886 /// ActOnCXXForRangeStmt - Check and build a C++11 for-range statement.
1887 ///
1888 /// C++11 [stmt.ranged]:
1889 ///   A range-based for statement is equivalent to
1890 ///
1891 ///   {
1892 ///     auto && __range = range-init;
1893 ///     for ( auto __begin = begin-expr,
1894 ///           __end = end-expr;
1895 ///           __begin != __end;
1896 ///           ++__begin ) {
1897 ///       for-range-declaration = *__begin;
1898 ///       statement
1899 ///     }
1900 ///   }
1901 ///
1902 /// The body of the loop is not available yet, since it cannot be analysed until
1903 /// we have determined the type of the for-range-declaration.
1904 StmtResult
1905 Sema::ActOnCXXForRangeStmt(SourceLocation ForLoc,
1906                            Stmt *First, SourceLocation ColonLoc, Expr *Range,
1907                            SourceLocation RParenLoc, BuildForRangeKind Kind) {
1908   if (!First)
1909     return StmtError();
1910 
1911   if (Range && ObjCEnumerationCollection(Range))
1912     return ActOnObjCForCollectionStmt(ForLoc, First, Range, RParenLoc);
1913 
1914   DeclStmt *DS = dyn_cast<DeclStmt>(First);
1915   assert(DS && "first part of for range not a decl stmt");
1916 
1917   if (!DS->isSingleDecl()) {
1918     Diag(DS->getStartLoc(), diag::err_type_defined_in_for_range);
1919     return StmtError();
1920   }
1921 
1922   Decl *LoopVar = DS->getSingleDecl();
1923   if (LoopVar->isInvalidDecl() || !Range ||
1924       DiagnoseUnexpandedParameterPack(Range, UPPC_Expression)) {
1925     LoopVar->setInvalidDecl();
1926     return StmtError();
1927   }
1928 
1929   // Build  auto && __range = range-init
1930   SourceLocation RangeLoc = Range->getLocStart();
1931   VarDecl *RangeVar = BuildForRangeVarDecl(*this, RangeLoc,
1932                                            Context.getAutoRRefDeductType(),
1933                                            "__range");
1934   if (FinishForRangeVarDecl(*this, RangeVar, Range, RangeLoc,
1935                             diag::err_for_range_deduction_failure)) {
1936     LoopVar->setInvalidDecl();
1937     return StmtError();
1938   }
1939 
1940   // Claim the type doesn't contain auto: we've already done the checking.
1941   DeclGroupPtrTy RangeGroup =
1942       BuildDeclaratorGroup(llvm::MutableArrayRef<Decl *>((Decl **)&RangeVar, 1),
1943                            /*TypeMayContainAuto=*/ false);
1944   StmtResult RangeDecl = ActOnDeclStmt(RangeGroup, RangeLoc, RangeLoc);
1945   if (RangeDecl.isInvalid()) {
1946     LoopVar->setInvalidDecl();
1947     return StmtError();
1948   }
1949 
1950   return BuildCXXForRangeStmt(ForLoc, ColonLoc, RangeDecl.get(),
1951                               /*BeginEndDecl=*/0, /*Cond=*/0, /*Inc=*/0, DS,
1952                               RParenLoc, Kind);
1953 }
1954 
1955 /// \brief Create the initialization, compare, and increment steps for
1956 /// the range-based for loop expression.
1957 /// This function does not handle array-based for loops,
1958 /// which are created in Sema::BuildCXXForRangeStmt.
1959 ///
1960 /// \returns a ForRangeStatus indicating success or what kind of error occurred.
1961 /// BeginExpr and EndExpr are set and FRS_Success is returned on success;
1962 /// CandidateSet and BEF are set and some non-success value is returned on
1963 /// failure.
1964 static Sema::ForRangeStatus BuildNonArrayForRange(Sema &SemaRef, Scope *S,
1965                                             Expr *BeginRange, Expr *EndRange,
1966                                             QualType RangeType,
1967                                             VarDecl *BeginVar,
1968                                             VarDecl *EndVar,
1969                                             SourceLocation ColonLoc,
1970                                             OverloadCandidateSet *CandidateSet,
1971                                             ExprResult *BeginExpr,
1972                                             ExprResult *EndExpr,
1973                                             Sema::BeginEndFunction *BEF) {
1974   DeclarationNameInfo BeginNameInfo(
1975       &SemaRef.PP.getIdentifierTable().get("begin"), ColonLoc);
1976   DeclarationNameInfo EndNameInfo(&SemaRef.PP.getIdentifierTable().get("end"),
1977                                   ColonLoc);
1978 
1979   LookupResult BeginMemberLookup(SemaRef, BeginNameInfo,
1980                                  Sema::LookupMemberName);
1981   LookupResult EndMemberLookup(SemaRef, EndNameInfo, Sema::LookupMemberName);
1982 
1983   if (CXXRecordDecl *D = RangeType->getAsCXXRecordDecl()) {
1984     // - if _RangeT is a class type, the unqualified-ids begin and end are
1985     //   looked up in the scope of class _RangeT as if by class member access
1986     //   lookup (3.4.5), and if either (or both) finds at least one
1987     //   declaration, begin-expr and end-expr are __range.begin() and
1988     //   __range.end(), respectively;
1989     SemaRef.LookupQualifiedName(BeginMemberLookup, D);
1990     SemaRef.LookupQualifiedName(EndMemberLookup, D);
1991 
1992     if (BeginMemberLookup.empty() != EndMemberLookup.empty()) {
1993       SourceLocation RangeLoc = BeginVar->getLocation();
1994       *BEF = BeginMemberLookup.empty() ? Sema::BEF_end : Sema::BEF_begin;
1995 
1996       SemaRef.Diag(RangeLoc, diag::err_for_range_member_begin_end_mismatch)
1997           << RangeLoc << BeginRange->getType() << *BEF;
1998       return Sema::FRS_DiagnosticIssued;
1999     }
2000   } else {
2001     // - otherwise, begin-expr and end-expr are begin(__range) and
2002     //   end(__range), respectively, where begin and end are looked up with
2003     //   argument-dependent lookup (3.4.2). For the purposes of this name
2004     //   lookup, namespace std is an associated namespace.
2005 
2006   }
2007 
2008   *BEF = Sema::BEF_begin;
2009   Sema::ForRangeStatus RangeStatus =
2010       SemaRef.BuildForRangeBeginEndCall(S, ColonLoc, ColonLoc, BeginVar,
2011                                         Sema::BEF_begin, BeginNameInfo,
2012                                         BeginMemberLookup, CandidateSet,
2013                                         BeginRange, BeginExpr);
2014 
2015   if (RangeStatus != Sema::FRS_Success)
2016     return RangeStatus;
2017   if (FinishForRangeVarDecl(SemaRef, BeginVar, BeginExpr->get(), ColonLoc,
2018                             diag::err_for_range_iter_deduction_failure)) {
2019     NoteForRangeBeginEndFunction(SemaRef, BeginExpr->get(), *BEF);
2020     return Sema::FRS_DiagnosticIssued;
2021   }
2022 
2023   *BEF = Sema::BEF_end;
2024   RangeStatus =
2025       SemaRef.BuildForRangeBeginEndCall(S, ColonLoc, ColonLoc, EndVar,
2026                                         Sema::BEF_end, EndNameInfo,
2027                                         EndMemberLookup, CandidateSet,
2028                                         EndRange, EndExpr);
2029   if (RangeStatus != Sema::FRS_Success)
2030     return RangeStatus;
2031   if (FinishForRangeVarDecl(SemaRef, EndVar, EndExpr->get(), ColonLoc,
2032                             diag::err_for_range_iter_deduction_failure)) {
2033     NoteForRangeBeginEndFunction(SemaRef, EndExpr->get(), *BEF);
2034     return Sema::FRS_DiagnosticIssued;
2035   }
2036   return Sema::FRS_Success;
2037 }
2038 
2039 /// Speculatively attempt to dereference an invalid range expression.
2040 /// If the attempt fails, this function will return a valid, null StmtResult
2041 /// and emit no diagnostics.
2042 static StmtResult RebuildForRangeWithDereference(Sema &SemaRef, Scope *S,
2043                                                  SourceLocation ForLoc,
2044                                                  Stmt *LoopVarDecl,
2045                                                  SourceLocation ColonLoc,
2046                                                  Expr *Range,
2047                                                  SourceLocation RangeLoc,
2048                                                  SourceLocation RParenLoc) {
2049   // Determine whether we can rebuild the for-range statement with a
2050   // dereferenced range expression.
2051   ExprResult AdjustedRange;
2052   {
2053     Sema::SFINAETrap Trap(SemaRef);
2054 
2055     AdjustedRange = SemaRef.BuildUnaryOp(S, RangeLoc, UO_Deref, Range);
2056     if (AdjustedRange.isInvalid())
2057       return StmtResult();
2058 
2059     StmtResult SR =
2060       SemaRef.ActOnCXXForRangeStmt(ForLoc, LoopVarDecl, ColonLoc,
2061                                    AdjustedRange.get(), RParenLoc,
2062                                    Sema::BFRK_Check);
2063     if (SR.isInvalid())
2064       return StmtResult();
2065   }
2066 
2067   // The attempt to dereference worked well enough that it could produce a valid
2068   // loop. Produce a fixit, and rebuild the loop with diagnostics enabled, in
2069   // case there are any other (non-fatal) problems with it.
2070   SemaRef.Diag(RangeLoc, diag::err_for_range_dereference)
2071     << Range->getType() << FixItHint::CreateInsertion(RangeLoc, "*");
2072   return SemaRef.ActOnCXXForRangeStmt(ForLoc, LoopVarDecl, ColonLoc,
2073                                       AdjustedRange.get(), RParenLoc,
2074                                       Sema::BFRK_Rebuild);
2075 }
2076 
2077 namespace {
2078 /// RAII object to automatically invalidate a declaration if an error occurs.
2079 struct InvalidateOnErrorScope {
2080   InvalidateOnErrorScope(Sema &SemaRef, Decl *D, bool Enabled)
2081       : Trap(SemaRef.Diags), D(D), Enabled(Enabled) {}
2082   ~InvalidateOnErrorScope() {
2083     if (Enabled && Trap.hasErrorOccurred())
2084       D->setInvalidDecl();
2085   }
2086 
2087   DiagnosticErrorTrap Trap;
2088   Decl *D;
2089   bool Enabled;
2090 };
2091 }
2092 
2093 /// BuildCXXForRangeStmt - Build or instantiate a C++11 for-range statement.
2094 StmtResult
2095 Sema::BuildCXXForRangeStmt(SourceLocation ForLoc, SourceLocation ColonLoc,
2096                            Stmt *RangeDecl, Stmt *BeginEnd, Expr *Cond,
2097                            Expr *Inc, Stmt *LoopVarDecl,
2098                            SourceLocation RParenLoc, BuildForRangeKind Kind) {
2099   Scope *S = getCurScope();
2100 
2101   DeclStmt *RangeDS = cast<DeclStmt>(RangeDecl);
2102   VarDecl *RangeVar = cast<VarDecl>(RangeDS->getSingleDecl());
2103   QualType RangeVarType = RangeVar->getType();
2104 
2105   DeclStmt *LoopVarDS = cast<DeclStmt>(LoopVarDecl);
2106   VarDecl *LoopVar = cast<VarDecl>(LoopVarDS->getSingleDecl());
2107 
2108   // If we hit any errors, mark the loop variable as invalid if its type
2109   // contains 'auto'.
2110   InvalidateOnErrorScope Invalidate(*this, LoopVar,
2111                                     LoopVar->getType()->isUndeducedType());
2112 
2113   StmtResult BeginEndDecl = BeginEnd;
2114   ExprResult NotEqExpr = Cond, IncrExpr = Inc;
2115 
2116   if (RangeVarType->isDependentType()) {
2117     // The range is implicitly used as a placeholder when it is dependent.
2118     RangeVar->markUsed(Context);
2119 
2120     // Deduce any 'auto's in the loop variable as 'DependentTy'. We'll fill
2121     // them in properly when we instantiate the loop.
2122     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check)
2123       LoopVar->setType(SubstAutoType(LoopVar->getType(), Context.DependentTy));
2124   } else if (!BeginEndDecl.get()) {
2125     SourceLocation RangeLoc = RangeVar->getLocation();
2126 
2127     const QualType RangeVarNonRefType = RangeVarType.getNonReferenceType();
2128 
2129     ExprResult BeginRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
2130                                                 VK_LValue, ColonLoc);
2131     if (BeginRangeRef.isInvalid())
2132       return StmtError();
2133 
2134     ExprResult EndRangeRef = BuildDeclRefExpr(RangeVar, RangeVarNonRefType,
2135                                               VK_LValue, ColonLoc);
2136     if (EndRangeRef.isInvalid())
2137       return StmtError();
2138 
2139     QualType AutoType = Context.getAutoDeductType();
2140     Expr *Range = RangeVar->getInit();
2141     if (!Range)
2142       return StmtError();
2143     QualType RangeType = Range->getType();
2144 
2145     if (RequireCompleteType(RangeLoc, RangeType,
2146                             diag::err_for_range_incomplete_type))
2147       return StmtError();
2148 
2149     // Build auto __begin = begin-expr, __end = end-expr.
2150     VarDecl *BeginVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
2151                                              "__begin");
2152     VarDecl *EndVar = BuildForRangeVarDecl(*this, ColonLoc, AutoType,
2153                                            "__end");
2154 
2155     // Build begin-expr and end-expr and attach to __begin and __end variables.
2156     ExprResult BeginExpr, EndExpr;
2157     if (const ArrayType *UnqAT = RangeType->getAsArrayTypeUnsafe()) {
2158       // - if _RangeT is an array type, begin-expr and end-expr are __range and
2159       //   __range + __bound, respectively, where __bound is the array bound. If
2160       //   _RangeT is an array of unknown size or an array of incomplete type,
2161       //   the program is ill-formed;
2162 
2163       // begin-expr is __range.
2164       BeginExpr = BeginRangeRef;
2165       if (FinishForRangeVarDecl(*this, BeginVar, BeginRangeRef.get(), ColonLoc,
2166                                 diag::err_for_range_iter_deduction_failure)) {
2167         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2168         return StmtError();
2169       }
2170 
2171       // Find the array bound.
2172       ExprResult BoundExpr;
2173       if (const ConstantArrayType *CAT = dyn_cast<ConstantArrayType>(UnqAT))
2174         BoundExpr = Owned(IntegerLiteral::Create(Context, CAT->getSize(),
2175                                                  Context.getPointerDiffType(),
2176                                                  RangeLoc));
2177       else if (const VariableArrayType *VAT =
2178                dyn_cast<VariableArrayType>(UnqAT))
2179         BoundExpr = VAT->getSizeExpr();
2180       else {
2181         // Can't be a DependentSizedArrayType or an IncompleteArrayType since
2182         // UnqAT is not incomplete and Range is not type-dependent.
2183         llvm_unreachable("Unexpected array type in for-range");
2184       }
2185 
2186       // end-expr is __range + __bound.
2187       EndExpr = ActOnBinOp(S, ColonLoc, tok::plus, EndRangeRef.get(),
2188                            BoundExpr.get());
2189       if (EndExpr.isInvalid())
2190         return StmtError();
2191       if (FinishForRangeVarDecl(*this, EndVar, EndExpr.get(), ColonLoc,
2192                                 diag::err_for_range_iter_deduction_failure)) {
2193         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2194         return StmtError();
2195       }
2196     } else {
2197       OverloadCandidateSet CandidateSet(RangeLoc);
2198       Sema::BeginEndFunction BEFFailure;
2199       ForRangeStatus RangeStatus =
2200           BuildNonArrayForRange(*this, S, BeginRangeRef.get(),
2201                                 EndRangeRef.get(), RangeType,
2202                                 BeginVar, EndVar, ColonLoc, &CandidateSet,
2203                                 &BeginExpr, &EndExpr, &BEFFailure);
2204 
2205       if (Kind == BFRK_Build && RangeStatus == FRS_NoViableFunction &&
2206           BEFFailure == BEF_begin) {
2207         // If the range is being built from an array parameter, emit a
2208         // a diagnostic that it is being treated as a pointer.
2209         if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Range)) {
2210           if (ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
2211             QualType ArrayTy = PVD->getOriginalType();
2212             QualType PointerTy = PVD->getType();
2213             if (PointerTy->isPointerType() && ArrayTy->isArrayType()) {
2214               Diag(Range->getLocStart(), diag::err_range_on_array_parameter)
2215                 << RangeLoc << PVD << ArrayTy << PointerTy;
2216               Diag(PVD->getLocation(), diag::note_declared_at);
2217               return StmtError();
2218             }
2219           }
2220         }
2221 
2222         // If building the range failed, try dereferencing the range expression
2223         // unless a diagnostic was issued or the end function is problematic.
2224         StmtResult SR = RebuildForRangeWithDereference(*this, S, ForLoc,
2225                                                        LoopVarDecl, ColonLoc,
2226                                                        Range, RangeLoc,
2227                                                        RParenLoc);
2228         if (SR.isInvalid() || SR.isUsable())
2229           return SR;
2230       }
2231 
2232       // Otherwise, emit diagnostics if we haven't already.
2233       if (RangeStatus == FRS_NoViableFunction) {
2234         Expr *Range = BEFFailure ? EndRangeRef.get() : BeginRangeRef.get();
2235         Diag(Range->getLocStart(), diag::err_for_range_invalid)
2236             << RangeLoc << Range->getType() << BEFFailure;
2237         CandidateSet.NoteCandidates(*this, OCD_AllCandidates, Range);
2238       }
2239       // Return an error if no fix was discovered.
2240       if (RangeStatus != FRS_Success)
2241         return StmtError();
2242     }
2243 
2244     assert(!BeginExpr.isInvalid() && !EndExpr.isInvalid() &&
2245            "invalid range expression in for loop");
2246 
2247     // C++11 [dcl.spec.auto]p7: BeginType and EndType must be the same.
2248     QualType BeginType = BeginVar->getType(), EndType = EndVar->getType();
2249     if (!Context.hasSameType(BeginType, EndType)) {
2250       Diag(RangeLoc, diag::err_for_range_begin_end_types_differ)
2251         << BeginType << EndType;
2252       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2253       NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2254     }
2255 
2256     Decl *BeginEndDecls[] = { BeginVar, EndVar };
2257     // Claim the type doesn't contain auto: we've already done the checking.
2258     DeclGroupPtrTy BeginEndGroup =
2259         BuildDeclaratorGroup(llvm::MutableArrayRef<Decl *>(BeginEndDecls, 2),
2260                              /*TypeMayContainAuto=*/ false);
2261     BeginEndDecl = ActOnDeclStmt(BeginEndGroup, ColonLoc, ColonLoc);
2262 
2263     const QualType BeginRefNonRefType = BeginType.getNonReferenceType();
2264     ExprResult BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2265                                            VK_LValue, ColonLoc);
2266     if (BeginRef.isInvalid())
2267       return StmtError();
2268 
2269     ExprResult EndRef = BuildDeclRefExpr(EndVar, EndType.getNonReferenceType(),
2270                                          VK_LValue, ColonLoc);
2271     if (EndRef.isInvalid())
2272       return StmtError();
2273 
2274     // Build and check __begin != __end expression.
2275     NotEqExpr = ActOnBinOp(S, ColonLoc, tok::exclaimequal,
2276                            BeginRef.get(), EndRef.get());
2277     NotEqExpr = ActOnBooleanCondition(S, ColonLoc, NotEqExpr.get());
2278     NotEqExpr = ActOnFinishFullExpr(NotEqExpr.get());
2279     if (NotEqExpr.isInvalid()) {
2280       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2281         << RangeLoc << 0 << BeginRangeRef.get()->getType();
2282       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2283       if (!Context.hasSameType(BeginType, EndType))
2284         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2285       return StmtError();
2286     }
2287 
2288     // Build and check ++__begin expression.
2289     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2290                                 VK_LValue, ColonLoc);
2291     if (BeginRef.isInvalid())
2292       return StmtError();
2293 
2294     IncrExpr = ActOnUnaryOp(S, ColonLoc, tok::plusplus, BeginRef.get());
2295     IncrExpr = ActOnFinishFullExpr(IncrExpr.get());
2296     if (IncrExpr.isInvalid()) {
2297       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2298         << RangeLoc << 2 << BeginRangeRef.get()->getType() ;
2299       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2300       return StmtError();
2301     }
2302 
2303     // Build and check *__begin  expression.
2304     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2305                                 VK_LValue, ColonLoc);
2306     if (BeginRef.isInvalid())
2307       return StmtError();
2308 
2309     ExprResult DerefExpr = ActOnUnaryOp(S, ColonLoc, tok::star, BeginRef.get());
2310     if (DerefExpr.isInvalid()) {
2311       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2312         << RangeLoc << 1 << BeginRangeRef.get()->getType();
2313       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2314       return StmtError();
2315     }
2316 
2317     // Attach  *__begin  as initializer for VD. Don't touch it if we're just
2318     // trying to determine whether this would be a valid range.
2319     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
2320       AddInitializerToDecl(LoopVar, DerefExpr.get(), /*DirectInit=*/false,
2321                            /*TypeMayContainAuto=*/true);
2322       if (LoopVar->isInvalidDecl())
2323         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2324     }
2325   }
2326 
2327   // Don't bother to actually allocate the result if we're just trying to
2328   // determine whether it would be valid.
2329   if (Kind == BFRK_Check)
2330     return StmtResult();
2331 
2332   return Owned(new (Context) CXXForRangeStmt(RangeDS,
2333                                      cast_or_null<DeclStmt>(BeginEndDecl.get()),
2334                                              NotEqExpr.take(), IncrExpr.take(),
2335                                              LoopVarDS, /*Body=*/0, ForLoc,
2336                                              ColonLoc, RParenLoc));
2337 }
2338 
2339 /// FinishObjCForCollectionStmt - Attach the body to a objective-C foreach
2340 /// statement.
2341 StmtResult Sema::FinishObjCForCollectionStmt(Stmt *S, Stmt *B) {
2342   if (!S || !B)
2343     return StmtError();
2344   ObjCForCollectionStmt * ForStmt = cast<ObjCForCollectionStmt>(S);
2345 
2346   ForStmt->setBody(B);
2347   return S;
2348 }
2349 
2350 /// FinishCXXForRangeStmt - Attach the body to a C++0x for-range statement.
2351 /// This is a separate step from ActOnCXXForRangeStmt because analysis of the
2352 /// body cannot be performed until after the type of the range variable is
2353 /// determined.
2354 StmtResult Sema::FinishCXXForRangeStmt(Stmt *S, Stmt *B) {
2355   if (!S || !B)
2356     return StmtError();
2357 
2358   if (isa<ObjCForCollectionStmt>(S))
2359     return FinishObjCForCollectionStmt(S, B);
2360 
2361   CXXForRangeStmt *ForStmt = cast<CXXForRangeStmt>(S);
2362   ForStmt->setBody(B);
2363 
2364   DiagnoseEmptyStmtBody(ForStmt->getRParenLoc(), B,
2365                         diag::warn_empty_range_based_for_body);
2366 
2367   return S;
2368 }
2369 
2370 StmtResult Sema::ActOnGotoStmt(SourceLocation GotoLoc,
2371                                SourceLocation LabelLoc,
2372                                LabelDecl *TheDecl) {
2373   getCurFunction()->setHasBranchIntoScope();
2374   TheDecl->markUsed(Context);
2375   return Owned(new (Context) GotoStmt(TheDecl, GotoLoc, LabelLoc));
2376 }
2377 
2378 StmtResult
2379 Sema::ActOnIndirectGotoStmt(SourceLocation GotoLoc, SourceLocation StarLoc,
2380                             Expr *E) {
2381   // Convert operand to void*
2382   if (!E->isTypeDependent()) {
2383     QualType ETy = E->getType();
2384     QualType DestTy = Context.getPointerType(Context.VoidTy.withConst());
2385     ExprResult ExprRes = Owned(E);
2386     AssignConvertType ConvTy =
2387       CheckSingleAssignmentConstraints(DestTy, ExprRes);
2388     if (ExprRes.isInvalid())
2389       return StmtError();
2390     E = ExprRes.take();
2391     if (DiagnoseAssignmentResult(ConvTy, StarLoc, DestTy, ETy, E, AA_Passing))
2392       return StmtError();
2393   }
2394 
2395   ExprResult ExprRes = ActOnFinishFullExpr(E);
2396   if (ExprRes.isInvalid())
2397     return StmtError();
2398   E = ExprRes.take();
2399 
2400   getCurFunction()->setHasIndirectGoto();
2401 
2402   return Owned(new (Context) IndirectGotoStmt(GotoLoc, StarLoc, E));
2403 }
2404 
2405 StmtResult
2406 Sema::ActOnContinueStmt(SourceLocation ContinueLoc, Scope *CurScope) {
2407   Scope *S = CurScope->getContinueParent();
2408   if (!S) {
2409     // C99 6.8.6.2p1: A break shall appear only in or as a loop body.
2410     return StmtError(Diag(ContinueLoc, diag::err_continue_not_in_loop));
2411   }
2412 
2413   return Owned(new (Context) ContinueStmt(ContinueLoc));
2414 }
2415 
2416 StmtResult
2417 Sema::ActOnBreakStmt(SourceLocation BreakLoc, Scope *CurScope) {
2418   Scope *S = CurScope->getBreakParent();
2419   if (!S) {
2420     // C99 6.8.6.3p1: A break shall appear only in or as a switch/loop body.
2421     return StmtError(Diag(BreakLoc, diag::err_break_not_in_loop_or_switch));
2422   }
2423 
2424   return Owned(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 const VarDecl *Sema::getCopyElisionCandidate(QualType ReturnType,
2446                                              Expr *E,
2447                                              bool AllowFunctionParameter) {
2448   QualType ExprType = E->getType();
2449   // - in a return statement in a function with ...
2450   // ... a class return type ...
2451   if (!ReturnType.isNull()) {
2452     if (!ReturnType->isRecordType())
2453       return 0;
2454     // ... the same cv-unqualified type as the function return type ...
2455     if (!Context.hasSameUnqualifiedType(ReturnType, ExprType))
2456       return 0;
2457   }
2458 
2459   // ... the expression is the name of a non-volatile automatic object
2460   // (other than a function or catch-clause parameter)) ...
2461   const DeclRefExpr *DR = dyn_cast<DeclRefExpr>(E->IgnoreParens());
2462   if (!DR || DR->refersToEnclosingLocal())
2463     return 0;
2464   const VarDecl *VD = dyn_cast<VarDecl>(DR->getDecl());
2465   if (!VD)
2466     return 0;
2467 
2468   // ...object (other than a function or catch-clause parameter)...
2469   if (VD->getKind() != Decl::Var &&
2470       !(AllowFunctionParameter && VD->getKind() == Decl::ParmVar))
2471     return 0;
2472   if (VD->isExceptionVariable()) return 0;
2473 
2474   // ...automatic...
2475   if (!VD->hasLocalStorage()) return 0;
2476 
2477   // ...non-volatile...
2478   if (VD->getType().isVolatileQualified()) return 0;
2479   if (VD->getType()->isReferenceType()) return 0;
2480 
2481   // __block variables can't be allocated in a way that permits NRVO.
2482   if (VD->hasAttr<BlocksAttr>()) return 0;
2483 
2484   // Variables with higher required alignment than their type's ABI
2485   // alignment cannot use NRVO.
2486   if (VD->hasAttr<AlignedAttr>() &&
2487       Context.getDeclAlign(VD) > Context.getTypeAlignInChars(VD->getType()))
2488     return 0;
2489 
2490   return VD;
2491 }
2492 
2493 /// \brief Perform the initialization of a potentially-movable value, which
2494 /// is the result of return value.
2495 ///
2496 /// This routine implements C++0x [class.copy]p33, which attempts to treat
2497 /// returned lvalues as rvalues in certain cases (to prefer move construction),
2498 /// then falls back to treating them as lvalues if that failed.
2499 ExprResult
2500 Sema::PerformMoveOrCopyInitialization(const InitializedEntity &Entity,
2501                                       const VarDecl *NRVOCandidate,
2502                                       QualType ResultType,
2503                                       Expr *Value,
2504                                       bool AllowNRVO) {
2505   // C++0x [class.copy]p33:
2506   //   When the criteria for elision of a copy operation are met or would
2507   //   be met save for the fact that the source object is a function
2508   //   parameter, and the object to be copied is designated by an lvalue,
2509   //   overload resolution to select the constructor for the copy is first
2510   //   performed as if the object were designated by an rvalue.
2511   ExprResult Res = ExprError();
2512   if (AllowNRVO &&
2513       (NRVOCandidate || getCopyElisionCandidate(ResultType, Value, true))) {
2514     ImplicitCastExpr AsRvalue(ImplicitCastExpr::OnStack,
2515                               Value->getType(), CK_NoOp, Value, VK_XValue);
2516 
2517     Expr *InitExpr = &AsRvalue;
2518     InitializationKind Kind
2519       = InitializationKind::CreateCopy(Value->getLocStart(),
2520                                        Value->getLocStart());
2521     InitializationSequence Seq(*this, Entity, Kind, InitExpr);
2522 
2523     //   [...] If overload resolution fails, or if the type of the first
2524     //   parameter of the selected constructor is not an rvalue reference
2525     //   to the object's type (possibly cv-qualified), overload resolution
2526     //   is performed again, considering the object as an lvalue.
2527     if (Seq) {
2528       for (InitializationSequence::step_iterator Step = Seq.step_begin(),
2529            StepEnd = Seq.step_end();
2530            Step != StepEnd; ++Step) {
2531         if (Step->Kind != InitializationSequence::SK_ConstructorInitialization)
2532           continue;
2533 
2534         CXXConstructorDecl *Constructor
2535         = cast<CXXConstructorDecl>(Step->Function.Function);
2536 
2537         const RValueReferenceType *RRefType
2538           = Constructor->getParamDecl(0)->getType()
2539                                                  ->getAs<RValueReferenceType>();
2540 
2541         // If we don't meet the criteria, break out now.
2542         if (!RRefType ||
2543             !Context.hasSameUnqualifiedType(RRefType->getPointeeType(),
2544                             Context.getTypeDeclType(Constructor->getParent())))
2545           break;
2546 
2547         // Promote "AsRvalue" to the heap, since we now need this
2548         // expression node to persist.
2549         Value = ImplicitCastExpr::Create(Context, Value->getType(),
2550                                          CK_NoOp, Value, 0, VK_XValue);
2551 
2552         // Complete type-checking the initialization of the return type
2553         // using the constructor we found.
2554         Res = Seq.Perform(*this, Entity, Kind, Value);
2555       }
2556     }
2557   }
2558 
2559   // Either we didn't meet the criteria for treating an lvalue as an rvalue,
2560   // above, or overload resolution failed. Either way, we need to try
2561   // (again) now with the return value expression as written.
2562   if (Res.isInvalid())
2563     Res = PerformCopyInitialization(Entity, SourceLocation(), Value);
2564 
2565   return Res;
2566 }
2567 
2568 /// \brief Determine whether the declared return type of the specified function
2569 /// contains 'auto'.
2570 static bool hasDeducedReturnType(FunctionDecl *FD) {
2571   const FunctionProtoType *FPT =
2572       FD->getTypeSourceInfo()->getType()->castAs<FunctionProtoType>();
2573   return FPT->getReturnType()->isUndeducedType();
2574 }
2575 
2576 /// ActOnCapScopeReturnStmt - Utility routine to type-check return statements
2577 /// for capturing scopes.
2578 ///
2579 StmtResult
2580 Sema::ActOnCapScopeReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
2581   // If this is the first return we've seen, infer the return type.
2582   // [expr.prim.lambda]p4 in C++11; block literals follow the same rules.
2583   CapturingScopeInfo *CurCap = cast<CapturingScopeInfo>(getCurFunction());
2584   QualType FnRetType = CurCap->ReturnType;
2585   LambdaScopeInfo *CurLambda = dyn_cast<LambdaScopeInfo>(CurCap);
2586 
2587   if (CurLambda && hasDeducedReturnType(CurLambda->CallOperator)) {
2588     // In C++1y, the return type may involve 'auto'.
2589     // FIXME: Blocks might have a return type of 'auto' explicitly specified.
2590     FunctionDecl *FD = CurLambda->CallOperator;
2591     if (CurCap->ReturnType.isNull())
2592       CurCap->ReturnType = FD->getReturnType();
2593 
2594     AutoType *AT = CurCap->ReturnType->getContainedAutoType();
2595     assert(AT && "lost auto type from lambda return type");
2596     if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
2597       FD->setInvalidDecl();
2598       return StmtError();
2599     }
2600     CurCap->ReturnType = FnRetType = FD->getReturnType();
2601   } else if (CurCap->HasImplicitReturnType) {
2602     // For blocks/lambdas with implicit return types, we check each return
2603     // statement individually, and deduce the common return type when the block
2604     // or lambda is completed.
2605     // FIXME: Fold this into the 'auto' codepath above.
2606     if (RetValExp && !isa<InitListExpr>(RetValExp)) {
2607       ExprResult Result = DefaultFunctionArrayLvalueConversion(RetValExp);
2608       if (Result.isInvalid())
2609         return StmtError();
2610       RetValExp = Result.take();
2611 
2612       if (!CurContext->isDependentContext())
2613         FnRetType = RetValExp->getType();
2614       else
2615         FnRetType = CurCap->ReturnType = Context.DependentTy;
2616     } else {
2617       if (RetValExp) {
2618         // C++11 [expr.lambda.prim]p4 bans inferring the result from an
2619         // initializer list, because it is not an expression (even
2620         // though we represent it as one). We still deduce 'void'.
2621         Diag(ReturnLoc, diag::err_lambda_return_init_list)
2622           << RetValExp->getSourceRange();
2623       }
2624 
2625       FnRetType = Context.VoidTy;
2626     }
2627 
2628     // Although we'll properly infer the type of the block once it's completed,
2629     // make sure we provide a return type now for better error recovery.
2630     if (CurCap->ReturnType.isNull())
2631       CurCap->ReturnType = FnRetType;
2632   }
2633   assert(!FnRetType.isNull());
2634 
2635   if (BlockScopeInfo *CurBlock = dyn_cast<BlockScopeInfo>(CurCap)) {
2636     if (CurBlock->FunctionType->getAs<FunctionType>()->getNoReturnAttr()) {
2637       Diag(ReturnLoc, diag::err_noreturn_block_has_return_expr);
2638       return StmtError();
2639     }
2640   } else if (CapturedRegionScopeInfo *CurRegion =
2641                  dyn_cast<CapturedRegionScopeInfo>(CurCap)) {
2642     Diag(ReturnLoc, diag::err_return_in_captured_stmt) << CurRegion->getRegionName();
2643     return StmtError();
2644   } else {
2645     assert(CurLambda && "unknown kind of captured scope");
2646     if (CurLambda->CallOperator->getType()->getAs<FunctionType>()
2647             ->getNoReturnAttr()) {
2648       Diag(ReturnLoc, diag::err_noreturn_lambda_has_return_expr);
2649       return StmtError();
2650     }
2651   }
2652 
2653   // Otherwise, verify that this result type matches the previous one.  We are
2654   // pickier with blocks than for normal functions because we don't have GCC
2655   // compatibility to worry about here.
2656   const VarDecl *NRVOCandidate = 0;
2657   if (FnRetType->isDependentType()) {
2658     // Delay processing for now.  TODO: there are lots of dependent
2659     // types we can conclusively prove aren't void.
2660   } else if (FnRetType->isVoidType()) {
2661     if (RetValExp && !isa<InitListExpr>(RetValExp) &&
2662         !(getLangOpts().CPlusPlus &&
2663           (RetValExp->isTypeDependent() ||
2664            RetValExp->getType()->isVoidType()))) {
2665       if (!getLangOpts().CPlusPlus &&
2666           RetValExp->getType()->isVoidType())
2667         Diag(ReturnLoc, diag::ext_return_has_void_expr) << "literal" << 2;
2668       else {
2669         Diag(ReturnLoc, diag::err_return_block_has_expr);
2670         RetValExp = 0;
2671       }
2672     }
2673   } else if (!RetValExp) {
2674     return StmtError(Diag(ReturnLoc, diag::err_block_return_missing_expr));
2675   } else if (!RetValExp->isTypeDependent()) {
2676     // we have a non-void block with an expression, continue checking
2677 
2678     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
2679     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
2680     // function return.
2681 
2682     // In C++ the return statement is handled via a copy initialization.
2683     // the C version of which boils down to CheckSingleAssignmentConstraints.
2684     NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, false);
2685     InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
2686                                                                    FnRetType,
2687                                                           NRVOCandidate != 0);
2688     ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
2689                                                      FnRetType, RetValExp);
2690     if (Res.isInvalid()) {
2691       // FIXME: Cleanup temporaries here, anyway?
2692       return StmtError();
2693     }
2694     RetValExp = Res.take();
2695     CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc);
2696   }
2697 
2698   if (RetValExp) {
2699     ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
2700     if (ER.isInvalid())
2701       return StmtError();
2702     RetValExp = ER.take();
2703   }
2704   ReturnStmt *Result = new (Context) ReturnStmt(ReturnLoc, RetValExp,
2705                                                 NRVOCandidate);
2706 
2707   // If we need to check for the named return value optimization,
2708   // or if we need to infer the return type,
2709   // save the return statement in our scope for later processing.
2710   if (CurCap->HasImplicitReturnType ||
2711       (getLangOpts().CPlusPlus && FnRetType->isRecordType() &&
2712        !CurContext->isDependentContext()))
2713     FunctionScopes.back()->Returns.push_back(Result);
2714 
2715   return Owned(Result);
2716 }
2717 
2718 /// Deduce the return type for a function from a returned expression, per
2719 /// C++1y [dcl.spec.auto]p6.
2720 bool Sema::DeduceFunctionTypeFromReturnExpr(FunctionDecl *FD,
2721                                             SourceLocation ReturnLoc,
2722                                             Expr *&RetExpr,
2723                                             AutoType *AT) {
2724   TypeLoc OrigResultType = FD->getTypeSourceInfo()->getTypeLoc().
2725     IgnoreParens().castAs<FunctionProtoTypeLoc>().getReturnLoc();
2726   QualType Deduced;
2727 
2728   if (RetExpr && isa<InitListExpr>(RetExpr)) {
2729     //  If the deduction is for a return statement and the initializer is
2730     //  a braced-init-list, the program is ill-formed.
2731     Diag(RetExpr->getExprLoc(),
2732          getCurLambda() ? diag::err_lambda_return_init_list
2733                         : diag::err_auto_fn_return_init_list)
2734         << RetExpr->getSourceRange();
2735     return true;
2736   }
2737 
2738   if (FD->isDependentContext()) {
2739     // C++1y [dcl.spec.auto]p12:
2740     //   Return type deduction [...] occurs when the definition is
2741     //   instantiated even if the function body contains a return
2742     //   statement with a non-type-dependent operand.
2743     assert(AT->isDeduced() && "should have deduced to dependent type");
2744     return false;
2745   } else if (RetExpr) {
2746     //  If the deduction is for a return statement and the initializer is
2747     //  a braced-init-list, the program is ill-formed.
2748     if (isa<InitListExpr>(RetExpr)) {
2749       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_return_init_list);
2750       return true;
2751     }
2752 
2753     //  Otherwise, [...] deduce a value for U using the rules of template
2754     //  argument deduction.
2755     DeduceAutoResult DAR = DeduceAutoType(OrigResultType, RetExpr, Deduced);
2756 
2757     if (DAR == DAR_Failed && !FD->isInvalidDecl())
2758       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_deduction_failure)
2759         << OrigResultType.getType() << RetExpr->getType();
2760 
2761     if (DAR != DAR_Succeeded)
2762       return true;
2763   } else {
2764     //  In the case of a return with no operand, the initializer is considered
2765     //  to be void().
2766     //
2767     // Deduction here can only succeed if the return type is exactly 'cv auto'
2768     // or 'decltype(auto)', so just check for that case directly.
2769     if (!OrigResultType.getType()->getAs<AutoType>()) {
2770       Diag(ReturnLoc, diag::err_auto_fn_return_void_but_not_auto)
2771         << OrigResultType.getType();
2772       return true;
2773     }
2774     // We always deduce U = void in this case.
2775     Deduced = SubstAutoType(OrigResultType.getType(), Context.VoidTy);
2776     if (Deduced.isNull())
2777       return true;
2778   }
2779 
2780   //  If a function with a declared return type that contains a placeholder type
2781   //  has multiple return statements, the return type is deduced for each return
2782   //  statement. [...] if the type deduced is not the same in each deduction,
2783   //  the program is ill-formed.
2784   if (AT->isDeduced() && !FD->isInvalidDecl()) {
2785     AutoType *NewAT = Deduced->getContainedAutoType();
2786     if (!FD->isDependentContext() &&
2787         !Context.hasSameType(AT->getDeducedType(), NewAT->getDeducedType())) {
2788       const LambdaScopeInfo *LambdaSI = getCurLambda();
2789       if (LambdaSI && LambdaSI->HasImplicitReturnType) {
2790         Diag(ReturnLoc, diag::err_typecheck_missing_return_type_incompatible)
2791           << NewAT->getDeducedType() << AT->getDeducedType()
2792           << true /*IsLambda*/;
2793       } else {
2794         Diag(ReturnLoc, diag::err_auto_fn_different_deductions)
2795           << (AT->isDecltypeAuto() ? 1 : 0)
2796           << NewAT->getDeducedType() << AT->getDeducedType();
2797       }
2798       return true;
2799     }
2800   } else if (!FD->isInvalidDecl()) {
2801     // Update all declarations of the function to have the deduced return type.
2802     Context.adjustDeducedFunctionResultType(FD, Deduced);
2803   }
2804 
2805   return false;
2806 }
2807 
2808 StmtResult
2809 Sema::ActOnReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
2810   // Check for unexpanded parameter packs.
2811   if (RetValExp && DiagnoseUnexpandedParameterPack(RetValExp))
2812     return StmtError();
2813 
2814   if (isa<CapturingScopeInfo>(getCurFunction()))
2815     return ActOnCapScopeReturnStmt(ReturnLoc, RetValExp);
2816 
2817   QualType FnRetType;
2818   QualType RelatedRetType;
2819   const AttrVec *Attrs = 0;
2820   bool isObjCMethod = false;
2821 
2822   if (const FunctionDecl *FD = getCurFunctionDecl()) {
2823     FnRetType = FD->getReturnType();
2824     if (FD->hasAttrs())
2825       Attrs = &FD->getAttrs();
2826     if (FD->isNoReturn())
2827       Diag(ReturnLoc, diag::warn_noreturn_function_has_return_expr)
2828         << FD->getDeclName();
2829   } else if (ObjCMethodDecl *MD = getCurMethodDecl()) {
2830     FnRetType = MD->getReturnType();
2831     isObjCMethod = true;
2832     if (MD->hasAttrs())
2833       Attrs = &MD->getAttrs();
2834     if (MD->hasRelatedResultType() && MD->getClassInterface()) {
2835       // In the implementation of a method with a related return type, the
2836       // type used to type-check the validity of return statements within the
2837       // method body is a pointer to the type of the class being implemented.
2838       RelatedRetType = Context.getObjCInterfaceType(MD->getClassInterface());
2839       RelatedRetType = Context.getObjCObjectPointerType(RelatedRetType);
2840     }
2841   } else // If we don't have a function/method context, bail.
2842     return StmtError();
2843 
2844   // FIXME: Add a flag to the ScopeInfo to indicate whether we're performing
2845   // deduction.
2846   if (getLangOpts().CPlusPlus1y) {
2847     if (AutoType *AT = FnRetType->getContainedAutoType()) {
2848       FunctionDecl *FD = cast<FunctionDecl>(CurContext);
2849       if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
2850         FD->setInvalidDecl();
2851         return StmtError();
2852       } else {
2853         FnRetType = FD->getReturnType();
2854       }
2855     }
2856   }
2857 
2858   bool HasDependentReturnType = FnRetType->isDependentType();
2859 
2860   ReturnStmt *Result = 0;
2861   if (FnRetType->isVoidType()) {
2862     if (RetValExp) {
2863       if (isa<InitListExpr>(RetValExp)) {
2864         // We simply never allow init lists as the return value of void
2865         // functions. This is compatible because this was never allowed before,
2866         // so there's no legacy code to deal with.
2867         NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2868         int FunctionKind = 0;
2869         if (isa<ObjCMethodDecl>(CurDecl))
2870           FunctionKind = 1;
2871         else if (isa<CXXConstructorDecl>(CurDecl))
2872           FunctionKind = 2;
2873         else if (isa<CXXDestructorDecl>(CurDecl))
2874           FunctionKind = 3;
2875 
2876         Diag(ReturnLoc, diag::err_return_init_list)
2877           << CurDecl->getDeclName() << FunctionKind
2878           << RetValExp->getSourceRange();
2879 
2880         // Drop the expression.
2881         RetValExp = 0;
2882       } else if (!RetValExp->isTypeDependent()) {
2883         // C99 6.8.6.4p1 (ext_ since GCC warns)
2884         unsigned D = diag::ext_return_has_expr;
2885         if (RetValExp->getType()->isVoidType()) {
2886           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2887           if (isa<CXXConstructorDecl>(CurDecl) ||
2888               isa<CXXDestructorDecl>(CurDecl))
2889             D = diag::err_ctor_dtor_returns_void;
2890           else
2891             D = diag::ext_return_has_void_expr;
2892         }
2893         else {
2894           ExprResult Result = Owned(RetValExp);
2895           Result = IgnoredValueConversions(Result.take());
2896           if (Result.isInvalid())
2897             return StmtError();
2898           RetValExp = Result.take();
2899           RetValExp = ImpCastExprToType(RetValExp,
2900                                         Context.VoidTy, CK_ToVoid).take();
2901         }
2902         // return of void in constructor/destructor is illegal in C++.
2903         if (D == diag::err_ctor_dtor_returns_void) {
2904           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2905           Diag(ReturnLoc, D)
2906             << CurDecl->getDeclName() << isa<CXXDestructorDecl>(CurDecl)
2907             << RetValExp->getSourceRange();
2908         }
2909         // return (some void expression); is legal in C++.
2910         else if (D != diag::ext_return_has_void_expr ||
2911             !getLangOpts().CPlusPlus) {
2912           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2913 
2914           int FunctionKind = 0;
2915           if (isa<ObjCMethodDecl>(CurDecl))
2916             FunctionKind = 1;
2917           else if (isa<CXXConstructorDecl>(CurDecl))
2918             FunctionKind = 2;
2919           else if (isa<CXXDestructorDecl>(CurDecl))
2920             FunctionKind = 3;
2921 
2922           Diag(ReturnLoc, D)
2923             << CurDecl->getDeclName() << FunctionKind
2924             << RetValExp->getSourceRange();
2925         }
2926       }
2927 
2928       if (RetValExp) {
2929         ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
2930         if (ER.isInvalid())
2931           return StmtError();
2932         RetValExp = ER.take();
2933       }
2934     }
2935 
2936     Result = new (Context) ReturnStmt(ReturnLoc, RetValExp, 0);
2937   } else if (!RetValExp && !HasDependentReturnType) {
2938     unsigned DiagID = diag::warn_return_missing_expr;  // C90 6.6.6.4p4
2939     // C99 6.8.6.4p1 (ext_ since GCC warns)
2940     if (getLangOpts().C99) DiagID = diag::ext_return_missing_expr;
2941 
2942     if (FunctionDecl *FD = getCurFunctionDecl())
2943       Diag(ReturnLoc, DiagID) << FD->getIdentifier() << 0/*fn*/;
2944     else
2945       Diag(ReturnLoc, DiagID) << getCurMethodDecl()->getDeclName() << 1/*meth*/;
2946     Result = new (Context) ReturnStmt(ReturnLoc);
2947   } else {
2948     assert(RetValExp || HasDependentReturnType);
2949     const VarDecl *NRVOCandidate = 0;
2950     if (!HasDependentReturnType && !RetValExp->isTypeDependent()) {
2951       // we have a non-void function with an expression, continue checking
2952 
2953       QualType RetType = (RelatedRetType.isNull() ? FnRetType : RelatedRetType);
2954 
2955       // C99 6.8.6.4p3(136): The return statement is not an assignment. The
2956       // overlap restriction of subclause 6.5.16.1 does not apply to the case of
2957       // function return.
2958 
2959       // In C++ the return statement is handled via a copy initialization,
2960       // the C version of which boils down to CheckSingleAssignmentConstraints.
2961       NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, false);
2962       InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
2963                                                                      RetType,
2964                                                             NRVOCandidate != 0);
2965       ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
2966                                                        RetType, RetValExp);
2967       if (Res.isInvalid()) {
2968         // FIXME: Clean up temporaries here anyway?
2969         return StmtError();
2970       }
2971       RetValExp = Res.takeAs<Expr>();
2972 
2973       // If we have a related result type, we need to implicitly
2974       // convert back to the formal result type.  We can't pretend to
2975       // initialize the result again --- we might end double-retaining
2976       // --- so instead we initialize a notional temporary.
2977       if (!RelatedRetType.isNull()) {
2978         Entity = InitializedEntity::InitializeRelatedResult(getCurMethodDecl(),
2979                                                             FnRetType);
2980         Res = PerformCopyInitialization(Entity, ReturnLoc, RetValExp);
2981         if (Res.isInvalid()) {
2982           // FIXME: Clean up temporaries here anyway?
2983           return StmtError();
2984         }
2985         RetValExp = Res.takeAs<Expr>();
2986       }
2987 
2988       CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc, isObjCMethod, Attrs,
2989                          getCurFunctionDecl());
2990     }
2991 
2992     if (RetValExp) {
2993       ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
2994       if (ER.isInvalid())
2995         return StmtError();
2996       RetValExp = ER.take();
2997     }
2998     Result = new (Context) ReturnStmt(ReturnLoc, RetValExp, NRVOCandidate);
2999   }
3000 
3001   // If we need to check for the named return value optimization, save the
3002   // return statement in our scope for later processing.
3003   if (getLangOpts().CPlusPlus && FnRetType->isRecordType() &&
3004       !CurContext->isDependentContext())
3005     FunctionScopes.back()->Returns.push_back(Result);
3006 
3007   return Owned(Result);
3008 }
3009 
3010 StmtResult
3011 Sema::ActOnObjCAtCatchStmt(SourceLocation AtLoc,
3012                            SourceLocation RParen, Decl *Parm,
3013                            Stmt *Body) {
3014   VarDecl *Var = cast_or_null<VarDecl>(Parm);
3015   if (Var && Var->isInvalidDecl())
3016     return StmtError();
3017 
3018   return Owned(new (Context) ObjCAtCatchStmt(AtLoc, RParen, Var, Body));
3019 }
3020 
3021 StmtResult
3022 Sema::ActOnObjCAtFinallyStmt(SourceLocation AtLoc, Stmt *Body) {
3023   return Owned(new (Context) ObjCAtFinallyStmt(AtLoc, Body));
3024 }
3025 
3026 StmtResult
3027 Sema::ActOnObjCAtTryStmt(SourceLocation AtLoc, Stmt *Try,
3028                          MultiStmtArg CatchStmts, Stmt *Finally) {
3029   if (!getLangOpts().ObjCExceptions)
3030     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@try";
3031 
3032   getCurFunction()->setHasBranchProtectedScope();
3033   unsigned NumCatchStmts = CatchStmts.size();
3034   return Owned(ObjCAtTryStmt::Create(Context, AtLoc, Try,
3035                                      CatchStmts.data(),
3036                                      NumCatchStmts,
3037                                      Finally));
3038 }
3039 
3040 StmtResult Sema::BuildObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw) {
3041   if (Throw) {
3042     ExprResult Result = DefaultLvalueConversion(Throw);
3043     if (Result.isInvalid())
3044       return StmtError();
3045 
3046     Result = ActOnFinishFullExpr(Result.take());
3047     if (Result.isInvalid())
3048       return StmtError();
3049     Throw = Result.take();
3050 
3051     QualType ThrowType = Throw->getType();
3052     // Make sure the expression type is an ObjC pointer or "void *".
3053     if (!ThrowType->isDependentType() &&
3054         !ThrowType->isObjCObjectPointerType()) {
3055       const PointerType *PT = ThrowType->getAs<PointerType>();
3056       if (!PT || !PT->getPointeeType()->isVoidType())
3057         return StmtError(Diag(AtLoc, diag::error_objc_throw_expects_object)
3058                          << Throw->getType() << Throw->getSourceRange());
3059     }
3060   }
3061 
3062   return Owned(new (Context) ObjCAtThrowStmt(AtLoc, Throw));
3063 }
3064 
3065 StmtResult
3066 Sema::ActOnObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw,
3067                            Scope *CurScope) {
3068   if (!getLangOpts().ObjCExceptions)
3069     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@throw";
3070 
3071   if (!Throw) {
3072     // @throw without an expression designates a rethrow (which much occur
3073     // in the context of an @catch clause).
3074     Scope *AtCatchParent = CurScope;
3075     while (AtCatchParent && !AtCatchParent->isAtCatchScope())
3076       AtCatchParent = AtCatchParent->getParent();
3077     if (!AtCatchParent)
3078       return StmtError(Diag(AtLoc, diag::error_rethrow_used_outside_catch));
3079   }
3080   return BuildObjCAtThrowStmt(AtLoc, Throw);
3081 }
3082 
3083 ExprResult
3084 Sema::ActOnObjCAtSynchronizedOperand(SourceLocation atLoc, Expr *operand) {
3085   ExprResult result = DefaultLvalueConversion(operand);
3086   if (result.isInvalid())
3087     return ExprError();
3088   operand = result.take();
3089 
3090   // Make sure the expression type is an ObjC pointer or "void *".
3091   QualType type = operand->getType();
3092   if (!type->isDependentType() &&
3093       !type->isObjCObjectPointerType()) {
3094     const PointerType *pointerType = type->getAs<PointerType>();
3095     if (!pointerType || !pointerType->getPointeeType()->isVoidType())
3096       return Diag(atLoc, diag::error_objc_synchronized_expects_object)
3097                << type << operand->getSourceRange();
3098   }
3099 
3100   // The operand to @synchronized is a full-expression.
3101   return ActOnFinishFullExpr(operand);
3102 }
3103 
3104 StmtResult
3105 Sema::ActOnObjCAtSynchronizedStmt(SourceLocation AtLoc, Expr *SyncExpr,
3106                                   Stmt *SyncBody) {
3107   // We can't jump into or indirect-jump out of a @synchronized block.
3108   getCurFunction()->setHasBranchProtectedScope();
3109   return Owned(new (Context) ObjCAtSynchronizedStmt(AtLoc, SyncExpr, SyncBody));
3110 }
3111 
3112 /// ActOnCXXCatchBlock - Takes an exception declaration and a handler block
3113 /// and creates a proper catch handler from them.
3114 StmtResult
3115 Sema::ActOnCXXCatchBlock(SourceLocation CatchLoc, Decl *ExDecl,
3116                          Stmt *HandlerBlock) {
3117   // There's nothing to test that ActOnExceptionDecl didn't already test.
3118   return Owned(new (Context) CXXCatchStmt(CatchLoc,
3119                                           cast_or_null<VarDecl>(ExDecl),
3120                                           HandlerBlock));
3121 }
3122 
3123 StmtResult
3124 Sema::ActOnObjCAutoreleasePoolStmt(SourceLocation AtLoc, Stmt *Body) {
3125   getCurFunction()->setHasBranchProtectedScope();
3126   return Owned(new (Context) ObjCAutoreleasePoolStmt(AtLoc, Body));
3127 }
3128 
3129 namespace {
3130 
3131 class TypeWithHandler {
3132   QualType t;
3133   CXXCatchStmt *stmt;
3134 public:
3135   TypeWithHandler(const QualType &type, CXXCatchStmt *statement)
3136   : t(type), stmt(statement) {}
3137 
3138   // An arbitrary order is fine as long as it places identical
3139   // types next to each other.
3140   bool operator<(const TypeWithHandler &y) const {
3141     if (t.getAsOpaquePtr() < y.t.getAsOpaquePtr())
3142       return true;
3143     if (t.getAsOpaquePtr() > y.t.getAsOpaquePtr())
3144       return false;
3145     else
3146       return getTypeSpecStartLoc() < y.getTypeSpecStartLoc();
3147   }
3148 
3149   bool operator==(const TypeWithHandler& other) const {
3150     return t == other.t;
3151   }
3152 
3153   CXXCatchStmt *getCatchStmt() const { return stmt; }
3154   SourceLocation getTypeSpecStartLoc() const {
3155     return stmt->getExceptionDecl()->getTypeSpecStartLoc();
3156   }
3157 };
3158 
3159 }
3160 
3161 /// ActOnCXXTryBlock - Takes a try compound-statement and a number of
3162 /// handlers and creates a try statement from them.
3163 StmtResult Sema::ActOnCXXTryBlock(SourceLocation TryLoc, Stmt *TryBlock,
3164                                   ArrayRef<Stmt *> Handlers) {
3165   // Don't report an error if 'try' is used in system headers.
3166   if (!getLangOpts().CXXExceptions &&
3167       !getSourceManager().isInSystemHeader(TryLoc))
3168       Diag(TryLoc, diag::err_exceptions_disabled) << "try";
3169 
3170   const unsigned NumHandlers = Handlers.size();
3171   assert(NumHandlers > 0 &&
3172          "The parser shouldn't call this if there are no handlers.");
3173 
3174   SmallVector<TypeWithHandler, 8> TypesWithHandlers;
3175 
3176   for (unsigned i = 0; i < NumHandlers; ++i) {
3177     CXXCatchStmt *Handler = cast<CXXCatchStmt>(Handlers[i]);
3178     if (!Handler->getExceptionDecl()) {
3179       if (i < NumHandlers - 1)
3180         return StmtError(Diag(Handler->getLocStart(),
3181                               diag::err_early_catch_all));
3182 
3183       continue;
3184     }
3185 
3186     const QualType CaughtType = Handler->getCaughtType();
3187     const QualType CanonicalCaughtType = Context.getCanonicalType(CaughtType);
3188     TypesWithHandlers.push_back(TypeWithHandler(CanonicalCaughtType, Handler));
3189   }
3190 
3191   // Detect handlers for the same type as an earlier one.
3192   if (NumHandlers > 1) {
3193     llvm::array_pod_sort(TypesWithHandlers.begin(), TypesWithHandlers.end());
3194 
3195     TypeWithHandler prev = TypesWithHandlers[0];
3196     for (unsigned i = 1; i < TypesWithHandlers.size(); ++i) {
3197       TypeWithHandler curr = TypesWithHandlers[i];
3198 
3199       if (curr == prev) {
3200         Diag(curr.getTypeSpecStartLoc(),
3201              diag::warn_exception_caught_by_earlier_handler)
3202           << curr.getCatchStmt()->getCaughtType().getAsString();
3203         Diag(prev.getTypeSpecStartLoc(),
3204              diag::note_previous_exception_handler)
3205           << prev.getCatchStmt()->getCaughtType().getAsString();
3206       }
3207 
3208       prev = curr;
3209     }
3210   }
3211 
3212   getCurFunction()->setHasBranchProtectedScope();
3213 
3214   // FIXME: We should detect handlers that cannot catch anything because an
3215   // earlier handler catches a superclass. Need to find a method that is not
3216   // quadratic for this.
3217   // Neither of these are explicitly forbidden, but every compiler detects them
3218   // and warns.
3219 
3220   return Owned(CXXTryStmt::Create(Context, TryLoc, TryBlock, Handlers));
3221 }
3222 
3223 StmtResult
3224 Sema::ActOnSEHTryBlock(bool IsCXXTry,
3225                        SourceLocation TryLoc,
3226                        Stmt *TryBlock,
3227                        Stmt *Handler) {
3228   assert(TryBlock && Handler);
3229 
3230   getCurFunction()->setHasBranchProtectedScope();
3231 
3232   return Owned(SEHTryStmt::Create(Context,IsCXXTry,TryLoc,TryBlock,Handler));
3233 }
3234 
3235 StmtResult
3236 Sema::ActOnSEHExceptBlock(SourceLocation Loc,
3237                           Expr *FilterExpr,
3238                           Stmt *Block) {
3239   assert(FilterExpr && Block);
3240 
3241   if(!FilterExpr->getType()->isIntegerType()) {
3242     return StmtError(Diag(FilterExpr->getExprLoc(),
3243                      diag::err_filter_expression_integral)
3244                      << FilterExpr->getType());
3245   }
3246 
3247   return Owned(SEHExceptStmt::Create(Context,Loc,FilterExpr,Block));
3248 }
3249 
3250 StmtResult
3251 Sema::ActOnSEHFinallyBlock(SourceLocation Loc,
3252                            Stmt *Block) {
3253   assert(Block);
3254   return Owned(SEHFinallyStmt::Create(Context,Loc,Block));
3255 }
3256 
3257 StmtResult Sema::BuildMSDependentExistsStmt(SourceLocation KeywordLoc,
3258                                             bool IsIfExists,
3259                                             NestedNameSpecifierLoc QualifierLoc,
3260                                             DeclarationNameInfo NameInfo,
3261                                             Stmt *Nested)
3262 {
3263   return new (Context) MSDependentExistsStmt(KeywordLoc, IsIfExists,
3264                                              QualifierLoc, NameInfo,
3265                                              cast<CompoundStmt>(Nested));
3266 }
3267 
3268 
3269 StmtResult Sema::ActOnMSDependentExistsStmt(SourceLocation KeywordLoc,
3270                                             bool IsIfExists,
3271                                             CXXScopeSpec &SS,
3272                                             UnqualifiedId &Name,
3273                                             Stmt *Nested) {
3274   return BuildMSDependentExistsStmt(KeywordLoc, IsIfExists,
3275                                     SS.getWithLocInContext(Context),
3276                                     GetNameFromUnqualifiedId(Name),
3277                                     Nested);
3278 }
3279 
3280 RecordDecl*
3281 Sema::CreateCapturedStmtRecordDecl(CapturedDecl *&CD, SourceLocation Loc,
3282                                    unsigned NumParams) {
3283   DeclContext *DC = CurContext;
3284   while (!(DC->isFunctionOrMethod() || DC->isRecord() || DC->isFileContext()))
3285     DC = DC->getParent();
3286 
3287   RecordDecl *RD = 0;
3288   if (getLangOpts().CPlusPlus)
3289     RD = CXXRecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc, /*Id=*/0);
3290   else
3291     RD = RecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc, /*Id=*/0);
3292 
3293   DC->addDecl(RD);
3294   RD->setImplicit();
3295   RD->startDefinition();
3296 
3297   CD = CapturedDecl::Create(Context, CurContext, NumParams);
3298   DC->addDecl(CD);
3299 
3300   // Build the context parameter
3301   assert(NumParams > 0 && "CapturedStmt requires context parameter");
3302   DC = CapturedDecl::castToDeclContext(CD);
3303   IdentifierInfo *VarName = &Context.Idents.get("__context");
3304   QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
3305   ImplicitParamDecl *Param
3306     = ImplicitParamDecl::Create(Context, DC, Loc, VarName, ParamType);
3307   DC->addDecl(Param);
3308 
3309   CD->setContextParam(Param);
3310 
3311   return RD;
3312 }
3313 
3314 static void buildCapturedStmtCaptureList(
3315     SmallVectorImpl<CapturedStmt::Capture> &Captures,
3316     SmallVectorImpl<Expr *> &CaptureInits,
3317     ArrayRef<CapturingScopeInfo::Capture> Candidates) {
3318 
3319   typedef ArrayRef<CapturingScopeInfo::Capture>::const_iterator CaptureIter;
3320   for (CaptureIter Cap = Candidates.begin(); Cap != Candidates.end(); ++Cap) {
3321 
3322     if (Cap->isThisCapture()) {
3323       Captures.push_back(CapturedStmt::Capture(Cap->getLocation(),
3324                                                CapturedStmt::VCK_This));
3325       CaptureInits.push_back(Cap->getInitExpr());
3326       continue;
3327     }
3328 
3329     assert(Cap->isReferenceCapture() &&
3330            "non-reference capture not yet implemented");
3331 
3332     Captures.push_back(CapturedStmt::Capture(Cap->getLocation(),
3333                                              CapturedStmt::VCK_ByRef,
3334                                              Cap->getVariable()));
3335     CaptureInits.push_back(Cap->getInitExpr());
3336   }
3337 }
3338 
3339 void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
3340                                     CapturedRegionKind Kind,
3341                                     unsigned NumParams) {
3342   CapturedDecl *CD = 0;
3343   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, NumParams);
3344 
3345   // Enter the capturing scope for this captured region.
3346   PushCapturedRegionScope(CurScope, CD, RD, Kind);
3347 
3348   if (CurScope)
3349     PushDeclContext(CurScope, CD);
3350   else
3351     CurContext = CD;
3352 
3353   PushExpressionEvaluationContext(PotentiallyEvaluated);
3354 }
3355 
3356 void Sema::ActOnCapturedRegionError() {
3357   DiscardCleanupsInEvaluationContext();
3358   PopExpressionEvaluationContext();
3359 
3360   CapturedRegionScopeInfo *RSI = getCurCapturedRegion();
3361   RecordDecl *Record = RSI->TheRecordDecl;
3362   Record->setInvalidDecl();
3363 
3364   SmallVector<Decl*, 4> Fields(Record->fields());
3365   ActOnFields(/*Scope=*/0, Record->getLocation(), Record, Fields,
3366               SourceLocation(), SourceLocation(), /*AttributeList=*/0);
3367 
3368   PopDeclContext();
3369   PopFunctionScopeInfo();
3370 }
3371 
3372 StmtResult Sema::ActOnCapturedRegionEnd(Stmt *S) {
3373   CapturedRegionScopeInfo *RSI = getCurCapturedRegion();
3374 
3375   SmallVector<CapturedStmt::Capture, 4> Captures;
3376   SmallVector<Expr *, 4> CaptureInits;
3377   buildCapturedStmtCaptureList(Captures, CaptureInits, RSI->Captures);
3378 
3379   CapturedDecl *CD = RSI->TheCapturedDecl;
3380   RecordDecl *RD = RSI->TheRecordDecl;
3381 
3382   CapturedStmt *Res = CapturedStmt::Create(getASTContext(), S,
3383                                            RSI->CapRegionKind, Captures,
3384                                            CaptureInits, CD, RD);
3385 
3386   CD->setBody(Res->getCapturedStmt());
3387   RD->completeDefinition();
3388 
3389   DiscardCleanupsInEvaluationContext();
3390   PopExpressionEvaluationContext();
3391 
3392   PopDeclContext();
3393   PopFunctionScopeInfo();
3394 
3395   return Owned(Res);
3396 }
3397