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