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