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