1 //===--- SemaLambda.cpp - Semantic Analysis for C++11 Lambdas -------------===//
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 C++ lambda expressions.
11 //
12 //===----------------------------------------------------------------------===//
13 #include "clang/Sema/DeclSpec.h"
14 #include "TypeLocBuilder.h"
15 #include "clang/AST/ASTLambda.h"
16 #include "clang/AST/ExprCXX.h"
17 #include "clang/Basic/TargetInfo.h"
18 #include "clang/Sema/Initialization.h"
19 #include "clang/Sema/Lookup.h"
20 #include "clang/Sema/Scope.h"
21 #include "clang/Sema/ScopeInfo.h"
22 #include "clang/Sema/SemaInternal.h"
23 #include "clang/Sema/SemaLambda.h"
24 using namespace clang;
25 using namespace sema;
26 
27 /// Examines the FunctionScopeInfo stack to determine the nearest
28 /// enclosing lambda (to the current lambda) that is 'capture-ready' for
29 /// the variable referenced in the current lambda (i.e. \p VarToCapture).
30 /// If successful, returns the index into Sema's FunctionScopeInfo stack
31 /// of the capture-ready lambda's LambdaScopeInfo.
32 ///
33 /// Climbs down the stack of lambdas (deepest nested lambda - i.e. current
34 /// lambda - is on top) to determine the index of the nearest enclosing/outer
35 /// lambda that is ready to capture the \p VarToCapture being referenced in
36 /// the current lambda.
37 /// As we climb down the stack, we want the index of the first such lambda -
38 /// that is the lambda with the highest index that is 'capture-ready'.
39 ///
40 /// A lambda 'L' is capture-ready for 'V' (var or this) if:
41 ///  - its enclosing context is non-dependent
42 ///  - and if the chain of lambdas between L and the lambda in which
43 ///    V is potentially used (i.e. the lambda at the top of the scope info
44 ///    stack), can all capture or have already captured V.
45 /// If \p VarToCapture is 'null' then we are trying to capture 'this'.
46 ///
47 /// Note that a lambda that is deemed 'capture-ready' still needs to be checked
48 /// for whether it is 'capture-capable' (see
49 /// getStackIndexOfNearestEnclosingCaptureCapableLambda), before it can truly
50 /// capture.
51 ///
52 /// \param FunctionScopes - Sema's stack of nested FunctionScopeInfo's (which a
53 ///  LambdaScopeInfo inherits from).  The current/deepest/innermost lambda
54 ///  is at the top of the stack and has the highest index.
55 /// \param VarToCapture - the variable to capture.  If NULL, capture 'this'.
56 ///
57 /// \returns An Optional<unsigned> Index that if evaluates to 'true' contains
58 /// the index (into Sema's FunctionScopeInfo stack) of the innermost lambda
59 /// which is capture-ready.  If the return value evaluates to 'false' then
60 /// no lambda is capture-ready for \p VarToCapture.
61 
62 static inline Optional<unsigned>
63 getStackIndexOfNearestEnclosingCaptureReadyLambda(
64     ArrayRef<const clang::sema::FunctionScopeInfo *> FunctionScopes,
65     VarDecl *VarToCapture) {
66   // Label failure to capture.
67   const Optional<unsigned> NoLambdaIsCaptureReady;
68 
69   // Ignore all inner captured regions.
70   unsigned CurScopeIndex = FunctionScopes.size() - 1;
71   while (CurScopeIndex > 0 && isa<clang::sema::CapturedRegionScopeInfo>(
72                                   FunctionScopes[CurScopeIndex]))
73     --CurScopeIndex;
74   assert(
75       isa<clang::sema::LambdaScopeInfo>(FunctionScopes[CurScopeIndex]) &&
76       "The function on the top of sema's function-info stack must be a lambda");
77 
78   // If VarToCapture is null, we are attempting to capture 'this'.
79   const bool IsCapturingThis = !VarToCapture;
80   const bool IsCapturingVariable = !IsCapturingThis;
81 
82   // Start with the current lambda at the top of the stack (highest index).
83   DeclContext *EnclosingDC =
84       cast<sema::LambdaScopeInfo>(FunctionScopes[CurScopeIndex])->CallOperator;
85 
86   do {
87     const clang::sema::LambdaScopeInfo *LSI =
88         cast<sema::LambdaScopeInfo>(FunctionScopes[CurScopeIndex]);
89     // IF we have climbed down to an intervening enclosing lambda that contains
90     // the variable declaration - it obviously can/must not capture the
91     // variable.
92     // Since its enclosing DC is dependent, all the lambdas between it and the
93     // innermost nested lambda are dependent (otherwise we wouldn't have
94     // arrived here) - so we don't yet have a lambda that can capture the
95     // variable.
96     if (IsCapturingVariable &&
97         VarToCapture->getDeclContext()->Equals(EnclosingDC))
98       return NoLambdaIsCaptureReady;
99 
100     // For an enclosing lambda to be capture ready for an entity, all
101     // intervening lambda's have to be able to capture that entity. If even
102     // one of the intervening lambda's is not capable of capturing the entity
103     // then no enclosing lambda can ever capture that entity.
104     // For e.g.
105     // const int x = 10;
106     // [=](auto a) {    #1
107     //   [](auto b) {   #2 <-- an intervening lambda that can never capture 'x'
108     //    [=](auto c) { #3
109     //       f(x, c);  <-- can not lead to x's speculative capture by #1 or #2
110     //    }; }; };
111     // If they do not have a default implicit capture, check to see
112     // if the entity has already been explicitly captured.
113     // If even a single dependent enclosing lambda lacks the capability
114     // to ever capture this variable, there is no further enclosing
115     // non-dependent lambda that can capture this variable.
116     if (LSI->ImpCaptureStyle == sema::LambdaScopeInfo::ImpCap_None) {
117       if (IsCapturingVariable && !LSI->isCaptured(VarToCapture))
118         return NoLambdaIsCaptureReady;
119       if (IsCapturingThis && !LSI->isCXXThisCaptured())
120         return NoLambdaIsCaptureReady;
121     }
122     EnclosingDC = getLambdaAwareParentOfDeclContext(EnclosingDC);
123 
124     assert(CurScopeIndex);
125     --CurScopeIndex;
126   } while (!EnclosingDC->isTranslationUnit() &&
127            EnclosingDC->isDependentContext() &&
128            isLambdaCallOperator(EnclosingDC));
129 
130   assert(CurScopeIndex < (FunctionScopes.size() - 1));
131   // If the enclosingDC is not dependent, then the immediately nested lambda
132   // (one index above) is capture-ready.
133   if (!EnclosingDC->isDependentContext())
134     return CurScopeIndex + 1;
135   return NoLambdaIsCaptureReady;
136 }
137 
138 /// Examines the FunctionScopeInfo stack to determine the nearest
139 /// enclosing lambda (to the current lambda) that is 'capture-capable' for
140 /// the variable referenced in the current lambda (i.e. \p VarToCapture).
141 /// If successful, returns the index into Sema's FunctionScopeInfo stack
142 /// of the capture-capable lambda's LambdaScopeInfo.
143 ///
144 /// Given the current stack of lambdas being processed by Sema and
145 /// the variable of interest, to identify the nearest enclosing lambda (to the
146 /// current lambda at the top of the stack) that can truly capture
147 /// a variable, it has to have the following two properties:
148 ///  a) 'capture-ready' - be the innermost lambda that is 'capture-ready':
149 ///     - climb down the stack (i.e. starting from the innermost and examining
150 ///       each outer lambda step by step) checking if each enclosing
151 ///       lambda can either implicitly or explicitly capture the variable.
152 ///       Record the first such lambda that is enclosed in a non-dependent
153 ///       context. If no such lambda currently exists return failure.
154 ///  b) 'capture-capable' - make sure the 'capture-ready' lambda can truly
155 ///  capture the variable by checking all its enclosing lambdas:
156 ///     - check if all outer lambdas enclosing the 'capture-ready' lambda
157 ///       identified above in 'a' can also capture the variable (this is done
158 ///       via tryCaptureVariable for variables and CheckCXXThisCapture for
159 ///       'this' by passing in the index of the Lambda identified in step 'a')
160 ///
161 /// \param FunctionScopes - Sema's stack of nested FunctionScopeInfo's (which a
162 /// LambdaScopeInfo inherits from).  The current/deepest/innermost lambda
163 /// is at the top of the stack.
164 ///
165 /// \param VarToCapture - the variable to capture.  If NULL, capture 'this'.
166 ///
167 ///
168 /// \returns An Optional<unsigned> Index that if evaluates to 'true' contains
169 /// the index (into Sema's FunctionScopeInfo stack) of the innermost lambda
170 /// which is capture-capable.  If the return value evaluates to 'false' then
171 /// no lambda is capture-capable for \p VarToCapture.
172 
173 Optional<unsigned> clang::getStackIndexOfNearestEnclosingCaptureCapableLambda(
174     ArrayRef<const sema::FunctionScopeInfo *> FunctionScopes,
175     VarDecl *VarToCapture, Sema &S) {
176 
177   const Optional<unsigned> NoLambdaIsCaptureCapable;
178 
179   const Optional<unsigned> OptionalStackIndex =
180       getStackIndexOfNearestEnclosingCaptureReadyLambda(FunctionScopes,
181                                                         VarToCapture);
182   if (!OptionalStackIndex)
183     return NoLambdaIsCaptureCapable;
184 
185   const unsigned IndexOfCaptureReadyLambda = OptionalStackIndex.getValue();
186   assert(((IndexOfCaptureReadyLambda != (FunctionScopes.size() - 1)) ||
187           S.getCurGenericLambda()) &&
188          "The capture ready lambda for a potential capture can only be the "
189          "current lambda if it is a generic lambda");
190 
191   const sema::LambdaScopeInfo *const CaptureReadyLambdaLSI =
192       cast<sema::LambdaScopeInfo>(FunctionScopes[IndexOfCaptureReadyLambda]);
193 
194   // If VarToCapture is null, we are attempting to capture 'this'
195   const bool IsCapturingThis = !VarToCapture;
196   const bool IsCapturingVariable = !IsCapturingThis;
197 
198   if (IsCapturingVariable) {
199     // Check if the capture-ready lambda can truly capture the variable, by
200     // checking whether all enclosing lambdas of the capture-ready lambda allow
201     // the capture - i.e. make sure it is capture-capable.
202     QualType CaptureType, DeclRefType;
203     const bool CanCaptureVariable =
204         !S.tryCaptureVariable(VarToCapture,
205                               /*ExprVarIsUsedInLoc*/ SourceLocation(),
206                               clang::Sema::TryCapture_Implicit,
207                               /*EllipsisLoc*/ SourceLocation(),
208                               /*BuildAndDiagnose*/ false, CaptureType,
209                               DeclRefType, &IndexOfCaptureReadyLambda);
210     if (!CanCaptureVariable)
211       return NoLambdaIsCaptureCapable;
212   } else {
213     // Check if the capture-ready lambda can truly capture 'this' by checking
214     // whether all enclosing lambdas of the capture-ready lambda can capture
215     // 'this'.
216     const bool CanCaptureThis =
217         !S.CheckCXXThisCapture(
218              CaptureReadyLambdaLSI->PotentialThisCaptureLocation,
219              /*Explicit*/ false, /*BuildAndDiagnose*/ false,
220              &IndexOfCaptureReadyLambda);
221     if (!CanCaptureThis)
222       return NoLambdaIsCaptureCapable;
223   }
224   return IndexOfCaptureReadyLambda;
225 }
226 
227 static inline TemplateParameterList *
228 getGenericLambdaTemplateParameterList(LambdaScopeInfo *LSI, Sema &SemaRef) {
229   if (LSI->GLTemplateParameterList)
230     return LSI->GLTemplateParameterList;
231 
232   if (!LSI->AutoTemplateParams.empty()) {
233     SourceRange IntroRange = LSI->IntroducerRange;
234     SourceLocation LAngleLoc = IntroRange.getBegin();
235     SourceLocation RAngleLoc = IntroRange.getEnd();
236     LSI->GLTemplateParameterList = TemplateParameterList::Create(
237         SemaRef.Context,
238         /*Template kw loc*/ SourceLocation(), LAngleLoc,
239         llvm::makeArrayRef((NamedDecl *const *)LSI->AutoTemplateParams.data(),
240                            LSI->AutoTemplateParams.size()),
241         RAngleLoc, nullptr);
242   }
243   return LSI->GLTemplateParameterList;
244 }
245 
246 CXXRecordDecl *Sema::createLambdaClosureType(SourceRange IntroducerRange,
247                                              TypeSourceInfo *Info,
248                                              bool KnownDependent,
249                                              LambdaCaptureDefault CaptureDefault) {
250   DeclContext *DC = CurContext;
251   while (!(DC->isFunctionOrMethod() || DC->isRecord() || DC->isFileContext()))
252     DC = DC->getParent();
253   bool IsGenericLambda = getGenericLambdaTemplateParameterList(getCurLambda(),
254                                                                *this);
255   // Start constructing the lambda class.
256   CXXRecordDecl *Class = CXXRecordDecl::CreateLambda(Context, DC, Info,
257                                                      IntroducerRange.getBegin(),
258                                                      KnownDependent,
259                                                      IsGenericLambda,
260                                                      CaptureDefault);
261   DC->addDecl(Class);
262 
263   return Class;
264 }
265 
266 /// Determine whether the given context is or is enclosed in an inline
267 /// function.
268 static bool isInInlineFunction(const DeclContext *DC) {
269   while (!DC->isFileContext()) {
270     if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(DC))
271       if (FD->isInlined())
272         return true;
273 
274     DC = DC->getLexicalParent();
275   }
276 
277   return false;
278 }
279 
280 MangleNumberingContext *
281 Sema::getCurrentMangleNumberContext(const DeclContext *DC,
282                                     Decl *&ManglingContextDecl) {
283   // Compute the context for allocating mangling numbers in the current
284   // expression, if the ABI requires them.
285   ManglingContextDecl = ExprEvalContexts.back().ManglingContextDecl;
286 
287   enum ContextKind {
288     Normal,
289     DefaultArgument,
290     DataMember,
291     StaticDataMember,
292     InlineVariable,
293     VariableTemplate
294   } Kind = Normal;
295 
296   // Default arguments of member function parameters that appear in a class
297   // definition, as well as the initializers of data members, receive special
298   // treatment. Identify them.
299   if (ManglingContextDecl) {
300     if (ParmVarDecl *Param = dyn_cast<ParmVarDecl>(ManglingContextDecl)) {
301       if (const DeclContext *LexicalDC
302           = Param->getDeclContext()->getLexicalParent())
303         if (LexicalDC->isRecord())
304           Kind = DefaultArgument;
305     } else if (VarDecl *Var = dyn_cast<VarDecl>(ManglingContextDecl)) {
306       if (Var->getDeclContext()->isRecord())
307         Kind = StaticDataMember;
308       else if (Var->getMostRecentDecl()->isInline())
309         Kind = InlineVariable;
310       else if (Var->getDescribedVarTemplate())
311         Kind = VariableTemplate;
312       else if (auto *VTS = dyn_cast<VarTemplateSpecializationDecl>(Var)) {
313         if (!VTS->isExplicitSpecialization())
314           Kind = VariableTemplate;
315       }
316     } else if (isa<FieldDecl>(ManglingContextDecl)) {
317       Kind = DataMember;
318     }
319   }
320 
321   // Itanium ABI [5.1.7]:
322   //   In the following contexts [...] the one-definition rule requires closure
323   //   types in different translation units to "correspond":
324   bool IsInNonspecializedTemplate =
325       inTemplateInstantiation() || CurContext->isDependentContext();
326   switch (Kind) {
327   case Normal: {
328     //  -- the bodies of non-exported nonspecialized template functions
329     //  -- the bodies of inline functions
330     if ((IsInNonspecializedTemplate &&
331          !(ManglingContextDecl && isa<ParmVarDecl>(ManglingContextDecl))) ||
332         isInInlineFunction(CurContext)) {
333       ManglingContextDecl = nullptr;
334       while (auto *CD = dyn_cast<CapturedDecl>(DC))
335         DC = CD->getParent();
336       return &Context.getManglingNumberContext(DC);
337     }
338 
339     ManglingContextDecl = nullptr;
340     return nullptr;
341   }
342 
343   case StaticDataMember:
344     //  -- the initializers of nonspecialized static members of template classes
345     if (!IsInNonspecializedTemplate) {
346       ManglingContextDecl = nullptr;
347       return nullptr;
348     }
349     // Fall through to get the current context.
350     LLVM_FALLTHROUGH;
351 
352   case DataMember:
353     //  -- the in-class initializers of class members
354   case DefaultArgument:
355     //  -- default arguments appearing in class definitions
356   case InlineVariable:
357     //  -- the initializers of inline variables
358   case VariableTemplate:
359     //  -- the initializers of templated variables
360     return &ExprEvalContexts.back().getMangleNumberingContext(Context);
361   }
362 
363   llvm_unreachable("unexpected context");
364 }
365 
366 MangleNumberingContext &
367 Sema::ExpressionEvaluationContextRecord::getMangleNumberingContext(
368     ASTContext &Ctx) {
369   assert(ManglingContextDecl && "Need to have a context declaration");
370   if (!MangleNumbering)
371     MangleNumbering = Ctx.createMangleNumberingContext();
372   return *MangleNumbering;
373 }
374 
375 CXXMethodDecl *Sema::startLambdaDefinition(CXXRecordDecl *Class,
376                                            SourceRange IntroducerRange,
377                                            TypeSourceInfo *MethodTypeInfo,
378                                            SourceLocation EndLoc,
379                                            ArrayRef<ParmVarDecl *> Params,
380                                            const bool IsConstexprSpecified) {
381   QualType MethodType = MethodTypeInfo->getType();
382   TemplateParameterList *TemplateParams =
383             getGenericLambdaTemplateParameterList(getCurLambda(), *this);
384   // If a lambda appears in a dependent context or is a generic lambda (has
385   // template parameters) and has an 'auto' return type, deduce it to a
386   // dependent type.
387   if (Class->isDependentContext() || TemplateParams) {
388     const FunctionProtoType *FPT = MethodType->castAs<FunctionProtoType>();
389     QualType Result = FPT->getReturnType();
390     if (Result->isUndeducedType()) {
391       Result = SubstAutoType(Result, Context.DependentTy);
392       MethodType = Context.getFunctionType(Result, FPT->getParamTypes(),
393                                            FPT->getExtProtoInfo());
394     }
395   }
396 
397   // C++11 [expr.prim.lambda]p5:
398   //   The closure type for a lambda-expression has a public inline function
399   //   call operator (13.5.4) whose parameters and return type are described by
400   //   the lambda-expression's parameter-declaration-clause and
401   //   trailing-return-type respectively.
402   DeclarationName MethodName
403     = Context.DeclarationNames.getCXXOperatorName(OO_Call);
404   DeclarationNameLoc MethodNameLoc;
405   MethodNameLoc.CXXOperatorName.BeginOpNameLoc
406     = IntroducerRange.getBegin().getRawEncoding();
407   MethodNameLoc.CXXOperatorName.EndOpNameLoc
408     = IntroducerRange.getEnd().getRawEncoding();
409   CXXMethodDecl *Method
410     = CXXMethodDecl::Create(Context, Class, EndLoc,
411                             DeclarationNameInfo(MethodName,
412                                                 IntroducerRange.getBegin(),
413                                                 MethodNameLoc),
414                             MethodType, MethodTypeInfo,
415                             SC_None,
416                             /*isInline=*/true,
417                             IsConstexprSpecified,
418                             EndLoc);
419   Method->setAccess(AS_public);
420 
421   // Temporarily set the lexical declaration context to the current
422   // context, so that the Scope stack matches the lexical nesting.
423   Method->setLexicalDeclContext(CurContext);
424   // Create a function template if we have a template parameter list
425   FunctionTemplateDecl *const TemplateMethod = TemplateParams ?
426             FunctionTemplateDecl::Create(Context, Class,
427                                          Method->getLocation(), MethodName,
428                                          TemplateParams,
429                                          Method) : nullptr;
430   if (TemplateMethod) {
431     TemplateMethod->setLexicalDeclContext(CurContext);
432     TemplateMethod->setAccess(AS_public);
433     Method->setDescribedFunctionTemplate(TemplateMethod);
434   }
435 
436   // Add parameters.
437   if (!Params.empty()) {
438     Method->setParams(Params);
439     CheckParmsForFunctionDef(Params,
440                              /*CheckParameterNames=*/false);
441 
442     for (auto P : Method->parameters())
443       P->setOwningFunction(Method);
444   }
445 
446   Decl *ManglingContextDecl;
447   if (MangleNumberingContext *MCtx =
448           getCurrentMangleNumberContext(Class->getDeclContext(),
449                                         ManglingContextDecl)) {
450     unsigned ManglingNumber = MCtx->getManglingNumber(Method);
451     Class->setLambdaMangling(ManglingNumber, ManglingContextDecl);
452   }
453 
454   return Method;
455 }
456 
457 void Sema::buildLambdaScope(LambdaScopeInfo *LSI,
458                                         CXXMethodDecl *CallOperator,
459                                         SourceRange IntroducerRange,
460                                         LambdaCaptureDefault CaptureDefault,
461                                         SourceLocation CaptureDefaultLoc,
462                                         bool ExplicitParams,
463                                         bool ExplicitResultType,
464                                         bool Mutable) {
465   LSI->CallOperator = CallOperator;
466   CXXRecordDecl *LambdaClass = CallOperator->getParent();
467   LSI->Lambda = LambdaClass;
468   if (CaptureDefault == LCD_ByCopy)
469     LSI->ImpCaptureStyle = LambdaScopeInfo::ImpCap_LambdaByval;
470   else if (CaptureDefault == LCD_ByRef)
471     LSI->ImpCaptureStyle = LambdaScopeInfo::ImpCap_LambdaByref;
472   LSI->CaptureDefaultLoc = CaptureDefaultLoc;
473   LSI->IntroducerRange = IntroducerRange;
474   LSI->ExplicitParams = ExplicitParams;
475   LSI->Mutable = Mutable;
476 
477   if (ExplicitResultType) {
478     LSI->ReturnType = CallOperator->getReturnType();
479 
480     if (!LSI->ReturnType->isDependentType() &&
481         !LSI->ReturnType->isVoidType()) {
482       if (RequireCompleteType(CallOperator->getBeginLoc(), LSI->ReturnType,
483                               diag::err_lambda_incomplete_result)) {
484         // Do nothing.
485       }
486     }
487   } else {
488     LSI->HasImplicitReturnType = true;
489   }
490 }
491 
492 void Sema::finishLambdaExplicitCaptures(LambdaScopeInfo *LSI) {
493   LSI->finishedExplicitCaptures();
494 }
495 
496 void Sema::addLambdaParameters(CXXMethodDecl *CallOperator, Scope *CurScope) {
497   // Introduce our parameters into the function scope
498   for (unsigned p = 0, NumParams = CallOperator->getNumParams();
499        p < NumParams; ++p) {
500     ParmVarDecl *Param = CallOperator->getParamDecl(p);
501 
502     // If this has an identifier, add it to the scope stack.
503     if (CurScope && Param->getIdentifier()) {
504       CheckShadow(CurScope, Param);
505 
506       PushOnScopeChains(Param, CurScope);
507     }
508   }
509 }
510 
511 /// If this expression is an enumerator-like expression of some type
512 /// T, return the type T; otherwise, return null.
513 ///
514 /// Pointer comparisons on the result here should always work because
515 /// it's derived from either the parent of an EnumConstantDecl
516 /// (i.e. the definition) or the declaration returned by
517 /// EnumType::getDecl() (i.e. the definition).
518 static EnumDecl *findEnumForBlockReturn(Expr *E) {
519   // An expression is an enumerator-like expression of type T if,
520   // ignoring parens and parens-like expressions:
521   E = E->IgnoreParens();
522 
523   //  - it is an enumerator whose enum type is T or
524   if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E)) {
525     if (EnumConstantDecl *D
526           = dyn_cast<EnumConstantDecl>(DRE->getDecl())) {
527       return cast<EnumDecl>(D->getDeclContext());
528     }
529     return nullptr;
530   }
531 
532   //  - it is a comma expression whose RHS is an enumerator-like
533   //    expression of type T or
534   if (BinaryOperator *BO = dyn_cast<BinaryOperator>(E)) {
535     if (BO->getOpcode() == BO_Comma)
536       return findEnumForBlockReturn(BO->getRHS());
537     return nullptr;
538   }
539 
540   //  - it is a statement-expression whose value expression is an
541   //    enumerator-like expression of type T or
542   if (StmtExpr *SE = dyn_cast<StmtExpr>(E)) {
543     if (Expr *last = dyn_cast_or_null<Expr>(SE->getSubStmt()->body_back()))
544       return findEnumForBlockReturn(last);
545     return nullptr;
546   }
547 
548   //   - it is a ternary conditional operator (not the GNU ?:
549   //     extension) whose second and third operands are
550   //     enumerator-like expressions of type T or
551   if (ConditionalOperator *CO = dyn_cast<ConditionalOperator>(E)) {
552     if (EnumDecl *ED = findEnumForBlockReturn(CO->getTrueExpr()))
553       if (ED == findEnumForBlockReturn(CO->getFalseExpr()))
554         return ED;
555     return nullptr;
556   }
557 
558   // (implicitly:)
559   //   - it is an implicit integral conversion applied to an
560   //     enumerator-like expression of type T or
561   if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(E)) {
562     // We can sometimes see integral conversions in valid
563     // enumerator-like expressions.
564     if (ICE->getCastKind() == CK_IntegralCast)
565       return findEnumForBlockReturn(ICE->getSubExpr());
566 
567     // Otherwise, just rely on the type.
568   }
569 
570   //   - it is an expression of that formal enum type.
571   if (const EnumType *ET = E->getType()->getAs<EnumType>()) {
572     return ET->getDecl();
573   }
574 
575   // Otherwise, nope.
576   return nullptr;
577 }
578 
579 /// Attempt to find a type T for which the returned expression of the
580 /// given statement is an enumerator-like expression of that type.
581 static EnumDecl *findEnumForBlockReturn(ReturnStmt *ret) {
582   if (Expr *retValue = ret->getRetValue())
583     return findEnumForBlockReturn(retValue);
584   return nullptr;
585 }
586 
587 /// Attempt to find a common type T for which all of the returned
588 /// expressions in a block are enumerator-like expressions of that
589 /// type.
590 static EnumDecl *findCommonEnumForBlockReturns(ArrayRef<ReturnStmt*> returns) {
591   ArrayRef<ReturnStmt*>::iterator i = returns.begin(), e = returns.end();
592 
593   // Try to find one for the first return.
594   EnumDecl *ED = findEnumForBlockReturn(*i);
595   if (!ED) return nullptr;
596 
597   // Check that the rest of the returns have the same enum.
598   for (++i; i != e; ++i) {
599     if (findEnumForBlockReturn(*i) != ED)
600       return nullptr;
601   }
602 
603   // Never infer an anonymous enum type.
604   if (!ED->hasNameForLinkage()) return nullptr;
605 
606   return ED;
607 }
608 
609 /// Adjust the given return statements so that they formally return
610 /// the given type.  It should require, at most, an IntegralCast.
611 static void adjustBlockReturnsToEnum(Sema &S, ArrayRef<ReturnStmt*> returns,
612                                      QualType returnType) {
613   for (ArrayRef<ReturnStmt*>::iterator
614          i = returns.begin(), e = returns.end(); i != e; ++i) {
615     ReturnStmt *ret = *i;
616     Expr *retValue = ret->getRetValue();
617     if (S.Context.hasSameType(retValue->getType(), returnType))
618       continue;
619 
620     // Right now we only support integral fixup casts.
621     assert(returnType->isIntegralOrUnscopedEnumerationType());
622     assert(retValue->getType()->isIntegralOrUnscopedEnumerationType());
623 
624     ExprWithCleanups *cleanups = dyn_cast<ExprWithCleanups>(retValue);
625 
626     Expr *E = (cleanups ? cleanups->getSubExpr() : retValue);
627     E = ImplicitCastExpr::Create(S.Context, returnType, CK_IntegralCast,
628                                  E, /*base path*/ nullptr, VK_RValue);
629     if (cleanups) {
630       cleanups->setSubExpr(E);
631     } else {
632       ret->setRetValue(E);
633     }
634   }
635 }
636 
637 void Sema::deduceClosureReturnType(CapturingScopeInfo &CSI) {
638   assert(CSI.HasImplicitReturnType);
639   // If it was ever a placeholder, it had to been deduced to DependentTy.
640   assert(CSI.ReturnType.isNull() || !CSI.ReturnType->isUndeducedType());
641   assert((!isa<LambdaScopeInfo>(CSI) || !getLangOpts().CPlusPlus14) &&
642          "lambda expressions use auto deduction in C++14 onwards");
643 
644   // C++ core issue 975:
645   //   If a lambda-expression does not include a trailing-return-type,
646   //   it is as if the trailing-return-type denotes the following type:
647   //     - if there are no return statements in the compound-statement,
648   //       or all return statements return either an expression of type
649   //       void or no expression or braced-init-list, the type void;
650   //     - otherwise, if all return statements return an expression
651   //       and the types of the returned expressions after
652   //       lvalue-to-rvalue conversion (4.1 [conv.lval]),
653   //       array-to-pointer conversion (4.2 [conv.array]), and
654   //       function-to-pointer conversion (4.3 [conv.func]) are the
655   //       same, that common type;
656   //     - otherwise, the program is ill-formed.
657   //
658   // C++ core issue 1048 additionally removes top-level cv-qualifiers
659   // from the types of returned expressions to match the C++14 auto
660   // deduction rules.
661   //
662   // In addition, in blocks in non-C++ modes, if all of the return
663   // statements are enumerator-like expressions of some type T, where
664   // T has a name for linkage, then we infer the return type of the
665   // block to be that type.
666 
667   // First case: no return statements, implicit void return type.
668   ASTContext &Ctx = getASTContext();
669   if (CSI.Returns.empty()) {
670     // It's possible there were simply no /valid/ return statements.
671     // In this case, the first one we found may have at least given us a type.
672     if (CSI.ReturnType.isNull())
673       CSI.ReturnType = Ctx.VoidTy;
674     return;
675   }
676 
677   // Second case: at least one return statement has dependent type.
678   // Delay type checking until instantiation.
679   assert(!CSI.ReturnType.isNull() && "We should have a tentative return type.");
680   if (CSI.ReturnType->isDependentType())
681     return;
682 
683   // Try to apply the enum-fuzz rule.
684   if (!getLangOpts().CPlusPlus) {
685     assert(isa<BlockScopeInfo>(CSI));
686     const EnumDecl *ED = findCommonEnumForBlockReturns(CSI.Returns);
687     if (ED) {
688       CSI.ReturnType = Context.getTypeDeclType(ED);
689       adjustBlockReturnsToEnum(*this, CSI.Returns, CSI.ReturnType);
690       return;
691     }
692   }
693 
694   // Third case: only one return statement. Don't bother doing extra work!
695   if (CSI.Returns.size() == 1)
696     return;
697 
698   // General case: many return statements.
699   // Check that they all have compatible return types.
700 
701   // We require the return types to strictly match here.
702   // Note that we've already done the required promotions as part of
703   // processing the return statement.
704   for (const ReturnStmt *RS : CSI.Returns) {
705     const Expr *RetE = RS->getRetValue();
706 
707     QualType ReturnType =
708         (RetE ? RetE->getType() : Context.VoidTy).getUnqualifiedType();
709     if (Context.getCanonicalFunctionResultType(ReturnType) ==
710           Context.getCanonicalFunctionResultType(CSI.ReturnType)) {
711       // Use the return type with the strictest possible nullability annotation.
712       auto RetTyNullability = ReturnType->getNullability(Ctx);
713       auto BlockNullability = CSI.ReturnType->getNullability(Ctx);
714       if (BlockNullability &&
715           (!RetTyNullability ||
716            hasWeakerNullability(*RetTyNullability, *BlockNullability)))
717         CSI.ReturnType = ReturnType;
718       continue;
719     }
720 
721     // FIXME: This is a poor diagnostic for ReturnStmts without expressions.
722     // TODO: It's possible that the *first* return is the divergent one.
723     Diag(RS->getBeginLoc(),
724          diag::err_typecheck_missing_return_type_incompatible)
725         << ReturnType << CSI.ReturnType << isa<LambdaScopeInfo>(CSI);
726     // Continue iterating so that we keep emitting diagnostics.
727   }
728 }
729 
730 QualType Sema::buildLambdaInitCaptureInitialization(SourceLocation Loc,
731                                                     bool ByRef,
732                                                     IdentifierInfo *Id,
733                                                     bool IsDirectInit,
734                                                     Expr *&Init) {
735   // Create an 'auto' or 'auto&' TypeSourceInfo that we can use to
736   // deduce against.
737   QualType DeductType = Context.getAutoDeductType();
738   TypeLocBuilder TLB;
739   TLB.pushTypeSpec(DeductType).setNameLoc(Loc);
740   if (ByRef) {
741     DeductType = BuildReferenceType(DeductType, true, Loc, Id);
742     assert(!DeductType.isNull() && "can't build reference to auto");
743     TLB.push<ReferenceTypeLoc>(DeductType).setSigilLoc(Loc);
744   }
745   TypeSourceInfo *TSI = TLB.getTypeSourceInfo(Context, DeductType);
746 
747   // Deduce the type of the init capture.
748   QualType DeducedType = deduceVarTypeFromInitializer(
749       /*VarDecl*/nullptr, DeclarationName(Id), DeductType, TSI,
750       SourceRange(Loc, Loc), IsDirectInit, Init);
751   if (DeducedType.isNull())
752     return QualType();
753 
754   // Are we a non-list direct initialization?
755   ParenListExpr *CXXDirectInit = dyn_cast<ParenListExpr>(Init);
756 
757   // Perform initialization analysis and ensure any implicit conversions
758   // (such as lvalue-to-rvalue) are enforced.
759   InitializedEntity Entity =
760       InitializedEntity::InitializeLambdaCapture(Id, DeducedType, Loc);
761   InitializationKind Kind =
762       IsDirectInit
763           ? (CXXDirectInit ? InitializationKind::CreateDirect(
764                                  Loc, Init->getBeginLoc(), Init->getEndLoc())
765                            : InitializationKind::CreateDirectList(Loc))
766           : InitializationKind::CreateCopy(Loc, Init->getBeginLoc());
767 
768   MultiExprArg Args = Init;
769   if (CXXDirectInit)
770     Args =
771         MultiExprArg(CXXDirectInit->getExprs(), CXXDirectInit->getNumExprs());
772   QualType DclT;
773   InitializationSequence InitSeq(*this, Entity, Kind, Args);
774   ExprResult Result = InitSeq.Perform(*this, Entity, Kind, Args, &DclT);
775 
776   if (Result.isInvalid())
777     return QualType();
778 
779   Init = Result.getAs<Expr>();
780   return DeducedType;
781 }
782 
783 VarDecl *Sema::createLambdaInitCaptureVarDecl(SourceLocation Loc,
784                                               QualType InitCaptureType,
785                                               IdentifierInfo *Id,
786                                               unsigned InitStyle, Expr *Init) {
787   TypeSourceInfo *TSI = Context.getTrivialTypeSourceInfo(InitCaptureType,
788       Loc);
789   // Create a dummy variable representing the init-capture. This is not actually
790   // used as a variable, and only exists as a way to name and refer to the
791   // init-capture.
792   // FIXME: Pass in separate source locations for '&' and identifier.
793   VarDecl *NewVD = VarDecl::Create(Context, CurContext, Loc,
794                                    Loc, Id, InitCaptureType, TSI, SC_Auto);
795   NewVD->setInitCapture(true);
796   NewVD->setReferenced(true);
797   // FIXME: Pass in a VarDecl::InitializationStyle.
798   NewVD->setInitStyle(static_cast<VarDecl::InitializationStyle>(InitStyle));
799   NewVD->markUsed(Context);
800   NewVD->setInit(Init);
801   return NewVD;
802 }
803 
804 FieldDecl *Sema::buildInitCaptureField(LambdaScopeInfo *LSI, VarDecl *Var) {
805   FieldDecl *Field = FieldDecl::Create(
806       Context, LSI->Lambda, Var->getLocation(), Var->getLocation(),
807       nullptr, Var->getType(), Var->getTypeSourceInfo(), nullptr, false,
808       ICIS_NoInit);
809   Field->setImplicit(true);
810   Field->setAccess(AS_private);
811   LSI->Lambda->addDecl(Field);
812 
813   LSI->addCapture(Var, /*isBlock*/false, Var->getType()->isReferenceType(),
814                   /*isNested*/false, Var->getLocation(), SourceLocation(),
815                   Var->getType(), Var->getInit());
816   return Field;
817 }
818 
819 void Sema::ActOnStartOfLambdaDefinition(LambdaIntroducer &Intro,
820                                         Declarator &ParamInfo,
821                                         Scope *CurScope) {
822   // Determine if we're within a context where we know that the lambda will
823   // be dependent, because there are template parameters in scope.
824   bool KnownDependent = false;
825   LambdaScopeInfo *const LSI = getCurLambda();
826   assert(LSI && "LambdaScopeInfo should be on stack!");
827 
828   // The lambda-expression's closure type might be dependent even if its
829   // semantic context isn't, if it appears within a default argument of a
830   // function template.
831   if (CurScope->getTemplateParamParent())
832     KnownDependent = true;
833 
834   // Determine the signature of the call operator.
835   TypeSourceInfo *MethodTyInfo;
836   bool ExplicitParams = true;
837   bool ExplicitResultType = true;
838   bool ContainsUnexpandedParameterPack = false;
839   SourceLocation EndLoc;
840   SmallVector<ParmVarDecl *, 8> Params;
841   if (ParamInfo.getNumTypeObjects() == 0) {
842     // C++11 [expr.prim.lambda]p4:
843     //   If a lambda-expression does not include a lambda-declarator, it is as
844     //   if the lambda-declarator were ().
845     FunctionProtoType::ExtProtoInfo EPI(Context.getDefaultCallingConvention(
846         /*IsVariadic=*/false, /*IsCXXMethod=*/true));
847     EPI.HasTrailingReturn = true;
848     EPI.TypeQuals |= DeclSpec::TQ_const;
849     // C++1y [expr.prim.lambda]:
850     //   The lambda return type is 'auto', which is replaced by the
851     //   trailing-return type if provided and/or deduced from 'return'
852     //   statements
853     // We don't do this before C++1y, because we don't support deduced return
854     // types there.
855     QualType DefaultTypeForNoTrailingReturn =
856         getLangOpts().CPlusPlus14 ? Context.getAutoDeductType()
857                                   : Context.DependentTy;
858     QualType MethodTy =
859         Context.getFunctionType(DefaultTypeForNoTrailingReturn, None, EPI);
860     MethodTyInfo = Context.getTrivialTypeSourceInfo(MethodTy);
861     ExplicitParams = false;
862     ExplicitResultType = false;
863     EndLoc = Intro.Range.getEnd();
864   } else {
865     assert(ParamInfo.isFunctionDeclarator() &&
866            "lambda-declarator is a function");
867     DeclaratorChunk::FunctionTypeInfo &FTI = ParamInfo.getFunctionTypeInfo();
868 
869     // C++11 [expr.prim.lambda]p5:
870     //   This function call operator is declared const (9.3.1) if and only if
871     //   the lambda-expression's parameter-declaration-clause is not followed
872     //   by mutable. It is neither virtual nor declared volatile. [...]
873     if (!FTI.hasMutableQualifier())
874       FTI.TypeQuals |= DeclSpec::TQ_const;
875 
876     MethodTyInfo = GetTypeForDeclarator(ParamInfo, CurScope);
877     assert(MethodTyInfo && "no type from lambda-declarator");
878     EndLoc = ParamInfo.getSourceRange().getEnd();
879 
880     ExplicitResultType = FTI.hasTrailingReturnType();
881 
882     if (FTIHasNonVoidParameters(FTI)) {
883       Params.reserve(FTI.NumParams);
884       for (unsigned i = 0, e = FTI.NumParams; i != e; ++i)
885         Params.push_back(cast<ParmVarDecl>(FTI.Params[i].Param));
886     }
887 
888     // Check for unexpanded parameter packs in the method type.
889     if (MethodTyInfo->getType()->containsUnexpandedParameterPack())
890       ContainsUnexpandedParameterPack = true;
891   }
892 
893   CXXRecordDecl *Class = createLambdaClosureType(Intro.Range, MethodTyInfo,
894                                                  KnownDependent, Intro.Default);
895 
896   CXXMethodDecl *Method =
897       startLambdaDefinition(Class, Intro.Range, MethodTyInfo, EndLoc, Params,
898                             ParamInfo.getDeclSpec().isConstexprSpecified());
899   if (ExplicitParams)
900     CheckCXXDefaultArguments(Method);
901 
902   // This represents the function body for the lambda function, check if we
903   // have to apply optnone due to a pragma.
904   AddRangeBasedOptnone(Method);
905 
906   // code_seg attribute on lambda apply to the method.
907   if (Attr *A = getImplicitCodeSegOrSectionAttrForFunction(Method, /*IsDefinition=*/true))
908     Method->addAttr(A);
909 
910   // Attributes on the lambda apply to the method.
911   ProcessDeclAttributes(CurScope, Method, ParamInfo);
912 
913   // CUDA lambdas get implicit attributes based on the scope in which they're
914   // declared.
915   if (getLangOpts().CUDA)
916     CUDASetLambdaAttrs(Method);
917 
918   // Introduce the function call operator as the current declaration context.
919   PushDeclContext(CurScope, Method);
920 
921   // Build the lambda scope.
922   buildLambdaScope(LSI, Method, Intro.Range, Intro.Default, Intro.DefaultLoc,
923                    ExplicitParams, ExplicitResultType, !Method->isConst());
924 
925   // C++11 [expr.prim.lambda]p9:
926   //   A lambda-expression whose smallest enclosing scope is a block scope is a
927   //   local lambda expression; any other lambda expression shall not have a
928   //   capture-default or simple-capture in its lambda-introducer.
929   //
930   // For simple-captures, this is covered by the check below that any named
931   // entity is a variable that can be captured.
932   //
933   // For DR1632, we also allow a capture-default in any context where we can
934   // odr-use 'this' (in particular, in a default initializer for a non-static
935   // data member).
936   if (Intro.Default != LCD_None && !Class->getParent()->isFunctionOrMethod() &&
937       (getCurrentThisType().isNull() ||
938        CheckCXXThisCapture(SourceLocation(), /*Explicit*/true,
939                            /*BuildAndDiagnose*/false)))
940     Diag(Intro.DefaultLoc, diag::err_capture_default_non_local);
941 
942   // Distinct capture names, for diagnostics.
943   llvm::SmallSet<IdentifierInfo*, 8> CaptureNames;
944 
945   // Handle explicit captures.
946   SourceLocation PrevCaptureLoc
947     = Intro.Default == LCD_None? Intro.Range.getBegin() : Intro.DefaultLoc;
948   for (auto C = Intro.Captures.begin(), E = Intro.Captures.end(); C != E;
949        PrevCaptureLoc = C->Loc, ++C) {
950     if (C->Kind == LCK_This || C->Kind == LCK_StarThis) {
951       if (C->Kind == LCK_StarThis)
952         Diag(C->Loc, !getLangOpts().CPlusPlus17
953                              ? diag::ext_star_this_lambda_capture_cxx17
954                              : diag::warn_cxx14_compat_star_this_lambda_capture);
955 
956       // C++11 [expr.prim.lambda]p8:
957       //   An identifier or this shall not appear more than once in a
958       //   lambda-capture.
959       if (LSI->isCXXThisCaptured()) {
960         Diag(C->Loc, diag::err_capture_more_than_once)
961             << "'this'" << SourceRange(LSI->getCXXThisCapture().getLocation())
962             << FixItHint::CreateRemoval(
963                    SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
964         continue;
965       }
966 
967       // C++2a [expr.prim.lambda]p8:
968       //  If a lambda-capture includes a capture-default that is =,
969       //  each simple-capture of that lambda-capture shall be of the form
970       //  "&identifier", "this", or "* this". [ Note: The form [&,this] is
971       //  redundant but accepted for compatibility with ISO C++14. --end note ]
972       if (Intro.Default == LCD_ByCopy && C->Kind != LCK_StarThis)
973         Diag(C->Loc, !getLangOpts().CPlusPlus2a
974                          ? diag::ext_equals_this_lambda_capture_cxx2a
975                          : diag::warn_cxx17_compat_equals_this_lambda_capture);
976 
977       // C++11 [expr.prim.lambda]p12:
978       //   If this is captured by a local lambda expression, its nearest
979       //   enclosing function shall be a non-static member function.
980       QualType ThisCaptureType = getCurrentThisType();
981       if (ThisCaptureType.isNull()) {
982         Diag(C->Loc, diag::err_this_capture) << true;
983         continue;
984       }
985 
986       CheckCXXThisCapture(C->Loc, /*Explicit=*/true, /*BuildAndDiagnose*/ true,
987                           /*FunctionScopeIndexToStopAtPtr*/ nullptr,
988                           C->Kind == LCK_StarThis);
989       if (!LSI->Captures.empty())
990         LSI->ExplicitCaptureRanges[LSI->Captures.size() - 1] = C->ExplicitRange;
991       continue;
992     }
993 
994     assert(C->Id && "missing identifier for capture");
995 
996     if (C->Init.isInvalid())
997       continue;
998 
999     VarDecl *Var = nullptr;
1000     if (C->Init.isUsable()) {
1001       Diag(C->Loc, getLangOpts().CPlusPlus14
1002                        ? diag::warn_cxx11_compat_init_capture
1003                        : diag::ext_init_capture);
1004 
1005       if (C->Init.get()->containsUnexpandedParameterPack())
1006         ContainsUnexpandedParameterPack = true;
1007       // If the initializer expression is usable, but the InitCaptureType
1008       // is not, then an error has occurred - so ignore the capture for now.
1009       // for e.g., [n{0}] { }; <-- if no <initializer_list> is included.
1010       // FIXME: we should create the init capture variable and mark it invalid
1011       // in this case.
1012       if (C->InitCaptureType.get().isNull())
1013         continue;
1014 
1015       unsigned InitStyle;
1016       switch (C->InitKind) {
1017       case LambdaCaptureInitKind::NoInit:
1018         llvm_unreachable("not an init-capture?");
1019       case LambdaCaptureInitKind::CopyInit:
1020         InitStyle = VarDecl::CInit;
1021         break;
1022       case LambdaCaptureInitKind::DirectInit:
1023         InitStyle = VarDecl::CallInit;
1024         break;
1025       case LambdaCaptureInitKind::ListInit:
1026         InitStyle = VarDecl::ListInit;
1027         break;
1028       }
1029       Var = createLambdaInitCaptureVarDecl(C->Loc, C->InitCaptureType.get(),
1030                                            C->Id, InitStyle, C->Init.get());
1031       // C++1y [expr.prim.lambda]p11:
1032       //   An init-capture behaves as if it declares and explicitly
1033       //   captures a variable [...] whose declarative region is the
1034       //   lambda-expression's compound-statement
1035       if (Var)
1036         PushOnScopeChains(Var, CurScope, false);
1037     } else {
1038       assert(C->InitKind == LambdaCaptureInitKind::NoInit &&
1039              "init capture has valid but null init?");
1040 
1041       // C++11 [expr.prim.lambda]p8:
1042       //   If a lambda-capture includes a capture-default that is &, the
1043       //   identifiers in the lambda-capture shall not be preceded by &.
1044       //   If a lambda-capture includes a capture-default that is =, [...]
1045       //   each identifier it contains shall be preceded by &.
1046       if (C->Kind == LCK_ByRef && Intro.Default == LCD_ByRef) {
1047         Diag(C->Loc, diag::err_reference_capture_with_reference_default)
1048             << FixItHint::CreateRemoval(
1049                 SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
1050         continue;
1051       } else if (C->Kind == LCK_ByCopy && Intro.Default == LCD_ByCopy) {
1052         Diag(C->Loc, diag::err_copy_capture_with_copy_default)
1053             << FixItHint::CreateRemoval(
1054                 SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
1055         continue;
1056       }
1057 
1058       // C++11 [expr.prim.lambda]p10:
1059       //   The identifiers in a capture-list are looked up using the usual
1060       //   rules for unqualified name lookup (3.4.1)
1061       DeclarationNameInfo Name(C->Id, C->Loc);
1062       LookupResult R(*this, Name, LookupOrdinaryName);
1063       LookupName(R, CurScope);
1064       if (R.isAmbiguous())
1065         continue;
1066       if (R.empty()) {
1067         // FIXME: Disable corrections that would add qualification?
1068         CXXScopeSpec ScopeSpec;
1069         if (DiagnoseEmptyLookup(CurScope, ScopeSpec, R,
1070                                 llvm::make_unique<DeclFilterCCC<VarDecl>>()))
1071           continue;
1072       }
1073 
1074       Var = R.getAsSingle<VarDecl>();
1075       if (Var && DiagnoseUseOfDecl(Var, C->Loc))
1076         continue;
1077     }
1078 
1079     // C++11 [expr.prim.lambda]p8:
1080     //   An identifier or this shall not appear more than once in a
1081     //   lambda-capture.
1082     if (!CaptureNames.insert(C->Id).second) {
1083       if (Var && LSI->isCaptured(Var)) {
1084         Diag(C->Loc, diag::err_capture_more_than_once)
1085             << C->Id << SourceRange(LSI->getCapture(Var).getLocation())
1086             << FixItHint::CreateRemoval(
1087                    SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
1088       } else
1089         // Previous capture captured something different (one or both was
1090         // an init-cpature): no fixit.
1091         Diag(C->Loc, diag::err_capture_more_than_once) << C->Id;
1092       continue;
1093     }
1094 
1095     // C++11 [expr.prim.lambda]p10:
1096     //   [...] each such lookup shall find a variable with automatic storage
1097     //   duration declared in the reaching scope of the local lambda expression.
1098     // Note that the 'reaching scope' check happens in tryCaptureVariable().
1099     if (!Var) {
1100       Diag(C->Loc, diag::err_capture_does_not_name_variable) << C->Id;
1101       continue;
1102     }
1103 
1104     // Ignore invalid decls; they'll just confuse the code later.
1105     if (Var->isInvalidDecl())
1106       continue;
1107 
1108     if (!Var->hasLocalStorage()) {
1109       Diag(C->Loc, diag::err_capture_non_automatic_variable) << C->Id;
1110       Diag(Var->getLocation(), diag::note_previous_decl) << C->Id;
1111       continue;
1112     }
1113 
1114     // C++11 [expr.prim.lambda]p23:
1115     //   A capture followed by an ellipsis is a pack expansion (14.5.3).
1116     SourceLocation EllipsisLoc;
1117     if (C->EllipsisLoc.isValid()) {
1118       if (Var->isParameterPack()) {
1119         EllipsisLoc = C->EllipsisLoc;
1120       } else {
1121         Diag(C->EllipsisLoc, diag::err_pack_expansion_without_parameter_packs)
1122           << SourceRange(C->Loc);
1123 
1124         // Just ignore the ellipsis.
1125       }
1126     } else if (Var->isParameterPack()) {
1127       ContainsUnexpandedParameterPack = true;
1128     }
1129 
1130     if (C->Init.isUsable()) {
1131       buildInitCaptureField(LSI, Var);
1132     } else {
1133       TryCaptureKind Kind = C->Kind == LCK_ByRef ? TryCapture_ExplicitByRef :
1134                                                    TryCapture_ExplicitByVal;
1135       tryCaptureVariable(Var, C->Loc, Kind, EllipsisLoc);
1136     }
1137     if (!LSI->Captures.empty())
1138       LSI->ExplicitCaptureRanges[LSI->Captures.size() - 1] = C->ExplicitRange;
1139   }
1140   finishLambdaExplicitCaptures(LSI);
1141 
1142   LSI->ContainsUnexpandedParameterPack = ContainsUnexpandedParameterPack;
1143 
1144   // Add lambda parameters into scope.
1145   addLambdaParameters(Method, CurScope);
1146 
1147   // Enter a new evaluation context to insulate the lambda from any
1148   // cleanups from the enclosing full-expression.
1149   PushExpressionEvaluationContext(
1150       ExpressionEvaluationContext::PotentiallyEvaluated);
1151 }
1152 
1153 void Sema::ActOnLambdaError(SourceLocation StartLoc, Scope *CurScope,
1154                             bool IsInstantiation) {
1155   LambdaScopeInfo *LSI = cast<LambdaScopeInfo>(FunctionScopes.back());
1156 
1157   // Leave the expression-evaluation context.
1158   DiscardCleanupsInEvaluationContext();
1159   PopExpressionEvaluationContext();
1160 
1161   // Leave the context of the lambda.
1162   if (!IsInstantiation)
1163     PopDeclContext();
1164 
1165   // Finalize the lambda.
1166   CXXRecordDecl *Class = LSI->Lambda;
1167   Class->setInvalidDecl();
1168   SmallVector<Decl*, 4> Fields(Class->fields());
1169   ActOnFields(nullptr, Class->getLocation(), Class, Fields, SourceLocation(),
1170               SourceLocation(), ParsedAttributesView());
1171   CheckCompletedCXXClass(Class);
1172 
1173   PopFunctionScopeInfo();
1174 }
1175 
1176 QualType Sema::getLambdaConversionFunctionResultType(
1177     const FunctionProtoType *CallOpProto) {
1178   // The function type inside the pointer type is the same as the call
1179   // operator with some tweaks. The calling convention is the default free
1180   // function convention, and the type qualifications are lost.
1181   const FunctionProtoType::ExtProtoInfo CallOpExtInfo =
1182       CallOpProto->getExtProtoInfo();
1183   FunctionProtoType::ExtProtoInfo InvokerExtInfo = CallOpExtInfo;
1184   CallingConv CC = Context.getDefaultCallingConvention(
1185       CallOpProto->isVariadic(), /*IsCXXMethod=*/false);
1186   InvokerExtInfo.ExtInfo = InvokerExtInfo.ExtInfo.withCallingConv(CC);
1187   InvokerExtInfo.TypeQuals = 0;
1188   assert(InvokerExtInfo.RefQualifier == RQ_None &&
1189       "Lambda's call operator should not have a reference qualifier");
1190   return Context.getFunctionType(CallOpProto->getReturnType(),
1191                                  CallOpProto->getParamTypes(), InvokerExtInfo);
1192 }
1193 
1194 /// Add a lambda's conversion to function pointer, as described in
1195 /// C++11 [expr.prim.lambda]p6.
1196 static void addFunctionPointerConversion(Sema &S,
1197                                          SourceRange IntroducerRange,
1198                                          CXXRecordDecl *Class,
1199                                          CXXMethodDecl *CallOperator) {
1200   // This conversion is explicitly disabled if the lambda's function has
1201   // pass_object_size attributes on any of its parameters.
1202   auto HasPassObjectSizeAttr = [](const ParmVarDecl *P) {
1203     return P->hasAttr<PassObjectSizeAttr>();
1204   };
1205   if (llvm::any_of(CallOperator->parameters(), HasPassObjectSizeAttr))
1206     return;
1207 
1208   // Add the conversion to function pointer.
1209   QualType InvokerFunctionTy = S.getLambdaConversionFunctionResultType(
1210       CallOperator->getType()->castAs<FunctionProtoType>());
1211   QualType PtrToFunctionTy = S.Context.getPointerType(InvokerFunctionTy);
1212 
1213   // Create the type of the conversion function.
1214   FunctionProtoType::ExtProtoInfo ConvExtInfo(
1215       S.Context.getDefaultCallingConvention(
1216       /*IsVariadic=*/false, /*IsCXXMethod=*/true));
1217   // The conversion function is always const.
1218   ConvExtInfo.TypeQuals = Qualifiers::Const;
1219   QualType ConvTy =
1220       S.Context.getFunctionType(PtrToFunctionTy, None, ConvExtInfo);
1221 
1222   SourceLocation Loc = IntroducerRange.getBegin();
1223   DeclarationName ConversionName
1224     = S.Context.DeclarationNames.getCXXConversionFunctionName(
1225         S.Context.getCanonicalType(PtrToFunctionTy));
1226   DeclarationNameLoc ConvNameLoc;
1227   // Construct a TypeSourceInfo for the conversion function, and wire
1228   // all the parameters appropriately for the FunctionProtoTypeLoc
1229   // so that everything works during transformation/instantiation of
1230   // generic lambdas.
1231   // The main reason for wiring up the parameters of the conversion
1232   // function with that of the call operator is so that constructs
1233   // like the following work:
1234   // auto L = [](auto b) {                <-- 1
1235   //   return [](auto a) -> decltype(a) { <-- 2
1236   //      return a;
1237   //   };
1238   // };
1239   // int (*fp)(int) = L(5);
1240   // Because the trailing return type can contain DeclRefExprs that refer
1241   // to the original call operator's variables, we hijack the call
1242   // operators ParmVarDecls below.
1243   TypeSourceInfo *ConvNamePtrToFunctionTSI =
1244       S.Context.getTrivialTypeSourceInfo(PtrToFunctionTy, Loc);
1245   ConvNameLoc.NamedType.TInfo = ConvNamePtrToFunctionTSI;
1246 
1247   // The conversion function is a conversion to a pointer-to-function.
1248   TypeSourceInfo *ConvTSI = S.Context.getTrivialTypeSourceInfo(ConvTy, Loc);
1249   FunctionProtoTypeLoc ConvTL =
1250       ConvTSI->getTypeLoc().getAs<FunctionProtoTypeLoc>();
1251   // Get the result of the conversion function which is a pointer-to-function.
1252   PointerTypeLoc PtrToFunctionTL =
1253       ConvTL.getReturnLoc().getAs<PointerTypeLoc>();
1254   // Do the same for the TypeSourceInfo that is used to name the conversion
1255   // operator.
1256   PointerTypeLoc ConvNamePtrToFunctionTL =
1257       ConvNamePtrToFunctionTSI->getTypeLoc().getAs<PointerTypeLoc>();
1258 
1259   // Get the underlying function types that the conversion function will
1260   // be converting to (should match the type of the call operator).
1261   FunctionProtoTypeLoc CallOpConvTL =
1262       PtrToFunctionTL.getPointeeLoc().getAs<FunctionProtoTypeLoc>();
1263   FunctionProtoTypeLoc CallOpConvNameTL =
1264     ConvNamePtrToFunctionTL.getPointeeLoc().getAs<FunctionProtoTypeLoc>();
1265 
1266   // Wire up the FunctionProtoTypeLocs with the call operator's parameters.
1267   // These parameter's are essentially used to transform the name and
1268   // the type of the conversion operator.  By using the same parameters
1269   // as the call operator's we don't have to fix any back references that
1270   // the trailing return type of the call operator's uses (such as
1271   // decltype(some_type<decltype(a)>::type{} + decltype(a){}) etc.)
1272   // - we can simply use the return type of the call operator, and
1273   // everything should work.
1274   SmallVector<ParmVarDecl *, 4> InvokerParams;
1275   for (unsigned I = 0, N = CallOperator->getNumParams(); I != N; ++I) {
1276     ParmVarDecl *From = CallOperator->getParamDecl(I);
1277 
1278     InvokerParams.push_back(ParmVarDecl::Create(
1279         S.Context,
1280         // Temporarily add to the TU. This is set to the invoker below.
1281         S.Context.getTranslationUnitDecl(), From->getBeginLoc(),
1282         From->getLocation(), From->getIdentifier(), From->getType(),
1283         From->getTypeSourceInfo(), From->getStorageClass(),
1284         /*DefaultArg=*/nullptr));
1285     CallOpConvTL.setParam(I, From);
1286     CallOpConvNameTL.setParam(I, From);
1287   }
1288 
1289   CXXConversionDecl *Conversion = CXXConversionDecl::Create(
1290       S.Context, Class, Loc,
1291       DeclarationNameInfo(ConversionName, Loc, ConvNameLoc), ConvTy, ConvTSI,
1292       /*isInline=*/true, /*isExplicit=*/false,
1293       /*isConstexpr=*/S.getLangOpts().CPlusPlus17,
1294       CallOperator->getBody()->getEndLoc());
1295   Conversion->setAccess(AS_public);
1296   Conversion->setImplicit(true);
1297 
1298   if (Class->isGenericLambda()) {
1299     // Create a template version of the conversion operator, using the template
1300     // parameter list of the function call operator.
1301     FunctionTemplateDecl *TemplateCallOperator =
1302             CallOperator->getDescribedFunctionTemplate();
1303     FunctionTemplateDecl *ConversionTemplate =
1304                   FunctionTemplateDecl::Create(S.Context, Class,
1305                                       Loc, ConversionName,
1306                                       TemplateCallOperator->getTemplateParameters(),
1307                                       Conversion);
1308     ConversionTemplate->setAccess(AS_public);
1309     ConversionTemplate->setImplicit(true);
1310     Conversion->setDescribedFunctionTemplate(ConversionTemplate);
1311     Class->addDecl(ConversionTemplate);
1312   } else
1313     Class->addDecl(Conversion);
1314   // Add a non-static member function that will be the result of
1315   // the conversion with a certain unique ID.
1316   DeclarationName InvokerName = &S.Context.Idents.get(
1317                                                  getLambdaStaticInvokerName());
1318   // FIXME: Instead of passing in the CallOperator->getTypeSourceInfo()
1319   // we should get a prebuilt TrivialTypeSourceInfo from Context
1320   // using FunctionTy & Loc and get its TypeLoc as a FunctionProtoTypeLoc
1321   // then rewire the parameters accordingly, by hoisting up the InvokeParams
1322   // loop below and then use its Params to set Invoke->setParams(...) below.
1323   // This would avoid the 'const' qualifier of the calloperator from
1324   // contaminating the type of the invoker, which is currently adjusted
1325   // in SemaTemplateDeduction.cpp:DeduceTemplateArguments.  Fixing the
1326   // trailing return type of the invoker would require a visitor to rebuild
1327   // the trailing return type and adjusting all back DeclRefExpr's to refer
1328   // to the new static invoker parameters - not the call operator's.
1329   CXXMethodDecl *Invoke = CXXMethodDecl::Create(
1330       S.Context, Class, Loc, DeclarationNameInfo(InvokerName, Loc),
1331       InvokerFunctionTy, CallOperator->getTypeSourceInfo(), SC_Static,
1332       /*IsInline=*/true,
1333       /*IsConstexpr=*/false, CallOperator->getBody()->getEndLoc());
1334   for (unsigned I = 0, N = CallOperator->getNumParams(); I != N; ++I)
1335     InvokerParams[I]->setOwningFunction(Invoke);
1336   Invoke->setParams(InvokerParams);
1337   Invoke->setAccess(AS_private);
1338   Invoke->setImplicit(true);
1339   if (Class->isGenericLambda()) {
1340     FunctionTemplateDecl *TemplateCallOperator =
1341             CallOperator->getDescribedFunctionTemplate();
1342     FunctionTemplateDecl *StaticInvokerTemplate = FunctionTemplateDecl::Create(
1343                           S.Context, Class, Loc, InvokerName,
1344                           TemplateCallOperator->getTemplateParameters(),
1345                           Invoke);
1346     StaticInvokerTemplate->setAccess(AS_private);
1347     StaticInvokerTemplate->setImplicit(true);
1348     Invoke->setDescribedFunctionTemplate(StaticInvokerTemplate);
1349     Class->addDecl(StaticInvokerTemplate);
1350   } else
1351     Class->addDecl(Invoke);
1352 }
1353 
1354 /// Add a lambda's conversion to block pointer.
1355 static void addBlockPointerConversion(Sema &S,
1356                                       SourceRange IntroducerRange,
1357                                       CXXRecordDecl *Class,
1358                                       CXXMethodDecl *CallOperator) {
1359   QualType FunctionTy = S.getLambdaConversionFunctionResultType(
1360       CallOperator->getType()->castAs<FunctionProtoType>());
1361   QualType BlockPtrTy = S.Context.getBlockPointerType(FunctionTy);
1362 
1363   FunctionProtoType::ExtProtoInfo ConversionEPI(
1364       S.Context.getDefaultCallingConvention(
1365           /*IsVariadic=*/false, /*IsCXXMethod=*/true));
1366   ConversionEPI.TypeQuals = Qualifiers::Const;
1367   QualType ConvTy = S.Context.getFunctionType(BlockPtrTy, None, ConversionEPI);
1368 
1369   SourceLocation Loc = IntroducerRange.getBegin();
1370   DeclarationName Name
1371     = S.Context.DeclarationNames.getCXXConversionFunctionName(
1372         S.Context.getCanonicalType(BlockPtrTy));
1373   DeclarationNameLoc NameLoc;
1374   NameLoc.NamedType.TInfo = S.Context.getTrivialTypeSourceInfo(BlockPtrTy, Loc);
1375   CXXConversionDecl *Conversion = CXXConversionDecl::Create(
1376       S.Context, Class, Loc, DeclarationNameInfo(Name, Loc, NameLoc), ConvTy,
1377       S.Context.getTrivialTypeSourceInfo(ConvTy, Loc),
1378       /*isInline=*/true, /*isExplicit=*/false,
1379       /*isConstexpr=*/false, CallOperator->getBody()->getEndLoc());
1380   Conversion->setAccess(AS_public);
1381   Conversion->setImplicit(true);
1382   Class->addDecl(Conversion);
1383 }
1384 
1385 static ExprResult performLambdaVarCaptureInitialization(
1386     Sema &S, const Capture &Capture, FieldDecl *Field,
1387     SourceLocation ImplicitCaptureLoc, bool IsImplicitCapture) {
1388   assert(Capture.isVariableCapture() && "not a variable capture");
1389 
1390   auto *Var = Capture.getVariable();
1391   SourceLocation Loc =
1392       IsImplicitCapture ? ImplicitCaptureLoc : Capture.getLocation();
1393 
1394   // C++11 [expr.prim.lambda]p21:
1395   //   When the lambda-expression is evaluated, the entities that
1396   //   are captured by copy are used to direct-initialize each
1397   //   corresponding non-static data member of the resulting closure
1398   //   object. (For array members, the array elements are
1399   //   direct-initialized in increasing subscript order.) These
1400   //   initializations are performed in the (unspecified) order in
1401   //   which the non-static data members are declared.
1402 
1403   // C++ [expr.prim.lambda]p12:
1404   //   An entity captured by a lambda-expression is odr-used (3.2) in
1405   //   the scope containing the lambda-expression.
1406   ExprResult RefResult = S.BuildDeclarationNameExpr(
1407       CXXScopeSpec(), DeclarationNameInfo(Var->getDeclName(), Loc), Var);
1408   if (RefResult.isInvalid())
1409     return ExprError();
1410   Expr *Ref = RefResult.get();
1411 
1412   auto Entity = InitializedEntity::InitializeLambdaCapture(
1413       Var->getIdentifier(), Field->getType(), Loc);
1414   InitializationKind InitKind = InitializationKind::CreateDirect(Loc, Loc, Loc);
1415   InitializationSequence Init(S, Entity, InitKind, Ref);
1416   return Init.Perform(S, Entity, InitKind, Ref);
1417 }
1418 
1419 ExprResult Sema::ActOnLambdaExpr(SourceLocation StartLoc, Stmt *Body,
1420                                  Scope *CurScope) {
1421   LambdaScopeInfo LSI = *cast<LambdaScopeInfo>(FunctionScopes.back());
1422   ActOnFinishFunctionBody(LSI.CallOperator, Body);
1423   return BuildLambdaExpr(StartLoc, Body->getEndLoc(), &LSI);
1424 }
1425 
1426 static LambdaCaptureDefault
1427 mapImplicitCaptureStyle(CapturingScopeInfo::ImplicitCaptureStyle ICS) {
1428   switch (ICS) {
1429   case CapturingScopeInfo::ImpCap_None:
1430     return LCD_None;
1431   case CapturingScopeInfo::ImpCap_LambdaByval:
1432     return LCD_ByCopy;
1433   case CapturingScopeInfo::ImpCap_CapturedRegion:
1434   case CapturingScopeInfo::ImpCap_LambdaByref:
1435     return LCD_ByRef;
1436   case CapturingScopeInfo::ImpCap_Block:
1437     llvm_unreachable("block capture in lambda");
1438   }
1439   llvm_unreachable("Unknown implicit capture style");
1440 }
1441 
1442 bool Sema::CaptureHasSideEffects(const Capture &From) {
1443   if (!From.isVLATypeCapture()) {
1444     Expr *Init = From.getInitExpr();
1445     if (Init && Init->HasSideEffects(Context))
1446       return true;
1447   }
1448 
1449   if (!From.isCopyCapture())
1450     return false;
1451 
1452   const QualType T = From.isThisCapture()
1453                          ? getCurrentThisType()->getPointeeType()
1454                          : From.getCaptureType();
1455 
1456   if (T.isVolatileQualified())
1457     return true;
1458 
1459   const Type *BaseT = T->getBaseElementTypeUnsafe();
1460   if (const CXXRecordDecl *RD = BaseT->getAsCXXRecordDecl())
1461     return !RD->isCompleteDefinition() || !RD->hasTrivialCopyConstructor() ||
1462            !RD->hasTrivialDestructor();
1463 
1464   return false;
1465 }
1466 
1467 bool Sema::DiagnoseUnusedLambdaCapture(SourceRange CaptureRange,
1468                                        const Capture &From) {
1469   if (CaptureHasSideEffects(From))
1470     return false;
1471 
1472   if (From.isVLATypeCapture())
1473     return false;
1474 
1475   auto diag = Diag(From.getLocation(), diag::warn_unused_lambda_capture);
1476   if (From.isThisCapture())
1477     diag << "'this'";
1478   else
1479     diag << From.getVariable();
1480   diag << From.isNonODRUsed();
1481   diag << FixItHint::CreateRemoval(CaptureRange);
1482   return true;
1483 }
1484 
1485 ExprResult Sema::BuildLambdaExpr(SourceLocation StartLoc, SourceLocation EndLoc,
1486                                  LambdaScopeInfo *LSI) {
1487   // Collect information from the lambda scope.
1488   SmallVector<LambdaCapture, 4> Captures;
1489   SmallVector<Expr *, 4> CaptureInits;
1490   SourceLocation CaptureDefaultLoc = LSI->CaptureDefaultLoc;
1491   LambdaCaptureDefault CaptureDefault =
1492       mapImplicitCaptureStyle(LSI->ImpCaptureStyle);
1493   CXXRecordDecl *Class;
1494   CXXMethodDecl *CallOperator;
1495   SourceRange IntroducerRange;
1496   bool ExplicitParams;
1497   bool ExplicitResultType;
1498   CleanupInfo LambdaCleanup;
1499   bool ContainsUnexpandedParameterPack;
1500   bool IsGenericLambda;
1501   {
1502     CallOperator = LSI->CallOperator;
1503     Class = LSI->Lambda;
1504     IntroducerRange = LSI->IntroducerRange;
1505     ExplicitParams = LSI->ExplicitParams;
1506     ExplicitResultType = !LSI->HasImplicitReturnType;
1507     LambdaCleanup = LSI->Cleanup;
1508     ContainsUnexpandedParameterPack = LSI->ContainsUnexpandedParameterPack;
1509     IsGenericLambda = Class->isGenericLambda();
1510 
1511     CallOperator->setLexicalDeclContext(Class);
1512     Decl *TemplateOrNonTemplateCallOperatorDecl =
1513         CallOperator->getDescribedFunctionTemplate()
1514         ? CallOperator->getDescribedFunctionTemplate()
1515         : cast<Decl>(CallOperator);
1516 
1517     TemplateOrNonTemplateCallOperatorDecl->setLexicalDeclContext(Class);
1518     Class->addDecl(TemplateOrNonTemplateCallOperatorDecl);
1519 
1520     PopExpressionEvaluationContext();
1521 
1522     // Translate captures.
1523     auto CurField = Class->field_begin();
1524     // True if the current capture has a used capture or default before it.
1525     bool CurHasPreviousCapture = CaptureDefault != LCD_None;
1526     SourceLocation PrevCaptureLoc = CurHasPreviousCapture ?
1527         CaptureDefaultLoc : IntroducerRange.getBegin();
1528 
1529     for (unsigned I = 0, N = LSI->Captures.size(); I != N; ++I, ++CurField) {
1530       const Capture &From = LSI->Captures[I];
1531 
1532       assert(!From.isBlockCapture() && "Cannot capture __block variables");
1533       bool IsImplicit = I >= LSI->NumExplicitCaptures;
1534 
1535       // Use source ranges of explicit captures for fixits where available.
1536       SourceRange CaptureRange = LSI->ExplicitCaptureRanges[I];
1537 
1538       // Warn about unused explicit captures.
1539       bool IsCaptureUsed = true;
1540       if (!CurContext->isDependentContext() && !IsImplicit && !From.isODRUsed()) {
1541         // Initialized captures that are non-ODR used may not be eliminated.
1542         bool NonODRUsedInitCapture =
1543             IsGenericLambda && From.isNonODRUsed() && From.getInitExpr();
1544         if (!NonODRUsedInitCapture) {
1545           bool IsLast = (I + 1) == LSI->NumExplicitCaptures;
1546           SourceRange FixItRange;
1547           if (CaptureRange.isValid()) {
1548             if (!CurHasPreviousCapture && !IsLast) {
1549               // If there are no captures preceding this capture, remove the
1550               // following comma.
1551               FixItRange = SourceRange(CaptureRange.getBegin(),
1552                                        getLocForEndOfToken(CaptureRange.getEnd()));
1553             } else {
1554               // Otherwise, remove the comma since the last used capture.
1555               FixItRange = SourceRange(getLocForEndOfToken(PrevCaptureLoc),
1556                                        CaptureRange.getEnd());
1557             }
1558           }
1559 
1560           IsCaptureUsed = !DiagnoseUnusedLambdaCapture(FixItRange, From);
1561         }
1562       }
1563 
1564       if (CaptureRange.isValid()) {
1565         CurHasPreviousCapture |= IsCaptureUsed;
1566         PrevCaptureLoc = CaptureRange.getEnd();
1567       }
1568 
1569       // Handle 'this' capture.
1570       if (From.isThisCapture()) {
1571         // Capturing 'this' implicitly with a default of '[=]' is deprecated,
1572         // because it results in a reference capture. Don't warn prior to
1573         // C++2a; there's nothing that can be done about it before then.
1574         if (getLangOpts().CPlusPlus2a && IsImplicit &&
1575             CaptureDefault == LCD_ByCopy) {
1576           Diag(From.getLocation(), diag::warn_deprecated_this_capture);
1577           Diag(CaptureDefaultLoc, diag::note_deprecated_this_capture)
1578               << FixItHint::CreateInsertion(
1579                      getLocForEndOfToken(CaptureDefaultLoc), ", this");
1580         }
1581 
1582         Captures.push_back(
1583             LambdaCapture(From.getLocation(), IsImplicit,
1584                           From.isCopyCapture() ? LCK_StarThis : LCK_This));
1585         CaptureInits.push_back(From.getInitExpr());
1586         continue;
1587       }
1588       if (From.isVLATypeCapture()) {
1589         Captures.push_back(
1590             LambdaCapture(From.getLocation(), IsImplicit, LCK_VLAType));
1591         CaptureInits.push_back(nullptr);
1592         continue;
1593       }
1594 
1595       VarDecl *Var = From.getVariable();
1596       LambdaCaptureKind Kind = From.isCopyCapture() ? LCK_ByCopy : LCK_ByRef;
1597       Captures.push_back(LambdaCapture(From.getLocation(), IsImplicit, Kind,
1598                                        Var, From.getEllipsisLoc()));
1599       Expr *Init = From.getInitExpr();
1600       if (!Init) {
1601         auto InitResult = performLambdaVarCaptureInitialization(
1602             *this, From, *CurField, CaptureDefaultLoc, IsImplicit);
1603         if (InitResult.isInvalid())
1604           return ExprError();
1605         Init = InitResult.get();
1606       }
1607       CaptureInits.push_back(Init);
1608     }
1609 
1610     // C++11 [expr.prim.lambda]p6:
1611     //   The closure type for a lambda-expression with no lambda-capture
1612     //   has a public non-virtual non-explicit const conversion function
1613     //   to pointer to function having the same parameter and return
1614     //   types as the closure type's function call operator.
1615     if (Captures.empty() && CaptureDefault == LCD_None)
1616       addFunctionPointerConversion(*this, IntroducerRange, Class,
1617                                    CallOperator);
1618 
1619     // Objective-C++:
1620     //   The closure type for a lambda-expression has a public non-virtual
1621     //   non-explicit const conversion function to a block pointer having the
1622     //   same parameter and return types as the closure type's function call
1623     //   operator.
1624     // FIXME: Fix generic lambda to block conversions.
1625     if (getLangOpts().Blocks && getLangOpts().ObjC1 && !IsGenericLambda)
1626       addBlockPointerConversion(*this, IntroducerRange, Class, CallOperator);
1627 
1628     // Finalize the lambda class.
1629     SmallVector<Decl*, 4> Fields(Class->fields());
1630     ActOnFields(nullptr, Class->getLocation(), Class, Fields, SourceLocation(),
1631                 SourceLocation(), ParsedAttributesView());
1632     CheckCompletedCXXClass(Class);
1633   }
1634 
1635   Cleanup.mergeFrom(LambdaCleanup);
1636 
1637   LambdaExpr *Lambda = LambdaExpr::Create(Context, Class, IntroducerRange,
1638                                           CaptureDefault, CaptureDefaultLoc,
1639                                           Captures,
1640                                           ExplicitParams, ExplicitResultType,
1641                                           CaptureInits, EndLoc,
1642                                           ContainsUnexpandedParameterPack);
1643   // If the lambda expression's call operator is not explicitly marked constexpr
1644   // and we are not in a dependent context, analyze the call operator to infer
1645   // its constexpr-ness, suppressing diagnostics while doing so.
1646   if (getLangOpts().CPlusPlus17 && !CallOperator->isInvalidDecl() &&
1647       !CallOperator->isConstexpr() &&
1648       !isa<CoroutineBodyStmt>(CallOperator->getBody()) &&
1649       !Class->getDeclContext()->isDependentContext()) {
1650     TentativeAnalysisScope DiagnosticScopeGuard(*this);
1651     CallOperator->setConstexpr(
1652         CheckConstexprFunctionDecl(CallOperator) &&
1653         CheckConstexprFunctionBody(CallOperator, CallOperator->getBody()));
1654   }
1655 
1656   // Emit delayed shadowing warnings now that the full capture list is known.
1657   DiagnoseShadowingLambdaDecls(LSI);
1658 
1659   if (!CurContext->isDependentContext()) {
1660     switch (ExprEvalContexts.back().Context) {
1661     // C++11 [expr.prim.lambda]p2:
1662     //   A lambda-expression shall not appear in an unevaluated operand
1663     //   (Clause 5).
1664     case ExpressionEvaluationContext::Unevaluated:
1665     case ExpressionEvaluationContext::UnevaluatedList:
1666     case ExpressionEvaluationContext::UnevaluatedAbstract:
1667     // C++1y [expr.const]p2:
1668     //   A conditional-expression e is a core constant expression unless the
1669     //   evaluation of e, following the rules of the abstract machine, would
1670     //   evaluate [...] a lambda-expression.
1671     //
1672     // This is technically incorrect, there are some constant evaluated contexts
1673     // where this should be allowed.  We should probably fix this when DR1607 is
1674     // ratified, it lays out the exact set of conditions where we shouldn't
1675     // allow a lambda-expression.
1676     case ExpressionEvaluationContext::ConstantEvaluated:
1677       // We don't actually diagnose this case immediately, because we
1678       // could be within a context where we might find out later that
1679       // the expression is potentially evaluated (e.g., for typeid).
1680       ExprEvalContexts.back().Lambdas.push_back(Lambda);
1681       break;
1682 
1683     case ExpressionEvaluationContext::DiscardedStatement:
1684     case ExpressionEvaluationContext::PotentiallyEvaluated:
1685     case ExpressionEvaluationContext::PotentiallyEvaluatedIfUsed:
1686       break;
1687     }
1688   }
1689 
1690   return MaybeBindToTemporary(Lambda);
1691 }
1692 
1693 ExprResult Sema::BuildBlockForLambdaConversion(SourceLocation CurrentLocation,
1694                                                SourceLocation ConvLocation,
1695                                                CXXConversionDecl *Conv,
1696                                                Expr *Src) {
1697   // Make sure that the lambda call operator is marked used.
1698   CXXRecordDecl *Lambda = Conv->getParent();
1699   CXXMethodDecl *CallOperator
1700     = cast<CXXMethodDecl>(
1701         Lambda->lookup(
1702           Context.DeclarationNames.getCXXOperatorName(OO_Call)).front());
1703   CallOperator->setReferenced();
1704   CallOperator->markUsed(Context);
1705 
1706   ExprResult Init = PerformCopyInitialization(
1707       InitializedEntity::InitializeLambdaToBlock(ConvLocation, Src->getType(),
1708                                                  /*NRVO=*/false),
1709       CurrentLocation, Src);
1710   if (!Init.isInvalid())
1711     Init = ActOnFinishFullExpr(Init.get());
1712 
1713   if (Init.isInvalid())
1714     return ExprError();
1715 
1716   // Create the new block to be returned.
1717   BlockDecl *Block = BlockDecl::Create(Context, CurContext, ConvLocation);
1718 
1719   // Set the type information.
1720   Block->setSignatureAsWritten(CallOperator->getTypeSourceInfo());
1721   Block->setIsVariadic(CallOperator->isVariadic());
1722   Block->setBlockMissingReturnType(false);
1723 
1724   // Add parameters.
1725   SmallVector<ParmVarDecl *, 4> BlockParams;
1726   for (unsigned I = 0, N = CallOperator->getNumParams(); I != N; ++I) {
1727     ParmVarDecl *From = CallOperator->getParamDecl(I);
1728     BlockParams.push_back(ParmVarDecl::Create(
1729         Context, Block, From->getBeginLoc(), From->getLocation(),
1730         From->getIdentifier(), From->getType(), From->getTypeSourceInfo(),
1731         From->getStorageClass(),
1732         /*DefaultArg=*/nullptr));
1733   }
1734   Block->setParams(BlockParams);
1735 
1736   Block->setIsConversionFromLambda(true);
1737 
1738   // Add capture. The capture uses a fake variable, which doesn't correspond
1739   // to any actual memory location. However, the initializer copy-initializes
1740   // the lambda object.
1741   TypeSourceInfo *CapVarTSI =
1742       Context.getTrivialTypeSourceInfo(Src->getType());
1743   VarDecl *CapVar = VarDecl::Create(Context, Block, ConvLocation,
1744                                     ConvLocation, nullptr,
1745                                     Src->getType(), CapVarTSI,
1746                                     SC_None);
1747   BlockDecl::Capture Capture(/*Variable=*/CapVar, /*ByRef=*/false,
1748                              /*Nested=*/false, /*Copy=*/Init.get());
1749   Block->setCaptures(Context, Capture, /*CapturesCXXThis=*/false);
1750 
1751   // Add a fake function body to the block. IR generation is responsible
1752   // for filling in the actual body, which cannot be expressed as an AST.
1753   Block->setBody(new (Context) CompoundStmt(ConvLocation));
1754 
1755   // Create the block literal expression.
1756   Expr *BuildBlock = new (Context) BlockExpr(Block, Conv->getConversionType());
1757   ExprCleanupObjects.push_back(Block);
1758   Cleanup.setExprNeedsCleanups(true);
1759 
1760   return BuildBlock;
1761 }
1762