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                                         OverloadCandidateSet::CSK_Normal);
2199       Sema::BeginEndFunction BEFFailure;
2200       ForRangeStatus RangeStatus =
2201           BuildNonArrayForRange(*this, S, BeginRangeRef.get(),
2202                                 EndRangeRef.get(), RangeType,
2203                                 BeginVar, EndVar, ColonLoc, &CandidateSet,
2204                                 &BeginExpr, &EndExpr, &BEFFailure);
2205 
2206       if (Kind == BFRK_Build && RangeStatus == FRS_NoViableFunction &&
2207           BEFFailure == BEF_begin) {
2208         // If the range is being built from an array parameter, emit a
2209         // a diagnostic that it is being treated as a pointer.
2210         if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Range)) {
2211           if (ParmVarDecl *PVD = dyn_cast<ParmVarDecl>(DRE->getDecl())) {
2212             QualType ArrayTy = PVD->getOriginalType();
2213             QualType PointerTy = PVD->getType();
2214             if (PointerTy->isPointerType() && ArrayTy->isArrayType()) {
2215               Diag(Range->getLocStart(), diag::err_range_on_array_parameter)
2216                 << RangeLoc << PVD << ArrayTy << PointerTy;
2217               Diag(PVD->getLocation(), diag::note_declared_at);
2218               return StmtError();
2219             }
2220           }
2221         }
2222 
2223         // If building the range failed, try dereferencing the range expression
2224         // unless a diagnostic was issued or the end function is problematic.
2225         StmtResult SR = RebuildForRangeWithDereference(*this, S, ForLoc,
2226                                                        LoopVarDecl, ColonLoc,
2227                                                        Range, RangeLoc,
2228                                                        RParenLoc);
2229         if (SR.isInvalid() || SR.isUsable())
2230           return SR;
2231       }
2232 
2233       // Otherwise, emit diagnostics if we haven't already.
2234       if (RangeStatus == FRS_NoViableFunction) {
2235         Expr *Range = BEFFailure ? EndRangeRef.get() : BeginRangeRef.get();
2236         Diag(Range->getLocStart(), diag::err_for_range_invalid)
2237             << RangeLoc << Range->getType() << BEFFailure;
2238         CandidateSet.NoteCandidates(*this, OCD_AllCandidates, Range);
2239       }
2240       // Return an error if no fix was discovered.
2241       if (RangeStatus != FRS_Success)
2242         return StmtError();
2243     }
2244 
2245     assert(!BeginExpr.isInvalid() && !EndExpr.isInvalid() &&
2246            "invalid range expression in for loop");
2247 
2248     // C++11 [dcl.spec.auto]p7: BeginType and EndType must be the same.
2249     QualType BeginType = BeginVar->getType(), EndType = EndVar->getType();
2250     if (!Context.hasSameType(BeginType, EndType)) {
2251       Diag(RangeLoc, diag::err_for_range_begin_end_types_differ)
2252         << BeginType << EndType;
2253       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2254       NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2255     }
2256 
2257     Decl *BeginEndDecls[] = { BeginVar, EndVar };
2258     // Claim the type doesn't contain auto: we've already done the checking.
2259     DeclGroupPtrTy BeginEndGroup =
2260         BuildDeclaratorGroup(llvm::MutableArrayRef<Decl *>(BeginEndDecls, 2),
2261                              /*TypeMayContainAuto=*/ false);
2262     BeginEndDecl = ActOnDeclStmt(BeginEndGroup, ColonLoc, ColonLoc);
2263 
2264     const QualType BeginRefNonRefType = BeginType.getNonReferenceType();
2265     ExprResult BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2266                                            VK_LValue, ColonLoc);
2267     if (BeginRef.isInvalid())
2268       return StmtError();
2269 
2270     ExprResult EndRef = BuildDeclRefExpr(EndVar, EndType.getNonReferenceType(),
2271                                          VK_LValue, ColonLoc);
2272     if (EndRef.isInvalid())
2273       return StmtError();
2274 
2275     // Build and check __begin != __end expression.
2276     NotEqExpr = ActOnBinOp(S, ColonLoc, tok::exclaimequal,
2277                            BeginRef.get(), EndRef.get());
2278     NotEqExpr = ActOnBooleanCondition(S, ColonLoc, NotEqExpr.get());
2279     NotEqExpr = ActOnFinishFullExpr(NotEqExpr.get());
2280     if (NotEqExpr.isInvalid()) {
2281       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2282         << RangeLoc << 0 << BeginRangeRef.get()->getType();
2283       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2284       if (!Context.hasSameType(BeginType, EndType))
2285         NoteForRangeBeginEndFunction(*this, EndExpr.get(), BEF_end);
2286       return StmtError();
2287     }
2288 
2289     // Build and check ++__begin expression.
2290     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2291                                 VK_LValue, ColonLoc);
2292     if (BeginRef.isInvalid())
2293       return StmtError();
2294 
2295     IncrExpr = ActOnUnaryOp(S, ColonLoc, tok::plusplus, BeginRef.get());
2296     IncrExpr = ActOnFinishFullExpr(IncrExpr.get());
2297     if (IncrExpr.isInvalid()) {
2298       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2299         << RangeLoc << 2 << BeginRangeRef.get()->getType() ;
2300       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2301       return StmtError();
2302     }
2303 
2304     // Build and check *__begin  expression.
2305     BeginRef = BuildDeclRefExpr(BeginVar, BeginRefNonRefType,
2306                                 VK_LValue, ColonLoc);
2307     if (BeginRef.isInvalid())
2308       return StmtError();
2309 
2310     ExprResult DerefExpr = ActOnUnaryOp(S, ColonLoc, tok::star, BeginRef.get());
2311     if (DerefExpr.isInvalid()) {
2312       Diag(RangeLoc, diag::note_for_range_invalid_iterator)
2313         << RangeLoc << 1 << BeginRangeRef.get()->getType();
2314       NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2315       return StmtError();
2316     }
2317 
2318     // Attach  *__begin  as initializer for VD. Don't touch it if we're just
2319     // trying to determine whether this would be a valid range.
2320     if (!LoopVar->isInvalidDecl() && Kind != BFRK_Check) {
2321       AddInitializerToDecl(LoopVar, DerefExpr.get(), /*DirectInit=*/false,
2322                            /*TypeMayContainAuto=*/true);
2323       if (LoopVar->isInvalidDecl())
2324         NoteForRangeBeginEndFunction(*this, BeginExpr.get(), BEF_begin);
2325     }
2326   }
2327 
2328   // Don't bother to actually allocate the result if we're just trying to
2329   // determine whether it would be valid.
2330   if (Kind == BFRK_Check)
2331     return StmtResult();
2332 
2333   return Owned(new (Context) CXXForRangeStmt(RangeDS,
2334                                      cast_or_null<DeclStmt>(BeginEndDecl.get()),
2335                                              NotEqExpr.take(), IncrExpr.take(),
2336                                              LoopVarDS, /*Body=*/0, ForLoc,
2337                                              ColonLoc, RParenLoc));
2338 }
2339 
2340 /// FinishObjCForCollectionStmt - Attach the body to a objective-C foreach
2341 /// statement.
2342 StmtResult Sema::FinishObjCForCollectionStmt(Stmt *S, Stmt *B) {
2343   if (!S || !B)
2344     return StmtError();
2345   ObjCForCollectionStmt * ForStmt = cast<ObjCForCollectionStmt>(S);
2346 
2347   ForStmt->setBody(B);
2348   return S;
2349 }
2350 
2351 /// FinishCXXForRangeStmt - Attach the body to a C++0x for-range statement.
2352 /// This is a separate step from ActOnCXXForRangeStmt because analysis of the
2353 /// body cannot be performed until after the type of the range variable is
2354 /// determined.
2355 StmtResult Sema::FinishCXXForRangeStmt(Stmt *S, Stmt *B) {
2356   if (!S || !B)
2357     return StmtError();
2358 
2359   if (isa<ObjCForCollectionStmt>(S))
2360     return FinishObjCForCollectionStmt(S, B);
2361 
2362   CXXForRangeStmt *ForStmt = cast<CXXForRangeStmt>(S);
2363   ForStmt->setBody(B);
2364 
2365   DiagnoseEmptyStmtBody(ForStmt->getRParenLoc(), B,
2366                         diag::warn_empty_range_based_for_body);
2367 
2368   return S;
2369 }
2370 
2371 StmtResult Sema::ActOnGotoStmt(SourceLocation GotoLoc,
2372                                SourceLocation LabelLoc,
2373                                LabelDecl *TheDecl) {
2374   getCurFunction()->setHasBranchIntoScope();
2375   TheDecl->markUsed(Context);
2376   return Owned(new (Context) GotoStmt(TheDecl, GotoLoc, LabelLoc));
2377 }
2378 
2379 StmtResult
2380 Sema::ActOnIndirectGotoStmt(SourceLocation GotoLoc, SourceLocation StarLoc,
2381                             Expr *E) {
2382   // Convert operand to void*
2383   if (!E->isTypeDependent()) {
2384     QualType ETy = E->getType();
2385     QualType DestTy = Context.getPointerType(Context.VoidTy.withConst());
2386     ExprResult ExprRes = Owned(E);
2387     AssignConvertType ConvTy =
2388       CheckSingleAssignmentConstraints(DestTy, ExprRes);
2389     if (ExprRes.isInvalid())
2390       return StmtError();
2391     E = ExprRes.take();
2392     if (DiagnoseAssignmentResult(ConvTy, StarLoc, DestTy, ETy, E, AA_Passing))
2393       return StmtError();
2394   }
2395 
2396   ExprResult ExprRes = ActOnFinishFullExpr(E);
2397   if (ExprRes.isInvalid())
2398     return StmtError();
2399   E = ExprRes.take();
2400 
2401   getCurFunction()->setHasIndirectGoto();
2402 
2403   return Owned(new (Context) IndirectGotoStmt(GotoLoc, StarLoc, E));
2404 }
2405 
2406 StmtResult
2407 Sema::ActOnContinueStmt(SourceLocation ContinueLoc, Scope *CurScope) {
2408   Scope *S = CurScope->getContinueParent();
2409   if (!S) {
2410     // C99 6.8.6.2p1: A break shall appear only in or as a loop body.
2411     return StmtError(Diag(ContinueLoc, diag::err_continue_not_in_loop));
2412   }
2413 
2414   return Owned(new (Context) ContinueStmt(ContinueLoc));
2415 }
2416 
2417 StmtResult
2418 Sema::ActOnBreakStmt(SourceLocation BreakLoc, Scope *CurScope) {
2419   Scope *S = CurScope->getBreakParent();
2420   if (!S) {
2421     // C99 6.8.6.3p1: A break shall appear only in or as a switch/loop body.
2422     return StmtError(Diag(BreakLoc, diag::err_break_not_in_loop_or_switch));
2423   }
2424 
2425   return Owned(new (Context) BreakStmt(BreakLoc));
2426 }
2427 
2428 /// \brief Determine whether the given expression is a candidate for
2429 /// copy elision in either a return statement or a throw expression.
2430 ///
2431 /// \param ReturnType If we're determining the copy elision candidate for
2432 /// a return statement, this is the return type of the function. If we're
2433 /// determining the copy elision candidate for a throw expression, this will
2434 /// be a NULL type.
2435 ///
2436 /// \param E The expression being returned from the function or block, or
2437 /// being thrown.
2438 ///
2439 /// \param AllowFunctionParameter Whether we allow function parameters to
2440 /// be considered NRVO candidates. C++ prohibits this for NRVO itself, but
2441 /// we re-use this logic to determine whether we should try to move as part of
2442 /// a return or throw (which does allow function parameters).
2443 ///
2444 /// \returns The NRVO candidate variable, if the return statement may use the
2445 /// NRVO, or NULL if there is no such candidate.
2446 const VarDecl *Sema::getCopyElisionCandidate(QualType ReturnType,
2447                                              Expr *E,
2448                                              bool AllowFunctionParameter) {
2449   QualType ExprType = E->getType();
2450   // - in a return statement in a function with ...
2451   // ... a class return type ...
2452   if (!ReturnType.isNull()) {
2453     if (!ReturnType->isRecordType())
2454       return 0;
2455     // ... the same cv-unqualified type as the function return type ...
2456     if (!Context.hasSameUnqualifiedType(ReturnType, ExprType))
2457       return 0;
2458   }
2459 
2460   // ... the expression is the name of a non-volatile automatic object
2461   // (other than a function or catch-clause parameter)) ...
2462   const DeclRefExpr *DR = dyn_cast<DeclRefExpr>(E->IgnoreParens());
2463   if (!DR || DR->refersToEnclosingLocal())
2464     return 0;
2465   const VarDecl *VD = dyn_cast<VarDecl>(DR->getDecl());
2466   if (!VD)
2467     return 0;
2468 
2469   // ...object (other than a function or catch-clause parameter)...
2470   if (VD->getKind() != Decl::Var &&
2471       !(AllowFunctionParameter && VD->getKind() == Decl::ParmVar))
2472     return 0;
2473   if (VD->isExceptionVariable()) return 0;
2474 
2475   // ...automatic...
2476   if (!VD->hasLocalStorage()) return 0;
2477 
2478   // ...non-volatile...
2479   if (VD->getType().isVolatileQualified()) return 0;
2480   if (VD->getType()->isReferenceType()) return 0;
2481 
2482   // __block variables can't be allocated in a way that permits NRVO.
2483   if (VD->hasAttr<BlocksAttr>()) return 0;
2484 
2485   // Variables with higher required alignment than their type's ABI
2486   // alignment cannot use NRVO.
2487   if (VD->hasAttr<AlignedAttr>() &&
2488       Context.getDeclAlign(VD) > Context.getTypeAlignInChars(VD->getType()))
2489     return 0;
2490 
2491   return VD;
2492 }
2493 
2494 /// \brief Perform the initialization of a potentially-movable value, which
2495 /// is the result of return value.
2496 ///
2497 /// This routine implements C++0x [class.copy]p33, which attempts to treat
2498 /// returned lvalues as rvalues in certain cases (to prefer move construction),
2499 /// then falls back to treating them as lvalues if that failed.
2500 ExprResult
2501 Sema::PerformMoveOrCopyInitialization(const InitializedEntity &Entity,
2502                                       const VarDecl *NRVOCandidate,
2503                                       QualType ResultType,
2504                                       Expr *Value,
2505                                       bool AllowNRVO) {
2506   // C++0x [class.copy]p33:
2507   //   When the criteria for elision of a copy operation are met or would
2508   //   be met save for the fact that the source object is a function
2509   //   parameter, and the object to be copied is designated by an lvalue,
2510   //   overload resolution to select the constructor for the copy is first
2511   //   performed as if the object were designated by an rvalue.
2512   ExprResult Res = ExprError();
2513   if (AllowNRVO &&
2514       (NRVOCandidate || getCopyElisionCandidate(ResultType, Value, true))) {
2515     ImplicitCastExpr AsRvalue(ImplicitCastExpr::OnStack,
2516                               Value->getType(), CK_NoOp, Value, VK_XValue);
2517 
2518     Expr *InitExpr = &AsRvalue;
2519     InitializationKind Kind
2520       = InitializationKind::CreateCopy(Value->getLocStart(),
2521                                        Value->getLocStart());
2522     InitializationSequence Seq(*this, Entity, Kind, InitExpr);
2523 
2524     //   [...] If overload resolution fails, or if the type of the first
2525     //   parameter of the selected constructor is not an rvalue reference
2526     //   to the object's type (possibly cv-qualified), overload resolution
2527     //   is performed again, considering the object as an lvalue.
2528     if (Seq) {
2529       for (InitializationSequence::step_iterator Step = Seq.step_begin(),
2530            StepEnd = Seq.step_end();
2531            Step != StepEnd; ++Step) {
2532         if (Step->Kind != InitializationSequence::SK_ConstructorInitialization)
2533           continue;
2534 
2535         CXXConstructorDecl *Constructor
2536         = cast<CXXConstructorDecl>(Step->Function.Function);
2537 
2538         const RValueReferenceType *RRefType
2539           = Constructor->getParamDecl(0)->getType()
2540                                                  ->getAs<RValueReferenceType>();
2541 
2542         // If we don't meet the criteria, break out now.
2543         if (!RRefType ||
2544             !Context.hasSameUnqualifiedType(RRefType->getPointeeType(),
2545                             Context.getTypeDeclType(Constructor->getParent())))
2546           break;
2547 
2548         // Promote "AsRvalue" to the heap, since we now need this
2549         // expression node to persist.
2550         Value = ImplicitCastExpr::Create(Context, Value->getType(),
2551                                          CK_NoOp, Value, 0, VK_XValue);
2552 
2553         // Complete type-checking the initialization of the return type
2554         // using the constructor we found.
2555         Res = Seq.Perform(*this, Entity, Kind, Value);
2556       }
2557     }
2558   }
2559 
2560   // Either we didn't meet the criteria for treating an lvalue as an rvalue,
2561   // above, or overload resolution failed. Either way, we need to try
2562   // (again) now with the return value expression as written.
2563   if (Res.isInvalid())
2564     Res = PerformCopyInitialization(Entity, SourceLocation(), Value);
2565 
2566   return Res;
2567 }
2568 
2569 /// \brief Determine whether the declared return type of the specified function
2570 /// contains 'auto'.
2571 static bool hasDeducedReturnType(FunctionDecl *FD) {
2572   const FunctionProtoType *FPT =
2573       FD->getTypeSourceInfo()->getType()->castAs<FunctionProtoType>();
2574   return FPT->getReturnType()->isUndeducedType();
2575 }
2576 
2577 /// ActOnCapScopeReturnStmt - Utility routine to type-check return statements
2578 /// for capturing scopes.
2579 ///
2580 StmtResult
2581 Sema::ActOnCapScopeReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
2582   // If this is the first return we've seen, infer the return type.
2583   // [expr.prim.lambda]p4 in C++11; block literals follow the same rules.
2584   CapturingScopeInfo *CurCap = cast<CapturingScopeInfo>(getCurFunction());
2585   QualType FnRetType = CurCap->ReturnType;
2586   LambdaScopeInfo *CurLambda = dyn_cast<LambdaScopeInfo>(CurCap);
2587 
2588   if (CurLambda && hasDeducedReturnType(CurLambda->CallOperator)) {
2589     // In C++1y, the return type may involve 'auto'.
2590     // FIXME: Blocks might have a return type of 'auto' explicitly specified.
2591     FunctionDecl *FD = CurLambda->CallOperator;
2592     if (CurCap->ReturnType.isNull())
2593       CurCap->ReturnType = FD->getReturnType();
2594 
2595     AutoType *AT = CurCap->ReturnType->getContainedAutoType();
2596     assert(AT && "lost auto type from lambda return type");
2597     if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
2598       FD->setInvalidDecl();
2599       return StmtError();
2600     }
2601     CurCap->ReturnType = FnRetType = FD->getReturnType();
2602   } else if (CurCap->HasImplicitReturnType) {
2603     // For blocks/lambdas with implicit return types, we check each return
2604     // statement individually, and deduce the common return type when the block
2605     // or lambda is completed.
2606     // FIXME: Fold this into the 'auto' codepath above.
2607     if (RetValExp && !isa<InitListExpr>(RetValExp)) {
2608       ExprResult Result = DefaultFunctionArrayLvalueConversion(RetValExp);
2609       if (Result.isInvalid())
2610         return StmtError();
2611       RetValExp = Result.take();
2612 
2613       if (!CurContext->isDependentContext())
2614         FnRetType = RetValExp->getType();
2615       else
2616         FnRetType = CurCap->ReturnType = Context.DependentTy;
2617     } else {
2618       if (RetValExp) {
2619         // C++11 [expr.lambda.prim]p4 bans inferring the result from an
2620         // initializer list, because it is not an expression (even
2621         // though we represent it as one). We still deduce 'void'.
2622         Diag(ReturnLoc, diag::err_lambda_return_init_list)
2623           << RetValExp->getSourceRange();
2624       }
2625 
2626       FnRetType = Context.VoidTy;
2627     }
2628 
2629     // Although we'll properly infer the type of the block once it's completed,
2630     // make sure we provide a return type now for better error recovery.
2631     if (CurCap->ReturnType.isNull())
2632       CurCap->ReturnType = FnRetType;
2633   }
2634   assert(!FnRetType.isNull());
2635 
2636   if (BlockScopeInfo *CurBlock = dyn_cast<BlockScopeInfo>(CurCap)) {
2637     if (CurBlock->FunctionType->getAs<FunctionType>()->getNoReturnAttr()) {
2638       Diag(ReturnLoc, diag::err_noreturn_block_has_return_expr);
2639       return StmtError();
2640     }
2641   } else if (CapturedRegionScopeInfo *CurRegion =
2642                  dyn_cast<CapturedRegionScopeInfo>(CurCap)) {
2643     Diag(ReturnLoc, diag::err_return_in_captured_stmt) << CurRegion->getRegionName();
2644     return StmtError();
2645   } else {
2646     assert(CurLambda && "unknown kind of captured scope");
2647     if (CurLambda->CallOperator->getType()->getAs<FunctionType>()
2648             ->getNoReturnAttr()) {
2649       Diag(ReturnLoc, diag::err_noreturn_lambda_has_return_expr);
2650       return StmtError();
2651     }
2652   }
2653 
2654   // Otherwise, verify that this result type matches the previous one.  We are
2655   // pickier with blocks than for normal functions because we don't have GCC
2656   // compatibility to worry about here.
2657   const VarDecl *NRVOCandidate = 0;
2658   if (FnRetType->isDependentType()) {
2659     // Delay processing for now.  TODO: there are lots of dependent
2660     // types we can conclusively prove aren't void.
2661   } else if (FnRetType->isVoidType()) {
2662     if (RetValExp && !isa<InitListExpr>(RetValExp) &&
2663         !(getLangOpts().CPlusPlus &&
2664           (RetValExp->isTypeDependent() ||
2665            RetValExp->getType()->isVoidType()))) {
2666       if (!getLangOpts().CPlusPlus &&
2667           RetValExp->getType()->isVoidType())
2668         Diag(ReturnLoc, diag::ext_return_has_void_expr) << "literal" << 2;
2669       else {
2670         Diag(ReturnLoc, diag::err_return_block_has_expr);
2671         RetValExp = 0;
2672       }
2673     }
2674   } else if (!RetValExp) {
2675     return StmtError(Diag(ReturnLoc, diag::err_block_return_missing_expr));
2676   } else if (!RetValExp->isTypeDependent()) {
2677     // we have a non-void block with an expression, continue checking
2678 
2679     // C99 6.8.6.4p3(136): The return statement is not an assignment. The
2680     // overlap restriction of subclause 6.5.16.1 does not apply to the case of
2681     // function return.
2682 
2683     // In C++ the return statement is handled via a copy initialization.
2684     // the C version of which boils down to CheckSingleAssignmentConstraints.
2685     NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, false);
2686     InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
2687                                                                    FnRetType,
2688                                                           NRVOCandidate != 0);
2689     ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
2690                                                      FnRetType, RetValExp);
2691     if (Res.isInvalid()) {
2692       // FIXME: Cleanup temporaries here, anyway?
2693       return StmtError();
2694     }
2695     RetValExp = Res.take();
2696     CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc);
2697   }
2698 
2699   if (RetValExp) {
2700     ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
2701     if (ER.isInvalid())
2702       return StmtError();
2703     RetValExp = ER.take();
2704   }
2705   ReturnStmt *Result = new (Context) ReturnStmt(ReturnLoc, RetValExp,
2706                                                 NRVOCandidate);
2707 
2708   // If we need to check for the named return value optimization,
2709   // or if we need to infer the return type,
2710   // save the return statement in our scope for later processing.
2711   if (CurCap->HasImplicitReturnType ||
2712       (getLangOpts().CPlusPlus && FnRetType->isRecordType() &&
2713        !CurContext->isDependentContext()))
2714     FunctionScopes.back()->Returns.push_back(Result);
2715 
2716   return Owned(Result);
2717 }
2718 
2719 /// Deduce the return type for a function from a returned expression, per
2720 /// C++1y [dcl.spec.auto]p6.
2721 bool Sema::DeduceFunctionTypeFromReturnExpr(FunctionDecl *FD,
2722                                             SourceLocation ReturnLoc,
2723                                             Expr *&RetExpr,
2724                                             AutoType *AT) {
2725   TypeLoc OrigResultType = FD->getTypeSourceInfo()->getTypeLoc().
2726     IgnoreParens().castAs<FunctionProtoTypeLoc>().getReturnLoc();
2727   QualType Deduced;
2728 
2729   if (RetExpr && isa<InitListExpr>(RetExpr)) {
2730     //  If the deduction is for a return statement and the initializer is
2731     //  a braced-init-list, the program is ill-formed.
2732     Diag(RetExpr->getExprLoc(),
2733          getCurLambda() ? diag::err_lambda_return_init_list
2734                         : diag::err_auto_fn_return_init_list)
2735         << RetExpr->getSourceRange();
2736     return true;
2737   }
2738 
2739   if (FD->isDependentContext()) {
2740     // C++1y [dcl.spec.auto]p12:
2741     //   Return type deduction [...] occurs when the definition is
2742     //   instantiated even if the function body contains a return
2743     //   statement with a non-type-dependent operand.
2744     assert(AT->isDeduced() && "should have deduced to dependent type");
2745     return false;
2746   } else if (RetExpr) {
2747     //  If the deduction is for a return statement and the initializer is
2748     //  a braced-init-list, the program is ill-formed.
2749     if (isa<InitListExpr>(RetExpr)) {
2750       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_return_init_list);
2751       return true;
2752     }
2753 
2754     //  Otherwise, [...] deduce a value for U using the rules of template
2755     //  argument deduction.
2756     DeduceAutoResult DAR = DeduceAutoType(OrigResultType, RetExpr, Deduced);
2757 
2758     if (DAR == DAR_Failed && !FD->isInvalidDecl())
2759       Diag(RetExpr->getExprLoc(), diag::err_auto_fn_deduction_failure)
2760         << OrigResultType.getType() << RetExpr->getType();
2761 
2762     if (DAR != DAR_Succeeded)
2763       return true;
2764   } else {
2765     //  In the case of a return with no operand, the initializer is considered
2766     //  to be void().
2767     //
2768     // Deduction here can only succeed if the return type is exactly 'cv auto'
2769     // or 'decltype(auto)', so just check for that case directly.
2770     if (!OrigResultType.getType()->getAs<AutoType>()) {
2771       Diag(ReturnLoc, diag::err_auto_fn_return_void_but_not_auto)
2772         << OrigResultType.getType();
2773       return true;
2774     }
2775     // We always deduce U = void in this case.
2776     Deduced = SubstAutoType(OrigResultType.getType(), Context.VoidTy);
2777     if (Deduced.isNull())
2778       return true;
2779   }
2780 
2781   //  If a function with a declared return type that contains a placeholder type
2782   //  has multiple return statements, the return type is deduced for each return
2783   //  statement. [...] if the type deduced is not the same in each deduction,
2784   //  the program is ill-formed.
2785   if (AT->isDeduced() && !FD->isInvalidDecl()) {
2786     AutoType *NewAT = Deduced->getContainedAutoType();
2787     if (!FD->isDependentContext() &&
2788         !Context.hasSameType(AT->getDeducedType(), NewAT->getDeducedType())) {
2789       const LambdaScopeInfo *LambdaSI = getCurLambda();
2790       if (LambdaSI && LambdaSI->HasImplicitReturnType) {
2791         Diag(ReturnLoc, diag::err_typecheck_missing_return_type_incompatible)
2792           << NewAT->getDeducedType() << AT->getDeducedType()
2793           << true /*IsLambda*/;
2794       } else {
2795         Diag(ReturnLoc, diag::err_auto_fn_different_deductions)
2796           << (AT->isDecltypeAuto() ? 1 : 0)
2797           << NewAT->getDeducedType() << AT->getDeducedType();
2798       }
2799       return true;
2800     }
2801   } else if (!FD->isInvalidDecl()) {
2802     // Update all declarations of the function to have the deduced return type.
2803     Context.adjustDeducedFunctionResultType(FD, Deduced);
2804   }
2805 
2806   return false;
2807 }
2808 
2809 StmtResult
2810 Sema::ActOnReturnStmt(SourceLocation ReturnLoc, Expr *RetValExp) {
2811   // Check for unexpanded parameter packs.
2812   if (RetValExp && DiagnoseUnexpandedParameterPack(RetValExp))
2813     return StmtError();
2814 
2815   if (isa<CapturingScopeInfo>(getCurFunction()))
2816     return ActOnCapScopeReturnStmt(ReturnLoc, RetValExp);
2817 
2818   QualType FnRetType;
2819   QualType RelatedRetType;
2820   const AttrVec *Attrs = 0;
2821   bool isObjCMethod = false;
2822 
2823   if (const FunctionDecl *FD = getCurFunctionDecl()) {
2824     FnRetType = FD->getReturnType();
2825     if (FD->hasAttrs())
2826       Attrs = &FD->getAttrs();
2827     if (FD->isNoReturn())
2828       Diag(ReturnLoc, diag::warn_noreturn_function_has_return_expr)
2829         << FD->getDeclName();
2830   } else if (ObjCMethodDecl *MD = getCurMethodDecl()) {
2831     FnRetType = MD->getReturnType();
2832     isObjCMethod = true;
2833     if (MD->hasAttrs())
2834       Attrs = &MD->getAttrs();
2835     if (MD->hasRelatedResultType() && MD->getClassInterface()) {
2836       // In the implementation of a method with a related return type, the
2837       // type used to type-check the validity of return statements within the
2838       // method body is a pointer to the type of the class being implemented.
2839       RelatedRetType = Context.getObjCInterfaceType(MD->getClassInterface());
2840       RelatedRetType = Context.getObjCObjectPointerType(RelatedRetType);
2841     }
2842   } else // If we don't have a function/method context, bail.
2843     return StmtError();
2844 
2845   // FIXME: Add a flag to the ScopeInfo to indicate whether we're performing
2846   // deduction.
2847   if (getLangOpts().CPlusPlus1y) {
2848     if (AutoType *AT = FnRetType->getContainedAutoType()) {
2849       FunctionDecl *FD = cast<FunctionDecl>(CurContext);
2850       if (DeduceFunctionTypeFromReturnExpr(FD, ReturnLoc, RetValExp, AT)) {
2851         FD->setInvalidDecl();
2852         return StmtError();
2853       } else {
2854         FnRetType = FD->getReturnType();
2855       }
2856     }
2857   }
2858 
2859   bool HasDependentReturnType = FnRetType->isDependentType();
2860 
2861   ReturnStmt *Result = 0;
2862   if (FnRetType->isVoidType()) {
2863     if (RetValExp) {
2864       if (isa<InitListExpr>(RetValExp)) {
2865         // We simply never allow init lists as the return value of void
2866         // functions. This is compatible because this was never allowed before,
2867         // so there's no legacy code to deal with.
2868         NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2869         int FunctionKind = 0;
2870         if (isa<ObjCMethodDecl>(CurDecl))
2871           FunctionKind = 1;
2872         else if (isa<CXXConstructorDecl>(CurDecl))
2873           FunctionKind = 2;
2874         else if (isa<CXXDestructorDecl>(CurDecl))
2875           FunctionKind = 3;
2876 
2877         Diag(ReturnLoc, diag::err_return_init_list)
2878           << CurDecl->getDeclName() << FunctionKind
2879           << RetValExp->getSourceRange();
2880 
2881         // Drop the expression.
2882         RetValExp = 0;
2883       } else if (!RetValExp->isTypeDependent()) {
2884         // C99 6.8.6.4p1 (ext_ since GCC warns)
2885         unsigned D = diag::ext_return_has_expr;
2886         if (RetValExp->getType()->isVoidType()) {
2887           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2888           if (isa<CXXConstructorDecl>(CurDecl) ||
2889               isa<CXXDestructorDecl>(CurDecl))
2890             D = diag::err_ctor_dtor_returns_void;
2891           else
2892             D = diag::ext_return_has_void_expr;
2893         }
2894         else {
2895           ExprResult Result = Owned(RetValExp);
2896           Result = IgnoredValueConversions(Result.take());
2897           if (Result.isInvalid())
2898             return StmtError();
2899           RetValExp = Result.take();
2900           RetValExp = ImpCastExprToType(RetValExp,
2901                                         Context.VoidTy, CK_ToVoid).take();
2902         }
2903         // return of void in constructor/destructor is illegal in C++.
2904         if (D == diag::err_ctor_dtor_returns_void) {
2905           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2906           Diag(ReturnLoc, D)
2907             << CurDecl->getDeclName() << isa<CXXDestructorDecl>(CurDecl)
2908             << RetValExp->getSourceRange();
2909         }
2910         // return (some void expression); is legal in C++.
2911         else if (D != diag::ext_return_has_void_expr ||
2912             !getLangOpts().CPlusPlus) {
2913           NamedDecl *CurDecl = getCurFunctionOrMethodDecl();
2914 
2915           int FunctionKind = 0;
2916           if (isa<ObjCMethodDecl>(CurDecl))
2917             FunctionKind = 1;
2918           else if (isa<CXXConstructorDecl>(CurDecl))
2919             FunctionKind = 2;
2920           else if (isa<CXXDestructorDecl>(CurDecl))
2921             FunctionKind = 3;
2922 
2923           Diag(ReturnLoc, D)
2924             << CurDecl->getDeclName() << FunctionKind
2925             << RetValExp->getSourceRange();
2926         }
2927       }
2928 
2929       if (RetValExp) {
2930         ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
2931         if (ER.isInvalid())
2932           return StmtError();
2933         RetValExp = ER.take();
2934       }
2935     }
2936 
2937     Result = new (Context) ReturnStmt(ReturnLoc, RetValExp, 0);
2938   } else if (!RetValExp && !HasDependentReturnType) {
2939     unsigned DiagID = diag::warn_return_missing_expr;  // C90 6.6.6.4p4
2940     // C99 6.8.6.4p1 (ext_ since GCC warns)
2941     if (getLangOpts().C99) DiagID = diag::ext_return_missing_expr;
2942 
2943     if (FunctionDecl *FD = getCurFunctionDecl())
2944       Diag(ReturnLoc, DiagID) << FD->getIdentifier() << 0/*fn*/;
2945     else
2946       Diag(ReturnLoc, DiagID) << getCurMethodDecl()->getDeclName() << 1/*meth*/;
2947     Result = new (Context) ReturnStmt(ReturnLoc);
2948   } else {
2949     assert(RetValExp || HasDependentReturnType);
2950     const VarDecl *NRVOCandidate = 0;
2951     if (!HasDependentReturnType && !RetValExp->isTypeDependent()) {
2952       // we have a non-void function with an expression, continue checking
2953 
2954       QualType RetType = (RelatedRetType.isNull() ? FnRetType : RelatedRetType);
2955 
2956       // C99 6.8.6.4p3(136): The return statement is not an assignment. The
2957       // overlap restriction of subclause 6.5.16.1 does not apply to the case of
2958       // function return.
2959 
2960       // In C++ the return statement is handled via a copy initialization,
2961       // the C version of which boils down to CheckSingleAssignmentConstraints.
2962       NRVOCandidate = getCopyElisionCandidate(FnRetType, RetValExp, false);
2963       InitializedEntity Entity = InitializedEntity::InitializeResult(ReturnLoc,
2964                                                                      RetType,
2965                                                             NRVOCandidate != 0);
2966       ExprResult Res = PerformMoveOrCopyInitialization(Entity, NRVOCandidate,
2967                                                        RetType, RetValExp);
2968       if (Res.isInvalid()) {
2969         // FIXME: Clean up temporaries here anyway?
2970         return StmtError();
2971       }
2972       RetValExp = Res.takeAs<Expr>();
2973 
2974       // If we have a related result type, we need to implicitly
2975       // convert back to the formal result type.  We can't pretend to
2976       // initialize the result again --- we might end double-retaining
2977       // --- so instead we initialize a notional temporary.
2978       if (!RelatedRetType.isNull()) {
2979         Entity = InitializedEntity::InitializeRelatedResult(getCurMethodDecl(),
2980                                                             FnRetType);
2981         Res = PerformCopyInitialization(Entity, ReturnLoc, RetValExp);
2982         if (Res.isInvalid()) {
2983           // FIXME: Clean up temporaries here anyway?
2984           return StmtError();
2985         }
2986         RetValExp = Res.takeAs<Expr>();
2987       }
2988 
2989       CheckReturnValExpr(RetValExp, FnRetType, ReturnLoc, isObjCMethod, Attrs,
2990                          getCurFunctionDecl());
2991     }
2992 
2993     if (RetValExp) {
2994       ExprResult ER = ActOnFinishFullExpr(RetValExp, ReturnLoc);
2995       if (ER.isInvalid())
2996         return StmtError();
2997       RetValExp = ER.take();
2998     }
2999     Result = new (Context) ReturnStmt(ReturnLoc, RetValExp, NRVOCandidate);
3000   }
3001 
3002   // If we need to check for the named return value optimization, save the
3003   // return statement in our scope for later processing.
3004   if (getLangOpts().CPlusPlus && FnRetType->isRecordType() &&
3005       !CurContext->isDependentContext())
3006     FunctionScopes.back()->Returns.push_back(Result);
3007 
3008   return Owned(Result);
3009 }
3010 
3011 StmtResult
3012 Sema::ActOnObjCAtCatchStmt(SourceLocation AtLoc,
3013                            SourceLocation RParen, Decl *Parm,
3014                            Stmt *Body) {
3015   VarDecl *Var = cast_or_null<VarDecl>(Parm);
3016   if (Var && Var->isInvalidDecl())
3017     return StmtError();
3018 
3019   return Owned(new (Context) ObjCAtCatchStmt(AtLoc, RParen, Var, Body));
3020 }
3021 
3022 StmtResult
3023 Sema::ActOnObjCAtFinallyStmt(SourceLocation AtLoc, Stmt *Body) {
3024   return Owned(new (Context) ObjCAtFinallyStmt(AtLoc, Body));
3025 }
3026 
3027 StmtResult
3028 Sema::ActOnObjCAtTryStmt(SourceLocation AtLoc, Stmt *Try,
3029                          MultiStmtArg CatchStmts, Stmt *Finally) {
3030   if (!getLangOpts().ObjCExceptions)
3031     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@try";
3032 
3033   getCurFunction()->setHasBranchProtectedScope();
3034   unsigned NumCatchStmts = CatchStmts.size();
3035   return Owned(ObjCAtTryStmt::Create(Context, AtLoc, Try,
3036                                      CatchStmts.data(),
3037                                      NumCatchStmts,
3038                                      Finally));
3039 }
3040 
3041 StmtResult Sema::BuildObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw) {
3042   if (Throw) {
3043     ExprResult Result = DefaultLvalueConversion(Throw);
3044     if (Result.isInvalid())
3045       return StmtError();
3046 
3047     Result = ActOnFinishFullExpr(Result.take());
3048     if (Result.isInvalid())
3049       return StmtError();
3050     Throw = Result.take();
3051 
3052     QualType ThrowType = Throw->getType();
3053     // Make sure the expression type is an ObjC pointer or "void *".
3054     if (!ThrowType->isDependentType() &&
3055         !ThrowType->isObjCObjectPointerType()) {
3056       const PointerType *PT = ThrowType->getAs<PointerType>();
3057       if (!PT || !PT->getPointeeType()->isVoidType())
3058         return StmtError(Diag(AtLoc, diag::error_objc_throw_expects_object)
3059                          << Throw->getType() << Throw->getSourceRange());
3060     }
3061   }
3062 
3063   return Owned(new (Context) ObjCAtThrowStmt(AtLoc, Throw));
3064 }
3065 
3066 StmtResult
3067 Sema::ActOnObjCAtThrowStmt(SourceLocation AtLoc, Expr *Throw,
3068                            Scope *CurScope) {
3069   if (!getLangOpts().ObjCExceptions)
3070     Diag(AtLoc, diag::err_objc_exceptions_disabled) << "@throw";
3071 
3072   if (!Throw) {
3073     // @throw without an expression designates a rethrow (which much occur
3074     // in the context of an @catch clause).
3075     Scope *AtCatchParent = CurScope;
3076     while (AtCatchParent && !AtCatchParent->isAtCatchScope())
3077       AtCatchParent = AtCatchParent->getParent();
3078     if (!AtCatchParent)
3079       return StmtError(Diag(AtLoc, diag::error_rethrow_used_outside_catch));
3080   }
3081   return BuildObjCAtThrowStmt(AtLoc, Throw);
3082 }
3083 
3084 ExprResult
3085 Sema::ActOnObjCAtSynchronizedOperand(SourceLocation atLoc, Expr *operand) {
3086   ExprResult result = DefaultLvalueConversion(operand);
3087   if (result.isInvalid())
3088     return ExprError();
3089   operand = result.take();
3090 
3091   // Make sure the expression type is an ObjC pointer or "void *".
3092   QualType type = operand->getType();
3093   if (!type->isDependentType() &&
3094       !type->isObjCObjectPointerType()) {
3095     const PointerType *pointerType = type->getAs<PointerType>();
3096     if (!pointerType || !pointerType->getPointeeType()->isVoidType())
3097       return Diag(atLoc, diag::error_objc_synchronized_expects_object)
3098                << type << operand->getSourceRange();
3099   }
3100 
3101   // The operand to @synchronized is a full-expression.
3102   return ActOnFinishFullExpr(operand);
3103 }
3104 
3105 StmtResult
3106 Sema::ActOnObjCAtSynchronizedStmt(SourceLocation AtLoc, Expr *SyncExpr,
3107                                   Stmt *SyncBody) {
3108   // We can't jump into or indirect-jump out of a @synchronized block.
3109   getCurFunction()->setHasBranchProtectedScope();
3110   return Owned(new (Context) ObjCAtSynchronizedStmt(AtLoc, SyncExpr, SyncBody));
3111 }
3112 
3113 /// ActOnCXXCatchBlock - Takes an exception declaration and a handler block
3114 /// and creates a proper catch handler from them.
3115 StmtResult
3116 Sema::ActOnCXXCatchBlock(SourceLocation CatchLoc, Decl *ExDecl,
3117                          Stmt *HandlerBlock) {
3118   // There's nothing to test that ActOnExceptionDecl didn't already test.
3119   return Owned(new (Context) CXXCatchStmt(CatchLoc,
3120                                           cast_or_null<VarDecl>(ExDecl),
3121                                           HandlerBlock));
3122 }
3123 
3124 StmtResult
3125 Sema::ActOnObjCAutoreleasePoolStmt(SourceLocation AtLoc, Stmt *Body) {
3126   getCurFunction()->setHasBranchProtectedScope();
3127   return Owned(new (Context) ObjCAutoreleasePoolStmt(AtLoc, Body));
3128 }
3129 
3130 namespace {
3131 
3132 class TypeWithHandler {
3133   QualType t;
3134   CXXCatchStmt *stmt;
3135 public:
3136   TypeWithHandler(const QualType &type, CXXCatchStmt *statement)
3137   : t(type), stmt(statement) {}
3138 
3139   // An arbitrary order is fine as long as it places identical
3140   // types next to each other.
3141   bool operator<(const TypeWithHandler &y) const {
3142     if (t.getAsOpaquePtr() < y.t.getAsOpaquePtr())
3143       return true;
3144     if (t.getAsOpaquePtr() > y.t.getAsOpaquePtr())
3145       return false;
3146     else
3147       return getTypeSpecStartLoc() < y.getTypeSpecStartLoc();
3148   }
3149 
3150   bool operator==(const TypeWithHandler& other) const {
3151     return t == other.t;
3152   }
3153 
3154   CXXCatchStmt *getCatchStmt() const { return stmt; }
3155   SourceLocation getTypeSpecStartLoc() const {
3156     return stmt->getExceptionDecl()->getTypeSpecStartLoc();
3157   }
3158 };
3159 
3160 }
3161 
3162 /// ActOnCXXTryBlock - Takes a try compound-statement and a number of
3163 /// handlers and creates a try statement from them.
3164 StmtResult Sema::ActOnCXXTryBlock(SourceLocation TryLoc, Stmt *TryBlock,
3165                                   ArrayRef<Stmt *> Handlers) {
3166   // Don't report an error if 'try' is used in system headers.
3167   if (!getLangOpts().CXXExceptions &&
3168       !getSourceManager().isInSystemHeader(TryLoc))
3169       Diag(TryLoc, diag::err_exceptions_disabled) << "try";
3170 
3171   const unsigned NumHandlers = Handlers.size();
3172   assert(NumHandlers > 0 &&
3173          "The parser shouldn't call this if there are no handlers.");
3174 
3175   SmallVector<TypeWithHandler, 8> TypesWithHandlers;
3176 
3177   for (unsigned i = 0; i < NumHandlers; ++i) {
3178     CXXCatchStmt *Handler = cast<CXXCatchStmt>(Handlers[i]);
3179     if (!Handler->getExceptionDecl()) {
3180       if (i < NumHandlers - 1)
3181         return StmtError(Diag(Handler->getLocStart(),
3182                               diag::err_early_catch_all));
3183 
3184       continue;
3185     }
3186 
3187     const QualType CaughtType = Handler->getCaughtType();
3188     const QualType CanonicalCaughtType = Context.getCanonicalType(CaughtType);
3189     TypesWithHandlers.push_back(TypeWithHandler(CanonicalCaughtType, Handler));
3190   }
3191 
3192   // Detect handlers for the same type as an earlier one.
3193   if (NumHandlers > 1) {
3194     llvm::array_pod_sort(TypesWithHandlers.begin(), TypesWithHandlers.end());
3195 
3196     TypeWithHandler prev = TypesWithHandlers[0];
3197     for (unsigned i = 1; i < TypesWithHandlers.size(); ++i) {
3198       TypeWithHandler curr = TypesWithHandlers[i];
3199 
3200       if (curr == prev) {
3201         Diag(curr.getTypeSpecStartLoc(),
3202              diag::warn_exception_caught_by_earlier_handler)
3203           << curr.getCatchStmt()->getCaughtType().getAsString();
3204         Diag(prev.getTypeSpecStartLoc(),
3205              diag::note_previous_exception_handler)
3206           << prev.getCatchStmt()->getCaughtType().getAsString();
3207       }
3208 
3209       prev = curr;
3210     }
3211   }
3212 
3213   getCurFunction()->setHasBranchProtectedScope();
3214 
3215   // FIXME: We should detect handlers that cannot catch anything because an
3216   // earlier handler catches a superclass. Need to find a method that is not
3217   // quadratic for this.
3218   // Neither of these are explicitly forbidden, but every compiler detects them
3219   // and warns.
3220 
3221   return Owned(CXXTryStmt::Create(Context, TryLoc, TryBlock, Handlers));
3222 }
3223 
3224 StmtResult
3225 Sema::ActOnSEHTryBlock(bool IsCXXTry,
3226                        SourceLocation TryLoc,
3227                        Stmt *TryBlock,
3228                        Stmt *Handler) {
3229   assert(TryBlock && Handler);
3230 
3231   getCurFunction()->setHasBranchProtectedScope();
3232 
3233   return Owned(SEHTryStmt::Create(Context,IsCXXTry,TryLoc,TryBlock,Handler));
3234 }
3235 
3236 StmtResult
3237 Sema::ActOnSEHExceptBlock(SourceLocation Loc,
3238                           Expr *FilterExpr,
3239                           Stmt *Block) {
3240   assert(FilterExpr && Block);
3241 
3242   if(!FilterExpr->getType()->isIntegerType()) {
3243     return StmtError(Diag(FilterExpr->getExprLoc(),
3244                      diag::err_filter_expression_integral)
3245                      << FilterExpr->getType());
3246   }
3247 
3248   return Owned(SEHExceptStmt::Create(Context,Loc,FilterExpr,Block));
3249 }
3250 
3251 StmtResult
3252 Sema::ActOnSEHFinallyBlock(SourceLocation Loc,
3253                            Stmt *Block) {
3254   assert(Block);
3255   return Owned(SEHFinallyStmt::Create(Context,Loc,Block));
3256 }
3257 
3258 StmtResult Sema::BuildMSDependentExistsStmt(SourceLocation KeywordLoc,
3259                                             bool IsIfExists,
3260                                             NestedNameSpecifierLoc QualifierLoc,
3261                                             DeclarationNameInfo NameInfo,
3262                                             Stmt *Nested)
3263 {
3264   return new (Context) MSDependentExistsStmt(KeywordLoc, IsIfExists,
3265                                              QualifierLoc, NameInfo,
3266                                              cast<CompoundStmt>(Nested));
3267 }
3268 
3269 
3270 StmtResult Sema::ActOnMSDependentExistsStmt(SourceLocation KeywordLoc,
3271                                             bool IsIfExists,
3272                                             CXXScopeSpec &SS,
3273                                             UnqualifiedId &Name,
3274                                             Stmt *Nested) {
3275   return BuildMSDependentExistsStmt(KeywordLoc, IsIfExists,
3276                                     SS.getWithLocInContext(Context),
3277                                     GetNameFromUnqualifiedId(Name),
3278                                     Nested);
3279 }
3280 
3281 RecordDecl*
3282 Sema::CreateCapturedStmtRecordDecl(CapturedDecl *&CD, SourceLocation Loc,
3283                                    unsigned NumParams) {
3284   DeclContext *DC = CurContext;
3285   while (!(DC->isFunctionOrMethod() || DC->isRecord() || DC->isFileContext()))
3286     DC = DC->getParent();
3287 
3288   RecordDecl *RD = 0;
3289   if (getLangOpts().CPlusPlus)
3290     RD = CXXRecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc, /*Id=*/0);
3291   else
3292     RD = RecordDecl::Create(Context, TTK_Struct, DC, Loc, Loc, /*Id=*/0);
3293 
3294   DC->addDecl(RD);
3295   RD->setImplicit();
3296   RD->startDefinition();
3297 
3298   CD = CapturedDecl::Create(Context, CurContext, NumParams);
3299   DC->addDecl(CD);
3300 
3301   // Build the context parameter
3302   assert(NumParams > 0 && "CapturedStmt requires context parameter");
3303   DC = CapturedDecl::castToDeclContext(CD);
3304   IdentifierInfo *VarName = &Context.Idents.get("__context");
3305   QualType ParamType = Context.getPointerType(Context.getTagDeclType(RD));
3306   ImplicitParamDecl *Param
3307     = ImplicitParamDecl::Create(Context, DC, Loc, VarName, ParamType);
3308   DC->addDecl(Param);
3309 
3310   CD->setContextParam(Param);
3311 
3312   return RD;
3313 }
3314 
3315 static void buildCapturedStmtCaptureList(
3316     SmallVectorImpl<CapturedStmt::Capture> &Captures,
3317     SmallVectorImpl<Expr *> &CaptureInits,
3318     ArrayRef<CapturingScopeInfo::Capture> Candidates) {
3319 
3320   typedef ArrayRef<CapturingScopeInfo::Capture>::const_iterator CaptureIter;
3321   for (CaptureIter Cap = Candidates.begin(); Cap != Candidates.end(); ++Cap) {
3322 
3323     if (Cap->isThisCapture()) {
3324       Captures.push_back(CapturedStmt::Capture(Cap->getLocation(),
3325                                                CapturedStmt::VCK_This));
3326       CaptureInits.push_back(Cap->getInitExpr());
3327       continue;
3328     }
3329 
3330     assert(Cap->isReferenceCapture() &&
3331            "non-reference capture not yet implemented");
3332 
3333     Captures.push_back(CapturedStmt::Capture(Cap->getLocation(),
3334                                              CapturedStmt::VCK_ByRef,
3335                                              Cap->getVariable()));
3336     CaptureInits.push_back(Cap->getInitExpr());
3337   }
3338 }
3339 
3340 void Sema::ActOnCapturedRegionStart(SourceLocation Loc, Scope *CurScope,
3341                                     CapturedRegionKind Kind,
3342                                     unsigned NumParams) {
3343   CapturedDecl *CD = 0;
3344   RecordDecl *RD = CreateCapturedStmtRecordDecl(CD, Loc, NumParams);
3345 
3346   // Enter the capturing scope for this captured region.
3347   PushCapturedRegionScope(CurScope, CD, RD, Kind);
3348 
3349   if (CurScope)
3350     PushDeclContext(CurScope, CD);
3351   else
3352     CurContext = CD;
3353 
3354   PushExpressionEvaluationContext(PotentiallyEvaluated);
3355 }
3356 
3357 void Sema::ActOnCapturedRegionError() {
3358   DiscardCleanupsInEvaluationContext();
3359   PopExpressionEvaluationContext();
3360 
3361   CapturedRegionScopeInfo *RSI = getCurCapturedRegion();
3362   RecordDecl *Record = RSI->TheRecordDecl;
3363   Record->setInvalidDecl();
3364 
3365   SmallVector<Decl*, 4> Fields(Record->fields());
3366   ActOnFields(/*Scope=*/0, Record->getLocation(), Record, Fields,
3367               SourceLocation(), SourceLocation(), /*AttributeList=*/0);
3368 
3369   PopDeclContext();
3370   PopFunctionScopeInfo();
3371 }
3372 
3373 StmtResult Sema::ActOnCapturedRegionEnd(Stmt *S) {
3374   CapturedRegionScopeInfo *RSI = getCurCapturedRegion();
3375 
3376   SmallVector<CapturedStmt::Capture, 4> Captures;
3377   SmallVector<Expr *, 4> CaptureInits;
3378   buildCapturedStmtCaptureList(Captures, CaptureInits, RSI->Captures);
3379 
3380   CapturedDecl *CD = RSI->TheCapturedDecl;
3381   RecordDecl *RD = RSI->TheRecordDecl;
3382 
3383   CapturedStmt *Res = CapturedStmt::Create(getASTContext(), S,
3384                                            RSI->CapRegionKind, Captures,
3385                                            CaptureInits, CD, RD);
3386 
3387   CD->setBody(Res->getCapturedStmt());
3388   RD->completeDefinition();
3389 
3390   DiscardCleanupsInEvaluationContext();
3391   PopExpressionEvaluationContext();
3392 
3393   PopDeclContext();
3394   PopFunctionScopeInfo();
3395 
3396   return Owned(Res);
3397 }
3398