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   Init = Result.getAs<Expr>();
779 
780   // The init-capture initialization is a full-expression that must be
781   // processed as one before we enter the declcontext of the lambda's
782   // call-operator.
783   Result = ActOnFinishFullExpr(Init, Loc, /*DiscardedValue*/ false,
784                                /*IsConstexpr*/ false,
785                                /*IsLambdaInitCaptureInitializer*/ true);
786   if (Result.isInvalid())
787     return QualType();
788 
789   Init = Result.getAs<Expr>();
790   return DeducedType;
791 }
792 
793 VarDecl *Sema::createLambdaInitCaptureVarDecl(SourceLocation Loc,
794                                               QualType InitCaptureType,
795                                               IdentifierInfo *Id,
796                                               unsigned InitStyle, Expr *Init) {
797   TypeSourceInfo *TSI = Context.getTrivialTypeSourceInfo(InitCaptureType,
798       Loc);
799   // Create a dummy variable representing the init-capture. This is not actually
800   // used as a variable, and only exists as a way to name and refer to the
801   // init-capture.
802   // FIXME: Pass in separate source locations for '&' and identifier.
803   VarDecl *NewVD = VarDecl::Create(Context, CurContext, Loc,
804                                    Loc, Id, InitCaptureType, TSI, SC_Auto);
805   NewVD->setInitCapture(true);
806   NewVD->setReferenced(true);
807   // FIXME: Pass in a VarDecl::InitializationStyle.
808   NewVD->setInitStyle(static_cast<VarDecl::InitializationStyle>(InitStyle));
809   NewVD->markUsed(Context);
810   NewVD->setInit(Init);
811   return NewVD;
812 }
813 
814 FieldDecl *Sema::buildInitCaptureField(LambdaScopeInfo *LSI, VarDecl *Var) {
815   FieldDecl *Field = FieldDecl::Create(
816       Context, LSI->Lambda, Var->getLocation(), Var->getLocation(),
817       nullptr, Var->getType(), Var->getTypeSourceInfo(), nullptr, false,
818       ICIS_NoInit);
819   Field->setImplicit(true);
820   Field->setAccess(AS_private);
821   LSI->Lambda->addDecl(Field);
822 
823   LSI->addCapture(Var, /*isBlock*/false, Var->getType()->isReferenceType(),
824                   /*isNested*/false, Var->getLocation(), SourceLocation(),
825                   Var->getType(), Var->getInit());
826   return Field;
827 }
828 
829 void Sema::ActOnStartOfLambdaDefinition(LambdaIntroducer &Intro,
830                                         Declarator &ParamInfo,
831                                         Scope *CurScope) {
832   // Determine if we're within a context where we know that the lambda will
833   // be dependent, because there are template parameters in scope.
834   bool KnownDependent = false;
835   LambdaScopeInfo *const LSI = getCurLambda();
836   assert(LSI && "LambdaScopeInfo should be on stack!");
837 
838   // The lambda-expression's closure type might be dependent even if its
839   // semantic context isn't, if it appears within a default argument of a
840   // function template.
841   if (CurScope->getTemplateParamParent())
842     KnownDependent = true;
843 
844   // Determine the signature of the call operator.
845   TypeSourceInfo *MethodTyInfo;
846   bool ExplicitParams = true;
847   bool ExplicitResultType = true;
848   bool ContainsUnexpandedParameterPack = false;
849   SourceLocation EndLoc;
850   SmallVector<ParmVarDecl *, 8> Params;
851   if (ParamInfo.getNumTypeObjects() == 0) {
852     // C++11 [expr.prim.lambda]p4:
853     //   If a lambda-expression does not include a lambda-declarator, it is as
854     //   if the lambda-declarator were ().
855     FunctionProtoType::ExtProtoInfo EPI(Context.getDefaultCallingConvention(
856         /*IsVariadic=*/false, /*IsCXXMethod=*/true));
857     EPI.HasTrailingReturn = true;
858     EPI.TypeQuals |= DeclSpec::TQ_const;
859     // C++1y [expr.prim.lambda]:
860     //   The lambda return type is 'auto', which is replaced by the
861     //   trailing-return type if provided and/or deduced from 'return'
862     //   statements
863     // We don't do this before C++1y, because we don't support deduced return
864     // types there.
865     QualType DefaultTypeForNoTrailingReturn =
866         getLangOpts().CPlusPlus14 ? Context.getAutoDeductType()
867                                   : Context.DependentTy;
868     QualType MethodTy =
869         Context.getFunctionType(DefaultTypeForNoTrailingReturn, None, EPI);
870     MethodTyInfo = Context.getTrivialTypeSourceInfo(MethodTy);
871     ExplicitParams = false;
872     ExplicitResultType = false;
873     EndLoc = Intro.Range.getEnd();
874   } else {
875     assert(ParamInfo.isFunctionDeclarator() &&
876            "lambda-declarator is a function");
877     DeclaratorChunk::FunctionTypeInfo &FTI = ParamInfo.getFunctionTypeInfo();
878 
879     // C++11 [expr.prim.lambda]p5:
880     //   This function call operator is declared const (9.3.1) if and only if
881     //   the lambda-expression's parameter-declaration-clause is not followed
882     //   by mutable. It is neither virtual nor declared volatile. [...]
883     if (!FTI.hasMutableQualifier())
884       FTI.TypeQuals |= DeclSpec::TQ_const;
885 
886     MethodTyInfo = GetTypeForDeclarator(ParamInfo, CurScope);
887     assert(MethodTyInfo && "no type from lambda-declarator");
888     EndLoc = ParamInfo.getSourceRange().getEnd();
889 
890     ExplicitResultType = FTI.hasTrailingReturnType();
891 
892     if (FTIHasNonVoidParameters(FTI)) {
893       Params.reserve(FTI.NumParams);
894       for (unsigned i = 0, e = FTI.NumParams; i != e; ++i)
895         Params.push_back(cast<ParmVarDecl>(FTI.Params[i].Param));
896     }
897 
898     // Check for unexpanded parameter packs in the method type.
899     if (MethodTyInfo->getType()->containsUnexpandedParameterPack())
900       ContainsUnexpandedParameterPack = true;
901   }
902 
903   CXXRecordDecl *Class = createLambdaClosureType(Intro.Range, MethodTyInfo,
904                                                  KnownDependent, Intro.Default);
905 
906   CXXMethodDecl *Method =
907       startLambdaDefinition(Class, Intro.Range, MethodTyInfo, EndLoc, Params,
908                             ParamInfo.getDeclSpec().isConstexprSpecified());
909   if (ExplicitParams)
910     CheckCXXDefaultArguments(Method);
911 
912   // This represents the function body for the lambda function, check if we
913   // have to apply optnone due to a pragma.
914   AddRangeBasedOptnone(Method);
915 
916   // code_seg attribute on lambda apply to the method.
917   if (Attr *A = getImplicitCodeSegOrSectionAttrForFunction(Method, /*IsDefinition=*/true))
918     Method->addAttr(A);
919 
920   // Attributes on the lambda apply to the method.
921   ProcessDeclAttributes(CurScope, Method, ParamInfo);
922 
923   // CUDA lambdas get implicit attributes based on the scope in which they're
924   // declared.
925   if (getLangOpts().CUDA)
926     CUDASetLambdaAttrs(Method);
927 
928   // Introduce the function call operator as the current declaration context.
929   PushDeclContext(CurScope, Method);
930 
931   // Build the lambda scope.
932   buildLambdaScope(LSI, Method, Intro.Range, Intro.Default, Intro.DefaultLoc,
933                    ExplicitParams, ExplicitResultType, !Method->isConst());
934 
935   // C++11 [expr.prim.lambda]p9:
936   //   A lambda-expression whose smallest enclosing scope is a block scope is a
937   //   local lambda expression; any other lambda expression shall not have a
938   //   capture-default or simple-capture in its lambda-introducer.
939   //
940   // For simple-captures, this is covered by the check below that any named
941   // entity is a variable that can be captured.
942   //
943   // For DR1632, we also allow a capture-default in any context where we can
944   // odr-use 'this' (in particular, in a default initializer for a non-static
945   // data member).
946   if (Intro.Default != LCD_None && !Class->getParent()->isFunctionOrMethod() &&
947       (getCurrentThisType().isNull() ||
948        CheckCXXThisCapture(SourceLocation(), /*Explicit*/true,
949                            /*BuildAndDiagnose*/false)))
950     Diag(Intro.DefaultLoc, diag::err_capture_default_non_local);
951 
952   // Distinct capture names, for diagnostics.
953   llvm::SmallSet<IdentifierInfo*, 8> CaptureNames;
954 
955   // Handle explicit captures.
956   SourceLocation PrevCaptureLoc
957     = Intro.Default == LCD_None? Intro.Range.getBegin() : Intro.DefaultLoc;
958   for (auto C = Intro.Captures.begin(), E = Intro.Captures.end(); C != E;
959        PrevCaptureLoc = C->Loc, ++C) {
960     if (C->Kind == LCK_This || C->Kind == LCK_StarThis) {
961       if (C->Kind == LCK_StarThis)
962         Diag(C->Loc, !getLangOpts().CPlusPlus17
963                              ? diag::ext_star_this_lambda_capture_cxx17
964                              : diag::warn_cxx14_compat_star_this_lambda_capture);
965 
966       // C++11 [expr.prim.lambda]p8:
967       //   An identifier or this shall not appear more than once in a
968       //   lambda-capture.
969       if (LSI->isCXXThisCaptured()) {
970         Diag(C->Loc, diag::err_capture_more_than_once)
971             << "'this'" << SourceRange(LSI->getCXXThisCapture().getLocation())
972             << FixItHint::CreateRemoval(
973                    SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
974         continue;
975       }
976 
977       // C++2a [expr.prim.lambda]p8:
978       //  If a lambda-capture includes a capture-default that is =,
979       //  each simple-capture of that lambda-capture shall be of the form
980       //  "&identifier", "this", or "* this". [ Note: The form [&,this] is
981       //  redundant but accepted for compatibility with ISO C++14. --end note ]
982       if (Intro.Default == LCD_ByCopy && C->Kind != LCK_StarThis)
983         Diag(C->Loc, !getLangOpts().CPlusPlus2a
984                          ? diag::ext_equals_this_lambda_capture_cxx2a
985                          : diag::warn_cxx17_compat_equals_this_lambda_capture);
986 
987       // C++11 [expr.prim.lambda]p12:
988       //   If this is captured by a local lambda expression, its nearest
989       //   enclosing function shall be a non-static member function.
990       QualType ThisCaptureType = getCurrentThisType();
991       if (ThisCaptureType.isNull()) {
992         Diag(C->Loc, diag::err_this_capture) << true;
993         continue;
994       }
995 
996       CheckCXXThisCapture(C->Loc, /*Explicit=*/true, /*BuildAndDiagnose*/ true,
997                           /*FunctionScopeIndexToStopAtPtr*/ nullptr,
998                           C->Kind == LCK_StarThis);
999       if (!LSI->Captures.empty())
1000         LSI->ExplicitCaptureRanges[LSI->Captures.size() - 1] = C->ExplicitRange;
1001       continue;
1002     }
1003 
1004     assert(C->Id && "missing identifier for capture");
1005 
1006     if (C->Init.isInvalid())
1007       continue;
1008 
1009     VarDecl *Var = nullptr;
1010     if (C->Init.isUsable()) {
1011       Diag(C->Loc, getLangOpts().CPlusPlus14
1012                        ? diag::warn_cxx11_compat_init_capture
1013                        : diag::ext_init_capture);
1014 
1015       if (C->Init.get()->containsUnexpandedParameterPack())
1016         ContainsUnexpandedParameterPack = true;
1017       // If the initializer expression is usable, but the InitCaptureType
1018       // is not, then an error has occurred - so ignore the capture for now.
1019       // for e.g., [n{0}] { }; <-- if no <initializer_list> is included.
1020       // FIXME: we should create the init capture variable and mark it invalid
1021       // in this case.
1022       if (C->InitCaptureType.get().isNull())
1023         continue;
1024 
1025       unsigned InitStyle;
1026       switch (C->InitKind) {
1027       case LambdaCaptureInitKind::NoInit:
1028         llvm_unreachable("not an init-capture?");
1029       case LambdaCaptureInitKind::CopyInit:
1030         InitStyle = VarDecl::CInit;
1031         break;
1032       case LambdaCaptureInitKind::DirectInit:
1033         InitStyle = VarDecl::CallInit;
1034         break;
1035       case LambdaCaptureInitKind::ListInit:
1036         InitStyle = VarDecl::ListInit;
1037         break;
1038       }
1039       Var = createLambdaInitCaptureVarDecl(C->Loc, C->InitCaptureType.get(),
1040                                            C->Id, InitStyle, C->Init.get());
1041       // C++1y [expr.prim.lambda]p11:
1042       //   An init-capture behaves as if it declares and explicitly
1043       //   captures a variable [...] whose declarative region is the
1044       //   lambda-expression's compound-statement
1045       if (Var)
1046         PushOnScopeChains(Var, CurScope, false);
1047     } else {
1048       assert(C->InitKind == LambdaCaptureInitKind::NoInit &&
1049              "init capture has valid but null init?");
1050 
1051       // C++11 [expr.prim.lambda]p8:
1052       //   If a lambda-capture includes a capture-default that is &, the
1053       //   identifiers in the lambda-capture shall not be preceded by &.
1054       //   If a lambda-capture includes a capture-default that is =, [...]
1055       //   each identifier it contains shall be preceded by &.
1056       if (C->Kind == LCK_ByRef && Intro.Default == LCD_ByRef) {
1057         Diag(C->Loc, diag::err_reference_capture_with_reference_default)
1058             << FixItHint::CreateRemoval(
1059                 SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
1060         continue;
1061       } else if (C->Kind == LCK_ByCopy && Intro.Default == LCD_ByCopy) {
1062         Diag(C->Loc, diag::err_copy_capture_with_copy_default)
1063             << FixItHint::CreateRemoval(
1064                 SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
1065         continue;
1066       }
1067 
1068       // C++11 [expr.prim.lambda]p10:
1069       //   The identifiers in a capture-list are looked up using the usual
1070       //   rules for unqualified name lookup (3.4.1)
1071       DeclarationNameInfo Name(C->Id, C->Loc);
1072       LookupResult R(*this, Name, LookupOrdinaryName);
1073       LookupName(R, CurScope);
1074       if (R.isAmbiguous())
1075         continue;
1076       if (R.empty()) {
1077         // FIXME: Disable corrections that would add qualification?
1078         CXXScopeSpec ScopeSpec;
1079         if (DiagnoseEmptyLookup(CurScope, ScopeSpec, R,
1080                                 llvm::make_unique<DeclFilterCCC<VarDecl>>()))
1081           continue;
1082       }
1083 
1084       Var = R.getAsSingle<VarDecl>();
1085       if (Var && DiagnoseUseOfDecl(Var, C->Loc))
1086         continue;
1087     }
1088 
1089     // C++11 [expr.prim.lambda]p8:
1090     //   An identifier or this shall not appear more than once in a
1091     //   lambda-capture.
1092     if (!CaptureNames.insert(C->Id).second) {
1093       if (Var && LSI->isCaptured(Var)) {
1094         Diag(C->Loc, diag::err_capture_more_than_once)
1095             << C->Id << SourceRange(LSI->getCapture(Var).getLocation())
1096             << FixItHint::CreateRemoval(
1097                    SourceRange(getLocForEndOfToken(PrevCaptureLoc), C->Loc));
1098       } else
1099         // Previous capture captured something different (one or both was
1100         // an init-cpature): no fixit.
1101         Diag(C->Loc, diag::err_capture_more_than_once) << C->Id;
1102       continue;
1103     }
1104 
1105     // C++11 [expr.prim.lambda]p10:
1106     //   [...] each such lookup shall find a variable with automatic storage
1107     //   duration declared in the reaching scope of the local lambda expression.
1108     // Note that the 'reaching scope' check happens in tryCaptureVariable().
1109     if (!Var) {
1110       Diag(C->Loc, diag::err_capture_does_not_name_variable) << C->Id;
1111       continue;
1112     }
1113 
1114     // Ignore invalid decls; they'll just confuse the code later.
1115     if (Var->isInvalidDecl())
1116       continue;
1117 
1118     if (!Var->hasLocalStorage()) {
1119       Diag(C->Loc, diag::err_capture_non_automatic_variable) << C->Id;
1120       Diag(Var->getLocation(), diag::note_previous_decl) << C->Id;
1121       continue;
1122     }
1123 
1124     // C++11 [expr.prim.lambda]p23:
1125     //   A capture followed by an ellipsis is a pack expansion (14.5.3).
1126     SourceLocation EllipsisLoc;
1127     if (C->EllipsisLoc.isValid()) {
1128       if (Var->isParameterPack()) {
1129         EllipsisLoc = C->EllipsisLoc;
1130       } else {
1131         Diag(C->EllipsisLoc, diag::err_pack_expansion_without_parameter_packs)
1132           << SourceRange(C->Loc);
1133 
1134         // Just ignore the ellipsis.
1135       }
1136     } else if (Var->isParameterPack()) {
1137       ContainsUnexpandedParameterPack = true;
1138     }
1139 
1140     if (C->Init.isUsable()) {
1141       buildInitCaptureField(LSI, Var);
1142     } else {
1143       TryCaptureKind Kind = C->Kind == LCK_ByRef ? TryCapture_ExplicitByRef :
1144                                                    TryCapture_ExplicitByVal;
1145       tryCaptureVariable(Var, C->Loc, Kind, EllipsisLoc);
1146     }
1147     if (!LSI->Captures.empty())
1148       LSI->ExplicitCaptureRanges[LSI->Captures.size() - 1] = C->ExplicitRange;
1149   }
1150   finishLambdaExplicitCaptures(LSI);
1151 
1152   LSI->ContainsUnexpandedParameterPack = ContainsUnexpandedParameterPack;
1153 
1154   // Add lambda parameters into scope.
1155   addLambdaParameters(Method, CurScope);
1156 
1157   // Enter a new evaluation context to insulate the lambda from any
1158   // cleanups from the enclosing full-expression.
1159   PushExpressionEvaluationContext(
1160       ExpressionEvaluationContext::PotentiallyEvaluated);
1161 }
1162 
1163 void Sema::ActOnLambdaError(SourceLocation StartLoc, Scope *CurScope,
1164                             bool IsInstantiation) {
1165   LambdaScopeInfo *LSI = cast<LambdaScopeInfo>(FunctionScopes.back());
1166 
1167   // Leave the expression-evaluation context.
1168   DiscardCleanupsInEvaluationContext();
1169   PopExpressionEvaluationContext();
1170 
1171   // Leave the context of the lambda.
1172   if (!IsInstantiation)
1173     PopDeclContext();
1174 
1175   // Finalize the lambda.
1176   CXXRecordDecl *Class = LSI->Lambda;
1177   Class->setInvalidDecl();
1178   SmallVector<Decl*, 4> Fields(Class->fields());
1179   ActOnFields(nullptr, Class->getLocation(), Class, Fields, SourceLocation(),
1180               SourceLocation(), ParsedAttributesView());
1181   CheckCompletedCXXClass(Class);
1182 
1183   PopFunctionScopeInfo();
1184 }
1185 
1186 QualType Sema::getLambdaConversionFunctionResultType(
1187     const FunctionProtoType *CallOpProto) {
1188   // The function type inside the pointer type is the same as the call
1189   // operator with some tweaks. The calling convention is the default free
1190   // function convention, and the type qualifications are lost.
1191   const FunctionProtoType::ExtProtoInfo CallOpExtInfo =
1192       CallOpProto->getExtProtoInfo();
1193   FunctionProtoType::ExtProtoInfo InvokerExtInfo = CallOpExtInfo;
1194   CallingConv CC = Context.getDefaultCallingConvention(
1195       CallOpProto->isVariadic(), /*IsCXXMethod=*/false);
1196   InvokerExtInfo.ExtInfo = InvokerExtInfo.ExtInfo.withCallingConv(CC);
1197   InvokerExtInfo.TypeQuals = 0;
1198   assert(InvokerExtInfo.RefQualifier == RQ_None &&
1199       "Lambda's call operator should not have a reference qualifier");
1200   return Context.getFunctionType(CallOpProto->getReturnType(),
1201                                  CallOpProto->getParamTypes(), InvokerExtInfo);
1202 }
1203 
1204 /// Add a lambda's conversion to function pointer, as described in
1205 /// C++11 [expr.prim.lambda]p6.
1206 static void addFunctionPointerConversion(Sema &S,
1207                                          SourceRange IntroducerRange,
1208                                          CXXRecordDecl *Class,
1209                                          CXXMethodDecl *CallOperator) {
1210   // This conversion is explicitly disabled if the lambda's function has
1211   // pass_object_size attributes on any of its parameters.
1212   auto HasPassObjectSizeAttr = [](const ParmVarDecl *P) {
1213     return P->hasAttr<PassObjectSizeAttr>();
1214   };
1215   if (llvm::any_of(CallOperator->parameters(), HasPassObjectSizeAttr))
1216     return;
1217 
1218   // Add the conversion to function pointer.
1219   QualType InvokerFunctionTy = S.getLambdaConversionFunctionResultType(
1220       CallOperator->getType()->castAs<FunctionProtoType>());
1221   QualType PtrToFunctionTy = S.Context.getPointerType(InvokerFunctionTy);
1222 
1223   // Create the type of the conversion function.
1224   FunctionProtoType::ExtProtoInfo ConvExtInfo(
1225       S.Context.getDefaultCallingConvention(
1226       /*IsVariadic=*/false, /*IsCXXMethod=*/true));
1227   // The conversion function is always const.
1228   ConvExtInfo.TypeQuals = Qualifiers::Const;
1229   QualType ConvTy =
1230       S.Context.getFunctionType(PtrToFunctionTy, None, ConvExtInfo);
1231 
1232   SourceLocation Loc = IntroducerRange.getBegin();
1233   DeclarationName ConversionName
1234     = S.Context.DeclarationNames.getCXXConversionFunctionName(
1235         S.Context.getCanonicalType(PtrToFunctionTy));
1236   DeclarationNameLoc ConvNameLoc;
1237   // Construct a TypeSourceInfo for the conversion function, and wire
1238   // all the parameters appropriately for the FunctionProtoTypeLoc
1239   // so that everything works during transformation/instantiation of
1240   // generic lambdas.
1241   // The main reason for wiring up the parameters of the conversion
1242   // function with that of the call operator is so that constructs
1243   // like the following work:
1244   // auto L = [](auto b) {                <-- 1
1245   //   return [](auto a) -> decltype(a) { <-- 2
1246   //      return a;
1247   //   };
1248   // };
1249   // int (*fp)(int) = L(5);
1250   // Because the trailing return type can contain DeclRefExprs that refer
1251   // to the original call operator's variables, we hijack the call
1252   // operators ParmVarDecls below.
1253   TypeSourceInfo *ConvNamePtrToFunctionTSI =
1254       S.Context.getTrivialTypeSourceInfo(PtrToFunctionTy, Loc);
1255   ConvNameLoc.NamedType.TInfo = ConvNamePtrToFunctionTSI;
1256 
1257   // The conversion function is a conversion to a pointer-to-function.
1258   TypeSourceInfo *ConvTSI = S.Context.getTrivialTypeSourceInfo(ConvTy, Loc);
1259   FunctionProtoTypeLoc ConvTL =
1260       ConvTSI->getTypeLoc().getAs<FunctionProtoTypeLoc>();
1261   // Get the result of the conversion function which is a pointer-to-function.
1262   PointerTypeLoc PtrToFunctionTL =
1263       ConvTL.getReturnLoc().getAs<PointerTypeLoc>();
1264   // Do the same for the TypeSourceInfo that is used to name the conversion
1265   // operator.
1266   PointerTypeLoc ConvNamePtrToFunctionTL =
1267       ConvNamePtrToFunctionTSI->getTypeLoc().getAs<PointerTypeLoc>();
1268 
1269   // Get the underlying function types that the conversion function will
1270   // be converting to (should match the type of the call operator).
1271   FunctionProtoTypeLoc CallOpConvTL =
1272       PtrToFunctionTL.getPointeeLoc().getAs<FunctionProtoTypeLoc>();
1273   FunctionProtoTypeLoc CallOpConvNameTL =
1274     ConvNamePtrToFunctionTL.getPointeeLoc().getAs<FunctionProtoTypeLoc>();
1275 
1276   // Wire up the FunctionProtoTypeLocs with the call operator's parameters.
1277   // These parameter's are essentially used to transform the name and
1278   // the type of the conversion operator.  By using the same parameters
1279   // as the call operator's we don't have to fix any back references that
1280   // the trailing return type of the call operator's uses (such as
1281   // decltype(some_type<decltype(a)>::type{} + decltype(a){}) etc.)
1282   // - we can simply use the return type of the call operator, and
1283   // everything should work.
1284   SmallVector<ParmVarDecl *, 4> InvokerParams;
1285   for (unsigned I = 0, N = CallOperator->getNumParams(); I != N; ++I) {
1286     ParmVarDecl *From = CallOperator->getParamDecl(I);
1287 
1288     InvokerParams.push_back(ParmVarDecl::Create(
1289         S.Context,
1290         // Temporarily add to the TU. This is set to the invoker below.
1291         S.Context.getTranslationUnitDecl(), From->getBeginLoc(),
1292         From->getLocation(), From->getIdentifier(), From->getType(),
1293         From->getTypeSourceInfo(), From->getStorageClass(),
1294         /*DefaultArg=*/nullptr));
1295     CallOpConvTL.setParam(I, From);
1296     CallOpConvNameTL.setParam(I, From);
1297   }
1298 
1299   CXXConversionDecl *Conversion = CXXConversionDecl::Create(
1300       S.Context, Class, Loc,
1301       DeclarationNameInfo(ConversionName, Loc, ConvNameLoc), ConvTy, ConvTSI,
1302       /*isInline=*/true, /*isExplicit=*/false,
1303       /*isConstexpr=*/S.getLangOpts().CPlusPlus17,
1304       CallOperator->getBody()->getEndLoc());
1305   Conversion->setAccess(AS_public);
1306   Conversion->setImplicit(true);
1307 
1308   if (Class->isGenericLambda()) {
1309     // Create a template version of the conversion operator, using the template
1310     // parameter list of the function call operator.
1311     FunctionTemplateDecl *TemplateCallOperator =
1312             CallOperator->getDescribedFunctionTemplate();
1313     FunctionTemplateDecl *ConversionTemplate =
1314                   FunctionTemplateDecl::Create(S.Context, Class,
1315                                       Loc, ConversionName,
1316                                       TemplateCallOperator->getTemplateParameters(),
1317                                       Conversion);
1318     ConversionTemplate->setAccess(AS_public);
1319     ConversionTemplate->setImplicit(true);
1320     Conversion->setDescribedFunctionTemplate(ConversionTemplate);
1321     Class->addDecl(ConversionTemplate);
1322   } else
1323     Class->addDecl(Conversion);
1324   // Add a non-static member function that will be the result of
1325   // the conversion with a certain unique ID.
1326   DeclarationName InvokerName = &S.Context.Idents.get(
1327                                                  getLambdaStaticInvokerName());
1328   // FIXME: Instead of passing in the CallOperator->getTypeSourceInfo()
1329   // we should get a prebuilt TrivialTypeSourceInfo from Context
1330   // using FunctionTy & Loc and get its TypeLoc as a FunctionProtoTypeLoc
1331   // then rewire the parameters accordingly, by hoisting up the InvokeParams
1332   // loop below and then use its Params to set Invoke->setParams(...) below.
1333   // This would avoid the 'const' qualifier of the calloperator from
1334   // contaminating the type of the invoker, which is currently adjusted
1335   // in SemaTemplateDeduction.cpp:DeduceTemplateArguments.  Fixing the
1336   // trailing return type of the invoker would require a visitor to rebuild
1337   // the trailing return type and adjusting all back DeclRefExpr's to refer
1338   // to the new static invoker parameters - not the call operator's.
1339   CXXMethodDecl *Invoke = CXXMethodDecl::Create(
1340       S.Context, Class, Loc, DeclarationNameInfo(InvokerName, Loc),
1341       InvokerFunctionTy, CallOperator->getTypeSourceInfo(), SC_Static,
1342       /*IsInline=*/true,
1343       /*IsConstexpr=*/false, CallOperator->getBody()->getEndLoc());
1344   for (unsigned I = 0, N = CallOperator->getNumParams(); I != N; ++I)
1345     InvokerParams[I]->setOwningFunction(Invoke);
1346   Invoke->setParams(InvokerParams);
1347   Invoke->setAccess(AS_private);
1348   Invoke->setImplicit(true);
1349   if (Class->isGenericLambda()) {
1350     FunctionTemplateDecl *TemplateCallOperator =
1351             CallOperator->getDescribedFunctionTemplate();
1352     FunctionTemplateDecl *StaticInvokerTemplate = FunctionTemplateDecl::Create(
1353                           S.Context, Class, Loc, InvokerName,
1354                           TemplateCallOperator->getTemplateParameters(),
1355                           Invoke);
1356     StaticInvokerTemplate->setAccess(AS_private);
1357     StaticInvokerTemplate->setImplicit(true);
1358     Invoke->setDescribedFunctionTemplate(StaticInvokerTemplate);
1359     Class->addDecl(StaticInvokerTemplate);
1360   } else
1361     Class->addDecl(Invoke);
1362 }
1363 
1364 /// Add a lambda's conversion to block pointer.
1365 static void addBlockPointerConversion(Sema &S,
1366                                       SourceRange IntroducerRange,
1367                                       CXXRecordDecl *Class,
1368                                       CXXMethodDecl *CallOperator) {
1369   QualType FunctionTy = S.getLambdaConversionFunctionResultType(
1370       CallOperator->getType()->castAs<FunctionProtoType>());
1371   QualType BlockPtrTy = S.Context.getBlockPointerType(FunctionTy);
1372 
1373   FunctionProtoType::ExtProtoInfo ConversionEPI(
1374       S.Context.getDefaultCallingConvention(
1375           /*IsVariadic=*/false, /*IsCXXMethod=*/true));
1376   ConversionEPI.TypeQuals = Qualifiers::Const;
1377   QualType ConvTy = S.Context.getFunctionType(BlockPtrTy, None, ConversionEPI);
1378 
1379   SourceLocation Loc = IntroducerRange.getBegin();
1380   DeclarationName Name
1381     = S.Context.DeclarationNames.getCXXConversionFunctionName(
1382         S.Context.getCanonicalType(BlockPtrTy));
1383   DeclarationNameLoc NameLoc;
1384   NameLoc.NamedType.TInfo = S.Context.getTrivialTypeSourceInfo(BlockPtrTy, Loc);
1385   CXXConversionDecl *Conversion = CXXConversionDecl::Create(
1386       S.Context, Class, Loc, DeclarationNameInfo(Name, Loc, NameLoc), ConvTy,
1387       S.Context.getTrivialTypeSourceInfo(ConvTy, Loc),
1388       /*isInline=*/true, /*isExplicit=*/false,
1389       /*isConstexpr=*/false, CallOperator->getBody()->getEndLoc());
1390   Conversion->setAccess(AS_public);
1391   Conversion->setImplicit(true);
1392   Class->addDecl(Conversion);
1393 }
1394 
1395 static ExprResult performLambdaVarCaptureInitialization(
1396     Sema &S, const Capture &Capture, FieldDecl *Field,
1397     SourceLocation ImplicitCaptureLoc, bool IsImplicitCapture) {
1398   assert(Capture.isVariableCapture() && "not a variable capture");
1399 
1400   auto *Var = Capture.getVariable();
1401   SourceLocation Loc =
1402       IsImplicitCapture ? ImplicitCaptureLoc : Capture.getLocation();
1403 
1404   // C++11 [expr.prim.lambda]p21:
1405   //   When the lambda-expression is evaluated, the entities that
1406   //   are captured by copy are used to direct-initialize each
1407   //   corresponding non-static data member of the resulting closure
1408   //   object. (For array members, the array elements are
1409   //   direct-initialized in increasing subscript order.) These
1410   //   initializations are performed in the (unspecified) order in
1411   //   which the non-static data members are declared.
1412 
1413   // C++ [expr.prim.lambda]p12:
1414   //   An entity captured by a lambda-expression is odr-used (3.2) in
1415   //   the scope containing the lambda-expression.
1416   ExprResult RefResult = S.BuildDeclarationNameExpr(
1417       CXXScopeSpec(), DeclarationNameInfo(Var->getDeclName(), Loc), Var);
1418   if (RefResult.isInvalid())
1419     return ExprError();
1420   Expr *Ref = RefResult.get();
1421 
1422   auto Entity = InitializedEntity::InitializeLambdaCapture(
1423       Var->getIdentifier(), Field->getType(), Loc);
1424   InitializationKind InitKind = InitializationKind::CreateDirect(Loc, Loc, Loc);
1425   InitializationSequence Init(S, Entity, InitKind, Ref);
1426   return Init.Perform(S, Entity, InitKind, Ref);
1427 }
1428 
1429 ExprResult Sema::ActOnLambdaExpr(SourceLocation StartLoc, Stmt *Body,
1430                                  Scope *CurScope) {
1431   LambdaScopeInfo LSI = *cast<LambdaScopeInfo>(FunctionScopes.back());
1432   ActOnFinishFunctionBody(LSI.CallOperator, Body);
1433   return BuildLambdaExpr(StartLoc, Body->getEndLoc(), &LSI);
1434 }
1435 
1436 static LambdaCaptureDefault
1437 mapImplicitCaptureStyle(CapturingScopeInfo::ImplicitCaptureStyle ICS) {
1438   switch (ICS) {
1439   case CapturingScopeInfo::ImpCap_None:
1440     return LCD_None;
1441   case CapturingScopeInfo::ImpCap_LambdaByval:
1442     return LCD_ByCopy;
1443   case CapturingScopeInfo::ImpCap_CapturedRegion:
1444   case CapturingScopeInfo::ImpCap_LambdaByref:
1445     return LCD_ByRef;
1446   case CapturingScopeInfo::ImpCap_Block:
1447     llvm_unreachable("block capture in lambda");
1448   }
1449   llvm_unreachable("Unknown implicit capture style");
1450 }
1451 
1452 bool Sema::CaptureHasSideEffects(const Capture &From) {
1453   if (!From.isVLATypeCapture()) {
1454     Expr *Init = From.getInitExpr();
1455     if (Init && Init->HasSideEffects(Context))
1456       return true;
1457   }
1458 
1459   if (!From.isCopyCapture())
1460     return false;
1461 
1462   const QualType T = From.isThisCapture()
1463                          ? getCurrentThisType()->getPointeeType()
1464                          : From.getCaptureType();
1465 
1466   if (T.isVolatileQualified())
1467     return true;
1468 
1469   const Type *BaseT = T->getBaseElementTypeUnsafe();
1470   if (const CXXRecordDecl *RD = BaseT->getAsCXXRecordDecl())
1471     return !RD->isCompleteDefinition() || !RD->hasTrivialCopyConstructor() ||
1472            !RD->hasTrivialDestructor();
1473 
1474   return false;
1475 }
1476 
1477 bool Sema::DiagnoseUnusedLambdaCapture(SourceRange CaptureRange,
1478                                        const Capture &From) {
1479   if (CaptureHasSideEffects(From))
1480     return false;
1481 
1482   if (From.isVLATypeCapture())
1483     return false;
1484 
1485   auto diag = Diag(From.getLocation(), diag::warn_unused_lambda_capture);
1486   if (From.isThisCapture())
1487     diag << "'this'";
1488   else
1489     diag << From.getVariable();
1490   diag << From.isNonODRUsed();
1491   diag << FixItHint::CreateRemoval(CaptureRange);
1492   return true;
1493 }
1494 
1495 ExprResult Sema::BuildLambdaExpr(SourceLocation StartLoc, SourceLocation EndLoc,
1496                                  LambdaScopeInfo *LSI) {
1497   // Collect information from the lambda scope.
1498   SmallVector<LambdaCapture, 4> Captures;
1499   SmallVector<Expr *, 4> CaptureInits;
1500   SourceLocation CaptureDefaultLoc = LSI->CaptureDefaultLoc;
1501   LambdaCaptureDefault CaptureDefault =
1502       mapImplicitCaptureStyle(LSI->ImpCaptureStyle);
1503   CXXRecordDecl *Class;
1504   CXXMethodDecl *CallOperator;
1505   SourceRange IntroducerRange;
1506   bool ExplicitParams;
1507   bool ExplicitResultType;
1508   CleanupInfo LambdaCleanup;
1509   bool ContainsUnexpandedParameterPack;
1510   bool IsGenericLambda;
1511   {
1512     CallOperator = LSI->CallOperator;
1513     Class = LSI->Lambda;
1514     IntroducerRange = LSI->IntroducerRange;
1515     ExplicitParams = LSI->ExplicitParams;
1516     ExplicitResultType = !LSI->HasImplicitReturnType;
1517     LambdaCleanup = LSI->Cleanup;
1518     ContainsUnexpandedParameterPack = LSI->ContainsUnexpandedParameterPack;
1519     IsGenericLambda = Class->isGenericLambda();
1520 
1521     CallOperator->setLexicalDeclContext(Class);
1522     Decl *TemplateOrNonTemplateCallOperatorDecl =
1523         CallOperator->getDescribedFunctionTemplate()
1524         ? CallOperator->getDescribedFunctionTemplate()
1525         : cast<Decl>(CallOperator);
1526 
1527     TemplateOrNonTemplateCallOperatorDecl->setLexicalDeclContext(Class);
1528     Class->addDecl(TemplateOrNonTemplateCallOperatorDecl);
1529 
1530     PopExpressionEvaluationContext();
1531 
1532     // Translate captures.
1533     auto CurField = Class->field_begin();
1534     // True if the current capture has a used capture or default before it.
1535     bool CurHasPreviousCapture = CaptureDefault != LCD_None;
1536     SourceLocation PrevCaptureLoc = CurHasPreviousCapture ?
1537         CaptureDefaultLoc : IntroducerRange.getBegin();
1538 
1539     for (unsigned I = 0, N = LSI->Captures.size(); I != N; ++I, ++CurField) {
1540       const Capture &From = LSI->Captures[I];
1541 
1542       assert(!From.isBlockCapture() && "Cannot capture __block variables");
1543       bool IsImplicit = I >= LSI->NumExplicitCaptures;
1544 
1545       // Use source ranges of explicit captures for fixits where available.
1546       SourceRange CaptureRange = LSI->ExplicitCaptureRanges[I];
1547 
1548       // Warn about unused explicit captures.
1549       bool IsCaptureUsed = true;
1550       if (!CurContext->isDependentContext() && !IsImplicit && !From.isODRUsed()) {
1551         // Initialized captures that are non-ODR used may not be eliminated.
1552         bool NonODRUsedInitCapture =
1553             IsGenericLambda && From.isNonODRUsed() && From.getInitExpr();
1554         if (!NonODRUsedInitCapture) {
1555           bool IsLast = (I + 1) == LSI->NumExplicitCaptures;
1556           SourceRange FixItRange;
1557           if (CaptureRange.isValid()) {
1558             if (!CurHasPreviousCapture && !IsLast) {
1559               // If there are no captures preceding this capture, remove the
1560               // following comma.
1561               FixItRange = SourceRange(CaptureRange.getBegin(),
1562                                        getLocForEndOfToken(CaptureRange.getEnd()));
1563             } else {
1564               // Otherwise, remove the comma since the last used capture.
1565               FixItRange = SourceRange(getLocForEndOfToken(PrevCaptureLoc),
1566                                        CaptureRange.getEnd());
1567             }
1568           }
1569 
1570           IsCaptureUsed = !DiagnoseUnusedLambdaCapture(FixItRange, From);
1571         }
1572       }
1573 
1574       if (CaptureRange.isValid()) {
1575         CurHasPreviousCapture |= IsCaptureUsed;
1576         PrevCaptureLoc = CaptureRange.getEnd();
1577       }
1578 
1579       // Handle 'this' capture.
1580       if (From.isThisCapture()) {
1581         // Capturing 'this' implicitly with a default of '[=]' is deprecated,
1582         // because it results in a reference capture. Don't warn prior to
1583         // C++2a; there's nothing that can be done about it before then.
1584         if (getLangOpts().CPlusPlus2a && IsImplicit &&
1585             CaptureDefault == LCD_ByCopy) {
1586           Diag(From.getLocation(), diag::warn_deprecated_this_capture);
1587           Diag(CaptureDefaultLoc, diag::note_deprecated_this_capture)
1588               << FixItHint::CreateInsertion(
1589                      getLocForEndOfToken(CaptureDefaultLoc), ", this");
1590         }
1591 
1592         Captures.push_back(
1593             LambdaCapture(From.getLocation(), IsImplicit,
1594                           From.isCopyCapture() ? LCK_StarThis : LCK_This));
1595         CaptureInits.push_back(From.getInitExpr());
1596         continue;
1597       }
1598       if (From.isVLATypeCapture()) {
1599         Captures.push_back(
1600             LambdaCapture(From.getLocation(), IsImplicit, LCK_VLAType));
1601         CaptureInits.push_back(nullptr);
1602         continue;
1603       }
1604 
1605       VarDecl *Var = From.getVariable();
1606       LambdaCaptureKind Kind = From.isCopyCapture() ? LCK_ByCopy : LCK_ByRef;
1607       Captures.push_back(LambdaCapture(From.getLocation(), IsImplicit, Kind,
1608                                        Var, From.getEllipsisLoc()));
1609       Expr *Init = From.getInitExpr();
1610       if (!Init) {
1611         auto InitResult = performLambdaVarCaptureInitialization(
1612             *this, From, *CurField, CaptureDefaultLoc, IsImplicit);
1613         if (InitResult.isInvalid())
1614           return ExprError();
1615         Init = InitResult.get();
1616       }
1617       CaptureInits.push_back(Init);
1618     }
1619 
1620     // C++11 [expr.prim.lambda]p6:
1621     //   The closure type for a lambda-expression with no lambda-capture
1622     //   has a public non-virtual non-explicit const conversion function
1623     //   to pointer to function having the same parameter and return
1624     //   types as the closure type's function call operator.
1625     if (Captures.empty() && CaptureDefault == LCD_None)
1626       addFunctionPointerConversion(*this, IntroducerRange, Class,
1627                                    CallOperator);
1628 
1629     // Objective-C++:
1630     //   The closure type for a lambda-expression has a public non-virtual
1631     //   non-explicit const conversion function to a block pointer having the
1632     //   same parameter and return types as the closure type's function call
1633     //   operator.
1634     // FIXME: Fix generic lambda to block conversions.
1635     if (getLangOpts().Blocks && getLangOpts().ObjC1 && !IsGenericLambda)
1636       addBlockPointerConversion(*this, IntroducerRange, Class, CallOperator);
1637 
1638     // Finalize the lambda class.
1639     SmallVector<Decl*, 4> Fields(Class->fields());
1640     ActOnFields(nullptr, Class->getLocation(), Class, Fields, SourceLocation(),
1641                 SourceLocation(), ParsedAttributesView());
1642     CheckCompletedCXXClass(Class);
1643   }
1644 
1645   Cleanup.mergeFrom(LambdaCleanup);
1646 
1647   LambdaExpr *Lambda = LambdaExpr::Create(Context, Class, IntroducerRange,
1648                                           CaptureDefault, CaptureDefaultLoc,
1649                                           Captures,
1650                                           ExplicitParams, ExplicitResultType,
1651                                           CaptureInits, EndLoc,
1652                                           ContainsUnexpandedParameterPack);
1653   // If the lambda expression's call operator is not explicitly marked constexpr
1654   // and we are not in a dependent context, analyze the call operator to infer
1655   // its constexpr-ness, suppressing diagnostics while doing so.
1656   if (getLangOpts().CPlusPlus17 && !CallOperator->isInvalidDecl() &&
1657       !CallOperator->isConstexpr() &&
1658       !isa<CoroutineBodyStmt>(CallOperator->getBody()) &&
1659       !Class->getDeclContext()->isDependentContext()) {
1660     TentativeAnalysisScope DiagnosticScopeGuard(*this);
1661     CallOperator->setConstexpr(
1662         CheckConstexprFunctionDecl(CallOperator) &&
1663         CheckConstexprFunctionBody(CallOperator, CallOperator->getBody()));
1664   }
1665 
1666   // Emit delayed shadowing warnings now that the full capture list is known.
1667   DiagnoseShadowingLambdaDecls(LSI);
1668 
1669   if (!CurContext->isDependentContext()) {
1670     switch (ExprEvalContexts.back().Context) {
1671     // C++11 [expr.prim.lambda]p2:
1672     //   A lambda-expression shall not appear in an unevaluated operand
1673     //   (Clause 5).
1674     case ExpressionEvaluationContext::Unevaluated:
1675     case ExpressionEvaluationContext::UnevaluatedList:
1676     case ExpressionEvaluationContext::UnevaluatedAbstract:
1677     // C++1y [expr.const]p2:
1678     //   A conditional-expression e is a core constant expression unless the
1679     //   evaluation of e, following the rules of the abstract machine, would
1680     //   evaluate [...] a lambda-expression.
1681     //
1682     // This is technically incorrect, there are some constant evaluated contexts
1683     // where this should be allowed.  We should probably fix this when DR1607 is
1684     // ratified, it lays out the exact set of conditions where we shouldn't
1685     // allow a lambda-expression.
1686     case ExpressionEvaluationContext::ConstantEvaluated:
1687       // We don't actually diagnose this case immediately, because we
1688       // could be within a context where we might find out later that
1689       // the expression is potentially evaluated (e.g., for typeid).
1690       ExprEvalContexts.back().Lambdas.push_back(Lambda);
1691       break;
1692 
1693     case ExpressionEvaluationContext::DiscardedStatement:
1694     case ExpressionEvaluationContext::PotentiallyEvaluated:
1695     case ExpressionEvaluationContext::PotentiallyEvaluatedIfUsed:
1696       break;
1697     }
1698   }
1699 
1700   return MaybeBindToTemporary(Lambda);
1701 }
1702 
1703 ExprResult Sema::BuildBlockForLambdaConversion(SourceLocation CurrentLocation,
1704                                                SourceLocation ConvLocation,
1705                                                CXXConversionDecl *Conv,
1706                                                Expr *Src) {
1707   // Make sure that the lambda call operator is marked used.
1708   CXXRecordDecl *Lambda = Conv->getParent();
1709   CXXMethodDecl *CallOperator
1710     = cast<CXXMethodDecl>(
1711         Lambda->lookup(
1712           Context.DeclarationNames.getCXXOperatorName(OO_Call)).front());
1713   CallOperator->setReferenced();
1714   CallOperator->markUsed(Context);
1715 
1716   ExprResult Init = PerformCopyInitialization(
1717       InitializedEntity::InitializeLambdaToBlock(ConvLocation, Src->getType(),
1718                                                  /*NRVO=*/false),
1719       CurrentLocation, Src);
1720   if (!Init.isInvalid())
1721     Init = ActOnFinishFullExpr(Init.get());
1722 
1723   if (Init.isInvalid())
1724     return ExprError();
1725 
1726   // Create the new block to be returned.
1727   BlockDecl *Block = BlockDecl::Create(Context, CurContext, ConvLocation);
1728 
1729   // Set the type information.
1730   Block->setSignatureAsWritten(CallOperator->getTypeSourceInfo());
1731   Block->setIsVariadic(CallOperator->isVariadic());
1732   Block->setBlockMissingReturnType(false);
1733 
1734   // Add parameters.
1735   SmallVector<ParmVarDecl *, 4> BlockParams;
1736   for (unsigned I = 0, N = CallOperator->getNumParams(); I != N; ++I) {
1737     ParmVarDecl *From = CallOperator->getParamDecl(I);
1738     BlockParams.push_back(ParmVarDecl::Create(
1739         Context, Block, From->getBeginLoc(), From->getLocation(),
1740         From->getIdentifier(), From->getType(), From->getTypeSourceInfo(),
1741         From->getStorageClass(),
1742         /*DefaultArg=*/nullptr));
1743   }
1744   Block->setParams(BlockParams);
1745 
1746   Block->setIsConversionFromLambda(true);
1747 
1748   // Add capture. The capture uses a fake variable, which doesn't correspond
1749   // to any actual memory location. However, the initializer copy-initializes
1750   // the lambda object.
1751   TypeSourceInfo *CapVarTSI =
1752       Context.getTrivialTypeSourceInfo(Src->getType());
1753   VarDecl *CapVar = VarDecl::Create(Context, Block, ConvLocation,
1754                                     ConvLocation, nullptr,
1755                                     Src->getType(), CapVarTSI,
1756                                     SC_None);
1757   BlockDecl::Capture Capture(/*Variable=*/CapVar, /*ByRef=*/false,
1758                              /*Nested=*/false, /*Copy=*/Init.get());
1759   Block->setCaptures(Context, Capture, /*CapturesCXXThis=*/false);
1760 
1761   // Add a fake function body to the block. IR generation is responsible
1762   // for filling in the actual body, which cannot be expressed as an AST.
1763   Block->setBody(new (Context) CompoundStmt(ConvLocation));
1764 
1765   // Create the block literal expression.
1766   Expr *BuildBlock = new (Context) BlockExpr(Block, Conv->getConversionType());
1767   ExprCleanupObjects.push_back(Block);
1768   Cleanup.setExprNeedsCleanups(true);
1769 
1770   return BuildBlock;
1771 }
1772